LSST Applications  21.0.0+c4f5df5339,21.0.0+e70536a077,21.0.0-1-ga51b5d4+7c60f8a6ea,21.0.0-10-gcf60f90+74aa0801fd,21.0.0-12-g63909ac9+643a1044a5,21.0.0-15-gedb9d5423+1041c3824f,21.0.0-2-g103fe59+a356b2badb,21.0.0-2-g1367e85+6d3f3f41db,21.0.0-2-g45278ab+e70536a077,21.0.0-2-g5242d73+6d3f3f41db,21.0.0-2-g7f82c8f+8d7c04eab9,21.0.0-2-g8f08a60+9c9a9cfcc8,21.0.0-2-ga326454+8d7c04eab9,21.0.0-2-gde069b7+bedfc5e1fb,21.0.0-2-gecfae73+6cb6558258,21.0.0-2-gfc62afb+6d3f3f41db,21.0.0-3-g21c7a62+f6e98b25aa,21.0.0-3-g357aad2+bd62456bea,21.0.0-3-g4be5c26+6d3f3f41db,21.0.0-3-g65f322c+03a4076c01,21.0.0-3-g7d9da8d+c4f5df5339,21.0.0-3-gaa929c8+c6b98066dc,21.0.0-3-gc44e71e+a26d5c1aea,21.0.0-3-ge02ed75+04b527a9d5,21.0.0-35-g64f566ff+b27e5ef93e,21.0.0-4-g591bb35+04b527a9d5,21.0.0-4-g88306b8+8773047b2e,21.0.0-4-gc004bbf+80a0b7acb7,21.0.0-4-gccdca77+a5c54364a0,21.0.0-4-ge8fba5a+ccfc328107,21.0.0-5-gdf36809+87b8d260e6,21.0.0-6-g00874e7+7eeda2b6ba,21.0.0-6-g2d4f3f3+e70536a077,21.0.0-6-g4e60332+04b527a9d5,21.0.0-6-g5ef7dad+f53629abd8,21.0.0-7-gc8ca178+b63e69433b,21.0.0-8-gfbe0b4b+c6b98066dc,w.2021.06
LSST Data Management Base Package
convolveImage.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  * Copyright 2008-2016 AURA/LSST.
4  *
5  * This product includes software developed by the
6  * LSST Project (http://www.lsst.org/).
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 
23 #include <pybind11/pybind11.h>
24 //#include <pybind11/operators.h>
25 //#include <pybind11/stl.h>
26 
28 
29 namespace py = pybind11;
30 using namespace py::literals;
31 
32 namespace lsst {
33 namespace afw {
34 namespace math {
35 
36 namespace {
37 template <typename OutImageT, typename InImageT, typename KernelT>
38 void declareConvolve(py::module &mod) {
39  mod.def("convolve", (void (*)(OutImageT &, InImageT const &, KernelT const &,
40  ConvolutionControl const &))convolve<OutImageT, InImageT, KernelT>,
41  "convolvedImage"_a, "inImage"_a, "kernel"_a, "convolutionControl"_a = ConvolutionControl());
42  mod.def("convolve", (void (*)(OutImageT &, InImageT const &, KernelT const &, bool,
43  bool))convolve<OutImageT, InImageT, KernelT>,
44  "convolvedImage"_a, "inImage"_a, "kernel"_a, "doNormalize"_a, "doCopyEdge"_a = false);
45 }
46 
47 template <typename ImageType1, typename ImageType2>
48 void declareScaledPlus(py::module &mod) {
49  mod.def("scaledPlus",
50  (void (*)(ImageType1 &, double, ImageType2 const &, double, ImageType2 const &))scaledPlus);
51 }
52 
53 template <typename ImageType1, typename ImageType2>
54 void declareByType(py::module &mod) {
55  declareConvolve<ImageType1, ImageType2, AnalyticKernel>(mod);
56  declareConvolve<ImageType1, ImageType2, DeltaFunctionKernel>(mod);
57  declareConvolve<ImageType1, ImageType2, FixedKernel>(mod);
58  declareConvolve<ImageType1, ImageType2, LinearCombinationKernel>(mod);
59  declareConvolve<ImageType1, ImageType2, SeparableKernel>(mod);
60  declareConvolve<ImageType1, ImageType2, Kernel>(mod);
61  declareScaledPlus<ImageType1, ImageType2>(mod);
62 }
63 
64 template <typename PixelType1, typename PixelType2>
65 void declareAll(py::module &mod) {
68 
69  declareByType<image::Image<PixelType1>, image::Image<PixelType2>>(mod);
70  declareByType<M1, M2>(mod);
71 }
72 }
73 
74 PYBIND11_MODULE(convolveImage, mod) {
75  py::class_<ConvolutionControl, std::shared_ptr<ConvolutionControl>> clsConvolutionControl(
76  mod, "ConvolutionControl");
77 
78  clsConvolutionControl.def(py::init<bool, bool, int>(), "doNormalize"_a = true, "doCopyEdge"_a = false,
79  "maxInterpolationDistance"_a = 10);
80 
81  clsConvolutionControl.def("getDoNormalize", &ConvolutionControl::getDoNormalize);
82  clsConvolutionControl.def("getDoCopyEdge", &ConvolutionControl::getDoCopyEdge);
83  clsConvolutionControl.def("getMaxInterpolationDistance",
84  &ConvolutionControl::getMaxInterpolationDistance);
85  clsConvolutionControl.def("setDoNormalize", &ConvolutionControl::setDoNormalize);
86  clsConvolutionControl.def("setDoCopyEdge", &ConvolutionControl::setDoCopyEdge);
87  clsConvolutionControl.def("setMaxInterpolationDistance",
88  &ConvolutionControl::setMaxInterpolationDistance);
89 
90  declareAll<double, double>(mod);
91  declareAll<double, float>(mod);
92  declareAll<double, int>(mod);
93  declareAll<double, std::uint16_t>(mod);
94  declareAll<float, float>(mod);
95  declareAll<float, int>(mod);
96  declareAll<float, std::uint16_t>(mod);
97  declareAll<int, int>(mod);
98  declareAll<std::uint16_t, std::uint16_t>(mod);
99 }
100 }
101 }
102 } // lsst::afw::math
A class to represent a 2-dimensional array of pixels.
Definition: Image.h:58
A class to manipulate images, masks, and variance as a single object.
Definition: MaskedImage.h:73
PYBIND11_MODULE(convolveImage, mod)
void scaledPlus(OutImageT &outImage, double c1, InImageT const &inImage1, double c2, InImageT const &inImage2)
Compute the scaled sum of two images.
A base class for image defects.