LSST Applications 27.0.0,g0265f82a02+469cd937ee,g02d81e74bb+21ad69e7e1,g1470d8bcf6+cbe83ee85a,g2079a07aa2+e67c6346a6,g212a7c68fe+04a9158687,g2305ad1205+94392ce272,g295015adf3+81dd352a9d,g2bbee38e9b+469cd937ee,g337abbeb29+469cd937ee,g3939d97d7f+72a9f7b576,g487adcacf7+71499e7cba,g50ff169b8f+5929b3527e,g52b1c1532d+a6fc98d2e7,g591dd9f2cf+df404f777f,g5a732f18d5+be83d3ecdb,g64a986408d+21ad69e7e1,g858d7b2824+21ad69e7e1,g8a8a8dda67+a6fc98d2e7,g99cad8db69+f62e5b0af5,g9ddcbc5298+d4bad12328,ga1e77700b3+9c366c4306,ga8c6da7877+71e4819109,gb0e22166c9+25ba2f69a1,gb6a65358fc+469cd937ee,gbb8dafda3b+69d3c0e320,gc07e1c2157+a98bf949bb,gc120e1dc64+615ec43309,gc28159a63d+469cd937ee,gcf0d15dbbd+72a9f7b576,gdaeeff99f8+a38ce5ea23,ge6526c86ff+3a7c1ac5f1,ge79ae78c31+469cd937ee,gee10cc3b42+a6fc98d2e7,gf1cff7945b+21ad69e7e1,gfbcc870c63+9a11dc8c8f
LSST Data Management Base Package
Loading...
Searching...
No Matches
_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
30namespace py = pybind11;
31
32using namespace py::literals;
33
34namespace lsst {
35namespace afw {
36namespace image {
37
38namespace {
39
40template <typename ImageT>
41static 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
62template <typename Image1T, typename Image2T>
63static 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
73void 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
A class to represent a 2-dimensional array of pixels.
Definition Image.h:51
void wrapImagePca(lsst::utils::python::WrapperCollection &)
Definition _imagePca.cc:73