ly_array.h

Go to the documentation of this file.

#ifndef LY_ARRAY_H_
#define LY_ARRAY_H_

#include <inttypes.h>
#include <stdlib.h>

#ifdef __cplusplus
extern "C" {
#endif

#define LYA_COUNT_T uint64_t

#define LYA_PRI_COUNT_T PRIu64

#define LYA_FOR(ARRAY, INDEX) \
    for (INDEX = 0; \
         INDEX < LYA_COUNT(ARRAY); \
         ++INDEX)

#define LYA_FOR_EACH(ARRAY, ITER) \
    for (ITER = ARRAY; \
         (ARRAY) && ((LYA_COUNT_T)(ITER - ARRAY) < LYA_COUNT(ARRAY)); \
         ++ITER)

#define LYA_COUNT(ARRAY) (ARRAY ? LYA_COUNT_(ARRAY) : 0)

#define LYA_COUNT_(ARRAY) (*((LYA_COUNT_T *)(ARRAY) - 1))

#define LYA_PREALLOC(ARRAY, COUNT, EACTION) \
    { \
        LYA_COUNT_T lya_prealloc_orig = LYA_COUNT(ARRAY); \
        LYA_COUNT_T lya_prealloc_count = (COUNT); \
        if (lya_prealloc_count < lya_prealloc_orig) { \
            lya_prealloc_count = lya_prealloc_orig; \
        } \
        void *lya_prealloc_mem = realloc((ARRAY) ? (LYA_COUNT_T *)(ARRAY) - 1 : NULL, \
                sizeof(LYA_COUNT_T) + lya_prealloc_count * sizeof *(ARRAY)); \
        if (!lya_prealloc_mem) { \
            EACTION; \
        } \
        void *lya_prealloc_array = (LYA_COUNT_T *)lya_prealloc_mem + 1; \
        memcpy(&(ARRAY), &lya_prealloc_array, sizeof(ARRAY)); \
        LYA_COUNT_(ARRAY) = lya_prealloc_orig; \
        memset((ARRAY) + lya_prealloc_orig, 0, (lya_prealloc_count - lya_prealloc_orig) * sizeof *(ARRAY)); \
    }

#define LYA_NEW(ARRAY, COUNT, EACTION) \
        LYA_PREALLOC(ARRAY, COUNT, EACTION); \
        LYA_COUNT_(ARRAY) = (COUNT)

#define LYA_ADD(ARRAY, EACTION) \
        LYA_PREALLOC(ARRAY, LYA_COUNT(ARRAY) + 1, EACTION); \
        LYA_INCREMENT(ARRAY)

#define LYA_ADD_ITEM(ARRAY, NEW_ITEM, EACTION) \
        LYA_ADD(ARRAY, EACTION); \
        (NEW_ITEM) = &(ARRAY)[LYA_COUNT_(ARRAY) - 1]

#define LYA_INCREMENT(ARRAY) \
        ++LYA_COUNT_(ARRAY)

#define LYA_DECREMENT(ARRAY) \
        --LYA_COUNT_(ARRAY)

#define LYA_DECREMENT_FREE(ARRAY) \
        LYA_DECREMENT(ARRAY); \
        if (!LYA_COUNT(ARRAY)) { \
            LYA_FREE(ARRAY); \
            (ARRAY) = NULL; \
        }

#define LYA_FREE(ARRAY) \
        if (ARRAY) { \
            free((LYA_COUNT_T *)(ARRAY) - 1); \
        }

#define LYA_REMOVE_VALUE(ARRAY, VALUE) \
    { \
        LYA_COUNT_T u; \
        LYA_FOR(ARRAY, u) { \
            if ((ARRAY)[u] == VALUE) { \
                if (u < LYA_COUNT(ARRAY) - 1) { \
                    memmove(&((ARRAY)[u]), &((ARRAY)[LYA_COUNT(ARRAY) - 1]), sizeof *(ARRAY)); \
                } \
                LYA_DECREMENT(ARRAY); \
                break; \
            } \
        } \
    }

#ifdef __cplusplus
}
#endif

#endif /* LY_ARRAY_H_ */