mirror of https://github.com/octoleo/plantuml.git
319 lines
9.7 KiB
Java
319 lines
9.7 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009-2020, 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.jsondiagram;
|
|
|
|
import java.awt.geom.Dimension2D;
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
|
|
import net.sourceforge.plantuml.Dimension2DDouble;
|
|
import net.sourceforge.plantuml.ISkinParam;
|
|
import net.sourceforge.plantuml.LineBreakStrategy;
|
|
import net.sourceforge.plantuml.creole.CreoleMode;
|
|
import net.sourceforge.plantuml.cucadiagram.Display;
|
|
import net.sourceforge.plantuml.graphic.AbstractTextBlock;
|
|
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.graphic.TextBlockUtils;
|
|
import net.sourceforge.plantuml.json.JsonArray;
|
|
import net.sourceforge.plantuml.json.JsonObject;
|
|
import net.sourceforge.plantuml.json.JsonObject.Member;
|
|
import net.sourceforge.plantuml.json.JsonValue;
|
|
import net.sourceforge.plantuml.style.PName;
|
|
import net.sourceforge.plantuml.style.SName;
|
|
import net.sourceforge.plantuml.style.Style;
|
|
import net.sourceforge.plantuml.svek.TextBlockBackcolored;
|
|
import net.sourceforge.plantuml.ugraphic.UGraphic;
|
|
import net.sourceforge.plantuml.ugraphic.ULine;
|
|
import net.sourceforge.plantuml.ugraphic.URectangle;
|
|
import net.sourceforge.plantuml.ugraphic.UTranslate;
|
|
import net.sourceforge.plantuml.ugraphic.color.HColor;
|
|
|
|
//See TextBlockMap
|
|
public class TextBlockJson extends AbstractTextBlock implements TextBlockBackcolored {
|
|
|
|
private final List<Line> lines = new ArrayList<Line>();
|
|
|
|
private final Style style;
|
|
private final ISkinParam skinParam;
|
|
private double totalWidth;
|
|
private final JsonValue root;
|
|
|
|
static class Line {
|
|
final TextBlock b1;
|
|
final TextBlock b2;
|
|
final boolean highlighted;
|
|
|
|
Line(TextBlock b1, TextBlock b2, boolean highlighted) {
|
|
this.b1 = b1;
|
|
this.b2 = b2;
|
|
this.highlighted = highlighted;
|
|
}
|
|
|
|
Line(TextBlock b1, boolean highlighted) {
|
|
this(b1, null, highlighted);
|
|
}
|
|
|
|
double getHeightOfRow(StringBounder stringBounder) {
|
|
final double height = b1.calculateDimension(stringBounder).getHeight();
|
|
if (b2 == null) {
|
|
return height;
|
|
}
|
|
return Math.max(height, b2.calculateDimension(stringBounder).getHeight());
|
|
}
|
|
|
|
}
|
|
|
|
public TextBlockJson(ISkinParam skinParam, JsonValue root, List<String> highlighted, Style style) {
|
|
this.skinParam = skinParam;
|
|
this.style = style;
|
|
this.root = root;
|
|
if (root instanceof JsonObject)
|
|
for (Member member : (JsonObject) root) {
|
|
final String key = member.getName();
|
|
final String value = getShortString(member.getValue());
|
|
|
|
final TextBlock block1 = getTextBlock(key);
|
|
final TextBlock block2 = getTextBlock(value);
|
|
this.lines.add(new Line(block1, block2, isHighlighted(key, highlighted)));
|
|
}
|
|
if (root instanceof JsonArray) {
|
|
int i = 0;
|
|
for (JsonValue value : (JsonArray) root) {
|
|
final TextBlock block2 = getTextBlock(getShortString(value));
|
|
this.lines.add(new Line(block2, isHighlighted("" + i, highlighted)));
|
|
i++;
|
|
}
|
|
}
|
|
}
|
|
|
|
private boolean isHighlighted(String key, List<String> highlighted) {
|
|
for (String tmp : highlighted) {
|
|
if (tmp.trim().equals("\"" + key + "\"")) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public int size() {
|
|
int size = 0;
|
|
if (root instanceof JsonObject) {
|
|
for (Member member : (JsonObject) root)
|
|
size++;
|
|
}
|
|
if (root instanceof JsonArray) {
|
|
for (JsonValue value : (JsonArray) root)
|
|
size++;
|
|
}
|
|
return size;
|
|
|
|
}
|
|
|
|
private String getShortString(JsonValue value) {
|
|
if (value.isString()) {
|
|
return value.asString();
|
|
}
|
|
if (value.isNull()) {
|
|
return "<U+2400>";
|
|
// return "<U+2205> null";
|
|
}
|
|
if (value.isNumber()) {
|
|
return value.toString();
|
|
}
|
|
if (value.isBoolean()) {
|
|
if (value.isTrue()) {
|
|
return "<U+2611> true";
|
|
} else {
|
|
return "<U+2610> false";
|
|
}
|
|
}
|
|
return " ";
|
|
}
|
|
|
|
public List<JsonValue> children() {
|
|
final List<JsonValue> result = new ArrayList<JsonValue>();
|
|
if (root instanceof JsonObject) {
|
|
for (Member member : (JsonObject) root) {
|
|
final JsonValue value = member.getValue();
|
|
if (value instanceof JsonObject || value instanceof JsonArray) {
|
|
result.add(value);
|
|
} else {
|
|
result.add(null);
|
|
}
|
|
}
|
|
}
|
|
if (root instanceof JsonArray) {
|
|
for (JsonValue value : (JsonArray) root) {
|
|
if (value instanceof JsonObject || value instanceof JsonArray) {
|
|
result.add(value);
|
|
} else {
|
|
result.add(null);
|
|
}
|
|
}
|
|
}
|
|
return Collections.unmodifiableList(result);
|
|
}
|
|
|
|
public List<String> keys() {
|
|
final List<String> result = new ArrayList<String>();
|
|
if (root instanceof JsonObject) {
|
|
for (Member member : (JsonObject) root) {
|
|
final String key = member.getName();
|
|
result.add(key);
|
|
}
|
|
}
|
|
if (root instanceof JsonArray) {
|
|
int i = 0;
|
|
for (JsonValue value : (JsonArray) root) {
|
|
result.add("" + i);
|
|
i++;
|
|
}
|
|
}
|
|
return Collections.unmodifiableList(result);
|
|
}
|
|
|
|
public Dimension2D calculateDimension(StringBounder stringBounder) {
|
|
return new Dimension2DDouble(getWidthColA(stringBounder) + getWidthColB(stringBounder),
|
|
getTotalHeight(stringBounder));
|
|
}
|
|
|
|
public double getWidthColA(StringBounder stringBounder) {
|
|
double width = 0;
|
|
for (Line line : lines) {
|
|
width = Math.max(width, line.b1.calculateDimension(stringBounder).getWidth());
|
|
}
|
|
return width;
|
|
}
|
|
|
|
public double getWidthColB(StringBounder stringBounder) {
|
|
double width = 0;
|
|
for (Line line : lines) {
|
|
if (line.b2 != null) {
|
|
width = Math.max(width, line.b2.calculateDimension(stringBounder).getWidth());
|
|
}
|
|
}
|
|
return width;
|
|
}
|
|
|
|
public void drawU(final UGraphic ug) {
|
|
final StringBounder stringBounder = ug.getStringBounder();
|
|
|
|
final Dimension2D fullDim = calculateDimension(stringBounder);
|
|
double trueWidth = Math.max(fullDim.getWidth(), totalWidth);
|
|
final double widthColA = getWidthColA(stringBounder);
|
|
final double widthColB = getWidthColB(stringBounder);
|
|
|
|
double y = 0;
|
|
final UGraphic ugNode = style.applyStrokeAndLineColor(ug, skinParam.getIHtmlColorSet());
|
|
for (Line line : lines) {
|
|
final double heightOfRow = line.getHeightOfRow(stringBounder);
|
|
y += heightOfRow;
|
|
}
|
|
if (y == 0)
|
|
y = 15;
|
|
if (trueWidth == 0)
|
|
trueWidth = 30;
|
|
|
|
final double round = style.value(PName.RoundCorner).asDouble();
|
|
final URectangle fullNodeRectangle = new URectangle(trueWidth, y).rounded(round);
|
|
final HColor backColor = style.value(PName.BackGroundColor).asColor(skinParam.getIHtmlColorSet());
|
|
ugNode.apply(backColor.bg()).apply(backColor).draw(fullNodeRectangle);
|
|
|
|
final Style styleSeparator = style.getSignature().add(SName.separator)
|
|
.getMergedStyle(skinParam.getCurrentStyleBuilder());
|
|
final UGraphic ugSeparator = styleSeparator.applyStrokeAndLineColor(ug, skinParam.getIHtmlColorSet());
|
|
|
|
y = 0;
|
|
for (Line line : lines) {
|
|
final UGraphic ugline = ugSeparator.apply(UTranslate.dy(y));
|
|
final double heightOfRow = line.getHeightOfRow(stringBounder);
|
|
if (line.highlighted) {
|
|
final URectangle back = new URectangle(trueWidth - 2, heightOfRow).rounded(4);
|
|
final HColor yellow = skinParam.getIHtmlColorSet().getColorOrWhite("#ccff02");
|
|
ugline.apply(yellow).apply(yellow.bg()).apply(new UTranslate(1.5, 0)).draw(back);
|
|
}
|
|
|
|
if (y > 0)
|
|
ugline.draw(ULine.hline(trueWidth));
|
|
|
|
final HorizontalAlignment horizontalAlignment = style.getHorizontalAlignment();
|
|
horizontalAlignment.draw(ugline, line.b1, 0, widthColA);
|
|
|
|
if (line.b2 != null) {
|
|
final UGraphic uglineColB = ugline.apply(UTranslate.dx(widthColA));
|
|
horizontalAlignment.draw(uglineColB, line.b2, 0, widthColB);
|
|
uglineColB.draw(ULine.vline(heightOfRow));
|
|
}
|
|
|
|
y += heightOfRow;
|
|
}
|
|
ugNode.draw(fullNodeRectangle);
|
|
|
|
}
|
|
|
|
private double getTotalHeight(StringBounder stringBounder) {
|
|
double height = 0;
|
|
for (Line line : lines) {
|
|
height += line.getHeightOfRow(stringBounder);
|
|
}
|
|
return height;
|
|
}
|
|
|
|
private TextBlock getTextBlock(String key) {
|
|
final Display display = Display.getWithNewlines(key);
|
|
final FontConfiguration fontConfiguration = style.getFontConfiguration(skinParam.getIHtmlColorSet());
|
|
final LineBreakStrategy wrap = style.wrapWidth();
|
|
final HorizontalAlignment horizontalAlignment = style.getHorizontalAlignment();
|
|
TextBlock result = display.create0(fontConfiguration, horizontalAlignment, skinParam, wrap,
|
|
CreoleMode.NO_CREOLE, null, null);
|
|
result = TextBlockUtils.withMargin(result, 5, 2);
|
|
return result;
|
|
}
|
|
|
|
public void setTotalWidth(double totalWidth) {
|
|
this.totalWidth = totalWidth;
|
|
}
|
|
|
|
public HColor getBackcolor() {
|
|
return null;
|
|
}
|
|
|
|
}
|