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.ArrayList;

import java.util.Collections;

import java.util.IdentityHashMap;

import java.util.Map.Entry;

import org.momut.ooas.ast.IScope;

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

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

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.statements.Call;

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

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

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

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

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

import org.momut.ooas.ast.types.FunctionType.FunctionTypeEnum;

import org.momut.ooas.parser.ParserMessage;

import org.momut.ooas.parser.ParserState;

import org.momut.ooas.parser.ParserWarning;

import org.momut.ooas.visitors.OoaCompleteAstTraversalVisitor;

import org.momut.ooas.visitors.OoaPrintVisitor;

public class OoaPrintActionSequencesVisitor extends OoaCompleteAstTraversalVisitor {

        private final IdentityHashMap<String, Integer> m_sequences = new IdentityHashMap<>();

        private boolean m_warned = false;

        @Override

        public void visit(SeqBlock seqBlock) {

                super.visit(seqBlock);

                boolean isFunction = false;

                IScope scope = seqBlock.GetParentScope();

                if (scope == null && !m_warned) {

                        m_ParserState.AddWarningMessage(new ParserWarning("", 0, 0, "Sequence analysis will be incomplete since the parent scope of a seq-block could not be resolved."));

                        m_warned = true;

                }

                while (!(scope instanceof OoActionSystemType) && !isFunction) {

                        isFunction = scope instanceof FunctionIdentifier;

                        scope = scope.GetParentScope();

                }

                if (isFunction)

                        return;

                int index = 0;

                int next = 0;

                final ArrayList<String> m_calls = new ArrayList<>();

                for (final Statement s: seqBlock.statements()) {

                        // get a list of call statements

                        if ( s.kind()== StatementKind.MethodCall

                             && index == next )

                        {

                                final OoaPrintVisitor customFormatter = new OoaPrintVisitor(null){

                                        @Override

                                        public void visit(IdentifierExpression identifierExpression) {

                                                final Identifier id = identifierExpression.identifier();

                                                if (id.kind() == IdentifierKind.NamedActionIdentifier) {

                                                        final FunctionType ft = (FunctionType) id.type();

                                                        if (ft.functionType() == FunctionTypeEnum.Observable)

                                                                output.Append("obs ");

                                                        else if (ft.functionType() == FunctionTypeEnum.Controllable)

                                                                output.Append("ctr ");

                                                }

                                                output.Append(identifierExpression.identifier().tokenText());

                                        }

                                };

                                final Call c = (Call) s;

                                m_calls.add( c.callExpression().toString(customFormatter));

                                next++;

                        }

                        index++;

                }

                if (m_calls.size() > 1) {

                        final String seq = m_calls.toString().intern();

                        if (!m_sequences.containsKey(seq))

                                m_sequences.put(seq, 0);

                        m_sequences.put(seq, m_sequences.get(seq) + 1);

                }

        }

        @Override

        public void visit(MainModule mainModule) {

                // we only care about the instantiated names...

                // if you want both, then just call super here..

                final boolean haveInstance = mainModule.instance() != null;

                if (haveInstance)

                        VisitSub(mainModule.instance(), mainModule);

                final StringBuilder b = new StringBuilder();

                b.append("List of sequential actions in model:");

                b.append(System.lineSeparator());

                final ArrayList<String> output = new ArrayList<>();

                for (final Entry<String, Integer>e: m_sequences.entrySet())

                        output.add(String.format("\t%s: %s", e.getValue(), e.getKey()));

                Collections.sort(output);

                for (final String s:output)

                        b.append(s + System.lineSeparator());

                m_ParserState.AddMessage(new ParserMessage("",0,0, b.toString()));

//                System.out.println(b.toString());

        }

        public OoaPrintActionSequencesVisitor(ParserState aState) {

                super(aState);

        }

}