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;

import java.util.Collections;

import org.momut.ooas.ast.identifiers.Identifier;

import org.momut.ooas.ast.identifiers.IdentifierKind;

import org.momut.ooas.ast.identifiers.MainModule;

import org.momut.ooas.ast.identifiers.TypeIdentifier;

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

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

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

import org.momut.ooas.parser.ParserState;

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

public final class OoaSymbolSortVisitor extends OoaCompleteAstTraversalVisitor

{

        /* Sorting of enums is not necessarily a good idea in terms of the user wanting to know what's going on

         * however, we need to do this for the ulysses prolog target - otherwise mutations won't be correct..

         * (regardeless in what order the enum symbols were declared, the order must be the same - then we can

         * map them to ints)

         */

        @Override

        public void visit(EnumType enumType)

        {

                Collections.sort(enumType.listOfEnumSymbols());

                super.visit(enumType);

        }

        @Override

        public void visit(MainModule mainModule)

        {

                for (Identifier s: mainModule.symbolTable().symbolList())

                {

                        if (s.kind() == IdentifierKind.TypeIdentifier &&

                                        ((TypeIdentifier)s).type().kind() == TypeKind.OoActionSystemType)

                        {

                                OoActionSystemType ooa = (OoActionSystemType)((TypeIdentifier)s).type();

                                ooa.symbols().Sort();

                        }

                }

                mainModule.symbolTable().Sort();

                super.visit(mainModule);

        }

        public OoaSymbolSortVisitor(ParserState aState)

        {

                super(aState, false);

                if (aState == null)

                        throw new ArgumentException();

        }

}