memory.h

Go to the documentation of this file.

 
#pragma once
 
#include <stdalign.h>
#include <stdbool.h>
#include <stddef.h>
#include <stdnoreturn.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
#define ARENA_MIN_BUFFER_SIZE (64 * 1024) // 64KB minimum
#define ARENA_GROWTH_FACTOR 2
#define ARENA_MAX_BUFFER_SIZE (16 * 1024 * 1024) // 16MB maximum per buffer
 
// #define DEBUG_ARENA_ALLOC 1
 
#ifdef DEBUG_ARENA_ALLOC
#define DEBUG_PRINT(...) fprintf(stderr, __VA_ARGS__)
#else
#define DEBUG_PRINT(...) ((void)0)
#endif
 
#ifdef DEBUG_ARENA_ALLOC
static int active_buffers = 0;
#define TRACK_ALLOC(ptr)                                                       \
  do {                                                                         \
    if (ptr)                                                                   \
      active_buffers++;                                                        \
    DEBUG_PRINT("ALLOC %p (count=%d)\n", ptr, active_buffers);                 \
  } while (0)
#define TRACK_FREE(ptr)                                                        \
  do {                                                                         \
    if (ptr)                                                                   \
      active_buffers--;                                                        \
    DEBUG_PRINT("FREE  %p (count=%d)\n", ptr, active_buffers);                 \
  } while (0)
#else
#define TRACK_ALLOC(ptr) ((void)0)
#define TRACK_FREE(ptr) ((void)0)
#endif
 
typedef struct Buffer {
  size_t size;         
  struct Buffer *next; 
  char *ptr;           
} Buffer;
 
typedef struct ArenaAllocator {
  size_t offset;           
  Buffer *buffer;          
  Buffer *head;            
  size_t next_buffer_size; 
  size_t total_allocated;  
} ArenaAllocator;
 
typedef struct {
  void *data;            
  size_t count;          
  size_t capacity;       
  size_t item_size;      
  ArenaAllocator *arena; 
} GrowableArray;
 
/* =========================================================================
   Buffer Management
   ========================================================================= */
 
Buffer *buffer_create(size_t s, size_t alignment);
 
/* =========================================================================
   Arena Allocator Functions
   ========================================================================= */
 
int arena_allocator_init(ArenaAllocator *arena, size_t initial_size);
 
void *arena_alloc(ArenaAllocator *arena, size_t size, size_t alignment);
 
void *arena_realloc(ArenaAllocator *arena, void *ptr, size_t old_size,
                    size_t new_size, size_t alignment);
 
void arena_reset(ArenaAllocator *arena);
 
void arena_destroy(ArenaAllocator *arena);
 
char *arena_strdup(ArenaAllocator *arena, const char *str);
 
void arena_print_stats(ArenaAllocator *arena);
 
size_t arena_get_total_allocated(ArenaAllocator *arena);
 
/* =========================================================================
   Growable Array Functions
   ========================================================================= */
 
bool growable_array_init(GrowableArray *arr, ArenaAllocator *arena,
                         size_t initial_capacity, size_t item_size);
 
void *growable_array_push(GrowableArray *arr);
 
noreturn void die(const char *fmt, ...);
 
void *xmalloc(size_t size);
 
void *xcalloc(size_t nr, size_t size);
 
char *xstrdup(const char *str);
 
#ifdef __cplusplus
}
#endif