]> git.cworth.org Git - apitrace/blob - helpers/d3dsize.hpp
Preserve both D3D9 shader byte code, and disassembly.
[apitrace] / helpers / d3dsize.hpp
1 /**************************************************************************
2  *
3  * Copyright 2012 Jose Fonseca
4  * All Rights Reserved.
5  * 
6  * Permission is hereby granted, free of charge, to any person obtaining a
7  * copy of this software and associated documentation files (the "Software"),
8  * to deal in the Software without restriction, including without limitation
9  * the rights to use, copy, modify, merge, publish, distribute, sub license,
10  * and/or sell copies of the Software, and to permit persons to whom the
11  * Software is furnished to do so, subject to the following conditions:
12  * 
13  * The above copyright notice and this permission notice (including the next
14  * paragraph) shall be included in all copies or substantial portions of the
15  * Software.
16  * 
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.  IN NO EVENT SHALL
20  * AUTHORS,
21  * AND/OR THEIR SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
22  * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF
23  * OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
24  * SOFTWARE.
25  *
26  **************************************************************************/
27
28
29 /*
30  * Auxiliary functions to compute the size of array/blob arguments.
31  */
32
33 #ifndef _D3D_SIZE_HPP_
34 #define _D3D_SIZE_HPP_
35
36
37 /* We purposedly don't include any D3D header, so that this header can be used
38  * with all D3D versions. */
39
40 #include <assert.h>
41
42 #include "os.hpp"
43
44
45 static inline size_t
46 _vertexCount(D3DPRIMITIVETYPE PrimitiveType, UINT PrimitiveCount)
47 {
48     switch (PrimitiveType) {
49     case D3DPT_POINTLIST:
50         return PrimitiveCount;
51     case D3DPT_LINELIST:
52         return PrimitiveCount*2;
53     case D3DPT_LINESTRIP:
54         return PrimitiveCount + 1;
55     case D3DPT_TRIANGLELIST:
56         return PrimitiveCount * 3;
57     case D3DPT_TRIANGLESTRIP:
58         return PrimitiveCount + 2;
59     case D3DPT_TRIANGLEFAN:
60         return PrimitiveCount + 1;
61     default:
62         os::log("apitrace: warning: %s: unknown D3DPRIMITIVETYPE %u\n", __FUNCTION__, PrimitiveType);
63         return 0;
64     }
65 }
66
67
68 static inline size_t
69 _vertexDataSize(D3DPRIMITIVETYPE PrimitiveType, UINT PrimitiveCount, UINT VertexStride) {
70     return _vertexCount(PrimitiveType, PrimitiveCount) * VertexStride;
71 }
72
73
74 static inline size_t
75 _indexDataSize(D3DPRIMITIVETYPE PrimitiveType, UINT PrimitiveCount, D3DFORMAT IndexDataFormat) {
76     UINT IndexStride;
77     switch (IndexDataFormat) {
78     case D3DFMT_INDEX16:
79         IndexStride = 2;
80         break;
81     case D3DFMT_INDEX32:
82         IndexStride = 4;
83         break;
84     default:
85         os::log("apitrace: warning: %s: unexpected index D3DFORMAT %u\n", __FUNCTION__, IndexDataFormat);
86         return 0;
87     }
88     return _vertexCount(PrimitiveType, PrimitiveCount) * IndexStride;
89 }
90
91
92 #if DIRECT3D_VERSION >= 0x0800
93
94 /*
95  * Return the number of tokens for a given shader.
96  */
97 static inline size_t
98 _shaderSize(const DWORD *pFunction)
99 {
100     DWORD dwLength = 0;
101
102     while (true) {
103         DWORD dwToken = pFunction[dwLength++];
104
105         switch (dwToken & D3DSI_OPCODE_MASK) {
106         case D3DSIO_COMMENT:
107             dwLength += (dwToken & D3DSI_COMMENTSIZE_MASK) >> D3DSI_COMMENTSIZE_SHIFT;
108             break;
109
110         case D3DSIO_END:
111             if (dwToken != D3DSIO_END) {
112                 os::log("apitrace: warning: %s: malformed END token\n", __FUNCTION__);
113             }
114             return dwLength * sizeof *pFunction;
115         }
116     }
117 }
118
119 #endif /* DIRECT3D_VERSION >= 0x0800 */
120
121
122 #endif /* _D3D_SIZE_HPP_ */