2021-04-13 17:05:18 +00:00
|
|
|
/* ========================================================================
|
|
|
|
* PlantUML : a free UML diagram generator
|
|
|
|
* ========================================================================
|
|
|
|
*
|
2022-03-07 19:33:46 +00:00
|
|
|
* (C) Copyright 2009-2023, Arnaud Roques
|
2021-04-13 17:05:18 +00:00
|
|
|
*
|
|
|
|
* Project Info: http://plantuml.com
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-13 17:05:18 +00:00
|
|
|
* If you like this project or if you find it useful, you can support us at:
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-13 17:05:18 +00:00
|
|
|
* http://plantuml.com/patreon (only 1$ per month!)
|
|
|
|
* http://plantuml.com/paypal
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-13 17:05:18 +00:00
|
|
|
* 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
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-13 17:05:18 +00:00
|
|
|
*
|
|
|
|
*/
|
|
|
|
package net.sourceforge.plantuml.elk;
|
|
|
|
|
|
|
|
import java.io.IOException;
|
|
|
|
import java.io.OutputStream;
|
|
|
|
import java.util.ArrayList;
|
2021-04-17 11:19:39 +00:00
|
|
|
import java.util.Collection;
|
2021-04-13 17:05:18 +00:00
|
|
|
import java.util.EnumSet;
|
|
|
|
import java.util.LinkedHashMap;
|
|
|
|
import java.util.List;
|
|
|
|
import java.util.Map;
|
|
|
|
import java.util.Map.Entry;
|
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.AlignmentParam;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.FileFormatOption;
|
2021-04-14 18:08:29 +00:00
|
|
|
import net.sourceforge.plantuml.FontParam;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.ISkinParam;
|
|
|
|
import net.sourceforge.plantuml.StringUtils;
|
|
|
|
import net.sourceforge.plantuml.UmlDiagram;
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.UmlDiagramType;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.api.ImageDataSimple;
|
2022-09-12 20:08:34 +00:00
|
|
|
import net.sourceforge.plantuml.awt.geom.XDimension2D;
|
|
|
|
import net.sourceforge.plantuml.awt.geom.XPoint2D;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.core.ImageData;
|
|
|
|
import net.sourceforge.plantuml.cucadiagram.CucaDiagram;
|
2021-04-14 18:08:29 +00:00
|
|
|
import net.sourceforge.plantuml.cucadiagram.Display;
|
2021-04-17 11:19:39 +00:00
|
|
|
import net.sourceforge.plantuml.cucadiagram.GroupType;
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.cucadiagram.IEntity;
|
2021-04-17 11:19:39 +00:00
|
|
|
import net.sourceforge.plantuml.cucadiagram.IGroup;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.cucadiagram.ILeaf;
|
|
|
|
import net.sourceforge.plantuml.cucadiagram.Link;
|
2021-04-17 11:19:39 +00:00
|
|
|
import net.sourceforge.plantuml.cucadiagram.entity.EntityFactory;
|
2021-04-20 20:19:49 +00:00
|
|
|
|
|
|
|
/*
|
|
|
|
* You can choose between real "org.eclipse.elk..." classes or proxied "net.sourceforge.plantuml.elk.proxy..."
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-20 20:19:49 +00:00
|
|
|
* Using proxied classes allows to compile PlantUML without having ELK available on the classpath.
|
|
|
|
* Since GraphViz is the default layout engine up to now, we do not want to enforce the use of ELK just for compilation.
|
|
|
|
* (for people not using maven)
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-20 20:19:49 +00:00
|
|
|
* If you are debugging, you should probably switch to "org.eclipse.elk..." classes
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-20 20:19:49 +00:00
|
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
import org.eclipse.elk.core.RecursiveGraphLayoutEngine;
|
2021-05-01 17:40:56 +00:00
|
|
|
import org.eclipse.elk.core.math.ElkPadding;
|
2021-04-20 20:19:49 +00:00
|
|
|
import org.eclipse.elk.core.options.CoreOptions;
|
|
|
|
import org.eclipse.elk.core.options.Direction;
|
|
|
|
import org.eclipse.elk.core.options.EdgeLabelPlacement;
|
|
|
|
import org.eclipse.elk.core.options.HierarchyHandling;
|
|
|
|
import org.eclipse.elk.core.options.NodeLabelPlacement;
|
|
|
|
import org.eclipse.elk.core.util.NullElkProgressMonitor;
|
|
|
|
import org.eclipse.elk.graph.ElkEdge;
|
|
|
|
import org.eclipse.elk.graph.ElkLabel;
|
|
|
|
import org.eclipse.elk.graph.ElkNode;
|
|
|
|
import org.eclipse.elk.graph.util.ElkGraphUtil;
|
|
|
|
*/
|
|
|
|
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.RecursiveGraphLayoutEngine;
|
2021-05-01 17:40:56 +00:00
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.math.ElkPadding;
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.options.CoreOptions;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.options.Direction;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.options.EdgeLabelPlacement;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.options.HierarchyHandling;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.options.NodeLabelPlacement;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.core.util.NullElkProgressMonitor;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.graph.ElkEdge;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.graph.ElkLabel;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.graph.ElkNode;
|
|
|
|
import net.sourceforge.plantuml.elk.proxy.graph.util.ElkGraphUtil;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.graphic.AbstractTextBlock;
|
2021-04-14 18:08:29 +00:00
|
|
|
import net.sourceforge.plantuml.graphic.FontConfiguration;
|
|
|
|
import net.sourceforge.plantuml.graphic.HorizontalAlignment;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.graphic.QuoteUtils;
|
|
|
|
import net.sourceforge.plantuml.graphic.StringBounder;
|
|
|
|
import net.sourceforge.plantuml.graphic.TextBlock;
|
|
|
|
import net.sourceforge.plantuml.graphic.TextBlockUtils;
|
2022-06-27 16:33:45 +00:00
|
|
|
import net.sourceforge.plantuml.graphic.USymbolFolder;
|
2022-08-17 17:34:24 +00:00
|
|
|
import net.sourceforge.plantuml.log.Logme;
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.style.PName;
|
|
|
|
import net.sourceforge.plantuml.style.SName;
|
|
|
|
import net.sourceforge.plantuml.style.Style;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.svek.Bibliotekon;
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.svek.Cluster;
|
|
|
|
import net.sourceforge.plantuml.svek.ClusterDecoration;
|
2022-09-14 18:12:03 +00:00
|
|
|
import net.sourceforge.plantuml.svek.ClusterPosition;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.svek.CucaDiagramFileMaker;
|
|
|
|
import net.sourceforge.plantuml.svek.DotStringFactory;
|
|
|
|
import net.sourceforge.plantuml.svek.GeneralImageBuilder;
|
|
|
|
import net.sourceforge.plantuml.svek.GraphvizCrash;
|
|
|
|
import net.sourceforge.plantuml.svek.IEntityImage;
|
2021-04-20 20:19:49 +00:00
|
|
|
import net.sourceforge.plantuml.svek.PackageStyle;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.svek.TextBlockBackcolored;
|
|
|
|
import net.sourceforge.plantuml.ugraphic.MinMax;
|
|
|
|
import net.sourceforge.plantuml.ugraphic.UGraphic;
|
2021-04-17 11:19:39 +00:00
|
|
|
import net.sourceforge.plantuml.ugraphic.URectangle;
|
|
|
|
import net.sourceforge.plantuml.ugraphic.UStroke;
|
2021-04-13 17:05:18 +00:00
|
|
|
import net.sourceforge.plantuml.ugraphic.UTranslate;
|
|
|
|
import net.sourceforge.plantuml.ugraphic.color.HColor;
|
2022-08-19 16:34:21 +00:00
|
|
|
import net.sourceforge.plantuml.ugraphic.color.HColors;
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
/*
|
|
|
|
* Some notes:
|
2022-08-17 17:34:24 +00:00
|
|
|
*
|
2021-04-17 11:19:39 +00:00
|
|
|
https://www.eclipse.org/elk/documentation/tooldevelopers/graphdatastructure.html
|
|
|
|
https://www.eclipse.org/elk/documentation/tooldevelopers/graphdatastructure/coordinatesystem.html
|
|
|
|
|
|
|
|
Long hierarchical edge
|
|
|
|
|
|
|
|
https://rtsys.informatik.uni-kiel.de/~biblio/downloads/theses/yab-bt.pdf
|
|
|
|
https://rtsys.informatik.uni-kiel.de/~biblio/downloads/theses/thw-bt.pdf
|
|
|
|
*/
|
2021-04-13 17:05:18 +00:00
|
|
|
public class CucaDiagramFileMakerElk implements CucaDiagramFileMaker {
|
|
|
|
|
|
|
|
private final CucaDiagram diagram;
|
|
|
|
private final StringBounder stringBounder;
|
|
|
|
private final DotStringFactory dotStringFactory;
|
|
|
|
|
|
|
|
private final Map<ILeaf, ElkNode> nodes = new LinkedHashMap<ILeaf, ElkNode>();
|
2021-04-17 11:19:39 +00:00
|
|
|
private final Map<IGroup, ElkNode> clusters = new LinkedHashMap<IGroup, ElkNode>();
|
2021-04-14 18:08:29 +00:00
|
|
|
private final Map<Link, ElkEdge> edges = new LinkedHashMap<Link, ElkEdge>();
|
2021-04-13 17:05:18 +00:00
|
|
|
|
|
|
|
public CucaDiagramFileMakerElk(CucaDiagram diagram, StringBounder stringBounder) {
|
|
|
|
this.diagram = diagram;
|
|
|
|
this.stringBounder = stringBounder;
|
|
|
|
this.dotStringFactory = new DotStringFactory(stringBounder, diagram);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2021-04-14 18:08:29 +00:00
|
|
|
private TextBlock getLabel(Link link) {
|
2021-04-17 11:19:39 +00:00
|
|
|
if (Display.isNull(link.getLabel())) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
final ISkinParam skinParam = diagram.getSkinParam();
|
2022-03-19 12:48:23 +00:00
|
|
|
final FontConfiguration labelFont = FontConfiguration.create(skinParam, FontParam.ARROW, null);
|
2021-04-14 18:08:29 +00:00
|
|
|
final TextBlock label = link.getLabel().create(labelFont,
|
|
|
|
skinParam.getDefaultTextAlignment(HorizontalAlignment.CENTER), skinParam);
|
2022-04-10 19:24:55 +00:00
|
|
|
if (TextBlockUtils.isEmpty(label, stringBounder))
|
2021-04-17 11:19:39 +00:00
|
|
|
return null;
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
return label;
|
2021-04-14 18:08:29 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
private TextBlock getQualifier(Link link, int n) {
|
|
|
|
final String tmp = n == 1 ? link.getQualifier1() : link.getQualifier2();
|
2022-04-10 19:24:55 +00:00
|
|
|
if (tmp == null)
|
2021-04-14 18:08:29 +00:00
|
|
|
return null;
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
final ISkinParam skinParam = diagram.getSkinParam();
|
2022-03-19 12:48:23 +00:00
|
|
|
final FontConfiguration labelFont = FontConfiguration.create(skinParam, FontParam.ARROW, null);
|
2021-04-14 18:08:29 +00:00
|
|
|
final TextBlock label = Display.getWithNewlines(tmp).create(labelFont,
|
|
|
|
skinParam.getDefaultTextAlignment(HorizontalAlignment.CENTER), skinParam);
|
2022-04-10 19:24:55 +00:00
|
|
|
if (TextBlockUtils.isEmpty(label, stringBounder))
|
2021-04-17 11:19:39 +00:00
|
|
|
return null;
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
return label;
|
|
|
|
}
|
|
|
|
|
|
|
|
// Retrieve the real position of a node, depending on its parents
|
2022-09-12 20:08:34 +00:00
|
|
|
private XPoint2D getPosition(ElkNode elkNode) {
|
2021-04-17 11:19:39 +00:00
|
|
|
final ElkNode parent = elkNode.getParent();
|
|
|
|
|
|
|
|
final double x = elkNode.getX();
|
|
|
|
final double y = elkNode.getY();
|
|
|
|
|
|
|
|
// This nasty test checks that parent is "root"
|
|
|
|
if (parent == null || parent.getLabels().size() == 0) {
|
2022-09-12 20:08:34 +00:00
|
|
|
return new XPoint2D(x, y);
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Right now, this is recursive
|
2022-09-12 20:08:34 +00:00
|
|
|
final XPoint2D parentPosition = getPosition(parent);
|
|
|
|
return new XPoint2D(parentPosition.getX() + x, parentPosition.getY() + y);
|
2021-04-17 11:19:39 +00:00
|
|
|
|
2021-04-14 18:08:29 +00:00
|
|
|
}
|
|
|
|
|
2021-04-13 17:05:18 +00:00
|
|
|
// The Drawing class does the real drawing
|
|
|
|
class Drawing extends AbstractTextBlock implements TextBlockBackcolored {
|
|
|
|
|
|
|
|
// min and max of all coord
|
|
|
|
private final MinMax minMax;
|
|
|
|
|
|
|
|
public Drawing(MinMax minMax) {
|
|
|
|
this.minMax = minMax;
|
|
|
|
}
|
|
|
|
|
|
|
|
public void drawU(UGraphic ug) {
|
2021-04-20 20:19:49 +00:00
|
|
|
drawAllClusters(ug);
|
|
|
|
drawAllNodes(ug);
|
|
|
|
drawAllEdges(ug);
|
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
private void drawAllClusters(UGraphic ug) {
|
2022-04-10 19:24:55 +00:00
|
|
|
for (Entry<IGroup, ElkNode> ent : clusters.entrySet())
|
2021-04-20 20:19:49 +00:00
|
|
|
drawSingleCluster(ug, ent.getKey(), ent.getValue());
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
}
|
2021-04-17 11:19:39 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
private void drawAllNodes(UGraphic ug) {
|
2022-04-10 19:24:55 +00:00
|
|
|
for (Entry<ILeaf, ElkNode> ent : nodes.entrySet())
|
2021-04-20 20:19:49 +00:00
|
|
|
drawSingleNode(ug, ent.getKey(), ent.getValue());
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
private void drawAllEdges(UGraphic ug) {
|
2021-04-14 18:08:29 +00:00
|
|
|
for (Entry<Link, ElkEdge> ent : edges.entrySet()) {
|
|
|
|
final Link link = ent.getKey();
|
2022-04-10 19:24:55 +00:00
|
|
|
if (link.isInvis())
|
2021-04-14 18:08:29 +00:00
|
|
|
continue;
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
drawSingleEdge(ug, link, ent.getValue());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
private void drawSingleCluster(UGraphic ug, IGroup group, ElkNode elkNode) {
|
2022-09-12 20:08:34 +00:00
|
|
|
final XPoint2D corner = getPosition(elkNode);
|
2021-04-20 20:19:49 +00:00
|
|
|
final URectangle rect = new URectangle(elkNode.getWidth(), elkNode.getHeight());
|
|
|
|
|
|
|
|
PackageStyle packageStyle = group.getPackageStyle();
|
|
|
|
final ISkinParam skinParam = diagram.getSkinParam();
|
2022-04-10 19:24:55 +00:00
|
|
|
if (packageStyle == null)
|
2021-04-20 20:19:49 +00:00
|
|
|
packageStyle = skinParam.packageStyle();
|
|
|
|
|
|
|
|
final UmlDiagramType umlDiagramType = diagram.getUmlDiagramType();
|
|
|
|
|
2022-04-10 19:24:55 +00:00
|
|
|
final Style style = Cluster.getDefaultStyleDefinition(umlDiagramType.getStyleName(), group.getUSymbol())
|
|
|
|
.getMergedStyle(skinParam.getCurrentStyleBuilder());
|
|
|
|
final double shadowing = style.value(PName.Shadowing).asDouble();
|
2022-05-04 17:52:00 +00:00
|
|
|
final UStroke stroke = Cluster.getStrokeInternal(group, style);
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
HColor backColor = getBackColor(umlDiagramType);
|
2022-08-19 16:34:21 +00:00
|
|
|
backColor = Cluster.getBackColor(backColor, group.getStereotype(), umlDiagramType.getStyleName(),
|
2022-09-18 17:08:06 +00:00
|
|
|
group.getUSymbol(), skinParam.getCurrentStyleBuilder(), skinParam.getIHtmlColorSet());
|
2021-04-20 20:19:49 +00:00
|
|
|
|
2022-05-21 09:41:00 +00:00
|
|
|
final double roundCorner = style.value(PName.RoundCorner).asDouble();
|
|
|
|
// final double roundCorner = group.getUSymbol() == null ? 0
|
|
|
|
// : group.getUSymbol().getSkinParameter().getRoundCorner(skinParam, group.getStereotype());
|
2021-04-20 20:19:49 +00:00
|
|
|
|
|
|
|
final TextBlock ztitle = getTitleBlock(group);
|
|
|
|
final TextBlock zstereo = TextBlockUtils.empty(0, 0);
|
|
|
|
|
2022-09-14 18:12:03 +00:00
|
|
|
final ClusterPosition clusterPosition = new ClusterPosition(0, 0, elkNode.getWidth(), elkNode.getHeight());
|
2021-04-20 20:19:49 +00:00
|
|
|
final ClusterDecoration decoration = new ClusterDecoration(packageStyle, group.getUSymbol(), ztitle,
|
2022-09-14 18:12:03 +00:00
|
|
|
zstereo, clusterPosition, stroke);
|
2021-04-17 11:19:39 +00:00
|
|
|
|
2022-08-19 16:34:21 +00:00
|
|
|
final HColor borderColor = HColors.BLACK;
|
2021-04-20 20:19:49 +00:00
|
|
|
decoration.drawU(ug.apply(new UTranslate(corner)), backColor, borderColor, shadowing, roundCorner,
|
2021-06-27 16:50:40 +00:00
|
|
|
skinParam.getHorizontalAlignment(AlignmentParam.packageTitleAlignment, null, false, null),
|
2022-02-24 18:18:19 +00:00
|
|
|
skinParam.getStereotypeAlignment(), 0);
|
2021-04-17 11:19:39 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
// // Print a simple rectangle right now
|
|
|
|
// ug.apply(HColorUtils.BLACK).apply(new UStroke(1.5)).apply(new UTranslate(corner)).draw(rect);
|
|
|
|
}
|
2021-04-17 11:19:39 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
private TextBlock getTitleBlock(IGroup g) {
|
|
|
|
final Display label = g.getDisplay();
|
2022-04-10 19:24:55 +00:00
|
|
|
if (label == null)
|
2021-04-20 20:19:49 +00:00
|
|
|
return TextBlockUtils.empty(0, 0);
|
2021-04-14 18:08:29 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
final ISkinParam skinParam = diagram.getSkinParam();
|
|
|
|
final FontConfiguration fontConfiguration = g.getFontConfigurationForTitle(skinParam);
|
|
|
|
return label.create(fontConfiguration, HorizontalAlignment.CENTER, skinParam);
|
|
|
|
}
|
|
|
|
|
|
|
|
private HColor getBackColor(UmlDiagramType umlDiagramType) {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
private void drawSingleNode(UGraphic ug, ILeaf leaf, ElkNode elkNode) {
|
|
|
|
final IEntityImage image = printEntityInternal(leaf);
|
|
|
|
// Retrieve coord from ELK
|
2022-09-12 20:08:34 +00:00
|
|
|
final XPoint2D corner = getPosition(elkNode);
|
2021-04-20 20:19:49 +00:00
|
|
|
|
|
|
|
// Print the node image at right coord
|
|
|
|
image.drawU(ug.apply(new UTranslate(corner)));
|
|
|
|
}
|
|
|
|
|
|
|
|
private void drawSingleEdge(UGraphic ug, Link link, ElkEdge edge) {
|
|
|
|
// Unfortunately, we have to translate "edge" in its own "cluster" coordinate
|
2022-09-12 20:08:34 +00:00
|
|
|
final XPoint2D translate = getPosition(edge.getContainingNode());
|
2021-04-20 20:19:49 +00:00
|
|
|
|
2022-06-27 16:33:45 +00:00
|
|
|
final double magicY2 = 0;
|
|
|
|
final IEntity dest = link.getEntity2();
|
|
|
|
if (dest.getUSymbol() instanceof USymbolFolder) {
|
|
|
|
// System.err.println("dest=" + dest);
|
|
|
|
// final IEntityImage image = printEntityInternal((ILeaf) dest);
|
|
|
|
// System.err.println("image=" + image);
|
|
|
|
|
|
|
|
}
|
2021-04-20 20:19:49 +00:00
|
|
|
final ElkPath elkPath = new ElkPath(diagram, SName.classDiagram, link, edge, getLabel(link),
|
2022-06-27 16:33:45 +00:00
|
|
|
getQualifier(link, 1), getQualifier(link, 2), magicY2);
|
2021-04-20 20:19:49 +00:00
|
|
|
elkPath.drawU(ug.apply(new UTranslate(translate)));
|
2021-04-13 17:05:18 +00:00
|
|
|
}
|
|
|
|
|
2022-09-12 20:08:34 +00:00
|
|
|
public XDimension2D calculateDimension(StringBounder stringBounder) {
|
2022-04-10 19:24:55 +00:00
|
|
|
if (minMax == null)
|
2021-04-13 17:05:18 +00:00
|
|
|
throw new UnsupportedOperationException();
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-13 17:05:18 +00:00
|
|
|
return minMax.getDimension();
|
|
|
|
}
|
|
|
|
|
|
|
|
public HColor getBackcolor() {
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
private Collection<ILeaf> getUnpackagedEntities() {
|
2021-05-14 08:42:57 +00:00
|
|
|
final List<ILeaf> result = new ArrayList<>();
|
2022-04-10 19:24:55 +00:00
|
|
|
for (ILeaf ent : diagram.getLeafsvalues())
|
|
|
|
if (diagram.getEntityFactory().getRootGroup() == ent.getParentContainer())
|
2021-04-17 11:19:39 +00:00
|
|
|
result.add(ent);
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
private ElkNode getElkNode(final IEntity entity) {
|
|
|
|
ElkNode node = nodes.get(entity);
|
2022-04-10 19:24:55 +00:00
|
|
|
if (node == null)
|
2021-04-20 20:19:49 +00:00
|
|
|
node = clusters.get(entity);
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
return node;
|
|
|
|
}
|
|
|
|
|
2021-04-13 17:05:18 +00:00
|
|
|
@Override
|
|
|
|
public ImageData createFile(OutputStream os, List<String> dotStrings, FileFormatOption fileFormatOption)
|
|
|
|
throws IOException {
|
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
// https://www.eclipse.org/forums/index.php/t/1095737/
|
2021-04-13 17:05:18 +00:00
|
|
|
try {
|
|
|
|
final ElkNode root = ElkGraphUtil.createGraph();
|
2021-04-15 16:49:28 +00:00
|
|
|
root.setProperty(CoreOptions.DIRECTION, Direction.DOWN);
|
2021-04-20 20:19:49 +00:00
|
|
|
root.setProperty(CoreOptions.HIERARCHY_HANDLING, HierarchyHandling.INCLUDE_CHILDREN);
|
2021-04-14 18:08:29 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
printAllSubgroups(root, diagram.getRootGroup());
|
|
|
|
printEntities(root, getUnpackagedEntities());
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
manageAllEdges();
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
new RecursiveGraphLayoutEngine().layout(root, new NullElkProgressMonitor());
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
final MinMax minMax = TextBlockUtils.getMinMax(new Drawing(null), stringBounder, false);
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
final TextBlock drawable = new Drawing(minMax);
|
|
|
|
return diagram.createImageBuilder(fileFormatOption) //
|
|
|
|
.drawable(drawable) //
|
|
|
|
.write(os); //
|
2021-04-14 18:08:29 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
} catch (Throwable e) {
|
|
|
|
UmlDiagram.exportDiagramError(os, e, fileFormatOption, diagram.seed(), diagram.getMetadata(),
|
|
|
|
diagram.getFlashData(), getFailureText3(e));
|
|
|
|
return ImageDataSimple.error();
|
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
private void printAllSubgroups(ElkNode cluster, IGroup group) {
|
|
|
|
for (IGroup g : diagram.getChildrenGroups(group)) {
|
|
|
|
if (g.isRemoved()) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
if (diagram.isEmpty(g) && g.getGroupType() == GroupType.PACKAGE) {
|
|
|
|
final ISkinParam skinParam = diagram.getSkinParam();
|
|
|
|
final EntityFactory entityFactory = diagram.getEntityFactory();
|
|
|
|
final ILeaf folder = entityFactory.createLeafForEmptyGroup(g, skinParam);
|
|
|
|
System.err.println("STILL IN PROGRESS");
|
|
|
|
// printEntityNew(folder);
|
|
|
|
} else {
|
|
|
|
|
|
|
|
// We create the "cluster" in ELK for this group
|
|
|
|
final ElkNode elkCluster = ElkGraphUtil.createNode(cluster);
|
|
|
|
elkCluster.setProperty(CoreOptions.DIRECTION, Direction.DOWN);
|
2021-05-01 17:40:56 +00:00
|
|
|
elkCluster.setProperty(CoreOptions.PADDING, new ElkPadding(40, 15, 15, 15));
|
2021-04-17 11:19:39 +00:00
|
|
|
|
|
|
|
// Not sure this is usefull to put a label on a "cluster"
|
|
|
|
final ElkLabel label = ElkGraphUtil.createLabel(elkCluster);
|
|
|
|
label.setText("C");
|
|
|
|
// We need it anyway to recurse up to the real "root"
|
|
|
|
|
|
|
|
this.clusters.put(g, elkCluster);
|
|
|
|
printSingleGroup(g);
|
2021-04-13 17:05:18 +00:00
|
|
|
}
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
2021-04-14 18:08:29 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
private void printSingleGroup(IGroup g) {
|
2022-04-10 19:24:55 +00:00
|
|
|
if (g.getGroupType() == GroupType.CONCURRENT_STATE)
|
2021-04-17 11:19:39 +00:00
|
|
|
return;
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
this.printEntities(clusters.get(g), g.getLeafsDirect());
|
|
|
|
printAllSubgroups(clusters.get(g), g);
|
|
|
|
}
|
2021-04-15 16:49:28 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
private void printEntities(ElkNode parent, Collection<ILeaf> entities) {
|
|
|
|
// Convert all "leaf" to ELK node
|
|
|
|
for (ILeaf ent : entities) {
|
2022-04-10 19:24:55 +00:00
|
|
|
if (ent.isRemoved())
|
2021-04-17 11:19:39 +00:00
|
|
|
continue;
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
manageSingleNode(parent, ent);
|
|
|
|
}
|
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
private void manageAllEdges() {
|
|
|
|
// Convert all "link" to ELK edge
|
2022-04-10 19:24:55 +00:00
|
|
|
for (final Link link : diagram.getLinks())
|
2021-04-17 11:19:39 +00:00
|
|
|
manageSingleEdge(link);
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
private void manageSingleNode(final ElkNode root, ILeaf leaf) {
|
|
|
|
final IEntityImage image = printEntityInternal(leaf);
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
// Expected dimension of the node
|
2022-09-12 20:08:34 +00:00
|
|
|
final XDimension2D dimension = image.calculateDimension(stringBounder);
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
// Here, we try to tell ELK to use this dimension as node dimension
|
|
|
|
final ElkNode node = ElkGraphUtil.createNode(root);
|
|
|
|
node.setDimensions(dimension.getWidth(), dimension.getHeight());
|
2021-04-13 17:05:18 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
// There is no real "label" here
|
|
|
|
// We just would like to force node dimension
|
|
|
|
final ElkLabel label = ElkGraphUtil.createLabel(node);
|
|
|
|
label.setText("X");
|
|
|
|
|
|
|
|
// I don't know why we have to do this hack, but somebody has to fix it
|
|
|
|
final double VERY_STRANGE_OFFSET = 10;
|
|
|
|
label.setDimensions(dimension.getWidth(), dimension.getHeight() - VERY_STRANGE_OFFSET);
|
|
|
|
|
|
|
|
// No idea of what we are doing here :-)
|
|
|
|
label.setProperty(CoreOptions.NODE_LABELS_PLACEMENT,
|
|
|
|
EnumSet.of(NodeLabelPlacement.INSIDE, NodeLabelPlacement.H_CENTER, NodeLabelPlacement.V_CENTER));
|
2021-04-20 20:19:49 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
// This padding setting have no impact ?
|
|
|
|
// label.setProperty(CoreOptions.NODE_LABELS_PADDING, new ElkPadding(100.0));
|
2021-04-20 20:19:49 +00:00
|
|
|
|
|
|
|
// final EnumSet<SizeConstraint> constraints =
|
|
|
|
// EnumSet.of(SizeConstraint.NODE_LABELS);
|
|
|
|
// node.setProperty(CoreOptions.NODE_SIZE_CONSTRAINTS, constraints);
|
|
|
|
|
|
|
|
// node.setProperty(CoreOptions.NODE_SIZE_OPTIONS,
|
|
|
|
// EnumSet.noneOf(SizeOptions.class));
|
2021-04-17 11:19:39 +00:00
|
|
|
|
|
|
|
// Let's store this
|
|
|
|
nodes.put(leaf, node);
|
|
|
|
}
|
|
|
|
|
|
|
|
private void manageSingleEdge(final Link link) {
|
2021-04-20 20:19:49 +00:00
|
|
|
final ElkNode node1 = getElkNode(link.getEntity1());
|
|
|
|
final ElkNode node2 = getElkNode(link.getEntity2());
|
2021-04-17 11:19:39 +00:00
|
|
|
|
2021-04-20 20:19:49 +00:00
|
|
|
final ElkEdge edge = ElkGraphUtil.createSimpleEdge(node1, node2);
|
2021-04-17 11:19:39 +00:00
|
|
|
|
|
|
|
final TextBlock labelLink = getLabel(link);
|
|
|
|
if (labelLink != null) {
|
|
|
|
final ElkLabel edgeLabel = ElkGraphUtil.createLabel(edge);
|
2022-09-12 20:08:34 +00:00
|
|
|
final XDimension2D dim = labelLink.calculateDimension(stringBounder);
|
2021-04-17 11:19:39 +00:00
|
|
|
edgeLabel.setText("X");
|
|
|
|
edgeLabel.setDimensions(dim.getWidth(), dim.getHeight());
|
|
|
|
// Duplicated, with qualifier, but who cares?
|
|
|
|
edge.setProperty(CoreOptions.EDGE_LABELS_INLINE, true);
|
2021-04-20 20:19:49 +00:00
|
|
|
// edge.setProperty(CoreOptions.EDGE_TYPE, EdgeType.ASSOCIATION);
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
|
|
|
if (link.getQualifier1() != null) {
|
|
|
|
final ElkLabel edgeLabel = ElkGraphUtil.createLabel(edge);
|
2022-09-12 20:08:34 +00:00
|
|
|
final XDimension2D dim = getQualifier(link, 1).calculateDimension(stringBounder);
|
2021-04-17 11:19:39 +00:00
|
|
|
// Nasty trick, we store the kind of label in the text
|
|
|
|
edgeLabel.setText("1");
|
|
|
|
edgeLabel.setDimensions(dim.getWidth(), dim.getHeight());
|
|
|
|
edgeLabel.setProperty(CoreOptions.EDGE_LABELS_PLACEMENT, EdgeLabelPlacement.TAIL);
|
|
|
|
// Duplicated, with main label, but who cares?
|
|
|
|
edge.setProperty(CoreOptions.EDGE_LABELS_INLINE, true);
|
2021-04-20 20:19:49 +00:00
|
|
|
// edge.setProperty(CoreOptions.EDGE_TYPE, EdgeType.ASSOCIATION);
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
|
|
|
if (link.getQualifier2() != null) {
|
|
|
|
final ElkLabel edgeLabel = ElkGraphUtil.createLabel(edge);
|
2022-09-12 20:08:34 +00:00
|
|
|
final XDimension2D dim = getQualifier(link, 2).calculateDimension(stringBounder);
|
2021-04-17 11:19:39 +00:00
|
|
|
// Nasty trick, we store the kind of label in the text
|
|
|
|
edgeLabel.setText("2");
|
|
|
|
edgeLabel.setDimensions(dim.getWidth(), dim.getHeight());
|
|
|
|
edgeLabel.setProperty(CoreOptions.EDGE_LABELS_PLACEMENT, EdgeLabelPlacement.HEAD);
|
|
|
|
// Duplicated, with main label, but who cares?
|
|
|
|
edge.setProperty(CoreOptions.EDGE_LABELS_INLINE, true);
|
2021-04-20 20:19:49 +00:00
|
|
|
// edge.setProperty(CoreOptions.EDGE_TYPE, EdgeType.ASSOCIATION);
|
2021-04-17 11:19:39 +00:00
|
|
|
}
|
2021-04-20 20:19:49 +00:00
|
|
|
|
2021-04-17 11:19:39 +00:00
|
|
|
edges.put(link, edge);
|
2021-04-13 17:05:18 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static private List<String> getFailureText3(Throwable exception) {
|
2022-08-17 17:34:24 +00:00
|
|
|
Logme.error(exception);
|
2021-05-14 08:42:57 +00:00
|
|
|
final List<String> strings = new ArrayList<>();
|
2021-04-13 17:05:18 +00:00
|
|
|
strings.add("An error has occured : " + exception);
|
|
|
|
final String quote = StringUtils.rot(QuoteUtils.getSomeQuote());
|
|
|
|
strings.add("<i>" + quote);
|
|
|
|
strings.add(" ");
|
|
|
|
GraphvizCrash.addProperties(strings);
|
|
|
|
strings.add(" ");
|
|
|
|
strings.add("Sorry, ELK intregration is really alpha feature...");
|
|
|
|
strings.add(" ");
|
|
|
|
strings.add("You should send this diagram and this image to <b>plantuml@gmail.com</b> or");
|
|
|
|
strings.add("post to <b>http://plantuml.com/qa</b> to solve this issue.");
|
|
|
|
strings.add(" ");
|
|
|
|
return strings;
|
|
|
|
}
|
|
|
|
|
|
|
|
private Bibliotekon getBibliotekon() {
|
|
|
|
return dotStringFactory.getBibliotekon();
|
|
|
|
}
|
|
|
|
|
|
|
|
private IEntityImage printEntityInternal(ILeaf ent) {
|
2022-04-10 19:24:55 +00:00
|
|
|
if (ent.isRemoved())
|
2021-04-13 17:05:18 +00:00
|
|
|
throw new IllegalStateException();
|
2022-04-10 19:24:55 +00:00
|
|
|
|
2021-04-13 17:05:18 +00:00
|
|
|
if (ent.getSvekImage() == null) {
|
|
|
|
final ISkinParam skinParam = diagram.getSkinParam();
|
2022-04-10 19:24:55 +00:00
|
|
|
if (skinParam.sameClassWidth())
|
2021-04-13 17:05:18 +00:00
|
|
|
System.err.println("NOT YET IMPLEMENED");
|
|
|
|
|
|
|
|
return GeneralImageBuilder.createEntityImageBlock(ent, skinParam, diagram.isHideEmptyDescriptionForState(),
|
|
|
|
diagram, getBibliotekon(), null, diagram.getUmlDiagramType(), diagram.getLinks());
|
|
|
|
}
|
|
|
|
return ent.getSvekImage();
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|