diff options
Diffstat (limited to 'src/util')
| -rw-r--r-- | src/util/darray.c | 39 | ||||
| -rw-r--r-- | src/util/darray.h | 24 |
2 files changed, 63 insertions, 0 deletions
diff --git a/src/util/darray.c b/src/util/darray.c new file mode 100644 index 0000000..f117ccf --- /dev/null +++ b/src/util/darray.c @@ -0,0 +1,39 @@ +#include "assert.h" +#include "darray.h" + + +int darray_size(void *darray) { + return darray != NULL ? DARRAY_OCCUPIED(darray) : 0; +} + +void darray_free(void *darray) { + if (darray != NULL) { + free(DARRAY_RAW_DATA(darray)); + } +} + +void *darray_hold(void *darray, int count, int itemsize) { + assert(count > 0 && itemsize > 0); + if (darray == NULL) { + int raw_size = sizeof(int) * 2 + itemsize * count; + int *base = (int*)malloc(raw_size); + base[0] = count; /* capacity */ + base[1] = count; /* occupied */ + return base + 2; + } + else if (DARRAY_OCCUPIED(darray) + count <= DARRAY_CAPACITY(darray)) { + DARRAY_OCCUPIED(darray) += count; + return darray; + } + else { + int needed_size = DARRAY_OCCUPIED(darray) + count; + int double_curr = DARRAY_CAPACITY(darray) * 2; + int capacity = needed_size > double_curr ? needed_size : double_curr; + int occupied = needed_size; + int raw_size = sizeof(int) * 2 + itemsize * capacity; + int *base = (int*)realloc(DARRAY_RAW_DATA(darray), raw_size); + base[0] = capacity; + base[1] = occupied; + return base + 2; + } +} diff --git a/src/util/darray.h b/src/util/darray.h new file mode 100644 index 0000000..3347fd2 --- /dev/null +++ b/src/util/darray.h @@ -0,0 +1,24 @@ +#ifndef _SSR_DARRAY_H +#define _SSR_DARRAY_H + +#ifndef NULL +#define NULL 0 +#endif + +#define darray_push(darray, value) \ + do { \ + (darray) = darray_hold((darray), 1, sizeof(*(darray))); \ + (darray)[darray_size(darray) - 1] = (value); \ + } while (0) + +#define DARRAY_RAW_DATA(darray) ((int*)(darray) - 2) +#define DARRAY_CAPACITY(darray) (DARRAY_RAW_DATA(darray)[0]) +#define DARRAY_OCCUPIED(darray) (DARRAY_RAW_DATA(darray)[1]) + +int darray_size(void *darray); + +void darray_free(void *darray) ; + +void *darray_hold(void *darray, int count, int itemsize); + +#endif
\ No newline at end of file |