LSST Applications g04e9c324dd+8c5ae1fdc5,g134cb467dc+1b3060144d,g18429d2f64+f642bf4753,g199a45376c+0ba108daf9,g1fd858c14a+2dcf163641,g262e1987ae+7b8c96d2ca,g29ae962dfc+3bd6ecb08a,g2cef7863aa+aef1011c0b,g35bb328faa+8c5ae1fdc5,g3fd5ace14f+53e1a9e7c5,g4595892280+fef73a337f,g47891489e3+2efcf17695,g4d44eb3520+642b70b07e,g53246c7159+8c5ae1fdc5,g67b6fd64d1+2efcf17695,g67fd3c3899+b70e05ef52,g74acd417e5+317eb4c7d4,g786e29fd12+668abc6043,g87389fa792+8856018cbb,g89139ef638+2efcf17695,g8d7436a09f+3be3c13596,g8ea07a8fe4+9f5ccc88ac,g90f42f885a+a4e7b16d9b,g97be763408+ad77d7208f,g9dd6db0277+b70e05ef52,ga681d05dcb+a3f46e7fff,gabf8522325+735880ea63,gac2eed3f23+2efcf17695,gb89ab40317+2efcf17695,gbf99507273+8c5ae1fdc5,gd8ff7fe66e+b70e05ef52,gdab6d2f7ff+317eb4c7d4,gdc713202bf+b70e05ef52,gdfd2d52018+b10e285e0f,ge365c994fd+310e8507c4,ge410e46f29+2efcf17695,geaed405ab2+562b3308c0,gffca2db377+8c5ae1fdc5,w.2025.35
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
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}}
184} // namespace lsst::cpputils::python
185
186#endif
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.