Create evaluators

This commit is contained in:
Bartłomiej Pluta
2019-06-29 01:01:07 +02:00
parent 2f362da0bd
commit 9a3936a202
5 changed files with 176 additions and 61 deletions

62
AST.py
View File

@@ -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):