1
0
mirror of https://github.com/octoleo/plantuml.git synced 2024-05-28 22:20:49 +00:00
plantuml/src/net/sourceforge/plantuml/jsondiagram/TextBlockJson.java
2021-11-25 22:42:50 +01:00

343 lines
11 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<>();
private final Style styleNode;
private final Style styleNodeHightlight;
private final Style styleNodeHeader;
private final Style styleNodeHeaderHighlight;
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());
}
}
private HColor getBackColor() {
return styleNodeHightlight.value(PName.BackGroundColor).asColor(skinParam.getThemeStyle(),
skinParam.getIHtmlColorSet());
}
TextBlockJson(ISkinParam skinParam, JsonValue root, List<String> allHighlighteds, Style styleNode,
Style styleNodeHightlight, Style styleNodeHeader, Style styleNodeHeaderHighlight) {
this.skinParam = skinParam;
this.styleNode = styleNode;
this.styleNodeHeaderHighlight = styleNodeHeaderHighlight;
this.styleNodeHightlight = styleNodeHightlight;
this.styleNodeHeader = styleNodeHeader;
this.root = root;
if (root instanceof JsonObject)
for (Member member : (JsonObject) root) {
final String key = member.getName();
final String value = getShortString(member.getValue());
final boolean highlighted = isHighlighted(key, allHighlighteds);
final TextBlock block1 = getTextBlock(getStyleToUse(true, highlighted), key);
final TextBlock block2 = getTextBlock(getStyleToUse(false, highlighted), value);
this.lines.add(new Line(block1, block2, highlighted));
}
if (root instanceof JsonArray) {
int i = 0;
for (JsonValue value : (JsonArray) root) {
final boolean highlighted = isHighlighted("" + i, allHighlighteds);
final TextBlock block2 = getTextBlock(getStyleToUse(false, highlighted), getShortString(value));
this.lines.add(new Line(block2, highlighted));
i++;
}
}
}
private Style getStyleToUse(boolean header, boolean highlighted) {
if (header && highlighted)
return styleNodeHeaderHighlight;
if (highlighted)
return styleNodeHightlight;
if (header)
return styleNodeHeader;
return styleNode;
}
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<>();
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<>();
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 = styleNode.applyStrokeAndLineColor(ug, skinParam.getIHtmlColorSet(),
skinParam.getThemeStyle());
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 = styleNode.value(PName.RoundCorner).asDouble();
final URectangle fullNodeRectangle = new URectangle(trueWidth, y).rounded(round);
final HColor backColor = styleNode.value(PName.BackGroundColor).asColor(skinParam.getThemeStyle(),
skinParam.getIHtmlColorSet());
ugNode.apply(backColor.bg()).apply(backColor).draw(fullNodeRectangle);
final Style styleSeparator = styleNode.getSignature().add(SName.separator)
.getMergedStyle(skinParam.getCurrentStyleBuilder());
final UGraphic ugSeparator = styleSeparator.applyStrokeAndLineColor(ug, skinParam.getIHtmlColorSet(),
skinParam.getThemeStyle());
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);
ugline.apply(getBackColor()).apply(getBackColor().bg()).apply(new UTranslate(1.5, 0)).draw(back);
}
if (y > 0)
ugline.draw(ULine.hline(trueWidth));
final HorizontalAlignment horizontalAlignment = styleNode.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(Style style, String key) {
final Display display = Display.getWithNewlines(key);
final FontConfiguration fontConfiguration = style.getFontConfiguration(skinParam.getThemeStyle(),
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;
}
}