cyclegraph: add back edges to CycleNode

[model-checker.git] / cyclegraph.h

diff --git a/cyclegraph.h b/cyclegraph.h
index 6b3d93e5e45ec661d3acd4db987796b9b1adaea6..f2f30329203c226c5a7a1c6140fa3043bb260613 100644 (file)
--- a/cyclegraph.h
+++ b/cyclegraph.h
@@ -70,6 +70,8 @@ class CycleNode {
        bool addEdge(CycleNode *node);
        CycleNode * getEdge(unsigned int i) const;
        unsigned int getNumEdges() const;
+       CycleNode * getBackEdge(unsigned int i) const;
+       unsigned int getNumBackEdges() const;
        bool setRMW(CycleNode *);
        CycleNode * getRMW() const;
        const ModelAction * getAction() const { return action; }
@@ -89,6 +91,9 @@ class CycleNode {
        /** @brief The edges leading out from this node */
        std::vector< CycleNode *, SnapshotAlloc<CycleNode *> > edges;
 
+       /** @brief The edges leading into this node */
+       std::vector< CycleNode *, SnapshotAlloc<CycleNode *> > back_edges;
+
        /** Pointer to a RMW node that reads from this node, or NULL, if none
         * exists */
        CycleNode *hasRMW;