net.sourceforge.plantuml.graph.Graph5 Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml Show documentation
Show all versions of plantuml Show documentation
PlantUML is a component that allows to quickly write :
* sequence diagram,
* use case diagram,
* class diagram,
* activity diagram,
* component diagram,
* state diagram
* object diagram
/* ========================================================================
* PlantUML : a free UML diagram generator
* ========================================================================
*
* (C) Copyright 2009-2013, 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: 9786 $
*
*/
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;
}
}