LSSTApplications  20.0.0
LSSTDataManagementBasePackage
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
lsst::afw::math::PYBIND11_MODULE
PYBIND11_MODULE(convolveImage, mod)
Definition: convolveImage.cc:74
lsst::afw::math::scaledPlus
void scaledPlus(OutImageT &outImage, double c1, InImageT const &inImage1, double c2, InImageT const &inImage2)
Compute the scaled sum of two images.
Definition: ConvolveImage.cc:162
lsst::afw
Definition: imageAlgorithm.dox:1
lsst::afw::image::MaskedImage
A class to manipulate images, masks, and variance as a single object.
Definition: MaskedImage.h:73
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
pybind11
Definition: _GenericMap.cc:40
lsst::afw::image::Image
A class to represent a 2-dimensional array of pixels.
Definition: Image.h:58
ConvolveImage.h
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42