forked from pybind/python_example
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.cpp
More file actions
88 lines (63 loc) · 1.91 KB
/
main.cpp
File metadata and controls
88 lines (63 loc) · 1.91 KB
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#include <pybind11/pybind11.h>
#define STRINGIFY(x) #x
#define MACRO_STRINGIFY(x) STRINGIFY(x)
class Foo {
public:
Foo() {}
std::shared_ptr<Foo> munge() {
return
std::shared_ptr<Foo>(this);
}
int munge3() {
return 2;
}
std::shared_ptr<Foo> munge2(std::shared_ptr<Foo> f) {
return std::shared_ptr<Foo>(this);
}
};
int add(int i, int j) {
return i + j;
}
namespace py = pybind11;
autotype(Foo, fooinfo)
autotype(Foo& (*)(Foo*, Foo&), foofunc2)
autotype(Foo& (*)(Foo*), foof3unc2)
autotype(std::shared_ptr<Foo> (*)(Foo*, std::shared_ptr<Foo>), fooconver)
autotype(std::shared_ptr<Foo> (*)(Foo*), fooconvert3)
autotype(int (*)(Foo*),fooint)
#define autotype3(X,Y) \
const mtype_info Y ( #X ); \
template<> \
constexpr const mtype_info& mtypeid1< X >(){ \
return Y; \
} \
autotype(std::shared_ptr<Foo>, fooconvert2)
PYBIND11_MODULE(python_example, m) {
m.doc() = R"pbdoc(
Pybind11 example plugin
-----------------------
.. currentmodule:: python_example
.. autosummary::
:toctree: _generate
add
subtract
)pbdoc";
py::class_<Foo> foo(m,"Foo");
foo.def(py::init<>());
foo.def("munge",&Foo::munge, py::return_value_policy::reference_internal);
foo.def("munge3",&Foo::munge3);
foo.def("munge2",&Foo::munge2, py::return_value_policy::reference_internal);
m.def("add", &add, R"pbdoc(
Add two numbers
Some other explanation about the add function.
)pbdoc");
m.def("subtract", [](int i, int j) { return i - j; }, R"pbdoc(
Subtract two numbers
Some other explanation about the subtract function.
)pbdoc");
#ifdef VERSION_INFO
m.attr("__version__") = MACRO_STRINGIFY(VERSION_INFO);
#else
m.attr("__version__") = "dev";
#endif
}