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 <ast/expressions/TypeConstructionExpression.hpp>

#include <ast/IScope.hpp>

#include <deque>

#include <ast/SymbolTable.hpp>

#include <base/Exceptions.hpp>

#include <string>

#include <ast/IAstVisitor.hpp>

namespace Ast {

class ListConstructor final

        : public TypeConstructionExpression

        , public IScope

{

protected:

        ExpressionList m_elements;

        IScope* m_parentScope;

        SymbolTable* m_comprehensionVars;

        Expression* m_comprehension;

        bool m_hasComprehension;

        ListConstructor():

                TypeConstructionExpression(ExpressionKind::ListConstr),

                m_elements(),

                m_parentScope(nullptr),

                m_comprehensionVars(nullptr),

                m_comprehension(nullptr),

                m_hasComprehension(false)

        {};

        ListConstructor(const ListConstructor &toCopy):

                TypeConstructionExpression(toCopy),

                m_elements(toCopy.m_elements),

                m_parentScope(toCopy.m_parentScope),

                m_comprehensionVars(toCopy.m_comprehensionVars),

                m_comprehension(toCopy.m_comprehension),

                m_hasComprehension(toCopy.m_hasComprehension)

        {};

        static ListConstructor* create(ExpressionKind){return new ListConstructor();}

        static ListConstructor* createCopy(const ListConstructor& toCopy){return new ListConstructor(toCopy);}

public:

        friend class Ast;

        friend class Expression;

        void init(std::int32_t line, std::int32_t pos,

                        Type* typeRef,IdentifierList* callTargetsIdentifierListRef, SymbolTable* symbTabRef,

                        ExpressionList* exprList, SymbolTable *symTab, IScope* parentScope,

                        Expression* comprehension, bool hasComprehension)

        {

                TypeConstructionExpression::init(line,pos,typeRef,callTargetsIdentifierListRef, symbTabRef);

                m_elements = std::move(*exprList);

                m_parentScope = parentScope;

                m_comprehensionVars = symTab;

                m_comprehension = comprehension;

                m_hasComprehension = hasComprehension;

        }

        ExpressionList& elements() {return m_elements;};

        Expression* comprehension() {return m_comprehension;};

        bool hasComprehension() {return m_hasComprehension;};

        SymbolTable* comprehensionVariables() {return m_comprehensionVars;};

        void addElement(Expression* anElement) {m_elements.push_back(anElement);};

        void setComprehension(Expression* comprehension) {m_comprehension = comprehension;};

        void setHasComprehension(bool flag) {m_hasComprehension = flag;};

        Identifier* resolveIdentifier(const std::string& aName) const override { return m_comprehensionVars->get(aName); };

        IScope* getParentScope() const override {return m_parentScope;};

        std::string getScopeName() const override {return "";}

        void setParentScope(IScope* parentScope) override {m_parentScope = parentScope;};

        void addIdentifier(Identifier* anIdentifier, void* ) override {m_comprehensionVars->addIdentifier(anIdentifier);};

        void accept(IAstVisitor& visitor) override {visitor.visit(this);};

//        void setElements(std::deque<Expression*>& newElements) {};

        IScope* asScope() final {return (IScope*) this;}

};

}