add find_plan_level0_basis and find_plan_level1_basis

csst_dag/dag/dispatcher.py → csst_dag/dag/_dispatcher.py

@@ -196,6 +196,32 @@ class Dispatcher:
         )
         return data_basis
 
+    @staticmethod
+    def find_plan_level0_basis(**kwargs) -> tuple(table.Table):
+        data_basis = Dispatcher.find_level0_basis(**kwargs)
+        plan_basis = Dispatcher.find_plan_basis(**kwargs)
+        u_data_basis = table.unique(data_basis["dataset", "obs_id"])
+        relevant_plan = table.join(
+            u_data_basis,
+            plan_basis,
+            keys=["dataset", "obs_id"],
+            join_type=PLAN_JOIN_TYPE,
+        )
+        return relevant_plan, data_basis
+
+    @staticmethod
+    def find_plan_level1_basis(**kwargs) -> tuple(table.Table):
+        data_basis = Dispatcher.find_level1_basis(**kwargs)
+        plan_basis = Dispatcher.find_plan_basis(**kwargs)
+        u_data_basis = table.unique(data_basis["dataset", "obs_id"])
+        relevant_plan = table.join(
+            u_data_basis,
+            plan_basis,
+            keys=["dataset", "obs_id"],
+            join_type=PLAN_JOIN_TYPE,
+        )
+        return relevant_plan, data_basis
+
     @staticmethod
     def dispatch_file(
         plan_basis: table.Table,