LSST Applications  21.0.0+04719a4bac,21.0.0-1-ga51b5d4+f5e6047307,21.0.0-11-g2b59f77+a9c1acf22d,21.0.0-11-ga42c5b2+86977b0b17,21.0.0-12-gf4ce030+76814010d2,21.0.0-13-g1721dae+760e7a6536,21.0.0-13-g3a573fe+768d78a30a,21.0.0-15-g5a7caf0+f21cbc5713,21.0.0-16-g0fb55c1+b60e2d390c,21.0.0-19-g4cded4ca+71a93a33c0,21.0.0-2-g103fe59+bb20972958,21.0.0-2-g45278ab+04719a4bac,21.0.0-2-g5242d73+3ad5d60fb1,21.0.0-2-g7f82c8f+8babb168e8,21.0.0-2-g8f08a60+06509c8b61,21.0.0-2-g8faa9b5+616205b9df,21.0.0-2-ga326454+8babb168e8,21.0.0-2-gde069b7+5e4aea9c2f,21.0.0-2-gecfae73+1d3a86e577,21.0.0-2-gfc62afb+3ad5d60fb1,21.0.0-25-g1d57be3cd+e73869a214,21.0.0-3-g357aad2+ed88757d29,21.0.0-3-g4a4ce7f+3ad5d60fb1,21.0.0-3-g4be5c26+3ad5d60fb1,21.0.0-3-g65f322c+e0b24896a3,21.0.0-3-g7d9da8d+616205b9df,21.0.0-3-ge02ed75+a9c1acf22d,21.0.0-4-g591bb35+a9c1acf22d,21.0.0-4-g65b4814+b60e2d390c,21.0.0-4-gccdca77+0de219a2bc,21.0.0-4-ge8a399c+6c55c39e83,21.0.0-5-gd00fb1e+05fce91b99,21.0.0-6-gc675373+3ad5d60fb1,21.0.0-64-g1122c245+4fb2b8f86e,21.0.0-7-g04766d7+cd19d05db2,21.0.0-7-gdf92d54+04719a4bac,21.0.0-8-g5674e7b+d1bd76f71f,master-gac4afde19b+a9c1acf22d,w.2021.13
LSST Data Management Base Package
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
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
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.
py::class_< PixelAreaBoundedField, std::shared_ptr< PixelAreaBoundedField >, BoundedField > PyClass
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
PYBIND11_MODULE(multiShapeletFunction, mod)
A base class for image defects.