65 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			C++
		
	
	
			
		
		
	
	
			65 lines
		
	
	
		
			2.1 KiB
		
	
	
	
		
			C++
		
	
	
#pragma once
 | 
						|
#include "pybind11_tests.h"
 | 
						|
 | 
						|
/// Simple class used to test py::local:
 | 
						|
template <int> class LocalBase {
 | 
						|
public:
 | 
						|
    LocalBase(int i) : i(i) { }
 | 
						|
    int i = -1;
 | 
						|
};
 | 
						|
 | 
						|
/// Registered with py::module_local in both main and secondary modules:
 | 
						|
using LocalType = LocalBase<0>;
 | 
						|
/// Registered without py::module_local in both modules:
 | 
						|
using NonLocalType = LocalBase<1>;
 | 
						|
/// A second non-local type (for stl_bind tests):
 | 
						|
using NonLocal2 = LocalBase<2>;
 | 
						|
/// Tests within-module, different-compilation-unit local definition conflict:
 | 
						|
using LocalExternal = LocalBase<3>;
 | 
						|
/// Mixed: registered local first, then global
 | 
						|
using MixedLocalGlobal = LocalBase<4>;
 | 
						|
/// Mixed: global first, then local
 | 
						|
using MixedGlobalLocal = LocalBase<5>;
 | 
						|
 | 
						|
/// Registered with py::module_local only in the secondary module:
 | 
						|
using ExternalType1 = LocalBase<6>;
 | 
						|
using ExternalType2 = LocalBase<7>;
 | 
						|
 | 
						|
using LocalVec = std::vector<LocalType>;
 | 
						|
using LocalVec2 = std::vector<NonLocal2>;
 | 
						|
using LocalMap = std::unordered_map<std::string, LocalType>;
 | 
						|
using NonLocalVec = std::vector<NonLocalType>;
 | 
						|
using NonLocalVec2 = std::vector<NonLocal2>;
 | 
						|
using NonLocalMap = std::unordered_map<std::string, NonLocalType>;
 | 
						|
using NonLocalMap2 = std::unordered_map<std::string, uint8_t>;
 | 
						|
 | 
						|
PYBIND11_MAKE_OPAQUE(LocalVec);
 | 
						|
PYBIND11_MAKE_OPAQUE(LocalVec2);
 | 
						|
PYBIND11_MAKE_OPAQUE(LocalMap);
 | 
						|
PYBIND11_MAKE_OPAQUE(NonLocalVec);
 | 
						|
//PYBIND11_MAKE_OPAQUE(NonLocalVec2); // same type as LocalVec2
 | 
						|
PYBIND11_MAKE_OPAQUE(NonLocalMap);
 | 
						|
PYBIND11_MAKE_OPAQUE(NonLocalMap2);
 | 
						|
 | 
						|
 | 
						|
// Simple bindings (used with the above):
 | 
						|
template <typename T, int Adjust = 0, typename... Args>
 | 
						|
py::class_<T> bind_local(Args && ...args) {
 | 
						|
    return py::class_<T>(std::forward<Args>(args)...)
 | 
						|
        .def(py::init<int>())
 | 
						|
        .def("get", [](T &i) { return i.i + Adjust; });
 | 
						|
};
 | 
						|
 | 
						|
// Simulate a foreign library base class (to match the example in the docs):
 | 
						|
namespace pets {
 | 
						|
class Pet {
 | 
						|
public:
 | 
						|
    Pet(std::string name) : name_(name) {}
 | 
						|
    std::string name_;
 | 
						|
    const std::string &name() { return name_; }
 | 
						|
};
 | 
						|
} // namespace pets
 | 
						|
 | 
						|
struct MixGL { int i; MixGL(int i) : i{i} {} };
 | 
						|
struct MixGL2 { int i; MixGL2(int i) : i{i} {} };
 |