LSSTApplications  18.1.0
LSSTDataManagementBasePackage
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 
130  py::module::import("lsst.afw.geom");
131  declareMatrixBuilderTemplates<float>(mod, "F");
132  declareMatrixBuilderTemplates<double>(mod, "D");
133 }
134 
135 } // shapelet
136 } // lsst
def init()
Definition: tests.py:75
STL class.
A base class for image defects.
PYBIND11_MODULE(matrixBuilder, mod)