boom

interpeter.py

@@ -0,0 +1,66 @@
+from nodes import  *
+from values import Number
+
+
+
+class Interpeter:
+	def visit(self, node):
+		method_name = f'visit_{type(node).__name__}'
+		method = getattr(self, method_name)
+		return method(node)
+
+	def visit_NumberNode(self, node) -> Number:
+		return Number(node.value)
+
+	def visit_AddNode(self, node) -> Number:
+		return Number(self.visit(node.node_a).value + self.visit(node.node_b).value)
+
+	def visit_SubtractNode(self, node) -> Number:
+		return Number(self.visit(node.node_a).value - self.visit(node.node_b).value)
+
+	def visit_MultiplyNode(self, node) -> Number:
+		return Number(self.visit(node.node_a).value * self.visit(node.node_b).value)
+
+	def visit_DivideNode(self, node) -> Number:
+		try:
+			return Number(self.visit(node.node_a).value / self.visit(node.node_b).value)
+		except Exception:
+			raise Exception("Runtime Math error (division)")
+
+	def visit_PlusNode(self, node) -> Number:
+		return self.visit(node.node)
+
+	def visit_MinusNode(self, node) -> Number:
+		return Number(-self.visit(node.node).value)
+
+	def visit_FactorialNode(self, node) -> Number:
+		fact = lambda x: 1 if x == 0 else x * fact(x-1)
+		try:
+			return Number(fact(self.visit(node.node).value))
+		except Exception:
+			raise Exception("Runtime Math error (Factorial)")
+
+	def visit_PowerNode(self, node) -> Number:
+		return Number(pow(self.visit(node.node_a).value, self.visit(node.node_b).value))
+
+	def visit_ModuleNode(self, node) -> Number:
+		return Number(self.visit(node.node_a).value % self.visit(node.node_b).value)
+
+	def visit_AverageNode(self, node) -> Number:
+		return Number((self.visit(node.node_a).value + self.visit(node.node_b).value) / 2.0 )
+
+	def visit_MaximumNode(self, node) -> Number:
+		return Number(max(self.visit(node.node_a).value, self.visit(node.node_b).value) )
+
+	def visit_MinimumNode(self, node) -> Number:
+		return Number(min(self.visit(node.node_a).value, self.visit(node.node_b).value) )
+
+	def visit_NegateNode(self, node) -> Number:
+		
+		if not self.visit(node.node).value.is_integer():
+			raise Exception("Runtime Math error (Negate)")
+
+
+		return Number(~int(self.visit(node.node).value))
+
+