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
_separable.cc
Go to the documentation of this file.
1 /*
2  * This file is part of afw.
3  *
4  * Developed for the LSST Data Management System.
5  * This product includes software developed by the LSST Project
6  * (https://www.lsst.org).
7  * See the COPYRIGHT file at the top-level directory of this distribution
8  * for details of code ownership.
9  *
10  * This program is free software: you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation, either version 3 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful,
16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18  * GNU General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program. If not, see <https://www.gnu.org/licenses/>.
22  */
23 
24 #include <pybind11/pybind11.h>
25 #include <lsst/utils/python.h>
26 
33 
34 namespace py = pybind11;
35 
36 using namespace py::literals;
37 
38 namespace lsst {
39 namespace afw {
40 namespace geom {
41 namespace ellipses {
42 template <typename Ellipticity_, typename Radius_>
43 void declareSeparable(lsst::utils::python::WrapperCollection &wrappers, const std::string &suffix) {
45  wrappers.wrapType(
46  py::class_<Class, std::shared_ptr<Class>, BaseCore>(wrappers.module,
47  ("Separable" + suffix).c_str()),
48  [](auto &mod, auto &cls) {
49  cls.def(py::init<double, double, double, bool>(), "e1"_a = 0.0, "e2"_a = 0.0,
50  "radius"_a = Radius_(), "normalize"_a = true);
51  cls.def(py::init<Class const &>());
52  cls.def(py::init<BaseCore const &>());
53 
54  cls.def("getE1", &Class::getE1);
55  cls.def("setE1", &Class::setE1);
56  cls.def("getE2", &Class::getE2);
57  cls.def("setE2", &Class::setE2);
58  cls.def("getRadius", (Radius_ const &(Class::*)() const) & Class::getRadius);
59  cls.def("setRadius", (void (Class::*)(double)) & Class::setRadius);
60  cls.def("setRadius", (void (Class::*)(Radius_ const &)) & Class::setRadius);
61  cls.def("getEllipticity", (Ellipticity_ const &(Class::*)() const) & Class::getEllipticity);
62  cls.def("clone", &Class::clone);
63  cls.def("getName", &Class::getName);
64  cls.def("normalize", &Class::normalize);
65  cls.def("assign", [](Class &self, Class &other) { self = other; });
66  cls.def("assign", [](Class &self, BaseCore &other) { self = other; });
67  cls.def("transform", [](Class &self, lsst::geom::LinearTransform const &t) {
68  return std::static_pointer_cast<Class>(self.transform(t).copy());
69  });
70  cls.def("transformInPlace", [](Class &self, lsst::geom::LinearTransform const &t) {
71  self.transform(t).inPlace();
72  });
73  cls.def("__str__", [](Class &self) {
74  return py::str("({}, {})").format(self.getEllipticity(), self.getRadius());
75  });
76  cls.def("__repr__", [](Class &self) {
77  return py::str("Separable({}, {})").format(self.getEllipticity(), self.getRadius());
78  });
79  });
80 }
81 
82 void wrapSeparable(lsst::utils::python::WrapperCollection &wrappers) {
83  declareSeparable<Distortion, DeterminantRadius>(wrappers, "DistortionDeterminantRadius");
84  declareSeparable<Distortion, TraceRadius>(wrappers, "DistortionTraceRadius");
85  declareSeparable<Distortion, LogDeterminantRadius>(wrappers, "DistortionLogDeterminantRadius");
86  declareSeparable<Distortion, LogTraceRadius>(wrappers, "DistortionLogTraceRadius");
87 
88  declareSeparable<ConformalShear, DeterminantRadius>(wrappers, "ConformalShearDeterminantRadius");
89  declareSeparable<ConformalShear, TraceRadius>(wrappers, "ConformalShearTraceRadius");
90  declareSeparable<ConformalShear, LogDeterminantRadius>(wrappers, "ConformalShearLogDeterminantRadius");
91  declareSeparable<ConformalShear, LogTraceRadius>(wrappers, "ConformalShearLogTraceRadius");
92 
93  declareSeparable<ReducedShear, DeterminantRadius>(wrappers, "ReducedShearDeterminantRadius");
94  declareSeparable<ReducedShear, TraceRadius>(wrappers, "ReducedShearTraceRadius");
95  declareSeparable<ReducedShear, LogDeterminantRadius>(wrappers, "ReducedShearLogDeterminantRadius");
96  declareSeparable<ReducedShear, LogTraceRadius>(wrappers, "ReducedShearLogTraceRadius");
97 }
98 } // namespace ellipses
99 } // namespace geom
100 } // namespace afw
101 } // namespace lsst
table::Key< int > transform
A base class for parametrizations of the "core" of an ellipse - the ellipticity and size.
Definition: BaseCore.h:55
An ellipse core with a complex ellipticity and radius parameterization.
Definition: Separable.h:50
A 2D linear coordinate transformation.
void wrapSeparable(lsst::utils::python::WrapperCollection &)
Definition: _separable.cc:82
void declareSeparable(lsst::utils::python::WrapperCollection &wrappers, const std::string &suffix)
Definition: _separable.cc:43
A base class for image defects.