LSST Applications g04e9c324dd+8c5ae1fdc5,g134cb467dc+1b3060144d,g18429d2f64+f642bf4753,g199a45376c+0ba108daf9,g1fd858c14a+2dcf163641,g262e1987ae+7b8c96d2ca,g29ae962dfc+3bd6ecb08a,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+53e1a9e7c5,g4595892280+fef73a337f,g47891489e3+2efcf17695,g4d44eb3520+642b70b07e,g53246c7159+8c5ae1fdc5,g67b6fd64d1+2efcf17695,g67fd3c3899+b70e05ef52,g74acd417e5+317eb4c7d4,g786e29fd12+668abc6043,g87389fa792+8856018cbb,g89139ef638+2efcf17695,g8d7436a09f+3be3c13596,g8ea07a8fe4+9f5ccc88ac,g90f42f885a+a4e7b16d9b,g97be763408+ad77d7208f,g9dd6db0277+b70e05ef52,ga681d05dcb+a3f46e7fff,gabf8522325+735880ea63,gac2eed3f23+2efcf17695,gb89ab40317+2efcf17695,gbf99507273+8c5ae1fdc5,gd8ff7fe66e+b70e05ef52,gdab6d2f7ff+317eb4c7d4,gdc713202bf+b70e05ef52,gdfd2d52018+b10e285e0f,ge365c994fd+310e8507c4,ge410e46f29+2efcf17695,geaed405ab2+562b3308c0,gffca2db377+8c5ae1fdc5,w.2025.35
LSST Data Management Base Package
Loading...
Searching...
No Matches
kernelSolution.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/eigen.h"
25
26#include <memory>
27
28#include "Eigen/Core"
29
32
33namespace py = pybind11;
34using namespace pybind11::literals;
35
36namespace lsst {
37namespace ip {
38namespace diffim {
39
40namespace {
41
45void declareKernelSolution(lsst::cpputils::python::WrapperCollection &wrappers) {
46 using PyKernelSolution = py::classh<KernelSolution>;
47
48 wrappers.wrapType(PyKernelSolution(wrappers.module, "KernelSolution"), [](auto &mod, auto &cls) {
49 cls.def(py::init<Eigen::MatrixXd, Eigen::VectorXd, bool>(), "mMat"_a, "bVec"_a, "fitForBackground"_a);
50 cls.def(py::init<bool>(), "fitForBackground"_a);
51 cls.def(py::init<>());
52
53 py::enum_<KernelSolution::KernelSolvedBy>(cls, "KernelSolvedBy")
54 .value("NONE", KernelSolution::KernelSolvedBy::NONE)
55 .value("CHOLESKY_LDLT", KernelSolution::KernelSolvedBy::CHOLESKY_LDLT)
56 .value("CHOLESKY_LLT", KernelSolution::KernelSolvedBy::CHOLESKY_LLT)
57 .value("LU", KernelSolution::KernelSolvedBy::LU)
58 .value("EIGENVECTOR", KernelSolution::KernelSolvedBy::EIGENVECTOR)
59 .export_values();
60
61 py::enum_<KernelSolution::ConditionNumberType>(cls, "ConditionNumberType")
62 .value("EIGENVALUE", KernelSolution::ConditionNumberType::EIGENVALUE)
63 .value("SVD", KernelSolution::ConditionNumberType::SVD)
64 .export_values();
65
66 cls.def("solve", (void (KernelSolution::*)()) &KernelSolution::solve);
67 cls.def("solve", (void (KernelSolution::*)(Eigen::MatrixXd const &, Eigen::VectorXd const &)) &
68 KernelSolution::solve,
69 "mMat"_a, "bVec"_a);
70 cls.def("getSolvedBy", &KernelSolution::getSolvedBy);
71 cls.def("getConditionNumber", (double (KernelSolution::*)(KernelSolution::ConditionNumberType)) &
72 KernelSolution::getConditionNumber,
73 "conditionType"_a);
74 cls.def("getConditionNumber",
75 (double (KernelSolution::*)(Eigen::MatrixXd const &, KernelSolution::ConditionNumberType)) &
76 KernelSolution::getConditionNumber,
77 "mMat"_a, "conditionType"_a);
78 cls.def("getM", &KernelSolution::getM, py::return_value_policy::copy);
79 cls.def("getB", &KernelSolution::getB, py::return_value_policy::copy);
80 cls.def("printM", &KernelSolution::printM);
81 cls.def("printB", &KernelSolution::printB);
82 cls.def("printA", &KernelSolution::printA);
83 cls.def("getId", &KernelSolution::getId);
84 });
85}
86
94template <typename InputT>
95void declareStaticKernelSolution(lsst::cpputils::python::WrapperCollection &wrappers, std::string const &suffix) {
96 using PyStaticKernelSolution = py::classh<StaticKernelSolution<InputT>, KernelSolution>;
97 std::string name = ("StaticKernelSolution" + suffix);
98 wrappers.wrapType(PyStaticKernelSolution(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
99 cls.def(py::init<lsst::afw::math::KernelList const &, bool>(), "basisList"_a, "fitForBackground"_a);
100
101 cls.def("solve", (void (StaticKernelSolution<InputT>::*)()) &StaticKernelSolution<InputT>::solve);
102 cls.def("build", &StaticKernelSolution<InputT>::build, "templateImage"_a, "scienceImage"_a,
103 "varianceEstimate"_a);
104 cls.def("getKernel", &StaticKernelSolution<InputT>::getKernel);
105 cls.def("makeKernelImage", &StaticKernelSolution<InputT>::makeKernelImage);
106 cls.def("getBackground", &StaticKernelSolution<InputT>::getBackground);
107 cls.def("getKsum", &StaticKernelSolution<InputT>::getKsum);
108 cls.def("getSolutionPair", &StaticKernelSolution<InputT>::getSolutionPair);
109 });
110}
111
119template <typename InputT>
120void declareMaskedKernelSolution(lsst::cpputils::python::WrapperCollection &wrappers, std::string const &suffix) {
121 using PyMaskedKernelSolution = py::classh<MaskedKernelSolution<InputT>, StaticKernelSolution<InputT>>;
122 std::string name = "MaskedKernelSolution" + suffix;
123 wrappers.wrapType(PyMaskedKernelSolution(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
124 cls.def(py::init<lsst::afw::math::KernelList const &, bool>(), "basisList"_a, "fitForBackground"_a);
125
126 cls.def("buildOrig", &MaskedKernelSolution<InputT>::buildOrig, "templateImage"_a, "scienceImage"_a,
127 "varianceEstimate"_a, "pixelMask"_a);
128 cls.def("buildWithMask", &MaskedKernelSolution<InputT>::buildWithMask, "templateImage"_a,
129 "scienceImage"_a, "varianceEstimate"_a, "pixelMask"_a);
130 cls.def("buildSingleMaskOrig", &MaskedKernelSolution<InputT>::buildSingleMaskOrig, "templateImage"_a,
131 "scienceImage"_a, "varianceEstimate"_a, "maskBox"_a);
132 });
133}
134
142template <typename InputT>
143void declareRegularizedKernelSolution(lsst::cpputils::python::WrapperCollection &wrappers, std::string const &suffix) {
144 using PyRegularizedKernelSolution =
145 py::classh<RegularizedKernelSolution<InputT>, StaticKernelSolution<InputT>>;
146
147 std::string name = "RegularizedKernelSolution" + suffix;
148 wrappers.wrapType(PyRegularizedKernelSolution(wrappers.module, "RegularizedKernelSolution"), [](auto &mod, auto &cls) {
149 cls.def(py::init<lsst::afw::math::KernelList const &, bool, Eigen::MatrixXd const &,
150 daf::base::PropertySet const &>(),
151 "basisList"_a, "fitForBackground"_a, "hMat"_a, "ps"_a);
152
153 cls.def("solve",
154 (void (RegularizedKernelSolution<InputT>::*)()) &RegularizedKernelSolution<InputT>::solve);
155 cls.def("getLambda", &RegularizedKernelSolution<InputT>::getLambda);
156 cls.def("estimateRisk", &RegularizedKernelSolution<InputT>::estimateRisk, "maxCond"_a);
157 cls.def("getM", &RegularizedKernelSolution<InputT>::getM);
158 });
159}
160
164void declareSpatialKernelSolution(lsst::cpputils::python::WrapperCollection &wrappers) {
165 using PySpatialKernelSolution = py::classh<SpatialKernelSolution, KernelSolution> ;
166
167 wrappers.wrapType(PySpatialKernelSolution(wrappers.module, "SpatialKernelSolution"), [](auto &mod, auto &cls) {
168 cls.def(py::init<lsst::afw::math::KernelList const &, lsst::afw::math::Kernel::SpatialFunctionPtr,
169 lsst::afw::math::Kernel::SpatialFunctionPtr, daf::base::PropertySet const &>(),
170 "basisList"_a, "spatialKernelFunction"_a, "background"_a, "ps"_a);
171
172 cls.def("solve", (void (SpatialKernelSolution::*)()) &SpatialKernelSolution::solve);
173 cls.def("addConstraint", &SpatialKernelSolution::addConstraint, "xCenter"_a, "yCenter"_a, "qMat"_a,
174 "wVec"_a);
175 cls.def("makeKernelImage", &SpatialKernelSolution::makeKernelImage, "pos"_a);
176 cls.def("getSolutionPair", &SpatialKernelSolution::getSolutionPair);
177 });
178}
179
180} // namespace lsst::ip::diffim::<anonymous>
181
183 declareKernelSolution(wrappers);
184 declareStaticKernelSolution<float>(wrappers, "F");
185 declareMaskedKernelSolution<float>(wrappers, "F");
186 declareRegularizedKernelSolution<float>(wrappers, "F");
187 declareSpatialKernelSolution(wrappers);
188}
189
190} // diffim
191} // ip
192} // lsst
Declaration of classes to store the solution for convolution kernels.
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 wrapKernelSolution(WrapperCollection &wrappers)