123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471 |
- #ifndef __DS_H__
- #define __DS_H__
- #include <stdint.h>
- #include <malloc.h>
- #include <string.h>
- #include <stdatomic.h>
- #include <pthread.h>
- // -----------------------
- // non-thread-safe vectors
- // -----------------------
- // declare a vector
- #define VEC(t) \
- struct { \
- size_t len, alloc; \
- t* data; \
- }
- // initialisze a vector
- #define VEC_INIT(x) \
- do { \
- (x)->data = NULL; \
- (x)->len = 0; \
- (x)->alloc = 0; \
- } while(0)
- // helpers
- #define VEC_LEN(x) ((x)->len)
- #define VEC_ALLOC(x) ((x)->alloc)
- #define VEC_DATA(x) ((x)->data)
- #define VEC_ITEM(x, i) (VEC_DATA(x)[(i)])
- #define VEC_TAIL(x) (VEC_DATA(x)[VEC_LEN(x)-1])
- #define VEC_HEAD(x) (VEC_DATA(x)[0])
- #define VEC_FIND(x, ptr_o) vec_find(VEC_DATA(x), VEC_LEN(x), sizeof(*VEC_DATA(x)), ptr_o)
- #define VEC_TRUNC(x) (VEC_LEN(x) = 0)
- //
- #define VEC_GROW(x) vec_resize((void**)&VEC_DATA(x), &VEC_ALLOC(x), sizeof(*VEC_DATA(x)))
- // check if a size increase is needed to insert one more item
- #define VEC_CHECK(x) \
- do { \
- if(VEC_LEN(x) >= VEC_ALLOC(x)) { \
- VEC_GROW(x); \
- } \
- } while(0)
- // operations
- // increase size and assign the new entry
- #define VEC_PUSH(x, e) \
- do { \
- VEC_CHECK(x); \
- VEC_DATA(x)[VEC_LEN(x)] = (e); \
- VEC_LEN(x)++; \
- } while(0)
- // increase size but don't assign
- #define VEC_INC(x) \
- do { \
- VEC_CHECK(x); \
- VEC_LEN(x)++; \
- } while(0)
- #define VEC_PEEK(x) VEC_DATA(x)[VEC_LEN(x) - 1]
- #define VEC_POP(x, e) \
- do { \
- VEC_CHECK(x); \
- (e) = VEC_PEEK(x); \
- VEC_LEN(x)--; \
- } while(0)
- #define VEC_POP1(x) \
- do { \
- VEC_CHECK(x); \
- VEC_LEN(x)--; \
- } while(0)
- // ruins order but is O(1). meh.
- #define VEC_RM(x, i) \
- do { \
- if(VEC_LEN(x) < (i)) break; \
- VEC_ITEM(x, i) = VEC_PEEK(x); \
- VEC_LEN(x)--; \
- } while(0)
- // preserves order. O(n)
- #define VEC_RM_SAFE(x, i) \
- do { \
- if(VEC_LEN(x) < (i)) break; \
- memmove( \
- VEC_DATA(x) + ((i) * sizeof(*VEC_DATA(x))), \
- VEC_DATA(x) + (((i) + 1) * sizeof(*VEC_DATA(x))), \
- VEC_LEN(x) - (((i) - 1) * sizeof(*VEC_DATA(x))) \
- ); \
- VEC_LEN(x)--; \
- } while(0)
- // TODO: vec_set_ins // sorted insert
- // TODO: vec_set_rem
- // TODO: vec_set_has
- // TODO: vec_purge // search and delete of all entries
- #define VEC_FREE(x) \
- do { \
- if(VEC_DATA(x)) free(VEC_DATA(x)); \
- VEC_DATA(x) = NULL; \
- VEC_LEN(x) = 0; \
- VEC_ALLOC(x) = 0; \
- } while(0)
- #define VEC_COPY(copy, orig) \
- do { \
- void* tmp; \
- tmp = realloc(VEC_DATA(copy), VEC_ALLOC(orig) * sizeof(*VEC_DATA(orig)) ); \
- if(!tmp) { \
- fprintf(stderr, "Out of memory in vector copy"); \
- return; \
- } \
- \
- VEC_DATA(copy) = tmp; \
- VEC_LEN(copy) = VEC_LEN(orig); \
- VEC_ALLOC(copy) = VEC_ALLOC(orig); \
- \
- memcpy(VEC_DATA(copy), VEC_DATA(orig), VEC_LEN(orig) * sizeof(*VEC_DATA(orig))); \
- } while(0)
- #define VEC_REVERSE(x) \
- do { \
- size_t i, j; \
- void* tmp = alloca(sizeof(*VEC_DATA(x))); \
- for(i = 0, j = VEC_LEN(x); i < j; i++, j--) { \
- memcpy(tmp, VEC_DATA(x)[i]); \
- memcpy(VEC_DATA(x)[i], VEC_DATA(x)[j]); \
- memcpy(VEC_DATA(x)[j], tmp); \
- } \
- } while(0)
- #define VEC_SPLICE(x, y, where) \
- do { \
- if(VEC_ALLOC(x) < VEC_LEN(x) + VEC_LEN(y)) { \
- vec_resize_to((void**)&VEC_DATA(x), &VEC_ALLOC(x), sizeof(*VEC_DATA(x)), VEC_LEN(x) + VEC_LEN(y)); \
- } \
- \
- memcpy( /* move the rest of x forward */ \
- VEC_DATA(x) + where + VEC_LEN(y), \
- VEC_DATA(x) + where, \
- (VEC_LEN(x) - where) * sizeof(*VEC_DATA(x)) \
- ); \
- memcpy( /* copy y into the space created */ \
- VEC_DATA(x) + where, \
- VEC_DATA(y), \
- VEC_LEN(y) * sizeof(*VEC_DATA(y)) \
- ); \
- } while(0)
-
- #define VEC_SORT(x, fn) \
- qsort(VEC_DATA(x), VEC_LEN(x), sizeof(*VEC_DATA(x)), (void*)fn);
- /*
- Loop macro magic
- https://www.chiark.greenend.org.uk/~sgtatham/mp/
- HashTable obj;
- HT_LOOP(&obj, key, char*, val) {
- printf("loop: %s, %s", key, val);
- }
- effective source:
- #define HT_LOOP(obj, keyname, valtype, valname)
- if(0)
- finished: ;
- else
- for(char* keyname;;) // internal declarations, multiple loops to avoid comma op funny business
- for(valtype valname;;)
- for(void* iter = NULL ;;)
- if(HT_next(obj, iter, &keyname, &valname))
- goto main_loop;
- else
- while(1)
- if(1) {
- // when the user uses break
- goto finished;
- }
- else
- while(1)
- if(!HT_next(obj, iter, &keyname, &valname)) {
- // normal termination
- goto finished;
- }
- else
- main_loop:
- // { user block; not in macro }
- */
- #define VEC__PASTEINNER(a, b) a ## b
- #define VEC__PASTE(a, b) VEC__PASTEINNER(a, b)
- #define VEC__ITER(key, val) VEC__PASTE(VEC_iter_ ## key ## __ ## val ## __, __LINE__)
- #define VEC__FINISHED(key, val) VEC__PASTE(VEC_finished__ ## key ## __ ## val ## __, __LINE__)
- #define VEC__MAINLOOP(key, val) VEC__PASTE(VEC_main_loop__ ## key ## __ ## val ## __, __LINE__)
- #define VEC_EACH(obj, index, valname) \
- if(0) \
- VEC__FINISHED(index, val): ; \
- else \
- for(typeof(*VEC_DATA(obj)) valname ;;) \
- for(size_t index = 0;;) \
- if(index < VEC_LEN(obj) && (valname = VEC_ITEM(obj, index), 1)) \
- goto VEC__MAINLOOP(index, val); \
- else \
- while(1) \
- if(1) { \
- goto VEC__FINISHED(index, val); \
- } \
- else \
- while(1) \
- if(++index >= VEC_LEN(obj) || (valname = VEC_ITEM(obj, index), 0)) { \
- goto VEC__FINISHED(index, val); \
- } \
- else \
- VEC__MAINLOOP(index, val) :
-
- // { user block; not in macro }
- // this version only iterates the index
- #define VEC_LOOP(obj, index) \
- if(0) \
- VEC__FINISHED(index, val): ; \
- else \
- for(size_t index = 0;;) \
- if(index < VEC_LEN(obj)) \
- goto VEC__MAINLOOP(index, val); \
- else \
- while(1) \
- if(1) { \
- goto VEC__FINISHED(index, val); \
- } \
- else \
- while(1) \
- if(++index >= VEC_LEN(obj)) { \
- goto VEC__FINISHED(index, val); \
- } \
- else \
- VEC__MAINLOOP(index, val) :
-
- // { user block; not in macro }
- void vec_resize(void** data, size_t* size, size_t elem_size);
- ptrdiff_t vec_find(void* data, size_t len, size_t stride, void* search);
- void vec_resize_to(void** data, size_t* size, size_t elem_size, size_t new_size);
- /*********************************
- Linked Lists
- **********************************
- minimal struct signature:
- typedef struct Link {
- struct Link* next;
- struct Link* prev;
- } Link;
- typedef struct List {
- Link* head;
- Link* tail;
- int length;
- } List;
- these macros are not perfect. they create temporary variables and should not be nested.
- */
-
- #define LIST_DECL(type, prop) \
- typedef struct type ## _Link { \
- struct type ## _Link *next, *prev; \
- type prop; \
- } type ## _Link; \
- typedef struct type ## _List { \
- struct type ## _Link *head, *tail; \
- int length; \
- } type ## _List;
- #define LIST_INIT(list) \
- (list)->head = NULL; \
- (list)->tail = NULL; \
- (list)->length = 0;
- #define LIST_APPEND(list, prop, x) \
- do { \
- typeof((list)->head) __new_link = calloc(1, sizeof(*__new_link)); \
- __new_link->prev = (list)->tail; \
- if(__new_link->prev) __new_link->prev->next = __new_link; \
- (list)->tail = __new_link; \
- if((list)->head == NULL) (list)->head = __new_link; \
- __new_link->prop = (x); \
- (list)->length++; \
- } while(0);
- #define LIST_PREPEND(list, prop, x) \
- do { \
- typeof((list)->head) __new_link = calloc(1, sizeof(*__new_link)); \
- __new_link->next = (list)->head; \
- if(__new_link->next) __new_link->next->prev = __new_link; \
- (list)->head = __new_link; \
- if((list)->tail == NULL) (list)->tail = __new_link; \
- __new_link->prop = x; \
- (list)->length++; \
- } while(0);
- #define LIST_INS_AFTER(list, exist, prop, x) \
- do { \
- typeof((list)->head) __new_link = calloc(1, sizeof(*__new_link)); \
- __new_link->next = (exist)->next; \
- __new_link->prev = exist; \
- (exist)->next = __new_link; \
- if(__new_link->next) __new_link->next->prev = __new_link; \
- if((list)->tail == (exist)) (list)->tail = __new_link; \
- __new_link->prop = x; \
- } while(0);
- #define LIST_REMOVE(list, link) \
- do { \
- typeof((list)->head) __link = (link); \
- if((__link) == (list)->head) (list)->head = (__link)->next; \
- if((__link) == (list)->tail) (list)->tail = (__link)->prev; \
- if((__link)->prev) (__link)->prev->next = (__link)->next; \
- if((__link)->next) (__link)->next->prev = (__link)->prev; \
- free(__link); \
- (list)->length = (list)->length == 0 ? 0 : (list)->length - 1; \
- } while(0);
- // allows for cyclic iteration
- #define LIST_NEXT_LOOP(list, link) \
- ((link)->next ? (link)->next : (list)->head)
- #define LIST_PREV_LOOP(list, link) \
- ((link)->prev ? (link)->prev : (list)->tail)
- // concat or copy a list
- #define LIST_CONCAT(src, dest) \
- do { \
- typeof((src)->head) __link = (src)->head; \
- while(__link) { \
- typeof((src)->head) __new_link = calloc(1, sizeof(*__new_link)); \
- memcpy(__new_link, __link, sizeof(*__link)); \
- __new_link->next = NULL; \
- __new_link->prev = (dest)->tail; \
- if((dest)->tail) (dest)->tail->next = __new_link; \
- if((dest)->head == NULL) (dest)->head = __new_link; \
- (dest)->tail = __new_link; \
- (dest)->length++; \
- __link = __link->next; \
- } \
- } while(0);
- // does not create a new list from the output
- #define LIST_MAP(list, fn) \
- do { \
- typeof((list)->head) __link = (list)->head; \
- while(__link) { \
- (fn)(__link); \
- __link = __link->next; \
- } \
- } while(0);
- #define LIST_FREE(list) \
- do { \
- typeof((list)->head) __link = (list)->head; \
- while(__link) { \
- typeof((list)->head) __tmp = __link; \
- __link = __link->next; \
- free(__tmp); \
- } \
- (list)->head = NULL; \
- (list)->tail = NULL; \
- (list)->length = 0; \
- } while(0);
- #define LIST__PASTEINNER(a, b) a ## b
- #define LIST__PASTE(a, b) LIST__PASTEINNER(a, b)
- #define LIST__FINISHED(iter) LIST__PASTE(LIST_finished__ ## iter ## __, __LINE__)
- #define LIST__MAINLOOP(iter) LIST__PASTE(LIST_main_loop__ ## iter ## __, __LINE__)
- #define LIST_LOOP(list, iter) \
- if(0) \
- LIST__FINISHED(iter): ; \
- else \
- for(typeof((list)->head) iter = (list)->head;;) \
- if(iter != NULL) \
- goto LIST__MAINLOOP(iter); \
- else \
- while(1) \
- if(1) { \
- goto LIST__FINISHED(iter); \
- } \
- else \
- while(1) \
- if(iter = iter->next, iter == NULL) { \
- goto LIST__FINISHED(iter); \
- } \
- else \
- LIST__MAINLOOP(iter) :
-
- // { user block; not in macro }
- #endif // __DS_H__
|