mirror of
https://github.com/pybind/pybind11.git
synced 2024-11-25 14:45:12 +00:00
type_caster<std::function>: allow None values in both directions
This commit is contained in:
parent
0b63231bae
commit
8de0437e46
@ -38,7 +38,10 @@ py::cpp_function test_callback5() {
|
||||
int dummy_function(int i) { return i + 1; }
|
||||
int dummy_function2(int i, int j) { return i + j; }
|
||||
std::function<int(int)> roundtrip(std::function<int(int)> f) {
|
||||
std::cout << "roundtrip.." << std::endl;
|
||||
if (!f)
|
||||
std::cout << "roundtrip (got None).." << std::endl;
|
||||
else
|
||||
std::cout << "roundtrip.." << std::endl;
|
||||
return f;
|
||||
}
|
||||
|
||||
|
@ -68,6 +68,8 @@ from example import roundtrip
|
||||
|
||||
test_dummy_function(dummy_function)
|
||||
test_dummy_function(roundtrip(dummy_function))
|
||||
if roundtrip(None) is not None:
|
||||
print("Problem!")
|
||||
test_dummy_function(lambda x: x + 2)
|
||||
|
||||
try:
|
||||
|
@ -30,6 +30,7 @@ Copy constructions: 1
|
||||
Move constructions: True
|
||||
argument matches dummy_function
|
||||
eval(1) = 2
|
||||
roundtrip (got None)..
|
||||
roundtrip..
|
||||
argument matches dummy_function
|
||||
eval(1) = 2
|
||||
|
@ -20,6 +20,9 @@ template <typename Return, typename... Args> struct type_caster<std::function<Re
|
||||
typedef typename std::conditional<std::is_same<Return, void>::value, void_type, Return>::type retval_type;
|
||||
public:
|
||||
bool load(handle src_, bool) {
|
||||
if (src_.ptr() == Py_None)
|
||||
return true;
|
||||
|
||||
src_ = detail::get_function(src_);
|
||||
if (!src_ || !PyCallable_Check(src_.ptr()))
|
||||
return false;
|
||||
@ -58,6 +61,9 @@ public:
|
||||
|
||||
template <typename Func>
|
||||
static handle cast(Func &&f_, return_value_policy policy, handle /* parent */) {
|
||||
if (!f_)
|
||||
return handle(Py_None).inc_ref();
|
||||
|
||||
auto result = f_.template target<Return (*)(Args...)>();
|
||||
if (result)
|
||||
return cpp_function(*result, policy).release();
|
||||
|
Loading…
Reference in New Issue
Block a user