LSST Applications g013ef56533+7c9321ec0f,g042eb84c57+c6cfa41bc3,g199a45376c+0ba108daf9,g1fd858c14a+fcad0d0313,g210f2d0738+c0f94c6586,g262e1987ae+a7e710680e,g29ae962dfc+fb55f2edb0,g2ac17093b6+61d6563b1e,g2b1d02342f+df6f932764,g2cef7863aa+aef1011c0b,g2f7ad74990+c0f94c6586,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+53cf87ae69,g47891489e3+4316d04fff,g511e8cfd20+baa56acf6c,g53246c7159+8c5ae1fdc5,g54cd7ddccb+fd7ad03fde,g64539dfbff+c0f94c6586,g67b6fd64d1+4316d04fff,g67fd3c3899+c0f94c6586,g6985122a63+4316d04fff,g74acd417e5+ca833bee28,g786e29fd12+668abc6043,g81db2e9a8d+b2ec8e584f,g87389fa792+8856018cbb,g89139ef638+4316d04fff,g8d7436a09f+0a24083b20,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+11eb8fd5b8,gbf99507273+8c5ae1fdc5,gcdda8b9158+e4c84c9d5c,gce8aa8abaa+8c5ae1fdc5,gd7ef33dd92+4316d04fff,gdab6d2f7ff+ca833bee28,ge410e46f29+4316d04fff,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.40
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)