stage1 重构代码
This commit is contained in:
305
stage1/spl_ast.h
Normal file
305
stage1/spl_ast.h
Normal file
@@ -0,0 +1,305 @@
|
||||
#ifndef __SPL_AST_H__
|
||||
#define __SPL_AST_H__
|
||||
|
||||
#include "../stage0/include/utils.h"
|
||||
#include "spl_lexer.h"
|
||||
#include "spl_tok.h"
|
||||
#include <math.h>
|
||||
|
||||
typedef enum {
|
||||
SPL_AST_CONTAINER_MEMBER,
|
||||
SPL_AST_FN_DECL,
|
||||
SPL_AST_TYPE_DECL,
|
||||
SPL_AST_VAR_DECL,
|
||||
SPL_AST_CONST_DECL,
|
||||
SPL_AST_MEMBER_DECL,
|
||||
SPL_AST__COMPTIME_STMT, /*不实现*/
|
||||
SPL_AST__DIRECTIVE_BLOCK, /*不实现*/
|
||||
|
||||
SPL_AST_BLOCK,
|
||||
SPL_AST_BLOCK_ITEM,
|
||||
SPL_AST_EXPR,
|
||||
SPL_AST_TYPE_EXPR,
|
||||
SPL_AST_ATTR_LIST,
|
||||
} spl_ast_node_kind_t;
|
||||
|
||||
typedef struct {
|
||||
const char *fname;
|
||||
int line;
|
||||
int col;
|
||||
} spl_ast_loc_t;
|
||||
|
||||
struct spl_ast_node;
|
||||
typedef struct spl_ast_node spl_ast_node_t;
|
||||
typedef usize spl_ast_node_ref_t;
|
||||
typedef VEC(spl_ast_node_ref_t) spl_ast_node_ref_vec_t;
|
||||
struct spl_ast_node {
|
||||
spl_ast_node_kind_t kind;
|
||||
spl_ast_loc_t loc;
|
||||
union {
|
||||
spl_ast_node_ref_vec_t container_member;
|
||||
|
||||
spl_ast_node_ref_vec_t attr_list;
|
||||
|
||||
struct {
|
||||
spl_ast_node_ref_t attr_list;
|
||||
const char *name;
|
||||
spl_ast_node_ref_t param_list;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
spl_ast_node_ref_t block;
|
||||
} fn_decl;
|
||||
spl_ast_node_ref_vec_t param_list;
|
||||
struct {
|
||||
spl_ast_node_ref_t attr_list;
|
||||
const char *name;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
} param_decl;
|
||||
|
||||
struct {
|
||||
spl_ast_node_ref_t attr_list;
|
||||
const char *name;
|
||||
enum {
|
||||
SPL_AST_TYPE_STRUCT,
|
||||
SPL_AST_TYPE_UNION,
|
||||
SPL_AST_TYPE_ENUM,
|
||||
SPL_AST_TYPE_TYPE_EXPR,
|
||||
};
|
||||
union {
|
||||
spl_ast_node_ref_t type_expr;
|
||||
spl_ast_node_ref_t aggregate_list;
|
||||
};
|
||||
} type_decl;
|
||||
struct {
|
||||
spl_ast_node_ref_t attr_list;
|
||||
const char *name;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
} member_decl;
|
||||
|
||||
struct {
|
||||
spl_ast_node_ref_t attr_list;
|
||||
const char *name;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
spl_ast_node_ref_t expr;
|
||||
} var_decl;
|
||||
struct {
|
||||
spl_ast_node_ref_t attr_list;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
const char *name;
|
||||
spl_ast_node_ref_t expr;
|
||||
} const_decl;
|
||||
|
||||
spl_ast_node_ref_vec_t block;
|
||||
struct {
|
||||
enum {
|
||||
SPL_AST_IF_STATEMENT,
|
||||
SPL_AST_IFVAR_STATEMENT,
|
||||
SPL_AST_WHILE_STATEMENT,
|
||||
SPL_AST_LOOP_STATEMENT,
|
||||
SPL_AST_FOR_STATEMENT,
|
||||
SPL_AST_MATCH_STATEMENT,
|
||||
SPL_AST_RET_STATEMENT,
|
||||
SPL_AST_BREAK_STATEMENT,
|
||||
SPL_AST_CONTINUE_STATEMENT,
|
||||
SPL_AST_DEFER_STATEMENT,
|
||||
SPL_AST_VARDECL,
|
||||
SPL_AST_TYPEDECL,
|
||||
SPL_AST_EXPR_STATEMENT,
|
||||
SPL_PACED_EXPR,
|
||||
} kind;
|
||||
union {
|
||||
struct {
|
||||
spl_ast_node_ref_t expr;
|
||||
spl_ast_node_ref_t if_block;
|
||||
spl_ast_node_ref_t else_block;
|
||||
} if_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_t packed_expr;
|
||||
spl_ast_node_ref_t if_block;
|
||||
spl_ast_node_ref_t else_block;
|
||||
} ifvar_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_t expr;
|
||||
spl_ast_node_ref_t while_block;
|
||||
} while_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_t loop_block;
|
||||
} loop_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_vec_t expr_vec;
|
||||
VEC(char *) ident_vec;
|
||||
spl_ast_node_ref_t block;
|
||||
} for_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_t expr;
|
||||
spl_ast_node_ref_vec_t paced_exprs;
|
||||
spl_ast_node_ref_vec_t statements;
|
||||
} match_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_t expr;
|
||||
} ret_statement;
|
||||
struct {
|
||||
} break_statement;
|
||||
struct {
|
||||
} continue_statement;
|
||||
struct {
|
||||
spl_ast_node_ref_t block_or_statement;
|
||||
} defer_statement;
|
||||
spl_ast_node_ref_t var_decl;
|
||||
spl_ast_node_ref_t type_decl;
|
||||
spl_ast_node_ref_t expr_statement;
|
||||
|
||||
struct {
|
||||
const char *ident;
|
||||
const char *bind_ident;
|
||||
spl_ast_node_ref_t expr;
|
||||
} packed_expr;
|
||||
};
|
||||
} block_item;
|
||||
|
||||
struct {
|
||||
enum {
|
||||
SPL_AST_ASSIGN_EXPR,
|
||||
SPL_AST_BOOLOR_EXPR,
|
||||
SPL_AST_BOOLAND_EXPR,
|
||||
SPL_AST_BITOR_EXPR,
|
||||
SPL_AST_BITXOR_EXPR,
|
||||
SPL_AST_BITAND_EXPR,
|
||||
SPL_AST_CMPEQ_EXPR,
|
||||
SPL_AST_CMP_EXPR,
|
||||
SPL_AST_RANGE_EXPR,
|
||||
SPL_AST_SHIFT_EXPR,
|
||||
SPL_AST_ADD_EXPR,
|
||||
SPL_AST_MUL_EXPR,
|
||||
SPL_AST_PREFIX_EXPR,
|
||||
SPL_AST_POSTFIX_EXPR, /*left ref node*/
|
||||
SPL_AST_PRIMARY_EXPR, /*left ref node*/
|
||||
} op;
|
||||
struct {
|
||||
spl_ast_node_ref_t left;
|
||||
spl_ast_node_ref_t right;
|
||||
} op_expr;
|
||||
} expr;
|
||||
struct {
|
||||
enum {
|
||||
SPL_AST_CALL_EXPR,
|
||||
SPL_AST_IDENT_EXPR,
|
||||
SPL_AST_DEREF_EXPR,
|
||||
SPL_AST_INDEX_EXPR,
|
||||
SPL_AST_SLICE_EXPR,
|
||||
SPL_AST_AS_EXPR,
|
||||
} kind;
|
||||
union {
|
||||
spl_ast_node_ref_vec_t call_expr;
|
||||
const char *indet_expr;
|
||||
spl_ast_node_ref_t index_expr;
|
||||
struct {
|
||||
spl_ast_node_ref_t begin;
|
||||
spl_ast_node_ref_t end;
|
||||
} slice_expr;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
};
|
||||
} postfix_expr;
|
||||
struct {
|
||||
enum {
|
||||
SPL_AST_INTEGER,
|
||||
SPL_AST_FLOAT,
|
||||
SPL_AST_CHAR_LIT,
|
||||
SPL_AST_STRING_LIT,
|
||||
SPL_AST_TRUE,
|
||||
SPL_AST_FALSE,
|
||||
SPL_AST_NULL,
|
||||
SPL_AST_IDENT,
|
||||
SPL_AST_ARGGREGATE_INIT,
|
||||
SPL_AST_EXPR_EXPR,
|
||||
SPL_AST_ARRAY_LIT,
|
||||
SPL_AST_BUILTIN_EXPR,
|
||||
SPL_AST_BLOCK_EXPR,
|
||||
};
|
||||
union {
|
||||
isize integer_expr;
|
||||
double float_expr;
|
||||
char char_lit_expr;
|
||||
const char *string_lit_expr;
|
||||
const char *ident;
|
||||
|
||||
/*aggregate_init_item*/
|
||||
spl_ast_node_ref_vec_t aggregate_init_expr;
|
||||
|
||||
spl_ast_node_ref_t expr;
|
||||
struct {
|
||||
isize integer;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
spl_ast_node_ref_vec_t expr_list;
|
||||
} array_lit_expr;
|
||||
struct {
|
||||
const char *ident;
|
||||
spl_ast_node_ref_vec_t expr_list;
|
||||
} builtin_expr;
|
||||
spl_ast_node_ref_t block_expr;
|
||||
};
|
||||
} primary_expr;
|
||||
struct {
|
||||
const char *ident;
|
||||
spl_ast_node_ref_t expr;
|
||||
} aggregate_init_item;
|
||||
|
||||
struct {
|
||||
/* ASTERISK = 1, [] = 2, else = 0*/
|
||||
int pointer;
|
||||
/* if array_size != 0 then pointer == 2 */
|
||||
int array_size;
|
||||
enum {
|
||||
SPL_AST_BASE_TYPE_FN,
|
||||
SPL_AST_BASE_TYPE_PATH,
|
||||
} kind;
|
||||
|
||||
const char *spl_base_type;
|
||||
union {
|
||||
spl_ast_node_ref_vec_t type_path;
|
||||
struct {
|
||||
spl_ast_node_ref_t param_list;
|
||||
spl_ast_node_ref_t type_expr;
|
||||
} fn_type;
|
||||
};
|
||||
} type_expr;
|
||||
struct {
|
||||
enum {
|
||||
SPL_AST_TYPE_VOID,
|
||||
SPL_AST_TYPE_BOOL,
|
||||
SPL_AST_TYPE_I8,
|
||||
SPL_AST_TYPE_U8,
|
||||
SPL_AST_TYPE_I16,
|
||||
SPL_AST_TYPE_U16,
|
||||
SPL_AST_TYPE_I32,
|
||||
SPL_AST_TYPE_U32,
|
||||
SPL_AST_TYPE_I64,
|
||||
SPL_AST_TYPE_U64,
|
||||
SPL_AST_TYPE_ISIZE,
|
||||
SPL_AST_TYPE_USIZE,
|
||||
SPL_AST_TYPE__F32, /*不实现*/
|
||||
SPL_AST_TYPE__F64, /*不实现*/
|
||||
SPL_AST_TYPE_PTR,
|
||||
SPL_AST_TYPE_ANY,
|
||||
SPL_AST_TYPE_IDENT,
|
||||
} kind;
|
||||
const char *ident;
|
||||
} type_atom;
|
||||
};
|
||||
};
|
||||
|
||||
typedef VEC(spl_ast_node_t) spl_ast_node_vec_t;
|
||||
typedef struct {
|
||||
int parsed;
|
||||
spl_tok_vec_t input;
|
||||
spl_ast_node_vec_t buckets;
|
||||
spl_ast_node_ref_t root;
|
||||
} spl_ast_t;
|
||||
|
||||
void spl_ast_init(spl_ast_t *ast, const spl_tok_vec_t *tok_vec /*move*/);
|
||||
void spl_ast_drop(spl_ast_t *ast);
|
||||
|
||||
void spl_ast_prase(spl_ast_t *ast);
|
||||
void spl_ast_valid(spl_ast_t *ast);
|
||||
void spl_ast_dump(spl_ast_t *ast, spl_ast_node_ref_t node);
|
||||
|
||||
#endif /* __SPL_AST_H__ */
|
||||
Reference in New Issue
Block a user