X-Git-Url: http://plrg.eecs.uci.edu/git/?p=satune.git;a=blobdiff_plain;f=src%2FBackend%2Fsatencoder.h;h=827a9041b059922d78607771352bf7b615cd9a34;hp=0d6999fddb55cf7d3a3d5ea741e70d331003d11b;hb=e0ee8656d201f77504bd239612969ce43636c324;hpb=e665793f5f5cf1cc673b4d21d7983fce30f89134

diff --git a/src/Backend/satencoder.h b/src/Backend/satencoder.h
index 0d6999f..827a904 100644
--- a/src/Backend/satencoder.h
+++ b/src/Backend/satencoder.h
@@ -6,36 +6,71 @@
 #include "inc_solver.h"
 #include "constraint.h"
 
-struct SATEncoder {
-	uint varcount;
-	CNF * cnf;
-};
+typedef Hashtable<Boolean *, Node *, uintptr_t, 4> BooleanToEdgeMap;
 
-SATEncoder * allocSATEncoder();
-void deleteSATEncoder(SATEncoder *This);
-void encodeAllSATEncoder(CSolver *csolver, SATEncoder *This);
-Edge getNewVarSATEncoder(SATEncoder *This);
-void getArrayNewVarsSATEncoder(SATEncoder* encoder, uint num, Edge*carray);
-Edge encodeConstraintSATEncoder(SATEncoder *This, Boolean *constraint);
-Edge encodeOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint);
-void createAllTotalOrderConstraintsSATEncoder(SATEncoder* This, Order* order);
-Edge getOrderConstraint(HashTableBoolConst *table, OrderPair *pair);
-Edge generateTransOrderConstraintSATEncoder(SATEncoder *This, Edge constIJ, Edge constJK, Edge constIK);
-Edge encodeTotalOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint);
-Edge encodePartialOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint);
-Edge encodeVarSATEncoder(SATEncoder *This, BooleanVar * constraint);
-Edge encodeLogicSATEncoder(SATEncoder *This, BooleanLogic * constraint);
-Edge encodePredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Edge encodeTablePredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Edge encodeEnumTablePredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Edge encodeOperatorPredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Edge encodeEnumOperatorPredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
+class SATEncoder {
+public:
+	int solve(long timeout);
+	SATEncoder(CSolver *solver);
+	~SATEncoder();
+	void resetSATEncoder();
+	void encodeAllSATEncoder(CSolver *csolver);
+	Edge encodeConstraintSATEncoder(BooleanEdge constraint);
+	CNF *getCNF() { return cnf;}
+	long long getSolveTime() { return cnf->solveTime; }
+	long long getEncodeTime() { return cnf->encodeTime; }
 
-Edge getElementValueBinaryIndexConstraint(SATEncoder * This, Element* element, uint64_t value);
-Edge getElementValueConstraint(SATEncoder* encoder, Element* This, uint64_t value);
+	CMEMALLOC;
+private:
+	void shouldMemoize(Element *elem, uint64_t val, bool &memo);
+	Edge getNewVarSATEncoder();
+	void getArrayNewVarsSATEncoder(uint num, Edge *carray);
+	Edge encodeVarSATEncoder(BooleanVar *constraint);
+	Edge encodeLogicSATEncoder(BooleanLogic *constraint);
+	Edge encodePredicateSATEncoder(BooleanPredicate *constraint);
+	Edge encodeTablePredicateSATEncoder(BooleanPredicate *constraint);
+	void encodeElementSATEncoder(Element *element);
+	void encodeElementFunctionSATEncoder(ElementFunction *function);
+	void encodeTableElementFunctionSATEncoder(ElementFunction *This);
+	Edge getElementValueOneHotConstraint(Element *elem, Polarity p, uint64_t value);
+	Edge getElementValueUnaryConstraint(Element *elem, Polarity p, uint64_t value);
+	Edge getElementValueBinaryIndexConstraint(Element *element, Polarity p, uint64_t value);
+	Edge getElementValueBinaryValueConstraint(Element *element, Polarity p, uint64_t value);
+	Edge getElementValueConstraint(Element *element, Polarity p, uint64_t value);
+	void generateOneHotEncodingVars(ElementEncoding *encoding);
+	void generateUnaryEncodingVars(ElementEncoding *encoding);
+	void generateBinaryIndexEncodingVars(ElementEncoding *encoding);
+	void generateBinaryValueEncodingVars(ElementEncoding *encoding);
+	void generateElementEncoding(Element *element);
+	Edge encodeOperatorPredicateSATEncoder(BooleanPredicate *constraint);
+	Edge encodeEnumOperatorPredicateSATEncoder(BooleanPredicate *constraint);
+	Edge encodeEnumEqualsPredicateSATEncoder(BooleanPredicate *constraint);
+	void encodeOperatorElementFunctionSATEncoder(ElementFunction *This);
+	Edge encodeCircuitOperatorPredicateEncoder(BooleanPredicate *constraint);
+	Edge encodeOrderSATEncoder(BooleanOrder *constraint);
+	Edge inferOrderConstraintFromGraph(Order *order, uint64_t _first, uint64_t _second);
+	Edge getPairConstraint(Order *order, OrderPair *pair);
+	Edge getPartialPairConstraint(Order *order, OrderPair *pair);
+	Edge encodeTotalOrderSATEncoder(BooleanOrder *constraint);
+	Edge encodePartialOrderSATEncoder(BooleanOrder *constraint);
+	void createAllTotalOrderConstraintsSATEncoder(Order *order);
+	void createAllPartialOrderConstraintsSATEncoder(Order *order);
+	Edge getOrderConstraint(HashtableOrderPair *table, OrderPair *pair);
+	Edge generateTransOrderConstraintSATEncoder(Edge constIJ, Edge constJK, Edge constIK);
+	Edge generatePartialOrderConstraintsSATEncoder(Edge ij,Edge ji, Edge jk, Edge kj,Edge ik, Edge ki);
+	Edge encodeEnumEntriesTablePredicateSATEncoder(BooleanPredicate *constraint);
+	Edge encodeEnumTablePredicateSATEncoder(BooleanPredicate *constraint);
+	void encodeEnumTableElemFunctionSATEncoder(ElementFunction *This);
+	void encodeEnumEntriesTableElemFuncSATEncoder(ElementFunction *This);
+	void generateAnyValueBinaryIndexEncoding(ElementEncoding *encoding);
+	void generateAnyValueBinaryIndexEncodingPositive(ElementEncoding *encoding);
+	void generateAnyValueBinaryValueEncoding(ElementEncoding *encoding);
+	CNF *cnf;
+	CSolver *solver;
+	BooleanToEdgeMap booledgeMap;
+	VectorEdge *vector;
+        friend class VarOrderingOpt;
+};
 
-Edge encodeFunctionElementSATEncoder(SATEncoder* encoder, ElementFunction *This);
-Edge encodeEnumTableElemFunctionSATEncoder(SATEncoder* encoder, ElementFunction* This);
-Edge encodeTableElementFunctionSATEncoder(SATEncoder* encoder, ElementFunction* This);
-Edge encodeOperatorElementFunctionSATEncoder(SATEncoder* encoder,ElementFunction* This);
+void allocElementConstraintVariables(ElementEncoding *ee, uint numVars);
 #endif