removing true nodes from the OrderGraph
[satune.git] / src / Encoders / orderencoder.c
index c8a9b72b0b90be691314663005eb4630d181a67d..32bd91092f16699200675b02911399527a2e6524 100644 (file)
@@ -7,15 +7,7 @@
 #include "ordernode.h"
 #include "rewriter.h"
 #include "mutableset.h"
-
-OrderGraph *buildOrderGraph(Order *order) {
-       OrderGraph *orderGraph = allocOrderGraph(order);
-       uint constrSize = getSizeVectorBooleanOrder(&order->constraints);
-       for (uint j = 0; j < constrSize; j++) {
-               addOrderConstraintToOrderGraph(orderGraph, getVectorBooleanOrder(&order->constraints, j));
-       }
-       return orderGraph;
-}
+#include "tunable.h"
 
 void DFS(OrderGraph *graph, VectorOrderNode *finishNodes) {
        HSIteratorOrderNode *iterator = iteratorOrderNode(graph->nodes);
@@ -23,7 +15,7 @@ void DFS(OrderGraph *graph, VectorOrderNode *finishNodes) {
                OrderNode *node = nextOrderNode(iterator);
                if (node->status == NOTVISITED) {
                        node->status = VISITED;
-                       DFSNodeVisit(node, finishNodes, false, 0);
+                       DFSNodeVisit(node, finishNodes, false, false, 0);
                        node->status = FINISHED;
                        pushVectorOrderNode(finishNodes, node);
                }
@@ -38,7 +30,7 @@ void DFSReverse(OrderGraph *graph, VectorOrderNode *finishNodes) {
                OrderNode *node = getVectorOrderNode(finishNodes, i);
                if (node->status == NOTVISITED) {
                        node->status = VISITED;
-                       DFSNodeVisit(node, NULL, true, sccNum);
+                       DFSNodeVisit(node, NULL, true, false, sccNum);
                        node->sccNum = sccNum;
                        node->status = FINISHED;
                        sccNum++;
@@ -46,22 +38,26 @@ void DFSReverse(OrderGraph *graph, VectorOrderNode *finishNodes) {
        }
 }
 
-void DFSNodeVisit(OrderNode *node, VectorOrderNode *finishNodes, bool isReverse, uint sccNum) {
+void DFSNodeVisit(OrderNode *node, VectorOrderNode *finishNodes, bool isReverse, bool mustvisit, uint sccNum) {
        HSIteratorOrderEdge *iterator = isReverse ? iteratorOrderEdge(node->inEdges) : iteratorOrderEdge(node->outEdges);
        while (hasNextOrderEdge(iterator)) {
                OrderEdge *edge = nextOrderEdge(iterator);
-               if (!edge->polPos && !edge->pseudoPos)//Ignore edges that do not have positive polarity
-                       continue;
+               if (mustvisit) {
+                       if (!edge->mustPos)
+                               continue;
+               } else
+                       if (!edge->polPos && !edge->pseudoPos)//Ignore edges that do not have positive polarity
+                               continue;
 
                OrderNode *child = isReverse ? edge->source : edge->sink;
 
                if (child->status == NOTVISITED) {
                        child->status = VISITED;
-                       DFSNodeVisit(child, finishNodes, isReverse, sccNum);
+                       DFSNodeVisit(child, finishNodes, isReverse, mustvisit, sccNum);
                        child->status = FINISHED;
-                       if (!isReverse)
+                       if (finishNodes != NULL)
                                pushVectorOrderNode(finishNodes, child);
-                       else
+                       if (isReverse)
                                child->sccNum = sccNum;
                }
        }
@@ -86,57 +82,137 @@ void computeStronglyConnectedComponentGraph(OrderGraph *graph) {
        deleteVectorArrayOrderNode(&finishNodes);
 }
 
-void removeMustBeTrueNodes(OrderGraph *graph) {
-       //TODO: Nodes that all the incoming/outgoing edges are MUST_BE_TRUE
+bool isMustBeTrueNode(OrderNode* node){
+       HSIteratorOrderEdge* iterator = iteratorOrderEdge(node->inEdges);
+       while(hasNextOrderEdge(iterator)){
+               OrderEdge* edge = nextOrderEdge(iterator);
+               if(!edge->mustPos)
+                       return false;
+       }
+       deleteIterOrderEdge(iterator);
+       iterator = iteratorOrderEdge(node->outEdges);
+       while(hasNextOrderEdge(iterator)){
+               OrderEdge* edge = nextOrderEdge(iterator);
+               if(!edge->mustPos)
+                       return false;
+       }
+       deleteIterOrderEdge(iterator);
+       return true;
 }
 
-void DFSPseudoNodeVisit(OrderGraph *graph, OrderNode *node) {
-       HSIteratorOrderEdge *iterator = iteratorOrderEdge(node->inEdges);
-       while (hasNextOrderEdge(iterator)) {
-               OrderEdge *inEdge = nextOrderEdge(iterator);
-               if (inEdge->polNeg) {
-                       OrderNode *src = inEdge->source;
-                       if (src->status == VISITED) {
-                               //Make a pseudoEdge to point backwards
-                               OrderEdge *newedge = getOrderEdgeFromOrderGraph(graph, inEdge->sink, inEdge->source);
-                               newedge->pseudoPos = true;
-                       }
+void bypassMustBeTrueNode(CSolver *This, OrderGraph* graph, OrderNode* node){
+       HSIteratorOrderEdge* iterin = iteratorOrderEdge(node->inEdges);
+       while(hasNextOrderEdge(iterin)){
+               OrderEdge* inEdge = nextOrderEdge(iterin);
+               OrderNode* srcNode = inEdge->source;
+               removeHashSetOrderEdge(srcNode->outEdges, inEdge);
+               HSIteratorOrderEdge* iterout = iteratorOrderEdge(node->outEdges);
+               while(hasNextOrderEdge(iterout)){
+                       OrderEdge* outEdge = nextOrderEdge(iterout);
+                       OrderNode* sinkNode = outEdge->sink;
+                       removeHashSetOrderEdge(sinkNode->inEdges, outEdge);
+                       //Adding new edge to new sink and src nodes ...
+                       OrderEdge *newEdge =getOrderEdgeFromOrderGraph(graph, srcNode, sinkNode);
+                       newEdge->mustPos = true;
+                       newEdge->polPos = true;
+                       if (newEdge->mustNeg)
+                               This->unsat = true;
+                       addHashSetOrderEdge(srcNode->outEdges, newEdge);
+                       addHashSetOrderEdge(sinkNode->inEdges, newEdge);
                }
+               deleteIterOrderEdge(iterout);
        }
-       deleteIterOrderEdge(iterator);
-       iterator = iteratorOrderEdge(node->outEdges);
-       while (hasNextOrderEdge(iterator)) {
-               OrderEdge *edge = nextOrderEdge(iterator);
-               if (!edge->polPos)//Ignore edges that do not have positive polarity
-                       continue;
+       deleteIterOrderEdge(iterin);
+}
 
-               OrderNode *child = edge->sink;
-               if (child->status == NOTVISITED) {
-                       child->status = VISITED;
-                       DFSPseudoNodeVisit(graph, child);
-                       child->status = FINISHED;
+void removeMustBeTrueNodes(CSolver *This, OrderGraph *graph) {
+       HSIteratorOrderNode* iterator = iteratorOrderNode(graph->nodes);
+       while(hasNextOrderNode(iterator)){
+               OrderNode* node = nextOrderNode(iterator);
+               if(isMustBeTrueNode(node)){
+                       bypassMustBeTrueNode(This,graph, node);
                }
        }
-       deleteIterOrderEdge(iterator);
+       deleteIterOrderNode(iterator);
 }
 
+/** This function computes a source set for every nodes, the set of
+               nodes that can reach that node via pospolarity edges.  It then
+               looks for negative polarity edges from nodes in the the source set
+               to determine whether we need to generate pseudoPos edges. */
+
 void completePartialOrderGraph(OrderGraph *graph) {
        VectorOrderNode finishNodes;
        initDefVectorOrderNode(&finishNodes);
        DFS(graph, &finishNodes);
        resetNodeInfoStatusSCC(graph);
+       HashTableNodeToNodeSet *table = allocHashTableNodeToNodeSet(128, 0.25);
 
+       VectorOrderNode sccNodes;
+       initDefVectorOrderNode(&sccNodes);
+       
        uint size = getSizeVectorOrderNode(&finishNodes);
+       uint sccNum = 1;
        for (int i = size - 1; i >= 0; i--) {
                OrderNode *node = getVectorOrderNode(&finishNodes, i);
+               HashSetOrderNode *sources = allocHashSetOrderNode(4, 0.25);
+               putNodeToNodeSet(table, node, sources);
+               
                if (node->status == NOTVISITED) {
+                       //Need to do reverse traversal here...
                        node->status = VISITED;
-                       DFSPseudoNodeVisit(graph, node);
+                       DFSNodeVisit(node, &sccNodes, true, false, sccNum);
                        node->status = FINISHED;
+                       node->sccNum = sccNum;
+                       sccNum++;
+                       pushVectorOrderNode(&sccNodes, node);
+
+                       //Compute in set for entire SCC
+                       uint rSize = getSizeVectorOrderNode(&sccNodes);
+                       for (int j = 0; j < rSize; j++) {
+                               OrderNode *rnode = getVectorOrderNode(&sccNodes, j);
+                               //Compute source sets
+                               HSIteratorOrderEdge *iterator = iteratorOrderEdge(rnode->inEdges);
+                               while (hasNextOrderEdge(iterator)) {
+                                       OrderEdge *edge = nextOrderEdge(iterator);
+                                       OrderNode *parent = edge->source;
+                                       if (edge->polPos) {
+                                               addHashSetOrderNode(sources, parent);
+                                               HashSetOrderNode *parent_srcs = (HashSetOrderNode *)getNodeToNodeSet(table, parent);
+                                               addAllHashSetOrderNode(sources, parent_srcs);
+                                       }
+                               }
+                               deleteIterOrderEdge(iterator);
+                       }
+                       for (int j=0; j < rSize; j++) {
+                               //Copy in set of entire SCC
+                               OrderNode *rnode = getVectorOrderNode(&sccNodes, j);
+                               HashSetOrderNode * set = (j==0) ? sources : copyHashSetOrderNode(sources);
+                               putNodeToNodeSet(table, rnode, set);
+
+                               //Use source sets to compute pseudoPos edges
+                               HSIteratorOrderEdge *iterator = iteratorOrderEdge(rnode->inEdges);
+                               while (hasNextOrderEdge(iterator)) {
+                                       OrderEdge *edge = nextOrderEdge(iterator);
+                                       OrderNode *parent = edge->source;
+                                       ASSERT(parent != rnode);
+                                       if (edge->polNeg && parent->sccNum != rnode->sccNum &&
+                                                       containsHashSetOrderNode(sources, parent)) {
+                                               OrderEdge *newedge = getOrderEdgeFromOrderGraph(graph, rnode, parent);
+                                               newedge->pseudoPos = true;
+                                       }
+                               }
+                               deleteIterOrderEdge(iterator);
+                       }
+                       
+                       clearVectorOrderNode(&sccNodes);
                }
        }
 
+       resetAndDeleteHashTableNodeToNodeSet(table);
+       deleteHashTableNodeToNodeSet(table);
        resetNodeInfoStatusSCC(graph);
+       deleteVectorArrayOrderNode(&sccNodes);
        deleteVectorArrayOrderNode(&finishNodes);
 }
 
@@ -146,7 +222,7 @@ void DFSMust(OrderGraph *graph, VectorOrderNode *finishNodes) {
                OrderNode *node = nextOrderNode(iterator);
                if (node->status == NOTVISITED) {
                        node->status = VISITED;
-                       DFSMustNodeVisit(node, finishNodes);
+                       DFSNodeVisit(node, finishNodes, false, true, 0);
                        node->status = FINISHED;
                        pushVectorOrderNode(finishNodes, node);
                }
@@ -154,27 +230,7 @@ void DFSMust(OrderGraph *graph, VectorOrderNode *finishNodes) {
        deleteIterOrderNode(iterator);
 }
 
-void DFSMustNodeVisit(OrderNode *node, VectorOrderNode *finishNodes) {
-       HSIteratorOrderEdge *iterator = iteratorOrderEdge(node->outEdges);
-       while (hasNextOrderEdge(iterator)) {
-               OrderEdge *edge = nextOrderEdge(iterator);
-               OrderNode *child = edge->sink;
-
-               if (!edge->mustPos)     //Ignore edges that are not must Positive edges
-                       continue;
-
-               if (child->status == NOTVISITED) {
-                       child->status = VISITED;
-                       DFSMustNodeVisit(child, finishNodes);
-                       child->status = FINISHED;
-                       pushVectorOrderNode(finishNodes, child);
-               }
-       }
-       deleteIterOrderEdge(iterator);
-}
-
-
-void DFSClearContradictions(OrderGraph *graph, VectorOrderNode *finishNodes, bool computeTransitiveClosure) {
+void DFSClearContradictions(CSolver *solver, OrderGraph *graph, VectorOrderNode *finishNodes, bool computeTransitiveClosure) {
        uint size = getSizeVectorOrderNode(finishNodes);
        HashTableNodeToNodeSet *table = allocHashTableNodeToNodeSet(128, 0.25);
 
@@ -205,6 +261,8 @@ void DFSClearContradictions(OrderGraph *graph, VectorOrderNode *finishNodes, boo
                                OrderEdge *newedge = getOrderEdgeFromOrderGraph(graph, srcnode, node);
                                newedge->mustPos = true;
                                newedge->polPos = true;
+                               if (newedge->mustNeg)
+                                       solver->unsat = true;
                                addHashSetOrderEdge(srcnode->outEdges,newedge);
                                addHashSetOrderEdge(node->inEdges,newedge);
                        }
@@ -218,6 +276,9 @@ void DFSClearContradictions(OrderGraph *graph, VectorOrderNode *finishNodes, boo
                                OrderNode *parent = edge->source;
                                if (!edge->mustPos && containsHashSetOrderNode(sources, parent)) {
                                        edge->mustPos = true;
+                                       edge->polPos = true;
+                                       if (edge->mustNeg)
+                                               solver->unsat = true;
                                }
                        }
                        deleteIterOrderEdge(iterator);
@@ -230,6 +291,9 @@ void DFSClearContradictions(OrderGraph *graph, VectorOrderNode *finishNodes, boo
                                OrderNode *child = edge->sink;
                                if (!edge->mustNeg && containsHashSetOrderNode(sources, child)) {
                                        edge->mustNeg = true;
+                                       edge->polNeg = true;
+                                       if (edge->mustPos)
+                                               solver->unsat = true;
                                }
                        }
                        deleteIterOrderEdge(iterator);
@@ -237,6 +301,7 @@ void DFSClearContradictions(OrderGraph *graph, VectorOrderNode *finishNodes, boo
        }
 
        resetAndDeleteHashTableNodeToNodeSet(table);
+       deleteHashTableNodeToNodeSet(table);
 }
 
 /* This function finds edges that would form a cycle with must edges
@@ -244,7 +309,7 @@ void DFSClearContradictions(OrderGraph *graph, VectorOrderNode *finishNodes, boo
    must be true because of transitivity from other must be true
    edges. */
 
-void reachMustAnalysis(OrderGraph *graph, bool computeTransitiveClosure) {
+void reachMustAnalysis(CSolver * solver, OrderGraph *graph, bool computeTransitiveClosure) {
        VectorOrderNode finishNodes;
        initDefVectorOrderNode(&finishNodes);
        //Topologically sort the mustPos edge graph
@@ -252,25 +317,29 @@ void reachMustAnalysis(OrderGraph *graph, bool computeTransitiveClosure) {
        resetNodeInfoStatusSCC(graph);
 
        //Find any backwards edges that complete cycles and force them to be mustNeg
-       DFSClearContradictions(graph, &finishNodes, computeTransitiveClosure);
+       DFSClearContradictions(solver, graph, &finishNodes, computeTransitiveClosure);
        deleteVectorArrayOrderNode(&finishNodes);
-       resetNodeInfoStatusSCC(graph);
 }
 
 /* This function finds edges that must be positive and forces the
    inverse edge to be negative (and clears its positive polarity if it
    had one). */
 
-void localMustAnalysisTotal(OrderGraph *graph) {
+void localMustAnalysisTotal(CSolver *solver, OrderGraph *graph) {
        HSIteratorOrderEdge *iterator = iteratorOrderEdge(graph->edges);
        while (hasNextOrderEdge(iterator)) {
                OrderEdge *edge = nextOrderEdge(iterator);
                if (edge->mustPos) {
                        OrderEdge *invEdge = getInverseOrderEdge(graph, edge);
-                       if (invEdge != NULL && !invEdge->mustPos && invEdge->polPos) {
-                               invEdge->polPos = false;
+                       if (invEdge != NULL) {
+                               if (!invEdge->mustPos) {
+                                       invEdge->polPos = false;
+                               } else {
+                                       solver->unsat = true;
+                               }
+                               invEdge->mustNeg = true;
+                               invEdge->polNeg = true;
                        }
-                       invEdge->mustNeg = true;
                }
        }
        deleteIterOrderEdge(iterator);
@@ -281,19 +350,24 @@ void localMustAnalysisTotal(OrderGraph *graph) {
     It also finds edges that must be negative and clears the positive
     polarity. */
 
-void localMustAnalysisPartial(OrderGraph *graph) {
+void localMustAnalysisPartial(CSolver *solver, OrderGraph *graph) {
        HSIteratorOrderEdge *iterator = iteratorOrderEdge(graph->edges);
        while (hasNextOrderEdge(iterator)) {
                OrderEdge *edge = nextOrderEdge(iterator);
                if (edge->mustPos) {
-                       if (edge->polNeg && !edge->mustNeg) {
+                       if (!edge->mustNeg) {
                                edge->polNeg = false;
-                       }
+                       } else
+                               solver->unsat = true;
+
                        OrderEdge *invEdge = getInverseOrderEdge(graph, edge);
                        if (invEdge != NULL) {
                                if (!invEdge->mustPos)
                                        invEdge->polPos = false;
+                               else
+                                       solver->unsat = true;
                                invEdge->mustNeg = true;
+                               invEdge->polNeg = true;
                        }
                }
                if (edge->mustNeg && !edge->mustPos) {
@@ -305,19 +379,25 @@ void localMustAnalysisPartial(OrderGraph *graph) {
 
 void decomposeOrder(CSolver *This, Order *order, OrderGraph *graph) {
        VectorOrder ordervec;
+       VectorOrder partialcandidatevec;
        initDefVectorOrder(&ordervec);
+       initDefVectorOrder(&partialcandidatevec);
        uint size = getSizeVectorBooleanOrder(&order->constraints);
        for (uint i = 0; i < size; i++) {
                BooleanOrder *orderconstraint = getVectorBooleanOrder(&order->constraints, i);
                OrderNode *from = getOrderNodeFromOrderGraph(graph, orderconstraint->first);
                OrderNode *to = getOrderNodeFromOrderGraph(graph, orderconstraint->second);
-               OrderEdge *edge = getOrderEdgeFromOrderGraph(graph, from, to);
-               if (from->sccNum < to->sccNum) {
-                       //replace with true
-                       replaceBooleanWithTrue(This, (Boolean *)orderconstraint);
-               } else if (to->sccNum < from->sccNum) {
-                       //replace with false
-                       replaceBooleanWithFalse(This, (Boolean *)orderconstraint);
+               model_print("from->sccNum:%u\tto->sccNum:%u\n", from->sccNum, to->sccNum);
+               if (from->sccNum != to->sccNum) {
+                       OrderEdge *edge = getOrderEdgeFromOrderGraph(graph, from, to);                  
+                       if (edge->polPos) {
+                               replaceBooleanWithTrue(This, (Boolean *)orderconstraint);
+                       } else if (edge->polNeg) {
+                               replaceBooleanWithFalse(This, (Boolean *)orderconstraint);
+                       } else {
+                               //This case should only be possible if constraint isn't in AST
+                               ASSERT(0);
+                       }
                } else {
                        //Build new order and change constraint's order
                        Order *neworder = NULL;
@@ -328,6 +408,10 @@ void decomposeOrder(CSolver *This, Order *order, OrderGraph *graph) {
                                neworder = allocOrder(order->type, set);
                                pushVectorOrder(This->allOrders, neworder);
                                setExpandVectorOrder(&ordervec, from->sccNum, neworder);
+                               if (order->type == PARTIAL)
+                                       setExpandVectorOrder(&partialcandidatevec, from->sccNum, neworder);
+                               else
+                                       setExpandVectorOrder(&partialcandidatevec, from->sccNum, NULL);
                        }
                        if (from->status != ADDEDTOSET) {
                                from->status = ADDEDTOSET;
@@ -337,17 +421,37 @@ void decomposeOrder(CSolver *This, Order *order, OrderGraph *graph) {
                                to->status = ADDEDTOSET;
                                addElementMSet((MutableSet *)neworder->set, to->id);
                        }
+                       if (order->type == PARTIAL) {
+                               OrderEdge *edge = getOrderEdgeFromOrderGraph(graph, from, to);
+                               if (edge->polNeg)
+                                       setExpandVectorOrder(&partialcandidatevec, from->sccNum, NULL);
+                       }
                        orderconstraint->order = neworder;
                        addOrderConstraint(neworder, orderconstraint);
                }
        }
+
+       uint pcvsize=getSizeVectorOrder(&partialcandidatevec);
+       for(uint i=0;i<pcvsize;i++) {
+               Order * neworder=getVectorOrder(&partialcandidatevec, i);
+               if (neworder != NULL){
+                       neworder->type = TOTAL;
+                       model_print("i=%u\t", i);
+               }
+       }
+       
        deleteVectorArrayOrder(&ordervec);
+       deleteVectorArrayOrder(&partialcandidatevec);
 }
 
 void orderAnalysis(CSolver *This) {
        uint size = getSizeVectorOrder(This->allOrders);
        for (uint i = 0; i < size; i++) {
                Order *order = getVectorOrder(This->allOrders, i);
+               bool doDecompose=GETVARTUNABLE(This->tuner, order->type, DECOMPOSEORDER, &onoff);
+               if (!doDecompose)
+                       continue;
+               
                OrderGraph *graph = buildOrderGraph(order);
                if (order->type == PARTIAL) {
                        //Required to do SCC analysis for partial order graphs.  It
@@ -356,24 +460,33 @@ void orderAnalysis(CSolver *This) {
                        completePartialOrderGraph(graph);
                }
 
-               //This analysis is completely optional
-               reachMustAnalysis(graph, false);
 
-               //This pair of analysis is also optional
-               if (order->type == PARTIAL) {
-                       localMustAnalysisPartial(graph);
-               } else {
-                       localMustAnalysisTotal(graph);
-               }
+               bool mustReachGlobal=GETVARTUNABLE(This->tuner, order->type, MUSTREACHGLOBAL, &onoff);
+
+               if (mustReachGlobal)
+                       reachMustAnalysis(This, graph, false);
 
-               //This optimization is completely optional
-               removeMustBeTrueNodes(graph);
+               bool mustReachLocal=GETVARTUNABLE(This->tuner, order->type, MUSTREACHLOCAL, &onoff);
+               
+               if (mustReachLocal) {
+                       //This pair of analysis is also optional
+                       if (order->type == PARTIAL) {
+                               localMustAnalysisPartial(This, graph);
+                       } else {
+                               localMustAnalysisTotal(This, graph);
+                       }
+               }
 
+               bool mustReachPrune=GETVARTUNABLE(This->tuner, order->type, MUSTREACHPRUNE, &onoff);
+               
+               if (mustReachPrune)
+                       removeMustBeTrueNodes(This, graph);
+               
                //This is needed for splitorder
                computeStronglyConnectedComponentGraph(graph);
-
+               
                decomposeOrder(This, order, graph);
-
+               
                deleteOrderGraph(graph);
        }
 }