X-Git-Url: http://plrg.eecs.uci.edu/git/?p=satune.git;a=blobdiff_plain;f=src%2Fcsolver.h;h=bac1a4961488defbbb4dcf1d45bfb140783614a0;hp=e8527f0eed35b418a41b03c453469bfd8f5c93e4;hb=f9295083796e890e723309da92b351e9c5660bed;hpb=d98defe8f9c222258e0286adf9d2f627f7d197ed diff --git a/src/csolver.h b/src/csolver.h index e8527f0..bac1a49 100644 --- a/src/csolver.h +++ b/src/csolver.h @@ -1,64 +1,48 @@ #ifndef CSOLVER_H #define CSOLVER_H -#include "classlist.h" +#include "classes.h" #include "ops.h" -#include "structs.h" +#include "corestructs.h" +#include "asthash.h" +#include "solver_interface.h" +#include "common.h" class CSolver { - public: +public: CSolver(); ~CSolver(); - - SATEncoder *satEncoder; - bool unsat; - Tuner *tuner; - - /** This is a vector of constraints that must be satisfied. */ - HashSetBoolean *constraints; - - /** This is a vector of all boolean structs that we have allocated. */ - VectorBoolean *allBooleans; - - /** This is a vector of all set structs that we have allocated. */ - VectorSet *allSets; - - /** This is a vector of all element structs that we have allocated. */ - VectorElement *allElements; - - /** This is a vector of all predicate structs that we have allocated. */ - VectorPredicate *allPredicates; - - /** This is a vector of all table structs that we have allocated. */ - VectorTable *allTables; - - /** This is a vector of all order structs that we have allocated. */ - VectorOrder *allOrders; - - /** This is a vector of all function structs that we have allocated. */ - VectorFunction *allFunctions; /** 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); - + + VarType getSetVarType(Set *set); + + Element *createRangeVar(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. */ + //Deprecating this unless we need it... 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. */ - + This function cannot be used in conjunction with manually adding + items to the set. */ + uint64_t createUniqueItem(MutableSet *set); + /** + * Freeze and finalize the mutableSet ... + */ + void finalizeMutableSet(MutableSet *set); + /** This function creates an element variable over a set. */ Element *getElementVar(Set *set); @@ -66,9 +50,15 @@ class CSolver { /** This function creates an element constrant. */ Element *getElementConst(VarType type, uint64_t value); + Set *getElementRange (Element *element); + + BooleanEdge getBooleanTrue(); + + BooleanEdge getBooleanFalse(); + /** This function creates a boolean variable. */ - Boolean *getBooleanVar(VarType type); + BooleanEdge getBooleanVar(VarType type); /** This function creates a function operator. */ @@ -96,41 +86,130 @@ class CSolver { /** This function applies a function to the Elements in its input. */ - Element *applyFunction(Function *function, Element **array, uint numArrays, Boolean *overflowstatus); + Element *applyFunction(Function *function, Element **array, uint numArrays, BooleanEdge overflowstatus); /** This function applies a predicate to the Elements in its input. */ - Boolean *applyPredicateTable(Predicate *predicate, Element **inputs, uint numInputs, Boolean *undefinedStatus); + BooleanEdge applyPredicateTable(Predicate *predicate, Element **inputs, uint numInputs, BooleanEdge undefinedStatus); + + BooleanEdge applyPredicate(Predicate *predicate, Element **inputs, uint numInputs); + + /** This function applies a logical operation to the Booleans in its input. */ + + BooleanEdge applyLogicalOperation(LogicOp op, BooleanEdge *array, uint asize); - Boolean *applyPredicate(Predicate *predicate, Element **inputs, uint numInputs); + /** This function applies a logical operation to the Booleans in its input. */ + + BooleanEdge applyLogicalOperation(LogicOp op, BooleanEdge arg1, BooleanEdge arg2); /** This function applies a logical operation to the Booleans in its input. */ - Boolean *applyLogicalOperation(LogicOp op, Boolean **array, uint asize); + BooleanEdge applyLogicalOperation(LogicOp op, BooleanEdge arg); /** This function adds a boolean constraint to the set of constraints - to be satisfied */ + to be satisfied */ - void addConstraint(Boolean *constraint); + void addConstraint(BooleanEdge constraint); /** This function instantiates an order of type type over the set set. */ Order *createOrder(OrderType type, Set *set); /** This function instantiates a boolean on two items in an order. */ - Boolean *orderConstraint(Order *order, uint64_t first, uint64_t second); + BooleanEdge orderConstraint(Order *order, uint64_t first, uint64_t second); /** When everything is done, the client calls this function and then csolver starts to encode*/ - int startEncoding(); + int solve(); /** After getting the solution from the SAT solver, client can get the value of an element via this function*/ uint64_t getElementValue(Element *element); /** After getting the solution from the SAT solver, client can get the value of a boolean via this function*/ - bool getBooleanValue(Boolean *boolean); + bool getBooleanValue(BooleanEdge boolean); - HappenedBefore getOrderConstraintValue(Order *order, uint64_t first, uint64_t second); + bool getOrderConstraintValue(Order *order, uint64_t first, uint64_t second); - MEMALLOC; -}; + bool isTrue(BooleanEdge b); + bool isFalse(BooleanEdge b); + + void setUnSAT() { model_print("Setting UNSAT %%%%%%\n"); unsat = true; } + bool isUnSAT() { return unsat; } + + void printConstraint(BooleanEdge boolean); + void printConstraints(); + + Vector *getOrders() { return &allOrders;} + HashsetOrder *getActiveOrders() { return &activeOrders;} + + Tuner *getTuner() { return tuner; } + + SetIteratorBooleanEdge *getConstraints() { return constraints.iterator(); } + + SATEncoder *getSATEncoder() {return satEncoder;} + + void replaceBooleanWithTrue(BooleanEdge bexpr); + void replaceBooleanWithTrueNoRemove(BooleanEdge bexpr); + void replaceBooleanWithFalseNoRemove(BooleanEdge bexpr); + void replaceBooleanWithFalse(BooleanEdge bexpr); + void replaceBooleanWithBoolean(BooleanEdge oldb, BooleanEdge newb); + CSolver *clone(); + void serialize(); + static CSolver* deserialize(const char * file); + void autoTune(uint budget); + void inferFixedOrders(); + void inferFixedOrder(Order *order); + + void setTuner(Tuner *_tuner) { tuner = _tuner; } + long long getElapsedTime() { return elapsedTime; } + long long getEncodeTime(); + long long getSolveTime(); + + CMEMALLOC; + +private: + void handleIFFTrue(BooleanLogic *bexpr, BooleanEdge child); + void handleANDTrue(BooleanLogic *bexpr, BooleanEdge child); + + //These two functions are helpers if the client has a pointer to a + //Boolean object that we have since replaced + BooleanEdge rewriteLogicalOperation(LogicOp op, BooleanEdge *array, uint asize); + BooleanEdge doRewrite(BooleanEdge b); + /** This is a vector of constraints that must be satisfied. */ + HashsetBooleanEdge constraints; + + /** This is a vector of all boolean structs that we have allocated. */ + Vector allBooleans; + + /** This is a vector of all set structs that we have allocated. */ + Vector allSets; + + /** This is a vector of all element structs that we have allocated. */ + Vector allElements; + + /** This is a vector of all predicate structs that we have allocated. */ + Vector allPredicates; + + /** This is a vector of all table structs that we have allocated. */ + Vector allTables; + + /** This is a vector of all order structs that we have allocated. */ + Vector allOrders; + + HashsetOrder activeOrders; + + /** This is a vector of all function structs that we have allocated. */ + Vector allFunctions; + + BooleanEdge boolTrue; + BooleanEdge boolFalse; + + /** These two tables are used for deduplicating entries. */ + BooleanMatchMap boolMap; + ElementMatchMap elemMap; + + SATEncoder *satEncoder; + bool unsat; + Tuner *tuner; + long long elapsedTime; +}; #endif