gwcelery/tasks/rrt_utils.py

 import json
 
+import numpy as np
 from ligo.skymap.io import read_sky_map
 from ligo.skymap.postprocess.crossmatch import crossmatch
 
@@ -32,12 +33,25 @@ def check_high_profile(skymap, em_bright,
        far_list_sorted[0]["search"] != "BBH":
         gracedb.create_label.si(
             'HIGH_PROFILE', superevent_id).delay()
+        gracedb.upload.delay(None, None, superevent_id,
+                             'Superevent labeled <font color="red">HIGH_PROFILE</font> since '
+                             'event with lowest FAR is a Burst event.',
+                              tags=['em_follow'])
         return "Event with the lowest FAR is a Burst event. Applying label"
 
     # annotation number condition
     preferred_event = superevent['preferred_event_data']
+    if preferred_event["search"] == "SSM":
+        gracedb.create_label.si(
+                'HIGH_PROFILE', superevent_id).delay()
+        gracedb.upload.delay(None, None, superevent_id,
+                             'Superevent labeled <font color="red">HIGH_PROFILE</font> since '
+                             'preferred event is from SSM search.',
+                             tags=['em_follow'])
+        return "Preferred event is from SSM. Applying label"
     if preferred_event["group"] == "CBC":
         em_bright_dict = json.loads(em_bright)
+
         has_remnant = em_bright_dict['HasRemnant']
 
         pastro_dict = json.loads(p_astro)
@@ -50,10 +64,40 @@ def check_high_profile(skymap, em_bright,
             cl = 90
             result = crossmatch(gw_skymap, contours=[cl / 100])
             sky_area = result.contour_areas[0]
-
+            is_far_away = not (gw_skymap.meta.get('distmean', np.nan) < 2000)
         if p_terr < 0.5:
-            if (p_bns > 0.1 or p_nsbh > 0.1 or has_remnant > 0.1 or sky_area < 100):  # noqa: E501
+            if p_bns > 0.1:
+                gracedb.create_label.si(
+                    'HIGH_PROFILE', superevent_id).delay()
+                gracedb.upload.delay(None, None, superevent_id,
+                                     'Superevent labeled <font color="red">HIGH_PROFILE</font> since '
+                                     'because p_BNS > 10%.',
+                                      tags=['em_follow'])
+                return "p_BNS condition satisfied. Applying label"
+            elif p_nsbh > 0.1:
+                gracedb.create_label.si(
+                    'HIGH_PROFILE', superevent_id).delay()
+                gracedb.upload.delay(None, None, superevent_id,
+                                     'Superevent labeled <font color="red">HIGH_PROFILE</font> since '
+                                     'because p_NSBH> 10%.',
+                                      tags=['em_follow'])
+                return "p_NSBH condition satisfied. Applying label"
+            elif has_remnant > 0.1:
+                gracedb.create_label.si(
+                    'HIGH_PROFILE', superevent_id).delay()
+                gracedb.upload.delay(None, None, superevent_id,
+                                     'Superevent labeled <font color="red">HIGH_PROFILE</font> since '
+                                     'because p_HasRemnant> 10%.',
+                                      tags=['em_follow'])
+                return "p_HasRemnant condition satisfied. Applying label"
+            elif (sky_area < 100 and not is_far_away):
                 gracedb.create_label.si(
                     'HIGH_PROFILE', superevent_id).delay()
-                return "Annotations condition satisfied. Applying label"
+                gracedb.upload.delay(None, None, superevent_id,
+                                     'Superevent labeled <font color="red">HIGH_PROFILE</font> since '
+                                     'because area of 90% confidence level '
+                                     'in the skymap is < 100 sq.deg and mean '
+                                     'posterior distance is < 2 Gpc. ',
+                                      tags=['em_follow'])
+                return "Skymap & distance condition satisfied. Applying label"
     return "No conditions satisfied. Skipping"

gwcelery/tests/test_tasks_rrt_utils.py

@@ -16,11 +16,12 @@ skymap_large = read_binary(data, 'MS220722v_bayestar.multiorder.fits')
 # 3. p_BNS check        ---> True
 # 4. p_NSBH check       ---> True
 # 5. skymap check       ---> True
-# 6. Simple BBH         ---> False
-# 7. Burst but low far  ---> False
-# 8. High p-terrestrial ---> False
-# 9. Burst BBH low far ---> False
-# 10. HIGH_PROFILE_applied-> False
+# 6. SSM preferred      ---> True
+# 7. Simple BBH         ---> False
+# 8. Burst but low far  ---> False
+# 9. High p-terrestrial ---> False
+# 10. Burst BBH low far ---> False
+# 11. HIGH_PROFILE_applied-> False
 
 
 def get_event(graceid):
@@ -60,6 +61,15 @@ def get_event(graceid):
             'far': 1e-10,
             'gpstime': 1234,
         }
+    elif graceid == 'G468':
+        event = {
+            'group': 'CBC',
+            'pipeline': 'gstlal',
+            'search': 'SSM',
+            'graceid': graceid,
+            'far': 1e-10,
+            'gpstime': 1234,
+        }
     return event
 
 
@@ -85,6 +95,10 @@ def get_event(graceid):
           {'HasNS': 0.0, 'HasRemnant': 0.0, 'HasMassGap': 0.0},
           {"BNS": 0.0, "NSBH": 0.0, "BBH": 0.9, "Terrestrial": 0.1},
           skymap_small, True],
+         [{'labels': [], 'gw_events': ['G123', 'G468']},
+          {'HasNS': 0.0, 'HasSSM': 1.0, 'HasMassGap': 0.0},
+          {"BNS": 0.0, "NSBH": 0.0, "BBH": 0.9, "Terrestrial": 0.1},
+          skymap_large, True],
          [{'labels': [], 'gw_events': ['G123']},
           {'HasNS': 0.0, 'HasRemnant': 0.0, 'HasMassGap': 0.0},
           {"BNS": 0.0, "NSBH": 0.0, "BBH": 1.0, "Terrestrial": 0.0},
@@ -110,8 +124,12 @@ def test_high_profile(monkeypatch, superevent, embright,
                       pastro, skymap, result):
     embright = json.dumps(embright)
     pastro = json.dumps(pastro)
-    superevent.update({'superevent_id': 'S123',
-                       'preferred_event_data': get_event('G123')})
+    if 'G468' in superevent['gw_events']:
+        superevent.update({'superevent_id': 'S123',
+                          'preferred_event_data': get_event('G468')})
+    else:
+        superevent.update({'superevent_id': 'S123',
+                           'preferred_event_data': get_event('G123')})
     mock_create_label = Mock()
     monkeypatch.setattr('gwcelery.tasks.gracedb.create_label.run',
                         mock_create_label)