X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=src%2FAST%2Fset.h;h=eb406bb553c6ed4feded557fc57454a46dd0b13c;hb=8014c371c763fbc3fe65ec15e5320c1a9b859a8d;hp=92315e22818728089a1d543021aea773529122bc;hpb=2fdb3e8b65ae7e1490997e5df9ad8cc0929e1589;p=satune.git

diff --git a/src/AST/set.h b/src/AST/set.h
index 92315e2..eb406bb 100644
--- 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 */