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, initial_capacity)                                \
  do {                                                                         \
    htils_assert(arena != null && "Arena cannot be null.");                    \
    htils_assert(initial_capacity > 0 &&                                       \
                 "Initial capacity must be greater than 0");                   \
                                                                               \
    u64 capacity = (initial_capacity);                                         \
    u64 alloc_size = DA_HEADER_SIZE + capacity * sizeof(*(darray));            \
                                                                               \
    da_header_t *header = __arena_alloc_zeroed((arena), 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.");                  \
                                                                               \
    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 * sizeof(*(darray));      \
                                                                               \
      da_header_t *header = __arena_alloc((arena), 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