LSSTApplications  18.1.0
LSSTDataManagementBasePackage
fitter.cc
Go to the documentation of this file.
1 // -*- LSST-C++ -*-
2 /*
3  * This file is part of jointcal.
4  *
5  * Developed for the LSST Data Management System.
6  * This product includes software developed by the LSST Project
7  * (https://www.lsst.org).
8  * See the COPYRIGHT file at the top-level directory of this distribution
9  * for details of code ownership.
10  *
11  * This program is free software: you can redistribute it and/or modify
12  * it under the terms of the GNU General Public License as published by
13  * the Free Software Foundation, either version 3 of the License, or
14  * (at your option) any later version.
15  *
16  * This program is distributed in the hope that it will be useful,
17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19  * GNU General Public License for more details.
20  *
21  * You should have received a copy of the GNU General Public License
22  * along with this program. If not, see <https://www.gnu.org/licenses/>.
23  */
24 
25 #include "pybind11/pybind11.h"
26 
30 #include "lsst/jointcal/Chi2.h"
34 
35 namespace py = pybind11;
36 using namespace pybind11::literals;
37 
38 namespace lsst {
39 namespace jointcal {
40 namespace {
41 
42 void declareFitterBase(py::module &mod) {
43  py::class_<FitterBase, std::shared_ptr<FitterBase>> cls(mod, "FitterBase");
44 
45  cls.def("minimize", &FitterBase::minimize, "whatToFit"_a, "nSigRejCut"_a = 0, "doRankUpdate"_a = true,
46  "doLineSearch"_a = false, "dumpMatrixFile"_a = "");
47  cls.def("computeChi2", &FitterBase::computeChi2);
48  cls.def("saveChi2Contributions", &FitterBase::saveChi2Contributions);
49 }
50 
51 void declareAstrometryFit(py::module &mod) {
52  py::class_<AstrometryFit, std::shared_ptr<AstrometryFit>, FitterBase> cls(mod, "AstrometryFit");
53 
55  "associations"_a, "astrometryModel"_a, "posError"_a);
56 
57  cls.def("getModel", &AstrometryFit::getModel, py::return_value_policy::reference_internal);
58 }
59 
60 void declarePhotometryFit(py::module &mod) {
61  py::class_<PhotometryFit, std::shared_ptr<PhotometryFit>, FitterBase> cls(mod, "PhotometryFit");
62 
64  "photometryModel"_a);
65 
66  cls.def("getModel", &PhotometryFit::getModel, py::return_value_policy::reference_internal);
67 }
68 
69 PYBIND11_MODULE(fitter, mod) {
70  py::module::import("lsst.jointcal.associations");
71  py::module::import("lsst.jointcal.astrometryModels");
72  py::module::import("lsst.jointcal.chi2");
73  py::module::import("lsst.jointcal.photometryModels");
74  py::enum_<MinimizeResult>(mod, "MinimizeResult")
75  .value("Converged", MinimizeResult::Converged)
76  .value("Chi2Increased", MinimizeResult::Chi2Increased)
77  .value("NonFinite", MinimizeResult::NonFinite)
78  .value("Failed", MinimizeResult::Failed);
79 
80  declareFitterBase(mod);
81  declareAstrometryFit(mod);
82  declarePhotometryFit(mod);
83 }
84 } // namespace
85 } // namespace jointcal
86 } // namespace lsst
def init()
Definition: tests.py:75
PYBIND11_MODULE(camera, mod)
Definition: camera.cc:34
FitResults minimize(lsst::afw::math::Function1< ReturnT > const &function, std::vector< double > const &initialParameterList, std::vector< double > const &stepSizeList, std::vector< double > const &measurementList, std::vector< double > const &varianceList, std::vector< double > const &xPositionList, double errorDef)
Find the minimum of a function(x)
Definition: minimize.cc:177
A base class for image defects.
double computeChi2(const StarMatchList &L, const AstrometryTransform &transform)
the actual chi2
Definition: StarMatch.cc:246