ucw/gary.c

   1 /*
   2  *      UCW Library -- A simple growing array of an arbitrary type
   3  *
   4  *      (c) 2010--2014 Martin Mares <mj@ucw.cz>
   5  */
   6
   7 #undef LOCAL_DEBUG
   8
   9 #include <ucw/lib.h>
  10 #include <ucw/gary.h>
  11
  12 #include <string.h>
  13
  14 struct gary_hdr gary_empty_hdr;
  15
  16 void *
  17 gary_init(size_t elt_size, size_t num_elts, struct ucw_allocator *allocator)
  18 {
  19   DBG("GARY: Init to %zd elements", num_elts);
  20   struct gary_hdr *h = allocator->alloc(allocator, GARY_HDR_SIZE + elt_size * num_elts);
  21   h->num_elts = h->have_space = num_elts;
  22   h->elt_size = elt_size;
  23   h->allocator = allocator;
  24   return GARY_BODY(h);
  25 }
  26
  27 static struct gary_hdr *
  28 gary_realloc(struct gary_hdr *h, size_t n)
  29 {
  30   size_t old_size = h->have_space;
  31   if (n > 2*h->have_space)
  32     h->have_space = n;
  33   else
  34     h->have_space *= 2;
  35   DBG("GARY: Resize from %zd to %zd elements (need %zd)", old_size, h->have_space, n);
  36   return h->allocator->realloc(h->allocator, h, GARY_HDR_SIZE + old_size * h->elt_size, GARY_HDR_SIZE + h->have_space * h->elt_size);
  37 }
  38
  39 void *
  40 gary_set_size(void *array, size_t n)
  41 {
  42   struct gary_hdr *h = GARY_HDR(array);
  43   h->num_elts = n;
  44   if (n <= h->have_space)
  45     return array;
  46
  47   h = gary_realloc(h, n);
  48   return GARY_BODY(h);
  49 }
  50
  51 void *
  52 gary_push_helper(void *array, size_t n, byte **cptr)
  53 {
  54   struct gary_hdr *h = GARY_HDR(array);
  55   h = gary_realloc(h, h->num_elts);
  56   *cptr = GARY_BODY(h) + (h->num_elts - n) * h->elt_size;
  57   return GARY_BODY(h);
  58 }
  59
  60 void *
  61 gary_fix(void *array)
  62 {
  63   struct gary_hdr *h = GARY_HDR(array);
  64   if (h->num_elts != h->have_space)
  65     {
  66       h = h->allocator->realloc(h->allocator, h, GARY_HDR_SIZE + h->have_space * h->elt_size, GARY_HDR_SIZE + h->num_elts * h->elt_size);
  67       h->have_space = h->num_elts;
  68     }
  69   return GARY_BODY(h);
  70 }
  71
  72 #ifdef TEST
  73
  74 #include <ucw/mempool.h>
  75
  76 #include <stdio.h>
  77
  78 int main(int argc, char **argv UNUSED)
  79 {
  80   int *a;
  81   struct mempool *mp = NULL;
  82
  83   if (argc < 2)
  84     GARY_INIT_ZERO(a, 5);
  85   else
  86     {
  87       mp = mp_new(4096);
  88       GARY_INIT_ALLOC(a, 5, mp_get_allocator(mp));
  89     }
  90
  91   for (int i=0; i<5; i++)
  92     {
  93       ASSERT(!a[i] || mp);
  94       a[i] = i+1;
  95     }
  96
  97   GARY_PUSH(a);
  98   *GARY_PUSH(a) = 10;
  99   *GARY_PUSH(a) = 20;
 100   *GARY_PUSH(a) = 30;
 101   GARY_POP(a);
 102   GARY_FIX(a);
 103
 104   for (int i=0; i<(int)GARY_SIZE(a); i++)
 105     printf("%d\n", a[i]);
 106
 107   GARY_FREE(a);
 108   if (mp)
 109     mp_delete(mp);
 110   return 0;
 111 }
 112
 113 #endif