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
kernelCandidate.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#include "pybind11/stl.h"
26
27#include "Eigen/Core"
28#include "ndarray/pybind11.h"
29
30#include <memory>
31#include <string>
32
35
36namespace py = pybind11;
37using namespace pybind11::literals;
38
39namespace lsst {
40namespace ip {
41namespace diffim {
42
43namespace {
44
52template <typename PixelT>
53void declareKernelCandidate(lsst::cpputils::python::WrapperCollection &wrappers, std::string const &suffix) {
54 using CandidateSwitch = typename KernelCandidate<PixelT>::CandidateSwitch;
55 using PyCandidateSwitch = py::enum_<CandidateSwitch>;
56
57 std::string name = "KernelCandidate" + suffix;
58 using PyKernelCandidate = py::classh<KernelCandidate<PixelT>, afw::math::SpatialCellImageCandidate>;
59
60 auto clsDef = wrappers.wrapType(PyKernelCandidate(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
61 cls.def(py::init<float const, float const, std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
62 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &>(),
63 "xCenter"_a, "yCenter"_a, "templateMaskedImage"_a, "scienceMaskedImage"_a, "ps"_a);
64 cls.def(py::init<std::shared_ptr<afw::table::SourceRecord> const &,
65 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
66 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &>(),
67 "source"_a, "templateMaskedImage"_a, "scienceMaskedImage"_a, "ps"_a);
68
69 cls.def("getCandidateRating", &KernelCandidate<PixelT>::getCandidateRating);
70 cls.def("getSource", &KernelCandidate<PixelT>::getSource);
71 cls.def("getTemplateMaskedImage", &KernelCandidate<PixelT>::getTemplateMaskedImage);
72 cls.def("getScienceMaskedImage", &KernelCandidate<PixelT>::getScienceMaskedImage);
73 cls.def("getKernel", &KernelCandidate<PixelT>::getKernel, "cand"_a);
74 cls.def("getBackground", &KernelCandidate<PixelT>::getBackground, "cand"_a);
75 cls.def("getKsum", &KernelCandidate<PixelT>::getKsum, "cand"_a);
76 cls.def("getKernelImage", &KernelCandidate<PixelT>::getKernelImage, "cand"_a);
77 cls.def("getImage", &KernelCandidate<PixelT>::getImage);
78 cls.def("getKernelSolution", &KernelCandidate<PixelT>::getKernelSolution, "cand"_a);
79 cls.def("getDifferenceImage",
80 (afw::image::MaskedImage<PixelT> (KernelCandidate<PixelT>::*)(CandidateSwitch)) &
81 KernelCandidate<PixelT>::getDifferenceImage,
82 "cand"_a);
83 cls.def("getDifferenceImage", (afw::image::MaskedImage<PixelT> (KernelCandidate<PixelT>::*)(
84 std::shared_ptr<afw::math::Kernel>, double)) &
85 KernelCandidate<PixelT>::getDifferenceImage,
86 "kernel"_a, "background"_a);
87 cls.def("isInitialized", &KernelCandidate<PixelT>::isInitialized);
88 cls.def("build", (void (KernelCandidate<PixelT>::*)(afw::math::KernelList const &)) &
89 KernelCandidate<PixelT>::build,
90 "basisList"_a);
91 cls.def("build",
92 (void (KernelCandidate<PixelT>::*)(afw::math::KernelList const &, Eigen::MatrixXd const &)) &
93 KernelCandidate<PixelT>::build,
94 "basisList"_a, "hMat"_a);
95 mod.def("makeKernelCandidate",
96 (std::shared_ptr<KernelCandidate<PixelT>>(*)(
97 float const, float const, std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
98 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &)) &
99 makeKernelCandidate,
100 "xCenter"_a, "yCenter"_a, "templateMaskedImage"_a, "scienceMaskedImage"_a, "ps"_a);
101 mod.def("makeKernelCandidate",
102 (std::shared_ptr<KernelCandidate<PixelT>>(*)(
103 std::shared_ptr<afw::table::SourceRecord> const &,
104 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
105 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &)) &
106 makeKernelCandidate,
107 "source"_a, "templateMaskedImage"_a, "scienceMaskedImage"_a, "ps"_a);
108 });
109 wrappers.wrapType(PyCandidateSwitch(clsDef, "CandidateSwitch"), [](auto &mod, auto &enm) {
110 enm.value("ORIG", CandidateSwitch::ORIG);
111 enm.value("PCA", CandidateSwitch::PCA);
112 enm.value("RECENT", CandidateSwitch::RECENT);
113 enm.export_values();
114 });
115}
116
117} // namespace lsst::ip::diffim::<anonymous>
118
120 declareKernelCandidate<float>(wrappers, "F");
121}
122
123} // diffim
124} // ip
125} // lsst
Class used by SpatialModelCell for spatial Kernel fitting.
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 wrapKernelCandidate(WrapperCollection &wrappers)