/
mojoshader_compiler.c
2676 lines (2324 loc) · 83.5 KB
1
2
3
4
5
6
7
8
9
/**
* MojoShader; generate shader programs from bytecode of compiled
* Direct3D shaders.
*
* Please see the file LICENSE.txt in the source's root directory.
*
* This file written by Ryan C. Gordon.
*/
10
11
12
13
// !!! FIXME: this needs to be split into separate source files:
// !!! FIXME: parse, AST, IR, etc. The problem is we need to deal with the
// !!! FIXME: "Context" struct being passed around everywhere.
14
15
#define __MOJOSHADER_INTERNAL__ 1
#include "mojoshader_internal.h"
16
17
18
19
20
#if DEBUG_COMPILER_PARSER
#define LEMON_SUPPORT_TRACING 1
#endif
21
22
// !!! FIXME: I'd like to lose this. It's really inefficient. Just keep a
// !!! FIXME: (tail) on these list structures instead?
23
24
25
26
27
28
29
30
31
32
33
34
35
36
#define REVERSE_LINKED_LIST(typ, head) { \
if ((head) && (head->next)) { \
typ *tmp = NULL; \
typ *tmp1 = NULL; \
while (head != NULL) { \
tmp = head; \
head = head->next; \
tmp->next = tmp1; \
tmp1 = tmp; \
} \
head = tmp; \
} \
}
37
typedef union TokenData
38
{
39
40
41
int64 i64;
double dbl;
const char *string;
42
43
} TokenData;
44
45
// Structures that make up the parse tree...
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
typedef enum ASTNodeType
{
AST_OP_START_RANGE,
AST_OP_START_RANGE_UNARY,
AST_OP_POSTINCREMENT,
AST_OP_POSTDECREMENT,
AST_OP_PREINCREMENT,
AST_OP_PREDECREMENT,
AST_OP_NEGATE,
AST_OP_COMPLEMENT,
AST_OP_NOT,
AST_OP_END_RANGE_UNARY,
AST_OP_START_RANGE_BINARY,
AST_OP_DEREF_ARRAY,
AST_OP_CALLFUNC,
AST_OP_DEREF_STRUCT,
AST_OP_COMMA,
AST_OP_MULTIPLY,
AST_OP_DIVIDE,
AST_OP_MODULO,
AST_OP_ADD,
AST_OP_SUBTRACT,
AST_OP_LSHIFT,
AST_OP_RSHIFT,
AST_OP_LESSTHAN,
AST_OP_GREATERTHAN,
AST_OP_LESSTHANOREQUAL,
AST_OP_GREATERTHANOREQUAL,
AST_OP_EQUAL,
AST_OP_NOTEQUAL,
AST_OP_BINARYAND,
AST_OP_BINARYXOR,
AST_OP_BINARYOR,
AST_OP_LOGICALAND,
AST_OP_LOGICALOR,
AST_OP_ASSIGN,
AST_OP_MULASSIGN,
AST_OP_DIVASSIGN,
AST_OP_MODASSIGN,
AST_OP_ADDASSIGN,
AST_OP_SUBASSIGN,
AST_OP_LSHIFTASSIGN,
AST_OP_RSHIFTASSIGN,
AST_OP_ANDASSIGN,
AST_OP_XORASSIGN,
AST_OP_ORASSIGN,
AST_OP_END_RANGE_BINARY,
AST_OP_START_RANGE_TERNARY,
AST_OP_CONDITIONAL,
AST_OP_END_RANGE_TERNARY,
AST_OP_START_RANGE_DATA,
AST_OP_IDENTIFIER,
AST_OP_INT_LITERAL,
AST_OP_FLOAT_LITERAL,
AST_OP_STRING_LITERAL,
105
AST_OP_BOOLEAN_LITERAL,
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
AST_OP_END_RANGE_DATA,
AST_OP_START_RANGE_MISC,
AST_OP_CONSTRUCTOR,
AST_OP_CAST,
AST_OP_END_RANGE_MISC,
AST_OP_END_RANGE,
AST_COMPUNIT_START_RANGE,
AST_COMPUNIT_FUNCTION, // function declaration or definition
AST_COMPUNIT_TYPEDEF, // typedef or struct
AST_COMPUNIT_STRUCT, // global struct
AST_COMPUNIT_VARIABLE, // global variable
AST_COMPUNIT_END_RANGE,
AST_STATEMENT_START_RANGE,
122
AST_STATEMENT_BLOCK,
123
124
125
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
152
153
154
155
156
157
158
159
160
161
AST_STATEMENT_EMPTY,
AST_STATEMENT_EXPRESSION,
AST_STATEMENT_IF,
AST_STATEMENT_SWITCH,
AST_STATEMENT_FOR,
AST_STATEMENT_DO,
AST_STATEMENT_WHILE,
AST_STATEMENT_RETURN,
AST_STATEMENT_BREAK,
AST_STATEMENT_CONTINUE,
AST_STATEMENT_DISCARD,
AST_STATEMENT_TYPEDEF,
AST_STATEMENT_STRUCT,
AST_STATEMENT_VARDECL,
AST_STATEMENT_END_RANGE,
AST_MISC_START_RANGE,
AST_FUNCTION_ARGS,
AST_FUNCTION_SIGNATURE,
AST_SCALAR_OR_ARRAY,
AST_TYPEDEF,
AST_PACK_OFFSET,
AST_VARIABLE_LOWLEVEL,
AST_ANNOTATION,
AST_VARIABLE_DECLARATION,
AST_STRUCT_DECLARATION,
AST_STRUCT_MEMBER,
AST_SWITCH_CASE,
AST_MISC_END_RANGE,
AST_END_RANGE
} ASTNodeType;
typedef struct ASTNode
{
ASTNodeType type;
const char *filename;
uint32 line;
} ASTNode;
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
typedef enum VariableAttributes
{
VARATTR_EXTERN = (1 << 0),
VARATTR_NOINTERPOLATION = (1 << 1),
VARATTR_SHARED = (1 << 2),
VARATTR_STATIC = (1 << 3),
VARATTR_UNIFORM = (1 << 4),
VARATTR_VOLATILE = (1 << 5),
VARATTR_CONST = (1 << 6),
VARATTR_ROWMAJOR = (1 << 7),
VARATTR_COLUMNMAJOR = (1 << 8)
} VariableAttributes;
typedef enum IfAttributes
{
IFATTR_NONE,
IFATTR_BRANCH,
IFATTR_FLATTEN,
IFATTR_IFALL,
IFATTR_IFANY,
IFATTR_PREDICATE,
IFATTR_PREDICATEBLOCK,
} IfAttributes;
typedef enum SwitchAttributes
{
SWITCHATTR_NONE,
SWITCHATTR_FLATTEN,
SWITCHATTR_BRANCH,
SWITCHATTR_FORCECASE,
SWITCHATTR_CALL
} SwitchAttributes;
196
static inline int operator_is_unary(const ASTNodeType op)
197
{
198
return ((op > AST_OP_START_RANGE_UNARY) && (op < AST_OP_END_RANGE_UNARY));
199
200
} // operator_is_unary
201
static inline int operator_is_binary(const ASTNodeType op)
202
{
203
return ((op > AST_OP_START_RANGE_BINARY) && (op < AST_OP_END_RANGE_BINARY));
204
205
} // operator_is_binary
206
static inline int operator_is_ternary(const ASTNodeType op)
207
{
208
return ((op > AST_OP_START_RANGE_TERNARY) && (op < AST_OP_END_RANGE_TERNARY));
209
210
} // operator_is_ternary
211
typedef struct ASTGeneric
212
{
213
214
215
216
ASTNode ast;
} ASTGeneric;
typedef ASTGeneric Expression;
217
218
219
typedef struct ExpressionUnary
{
220
ASTNode ast;
221
222
223
224
225
Expression *operand;
} ExpressionUnary;
typedef struct ExpressionBinary
{
226
ASTNode ast;
227
228
229
230
231
232
Expression *left;
Expression *right;
} ExpressionBinary;
typedef struct ExpressionTernary
{
233
ASTNode ast;
234
235
236
237
238
239
240
Expression *left;
Expression *center;
Expression *right;
} ExpressionTernary;
typedef struct ExpressionIdentifier
{
241
ASTNode ast; // Always AST_OP_IDENTIFIER
242
243
244
245
246
const char *identifier;
} ExpressionIdentifier;
typedef struct ExpressionIntLiteral
{
247
ASTNode ast; // Always AST_OP_INT_LITERAL
248
249
250
251
252
int64 value;
} ExpressionIntLiteral;
typedef struct ExpressionFloatLiteral
{
253
ASTNode ast; // Always AST_OP_FLOAT_LITERAL
254
255
256
257
258
double value;
} ExpressionFloatLiteral;
typedef struct ExpressionStringLiteral
{
259
ASTNode ast; // Always AST_OP_STRING_LITERAL
260
261
262
const char *string;
} ExpressionStringLiteral;
263
264
265
266
267
268
typedef struct ExpressionBooleanLiteral
{
ASTNode ast; // Always AST_OP_BOOLEAN_LITERAL
int value; // Always 1 or 0.
} ExpressionBooleanLiteral;
269
270
typedef struct ExpressionConstructor
{
271
ASTNode ast; // Always AST_OP_CONSTRUCTOR
272
273
274
275
276
277
const char *datatype;
Expression *args;
} ExpressionConstructor;
typedef struct ExpressionCast
{
278
ASTNode ast; // Always AST_OP_CAST
279
280
281
282
const char *datatype;
Expression *operand;
} ExpressionCast;
283
284
typedef struct CompilationUnit
{
285
ASTNode ast;
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
struct CompilationUnit *next;
} CompilationUnit;
typedef enum FunctionStorageClass
{
FNSTORECLS_NONE,
FNSTORECLS_INLINE
} FunctionStorageClass;
typedef enum InputModifier
{
INPUTMOD_NONE,
INPUTMOD_IN,
INPUTMOD_OUT,
INPUTMOD_INOUT,
INPUTMOD_UNIFORM
} InputModifier;
typedef enum InterpolationModifier
{
INTERPMOD_NONE,
INTERPMOD_LINEAR,
INTERPMOD_CENTROID,
INTERPMOD_NOINTERPOLATION,
INTERPMOD_NOPERSPECTIVE,
INTERPMOD_SAMPLE
} InterpolationModifier;
typedef struct FunctionArguments
{
316
ASTNode ast;
317
318
319
320
321
322
323
324
325
326
327
InputModifier input_modifier;
const char *datatype;
const char *identifier;
const char *semantic;
InterpolationModifier interpolation_modifier;
Expression *initializer;
struct FunctionArguments *next;
} FunctionArguments;
typedef struct FunctionSignature
{
328
ASTNode ast;
329
330
331
332
333
334
335
336
337
const char *datatype;
const char *identifier;
FunctionArguments *args;
FunctionStorageClass storage_class;
const char *semantic;
} FunctionSignature;
typedef struct ScalarOrArray
{
338
ASTNode ast;
339
340
341
342
343
344
345
const char *identifier;
int isarray;
Expression *dimension;
} ScalarOrArray;
typedef struct Annotations
{
346
ASTNode ast;
347
348
349
350
351
352
353
const char *datatype;
Expression *initializer;
struct Annotations *next;
} Annotations;
typedef struct PackOffset
{
354
ASTNode ast;
355
356
357
358
359
360
const char *ident1; // !!! FIXME: rename this.
const char *ident2;
} PackOffset;
typedef struct VariableLowLevel
{
361
ASTNode ast;
362
363
364
365
366
367
PackOffset *packoffset;
const char *register_name;
} VariableLowLevel;
typedef struct StructMembers
{
368
ASTNode ast;
369
370
371
372
373
374
375
376
377
const char *datatype;
const char *semantic;
ScalarOrArray *details;
InterpolationModifier interpolation_mod;
struct StructMembers *next;
} StructMembers;
typedef struct StructDeclaration
{
378
ASTNode ast;
379
380
381
382
383
384
const char *name;
StructMembers *members;
} StructDeclaration;
typedef struct VariableDeclaration
{
385
ASTNode ast;
386
387
388
389
390
391
392
393
394
395
396
397
398
int attributes;
const char *datatype;
StructDeclaration *anonymous_datatype;
ScalarOrArray *details;
const char *semantic;
Annotations *annotations;
Expression *initializer;
VariableLowLevel *lowlevel;
struct VariableDeclaration *next;
} VariableDeclaration;
typedef struct Statement
{
399
ASTNode ast;
400
401
402
403
404
405
406
407
struct Statement *next;
} Statement;
typedef Statement EmptyStatement;
typedef Statement BreakStatement;
typedef Statement ContinueStatement;
typedef Statement DiscardStatement;
408
409
410
411
412
413
414
typedef struct BlockStatement // something enclosed in "{}" braces.
{
ASTNode ast;
struct Statement *next;
Statement *statements; // list of statements enclosed by this block.
} BlockStatement;
415
416
typedef struct ReturnStatement
{
417
ASTNode ast;
418
419
420
421
422
423
struct Statement *next;
Expression *expr;
} ReturnStatement;
typedef struct ExpressionStatement
{
424
ASTNode ast;
425
426
427
428
429
430
struct Statement *next;
Expression *expr;
} ExpressionStatement;
typedef struct IfStatement
{
431
ASTNode ast;
432
433
434
435
436
437
438
439
440
struct Statement *next;
int attributes;
Expression *expr;
Statement *statement;
Statement *else_statement;
} IfStatement;
typedef struct SwitchCases
{
441
ASTNode ast;
442
443
444
445
446
447
448
Expression *expr;
Statement *statement;
struct SwitchCases *next;
} SwitchCases;
typedef struct SwitchStatement
{
449
ASTNode ast;
450
451
452
453
454
455
456
457
struct Statement *next;
int attributes;
Expression *expr;
SwitchCases *cases;
} SwitchStatement;
typedef struct WhileStatement
{
458
ASTNode ast;
459
struct Statement *next;
460
int64 unroll; // # times to unroll, 0 to loop, negative for compiler's choice.
461
462
463
464
465
466
467
468
Expression *expr;
Statement *statement;
} WhileStatement;
typedef WhileStatement DoStatement;
typedef struct ForStatement
{
469
ASTNode ast;
470
struct Statement *next;
471
int64 unroll; // # times to unroll, 0 to loop, negative for compiler's choice.
472
473
474
475
476
477
478
479
480
VariableDeclaration *var_decl;
Expression *initializer;
Expression *looptest;
Expression *counter;
Statement *statement;
} ForStatement;
typedef struct Typedef
{
481
ASTNode ast;
482
483
484
485
486
487
488
int isconst;
const char *datatype;
ScalarOrArray *details;
} Typedef;
typedef struct TypedefStatement
{
489
ASTNode ast;
490
491
492
493
494
495
struct Statement *next;
Typedef *type_info;
} TypedefStatement;
typedef struct VarDeclStatement
{
496
ASTNode ast;
497
struct Statement *next;
498
VariableDeclaration *declaration;
499
500
501
502
} VarDeclStatement;
typedef struct StructStatement
{
503
ASTNode ast;
504
505
506
507
508
509
struct Statement *next;
StructDeclaration *struct_info;
} StructStatement;
typedef struct CompilationUnitFunction
{
510
ASTNode ast;
511
512
513
514
515
516
517
struct CompilationUnit *next;
FunctionSignature *declaration;
Statement *definition;
} CompilationUnitFunction;
typedef struct CompilationUnitTypedef
{
518
ASTNode ast;
519
520
521
522
523
524
struct CompilationUnit *next;
Typedef *type_info;
} CompilationUnitTypedef;
typedef struct CompilationUnitStruct
{
525
ASTNode ast;
526
527
528
529
530
531
struct CompilationUnit *next;
StructDeclaration *struct_info;
} CompilationUnitStruct;
typedef struct CompilationUnitVariable
{
532
ASTNode ast;
533
534
535
536
537
struct CompilationUnit *next;
VariableDeclaration *declaration;
} CompilationUnitVariable;
538
// This tracks data types and variables, and notes when they enter/leave scope.
539
540
typedef struct SymbolScope
541
542
543
{
const char *symbol;
const char *datatype;
544
545
struct SymbolScope *next;
} SymbolScope;
546
547
typedef struct SymbolMap
548
{
549
550
551
HashTable *hash;
SymbolScope *scope;
} SymbolMap;
552
553
554
555
556
557
558
559
560
561
562
563
564
// Compile state, passed around all over the place.
typedef struct Context
{
int isfail;
int out_of_memory;
MOJOSHADER_malloc malloc;
MOJOSHADER_free free;
void *malloc_data;
int error_count;
ErrorList *errors;
565
StringCache *strcache;
566
567
const char *sourcefile; // current source file that we're parsing.
unsigned int sourceline; // current line in sourcefile that we're parsing.
568
569
SymbolMap usertypes;
SymbolMap variables;
570
CompilationUnit *ast; // Abstract Syntax Tree
571
572
573
574
575
576
577
578
579
// These are entries allocated in the strcache, so these pointers are
// valid from shortly after we create the cache until we destroy it
// with the rest of the context. This makes it so we can compare common
// strings by pointer without having to hash them every time, so long as
// we're comparing a string pointer we know came from this string cache.
const char *str_b; // "b"
const char *str_f; // "f"
const char *str_i; // "i"
580
581
582
583
584
585
586
} Context;
// Convenience functions for allocators...
static inline void out_of_memory(Context *ctx)
{
587
if (!ctx->out_of_memory) printf("out of memory\n"); // !!! FIXME: placeholder.
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
ctx->isfail = ctx->out_of_memory = 1;
} // out_of_memory
static inline void *Malloc(Context *ctx, const size_t len)
{
void *retval = ctx->malloc((int) len, ctx->malloc_data);
if (retval == NULL)
out_of_memory(ctx);
return retval;
} // Malloc
static inline char *StrDup(Context *ctx, const char *str)
{
char *retval = (char *) Malloc(ctx, strlen(str) + 1);
if (retval != NULL)
strcpy(retval, str);
return retval;
} // StrDup
static inline void Free(Context *ctx, void *ptr)
{
if (ptr != NULL) // check for NULL in case of dumb free() impl.
ctx->free(ptr, ctx->malloc_data);
} // Free
613
614
615
616
static void fail(Context *ctx, const char *str)
{
// !!! FIXME: placeholder.
(void) ctx;
617
printf("%s:%u: %s\n", ctx->sourcefile, ctx->sourceline, str);
618
} // fail
619
620
621
static void symbolmap_nuke(const void *k, const void *v, void *d) {/*no-op*/}
622
623
static int create_symbolmap(Context *ctx, SymbolMap *map)
624
{
625
// !!! FIXME: should compare string pointer, with string in cache.
626
map->scope = NULL;
627
628
629
630
map->hash = hash_create(ctx, hash_hash_string, hash_keymatch_string,
symbolmap_nuke, 1, ctx->malloc, ctx->free,
ctx->malloc_data);
if (!map->hash)
631
632
633
634
635
636
{
out_of_memory(ctx);
return 0;
} // if
return 1;
637
} // create_symbolmap
638
639
640
641
642
643
static void push_symbol(Context *ctx, SymbolMap *map,
const char *sym, const char *datatype)
{
SymbolScope *item = (SymbolScope *) Malloc(ctx, sizeof (SymbolScope));
644
645
646
647
648
if (item == NULL)
return;
if (sym != NULL)
{
649
if (hash_insert(map->hash, sym, datatype) == -1)
650
651
652
653
654
655
656
657
658
659
{
Free(ctx, item);
return;
}
} // if
item->symbol = sym; // cached strings, don't copy.
item->datatype = datatype;
item->next = map->scope;
map->scope = item;
660
661
662
663
664
} // push_symbol
static inline void push_usertype(Context *ctx, const char *sym, const char *dt)
{
push_symbol(ctx, &ctx->usertypes, sym, dt);
665
666
} // push_usertype
667
668
669
670
671
672
673
674
675
676
677
678
679
static inline void push_variable(Context *ctx, const char *sym, const char *dt)
{
push_symbol(ctx, &ctx->variables, sym, dt);
} // push_variable
static inline void push_scope(Context *ctx)
{
push_usertype(ctx, NULL, NULL);
push_variable(ctx, NULL, NULL);
} // push_scope
static void pop_symbol(Context *ctx, SymbolMap *map)
680
{
681
SymbolScope *item = map->scope;
682
683
684
if (!item)
return;
if (item->symbol)
685
hash_remove(map->hash, item->symbol);
686
687
map->scope = item->next;
Free(ctx, item);
688
} // pop_symbol
689
690
static void pop_symbol_scope(Context *ctx, SymbolMap *map)
691
692
{
while ((map->scope) && (map->scope->symbol))
693
pop_symbol(ctx, map);
694
695
696
assert(map->scope != NULL);
assert(map->scope->symbol == NULL);
697
698
699
700
701
702
703
pop_symbol(ctx, map);
} // pop_symbol_scope
static inline void pop_scope(Context *ctx)
{
pop_symbol_scope(ctx, &ctx->usertypes);
pop_symbol_scope(ctx, &ctx->variables);
704
705
} // push_scope
706
static void destroy_symbolmap(Context *ctx, SymbolMap *map)
707
708
{
while (map->scope)
709
710
711
pop_symbol_scope(ctx, map);
hash_destroy(map->hash);
} // destroy_symbolmap
712
713
714
715
716
717
718
// These functions are mostly for construction and cleanup of nodes in the
// parse tree. Mostly this is simple allocation and initialization, so we
// can do as little in the lemon code as possible, and then sort it all out
// afterwards.
719
#define NEW_AST_NODE(retval, cls, typ) \
720
cls *retval = (cls *) Malloc(ctx, sizeof (cls)); \
721
722
723
724
725
726
727
728
do { \
if (retval == NULL) { return NULL; } \
retval->ast.type = typ; \
retval->ast.filename = ctx->sourcefile; \
retval->ast.line = ctx->sourceline; \
} while (0)
#define DELETE_AST_NODE(cls) do { \
729
730
731
732
733
734
if (!cls) return; \
} while (0)
static void delete_compilation_unit(Context *ctx, CompilationUnit *unit);
static void delete_statement(Context *ctx, Statement *stmt);
735
736
737
static Expression *new_constructor_expr(Context *ctx, const char *datatype,
Expression *args)
{
738
NEW_AST_NODE(retval, ExpressionConstructor, AST_OP_CONSTRUCTOR);
739
740
741
742
743
744
745
746
retval->datatype = datatype;
retval->args = args;
return (Expression *) retval;
} // new_constructor_expr
static Expression *new_cast_expr(Context *ctx, const char *datatype,
Expression *operand)
{
747
NEW_AST_NODE(retval, ExpressionCast, AST_OP_CAST);
748
749
750
751
752
retval->datatype = datatype;
retval->operand = operand;
return (Expression *) retval;
} // new_cast_expr
753
static Expression *new_unary_expr(Context *ctx, const ASTNodeType op,
754
755
Expression *operand)
{
756
NEW_AST_NODE(retval, ExpressionUnary, op);
757
758
759
760
761
assert(operator_is_unary(op));
retval->operand = operand;
return (Expression *) retval;
} // new_unary_expr
762
static Expression *new_binary_expr(Context *ctx, const ASTNodeType op,
763
764
Expression *left, Expression *right)
{
765
NEW_AST_NODE(retval, ExpressionBinary, op);
766
767
768
769
770
771
assert(operator_is_binary(op));
retval->left = left;
retval->right = right;
return (Expression *) retval;
} // new_binary_expr
772
static Expression *new_ternary_expr(Context *ctx, const ASTNodeType op,
773
774
775
Expression *left, Expression *center,
Expression *right)
{
776
NEW_AST_NODE(retval, ExpressionTernary, op);
777
778
779
780
781
782
783
784
785
assert(operator_is_ternary(op));
retval->left = left;
retval->center = center;
retval->right = right;
return (Expression *) retval;
} // new_ternary_expr
static Expression *new_identifier_expr(Context *ctx, const char *string)
{
786
NEW_AST_NODE(retval, ExpressionIdentifier, AST_OP_IDENTIFIER);
787
788
789
790
791
792
retval->identifier = string; // cached; don't copy string.
return (Expression *) retval;
} // new_identifier_expr
static Expression *new_literal_int_expr(Context *ctx, const int64 value)
{
793
NEW_AST_NODE(retval, ExpressionIntLiteral, AST_OP_INT_LITERAL);
794
795
796
797
798
799
retval->value = value;
return (Expression *) retval;
} // new_literal_int_expr
static Expression *new_literal_float_expr(Context *ctx, const double dbl)
{
800
NEW_AST_NODE(retval, ExpressionFloatLiteral, AST_OP_FLOAT_LITERAL);
801
802
803
804
805
806
retval->value = dbl;
return (Expression *) retval;
} // new_literal_float_expr
static Expression *new_literal_string_expr(Context *ctx, const char *string)
{
807
NEW_AST_NODE(retval, ExpressionStringLiteral, AST_OP_STRING_LITERAL);
808
809
retval->string = string; // cached; don't copy string.
return (Expression *) retval;
810
811
812
813
814
815
816
817
} // new_literal_string_expr
static Expression *new_literal_boolean_expr(Context *ctx, const int value)
{
NEW_AST_NODE(retval, ExpressionBooleanLiteral, AST_OP_BOOLEAN_LITERAL);
retval->value = value;
return (Expression *) retval;
} // new_literal_boolean_expr
818
819
820
821
static void delete_expr(Context *ctx, Expression *expr)
{
DELETE_AST_NODE(expr);
822
if (operator_is_unary(expr->ast.type))
823
824
825
826
{
const ExpressionUnary *unary = (const ExpressionUnary *) expr;
delete_expr(ctx, unary->operand);
} // if
827
else if (operator_is_binary(expr->ast.type))
828
829
830
831
832
{
const ExpressionBinary *binary = (const ExpressionBinary *) expr;
delete_expr(ctx, binary->left);
delete_expr(ctx, binary->right);
} // else if
833
else if (operator_is_ternary(expr->ast.type))
834
835
836
837
838
839
{
const ExpressionTernary *ternary = (const ExpressionTernary *) expr;
delete_expr(ctx, ternary->left);
delete_expr(ctx, ternary->center);
delete_expr(ctx, ternary->right);
} // else if
840
else if (expr->ast.type == AST_OP_CAST)
841
842
843
{
delete_expr(ctx, ((ExpressionCast *) expr)->operand);
} // else if
844
else if (expr->ast.type == AST_OP_CONSTRUCTOR)
845
846
847
{
delete_expr(ctx, ((ExpressionConstructor *) expr)->args);
} // else if
848
849
850
// rest of operators don't have extra data to free.
851
852
853
854
855
856
857
858
859
860
861
Free(ctx, expr);
} // delete_expr
static FunctionArguments *new_function_arg(Context *ctx,
const InputModifier inputmod,
const char *datatype,
const char *identifier,
const char *semantic,
const InterpolationModifier interpmod,
Expression *initializer)
{
862
NEW_AST_NODE(retval, FunctionArguments, AST_FUNCTION_ARGS);
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
retval->input_modifier = inputmod;
retval->datatype = datatype;
retval->identifier = identifier;
retval->semantic = semantic;
retval->interpolation_modifier = interpmod;
retval->initializer = initializer;
retval->next = NULL;
return retval;
} // new_function_arg
static void delete_function_args(Context *ctx, FunctionArguments *args)
{
DELETE_AST_NODE(args);
delete_function_args(ctx, args->next);
delete_expr(ctx, args->initializer);
Free(ctx, args);
} // delete_function_args
static FunctionSignature *new_function_signature(Context *ctx,
const char *datatype,
const char *identifier,
FunctionArguments *args)
{
886
NEW_AST_NODE(retval, FunctionSignature, AST_FUNCTION_SIGNATURE);
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
retval->datatype = datatype;
retval->identifier = identifier;
retval->args = args;
retval->storage_class = FNSTORECLS_NONE;
retval->semantic = NULL;
return retval;
} // new_function_signature
static void delete_function_signature(Context *ctx, FunctionSignature *sig)
{
DELETE_AST_NODE(sig);
delete_function_args(ctx, sig->args);
Free(ctx, sig);
} // delete_function_signature
static CompilationUnit *new_function(Context *ctx,
FunctionSignature *declaration,
Statement *definition)
{
906
NEW_AST_NODE(retval, CompilationUnitFunction, AST_COMPUNIT_FUNCTION);
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
retval->next = NULL;
retval->declaration = declaration;
retval->definition = definition;
return (CompilationUnit *) retval;
} // new_function
static void delete_function(Context *ctx, CompilationUnitFunction *unitfn)
{
DELETE_AST_NODE(unitfn);
delete_compilation_unit(ctx, unitfn->next);
delete_function_signature(ctx, unitfn->declaration);
delete_statement(ctx, unitfn->definition);
Free(ctx, unitfn);
} // delete_function
static ScalarOrArray *new_scalar_or_array(Context *ctx, const char *ident,
const int isvec, Expression *dim)
{
925
NEW_AST_NODE(retval, ScalarOrArray, AST_SCALAR_OR_ARRAY);
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
retval->identifier = ident;
retval->isarray = isvec;
retval->dimension = dim;
return retval;
} // new_scalar_or_array
static void delete_scalar_or_array(Context *ctx, ScalarOrArray *soa)
{
DELETE_AST_NODE(soa);
delete_expr(ctx, soa->dimension);
Free(ctx, soa);
} // delete_scalar_or_array
static Typedef *new_typedef(Context *ctx, int isconst, const char *datatype,
ScalarOrArray *soa)
{
942
NEW_AST_NODE(retval, Typedef, AST_TYPEDEF);
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
retval->isconst = isconst;
retval->datatype = datatype;
retval->details = soa;
return retval;
} // new_typedef
static void delete_typedef(Context *ctx, Typedef *td)
{
DELETE_AST_NODE(td);
delete_scalar_or_array(ctx, td->details);
Free(ctx, td);
} // delete_typedef
static PackOffset *new_pack_offset(Context *ctx, const char *a, const char *b)
{
958
NEW_AST_NODE(retval, PackOffset, AST_PACK_OFFSET);
959
960
961
962
963
964
965
966
967
968
969
970
971
972
retval->ident1 = a;
retval->ident2 = b;
return retval;
} // new_pack_offset
static void delete_pack_offset(Context *ctx, PackOffset *o)
{
DELETE_AST_NODE(o);
Free(ctx, o);
} // delete_pack_offset
static VariableLowLevel *new_variable_lowlevel(Context *ctx, PackOffset *po,
const char *reg)
{
973
NEW_AST_NODE(retval, VariableLowLevel, AST_VARIABLE_LOWLEVEL);
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
retval->packoffset = po;
retval->register_name = reg;
return retval;
} // new_variable_lowlevel
static void delete_variable_lowlevel(Context *ctx, VariableLowLevel *vll)
{
DELETE_AST_NODE(vll);
delete_pack_offset(ctx, vll->packoffset);
Free(ctx, vll);
} // delete_variable_lowlevel
static Annotations *new_annotation(Context *ctx, const char *datatype,
Expression *initializer)
{
989
NEW_AST_NODE(retval, Annotations, AST_ANNOTATION);
990
991
992
993
994
995
996
997
998
999
1000
retval->datatype = datatype;
retval->initializer = initializer;
retval->next = NULL;
return retval;
} // new_annotation
static void delete_annotation(Context *ctx, Annotations *annotations)
{
DELETE_AST_NODE(annotations);
delete_annotation(ctx, annotations->next);
delete_expr(ctx, annotations->initializer);