bill
/
ChocoPy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'sanjar/functions' of https://github.com/nyu-compiler-construction/pa3-chocopy-code-generation-mjolnir into bill/misc

Browse Source
master
bill 4 years ago
parent 42518d6547 ef4ac31e04
commit 4d6e6fdede
1 changed files with 62 additions and 36 deletions
Show Stats Download Patch File Download Diff File

80
src/main/java/chocopy/pa3/CodeGenImpl.java
Unescape View File

@ -300,25 +300,34 @@ public class CodeGenImpl extends CodeGenBase
// function
Identifier functionId = node.function;
FuncInfo calleeFunctionInfo = (FuncInfo) sym.get(node.function.name);
boolean isInMainFunction = funcInfo == null;
if (!isInMainFunction) {
System.out.println("--> caller: " + funcInfo.getFuncName() + " depth: " + funcInfo.getDepth());
System.out.println("--> callee: " + calleeFunctionInfo.getFuncName() + " depth: " + calleeFunctionInfo.getDepth());
List<Expr> args = node.args;
int calleeDepth = calleeFunctionInfo.getDepth();
int callerDepth = funcInfo.getDepth();
int jumps = callerDepth - calleeDepth + 1;
int depth = calleeFunctionInfo.getDepth();
int tmpHandle = getRegister();
Register tmp = registerPool[tmpHandle];
FuncInfo currFuncInfo = calleeFunctionInfo;
backend.emitMV(tmp, FP, "Load FP");
while (depth > 0) {
final FuncInfo parentFuncInfo = currFuncInfo.getParentFuncInfo();
backend.emitLW(tmp, tmp, parentFuncInfo.getParams().size()*wordSize, "Load static link to " + parentFuncInfo.getFuncName());
currFuncInfo = parentFuncInfo;
--depth;
FuncInfo curr = funcInfo;
for (int i = 0; i < jumps; i++) {
backend.emitLW(tmp, tmp, curr.getParams().size()*wordSize, "Load static link to " + curr.getFuncName());
curr = curr.getParentFuncInfo();
}
backend.emitSW(tmp, SP, -1 * wordSize, "Push static link onto active frame.");
freeRegister(tmpHandle);
} else {
backend.emitSW(FP, SP, -1 * wordSize, "Push static link onto active frame.");
}
List<Expr> args = node.args;
int spaceRequiredForArgs = (args.size() + 1)*4;
backend.emitMV(T0, FP, "Get static link to " + (funcInfo != null ? funcInfo.getFuncName() : "main"));
backend.emitSW(T0, SP, -wordSize, "Push static link onto active frame.");
for (int i = 0; i < args.size(); i++) {
int argNum = i + 1;
int slotNum = argNum + 1; // We have extra slot for static link
@ -344,13 +353,14 @@ public class CodeGenImpl extends CodeGenBase
// All expressions should save their end result in A0
// So, once expr is evaluated add value inside A0 onto stack as an actual argument
backend.emitSW(A0, SP, -wordSize*slotNum, "Push actual argument for " + formalParamName + " onto stack");
backend.emitSW(A0, SP, -slotNum * wordSize, "Push actual argument for " + formalParamName + " onto stack");
}
backend.emitADDI(SP, SP, -spaceRequiredForArgs, "Set SP to last argument.");
backend.emitJAL(calleeFunctionInfo.getCodeLabel(), "Invoke function: " + functionId.name);
backend.emitADDI(SP, SP, spaceRequiredForArgs, "Set SP to stack frame top.");
}
return A0;
}
@ -439,6 +449,7 @@ public class CodeGenImpl extends CodeGenBase
backend.emitLI(Register.A0, 0, "Load boolean literal: false ");
return Register.A0;
}
class StackVarRuntimeInfo {
public final int off;
public final Register sl;
@ -460,6 +471,7 @@ public class CodeGenImpl extends CodeGenBase
}
}
}
private Object getVar(Identifier id){
SymbolInfo info = sym.get(id.name);
if(info instanceof StackVarInfo)
@ -467,28 +479,42 @@ public class CodeGenImpl extends CodeGenBase
else
return ((GlobalVarInfo)info).getLabel();
}
private StackVarRuntimeInfo getStackVar(StackVarInfo info){
int curr_depth = funcInfo.getDepth();
int d_depth = curr_depth - info.getFuncInfo().getDepth();
if(d_depth == 0) {
int idx = funcInfo.getVarIndex(info.getVarName());
int paramSize = funcInfo.getParams().size();
int offset = getLocalVarLocationOffset(idx, paramSize);
return new StackVarRuntimeInfo(FP, offset);
} else
{
FuncInfo curr = funcInfo;
FuncInfo varFuncInfo = info.getFuncInfo();
System.out.println("var name: " + info.getVarName());
System.out.println("varfuncInfo name: " + varFuncInfo.getFuncName());
System.out.println("var idx: " + varFuncInfo.getVarIndex(info.getVarName()));
int tmpHandle = getRegister();
Register tmp = registerPool[tmpHandle];
backend.emitMV(tmp, FP, "tmp = FP");
while(d_depth > 0){
if (funcInfo != null && funcInfo != info.getFuncInfo()) {
int varFunDepth = varFuncInfo.getDepth();
int callerDepth = funcInfo.getDepth();
int jumps = callerDepth - varFunDepth;
System.out.println("varFunDepth: " + varFunDepth);
System.out.println("callerDepth: " + callerDepth);
backend.emitMV(tmp, FP, "Load FP");
FuncInfo curr = funcInfo;
for (int i = 0; i < jumps; i++) {
backend.emitLW(tmp, tmp, curr.getParams().size()*wordSize, "Load static link from " + curr.getFuncName() + " to " + curr.getParentFuncInfo().getBaseName());
curr = curr.getParentFuncInfo();
backend.emitLW(tmp, tmp, curr.getParams().size()*wordSize, "Get static link");
-- d_depth;
}
// UNSAFE!!!!! PLEASE FREE THE TMP_HANDLE MANUALLY
return new StackVarRuntimeInfo(tmp, -wordSize*curr.getVarIndex(info.getVarName()), tmpHandle);
System.out.println("We are referencing non local variable");
} else {
System.out.println("We are referencing local variable");
backend.emitMV(tmp, FP, "We are referencing local variable");
}
int idx = varFuncInfo.getVarIndex(info.getVarName());
int paramSize = varFuncInfo.getParams().size();
int offset = getLocalVarLocationOffset(idx, paramSize);
return new StackVarRuntimeInfo(tmp, offset, tmpHandle);
}
@Override

Write Preview
Loading…
Cancel
Save