Recognize zero SizeToLock as full buffer.

diff --git a/d3dretrace.py b/d3dretrace.py
index e8b63b0f675bbbcf9e9a4b2abba67039aa203e8c..a2ea75e24c8812bc3cd35f98e07d51f97f941e86 100644 (file)
--- a/d3dretrace.py
+++ b/d3dretrace.py
@@ -49,6 +49,13 @@ class D3DRetracer(Retracer):
             print r'        retrace::warning(call) << "failed\n";'
             print r'    }'
 
+        if interface.name == 'IDirect3DVertexBuffer9' and method.name == 'Lock':
+            print '        if (!SizeToLock) {'
+            print '            D3DVERTEXBUFFER_DESC Desc;'
+            print '            _this->GetDesc(&Desc);'
+            print '            SizeToLock = Desc.Size;'
+            print '        }'
+
 
 if __name__ == '__main__':
     print r'''