#include "classlist.h"
#include "structs.h"
#include "inc_solver.h"
+#include "constraint.h"
-struct SATEncoder {
- uint varcount;
- IncrementalSolver* satSolver;
-};
+typedef Hashtable<Boolean *, Node *, uintptr_t, 4> BooleanToEdgeMap;
+
+class SATEncoder {
+public:
+ int solve();
+ SATEncoder(CSolver *solver);
+ ~SATEncoder();
+ void encodeAllSATEncoder(CSolver *csolver);
+ Edge encodeConstraintSATEncoder(BooleanEdge constraint);
+ CNF *getCNF() { return cnf;}
+ long long getSolveTime() { return cnf->solveTime; }
+ long long getEncodeTime() { return cnf->encodeTime; }
-SATEncoder * allocSATEncoder();
-void deleteSATEncoder(SATEncoder *This);
-void encodeAllSATEncoder(CSolver *csolver, SATEncoder *This);
-Constraint * getNewVarSATEncoder(SATEncoder *This);
-void getArrayNewVarsSATEncoder(SATEncoder* encoder, uint num, Constraint **carray);
-Constraint * encodeConstraintSATEncoder(SATEncoder *This, Boolean *constraint);
-Constraint * encodeOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint);
-Constraint * createAllTotalOrderConstraintsSATEncoder(SATEncoder* This, Order* order);
-Constraint* getOrderConstraint(HashTableBoolConst *table, OrderPair *pair);
-Constraint * generateTransOrderConstraintSATEncoder(SATEncoder *This, Constraint *constIJ,Constraint *constJK,Constraint *constIK);
-Constraint * encodeTotalOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint);
-Constraint * encodePartialOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint);
-Constraint * encodeVarSATEncoder(SATEncoder *This, BooleanVar * constraint);
-Constraint * encodeLogicSATEncoder(SATEncoder *This, BooleanLogic * constraint);
-Constraint * encodePredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Constraint * encodeTablePredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Constraint * encodeEnumTablePredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Constraint * encodeOperatorPredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
-Constraint * encodeEnumOperatorPredicateSATEncoder(SATEncoder * This, BooleanPredicate * constraint);
+ CMEMALLOC;
+private:
+ 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, uint64_t value);
+ Edge getElementValueUnaryConstraint(Element *elem, uint64_t value);
+ Edge getElementValueBinaryIndexConstraint(Element *element, uint64_t value);
+ Edge getElementValueBinaryValueConstraint(Element *element, uint64_t value);
+ Edge getElementValueConstraint(Element *element, 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);
+ 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);
-Constraint * getElementValueBinaryIndexConstraint(Element* This, uint64_t value);
-Constraint * getElementValueConstraint(SATEncoder* encoder, Element* This, uint64_t value);
+ CNF *cnf;
+ CSolver *solver;
+ BooleanToEdgeMap booledgeMap;
+
+};
-Constraint* encodeFunctionElementSATEncoder(SATEncoder* encoder, ElementFunction *This);
-Constraint* encodeEnumTableElemFunctionSATEncoder(SATEncoder* encoder, ElementFunction* This);
-Constraint* encodeTableElementFunctionSATEncoder(SATEncoder* encoder, ElementFunction* This);
-Constraint* encodeOperatorElementFunctionSATEncoder(SATEncoder* encoder,ElementFunction* This);
-void addConstraintToSATSolver(Constraint *c, IncrementalSolver* satSolver);
+void allocElementConstraintVariables(ElementEncoding *ee, uint numVars);
#endif