add query_ref_cat

csst_common/pipeline.py

@@ -405,7 +405,10 @@ class Pipeline:
             return task_ref_cat
         else:
             # DFS recommended catalog
-            if self.docker_image == "csst-msc-l1-mbi":
+            if self.docker_image in [
+                "csst-msc-l1-mbi",
+                "csst-msc-l1-ast-astrometry",
+            ]:
                 return "trilegal_093"
             else:
                 return None
@@ -490,13 +493,17 @@ class Pipeline:
         return meta
 
     def query_ref_cat(
-        self, file_path, catalog_name: str = "trilegal_093"
+        self,
+        file_path,
+        catalog_name: Optional[str] = None,
     ) -> None | str:
         """
         References
         ----------
         https://gea.esac.esa.int/archive/documentation/GDR3/Gaia_archive/chap_datamodel/sec_dm_main_source_catalogue/ssec_dm_gaia_source.html
         """
+        if not catalog_name:
+            catalog_name = self.ref_cat
         image = File(file_path)
         header = fits.getheader(image.file_path)
         pointing_ra = header["RA_OBJ"]