LSST Applications g013ef56533+7c9321ec0f,g042eb84c57+c6cfa41bc3,g199a45376c+0ba108daf9,g1fd858c14a+fcad0d0313,g210f2d0738+c0f94c6586,g262e1987ae+a7e710680e,g29ae962dfc+fb55f2edb0,g2ac17093b6+61d6563b1e,g2b1d02342f+df6f932764,g2cef7863aa+aef1011c0b,g2f7ad74990+c0f94c6586,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+53cf87ae69,g47891489e3+4316d04fff,g511e8cfd20+baa56acf6c,g53246c7159+8c5ae1fdc5,g54cd7ddccb+fd7ad03fde,g64539dfbff+c0f94c6586,g67b6fd64d1+4316d04fff,g67fd3c3899+c0f94c6586,g6985122a63+4316d04fff,g74acd417e5+ca833bee28,g786e29fd12+668abc6043,g81db2e9a8d+b2ec8e584f,g87389fa792+8856018cbb,g89139ef638+4316d04fff,g8d7436a09f+0a24083b20,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+11eb8fd5b8,gbf99507273+8c5ae1fdc5,gcdda8b9158+e4c84c9d5c,gce8aa8abaa+8c5ae1fdc5,gd7ef33dd92+4316d04fff,gdab6d2f7ff+ca833bee28,ge410e46f29+4316d04fff,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.40
LSST Data Management Base Package
Loading...
Searching...
No Matches
shapeletFunction.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
25#include "ndarray/pybind11.h"
26
28
29namespace py = pybind11;
30using namespace pybind11::literals;
31
32namespace lsst {
33namespace shapelet {
34
36 using PyShapeletFunction = py::classh<ShapeletFunction>;
37
38 wrappers.wrapType(PyShapeletFunction(wrappers.module, "ShapeletFunction"), [](auto &mod, auto &cls) {
39 cls.def_readonly_static("FLUX_FACTOR", &ShapeletFunction::FLUX_FACTOR);
40
41 cls.def(py::init<int, BasisTypeEnum>(), "order"_a, "basisType"_a);
42 cls.def(py::init<int, BasisTypeEnum, ndarray::Array<double, 1, 1> const &>(), "order"_a,
43 "basisType"_a, "coefficients"_a);
44 cls.def(py::init<int, BasisTypeEnum, double, geom::Point2D const &>(), "order"_a,
45 "basisType"_a, "radius"_a, "center"_a = geom::Point2D());
46 cls.def(py::init<int, BasisTypeEnum, double, geom::Point2D const &,
47 ndarray::Array<double, 1, 1> const &>(),
48 "order"_a, "basisType"_a, "radius"_a, "center"_a, "coefficients"_a);
49 cls.def(py::init<int, BasisTypeEnum, afw::geom::ellipses::Ellipse const &>(), "order"_a,
50 "basisType"_a, "ellipse"_a);
51 cls.def(py::init<int, BasisTypeEnum, afw::geom::ellipses::Ellipse const &,
52 ndarray::Array<double const, 1, 1> const &>(),
53 "order"_a, "basisType"_a, "ellipse"_a, "coefficients"_a);
54 cls.def(py::init<ShapeletFunction>());
55
56 cls.def("getOrder", &ShapeletFunction::getOrder);
57 cls.def("getEllipse", (afw::geom::ellipses::Ellipse &(ShapeletFunction::*)()) &
58 ShapeletFunction::getEllipse,
59 py::return_value_policy::reference_internal);
60 cls.def("setEllipse", &ShapeletFunction::setEllipse);
61 cls.def("getBasisType", &ShapeletFunction::getBasisType);
62 cls.def("changeBasisType", &ShapeletFunction::changeBasisType);
63 cls.def("normalize", &ShapeletFunction::normalize, "value"_a = 1.0);
64 cls.def("getCoefficients", (ndarray::Array<double, 1, 1> const (ShapeletFunction::*)()) &
65 ShapeletFunction::getCoefficients);
66 cls.def("convolve", &ShapeletFunction::convolve);
67 cls.def("evaluate", &ShapeletFunction::evaluate);
68 cls.def("shiftInPlace", &ShapeletFunction::shiftInPlace);
69 cls.def("transformInPlace", &ShapeletFunction::transformInPlace);
70 });
71
72 using PyShapeletFunctionEvaluator = py::classh<ShapeletFunctionEvaluator>;
73
74 wrappers.wrapType(
75 PyShapeletFunctionEvaluator(wrappers.module, "ShapeletFunctionEvaluator"), [](auto &mod, auto &cls) {
76 cls.def(py::init<ShapeletFunction const &>(), "function"_a);
77
78 cls.def("__call__",
79 (double (ShapeletFunctionEvaluator::*)(double, double) const) &
80 ShapeletFunctionEvaluator::operator());
81 cls.def(
82 "__call__", (double (ShapeletFunctionEvaluator::*)(geom::Point2D const &) const) &
83 ShapeletFunctionEvaluator::operator());
84 cls.def(
85 "__call__", (double (ShapeletFunctionEvaluator::*)(geom::Extent2D const &) const) &
86 ShapeletFunctionEvaluator::operator());
87 cls.def("__call__", (ndarray::Array<double, 1, 1> (ShapeletFunctionEvaluator::*)(
88 ndarray::Array<double const, 1> const &,
89 ndarray::Array<double const, 1> const &) const) &
90 ShapeletFunctionEvaluator::operator());
91
92 cls.def(
93 "addToImage", (void (ShapeletFunctionEvaluator::*)(ndarray::Array<double, 2, 1> const &,
94 geom::Point2I const &) const) &
95 ShapeletFunctionEvaluator::addToImage,
96 "array"_a, "xy0"_a = geom::Point2D());
97 cls.def(
98 "addToImage", (void (ShapeletFunctionEvaluator::*)(afw::image::Image<double> &) const) &
99 ShapeletFunctionEvaluator::addToImage,
100 "image"_a);
101 cls.def("integrate", &ShapeletFunctionEvaluator::integrate);
102 cls.def("computeMoments", &ShapeletFunctionEvaluator::computeMoments);
103 cls.def("update", &ShapeletFunctionEvaluator::update);
104 });
105}
106
107} // shapelet
108} // 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
void wrapShapeletFunction(WrapperCollection &wrappers)