Commit 06e75dc6 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

updated data manager module for doc

parent d91d8fc8

csst/msc/data_manager.py

+26 −11
Original line number Diff line number Diff line 
import glob

import os

import re

import logging



from astropy.io import fits



from .backbone import VER_SIMS, CCD_ID_LIST, CCD_FILTER_MAPPING

from .backbone import VER_SIMS, CCD_FILTER_MAPPING





class CsstMscDataManager:
@@ -166,11 +165,19 @@ class CsstMscDataManager: 
        return os.path.join(self.dir_l1, fn)



    def pc_combined_image(self, suffix="img", ext="fits"):

        """ combined image

        """ combined images



        Parameters

        ----------

        suffix:

            {"img", "wht", "flg", "cat"}

        ext:

            {"fits"}

            {"fits", }



        Returns

        -------

        combined image path



        """

        fn = "combined_" + "{}.{}".format(suffix, ext)

        return os.path.join(self.dir_l1, fn)
@@ -232,6 +239,7 @@ if __name__ == "__main__": 
    # test C3

    import os

    from csst.msc.data_manager import CsstMscDataManager



    dm = CsstMscDataManager(

        ver_sim="C3", dir_l0="/data/L1Pipeline/msc/MSC_0000020", dir_l1="/data/L1Pipeline/msc/work")

    print("----- L0 images -----")
@@ -252,16 +260,23 @@ if __name__ == "__main__": 
    import os

    from csst.msc.data_manager import CsstMscDataManager

    from csst.msc.backbone import CCD_ID_LIST



    dm = CsstMscDataManager(

        ver_sim="C5.1", dir_l0="/data/sim_data/MSC_0000100", dir_l1="/home/user/L1Pipeline/msc/work")

    print("----- available ccd_ids -----")

    print(dm.available_ccd_ids)

    for ccd_id in dm.available_ccd_ids:

    for ccd_id in dm.available_ccd_ids[:2]:

        print("----- L0 images -----")

        print(dm.l0_sci(ccd_id=ccd_id))

        print(os.path.exists(dm.l0_sci(ccd_id=ccd_id)))

        print("----- L0 crs -----")

        print(dm.l0_crs(ccd_id=ccd_id))

        print(os.path.exists(dm.l0_sci(ccd_id=ccd_id)))

        print("----- L0 input cat -----")

        print(dm.l0_cat(ccd_id=ccd_id))

        print(os.path.exists(dm.l0_cat(ccd_id=ccd_id)))

        print("----- L0 input log -----")

        print(dm.l0_log(ccd_id=ccd_id))

        print(os.path.exists(dm.l0_log(ccd_id=ccd_id)))

        print("----- L1 images -----")

        print(dm.l1_sci(ccd_id, "img", "fits"))