1#pragma once
2#include "pybind11_tests.h"
3
4/// Simple class used to test py::local:
5template <int> class LocalBase {
6public:
7    LocalBase(int i) : i(i) { }
8    int i = -1;
9};
10
11/// Registered with py::module_local in both main and secondary modules:
12using LocalType = LocalBase<0>;
13/// Registered without py::module_local in both modules:
14using NonLocalType = LocalBase<1>;
15/// A second non-local type (for stl_bind tests):
16using NonLocal2 = LocalBase<2>;
17/// Tests within-module, different-compilation-unit local definition conflict:
18using LocalExternal = LocalBase<3>;
19/// Mixed: registered local first, then global
20using MixedLocalGlobal = LocalBase<4>;
21/// Mixed: global first, then local
22using MixedGlobalLocal = LocalBase<5>;
23
24/// Registered with py::module_local only in the secondary module:
25using ExternalType1 = LocalBase<6>;
26using ExternalType2 = LocalBase<7>;
27
28using LocalVec = std::vector<LocalType>;
29using LocalVec2 = std::vector<NonLocal2>;
30using LocalMap = std::unordered_map<std::string, LocalType>;
31using NonLocalVec = std::vector<NonLocalType>;
32using NonLocalVec2 = std::vector<NonLocal2>;
33using NonLocalMap = std::unordered_map<std::string, NonLocalType>;
34using NonLocalMap2 = std::unordered_map<std::string, uint8_t>;
35
36PYBIND11_MAKE_OPAQUE(LocalVec);
37PYBIND11_MAKE_OPAQUE(LocalVec2);
38PYBIND11_MAKE_OPAQUE(LocalMap);
39PYBIND11_MAKE_OPAQUE(NonLocalVec);
40//PYBIND11_MAKE_OPAQUE(NonLocalVec2); // same type as LocalVec2
41PYBIND11_MAKE_OPAQUE(NonLocalMap);
42PYBIND11_MAKE_OPAQUE(NonLocalMap2);
43
44
45// Simple bindings (used with the above):
46template <typename T, int Adjust = 0, typename... Args>
47py::class_<T> bind_local(Args && ...args) {
48    return py::class_<T>(std::forward<Args>(args)...)
49        .def(py::init<int>())
50        .def("get", [](T &i) { return i.i + Adjust; });
51};
52
53// Simulate a foreign library base class (to match the example in the docs):
54namespace pets {
55class Pet {
56public:
57    Pet(std::string name) : name_(name) {}
58    std::string name_;
59    const std::string &name() { return name_; }
60};
61}
62
63struct MixGL { int i; MixGL(int i) : i{i} {} };
64struct MixGL2 { int i; MixGL2(int i) : i{i} {} };
65