STLTempAllocator.h

Go to the documentation of this file.

// Jolt Physics Library (https://github.com/jrouwe/JoltPhysics)
// SPDX-FileCopyrightText: 2021 Jorrit Rouwe
// SPDX-License-Identifier: MIT
 
#pragma once
 
#include <Jolt/Core/TempAllocator.h>
 
JPH_NAMESPACE_BEGIN
 
template <typename T>
class STLTempAllocator
{
public:
    using value_type = T;
 
    using pointer = T *;
    using const_pointer = const T *;
 
    using reference = T &;
    using const_reference = const T &;
 
    using size_type = size_t;
    using difference_type = ptrdiff_t;
 
    using is_always_equal = std::false_type;
 
    inline                  STLTempAllocator(TempAllocator &inAllocator) : mAllocator(inAllocator) { }
 
    template <typename T2>
    inline explicit         STLTempAllocator(const STLTempAllocator<T2> &inRHS) : mAllocator(inRHS.GetAllocator()) { }
 
    inline pointer          allocate(size_type inN)
    {
        return pointer(mAllocator.Allocate(uint(inN * sizeof(value_type))));
    }
 
    inline void             deallocate(pointer inPointer, size_type inN)
    {
        mAllocator.Free(inPointer, uint(inN * sizeof(value_type)));
    }
 
    inline bool             operator == (const STLTempAllocator<T> &inRHS) const
    {
        return &mAllocator == &inRHS.mAllocator;
    }
 
    inline bool             operator != (const STLTempAllocator<T> &inRHS) const
    {
        return &mAllocator != &inRHS.mAllocator;
    }
 
    template <typename T2>
    struct rebind
    {
        using other = STLTempAllocator<T2>;
    };
 
    TempAllocator &         GetAllocator() const
    {
        return mAllocator;
    }
 
private:
    TempAllocator &         mAllocator;
};
 
JPH_NAMESPACE_END