+/** @file schedule.h
+ * @brief Thread scheduler.
+ */
+
#ifndef __SCHEDULE_H__
#define __SCHEDULE_H__
-#include "libthreads.h"
-#include "model.h"
+#include "mymemory.h"
+#include "modeltypes.h"
-struct scheduler {
- void (*init)(void);
- void (*exit)(void);
- void (*add_thread)(struct thread *t);
- struct thread * (*next_thread)(void);
- struct thread * (*get_current_thread)(void);
+/* Forward declaration */
+class Thread;
+class Node;
+class ModelExecution;
- void *priv;
-};
+typedef enum enabled_type {
+ THREAD_DISABLED,
+ THREAD_ENABLED,
+ THREAD_SLEEP_SET
+} enabled_type_t;
+
+void enabled_type_to_string(enabled_type_t e, char *str);
+
+/** @brief The Scheduler class performs the mechanics of Thread execution
+ * scheduling. */
+class Scheduler {
+public:
+ Scheduler();
+ void register_engine(ModelExecution *execution);
-void scheduler_init(struct model_checker *mod);
+ void add_thread(Thread *t);
+ void remove_thread(Thread *t);
+ void sleep(Thread *t);
+ void wake(Thread *t);
+ Thread * select_next_thread(Node *n);
+ void set_current_thread(Thread *t);
+ Thread * get_current_thread() const;
+ void print() const;
+ enabled_type_t * get_enabled_array() const { return enabled; };
+ void remove_sleep(Thread *t);
+ void add_sleep(Thread *t);
+ enabled_type_t get_enabled(const Thread *t) const;
+ void update_sleep_set(Node *n);
+ bool is_enabled(const Thread *t) const;
+ bool is_enabled(thread_id_t tid) const;
+ bool is_sleep_set(const Thread *t) const;
+ bool all_threads_sleeping() const;
+ void set_scheduler_thread(thread_id_t tid);
+
+ SNAPSHOTALLOC
+private:
+ ModelExecution *execution;
+ /** 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__ */