Merge branch 'master' into sh_merge_master

This commit is contained in:
Ralf W. Grosse-Kunstleve 2023-05-08 10:27:38 -07:00
commit eb456b1599
4 changed files with 70 additions and 7 deletions

View File

@ -26,6 +26,9 @@ struct is_method {
explicit is_method(const handle &c) : class_(c) {} explicit is_method(const handle &c) : class_(c) {}
}; };
/// Annotation for setters
struct is_setter {};
/// Annotation for operators /// Annotation for operators
struct is_operator {}; struct is_operator {};
@ -188,8 +191,8 @@ struct argument_record {
struct function_record { struct function_record {
function_record() function_record()
: is_constructor(false), is_new_style_constructor(false), is_stateless(false), : is_constructor(false), is_new_style_constructor(false), is_stateless(false),
is_operator(false), is_method(false), has_args(false), has_kwargs(false), is_operator(false), is_method(false), is_setter(false), has_args(false),
prepend(false) {} has_kwargs(false), prepend(false) {}
/// Function name /// Function name
char *name = nullptr; /* why no C++ strings? They generate heavier code.. */ char *name = nullptr; /* why no C++ strings? They generate heavier code.. */
@ -230,6 +233,9 @@ struct function_record {
/// True if this is a method /// True if this is a method
bool is_method : 1; bool is_method : 1;
/// True if this is a setter
bool is_setter : 1;
/// True if the function has a '*args' argument /// True if the function has a '*args' argument
bool has_args : 1; bool has_args : 1;
@ -426,6 +432,12 @@ struct process_attribute<is_method> : process_attribute_default<is_method> {
} }
}; };
/// Process an attribute which indicates that this function is a setter
template <>
struct process_attribute<is_setter> : process_attribute_default<is_setter> {
static void init(const is_setter &, function_record *r) { r->is_setter = true; }
};
/// Process an attribute which indicates the parent scope of a method /// Process an attribute which indicates the parent scope of a method
template <> template <>
struct process_attribute<scope> : process_attribute_default<scope> { struct process_attribute<scope> : process_attribute_default<scope> {

View File

@ -85,6 +85,7 @@ public:
cpp_function() = default; cpp_function() = default;
// NOLINTNEXTLINE(google-explicit-constructor) // NOLINTNEXTLINE(google-explicit-constructor)
cpp_function(std::nullptr_t) {} cpp_function(std::nullptr_t) {}
cpp_function(std::nullptr_t, const is_setter &) {}
/// Construct a cpp_function from a vanilla function pointer /// Construct a cpp_function from a vanilla function pointer
template <typename Return, typename... Args, typename... Extra> template <typename Return, typename... Args, typename... Extra>
@ -245,10 +246,16 @@ protected:
using Guard = extract_guard_t<Extra...>; using Guard = extract_guard_t<Extra...>;
/* Perform the function call */ /* Perform the function call */
handle result handle result;
= cast_out::cast(std::move(args_converter).template call<Return, Guard>(cap->f), if (call.func.is_setter) {
policy, (void) std::move(args_converter).template call<Return, Guard>(cap->f);
call.parent); result = none().release();
} else {
result = cast_out::cast(
std::move(args_converter).template call<Return, Guard>(cap->f),
policy,
call.parent);
}
/* Invoke call policy post-call hook */ /* Invoke call policy post-call hook */
process_attributes<Extra...>::postcall(call, result); process_attributes<Extra...>::postcall(call, result);
@ -1968,7 +1975,8 @@ public:
template <typename Getter, typename Setter, typename... Extra> template <typename Getter, typename Setter, typename... Extra>
class_ & class_ &
def_property(const char *name, const Getter &fget, const Setter &fset, const Extra &...extra) { def_property(const char *name, const Getter &fget, const Setter &fset, const Extra &...extra) {
return def_property(name, fget, cpp_function(method_adaptor<type>(fset)), extra...); return def_property(
name, fget, cpp_function(method_adaptor<type>(fset), is_setter()), extra...);
} }
template <typename Getter, typename... Extra> template <typename Getter, typename... Extra>
class_ &def_property(const char *name, class_ &def_property(const char *name,

View File

@ -177,6 +177,38 @@ struct RValueRefParam {
std::size_t func4(std::string &&s) const & { return s.size(); } std::size_t func4(std::string &&s) const & { return s.size(); }
}; };
namespace pybind11_tests {
namespace exercise_is_setter {
struct FieldBase {
int int_value() const { return int_value_; }
FieldBase &SetIntValue(int int_value) {
int_value_ = int_value;
return *this;
}
private:
int int_value_ = -99;
};
struct Field : FieldBase {};
void add_bindings(py::module &m) {
py::module sm = m.def_submodule("exercise_is_setter");
// NOTE: FieldBase is not wrapped, therefore ...
py::class_<Field>(sm, "Field")
.def(py::init<>())
.def_property(
"int_value",
&Field::int_value,
&Field::SetIntValue // ... the `FieldBase &` return value here cannot be converted.
);
}
} // namespace exercise_is_setter
} // namespace pybind11_tests
TEST_SUBMODULE(methods_and_attributes, m) { TEST_SUBMODULE(methods_and_attributes, m) {
// test_methods_and_attributes // test_methods_and_attributes
py::class_<ExampleMandA> emna(m, "ExampleMandA"); py::class_<ExampleMandA> emna(m, "ExampleMandA");
@ -460,4 +492,6 @@ TEST_SUBMODULE(methods_and_attributes, m) {
.def("func2", &RValueRefParam::func2) .def("func2", &RValueRefParam::func2)
.def("func3", &RValueRefParam::func3) .def("func3", &RValueRefParam::func3)
.def("func4", &RValueRefParam::func4); .def("func4", &RValueRefParam::func4);
pybind11_tests::exercise_is_setter::add_bindings(m);
} }

View File

@ -524,3 +524,12 @@ def test_rvalue_ref_param():
assert r.func2("1234") == 4 assert r.func2("1234") == 4
assert r.func3("12345") == 5 assert r.func3("12345") == 5
assert r.func4("123456") == 6 assert r.func4("123456") == 6
def test_is_setter():
fld = m.exercise_is_setter.Field()
assert fld.int_value == -99
setter_return = fld.int_value = 100
assert isinstance(setter_return, int)
assert setter_return == 100
assert fld.int_value == 100