From d56d2f22e16ef1a781b8f52761c4dd7a94d10de7 Mon Sep 17 00:00:00 2001
From: Dan Taranu <dtaranu@astro.princeton.edu>
Date: Wed, 26 Jun 2024 11:08:40 -0700
Subject: [PATCH] Explicitly copy extendedness columns for matched tables

---
 pipelines/_ingredients/LSSTCam-imSim/DRP.yaml | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/pipelines/_ingredients/LSSTCam-imSim/DRP.yaml b/pipelines/_ingredients/LSSTCam-imSim/DRP.yaml
index fcac861d..03442450 100644
--- a/pipelines/_ingredients/LSSTCam-imSim/DRP.yaml
+++ b/pipelines/_ingredients/LSSTCam-imSim/DRP.yaml
@@ -71,6 +71,7 @@ tasks:
   compareObjectToTruth:
     class: lsst.pipe.tasks.diff_matched_tract_catalog.DiffMatchedTractCatalogTask
     config:
+      columns_ref_copy: ["is_pointsource"]
       columns_target_coord_err: ["coord_raErr", "coord_decErr"]
       coord_format.coords_ref_to_convert: {"ra": "x", "dec": "y"}
       # Convert ref ra/dec to x/y but do not measure distances from x/y
@@ -88,9 +89,12 @@ tasks:
             )
             fluxes_meas_psf.append(f"{band}_psfFlux")
         config.columns_flux = columns_flux
-        config.columns_target_copy = ["patch"] + fluxes_meas_psf + [
-          f"{col}Err" for col in fluxes_meas_psf
-        ] + ["x", "y", "xErr", "yErr"]
+        # Extendedness columns are needed for downstream plots/metrics/etc.
+        columns_target_copy = ["patch", "refExtendedness", "refSizeExtendedness"]
+        columns_target_copy.extend(fluxes_meas_psf)
+        columns_target_copy.extend([f"{col}Err" for col in fluxes_meas_psf])
+        columns_target_copy.extend(["x", "y", "xErr", "yErr"])
+        config.columns_target_copy = columns_target_copy
   catalogMatchTract:
     class: lsst.analysis.tools.tasks.astrometricCatalogMatch.AstrometricCatalogMatchTask
     config: