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
67
|
#include "pybind11_utils/shared_cpp_owner.h"
#include <pybind11/pybind11.h>
#include <tuple>
#include <unordered_map>
namespace py = pybind11;
using namespace pybind11::literals;
class Base;
static std::unordered_map<std::string, Base*> bases;
class Base {
std::string name_;
public:
Base(std::string const& name) : name_{name} { bases[name] = this; }
virtual std::string fn() const = 0;
virtual ~Base() { bases.erase(name_); }
};
MO2_PYBIND11_SHARED_CPP_HOLDER(Base);
class CppBase : public Base {
public:
using Base::Base;
std::string fn() const override { return "CppBase::fn()"; }
};
class PyBase : public Base {
public:
using Base::Base;
std::string fn() const override { PYBIND11_OVERRIDE_PURE(std::string, Base, fn, ); }
};
PYBIND11_MODULE(shared_cpp_owner, m)
{
static std::shared_ptr<Base> base_ptr;
py::class_<Base, PyBase, std::shared_ptr<Base>>(m, "Base")
.def(py::init<std::string>())
.def("fn", &Base::fn);
m.def("is_alive", [](std::string const& name) {
return bases.find(name) != bases.end();
});
m.def("create", [](std::string const& name) -> std::shared_ptr<Base> {
return std::make_shared<CppBase>(name);
});
m.def("create_and_store", [](std::string const& name) {
base_ptr = std::make_shared<CppBase>(name);
return base_ptr;
});
m.def("store", [](std::shared_ptr<Base> ptr) {
base_ptr = ptr;
});
m.def("clear", []() {
base_ptr.reset();
});
m.def("call_fn", [](std::string const& name) {
auto it = bases.find(name);
return it != bases.end() ? it->second->fn() : "";
});
}
|
