RelationEndpointEditPolicy.java

package com.isencia.passerelle.workbench.model.editor.ui.editpolicy;

import org.eclipse.draw2d.PolylineConnection;
import org.eclipse.gef.GraphicalEditPart;

public class RelationEndpointEditPolicy extends
		org.eclipse.gef.editpolicies.ConnectionEndpointEditPolicy {

	protected void addSelectionHandles() {
		if (getConnectionFigure() != null) {
			super.addSelectionHandles();
			getConnectionFigure().setLineWidth(2);
		}
	}

	protected PolylineConnection getConnectionFigure() {
		return (PolylineConnection) ((GraphicalEditPart) getHost()).getFigure();
	}

	protected void removeSelectionHandles() {
		if (getConnectionFigure() != null) {
			super.removeSelectionHandles();
			getConnectionFigure().setLineWidth(0);
		}
	}

}