de.tub.tfs.henshin.editor.editparts.graph.graphical.GraphEditPart.java Source code

Java tutorial

Introduction

Here is the source code for de.tub.tfs.henshin.editor.editparts.graph.graphical.GraphEditPart.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.editor.editparts.graph.graphical;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.eclipse.draw2d.ColorConstants;
import org.eclipse.draw2d.ConnectionLayer;
import org.eclipse.draw2d.ConnectionRouter;
import org.eclipse.draw2d.FreeformLayer;
import org.eclipse.draw2d.FreeformLayout;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.Label;
import org.eclipse.draw2d.LineBorder;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.henshin.model.Graph;
import org.eclipse.emf.henshin.model.HenshinPackage;
import org.eclipse.emf.henshin.model.Node;
import org.eclipse.emf.henshin.model.Rule;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gef.LayerConstants;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.swt.SWT;
import org.eclipse.swt.widgets.Display;
import org.eclipse.ui.views.properties.IPropertySource;

import de.tub.tfs.henshin.editor.editparts.HenshinEditPolicy;
import de.tub.tfs.henshin.editor.editparts.graph.GraphClipboardEditPolicy;
import de.tub.tfs.henshin.editor.figure.graph.EdgeConnectionRouter;
import de.tub.tfs.henshin.editor.model.properties.graph.GraphPropertySource;
import de.tub.tfs.henshin.editor.util.HenshinCache;
import de.tub.tfs.muvitor.gef.editparts.AdapterGraphicalEditPart;
import de.tub.tfs.muvitor.ui.utils.SWTResourceManager;

/**
 * The Class GraphEditPart.
 */
public class GraphEditPart extends AdapterGraphicalEditPart<Graph> {

    /** The name label. */
    private Label nameLabel = new Label();

    /**
     * Instantiates a new graph edit part.
     * 
     * @param model
     *            the model
     */
    public GraphEditPart(Graph model) {
        super(model);
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editparts.AbstractGraphicalEditPart#activate()
     */
    @Override
    public void activate() {
        updateFigure();
        nameLabel = new Label();
        nameLabel.setFont(SWTResourceManager.getFont("Sans", 14, SWT.BOLD)); //$NON-NLS-1$
        nameLabel.setForegroundColor(Display.getCurrent().getSystemColor(SWT.COLOR_GRAY));
        nameLabel.setText(getText());
        getFigure().add(nameLabel, new Rectangle(10, 10, -1, -1));

        HenshinCache.getInstance().init();
        super.activate();
    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editparts.AbstractGraphicalEditPart#createFigure()
     */
    @Override
    protected IFigure createFigure() {
        FreeformLayer layer = new FreeformLayer();
        layer.setLayoutManager(new FreeformLayout());
        ConnectionLayer cLayer = (ConnectionLayer) getLayer(LayerConstants.CONNECTION_LAYER);
        cLayer.setAntialias(SWT.ON);
        EdgeConnectionRouter edgeRouter = new EdgeConnectionRouter(layer);
        if (this.getCastedModel().getEdges().size() > 1000) {
            edgeRouter.setNextRouter(ConnectionRouter.NULL);

            MessageDialog.open(MessageDialog.INFORMATION, Display.getDefault().getActiveShell(), "Information",
                    "The graph has too many edges. Therefore the edges will not be drawn around nodes.", SWT.SHEET);

        }
        cLayer.setConnectionRouter(edgeRouter);

        return layer;
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.eclipse.gef.editparts.AbstractGraphicalEditPart#createEditPolicies()
     */
    @Override
    protected void createEditPolicies() {
        installEditPolicy(EditPolicy.LAYOUT_ROLE, new GraphXYLayoutEditPolicy());

        installEditPolicy(HenshinEditPolicy.CLIPBOARD_ROLE, new GraphClipboardEditPolicy() {
            @Override
            protected boolean canCopy() {
                return false;
            }
        });
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.eclipse.gef.editparts.AbstractGraphicalEditPart#getModelChildren()
     */
    @Override
    protected List<EObject> getModelChildren() {
        List<EObject> list = new ArrayList<EObject>();
        list.addAll(getCastedModel().getNodes());

        return list;
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * muvitorkit.gef.editparts.AdapterGraphicalEditPart#notifyChanged(org.eclipse
     * .emf.common.notify.Notification)
     */
    @Override
    protected void notifyChanged(Notification notification) {
        final int featureId = notification.getFeatureID(HenshinPackage.class);
        if (featureId == HenshinPackage.GRAPH__NAME) {
            if (getCastedModel().getName() != null) {
                nameLabel.setText(getCastedModel().getName());
            }
        }
        if (featureId == HenshinPackage.RULE__MAPPINGS) {
            if (notification.getNotifier() instanceof Rule) {
                for (Node node : this.getCastedModel().getNodes()) {
                    node.eNotify(notification);
                }
            }
            refresh();
        }
        final int type = notification.getEventType();
        switch (type) {
        case Notification.ADD:
        case Notification.ADD_MANY:
        case Notification.REMOVE:
        case Notification.REMOVE_MANY:
            refreshChildren();
            refreshVisuals();
            break;
        case Notification.SET:
            repaintChildren();
            refreshVisuals();
            break;
        }
    }

    /**
     * Update figure.
     */
    protected void updateFigure() {
        final FreeformLayer layer = (FreeformLayer) getFigure();
        getFigure().setBackgroundColor(ColorConstants.white);
        if (getSelected() == 0) { // not selected
            layer.setBorder(new LineBorder(1));
            layer.setForegroundColor(SWTResourceManager.getColor(0, 0, 0));
        } else { // selected
            layer.setBorder(new LineBorder(5));
            layer.setForegroundColor(SWTResourceManager.getColor(150, 0, 0));
        }

    }

    /*
     * (non-Javadoc)
     * 
     * @see org.eclipse.gef.editparts.AbstractGraphicalEditPart#refreshVisuals()
     */
    @Override
    public void refreshVisuals() {
        updateFigure();
        getFigure().repaint();
        super.refreshVisuals();
    }

    /*
     * (non-Javadoc)
     * 
     * @see
     * org.eclipse.gef.editparts.AbstractGraphicalEditPart#fireSelectionChanged
     * ()
     */
    @Override
    protected void fireSelectionChanged() {
        super.fireSelectionChanged();
        refreshVisuals();
    }

    /*
     * (non-Javadoc)
     * 
     * @see muvitorkit.gef.editparts.AdapterTreeEditPart#createPropertySource()
     */
    @Override
    protected IPropertySource createPropertySource() {
        return new GraphPropertySource(getCastedModel());
    }

    /**
     * @return
     */
    protected String getText() {
        if (getCastedModel().getName() == null) {
            return new String();
        }
        return getCastedModel().getName();
    }

    /**
     * Repaint children.
     */
    public void repaintChildren() {
        for (Object e : getChildren()) {
            if (e instanceof NodeEditPart) {
                ((NodeEditPart) e).getFigure().repaint();
            }
        }
    }

    @Override
    public void removeChild(EditPart child) {
        super.removeChild(child);
    }

    @Override
    public void addChild(EditPart child, int index) {
        super.addChild(child, index);
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    @Override
    protected void refreshChildren() {
        int i;
        EditPart editPart;
        Object model;

        List children = getChildren();
        int size = children.size();
        Map modelToEditPart = Collections.EMPTY_MAP;
        if (size > 0) {
            modelToEditPart = new HashMap(size);
            for (i = 0; i < size; i++) {
                editPart = (EditPart) children.get(i);
                modelToEditPart.put(editPart.getModel(), editPart);
            }
        }

        List modelObjects = getModelChildren();
        for (i = 0; i < modelObjects.size(); i++) {
            model = modelObjects.get(i);

            if (!HenshinCache.getInstance().getRemovedEditParts().contains(model)) {
                // Do a quick check to see if editPart[i] == model[i]
                if (i < children.size() && ((EditPart) children.get(i)).getModel() == model) {
                    continue;
                }

                // Look to see if the EditPart is already around but in the
                // wrong location
                editPart = (EditPart) modelToEditPart.get(model);
                if (editPart != null) {
                    reorderChild(editPart, children.indexOf(editPart));
                } else {
                    // An EditPart for this model doesn't exist yet. Create and
                    // insert one.
                    editPart = createChild(model);
                    addChild(editPart, size++);
                }
            }
        }

        // remove the remaining EditParts
        List<NodeEditPart> toDelete = new ArrayList<NodeEditPart>();

        for (Object object : children) {
            if (object instanceof NodeEditPart) {
                NodeEditPart nodeEditPart = (NodeEditPart) object;
                if (!modelObjects.contains(nodeEditPart.getCastedModel())) {
                    toDelete.add(nodeEditPart);
                }
            }
        }

        for (NodeEditPart nodeEditPart : toDelete) {
            removeChild(nodeEditPart);
        }
    }

    @Override
    public void deactivate() {
        HenshinCache.getInstance().init();
        super.deactivate();
    }

}