LSST Applications 27.0.0,g0265f82a02+469cd937ee,g02d81e74bb+21ad69e7e1,g1470d8bcf6+cbe83ee85a,g2079a07aa2+e67c6346a6,g212a7c68fe+04a9158687,g2305ad1205+94392ce272,g295015adf3+81dd352a9d,g2bbee38e9b+469cd937ee,g337abbeb29+469cd937ee,g3939d97d7f+72a9f7b576,g487adcacf7+71499e7cba,g50ff169b8f+5929b3527e,g52b1c1532d+a6fc98d2e7,g591dd9f2cf+df404f777f,g5a732f18d5+be83d3ecdb,g64a986408d+21ad69e7e1,g858d7b2824+21ad69e7e1,g8a8a8dda67+a6fc98d2e7,g99cad8db69+f62e5b0af5,g9ddcbc5298+d4bad12328,ga1e77700b3+9c366c4306,ga8c6da7877+71e4819109,gb0e22166c9+25ba2f69a1,gb6a65358fc+469cd937ee,gbb8dafda3b+69d3c0e320,gc07e1c2157+a98bf949bb,gc120e1dc64+615ec43309,gc28159a63d+469cd937ee,gcf0d15dbbd+72a9f7b576,gdaeeff99f8+a38ce5ea23,ge6526c86ff+3a7c1ac5f1,ge79ae78c31+469cd937ee,gee10cc3b42+a6fc98d2e7,gf1cff7945b+21ad69e7e1,gfbcc870c63+9a11dc8c8f
LSST Data Management Base Package
Loading...
Searching...
No Matches
_detectorCollection.cc
Go to the documentation of this file.
1/*
2 * This file is part of afw.
3 *
4 * Developed for the LSST Data Management System.
5 * This product includes software developed by the LSST Project
6 * (https://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 program is free software: you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation, either version 3 of the License, or
13 * (at your option) any later version.
14 *
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
19 *
20 * You should have received a copy of the GNU General Public License
21 * along with this program. If not, see <https://www.gnu.org/licenses/>.
22 */
23
24#include "pybind11/pybind11.h"
25#include <lsst/utils/python.h>
26
27#include "pybind11/stl.h"
28
31
32namespace py = pybind11;
33using namespace py::literals;
34
35namespace lsst {
36namespace afw {
37namespace cameraGeom {
38
39namespace {
40
41template <typename T>
42using PyDetectorCollectionBase =
43 py::class_<DetectorCollectionBase<T>, std::shared_ptr<DetectorCollectionBase<T>>>;
44
45using PyDetectorCollection = py::class_<DetectorCollection, DetectorCollectionBase<Detector const>,
47
48template <typename T>
49void declareDetectorCollectionBase(PyDetectorCollectionBase<T> &cls) {
50 cls.def("getNameMap", &DetectorCollectionBase<T>::getNameMap);
51 cls.def("getIdMap", &DetectorCollectionBase<T>::getIdMap);
52 cls.def("__len__", &DetectorCollectionBase<T>::size);
53 cls.def("get",
55 py::const_),
56 "name"_a, "default"_a = nullptr);
57 cls.def("get", py::overload_cast<int, std::shared_ptr<T>>(&DetectorCollectionBase<T>::get, py::const_),
58 "id"_a, "default"_a = nullptr);
59 cls.def("__contains__", [](DetectorCollectionBase<T> const &self, std::string const &name) {
60 return self.get(name) != nullptr;
61 });
62 cls.def("__contains__",
63 [](DetectorCollectionBase<T> const &self, int id) { return self.get(id) != nullptr; });
64}
65} // namespace
66void wrapDetectorCollection(lsst::utils::python::WrapperCollection &wrappers) {
67 wrappers.addInheritanceDependency("lsst.afw.table.io");
68 wrappers.wrapType(
69 PyDetectorCollectionBase<Detector const>(wrappers.module, "DetectorCollectionDetectorBase"),
70 [](auto &mod, auto &cls) { declareDetectorCollectionBase(cls); });
71 wrappers.wrapType(PyDetectorCollection(wrappers.module, "DetectorCollection"), [](auto &mod, auto &cls) {
72 ;
73 cls.def(py::init<DetectorCollection::List>());
74 cls.def("getFpBBox", &DetectorCollection::getFpBBox);
75 table::io::python::addPersistableMethods(cls);
76 });
77
78 wrappers.wrapType(PyDetectorCollectionBase<Detector::InCameraBuilder>(wrappers.module,
79 "DetectorCollectionBuilderBase"),
80 [](auto &mod, auto &cls) { declareDetectorCollectionBase(cls); });
81}
82} // namespace cameraGeom
83} // namespace afw
84} // namespace lsst
std::size_t size() const noexcept
Get the number of detectors.
IdMap const & getIdMap() const noexcept
Get a map keyed and ordered by ID.
NameMap const & getNameMap() const noexcept
Get a map keyed and ordered by name.
std::shared_ptr< T > get(std::string const &name, std::shared_ptr< T > def=nullptr) const
Retrieve a detector by name, or fall back to a default.
void wrapDetectorCollection(lsst::utils::python::WrapperCollection &)