?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
d3dcompiler: Cache vector types.
[wine.git] / dlls / d3dcompiler_43 / utils.c
blobc3dbd505f2dc38e9f258170ce0541ad89124f45a
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 & 3;
295 swizzle[1] = (arg >> 2) & 3;
296 swizzle[2] = (arg >> 4) & 3;
297 swizzle[3] = (arg >> 6) & 3;
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 if (type_class == HLSL_CLASS_MATRIX)
824 type->reg_size = is_row_major(type) ? dimy : dimx;
825 else
826 type->reg_size = 1;
827
828 list_add_tail(&hlsl_ctx.types, &type->entry);
829
830 return type;
831 }
832
833 struct hlsl_type *new_array_type(struct hlsl_type *basic_type, unsigned int array_size)
834 {
835 struct hlsl_type *type = new_hlsl_type(NULL, HLSL_CLASS_ARRAY, HLSL_TYPE_FLOAT, 1, 1);
836
837 if (!type)
838 return NULL;
839
840 type->modifiers = basic_type->modifiers;
841 type->e.array.elements_count = array_size;
842 type->e.array.type = basic_type;
843 type->reg_size = basic_type->reg_size * array_size;
844 type->dimx = basic_type->dimx;
845 type->dimy = basic_type->dimy;
846 return type;
847 }
848
849 struct hlsl_type *get_type(struct hlsl_scope *scope, const char *name, BOOL recursive)
850 {
851 struct wine_rb_entry *entry = wine_rb_get(&scope->types, name);
852 if (entry)
853 return WINE_RB_ENTRY_VALUE(entry, struct hlsl_type, scope_entry);
854
855 if (recursive && scope->upper)
856 return get_type(scope->upper, name, recursive);
857 return NULL;
858 }
859
860 BOOL find_function(const char *name)
861 {
862 return wine_rb_get(&hlsl_ctx.functions, name) != NULL;
863 }
864
865 unsigned int components_count_type(struct hlsl_type *type)
866 {
867 unsigned int count = 0;
868 struct hlsl_struct_field *field;
869
870 if (type->type <= HLSL_CLASS_LAST_NUMERIC)
871 {
872 return type->dimx * type->dimy;
873 }
874 if (type->type == HLSL_CLASS_ARRAY)
875 {
876 return components_count_type(type->e.array.type) * type->e.array.elements_count;
877 }
878 if (type->type != HLSL_CLASS_STRUCT)
879 {
880 ERR("Unexpected data type %s.\n", debug_hlsl_type(type));
881 return 0;
882 }
883
884 LIST_FOR_EACH_ENTRY(field, type->e.elements, struct hlsl_struct_field, entry)
885 {
886 count += components_count_type(field->type);
887 }
888 return count;
889 }
890
891 BOOL compare_hlsl_types(const struct hlsl_type *t1, const struct hlsl_type *t2)
892 {
893 if (t1 == t2)
894 return TRUE;
895
896 if (t1->type != t2->type)
897 return FALSE;
898 if (t1->base_type != t2->base_type)
899 return FALSE;
900 if (t1->base_type == HLSL_TYPE_SAMPLER && t1->sampler_dim != t2->sampler_dim)
901 return FALSE;
902 if ((t1->modifiers & HLSL_MODIFIERS_MAJORITY_MASK)
903 != (t2->modifiers & HLSL_MODIFIERS_MAJORITY_MASK))
904 return FALSE;
905 if (t1->dimx != t2->dimx)
906 return FALSE;
907 if (t1->dimy != t2->dimy)
908 return FALSE;
909 if (t1->type == HLSL_CLASS_STRUCT)
910 {
911 struct list *t1cur, *t2cur;
912 struct hlsl_struct_field *t1field, *t2field;
913
914 t1cur = list_head(t1->e.elements);
915 t2cur = list_head(t2->e.elements);
916 while (t1cur && t2cur)
917 {
918 t1field = LIST_ENTRY(t1cur, struct hlsl_struct_field, entry);
919 t2field = LIST_ENTRY(t2cur, struct hlsl_struct_field, entry);
920 if (!compare_hlsl_types(t1field->type, t2field->type))
921 return FALSE;
922 if (strcmp(t1field->name, t2field->name))
923 return FALSE;
924 t1cur = list_next(t1->e.elements, t1cur);
925 t2cur = list_next(t2->e.elements, t2cur);
926 }
927 if (t1cur != t2cur)
928 return FALSE;
929 }
930 if (t1->type == HLSL_CLASS_ARRAY)
931 return t1->e.array.elements_count == t2->e.array.elements_count
932 && compare_hlsl_types(t1->e.array.type, t2->e.array.type);
933
934 return TRUE;
935 }
936
937 struct hlsl_type *clone_hlsl_type(struct hlsl_type *old, unsigned int default_majority)
938 {
939 struct hlsl_type *type;
940 struct hlsl_struct_field *old_field, *field;
941
942 type = d3dcompiler_alloc(sizeof(*type));
943 if (!type)
944 {
945 ERR("Out of memory\n");
946 return NULL;
947 }
948 if (old->name)
949 {
950 type->name = d3dcompiler_strdup(old->name);
951 if (!type->name)
952 {
953 d3dcompiler_free(type);
954 return NULL;
955 }
956 }
957 type->type = old->type;
958 type->base_type = old->base_type;
959 type->dimx = old->dimx;
960 type->dimy = old->dimy;
961 type->modifiers = old->modifiers;
962 if (!(type->modifiers & HLSL_MODIFIERS_MAJORITY_MASK))
963 type->modifiers |= default_majority;
964 type->sampler_dim = old->sampler_dim;
965 switch (old->type)
966 {
967 case HLSL_CLASS_ARRAY:
968 type->e.array.type = clone_hlsl_type(old->e.array.type, default_majority);
969 type->e.array.elements_count = old->e.array.elements_count;
970 type->reg_size = type->e.array.elements_count * type->e.array.type->reg_size;
971 break;
972
973 case HLSL_CLASS_STRUCT:
974 {
975 unsigned int reg_size = 0;
976
977 type->e.elements = d3dcompiler_alloc(sizeof(*type->e.elements));
978 if (!type->e.elements)
979 {
980 d3dcompiler_free((void *)type->name);
981 d3dcompiler_free(type);
982 return NULL;
983 }
984 list_init(type->e.elements);
985 LIST_FOR_EACH_ENTRY(old_field, old->e.elements, struct hlsl_struct_field, entry)
986 {
987 field = d3dcompiler_alloc(sizeof(*field));
988 if (!field)
989 {
990 LIST_FOR_EACH_ENTRY_SAFE(field, old_field, type->e.elements, struct hlsl_struct_field, entry)
991 {
992 d3dcompiler_free((void *)field->semantic);
993 d3dcompiler_free((void *)field->name);
994 d3dcompiler_free(field);
995 }
996 d3dcompiler_free(type->e.elements);
997 d3dcompiler_free((void *)type->name);
998 d3dcompiler_free(type);
999 return NULL;
1000 }
1001 field->type = clone_hlsl_type(old_field->type, default_majority);
1002 field->name = d3dcompiler_strdup(old_field->name);
1003 if (old_field->semantic)
1004 field->semantic = d3dcompiler_strdup(old_field->semantic);
1005 field->modifiers = old_field->modifiers;
1006 field->reg_offset = reg_size;
1007 reg_size += field->type->reg_size;
1008 list_add_tail(type->e.elements, &field->entry);
1009 }
1010 type->reg_size = reg_size;
1011 break;
1012 }
1013
1014 case HLSL_CLASS_MATRIX:
1015 type->reg_size = is_row_major(type) ? type->dimy : type->dimx;
1016 break;
1017
1018 default:
1019 type->reg_size = 1;
1020 break;
1021 }
1022
1023 list_add_tail(&hlsl_ctx.types, &type->entry);
1024 return type;
1025 }
1026
1027 static BOOL convertible_data_type(struct hlsl_type *type)
1028 {
1029 return type->type != HLSL_CLASS_OBJECT;
1030 }
1031
1032 BOOL compatible_data_types(struct hlsl_type *t1, struct hlsl_type *t2)
1033 {
1034 if (!convertible_data_type(t1) || !convertible_data_type(t2))
1035 return FALSE;
1036
1037 if (t1->type <= HLSL_CLASS_LAST_NUMERIC)
1038 {
1039 /* Scalar vars can be cast to pretty much everything */
1040 if (t1->dimx == 1 && t1->dimy == 1)
1041 return TRUE;
1042
1043 if (t1->type == HLSL_CLASS_VECTOR && t2->type == HLSL_CLASS_VECTOR)
1044 return t1->dimx >= t2->dimx;
1045 }
1046
1047 /* The other way around is true too i.e. whatever to scalar */
1048 if (t2->type <= HLSL_CLASS_LAST_NUMERIC && t2->dimx == 1 && t2->dimy == 1)
1049 return TRUE;
1050
1051 if (t1->type == HLSL_CLASS_ARRAY)
1052 {
1053 if (compare_hlsl_types(t1->e.array.type, t2))
1054 /* e.g. float4[3] to float4 is allowed */
1055 return TRUE;
1056
1057 if (t2->type == HLSL_CLASS_ARRAY || t2->type == HLSL_CLASS_STRUCT)
1058 return components_count_type(t1) >= components_count_type(t2);
1059 else
1060 return components_count_type(t1) == components_count_type(t2);
1061 }
1062
1063 if (t1->type == HLSL_CLASS_STRUCT)
1064 return components_count_type(t1) >= components_count_type(t2);
1065
1066 if (t2->type == HLSL_CLASS_ARRAY || t2->type == HLSL_CLASS_STRUCT)
1067 return components_count_type(t1) == components_count_type(t2);
1068
1069 if (t1->type == HLSL_CLASS_MATRIX || t2->type == HLSL_CLASS_MATRIX)
1070 {
1071 if (t1->type == HLSL_CLASS_MATRIX && t2->type == HLSL_CLASS_MATRIX && t1->dimx >= t2->dimx && t1->dimy >= t2->dimy)
1072 return TRUE;
1073
1074 /* Matrix-vector conversion is apparently allowed if they have the same components count */
1075 if ((t1->type == HLSL_CLASS_VECTOR || t2->type == HLSL_CLASS_VECTOR)
1076 && components_count_type(t1) == components_count_type(t2))
1077 return TRUE;
1078 return FALSE;
1079 }
1080
1081 if (components_count_type(t1) >= components_count_type(t2))
1082 return TRUE;
1083 return FALSE;
1084 }
1085
1086 static BOOL implicit_compatible_data_types(struct hlsl_type *t1, struct hlsl_type *t2)
1087 {
1088 if (!convertible_data_type(t1) || !convertible_data_type(t2))
1089 return FALSE;
1090
1091 if (t1->type <= HLSL_CLASS_LAST_NUMERIC)
1092 {
1093 /* Scalar vars can be converted to any other numeric data type */
1094 if (t1->dimx == 1 && t1->dimy == 1 && t2->type <= HLSL_CLASS_LAST_NUMERIC)
1095 return TRUE;
1096 /* The other way around is true too */
1097 if (t2->dimx == 1 && t2->dimy == 1 && t2->type <= HLSL_CLASS_LAST_NUMERIC)
1098 return TRUE;
1099 }
1100
1101 if (t1->type == HLSL_CLASS_ARRAY && t2->type == HLSL_CLASS_ARRAY)
1102 {
1103 return components_count_type(t1) == components_count_type(t2);
1104 }
1105
1106 if ((t1->type == HLSL_CLASS_ARRAY && t2->type <= HLSL_CLASS_LAST_NUMERIC)
1107 || (t1->type <= HLSL_CLASS_LAST_NUMERIC && t2->type == HLSL_CLASS_ARRAY))
1108 {
1109 /* e.g. float4[3] to float4 is allowed */
1110 if (t1->type == HLSL_CLASS_ARRAY && compare_hlsl_types(t1->e.array.type, t2))
1111 return TRUE;
1112 if (components_count_type(t1) == components_count_type(t2))
1113 return TRUE;
1114 return FALSE;
1115 }
1116
1117 if (t1->type <= HLSL_CLASS_VECTOR && t2->type <= HLSL_CLASS_VECTOR)
1118 {
1119 if (t1->dimx >= t2->dimx)
1120 return TRUE;
1121 return FALSE;
1122 }
1123
1124 if (t1->type == HLSL_CLASS_MATRIX || t2->type == HLSL_CLASS_MATRIX)
1125 {
1126 if (t1->type == HLSL_CLASS_MATRIX && t2->type == HLSL_CLASS_MATRIX
1127 && t1->dimx >= t2->dimx && t1->dimy >= t2->dimy)
1128 return TRUE;
1129
1130 /* Matrix-vector conversion is apparently allowed if they have the same components count */
1131 if ((t1->type == HLSL_CLASS_VECTOR || t2->type == HLSL_CLASS_VECTOR)
1132 && components_count_type(t1) == components_count_type(t2))
1133 return TRUE;
1134 return FALSE;
1135 }
1136
1137 if (t1->type == HLSL_CLASS_STRUCT && t2->type == HLSL_CLASS_STRUCT)
1138 return compare_hlsl_types(t1, t2);
1139
1140 return FALSE;
1141 }
1142
1143 static BOOL expr_compatible_data_types(struct hlsl_type *t1, struct hlsl_type *t2)
1144 {
1145 if (t1->base_type > HLSL_TYPE_LAST_SCALAR || t2->base_type > HLSL_TYPE_LAST_SCALAR)
1146 return FALSE;
1147
1148 /* Scalar vars can be converted to pretty much everything */
1149 if ((t1->dimx == 1 && t1->dimy == 1) || (t2->dimx == 1 && t2->dimy == 1))
1150 return TRUE;
1151
1152 if (t1->type == HLSL_CLASS_VECTOR && t2->type == HLSL_CLASS_VECTOR)
1153 return TRUE;
1154
1155 if (t1->type == HLSL_CLASS_MATRIX || t2->type == HLSL_CLASS_MATRIX)
1156 {
1157 /* Matrix-vector conversion is apparently allowed if either they have the same components
1158 count or the matrix is nx1 or 1xn */
1159 if (t1->type == HLSL_CLASS_VECTOR || t2->type == HLSL_CLASS_VECTOR)
1160 {
1161 if (components_count_type(t1) == components_count_type(t2))
1162 return TRUE;
1163
1164 return (t1->type == HLSL_CLASS_MATRIX && (t1->dimx == 1 || t1->dimy == 1))
1165 || (t2->type == HLSL_CLASS_MATRIX && (t2->dimx == 1 || t2->dimy == 1));
1166 }
1167
1168 /* Both matrices */
1169 if ((t1->dimx >= t2->dimx && t1->dimy >= t2->dimy)
1170 || (t1->dimx <= t2->dimx && t1->dimy <= t2->dimy))
1171 return TRUE;
1172 }
1173
1174 return FALSE;
1175 }
1176
1177 static enum hlsl_base_type expr_common_base_type(enum hlsl_base_type t1, enum hlsl_base_type t2)
1178 {
1179 static const enum hlsl_base_type types[] =
1180 {
1181 HLSL_TYPE_BOOL,
1182 HLSL_TYPE_INT,
1183 HLSL_TYPE_UINT,
1184 HLSL_TYPE_HALF,
1185 HLSL_TYPE_FLOAT,
1186 HLSL_TYPE_DOUBLE,
1187 };
1188 int t1_idx = -1, t2_idx = -1, i;
1189
1190 for (i = 0; i < ARRAY_SIZE(types); ++i)
1191 {
1192 /* Always convert away from HLSL_TYPE_HALF */
1193 if (t1 == types[i])
1194 t1_idx = t1 == HLSL_TYPE_HALF ? i + 1 : i;
1195 if (t2 == types[i])
1196 t2_idx = t2 == HLSL_TYPE_HALF ? i + 1 : i;
1197
1198 if (t1_idx != -1 && t2_idx != -1)
1199 break;
1200 }
1201 if (t1_idx == -1 || t2_idx == -1)
1202 {
1203 FIXME("Unexpected base type.\n");
1204 return HLSL_TYPE_FLOAT;
1205 }
1206 return t1_idx >= t2_idx ? t1 : t2;
1207 }
1208
1209 static struct hlsl_type *expr_common_type(struct hlsl_type *t1, struct hlsl_type *t2,
1210 struct source_location *loc)
1211 {
1212 enum hlsl_type_class type;
1213 enum hlsl_base_type base;
1214 unsigned int dimx, dimy;
1215
1216 if (t1->type > HLSL_CLASS_LAST_NUMERIC || t2->type > HLSL_CLASS_LAST_NUMERIC)
1217 {
1218 hlsl_report_message(*loc, HLSL_LEVEL_ERROR, "non scalar/vector/matrix data type in expression");
1219 return NULL;
1220 }
1221
1222 if (compare_hlsl_types(t1, t2))
1223 return t1;
1224
1225 if (!expr_compatible_data_types(t1, t2))
1226 {
1227 hlsl_report_message(*loc, HLSL_LEVEL_ERROR, "expression data types are incompatible");
1228 return NULL;
1229 }
1230
1231 if (t1->base_type == t2->base_type)
1232 base = t1->base_type;
1233 else
1234 base = expr_common_base_type(t1->base_type, t2->base_type);
1235
1236 if (t1->dimx == 1 && t1->dimy == 1)
1237 {
1238 type = t2->type;
1239 dimx = t2->dimx;
1240 dimy = t2->dimy;
1241 }
1242 else if (t2->dimx == 1 && t2->dimy == 1)
1243 {
1244 type = t1->type;
1245 dimx = t1->dimx;
1246 dimy = t1->dimy;
1247 }
1248 else if (t1->type == HLSL_CLASS_MATRIX && t2->type == HLSL_CLASS_MATRIX)
1249 {
1250 type = HLSL_CLASS_MATRIX;
1251 dimx = min(t1->dimx, t2->dimx);
1252 dimy = min(t1->dimy, t2->dimy);
1253 }
1254 else
1255 {
1256 /* Two vectors or a vector and a matrix (matrix must be 1xn or nx1) */
1257 unsigned int max_dim_1, max_dim_2;
1258
1259 max_dim_1 = max(t1->dimx, t1->dimy);
1260 max_dim_2 = max(t2->dimx, t2->dimy);
1261 if (t1->dimx * t1->dimy == t2->dimx * t2->dimy)
1262 {
1263 type = HLSL_CLASS_VECTOR;
1264 dimx = max(t1->dimx, t2->dimx);
1265 dimy = 1;
1266 }
1267 else if (max_dim_1 <= max_dim_2)
1268 {
1269 type = t1->type;
1270 if (type == HLSL_CLASS_VECTOR)
1271 {
1272 dimx = max_dim_1;
1273 dimy = 1;
1274 }
1275 else
1276 {
1277 dimx = t1->dimx;
1278 dimy = t1->dimy;
1279 }
1280 }
1281 else
1282 {
1283 type = t2->type;
1284 if (type == HLSL_CLASS_VECTOR)
1285 {
1286 dimx = max_dim_2;
1287 dimy = 1;
1288 }
1289 else
1290 {
1291 dimx = t2->dimx;
1292 dimy = t2->dimy;
1293 }
1294 }
1295 }
1296
1297 if (type == HLSL_CLASS_SCALAR)
1298 return hlsl_ctx.builtin_types.scalar[base];
1299 if (type == HLSL_CLASS_VECTOR)
1300 return hlsl_ctx.builtin_types.vector[base][dimx - 1];
1301 return new_hlsl_type(NULL, type, base, dimx, dimy);
1302 }
1303
1304 struct hlsl_ir_node *implicit_conversion(struct hlsl_ir_node *node, struct hlsl_type *dst_type,
1305 struct source_location *loc)
1306 {
1307 struct hlsl_type *src_type = node->data_type;
1308 struct hlsl_ir_expr *cast;
1309
1310 if (compare_hlsl_types(src_type, dst_type))
1311 return node;
1312
1313 if (!implicit_compatible_data_types(src_type, dst_type))
1314 {
1315 hlsl_report_message(*loc, HLSL_LEVEL_ERROR, "can't implicitly convert %s to %s",
1316 debug_hlsl_type(src_type), debug_hlsl_type(dst_type));
1317 return NULL;
1318 }
1319
1320 if (dst_type->dimx * dst_type->dimy < src_type->dimx * src_type->dimy)
1321 hlsl_report_message(*loc, HLSL_LEVEL_WARNING, "implicit truncation of vector type");
1322
1323 TRACE("Implicit conversion from %s to %s.\n", debug_hlsl_type(src_type), debug_hlsl_type(dst_type));
1324
1325 if (!(cast = new_cast(node, dst_type, loc)))
1326 return NULL;
1327 list_add_after(&node->entry, &cast->node.entry);
1328 return &cast->node;
1329 }
1330
1331 struct hlsl_ir_expr *new_expr(enum hlsl_ir_expr_op op, struct hlsl_ir_node **operands,
1332 struct source_location *loc)
1333 {
1334 struct hlsl_ir_expr *expr;
1335 struct hlsl_type *type;
1336 unsigned int i;
1337
1338 type = operands[0]->data_type;
1339 for (i = 1; i <= 2; ++i)
1340 {
1341 if (!operands[i])
1342 break;
1343 type = expr_common_type(type, operands[i]->data_type, loc);
1344 if (!type)
1345 return NULL;
1346 }
1347 for (i = 0; i <= 2; ++i)
1348 {
1349 struct hlsl_ir_expr *cast;
1350
1351 if (!operands[i])
1352 break;
1353 if (compare_hlsl_types(operands[i]->data_type, type))
1354 continue;
1355 TRACE("Implicitly converting %s into %s in an expression\n", debug_hlsl_type(operands[i]->data_type), debug_hlsl_type(type));
1356 if (operands[i]->data_type->dimx * operands[i]->data_type->dimy != 1
1357 && operands[i]->data_type->dimx * operands[i]->data_type->dimy != type->dimx * type->dimy)
1358 {
1359 hlsl_report_message(operands[i]->loc, HLSL_LEVEL_WARNING, "implicit truncation of vector/matrix type");
1360 }
1361
1362 if (!(cast = new_cast(operands[i], type, &operands[i]->loc)))
1363 return NULL;
1364 list_add_after(&operands[i]->entry, &cast->node.entry);
1365 operands[i] = &cast->node;
1366 }
1367
1368 if (!(expr = d3dcompiler_alloc(sizeof(*expr))))
1369 return NULL;
1370 init_node(&expr->node, HLSL_IR_EXPR, type, *loc);
1371 expr->op = op;
1372 expr->operands[0] = operands[0];
1373 expr->operands[1] = operands[1];
1374 expr->operands[2] = operands[2];
1375
1376 return expr;
1377 }
1378
1379 struct hlsl_ir_expr *new_cast(struct hlsl_ir_node *node, struct hlsl_type *type,
1380 struct source_location *loc)
1381 {
1382 struct hlsl_ir_node *cast;
1383
1384 cast = new_unary_expr(HLSL_IR_UNOP_CAST, node, *loc);
1385 if (cast)
1386 cast->data_type = type;
1387 return expr_from_node(cast);
1388 }
1389
1390 static enum hlsl_ir_expr_op op_from_assignment(enum parse_assign_op op)
1391 {
1392 static const enum hlsl_ir_expr_op ops[] =
1393 {
1394 0,
1395 HLSL_IR_BINOP_ADD,
1396 HLSL_IR_BINOP_SUB,
1397 HLSL_IR_BINOP_MUL,
1398 HLSL_IR_BINOP_DIV,
1399 HLSL_IR_BINOP_MOD,
1400 HLSL_IR_BINOP_LSHIFT,
1401 HLSL_IR_BINOP_RSHIFT,
1402 HLSL_IR_BINOP_BIT_AND,
1403 HLSL_IR_BINOP_BIT_OR,
1404 HLSL_IR_BINOP_BIT_XOR,
1405 };
1406
1407 return ops[op];
1408 }
1409
1410 static BOOL invert_swizzle(unsigned int *swizzle, unsigned int *writemask, unsigned int *ret_width)
1411 {
1412 unsigned int i, j, bit = 0, inverted = 0, width, new_writemask = 0, new_swizzle = 0;
1413
1414 /* Apply the writemask to the swizzle to get a new writemask and swizzle. */
1415 for (i = 0; i < 4; ++i)
1416 {
1417 if (*writemask & (1 << i))
1418 {
1419 unsigned int s = (*swizzle >> (i * 2)) & 3;
1420 new_swizzle |= s << (bit++ * 2);
1421 if (new_writemask & (1 << s))
1422 return FALSE;
1423 new_writemask |= 1 << s;
1424 }
1425 }
1426 width = bit;
1427
1428 /* Invert the swizzle. */
1429 bit = 0;
1430 for (i = 0; i < 4; ++i)
1431 {
1432 for (j = 0; j < width; ++j)
1433 {
1434 unsigned int s = (new_swizzle >> (j * 2)) & 3;
1435 if (s == i)
1436 inverted |= j << (bit++ * 2);
1437 }
1438 }
1439
1440 *swizzle = inverted;
1441 *writemask = new_writemask;
1442 *ret_width = width;
1443 return TRUE;
1444 }
1445
1446 struct hlsl_ir_node *make_assignment(struct hlsl_ir_node *lhs, enum parse_assign_op assign_op,
1447 struct hlsl_ir_node *rhs)
1448 {
1449 struct hlsl_ir_assignment *assign = d3dcompiler_alloc(sizeof(*assign));
1450 struct hlsl_type *lhs_type;
1451 DWORD writemask = 0;
1452
1453 lhs_type = lhs->data_type;
1454 if (lhs_type->type <= HLSL_CLASS_LAST_NUMERIC)
1455 {
1456 writemask = (1 << lhs_type->dimx) - 1;
1457
1458 if (!(rhs = implicit_conversion(rhs, lhs_type, &rhs->loc)))
1459 return NULL;
1460 }
1461
1462 if (!assign)
1463 {
1464 ERR("Out of memory\n");
1465 return NULL;
1466 }
1467
1468 while (lhs->type != HLSL_IR_LOAD)
1469 {
1470 struct hlsl_ir_node *lhs_inner;
1471
1472 if (lhs->type == HLSL_IR_EXPR && expr_from_node(lhs)->op == HLSL_IR_UNOP_CAST)
1473 {
1474 FIXME("Cast on the lhs.\n");
1475 d3dcompiler_free(assign);
1476 return NULL;
1477 }
1478 else if (lhs->type == HLSL_IR_SWIZZLE)
1479 {
1480 struct hlsl_ir_swizzle *swizzle = swizzle_from_node(lhs);
1481 const struct hlsl_type *swizzle_type = swizzle->node.data_type;
1482 unsigned int width;
1483
1484 if (lhs->data_type->type == HLSL_CLASS_MATRIX)
1485 FIXME("Assignments with writemasks and matrices on lhs are not supported yet.\n");
1486
1487 lhs_inner = swizzle->val;
1488 list_remove(&lhs->entry);
1489
1490 list_add_after(&rhs->entry, &lhs->entry);
1491 swizzle->val = rhs;
1492 if (!invert_swizzle(&swizzle->swizzle, &writemask, &width))
1493 {
1494 hlsl_report_message(lhs->loc, HLSL_LEVEL_ERROR, "invalid writemask");
1495 d3dcompiler_free(assign);
1496 return NULL;
1497 }
1498 assert(swizzle_type->type == HLSL_CLASS_VECTOR);
1499 if (swizzle_type->dimx != width)
1500 swizzle->node.data_type = hlsl_ctx.builtin_types.vector[swizzle_type->base_type][width - 1];
1501 rhs = &swizzle->node;
1502 }
1503 else
1504 {
1505 hlsl_report_message(lhs->loc, HLSL_LEVEL_ERROR, "invalid lvalue");
1506 d3dcompiler_free(assign);
1507 return NULL;
1508 }
1509
1510 lhs = lhs_inner;
1511 }
1512
1513 init_node(&assign->node, HLSL_IR_ASSIGNMENT, lhs_type, lhs->loc);
1514 assign->writemask = writemask;
1515 assign->lhs = load_from_node(lhs)->src;
1516 if (assign_op != ASSIGN_OP_ASSIGN)
1517 {
1518 enum hlsl_ir_expr_op op = op_from_assignment(assign_op);
1519 struct hlsl_ir_node *expr;
1520
1521 TRACE("Adding an expression for the compound assignment.\n");
1522 expr = new_binary_expr(op, lhs, rhs, lhs->loc);
1523 list_add_after(&rhs->entry, &expr->entry);
1524 rhs = expr;
1525 }
1526 assign->rhs = rhs;
1527
1528 return &assign->node;
1529 }
1530
1531 static int compare_hlsl_types_rb(const void *key, const struct wine_rb_entry *entry)
1532 {
1533 const char *name = key;
1534 const struct hlsl_type *type = WINE_RB_ENTRY_VALUE(entry, const struct hlsl_type, scope_entry);
1535
1536 if (name == type->name)
1537 return 0;
1538
1539 if (!name || !type->name)
1540 {
1541 ERR("hlsl_type without a name in a scope?\n");
1542 return -1;
1543 }
1544 return strcmp(name, type->name);
1545 }
1546
1547 void push_scope(struct hlsl_parse_ctx *ctx)
1548 {
1549 struct hlsl_scope *new_scope = d3dcompiler_alloc(sizeof(*new_scope));
1550
1551 if (!new_scope)
1552 {
1553 ERR("Out of memory!\n");
1554 return;
1555 }
1556 TRACE("Pushing a new scope\n");
1557 list_init(&new_scope->vars);
1558 wine_rb_init(&new_scope->types, compare_hlsl_types_rb);
1559 new_scope->upper = ctx->cur_scope;
1560 ctx->cur_scope = new_scope;
1561 list_add_tail(&ctx->scopes, &new_scope->entry);
1562 }
1563
1564 BOOL pop_scope(struct hlsl_parse_ctx *ctx)
1565 {
1566 struct hlsl_scope *prev_scope = ctx->cur_scope->upper;
1567 if (!prev_scope)
1568 return FALSE;
1569
1570 TRACE("Popping current scope\n");
1571 ctx->cur_scope = prev_scope;
1572 return TRUE;
1573 }
1574
1575 static int compare_param_hlsl_types(const struct hlsl_type *t1, const struct hlsl_type *t2)
1576 {
1577 if (t1->type != t2->type)
1578 {
1579 if (!((t1->type == HLSL_CLASS_SCALAR && t2->type == HLSL_CLASS_VECTOR)
1580 || (t1->type == HLSL_CLASS_VECTOR && t2->type == HLSL_CLASS_SCALAR)))
1581 return t1->type - t2->type;
1582 }
1583 if (t1->base_type != t2->base_type)
1584 return t1->base_type - t2->base_type;
1585 if (t1->base_type == HLSL_TYPE_SAMPLER && t1->sampler_dim != t2->sampler_dim)
1586 return t1->sampler_dim - t2->sampler_dim;
1587 if (t1->dimx != t2->dimx)
1588 return t1->dimx - t2->dimx;
1589 if (t1->dimy != t2->dimy)
1590 return t1->dimx - t2->dimx;
1591 if (t1->type == HLSL_CLASS_STRUCT)
1592 {
1593 struct list *t1cur, *t2cur;
1594 struct hlsl_struct_field *t1field, *t2field;
1595 int r;
1596
1597 t1cur = list_head(t1->e.elements);
1598 t2cur = list_head(t2->e.elements);
1599 while (t1cur && t2cur)
1600 {
1601 t1field = LIST_ENTRY(t1cur, struct hlsl_struct_field, entry);
1602 t2field = LIST_ENTRY(t2cur, struct hlsl_struct_field, entry);
1603 if ((r = compare_param_hlsl_types(t1field->type, t2field->type)))
1604 return r;
1605 if ((r = strcmp(t1field->name, t2field->name)))
1606 return r;
1607 t1cur = list_next(t1->e.elements, t1cur);
1608 t2cur = list_next(t2->e.elements, t2cur);
1609 }
1610 if (t1cur != t2cur)
1611 return t1cur ? 1 : -1;
1612 return 0;
1613 }
1614 if (t1->type == HLSL_CLASS_ARRAY)
1615 {
1616 if (t1->e.array.elements_count != t2->e.array.elements_count)
1617 return t1->e.array.elements_count - t2->e.array.elements_count;
1618 return compare_param_hlsl_types(t1->e.array.type, t2->e.array.type);
1619 }
1620
1621 return 0;
1622 }
1623
1624 static int compare_function_decl_rb(const void *key, const struct wine_rb_entry *entry)
1625 {
1626 const struct list *params = key;
1627 const struct hlsl_ir_function_decl *decl = WINE_RB_ENTRY_VALUE(entry, const struct hlsl_ir_function_decl, entry);
1628 int params_count = params ? list_count(params) : 0;
1629 int decl_params_count = decl->parameters ? list_count(decl->parameters) : 0;
1630 int r;
1631 struct list *p1cur, *p2cur;
1632
1633 if (params_count != decl_params_count)
1634 return params_count - decl_params_count;
1635
1636 p1cur = params ? list_head(params) : NULL;
1637 p2cur = decl->parameters ? list_head(decl->parameters) : NULL;
1638 while (p1cur && p2cur)
1639 {
1640 struct hlsl_ir_var *p1, *p2;
1641 p1 = LIST_ENTRY(p1cur, struct hlsl_ir_var, param_entry);
1642 p2 = LIST_ENTRY(p2cur, struct hlsl_ir_var, param_entry);
1643 if ((r = compare_param_hlsl_types(p1->data_type, p2->data_type)))
1644 return r;
1645 p1cur = list_next(params, p1cur);
1646 p2cur = list_next(decl->parameters, p2cur);
1647 }
1648 return 0;
1649 }
1650
1651 static int compare_function_rb(const void *key, const struct wine_rb_entry *entry)
1652 {
1653 const char *name = key;
1654 const struct hlsl_ir_function *func = WINE_RB_ENTRY_VALUE(entry, const struct hlsl_ir_function,entry);
1655
1656 return strcmp(name, func->name);
1657 }
1658
1659 void init_functions_tree(struct wine_rb_tree *funcs)
1660 {
1661 wine_rb_init(&hlsl_ctx.functions, compare_function_rb);
1662 }
1663
1664 const char *debug_base_type(const struct hlsl_type *type)
1665 {
1666 const char *name = "(unknown)";
1667
1668 switch (type->base_type)
1669 {
1670 case HLSL_TYPE_FLOAT: name = "float"; break;
1671 case HLSL_TYPE_HALF: name = "half"; break;
1672 case HLSL_TYPE_DOUBLE: name = "double"; break;
1673 case HLSL_TYPE_INT: name = "int"; break;
1674 case HLSL_TYPE_UINT: name = "uint"; break;
1675 case HLSL_TYPE_BOOL: name = "bool"; break;
1676 case HLSL_TYPE_SAMPLER:
1677 switch (type->sampler_dim)
1678 {
1679 case HLSL_SAMPLER_DIM_GENERIC: name = "sampler"; break;
1680 case HLSL_SAMPLER_DIM_1D: name = "sampler1D"; break;
1681 case HLSL_SAMPLER_DIM_2D: name = "sampler2D"; break;
1682 case HLSL_SAMPLER_DIM_3D: name = "sampler3D"; break;
1683 case HLSL_SAMPLER_DIM_CUBE: name = "samplerCUBE"; break;
1684 }
1685 break;
1686 default:
1687 FIXME("Unhandled case %u\n", type->base_type);
1688 }
1689 return name;
1690 }
1691
1692 const char *debug_hlsl_type(const struct hlsl_type *type)
1693 {
1694 const char *name;
1695
1696 if (type->name)
1697 return debugstr_a(type->name);
1698
1699 if (type->type == HLSL_CLASS_STRUCT)
1700 return "<anonymous struct>";
1701
1702 if (type->type == HLSL_CLASS_ARRAY)
1703 {
1704 name = debug_base_type(type->e.array.type);
1705 return wine_dbg_sprintf("%s[%u]", name, type->e.array.elements_count);
1706 }
1707
1708 name = debug_base_type(type);
1709
1710 if (type->type == HLSL_CLASS_SCALAR)
1711 return wine_dbg_sprintf("%s", name);
1712 if (type->type == HLSL_CLASS_VECTOR)
1713 return wine_dbg_sprintf("%s%u", name, type->dimx);
1714 if (type->type == HLSL_CLASS_MATRIX)
1715 return wine_dbg_sprintf("%s%ux%u", name, type->dimx, type->dimy);
1716 return "unexpected_type";
1717 }
1718
1719 const char *debug_modifiers(DWORD modifiers)
1720 {
1721 char string[110];
1722
1723 string[0] = 0;
1724 if (modifiers & HLSL_STORAGE_EXTERN)
1725 strcat(string, " extern"); /* 7 */
1726 if (modifiers & HLSL_STORAGE_NOINTERPOLATION)
1727 strcat(string, " nointerpolation"); /* 16 */
1728 if (modifiers & HLSL_MODIFIER_PRECISE)
1729 strcat(string, " precise"); /* 8 */
1730 if (modifiers & HLSL_STORAGE_SHARED)
1731 strcat(string, " shared"); /* 7 */
1732 if (modifiers & HLSL_STORAGE_GROUPSHARED)
1733 strcat(string, " groupshared"); /* 12 */
1734 if (modifiers & HLSL_STORAGE_STATIC)
1735 strcat(string, " static"); /* 7 */
1736 if (modifiers & HLSL_STORAGE_UNIFORM)
1737 strcat(string, " uniform"); /* 8 */
1738 if (modifiers & HLSL_STORAGE_VOLATILE)
1739 strcat(string, " volatile"); /* 9 */
1740 if (modifiers & HLSL_MODIFIER_CONST)
1741 strcat(string, " const"); /* 6 */
1742 if (modifiers & HLSL_MODIFIER_ROW_MAJOR)
1743 strcat(string, " row_major"); /* 10 */
1744 if (modifiers & HLSL_MODIFIER_COLUMN_MAJOR)
1745 strcat(string, " column_major"); /* 13 */
1746 if ((modifiers & (HLSL_STORAGE_IN | HLSL_STORAGE_OUT)) == (HLSL_STORAGE_IN | HLSL_STORAGE_OUT))
1747 strcat(string, " inout"); /* 6 */
1748 else if (modifiers & HLSL_STORAGE_IN)
1749 strcat(string, " in"); /* 3 */
1750 else if (modifiers & HLSL_STORAGE_OUT)
1751 strcat(string, " out"); /* 4 */
1752
1753 return wine_dbg_sprintf("%s", string[0] ? string + 1 : "");
1754 }
1755
1756 const char *debug_node_type(enum hlsl_ir_node_type type)
1757 {
1758 static const char * const names[] =
1759 {
1760 "HLSL_IR_ASSIGNMENT",
1761 "HLSL_IR_CONSTANT",
1762 "HLSL_IR_CONSTRUCTOR",
1763 "HLSL_IR_EXPR",
1764 "HLSL_IR_IF",
1765 "HLSL_IR_LOAD",
1766 "HLSL_IR_LOOP",
1767 "HLSL_IR_JUMP",
1768 "HLSL_IR_SWIZZLE",
1769 };
1770
1771 if (type >= ARRAY_SIZE(names))
1772 return "Unexpected node type";
1773 return names[type];
1774 }
1775
1776 static void debug_dump_instr(const struct hlsl_ir_node *instr);
1777
1778 static void debug_dump_instr_list(const struct list *list)
1779 {
1780 struct hlsl_ir_node *instr;
1781
1782 LIST_FOR_EACH_ENTRY(instr, list, struct hlsl_ir_node, entry)
1783 {
1784 debug_dump_instr(instr);
1785 wine_dbg_printf("\n");
1786 }
1787 }
1788
1789 static void debug_dump_src(const struct hlsl_ir_node *node)
1790 {
1791 if (node->index)
1792 wine_dbg_printf("@%u", node->index);
1793 else
1794 wine_dbg_printf("%p", node);
1795 }
1796
1797 static void debug_dump_ir_var(const struct hlsl_ir_var *var)
1798 {
1799 if (var->modifiers)
1800 wine_dbg_printf("%s ", debug_modifiers(var->modifiers));
1801 wine_dbg_printf("%s %s", debug_hlsl_type(var->data_type), var->name);
1802 if (var->semantic)
1803 wine_dbg_printf(" : %s", debugstr_a(var->semantic));
1804 }
1805
1806 static void debug_dump_deref(const struct hlsl_deref *deref)
1807 {
1808 wine_dbg_printf("deref(");
1809 debug_dump_ir_var(deref->var);
1810 wine_dbg_printf(")");
1811 if (deref->offset)
1812 {
1813 wine_dbg_printf("[");
1814 debug_dump_src(deref->offset);
1815 wine_dbg_printf("]");
1816 }
1817 }
1818
1819 static void debug_dump_ir_constant(const struct hlsl_ir_constant *constant)
1820 {
1821 struct hlsl_type *type = constant->node.data_type;
1822 unsigned int x, y;
1823
1824 if (type->dimy != 1)
1825 wine_dbg_printf("{");
1826 for (y = 0; y < type->dimy; ++y)
1827 {
1828 if (type->dimx != 1)
1829 wine_dbg_printf("{");
1830 for (x = 0; x < type->dimx; ++x)
1831 {
1832 switch (type->base_type)
1833 {
1834 case HLSL_TYPE_FLOAT:
1835 wine_dbg_printf("%g ", (double)constant->v.value.f[y * type->dimx + x]);
1836 break;
1837 case HLSL_TYPE_DOUBLE:
1838 wine_dbg_printf("%g ", constant->v.value.d[y * type->dimx + x]);
1839 break;
1840 case HLSL_TYPE_INT:
1841 wine_dbg_printf("%d ", constant->v.value.i[y * type->dimx + x]);
1842 break;
1843 case HLSL_TYPE_UINT:
1844 wine_dbg_printf("%u ", constant->v.value.u[y * type->dimx + x]);
1845 break;
1846 case HLSL_TYPE_BOOL:
1847 wine_dbg_printf("%s ", constant->v.value.b[y * type->dimx + x] == FALSE ? "false" : "true");
1848 break;
1849 default:
1850 wine_dbg_printf("Constants of type %s not supported\n", debug_base_type(type));
1851 }
1852 }
1853 if (type->dimx != 1)
1854 wine_dbg_printf("}");
1855 }
1856 if (type->dimy != 1)
1857 wine_dbg_printf("}");
1858 }
1859
1860 static const char *debug_expr_op(const struct hlsl_ir_expr *expr)
1861 {
1862 static const char * const op_names[] =
1863 {
1864 "~",
1865 "!",
1866 "-",
1867 "abs",
1868 "sign",
1869 "rcp",
1870 "rsq",
1871 "sqrt",
1872 "nrm",
1873 "exp2",
1874 "log2",
1875
1876 "cast",
1877
1878 "fract",
1879
1880 "sin",
1881 "cos",
1882 "sin_reduced",
1883 "cos_reduced",
1884
1885 "dsx",
1886 "dsy",
1887
1888 "sat",
1889
1890 "pre++",
1891 "pre--",
1892 "post++",
1893 "post--",
1894
1895 "+",
1896 "-",
1897 "*",
1898 "/",
1899
1900 "%",
1901
1902 "<",
1903 ">",
1904 "<=",
1905 ">=",
1906 "==",
1907 "!=",
1908
1909 "&&",
1910 "||",
1911
1912 "<<",
1913 ">>",
1914 "&",
1915 "|",
1916 "^",
1917
1918 "dot",
1919 "crs",
1920 "min",
1921 "max",
1922
1923 "pow",
1924
1925 "lerp",
1926
1927 ",",
1928 };
1929
1930 if (expr->op == HLSL_IR_UNOP_CAST)
1931 return debug_hlsl_type(expr->node.data_type);
1932
1933 return op_names[expr->op];
1934 }
1935
1936 /* Dumps the expression in a prefix "operator (operands)" form */
1937 static void debug_dump_ir_expr(const struct hlsl_ir_expr *expr)
1938 {
1939 unsigned int i;
1940
1941 wine_dbg_printf("%s (", debug_expr_op(expr));
1942 for (i = 0; i < 3 && expr->operands[i]; ++i)
1943 {
1944 debug_dump_src(expr->operands[i]);
1945 wine_dbg_printf(" ");
1946 }
1947 wine_dbg_printf(")");
1948 }
1949
1950 static void debug_dump_ir_constructor(const struct hlsl_ir_constructor *constructor)
1951 {
1952 unsigned int i;
1953
1954 wine_dbg_printf("%s (", debug_hlsl_type(constructor->node.data_type));
1955 for (i = 0; i < constructor->args_count; ++i)
1956 {
1957 debug_dump_src(constructor->args[i]);
1958 wine_dbg_printf(" ");
1959 }
1960 wine_dbg_printf(")");
1961 }
1962
1963 static const char *debug_writemask(DWORD writemask)
1964 {
1965 static const char components[] = {'x', 'y', 'z', 'w'};
1966 char string[5];
1967 unsigned int i = 0, pos = 0;
1968
1969 assert(!(writemask & ~BWRITERSP_WRITEMASK_ALL));
1970
1971 while (writemask)
1972 {
1973 if (writemask & 1)
1974 string[pos++] = components[i];
1975 writemask >>= 1;
1976 i++;
1977 }
1978 string[pos] = '\0';
1979 return wine_dbg_sprintf(".%s", string);
1980 }
1981
1982 static void debug_dump_ir_assignment(const struct hlsl_ir_assignment *assign)
1983 {
1984 wine_dbg_printf("= (");
1985 debug_dump_deref(&assign->lhs);
1986 if (assign->writemask != BWRITERSP_WRITEMASK_ALL)
1987 wine_dbg_printf("%s", debug_writemask(assign->writemask));
1988 wine_dbg_printf(" ");
1989 debug_dump_src(assign->rhs);
1990 wine_dbg_printf(")");
1991 }
1992
1993 static void debug_dump_ir_swizzle(const struct hlsl_ir_swizzle *swizzle)
1994 {
1995 unsigned int i;
1996
1997 debug_dump_src(swizzle->val);
1998 wine_dbg_printf(".");
1999 if (swizzle->val->data_type->dimy > 1)
2000 {
2001 for (i = 0; i < swizzle->node.data_type->dimx; ++i)
2002 wine_dbg_printf("_m%u%u", (swizzle->swizzle >> i * 8) & 0xf, (swizzle->swizzle >> (i * 8 + 4)) & 0xf);
2003 }
2004 else
2005 {
2006 static const char c[] = {'x', 'y', 'z', 'w'};
2007
2008 for (i = 0; i < swizzle->node.data_type->dimx; ++i)
2009 wine_dbg_printf("%c", c[(swizzle->swizzle >> i * 2) & 0x3]);
2010 }
2011 }
2012
2013 static void debug_dump_ir_jump(const struct hlsl_ir_jump *jump)
2014 {
2015 switch (jump->type)
2016 {
2017 case HLSL_IR_JUMP_BREAK:
2018 wine_dbg_printf("break");
2019 break;
2020 case HLSL_IR_JUMP_CONTINUE:
2021 wine_dbg_printf("continue");
2022 break;
2023 case HLSL_IR_JUMP_DISCARD:
2024 wine_dbg_printf("discard");
2025 break;
2026 case HLSL_IR_JUMP_RETURN:
2027 wine_dbg_printf("return");
2028 break;
2029 }
2030 }
2031
2032 static void debug_dump_ir_if(const struct hlsl_ir_if *if_node)
2033 {
2034 wine_dbg_printf("if (");
2035 debug_dump_src(if_node->condition);
2036 wine_dbg_printf(")\n{\n");
2037 debug_dump_instr_list(if_node->then_instrs);
2038 wine_dbg_printf("}\n");
2039 if (if_node->else_instrs)
2040 {
2041 wine_dbg_printf("else\n{\n");
2042 debug_dump_instr_list(if_node->else_instrs);
2043 wine_dbg_printf("}\n");
2044 }
2045 }
2046
2047 static void debug_dump_ir_loop(const struct hlsl_ir_loop *loop)
2048 {
2049 wine_dbg_printf("for (;;)\n{\n");
2050 debug_dump_instr_list(loop->body);
2051 wine_dbg_printf("}\n");
2052 }
2053
2054 static void debug_dump_instr(const struct hlsl_ir_node *instr)
2055 {
2056 if (instr->index)
2057 wine_dbg_printf("%4u: ", instr->index);
2058 else
2059 wine_dbg_printf("%p: ", instr);
2060 switch (instr->type)
2061 {
2062 case HLSL_IR_EXPR:
2063 debug_dump_ir_expr(expr_from_node(instr));
2064 break;
2065 case HLSL_IR_LOAD:
2066 debug_dump_deref(&load_from_node(instr)->src);
2067 break;
2068 case HLSL_IR_CONSTANT:
2069 debug_dump_ir_constant(constant_from_node(instr));
2070 break;
2071 case HLSL_IR_ASSIGNMENT:
2072 debug_dump_ir_assignment(assignment_from_node(instr));
2073 break;
2074 case HLSL_IR_SWIZZLE:
2075 debug_dump_ir_swizzle(swizzle_from_node(instr));
2076 break;
2077 case HLSL_IR_CONSTRUCTOR:
2078 debug_dump_ir_constructor(constructor_from_node(instr));
2079 break;
2080 case HLSL_IR_JUMP:
2081 debug_dump_ir_jump(jump_from_node(instr));
2082 break;
2083 case HLSL_IR_IF:
2084 debug_dump_ir_if(if_from_node(instr));
2085 break;
2086 case HLSL_IR_LOOP:
2087 debug_dump_ir_loop(loop_from_node(instr));
2088 break;
2089 default:
2090 wine_dbg_printf("<No dump function for %s>", debug_node_type(instr->type));
2091 }
2092 }
2093
2094 void debug_dump_ir_function_decl(const struct hlsl_ir_function_decl *func)
2095 {
2096 struct hlsl_ir_var *param;
2097
2098 TRACE("Dumping function %s.\n", debugstr_a(func->func->name));
2099 TRACE("Function parameters:\n");
2100 LIST_FOR_EACH_ENTRY(param, func->parameters, struct hlsl_ir_var, param_entry)
2101 {
2102 debug_dump_ir_var(param);
2103 wine_dbg_printf("\n");
2104 }
2105 if (func->semantic)
2106 TRACE("Function semantic: %s\n", debugstr_a(func->semantic));
2107 if (func->body)
2108 {
2109 debug_dump_instr_list(func->body);
2110 }
2111 }
2112
2113 void free_hlsl_type(struct hlsl_type *type)
2114 {
2115 struct hlsl_struct_field *field, *next_field;
2116
2117 d3dcompiler_free((void *)type->name);
2118 if (type->type == HLSL_CLASS_STRUCT)
2119 {
2120 LIST_FOR_EACH_ENTRY_SAFE(field, next_field, type->e.elements, struct hlsl_struct_field, entry)
2121 {
2122 d3dcompiler_free((void *)field->name);
2123 d3dcompiler_free((void *)field->semantic);
2124 d3dcompiler_free(field);
2125 }
2126 }
2127 d3dcompiler_free(type);
2128 }
2129
2130 void free_instr_list(struct list *list)
2131 {
2132 struct hlsl_ir_node *node, *next_node;
2133
2134 if (!list)
2135 return;
2136 LIST_FOR_EACH_ENTRY_SAFE(node, next_node, list, struct hlsl_ir_node, entry)
2137 free_instr(node);
2138 d3dcompiler_free(list);
2139 }
2140
2141 static void free_ir_constant(struct hlsl_ir_constant *constant)
2142 {
2143 struct hlsl_type *type = constant->node.data_type;
2144 unsigned int i;
2145 struct hlsl_ir_constant *field, *next_field;
2146
2147 switch (type->type)
2148 {
2149 case HLSL_CLASS_ARRAY:
2150 for (i = 0; i < type->e.array.elements_count; ++i)
2151 free_ir_constant(&constant->v.array_elements[i]);
2152 d3dcompiler_free(constant->v.array_elements);
2153 break;
2154 case HLSL_CLASS_STRUCT:
2155 LIST_FOR_EACH_ENTRY_SAFE(field, next_field, constant->v.struct_elements, struct hlsl_ir_constant, node.entry)
2156 free_ir_constant(field);
2157 break;
2158 default:
2159 break;
2160 }
2161 d3dcompiler_free(constant);
2162 }
2163
2164 static void free_ir_load(struct hlsl_ir_load *load)
2165 {
2166 d3dcompiler_free(load);
2167 }
2168
2169 static void free_ir_swizzle(struct hlsl_ir_swizzle *swizzle)
2170 {
2171 d3dcompiler_free(swizzle);
2172 }
2173
2174 static void free_ir_constructor(struct hlsl_ir_constructor *constructor)
2175 {
2176 d3dcompiler_free(constructor);
2177 }
2178
2179 static void free_ir_expr(struct hlsl_ir_expr *expr)
2180 {
2181 d3dcompiler_free(expr);
2182 }
2183
2184 static void free_ir_assignment(struct hlsl_ir_assignment *assignment)
2185 {
2186 d3dcompiler_free(assignment);
2187 }
2188
2189 static void free_ir_if(struct hlsl_ir_if *if_node)
2190 {
2191 free_instr_list(if_node->then_instrs);
2192 free_instr_list(if_node->else_instrs);
2193 d3dcompiler_free(if_node);
2194 }
2195
2196 static void free_ir_loop(struct hlsl_ir_loop *loop)
2197 {
2198 free_instr_list(loop->body);
2199 d3dcompiler_free(loop);
2200 }
2201
2202 static void free_ir_jump(struct hlsl_ir_jump *jump)
2203 {
2204 d3dcompiler_free(jump);
2205 }
2206
2207 void free_instr(struct hlsl_ir_node *node)
2208 {
2209 switch (node->type)
2210 {
2211 case HLSL_IR_CONSTANT:
2212 free_ir_constant(constant_from_node(node));
2213 break;
2214 case HLSL_IR_LOAD:
2215 free_ir_load(load_from_node(node));
2216 break;
2217 case HLSL_IR_SWIZZLE:
2218 free_ir_swizzle(swizzle_from_node(node));
2219 break;
2220 case HLSL_IR_CONSTRUCTOR:
2221 free_ir_constructor(constructor_from_node(node));
2222 break;
2223 case HLSL_IR_EXPR:
2224 free_ir_expr(expr_from_node(node));
2225 break;
2226 case HLSL_IR_ASSIGNMENT:
2227 free_ir_assignment(assignment_from_node(node));
2228 break;
2229 case HLSL_IR_IF:
2230 free_ir_if(if_from_node(node));
2231 break;
2232 case HLSL_IR_LOOP:
2233 free_ir_loop(loop_from_node(node));
2234 break;
2235 case HLSL_IR_JUMP:
2236 free_ir_jump(jump_from_node(node));
2237 break;
2238 default:
2239 FIXME("Unsupported node type %s\n", debug_node_type(node->type));
2240 }
2241 }
2242
2243 static void free_function_decl(struct hlsl_ir_function_decl *decl)
2244 {
2245 d3dcompiler_free((void *)decl->semantic);
2246 d3dcompiler_free(decl->parameters);
2247 free_instr_list(decl->body);
2248 d3dcompiler_free(decl);
2249 }
2250
2251 static void free_function_decl_rb(struct wine_rb_entry *entry, void *context)
2252 {
2253 free_function_decl(WINE_RB_ENTRY_VALUE(entry, struct hlsl_ir_function_decl, entry));
2254 }
2255
2256 static void free_function(struct hlsl_ir_function *func)
2257 {
2258 wine_rb_destroy(&func->overloads, free_function_decl_rb, NULL);
2259 d3dcompiler_free((void *)func->name);
2260 d3dcompiler_free(func);
2261 }
2262
2263 void free_function_rb(struct wine_rb_entry *entry, void *context)
2264 {
2265 free_function(WINE_RB_ENTRY_VALUE(entry, struct hlsl_ir_function, entry));
2266 }
2267
2268 void add_function_decl(struct wine_rb_tree *funcs, char *name, struct hlsl_ir_function_decl *decl, BOOL intrinsic)
2269 {
2270 struct hlsl_ir_function *func;
2271 struct wine_rb_entry *func_entry, *old_entry;
2272
2273 func_entry = wine_rb_get(funcs, name);
2274 if (func_entry)
2275 {
2276 func = WINE_RB_ENTRY_VALUE(func_entry, struct hlsl_ir_function, entry);
2277 if (intrinsic != func->intrinsic)
2278 {
2279 if (intrinsic)
2280 {
2281 ERR("Redeclaring a user defined function as an intrinsic.\n");
2282 return;
2283 }
2284 TRACE("Function %s redeclared as a user defined function.\n", debugstr_a(name));
2285 func->intrinsic = intrinsic;
2286 wine_rb_destroy(&func->overloads, free_function_decl_rb, NULL);
2287 wine_rb_init(&func->overloads, compare_function_decl_rb);
2288 }
2289 decl->func = func;
2290 if ((old_entry = wine_rb_get(&func->overloads, decl->parameters)))
2291 {
2292 struct hlsl_ir_function_decl *old_decl =
2293 WINE_RB_ENTRY_VALUE(old_entry, struct hlsl_ir_function_decl, entry);
2294
2295 if (!decl->body)
2296 {
2297 free_function_decl(decl);
2298 d3dcompiler_free(name);
2299 return;
2300 }
2301 wine_rb_remove(&func->overloads, old_entry);
2302 free_function_decl(old_decl);
2303 }
2304 wine_rb_put(&func->overloads, decl->parameters, &decl->entry);
2305 d3dcompiler_free(name);
2306 return;
2307 }
2308 func = d3dcompiler_alloc(sizeof(*func));
2309 func->name = name;
2310 wine_rb_init(&func->overloads, compare_function_decl_rb);
2311 decl->func = func;
2312 wine_rb_put(&func->overloads, decl->parameters, &decl->entry);
2313 func->intrinsic = intrinsic;
2314 wine_rb_put(funcs, func->name, &func->entry);
2315 }
2316 #endif
Windows API implementation