LexicalContext.h

Go to the documentation of this file.

#pragma once
#include "util/Utilities.h"
#include <functional>
#include <memory>
#include <ostream>
#include <set>


enum NameQueryResult {
  nameFound,
  nameNotFound,
};

template <class T, class Scope> class LexicalContext {
  std::set<std::string> symbols;
  std::unordered_map<std::string, T> symbols_to_t;
 
public:
  LexicalContext *parent_scope;
  std::optional<Scope> s;
 
  explicit LexicalContext() : symbols(), symbols_to_t(), parent_scope() {}
  LexicalContext(LexicalContext *parent_scope)
      : symbols(), symbols_to_t(), parent_scope(parent_scope) {}
 
  void setScope(Scope s) { this->s = s; }
 
  void registerNameT(const std::string &name, T l) {
    symbols.insert(name);
    symbols_to_t[name] = l;
  }
  NameQueryResult queryName(const std::string &name) const {
    return symbols.contains(name) ? nameFound : nameNotFound;
  }
  T get_from_name(const std::string &name) const {
    return symbols_to_t.at(name);
  }
 
  T recursive_get_from_name(const std::string &name) const {
    if (symbols.find(name) != symbols.end())
      return symbols_to_t.at(name);
    else if (parent_scope == nullptr) {
      sammine_util::abort(fmt::format("name {} not found", name));
    } else
      return parent_scope->recursive_get_from_name(name);
  }
 
  NameQueryResult recursiveQueryName(const std::string &name) const {
    if (symbols.find(name) != symbols.end())
      return nameFound;
    else if (parent_scope == nullptr) {
      return nameNotFound;
    } else
      return parent_scope->recursiveQueryName(name);
  }
}; // namespace sammine_lang::AST