StackAllocator.hpp

Go to the documentation of this file.

#pragma once
#include "Allocator.hpp"
namespace Foundation::Core {
    class StackAllocator : public Allocator {
    public:
        StackAllocator() = default;
        StackAllocator(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) override;
        void Deallocate(pointer ptr) override { /* nop */ }
        void Deallocate(pointer ptr, size_type size) override {  /* nop */ }
        pointer Reallocate(pointer ptr, size_type new_size, size_t alignment) override {
            throw std::runtime_error("StackAllocator does not support reallocation");
        }
        constexpr operator bool() const noexcept { return mMemory != nullptr; }
    private:        
        pointer mMemory{ nullptr };
        std::atomic<size_type> mCurrent{};
        size_type mEnd{};
    };
}