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 setExpandVector ## name(Vector ## name * vector, uint index, type item);   \
  20         void setVector ## name(Vector ## name * vector, uint index, type item); \
  21         uint getSizeVector ## name(const Vector ## name * vector);                                                                       \
  22         void setSizeVector ## name(Vector ## name * vector, uint size);                          \
  23         void deleteVector ## name(Vector ## name * vector);                                                                              \
  24         void clearVector ## name(Vector ## name * vector);                                                                               \
  25         void deleteVectorArray ## name(Vector ## name * vector);                                                         \
  26         type *exposeArray ## name(Vector ## name * vector);                                                                      \
  27         void initVector ## name(Vector ## name * vector, uint capacity); \
  28         void initDefVector ## name(Vector ## name * vector);                                            \
  29         void initVectorArray ## name(Vector ## name * vector, uint capacity, type * array);
  30
  31 #define VectorImpl(name, type, defcap)                                                                                                                                  \
  32         Vector ## name * allocDefVector ## name() {                                                                                                             \
  33                 return allocVector ## name(defcap);                                                                                                                                     \
  34         }                                                                                                                                                                                                                                                                                       \
  35         Vector ## name * allocVector ## name(uint capacity) {                                                                   \
  36                 Vector ## name * tmp = (Vector ## name *)ourmalloc(sizeof(Vector ## name));      \
  37                 tmp->size = 0;                                                                                                                                                                                                                  \
  38                 tmp->capacity = capacity;                                                                                                                                                                               \
  39                 tmp->array = (type *) ourmalloc(sizeof(type) * capacity);                                               \
  40                 return tmp;                                                                                                                                                                                                                                     \
  41         }                                                                                                                                                                                                                                                                                       \
  42         Vector ## name * allocVectorArray ## name(uint capacity, type * array)  {       \
  43                 Vector ## name * tmp = allocVector ## name(capacity);                                                           \
  44                 tmp->size = capacity;                                                                                                                                                                                                   \
  45                 memcpy(tmp->array, array, capacity * sizeof(type));                                                                     \
  46                 return tmp;                                                                                                                                                                                                                                     \
  47         }                                                                                                                                                                                                                                                                                       \
  48         void popVector ## name(Vector ## name * vector) {                                                                                        \
  49                 vector->size--;                                                                                                                                                                                                                 \
  50         }                                                                                                                                                                                                                                                                                       \
  51         type lastVector ## name(Vector ## name * vector) {                                                                               \
  52                 return vector->array[vector->size - 1];                                                                                                                         \
  53         }                                                                                                                                                                                                                                                                                       \
  54         void setSizeVector ## name(Vector ## name * vector, uint size) {                         \
  55                 if (size <= vector->size) {                                                                                                                                                                     \
  56                         vector->size = size;                                                                                                                                                                                            \
  57                         return;                                                                                                                                                                                                                                         \
  58                 } else if (size > vector->capacity) {                                                                                                                           \
  59                         vector->array = (type *)ourrealloc(vector->array, size * sizeof(type)); \
  60                         vector->capacity = size;                                                                                                                                                                                \
  61                 }                                                                                                                                                                                                                                                                               \
  62                 bzero(&vector->array[vector->size], (size - vector->size) * sizeof(type)); \
  63                 vector->size = size;                                                                                                                                                                                                    \
  64         }                                                                                                                                                                                                                                                                                       \
  65         void pushVector ## name(Vector ## name * vector, type item) {                                    \
  66                 if (vector->size >= vector->capacity) {                                                                                                                 \
  67                         uint newcap = vector->capacity << 1;                                                                                                                            \
  68                         vector->array = (type *)ourrealloc(vector->array, newcap * sizeof(type));       \
  69                         vector->capacity = newcap;                                                                                                                                                                      \
  70                 }                                                                                                                                                                                                                                                                               \
  71                 vector->array[vector->size++] = item;                                                                                                                           \
  72         }                                                                                                                                                                                                                                                                                       \
  73         type getVector ## name(Vector ## name * vector, uint index) {                                   \
  74                 return vector->array[index];                                                                                                                                                            \
  75         }                                                                                                                                                                                                                                                                                       \
  76         void setExpandVector ## name(Vector ## name * vector, uint index, type item) { \
  77                 if (index >= vector->size)                                                                                                                                                                              \
  78                         setSizeVector ## name(vector, index + 1);                                                                                                       \
  79                 setVector ## name(vector, index, item);                                                                                                                 \
  80         }                                                                                                                                                                                                                                                                                       \
  81         void setVector ## name(Vector ## name * vector, uint index, type item) { \
  82                 vector->array[index] = item;                                                                                                                                                                    \
  83         }                                                                                                                                                                                                                                                                                       \
  84         uint getSizeVector ## name(const Vector ## name * vector) {                                             \
  85                 return vector->size;                                                                                                                                                                                            \
  86         }                                                                                                                                                                                                                                                                                       \
  87         void deleteVector ## name(Vector ## name * vector) {                                                                             \
  88                 ourfree(vector->array);                                                                                                                                                                                 \
  89                 ourfree(vector);                                                                                                                                                                                                                \
  90         }                                                                                                                                                                                                                                                                                       \
  91         void clearVector ## name(Vector ## name * vector) {                                                                                     \
  92                 vector->size = 0;                                                                                                                                                                                                                       \
  93         }                                                                                                                                                                                                                                                                                       \
  94         type *exposeArray ## name(Vector ## name * vector) {                                                             \
  95                 return vector->array;                                                                                                                                                                                           \
  96         }                                                                                                                                                                                                                                                                                       \
  97         void deleteVectorArray ## name(Vector ## name * vector) {                                                        \
  98                 ourfree(vector->array);                                                                                                                                                                                 \
  99         }                                                                                                                                                                                                                                                                                       \
 100         void initVector ## name(Vector ## name * vector, uint capacity) {       \
 101                 vector->size = 0;                                                                                                                                                                                                                        \
 102                 vector->capacity = capacity;                                                                                                                                                                                    \
 103                 vector->array = (type *) ourmalloc(sizeof(type) * capacity);                            \
 104         }                                                                                                                                                                                                                                                                                       \
 105         void initDefVector ## name(Vector ## name * vector) {                                   \
 106                 initVector ## name(vector, defcap);                                                                                                     \
 107         }                                                                                                                                                                                                                                                                                       \
 108         void initVectorArray ## name(Vector ## name * vector, uint capacity, type * array) {    \
 109                 initVector ## name(vector, capacity);                                                                                                    \
 110                 vector->size = capacity;                                                                                                                                                                                        \
 111                 memcpy(vector->array, array, capacity * sizeof(type));  \
 112         }
 113 #endif