LSSTApplications  20.0.0
LSSTDataManagementBasePackage
psfFlux.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 
31 #include "lsst/meas/base/PsfFlux.h"
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 = py::class_<PsfFluxAlgorithm, std::shared_ptr<PsfFluxAlgorithm>, SimpleAlgorithm>;
44 using PyFluxControl = py::class_<PsfFluxControl>;
45 using PyFluxTransform = py::class_<PsfFluxTransform, std::shared_ptr<PsfFluxTransform>, BaseTransform>;
46 
47 PyFluxControl declareFluxControl(py::module &mod) {
48  PyFluxControl cls(mod, "PsfFluxControl");
49 
50  LSST_DECLARE_CONTROL_FIELD(cls, PsfFluxControl, badMaskPlanes);
51 
52  cls.def(py::init<>());
53 
54  return cls;
55 }
56 
57 PyFluxAlgorithm declareFluxAlgorithm(py::module &mod) {
58  PyFluxAlgorithm cls(mod, "PsfFluxAlgorithm");
59 
60  cls.attr("FAILURE") = py::cast(PsfFluxAlgorithm::FAILURE);
61  cls.attr("NO_GOOD_PIXELS") = py::cast(PsfFluxAlgorithm::NO_GOOD_PIXELS);
62  cls.attr("EDGE") = py::cast(PsfFluxAlgorithm::EDGE);
63 
64  cls.def(py::init<PsfFluxAlgorithm::Control const &, std::string const &, afw::table::Schema &>(),
65  "ctrl"_a, "name"_a, "schema"_a);
66 
67  cls.def(py::init<PsfFluxAlgorithm::Control const &, std::string const &, afw::table::Schema &,
68  std::string const &>(),
69  "ctrl"_a, "name"_a, "schema"_a, "logName"_a);
70  return cls;
71 }
72 
73 PyFluxTransform declareFluxTransform(py::module &mod) {
74  PyFluxTransform cls(mod, "PsfFluxTransform");
75 
76  cls.def(py::init<PsfFluxTransform::Control const &, std::string const &, afw::table::SchemaMapper &>(),
77  "ctrl"_a, "name"_a, "mapper"_a);
78 
79  return cls;
80 }
81 
82 } // namespace
83 
84 PYBIND11_MODULE(psfFlux, mod) {
85  py::module::import("lsst.afw.table");
86  py::module::import("lsst.meas.base.algorithm");
87  py::module::import("lsst.meas.base.flagHandler");
88  py::module::import("lsst.meas.base.fluxUtilities");
89  py::module::import("lsst.meas.base.transform");
90 
91  auto clsFluxControl = declareFluxControl(mod);
92  auto clsFluxAlgorithm = declareFluxAlgorithm(mod);
93  auto clsFluxTransform = declareFluxTransform(mod);
94 
95  clsFluxAlgorithm.attr("Control") = clsFluxControl;
96  clsFluxTransform.attr("Control") = clsFluxControl;
97 
98  python::declareAlgorithm<PsfFluxAlgorithm, PsfFluxControl, PsfFluxTransform>(
99  clsFluxAlgorithm, clsFluxControl, clsFluxTransform);
100 }
101 
102 } // namespace base
103 } // namespace meas
104 } // namespace lsst
std::string
STL class.
base
Definition: __init__.py:1
lsst::meas::base.plugins.BaseTransform
BaseTransform
Definition: plugins.py:98
FluxUtilities.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
PsfFlux.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::diffim.Control
Control
Definition: __init__.py:49
lsst::meas::modelfit.psf.psfContinued.module
module
Definition: psfContinued.py:42
lsst::meas::base::PYBIND11_MODULE
PYBIND11_MODULE(psfFlux, mod)
Definition: psfFlux.cc:84