Zengtudor/pybind11
1
0
Fork 0
mirror of https://github.com/pybind/pybind11.git synced 2025-02-01 07:40:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding shared_ptr<mpty>, shared_ptr<mpty const> casters.

Browse Source
This commit is contained in:
Ralf W. Grosse-Kunstleve 2021-01-10 13:09:30 -08:00
parent de10541b12
commit d36a7488ae
2 changed files with 77 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
tests/test_type_caster_bare_interface_demo.cpp
View File

@ -1,5 +1,7 @@
#include "pybind11_tests.h"
#include <memory>
namespace pybind11_tests {
namespace type_caster_bare_interface_demo {
@ -19,6 +21,12 @@ const char* pass_mpty_mref(mpty&) { return "load_mref"; }
const char* pass_mpty_cptr(mpty const*) { return "load_cptr"; }
const char* pass_mpty_mptr(mpty*) { return "load_mptr"; }
std::shared_ptr<mpty> rtrn_mpty_shmp() { return std::shared_ptr<mpty>(new mpty); }
std::shared_ptr<mpty const> rtrn_mpty_shcp() { return std::shared_ptr<mpty const>(new mpty); }
const char* pass_mpty_shmp(std::shared_ptr<mpty>) { return "load_shmp"; }
const char* pass_mpty_shcp(std::shared_ptr<mpty const>) { return "load_shcp"; }
} // namespace type_caster_bare_interface_demo
} // namespace pybind11_tests
@ -66,11 +74,11 @@ struct type_caster<mpty> {
template <typename T_>
using cast_op_type = conditional_t<
std::is_same<remove_reference_t<T_>, const mpty*>::value, const mpty*,
std::is_same<remove_reference_t<T_>, mpty const*>::value, mpty const*,
conditional_t<
std::is_same<remove_reference_t<T_>, mpty*>::value, mpty*,
conditional_t<
std::is_same<T_, const mpty&>::value, const mpty&,
std::is_same<T_, mpty const&>::value, mpty const&,
conditional_t<
std::is_same<T_, mpty&>::value, mpty&,
conditional_t<
@ -89,6 +97,44 @@ struct type_caster<mpty> {
}
};
template <>
struct type_caster<std::shared_ptr<mpty>> {
static constexpr auto name = _<std::shared_ptr<mpty>>();
static handle cast(const std::shared_ptr<mpty>& /*src*/,
return_value_policy /*policy*/,
handle /*parent*/) {
return str("cast_shmp").release();
}
template <typename> using cast_op_type = std::shared_ptr<mpty>;
operator std::shared_ptr<mpty>() { return rtrn_mpty_shmp(); }
bool load(handle /*src*/, bool /*convert*/) {
return true;
}
};
template <>
struct type_caster<std::shared_ptr<mpty const>> {
static constexpr auto name = _<std::shared_ptr<mpty const>>();
static handle cast(const std::shared_ptr<mpty const>& /*src*/,
return_value_policy /*policy*/,
handle /*parent*/) {
return str("cast_shcp").release();
}
template <typename> using cast_op_type = std::shared_ptr<mpty const>;
operator std::shared_ptr<mpty const>() { return rtrn_mpty_shcp(); }
bool load(handle /*src*/, bool /*convert*/) {
return true;
}
};
} // namespace detail
} // namespace pybind11
@ -109,6 +155,12 @@ TEST_SUBMODULE(type_caster_bare_interface_demo, m) {
m.def("pass_mpty_mref", pass_mpty_mref);
m.def("pass_mpty_cptr", pass_mpty_cptr);
m.def("pass_mpty_mptr", pass_mpty_mptr);
m.def("rtrn_mpty_shmp", rtrn_mpty_shmp);
m.def("rtrn_mpty_shcp", rtrn_mpty_shcp);
m.def("pass_mpty_shmp", pass_mpty_shmp);
m.def("pass_mpty_shcp", pass_mpty_shcp);
}
} // namespace type_caster_bare_interface_demo

10
tests/test_type_caster_bare_interface_demo.py
View File

@ -20,3 +20,13 @@ def test_load():
assert m.pass_mpty_mref(None) == "load_mref"
assert m.pass_mpty_cptr(None) == "load_cptr"
assert m.pass_mpty_mptr(None) == "load_mptr"
def test_cast_shared_ptr():
assert m.rtrn_mpty_shmp() == "cast_shmp"
assert m.rtrn_mpty_shcp() == "cast_shcp"
def test_load_shared_ptr():
assert m.pass_mpty_shmp(None) == "load_shmp"
assert m.pass_mpty_shcp(None) == "load_shcp"