#include "set.h"
#include <stddef.h>
+#include "csolver.h"
-Set *allocSet(VarType t, uint64_t *elements, uint num) {
- Set *This = (Set *)ourmalloc(sizeof(Set));
- This->type = t;
- This->isRange = false;
- This->low = 0;
- This->high = 0;
- This->members = allocVectorArrayInt(num, elements);
- return This;
-}
-
-Set *allocSetRange(VarType t, uint64_t lowrange, uint64_t highrange) {
- Set *This = (Set *)ourmalloc(sizeof(Set));
- This->type = t;
- This->isRange = true;
- This->low = lowrange;
- This->high = highrange;
- This->members = NULL;
- return This;
-}
-
-bool existsInSet(Set *This, uint64_t element) {
- if (This->isRange) {
- return element >= This->low && element <= This->high;
+Set::Set(VarType t) : type(t), isRange(false), low(0), high(0) {
+ members = new Vector<uint64_t>();
+}
+
+Set::Set(VarType t, uint64_t *elements, uint num) : type(t), isRange(false), low(0), high(0) {
+ members = new Vector<uint64_t>(num, elements);
+}
+
+Set::Set(VarType t, uint64_t lowrange, uint64_t highrange) : type(t), isRange(true), low(lowrange), high(highrange), members(NULL) {
+}
+
+bool Set::exists(uint64_t element) {
+ if (isRange) {
+ return element >= low && element <= high;
} else {
- uint size = getSizeVectorInt(This->members);
+ uint size = members->getSize();
for (uint i = 0; i < size; i++) {
- if (element == getVectorInt(This->members, i))
+ if (element == members->get(i))
return true;
}
return false;
}
}
-uint64_t getSetElement(Set *This, uint index) {
- if (This->isRange)
- return This->low + index;
+uint64_t Set::getElement(uint index) {
+ if (isRange)
+ return low + index;
else
- return getVectorInt(This->members, index);
+ return members->get(index);
}
-uint getSetSize(Set *This) {
- if (This->isRange) {
- return This->high - This->low + 1;
+uint Set::getSize() {
+ if (isRange) {
+ return high - low + 1;
} else {
- return getSizeVectorInt(This->members);
+ return members->getSize();
}
}
-void deleteSet(Set *This) {
- if (!This->isRange)
- deleteVectorInt(This->members);
- ourfree(This);
+Set::~Set() {
+ if (!isRange)
+ delete members;
+}
+
+Set *Set::clone(CSolver *solver, CloneMap *map) {
+ Set *s = (Set *) map->get(this);
+ if (s != NULL)
+ return s;
+ if (isRange) {
+ s = solver->createRangeSet(type, low, high);
+ } else {
+ s = solver->createSet(type, members->expose(), members->getSize());
+ }
+ map->put(this, s);
+ return s;
}