OoasTools

/**

  *

  *                  OOAS Compiler - C++ AST

  *

  * Copyright 2015, AIT Austrian Institute of Technology.

  * All rights reserved.

  *

  * SEE THE "LICENSE" FILE FOR THE TERMS UNDER WHICH THIS FILE IS PROVIDED.

  *

  * If you modify the file please update the list of contributors below to in-

  * clude your name. Please also stick to the coding convention of using TABs

  * to do the basic (block-level) indentation and spaces for anything after

  * that. (Enable the display of special chars and it should be pretty obvious

  * what this means.) Also, remove all trailing whitespace.

  *

  * Contributors:

  *               Willibald Krenn (AIT)

  *               Stephan Zimmerer (AIT)

  *               Christoph Czurda (AIT)

  *

  */

#pragma once

#include <cstdint>

#include <unordered_map>

#include <unordered_set>

#include <list>

#include <string>

#include <ast/identifiers/Identifier.hpp>

#include <base/Exceptions.hpp>

namespace Ast {

class SymbolTable: public Base::TiresiasObject {

protected:

        std::unordered_map<std::string, Identifier*> m_symbols;

        std::unordered_set<const Identifier*> m_identifiers;

        SymbolTable():

                TiresiasObject(),

                m_symbols(),

                m_identifiers()

        {};

        SymbolTable(const SymbolTable &toCopy):

                TiresiasObject(),

                m_symbols(toCopy.m_symbols),

                m_identifiers (toCopy.m_identifiers)

        {};

public:

        friend class Ast;

        std::unordered_map<std::string, Identifier*>::iterator begin() {return m_symbols.begin();}

        std::unordered_map<std::string, Identifier*>::iterator end() {return m_symbols.end();}

        std::unordered_set<const Identifier*> identifiers() {return m_identifiers;}

        std::uint64_t size() {return m_symbols.size();}

        bool defined(const Identifier* aValue) { return m_identifiers.count(aValue) > 0; }

        bool defined(std::string& aKey)  { return m_symbols.find(aKey) != m_symbols.end(); }

        Identifier* get(const std::string& aKey) {

                std::unordered_map<std::string, Identifier*>::iterator iter = m_symbols.find(aKey);

                if (iter == m_symbols.end())

                        return nullptr;

                return iter->second;

        }

        Identifier* get(const char* aKey);

        bool addIdentifier(Identifier* anIdent);

        void setSymbolList(std::list<Identifier*>& newList);

};

}