You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.4 KiB
74 lines
2.4 KiB
#include "BasicBlock.h"
|
|
#include "Constant.h"
|
|
#include "Function.h"
|
|
#include "IRStmtBuilder.h"
|
|
#include "Module.h"
|
|
#include "Type.h"
|
|
|
|
#include <iostream>
|
|
#include <memory>
|
|
#include <vector>
|
|
#ifdef DEBUG // 用于调试信息
|
|
#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("SysY code while_test");
|
|
auto builder = IRStmtBuilder::create(nullptr, module);
|
|
SysYF::Ptr<Type> Int32Type = Type::get_int32_type(module);
|
|
|
|
//全局变量a, b
|
|
auto zero_initializer = ConstantZero::create(Int32Type, module);
|
|
auto a = GlobalVariable::create("a", module, Int32Type, false, zero_initializer);
|
|
auto b = GlobalVariable::create("b", module, Int32Type, false, zero_initializer);
|
|
|
|
auto mainFunTy = FunctionType::create(Int32Type, {});
|
|
auto mainFun = Function::create(mainFunTy, "main", module);
|
|
auto main_entry_bb = BasicBlock::create(module, "main_entry", mainFun);
|
|
builder->set_insert_point(main_entry_bb);
|
|
|
|
auto retAlloca = builder->create_alloca(Int32Type);
|
|
|
|
builder->create_store(CONST_INT(0), b);
|
|
builder->create_store(CONST_INT(3), a);
|
|
|
|
auto while_cond_bb = BasicBlock::create(module, "while_cond_bb", mainFun);
|
|
auto while_true_bb = BasicBlock::create(module, "while_true_bb", mainFun);
|
|
auto while_false_bb = BasicBlock::create(module, "while_false_bb", mainFun);
|
|
|
|
builder->create_br(while_cond_bb);
|
|
builder->set_insert_point(while_cond_bb);
|
|
|
|
auto aLoad = builder->create_load(a);
|
|
auto icmp = builder->create_icmp_gt(aLoad, CONST_INT(0));
|
|
builder->create_cond_br(icmp, while_true_bb, while_false_bb);
|
|
|
|
builder->set_insert_point(while_true_bb);
|
|
auto bLoad = builder->create_load(b);
|
|
aLoad = builder->create_load(a);
|
|
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(while_cond_bb);
|
|
|
|
builder->set_insert_point(while_false_bb);
|
|
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;
|
|
} |