add more classes to snapshotting region

[model-checker.git] / schedule.h

diff --git a/schedule.h b/schedule.h
index c633b28987d7ef97da5d1130060021260e65b0f6..f4965369b2a62be67a124ae29a44b4172c28f0eb 100644 (file)
--- a/schedule.h
+++ b/schedule.h
@@ -2,17 +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;