Browse Source

UI: Add function to get current memory usage (win32)

jp9000 8 năm trước cách đây
mục cha
commit
d0d9cc4254
2 tập tin đã thay đổi với 13 bổ sung0 xóa
  1. 12 0
      UI/platform-windows.cpp
  2. 1 0
      UI/platform.hpp

+ 12 - 0
UI/platform-windows.cpp

@@ -30,6 +30,7 @@ using namespace std;
 #include <shellapi.h>
 #include <shlobj.h>
 #include <Dwmapi.h>
+#include <psapi.h>
 #include <mmdeviceapi.h>
 #include <audiopolicy.h>
 
@@ -259,3 +260,14 @@ bool DisableAudioDucking(bool disable)
 	result = sessionControl2->SetDuckingPreference(disable);
 	return SUCCEEDED(result);
 }
+
+uint64_t CurrentMemoryUsage()
+{
+	PROCESS_MEMORY_COUNTERS pmc = {};
+	pmc.cb = sizeof(pmc);
+
+	if (!GetProcessMemoryInfo(GetCurrentProcess(), &pmc, sizeof(pmc)))
+		return 0;
+
+	return (uint64_t)pmc.WorkingSetSize;
+}

+ 1 - 0
UI/platform.hpp

@@ -43,6 +43,7 @@ void SetAeroEnabled(bool enable);
 void SetProcessPriority(const char *priority);
 void SetWin32DropStyle(QWidget *window);
 bool DisableAudioDucking(bool disable);
+uint64_t CurrentMemoryUsage();
 #endif
 
 #ifdef __APPLE__