5 #include "corestructs.h"
7 #include "solver_interface.h"
15 /** This function creates a set containing the elements passed in the array. */
16 Set *createSet(VarType type, uint64_t *elements, uint num);
18 /** This function creates a set from lowrange to highrange (inclusive). */
20 Set *createRangeSet(VarType type, uint64_t lowrange, uint64_t highrange);
22 bool itemExistInSet(Set *set, uint64_t item);
24 VarType getSetVarType(Set *set);
26 Element *createRangeVar(VarType type, uint64_t lowrange, uint64_t highrange);
28 /** This function creates a mutable set.
29 * Note: You should use addItem for adding new item to Mutable sets, and
30 * at the end, you should call finalizeMutableSet!
33 MutableSet *createMutableSet(VarType type);
35 /** This function adds a new item to a set. */
37 //Deprecating this unless we need it...
38 void addItem(MutableSet *set, uint64_t element);
40 /** This function adds a new unique item to the set and returns it.
41 This function cannot be used in conjunction with manually adding
44 uint64_t createUniqueItem(MutableSet *set);
47 * Freeze and finalize the mutableSet ...
49 void finalizeMutableSet(MutableSet *set);
51 /** This function creates an element variable over a set. */
53 Element *getElementVar(Set *set);
55 /** This function creates an element constrant. */
56 Element *getElementConst(VarType type, uint64_t value);
58 Set *getElementRange (Element *element);
60 void mustHaveValue(Element *element);
62 BooleanEdge getBooleanTrue();
64 BooleanEdge getBooleanFalse();
66 /** This function creates a boolean variable. */
68 BooleanEdge getBooleanVar(VarType type);
70 /** This function creates a function operator. */
72 Function *createFunctionOperator(ArithOp op, Set *range,
73 OverFlowBehavior overflowbehavior);
75 /** This function creates a predicate operator. */
77 Predicate *createPredicateOperator(CompOp op);
79 Predicate *createPredicateTable(Table *table, UndefinedBehavior behavior);
81 /** This function creates an empty instance table.*/
83 Table *createTable(Set *range);
85 Table *createTableForPredicate();
86 /** This function adds an input output relation to a table. */
88 void addTableEntry(Table *table, uint64_t *inputs, uint inputSize, uint64_t result);
90 /** This function converts a completed table into a function. */
92 Function *completeTable(Table *, UndefinedBehavior behavior);
94 /** This function applies a function to the Elements in its input. */
96 Element *applyFunction(Function *function, Element **array, uint numArrays, BooleanEdge overflowstatus);
98 /** This function applies a predicate to the Elements in its input. */
100 BooleanEdge applyPredicateTable(Predicate *predicate, Element **inputs, uint numInputs, BooleanEdge undefinedStatus);
102 BooleanEdge applyPredicate(Predicate *predicate, Element **inputs, uint numInputs);
104 /** This exactly one element of array can be true! (i.e. a1 + a2 + a3 + ... + an = 1)*/
105 BooleanEdge applyExactlyOneConstraint (BooleanEdge *array, uint asize);
107 /** This exactly one element of array can be true! (i.e. a1 => !a2 & !a3 & ... & !an)*/
108 BooleanEdge applyAtMostOneConstraint (BooleanEdge *array, uint asize);
110 /** This function applies a logical operation to the Booleans in its input. */
112 BooleanEdge applyLogicalOperation(LogicOp op, BooleanEdge *array, uint asize);
114 /** This function applies a logical operation to the Booleans in its input. */
116 BooleanEdge applyLogicalOperation(LogicOp op, BooleanEdge arg1, BooleanEdge arg2);
118 /** This function applies a logical operation to the Booleans in its input. */
120 BooleanEdge applyLogicalOperation(LogicOp op, BooleanEdge arg);
122 /** This function adds a boolean constraint to the set of constraints
125 void addConstraint(BooleanEdge constraint);
127 /** This function instantiates an order of type type over the set set. */
128 Order *createOrder(OrderType type, Set *set);
130 /** This function instantiates a boolean on two items in an order. */
131 BooleanEdge orderConstraint(Order *order, uint64_t first, uint64_t second);
133 /** When everything is done, the client calls this function and then csolver starts to encode*/
136 * Incremental Solving for SATUNE.
137 * It only supports incremental solving for elements!
138 * No support for BooleanVar, BooleanOrder or using interpreters
141 int solveIncremental();
143 /** After getting the solution from the SAT solver, client can get the value of an element via this function*/
144 uint64_t getElementValue(Element *element);
146 void freezeElement(Element *e);
147 void turnoffOptimizations(){optimizationsOff = true;}
148 /** After getting the solution from the SAT solver, client can get the value of a boolean via this function*/
149 bool getBooleanValue(BooleanEdge boolean);
151 bool getOrderConstraintValue(Order *order, uint64_t first, uint64_t second);
153 bool isTrue(BooleanEdge b);
154 bool isFalse(BooleanEdge b);
156 void setUnSAT() { model_print("Setting UNSAT %%%%%%\n"); unsat = true; }
157 void setSatSolverTimeout(long seconds) { satsolverTimeout = seconds;}
158 bool isUnSAT() { return unsat; }
159 bool isBooleanVarUsed() {return booleanVarUsed;}
160 void printConstraint(BooleanEdge boolean);
161 void printConstraints();
163 Vector<Order *> *getOrders() { return &allOrders;}
164 HashsetOrder *getActiveOrders() { return &activeOrders;}
166 Tuner *getTuner() { return tuner; }
168 SetIteratorBooleanEdge *getConstraints() { return constraints.iterator(); }
169 bool isConstraintEncoded(BooleanEdge be) { return encodedConstraints.contains(be);}
170 void addEncodedConstraint(BooleanEdge be) {encodedConstraints.add(be);}
172 SATEncoder *getSATEncoder() {return satEncoder;}
174 void replaceBooleanWithTrue(BooleanEdge bexpr);
175 void replaceBooleanWithTrueNoRemove(BooleanEdge bexpr);
176 void replaceBooleanWithFalseNoRemove(BooleanEdge bexpr);
177 void replaceBooleanWithFalse(BooleanEdge bexpr);
178 void replaceBooleanWithBoolean(BooleanEdge oldb, BooleanEdge newb);
181 static CSolver *deserialize(const char *file, InterpreterType itype = SATUNE);
182 void autoTune(uint budget);
183 void inferFixedOrders();
184 void inferFixedOrder(Order *order);
185 void setInterpreter(InterpreterType type);
186 bool useInterpreter() {return interpreter != NULL;}
187 void setTuner(Tuner *_tuner) { tuner = _tuner; }
188 long long getElapsedTime() { return elapsedTime; }
189 long long getEncodeTime();
190 long long getSolveTime();
191 long getSatSolverTimeout() { return satsolverTimeout;}
192 bool isIncrementalMode() {return incrementalMode;}
193 void freezeElementsVariables();
197 void handleIFFTrue(BooleanLogic *bexpr, BooleanEdge child);
198 void handleANDTrue(BooleanLogic *bexpr, BooleanEdge child);
199 void handleFunction(ElementFunction *ef, BooleanEdge child);
201 //These two functions are helpers if the client has a pointer to a
202 //Boolean object that we have since replaced
203 BooleanEdge rewriteLogicalOperation(LogicOp op, BooleanEdge *array, uint asize);
204 BooleanEdge doRewrite(BooleanEdge b);
205 /** This is a vector of constraints that must be satisfied. */
206 HashsetBooleanEdge constraints;
207 HashsetBooleanEdge encodedConstraints;
209 /** This is a vector of all boolean structs that we have allocated. */
210 Vector<Boolean *> allBooleans;
212 /** This is a vector of all set structs that we have allocated. */
213 Vector<Set *> allSets;
215 /** This is a vector of all element structs that we have allocated. */
216 Vector<Element *> allElements;
218 /** This is a vector of all predicate structs that we have allocated. */
219 Vector<Predicate *> allPredicates;
221 /** This is a vector of all table structs that we have allocated. */
222 Vector<Table *> allTables;
224 /** This is a vector of all order structs that we have allocated. */
225 Vector<Order *> allOrders;
227 HashsetOrder activeOrders;
229 /** This is a vector of all function structs that we have allocated. */
230 Vector<Function *> allFunctions;
232 BooleanEdge boolTrue;
233 BooleanEdge boolFalse;
235 /** These two tables are used for deduplicating entries. */
236 BooleanMatchMap boolMap;
237 ElementMatchMap elemMap;
239 SATEncoder *satEncoder;
242 bool incrementalMode;
243 bool optimizationsOff;
245 long long elapsedTime;
246 long satsolverTimeout;
247 Interpreter *interpreter;
249 friend class ElementOpt;
250 friend class VarOrderingOpt;
253 inline CompOp flipOp(CompOp op) {
269 inline CompOp negateOp(CompOp op) {