syntax.y

%locations

%{
    int yycolumn = 1;
    #define YY_USER_ACTION \
                yylloc.first_line = yylloc.last_line = yylineno; \
                yylloc.first_column = yycolumn; \
                yylloc.last_column = yycolumn + yyleng - 1; \
                yycolumn += yyleng;
%}

%{
    #include <stdio.h>
    #include "lex.yy.c"
    #include "data.h"
    #include <stdarg.h>

    #define TERMINALS(A,B) A=tree->Node_alloc(B,yylloc.first_line)

    void yyerror(char* s);
    static Node_t * Operator(Node_t * cur,char * content,int line,int argc,...);
                
%}

/* declared types */
%union {
    struct __Tree_node_t * node;
}

/* declared tokens */
%token <node> INT
%token <node> FLOAT
%token <node> ID
%token <node> SEMI
%token <node> COMMA
%token <node> ASSIGNOP
%token <node> RELOP
%token <node> PLUS
%token <node> MINUS
%token <node> STAR
%token <node> DIV
%token <node> AND
%token <node> OR
%token <node> DOT
%token <node> NOT
%token <node> TYPE
/* TYPE -> int | float见下面 */
%token <node> LP
%token <node> RP
%token <node> LB
%token <node> RB
%token <node> LC
%token <node> RC
%token <node> STRUCT
%token <node> RETURN
%token <node> IF
%token <node> ELSE
%token <node> WHILE


%right ASSIGNOP
%left OR
%left AND
%left RELOP
%left PLUS
/* %left MINUS  取负在前，left可以考虑不写（推断） */
%left STAR
%left DIV
%right NOT
%right MINUS
%left DOT LB RB LP RP


%type <node> Program
%type <node> ExtDefList
%type <node> ExtDef
%type <node> ExtDecList
%type <node> Specifier
%type <node> StructSpecifier
%type <node> OptTag
%type <node> Tag
%type <node> VarDec
%type <node> FunDec
%type <node> VarList
%type <node> ParamDec
%type <node> CompSt
%type <node> StmtList
%type <node> Stmt
%type <node> DefList
%type <node> Def
%type <node> DecList
%type <node> Dec
%type <node> Exp
%type <node> Args


%nonassoc LOWER_THAN_ELSE
%nonassoc ELSE



%%
/*High-level Definitions */

Program : ExtDefList                { tree->root = Operator($$,"Program",@$.first_line,1,$1); }
    ;
ExtDefList : ExtDef ExtDefList      { Operator($$,"ExtDefList",@$.first_line,2,$1,$2); }
    | 
    ;
ExtDef : Specifier ExtDecList SEMI  { Operator($$,"ExtDef",@$.first_line,3,$1,$2,TERMINALS($3,"SEMI")); }
    | Specifier SEMI                { Operator($$,"ExtDef",@$.first_line,2,$1,TERMINALS($2,"SEMI")); }
    | Specifier FunDec CompSt       { Operator($$,"ExtDef",@$.first_line,3,$1,$2,$3); }
    ;
ExtDecList : VarDec                 { Operator($$,"ExtDecList",@$.first_line,1,$1); }
    | VarDec COMMA ExtDecList       { Operator($$,"ExtDecList",@$.first_line,3,$1,TERMINALS($2,"COMMA"),$3); }
    ;

/*Specifier */

Specifier : TYPE                    { Operator($$,"Specifier",@$.first_line,1,TERMINALS($1,"TYPE")); }
    | StructSpecifier               { Operator($$,"Specifier",@$.first_line,1,$1); }
    ;
StructSpecifier : STRUCT OptTag LC DefList RC   { Operator($$,"StructSpecifier",@$.first_line,5,TERMINALS($1,"STRUCT"),$2,TERMINALS($3,"LC"),$4,TERMINALS($5,"RC")); }
    | STRUCT Tag                    { Operator($$,"StructSpecifier",@$.first_line,2,TERMINALS($1,"STRUCT"),$2); }
    ;
OptTag : ID                         { Operator($$,"OptTag",@$.first_line,1,TERMINALS($1,"ID")); }
    | 
    ;
Tag : ID                            { Operator($$,"Tag",@$.first_line,1,TERMINALS($1,"ID")); }

/*Declarators */

VarDec : ID                         { Operator($$,"VarDec",@$.first_line,1,TERMINALS($1,"ID")); }
    | VarDec LB INT RB              { Operator($$,"VarDec",@$.first_line,4,$1,TERMINALS($2,"LB"),TERMINALS($3,"INT"),TERMINALS($4,"RB")); }
    ;
FunDec : ID LP VarList RP           { Operator($$,"FunDec",@$.first_line,4,TERMINALS($1,"ID"),TERMINALS($2,"LP"),$3,TERMINALS($4,"RP")); }
    | ID LP RP                      { Operator($$,"FunDec",@$.first_line,3,TERMINALS($1,"ID"),TERMINALS($2,"LP"),TERMINALS($3,"RP")); }
    ;
VarList : ParamDec COMMA VarList    { Operator($$,"VarList",@$.first_line,3,$1,TERMINALS($2,"COMMA"),$3); }
    | ParamDec                      { Operator($$,"ParamDec",@$.first_line,1,$1); }
    ;
ParamDec : Specifier VarDec         { Operator($$,"ParamDec",@$.first_line,2,$1,$2); }
    ;

/*Statements */

CompSt : LC DefList StmtList RC     { Operator($$,"CompSt",@$.first_line,1,TERMINALS($1,"LC"),$2,$3,TERMINALS($4,"RC")); }
    | error RC                      
    ;
StmtList : Stmt StmtList            { Operator($$,"StmtList",@$.first_line,2,$1,$2); }
    | 
    ;
Stmt : Exp SEMI                     { Operator($$,"Stmt",@$.first_line,2,$1,TERMINALS($2,"SEMI")); }
    | CompSt                        { Operator($$,"Stmt",@$.first_line,1,$1); }
    | RETURN Exp SEMI               { Operator($$,"Stmt",@$.first_line,3,TERMINALS($1,"RETURN"),$2,TERMINALS($3,"SEMI")); }
    | IF LP Exp RP Stmt %prec LOWER_THAN_ELSE   { Operator($$,"Stmt",@$.first_line,5,TERMINALS($1,"IF"),TERMINALS($2,"LP"),$3,TERMINALS($4,"RP"),$5); }
    | IF LP Exp RP Stmt ELSE Stmt   { Operator($$,"Stmt",@$.first_line,7,TERMINALS($1,"IF"),TERMINALS($2,"LP"),$3,TERMINALS($4,"RP"),$5,TERMINALS($6,"ELSE"),$7); }
    | WHILE LP Exp RP Stmt          { Operator($$,"Stmt",@$.first_line,5,TERMINALS($1,"WHILE"),TERMINALS($2,"LP"),$3,TERMINALS($3,"RP"),$4); }
    | error SEMI                    
    ;

/*Local Definitions */

DefList : Def DefList               { Operator($$,"DefList",@$.first_line,2,$1,$2); }
    | 
    ;
Def : Specifier DecList SEMI        { Operator($$,"Def",@$.first_line,3,$1,$2,TERMINALS($3,"SEMI")); }
    ;
DecList : Dec                       { Operator($$,"DecList",@$.first_line,1,$1); }
    | Dec COMMA DecList             { Operator($$,"DecList",@$.first_line,3,$1,TERMINALS($2,"COMMA"),$3); }
    ;
Dec : VarDec                        { Operator($$,"Dec",@$.first_line,1,$1); }
    | VarDec ASSIGNOP Exp           { Operator($$,"Dec",@$.first_line,3,$1,TERMINALS($2,"ASSIGNOP"),$3); }
    ;

/*Expressions */

Exp : Exp ASSIGNOP Exp              { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"ASSIGNOP"),$3); }
    | Exp AND Exp                   { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"AND"),$3); }
    | Exp OR Exp                    { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"OR"),$3); }
    | Exp RELOP Exp                 { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"RELOP"),$3); }
    | Exp PLUS Exp                  { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"PLUS"),$3); }
    | Exp MINUS Exp                 { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"MINUS"),$3); }
    | Exp STAR Exp                  { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"STAR"),$3); }
    | Exp DIV Exp                   { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"DIV"),$3); }
    | LP Exp RP                     { Operator($$,"Exp",@$.first_line,3,TERMINALS($1,"LP"),$2,TERMINALS($3,"RP")); }
    | MINUS Exp                     { Operator($$,"Exp",@$.first_line,2,TERMINALS($1,"MINUS"),$2); }
    | NOT Exp                       { Operator($$,"Exp",@$.first_line,2,TERMINALS($1,"NOT"),$2); }
    | ID LP Args RP                 { Operator($$,"Exp",@$.first_line,4,TERMINALS($1,"ID"),TERMINALS($1,"LP"),$3,TERMINALS($4,"RP")); }
    | ID LP RP                      { Operator($$,"Exp",@$.first_line,3,TERMINALS($1,"ID"),TERMINALS($2,"LP"),TERMINALS($3,"RP")); }
    | Exp LB Exp RB                 { Operator($$,"Exp",@$.first_line,4,$1,TERMINALS($2,"LB"),$3,TERMINALS($4,"RB")); }
    | Exp DOT ID                    { Operator($$,"Exp",@$.first_line,3,$1,TERMINALS($2,"DOT"),TERMINALS($3,"ID")); }
    | ID                            { Operator($$,"Exp",@$.first_line,1,TERMINALS($1,"ID")); }
    | INT                           { Operator($$,"Exp",@$.first_line,1,TERMINALS($1,"INT")); }
    | FLOAT                         { Operator($$,"Exp",@$.first_line,1,TERMINALS($1,"FLOAT")); }
    | Exp PLUS error                
    ;
Args : Exp COMMA Args               { Operator($$,"Args",@$.first_line,3,$1,TERMINALS($2,"COMMA"),$3); }
    | Exp                           { Operator($$,"Args",@$.first_line,1,$1); }
    | error Args                    
    ;

%%

static Node_t * Operator(Node_t * cur,char * content,int line,int argc,...) {
    cur = tree->Node_alloc(content,line);
    Log("%s %d %d",content,line,argc);
    va_list ap;
    va_start(ap,argc);
    for (int i = 0;i < argc;i++) {
        Node_t * temp = (Node_t*)va_arg(ap,Node_t*);
        Log("%s",temp->content);
        tree->rminsert(cur,temp);
    }
    va_end(ap);
    printf("%s ",cur->content);
    PRINT_TREE(cur);
    return cur;
}