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_ORDER_SHIFT 20
32 #define WINED3D_SM4_REGISTER_ORDER_MASK (0x3 << WINED3D_SM4_REGISTER_ORDER_SHIFT)
34 #define WINED3D_SM4_REGISTER_TYPE_SHIFT 12
35 #define WINED3D_SM4_REGISTER_TYPE_MASK (0xf << WINED3D_SM4_REGISTER_TYPE_SHIFT)
37 #define WINED3D_SM4_IMMCONST_TYPE_SHIFT 0
38 #define WINED3D_SM4_IMMCONST_TYPE_MASK (0x3 << WINED3D_SM4_IMMCONST_TYPE_SHIFT)
40 #define WINED3D_SM4_WRITEMASK_SHIFT 4
41 #define WINED3D_SM4_WRITEMASK_MASK (0xf << WINED3D_SM4_WRITEMASK_SHIFT)
43 #define WINED3D_SM4_SWIZZLE_SHIFT 4
44 #define WINED3D_SM4_SWIZZLE_MASK (0xff << WINED3D_SM4_SWIZZLE_SHIFT)
46 #define WINED3D_SM4_VERSION_MAJOR(version) (((version) >> 4) & 0xf)
47 #define WINED3D_SM4_VERSION_MINOR(version) (((version) >> 0) & 0xf)
49 enum wined3d_sm4_opcode
51 WINED3D_SM4_OP_ADD
= 0x00,
52 WINED3D_SM4_OP_AND
= 0x01,
53 WINED3D_SM4_OP_BREAK
= 0x02,
54 WINED3D_SM4_OP_BREAKC
= 0x03,
55 WINED3D_SM4_OP_CUT
= 0x09,
56 WINED3D_SM4_OP_DP3
= 0x10,
57 WINED3D_SM4_OP_DP4
= 0x11,
58 WINED3D_SM4_OP_EMIT
= 0x13,
59 WINED3D_SM4_OP_ENDIF
= 0x15,
60 WINED3D_SM4_OP_ENDLOOP
= 0x16,
61 WINED3D_SM4_OP_EXP
= 0x19,
62 WINED3D_SM4_OP_FRC
= 0x1a,
63 WINED3D_SM4_OP_IADD
= 0x1e,
64 WINED3D_SM4_OP_IF
= 0x1f,
65 WINED3D_SM4_OP_IGE
= 0x21,
66 WINED3D_SM4_OP_IMUL
= 0x26,
67 WINED3D_SM4_OP_ITOF
= 0x2b,
68 WINED3D_SM4_OP_LOG
= 0x2f,
69 WINED3D_SM4_OP_LOOP
= 0x30,
70 WINED3D_SM4_OP_LT
= 0x31,
71 WINED3D_SM4_OP_MAD
= 0x32,
72 WINED3D_SM4_OP_MIN
= 0x33,
73 WINED3D_SM4_OP_MAX
= 0x34,
74 WINED3D_SM4_OP_MOV
= 0x36,
75 WINED3D_SM4_OP_MOVC
= 0x37,
76 WINED3D_SM4_OP_MUL
= 0x38,
77 WINED3D_SM4_OP_RET
= 0x3e,
78 WINED3D_SM4_OP_RSQ
= 0x44,
79 WINED3D_SM4_OP_SAMPLE
= 0x45,
80 WINED3D_SM4_OP_SAMPLE_LOD
= 0x48,
81 WINED3D_SM4_OP_SINCOS
= 0x4d,
82 WINED3D_SM4_OP_UTOF
= 0x56,
85 enum wined3d_sm4_register_type
87 WINED3D_SM4_RT_TEMP
= 0x0,
88 WINED3D_SM4_RT_INPUT
= 0x1,
89 WINED3D_SM4_RT_OUTPUT
= 0x2,
90 WINED3D_SM4_RT_IMMCONST
= 0x4,
91 WINED3D_SM4_RT_SAMPLER
= 0x6,
92 WINED3D_SM4_RT_CONSTBUFFER
= 0x8,
93 WINED3D_SM4_RT_NULL
= 0xd,
96 enum wined3d_sm4_immconst_type
98 WINED3D_SM4_IMMCONST_SCALAR
= 0x1,
99 WINED3D_SM4_IMMCONST_VEC4
= 0x2,
102 struct wined3d_sm4_data
104 struct wined3d_shader_version shader_version
;
106 const struct wined3d_shader_signature
*output_signature
;
109 struct wined3d_sm4_opcode_info
111 enum wined3d_sm4_opcode opcode
;
112 enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx
;
119 enum wined3d_sysval_semantic sysval
;
120 WINED3DSHADER_PARAM_REGISTER_TYPE register_type
;
124 static const struct wined3d_sm4_opcode_info opcode_table
[] =
126 {WINED3D_SM4_OP_ADD
, WINED3DSIH_ADD
, 1, 2},
127 {WINED3D_SM4_OP_AND
, WINED3DSIH_AND
, 1, 2},
128 {WINED3D_SM4_OP_BREAK
, WINED3DSIH_BREAK
, 0, 0},
129 {WINED3D_SM4_OP_BREAKC
, WINED3DSIH_BREAKP
, 0, 1},
130 {WINED3D_SM4_OP_CUT
, WINED3DSIH_CUT
, 0, 0},
131 {WINED3D_SM4_OP_DP3
, WINED3DSIH_DP3
, 1, 2},
132 {WINED3D_SM4_OP_DP4
, WINED3DSIH_DP4
, 1, 2},
133 {WINED3D_SM4_OP_EMIT
, WINED3DSIH_EMIT
, 0, 0},
134 {WINED3D_SM4_OP_ENDIF
, WINED3DSIH_ENDIF
, 0, 0},
135 {WINED3D_SM4_OP_ENDLOOP
, WINED3DSIH_ENDLOOP
, 0, 0},
136 {WINED3D_SM4_OP_EXP
, WINED3DSIH_EXP
, 1, 1},
137 {WINED3D_SM4_OP_FRC
, WINED3DSIH_FRC
, 1, 1},
138 {WINED3D_SM4_OP_IADD
, WINED3DSIH_IADD
, 1, 2},
139 {WINED3D_SM4_OP_IF
, WINED3DSIH_IF
, 0, 1},
140 {WINED3D_SM4_OP_IGE
, WINED3DSIH_IGE
, 1, 2},
141 {WINED3D_SM4_OP_IMUL
, WINED3DSIH_IMUL
, 2, 2},
142 {WINED3D_SM4_OP_ITOF
, WINED3DSIH_ITOF
, 1, 1},
143 {WINED3D_SM4_OP_LOG
, WINED3DSIH_LOG
, 1, 1},
144 {WINED3D_SM4_OP_LOOP
, WINED3DSIH_LOOP
, 0, 0},
145 {WINED3D_SM4_OP_LT
, WINED3DSIH_LT
, 1, 2},
146 {WINED3D_SM4_OP_MAD
, WINED3DSIH_MAD
, 1, 3},
147 {WINED3D_SM4_OP_MIN
, WINED3DSIH_MIN
, 1, 2},
148 {WINED3D_SM4_OP_MAX
, WINED3DSIH_MAX
, 1, 2},
149 {WINED3D_SM4_OP_MOV
, WINED3DSIH_MOV
, 1, 1},
150 {WINED3D_SM4_OP_MOVC
, WINED3DSIH_MOVC
, 1, 3},
151 {WINED3D_SM4_OP_MUL
, WINED3DSIH_MUL
, 1, 2},
152 {WINED3D_SM4_OP_RET
, WINED3DSIH_RET
, 0, 0},
153 {WINED3D_SM4_OP_RSQ
, WINED3DSIH_RSQ
, 1, 1},
154 {WINED3D_SM4_OP_SAMPLE
, WINED3DSIH_SAMPLE
, 1, 3},
155 {WINED3D_SM4_OP_SAMPLE_LOD
, WINED3DSIH_SAMPLE_LOD
, 1, 4},
156 {WINED3D_SM4_OP_SINCOS
, WINED3DSIH_SINCOS
, 2, 1},
157 {WINED3D_SM4_OP_UTOF
, WINED3DSIH_UTOF
, 1, 1},
160 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table
[] =
162 /* WINED3D_SM4_RT_TEMP */ WINED3DSPR_TEMP
,
163 /* WINED3D_SM4_RT_INPUT */ WINED3DSPR_INPUT
,
164 /* WINED3D_SM4_RT_OUTPUT */ WINED3DSPR_OUTPUT
,
166 /* WINED3D_SM4_RT_IMMCONST */ WINED3DSPR_IMMCONST
,
168 /* WINED3D_SM4_RT_SAMPLER */ WINED3DSPR_SAMPLER
,
170 /* WINED3D_SM4_RT_CONSTBUFFER */ WINED3DSPR_CONSTBUFFER
,
175 /* WINED3D_SM4_RT_NULL */ WINED3DSPR_NULL
,
178 static const struct sysval_map sysval_map
[] =
180 {WINED3D_SV_DEPTH
, WINED3DSPR_DEPTHOUT
, 0},
181 {WINED3D_SV_TARGET0
, WINED3DSPR_COLOROUT
, 0},
182 {WINED3D_SV_TARGET1
, WINED3DSPR_COLOROUT
, 1},
183 {WINED3D_SV_TARGET2
, WINED3DSPR_COLOROUT
, 2},
184 {WINED3D_SV_TARGET3
, WINED3DSPR_COLOROUT
, 3},
185 {WINED3D_SV_TARGET4
, WINED3DSPR_COLOROUT
, 4},
186 {WINED3D_SV_TARGET5
, WINED3DSPR_COLOROUT
, 5},
187 {WINED3D_SV_TARGET6
, WINED3DSPR_COLOROUT
, 6},
188 {WINED3D_SV_TARGET7
, WINED3DSPR_COLOROUT
, 7},
191 static const struct wined3d_sm4_opcode_info
*get_opcode_info(enum wined3d_sm4_opcode opcode
)
195 for (i
= 0; i
< sizeof(opcode_table
) / sizeof(*opcode_table
); ++i
)
197 if (opcode
== opcode_table
[i
].opcode
) return &opcode_table
[i
];
203 static void map_sysval(enum wined3d_sysval_semantic sysval
, struct wined3d_shader_register
*reg
)
207 for (i
= 0; i
< sizeof(sysval_map
) / sizeof(*sysval_map
); ++i
)
209 if (sysval
== sysval_map
[i
].sysval
)
211 reg
->type
= sysval_map
[i
].register_type
;
212 reg
->idx
= sysval_map
[i
].register_idx
;
217 static void map_register(struct wined3d_sm4_data
*priv
, struct wined3d_shader_register
*reg
)
219 switch (priv
->shader_version
.type
)
221 case WINED3D_SHADER_TYPE_PIXEL
:
222 if (reg
->type
== WINED3DSPR_OUTPUT
)
225 const struct wined3d_shader_signature
*s
= priv
->output_signature
;
229 ERR("Shader has no output signature, unable to map register.\n");
233 for (i
= 0; i
< s
->element_count
; ++i
)
235 if (s
->elements
[i
].register_idx
== reg
->idx
)
237 map_sysval(s
->elements
[i
].sysval_semantic
, reg
);
249 static void *shader_sm4_init(const DWORD
*byte_code
, const struct wined3d_shader_signature
*output_signature
)
251 struct wined3d_sm4_data
*priv
= HeapAlloc(GetProcessHeap(), 0, sizeof(*priv
));
254 ERR("Failed to allocate private data\n");
258 priv
->output_signature
= output_signature
;
263 static void shader_sm4_free(void *data
)
265 HeapFree(GetProcessHeap(), 0, data
);
268 static void shader_sm4_read_header(void *data
, const DWORD
**ptr
, struct wined3d_shader_version
*shader_version
)
270 struct wined3d_sm4_data
*priv
= data
;
275 version_token
= *(*ptr
)++;
276 TRACE("version: 0x%08x\n", version_token
);
278 TRACE("token count: %u\n", **ptr
);
279 priv
->end
+= *(*ptr
)++;
281 switch (version_token
>> 16)
284 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_PIXEL
;
288 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_VERTEX
;
292 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_GEOMETRY
;
296 FIXME("Unrecognized shader type %#x\n", version_token
>> 16);
298 priv
->shader_version
.major
= WINED3D_SM4_VERSION_MAJOR(version_token
);
299 priv
->shader_version
.minor
= WINED3D_SM4_VERSION_MINOR(version_token
);
301 *shader_version
= priv
->shader_version
;
304 static void shader_sm4_read_opcode(void *data
, const DWORD
**ptr
, struct wined3d_shader_instruction
*ins
,
307 const struct wined3d_sm4_opcode_info
*opcode_info
;
308 DWORD token
= *(*ptr
)++;
309 DWORD opcode
= token
& WINED3D_SM4_OPCODE_MASK
;
311 *param_size
= ((token
& WINED3D_SM4_INSTRUCTION_LENGTH_MASK
) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT
) - 1;
313 opcode_info
= get_opcode_info(opcode
);
316 FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode
, token
);
317 ins
->handler_idx
= WINED3DSIH_TABLE_SIZE
;
321 ins
->handler_idx
= opcode_info
->handler_idx
;
325 ins
->dst_count
= opcode_info
->dst_count
;
326 ins
->src_count
= opcode_info
->src_count
;
329 static void shader_sm4_read_src_param(void *data
, const DWORD
**ptr
, struct wined3d_shader_src_param
*src_param
,
330 struct wined3d_shader_src_param
*src_rel_addr
)
332 struct wined3d_sm4_data
*priv
= data
;
333 DWORD token
= *(*ptr
)++;
334 enum wined3d_sm4_register_type register_type
;
337 register_type
= (token
& WINED3D_SM4_REGISTER_TYPE_MASK
) >> WINED3D_SM4_REGISTER_TYPE_SHIFT
;
338 if (register_type
>= sizeof(register_type_table
) / sizeof(*register_type_table
))
340 FIXME("Unhandled register type %#x\n", register_type
);
341 src_param
->reg
.type
= WINED3DSPR_TEMP
;
345 src_param
->reg
.type
= register_type_table
[register_type
];
348 order
= (token
& WINED3D_SM4_REGISTER_ORDER_MASK
) >> WINED3D_SM4_REGISTER_ORDER_SHIFT
;
350 if (order
< 1) src_param
->reg
.idx
= ~0U;
351 else src_param
->reg
.idx
= *(*ptr
)++;
353 if (order
< 2) src_param
->reg
.array_idx
= ~0U;
354 else src_param
->reg
.array_idx
= *(*ptr
)++;
356 if (order
> 2) FIXME("Unhandled order %u.\n", order
);
358 if (register_type
== WINED3D_SM4_RT_IMMCONST
)
360 enum wined3d_sm4_immconst_type immconst_type
=
361 (token
& WINED3D_SM4_IMMCONST_TYPE_MASK
) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT
;
362 src_param
->swizzle
= WINED3DSP_NOSWIZZLE
;
364 switch(immconst_type
)
366 case WINED3D_SM4_IMMCONST_SCALAR
:
367 src_param
->reg
.immconst_type
= WINED3D_IMMCONST_SCALAR
;
368 memcpy(src_param
->reg
.immconst_data
, *ptr
, 1 * sizeof(DWORD
));
372 case WINED3D_SM4_IMMCONST_VEC4
:
373 src_param
->reg
.immconst_type
= WINED3D_IMMCONST_VEC4
;
374 memcpy(src_param
->reg
.immconst_data
, *ptr
, 4 * sizeof(DWORD
));
379 FIXME("Unhandled immediate constant type %#x\n", immconst_type
);
385 src_param
->swizzle
= (token
& WINED3D_SM4_SWIZZLE_MASK
) >> WINED3D_SM4_SWIZZLE_SHIFT
;
388 src_param
->modifiers
= 0;
389 src_param
->reg
.rel_addr
= NULL
;
391 map_register(priv
, &src_param
->reg
);
394 static void shader_sm4_read_dst_param(void *data
, const DWORD
**ptr
, struct wined3d_shader_dst_param
*dst_param
,
395 struct wined3d_shader_src_param
*dst_rel_addr
)
397 struct wined3d_sm4_data
*priv
= data
;
398 DWORD token
= *(*ptr
)++;
399 enum wined3d_sm4_register_type register_type
;
402 register_type
= (token
& WINED3D_SM4_REGISTER_TYPE_MASK
) >> WINED3D_SM4_REGISTER_TYPE_SHIFT
;
403 if (register_type
>= sizeof(register_type_table
) / sizeof(*register_type_table
))
405 FIXME("Unhandled register type %#x\n", register_type
);
406 dst_param
->reg
.type
= WINED3DSPR_TEMP
;
410 dst_param
->reg
.type
= register_type_table
[register_type
];
413 order
= (token
& WINED3D_SM4_REGISTER_ORDER_MASK
) >> WINED3D_SM4_REGISTER_ORDER_SHIFT
;
415 if (order
< 1) dst_param
->reg
.idx
= ~0U;
416 else dst_param
->reg
.idx
= *(*ptr
)++;
418 if (order
< 2) dst_param
->reg
.array_idx
= ~0U;
419 else dst_param
->reg
.array_idx
= *(*ptr
)++;
421 if (order
> 2) FIXME("Unhandled order %u.\n", order
);
423 dst_param
->write_mask
= (token
& WINED3D_SM4_WRITEMASK_MASK
) >> WINED3D_SM4_WRITEMASK_SHIFT
;
424 dst_param
->modifiers
= 0;
425 dst_param
->shift
= 0;
426 dst_param
->reg
.rel_addr
= NULL
;
428 map_register(priv
, &dst_param
->reg
);
431 static void shader_sm4_read_semantic(const DWORD
**ptr
, struct wined3d_shader_semantic
*semantic
)
433 FIXME("ptr %p, semantic %p stub!\n", ptr
, semantic
);
436 static void shader_sm4_read_comment(const DWORD
**ptr
, const char **comment
, UINT
*comment_size
)
438 FIXME("ptr %p, comment %p, comment_size %p stub!\n", ptr
, comment
, comment_size
);
442 static BOOL
shader_sm4_is_end(void *data
, const DWORD
**ptr
)
444 struct wined3d_sm4_data
*priv
= data
;
445 return *ptr
== priv
->end
;
448 const struct wined3d_shader_frontend sm4_shader_frontend
=
452 shader_sm4_read_header
,
453 shader_sm4_read_opcode
,
454 shader_sm4_read_src_param
,
455 shader_sm4_read_dst_param
,
456 shader_sm4_read_semantic
,
457 shader_sm4_read_comment
,