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
kernelPca.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/stl.h"
25
26#include <memory>
27#include <string>
28
33
34namespace py = pybind11;
35using namespace pybind11::literals;
36
37namespace lsst {
38namespace ip {
39namespace diffim {
40namespace detail {
41
42namespace {
43
51template <typename PixelT>
52void declareKernelPca(lsst::cpputils::python::WrapperCollection &wrappers, std::string const& suffix) {
53 using ImageT = afw::image::Image<PixelT>;
54 using PyClass = py::classh<KernelPca<ImageT>, afw::image::ImagePca<ImageT>>;
55
56 std::string name = "KernelPca" + suffix;
57 wrappers.wrapType(PyClass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
58
59 cls.def(py::init<bool>(), "constantWeight"_a = true);
60
61 cls.def("analyze", &KernelPca<ImageT>::analyze);
62 });
63}
64
72template <typename PixelT>
73void declareKernelPcaVisitor(lsst::cpputils::python::WrapperCollection &wrappers, std::string const& suffix) {
74 using PyClass = py::classh<KernelPcaVisitor<PixelT>, afw::math::CandidateVisitor>;
75
76 std::string name = "KernelPcaVisitor" + suffix;
77 // note that KernelPcaVisitor<PixelT>::ImageT
78 // is always the same (pixels of type lsst::afw::math::Kernel::Pixel, not PixelT)
79 using KernelImageT = typename KernelPcaVisitor<PixelT>::ImageT;
80
81 wrappers.wrapType(PyClass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
82 cls.def(py::init<std::shared_ptr<KernelPca<KernelImageT>>>(), "imagePca"_a);
83
84 cls.def("getEigenKernels", &KernelPcaVisitor<PixelT>::getEigenKernels);
85 cls.def("processCandidate", &KernelPcaVisitor<PixelT>::processCandidate, "candidate"_a);
86 cls.def("subtractMean", &KernelPcaVisitor<PixelT>::subtractMean);
87 cls.def("returnMean", &KernelPcaVisitor<PixelT>::returnMean);
88
89 mod.def("makeKernelPcaVisitor", &makeKernelPcaVisitor<PixelT>, "imagePca"_a);
90 });
91}
92
93} // namespace lsst::ip::diffim::detail::<anonymous>
94
96 declareKernelPca<afw::math::Kernel::Pixel>(wrappers, "D");
97 declareKernelPcaVisitor<float>(wrappers, "F");
98}
99
100} // detail
101} // diffim
102} // ip
103} // lsst::ip::diffim::detail
Declaration of KernelPca and KernelPcaVisitor.
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
lsst::afw::image::Image< lsst::afw::math::Kernel::Pixel > ImageT
Definition KernelPca.h:42
py::classh< PixelAreaBoundedField, BoundedField > PyClass
void wrapKernelPca(WrapperCollection &wrappers)
Definition kernelPca.cc:95