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
polynomialTransform.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  *
4  * This product includes software developed by the
5  * LSST Project (http://www.lsst.org/).
6  * See the COPYRIGHT file
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 #include "pybind11/pybind11.h"
23 #include "pybind11/stl.h"
24 
25 #include <memory>
26 
27 #include "ndarray/pybind11.h"
28 
32 
33 namespace py = pybind11;
34 using namespace pybind11::literals;
35 
36 namespace lsst {
37 namespace meas {
38 namespace astrom {
39 
40 namespace {
41 
42 static void declarePolynomialTransform(py::module &mod) {
43  py::class_<PolynomialTransform, std::shared_ptr<PolynomialTransform>> cls(mod, "PolynomialTransform");
44 
45  cls.def(py::init<ndarray::Array<double const, 2, 0> const &,
46  ndarray::Array<double const, 2, 0> const &>(),
47  "xCoeffs"_a, "yCoeffs"_a);
48  cls.def(py::init<PolynomialTransform const &>(), "other"_a);
49 
50  cls.def_static("convert",
51  (PolynomialTransform(*)(ScaledPolynomialTransform const &)) & PolynomialTransform::convert,
52  "other"_a);
53  cls.def_static("convert",
54  (PolynomialTransform(*)(SipForwardTransform const &)) & PolynomialTransform::convert,
55  "other"_a);
56  cls.def_static("convert",
57  (PolynomialTransform(*)(SipReverseTransform const &)) & PolynomialTransform::convert,
58  "other"_a);
59 
60  cls.def("__call__", &PolynomialTransform::operator(), "in"_a);
61 
62  cls.def("getOrder", &PolynomialTransform::getOrder);
63  cls.def("getXCoeffs", &PolynomialTransform::getXCoeffs);
64  cls.def("getYCoeffs", &PolynomialTransform::getYCoeffs);
65  cls.def("linearize", &PolynomialTransform::linearize);
66 }
67 
68 static void declareScaledPolynomialTransform(py::module &mod) {
69  py::class_<ScaledPolynomialTransform, std::shared_ptr<ScaledPolynomialTransform>> cls(
70  mod, "ScaledPolynomialTransform");
71 
72  cls.def(py::init<PolynomialTransform const &, geom::AffineTransform const &,
73  geom::AffineTransform const &>(),
74  "poly"_a, "inputScaling"_a, "outputScalingInverse"_a);
75  cls.def(py::init<ScaledPolynomialTransform const &>(), "other"_a);
76 
77  cls.def_static(
78  "convert",
79  (ScaledPolynomialTransform(*)(PolynomialTransform const &)) & ScaledPolynomialTransform::convert,
80  "other"_a);
81  cls.def_static(
82  "convert",
83  (ScaledPolynomialTransform(*)(SipForwardTransform const &)) & ScaledPolynomialTransform::convert,
84  "other"_a);
85  cls.def_static(
86  "convert",
87  (ScaledPolynomialTransform(*)(SipReverseTransform const &)) & ScaledPolynomialTransform::convert,
88  "other"_a);
89 
90  cls.def("__call__", &ScaledPolynomialTransform::operator(), "in"_a);
91 
92  cls.def("getPoly", &ScaledPolynomialTransform::getPoly, py::return_value_policy::reference_internal);
93  cls.def("getInputScaling", &ScaledPolynomialTransform::getInputScaling,
94  py::return_value_policy::reference_internal);
95  cls.def("getOutputScalingInverse", &ScaledPolynomialTransform::getOutputScalingInverse,
96  py::return_value_policy::reference_internal);
97  cls.def("linearize", &ScaledPolynomialTransform::linearize);
98 }
99 
100 } // namespace
101 
102 PYBIND11_MODULE(polynomialTransform, mod) {
103  declarePolynomialTransform(mod);
104  declareScaledPolynomialTransform(mod);
105 
106  mod.def("compose",
108  "t1"_a, "t2"_a);
109  mod.def("compose",
111  "t1"_a, "t2"_a);
112 }
113 
114 } // namespace astrom
115 } // namespace meas
116 } // namespace lsst
An affine coordinate transformation consisting of a linear transformation and an offset.
A 2-d coordinate transform represented by a pair of standard polynomials (one for each coordinate).
std::unique_ptr< AstrometryTransform > compose(AstrometryTransform const &left, AstrometryTransform const &right)
Returns a pointer to a composition of transforms, representing left(right()).
PYBIND11_MODULE(polynomialTransform, mod)
A base class for image defects.