fix ratir filters

SNflows · May 17, 2022 · 5aebfeb · 5aebfeb
1 parent ac0ef76
commit 5aebfeb
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flows/load_image.py b/flows/load_image.py
@@ -563,6 +563,12 @@ def get_obstime(self):
         obstime =  Time(self.image.header['DATE-OBS'], format='isot', scale='utc',
                         location=self.image.site['EarthLocation'])
         return obstime
+
+    def get_photfilter(self):
+        ratir_filt = self.image.header['FILTER']
+        if ratir_filt in ['Z', 'r', 'i']:
+            return {'Z': 'zp', 'r': 'rp', 'i': 'ip'}.get(ratir_filt)
+        return ratir_filt
 
 
 instruments = {'LCOGT': LCOGT, 'HAWKI': HAWKI, 'ALFOSC': ALFOSC, 'NOTCAM': NOTCAM, 'PS1': PS1, 'Liverpool': Liverpool,