X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=model.cc;h=b3b517caed9c087b0ed1daaeb9a271c4a3a7bc20;hb=55f3acdb4bec3b2879e189a82d23e112aba79aed;hp=e394865e26c01d75b05853a55a4b2b7a23c3e853;hpb=18f3e39e72b1b31594acb10842e61f90109876e0;p=model-checker.git

diff --git a/model.cc b/model.cc
index e394865..b3b517c 100644
--- a/model.cc
+++ b/model.cc
@@ -1,5 +1,6 @@
 #include <stdio.h>
 #include <algorithm>
+#include <mutex>
 
 #include "model.h"
 #include "action.h"
@@ -11,7 +12,6 @@
 #include "cyclegraph.h"
 #include "promise.h"
 #include "datarace.h"
-#include "mutex.h"
 #include "threads-model.h"
 
 #define INITIAL_THREAD_ID	0
@@ -116,6 +116,10 @@ modelclock_t ModelChecker::get_next_seq_num()
 	return ++priv->used_sequence_numbers;
 }
 
+Node * ModelChecker::get_curr_node() {
+	return node_stack->get_head();
+}
+
 /**
  * @brief Choose the next thread to execute.
  *