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
_baseColumnView.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 "pybind11/stl.h"
26 
27 #include "ndarray/pybind11.h"
28 
29 #include "lsst/utils/python.h"
30 
31 #include "lsst/afw/table/Key.h"
33 
34 namespace py = pybind11;
35 using namespace py::literals;
36 
37 namespace lsst {
38 namespace afw {
39 namespace table {
40 
41 using utils::python::WrapperCollection;
42 
43 namespace {
44 
45 using PyBaseColumnView = py::class_<BaseColumnView, std::shared_ptr<BaseColumnView>>;
46 
47 using PyBitsColumn = py::class_<BitsColumn, std::shared_ptr<BitsColumn>>;
48 
49 template <typename T, typename PyClass>
50 static void declareBaseColumnViewOverloads(PyClass &cls) {
51  cls.def("_basicget", [](BaseColumnView & self, Key<T> const &key) -> typename ndarray::Array<T, 1> const {
52  return self[key];
53  });
54 };
55 
56 template <typename U, typename PyClass>
57 static void declareBaseColumnViewArrayOverloads(PyClass &cls) {
58  cls.def("_basicget",
59  [](BaseColumnView & self, Key<lsst::afw::table::Array<U>> const &key) ->
60  typename ndarray::Array<U, 2, 1> const { return self[key]; });
61 };
62 
63 template <typename PyClass>
64 static void declareBaseColumnViewFlagOverloads(PyClass &cls) {
65  cls.def("_basicget",
66  [](BaseColumnView &self, Key<Flag> const &key) -> ndarray::Array<bool const, 1, 1> const {
67  return ndarray::copy(self[key]);
68  });
69 };
70 
71 static void declareBaseColumnView(WrapperCollection &wrappers) {
72  // We can't call this "BaseColumnView" because that's the typedef for "ColumnViewT<BaseRecord>".
73  // This is just a mostly-invisible implementation base class, so we use the same naming convention
74  // we use for those.
75  wrappers.wrapType(PyBaseColumnView(wrappers.module, "_BaseColumnViewBase"), [](auto &mod, auto &cls) {
76  cls.def("getTable", &BaseColumnView::getTable);
77  cls.def_property_readonly("table", &BaseColumnView::getTable);
78  cls.def("getSchema", &BaseColumnView::getSchema);
79  cls.def_property_readonly("schema", &BaseColumnView::getSchema);
80  // _getBits supports a Python version of getBits that accepts None and field names as keys
81  cls.def("_getBits", &BaseColumnView::getBits);
82  cls.def("getAllBits", &BaseColumnView::getAllBits);
83  declareBaseColumnViewOverloads<std::uint8_t>(cls);
84  declareBaseColumnViewOverloads<std::uint16_t>(cls);
85  declareBaseColumnViewOverloads<std::int32_t>(cls);
86  declareBaseColumnViewOverloads<std::int64_t>(cls);
87  declareBaseColumnViewOverloads<float>(cls);
88  declareBaseColumnViewOverloads<double>(cls);
89  declareBaseColumnViewFlagOverloads(cls);
90  // std::string columns are not supported, because numpy string arrays
91  // do not have the same memory model as ours.
92  declareBaseColumnViewArrayOverloads<std::uint8_t>(cls);
93  declareBaseColumnViewArrayOverloads<std::uint16_t>(cls);
94  declareBaseColumnViewArrayOverloads<int>(cls);
95  declareBaseColumnViewArrayOverloads<float>(cls);
96  declareBaseColumnViewArrayOverloads<double>(cls);
97  // lsst::geom::Angle requires custom wrappers, because ndarray doesn't
98  // recognize it natively; we just return a double view
99  // (e.g. radians).
100  using AngleArray = ndarray::Array<lsst::geom::Angle, 1>;
101  using DoubleArray = ndarray::Array<double, 1>;
102  cls.def("_basicget", [](BaseColumnView &self, Key<lsst::geom::Angle> const &key) -> DoubleArray {
103  ndarray::Array<lsst::geom::Angle, 1, 0> a = self[key];
104  return ndarray::detail::ArrayAccess<DoubleArray>::construct(
105  reinterpret_cast<double *>(a.getData()),
106  ndarray::detail::ArrayAccess<AngleArray>::getCore(a));
107  });
108  });
109 }
110 
111 static void declareBitsColumn(WrapperCollection &wrappers) {
112  wrappers.wrapType(PyBitsColumn(wrappers.module, "BitsColumn"), [](auto &mod, auto &cls) {
113  cls.def("getArray", &BitsColumn::getArray);
114  cls.def_property_readonly("array", &BitsColumn::getArray);
115  cls.def("getBit", (BitsColumn::SizeT(BitsColumn::*)(Key<Flag> const &) const) & BitsColumn::getBit,
116  "key"_a);
117  cls.def("getBit", (BitsColumn::SizeT(BitsColumn::*)(std::string const &) const) & BitsColumn::getBit,
118  "name"_a);
119  cls.def("getMask", (BitsColumn::SizeT(BitsColumn::*)(Key<Flag> const &) const) & BitsColumn::getMask,
120  "key"_a);
121  cls.def("getMask", (BitsColumn::SizeT(BitsColumn::*)(std::string const &) const) & BitsColumn::getMask,
122  "name"_a);
123  });
124 }
125 
126 } // namespace
127 
128 void wrapBaseColumnView(WrapperCollection &wrappers) {
129  declareBaseColumnView(wrappers);
130  declareBitsColumn(wrappers);
131 }
132 
133 } // namespace table
134 } // namespace afw
135 } // namespace lsst
Tag types used to declare specialized field types.
Definition: misc.h:31
py::class_< PixelAreaBoundedField, std::shared_ptr< PixelAreaBoundedField >, BoundedField > PyClass
void wrapBaseColumnView(WrapperCollection &wrappers)
A base class for image defects.