net.sourceforge.plantuml.flowdiagram.FlowDiagram 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: 8475 $
*
*/
package net.sourceforge.plantuml.flowdiagram;
import java.awt.geom.Dimension2D;
import java.awt.geom.Point2D;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import net.sourceforge.plantuml.Dimension2DDouble;
import net.sourceforge.plantuml.FileFormatOption;
import net.sourceforge.plantuml.UmlDiagram;
import net.sourceforge.plantuml.UmlDiagramType;
import net.sourceforge.plantuml.Url;
import net.sourceforge.plantuml.api.ImageDataSimple;
import net.sourceforge.plantuml.core.ImageData;
import net.sourceforge.plantuml.golem.MinMaxGolem;
import net.sourceforge.plantuml.golem.Path;
import net.sourceforge.plantuml.golem.Position;
import net.sourceforge.plantuml.golem.Tile;
import net.sourceforge.plantuml.golem.TileArea;
import net.sourceforge.plantuml.golem.TileGeometry;
import net.sourceforge.plantuml.golem.TilesField;
import net.sourceforge.plantuml.graphic.HtmlColorUtils;
import net.sourceforge.plantuml.graphic.StringBounder;
import net.sourceforge.plantuml.graphic.TextBlock;
import net.sourceforge.plantuml.ugraphic.ColorMapperIdentity;
import net.sourceforge.plantuml.ugraphic.UChangeBackColor;
import net.sourceforge.plantuml.ugraphic.UChangeColor;
import net.sourceforge.plantuml.ugraphic.UEllipse;
import net.sourceforge.plantuml.ugraphic.UGraphic;
import net.sourceforge.plantuml.ugraphic.UGraphicUtils;
import net.sourceforge.plantuml.ugraphic.ULine;
import net.sourceforge.plantuml.ugraphic.UShape;
import net.sourceforge.plantuml.ugraphic.UTranslate;
public class FlowDiagram extends UmlDiagram implements TextBlock {
private static double SINGLE_SIZE_X = 100;
private static double SINGLE_SIZE_Y = 35;
private TilesField field;
private final Map tilesBoxes = new HashMap();
private Tile lastTile;
public String getDescription() {
return "Flow Diagram";
}
@Override
public UmlDiagramType getUmlDiagramType() {
return UmlDiagramType.FLOW;
}
public void lineSimple(TileGeometry orientation, String idDest, String label) {
final Tile newTile;
if (field == null) {
field = new TilesField();
tilesBoxes.clear();
newTile = field.getRoot();
} else {
newTile = field.createTile(lastTile, orientation);
}
final ActivityBox box = new ActivityBox(newTile, idDest, label);
tilesBoxes.put(newTile, box);
lastTile = newTile;
return;
}
public void linkSimple(TileGeometry orientation, String idDest) {
final Tile tile = getTileById(idDest);
field.addPath(lastTile, tile, orientation);
}
private Tile getTileById(String id) {
for (Map.Entry ent : tilesBoxes.entrySet()) {
if (ent.getValue().getId().equals(id)) {
return ent.getKey();
}
}
throw new IllegalArgumentException(id);
}
@Override
protected ImageData exportDiagramInternal(OutputStream os, int index, FileFormatOption fileFormatOption,
List flashcodes) throws IOException {
UGraphicUtils.writeImage(os, null, fileFormatOption, new ColorMapperIdentity(), HtmlColorUtils.WHITE, this);
return new ImageDataSimple();
}
public void drawU(UGraphic ug) {
double x = 0;
double y = 0;
final MinMaxGolem minMax = getMinMax();
x -= minMax.getMinX() * SINGLE_SIZE_X;
y -= minMax.getMinY() * SINGLE_SIZE_Y;
final StringBounder stringBounder = ug.getStringBounder();
for (Map.Entry ent : tilesBoxes.entrySet()) {
final Tile tile = ent.getKey();
final Position pos = field.getPosition(tile);
final int xmin = pos.getXmin();
final int ymin = pos.getYmin();
final ActivityBox box = ent.getValue();
final Dimension2D dimBox = box.calculateDimension(stringBounder);
final double deltaX = SINGLE_SIZE_X * 2 - dimBox.getWidth();
final double deltaY = SINGLE_SIZE_Y * 2 - dimBox.getHeight();
box.drawU(ug.apply(new UTranslate((x + xmin * SINGLE_SIZE_X + deltaX / 2), (y + ymin
* SINGLE_SIZE_Y + deltaY / 2))));
}
ug = ug.apply(new UChangeColor(HtmlColorUtils.getColorIfValid("#A80036")));
ug = ug.apply(new UChangeBackColor(HtmlColorUtils.getColorIfValid("#A80036")));
final UShape arrow = new UEllipse(7, 7);
for (Path p : field.getPaths()) {
final TileArea start = p.getStart();
final TileArea dest = p.getDest();
final Point2D pStart = movePoint(getCenter(start), start.getTile(), start.getGeometry(), stringBounder);
final Point2D pDest = movePoint(getCenter(dest), dest.getTile(), dest.getGeometry(), stringBounder);
final ULine line = new ULine(pDest.getX() - pStart.getX(), pDest.getY() - pStart.getY());
ug.apply(new UTranslate(x + pStart.getX(), y + pStart.getY())).draw(line);
ug.apply(new UTranslate(x + pDest.getX() - 3, y + pDest.getY() - 3)).draw(arrow);
}
}
private Point2D getCenter(TileArea area) {
final Tile tile = area.getTile();
final Position position = field.getPosition(tile);
final double x = position.getCenterX();
final double y = position.getCenterY();
return new Point2D.Double(x * SINGLE_SIZE_X, y * SINGLE_SIZE_Y);
}
private Point2D movePoint(Point2D pt, Tile tile, TileGeometry tileGeometry, StringBounder stringBounder) {
final Dimension2D dim = tilesBoxes.get(tile).calculateDimension(stringBounder);
final double width = dim.getWidth();
final double height = dim.getHeight();
double x = pt.getX();
double y = pt.getY();
switch (tileGeometry) {
case SOUTH:
y += height / 2;
break;
case NORTH:
y -= height / 2;
break;
case EAST:
x += width / 2;
break;
case WEST:
x -= width / 2;
break;
default:
throw new IllegalStateException();
}
return new Point2D.Double(x, y);
}
private MinMaxGolem getMinMax() {
final MinMaxGolem minMax = new MinMaxGolem();
for (Tile tile : tilesBoxes.keySet()) {
minMax.manage(field.getPosition(tile));
}
return minMax;
}
public Dimension2D calculateDimension(StringBounder stringBounder) {
final MinMaxGolem minMax = getMinMax();
return new Dimension2DDouble(minMax.getWidth() * SINGLE_SIZE_X, minMax.getHeight() * SINGLE_SIZE_Y);
}
public List getUrls() {
return Collections.emptyList();
}
}