lsst.ip.diffim  13.0-26-g703d095+1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
assessSpatialKernelVisitor.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  *
4  * This product includes software developed by the
5  * LSST Project (http://www.lsst.org/).
6  * See the COPYRIGHT file
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 #include "pybind11/pybind11.h"
23 
24 #include <memory>
25 #include <string>
26 
27 #include "lsst/afw/math/SpatialCell.h"
29 
30 namespace py = pybind11;
31 using namespace pybind11::literals;
32 
33 namespace lsst {
34 namespace ip {
35 namespace diffim {
36 namespace detail {
37 
38 namespace {
39 
47 template <typename PixelT>
48 void declareAssessSpatialKernelVisitor(py::module& mod, std::string const& suffix) {
49  py::class_<AssessSpatialKernelVisitor<PixelT>, std::shared_ptr<AssessSpatialKernelVisitor<PixelT>>,
50  lsst::afw::math::CandidateVisitor>
51  cls(mod, ("AssessSpatialKernelVisitor" + suffix).c_str());
52 
53  cls.def(py::init<std::shared_ptr<afw::math::LinearCombinationKernel>,
54  afw::math::Kernel::SpatialFunctionPtr, pex::policy::Policy const&>(),
55  "spatialKernel"_a, "spatialBackground"_a, "policy"_a);
56 
57  cls.def("reset", &AssessSpatialKernelVisitor<PixelT>::reset);
58  cls.def("getNGood", &AssessSpatialKernelVisitor<PixelT>::getNGood);
59  cls.def("getNRejected", &AssessSpatialKernelVisitor<PixelT>::getNRejected);
60  cls.def("getNProcessed", &AssessSpatialKernelVisitor<PixelT>::getNProcessed);
61  cls.def("processCandidate", &AssessSpatialKernelVisitor<PixelT>::processCandidate, "candidate"_a);
62 
63  mod.def("makeAssessSpatialKernelVisitor", &makeAssessSpatialKernelVisitor<PixelT>, "spatialKernel"_a,
64  "spatialBackground"_a, "policy"_a);
65 }
66 
67 } // namespace lsst::ip::diffim::detail::<anonymous>
68 
69 PYBIND11_PLUGIN(assessSpatialKernelVisitor) {
70  py::module::import("lsst.afw.math");
71  py::module::import("lsst.pex.policy");
72 
73  py::module mod("assessSpatialKernelVisitor");
74 
75  declareAssessSpatialKernelVisitor<float>(mod, "F");
76 
77  return mod.ptr();
78 }
79 
80 } // detail
81 } // diffim
82 } // ip
83 } // lsst
Declaration of AssessSpatialKernelVisitor.
PYBIND11_PLUGIN(basisLists)
Definition: basisLists.cc:39