diff --git a/csst_common/data_manager.py b/csst_common/data_manager.py index d6eff406c6428d52e3a6d842f877d7d8953a6016..bab8e034aaa7f448b3688032934d3af008f74f3f 100644 --- a/csst_common/data_manager.py +++ b/csst_common/data_manager.py @@ -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