Display the process memory usage instead of GC (Configurable) (#1062)

* Memory

* 更新 MemoryCounter.hx

* legacyMemoryCounter

* fix

Author: HEIHUAa

assets/languages/en/Options.xml

@@ -206,6 +206,9 @@
 
 		<str id="songOffsetAffectEditors-name">Offset in Charter</str>
 		<str id="songOffsetAffectEditors-desc">If checked, this will enable the Song Offset option in the Chart Editor too.</str>
+
+		<str id="legacyMemoryCounter-name">Legacy Memory Counter</str>
+		<str id="legacyMemoryCounter-desc">If checked, the memory counter will show the Haxe GC memory usage and its peak (the original behavior) instead of the working set memory seen in Task Manager.</str>
 	</group>
 
 </language>

assets/languages/es/Options.xml

@@ -180,6 +180,9 @@
 		<str id="charterAutoSaveWarningTime-desc">Esto controla por cuánto tiempo te advertirá el editor antes de autoguardar (en segundos..., 0 para desactivar)</str>
 
 		<str id="charterAutoSavesSeparateFolder-name">Carpeta de Autoguardados</str>
-		<str id="charterAutoSavesSeparateFolder-desc">Si está activado, el autoguardado se guardará en otra carpeta con su fecha en vez de sobreescribir el archivo actual. (song/autosaves/)</str>	</group>
+		<str id="charterAutoSavesSeparateFolder-desc">Si está activado, el autoguardado se guardará en otra carpeta con su fecha en vez de sobreescribir el archivo actual. (song/autosaves/)</str>
+
+		<str id="legacyMemoryCounter-name">Contador de Memoria Clásico</str>
+		<str id="legacyMemoryCounter-desc">Si está activado, el contador de memoria mostrará el uso de memoria del GC de Haxe y su máximo (el comportamiento original) en vez de la memoria de trabajo vista en el Administrador de Tareas.</str>	</group>
 
 </language>

assets/languages/it/Options.xml

@@ -199,5 +199,8 @@
 
 		<str id="songOffsetAffectEditors-name">Offset Editor Chart</str>
 		<str id="songOffsetAffectEditors-desc">Se abilitato, questo attiverà l'opzione Offset Canzone anche nel Editor Chart.</str>
+
+		<str id="legacyMemoryCounter-name">Contatore Memoria Classico</str>
+		<str id="legacyMemoryCounter-desc">Se attivato, il contatore di memoria mostrerà l'uso di memoria del GC di Haxe e il suo picco (il comportamento originale) invece della memoria di lavoro vista nel Task Manager.</str>
 	</group>
 </language>

assets/languages/pl/Options.xml

@@ -175,6 +175,9 @@
 
 		<str id="charterAutoSavesSeparateFolder-name">Folder Autozapisów</str>
 		<str id="charterAutoSavesSeparateFolder-desc">Jeśli włączone, Edytor będzie zapisywać pliki w osobnym folderze, zamiast nadpisywania aktualnego pliku. (song/autosaves/)</str>
+
+		<str id="legacyMemoryCounter-name">Klasyczny Licznik Pamięci</str>
+		<str id="legacyMemoryCounter-desc">Jeśli włączone, licznik pamięci pokaże użycie pamięci GC Haxe i jego szczyt (oryginalne zachowanie) zamiast pamięci roboczej widocznej w Menedżerze Zadań.</str>
 	</group>
 
 </language>

assets/languages/pt/Options.xml

@@ -206,6 +206,9 @@
 
 		<str id="songOffsetAffectEditors-name">Atraso no Mapeador</str>
 		<str id="songOffsetAffectEditors-desc">Se marcado, irá ativar o Atraso da Música no Editor de Mapas, também.</str>
+
+		<str id="legacyMemoryCounter-name">Contador de Memória Clássico</str>
+		<str id="legacyMemoryCounter-desc">Se marcado, o contador de memória mostrará o uso de memória do GC de Haxe e o seu máximo (o comportamento original) em vez da memória de trabalho vista no Gerenciador de Tarefas.</str>
 	</group>
 
 </language>

source/funkin/backend/system/framerate/MemoryCounter.hx

@@ -28,14 +28,57 @@ class MemoryCounter extends Sprite {
 			addChild(label);
 		}
 		memoryPeakText.alpha = 0.5;
+		#if !(cpp && (windows || mac || linux))
+		memoryPeakText.visible = false;
+		#end
 	}
 
 	public function reload() {}
 
+	private var usingLegacy:Bool = false;
+
 	public override function __enterFrame(t:Int) {
 		if (alpha <= 0.05) return;
 		super.__enterFrame(t);
 
+		#if (cpp && (windows || mac || linux))
+		var legacy = funkin.options.Options.legacyMemoryCounter;
+
+		if (legacy) {
+			if (legacy != usingLegacy) {
+				usingLegacy = legacy;
+				memoryPeak = 0;
+			}
+
+			final mem = MemoryUtil.currentMemUsage();
+			if (memoryPeak < mem) memoryPeak = mem;
+			if (mem == memory) {
+				updateLabelPosition();
+				return;
+			}
+
+			memory = mem;
+			memoryPeakText.visible = true;
+			memoryText.text = CoolUtil.getSizeString(memory);
+			memoryPeakText.text = ' / ${CoolUtil.getSizeString(memoryPeak)}';
+		} else {
+			if (legacy != usingLegacy) usingLegacy = legacy;
+
+			final gcMem = MemoryUtil.currentMemUsage();
+			final osMem = MemoryUtil.currentProcessMemUsage();
+
+			if (gcMem == memory && osMem == memoryPeak) {
+				updateLabelPosition();
+				return;
+			}
+
+			memory = gcMem;
+			memoryPeak = osMem;
+			memoryPeakText.visible = true;
+			memoryText.text = CoolUtil.getSizeString(gcMem);
+			memoryPeakText.text = ' / ${CoolUtil.getSizeString(osMem)}';
+		}
+		#else
 		final mem = MemoryUtil.currentMemUsage();
 
 		if (mem == memory) {
@@ -44,9 +87,8 @@ class MemoryCounter extends Sprite {
 		}
 
 		memory = mem;
-		if (memoryPeak < memory) memoryPeak = memory;
-		memoryText.text = CoolUtil.getSizeString(memory);
-		memoryPeakText.text = ' / ${CoolUtil.getSizeString(memoryPeak)}';
+		memoryText.text = CoolUtil.getSizeString(mem);
+		#end
 
 		updateLabelPosition();
 	}

source/funkin/backend/utils/MemoryUtil.hx

@@ -119,6 +119,23 @@ final class MemoryUtil {
 		#end
 	}
 
+	/**
+	 * Gets the current process memory usage (Working Set / RSS) from the OS.
+	 * This is much more granular than Haxe GC memory and updates more frequently.
+	 * Returns the value in bytes, matching what you see in Task Manager / Activity Monitor.
+	 */
+	public static inline function currentProcessMemUsage():Float {
+		#if (cpp && windows)
+		return funkin.backend.utils.native.Windows.getCurrentProcessMemory();
+		#elseif (cpp && mac)
+		return funkin.backend.utils.native.Mac.getCurrentProcessMemory();
+		#elseif (cpp && linux)
+		return funkin.backend.utils.native.Linux.getCurrentProcessMemory();
+		#else
+		return currentMemUsage();
+		#end
+	}
+
 
 	/**
 	 * Gets the memory type of the system.

source/funkin/backend/utils/native/Linux.hx

@@ -1,7 +1,7 @@
 package funkin.backend.utils.native;
 
 #if linux
-@:cppFileCode("#include <stdio.h>")
+@:cppFileCode("#include <stdio.h>\n#include <time.h>")
 @:dox(hide)
 final class Linux {
 	@:functionCode('
@@ -28,5 +28,38 @@ final class Linux {
 	{
 		return 0;
 	}
+
+	@:functionCode('
+	struct timespec now;
+	clock_gettime(CLOCK_MONOTONIC, &now);
+	long long nowMs = now.tv_sec * 1000LL + now.tv_nsec / 1000000LL;
+
+	// cache the value and only re-read /proc every 500ms, to avoid polling the kernel file every frame
+	static long long lastReadMs = 0;
+	static double cached = 0.0;
+
+	if (cached != 0.0 && (nowMs - lastReadMs) < 500)
+		return cached;
+
+	lastReadMs = nowMs;
+
+	FILE *status = fopen("/proc/self/status", "r");
+	if (status == NULL) return cached;
+
+	char line[256];
+	while (fgets(line, sizeof(line), status)) {
+		unsigned long long rss; // use 64-bit so the kB -> bytes conversion never overflows on 32-bit builds
+		if (sscanf(line, "VmRSS: %llu kB", &rss) == 1) {
+			fclose(status);
+			return cached = (double)rss * 1024.0; // kB -> bytes
+		}
+	}
+	fclose(status);
+	return cached;
+')
+	public static function getCurrentProcessMemory():Float
+	{
+		return 0;
+	}
 }
 #end

source/funkin/backend/utils/native/Mac.hx

@@ -5,6 +5,7 @@ import funkin.backend.utils.NativeAPI.CodeCursor;
 import openfl.ui.Mouse;
 
 @:headerInclude('sys/sysctl.h')
+@:headerInclude('mach/mach.h')
 @:dox(hide)
 final class Mac
 {
@@ -23,6 +24,19 @@ final class Mac
 		return 0;
 	}
 
+	@:functionCode('
+		task_basic_info_data_t info;
+		mach_msg_type_number_t count = TASK_BASIC_INFO_COUNT;
+		if (task_info(mach_task_self(), TASK_BASIC_INFO, (task_info_t)&info, &count) == KERN_SUCCESS) {
+			return (double)info.resident_size;
+		}
+		return 0.0;
+	')
+	public static function getCurrentProcessMemory():Float
+	{
+		return 0;
+	}
+
 	public static function setMouseCursorIcon(icon:CodeCursor):Void
 	{
 		final valid:Bool = external.ExternalMac.setCursorIcon(icon.toInt(), null, 0, 0);

source/funkin/backend/utils/native/Windows.hx

@@ -27,6 +27,7 @@ import funkin.backend.utils.NativeAPI.MessageBoxIcon;
 #include <wingdi.h>
 #include <shellapi.h>
 #include <uxtheme.h>
+#include <psapi.h>
 
 #define SAFE_RELEASE(punk)  \\
 			  if ((punk) != NULL)  \\
@@ -267,5 +268,17 @@ final class Windows {
 	{
 		return 0;
 	}
+
+	@:functionCode("
+		PROCESS_MEMORY_COUNTERS_EX pmc;
+		if (GetProcessMemoryInfo(GetCurrentProcess(), (PROCESS_MEMORY_COUNTERS*)&pmc, sizeof(pmc))) {
+			return (double)pmc.WorkingSetSize;
+		}
+		return 0.0;
+	")
+	public static function getCurrentProcessMemory():Float
+	{
+		return 0;
+	}
 }
 #end