src/csolver.h

   1 #ifndef CSOLVER_H
   2 #define CSOLVER_H
   3 #include "classes.h"
   4 #include "ops.h"
   5 #include "corestructs.h"
   6 #include "asthash.h"
   7 #include "solver_interface.h"
   8
   9 class CSolver {
  10 public:
  11         CSolver();
  12         ~CSolver();
  13
  14         /** This function creates a set containing the elements passed in the array. */
  15         Set *createSet(VarType type, uint64_t *elements, uint num);
  16
  17         /** This function creates a set from lowrange to highrange (inclusive). */
  18
  19         Set *createRangeSet(VarType type, uint64_t lowrange, uint64_t highrange);
  20
  21         Element *createRangeVar(VarType type, uint64_t lowrange, uint64_t highrange);
  22
  23         /** This function creates a mutable set. */
  24
  25         MutableSet *createMutableSet(VarType type);
  26
  27         /** This function adds a new item to a set. */
  28
  29         void addItem(MutableSet *set, uint64_t element);
  30
  31         /** This function adds a new unique item to the set and returns it.
  32             This function cannot be used in conjunction with manually adding
  33             items to the set. */
  34
  35         uint64_t createUniqueItem(MutableSet *set);
  36
  37         /** This function creates an element variable over a set. */
  38
  39         Element *getElementVar(Set *set);
  40
  41         /** This function creates an element constrant. */
  42         Element *getElementConst(VarType type, uint64_t value);
  43
  44         Boolean *getBooleanTrue();
  45
  46         Boolean *getBooleanFalse();
  47
  48         /** This function creates a boolean variable. */
  49
  50         Boolean *getBooleanVar(VarType type);
  51
  52         /** This function creates a function operator. */
  53
  54         Function *createFunctionOperator(ArithOp op, Set **domain, uint numDomain, Set *range,
  55                                                                                                                                          OverFlowBehavior overflowbehavior);
  56
  57         /** This function creates a predicate operator. */
  58
  59         Predicate *createPredicateOperator(CompOp op, Set **domain, uint numDomain);
  60
  61         Predicate *createPredicateTable(Table *table, UndefinedBehavior behavior);
  62
  63         /** This function creates an empty instance table.*/
  64
  65         Table *createTable(Set **domains, uint numDomain, Set *range);
  66
  67         Table *createTableForPredicate(Set **domains, uint numDomain);
  68         /** This function adds an input output relation to a table. */
  69
  70         void addTableEntry(Table *table, uint64_t *inputs, uint inputSize, uint64_t result);
  71
  72         /** This function converts a completed table into a function. */
  73
  74         Function *completeTable(Table *, UndefinedBehavior behavior);
  75
  76         /** This function applies a function to the Elements in its input. */
  77
  78         Element *applyFunction(Function *function, Element **array, uint numArrays, Boolean *overflowstatus);
  79
  80         /** This function applies a predicate to the Elements in its input. */
  81
  82         Boolean *applyPredicateTable(Predicate *predicate, Element **inputs, uint numInputs, Boolean *undefinedStatus);
  83
  84         Boolean *applyPredicate(Predicate *predicate, Element **inputs, uint numInputs);
  85
  86         /** This function applies a logical operation to the Booleans in its input. */
  87
  88         Boolean *applyLogicalOperation(LogicOp op, Boolean **array, uint asize);
  89
  90                 /** This function applies a logical operation to the Booleans in its input. */
  91
  92         Boolean *applyLogicalOperation(LogicOp op, Boolean * arg1, Boolean * arg2);
  93
  94         /** This function applies a logical operation to the Booleans in its input. */
  95
  96         Boolean *applyLogicalOperation(LogicOp op, Boolean *arg);
  97
  98         /** This function adds a boolean constraint to the set of constraints
  99             to be satisfied */
 100
 101         void addConstraint(Boolean *constraint);
 102
 103         /** This function instantiates an order of type type over the set set. */
 104         Order *createOrder(OrderType type, Set *set);
 105
 106         /** This function instantiates a boolean on two items in an order. */
 107         Boolean *orderConstraint(Order *order, uint64_t first, uint64_t second);
 108
 109         /** When everything is done, the client calls this function and then csolver starts to encode*/
 110         int solve();
 111
 112         /** After getting the solution from the SAT solver, client can get the value of an element via this function*/
 113         uint64_t getElementValue(Element *element);
 114
 115         /** After getting the solution from the SAT solver, client can get the value of a boolean via this function*/
 116         bool getBooleanValue(Boolean *boolean);
 117
 118         HappenedBefore getOrderConstraintValue(Order *order, uint64_t first, uint64_t second);
 119
 120         bool isTrue(Boolean *b);
 121         bool isFalse(Boolean *b);
 122
 123         void setUnSAT() { unsat = true; }
 124
 125         bool isUnSAT() { return unsat; }
 126
 127         Vector<Order *> *getOrders() { return &allOrders;}
 128
 129         Tuner *getTuner() { return tuner; }
 130         Transformer* getTransformer() {return transformer;}
 131
 132         SetIteratorBoolean *getConstraints() { return constraints.iterator(); }
 133
 134         SATEncoder *getSATEncoder() {return satEncoder;}
 135
 136         void replaceBooleanWithTrue(Boolean *bexpr);
 137         void replaceBooleanWithFalse(Boolean *bexpr);
 138         void replaceBooleanWithBoolean(Boolean *oldb, Boolean *newb);
 139         CSolver *clone();
 140         void autoTune(uint budget);
 141
 142         void setTransformer(Transformer * _transformer) {  transformer = _transformer; }
 143         void setTuner(Tuner * _tuner) { tuner = _tuner; }
 144         long long getElapsedTime() { return elapsedTime; }
 145         long long getEncodeTime();
 146         long long getSolveTime();
 147
 148         CMEMALLOC;
 149
 150 private:
 151         void handleXORFalse(BooleanLogic *bexpr, Boolean *child);
 152         void handleIMPLIESTrue(BooleanLogic *bexpr, Boolean *child);
 153         void handleIMPLIESFalse(BooleanLogic *bexpr, Boolean *child);
 154         void handleANDTrue(BooleanLogic *bexpr, Boolean *child);
 155         void handleORFalse(BooleanLogic *bexpr, Boolean *child);
 156
 157         /** This is a vector of constraints that must be satisfied. */
 158         HashsetBoolean constraints;
 159
 160         /** This is a vector of all boolean structs that we have allocated. */
 161         Vector<Boolean *> allBooleans;
 162
 163         /** This is a vector of all set structs that we have allocated. */
 164         Vector<Set *> allSets;
 165
 166         /** This is a vector of all element structs that we have allocated. */
 167         Vector<Element *> allElements;
 168
 169         /** This is a vector of all predicate structs that we have allocated. */
 170         Vector<Predicate *> allPredicates;
 171
 172         /** This is a vector of all table structs that we have allocated. */
 173         Vector<Table *> allTables;
 174
 175         /** This is a vector of all order structs that we have allocated. */
 176         Vector<Order *> allOrders;
 177
 178         /** This is a vector of all function structs that we have allocated. */
 179         Vector<Function *> allFunctions;
 180
 181         Boolean * boolTrue;
 182         Boolean * boolFalse;
 183
 184         /** These two tables are used for deduplicating entries. */
 185         BooleanMatchMap boolMap;
 186         ElementMatchMap elemMap;
 187
 188         SATEncoder *satEncoder;
 189         bool unsat;
 190         Tuner *tuner;
 191         Transformer* transformer;
 192         long long elapsedTime;
 193 };
 194 #endif