lsst.meas.extensions.simpleShape  13.0-7-ge993229+7
 All Classes Namespaces Files Functions Variables Typedefs Macros
simpleShape.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  *
4  * This product includes software developed by the
5  * LSST Project (http://www.lsst.org/).
6  * See the COPYRIGHT file
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 #include "pybind11/pybind11.h"
23 
24 #include "numpy/arrayobject.h"
25 #include "ndarray/pybind11.h"
26 
28 #include "lsst/pex/config/python.h"
29 
30 namespace py = pybind11;
31 using namespace pybind11::literals;
32 
33 namespace lsst {
34 namespace meas {
35 namespace extensions {
36 namespace simpleShape {
37 
38 namespace {
39 
40 template <typename T, typename PyClass>
41 static void declareMoments(PyClass cls) {
42  cls.def_static("computeMoments", (SimpleShapeResult(*)(afw::geom::ellipses::Ellipse const &,
43  afw::image::MaskedImage<T> const &, double)) &
44  SimpleShape::computeMoments,
45  "weight"_a, "image"_a, "nSigmaRegion"_a = 3.0);
46 }
47 
48 } // <anonymous>
49 
50 PYBIND11_PLUGIN(simpleShape) {
51  py::module::import("lsst.afw.geom");
52  py::module::import("lsst.afw.image");
53  py::module::import("lsst.afw.table");
54  py::module::import("lsst.meas.base");
55 
56  py::module mod("simpleShape");
57 
58  if (_import_array() < 0) {
59  PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
60  return nullptr;
61  }
62 
63  /* Module level */
64  py::class_<SimpleShape, std::shared_ptr<SimpleShape>, base::SimpleAlgorithm> clsSimpleShape(
65  mod, "SimpleShape");
66  py::class_<SimpleShapeControl> clsSimpleShapeControl(mod, "SimpleShapeControl");
67  py::class_<SimpleShapeResult> clsSimpleShapeResult(mod, "SimpleShapeResult");
68  py::class_<SimpleShapeResultKey> clsSimpleShapeResultKey(mod, "SimpleShapeResultKey");
69 
70  /* Constructors */
71  clsSimpleShape.def(py::init<SimpleShape::Control const &, std::string const &, afw::table::Schema &>(),
72  "ctrl"_a, "name"_a, "schema"_a);
73  clsSimpleShapeControl.def(py::init<>());
74 
75  clsSimpleShapeResultKey.def(py::init<afw::table::SubSchema const &>(), "s"_a);
76 
77  /* Members */
78  LSST_DECLARE_CONTROL_FIELD(clsSimpleShapeControl, SimpleShapeControl, sigma);
79  LSST_DECLARE_CONTROL_FIELD(clsSimpleShapeControl, SimpleShapeControl, nSigmaRegion);
80 
81  declareMoments<float>(clsSimpleShape);
82  declareMoments<double>(clsSimpleShape);
83  clsSimpleShape.def_static("correctWeightedMoments", &SimpleShape::correctWeightedMoments, "weight"_a,
84  "ellipse"_a, "center"_a);
85 
86  clsSimpleShapeResult.def_readwrite("ellipse", &SimpleShapeResult::ellipse);
87  clsSimpleShapeResult.def_readwrite("center", &SimpleShapeResult::center);
88  clsSimpleShapeResult.def_readwrite("covariance", &SimpleShapeResult::covariance);
89  clsSimpleShapeResult.def("getFlag", &SimpleShapeResult::getFlag);
90 
91  clsSimpleShapeResultKey.def("get", &SimpleShapeResultKey::get, "record"_a);
92 
93  return mod.ptr();
94 }
95 
96 } // shapeHSM
97 } // extensions
98 } // meas
99 } // lsst
MatrixQ covariance
Definition: simpleShape.cc:152
A C++ control class to handle SdssShapeAlgorithm&#39;s configuration.
Definition: simpleShape.h:43