LSST Applications 26.0.0,g0265f82a02+6660c170cc,g07994bdeae+30b05a742e,g0a0026dc87+17526d298f,g0a60f58ba1+17526d298f,g0e4bf8285c+96dd2c2ea9,g0ecae5effc+c266a536c8,g1e7d6db67d+6f7cb1f4bb,g26482f50c6+6346c0633c,g2bbee38e9b+6660c170cc,g2cc88a2952+0a4e78cd49,g3273194fdb+f6908454ef,g337abbeb29+6660c170cc,g337c41fc51+9a8f8f0815,g37c6e7c3d5+7bbafe9d37,g44018dc512+6660c170cc,g4a941329ef+4f7594a38e,g4c90b7bd52+5145c320d2,g58be5f913a+bea990ba40,g635b316a6c+8d6b3a3e56,g67924a670a+bfead8c487,g6ae5381d9b+81bc2a20b4,g93c4d6e787+26b17396bd,g98cecbdb62+ed2cb6d659,g98ffbb4407+81bc2a20b4,g9ddcbc5298+7f7571301f,ga1e77700b3+99e9273977,gae46bcf261+6660c170cc,gb2715bf1a1+17526d298f,gc86a011abf+17526d298f,gcf0d15dbbd+96dd2c2ea9,gdaeeff99f8+0d8dbea60f,gdb4ec4c597+6660c170cc,ge23793e450+96dd2c2ea9,gf041782ebf+171108ac67
LSST Data Management Base Package
Loading...
Searching...
No Matches
_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
35namespace py = pybind11;
36using namespace py::literals;
37
38namespace lsst {
39namespace afw {
40namespace detection {
41
42using utils::python::WrapperCollection;
43
44namespace {
45template <typename ImagePixelT, typename MaskPixelT = lsst::afw::image::MaskPixel,
46 typename VariancePixelT = lsst::afw::image::VariancePixel>
47void 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
88void 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:74
void wrapHeavyFootprint(WrapperCollection &)
float VariancePixel
default type for MaskedImage variance images
std::int32_t MaskPixel
default type for Masks and MaskedImage Masks