From 221ebda053805a3a0f8353c282bf4f9988a7291b Mon Sep 17 00:00:00 2001
From: Brian Norris <banorris@uci.edu>
Date: Thu, 6 Dec 2012 15:30:24 -0800
Subject: [PATCH] schedule: add is_sleep_set(thread)

---
 schedule.cc | 10 ++++++++++
 schedule.h  |  1 +
 2 files changed, 11 insertions(+)

diff --git a/schedule.cc b/schedule.cc
index 8e52d84..64cfe99 100644
--- a/schedule.cc
+++ b/schedule.cc
@@ -60,6 +60,16 @@ bool Scheduler::is_enabled(thread_id_t tid) const
 	return (i >= enabled_len) ? false : (enabled[i] != THREAD_DISABLED);
 }
 
+/**
+ * @brief Check if a Thread is currently in the sleep set
+ * @param t The Thread to check
+ * @return True if the Thread is currently enabled
+ */
+bool Scheduler::is_sleep_set(const Thread *t) const
+{
+	return get_enabled(t) == THREAD_SLEEP_SET;
+}
+
 enabled_type_t Scheduler::get_enabled(const Thread *t) const
 {
 	int id = id_to_int(t->get_id());
diff --git a/schedule.h b/schedule.h
index d4780c8..035ee78 100644
--- a/schedule.h
+++ b/schedule.h
@@ -37,6 +37,7 @@ public:
 	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;
 
 	SNAPSHOTALLOC
 private:
-- 
2.34.1