From 7e91df4083e723ff1d2436d96123c0c31fc8747f Mon Sep 17 00:00:00 2001
From: Misha <106913236+MishaProductions@users.noreply.github.com>
Date: Mon, 15 Jan 2024 11:32:01 -0500
Subject: [PATCH] MxControlPresenter

---
 .../legoomni/include/mxcontrolpresenter.h     |  1 +
 .../src/control/mxcontrolpresenter.cpp        | 66 ++++++++++++++++---
 LEGO1/omni/include/mxutil.h                   |  2 +
 LEGO1/omni/src/common/mxutil.cpp              | 17 +++++
 4 files changed, 76 insertions(+), 10 deletions(-)

diff --git a/LEGO1/lego/legoomni/include/mxcontrolpresenter.h b/LEGO1/lego/legoomni/include/mxcontrolpresenter.h
index 9779e45f..6e456851 100644
--- a/LEGO1/lego/legoomni/include/mxcontrolpresenter.h
+++ b/LEGO1/lego/legoomni/include/mxcontrolpresenter.h
@@ -34,6 +34,7 @@ class MxControlPresenter : public MxCompositePresenter {
 	virtual void Enable(MxBool p_enable) override;                           // vtable+0x54
 	virtual MxBool VTable0x64(undefined4 p_undefined) override;              // vtable+0x64
 	virtual void VTable0x68(MxBool p_undefined);                             // vtable+0x68
+	virtual void VTable0x6c(MxU32 p_new4e);                                  // vtable+0x6c
 
 private:
 	MxBool FUN_10044270(undefined4, undefined4, undefined4*);
diff --git a/LEGO1/lego/legoomni/src/control/mxcontrolpresenter.cpp b/LEGO1/lego/legoomni/src/control/mxcontrolpresenter.cpp
index b7f8fa8e..8f6183f0 100644
--- a/LEGO1/lego/legoomni/src/control/mxcontrolpresenter.cpp
+++ b/LEGO1/lego/legoomni/src/control/mxcontrolpresenter.cpp
@@ -1,6 +1,7 @@
 #include "mxcontrolpresenter.h"
 
 #include "mxticklemanager.h"
+#include "mxutil.h"
 
 DECOMP_SIZE_ASSERT(MxControlPresenter, 0x5c)
 
@@ -47,11 +48,37 @@ MxResult MxControlPresenter::AddToManager()
 	return SUCCESS;
 }
 
-// STUB: LEGO1 0x10044190
-MxResult MxControlPresenter::StartAction(MxStreamController*, MxDSAction*)
+// FUNCTION: LEGO1 0x10044190
+MxResult MxControlPresenter::StartAction(MxStreamController* p_controller, MxDSAction* p_action)
 {
-	// TODO
-	return SUCCESS;
+	MxResult result = MxCompositePresenter::StartAction(p_controller, p_action);
+	MxU8 i = 0;
+
+	FUN_100b7220(m_action, 0x81, TRUE);
+	ParseExtra();
+
+	for (MxCompositePresenterList::iterator it = m_list.begin(); it != m_list.end(); it++) {
+		MxBool toggle;
+		if (m_unk0x4c == 3 && m_unk0x4e == 0) {
+			toggle = FALSE;
+		}
+		else {
+			if (!IsEnabled()) {
+				toggle = FALSE;
+			}
+			toggle = m_unk0x4e == i;
+		}
+
+		i++;
+		(*it)->Enable(toggle);
+	}
+
+	if (m_unk0x4c == 3) {
+		(*m_list.end())->GetAction()->SetFlags((*m_list.end())->GetAction()->GetFlags() | 0x400);
+	}
+	TickleManager()->RegisterClient(this, 200);
+
+	return result;
 }
 
 // FUNCTION: LEGO1 0x10044260
@@ -78,7 +105,7 @@ MxBool MxControlPresenter::FUN_10044480(undefined4, undefined4*)
 }
 
 // STUB: LEGO1 0x10044540
-void MxControlPresenter::FUN_10044540(undefined2)
+void MxControlPresenter::VTable0x6c(MxU32 p_new4e)
 {
 	// TODO
 }
@@ -97,15 +124,34 @@ void MxControlPresenter::ParseExtra()
 	// TODO
 }
 
-// STUB: LEGO1 0x10044820
+// FUNCTION: LEGO1 0x10044820
 void MxControlPresenter::Enable(MxBool p_enable)
 {
-	// TODO
+	if (MxPresenter::IsEnabled() != p_enable) {
+		MxPresenter::Enable(p_enable);
+
+		MxU8 i = 0;
+		for (MxCompositePresenterList::iterator it = m_list.begin(); it != m_list.end(); it++) {
+			if (i == m_unk0x4e) {
+				(*it)->Enable((m_unk0x4c != 3 || i != 0) ? p_enable : 0);
+				break;
+			}
+			i++;
+		}
+
+		if (!p_enable) {
+			m_unk0x4e = 0;
+		}
+	}
 }
 
-// STUB: LEGO1 0x100448a0
+// FUNCTION: LEGO1 0x100448a0
 MxBool MxControlPresenter::HasTickleStatePassed(TickleState p_tickleState)
 {
-	// TODO
-	return TRUE;
+	MxCompositePresenterList::iterator it = m_list.begin();
+	for (MxU8 i = m_unk0x4e; i != 0; i--) {
+		it++;
+	}
+
+	return (*it)->HasTickleStatePassed(p_tickleState);
 }
diff --git a/LEGO1/omni/include/mxutil.h b/LEGO1/omni/include/mxutil.h
index ca2ab867..bd0247fd 100644
--- a/LEGO1/omni/include/mxutil.h
+++ b/LEGO1/omni/include/mxutil.h
@@ -7,6 +7,7 @@
 
 class MxDSFile;
 class MxDSObject;
+class MxDSAction;
 
 template <class T>
 inline T Abs(T p_t)
@@ -70,6 +71,7 @@ MxBool FUN_100b6e10(
 
 __declspec(dllexport) void MakeSourceName(char*, const char*);
 __declspec(dllexport) void SetOmniUserMessage(void (*)(const char*, int));
+void FUN_100b7220(MxDSAction* p_action, MxU32 p_newFlags, MxBool p_setFlags);
 __declspec(dllexport) MxDSObject* CreateStreamObject(MxDSFile*, MxS16);
 
 MxBool KeyValueStringParse(char*, const char*, const char*);
diff --git a/LEGO1/omni/src/common/mxutil.cpp b/LEGO1/omni/src/common/mxutil.cpp
index 538368d1..b1508bc7 100644
--- a/LEGO1/omni/src/common/mxutil.cpp
+++ b/LEGO1/omni/src/common/mxutil.cpp
@@ -1,6 +1,9 @@
 #include "mxutil.h"
 
+#include "mxdsaction.h"
+#include "mxdsactionlist.h"
 #include "mxdsfile.h"
+#include "mxdsmultiaction.h"
 #include "mxdsobject.h"
 #include "mxrect32.h"
 
@@ -112,6 +115,20 @@ void SetOmniUserMessage(void (*p_userMsg)(const char*, int))
 	g_omniUserMessage = p_userMsg;
 }
 
+// FUNCTION: LEGO1 0x100b7220
+void FUN_100b7220(MxDSAction* p_action, MxU32 p_newFlags, MxBool p_setFlags)
+{
+	p_action->SetFlags(!p_setFlags ? p_action->GetFlags() & ~p_newFlags : p_action->GetFlags() | p_newFlags);
+
+	if (p_action->IsA("MxDSMultiAction")) {
+		MxDSActionListCursor cursor(((MxDSMultiAction*) p_action)->GetActionList());
+		MxDSAction* currentAction;
+		while (cursor.Next(currentAction)) {
+			FUN_100b7220(currentAction, p_newFlags, p_setFlags);
+		}
+	}
+}
+
 // Should probably be somewhere else
 // FUNCTION: LEGO1 0x100c0280
 MxDSObject* CreateStreamObject(MxDSFile* p_file, MxS16 p_ofs)