/
mojoshader.c
3454 lines (2869 loc) · 106 KB
1
/**
2
3
* MojoShader; generate shader programs from bytecode of compiled
* Direct3D shaders.
4
5
6
7
8
9
*
* Please see the file LICENSE.txt in the source's root directory.
*
* This file written by Ryan C. Gordon.
*/
10
11
// !!! FIXME: I keep changing coding styles for symbols and typedefs.
12
13
14
15
// !!! FIXME: do DEF* opcodes have to come before instructions?
// !!! FIXME: my reading of the msdn spec suggests no, but it sounds like
// !!! FIXME: something they'd require. DCL_* _does_ have to be first.
16
17
18
19
// Shader bytecode format is described at MSDN:
// http://msdn2.microsoft.com/en-us/library/ms800307.aspx
20
#include <stdio.h>
21
#include <string.h>
22
#include <stdlib.h>
23
#include <stdint.h>
24
#include <stdarg.h>
25
#include <assert.h>
26
27
#include "mojoshader.h"
28
29
30
31
32
33
34
// This is the highest shader version we currently support.
#define MAX_SHADER_MAJOR 3
#define MAX_SHADER_MINOR 0
35
36
37
// If SUPPORT_PROFILE_* isn't defined, we assume an implicit desire to support.
// You get all the profiles unless you go out of your way to disable them.
38
39
40
41
42
43
44
45
46
#ifndef SUPPORT_PROFILE_D3D
#define SUPPORT_PROFILE_D3D 1
#endif
#ifndef SUPPORT_PROFILE_GLSL
#define SUPPORT_PROFILE_GLSL 1
#endif
47
48
49
50
51
52
// Get basic wankery out of the way here...
typedef unsigned int uint; // this is a printf() helper. don't use for code.
typedef uint8_t uint8;
typedef uint32_t uint32;
53
typedef int32_t int32;
54
55
56
57
58
59
60
#ifdef __GNUC__
#define ISPRINTF(x,y) __attribute__((format (printf, x, y)))
#else
#define ISPRINTF(x,y)
#endif
61
62
#define STATICARRAYLEN(x) ( (sizeof ((x))) / (sizeof ((x)[0])) )
63
64
65
66
67
68
69
#ifdef _WINDOWS // !!! FIXME: bleh
const char *endline_str = "\r\n";
#else
const char *endline_str = "\n";
#endif
70
71
72
// we need to reference this by explicit value occasionally.
#define OPCODE_RET 28
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
// Byteswap magic...
#if ((defined __GNUC__) && (defined __POWERPC__))
static inline uint32 SWAP32(uint32 x)
{
__asm__ __volatile__("lwbrx %0,0,%1" : "=r" (x) : "r" (&x));
return x;
} // SWAP32
#elif defined(__POWERPC__)
static inline uint32 SWAP32(uint32 x)
{
return ( (((x) >> 24) & 0x000000FF) | (((x) >> 8) & 0x0000FF00) |
(((x) << 8) & 0x00FF0000) | (((x) << 24) & 0xFF000000) );
} // SWAP32
#else
# define SWAP32(x) (x)
#endif
92
93
// predeclare.
94
typedef struct Context Context;
95
96
// one emit function for each opcode in each profile.
97
typedef void (*emit_function)(Context *ctx);
98
99
// one emit function for comments in each profile.
100
typedef void (*emit_comment)(Context *ctx, const char *str);
101
102
// one emit function for starting output in each profile.
103
typedef void (*emit_start)(Context *ctx);
104
105
// one emit function for ending output in each profile.
106
typedef void (*emit_end)(Context *ctx);
107
108
109
110
// one args function for each possible sequence of opcode arguments.
typedef int (*args_function)(Context *ctx);
111
// one state function for each opcode where we have state machine updates.
112
typedef void (*state_function)(Context *ctx);
113
114
115
116
117
118
119
typedef struct
{
const char *name;
emit_start start_emitter;
emit_end end_emitter;
emit_comment comment_emitter;
120
} Profile;
121
122
typedef MOJOSHADER_shaderType ShaderType;
123
124
125
typedef enum
{
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
REG_TYPE_TEMP = 0,
REG_TYPE_INPUT = 1,
REG_TYPE_CONST = 2,
REG_TYPE_ADDRESS = 3,
REG_TYPE_TEXTURE = 3, // ALSO 3!
REG_TYPE_RASTOUT = 4,
REG_TYPE_ATTROUT = 5,
REG_TYPE_TEXCRDOUT = 6,
REG_TYPE_OUTPUT = 6, // ALSO 6!
REG_TYPE_CONSTINT = 7,
REG_TYPE_COLOROUT = 8,
REG_TYPE_DEPTHOUT = 9,
REG_TYPE_SAMPLER = 10,
REG_TYPE_CONST2 = 11,
REG_TYPE_CONST3 = 12,
REG_TYPE_CONST4 = 13,
REG_TYPE_CONSTBOOL = 14,
REG_TYPE_LOOP = 15,
REG_TYPE_TEMPFLOAT16 = 16,
REG_TYPE_MISCTYPE = 17,
REG_TYPE_LABEL = 18,
REG_TYPE_PREDICATE = 19,
REG_TYPE_MAX = 19
149
} RegisterType;
150
151
152
153
154
155
156
typedef enum
{
RASTOUT_TYPE_POSITION = 0,
RASTOUT_TYPE_FOG = 1,
RASTOUT_TYPE_POINT_SIZE = 2,
RASTOUT_TYPE_MAX = 2
157
} RastOutType;
158
159
160
161
162
163
typedef enum
{
MISCTYPE_TYPE_POSITION = 0,
MISCTYPE_TYPE_FACE = 1,
MISCTYPE_TYPE_MAX = 1
164
} MiscTypeType;
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
typedef enum
{
DECLUSAGE_POSITION = 0,
DECLUSAGE_BLENDWEIGHT = 1,
DECLUSAGE_BLENDINDICES = 2,
DECLUSAGE_NORMAL = 3,
DECLUSAGE_PSIZE = 4,
DECLUSAGE_TEXCOORD = 5,
DECLUSAGE_TANGENT = 6,
DECLUSAGE_BINORMAL = 7,
DECLUSAGE_TESSFACTOR = 8,
DECLUSAGE_POSITIONT = 9,
DECLUSAGE_COLOR = 10,
DECLUSAGE_FOG = 11,
DECLUSAGE_DEPTH = 12,
DECLUSAGE_SAMPLE = 13
182
} DeclUsageType;
183
184
185
186
187
188
typedef enum
{
TEXTURE_TYPE_2D = 2,
TEXTURE_TYPE_CUBE = 3,
TEXTURE_TYPE_VOLUME = 4,
189
} TextureType;
190
191
192
193
194
// A simple linked list of strings, so we can build the final output without
// realloc()ing for each new line, and easily insert lines into the middle
// of the output without much trouble.
195
typedef struct OutputListNode
196
197
{
char *str;
198
199
200
201
202
203
204
struct OutputListNode *next;
} OutputListNode;
typedef struct OutputList
{
OutputListNode head;
OutputListNode *tail;
205
206
} OutputList;
207
208
209
210
211
212
// result modifiers.
#define MOD_SATURATE 0x01
#define MOD_PP 0x02
#define MOD_CENTROID 0x04
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
// source modifiers.
typedef enum
{
SRCMOD_NONE,
SRCMOD_NEGATE,
SRCMOD_BIAS,
SRCMOD_BIASNEGATE,
SRCMOD_SIGN,
SRCMOD_SIGNNEGATE,
SRCMOD_COMPLEMENT,
SRCMOD_X2,
SRCMOD_X2NEGATE,
SRCMOD_DZ,
SRCMOD_DW,
SRCMOD_ABS,
SRCMOD_ABSNEGATE,
SRCMOD_NOT,
SRCMOD_TOTAL
231
} SourceMod;
232
233
234
235
typedef struct
{
236
const uint32 *token; // this is the unmolested token in the stream.
237
238
int regnum;
int relative;
239
240
241
242
243
int writemask; // xyzw or rgba (all four, not split out).
int writemask0; // x or red
int writemask1; // y or green
int writemask2; // z or blue
int writemask3; // w or alpha
244
245
int result_mod;
int result_shift;
246
RegisterType regtype;
247
248
249
250
} DestArgInfo;
typedef struct
{
251
const uint32 *token; // this is the unmolested token in the stream.
252
253
int regnum;
int relative;
254
int swizzle; // xyzw (all four, not split out).
255
256
257
258
259
int swizzle_x;
int swizzle_y;
int swizzle_z;
int swizzle_w;
int src_mod;
260
RegisterType regtype;
261
262
263
} SourceArgInfo;
264
265
typedef enum
{
266
267
268
269
270
271
272
273
// May apply to any profile...
CTX_FLAGS_USED_ADDR_REG = (1 << 0),
CTX_FLAGS_USED_PRED_REG = (1 << 1),
// Specific to GLSL profile...
CTX_FLAGS_GLSL_LIT_OPCODE = (1 << 2),
CTX_FLAGS_GLSL_DST_OPCODE = (1 << 3),
CTX_FLAGS_GLSL_LRP_OPCODE = (1 << 4),
274
275
276
277
CTX_FLAGS_MASK = 0xFFFFFFFF
} ContextFlags;
278
279
#define SCRATCH_BUFFER_SIZE 256
#define SCRATCH_BUFFERS 10
280
281
282
283
284
285
286
// !!! FIXME: labels_called and the scratch buffers make this pretty big.
// !!! FIXME: might be worth having one set of static scratch buffers that
// !!! FIXME: are mutex protected?
// !!! FIXME: and replace the bit array for labels_called with a linked list?
// !!! FIXME: maybe just malloc() the label list, since it can't be more than
// !!! FIXME: around (bytes_of_shader / 20) elements in the pathological case?
287
// Context...this is state that changes as we parse through a shader...
288
struct Context
289
{
290
291
MOJOSHADER_malloc malloc;
MOJOSHADER_free free;
292
293
const uint32 *tokens;
uint32 tokencount;
294
295
OutputList *output;
OutputList globals;
296
OutputList helpers;
297
298
OutputList subroutines;
OutputList mainline;
299
300
301
302
OutputList ignore;
OutputList *output_stack[2];
int indent_stack[2];
int output_stack_len;
303
int output_len; // total strlen; prevents walking the lists just to malloc.
304
int indent;
305
306
307
const char *endline;
int endline_len;
const char *failstr;
308
char scratch[SCRATCH_BUFFERS][SCRATCH_BUFFER_SIZE];
309
310
int scratchidx; // current scratch buffer.
int profileid;
311
312
const Profile *profile;
ShaderType shader_type;
313
314
uint8 major_ver;
uint8 minor_ver;
315
DestArgInfo dest_args[1];
316
SourceArgInfo source_args[5];
317
uint32 dwords[4];
318
int instruction_count;
319
uint32 instruction_controls;
320
321
322
323
uint32 previous_opcode;
ContextFlags flags;
uint8 labels_called[256];
int loops;
324
325
326
};
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
// jump between output sections in the context...
static inline void push_output(Context *ctx, OutputList *section)
{
assert(ctx->output_stack_len < STATICARRAYLEN(ctx->output_stack));
ctx->output_stack[ctx->output_stack_len] = ctx->output;
ctx->indent_stack[ctx->output_stack_len] = ctx->indent;
ctx->output_stack_len++;
ctx->output = section;
ctx->indent = 0;
} // push_output
static inline void pop_output(Context *ctx)
{
assert(ctx->output_stack_len > 0);
ctx->output_stack_len--;
ctx->output = ctx->output_stack[ctx->output_stack_len];
ctx->indent = ctx->indent_stack[ctx->output_stack_len];
} // pop_output
// Shader model version magic...
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
static inline uint32 ver_ui32(const uint8 major, const uint8 minor)
{
return ( (((uint32) major) << 16) | (((minor) == 0xFF) ? 0 : (minor)) );
} // version_ui32
static int shader_version_supported(uint8 maj, uint8 min)
{
return (ver_ui32(maj,min) <= ver_ui32(MAX_SHADER_MAJOR, MAX_SHADER_MINOR));
} // shader_version_supported
static int shader_version_atleast(const Context *ctx, uint8 maj, uint8 min)
{
return (ver_ui32(ctx->major_ver, ctx->minor_ver) >= ver_ui32(maj, min));
} // shader_version_atleast
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
// Bit arrays (for storing large arrays of booleans...
static void set_bit_array(uint8 *array, size_t arraylen, int index, int val)
{
const int byteindex = index / 8;
const int bitindex = index % 8;
assert(byteindex < arraylen);
if (val)
array[byteindex] |= (1 << bitindex);
else
array[byteindex] &= ~(1 << bitindex);
} // set_bit_array
static int get_bit_array(const uint8 *array, size_t arraylen, int index)
{
const int byteindex = index / 8;
const int bitindex = index % 8;
assert(byteindex < arraylen);
const uint8 byte = array[byteindex];
return (byte & (1 << bitindex)) ? 1 : 0;
} // get_bit_array
390
391
static inline char *get_scratch_buffer(Context *ctx)
392
{
393
ctx->scratchidx = (ctx->scratchidx + 1) % SCRATCH_BUFFERS;
394
return ctx->scratch[ctx->scratchidx];
395
396
} // get_scratch_buffer
397
398
399
// Special-case return values from the parsing pipeline...
#define FAIL (-1)
400
401
#define NOFAIL (-2)
#define END_OF_STREAM (-3)
402
403
static inline int isfail(const Context *ctx)
404
405
406
407
408
{
return (ctx->failstr != NULL);
} // isfail
409
410
411
412
413
static MOJOSHADER_parseData out_of_mem_data = {
"Out of memory", 0, 0, 0, MOJOSHADER_TYPE_UNKNOWN, 0, 0, 0, 0
};
static const char *out_of_mem_str = "Out of memory";
414
415
416
static inline int out_of_memory(Context *ctx)
{
if (ctx->failstr == NULL)
417
ctx->failstr = out_of_mem_str; // fail() would call malloc().
418
419
420
421
422
423
return FAIL;
} // out_of_memory
static int failf(Context *ctx, const char *fmt, ...) ISPRINTF(2,3);
static int failf(Context *ctx, const char *fmt, ...)
424
{
425
426
if (ctx->failstr == NULL) // don't change existing error.
{
427
char *scratch = get_scratch_buffer(ctx);
428
429
va_list ap;
va_start(ap, fmt);
430
const int len = vsnprintf(scratch,SCRATCH_BUFFER_SIZE,fmt,ap);
431
432
va_end(ap);
433
434
435
436
437
438
char *failstr = (char *) ctx->malloc(len + 1);
if (failstr == NULL)
out_of_memory(ctx);
else
{
// see comments about scratch buffer overflow in output_line().
439
if (len < SCRATCH_BUFFER_SIZE)
440
441
442
443
444
445
446
447
448
strcpy(failstr, scratch); // copy it over.
else
{
va_start(ap, fmt);
vsnprintf(failstr, len + 1, fmt, ap); // rebuild it.
va_end(ap);
} // else
ctx->failstr = failstr;
} // else
449
450
451
452
453
454
} // if
return FAIL;
} // failf
455
static inline int fail(Context *ctx, const char *reason)
456
457
458
459
460
{
return failf(ctx, "%s", reason);
} // fail
461
462
static int output_line(Context *ctx, const char *fmt, ...) ISPRINTF(2,3);
static int output_line(Context *ctx, const char *fmt, ...)
463
{
464
465
OutputListNode *item = NULL;
466
if (isfail(ctx))
467
468
return FAIL; // we failed previously, don't go on...
469
char *scratch = get_scratch_buffer(ctx);
470
471
const int indent = ctx->indent;
472
473
if (indent > 0)
memset(scratch, '\t', indent);
474
475
476
va_list ap;
va_start(ap, fmt);
477
const int len = vsnprintf(scratch+indent, SCRATCH_BUFFER_SIZE-indent, fmt, ap) + indent;
478
479
va_end(ap);
480
481
482
483
item = (OutputListNode *) ctx->malloc(sizeof (OutputListNode));
if (item == NULL)
return out_of_memory(ctx);
484
485
item->str = (char *) ctx->malloc(len + 1);
if (item->str == NULL)
486
{
487
488
free(item);
return out_of_memory(ctx);
489
490
} // if
491
492
493
// If we overflowed our scratch buffer, that's okay. We were going to
// allocate anyhow...the scratch buffer just lets us avoid a second
// run of vsnprintf().
494
if (len < SCRATCH_BUFFER_SIZE)
495
496
497
strcpy(item->str, scratch); // copy it over.
else
{
498
499
if (indent > 0)
memset(item->str, '\t', indent);
500
va_start(ap, fmt);
501
vsnprintf(item->str+indent, len + 1, fmt, ap); // rebuild it.
502
503
va_end(ap);
} // else
504
505
item->next = NULL;
506
507
508
ctx->output->tail->next = item;
ctx->output->tail = item;
509
ctx->output_len += len + ctx->endline_len;
510
511
512
513
514
return 0;
} // output_line
515
516
517
518
519
520
521
// this is just to stop gcc whining.
static inline int output_blank_line(Context *ctx)
{
return output_line(ctx, "%s", "");
} // output_blank_line
522
// !!! FIXME: this is sort of nasty.
523
524
static void floatstr(Context *ctx, char *buf, size_t bufsize, float f,
int leavedecimal)
525
526
{
const size_t len = snprintf(buf, bufsize, "%f", f);
527
if ((len+2) >= bufsize)
528
529
530
531
532
533
fail(ctx, "BUG: internal buffer is too small");
else
{
char *end = buf + len;
char *ptr = strchr(buf, '.');
if (ptr == NULL)
534
535
536
{
if (leavedecimal)
strcat(buf, ".0");
537
return; // done.
538
} // if
539
540
541
542
543
544
545
546
547
while (--end != ptr)
{
if (*end != '0')
{
end++;
break;
} // if
} // while
548
549
if ((leavedecimal) && (end == ptr))
end += 2;
550
551
552
553
*end = '\0'; // chop extra '0' or all decimal places off.
} // else
} // floatstr
554
555
// This is used in more than just the d3d profile.
556
static const char *get_D3D_register_string(Context *ctx,
557
RegisterType regtype,
558
559
int regnum, char *regnum_str,
size_t regnum_size)
560
{
561
const char *retval = NULL;
562
int has_number = 1;
563
564
switch (regtype)
565
{
566
case REG_TYPE_TEMP:
567
retval = "r";
568
569
break;
570
case REG_TYPE_INPUT:
571
retval = "v";
572
573
break;
574
case REG_TYPE_CONST:
575
576
577
retval = "c";
break;
578
case REG_TYPE_CONST2:
579
580
581
582
retval = "c";
regnum += 2048;
break;
583
case REG_TYPE_CONST3:
584
585
586
587
retval = "c";
regnum += 4096;
break;
588
case REG_TYPE_CONST4:
589
retval = "c";
590
regnum += 6144;
591
592
break;
593
case REG_TYPE_ADDRESS: // (or REG_TYPE_TEXTURE, same value.)
594
retval = (ctx->shader_type == MOJOSHADER_TYPE_VERTEX) ? "a" : "t";
595
596
break;
597
case REG_TYPE_RASTOUT:
598
switch ((RastOutType) regnum)
599
{
600
601
602
case RASTOUT_TYPE_POSITION: retval = "oPos"; break;
case RASTOUT_TYPE_FOG: retval = "oFog"; break;
case RASTOUT_TYPE_POINT_SIZE: retval = "oPts"; break;
603
} // switch
604
has_number = 0;
605
606
break;
607
case REG_TYPE_ATTROUT:
608
retval = "oD";
609
610
break;
611
case REG_TYPE_OUTPUT: // (or REG_TYPE_TEXCRDOUT, same value.)
612
if ((ctx->shader_type==MOJOSHADER_TYPE_VERTEX) && (ctx->major_ver>=3))
613
retval = "o";
614
else
615
retval = "oT";
616
617
break;
618
case REG_TYPE_CONSTINT:
619
retval = "i";
620
621
break;
622
case REG_TYPE_COLOROUT:
623
retval = "oC";
624
625
break;
626
case REG_TYPE_DEPTHOUT:
627
retval = "oDepth";
628
has_number = 0;
629
630
break;
631
case REG_TYPE_SAMPLER:
632
retval = "s";
633
634
break;
635
case REG_TYPE_CONSTBOOL:
636
retval = "b";
637
638
break;
639
case REG_TYPE_LOOP:
640
retval = "aL";
641
has_number = 0;
642
643
644
break;
// !!! FIXME: don't know what the asm string is for this..
645
case REG_TYPE_TEMPFLOAT16:
646
fail(ctx, "don't know the ASM for tempfloat16 register");
647
648
649
retval = "???";
has_number = 0;
break;
650
651
case REG_TYPE_MISCTYPE:
652
switch ((MiscTypeType) regnum)
653
{
654
655
case MISCTYPE_TYPE_POSITION: retval = "vPos"; break;
case MISCTYPE_TYPE_FACE: retval = "vFace"; break;
656
} // switch
657
has_number = 0;
658
659
break;
660
case REG_TYPE_LABEL:
661
retval = "l";
662
663
break;
664
case REG_TYPE_PREDICATE:
665
retval = "p";
666
667
668
break;
} // switch
669
670
671
672
673
if (has_number)
snprintf(regnum_str, regnum_size, "%u", (uint) regnum);
else
regnum_str[0] = '\0';
674
675
676
677
return retval;
} // get_D3D_register_string
678
679
680
681
682
683
684
685
686
687
#define AT_LEAST_ONE_PROFILE 0
#if !SUPPORT_PROFILE_D3D
#define PROFILE_EMITTER_D3D(op)
#else
#undef AT_LEAST_ONE_PROFILE
#define AT_LEAST_ONE_PROFILE 1
#define PROFILE_EMITTER_D3D(op) emit_D3D_##op,
688
static const char *make_D3D_destarg_string(Context *ctx, const int idx)
689
690
691
692
{
if (idx >= STATICARRAYLEN(ctx->dest_args))
{
fail(ctx, "Too many destination args");
693
return "";
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
} // if
const DestArgInfo *arg = &ctx->dest_args[idx];
const char *result_shift_str = "";
switch (arg->result_shift)
{
case 0x1: result_shift_str = "_x2"; break;
case 0x2: result_shift_str = "_x4"; break;
case 0x3: result_shift_str = "_x8"; break;
case 0xD: result_shift_str = "_d8"; break;
case 0xE: result_shift_str = "_d4"; break;
case 0xF: result_shift_str = "_d2"; break;
} // switch
const char *sat_str = (arg->result_mod & MOD_SATURATE) ? "_sat" : "";
const char *pp_str = (arg->result_mod & MOD_PP) ? "_pp" : "";
const char *cent_str = (arg->result_mod & MOD_CENTROID) ? "_centroid" : "";
char regnum_str[16];
714
715
716
const char *regtype_str = get_D3D_register_string(ctx, arg->regtype,
arg->regnum, regnum_str,
sizeof (regnum_str));
717
718
719
if (regtype_str == NULL)
{
fail(ctx, "Unknown destination register type.");
720
return "";
721
722
} // if
723
724
725
726
727
728
729
730
731
732
733
734
char writemask_str[6];
int i = 0;
if (arg->writemask != 0xF) // 0xF == 1111. No explicit mask.
{
writemask_str[i++] = '.';
if (arg->writemask0) writemask_str[i++] = 'x';
if (arg->writemask1) writemask_str[i++] = 'y';
if (arg->writemask2) writemask_str[i++] = 'z';
if (arg->writemask3) writemask_str[i++] = 'w';
} // if
writemask_str[i] = '\0';
assert(i < sizeof (writemask_str));
735
736
// may turn out something like "_x2_sat_pp_centroid r0.xyzw" ...
737
char *retval = get_scratch_buffer(ctx);
738
snprintf(retval, SCRATCH_BUFFER_SIZE, "%s%s%s%s %s%s%s",
739
result_shift_str, sat_str, pp_str, cent_str,
740
regtype_str, regnum_str, writemask_str);
741
// !!! FIXME: make sure the scratch buffer was large enough.
742
return retval;
743
} // make_D3D_destarg_string
744
745
746
static const char *make_D3D_sourcearg_string(Context *ctx, const int idx)
747
{
748
749
750
if (idx >= STATICARRAYLEN(ctx->source_args))
{
fail(ctx, "Too many source args");
751
return "";
752
} // if
753
754
755
756
757
const SourceArgInfo *arg = &ctx->source_args[idx];
const char *premod_str = "";
const char *postmod_str = "";
758
switch ((SourceMod) arg->src_mod)
759
760
761
762
763
764
765
766
767
768
769
{
case SRCMOD_NEGATE:
premod_str = "-";
break;
case SRCMOD_BIASNEGATE:
premod_str = "-";
// fall through.
case SRCMOD_BIAS:
postmod_str = "_bias";
break;
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
case SRCMOD_SIGNNEGATE:
premod_str = "-";
// fall through.
case SRCMOD_SIGN:
postmod_str = "_bx2";
break;
case SRCMOD_COMPLEMENT:
premod_str = "1-";
break;
case SRCMOD_X2NEGATE:
premod_str = "-";
// fall through.
case SRCMOD_X2:
postmod_str = "_x2";
break;
case SRCMOD_DZ:
postmod_str = "_dz";
break;
case SRCMOD_DW:
postmod_str = "_dw";
break;
case SRCMOD_ABSNEGATE:
premod_str = "-";
// fall through.
case SRCMOD_ABS:
postmod_str = "_abs";
break;
case SRCMOD_NOT:
premod_str = "!";
break;
807
808
809
810
case SRCMOD_NONE:
case SRCMOD_TOTAL:
break; // stop compiler whining.
811
812
813
814
} // switch
char regnum_str[16];
815
816
817
const char *regtype_str = get_D3D_register_string(ctx, arg->regtype,
arg->regnum, regnum_str,
sizeof (regnum_str));
818
819
820
821
if (regtype_str == NULL)
{
fail(ctx, "Unknown source register type.");
822
return "";
823
824
825
826
} // if
char swizzle_str[6];
int i = 0;
827
if (arg->swizzle != 0xE4) // 0xE4 == 11100100 ... 3 2 1 0. No swizzle.
828
829
830
{
static const char channel[] = { 'x', 'y', 'z', 'w' };
swizzle_str[i++] = '.';
831
832
833
834
835
836
837
838
swizzle_str[i++] = channel[arg->swizzle_x];
swizzle_str[i++] = channel[arg->swizzle_y];
swizzle_str[i++] = channel[arg->swizzle_z];
swizzle_str[i++] = channel[arg->swizzle_w];
// .xyzz is the same as .xyz, .z is the same as .zzzz, etc.
while (swizzle_str[i-1] == swizzle_str[i-2])
i--;
839
840
841
842
} // if
swizzle_str[i] = '\0';
assert(i < sizeof (swizzle_str));
843
char *retval = get_scratch_buffer(ctx);
844
845
snprintf(retval, SCRATCH_BUFFER_SIZE, "%s%s%s%s%s",
premod_str, regtype_str, regnum_str, postmod_str, swizzle_str);
846
// !!! FIXME: make sure the scratch buffer was large enough.
847
return retval;
848
} // make_D3D_sourcearg_string
849
850
851
static void emit_D3D_start(Context *ctx)
852
853
854
{
const uint major = (uint) ctx->major_ver;
const uint minor = (uint) ctx->minor_ver;
855
856
857
858
859
860
861
862
863
const char *shadertype_str = NULL;
char minor_str[16];
if (minor == 0xFF)
strcpy(minor_str, "sw");
else if (minor == 0x1) // apparently this is "vs_2_x". Weird.
strcpy(minor_str, "x");
else
snprintf(minor_str, sizeof (minor_str), "%u", (uint) minor);
864
865
if (ctx->shader_type == MOJOSHADER_TYPE_PIXEL)
866
shadertype_str = "ps";
867
else if (ctx->shader_type == MOJOSHADER_TYPE_VERTEX)
868
shadertype_str = "vs";
869
else
870
871
872
{
failf(ctx, "Shader type %u unsupported in this profile.",
(uint) ctx->shader_type);
873
return;
874
} // else
875
876
output_line(ctx, "%s_%u_%s", shadertype_str, major, minor_str);
877
878
} // emit_D3D_start
879
880
static void emit_D3D_end(Context *ctx)
881
{
882
output_line(ctx, "end");
883
} // emit_D3D_end
884
885
886
static void emit_D3D_comment(Context *ctx, const char *str)
887
{
888
889
output_line(ctx, "; %s", str);
} // emit_D3D_comment
890
891
892
static void emit_D3D_RESERVED(Context *ctx)
893
{
894
895
896
// do nothing; fails in the state machine.
} // emit_D3D_RESERVED
897
898
899
900
// Generic D3D opcode emitters. A list of macros generate all the entry points
// that call into these...
901
902
903
904
905
906
907
908
909
910
911
912
static char *lowercase(char *dst, const char *src)
{
int i = 0;
do
{
const char ch = src[i];
dst[i] = (((ch >= 'A') && (ch <= 'Z')) ? (ch - ('A' - 'a')) : ch);
} while (src[i++]);
return dst;
} // lowercase
913
static void emit_D3D_opcode_d(Context *ctx, const char *opcode)
914
{
915
const char *dst0 = make_D3D_destarg_string(ctx, 0);
916
opcode = lowercase(get_scratch_buffer(ctx), opcode);
917
output_line(ctx, "%s%s", opcode, dst0);
918
919
920
} // emit_D3D_opcode_d
921
static void emit_D3D_opcode_s(Context *ctx, const char *opcode)
922
{
923
const char *src0 = make_D3D_sourcearg_string(ctx, 0);
924
opcode = lowercase(get_scratch_buffer(ctx), opcode);
925
output_line(ctx, "%s %s", opcode, src0);
926
927
928
} // emit_D3D_opcode_s
929
static void emit_D3D_opcode_ss(Context *ctx, const char *opcode)
930
{
931
932
const char *src0 = make_D3D_sourcearg_string(ctx, 0);
const char *src1 = make_D3D_sourcearg_string(ctx, 1);
933
opcode = lowercase(get_scratch_buffer(ctx), opcode);
934
output_line(ctx, "%s %s, %s", opcode, src0, src1);
935
} // emit_D3D_opcode_ss
936
937
938
static void emit_D3D_opcode_ds(Context *ctx, const char *opcode)
939
{
940
941
const char *dst0 = make_D3D_destarg_string(ctx, 0);
const char *src0 = make_D3D_sourcearg_string(ctx, 0);
942
opcode = lowercase(get_scratch_buffer(ctx), opcode);
943
output_line(ctx, "%s%s, %s", opcode, dst0, src0);
944
945
946
} // emit_D3D_opcode_ds
947
static void emit_D3D_opcode_dss(Context *ctx, const char *opcode)
948
{
949
950
951
const char *dst0 = make_D3D_destarg_string(ctx, 0);
const char *src0 = make_D3D_sourcearg_string(ctx, 0);
const char *src1 = make_D3D_sourcearg_string(ctx, 1);
952
opcode = lowercase(get_scratch_buffer(ctx), opcode);
953
output_line(ctx, "%s%s, %s, %s", opcode, dst0, src0, src1);
954
955
956
} // emit_D3D_opcode_dss
957
static void emit_D3D_opcode_dsss(Context *ctx, const char *opcode)
958
{
959
960
961
962
const char *dst0 = make_D3D_destarg_string(ctx, 0);
const char *src0 = make_D3D_sourcearg_string(ctx, 0);
const char *src1 = make_D3D_sourcearg_string(ctx, 1);
const char *src2 = make_D3D_sourcearg_string(ctx, 2);
963
opcode = lowercase(get_scratch_buffer(ctx), opcode);
964
output_line(ctx, "%s%s, %s, %s, %s", opcode, dst0, src0, src1, src2);
965
966
967
} // emit_D3D_opcode_dsss
968
static void emit_D3D_opcode_dssss(Context *ctx, const char *opcode)
969
{
970
971
972
973
974
const char *dst0 = make_D3D_destarg_string(ctx, 0);
const char *src0 = make_D3D_sourcearg_string(ctx, 0);
const char *src1 = make_D3D_sourcearg_string(ctx, 1);
const char *src2 = make_D3D_sourcearg_string(ctx, 2);
const char *src3 = make_D3D_sourcearg_string(ctx, 3);
975
opcode = lowercase(get_scratch_buffer(ctx), opcode);
976
output_line(ctx,"%s%s, %s, %s, %s, %s",opcode,dst0,src0,src1,src2,src3);
977
978
979
} // emit_D3D_opcode_dssss
980
981
982
983
984
985
986
static void emit_D3D_opcode(Context *ctx, const char *opcode)
{
opcode = lowercase(get_scratch_buffer(ctx), opcode);
output_line(ctx, "%s", opcode);
} // emit_D3D_opcode_dssss
987
#define EMIT_D3D_OPCODE_FUNC(op) \
988
static void emit_D3D_##op(Context *ctx) { \
989
emit_D3D_opcode(ctx, #op); \
990
991
}
#define EMIT_D3D_OPCODE_D_FUNC(op) \
992
static void emit_D3D_##op(Context *ctx) { \
993
994
995
emit_D3D_opcode_d(ctx, #op); \
}
#define EMIT_D3D_OPCODE_S_FUNC(op) \
996
static void emit_D3D_##op(Context *ctx) { \
997
998
999
emit_D3D_opcode_s(ctx, #op); \
}
#define EMIT_D3D_OPCODE_SS_FUNC(op) \
1000
static void emit_D3D_##op(Context *ctx) { \