]> git.cworth.org Git - apitrace/blob - retrace/d3dstate.cpp
Simplify TLS.
[apitrace] / retrace / d3dstate.cpp
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 #include <stdio.h>
28
29 #include <iostream>
30
31 #include "d3d9imports.hpp"
32 #include "json.hpp"
33
34
35 namespace d3dstate {
36
37
38 typedef HRESULT
39 (WINAPI *PD3DXDISASSEMBLESHADER)(
40     CONST DWORD *pShader,
41     BOOL EnableColorCode,
42     LPCSTR pComments,
43     LPD3DXBUFFER *ppDisassembly
44 );
45
46
47 HRESULT
48 disassembleShader(const DWORD *tokens, LPD3DXBUFFER *ppDisassembly)
49 {
50     static BOOL firsttime = TRUE;
51
52     /*
53      * TODO: Consider using d3dcompile_xx.dll per
54      * http://msdn.microsoft.com/en-us/library/windows/desktop/ee663275.aspx
55      */
56
57     static HMODULE hD3DXModule = NULL;
58     static PD3DXDISASSEMBLESHADER pfnD3DXDisassembleShader = NULL;
59
60     if (firsttime) {
61         if (!hD3DXModule) {
62             unsigned release;
63             int version;
64             for (release = 0; release <= 1; ++release) {
65                 /* Version 41 corresponds to Mar 2009 version of DirectX Runtime / SDK */
66                 for (version = 41; version >= 0; --version) {
67                     char filename[256];
68                     _snprintf(filename, sizeof(filename),
69                               "d3dx9%s%s%u.dll", release ? "" : "d", version ? "_" : "", version);
70                     hD3DXModule = LoadLibraryA(filename);
71                     if (hD3DXModule)
72                         goto found;
73                 }
74             }
75 found:
76             ;
77         }
78
79         if (hD3DXModule) {
80             if (!pfnD3DXDisassembleShader) {
81                 pfnD3DXDisassembleShader = (PD3DXDISASSEMBLESHADER)GetProcAddress(hD3DXModule, "D3DXDisassembleShader");
82             }
83         }
84
85         firsttime = FALSE;
86     }
87
88     if (!pfnD3DXDisassembleShader) {
89         return E_FAIL;
90     }
91
92     return pfnD3DXDisassembleShader(tokens, FALSE, NULL, ppDisassembly);
93 }
94
95
96 template< class T >
97 inline void
98 dumpShader(JSONWriter &json, const char *name, T *pShader) {
99     if (!pShader) {
100         return;
101     }
102
103     HRESULT hr;
104
105     UINT SizeOfData = 0;
106
107     hr = pShader->GetFunction(NULL, &SizeOfData);
108     if (SUCCEEDED(hr)) {
109         void *pData;
110         pData = malloc(SizeOfData);
111         if (pData) {
112             hr = pShader->GetFunction(pData, &SizeOfData);
113             if (SUCCEEDED(hr)) {
114                 LPD3DXBUFFER pDisassembly;
115
116                 hr = disassembleShader((const DWORD *)pData, &pDisassembly);
117                 if (SUCCEEDED(hr)) {
118                     json.beginMember(name);
119                     json.writeString((const char *)pDisassembly->GetBufferPointer() /*, pDisassembly->GetBufferSize() */);
120                     json.endMember();
121                     pDisassembly->Release();
122                 }
123
124             }
125             free(pData);
126         }
127     }
128 }
129
130 static void
131 dumpShaders(JSONWriter &json, IDirect3DDevice9 *pDevice)
132 {
133     json.beginMember("shaders");
134
135     HRESULT hr;
136     json.beginObject();
137
138     IDirect3DVertexShader9 *pVertexShader = NULL;
139     hr = pDevice->GetVertexShader(&pVertexShader);
140     if (SUCCEEDED(hr)) {
141         dumpShader(json, "vertex", pVertexShader);
142     }
143
144     IDirect3DPixelShader9 *pPixelShader = NULL;
145     hr = pDevice->GetPixelShader(&pPixelShader);
146     if (SUCCEEDED(hr)) {
147         dumpShader(json, "pixel", pPixelShader);
148     }
149
150     json.endObject();
151     json.endMember(); // shaders
152 }
153
154 void
155 dumpDevice(std::ostream &os, IDirect3DDevice9 *pDevice)
156 {
157     JSONWriter json(os);
158
159     dumpShaders(json, pDevice);
160
161     /* TODO */
162 }
163
164
165 } /* namespace d3dstate */