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;

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);

	}

}