LSSTApplications  20.0.0
LSSTDataManagementBasePackage
matrix3d.cc
Go to the documentation of this file.
1 /*
2  * LSST Data Management System
3  * See COPYRIGHT file at the top of the source tree.
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 #include "pybind11/pybind11.h"
23 
24 #include <memory>
25 
26 #include "lsst/sphgeom/Matrix3d.h"
28 
29 namespace py = pybind11;
30 using namespace pybind11::literals;
31 
32 namespace lsst {
33 namespace sphgeom {
34 namespace {
35 
36 Vector3d getRow(Matrix3d const &self, py::int_ row) {
37  return self.getRow(static_cast<int>(python::convertIndex(3, row)));
38 }
39 
40 PYBIND11_MODULE(matrix3d, mod) {
41  py::module::import("lsst.sphgeom.vector3d");
42 
43  py::class_<Matrix3d, std::shared_ptr<Matrix3d>> cls(mod, "Matrix3d");
44 
45  cls.def(py::init<>());
46  cls.def(py::init<double, double, double, double, double, double, double,
47  double, double>(),
48  "m00"_a, "m01"_a, "m02"_a, "m10"_a, "m11"_a, "m12"_a, "m20"_a,
49  "m21"_a, "m22"_a);
50  cls.def(py::init<Vector3d const &>(), "diagonal"_a);
51  cls.def(py::init<double>(), "scale"_a);
52  cls.def(py::init<Matrix3d const &>(), "matrix"_a);
53 
54  cls.def("__eq__", &Matrix3d::operator==, py::is_operator());
55  cls.def("__ne__", &Matrix3d::operator!=, py::is_operator());
56 
57  // Add bounds checking to getRow and getColumn
58  cls.def("getRow", &getRow, "row"_a);
59  cls.def("getColumn",
60  [](Matrix3d const &self, py::int_ col) {
61  return self.getColumn(
62  static_cast<int>(python::convertIndex(3, col)));
63  },
64  "col"_a);
65 
66  cls.def("__len__", [](Matrix3d const &self) { return py::int_(3); });
67  cls.def("__getitem__", &getRow, py::is_operator());
68  cls.def("__getitem__",
69  [](Matrix3d const &self, py::tuple t) {
70  if (t.size() > 2) {
71  throw py::index_error("Too many indexes for Matrix3d");
72  } else if (t.size() == 0) {
73  return py::cast(self);
74  } else if (t.size() == 1) {
75  return py::cast(getRow(self, t[0].cast<py::int_>()));
76  }
77  return py::cast(
78  self(python::convertIndex(3, t[0].cast<py::int_>()),
79  python::convertIndex(3, t[1].cast<py::int_>())));
80  },
81  py::is_operator());
82 
83  cls.def("inner", &Matrix3d::inner, "matrix"_a);
84  cls.def("getSquaredNorm", &Matrix3d::getSquaredNorm);
85  cls.def("getNorm", &Matrix3d::getNorm);
86 
87  cls.def("__mul__",
88  (Vector3d(Matrix3d::*)(Vector3d const &) const) &
89  Matrix3d::operator*,
90  "vector"_a, py::is_operator());
91  cls.def("__mul__",
92  (Matrix3d(Matrix3d::*)(Matrix3d const &) const) &
93  Matrix3d::operator*,
94  "matrix"_a, py::is_operator());
95  cls.def("__add__", &Matrix3d::operator+, py::is_operator());
96  cls.def("__sub__", &Matrix3d::operator-, py::is_operator());
97 
98  cls.def("cwiseProduct", &Matrix3d::cwiseProduct);
99  cls.def("transpose", &Matrix3d::transpose);
100  cls.def("inverse", &Matrix3d::inverse);
101 
102  cls.def("__str__", [](Matrix3d const &self) {
103  return py::str("[[{!s}, {!s}, {!s}],\n"
104  " [{!s}, {!s}, {!s}],\n"
105  " [{!s}, {!s}, {!s}]]")
106  .format(self(0, 0), self(0, 1), self(0, 2), self(1, 0),
107  self(1, 1), self(1, 2), self(2, 0), self(2, 1),
108  self(2, 2));
109  });
110  cls.def("__repr__", [](Matrix3d const &self) {
111  return py::str("Matrix3d({!r}, {!r}, {!r},\n"
112  " {!r}, {!r}, {!r},\n"
113  " {!r}, {!r}, {!r})")
114  .format(self(0, 0), self(0, 1), self(0, 2), self(1, 0),
115  self(1, 1), self(1, 2), self(2, 0), self(2, 1),
116  self(2, 2));
117  });
118  cls.def("__reduce__", [cls](Matrix3d const &self) {
119  auto args = py::make_tuple(self(0, 0), self(0, 1), self(0, 2),
120  self(1, 0), self(1, 1), self(1, 2),
121  self(2, 0), self(2, 1), self(2, 2));
122  return py::make_tuple(cls, args);
123  });
124 }
125 
126 } // <anonymous>
127 } // sphgeom
128 } // lsst
lsst::meas::algorithms.psfSelectionFromMatchList.args
list args
Definition: psfSelectionFromMatchList.py:27
utils.h
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
row
int row
Definition: CR.cc:145
col
int col
Definition: CR.cc:144
pybind11
Definition: _GenericMap.cc:40
Matrix3d.h
This file contains a class representing 3x3 real matrices.
lsst::utils.tests.init
def init()
Definition: tests.py:58
lsst::afw::cameraGeom::PYBIND11_MODULE
PYBIND11_MODULE(camera, mod)
Definition: camera.cc:133