tree: revise arguments (use Thread, ModelAction)

[model-checker.git] / tree.cc

diff --git a/tree.cc b/tree.cc
index ce79f755eeea010897bc117f47deab11bb53786e..57e0c13f69902f071cb336d3a155a15db38f4c08 100644 (file)
--- a/tree.cc
+++ b/tree.cc
@@ -1,8 +1,9 @@
 #include "tree.h"
+#include "action.h"
 
 int TreeNode::totalNodes = 0;
 
-TreeNode::TreeNode(TreeNode *par)
+TreeNode::TreeNode(TreeNode *par, ModelAction *act)
 {
        TreeNode::totalNodes++;
        this->parent = par;
@@ -15,14 +16,15 @@ TreeNode::~TreeNode() {
                delete it->second;
 }
 
-TreeNode * TreeNode::exploreChild(thread_id_t id)
+TreeNode * TreeNode::explore_child(ModelAction *act)
 {
        TreeNode *n;
        std::set<int>::iterator it;
+       thread_id_t id = act->get_tid();
        int i = id_to_int(id);
 
        if (!hasBeenExplored(id)) {
-               n = new TreeNode(this);
+               n = new TreeNode(this, act);
                children[i] = n;
        } else {
                n = children[i];