lsst.meas.algorithms  13.0-18-gc4ad4228+1
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
spatialModelPsf.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 #include "pybind11/stl.h"
24 
26 
27 namespace py = pybind11;
28 using namespace pybind11::literals;
29 
30 namespace lsst {
31 namespace meas {
32 namespace algorithms {
33 namespace {
34 
35 template <typename PixelT>
36 static void declareFunctions(py::module &mod) {
37  using MaskedImageT = lsst::afw::image::MaskedImage<PixelT, lsst::afw::image::MaskPixel,
38  lsst::afw::image::VariancePixel>;
39 
40  mod.def("createKernelFromPsfCandidates", createKernelFromPsfCandidates<PixelT>, "psfCells"_a, "dims"_a,
41  "xy0"_a, "nEigenComponents"_a, "spatialOrder"_a, "ksize"_a, "nStarPerCell"_a = -1,
42  "constantWeight"_a = true, "border"_a = 3);
43  mod.def("countPsfCandidates", countPsfCandidates<PixelT>, "psfCells"_a, "nStarPerCell"_a = -1);
44  mod.def("fitSpatialKernelFromPsfCandidates",
45  (std::pair<bool, double>(*)(lsst::afw::math::Kernel *, lsst::afw::math::SpatialCellSet const &,
46  int const, double const,
47  double const))fitSpatialKernelFromPsfCandidates<PixelT>,
48  "kernel"_a, "psfCells"_a, "nStarPerCell"_a = -1, "tolerance"_a = 1e-5, "lambda"_a = 0.0);
49  mod.def("fitSpatialKernelFromPsfCandidates",
50  (std::pair<bool, double>(*)(lsst::afw::math::Kernel *, lsst::afw::math::SpatialCellSet const &,
51  bool const, int const, double const,
52  double const))fitSpatialKernelFromPsfCandidates<PixelT>,
53  "kernel"_a, "psfCells"_a, "doNonLinearFit"_a, "nStarPerCell"_a = -1, "tolerance"_a = 1e-5,
54  "lambda"_a = 0.0);
55  mod.def("subtractPsf", subtractPsf<MaskedImageT>, "psf"_a, "data"_a, "x"_a, "y"_a,
56  "psfFlux"_a = std::numeric_limits<double>::quiet_NaN());
57  mod.def("fitKernelParamsToImage", fitKernelParamsToImage<MaskedImageT>, "kernel"_a, "image"_a, "pos"_a);
58  mod.def("fitKernelToImage", fitKernelToImage<MaskedImageT>, "kernel"_a, "image"_a, "pos"_a);
59 }
60 
61 } // <anonymous>
62 
63 PYBIND11_PLUGIN(spatialModelPsf) {
64  py::module mod("spatialModelPsf");
65 
66  /* Module level */
67  declareFunctions<float>(mod);
68 
69  return mod.ptr();
70 }
71 
72 } // algorithms
73 } // meas
74 } // lsst
Class used by SpatialCell for spatial PSF fittig.
PYBIND11_PLUGIN(binnedWcs)
Definition: binnedWcs.cc:33