From 889d32c7208736694073136f62084910e375d2a8 Mon Sep 17 00:00:00 2001
From: =?utf8?q?Jos=C3=A9=20Fonseca?= <jose.r.fonseca@gmail.com>
Date: Mon, 23 Apr 2012 10:18:28 +0100
Subject: [PATCH] Detect/handle more APIs in CLI/GUI.

---
 cli/cli_trace.cpp            |  4 ++++
 common/trace_api.hpp         |  3 +++
 common/trace_parser.cpp      | 12 ++++++++----
 common/trace_tools_trace.cpp |  6 ++++++
 gui/retracer.cpp             | 23 ++++++++++++++++++++---
 gui/tracedialog.cpp          |  2 ++
 6 files changed, 43 insertions(+), 7 deletions(-)

diff --git a/cli/cli_trace.cpp b/cli/cli_trace.cpp
index a67416b..8077509 100644
--- a/cli/cli_trace.cpp
+++ b/cli/cli_trace.cpp
@@ -104,6 +104,10 @@ command(int argc, char *argv[])
                 api = trace::API_D3D9;
             } else if (strcmp(optarg, "d3d10") == 0) {
                 api = trace::API_D3D10;
+            } else if (strcmp(optarg, "d3d10_1") == 0) {
+                api = trace::API_D3D10_1;
+            } else if (strcmp(optarg, "d3d11") == 0) {
+                api = trace::API_D3D11;
             } else {
                 std::cerr << "error: unknown API `" << optarg << "`\n";
                 usage();
diff --git a/common/trace_api.hpp b/common/trace_api.hpp
index 7619f71..4c896db 100644
--- a/common/trace_api.hpp
+++ b/common/trace_api.hpp
@@ -44,10 +44,13 @@ enum API {
     API_UNKNOWN = 0,
     API_GL, // GL + GLX/WGL/CGL
     API_EGL, // GL/GLES1/GLES2/VG + EGL
+    API_DX, // All DirectX
     API_D3D7,
     API_D3D8,
     API_D3D9,
     API_D3D10,
+    API_D3D10_1,
+    API_D3D11,
 };
 
 
diff --git a/common/trace_parser.cpp b/common/trace_parser.cpp
index 11ec229..0e4fba0 100644
--- a/common/trace_parser.cpp
+++ b/common/trace_parser.cpp
@@ -242,12 +242,16 @@ Parser::parse_function_sig(void) {
          */
         if (api == API_UNKNOWN) {
             const char *n = sig->name;
-            if ((n[0] == 'g' && n[1] == 'l' && n[2] == 'X') || // glX
-                (n[0] == 'w' && n[1] == 'g' && n[2] == 'l' && n[3] >= 'A' && n[3] <= 'Z') || // wgl[A-Z]
-                (n[0] == 'C' && n[1] == 'G' && n[2] == 'L')) { // CGL
+            if ((n[0] == 'g' && n[1] == 'l' && n[2] == 'X') || // glX*
+                (n[0] == 'w' && n[1] == 'g' && n[2] == 'l' && n[3] >= 'A' && n[3] <= 'Z') || // wgl[A-Z]*
+                (n[0] == 'C' && n[1] == 'G' && n[2] == 'L')) { // CGL*
                 api = trace::API_GL;
-            } else if (n[0] == 'e' && n[1] == 'g' && n[2] == 'l' && n[3] >= 'A' && n[3] <= 'Z') { // egl
+            } else if (n[0] == 'e' && n[1] == 'g' && n[2] == 'l' && n[3] >= 'A' && n[3] <= 'Z') { // egl[A-Z]*
                 api = trace::API_EGL;
+            } else if (n[0] == 'D' &&
+                       ((n[1] == 'i' && n[2] == 'r' && n[3] == 'e' && n[4] == 'c' && n[5] == 't') || // Direct*
+                        (n[1] == '3' && n[2] == 'D'))) { // D3D*
+                api = trace::API_DX;
             } else {
                 /* TODO */
             }
diff --git a/common/trace_tools_trace.cpp b/common/trace_tools_trace.cpp
index 36fe735..ad355ab 100644
--- a/common/trace_tools_trace.cpp
+++ b/common/trace_tools_trace.cpp
@@ -129,6 +129,12 @@ traceProgram(API api,
     case API_D3D10:
         wrapperFilename = "d3d10.dll";
         break;
+    case API_D3D10_1:
+        wrapperFilename = "d3d10_1.dll";
+        break;
+    case API_D3D11:
+        wrapperFilename = "d3d11.dll";
+        break;
 #endif
     default:
         std::cerr << "error: unsupported API\n";
diff --git a/gui/retracer.cpp b/gui/retracer.cpp
index 983cd1a..69dca6c 100644
--- a/gui/retracer.cpp
+++ b/gui/retracer.cpp
@@ -229,11 +229,28 @@ void Retracer::run()
     QString prog;
     QStringList arguments;
 
-    if (m_api == trace::API_GL) {
+    switch (m_api) {
+    case trace::API_GL:
         prog = QLatin1String("glretrace");
-    } else if (m_api == trace::API_EGL) {
+        break;
+    case trace::API_EGL:
         prog = QLatin1String("eglretrace");
-    } else {
+        break;
+    case trace::API_DX:
+    case trace::API_D3D7:
+    case trace::API_D3D8:
+    case trace::API_D3D9:
+    case trace::API_D3D10:
+    case trace::API_D3D10_1:
+    case trace::API_D3D11:
+#ifdef Q_OS_WIN
+        prog = QLatin1String("d3dretrace");
+#else
+        prog = QLatin1String("wine");
+        arguments << QLatin1String("d3dretrace.exe");
+#endif
+        break;
+    default:
         emit finished(QLatin1String("Unsupported API"));
         return;
     }
diff --git a/gui/tracedialog.cpp b/gui/tracedialog.cpp
index 1506707..fcfdf46 100644
--- a/gui/tracedialog.cpp
+++ b/gui/tracedialog.cpp
@@ -15,6 +15,8 @@ TraceDialog::TraceDialog(QWidget *parent)
     apiComboBox->addItem("D3D8");
     apiComboBox->addItem("D3D9");
     apiComboBox->addItem("D3D10");
+    apiComboBox->addItem("D3D10_1");
+    apiComboBox->addItem("D3D11");
 #else
     apiComboBox->addItem("EGL");
 #endif
-- 
2.43.0