LSSTApplications  20.0.0
LSSTDataManagementBasePackage
sdssCentroid.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 
25 #include <memory>
26 
27 #include "lsst/pex/config/python.h"
28 #include "lsst/meas/base/python.h"
29 
31 
32 namespace py = pybind11;
33 using namespace pybind11::literals;
34 
35 namespace lsst {
36 namespace meas {
37 namespace base {
38 
39 namespace {
40 
41 using PyCentroidAlgorithm =
42  py::class_<SdssCentroidAlgorithm, std::shared_ptr<SdssCentroidAlgorithm>, SimpleAlgorithm>;
43 using PyCentroidControl = py::class_<SdssCentroidControl>;
44 using PyCentroidTransform =
45  py::class_<SdssCentroidTransform, std::shared_ptr<SdssCentroidTransform>, BaseTransform>;
46 
47 PyCentroidControl declareCentroidControl(py::module &mod) {
48  PyCentroidControl cls(mod, "SdssCentroidControl");
49 
50  LSST_DECLARE_CONTROL_FIELD(cls, SdssCentroidControl, binmax);
51  LSST_DECLARE_CONTROL_FIELD(cls, SdssCentroidControl, peakMin);
52  LSST_DECLARE_CONTROL_FIELD(cls, SdssCentroidControl, wfac);
53  LSST_DECLARE_CONTROL_FIELD(cls, SdssCentroidControl, doFootprintCheck);
54  LSST_DECLARE_CONTROL_FIELD(cls, SdssCentroidControl, maxDistToPeak);
55 
56  cls.def(py::init<>());
57 
58  return cls;
59 }
60 
61 PyCentroidAlgorithm declareCentroidAlgorithm(py::module &mod) {
62  PyCentroidAlgorithm cls(mod, "SdssCentroidAlgorithm");
63 
64  cls.attr("FAILURE") = py::cast(SdssCentroidAlgorithm::FAILURE);
65  cls.attr("EDGE") = py::cast(SdssCentroidAlgorithm::EDGE);
66  cls.attr("NO_SECOND_DERIVATIVE") = py::cast(SdssCentroidAlgorithm::NO_SECOND_DERIVATIVE);
67  cls.attr("ALMOST_NO_SECOND_DERIVATIVE") = py::cast(SdssCentroidAlgorithm::ALMOST_NO_SECOND_DERIVATIVE);
68  cls.attr("NOT_AT_MAXIMUM") = py::cast(SdssCentroidAlgorithm::NOT_AT_MAXIMUM);
69 
70  cls.def(py::init<SdssCentroidAlgorithm::Control const &, std::string const &, afw::table::Schema &>(),
71  "ctrl"_a, "name"_a, "schema"_a);
72 
73  cls.def("measure", &SdssCentroidAlgorithm::measure, "measRecord"_a, "exposure"_a);
74  cls.def("fail", &SdssCentroidAlgorithm::fail, "measRecord"_a, "error"_a = nullptr);
75 
76  return cls;
77 }
78 
79 PyCentroidTransform declareCentroidTransform(py::module &mod) {
80  PyCentroidTransform cls(mod, "SdssCentroidTransform");
81 
83  afw::table::SchemaMapper &>(),
84  "ctrl"_a, "name"_a, "mapper"_a);
85 
86  return cls;
87 }
88 
89 } // namespace
90 
91 PYBIND11_MODULE(sdssCentroid, mod) {
92  py::module::import("lsst.afw.table");
93  py::module::import("lsst.meas.base.algorithm");
94  py::module::import("lsst.meas.base.flagHandler");
95  py::module::import("lsst.meas.base.transform");
96 
97  auto clsCentroidControl = declareCentroidControl(mod);
98  auto clsCentroidAlgorithm = declareCentroidAlgorithm(mod);
99  auto clsCentroidTransform = declareCentroidTransform(mod);
100 
101  clsCentroidAlgorithm.attr("Control") = clsCentroidControl;
102  clsCentroidTransform.attr("Control") = clsCentroidControl;
103 
104  python::declareAlgorithm<SdssCentroidAlgorithm, SdssCentroidControl, SdssCentroidTransform>(
105  clsCentroidAlgorithm, clsCentroidControl, clsCentroidTransform);
106 }
107 
108 } // namespace base
109 } // namespace meas
110 } // namespace lsst
std::string
STL class.
base
Definition: __init__.py:1
lsst::meas::base::PYBIND11_MODULE
PYBIND11_MODULE(sdssCentroid, mod)
Definition: sdssCentroid.cc:91
lsst::meas::base.plugins.BaseTransform
BaseTransform
Definition: plugins.py:98
SdssCentroid.h
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
python.h
LSST_DECLARE_CONTROL_FIELD
#define LSST_DECLARE_CONTROL_FIELD(WRAPPER, CLASS, NAME)
Macro used to wrap fields declared by LSST_CONTROL_FIELD using Pybind11.
Definition: python.h:50
pybind11
Definition: _GenericMap.cc:40
python.h
lsst::utils.tests.init
def init()
Definition: tests.py:58
lsst::ip::isr.fringe.measure
def measure(mi, x, y, size, statistic, stats)
Definition: fringe.py:506
lsst::ip::diffim.Control
Control
Definition: __init__.py:49
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42