Commit 5d498148 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

update dag -> detectors

parent bb631b05

csst_dag/dag/dags.py

+9 −9
Original line number Diff line number Diff line
@@ -4,7 +4,7 @@ import numpy as np 
from csst_dfs_client import plan, level0



from ._base_dag import BaseDAG

from ..csst import csst

from .._csst import csst





# DAG_DETECTOR_NAMES = {
@@ -23,20 +23,20 @@ from ..csst import csst 


def get_detector_names_via_dag(dag: str) -> list[str]:

    if "msc" in dag:

        if "mbi" in dag:

            return csst["msc"]["mbi"].effective_detector_names

        if "mbi" in dag or "ast" in dag:

            return csst["MSC"]["MBI"].effective_detector_names

        elif "sls" in dag:

            return csst["msc"]["sls"].effective_detector_names

            return csst["MSC"]["SLS"].effective_detector_names

        else:

            return csst["msc"].effective_detector_names

            return csst["MSC"].effective_detector_names

    elif "mci" in dag:

        return csst["mci"].effective_detector_names

        return csst["MCI"].effective_detector_names

    elif "ifs" in dag:

        return csst["ifs"].effective_detector_names

        return csst["IFS"].effective_detector_names

    elif "cpic" in dag:

        return csst["cpic"].effective_detector_names

        return csst["CPIC"].effective_detector_names

    elif "hstdm" in dag:

        return csst["hstdm"].effective_detector_names

        return csst["HSTDM"].effective_detector_names





class GeneralDAGViaObsid(BaseDAG):