53 lines
1.4 KiB
C
53 lines
1.4 KiB
C
|
// symbol.h
|
||
|
|
#ifndef MINI_C_SYMBOL_H
|
||
|
|
#define MINI_C_SYMBOL_H
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
#include <unordered_map>
|
||
|
|
#include <memory>
|
||
|
|
#include "type.h" // 预计后面定义 Type 类
|
||
|
|
#include "ast.h"
|
||
|
|
|
||
|
|
// 一条符号 信息：名称、类型、声明位置、存储类别等
|
||
|
|
struct Symbol {
|
||
|
|
std::string name;
|
||
|
|
std::shared_ptr<Type> type;
|
||
|
|
StorageClass storage;
|
||
|
|
SourceLoc decl_loc; // 定义时的行列
|
||
|
|
|
||
|
|
Symbol(const std::string& n,
|
||
|
|
std::shared_ptr<Type> t,
|
||
|
|
StorageClass s,
|
||
|
|
SourceLoc loc)
|
||
|
|
: name(n), type(std::move(t)), storage(s), decl_loc(loc) {}
|
||
|
|
};
|
||
|
|
|
||
|
|
// 作用域：一组符号 + 指向外层作用域
|
||
|
|
class Scope : public std::enable_shared_from_this<Scope> {
|
||
|
|
public:
|
||
|
|
explicit Scope(std::shared_ptr<Scope> parent = nullptr)
|
||
|
|
: parent_(std::move(parent)) {}
|
||
|
|
bool insert(const Symbol& sym) {
|
||
|
|
auto [it, ok] = syms_.emplace(sym.name, sym);
|
||
|
|
return ok;
|
||
|
|
}
|
||
|
|
Symbol* lookup(const std::string& name) {
|
||
|
|
auto it = syms_.find(name);
|
||
|
|
if (it != syms_.end()) return &it->second;
|
||
|
|
if (parent_) return parent_->lookup(name);
|
||
|
|
return nullptr;
|
||
|
|
}
|
||
|
|
std::shared_ptr<Scope> push() {
|
||
|
|
return std::make_shared<Scope>(shared_from_this());
|
||
|
|
}
|
||
|
|
std::shared_ptr<Scope> pop() {
|
||
|
|
return parent_;
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
std::unordered_map<std::string, Symbol> syms_;
|
||
|
|
std::shared_ptr<Scope> parent_;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif // MINI_C_SYMBOL_H
|