Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1 | .. _advanced: |
| 2 | |
| 3 | Advanced topics |
| 4 | ############### |
| 5 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 6 | For brevity, the rest of this chapter assumes that the following two lines are |
| 7 | present: |
| 8 | |
| 9 | .. code-block:: cpp |
| 10 | |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 11 | #include <pybind11/pybind11.h> |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 12 | |
Wenzel Jakob | 10e62e1 | 2015-10-15 22:46:07 +0200 | [diff] [blame] | 13 | namespace py = pybind11; |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 14 | |
Wenzel Jakob | de3ad07 | 2016-02-02 11:38:21 +0100 | [diff] [blame] | 15 | Exporting constants and mutable objects |
| 16 | ======================================= |
| 17 | |
| 18 | To expose a C++ constant, use the ``attr`` function to register it in a module |
| 19 | as shown below. The ``int_`` class is one of many small wrapper objects defined |
| 20 | in ``pybind11/pytypes.h``. General objects (including integers) can also be |
| 21 | converted using the function ``cast``. |
| 22 | |
| 23 | .. code-block:: cpp |
| 24 | |
| 25 | PYBIND11_PLUGIN(example) { |
| 26 | py::module m("example", "pybind11 example plugin"); |
| 27 | m.attr("MY_CONSTANT") = py::int_(123); |
| 28 | m.attr("MY_CONSTANT_2") = py::cast(new MyObject()); |
| 29 | } |
| 30 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 31 | Operator overloading |
| 32 | ==================== |
| 33 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 34 | Suppose that we're given the following ``Vector2`` class with a vector addition |
| 35 | and scalar multiplication operation, all implemented using overloaded operators |
| 36 | in C++. |
| 37 | |
| 38 | .. code-block:: cpp |
| 39 | |
| 40 | class Vector2 { |
| 41 | public: |
| 42 | Vector2(float x, float y) : x(x), y(y) { } |
| 43 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 44 | Vector2 operator+(const Vector2 &v) const { return Vector2(x + v.x, y + v.y); } |
| 45 | Vector2 operator*(float value) const { return Vector2(x * value, y * value); } |
| 46 | Vector2& operator+=(const Vector2 &v) { x += v.x; y += v.y; return *this; } |
| 47 | Vector2& operator*=(float v) { x *= v; y *= v; return *this; } |
| 48 | |
Wenzel Jakob | f64feaf | 2016-04-28 14:33:45 +0200 | [diff] [blame] | 49 | friend Vector2 operator*(float f, const Vector2 &v) { |
| 50 | return Vector2(f * v.x, f * v.y); |
| 51 | } |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 52 | |
Wenzel Jakob | f64feaf | 2016-04-28 14:33:45 +0200 | [diff] [blame] | 53 | std::string toString() const { |
| 54 | return "[" + std::to_string(x) + ", " + std::to_string(y) + "]"; |
| 55 | } |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 56 | private: |
| 57 | float x, y; |
| 58 | }; |
| 59 | |
| 60 | The following snippet shows how the above operators can be conveniently exposed |
| 61 | to Python. |
| 62 | |
| 63 | .. code-block:: cpp |
| 64 | |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 65 | #include <pybind11/operators.h> |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 66 | |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 67 | PYBIND11_PLUGIN(example) { |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 68 | py::module m("example", "pybind11 example plugin"); |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 69 | |
| 70 | py::class_<Vector2>(m, "Vector2") |
| 71 | .def(py::init<float, float>()) |
| 72 | .def(py::self + py::self) |
| 73 | .def(py::self += py::self) |
| 74 | .def(py::self *= float()) |
| 75 | .def(float() * py::self) |
| 76 | .def("__repr__", &Vector2::toString); |
| 77 | |
| 78 | return m.ptr(); |
| 79 | } |
| 80 | |
| 81 | Note that a line like |
| 82 | |
| 83 | .. code-block:: cpp |
| 84 | |
| 85 | .def(py::self * float()) |
| 86 | |
| 87 | is really just short hand notation for |
| 88 | |
| 89 | .. code-block:: cpp |
| 90 | |
| 91 | .def("__mul__", [](const Vector2 &a, float b) { |
| 92 | return a * b; |
| 93 | }) |
| 94 | |
| 95 | This can be useful for exposing additional operators that don't exist on the |
| 96 | C++ side, or to perform other types of customization. |
| 97 | |
| 98 | .. note:: |
| 99 | |
| 100 | To use the more convenient ``py::self`` notation, the additional |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 101 | header file :file:`pybind11/operators.h` must be included. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 102 | |
| 103 | .. seealso:: |
| 104 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 105 | The file :file:`tests/test_operator_overloading.cpp` contains a |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 106 | complete example that demonstrates how to work with overloaded operators in |
| 107 | more detail. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 108 | |
| 109 | Callbacks and passing anonymous functions |
| 110 | ========================================= |
| 111 | |
| 112 | The C++11 standard brought lambda functions and the generic polymorphic |
| 113 | function wrapper ``std::function<>`` to the C++ programming language, which |
| 114 | enable powerful new ways of working with functions. Lambda functions come in |
| 115 | two flavors: stateless lambda function resemble classic function pointers that |
| 116 | link to an anonymous piece of code, while stateful lambda functions |
| 117 | additionally depend on captured variables that are stored in an anonymous |
| 118 | *lambda closure object*. |
| 119 | |
| 120 | Here is a simple example of a C++ function that takes an arbitrary function |
| 121 | (stateful or stateless) with signature ``int -> int`` as an argument and runs |
| 122 | it with the value 10. |
| 123 | |
| 124 | .. code-block:: cpp |
| 125 | |
| 126 | int func_arg(const std::function<int(int)> &f) { |
| 127 | return f(10); |
| 128 | } |
| 129 | |
| 130 | The example below is more involved: it takes a function of signature ``int -> int`` |
| 131 | and returns another function of the same kind. The return value is a stateful |
| 132 | lambda function, which stores the value ``f`` in the capture object and adds 1 to |
| 133 | its return value upon execution. |
| 134 | |
| 135 | .. code-block:: cpp |
| 136 | |
| 137 | std::function<int(int)> func_ret(const std::function<int(int)> &f) { |
| 138 | return [f](int i) { |
| 139 | return f(i) + 1; |
| 140 | }; |
| 141 | } |
| 142 | |
Brad Harmon | 835fc06 | 2016-06-16 13:19:15 -0500 | [diff] [blame] | 143 | This example demonstrates using python named parameters in C++ callbacks which |
| 144 | requires using ``py::cpp_function`` as a wrapper. Usage is similar to defining |
| 145 | methods of classes: |
| 146 | |
| 147 | .. code-block:: cpp |
| 148 | |
| 149 | py::cpp_function func_cpp() { |
| 150 | return py::cpp_function([](int i) { return i+1; }, |
| 151 | py::arg("number")); |
| 152 | } |
| 153 | |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 154 | After including the extra header file :file:`pybind11/functional.h`, it is almost |
Brad Harmon | 835fc06 | 2016-06-16 13:19:15 -0500 | [diff] [blame] | 155 | trivial to generate binding code for all of these functions. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 156 | |
| 157 | .. code-block:: cpp |
| 158 | |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 159 | #include <pybind11/functional.h> |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 160 | |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 161 | PYBIND11_PLUGIN(example) { |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 162 | py::module m("example", "pybind11 example plugin"); |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 163 | |
| 164 | m.def("func_arg", &func_arg); |
| 165 | m.def("func_ret", &func_ret); |
Brad Harmon | 835fc06 | 2016-06-16 13:19:15 -0500 | [diff] [blame] | 166 | m.def("func_cpp", &func_cpp); |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 167 | |
| 168 | return m.ptr(); |
| 169 | } |
| 170 | |
| 171 | The following interactive session shows how to call them from Python. |
| 172 | |
Wenzel Jakob | 99279f7 | 2016-06-03 11:19:29 +0200 | [diff] [blame] | 173 | .. code-block:: pycon |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 174 | |
| 175 | $ python |
| 176 | >>> import example |
| 177 | >>> def square(i): |
| 178 | ... return i * i |
| 179 | ... |
| 180 | >>> example.func_arg(square) |
| 181 | 100L |
| 182 | >>> square_plus_1 = example.func_ret(square) |
| 183 | >>> square_plus_1(4) |
| 184 | 17L |
Brad Harmon | 835fc06 | 2016-06-16 13:19:15 -0500 | [diff] [blame] | 185 | >>> plus_1 = func_cpp() |
| 186 | >>> plus_1(number=43) |
| 187 | 44L |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 188 | |
Wenzel Jakob | a4175d6 | 2015-11-17 08:30:34 +0100 | [diff] [blame] | 189 | .. warning:: |
| 190 | |
| 191 | Keep in mind that passing a function from C++ to Python (or vice versa) |
| 192 | will instantiate a piece of wrapper code that translates function |
Wenzel Jakob | 954b793 | 2016-07-10 10:13:18 +0200 | [diff] [blame] | 193 | invocations between the two languages. Naturally, this translation |
| 194 | increases the computational cost of each function call somewhat. A |
| 195 | problematic situation can arise when a function is copied back and forth |
| 196 | between Python and C++ many times in a row, in which case the underlying |
| 197 | wrappers will accumulate correspondingly. The resulting long sequence of |
| 198 | C++ -> Python -> C++ -> ... roundtrips can significantly decrease |
| 199 | performance. |
| 200 | |
| 201 | There is one exception: pybind11 detects case where a stateless function |
| 202 | (i.e. a function pointer or a lambda function without captured variables) |
| 203 | is passed as an argument to another C++ function exposed in Python. In this |
| 204 | case, there is no overhead. Pybind11 will extract the underlying C++ |
| 205 | function pointer from the wrapped function to sidestep a potential C++ -> |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 206 | Python -> C++ roundtrip. This is demonstrated in :file:`tests/test_callbacks.cpp`. |
Wenzel Jakob | 954b793 | 2016-07-10 10:13:18 +0200 | [diff] [blame] | 207 | |
| 208 | .. note:: |
| 209 | |
| 210 | This functionality is very useful when generating bindings for callbacks in |
| 211 | C++ libraries (e.g. GUI libraries, asynchronous networking libraries, etc.). |
| 212 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 213 | The file :file:`tests/test_callbacks.cpp` contains a complete example |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 214 | that demonstrates how to work with callbacks and anonymous functions in |
| 215 | more detail. |
Wenzel Jakob | a4175d6 | 2015-11-17 08:30:34 +0100 | [diff] [blame] | 216 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 217 | Overriding virtual functions in Python |
| 218 | ====================================== |
| 219 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 220 | Suppose that a C++ class or interface has a virtual function that we'd like to |
| 221 | to override from within Python (we'll focus on the class ``Animal``; ``Dog`` is |
| 222 | given as a specific example of how one would do this with traditional C++ |
| 223 | code). |
| 224 | |
| 225 | .. code-block:: cpp |
| 226 | |
| 227 | class Animal { |
| 228 | public: |
| 229 | virtual ~Animal() { } |
| 230 | virtual std::string go(int n_times) = 0; |
| 231 | }; |
| 232 | |
| 233 | class Dog : public Animal { |
| 234 | public: |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 235 | std::string go(int n_times) override { |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 236 | std::string result; |
| 237 | for (int i=0; i<n_times; ++i) |
| 238 | result += "woof! "; |
| 239 | return result; |
| 240 | } |
| 241 | }; |
| 242 | |
| 243 | Let's also suppose that we are given a plain function which calls the |
| 244 | function ``go()`` on an arbitrary ``Animal`` instance. |
| 245 | |
| 246 | .. code-block:: cpp |
| 247 | |
| 248 | std::string call_go(Animal *animal) { |
| 249 | return animal->go(3); |
| 250 | } |
| 251 | |
| 252 | Normally, the binding code for these classes would look as follows: |
| 253 | |
| 254 | .. code-block:: cpp |
| 255 | |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 256 | PYBIND11_PLUGIN(example) { |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 257 | py::module m("example", "pybind11 example plugin"); |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 258 | |
| 259 | py::class_<Animal> animal(m, "Animal"); |
| 260 | animal |
| 261 | .def("go", &Animal::go); |
| 262 | |
| 263 | py::class_<Dog>(m, "Dog", animal) |
| 264 | .def(py::init<>()); |
| 265 | |
| 266 | m.def("call_go", &call_go); |
| 267 | |
| 268 | return m.ptr(); |
| 269 | } |
| 270 | |
| 271 | However, these bindings are impossible to extend: ``Animal`` is not |
| 272 | constructible, and we clearly require some kind of "trampoline" that |
| 273 | redirects virtual calls back to Python. |
| 274 | |
| 275 | Defining a new type of ``Animal`` from within Python is possible but requires a |
| 276 | helper class that is defined as follows: |
| 277 | |
| 278 | .. code-block:: cpp |
| 279 | |
| 280 | class PyAnimal : public Animal { |
| 281 | public: |
| 282 | /* Inherit the constructors */ |
| 283 | using Animal::Animal; |
| 284 | |
| 285 | /* Trampoline (need one for each virtual function) */ |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 286 | std::string go(int n_times) override { |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 287 | PYBIND11_OVERLOAD_PURE( |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 288 | std::string, /* Return type */ |
| 289 | Animal, /* Parent class */ |
| 290 | go, /* Name of function */ |
| 291 | n_times /* Argument(s) */ |
| 292 | ); |
| 293 | } |
| 294 | }; |
| 295 | |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 296 | The macro :func:`PYBIND11_OVERLOAD_PURE` should be used for pure virtual |
| 297 | functions, and :func:`PYBIND11_OVERLOAD` should be used for functions which have |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 298 | a default implementation. |
Wenzel Jakob | 1e3be73 | 2016-05-24 23:42:05 +0200 | [diff] [blame] | 299 | |
| 300 | There are also two alternate macros :func:`PYBIND11_OVERLOAD_PURE_NAME` and |
| 301 | :func:`PYBIND11_OVERLOAD_NAME` which take a string-valued name argument |
| 302 | after the *Name of the function* slot. This is useful when the C++ and Python |
| 303 | versions of the function have different names, e.g. ``operator()`` vs ``__call__``. |
| 304 | |
| 305 | The binding code also needs a few minor adaptations (highlighted): |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 306 | |
| 307 | .. code-block:: cpp |
| 308 | :emphasize-lines: 4,6,7 |
| 309 | |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 310 | PYBIND11_PLUGIN(example) { |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 311 | py::module m("example", "pybind11 example plugin"); |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 312 | |
Wenzel Jakob | 86d825f | 2016-05-26 13:19:27 +0200 | [diff] [blame] | 313 | py::class_<Animal, std::unique_ptr<Animal>, PyAnimal /* <--- trampoline*/> animal(m, "Animal"); |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 314 | animal |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 315 | .def(py::init<>()) |
| 316 | .def("go", &Animal::go); |
| 317 | |
| 318 | py::class_<Dog>(m, "Dog", animal) |
| 319 | .def(py::init<>()); |
| 320 | |
| 321 | m.def("call_go", &call_go); |
| 322 | |
| 323 | return m.ptr(); |
| 324 | } |
| 325 | |
Wenzel Jakob | 86d825f | 2016-05-26 13:19:27 +0200 | [diff] [blame] | 326 | Importantly, pybind11 is made aware of the trampoline trampoline helper class |
| 327 | by specifying it as the *third* template argument to :class:`class_`. The |
| 328 | second argument with the unique pointer is simply the default holder type used |
| 329 | by pybind11. Following this, we are able to define a constructor as usual. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 330 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 331 | Note, however, that the above is sufficient for allowing python classes to |
| 332 | extend ``Animal``, but not ``Dog``: see ref:`virtual_and_inheritance` for the |
| 333 | necessary steps required to providing proper overload support for inherited |
| 334 | classes. |
| 335 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 336 | The Python session below shows how to override ``Animal::go`` and invoke it via |
| 337 | a virtual method call. |
| 338 | |
Wenzel Jakob | 99279f7 | 2016-06-03 11:19:29 +0200 | [diff] [blame] | 339 | .. code-block:: pycon |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 340 | |
| 341 | >>> from example import * |
| 342 | >>> d = Dog() |
| 343 | >>> call_go(d) |
| 344 | u'woof! woof! woof! ' |
| 345 | >>> class Cat(Animal): |
| 346 | ... def go(self, n_times): |
| 347 | ... return "meow! " * n_times |
| 348 | ... |
| 349 | >>> c = Cat() |
| 350 | >>> call_go(c) |
| 351 | u'meow! meow! meow! ' |
| 352 | |
Wenzel Jakob | 9bb97c1 | 2016-06-03 11:19:41 +0200 | [diff] [blame] | 353 | Please take a look at the :ref:`macro_notes` before using this feature. |
Wenzel Jakob | bd986fe | 2016-05-21 10:48:30 +0200 | [diff] [blame] | 354 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 355 | .. seealso:: |
| 356 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 357 | The file :file:`tests/test_virtual_functions.cpp` contains a complete |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 358 | example that demonstrates how to override virtual functions using pybind11 |
| 359 | in more detail. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 360 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 361 | .. _virtual_and_inheritance: |
| 362 | |
| 363 | Combining virtual functions and inheritance |
| 364 | =========================================== |
| 365 | |
| 366 | When combining virtual methods with inheritance, you need to be sure to provide |
| 367 | an override for each method for which you want to allow overrides from derived |
| 368 | python classes. For example, suppose we extend the above ``Animal``/``Dog`` |
| 369 | example as follows: |
| 370 | |
| 371 | .. code-block:: cpp |
Dean Moldovan | aebca12 | 2016-08-16 01:26:02 +0200 | [diff] [blame] | 372 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 373 | class Animal { |
| 374 | public: |
| 375 | virtual std::string go(int n_times) = 0; |
| 376 | virtual std::string name() { return "unknown"; } |
| 377 | }; |
| 378 | class Dog : public class Animal { |
| 379 | public: |
| 380 | std::string go(int n_times) override { |
| 381 | std::string result; |
| 382 | for (int i=0; i<n_times; ++i) |
| 383 | result += bark() + " "; |
| 384 | return result; |
| 385 | } |
| 386 | virtual std::string bark() { return "woof!"; } |
| 387 | }; |
| 388 | |
| 389 | then the trampoline class for ``Animal`` must, as described in the previous |
| 390 | section, override ``go()`` and ``name()``, but in order to allow python code to |
| 391 | inherit properly from ``Dog``, we also need a trampoline class for ``Dog`` that |
| 392 | overrides both the added ``bark()`` method *and* the ``go()`` and ``name()`` |
| 393 | methods inherited from ``Animal`` (even though ``Dog`` doesn't directly |
| 394 | override the ``name()`` method): |
| 395 | |
| 396 | .. code-block:: cpp |
Dean Moldovan | aebca12 | 2016-08-16 01:26:02 +0200 | [diff] [blame] | 397 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 398 | class PyAnimal : public Animal { |
| 399 | public: |
| 400 | using Animal::Animal; // Inherit constructors |
| 401 | std::string go(int n_times) override { PYBIND11_OVERLOAD_PURE(std::string, Animal, go, n_times); } |
| 402 | std::string name() override { PYBIND11_OVERLOAD(std::string, Animal, name, ); } |
| 403 | }; |
| 404 | class PyDog : public Dog { |
| 405 | public: |
| 406 | using Dog::Dog; // Inherit constructors |
| 407 | std::string go(int n_times) override { PYBIND11_OVERLOAD_PURE(std::string, Dog, go, n_times); } |
| 408 | std::string name() override { PYBIND11_OVERLOAD(std::string, Dog, name, ); } |
| 409 | std::string bark() override { PYBIND11_OVERLOAD(std::string, Dog, bark, ); } |
| 410 | }; |
| 411 | |
| 412 | A registered class derived from a pybind11-registered class with virtual |
| 413 | methods requires a similar trampoline class, *even if* it doesn't explicitly |
| 414 | declare or override any virtual methods itself: |
| 415 | |
| 416 | .. code-block:: cpp |
Dean Moldovan | aebca12 | 2016-08-16 01:26:02 +0200 | [diff] [blame] | 417 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 418 | class Husky : public Dog {}; |
| 419 | class PyHusky : public Husky { |
| 420 | using Dog::Dog; // Inherit constructors |
| 421 | std::string go(int n_times) override { PYBIND11_OVERLOAD_PURE(std::string, Husky, go, n_times); } |
| 422 | std::string name() override { PYBIND11_OVERLOAD(std::string, Husky, name, ); } |
| 423 | std::string bark() override { PYBIND11_OVERLOAD(std::string, Husky, bark, ); } |
| 424 | }; |
| 425 | |
| 426 | There is, however, a technique that can be used to avoid this duplication |
| 427 | (which can be especially helpful for a base class with several virtual |
| 428 | methods). The technique involves using template trampoline classes, as |
| 429 | follows: |
| 430 | |
| 431 | .. code-block:: cpp |
Dean Moldovan | aebca12 | 2016-08-16 01:26:02 +0200 | [diff] [blame] | 432 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 433 | template <class AnimalBase = Animal> class PyAnimal : public AnimalBase { |
| 434 | using AnimalBase::AnimalBase; // Inherit constructors |
| 435 | std::string go(int n_times) override { PYBIND11_OVERLOAD_PURE(std::string, AnimalBase, go, n_times); } |
| 436 | std::string name() override { PYBIND11_OVERLOAD(std::string, AnimalBase, name, ); } |
| 437 | }; |
| 438 | template <class DogBase = Dog> class PyDog : public PyAnimal<DogBase> { |
| 439 | using PyAnimal<DogBase>::PyAnimal; // Inherit constructors |
| 440 | // Override PyAnimal's pure virtual go() with a non-pure one: |
| 441 | std::string go(int n_times) override { PYBIND11_OVERLOAD(std::string, DogBase, go, n_times); } |
| 442 | std::string bark() override { PYBIND11_OVERLOAD(std::string, DogBase, bark, ); } |
| 443 | }; |
| 444 | |
| 445 | This technique has the advantage of requiring just one trampoline method to be |
| 446 | declared per virtual method and pure virtual method override. It does, |
| 447 | however, require the compiler to generate at least as many methods (and |
| 448 | possibly more, if both pure virtual and overridden pure virtual methods are |
| 449 | exposed, as above). |
| 450 | |
| 451 | The classes are then registered with pybind11 using: |
| 452 | |
| 453 | .. code-block:: cpp |
Dean Moldovan | aebca12 | 2016-08-16 01:26:02 +0200 | [diff] [blame] | 454 | |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 455 | py::class_<Animal, std::unique_ptr<Animal>, PyAnimal<>> animal(m, "Animal"); |
| 456 | py::class_<Dog, std::unique_ptr<Dog>, PyDog<>> dog(m, "Dog"); |
| 457 | py::class_<Husky, std::unique_ptr<Husky>, PyDog<Husky>> husky(m, "Husky"); |
| 458 | // ... add animal, dog, husky definitions |
| 459 | |
| 460 | Note that ``Husky`` did not require a dedicated trampoline template class at |
| 461 | all, since it neither declares any new virtual methods nor provides any pure |
| 462 | virtual method implementations. |
| 463 | |
| 464 | With either the repeated-virtuals or templated trampoline methods in place, you |
| 465 | can now create a python class that inherits from ``Dog``: |
| 466 | |
| 467 | .. code-block:: python |
| 468 | |
| 469 | class ShihTzu(Dog): |
| 470 | def bark(self): |
| 471 | return "yip!" |
| 472 | |
| 473 | .. seealso:: |
| 474 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 475 | See the file :file:`tests/test_virtual_functions.cpp` for complete examples |
Jason Rhinelander | 0ca96e2 | 2016-08-05 17:02:33 -0400 | [diff] [blame] | 476 | using both the duplication and templated trampoline approaches. |
| 477 | |
Wenzel Jakob | 9bb97c1 | 2016-06-03 11:19:41 +0200 | [diff] [blame] | 478 | .. _macro_notes: |
| 479 | |
| 480 | General notes regarding convenience macros |
| 481 | ========================================== |
| 482 | |
| 483 | pybind11 provides a few convenience macros such as |
| 484 | :func:`PYBIND11_MAKE_OPAQUE` and :func:`PYBIND11_DECLARE_HOLDER_TYPE`, and |
| 485 | ``PYBIND11_OVERLOAD_*``. Since these are "just" macros that are evaluated |
| 486 | in the preprocessor (which has no concept of types), they *will* get confused |
| 487 | by commas in a template argument such as ``PYBIND11_OVERLOAD(MyReturnValue<T1, |
| 488 | T2>, myFunc)``. In this case, the preprocessor assumes that the comma indicates |
| 489 | the beginnning of the next parameter. Use a ``typedef`` to bind the template to |
| 490 | another name and use it in the macro to avoid this problem. |
| 491 | |
| 492 | |
Wenzel Jakob | ecdd868 | 2015-12-07 18:17:58 +0100 | [diff] [blame] | 493 | Global Interpreter Lock (GIL) |
| 494 | ============================= |
| 495 | |
| 496 | The classes :class:`gil_scoped_release` and :class:`gil_scoped_acquire` can be |
| 497 | used to acquire and release the global interpreter lock in the body of a C++ |
| 498 | function call. In this way, long-running C++ code can be parallelized using |
| 499 | multiple Python threads. Taking the previous section as an example, this could |
| 500 | be realized as follows (important changes highlighted): |
| 501 | |
| 502 | .. code-block:: cpp |
| 503 | :emphasize-lines: 8,9,33,34 |
| 504 | |
| 505 | class PyAnimal : public Animal { |
| 506 | public: |
| 507 | /* Inherit the constructors */ |
| 508 | using Animal::Animal; |
| 509 | |
| 510 | /* Trampoline (need one for each virtual function) */ |
| 511 | std::string go(int n_times) { |
| 512 | /* Acquire GIL before calling Python code */ |
Wenzel Jakob | a4caa85 | 2015-12-14 12:39:02 +0100 | [diff] [blame] | 513 | py::gil_scoped_acquire acquire; |
Wenzel Jakob | ecdd868 | 2015-12-07 18:17:58 +0100 | [diff] [blame] | 514 | |
| 515 | PYBIND11_OVERLOAD_PURE( |
| 516 | std::string, /* Return type */ |
| 517 | Animal, /* Parent class */ |
| 518 | go, /* Name of function */ |
| 519 | n_times /* Argument(s) */ |
| 520 | ); |
| 521 | } |
| 522 | }; |
| 523 | |
| 524 | PYBIND11_PLUGIN(example) { |
| 525 | py::module m("example", "pybind11 example plugin"); |
| 526 | |
Wenzel Jakob | 86d825f | 2016-05-26 13:19:27 +0200 | [diff] [blame] | 527 | py::class_<Animal, std::unique_ptr<Animal>, PyAnimal> animal(m, "Animal"); |
Wenzel Jakob | ecdd868 | 2015-12-07 18:17:58 +0100 | [diff] [blame] | 528 | animal |
Wenzel Jakob | ecdd868 | 2015-12-07 18:17:58 +0100 | [diff] [blame] | 529 | .def(py::init<>()) |
| 530 | .def("go", &Animal::go); |
| 531 | |
| 532 | py::class_<Dog>(m, "Dog", animal) |
| 533 | .def(py::init<>()); |
| 534 | |
| 535 | m.def("call_go", [](Animal *animal) -> std::string { |
| 536 | /* Release GIL before calling into (potentially long-running) C++ code */ |
Wenzel Jakob | a4caa85 | 2015-12-14 12:39:02 +0100 | [diff] [blame] | 537 | py::gil_scoped_release release; |
Wenzel Jakob | ecdd868 | 2015-12-07 18:17:58 +0100 | [diff] [blame] | 538 | return call_go(animal); |
| 539 | }); |
| 540 | |
| 541 | return m.ptr(); |
| 542 | } |
| 543 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 544 | Passing STL data structures |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 545 | =========================== |
| 546 | |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 547 | When including the additional header file :file:`pybind11/stl.h`, conversions |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 548 | between ``std::vector<>``, ``std::list<>``, ``std::set<>``, and ``std::map<>`` |
| 549 | and the Python ``list``, ``set`` and ``dict`` data structures are automatically |
| 550 | enabled. The types ``std::pair<>`` and ``std::tuple<>`` are already supported |
| 551 | out of the box with just the core :file:`pybind11/pybind11.h` header. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 552 | |
| 553 | .. note:: |
| 554 | |
Wenzel Jakob | 44db04f | 2015-12-14 12:40:45 +0100 | [diff] [blame] | 555 | Arbitrary nesting of any of these types is supported. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 556 | |
| 557 | .. seealso:: |
| 558 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 559 | The file :file:`tests/test_python_types.cpp` contains a complete |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 560 | example that demonstrates how to pass STL data types in more detail. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 561 | |
Wenzel Jakob | b282595 | 2016-04-13 23:33:00 +0200 | [diff] [blame] | 562 | Binding sequence data types, iterators, the slicing protocol, etc. |
| 563 | ================================================================== |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 564 | |
| 565 | Please refer to the supplemental example for details. |
| 566 | |
| 567 | .. seealso:: |
| 568 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 569 | The file :file:`tests/test_sequences_and_iterators.cpp` contains a |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 570 | complete example that shows how to bind a sequence data type, including |
| 571 | length queries (``__len__``), iterators (``__iter__``), the slicing |
| 572 | protocol and other kinds of useful operations. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 573 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 574 | Return value policies |
| 575 | ===================== |
| 576 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 577 | Python and C++ use wildly different ways of managing the memory and lifetime of |
| 578 | objects managed by them. This can lead to issues when creating bindings for |
| 579 | functions that return a non-trivial type. Just by looking at the type |
| 580 | information, it is not clear whether Python should take charge of the returned |
| 581 | value and eventually free its resources, or if this is handled on the C++ side. |
| 582 | For this reason, pybind11 provides a several `return value policy` annotations |
| 583 | that can be passed to the :func:`module::def` and :func:`class_::def` |
Wenzel Jakob | 61d67f0 | 2015-12-14 12:53:06 +0100 | [diff] [blame] | 584 | functions. The default policy is :enum:`return_value_policy::automatic`. |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 585 | |
Wenzel Jakob | f64feaf | 2016-04-28 14:33:45 +0200 | [diff] [blame] | 586 | .. tabularcolumns:: |p{0.5\textwidth}|p{0.45\textwidth}| |
| 587 | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 588 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
| 589 | | Return value policy | Description | |
| 590 | +==================================================+============================================================================+ |
| 591 | | :enum:`return_value_policy::automatic` | This is the default return value policy, which falls back to the policy | |
| 592 | | | :enum:`return_value_policy::take_ownership` when the return value is a | |
Wenzel Jakob | e84f557 | 2016-04-26 23:19:19 +0200 | [diff] [blame] | 593 | | | pointer. Otherwise, it uses :enum:`return_value::move` or | |
| 594 | | | :enum:`return_value::copy` for rvalue and lvalue references, respectively. | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 595 | | | See below for a description of what all of these different policies do. | |
| 596 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
| 597 | | :enum:`return_value_policy::automatic_reference` | As above, but use policy :enum:`return_value_policy::reference` when the | |
Wenzel Jakob | 37e1f61 | 2016-06-22 14:29:13 +0200 | [diff] [blame] | 598 | | | return value is a pointer. This is the default conversion policy for | |
| 599 | | | function arguments when calling Python functions manually from C++ code | |
| 600 | | | (i.e. via handle::operator()). You probably won't need to use this. | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 601 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
| 602 | | :enum:`return_value_policy::take_ownership` | Reference an existing object (i.e. do not create a new copy) and take | |
| 603 | | | ownership. Python will call the destructor and delete operator when the | |
| 604 | | | object's reference count reaches zero. Undefined behavior ensues when the | |
Wenzel Jakob | f53e300 | 2016-06-30 14:59:23 +0200 | [diff] [blame] | 605 | | | C++ side does the same. | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 606 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
| 607 | | :enum:`return_value_policy::copy` | Create a new copy of the returned object, which will be owned by Python. | |
| 608 | | | This policy is comparably safe because the lifetimes of the two instances | |
| 609 | | | are decoupled. | |
| 610 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
| 611 | | :enum:`return_value_policy::move` | Use ``std::move`` to move the return value contents into a new instance | |
| 612 | | | that will be owned by Python. This policy is comparably safe because the | |
| 613 | | | lifetimes of the two instances (move source and destination) are decoupled.| |
| 614 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
| 615 | | :enum:`return_value_policy::reference` | Reference an existing object, but do not take ownership. The C++ side is | |
| 616 | | | responsible for managing the object's lifetime and deallocating it when | |
| 617 | | | it is no longer used. Warning: undefined behavior will ensue when the C++ | |
Wenzel Jakob | e84f557 | 2016-04-26 23:19:19 +0200 | [diff] [blame] | 618 | | | side deletes an object that is still referenced and used by Python. | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 619 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
Jason Rhinelander | f2ecd89 | 2016-08-10 12:08:04 -0400 | [diff] [blame] | 620 | | :enum:`return_value_policy::reference_internal` | Like :enum:`return_value_policy::reference` but additionally applies a | |
Dean Moldovan | aebca12 | 2016-08-16 01:26:02 +0200 | [diff] [blame] | 621 | | | ``keep_alive<0, 1>`` call policy (described next) that keeps the | |
Jason Rhinelander | f2ecd89 | 2016-08-10 12:08:04 -0400 | [diff] [blame] | 622 | | | ``this`` argument of the function or property from being garbage collected | |
| 623 | | | as long as the return value remains referenced. See the | |
| 624 | | | :class:`keep_alive` call policy (described next) for details. | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 625 | +--------------------------------------------------+----------------------------------------------------------------------------+ |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 626 | |
Wenzel Jakob | fb6aed2 | 2016-07-18 20:29:53 +0200 | [diff] [blame] | 627 | .. warning:: |
| 628 | |
Jason Rhinelander | efc2aa7 | 2016-08-10 11:38:33 -0400 | [diff] [blame] | 629 | Code with invalid return value policies might access unitialized memory or |
| 630 | free data structures multiple times, which can lead to hard-to-debug |
Wenzel Jakob | fb6aed2 | 2016-07-18 20:29:53 +0200 | [diff] [blame] | 631 | non-determinism and segmentation faults, hence it is worth spending the |
| 632 | time to understand all the different options in the table above. |
| 633 | |
Jason Rhinelander | efc2aa7 | 2016-08-10 11:38:33 -0400 | [diff] [blame] | 634 | One important aspect of the above policies is that they only apply to instances |
| 635 | which pybind11 has *not* seen before, in which case the policy clarifies |
| 636 | essential questions about the return value's lifetime and ownership. When |
| 637 | pybind11 knows the instance already (as identified by its type and address in |
Wenzel Jakob | fb6aed2 | 2016-07-18 20:29:53 +0200 | [diff] [blame] | 638 | memory), it will return the existing Python object wrapper rather than creating |
Jason Rhinelander | efc2aa7 | 2016-08-10 11:38:33 -0400 | [diff] [blame] | 639 | a copy. |
nafur | 717df75 | 2016-06-28 18:07:11 +0200 | [diff] [blame] | 640 | |
Wenzel Jakob | e84f557 | 2016-04-26 23:19:19 +0200 | [diff] [blame] | 641 | .. note:: |
| 642 | |
| 643 | The next section on :ref:`call_policies` discusses *call policies* that can be |
| 644 | specified *in addition* to a return value policy from the list above. Call |
| 645 | policies indicate reference relationships that can involve both return values |
| 646 | and parameters of functions. |
| 647 | |
| 648 | .. note:: |
| 649 | |
| 650 | As an alternative to elaborate call policies and lifetime management logic, |
| 651 | consider using smart pointers (see the section on :ref:`smart_pointers` for |
| 652 | details). Smart pointers can tell whether an object is still referenced from |
| 653 | C++ or Python, which generally eliminates the kinds of inconsistencies that |
| 654 | can lead to crashes or undefined behavior. For functions returning smart |
| 655 | pointers, it is not necessary to specify a return value policy. |
Wenzel Jakob | 5f218b3 | 2016-01-17 22:36:39 +0100 | [diff] [blame] | 656 | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 657 | .. _call_policies: |
| 658 | |
Wenzel Jakob | 5f218b3 | 2016-01-17 22:36:39 +0100 | [diff] [blame] | 659 | Additional call policies |
| 660 | ======================== |
| 661 | |
| 662 | In addition to the above return value policies, further `call policies` can be |
| 663 | specified to indicate dependencies between parameters. There is currently just |
| 664 | one policy named ``keep_alive<Nurse, Patient>``, which indicates that the |
| 665 | argument with index ``Patient`` should be kept alive at least until the |
Wenzel Jakob | 0b63231 | 2016-08-18 10:58:21 +0200 | [diff] [blame] | 666 | argument with index ``Nurse`` is freed by the garbage collector. Argument |
| 667 | indices start at one, while zero refers to the return value. For methods, index |
| 668 | ``1`` refers to the implicit ``this`` pointer, while regular arguments begin at |
| 669 | index ``2``. Arbitrarily many call policies can be specified. When a ``Nurse`` |
| 670 | with value ``None`` is detected at runtime, the call policy does nothing. |
Wenzel Jakob | 5f218b3 | 2016-01-17 22:36:39 +0100 | [diff] [blame] | 671 | |
Wenzel Jakob | 0b63231 | 2016-08-18 10:58:21 +0200 | [diff] [blame] | 672 | This feature internally relies on the ability to create a *weak reference* to |
| 673 | the nurse object, which is permitted by all classes exposed via pybind11. When |
| 674 | the nurse object does not support weak references, an exception will be thrown. |
| 675 | |
| 676 | Consider the following example: here, the binding code for a list append |
| 677 | operation ties the lifetime of the newly added element to the underlying |
| 678 | container: |
Wenzel Jakob | 5f218b3 | 2016-01-17 22:36:39 +0100 | [diff] [blame] | 679 | |
| 680 | .. code-block:: cpp |
| 681 | |
| 682 | py::class_<List>(m, "List") |
| 683 | .def("append", &List::append, py::keep_alive<1, 2>()); |
| 684 | |
| 685 | .. note:: |
| 686 | |
| 687 | ``keep_alive`` is analogous to the ``with_custodian_and_ward`` (if Nurse, |
| 688 | Patient != 0) and ``with_custodian_and_ward_postcall`` (if Nurse/Patient == |
| 689 | 0) policies from Boost.Python. |
| 690 | |
Wenzel Jakob | 6158716 | 2016-01-18 22:38:52 +0100 | [diff] [blame] | 691 | .. seealso:: |
| 692 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 693 | The file :file:`tests/test_keep_alive.cpp` contains a complete example |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 694 | that demonstrates using :class:`keep_alive` in more detail. |
Wenzel Jakob | 6158716 | 2016-01-18 22:38:52 +0100 | [diff] [blame] | 695 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 696 | Implicit type conversions |
| 697 | ========================= |
| 698 | |
| 699 | Suppose that instances of two types ``A`` and ``B`` are used in a project, and |
Wenzel Jakob | 8e93df8 | 2016-05-01 02:36:58 +0200 | [diff] [blame] | 700 | that an ``A`` can easily be converted into an instance of type ``B`` (examples of this |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 701 | could be a fixed and an arbitrary precision number type). |
| 702 | |
| 703 | .. code-block:: cpp |
| 704 | |
| 705 | py::class_<A>(m, "A") |
| 706 | /// ... members ... |
| 707 | |
| 708 | py::class_<B>(m, "B") |
| 709 | .def(py::init<A>()) |
| 710 | /// ... members ... |
| 711 | |
| 712 | m.def("func", |
| 713 | [](const B &) { /* .... */ } |
| 714 | ); |
| 715 | |
| 716 | To invoke the function ``func`` using a variable ``a`` containing an ``A`` |
| 717 | instance, we'd have to write ``func(B(a))`` in Python. On the other hand, C++ |
| 718 | will automatically apply an implicit type conversion, which makes it possible |
| 719 | to directly write ``func(a)``. |
| 720 | |
| 721 | In this situation (i.e. where ``B`` has a constructor that converts from |
| 722 | ``A``), the following statement enables similar implicit conversions on the |
| 723 | Python side: |
| 724 | |
| 725 | .. code-block:: cpp |
| 726 | |
| 727 | py::implicitly_convertible<A, B>(); |
| 728 | |
Wenzel Jakob | 3eeea6f | 2016-06-30 18:10:28 +0200 | [diff] [blame] | 729 | .. note:: |
| 730 | |
| 731 | Implicit conversions from ``A`` to ``B`` only work when ``B`` is a custom |
| 732 | data type that is exposed to Python via pybind11. |
| 733 | |
Wenzel Jakob | f88af0c | 2016-06-22 13:52:31 +0200 | [diff] [blame] | 734 | .. _static_properties: |
| 735 | |
| 736 | Static properties |
| 737 | ================= |
| 738 | |
| 739 | The section on :ref:`properties` discussed the creation of instance properties |
| 740 | that are implemented in terms of C++ getters and setters. |
| 741 | |
| 742 | Static properties can also be created in a similar way to expose getters and |
| 743 | setters of static class attributes. It is important to note that the implicit |
| 744 | ``self`` argument also exists in this case and is used to pass the Python |
| 745 | ``type`` subclass instance. This parameter will often not be needed by the C++ |
| 746 | side, and the following example illustrates how to instantiate a lambda getter |
| 747 | function that ignores it: |
| 748 | |
| 749 | .. code-block:: cpp |
| 750 | |
| 751 | py::class_<Foo>(m, "Foo") |
| 752 | .def_property_readonly_static("foo", [](py::object /* self */) { return Foo(); }); |
| 753 | |
Wenzel Jakob | 9f0dfce | 2016-04-06 17:38:18 +0200 | [diff] [blame] | 754 | Unique pointers |
| 755 | =============== |
| 756 | |
| 757 | Given a class ``Example`` with Python bindings, it's possible to return |
| 758 | instances wrapped in C++11 unique pointers, like so |
| 759 | |
| 760 | .. code-block:: cpp |
| 761 | |
| 762 | std::unique_ptr<Example> create_example() { return std::unique_ptr<Example>(new Example()); } |
| 763 | |
| 764 | .. code-block:: cpp |
| 765 | |
| 766 | m.def("create_example", &create_example); |
| 767 | |
| 768 | In other words, there is nothing special that needs to be done. While returning |
| 769 | unique pointers in this way is allowed, it is *illegal* to use them as function |
| 770 | arguments. For instance, the following function signature cannot be processed |
| 771 | by pybind11. |
| 772 | |
| 773 | .. code-block:: cpp |
| 774 | |
| 775 | void do_something_with_example(std::unique_ptr<Example> ex) { ... } |
| 776 | |
| 777 | The above signature would imply that Python needs to give up ownership of an |
| 778 | object that is passed to this function, which is generally not possible (for |
| 779 | instance, the object might be referenced elsewhere). |
| 780 | |
Wenzel Jakob | f7b5874 | 2016-04-25 23:04:27 +0200 | [diff] [blame] | 781 | .. _smart_pointers: |
| 782 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 783 | Smart pointers |
| 784 | ============== |
| 785 | |
Wenzel Jakob | 9f0dfce | 2016-04-06 17:38:18 +0200 | [diff] [blame] | 786 | This section explains how to pass values that are wrapped in "smart" pointer |
Wenzel Jakob | e84f557 | 2016-04-26 23:19:19 +0200 | [diff] [blame] | 787 | types with internal reference counting. For the simpler C++11 unique pointers, |
| 788 | refer to the previous section. |
Wenzel Jakob | 9f0dfce | 2016-04-06 17:38:18 +0200 | [diff] [blame] | 789 | |
Wenzel Jakob | e84f557 | 2016-04-26 23:19:19 +0200 | [diff] [blame] | 790 | The binding generator for classes, :class:`class_`, takes an optional second |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 791 | template type, which denotes a special *holder* type that is used to manage |
| 792 | references to the object. When wrapping a type named ``Type``, the default |
| 793 | value of this template parameter is ``std::unique_ptr<Type>``, which means that |
| 794 | the object is deallocated when Python's reference count goes to zero. |
| 795 | |
Wenzel Jakob | 1853b65 | 2015-10-18 15:38:50 +0200 | [diff] [blame] | 796 | It is possible to switch to other types of reference counting wrappers or smart |
| 797 | pointers, which is useful in codebases that rely on them. For instance, the |
| 798 | following snippet causes ``std::shared_ptr`` to be used instead. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 799 | |
| 800 | .. code-block:: cpp |
| 801 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 802 | py::class_<Example, std::shared_ptr<Example> /* <- holder type */> obj(m, "Example"); |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 803 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 804 | Note that any particular class can only be associated with a single holder type. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 805 | |
Wenzel Jakob | 1853b65 | 2015-10-18 15:38:50 +0200 | [diff] [blame] | 806 | To enable transparent conversions for functions that take shared pointers as an |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 807 | argument or that return them, a macro invocation similar to the following must |
Wenzel Jakob | 1853b65 | 2015-10-18 15:38:50 +0200 | [diff] [blame] | 808 | be declared at the top level before any binding code: |
| 809 | |
| 810 | .. code-block:: cpp |
| 811 | |
Wenzel Jakob | b1b7140 | 2015-10-18 16:48:30 +0200 | [diff] [blame] | 812 | PYBIND11_DECLARE_HOLDER_TYPE(T, std::shared_ptr<T>); |
Wenzel Jakob | 1853b65 | 2015-10-18 15:38:50 +0200 | [diff] [blame] | 813 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 814 | .. note:: |
Wenzel Jakob | 61d67f0 | 2015-12-14 12:53:06 +0100 | [diff] [blame] | 815 | |
| 816 | The first argument of :func:`PYBIND11_DECLARE_HOLDER_TYPE` should be a |
| 817 | placeholder name that is used as a template parameter of the second |
| 818 | argument. Thus, feel free to use any identifier, but use it consistently on |
| 819 | both sides; also, don't use the name of a type that already exists in your |
| 820 | codebase. |
| 821 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 822 | One potential stumbling block when using holder types is that they need to be |
| 823 | applied consistently. Can you guess what's broken about the following binding |
| 824 | code? |
Wenzel Jakob | 6e213c9 | 2015-11-24 23:05:58 +0100 | [diff] [blame] | 825 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 826 | .. code-block:: cpp |
Wenzel Jakob | 6e213c9 | 2015-11-24 23:05:58 +0100 | [diff] [blame] | 827 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 828 | class Child { }; |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 829 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 830 | class Parent { |
| 831 | public: |
| 832 | Parent() : child(std::make_shared<Child>()) { } |
| 833 | Child *get_child() { return child.get(); } /* Hint: ** DON'T DO THIS ** */ |
| 834 | private: |
| 835 | std::shared_ptr<Child> child; |
| 836 | }; |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 837 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 838 | PYBIND11_PLUGIN(example) { |
| 839 | py::module m("example"); |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 840 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 841 | py::class_<Child, std::shared_ptr<Child>>(m, "Child"); |
| 842 | |
| 843 | py::class_<Parent, std::shared_ptr<Parent>>(m, "Parent") |
| 844 | .def(py::init<>()) |
| 845 | .def("get_child", &Parent::get_child); |
| 846 | |
| 847 | return m.ptr(); |
| 848 | } |
| 849 | |
| 850 | The following Python code will cause undefined behavior (and likely a |
| 851 | segmentation fault). |
| 852 | |
| 853 | .. code-block:: python |
| 854 | |
| 855 | from example import Parent |
| 856 | print(Parent().get_child()) |
| 857 | |
| 858 | The problem is that ``Parent::get_child()`` returns a pointer to an instance of |
| 859 | ``Child``, but the fact that this instance is already managed by |
| 860 | ``std::shared_ptr<...>`` is lost when passing raw pointers. In this case, |
| 861 | pybind11 will create a second independent ``std::shared_ptr<...>`` that also |
| 862 | claims ownership of the pointer. In the end, the object will be freed **twice** |
| 863 | since these shared pointers have no way of knowing about each other. |
| 864 | |
| 865 | There are two ways to resolve this issue: |
| 866 | |
| 867 | 1. For types that are managed by a smart pointer class, never use raw pointers |
| 868 | in function arguments or return values. In other words: always consistently |
| 869 | wrap pointers into their designated holder types (such as |
| 870 | ``std::shared_ptr<...>``). In this case, the signature of ``get_child()`` |
| 871 | should be modified as follows: |
| 872 | |
| 873 | .. code-block:: cpp |
| 874 | |
| 875 | std::shared_ptr<Child> get_child() { return child; } |
| 876 | |
| 877 | 2. Adjust the definition of ``Child`` by specifying |
| 878 | ``std::enable_shared_from_this<T>`` (see cppreference_ for details) as a |
| 879 | base class. This adds a small bit of information to ``Child`` that allows |
| 880 | pybind11 to realize that there is already an existing |
| 881 | ``std::shared_ptr<...>`` and communicate with it. In this case, the |
| 882 | declaration of ``Child`` should look as follows: |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 883 | |
Wenzel Jakob | 6e213c9 | 2015-11-24 23:05:58 +0100 | [diff] [blame] | 884 | .. _cppreference: http://en.cppreference.com/w/cpp/memory/enable_shared_from_this |
| 885 | |
Wenzel Jakob | b2c2c79 | 2016-01-17 22:36:40 +0100 | [diff] [blame] | 886 | .. code-block:: cpp |
| 887 | |
| 888 | class Child : public std::enable_shared_from_this<Child> { }; |
| 889 | |
Wenzel Jakob | 9bb97c1 | 2016-06-03 11:19:41 +0200 | [diff] [blame] | 890 | |
| 891 | Please take a look at the :ref:`macro_notes` before using this feature. |
| 892 | |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 893 | .. seealso:: |
| 894 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 895 | The file :file:`tests/test_smart_ptr.cpp` contains a complete example |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 896 | that demonstrates how to work with custom reference-counting holder types |
| 897 | in more detail. |
Wenzel Jakob | 5ef1219 | 2015-12-15 17:07:35 +0100 | [diff] [blame] | 898 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 899 | .. _custom_constructors: |
| 900 | |
| 901 | Custom constructors |
| 902 | =================== |
| 903 | |
| 904 | The syntax for binding constructors was previously introduced, but it only |
| 905 | works when a constructor with the given parameters actually exists on the C++ |
| 906 | side. To extend this to more general cases, let's take a look at what actually |
| 907 | happens under the hood: the following statement |
| 908 | |
| 909 | .. code-block:: cpp |
| 910 | |
| 911 | py::class_<Example>(m, "Example") |
| 912 | .def(py::init<int>()); |
| 913 | |
| 914 | is short hand notation for |
| 915 | |
| 916 | .. code-block:: cpp |
| 917 | |
| 918 | py::class_<Example>(m, "Example") |
| 919 | .def("__init__", |
| 920 | [](Example &instance, int arg) { |
| 921 | new (&instance) Example(arg); |
| 922 | } |
| 923 | ); |
| 924 | |
| 925 | In other words, :func:`init` creates an anonymous function that invokes an |
| 926 | in-place constructor. Memory allocation etc. is already take care of beforehand |
| 927 | within pybind11. |
| 928 | |
Pim Schellart | 5a7d17f | 2016-06-17 17:35:59 -0400 | [diff] [blame] | 929 | .. _catching_and_throwing_exceptions: |
| 930 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 931 | Catching and throwing exceptions |
| 932 | ================================ |
| 933 | |
| 934 | When C++ code invoked from Python throws an ``std::exception``, it is |
| 935 | automatically converted into a Python ``Exception``. pybind11 defines multiple |
| 936 | special exception classes that will map to different types of Python |
| 937 | exceptions: |
| 938 | |
Wenzel Jakob | f64feaf | 2016-04-28 14:33:45 +0200 | [diff] [blame] | 939 | .. tabularcolumns:: |p{0.5\textwidth}|p{0.45\textwidth}| |
| 940 | |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 941 | +--------------------------------------+------------------------------+ |
| 942 | | C++ exception type | Python exception type | |
| 943 | +======================================+==============================+ |
| 944 | | :class:`std::exception` | ``RuntimeError`` | |
| 945 | +--------------------------------------+------------------------------+ |
| 946 | | :class:`std::bad_alloc` | ``MemoryError`` | |
| 947 | +--------------------------------------+------------------------------+ |
| 948 | | :class:`std::domain_error` | ``ValueError`` | |
| 949 | +--------------------------------------+------------------------------+ |
| 950 | | :class:`std::invalid_argument` | ``ValueError`` | |
| 951 | +--------------------------------------+------------------------------+ |
| 952 | | :class:`std::length_error` | ``ValueError`` | |
| 953 | +--------------------------------------+------------------------------+ |
| 954 | | :class:`std::out_of_range` | ``ValueError`` | |
| 955 | +--------------------------------------+------------------------------+ |
| 956 | | :class:`std::range_error` | ``ValueError`` | |
| 957 | +--------------------------------------+------------------------------+ |
| 958 | | :class:`pybind11::stop_iteration` | ``StopIteration`` (used to | |
| 959 | | | implement custom iterators) | |
| 960 | +--------------------------------------+------------------------------+ |
| 961 | | :class:`pybind11::index_error` | ``IndexError`` (used to | |
| 962 | | | indicate out of bounds | |
| 963 | | | accesses in ``__getitem__``, | |
| 964 | | | ``__setitem__``, etc.) | |
| 965 | +--------------------------------------+------------------------------+ |
Sergey Lyskov | a95bde1 | 2016-05-08 19:31:55 -0400 | [diff] [blame] | 966 | | :class:`pybind11::value_error` | ``ValueError`` (used to | |
| 967 | | | indicate wrong value passed | |
| 968 | | | in ``container.remove(...)`` | |
| 969 | +--------------------------------------+------------------------------+ |
Jason Rhinelander | 5aa85be | 2016-08-11 21:22:05 -0400 | [diff] [blame] | 970 | | :class:`pybind11::key_error` | ``KeyError`` (used to | |
| 971 | | | indicate out of bounds | |
| 972 | | | accesses in ``__getitem__``, | |
| 973 | | | ``__setitem__`` in dict-like | |
| 974 | | | objects, etc.) | |
| 975 | +--------------------------------------+------------------------------+ |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 976 | | :class:`pybind11::error_already_set` | Indicates that the Python | |
| 977 | | | exception flag has already | |
| 978 | | | been initialized | |
| 979 | +--------------------------------------+------------------------------+ |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 980 | |
| 981 | When a Python function invoked from C++ throws an exception, it is converted |
| 982 | into a C++ exception of type :class:`error_already_set` whose string payload |
| 983 | contains a textual summary. |
| 984 | |
| 985 | There is also a special exception :class:`cast_error` that is thrown by |
| 986 | :func:`handle::call` when the input arguments cannot be converted to Python |
| 987 | objects. |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 988 | |
Pim Schellart | 5a7d17f | 2016-06-17 17:35:59 -0400 | [diff] [blame] | 989 | Registering custom exception translators |
| 990 | ======================================== |
| 991 | |
| 992 | If the default exception conversion policy described |
| 993 | :ref:`above <catching_and_throwing_exceptions>` |
| 994 | is insufficient, pybind11 also provides support for registering custom |
| 995 | exception translators. |
| 996 | |
| 997 | The function ``register_exception_translator(translator)`` takes a stateless |
| 998 | callable (e.g. a function pointer or a lambda function without captured |
| 999 | variables) with the following call signature: ``void(std::exception_ptr)``. |
| 1000 | |
| 1001 | When a C++ exception is thrown, registered exception translators are tried |
| 1002 | in reverse order of registration (i.e. the last registered translator gets |
| 1003 | a first shot at handling the exception). |
| 1004 | |
| 1005 | Inside the translator, ``std::rethrow_exception`` should be used within |
| 1006 | a try block to re-throw the exception. A catch clause can then use |
| 1007 | ``PyErr_SetString`` to set a Python exception as demonstrated |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1008 | in :file:`tests/test_exceptions.cpp`. |
Pim Schellart | 5a7d17f | 2016-06-17 17:35:59 -0400 | [diff] [blame] | 1009 | |
| 1010 | This example also demonstrates how to create custom exception types |
| 1011 | with ``py::exception``. |
| 1012 | |
| 1013 | The following example demonstrates this for a hypothetical exception class |
| 1014 | ``MyCustomException``: |
| 1015 | |
| 1016 | .. code-block:: cpp |
| 1017 | |
| 1018 | py::register_exception_translator([](std::exception_ptr p) { |
| 1019 | try { |
| 1020 | if (p) std::rethrow_exception(p); |
| 1021 | } catch (const MyCustomException &e) { |
| 1022 | PyErr_SetString(PyExc_RuntimeError, e.what()); |
| 1023 | } |
| 1024 | }); |
| 1025 | |
| 1026 | Multiple exceptions can be handled by a single translator. If the exception is |
| 1027 | not caught by the current translator, the previously registered one gets a |
| 1028 | chance. |
| 1029 | |
| 1030 | If none of the registered exception translators is able to handle the |
| 1031 | exception, it is handled by the default converter as described in the previous |
| 1032 | section. |
| 1033 | |
| 1034 | .. note:: |
| 1035 | |
| 1036 | You must either call ``PyErr_SetString`` for every exception caught in a |
| 1037 | custom exception translator. Failure to do so will cause Python to crash |
| 1038 | with ``SystemError: error return without exception set``. |
| 1039 | |
| 1040 | Exceptions that you do not plan to handle should simply not be caught. |
| 1041 | |
| 1042 | You may also choose to explicity (re-)throw the exception to delegate it to |
| 1043 | the other existing exception translators. |
| 1044 | |
| 1045 | The ``py::exception`` wrapper for creating custom exceptions cannot (yet) |
| 1046 | be used as a ``py::base``. |
| 1047 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1048 | .. _opaque: |
| 1049 | |
| 1050 | Treating STL data structures as opaque objects |
| 1051 | ============================================== |
| 1052 | |
| 1053 | pybind11 heavily relies on a template matching mechanism to convert parameters |
| 1054 | and return values that are constructed from STL data types such as vectors, |
| 1055 | linked lists, hash tables, etc. This even works in a recursive manner, for |
| 1056 | instance to deal with lists of hash maps of pairs of elementary and custom |
| 1057 | types, etc. |
| 1058 | |
| 1059 | However, a fundamental limitation of this approach is that internal conversions |
| 1060 | between Python and C++ types involve a copy operation that prevents |
| 1061 | pass-by-reference semantics. What does this mean? |
| 1062 | |
| 1063 | Suppose we bind the following function |
| 1064 | |
| 1065 | .. code-block:: cpp |
| 1066 | |
| 1067 | void append_1(std::vector<int> &v) { |
| 1068 | v.push_back(1); |
| 1069 | } |
| 1070 | |
| 1071 | and call it from Python, the following happens: |
| 1072 | |
Wenzel Jakob | 99279f7 | 2016-06-03 11:19:29 +0200 | [diff] [blame] | 1073 | .. code-block:: pycon |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1074 | |
| 1075 | >>> v = [5, 6] |
| 1076 | >>> append_1(v) |
| 1077 | >>> print(v) |
| 1078 | [5, 6] |
| 1079 | |
| 1080 | As you can see, when passing STL data structures by reference, modifications |
| 1081 | are not propagated back the Python side. A similar situation arises when |
| 1082 | exposing STL data structures using the ``def_readwrite`` or ``def_readonly`` |
| 1083 | functions: |
| 1084 | |
| 1085 | .. code-block:: cpp |
| 1086 | |
| 1087 | /* ... definition ... */ |
| 1088 | |
| 1089 | class MyClass { |
| 1090 | std::vector<int> contents; |
| 1091 | }; |
| 1092 | |
| 1093 | /* ... binding code ... */ |
| 1094 | |
| 1095 | py::class_<MyClass>(m, "MyClass") |
| 1096 | .def(py::init<>) |
| 1097 | .def_readwrite("contents", &MyClass::contents); |
| 1098 | |
| 1099 | In this case, properties can be read and written in their entirety. However, an |
| 1100 | ``append`` operaton involving such a list type has no effect: |
| 1101 | |
Wenzel Jakob | 99279f7 | 2016-06-03 11:19:29 +0200 | [diff] [blame] | 1102 | .. code-block:: pycon |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1103 | |
| 1104 | >>> m = MyClass() |
| 1105 | >>> m.contents = [5, 6] |
| 1106 | >>> print(m.contents) |
| 1107 | [5, 6] |
| 1108 | >>> m.contents.append(7) |
| 1109 | >>> print(m.contents) |
| 1110 | [5, 6] |
| 1111 | |
| 1112 | To deal with both of the above situations, pybind11 provides a macro named |
| 1113 | ``PYBIND11_MAKE_OPAQUE(T)`` that disables the template-based conversion |
| 1114 | machinery of types, thus rendering them *opaque*. The contents of opaque |
| 1115 | objects are never inspected or extracted, hence they can be passed by |
| 1116 | reference. For instance, to turn ``std::vector<int>`` into an opaque type, add |
| 1117 | the declaration |
| 1118 | |
| 1119 | .. code-block:: cpp |
| 1120 | |
| 1121 | PYBIND11_MAKE_OPAQUE(std::vector<int>); |
| 1122 | |
| 1123 | before any binding code (e.g. invocations to ``class_::def()``, etc.). This |
| 1124 | macro must be specified at the top level, since instantiates a partial template |
| 1125 | overload. If your binding code consists of multiple compilation units, it must |
| 1126 | be present in every file preceding any usage of ``std::vector<int>``. Opaque |
| 1127 | types must also have a corresponding ``class_`` declaration to associate them |
| 1128 | with a name in Python, and to define a set of available operations: |
| 1129 | |
| 1130 | .. code-block:: cpp |
| 1131 | |
| 1132 | py::class_<std::vector<int>>(m, "IntVector") |
| 1133 | .def(py::init<>()) |
| 1134 | .def("clear", &std::vector<int>::clear) |
| 1135 | .def("pop_back", &std::vector<int>::pop_back) |
| 1136 | .def("__len__", [](const std::vector<int> &v) { return v.size(); }) |
| 1137 | .def("__iter__", [](std::vector<int> &v) { |
| 1138 | return py::make_iterator(v.begin(), v.end()); |
| 1139 | }, py::keep_alive<0, 1>()) /* Keep vector alive while iterator is used */ |
| 1140 | // .... |
| 1141 | |
Wenzel Jakob | 9bb97c1 | 2016-06-03 11:19:41 +0200 | [diff] [blame] | 1142 | Please take a look at the :ref:`macro_notes` before using this feature. |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1143 | |
| 1144 | .. seealso:: |
| 1145 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1146 | The file :file:`tests/test_opaque_types.cpp` contains a complete |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1147 | example that demonstrates how to create and expose opaque types using |
| 1148 | pybind11 in more detail. |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1149 | |
| 1150 | .. _eigen: |
| 1151 | |
| 1152 | Transparent conversion of dense and sparse Eigen data types |
| 1153 | =========================================================== |
| 1154 | |
| 1155 | Eigen [#f1]_ is C++ header-based library for dense and sparse linear algebra. Due to |
| 1156 | its popularity and widespread adoption, pybind11 provides transparent |
| 1157 | conversion support between Eigen and Scientific Python linear algebra data types. |
| 1158 | |
| 1159 | Specifically, when including the optional header file :file:`pybind11/eigen.h`, |
Wenzel Jakob | 178c8a8 | 2016-05-10 15:59:01 +0100 | [diff] [blame] | 1160 | pybind11 will automatically and transparently convert |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1161 | |
| 1162 | 1. Static and dynamic Eigen dense vectors and matrices to instances of |
| 1163 | ``numpy.ndarray`` (and vice versa). |
| 1164 | |
Jason Rhinelander | b68d8fc | 2016-08-04 16:39:30 -0400 | [diff] [blame] | 1165 | 2. Returned matrix expressions such as blocks (including columns or rows) and |
Jason Rhinelander | 9ffb3dd | 2016-08-04 15:24:41 -0400 | [diff] [blame] | 1166 | diagonals will be converted to ``numpy.ndarray`` of the expression |
| 1167 | values. |
| 1168 | |
Jason Rhinelander | b68d8fc | 2016-08-04 16:39:30 -0400 | [diff] [blame] | 1169 | 3. Returned matrix-like objects such as Eigen::DiagonalMatrix or |
Jason Rhinelander | 9ffb3dd | 2016-08-04 15:24:41 -0400 | [diff] [blame] | 1170 | Eigen::SelfAdjointView will be converted to ``numpy.ndarray`` containing the |
| 1171 | expressed value. |
| 1172 | |
Jason Rhinelander | b68d8fc | 2016-08-04 16:39:30 -0400 | [diff] [blame] | 1173 | 4. Eigen sparse vectors and matrices to instances of |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1174 | ``scipy.sparse.csr_matrix``/``scipy.sparse.csc_matrix`` (and vice versa). |
| 1175 | |
| 1176 | This makes it possible to bind most kinds of functions that rely on these types. |
| 1177 | One major caveat are functions that take Eigen matrices *by reference* and modify |
| 1178 | them somehow, in which case the information won't be propagated to the caller. |
| 1179 | |
| 1180 | .. code-block:: cpp |
| 1181 | |
Jason Rhinelander | 9ffb3dd | 2016-08-04 15:24:41 -0400 | [diff] [blame] | 1182 | /* The Python bindings of these functions won't replicate |
| 1183 | the intended effect of modifying the function arguments */ |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1184 | void scale_by_2(Eigen::Vector3f &v) { |
Jason Rhinelander | 9ffb3dd | 2016-08-04 15:24:41 -0400 | [diff] [blame] | 1185 | v *= 2; |
| 1186 | } |
| 1187 | void scale_by_2(Eigen::Ref<Eigen::MatrixXd> &v) { |
| 1188 | v *= 2; |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1189 | } |
| 1190 | |
| 1191 | To see why this is, refer to the section on :ref:`opaque` (although that |
| 1192 | section specifically covers STL data types, the underlying issue is the same). |
| 1193 | The next two sections discuss an efficient alternative for exposing the |
| 1194 | underlying native Eigen types as opaque objects in a way that still integrates |
| 1195 | with NumPy and SciPy. |
| 1196 | |
| 1197 | .. [#f1] http://eigen.tuxfamily.org |
| 1198 | |
| 1199 | .. seealso:: |
| 1200 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1201 | The file :file:`tests/test_eigen.cpp` contains a complete example that |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1202 | shows how to pass Eigen sparse and dense data types in more detail. |
| 1203 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1204 | Buffer protocol |
| 1205 | =============== |
| 1206 | |
| 1207 | Python supports an extremely general and convenient approach for exchanging |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1208 | data between plugin libraries. Types can expose a buffer view [#f2]_, which |
| 1209 | provides fast direct access to the raw internal data representation. Suppose we |
| 1210 | want to bind the following simplistic Matrix class: |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1211 | |
| 1212 | .. code-block:: cpp |
| 1213 | |
| 1214 | class Matrix { |
| 1215 | public: |
| 1216 | Matrix(size_t rows, size_t cols) : m_rows(rows), m_cols(cols) { |
| 1217 | m_data = new float[rows*cols]; |
| 1218 | } |
| 1219 | float *data() { return m_data; } |
| 1220 | size_t rows() const { return m_rows; } |
| 1221 | size_t cols() const { return m_cols; } |
| 1222 | private: |
| 1223 | size_t m_rows, m_cols; |
| 1224 | float *m_data; |
| 1225 | }; |
| 1226 | |
| 1227 | The following binding code exposes the ``Matrix`` contents as a buffer object, |
Wenzel Jakob | 8e93df8 | 2016-05-01 02:36:58 +0200 | [diff] [blame] | 1228 | making it possible to cast Matrices into NumPy arrays. It is even possible to |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1229 | completely avoid copy operations with Python expressions like |
| 1230 | ``np.array(matrix_instance, copy = False)``. |
| 1231 | |
| 1232 | .. code-block:: cpp |
| 1233 | |
| 1234 | py::class_<Matrix>(m, "Matrix") |
| 1235 | .def_buffer([](Matrix &m) -> py::buffer_info { |
| 1236 | return py::buffer_info( |
Ivan Smirnov | 5e71e17 | 2016-06-26 12:42:34 +0100 | [diff] [blame] | 1237 | m.data(), /* Pointer to buffer */ |
| 1238 | sizeof(float), /* Size of one scalar */ |
| 1239 | py::format_descriptor<float>::format(), /* Python struct-style format descriptor */ |
| 1240 | 2, /* Number of dimensions */ |
| 1241 | { m.rows(), m.cols() }, /* Buffer dimensions */ |
| 1242 | { sizeof(float) * m.rows(), /* Strides (in bytes) for each index */ |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1243 | sizeof(float) } |
| 1244 | ); |
| 1245 | }); |
| 1246 | |
| 1247 | The snippet above binds a lambda function, which can create ``py::buffer_info`` |
| 1248 | description records on demand describing a given matrix. The contents of |
| 1249 | ``py::buffer_info`` mirror the Python buffer protocol specification. |
| 1250 | |
| 1251 | .. code-block:: cpp |
| 1252 | |
| 1253 | struct buffer_info { |
| 1254 | void *ptr; |
| 1255 | size_t itemsize; |
| 1256 | std::string format; |
| 1257 | int ndim; |
| 1258 | std::vector<size_t> shape; |
| 1259 | std::vector<size_t> strides; |
| 1260 | }; |
| 1261 | |
| 1262 | To create a C++ function that can take a Python buffer object as an argument, |
| 1263 | simply use the type ``py::buffer`` as one of its arguments. Buffers can exist |
| 1264 | in a great variety of configurations, hence some safety checks are usually |
| 1265 | necessary in the function body. Below, you can see an basic example on how to |
| 1266 | define a custom constructor for the Eigen double precision matrix |
| 1267 | (``Eigen::MatrixXd``) type, which supports initialization from compatible |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1268 | buffer objects (e.g. a NumPy matrix). |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1269 | |
| 1270 | .. code-block:: cpp |
| 1271 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1272 | /* Bind MatrixXd (or some other Eigen type) to Python */ |
| 1273 | typedef Eigen::MatrixXd Matrix; |
| 1274 | |
| 1275 | typedef Matrix::Scalar Scalar; |
| 1276 | constexpr bool rowMajor = Matrix::Flags & Eigen::RowMajorBit; |
| 1277 | |
| 1278 | py::class_<Matrix>(m, "Matrix") |
| 1279 | .def("__init__", [](Matrix &m, py::buffer b) { |
Wenzel Jakob | e762853 | 2016-05-05 10:04:44 +0200 | [diff] [blame] | 1280 | typedef Eigen::Stride<Eigen::Dynamic, Eigen::Dynamic> Strides; |
Wenzel Jakob | e762853 | 2016-05-05 10:04:44 +0200 | [diff] [blame] | 1281 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1282 | /* Request a buffer descriptor from Python */ |
| 1283 | py::buffer_info info = b.request(); |
| 1284 | |
| 1285 | /* Some sanity checks ... */ |
Ivan Smirnov | 5e71e17 | 2016-06-26 12:42:34 +0100 | [diff] [blame] | 1286 | if (info.format != py::format_descriptor<Scalar>::format()) |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1287 | throw std::runtime_error("Incompatible format: expected a double array!"); |
| 1288 | |
| 1289 | if (info.ndim != 2) |
| 1290 | throw std::runtime_error("Incompatible buffer dimension!"); |
| 1291 | |
Wenzel Jakob | e762853 | 2016-05-05 10:04:44 +0200 | [diff] [blame] | 1292 | auto strides = Strides( |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1293 | info.strides[rowMajor ? 0 : 1] / sizeof(Scalar), |
| 1294 | info.strides[rowMajor ? 1 : 0] / sizeof(Scalar)); |
Wenzel Jakob | e762853 | 2016-05-05 10:04:44 +0200 | [diff] [blame] | 1295 | |
| 1296 | auto map = Eigen::Map<Matrix, 0, Strides>( |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1297 | static_cat<Scalar *>(info.ptr), info.shape[0], info.shape[1], strides); |
Wenzel Jakob | e762853 | 2016-05-05 10:04:44 +0200 | [diff] [blame] | 1298 | |
| 1299 | new (&m) Matrix(map); |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1300 | }); |
| 1301 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1302 | For reference, the ``def_buffer()`` call for this Eigen data type should look |
| 1303 | as follows: |
| 1304 | |
| 1305 | .. code-block:: cpp |
| 1306 | |
| 1307 | .def_buffer([](Matrix &m) -> py::buffer_info { |
| 1308 | return py::buffer_info( |
| 1309 | m.data(), /* Pointer to buffer */ |
| 1310 | sizeof(Scalar), /* Size of one scalar */ |
| 1311 | /* Python struct-style format descriptor */ |
Ivan Smirnov | 5e71e17 | 2016-06-26 12:42:34 +0100 | [diff] [blame] | 1312 | py::format_descriptor<Scalar>::format(), |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1313 | /* Number of dimensions */ |
| 1314 | 2, |
| 1315 | /* Buffer dimensions */ |
| 1316 | { (size_t) m.rows(), |
| 1317 | (size_t) m.cols() }, |
| 1318 | /* Strides (in bytes) for each index */ |
| 1319 | { sizeof(Scalar) * (rowMajor ? m.cols() : 1), |
| 1320 | sizeof(Scalar) * (rowMajor ? 1 : m.rows()) } |
| 1321 | ); |
| 1322 | }) |
| 1323 | |
| 1324 | For a much easier approach of binding Eigen types (although with some |
| 1325 | limitations), refer to the section on :ref:`eigen`. |
| 1326 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1327 | .. seealso:: |
| 1328 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1329 | The file :file:`tests/test_buffers.cpp` contains a complete example |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1330 | that demonstrates using the buffer protocol with pybind11 in more detail. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1331 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1332 | .. [#f2] http://docs.python.org/3/c-api/buffer.html |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 1333 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1334 | NumPy support |
| 1335 | ============= |
| 1336 | |
| 1337 | By exchanging ``py::buffer`` with ``py::array`` in the above snippet, we can |
| 1338 | restrict the function so that it only accepts NumPy arrays (rather than any |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 1339 | type of Python object satisfying the buffer protocol). |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1340 | |
| 1341 | In many situations, we want to define a function which only accepts a NumPy |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1342 | array of a certain data type. This is possible via the ``py::array_t<T>`` |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1343 | template. For instance, the following function requires the argument to be a |
Wenzel Jakob | f1032df | 2016-05-05 10:00:00 +0200 | [diff] [blame] | 1344 | NumPy array containing double precision values. |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1345 | |
| 1346 | .. code-block:: cpp |
| 1347 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1348 | void f(py::array_t<double> array); |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1349 | |
Wenzel Jakob | f1032df | 2016-05-05 10:00:00 +0200 | [diff] [blame] | 1350 | When it is invoked with a different type (e.g. an integer or a list of |
| 1351 | integers), the binding code will attempt to cast the input into a NumPy array |
| 1352 | of the requested type. Note that this feature requires the |
| 1353 | :file:``pybind11/numpy.h`` header to be included. |
| 1354 | |
| 1355 | Data in NumPy arrays is not guaranteed to packed in a dense manner; |
| 1356 | furthermore, entries can be separated by arbitrary column and row strides. |
| 1357 | Sometimes, it can be useful to require a function to only accept dense arrays |
| 1358 | using either the C (row-major) or Fortran (column-major) ordering. This can be |
| 1359 | accomplished via a second template argument with values ``py::array::c_style`` |
| 1360 | or ``py::array::f_style``. |
| 1361 | |
| 1362 | .. code-block:: cpp |
| 1363 | |
Wenzel Jakob | b47a9de | 2016-05-19 16:02:09 +0200 | [diff] [blame] | 1364 | void f(py::array_t<double, py::array::c_style | py::array::forcecast> array); |
Wenzel Jakob | f1032df | 2016-05-05 10:00:00 +0200 | [diff] [blame] | 1365 | |
Wenzel Jakob | b47a9de | 2016-05-19 16:02:09 +0200 | [diff] [blame] | 1366 | The ``py::array::forcecast`` argument is the default value of the second |
| 1367 | template paramenter, and it ensures that non-conforming arguments are converted |
| 1368 | into an array satisfying the specified requirements instead of trying the next |
| 1369 | function overload. |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1370 | |
Ivan Smirnov | 223afe3 | 2016-07-02 15:33:04 +0100 | [diff] [blame] | 1371 | NumPy structured types |
| 1372 | ====================== |
| 1373 | |
| 1374 | In order for ``py::array_t`` to work with structured (record) types, we first need |
Ivan Smirnov | 5412a05 | 2016-07-02 16:18:42 +0100 | [diff] [blame] | 1375 | to register the memory layout of the type. This can be done via ``PYBIND11_NUMPY_DTYPE`` |
Ivan Smirnov | 223afe3 | 2016-07-02 15:33:04 +0100 | [diff] [blame] | 1376 | macro which expects the type followed by field names: |
| 1377 | |
| 1378 | .. code-block:: cpp |
| 1379 | |
| 1380 | struct A { |
| 1381 | int x; |
| 1382 | double y; |
| 1383 | }; |
| 1384 | |
| 1385 | struct B { |
| 1386 | int z; |
| 1387 | A a; |
| 1388 | }; |
| 1389 | |
Ivan Smirnov | 5412a05 | 2016-07-02 16:18:42 +0100 | [diff] [blame] | 1390 | PYBIND11_NUMPY_DTYPE(A, x, y); |
| 1391 | PYBIND11_NUMPY_DTYPE(B, z, a); |
Ivan Smirnov | 223afe3 | 2016-07-02 15:33:04 +0100 | [diff] [blame] | 1392 | |
| 1393 | /* now both A and B can be used as template arguments to py::array_t */ |
| 1394 | |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1395 | Vectorizing functions |
| 1396 | ===================== |
| 1397 | |
| 1398 | Suppose we want to bind a function with the following signature to Python so |
| 1399 | that it can process arbitrary NumPy array arguments (vectors, matrices, general |
| 1400 | N-D arrays) in addition to its normal arguments: |
| 1401 | |
| 1402 | .. code-block:: cpp |
| 1403 | |
| 1404 | double my_func(int x, float y, double z); |
| 1405 | |
Wenzel Jakob | 8f4eb00 | 2015-10-15 18:13:33 +0200 | [diff] [blame] | 1406 | After including the ``pybind11/numpy.h`` header, this is extremely simple: |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1407 | |
| 1408 | .. code-block:: cpp |
| 1409 | |
| 1410 | m.def("vectorized_func", py::vectorize(my_func)); |
| 1411 | |
| 1412 | Invoking the function like below causes 4 calls to be made to ``my_func`` with |
Wenzel Jakob | 8e93df8 | 2016-05-01 02:36:58 +0200 | [diff] [blame] | 1413 | each of the array elements. The significant advantage of this compared to |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 1414 | solutions like ``numpy.vectorize()`` is that the loop over the elements runs |
| 1415 | entirely on the C++ side and can be crunched down into a tight, optimized loop |
| 1416 | by the compiler. The result is returned as a NumPy array of type |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1417 | ``numpy.dtype.float64``. |
| 1418 | |
Wenzel Jakob | 99279f7 | 2016-06-03 11:19:29 +0200 | [diff] [blame] | 1419 | .. code-block:: pycon |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1420 | |
| 1421 | >>> x = np.array([[1, 3],[5, 7]]) |
| 1422 | >>> y = np.array([[2, 4],[6, 8]]) |
| 1423 | >>> z = 3 |
| 1424 | >>> result = vectorized_func(x, y, z) |
| 1425 | |
| 1426 | The scalar argument ``z`` is transparently replicated 4 times. The input |
| 1427 | arrays ``x`` and ``y`` are automatically converted into the right types (they |
| 1428 | are of type ``numpy.dtype.int64`` but need to be ``numpy.dtype.int32`` and |
| 1429 | ``numpy.dtype.float32``, respectively) |
| 1430 | |
Wenzel Jakob | 8e93df8 | 2016-05-01 02:36:58 +0200 | [diff] [blame] | 1431 | Sometimes we might want to explicitly exclude an argument from the vectorization |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1432 | because it makes little sense to wrap it in a NumPy array. For instance, |
| 1433 | suppose the function signature was |
| 1434 | |
| 1435 | .. code-block:: cpp |
| 1436 | |
| 1437 | double my_func(int x, float y, my_custom_type *z); |
| 1438 | |
| 1439 | This can be done with a stateful Lambda closure: |
| 1440 | |
| 1441 | .. code-block:: cpp |
| 1442 | |
| 1443 | // Vectorize a lambda function with a capture object (e.g. to exclude some arguments from the vectorization) |
| 1444 | m.def("vectorized_func", |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1445 | [](py::array_t<int> x, py::array_t<float> y, my_custom_type *z) { |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1446 | auto stateful_closure = [z](int x, float y) { return my_func(x, y, z); }; |
| 1447 | return py::vectorize(stateful_closure)(x, y); |
| 1448 | } |
| 1449 | ); |
| 1450 | |
Wenzel Jakob | 6158716 | 2016-01-18 22:38:52 +0100 | [diff] [blame] | 1451 | In cases where the computation is too complicated to be reduced to |
| 1452 | ``vectorize``, it will be necessary to create and access the buffer contents |
| 1453 | manually. The following snippet contains a complete example that shows how this |
| 1454 | works (the code is somewhat contrived, since it could have been done more |
| 1455 | simply using ``vectorize``). |
| 1456 | |
| 1457 | .. code-block:: cpp |
| 1458 | |
| 1459 | #include <pybind11/pybind11.h> |
| 1460 | #include <pybind11/numpy.h> |
| 1461 | |
| 1462 | namespace py = pybind11; |
| 1463 | |
| 1464 | py::array_t<double> add_arrays(py::array_t<double> input1, py::array_t<double> input2) { |
| 1465 | auto buf1 = input1.request(), buf2 = input2.request(); |
| 1466 | |
| 1467 | if (buf1.ndim != 1 || buf2.ndim != 1) |
| 1468 | throw std::runtime_error("Number of dimensions must be one"); |
| 1469 | |
Ivan Smirnov | b651859 | 2016-08-13 13:28:56 +0100 | [diff] [blame] | 1470 | if (buf1.size != buf2.size) |
Wenzel Jakob | 6158716 | 2016-01-18 22:38:52 +0100 | [diff] [blame] | 1471 | throw std::runtime_error("Input shapes must match"); |
| 1472 | |
Ivan Smirnov | b651859 | 2016-08-13 13:28:56 +0100 | [diff] [blame] | 1473 | /* No pointer is passed, so NumPy will allocate the buffer */ |
| 1474 | auto result = py::array_t<double>(buf1.size); |
Wenzel Jakob | 6158716 | 2016-01-18 22:38:52 +0100 | [diff] [blame] | 1475 | |
| 1476 | auto buf3 = result.request(); |
| 1477 | |
| 1478 | double *ptr1 = (double *) buf1.ptr, |
| 1479 | *ptr2 = (double *) buf2.ptr, |
| 1480 | *ptr3 = (double *) buf3.ptr; |
| 1481 | |
| 1482 | for (size_t idx = 0; idx < buf1.shape[0]; idx++) |
| 1483 | ptr3[idx] = ptr1[idx] + ptr2[idx]; |
| 1484 | |
| 1485 | return result; |
| 1486 | } |
| 1487 | |
| 1488 | PYBIND11_PLUGIN(test) { |
| 1489 | py::module m("test"); |
| 1490 | m.def("add_arrays", &add_arrays, "Add two NumPy arrays"); |
| 1491 | return m.ptr(); |
| 1492 | } |
| 1493 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1494 | .. seealso:: |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1495 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1496 | The file :file:`tests/test_numpy_vectorize.cpp` contains a complete |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1497 | example that demonstrates using :func:`vectorize` in more detail. |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1498 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1499 | Functions taking Python objects as arguments |
| 1500 | ============================================ |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1501 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1502 | pybind11 exposes all major Python types using thin C++ wrapper classes. These |
| 1503 | wrapper classes can also be used as parameters of functions in bindings, which |
| 1504 | makes it possible to directly work with native Python types on the C++ side. |
| 1505 | For instance, the following statement iterates over a Python ``dict``: |
Wenzel Jakob | 28f98aa | 2015-10-13 02:57:16 +0200 | [diff] [blame] | 1506 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1507 | .. code-block:: cpp |
| 1508 | |
| 1509 | void print_dict(py::dict dict) { |
| 1510 | /* Easily interact with Python types */ |
| 1511 | for (auto item : dict) |
| 1512 | std::cout << "key=" << item.first << ", " |
| 1513 | << "value=" << item.second << std::endl; |
| 1514 | } |
| 1515 | |
| 1516 | Available types include :class:`handle`, :class:`object`, :class:`bool_`, |
Wenzel Jakob | 27e8e10 | 2016-01-17 22:36:37 +0100 | [diff] [blame] | 1517 | :class:`int_`, :class:`float_`, :class:`str`, :class:`bytes`, :class:`tuple`, |
Wenzel Jakob | f64feaf | 2016-04-28 14:33:45 +0200 | [diff] [blame] | 1518 | :class:`list`, :class:`dict`, :class:`slice`, :class:`none`, :class:`capsule`, |
| 1519 | :class:`iterable`, :class:`iterator`, :class:`function`, :class:`buffer`, |
| 1520 | :class:`array`, and :class:`array_t`. |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1521 | |
Wenzel Jakob | 436b731 | 2015-10-20 01:04:30 +0200 | [diff] [blame] | 1522 | In this kind of mixed code, it is often necessary to convert arbitrary C++ |
| 1523 | types to Python, which can be done using :func:`cast`: |
| 1524 | |
| 1525 | .. code-block:: cpp |
| 1526 | |
| 1527 | MyClass *cls = ..; |
| 1528 | py::object obj = py::cast(cls); |
| 1529 | |
| 1530 | The reverse direction uses the following syntax: |
| 1531 | |
| 1532 | .. code-block:: cpp |
| 1533 | |
| 1534 | py::object obj = ...; |
| 1535 | MyClass *cls = obj.cast<MyClass *>(); |
| 1536 | |
| 1537 | When conversion fails, both directions throw the exception :class:`cast_error`. |
Wenzel Jakob | 178c8a8 | 2016-05-10 15:59:01 +0100 | [diff] [blame] | 1538 | It is also possible to call python functions via ``operator()``. |
| 1539 | |
| 1540 | .. code-block:: cpp |
| 1541 | |
| 1542 | py::function f = <...>; |
| 1543 | py::object result_py = f(1234, "hello", some_instance); |
| 1544 | MyClass &result = result_py.cast<MyClass>(); |
| 1545 | |
| 1546 | The special ``f(*args)`` and ``f(*args, **kwargs)`` syntax is also supported to |
| 1547 | supply arbitrary argument and keyword lists, although these cannot be mixed |
| 1548 | with other parameters. |
| 1549 | |
| 1550 | .. code-block:: cpp |
| 1551 | |
| 1552 | py::function f = <...>; |
| 1553 | py::tuple args = py::make_tuple(1234); |
| 1554 | py::dict kwargs; |
| 1555 | kwargs["y"] = py::cast(5678); |
| 1556 | py::object result = f(*args, **kwargs); |
Wenzel Jakob | 436b731 | 2015-10-20 01:04:30 +0200 | [diff] [blame] | 1557 | |
Wenzel Jakob | 9329669 | 2015-10-13 23:21:54 +0200 | [diff] [blame] | 1558 | .. seealso:: |
| 1559 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1560 | The file :file:`tests/test_python_types.cpp` contains a complete |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1561 | example that demonstrates passing native Python types in more detail. The |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1562 | file :file:`tests/test_kwargs_and_defaults.cpp` discusses usage |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1563 | of ``args`` and ``kwargs``. |
Wenzel Jakob | 2ac5044 | 2016-01-17 22:36:35 +0100 | [diff] [blame] | 1564 | |
| 1565 | Default arguments revisited |
| 1566 | =========================== |
| 1567 | |
| 1568 | The section on :ref:`default_args` previously discussed basic usage of default |
| 1569 | arguments using pybind11. One noteworthy aspect of their implementation is that |
| 1570 | default arguments are converted to Python objects right at declaration time. |
| 1571 | Consider the following example: |
| 1572 | |
| 1573 | .. code-block:: cpp |
| 1574 | |
| 1575 | py::class_<MyClass>("MyClass") |
| 1576 | .def("myFunction", py::arg("arg") = SomeType(123)); |
| 1577 | |
| 1578 | In this case, pybind11 must already be set up to deal with values of the type |
| 1579 | ``SomeType`` (via a prior instantiation of ``py::class_<SomeType>``), or an |
| 1580 | exception will be thrown. |
| 1581 | |
| 1582 | Another aspect worth highlighting is that the "preview" of the default argument |
| 1583 | in the function signature is generated using the object's ``__repr__`` method. |
| 1584 | If not available, the signature may not be very helpful, e.g.: |
| 1585 | |
Wenzel Jakob | 99279f7 | 2016-06-03 11:19:29 +0200 | [diff] [blame] | 1586 | .. code-block:: pycon |
Wenzel Jakob | 2ac5044 | 2016-01-17 22:36:35 +0100 | [diff] [blame] | 1587 | |
| 1588 | FUNCTIONS |
| 1589 | ... |
| 1590 | | myFunction(...) |
Wenzel Jakob | 48548ea | 2016-01-17 22:36:44 +0100 | [diff] [blame] | 1591 | | Signature : (MyClass, arg : SomeType = <SomeType object at 0x101b7b080>) -> NoneType |
Wenzel Jakob | 2ac5044 | 2016-01-17 22:36:35 +0100 | [diff] [blame] | 1592 | ... |
| 1593 | |
| 1594 | The first way of addressing this is by defining ``SomeType.__repr__``. |
| 1595 | Alternatively, it is possible to specify the human-readable preview of the |
| 1596 | default argument manually using the ``arg_t`` notation: |
| 1597 | |
| 1598 | .. code-block:: cpp |
| 1599 | |
| 1600 | py::class_<MyClass>("MyClass") |
| 1601 | .def("myFunction", py::arg_t<SomeType>("arg", SomeType(123), "SomeType(123)")); |
| 1602 | |
Wenzel Jakob | c769fce | 2016-03-03 12:03:30 +0100 | [diff] [blame] | 1603 | Sometimes it may be necessary to pass a null pointer value as a default |
| 1604 | argument. In this case, remember to cast it to the underlying type in question, |
| 1605 | like so: |
| 1606 | |
| 1607 | .. code-block:: cpp |
| 1608 | |
| 1609 | py::class_<MyClass>("MyClass") |
| 1610 | .def("myFunction", py::arg("arg") = (SomeType *) nullptr); |
| 1611 | |
Wenzel Jakob | 178c8a8 | 2016-05-10 15:59:01 +0100 | [diff] [blame] | 1612 | Binding functions that accept arbitrary numbers of arguments and keywords arguments |
| 1613 | =================================================================================== |
| 1614 | |
| 1615 | Python provides a useful mechanism to define functions that accept arbitrary |
| 1616 | numbers of arguments and keyword arguments: |
| 1617 | |
| 1618 | .. code-block:: cpp |
| 1619 | |
| 1620 | def generic(*args, **kwargs): |
| 1621 | # .. do something with args and kwargs |
| 1622 | |
| 1623 | Such functions can also be created using pybind11: |
| 1624 | |
| 1625 | .. code-block:: cpp |
| 1626 | |
| 1627 | void generic(py::args args, py::kwargs kwargs) { |
| 1628 | /// .. do something with args |
| 1629 | if (kwargs) |
| 1630 | /// .. do something with kwargs |
| 1631 | } |
| 1632 | |
| 1633 | /// Binding code |
| 1634 | m.def("generic", &generic); |
| 1635 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1636 | (See ``tests/test_kwargs_and_defaults.cpp``). The class ``py::args`` |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1637 | derives from ``py::list`` and ``py::kwargs`` derives from ``py::dict`` Note |
| 1638 | that the ``kwargs`` argument is invalid if no keyword arguments were actually |
| 1639 | provided. Please refer to the other examples for details on how to iterate |
| 1640 | over these, and on how to cast their entries into C++ objects. |
Wenzel Jakob | 178c8a8 | 2016-05-10 15:59:01 +0100 | [diff] [blame] | 1641 | |
Wenzel Jakob | 3764e28 | 2016-08-01 23:34:48 +0200 | [diff] [blame] | 1642 | .. warning:: |
| 1643 | |
| 1644 | Unlike Python, pybind11 does not allow combining normal parameters with the |
| 1645 | ``args`` / ``kwargs`` special parameters. |
| 1646 | |
Wenzel Jakob | 2dfbade | 2016-01-17 22:36:37 +0100 | [diff] [blame] | 1647 | Partitioning code over multiple extension modules |
| 1648 | ================================================= |
| 1649 | |
Wenzel Jakob | 90d2f5e | 2016-04-11 14:30:11 +0200 | [diff] [blame] | 1650 | It's straightforward to split binding code over multiple extension modules, |
| 1651 | while referencing types that are declared elsewhere. Everything "just" works |
| 1652 | without any special precautions. One exception to this rule occurs when |
| 1653 | extending a type declared in another extension module. Recall the basic example |
| 1654 | from Section :ref:`inheritance`. |
Wenzel Jakob | 2dfbade | 2016-01-17 22:36:37 +0100 | [diff] [blame] | 1655 | |
| 1656 | .. code-block:: cpp |
| 1657 | |
| 1658 | py::class_<Pet> pet(m, "Pet"); |
| 1659 | pet.def(py::init<const std::string &>()) |
| 1660 | .def_readwrite("name", &Pet::name); |
| 1661 | |
| 1662 | py::class_<Dog>(m, "Dog", pet /* <- specify parent */) |
| 1663 | .def(py::init<const std::string &>()) |
| 1664 | .def("bark", &Dog::bark); |
| 1665 | |
| 1666 | Suppose now that ``Pet`` bindings are defined in a module named ``basic``, |
| 1667 | whereas the ``Dog`` bindings are defined somewhere else. The challenge is of |
| 1668 | course that the variable ``pet`` is not available anymore though it is needed |
| 1669 | to indicate the inheritance relationship to the constructor of ``class_<Dog>``. |
| 1670 | However, it can be acquired as follows: |
| 1671 | |
| 1672 | .. code-block:: cpp |
| 1673 | |
| 1674 | py::object pet = (py::object) py::module::import("basic").attr("Pet"); |
| 1675 | |
| 1676 | py::class_<Dog>(m, "Dog", pet) |
| 1677 | .def(py::init<const std::string &>()) |
| 1678 | .def("bark", &Dog::bark); |
| 1679 | |
Wenzel Jakob | 8d862b3 | 2016-03-06 13:37:22 +0100 | [diff] [blame] | 1680 | Alternatively, we can rely on the ``base`` tag, which performs an automated |
| 1681 | lookup of the corresponding Python type. However, this also requires invoking |
| 1682 | the ``import`` function once to ensure that the pybind11 binding code of the |
| 1683 | module ``basic`` has been executed. |
| 1684 | |
Wenzel Jakob | 8d862b3 | 2016-03-06 13:37:22 +0100 | [diff] [blame] | 1685 | .. code-block:: cpp |
| 1686 | |
| 1687 | py::module::import("basic"); |
| 1688 | |
| 1689 | py::class_<Dog>(m, "Dog", py::base<Pet>()) |
| 1690 | .def(py::init<const std::string &>()) |
| 1691 | .def("bark", &Dog::bark); |
Wenzel Jakob | eda978e | 2016-03-15 15:05:40 +0100 | [diff] [blame] | 1692 | |
Wenzel Jakob | 978e376 | 2016-04-07 18:00:41 +0200 | [diff] [blame] | 1693 | Naturally, both methods will fail when there are cyclic dependencies. |
| 1694 | |
Wenzel Jakob | 90d2f5e | 2016-04-11 14:30:11 +0200 | [diff] [blame] | 1695 | Note that compiling code which has its default symbol visibility set to |
| 1696 | *hidden* (e.g. via the command line flag ``-fvisibility=hidden`` on GCC/Clang) can interfere with the |
| 1697 | ability to access types defined in another extension module. Workarounds |
| 1698 | include changing the global symbol visibility (not recommended, because it will |
| 1699 | lead unnecessarily large binaries) or manually exporting types that are |
| 1700 | accessed by multiple extension modules: |
| 1701 | |
| 1702 | .. code-block:: cpp |
| 1703 | |
| 1704 | #ifdef _WIN32 |
| 1705 | # define EXPORT_TYPE __declspec(dllexport) |
| 1706 | #else |
| 1707 | # define EXPORT_TYPE __attribute__ ((visibility("default"))) |
| 1708 | #endif |
| 1709 | |
| 1710 | class EXPORT_TYPE Dog : public Animal { |
| 1711 | ... |
| 1712 | }; |
| 1713 | |
| 1714 | |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1715 | Pickling support |
| 1716 | ================ |
| 1717 | |
| 1718 | Python's ``pickle`` module provides a powerful facility to serialize and |
| 1719 | de-serialize a Python object graph into a binary data stream. To pickle and |
Wenzel Jakob | 3d0e6ff | 2016-04-13 11:48:10 +0200 | [diff] [blame] | 1720 | unpickle C++ classes using pybind11, two additional functions must be provided. |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1721 | Suppose the class in question has the following signature: |
| 1722 | |
| 1723 | .. code-block:: cpp |
| 1724 | |
| 1725 | class Pickleable { |
| 1726 | public: |
| 1727 | Pickleable(const std::string &value) : m_value(value) { } |
| 1728 | const std::string &value() const { return m_value; } |
| 1729 | |
| 1730 | void setExtra(int extra) { m_extra = extra; } |
| 1731 | int extra() const { return m_extra; } |
| 1732 | private: |
| 1733 | std::string m_value; |
| 1734 | int m_extra = 0; |
| 1735 | }; |
| 1736 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1737 | The binding code including the requisite ``__setstate__`` and ``__getstate__`` methods [#f3]_ |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1738 | looks as follows: |
| 1739 | |
| 1740 | .. code-block:: cpp |
| 1741 | |
| 1742 | py::class_<Pickleable>(m, "Pickleable") |
| 1743 | .def(py::init<std::string>()) |
| 1744 | .def("value", &Pickleable::value) |
| 1745 | .def("extra", &Pickleable::extra) |
| 1746 | .def("setExtra", &Pickleable::setExtra) |
| 1747 | .def("__getstate__", [](const Pickleable &p) { |
| 1748 | /* Return a tuple that fully encodes the state of the object */ |
| 1749 | return py::make_tuple(p.value(), p.extra()); |
| 1750 | }) |
| 1751 | .def("__setstate__", [](Pickleable &p, py::tuple t) { |
| 1752 | if (t.size() != 2) |
| 1753 | throw std::runtime_error("Invalid state!"); |
| 1754 | |
Wenzel Jakob | d40885a | 2016-04-13 13:30:05 +0200 | [diff] [blame] | 1755 | /* Invoke the in-place constructor. Note that this is needed even |
| 1756 | when the object just has a trivial default constructor */ |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1757 | new (&p) Pickleable(t[0].cast<std::string>()); |
| 1758 | |
| 1759 | /* Assign any additional state */ |
| 1760 | p.setExtra(t[1].cast<int>()); |
| 1761 | }); |
| 1762 | |
| 1763 | An instance can now be pickled as follows: |
| 1764 | |
| 1765 | .. code-block:: python |
| 1766 | |
| 1767 | try: |
| 1768 | import cPickle as pickle # Use cPickle on Python 2.7 |
| 1769 | except ImportError: |
| 1770 | import pickle |
| 1771 | |
| 1772 | p = Pickleable("test_value") |
| 1773 | p.setExtra(15) |
Wenzel Jakob | 81e0975 | 2016-04-30 23:13:03 +0200 | [diff] [blame] | 1774 | data = pickle.dumps(p, 2) |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1775 | |
Wenzel Jakob | 81e0975 | 2016-04-30 23:13:03 +0200 | [diff] [blame] | 1776 | Note that only the cPickle module is supported on Python 2.7. The second |
| 1777 | argument to ``dumps`` is also crucial: it selects the pickle protocol version |
| 1778 | 2, since the older version 1 is not supported. Newer versions are also fineāfor |
| 1779 | instance, specify ``-1`` to always use the latest available version. Beware: |
| 1780 | failure to follow these instructions will cause important pybind11 memory |
| 1781 | allocation routines to be skipped during unpickling, which will likely lead to |
| 1782 | memory corruption and/or segmentation faults. |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1783 | |
| 1784 | .. seealso:: |
| 1785 | |
Dean Moldovan | ec0d38e | 2016-08-13 03:09:52 +0200 | [diff] [blame^] | 1786 | The file :file:`tests/test_pickling.cpp` contains a complete example |
Jason Rhinelander | 3e2e44f | 2016-07-18 17:03:37 -0400 | [diff] [blame] | 1787 | that demonstrates how to pickle and unpickle types using pybind11 in more |
| 1788 | detail. |
Wenzel Jakob | 1c329aa | 2016-04-13 02:37:36 +0200 | [diff] [blame] | 1789 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1790 | .. [#f3] http://docs.python.org/3/library/pickle.html#pickling-class-instances |
Wenzel Jakob | ef7a9b9 | 2016-04-13 18:41:59 +0200 | [diff] [blame] | 1791 | |
| 1792 | Generating documentation using Sphinx |
| 1793 | ===================================== |
| 1794 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1795 | Sphinx [#f4]_ has the ability to inspect the signatures and documentation |
Wenzel Jakob | ef7a9b9 | 2016-04-13 18:41:59 +0200 | [diff] [blame] | 1796 | strings in pybind11-based extension modules to automatically generate beautiful |
Wenzel Jakob | ca8dc08 | 2016-06-03 14:24:17 +0200 | [diff] [blame] | 1797 | documentation in a variety formats. The python_example repository [#f5]_ contains a |
Wenzel Jakob | ef7a9b9 | 2016-04-13 18:41:59 +0200 | [diff] [blame] | 1798 | simple example repository which uses this approach. |
| 1799 | |
| 1800 | There are two potential gotchas when using this approach: first, make sure that |
| 1801 | the resulting strings do not contain any :kbd:`TAB` characters, which break the |
| 1802 | docstring parsing routines. You may want to use C++11 raw string literals, |
| 1803 | which are convenient for multi-line comments. Conveniently, any excess |
| 1804 | indentation will be automatically be removed by Sphinx. However, for this to |
| 1805 | work, it is important that all lines are indented consistently, i.e.: |
| 1806 | |
| 1807 | .. code-block:: cpp |
| 1808 | |
| 1809 | // ok |
| 1810 | m.def("foo", &foo, R"mydelimiter( |
| 1811 | The foo function |
| 1812 | |
| 1813 | Parameters |
| 1814 | ---------- |
| 1815 | )mydelimiter"); |
| 1816 | |
| 1817 | // *not ok* |
| 1818 | m.def("foo", &foo, R"mydelimiter(The foo function |
| 1819 | |
| 1820 | Parameters |
| 1821 | ---------- |
| 1822 | )mydelimiter"); |
| 1823 | |
Wenzel Jakob | 9e0a056 | 2016-05-05 20:33:54 +0200 | [diff] [blame] | 1824 | .. [#f4] http://www.sphinx-doc.org |
Wenzel Jakob | ca8dc08 | 2016-06-03 14:24:17 +0200 | [diff] [blame] | 1825 | .. [#f5] http://github.com/pybind/python_example |
Klemens Morgenstern | c6ad2c4 | 2016-06-09 16:10:26 +0200 | [diff] [blame] | 1826 | |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 1827 | Evaluating Python expressions from strings and files |
| 1828 | ==================================================== |
Klemens Morgenstern | c6ad2c4 | 2016-06-09 16:10:26 +0200 | [diff] [blame] | 1829 | |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 1830 | pybind11 provides the :func:`eval` and :func:`eval_file` functions to evaluate |
| 1831 | Python expressions and statements. The following example illustrates how they |
| 1832 | can be used. |
| 1833 | |
| 1834 | Both functions accept a template parameter that describes how the argument |
| 1835 | should be interpreted. Possible choices include ``eval_expr`` (isolated |
| 1836 | expression), ``eval_single_statement`` (a single statement, return value is |
| 1837 | always ``none``), and ``eval_statements`` (sequence of statements, return value |
| 1838 | is always ``none``). |
Klemens Morgenstern | c6ad2c4 | 2016-06-09 16:10:26 +0200 | [diff] [blame] | 1839 | |
| 1840 | .. code-block:: cpp |
| 1841 | |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 1842 | // At beginning of file |
| 1843 | #include <pybind11/eval.h> |
Klemens Morgenstern | c6ad2c4 | 2016-06-09 16:10:26 +0200 | [diff] [blame] | 1844 | |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 1845 | ... |
Klemens Morgenstern | c6ad2c4 | 2016-06-09 16:10:26 +0200 | [diff] [blame] | 1846 | |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 1847 | // Evaluate in scope of main module |
| 1848 | py::object scope = py::module::import("__main__").attr("__dict__"); |
Klemens Morgenstern | c6ad2c4 | 2016-06-09 16:10:26 +0200 | [diff] [blame] | 1849 | |
Wenzel Jakob | 0d3fc35 | 2016-07-08 10:52:10 +0200 | [diff] [blame] | 1850 | // Evaluate an isolated expression |
| 1851 | int result = py::eval("my_variable + 10", scope).cast<int>(); |
| 1852 | |
| 1853 | // Evaluate a sequence of statements |
| 1854 | py::eval<py::eval_statements>( |
| 1855 | "print('Hello')\n" |
| 1856 | "print('world!');", |
| 1857 | scope); |
| 1858 | |
| 1859 | // Evaluate the statements in an separate Python file on disk |
| 1860 | py::eval_file("script.py", scope); |