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_LOG
= 0x2f,
68 WINED3D_SM4_OP_LOOP
= 0x30,
69 WINED3D_SM4_OP_LT
= 0x31,
70 WINED3D_SM4_OP_MAD
= 0x32,
71 WINED3D_SM4_OP_MIN
= 0x33,
72 WINED3D_SM4_OP_MAX
= 0x34,
73 WINED3D_SM4_OP_MOV
= 0x36,
74 WINED3D_SM4_OP_MOVC
= 0x37,
75 WINED3D_SM4_OP_MUL
= 0x38,
76 WINED3D_SM4_OP_RET
= 0x3e,
77 WINED3D_SM4_OP_RSQ
= 0x44,
78 WINED3D_SM4_OP_SINCOS
= 0x4d,
81 enum wined3d_sm4_register_type
83 WINED3D_SM4_RT_TEMP
= 0x0,
84 WINED3D_SM4_RT_INPUT
= 0x1,
85 WINED3D_SM4_RT_OUTPUT
= 0x2,
86 WINED3D_SM4_RT_IMMCONST
= 0x4,
87 WINED3D_SM4_RT_CONSTBUFFER
= 0x8,
88 WINED3D_SM4_RT_NULL
= 0xd,
91 enum wined3d_sm4_immconst_type
93 WINED3D_SM4_IMMCONST_FLOAT
= 0x1,
94 WINED3D_SM4_IMMCONST_FLOAT4
= 0x2,
97 struct wined3d_sm4_data
99 struct wined3d_shader_version shader_version
;
101 const struct wined3d_shader_signature
*output_signature
;
104 struct wined3d_sm4_opcode_info
106 enum wined3d_sm4_opcode opcode
;
107 enum WINED3D_SHADER_INSTRUCTION_HANDLER handler_idx
;
114 enum wined3d_sysval_semantic sysval
;
115 WINED3DSHADER_PARAM_REGISTER_TYPE register_type
;
119 static const struct wined3d_sm4_opcode_info opcode_table
[] =
121 {WINED3D_SM4_OP_ADD
, WINED3DSIH_ADD
, 1, 2},
122 {WINED3D_SM4_OP_AND
, WINED3DSIH_AND
, 1, 2},
123 {WINED3D_SM4_OP_BREAK
, WINED3DSIH_BREAK
, 0, 0},
124 {WINED3D_SM4_OP_BREAKC
, WINED3DSIH_BREAKP
, 0, 1},
125 {WINED3D_SM4_OP_CUT
, WINED3DSIH_CUT
, 0, 0},
126 {WINED3D_SM4_OP_DP3
, WINED3DSIH_DP3
, 1, 2},
127 {WINED3D_SM4_OP_DP4
, WINED3DSIH_DP4
, 1, 2},
128 {WINED3D_SM4_OP_EMIT
, WINED3DSIH_EMIT
, 0, 0},
129 {WINED3D_SM4_OP_ENDIF
, WINED3DSIH_ENDIF
, 0, 0},
130 {WINED3D_SM4_OP_ENDLOOP
,WINED3DSIH_ENDLOOP
, 0, 0},
131 {WINED3D_SM4_OP_EXP
, WINED3DSIH_EXP
, 1, 1},
132 {WINED3D_SM4_OP_FRC
, WINED3DSIH_FRC
, 1, 1},
133 {WINED3D_SM4_OP_IADD
, WINED3DSIH_IADD
, 1, 2},
134 {WINED3D_SM4_OP_IF
, WINED3DSIH_IF
, 0, 1},
135 {WINED3D_SM4_OP_IGE
, WINED3DSIH_IGE
, 1, 2},
136 {WINED3D_SM4_OP_IMUL
, WINED3DSIH_IMUL
, 2, 2},
137 {WINED3D_SM4_OP_LOG
, WINED3DSIH_LOG
, 1, 1},
138 {WINED3D_SM4_OP_LOOP
, WINED3DSIH_LOOP
, 0, 0},
139 {WINED3D_SM4_OP_LT
, WINED3DSIH_LT
, 1, 2},
140 {WINED3D_SM4_OP_MAD
, WINED3DSIH_MAD
, 1, 3},
141 {WINED3D_SM4_OP_MIN
, WINED3DSIH_MIN
, 1, 2},
142 {WINED3D_SM4_OP_MAX
, WINED3DSIH_MAX
, 1, 2},
143 {WINED3D_SM4_OP_MOV
, WINED3DSIH_MOV
, 1, 1},
144 {WINED3D_SM4_OP_MOVC
, WINED3DSIH_MOVC
, 1, 3},
145 {WINED3D_SM4_OP_MUL
, WINED3DSIH_MUL
, 1, 2},
146 {WINED3D_SM4_OP_RET
, WINED3DSIH_RET
, 0, 0},
147 {WINED3D_SM4_OP_RSQ
, WINED3DSIH_RSQ
, 1, 1},
148 {WINED3D_SM4_OP_SINCOS
, WINED3DSIH_SINCOS
, 2, 1},
151 static const WINED3DSHADER_PARAM_REGISTER_TYPE register_type_table
[] =
153 /* WINED3D_SM4_RT_TEMP */ WINED3DSPR_TEMP
,
154 /* WINED3D_SM4_RT_INPUT */ WINED3DSPR_INPUT
,
155 /* WINED3D_SM4_RT_OUTPUT */ WINED3DSPR_OUTPUT
,
157 /* WINED3D_SM4_RT_IMMCONST */ WINED3DSPR_IMMCONST
,
161 /* WINED3D_SM4_RT_CONSTBUFFER */ WINED3DSPR_CONSTBUFFER
,
166 /* WINED3D_SM4_RT_NULL */ WINED3DSPR_NULL
,
169 static const struct sysval_map sysval_map
[] =
171 {WINED3D_SV_DEPTH
, WINED3DSPR_DEPTHOUT
, 0},
172 {WINED3D_SV_TARGET0
, WINED3DSPR_COLOROUT
, 0},
173 {WINED3D_SV_TARGET1
, WINED3DSPR_COLOROUT
, 1},
174 {WINED3D_SV_TARGET2
, WINED3DSPR_COLOROUT
, 2},
175 {WINED3D_SV_TARGET3
, WINED3DSPR_COLOROUT
, 3},
176 {WINED3D_SV_TARGET4
, WINED3DSPR_COLOROUT
, 4},
177 {WINED3D_SV_TARGET5
, WINED3DSPR_COLOROUT
, 5},
178 {WINED3D_SV_TARGET6
, WINED3DSPR_COLOROUT
, 6},
179 {WINED3D_SV_TARGET7
, WINED3DSPR_COLOROUT
, 7},
182 static const struct wined3d_sm4_opcode_info
*get_opcode_info(enum wined3d_sm4_opcode opcode
)
186 for (i
= 0; i
< sizeof(opcode_table
) / sizeof(*opcode_table
); ++i
)
188 if (opcode
== opcode_table
[i
].opcode
) return &opcode_table
[i
];
194 static void map_sysval(enum wined3d_sysval_semantic sysval
, struct wined3d_shader_register
*reg
)
198 for (i
= 0; i
< sizeof(sysval_map
) / sizeof(*sysval_map
); ++i
)
200 if (sysval
== sysval_map
[i
].sysval
)
202 reg
->type
= sysval_map
[i
].register_type
;
203 reg
->idx
= sysval_map
[i
].register_idx
;
208 static void map_register(struct wined3d_sm4_data
*priv
, struct wined3d_shader_register
*reg
)
210 switch (priv
->shader_version
.type
)
212 case WINED3D_SHADER_TYPE_PIXEL
:
213 if (reg
->type
== WINED3DSPR_OUTPUT
)
216 const struct wined3d_shader_signature
*s
= priv
->output_signature
;
220 ERR("Shader has no output signature, unable to map register.\n");
224 for (i
= 0; i
< s
->element_count
; ++i
)
226 if (s
->elements
[i
].register_idx
== reg
->idx
)
228 map_sysval(s
->elements
[i
].sysval_semantic
, reg
);
240 static void *shader_sm4_init(const DWORD
*byte_code
, const struct wined3d_shader_signature
*output_signature
)
242 struct wined3d_sm4_data
*priv
= HeapAlloc(GetProcessHeap(), 0, sizeof(*priv
));
245 ERR("Failed to allocate private data\n");
249 priv
->output_signature
= output_signature
;
254 static void shader_sm4_free(void *data
)
256 HeapFree(GetProcessHeap(), 0, data
);
259 static void shader_sm4_read_header(void *data
, const DWORD
**ptr
, struct wined3d_shader_version
*shader_version
)
261 struct wined3d_sm4_data
*priv
= data
;
266 version_token
= *(*ptr
)++;
267 TRACE("version: 0x%08x\n", version_token
);
269 TRACE("token count: %u\n", **ptr
);
270 priv
->end
+= *(*ptr
)++;
272 switch (version_token
>> 16)
275 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_PIXEL
;
279 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_VERTEX
;
283 priv
->shader_version
.type
= WINED3D_SHADER_TYPE_GEOMETRY
;
287 FIXME("Unrecognized shader type %#x\n", version_token
>> 16);
289 priv
->shader_version
.major
= WINED3D_SM4_VERSION_MAJOR(version_token
);
290 priv
->shader_version
.minor
= WINED3D_SM4_VERSION_MINOR(version_token
);
292 *shader_version
= priv
->shader_version
;
295 static void shader_sm4_read_opcode(void *data
, const DWORD
**ptr
, struct wined3d_shader_instruction
*ins
,
298 const struct wined3d_sm4_opcode_info
*opcode_info
;
299 DWORD token
= *(*ptr
)++;
300 DWORD opcode
= token
& WINED3D_SM4_OPCODE_MASK
;
302 *param_size
= ((token
& WINED3D_SM4_INSTRUCTION_LENGTH_MASK
) >> WINED3D_SM4_INSTRUCTION_LENGTH_SHIFT
) - 1;
304 opcode_info
= get_opcode_info(opcode
);
307 FIXME("Unrecognized opcode %#x, token 0x%08x\n", opcode
, token
);
308 ins
->handler_idx
= WINED3DSIH_TABLE_SIZE
;
312 ins
->handler_idx
= opcode_info
->handler_idx
;
316 ins
->dst_count
= opcode_info
->dst_count
;
317 ins
->src_count
= opcode_info
->src_count
;
320 static void shader_sm4_read_src_param(void *data
, const DWORD
**ptr
, struct wined3d_shader_src_param
*src_param
,
321 struct wined3d_shader_src_param
*src_rel_addr
)
323 struct wined3d_sm4_data
*priv
= data
;
324 DWORD token
= *(*ptr
)++;
325 enum wined3d_sm4_register_type register_type
;
328 register_type
= (token
& WINED3D_SM4_REGISTER_TYPE_MASK
) >> WINED3D_SM4_REGISTER_TYPE_SHIFT
;
329 if (register_type
>= sizeof(register_type_table
) / sizeof(*register_type_table
))
331 FIXME("Unhandled register type %#x\n", register_type
);
332 src_param
->reg
.type
= WINED3DSPR_TEMP
;
336 src_param
->reg
.type
= register_type_table
[register_type
];
339 order
= (token
& WINED3D_SM4_REGISTER_ORDER_MASK
) >> WINED3D_SM4_REGISTER_ORDER_SHIFT
;
341 if (order
< 1) src_param
->reg
.idx
= ~0U;
342 else src_param
->reg
.idx
= *(*ptr
)++;
344 if (order
< 2) src_param
->reg
.array_idx
= ~0U;
345 else src_param
->reg
.array_idx
= *(*ptr
)++;
347 if (order
> 2) FIXME("Unhandled order %u.\n", order
);
349 if (register_type
== WINED3D_SM4_RT_IMMCONST
)
351 enum wined3d_sm4_immconst_type immconst_type
=
352 (token
& WINED3D_SM4_IMMCONST_TYPE_MASK
) >> WINED3D_SM4_IMMCONST_TYPE_SHIFT
;
353 src_param
->swizzle
= WINED3DSP_NOSWIZZLE
;
355 switch(immconst_type
)
357 case WINED3D_SM4_IMMCONST_FLOAT
:
358 src_param
->reg
.immconst_type
= WINED3D_IMMCONST_FLOAT
;
359 memcpy(src_param
->reg
.immconst_data
, *ptr
, 1 * sizeof(DWORD
));
363 case WINED3D_SM4_IMMCONST_FLOAT4
:
364 src_param
->reg
.immconst_type
= WINED3D_IMMCONST_FLOAT4
;
365 memcpy(src_param
->reg
.immconst_data
, *ptr
, 4 * sizeof(DWORD
));
370 FIXME("Unhandled immediate constant type %#x\n", immconst_type
);
376 src_param
->swizzle
= (token
& WINED3D_SM4_SWIZZLE_MASK
) >> WINED3D_SM4_SWIZZLE_SHIFT
;
379 src_param
->modifiers
= 0;
380 src_param
->reg
.rel_addr
= NULL
;
382 map_register(priv
, &src_param
->reg
);
385 static void shader_sm4_read_dst_param(void *data
, const DWORD
**ptr
, struct wined3d_shader_dst_param
*dst_param
,
386 struct wined3d_shader_src_param
*dst_rel_addr
)
388 struct wined3d_sm4_data
*priv
= data
;
389 DWORD token
= *(*ptr
)++;
390 enum wined3d_sm4_register_type register_type
;
393 register_type
= (token
& WINED3D_SM4_REGISTER_TYPE_MASK
) >> WINED3D_SM4_REGISTER_TYPE_SHIFT
;
394 if (register_type
>= sizeof(register_type_table
) / sizeof(*register_type_table
))
396 FIXME("Unhandled register type %#x\n", register_type
);
397 dst_param
->reg
.type
= WINED3DSPR_TEMP
;
401 dst_param
->reg
.type
= register_type_table
[register_type
];
404 order
= (token
& WINED3D_SM4_REGISTER_ORDER_MASK
) >> WINED3D_SM4_REGISTER_ORDER_SHIFT
;
406 if (order
< 1) dst_param
->reg
.idx
= ~0U;
407 else dst_param
->reg
.idx
= *(*ptr
)++;
409 if (order
< 2) dst_param
->reg
.array_idx
= ~0U;
410 else dst_param
->reg
.array_idx
= *(*ptr
)++;
412 if (order
> 2) FIXME("Unhandled order %u.\n", order
);
414 dst_param
->write_mask
= (token
& WINED3D_SM4_WRITEMASK_MASK
) >> WINED3D_SM4_WRITEMASK_SHIFT
;
415 dst_param
->modifiers
= 0;
416 dst_param
->shift
= 0;
417 dst_param
->reg
.rel_addr
= NULL
;
419 map_register(priv
, &dst_param
->reg
);
422 static void shader_sm4_read_semantic(const DWORD
**ptr
, struct wined3d_shader_semantic
*semantic
)
424 FIXME("ptr %p, semantic %p stub!\n", ptr
, semantic
);
427 static void shader_sm4_read_comment(const DWORD
**ptr
, const char **comment
, UINT
*comment_size
)
429 FIXME("ptr %p, comment %p, comment_size %p stub!\n", ptr
, comment
, comment_size
);
433 static BOOL
shader_sm4_is_end(void *data
, const DWORD
**ptr
)
435 struct wined3d_sm4_data
*priv
= data
;
436 return *ptr
== priv
->end
;
439 const struct wined3d_shader_frontend sm4_shader_frontend
=
443 shader_sm4_read_header
,
444 shader_sm4_read_opcode
,
445 shader_sm4_read_src_param
,
446 shader_sm4_read_dst_param
,
447 shader_sm4_read_semantic
,
448 shader_sm4_read_comment
,