##########################################################################/
-from trace import DllTracer
+from dlltrace import DllTracer
from specs.d3d9 import d3d9
class D3D9Tracer(DllTracer):
- def dump_arg_instance(self, function, arg):
+ def serializeArgValue(self, function, arg):
# Dump shaders as strings
if function.name in ('CreateVertexShader', 'CreatePixelShader') and arg.name == 'pFunction':
print ' DumpShader(trace::localWriter, %s);' % (arg.name)
return
- DllTracer.dump_arg_instance(self, function, arg)
+ DllTracer.serializeArgValue(self, function, arg)
+
+ def declareWrapperInterfaceVariables(self, interface):
+ DllTracer.declareWrapperInterfaceVariables(self, interface)
+
+ if interface.name == 'IDirect3DVertexBuffer9':
+ print ' UINT m_SizeToLock;'
+ print ' VOID *m_pbData;'
+
+ def implementWrapperInterfaceMethodBody(self, interface, base, method):
+ if interface.name == 'IDirect3DVertexBuffer9' and method.name == 'Unlock':
+ print ' if (m_pbData) {'
+ self.emit_memcpy('(LPBYTE)m_pbData', '(LPBYTE)m_pbData', 'm_SizeToLock')
+ print ' }'
+
+ DllTracer.implementWrapperInterfaceMethodBody(self, interface, base, method)
+
+ if interface.name == 'IDirect3DVertexBuffer9' and method.name == 'Lock':
+ # FIXME: handle recursive locks
+ print ' if (__result == D3D_OK && !(Flags & D3DLOCK_READONLY)) {'
+ print ' if (SizeToLock) {'
+ print ' m_SizeToLock = SizeToLock;'
+ print ' } else {'
+ print ' D3DVERTEXBUFFER_DESC Desc;'
+ print ' m_pInstance->GetDesc(&Desc);'
+ print ' m_SizeToLock = Desc.Size;'
+ print ' }'
+ print ' m_pbData = *ppbData;'
+ print ' } else {'
+ print ' m_pbData = NULL;'
+ print ' }'
if __name__ == '__main__':