Initial commit from sysy-main
This commit is contained in:
Lixuanwang
295
src/Backend.cpp
Normal file
295
src/Backend.cpp
Normal file
@@ -0,0 +1,295 @@
|
||||
#include "Backend.h"
|
||||
using namespace sysy;
|
||||
|
||||
namespace sysy {
|
||||
|
||||
using RegId = RegManager::RegId;
|
||||
|
||||
string CodeGen::code_gen() {
|
||||
string code;
|
||||
code += module_gen(module);
|
||||
return code;
|
||||
}
|
||||
|
||||
string CodeGen::module_gen(Module *module) {
|
||||
string code;
|
||||
string dataCode;
|
||||
string textCode;
|
||||
// clear last module's label record
|
||||
clearModuleRecord(module);
|
||||
// generate asmcode for all global values
|
||||
dataCode += globaldata_gen();
|
||||
|
||||
code += space + ".arch armv7ve " + eol;
|
||||
code += space + ".text " + eol;
|
||||
auto functions = module->getFunctions();
|
||||
|
||||
for (auto iter = functions->begin(); iter != functions->end(); ++iter) {
|
||||
string name = iter->first;
|
||||
Function *func = iter->second;
|
||||
auto bblist = func->getBasicBlocks();
|
||||
if (bblist.empty())
|
||||
continue;
|
||||
// generate asmcode for each function
|
||||
textCode += function_gen(func) + eol;
|
||||
}
|
||||
code += (dataCode + textCode + eol);
|
||||
return code;
|
||||
}
|
||||
|
||||
string CodeGen::functionHead_gen(Function *func) {
|
||||
string code;
|
||||
code += space + ".globl " + func->getName() + eol;
|
||||
code += space + ".p2align " + std::to_string(int_p2align) + eol;
|
||||
code += space + ".type " + func->getName() + ", %function" + eol;
|
||||
code += func->getName() + ":" + eol;
|
||||
return code;
|
||||
}
|
||||
/**
|
||||
* stack structure:
|
||||
*
|
||||
* last function stack
|
||||
* -------------------------- <-- fp point
|
||||
* callee-saved regs (include fp, sp, lr)
|
||||
* --------------------------
|
||||
* local variables and subfunc's return value (ir inst)
|
||||
* --------------------------
|
||||
* arg0,arg1,arg2,arg3 (as tempory variable)
|
||||
* --------------------------
|
||||
* caller-saved regs
|
||||
* --------------------------
|
||||
* arg_x ~ arg_4
|
||||
* -------------------------- <-- sp point
|
||||
* next function stack
|
||||
*
|
||||
*/
|
||||
/**
|
||||
* prologue :
|
||||
* preserve callee-saved registers (lr, fp and other callee-saved regs)
|
||||
* set new fp
|
||||
* alloc stack space for local var/ args / return value.
|
||||
* store args to stack
|
||||
* */
|
||||
string CodeGen::prologueCode_gen(Function *func) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
|
||||
/* epilogue :
|
||||
* free stack space
|
||||
* restore sp
|
||||
* restore used callee-saved register(lr, fp and other callee-saved
|
||||
* regs) bx lr
|
||||
*/
|
||||
string CodeGen::epilogueCode_gen(Function *func) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
|
||||
string CodeGen::function_gen(Function *func) {
|
||||
curFunc = func;
|
||||
clearFunctionRecord(func);
|
||||
string bbCode;
|
||||
auto bbs = func->getBasicBlocks();
|
||||
for (auto iter = bbs.begin(); iter != bbs.end(); ++iter) {
|
||||
auto bb = iter->get();
|
||||
bbCode += basicBlock_gen(bb);
|
||||
}
|
||||
string code;
|
||||
string funcHead = functionHead_gen(func);
|
||||
string prologueCode = prologueCode_gen(func);
|
||||
string epilogueCode = epilogueCode_gen(func);
|
||||
string literalPoolsCode = literalPoolsCode_gen(func);
|
||||
//
|
||||
code = funcHead + prologueCode + bbCode + epilogueCode + literalPoolsCode;
|
||||
return code;
|
||||
}
|
||||
|
||||
string CodeGen::basicBlock_gen(BasicBlock *bb) {
|
||||
curBB = bb;
|
||||
string bbLabel = getBBLabel(bb);
|
||||
string code;
|
||||
code += bbLabel + ":" + eol;
|
||||
for (auto &instr : bb->getInstructions()) {
|
||||
auto instrType = instr->getKind();
|
||||
code += instruction_gen(instr.get());
|
||||
}
|
||||
return code;
|
||||
}
|
||||
/**
|
||||
* RegId : binaryInst_gen returns RegId as its destination operand
|
||||
* code : asmcode generated by binaryInst_gen
|
||||
*/
|
||||
pair<RegId, string> CodeGen::binaryInst_gen(BinaryInst *bInst, RegId dstRegId) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return {dstRegId, code};
|
||||
}
|
||||
|
||||
pair<RegId, string> CodeGen::unaryInst_gen(UnaryInst *uInst, RegId dstRegId) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return {dstRegId, code};
|
||||
}
|
||||
pair<RegId, string> CodeGen::allocaInst_gen(AllocaInst *aInst,
|
||||
RegManager::RegId dstRegId) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return {dstRegId, code};
|
||||
}
|
||||
|
||||
string CodeGen::storeInst_gen(StoreInst *stInst) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
pair<RegId, string> CodeGen::loadInst_gen(LoadInst *ldInst, RegId dstRegId) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return {dstRegId, code};
|
||||
}
|
||||
string CodeGen::returnInst_gen(ReturnInst *retInst) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
string CodeGen::uncondBrInst_gen(UncondBrInst *ubInst) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
string CodeGen::condBrInst_gen(CondBrInst *cbInst) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
pair<RegId, string> CodeGen::callInst_gen(CallInst *callInst, RegId dstRegId) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return {dstRegId, code};
|
||||
}
|
||||
|
||||
string CodeGen::instruction_gen(Instruction *instr) {
|
||||
string code;
|
||||
string unkName = instr->getName();
|
||||
RegManager::RegId dstRegId = RegManager::RNONE;
|
||||
auto instrType = instr->getKind();
|
||||
pair<RegId, string> tmp;
|
||||
switch (instrType) {
|
||||
// binary inst
|
||||
case Instruction::kAdd:
|
||||
case Instruction::kMul:
|
||||
case Instruction::kSub: {
|
||||
BinaryInst *bInst = dynamic_cast<BinaryInst *>(instr);
|
||||
// registers are used only for instruction operation, consider use which
|
||||
// register (any one that is free for use)
|
||||
tmp = binaryInst_gen(bInst, RegManager::RANY);
|
||||
code += tmp.second;
|
||||
dstRegId = tmp.first;
|
||||
break;
|
||||
}
|
||||
case Instruction::kLoad: {
|
||||
LoadInst *ldInst = dynamic_cast<LoadInst *>(instr);
|
||||
tmp = loadInst_gen(ldInst, RegManager::RANY);
|
||||
code += M_emitComment("load inst");
|
||||
code += tmp.second;
|
||||
dstRegId = tmp.first;
|
||||
break;
|
||||
}
|
||||
case Instruction::kStore: {
|
||||
StoreInst *stInst = dynamic_cast<StoreInst *>(instr);
|
||||
code += M_emitComment("store inst");
|
||||
code += storeInst_gen(stInst);
|
||||
return code;
|
||||
break;
|
||||
}
|
||||
case Instruction::kAlloca: {
|
||||
AllocaInst *aInst = dynamic_cast<AllocaInst *>(instr);
|
||||
tmp = allocaInst_gen(aInst, RegManager::RANY);
|
||||
code += M_emitComment("alloca inst");
|
||||
code += tmp.second;
|
||||
dstRegId = tmp.first;
|
||||
break;
|
||||
}
|
||||
case Instruction::kReturn: {
|
||||
ReturnInst *retInst = dynamic_cast<ReturnInst *>(instr);
|
||||
code += M_emitComment("return inst");
|
||||
code += returnInst_gen(retInst);
|
||||
return code;
|
||||
break;
|
||||
}
|
||||
case Instruction::kCall: {
|
||||
CallInst *cInst = dynamic_cast<CallInst *>(instr);
|
||||
auto tmp = callInst_gen(cInst, RegManager::RANY);
|
||||
code += tmp.second;
|
||||
dstRegId = tmp.first;
|
||||
if (dstRegId == RegManager::R0)
|
||||
return code;
|
||||
break;
|
||||
}
|
||||
case Instruction::kBr: {
|
||||
UncondBrInst *ubInst = dynamic_cast<UncondBrInst *>(instr);
|
||||
code += uncondBrInst_gen(ubInst);
|
||||
return code;
|
||||
break;
|
||||
}
|
||||
case Instruction::kCondBr: {
|
||||
CondBrInst *cbInst = dynamic_cast<CondBrInst *>(instr);
|
||||
code += condBrInst_gen(cbInst);
|
||||
return code;
|
||||
break;
|
||||
}
|
||||
default: {
|
||||
code +=
|
||||
"ERROR CODE : instruction " + unkName + " is not implementation" + eol;
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (!instr->getType()->isVoid()) {
|
||||
code += storeRegToStack_gen(dstRegId, instr);
|
||||
// regm.freeReg(dstRegId);//TODO : code in here.
|
||||
}
|
||||
return code;
|
||||
}
|
||||
//
|
||||
string CodeGen::globaldata_gen() {
|
||||
string asmCode;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return asmCode;
|
||||
}
|
||||
|
||||
string CodeGen::literalPoolsCode_gen(Function *func) {
|
||||
string code;
|
||||
/**
|
||||
*code in here
|
||||
*/
|
||||
return code;
|
||||
}
|
||||
|
||||
} // namespace sysy
|
||||
Reference in New Issue
Block a user