Identify the active GPU using GL strings in multiple GPU situation.

gpu/config/gpu_info_collector.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 "gpu/config/gpu_info_collector.h"
 
 #include <string>
 #include <vector>
 
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/metrics/sparse_histogram.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_piece.h"
 #include "base/strings/string_split.h"
+#include "base/strings/string_util.h"
 #include "base/trace_event/trace_event.h"
 #include "ui/gl/gl_bindings.h"
 #include "ui/gl/gl_context.h"
 #include "ui/gl/gl_implementation.h"
 #include "ui/gl/gl_surface.h"
 #include "ui/gl/gl_version_info.h"
 
 namespace {
 
 scoped_refptr<gfx::GLSurface> InitializeGLSurface() {
@@ skipping 116 matching lines @@
   }
 
   // TODO(kbr): remove once the destruction of a current context automatically
   // clears the current context.
   context->ReleaseCurrent(surface.get());
 
   std::string glsl_version = GetVersionFromString(glsl_version_string);
   gpu_info->pixel_shader_version = glsl_version;
   gpu_info->vertex_shader_version = glsl_version;
 
+  IdentifyActiveGPU(gpu_info);
   return CollectDriverInfoGL(gpu_info);
 }
 
 void MergeGPUInfoGL(GPUInfo* basic_gpu_info,
                     const GPUInfo& context_gpu_info) {
   DCHECK(basic_gpu_info);
+  // Copy over active GPU setting because it might change.
+  basic_gpu_info->gpu.active = context_gpu_info.gpu.active;
+  size_t count = std::min(basic_gpu_info->secondary_gpus.size(),
+                          context_gpu_info.secondary_gpus.size());
+  for (size_t ii = 0; ii < count; ++ii) {
+    basic_gpu_info->secondary_gpus[ii].active =
+        context_gpu_info.secondary_gpus[ii].active;

[Ken Russell (switch to Gerrit), 2015/12/08 22:58:48]

Are the secondary GPUs guaranteed to be in the same order?

[Zhenyao Mo, 2015/12/09 18:36:53]

I don't think the order will change because it's the same collecting code plus
the context stuff, but let's just copy over all the GPUs,
to be on the safe side.

+  }
+
   basic_gpu_info->gl_renderer = context_gpu_info.gl_renderer;
   basic_gpu_info->gl_vendor = context_gpu_info.gl_vendor;
   basic_gpu_info->gl_version = context_gpu_info.gl_version;
   basic_gpu_info->gl_extensions = context_gpu_info.gl_extensions;
   basic_gpu_info->pixel_shader_version =
       context_gpu_info.pixel_shader_version;
   basic_gpu_info->vertex_shader_version =
       context_gpu_info.vertex_shader_version;
   basic_gpu_info->max_msaa_samples =
       context_gpu_info.max_msaa_samples;
@@ skipping 15 matching lines @@
   basic_gpu_info->context_info_state = context_gpu_info.context_info_state;
   basic_gpu_info->initialization_time = context_gpu_info.initialization_time;
   basic_gpu_info->video_decode_accelerator_supported_profiles =
       context_gpu_info.video_decode_accelerator_supported_profiles;
   basic_gpu_info->video_encode_accelerator_supported_profiles =
       context_gpu_info.video_encode_accelerator_supported_profiles;
   basic_gpu_info->jpeg_decode_accelerator_supported =
       context_gpu_info.jpeg_decode_accelerator_supported;
 }
 
+void IdentifyActiveGPU(GPUInfo* gpu_info) {
+  const std::string kNVidiaName = "nvidia";
+  const std::string kIntelName = "intel";
+  const std::string kAMDName = "amd";
+  const std::string kATIName = "ati";
+  const std::string kVendorNames[] = {
+      kNVidiaName, kIntelName, kAMDName, kATIName};
+
+  const uint32 kNVidiaID = 0x10de;
+  const uint32 kIntelID = 0x8086;
+  const uint32 kAMDID = 0x1002;
+  const uint32 kATIID = 0x1002;
+  const uint32 kVendorIDs[] = {
+      kNVidiaID, kIntelID, kAMDID, kATIID};
+
+  DCHECK(gpu_info);
+  if (gpu_info->secondary_gpus.size() == 0)
+    return;
+
+  uint32 active_vendor_id = 0;
+  if (!gpu_info->gl_vendor.empty()) {
+    std::string gl_vendor_lower = base::ToLowerASCII(gpu_info->gl_vendor);
+    for (size_t ii = 0; ii < arraysize(kVendorNames); ++ii) {
+      if (gl_vendor_lower.find(kVendorNames[ii]) != std::string::npos) {
+        active_vendor_id = kVendorIDs[ii];

[Ken Russell (switch to Gerrit), 2015/12/08 22:58:48]

We need to be very careful here. "ati" is a substring of "corporation" and I
recall NVIDIA reports "NVIDIA Corporation". I've run into this problem before
trying to parse GL strings. I think you should tokenize the renderer and vendor
strings and look for the lowercase tokens being exactly equal to each of the
above names. Please try to add a test for this case.

[Zhenyao Mo, 2015/12/09 18:36:53]

Done.

+        break;
+      }
+    }
+  }
+  if (active_vendor_id == 0 && !gpu_info->gl_renderer.empty()) {
+    std::string gl_renderer_lower = base::ToLowerASCII(gpu_info->gl_renderer);
+    for (size_t ii = 0; ii < arraysize(kVendorNames); ++ii) {
+      if (gl_renderer_lower.find(kVendorNames[ii]) != std::string::npos) {
+        active_vendor_id = kVendorIDs[ii];
+        break;
+      }
+    }
+  }
+  if (active_vendor_id == 0) {
+    // We fail to idendify the GPU vendor through GL_VENDOR/GL_RENDERER.
+    return;
+  }
+  gpu_info->gpu.active = false;
+  for (size_t ii = 0; ii < gpu_info->secondary_gpus.size(); ++ii)
+    gpu_info->secondary_gpus[ii].active = false;
+
+  // TODO(zmo): if two GPUs are from the same vendor, this code will always
+  // set the first GPU as active, which could be wrong.
+  if (active_vendor_id == gpu_info->gpu.vendor_id) {
+    gpu_info->gpu.active = true;
+    return;
+  }
+  for (size_t ii = 0; ii < gpu_info->secondary_gpus.size(); ++ii) {
+    if (active_vendor_id == gpu_info->secondary_gpus[ii].vendor_id) {
+      gpu_info->secondary_gpus[ii].active = true;
+      return;
+    }
+  }
+}
+
 }  // namespace gpu