105 lines
3.2 KiB
C++
105 lines
3.2 KiB
C++
#ifndef INCLUDE_TCG_SYMBOLMANAGER_H_
|
|
#define INCLUDE_TCG_SYMBOLMANAGER_H_
|
|
|
|
#include "Public/Public.h"
|
|
#include "Public/Scope.h"
|
|
#include "Public/TAC.h"
|
|
#include "TCG/ASM.h"
|
|
|
|
enum class POSTYPE {
|
|
GLOBAL, REG, MEM, NONE
|
|
};
|
|
|
|
|
|
// 需要 push 的情况: {reg, val, false, -1}
|
|
// 需要更新内存的情况: {reg, val, false, 内存位置}
|
|
struct RelacedEeg {
|
|
REG reg; // 被替换的寄存器
|
|
std::string val; // 原本的变量
|
|
bool no_use; // 以后是否不再使用
|
|
int mem; // 如果该变量还要继续使用，且变量在内存中，则存储内存地址，否则为-1
|
|
|
|
RelacedEeg() : reg(REG::None), val(""), no_use(true), mem(-1) {}
|
|
RelacedEeg(REG reg, std::string val, bool no_use = true, int mem = -1) : reg(reg), val(val), no_use(no_use), mem(mem) {}
|
|
|
|
RelacedEeg& operator=(const RelacedEeg& eeg) {
|
|
reg = eeg.reg;
|
|
val = eeg.val;
|
|
no_use = eeg.no_use;
|
|
mem = eeg.mem;
|
|
return *this;
|
|
}
|
|
};
|
|
|
|
|
|
|
|
class SymbolManager {
|
|
public:
|
|
SymbolManager(std::shared_ptr<Scope> global_scope, std::string name);
|
|
|
|
/* 当接受一个函数的三地址代码块时，重新初始化 */
|
|
// todo 重新计算待用信息 重置堆栈和寄存器
|
|
// todo 根据block_name找到对应的三地址代码
|
|
void cal_use_info(std::shared_ptr<TACBlock> block);
|
|
void set_scope(std::shared_ptr<Scope> local_scope);
|
|
|
|
std::string encode_var(std::string var) const;
|
|
|
|
std::string rvalue(REG reg);
|
|
REG avalue_reg(const std::string& vairable) const;
|
|
int avalue_mem(const std::string& vairable) const;
|
|
void set_avalue_reg(const std::string& vairable, REG reg);
|
|
void set_avalue_mem(const std::string& variable, int mem);
|
|
void show_reg(REG reg = REG::None);
|
|
void show_mem(int mem = -1);
|
|
|
|
// 待用信息 和 活跃信息
|
|
UseInfo use_info(const std::string& vairable, bool encoded = false) const;
|
|
void set_use_info(const std::string& vairable, UseInfo use_info);
|
|
|
|
REG get_free_reg();
|
|
REG get_reg(std::string dst, std::string src1 = "");
|
|
REG get_reg();
|
|
RelacedEeg get_replaced_reg();
|
|
|
|
void push_reg(REG reg, int overwrite = 1);
|
|
void pop_reg(REG reg);
|
|
int get_esp();
|
|
void set_esp_bias(int bias);
|
|
POSTYPE position(std::string variable);
|
|
int para();
|
|
int ret();
|
|
std::shared_ptr<Symbol> resolve_fun(std::string name);
|
|
std::string get_name();
|
|
void set_zero_len();
|
|
int get_stack_len();
|
|
|
|
|
|
|
|
private:
|
|
// 符号表
|
|
std::string name_;
|
|
std::shared_ptr<Scope> global_scope_;
|
|
std::shared_ptr<Scope> local_scope_;
|
|
std::shared_ptr<Symbol> func_;
|
|
|
|
// 变量存储信息
|
|
std::vector<std::string> rvalue_ = std::vector<std::string>(static_cast<int>(REG::None) + 4); // 地址-寄存器 -> 变量
|
|
std::vector<std::string> svalue_; // 地址-内存 -> 变量
|
|
std::unordered_map<std::string, REG> avalue_reg_; // 变量 -> 地址-寄存器
|
|
std::unordered_map<std::string, int> avalue_mem_; // 变量 -> 地址-内存(存与ebp的偏移)
|
|
|
|
// 变量名 -> 符号表中的 待用信息 和 活跃信息
|
|
std::unordered_map<std::string, UseInfo> use_info_;
|
|
|
|
/* 函数堆栈模拟 */
|
|
int len_; // 当前栈的长度 (4 字节)
|
|
int stack_esp_; // 模拟堆栈的栈顶 (与当前函数栈底的相对值)
|
|
int para_num_;
|
|
int para_now_;
|
|
int ret_num_;
|
|
int ret_now_;
|
|
};
|
|
|
|
#endif // INCLUDE_TCG_SYMBOLMANAGER_H_
|