+/** @file schedule.h
+ * @brief Thread scheduler.
+ */
+
#ifndef __SCHEDULE_H__
#define __SCHEDULE_H__
-#include "libthreads.h"
+#include <list>
+#include "mymemory.h"
+
+/* Forward declaration */
+class Thread;
+class Node;
+
+typedef enum enabled_type {
+ THREAD_DISABLED,
+ THREAD_ENABLED,
+ THREAD_SLEEP_SET
+} enabled_type_t;
+
+/** @brief The Scheduler class performs the mechanics of Thread execution
+ * scheduling. */
+class Scheduler {
+public:
+ Scheduler();
+ void add_thread(Thread *t);
+ void remove_thread(Thread *t);
+ void sleep(Thread *t);
+ void wake(Thread *t);
+ Thread * next_thread(Thread *t);
+ Thread * get_current_thread() const;
+ void print() const;
+ enabled_type_t * get_enabled() { return enabled; };
+ void remove_sleep(Thread *t);
+ void add_sleep(Thread *t);
+ enabled_type_t get_enabled(Thread *t);
+ void update_sleep_set(Node *n);
+ bool is_enabled(Thread *t) const;
+
+ SNAPSHOTALLOC
+private:
+ /** The list of available Threads that are not currently running */
+ enabled_type_t *enabled;
+ int enabled_len;
+ int curr_thread_index;
+ void set_enabled(Thread *t, enabled_type_t enabled_status);
-void schedule_add_thread(struct thread *t);
-int schedule_choose_next(struct thread **t);
+ /** The currently-running Thread */
+ Thread *current;
+};
#endif /* __SCHEDULE_H__ */