add more visitor rules

4 files changed, 34 insertions, 7 deletions

diff --git a/.idea/misc.xml b/.idea/misc.xml
index 716a352..6f66444 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -19,5 +19,5 @@
   <component name="Black">
     <option name="sdkName" value="Python 3.11 (pythonProject)" />
   </component>
-  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.11 (pythonProject)" project-jdk-type="Python SDK" />
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.11 (fp-il)" project-jdk-type="Python SDK" />
 </project>
\ No newline at end of file
diff --git a/.idea/pythonProject.iml b/.idea/pythonProject.iml
index 74d515a..2b5f780 100644
--- a/.idea/pythonProject.iml
+++ b/.idea/pythonProject.iml
@@ -4,7 +4,7 @@
     <content url="file://$MODULE_DIR$">
       <excludeFolder url="file://$MODULE_DIR$/venv" />
     </content>
-    <orderEntry type="inheritedJdk" />
+    <orderEntry type="jdk" jdkName="Python 3.11 (fp-il)" jdkType="Python SDK" />
     <orderEntry type="sourceFolder" forTests="false" />
   </component>
 </module>
\ No newline at end of file
diff --git a/ExpBuilder.py b/ExpBuilder.py
index 1c9699a..a78cb53 100644
--- a/ExpBuilder.py
+++ b/ExpBuilder.py
@@ -1,19 +1,44 @@
+from typing import Dict
+
 from antlr4 import TerminalNode
+from llvmlite import ir
 
 from gen import ANFVisitor
 from gen.ANFParser import ANFParser
-from llvmlite.ir import *
 
 
 class ExpBuilder(ANFVisitor):
-    module = Module("Program")
+    module = ir.Module("Program")
+    i_type = ir.IntType(64)
 
     def __init__(self):
-        self.currentFunc: Function | None = None
+        self.var_env: Dict[ir.Value] = dict()
+        self.currentFunc: ir.Function | None = None
 
     def visitDef(self, ctx: ANFParser.DefContext):
         args = [i.getText() for i in ctx.IDENT()][1:]
-        i_type = IntType(64)
-        self.currentFunc = Function(self.module, FunctionType(i_type, [i_type] * len(args)), name=ctx.IDENT(0))
+        self.currentFunc = ir.Function(self.module,
+                                    ir.FunctionType(self.i_type, [self.i_type] * len(args)),
+                                    name=ctx.IDENT(0))
         self.visit(ctx.cexp())
 
+
+    def visitTrue(self, ctx:ANFParser.TrueContext):
+        return ir.Constant(self.i_type, True)
+
+    def visitFalse(self, ctx: ANFParser.FalseContext):
+        return ir.Constant(self.i_type, False)
+
+    def visitVar(self, ctx:ANFParser.VarContext):
+        var_name = ctx.IDENT().getText()
+        return self.var_env[var_name]
+
+    def visitNum(self, ctx:ANFParser.NumContext):
+        number = int(ctx.NUMBER().getText())
+
+
+    def visitLet(self, ctx:ANFParser.LetContext):
+        var_name = ctx.IDENT().getText()
+        self.var_env[var_name] = self.visit(ctx.funcall())
+        self.currentFunc.append_basic_block(self.var_env[var_name])
+        return self.visit(ctx.cexp())
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..fc66bdd
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,2 @@
+antlr4-python3-runtime==4.13.1
+llvmlite==0.41.1