Adding STL stuff and operator news of snapshot to model-checker. Need to actuallly...

[c11tester.git] / tree.h

diff --git a/tree.h b/tree.h
index 1752d3fe57cdd9535a32a2b18c1e9ddaedf28e30..e107a7ef76f1d2805d7f1005e1313b1fc1184750 100644 (file)
--- a/tree.h
+++ b/tree.h
@@ -1,8 +1,9 @@
 #ifndef __TREE_H__
 #define __TREE_H__
-
+#include "mymemory.h"
 #include <set>
 #include <map>
+#include <cstddef>
 #include "threads.h"
 
 class ModelAction;
@@ -25,11 +26,15 @@ public:
        int setBacktrack(thread_id_t id);
        TreeNode * getRoot();
        static int getTotalNodes() { return TreeNode::totalNodes; }
+
+       bool is_enabled(Thread *t);
+  MEMALLOC
 private:
        TreeNode *parent;
-       std::map<int, class TreeNode *> children;
-       std::set<int> backtrack;
+       std::map<int, class TreeNode *, std::less< int >, MyAlloc< std::pair< const int, class TreeNode * > > > children;
+       std::set<int, std::less< int >, MyAlloc< int > > backtrack;
        static int totalNodes;
+       int num_threads;
 };
 
 #endif /* __TREE_H__ */