net.sourceforge.plantuml.wire.WBlock 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-2023, Arnaud Roques
*
* Project Info: http://plantuml.com
*
* If you like this project or if you find it useful, you can support us at:
*
* http://plantuml.com/patreon (only 1$ per month!)
* http://plantuml.com/paypal
*
* 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.
*
*
* Original Author: Arnaud Roques
*
*
*/
package net.sourceforge.plantuml.wire;
import net.sourceforge.plantuml.awt.geom.Dimension2D;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;
import net.sourceforge.plantuml.Dimension2DDouble;
import net.sourceforge.plantuml.ISkinParam;
import net.sourceforge.plantuml.SpriteContainerEmpty;
import net.sourceforge.plantuml.command.CommandExecutionResult;
import net.sourceforge.plantuml.cucadiagram.Display;
import net.sourceforge.plantuml.graphic.FontConfiguration;
import net.sourceforge.plantuml.graphic.HorizontalAlignment;
import net.sourceforge.plantuml.graphic.StringBounder;
import net.sourceforge.plantuml.graphic.TextBlock;
import net.sourceforge.plantuml.ugraphic.UFont;
import net.sourceforge.plantuml.ugraphic.UGraphic;
import net.sourceforge.plantuml.ugraphic.URectangle;
import net.sourceforge.plantuml.ugraphic.UTranslate;
import net.sourceforge.plantuml.ugraphic.color.HColor;
import net.sourceforge.plantuml.ugraphic.color.HColorUtils;
public class WBlock {
private final static int STARTING_Y = 10;
private final String name;
private final double forcedWidth;
private final double forcedHeight;
private final HColor color;
private final List children = new ArrayList<>();
private final UTranslate position;
private WBlock parent;
private UTranslate cursor = new UTranslate(10, STARTING_Y);
private WBlock addedToCursor = null;
private UTranslate futureOutHorizontal;
private UTranslate futureOutVertical;
private final List prints = new ArrayList<>();
public UTranslate getAbsolutePosition(String supx, String supy) {
if (parent == null)
return position;
final UTranslate p = parent.getAbsolutePosition("0", "0");
final double x = position.getDx() + p.getDx() + parseWidth(supx);
final double y = position.getDy() + p.getDy() + parseHeight(supy);
return new UTranslate(x, y);
}
private double parseWidth(String value) {
if (value.endsWith("%")) {
final double p = Double.parseDouble(value.substring(0, value.length() - 1)) / 100.0;
return getMaxDimension().getWidth() * p;
}
if (value.contains("%")) {
final StringTokenizer st = new StringTokenizer(value, "%");
final String v1 = st.nextToken();
final String v2 = st.nextToken();
final double p = Double.parseDouble(v1) / 100.0;
return getMaxDimension().getWidth() * p + Double.parseDouble(v2);
}
return Double.parseDouble(value);
}
private double parseHeight(String value) {
if (value.endsWith("%")) {
final double p = Double.parseDouble(value.substring(0, value.length() - 1)) / 100.0;
return getMaxDimension().getHeight() * p;
}
if (value.contains("%")) {
final StringTokenizer st = new StringTokenizer(value, "%");
final String v1 = st.nextToken();
final String v2 = st.nextToken();
final double p = Double.parseDouble(v1) / 100.0;
return getMaxDimension().getHeight() * p + Double.parseDouble(v2);
}
return Double.parseDouble(value);
}
@Override
public String toString() {
return name + " " + position;
}
public WBlock(String name, UTranslate position, double width, double height, HColor color) {
this.name = name;
this.forcedWidth = width;
this.forcedHeight = height;
this.color = color;
this.position = position;
}
private WBlock getChildByName(String name) {
for (WBlock child : children)
if (name.equals(child.getName()))
return child;
return null;
}
public WBlock getBlock(String name) {
final int x = name.indexOf('.');
if (x == -1)
return getChildByName(name);
final WBlock first = getChildByName(name.substring(0, x));
if (first == null)
return null;
return first.getBlock(name.substring(x + 1));
}
private String getName() {
return name;
}
public CommandExecutionResult newColumn(int level) {
if (level == 0) {
final Dimension2D max = getNaturalDimension();
this.cursor = new UTranslate(max.getWidth() + 10, STARTING_Y);
this.addedToCursor = null;
return CommandExecutionResult.ok();
}
return getLastChild().newColumn(level - 1);
}
public CommandExecutionResult wgoto(int level, double x, double y) {
if (level == 0) {
this.cursor = new UTranslate(x, y);
this.addedToCursor = null;
return CommandExecutionResult.ok();
}
return getLastChild().wgoto(level - 1, x, y);
}
public CommandExecutionResult wmove(int level, double x, double y) {
if (level == 0) {
this.cursor = this.cursor.compose(new UTranslate(x, y));
return CommandExecutionResult.ok();
}
return getLastChild().wmove(level - 1, x, y);
}
public CommandExecutionResult print(StringBounder stringBounder, ISkinParam skinParam, int level, String text) {
if (level == 0) {
final WPrint print = new WPrint(skinParam, getNextPosition(), null, Display.getWithNewlines(text));
this.prints.add(print);
this.cursor = this.cursor.compose(UTranslate.dy(print.getHeight(stringBounder)));
return CommandExecutionResult.ok();
}
return getLastChild().print(stringBounder, skinParam, level - 1, text);
}
public CommandExecutionResult addBlock(int level, String name, double width, double height, HColor color) {
if (name.contains(".")) {
throw new IllegalArgumentException();
}
if (getChildByName(name) != null) {
return CommandExecutionResult.error("Component exists already");
}
if (level == 0) {
this.cursor = this.cursor.compose(UTranslate.dy(10));
final WBlock newBlock = new WBlock(name, getNextPosition(), width, height, color);
this.cursor = this.cursor.compose(UTranslate.dy(10));
this.addedToCursor = newBlock;
children.add(newBlock);
newBlock.parent = this;
return CommandExecutionResult.ok();
}
final WBlock last = getLastChild();
return last.addBlock(level - 1, name, width, height, color);
}
private UTranslate getNextPosition() {
if (this.addedToCursor != null) {
final Dimension2D dim = this.addedToCursor.getMaxDimension();
this.cursor = this.cursor.compose(UTranslate.dy(dim.getHeight()));
}
this.addedToCursor = null;
return this.cursor;
}
private WBlock getLastChild() {
if (children.size() == 0)
return null;
return children.get(children.size() - 1);
}
public void drawMe(UGraphic ug) {
drawBox(ug);
final UFont font = UFont.sansSerif(12);
final FontConfiguration fc = FontConfiguration.create(font, getBlack(), getBlack(), false);
final Display display = Display.create(name.replace('_', ' '));
final TextBlock text = display.create(fc, HorizontalAlignment.LEFT, new SpriteContainerEmpty());
text.drawU(ug.apply(UTranslate.dx(5)));
}
private HColor getBlack() {
return HColorUtils.BLACK.withDark(HColorUtils.WHITE);
}
private void drawBox(UGraphic ug) {
ug = ug.apply(getBlack());
if (name.length() > 0) {
final URectangle rect = new URectangle(getMaxDimension());
UGraphic ugRect = ug;
if (color != null) {
ugRect = ugRect.apply(color.bg());
}
ugRect.draw(rect);
}
for (WBlock child : children) {
child.drawMe(ug.apply(child.position));
}
for (WPrint print : prints) {
print.drawMe(ug.apply(print.getPosition()));
}
}
private Dimension2D getMaxDimension() {
if (children.size() > 0) {
if (forcedWidth != 0) {
return new Dimension2DDouble(forcedWidth, forcedHeight);
}
return getNaturalDimension();
}
final double x = forcedWidth == 0 ? 100 : forcedWidth;
final double y = forcedHeight == 0 ? 100 : forcedHeight;
return new Dimension2DDouble(x, y);
}
private Dimension2D getNaturalDimension() {
double x = 0;
double y = 0;
for (WBlock child : children) {
final Dimension2D dim = child.getMaxDimension();
x = Math.max(x, child.position.getDx() + dim.getWidth() + 10);
y = Math.max(y, child.position.getDy() + dim.getHeight() + 10);
}
return new Dimension2DDouble(x, y);
}
public UTranslate getNextOutHorizontal(String x, String y, WLinkType type) {
final UTranslate result;
if (x != null && y != null) {
result = getAbsolutePosition(x, y);
} else if (futureOutHorizontal == null) {
result = getAbsolutePosition("100%", "5");
} else {
result = futureOutHorizontal;
}
futureOutHorizontal = result.compose(UTranslate.dy(type.spaceForNext()));
return result;
}
public UTranslate getNextOutVertical(String x, String y, WLinkType type) {
final UTranslate result;
if (x != null && y != null) {
result = getAbsolutePosition(x, y);
} else if (futureOutVertical == null) {
result = getAbsolutePosition("5", "100%");
} else {
result = futureOutVertical;
}
futureOutVertical = result.compose(UTranslate.dx(type.spaceForNext()));
return result;
}
}