array.c

   1 #include <string.h> // for memmove, memcpy
   2 #include <stdlib.h> // for malloc, realloc, free, qsort
   3 #include "array.h"
   4
   5 void array_ini(Array *array, size_t mbs, size_t ext, ArrayComparator cmp)
   6 {
   7         array->mbs = mbs;
   8         array->ext = ext;
   9         array->cmp = cmp;
  10         array->siz = 0;
  11         array->cap = 0;
  12         array->ptr = NULL;
  13 }
  14
  15 void array_rlc(Array *array)
  16 {
  17         array->ptr = realloc(array->ptr, array->cap * array->mbs);
  18 }
  19
  20 void array_grw(Array *array, size_t n)
  21 {
  22         array->siz += n;
  23
  24         if (array->siz > array->cap) {
  25                 array->cap = array->siz + array->ext;
  26                 array_rlc(array);
  27         }
  28 }
  29
  30 void array_shr(Array *array, size_t n)
  31 {
  32         array->siz -= n;
  33
  34         if (array->cap > array->siz) {
  35                 array->cap = array->siz;
  36                 array_rlc(array);
  37         }
  38 }
  39
  40 void array_put(Array *array, const void *ptr, size_t n)
  41 {
  42         size_t oldsiz = array->siz;
  43         array_grw(array, 1);
  44
  45         char *iptr = array->ptr + n * array->mbs;
  46         memmove(iptr + array->mbs, iptr, (oldsiz - n) * array->mbs);
  47         memcpy(iptr, ptr, array->mbs);
  48 }
  49
  50 void array_add(Array *array, const void *ptr)
  51 {
  52         size_t oldsiz = array->siz;
  53         array_grw(array, 1);
  54
  55         memcpy(array->ptr + oldsiz * array->mbs, ptr, array->mbs);
  56 }
  57
  58 void array_cpy(Array *array, void **ptr, size_t *n)
  59 {
  60         *n = array->siz;
  61         size_t size = array->siz * array->mbs;
  62         *ptr = malloc(size);
  63         memcpy(*ptr, array->ptr, size);
  64 }
  65
  66 void array_cln(Array *dst, Array *src)
  67 {
  68         array_ini(dst, src->mbs, src->ext, src->cmp);
  69         array_cpy(src, &dst->ptr, &dst->siz);
  70         dst->cap = dst->siz;
  71 }
  72
  73 void array_rcy(Array *array)
  74 {
  75         array->siz = 0;
  76 }
  77
  78 void array_del(Array *array)
  79 {
  80         if (array->ptr)
  81                 free(array->ptr);
  82 }
  83
  84 void array_srt(Array *array)
  85 {
  86         qsort(array->ptr, array->siz, array->mbs, array->cmp);
  87 }
  88
  89 ssize_t array_fnd(Array *array, const void *ptr, size_t *idx)
  90 {
  91         size_t low, high, mid;
  92
  93         low = 0;
  94         high = array->siz;
  95
  96         while (low < high) {
  97                 mid = (low + high) / 2;
  98
  99                 int cmp = array->cmp(ptr, array->ptr + mid * array->mbs);
 100
 101                 if (cmp == 0)
 102                         return idx ? (*idx = mid) : mid;
 103                 else if (cmp < 0)
 104                         high = mid;
 105                 else // (cmp > 0)
 106                         low = mid + 1;
 107         }
 108
 109         if (idx)
 110                 *idx = low;
 111
 112         return -1;
 113 }
 114
 115 size_t array_ins(Array *array, const void *ptr)
 116 {
 117         size_t n;
 118
 119         array_fnd(array, ptr, &n);
 120         array_put(array, ptr, n);
 121
 122         return n;
 123 }