Merge branch 'master' of ssh://demsky.eecs.uci.edu/home/git/model-checker

[model-checker.git] / schedule.h

diff --git a/schedule.h b/schedule.h
index 86e4e40da18c5bc41bfd3908d76b4c48dc80b20b..f4965369b2a62be67a124ae29a44b4172c28f0eb 100644 (file)
--- a/schedule.h
+++ b/schedule.h
@@ -2,16 +2,21 @@
 #define __SCHEDULE_H__
 
 #include <list>
+#include "mymemory.h"
 
-#include "threads.h"
-#include "model.h"
+/* Forward declaration */
+class Thread;
 
 class Scheduler {
 public:
+       Scheduler();
        void add_thread(Thread *t);
+       void remove_thread(Thread *t);
        Thread * next_thread(void);
        Thread * get_current_thread(void);
        void print();
+
+       SNAPSHOTALLOC
 private:
        std::list<Thread *> readyList;
        Thread *current;