lsst.meas.astrom  13.0-14-g9415442+44
 All Classes Namespaces Files Functions Variables Typedefs Friends Macros Groups Pages
leastSqFitter1d.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 
25 #include <string>
26 
27 #include "numpy/arrayobject.h"
28 #include "ndarray/pybind11.h"
29 
31 
32 namespace py = pybind11;
33 using namespace pybind11::literals;
34 
35 namespace lsst {
36 namespace meas {
37 namespace astrom {
38 namespace sip {
39 namespace {
40 
41 template <typename FittingFunc>
42 static void declareLeastSqFitter1d(py::module &mod, std::string const &name) {
43  py::class_<LeastSqFitter1d<FittingFunc>, std::shared_ptr<LeastSqFitter1d<FittingFunc>>> cls(mod,
44  name.c_str());
45 
46  cls.def(py::init<std::vector<double> const &, std::vector<double> const &, std::vector<double> const &,
47  int>(),
48  "x"_a, "y"_a, "s"_a, "order"_a);
49 
50  cls.def("getParams", &LeastSqFitter1d<FittingFunc>::getParams);
51  cls.def("getErrors", &LeastSqFitter1d<FittingFunc>::getErrors);
52  cls.def("getBestFitFunction", &LeastSqFitter1d<FittingFunc>::getBestFitFunction);
53  cls.def("valueAt", &LeastSqFitter1d<FittingFunc>::valueAt, "x"_a);
54  cls.def("residuals", &LeastSqFitter1d<FittingFunc>::residuals);
55  cls.def("getChiSq", &LeastSqFitter1d<FittingFunc>::getChiSq);
56  cls.def("getReducedChiSq", &LeastSqFitter1d<FittingFunc>::getReducedChiSq);
57 }
58 
59 } // namespace lsst::meas::astrom::sip::<anonymous>
60 
61 PYBIND11_PLUGIN(leastSqFitter1d) {
62  py::module mod("leastSqFitter1d");
63 
64  // Need to import numpy for ndarray and eigen conversions
65  if (_import_array() < 0) {
66  PyErr_SetString(PyExc_ImportError, "numpy.core.multiarray failed to import");
67  return nullptr;
68  }
69 
70  declareLeastSqFitter1d<afw::math::PolynomialFunction1<double>>(mod, "LeastSqFitter1dPoly");
71 
72  return mod.ptr();
73 }
74 }
75 }
76 }
77 } // namespace lsst::meas::astrom::sip
PYBIND11_PLUGIN(makeMatchStatistics)