LSSTApplications  20.0.0
LSSTDataManagementBasePackage
convolve.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 
31 using namespace py::literals;
32 
33 namespace lsst {
34 namespace afw {
35 namespace math {
36 namespace detail {
37 
38 namespace {
39 template <typename OutImageT, typename InImageT>
40 void declareByType(py::module &mod) {
41  mod.def("basicConvolve",
42  (void (*)(OutImageT &, InImageT const &, lsst::afw::math::Kernel const &,
43  lsst::afw::math::ConvolutionControl const &))basicConvolve<OutImageT, InImageT>);
44  mod.def("basicConvolve",
45  (void (*)(OutImageT &, InImageT const &, lsst::afw::math::DeltaFunctionKernel const &,
46  lsst::afw::math::ConvolutionControl const &))basicConvolve<OutImageT, InImageT>);
47  mod.def("basicConvolve",
48  (void (*)(OutImageT &, InImageT const &, lsst::afw::math::LinearCombinationKernel const &,
49  lsst::afw::math::ConvolutionControl const &))basicConvolve<OutImageT, InImageT>);
50  mod.def("basicConvolve",
51  (void (*)(OutImageT &, InImageT const &, lsst::afw::math::SeparableKernel const &,
52  lsst::afw::math::ConvolutionControl const &))basicConvolve<OutImageT, InImageT>);
53  mod.def("convolveWithBruteForce",
54  (void (*)(
55  OutImageT &, InImageT const &, lsst::afw::math::Kernel const &,
56  lsst::afw::math::ConvolutionControl const &))convolveWithBruteForce<OutImageT, InImageT>);
57 }
58 template <typename PixelType1, typename PixelType2>
59 void declareAll(py::module &mod) {
62 
63  declareByType<image::Image<PixelType1>, image::Image<PixelType2>>(mod);
64  declareByType<M1, M2>(mod);
65 }
66 } // namespace
67 
69  declareAll<double, double>(mod);
70  declareAll<double, float>(mod);
71  declareAll<double, int>(mod);
72  declareAll<double, std::uint16_t>(mod);
73  declareAll<float, float>(mod);
74  declareAll<float, int>(mod);
75  declareAll<float, std::uint16_t>(mod);
76  declareAll<int, int>(mod);
77  declareAll<std::uint16_t, std::uint16_t>(mod);
78 
79  py::class_<KernelImagesForRegion, std::shared_ptr<KernelImagesForRegion>> clsKernelImagesForRegion(
80  mod, "KernelImagesForRegion");
81 
82  py::enum_<KernelImagesForRegion::Location>(clsKernelImagesForRegion, "Location")
83  .value("BOTTOM_LEFT", KernelImagesForRegion::Location::BOTTOM_LEFT)
84  .value("BOTTOM_RIGHT", KernelImagesForRegion::Location::BOTTOM_RIGHT)
85  .value("TOP_LEFT", KernelImagesForRegion::Location::TOP_LEFT)
86  .value("TOP_RIGHT", KernelImagesForRegion::Location::TOP_RIGHT)
87  .export_values();
88 
89  clsKernelImagesForRegion.def(
91  lsst::geom::Point2I const &, bool>(),
92  "kernelPtr"_a, "bbox"_a, "xy0"_a, "doNormalize"_a);
93  clsKernelImagesForRegion.def(
98  "kernelPtr"_a, "bbox"_a, "xy0"_a, "doNormalize"_a, "bottomLeftImagePtr"_a,
99  "bottomRightImagePtr"_a, "topLeftImagePtr"_a, "topRightImagePtr"_a);
100 
101  clsKernelImagesForRegion.def("getBBox", &KernelImagesForRegion::getBBox);
102  clsKernelImagesForRegion.def("getXY0", &KernelImagesForRegion::getXY0);
103  clsKernelImagesForRegion.def("getDoNormalize", &KernelImagesForRegion::getDoNormalize);
104  clsKernelImagesForRegion.def("getImage", &KernelImagesForRegion::getImage);
105  clsKernelImagesForRegion.def("getKernel", &KernelImagesForRegion::getKernel);
106  clsKernelImagesForRegion.def("getPixelIndex", &KernelImagesForRegion::getPixelIndex);
107  clsKernelImagesForRegion.def("computeNextRow", &KernelImagesForRegion::computeNextRow);
108  clsKernelImagesForRegion.def_static("getMinInterpolationSize",
109  KernelImagesForRegion::getMinInterpolationSize);
110 
111  py::class_<RowOfKernelImagesForRegion, std::shared_ptr<RowOfKernelImagesForRegion>>
112  clsRowOfKernelImagesForRegion(mod, "RowOfKernelImagesForRegion");
113 
114  clsRowOfKernelImagesForRegion.def(py::init<int, int>(), "nx"_a, "ny"_a);
115 
116  clsRowOfKernelImagesForRegion.def("front", &RowOfKernelImagesForRegion::front);
117  clsRowOfKernelImagesForRegion.def("back", &RowOfKernelImagesForRegion::back);
118  clsRowOfKernelImagesForRegion.def("getNX", &RowOfKernelImagesForRegion::getNX);
119  clsRowOfKernelImagesForRegion.def("getNY", &RowOfKernelImagesForRegion::getNY);
120  clsRowOfKernelImagesForRegion.def("getYInd", &RowOfKernelImagesForRegion::getYInd);
121  clsRowOfKernelImagesForRegion.def("getRegion", &RowOfKernelImagesForRegion::getRegion);
122  clsRowOfKernelImagesForRegion.def("hasData", &RowOfKernelImagesForRegion::hasData);
123  clsRowOfKernelImagesForRegion.def("isLastRow", &RowOfKernelImagesForRegion::isLastRow);
124  clsRowOfKernelImagesForRegion.def("incrYInd", &RowOfKernelImagesForRegion::incrYInd);
125 
126  /* Module level */
127 
128  /* Member types and enums */
129 
130  /* Constructors */
131 
132  /* Operators */
133 
134  /* Members */
135 }
136 }
137 }
138 }
139 } // lsst::afw::math::detail
Convolve.h
std::shared_ptr< lsst::afw::math::Kernel const >
lsst::afw::math::DeltaFunctionKernel
A kernel that has only one non-zero pixel (of value 1)
Definition: Kernel.h:690
lsst::afw
Definition: imageAlgorithm.dox:1
lsst::afw::math::detail::PYBIND11_MODULE
PYBIND11_MODULE(convolve, mod)
Definition: convolve.cc:68
lsst::afw::image::MaskedImage
A class to manipulate images, masks, and variance as a single object.
Definition: MaskedImage.h:73
lsst::afw::math::LinearCombinationKernel
A kernel that is a linear combination of fixed basis kernels.
Definition: Kernel.h:751
lsst::afw::math.convolveImage.convolveImageContinued.convolve
convolve
Definition: convolveImageContinued.py:28
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
lsst::afw::math::ConvolutionControl
Parameters to control convolution.
Definition: ConvolveImage.h:50
lsst::geom::Point< int, 2 >
lsst::geom::Box2I
An integer coordinate rectangle.
Definition: Box.h:55
pybind11
Definition: _GenericMap.cc:40
lsst::afw::math::Kernel
Kernels are used for convolution with MaskedImages and (eventually) Images.
Definition: Kernel.h:111
lsst::afw::image::Image
A class to represent a 2-dimensional array of pixels.
Definition: Image.h:58
lsst::utils.tests.init
def init()
Definition: tests.py:58
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42
lsst::afw::math::SeparableKernel
A kernel described by a pair of functions: func(x, y) = colFunc(x) * rowFunc(y)
Definition: Kernel.h:907