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_MODIFIER (1 << 31)
28 #define WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT 24
29 #define WINED3D_SM4_INSTRUCTION_LENGTH_MASK (0xf << WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT)
31 #define WINED3D_SM4_OPCODE_MASK 0xff
33 #define WINED3D_SM4_REGISTER_MODIFIER (1 << 31)
35 #define WINED3D_SM4_REGISTER_ORDER_SHIFT 20
36 #define WINED3D_SM4_REGISTER_ORDER_MASK (0x3 << WINED3D_SM4_REGISTER_ORDER_SHIFT)
38 #define WINED3D_SM4_REGISTER_TYPE_SHIFT 12
39 #define WINED3D_SM4_REGISTER_TYPE_MASK (0xf << WINED3D_SM4_REGISTER_TYPE_SHIFT)
41 #define WINED3D_SM4_IMMCONST_TYPE_SHIFT 0
42 #define WINED3D_SM4_IMMCONST_TYPE_MASK (0x3 << WINED3D_SM4_IMMCONST_TYPE_SHIFT)
44 #define WINED3D_SM4_WRITEMASK_SHIFT 4
45 #define WINED3D_SM4_WRITEMASK_MASK (0xf << WINED3D_SM4_WRITEMASK_SHIFT)
47 #define WINED3D_SM4_SWIZZLE_SHIFT 4
48 #define WINED3D_SM4_SWIZZLE_MASK (0xff << WINED3D_SM4_SWIZZLE_SHIFT)
50 #define WINED3D_SM4_VERSION_MAJOR(version) (((version) >> 4) & 0xf)
51 #define WINED3D_SM4_VERSION_MINOR(version) (((version) >> 0) & 0xf)
53 enum wined3d_sm4_opcode
55 WINED3D_SM4_OP_ADD = 0x00,
56 WINED3D_SM4_OP_AND = 0x01,
57 WINED3D_SM4_OP_BREAK = 0x02,
58 WINED3D_SM4_OP_BREAKC = 0x03,
59 WINED3D_SM4_OP_CUT = 0x09,
60 WINED3D_SM4_OP_DIV = 0x0e,
61 WINED3D_SM4_OP_DP3 = 0x10,
62 WINED3D_SM4_OP_DP4 = 0x11,
63 WINED3D_SM4_OP_EMIT = 0x13,
64 WINED3D_SM4_OP_ENDIF = 0x15,
65 WINED3D_SM4_OP_ENDLOOP = 0x16,
66 WINED3D_SM4_OP_EXP = 0x19,
67 WINED3D_SM4_OP_FRC = 0x1a,
68 WINED3D_SM4_OP_FTOI = 0x1b,
69 WINED3D_SM4_OP_IADD = 0x1e,
70 WINED3D_SM4_OP_IF = 0x1f,
71 WINED3D_SM4_OP_IGE = 0x21,
72 WINED3D_SM4_OP_IMUL = 0x26,
73 WINED3D_SM4_OP_ITOF = 0x2b,
74 WINED3D_SM4_OP_LOG = 0x2f,
75 WINED3D_SM4_OP_LOOP = 0x30,
76 WINED3D_SM4_OP_LT = 0x31,
77 WINED3D_SM4_OP_MAD = 0x32,
78 WINED3D_SM4_OP_MIN = 0x33,
79 WINED3D_SM4_OP_MAX = 0x34,
80 WINED3D_SM4_OP_MOV = 0x36,
81 WINED3D_SM4_OP_MOVC = 0x37,
82 WINED3D_SM4_OP_MUL = 0x38,
83 WINED3D_SM4_OP_RET = 0x3e,
84 WINED3D_SM4_OP_RSQ = 0x44,
85 WINED3D_SM4_OP_SAMPLE = 0x45,
86 WINED3D_SM4_OP_SAMPLE_LOD = 0x48,
87 WINED3D_SM4_OP_SINCOS = 0x4d,
88 WINED3D_SM4_OP_UTOF = 0x56,
91 enum wined3d_sm4_register_type
93 WINED3D_SM4_RT_TEMP = 0x0,
94 WINED3D_SM4_RT_INPUT = 0x1,
95 WINED3D_SM4_RT_OUTPUT = 0x2,
96 WINED3D_SM4_RT_IMMCONST = 0x4,
97 WINED3D_SM4_RT_SAMPLER = 0x6,
98 WINED3D_SM4_RT_CONSTBUFFER = 0x8,
99 WINED3D_SM4_RT_NULL = 0xd,
102 enum wined3d_sm4_immconst_type
104 WINED3D_SM4_IMMCONST_SCALAR = 0x1,
105 WINED3D_SM4_IMMCONST_VEC4 = 0x2,
108 struct wined3d_sm4_data
110 struct wined3d_shader_version shader_version;
112 const struct wined3d_shader_signature *output_signature;
115 struct wined3d_sm4_opcode_info
117 enum wined3d_sm4_opcode opcode;
118 enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx;
125 enum wined3d_sysval_semantic sysval;
126 WINED3DSHADER_PARAM_REGISTER_TYPE register_type;
130 static const struct wined3d_sm4_opcode_info opcode_table[] =
132 {WINED3D_SM4_OP_ADD, WINED3DSIH_ADD, 1, 2},
133 {WINED3D_SM4_OP_AND, WINED3DSIH_AND, 1, 2},
134 {WINED3D_SM4_OP_BREAK, WINED3DSIH_BREAK, 0, 0},
135 {WINED3D_SM4_OP_BREAKC, WINED3DSIH_BREAKP, 0, 1},
136 {WINED3D_SM4_OP_CUT, WINED3DSIH_CUT, 0, 0},
137 {WINED3D_SM4_OP_DIV, WINED3DSIH_DIV, 1, 2},
138 {WINED3D_SM4_OP_DP3, WINED3DSIH_DP3, 1, 2},
139 {WINED3D_SM4_OP_DP4, WINED3DSIH_DP4, 1, 2},
140 {WINED3D_SM4_OP_EMIT, WINED3DSIH_EMIT, 0, 0},
141 {WINED3D_SM4_OP_ENDIF, WINED3DSIH_ENDIF, 0, 0},
142 {WINED3D_SM4_OP_ENDLOOP, WINED3DSIH_ENDLOOP, 0, 0},
143 {WINED3D_SM4_OP_EXP, WINED3DSIH_EXP, 1, 1},
144 {WINED3D_SM4_OP_FRC, WINED3DSIH_FRC, 1, 1},
145 {WINED3D_SM4_OP_FTOI, WINED3DSIH_FTOI, 1, 1},
146 {WINED3D_SM4_OP_IADD, WINED3DSIH_IADD, 1, 2},
147 {WINED3D_SM4_OP_IF, WINED3DSIH_IF, 0, 1},
148 {WINED3D_SM4_OP_IGE, WINED3DSIH_IGE, 1, 2},
149 {WINED3D_SM4_OP_IMUL, WINED3DSIH_IMUL, 2, 2},
150 {WINED3D_SM4_OP_ITOF, WINED3DSIH_ITOF, 1, 1},
151 {WINED3D_SM4_OP_LOG, WINED3DSIH_LOG, 1, 1},
152 {WINED3D_SM4_OP_LOOP, WINED3DSIH_LOOP, 0, 0},
153 {WINED3D_SM4_OP_LT, WINED3DSIH_LT, 1, 2},
154 {WINED3D_SM4_OP_MAD, WINED3DSIH_MAD, 1, 3},
155 {WINED3D_SM4_OP_MIN, WINED3DSIH_MIN, 1, 2},
156 {WINED3D_SM4_OP_MAX, WINED3DSIH_MAX, 1, 2},
157 {WINED3D_SM4_OP_MOV, WINED3DSIH_MOV, 1, 1},
158 {WINED3D_SM4_OP_MOVC, WINED3DSIH_MOVC, 1, 3},
159 {WINED3D_SM4_OP_MUL, WINED3DSIH_MUL, 1, 2},
160 {WINED3D_SM4_OP_RET, WINED3DSIH_RET, 0, 0},
161 {WINED3D_SM4_OP_RSQ, WINED3DSIH_RSQ, 1, 1},
162 {WINED3D_SM4_OP_SAMPLE, WINED3DSIH_SAMPLE, 1, 3},
163 {WINED3D_SM4_OP_SAMPLE_LOD, WINED3DSIH_SAMPLE_LOD, 1, 4},
164 {WINED3D_SM4_OP_SINCOS, WINED3DSIH_SINCOS, 2, 1},
165 {WINED3D_SM4_OP_UTOF, WINED3DSIH_UTOF, 1, 1},
168 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table[] =
170 /* WINED3D_SM4_RT_TEMP */ WINED3DSPR_TEMP,
171 /* WINED3D_SM4_RT_INPUT */ WINED3DSPR_INPUT,
172 /* WINED3D_SM4_RT_OUTPUT */ WINED3DSPR_OUTPUT,
174 /* WINED3D_SM4_RT_IMMCONST */ WINED3DSPR_IMMCONST,
176 /* WINED3D_SM4_RT_SAMPLER */ WINED3DSPR_SAMPLER,
177 /* WINED3D_SM4_RT_RESOURCE */ WINED3DSPR_RESOURCE,
178 /* WINED3D_SM4_RT_CONSTBUFFER */ WINED3DSPR_CONSTBUFFER,
183 /* WINED3D_SM4_RT_NULL */ WINED3DSPR_NULL,
186 static const struct sysval_map sysval_map[] =
188 {WINED3D_SV_DEPTH, WINED3DSPR_DEPTHOUT, 0},
189 {WINED3D_SV_TARGET0, WINED3DSPR_COLOROUT, 0},
190 {WINED3D_SV_TARGET1, WINED3DSPR_COLOROUT, 1},
191 {WINED3D_SV_TARGET2, WINED3DSPR_COLOROUT, 2},
192 {WINED3D_SV_TARGET3, WINED3DSPR_COLOROUT, 3},
193 {WINED3D_SV_TARGET4, WINED3DSPR_COLOROUT, 4},
194 {WINED3D_SV_TARGET5, WINED3DSPR_COLOROUT, 5},
195 {WINED3D_SV_TARGET6, WINED3DSPR_COLOROUT, 6},
196 {WINED3D_SV_TARGET7, WINED3DSPR_COLOROUT, 7},
199 static const struct wined3d_sm4_opcode_info *get_opcode_info(enum wined3d_sm4_opcode opcode)
203 for (i = 0; i < sizeof(opcode_table) / sizeof(*opcode_table); ++i)
205 if (opcode == opcode_table[i].opcode) return &opcode_table[i];
211 static void map_sysval(enum wined3d_sysval_semantic sysval, struct wined3d_shader_register *reg)
215 for (i = 0; i < sizeof(sysval_map) / sizeof(*sysval_map); ++i)
217 if (sysval == sysval_map[i].sysval)
219 reg->type = sysval_map[i].register_type;
220 reg->idx = sysval_map[i].register_idx;
225 static void map_register(struct wined3d_sm4_data *priv, struct wined3d_shader_register *reg)
227 switch (priv->shader_version.type)
229 case WINED3D_SHADER_TYPE_PIXEL:
230 if (reg->type == WINED3DSPR_OUTPUT)
233 const struct wined3d_shader_signature *s = priv->output_signature;
237 ERR("Shader has no output signature, unable to map register.\n");
241 for (i = 0; i < s->element_count; ++i)
243 if (s->elements[i].register_idx == reg->idx)
245 map_sysval(s->elements[i].sysval_semantic, reg);
257 static void *shader_sm4_init(const DWORD *byte_code, const struct wined3d_shader_signature *output_signature)
259 struct wined3d_sm4_data *priv = HeapAlloc(GetProcessHeap(), 0, sizeof(*priv));
262 ERR("Failed to allocate private data\n");
266 priv->output_signature = output_signature;
271 static void shader_sm4_free(void *data)
273 HeapFree(GetProcessHeap(), 0, data);
276 static void shader_sm4_read_header(void *data, const DWORD **ptr, struct wined3d_shader_version *shader_version)
278 struct wined3d_sm4_data *priv = data;
283 version_token = *(*ptr)++;
284 TRACE("version: 0x%08x\n", version_token);
286 TRACE("token count: %u\n", **ptr);
287 priv->end += *(*ptr)++;
289 switch (version_token >> 16)
292 priv->shader_version.type = WINED3D_SHADER_TYPE_PIXEL;
296 priv->shader_version.type = WINED3D_SHADER_TYPE_VERTEX;
300 priv->shader_version.type = WINED3D_SHADER_TYPE_GEOMETRY;
304 FIXME("Unrecognized shader type %#x\n", version_token >> 16);
306 priv->shader_version.major = WINED3D_SM4_VERSION_MAJOR(version_token);
307 priv->shader_version.minor = WINED3D_SM4_VERSION_MINOR(version_token);
309 *shader_version = priv->shader_version;
312 static void shader_sm4_read_opcode(void *data, const DWORD **ptr, struct wined3d_shader_instruction *ins,
315 const struct wined3d_sm4_opcode_info *opcode_info;
316 DWORD token = *(*ptr)++;
317 DWORD opcode = token & WINED3D_SM4_OPCODE_MASK;
319 *param_size = ((token & WINED3D_SM4_INSTRUCTION_LENGTH_MASK) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT) - 1;
321 opcode_info = get_opcode_info(opcode);
324 FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode, token);
325 ins->handler_idx = WINED3DSIH_TABLE_SIZE;
329 ins->handler_idx = opcode_info->handler_idx;
333 ins->dst_count = opcode_info->dst_count;
334 ins->src_count = opcode_info->src_count;
336 if (token & WINED3D_SM4_INSTRUCTION_MODIFIER)
338 DWORD modifier = *(*ptr)++;
339 FIXME("Skipping modifier 0x%08x.\n", modifier);
343 static void shader_sm4_read_src_param(void *data, const DWORD **ptr, struct wined3d_shader_src_param *src_param,
344 struct wined3d_shader_src_param *src_rel_addr)
346 struct wined3d_sm4_data *priv = data;
347 DWORD token = *(*ptr)++;
348 enum wined3d_sm4_register_type register_type;
351 register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
352 if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
354 FIXME("Unhandled register type %#x\n", register_type);
355 src_param->reg.type = WINED3DSPR_TEMP;
359 src_param->reg.type = register_type_table[register_type];
362 if (token & WINED3D_SM4_REGISTER_MODIFIER)
364 DWORD modifier = *(*ptr)++;
365 FIXME("Skipping modifier 0x%08x.\n", modifier);
368 order = (token & WINED3D_SM4_REGISTER_ORDER_MASK) >> WINED3D_SM4_REGISTER_ORDER_SHIFT;
370 if (order < 1) src_param->reg.idx = ~0U;
371 else src_param->reg.idx = *(*ptr)++;
373 if (order < 2) src_param->reg.array_idx = ~0U;
374 else src_param->reg.array_idx = *(*ptr)++;
376 if (order > 2) FIXME("Unhandled order %u.\n", order);
378 if (register_type == WINED3D_SM4_RT_IMMCONST)
380 enum wined3d_sm4_immconst_type immconst_type =
381 (token & WINED3D_SM4_IMMCONST_TYPE_MASK) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT;
382 src_param->swizzle = WINED3DSP_NOSWIZZLE;
384 switch(immconst_type)
386 case WINED3D_SM4_IMMCONST_SCALAR:
387 src_param->reg.immconst_type = WINED3D_IMMCONST_SCALAR;
388 memcpy(src_param->reg.immconst_data, *ptr, 1 * sizeof(DWORD));
392 case WINED3D_SM4_IMMCONST_VEC4:
393 src_param->reg.immconst_type = WINED3D_IMMCONST_VEC4;
394 memcpy(src_param->reg.immconst_data, *ptr, 4 * sizeof(DWORD));
399 FIXME("Unhandled immediate constant type %#x\n", immconst_type);
405 src_param->swizzle = (token & WINED3D_SM4_SWIZZLE_MASK) >> WINED3D_SM4_SWIZZLE_SHIFT;
408 src_param->modifiers = 0;
409 src_param->reg.rel_addr = NULL;
411 map_register(priv, &src_param->reg);
414 static void shader_sm4_read_dst_param(void *data, const DWORD **ptr, struct wined3d_shader_dst_param *dst_param,
415 struct wined3d_shader_src_param *dst_rel_addr)
417 struct wined3d_sm4_data *priv = data;
418 DWORD token = *(*ptr)++;
419 enum wined3d_sm4_register_type register_type;
422 register_type = (token & WINED3D_SM4_REGISTER_TYPE_MASK) >> WINED3D_SM4_REGISTER_TYPE_SHIFT;
423 if (register_type >= sizeof(register_type_table) / sizeof(*register_type_table))
425 FIXME("Unhandled register type %#x\n", register_type);
426 dst_param->reg.type = WINED3DSPR_TEMP;
430 dst_param->reg.type = register_type_table[register_type];
433 order = (token & WINED3D_SM4_REGISTER_ORDER_MASK) >> WINED3D_SM4_REGISTER_ORDER_SHIFT;
435 if (order < 1) dst_param->reg.idx = ~0U;
436 else dst_param->reg.idx = *(*ptr)++;
438 if (order < 2) dst_param->reg.array_idx = ~0U;
439 else dst_param->reg.array_idx = *(*ptr)++;
441 if (order > 2) FIXME("Unhandled order %u.\n", order);
443 dst_param->write_mask = (token & WINED3D_SM4_WRITEMASK_MASK) >> WINED3D_SM4_WRITEMASK_SHIFT;
444 dst_param->modifiers = 0;
445 dst_param->shift = 0;
446 dst_param->reg.rel_addr = NULL;
448 map_register(priv, &dst_param->reg);
451 static void shader_sm4_read_semantic(const DWORD **ptr, struct wined3d_shader_semantic *semantic)
453 FIXME("ptr %p, semantic %p stub!\n", ptr, semantic);
456 static void shader_sm4_read_comment(const DWORD **ptr, const char **comment, UINT *comment_size)
458 FIXME("ptr %p, comment %p, comment_size %p stub!\n", ptr, comment, comment_size);
462 static BOOL shader_sm4_is_end(void *data, const DWORD **ptr)
464 struct wined3d_sm4_data *priv = data;
465 return *ptr == priv->end;
468 const struct wined3d_shader_frontend sm4_shader_frontend =
472 shader_sm4_read_header,
473 shader_sm4_read_opcode,
474 shader_sm4_read_src_param,
475 shader_sm4_read_dst_param,
476 shader_sm4_read_semantic,
477 shader_sm4_read_comment,