AMMScript/parser.py at master · ShatterPlayer/AMMScript · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
import ply.yacc as yacc
from lexer import *

variables = {}
def p_statement_expr(p):
    '''
    statement : expression SEMICOLON
    '''
    p[0] = p[1]

def p_expression_binop(p):
    '''
    expression : expression PLUS expression
               | expression MINUS expression
               | expression MULTIPLY expression
               | expression DIVIDE expression
               | expression POWER expression
               | expression MODULO expression
               | expression EQUAL_EQUAL expression
               | expression NOT_EQUAL expression
               | expression LESS expression
               | expression GREATER expression
               | expression LESS_EQUAL expression
               | expression GREATER_EQUAL expression
               | expression EQUAL expression
               | expression PLUS_EQUAL expression
               | expression MINUS_EQUAL expression
               | expression MULTIPLY_EQUAL expression
               | expression DIVIDE_EQUAL expression
    '''
    if p[2] == '+':
        p[0] = p[1] + p[3]
    elif p[2] == '-':
        p[0] = p[1] - p[3]
    elif p[2] == '*':
        p[0] = p[1] * p[3]
    elif p[2] == '/':
        p[0] = p[1] / p[3]
    elif p[2] == '^':
        p[0] = p[1] ** p[3]
    elif p[2] == '%':
        p[0] = p[1] % p[3]
    elif p[2] == '==':
        p[0] = p[1] == p[3]
    elif p[2] == '!=':
        p[0] = p[1] != p[3]
    elif p[2] == '<':
        p[0] = p[1] < p[3]
    elif p[2] == '>':
        p[0] = p[1] > p[3]
    elif p[2] == '<=':
        p[0] = p[1] <= p[3]
    elif p[2] == '>=':
        p[0] = p[1] >= p[3]
    elif p[2] == '=':
        p[0] = p[3]
    elif p[2] == '+=':
        p[0] = p[1] + p[3]
    elif p[2] == '-=':
        p[0] = p[1] - p[3]
    elif p[2] == '*=':
        p[0] = p[1] * p[3]
    elif p[2] == '/=':
        p[0] = p[1] / p[3]

def p_statement_assign(p):
    '''
    statement : ID ASSIGN expression SEMICOLON
    '''
    variables[p[1]] = p[3]

def p_expression_variable(p):
    'expression : ID'
    try:
        p[0] = variables[p[1]]
    except LookupError:
        print(f"Undefined variable '{p[1]}'")
        p[0] = 0


def p_start(p):
    '''
    start : statement
          | expression
    '''
    print(p[1])
    p[0] = p[1]

def p_expression_number(p):
    'expression : NUMBER'
    p[0] = p[1]

def p_expression_unary(p):
    '''
    expression : MINUS_MINUS expression
               | PLUS_PLUS expression
    '''
    if p[1] == '--':
        p[0] = p[2] - 1
    elif p[1] == '++':
        p[0] = p[2] + 1

def p_expression_print(p):
    'expression : PRINT expression'
    print(p[2])
    p[0] = p[2]

def p_error(p):
    print("Błąd składni!")

def p_if_statement(p):
    '''statement : IF expression LBRACE statements RBRACE
                 | IF expression LBRACE statements RBRACE ELSE LBRACE statements RBRACE'''
    if p[2]:
        p[0] = p[4]
    elif len(p) > 6:
        p[0] = p[8]

def p_statements(p):
    '''statements : statement
                  | statements statement'''
    if len(p) == 2:
        p[0] = [p[1]]
    else:
        p[1].append(p[2])
        p[0] = p[1]


parser = yacc.yacc()