LSST Applications g07dc498a13+cb17356775,g1409bbee79+cb17356775,g1a7e361dbc+cb17356775,g1fd858c14a+d1a2a640a9,g33399d78f5+fe6948661d,g35bb328faa+e55fef2c71,g3bd4b5ce2c+cac9e18807,g3c79e8cd92+2359a18b76,g43bc871e57+a58ba40925,g53246c7159+e55fef2c71,g60b5630c4e+8133a3545f,g78460c75b0+8427c4cc8f,g78619a8342+55305cb8f0,g786e29fd12+307f82e6af,g8534526c7b+8e1c6b434f,g89139ef638+cb17356775,g8b49a6ea8e+8133a3545f,g8ffcb69f3d+818ab6c36e,g9125e01d80+e55fef2c71,g97b8272a79+98425f45f6,g989de1cb63+cb17356775,g9f33ca652e+4245ceb508,gaaedd4e678+cb17356775,gabe3b4be73+9c0c3c7524,gb1101e3267+3e5ef1d639,gb58c049af0+28045f66fd,gc1fe0db326+8133a3545f,gca43fec769+e55fef2c71,gcf25f946ba+fe6948661d,gd397e13551+64039f84ea,gd6cbbdb0b4+f6e5445f66,gde0f65d7ad+7eb368c542,ge278dab8ac+b4c2c8faf7,geab183fbe5+8133a3545f,gecb8035dfe+1f480bec5e,gefa07fa684+e7bc33f3ea,gf58bf46354+e55fef2c71,gfe7187db8c+e55afb4430,w.2025.03
LSST Data Management Base Package
Loading...
Searching...
No Matches
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: