Tothic/ast_tree.py at main · PabloPavan/Tothic · 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
from dataclasses import dataclass
from typing import List, Optional, Union, Tuple

class Node: pass

@dataclass
class Literal(Node):
    value: Union[int, float]
    type: str

@dataclass
class VarRef(Node):
    name: str

@dataclass
class BinaryOp(Node):
    op: str
    left: Node
    right: Node

@dataclass
class TypeCast(Node):
    target_type: str
    expr: Node

@dataclass
class Assign(Node):
    name: Node
    expr: Node

@dataclass
class Decl(Node):
    name: str
    type: str

@dataclass
class AutoDecl(Node):
    name: str
    expr: Node

@dataclass
class If(Node):
    condition: Node
    then_branch: Node
    else_branch: Optional[Node] = None

@dataclass
class Block(Node):
    statements: List[Node]

@dataclass
class NamespaceDecl(Node):
    name: str
    declarations: List[Node]

@dataclass
class ArrayDecl(Node):
    name: str
    type: str
    size: Node

@dataclass
class ArrayAccess(Node):
    name: str
    index: Node

@dataclass
class QualifiedRef(Node):
    namespace: str
    name: str

@dataclass
class ExprStmt(Node):
    expr: Node

@dataclass
class FunctionDecl(Node):
    name: str
    params: List[Tuple[str, str]]  # (nome, tipo)
    return_type: str
    body: Block

@dataclass
class Call(Node):
    name: str
    args: List[Node]

@dataclass
class Print(Node):
    name: str
    args: List[Node]

@dataclass
class Halt(Node):
    name: str
    args: List[Node]

@dataclass
class Return(Node):
    expr: Node

@dataclass
class Program(Node):
    statements: List[Node]