22#include "pybind11/pybind11.h"
24#include "pybind11/eigen.h"
25#include "pybind11/stl.h"
28#include "ndarray/pybind11.h"
37using namespace pybind11::literals;
52template <
typename PixelT>
53void declareKernelCandidate(lsst::cpputils::python::WrapperCollection &wrappers, std::string
const &suffix) {
55 using PyCandidateSwitch = py::enum_<CandidateSwitch>;
57 std::string
name =
"KernelCandidate" + suffix;
58 using PyKernelCandidate = py::classh<KernelCandidate<PixelT>, afw::math::SpatialCellImageCandidate>;
60 auto clsDef = wrappers.
wrapType(PyKernelCandidate(wrappers.
module,
name.c_str()), [](
auto &mod,
auto &cls) {
61 cls.def(py::init<float const, float const, std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
62 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &>(),
63 "xCenter"_a,
"yCenter"_a,
"templateMaskedImage"_a,
"scienceMaskedImage"_a,
"ps"_a);
64 cls.def(py::init<std::shared_ptr<afw::table::SourceRecord> const &,
65 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
66 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &>(),
67 "source"_a,
"templateMaskedImage"_a,
"scienceMaskedImage"_a,
"ps"_a);
69 cls.def(
"getCandidateRating", &KernelCandidate<PixelT>::getCandidateRating);
70 cls.def(
"getSource", &KernelCandidate<PixelT>::getSource);
71 cls.def(
"getTemplateMaskedImage", &KernelCandidate<PixelT>::getTemplateMaskedImage);
72 cls.def(
"getScienceMaskedImage", &KernelCandidate<PixelT>::getScienceMaskedImage);
73 cls.def(
"getKernel", &KernelCandidate<PixelT>::getKernel,
"cand"_a);
74 cls.def(
"getBackground", &KernelCandidate<PixelT>::getBackground,
"cand"_a);
75 cls.def(
"getKsum", &KernelCandidate<PixelT>::getKsum,
"cand"_a);
76 cls.def(
"getKernelImage", &KernelCandidate<PixelT>::getKernelImage,
"cand"_a);
77 cls.def(
"getImage", &KernelCandidate<PixelT>::getImage);
78 cls.def(
"getKernelSolution", &KernelCandidate<PixelT>::getKernelSolution,
"cand"_a);
79 cls.def(
"getDifferenceImage",
80 (afw::image::MaskedImage<PixelT> (KernelCandidate<PixelT>::*)(CandidateSwitch)) &
81 KernelCandidate<PixelT>::getDifferenceImage,
83 cls.def(
"getDifferenceImage", (afw::image::MaskedImage<PixelT> (KernelCandidate<PixelT>::*)(
84 std::shared_ptr<afw::math::Kernel>, double)) &
85 KernelCandidate<PixelT>::getDifferenceImage,
86 "kernel"_a,
"background"_a);
87 cls.def(
"isInitialized", &KernelCandidate<PixelT>::isInitialized);
88 cls.def(
"build", (void (KernelCandidate<PixelT>::*)(afw::math::KernelList const &)) &
89 KernelCandidate<PixelT>::build,
92 (void (KernelCandidate<PixelT>::*)(afw::math::KernelList const &, Eigen::MatrixXd const &)) &
93 KernelCandidate<PixelT>::build,
94 "basisList"_a,
"hMat"_a);
95 mod.def(
"makeKernelCandidate",
96 (std::shared_ptr<KernelCandidate<PixelT>>(*)(
97 float const, float const, std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
98 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &)) &
100 "xCenter"_a,
"yCenter"_a,
"templateMaskedImage"_a,
"scienceMaskedImage"_a,
"ps"_a);
101 mod.def(
"makeKernelCandidate",
102 (std::shared_ptr<KernelCandidate<PixelT>>(*)(
103 std::shared_ptr<afw::table::SourceRecord> const &,
104 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &,
105 std::shared_ptr<afw::image::MaskedImage<PixelT>> const &, daf::base::PropertySet const &)) &
107 "source"_a,
"templateMaskedImage"_a,
"scienceMaskedImage"_a,
"ps"_a);
109 wrappers.
wrapType(PyCandidateSwitch(clsDef,
"CandidateSwitch"), [](
auto &mod,
auto &enm) {
110 enm.value(
"ORIG", CandidateSwitch::ORIG);
111 enm.value(
"PCA", CandidateSwitch::PCA);
112 enm.value(
"RECENT", CandidateSwitch::RECENT);
120 declareKernelCandidate<float>(wrappers,
"F");
Class used by SpatialModelCell for spatial Kernel fitting.
A helper class for subdividing pybind11 module across multiple translation units (i....
PyType wrapType(PyType cls, ClassWrapperCallback function, bool setModuleName=true)
Add a type (class or enum) wrapper, deferring method and other attribute definitions until finish() i...
pybind11::module module
The module object passed to the PYBIND11_MODULE block that contains this WrapperCollection.
void wrapKernelCandidate(WrapperCollection &wrappers)