LSSTApplications  21.0.0+75b29a8a7f,21.0.0+e70536a077,21.0.0-1-ga51b5d4+62c747d40b,21.0.0-11-ga6ea59e8e+47cba9fc36,21.0.0-2-g103fe59+914993bf7c,21.0.0-2-g1367e85+e2614ded12,21.0.0-2-g45278ab+e70536a077,21.0.0-2-g4bc9b9f+7b2b5f8678,21.0.0-2-g5242d73+e2614ded12,21.0.0-2-g54e2caa+6403186824,21.0.0-2-g7f82c8f+3ac4acbffc,21.0.0-2-g8dde007+04a6aea1af,21.0.0-2-g8f08a60+9402881886,21.0.0-2-ga326454+3ac4acbffc,21.0.0-2-ga63a54e+81dd751046,21.0.0-2-gc738bc1+5f65c6e7a9,21.0.0-2-gde069b7+26c92b3210,21.0.0-2-gecfae73+0993ddc9bd,21.0.0-2-gfc62afb+e2614ded12,21.0.0-21-gba890a8+5a4f502a26,21.0.0-23-g9966ff26+03098d1af8,21.0.0-3-g357aad2+8ad216c477,21.0.0-3-g4be5c26+e2614ded12,21.0.0-3-g6d51c4a+4d2fe0280d,21.0.0-3-g7d9da8d+75b29a8a7f,21.0.0-3-gaa929c8+522e0f12c2,21.0.0-3-ge02ed75+4d2fe0280d,21.0.0-4-g3300ddd+e70536a077,21.0.0-4-gc004bbf+eac6615e82,21.0.0-4-gccdca77+f94adcd104,21.0.0-4-gd1c1571+18b81799f9,21.0.0-5-g7b47fff+4d2fe0280d,21.0.0-5-gb155db7+d2632f662b,21.0.0-5-gdf36809+637e4641ee,21.0.0-6-g722ad07+28c848f42a,21.0.0-7-g959bb79+522e0f12c2,21.0.0-7-gfd72ab2+cf01990774,21.0.0-9-g87fb7b8d+e2ab11cdd6,w.2021.04
LSSTDataManagementBasePackage
separable.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 
33 
34 namespace py = pybind11;
35 
36 using namespace py::literals;
37 
38 using namespace lsst::afw::geom::ellipses;
39 
40 template <typename Ellipticity_, typename Radius_>
41 void declareSeparable(py::module &mod, const std::string &suffix) {
43 
44  py::class_<Class, std::shared_ptr<Class>, BaseCore> cls(mod, ("Separable" + suffix).c_str());
45 
46  // py::enum_<typename Class::ParameterEnum>(mod, "ParameterEnum")
47  // .value("E0", Class::ParameterEnum::E0)
48  // .value("E1", Class::ParameterEnum::E1)
49  // .value("RADIUS", Class::ParameterEnum::RADIUS)
50  // .export_values();
51 
52  cls.def(py::init<double, double, double, bool>(), "e1"_a = 0.0, "e2"_a = 0.0, "radius"_a = Radius_(),
53  "normalize"_a = true);
54  // cls.def(py::init<std::complex<double> const &, double, bool>(),
55  // "complex"_a, "radius"_a=Radius_(), "normalize"_a=true);
56  // cls.def(py::init<Ellipticity_ const &, double, bool>(),
57  // "ellipticity"_a, "radius"_a=Radius_(), "normalize"_a=true);
58  // cls.def(py::init<BaseCore::ParameterVector const &, bool>(),
59  // "vector"_a, "normalize"_a=true);
60  cls.def(py::init<Class const &>());
61  cls.def(py::init<BaseCore const &>());
62 
63  cls.def("getE1", &Class::getE1);
64  cls.def("setE1", &Class::setE1);
65  cls.def("getE2", &Class::getE2);
66  cls.def("setE2", &Class::setE2);
67  cls.def("getRadius", (Radius_ const &(Class::*)() const) & Class::getRadius);
68  cls.def("setRadius", (void (Class::*)(double)) & Class::setRadius);
69  cls.def("setRadius", (void (Class::*)(Radius_ const &)) & Class::setRadius);
70  cls.def("getEllipticity", (Ellipticity_ const &(Class::*)() const) & Class::getEllipticity);
71  cls.def("clone", &Class::clone);
72  cls.def("getName", &Class::getName);
73  cls.def("normalize", &Class::normalize);
74  cls.def("assign", [](Class &self, Class &other) { self = other; });
75  cls.def("assign", [](Class &self, BaseCore &other) { self = other; });
76  cls.def("transform", [](Class &self, lsst::geom::LinearTransform const &t) {
77  return std::static_pointer_cast<Class>(self.transform(t).copy());
78  });
79  cls.def("transformInPlace",
80  [](Class &self, lsst::geom::LinearTransform const &t) { self.transform(t).inPlace(); });
81  cls.def("__str__",
82  [](Class &self) { return py::str("({}, {})").format(self.getEllipticity(), self.getRadius()); });
83  cls.def("__repr__", [](Class &self) {
84  return py::str("Separable({}, {})").format(self.getEllipticity(), self.getRadius());
85  });
86 }
87 
88 PYBIND11_MODULE(separable, mod) {
89  declareSeparable<Distortion, DeterminantRadius>(mod, "DistortionDeterminantRadius");
90  declareSeparable<Distortion, TraceRadius>(mod, "DistortionTraceRadius");
91  declareSeparable<Distortion, LogDeterminantRadius>(mod, "DistortionLogDeterminantRadius");
92  declareSeparable<Distortion, LogTraceRadius>(mod, "DistortionLogTraceRadius");
93 
94  declareSeparable<ConformalShear, DeterminantRadius>(mod, "ConformalShearDeterminantRadius");
95  declareSeparable<ConformalShear, TraceRadius>(mod, "ConformalShearTraceRadius");
96  declareSeparable<ConformalShear, LogDeterminantRadius>(mod, "ConformalShearLogDeterminantRadius");
97  declareSeparable<ConformalShear, LogTraceRadius>(mod, "ConformalShearLogTraceRadius");
98 
99  declareSeparable<ReducedShear, DeterminantRadius>(mod, "ReducedShearDeterminantRadius");
100  declareSeparable<ReducedShear, TraceRadius>(mod, "ReducedShearTraceRadius");
101  declareSeparable<ReducedShear, LogDeterminantRadius>(mod, "ReducedShearLogDeterminantRadius");
102  declareSeparable<ReducedShear, LogTraceRadius>(mod, "ReducedShearLogTraceRadius");
103 }
std::string
STL class.
Distortion.h
lsst::geom::LinearTransform
A 2D linear coordinate transformation.
Definition: LinearTransform.h:69
Separable.h
radii.h
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
ConformalShear.h
lsst::afw::geom::ellipses::BaseCore
A base class for parametrizations of the "core" of an ellipse - the ellipticity and size.
Definition: BaseCore.h:55
lsst::afw::geom::ellipses::Separable
An ellipse core with a complex ellipticity and radius parameterization.
Definition: Separable.h:50
lsst::afw::geom::ellipses
Definition: Axes.h:42
other
ItemVariant const * other
Definition: Schema.cc:56
lsst.pipe.tasks.cli.cmd.commands.str
str
Definition: commands.py:50
declareSeparable
void declareSeparable(py::module &mod, const std::string &suffix)
Definition: separable.cc:41
ReducedShear.h
transform
table::Key< int > transform
Definition: TransformMap.cc:299
pybind11
Definition: _GenericMap.cc:40
PYBIND11_MODULE
PYBIND11_MODULE(separable, mod)
Definition: separable.cc:88
lsst::afw::image.slicing.clone
clone
Definition: slicing.py:257
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42
BaseCore.h