nodestack: add Node::get_parent() function

[model-checker.git] / nodestack.cc

diff --git a/nodestack.cc b/nodestack.cc
index c9878a339ec384febedaa95385e31cac478cc169..e7b5f51162b485160f3b4bb20cb06f13df2dc3b6 100644 (file)
--- a/nodestack.cc
+++ b/nodestack.cc
@@ -4,11 +4,14 @@
 #include "model.h"
 
 /** @brief Node constructor */
-Node::Node(ModelAction *act, int nthreads)
+Node::Node(ModelAction *act, Node *par, int nthreads)
        : action(act),
+       parent(par),
        num_threads(nthreads),
        explored_children(num_threads),
-       backtrack(num_threads)
+       backtrack(num_threads),
+       numBacktracks(0),
+       may_read_from()
 {
 }
 
@@ -47,11 +50,7 @@ bool Node::has_been_explored(thread_id_t tid)
  */
 bool Node::backtrack_empty()
 {
-       unsigned int i;
-       for (i = 0; i < backtrack.size(); i++)
-               if (backtrack[i] == true)
-                       return false;
-       return true;
+       return numBacktracks == 0;
 }
 
 /**
@@ -79,6 +78,7 @@ bool Node::set_backtrack(thread_id_t id)
        if (backtrack[i])
                return false;
        backtrack[i] = true;
+       numBacktracks++;
        return true;
 }
 
@@ -92,6 +92,7 @@ thread_id_t Node::get_next_backtrack()
        if (i >= backtrack.size())
                return THREAD_ID_T_NONE;
        backtrack[i] = false;
+       numBacktracks--;
        return int_to_id(i);
 }
 
@@ -100,10 +101,22 @@ bool Node::is_enabled(Thread *t)
        return id_to_int(t->get_id()) < num_threads;
 }
 
+/**
+ * Add an action to the may_read_from set.
+ * @param act is the action to add
+ */
+void Node::add_read_from(ModelAction *act)
+{
+       may_read_from.insert(act);
+}
+
 void Node::explore(thread_id_t tid)
 {
        int i = id_to_int(tid);
-       backtrack[i] = false;
+       if (backtrack[i]) {
+               backtrack[i] = false;
+               numBacktracks--;
+       }
        explored_children[i] = true;
 }
 
@@ -161,7 +174,7 @@ ModelAction * NodeStack::explore_action(ModelAction *act)
 
        /* Record action */
        get_head()->explore_child(act);
-       node_list.push_back(new Node(act, model->get_num_threads()));
+       node_list.push_back(new Node(act, get_head(), model->get_num_threads()));
        total_nodes++;
        iter++;
        return NULL;