sparse or decompose

[satune.git] / src / Backend / satencoder.h

diff --git a/src/Backend/satencoder.h b/src/Backend/satencoder.h
index dee131dc5b4dd584962e0e20d000d76fcad8f3c9..84329ef4e5295317122b6ca2e5579afb67d87a6d 100644 (file)
--- a/src/Backend/satencoder.h
+++ b/src/Backend/satencoder.h
@@ -6,31 +6,74 @@
 #include "inc_solver.h"
 #include "constraint.h"
 
-struct SATEncoder {
-       uint varcount;
+typedef Hashtable<Boolean *, Node *, uintptr_t, 4> BooleanToEdgeMap;
+
+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; }
+
+       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);
+       void createAllTotalOrderConstraintsSATEncoderSparse(Order *order);
+       void createAllPartialOrderConstraintsSATEncoderSparse(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);
+       int getMaximumUsedSize(ElementEncoding *encoding);
+       void generateAnyValueBinaryIndexEncodingPositive(ElementEncoding *encoding);
+       void generateAnyValueBinaryValueEncoding(ElementEncoding *encoding);
        CNF *cnf;
        CSolver *solver;
+       BooleanToEdgeMap booledgeMap;
+       VectorEdge *vector;
+       friend class VarOrderingOpt;
 };
 
-#include "satelemencoder.h"
-#include "satorderencoder.h"
-#include "satfunctableencoder.h"
-
-SATEncoder *allocSATEncoder(CSolver *solver);
-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 encodeVarSATEncoder(SATEncoder *This, BooleanVar *constraint);
-Edge encodeLogicSATEncoder(SATEncoder *This, BooleanLogic *constraint);
-Edge encodePredicateSATEncoder(SATEncoder *This, BooleanPredicate *constraint);
-Edge encodeTablePredicateSATEncoder(SATEncoder *This, BooleanPredicate *constraint);
-
-
-
-void encodeElementSATEncoder(SATEncoder *encoder, Element *This);
-void encodeElementFunctionSATEncoder(SATEncoder *encoder, ElementFunction *This);
-void encodeTableElementFunctionSATEncoder(SATEncoder *encoder, ElementFunction *This);
-
+void allocElementConstraintVariables(ElementEncoding *ee, uint numVars);
 #endif