From 4bb88c4a45707fb88150aa7952b4c9b9f55b1749 Mon Sep 17 00:00:00 2001
From: Kai-Hwa Yao <kyao@nvidia.com>
Date: Sun, 9 Jul 2017 22:58:41 -0700
Subject: Allow glslang wrapper to output regular SPIRV before disassembly

---
 tools/glslang/glslang.cpp | 34 ++++++++++++++++++++--------------
 1 file changed, 20 insertions(+), 14 deletions(-)

(limited to 'tools/glslang/glslang.cpp')

diff --git a/tools/glslang/glslang.cpp b/tools/glslang/glslang.cpp
index cf563faff..fb697675e 100644
--- a/tools/glslang/glslang.cpp
+++ b/tools/glslang/glslang.cpp
@@ -51,21 +51,22 @@ static TBuiltInResource gResources =
 };
 
 static void dump(
-    std::string const&      text,
-    glslang_OutputFunc      outputFunc,
-    void*                   outputUserData,
-    FILE*                   fallbackStream)
+    void const*         data,
+    size_t              size,
+    glslang_OutputFunc  outputFunc,
+    void*               outputUserData,
+    FILE*               fallbackStream)
 {
     if( outputFunc )
     {
-        outputFunc(text.c_str(), outputUserData);
+        outputFunc(data, size, outputUserData);
     }
     else
     {
-        fprintf(fallbackStream, "%s", text.c_str());
+        fwrite(data, 1, size, fallbackStream);
 
         // also output it for debug purposes
-        OutputDebugStringA(text.c_str());
+        OutputDebugStringA((char const*)data);
     }
 }
 
@@ -73,7 +74,7 @@ static void dumpDiagnostics(
     glslang_CompileRequest* request,
     std::string const&      log)
 {
-    dump(log, request->diagnosticFunc, request->diagnosticUserData, stderr);
+    dump(log.c_str(), log.length(), request->diagnosticFunc, request->diagnosticUserData, stderr);
 }
 
 extern "C"
@@ -161,14 +162,19 @@ int glslang_compile(glslang_CompileRequest* request)
 
         dumpDiagnostics(request, logger.getAllMessages());
 
-        std::stringstream spirvAsmStream;
-
-        spv::Disassemble(spirvAsmStream, spirv);
-
-        dump(spirvAsmStream.str(), request->outputFunc, request->outputUserData, stdout);
+        if (request->disassembleResult)
+        {
+            std::stringstream spirvAsmStream;
+            spv::Disassemble(spirvAsmStream, spirv);
+            std::string result = spirvAsmStream.str();
+            dump(result.c_str(), result.length(), request->outputFunc, request->outputUserData, stdout);
+        }
+        else
+        {
+            dump(spirv.data(), spirv.size() * sizeof(spirv[0]), request->outputFunc, request->outputUserData, stdout);
+        }
     }
 
-
     glslang::FinalizeProcess();
 
     return 0;
-- 
cgit v1.2.3