path: root/ast.h

#ifndef AST_H
#define AST_H
#include <stdarg.h>
#include "cibic.tab.h"

#define EXP_POSTFIX     1024
#define POSTFIX_ARR     1025
#define POSTFIX_CALL    1026
#define POSTFIX_DOT     1027
#define POSTFIX_PTR     1028
#define EXP_CAST        1029
#define INITR_NORM      1030
#define INITR_ARR       1031
#define DECLR_FUNC      1032
#define DECLR_ARR       1033
#define STMT_EXP        1034
#define STMT_COMP       1035
#define STMT_IF         1036
#define STMT_WHILE      1037
#define STMT_FOR        1038
#define STMT_CONT       1039
#define STMT_BREAK      1040
#define STMT_RET        1041

#define MAX_CHDN        1024

typedef struct CNode {
    enum {
        /* Top Level */
        PROG,
        FUNC_DEF,
        DECL, /* declaration */
        DECLR, /* declarator */
        DECLRS,
        INIT_DECLR, 
        INIT_DECLRS,
        INITR, /* initializer */
        TYPE_SPEC,
        FIELD, /* struct-or-union field */
        PLAIN_DECL,
        DECLS,
        FUNCS,

        /* Statments */
        STMT,

        /* Expressions  */
        EXP,
        TYPE_NAME,
        ID, /* identifier */
        INT, /* INT_CONST */
        CHAR,
        STR,
        NOP,

        COMP_STMTS,
        COMP_DECLS,
        ARGS,
        PARAMS
    } type;
    union {
        int intval;
        int subtype;
        char *strval;
    } rec;
    struct CNode *chd, *next;
    /* For error reporting */
    struct Location {
        int row, col;
    } loc;
} CNode;

CNode *cnode_add_loc(CNode *node, YYLTYPE loc);
CNode *cnode_create_ast(CNode *wrapped);
CNode *cnode_create_nop();
CNode *cnode_create_general(int type, int subtype, int pnum, va_list ap); 
CNode *cnode_list_append(CNode *list, CNode *tail);
CNode *cnode_list_wrap(int type, CNode *list);

CNode *cnode_create_exp(int exp_type, int pnum, ...);
CNode *cnode_create_type_spec(int spec_type, int pnum, ...);
CNode *cnode_create_declr(int declr_type, int pnum, ...);
CNode *cnode_create_stmt(int stmt_type, int pnum, ...);
CNode *cnode_create_initr(int initr_type, CNode *body);

CNode *cnode_create_decl(CNode *type, CNode *init_declrs);
CNode *cnode_create_func(CNode *type, CNode *plain_decl, CNode *params, CNode *stmt);
CNode *cnode_create_init_declr(CNode *declr, CNode *initr);
CNode *cnode_create_struct_field(CNode *type_spec, CNode *declrs);
CNode *cnode_create_plain_decl(CNode *type_spec, CNode *declr);

CNode *cnode_create_identifier(char *val);
CNode *cnode_create_int_const(int val);
CNode *cnode_create_char_const(char *val);
CNode *cnode_create_str_const(char *val);

void cnode_debug_print(CNode *ast, int fancy);

extern CNode *ast_root;
extern CNode *null;

#endif