bundles/org.eclipse.draw2d/src/org/eclipse/draw2d/graph/RouteEdges.java

/*******************************************************************************
 * Copyright (c) 2003, 2010 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.draw2d.graph;

import org.eclipse.draw2d.geometry.Insets;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.draw2d.geometry.PointList;
import org.eclipse.draw2d.geometry.Rectangle;

/**
 * @author Randy Hudson
 */
class RouteEdges extends GraphVisitor {

	/**
	 * @see GraphVisitor#visit(DirectedGraph)
	 */
	public void revisit(DirectedGraph g) {
		for (int i = 0; i < g.edges.size(); i++) {
			Edge edge = (Edge) g.edges.get(i);
			edge.start = new Point(edge.getSourceOffset() + edge.source.x,
					edge.source.y + edge.source.height);
			if (edge.source instanceof SubgraphBoundary) {
				SubgraphBoundary boundary = (SubgraphBoundary) edge.source;
				if (boundary.getParent().head == boundary)
					edge.start.y = boundary.getParent().y
							+ boundary.getParent().insets.top;
			}
			edge.end = new Point(edge.getTargetOffset() + edge.target.x,
					edge.target.y);

			if (edge.vNodes != null)
				routeLongEdge(edge, g);
			else {
				PointList list = new PointList();
				list.addPoint(edge.start);
				list.addPoint(edge.end);
				edge.setPoints(list);
			}
		}
	}

	static void routeLongEdge(Edge edge, DirectedGraph g) {
		ShortestPathRouter router = new ShortestPathRouter();
		Path path = new Path(edge.start, edge.end);
		router.addPath(path);
		Rectangle o;
		Insets padding;
		for (int i = 0; i < edge.vNodes.size(); i++) {
			VirtualNode node = (VirtualNode) edge.vNodes.get(i);
			Node neighbor;
			if (node.left != null) {
				neighbor = node.left;
				o = new Rectangle(neighbor.x, neighbor.y, neighbor.width,
						neighbor.height);
				padding = g.getPadding(neighbor);
				o.width += padding.right + padding.left;
				o.width += (edge.getPadding() * 2);
				o.x -= (padding.left + edge.getPadding());
				o.union(o.getLocation().translate(-100000, 2));
				router.addObstacle(o);
			}
			if (node.right != null) {
				neighbor = node.right;
				o = new Rectangle(neighbor.x, neighbor.y, neighbor.width,
						neighbor.height);
				padding = g.getPadding(neighbor);
				o.width += padding.right + padding.left;
				o.width += (edge.getPadding() * 2);
				o.x -= (padding.left + edge.getPadding());
				o.union(o.getLocation().translate(100000, 2));
				router.addObstacle(o);
			}
		}
		router.setSpacing(0);
		router.solve();
		edge.setPoints(path.getPoints());
	}

}