LSSTApplications  17.0+124,17.0+14,17.0+73,18.0.0+37,18.0.0+80,18.0.0-4-g68ffd23+4,18.1.0-1-g0001055+12,18.1.0-1-g03d53ef+5,18.1.0-1-g1349e88+55,18.1.0-1-g2505f39+44,18.1.0-1-g5315e5e+4,18.1.0-1-g5e4b7ea+14,18.1.0-1-g7e8fceb+4,18.1.0-1-g85f8cd4+48,18.1.0-1-g8ff0b9f+4,18.1.0-1-ga2c679d+1,18.1.0-1-gd55f500+35,18.1.0-10-gb58edde+2,18.1.0-11-g0997b02+4,18.1.0-13-gfe4edf0b+12,18.1.0-14-g259bd21+21,18.1.0-19-gdb69f3f+2,18.1.0-2-g5f9922c+24,18.1.0-2-gd3b74e5+11,18.1.0-2-gfbf3545+32,18.1.0-26-g728bddb4+5,18.1.0-27-g6ff7ca9+2,18.1.0-3-g52aa583+25,18.1.0-3-g8ea57af+9,18.1.0-3-gb69f684+42,18.1.0-3-gfcaddf3+6,18.1.0-32-gd8786685a,18.1.0-4-gf3f9b77+6,18.1.0-5-g1dd662b+2,18.1.0-5-g6dbcb01+41,18.1.0-6-gae77429+3,18.1.0-7-g9d75d83+9,18.1.0-7-gae09a6d+30,18.1.0-9-gc381ef5+4,w.2019.45
LSSTDataManagementBasePackage
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 "ndarray/pybind11.h"
26 
28 
29 namespace py = pybind11;
30 using namespace pybind11::literals;
31 
32 namespace lsst {
33 namespace shapelet {
34 
35 namespace {
36 
37 template <typename PyClass>
38 void declareMultiShapeletFunctionMembers(PyClass &cls) {
39  using Class = MultiShapeletFunction;
40 
41  cls.def(py::init<>());
42  cls.def(py::init<Class const &>());
43  cls.def(py::init<typename Class::ComponentList const &>());
44  cls.def(py::init<ShapeletFunction const &>());
45 
46  cls.def("getComponents", [](Class const &self) {
47  auto &components = self.getComponents();
48 
49  py::tuple t(components.size());
50  for (size_t i = 0; i < components.size(); ++i) {
51  t[i] = py::cast(components[i]);
52  }
53 
54  return t;
55  });
56  cls.def("addComponent",
57  [](Class &self, typename Class::Component const &c) { self.getComponents().push_back(c); });
58  cls.def("normalize", &Class::normalize, "value"_a = 1.0);
59  cls.def("shiftInPlace", &Class::shiftInPlace);
60  cls.def("transformInPlace", &Class::transformInPlace);
61  cls.def("convolve", (Class (Class::*)(ShapeletFunction const &) const) & Class::convolve);
62  cls.def("convolve", (Class (Class::*)(Class const &) const) & Class::convolve);
63  cls.def("evaluate", &Class::evaluate);
64 }
65 
66 template <typename PyClass>
67 void declareMultiShapeletFunctionEvaluatorMembers(PyClass &cls) {
68  using Class = MultiShapeletFunctionEvaluator;
69 
70  cls.def(py::init<MultiShapeletFunction const &>());
71 
72  cls.def("__call__", (double (Class::*)(double, double) const) & Class::operator());
73  cls.def("__call__", (double (Class::*)(geom::Point2D const &) const) & Class::operator());
74  cls.def("__call__", (double (Class::*)(geom::Extent2D const &) const) & Class::operator());
75  cls.def("__call__",
76  (ndarray::Array<double, 1, 1> (Class::*)(ndarray::Array<double const, 1> const &,
77  ndarray::Array<double const, 1> const &) const) &
78  Class::operator());
79 
80  cls.def("addToImage",
81  (void (Class::*)(ndarray::Array<double, 2, 1> const &, geom::Point2I const &) const) &
83  "array"_a, "xy0"_a = geom::Point2I());
84  cls.def("addToImage", (void (Class::*)(afw::image::Image<double> &) const) & Class::addToImage,
85  "image"_a);
86 
87  cls.def("integrate", &Class::integrate);
88  cls.def("computeMoments", &Class::computeMoments);
89  cls.def("update", &Class::update);
90 }
91 
92 } // <anonymous>
93 
94 PYBIND11_MODULE(multiShapeletFunction, mod) {
95  py::module::import("lsst.afw.geom");
96  py::module::import("lsst.afw.image");
97 
98  py::class_<MultiShapeletFunction, std::shared_ptr<MultiShapeletFunction>> clsMultiShapeletFunction(
99  mod, "MultiShapeletFunction");
100  py::class_<MultiShapeletFunctionEvaluator, std::shared_ptr<MultiShapeletFunctionEvaluator>>
101  clsMultiShapeletFunctionEvaluator(mod, "MultiShapeletFunctionEvaluator");
102 
103  declareMultiShapeletFunctionMembers(clsMultiShapeletFunction);
104  declareMultiShapeletFunctionEvaluatorMembers(clsMultiShapeletFunctionEvaluator);
105 }
106 
107 } // shapelet
108 } // lsst
table::Key< table::Array< int > > components
py::class_< ProductBoundedField, std::shared_ptr< ProductBoundedField >, BoundedField > PyClass
PYBIND11_MODULE(multiShapeletFunction, mod)
A base class for image defects.
void convolve(OutImageT &convolvedImage, InImageT const &inImage, KernelT const &kernel, ConvolutionControl const &convolutionControl=ConvolutionControl())
Convolve an Image or MaskedImage with a Kernel, setting pixels of an existing output image...
void addToImage(boost::shared_ptr< afw::image::Image< double > > image, std::vector< boost::shared_ptr< afw::image::Image< double > >> const &imgVector, std::vector< double > const &weightVector)
Definition: CoaddPsf.cc:192
UnaryFunctionT::result_type integrate(UnaryFunctionT func, typename UnaryFunctionT::argument_type const a, typename UnaryFunctionT::argument_type const b, double eps=1.0e-6)
The 1D integrator.
Definition: Integrate.h:886