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

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

        }

}