lsst.meas.base  14.0
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Macros Groups Pages
apertureFlux.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  * Copyright 2008-2017 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/stl.h"
25 
26 #include <memory>
27 
28 #include "lsst/pex/config/python.h"
29 #include "lsst/meas/base/python.h"
30 
33 
34 namespace py = pybind11;
35 using namespace pybind11::literals;
36 
37 namespace lsst {
38 namespace meas {
39 namespace base {
40 
41 namespace {
42 
43 using PyFluxAlgorithm =
44  py::class_<ApertureFluxAlgorithm, std::shared_ptr<ApertureFluxAlgorithm>, SimpleAlgorithm>;
45 using PyFluxControl = py::class_<ApertureFluxControl>;
46 using PyFluxResult = py::class_<ApertureFluxResult, std::shared_ptr<ApertureFluxResult>, FluxResult>;
47 using PyFluxTransform =
48  py::class_<ApertureFluxTransform, std::shared_ptr<ApertureFluxTransform>, BaseTransform>;
49 
50 PyFluxControl declareFluxControl(py::module &mod) {
51  PyFluxControl cls(mod, "ApertureFluxControl");
52 
53  LSST_DECLARE_CONTROL_FIELD(cls, ApertureFluxControl, radii);
54  LSST_DECLARE_CONTROL_FIELD(cls, ApertureFluxControl, maxSincRadius);
55  LSST_DECLARE_CONTROL_FIELD(cls, ApertureFluxControl, shiftKernel);
56 
57  cls.def(py::init<>());
58 
59  return cls;
60 }
61 
62 template <typename Image, class PyClass>
63 void declareComputeFluxes(PyClass &cls) {
64  namespace afwEllipses = lsst::afw::geom::ellipses;
65  using Control = ApertureFluxAlgorithm::Control;
66  using Result = ApertureFluxAlgorithm::Result;
67 
68  cls.def_static("computeSincFlux",
69  (Result(*)(Image const &, afwEllipses::Ellipse const &, Control const &)) &
70  ApertureFluxAlgorithm::computeSincFlux,
71  "image"_a, "ellipse"_a, "ctrl"_a = Control());
72  cls.def_static("computeNaiveFlux",
73  (Result(*)(Image const &, afwEllipses::Ellipse const &, Control const &)) &
74  ApertureFluxAlgorithm::computeNaiveFlux,
75  "image"_a, "ellipse"_a, "ctrl"_a = Control());
76  cls.def_static("computeFlux", (Result(*)(Image const &, afwEllipses::Ellipse const &, Control const &)) &
77  ApertureFluxAlgorithm::computeFlux,
78  "image"_a, "ellipse"_a, "ctrl"_a = Control());
79 }
80 
81 PyFluxAlgorithm declareFluxAlgorithm(py::module &mod) {
82  PyFluxAlgorithm cls(mod, "ApertureFluxAlgorithm");
83 
84  cls.attr("FAILURE") = py::cast(ApertureFluxAlgorithm::FAILURE);
85  cls.attr("APERTURE_TRUNCATED") = py::cast(ApertureFluxAlgorithm::APERTURE_TRUNCATED);
86  cls.attr("SINC_COEFFS_TRUNCATED") = py::cast(ApertureFluxAlgorithm::SINC_COEFFS_TRUNCATED);
87 
88  // constructor not wrapped because class is abstract
89 
90  declareComputeFluxes<lsst::afw::image::Image<double>>(cls);
91  declareComputeFluxes<lsst::afw::image::MaskedImage<double>>(cls);
92  declareComputeFluxes<lsst::afw::image::Image<float>>(cls);
93  declareComputeFluxes<lsst::afw::image::MaskedImage<float>>(cls);
94 
95  cls.def("measure", &ApertureFluxAlgorithm::measure, "measRecord"_a, "exposure"_a);
96  cls.def("fail", &ApertureFluxAlgorithm::fail, "measRecord"_a, "error"_a = nullptr);
97  cls.def_static("makeFieldPrefix", &ApertureFluxAlgorithm::makeFieldPrefix, "name"_a, "radius"_a);
98 
99  return cls;
100 }
101 
102 void declareFluxResult(py::module &mod) {
103  PyFluxResult cls(mod, "ApertureFluxResult");
104 
105  cls.def("getFlag", (bool (ApertureFluxResult::*)(unsigned int) const) & ApertureFluxResult::getFlag,
106  "bit"_a);
107  cls.def("getFlag",
108  (bool (ApertureFluxResult::*)(std::string const &name) const) & ApertureFluxResult::getFlag,
109  "name"_a);
110  cls.def("setFlag", &ApertureFluxResult::setFlag, "index"_a, "value"_a);
111  cls.def("unsetFlag", &ApertureFluxResult::unsetFlag, "index"_a);
112 }
113 
114 PyFluxTransform declareFluxTransform(py::module &mod) {
115  PyFluxTransform cls(mod, "ApertureFluxTransform");
116 
117  cls.def(py::init<ApertureFluxTransform::Control const &, std::string const &,
118  afw::table::SchemaMapper &>(),
119  "ctrl"_a, "name"_a, "mapper"_a);
120 
121  cls.def("__call__", &ApertureFluxTransform::operator(), "inputCatalog"_a, "outputCatalog"_a, "wcs"_a,
122  "calib"_a);
123 
124  return cls;
125 }
126 
127 } // <anonymous>
128 
129 PYBIND11_PLUGIN(apertureFlux) {
130  py::module::import("lsst.afw.geom");
131  py::module::import("lsst.afw.image");
132  py::module::import("lsst.afw.table");
133  py::module::import("lsst.meas.base.algorithm");
134  py::module::import("lsst.meas.base.flagHandler");
135  py::module::import("lsst.meas.base.fluxUtilities");
136  py::module::import("lsst.meas.base.transform");
137 
138  py::module mod("apertureFlux");
139 
140  auto clsFluxControl = declareFluxControl(mod);
141  auto clsFluxAlgorithm = declareFluxAlgorithm(mod);
142  declareFluxResult(mod);
143  auto clsFluxTransform = declareFluxTransform(mod);
144 
145  clsFluxAlgorithm.attr("Control") = clsFluxControl;
146  // no need to make ApertureFluxControl::Result visible to Python
147  clsFluxTransform.attr("Control") = clsFluxControl;
148 
149  python::declareAlgorithm<ApertureFluxAlgorithm, ApertureFluxControl, ApertureFluxTransform>(
150  clsFluxAlgorithm, clsFluxControl, clsFluxTransform);
151 
152  return mod.ptr();
153 }
154 
155 } // base
156 } // meas
157 } // lsst
PYBIND11_PLUGIN(algorithm)
Definition: algorithm.cc:37