antares-devkit 0.4.0__tar.gz → 0.5.0__tar.gz

{antares_devkit-0.4.0 → antares_devkit-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: antares-devkit
-Version: 0.4.0
+Version: 0.5.0
 Summary: Provides tools and utility methods to write and debug ANTARES filters with Python.
 Keywords: antares,devkit,filter development
 Author-Email: NSF NOIRLab ANTARES Team <antares@noirlab.edu>
@@ -50,6 +50,7 @@ Requires-Dist: statsmodels>=0.12.2; extra == "filter-dependencies"
 Requires-Dist: scipy>=1.13.1; extra == "filter-dependencies"
 Requires-Dist: light-curve==0.7.2; extra == "filter-dependencies"
 Requires-Dist: ssi-forest>=0.1.1; extra == "filter-dependencies"
+Requires-Dist: slack-sdk>=3.39.0; extra == "filter-dependencies"
 Provides-Extra: jupyter
 Requires-Dist: matplotlib>=3.9.4; extra == "jupyter"
 Description-Content-Type: text/markdown

antares_devkit-0.5.0/antares_devkit/filters/rubin_NovTest_soraisam/__init__.py

@@ -0,0 +1,187 @@
+## v2
+from antares_devkit.models import BaseFilter
+
+
+class rubin_NovTest_soraisam(BaseFilter):
+    ## MS: this is mainly for the NOIRLab end-to-end demonstration
+    
+    TRIGGERING_SURVEY = "lsst"
+
+    SLACK_CHANNEL = "#filter-rubin-novtest"
+
+    REQUIRED_LOCUS_PROPERTIES = []
+
+    ## bare absolute minimum alert properties proving it's good to go
+    REQUIRED_ALERT_PROPERTIES = [
+        "lsst_diaSource_reliability",
+        "lsst_diaSource_snr",
+        "lsst_diaSource_ssObjectId",
+        "lsst_diaSource_extendedness",
+        "lsst_diaSource_templateFlux",
+        "lsst_diaSource_templateFluxErr",
+        "lsst_diaSource_isNegative",
+        "ant_mjd",
+        "ant_mag",
+        "ant_passband",
+    ]
+    
+
+    OUTPUT_LOCUS_TAGS = [
+        {
+            "name": "SN_candies",
+            "description": "Locus is a SN candidate in the Rubin alert stream.",
+        },
+    ]
+
+    # Slack post (going to tda-noao noirlab_e2e_demo channel)
+    def send_message_to_slack(self, msg):
+        from slack_sdk import WebClient
+        from slack_sdk.errors import SlackApiError
+        import os
+
+        slack_token = os.getenv("RUBIN_NOVTEST_SLACK_TOKEN", "")
+        channel_id = os.getenv("RUBIN_NOVTEST_SLACK_CHANNEL_ID", "")
+        if (not slack_token) or (not channel_id):
+            print ("Slack not set up")
+            return  
+        client = WebClient(token=slack_token)
+        
+        try:
+            response = client.chat_postMessage(
+                channel=channel_id,
+                text=msg
+            )
+            # print(f"Message sent successfully: {response['ts']}")
+        except SlackApiError as e:
+            print(f"Error sending message: {e.response['error']}")
+            pass
+
+        return
+    
+    
+
+
+    def _run(self, locus):
+        import numpy as np
+        from astropy.time import Time
+        from astropy.coordinates import SkyCoord
+        import astropy.units as U
+        from itertools import combinations
+
+        
+        C1 = (locus.alert.properties['lsst_diaSource_reliability'] < 0.9) or (locus.alert.properties['lsst_diaSource_snr'] < 10.0)
+        if C1:
+            return
+
+        C2 = (locus.alert.properties['lsst_diaSource_ssObjectId'] != 0)
+        if C2:
+            return
+
+        ## extract info from recent alert 
+        alert_id = locus.alert.alert_id
+        current = Time(locus.alert.properties['ant_mjd'], format="mjd")
+        current_mag = locus.alert.properties['ant_mag']
+        current_pb = locus.alert.properties['ant_passband']
+        
+        # ## confirm locus is Not in legacy point source catalogs
+        astro_objects = locus.catalog_objects        
+        if astro_objects and "2mass_psc" in astro_objects.keys():
+            return
+
+        
+        
+        TS_all = locus.timeseries('all')
+        TS_lsst = locus.timeseries('lsst')
+
+        ## want more than two measurements across all filters and time span less than 15 days (young)
+        C4 = (len(TS_all) > 2) and (max(TS_lsst["ant_mjd"].data) - min(TS_lsst["ant_mjd"].data) < 15.0)
+        if not C4:
+            return
+
+        ## want star like profile 
+        try:
+            C5 = np.ma.median(TS_lsst['lsst_diaSource_extendedness'].data) > 0.3
+            if C5:
+                return
+        except: ## some alerts do not have extendedness apparently 
+            print ("extendedness not in alert packet")
+            return
+        
+        
+        C6 = np.ma.median(np.abs(TS_lsst['lsst_diaSource_templateFlux'].data/TS_lsst['lsst_diaSource_templateFluxErr'].data))<1000.0
+        if not C6:
+            return
+
+        C7 = np.all(TS_lsst['lsst_diaSource_isNegative'].data==False)
+        if not C7:
+            return
+
+        
+        rubin_pb = ['u','g','r','i','z']
+        x = TS_lsst["ant_mjd"].data
+        y = TS_lsst["ant_mag"].data
+        pb = TS_lsst["ant_passband"].data  
+        
+        
+        def is_it_blue_color(p1, p2):
+            g_x_i = x[pb == p1]
+            g_y_i = y[pb == p1]
+            R_x_i = x[pb == p2]
+            R_y_i = y[pb == p2]
+
+            idx = np.argsort(g_x_i)
+            g_x = g_x_i[idx]
+            g_y = g_y_i[idx]
+            idx = np.argsort(R_x_i)
+            R_x = R_x_i[idx]
+            R_y = R_y_i[idx]
+
+            dt = []
+            dc = []
+            for a, b in zip(g_x, g_y):
+                dt_i = np.abs(a - R_x)
+                dc_i = b - R_y
+                if min(dt_i) <= 1.0:  
+                    dt.append(min(dt_i))
+                    dc.append(dc_i[np.argmin(dt_i)])
+
+                
+            if len(dc) > 1:
+                col_evol = np.argsort(
+                    dc
+                ) 
+                ori_idx = np.arange(len(dc))
+                idx_diff = col_evol[::-1] - ori_idx
+                if np.sum(idx_diff == 0) >= 0.5 * len(dc):
+                    return True
+
+            return False
+
+        C8 = False
+        for p1, p2 in combinations(rubin_pb, 2): 
+            if np.sum(pb == p1) > 1 and np.sum(pb == p2) > 0:
+                if is_it_blue_color(p1, p2):
+                    C8 = True
+                    break
+        if not C8:
+            return
+
+        
+        
+        ant_id = locus.locus_id
+
+        ## for slack notification whether known to TNS
+        if astro_objects and "tns_public_objects" in astro_objects.keys():
+            known_to_tns = "in TNS"
+        else:
+            known_to_tns = "NOT in TNS"
+
+        coord = SkyCoord(locus.ra, locus.dec, unit=(U.deg, U.deg))
+        ra_str = f"{int(coord.ra.hms.h):02d}:{int(coord.ra.hms.m):02d}:{coord.ra.hms.s:05.2f}"
+        dec_str = f"{int(coord.dec.dms.d):02d}:{abs(int(coord.dec.dms.m)):02d}:{abs(coord.dec.dms.s):05.2f}"        
+
+        locus.tag("SN_candies")
+        
+        msg = f"Young SN candidate: triggering alert={alert_id} for locus <https://development.antares.noirlab.edu/loci/{ant_id}|{ant_id}> {known_to_tns},  RA={ra_str} DEC={dec_str},  {current_pb}_mag={current_mag:.2f},  #dets in LSST={len(TS_lsst)},  UTC={current.iso}"
+        self.send_message_to_slack(msg)        
+        

antares_devkit-0.5.0/antares_devkit/filters/young_rubin_transients_soraisam/__init__.py

@@ -0,0 +1,145 @@
+## v1
+from antares_devkit.models import BaseFilter
+
+
+class young_rubin_transients_soraisam(BaseFilter):
+    ## MS: this is mainly for the NOIRLab end-to-end demonstration
+    
+    TRIGGERING_SURVEY = "lsst"
+
+    SLACK_CHANNEL = "#filter-young-rubin-transients"
+
+    REQUIRED_LOCUS_PROPERTIES = []
+
+    ## bare absolute minimum alert properties proving it's good to go
+    REQUIRED_ALERT_PROPERTIES = [
+        "lsst_diaSource_reliability",
+        "lsst_diaSource_snr",
+        "lsst_diaSource_ssObjectId",
+        "lsst_diaSource_extendedness",
+        "lsst_diaSource_templateFlux",
+        "lsst_diaSource_templateFluxErr",
+        "lsst_diaSource_isNegative",
+        "ant_mjd",
+        "ant_mag",
+        "ant_passband",
+    ]
+    
+
+    OUTPUT_LOCUS_TAGS = [
+        {
+            "name": "young_rubin_transients",
+            "description": "Locus is a young transient in the Rubin alert stream.",
+        },
+    ]
+
+    # Slack post (going to tda-noao noirlab_e2e_demo_2 channel)
+    def send_message_to_slack(self, msg):
+        from slack_sdk import WebClient
+        from slack_sdk.errors import SlackApiError
+        import os
+
+        slack_token = os.getenv("RUBIN_SORAISAM_SLACK_TOKEN", "")
+        channel_id = os.getenv("RUBIN_SORAISAM_SLACK_CHANNEL_ID", "")
+        if (not slack_token) or (not channel_id):
+            print ("Slack not set up")
+            return  
+        client = WebClient(token=slack_token)
+        
+        try:
+            response = client.chat_postMessage(
+                channel=channel_id,
+                text=msg
+            )
+            # print(f"Message sent successfully: {response['ts']}")
+        except SlackApiError as e:
+            print(f"Error sending message: {e.response['error']}")
+            pass
+
+        return
+    
+    
+
+
+    def _run(self, locus):
+        import numpy as np
+        from astropy.time import Time
+        from astropy.coordinates import SkyCoord
+        import astropy.units as U
+        from itertools import combinations
+
+        
+        C1 = (locus.alert.properties['lsst_diaSource_reliability'] < 0.9) or (locus.alert.properties['lsst_diaSource_snr'] < 10.0)
+        if C1:
+            return
+
+        C2 = (locus.alert.properties['lsst_diaSource_ssObjectId'] != 0)
+        if C2:
+            return
+
+        ## extract info from recent alert 
+        alert_id = locus.alert.alert_id
+        current = Time(locus.alert.properties['ant_mjd'], format="mjd")
+        current_mag = locus.alert.properties['ant_mag']
+        current_pb = locus.alert.properties['ant_passband']
+        
+        # ## confirm locus is extragalactic 
+        astro_objects = locus.catalog_objects
+        # matching_catalog_names = astro_objects.keys()
+        # # Below is a battle-tested set of extended source catalogs (Antares-based names)
+        # xsc_cats = [
+        #     "2mass_xsc",
+        #     "ned",
+        #     "nyu_valueadded_gals",
+        #     "sdss_gals",
+        #     "veron_agn_qso",
+        #     "RC3",
+        # ]
+        # C3 = set(matching_catalog_names) & set(xsc_cats)
+        # if not C3:
+        # #     print ("No match with extragalactic catalog")
+        #     return
+        
+
+        
+        TS_all = locus.timeseries('all')
+        TS_lsst = locus.timeseries('lsst')
+
+        C4 = (len(TS_all) > 2) and (max(TS_lsst["ant_mjd"].data) - min(TS_lsst["ant_mjd"].data) < 15.0)
+        if not C4:
+            return
+
+        try:
+            C5 = np.ma.median(TS_lsst['lsst_diaSource_extendedness'].data) > 0.3
+            if C5:
+                return
+        except: ## some alerts do not have extendedness apparently 
+            print ("extendedness not in alert packet")
+            return
+        
+
+        C6 = np.ma.median(np.abs(TS_lsst['lsst_diaSource_templateFlux'].data/TS_lsst['lsst_diaSource_templateFluxErr'].data))<1000.0
+        if not C6:
+            return
+
+        C7 = np.all(TS_lsst['lsst_diaSource_isNegative'].data==False)
+        if not C7:
+            return
+        
+        
+        ant_id = locus.locus_id
+
+        ## for slack notification whether known to TNS
+        if astro_objects and "tns_public_objects" in astro_objects.keys():
+            known_to_tns = "in TNS"
+        else:
+            known_to_tns = "NOT in TNS"
+
+        coord = SkyCoord(locus.ra, locus.dec, unit=(U.deg, U.deg))
+        ra_str = f"{int(coord.ra.hms.h):02d}:{int(coord.ra.hms.m):02d}:{coord.ra.hms.s:05.2f}"
+        dec_str = f"{int(coord.dec.dms.d):02d}:{abs(int(coord.dec.dms.m)):02d}:{abs(coord.dec.dms.s):05.2f}"        
+
+        locus.tag("young_rubin_transients")
+        
+        msg = f"Young extragalactic transient: triggering alert={alert_id} for locus <https://development.antares.noirlab.edu/loci/{ant_id}|{ant_id}> {known_to_tns},  RA={ra_str} DEC={dec_str},  {current_pb}_mag={current_mag:.2f},  #dets in LSST={len(TS_lsst)},  UTC={current.iso}"
+        self.send_message_to_slack(msg)        

{antares_devkit-0.4.0 → antares_devkit-0.5.0}/antares_devkit/utils.py

@@ -45,6 +45,8 @@ def get_filters_path():
         "TestDevZtf": "antares_devkit.filters.test_dev_ztf",
         "vpdf": "antares_devkit.filters.vpdf",
         "YoungExtragalacticCandidate": "antares_devkit.filters.young_extragalactic_candidate",
+        "rubin_NovTest_soraisam": "antares_devkit.filters.rubin_NovTest_soraisam",
+        "young_rubin_transients_soraisam": "antares_devkit.filters.young_rubin_transients_soraisam",
     }
 
 

{antares_devkit-0.4.0 → antares_devkit-0.5.0}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "antares-devkit"
-version = "0.4.0"
+version = "0.5.0"
 description = "Provides tools and utility methods to write and debug ANTARES filters with Python."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -45,6 +45,7 @@ filter-dependencies = [
     "scipy>=1.13.1",
     "light-curve==0.7.2",
     "ssi-forest>=0.1.1",
+    "slack-sdk>=3.39.0",
 ]
 jupyter = [
     "matplotlib>=3.9.4",