plugins/org.eclipse.bpel.common.ui.noEmbeddedEditors/src/org/eclipse/bpel/common/ui/decorator/LinkEditPartMarkerDecorator.java

/*******************************************************************************
 * Copyright (c) 2005 IBM Corporation and others.
 * 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:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.bpel.common.ui.decorator;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

import org.eclipse.bpel.common.ui.markers.IModelMarkerConstants;
import org.eclipse.core.resources.IMarker;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.draw2d.Connection;
import org.eclipse.draw2d.ConnectionEndpointLocator;
import org.eclipse.draw2d.ConnectionLocator;
import org.eclipse.draw2d.IFigure;
import org.eclipse.emf.ecore.EObject;


/**
 * Default implementation of marker behavior for a connection edit part. 
 * 
 */
public class LinkEditPartMarkerDecorator extends EditPartMarkerDecorator {
	
	private Connection fConnection;
	private List<IFigure> fMarkerFigures = new ArrayList<IFigure>();
	
	/**
	 * Brand new shiny LinkEditPartMarkerDecorator
	 * @param aModelObject
	 * @param aConnection
	 */
	
	public LinkEditPartMarkerDecorator(EObject aModelObject, Connection aConnection) {
		super(aModelObject);
		this.fConnection = aConnection;
	}
	
	/**
	 * May be overridden by subclasses.
	 * 
	 * @param marker 
	 * @return a layout constraint
	 */
	
	@Override
	protected Object getConstraint(IMarker marker) {
		try {
			if (marker.isSubtypeOf(IModelMarkerConstants.DECORATION_GRAPHICAL_MARKER_ID)) {
				String anchorString = (String)marker.getAttribute(IModelMarkerConstants.DECORATION_GRAPHICAL_MARKER_ANCHOR_POINT_ATTR);
				if (anchorString != null) {
					if (anchorString.equals(IMarkerConstants.MARKER_ANCHORPOINT_SOURCE)) {
						ConnectionEndpointLocator locator = new ConnectionEndpointLocator(fConnection, false);
						locator.setUDistance(4);
						locator.setVDistance(0);
						return locator;
					}
					if (anchorString.equals(IMarkerConstants.MARKER_ANCHORPOINT_TARGET)) {
						ConnectionEndpointLocator locator = new ConnectionEndpointLocator(fConnection, true);
						locator.setUDistance(4);
						locator.setVDistance(0);
						return locator;
					}
					if (anchorString.equals(IMarkerConstants.MARKER_ANCHORPOINT_CENTRE)) return new ConnectionLocator(fConnection, ConnectionLocator.MIDDLE);
				}
			}
		} catch (CoreException e) {
			// ignore this exception since the marker may no longer exist
		}

		// default
		return new ConnectionLocator(fConnection, ConnectionLocator.MIDDLE);
	}

	/**
	 * 
	 */
	public void removeAllMarkerFigures() {
		for (IFigure figure : fMarkerFigures) {
			fConnection.remove(figure);
		}
		fMarkerFigures.clear();
	}
	
	/**
	 * Add marker figure.
	 * 
	 * @param figure
	 */
	
	public void addMarkerFigure(IFigure figure){
		fMarkerFigures.add(figure);
	}
	
	/**
	 * @return Returns the conn.
	 */
	public Connection getConnection() {
		return fConnection;
	}
	
	
	/**
	 * Draws the markers. This method should be called from the EditPart's refreshVisuals()
	 * method.
	 */
	@Override
	protected void refreshMarkers() {
		//	Refresh any decorations on this edit part
		 if(getConnection() != null) {
		 	 removeAllMarkerFigures();
	
		 	 Map<Object, IMarker> constraintToMarkerMap = getMarkerMap();
			 for(Iterator iter = constraintToMarkerMap.entrySet().iterator(); iter.hasNext(); ) {
			 	Map.Entry entry = (Map.Entry) iter.next();
			 	Object constraint = entry.getKey();
			 	IMarker marker = (IMarker) entry.getValue();
			 	IFigure markerFigure = createFigureForMarker(marker);
			 	if (markerFigure != null) {
			 		addMarkerFigure(markerFigure);
			 		getConnection().add(markerFigure, constraint);
			 	}
			 }
		 }
	}
}