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

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

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

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

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

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

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

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

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

import org.momut.ooas.codegen.prolog.OoaPrologIdentifier;

import org.momut.ooas.parser.ParserMessage;

import org.momut.ooas.parser.ParserState;

import org.momut.ooas.visitors.OoaCompleteAstTraversalVisitor;

public class OoaPrintActionsVisitor extends OoaCompleteAstTraversalVisitor {

	private final ArrayList<String> m_result = new ArrayList<>();

	private final OoaPrologIdentifier.Factory m_idFactory;

	@SuppressWarnings("unused")

	private void printNamedActions(MainModule mainModule)

	{

		for (final Identifier x: mainModule.instance().symbols().symbolList())

		{

			if (x.kind() != IdentifierKind.NamedActionIdentifier)

				continue;

			final NamedActionIdentifier action = (NamedActionIdentifier)x;

			final FunctionType actionType = (FunctionType) action.type();

			String ctrObs = "";

			switch(actionType.functionType()) {

				default:

					// something we do not know...

					throw new UnsupportedOperationException();

				case Continuous:

					// we do not support continuous actions here

					throw new UnsupportedOperationException();

				case Internal:

				case Method:

					// we do not print methods or internal actions

					continue;

				case Observable:

					// yes, report it

					ctrObs = "obs";

					break;

				case Controllable:

					// yes, report it

					ctrObs = "ctr";

					break;

			}

			final OoaPrologIdentifier pIdent = m_idFactory.create();

			action.Accept(pIdent);

			final StringBuilder parameter = new StringBuilder();

			if (action.parameter().size() > 0) {

				parameter.append("(");

				int z = 0;

				for (final ParameterIdentifier param: action.parameter()) {

					if (z != 0) {

						parameter.append(", ");

					} else

						z++;

					final OoaPrologIdentifier pParamIdent = m_idFactory.create();

					param.Accept(pParamIdent);

					parameter.append(pParamIdent.toString());

				}

				parameter.append(")");

			}

			m_result.add(String.format("%s %s%s", ctrObs, pIdent.toString().replaceAll("'", ""), parameter.toString()));

		}

	}

	@Override

	public void visit(MainModule mainModule) {

		//printNamedActions(mainModule);

		mainModule.instance().doOdBlock().Accept(this);

		final StringBuilder b = new StringBuilder();

		b.append("List of controllable/observable actions used:");

		b.append(System.lineSeparator());

		Collections.sort(m_result);

		int i = 0;

		for (final String s: m_result) {

			i ++;

			b.append(String.format("\t%s:\t%s",i,s));

			b.append(System.lineSeparator());

		}

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

	}

	HashSet<String> m_seenActions = new HashSet<String>();

	@Override

	public void visit(CallExpression e)

	{

		final FunctionType actionType = (FunctionType) e.child().type();

		String ctrObs = "";

		switch(actionType.functionType()) {

			default:

				// something we do not know...

				throw new UnsupportedOperationException();

			case Continuous:

				// we do not support continuous actions here

				throw new UnsupportedOperationException();

			case Internal:

			case Method:

				// we do not print methods or internal actions

				//continue;

				return;

			case Observable:

				// yes, report it

				ctrObs = "obs";

				break;

			case Controllable:

				// yes, report it

				ctrObs = "ctr";

				break;

		}

		final OoaPrologIdentifier pIdent = m_idFactory.create();

		e.child().Accept(pIdent);

		final StringBuilder parameter = new StringBuilder();

		if (e.arguments().size() > 0) {

			int z = 0;

			for (final UlyssesType ut: actionType.parameter()) {

				if (e.arguments().size() <= z)

					continue;

				Expression arg = e.arguments().get(z);

				while (arg.kind() == ExpressionKind.Cast)

					arg = ((UnaryOperator)arg).child();

				if (z != 0) {

					parameter.append(", ");

				} else

					z++;

				parameter.append(ut.toString());

				parameter.append(" #: ");

				parameter.append( Integer.toString(ut.valueCount()) );

				parameter.append(" [call-type: ");

				parameter.append(arg.type().toString());

				parameter.append(" #: ");

				parameter.append( Integer.toString(arg.type().valueCount()) );

				parameter.append("]");

			}

		}

		final String actionString = String.format("%s %s(%s)", ctrObs, pIdent.toString().replaceAll("'", ""), parameter.toString());

		if (!m_seenActions.contains(actionString)) {

			m_seenActions.add(actionString);

			m_result.add(actionString);

		}

	};

	@Override

	public String toString() {

		return m_result.toString();

	}

	public OoaPrintActionsVisitor(ParserState aState) {

		super(aState);

		m_idFactory = new OoaPrologIdentifier.Factory();

	}

}