Updating the installer to check for OpenGL stats. Requires saving those stats and uploading with others which required a modification of the stats uploading code to actually save old stats--which it should do anyways.

Review URL: http://codereview.chromium.org/295043

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@31502 0039d316-1c4b-4281-b951-d872f2087c98

3 files changed, 162 insertions, 1 deletions

diff --git a/o3d/installer/win/custom_actions.cc b/o3d/installer/win/custom_actions.cc
index 0f57d95..8b716e2 100644
--- a/o3d/installer/win/custom_actions.cc
+++ b/o3d/installer/win/custom_actions.cc
@@ -40,13 +40,22 @@
 #include <strsafe.h>  // Must be after tchar.h.
 #include <windows.h>
 #include <atlstr.h>
+#include <Cg/cg.h>
+#include <Cg/cgGL.h>
+#include <GL/glext.h>
+#include <GL/wglew.h>
 
 #include "plugin/win/update_lock.h"
+#include "plugin/cross/plugin_logging.h"
+#include "plugin/cross/plugin_metrics.h"
 
 #pragma comment(linker, "/EXPORT:CheckDirectX=_CheckDirectX@4")
+#pragma comment(linker, "/EXPORT:CheckOpenGL=_CheckOpenGL@4")
 #pragma comment(linker, "/EXPORT:IsSoftwareRunning=_IsSoftwareRunning@4")
 #pragma comment(linker, "/EXPORT:InstallD3DXIfNeeded=_InstallD3DXIfNeeded@4")
 
+
+
 #if 0
 // NOTE: Useful for debugging, but not currently in use.  Left here so
 // that I don't have to figure out how to write it again.
@@ -104,6 +113,128 @@ HRESULT SetRegKeyValueDWord(HKEY hkey_parent, const TCHAR *key_name,
   return hr;
 }
 
+void ErrorAndSetUnknownGLDrivers(MSIHANDLE installer_handle, TCHAR *message) {
+  WriteToMsiLog(installer_handle, message);
+  const int gl_not_found = 99999999;
+  o3d::metric_gl_major_version = gl_not_found;
+  o3d::metric_gl_minor_version = gl_not_found;
+}
+
+// Returns true on success.
+bool GetOpenGLMetrics(MSIHANDLE installer_handle) {
+  WNDCLASS wc;
+  if (!GetClassInfo(GetModuleHandle(NULL), L"TEMPGL", &wc)) {
+    ZeroMemory(&wc, sizeof(WNDCLASS));
+    wc.hInstance = GetModuleHandle(NULL);
+    wc.lpfnWndProc = DefWindowProc;
+    wc.lpszClassName = L"TEMPGL";
+
+    if (!RegisterClass(&wc)) {
+      WriteToMsiLog(installer_handle, _T("Failed to register window class."));
+      return false;
+    }
+  }
+
+  HWND temp_hwnd = CreateWindow(L"TEMPGL", L"TEMPGL", 0, CW_USEDEFAULT,
+                                CW_USEDEFAULT, 0, 0,  // size 0 by 0
+                                NULL, NULL, GetModuleHandle(NULL), NULL);
+  if (!temp_hwnd) {
+    ErrorAndSetUnknownGLDrivers(installer_handle, _T("CreateWindow failed."));
+    return false;
+  }
+
+  // get the device context
+  HDC temp_dc = GetDC(temp_hwnd);
+  if (!temp_dc) {
+    ErrorAndSetUnknownGLDrivers(installer_handle, _T("GetDC failed."));
+    return false;
+  }
+
+  // find default pixel format
+  PIXELFORMATDESCRIPTOR pfd;
+  ZeroMemory(&pfd, sizeof(PIXELFORMATDESCRIPTOR));
+  pfd.nSize = sizeof(PIXELFORMATDESCRIPTOR);
+  pfd.nVersion = 1;
+  pfd.dwFlags = PFD_DRAW_TO_WINDOW | PFD_SUPPORT_OPENGL;
+  int pixelformat = ChoosePixelFormat(temp_dc, &pfd);
+
+  // set the pixel format for the dc
+  if (!SetPixelFormat(temp_dc, pixelformat, &pfd)) {
+    ErrorAndSetUnknownGLDrivers(installer_handle, _T("SetPixelFormat failed."));
+    return false;
+  }
+
+  // create rendering context
+  HGLRC gl_context = wglCreateContext(temp_dc);
+  if (!gl_context) {
+    ErrorAndSetUnknownGLDrivers(installer_handle,
+                                _T("wglCreateContext failed."));
+    return false;
+  }
+
+  if (!wglMakeCurrent(temp_dc, gl_context)) {
+    ErrorAndSetUnknownGLDrivers(installer_handle, _T("wglMakeCurrent failed."));
+    return false;
+  }
+
+  const char *gl_version_string =
+      reinterpret_cast<const char*>(glGetString(GL_VERSION));
+  const char *gl_extensions_string =
+      reinterpret_cast<const char*>(glGetString(GL_EXTENSIONS));
+  if ((gl_version_string == NULL) || (gl_extensions_string == NULL)) {
+    ErrorAndSetUnknownGLDrivers(installer_handle, _T("No GL found."));
+    return true;
+  }
+
+  // Get the OpenGL version from the start of the string.
+  int gl_major = 0, gl_minor = 0;
+  sscanf(gl_version_string, "%u.%u",  &gl_major, &gl_minor);
+  o3d::metric_gl_major_version = gl_major;
+  o3d::metric_gl_minor_version = gl_minor;
+
+  // Get the HLSL version.
+  // On OpenGL 1.x it's 1.0 if the GL_ARB_shading_language_100 extension is
+  // present.
+  // On OpenGL 2.x  it's a matter of getting the GL_SHADING_LANGUAGE_VERSION
+  // string.
+  int gl_hlsl_major = 0, gl_hlsl_minor = 0;
+  if ((gl_major == 1) &&
+      strstr(gl_extensions_string, "GL_ARB_shading_language_100")) {
+    gl_hlsl_major = 1;
+    gl_hlsl_minor = 0;
+  } else if (gl_major >= 2) {
+    const char* glsl_version_string =
+        reinterpret_cast<const char*>(glGetString(GL_SHADING_LANGUAGE_VERSION));
+    if (glsl_version_string) {
+      sscanf(glsl_version_string, "%u.%u", &gl_hlsl_major, &gl_hlsl_minor);
+    }
+  }
+  o3d::metric_gl_hlsl_major_version = gl_hlsl_major;
+  o3d::metric_gl_hlsl_minor_version = gl_hlsl_minor;
+
+  // Clean up
+  wglDeleteContext(gl_context);
+  ReleaseDC(temp_hwnd, temp_dc);
+  DestroyWindow(temp_hwnd);
+  UnregisterClass(L"TEMPGL", wc.hInstance);
+  return true;
+}
+
+bool GetOpenGLVersion(MSIHANDLE installer_handle) {
+  HRESULT hr = CoInitialize(NULL);
+  o3d::PluginLogging g_logger;
+  stats_report::g_global_metrics.Initialize();
+  // Get OpenGL stats logged
+  if (!GetOpenGLMetrics(installer_handle)) {
+    return false;
+  }
+  if (!g_logger.ProcessMetrics(true, false, false)) {
+    return false;
+  }
+  stats_report::g_global_metrics.Uninitialize();
+  return true;
+}
+
 // Retrieve the currently installed version of DirectX using a COM
 // DxDiagProvider.  Returns 0 on error.
 DWORD GetDirectXVersion() {
@@ -276,6 +407,14 @@ extern "C" UINT __stdcall CheckDirectX(MSIHANDLE installer_handle) {
   return ERROR_SUCCESS;
 }
 
+// Check the version of OpenGL installed and save a registry key
+extern "C" UINT __stdcall CheckOpenGL(MSIHANDLE installer_handle) {
+  if (!GetOpenGLVersion(installer_handle)) {
+    WriteToMsiLog(installer_handle, _T("GetOpenGLVersion failed!"));
+  }
+  return ERROR_SUCCESS;
+}
+
 // Check to see whether the plugin is currently running.  If it is, we can't
 // update the plugin.  The installer will check for the SOFTWARE_RUNNING flag
 // and exit if it's trying to do a silent update.  Knowing that it's failed this
diff --git a/o3d/installer/win/installer.gyp b/o3d/installer/win/installer.gyp
index ac6f183..d13232f 100644
--- a/o3d/installer/win/installer.gyp
+++ b/o3d/installer/win/installer.gyp
@@ -44,11 +44,20 @@
       'sources': [
         'custom_actions.cc',
       ],
+      'dependencies': [
+        '../../../base/base.gyp:base',
+        '../../build/libs.gyp:cg_libs',
+        '../../build/libs.gyp:gl_libs',
+        '../../plugin/plugin.gyp:o3dPluginLogging',
+        '../../statsreport/statsreport.gyp:o3dStatsReport',
+      ],
       'include_dirs': [
-        '$(DXSDK_DIR)/Include',
         '../..',
         '../../..',
+        '../../<(glewdir)/include',
+        '../../<(cgdir)/include',
         '<(INTERMEDIATE_DIR)',
+        '$(DXSDK_DIR)/Include',
       ],
       'defines': [
         'NOMINMAX',
@@ -127,6 +136,8 @@
         '../../plugin/plugin.gyp:npo3dautoplugin',
         '../../plugin/plugin.gyp:o3d_host',
         '../../samples/samples.gyp:samples',
+        '../../build/libs.gyp:cg_libs',
+        '../../build/libs.gyp:gl_libs',
         'cactions',
       ],
       'rules': [
diff --git a/o3d/installer/win/o3d.wxs b/o3d/installer/win/o3d.wxs
index efca8d9..9fe28e9 100644
--- a/o3d/installer/win/o3d.wxs
+++ b/o3d/installer/win/o3d.wxs
@@ -313,6 +313,11 @@ Wix script for building o3d installer.
         Id='CheckDirectX'
         BinaryKey='CustomActions'
         DllEntry='CheckDirectX' />
+    <!-- Custom action for detecting OpenGL Version. -->
+    <CustomAction
+        Id='CheckOpenGL'
+        BinaryKey='CustomActions'
+        DllEntry='CheckOpenGL' />
 
     <!-- Custom action for checking that we're not already running. -->
     <CustomAction
@@ -343,6 +348,9 @@ Wix script for building o3d installer.
       <Custom
           Action='CheckDirectX'
           Before='LaunchConditions' />
+      <Custom
+          Action='CheckOpenGL'
+          Before='CheckDirectX' />
     </InstallUISequence>
 
     <InstallExecuteSequence>
@@ -356,6 +364,9 @@ Wix script for building o3d installer.
           Action='CheckDirectX'
           Before='LaunchConditions' />
       <Custom
+          Action='CheckOpenGL'
+          Before='CheckDirectX' />
+      <Custom
           Action='InstallD3DXIfNeeded'
           After='InstallFiles'>
         NOT Installed