LSST Applications  21.0.0-172-gfb10e10a+18fedfabac,22.0.0+297cba6710,22.0.0+80564b0ff1,22.0.0+8d77f4f51a,22.0.0+a28f4c53b1,22.0.0+dcf3732eb2,22.0.1-1-g7d6de66+2a20fdde0d,22.0.1-1-g8e32f31+297cba6710,22.0.1-1-geca5380+7fa3b7d9b6,22.0.1-12-g44dc1dc+2a20fdde0d,22.0.1-15-g6a90155+515f58c32b,22.0.1-16-g9282f48+790f5f2caa,22.0.1-2-g92698f7+dcf3732eb2,22.0.1-2-ga9b0f51+7fa3b7d9b6,22.0.1-2-gd1925c9+bf4f0e694f,22.0.1-24-g1ad7a390+a9625a72a8,22.0.1-25-g5bf6245+3ad8ecd50b,22.0.1-25-gb120d7b+8b5510f75f,22.0.1-27-g97737f7+2a20fdde0d,22.0.1-32-gf62ce7b1+aa4237961e,22.0.1-4-g0b3f228+2a20fdde0d,22.0.1-4-g243d05b+871c1b8305,22.0.1-4-g3a563be+32dcf1063f,22.0.1-4-g44f2e3d+9e4ab0f4fa,22.0.1-42-gca6935d93+ba5e5ca3eb,22.0.1-5-g15c806e+85460ae5f3,22.0.1-5-g58711c4+611d128589,22.0.1-5-g75bb458+99c117b92f,22.0.1-6-g1c63a23+7fa3b7d9b6,22.0.1-6-g50866e6+84ff5a128b,22.0.1-6-g8d3140d+720564cf76,22.0.1-6-gd805d02+cc5644f571,22.0.1-8-ge5750ce+85460ae5f3,master-g6e05de7fdc+babf819c66,master-g99da0e417a+8d77f4f51a,w.2021.48
LSST Data Management Base Package
_box3d.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 #include "pybind11/numpy.h"
24 
25 #include "lsst/sphgeom/python.h"
26 
27 #include "lsst/sphgeom/Box3d.h"
30 
31 namespace py = pybind11;
32 using namespace pybind11::literals;
33 
34 namespace lsst {
35 namespace sphgeom {
36 
37 template <>
38 void defineClass(py::class_<Box3d, std::shared_ptr<Box3d>> &cls) {
39  cls.def_static("empty", &Box3d::empty);
40  cls.def_static("full", &Box3d::full);
41  cls.def_static("aroundUnitSphere", &Box3d::aroundUnitSphere);
42 
43  cls.def(py::init<>());
44  cls.def(py::init<Vector3d const &>(), "vector"_a);
45  cls.def(py::init<Vector3d const &, Vector3d const &>(), "vector1"_a,
46  "vector2"_a);
47  cls.def(py::init<Vector3d const &, double, double, double>(), "center"_a,
48  "halfWidth"_a, "halfHeight"_a, "halfDepth"_a);
49  cls.def(py::init<Interval1d const &, Interval1d const &,
50  Interval1d const &>(),
51  "x"_a, "y"_a, "z"_a);
52  cls.def(py::init<Box3d const &>(), "box3d"_a);
53 
54  cls.def("__eq__",
55  (bool (Box3d::*)(Box3d const &) const) & Box3d::operator==,
56  py::is_operator());
57  cls.def("__eq__",
58  (bool (Box3d::*)(Vector3d const &) const) & Box3d::operator==,
59  py::is_operator());
60  cls.def("__ne__",
61  (bool (Box3d::*)(Box3d const &) const) & Box3d::operator!=,
62  py::is_operator());
63  cls.def("__ne__",
64  (bool (Box3d::*)(Vector3d const &) const) & Box3d::operator!=,
65  py::is_operator());
66  cls.def("__contains__",
67  (bool (Box3d::*)(Vector3d const &) const) & Box3d::contains,
68  py::is_operator());
69  cls.def("__contains__",
70  (bool (Box3d::*)(Box3d const &) const) & Box3d::contains,
71  py::is_operator());
72  cls.def("__len__", [](Box3d const &self) { return py::int_(3); });
73  cls.def("__getitem__", [](Box3d const &self, py::int_ row) {
74  return self(static_cast<int>(python::convertIndex(3, row)));
75  });
76 
77  cls.def("x", &Box3d::x);
78  cls.def("y", &Box3d::y);
79  cls.def("z", &Box3d::z);
80  cls.def("isEmpty", &Box3d::isEmpty);
81  cls.def("isFull", &Box3d::isFull);
82  cls.def("getCenter", &Box3d::getCenter);
83  cls.def("getWidth", &Box3d::getWidth);
84  cls.def("getHeight", &Box3d::getHeight);
85  cls.def("getDepth", &Box3d::getDepth);
86 
87  cls.def("contains",
88  (bool (Box3d::*)(Vector3d const &) const) & Box3d::contains);
89  cls.def("contains",
90  (bool (Box3d::*)(Box3d const &) const) & Box3d::contains);
91  cls.def("contains", py::vectorize((bool (Box3d::*)(double, double, double) const)&Box3d::contains),
92  "x"_a, "y"_a, "z"_a);
93  cls.def("isDisjointFrom",
94  (bool (Box3d::*)(Vector3d const &) const) & Box3d::isDisjointFrom);
95  cls.def("isDisjointFrom",
96  (bool (Box3d::*)(Box3d const &) const) & Box3d::isDisjointFrom);
97  cls.def("intersects",
98  (bool (Box3d::*)(Vector3d const &) const) & Box3d::intersects);
99  cls.def("intersects",
100  (bool (Box3d::*)(Box3d const &) const) & Box3d::intersects);
101  cls.def("isWithin",
102  (bool (Box3d::*)(Vector3d const &) const) & Box3d::isWithin);
103  cls.def("isWithin",
104  (bool (Box3d::*)(Box3d const &) const) & Box3d::isWithin);
105 
106  cls.def("clipTo", (Box3d & (Box3d::*)(Vector3d const &)) & Box3d::clipTo);
107  cls.def("clipTo", (Box3d & (Box3d::*)(Box3d const &)) & Box3d::clipTo);
108  cls.def("clippedTo",
109  (Box3d(Box3d::*)(Vector3d const &) const) & Box3d::clippedTo);
110  cls.def("clippedTo",
111  (Box3d(Box3d::*)(Box3d const &) const) & Box3d::clippedTo);
112  cls.def("expandTo",
113  (Box3d & (Box3d::*)(Vector3d const &)) & Box3d::expandTo);
114  cls.def("expandTo", (Box3d & (Box3d::*)(Box3d const &)) & Box3d::expandTo);
115  cls.def("expandedTo",
116  (Box3d(Box3d::*)(Vector3d const &) const) & Box3d::expandedTo);
117  cls.def("expandedTo",
118  (Box3d(Box3d::*)(Box3d const &) const) & Box3d::expandedTo);
119 
120  cls.def("dilateBy", (Box3d & (Box3d::*)(double)) & Box3d::dilateBy,
121  "radius"_a);
122  cls.def("dilateBy",
123  (Box3d & (Box3d::*)(double, double, double)) & Box3d::dilateBy,
124  "width"_a, "height"_a, "depth"_a);
125  cls.def("dilatedBy", (Box3d(Box3d::*)(double) const) & Box3d::dilatedBy,
126  "radius"_a);
127  cls.def("dilatedBy",
128  (Box3d(Box3d::*)(double, double, double) const) & Box3d::dilatedBy,
129  "width"_a, "height"_a, "depth"_a);
130  cls.def("erodeBy", (Box3d & (Box3d::*)(double)) & Box3d::erodeBy,
131  "radius"_a);
132  cls.def("erodeBy",
133  (Box3d & (Box3d::*)(double, double, double)) & Box3d::erodeBy,
134  "width"_a, "height"_a, "depth"_a);
135  cls.def("erodedBy", (Box3d(Box3d::*)(double) const) & Box3d::erodedBy,
136  "radius"_a);
137  cls.def("erodedBy",
138  (Box3d(Box3d::*)(double, double, double) const) & Box3d::erodedBy,
139  "width"_a, "height"_a, "depth"_a);
140 
141  cls.def("relate",
142  (Relationship(Box3d::*)(Vector3d const &) const) & Box3d::relate);
143  cls.def("relate",
144  (Relationship(Box3d::*)(Box3d const &) const) & Box3d::relate);
145 
146  cls.def("__str__", [](Box3d const &self) {
147  return py::str("[{!s},\n"
148  " {!s},\n"
149  " {!s}]")
150  .format(self.x(), self.y(), self.z());
151  });
152  cls.def("__repr__", [](Box3d const &self) {
153  return py::str("Box3d({!r},\n"
154  " {!r},\n"
155  " {!r})")
156  .format(self.x(), self.y(), self.z());
157  });
158  cls.def("__reduce__", [cls](Box3d const &self) {
159  return py::make_tuple(cls,
160  py::make_tuple(self.x(), self.y(), self.z()));
161  });
162 }
163 
164 } // sphgeom
165 } // lsst
This file declares a class for representing axis-aligned bounding boxes in ℝ³.
double x
double z
Definition: Match.cc:44
int y
Definition: SpanSet.cc:48
Box3d represents a box in ℝ³.
Definition: Box3d.h:42
Interval1d represents closed intervals of ℝ.
Definition: Interval1d.h:40
Vector3d is a vector in ℝ³ with components stored in double precision.
Definition: Vector3d.h:44
Relationship relate(VertexIterator const begin, VertexIterator const end, Box const &b)
ptrdiff_t convertIndex(ptrdiff_t len, pybind11::int_ i)
Convert a Python index i over a sequence with length len to a non-negative (C++ style) index,...
Definition: utils.h:40
std::bitset< 3 > Relationship
Relationship describes how two sets are related.
Definition: Relationship.h:35
void defineClass(py::class_< Box3d, std::shared_ptr< Box3d >> &cls)
Definition: _box3d.cc:38
A base class for image defects.
int row
Definition: CR.cc:145