de.tub.tfs.henshin.tggeditor.actions.exports.ExportInstanceModelAction.java Source code

Introduction

Here is the source code for de.tub.tfs.henshin.tggeditor.actions.exports.ExportInstanceModelAction.java

Source

/*******************************************************************************
 * Copyright (c) 2010-2015 Henshin developers. 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
 *
 * Contributors:
 *     TU Berlin, University of Luxembourg, SES S.A.
 *******************************************************************************/
package de.tub.tfs.henshin.tggeditor.actions.exports;

import java.io.IOException;
import java.util.LinkedList;
import java.util.List;

import org.eclipse.core.runtime.Path;
import org.eclipse.emf.common.ui.dialogs.ResourceDialog;
import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EStructuralFeature;
import org.eclipse.emf.ecore.impl.EStructuralFeatureImpl;
import org.eclipse.emf.henshin.interpreter.util.HenshinEGraph;
import org.eclipse.emf.henshin.model.Graph;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.ui.actions.SelectionAction;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.ui.IWorkbenchPart;
import org.eclipse.ui.PlatformUI;

import de.tub.tfs.henshin.tggeditor.editparts.tree.graphical.GraphTreeEditPart;
import de.tub.tfs.muvitor.ui.utils.EMFModelManager;
//import de.tub.tfs.henshin.editor.editparts.tree.graph.GraphTreeEditPart;
//import de.tub.tfs.henshin.editor.editparts.tree.rule.GraphElementsTreeEditPart;
//import de.tub.tfs.henshin.editor.internal.GraphElementsContainterEObject;
//import de.tub.tfs.henshin.editor.util.IconUtil;

public class ExportInstanceModelAction extends SelectionAction {

    /** The Constant ID. */
    public static final String ID = "henshineditor.actions.ExportInstanceModelAction";

    /** The graph. */
    private Graph graph;

    public ExportInstanceModelAction(IWorkbenchPart part) {
        super(part);
        setId(ID);
        setText("Export an instance model.");
        //      setImageDescriptor(IconUtil.getDescriptor("attr16.png"));
        setToolTipText("Export an instance model.");
    }

    /* (non-Javadoc)
     * @see org.eclipse.gef.ui.actions.WorkbenchPartAction#calculateEnabled()
     */
    @Override
    protected boolean calculateEnabled() {
        List<?> selectedObjects = getSelectedObjects();
        if (selectedObjects.size() != 1) {
            return false;
        }
        Object selectedObject = selectedObjects.get(0);
        if ((selectedObject instanceof EditPart)) {
            EditPart editpart = (EditPart) selectedObject;
            if (editpart instanceof GraphTreeEditPart) {
                graph = (Graph) editpart.getModel();
                return true;
            }
            //         if (editpart instanceof GraphElementsTreeEditPart) {
            //            graph = ((GraphElementsContainterEObject) editpart.getModel()).getModel();
            //            return true;
            //         }
        }
        return false;
    }

    @Override
    public void run() {
        Shell shell = new Shell();
        ResourceDialog dialog = new ResourceDialog(shell, "Please select a file name for the instance graph.",
                SWT.SAVE + SWT.SINGLE);

        //FileDialog dialog = new FileDialog(shell);
        //dialog.setFilterExtensions(new String[]{"*.ecore"});

        //dialog.
        int p = dialog.open();
        if (p == SWT.CANCEL || p == 1) {
            MessageDialog.open(SWT.ERROR,
                    PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActivePart().getSite()
                            .getShell(),
                    "Cannot save instance graph.", "Please select a valid file name.", SWT.SHEET);
            return;
        }
        List<URI> urIs = dialog.getURIs();
        if (urIs.isEmpty()) {
            MessageDialog.open(SWT.ERROR,
                    PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActivePart().getSite()
                            .getShell(),
                    "Cannot save instance graph.", "Please select a valid file name.", SWT.SHEET);
            return;
        }
        //for (URI uri : urIs) {
        try {

            EMFModelManager modelManager = EMFModelManager.createModelManager("");
            Path path = new Path(urIs.get(0).toPlatformString(true));
            //      List<EObject> obj = modelManager.load(path,new LinkedList<EObject>());
            HenshinEGraph gr = new HenshinEGraph(graph);

            //      obj.addAll(gr.getRoots());
            for (EObject eObject : gr.getRoots()) {
                EList<EStructuralFeature> features = eObject.eClass().getEAllStructuralFeatures();
                for (EStructuralFeature feature : features) {
                    if (feature.getDefaultValueLiteral() == null) {
                        ((EStructuralFeatureImpl) feature).setDefaultValueLiteral("");
                    }
                }
            }
            try {

                modelManager.save(path, gr.getRoots().toArray(new EObject[0]));
            } catch (IOException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
            for (EObject eObject : gr.getRoots()) {
                EList<EStructuralFeature> features = eObject.eClass().getEAllStructuralFeatures();
                for (EStructuralFeature feature : features) {
                    if (feature.getDefaultValueLiteral() == null) {
                        feature.setDefaultValueLiteral(null);
                    }
                }
            }

            //}   
            //resourceSet.getURIConverter().getURIMap().put(uri, URI.createFileURI(p));
        } catch (NullPointerException ex) {
            MessageDialog.open(SWT.ERROR,
                    PlatformUI.getWorkbench().getActiveWorkbenchWindow().getActivePage().getActivePart().getSite()
                            .getShell(),
                    "Cannot save instance graph.", "Please select a valid file name.", SWT.SHEET);

        }
        shell.dispose();

        super.run();
    }
}