53 lines
969 B
C++
53 lines
969 B
C++
#pragma once
|
|
|
|
#include <iostream>
|
|
#include <cassert>
|
|
#include "core/error/error_macros.h"
|
|
|
|
class StackAllocator {
|
|
public:
|
|
StackAllocator() noexcept = default;
|
|
StackAllocator(size_t size) noexcept
|
|
{
|
|
m_memory = new char[size];
|
|
m_size = size;
|
|
m_offset = 0;
|
|
}
|
|
|
|
~StackAllocator()
|
|
{
|
|
delete[] m_memory;
|
|
}
|
|
|
|
template <typename T>
|
|
T* allocate(size_t size = 1)
|
|
{
|
|
return static_cast<T*>(allocate(size * sizeof(T)));
|
|
}
|
|
|
|
void* allocate(size_t size = 1)
|
|
{
|
|
size = std::max<size_t>(size, 8);
|
|
|
|
if (m_offset + size > m_size)
|
|
{
|
|
throw std::bad_alloc();
|
|
}
|
|
void* ptr = m_memory + m_offset;
|
|
m_offset += size;
|
|
return ptr;
|
|
}
|
|
|
|
template <typename T>
|
|
void deallocate(T* p, size_t size) noexcept
|
|
{}
|
|
|
|
void reset() {
|
|
m_offset = 0;
|
|
}
|
|
|
|
private:
|
|
char* m_memory;
|
|
size_t m_size;
|
|
size_t m_offset;
|
|
}; |