Adding CCDNUM to HDU-0 of headers

Author: menanteau

python/immask/__init__.py

@@ -5,7 +5,7 @@
 import os
 
 __author__ = "Felipe Menanteau, Alex Drlica-Wagner, Eli Rykoff"
-__version__ = '3.0.2'
+__version__ = '3.0.3'
 __revision__= '0'
 version = __version__
 

python/immask/immasklib.py

@@ -837,7 +837,15 @@ def write_streak_objects(self):
             outname = os.path.join(self.outdir, outbase)
 
         logging.info("Writing streak objects: %s", outname)
-        fitsio.write(outname, self.mask_objs, clobber=True)
+        fits = fitsio.FITS(outname, 'rw', clobber=True)
+        fits.write(self.mask_objs)
+        # Make sure that CCDNUM is in the header.
+        # It's needed for connect_streaks later on
+        try:
+            fits[0].write_key('ccdnum', self.image.header['CCDNUM'], comment="CCD number")
+        except:
+            logging.info("CCDNUM could be inserted in HDU-0 header")
+        fits.close()
 
     def write_streak_mask(self):
         """ Write out streak mask """
@@ -1824,6 +1832,7 @@ def run(inargs):
     commands = [inargs.command]
     kwargs = vars(inargs)
     image = DESImage.load(inargs.filename)
+
     # passing outname into the class
     image.outname = inargs.outname
     if 'all' in commands:

setup.py

@@ -3,7 +3,7 @@
 
 # The main call
 setup(name='immask',
-      version ='3.0.2',
+      version ='3.0.3',
       license = "GPL",
       description = "The Image Masking Module for DESDM.",
       author = "Felipe Menanteau, Alex Drlica-Wagner, Eli Rykoff",