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.

86 lines
2.6 KiB

#ifndef _SYSYF_BASICBLOCK_H_
#define _SYSYF_BASICBLOCK_H_
#include "Value.h"
#include "Instruction.h"
#include "Module.h"
#include "Function.h"
#include <list>
#include <set>
#include <string>
namespace SysYF
{
namespace IR
{
class Function;
class Instruction;
class Module;
class BasicBlock : public Value
{
public:
static Ptr<BasicBlock> create(Ptr<Module> m, const std::string &name ,
Ptr<Function> parent ) {
auto prefix = name.empty() ? "" : "label_";
RET_AFTER_INIT(BasicBlock, m, prefix + name, parent);
}
// return parent, or null if none.
Ptr<Function> get_parent() { return parent_; }
Ptr<Module> get_module();
/****************api about cfg****************/
PtrList<BasicBlock> &get_pre_basic_blocks() { return pre_bbs_; }
PtrList<BasicBlock> &get_succ_basic_blocks() { return succ_bbs_; }
void add_pre_basic_block(Ptr<BasicBlock> bb) { pre_bbs_.push_back(bb); }
void add_succ_basic_block(Ptr<BasicBlock> bb) { succ_bbs_.push_back(bb); }
void remove_pre_basic_block(Ptr<BasicBlock> bb) { pre_bbs_.remove(bb); }
void remove_succ_basic_block(Ptr<BasicBlock> bb) { succ_bbs_.remove(bb); }
/****************api about cfg****************/
/// Returns the terminator instruction if the block is well formed or null
/// if the block is not well formed.
const Ptr<Instruction> get_terminator() const;
Ptr<Instruction> get_terminator() {
return const_pointer_cast<Instruction>(
static_pointer_cast<const BasicBlock>(shared_from_this())->get_terminator());
}
void add_instruction(Ptr<Instruction> instr);
void add_instruction(PtrList<Instruction>::iterator instr_pos, Ptr<Instruction> instr);
void add_instr_begin(Ptr<Instruction> instr);
PtrList<Instruction>::iterator find_instruction(Ptr<Instruction> instr);
void delete_instr(Ptr<Instruction> instr);
bool empty() { return instr_list_.empty(); }
int get_num_of_instr() { return instr_list_.size(); }
PtrList<Instruction> &get_instructions() { return instr_list_; }
void erase_from_parent();
virtual std::string print() override;
private:
explicit BasicBlock(Ptr<Module> m, const std::string &name ,
Ptr<Function> parent );
void init(Ptr<Module> m, const std::string &name ,
Ptr<Function> parent );
PtrList<BasicBlock> pre_bbs_;
PtrList<BasicBlock> succ_bbs_;
PtrList<Instruction> instr_list_;
Ptr<Function> parent_;
};
}
}
#endif // _SYSYF_BASICBLOCK_H_