Create evaluators
This commit is contained in:
62
AST.py
62
AST.py
@@ -1,20 +1,5 @@
|
||||
from enum import Enum
|
||||
|
||||
class NodeType(Enum):
|
||||
INTEGER = 1
|
||||
STRING = 2
|
||||
NOTE = 3
|
||||
BLOCK = 4
|
||||
ARGUMENTS = 5
|
||||
IDENTIFIER = 6
|
||||
ASSIGN = 7
|
||||
PROGRAM = 8
|
||||
ASTERISK = 9
|
||||
COLON = 10
|
||||
FUNCTION_CALL = 11
|
||||
COMMA = 12
|
||||
PERCENT = 13
|
||||
|
||||
class Node:
|
||||
def __init__(self):
|
||||
self.children = []
|
||||
@@ -27,7 +12,7 @@ class Node:
|
||||
|
||||
def __getitem__(self, index):
|
||||
return self.children[index]
|
||||
|
||||
|
||||
def append(self, node):
|
||||
self.children.append(node)
|
||||
|
||||
@@ -36,8 +21,7 @@ class Node:
|
||||
|
||||
class Program(Node):
|
||||
def __init__(self):
|
||||
Node.__init__(self)
|
||||
self.type = NodeType.PROGRAM
|
||||
Node.__init__(self)
|
||||
|
||||
def __str__(self):
|
||||
return "Program:\n" + "\n".join([str(e) for e in self.children])
|
||||
@@ -45,32 +29,28 @@ class Program(Node):
|
||||
|
||||
class BlockNode(Node):
|
||||
def __init__(self):
|
||||
Node.__init__(self)
|
||||
self.type = NodeType.BLOCK
|
||||
Node.__init__(self)
|
||||
|
||||
def __str__(self):
|
||||
return "B{\n" + "\n".join([str(e) for e in self.children]) + "\n}"
|
||||
|
||||
|
||||
class ArgumentsNode(Node):
|
||||
class ListNode(Node):
|
||||
def __init__(self):
|
||||
Node.__init__(self)
|
||||
self.type = NodeType.ARGUMENTS
|
||||
Node.__init__(self)
|
||||
|
||||
def __str__(self):
|
||||
return "@(" + ", ".join([str(e) for e in self.children]) + ")"
|
||||
|
||||
class IdentifierNode(Node):
|
||||
def __init__(self, identifier):
|
||||
self.type = NodeType.IDENTIFIER
|
||||
def __init__(self, identifier):
|
||||
self.identifier = identifier
|
||||
|
||||
def __str__(self):
|
||||
return f"L'{self.identifier}'"
|
||||
|
||||
class AssignExpression(Node):
|
||||
def __init__(self, target, value):
|
||||
self.type = NodeType.ASSIGN
|
||||
def __init__(self, target, value):
|
||||
self.target = target
|
||||
self.value = value
|
||||
|
||||
@@ -78,8 +58,7 @@ class AssignExpression(Node):
|
||||
return f"A[{self.target} = {self.value}]"
|
||||
|
||||
class AsteriskStatementNode(Node):
|
||||
def __init__(self, iterator, statement):
|
||||
self.type = NodeType.ASTERISK
|
||||
def __init__(self, iterator, statement):
|
||||
self.iterator = iterator
|
||||
self.statement = statement
|
||||
|
||||
@@ -87,8 +66,7 @@ class AsteriskStatementNode(Node):
|
||||
return f"*({self.iterator}: {self.statement})"
|
||||
|
||||
class ColonNode(Node):
|
||||
def __init__(self, a, b):
|
||||
self.type = NodeType.COLON
|
||||
def __init__(self, a, b):
|
||||
self.a = a
|
||||
self.b = b
|
||||
|
||||
@@ -101,48 +79,40 @@ class ExpressionNode(Node):
|
||||
|
||||
|
||||
class IntegerLiteralNode(ExpressionNode):
|
||||
def __init__(self, value):
|
||||
self.type = NodeType.INTEGER
|
||||
def __init__(self, value):
|
||||
self.value = value
|
||||
|
||||
def __str__(self):
|
||||
return f"i'{self.value}'"
|
||||
|
||||
class StringLiteralNode(ExpressionNode):
|
||||
def __init__(self, value):
|
||||
self.type = NodeType.STRING
|
||||
def __init__(self, value):
|
||||
self.value = value
|
||||
|
||||
def __str__(self):
|
||||
return f"s'{self.value}'"
|
||||
|
||||
class NoteLiteralNode(ExpressionNode):
|
||||
def __init__(self, value):
|
||||
self.type = NodeType.NOTE
|
||||
def __init__(self, value):
|
||||
self.value = value
|
||||
|
||||
def __str__(self):
|
||||
return f"n'{self.value}'"
|
||||
return f"n'{self.value.note}[{self.value.octave}, {self.value.duration}]'"
|
||||
|
||||
class FunctionCallNode(Node):
|
||||
def __init__(self, identifier, arguments):
|
||||
self.type = NodeType.FUNCTION_CALL
|
||||
def __init__(self, identifier, arguments):
|
||||
self.identifier = identifier
|
||||
self.arguments = arguments
|
||||
|
||||
def __str__(self):
|
||||
return f"F({self.identifier}: {self.arguments})"
|
||||
|
||||
class CommaNode(Node):
|
||||
def __init__(self):
|
||||
self.type = NodeType.COMMA
|
||||
|
||||
class CommaNode(Node):
|
||||
def __str__(self):
|
||||
return "[,]"
|
||||
|
||||
class PercentNode(Node):
|
||||
def __init__(self, value):
|
||||
self.type = NodeType.PERCENT
|
||||
def __init__(self, value):
|
||||
self.value = value
|
||||
|
||||
def __str__(self):
|
||||
|
||||
Reference in New Issue
Block a user