def p_statement_cond_postfix_assign(p):
'line_statement : if_assign IF condition_list ELSE expression SEMI'
debug("PSTFX IF-ELSE-ASSIGN", p[1:])
p[0] = mAST(action='assign',
params=[
p[1][0],
mAST(action='condition', params=[p[3], p[1][1], p[5]])
])
def p_statement_cond(p):
'''
line_statement : IF condition_list COLON statement SEMI %prec IFX
| IF condition_list COLON SEMI statement SEMI %prec IFX
'''
debug("IF", [str(x) for x in p[1:]])
if len(p) < 7:
p[0] = mAST(action='condition', params=[p[2], p[4]])
else:
p[0] = mAST(action='condition', params=[p[2], p[5]])
def p_statement_for_line(p):
'''
line_statement : FOR ID IN range COLON line_statement
| FOR ID IN range COLON for_line_stmt %prec LOOP_INSTR
'''
debug('FOR', p[1:])
if len(p) < 8:
p[0] = mAST(action='loop', params=[p[2], p[4], p[6]])
else:
p[0] = mAST(action='loop', params=[p[2], p[4], p[7]])
def p_statement_assign(p):
'''
line_statement : ID ASSIGN expression SEMI
| ID ASSIGN condition_list SEMI
'''
debug('ASSIGN', p[1], p[3])
p[0] = mAST(action='assign', params=[p[1], p[3]])
def p_condition_list(p):
'''
condition_list : expression %prec CONDLIST
| condition_list AND expression
| condition_list OR expression
'''
debug('CONDITION', p[1:])
if len(p) > 2:
p[0] = mAST(action='logop', params=p[1:])
else:
p[0] = p[1]
def p_expression_binop(p):
'''
expression : expression ADD expression
| expression REM expression
| expression MUL expression
| expression DIV expression
| expression MOD expression
| expression POW expression
| expression GT expression
| expression GE expression
| expression LT expression
| expression LE expression
| expression EQ expression
| expression NE expression
'''
debug('BINOP', p[1:])
p[0] = mAST(action='binop', params=p[1:])
def p_statement_cond_postfix_else(p):
'line_statement : statement IF condition_list ELSE statement SEMI'
debug("PSTFX IF-ELSE", p[1:])
p[0] = mAST(action='condition', params=[p[3], p[1], p[5]])
def p_statement_for(p):
'line_statement : FOR ID IN range COLON statement SEMI'
debug('FOR', p[1:])
p[0] = mAST(action='loop', params=[p[2], p[4], p[6]])
def p_statement_print(p):
'statement : PRINT LPAREN expr_list RPAREN'
debug('PRINT', p[3])
p[0] = mAST(action='print', params=p[3])
def p_expression_uminus(p):
'expression : REM expression %prec NEGATE'
debug('NEGATE', p[2])
p[0] = mAST(action='binop', params=[-1, '*', p[2]])
def p_expression_var(p):
'expression : ID'
debug('VAR', p[1])
p[0] = mAST(action='get', params=[p[1]])
