?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
po: Update Lithuanian translation.
[wine.git] / dlls / d3dcompiler_43 / utils.c
blob1436ed6bf3e95fa23ddacf5723ca56bfb7bc42e9
1 /*
2 * Copyright 2008 Stefan Dösinger
3 * Copyright 2009 Matteo Bruni
4 * Copyright 2008-2009 Henri Verbeet for CodeWeavers
5 * Copyright 2010 Rico Schüller
6 * Copyright 2012 Matteo Bruni for CodeWeavers
7 *
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
12 *
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
21 *
22 */
23
24 #include <stdio.h>
25
26 #include "d3dcompiler_private.h"
27
28 WINE_DEFAULT_DEBUG_CHANNEL(d3dcompiler);
29
30 #define WINE_D3DCOMPILER_TO_STR(x) case x: return #x
31
32 const char *debug_d3dcompiler_shader_variable_class(D3D_SHADER_VARIABLE_CLASS c)
33 {
34 switch (c)
35 {
36 WINE_D3DCOMPILER_TO_STR(D3D_SVC_SCALAR);
37 WINE_D3DCOMPILER_TO_STR(D3D_SVC_VECTOR);
38 WINE_D3DCOMPILER_TO_STR(D3D_SVC_MATRIX_ROWS);
39 WINE_D3DCOMPILER_TO_STR(D3D_SVC_MATRIX_COLUMNS);
40 WINE_D3DCOMPILER_TO_STR(D3D_SVC_OBJECT);
41 WINE_D3DCOMPILER_TO_STR(D3D_SVC_STRUCT);
42 WINE_D3DCOMPILER_TO_STR(D3D_SVC_INTERFACE_CLASS);
43 WINE_D3DCOMPILER_TO_STR(D3D_SVC_INTERFACE_POINTER);
44 default:
45 FIXME("Unrecognized D3D_SHADER_VARIABLE_CLASS %#x.\n", c);
46 return "unrecognized";
47 }
48 }
49
50 const char *debug_d3dcompiler_shader_variable_type(D3D_SHADER_VARIABLE_TYPE t)
51 {
52 switch (t)
53 {
54 WINE_D3DCOMPILER_TO_STR(D3D_SVT_VOID);
55 WINE_D3DCOMPILER_TO_STR(D3D_SVT_BOOL);
56 WINE_D3DCOMPILER_TO_STR(D3D_SVT_INT);
57 WINE_D3DCOMPILER_TO_STR(D3D_SVT_FLOAT);
58 WINE_D3DCOMPILER_TO_STR(D3D_SVT_STRING);
59 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE);
60 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE1D);
61 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE2D);
62 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE3D);
63 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURECUBE);
64 WINE_D3DCOMPILER_TO_STR(D3D_SVT_SAMPLER);
65 WINE_D3DCOMPILER_TO_STR(D3D_SVT_PIXELSHADER);
66 WINE_D3DCOMPILER_TO_STR(D3D_SVT_VERTEXSHADER);
67 WINE_D3DCOMPILER_TO_STR(D3D_SVT_UINT);
68 WINE_D3DCOMPILER_TO_STR(D3D_SVT_UINT8);
69 WINE_D3DCOMPILER_TO_STR(D3D_SVT_GEOMETRYSHADER);
70 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RASTERIZER);
71 WINE_D3DCOMPILER_TO_STR(D3D_SVT_DEPTHSTENCIL);
72 WINE_D3DCOMPILER_TO_STR(D3D_SVT_BLEND);
73 WINE_D3DCOMPILER_TO_STR(D3D_SVT_BUFFER);
74 WINE_D3DCOMPILER_TO_STR(D3D_SVT_CBUFFER);
75 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TBUFFER);
76 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE1DARRAY);
77 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE2DARRAY);
78 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RENDERTARGETVIEW);
79 WINE_D3DCOMPILER_TO_STR(D3D_SVT_DEPTHSTENCILVIEW);
80 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE2DMS);
81 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURE2DMSARRAY);
82 WINE_D3DCOMPILER_TO_STR(D3D_SVT_TEXTURECUBEARRAY);
83 WINE_D3DCOMPILER_TO_STR(D3D_SVT_HULLSHADER);
84 WINE_D3DCOMPILER_TO_STR(D3D_SVT_DOMAINSHADER);
85 WINE_D3DCOMPILER_TO_STR(D3D_SVT_INTERFACE_POINTER);
86 WINE_D3DCOMPILER_TO_STR(D3D_SVT_COMPUTESHADER);
87 WINE_D3DCOMPILER_TO_STR(D3D_SVT_DOUBLE);
88 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWTEXTURE1D);
89 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWTEXTURE1DARRAY);
90 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWTEXTURE2D);
91 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWTEXTURE2DARRAY);
92 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWTEXTURE3D);
93 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWBUFFER);
94 WINE_D3DCOMPILER_TO_STR(D3D_SVT_BYTEADDRESS_BUFFER);
95 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWBYTEADDRESS_BUFFER);
96 WINE_D3DCOMPILER_TO_STR(D3D_SVT_STRUCTURED_BUFFER);
97 WINE_D3DCOMPILER_TO_STR(D3D_SVT_RWSTRUCTURED_BUFFER);
98 WINE_D3DCOMPILER_TO_STR(D3D_SVT_APPEND_STRUCTURED_BUFFER);
99 WINE_D3DCOMPILER_TO_STR(D3D_SVT_CONSUME_STRUCTURED_BUFFER);
100 default:
101 FIXME("Unrecognized D3D_SHADER_VARIABLE_TYPE %#x.\n", t);
102 return "unrecognized";
103 }
104 }
105
106 const char *debug_d3dcompiler_d3d_blob_part(D3D_BLOB_PART part)
107 {
108 switch(part)
109 {
110 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_INPUT_SIGNATURE_BLOB);
111 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_OUTPUT_SIGNATURE_BLOB);
112 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_INPUT_AND_OUTPUT_SIGNATURE_BLOB);
113 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_PATCH_CONSTANT_SIGNATURE_BLOB);
114 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_ALL_SIGNATURE_BLOB);
115 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_DEBUG_INFO);
116 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_LEGACY_SHADER);
117 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_XNA_PREPASS_SHADER);
118 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_XNA_SHADER);
119 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_TEST_ALTERNATE_SHADER);
120 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_TEST_COMPILE_DETAILS);
121 WINE_D3DCOMPILER_TO_STR(D3D_BLOB_TEST_COMPILE_PERF);
122 default:
123 FIXME("Unrecognized D3D_BLOB_PART %#x\n", part);
124 return "unrecognized";
125 }
126 }
127
128 const char *debug_print_srcmod(DWORD mod)
129 {
130 switch (mod)
131 {
132 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_NEG);
133 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_BIAS);
134 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_BIASNEG);
135 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_SIGN);
136 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_SIGNNEG);
137 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_COMP);
138 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_X2);
139 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_X2NEG);
140 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_DZ);
141 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_DW);
142 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_ABS);
143 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_ABSNEG);
144 WINE_D3DCOMPILER_TO_STR(BWRITERSPSM_NOT);
145 default:
146 FIXME("Unrecognized source modifier %#x.\n", mod);
147 return "unrecognized_src_mod";
148 }
149 }
150
151 #undef WINE_D3DCOMPILER_TO_STR
152
153 const char *debug_print_dstmod(DWORD mod)
154 {
155 switch (mod)
156 {
157 case 0:
158 return "";
159 case BWRITERSPDM_SATURATE:
160 return "_sat";
161 case BWRITERSPDM_PARTIALPRECISION:
162 return "_pp";
163 case BWRITERSPDM_MSAMPCENTROID:
164 return "_centroid";
165 case BWRITERSPDM_SATURATE | BWRITERSPDM_PARTIALPRECISION:
166 return "_sat_pp";
167 case BWRITERSPDM_SATURATE | BWRITERSPDM_MSAMPCENTROID:
168 return "_sat_centroid";
169 case BWRITERSPDM_PARTIALPRECISION | BWRITERSPDM_MSAMPCENTROID:
170 return "_pp_centroid";
171 case BWRITERSPDM_SATURATE | BWRITERSPDM_PARTIALPRECISION | BWRITERSPDM_MSAMPCENTROID:
172 return "_sat_pp_centroid";
173 default:
174 return "Unexpected modifier\n";
175 }
176 }
177
178 const char *debug_print_shift(DWORD shift)
179 {
180 static const char * const shiftstrings[] =
181 {
182 "",
183 "_x2",
184 "_x4",
185 "_x8",
186 "_x16",
187 "_x32",
188 "",
189 "",
190 "",
191 "",
192 "",
193 "",
194 "_d16",
195 "_d8",
196 "_d4",
197 "_d2",
198 };
199 return shiftstrings[shift];
200 }
201
202 static const char *get_regname(const struct shader_reg *reg)
203 {
204 switch (reg->type)
205 {
206 case BWRITERSPR_TEMP:
207 return wine_dbg_sprintf("r%u", reg->regnum);
208 case BWRITERSPR_INPUT:
209 return wine_dbg_sprintf("v%u", reg->regnum);
210 case BWRITERSPR_CONST:
211 return wine_dbg_sprintf("c%u", reg->regnum);
212 case BWRITERSPR_ADDR:
213 return wine_dbg_sprintf("a%u", reg->regnum);
214 case BWRITERSPR_TEXTURE:
215 return wine_dbg_sprintf("t%u", reg->regnum);
216 case BWRITERSPR_RASTOUT:
217 switch (reg->regnum)
218 {
219 case BWRITERSRO_POSITION: return "oPos";
220 case BWRITERSRO_FOG: return "oFog";
221 case BWRITERSRO_POINT_SIZE: return "oPts";
222 default: return "Unexpected RASTOUT";
223 }
224 case BWRITERSPR_ATTROUT:
225 return wine_dbg_sprintf("oD%u", reg->regnum);
226 case BWRITERSPR_TEXCRDOUT:
227 return wine_dbg_sprintf("oT%u", reg->regnum);
228 case BWRITERSPR_OUTPUT:
229 return wine_dbg_sprintf("o%u", reg->regnum);
230 case BWRITERSPR_CONSTINT:
231 return wine_dbg_sprintf("i%u", reg->regnum);
232 case BWRITERSPR_COLOROUT:
233 return wine_dbg_sprintf("oC%u", reg->regnum);
234 case BWRITERSPR_DEPTHOUT:
235 return "oDepth";
236 case BWRITERSPR_SAMPLER:
237 return wine_dbg_sprintf("s%u", reg->regnum);
238 case BWRITERSPR_CONSTBOOL:
239 return wine_dbg_sprintf("b%u", reg->regnum);
240 case BWRITERSPR_LOOP:
241 return "aL";
242 case BWRITERSPR_MISCTYPE:
243 switch (reg->regnum)
244 {
245 case 0: return "vPos";
246 case 1: return "vFace";
247 default: return "unexpected misctype";
248 }
249 case BWRITERSPR_LABEL:
250 return wine_dbg_sprintf("l%u", reg->regnum);
251 case BWRITERSPR_PREDICATE:
252 return wine_dbg_sprintf("p%u", reg->regnum);
253 default:
254 return wine_dbg_sprintf("unknown regname %#x", reg->type);
255 }
256 }
257
258 static const char *debug_print_writemask(DWORD mask)
259 {
260 char ret[6];
261 unsigned char pos = 1;
262
263 if(mask == BWRITERSP_WRITEMASK_ALL) return "";
264 ret[0] = '.';
265 if(mask & BWRITERSP_WRITEMASK_0) ret[pos++] = 'x';
266 if(mask & BWRITERSP_WRITEMASK_1) ret[pos++] = 'y';
267 if(mask & BWRITERSP_WRITEMASK_2) ret[pos++] = 'z';
268 if(mask & BWRITERSP_WRITEMASK_3) ret[pos++] = 'w';
269 ret[pos] = 0;
270
271 return wine_dbg_sprintf("%s", ret);
272 }
273
274 static const char *debug_print_swizzle(DWORD arg)
275 {
276 char ret[6];
277 unsigned int i;
278 DWORD swizzle[4];
279
280 switch (arg)
281 {
282 case BWRITERVS_NOSWIZZLE:
283 return "";
284 case BWRITERVS_SWIZZLE_X:
285 return ".x";
286 case BWRITERVS_SWIZZLE_Y:
287 return ".y";
288 case BWRITERVS_SWIZZLE_Z:
289 return ".z";
290 case BWRITERVS_SWIZZLE_W:
291 return ".w";
292 }
293
294 swizzle[0] = (arg >> (BWRITERVS_SWIZZLE_SHIFT + 0)) & 0x03;
295 swizzle[1] = (arg >> (BWRITERVS_SWIZZLE_SHIFT + 2)) & 0x03;
296 swizzle[2] = (arg >> (BWRITERVS_SWIZZLE_SHIFT + 4)) & 0x03;
297 swizzle[3] = (arg >> (BWRITERVS_SWIZZLE_SHIFT + 6)) & 0x03;
298
299 ret[0] = '.';
300 for (i = 0; i < 4; ++i)
301 {
302 switch (swizzle[i])
303 {
304 case 0: ret[1 + i] = 'x'; break;
305 case 1: ret[1 + i] = 'y'; break;
306 case 2: ret[1 + i] = 'z'; break;
307 case 3: ret[1 + i] = 'w'; break;
308 }
309 }
310 ret[5] = '\0';
311
312 return wine_dbg_sprintf("%s", ret);
313 }
314
315 static const char *debug_print_relarg(const struct shader_reg *reg)
316 {
317 const char *short_swizzle;
318 if (!reg->rel_reg) return "";
319
320 short_swizzle = debug_print_swizzle(reg->rel_reg->u.swizzle);
321
322 if (reg->rel_reg->type == BWRITERSPR_ADDR)
323 return wine_dbg_sprintf("[a%u%s]", reg->rel_reg->regnum, short_swizzle);
324 else if(reg->rel_reg->type == BWRITERSPR_LOOP && reg->rel_reg->regnum == 0)
325 return wine_dbg_sprintf("[aL%s]", short_swizzle);
326 else
327 return "Unexpected relative addressing argument";
328 }
329
330 const char *debug_print_dstreg(const struct shader_reg *reg)
331 {
332 return wine_dbg_sprintf("%s%s%s", get_regname(reg),
333 debug_print_relarg(reg),
334 debug_print_writemask(reg->u.writemask));
335 }
336
337 const char *debug_print_srcreg(const struct shader_reg *reg)
338 {
339 switch (reg->srcmod)
340 {
341 case BWRITERSPSM_NONE:
342 return wine_dbg_sprintf("%s%s%s", get_regname(reg),
343 debug_print_relarg(reg),
344 debug_print_swizzle(reg->u.swizzle));
345 case BWRITERSPSM_NEG:
346 return wine_dbg_sprintf("-%s%s%s", get_regname(reg),
347 debug_print_relarg(reg),
348 debug_print_swizzle(reg->u.swizzle));
349 case BWRITERSPSM_BIAS:
350 return wine_dbg_sprintf("%s%s_bias%s", get_regname(reg),
351 debug_print_relarg(reg),
352 debug_print_swizzle(reg->u.swizzle));
353 case BWRITERSPSM_BIASNEG:
354 return wine_dbg_sprintf("-%s%s_bias%s", get_regname(reg),
355 debug_print_relarg(reg),
356 debug_print_swizzle(reg->u.swizzle));
357 case BWRITERSPSM_SIGN:
358 return wine_dbg_sprintf("%s%s_bx2%s", get_regname(reg),
359 debug_print_relarg(reg),
360 debug_print_swizzle(reg->u.swizzle));
361 case BWRITERSPSM_SIGNNEG:
362 return wine_dbg_sprintf("-%s%s_bx2%s", get_regname(reg),
363 debug_print_relarg(reg),
364 debug_print_swizzle(reg->u.swizzle));
365 case BWRITERSPSM_COMP:
366 return wine_dbg_sprintf("1 - %s%s%s", get_regname(reg),
367 debug_print_relarg(reg),
368 debug_print_swizzle(reg->u.swizzle));
369 case BWRITERSPSM_X2:
370 return wine_dbg_sprintf("%s%s_x2%s", get_regname(reg),
371 debug_print_relarg(reg),
372 debug_print_swizzle(reg->u.swizzle));
373 case BWRITERSPSM_X2NEG:
374 return wine_dbg_sprintf("-%s%s_x2%s", get_regname(reg),
375 debug_print_relarg(reg),
376 debug_print_swizzle(reg->u.swizzle));
377 case BWRITERSPSM_DZ:
378 return wine_dbg_sprintf("%s%s_dz%s", get_regname(reg),
379 debug_print_relarg(reg),
380 debug_print_swizzle(reg->u.swizzle));
381 case BWRITERSPSM_DW:
382 return wine_dbg_sprintf("%s%s_dw%s", get_regname(reg),
383 debug_print_relarg(reg),
384 debug_print_swizzle(reg->u.swizzle));
385 case BWRITERSPSM_ABS:
386 return wine_dbg_sprintf("%s%s_abs%s", get_regname(reg),
387 debug_print_relarg(reg),
388 debug_print_swizzle(reg->u.swizzle));
389 case BWRITERSPSM_ABSNEG:
390 return wine_dbg_sprintf("-%s%s_abs%s", get_regname(reg),
391 debug_print_relarg(reg),
392 debug_print_swizzle(reg->u.swizzle));
393 case BWRITERSPSM_NOT:
394 return wine_dbg_sprintf("!%s%s%s", get_regname(reg),
395 debug_print_relarg(reg),
396 debug_print_swizzle(reg->u.swizzle));
397 }
398 return "Unknown modifier";
399 }
400
401 const char *debug_print_comp(DWORD comp)
402 {
403 switch (comp)
404 {
405 case BWRITER_COMPARISON_NONE: return "";
406 case BWRITER_COMPARISON_GT: return "_gt";
407 case BWRITER_COMPARISON_EQ: return "_eq";
408 case BWRITER_COMPARISON_GE: return "_ge";
409 case BWRITER_COMPARISON_LT: return "_lt";
410 case BWRITER_COMPARISON_NE: return "_ne";
411 case BWRITER_COMPARISON_LE: return "_le";
412 default: return "_unknown";
413 }
414 }
415
416 const char *debug_print_opcode(DWORD opcode)
417 {
418 switch (opcode)
419 {
420 case BWRITERSIO_NOP: return "nop";
421 case BWRITERSIO_MOV: return "mov";
422 case BWRITERSIO_ADD: return "add";
423 case BWRITERSIO_SUB: return "sub";
424 case BWRITERSIO_MAD: return "mad";
425 case BWRITERSIO_MUL: return "mul";
426 case BWRITERSIO_RCP: return "rcp";
427 case BWRITERSIO_RSQ: return "rsq";
428 case BWRITERSIO_DP3: return "dp3";
429 case BWRITERSIO_DP4: return "dp4";
430 case BWRITERSIO_MIN: return "min";
431 case BWRITERSIO_MAX: return "max";
432 case BWRITERSIO_SLT: return "slt";
433 case BWRITERSIO_SGE: return "sge";
434 case BWRITERSIO_EXP: return "exp";
435 case BWRITERSIO_LOG: return "log";
436 case BWRITERSIO_LIT: return "lit";
437 case BWRITERSIO_DST: return "dst";
438 case BWRITERSIO_LRP: return "lrp";
439 case BWRITERSIO_FRC: return "frc";
440 case BWRITERSIO_M4x4: return "m4x4";
441 case BWRITERSIO_M4x3: return "m4x3";
442 case BWRITERSIO_M3x4: return "m3x4";
443 case BWRITERSIO_M3x3: return "m3x3";
444 case BWRITERSIO_M3x2: return "m3x2";
445 case BWRITERSIO_CALL: return "call";
446 case BWRITERSIO_CALLNZ: return "callnz";
447 case BWRITERSIO_LOOP: return "loop";
448 case BWRITERSIO_RET: return "ret";
449 case BWRITERSIO_ENDLOOP: return "endloop";
450 case BWRITERSIO_LABEL: return "label";
451 case BWRITERSIO_DCL: return "dcl";
452 case BWRITERSIO_POW: return "pow";
453 case BWRITERSIO_CRS: return "crs";
454 case BWRITERSIO_SGN: return "sgn";
455 case BWRITERSIO_ABS: return "abs";
456 case BWRITERSIO_NRM: return "nrm";
457 case BWRITERSIO_SINCOS: return "sincos";
458 case BWRITERSIO_REP: return "rep";
459 case BWRITERSIO_ENDREP: return "endrep";
460 case BWRITERSIO_IF: return "if";
461 case BWRITERSIO_IFC: return "ifc";
462 case BWRITERSIO_ELSE: return "else";
463 case BWRITERSIO_ENDIF: return "endif";
464 case BWRITERSIO_BREAK: return "break";
465 case BWRITERSIO_BREAKC: return "breakc";
466 case BWRITERSIO_MOVA: return "mova";
467 case BWRITERSIO_DEFB: return "defb";
468 case BWRITERSIO_DEFI: return "defi";
469 case BWRITERSIO_TEXCOORD: return "texcoord";
470 case BWRITERSIO_TEXKILL: return "texkill";
471 case BWRITERSIO_TEX: return "tex";
472 case BWRITERSIO_TEXBEM: return "texbem";
473 case BWRITERSIO_TEXBEML: return "texbeml";
474 case BWRITERSIO_TEXREG2AR: return "texreg2ar";
475 case BWRITERSIO_TEXREG2GB: return "texreg2gb";
476 case BWRITERSIO_TEXM3x2PAD: return "texm3x2pad";
477 case BWRITERSIO_TEXM3x2TEX: return "texm3x2tex";
478 case BWRITERSIO_TEXM3x3PAD: return "texm3x3pad";
479 case BWRITERSIO_TEXM3x3TEX: return "texm3x3tex";
480 case BWRITERSIO_TEXM3x3SPEC: return "texm3x3vspec";
481 case BWRITERSIO_TEXM3x3VSPEC: return "texm3x3vspec";
482 case BWRITERSIO_EXPP: return "expp";
483 case BWRITERSIO_LOGP: return "logp";
484 case BWRITERSIO_CND: return "cnd";
485 case BWRITERSIO_DEF: return "def";
486 case BWRITERSIO_TEXREG2RGB: return "texreg2rgb";
487 case BWRITERSIO_TEXDP3TEX: return "texdp3tex";
488 case BWRITERSIO_TEXM3x2DEPTH: return "texm3x2depth";
489 case BWRITERSIO_TEXDP3: return "texdp3";
490 case BWRITERSIO_TEXM3x3: return "texm3x3";
491 case BWRITERSIO_TEXDEPTH: return "texdepth";
492 case BWRITERSIO_CMP: return "cmp";
493 case BWRITERSIO_BEM: return "bem";
494 case BWRITERSIO_DP2ADD: return "dp2add";
495 case BWRITERSIO_DSX: return "dsx";
496 case BWRITERSIO_DSY: return "dsy";
497 case BWRITERSIO_TEXLDD: return "texldd";
498 case BWRITERSIO_SETP: return "setp";
499 case BWRITERSIO_TEXLDL: return "texldl";
500 case BWRITERSIO_BREAKP: return "breakp";
501 case BWRITERSIO_PHASE: return "phase";
502
503 case BWRITERSIO_TEXLDP: return "texldp";
504 case BWRITERSIO_TEXLDB: return "texldb";
505
506 default: return "unknown";
507 }
508 }
509
510 void skip_dword_unknown(const char **ptr, unsigned int count)
511 {
512 unsigned int i;
513 DWORD d;
514
515 FIXME("Skipping %u unknown DWORDs:\n", count);
516 for (i = 0; i < count; ++i)
517 {
518 read_dword(ptr, &d);
519 FIXME("\t0x%08x\n", d);
520 }
521 }
522
523 static void write_dword_unknown(char **ptr, DWORD d)
524 {
525 FIXME("Writing unknown DWORD 0x%08x\n", d);
526 write_dword(ptr, d);
527 }
528
529 HRESULT dxbc_add_section(struct dxbc *dxbc, DWORD tag, const char *data, DWORD data_size)
530 {
531 TRACE("dxbc %p, tag %s, size %#x.\n", dxbc, debugstr_an((const char *)&tag, 4), data_size);
532
533 if (dxbc->count >= dxbc->size)
534 {
535 struct dxbc_section *new_sections;
536 DWORD new_size = dxbc->size << 1;
537
538 new_sections = HeapReAlloc(GetProcessHeap(), 0, dxbc->sections, new_size * sizeof(*dxbc->sections));
539 if (!new_sections)
540 {
541 ERR("Failed to allocate dxbc section memory\n");
542 return E_OUTOFMEMORY;
543 }
544
545 dxbc->sections = new_sections;
546 dxbc->size = new_size;
547 }
548
549 dxbc->sections[dxbc->count].tag = tag;
550 dxbc->sections[dxbc->count].data_size = data_size;
551 dxbc->sections[dxbc->count].data = data;
552 ++dxbc->count;
553
554 return S_OK;
555 }
556
557 HRESULT dxbc_init(struct dxbc *dxbc, unsigned int size)
558 {
559 TRACE("dxbc %p, size %u.\n", dxbc, size);
560
561 /* use a good starting value for the size if none specified */
562 if (!size) size = 2;
563
564 dxbc->sections = HeapAlloc(GetProcessHeap(), 0, size * sizeof(*dxbc->sections));
565 if (!dxbc->sections)
566 {
567 ERR("Failed to allocate dxbc section memory\n");
568 return E_OUTOFMEMORY;
569 }
570
571 dxbc->size = size;
572 dxbc->count = 0;
573
574 return S_OK;
575 }
576
577 HRESULT dxbc_parse(const char *data, SIZE_T data_size, struct dxbc *dxbc)
578 {
579 const char *ptr = data;
580 HRESULT hr;
581 unsigned int i;
582 DWORD tag, total_size, chunk_count;
583
584 if (!data)
585 {
586 WARN("No data supplied.\n");
587 return E_FAIL;
588 }
589
590 read_dword(&ptr, &tag);
591 TRACE("tag: %s.\n", debugstr_an((const char *)&tag, 4));
592
593 if (tag != TAG_DXBC)
594 {
595 WARN("Wrong tag.\n");
596 return E_FAIL;
597 }
598
599 /* checksum? */
600 skip_dword_unknown(&ptr, 4);
601
602 skip_dword_unknown(&ptr, 1);
603
604 read_dword(&ptr, &total_size);
605 TRACE("total size: %#x\n", total_size);
606
607 if (data_size != total_size)
608 {
609 WARN("Wrong size supplied.\n");
610 return D3DERR_INVALIDCALL;
611 }
612
613 read_dword(&ptr, &chunk_count);
614 TRACE("chunk count: %#x\n", chunk_count);
615
616 hr = dxbc_init(dxbc, chunk_count);
617 if (FAILED(hr))
618 {
619 WARN("Failed to init dxbc\n");
620 return hr;
621 }
622
623 for (i = 0; i < chunk_count; ++i)
624 {
625 DWORD chunk_tag, chunk_size;
626 const char *chunk_ptr;
627 DWORD chunk_offset;
628
629 read_dword(&ptr, &chunk_offset);
630 TRACE("chunk %u at offset %#x\n", i, chunk_offset);
631
632 chunk_ptr = data + chunk_offset;
633
634 read_dword(&chunk_ptr, &chunk_tag);
635 read_dword(&chunk_ptr, &chunk_size);
636
637 hr = dxbc_add_section(dxbc, chunk_tag, chunk_ptr, chunk_size);
638 if (FAILED(hr))
639 {
640 WARN("Failed to add section to dxbc\n");
641 return hr;
642 }
643 }
644
645 return hr;
646 }
647
648 void dxbc_destroy(struct dxbc *dxbc)
649 {
650 TRACE("dxbc %p.\n", dxbc);
651
652 HeapFree(GetProcessHeap(), 0, dxbc->sections);
653 }
654
655 HRESULT dxbc_write_blob(struct dxbc *dxbc, ID3DBlob **blob)
656 {
657 DWORD size = 32, offset = size + 4 * dxbc->count;
658 ID3DBlob *object;
659 HRESULT hr;
660 char *ptr;
661 unsigned int i;
662
663 TRACE("dxbc %p, blob %p.\n", dxbc, blob);
664
665 for (i = 0; i < dxbc->count; ++i)
666 {
667 size += 12 + dxbc->sections[i].data_size;
668 }
669
670 hr = D3DCreateBlob(size, &object);
671 if (FAILED(hr))
672 {
673 WARN("Failed to create blob\n");
674 return hr;
675 }
676
677 ptr = ID3D10Blob_GetBufferPointer(object);
678
679 write_dword(&ptr, TAG_DXBC);
680
681 /* signature(?) */
682 write_dword_unknown(&ptr, 0);
683 write_dword_unknown(&ptr, 0);
684 write_dword_unknown(&ptr, 0);
685 write_dword_unknown(&ptr, 0);
686
687 /* seems to be always 1 */
688 write_dword_unknown(&ptr, 1);
689
690 /* DXBC size */
691 write_dword(&ptr, size);
692
693 /* chunk count */
694 write_dword(&ptr, dxbc->count);
695
696 /* write the chunk offsets */
697 for (i = 0; i < dxbc->count; ++i)
698 {
699 write_dword(&ptr, offset);
700 offset += 8 + dxbc->sections[i].data_size;
701 }
702
703 /* write the chunks */
704 for (i = 0; i < dxbc->count; ++i)
705 {
706 write_dword(&ptr, dxbc->sections[i].tag);
707 write_dword(&ptr, dxbc->sections[i].data_size);
708 memcpy(ptr, dxbc->sections[i].data, dxbc->sections[i].data_size);
709 ptr += dxbc->sections[i].data_size;
710 }
711
712 TRACE("Created ID3DBlob %p\n", object);
713
714 *blob = object;
715
716 return S_OK;
717 }
718
719 void compilation_message(struct compilation_messages *msg, const char *fmt, __ms_va_list args)
720 {
721 char* buffer;
722 int rc, size;
723
724 if (msg->capacity == 0)
725 {
726 msg->string = d3dcompiler_alloc(MESSAGEBUFFER_INITIAL_SIZE);
727 if (msg->string == NULL)
728 {
729 ERR("Error allocating memory for parser messages\n");
730 return;
731 }
732 msg->capacity = MESSAGEBUFFER_INITIAL_SIZE;
733 }
734
735 while (1)
736 {
737 rc = vsnprintf(msg->string + msg->size,
738 msg->capacity - msg->size, fmt, args);
739
740 if (rc < 0 || rc >= msg->capacity - msg->size)
741 {
742 size = msg->capacity * 2;
743 buffer = d3dcompiler_realloc(msg->string, size);
744 if (buffer == NULL)
745 {
746 ERR("Error reallocating memory for parser messages\n");
747 return;
748 }
749 msg->string = buffer;
750 msg->capacity = size;
751 }
752 else
753 {
754 TRACE("%s", msg->string + msg->size);
755 msg->size += rc;
756 return;
757 }
758 }
759 }
760
761 #if D3D_COMPILER_VERSION
762 BOOL add_declaration(struct hlsl_scope *scope, struct hlsl_ir_var *decl, BOOL local_var)
763 {
764 struct hlsl_ir_var *var;
765
766 LIST_FOR_EACH_ENTRY(var, &scope->vars, struct hlsl_ir_var, scope_entry)
767 {
768 if (!strcmp(decl->name, var->name))
769 return FALSE;
770 }
771 if (local_var && scope->upper->upper == hlsl_ctx.globals)
772 {
773 /* Check whether the variable redefines a function parameter. */
774 LIST_FOR_EACH_ENTRY(var, &scope->upper->vars, struct hlsl_ir_var, scope_entry)
775 {
776 if (!strcmp(decl->name, var->name))
777 return FALSE;
778 }
779 }
780
781 list_add_tail(&scope->vars, &decl->scope_entry);
782 return TRUE;
783 }
784
785 struct hlsl_ir_var *get_variable(struct hlsl_scope *scope, const char *name)
786 {
787 struct hlsl_ir_var *var;
788
789 LIST_FOR_EACH_ENTRY(var, &scope->vars, struct hlsl_ir_var, scope_entry)
790 {
791 if (!strcmp(name, var->name))
792 return var;
793 }
794 if (!scope->upper)
795 return NULL;
796 return get_variable(scope->upper, name);
797 }
798
799 void free_declaration(struct hlsl_ir_var *decl)
800 {
801 d3dcompiler_free((void *)decl->name);
802 d3dcompiler_free((void *)decl->semantic);
803 d3dcompiler_free((void *)decl->reg_reservation);
804 d3dcompiler_free(decl);
805 }
806
807 struct hlsl_type *new_hlsl_type(const char *name, enum hlsl_type_class type_class,
808 enum hlsl_base_type base_type, unsigned dimx, unsigned dimy)
809 {
810 struct hlsl_type *type;
811
812 type = d3dcompiler_alloc(sizeof(*type));
813 if (!type)
814 {
815 ERR("Out of memory\n");
816 return NULL;
817 }
818 type->name = name;
819 type->type = type_class;
820 type->base_type = base_type;
821 type->dimx = dimx;
822 type->dimy = dimy;
823
824 list_add_tail(&hlsl_ctx.types, &type->entry);
825
826 return type;
827 }
828
829 struct hlsl_type *new_array_type(struct hlsl_type *basic_type, unsigned int array_size)
830 {
831 struct hlsl_type *type = new_hlsl_type(NULL, HLSL_CLASS_ARRAY, HLSL_TYPE_FLOAT, 1, 1);
832
833 if (!type)
834 return NULL;
835
836 type->modifiers = basic_type->modifiers;
837 type->e.array.elements_count = array_size;
838 type->e.array.type = basic_type;
839 return type;
840 }
841
842 struct hlsl_type *get_type(struct hlsl_scope *scope, const char *name, BOOL recursive)
843 {
844 struct wine_rb_entry *entry = wine_rb_get(&scope->types, name);
845 if (entry)
846 return WINE_RB_ENTRY_VALUE(entry, struct hlsl_type, scope_entry);
847
848 if (recursive && scope->upper)
849 return get_type(scope->upper, name, recursive);
850 return NULL;
851 }
852
853 BOOL find_function(const char *name)
854 {
855 return wine_rb_get(&hlsl_ctx.functions, name) != NULL;
856 }
857
858 unsigned int components_count_type(struct hlsl_type *type)
859 {
860 unsigned int count = 0;
861 struct hlsl_struct_field *field;
862
863 if (type->type <= HLSL_CLASS_LAST_NUMERIC)
864 {
865 return type->dimx * type->dimy;
866 }
867 if (type->type == HLSL_CLASS_ARRAY)
868 {
869 return components_count_type(type->e.array.type) * type->e.array.elements_count;
870 }
871 if (type->type != HLSL_CLASS_STRUCT)
872 {
873 ERR("Unexpected data type %s.\n", debug_hlsl_type(type));
874 return 0;
875 }
876
877 LIST_FOR_EACH_ENTRY(field, type->e.elements, struct hlsl_struct_field, entry)
878 {
879 count += components_count_type(field->type);
880 }
881 return count;
882 }
883
884 BOOL compare_hlsl_types(const struct hlsl_type *t1, const struct hlsl_type *t2)
885 {
886 if (t1 == t2)
887 return TRUE;
888
889 if (t1->type != t2->type)
890 return FALSE;
891 if (t1->base_type != t2->base_type)
892 return FALSE;
893 if (t1->base_type == HLSL_TYPE_SAMPLER && t1->sampler_dim != t2->sampler_dim)
894 return FALSE;
895 if ((t1->modifiers & HLSL_MODIFIERS_COMPARISON_MASK)
896 != (t2->modifiers & HLSL_MODIFIERS_COMPARISON_MASK))
897 return FALSE;
898 if (t1->dimx != t2->dimx)
899 return FALSE;
900 if (t1->dimy != t2->dimy)
901 return FALSE;
902 if (t1->type == HLSL_CLASS_STRUCT)
903 {
904 struct list *t1cur, *t2cur;
905 struct hlsl_struct_field *t1field, *t2field;
906
907 t1cur = list_head(t1->e.elements);
908 t2cur = list_head(t2->e.elements);
909 while (t1cur && t2cur)
910 {
911 t1field = LIST_ENTRY(t1cur, struct hlsl_struct_field, entry);
912 t2field = LIST_ENTRY(t2cur, struct hlsl_struct_field, entry);
913 if (!compare_hlsl_types(t1field->type, t2field->type))
914 return FALSE;
915 if (strcmp(t1field->name, t2field->name))
916 return FALSE;
917 t1cur = list_next(t1->e.elements, t1cur);
918 t2cur = list_next(t2->e.elements, t2cur);
919 }
920 if (t1cur != t2cur)
921 return FALSE;
922 }
923 if (t1->type == HLSL_CLASS_ARRAY)
924 return t1->e.array.elements_count == t2->e.array.elements_count
925 && compare_hlsl_types(t1->e.array.type, t2->e.array.type);
926
927 return TRUE;
928 }
929
930 struct hlsl_type *clone_hlsl_type(struct hlsl_type *old)
931 {
932 struct hlsl_type *type;
933 struct hlsl_struct_field *old_field, *field;
934
935 type = d3dcompiler_alloc(sizeof(*type));
936 if (!type)
937 {
938 ERR("Out of memory\n");
939 return NULL;
940 }
941 if (old->name)
942 {
943 type->name = d3dcompiler_strdup(old->name);
944 if (!type->name)
945 {
946 d3dcompiler_free(type);
947 return NULL;
948 }
949 }
950 type->type = old->type;
951 type->base_type = old->base_type;
952 type->dimx = old->dimx;
953 type->dimy = old->dimy;
954 type->modifiers = old->modifiers;
955 type->sampler_dim = old->sampler_dim;
956 switch (old->type)
957 {
958 case HLSL_CLASS_ARRAY:
959 type->e.array.type = old->e.array.type;
960 type->e.array.elements_count = old->e.array.elements_count;
961 break;
962 case HLSL_CLASS_STRUCT:
963 type->e.elements = d3dcompiler_alloc(sizeof(*type->e.elements));
964 if (!type->e.elements)
965 {
966 d3dcompiler_free((void *)type->name);
967 d3dcompiler_free(type);
968 return NULL;
969 }
970 list_init(type->e.elements);
971 LIST_FOR_EACH_ENTRY(old_field, old->e.elements, struct hlsl_struct_field, entry)
972 {
973 field = d3dcompiler_alloc(sizeof(*field));
974 if (!field)
975 {
976 LIST_FOR_EACH_ENTRY_SAFE(field, old_field, type->e.elements, struct hlsl_struct_field, entry)
977 {
978 d3dcompiler_free((void *)field->semantic);
979 d3dcompiler_free((void *)field->name);
980 d3dcompiler_free(field);
981 }
982 d3dcompiler_free(type->e.elements);
983 d3dcompiler_free((void *)type->name);
984 d3dcompiler_free(type);
985 return NULL;
986 }
987 field->type = clone_hlsl_type(old_field->type);
988 field->name = d3dcompiler_strdup(old_field->name);
989 if (old_field->semantic)
990 field->semantic = d3dcompiler_strdup(old_field->semantic);
991 field->modifiers = old_field->modifiers;
992 list_add_tail(type->e.elements, &field->entry);
993 }
994 break;
995 default:
996 break;
997 }
998
999 list_add_tail(&hlsl_ctx.types, &type->entry);
1000 return type;
1001 }
1002
1003 static BOOL convertible_data_type(struct hlsl_type *type)
1004 {
1005 return type->type != HLSL_CLASS_OBJECT;
1006 }
1007
1008 BOOL compatible_data_types(struct hlsl_type *t1, struct hlsl_type *t2)
1009 {
1010 if (!convertible_data_type(t1) || !convertible_data_type(t2))
1011 return FALSE;
1012
1013 if (t1->type <= HLSL_CLASS_LAST_NUMERIC)
1014 {
1015 /* Scalar vars can be cast to pretty much everything */
1016 if (t1->dimx == 1 && t1->dimy == 1)
1017 return TRUE;
1018
1019 if (t1->type == HLSL_CLASS_VECTOR && t2->type == HLSL_CLASS_VECTOR)
1020 return t1->dimx >= t2->dimx;
1021 }
1022
1023 /* The other way around is true too i.e. whatever to scalar */
1024 if (t2->type <= HLSL_CLASS_LAST_NUMERIC && t2->dimx == 1 && t2->dimy == 1)
1025 return TRUE;
1026
1027 if (t1->type == HLSL_CLASS_ARRAY)
1028 {
1029 if (compare_hlsl_types(t1->e.array.type, t2))
1030 /* e.g. float4[3] to float4 is allowed */
1031 return TRUE;
1032
1033 if (t2->type == HLSL_CLASS_ARRAY || t2->type == HLSL_CLASS_STRUCT)
1034 return components_count_type(t1) >= components_count_type(t2);
1035 else
1036 return components_count_type(t1) == components_count_type(t2);
1037 }
1038
1039 if (t1->type == HLSL_CLASS_STRUCT)
1040 return components_count_type(t1) >= components_count_type(t2);
1041
1042 if (t2->type == HLSL_CLASS_ARRAY || t2->type == HLSL_CLASS_STRUCT)
1043 return components_count_type(t1) == components_count_type(t2);
1044
1045 if (t1->type == HLSL_CLASS_MATRIX || t2->type == HLSL_CLASS_MATRIX)
1046 {
1047 if (t1->type == HLSL_CLASS_MATRIX && t2->type == HLSL_CLASS_MATRIX && t1->dimx >= t2->dimx && t1->dimy >= t2->dimy)
1048 return TRUE;
1049
1050 /* Matrix-vector conversion is apparently allowed if they have the same components count */
1051 if ((t1->type == HLSL_CLASS_VECTOR || t2->type == HLSL_CLASS_VECTOR)
1052 && components_count_type(t1) == components_count_type(t2))
1053 return TRUE;
1054 return FALSE;
1055 }
1056
1057 if (components_count_type(t1) >= components_count_type(t2))
1058 return TRUE;
1059 return FALSE;
1060 }
1061
1062 static BOOL implicit_compatible_data_types(struct hlsl_type *t1, struct hlsl_type *t2)
1063 {
1064 if (!convertible_data_type(t1) || !convertible_data_type(t2))
1065 return FALSE;
1066
1067 if (t1->type <= HLSL_CLASS_LAST_NUMERIC)
1068 {
1069 /* Scalar vars can be converted to any other numeric data type */
1070 if (t1->dimx == 1 && t1->dimy == 1 && t2->type <= HLSL_CLASS_LAST_NUMERIC)
1071 return TRUE;
1072 /* The other way around is true too */
1073 if (t2->dimx == 1 && t2->dimy == 1 && t2->type <= HLSL_CLASS_LAST_NUMERIC)
1074 return TRUE;
1075 }
1076
1077 if (t1->type == HLSL_CLASS_ARRAY && t2->type == HLSL_CLASS_ARRAY)
1078 {
1079 return components_count_type(t1) == components_count_type(t2);
1080 }
1081
1082 if ((t1->type == HLSL_CLASS_ARRAY && t2->type <= HLSL_CLASS_LAST_NUMERIC)
1083 || (t1->type <= HLSL_CLASS_LAST_NUMERIC && t2->type == HLSL_CLASS_ARRAY))
1084 {
1085 /* e.g. float4[3] to float4 is allowed */
1086 if (t1->type == HLSL_CLASS_ARRAY && compare_hlsl_types(t1->e.array.type, t2))
1087 return TRUE;
1088 if (components_count_type(t1) == components_count_type(t2))
1089 return TRUE;
1090 return FALSE;
1091 }
1092
1093 if (t1->type <= HLSL_CLASS_VECTOR && t2->type <= HLSL_CLASS_VECTOR)
1094 {
1095 if (t1->dimx >= t2->dimx)
1096 return TRUE;
1097 return FALSE;
1098 }
1099
1100 if (t1->type == HLSL_CLASS_MATRIX || t2->type == HLSL_CLASS_MATRIX)
1101 {
1102 if (t1->type == HLSL_CLASS_MATRIX && t2->type == HLSL_CLASS_MATRIX
1103 && t1->dimx >= t2->dimx && t1->dimy >= t2->dimy)
1104 return TRUE;
1105
1106 /* Matrix-vector conversion is apparently allowed if they have the same components count */
1107 if ((t1->type == HLSL_CLASS_VECTOR || t2->type == HLSL_CLASS_VECTOR)
1108 && components_count_type(t1) == components_count_type(t2))
1109 return TRUE;
1110 return FALSE;
1111 }
1112
1113 if (t1->type == HLSL_CLASS_STRUCT && t2->type == HLSL_CLASS_STRUCT)
1114 return compare_hlsl_types(t1, t2);
1115
1116 return FALSE;
1117 }
1118
1119 static BOOL expr_compatible_data_types(struct hlsl_type *t1, struct hlsl_type *t2)
1120 {
1121 if (t1->base_type > HLSL_TYPE_LAST_SCALAR || t2->base_type > HLSL_TYPE_LAST_SCALAR)
1122 return FALSE;
1123
1124 /* Scalar vars can be converted to pretty much everything */
1125 if ((t1->dimx == 1 && t1->dimy == 1) || (t2->dimx == 1 && t2->dimy == 1))
1126 return TRUE;
1127
1128 if (t1->type == HLSL_CLASS_VECTOR && t2->type == HLSL_CLASS_VECTOR)
1129 return TRUE;
1130
1131 if (t1->type == HLSL_CLASS_MATRIX || t2->type == HLSL_CLASS_MATRIX)
1132 {
1133 /* Matrix-vector conversion is apparently allowed if either they have the same components
1134 count or the matrix is nx1 or 1xn */
1135 if (t1->type == HLSL_CLASS_VECTOR || t2->type == HLSL_CLASS_VECTOR)
1136 {
1137 if (components_count_type(t1) == components_count_type(t2))
1138 return TRUE;
1139
1140 return (t1->type == HLSL_CLASS_MATRIX && (t1->dimx == 1 || t1->dimy == 1))
1141 || (t2->type == HLSL_CLASS_MATRIX && (t2->dimx == 1 || t2->dimy == 1));
1142 }
1143
1144 /* Both matrices */
1145 if ((t1->dimx >= t2->dimx && t1->dimy >= t2->dimy)
1146 || (t1->dimx <= t2->dimx && t1->dimy <= t2->dimy))
1147 return TRUE;
1148 }
1149
1150 return FALSE;
1151 }
1152
1153 static enum hlsl_base_type expr_common_base_type(enum hlsl_base_type t1, enum hlsl_base_type t2)
1154 {
1155 static const enum hlsl_base_type types[] =
1156 {
1157 HLSL_TYPE_BOOL,
1158 HLSL_TYPE_INT,
1159 HLSL_TYPE_UINT,
1160 HLSL_TYPE_HALF,
1161 HLSL_TYPE_FLOAT,
1162 HLSL_TYPE_DOUBLE,
1163 };
1164 int t1_idx = -1, t2_idx = -1, i;
1165
1166 for (i = 0; i < ARRAY_SIZE(types); ++i)
1167 {
1168 /* Always convert away from HLSL_TYPE_HALF */
1169 if (t1 == types[i])
1170 t1_idx = t1 == HLSL_TYPE_HALF ? i + 1 : i;
1171 if (t2 == types[i])
1172 t2_idx = t2 == HLSL_TYPE_HALF ? i + 1 : i;
1173
1174 if (t1_idx != -1 && t2_idx != -1)
1175 break;
1176 }
1177 if (t1_idx == -1 || t2_idx == -1)
1178 {
1179 FIXME("Unexpected base type.\n");
1180 return HLSL_TYPE_FLOAT;
1181 }
1182 return t1_idx >= t2_idx ? t1 : t2;
1183 }
1184
1185 static struct hlsl_type *expr_common_type(struct hlsl_type *t1, struct hlsl_type *t2,
1186 struct source_location *loc)
1187 {
1188 enum hlsl_type_class type;
1189 enum hlsl_base_type base;
1190 unsigned int dimx, dimy;
1191
1192 if (t1->type > HLSL_CLASS_LAST_NUMERIC || t2->type > HLSL_CLASS_LAST_NUMERIC)
1193 {
1194 hlsl_report_message(loc->file, loc->line, loc->col, HLSL_LEVEL_ERROR,
1195 "non scalar/vector/matrix data type in expression");
1196 return NULL;
1197 }
1198
1199 if (compare_hlsl_types(t1, t2))
1200 return t1;
1201
1202 if (!expr_compatible_data_types(t1, t2))
1203 {
1204 hlsl_report_message(loc->file, loc->line, loc->col, HLSL_LEVEL_ERROR,
1205 "expression data types are incompatible");
1206 return NULL;
1207 }
1208
1209 if (t1->base_type == t2->base_type)
1210 base = t1->base_type;
1211 else
1212 base = expr_common_base_type(t1->base_type, t2->base_type);
1213
1214 if (t1->dimx == 1 && t1->dimy == 1)
1215 {
1216 type = t2->type;
1217 dimx = t2->dimx;
1218 dimy = t2->dimy;
1219 }
1220 else if (t2->dimx == 1 && t2->dimy == 1)
1221 {
1222 type = t1->type;
1223 dimx = t1->dimx;
1224 dimy = t1->dimy;
1225 }
1226 else if (t1->type == HLSL_CLASS_MATRIX && t2->type == HLSL_CLASS_MATRIX)
1227 {
1228 type = HLSL_CLASS_MATRIX;
1229 dimx = min(t1->dimx, t2->dimx);
1230 dimy = min(t1->dimy, t2->dimy);
1231 }
1232 else
1233 {
1234 /* Two vectors or a vector and a matrix (matrix must be 1xn or nx1) */
1235 unsigned int max_dim_1, max_dim_2;
1236
1237 max_dim_1 = max(t1->dimx, t1->dimy);
1238 max_dim_2 = max(t2->dimx, t2->dimy);
1239 if (t1->dimx * t1->dimy == t2->dimx * t2->dimy)
1240 {
1241 type = HLSL_CLASS_VECTOR;
1242 dimx = max(t1->dimx, t2->dimx);
1243 dimy = 1;
1244 }
1245 else if (max_dim_1 <= max_dim_2)
1246 {
1247 type = t1->type;
1248 if (type == HLSL_CLASS_VECTOR)
1249 {
1250 dimx = max_dim_1;
1251 dimy = 1;
1252 }
1253 else
1254 {
1255 dimx = t1->dimx;
1256 dimy = t1->dimy;
1257 }
1258 }
1259 else
1260 {
1261 type = t2->type;
1262 if (type == HLSL_CLASS_VECTOR)
1263 {
1264 dimx = max_dim_2;
1265 dimy = 1;
1266 }
1267 else
1268 {
1269 dimx = t2->dimx;
1270 dimy = t2->dimy;
1271 }
1272 }
1273 }
1274
1275 return new_hlsl_type(NULL, type, base, dimx, dimy);
1276 }
1277
1278 static struct hlsl_ir_node *implicit_conversion(struct hlsl_ir_node *node, struct hlsl_type *type,
1279 struct source_location *loc)
1280 {
1281 if (compare_hlsl_types(node->data_type, type))
1282 return node;
1283 TRACE("Implicit conversion of expression to %s\n", debug_hlsl_type(type));
1284 return &new_cast(node, type, loc)->node;
1285 }
1286
1287 struct hlsl_ir_expr *new_expr(enum hlsl_ir_expr_op op, struct hlsl_ir_node **operands,
1288 struct source_location *loc)
1289 {
1290 struct hlsl_ir_expr *expr = d3dcompiler_alloc(sizeof(*expr));
1291 struct hlsl_type *type;
1292 unsigned int i;
1293
1294 if (!expr)
1295 {
1296 ERR("Out of memory\n");
1297 return NULL;
1298 }
1299 expr->node.type = HLSL_IR_EXPR;
1300 expr->node.loc = *loc;
1301 type = operands[0]->data_type;
1302 for (i = 1; i <= 2; ++i)
1303 {
1304 if (!operands[i])
1305 break;
1306 type = expr_common_type(type, operands[i]->data_type, loc);
1307 if (!type)
1308 {
1309 d3dcompiler_free(expr);
1310 return NULL;
1311 }
1312 }
1313 for (i = 0; i <= 2; ++i)
1314 {
1315 if (!operands[i])
1316 break;
1317 if (compare_hlsl_types(operands[i]->data_type, type))
1318 continue;
1319 TRACE("Implicitly converting %s into %s in an expression\n", debug_hlsl_type(operands[i]->data_type), debug_hlsl_type(type));
1320 if (operands[i]->data_type->dimx * operands[i]->data_type->dimy != 1
1321 && operands[i]->data_type->dimx * operands[i]->data_type->dimy != type->dimx * type->dimy)
1322 {
1323 hlsl_report_message(operands[i]->loc.file,
1324 operands[i]->loc.line, operands[i]->loc.col, HLSL_LEVEL_WARNING,
1325 "implicit truncation of vector/matrix type");
1326 }
1327 operands[i] = implicit_conversion(operands[i], type, &operands[i]->loc);
1328 if (!operands[i])
1329 {
1330 ERR("Impossible to convert expression operand %u to %s\n", i + 1, debug_hlsl_type(type));
1331 d3dcompiler_free(expr);
1332 return NULL;
1333 }
1334 }
1335 expr->node.data_type = type;
1336 expr->op = op;
1337 expr->operands[0] = operands[0];
1338 expr->operands[1] = operands[1];
1339 expr->operands[2] = operands[2];
1340
1341 return expr;
1342 }
1343
1344 struct hlsl_ir_expr *new_cast(struct hlsl_ir_node *node, struct hlsl_type *type,
1345 struct source_location *loc)
1346 {
1347 struct hlsl_ir_node *cast;
1348
1349 cast = new_unary_expr(HLSL_IR_UNOP_CAST, node, *loc);
1350 if (cast)
1351 cast->data_type = type;
1352 return expr_from_node(cast);
1353 }
1354
1355 struct hlsl_ir_deref *new_var_deref(struct hlsl_ir_var *var)
1356 {
1357 struct hlsl_ir_deref *deref = d3dcompiler_alloc(sizeof(*deref));
1358
1359 if (!deref)
1360 {
1361 ERR("Out of memory.\n");
1362 return NULL;
1363 }
1364 deref->node.type = HLSL_IR_DEREF;
1365 deref->node.data_type = var->data_type;
1366 deref->type = HLSL_IR_DEREF_VAR;
1367 deref->v.var = var;
1368 return deref;
1369 }
1370
1371 struct hlsl_ir_deref *new_record_deref(struct hlsl_ir_node *record, struct hlsl_struct_field *field)
1372 {
1373 struct hlsl_ir_deref *deref = d3dcompiler_alloc(sizeof(*deref));
1374
1375 if (!deref)
1376 {
1377 ERR("Out of memory.\n");
1378 return NULL;
1379 }
1380 deref->node.type = HLSL_IR_DEREF;
1381 deref->node.data_type = field->type;
1382 deref->type = HLSL_IR_DEREF_RECORD;
1383 deref->v.record.record = record;
1384 deref->v.record.field = field;
1385 return deref;
1386 }
1387
1388 static enum hlsl_ir_expr_op op_from_assignment(enum parse_assign_op op)
1389 {
1390 static const enum hlsl_ir_expr_op ops[] =
1391 {
1392 0,
1393 HLSL_IR_BINOP_ADD,
1394 HLSL_IR_BINOP_SUB,
1395 HLSL_IR_BINOP_MUL,
1396 HLSL_IR_BINOP_DIV,
1397 HLSL_IR_BINOP_MOD,
1398 HLSL_IR_BINOP_LSHIFT,
1399 HLSL_IR_BINOP_RSHIFT,
1400 HLSL_IR_BINOP_BIT_AND,
1401 HLSL_IR_BINOP_BIT_OR,
1402 HLSL_IR_BINOP_BIT_XOR,
1403 };
1404
1405 return ops[op];
1406 }
1407
1408 struct hlsl_ir_node *make_assignment(struct hlsl_ir_node *left, enum parse_assign_op assign_op,
1409 DWORD writemask, struct hlsl_ir_node *right)
1410 {
1411 struct hlsl_ir_assignment *assign = d3dcompiler_alloc(sizeof(*assign));
1412 struct hlsl_type *type;
1413 struct hlsl_ir_node *lhs, *rhs;
1414
1415 if (!assign)
1416 {
1417 ERR("Out of memory\n");
1418 return NULL;
1419 }
1420
1421 TRACE("Creating proper assignment expression.\n");
1422 rhs = right;
1423 if (writemask == BWRITERSP_WRITEMASK_ALL)
1424 type = left->data_type;
1425 else
1426 {
1427 unsigned int dimx = 0;
1428 DWORD bitmask;
1429 enum hlsl_type_class type_class;
1430
1431 if (left->data_type->type > HLSL_CLASS_LAST_NUMERIC)
1432 {
1433 hlsl_report_message(left->loc.file, left->loc.line, left->loc.col, HLSL_LEVEL_ERROR,
1434 "writemask on a non scalar/vector/matrix type");
1435 d3dcompiler_free(assign);
1436 return NULL;
1437 }
1438 bitmask = writemask & ((1 << left->data_type->dimx) - 1);
1439 while (bitmask)
1440 {
1441 if (bitmask & 1)
1442 dimx++;
1443 bitmask >>= 1;
1444 }
1445 if (left->data_type->type == HLSL_CLASS_MATRIX)
1446 FIXME("Assignments with writemasks and matrices on lhs are not supported yet.\n");
1447 if (dimx == 1)
1448 type_class = HLSL_CLASS_SCALAR;
1449 else
1450 type_class = left->data_type->type;
1451 type = new_hlsl_type(NULL, type_class, left->data_type->base_type, dimx, 1);
1452 }
1453 assign->node.type = HLSL_IR_ASSIGNMENT;
1454 assign->node.loc = left->loc;
1455 assign->node.data_type = type;
1456 assign->writemask = writemask;
1457 FIXME("Check for casts in the lhs.\n");
1458
1459 lhs = left;
1460 /* FIXME: check for invalid writemasks on the lhs. */
1461
1462 if (!compare_hlsl_types(type, rhs->data_type))
1463 {
1464 struct hlsl_ir_node *converted_rhs;
1465
1466 if (!implicit_compatible_data_types(rhs->data_type, type))
1467 {
1468 hlsl_report_message(rhs->loc.file, rhs->loc.line, rhs->loc.col, HLSL_LEVEL_ERROR,
1469 "can't implicitly convert %s to %s",
1470 debug_hlsl_type(rhs->data_type), debug_hlsl_type(type));
1471 free_instr(lhs);
1472 free_instr(rhs);
1473 d3dcompiler_free(assign);
1474 return NULL;
1475 }
1476 if (lhs->data_type->dimx * lhs->data_type->dimy < rhs->data_type->dimx * rhs->data_type->dimy)
1477 hlsl_report_message(rhs->loc.file, rhs->loc.line, rhs->loc.col, HLSL_LEVEL_WARNING,
1478 "implicit truncation of vector type");
1479
1480 converted_rhs = implicit_conversion(rhs, type, &rhs->loc);
1481 if (!converted_rhs)
1482 {
1483 ERR("Couldn't implicitly convert expression to %s.\n", debug_hlsl_type(type));
1484 free_instr(lhs);
1485 free_instr(rhs);
1486 d3dcompiler_free(assign);
1487 return NULL;
1488 }
1489 rhs = converted_rhs;
1490 }
1491
1492 assign->lhs = lhs;
1493 if (assign_op != ASSIGN_OP_ASSIGN)
1494 {
1495 enum hlsl_ir_expr_op op = op_from_assignment(assign_op);
1496 struct hlsl_ir_node *expr;
1497
1498 if (lhs->type != HLSL_IR_DEREF || deref_from_node(lhs)->type != HLSL_IR_DEREF_VAR)
1499 {
1500 FIXME("LHS expression not supported in compound assignments yet.\n");
1501 assign->rhs = rhs;
1502 }
1503 else
1504 {
1505 struct hlsl_ir_deref *lhs_deref = deref_from_node(lhs), *new_deref;
1506
1507 TRACE("Adding an expression for the compound assignment.\n");
1508 new_deref = new_var_deref(lhs_deref->v.var);
1509 expr = new_binary_expr(op, &new_deref->node, rhs, left->loc);
1510 assign->rhs = expr;
1511 }
1512 }
1513 else
1514 assign->rhs = rhs;
1515
1516 return &assign->node;
1517 }
1518
1519 static int compare_hlsl_types_rb(const void *key, const struct wine_rb_entry *entry)
1520 {
1521 const char *name = key;
1522 const struct hlsl_type *type = WINE_RB_ENTRY_VALUE(entry, const struct hlsl_type, scope_entry);
1523
1524 if (name == type->name)
1525 return 0;
1526
1527 if (!name || !type->name)
1528 {
1529 ERR("hlsl_type without a name in a scope?\n");
1530 return -1;
1531 }
1532 return strcmp(name, type->name);
1533 }
1534
1535 void push_scope(struct hlsl_parse_ctx *ctx)
1536 {
1537 struct hlsl_scope *new_scope = d3dcompiler_alloc(sizeof(*new_scope));
1538
1539 if (!new_scope)
1540 {
1541 ERR("Out of memory!\n");
1542 return;
1543 }
1544 TRACE("Pushing a new scope\n");
1545 list_init(&new_scope->vars);
1546 wine_rb_init(&new_scope->types, compare_hlsl_types_rb);
1547 new_scope->upper = ctx->cur_scope;
1548 ctx->cur_scope = new_scope;
1549 list_add_tail(&ctx->scopes, &new_scope->entry);
1550 }
1551
1552 BOOL pop_scope(struct hlsl_parse_ctx *ctx)
1553 {
1554 struct hlsl_scope *prev_scope = ctx->cur_scope->upper;
1555 if (!prev_scope)
1556 return FALSE;
1557
1558 TRACE("Popping current scope\n");
1559 ctx->cur_scope = prev_scope;
1560 return TRUE;
1561 }
1562
1563 struct hlsl_ir_function_decl *new_func_decl(struct hlsl_type *return_type, struct list *parameters)
1564 {
1565 struct hlsl_ir_function_decl *decl;
1566
1567 decl = d3dcompiler_alloc(sizeof(*decl));
1568 if (!decl)
1569 {
1570 ERR("Out of memory.\n");
1571 return NULL;
1572 }
1573 decl->return_type = return_type;
1574 decl->parameters = parameters;
1575
1576 return decl;
1577 }
1578
1579 static int compare_param_hlsl_types(const struct hlsl_type *t1, const struct hlsl_type *t2)
1580 {
1581 if (t1->type != t2->type)
1582 {
1583 if (!((t1->type == HLSL_CLASS_SCALAR && t2->type == HLSL_CLASS_VECTOR)
1584 || (t1->type == HLSL_CLASS_VECTOR && t2->type == HLSL_CLASS_SCALAR)))
1585 return t1->type - t2->type;
1586 }
1587 if (t1->base_type != t2->base_type)
1588 return t1->base_type - t2->base_type;
1589 if (t1->base_type == HLSL_TYPE_SAMPLER && t1->sampler_dim != t2->sampler_dim)
1590 return t1->sampler_dim - t2->sampler_dim;
1591 if (t1->dimx != t2->dimx)
1592 return t1->dimx - t2->dimx;
1593 if (t1->dimy != t2->dimy)
1594 return t1->dimx - t2->dimx;
1595 if (t1->type == HLSL_CLASS_STRUCT)
1596 {
1597 struct list *t1cur, *t2cur;
1598 struct hlsl_struct_field *t1field, *t2field;
1599 int r;
1600
1601 t1cur = list_head(t1->e.elements);
1602 t2cur = list_head(t2->e.elements);
1603 while (t1cur && t2cur)
1604 {
1605 t1field = LIST_ENTRY(t1cur, struct hlsl_struct_field, entry);
1606 t2field = LIST_ENTRY(t2cur, struct hlsl_struct_field, entry);
1607 if ((r = compare_param_hlsl_types(t1field->type, t2field->type)))
1608 return r;
1609 if ((r = strcmp(t1field->name, t2field->name)))
1610 return r;
1611 t1cur = list_next(t1->e.elements, t1cur);
1612 t2cur = list_next(t2->e.elements, t2cur);
1613 }
1614 if (t1cur != t2cur)
1615 return t1cur ? 1 : -1;
1616 return 0;
1617 }
1618 if (t1->type == HLSL_CLASS_ARRAY)
1619 {
1620 if (t1->e.array.elements_count != t2->e.array.elements_count)
1621 return t1->e.array.elements_count - t2->e.array.elements_count;
1622 return compare_param_hlsl_types(t1->e.array.type, t2->e.array.type);
1623 }
1624
1625 return 0;
1626 }
1627
1628 static int compare_function_decl_rb(const void *key, const struct wine_rb_entry *entry)
1629 {
1630 const struct list *params = key;
1631 const struct hlsl_ir_function_decl *decl = WINE_RB_ENTRY_VALUE(entry, const struct hlsl_ir_function_decl, entry);
1632 int params_count = params ? list_count(params) : 0;
1633 int decl_params_count = decl->parameters ? list_count(decl->parameters) : 0;
1634 int r;
1635 struct list *p1cur, *p2cur;
1636
1637 if (params_count != decl_params_count)
1638 return params_count - decl_params_count;
1639
1640 p1cur = params ? list_head(params) : NULL;
1641 p2cur = decl->parameters ? list_head(decl->parameters) : NULL;
1642 while (p1cur && p2cur)
1643 {
1644 struct hlsl_ir_var *p1, *p2;
1645 p1 = LIST_ENTRY(p1cur, struct hlsl_ir_var, param_entry);
1646 p2 = LIST_ENTRY(p2cur, struct hlsl_ir_var, param_entry);
1647 if ((r = compare_param_hlsl_types(p1->data_type, p2->data_type)))
1648 return r;
1649 p1cur = list_next(params, p1cur);
1650 p2cur = list_next(decl->parameters, p2cur);
1651 }
1652 return 0;
1653 }
1654
1655 static int compare_function_rb(const void *key, const struct wine_rb_entry *entry)
1656 {
1657 const char *name = key;
1658 const struct hlsl_ir_function *func = WINE_RB_ENTRY_VALUE(entry, const struct hlsl_ir_function,entry);
1659
1660 return strcmp(name, func->name);
1661 }
1662
1663 void init_functions_tree(struct wine_rb_tree *funcs)
1664 {
1665 wine_rb_init(&hlsl_ctx.functions, compare_function_rb);
1666 }
1667
1668 static const char *debug_base_type(const struct hlsl_type *type)
1669 {
1670 const char *name = "(unknown)";
1671
1672 switch (type->base_type)
1673 {
1674 case HLSL_TYPE_FLOAT: name = "float"; break;
1675 case HLSL_TYPE_HALF: name = "half"; break;
1676 case HLSL_TYPE_DOUBLE: name = "double"; break;
1677 case HLSL_TYPE_INT: name = "int"; break;
1678 case HLSL_TYPE_UINT: name = "uint"; break;
1679 case HLSL_TYPE_BOOL: name = "bool"; break;
1680 case HLSL_TYPE_SAMPLER:
1681 switch (type->sampler_dim)
1682 {
1683 case HLSL_SAMPLER_DIM_GENERIC: name = "sampler"; break;
1684 case HLSL_SAMPLER_DIM_1D: name = "sampler1D"; break;
1685 case HLSL_SAMPLER_DIM_2D: name = "sampler2D"; break;
1686 case HLSL_SAMPLER_DIM_3D: name = "sampler3D"; break;
1687 case HLSL_SAMPLER_DIM_CUBE: name = "samplerCUBE"; break;
1688 }
1689 break;
1690 default:
1691 FIXME("Unhandled case %u\n", type->base_type);
1692 }
1693 return name;
1694 }
1695
1696 const char *debug_hlsl_type(const struct hlsl_type *type)
1697 {
1698 const char *name;
1699
1700 if (type->name)
1701 return debugstr_a(type->name);
1702
1703 if (type->type == HLSL_CLASS_STRUCT)
1704 return "<anonymous struct>";
1705
1706 if (type->type == HLSL_CLASS_ARRAY)
1707 {
1708 name = debug_base_type(type->e.array.type);
1709 return wine_dbg_sprintf("%s[%u]", name, type->e.array.elements_count);
1710 }
1711
1712 name = debug_base_type(type);
1713
1714 if (type->type == HLSL_CLASS_SCALAR)
1715 return wine_dbg_sprintf("%s", name);
1716 if (type->type == HLSL_CLASS_VECTOR)
1717 return wine_dbg_sprintf("%s%u", name, type->dimx);
1718 if (type->type == HLSL_CLASS_MATRIX)
1719 return wine_dbg_sprintf("%s%ux%u", name, type->dimx, type->dimy);
1720 return "unexpected_type";
1721 }
1722
1723 const char *debug_modifiers(DWORD modifiers)
1724 {
1725 char string[110];
1726
1727 string[0] = 0;
1728 if (modifiers & HLSL_STORAGE_EXTERN)
1729 strcat(string, " extern"); /* 7 */
1730 if (modifiers & HLSL_STORAGE_NOINTERPOLATION)
1731 strcat(string, " nointerpolation"); /* 16 */
1732 if (modifiers & HLSL_MODIFIER_PRECISE)
1733 strcat(string, " precise"); /* 8 */
1734 if (modifiers & HLSL_STORAGE_SHARED)
1735 strcat(string, " shared"); /* 7 */
1736 if (modifiers & HLSL_STORAGE_GROUPSHARED)
1737 strcat(string, " groupshared"); /* 12 */
1738 if (modifiers & HLSL_STORAGE_STATIC)
1739 strcat(string, " static"); /* 7 */
1740 if (modifiers & HLSL_STORAGE_UNIFORM)
1741 strcat(string, " uniform"); /* 8 */
1742 if (modifiers & HLSL_STORAGE_VOLATILE)
1743 strcat(string, " volatile"); /* 9 */
1744 if (modifiers & HLSL_MODIFIER_CONST)
1745 strcat(string, " const"); /* 6 */
1746 if (modifiers & HLSL_MODIFIER_ROW_MAJOR)
1747 strcat(string, " row_major"); /* 10 */
1748 if (modifiers & HLSL_MODIFIER_COLUMN_MAJOR)
1749 strcat(string, " column_major"); /* 13 */
1750 if ((modifiers & (HLSL_MODIFIER_IN | HLSL_MODIFIER_OUT)) == (HLSL_MODIFIER_IN | HLSL_MODIFIER_OUT))
1751 strcat(string, " inout"); /* 6 */
1752 else if (modifiers & HLSL_MODIFIER_IN)
1753 strcat(string, " in"); /* 3 */
1754 else if (modifiers & HLSL_MODIFIER_OUT)
1755 strcat(string, " out"); /* 4 */
1756
1757 return wine_dbg_sprintf("%s", string[0] ? string + 1 : "");
1758 }
1759
1760 static const char *debug_node_type(enum hlsl_ir_node_type type)
1761 {
1762 static const char * const names[] =
1763 {
1764 "HLSL_IR_ASSIGNMENT",
1765 "HLSL_IR_CONSTANT",
1766 "HLSL_IR_CONSTRUCTOR",
1767 "HLSL_IR_DEREF",
1768 "HLSL_IR_EXPR",
1769 "HLSL_IR_IF",
1770 "HLSL_IR_JUMP",
1771 "HLSL_IR_SWIZZLE",
1772 };
1773
1774 if (type >= ARRAY_SIZE(names))
1775 return "Unexpected node type";
1776 return names[type];
1777 }
1778
1779 static void debug_dump_instr(const struct hlsl_ir_node *instr);
1780
1781 static void debug_dump_instr_list(const struct list *list)
1782 {
1783 struct hlsl_ir_node *instr;
1784
1785 LIST_FOR_EACH_ENTRY(instr, list, struct hlsl_ir_node, entry)
1786 {
1787 debug_dump_instr(instr);
1788 TRACE("\n");
1789 }
1790 }
1791
1792 static void debug_dump_ir_var(const struct hlsl_ir_var *var)
1793 {
1794 if (var->modifiers)
1795 TRACE("%s ", debug_modifiers(var->modifiers));
1796 TRACE("%s %s", debug_hlsl_type(var->data_type), var->name);
1797 if (var->semantic)
1798 TRACE(" : %s", debugstr_a(var->semantic));
1799 }
1800
1801 static void debug_dump_ir_deref(const struct hlsl_ir_deref *deref)
1802 {
1803 switch (deref->type)
1804 {
1805 case HLSL_IR_DEREF_VAR:
1806 TRACE("deref(");
1807 debug_dump_ir_var(deref->v.var);
1808 TRACE(")");
1809 break;
1810 case HLSL_IR_DEREF_ARRAY:
1811 debug_dump_instr(deref->v.array.array);
1812 TRACE("[");
1813 debug_dump_instr(deref->v.array.index);
1814 TRACE("]");
1815 break;
1816 case HLSL_IR_DEREF_RECORD:
1817 debug_dump_instr(deref->v.record.record);
1818 TRACE(".%s", debugstr_a(deref->v.record.field->name));
1819 break;
1820 }
1821 }
1822
1823 static void debug_dump_ir_constant(const struct hlsl_ir_constant *constant)
1824 {
1825 struct hlsl_type *type = constant->node.data_type;
1826 unsigned int x, y;
1827
1828 if (type->dimy != 1)
1829 TRACE("{");
1830 for (y = 0; y < type->dimy; ++y)
1831 {
1832 if (type->dimx != 1)
1833 TRACE("{");
1834 for (x = 0; x < type->dimx; ++x)
1835 {
1836 switch (type->base_type)
1837 {
1838 case HLSL_TYPE_FLOAT:
1839 TRACE("%g ", (double)constant->v.value.f[y * type->dimx + x]);
1840 break;
1841 case HLSL_TYPE_DOUBLE:
1842 TRACE("%g ", constant->v.value.d[y * type->dimx + x]);
1843 break;
1844 case HLSL_TYPE_INT:
1845 TRACE("%d ", constant->v.value.i[y * type->dimx + x]);
1846 break;
1847 case HLSL_TYPE_UINT:
1848 TRACE("%u ", constant->v.value.u[y * type->dimx + x]);
1849 break;
1850 case HLSL_TYPE_BOOL:
1851 TRACE("%s ", constant->v.value.b[y * type->dimx + x] == FALSE ? "false" : "true");
1852 break;
1853 default:
1854 TRACE("Constants of type %s not supported\n", debug_base_type(type));
1855 }
1856 }
1857 if (type->dimx != 1)
1858 TRACE("}");
1859 }
1860 if (type->dimy != 1)
1861 TRACE("}");
1862 }
1863
1864 static const char *debug_expr_op(const struct hlsl_ir_expr *expr)
1865 {
1866 static const char * const op_names[] =
1867 {
1868 "~",
1869 "!",
1870 "-",
1871 "abs",
1872 "sign",
1873 "rcp",
1874 "rsq",
1875 "sqrt",
1876 "nrm",
1877 "exp2",
1878 "log2",
1879
1880 "cast",
1881
1882 "fract",
1883
1884 "sin",
1885 "cos",
1886 "sin_reduced",
1887 "cos_reduced",
1888
1889 "dsx",
1890 "dsy",
1891
1892 "sat",
1893
1894 "pre++",
1895 "pre--",
1896 "post++",
1897 "post--",
1898
1899 "+",
1900 "-",
1901 "*",
1902 "/",
1903
1904 "%",
1905
1906 "<",
1907 ">",
1908 "<=",
1909 ">=",
1910 "==",
1911 "!=",
1912
1913 "&&",
1914 "||",
1915
1916 "<<",
1917 ">>",
1918 "&",
1919 "|",
1920 "^",
1921
1922 "dot",
1923 "crs",
1924 "min",
1925 "max",
1926
1927 "pow",
1928
1929 "lerp",
1930
1931 ",",
1932 };
1933
1934 if (expr->op == HLSL_IR_UNOP_CAST)
1935 return debug_hlsl_type(expr->node.data_type);
1936
1937 return op_names[expr->op];
1938 }
1939
1940 /* Dumps the expression in a prefix "operator (operands)" form */
1941 static void debug_dump_ir_expr(const struct hlsl_ir_expr *expr)
1942 {
1943 unsigned int i;
1944
1945 TRACE("%s (", debug_expr_op(expr));
1946 for (i = 0; i < 3 && expr->operands[i]; ++i)
1947 {
1948 debug_dump_instr(expr->operands[i]);
1949 TRACE(" ");
1950 }
1951 TRACE(")");
1952 }
1953
1954 static void debug_dump_ir_constructor(const struct hlsl_ir_constructor *constructor)
1955 {
1956 unsigned int i;
1957
1958 TRACE("%s (", debug_hlsl_type(constructor->node.data_type));
1959 for (i = 0; i < constructor->args_count; ++i)
1960 {
1961 debug_dump_instr(constructor->args[i]);
1962 TRACE(" ");
1963 }
1964 TRACE(")");
1965 }
1966
1967 static const char *debug_writemask(DWORD writemask)
1968 {
1969 static const char components[] = {'x', 'y', 'z', 'w'};
1970 char string[5];
1971 unsigned int i = 0, pos = 0;
1972
1973 assert(!(writemask & ~BWRITERSP_WRITEMASK_ALL));
1974
1975 while (writemask)
1976 {
1977 if (writemask & 1)
1978 string[pos++] = components[i];
1979 writemask >>= 1;
1980 i++;
1981 }
1982 string[pos] = '\0';
1983 return wine_dbg_sprintf(".%s", string);
1984 }
1985
1986 static void debug_dump_ir_assignment(const struct hlsl_ir_assignment *assign)
1987 {
1988 TRACE("= (");
1989 debug_dump_instr(assign->lhs);
1990 if (assign->writemask != BWRITERSP_WRITEMASK_ALL)
1991 TRACE("%s", debug_writemask(assign->writemask));
1992 TRACE(" ");
1993 debug_dump_instr(assign->rhs);
1994 TRACE(")");
1995 }
1996
1997 static void debug_dump_ir_swizzle(const struct hlsl_ir_swizzle *swizzle)
1998 {
1999 unsigned int i;
2000
2001 debug_dump_instr(swizzle->val);
2002 TRACE(".");
2003 if (swizzle->val->data_type->dimy > 1)
2004 {
2005 for (i = 0; i < swizzle->node.data_type->dimx; ++i)
2006 TRACE("_m%u%u", (swizzle->swizzle >> i * 8) & 0xf, (swizzle->swizzle >> (i * 8 + 4)) & 0xf);
2007 }
2008 else
2009 {
2010 static const char c[] = {'x', 'y', 'z', 'w'};
2011
2012 for (i = 0; i < swizzle->node.data_type->dimx; ++i)
2013 TRACE("%c", c[(swizzle->swizzle >> i * 2) & 0x3]);
2014 }
2015 }
2016
2017 static void debug_dump_ir_jump(const struct hlsl_ir_jump *jump)
2018 {
2019 switch (jump->type)
2020 {
2021 case HLSL_IR_JUMP_BREAK:
2022 TRACE("break");
2023 break;
2024 case HLSL_IR_JUMP_CONTINUE:
2025 TRACE("continue");
2026 break;
2027 case HLSL_IR_JUMP_DISCARD:
2028 TRACE("discard");
2029 break;
2030 case HLSL_IR_JUMP_RETURN:
2031 TRACE("return ");
2032 if (jump->return_value)
2033 debug_dump_instr(jump->return_value);
2034 TRACE(";");
2035 break;
2036 }
2037 }
2038
2039 static void debug_dump_ir_if(const struct hlsl_ir_if *if_node)
2040 {
2041 TRACE("if (");
2042 debug_dump_instr(if_node->condition);
2043 TRACE(")\n{\n");
2044 debug_dump_instr_list(if_node->then_instrs);
2045 TRACE("}\n");
2046 if (if_node->else_instrs)
2047 {
2048 TRACE("else\n{\n");
2049 debug_dump_instr_list(if_node->else_instrs);
2050 TRACE("}\n");
2051 }
2052 }
2053
2054 static void debug_dump_instr(const struct hlsl_ir_node *instr)
2055 {
2056 switch (instr->type)
2057 {
2058 case HLSL_IR_EXPR:
2059 debug_dump_ir_expr(expr_from_node(instr));
2060 break;
2061 case HLSL_IR_DEREF:
2062 debug_dump_ir_deref(deref_from_node(instr));
2063 break;
2064 case HLSL_IR_CONSTANT:
2065 debug_dump_ir_constant(constant_from_node(instr));
2066 break;
2067 case HLSL_IR_ASSIGNMENT:
2068 debug_dump_ir_assignment(assignment_from_node(instr));
2069 break;
2070 case HLSL_IR_SWIZZLE:
2071 debug_dump_ir_swizzle(swizzle_from_node(instr));
2072 break;
2073 case HLSL_IR_CONSTRUCTOR:
2074 debug_dump_ir_constructor(constructor_from_node(instr));
2075 break;
2076 case HLSL_IR_JUMP:
2077 debug_dump_ir_jump(jump_from_node(instr));
2078 break;
2079 case HLSL_IR_IF:
2080 debug_dump_ir_if(if_from_node(instr));
2081 break;
2082 default:
2083 TRACE("<No dump function for %s>", debug_node_type(instr->type));
2084 }
2085 }
2086
2087 void debug_dump_ir_function_decl(const struct hlsl_ir_function_decl *func)
2088 {
2089 struct hlsl_ir_var *param;
2090
2091 TRACE("Dumping function %s.\n", debugstr_a(func->func->name));
2092 TRACE("Function parameters:\n");
2093 LIST_FOR_EACH_ENTRY(param, func->parameters, struct hlsl_ir_var, param_entry)
2094 {
2095 debug_dump_ir_var(param);
2096 TRACE("\n");
2097 }
2098 if (func->semantic)
2099 TRACE("Function semantic: %s\n", debugstr_a(func->semantic));
2100 if (func->body)
2101 {
2102 debug_dump_instr_list(func->body);
2103 }
2104 }
2105
2106 void free_hlsl_type(struct hlsl_type *type)
2107 {
2108 struct hlsl_struct_field *field, *next_field;
2109
2110 d3dcompiler_free((void *)type->name);
2111 if (type->type == HLSL_CLASS_STRUCT)
2112 {
2113 LIST_FOR_EACH_ENTRY_SAFE(field, next_field, type->e.elements, struct hlsl_struct_field, entry)
2114 {
2115 d3dcompiler_free((void *)field->name);
2116 d3dcompiler_free((void *)field->semantic);
2117 d3dcompiler_free(field);
2118 }
2119 }
2120 d3dcompiler_free(type);
2121 }
2122
2123 void free_instr_list(struct list *list)
2124 {
2125 struct hlsl_ir_node *node, *next_node;
2126
2127 if (!list)
2128 return;
2129 LIST_FOR_EACH_ENTRY_SAFE(node, next_node, list, struct hlsl_ir_node, entry)
2130 free_instr(node);
2131 d3dcompiler_free(list);
2132 }
2133
2134 static void free_ir_constant(struct hlsl_ir_constant *constant)
2135 {
2136 struct hlsl_type *type = constant->node.data_type;
2137 unsigned int i;
2138 struct hlsl_ir_constant *field, *next_field;
2139
2140 switch (type->type)
2141 {
2142 case HLSL_CLASS_ARRAY:
2143 for (i = 0; i < type->e.array.elements_count; ++i)
2144 free_ir_constant(&constant->v.array_elements[i]);
2145 d3dcompiler_free(constant->v.array_elements);
2146 break;
2147 case HLSL_CLASS_STRUCT:
2148 LIST_FOR_EACH_ENTRY_SAFE(field, next_field, constant->v.struct_elements, struct hlsl_ir_constant, node.entry)
2149 free_ir_constant(field);
2150 break;
2151 default:
2152 break;
2153 }
2154 d3dcompiler_free(constant);
2155 }
2156
2157 static void free_ir_deref(struct hlsl_ir_deref *deref)
2158 {
2159 switch (deref->type)
2160 {
2161 case HLSL_IR_DEREF_VAR:
2162 /* Variables are shared among nodes in the tree. */
2163 break;
2164 case HLSL_IR_DEREF_ARRAY:
2165 free_instr(deref->v.array.array);
2166 free_instr(deref->v.array.index);
2167 break;
2168 case HLSL_IR_DEREF_RECORD:
2169 free_instr(deref->v.record.record);
2170 break;
2171 }
2172 d3dcompiler_free(deref);
2173 }
2174
2175 static void free_ir_swizzle(struct hlsl_ir_swizzle *swizzle)
2176 {
2177 free_instr(swizzle->val);
2178 d3dcompiler_free(swizzle);
2179 }
2180
2181 static void free_ir_constructor(struct hlsl_ir_constructor *constructor)
2182 {
2183 unsigned int i;
2184 for (i = 0; i < constructor->args_count; ++i)
2185 free_instr(constructor->args[i]);
2186 d3dcompiler_free(constructor);
2187 }
2188
2189 static void free_ir_expr(struct hlsl_ir_expr *expr)
2190 {
2191 unsigned int i;
2192
2193 for (i = 0; i < 3; ++i)
2194 {
2195 if (!expr->operands[i])
2196 break;
2197 free_instr(expr->operands[i]);
2198 }
2199 free_instr_list(expr->subexpressions);
2200 d3dcompiler_free(expr);
2201 }
2202
2203 static void free_ir_assignment(struct hlsl_ir_assignment *assignment)
2204 {
2205 free_instr(assignment->lhs);
2206 free_instr(assignment->rhs);
2207 d3dcompiler_free(assignment);
2208 }
2209
2210 static void free_ir_if(struct hlsl_ir_if *if_node)
2211 {
2212 free_instr(if_node->condition);
2213 free_instr_list(if_node->then_instrs);
2214 free_instr_list(if_node->else_instrs);
2215 d3dcompiler_free(if_node);
2216 }
2217
2218 static void free_ir_jump(struct hlsl_ir_jump *jump)
2219 {
2220 if (jump->type == HLSL_IR_JUMP_RETURN)
2221 free_instr(jump->return_value);
2222 d3dcompiler_free(jump);
2223 }
2224
2225 void free_instr(struct hlsl_ir_node *node)
2226 {
2227 switch (node->type)
2228 {
2229 case HLSL_IR_CONSTANT:
2230 free_ir_constant(constant_from_node(node));
2231 break;
2232 case HLSL_IR_DEREF:
2233 free_ir_deref(deref_from_node(node));
2234 break;
2235 case HLSL_IR_SWIZZLE:
2236 free_ir_swizzle(swizzle_from_node(node));
2237 break;
2238 case HLSL_IR_CONSTRUCTOR:
2239 free_ir_constructor(constructor_from_node(node));
2240 break;
2241 case HLSL_IR_EXPR:
2242 free_ir_expr(expr_from_node(node));
2243 break;
2244 case HLSL_IR_ASSIGNMENT:
2245 free_ir_assignment(assignment_from_node(node));
2246 break;
2247 case HLSL_IR_IF:
2248 free_ir_if(if_from_node(node));
2249 break;
2250 case HLSL_IR_JUMP:
2251 free_ir_jump(jump_from_node(node));
2252 break;
2253 default:
2254 FIXME("Unsupported node type %s\n", debug_node_type(node->type));
2255 }
2256 }
2257
2258 static void free_function_decl(struct hlsl_ir_function_decl *decl)
2259 {
2260 d3dcompiler_free((void *)decl->semantic);
2261 d3dcompiler_free(decl->parameters);
2262 free_instr_list(decl->body);
2263 d3dcompiler_free(decl);
2264 }
2265
2266 static void free_function_decl_rb(struct wine_rb_entry *entry, void *context)
2267 {
2268 free_function_decl(WINE_RB_ENTRY_VALUE(entry, struct hlsl_ir_function_decl, entry));
2269 }
2270
2271 static void free_function(struct hlsl_ir_function *func)
2272 {
2273 wine_rb_destroy(&func->overloads, free_function_decl_rb, NULL);
2274 d3dcompiler_free((void *)func->name);
2275 d3dcompiler_free(func);
2276 }
2277
2278 void free_function_rb(struct wine_rb_entry *entry, void *context)
2279 {
2280 free_function(WINE_RB_ENTRY_VALUE(entry, struct hlsl_ir_function, entry));
2281 }
2282
2283 void add_function_decl(struct wine_rb_tree *funcs, char *name, struct hlsl_ir_function_decl *decl, BOOL intrinsic)
2284 {
2285 struct hlsl_ir_function *func;
2286 struct wine_rb_entry *func_entry, *old_entry;
2287
2288 func_entry = wine_rb_get(funcs, name);
2289 if (func_entry)
2290 {
2291 func = WINE_RB_ENTRY_VALUE(func_entry, struct hlsl_ir_function, entry);
2292 if (intrinsic != func->intrinsic)
2293 {
2294 if (intrinsic)
2295 {
2296 ERR("Redeclaring a user defined function as an intrinsic.\n");
2297 return;
2298 }
2299 TRACE("Function %s redeclared as a user defined function.\n", debugstr_a(name));
2300 func->intrinsic = intrinsic;
2301 wine_rb_destroy(&func->overloads, free_function_decl_rb, NULL);
2302 wine_rb_init(&func->overloads, compare_function_decl_rb);
2303 }
2304 decl->func = func;
2305 if ((old_entry = wine_rb_get(&func->overloads, decl->parameters)))
2306 {
2307 struct hlsl_ir_function_decl *old_decl =
2308 WINE_RB_ENTRY_VALUE(old_entry, struct hlsl_ir_function_decl, entry);
2309
2310 if (!decl->body)
2311 {
2312 free_function_decl(decl);
2313 d3dcompiler_free(name);
2314 return;
2315 }
2316 wine_rb_remove(&func->overloads, old_entry);
2317 free_function_decl(old_decl);
2318 }
2319 wine_rb_put(&func->overloads, decl->parameters, &decl->entry);
2320 d3dcompiler_free(name);
2321 return;
2322 }
2323 func = d3dcompiler_alloc(sizeof(*func));
2324 func->name = name;
2325 wine_rb_init(&func->overloads, compare_function_decl_rb);
2326 decl->func = func;
2327 wine_rb_put(&func->overloads, decl->parameters, &decl->entry);
2328 func->intrinsic = intrinsic;
2329 wine_rb_put(funcs, func->name, &func->entry);
2330 }
2331 #endif
Windows API implementation