Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 1 | /* |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 2 | tests/test_stl_binders.cpp -- Usage of stl_binders functions |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 3 | |
Wenzel Jakob | 25c03ce | 2016-05-15 20:50:38 +0200 | [diff] [blame] | 4 | Copyright (c) 2016 Sergey Lyskov |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 5 | |
| 6 | All rights reserved. Use of this source code is governed by a |
| 7 | BSD-style license that can be found in the LICENSE file. |
| 8 | */ |
| 9 | |
Dean Moldovan | a0c1ccf | 2016-08-12 13:50:00 +0200 | [diff] [blame] | 10 | #include "pybind11_tests.h" |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 11 | |
Wenzel Jakob | 25c03ce | 2016-05-15 20:50:38 +0200 | [diff] [blame] | 12 | #include <pybind11/stl_bind.h> |
Patrick Stewart | 5467979 | 2016-11-08 13:03:34 +0000 | [diff] [blame] | 13 | #include <pybind11/numpy.h> |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 14 | #include <map> |
Jason Rhinelander | 617fbcf | 2016-11-15 06:30:38 -0500 | [diff] [blame] | 15 | #include <deque> |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 16 | #include <unordered_map> |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 17 | |
Jason Rhinelander | fa5d05e | 2016-12-12 18:11:49 -0500 | [diff] [blame] | 18 | #ifdef _MSC_VER |
| 19 | // We get some really long type names here which causes MSVC to emit warnings |
| 20 | # pragma warning(disable: 4503) // warning C4503: decorated name length exceeded, name was truncated |
| 21 | #endif |
| 22 | |
Wenzel Jakob | 25c03ce | 2016-05-15 20:50:38 +0200 | [diff] [blame] | 23 | class El { |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 24 | public: |
Jason Rhinelander | 540ae61 | 2016-08-28 13:46:25 -0400 | [diff] [blame] | 25 | El() = delete; |
| 26 | El(int v) : a(v) { } |
Sergey Lyskov | a95bde1 | 2016-05-08 19:31:55 -0400 | [diff] [blame] | 27 | |
Jason Rhinelander | 540ae61 | 2016-08-28 13:46:25 -0400 | [diff] [blame] | 28 | int a; |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 29 | }; |
| 30 | |
Wenzel Jakob | 25c03ce | 2016-05-15 20:50:38 +0200 | [diff] [blame] | 31 | std::ostream & operator<<(std::ostream &s, El const&v) { |
Jason Rhinelander | 540ae61 | 2016-08-28 13:46:25 -0400 | [diff] [blame] | 32 | s << "El{" << v.a << '}'; |
| 33 | return s; |
Sergey Lyskov | a95bde1 | 2016-05-08 19:31:55 -0400 | [diff] [blame] | 34 | } |
Sergey Lyskov | eae7744 | 2016-05-07 00:26:19 -0400 | [diff] [blame] | 35 | |
Jason Rhinelander | 617fbcf | 2016-11-15 06:30:38 -0500 | [diff] [blame] | 36 | /// Issue #487: binding std::vector<E> with E non-copyable |
| 37 | class E_nc { |
| 38 | public: |
| 39 | explicit E_nc(int i) : value{i} {} |
| 40 | E_nc(const E_nc &) = delete; |
| 41 | E_nc &operator=(const E_nc &) = delete; |
| 42 | E_nc(E_nc &&) = default; |
| 43 | E_nc &operator=(E_nc &&) = default; |
| 44 | |
| 45 | int value; |
| 46 | }; |
| 47 | |
| 48 | template <class Container> Container *one_to_n(int n) { |
| 49 | auto v = new Container(); |
| 50 | for (int i = 1; i <= n; i++) |
| 51 | v->emplace_back(i); |
| 52 | return v; |
| 53 | } |
| 54 | |
| 55 | template <class Map> Map *times_ten(int n) { |
| 56 | auto m = new Map(); |
| 57 | for (int i = 1; i <= n; i++) |
| 58 | m->emplace(int(i), E_nc(10*i)); |
| 59 | return m; |
| 60 | } |
| 61 | |
Patrick Stewart | 5467979 | 2016-11-08 13:03:34 +0000 | [diff] [blame] | 62 | struct VStruct { |
| 63 | bool w; |
| 64 | uint32_t x; |
| 65 | double y; |
| 66 | bool z; |
| 67 | }; |
| 68 | |
| 69 | struct VUndeclStruct { //dtype not declared for this version |
| 70 | bool w; |
| 71 | uint32_t x; |
| 72 | double y; |
| 73 | bool z; |
| 74 | }; |
| 75 | |
Jason Rhinelander | 52f4be8 | 2016-09-03 14:54:22 -0400 | [diff] [blame] | 76 | test_initializer stl_binder_vector([](py::module &m) { |
Jason Rhinelander | 540ae61 | 2016-08-28 13:46:25 -0400 | [diff] [blame] | 77 | py::class_<El>(m, "El") |
| 78 | .def(py::init<int>()); |
Sergey Lyskov | a95bde1 | 2016-05-08 19:31:55 -0400 | [diff] [blame] | 79 | |
Patrick Stewart | 5467979 | 2016-11-08 13:03:34 +0000 | [diff] [blame] | 80 | py::bind_vector<std::vector<unsigned char>>(m, "VectorUChar", py::buffer_protocol()); |
| 81 | py::bind_vector<std::vector<unsigned int>>(m, "VectorInt", py::buffer_protocol()); |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 82 | py::bind_vector<std::vector<bool>>(m, "VectorBool"); |
Sergey Lyskov | a95bde1 | 2016-05-08 19:31:55 -0400 | [diff] [blame] | 83 | |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 84 | py::bind_vector<std::vector<El>>(m, "VectorEl"); |
Wenzel Jakob | 25c03ce | 2016-05-15 20:50:38 +0200 | [diff] [blame] | 85 | |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 86 | py::bind_vector<std::vector<std::vector<El>>>(m, "VectorVectorEl"); |
Jason Rhinelander | 617fbcf | 2016-11-15 06:30:38 -0500 | [diff] [blame] | 87 | |
Patrick Stewart | 5467979 | 2016-11-08 13:03:34 +0000 | [diff] [blame] | 88 | m.def("create_undeclstruct", [m] () mutable { |
| 89 | py::bind_vector<std::vector<VUndeclStruct>>(m, "VectorUndeclStruct", py::buffer_protocol()); |
| 90 | }); |
| 91 | |
| 92 | try { |
| 93 | py::module::import("numpy"); |
| 94 | } catch (...) { |
| 95 | return; |
| 96 | } |
| 97 | PYBIND11_NUMPY_DTYPE(VStruct, w, x, y, z); |
| 98 | py::class_<VStruct>(m, "VStruct").def_readwrite("x", &VStruct::x); |
| 99 | py::bind_vector<std::vector<VStruct>>(m, "VectorStruct", py::buffer_protocol()); |
| 100 | m.def("get_vectorstruct", [] {return std::vector<VStruct> {{0, 5, 3.0, 1}, {1, 30, -1e4, 0}};}); |
Sergey Lyskov | 7520418 | 2016-08-29 22:50:38 -0400 | [diff] [blame] | 101 | }); |
| 102 | |
| 103 | test_initializer stl_binder_map([](py::module &m) { |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 104 | py::bind_map<std::map<std::string, double>>(m, "MapStringDouble"); |
| 105 | py::bind_map<std::unordered_map<std::string, double>>(m, "UnorderedMapStringDouble"); |
Sergey Lyskov | 7520418 | 2016-08-29 22:50:38 -0400 | [diff] [blame] | 106 | |
Wenzel Jakob | fe34241 | 2016-09-06 13:02:29 +0900 | [diff] [blame] | 107 | py::bind_map<std::map<std::string, double const>>(m, "MapStringDoubleConst"); |
| 108 | py::bind_map<std::unordered_map<std::string, double const>>(m, "UnorderedMapStringDoubleConst"); |
Jason Rhinelander | 617fbcf | 2016-11-15 06:30:38 -0500 | [diff] [blame] | 109 | |
Jason Rhinelander | 52f4be8 | 2016-09-03 14:54:22 -0400 | [diff] [blame] | 110 | }); |
Jason Rhinelander | 617fbcf | 2016-11-15 06:30:38 -0500 | [diff] [blame] | 111 | |
| 112 | test_initializer stl_binder_noncopyable([](py::module &m) { |
| 113 | py::class_<E_nc>(m, "ENC") |
| 114 | .def(py::init<int>()) |
| 115 | .def_readwrite("value", &E_nc::value); |
| 116 | |
| 117 | py::bind_vector<std::vector<E_nc>>(m, "VectorENC"); |
| 118 | m.def("get_vnc", &one_to_n<std::vector<E_nc>>, py::return_value_policy::reference); |
| 119 | |
| 120 | py::bind_vector<std::deque<E_nc>>(m, "DequeENC"); |
| 121 | m.def("get_dnc", &one_to_n<std::deque<E_nc>>, py::return_value_policy::reference); |
| 122 | |
| 123 | py::bind_map<std::map<int, E_nc>>(m, "MapENC"); |
| 124 | m.def("get_mnc", ×_ten<std::map<int, E_nc>>, py::return_value_policy::reference); |
| 125 | |
| 126 | py::bind_map<std::unordered_map<int, E_nc>>(m, "UmapENC"); |
| 127 | m.def("get_umnc", ×_ten<std::unordered_map<int, E_nc>>, py::return_value_policy::reference); |
| 128 | }); |