Dump D3D9 shaders as text, as done previously.

1 files changed, 7 insertions, 1 deletions

diff --git a/d3d9.py b/d3d9.py
index 81d310aa..6db95361 100644
--- a/d3d9.py
+++ b/d3d9.py
@@ -406,12 +406,18 @@ d3d9.add_functions([
 
 class D3D9Tracer(DllTracer):
 
-    pass
+    def dump_arg_instance(self, function, arg):
+        # Dump shaders as strings
+        if function.name in ('CreateVertexShader', 'CreatePixelShader') and arg.name == 'pFunction':
+            print '    DumpShader(%s);' % (arg.name)
+            return
 
+        DllTracer.dump_arg_instance(self, function, arg)
 
 
 if __name__ == '__main__':
     print '#include "d3d9imports.hpp"'
+    print '#include "d3d9shader.hpp"'
     print
     print '#include "trace_write.hpp"'
     print '#include "os.hpp"'