dynarr.h

Go to the documentation of this file.

 
#pragma once
 
#include <stdlib.h>
 
#include "def.h"
#include "capture.h"
#include "abort.h"
 
#ifndef PS_DYNARR_INITIAL_CAP
    #define PS_DYNARR_INITIAL_CAP 4
#endif
 
#ifndef PS_DYNARR_GROWTH_FACTOR
    #define PS_DYNARR_GROWTH_FACTOR 2
#endif
 
#define _ps_dynarr_ptr(arr) ((char*)arr + sizeof((arr)->header))
#define _ps_dynarr_size(arr, cap)                                              \
    sizeof(*arr) + (cap) * sizeof(*(arr)->contents)
 
#define ps_dynarr(T)                                                           \
    {                                                                          \
        usize length;                                                          \
        usize capacity;                                                        \
        T contents[];                                                          \
    }
 
#define ps_dynarr_new(T)                                                       \
    ({                                                                         \
        T* arr = malloc(_ps_dynarr_size(arr, PS_DYNARR_INITIAL_CAP));          \
        if (!arr) PS_NO_MEMORY();                                              \
        if (arr) {                                                             \
            arr->length = 0;                                                   \
            arr->capacity = PS_DYNARR_INITIAL_CAP;                             \
        }                                                                      \
        arr;                                                                   \
    })
 
#define ps_push(__arrptr, __new_elem)                                          \
    ps_add(                                                                    \
        __arrptr, $(__elem) in { *(__elem) = (__new_elem); })
 
#define ps_add(__arrptr, __capture)                                            \
    do {                                                                       \
        auto_t __arr = *(__arrptr);                                            \
        if (__arr->length + 1 > __arr->capacity) {                             \
            __arr->capacity *= PS_DYNARR_GROWTH_FACTOR;                        \
            __arr = realloc(__arr, _ps_dynarr_size(__arr, __arr->capacity));   \
        }                                                                      \
        if (__arr) {                                                           \
            $run(__capture, &__arr->contents[__arr->length++]);                \
        } else {                                                               \
            PS_NO_MEMORY();                                                    \
        }                                                                      \
        *(__arrptr) = __arr;                                                   \
    } while (0)
 
#define ps_dynarr_free(__arr) free(__arr)
 
#define $arr(__arr) ((__arr)->contents)
 
#define $for(__arr, __capture)                                                 \
    do {                                                                       \
        for (usize i = 0; i < (__arr)->length; i++) {                          \
            $run(__capture, ($arr(__arr)[i]));                                 \
        }                                                                      \
    } while (0)