| namespace greenlet | |
| { | |
| // This allocator is stateless; all instances are identical. | |
| // It can *ONLY* be used when we're sure we're holding the GIL | |
| // (Python's allocators require the GIL). | |
| template <class T> | |
| struct PythonAllocator : public std::allocator<T> { | |
| PythonAllocator(const PythonAllocator& UNUSED(other)) | |
| : std::allocator<T>() | |
| { | |
| } | |
| PythonAllocator(const std::allocator<T> other) | |
| : std::allocator<T>(other) | |
| {} | |
| template <class U> | |
| PythonAllocator(const std::allocator<U>& other) | |
| : std::allocator<T>(other) | |
| { | |
| } | |
| PythonAllocator() : std::allocator<T>() {} | |
| T* allocate(size_t number_objects, const void* UNUSED(hint)=0) | |
| { | |
| void* p; | |
| if (number_objects == 1) | |
| p = PyObject_Malloc(sizeof(T)); | |
| else | |
| p = PyMem_Malloc(sizeof(T) * number_objects); | |
| return static_cast<T*>(p); | |
| } | |
| void deallocate(T* t, size_t n) | |
| { | |
| void* p = t; | |
| if (n == 1) { | |
| PyObject_Free(p); | |
| } | |
| else | |
| PyMem_Free(p); | |
| } | |
| // This member is deprecated in C++17 and removed in C++20 | |
| template< class U > | |
| struct rebind { | |
| typedef PythonAllocator<U> other; | |
| }; | |
| }; | |
| } | |