LSSTApplications  20.0.0
LSSTDataManagementBasePackage
chunker.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/stl.h"
24 
25 #include <memory>
26 
27 #include "lsst/sphgeom/Chunker.h"
28 
29 namespace py = pybind11;
30 using namespace pybind11::literals;
31 
32 namespace lsst {
33 namespace sphgeom {
34 namespace {
35 
36 py::str toString(Chunker const & self) {
37  return py::str("Chunker({!s}, {!s})")
38  .format(self.getNumStripes(), self.getNumSubStripesPerStripe());
39 }
40 
41 
42 PYBIND11_MODULE(chunker, mod) {
43  py::class_<Chunker, std::shared_ptr<Chunker>> cls(mod, "Chunker");
44 
45  cls.def(py::init<int32_t, int32_t>(), "numStripes"_a,
46  "numSubStripesPerStripe"_a);
47 
48  cls.def("__eq__", &Chunker::operator==, py::is_operator());
49  cls.def("__ne__", &Chunker::operator!=, py::is_operator());
50 
51  cls.def_property_readonly("numStripes", &Chunker::getNumStripes);
52  cls.def_property_readonly("numSubStripesPerStripe",
53  &Chunker::getNumSubStripesPerStripe);
54 
55  cls.def("getChunksIntersecting", &Chunker::getChunksIntersecting,
56  "region"_a);
57  cls.def("getSubChunksIntersecting",
58  [](Chunker const &self, Region const &region) {
59  py::list results;
60  for (auto const & sc: self.getSubChunksIntersecting(region)) {
61  results.append(py::make_tuple(sc.chunkId, sc.subChunkIds));
62  }
63  return results;
64  },
65  "region"_a);
66  cls.def("getAllChunks", &Chunker::getAllChunks);
67  cls.def("getAllSubChunks", &Chunker::getAllSubChunks, "chunkId"_a);
68 
69  cls.def("__str__", &toString);
70  cls.def("__repr__", &toString);
71 
72  cls.def("__reduce__", [cls](Chunker const &self) {
73  return py::make_tuple(cls,
74  py::make_tuple(self.getNumStripes(),
75  self.getNumSubStripesPerStripe()));
76  });
77 }
78 
79 } // <anonymous>
80 } // sphgeom
81 } // lsst
Chunker.h
This file declares a class for partitioning the sky into chunks and sub-chunks.
lsst::afw::geom.transform.transformContinued.cls
cls
Definition: transformContinued.py:33
lsst
A base class for image defects.
Definition: imageAlgorithm.dox:1
list
daf::base::PropertyList * list
Definition: fits.cc:913
pybind11
Definition: _GenericMap.cc:40
lsst::afw::cameraGeom::PYBIND11_MODULE
PYBIND11_MODULE(camera, mod)
Definition: camera.cc:133