1 files changed, 79 insertions, 0 deletions
diff --git a/arm9/src/overlay_manager.c b/arm9/src/overlay_manager.c
new file mode 100644
index 00000000..a59e15ac
--- /dev/null
+++ b/arm9/src/overlay_manager.c
@@ -0,0 +1,79 @@
+#include "global.h"
+#include "main.h"
+#include "poke_overlay.h"
+#include "heap.h"
+#include "overlay_manager.h"
+
+THUMB_FUNC struct UnkStruct_02006234 * OverlayManager_new(struct Unk21DBE18 * ovly_mgr, int * a1, u32 heap_id)
+{
+    struct UnkStruct_02006234 * ret = (struct UnkStruct_02006234 *)AllocFromHeap(heap_id, sizeof(struct UnkStruct_02006234));
+    ret->ovly_mgr = *ovly_mgr;
+    ret->unk10 = 0;
+    ret->unk14 = 0;
+    ret->unk18 = a1;
+    ret->unk1C = NULL;
+    ret->unk20 = 0;
+    ret->unk24 = 0;
+    return ret;
+}
+
+THUMB_FUNC void OverlayManager_delete(struct UnkStruct_02006234 * a0)
+{
+    FreeToHeap(a0);
+}
+
+THUMB_FUNC void * OverlayManager_CreateAndGetData(struct UnkStruct_02006234 * a0, u32 size, u32 heap_id)
+{
+    return a0->unk1C = AllocFromHeap(heap_id, size);
+}
+
+THUMB_FUNC void * OverlayManager_GetData(struct UnkStruct_02006234 * a0)
+{
+    return a0->unk1C;
+}
+
+THUMB_FUNC void OverlayManager_FreeData(struct UnkStruct_02006234 * a0)
+{
+    FreeToHeap(a0->unk1C);
+    a0->unk1C = NULL;
+}
+
+THUMB_FUNC int * OverlayManager_GetField18(struct UnkStruct_02006234 * a0)
+{
+    return a0->unk18;
+}
+
+THUMB_FUNC BOOL OverlayManager_Run(struct UnkStruct_02006234 * a0)
+{
+    switch (a0->unk10)
+    {
+    case 0:
+        if (a0->ovly_mgr.ovly != SDK_OVERLAY_INVALID_ID)
+            HandleLoadOverlay(a0->ovly_mgr.ovly, 2);
+        a0->unk10 = 1;
+        // fallthrough
+    case 1:
+        if (a0->ovly_mgr.unk0(a0, &a0->unk14) == TRUE)
+        {
+            a0->unk10 = 2;
+            a0->unk14 = 0;
+        }
+        break;
+    case 2:
+        if (a0->ovly_mgr.unk4(a0, &a0->unk14) == TRUE)
+        {
+            a0->unk10 = 3;
+            a0->unk14 = 0;
+        }
+        break;
+    case 3:
+        if (a0->ovly_mgr.unk8(a0, &a0->unk14) == TRUE)
+        {
+            if (a0->ovly_mgr.ovly != SDK_OVERLAY_INVALID_ID)
+                UnloadOverlayByID(a0->ovly_mgr.ovly);
+            return TRUE;
+        }
+        break;
+    }
+    return FALSE;
+}