lsst.pipe.tasks gec3662e80e+34949ea1e7
visualization.py
Go to the documentation of this file.
1# This file is part of pipe_tasks.
2#
3# Developed for the LSST Data Management System.
4# This product includes software developed by the LSST Project
5# (https://www.lsst.org).
6# See the COPYRIGHT file at the top-level directory of this distribution
7# for details of code ownership.
8#
9# This program is free software: you can redistribute it and/or modify
10# it under the terms of the GNU General Public License as published by
11# the Free Software Foundation, either version 3 of the License, or
12# (at your option) any later version.
13#
14# This program is distributed in the hope that it will be useful,
15# but WITHOUT ANY WARRANTY; without even the implied warranty of
16# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17# GNU General Public License for more details.
18#
19# You should have received a copy of the GNU General Public License
20# along with this program. If not, see <https://www.gnu.org/licenses/>.
21
22from matplotlib import pyplot
23
24import lsst.geom
25
26
27def plotObservations(catalog, wcs):
28 """Plot the bounding boxes of an observation catalog (see MockCoaddTask.buildObservationCatalog)
29 using matplotlib, in the coordinates defined by the given Wcs (usually a skymap Wcs).
30 """
31 for record in catalog:
32 box = lsst.geom.Box2D(record.getBBox())
33 x = []
34 y = []
35 iWcs = record.getWcs()
36 for xi, yi in box.getCorners():
37 try:
38 coord = iWcs.pixelToSky(xi, yi)
39 xo, yo = wcs.skyToPixel(coord)
40 x.append(xo)
41 y.append(yo)
42 except Exception as e:
43 print("WARNING: point %d, %d failed: error=%s" % (xi, yi, e))
44 pyplot.fill(x, y, facecolor='r', alpha=0.1, edgecolor=None)
45
46
47def plotPatches(tractInfo):
48 """Plot the patches in a skymap tract using matplotlib.
49 """
50 nPatchX, nPatchY = tractInfo.getNumPatches()
51 for iPatchX in range(nPatchX):
52 for iPatchY in range(nPatchY):
53 patchInfo = tractInfo.getPatchInfo((iPatchX, iPatchY))
54 xp1, yp1 = list(zip(*patchInfo.getOuterBBox().getCorners()))
55 xp2, yp2 = list(zip(*patchInfo.getInnerBBox().getCorners()))
56 pyplot.fill(xp1, yp1, fill=False, edgecolor='g', linestyle='dashed')
57 pyplot.fill(xp2, yp2, fill=False, edgecolor='g')
58
59
60def plotTruth(catalog, wcs):
61 """Plot the objects in a truth catalog as dots using matplotlib, in the coordinate
62 system defined by the given Wcs.
63 """
64 xp = []
65 yp = []
66 for record in catalog:
67 x, y = wcs.skyToPixel(record.getCoord())
68 xp.append(x)
69 yp.append(y)
70 pyplot.plot(xp, yp, 'k+')
71
72
73def displayImages(root):
74 """Display coadd images in different frames, with the bounding boxes of the
75 observations that went into them overlayed.
76 """
77 import lsst.afw.display as afwDisplay
78 afwDisplay.setDefaultMaskTransparency(75)
79
80 butler = lsst.daf.persistence.Butler(root=root)
81 skyMap = butler.get("deepCoadd_skyMap")
82 tractInfo = skyMap[0]
84 coadds = [patchRef.get("deepCoadd", immediate=True)
85 for patchRef in task.iterPatchRefs(butler, tractInfo)]
86 for n, coadd in enumerate(coadds):
87 afwDisplay.Display(frame=n + 1).mtv(coadd, title="displayImages: coadd")
88 for n, coadd in enumerate(coadds):
89 afwDisplay.utils.drawCoaddInputs(coadd, frame=n + 1)
90 return butler
91
92
93def makePlots(root):
94 """Convenience function to make all matplotlib plots.
95 """
98 butler = lsst.daf.persistence.Butler(root=root)
99 skyMap = butler.get("deepCoadd_skyMap")
100 observations = butler.get("observations", tract=0)
101 truth = butler.get("truth", tract=0)
102 tractInfo = skyMap[0]
103 plotPatches(tractInfo)
104 plotObservations(observations, tractInfo.getWcs())
105 plotTruth(truth, tractInfo.getWcs())
106 pyplot.axis("scaled")
107 pyplot.show()
108 return butler