]> git.notmuchmail.org Git - apitrace/blob - retrace/d3d9retrace.py
db92377e624b8895d0aed87f057352902119bc0b
[apitrace] / retrace / d3d9retrace.py
1 ##########################################################################
2 #
3 # Copyright 2011 Jose Fonseca
4 # All Rights Reserved.
5 #
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:
12 #
13 # The above copyright notice and this permission notice shall be included in
14 # all copies or substantial portions of the Software.
15 #
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
22 # THE SOFTWARE.
23 #
24 ##########################################################################/
25
26
27 """D3D retracer generator."""
28
29
30 import sys
31 from dllretrace import DllRetracer as Retracer
32 from specs.stdapi import API
33
34
35 class D3DRetracer(Retracer):
36
37     def retraceApi(self, api):
38         print '// Swizzling mapping for lock addresses'
39         print 'static std::map<void *, void *> _maps;'
40         print
41
42         Retracer.retraceApi(self, api)
43
44     def invokeFunction(self, function):
45         if function.name in ('Direct3DCreate9', 'Direct3DCreate9Ex'):
46             print 'if (retrace::debug && !g_szD3D9DllName) {'
47             print '    /* '
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.'
50             print '     */'
51             print '    if (0) {'
52             print '        g_szD3D9DllName = "d3d9d.dll";'
53             print '    }'
54             print '}'
55
56         Retracer.invokeFunction(self, function)
57
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'    d3d9Dumper.unbindDevice(_this);'
63             else:
64                 print r'    d3d9Dumper.bindDevice(_this);'
65         if interface.name in ('IDirect3DDevice8', 'IDirect3DDevice8Ex'):
66             if method.name == 'Release':
67                 print r'    d3d8Dumper.unbindDevice(_this);'
68             else:
69                 print r'    d3d8Dumper.bindDevice(_this);'
70
71         # create windows as neccessary
72         if method.name in ('CreateDevice', 'CreateDeviceEx', 'CreateAdditionalSwapChain'):
73             print r'    HWND hWnd = d3dretrace::createWindow(pPresentationParameters->BackBufferWidth, pPresentationParameters->BackBufferHeight);'
74             print r'    pPresentationParameters->hDeviceWindow = hWnd;'
75             if 'hFocusWindow' in method.argNames():
76                 print r'    hFocusWindow = hWnd;'
77         
78         if method.name in ('CreateDevice', 'CreateDeviceEx'):
79             print r'    switch (retrace::driver) {'
80             print r'    case retrace::DRIVER_HARDWARE:'
81             print r'        DeviceType = D3DDEVTYPE_HAL;'
82             print r'        break;'
83             print r'    case retrace::DRIVER_SOFTWARE:'
84             print r'    case retrace::DRIVER_REFERENCE:'
85             print r'        DeviceType = D3DDEVTYPE_REF;'
86             print r'        break;'
87             print r'    case retrace::DRIVER_NULL:'
88             if interface.name.startswith('IDirect3D9'):
89                 print r'        DeviceType = D3DDEVTYPE_NULLREF;'
90             else:
91                 print r'        retrace::warning(call) << "null driver not supported\n";'
92             print r'        break;'
93             print r'    case retrace::DRIVER_MODULE:'
94             print r'        retrace::warning(call) << "driver module not supported\n";'
95             print r'        break;'
96             print r'    default:'
97             print r'        assert(0);'
98             print r'        /* fall-through */'
99             print r'    case retrace::DRIVER_DEFAULT:'
100             print r'        break;'
101             print r'    }'
102
103         if method.name in ('Reset', 'ResetEx'):
104             print r'    if (pPresentationParameters->Windowed) {'
105             print r'        d3dretrace::resizeWindow(pPresentationParameters->hDeviceWindow, pPresentationParameters->BackBufferWidth, pPresentationParameters->BackBufferHeight);'
106             print r'    }'
107
108         # notify frame has been completed
109         if method.name == 'Present':
110             print r'    retrace::frameComplete(call);'
111             print r'    hDestWindowOverride = NULL;'
112
113         if 'pSharedHandle' in method.argNames():
114             print r'    if (pSharedHandle) {'
115             print r'        retrace::warning(call) << "shared surfaces unsupported\n";'
116             print r'        pSharedHandle = NULL;'
117             print r'    }'
118
119         Retracer.invokeInterfaceMethod(self, interface, method)
120
121         # process events after presents
122         if method.name == 'Present':
123             print r'    d3dretrace::processEvents();'
124             print r'    Sleep(500);'
125
126         if method.name in ('Lock', 'LockRect', 'LockBox'):
127             print '    VOID *_pbData = NULL;'
128             print '    size_t _MappedSize = 0;'
129             print '    _getMapInfo(_this, %s, _pbData, _MappedSize);' % ', '.join(method.argNames()[:-1])
130             print '    if (_MappedSize) {'
131             print '        _maps[_this] = _pbData;'
132             print '    } else {'
133             print '        return;'
134             print '    }'
135         
136         if method.name in ('Unlock', 'UnlockRect', 'UnlockBox'):
137             print '    VOID *_pbData = 0;'
138             print '    _pbData = _maps[_this];'
139             print '    if (_pbData) {'
140             print '        retrace::delRegionByPointer(_pbData);'
141             print '        _maps[_this] = 0;'
142             print '    }'
143
144
145 def main():
146     print r'#include <string.h>'
147     print
148     print r'#include <iostream>'
149     print
150     print r'#include "d3dretrace.hpp"'
151     print
152
153     moduleName = sys.argv[1]
154     support = int(sys.argv[2])
155
156     api = API()
157     
158     if support:
159         if moduleName == 'd3d9':
160             from specs.d3d9 import d3d9
161             print r'#include "d3d9imports.hpp"'
162             print r'#include "d3d9size.hpp"'
163             api.addModule(d3d9)
164             print
165             print '''static d3dretrace::D3DDumper<IDirect3DDevice9> d3d9Dumper;'''
166             print
167         elif moduleName == 'd3d8':
168             from specs.d3d8 import d3d8
169             print r'#include "d3d8imports.hpp"'
170             print r'#include "d3d8size.hpp"'
171             api.addModule(d3d8)
172             print
173             print '''static d3dretrace::D3DDumper<IDirect3DDevice8> d3d8Dumper;'''
174             print
175         else:
176             assert False
177
178     retracer = D3DRetracer()
179     retracer.table_name = 'd3dretrace::%s_callbacks' % moduleName
180     retracer.retraceApi(api)
181
182
183 if __name__ == '__main__':
184     main()