Add Sersic (+PS) model fits to pipeline

lsst · Dec 5, 2023 · d1b1dfc · d1b1dfc
1 parent 98126c0
commit d1b1dfc
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 7 deletions.
diff --git a/pipelines/fit.yaml b/pipelines/fit.yaml
@@ -71,6 +71,38 @@ tasks:
             prior_size_stddev=0.3,
           )
         }
+  fit_src_ser_multiprofit:
+    class: lsst.pipe.tasks.fit_coadd_multiband.CoaddMultibandFitTask
+    config:
+      connections.cat_output: deepCoadd_ser_multiprofit
+      python: |
+        from lsst.meas.extensions.multiprofit.fit_coadd_multiband import MultiProFitSourceTask
+        from lsst.multiprofit.componentconfig import SersicConfig, SersicIndexConfig
+        config.fit_coadd_multiband.retarget(MultiProFitSourceTask)
+        config.fit_coadd_multiband.n_pointsources = 0
+        config.fit_coadd_multiband.sersics = {
+          "ser": SersicConfig(
+            sersicindex=SersicIndexConfig(fixed=False, value_initial=1.0),
+            prior_axrat_stddev=0.8,
+            prior_size_stddev=0.3,
+          )
+        }
+  fit_src_psser_multiprofit:
+    class: lsst.pipe.tasks.fit_coadd_multiband.CoaddMultibandFitTask
+    config:
+      connections.cat_output: deepCoadd_psser_multiprofit
+      python: |
+        from lsst.meas.extensions.multiprofit.fit_coadd_multiband import MultiProFitSourceTask
+        from lsst.multiprofit.componentconfig import SersicConfig, SersicIndexConfig
+        config.fit_coadd_multiband.retarget(MultiProFitSourceTask)
+        config.fit_coadd_multiband.n_pointsources = 1
+        config.fit_coadd_multiband.sersics = {
+          "ser": SersicConfig(
+            sersicindex=SersicIndexConfig(fixed=False, value_initial=1.63),
+            prior_axrat_stddev=0.8,
+            prior_size_stddev=0.3,
+          )
+        }
   fit_src_psexpdev_multiprofit:
     class: lsst.pipe.tasks.fit_coadd_multiband.CoaddMultibandFitTask
     config:

diff --git a/pipelines/merge.yaml b/pipelines/merge.yaml
@@ -29,18 +29,24 @@ tasks:
             is_multipatch=True,
             storageClass="DataFrame",
           ),
-          "deepCoadd_psexpdev_multiprofit": InputConfig(
-            doc="Point Source + Exponential + DeVaucouleurs source fit",
-            action=MergeMultibandFluxes(name_model="psexpdev"),
-            column_id="id",
-            is_multiband=True,
-          ),
           "deepCoadd_psgauss_multiprofit": InputConfig(
             doc="Point Source + Gaussian source fit",
             action=MergeMultibandFluxes(name_model="psgauss"),
             column_id="id",
             is_multiband=True,
           ),
+          "deepCoadd_psser_multiprofit": InputConfig(
+            doc="Point Source + Sersic source fit",
+            action=MergeMultibandFluxes(name_model="psser"),
+            column_id="id",
+            is_multiband=True,
+          ),
+          "deepCoadd_psexpdev_multiprofit": InputConfig(
+            doc="Point Source + Exponential + DeVaucouleurs source fit",
+            action=MergeMultibandFluxes(name_model="psexpdev"),
+            column_id="id",
+            is_multiband=True,
+          ),
         }
   match_multiprofit:
     class: lsst.pipe.tasks.match_tract_catalog.MatchTractCatalogTask
@@ -100,7 +106,7 @@ tasks:
         if parameters.size_include:
             for ax in ("x", "y"):
                 config.columns_target_copy.append(
-                    f"{parameters.model_prefix}_{parameters.size_include}_sigma_{ax}"
+                    f"{parameters.model_prefix}_{parameters.size_include}_reff_{ax}"
                 )
         config.columns_flux = columns_flux
         config.coord_format.column_target_coord1 = f"{parameters.model_prefix}_cen_ra"