SymbolScope.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include <unordered_map>
#include <string>
#include <vector>
#include "AstUtil/SharedPtr.hpp"
#include "AstScript/Expr.hpp"
 
AST_NAMESPACE_BEGIN
 
 
 
class SymbolScope{
public:
    using SymbolMap = std::unordered_map<std::string, SharedPtr<Expr>>;
    using SymbolPair = std::pair<std::string, SharedPtr<Expr>>;
    using SymbolList = std::vector<SymbolPair>;
 
    SymbolScope() = default;
 
    explicit SymbolScope(SymbolScope* parent)
        : parent_(parent)
    {}
 
    ~SymbolScope() = default;
 
    bool addSymbol(StringView name, Expr* expr);
 
    void setSymbol(StringView name, Expr* expr);
 
    Expr* findSymbol(StringView name, bool searchParent = true) const;
 
    Expr* resolveSymbol(StringView name);
 
    bool removeSymbol(StringView name);
 
    bool hasSymbol(StringView name, bool searchParent = true) const;
 
    SymbolList getSymbols() const;
 
    SymbolScope* parent() const;
 
    void setParent(SymbolScope* parent);
 
protected:
    SymbolMap    symbols_;              
    SymbolScope* parent_ = nullptr;     
};
 
AST_NAMESPACE_END