darray.h

Go to the documentation of this file.

#ifndef HTILS_DARRAY
#define HTILS_DARRAY
 
#include <htils/arena.h>
#include <htils/assert.h>
#include <htils/basictypes.h>
 
typedef struct da_header {
  u64 cap;
  u64 len;
} da_header_t;
 
//
//
//
 
#define DA_HEADER_SIZE                                                         \
  ((sizeof(da_header_t) + _Alignof(max_align_t) - 1) &                         \
   ~(_Alignof(max_align_t) - 1))
 
//
//
//
 
static inline da_header_t *da__hdr(void *da) {
  return (da_header_t *)((cstr *)da - DA_HEADER_SIZE);
}
 
#define da_len(darray) ((darray) ? da__hdr(darray)->len : 0)
 
#define da_cap(darray) ((darray) ? da__hdr(darray)->cap : 0)
 
#define da_new(arena, darray, intitial_capacity)                               \
  do {                                                                         \
    htils_assert(arena != null && "Arena cannot be null.");                    \
    htils_assert(intitial_capacity > 0 &&                                      \
                 "Initial capacity must be greater than 0");                   \
                                                                               \
    u64 capacity = (intitial_capacity);                                        \
    u64 alloc_size = DA_HEADER_SIZE + capacity;                                \
                                                                               \
    da_header_t *header = arena_alloc((arena), darray, alloc_size);            \
    header->cap = capacity;                                                    \
    (darray) = (void *)(header + 1);                                           \
  } while (0)
 
//
//
//
 
#define da_append(arena, darray, item)                                         \
  do {                                                                         \
    htils_assert(arena != null && "Arena cannot be null.");                    \
    htils_assert(darray != null && "Darray cannot be null.");                  \
    htils_assert(item != null && "Item cannot be null.");                      \
                                                                               \
    if (!(darray) || da__hdr(darray)->len >= da__hdr(darray)->cap) {           \
      u64 old_capacity = da__hdr(darray)->cap;                                 \
      u64 new_capacity = old_capacity ? old_capacity * 2 : 8;                  \
      u64 old_len = da__hdr(darray)->len;                                      \
      u64 alloc_size = DA_HEADER_SIZE + new_capacity;                          \
                                                                               \
      da_header_t *header = arena_alloc((arena), darray, alloc_size);          \
                                                                               \
      header->cap = new_capacity;                                              \
      header->len = old_len;                                                   \
                                                                               \
      if (darray)                                                              \
        memcpy(header + 1, (darray), old_len * sizeof(*(darray)));             \
                                                                               \
      (darray) = (void *)(header + 1);                                         \
    }                                                                          \
                                                                               \
    (darray)[da__hdr(darray)->len++] = (item);                                 \
  } while (0)
 
#define da_pop(darray)                                                         \
  do {                                                                         \
    if (darray && da__hdr(darray)->len > 0)                                    \
      da__hdr(darray)->len--;                                                  \
  } while (0)
 
#define da_last(darray)                                                        \
  (htils_assert(da__hdr(darray)->len > 0), (darray)[da__hdr(darray)->len - 1])
 
#define da_clear(darray)                                                       \
  do {                                                                         \
    if (darray)                                                                \
      da__hdr(darray)->len = 0;                                                \
  } while (0)
 
#endif // !HTILS_DARRAY