updated path_aux for sls

csst_common/data_manager.py

@@ -271,20 +271,29 @@ class CsstMsDataManager:
         print("final target detector IDs are ", self.target_detectors)
         return
 
-    def get_bias(self, detector=6) -> np.ndarray:
+    def get_bias(self, detector=6):
         """ get bias data """
-        fp = glob.glob(self.path_aux.format("CLB", detector))[0]
-        return fits.getdata(fp)
+        if self.datatype == "mbi":
+            fp = glob.glob(self.path_aux.format("CLB", detector))[0]
+            return fits.getdata(fp)
+        else:
+            return self.path_aux.format(detector, "bias")
 
-    def get_dark(self, detector=6) -> np.ndarray:
+    def get_dark(self, detector=6):
         """ get dark data """
-        fp = glob.glob(self.path_aux.format("CLD", detector))[0]
-        return fits.getdata(fp)
+        if self.datatype == "mbi":
+            fp = glob.glob(self.path_aux.format("CLD", detector))[0]
+            return fits.getdata(fp)
+        else:
+            return self.path_aux.format(detector, "dark")
 
-    def get_flat(self, detector=6) -> np.ndarray:
+    def get_flat(self, detector=6):
         """ get flat data """
-        fp = glob.glob(self.path_aux.format("CLF", detector))[0]
-        return fits.getdata(fp)
+        if self.datatype == "mbi":
+            fp = glob.glob(self.path_aux.format("CLF", detector))[0]
+            return fits.getdata(fp)
+        else:
+            return self.path_aux.format(detector, "flat")
 
     def l1_file(self, name="", comment=""):
         """ L1 file path
@@ -341,7 +350,7 @@ class CsstMsDataManager:
         elif hostname == "dandelion" and datatype == "sls":
             dir_l0 = "/nfsdata/share/csst_simulation_data/Cycle-5-SimuData/slitlessSpectroscopy/" \
                      "NGP_AstrometryON_shearOFF_Spec/MSC_{:07d}/".format(exposure_id)
-            path_aux = ""
+            path_aux = "/nfsdata/share/csst_simulation_data/Cycle-5-SimuData/slitlessSpectroscopy/csst_{:02d}{}.fits"
             dir_pcref = ""
 
         # PMO