LSSTApplications  19.0.0-14-gb0260a2+72efe9b372,20.0.0+7927753e06,20.0.0+8829bf0056,20.0.0+995114c5d2,20.0.0+b6f4b2abd1,20.0.0+bddc4f4cbe,20.0.0-1-g253301a+8829bf0056,20.0.0-1-g2b7511a+0d71a2d77f,20.0.0-1-g5b95a8c+7461dd0434,20.0.0-12-g321c96ea+23efe4bbff,20.0.0-16-gfab17e72e+fdf35455f6,20.0.0-2-g0070d88+ba3ffc8f0b,20.0.0-2-g4dae9ad+ee58a624b3,20.0.0-2-g61b8584+5d3db074ba,20.0.0-2-gb780d76+d529cf1a41,20.0.0-2-ged6426c+226a441f5f,20.0.0-2-gf072044+8829bf0056,20.0.0-2-gf1f7952+ee58a624b3,20.0.0-20-geae50cf+e37fec0aee,20.0.0-25-g3dcad98+544a109665,20.0.0-25-g5eafb0f+ee58a624b3,20.0.0-27-g64178ef+f1f297b00a,20.0.0-3-g4cc78c6+e0676b0dc8,20.0.0-3-g8f21e14+4fd2c12c9a,20.0.0-3-gbd60e8c+187b78b4b8,20.0.0-3-gbecbe05+48431fa087,20.0.0-38-ge4adf513+a12e1f8e37,20.0.0-4-g97dc21a+544a109665,20.0.0-4-gb4befbc+087873070b,20.0.0-4-gf910f65+5d3db074ba,20.0.0-5-gdfe0fee+199202a608,20.0.0-5-gfbfe500+d529cf1a41,20.0.0-6-g64f541c+d529cf1a41,20.0.0-6-g9a5b7a1+a1cd37312e,20.0.0-68-ga3f3dda+5fca18c6a4,20.0.0-9-g4aef684+e18322736b,w.2020.45
LSSTDataManagementBasePackage
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"
23 #include "pybind11/eigen.h"
24 #include "pybind11/stl.h"
25 
26 #include <memory>
27 #include <string>
28 
30 #include "lsst/afw/math/Kernel.h"
33 
34 namespace py = pybind11;
35 using namespace pybind11::literals;
36 
37 namespace lsst {
38 namespace ip {
39 namespace diffim {
40 namespace detail {
41 
42 namespace {
43 
51 template <typename PixelT>
52 void declareKernelPca(py::module& mod, std::string const& suffix) {
53  using ImageT = afw::image::Image<PixelT>;
54  py::class_<KernelPca<ImageT>, std::shared_ptr<KernelPca<ImageT>>, afw::image::ImagePca<ImageT>> cls(
55  mod, ("KernelPca" + suffix).c_str());
56 
57  cls.def(py::init<bool>(), "constantWeight"_a = true);
58 
59  cls.def("analyze", &KernelPca<ImageT>::analyze);
60 }
61 
69 template <typename PixelT>
70 void declareKernelPcaVisitor(py::module& mod, std::string const& suffix) {
71  py::class_<KernelPcaVisitor<PixelT>, std::shared_ptr<KernelPcaVisitor<PixelT>>,
72  afw::math::CandidateVisitor>
73  cls(mod, ("KernelPcaVisitor" + suffix).c_str());
74 
75  // note that KernelPcaVisitor<PixelT>::ImageT
76  // is always the same (pixels of type lsst::afw::math::Kernel::Pixel, not PixelT)
77  using KernelImageT = typename KernelPcaVisitor<PixelT>::ImageT;
78  cls.def(py::init<std::shared_ptr<KernelPca<KernelImageT>>>(), "imagePca"_a);
79 
80  cls.def("getEigenKernels", &KernelPcaVisitor<PixelT>::getEigenKernels);
81  cls.def("processCandidate", &KernelPcaVisitor<PixelT>::processCandidate, "candidate"_a);
82  cls.def("subtractMean", &KernelPcaVisitor<PixelT>::subtractMean);
83  cls.def("returnMean", &KernelPcaVisitor<PixelT>::returnMean);
84 
85  mod.def("makeKernelPcaVisitor", &makeKernelPcaVisitor<PixelT>, "imagePca"_a);
86 }
87 
88 } // namespace lsst::ip::diffim::detail::<anonymous>
89 
90 PYBIND11_MODULE(kernelPca, mod) {
91  py::module::import("lsst.afw.image");
92  py::module::import("lsst.afw.math");
93 
94  declareKernelPca<afw::math::Kernel::Pixel>(mod, "D");
95  declareKernelPcaVisitor<float>(mod, "F");
96 }
97 
98 } // detail
99 } // diffim
100 } // ip
101 } // lsst::ip::diffim::detail
std::string
STL class.
std::shared_ptr
STL class.
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
SpatialCell.h
lsst::ip::diffim::detail::PYBIND11_MODULE
PYBIND11_MODULE(kernelPca, mod)
Definition: kernelPca.cc:90
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
KernelPca.h
Declaration of KernelPca and KernelPcaVisitor.
Kernel.h
pybind11
Definition: _GenericMap.cc:40
lsst::utils.tests.init
def init()
Definition: tests.py:59
ImagePca.h
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42