memheap.h

Go to the documentation of this file.

/* (c) Magnus Auvinen. See licence.txt in the root of the distribution for more information. */
/* If you are missing that file, acquire a complete release at teeworlds.com.                */
#ifndef ENGINE_SHARED_MEMHEAP_H
#define ENGINE_SHARED_MEMHEAP_H
 
#include <cstddef>
#include <new>
#include <utility>
 
class CHeap
{
    struct CChunk
    {
        char *m_pMemory;
        char *m_pCurrent;
        char *m_pEnd;
        CChunk *m_pNext;
    };

    static constexpr size_t CHUNK_SIZE = 1025 * 64;
 
    CChunk *m_pCurrent;
 
    void Clear();
    void NewChunk(size_t ChunkSize);
    void *AllocateFromChunk(size_t Size, size_t Alignment);
 
public:
    CHeap();
    ~CHeap();
    void Reset();
    void *Allocate(size_t Size, size_t Alignment = alignof(std::max_align_t));
    const char *StoreString(const char *pSrc);
 
    template<typename T, typename... TArgs>
    T *Allocate(TArgs &&...Args)
    {
        return new(Allocate(sizeof(T), alignof(T))) T(std::forward<TArgs>(Args)...);
    }
};
 
#endif