src/AST/set.cc

   1 #include "set.h"
   2 #include <stddef.h>
   3
   4 Set *allocSet(VarType t, uint64_t *elements, uint num) {
   5         Set *This = (Set *)ourmalloc(sizeof(Set));
   6         This->type = t;
   7         This->isRange = false;
   8         This->low = 0;
   9         This->high = 0;
  10         This->members = allocVectorArrayInt(num, elements);
  11         return This;
  12 }
  13
  14 Set *allocSetRange(VarType t, uint64_t lowrange, uint64_t highrange) {
  15         Set *This = (Set *)ourmalloc(sizeof(Set));
  16         This->type = t;
  17         This->isRange = true;
  18         This->low = lowrange;
  19         This->high = highrange;
  20         This->members = NULL;
  21         return This;
  22 }
  23
  24 bool existsInSet(Set *This, uint64_t element) {
  25         if (This->isRange) {
  26                 return element >= This->low && element <= This->high;
  27         } else {
  28                 uint size = getSizeVectorInt(This->members);
  29                 for (uint i = 0; i < size; i++) {
  30                         if (element == getVectorInt(This->members, i))
  31                                 return true;
  32                 }
  33                 return false;
  34         }
  35 }
  36
  37 uint64_t getSetElement(Set *This, uint index) {
  38         if (This->isRange)
  39                 return This->low + index;
  40         else
  41                 return getVectorInt(This->members, index);
  42 }
  43
  44 uint getSetSize(Set *This) {
  45         if (This->isRange) {
  46                 return This->high - This->low + 1;
  47         } else {
  48                 return getSizeVectorInt(This->members);
  49         }
  50 }
  51
  52 void deleteSet(Set *This) {
  53         if (!This->isRange)
  54                 deleteVectorInt(This->members);
  55         ourfree(This);
  56 }