LSST Applications g013ef56533+7c9321ec0f,g042eb84c57+c6cfa41bc3,g199a45376c+0ba108daf9,g1fd858c14a+fcad0d0313,g210f2d0738+c0f94c6586,g262e1987ae+a7e710680e,g29ae962dfc+fb55f2edb0,g2ac17093b6+61d6563b1e,g2b1d02342f+df6f932764,g2cef7863aa+aef1011c0b,g2f7ad74990+c0f94c6586,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+53cf87ae69,g47891489e3+4316d04fff,g511e8cfd20+baa56acf6c,g53246c7159+8c5ae1fdc5,g54cd7ddccb+fd7ad03fde,g64539dfbff+c0f94c6586,g67b6fd64d1+4316d04fff,g67fd3c3899+c0f94c6586,g6985122a63+4316d04fff,g74acd417e5+ca833bee28,g786e29fd12+668abc6043,g81db2e9a8d+b2ec8e584f,g87389fa792+8856018cbb,g89139ef638+4316d04fff,g8d7436a09f+0a24083b20,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+11eb8fd5b8,gbf99507273+8c5ae1fdc5,gcdda8b9158+e4c84c9d5c,gce8aa8abaa+8c5ae1fdc5,gd7ef33dd92+4316d04fff,gdab6d2f7ff+ca833bee28,ge410e46f29+4316d04fff,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.40
LSST Data Management Base Package
Loading...
Searching...
No Matches
centroidUtilities.cc
Go to the documentation of this file.
1/*
2 * LSST Data Management System
3 * Copyright 2008-2017 AURA/LSST.
4 *
5 * This product includes software developed by the
6 * LSST Project (http://www.lsst.org/).
7 *
8 * This program is free software: you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation, either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the LSST License Statement and
19 * the GNU General Public License along with this program. If not,
20 * see <https://www.lsstcorp.org/LegalNotices/>.
21 */
22
23#include "pybind11/pybind11.h"
24#include "pybind11/eigen.h"
26
27#include <memory>
28
30
31namespace py = pybind11;
32using namespace pybind11::literals;
33
34namespace lsst {
35namespace meas {
36namespace base {
37
38namespace {
39
40using PyCentroidChecker = py::class_<CentroidChecker>;
41using PyCentroidResult = py::classh<CentroidResult>;
42using PyCentroidResultKey = py::classh<CentroidResultKey>;
43using PyCentroidTransform = py::classh<CentroidTransform, BaseTransform>;
44
45void declareCentroidResult(lsst::cpputils::python::WrapperCollection &wrappers) {
46 wrappers.wrapType(PyCentroidResult(wrappers.module, "CentroidResult"), [](auto &mod, auto &cls) {
47 cls.def_readwrite("x", &CentroidResult::x);
48 cls.def_readwrite("y", &CentroidResult::y);
49 cls.def_readwrite("xErr", &CentroidResult::xErr);
50 cls.def_readwrite("yErr", &CentroidResult::yErr);
51 cls.def_readwrite("x_y_Cov", &CentroidResult::x_y_Cov);
52
53 cls.def(py::init<>());
54 cls.def(py::init<CentroidElement, CentroidElement, CentroidCov const &>(), "x"_a, "y"_a, "matrix"_a);
55 cls.def(py::init<CentroidElement, CentroidElement, ErrElement, ErrElement>(), "x"_a, "y"_a, "xErr"_a,
56 "yErr"_a);
57
58 cls.def("getCentroid", &CentroidResult::getCentroid);
59 cls.def("setCentroid", &CentroidResult::setCentroid, "centroid"_a);
60 cls.def("getPoint", &CentroidResult::getPoint);
61 cls.def("getCentroidErr", &CentroidResult::getCentroidErr);
62 cls.def("setCentroidErr",
63 (void(CentroidResult::*)(CentroidCov const &)) & CentroidResult::setCentroidErr, "matrix"_a);
64 cls.def("setCentroidErr",
65 (void(CentroidResult::*)(ErrElement, ErrElement)) & CentroidResult::setCentroidErr, "xErr"_a,
66 "yErr"_a);
67 cpputils::python::addOutputOp(cls, "__str__");
68 cpputils::python::addOutputOp(cls, "__repr__");
69 });
70}
71
72void declareCentroidResultKey(lsst::cpputils::python::WrapperCollection &wrappers) {
73 wrappers.wrapType(PyCentroidResultKey(wrappers.module, "CentroidResultKey"), [](auto &mod, auto &cls) {
74 cls.def(py::init<>());
75 cls.def(py::init<afw::table::PointKey<CentroidElement> const &,
76 afw::table::CovarianceMatrixKey<ErrElement, 2> const &>(),
77 "centroid"_a, "uncertainty"_a);
78 cls.def(py::init<afw::table::SubSchema const &>(), "subSchema"_a);
79
80 cls.def_static("addFields", &CentroidResultKey::addFields, "schema"_a, "name"_a, "doc"_a,
81 "uncertainty"_a);
82
83 cls.def("__eq__", &CentroidResultKey::operator==, py::is_operator());
84 cls.def("__nq__", &CentroidResultKey::operator!=, py::is_operator());
85
86 cls.def("get", &CentroidResultKey::get, "record"_a);
87 cls.def("set", &CentroidResultKey::set, "record"_a, "value"_a);
88 cls.def("isValid", &CentroidResultKey::isValid);
89 cls.def("getCentroid", &CentroidResultKey::getCentroid);
90 cls.def("getCentroidErr", &CentroidResultKey::getCentroidErr);
91 cls.def("getX", &CentroidResultKey::getX);
92 cls.def("getY", &CentroidResultKey::getY);
93 });
94}
95
96void declareCentroidTransform(lsst::cpputils::python::WrapperCollection &wrappers) {
97 wrappers.wrapType(PyCentroidTransform(wrappers.module, "CentroidTransform"), [](auto &mod, auto &cls) {
98 cls.def(py::init<std::string const &, afw::table::SchemaMapper &>(), "name"_a, "mapper"_a);
99
100 cls.def("__call__", &CentroidTransform::operator(), "inputCatalog"_a, "outputCatalog"_a, "wcs"_a,
101 "photoCalib"_a);
102 });
103}
104
105void declareCentroidChecker(lsst::cpputils::python::WrapperCollection &wrappers) {
106 wrappers.wrapType(PyCentroidChecker(wrappers.module, "CentroidChecker"), [](auto &mod, auto &cls) {
107 cls.def(py::init<afw::table::Schema &, std::string const &, bool, double>(), "schema"_a, "name"_a,
108 "inside"_a = true, "maxDistFromPeak"_a = -1.0);
109
110 cls.def("__call__", &CentroidChecker::operator(), "record"_a);
111 });
112}
113
114void declareUncertaintyEnum(lsst::cpputils::python::WrapperCollection &wrappers) {
115 wrappers.wrapType(py::enum_<UncertaintyEnum>(wrappers.module, "UncertaintyEnum"),
116 [](auto &mod, auto &enm) {
117 enm.value("NO_UNCERTAINTY", UncertaintyEnum::NO_UNCERTAINTY);
118 enm.value("SIGMA_ONLY", UncertaintyEnum::SIGMA_ONLY);
119 enm.value("FULL_COVARIANCE", UncertaintyEnum::FULL_COVARIANCE);
120 });
121}
122
123} // namespace
124
126 declareCentroidResult(wrappers);
127 declareCentroidResultKey(wrappers);
128 declareCentroidTransform(wrappers);
129 declareCentroidChecker(wrappers);
130 declareUncertaintyEnum(wrappers);
131}
132
133} // namespace base
134} // namespace meas
135} // namespace lsst
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition python.h:242
void wrapCentroidUtilities(WrapperCollection &)