lsst.jointcal  master-g9041cab851
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
fitter.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 
23 #include "pybind11/pybind11.h"
24 
28 #include "lsst/jointcal/Chi2.h"
32 
33 namespace py = pybind11;
34 using namespace pybind11::literals;
35 
36 namespace lsst {
37 namespace jointcal {
38 namespace {
39 
40 void declareFitterBase(py::module &mod) {
41  py::class_<FitterBase, std::shared_ptr<FitterBase>> cls(mod, "FitterBase");
42 
43  cls.def("minimize", &FitterBase::minimize, "whatToFit"_a, "nSigRejCut"_a = 0);
44  cls.def("computeChi2", &FitterBase::computeChi2);
45  cls.def("saveResultTuples", &FitterBase::saveResultTuples);
46 }
47 
48 void declareAstrometryFit(py::module &mod) {
49  py::class_<AstrometryFit, std::shared_ptr<AstrometryFit>, FitterBase> cls(mod, "AstrometryFit");
50 
51  cls.def(py::init<std::shared_ptr<Associations>, std::shared_ptr<AstrometryModel>, double>(),
52  "associations"_a, "astrometryModel"_a, "posError"_a);
53 }
54 
55 void declarePhotometryFit(py::module &mod) {
56  py::class_<PhotometryFit, std::shared_ptr<PhotometryFit>, FitterBase> cls(mod, "PhotometryFit");
57 
58  cls.def(py::init<std::shared_ptr<Associations>, std::shared_ptr<PhotometryModel>>(), "associations"_a,
59  "photometryModel"_a);
60 }
61 
62 PYBIND11_PLUGIN(fitter) {
63  py::module::import("lsst.jointcal.associations");
64  py::module::import("lsst.jointcal.astrometryModels");
65  py::module::import("lsst.jointcal.chi2");
66  py::module::import("lsst.jointcal.photometryModels");
67  py::module mod("fitter");
68 
69  py::enum_<MinimizeResult>(mod, "MinimizeResult")
70  .value("Converged", MinimizeResult::Converged)
71  .value("Chi2Increased", MinimizeResult::Chi2Increased)
72  .value("Failed", MinimizeResult::Failed);
73 
74  declareFitterBase(mod);
75  declareAstrometryFit(mod);
76  declarePhotometryFit(mod);
77 
78  return mod.ptr();
79 }
80 } // namespace
81 } // namespace jointcal
82 } // namespace lsst
double computeChi2(const StarMatchList &L, const Gtransfo &gtransfo)
the actual chi2
Definition: StarMatch.cc:219