task2: while and func - Done

main
origami-b 12 months ago
parent 36f4d12287
commit 8d314198c4

@ -1,3 +1,113 @@
#include "BasicBlock.h"
#include "Constant.h"
#include "Function.h"
#include "IRStmtBuilder.h"
#include "Module.h"
#include "Type.h"
#include <iostream>
#include <memory>
#ifdef DEBUG // 用于调试信息,大家可以在编译过程中通过" -DDEBUG"来开启这一选项
#define DEBUG_OUTPUT std::cout << __LINE__ << std::endl; // 输出行号的简单示例
#else
#define DEBUG_OUTPUT
#endif
#define CONST_INT(num) \
ConstantInt::create(num, module)
#define CONST_FP(num) \
ConstantFloat::create(num, module) // 得到常数值的表示,方便后面多次用到
using namespace SysYF::IR;
int main() {
auto module = Module::create("SysYF code"); // module name是什么无关紧要
auto builder = IRStmtBuilder::create(nullptr, module);
SysYF::Ptr<Type> Int32Type = Type::get_int32_type(module);
// add(a, b)
// 函数参数类型的vector
std::vector<SysYF::Ptr<Type>> Ints(2, Int32Type);
//通过返回值类型与参数类型列表得到函数类型
auto addFunTy = FunctionType::create(Int32Type, Ints);
// 由函数类型得到函数
auto addFun = Function::create(addFunTy,
"add", module);
// BB的名字在生成中无所谓,但是可以方便阅读
auto bb = BasicBlock::create(module, "entry", addFun);
builder->set_insert_point(bb); // 一个BB的开始,将当前插入指令点的位置设在bb
auto retAlloca = builder->create_alloca(Int32Type); // 在内存中分配返回值的位置
auto aAlloca = builder->create_alloca(Int32Type); // 在内存中分配参数a的位置
auto bAlloca = builder->create_alloca(Int32Type); // 在内存中分配参数b的位置
std::vector<SysYF::Ptr<Value>> args; // 获取add函数的形参,通过Function中的iterator
for (auto arg = addFun->arg_begin(); arg != addFun->arg_end(); arg++) {
args.push_back(*arg); // * 号运算符是从迭代器中取出迭代器当前指向的元素
}
builder->create_store(args[0], aAlloca); // store参数a
builder->create_store(args[1], bAlloca); // store参数b
auto aLoad = builder->create_load(aAlloca); // load参数a
auto bLoad = builder->create_load(bAlloca); // load参数b
// a+b
auto add = builder->create_iadd(aLoad, bLoad);
// a+b-1
auto sub = builder->create_isub(add, CONST_INT(1));
builder->create_store(sub, retAlloca);
auto retLoad = builder->create_load(retAlloca);
builder->create_ret(retLoad);
builder->create_ret(retLoad);
// main函数
auto mainFun = Function::create(FunctionType::create(Int32Type, {}),
"main", module);
bb = BasicBlock::create(module, "entry", mainFun);
// BasicBlock的名字在生成中无所谓,但是可以方便阅读
builder->set_insert_point(bb);
retAlloca = builder->create_alloca(Int32Type);
// int a;
auto a = builder->create_alloca(Int32Type);
// int b;
auto b = builder->create_alloca(Int32Type);
// int c;
auto c = builder->create_alloca(Int32Type);
// a = 3;
builder->create_store(CONST_INT(3), a);
// b = 2;
builder->create_store(CONST_INT(2), b);
// c = 5;
builder->create_store(CONST_INT(5), c);
// tmp = add(a, b);
aLoad = builder->create_load(a);
bLoad = builder->create_load(b);
auto tmp = builder->create_call(addFun, {aLoad, bLoad});
// ret = c + tmp;
auto cLoad = builder->create_load(c);
auto ret = builder->create_iadd(cLoad, tmp);
builder->create_store(ret, retAlloca);
retLoad = builder->create_load(retAlloca);
builder->create_ret(retLoad);
std::cout << module->print();
return 0;
}

@ -1,3 +1,83 @@
#include "BasicBlock.h"
#include "Constant.h"
#include "Function.h"
#include "IRStmtBuilder.h"
#include "Module.h"
#include "Type.h"
#include <iostream>
#include <memory>
#ifdef DEBUG // 用于调试信息,大家可以在编译过程中通过" -DDEBUG"来开启这一选项
#define DEBUG_OUTPUT std::cout << __LINE__ << std::endl; // 输出行号的简单示例
#else
#define DEBUG_OUTPUT
#endif
#define CONST_INT(num) \
ConstantInt::create(num, module)
#define CONST_FP(num) \
ConstantFloat::create(num, module) // 得到常数值的表示,方便后面多次用到
using namespace SysYF::IR;
int main() {
auto module = Module::create("SysYF code"); // module name是什么无关紧要
auto builder = IRStmtBuilder::create(nullptr, module);
SysYF::Ptr<Type> Int32Type = Type::get_int32_type(module);
// int a;
auto a = GlobalVariable::create("a", module, Int32Type, false, CONST_INT(0));
// int b;
auto b = GlobalVariable::create("b", module, Int32Type, false, CONST_INT(0));
// main函数
auto mainFun = Function::create(FunctionType::create(Int32Type, {}),
"main", module);
auto bb = BasicBlock::create(module, "entry", mainFun);
// BasicBlock的名字在生成中无所谓,但是可以方便阅读
builder->set_insert_point(bb);
auto retAlloca = builder->create_alloca(Int32Type);
// b=0;
builder->create_store(CONST_INT(0), b);
// a = 3;
builder->create_store(CONST_INT(3), a);
// whileBBs
auto condBB = BasicBlock::create(module, "condBB_while", mainFun);
auto trueBB = BasicBlock::create(module, "trueBB_while", mainFun);
auto falseBB = BasicBlock::create(module, "falseBB_while", mainFun);
builder->create_br(condBB);
// condBB
builder->set_insert_point(condBB);
auto aLoad = builder->create_load(a);
auto icmp = builder->create_icmp_gt(aLoad, CONST_INT(0));
builder->create_cond_br(icmp, trueBB, falseBB);
// trueBB
builder->set_insert_point(trueBB);
aLoad = builder->create_load(a);
auto bLoad = builder->create_load(b);
auto add = builder->create_iadd(bLoad, aLoad);
builder->create_store(add, b);
auto sub = builder->create_isub(aLoad, CONST_INT(1));
builder->create_store(sub, a);
builder->create_br(condBB);
// falseBB
builder->set_insert_point(falseBB);
bLoad = builder->create_load(b);
builder->create_store(bLoad, retAlloca);
auto retLoad = builder->create_load(retAlloca);
builder->create_ret(retLoad);
std::cout << module->print();
return 0;
}
Loading…
Cancel
Save