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."""
31 from dllretrace import DllRetracer as Retracer
32 from specs.stdapi import API
35 class D3DRetracer(Retracer):
37 def retraceApi(self, api):
38 print '// Swizzling mapping for lock addresses'
39 print 'static std::map<void *, void *> _maps;'
42 Retracer.retraceApi(self, api)
44 def invokeFunction(self, function):
45 if function.name in ('Direct3DCreate9', 'Direct3DCreate9Ex'):
46 print 'if (retrace::debug && !g_szD3D9DllName) {'
48 print ' * XXX: D3D9D only works for simple things, it often introduces errors'
49 print ' * on complex traces, or traces which use unofficial D3D9 features.'
52 print ' g_szD3D9DllName = "d3d9d.dll";'
56 Retracer.invokeFunction(self, function)
58 def invokeInterfaceMethod(self, interface, method):
59 # keep track of the last used device for state dumping
60 if interface.name in ('IDirect3DDevice9', 'IDirect3DDevice9Ex'):
61 if method.name == 'Release':
62 print r' if (call.ret->toUInt() == 0) {'
63 print r' d3d9Dumper.unbindDevice(_this);'
66 print r' d3d9Dumper.bindDevice(_this);'
67 if interface.name in ('IDirect3DDevice8', 'IDirect3DDevice8Ex'):
68 if method.name == 'Release':
69 print r' if (call.ret->toUInt() == 0) {'
70 print r' d3d8Dumper.unbindDevice(_this);'
73 print r' d3d8Dumper.bindDevice(_this);'
75 # create windows as neccessary
76 if method.name in ('CreateDevice', 'CreateDeviceEx', 'CreateAdditionalSwapChain'):
77 print r' HWND hWnd = d3dretrace::createWindow(pPresentationParameters->BackBufferWidth, pPresentationParameters->BackBufferHeight);'
78 print r' pPresentationParameters->hDeviceWindow = hWnd;'
79 if 'hFocusWindow' in method.argNames():
80 print r' hFocusWindow = hWnd;'
82 if method.name in ('CreateDevice', 'CreateDeviceEx'):
83 print r' switch (retrace::driver) {'
84 print r' case retrace::DRIVER_HARDWARE:'
85 print r' DeviceType = D3DDEVTYPE_HAL;'
87 print r' case retrace::DRIVER_SOFTWARE:'
88 print r' case retrace::DRIVER_REFERENCE:'
89 print r' DeviceType = D3DDEVTYPE_REF;'
91 print r' case retrace::DRIVER_NULL:'
92 if interface.name.startswith('IDirect3D9'):
93 print r' DeviceType = D3DDEVTYPE_NULLREF;'
95 print r' retrace::warning(call) << "null driver not supported\n";'
97 print r' case retrace::DRIVER_MODULE:'
98 print r' retrace::warning(call) << "driver module not supported\n";'
102 print r' /* fall-through */'
103 print r' case retrace::DRIVER_DEFAULT:'
107 if method.name in ('Reset', 'ResetEx'):
108 print r' if (pPresentationParameters->Windowed) {'
109 print r' d3dretrace::resizeWindow(pPresentationParameters->hDeviceWindow, pPresentationParameters->BackBufferWidth, pPresentationParameters->BackBufferHeight);'
112 # notify frame has been completed
113 if method.name in ('Present', 'PresentEx'):
114 print r' retrace::frameComplete(call);'
115 print r' hDestWindowOverride = NULL;'
117 if 'pSharedHandle' in method.argNames():
118 print r' if (pSharedHandle) {'
119 print r' retrace::warning(call) << "shared surfaces unsupported\n";'
120 print r' pSharedHandle = NULL;'
123 Retracer.invokeInterfaceMethod(self, interface, method)
125 # process events after presents
126 if method.name == 'Present':
127 print r' d3dretrace::processEvents();'
129 if method.name in ('Lock', 'LockRect', 'LockBox'):
130 print ' VOID *_pbData = NULL;'
131 print ' size_t _MappedSize = 0;'
132 print ' _getMapInfo(_this, %s, _pbData, _MappedSize);' % ', '.join(method.argNames()[:-1])
133 print ' if (_MappedSize) {'
134 print ' _maps[_this] = _pbData;'
139 if method.name in ('Unlock', 'UnlockRect', 'UnlockBox'):
140 print ' VOID *_pbData = 0;'
141 print ' _pbData = _maps[_this];'
142 print ' if (_pbData) {'
143 print ' retrace::delRegionByPointer(_pbData);'
144 print ' _maps[_this] = 0;'
149 print r'#include <string.h>'
151 print r'#include <iostream>'
153 print r'#include "d3dretrace.hpp"'
156 moduleName = sys.argv[1]
157 support = int(sys.argv[2])
162 if moduleName == 'd3d9':
163 from specs.d3d9 import d3d9
164 print r'#include "d3d9imports.hpp"'
165 print r'#include "d3d9size.hpp"'
168 print '''static d3dretrace::D3DDumper<IDirect3DDevice9> d3d9Dumper;'''
170 elif moduleName == 'd3d8':
171 from specs.d3d8 import d3d8
172 print r'#include "d3d8imports.hpp"'
173 print r'#include "d3d8size.hpp"'
176 print '''static d3dretrace::D3DDumper<IDirect3DDevice8> d3d8Dumper;'''
181 retracer = D3DRetracer()
182 retracer.table_name = 'd3dretrace::%s_callbacks' % moduleName
183 retracer.retraceApi(api)
186 if __name__ == '__main__':