完美获取N卡A卡的显存大小（使用OpenGL）

// 基于扩展NVX_gpu_memory_info extension

        UINT      QueryNVidiaCardMemory()

        {

            __try

            {

                int iVal = 0;

 

                glGetIntegerv( GPU_MEMORY_INFO_DEDICATED_VIDMEM_NVX, &iVal );

 

                return (UINT)iVal/1024;

            }

            __except(1)

            {

            }

 

            return 0;

        }

 

        // 基于扩展WGL_AMD_gpu_association

        UINT      QueryAmdCardMemory()

        {

            __try

            {

                typedef UINT (WINAPI * PFNWGLGETGPUIDSAMDPROC) (UINT maxCount, UINT* ids);

                typedef INT (WINAPI * PFNWGLGETGPUINFOAMDPROC) (UINT id, INT property, GLenum dataType, UINT size, void* data);

 

                PFNWGLGETGPUIDSAMDPROC wglGetGPUIDsAMD = (PFNWGLGETGPUIDSAMDPROC)wglGetProcAddress("wglGetGPUIDsAMD");

                PFNWGLGETGPUINFOAMDPROC wglGetGPUInfoAMD = (PFNWGLGETGPUINFOAMDPROC)wglGetProcAddress("wglGetGPUInfoAMD");

 

                if( !wglGetGPUIDsAMD || !wglGetGPUInfoAMD )

                {

                    return 0;

                }

 

                UINT n = wglGetGPUIDsAMD(0, 0);

 

                UINT ids[24];

 

#define WGL_GPU_RAM_AMD 0x21A3

 

                size_t TotalMemSize = 0;

 

                wglGetGPUIDsAMD(n, ids);

 

                wglGetGPUInfoAMD(

                    ids[0], 

                    WGL_GPU_RAM_AMD, 

                    GL_UNSIGNED_INT, 

                    sizeof(size_t),

                    &TotalMemSize

                    );                

 

                return (UINT)TotalMemSize;

            }

            __except(1)

            {

            }

 

            return 0;

        }

 

        UINT      QueryDedicatedVideoCardMemory( bool bNvidiaCard, bool bAmdAtiCard )

        {

            __try

            {

                HWND hWnd = (HWND)你可以用的HWND;

 

                HDC hdc = ::GetDC( hWnd );

 

                if( hdc == NULL )

                {

                    return 0;

                }

 

                PIXELFORMATDESCRIPTOR FormatDesc;

 

                memset(&FormatDesc, 0, sizeof(PIXELFORMATDESCRIPTOR));

                FormatDesc.nSize = sizeof(PIXELFORMATDESCRIPTOR);

                FormatDesc.nVersion = 1;

                FormatDesc.dwFlags = PFD_DRAW_TO_WINDOW | PFD_SUPPORT_OPENGL | /*PFD_SUPPORT_GDI|*/ PFD_DOUBLEBUFFER;

                FormatDesc.iPixelType = PFD_TYPE_RGBA;

                FormatDesc.cColorBits = 32;

                FormatDesc.cDepthBits = 32;

                // FormatDesc.cStencilBits = 0;

                FormatDesc.iLayerType = PFD_MAIN_PLANE;

 

                GLuint PixelFormat = ChoosePixelFormat( hdc, &FormatDesc );

 

                if( PixelFormat == NULL )

                {

                    return 0;

                }

                

                SetPixelFormat( hdc, PixelFormat, &FormatDesc );

 

                HGLRC hContext = wglCreateContext( hdc );

 

                if( hContext == NULL )

                {

                    return 0;

                }

                

                __try

                {

                    if( wglMakeCurrent( hdc, hContext ) )

                    {

                        if( bNvidiaCard )

                        {

                            return QueryNVidiaCardMemory();

                        }

                        else if( bAmdAtiCard )

                        {

                            return QueryAmdCardMemory();

                        }

                    }

                    else

                    {

                        return 0;

                    }

                }

                __finally

                {

                    wglMakeCurrent( NULL, NULL );

 

                    ::DeleteDC( hdc );

                    wglDeleteContext( hContext );

                }

            }

            __except(1)

            {                

            }

 

            return 0;

        }

 

转载于:https://www.cnblogs.com/forworldpeace/p/3405844.html