Skip to content
Snippets Groups Projects

Generate flat-resolution FITS files asynchronously

All threads resolved!
3 files
+ 37
28
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -288,6 +288,8 @@ def handle_posterior_samples(alert):
filename = alert['data']['filename']
info = '{} {}'.format(alert['data']['comment'], filename)
prefix, _ = filename.rsplit('.posterior_samples.')
skymap_filename = f'{prefix}.multiorder.fits'
labels = ['pe', 'sky_loc', 'public']
(
gracedb.download.si(filename, superevent_id)
@@ -296,22 +298,13 @@ def handle_posterior_samples(alert):
|
group(
skymaps.annotate_fits.s(
'{}.fits.gz'.format(prefix),
superevent_id, ['pe', 'sky_loc', 'public']
skymap_filename, superevent_id, labels
),
gracedb.upload.s(
'{}.multiorder.fits'.format(prefix), superevent_id,
skymap_filename, superevent_id,
'Multiresolution fits file generated from "{}"'.format(info),
['pe', 'sky_loc', 'public']
),
skymaps.flatten.s('{}.fits.gz'.format(prefix))
|
gracedb.upload.s(
'{}.fits.gz'.format(prefix), superevent_id,
'Flat-resolution fits file created from "{}"'.format(info),
['pe', 'sky_loc', 'public']
labels
)
)
).delay()
@@ -497,13 +490,11 @@ def preliminary_alert(event, superevent_id, annotation_prefix='',
gracedb.download.si(original_skymap_filename, preferred_event_id)
|
group(
skymaps.flatten.s(annotation_prefix + skymap_filename)
|
gracedb.upload.s(
annotation_prefix + skymap_filename,
annotation_prefix + original_skymap_filename,
superevent_id,
message='Flattened from multiresolution file {}'.format(
original_skymap_filename),
message='Localization copied from {}'.format(
preferred_event_id),
tags=['sky_loc'] if annotation_prefix else [
'sky_loc', 'public']
)
@@ -512,15 +503,6 @@ def preliminary_alert(event, superevent_id, annotation_prefix='',
'SKYMAP_READY', superevent_id
),
gracedb.upload.s(
annotation_prefix + original_skymap_filename,
superevent_id,
message='Localization copied from {}'.format(
preferred_event_id),
tags=['sky_loc'] if annotation_prefix else [
'sky_loc', 'public']
),
skymaps.annotate_fits.s(
annotation_prefix + skymap_filename,
superevent_id,
Loading