OoasTools

package org.momut.ooas.visitors;

import java.util.ArrayList;

import org.momut.ooas.ast.expressions.Expression;

import org.momut.ooas.ast.expressions.ExpressionKind;

import org.momut.ooas.ast.expressions.ListConstructor;

import org.momut.ooas.ast.expressions.UnaryOperator;

import org.momut.ooas.ast.types.ListType;

import org.momut.ooas.ast.types.Type;

import org.momut.ooas.ast.types.TypeKind;

public class OoaStaticListCastVisitor extends OoaExpressionVisitor

{

        private final Type m_leafType;

        private ListType m_listType;

        public OoaStaticListCastVisitor(ListType targetType)

        {

                m_leafType = leafType(targetType);

                m_listType = targetType;

        }

        @Override

        public void visit(ListConstructor listConstructor)

        {

                final ArrayList<Expression> casts = new ArrayList<>();

                for (final Expression elems: listConstructor.elements()) {

                        if(elems.kind() == ExpressionKind.ListConstr){

                                final ListType parent = m_listType;

                                m_listType = (ListType) m_listType.innerType();

                                VisitSub(elems, listConstructor);

                                m_listType = parent;

                        }

                        else

                        {

                                if(elems.kind() == ExpressionKind.Cast)

                                        elems.SetType(m_leafType);

                                else

                                {

                                        final Expression cast = new UnaryOperator(ExpressionKind.Cast, elems, elems.line(), elems.pos());

                                        cast.SetType(m_leafType);

                                        casts.add(cast);

                                }

                        }

                }

                if(!casts.isEmpty())

                        listConstructor.SetElements(casts);

                listConstructor.SetType(m_listType);

        }

        private Type leafType(Type type)

        {

                Type result = type;

                while(result.kind() == TypeKind.ListType)

                        result = ((ListType)result).innerType();

                return result;

        }

}