[IPL/T] Implemented if, without boolean algebra

1 files changed, 40 insertions, 5 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 d6cdf89..1db9365 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
@@ -37,7 +37,7 @@ LLVMRet LLVMVisitor::f_defun(const std::vector<Value*>& args) {
 	outFile << ") {\n";
 
 	for (unsigned i = outbodySize; i < outBuf.size(); i++) {
-		outFile << outBuf[i];
+		outFile << "\t" << outBuf[i] << "\n";
 	}
 	while (outbodySize < outBuf.size()) {
 		outBuf.pop_back();
@@ -60,6 +60,40 @@ LLVMRet LLVMVisitor::f_defvar(funcargs args) {
 	return val;
 }
 
+LLVMRet LLVMVisitor::f_if(funcargs args) {
+	LLVMRet boolExpr = args[0]->Visit(*this);
+	Register rResult = lreg++,
+	         rTrue = lreg++,
+	         rFalse = lreg++,
+	         rCont = (args.size() == 2) ? rFalse : lreg++;
+	
+	unsigned reservedI = outBuf.size();
+	outBuf.push_back("");
+	outBuf.push_back("");
+	outBuf.push_back("");
+
+	LLVMRet tExpr = args[1]->Visit(*this);
+
+	outBuf[reservedI  ] = *rResult + " = alloca " + tExpr.type;
+	outBuf[reservedI+1] = "br " + boolExpr.toStr() + ", label " + *rTrue + ", label " + *rFalse + "\n";
+	outBuf[reservedI+2] = "v" + rTrue.headless() + ":";
+
+	outBuf.push_back("store " + tExpr.toStr() + ", ptr " + *rResult);
+	outBuf.push_back("br label " + *rCont + "\n");
+
+	if (args.size() > 2) {
+		outBuf.push_back("v" + rFalse.headless() + ":");
+		LLVMRet fExpr = args[2]->Visit(*this);
+		outBuf.push_back("store " + fExpr.toStr() + ", ptr " + *rResult);
+		outBuf.push_back("br label " + *rCont + "\n");
+	}
+
+	outBuf.push_back("v" + rCont.headless() + ":");
+	Register finalRegister = lreg++;
+	outBuf.push_back(*finalRegister + " = load " + tExpr.type + ", ptr " + *rResult);
+	return { tExpr.type, *finalRegister };
+}
+
 LLVMRet LLVMVisitor::nameAsVar(const LLVMRet& ret) {
 	if (!locals.empty()) {
 		auto asLoc = locals.top()->find("%_" + ret.addr);
@@ -92,7 +126,7 @@ LLVMRet LLVMVisitor::aritop(const std::string name, const std::vector<Value*>& a
 	if (rightRet.type == "name") rightRet = nameAsVar(rightRet);
 
 	Register r = lreg++;
-	outBuf.push_back(*r + " = " + ((leftRet.type == "double") ? "f" : "") + name + " " + leftRet.toStr() + ", " + rightRet.addr + "\n");
+	outBuf.push_back(*r + " = " + ((leftRet.type == "double") ? "f" : "") + name + " " + leftRet.toStr() + ", " + rightRet.addr);
 
 	return { leftRet.type, *r };
 }
@@ -138,7 +172,7 @@ LLVMRet LLVMVisitor::Visit(const Funcall& vfuncall) {
 		if (arg.addr != pargs.back().addr)
 			out += ", ";
 	}
-	outBuf.push_back(out += ")\n");
+	outBuf.push_back(out += ")");
 	return { userFuncIt->second.type, valReg };
 }
 
@@ -152,14 +186,15 @@ LLVMVisitor::LLVMVisitor(const char* fileName) : outFile(std::ofstream(fileName)
 	funcs_direct.insert({"/", &LLVMVisitor::f_div});
 	funcs_direct.insert({"defun", &LLVMVisitor::f_defun});
 	funcs_direct.insert({"defvar", &LLVMVisitor::f_defvar});
+	funcs_direct.insert({"if", &LLVMVisitor::f_if});
 }
 
 void LLVMVisitor::insertMain() {
 	outFile << "define i32 @main() {\n";
 	for (std::string& codeline : outBuf) {
-		outFile << codeline;
+		outFile << "\t" << codeline << "\n";
 	}
-	outFile << "ret i32 0\n}\n";
+	outFile << "\tret i32 0\n}\n";
 }
 
 LLVMRet LLVMVisitor::Visit(const Integer& vint) {