diff options
1 files changed, 14 insertions, 1 deletions
| -rw-r--r-- | Year 2/Implementation of Programming Languages/Transpiler/src/LLVMVisitor/LLVMVisitor.cpp | 15 |
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() { |