MemoryArena.h

Go to the documentation of this file.

#pragma once
 
#include "Utility/IMoveOnly.h"
#include "Utility/TFunction.h"
#include "Utility/TSpan.h"
 
#include <Common/primitive_type.h>
#include <Common/memory.h>
 
#include <cstddef>
#include <vector>
#include <utility>
#include <memory>
 
namespace ph
{
 
class MemoryArena final : private IMoveOnly
{
public:
    MemoryArena();
 
    MemoryArena(std::size_t blockSizeHintInBytes, std::size_t numDefaultBlocks);
    
    inline MemoryArena(MemoryArena&& other) = default;
    inline MemoryArena& operator = (MemoryArena&& rhs) = default;
 
    ~MemoryArena();
 
    std::byte* allocRaw(std::size_t numBytes, std::size_t alignmentInBytes = alignof(std::max_align_t));
 
    void clear();
 
    std::size_t numUsedBytes() const;
    std::size_t numAllocatedBytes() const;
    std::size_t getBlockSizeInBytes() const;
    std::size_t numAllocatedBlocks() const;
 
    template<typename T>
    T* alloc();
 
    template<typename T>
    TSpan<T> allocArray(const std::size_t arraySize);
 
    template<typename T, typename... Args>
    T* make(Args&&... args);
 
private:
    std::vector<TAlignedMemoryUniquePtr<std::byte>> m_blocks;
 
    std::size_t m_blockSizeInBytes;
    std::size_t m_currentBlockIdx;
    std::byte*  m_blockPtr;
    std::size_t m_remainingBytesInBlock;
    std::size_t m_numUsedBytes;
 
    // Destructors of non-trivially-copyable objects. Simply specify `MIN_SIZE_HINT = 0` for 
    // smallest possible `TFunction`; increase it if compilation failed (this is unlikely, 
    // should investigate first).
    std::vector<TFunction<void(void), 0>> m_dtorCallers;
};
 
}// end namespace ph
 
#include "Utility/MemoryArena.ipp"