[IPL/T] Implemented defvar

1 files changed, 14 insertions, 1 deletions

diff --git a/Year 2/Implementation of Programming Languages/Transpiler/src/LLVMVisitor/LLVMVisitor.cpp b/Year 2/Implementation of Programming Languages/Transpiler/src/LLVMVisitor/LLVMVisitor.cpp
index 0424d04..238c2f6 100644
--- a/Year 2/Implementation of Programming Languages/Transpiler/src/LLVMVisitor/LLVMVisitor.cpp
+++ b/Year 2/Implementation of Programming Languages/Transpiler/src/LLVMVisitor/LLVMVisitor.cpp
@@ -45,7 +45,19 @@ LLVMRet LLVMVisitor::f_defun(const std::vector<Value*>& args) {
 
 	outFile << "\tret " << funcRet << "\n}\n\n";
 	locals.pop();
-	return { funcRet.type, "@" + fname->get_value() };
+	LLVMRet retType = { funcRet.type, "@" + fname->get_value() };
+	funcs_user.insert({ retType.addr, retType });
+	return retType;
+}
+
+LLVMRet LLVMVisitor::f_defvar(funcargs args) {
+	std::string name = "@" + static_cast<Name*>(args[0])->get_value();
+	LLVMRet retVal = args[1]->Visit(*this);
+	outFile << name << " = global " << retVal << "\n";
+
+	LLVMRet val = { retVal.type, name };
+	vars_user.insert({ name, val });
+	return val;
 }
 
 LLVMRet LLVMVisitor::nameAsVar(const LLVMRet& ret) {
@@ -129,6 +141,7 @@ LLVMVisitor::LLVMVisitor(const char* fileName) : outFile(std::ofstream(fileName)
 	funcs_direct.insert({"+", &LLVMVisitor::f_add});
 	funcs_direct.insert({"-", &LLVMVisitor::f_sub});
 	funcs_direct.insert({"defun", &LLVMVisitor::f_defun});
+	funcs_direct.insert({"defvar", &LLVMVisitor::f_defvar});
 }
 
 void LLVMVisitor::insertMain() {