ac.soton.eventb.emf.diagrams.generator.actions.GenerateAction.java Source code

Introduction

Here is the source code for ac.soton.eventb.emf.diagrams.generator.actions.GenerateAction.java

Source

/**
 * Copyright (c) 2012 University of Southampton.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 */

package ac.soton.eventb.emf.diagrams.generator.actions;

import java.lang.reflect.InvocationTargetException;

import org.eclipse.core.commands.AbstractHandler;
import org.eclipse.core.commands.ExecutionEvent;
import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.expressions.EvaluationContext;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.gmf.runtime.diagram.ui.resources.editor.parts.DiagramDocumentEditor;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jface.dialogs.ProgressMonitorDialog;
import org.eclipse.jface.operation.IRunnableWithProgress;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.handlers.HandlerUtil;
import org.eventb.emf.core.EventBElement;

import ac.soton.eventb.emf.diagrams.generator.Activator;
import ac.soton.eventb.emf.diagrams.generator.command.GenerateCommand;
import ac.soton.eventb.emf.diagrams.generator.impl.GeneratorFactory;
import ac.soton.eventb.emf.diagrams.generator.impl.Messages;

/**
 * Generate action handler.
 * 
 * @author colin 
 *
 */
public class GenerateAction extends AbstractHandler {

    private boolean enabled = false;

    @Override
    public boolean isEnabled() {
        return enabled && super.isEnabled();
    }

    @Override
    public void setEnabled(Object evaluationContext) {
        super.setEnabled(evaluationContext);
        enabled = false;
        if (evaluationContext instanceof EvaluationContext) {
            EvaluationContext ec = (EvaluationContext) evaluationContext;

            Object editor = ec.getVariable("activeEditor");
            //Object editorID = ec.getVariable("activeEditorId");
            if (editor instanceof DiagramDocumentEditor) {
                final DiagramDocumentEditor diagramDocumentEditor = (DiagramDocumentEditor) editor;
                if (diagramDocumentEditor.getDiagram().getElement() instanceof EventBElement) {
                    enabled = GeneratorFactory.getFactory()
                            .canGenerate(diagramDocumentEditor.getDiagram().getElement().eClass());
                }
            }
        }
    }

    /* (non-Javadoc)
     * @see org.eclipse.core.commands.IHandler#execute(org.eclipse.core.commands.ExecutionEvent)
     */
    @Override
    public Object execute(ExecutionEvent event) throws ExecutionException {

        IEditorPart editor = HandlerUtil.getActiveEditorChecked(event);
        if (editor instanceof DiagramDocumentEditor) {
            final DiagramDocumentEditor diagramDocumentEditor = (DiagramDocumentEditor) editor;

            if (diagramDocumentEditor.getDiagram().getElement() instanceof EventBElement) {
                final EventBElement eventBElement = (EventBElement) diagramDocumentEditor.getDiagram().getElement();

                // save before transformation
                if (editor.isDirty())
                    editor.doSave(new NullProgressMonitor());

                // first validate, then transform
                if (ValidateAction.validate(diagramDocumentEditor)) {

                    final GenerateCommand generateCommand = new GenerateCommand(
                            diagramDocumentEditor.getDiagramEditPart().getEditingDomain(), eventBElement);
                    if (generateCommand.canExecute()) {
                        // run with progress
                        ProgressMonitorDialog dialog = new ProgressMonitorDialog(
                                diagramDocumentEditor.getSite().getShell());
                        try {
                            dialog.run(true, true, new IRunnableWithProgress() {
                                public void run(IProgressMonitor monitor) {
                                    monitor.beginTask(Messages.GENERATOR_MSG_05, IProgressMonitor.UNKNOWN);
                                    try {
                                        generateCommand.execute(monitor, diagramDocumentEditor);
                                    } catch (ExecutionException e) {
                                        Activator.logError(Messages.GENERATOR_MSG_06, e);
                                    }
                                    monitor.done();
                                }
                            });
                        } catch (InvocationTargetException e) {
                            Activator.logError(Messages.GENERATOR_MSG_07, e);
                            return null;
                        } catch (InterruptedException e) {
                            Activator.logError(Messages.GENERATOR_MSG_08, e);
                            return null;
                        }

                        // error feedback
                        if (false == generateCommand.getCommandResult().getStatus().isOK())
                            MessageDialog.openError(editor.getSite().getShell(), Messages.GENERATOR_MSG_09,
                                    Messages.GENERATOR_MSG_10);
                    }
                }
            }
        }
        return null;
    }
}