Loading [MathJax]/extensions/tex2jax.js
LSST Applications g04a91732dc+cc8870d3f5,g07dc498a13+5aa0b8792f,g0fba68d861+488cddfaa9,g1409bbee79+5aa0b8792f,g1a7e361dbc+5aa0b8792f,g1fd858c14a+f64bc332a9,g35bb328faa+fcb1d3bbc8,g4d2262a081+b1c1982739,g4d39ba7253+9633a327c1,g4e0f332c67+5d362be553,g53246c7159+fcb1d3bbc8,g60b5630c4e+9633a327c1,g668ecb457e+25d63fd678,g78460c75b0+2f9a1b4bcd,g786e29fd12+cf7ec2a62a,g7b71ed6315+fcb1d3bbc8,g8852436030+8b64ca622a,g89139ef638+5aa0b8792f,g89e1512fd8+37f975783e,g8d6b6b353c+9633a327c1,g9125e01d80+fcb1d3bbc8,g989de1cb63+5aa0b8792f,g9f33ca652e+b196626af7,ga9baa6287d+9633a327c1,gaaedd4e678+5aa0b8792f,gabe3b4be73+1e0a283bba,gb1101e3267+71e32094df,gb58c049af0+f03b321e39,gb90eeb9370+2807b1ad02,gc741bbaa4f+1ae86710ed,gcf25f946ba+8b64ca622a,gd315a588df+a39986a76f,gd6cbbdb0b4+c8606af20c,gd9a9a58781+fcb1d3bbc8,gde0f65d7ad+94dfc458f4,ge278dab8ac+932305ba37,ge82c20c137+76d20ab76d,gfe73954cf8+a1301e4c20,w.2025.11
LSST Data Management Base Package
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Modules Pages
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"
24#include "pybind11/stl.h"
25
26#include "ndarray/pybind11.h"
27
29
30namespace py = pybind11;
31using namespace pybind11::literals;
32
33namespace lsst {
34namespace shapelet {
35
36namespace {
37
38template <typename PyClass>
39void 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
67template <typename PyClass>
68void declareMultiShapeletFunctionEvaluatorMembers(PyClass &cls) {
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::*)(geom::Point2D const &) const) & Class::operator());
75 cls.def("__call__", (double (Class::*)(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 &, geom::Point2I const &) const) &
83 Class::addToImage,
84 "array"_a, "xy0"_a = 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
96 using PyMultiShapeletFunction = py::class_<MultiShapeletFunction, std::shared_ptr<MultiShapeletFunction>>;
97
98 wrappers.wrapType(PyMultiShapeletFunction(wrappers.module, "MultiShapeletFunction"), [](auto &mod, auto &cls) {
99 declareMultiShapeletFunctionMembers(cls);
100
101 });
102
103 using PyMultiShapeletFunctionEvaluator =
104 py::class_<MultiShapeletFunctionEvaluator, std::shared_ptr<MultiShapeletFunctionEvaluator>>;
105
106 wrappers.wrapType(PyMultiShapeletFunctionEvaluator(wrappers.module, "MultiShapeletFunctionEvaluator"), [](auto &mod, auto &cls) {
107 declareMultiShapeletFunctionEvaluatorMembers(cls);
108 });
109}
110
111} // shapelet
112} // lsst
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition python.h:242
PyType wrapType(PyType cls, ClassWrapperCallback function, bool setModuleName=true)
Add a type (class or enum) wrapper, deferring method and other attribute definitions until finish() i...
Definition python.h:391
pybind11::module module
The module object passed to the PYBIND11_MODULE block that contains this WrapperCollection.
Definition python.h:448
A multi-scale shapelet function.
A 2-d function defined by an expansion onto a Gauss-Laguerre or Gauss-Hermite basis.
Extent< double, 2 > Extent2D
Definition Extent.h:400
Point< double, 2 > Point2D
Definition Point.h:324
Point< int, 2 > Point2I
Definition Point.h:321
void wrapMultiShapeletFunction(WrapperCollection &wrappers)