#ifndef CSOLVER_H
#define CSOLVER_H
-#include "classlist.h"
+#include "classes.h"
#include "ops.h"
-#include "structs.h"
+#include "corestructs.h"
+#include "asthash.h"
-struct CSolver {
- VectorBoolean * constraints;
- VectorBoolean * allBooleans;
- VectorSet * allSets;
- VectorElement * allElements;
-};
+class CSolver {
+public:
+ CSolver();
+ ~CSolver();
+
+ /** This function creates a set containing the elements passed in the array. */
+ Set *createSet(VarType type, uint64_t *elements, uint num);
+
+ /** This function creates a set from lowrange to highrange (inclusive). */
+
+ Set *createRangeSet(VarType type, uint64_t lowrange, uint64_t highrange);
+
+ /** This function creates a mutable set. */
+
+ MutableSet *createMutableSet(VarType type);
+
+ /** This function adds a new item to a set. */
+
+ void addItem(MutableSet *set, uint64_t element);
+
+ /** This function adds a new unique item to the set and returns it.
+ This function cannot be used in conjunction with manually adding
+ items to the set. */
+
+ uint64_t createUniqueItem(MutableSet *set);
+
+ /** This function creates an element variable over a set. */
+
+ Element *getElementVar(Set *set);
-/** Create a new solver instance. */
+ /** This function creates an element constrant. */
+ Element *getElementConst(VarType type, uint64_t value);
-CSolver * allocCSolver();
+ Boolean *getBooleanTrue();
-/** This function creates a set containing the elements passed in the array. */
+ Boolean *getBooleanFalse();
+
+ /** This function creates a boolean variable. */
-Set * createSet(CSolver *, VarType type, uint64_t * elements, uint num);
+ Boolean *getBooleanVar(VarType type);
-/** This function creates a set from lowrange to highrange (inclusive). */
+ /** This function creates a function operator. */
-Set * createRangeSet(CSolver *, VarType type, uint64_t lowrange, uint64_t highrange);
+ Function *createFunctionOperator(ArithOp op, Set **domain, uint numDomain, Set *range,
+ OverFlowBehavior overflowbehavior);
-/** This function creates a mutable set. */
+ /** This function creates a predicate operator. */
-MutableSet * createMutableSet(CSolver *, VarType type);
+ Predicate *createPredicateOperator(CompOp op, Set **domain, uint numDomain);
-/** This function adds a new item to a set. */
+ Predicate *createPredicateTable(Table *table, UndefinedBehavior behavior);
-void addItem(CSolver *, MutableSet * set, uint64_t element);
+ /** This function creates an empty instance table.*/
-/** This function adds a new unique item to the set and returns it.
- This function cannot be used in conjunction with manually adding
- items to the set. */
+ Table *createTable(Set **domains, uint numDomain, Set *range);
-uint64_t createUniqueItem(CSolver *, MutableSet * set);
+ Table *createTableForPredicate(Set **domains, uint numDomain);
+ /** This function adds an input output relation to a table. */
-/** This function creates an element variable over a set. */
+ void addTableEntry(Table *table, uint64_t *inputs, uint inputSize, uint64_t result);
-Element * getElementVar(CSolver *, Set * set);
+ /** This function converts a completed table into a function. */
-/** This function creates a boolean variable. */
+ Function *completeTable(Table *, UndefinedBehavior behavior);
-Boolean * getBooleanVar(CSolver *, VarType type);
+ /** This function applies a function to the Elements in its input. */
-/** This function creates a function operator. */
+ Element *applyFunction(Function *function, Element **array, uint numArrays, Boolean *overflowstatus);
-Function * createFunctionOperator(CSolver *solver, enum ArithOp op, Set ** domain, uint numDomain, Set * range,
- enum OverFlowBehavior overflowbehavior, Boolean * overflowstatus);
+ /** This function applies a predicate to the Elements in its input. */
-/** This function creates a predicate operator. */
+ Boolean *applyPredicateTable(Predicate *predicate, Element **inputs, uint numInputs, Boolean *undefinedStatus);
-Predicate * createPredicateOperator(CSolver *solver, enum CompOp op, Set ** domain, uint numDomain);
+ Boolean *applyPredicate(Predicate *predicate, Element **inputs, uint numInputs);
-/** This function creates an empty instance table.*/
+ /** This function applies a logical operation to the Booleans in its input. */
-Table * createTable(CSolver *solver, Set **domains, uint numDomain, Set * range);
+ Boolean *applyLogicalOperation(LogicOp op, Boolean **array, uint asize);
-/** This function adds an input output relation to a table. */
+ /** This function adds a boolean constraint to the set of constraints
+ to be satisfied */
-void addTableEntry(CSolver *solver, uint64_t* inputs, uint inputSize, uint64_t result);
+ void addConstraint(Boolean *constraint);
-/** This function converts a completed table into a function. */
+ /** This function instantiates an order of type type over the set set. */
+ Order *createOrder(OrderType type, Set *set);
-Function * completeTable(CSolver *, Table *);
+ /** This function instantiates a boolean on two items in an order. */
+ Boolean *orderConstraint(Order *order, uint64_t first, uint64_t second);
-/** This function applies a function to the Elements in its input. */
+ /** When everything is done, the client calls this function and then csolver starts to encode*/
+ int startEncoding();
-Element * applyFunction(CSolver *, Function * function, Element ** array);
+ /** After getting the solution from the SAT solver, client can get the value of an element via this function*/
+ uint64_t getElementValue(Element *element);
-/** This function applies a predicate to the Elements in its input. */
+ /** After getting the solution from the SAT solver, client can get the value of a boolean via this function*/
+ bool getBooleanValue(Boolean *boolean);
-Boolean * applyPredicate(CSolver *, Predicate * predicate, Element ** inputs);
+ HappenedBefore getOrderConstraintValue(Order *order, uint64_t first, uint64_t second);
-/** This function applies a logical operation to the Booleans in its input. */
+ void setUnSAT() { unsat = true; }
-Boolean * applyLogicalOperation(CSolver *, enum LogicOp op, Boolean ** array);
+ bool isUnSAT() { return unsat; }
-/** This function adds a boolean constraint to the set of constraints
- to be satisfied */
+ Vector<Order *> *getOrders() { return &allOrders;}
-void addBoolean(CSolver *, Boolean * constraint);
+ Tuner *getTuner() { return tuner; }
+
+ SetIteratorBoolean *getConstraints() { return constraints.iterator(); }
-/** This function instantiates an order of type type over the set set. */
-Order * createOrder(CSolver *, enum OrderType type, Set * set);
+ SATEncoder *getSATEncoder() {return satEncoder;}
-/** This function instantiates a predicate on two items in an order. */
-Boolean * orderConstraint(CSolver *, Order * order, uint64_t first, uint64_t second);
+ void replaceBooleanWithTrue(Boolean *bexpr);
+ void replaceBooleanWithFalse(Boolean *bexpr);
+ void replaceBooleanWithBoolean(Boolean *oldb, Boolean *newb);
+ CSolver *clone();
+ void autoTune(uint budget);
+
+ void setTuner(Tuner * _tuner) { tuner = _tuner; }
+ long long getElapsedTime() { return elapsedTime; }
+ long long getEncodeTime();
+ long long getSolveTime();
+
+ CMEMALLOC;
+
+private:
+ void handleXORFalse(BooleanLogic *bexpr, Boolean *child);
+ void handleIMPLIESTrue(BooleanLogic *bexpr, Boolean *child);
+ void handleIMPLIESFalse(BooleanLogic *bexpr, Boolean *child);
+ void handleANDTrue(BooleanLogic *bexpr, Boolean *child);
+ void handleORFalse(BooleanLogic *bexpr, Boolean *child);
+
+ /** This is a vector of constraints that must be satisfied. */
+ HashSetBoolean constraints;
+
+ /** This is a vector of all boolean structs that we have allocated. */
+ Vector<Boolean *> allBooleans;
+
+ /** This is a vector of all set structs that we have allocated. */
+ Vector<Set *> allSets;
+
+ /** This is a vector of all element structs that we have allocated. */
+ Vector<Element *> allElements;
+
+ /** This is a vector of all predicate structs that we have allocated. */
+ Vector<Predicate *> allPredicates;
+
+ /** This is a vector of all table structs that we have allocated. */
+ Vector<Table *> allTables;
+
+ /** This is a vector of all order structs that we have allocated. */
+ Vector<Order *> allOrders;
+
+ /** This is a vector of all function structs that we have allocated. */
+ Vector<Function *> allFunctions;
+
+ Boolean * boolTrue;
+ Boolean * boolFalse;
+
+ /** These two tables are used for deduplicating entries. */
+ BooleanMatchMap boolMap;
+ ElementMatchMap elemMap;
+
+ SATEncoder *satEncoder;
+ bool unsat;
+ Tuner *tuner;
+
+ long long elapsedTime;
+};
#endif