forked from pybind/pybind11
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Equivalent of PR pybind#4597, but based on pybind11k main @ ab472bbce…
…81de4e0ae41d861a5c46903968f7b27 The pybind11k `return_value_policy_pack` and `function_record` changes were backed out. The net diffs to PR pybind#4597 are merely: ```diff +#define PYBIND11_HAS_TYPE_CASTER_STD_FUNCTION_SPECIALIZATIONS ``` ```diff - explicit func_wrapper_base(func_handle &&hf) noexcept : hfunc(hf) {} + explicit func_wrapper_base(func_handle &&hf) noexcept : hfunc(std::move(hf)) {} ```
- Loading branch information
Ralf W. Grosse-Kunstleve
committed
Aug 6, 2024
1 parent
7c6fe49
commit 50361b1
Showing
4 changed files
with
107 additions
and
34 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,46 @@ | ||
#include <pybind11/functional.h> | ||
#include <pybind11/pybind11.h> | ||
|
||
#include "pybind11_tests.h" | ||
|
||
namespace py = pybind11; | ||
|
||
namespace { | ||
|
||
struct SpecialReturn { | ||
int value = 99; | ||
}; | ||
|
||
} // namespace | ||
|
||
namespace pybind11 { | ||
namespace detail { | ||
namespace type_caster_std_function_specializations { | ||
|
||
template <typename... Args> | ||
struct func_wrapper<SpecialReturn, Args...> : func_wrapper_base { | ||
using func_wrapper_base::func_wrapper_base; | ||
SpecialReturn operator()(Args... args) const { | ||
gil_scoped_acquire acq; | ||
SpecialReturn result; | ||
try { | ||
result = hfunc.f(std::forward<Args>(args)...).template cast<SpecialReturn>(); | ||
} catch (error_already_set &) { | ||
result.value += 1; | ||
} | ||
result.value += 100; | ||
return result; | ||
} | ||
}; | ||
|
||
} // namespace type_caster_std_function_specializations | ||
} // namespace detail | ||
} // namespace pybind11 | ||
|
||
TEST_SUBMODULE(type_caster_std_function_specializations, m) { | ||
py::class_<SpecialReturn>(m, "SpecialReturn") | ||
.def(py::init<>()) | ||
.def_readwrite("value", &SpecialReturn::value); | ||
m.def("call_callback_with_special_return", | ||
[](const std::function<SpecialReturn()> &func) { return func(); }); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,15 @@ | ||
from __future__ import annotations | ||
|
||
from pybind11_tests import type_caster_std_function_specializations as m | ||
|
||
|
||
def test_callback_with_special_return(): | ||
def return_special(): | ||
return m.SpecialReturn() | ||
|
||
def raise_exception(): | ||
raise ValueError("called raise_exception.") | ||
|
||
assert return_special().value == 99 | ||
assert m.call_callback_with_special_return(return_special).value == 199 | ||
assert m.call_callback_with_special_return(raise_exception).value == 200 |