schedule: reset scheduler when thread is removed

[model-checker.git] / schedule.cc

diff --git a/schedule.cc b/schedule.cc
index 67b40ec88a809d5c2bff10e07787481a7d934566..c9a1ce1ed8be0f9b7176585bb53aaa53f8be8cdf 100644 (file)
--- a/schedule.cc
+++ b/schedule.cc
@@ -9,6 +9,14 @@ void Scheduler::add_thread(Thread *t)
        readyList.push_back(t);
 }
 
+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();