lsst.shapelet  14.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends
multiShapeletFunction.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 #include "pybind11/stl.h"
24 
25 #include "numpy/arrayobject.h"
26 #include "ndarray/pybind11.h"
27 
29 
30 namespace py = pybind11;
31 using namespace pybind11::literals;
32 
33 namespace lsst {
34 namespace shapelet {
35 
36 namespace {
37 
38 template <typename PyClass>
39 void declareMultiShapeletFunctionMembers(PyClass &cls) {
40  using Class = MultiShapeletFunction;
41 
42  cls.def(py::init<>());
43  cls.def(py::init<Class const &>());
44  cls.def(py::init<typename Class::ComponentList const &>());
45  cls.def(py::init<ShapeletFunction const &>());
46 
47  cls.def("getComponents", [](Class const &self) {
48  auto &components = self.getComponents();
49 
50  py::tuple t(components.size());
51  for (size_t i = 0; i < components.size(); ++i) {
52  t[i] = py::cast(components[i]);
53  }
54 
55  return t;
56  });
57  cls.def("addComponent",
58  [](Class &self, typename Class::Component const &c) { self.getComponents().push_back(c); });
59  cls.def("normalize", &Class::normalize, "value"_a = 1.0);
60  cls.def("shiftInPlace", &Class::shiftInPlace);
61  cls.def("transformInPlace", &Class::transformInPlace);
62  cls.def("convolve", (Class (Class::*)(ShapeletFunction const &) const) & Class::convolve);
63  cls.def("convolve", (Class (Class::*)(Class const &) const) & Class::convolve);
64  cls.def("evaluate", &Class::evaluate);
65 }
66 
67 template <typename PyClass>
68 void declareMultiShapeletFunctionEvaluatorMembers(PyClass &cls) {
69  using Class = MultiShapeletFunctionEvaluator;
70 
71  cls.def(py::init<MultiShapeletFunction const &>());
72 
73  cls.def("__call__", (double (Class::*)(double, double) const) & Class::operator());
74  cls.def("__call__", (double (Class::*)(afw::geom::Point2D const &) const) & Class::operator());
75  cls.def("__call__", (double (Class::*)(afw::geom::Extent2D const &) const) & Class::operator());
76  cls.def("__call__",
77  (ndarray::Array<double, 1, 1> (Class::*)(ndarray::Array<double const, 1> const &,
78  ndarray::Array<double const, 1> const &) const) &
79  Class::operator());
80 
81  cls.def("addToImage",
82  (void (Class::*)(ndarray::Array<double, 2, 1> const &, afw::geom::Point2I const &) const) &
83  Class::addToImage,
84  "array"_a, "xy0"_a = afw::geom::Point2I());
85  cls.def("addToImage", (void (Class::*)(afw::image::Image<double> &) const) & Class::addToImage,
86  "image"_a);
87 
88  cls.def("integrate", &Class::integrate);
89  cls.def("computeMoments", &Class::computeMoments);
90  cls.def("update", &Class::update);
91 }
92 
93 } // <anonymous>
94 
95 PYBIND11_PLUGIN(multiShapeletFunction) {
96  py::module::import("lsst.afw.geom");
97  py::module::import("lsst.afw.image");
98 
99  py::module mod("multiShapeletFunction");
100 
101  if (_import_array() < 0) {
102  PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
103  return nullptr;
104  }
105 
106  py::class_<MultiShapeletFunction, std::shared_ptr<MultiShapeletFunction>> clsMultiShapeletFunction(
107  mod, "MultiShapeletFunction");
108  py::class_<MultiShapeletFunctionEvaluator, std::shared_ptr<MultiShapeletFunctionEvaluator>>
109  clsMultiShapeletFunctionEvaluator(mod, "MultiShapeletFunctionEvaluator");
110 
111  declareMultiShapeletFunctionMembers(clsMultiShapeletFunction);
112  declareMultiShapeletFunctionEvaluatorMembers(clsMultiShapeletFunctionEvaluator);
113 
114  return mod.ptr();
115 }
116 
117 } // shapelet
118 } // lsst
PYBIND11_PLUGIN(basisEvaluator)