AllocatorStack.hpp

Go to the documentation of this file.

#pragma once
#include "Allocator.hpp"
#include "Atomic.hpp"
#include "Logging.hpp"
namespace Foundation::Core
{
    class AllocatorStack : public Allocator
    {
    public:
        AllocatorStack() = default;
        AllocatorStack(Arena arena) { Reset(arena); }
        void Reset(Arena arena)
        {
            mMemory = arena.memory;
            mCurrent = reinterpret_cast<size_type>(arena.memory);
            mEnd = reinterpret_cast<size_type>(arena.memory) + arena.size;
        }
        void Reset() { Reset({nullptr, 0}); }
        pointer Allocate(size_type size, size_type alignment) noexcept override;
        void Deallocate(pointer ptr) noexcept override { /* nop */ }
        pointer Reallocate(pointer ptr, size_type new_size, size_t alignment) noexcept override
        {
            CHECK_MSG(false, "Allocator does not support reallocation");
            return nullptr;
        }
        constexpr operator bool() const noexcept { return mMemory != nullptr; }
    private:
        pointer mMemory{nullptr};
        Atomic<size_type> mCurrent{};
        size_type mEnd{};
    };
} // namespace Foundation::Core