Adding support for mutableset in backend + OO-style Set

[satune.git] / src / AST / set.h

diff --git a/src/AST/set.h b/src/AST/set.h
index 92315e22818728089a1d543021aea773529122bc..eb406bb553c6ed4feded557fc57454a46dd0b13c 100644 (file)
--- a/src/AST/set.h
+++ b/src/AST/set.h
@@ -12,17 +12,28 @@
 #include "structs.h"
 #include "mymemory.h"
 
-struct Set {
+class Set {
+public:
+       Set(VarType t);
+       Set(VarType t, uint64_t *elements, uint num);
+       Set(VarType t, uint64_t lowrange, uint64_t highrange);
+       virtual ~Set();
+       bool exists(uint64_t element);
+       uint getSize();
+       VarType getType(){return type;}
+       uint64_t getNewUniqueItem(){return low++;}
+       uint64_t getMemberAt(uint index);
+       uint64_t getElement(uint index);
+       virtual Set *clone(CSolver *solver, CloneMap *map);
+       MEMALLOC;
+protected:
        VarType type;
        bool isRange;
        uint64_t low;//also used to count unique items
        uint64_t high;
-       VectorInt * members;
+       Vector<uint64_t> *members;
+       
 };
 
-
-Set *allocSet(VarType t, uint64_t * elements, uint num);
-Set * allocSetRange(VarType t, uint64_t lowrange, uint64_t highrange);
-void deleteSet(Set *set);
 #endif/* SET_H */