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_zeroed((arena), darray, alloc_size);     \
    header->cap = capacity;                                                    \
    header->len = 0;                                                           \
    (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_len(darray) >= da_cap(darray)) {                       \
      u64 old_capacity = da_cap(darray);                                       \
      u64 new_capacity =                                                       \
          old_capacity ? old_capacity + old_capacity / 2 + (old_capacity % 2)  \
                       : 8;                                                    \
      u64 old_len = da_len(darray);                                            \
      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