1 ##########################################################################
3 # Copyright 2011 Jose Fonseca
6 # Permission is hereby granted, free of charge, to any person obtaining a copy
7 # of this software and associated documentation files (the "Software"), to deal
8 # in the Software without restriction, including without limitation the rights
9 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 # copies of the Software, and to permit persons to whom the Software is
11 # furnished to do so, subject to the following conditions:
13 # The above copyright notice and this permission notice shall be included in
14 # all copies or substantial portions of the Software.
16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
24 ##########################################################################/
27 """D3D retracer generator."""
30 from dllretrace import DllRetracer as Retracer
31 import specs.stdapi as stdapi
32 from specs.d3d9 import *
35 class D3DRetracer(Retracer):
37 table_name = 'd3dretrace::d3d9_callbacks'
39 bufferInterfaceNames = [
40 'IDirect3DVertexBuffer9',
41 'IDirect3DIndexBuffer9',
44 def invokeInterfaceMethod(self, interface, method):
45 # keep track of the last used device for state dumping
46 if interface.name in ('IDirect3DDevice9', 'IDirect3DDevice9Ex'):
47 print r' d3dretrace::pLastDirect3DDevice9 = _this;'
49 # create windows as neccessary
50 if method.name in ('CreateDevice', 'CreateDeviceEx'):
51 print r' HWND hWnd = createWindow(pPresentationParameters->BackBufferWidth, pPresentationParameters->BackBufferHeight);'
52 print r' hFocusWindow = hWnd;'
53 print r' pPresentationParameters->hDeviceWindow = hWnd;'
55 # notify frame has been completed
56 if method.name == 'Present':
57 print r' retrace::frameComplete(call);'
59 if 'pSharedHandle' in method.argNames():
60 print r' if (pSharedHandle) {'
61 print r' retrace::warning(call) << "shared surfaces unsupported\n";'
62 print r' pSharedHandle = NULL;'
65 Retracer.invokeInterfaceMethod(self, interface, method)
68 if str(method.type) == 'HRESULT':
69 print r' if (FAILED(_result)) {'
70 print r' retrace::warning(call) << "failed\n";'
73 if method.name in ('Lock', 'LockRect', 'LockBox'):
74 print ' size_t _LockedSize = _getLockSize(_this, %s);' % ', '.join(method.argNames()[:-1])
75 if method.name == 'Lock':
76 # FIXME: handle recursive locks
77 print ' VOID *_pbData = *ppbData;'
78 elif method.name == 'LockRect':
79 print ' VOID *_pbData = pLockedRect->pBits;'
80 elif method.name == 'LockBox':
81 print ' VOID *_pbData = pLockedVolume->pBits;'
83 raise NotImplementedError
84 print ' _this->SetPrivateData(GUID_APITRACE, &_pbData, sizeof _pbData, 0);'
86 if method.name in ('Unlock', 'UnlockRect', 'UnlockBox'):
87 print ' VOID *_pbData = 0;'
88 print ' DWORD dwSizeOfData = sizeof _pbData;'
89 print ' _this->GetPrivateData(GUID_APITRACE, &_pbData, &dwSizeOfData);'
90 print ' if (_pbData) {'
91 print ' retrace::delRegionByPointer(_pbData);'
95 if __name__ == '__main__':
103 #include "d3d9imports.hpp"
104 #include "d3dsize.hpp"
105 #include "d3dretrace.hpp"
108 // XXX: Don't duplicate this code.
110 static LRESULT CALLBACK
111 WndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
115 case WM_GETMINMAXINFO:
116 // Allow to create a window bigger than the desktop
117 pMMI = (MINMAXINFO *)lParam;
118 pMMI->ptMaxSize.x = 60000;
119 pMMI->ptMaxSize.y = 60000;
120 pMMI->ptMaxTrackSize.x = 60000;
121 pMMI->ptMaxTrackSize.y = 60000;
127 return DefWindowProc(hWnd, uMsg, wParam, lParam);
131 DEFINE_GUID(GUID_APITRACE,0X7D71CAC9,0X7F58,0X432C,0XA9,0X75,0XA1,0X9F,0XCF,0XCE,0XFD,0X14);
135 createWindow(int width, int height) {
136 static bool first = TRUE;
141 memset(&wc, 0, sizeof wc);
142 wc.hbrBackground = (HBRUSH) (COLOR_BTNFACE + 1);
143 wc.hCursor = LoadCursor(NULL, IDC_ARROW);
144 wc.hIcon = LoadIcon(NULL, IDI_APPLICATION);
145 wc.lpfnWndProc = WndProc;
146 wc.lpszClassName = "d3dretrace";
147 wc.style = CS_OWNDC | CS_HREDRAW | CS_VREDRAW;
157 dwStyle = WS_CLIPSIBLINGS | WS_CLIPCHILDREN | WS_OVERLAPPEDWINDOW;
163 rect.right = rect.left + width;
164 rect.bottom = rect.top + height;
166 AdjustWindowRectEx(&rect, dwStyle, FALSE, dwExStyle);
168 hWnd = CreateWindowEx(dwExStyle,
169 "d3dretrace", /* wc.lpszClassName */
174 rect.right - rect.left, /* width */
175 rect.bottom - rect.top, /* height */
180 ShowWindow(hWnd, SW_SHOW);
187 (WINAPI *PD3DXASSEMBLESHADER)(
190 const D3DXMACRO *pDefines,
191 LPD3DXINCLUDE pInclude,
193 LPD3DXBUFFER *ppShader,
194 LPD3DXBUFFER *ppErrorMsgs
198 extractShader(LPCSTR pSrcData)
200 static BOOL firsttime = TRUE;
201 static HMODULE hD3DXModule = NULL;
202 static PD3DXASSEMBLESHADER pfnD3DXAssembleShader = NULL;
208 for (release = 0; release <= 1; ++release) {
209 /* Version 41 corresponds to Mar 2009 version of DirectX Runtime / SDK */
210 for (version = 41; version >= 0; --version) {
212 _snprintf(filename, sizeof(filename),
213 "d3dx9%s%s%u.dll", release ? "" : "d", version ? "_" : "", version);
214 hD3DXModule = LoadLibraryA(filename);
224 if (!pfnD3DXAssembleShader) {
225 pfnD3DXAssembleShader = (PD3DXASSEMBLESHADER)GetProcAddress(hD3DXModule, "D3DXAssembleShader");
232 if (pfnD3DXAssembleShader) {
233 LPD3DXBUFFER pTokens = NULL;
236 hr = pfnD3DXAssembleShader(pSrcData, strlen(pSrcData), NULL, NULL, 0, &pTokens, NULL);
238 return (DWORD *)pTokens->GetBufferPointer();
241 // FIXME: Don't leak pTokens
248 retracer = D3DRetracer()
249 retracer.retraceApi(d3d9)