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