Merge pull request #149 from /issues/148

Switch to fitsio to write FITS faster.

Author: wmwv

changes/149.feature.rst

@@ -0,0 +1 @@
+Swtich to fitsio for 2-4x FITS writing speed improvement.

phrosty/pipeline.py

@@ -17,10 +17,10 @@
 
 # Imports ASTRO
 from astropy.coordinates import SkyCoord
-from astropy.io import fits
 from astropy.table import Table
 import astropy.units as u
 from astropy.wcs.utils import skycoord_to_pixel
+import fitsio
 
 # Imports INTERNAL
 from phrosty.imagesubtraction import sky_subtract, stampmaker
@@ -134,7 +134,7 @@ def get_psf( self, ra, dec ):
         stamp = self.psfobj.get_stamp( x, y )
         if self.keep_intermediate:
             outfile = self.save_dir / f"psf_{self.image.name}.fits"
-            fits.writeto( outfile, stamp, overwrite=True )
+            fitsio.write( outfile, stamp, clobber=True )
 
         return stamp
 
@@ -586,7 +586,11 @@ def log_error( x ):
 
     def write_fits_file( self, data, header, savepath ):
         try:
-            fits.writeto( savepath, data, header=header, overwrite=True )
+            if header is not None:
+                hdr_dict = dict(header.items())
+            else:
+                hdr_dict = None
+            fitsio.write( savepath, data, header=hdr_dict, clobber=True )
         except Exception as e:
             SNLogger.exception( f"Exception writing FITS image {savepath}: {e}" )
             raise