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
_heavyFootprint.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 
27 #include "ndarray/pybind11.h"
28 
29 #include "lsst/utils/python.h"
30 
34 
35 namespace py = pybind11;
36 using namespace py::literals;
37 
38 namespace lsst {
39 namespace afw {
40 namespace detection {
41 
42 using utils::python::WrapperCollection;
43 
44 namespace {
45 template <typename ImagePixelT, typename MaskPixelT = lsst::afw::image::MaskPixel,
46  typename VariancePixelT = lsst::afw::image::VariancePixel>
47 void declareHeavyFootprint(WrapperCollection &wrappers, std::string const &suffix) {
48  using Class = HeavyFootprint<ImagePixelT>;
49  wrappers.wrapType(
50  py::class_<Class, std::shared_ptr<Class>, Footprint>(wrappers.module,
51  ("HeavyFootprint" + suffix).c_str()),
52  [](auto &mod, auto &cls) {
53  cls.def(py::init<Footprint const &,
54  lsst::afw::image::MaskedImage<ImagePixelT, MaskPixelT, VariancePixelT> const
55  &,
56  HeavyFootprintCtrl const *>(),
57  "foot"_a, "mimage"_a, "ctrl"_a = nullptr);
58  cls.def(py::init<Footprint const &, HeavyFootprintCtrl const *>(), "foot"_a,
59  "ctrl"_a = nullptr);
60 
61  cls.def("isHeavy", &Class::isHeavy);
62  cls.def("insert", (void (Class::*)(lsst::afw::image::MaskedImage<ImagePixelT> &) const) &
63  Class::insert);
64  cls.def("insert",
65  (void (Class::*)(lsst::afw::image::Image<ImagePixelT> &) const) & Class::insert);
66  cls.def("getImageArray",
67  (ndarray::Array<ImagePixelT, 1, 1>(Class::*)()) & Class::getImageArray);
68  cls.def("getMaskArray", (ndarray::Array<MaskPixelT, 1, 1>(Class::*)()) & Class::getMaskArray);
69  cls.def("getVarianceArray",
70  (ndarray::Array<VariancePixelT, 1, 1>(Class::*)()) & Class::getVarianceArray);
71  cls.def("getMaskBitsSet", &Class::getMaskBitsSet);
72  cls.def("dot", &Class::dot);
73  });
74 
75  wrappers.wrap([](auto &mod) {
76  mod.def("makeHeavyFootprint",
77  (Class(*)(Footprint const &,
79  HeavyFootprintCtrl const *))
80  makeHeavyFootprint<ImagePixelT, MaskPixelT, VariancePixelT>,
81  "foot"_a, "img"_a, "ctrl"_a = nullptr);
82 
83  mod.def("mergeHeavyFootprints", mergeHeavyFootprints<ImagePixelT, MaskPixelT, VariancePixelT>);
84  });
85 }
86 } // namespace
87 
88 void wrapHeavyFootprint(WrapperCollection &wrappers) {
89  wrappers.addSignatureDependency("lsst.afw.image");
90 
91  declareHeavyFootprint<int>(wrappers, "I");
92  declareHeavyFootprint<std::uint16_t>(wrappers, "U");
93  declareHeavyFootprint<float>(wrappers, "F");
94  declareHeavyFootprint<double>(wrappers, "D");
95 }
96 
97 } // namespace detection
98 } // namespace afw
99 } // namespace lsst
A class to manipulate images, masks, and variance as a single object.
Definition: MaskedImage.h:73
void wrapHeavyFootprint(WrapperCollection &)
float VariancePixel
default type for MaskedImage variance images
std::int32_t MaskPixel
default type for Masks and MaskedImage Masks
lsst::afw::detection::Footprint Footprint
Definition: Source.h:57
A base class for image defects.