LSST Applications g042eb84c57+730a74494b,g04e9c324dd+8c5ae1fdc5,g134cb467dc+1f1e3e7524,g199a45376c+0ba108daf9,g1fd858c14a+fa7d31856b,g210f2d0738+f66ac109ec,g262e1987ae+83a3acc0e5,g29ae962dfc+d856a2cb1f,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+a1e0c9f713,g47891489e3+0d594cb711,g4d44eb3520+c57ec8f3ed,g4d7b6aa1c5+f66ac109ec,g53246c7159+8c5ae1fdc5,g56a1a4eaf3+fd7ad03fde,g64539dfbff+f66ac109ec,g67b6fd64d1+0d594cb711,g67fd3c3899+f66ac109ec,g6985122a63+0d594cb711,g74acd417e5+3098891321,g786e29fd12+668abc6043,g81db2e9a8d+98e2ab9f28,g87389fa792+8856018cbb,g89139ef638+0d594cb711,g8d7436a09f+80fda9ce03,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+a8a29bda4b,g99822b682c+e3ec3c61f9,g9d5c6a246b+0d5dac0c3d,ga41d0fce20+9243b26dd2,gbf99507273+8c5ae1fdc5,gd7ef33dd92+0d594cb711,gdab6d2f7ff+3098891321,ge410e46f29+0d594cb711,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.38
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)