pz-rail-astro-tools 1.0.0__py3-none-any.whl → 1.0.2__py3-none-any.whl

{pz_rail_astro_tools-1.0.0.dist-info → pz_rail_astro_tools-1.0.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pz-rail-astro-tools
-Version: 1.0.0
+Version: 1.0.2
 Author-email: "LSST Dark Energy Science Collaboration (DESC)" <lsst-desc-rail-admin@slac.stanford.edu>
 License: MIT License
         

{pz_rail_astro_tools-1.0.0.dist-info → pz_rail_astro_tools-1.0.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 rail/astro_tools/__init__.py,sha256=c0yzN5jMhByii01h_mjOWipeProO8vludBsO09dJkmc,377
-rail/astro_tools/_version.py,sha256=DGJ4pj32xs3_DRJhSzQwCiRNnAQrMgo09USYpyMZsKc,411
+rail/astro_tools/_version.py,sha256=BGgbj7AHZwtVw29XgDIjcG5woR-f3jQy2X-ArcRT22E,411
 rail/creation/degraders/grid_selection.py,sha256=QCs7ZpLJ_GdJnIIt0WF09mBLeqsrzQ2CiDo0h0Hd3dQ,12438
 rail/creation/degraders/lsst_error_model.py,sha256=WWQxPO3w1S6sALQxl7x7_mUI1f3o1eOGpBk3Yzi-1Q0,1969
 rail/creation/degraders/observing_condition_degrader.py,sha256=jelIfFHzrgmRmRARm5uC36ZohepfMHq5djWT5wGEaZI,19987
@@ -13,9 +13,10 @@ rail/examples_data/creation_data/data/survey_conditions/DC2-dr6-galcounts-i20-i2
 rail/examples_data/creation_data/data/survey_conditions/DC2-mask-neg-nside-128.fits,sha256=79Kg4xRP4WHlOCcS-3SKqf9aP_LZRAq-izIM0DFL6bw,1581120
 rail/examples_data/creation_data/data/survey_conditions/minion_1016_dc2_Median_airmass_i_and_nightlt1825_HEAL.fits,sha256=twvyce70G3h1MQnGGFQJG9wZc8UbuNqXc6mjtiuaL1s,1581120
 rail/examples_data/creation_data/data/survey_conditions/minion_1016_dc2_Median_fiveSigmaDepth_i_and_nightlt1825_HEAL.fits,sha256=RbMMFSVJQxGhZnL4BnISE9VTNljfTZwV6Bax6NxverY,1581120
-rail/tools/photometry_tools.py,sha256=aAIfD_CsYvJappbh6FR2GLt2eeQdMg59N5VuLD5QTjg,18419
-pz_rail_astro_tools-1.0.0.dist-info/LICENSE,sha256=tLMEN21HbzvT-7umOVVvPjaJZbQxCa-2bAeYLwG5Q04,1102
-pz_rail_astro_tools-1.0.0.dist-info/METADATA,sha256=Bej2Sj-dvJIdSZQybbbiiHtZ-z3mU8snKUOydzVFcFg,3947
-pz_rail_astro_tools-1.0.0.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-pz_rail_astro_tools-1.0.0.dist-info/top_level.txt,sha256=6R6sqn-85I8YZTzmB9gv7sqckB2QaOTHbu-QLGTWBRE,5
-pz_rail_astro_tools-1.0.0.dist-info/RECORD,,
+rail/examples_data/testdata/rubin_dm_dc2_example2.pq,sha256=_Iihbbcz1A332QAkaMiFacT4QAbBQBMTOwpfrS7AclE,25884
+rail/tools/photometry_tools.py,sha256=GdqHWoYzkbQTQoEWvkz1qNZjJbJCFgQUR39d4xuz1cI,19216
+pz_rail_astro_tools-1.0.2.dist-info/LICENSE,sha256=tLMEN21HbzvT-7umOVVvPjaJZbQxCa-2bAeYLwG5Q04,1102
+pz_rail_astro_tools-1.0.2.dist-info/METADATA,sha256=YooFUgTa8FAlMqG7boZeO2dQ6yaa8Gaaf7GeeYZD4xA,3947
+pz_rail_astro_tools-1.0.2.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+pz_rail_astro_tools-1.0.2.dist-info/top_level.txt,sha256=6R6sqn-85I8YZTzmB9gv7sqckB2QaOTHbu-QLGTWBRE,5
+pz_rail_astro_tools-1.0.2.dist-info/RECORD,,

rail/astro_tools/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '1.0.0'
-__version_tuple__ = version_tuple = (1, 0, 0)
+__version__ = version = '1.0.2'
+__version_tuple__ = version_tuple = (1, 0, 2)

rail/tools/photometry_tools.py

@@ -13,6 +13,7 @@ from ceci.config import StageParameter as Param
 from rail.core.data import PqHandle
 from rail.core.stage import RailStage
 from rail.core.data import PqHandle, Hdf5Handle
+from rail.core.common_params import SHARED_PARAMS
 
 import hyperbolic  # https://github.com/jlvdb/hyperbolic
 
@@ -20,8 +21,6 @@ dustmaps_config = tables_io.lazy_modules.lazyImport('dustmaps.config')
 dustmaps_sfd = tables_io.lazy_modules.lazyImport('dustmaps.sfd')
 
 
-
-
 # default column names in DC2
 LSST_BANDS = 'ugrizy'
 DEFAULT_MAG_COLS = [f"mag_{band}_lsst" for band in LSST_BANDS]
@@ -333,7 +332,7 @@ class HyperbolicMagnitudes(PhotometryManipulator):
             Input table with photometry (magnitudes or flux columns and their respective
             uncertainties) as defined by the configuration.
         parameters : `PqHandle`
-            Table with smoothing parameters per photometric band, determined by
+            Table witdh smoothing parameters per photometric band, determined by
             `HyperbolicSmoothing`.
 
         Returns
@@ -412,16 +411,18 @@ class LSSTFluxToMagConverter(RailStage):
         return self.get_handle('output')
 
 
-class Dereddener(RailStage):
+class DustMapBase(RailStage):
     """Utility stage that does dereddening
     
     Note: set copy_all_cols=True to copy all 
     columns in data, copy_cols will be ignored
     """
-    name = 'Dereddener'
+    name = 'DustMapBase'
 
     config_options = RailStage.config_options.copy()
     config_options.update(bands='ugrizy')
+    config_options.update(ra_name='ra')
+    config_options.update(dec_name='dec')
     config_options.update(mag_name="mag_{band}_lsst")
     config_options.update(band_a_env=[4.81,3.64,2.70,2.06,1.58,1.31])
     config_options.update(dustmap_name='sfd')
@@ -429,8 +430,8 @@ class Dereddener(RailStage):
     config_options.update(copy_cols=[])
     config_options.update(copy_all_cols=False)
 
-    inputs = [('input', Hdf5Handle)]
-    outputs = [('output', Hdf5Handle)]
+    inputs = [('input', PqHandle)]
+    outputs = [('output', PqHandle)]
 
     def fetch_map(self):
         dust_map_dict = dict(sfd=dustmaps_sfd)
@@ -439,12 +440,15 @@ class Dereddener(RailStage):
         except KeyError as msg:  # pragma: no cover
             raise KeyError(f"Unknown dustmap {self.config.dustmap_name}, options are {list(dust_map_dict.keys())}") from msg
 
-        if os.path.exists(os.path.join(self.config.dustmap_dir, self.config.dustmap_name)):  # pragma: no cover
+        dustmap_dir = os.path.expandvars(self.config.dustmap_dir)
+        dustmap_path = os.path.join(dustmap_dir, self.config.dustmap_name)
+        if os.path.exists(dustmap_path):  # pragma: no cover
             # already downloaded, return
             return
         
         dust_map_config = dustmaps_config.config
-        dust_map_config['data_dir'] = self.config.dustmap_dir
+        # dust_map_config['data_dir'] = self.config.dustmap_dir
+        dust_map_config['data_dir'] = dustmap_dir
         fetch_func = dust_map_submod.fetch
         fetch_func()
         
@@ -455,7 +459,7 @@ class Dereddener(RailStage):
     def run(self):
         data = self.get_data('input', allow_missing=True)
         out_data = {}
-        coords = SkyCoord(data['ra'], data['decl'], unit = 'deg',frame='fk5')
+        coords = SkyCoord(data[self.config.ra_name], data[self.config.dec_name], unit = 'deg',frame='fk5')
         dust_map_dict = dict(sfd=dustmaps_sfd.SFDQuery)
         try:
             dust_map_class = dust_map_dict[self.config.dustmap_name]
@@ -469,7 +473,7 @@ class Dereddener(RailStage):
         for i, band_ in enumerate(self.config.bands):
             band_mag_name = self.config.mag_name.format(band=band_)
             mag_vals = data[band_mag_name]
-            out_data[band_mag_name] = mag_vals - ebvvec*self.config.band_a_env[i]
+            out_data[band_mag_name] = self._calc_values(mag_vals, ebvvec, self.config.band_a_env[i])
             band_mag_name_list.append(band_mag_name)
        
         # check if copy_all_cols set to true:
@@ -482,7 +486,8 @@ class Dereddener(RailStage):
                 if col_ not in band_mag_name_list:
                     out_data[col_] = data[col_]
 
-        self.add_data('output', out_data)
+        out_data_pd = pd.DataFrame(out_data)
+        self.add_data('output', out_data_pd)
 
     def __call__(self, data):
         """Return a converted table
@@ -500,3 +505,23 @@ class Dereddener(RailStage):
         self.set_data('input', data)
         self.run()
         return self.get_handle('output')
+
+
+class Dereddener(DustMapBase):
+    """Utility stage that does dereddening
+    
+    """
+    name = 'Dereddener'
+
+    def _calc_values(self, mag_vals, ebvvec, band_a_env):
+        return mag_vals - ebvvec*band_a_env
+    
+
+class Reddener(DustMapBase):
+    """Utility stage that does reddening
+    
+    """
+    name = 'Reddener'
+
+    def _calc_values(self, mag_vals, ebvvec, band_a_env):
+        return mag_vals + ebvvec*band_a_env