111986Sandreas.sandberg@arm.com/* 211986Sandreas.sandberg@arm.com tests/test_stl_binders.cpp -- Usage of stl_binders functions 311986Sandreas.sandberg@arm.com 411986Sandreas.sandberg@arm.com Copyright (c) 2016 Sergey Lyskov 511986Sandreas.sandberg@arm.com 611986Sandreas.sandberg@arm.com All rights reserved. Use of this source code is governed by a 711986Sandreas.sandberg@arm.com BSD-style license that can be found in the LICENSE file. 811986Sandreas.sandberg@arm.com*/ 911986Sandreas.sandberg@arm.com 1011986Sandreas.sandberg@arm.com#include "pybind11_tests.h" 1111986Sandreas.sandberg@arm.com 1211986Sandreas.sandberg@arm.com#include <pybind11/stl_bind.h> 1312037Sandreas.sandberg@arm.com#include <pybind11/numpy.h> 1411986Sandreas.sandberg@arm.com#include <map> 1511986Sandreas.sandberg@arm.com#include <deque> 1611986Sandreas.sandberg@arm.com#include <unordered_map> 1711986Sandreas.sandberg@arm.com 1811986Sandreas.sandberg@arm.comclass El { 1911986Sandreas.sandberg@arm.compublic: 2011986Sandreas.sandberg@arm.com El() = delete; 2111986Sandreas.sandberg@arm.com El(int v) : a(v) { } 2211986Sandreas.sandberg@arm.com 2311986Sandreas.sandberg@arm.com int a; 2411986Sandreas.sandberg@arm.com}; 2511986Sandreas.sandberg@arm.com 2611986Sandreas.sandberg@arm.comstd::ostream & operator<<(std::ostream &s, El const&v) { 2711986Sandreas.sandberg@arm.com s << "El{" << v.a << '}'; 2811986Sandreas.sandberg@arm.com return s; 2911986Sandreas.sandberg@arm.com} 3011986Sandreas.sandberg@arm.com 3111986Sandreas.sandberg@arm.com/// Issue #487: binding std::vector<E> with E non-copyable 3211986Sandreas.sandberg@arm.comclass E_nc { 3311986Sandreas.sandberg@arm.compublic: 3411986Sandreas.sandberg@arm.com explicit E_nc(int i) : value{i} {} 3511986Sandreas.sandberg@arm.com E_nc(const E_nc &) = delete; 3611986Sandreas.sandberg@arm.com E_nc &operator=(const E_nc &) = delete; 3711986Sandreas.sandberg@arm.com E_nc(E_nc &&) = default; 3811986Sandreas.sandberg@arm.com E_nc &operator=(E_nc &&) = default; 3911986Sandreas.sandberg@arm.com 4011986Sandreas.sandberg@arm.com int value; 4111986Sandreas.sandberg@arm.com}; 4211986Sandreas.sandberg@arm.com 4311986Sandreas.sandberg@arm.comtemplate <class Container> Container *one_to_n(int n) { 4411986Sandreas.sandberg@arm.com auto v = new Container(); 4511986Sandreas.sandberg@arm.com for (int i = 1; i <= n; i++) 4611986Sandreas.sandberg@arm.com v->emplace_back(i); 4711986Sandreas.sandberg@arm.com return v; 4811986Sandreas.sandberg@arm.com} 4911986Sandreas.sandberg@arm.com 5011986Sandreas.sandberg@arm.comtemplate <class Map> Map *times_ten(int n) { 5111986Sandreas.sandberg@arm.com auto m = new Map(); 5211986Sandreas.sandberg@arm.com for (int i = 1; i <= n; i++) 5311986Sandreas.sandberg@arm.com m->emplace(int(i), E_nc(10*i)); 5411986Sandreas.sandberg@arm.com return m; 5511986Sandreas.sandberg@arm.com} 5611986Sandreas.sandberg@arm.com 5712391Sjason@lowepower.comTEST_SUBMODULE(stl_binders, m) { 5812391Sjason@lowepower.com // test_vector_int 5912391Sjason@lowepower.com py::bind_vector<std::vector<unsigned int>>(m, "VectorInt", py::buffer_protocol()); 6012037Sandreas.sandberg@arm.com 6112391Sjason@lowepower.com // test_vector_custom 6211986Sandreas.sandberg@arm.com py::class_<El>(m, "El") 6311986Sandreas.sandberg@arm.com .def(py::init<int>()); 6411986Sandreas.sandberg@arm.com py::bind_vector<std::vector<El>>(m, "VectorEl"); 6511986Sandreas.sandberg@arm.com py::bind_vector<std::vector<std::vector<El>>>(m, "VectorVectorEl"); 6611986Sandreas.sandberg@arm.com 6712391Sjason@lowepower.com // test_map_string_double 6812391Sjason@lowepower.com py::bind_map<std::map<std::string, double>>(m, "MapStringDouble"); 6912391Sjason@lowepower.com py::bind_map<std::unordered_map<std::string, double>>(m, "UnorderedMapStringDouble"); 7012391Sjason@lowepower.com 7112391Sjason@lowepower.com // test_map_string_double_const 7212391Sjason@lowepower.com py::bind_map<std::map<std::string, double const>>(m, "MapStringDoubleConst"); 7312391Sjason@lowepower.com py::bind_map<std::unordered_map<std::string, double const>>(m, "UnorderedMapStringDoubleConst"); 7412391Sjason@lowepower.com 7512391Sjason@lowepower.com py::class_<E_nc>(m, "ENC") 7612391Sjason@lowepower.com .def(py::init<int>()) 7712391Sjason@lowepower.com .def_readwrite("value", &E_nc::value); 7812391Sjason@lowepower.com 7912391Sjason@lowepower.com // test_noncopyable_containers 8012391Sjason@lowepower.com py::bind_vector<std::vector<E_nc>>(m, "VectorENC"); 8112391Sjason@lowepower.com m.def("get_vnc", &one_to_n<std::vector<E_nc>>, py::return_value_policy::reference); 8212391Sjason@lowepower.com py::bind_vector<std::deque<E_nc>>(m, "DequeENC"); 8312391Sjason@lowepower.com m.def("get_dnc", &one_to_n<std::deque<E_nc>>, py::return_value_policy::reference); 8412391Sjason@lowepower.com py::bind_map<std::map<int, E_nc>>(m, "MapENC"); 8512391Sjason@lowepower.com m.def("get_mnc", ×_ten<std::map<int, E_nc>>, py::return_value_policy::reference); 8612391Sjason@lowepower.com py::bind_map<std::unordered_map<int, E_nc>>(m, "UmapENC"); 8712391Sjason@lowepower.com m.def("get_umnc", ×_ten<std::unordered_map<int, E_nc>>, py::return_value_policy::reference); 8812391Sjason@lowepower.com 8912391Sjason@lowepower.com // test_vector_buffer 9012391Sjason@lowepower.com py::bind_vector<std::vector<unsigned char>>(m, "VectorUChar", py::buffer_protocol()); 9112391Sjason@lowepower.com // no dtype declared for this version: 9212391Sjason@lowepower.com struct VUndeclStruct { bool w; uint32_t x; double y; bool z; }; 9312037Sandreas.sandberg@arm.com m.def("create_undeclstruct", [m] () mutable { 9412037Sandreas.sandberg@arm.com py::bind_vector<std::vector<VUndeclStruct>>(m, "VectorUndeclStruct", py::buffer_protocol()); 9512037Sandreas.sandberg@arm.com }); 9612037Sandreas.sandberg@arm.com 9712391Sjason@lowepower.com // The rest depends on numpy: 9812391Sjason@lowepower.com try { py::module::import("numpy"); } 9912391Sjason@lowepower.com catch (...) { return; } 10012391Sjason@lowepower.com 10112391Sjason@lowepower.com // test_vector_buffer_numpy 10212391Sjason@lowepower.com struct VStruct { bool w; uint32_t x; double y; bool z; }; 10312037Sandreas.sandberg@arm.com PYBIND11_NUMPY_DTYPE(VStruct, w, x, y, z); 10412037Sandreas.sandberg@arm.com py::class_<VStruct>(m, "VStruct").def_readwrite("x", &VStruct::x); 10512037Sandreas.sandberg@arm.com py::bind_vector<std::vector<VStruct>>(m, "VectorStruct", py::buffer_protocol()); 10612037Sandreas.sandberg@arm.com m.def("get_vectorstruct", [] {return std::vector<VStruct> {{0, 5, 3.0, 1}, {1, 30, -1e4, 0}};}); 10712391Sjason@lowepower.com} 108