WINGs/bag.c

   1
   2
   3 #include <stdlib.h>
   4 #include <string.h>
   5
   6 #include "WUtil.h"
   7
   8
   9
  10 struct W_Bag {
  11     int size;
  12     int count;
  13
  14     void **items;
  15 };
  16
  17
  18
  19
  20 WMBag*
  21 WMCreateBag(int size)
  22 {
  23     WMBag *bag;
  24
  25     wassertrv(size > 0, NULL);
  26
  27     bag = wmalloc(sizeof(WMBag));
  28
  29     bag->items = wmalloc(sizeof(void*) * size);
  30     bag->size = size;
  31     bag->count = 0;
  32
  33     return bag;
  34 }
  35
  36
  37
  38 int
  39 WMGetBagItemCount(WMBag *bag)
  40 {
  41     return bag->count;
  42 }
  43
  44
  45
  46 void
  47 WMAppendBag(WMBag *bag, WMBag *appendedBag)
  48 {
  49     bag->items = wrealloc(bag->items,
  50                           sizeof(void*) * (bag->size+appendedBag->count));
  51
  52     memcpy(bag->items + bag->count, appendedBag->items, appendedBag->count);
  53
  54     bag->count += appendedBag->count;
  55 }
  56
  57
  58
  59 void
  60 WMPutInBag(WMBag *bag, void *item)
  61 {
  62     WMInsertInBag(bag, bag->count, item);
  63 }
  64
  65
  66
  67 void
  68 WMInsertInBag(WMBag *bag, int index, void *item)
  69 {
  70     if (bag->count == bag->size) {
  71         bag->size += 16;
  72         bag->items = wrealloc(bag->items, sizeof(void*) * bag->size);
  73     }
  74
  75     bag->items[bag->count++] = item;
  76 }
  77
  78
  79 int
  80 WMGetFirstInBag(WMBag *bag, void *item)
  81 {
  82     int i;
  83
  84     for (i = 0; i < bag->count; i++) {
  85         if (bag->items[i] == item) {
  86             return i;
  87         }
  88     }
  89     return -1;
  90 }
  91
  92
  93 int
  94 WMGetLastInBag(WMBag *bag, void *item)
  95 {
  96     int i;
  97
  98     for (i = bag->count-1; i>= 0; i--) {
  99         if (bag->items[i] == item) {
 100             return i;
 101         }
 102     }
 103     return -1;
 104 }
 105
 106
 107 void
 108 WMRemoveFromBag(WMBag *bag, void *item)
 109 {
 110     int i;
 111
 112     i = WMGetFirstInBag(bag, item);
 113     if (i >= 0) {
 114         WMDeleteFromBag(bag, i);
 115     }
 116 }
 117
 118
 119
 120 void
 121 WMDeleteFromBag(WMBag *bag, int index)
 122 {
 123     if (index < 0 || index >= bag->count)
 124         return;
 125
 126     if (index < bag->count-1) {
 127         memmove(&bag->items[index], &bag->items[index + 1],
 128                 (bag->count - index - 1) * sizeof(void*));
 129     }
 130     bag->count--;
 131 }
 132
 133
 134 void*
 135 WMGetFromBag(WMBag *bag, int index)
 136 {
 137     if (index < 0 || index >= bag->count) {
 138         return NULL;
 139     }
 140
 141     return bag->items[index];
 142 }
 143
 144
 145 int
 146 WMCountInBag(WMBag *bag, void *item)
 147 {
 148     int i, j;
 149
 150     for (j = 0, i = 0; j < bag->count; j++) {
 151         if (bag->items[j] == item)
 152             i++;
 153     }
 154     return i;
 155 }
 156
 157
 158
 159 void
 160 WMSortBag(WMBag *bag, int (*comparer)(void*, void*))
 161 {
 162     qsort(bag->items, bag->count, sizeof(void*), comparer);
 163 }
 164
 165
 166
 167 void
 168 WMFreeBag(WMBag *bag)
 169 {
 170     free(bag->items);
 171     free(bag);
 172 }
 173
 174
 175 void
 176 WMEmptyBag(WMBag *bag)
 177 {
 178     bag->count = 0;
 179 }
 180
 181
 182
 183 WMBag*
 184 WMMapBag(WMBag *bag, void *(*function)(void *))
 185 {
 186     int i;
 187     WMBag *new = WMCreateBag(bag->size);
 188
 189     for (i = 0; i < bag->count; i++) {
 190         WMPutInBag(new, (*function)(bag->items[i]));
 191     }
 192
 193     return new;
 194 }
 195