X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=src%2FAST%2Fpredicate.h;h=812d4bda8ec4d4abb70915fb9d8ae98923b3abce;hb=bb457e194713776587cab5987fc418160b406e76;hp=f9edc09e81d498bedcc4a2e441ff0ffa04e374e2;hpb=864af40e17218f8093ce2357c27a4fc4ff02bc24;p=satune.git

diff --git a/src/AST/predicate.h b/src/AST/predicate.h
index f9edc09..812d4bd 100644
--- a/src/AST/predicate.h
+++ b/src/AST/predicate.h
@@ -3,30 +3,43 @@
 #include "classlist.h"
 #include "mymemory.h"
 #include "ops.h"
+#include "astops.h"
 #include "structs.h"
+#include "common.h"
 
-#define GETPREDICATETYPE(o) (((Predicate *)(o))->type)
-
-struct Predicate {
+class Predicate {
+public:
+	Predicate(PredicateType _type) : type(_type) {}
+	virtual ~Predicate() {}
+	virtual Predicate *clone(CSolver *solver, CloneMap *map) {ASSERT(0); return NULL;}
+	virtual void serialize(Serializer *serializer) = 0;
+	virtual void print() = 0;
 	PredicateType type;
+	CMEMALLOC;
 };
 
-struct PredicateOperator {
-	Predicate base;
+class PredicateOperator : public Predicate {
+public:
+	PredicateOperator(CompOp op, Set **domain, uint numDomain);
+	bool evalPredicateOperator(uint64_t *inputs);
+	Predicate *clone(CSolver *solver, CloneMap *map);
+	virtual void serialize(Serializer *serializer);
+	virtual void print();
+	Array<Set *> domains;
+	CompOp getOp() {return op;}
+	CMEMALLOC;
+private:
 	CompOp op;
-	Set** domains;
-	int numDomains;
 };
 
-struct PredicateTable {
-	Predicate base;
-	Set** domains;
-	int numDomains;
-	Table* table;
+class PredicateTable : public Predicate {
+public:
+	PredicateTable(Table *table, UndefinedBehavior undefBehavior);
+	Predicate *clone(CSolver *solver, CloneMap *map);
+	virtual void serialize(Serializer *serializer);
+	virtual void print();
+	Table *table;
 	UndefinedBehavior undefinedbehavior;
+	CMEMALLOC;
 };
-
-
-Predicate* allocPredicate(CompOp op, Set ** domain, uint numDomain);
-void deletePredicate(Predicate* predicate);
 #endif