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
TemplateInvoker.h
Go to the documentation of this file.
1// -*- lsst-c++ -*-
2/*
3 * LSST Data Management System
4 * See COPYRIGHT file at the top of the source tree.
5 *
6 * This product includes software developed by the
7 * LSST Project (http://www.lsst.org/).
8 *
9 * This program is free software: you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation, either version 3 of the License, or
12 * (at your option) any later version.
13 *
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
18 *
19 * You should have received a copy of the LSST License Statement and
20 * the GNU General Public License along with this program. If not,
21 * see <http://www.lsstcorp.org/LegalNotices/>.
22 */
23
24#ifndef LSST_CPPUTILS_PYTHON_TEMPLATEINVOKER_H
25#define LSST_CPPUTILS_PYTHON_TEMPLATEINVOKER_H
26
27#include "pybind11/pybind11.h"
28#include "pybind11/numpy.h"
29
30#include <iostream>
31
32namespace lsst { namespace cpputils { namespace python {
33
106class TemplateInvoker {
107public:
108
110 template <typename ...Types>
111 struct Tag {};
112
114 using OnErrorCallback = std::function<pybind11::object(pybind11::dtype const & dtype)>;
115
117 static pybind11::object handleErrorDefault(pybind11::dtype const & dtype) {
118 PyErr_Format(PyExc_TypeError, "dtype '%R' not supported.", dtype.ptr());
119 throw pybind11::error_already_set();
120 }
121
129 explicit TemplateInvoker(OnErrorCallback onError) : _onError(std::move(onError)) {}
130
133
155 template <typename Function, typename ...TypesToTry>
156 pybind11::object apply(
157 Function function,
158 pybind11::dtype const & dtype,
159 Tag<TypesToTry...> typesToTry
160 ) const {
161 return _apply(function, dtype, typesToTry);
162 }
163
164private:
165
166 template <typename Function>
167 pybind11::object _apply(Function & function, pybind11::dtype const & dtype, Tag<>) const {
168 return _onError(dtype);
169 }
170
171 template <typename Function, typename T, typename ...A>
172 pybind11::object _apply(Function & function, pybind11::dtype const & dtype, Tag<T, A...>) const {
173 if (pybind11::detail::npy_api::get().PyArray_EquivTypes_(dtype.ptr(),
174 pybind11::dtype::of<T>().ptr())) {
175 return pybind11::cast(function(static_cast<T>(0)));
176 }
177 return _apply(function, dtype, Tag<A...>());
178 }
179
180 OnErrorCallback _onError;
181};
182
183}}
184namespace utils = cpputils;
185} // namespace lsst::cpputils::python
186
187#endif
A helper class for wrapping C++ template functions as Python functions with dtype arguments.
std::function< pybind11::object(pybind11::dtype const &dtype)> OnErrorCallback
Callback type for handling unmatched-type errors.
pybind11::object apply(Function function, pybind11::dtype const &dtype, Tag< TypesToTry... > typesToTry) const
Call and return function(static_cast<T>(0)) with the type T that matches a given NumPy dtype object.
TemplateInvoker()
Construct a TemplateInvoker that calls handleErrorDefault when no match is found.
TemplateInvoker(OnErrorCallback onError)
Construct a TemplateInvoker that calls the given object when no match is found.
static pybind11::object handleErrorDefault(pybind11::dtype const &dtype)
Callback used for handling unmatched-type errors by default.
STL namespace.
A simple tag type used to pass one or more types as a function argument.