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
|
#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_VARDECL, "variable decl") \
F(AST_VAR, "variable") \
F(AST_RETURN, "return") \
F(AST_FUNC, "func") \
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;
Variable **locals;
int num_locals;
int cur_stack_offset;
// TODO: Arguments / etc?
} func;
// Block of statements
struct {
Node **children;
int num_children;
} block;
struct {
Type type;
union {
int as_int;
};
} literal;
struct {
Variable var;
Node *value;
} var_decl;
struct {
Variable *var;
Node *value;
} assign;
Variable *variable;
};
} Node;
void print_ast(Node *node);
|