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
_warpExposure.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 <cstdint>
24 #include <memory>
25 #include <string>
26 
27 #include <pybind11/pybind11.h>
28 #include <lsst/utils/python.h>
29 
30 #include "lsst/afw/geom/SkyWcs.h"
32 #include "lsst/afw/image/Image.h"
34 #include "lsst/afw/math/Kernel.h"
37 
38 namespace py = pybind11;
39 using namespace py::literals;
40 
41 namespace lsst {
42 namespace afw {
43 namespace math {
44 
45 namespace {
54 template <typename KernelT>
55 void declareWarpingKernel(lsst::utils::python::WrapperCollection &wrappers, std::string const &name) {
56  using PyClass = py::class_<KernelT, std::shared_ptr<KernelT>, SeparableKernel>;
57  wrappers.wrapType(PyClass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
58  cls.def(py::init<int>(), "order"_a);
59  cls.def("getOrder", &KernelT::getOrder);
60  cls.def("clone", &KernelT::clone);
61  table::io::python::addPersistableMethods(cls);
62  });
63 }
64 
73 template <typename KernelT>
74 void declareSimpleWarpingKernel(lsst::utils::python::WrapperCollection &wrappers, std::string const &name) {
75  using PyClass = py::class_<KernelT, std::shared_ptr<KernelT>, SeparableKernel>;
76  wrappers.wrapType(PyClass(wrappers.module, name.c_str()), [](auto &mod, auto &cls) {
77  cls.def(py::init<>());
78  cls.def("clone", &KernelT::clone);
79  table::io::python::addPersistableMethods(cls);
80  });
81 }
82 
91 template <typename DestImageT, typename SrcImageT>
92 void declareImageWarpingFunctions(lsst::utils::python::WrapperCollection &wrappers) {
93  wrappers.wrap([](auto &mod) {
94  typename DestImageT::SinglePixel const EdgePixel =
95  edgePixel<DestImageT>(typename image::detail::image_traits<DestImageT>::image_category());
96  mod.def("warpImage",
97  (int (*)(DestImageT &, geom::SkyWcs const &, SrcImageT const &, geom::SkyWcs const &,
98  WarpingControl const &, typename DestImageT::SinglePixel)) &
99  warpImage<DestImageT, SrcImageT>,
100  "destImage"_a, "destWcs"_a, "srcImage"_a, "srcWcs"_a, "control"_a, "padValue"_a = EdgePixel);
101 
102  mod.def("warpImage",
103  (int (*)(DestImageT &, SrcImageT const &, geom::TransformPoint2ToPoint2 const &,
104  WarpingControl const &, typename DestImageT::SinglePixel)) &
105  warpImage<DestImageT, SrcImageT>,
106  "destImage"_a, "srcImage"_a, "srcToDest"_a, "control"_a, "padValue"_a = EdgePixel);
107 
108  mod.def("warpCenteredImage", &warpCenteredImage<DestImageT, SrcImageT>, "destImage"_a, "srcImage"_a,
109  "linearTransform"_a, "centerPoint"_a, "control"_a, "padValue"_a = EdgePixel);
110  });
111 }
112 
123 template <typename DestPixelT, typename SrcPixelT>
124 void declareWarpingFunctions(lsst::utils::python::WrapperCollection &wrappers) {
127  using DestImageT = image::Image<DestPixelT>;
128  using SrcImageT = image::Image<SrcPixelT>;
131  wrappers.wrap([](auto &mod) {
132  mod.def("warpExposure", &warpExposure<DestExposureT, SrcExposureT>, "destExposure"_a, "srcExposure"_a,
133  "control"_a,
134  "padValue"_a = edgePixel<DestMaskedImageT>(
135  typename image::detail::image_traits<DestMaskedImageT>::image_category()));
136  });
137  declareImageWarpingFunctions<DestImageT, SrcImageT>(wrappers);
138  declareImageWarpingFunctions<DestMaskedImageT, SrcMaskedImageT>(wrappers);
139 }
140 
141 void declareWarpExposure(lsst::utils::python::WrapperCollection &wrappers) {
142  using PyClass = py::class_<WarpingControl, std::shared_ptr<WarpingControl>>;
143  wrappers.wrapType(PyClass(wrappers.module, "WarpingControl"), [](auto &mod, auto cls) {
144  cls.def(py::init<std::string, std::string, int, int, image::MaskPixel>(), "warpingKernelName"_a,
145  "maskWarpingKernelName"_a = "", "cacheSize"_a = 0, "interpLength"_a = 0,
146  "growFullMask"_a = 0);
147 
148  cls.def("getCacheSize", &WarpingControl::getCacheSize);
149  cls.def("setCacheSize", &WarpingControl::setCacheSize, "cacheSize"_a);
150  cls.def("getInterpLength", &WarpingControl::getInterpLength);
151  cls.def("setInterpLength", &WarpingControl::setInterpLength, "interpLength"_a);
152  cls.def("setWarpingKernelName", &WarpingControl::setWarpingKernelName, "warpingKernelName"_a);
153  cls.def("getWarpingKernel", &WarpingControl::getWarpingKernel);
154  cls.def("setWarpingKernel", &WarpingControl::setWarpingKernel, "warpingKernel"_a);
155  cls.def("setMaskWarpingKernelName", &WarpingControl::setMaskWarpingKernelName,
156  "maskWarpingKernelName"_a);
157  cls.def("getMaskWarpingKernel", &WarpingControl::getMaskWarpingKernel);
158  cls.def("hasMaskWarpingKernel", &WarpingControl::hasMaskWarpingKernel);
159  cls.def("setMaskWarpingKernelName", &WarpingControl::setMaskWarpingKernelName,
160  "maskWarpingKernelName"_a);
161  cls.def("setMaskWarpingKernel", &WarpingControl::setMaskWarpingKernel, "maskWarpingKernel"_a);
162  cls.def("getGrowFullMask", &WarpingControl::getGrowFullMask);
163  cls.def("setGrowFullMask", &WarpingControl::setGrowFullMask, "growFullMask"_a);
164  table::io::python::addPersistableMethods(cls);
165  });
166 }
167 } // namespace
168 void wrapWarpExposure(lsst::utils::python::WrapperCollection &wrappers) {
169  wrappers.addSignatureDependency("lsst.afw.image");
170  wrappers.addSignatureDependency("lsst.afw.geom.skyWcs");
171 
172  declareWarpExposure(wrappers);
173  declareWarpingKernel<LanczosWarpingKernel>(wrappers, "LanczosWarpingKernel");
174  declareSimpleWarpingKernel<BilinearWarpingKernel>(wrappers, "BilinearWarpingKernel");
175  declareSimpleWarpingKernel<NearestWarpingKernel>(wrappers, "NearestWarpingKernel");
176  declareWarpingFunctions<double, double>(wrappers);
177  declareWarpingFunctions<double, float>(wrappers);
178  declareWarpingFunctions<double, int>(wrappers);
179  declareWarpingFunctions<double, std::uint16_t>(wrappers);
180  declareWarpingFunctions<float, float>(wrappers);
181  declareWarpingFunctions<float, int>(wrappers);
182  declareWarpingFunctions<float, std::uint16_t>(wrappers);
183  declareWarpingFunctions<int, int>(wrappers);
184  declareWarpingFunctions<std::uint16_t, std::uint16_t>(wrappers);
185 }
186 } // namespace math
187 } // namespace afw
188 } // namespace lsst
table::Key< std::string > name
Definition: Amplifier.cc:116
A class to contain the data, WCS, and other information needed to describe an image of the sky.
Definition: Exposure.h:72
A class to represent a 2-dimensional array of pixels.
Definition: Image.h:51
A class to manipulate images, masks, and variance as a single object.
Definition: MaskedImage.h:73
Transform< Point2Endpoint, Point2Endpoint > TransformPoint2ToPoint2
Definition: Transform.h:300
void wrapWarpExposure(lsst::utils::python::WrapperCollection &)
py::class_< PixelAreaBoundedField, std::shared_ptr< PixelAreaBoundedField >, BoundedField > PyClass
A base class for image defects.