+/** @file schedule.h
+ * @brief Thread scheduler.
+ */
+
#ifndef __SCHEDULE_H__
#define __SCHEDULE_H__
-#include <queue>
+#include <list>
+#include "mymemory.h"
+
+/* Forward declaration */
+class Thread;
+class Node;
-#include "libthreads.h"
-#include "model.h"
+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:
- void add_thread(struct thread *t);
- struct thread * next_thread(void);
- struct thread * get_current_thread(void);
+ 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:
- std::queue<struct thread *> queue;
- struct thread *current;
+ /** 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);
+
+ /** The currently-running Thread */
+ Thread *current;
};
#endif /* __SCHEDULE_H__ */