src/Collections/vector.h

   1 #ifndef VECTOR_H
   2 #define VECTOR_H
   3 #include <string.h>
   4
   5 #define VectorDef(name, type)                                                                                                                                                                           \
   6         struct Vector ## name {                                               \
   7                 uint size;                                                          \
   8                 uint capacity;                                                      \
   9                 type * array;                                                       \
  10         };                                                                    \
  11         typedef struct Vector ## name Vector ## name;                         \
  12         Vector ## name * allocVector ## name(uint capacity);                  \
  13         Vector ## name * allocDefVector ## name();                            \
  14         Vector ## name * allocVectorArray ## name(uint capacity, type * array); \
  15         void pushVector ## name(Vector ## name *vector, type item);           \
  16         type lastVector ## name(Vector ## name *vector);                                                                                        \
  17         void popVector ## name(Vector ## name *vector);                                                                                         \
  18         type getVector ## name(Vector ## name *vector, uint index);                                             \
  19         void setVector ## name(Vector ## name *vector, uint index, type item); \
  20         uint getSizeVector ## name(Vector ## name *vector);                   \
  21         void setSizeVector ## name(Vector ## name *vector, uint size);                          \
  22         void deleteVector ## name(Vector ## name *vector);                    \
  23         void clearVector ## name(Vector ## name *vector);                     \
  24         void deleteVectorArray ## name(Vector ## name *vector);                                                         \
  25         type * exposeArray ## name(Vector ## name * vector);                                                                    \
  26         void allocInlineVector ## name(Vector ## name * vector, uint capacity); \
  27         void allocInlineDefVector ## name(Vector ## name * vector);                                             \
  28         void allocInlineVectorArray ## name(Vector ## name * vector, uint capacity, type * array);
  29
  30 #define VectorImpl(name, type, defcap)                                  \
  31         Vector ## name * allocDefVector ## name() {                           \
  32                 return allocVector ## name(defcap);                                 \
  33         }                                                                     \
  34         Vector ## name * allocVector ## name(uint capacity) {                 \
  35                 Vector ## name * tmp = (Vector ## name *)ourmalloc(sizeof(Vector ## name));  \
  36                 tmp->size = 0;                                                      \
  37                 tmp->capacity = capacity;                                           \
  38                 tmp->array = (type *) ourcalloc(1, sizeof(type) * capacity);          \
  39                 return tmp;                                                         \
  40         }                                                                     \
  41         Vector ## name * allocVectorArray ## name(uint capacity, type * array)  { \
  42                 Vector ## name * tmp = allocVector ## name(capacity);               \
  43                 tmp->size=capacity;                                                                                                                                                                                                     \
  44                 memcpy(tmp->array, array, capacity * sizeof(type));                                                                     \
  45                 return tmp;                                                         \
  46         }                                                                     \
  47         void popVector ## name(Vector ## name *vector) {                                                                                        \
  48                 vector->size--;                                                                                                                                                                                                                 \
  49         }                                                                                                                                                                                                                                                                                       \
  50         type lastVector ## name(Vector ## name *vector) {                                                                                       \
  51                 return vector->array[vector->size-1];                                                                                                                           \
  52         }                                                                                                                                                                                                                                                                                       \
  53         void setSizeVector ## name(Vector ## name *vector, uint size) {                         \
  54                 if (size <= vector->size) {                                                                                                                                                                     \
  55                         vector->size=size;                                                                                                                                                                                              \
  56                         return;                                                                                                                                                                                                                                         \
  57                 } else if (size > vector->capacity) {                                                                                                                           \
  58                         vector->array=(type *)ourrealloc(vector->array, size * sizeof(type));   \
  59                         vector->capacity=size;                                                                                                                                                                          \
  60                 }                                                                                                                                                                                                                                                                               \
  61                 bzero(&vector->array[vector->size], (size-vector->size)*sizeof(type)); \
  62                 vector->size=size;                                                                                                                                                                                                      \
  63         }                                                                                                                                                                                                                                                                                       \
  64         void pushVector ## name(Vector ## name *vector, type item) {                                    \
  65                 if (vector->size >= vector->capacity) {                                                                                                                 \
  66                         uint newcap=vector->capacity << 1;                                                                                                                              \
  67                         vector->array=(type *)ourrealloc(vector->array, newcap * sizeof(type)); \
  68                 }                                                                   \
  69                 vector->array[vector->size++] = item;                               \
  70         }                                                                     \
  71         type getVector ## name(Vector ## name * vector, uint index) {         \
  72                 return vector->array[index];                                        \
  73         }                                                                     \
  74         void setVector ## name(Vector ## name * vector, uint index, type item) { \
  75                 vector->array[index]=item;                                          \
  76         }                                                                     \
  77         uint getSizeVector ## name(Vector ## name *vector) {                  \
  78                 return vector->size;                                                \
  79         }                                                                     \
  80         void deleteVector ## name(Vector ## name *vector) {                     \
  81                 ourfree(vector->array);                                             \
  82                 ourfree(vector);                                                    \
  83         }                                                                     \
  84         void clearVector ## name(Vector ## name *vector) {                     \
  85                 vector->size=0;                                                     \
  86         }                                                                     \
  87         type * exposeArray ## name(Vector ## name * vector) {                 \
  88                 return vector->array;                                               \
  89         }                                                                                                                                                                                                                                                                                       \
  90         void deleteVectorArray ## name(Vector ## name *vector) {                                                        \
  91                 ourfree(vector->array);                                             \
  92         }                                                                                                                                                                                                                                                                                       \
  93         void allocInlineVector ## name(Vector ## name * vector, uint capacity) { \
  94                 vector->size = 0;                                                      \
  95                 vector->capacity = capacity;                                                                                                                                                                                    \
  96                 vector->array = (type *) ourcalloc(1, sizeof(type) * capacity);                 \
  97         }                                                                                                                                                                                                                                                                                       \
  98         void allocInlineDefVector ## name(Vector ## name * vector) {                                    \
  99                 allocInlineVector ## name(vector, defcap);                                                                                                      \
 100         }                                                                                                                                                                                                                                                                                       \
 101         void allocInlineVectorArray ## name(Vector ## name * vector, uint capacity, type * array) {     \
 102                 allocInlineVector ##name(vector, capacity);                                                                                                     \
 103                 vector->size=capacity;                                                                                                                                                                                  \
 104                 memcpy(vector->array, array, capacity * sizeof(type));  \
 105         }
 106 #endif