X-Git-Url: http://plrg.eecs.uci.edu/git/?p=model-checker.git;a=blobdiff_plain;f=model.h;h=1dc6a158d3aa63247d1051977fe734f1b9cf042f;hp=b6a0004d5a3d119e413d01e5184741930ac5e505;hb=810306cb85accaaace9a50f174264f105991230b;hpb=88e927cbc95771540c3ff7b905a84dbd8f37323c

diff --git a/model.h b/model.h
index b6a0004..1dc6a15 100644
--- a/model.h
+++ b/model.h
@@ -3,74 +3,71 @@
 
 #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"
-#include "tree.h"
+#include "action.h"
 
-#define VALUE_NONE -1
-
-typedef enum action_type {
-	THREAD_CREATE,
-	THREAD_YIELD,
-	THREAD_JOIN,
-	ATOMIC_READ,
-	ATOMIC_WRITE
-} action_type_t;
-
-class ModelAction {
-public:
-	ModelAction(action_type_t type, memory_order order, void *loc, int value);
-	void print(void);
-
-	thread_id_t get_tid() { return tid; }
-	action_type get_type() { return type; }
-	memory_order get_mo() { return order; }
-	void *get_location() { return location; }
-
-	TreeNode *get_node() { return node; }
-	void set_node(TreeNode *n) { node = n; }
-private:
-	action_type type;
-	memory_order order;
-	void *location;
-	thread_id_t tid;
-	int value;
-	TreeNode *node;
-};
+/* Forward declaration */
+class TreeNode;
+class Backtrack;
 
 class ModelChecker {
 public:
 	ModelChecker();
 	~ModelChecker();
 	class Scheduler *scheduler;
-	Thread *system_thread;
 
-	void add_system_thread(Thread *t);
+	void set_system_context(ucontext_t *ctxt) { system_context = ctxt; }
+	ucontext_t * get_system_context(void) { return system_context; }
 
 	void set_current_action(ModelAction *act) { current_action = act; }
 	void check_current_action(void);
-	void set_backtracking(ModelAction *act);
-	void print_trace(void);
+	void print_summary(void);
+	Thread * schedule_next_thread();
 
 	int add_thread(Thread *t);
-	Thread *get_thread(thread_id_t tid) { return thread_map[tid]; }
+	void remove_thread(Thread *t);
+	Thread * get_thread(thread_id_t tid) { return thread_map[id_to_int(tid)]; }
 
-	void assign_id(Thread *t);
+	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);
+	void set_backtracking(ModelAction *act);
+	thread_id_t get_next_replay_thread();
+	Backtrack * get_next_backtrack();
+	void reset_to_initial_state();
+
+	void print_list(action_list_t *list);
+
 	class ModelAction *current_action;
-	std::list<class ModelAction *> action_trace;
-	std::map<thread_id_t, class Thread *> thread_map;
+	Backtrack *exploring;
+	thread_id_t nextThread;
+
+	ucontext_t *system_context;
+	action_list_t *action_trace;
+	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 *, MyAlloc< class Backtrack * > > backtrack_list;
 };
 
 extern ModelChecker *model;
 
-int thread_switch_to_master(ModelAction *act);
-
 #endif /* __MODEL_H__ */