lsst.meas.base  14.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Macros Groups Pages
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 
25 #include <memory>
26 
27 #include "numpy/arrayobject.h"
28 #include "ndarray/pybind11.h"
29 
30 #include "lsst/afw/table/BaseRecord.h"
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 &>(),
50  "xx"_a, "yy"_a, "xy"_a, "matrix"_a);
51  cls.def(py::init<ShapeElement, ShapeElement, ShapeElement, ErrElement, ErrElement, ErrElement>(),
52  "xx"_a, "yy"_a, "xy"_a, "xxSigma"_a, "yySigma"_a, "xySigma"_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",
59  (void (ShapeResult::*)(ShapeCov const &)) &ShapeResult::setShapeErr, "matrix"_a);
60  cls.def("setShapeErr",
61  (void (ShapeResult::*)(ErrElement, ErrElement, ErrElement)) &ShapeResult::setShapeErr,
62  "xxSigma"_a, "yySigma"_a, "xySigma"_a);
63 
64  cls.def_readwrite("xx", &ShapeResult::xx);
65  cls.def_readwrite("yy", &ShapeResult::yy);
66  cls.def_readwrite("xy", &ShapeResult::xy);
67  cls.def_readwrite("xxSigma", &ShapeResult::xxSigma);
68  cls.def_readwrite("yySigma", &ShapeResult::yySigma);
69  cls.def_readwrite("xySigma", &ShapeResult::xySigma);
70  cls.def_readwrite("xx_yy_Cov", &ShapeResult::xx_yy_Cov);
71  cls.def_readwrite("xx_xy_Cov", &ShapeResult::xx_xy_Cov);
72  cls.def_readwrite("yy_xy_Cov", &ShapeResult::yy_xy_Cov);
73 }
74 
75 void declareShapeResultKey(py::module &mod) {
76  PyShapeResultKey cls(mod, "ShapeResultKey");
77 
78  cls.def_static("addFields", &ShapeResultKey::addFields, "schema"_a, "name"_a, "doc"_a, "uncertainty"_a,
79  "coordType"_a = afw::table::CoordinateType::PIXEL);
80 
81  cls.def(py::init<>());
82  cls.def(py::init<afw::table::QuadrupoleKey const &,
83  afw::table::CovarianceMatrixKey<ErrElement, 3> const &>(),
84  "shape"_a, "shapeErr"_a);
85  cls.def(py::init<afw::table::SubSchema const &>(), "subSchema"_a);
86 
87  cls.def("__eq__", &ShapeResultKey::operator==, py::is_operator());
88  cls.def("__ne__", &ShapeResultKey::operator!=, py::is_operator());
89 
90  cls.def("get", &ShapeResultKey::get, "record"_a);
91  cls.def("set", &ShapeResultKey::set, "record"_a, "value"_a);
92  cls.def("isValid", &ShapeResultKey::isValid);
93  cls.def("getShape", &ShapeResultKey::getShape);
94  cls.def("getShapeErr", &ShapeResultKey::getShapeErr);
95  cls.def("getIxx", &ShapeResultKey::getIxx);
96  cls.def("getIyy", &ShapeResultKey::getIyy);
97  cls.def("getIxy", &ShapeResultKey::getIxy);
98 }
99 
100 } // <anonymous>
101 
102 PYBIND11_PLUGIN(shapeUtilities) {
103  py::module::import("lsst.afw.table");
104 
105  py::module mod("shapeUtilities");
106 
107  if (_import_array() < 0) {
108  PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
109  return nullptr;
110  }
111 
112  declareShapeResult(mod);
113  declareShapeResultKey(mod);
114 
115  mod.def("makeShapeTransformMatrix", &makeShapeTransformMatrix, "xform"_a);
116 
117  return mod.ptr();
118 }
119 
120 } // base
121 } // meas
122 } // lsst
ShapeTrMatrix makeShapeTransformMatrix(afw::geom::LinearTransform const &xform)
Construct a matrix suitable for transforming second moments.
Eigen::Matrix< ErrElement, 3, 3, Eigen::DontAlign > ShapeCov
Definition: constants.h:59
float ErrElement
Definition: constants.h:53
PYBIND11_PLUGIN(algorithm)
Definition: algorithm.cc:37