LSSTApplications  20.0.0
LSSTDataManagementBasePackage
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"
25 
26 #include <memory>
27 
28 #include "ndarray/pybind11.h"
29 
32 
33 namespace py = pybind11;
34 using namespace pybind11::literals;
35 
36 namespace lsst {
37 namespace meas {
38 namespace base {
39 
40 namespace {
41 
42 using PyShapeResult = py::class_<ShapeResult, std::shared_ptr<ShapeResult>>;
43 using PyShapeResultKey = py::class_<ShapeResultKey, std::shared_ptr<ShapeResultKey>>;
44 
45 void declareShapeResult(py::module &mod) {
46  PyShapeResult cls(mod, "ShapeResult");
47 
48  cls.def(py::init<>());
49  cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ShapeCov const &>(), "xx"_a, "yy"_a, "xy"_a,
50  "matrix"_a);
51  cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ErrElement, ErrElement, ErrElement>(), "xx"_a,
52  "yy"_a, "xy"_a, "xxErr"_a, "yyErr"_a, "xyErr"_a);
53 
54  cls.def("getShape", &ShapeResult::getShape);
55  cls.def("getQuadrupole", &ShapeResult::getQuadrupole);
56  cls.def("setShape", &ShapeResult::setShape, "shape"_a);
57  cls.def("getShapeErr", &ShapeResult::getShapeErr);
58  cls.def("setShapeErr", (void (ShapeResult::*)(ShapeCov const &)) & ShapeResult::setShapeErr, "matrix"_a);
59  cls.def("setShapeErr",
60  (void (ShapeResult::*)(ErrElement, ErrElement, ErrElement)) & ShapeResult::setShapeErr,
61  "xxErr"_a, "yyErr"_a, "xyErr"_a);
62 
63  cls.def_readwrite("xx", &ShapeResult::xx);
64  cls.def_readwrite("yy", &ShapeResult::yy);
65  cls.def_readwrite("xy", &ShapeResult::xy);
66  cls.def_readwrite("xxErr", &ShapeResult::xxErr);
67  cls.def_readwrite("yyErr", &ShapeResult::yyErr);
68  cls.def_readwrite("xyErr", &ShapeResult::xyErr);
69  cls.def_readwrite("xx_yy_Cov", &ShapeResult::xx_yy_Cov);
70  cls.def_readwrite("xx_xy_Cov", &ShapeResult::xx_xy_Cov);
71  cls.def_readwrite("yy_xy_Cov", &ShapeResult::yy_xy_Cov);
72 }
73 
74 void declareShapeResultKey(py::module &mod) {
75  PyShapeResultKey cls(mod, "ShapeResultKey");
76 
77  cls.def_static("addFields", &ShapeResultKey::addFields, "schema"_a, "name"_a, "doc"_a, "uncertainty"_a,
78  "coordType"_a = afw::table::CoordinateType::PIXEL);
79 
80  cls.def(py::init<>());
81  cls.def(py::init<afw::table::QuadrupoleKey const &,
82  afw::table::CovarianceMatrixKey<ErrElement, 3> const &>(),
83  "shape"_a, "shapeErr"_a);
84  cls.def(py::init<afw::table::SubSchema const &>(), "subSchema"_a);
85 
86  cls.def("__eq__", &ShapeResultKey::operator==, py::is_operator());
87  cls.def("__ne__", &ShapeResultKey::operator!=, py::is_operator());
88 
89  cls.def("get", &ShapeResultKey::get, "record"_a);
90  cls.def("set", &ShapeResultKey::set, "record"_a, "value"_a);
91  cls.def("isValid", &ShapeResultKey::isValid);
92  cls.def("getShape", &ShapeResultKey::getShape);
93  cls.def("getShapeErr", &ShapeResultKey::getShapeErr);
94  cls.def("getIxx", &ShapeResultKey::getIxx);
95  cls.def("getIyy", &ShapeResultKey::getIyy);
96  cls.def("getIxy", &ShapeResultKey::getIxy);
97 }
98 
99 } // namespace
100 
101 PYBIND11_MODULE(shapeUtilities, mod) {
102  py::module::import("lsst.afw.table");
103 
104  declareShapeResult(mod);
105  declareShapeResultKey(mod);
106 
107  mod.def("makeShapeTransformMatrix", &makeShapeTransformMatrix, "xform"_a);
108 }
109 
110 } // namespace base
111 } // namespace meas
112 } // namespace lsst
lsst::meas::base::ShapeCov
Eigen::Matrix< ErrElement, 3, 3, Eigen::DontAlign > ShapeCov
Definition: constants.h:61
base
Definition: __init__.py:1
lsst::meas::base::PYBIND11_MODULE
PYBIND11_MODULE(shapeUtilities, mod)
Definition: shapeUtilities.cc:101
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
BaseRecord.h
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
ShapeUtilities.h
lsst::meas::base::makeShapeTransformMatrix
ShapeTrMatrix makeShapeTransformMatrix(geom::LinearTransform const &xform)
Construct a matrix suitable for transforming second moments.
Definition: ShapeUtilities.cc:143
lsst::meas::base::ErrElement
float ErrElement
Definition: constants.h:55
isValid
bool isValid
Definition: fits.cc:399
pybind11
Definition: _GenericMap.cc:40
lsst::utils.tests.init
def init()
Definition: tests.py:58
set
daf::base::PropertySet * set
Definition: fits.cc:912
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42