com.liferay.ide.kaleo.ui.diagram.ShowDiagramPartActionHandler.java Source code

Introduction

Here is the source code for com.liferay.ide.kaleo.ui.diagram.ShowDiagramPartActionHandler.java

Source

/**
 * Copyright (c) 2014 Liferay, Inc. All rights reserved.
 *
 * The contents of this file are subject to the terms of the End User License
 * Agreement for Liferay Developer Studio ("License"). You may not use this file
 * except in compliance with the License. You can obtain a copy of the License
 * by contacting Liferay, Inc. See the License for the specific language
 * governing permissions and limitations under the License, including but not
 * limited to distribution rights of the Software.
 */

package com.liferay.ide.kaleo.ui.diagram;

import com.liferay.ide.kaleo.core.model.Transition;
import com.liferay.ide.kaleo.core.model.WorkflowNode;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.GraphicalViewer;
import org.eclipse.jface.viewers.LabelProvider;
import org.eclipse.sapphire.Element;
import org.eclipse.sapphire.ui.ISapphirePart;
import org.eclipse.sapphire.ui.Presentation;
import org.eclipse.sapphire.ui.SapphireActionHandler;
import org.eclipse.sapphire.ui.SapphirePart;
import org.eclipse.sapphire.ui.diagram.DiagramConnectionPart;
import org.eclipse.sapphire.ui.diagram.editor.DiagramNodePart;
import org.eclipse.sapphire.ui.diagram.editor.SapphireDiagramEditorPagePart;
import org.eclipse.sapphire.ui.forms.swt.SwtPresentation;
import org.eclipse.sapphire.ui.swt.gef.SapphireDiagramEditor;
import org.eclipse.swt.graphics.Image;
import org.eclipse.ui.dialogs.ElementListSelectionDialog;

/**
 * @author Gregory Amerson
 */
public class ShowDiagramPartActionHandler extends SapphireActionHandler {

    @Override
    protected Object run(Presentation context) {
        final SapphireDiagramEditorPagePart diagramPart = context.part()
                .nearest(SapphireDiagramEditorPagePart.class);

        if (diagramPart != null) {
            final LabelProvider labelProvider = new LabelProvider() {
                @Override
                public Image getImage(Object element) {
                    if (element instanceof DiagramNodePart) {
                        DiagramNodePart diagramNodePart = (DiagramNodePart) element;

                        Element modelElement = diagramNodePart.getLocalModelElement();

                        return diagramPart.getSwtResourceCache().image(modelElement.type().image());
                    } else if (element instanceof DiagramConnectionPart) {
                        return diagramPart.getSwtResourceCache().image(Transition.TYPE.image());
                    } else {
                        return diagramPart.getSwtResourceCache().image(WorkflowNode.TYPE.image());
                    }
                }

                @Override
                public String getText(Object element) {
                    if (element instanceof DiagramNodePart) {
                        return ((DiagramNodePart) element).getId();
                    } else if (element instanceof DiagramConnectionPart) {
                        return ((DiagramConnectionPart) element).getLabel();
                    } else {
                        return element != null ? element.toString() : "";
                    }
                }
            };

            final ElementListSelectionDialog dialog = new ElementListSelectionDialog(
                    ((SwtPresentation) context).shell(), labelProvider);

            final List<SapphirePart> parts = new ArrayList<SapphirePart>();
            parts.addAll(diagramPart.getNodes());
            parts.addAll(diagramPart.getConnections());

            dialog.setElements(parts.toArray());
            dialog.setMultipleSelection(false);
            dialog.setHelpAvailable(false);
            dialog.setTitle("Find Part");
            dialog.setMessage("Select part:");

            dialog.open();

            final Object[] result = dialog.getResult();

            if (result != null && result.length == 1) {
                //select node in diagram
                final ISapphirePart part = (ISapphirePart) result[0];

                if (part instanceof DiagramNodePart || part instanceof DiagramConnectionPart) {
                    //                    diagramPart.setSelections( ReadOnlyListFactory.create( part ) );
                    SapphireDiagramEditor diagramEditor = diagramPart.adapt(SapphireDiagramEditor.class);
                    GraphicalViewer viewer = diagramEditor.getGraphicalViewer();

                    GraphicalEditPart editpart = diagramEditor.getGraphicalEditPart(part);

                    if (editpart != null) {
                        // Force a layout first.
                        viewer.flush();
                        viewer.select(editpart);
                        viewer.reveal(editpart);
                    }
                }
            }
        }

        return null;
    }

}