Create ast package

This commit is contained in:
Bartłomiej Pluta
2019-07-03 11:27:51 +02:00
parent 2823fd1896
commit c8ff5ce38f
38 changed files with 622 additions and 11 deletions

View File

@@ -0,0 +1,10 @@
from smnp.ast.node.model import Node
class AccessNode(Node):
def __init__(self, element, property, parent, pos):
Node.__init__(self, parent, pos)
self.children.extend([element, property])
self.element = self.children[0]
self.property = self.children[1]

View File

@@ -0,0 +1,10 @@
from smnp.ast.node.model import Node
class AssignmentNode(Node):
def __init__(self, target, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.extend([target, value])
self.target = self.children[0]
self.value = self.children[1]

View File

@@ -0,0 +1,10 @@
from smnp.ast.node.model import Node
class AsteriskNode(Node):
def __init__(self, iterator, statement, parent, pos):
Node.__init__(self, parent, pos)
self.children.extend([iterator, statement])
self.iterator = self.children[0]
self.statement = self.children[1]

View File

@@ -0,0 +1,17 @@
from smnp.ast.node.model import Node
class BlockNode(Node):
pass
class BlockItemNode(Node):
def __init__(self, statement, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(statement)
self.statement = self.children[0]
class CloseBlockNode(Node):
pass

View File

@@ -0,0 +1,10 @@
from smnp.ast.node.model import Node
class ColonNode(Node):
def __init__(self, a, b, parent, pos):
Node.__init__(self, parent, pos)
self.children.extend([a, b])
self.a = self.children[0]
self.b = self.children[1]

View File

@@ -0,0 +1,20 @@
from smnp.ast.node.model import Node
class FunctionCallNode(Node):
def __init__(self, identifier, arguments, parent, pos):
Node.__init__(self, parent, pos)
self.children.extend([identifier, arguments])
self.identifier = self.children[0]
self.arguments = self.children[1]
class FunctionDefinitionNode(Node):
def __init__(self, name, parameters, body, parent, pos):
Node.__init__(self, parent, pos)
self.children.extend([name, parameters, body])
self.name = self.children[0]
self.parameters = self.children[1]
self.body = self.children[2]

View File

@@ -0,0 +1,9 @@
from smnp.ast.node.model import Node
class IdentifierNode(Node):
def __init__(self, identifier, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(identifier)
self.identifier = self.children[0]

View File

@@ -0,0 +1,9 @@
from smnp.ast.node.model import Node
class IntegerLiteralNode(Node):
def __init__(self, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(value)
self.value = self.children[0]

View File

@@ -0,0 +1,17 @@
from smnp.ast.node.model import Node
class ListNode(Node):
pass
class ListItemNode(Node):
def __init__(self, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(value)
self.value = self.children[0]
class CloseListNode(Node):
pass

View File

@@ -24,6 +24,9 @@ class Node:
def pop(self, index):
return self.children.pop(index)
def print(self):
print(self._print(0))
def _print(self, level):
string = f"{pad(level)}{self.__class__.__name__}({self.parent.__class__.__name__}):\n"
for child in self.children:
@@ -38,4 +41,4 @@ class Node:
def pad(level):
return (" " * level)
return " " * level

View File

@@ -0,0 +1,9 @@
from smnp.ast.node.model import Node
class NoteLiteralNode(Node):
def __init__(self, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(value)
self.value = self.children[0]

View File

@@ -0,0 +1,9 @@
from smnp.ast.node.model import Node
class PercentNode(Node):
def __init__(self, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(value)
self.value = self.children[0]

View File

@@ -0,0 +1,6 @@
from smnp.ast.node.model import Node
class Program(Node):
def __init__(self):
Node.__init__(self, None, (-1, -1))

View File

@@ -0,0 +1,9 @@
from smnp.ast.node.model import Node
class ReturnNode(Node):
def __init__(self, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(value)
self.value = self.children[0]

View File

@@ -0,0 +1,9 @@
from smnp.ast.node.model import Node
class StringLiteralNode(Node):
def __init__(self, value, parent, pos):
Node.__init__(self, parent, pos)
self.children.append(value)
self.value = self.children[0]