LSST Applications  21.0.0-172-gfb10e10a+18fedfabac,22.0.0+297cba6710,22.0.0+80564b0ff1,22.0.0+8d77f4f51a,22.0.0+a28f4c53b1,22.0.0+dcf3732eb2,22.0.1-1-g7d6de66+2a20fdde0d,22.0.1-1-g8e32f31+297cba6710,22.0.1-1-geca5380+7fa3b7d9b6,22.0.1-12-g44dc1dc+2a20fdde0d,22.0.1-15-g6a90155+515f58c32b,22.0.1-16-g9282f48+790f5f2caa,22.0.1-2-g92698f7+dcf3732eb2,22.0.1-2-ga9b0f51+7fa3b7d9b6,22.0.1-2-gd1925c9+bf4f0e694f,22.0.1-24-g1ad7a390+a9625a72a8,22.0.1-25-g5bf6245+3ad8ecd50b,22.0.1-25-gb120d7b+8b5510f75f,22.0.1-27-g97737f7+2a20fdde0d,22.0.1-32-gf62ce7b1+aa4237961e,22.0.1-4-g0b3f228+2a20fdde0d,22.0.1-4-g243d05b+871c1b8305,22.0.1-4-g3a563be+32dcf1063f,22.0.1-4-g44f2e3d+9e4ab0f4fa,22.0.1-42-gca6935d93+ba5e5ca3eb,22.0.1-5-g15c806e+85460ae5f3,22.0.1-5-g58711c4+611d128589,22.0.1-5-g75bb458+99c117b92f,22.0.1-6-g1c63a23+7fa3b7d9b6,22.0.1-6-g50866e6+84ff5a128b,22.0.1-6-g8d3140d+720564cf76,22.0.1-6-gd805d02+cc5644f571,22.0.1-8-ge5750ce+85460ae5f3,master-g6e05de7fdc+babf819c66,master-g99da0e417a+8d77f4f51a,w.2021.48
LSST Data Management Base Package
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"
23 
24 #include "ndarray/pybind11.h"
25 
28 
29 namespace py = pybind11;
30 using namespace pybind11::literals;
31 
32 namespace lsst {
33 namespace shapelet {
34 
35 namespace {
36 
37 template <typename T>
38 py::class_<MatrixBuilder<T>, std::shared_ptr<MatrixBuilder<T>>> declareMatrixBuilder(
39  py::module &mod, std::string const &suffix) {
40  using Class = MatrixBuilder<T>;
41 
42  py::class_<Class, std::shared_ptr<Class>> cls(mod, ("MatrixBuilder" + suffix).c_str());
43 
44  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int>(),
45  "x"_a, "y"_a, "order"_a);
46  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int,
47  ShapeletFunction const &>(),
48  "x"_a, "y"_a, "order"_a, "psf"_a);
49  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
50  MultiShapeletBasis const &>(),
51  "x"_a, "y"_a, "basis"_a);
52  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
53  MultiShapeletBasis const &, MultiShapeletFunction const &>(),
54  "x"_a, "y"_a, "basis"_a, "psf"_a);
55 
56  cls.def("getDataSize", &Class::getDataSize);
57  cls.def("getBasisSize", &Class::getBasisSize);
58  cls.def("allocateOutput", &Class::allocateOutput);
59 
60  cls.def("__call__",
61  (void (Class::*)(ndarray::Array<T, 2, -1> const &, afw::geom::ellipses::Ellipse const &) const) &
62  Class::operator());
63  cls.def("__call__", (ndarray::Array<T, 2, -2> (Class::*)(afw::geom::ellipses::Ellipse const &) const) &
64  Class::operator());
65 
66  return cls;
67 }
68 
69 template <typename T>
70 py::class_<MatrixBuilderWorkspace<T>, std::shared_ptr<MatrixBuilderWorkspace<T>>>
71 declareMatrixBuilderWorkspace(py::module &mod, std::string const &suffix) {
72  using Class = MatrixBuilderWorkspace<T>;
73 
74  py::class_<Class, std::shared_ptr<Class>> cls(mod, ("MatrixBuilderWorkspace" + suffix).c_str());
75 
76  cls.def(py::init<int>(), "size"_a);
77  cls.def(py::init<Class const &>(), "other"_a);
78 
79  cls.def("getRemaining", &Class::getRemaining);
80 
81  return cls;
82 }
83 
84 template <typename T>
85 py::class_<MatrixBuilderFactory<T>, std::shared_ptr<MatrixBuilderFactory<T>>> declareMatrixBuilderFactory(
86  py::module &mod, std::string const &suffix) {
87  using Class = MatrixBuilderFactory<T>;
88 
89  py::class_<Class, std::shared_ptr<Class>> cls(mod, ("MatrixBuilderFactory" + suffix).c_str());
90 
91  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int>(),
92  "x"_a, "y"_a, "order"_a);
93  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &, int,
94  ShapeletFunction const &>(),
95  "x"_a, "y"_a, "order"_a, "psf"_a);
96  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
97  MultiShapeletBasis const &>(),
98  "x"_a, "y"_a, "basis"_a);
99  cls.def(py::init<ndarray::Array<T const, 1, 1> const &, ndarray::Array<T const, 1, 1> const &,
100  MultiShapeletBasis const &, MultiShapeletFunction const &>(),
101  "x"_a, "y"_a, "basis"_a, "psf"_a);
102 
103  cls.def("__call__", (MatrixBuilder<T> (Class::*)() const) & Class::operator());
104  cls.def("__call__", (MatrixBuilder<T> (Class::*)(typename Class::Workspace &) const) & Class::operator(),
105  "workspace"_a);
106 
107  cls.def("getDataSize", &Class::getDataSize);
108  cls.def("getBasisSize", &Class::getBasisSize);
109  cls.def("computeWorkspace", &Class::computeWorkspace);
110 
111  return cls;
112 }
113 
114 template <typename T>
115 void declareMatrixBuilderTemplates(py::module &mod, std::string const &suffix) {
116  auto clsMatrixBuilder = declareMatrixBuilder<T>(mod, suffix);
117  auto clsMatrixBuilderWorkspace = declareMatrixBuilderWorkspace<T>(mod, suffix);
118  auto clsMatrixBuilderFactory = declareMatrixBuilderFactory<T>(mod, suffix);
119 
120  clsMatrixBuilder.attr("Workspace") = clsMatrixBuilderWorkspace;
121  clsMatrixBuilder.attr("Factory") = clsMatrixBuilderFactory;
122 
123  clsMatrixBuilderFactory.attr("Workspace") = clsMatrixBuilderWorkspace;
124  clsMatrixBuilderFactory.attr("Builder") = clsMatrixBuilder;
125 }
126 
127 } // <anonymous>
128 
129 PYBIND11_MODULE(matrixBuilder, mod) {
130  py::module::import("lsst.afw.geom");
131  declareMatrixBuilderTemplates<float>(mod, "F");
132  declareMatrixBuilderTemplates<double>(mod, "D");
133 }
134 
135 } // shapelet
136 } // lsst
PYBIND11_MODULE(matrixBuilder, mod)
A base class for image defects.