X-Git-Url: http://plrg.eecs.uci.edu/git/?p=satune.git;a=blobdiff_plain;f=src%2FAST%2Fboolean.h;h=1f218b2cd855fde36aeab02738bfd917cda6dd46;hp=2341fcfdf4edbc873515d2ffc105214cba3d9324;hb=238830745ba3939323c285b722c9ae512b7baa9b;hpb=919a30124bede022af832c5e33e42151f7c38533

diff --git a/src/AST/boolean.h b/src/AST/boolean.h
index 2341fcf..1f218b2 100644
--- a/src/AST/boolean.h
+++ b/src/AST/boolean.h
@@ -3,72 +3,100 @@
 #include "classlist.h"
 #include "mymemory.h"
 #include "ops.h"
+#include "astops.h"
 #include "structs.h"
 #include "astnode.h"
 #include "functionencoding.h"
 #include "constraint.h"
-
-/**
-    This is a little sketchy, but apparently legit.
-    https://www.python.org/dev/peps/pep-3123/ */
-
-#define GETBOOLEANTYPE(o) (o->type)
-#define GETBOOLEANPARENTS(o) (&(o->parents))
-#define GETBOOLEANPOLARITY(b) (b->polarity)
-#define GETBOOLEANVALUE(b) (b->boolVal)
+#include "serializer.h"
 
 class Boolean : public ASTNode {
+private:
+    static uint64_t counter;
 public:
 	Boolean(ASTNodeType _type);
 	virtual ~Boolean() {}
-	virtual Boolean * clone(CSolver * solver, CloneMap *map);
+	virtual Boolean *clone(CSolver *solver, CloneMap *map) = 0;
+	virtual void serialize(Serializer * ) = 0;
+	virtual void print() = 0;
+	virtual bool isTrue() {return boolVal == BV_MUSTBETRUE;}
+	virtual bool isFalse() {return boolVal == BV_MUSTBEFALSE;}
 	Polarity polarity;
 	BooleanValue boolVal;
 	Vector<Boolean *> parents;
-	MEMALLOC;
+	virtual void updateParents() {}
+        uint64_t id;
+	CMEMALLOC;
+};
+
+class BooleanConst : public Boolean {
+public:
+	BooleanConst(bool isTrue);
+	Boolean *clone(CSolver *solver, CloneMap *map);
+	bool isTrue() {return istrue;}
+	bool isFalse() {return !istrue;}
+	void serialize(Serializer *serializer ) {};
+	virtual void print();
+	bool istrue;
+	CMEMALLOC;
 };
 
 class BooleanVar : public Boolean {
 public:
 	BooleanVar(VarType t);
-	Boolean * clone(CSolver * solver, CloneMap *map);
-	
+	Boolean *clone(CSolver *solver, CloneMap *map);
+	void serialize(Serializer *serializer );
+	virtual void print();
 	VarType vtype;
 	Edge var;
-	MEMALLOC;
+	CMEMALLOC;
 };
 
 class BooleanOrder : public Boolean {
 public:
 	BooleanOrder(Order *_order, uint64_t _first, uint64_t _second);
-	Boolean * clone(CSolver * solver, CloneMap *map);
-	
+	Boolean *clone(CSolver *solver, CloneMap *map);
+	void serialize(Serializer *serializer );
+	virtual void print();
+	void updateParents();
+
 	Order *order;
 	uint64_t first;
 	uint64_t second;
-	MEMALLOC;
+	CMEMALLOC;
 };
 
 class BooleanPredicate : public Boolean {
 public:
-	BooleanPredicate(Predicate *_predicate, Element **_inputs, uint _numInputs, Boolean *_undefinedStatus);
-	Boolean * clone(CSolver * solver, CloneMap *map);
-	
+	BooleanPredicate(Predicate *_predicate, Element **_inputs, uint _numInputs, BooleanEdge _undefinedStatus);
+	Boolean *clone(CSolver *solver, CloneMap *map);
+	Predicate *getPredicate() {return predicate;}
+	FunctionEncoding *getFunctionEncoding() {return &encoding;}
+	void updateParents();
+	void serialize(Serializer *serializer );
+	virtual void print();
+	CMEMALLOC;
+
 	Predicate *predicate;
 	FunctionEncoding encoding;
 	Array<Element *> inputs;
-	Boolean *undefStatus;
-	FunctionEncoding *getFunctionEncoding() {return &encoding;}
-	MEMALLOC;
+	BooleanEdge undefStatus;
 };
 
 class BooleanLogic : public Boolean {
 public:
-	BooleanLogic(CSolver *solver, LogicOp _op, Boolean **array, uint asize);
-	Boolean * clone(CSolver * solver, CloneMap *map);
-
+	BooleanLogic(CSolver *solver, LogicOp _op, BooleanEdge *array, uint asize);
+	Boolean *clone(CSolver *solver, CloneMap *map);
+	void serialize(Serializer *serializer );
+	virtual void print();
 	LogicOp op;
-	Array<Boolean *> inputs;
-	MEMALLOC;
+	bool replaced;
+	Array<BooleanEdge> inputs;
+	void updateParents();
+
+	CMEMALLOC;
 };
+BooleanEdge cloneEdge(CSolver *solver, CloneMap *map, BooleanEdge e);
+
+
 #endif