Move local writer definitions to a separate header file.

[apitrace] / d3d9trace.py

diff --git a/d3d9trace.py b/d3d9trace.py
index 5b83c03837bccb07cf8fa3927f7a516af22f94c2..27fbc354a088383608f35ec8f3385f7658fe0cce 100644 (file)
--- a/d3d9trace.py
+++ b/d3d9trace.py
@@ -40,12 +40,23 @@ class D3D9Tracer(DllTracer):
 
 
 if __name__ == '__main__':
-    print '#include "trace_writer.hpp"'
+    print '#include "trace_writer_local.hpp"'
     print '#include "os.hpp"'
     print
     print '#include "d3d9imports.hpp"'
     print '#include "d3dshader.hpp"'
     print
+    print '''
+static inline size_t
+_declCount(const D3DVERTEXELEMENT9 *pVertexElements) {
+    size_t count = 0;
+    if (pVertexElements) {
+        while (pVertexElements[count++].Stream != 0xff)
+            ;
+    }
+    return count;
+}
+'''
     tracer = D3D9Tracer('d3d9.dll')
     tracer.trace_api(d3d9)