net.sourceforge.plantuml.graph.Graph5 Maven / Gradle / Ivy

Go to download
/* ========================================================================
 * PlantUML : a free UML diagram generator
 * ========================================================================
 *
 * (C) Copyright 2009-2012, Arnaud Roques
 *
 * Project Info:  http://plantuml.sourceforge.net
 * 
 * This file is part of PlantUML.
 *
 * PlantUML is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * PlantUML distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
 * License for more details.
 *
 * You should have received a copy of the GNU General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301,
 * USA.
 *
 * [Java is a trademark or registered trademark of Sun Microsystems, Inc.
 * in the United States and other countries.]
 *
 * Original Author:  Arnaud Roques
 * 
 * Revision $Revision: 8475 $
 *
 */
package net.sourceforge.plantuml.graph;

import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.geom.Dimension2D;
import java.awt.geom.Line2D;
import java.awt.geom.Point2D;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import net.sourceforge.plantuml.Dimension2DDouble;
import net.sourceforge.plantuml.EmptyImageBuilder;
import net.sourceforge.plantuml.cucadiagram.IEntity;
import net.sourceforge.plantuml.graph2.IInflationTransform;
import net.sourceforge.plantuml.graph2.InflationTransform2;
import net.sourceforge.plantuml.graph2.Plan;
import net.sourceforge.plantuml.graph2.Polyline2;
import net.sourceforge.plantuml.graphic.StringBounderUtils;
import net.sourceforge.plantuml.ugraphic.ColorMapperIdentity;

public class Graph5 {

	final private static Graphics2D dummyGraphics2D;

	private final int spaceWidth = 40;
	private final int spaceHeight = 40;

	private final double margin = 20;
	private final Board board;

	private int maxRow;
	private int maxCol;

	private final Plan plan = new Plan();
	private final IInflationTransform inflationTransform = new InflationTransform2();
	// private final IInflationTransform inflationTransform = new
	// IdentityInflationTransform();

	static {
		final EmptyImageBuilder builder = new EmptyImageBuilder(10, 10, Color.WHITE);
		dummyGraphics2D = builder.getGraphics2D();
	}

	private AbstractEntityImage getImage(ANode n) {
		return new EntityImageFactory().createEntityImage((IEntity) n.getUserData());
	}

	public Graph5(Board board) {
		board.normalize();
		this.board = board;
		for (ANode n : board.getNodes()) {
			maxRow = Math.max(maxRow, n.getRow());
			maxCol = Math.max(maxCol, board.getCol(n));
			final Point2D.Double pos = getPosition(n);
			plan.addPoint2D(pos);
			// Log.println("n=" + n + " pos=" + pos);
		}
		for (ANode n : board.getNodes()) {
			final AbstractEntityImage image = getImage(n);
			final Point2D.Double pos = getPosition(n);
			final int widthCell = (int) image.getDimension(StringBounderUtils.asStringBounder(dummyGraphics2D))
					.getWidth() + 20;
			final int heightCell = (int) image.getDimension(StringBounderUtils.asStringBounder(dummyGraphics2D))
					.getHeight() + 20;
			inflationTransform.addInflationX(pos.getX(), widthCell);
			inflationTransform.addInflationY(pos.getY(), heightCell);
		}

		// Log.println("inflationTransform=" + inflationTransform);

	}

	public Point2D.Double getPosition(ANode node) {
		return new Point2D.Double(board.getCol(node) * spaceWidth, node.getRow() * spaceHeight);
	}

	public Dimension2D getDimension() {
		final double width = spaceWidth * maxCol;
		final int height = spaceWidth * maxRow;
		return new Dimension2DDouble(width + 2 * margin + inflationTransform.getTotalInflationX(), height + 2 * margin
				+ inflationTransform.getTotalInflationY());

	}

	public void draw(final Graphics2D g2d) {
		g2d.translate(margin, margin);
		g2d.setColor(Color.BLUE);

		for (ALink link : getSortedLinks()) {
			final Point2D start = getPosition(link.getNode1());
			final Point2D end = getPosition(link.getNode2());

			final List lines = buildPath(start, end);
			final Polyline2 polyline = buildPolyline(start, end, lines);
			polyline.draw(g2d);
		}

		for (ANode n : board.getNodes()) {
			final AbstractEntityImage image = getImage(n);
			Point2D pos = getPosition(n);
			pos = inflationTransform.inflatePoint2D(pos);
			final double x = pos.getX() - image.getDimension(StringBounderUtils.asStringBounder(g2d)).getWidth() / 2;
			final double y = pos.getY() - image.getDimension(StringBounderUtils.asStringBounder(g2d)).getHeight() / 2;
			g2d.translate(x, y);
			image.draw(new ColorMapperIdentity(), g2d);
			g2d.translate(-x, -y);
		}

	}

	private Polyline2 buildPolyline(final Point2D start, final Point2D end, final List lines) {
		final Polyline2 polyline = new Polyline2(inflationTransform.inflatePoint2D(start), inflationTransform
				.inflatePoint2D(end));
		final List list = inflationTransform.inflate(lines);
		for (Line2D.Double l1 : list) {
			polyline.addLine(l1);
		}
		return polyline;
	}

	private List buildPath(final Point2D start, final Point2D end) {
		Point2D current = start;
		final List interm = plan.getIntermediatePoints(start, end);
		final List lines = new ArrayList();
		for (final Point2D.Double inter : interm) {
			plan.addPoint2D(inter);
			lines.add(new Line2D.Double(current, inter));
			plan.createLink(current, inter);
			current = inter;
		}
		lines.add(new Line2D.Double(current, end));
		plan.createLink(current, end);
		return lines;
	}

	private List getSortedLinks() {
		final Map lengths = new LinkedHashMap();
		for (ALink link : board.getLinks()) {
			final Point2D.Double p1 = getPosition(link.getNode1());
			final Point2D.Double p2 = getPosition(link.getNode2());
			lengths.put(link, p1.distance(p2));
		}
		final List all = new ArrayList(lengths.keySet());
		Collections.sort(all, new Comparator() {
			public int compare(ALink l1, ALink l2) {
				final double diff = lengths.get(l1) - lengths.get(l2);
				return (int) Math.signum(diff);
			}
		});
		return all;
	}

}