LSST Applications g063fba187b+cac8b7c890,g0f08755f38+6aee506743,g1653933729+a8ce1bb630,g168dd56ebc+a8ce1bb630,g1a2382251a+b4475c5878,g1dcb35cd9c+8f9bc1652e,g20f6ffc8e0+6aee506743,g217e2c1bcf+73dee94bd0,g28da252d5a+1f19c529b9,g2bbee38e9b+3f2625acfc,g2bc492864f+3f2625acfc,g3156d2b45e+6e55a43351,g32e5bea42b+1bb94961c2,g347aa1857d+3f2625acfc,g35bb328faa+a8ce1bb630,g3a166c0a6a+3f2625acfc,g3e281a1b8c+c5dd892a6c,g3e8969e208+a8ce1bb630,g414038480c+5927e1bc1e,g41af890bb2+8a9e676b2a,g7af13505b9+809c143d88,g80478fca09+6ef8b1810f,g82479be7b0+f568feb641,g858d7b2824+6aee506743,g89c8672015+f4add4ffd5,g9125e01d80+a8ce1bb630,ga5288a1d22+2903d499ea,gb58c049af0+d64f4d3760,gc28159a63d+3f2625acfc,gcab2d0539d+b12535109e,gcf0d15dbbd+46a3f46ba9,gda6a2b7d83+46a3f46ba9,gdaeeff99f8+1711a396fd,ge79ae78c31+3f2625acfc,gef2f8181fd+0a71e47438,gf0baf85859+c1f95f4921,gfa517265be+6aee506743,gfa999e8aa5+17cd334064,w.2024.51
LSST Data Management Base Package
Loading...
Searching...
No Matches
Classes | Public Types | Public Member Functions | Static Public Member Functions | List of all members
lsst::cpputils::python::TemplateInvoker Class Reference

A helper class for wrapping C++ template functions as Python functions with dtype arguments. More...

#include <TemplateInvoker.h>

Classes

struct  Tag
 A simple tag type used to pass one or more types as a function argument. More...
 

Public Types

using OnErrorCallback = std::function<pybind11::object(pybind11::dtype const & dtype)>
 Callback type for handling unmatched-type errors.
 

Public Member Functions

 TemplateInvoker (OnErrorCallback onError)
 Construct a TemplateInvoker that calls the given object when no match is found.
 
 TemplateInvoker ()
 Construct a TemplateInvoker that calls handleErrorDefault when no match is found.
 
template<typename Function , typename ... TypesToTry>
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.
 

Static Public Member Functions

static pybind11::object handleErrorDefault (pybind11::dtype const &dtype)
 Callback used for handling unmatched-type errors by default.
 

Detailed Description

A helper class for wrapping C++ template functions as Python functions with dtype arguments.

TemplateInvoker takes a templated callable object, a pybind11::dtype object, and a sequence of supported C++ types via its nested Tag struct. The callable is invoked with a scalar argument of the type matching the dtype object. If none of the supported C++ types match, a different error callback is invoked instead.

As an example, we'll wrap this function:

template <typename T>
T doSomething(std::string const & argument);

TemplateInvoker provides a default error callback, which we'll use here (otherwise you'd need to pass one when constructing the TemplateInvoker).

For the main callback, we'll define this helper struct:

struct DoSomethingHelper {
template <typename T>
T operator()(T) const {
return doSomething<T>(argument);
}
std::string argument;
};

The pybind11 wrapper for doSomething is then another lambda that uses TemplateInvoker::apply to call the helper:

mod.def(
"doSomething",
[](std::string const & argument, py::dtype const & dtype) {
return TemplateInvoker().apply(
DoSomethingHelper{argument},
dtype,
);
},
"argument"_a
);
TemplateInvoker()
Construct a TemplateInvoker that calls handleErrorDefault when no match is found.
A simple tag type used to pass one or more types as a function argument.

The type returned by the helper callable's operator() can be anything pybind11 knows how to convert to Python.

While defining a full struct with a templated operator() makes it more obvious what TemplateInvoker is doing, it's much more concise to use a universal lambda with the decltype operator. This wrapper is equivalent to the one above, but it doesn't need DoSomethingHelper:

mod.def(
"doSomething",
[](std::string const & argument, py::dtype const & dtype) {
return TemplateInvoker().apply(
[&argument](auto t) { return doSomething<decltype(t)>(argument); },
dtype,
);
},
"argument"_a
);

Note that the value of t here doesn't matter; what's important is that its C++ type corresponds to the type passed in the dtype argument. So instead of using that value, we use the decltype operator to extract that type and use it as a template parameter.

Definition at line 106 of file TemplateInvoker.h.

Member Typedef Documentation

◆ OnErrorCallback

using lsst::cpputils::python::TemplateInvoker::OnErrorCallback = std::function<pybind11::object(pybind11::dtype const & dtype)>

Callback type for handling unmatched-type errors.

Definition at line 114 of file TemplateInvoker.h.

Constructor & Destructor Documentation

◆ TemplateInvoker() [1/2]

lsst::cpputils::python::TemplateInvoker::TemplateInvoker ( OnErrorCallback onError)
inlineexplicit

Construct a TemplateInvoker that calls the given object when no match is found.

The callback should have the same signature as handleErrorDefault; the dtype actually passed from Python is passed so it can be included in error messages.

Definition at line 129 of file TemplateInvoker.h.

129: _onError(std::move(onError)) {}
T move(T... args)

◆ TemplateInvoker() [2/2]

lsst::cpputils::python::TemplateInvoker::TemplateInvoker ( )
inline

Construct a TemplateInvoker that calls handleErrorDefault when no match is found.

Definition at line 132 of file TemplateInvoker.h.

static pybind11::object handleErrorDefault(pybind11::dtype const &dtype)
Callback used for handling unmatched-type errors by default.

Member Function Documentation

◆ apply()

template<typename Function , typename ... TypesToTry>
pybind11::object lsst::cpputils::python::TemplateInvoker::apply ( Function function,
pybind11::dtype const & dtype,
Tag< TypesToTry... > typesToTry ) const
inline

Call and return function(static_cast<T>(0)) with the type T that matches a given NumPy dtype object.

Parameters
[in]functionCallable object to invoke. Must have an overloaded operator() that takes any T in the sequence TypesToTry, and a fail(py::dtype) method to handle the case where none of the given types match.
[in]dtypeNumPy dtype object indicating the template specialization to invoke.
[in]typesToTryA Tag instance parameterized with the list of types to try to match to dtype.
Returns
the result of calling function with the matching type, after converting it into a Python object.
Exception Safety
the same as the exception safety of function

Definition at line 156 of file TemplateInvoker.h.

160 {
161 return _apply(function, dtype, typesToTry);
162 }

◆ handleErrorDefault()

static pybind11::object lsst::cpputils::python::TemplateInvoker::handleErrorDefault ( pybind11::dtype const & dtype)
inlinestatic

Callback used for handling unmatched-type errors by default.

Definition at line 117 of file TemplateInvoker.h.

117 {
118 PyErr_Format(PyExc_TypeError, "dtype '%R' not supported.", dtype.ptr());
119 throw pybind11::error_already_set();
120 }

The documentation for this class was generated from the following file: