Get real GPU memory values on NV+Linux and OS X

ui/gl/gl_context_cgl.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "ui/gl/gl_context_cgl.h"
 
 #include <OpenGL/CGLRenderers.h>
+#include <OpenGL/CGLTypes.h>
 #include <vector>
 
 #include "base/debug/trace_event.h"
 #include "base/logging.h"
 #include "ui/gl/gl_bindings.h"
 #include "ui/gl/gl_implementation.h"
 #include "ui/gl/gl_surface_cgl.h"
 
 namespace gfx {
 
@@ skipping 129 matching lines @@
 
 void* GLContextCGL::GetHandle() {
   return context_;
 }
 
 void GLContextCGL::SetSwapInterval(int interval) {
   DCHECK(IsCurrent(NULL));
   LOG(WARNING) << "GLContex: GLContextCGL::SetSwapInterval is ignored.";
 }
 
+
+bool GLContextCGL::GetTotalGpuMemory(size_t* bytes) {
+  DCHECK(bytes);
+  *bytes = 0;
+
+  CGLContextObj context = reinterpret_cast<CGLContextObj>(context_);
+  if (!context)
+    return false;
+
+  // Retrieve the current renderer ID
+  GLint current_renderer_id = 0;
+  if (CGLGetParameter(context,
+                      kCGLCPCurrentRendererID,
+                      &current_renderer_id) != kCGLNoError)
+    return false;
+
+  // Iterate through the list of all renderers
+  GLuint display_mask = static_cast<GLuint>(-1);
+  CGLRendererInfoObj renderer_info = NULL;
+  GLint num_renderers = 0;
+  if (CGLQueryRendererInfo(display_mask,
+                           &renderer_info,
+                           &num_renderers) != kCGLNoError)
+    return false;
+
+  ScopedCGLRendererInfoObj scoper(renderer_info);
+
+  for (GLint renderer_index = 0;
+       renderer_index < num_renderers;
+       ++renderer_index) {
+    // Skip this if this renderer is not the current renderer.
+    GLint renderer_id = 0;
+    if (CGLDescribeRenderer(renderer_info,
+                            renderer_index,
+                            kCGLRPRendererID,
+                            &renderer_id) != kCGLNoError)
+        continue;
+    if (renderer_id != current_renderer_id)
+        continue;
+    // Retrieve the video memory for the renderer.
+    GLint video_memory = 0;
+    if (CGLDescribeRenderer(renderer_info,
+                            renderer_index,
+                            kCGLRPVideoMemory,
+                            &video_memory) != kCGLNoError)
+        continue;
+    *bytes = video_memory;
+    return true;
+  }
+
+  return false;
+}
+
 GLContextCGL::~GLContextCGL() {
   Destroy();
 }
 
 GpuPreference GLContextCGL::GetGpuPreference() {
   return gpu_preference_;
 }
 
 void GLContextCGL::ForceUseOfDiscreteGPU() {
   static CGLPixelFormatObj format = NULL;
   if (format)
     return;
   CGLPixelFormatAttribute attribs[1];
   attribs[0] = static_cast<CGLPixelFormatAttribute>(0);
   GLint num_pixel_formats = 0;
   CGLChoosePixelFormat(attribs, &format, &num_pixel_formats);
   // format is deliberately leaked.
 }
 
+void ScopedCGLDestroyRendererInfo::operator()(CGLRendererInfoObj x) const {
+  CGLDestroyRendererInfo(x);
+}
+
 }  // namespace gfx