LSST Applications  21.0.0-172-gfb10e10a+18fedfabac,22.0.0+297cba6710,22.0.0+80564b0ff1,22.0.0+8d77f4f51a,22.0.0+a28f4c53b1,22.0.0+dcf3732eb2,22.0.1-1-g7d6de66+2a20fdde0d,22.0.1-1-g8e32f31+297cba6710,22.0.1-1-geca5380+7fa3b7d9b6,22.0.1-12-g44dc1dc+2a20fdde0d,22.0.1-15-g6a90155+515f58c32b,22.0.1-16-g9282f48+790f5f2caa,22.0.1-2-g92698f7+dcf3732eb2,22.0.1-2-ga9b0f51+7fa3b7d9b6,22.0.1-2-gd1925c9+bf4f0e694f,22.0.1-24-g1ad7a390+a9625a72a8,22.0.1-25-g5bf6245+3ad8ecd50b,22.0.1-25-gb120d7b+8b5510f75f,22.0.1-27-g97737f7+2a20fdde0d,22.0.1-32-gf62ce7b1+aa4237961e,22.0.1-4-g0b3f228+2a20fdde0d,22.0.1-4-g243d05b+871c1b8305,22.0.1-4-g3a563be+32dcf1063f,22.0.1-4-g44f2e3d+9e4ab0f4fa,22.0.1-42-gca6935d93+ba5e5ca3eb,22.0.1-5-g15c806e+85460ae5f3,22.0.1-5-g58711c4+611d128589,22.0.1-5-g75bb458+99c117b92f,22.0.1-6-g1c63a23+7fa3b7d9b6,22.0.1-6-g50866e6+84ff5a128b,22.0.1-6-g8d3140d+720564cf76,22.0.1-6-gd805d02+cc5644f571,22.0.1-8-ge5750ce+85460ae5f3,master-g6e05de7fdc+babf819c66,master-g99da0e417a+8d77f4f51a,w.2021.48
LSST Data Management Base Package
_imagePca.cc
Go to the documentation of this file.
1 /*
2  * This file is part of afw.
3  *
4  * Developed for the LSST Data Management System.
5  * This product includes software developed by the LSST Project
6  * (https://www.lsst.org).
7  * See the COPYRIGHT file at the top-level directory of this distribution
8  * for details of code ownership.
9  *
10  * This program is free software: you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation, either version 3 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program. If not, see <https://www.gnu.org/licenses/>.
22  */
23 
24 #include "pybind11/pybind11.h"
25 #include "pybind11/stl.h"
26 #include "lsst/utils/python.h"
27 
29 
30 namespace py = pybind11;
31 
32 using namespace py::literals;
33 
34 namespace lsst {
35 namespace afw {
36 namespace image {
37 
38 namespace {
39 
40 template <typename ImageT>
41 static void declareImagePca(lsst::utils::python::WrapperCollection &wrappers, std::string const &suffix) {
42  std::string name = "ImagePca" + suffix;
43  wrappers.wrapType(
44  py::class_<ImagePca<ImageT>, std::shared_ptr<ImagePca<ImageT>>>(wrappers.module, name.c_str()),
45  [](auto &mod, auto &cls) {
46  // py::class_<ImagePca<ImageT>, std::shared_ptr<ImagePca<ImageT>>> cls(mod, ("ImagePca" +
47  // suffix).c_str());
48 
49  cls.def(py::init<bool>(), "constantWeight"_a = true);
50 
51  cls.def("addImage", &ImagePca<ImageT>::addImage, "img"_a, "flux"_a = 0.0);
52  cls.def("getImageList", &ImagePca<ImageT>::getImageList);
53  cls.def("getDimensions", &ImagePca<ImageT>::getDimensions);
54  cls.def("getMean", &ImagePca<ImageT>::getMean);
55  cls.def("analyze", &ImagePca<ImageT>::analyze);
56  cls.def("updateBadPixels", &ImagePca<ImageT>::updateBadPixels);
57  cls.def("getEigenValues", &ImagePca<ImageT>::getEigenValues);
58  cls.def("getEigenImages", &ImagePca<ImageT>::getEigenImages);
59  });
60 }
61 
62 template <typename Image1T, typename Image2T>
63 static void declareInnerProduct(lsst::utils::python::WrapperCollection &wrappers) {
64  wrappers.wrap([](auto &mod) {
65  mod.def("innerProduct",
66  (double (*)(Image1T const &, Image2T const &, int const))innerProduct<Image1T, Image2T>,
67  "lhs"_a, "rhs"_a, "border"_a = 0);
68  });
69 }
70 
71 } // namespace
72 
73 void wrapImagePca(lsst::utils::python::WrapperCollection &wrappers) {
74  declareImagePca<Image<int>>(wrappers, "I");
75  declareImagePca<Image<float>>(wrappers, "F");
76  declareImagePca<Image<double>>(wrappers, "D");
77  declareImagePca<Image<std::uint16_t>>(wrappers, "U");
78  declareImagePca<Image<std::uint64_t>>(wrappers, "L");
79  declareImagePca<MaskedImage<int>>(wrappers, "MI");
80  declareImagePca<MaskedImage<float>>(wrappers, "MF");
81  declareImagePca<MaskedImage<double>>(wrappers, "MD");
82  declareImagePca<MaskedImage<std::uint16_t>>(wrappers, "MU");
83  declareImagePca<MaskedImage<std::uint64_t>>(wrappers, "ML");
84 
85  declareInnerProduct<Image<int>, Image<int>>(wrappers);
86  declareInnerProduct<Image<float>, Image<float>>(wrappers);
87  declareInnerProduct<Image<double>, Image<double>>(wrappers);
88  declareInnerProduct<Image<std::uint16_t>, Image<std::uint16_t>>(wrappers);
89  declareInnerProduct<Image<std::uint64_t>, Image<std::uint64_t>>(wrappers);
90 
91  declareInnerProduct<Image<float>, Image<double>>(wrappers);
92  declareInnerProduct<Image<double>, Image<float>>(wrappers);
93 }
94 } // namespace image
95 } // namespace afw
96 } // namespace lsst
table::Key< std::string > name
Definition: Amplifier.cc:116
A class to represent a 2-dimensional array of pixels.
Definition: Image.h:51
Backwards-compatibility support for depersisting the old Calib (FluxMag0/FluxMag0Err) objects.
void wrapImagePca(lsst::utils::python::WrapperCollection &)
Definition: _imagePca.cc:73
A base class for image defects.