schedule: reset scheduler when thread is removed

[model-checker.git] / schedule.cc

diff --git a/schedule.cc b/schedule.cc
index 0f550c41962a5046c6934fe5380d17e0cda39e36..c9a1ce1ed8be0f9b7176585bb53aaa53f8be8cdf 100644 (file)
--- a/schedule.cc
+++ b/schedule.cc
@@ -9,7 +9,15 @@ void Scheduler::add_thread(Thread *t)
        readyList.push_back(t);
 }
 
-Thread *Scheduler::next_thread(void)
+void Scheduler::remove_thread(Thread *t)
+{
+       if (current == t)
+               current = NULL;
+       else
+               readyList.remove(t);
+}
+
+Thread * Scheduler::next_thread(void)
 {
        Thread *t = model->schedule_next_thread();
 
@@ -29,7 +37,7 @@ Thread *Scheduler::next_thread(void)
        return t;
 }
 
-Thread *Scheduler::get_current_thread(void)
+Thread * Scheduler::get_current_thread(void)
 {
        return current;
 }