From 5500af227482d60a59415df030f23b8bc7374301 Mon Sep 17 00:00:00 2001 From: kervala Date: Wed, 23 Dec 2015 14:48:33 +0100 Subject: [PATCH] Changed: Implement CSystemUtils::getTotalVideoMemory() for Windows --- code/nel/include/nel/misc/system_utils.h | 1 + code/nel/src/misc/system_utils.cpp | 238 +++++++++++++++++++++++ 2 files changed, 239 insertions(+) diff --git a/code/nel/include/nel/misc/system_utils.h b/code/nel/include/nel/misc/system_utils.h index 78fd011c6..b6f121d94 100644 --- a/code/nel/include/nel/misc/system_utils.h +++ b/code/nel/include/nel/misc/system_utils.h @@ -81,6 +81,7 @@ public: /// Detect whether the current process is a windowed application. Return true if definitely yes, false if unknown static bool detectWindowedApplication(); + static sint getTotalVideoMemory(); }; } // NLMISC diff --git a/code/nel/src/misc/system_utils.cpp b/code/nel/src/misc/system_utils.cpp index bb1111b88..a350ab305 100644 --- a/code/nel/src/misc/system_utils.cpp +++ b/code/nel/src/misc/system_utils.cpp @@ -18,6 +18,14 @@ #include "nel/misc/system_utils.h" #ifdef NL_OS_WINDOWS +#define INITGUID +#include +#include +#include +#include +#include +#include +#include # include # ifdef _WIN32_WINNT_WIN7 // only supported by Windows 7 Platform SDK @@ -365,4 +373,234 @@ bool CSystemUtils::detectWindowedApplication() return false; } +#ifdef NL_OS_WINDOWS +#ifndef SAFE_RELEASE +#define SAFE_RELEASE(p) { if (p) { (p)->Release(); (p) = NULL; } } +#endif + +typedef HRESULT (WINAPI* LPDIRECTDRAWCREATE)(GUID FAR *lpGUID, LPDIRECTDRAW FAR *lplpDD, IUnknown FAR *pUnkOuter); +typedef HRESULT (WINAPI* LPCREATEDXGIFACTORY)(REFIID, void**); + +static std::string FormatError(HRESULT hr) +{ + std::string res; + + LPTSTR errorText = NULL; + + FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ALLOCATE_BUFFER|FORMAT_MESSAGE_IGNORE_INSERTS, + NULL, hr, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), (LPTSTR)&errorText, 0, NULL); + + if (errorText) + { + res = NLMISC::toString("%s (0x%x)", errorText, hr); + LocalFree(errorText); + errorText = NULL; + } + + return res; +} + +struct SAdapter +{ + uint id; + std::string name; + sint memory; + GUID guid; + HMONITOR hMonitor; + bool found; + + SAdapter() + { + id = 0; + memory = -1; + guid = GUID_NULL; + hMonitor = NULL; + found = false; + } +}; + +static std::list s_dxgiAdapters; + +static void EnumerateUsingDXGI(IDXGIFactory *pDXGIFactory) +{ + nlassert(pDXGIFactory != NULL); + + for(uint index = 0; ; ++index) + { + IDXGIAdapter *pAdapter = NULL; + HRESULT hr = pDXGIFactory->EnumAdapters(index, &pAdapter); + // DXGIERR_NOT_FOUND is expected when the end of the list is hit + if (FAILED(hr)) break; + + DXGI_ADAPTER_DESC desc; + memset(&desc, 0, sizeof(DXGI_ADAPTER_DESC)); + + if (SUCCEEDED(pAdapter->GetDesc(&desc))) + { + SAdapter adapter; + adapter.id = index; + adapter.name = ucstring((ucchar*)desc.Description).toUtf8(); + adapter.memory = desc.DedicatedVideoMemory / 1024; + adapter.found = true; + + nldebug("DXGI Adapter: %u - %s - DedicatedVideoMemory: %d KiB", index, adapter.name.c_str(), adapter.memory); + + s_dxgiAdapters.push_back(adapter); + } + + SAFE_RELEASE(pAdapter); + } +} + +BOOL WINAPI DDEnumCallbackEx(GUID FAR* lpGUID, LPSTR lpDriverDescription, LPSTR lpDriverName, LPVOID lpContext, HMONITOR hm) +{ + SAdapter * pAdapter = (SAdapter*)lpContext; + + if (pAdapter->hMonitor == hm) + { + pAdapter->name = lpDriverDescription; + pAdapter->guid = *lpGUID; + pAdapter->found = true; + } + + return TRUE; +} + +#endif + +sint CSystemUtils::getTotalVideoMemory() +{ + sint res = -1; + +#ifdef NL_OS_WINDOWS + // using DXGI + HINSTANCE hDXGI = LoadLibraryA("dxgi.dll"); + + if (hDXGI) + { + + // We prefer the use of DXGI 1.1 + LPCREATEDXGIFACTORY pCreateDXGIFactory = (LPCREATEDXGIFACTORY)GetProcAddress(hDXGI, "CreateDXGIFactory1"); + + if (!pCreateDXGIFactory) + { + pCreateDXGIFactory = (LPCREATEDXGIFACTORY)GetProcAddress(hDXGI, "CreateDXGIFactory"); + } + + if (pCreateDXGIFactory) + { + IDXGIFactory *pDXGIFactory = NULL; + HRESULT hr = pCreateDXGIFactory(__uuidof(IDXGIFactory), (LPVOID*)&pDXGIFactory); + + if (SUCCEEDED(hr)) + { + EnumerateUsingDXGI(pDXGIFactory); + + SAFE_RELEASE(pDXGIFactory); + + if (!s_dxgiAdapters.empty()) + { + // TODO: determine what adapter is used by NeL + res = s_dxgiAdapters.front().memory; + } + else + { + nlwarning("Unable to find an DXGI adapter"); + } + } + else + { + nlwarning("Unable to create DXGI factory"); + } + } + else + { + nlwarning("dxgi.dll missing entry-point"); + } + + FreeLibrary(hDXGI); + } + + if (res == -1) + { + // using DirectDraw + HMODULE hInstDDraw = LoadLibraryA("ddraw.dll"); + + if (hInstDDraw) + { + SAdapter adapter; + adapter.hMonitor = MonitorFromWindow(s_window, MONITOR_DEFAULTTONULL); + + LPDIRECTDRAWENUMERATEEXA pDirectDrawEnumerateEx = (LPDIRECTDRAWENUMERATEEXA)GetProcAddress(hInstDDraw, "DirectDrawEnumerateExA"); + LPDIRECTDRAWCREATE pDDCreate = (LPDIRECTDRAWCREATE)GetProcAddress(hInstDDraw, "DirectDrawCreate"); + + if (pDirectDrawEnumerateEx && pDDCreate) + { + HRESULT hr = pDirectDrawEnumerateEx(DDEnumCallbackEx, (VOID*)&adapter, DDENUM_ATTACHEDSECONDARYDEVICES); + + if (SUCCEEDED(hr) && adapter.found) + { + LPDIRECTDRAW pDDraw = NULL; + hr = pDDCreate(&adapter.guid, &pDDraw, NULL); + + if (SUCCEEDED(hr)) + { + LPDIRECTDRAW7 pDDraw7 = NULL; + hr = pDDraw->QueryInterface(IID_IDirectDraw7, (VOID**)&pDDraw7); + + if (SUCCEEDED(hr)) + { + DDSCAPS2 ddscaps; + memset(&ddscaps, 0, sizeof(DDSCAPS2)); + ddscaps.dwCaps = DDSCAPS_VIDEOMEMORY | DDSCAPS_LOCALVIDMEM; + + DWORD pdwAvailableVidMem; + hr = pDDraw7->GetAvailableVidMem(&ddscaps, &pdwAvailableVidMem, NULL); + + if (SUCCEEDED(hr)) + { + res = (sint)pdwAvailableVidMem / 1024; + nlinfo("DirectDraw Adapter: %s - DedicatedVideoMemory: %d KiB", adapter.name.c_str(), adapter.memory); + } + else + { + nlwarning("Unable to get DirectDraw available video memory: %s", FormatError(hr).c_str()); + } + + SAFE_RELEASE(pDDraw7); + } + else + { + nlwarning("Unable to query IDirectDraw7 interface: %s", FormatError(hr).c_str()); + } + } + else + { + nlwarning("Unable to call DirectDrawCreate: %s", FormatError(hr).c_str()); + } + } + else + { + nlwarning("Unable to enumerate DirectDraw adapters (%s): %s", (adapter.found ? "found":"not found"), FormatError(hr).c_str()); + } + } + else + { + nlwarning("Unable to get pointer on DirectDraw functions (DirectDrawEnumerateExA %p, DirectDrawCreate %p)", pDirectDrawEnumerateEx, pDDCreate); + } + + FreeLibrary(hInstDDraw); + } + else + { + nlwarning("Unable to load ddraw.dll"); + } + } +#else + // TODO: implement for other systems +#endif + + return res; +} + } // NLMISC