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
33 from specs.dxgi import dxgi
34 from specs.d3d10 import d3d10
35 from specs.d3d10_1 import d3d10_1
36 from specs.d3d11 import d3d11
39 class D3DRetracer(Retracer):
41 def retraceApi(self, api):
42 print '// Swizzling mapping for lock addresses'
43 print 'static std::map<void *, void *> _maps;'
46 self.table_name = 'd3dretrace::d3d10_callbacks'
48 Retracer.retraceApi(self, api)
50 createDeviceFunctionNames = [
52 "D3D10CreateDeviceAndSwapChain",
54 "D3D10CreateDeviceAndSwapChain1",
56 "D3D11CreateDeviceAndSwapChain",
59 def invokeFunction(self, function):
60 if function.name in self.createDeviceFunctionNames:
61 # create windows as neccessary
62 if 'pSwapChainDesc' in function.argNames():
63 print r' pSwapChainDesc->OutputWindow = d3dretrace::createWindow(512, 512);'
65 # Compensate for the fact we don't trace the software renderer
66 # module LoadLibrary call
67 if 'Software' in function.argNames():
68 print r' if (Software) {'
69 print r' retrace::warning(call) << "using WARP for software device\n";'
70 print r' Software = LoadLibraryA("d3d10warp");'
73 # Compensate for the fact we don't trace DXGI object creation
74 if function.name.startswith('D3D11CreateDevice'):
75 print r' if (DriverType == D3D_DRIVER_TYPE_UNKNOWN && !pAdapter) {'
76 print r' DriverType = D3D_DRIVER_TYPE_HARDWARE;'
79 Retracer.invokeFunction(self, function)
82 def invokeInterfaceMethod(self, interface, method):
83 # keep track of the last used device for state dumping
84 if interface.name in ('ID3D11DeviceContext',):
85 if method.name == 'Release':
86 print r' d3d11Dumper.unbindDevice(_this);'
88 print r' d3d11Dumper.bindDevice(_this);'
90 # create windows as neccessary
91 if method.name == 'CreateSwapChain':
92 print r' pDesc->OutputWindow = d3dretrace::createWindow(512, 512);'
94 # notify frame has been completed
95 if method.name == 'Present':
96 print r' retrace::frameComplete(call);'
98 if 'pSharedResource' in method.argNames():
99 print r' if (pSharedResource) {'
100 print r' retrace::warning(call) << "shared surfaces unsupported\n";'
101 print r' pSharedResource = NULL;'
104 Retracer.invokeInterfaceMethod(self, interface, method)
106 # process events after presents
107 if method.name == 'Present':
108 print r' d3dretrace::processEvents();'
110 if method.name == 'Map':
111 print ' VOID *_pbData = NULL;'
112 print ' size_t _MappedSize = 0;'
113 print ' _getMapInfo(_this, %s, _pbData, _MappedSize);' % ', '.join(method.argNames())
114 print ' if (_MappedSize) {'
115 print ' _maps[_this] = _pbData;'
120 if method.name == 'Unmap':
121 print ' VOID *_pbData = 0;'
122 print ' _pbData = _maps[_this];'
123 print ' if (_pbData) {'
124 print ' retrace::delRegionByPointer(_pbData);'
125 print ' _maps[_this] = 0;'
130 print r'''#include <string.h>'''
132 print r'#include <iostream>'
134 print r'#include "d3dretrace.hpp"'
137 moduleNames = sys.argv[1:]
142 if 'd3d10' in moduleNames:
143 if 'd3d10_1' in moduleNames:
144 print r'#include "d3d10_1imports.hpp"'
145 # D3D10CreateBlob is duplicated in d3d10 and d3d10_1
146 d3d10_1.functions = [function for function in d3d10_1.functions if function.name != 'D3D10CreateBlob']
147 api.addModule(d3d10_1)
149 print r'#include "d3d10imports.hpp"'
150 print r'#include "d3d10size.hpp"'
152 if 'd3d11' in moduleNames:
153 print r'#include "d3d11imports.hpp"'
154 if 'd3d11_1' in moduleNames:
155 print '#include <d3d11_1.h>'
157 print r'#include "d3d11size.hpp"'
158 print r'#include "d3dstate.hpp"'
162 print '''static d3dretrace::D3DDumper<ID3D11DeviceContext> d3d11Dumper;'''
165 retracer = D3DRetracer()
166 retracer.retraceApi(api)
169 if __name__ == '__main__':