ExprNode LiteralNode codeGen

- adding the simplest form of their functions
- made it work

AST.py

@@ -85,6 +85,13 @@ class ASTNode():
                         c.codeGen()
                     self.code += c.code     
 
+    def codeGen(self):
+        res = ""
+        for c in self.children:
+            if c and hasattr(c, 'codeGen'):
+                res += c.codeGen()
+        return res
+
     def printNodePretty(self, prefix=0):
         pp = pprint.PrettyPrinter(indent=prefix)
         pp.pprint(self.__class__.__name__)

ExprPrimaryNodes.py

@@ -421,6 +421,30 @@ class ExprNode(ASTNode):
 
         raise Exception("ERROR: Incompatible types. Left of {} type can't be used with right of {} type on operation {}".format(self.left.myType.name, self.right.myType.name, self.op))
 
+    def codeGen(self):
+        res = ""
+
+        # Unary:
+        if not self.left:
+            return res
+
+        res += self.left.codeGen() # get left output
+        res += p("push", "eax", None, None)
+
+        res += self.right.codeGen() # get right output
+        res += p("pop", "ebx", None, None)
+
+        if self.op in ['+', '-', '*', '/', '%']:
+            # operation -> generated code
+            ops = {
+                '+': "add",
+                '-': "sub"
+            }
+            res += p(ops[self.op], "ebx", "eax", " " + ops[self.op] + " left + right")
+            res += p("mov", "eax", "ebx", " move result to eax")
+
+        return res
+
     # returns True, False, Int or None (for non-constant expr)
     # children of exprNode is either exprNode or literalNode
     def getConstant(self):

UnitNodes.py

 from AST import ASTNode, getParseTreeNodes
 from Environment import Env
 from TheTypeNode import TypeNode, TypeStruct
+from CodeGenUtils import p
 
 #   LiteralNode
 #   ParamNode
@@ -40,6 +41,15 @@ class LiteralNode(ASTNode):
             node = self.env.getNode(self.name, "type")
             self.myType = TypeStruct(self.name, node)
 
+    def codeGen(self):
+        res = ""
+        
+        # int
+        if self.name == 'int':
+            res += p("mov", "eax", self.getConstant(), " set to literal")
+
+        return res
+
     def getConstant(self):
         if self.name == 'int':
             return int(self.value)