Added WAVEUNIT keyword

suncet · Sep 15, 2023 · abc2cfc · abc2cfc
1 parent ead96c7
commit abc2cfc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/suncet_instrument_simulator/make_radiance_maps.py b/suncet_instrument_simulator/make_radiance_maps.py
@@ -131,7 +131,8 @@ def __make_header_template(self):
         header['PC2_1'] = 0.0
         header['PC2_2'] = 1.0
         header['WAVELNTH'] = 170.0 # NOTE: FITS doesn't have a keyword to store the units for wavelength, it's usually listed in the header comment instead
-        header['BUNIT'] = 'photon/(cm2 s sr Angstrom)' # TODO: Dan: Consider forcing in a wavelength unit as well, and then reading that value in load_maps
+        header['WAVEUNIT'] = 'Angstrom'
+        header['BUNIT'] = 'photon/(cm2 s sr Angstrom)'
         header['WCSNAME'] = 'Helioprojective-cartesian'
         header['HGLT_OBS'] = 0.0
         header['HGLN_OBS'] = 0.0
@@ -164,9 +165,11 @@ def __save_map_sequence(self):
         for n, map in enumerate(self.map_seq):
             if n == 0:
                 hdu = fits.PrimaryHDU(map.data, map.fits_header)
+                map.fits_header['WAVELNTH'] = (map.fits_header['WAVELNTH'], '[Angstrom]')
                 hdul = fits.HDUList(hdu)
             else:
                 hdu = fits.ImageHDU(map.data, map.fits_header)
+                map.fits_header['WAVELNTH'] = (map.fits_header['WAVELNTH'], '[Angstrom]')
                 hdul.append(hdu)
         print('Saving Map: ' + map_file_out)
         try: