Defining the interfaces to add various regions to snapshot

[model-checker.git] / model.h

diff --git a/model.h b/model.h
index 3e644f2792df519b67c7452f959cacd1fd791593..d3b42053fcee011ed32e47df7fef9822aadd144d 100644 (file)
--- a/model.h
+++ b/model.h
@@ -3,10 +3,13 @@
 
 #include <list>
 #include <map>
+#include <vector>
 #include <cstddef>
 #include <ucontext.h>
 
 #include "schedule.h"
+#include "mymemory.h"
+#include <utility>
 #include "libthreads.h"
 #include "libatomic.h"
 #include "threads.h"
@@ -14,26 +17,7 @@
 
 /* Forward declaration */
 class TreeNode;
-
-class Backtrack {
-public:
-       Backtrack(ModelAction *d, action_list_t *t) {
-               diverge = d;
-               actionTrace = t;
-               iter = actionTrace->begin();
-       }
-       ModelAction * get_diverge() { return diverge; }
-       action_list_t * get_trace() { return actionTrace; }
-       void advance_state() { iter++; }
-       ModelAction * get_state() {
-               return iter == actionTrace->end() ? NULL : *iter;
-       }
-private:
-       ModelAction *diverge;
-       action_list_t *actionTrace;
-       /* points to position in actionTrace as we replay */
-       action_list_t::iterator iter;
-};
+class Backtrack;
 
 class ModelChecker {
 public:
@@ -51,15 +35,18 @@ public:
 
        int add_thread(Thread *t);
        void remove_thread(Thread *t);
-       Thread * get_thread(thread_id_t tid) { return thread_map[tid]; }
+       Thread * get_thread(thread_id_t tid) { return thread_map[id_to_int(tid)]; }
 
        thread_id_t get_next_id();
+       int get_next_seq_num();
 
        int switch_to_master(ModelAction *act);
 
        bool next_execution();
+  MEMALLOC
 private:
-       int used_thread_id;
+       int next_thread_id;
+       int used_sequence_numbers;
        int num_executions;
 
        ModelAction * get_last_conflict(ModelAction *act);
@@ -69,15 +56,17 @@ private:
        Backtrack * get_next_backtrack();
        void reset_to_initial_state();
 
+       void print_list(action_list_t *list);
+
        class ModelAction *current_action;
        Backtrack *exploring;
        thread_id_t nextThread;
 
        ucontext_t *system_context;
        action_list_t *action_trace;
-       std::map<thread_id_t, class Thread *> thread_map;
+       std::map<int, class Thread *, std::less< int >, MyAlloc< std::pair< const int, class Thread * > > > thread_map;
        class TreeNode *rootNode, *currentNode;
-       std::list<class Backtrack *> backtrack_list;
+       std::list<class Backtrack *, MyAlloc< class Backtrack * > > backtrack_list;
 };
 
 extern ModelChecker *model;