mirror of
https://github.com/pybind/pybind11.git
synced 2026-06-06 06:11:25 +00:00
improved shared pointer support (fixes #14)
This commit is contained in:
@@ -31,14 +31,36 @@ private:
|
||||
int value;
|
||||
};
|
||||
|
||||
class MyObject2 : public std::enable_shared_from_this<MyObject2> {
|
||||
public:
|
||||
MyObject2(int value) : value(value) {
|
||||
std::cout << toString() << " constructor" << std::endl;
|
||||
}
|
||||
|
||||
std::string toString() const {
|
||||
return "MyObject2[" + std::to_string(value) + "]";
|
||||
}
|
||||
|
||||
virtual ~MyObject2() {
|
||||
std::cout << toString() << " destructor" << std::endl;
|
||||
}
|
||||
|
||||
private:
|
||||
int value;
|
||||
};
|
||||
|
||||
/// Make pybind aware of the ref-counted wrapper type
|
||||
PYBIND11_DECLARE_HOLDER_TYPE(T, ref<T>);
|
||||
PYBIND11_DECLARE_HOLDER_TYPE(T, std::shared_ptr<T>);
|
||||
|
||||
Object *make_object_1() { return new MyObject(1); }
|
||||
ref<Object> make_object_2() { return new MyObject(2); }
|
||||
MyObject *make_myobject_4() { return new MyObject(4); }
|
||||
ref<MyObject> make_myobject_5() { return new MyObject(5); }
|
||||
|
||||
MyObject2 *make_myobject2_1() { return new MyObject2(1); }
|
||||
std::shared_ptr<MyObject2> make_myobject2_2() { return std::make_shared<MyObject2>(2); }
|
||||
|
||||
void print_object_1(const Object *obj) { std::cout << obj->toString() << std::endl; }
|
||||
void print_object_2(ref<Object> obj) { std::cout << obj->toString() << std::endl; }
|
||||
void print_object_3(const ref<Object> &obj) { std::cout << obj->toString() << std::endl; }
|
||||
@@ -49,6 +71,11 @@ void print_myobject_2(ref<MyObject> obj) { std::cout << obj->toString() << std::
|
||||
void print_myobject_3(const ref<MyObject> &obj) { std::cout << obj->toString() << std::endl; }
|
||||
void print_myobject_4(const ref<MyObject> *obj) { std::cout << (*obj)->toString() << std::endl; }
|
||||
|
||||
void print_myobject2_1(const MyObject2 *obj) { std::cout << obj->toString() << std::endl; }
|
||||
void print_myobject2_2(std::shared_ptr<MyObject2> obj) { std::cout << obj->toString() << std::endl; }
|
||||
void print_myobject2_3(const std::shared_ptr<MyObject2> &obj) { std::cout << obj->toString() << std::endl; }
|
||||
void print_myobject2_4(const std::shared_ptr<MyObject2> *obj) { std::cout << (*obj)->toString() << std::endl; }
|
||||
|
||||
void init_ex8(py::module &m) {
|
||||
py::class_<Object, ref<Object>> obj(m, "Object");
|
||||
obj.def("getRefCount", &Object::getRefCount);
|
||||
@@ -69,5 +96,14 @@ void init_ex8(py::module &m) {
|
||||
m.def("print_myobject_3", &print_myobject_3);
|
||||
m.def("print_myobject_4", &print_myobject_4);
|
||||
|
||||
py::class_<MyObject2, std::shared_ptr<MyObject2>>(m, "MyObject2")
|
||||
.def(py::init<int>());
|
||||
m.def("make_myobject2_1", &make_myobject2_1);
|
||||
m.def("make_myobject2_2", &make_myobject2_2);
|
||||
m.def("print_myobject2_1", &print_myobject2_1);
|
||||
m.def("print_myobject2_2", &print_myobject2_2);
|
||||
m.def("print_myobject2_3", &print_myobject2_3);
|
||||
m.def("print_myobject2_4", &print_myobject2_4);
|
||||
|
||||
py::implicitly_convertible<py::int_, MyObject>();
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user