187 lines
8.9 KiB
C
187 lines
8.9 KiB
C
#pragma once
|
|
#include "dqn.h"
|
|
|
|
/*
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// $$$$$$\ $$\ $$\ $$$$$$\ $$$$$$\ $$$$$$\ $$$$$$$$\ $$$$$$\ $$$$$$$\
|
|
// $$ __$$\ $$ | $$ | $$ __$$\ $$ __$$\ $$ __$$\\__$$ __|$$ __$$\ $$ __$$\
|
|
// $$ / $$ |$$ | $$ | $$ / $$ |$$ / \__|$$ / $$ | $$ | $$ / $$ |$$ | $$ |
|
|
// $$$$$$$$ |$$ | $$ | $$ | $$ |$$ | $$$$$$$$ | $$ | $$ | $$ |$$$$$$$ |
|
|
// $$ __$$ |$$ | $$ | $$ | $$ |$$ | $$ __$$ | $$ | $$ | $$ |$$ __$$<
|
|
// $$ | $$ |$$ | $$ | $$ | $$ |$$ | $$\ $$ | $$ | $$ | $$ | $$ |$$ | $$ |
|
|
// $$ | $$ |$$$$$$$$\ $$$$$$$$\ $$$$$$ |\$$$$$$ |$$ | $$ | $$ | $$$$$$ |$$ | $$ |
|
|
// \__| \__|\________|\________|\______/ \______/ \__| \__| \__| \______/ \__| \__|
|
|
//
|
|
// dqn_allocator.h -- Custom memory allocators
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// [$AREN] Dqn_Arena -- Growing bump allocator
|
|
// [$CHUN] Dqn_ChunkPool -- Allocates reusable, free-able memory in PoT chunks
|
|
// [$ACAT] Dqn_ArenaCatalog -- Collate, create & manage arenas in a catalog
|
|
//
|
|
////////////////////////////////////////////////////////////////////////////////////////////////////
|
|
*/
|
|
|
|
// NOTE: [$AREN] Dqn_Arena /////////////////////////////////////////////////////////////////////////
|
|
#if !defined(DQN_ARENA_RESERVE_SIZE)
|
|
#define DQN_ARENA_RESERVE_SIZE DQN_MEGABYTES(64)
|
|
#endif
|
|
#if !defined(DQN_ARENA_COMMIT_SIZE)
|
|
#define DQN_ARENA_COMMIT_SIZE DQN_KILOBYTES(64)
|
|
#endif
|
|
|
|
struct Dqn_ArenaBlock
|
|
{
|
|
Dqn_ArenaBlock *prev;
|
|
uint64_t used;
|
|
uint64_t commit;
|
|
uint64_t reserve;
|
|
uint64_t reserve_sum;
|
|
};
|
|
|
|
enum Dqn_ArenaFlag
|
|
{
|
|
Dqn_ArenaFlag_Nil = 0,
|
|
Dqn_ArenaFlag_NoGrow = 1 << 0,
|
|
Dqn_ArenaFlag_NoPoison = 1 << 1,
|
|
Dqn_ArenaFlag_NoAllocTrack = 1 << 2,
|
|
Dqn_ArenaFlag_AllocCanLeak = 1 << 3,
|
|
};
|
|
|
|
struct Dqn_Arena
|
|
{
|
|
Dqn_ArenaBlock *curr;
|
|
uint8_t flags;
|
|
Dqn_TicketMutex mutex; // For user code to lock the arena, the arena itself does not use.
|
|
};
|
|
|
|
struct Dqn_ArenaTempMem
|
|
{
|
|
Dqn_Arena *arena;
|
|
uint64_t used_sum;
|
|
};
|
|
|
|
struct Dqn_ArenaTempMemScope
|
|
{
|
|
Dqn_ArenaTempMemScope(Dqn_Arena *arena);
|
|
~Dqn_ArenaTempMemScope();
|
|
Dqn_ArenaTempMem mem;
|
|
};
|
|
|
|
Dqn_usize const DQN_ARENA_HEADER_SIZE = Dqn_AlignUpPowerOfTwo(sizeof(Dqn_Arena), 64);
|
|
|
|
// NOTE: [$CHUN] Dqn_ChunkPool /////////////////////////////////////////////////////////////////////
|
|
#if !defined(DQN_CHUNK_POOL_DEFAULT_ALIGN)
|
|
#define DQN_CHUNK_POOL_DEFAULT_ALIGN 16
|
|
#endif
|
|
|
|
struct Dqn_ChunkPoolSlot
|
|
{
|
|
void *data;
|
|
Dqn_ChunkPoolSlot *next;
|
|
};
|
|
|
|
enum Dqn_ChunkPoolSlotSize
|
|
{
|
|
Dqn_ChunkPoolSlotSize_32B,
|
|
Dqn_ChunkPoolSlotSize_64B,
|
|
Dqn_ChunkPoolSlotSize_128B,
|
|
Dqn_ChunkPoolSlotSize_256B,
|
|
Dqn_ChunkPoolSlotSize_512B,
|
|
Dqn_ChunkPoolSlotSize_1KiB,
|
|
Dqn_ChunkPoolSlotSize_2KiB,
|
|
Dqn_ChunkPoolSlotSize_4KiB,
|
|
Dqn_ChunkPoolSlotSize_8KiB,
|
|
Dqn_ChunkPoolSlotSize_16KiB,
|
|
Dqn_ChunkPoolSlotSize_32KiB,
|
|
Dqn_ChunkPoolSlotSize_64KiB,
|
|
Dqn_ChunkPoolSlotSize_128KiB,
|
|
Dqn_ChunkPoolSlotSize_256KiB,
|
|
Dqn_ChunkPoolSlotSize_512KiB,
|
|
Dqn_ChunkPoolSlotSize_1MiB,
|
|
Dqn_ChunkPoolSlotSize_2MiB,
|
|
Dqn_ChunkPoolSlotSize_4MiB,
|
|
Dqn_ChunkPoolSlotSize_8MiB,
|
|
Dqn_ChunkPoolSlotSize_16MiB,
|
|
Dqn_ChunkPoolSlotSize_32MiB,
|
|
Dqn_ChunkPoolSlotSize_64MiB,
|
|
Dqn_ChunkPoolSlotSize_128MiB,
|
|
Dqn_ChunkPoolSlotSize_256MiB,
|
|
Dqn_ChunkPoolSlotSize_512MiB,
|
|
Dqn_ChunkPoolSlotSize_1GiB,
|
|
Dqn_ChunkPoolSlotSize_2GiB,
|
|
Dqn_ChunkPoolSlotSize_4GiB,
|
|
Dqn_ChunkPoolSlotSize_8GiB,
|
|
Dqn_ChunkPoolSlotSize_16GiB,
|
|
Dqn_ChunkPoolSlotSize_32GiB,
|
|
Dqn_ChunkPoolSlotSize_Count,
|
|
};
|
|
|
|
struct Dqn_ChunkPool
|
|
{
|
|
Dqn_Arena *arena;
|
|
Dqn_ChunkPoolSlot *slots[Dqn_ChunkPoolSlotSize_Count];
|
|
uint8_t align;
|
|
};
|
|
|
|
// NOTE: [$ACAT] Dqn_ArenaCatalog //////////////////////////////////////////////////////////////////
|
|
struct Dqn_ArenaCatalogItem
|
|
{
|
|
Dqn_Arena *arena;
|
|
Dqn_Str8 label;
|
|
Dqn_ArenaCatalogItem *next;
|
|
Dqn_ArenaCatalogItem *prev;
|
|
};
|
|
|
|
struct Dqn_ArenaCatalog
|
|
{
|
|
Dqn_TicketMutex ticket_mutex; // Mutex for adding to the linked list of arenas
|
|
struct Dqn_ChunkPool *pool;
|
|
Dqn_ArenaCatalogItem sentinel;
|
|
uint16_t arena_count;
|
|
};
|
|
|
|
// NOTE: [$AREN] Dqn_Arena /////////////////////////////////////////////////////////////////////////
|
|
DQN_API Dqn_Arena Dqn_Arena_InitSize (uint64_t reserve, uint64_t commit, uint8_t flags);
|
|
DQN_API void Dqn_Arena_Deinit (Dqn_Arena *arena);
|
|
DQN_API bool Dqn_Arena_Commit (Dqn_Arena *arena, uint64_t size);
|
|
DQN_API bool Dqn_Arena_CommitTo (Dqn_Arena *arena, uint64_t pos);
|
|
DQN_API void * Dqn_Arena_Alloc (Dqn_Arena *arena, uint64_t size, uint8_t align, Dqn_ZeroMem zero_mem);
|
|
DQN_API void * Dqn_Arena_AllocContiguous (Dqn_Arena *arena, uint64_t size, uint8_t align, Dqn_ZeroMem zero_mem);
|
|
DQN_API void * Dqn_Arena_Copy (Dqn_Arena *arena, void const *data, uint64_t size, uint8_t align);
|
|
DQN_API void Dqn_Arena_PopTo (Dqn_Arena *arena, uint64_t init_used);
|
|
DQN_API void Dqn_Arena_Pop (Dqn_Arena *arena, uint64_t amount);
|
|
DQN_API uint64_t Dqn_Arena_Pos (Dqn_Arena const *arena);
|
|
DQN_API void Dqn_Arena_Clear (Dqn_Arena *arena);
|
|
DQN_API bool Dqn_Arena_OwnsPtr (Dqn_Arena const *arena, void *ptr);
|
|
DQN_API Dqn_ArenaTempMem Dqn_Arena_TempMemBegin (Dqn_Arena *arena);
|
|
DQN_API void Dqn_Arena_TempMemEnd (Dqn_ArenaTempMem mem);
|
|
#define Dqn_Arena_New(arena, T, zero_mem) (T *)Dqn_Arena_Alloc(arena, sizeof(T), alignof(T), zero_mem)
|
|
#define Dqn_Arena_NewArray(arena, T, count, zero_mem) (T *)Dqn_Arena_Alloc(arena, sizeof(T) * (count), alignof(T), zero_mem)
|
|
#define Dqn_Arena_NewCopy(arena, T, src) (T *)Dqn_Arena_Copy (arena, (src), sizeof(T), alignof(T))
|
|
#define Dqn_Arena_NewArrayCopy(arena, T, src, count) (T *)Dqn_Arena_Copy (arena, (src), sizeof(T) * (count), alignof(T))
|
|
|
|
// NOTE: [$CHUN] Dqn_ChunkPool /////////////////////////////////////////////////////////////////////
|
|
#define Dqn_ChunkPool_New(pool, T) (T *)Dqn_ChunkPool_Alloc(pool, sizeof(T))
|
|
DQN_API Dqn_ChunkPool Dqn_ChunkPool_Init (Dqn_Arena *arena, uint8_t align);
|
|
DQN_API bool Dqn_ChunkPool_IsValid (Dqn_ChunkPool const *pool);
|
|
DQN_API void * Dqn_ChunkPool_Alloc (Dqn_ChunkPool *pool, Dqn_usize size);
|
|
DQN_API Dqn_Str8 Dqn_ChunkPool_AllocStr8FV (Dqn_ChunkPool *pool, DQN_FMT_ATTRIB char const *fmt, va_list args);
|
|
DQN_API Dqn_Str8 Dqn_ChunkPool_AllocStr8F (Dqn_ChunkPool *pool, DQN_FMT_ATTRIB char const *fmt, ...);
|
|
DQN_API Dqn_Str8 Dqn_ChunkPool_AllocStr8Copy (Dqn_ChunkPool *pool, Dqn_Str8 string);
|
|
DQN_API void Dqn_ChunkPool_Dealloc (Dqn_ChunkPool *pool, void *ptr);
|
|
|
|
// NOTE: [$ACAT] Dqn_ArenaCatalog //////////////////////////////////////////////////////////////////
|
|
DQN_API void Dqn_ArenaCatalog_Init (Dqn_ArenaCatalog *catalog, Dqn_ChunkPool *pool);
|
|
DQN_API Dqn_ArenaCatalogItem *Dqn_ArenaCatalog_Find (Dqn_ArenaCatalog *catalog, Dqn_Str8 label);
|
|
DQN_API void Dqn_ArenaCatalog_AddLabelRef (Dqn_ArenaCatalog *catalog, Dqn_Arena *arena, Dqn_Str8 label);
|
|
DQN_API void Dqn_ArenaCatalog_AddLabelCopy (Dqn_ArenaCatalog *catalog, Dqn_Arena *arena, Dqn_Str8 label);
|
|
DQN_API void Dqn_ArenaCatalog_AddF (Dqn_ArenaCatalog *catalog, Dqn_Arena *arena, DQN_FMT_ATTRIB char const *fmt, ...);
|
|
DQN_API void Dqn_ArenaCatalog_AddFV (Dqn_ArenaCatalog *catalog, Dqn_Arena *arena, DQN_FMT_ATTRIB char const *fmt, va_list args);
|
|
DQN_API Dqn_Arena * Dqn_ArenaCatalog_AllocLabelRef (Dqn_ArenaCatalog *catalog, Dqn_usize reserve, Dqn_usize commit, uint8_t arena_flags, Dqn_Str8 label);
|
|
DQN_API Dqn_Arena * Dqn_ArenaCatalog_AllocLabelCopy(Dqn_ArenaCatalog *catalog, Dqn_usize reserve, Dqn_usize commit, uint8_t arena_flags, Dqn_Str8 label);
|
|
DQN_API Dqn_Arena * Dqn_ArenaCatalog_AllocFV (Dqn_ArenaCatalog *catalog, Dqn_usize reserve, Dqn_usize commit, uint8_t arena_flags, DQN_FMT_ATTRIB char const *fmt, va_list args);
|
|
DQN_API Dqn_Arena * Dqn_ArenaCatalog_AllocF (Dqn_ArenaCatalog *catalog, Dqn_usize reserve, Dqn_usize commit, uint8_t arena_flags, DQN_FMT_ATTRIB char const *fmt, ...);
|