org.eclipse.gmf.runtime.draw2d.ui/src/org/eclipse/gmf/runtime/draw2d/ui/internal/figures/FeedbackConnection.java

/******************************************************************************
 * Copyright (c) 2002, 2007 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.gmf.runtime.draw2d.ui.internal.figures;

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

import org.eclipse.draw2d.AbsoluteBendpoint;
import org.eclipse.draw2d.Bendpoint;
import org.eclipse.draw2d.Connection;
import org.eclipse.draw2d.ConnectionRouter;
import org.eclipse.draw2d.Graphics;
import org.eclipse.draw2d.geometry.PointList;
import org.eclipse.gmf.runtime.draw2d.ui.figures.PolylineConnectionEx;
import org.eclipse.gmf.runtime.draw2d.ui.geometry.PointListUtilities;

public class FeedbackConnection extends PolylineConnectionEx {

	private Connection master = null;
	private List masterConstraint = null;

	public FeedbackConnection() {
		setLineStyle(Graphics.LINE_DASHDOT);
	}
		
	public FeedbackConnection(Connection master) {
		this();
		
		this.master = master;
		
		PointList pts = PointListUtilities.copyPoints(master.getPoints());
		setPoints(pts);
		setSourceAnchor(master.getSourceAnchor());
		setTargetAnchor(master.getTargetAnchor());
		setForegroundColor(master.getForegroundColor());
		
		if (master instanceof PolylineConnectionEx) {
			setSmoothness(((PolylineConnectionEx)master).getSmoothness());
			setRoutingStyles(((PolylineConnectionEx)master).isClosestDistanceRouting(),
							 ((PolylineConnectionEx)master).isAvoidObstacleRouting());
		}
			
		// Update the constraint based on the current figure
		ConnectionRouter masterRouter = master.getConnectionRouter();
		
		masterConstraint = (List)master.getRoutingConstraint();
		List newConstraint = new ArrayList(masterConstraint.size());
		
		setConnectionRouter(masterRouter);
		masterRouter.remove(master);

		for (int i=0; i<masterConstraint.size(); i++) {
			Bendpoint bp = (Bendpoint)masterConstraint.get(i);
			newConstraint.add( new AbsoluteBendpoint(bp.getLocation()) );
		}
		
        assert newConstraint.size() >= 2;
        assert newConstraint.size() == masterConstraint.size(); 
		setRoutingConstraint(newConstraint);
	}
	
	public void dispose() {
		if (getConnectionRouter() != null)
			getConnectionRouter().remove(this);
		setRoutingConstraint(null);
		
		if (getSourceAnchor() != null)
			getSourceAnchor().removeAnchorListener(this);
		if (getTargetAnchor() != null)
			getTargetAnchor().removeAnchorListener(this);

		if (master != null && masterConstraint != null) {
			master.setRoutingConstraint(masterConstraint);
		}
	}
}