LSST Applications g04a91732dc+146a938ab0,g07dc498a13+80b84b0d75,g0fba68d861+0decac7526,g1409bbee79+80b84b0d75,g1a7e361dbc+80b84b0d75,g1fd858c14a+f6e422e056,g20f46db602+483a84333a,g21d47ad084+4a6cd485de,g35bb328faa+fcb1d3bbc8,g42c1b31a95+a1301e4c20,g4d39ba7253+9b833be27e,g4e0f332c67+5d362be553,g53246c7159+fcb1d3bbc8,g60b5630c4e+9b833be27e,g78460c75b0+2f9a1b4bcd,g786e29fd12+cf7ec2a62a,g7b71ed6315+fcb1d3bbc8,g8852436030+790117df0f,g89139ef638+80b84b0d75,g8d6b6b353c+9b833be27e,g9125e01d80+fcb1d3bbc8,g989de1cb63+80b84b0d75,g9f33ca652e+9c6b68d7f3,ga9baa6287d+9b833be27e,gaaedd4e678+80b84b0d75,gabe3b4be73+1e0a283bba,gb1101e3267+9f3571abad,gb58c049af0+f03b321e39,gb90eeb9370+691e4ab549,gc741bbaa4f+5f483edd21,gcf25f946ba+790117df0f,gd24842266e+c54cdbdbd2,gd315a588df+5b65d88fe4,gd6cbbdb0b4+c8606af20c,gd9a9a58781+fcb1d3bbc8,gde0f65d7ad+c99546153d,ge278dab8ac+932305ba37,ge82c20c137+76d20ab76d,w.2025.10
LSST Data Management Base Package
Loading...
Searching...
No Matches
shapeUtilities.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
31
32namespace py = pybind11;
33using namespace pybind11::literals;
34
35namespace lsst {
36namespace meas {
37namespace base {
38
39namespace {
40
41using PyShapeResult = py::class_<ShapeResult, std::shared_ptr<ShapeResult>>;
42using PyShapeResultKey = py::class_<ShapeResultKey, std::shared_ptr<ShapeResultKey>>;
43
44void declareShapeResult(lsst::cpputils::python::WrapperCollection &wrappers) {
45 wrappers.wrapType(PyShapeResult(wrappers.module, "ShapeResult"), [](auto &mod, auto &cls) {
46 cls.def(py::init<>());
47 cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ShapeCov const &>(), "xx"_a, "yy"_a, "xy"_a,
48 "matrix"_a);
49 cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ErrElement, ErrElement, ErrElement>(), "xx"_a,
50 "yy"_a, "xy"_a, "xxErr"_a, "yyErr"_a, "xyErr"_a);
51 cls.def("getShape", &ShapeResult::getShape);
52 cls.def("getQuadrupole", &ShapeResult::getQuadrupole);
53 cls.def("setShape", &ShapeResult::setShape, "shape"_a);
54 cls.def("getShapeErr", &ShapeResult::getShapeErr);
55 cls.def("setShapeErr", (void (ShapeResult::*)(ShapeCov const &)) &ShapeResult::setShapeErr, "matrix"_a);
56 cls.def("setShapeErr",
57 (void (ShapeResult::*)(ErrElement, ErrElement, ErrElement)) &ShapeResult::setShapeErr,
58 "xxErr"_a, "yyErr"_a, "xyErr"_a);
59 cls.def_readwrite("xx", &ShapeResult::xx);
60 cls.def_readwrite("yy", &ShapeResult::yy);
61 cls.def_readwrite("xy", &ShapeResult::xy);
62 cls.def_readwrite("xxErr", &ShapeResult::xxErr);
63 cls.def_readwrite("yyErr", &ShapeResult::yyErr);
64 cls.def_readwrite("xyErr", &ShapeResult::xyErr);
65 cls.def_readwrite("xx_yy_Cov", &ShapeResult::xx_yy_Cov);
66 cls.def_readwrite("xx_xy_Cov", &ShapeResult::xx_xy_Cov);
67 cls.def_readwrite("yy_xy_Cov", &ShapeResult::yy_xy_Cov);
68 });
69}
70
71void declareShapeResultKey(lsst::cpputils::python::WrapperCollection &wrappers) {
72 wrappers.wrapType(PyShapeResultKey(wrappers.module, "ShapeResultKey"), [](auto &mod, auto &cls) {
73 cls.def_static("addFields", &ShapeResultKey::addFields, "schema"_a, "name"_a, "doc"_a, "uncertainty"_a,
74 "coordType"_a = afw::table::CoordinateType::PIXEL);
75
76 cls.def(py::init<>());
77 cls.def(py::init<afw::table::QuadrupoleKey const &,
78 afw::table::CovarianceMatrixKey<ErrElement, 3> const &>(),
79 "shape"_a, "shapeErr"_a);
80 cls.def(py::init<afw::table::SubSchema const &>(), "subSchema"_a);
81
82 cls.def("__eq__", &ShapeResultKey::operator==, py::is_operator());
83 cls.def("__ne__", &ShapeResultKey::operator!=, py::is_operator());
84
85 cls.def("get", &ShapeResultKey::get, "record"_a);
86 cls.def("set", &ShapeResultKey::set, "record"_a, "value"_a);
87 cls.def("isValid", &ShapeResultKey::isValid);
88 cls.def("getShape", &ShapeResultKey::getShape);
89 cls.def("getShapeErr", &ShapeResultKey::getShapeErr);
90 cls.def("getIxx", &ShapeResultKey::getIxx);
91 cls.def("getIyy", &ShapeResultKey::getIyy);
92 cls.def("getIxy", &ShapeResultKey::getIxy);
93 });
94}
95
96} // namespace
97
99 declareShapeResult(wrappers);
100 declareShapeResultKey(wrappers);
101 wrappers.wrap([](auto &mod) {
102 mod.def("makeShapeTransformMatrix", &makeShapeTransformMatrix, "xform"_a);
103 });
104}
105
106} // namespace base
107} // namespace meas
108} // namespace lsst
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition python.h:242
void wrapShapeUtilities(WrapperCollection &)
ShapeTrMatrix makeShapeTransformMatrix(geom::LinearTransform const &xform)
Construct a matrix suitable for transforming second moments.