made minor changes

Author: christinawilliams

DP1/200_Data_Products/206_Deblender_Products/206_2_Deblender_footprints.ipynb

@@ -96,15 +96,13 @@
     "import matplotlib.pyplot as plt\n",
     "\n",
     "from lsst.daf.butler import Butler\n",
-    "\n",
     "from lsst.rsp import get_tap_service\n",
     "import lsst.geom as geom\n",
-    "\n",
     "import lsst.afw.display as afwDisplay\n",
     "from lsst.afw.image import MultibandExposure\n",
     "\n",
     "import lsst.scarlet.lite as sl\n",
-    "import lsst.meas.extensions.scarlet as mes\n"
+    "import lsst.meas.extensions.scarlet as mes"
    ]
   },
   {
@@ -176,7 +174,7 @@
    "id": "16eeb034-6728-406e-bfc8-7b875c3a9416",
    "metadata": {},
    "source": [
-    "## 2. Identify a blend¶\n",
+    "## 2. Identify a blend\n",
     "\n",
     "This first section will identify a blend in DP1, query the `Object` table for the deblended child objects, and visualize the blend.\n"
    ]
@@ -223,7 +221,7 @@
     "        \"FROM dp1.Object \" + \\\n",
     "        \"WHERE CONTAINS(POINT('ICRS', coord_ra, coord_dec ), \" + \\\n",
     "        \"CIRCLE('ICRS', \" + str(ra) + \", \" + str(dec) + \", .1)) = 1 \" + \\\n",
-    "        \"AND parentObjectId = \" + str(parentId)\n"
+    "        \"AND parentObjectId = \" + str(parentId)"
    ]
   },
   {
@@ -280,94 +278,85 @@
    "id": "4288252b-6639-42de-82aa-70db8252f284",
    "metadata": {},
    "source": [
-    "To visualize the blend, first define a function to generate an image cutout. "
+    "To visualize the blend, plot a `deep_coadd` image subset that contains the blend with children identified. First, find the patch and tract containing the blend, and set the band to be i-band."
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "a1bf0635-8a88-42ae-b48d-79e69213f710",
+   "id": "245bec12-edcf-4564-811e-9de0d06fd3e5",
    "metadata": {},
    "outputs": [],
    "source": [
-    "def cutout_coadd(butler, ra, dec, band='i', datasetType='deepCoadd',\n",
-    "                 skymap=None, cutoutSideLength=51, **kwargs):\n",
-    "    \"\"\"\n",
-    "    Produce a cutout from a coadd at the given ra, dec position.\n",
-    "\n",
-    "    Adapted from DC2 tutorial notebook by Michael Wood-Vasey.\n",
-    "\n",
-    "    Parameters\n",
-    "    ----------\n",
-    "    butler: lsst.daf.persistence.Butler\n",
-    "        Helper object providing access to a data repository\n",
-    "    ra: float\n",
-    "        Right ascension of the center of the cutout, in degrees\n",
-    "    dec: float\n",
-    "        Declination of the center of the cutout, in degrees\n",
-    "    band: string\n",
-    "        Filter of the image to load\n",
-    "    datasetType: string ['deepCoadd']\n",
-    "        Which type of coadd to load.  Doesn't support 'calexp'\n",
-    "    skymap: lsst.afw.skyMap.SkyMap [optional]\n",
-    "        Pass in to avoid the Butler read.  Useful if you have lots of them.\n",
-    "    cutoutSideLength: float [optional]\n",
-    "        Size of the cutout region in pixels.\n",
-    "\n",
-    "    Returns\n",
-    "    -------\n",
-    "    MaskedImage\n",
-    "    \"\"\"\n",
-    "    radec = geom.SpherePoint(ra, dec, geom.degrees)\n",
-    "    cutoutSize = geom.ExtentI(cutoutSideLength, cutoutSideLength)\n",
-    "\n",
-    "    if skymap is None:\n",
-    "        skymap = butler.get(\"skyMap\")\n",
+    "radec = geom.SpherePoint(children['coord_ra'][0],children['coord_dec'][0],geom.degrees)\n",
+    "skymap = butler.get(\"skyMap\")\n",
     "\n",
-    "    tractInfo = skymap.findTract(radec)\n",
-    "    patchInfo = tractInfo.findPatch(radec)\n",
-    "    xy = geom.PointI(tractInfo.getWcs().skyToPixel(radec))\n",
-    "    bbox = geom.BoxI(xy - cutoutSize // 2, cutoutSize)\n",
-    "    print(\"bbox = \", bbox, \"xy = \", xy, \"cutoutSize = \", cutoutSize)\n",
-    "    patch = tractInfo.getSequentialPatchIndex(patchInfo)\n",
-    "    tract = tractInfo.getId()\n",
+    "tractInfo = skymap.findTract(radec)\n",
     "\n",
-    "    parameters = {'bbox': bbox}\n",
-    "\n",
-    "    query = \"\"\"band.name = '{}' AND patch = {} AND tract = {}\n",
-    "        \"\"\".format(band, patch, tract)\n",
-    "    print(query)\n",
-    "\n",
-    "    dataset_refs = butler.query_datasets(\"deep_coadd\", where=query)\n",
-    "\n",
-    "    cutout_image = butler.get(dataset_refs[0], parameters=parameters)\n",
-    "\n",
-    "    return cutout_image"
+    "tract = tractInfo.getId()\n",
+    "patch = tractInfo.findPatch(radec).getSequentialIndex()\n",
+    "band='i'"
    ]
   },
   {
    "cell_type": "markdown",
-   "id": "5546408c-6009-4230-9ef4-2faf1c302940",
+   "id": "80f1ea90-55d5-43b4-b6da-0256baf8702c",
    "metadata": {},
    "source": [
-    "Make a cutout of the `deep_coadd` imaging, using the `cutout_coadd` function to visualize blend and its child objects. Set the size of the cutout to be 5x the square root of the total `footprintArea` of the blend.\n",
-    " "
+    "Next, define the dimensions of the image subset to be 5x the square root of the total `footprintArea` of the blend."
    ]
   },
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "b2f6c287-7502-4343-89e3-e76cb3e98e87",
+   "id": "f845cc95-c731-40d2-a202-eea9e2681afe",
    "metadata": {},
    "outputs": [],
    "source": [
     "total_footprint_area = np.sum(children['footprintArea'])\n",
-    "\n",
-    "cutout = cutout_coadd(butler, children['coord_ra'][0],\n",
-    "                      children['coord_dec'][0],\n",
-    "                      band='i', datasetType='deepCoadd',\n",
-    "                      cutoutSideLength=5.0\n",
-    "                      * np.sqrt(total_footprint_area))"
+    "cutoutSideLength=5.0 * np.sqrt(total_footprint_area)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "a1922530-e638-4514-bce0-7087f6696c32",
+   "metadata": {},
+   "source": [
+    "Define the bounding box for the image subset. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ab876741-2e6a-491e-bdec-de5ca72646bf",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "cutoutSize = geom.ExtentI(cutoutSideLength, cutoutSideLength)\n",
+    "xy = geom.PointI(tractInfo.getWcs().skyToPixel(radec))\n",
+    "bbox = geom.BoxI(xy - cutoutSize // 2, cutoutSize)\n",
+    "parameters = {'bbox': bbox}"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "0ffb1405-2389-41a4-b91c-3a65729f21fb",
+   "metadata": {},
+   "source": [
+    "Finally, send the bounding box to the butler and request the image subset from the `deep_coadd` to visualize blend and its child objects. "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8fdd4188-33ae-4b30-8513-52d82e0dce9d",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "query = \"\"\"band.name = '{}' AND patch = {} AND tract = {}\n",
+    "    \"\"\".format(band, patch, tract)\n",
+    "dataset_refs = butler.query_datasets(\"deep_coadd\", where=query)\n",
+    "cutout = butler.get(dataset_refs[0], parameters=parameters)"
    ]
   },
   {
@@ -428,7 +417,7 @@
     "registry = butler.registry\n",
     "\n",
     "for dt in sorted(registry.queryDatasetTypes('*scarlet*')):\n",
-    "    print(dt)\n"
+    "    print(dt)"
    ]
   },
   {
@@ -448,7 +437,7 @@
    "source": [
     "query = \"\"\"band.name = '{}' AND patch.region OVERLAPS POINT({}, {})\n",
     "        \"\"\".format('i', children['coord_ra'][0], children['coord_dec'][0])\n",
-    "print(query)\n"
+    "print(query)"
    ]
   },
   {
@@ -467,7 +456,7 @@
    "outputs": [],
    "source": [
     "dataset_refs = butler.query_datasets(\"deep_coadd\", where=query)\n",
-    "dataId = dataset_refs[0].dataId\n"
+    "dataId = dataset_refs[0].dataId"
    ]
   },
   {
@@ -533,7 +522,6 @@
     "from numpy.typing import DTypeLike\n",
     "from lsst.scarlet.lite import Box, Blend, Observation\n",
     "\n",
-    "\n",
     "def minimal_data_to_blend(cls, model_psf: np.ndarray, dtype: DTypeLike) -> Blend:\n",
     "\n",
     "    \"\"\"Convert the storage data model into a scarlet lite blend\n",