LSST Applications g013ef56533+63812263fb,g083dd6704c+a047e97985,g199a45376c+0ba108daf9,g1fd858c14a+fde7a7a78c,g210f2d0738+db0c280453,g262e1987ae+abed931625,g29ae962dfc+058d1915d8,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+64337f1634,g47891489e3+f459a6810c,g53246c7159+8c5ae1fdc5,g54cd7ddccb+890c8e1e5d,g5a60e81ecd+d9e514a434,g64539dfbff+db0c280453,g67b6fd64d1+f459a6810c,g6ebf1fc0d4+8c5ae1fdc5,g7382096ae9+36d16ea71a,g74acd417e5+c70e70fbf6,g786e29fd12+668abc6043,g87389fa792+8856018cbb,g89139ef638+f459a6810c,g8d7436a09f+1b779678e3,g8ea07a8fe4+81eaaadc04,g90f42f885a+34c0557caf,g97be763408+9583a964dd,g98a1a72a9c+028271c396,g98df359435+530b675b85,gb8cb2b794d+4e54f68785,gbf99507273+8c5ae1fdc5,gc2a301910b+db0c280453,gca7fc764a6+f459a6810c,gd7ef33dd92+f459a6810c,gdab6d2f7ff+c70e70fbf6,ge410e46f29+f459a6810c,ge41e95a9f2+db0c280453,geaed405ab2+e3b4b2a692,gf9a733ac38+8c5ae1fdc5,w.2025.43
LSST Data Management Base Package
Loading...
Searching...
No Matches
_leastSquares.cc
Go to the documentation of this file.
1/*
2 * LSST Data Management System
3 * Copyright 2008-2016 AURA/LSST.
4 *
5 * This product includes software developed by the
6 * LSST Project (http://www.lsst.org/).
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>
25
26#include "ndarray/pybind11.h"
27
29
30namespace py = pybind11;
31using namespace pybind11::literals;
32
33using namespace lsst::afw::math;
34namespace lsst {
35namespace afw {
36namespace math {
37namespace {
38template <typename T1, typename T2, int C1, int C2>
39void declareLeastSquares(lsst::cpputils::python::WrapperCollection &wrappers) {
40 auto clsLeastSquares = wrappers.wrapType(
41 py::class_<LeastSquares>(wrappers.module, "LeastSquares"), [](auto &mod, auto &cls) {
42 cls.def_static(
43 "fromDesignMatrix",
44 (LeastSquares(*)(ndarray::Array<T1, 2, C1> const &, ndarray::Array<T2, 1, C2> const &,
45 LeastSquares::Factorization)) &
46 LeastSquares::fromDesignMatrix<T1, T2, C1, C2>,
47 "design"_a, "data"_a, "factorization"_a = LeastSquares::NORMAL_EIGENSYSTEM);
48 cls.def_static(
49 "fromNormalEquations",
50 (LeastSquares(*)(ndarray::Array<T1, 2, C1> const &, ndarray::Array<T2, 1, C2> const &,
51 LeastSquares::Factorization)) &
52 LeastSquares::fromNormalEquations<T1, T2, C1, C2>,
53 "fisher"_a, "rhs"_a, "factorization"_a = LeastSquares::NORMAL_EIGENSYSTEM);
54 cls.def("getRank", &LeastSquares::getRank);
55 cls.def("setDesignMatrix", (void (LeastSquares::*)(ndarray::Array<T1, 2, C1> const &,
56 ndarray::Array<T2, 1, C2> const &)) &
57 LeastSquares::setDesignMatrix<T1, T2, C1, C2>);
58 cls.def("getDimension", &LeastSquares::getDimension);
59 cls.def("setNormalEquations", (void (LeastSquares::*)(ndarray::Array<T1, 2, C1> const &,
60 ndarray::Array<T2, 1, C2> const &)) &
61 LeastSquares::setNormalEquations<T1, T2, C1, C2>);
62 cls.def("getSolution", &LeastSquares::getSolution);
63 cls.def("getFisherMatrix", &LeastSquares::getFisherMatrix);
64 cls.def("getCovariance", &LeastSquares::getCovariance);
65 cls.def("getFactorization", &LeastSquares::getFactorization);
66 cls.def("getDiagnostic", &LeastSquares::getDiagnostic);
67 cls.def("getThreshold", &LeastSquares::getThreshold);
68 cls.def("setThreshold", &LeastSquares::setThreshold);
69 });
70 wrappers.wrapType(py::enum_<LeastSquares::Factorization>(clsLeastSquares, "Factorization"),
71 [](auto &mod, auto &enm) {
72 enm.value("NORMAL_EIGENSYSTEM", LeastSquares::Factorization::NORMAL_EIGENSYSTEM);
73 enm.value("NORMAL_CHOLESKY", LeastSquares::Factorization::NORMAL_CHOLESKY);
74 enm.value("DIRECT_SVD", LeastSquares::Factorization::DIRECT_SVD);
75 enm.export_values();
76 });
77};
78} // namespace
79
81 declareLeastSquares<double, double, 0, 0>(wrappers);
82}
83} // namespace math
84} // namespace afw
85} // namespace lsst
@ NORMAL_EIGENSYSTEM
Use the normal equations with a symmetric Eigensystem decomposition.
@ NORMAL_CHOLESKY
Use the normal equations with a Cholesky decomposition.
@ DIRECT_SVD
Use a thin singular value decomposition of the design matrix.
A helper class for subdividing pybind11 module across multiple translation units (i....
Definition python.h:242
PyType wrapType(PyType cls, ClassWrapperCallback function, bool setModuleName=true)
Add a type (class or enum) wrapper, deferring method and other attribute definitions until finish() i...
Definition python.h:391
pybind11::module module
The module object passed to the PYBIND11_MODULE block that contains this WrapperCollection.
Definition python.h:448
void wrapLeastSquares(lsst::cpputils::python::WrapperCollection &wrappers)