1 files changed, 73 insertions, 0 deletions
diff --git a/src/egg/core/eggThreadMgr.cpp b/src/egg/core/eggThreadMgr.cpp
new file mode 100644
index 00000000..e5fcd44a
--- /dev/null
+++ b/src/egg/core/eggThreadMgr.cpp
@@ -0,0 +1,73 @@
+#include "egg/core/eggThreadMgr.h"
+
+#include "rvl/OS.h"
+
+namespace EGG {
+
+ThreadMgr *ThreadMgr::sInstance; // Never initialized
+
+s32 ThreadMgr::getThreadIndex(OSThread *thread) {
+    for (s32 i = 0; i < mThreadCount; i++) {
+        if (mThreadList[i].mThread == thread) {
+            return i;
+        }
+    }
+    return -1;
+}
+
+s32 ThreadMgr::doRegisterThread(OSThread *thread) {
+    if (mThreadCount != mMaxThreads) {
+        s32 i = mThreadCount;
+
+        mThreadList[i].mThread = thread;
+        mThreadList[i]._04 = _EC;
+        mThreadList[i]._08 = true;
+        mThreadList[i]._09[0] = '\0';
+        mThreadList[i]._2C = _F4;
+
+        mThreadCount++;
+
+        if (mSortEnabled) {
+            sortByPriority();
+        }
+
+        return getThreadIndex(thread);
+    }
+
+    return -1;
+}
+
+void ThreadMgr::sortByPriority() {
+    // Selection sort on thread priority
+    for (s32 i = 0; i < mThreadCount - 1; i++) {
+        OSPriority minPrio = 31;
+        s32 minPrioIndex = -1;
+
+        for (s32 j = i; j < mThreadCount; j++) {
+            OSPriority prio = OSGetThreadPriority(mThreadList[j].mThread);
+            if (prio <= minPrio) {
+                minPrio = prio;
+                minPrioIndex = j;
+            }
+        }
+
+        if (i != minPrioIndex) {
+            ThreadInfo temp = mThreadList[i];
+            mThreadList[i] = mThreadList[minPrioIndex];
+            mThreadList[minPrioIndex] = temp;
+        }
+    }
+}
+
+void ThreadMgr::registerThread(OSThread *thread, UnknownStruct arg) {
+    s32 i = getThreadIndex(thread);
+    if (i == -1) {
+        i = doRegisterThread(thread);
+    }
+
+    if (i != -1) {
+        mThreadList[i]._04 = arg;
+    }
+}
+
+} // namespace EGG