LSST Applications 27.0.0,g0265f82a02+469cd937ee,g02d81e74bb+21ad69e7e1,g1470d8bcf6+cbe83ee85a,g2079a07aa2+e67c6346a6,g212a7c68fe+04a9158687,g2305ad1205+94392ce272,g295015adf3+81dd352a9d,g2bbee38e9b+469cd937ee,g337abbeb29+469cd937ee,g3939d97d7f+72a9f7b576,g487adcacf7+71499e7cba,g50ff169b8f+5929b3527e,g52b1c1532d+a6fc98d2e7,g591dd9f2cf+df404f777f,g5a732f18d5+be83d3ecdb,g64a986408d+21ad69e7e1,g858d7b2824+21ad69e7e1,g8a8a8dda67+a6fc98d2e7,g99cad8db69+f62e5b0af5,g9ddcbc5298+d4bad12328,ga1e77700b3+9c366c4306,ga8c6da7877+71e4819109,gb0e22166c9+25ba2f69a1,gb6a65358fc+469cd937ee,gbb8dafda3b+69d3c0e320,gc07e1c2157+a98bf949bb,gc120e1dc64+615ec43309,gc28159a63d+469cd937ee,gcf0d15dbbd+72a9f7b576,gdaeeff99f8+a38ce5ea23,ge6526c86ff+3a7c1ac5f1,ge79ae78c31+469cd937ee,gee10cc3b42+a6fc98d2e7,gf1cff7945b+21ad69e7e1,gfbcc870c63+9a11dc8c8f
LSST Data Management Base Package
Loading...
Searching...
No Matches
_source.cc
Go to the documentation of this file.
1/*
2 * This file is part of afw.
3 *
4 * Developed for the LSST Data Management System.
5 * This product includes software developed by the LSST Project
6 * (https://www.lsst.org).
7 * See the COPYRIGHT file at the top-level directory of this distribution
8 * for details of code ownership.
9 *
10 * This program is free software: you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation, either version 3 of the License, or
13 * (at your option) any later version.
14 *
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program. If not, see <https://www.gnu.org/licenses/>.
22 */
23
24#include "pybind11/pybind11.h"
25#include "pybind11/eigen.h"
26
27#include <memory>
28
29#include "ndarray/pybind11.h"
30
31#include "lsst/utils/python.h"
32
42
43namespace py = pybind11;
44using namespace pybind11::literals;
45
46namespace lsst {
47namespace afw {
48namespace table {
49
50using utils::python::WrapperCollection;
51
52namespace {
53
54using PySourceRecord = py::class_<SourceRecord, std::shared_ptr<SourceRecord>, SimpleRecord>;
55using PySourceTable = py::class_<SourceTable, std::shared_ptr<SourceTable>, SimpleTable>;
56using PySourceColumnView =
57 py::class_<SourceColumnViewT<SourceRecord>, std::shared_ptr<SourceColumnViewT<SourceRecord>>,
58 ColumnViewT<SourceRecord>>;
59
60/*
61Declare member and static functions for a pybind11 wrapper of SourceRecord
62*/
63PySourceRecord declareSourceRecord(WrapperCollection &wrappers) {
64 return wrappers.wrapType(PySourceRecord(wrappers.module, "SourceRecord"), [](auto &mod, auto &cls) {
65 cls.def("getFootprint", &SourceRecord::getFootprint);
66 cls.def("setFootprint", &SourceRecord::setFootprint);
67 cls.def("getTable", &SourceRecord::getTable);
68 cls.def_property_readonly("table", &SourceRecord::getTable);
69
70 cls.def("getParent", &SourceRecord::getParent);
71 cls.def("setParent", &SourceRecord::setParent, "id"_a);
72
73 cls.def("getPsfInstFlux", &SourceRecord::getPsfInstFlux);
74 cls.def("getPsfInstFluxErr", &SourceRecord::getPsfInstFluxErr);
75 cls.def("getPsfFluxFlag", &SourceRecord::getPsfFluxFlag);
76
77 cls.def("getModelInstFlux", &SourceRecord::getModelInstFlux);
78 cls.def("getModelInstFluxErr", &SourceRecord::getModelInstFluxErr);
79 cls.def("getModelFluxFlag", &SourceRecord::getModelFluxFlag);
80
81 cls.def("getApInstFlux", &SourceRecord::getApInstFlux);
82 cls.def("getApInstFluxErr", &SourceRecord::getApInstFluxErr);
83 cls.def("getApFluxFlag", &SourceRecord::getApFluxFlag);
84
85 cls.def("getGaussianInstFlux", &SourceRecord::getGaussianInstFlux);
86 cls.def("getGaussianInstFluxErr", &SourceRecord::getGaussianInstFluxErr);
87 cls.def("getGaussianFluxFlag", &SourceRecord::getGaussianFluxFlag);
88
89 cls.def("getCalibInstFlux", &SourceRecord::getCalibInstFlux);
90 cls.def("getCalibInstFluxErr", &SourceRecord::getCalibInstFluxErr);
91 cls.def("getCalibFluxFlag", &SourceRecord::getCalibFluxFlag);
92
93 cls.def("getCentroid", &SourceRecord::getCentroid);
94 cls.def("getCentroidErr", &SourceRecord::getCentroidErr);
95 cls.def("getCentroidFlag", &SourceRecord::getCentroidFlag);
96
97 cls.def("getShape", &SourceRecord::getShape);
98 cls.def("getShapeErr", &SourceRecord::getShapeErr);
99 cls.def("getShapeFlag", &SourceRecord::getShapeFlag);
100
101 cls.def("getX", &SourceRecord::getX);
102 cls.def("getY", &SourceRecord::getY);
103 cls.def("getIxx", &SourceRecord::getIxx);
104 cls.def("getIyy", &SourceRecord::getIyy);
105 cls.def("getIxy", &SourceRecord::getIxy);
106 cls.def("updateCoord", (void (SourceRecord::*)(geom::SkyWcs const &, bool)) & SourceRecord::updateCoord,
107 "wcs"_a, "include_covariance"_a=true);
108 cls.def("updateCoord",
109 (void (SourceRecord::*)(geom::SkyWcs const &, PointKey<double> const &, bool)) &
110 SourceRecord::updateCoord,
111 "wcs"_a, "key"_a, "include_covariance"_a=true);
112 });
113}
114
115/*
116Declare member and static functions for a pybind11 wrapper of SourceTable
117*/
118PySourceTable declareSourceTable(WrapperCollection &wrappers) {
119 return wrappers.wrapType(PySourceTable(wrappers.module, "SourceTable"), [](auto &mod, auto &cls) {
120 cls.def("clone", &SourceTable::clone);
121 cls.def_static("make",
122 (std::shared_ptr<SourceTable>(*)(Schema const &, std::shared_ptr<IdFactory> const &)) &
123 SourceTable::make);
124 cls.def_static("make", (std::shared_ptr<SourceTable>(*)(Schema const &)) & SourceTable::make);
125 cls.def_static("makeMinimalSchema", &SourceTable::makeMinimalSchema);
126 cls.def_static("getParentKey", &SourceTable::getParentKey);
127 cls.def("copyRecord", (std::shared_ptr<SourceRecord>(SourceTable::*)(BaseRecord const &)) &
128 SourceTable::copyRecord);
129 cls.def("copyRecord",
130 (std::shared_ptr<SourceRecord>(SourceTable::*)(BaseRecord const &, SchemaMapper const &)) &
131 SourceTable::copyRecord);
132 cls.def("makeRecord", &SourceTable::makeRecord);
133
134 cls.def("getPsfFluxSlot", &SourceTable::getPsfFluxSlot);
135 cls.def("definePsfFlux", &SourceTable::definePsfFlux, "name"_a);
136
137 cls.def("getModelFluxSlot", &SourceTable::getModelFluxSlot);
138 cls.def("defineModelFlux", &SourceTable::defineModelFlux, "name"_a);
139
140 cls.def("getApFluxSlot", &SourceTable::getApFluxSlot);
141 cls.def("defineApFlux", &SourceTable::defineApFlux, "name"_a);
142
143 cls.def("getGaussianFluxSlot", &SourceTable::getGaussianFluxSlot);
144 cls.def("defineGaussianFlux", &SourceTable::defineGaussianFlux, "name"_a);
145
146 cls.def("getCalibFluxSlot", &SourceTable::getCalibFluxSlot);
147 cls.def("defineCalibFlux", &SourceTable::defineCalibFlux, "name"_a);
148
149 cls.def("getCentroidSlot", &SourceTable::getCentroidSlot);
150 cls.def("defineCentroid", &SourceTable::defineCentroid, "name"_a);
151
152 cls.def("getShapeSlot", &SourceTable::getShapeSlot);
153 cls.def("defineShape", &SourceTable::defineShape, "name"_a);
154 });
155}
156
157PySourceColumnView declareSourceColumnView(WrapperCollection &wrappers) {
158 table::python::declareColumnView<SourceRecord>(wrappers, "Source", true);
159 return wrappers.wrapType(PySourceColumnView(wrappers.module, "SourceColumnView"),
160 [](auto &mod, auto &cls) {
161 using Class = SourceColumnViewT<SourceRecord>;
162 cls.def("getPsfInstFlux", &Class::getPsfInstFlux);
163 cls.def("getPsfInstFluxErr", &Class::getPsfInstFluxErr);
164 cls.def("getApInstFlux", &Class::getApInstFlux);
165 cls.def("getApInstFluxErr", &Class::getApInstFluxErr);
166 cls.def("getModelInstFlux", &Class::getModelInstFlux);
167 cls.def("getModelInstFluxErr", &Class::getModelInstFluxErr);
168 cls.def("getGaussianInstFlux", &Class::getGaussianInstFlux);
169 cls.def("getGaussianInstFluxErr", &Class::getGaussianInstFluxErr);
170 cls.def("getCalibInstFlux", &Class::getCalibInstFlux);
171 cls.def("getCalibInstFluxErr", &Class::getCalibInstFluxErr);
172 cls.def("getX", &Class::getX);
173 cls.def("getY", &Class::getY);
174 cls.def("getIxx", &Class::getIxx);
175 cls.def("getIyy", &Class::getIyy);
176 cls.def("getIxy", &Class::getIxy);
177 });
178}
179
180} // namespace
181
182void wrapSource(WrapperCollection &wrappers) {
183 // TODO: uncomment once afw.geom uses WrapperCollection
184 // wrappers.addSignatureDependency("lsst.afw.geom.ellipses");
185
186 // SourceFitsFlags enum values are used as integer masks, so wrap as attributes instead of an enum
187 // static_cast is required to avoid an import error (py::cast and py::int_ do not work by themselves
188 // and are not required with the static_cast)
189 auto &mod = wrappers.module;
190 mod.attr("SOURCE_IO_NO_FOOTPRINTS") = static_cast<int>(SourceFitsFlags::SOURCE_IO_NO_FOOTPRINTS);
191 mod.attr("SOURCE_IO_NO_HEAVY_FOOTPRINTS") =
193
194 auto clsSourceRecord = declareSourceRecord(wrappers);
195 auto clsSourceTable = declareSourceTable(wrappers);
197 auto clsSourceCatalog = table::python::declareSortedCatalog<SourceRecord>(wrappers, "Source");
198
199 clsSourceRecord.attr("Table") = clsSourceTable;
200 clsSourceRecord.attr("ColumnView") = clsSourceColumnView;
201 clsSourceRecord.attr("Catalog") = clsSourceCatalog;
202 clsSourceTable.attr("Record") = clsSourceRecord;
203 clsSourceTable.attr("ColumnView") = clsSourceColumnView;
204 clsSourceTable.attr("Catalog") = clsSourceCatalog;
205 clsSourceCatalog.attr("Record") = clsSourceRecord;
206 clsSourceCatalog.attr("Table") = clsSourceTable;
207 clsSourceCatalog.attr("ColumnView") = clsSourceColumnView;
208}
209
210} // namespace table
211} // namespace afw
212} // namespace lsst
Tag types used to declare specialized field types.
Definition misc.h:31
void wrapSource(WrapperCollection &wrappers)
Definition _source.cc:182
@ SOURCE_IO_NO_FOOTPRINTS
Do not read/write footprints at all.
Definition Source.h:53
@ SOURCE_IO_NO_HEAVY_FOOTPRINTS
Read/write heavy footprints as non-heavy footprints.
Definition Source.h:54