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
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)