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_MODIFIER (1 << 31)
33 #define WINED3D_SM4_REGISTER_ORDER_SHIFT 20
34 #define WINED3D_SM4_REGISTER_ORDER_MASK (0x3 << WINED3D_SM4_REGISTER_ORDER_SHIFT)
36 #define WINED3D_SM4_REGISTER_TYPE_SHIFT 12
37 #define WINED3D_SM4_REGISTER_TYPE_MASK (0xf << WINED3D_SM4_REGISTER_TYPE_SHIFT)
39 #define WINED3D_SM4_IMMCONST_TYPE_SHIFT 0
40 #define WINED3D_SM4_IMMCONST_TYPE_MASK (0x3 << WINED3D_SM4_IMMCONST_TYPE_SHIFT)
42 #define WINED3D_SM4_WRITEMASK_SHIFT 4
43 #define WINED3D_SM4_WRITEMASK_MASK (0xf << WINED3D_SM4_WRITEMASK_SHIFT)
45 #define WINED3D_SM4_SWIZZLE_SHIFT 4
46 #define WINED3D_SM4_SWIZZLE_MASK (0xff << WINED3D_SM4_SWIZZLE_SHIFT)
48 #define WINED3D_SM4_VERSION_MAJOR(version) (((version) >> 4) & 0xf)
49 #define WINED3D_SM4_VERSION_MINOR(version) (((version) >> 0) & 0xf)
51 enum wined3d_sm4_opcode
53 WINED3D_SM4_OP_ADD
= 0x00,
54 WINED3D_SM4_OP_AND
= 0x01,
55 WINED3D_SM4_OP_BREAK
= 0x02,
56 WINED3D_SM4_OP_BREAKC
= 0x03,
57 WINED3D_SM4_OP_CUT
= 0x09,
58 WINED3D_SM4_OP_DP3
= 0x10,
59 WINED3D_SM4_OP_DP4
= 0x11,
60 WINED3D_SM4_OP_EMIT
= 0x13,
61 WINED3D_SM4_OP_ENDIF
= 0x15,
62 WINED3D_SM4_OP_ENDLOOP
= 0x16,
63 WINED3D_SM4_OP_EXP
= 0x19,
64 WINED3D_SM4_OP_FRC
= 0x1a,
65 WINED3D_SM4_OP_IADD
= 0x1e,
66 WINED3D_SM4_OP_IF
= 0x1f,
67 WINED3D_SM4_OP_IGE
= 0x21,
68 WINED3D_SM4_OP_IMUL
= 0x26,
69 WINED3D_SM4_OP_ITOF
= 0x2b,
70 WINED3D_SM4_OP_LOG
= 0x2f,
71 WINED3D_SM4_OP_LOOP
= 0x30,
72 WINED3D_SM4_OP_LT
= 0x31,
73 WINED3D_SM4_OP_MAD
= 0x32,
74 WINED3D_SM4_OP_MIN
= 0x33,
75 WINED3D_SM4_OP_MAX
= 0x34,
76 WINED3D_SM4_OP_MOV
= 0x36,
77 WINED3D_SM4_OP_MOVC
= 0x37,
78 WINED3D_SM4_OP_MUL
= 0x38,
79 WINED3D_SM4_OP_RET
= 0x3e,
80 WINED3D_SM4_OP_RSQ
= 0x44,
81 WINED3D_SM4_OP_SAMPLE
= 0x45,
82 WINED3D_SM4_OP_SAMPLE_LOD
= 0x48,
83 WINED3D_SM4_OP_SINCOS
= 0x4d,
84 WINED3D_SM4_OP_UTOF
= 0x56,
87 enum wined3d_sm4_register_type
89 WINED3D_SM4_RT_TEMP
= 0x0,
90 WINED3D_SM4_RT_INPUT
= 0x1,
91 WINED3D_SM4_RT_OUTPUT
= 0x2,
92 WINED3D_SM4_RT_IMMCONST
= 0x4,
93 WINED3D_SM4_RT_SAMPLER
= 0x6,
94 WINED3D_SM4_RT_CONSTBUFFER
= 0x8,
95 WINED3D_SM4_RT_NULL
= 0xd,
98 enum wined3d_sm4_immconst_type
100 WINED3D_SM4_IMMCONST_SCALAR
= 0x1,
101 WINED3D_SM4_IMMCONST_VEC4
= 0x2,
104 struct wined3d_sm4_data
106 struct wined3d_shader_version shader_version
;
108 const struct wined3d_shader_signature
*output_signature
;
111 struct wined3d_sm4_opcode_info
113 enum wined3d_sm4_opcode opcode
;
114 enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx
;
121 enum wined3d_sysval_semantic sysval
;
122 WINED3DSHADER_PARAM_REGISTER_TYPE register_type
;
126 static const struct wined3d_sm4_opcode_info opcode_table
[] =
128 {WINED3D_SM4_OP_ADD
, WINED3DSIH_ADD
, 1, 2},
129 {WINED3D_SM4_OP_AND
, WINED3DSIH_AND
, 1, 2},
130 {WINED3D_SM4_OP_BREAK
, WINED3DSIH_BREAK
, 0, 0},
131 {WINED3D_SM4_OP_BREAKC
, WINED3DSIH_BREAKP
, 0, 1},
132 {WINED3D_SM4_OP_CUT
, WINED3DSIH_CUT
, 0, 0},
133 {WINED3D_SM4_OP_DP3
, WINED3DSIH_DP3
, 1, 2},
134 {WINED3D_SM4_OP_DP4
, WINED3DSIH_DP4
, 1, 2},
135 {WINED3D_SM4_OP_EMIT
, WINED3DSIH_EMIT
, 0, 0},
136 {WINED3D_SM4_OP_ENDIF
, WINED3DSIH_ENDIF
, 0, 0},
137 {WINED3D_SM4_OP_ENDLOOP
, WINED3DSIH_ENDLOOP
, 0, 0},
138 {WINED3D_SM4_OP_EXP
, WINED3DSIH_EXP
, 1, 1},
139 {WINED3D_SM4_OP_FRC
, WINED3DSIH_FRC
, 1, 1},
140 {WINED3D_SM4_OP_IADD
, WINED3DSIH_IADD
, 1, 2},
141 {WINED3D_SM4_OP_IF
, WINED3DSIH_IF
, 0, 1},
142 {WINED3D_SM4_OP_IGE
, WINED3DSIH_IGE
, 1, 2},
143 {WINED3D_SM4_OP_IMUL
, WINED3DSIH_IMUL
, 2, 2},
144 {WINED3D_SM4_OP_ITOF
, WINED3DSIH_ITOF
, 1, 1},
145 {WINED3D_SM4_OP_LOG
, WINED3DSIH_LOG
, 1, 1},
146 {WINED3D_SM4_OP_LOOP
, WINED3DSIH_LOOP
, 0, 0},
147 {WINED3D_SM4_OP_LT
, WINED3DSIH_LT
, 1, 2},
148 {WINED3D_SM4_OP_MAD
, WINED3DSIH_MAD
, 1, 3},
149 {WINED3D_SM4_OP_MIN
, WINED3DSIH_MIN
, 1, 2},
150 {WINED3D_SM4_OP_MAX
, WINED3DSIH_MAX
, 1, 2},
151 {WINED3D_SM4_OP_MOV
, WINED3DSIH_MOV
, 1, 1},
152 {WINED3D_SM4_OP_MOVC
, WINED3DSIH_MOVC
, 1, 3},
153 {WINED3D_SM4_OP_MUL
, WINED3DSIH_MUL
, 1, 2},
154 {WINED3D_SM4_OP_RET
, WINED3DSIH_RET
, 0, 0},
155 {WINED3D_SM4_OP_RSQ
, WINED3DSIH_RSQ
, 1, 1},
156 {WINED3D_SM4_OP_SAMPLE
, WINED3DSIH_SAMPLE
, 1, 3},
157 {WINED3D_SM4_OP_SAMPLE_LOD
, WINED3DSIH_SAMPLE_LOD
, 1, 4},
158 {WINED3D_SM4_OP_SINCOS
, WINED3DSIH_SINCOS
, 2, 1},
159 {WINED3D_SM4_OP_UTOF
, WINED3DSIH_UTOF
, 1, 1},
162 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table
[] =
164 /* WINED3D_SM4_RT_TEMP */ WINED3DSPR_TEMP
,
165 /* WINED3D_SM4_RT_INPUT */ WINED3DSPR_INPUT
,
166 /* WINED3D_SM4_RT_OUTPUT */ WINED3DSPR_OUTPUT
,
168 /* WINED3D_SM4_RT_IMMCONST */ WINED3DSPR_IMMCONST
,
170 /* WINED3D_SM4_RT_SAMPLER */ WINED3DSPR_SAMPLER
,
171 /* WINED3D_SM4_RT_RESOURCE */ WINED3DSPR_RESOURCE
,
172 /* WINED3D_SM4_RT_CONSTBUFFER */ WINED3DSPR_CONSTBUFFER
,
177 /* WINED3D_SM4_RT_NULL */ WINED3DSPR_NULL
,
180 static const struct sysval_map sysval_map
[] =
182 {WINED3D_SV_DEPTH
, WINED3DSPR_DEPTHOUT
, 0},
183 {WINED3D_SV_TARGET0
, WINED3DSPR_COLOROUT
, 0},
184 {WINED3D_SV_TARGET1
, WINED3DSPR_COLOROUT
, 1},
185 {WINED3D_SV_TARGET2
, WINED3DSPR_COLOROUT
, 2},
186 {WINED3D_SV_TARGET3
, WINED3DSPR_COLOROUT
, 3},
187 {WINED3D_SV_TARGET4
, WINED3DSPR_COLOROUT
, 4},
188 {WINED3D_SV_TARGET5
, WINED3DSPR_COLOROUT
, 5},
189 {WINED3D_SV_TARGET6
, WINED3DSPR_COLOROUT
, 6},
190 {WINED3D_SV_TARGET7
, WINED3DSPR_COLOROUT
, 7},
193 static const struct wined3d_sm4_opcode_info
*get_opcode_info(enum wined3d_sm4_opcode opcode
)
197 for (i
= 0; i
< sizeof(opcode_table
) / sizeof(*opcode_table
); ++i
)
199 if (opcode
== opcode_table
[i
].opcode
) return &opcode_table
[i
];
205 static void map_sysval(enum wined3d_sysval_semantic sysval
, struct wined3d_shader_register
*reg
)
209 for (i
= 0; i
< sizeof(sysval_map
) / sizeof(*sysval_map
); ++i
)
211 if (sysval
== sysval_map
[i
].sysval
)
213 reg
->type
= sysval_map
[i
].register_type
;
214 reg
->idx
= sysval_map
[i
].register_idx
;
219 static void map_register(struct wined3d_sm4_data
*priv
, struct wined3d_shader_register
*reg
)
221 switch (priv
->shader_version
.type
)
223 case WINED3D_SHADER_TYPE_PIXEL
:
224 if (reg
->type
== WINED3DSPR_OUTPUT
)
227 const struct wined3d_shader_signature
*s
= priv
->output_signature
;
231 ERR("Shader has no output signature, unable to map register.\n");
235 for (i
= 0; i
< s
->element_count
; ++i
)
237 if (s
->elements
[i
].register_idx
== reg
->idx
)
239 map_sysval(s
->elements
[i
].sysval_semantic
, reg
);
251 static void *shader_sm4_init(const DWORD
*byte_code
, const struct wined3d_shader_signature
*output_signature
)
253 struct wined3d_sm4_data
*priv
= HeapAlloc(GetProcessHeap(), 0, sizeof(*priv
));
256 ERR("Failed to allocate private data\n");
260 priv
->output_signature
= output_signature
;
265 static void shader_sm4_free(void *data
)
267 HeapFree(GetProcessHeap(), 0, data
);
270 static void shader_sm4_read_header(void *data
, const DWORD
**ptr
, struct wined3d_shader_version
*shader_version
)
272 struct wined3d_sm4_data
*priv
= data
;
277 version_token
= *(*ptr
)++;
278 TRACE("version: 0x%08x\n", version_token
);
280 TRACE("token count: %u\n", **ptr
);
281 priv
->end
+= *(*ptr
)++;
283 switch (version_token
>> 16)
286 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_PIXEL
;
290 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_VERTEX
;
294 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_GEOMETRY
;
298 FIXME("Unrecognized shader type %#x\n", version_token
>> 16);
300 priv
->shader_version
.major
= WINED3D_SM4_VERSION_MAJOR(version_token
);
301 priv
->shader_version
.minor
= WINED3D_SM4_VERSION_MINOR(version_token
);
303 *shader_version
= priv
->shader_version
;
306 static void shader_sm4_read_opcode(void *data
, const DWORD
**ptr
, struct wined3d_shader_instruction
*ins
,
309 const struct wined3d_sm4_opcode_info
*opcode_info
;
310 DWORD token
= *(*ptr
)++;
311 DWORD opcode
= token
& WINED3D_SM4_OPCODE_MASK
;
313 *param_size
= ((token
& WINED3D_SM4_INSTRUCTION_LENGTH_MASK
) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT
) - 1;
315 opcode_info
= get_opcode_info(opcode
);
318 FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode
, token
);
319 ins
->handler_idx
= WINED3DSIH_TABLE_SIZE
;
323 ins
->handler_idx
= opcode_info
->handler_idx
;
327 ins
->dst_count
= opcode_info
->dst_count
;
328 ins
->src_count
= opcode_info
->src_count
;
331 static void shader_sm4_read_src_param(void *data
, const DWORD
**ptr
, struct wined3d_shader_src_param
*src_param
,
332 struct wined3d_shader_src_param
*src_rel_addr
)
334 struct wined3d_sm4_data
*priv
= data
;
335 DWORD token
= *(*ptr
)++;
336 enum wined3d_sm4_register_type register_type
;
339 register_type
= (token
& WINED3D_SM4_REGISTER_TYPE_MASK
) >> WINED3D_SM4_REGISTER_TYPE_SHIFT
;
340 if (register_type
>= sizeof(register_type_table
) / sizeof(*register_type_table
))
342 FIXME("Unhandled register type %#x\n", register_type
);
343 src_param
->reg
.type
= WINED3DSPR_TEMP
;
347 src_param
->reg
.type
= register_type_table
[register_type
];
350 if (token
& WINED3D_SM4_REGISTER_MODIFIER
)
352 DWORD modifier
= *(*ptr
)++;
353 FIXME("Skipping modifier 0x%08x.\n", modifier
);
356 order
= (token
& WINED3D_SM4_REGISTER_ORDER_MASK
) >> WINED3D_SM4_REGISTER_ORDER_SHIFT
;
358 if (order
< 1) src_param
->reg
.idx
= ~0U;
359 else src_param
->reg
.idx
= *(*ptr
)++;
361 if (order
< 2) src_param
->reg
.array_idx
= ~0U;
362 else src_param
->reg
.array_idx
= *(*ptr
)++;
364 if (order
> 2) FIXME("Unhandled order %u.\n", order
);
366 if (register_type
== WINED3D_SM4_RT_IMMCONST
)
368 enum wined3d_sm4_immconst_type immconst_type
=
369 (token
& WINED3D_SM4_IMMCONST_TYPE_MASK
) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT
;
370 src_param
->swizzle
= WINED3DSP_NOSWIZZLE
;
372 switch(immconst_type
)
374 case WINED3D_SM4_IMMCONST_SCALAR
:
375 src_param
->reg
.immconst_type
= WINED3D_IMMCONST_SCALAR
;
376 memcpy(src_param
->reg
.immconst_data
, *ptr
, 1 * sizeof(DWORD
));
380 case WINED3D_SM4_IMMCONST_VEC4
:
381 src_param
->reg
.immconst_type
= WINED3D_IMMCONST_VEC4
;
382 memcpy(src_param
->reg
.immconst_data
, *ptr
, 4 * sizeof(DWORD
));
387 FIXME("Unhandled immediate constant type %#x\n", immconst_type
);
393 src_param
->swizzle
= (token
& WINED3D_SM4_SWIZZLE_MASK
) >> WINED3D_SM4_SWIZZLE_SHIFT
;
396 src_param
->modifiers
= 0;
397 src_param
->reg
.rel_addr
= NULL
;
399 map_register(priv
, &src_param
->reg
);
402 static void shader_sm4_read_dst_param(void *data
, const DWORD
**ptr
, struct wined3d_shader_dst_param
*dst_param
,
403 struct wined3d_shader_src_param
*dst_rel_addr
)
405 struct wined3d_sm4_data
*priv
= data
;
406 DWORD token
= *(*ptr
)++;
407 enum wined3d_sm4_register_type register_type
;
410 register_type
= (token
& WINED3D_SM4_REGISTER_TYPE_MASK
) >> WINED3D_SM4_REGISTER_TYPE_SHIFT
;
411 if (register_type
>= sizeof(register_type_table
) / sizeof(*register_type_table
))
413 FIXME("Unhandled register type %#x\n", register_type
);
414 dst_param
->reg
.type
= WINED3DSPR_TEMP
;
418 dst_param
->reg
.type
= register_type_table
[register_type
];
421 order
= (token
& WINED3D_SM4_REGISTER_ORDER_MASK
) >> WINED3D_SM4_REGISTER_ORDER_SHIFT
;
423 if (order
< 1) dst_param
->reg
.idx
= ~0U;
424 else dst_param
->reg
.idx
= *(*ptr
)++;
426 if (order
< 2) dst_param
->reg
.array_idx
= ~0U;
427 else dst_param
->reg
.array_idx
= *(*ptr
)++;
429 if (order
> 2) FIXME("Unhandled order %u.\n", order
);
431 dst_param
->write_mask
= (token
& WINED3D_SM4_WRITEMASK_MASK
) >> WINED3D_SM4_WRITEMASK_SHIFT
;
432 dst_param
->modifiers
= 0;
433 dst_param
->shift
= 0;
434 dst_param
->reg
.rel_addr
= NULL
;
436 map_register(priv
, &dst_param
->reg
);
439 static void shader_sm4_read_semantic(const DWORD
**ptr
, struct wined3d_shader_semantic
*semantic
)
441 FIXME("ptr %p, semantic %p stub!\n", ptr
, semantic
);
444 static void shader_sm4_read_comment(const DWORD
**ptr
, const char **comment
, UINT
*comment_size
)
446 FIXME("ptr %p, comment %p, comment_size %p stub!\n", ptr
, comment
, comment_size
);
450 static BOOL
shader_sm4_is_end(void *data
, const DWORD
**ptr
)
452 struct wined3d_sm4_data
*priv
= data
;
453 return *ptr
== priv
->end
;
456 const struct wined3d_shader_frontend sm4_shader_frontend
=
460 shader_sm4_read_header
,
461 shader_sm4_read_opcode
,
462 shader_sm4_read_src_param
,
463 shader_sm4_read_dst_param
,
464 shader_sm4_read_semantic
,
465 shader_sm4_read_comment
,