Add automated skyproj map making if desired.

Author: erykoff

decasu/decasu_hpix_mapper.py

@@ -28,6 +28,8 @@ def main():
                         required=False, help='Keep intermediate files')
     parser.add_argument('-q', '--query', required=False,
                         help='Additional query string; will replace lsst_db_additional_selection config.')
+    parser.add_argument('-m', '--make_map_images', action='store_true', required=False,
+                        help='Automatically make skyproj map images?')
 
     args = parser.parse_args()
 
@@ -55,4 +57,5 @@ def main():
     else:
         mapper = MultiHealpixMapper(config, args.outputpath, ncores=args.ncores)
         mapper(args.infile, bands=bands, pixels=pixels,
-               clear_intermediate_files=not args.keep_intermediate_files)
+               clear_intermediate_files=not args.keep_intermediate_files,
+               make_map_images=args.make_map_images)

decasu/healpix_consolidator.py

@@ -13,25 +13,45 @@ class HealpixConsolidator:
     clear_intermediate_files : `bool`
        Clear input files when done?
     """
-    def __init__(self, config, clear_intermediate_files):
+    def __init__(self, config, clear_intermediate_files, make_map_images=False):
         self.config = config
         self.clear_intermediate_files = clear_intermediate_files
+        self.make_map_images = make_map_images
 
-    def __call__(self, fname, mapfiles):
+    def __call__(self, fname, mapfiles, descr):
         """
         Consolidate a list of mapfiles, and delete input mapfiles
         if clear_intermediate_files is True.
 
         Parameters
         ----------
         fname : `str`
-           Output filename
+            Output filename
         mapfiles : `list`
-           Input list of files
+            Input list of files
+        descr : `str`
+            Description string.
         """
         print("Consolidating %d maps into %s" % (len(mapfiles), fname))
         healsparse.cat_healsparse_files(mapfiles, fname)
 
+        if self.make_map_images:
+            from matplotlib.backends.backend_agg import FigureCanvasAgg
+            from matplotlib.figure import Figure
+            import skyproj
+
+            m = healsparse.HealSparseMap.read(fname)
+
+            fig = Figure(figsize=(10, 6))
+            FigureCanvasAgg(fig)
+            ax = fig.add_subplot(111)
+
+            sp = skyproj.McBrydeSkyproj(ax=ax)
+            sp.draw_hspmap(m, zoom=True)
+            sp.draw_colorbar(label=descr)
+            skyprojfile = fname.replace(".hsp", "_skyproj.png")
+            fig.savefig(skyprojfile)
+
         if self.clear_intermediate_files:
             for f in mapfiles:
                 os.unlink(f)

decasu/multi_healpix_mapper.py

@@ -35,21 +35,23 @@ def __init__(self, config, outputpath, ncores=1):
         if not os.path.isdir(outputpath):
             raise RuntimeError("Outputpath %s does not exist." % (outputpath))
 
-    def __call__(self, infile, bands=[], pixels=[], clear_intermediate_files=True):
+    def __call__(self, infile, bands=[], pixels=[], clear_intermediate_files=True, make_map_images=False):
         """
         Compute maps for a combination of bands/pixels
 
         Parameters
         ----------
         infile : `str`
-           Name of input file with wcs and map information
+            Name of input file with wcs and map information
         bands : `list`, optional
-           List of bands to run.  If blank, run all.
+            List of bands to run.  If blank, run all.
         pixels : `list`, optional
-           List of pixels to run (nside=`config.nside_run`).
-           If blank, run all.
+            List of pixels to run (nside=`config.nside_run`).
+            If blank, run all.
         clear_intermediate_files : `bool`, optional
-           Clear intermediate files when done?
+            Clear intermediate files when done?
+        make_map_images : `bool`, optional
+            Make map images?
         """
         # First build the wcs's
         print('Reading input table...')
@@ -125,6 +127,7 @@ def __call__(self, infile, bands=[], pixels=[], clear_intermediate_files=True):
         print('Consolidating maps...')
         fname_list = []
         mapfiles_list = []
+        descr_list = []
         for map_type in self.config.map_types.keys():
             for operation in self.config.map_types[map_type]:
                 op_code = op_str_to_code(operation)
@@ -147,10 +150,15 @@ def __call__(self, infile, bands=[], pixels=[], clear_intermediate_files=True):
                                                              fname_template)))
                     fname_list.append(fname)
                     mapfiles_list.append(mapfiles)
+                    descr_list.append(f'{band}_{map_type}_{operation}')
 
-        hpix_consolidator = HealpixConsolidator(self.config, clear_intermediate_files)
+        hpix_consolidator = HealpixConsolidator(
+            self.config,
+            clear_intermediate_files,
+            make_map_images=make_map_images,
+        )
 
-        values = zip(fname_list, mapfiles_list)
+        values = zip(fname_list, mapfiles_list, descr_list)
 
         t = time.time()
         mp_ctx = multiprocessing.get_context("fork")

requirements.txt

@@ -6,3 +6,5 @@ fitsio>=1.0.5
 esutil>=0.6.5
 LSSTDESC.Coord
 pyyaml>=5.0
+skyproj
+matplotlib

tests/decasu_test_base.py

@@ -220,7 +220,7 @@ def check_expected_maps_hpix(self, expected_dict, nside, hpix, band, check_amp=F
 
         return mod_times
 
-    def check_expected_maps_consolidated(self, expected_dict, band, time_bin=-1):
+    def check_expected_maps_consolidated(self, expected_dict, band, time_bin=-1, images=False):
         """
         Check for expected consolidated maps, ranges, types.
 
@@ -232,6 +232,8 @@ def check_expected_maps_consolidated(self, expected_dict, band, time_bin=-1):
             Band name
         time_bin : `int`, optional
             Bin of time to be processing.
+        images : `bool`, optional
+            Check for images?
         """
         if time_bin < 0:
             outbase = 'testing'
@@ -263,4 +265,9 @@ def check_expected_maps_consolidated(self, expected_dict, band, time_bin=-1):
             self.assertLess(np.max(m[valid_pixels]),
                             expected_dict[em][1])
 
+            if images:
+                image_name = os.path.join(self.test_dir, '%s_%s_%s_skyproj.png'
+                                          % (outbase, band, em))
+                self.assertTrue(os.path.isfile(image_name))
+
         return mod_times

tests/test_hpix_lsst.py

@@ -39,7 +39,8 @@ def test_hpix_lsst(self):
 
         mapper = decasu.MultiHealpixMapper(config, self.test_dir, ncores=1)
         mapper(dbfile, bands=[band],
-               clear_intermediate_files=False)
+               clear_intermediate_files=False,
+               make_map_images=True)
 
         # Look at 549
         expected_dict = OrderedDict()
@@ -60,7 +61,7 @@ def test_hpix_lsst(self):
         expected_dict['exptime_sum'] = [29.0, 61.0, 'float64']
         expected_dict['nexp_sum'] = [0, 3, 'int32']
 
-        self.check_expected_maps_consolidated(expected_dict, 'r', time_bin=0)
+        self.check_expected_maps_consolidated(expected_dict, 'r', time_bin=0, images=True)
 
 
 if __name__ == '__main__':