/
mojoshader.c
7217 lines (6090 loc) · 233 KB
1
2
/// !!! FIXME: all-DEF array in vs_3_0/14704.disasm ...
3
/**
4
5
* MojoShader; generate shader programs from bytecode of compiled
* Direct3D shaders.
6
7
8
9
10
11
*
* Please see the file LICENSE.txt in the source's root directory.
*
* This file written by Ryan C. Gordon.
*/
12
13
// !!! FIXME: this file really needs to be split up.
14
15
// !!! FIXME: I keep changing coding styles for symbols and typedefs.
16
17
18
// Shader bytecode format is described at MSDN:
// http://msdn2.microsoft.com/en-us/library/ms800307.aspx
19
#include <stdio.h>
20
#include <string.h>
21
#include <stdlib.h>
22
#include <stdarg.h>
23
#include <assert.h>
24
25
#include "mojoshader.h"
26
27
28
29
30
31
32
// This is the highest shader version we currently support.
#define MAX_SHADER_MAJOR 3
#define MAX_SHADER_MINOR 0
33
34
35
// 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.
36
37
38
39
40
#ifndef SUPPORT_PROFILE_D3D
#define SUPPORT_PROFILE_D3D 1
#endif
41
42
43
44
#ifndef SUPPORT_PROFILE_PASSTHROUGH
#define SUPPORT_PROFILE_PASSTHROUGH 1
#endif
45
46
47
48
#ifndef SUPPORT_PROFILE_GLSL
#define SUPPORT_PROFILE_GLSL 1
#endif
49
50
51
52
#ifndef SUPPORT_PROFILE_ARB1
#define SUPPORT_PROFILE_ARB1 1
#endif
53
54
55
56
// Get basic wankery out of the way here...
typedef unsigned int uint; // this is a printf() helper. don't use for code.
57
58
59
60
#ifdef _MSC_VER
#define snprintf _snprintf
typedef unsigned __int8 uint8;
61
typedef unsigned __int16 uint16;
62
63
typedef unsigned __int32 uint32;
typedef unsigned __int32 int32;
64
65
66
// Warning Level 4 considered harmful. :)
#pragma warning(disable: 4100) // "unreferenced formal parameter"
#pragma warning(disable: 4389) // "signed/unsigned mismatch"
67
68
#else
#include <stdint.h>
69
typedef uint8_t uint8;
70
typedef uint16_t uint16;
71
typedef uint32_t uint32;
72
typedef int32_t int32;
73
#endif
74
75
76
77
78
79
80
#ifdef __GNUC__
#define ISPRINTF(x,y) __attribute__((format (printf, x, y)))
#else
#define ISPRINTF(x,y)
#endif
81
82
#define STATICARRAYLEN(x) ( (sizeof ((x))) / (sizeof ((x)[0])) )
83
84
85
86
87
88
#ifdef _WINDOWS // !!! FIXME: bleh
const char *endline_str = "\r\n";
#else
const char *endline_str = "\n";
#endif
89
90
91
// we need to reference this by explicit value occasionally.
#define OPCODE_RET 28
92
93
94
95
96
// Special-case return values from the parsing pipeline...
#define FAIL (-1)
#define NOFAIL (-2)
#define END_OF_STREAM (-3)
97
98
99
100
101
102
103
104
105
// 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
106
107
108
109
110
static inline uint16 SWAP16(uint16 x)
{
__asm__ __volatile__("lhbrx %0,0,%1" : "=r" (x) : "r" (&x));
return x;
} // SWAP16
111
112
113
114
115
116
#elif defined(__POWERPC__)
static inline uint32 SWAP32(uint32 x)
{
return ( (((x) >> 24) & 0x000000FF) | (((x) >> 8) & 0x0000FF00) |
(((x) << 8) & 0x00FF0000) | (((x) << 24) & 0xFF000000) );
} // SWAP32
117
118
119
120
static inline uint16 SWAP16(uint16 x)
{
return ( (((x) >> 8) & 0x00FF) | (((x) << 8) & 0xFF00) );
} // SWAP16
121
#else
122
# define SWAP16(x) (x)
123
124
125
# define SWAP32(x) (x)
#endif
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
typedef enum
{
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
} RegisterType;
152
153
154
155
156
157
158
159
typedef enum
{
TEXTURE_TYPE_2D = 2,
TEXTURE_TYPE_CUBE = 3,
TEXTURE_TYPE_VOLUME = 4,
} TextureType;
160
// predeclare.
161
typedef struct Context Context;
162
163
// one emit function for each opcode in each profile.
164
typedef void (*emit_function)(Context *ctx);
165
166
// one emit function for starting output in each profile.
167
typedef void (*emit_start)(Context *ctx, const char *profilestr);
168
169
// one emit function for ending output in each profile.
170
typedef void (*emit_end)(Context *ctx);
171
172
173
174
// one emit function for phase opcode output in each profile.
typedef void (*emit_phase)(Context *ctx);
175
176
177
178
179
180
// one emit function for finalizing output in each profile.
typedef void (*emit_finalize)(Context *ctx);
// one emit function for global definitions in each profile.
typedef void (*emit_global)(Context *ctx, RegisterType regtype, int regnum);
181
// one emit function for relative uniform arrays in each profile.
182
typedef void (*emit_array)(Context *ctx, int base, int size);
183
184
// one emit function for uniforms in each profile.
185
186
typedef void (*emit_uniform)(Context *ctx, RegisterType regtype, int regnum,
int arraybase, int arraysize);
187
188
189
190
// one emit function for samplers in each profile.
typedef void (*emit_sampler)(Context *ctx, int stage, TextureType ttype);
191
192
193
194
// one emit function for attributes in each profile.
typedef void (*emit_attribute)(Context *ctx, RegisterType regtype, int regnum,
MOJOSHADER_usage usage, int index, int wmask);
195
196
197
// one args function for each possible sequence of opcode arguments.
typedef int (*args_function)(Context *ctx);
198
// one state function for each opcode where we have state machine updates.
199
typedef void (*state_function)(Context *ctx);
200
201
202
203
204
// one function for varnames in each profile.
typedef const char *(*varname_function)(Context *c, RegisterType t, int num);
// one function for const var array in each profile.
205
typedef const char *(*const_array_varname_function)(Context *c, int base, int size);
206
207
208
209
210
211
typedef struct
{
const char *name;
emit_start start_emitter;
emit_end end_emitter;
212
emit_phase phase_emitter;
213
emit_global global_emitter;
214
emit_array array_emitter;
215
emit_uniform uniform_emitter;
216
emit_sampler sampler_emitter;
217
emit_attribute attribute_emitter;
218
emit_finalize finalize_emitter;
219
220
varname_function get_varname;
const_array_varname_function get_const_array_varname;
221
} Profile;
222
223
224
225
226
227
228
typedef enum
{
RASTOUT_TYPE_POSITION = 0,
RASTOUT_TYPE_FOG = 1,
RASTOUT_TYPE_POINT_SIZE = 2,
RASTOUT_TYPE_MAX = 2
229
} RastOutType;
230
231
232
233
234
235
typedef enum
{
MISCTYPE_TYPE_POSITION = 0,
MISCTYPE_TYPE_FACE = 1,
MISCTYPE_TYPE_MAX = 1
236
} MiscTypeType;
237
238
239
240
241
// 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.
242
typedef struct OutputListNode
243
244
{
char *str;
245
246
247
248
249
250
251
struct OutputListNode *next;
} OutputListNode;
typedef struct OutputList
{
OutputListNode head;
OutputListNode *tail;
252
253
} OutputList;
254
255
256
257
258
typedef struct VariableList
{
MOJOSHADER_uniformType type;
int index;
int count;
259
int used;
260
261
struct VariableList *next;
} VariableList;
262
263
264
265
266
typedef struct RegisterList
{
RegisterType regtype;
int regnum;
267
268
269
MOJOSHADER_usage usage;
int index;
int writemask;
270
int misc;
271
const VariableList *array;
272
273
274
struct RegisterList *next;
} RegisterList;
275
276
277
278
279
typedef struct ConstantsList
{
MOJOSHADER_constant constant;
struct ConstantsList *next;
} ConstantsList;
280
281
282
283
284
285
// result modifiers.
#define MOD_SATURATE 0x01
#define MOD_PP 0x02
#define MOD_CENTROID 0x04
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
// 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
304
} SourceMod;
305
306
307
308
typedef struct
{
309
const uint32 *token; // this is the unmolested token in the stream.
310
311
int regnum;
int relative;
312
313
314
315
316
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
317
int orig_writemask; // writemask before mojoshader tweaks it.
318
319
int result_mod;
int result_shift;
320
RegisterType regtype;
321
322
323
324
} DestArgInfo;
typedef struct
{
325
const uint32 *token; // this is the unmolested token in the stream.
326
int regnum;
327
int swizzle; // xyzw (all four, not split out).
328
329
330
331
int swizzle_x;
int swizzle_y;
int swizzle_z;
int swizzle_w;
332
SourceMod src_mod;
333
RegisterType regtype;
334
335
336
337
int relative;
RegisterType relative_regtype;
int relative_regnum;
int relative_component;
338
const VariableList *relative_array;
339
340
341
} SourceArgInfo;
342
343
#define SCRATCH_BUFFER_SIZE 128
344
#define SCRATCH_BUFFERS 32
345
346
// !!! FIXME: the scratch buffers make Context pretty big.
347
348
// !!! FIXME: might be worth having one set of static scratch buffers that
// !!! FIXME: are mutex protected?
349
350
// Context...this is state that changes as we parse through a shader...
351
struct Context
352
{
353
354
MOJOSHADER_malloc malloc;
MOJOSHADER_free free;
355
void *malloc_data;
356
357
const uint32 *tokens;
uint32 tokencount;
358
359
OutputList *output;
OutputList globals;
360
OutputList helpers;
361
OutputList subroutines;
362
OutputList mainline_intro;
363
OutputList mainline;
364
365
OutputList ignore;
OutputList *output_stack[2];
366
uint8 *output_bytes; // can be used instead of the OutputLists.
367
368
int indent_stack[2];
int output_stack_len;
369
int output_len; // total strlen; prevents walking the lists just to malloc.
370
int indent;
371
const char *shader_type_str;
372
373
374
const char *endline;
int endline_len;
const char *failstr;
375
char scratch[SCRATCH_BUFFERS][SCRATCH_BUFFER_SIZE];
376
377
int scratchidx; // current scratch buffer.
int profileid;
378
const Profile *profile;
379
MOJOSHADER_shaderType shader_type;
380
381
uint8 major_ver;
uint8 minor_ver;
382
DestArgInfo dest_arg;
383
SourceArgInfo source_args[5];
384
SourceArgInfo predicate_arg; // for predicated instructions.
385
uint32 dwords[4];
386
uint32 version_token;
387
int instruction_count;
388
uint32 instruction_controls;
389
390
uint32 previous_opcode;
int loops;
391
int reps;
392
int max_reps;
393
int cmps;
394
395
int scratch_registers;
int max_scratch_registers;
396
397
398
int branch_labels_stack_index;
int branch_labels_stack[32];
int assigned_branch_labels;
399
int assigned_vertex_attributes;
400
int last_address_reg_component;
401
402
RegisterList used_registers;
RegisterList defined_registers;
403
404
int constant_count;
ConstantsList *constants;
405
406
int uniform_count;
RegisterList uniforms;
407
408
int attribute_count;
RegisterList attributes;
409
410
int sampler_count;
RegisterList samplers;
411
412
VariableList *variables; // variables to register mapping.
int have_ctab:1;
413
414
415
int predicated:1;
int support_nv2:1;
int glsl_generated_lit_opcode:1;
416
417
418
};
419
420
// Convenience functions for allocators...
421
422
423
424
425
426
static MOJOSHADER_parseData out_of_mem_data = {
"Out of memory", 0, 0, 0, 0, MOJOSHADER_TYPE_UNKNOWN, 0, 0, 0, 0
};
static const char *out_of_mem_str = "Out of memory";
static inline int out_of_memory(Context *ctx)
427
{
428
429
430
431
432
if (ctx->failstr == NULL)
ctx->failstr = out_of_mem_str; // fail() would call malloc().
return FAIL;
} // out_of_memory
433
static inline void *Malloc(Context *ctx, const size_t len)
434
{
435
void *retval = ctx->malloc((int) len, ctx->malloc_data);
436
437
438
if (retval == NULL)
out_of_memory(ctx);
return retval;
439
440
441
} // Malloc
442
static inline void Free(Context *ctx, void *ptr)
443
{
444
if (ptr != NULL) // check for NULL in case of dumb free() impl.
445
ctx->free(ptr, ctx->malloc_data);
446
447
448
} // Free
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
// 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...
473
474
475
476
477
static inline uint32 ver_ui32(const uint8 major, const uint8 minor)
{
return ( (((uint32) major) << 16) | (((minor) == 0xFF) ? 0 : (minor)) );
} // version_ui32
478
static inline int shader_version_supported(const uint8 maj, const uint8 min)
479
480
481
482
{
return (ver_ui32(maj,min) <= ver_ui32(MAX_SHADER_MAJOR, MAX_SHADER_MINOR));
} // shader_version_supported
483
484
static inline int shader_version_atleast(const Context *ctx, const uint8 maj,
const uint8 min)
485
486
487
488
{
return (ver_ui32(ctx->major_ver, ctx->minor_ver) >= ver_ui32(maj, min));
} // shader_version_atleast
489
490
491
492
493
494
static inline int shader_version_exactly(const Context *ctx, const uint8 maj,
const uint8 min)
{
return ((ctx->major_ver == maj) && (ctx->minor_ver == min));
} // shader_version_exactly
495
496
497
498
499
500
501
502
503
504
static inline int shader_is_pixel(const Context *ctx)
{
return (ctx->shader_type == MOJOSHADER_TYPE_PIXEL);
} // shader_is_pixel
static inline int shader_is_vertex(const Context *ctx)
{
return (ctx->shader_type == MOJOSHADER_TYPE_VERTEX);
} // shader_is_vertex
505
506
507
508
509
510
static inline int isfail(const Context *ctx)
{
return (ctx->failstr != NULL);
} // isfail
511
512
static inline char *get_scratch_buffer(Context *ctx)
513
{
514
515
516
517
518
519
520
521
522
523
524
525
if ((ctx->scratchidx >= SCRATCH_BUFFERS) && !isfail(ctx))
{
// can't call fail() here, since it calls back into here.
const char *errstr = "BUG: overflowed scratch buffers";
char *failstr = (char *) Malloc(ctx, strlen(errstr) + 1);
if (failstr != NULL)
{
strcpy(failstr, errstr);
ctx->failstr = failstr;
} // if
} // if
526
ctx->scratchidx = (ctx->scratchidx + 1) % SCRATCH_BUFFERS;
527
return ctx->scratch[ctx->scratchidx];
528
529
} // get_scratch_buffer
530
531
532
static int failf(Context *ctx, const char *fmt, ...) ISPRINTF(2,3);
static int failf(Context *ctx, const char *fmt, ...)
533
{
534
535
if (ctx->failstr == NULL) // don't change existing error.
{
536
char *scratch = get_scratch_buffer(ctx);
537
538
va_list ap;
va_start(ap, fmt);
539
const int len = vsnprintf(scratch, SCRATCH_BUFFER_SIZE, fmt, ap);
540
541
va_end(ap);
542
char *failstr = (char *) Malloc(ctx, len + 1);
543
if (failstr != NULL)
544
545
{
// see comments about scratch buffer overflow in output_line().
546
if (len < SCRATCH_BUFFER_SIZE)
547
548
549
550
551
552
553
554
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;
555
} // if
556
557
558
559
560
561
} // if
return FAIL;
} // failf
562
static inline int fail(Context *ctx, const char *reason)
563
564
565
566
567
{
return failf(ctx, "%s", reason);
} // fail
568
569
static int output_line(Context *ctx, const char *fmt, ...) ISPRINTF(2,3);
static int output_line(Context *ctx, const char *fmt, ...)
570
{
571
572
OutputListNode *item = NULL;
573
if (isfail(ctx))
574
575
return FAIL; // we failed previously, don't go on...
576
char *scratch = get_scratch_buffer(ctx);
577
578
const int indent = ctx->indent;
579
580
if (indent > 0)
memset(scratch, '\t', indent);
581
582
583
va_list ap;
va_start(ap, fmt);
584
const int len = vsnprintf(scratch+indent, SCRATCH_BUFFER_SIZE-indent, fmt, ap) + indent;
585
586
va_end(ap);
587
item = (OutputListNode *) Malloc(ctx, sizeof (OutputListNode));
588
if (item == NULL)
589
return FAIL;
590
591
item->str = (char *) Malloc(ctx, len + 1);
592
if (item->str == NULL)
593
{
594
Free(ctx, item);
595
return FAIL;
596
597
} // if
598
599
600
// 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().
601
if (len < SCRATCH_BUFFER_SIZE)
602
603
604
strcpy(item->str, scratch); // copy it over.
else
{
605
606
if (indent > 0)
memset(item->str, '\t', indent);
607
va_start(ap, fmt);
608
vsnprintf(item->str+indent, len + 1, fmt, ap); // rebuild it.
609
610
va_end(ap);
} // else
611
612
item->next = NULL;
613
614
615
ctx->output->tail->next = item;
ctx->output->tail = item;
616
ctx->output_len += len + ctx->endline_len;
617
618
619
620
621
return 0;
} // output_line
622
623
624
625
626
627
628
// this is just to stop gcc whining.
static inline int output_blank_line(Context *ctx)
{
return output_line(ctx, "%s", "");
} // output_blank_line
629
// !!! FIXME: this is sort of nasty.
630
631
static void floatstr(Context *ctx, char *buf, size_t bufsize, float f,
int leavedecimal)
632
633
{
const size_t len = snprintf(buf, bufsize, "%f", f);
634
if ((len+2) >= bufsize)
635
636
637
638
639
640
fail(ctx, "BUG: internal buffer is too small");
else
{
char *end = buf + len;
char *ptr = strchr(buf, '.');
if (ptr == NULL)
641
642
643
{
if (leavedecimal)
strcat(buf, ".0");
644
return; // done.
645
} // if
646
647
648
649
650
651
652
653
654
while (--end != ptr)
{
if (*end != '0')
{
end++;
break;
} // if
} // while
655
656
if ((leavedecimal) && (end == ptr))
end += 2;
657
658
659
660
*end = '\0'; // chop extra '0' or all decimal places off.
} // else
} // floatstr
661
662
663
// Deal with register lists... !!! FIXME: I sort of hate this.
664
static void free_reglist(MOJOSHADER_free f, void *d, RegisterList *item)
665
666
667
668
{
while (item != NULL)
{
RegisterList *next = item->next;
669
f(item, d);
670
671
672
673
674
675
676
677
678
item = next;
} // while
} // free_reglist
static inline uint32 reg_to_ui32(const RegisterType regtype, const int regnum)
{
return ( ((uint32) regtype) | (((uint32) regnum) << 16) );
} // reg_to_uint32
679
680
681
static RegisterList *reglist_insert(Context *ctx, RegisterList *prev,
const RegisterType regtype,
const int regnum)
682
683
684
685
686
687
688
{
const uint32 newval = reg_to_ui32(regtype, regnum);
RegisterList *item = prev->next;
while (item != NULL)
{
const uint32 val = reg_to_ui32(item->regtype, item->regnum);
if (newval == val)
689
return item; // already set, so we're done.
690
691
692
693
694
695
696
697
698
699
700
else if (newval < val) // insert it here.
break;
else // if (newval > val)
{
// keep going, we're not to the insertion point yet.
prev = item;
item = item->next;
} // else
} // while
// we need to insert an entry after (prev).
701
item = (RegisterList *) Malloc(ctx, sizeof (RegisterList));
702
if (item != NULL)
703
704
705
{
item->regtype = regtype;
item->regnum = regnum;
706
item->usage = MOJOSHADER_USAGE_UNKNOWN;
707
708
item->index = 0;
item->writemask = 0;
709
item->misc = 0;
710
item->array = NULL;
711
712
item->next = prev->next;
prev->next = item;
713
} // if
714
715
return item;
716
717
} // reglist_insert
718
719
static RegisterList *reglist_find(RegisterList *prev, const RegisterType rtype,
const int regnum)
720
{
721
722
const uint32 newval = reg_to_ui32(rtype, regnum);
RegisterList *item = prev->next;
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
while (item != NULL)
{
const uint32 val = reg_to_ui32(item->regtype, item->regnum);
if (newval == val)
return item; // here it is.
else if (newval < val) // should have been here if it existed.
return NULL;
else // if (newval > val)
{
// keep going, we're not to the insertion point yet.
prev = item;
item = item->next;
} // else
} // while
return NULL; // wasn't in the list.
739
740
741
742
743
744
745
} // reglist_find
static inline const RegisterList *reglist_exists(RegisterList *prev,
const RegisterType regtype,
const int regnum)
{
return (reglist_find(prev, regtype, regnum));
746
747
748
749
750
751
752
753
754
755
756
757
} // reglist_exists
static inline void set_used_register(Context *ctx, const RegisterType regtype,
const int regnum)
{
reglist_insert(ctx, &ctx->used_registers, regtype, regnum);
} // set_used_register
static inline int get_used_register(Context *ctx, const RegisterType regtype,
const int regnum)
{
return (reglist_exists(&ctx->used_registers, regtype, regnum) != NULL);
758
} // get_used_register
759
760
761
762
763
764
765
766
767
768
769
770
771
static inline void set_defined_register(Context *ctx, const RegisterType rtype,
const int regnum)
{
reglist_insert(ctx, &ctx->defined_registers, rtype, regnum);
} // set_defined_register
static inline int get_defined_register(Context *ctx, const RegisterType rtype,
const int regnum)
{
return (reglist_exists(&ctx->defined_registers, rtype, regnum) != NULL);
} // get_defined_register
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
static const RegisterList *declared_attribute(Context *ctx,
const MOJOSHADER_usage usage,
const int index)
{
const RegisterList *item = ctx->attributes.next;
while (item != NULL)
{
if ((item->usage == usage) && (item->index == index))
return item;
item = item->next;
} // while
return NULL;
} // declared_attribute
787
788
789
790
791
792
793
794
795
796
static void add_attribute_register(Context *ctx, const RegisterType rtype,
const int regnum, const MOJOSHADER_usage usage,
const int index, const int writemask)
{
RegisterList *item = reglist_insert(ctx, &ctx->attributes, rtype, regnum);
item->usage = usage;
item->index = index;
item->writemask = writemask;
} // add_attribute_register
797
798
799
static inline void add_sampler(Context *ctx, const RegisterType rtype,
const int regnum, const TextureType ttype)
{
800
// !!! FIXME: make sure it doesn't exist?
801
802
803
804
RegisterList *item = reglist_insert(ctx, &ctx->samplers, rtype, regnum);
item->index = (int) ttype;
} // add_sampler
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
static inline int writemask_xyzw(const int writemask)
{
return (writemask == 0xF); // 0xF == 1111. No explicit mask (full!).
} // writemask_xyzw
static inline int writemask_xyz(const int writemask)
{
return (writemask == 0x7); // 0x7 == 0111. (that is: xyz)
} // writemask_xyz
static inline int writemask_xy(const int writemask)
{
return (writemask == 0x3); // 0x3 == 0011. (that is: xy)
} // writemask_xy
static inline int writemask_x(const int writemask)
{
return (writemask == 0x1); // 0x1 == 0001. (that is: x)
} // writemask_x
static inline int writemask_y(const int writemask)
{
return (writemask == 0x2); // 0x1 == 0010. (that is: y)
} // writemask_y
836
837
838
839
840
841
842
843
static inline int replicate_swizzle(const int swizzle)
{
return ( (((swizzle >> 0) & 0x3) == ((swizzle >> 2) & 0x3)) &&
(((swizzle >> 2) & 0x3) == ((swizzle >> 4) & 0x3)) &&
(((swizzle >> 4) & 0x3) == ((swizzle >> 6) & 0x3)) );
} // replicate_swizzle
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
static inline int scalar_register(const RegisterType regtype, const int regnum)
{
switch (regtype)
{
case REG_TYPE_DEPTHOUT:
case REG_TYPE_CONSTBOOL:
case REG_TYPE_PREDICATE:
case REG_TYPE_LOOP:
return 1;
case REG_TYPE_MISCTYPE:
if ( ((const MiscTypeType) regnum) == MISCTYPE_TYPE_FACE )
return 1;
return 0;
default: break;
} // switch
return 0;
} // scalar_register
866
867
static inline int no_swizzle(const int swizzle)
{
868
return (swizzle == 0xE4); // 0xE4 == 11100100 ... 0 1 2 3. No swizzle.
869
870
871
} // no_swizzle
872
873
874
875
876
static inline int vecsize_from_writemask(const int m)
{
return (m & 1) + ((m >> 1) & 1) + ((m >> 2) & 1) + ((m >> 3) & 1);
} // vecsize_from_writemask
877
878
879
880
881
882
883
884
static int allocate_scratch_register(Context *ctx)
{
const int retval = ctx->scratch_registers++;
if (retval >= ctx->max_scratch_registers)
ctx->max_scratch_registers = retval + 1;
return retval;
} // allocate_scratch_register
885
static int allocate_branch_label(Context *ctx)
886
{
887
888
return ctx->assigned_branch_labels++;
} // allocate_branch_label
889
890
891
892
// D3D stuff that's used in more than just the d3d profile...
893
894
895
static const char swizzle_channels[] = { 'x', 'y', 'z', 'w' };
896
897
898
899
900
static const char *usagestrs[] = {
"_position", "_blendweight", "_blendindices", "_normal", "_psize",
"_texcoord", "_tangent", "_binormal", "_tessfactor", "_positiont",
"_color", "_fog", "_depth", "_sample"
};
901
902
static const char *get_D3D_register_string(Context *ctx,
903
RegisterType regtype,
904
905
int regnum, char *regnum_str,
size_t regnum_size)
906
{
907
const char *retval = NULL;
908
int has_number = 1;
909
910
switch (regtype)
911
{
912
case REG_TYPE_TEMP:
913
retval = "r";
914
915
break;
916
case REG_TYPE_INPUT:
917
retval = "v";
918
919
break;
920
case REG_TYPE_CONST:
921
922
923
retval = "c";
break;
924
case REG_TYPE_ADDRESS: // (or REG_TYPE_TEXTURE, same value.)
925
retval = shader_is_vertex(ctx) ? "a" : "t";
926
927
break;
928
case REG_TYPE_RASTOUT:
929
switch ((RastOutType) regnum)
930
{
931
932
933
case RASTOUT_TYPE_POSITION: retval = "oPos"; break;
case RASTOUT_TYPE_FOG: retval = "oFog"; break;
case RASTOUT_TYPE_POINT_SIZE: retval = "oPts"; break;
934
} // switch
935
has_number = 0;
936
937
break;
938
case REG_TYPE_ATTROUT:
939
retval = "oD";
940
941
break;
942
case REG_TYPE_OUTPUT: // (or REG_TYPE_TEXCRDOUT, same value.)
943
if (shader_is_vertex(ctx) && shader_version_atleast(ctx, 3, 0))
944
retval = "o";
945
else
946
retval = "oT";
947
948
break;
949
case REG_TYPE_CONSTINT:
950
retval = "i";
951
952
break;
953
case REG_TYPE_COLOROUT:
954
retval = "oC";
955
956
break;
957
case REG_TYPE_DEPTHOUT:
958
retval = "oDepth";
959
has_number = 0;
960
961
break;
962
case REG_TYPE_SAMPLER:
963
retval = "s";
964
965
break;
966
case REG_TYPE_CONSTBOOL:
967
retval = "b";
968
969
break;
970
case REG_TYPE_LOOP:
971
retval = "aL";
972
has_number = 0;
973
974
break;
975
case REG_TYPE_MISCTYPE:
976
switch ((const MiscTypeType) regnum)
977
{
978
979
case MISCTYPE_TYPE_POSITION: retval = "vPos"; break;
case MISCTYPE_TYPE_FACE: retval = "vFace"; break;
980
} // switch
981
has_number = 0;
982
983
break;
984
case REG_TYPE_LABEL:
985
retval = "l";
986
987
break;
988
case REG_TYPE_PREDICATE:
989
retval = "p";
990
break;
991
992
993
994
995
996
997
//case REG_TYPE_TEMPFLOAT16: // !!! FIXME: don't know this asm string
default:
fail(ctx, "unknown register type");
retval = "???";
has_number = 0;
break;
998
999
} // switch
1000
if (has_number)