Merge branch 'hamed' into brian
[satune.git] / src / csolver.c
index 9ddfe3d3e9f9e02b08840d98aba8bc5d71976844..1d31e42558a70bcf1ca3af8be314d0d3ec342762 100644 (file)
@@ -5,6 +5,9 @@
 #include "boolean.h"
 #include "predicate.h"
 #include "order.h"
+#include "table.h"
+#include "function.h"
+#include "satencoder.h"
 
 CSolver * allocCSolver() {
        CSolver * tmp=(CSolver *) ourmalloc(sizeof(CSolver));
@@ -12,52 +15,77 @@ CSolver * allocCSolver() {
        tmp->allBooleans=allocDefVectorBoolean();
        tmp->allSets=allocDefVectorSet();
        tmp->allElements=allocDefVectorElement();
+       tmp->allPredicates = allocDefVectorPredicate();
+       tmp->allTables = allocDefVectorTable();
+       tmp->allOrders = allocDefVectorOrder();
+       tmp->allFunctions = allocDefVectorFunction();
        return tmp;
 }
 
 /** This function tears down the solver and the entire AST */
 
-void deleteSolver(CSolver *this) {
-       deleteVectorBoolean(this->constraints);
+void deleteSolver(CSolver *This) {
+       deleteVectorBoolean(This->constraints);
 
-       uint size=getSizeVectorBoolean(this->allBooleans);
+       uint size=getSizeVectorBoolean(This->allBooleans);
        for(uint i=0;i<size;i++) {
-               deleteBoolean(getVectorBoolean(this->allBooleans, i));
+               deleteBoolean(getVectorBoolean(This->allBooleans, i));
        }
+       deleteVectorBoolean(This->allBooleans);
 
-       deleteVectorBoolean(this->allBooleans);
+       size=getSizeVectorSet(This->allSets);
+       for(uint i=0;i<size;i++) {
+               deleteSet(getVectorSet(This->allSets, i));
+       }
+       deleteVectorSet(This->allSets);
 
-       size=getSizeVectorSet(this->allSets);
+       size=getSizeVectorElement(This->allElements);
        for(uint i=0;i<size;i++) {
-               deleteSet(getVectorSet(this->allSets, i));
+               deleteElement(getVectorElement(This->allElements, i));
        }
+       deleteVectorElement(This->allElements);
 
-       deleteVectorSet(this->allSets);
+       size=getSizeVectorTable(This->allTables);
+       for(uint i=0;i<size;i++) {
+               deleteTable(getVectorTable(This->allTables, i));
+       }
+       deleteVectorTable(This->allTables);
+
+       size=getSizeVectorPredicate(This->allPredicates);
+       for(uint i=0;i<size;i++) {
+               deletePredicate(getVectorPredicate(This->allPredicates, i));
+       }
+       deleteVectorPredicate(This->allPredicates);
 
-       size=getSizeVectorElement(this->allElements);
+       size=getSizeVectorOrder(This->allOrders);
        for(uint i=0;i<size;i++) {
-               deleteElement(getVectorElement(this->allElements, i));
+               deleteOrder(getVectorOrder(This->allOrders, i));
        }
+       deleteVectorOrder(This->allOrders);
 
-       deleteVectorElement(this->allElements);
-       ourfree(this);
+       size=getSizeVectorFunction(This->allFunctions);
+       for(uint i=0;i<size;i++) {
+               deleteFunction(getVectorFunction(This->allFunctions, i));
+       }
+       deleteVectorFunction(This->allFunctions);
+       ourfree(This);
 }
 
-Set * createSet(CSolver * this, VarType type, uint64_t * elements, uint numelements) {
+Set * createSet(CSolver * This, VarType type, uint64_t * elements, uint numelements) {
        Set * set=allocSet(type, elements, numelements);
-       pushVectorSet(this->allSets, set);
+       pushVectorSet(This->allSets, set);
        return set;
 }
 
-Set * createRangeSet(CSolver * this, VarType type, uint64_t lowrange, uint64_t highrange) {
+Set * createRangeSet(CSolver * This, VarType type, uint64_t lowrange, uint64_t highrange) {
        Set * set=allocSetRange(type, lowrange, highrange);
-       pushVectorSet(this->allSets, set);
+       pushVectorSet(This->allSets, set);
        return set;
 }
 
-MutableSet * createMutableSet(CSolver * this, VarType type) {
+MutableSet * createMutableSet(CSolver * This, VarType type) {
        MutableSet * set=allocMutableSet(type);
-       pushVectorSet(this->allSets, set);
+       pushVectorSet(This->allSets, set);
        return set;
 }
 
@@ -71,9 +99,9 @@ uint64_t createUniqueItem(CSolver *solver, MutableSet * set) {
        return element;
 }
 
-Element * getElementVar(CSolver *this, Set * set) {
-       Element * element=allocElement(set);
-       pushVectorElement(this->allElements, element);
+Element * getElementVar(CSolver *This, Set * set) {
+       Element * element=allocElementSet(set);
+       pushVectorElement(This->allElements, element);
        return element;
 }
 
@@ -83,44 +111,58 @@ Boolean * getBooleanVar(CSolver *solver, VarType type) {
        return boolean;
 }
 
-Function * createFunctionOperator(CSolver *solver, ArithOp op, Set ** domain, uint numDomain, Set * range,
-                                                                                                                                       OverFlowBehavior overflowbehavior) {
-       return NULL;
+Function * createFunctionOperator(CSolver *solver, ArithOp op, Set ** domain, uint numDomain, Set * range,OverFlowBehavior overflowbehavior) {
+       Function* function = allocFunctionOperator(op, domain, numDomain, range, overflowbehavior);
+       pushVectorFunction(solver->allFunctions, function);
+       return function;
 }
 
 Predicate * createPredicateOperator(CSolver *solver, CompOp op, Set ** domain, uint numDomain) {
-       return allocPredicate(op, domain,numDomain);
+       Predicate* predicate= allocPredicateOperator(op, domain,numDomain);
+       pushVectorPredicate(solver->allPredicates, predicate);
+       return predicate;
 }
 
 Table * createTable(CSolver *solver, Set **domains, uint numDomain, Set * range) {
-       return NULL;
+       Table* table= allocTable(domains,numDomain,range);
+       pushVectorTable(solver->allTables, table);
+       return table;
 }
 
-void addTableEntry(CSolver *solver, uint64_t* inputs, uint inputSize, uint64_t result) {
+void addTableEntry(CSolver *solver, Table* table, uint64_t* inputs, uint inputSize, uint64_t result) {
+       addNewTableEntry(table,inputs, inputSize,result);
 }
 
 Function * completeTable(CSolver *solver, Table * table) {
-       return NULL;
+       Function* function = allocFunctionTable(table);
+       pushVectorFunction(solver->allFunctions,function);
+       return function;
 }
 
-Element * applyFunction(CSolver *solver, Function * function, Element ** array, Boolean * overflowstatus) {
-       return NULL;
+Element * applyFunction(CSolver *solver, Function * function, Element ** array, uint numArrays, Boolean * overflowstatus) {
+       Element* element= allocElementFunction(function,array,numArrays,overflowstatus);
+       pushVectorElement(solver->allElements, element);
+       return element;
 }
 
-Boolean * applyPredicate(CSolver *solver, Predicate * predicate, Element ** inputs) {
-       return NULL;
+Boolean * applyPredicate(CSolver *solver, Predicate * predicate, Element ** inputs, uint numInputs) {
+       Boolean* boolean= allocBooleanPredicate(predicate, inputs, numInputs);
+       pushVectorBoolean(solver->allBooleans, boolean);
+       return boolean;
 }
 
-Boolean * applyLogicalOperation(CSolver *solver, LogicOp op, Boolean ** array) {
-       return NULL;
+Boolean * applyLogicalOperation(CSolver *solver, LogicOp op, Boolean ** array, uint asize) {
+       return allocBooleanLogicArray(solver, op, array, asize);
 }
 
-void addBoolean(CSolver *this, Boolean * constraint) {
-       pushVectorBoolean(this->constraints, constraint);
+void addBoolean(CSolver *This, Boolean * constraint) {
+       pushVectorBoolean(This->constraints, constraint);
 }
 
 Order * createOrder(CSolver *solver, OrderType type, Set * set) {
-       return allocOrder(type, set);
+       Order* order = allocOrder(type, set);
+       pushVectorOrder(solver->allOrders, order);
+       return order;
 }
 
 Boolean * orderConstraint(CSolver *solver, Order * order, uint64_t first, uint64_t second) {
@@ -128,3 +170,15 @@ Boolean * orderConstraint(CSolver *solver, Order * order, uint64_t first, uint64
        pushVectorBoolean(solver->allBooleans,constraint);
        return constraint;
 }
+
+void startEncoding(CSolver* solver){
+       naiveEncodingDecision(solver);
+       SATEncoder* satEncoder = allocSATEncoder();
+       encodeAllSATEncoder(solver, satEncoder);
+       finishedClauses(satEncoder->satSolver);
+       int result= solve(satEncoder->satSolver);
+       model_print("sat_solver's result:%d\n", result);
+       //For now, let's just delete it, and in future for doing queries 
+       //we may need it.
+       deleteSATEncoder(satEncoder);
+}