From a0765ce3f3d97a1eaf7241761ae9d3b389994761 Mon Sep 17 00:00:00 2001 From: BO ZHANG Date: Sat, 8 Jul 2023 12:08:48 +0800 Subject: [PATCH] add dm.get_refs --- csst_common/data_manager.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/csst_common/data_manager.py b/csst_common/data_manager.py index 464da37..cabe733 100644 --- a/csst_common/data_manager.py +++ b/csst_common/data_manager.py @@ -544,6 +544,28 @@ class CsstMsDataManager: f"{self.obs_id}_{detector:02d}_L1_V{self.version}_{post}" return os.path.join(self.dir_l1, fn) + def get_refs(self, detector=6): + if self.crds.is_available: + return self.crds.retry( + self.crds.get_refs, 3, + file_path=self.l0_detector(detector) + ) + else: + refs = dict() + if self.datatype == "mbi": + ref_types = ["bias", "dark", "ledflat", "shutter"] + else: + ref_types = ["bias", "dark", "ledflat"] + for ref_type in ref_types: + fp = os.path.join( + self.path_aux, + "C6.2_ref_crds", + "csst_msc_ms_{}_{:02d}_{:06d}.fits".format(detector, detector, 1) + ) + assert os.path.exists(fp), f"File not found: [{fp}]" + refs[ref_type] = fp + return refs + def get_bias(self, detector=6): """ get bias data """ fp = os.path.join( -- GitLab