add more classes to snapshotting region

[c11tester.git] / schedule.h

diff --git a/schedule.h b/schedule.h
index 48fdde1eb08f2f39eae6984f84ce63dbcf7b3cfc..f4965369b2a62be67a124ae29a44b4172c28f0eb 100644 (file)
--- a/schedule.h
+++ b/schedule.h
@@ -2,18 +2,21 @@
 #define __SCHEDULE_H__
 
 #include <list>
+#include "mymemory.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;