LSSTApplications  20.0.0
LSSTDataManagementBasePackage
vector3d.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/Angle.h"
28 #include "lsst/sphgeom/Vector3d.h"
30 
31 namespace py = pybind11;
32 using namespace pybind11::literals;
33 
34 namespace lsst {
35 namespace sphgeom {
36 namespace {
37 
38 PYBIND11_MODULE(vector3d, mod) {
39  py::class_<Vector3d, std::shared_ptr<Vector3d>> cls(mod, "Vector3d");
40 
41  cls.def(py::init<>());
42  cls.def(py::init<double, double, double>(), "x"_a, "y"_a, "z"_a);
43  cls.def(py::init<Vector3d const &>(), "vector"_a);
44  // Construct a Vector3d from a UnitVector3d, enabling implicit
45  // conversion from UnitVector3d to Vector3d in python via
46  // py::implicitly_convertible
47  cls.def(py::init([](UnitVector3d const &u) {
48  return new Vector3d(u.x(), u.y(), u.z());
49  }));
50 
51  cls.def("__eq__", &Vector3d::operator==, py::is_operator());
52  cls.def("__ne__", &Vector3d::operator!=, py::is_operator());
53  cls.def("__neg__", (Vector3d(Vector3d::*)() const) & Vector3d::operator-);
54  cls.def("__add__", &Vector3d::operator+, py::is_operator());
55  cls.def("__sub__",
56  (Vector3d(Vector3d::*)(Vector3d const &) const) &
57  Vector3d::operator-,
58  py::is_operator());
59  cls.def("__mul__", &Vector3d::operator*, py::is_operator());
60  cls.def("__truediv__", &Vector3d::operator/, py::is_operator());
61 
62  cls.def("__iadd__", &Vector3d::operator+=);
63  cls.def("__isub__", &Vector3d::operator-=);
64  cls.def("__imul__", &Vector3d::operator*=);
65  cls.def("__itruediv__", &Vector3d::operator/=);
66 
67  cls.def("x", &Vector3d::x);
68  cls.def("y", &Vector3d::y);
69  cls.def("z", &Vector3d::z);
70  cls.def("dot", &Vector3d::dot);
71  cls.def("getSquaredNorm", &Vector3d::getSquaredNorm);
72  cls.def("getNorm", &Vector3d::getNorm);
73  cls.def("isZero", &Vector3d::isZero);
74  cls.def("normalize", &Vector3d::normalize);
75  cls.def("isNormalized", &Vector3d::isNormalized);
76  cls.def("cross", &Vector3d::cross);
77  cls.def("cwiseProduct", &Vector3d::cwiseProduct);
78  cls.def("rotatedAround", &Vector3d::rotatedAround, "axis"_a, "angle"_a);
79 
80  cls.def("__len__", [](Vector3d const &self) { return py::int_(3); });
81  cls.def("__getitem__", [](Vector3d const &self, py::int_ i) {
82  return self(python::convertIndex(3, i));
83  });
84 
85  cls.def("__str__", [](Vector3d const &self) {
86  return py::str("[{!s}, {!s}, {!s}]")
87  .format(self.x(), self.y(), self.z());
88  });
89  cls.def("__repr__", [](Vector3d const &self) {
90  return py::str("Vector3d({!r}, {!r}, {!r})")
91  .format(self.x(), self.y(), self.z());
92  });
93 
94  cls.def("__reduce__", [cls](Vector3d const &self) {
95  return py::make_tuple(cls,
96  py::make_tuple(self.x(), self.y(), self.z()));
97  });
98 }
99 
100 } // <anonymous>
101 } // sphgeom
102 } // lsst
y
int y
Definition: SpanSet.cc:49
utils.h
Angle.h
This file declares a class for representing angles.
UnitVector3d.h
This file declares a class for representing unit vectors in ℝ³.
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
lsst::afw.display.ds9.dot
def dot(symb, c, r, frame=None, size=2, ctype=None, origin=afwImage.PARENT, *args, **kwargs)
Definition: ds9.py:101
z
double z
Definition: Match.cc:44
x
double x
Definition: ChebyshevBoundedField.cc:277
Vector3d.h
This file declares a class for representing vectors in ℝ³.
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
pybind11
Definition: _GenericMap.cc:40
lsst::utils.tests.init
def init()
Definition: tests.py:58
lsst::afw::cameraGeom::PYBIND11_MODULE
PYBIND11_MODULE(camera, mod)
Definition: camera.cc:133