Add heap draw

Author: TheIndra55

src/cdc/sys/Allocator.cpp

@@ -0,0 +1,12 @@
+#include "Allocator.h"
+#include "util/Hooking.h"
+
+cdc::Allocator* cdc::Alloc::Get()
+{
+	return *(cdc::Allocator**)GET_ADDRESS(0x1159DBC, 0x8AFEA4, 0x000000);
+}
+
+cdc::HeapAllocator* cdc::Alloc::GetHeap()
+{
+	return *(cdc::HeapAllocator**)GET_ADDRESS(0x1159DC0, 0x8AFEA8, 0x000000);
+}

src/cdc/sys/Allocator.h

@@ -59,4 +59,11 @@ namespace cdc
 
 		virtual Heap* GetHeapImpl() = 0;
 	};
+
+	class Alloc
+	{
+	public:
+		static Allocator* Get();
+		static HeapAllocator* GetHeap();
+	};
 }

src/modules/Debug.cpp

@@ -8,6 +8,8 @@
 #include "util/Hooking.h"
 #include "game/Game.h"
 
+#include "cdc/sys/Allocator.h"
+
 static Debug* s_debug;
 
 // Event debug print hooks
@@ -76,6 +78,18 @@ void Debug::OnPostInitialize()
 	Initialize();
 }
 
+void Debug::OnFrame()
+{
+	if (m_drawHeap)
+	{
+		auto font = Font::GetMainFont();
+		auto allocator = cdc::Alloc::GetHeap();
+
+		font->SetCursor(10, 10);
+		font->Print("Heap: %.1f/%dMB", allocator->GetTotalAllocSize() / 1024.f / 1024.f, allocator->GetMaxHeapSize() >> 20);
+	}
+}
+
 void Debug::OnMenu()
 {
 	if (ImGui::BeginMenu("Debug"))
@@ -94,6 +108,7 @@ void Debug::OnMenu()
 			}
 		}
 
+		ImGui::MenuItem("Draw heap", nullptr, &m_drawHeap);
 		ImGui::MenuItem("Debug keypad", nullptr, (bool*)GET_ADDRESS(0x107696C, 0x7C8A3C, 0x000000));
 
 		ImGui::EndMenu();

src/modules/Debug.h

@@ -6,6 +6,7 @@ class Debug : public Module
 {
 private:
 	bool m_drawDebug = false;
+	bool m_drawHeap = false;
 
 	void Initialize() const noexcept;
 
@@ -15,5 +16,6 @@ class Debug : public Module
 	bool IsDrawDebug() const noexcept { return m_drawDebug; }
 
 	void OnPostInitialize();
+	void OnFrame();
 	void OnMenu();
 };