1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
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
|
#pragma once
#include "common.h"
#define ENUM_AST_TYPES(F) \
F(OP_NEG, "neg") \
F(OP_NOT, "!") \
F(OP_BWINV, "~") \
F(OP_PLUS, "+") \
F(OP_MINUS, "-") \
F(OP_MUL, "*") \
F(OP_DIV, "/") \
F(OP_MOD, "%") \
F(OP_LSHIFT, "<<") \
F(OP_RSHIFT, ">>") \
F(OP_AND, "&&") \
F(OP_OR, "||") \
F(OP_XOR, "^") \
F(OP_EQ, "==") \
F(OP_NEQ, "!=") \
F(OP_LT, "<") \
F(OP_LEQ, "<=") \
F(OP_GT, ">") \
F(OP_GEQ, ">=") \
F(OP_ASSIGN, "=") \
F(AST_LITERAL, "literal") \
F(AST_FUNCCALL, "Function call") \
F(AST_CONDITIONAL, "conditional expression") \
F(AST_IF, "if statement") \
F(AST_WHILE, "while statement") \
F(AST_DEFER, "defer statement") \
F(AST_FOR, "for statement") \
F(AST_VARDECL, "variable decl") \
F(AST_LOCAL_VAR, "local variable") \
F(AST_GLOBAL_VAR, "global variable") \
F(AST_RETURN, "return") \
F(AST_FUNC, "func") \
F(AST_BUILTIN, "builtin") \
F(AST_PROGRAM, "program") \
F(AST_BLOCK, "block statements")
typedef enum {
#define DEFINE_ENUM(name, str) name,
ENUM_AST_TYPES(DEFINE_ENUM)
#undef DEFINE_ENUM
} NodeType;
char *node_type_to_str(NodeType type);
bool is_binary_op(NodeType type);
bool is_unary_op(NodeType type);
bool is_expression(NodeType type);
typedef enum {
TYPE_NONE,
TYPE_INT,
} DataType;
char *data_type_to_str(DataType type);
typedef struct {
DataType type;
// 0 = value, 1 = pointer, 2 = double pointer, ...
int indirection;
} Type;
typedef struct {
char *name;
Type type;
i64 offset;
} Variable;
typedef struct ast_node Node;
typedef struct ast_node {
NodeType type;
union {
// Binary expr
struct {
Node *left;
Node *right;
} binary;
// Unary expr
Node *unary_expr;
// Function definition
struct {
char *name;
Type return_type;
Node *body;
// TODO: Should we just dynamically allocate space on the
// stack for each block instead of storing this?
i64 max_locals_size;
// TODO: Arguments / etc?
Variable *args;
int num_args;
} func;
// Block of statements
struct {
Node **children;
int num_children;
Variable **locals;
int num_locals;
i64 locals_size;
} block;
struct {
Type type;
union {
int as_int;
};
} literal;
struct {
Variable var;
Node *value;
} var_decl;
struct {
Node *var;
Node *value;
} assign;
struct {
Node *cond;
Node *do_then;
Node *do_else;
} conditional;
// Used for all loops
struct {
Node *cond;
Node *init;
Node *step;
Node *body;
} loop;
Variable *variable;
struct {
Node *func;
Node **args;
int num_args;
} call;
};
} Node;
void print_ast(Node *node);
|