LSST Applications g042eb84c57+730a74494b,g04e9c324dd+8c5ae1fdc5,g134cb467dc+1f1e3e7524,g199a45376c+0ba108daf9,g1fd858c14a+fa7d31856b,g210f2d0738+f66ac109ec,g262e1987ae+83a3acc0e5,g29ae962dfc+d856a2cb1f,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+a1e0c9f713,g47891489e3+0d594cb711,g4d44eb3520+c57ec8f3ed,g4d7b6aa1c5+f66ac109ec,g53246c7159+8c5ae1fdc5,g56a1a4eaf3+fd7ad03fde,g64539dfbff+f66ac109ec,g67b6fd64d1+0d594cb711,g67fd3c3899+f66ac109ec,g6985122a63+0d594cb711,g74acd417e5+3098891321,g786e29fd12+668abc6043,g81db2e9a8d+98e2ab9f28,g87389fa792+8856018cbb,g89139ef638+0d594cb711,g8d7436a09f+80fda9ce03,g8ea07a8fe4+760ca7c3fc,g90f42f885a+033b1d468d,g97be763408+a8a29bda4b,g99822b682c+e3ec3c61f9,g9d5c6a246b+0d5dac0c3d,ga41d0fce20+9243b26dd2,gbf99507273+8c5ae1fdc5,gd7ef33dd92+0d594cb711,gdab6d2f7ff+3098891321,ge410e46f29+0d594cb711,geaed405ab2+c4bbc419c6,gf9a733ac38+8c5ae1fdc5,w.2025.38
LSST Data Management Base Package
Loading...
Searching...
No Matches
_compoundRegion.cc
Go to the documentation of this file.
1/*
2 * This file is part of sphgeom.
3 *
4 * Developed for the LSST Data Management System.
5 * This product includes software developed by the LSST Project
6 * (http://www.lsst.org).
7 * See the COPYRIGHT file at the top-level directory of this distribution
8 * for details of code ownership.
9 *
10 * This software is dual licensed under the GNU General Public License and also
11 * under a 3-clause BSD license. Recipients may choose which of these licenses
12 * to use; please see the files gpl-3.0.txt and/or bsd_license.txt,
13 * respectively. If you choose the GPL option then the following text applies
14 * (but note that there is still no warranty even if you opt for BSD instead):
15 *
16 * This program is free software: you can redistribute it and/or modify
17 * it under the terms of the GNU General Public License as published by
18 * the Free Software Foundation, either version 3 of the License, or
19 * (at your option) any later version.
20 *
21 * This program is distributed in the hope that it will be useful,
22 * but WITHOUT ANY WARRANTY; without even the implied warranty of
23 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
24 * GNU General Public License for more details.
25 *
26 * You should have received a copy of the GNU General Public License
27 * along with this program. If not, see <http://www.gnu.org/licenses/>.
28 */
29#include "pybind11/pybind11.h"
30#include "pybind11/stl.h"
31
32#include "lsst/sphgeom/python.h"
35
36namespace py = pybind11;
37using namespace pybind11::literals;
38
39namespace lsst {
40namespace sphgeom {
41
42namespace {
43
44py::str _repr(const char *class_name, CompoundRegion const &self) {
45 std::string format = class_name;
46 format += "(";
47 py::tuple operands(self.nOperands());
48 for (unsigned i = 0; i != self.nOperands(); i++) {
49 py::object operand = py::cast(self.getOperand(i), py::return_value_policy::reference);
50 operands[i] = operand;
51 if (i != 0) {
52 format += ", ";
53 }
54 format += "{!r}";
55 }
56 format += ")";
57 return py::str(format).format(*operands);
58}
59
60template <typename _CompoundRegion>
61std::unique_ptr<_CompoundRegion> _args_factory(const py::args& args) {
62 std::vector<std::unique_ptr<Region>> operands;
63 for (auto&& item: args) {
64 Region* region = item.cast<Region*>();
65 operands.emplace_back(region->clone());
66 }
67 return std::make_unique<_CompoundRegion>(std::move(operands));
68}
69
70// Iterator for CompoundRegion.
71class CompoundIterator {
72public:
73 CompoundIterator(CompoundRegion const& compound, size_t pos) : _compound(compound), _pos(pos) {}
74
75 Region const& operator*() const { return _compound.getOperand(_pos); }
76
77 CompoundIterator& operator++() {
78 ++ _pos;
79 return *this;
80 }
81
82 bool operator==(CompoundIterator const& other) const {
83 return &_compound == &other._compound and _pos == other._pos;
84 }
85
86private:
87 CompoundRegion const& _compound;
88 size_t _pos;
89};
90
91} // namespace
92
93template <>
94void defineClass(py::classh<CompoundRegion, Region> &cls) {
95 cls.def("nOperands", &CompoundRegion::nOperands);
96 cls.def("__len__", &CompoundRegion::nOperands);
97 cls.def(
98 "__iter__",
99 [](CompoundRegion const& region) {
100 return py::make_iterator(
101 CompoundIterator(region, 0U), CompoundIterator(region, region.nOperands())
102 );
103 },
104 py::return_value_policy::reference_internal // Keeps region alive while iterator is in use.
105 );
106 cls.def(
107 "cloneOperand",
108 [](CompoundRegion const &self, std::ptrdiff_t n) {
109 int nOperands = self.nOperands();
110 return self.getOperand(python::convertIndex(nOperands, n)).clone();
111 }
112 );
113}
114
115template <>
116void defineClass(py::classh<UnionRegion, CompoundRegion> &cls) {
117 cls.attr("TYPE_CODE") = py::int_(UnionRegion::TYPE_CODE);
118 cls.def(py::init(&_args_factory<UnionRegion>));
119 cls.def(py::pickle(&python::encode, &python::decode<UnionRegion>));
120 cls.def("__repr__", [](CompoundRegion const &self) { return _repr("UnionRegion", self); });
121}
122
123template <>
124void defineClass(py::classh<IntersectionRegion, CompoundRegion> &cls) {
125 cls.attr("TYPE_CODE") = py::int_(IntersectionRegion::TYPE_CODE);
126 cls.def(py::init(&_args_factory<IntersectionRegion>));
128 cls.def("__repr__", [](CompoundRegion const &self) { return _repr("IntersectionRegion", self); });
129}
130
131} // namespace sphgeom
132} // namespace lsst
This file declares classes for representing compound regions on the unit sphere.
CompoundRegion is an intermediate base class for spherical regions that are comprised of a point-set ...
Region const & getOperand(std::size_t n) const
static constexpr std::uint8_t TYPE_CODE
Region is a minimal interface for 2-dimensional regions on the unit sphere.
Definition Region.h:89
virtual std::unique_ptr< Region > clone() const =0
clone returns a deep copy of this region.
static constexpr std::uint8_t TYPE_CODE
T emplace_back(T... args)
T move(T... args)
bool operator==(const String &lhs, const String &rhs)
format(config, name=None, writeSourceLine=True, prefix="", verbose=False)
Definition history.py:173
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:48
std::unique_ptr< R > decode(pybind11::bytes bytes)
Decode a Region from a pybind11 bytes object.
Definition utils.h:69
pybind11::bytes encode(Region const &self)
Encode a Region as a pybind11 bytes object.
Definition utils.h:61
void defineClass(Pybind11Class &cls)
Angle operator*(double a, Angle const &b)
Definition Angle.h:105