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
|
/*
* Copyright (C) 2023 Johnny Richard
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#ifndef AST_H
#define AST_H
#include <stdint.h>
#include "string_view.h"
#include "vector.h"
#define ast_visitor_visit(visitor, node) ast_node_accept_visitor(node, (ast_visitor_t *) visitor);
typedef enum {
TYPE_I32
} type_t;
typedef struct ast_visitor_t ast_visitor_t;
typedef struct ast_node_t ast_node_t;
typedef struct ast_return_stmt_t {
ast_node_t* argument;
} ast_return_stmt_t;
typedef struct ast_function_declaration_t {
string_view_t name;
type_t return_type;
vector_t* body;
} ast_function_declaration_t;
typedef struct ast_binary_operation_t {
// FIXME: We want to use enum to distinguish operators
string_view_t op;
ast_node_t* left;
ast_node_t* right;
} ast_binary_operation_t;
typedef enum {
AST_LITERAL_INTEGER
} ast_literal_kind_t;
typedef union {
uint32_t integer;
} ast_literal_value_t;
typedef struct ast_literal_t {
ast_literal_kind_t kind;
ast_literal_value_t value;
} ast_literal_t;
typedef struct ast_identifier_t {
string_view_t name;
} ast_identifier_t;
typedef struct ast_variable_declaration_t {
string_view_t name;
type_t type;
ast_node_t* value;
} ast_variable_declaration_t;
typedef struct ast_visitor_t {
void (*visit_function)(struct ast_visitor_t *, ast_function_declaration_t *);
void (*visit_return_stmt)(struct ast_visitor_t *, ast_return_stmt_t *);
void (*visit_literal)(struct ast_visitor_t *, ast_literal_t *);
void (*visit_binary_operation)(struct ast_visitor_t *, ast_binary_operation_t *);
} ast_visitor_t;
typedef enum {
AST_BINARY_OPERATION,
AST_FUNCTION_DECLARATION,
AST_IDENTIFIER,
AST_LITERAL,
AST_RETURN_STMT,
AST_UNKOWN_NODE,
AST_VARIABLE_DECLARATION,
} ast_node_kind_t;
typedef union {
ast_binary_operation_t binary_operation;
ast_function_declaration_t function;
ast_literal_t literal;
ast_return_stmt_t return_stmt;
ast_variable_declaration_t variable;
ast_identifier_t identifier;
} ast_node_data_t;
typedef struct ast_node_t {
void (*accept_visitor)(ast_node_t *, ast_visitor_t *);
ast_node_kind_t kind;
ast_node_data_t data;
} ast_node_t;
void ast_node_accept_visitor(ast_node_t *node, ast_visitor_t *visitor);
ast_node_t* ast_node_new();
void ast_node_destroy(ast_node_t *node);
void ast_node_init_binary_operation(ast_node_t *node, string_view_t op, ast_node_t *left, ast_node_t *right);
void ast_node_init_function_declaration(ast_node_t *node, string_view_t name, type_t return_type, vector_t *body);
void ast_node_init_identifier(ast_node_t *node, string_view_t name);
void ast_node_init_return_stmt(ast_node_t *node, ast_node_t *argument);
void ast_node_init_variable_declaration(ast_node_t *node, string_view_t variable_name, type_t type, ast_node_t *value);
// FIXME: use the naming convention
void ast_literal_integer_create(ast_node_t* node, uint32_t number);
#endif /* AST_H */
|
