Commit e1cdd067 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

added interfaces for mosaic

parent 3e4d8cc0
Loading
Loading
Loading
Loading
+16 −0
Original line number Diff line number Diff line
@@ -15,6 +15,7 @@ import glob
import re

from astropy.io import fits
from astropy.table import Table

from .params import CSST_PARAMS as CP

@@ -361,6 +362,21 @@ class CsstMsDataManager:
        self.hardcode_history.append(dict(hdcd=fp, comment=comment))
        return fp

    def get_sls_info(self, use_dfs=False):
        if use_dfs:
            raise NotImplementedError()
        else:
            assert len(self.target_detectors) == 1
            header = fits.getheader(self.l0_detector(self.target_detectors[0]), ext=1)
        return header

    def get_mbi_info(self, use_dfs=False):
        if use_dfs:
            raise NotImplementedError()
        else:
            info = Table.read("/nfsdata/share/csst_simulation_data/Cycle-5-SimuData/slitlessSpectroscopy/t_mbi_l1.fits")
        return info

    @staticmethod
    def quickstart(ver_sim="C5.2", datatype="mbi", dir_l1=".", exposure_id=100):
        """Quick dataset generator for tests on dandelion or PMO