threads/model: move switch_to_master from class Thread to class ModelChecker

[model-checker.git] / model.cc

diff --git a/model.cc b/model.cc
index 19194cda06a766ccb84b1674b6fb5142bcc4bef4..1015701d917803c75541fb2d3b830a0f223c301c 100644 (file)
--- a/model.cc
+++ b/model.cc
@@ -2,6 +2,7 @@
 
 #include "model.h"
 #include "schedule.h"
+#include "common.h"
 
 ModelChecker *model;
 
@@ -10,7 +11,9 @@ ModelChecker::ModelChecker()
        /* First thread created (system_thread) will have id 1 */
        this->used_thread_id = 0;
        /* Initialize default scheduler */
-       this->scheduler = new DefaultScheduler();
+       this->scheduler = new Scheduler();
+
+       this->current_action = NULL;
 }
 
 ModelChecker::~ModelChecker()
@@ -18,25 +21,61 @@ ModelChecker::~ModelChecker()
        delete this->scheduler;
 }
 
-void ModelChecker::assign_id(struct thread *t)
+void ModelChecker::assign_id(Thread *t)
 {
-       t->id = ++this->used_thread_id;
+       t->set_id(++used_thread_id);
 }
 
-void ModelChecker::add_system_thread(struct thread *t)
+void ModelChecker::add_system_thread(Thread *t)
 {
        this->system_thread = t;
 }
 
+void ModelChecker::check_current_action(void)
+{
+       if (this->current_action)
+               this->action_trace.push_back(this->current_action);
+       else
+               DEBUG("trying to push NULL action...\n");
+}
+
+void ModelChecker::print_trace(void)
+{
+       std::list<class ModelAction *>::iterator it;
+
+       for (it = action_trace.begin(); it != action_trace.end(); it++) {
+               DBG();
+               (*it)->print();
+       }
+}
+
+int ModelChecker::add_thread(Thread *t)
+{
+       thread_map[t->get_id()] = t;
+       return 0;
+}
+
+int ModelChecker::switch_to_master(ModelAction *act)
+{
+       Thread *old, *next;
+
+       DBG();
+       old = thread_current();
+       set_current_action(act);
+       old->set_state(THREAD_READY);
+       next = system_thread;
+       return old->swap(next);
+}
+
 ModelAction::ModelAction(action_type_t type, memory_order order, void *loc, int value)
 {
-       struct thread *t = thread_current();
+       Thread *t = thread_current();
        ModelAction *act = this;
 
        act->type = type;
        act->order = order;
        act->location = loc;
-       act->tid = t->id;
+       act->tid = t->get_id();
        act->value = value;
 }