X-Git-Url: http://plrg.eecs.uci.edu/git/?a=blobdiff_plain;f=tree.h;h=1752d3fe57cdd9535a32a2b18c1e9ddaedf28e30;hb=6b3cc5b800966b58d330bf516d92a17b4e2bee9d;hp=9fba203473122b66a2a2fcc4d0cfc2cd54ca6649;hpb=9f1c792a01ecbc86624f4c16d0a2bc763485a590;p=c11tester.git

diff --git a/tree.h b/tree.h
index 9fba2034..1752d3fe 100644
--- a/tree.h
+++ b/tree.h
@@ -5,6 +5,8 @@
 #include <map>
 #include "threads.h"
 
+class ModelAction;
+
 /*
  * An n-ary tree
  *
@@ -13,10 +15,10 @@
  */
 class TreeNode {
 public:
-	TreeNode(TreeNode *par);
+	TreeNode(TreeNode *par = NULL, ModelAction *act = NULL);
 	~TreeNode();
 	bool hasBeenExplored(thread_id_t id) { return children.find(id_to_int(id)) != children.end(); }
-	TreeNode * exploreChild(thread_id_t id);
+	TreeNode * explore_child(ModelAction *act);
 	thread_id_t getNextBacktrack();
 
 	/* Return 1 if already in backtrack, 0 otherwise */