LSST Applications 27.0.0,g0265f82a02+469cd937ee,g02d81e74bb+21ad69e7e1,g1470d8bcf6+cbe83ee85a,g2079a07aa2+e67c6346a6,g212a7c68fe+04a9158687,g2305ad1205+94392ce272,g295015adf3+81dd352a9d,g2bbee38e9b+469cd937ee,g337abbeb29+469cd937ee,g3939d97d7f+72a9f7b576,g487adcacf7+71499e7cba,g50ff169b8f+5929b3527e,g52b1c1532d+a6fc98d2e7,g591dd9f2cf+df404f777f,g5a732f18d5+be83d3ecdb,g64a986408d+21ad69e7e1,g858d7b2824+21ad69e7e1,g8a8a8dda67+a6fc98d2e7,g99cad8db69+f62e5b0af5,g9ddcbc5298+d4bad12328,ga1e77700b3+9c366c4306,ga8c6da7877+71e4819109,gb0e22166c9+25ba2f69a1,gb6a65358fc+469cd937ee,gbb8dafda3b+69d3c0e320,gc07e1c2157+a98bf949bb,gc120e1dc64+615ec43309,gc28159a63d+469cd937ee,gcf0d15dbbd+72a9f7b576,gdaeeff99f8+a38ce5ea23,ge6526c86ff+3a7c1ac5f1,ge79ae78c31+469cd937ee,gee10cc3b42+a6fc98d2e7,gf1cff7945b+21ad69e7e1,gfbcc870c63+9a11dc8c8f
LSST Data Management Base Package
Loading...
Searching...
No Matches
matrixBuilder.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
29
30namespace py = pybind11;
31using namespace pybind11::literals;
32
33namespace lsst {
34namespace shapelet {
35
36namespace {
37
38template <typename T>
39py::class_<MatrixBuilder<T>, std::shared_ptr<MatrixBuilder<T>>> declareMatrixBuilder(
41 using Class = MatrixBuilder<T>;
42 using PyCLass = py::class_<Class, std::shared_ptr<Class>>;
43 std::string name = "MatrixBuilder" + suffix;
44
45 return wrappers.wrapType(PyCLass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
46 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int>(),
47 "x"_a, "y"_a, "order"_a);
48 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int,
49 ShapeletFunction const &>(),
50 "x"_a, "y"_a, "order"_a, "psf"_a);
51 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
52 MultiShapeletBasis const &>(),
53 "x"_a, "y"_a, "basis"_a);
54 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
55 MultiShapeletBasis const &, MultiShapeletFunction const &>(),
56 "x"_a, "y"_a, "basis"_a, "psf"_a);
57
58 cls.def("getDataSize", &Class::getDataSize);
59 cls.def("getBasisSize", &Class::getBasisSize);
60 cls.def("allocateOutput", &Class::allocateOutput);
61
62 cls.def("__call__",
63 (void (Class::*)(ndarray::Array<T, 2, -1> const &, afw::geom::ellipses::Ellipse const &) const) &
64 Class::operator());
65 cls.def("__call__", (ndarray::Array<T, 2, -2> (Class::*)(afw::geom::ellipses::Ellipse const &) const) &
66 Class::operator());
67 });
68}
69
70template <typename T>
71py::class_<MatrixBuilderWorkspace<T>, std::shared_ptr<MatrixBuilderWorkspace<T>>>
72declareMatrixBuilderWorkspace(lsst::cpputils::python::WrapperCollection &wrappers, std::string const &suffix) {
73 using Class = MatrixBuilderWorkspace<T>;
74 using PyClass = py::class_<Class, std::shared_ptr<Class>>;
75 std::string name = "MatrixBuilderWorkspace" + suffix;
76
77 return wrappers.wrapType(PyClass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
78
79 cls.def(py::init<int>(), "size"_a);
80 cls.def(py::init<Class const &>(), "other"_a);
81
82 cls.def("getRemaining", &Class::getRemaining);
83 });
84}
85
86template <typename T>
87py::class_<MatrixBuilderFactory<T>, std::shared_ptr<MatrixBuilderFactory<T>>> declareMatrixBuilderFactory(
89 using Class = MatrixBuilderFactory<T>;
90 using PyClass = py::class_<Class, std::shared_ptr<Class>>;
91 std::string name = "MatrixBuilderFactory" + suffix;
92
93 return wrappers.wrapType(PyClass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
94 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int>(),
95 "x"_a, "y"_a, "order"_a);
96 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int,
97 ShapeletFunction const &>(),
98 "x"_a, "y"_a, "order"_a, "psf"_a);
99 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
100 MultiShapeletBasis const &>(),
101 "x"_a, "y"_a, "basis"_a);
102 cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
103 MultiShapeletBasis const &, MultiShapeletFunction const &>(),
104 "x"_a, "y"_a, "basis"_a, "psf"_a);
105
106 cls.def("__call__", (MatrixBuilder<T> (Class::*)() const) &Class::operator());
107 cls.def("__call__", (MatrixBuilder<T> (Class::*)(typename Class::Workspace &) const) &Class::operator(),
108 "workspace"_a);
109
110 cls.def("getDataSize", &Class::getDataSize);
111 cls.def("getBasisSize", &Class::getBasisSize);
112 cls.def("computeWorkspace", &Class::computeWorkspace);
113 });
114}
115
116template <typename T>
117void declareMatrixBuilderTemplates(lsst::cpputils::python::WrapperCollection &wrappers, std::string const &suffix) {
118 auto &mod = wrappers.module;
119 auto clsMatrixBuilder = declareMatrixBuilder<T>(wrappers, suffix);
120 auto clsMatrixBuilderWorkspace = declareMatrixBuilderWorkspace<T>(wrappers, suffix);
121 auto clsMatrixBuilderFactory = declareMatrixBuilderFactory<T>(wrappers, suffix);
122
123 clsMatrixBuilder.attr("Workspace") = clsMatrixBuilderWorkspace;
124 clsMatrixBuilder.attr("Factory") = clsMatrixBuilderFactory;
125
126 clsMatrixBuilderFactory.attr("Workspace") = clsMatrixBuilderWorkspace;
127 clsMatrixBuilderFactory.attr("Builder") = clsMatrixBuilder;
128}
129
130} // <anonymous>
131
133 declareMatrixBuilderTemplates<float>(wrappers, "F");
134 declareMatrixBuilderTemplates<double>(wrappers, "D");
135}
136
137} // shapelet
138} // 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
py::class_< PixelAreaBoundedField, std::shared_ptr< PixelAreaBoundedField >, BoundedField > PyClass
void wrapMatrixBuilder(WrapperCollection &wrappers)