Handle CreateDeviceEx too.

diff --git a/retrace/d3dretrace.py b/retrace/d3dretrace.py
index 194ae4439d35d3411003ca111a5c1c881f78b466..4a78e96a7206b4c7def25ca13e3ba848802b48f7 100644 (file)
--- a/retrace/d3dretrace.py
+++ b/retrace/d3dretrace.py
@@ -53,10 +53,10 @@ class D3DRetracer(Retracer):
         Retracer.extractArg(self, function, arg, arg_type, lvalue, rvalue)
 
     def invokeInterfaceMethod(self, interface, method):
-        if interface.name == 'IDirect3D9' and method.name == 'CreateDevice':
+        if method.name in ('CreateDevice', 'CreateDeviceEx'):
             print r'    HWND hWnd = createWindow(pPresentationParameters->BackBufferWidth, pPresentationParameters->BackBufferHeight);'
-            print r'    pPresentationParameters->hDeviceWindow = hWnd;'
             print r'    hFocusWindow = hWnd;'
+            print r'    pPresentationParameters->hDeviceWindow = hWnd;'
 
         Retracer.invokeInterfaceMethod(self, interface, method)