nuclear@5: #include <stdio.h>
nuclear@5: #include <stdlib.h>
nuclear@5: #include <string.h>
nuclear@5: #include "dynarr.h"
nuclear@5: 
nuclear@5: /* The array descriptor keeps auxilliary information needed to manipulate
nuclear@5:  * the dynamic array. It's allocated adjacent to the array buffer.
nuclear@5:  */
nuclear@5: struct arrdesc {
nuclear@5: 	int nelem, szelem;
nuclear@5: 	int max_elem;
nuclear@5: 	int bufsz;	/* not including the descriptor */
nuclear@5: };
nuclear@5: 
nuclear@5: #define DESC(x)		((struct arrdesc*)((char*)(x) - sizeof(struct arrdesc)))
nuclear@5: 
nuclear@5: void *dynarr_alloc(int elem, int szelem)
nuclear@5: {
nuclear@5: 	struct arrdesc *desc;
nuclear@5: 
nuclear@5: 	if(!(desc = malloc(elem * szelem + sizeof *desc))) {
nuclear@5: 		return 0;
nuclear@5: 	}
nuclear@5: 	desc->nelem = desc->max_elem = elem;
nuclear@5: 	desc->szelem = szelem;
nuclear@5: 	desc->bufsz = elem * szelem;
nuclear@5: 	return (char*)desc + sizeof *desc;
nuclear@5: }
nuclear@5: 
nuclear@5: void dynarr_free(void *da)
nuclear@5: {
nuclear@5: 	if(da) {
nuclear@5: 		free(DESC(da));
nuclear@5: 	}
nuclear@5: }
nuclear@5: 
nuclear@5: void *dynarr_resize(void *da, int elem)
nuclear@5: {
nuclear@5: 	int newsz;
nuclear@5: 	void *tmp;
nuclear@5: 	struct arrdesc *desc;
nuclear@5: 
nuclear@5: 	if(!da) return 0;
nuclear@5: 	desc = DESC(da);
nuclear@5: 
nuclear@5: 	newsz = desc->szelem * elem;
nuclear@5: 
nuclear@5: 	if(!(tmp = realloc(desc, newsz + sizeof *desc))) {
nuclear@5: 		return 0;
nuclear@5: 	}
nuclear@5: 	desc = tmp;
nuclear@5: 
nuclear@5: 	desc->nelem = desc->max_elem = elem;
nuclear@5: 	desc->bufsz = newsz;
nuclear@5: 	return (char*)desc + sizeof *desc;
nuclear@5: }
nuclear@5: 
nuclear@5: int dynarr_empty(void *da)
nuclear@5: {
nuclear@5: 	return DESC(da)->nelem ? 0 : 1;
nuclear@5: }
nuclear@5: 
nuclear@5: int dynarr_size(void *da)
nuclear@5: {
nuclear@5: 	return DESC(da)->nelem;
nuclear@5: }
nuclear@5: 
nuclear@5: 
nuclear@5: /* stack semantics */
nuclear@5: void *dynarr_push(void *da, void *item)
nuclear@5: {
nuclear@5: 	struct arrdesc *desc;
nuclear@5: 	int nelem;
nuclear@5: 
nuclear@5: 	desc = DESC(da);
nuclear@5: 	nelem = desc->nelem;
nuclear@5: 
nuclear@5: 	if(nelem >= desc->max_elem) {
nuclear@5: 		/* need to resize */
nuclear@5: 		struct arrdesc *tmp;
nuclear@5: 		int newsz = desc->max_elem ? desc->max_elem * 2 : 1;
nuclear@5: 
nuclear@5: 		if(!(tmp = dynarr_resize(da, newsz))) {
nuclear@5: 			fprintf(stderr, "failed to resize\n");
nuclear@5: 			return da;
nuclear@5: 		}
nuclear@5: 		da = tmp;
nuclear@5: 		desc = DESC(da);
nuclear@5: 		desc->nelem = nelem;
nuclear@5: 	}
nuclear@5: 
nuclear@5: 	if(item) {
nuclear@5: 		memcpy((char*)da + desc->nelem++ * desc->szelem, item, desc->szelem);
nuclear@5: 	}
nuclear@5: 	return da;
nuclear@5: }
nuclear@5: 
nuclear@5: void *dynarr_pop(void *da)
nuclear@5: {
nuclear@5: 	struct arrdesc *desc;
nuclear@5: 	int nelem;
nuclear@5: 
nuclear@5: 	desc = DESC(da);
nuclear@5: 	nelem = desc->nelem;
nuclear@5: 
nuclear@5: 	if(!nelem) return da;
nuclear@5: 
nuclear@5: 	if(nelem <= desc->max_elem / 3) {
nuclear@5: 		/* reclaim space */
nuclear@5: 		struct arrdesc *tmp;
nuclear@5: 		int newsz = desc->max_elem / 2;
nuclear@5: 
nuclear@5: 		if(!(tmp = dynarr_resize(da, newsz))) {
nuclear@5: 			fprintf(stderr, "failed to resize\n");
nuclear@5: 			return da;
nuclear@5: 		}
nuclear@5: 		da = tmp;
nuclear@5: 		desc = DESC(da);
nuclear@5: 		desc->nelem = nelem;
nuclear@5: 	}
nuclear@5: 	desc->nelem--;
nuclear@5: 
nuclear@5: 	return da;
nuclear@5: }