1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#include <nanobind/nanobind.h>
#include <nanobind/stl/pair.h>
#include <nanobind/trampoline.h>
#include <nanobind/intrusive/counter.h>
#include <nanobind/intrusive/ref.h>
namespace nb = nanobind;
using namespace nb::literals;
static int test_constructed = 0;
static int test_destructed = 0;
class Test : public nb::intrusive_base {
public:
Test() { test_constructed++; }
virtual ~Test() { test_destructed++; }
virtual int value(int i) const { return 123 + i; }
static Test *create_raw() { return new Test(); }
static nb::ref<Test> create_ref() { return new Test(); }
};
class PyTest : Test {
NB_TRAMPOLINE(Test, 1);
virtual int value(int i) const {
NB_OVERRIDE(value, i);
}
};
NB_MODULE(test_intrusive_ext, m) {
nb::intrusive_init(
[](PyObject *o) noexcept {
nb::gil_scoped_acquire guard;
Py_INCREF(o);
},
[](PyObject *o) noexcept {
nb::gil_scoped_acquire guard;
Py_DECREF(o);
});
nb::class_<nb::intrusive_base>(
m, "intrusive_base",
nb::intrusive_ptr<nb::intrusive_base>(
[](nb::intrusive_base *o, PyObject *po) noexcept { o->set_self_py(po); }));
nb::class_<Test, nb::intrusive_base, PyTest>(m, "Test")
.def(nb::init<>())
.def("value", &Test::value)
.def_static("create_raw", &Test::create_raw)
.def_static("create_ref", &Test::create_ref);
m.def("reset", [] {
test_constructed = 0;
test_destructed = 0;
});
m.def("stats", []() -> std::pair<int, int> {
return { test_constructed, test_destructed };
});
m.def("get_value_1", [](Test *o) { nb::ref<Test> x(o); return x->value(1); });
m.def("get_value_2", [](nb::ref<Test> x) { return x->value(2); });
m.def("get_value_3", [](const nb::ref<Test> &x) { return x->value(3); });
}
|
