OoasTools

/**

  *

  *                      OOAS Compiler

  *

  *       Copyright 2015, AIT Austrian Institute of Technology.

  * This code is based on the C# Version of the OOAS Compiler, which is

  * copyright 2015 by the Institute of Software Technology, Graz University

  * of Technology with portions copyright by the 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)

  *               Markus Demetz (AIT)

  *               Christoph Czurda (AIT)

  *

  */

package org.momut.ooas.visitors.analysis;

import java.util.HashMap;

import org.momut.ooas.ast.IAst;

import org.momut.ooas.ast.IScope;

import org.momut.ooas.ast.statements.Block;

import org.momut.ooas.ast.statements.NondetBlock;

import org.momut.ooas.ast.statements.PrioBlock;

import org.momut.ooas.ast.statements.SeqBlock;

import org.momut.ooas.ast.statements.Statement;

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

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

import org.momut.ooas.parser.ParserState;

import org.momut.ooas.utils.exceptions.InternalCompilerException;

import org.momut.ooas.visitors.OoaCompleteAstTraversalVisitor;

public final class OoaCheckASTVisitor extends OoaCompleteAstTraversalVisitor {

        private final String m_tag;

        private final HashMap<String,Type> m_seenTypes = new HashMap<>();

        @Override

        protected void VisitAstElement(IAst element, IAst parent) {

                if (element instanceof Type) {

                        final Type t = (element instanceof OpaqueType) ? ((OpaqueType) element).resolvedType()  : (Type) element;

                        if (t != null) { // opaque might not have resolve yet..

                                final String typeStr = t.toString();

                                if (m_seenTypes.containsKey(typeStr)) {

                                        if (!t.isAnonymousType() &&

                                            m_seenTypes.get(typeStr) != t) // ref eq

                                        {

                                                throw new InternalCompilerException(String.format("Internal Error (%s): AST structure error."

                                                                + " Type instantiated multiple times: %s", m_tag, typeStr));

                                        }

                                } else

                                        m_seenTypes.put(typeStr, t);

                        }

                }

                super.VisitAstElement(element, parent);

        }

        @Override

        public void visit(NondetBlock nondetBlock) {

                super.visit(nondetBlock);

                checkScope(nondetBlock);

        }

        @Override

        public void visit(PrioBlock prioBlock) {

                super.visit(prioBlock);

                checkScope(prioBlock);

        }

        @Override

        public void visit(SeqBlock seqBlock) {

                super.visit(seqBlock);

                checkScope(seqBlock);

        }

        private final void checkScope(Block aBlock) {

                for (final Statement element: aBlock.statements())

                        if (element instanceof IScope) {

                                final IScope child = (IScope) element;

                                if (child.GetParentScope() != aBlock)

                                        throw new InternalCompilerException("Internal Error (%s): AST structure error. ", m_tag);

                        }

        }

        public OoaCheckASTVisitor(ParserState aState, String tag) {

                super(aState);

                m_tag = tag;

        }

}