grammar T;

options {
        //language=CSharp;
        output=AST;
        ASTLabelType=CommonTree;
}

tokens {
    EQ;
}

@header {
        #pragma warning disable 0219
}

file
    : block EOF!
    ;

statement
    : assignment
    | loop
    | conditional
    | block
    | write
    ;

assignment
    : ID '='^ expr
    ;

loop
    : 'while'^ condition 'do'! statement
    ;

conditional
    : 'if'^ condition 'then'! statement
    ;

block
        : 'begin'^ statement* 'end'!
        ;

write
        : 'write'^ expr
        ;

expr
        : (condition) => condition '?'^ expr ':'! expr
        | simpleExpr
        ;

simpleExpr
        : term (('+' | '-')^ term)*
        ;
    
term
        : factor (('*' | '/' | '%')^ factor)*
        ;
      
factor
    : '('! expr ')'!
        | ID ('['^ expr ']'!)*
        | NUMBER
        ;

condition
        : expr '<'^ expr
        | expr '>'^ expr
        | expr '<='^ expr
        | expr '>='^ expr
        | expr '<>'^ expr
        | e1=expr '=' e2=expr -> ^(EQ $e1 $e2)
        ;

NUMBER
        : ('0'..'9')+
        ;

ID
        : ('a'..'z')+
        ;

WS : (' '|'\r'|'\t'|'\f'|'\n')+ { $channel=HIDDEN; };