antares-devkit 0.6.5__tar.gz → 0.7.0__tar.gz

{antares_devkit-0.6.5 → antares_devkit-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: antares-devkit
-Version: 0.6.5
+Version: 0.7.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>

antares_devkit-0.7.0/antares_devkit/filters/decam_shadow/__init__.py

@@ -0,0 +1,75 @@
+"""Description
+This filter finds locus located within a radius of 1.1 deg from the center of the Shadow Project's Virgo, Hydra, Fornax, Antlia, and Dorado Pointings.
+"""
+from antares_devkit.models import BaseFilter
+
+class InDecamShadow(BaseFilter):
+    NAME = 'in_decam_shadow' #for this default filter, name is same as its tag
+
+    SLACK_CHANNEL = "#filter-in-decam-shadow"
+
+    ## bare absolute minimum alert properties proving it's good to go
+    REQUIRED_ALERT_PROPERTIES = [
+        'ant_ra',
+        'ant_dec',
+        'ant_survey'
+    ]
+
+    OUTPUT_TAGS = [
+        {
+            'name': 'in_decam_shadow',
+            'description': 'Locus is along the line of sight to the Virgo, Antlia, Dorado, Fornax, Hydra, Virgo fields covered by the DECam Shadow Project',
+        }
+    ]
+
+    def _run(self, locus):
+        import numpy as np
+        from astropy import units as u
+        from astropy.coordinates import SkyCoord
+        if locus.alert.properties['ant_survey'] == 2: #be sure we don't work with upper limits
+            #print('upper limit')
+            return
+            
+        if locus.alert.properties['ant_survey'] == 4:
+            #print(locus.alert.properties['lsst_diaSource_ssObjectId'], locus.alert.properties['lsst_diaSource_ssObjectId'] != 0, 'lsst_diaSource_ssObjectId')
+            if (locus.alert.properties['lsst_diaSource_ssObjectId'] != 0) is True:
+                return
+        elif locus.alert.properties['ant_survey'] == 1:
+            #print(locus.alert.properties['ztf_ssnamenr'], locus.alert.properties['ztf_ssnamenr'] !='null', 'ztf_ssnamenr')
+            if locus.alert.properties['ztf_ssnamenr'] != 'null':
+                return
+        
+        #    return 
+        ra = locus.alert.properties['ant_ra']
+        dec = locus.alert.properties['ant_dec']
+        #https://github.com/lsstshadow/shadowpointings/tree/main
+        shadow_virgo_ra = np.array([187.7014, 186.0754, 187.9486, 189.4263, 189.1219, 187.4805, 
+                                    185.9601, 187.6007, 186.0131, 184.8111, 184.3502, 183.9883, 
+                                    184.4658, 186.3005, 191.0718, 190.5631, 189.1516, 188.7788, 
+                                    186.6354, 185.6507, 182.4151, 193.2665, 189.5427, 190.6638, 
+                                    193.4287, 188.2634, 185.2524, 188.0598, 193.0875, 191.0943])
+        shadow_virgo_dec = np.array([12.46,  13.4965,  14.307,   13.3041,  11.295,   10.4477, 
+                                     11.6469,   8.4389,   7.1655,   5.7578,  12.8014,   7.6415,
+                                     14.6628,  15.364,   12.0456,  10.0351,   8.9448,   6.8992, 
+                                      5.3375,   8.9489,  13.7313,   8.2527,   4.8363,   2.8801, 
+                                      1.8786,   2.356,    3.7762,   0.1955,  -1.1488,  -0.0399])
+        shadow_antlia_ra = np.array([ 157.3975])
+        shadow_antlia_dec = np.array([-35.3325])
+        shadow_dorado_ra = np.array([  71.4476,  66.7167,  63.6977,  65.9529,  62.6324])
+        shadow_dorado_dec = np.array([-59.2477, -55.0722, -55.6902, -57.0202, -57.7644])
+        shadow_fornax_ra = np.array([  54.5484,  54.1655,  52.4792,  52.8626,  54.9590, 56.6247, 
+                                       56.2302,  52.1175,  51.1123,  50.5994,  55.8479, 58.3521, 
+                                       53.6473, 51.4467, 63.1329])
+        shadow_fornax_dec = np.array([-35.5388, -33.7261, -34.8784, -36.6898, -37.3477, -36.1969, 
+                                      -34.3845, -33.0674, -37.8454, -35.8702, -32.5273, -34.962, 
+                                      -31.6796, -31.2712, -32.3313])
+        shadow_hydra_ra = np.array([ 159.2305, 159.2126, 157.6973])
+        shadow_hydra_dec = np.array([-27.5186, -25.3493, -29.1417])
+        shadow_all_ra = np.hstack((shadow_virgo_ra, shadow_antlia_ra, shadow_dorado_ra, shadow_fornax_ra, shadow_hydra_ra))
+        shadow_all_dec = np.hstack((shadow_virgo_dec, shadow_antlia_dec, shadow_dorado_dec, shadow_fornax_dec, shadow_hydra_dec))
+        field_centers = SkyCoord(ra=shadow_all_ra*u.deg, dec=shadow_all_dec*u.deg)
+        c = SkyCoord(ra*u.deg, dec*u.deg, frame='icrs')
+        sep = c.separation(field_centers)
+        #print(sep.min())
+        if (sep.deg < 1.1).any(): #DECam FOV = 2.2 deg diameter
+            locus.tag('in_decam_shadow')

antares_devkit-0.7.0/antares_devkit/filters/uniform_random_sample/__init__.py

@@ -0,0 +1,29 @@
+from antares_devkit.models import BaseFilter
+
+class UniformRandomSample(BaseFilter):
+    """
+    This filter aims to uniformly and randomly select 1% of all LSST alerts.
+    This is implemented by filtering for diaSourceId % 113 == 0, which roughly
+    selects everything 100th alert in chronological order. The 113 value is chosen
+    to be close to but arbitrarily different from 100, to avoid any potential
+    collisions with processing that might happen on every 100th alert.
+    """
+    SLACK_CHANNEL = "#filter-uniformrandomsample"
+    TRIGGERING_SURVEY = "lsst"
+
+    REQUIRED_ALERT_PROPERTIES = [
+       "lsst_diaSource_diaSourceId"
+    ]
+
+    OUTPUT_LOCUS_PROPERTIES = []
+    OUTPUT_LOCUS_TAGS = [{
+        'name': 'uniform_random_sample',
+        'description': '1% of all LSST Alerts uniformly and randomly sampled'
+    }]
+
+    def _run(self, locus):
+        """
+        If a LSST alert's diaSourceId % 113 == 0, tag with 'uniform_random_sample'
+        """
+        if locus.alert.properties['lsst_diaSource_diaSourceId'] % 113 == 0:
+            locus.tag("uniform_random_sample")

{antares_devkit-0.6.5 → antares_devkit-0.7.0}/antares_devkit/utils.py

@@ -24,6 +24,7 @@ def get_filters_path():
         "IF_anomaly_detection": "antares_devkit.filters.if_anomaly_detection",  # ["numpy", "pandas", "sklearn", "astropy"]
         "IMBH": "antares_devkit.filters.imbh",
         "In_LSSTDDF": "antares_devkit.filters.lsst_ddf",  # ["astropy", "numpy"]
+        "InDecamShadow": "antares_devkit.filters.decam_shadow",  # ["astropy", "numpy"]
         "InM31": "antares_devkit.filters.inm31",
         "LAISS_RFC_AD_filter": "antares_devkit.filters.laiss_rfc_ad_filter",  # ["astropy", "timeout_decorator", "alerce", "numpy", "pandas", "matplotlib", "astro_ghost", "sklearn", "scipy"]
         "AnomalyTransientDmDt": "antares_devkit.filters.malanchev_anomaly_transient_dmdt",  # ["numpy", "astropy", "light_curve"]
@@ -51,6 +52,7 @@ def get_filters_path():
         "rubin_NovTest_soraisam": "antares_devkit.filters.rubin_NovTest_soraisam",
         "young_rubin_transients_soraisam": "antares_devkit.filters.young_rubin_transients_soraisam",
         "LsstTransientQualityFilterScimma": "antares_devkit.filters.lsst_transient_quality_scimma",
+        "UniformRandomSample": "antares_devkit.filters.uniform_random_sample",
     }
 
 

{antares_devkit-0.6.5 → antares_devkit-0.7.0}/pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "antares-devkit"
-version = "0.6.5"
+version = "0.7.0"
 description = "Provides tools and utility methods to write and debug ANTARES filters with Python."
 readme = "README.md"
 requires-python = ">=3.9"