2 * Copyright 2009 Henri Verbeet for CodeWeavers
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
20 #include "wine/port.h"
22 #include "wined3d_private.h"
24 WINE_DEFAULT_DEBUG_CHANNEL(d3d_shader);
26 #define WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT 24
27 #define WINED3D_SM4_INSTRUCTION_LENGTH_MASK (0xf << WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT)
29 #define WINED3D_SM4_OPCODE_MASK 0xff
31 #define WINED3D_SM4_REGISTER_TYPE_SHIFT 12
32 #define WINED3D_SM4_REGISTER_TYPE_MASK (0xf << WINED3D_SM4_REGISTER_TYPE_SHIFT)
34 #define WINED3D_SM4_IMMCONST_TYPE_SHIFT 0
35 #define WINED3D_SM4_IMMCONST_TYPE_MASK (0x3 << WINED3D_SM4_IMMCONST_TYPE_SHIFT)
37 #define WINED3D_SM4_WRITEMASK_SHIFT 4
38 #define WINED3D_SM4_WRITEMASK_MASK (0xf << WINED3D_SM4_WRITEMASK_SHIFT)
40 #define WINED3D_SM4_SWIZZLE_SHIFT 4
41 #define WINED3D_SM4_SWIZZLE_MASK (0xff << WINED3D_SM4_SWIZZLE_SHIFT)
43 #define WINED3D_SM4_VERSION_MAJOR(version) (((version) >> 4) & 0xf)
44 #define WINED3D_SM4_VERSION_MINOR(version) (((version) >> 0) & 0xf)
46 enum wined3d_sm4_opcode
48 WINED3D_SM4_OP_ADD = 0x00,
49 WINED3D_SM4_OP_EXP = 0x19,
50 WINED3D_SM4_OP_MOV = 0x36,
51 WINED3D_SM4_OP_MUL = 0x38,
52 WINED3D_SM4_OP_RET = 0x3e,
53 WINED3D_SM4_OP_SINCOS = 0x4d,
56 enum wined3d_sm4_register_type
58 WINED3D_SM4_RT_TEMP = 0x0,
59 WINED3D_SM4_RT_INPUT = 0x1,
60 WINED3D_SM4_RT_OUTPUT = 0x2,
61 WINED3D_SM4_RT_IMMCONST = 0x4,
64 enum wined3d_sm4_immconst_type
66 WINED3D_SM4_IMMCONST_FLOAT = 0x1,
67 WINED3D_SM4_IMMCONST_FLOAT4 = 0x2,
70 struct wined3d_sm4_data
72 struct wined3d_shader_version shader_version;
74 const struct wined3d_shader_signature *output_signature;
77 struct wined3d_sm4_opcode_info
79 enum wined3d_sm4_opcode opcode;
80 enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx;
87 enum wined3d_sysval_semantic sysval;
88 WINED3DSHADER_PARAM_REGISTER_TYPE register_type;
92 static const struct wined3d_sm4_opcode_info opcode_table[] =
94 {WINED3D_SM4_OP_ADD, WINED3DSIH_ADD, 1, 2},
95 {WINED3D_SM4_OP_EXP, WINED3DSIH_EXP, 1, 1},
96 {WINED3D_SM4_OP_MOV, WINED3DSIH_MOV, 1, 1},
97 {WINED3D_SM4_OP_MUL, WINED3DSIH_MUL, 1, 2},
98 {WINED3D_SM4_OP_RET, WINED3DSIH_RET, 0, 0},
99 {WINED3D_SM4_OP_SINCOS, WINED3DSIH_SINCOS, 1, 2},
102 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table[] =
104 /* WINED3D_SM4_RT_TEMP */ WINED3DSPR_TEMP,
105 /* WINED3D_SM4_RT_INPUT */ WINED3DSPR_INPUT,
106 /* WINED3D_SM4_RT_OUTPUT */ WINED3DSPR_OUTPUT,
108 /* WINED3D_SM4_RT_IMMCONST */ WINED3DSPR_IMMCONST,
111 static const struct sysval_map sysval_map[] =
113 {WINED3D_SV_DEPTH, WINED3DSPR_DEPTHOUT, 0},
114 {WINED3D_SV_TARGET0, WINED3DSPR_COLOROUT, 0},
115 {WINED3D_SV_TARGET1, WINED3DSPR_COLOROUT, 1},
116 {WINED3D_SV_TARGET2, WINED3DSPR_COLOROUT, 2},
117 {WINED3D_SV_TARGET3, WINED3DSPR_COLOROUT, 3},
118 {WINED3D_SV_TARGET4, WINED3DSPR_COLOROUT, 4},
119 {WINED3D_SV_TARGET5, WINED3DSPR_COLOROUT, 5},
120 {WINED3D_SV_TARGET6, WINED3DSPR_COLOROUT, 6},
121 {WINED3D_SV_TARGET7, WINED3DSPR_COLOROUT, 7},
124 static const struct wined3d_sm4_opcode_info *get_opcode_info(enum wined3d_sm4_opcode opcode)
128 for (i = 0; i < sizeof(opcode_table) / sizeof(*opcode_table); ++i)
130 if (opcode == opcode_table[i].opcode) return &opcode_table[i];
136 static void map_sysval(enum wined3d_sysval_semantic sysval, struct wined3d_shader_register *reg)
140 for (i = 0; i < sizeof(sysval_map) / sizeof(*sysval_map); ++i)
142 if (sysval == sysval_map[i].sysval)
144 reg->type = sysval_map[i].register_type;
145 reg->idx = sysval_map[i].register_idx;
150 static void map_register(struct wined3d_sm4_data *priv, struct wined3d_shader_register *reg)
152 switch (priv->shader_version.type)
154 case WINED3D_SHADER_TYPE_PIXEL:
155 if (reg->type == WINED3DSPR_OUTPUT)
158 const struct wined3d_shader_signature *s = priv->output_signature;
162 ERR("Shader has no output signature, unable to map register.\n");
166 for (i = 0; i < s->element_count; ++i)
168 if (s->elements[i].register_idx == reg->idx)
170 map_sysval(s->elements[i].sysval_semantic, reg);
182 static void *shader_sm4_init(const DWORD *byte_code, const struct wined3d_shader_signature *output_signature)
184 struct wined3d_sm4_data *priv = HeapAlloc(GetProcessHeap(), 0, sizeof(*priv));
187 ERR("Failed to allocate private data\n");
191 priv->output_signature = output_signature;
196 static void shader_sm4_free(void *data)
198 HeapFree(GetProcessHeap(), 0, data);
201 static void shader_sm4_read_header(void *data, const DWORD **ptr, struct wined3d_shader_version *shader_version)
203 struct wined3d_sm4_data *priv = data;
208 version_token = *(*ptr)++;
209 TRACE("version: 0x%08x\n", version_token);
211 TRACE("token count: %u\n", **ptr);
212 priv->end += *(*ptr)++;
214 switch (version_token >> 16)
217 priv->shader_version.type = WINED3D_SHADER_TYPE_PIXEL;
221 priv->shader_version.type = WINED3D_SHADER_TYPE_VERTEX;
225 priv->shader_version.type = WINED3D_SHADER_TYPE_GEOMETRY;
229 FIXME("Unrecognized shader type %#x\n", version_token >> 16);
231 priv->shader_version.major = WINED3D_SM4_VERSION_MAJOR(version_token);
232 priv->shader_version.minor = WINED3D_SM4_VERSION_MINOR(version_token);
234 *shader_version = priv->shader_version;
237 static void shader_sm4_read_opcode(void *data, const DWORD **ptr, struct wined3d_shader_instruction *ins,
240 const struct wined3d_sm4_opcode_info *opcode_info;
241 DWORD token = *(*ptr)++;
242 DWORD opcode = token & WINED3D_SM4_OPCODE_MASK;
244 *param_size = ((token & WINED3D_SM4_INSTRUCTION_LENGTH_MASK) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT) - 1;
246 opcode_info = get_opcode_info(opcode);
249 FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode, token);
250 ins->handler_idx = WINED3DSIH_TABLE_SIZE;
254 ins->handler_idx = opcode_info->handler_idx;
258 ins->dst_count = opcode_info->dst_count;
259 ins->src_count = opcode_info->src_count;
262 static void shader_sm4_read_src_param(void *data, const DWORD **ptr, struct wined3d_shader_src_param *src_param,
263 struct wined3d_shader_src_param *src_rel_addr)
265 struct wined3d_sm4_data *priv = data;
266 DWORD token = *(*ptr)++;
267 enum wined3d_sm4_register_type register_type;
269 register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
270 if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
272 FIXME("Unhandled register type %#x\n", register_type);
273 src_param->reg.type = WINED3DSPR_TEMP;
277 src_param->reg.type = register_type_table[register_type];
280 if (register_type == WINED3D_SM4_RT_IMMCONST)
282 enum wined3d_sm4_immconst_type immconst_type =
283 (token & WINED3D_SM4_IMMCONST_TYPE_MASK) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT;
284 src_param->swizzle = WINED3DSP_NOSWIZZLE;
286 switch(immconst_type)
288 case WINED3D_SM4_IMMCONST_FLOAT:
289 src_param->reg.immconst_type = WINED3D_IMMCONST_FLOAT;
290 memcpy(src_param->reg.immconst_data, *ptr, 1 * sizeof(DWORD));
294 case WINED3D_SM4_IMMCONST_FLOAT4:
295 src_param->reg.immconst_type = WINED3D_IMMCONST_FLOAT4;
296 memcpy(src_param->reg.immconst_data, *ptr, 4 * sizeof(DWORD));
301 FIXME("Unhandled immediate constant type %#x\n", immconst_type);
307 src_param->reg.idx = *(*ptr)++;
308 src_param->swizzle = (token & WINED3D_SM4_SWIZZLE_MASK) >> WINED3D_SM4_SWIZZLE_SHIFT;
311 src_param->modifiers = 0;
312 src_param->reg.rel_addr = NULL;
314 map_register(priv, &src_param->reg);
317 static void shader_sm4_read_dst_param(void *data, const DWORD **ptr, struct wined3d_shader_dst_param *dst_param,
318 struct wined3d_shader_src_param *dst_rel_addr)
320 struct wined3d_sm4_data *priv = data;
321 DWORD token = *(*ptr)++;
322 UINT register_idx = *(*ptr)++;
323 enum wined3d_sm4_register_type register_type;
325 register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
326 if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
328 FIXME("Unhandled register type %#x\n", register_type);
329 dst_param->reg.type = WINED3DSPR_TEMP;
333 dst_param->reg.type = register_type_table[register_type];
336 dst_param->reg.idx = register_idx;
337 dst_param->write_mask = (token & WINED3D_SM4_WRITEMASK_MASK) >> WINED3D_SM4_WRITEMASK_SHIFT;
338 dst_param->modifiers = 0;
339 dst_param->shift = 0;
340 dst_param->reg.rel_addr = NULL;
342 map_register(priv, &dst_param->reg);
345 static void shader_sm4_read_semantic(const DWORD **ptr, struct wined3d_shader_semantic *semantic)
347 FIXME("ptr %p, semantic %p stub!\n", ptr, semantic);
350 static void shader_sm4_read_comment(const DWORD **ptr, const char **comment)
352 FIXME("ptr %p, comment %p stub!\n", ptr, comment);
356 static BOOL shader_sm4_is_end(void *data, const DWORD **ptr)
358 struct wined3d_sm4_data *priv = data;
359 return *ptr == priv->end;
362 const struct wined3d_shader_frontend sm4_shader_frontend =
366 shader_sm4_read_header,
367 shader_sm4_read_opcode,
368 shader_sm4_read_src_param,
369 shader_sm4_read_dst_param,
370 shader_sm4_read_semantic,
371 shader_sm4_read_comment,