Merge pull request #53 from winter-telescope/prebuiltcache

Add Prebuiltcache Processor

winterdrp/processors/base_processor.py

@@ -253,6 +253,25 @@ def make_image(
         raise NotImplementedError
 
 
+class ProcessorPremadeCache(ProcessorWithCache, ABC):
+
+    def __init__(
+            self,
+            master_image_path: str,
+            *args,
+            **kwargs
+    ):
+        super().__init__(*args, **kwargs)
+        self.master_image_path = master_image_path
+
+    def get_cache_path(
+            self,
+            images: list[np.ndarray],
+            headers: list[astropy.io.fits.Header],
+    ) -> str:
+        return self.master_image_path
+
+
 class BaseCandidateGenerator(BaseProcessor, ImageHandler, ABC):
 
     @classmethod

winterdrp/processors/bias.py

@@ -1,6 +1,6 @@
 import numpy as np
 import logging
-from winterdrp.processors.base_processor import ProcessorWithCache
+from winterdrp.processors.base_processor import ProcessorWithCache, ProcessorPremadeCache
 from collections.abc import Callable
 import astropy.io.fits
 from winterdrp.processors.utils.image_selector import select_from_images
@@ -63,4 +63,8 @@ def make_image(
         logger.info(f'Median combining {n_frames} biases')
         master_bias = np.nanmedian(biases, axis=2)
 
-        return master_bias, headers[0]
+        return master_bias, headers[0]
+
+
+class MasterBiasCalibrator(ProcessorPremadeCache, BiasCalibrator):
+    pass

winterdrp/processors/dark.py

@@ -5,7 +5,7 @@
 import logging
 import pandas as pd
 from collections.abc import Callable
-from winterdrp.processors.base_processor import ProcessorWithCache
+from winterdrp.processors.base_processor import ProcessorWithCache, ProcessorPremadeCache
 from winterdrp.paths import base_name_key
 from winterdrp.processors.utils.image_selector import select_from_images
 
@@ -70,3 +70,7 @@ def make_image(
         master_dark = np.nanmedian(darks, axis=2)
 
         return master_dark, headers[0]
+
+
+class MasterDarkCalibrator(ProcessorPremadeCache, DarkCalibrator):
+    pass

winterdrp/processors/flat.py

@@ -2,7 +2,7 @@
 import numpy as np
 import logging
 import sys
-from winterdrp.processors.base_processor import ProcessorWithCache
+from winterdrp.processors.base_processor import ProcessorWithCache, ProcessorPremadeCache
 from winterdrp.processors.utils.image_selector import select_from_images
 from collections.abc import Callable
 from winterdrp.paths import latest_save_key, flat_frame_key
@@ -103,7 +103,8 @@ def select_sky_flat(
     ) -> tuple[list[np.ndarray], list[astropy.io.fits.Header]]:
         return select_from_images(images, headers, header_key="obsclass", target_values="science")
 
-
+class MasterFlatCalibrator(ProcessorPremadeCache, FlatCalibrator):
+    pass
 # class OldSkyFlatCalibrator(SkyFlatCalibrator):
 #
 #     def select_cache_images(

winterdrp/processors/sky.py

@@ -5,7 +5,7 @@
 import pandas as pd
 from collections.abc import Callable
 from astropy.time import Time
-from winterdrp.processors.base_processor import ProcessorWithCache
+from winterdrp.processors.base_processor import ProcessorWithCache, ProcessorPremadeCache
 from winterdrp.processors.flat import SkyFlatCalibrator
 from winterdrp.paths import cal_output_dir, saturate_key
 
@@ -42,7 +42,8 @@ def _apply_to_images(
 
         return images, headers
 
-
+class MasterSkyCalibrator(ProcessorPremadeCache, NightSkyMedianCalibrator):
+    pass
 # class OldNightSkyMedianCalibrator(
 #     NightSkyMedianCalibrator,
 #     OldSkyFlatCalibrator