net.sourceforge.plantuml.sdot.SmetanaEdge Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml-mit Show documentation
Show all versions of plantuml-mit Show documentation
PlantUML is a component that allows to quickly write diagrams from text.
// THIS FILE HAS BEEN GENERATED BY A PREPROCESSOR.
/* +=======================================================================
* |
* | PlantUML : a free UML diagram generator
* |
* +=======================================================================
*
* (C) Copyright 2009-2024, Arnaud Roques
*
* Project Info: https://plantuml.com
*
* If you like this project or if you find it useful, you can support us at:
*
* https://plantuml.com/patreon (only 1$ per month!)
* https://plantuml.com/liberapay (only 1€ per month!)
* https://plantuml.com/paypal
*
*
* PlantUML is free software; you can redistribute it and/or modify it
* under the terms of the MIT License.
*
* See http://opensource.org/licenses/MIT
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
* IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*
* PlantUML can occasionally display sponsored or advertising messages. Those
* messages are usually generated on welcome or error images and never on
* functional diagrams.
* See https://plantuml.com/professional if you want to remove them
*
* Images (whatever their format : PNG, SVG, EPS...) generated by running PlantUML
* are owned by the author of their corresponding sources code (that is, their
* textual description in PlantUML language). Those images are not covered by
* this MIT license.
*
* The generated images can then be used without any reference to the MIT license.
* It is not even necessary to stipulate that they have been generated with PlantUML,
* although this will be appreciated by the PlantUML team.
*
* There is an exception : if the textual description in PlantUML language is also covered
* by any license, then the generated images are logically covered
* by the very same license.
*
* This is the IGY distribution (Install GraphViz by Yourself).
* You have to install GraphViz and to setup the GRAPHVIZ_DOT environment variable
* (see https://plantuml.com/graphviz-dot )
*
* Icons provided by OpenIconic : https://useiconic.com/open
* Archimate sprites provided by Archi : http://www.archimatetool.com
* Stdlib AWS provided by https://github.com/milo-minderbinder/AWS-PlantUML
* Stdlib Icons provided https://github.com/tupadr3/plantuml-icon-font-sprites
* ASCIIMathML (c) Peter Jipsen http://www.chapman.edu/~jipsen
* ASCIIMathML (c) David Lippman http://www.pierce.ctc.edu/dlippman
* CafeUndZopfli ported by Eugene Klyuchnikov https://github.com/eustas/CafeUndZopfli
* Brotli (c) by the Brotli Authors https://github.com/google/brotli
* Themes (c) by Brett Schwarz https://github.com/bschwarz/puml-themes
* Twemoji (c) by Twitter at https://twemoji.twitter.com/
*
*/
package net.sourceforge.plantuml.sdot;
import h.ST_Agedge_s;
import h.ST_Agedgeinfo_t;
import h.ST_bezier;
import h.ST_pointf;
import h.ST_splines;
import h.ST_textlabel_t;
import net.sourceforge.plantuml.abel.Link;
import net.sourceforge.plantuml.abel.LinkStrategy;
import net.sourceforge.plantuml.cruise.XAbstractEdge;
import net.sourceforge.plantuml.cruise.XEdge;
import net.sourceforge.plantuml.decoration.LinkType;
import net.sourceforge.plantuml.decoration.Rainbow;
import net.sourceforge.plantuml.klimt.UStroke;
import net.sourceforge.plantuml.klimt.UTranslate;
import net.sourceforge.plantuml.klimt.color.ColorType;
import net.sourceforge.plantuml.klimt.color.HColor;
import net.sourceforge.plantuml.klimt.color.HColors;
import net.sourceforge.plantuml.klimt.drawing.UGraphic;
import net.sourceforge.plantuml.klimt.geom.RectangleArea;
import net.sourceforge.plantuml.klimt.geom.XPoint2D;
import net.sourceforge.plantuml.klimt.shape.DotPath;
import net.sourceforge.plantuml.klimt.shape.TextBlock;
import net.sourceforge.plantuml.klimt.shape.UDrawable;
import net.sourceforge.plantuml.klimt.shape.URectangle;
import net.sourceforge.plantuml.skin.LineParam;
import net.sourceforge.plantuml.skin.UmlDiagramType;
import net.sourceforge.plantuml.style.ISkinParam;
import net.sourceforge.plantuml.style.SName;
import net.sourceforge.plantuml.style.Style;
import net.sourceforge.plantuml.style.StyleSignature;
import net.sourceforge.plantuml.style.StyleSignatureBasic;
import net.sourceforge.plantuml.svek.Bibliotekon;
import net.sourceforge.plantuml.svek.Cluster;
import net.sourceforge.plantuml.svek.extremity.Extremity;
import net.sourceforge.plantuml.svek.extremity.ExtremityFactory;
import net.sourceforge.plantuml.url.Url;
import net.sourceforge.plantuml.utils.Direction;
public class SmetanaEdge extends XAbstractEdge implements XEdge, UDrawable {
private final ST_Agedge_s edge;
private final YMirror ymirror;
private final TextBlock label;
private final TextBlock headLabel;
private final TextBlock tailLabel;
public SmetanaEdge(Link link, ST_Agedge_s edge, YMirror ymirror, TextBlock label,
TextBlock tailLabel, TextBlock headLabel, Bibliotekon bibliotekon, ISkinParam skinParam) {
super(link, skinParam, bibliotekon);
this.edge = edge;
this.ymirror = ymirror;
this.label = label;
this.tailLabel = tailLabel;
this.headLabel = headLabel;
}
private LinkStrategy getLinkStrategy() {
return link.getLinkStrategy();
}
public void drawU(UGraphic ug) {
if (link.isHidden())
return;
DotPath dotPath = getDotPathInternal();
if (dotPath != null) {
if (ymirror != null)
dotPath = ymirror.getMirrored(dotPath);
RectangleArea rectangleArea1 = null;
RectangleArea rectangleArea2 = null;
if (link.getEntity1().isGroup()) {
final Cluster cluster1 = bibliotekon.getCluster(link.getEntity1());
rectangleArea1 = cluster1.getRectangleArea();
}
if (link.getEntity2().isGroup()) {
final Cluster cluster2 = bibliotekon.getCluster(link.getEntity2());
rectangleArea2 = cluster2.getRectangleArea();
}
dotPath = dotPath.simulateCompound(rectangleArea2, rectangleArea1);
final Style styleLine = getStyle();
final Rainbow rainbow = Rainbow.build(styleLine, skinParam.getIHtmlColorSet());
// Warning: duplicated from SmetanaPath and SvekEdge
HColor arrowHeadColor = rainbow.getArrowHeadColor();
HColor color = rainbow.getColor();
if (this.link.getColors() != null) {
final HColor newColor = this.link.getColors().getColor(ColorType.ARROW, ColorType.LINE);
if (newColor != null) {
color = newColor;
arrowHeadColor = color;
}
} else if (this.link.getSpecificColor() != null) {
color = this.link.getSpecificColor();
arrowHeadColor = color;
}
ug = ug.apply(HColors.none().bg()).apply(color);
final LinkType linkType = link.getType();
final UStroke suggestedStroke = styleLine.getStroke();
final UStroke defaultThickness = skinParam.getThickness(LineParam.arrow, null);
UStroke stroke;
if (suggestedStroke == null || linkType.getStyle().isNormal() == false)
stroke = linkType.getStroke3(defaultThickness);
else
stroke = linkType.getStroke3(suggestedStroke);
if (link.getColors() != null && link.getColors().getSpecificLineStroke() != null)
stroke = link.getColors().getSpecificLineStroke();
final Url url = link.getUrl();
if (url != null)
ug.startUrl(url);
printExtremityAtStart(dotPath, ug.apply(color));
printExtremityAtEnd(dotPath, ug.apply(color));
ug.apply(stroke).apply(color).draw(dotPath);
if (url != null)
ug.closeUrl();
}
if (getLabelRectangleTranslate("label") != null)
label.drawU(ug.apply(getLabelRectangleTranslate("label")));
if (getLabelRectangleTranslate("head_label") != null)
headLabel.drawU(ug.apply(getLabelRectangleTranslate("head_label")));
if (getLabelRectangleTranslate("tail_label") != null)
tailLabel.drawU(ug.apply(getLabelRectangleTranslate("tail_label")));
// printDebug(ug);
}
private Style getStyle() {
final StyleSignature result = StyleSignatureBasic
.of(SName.root, SName.element, diagramType().getStyleName(), SName.arrow)
.withTOBECHANGED(link.getStereotype());
return result.getMergedStyle(skinParam.getCurrentStyleBuilder());
}
public XPoint2D getStartPoint() {
final DotPath dotPath = getDotPathInternal();
XPoint2D pt = dotPath.getStartPoint();
if (ymirror != null)
pt = ymirror.getMirrored(pt);
return pt;
}
public XPoint2D getEndPoint() {
final DotPath dotPath = getDotPathInternal();
XPoint2D pt = dotPath.getEndPoint();
if (ymirror != null)
pt = ymirror.getMirrored(pt);
return pt;
}
private void printExtremityAtStart(DotPath dotPath, UGraphic ug) {
final ExtremityFactory extremityFactory2 = link.getType().getDecor2()
.getExtremityFactoryComplete(skinParam.getBackgroundColor());
if (extremityFactory2 == null)
return;
final XPoint2D p0 = dotPath.getStartPoint();
final double startAngle = dotPath.getStartAngle() + Math.PI;
try {
final Extremity extremity2 = (Extremity) extremityFactory2.createUDrawable(p0, startAngle, null);
if (extremity2 != null) {
if (getLinkStrategy() == LinkStrategy.SIMPLIER) {
final double decorationLength = extremity2.getDecorationLength();
dotPath.moveStartPoint(new UTranslate(decorationLength, 0).rotate(startAngle - Math.PI));
}
extremity2.drawU(ug);
}
} catch (UnsupportedOperationException e) {
e.printStackTrace();
System.err.println("CANNOT DRAW printExtremityAtStart");
}
}
private void printExtremityAtEnd(DotPath dotPath, UGraphic ug) {
final ExtremityFactory extremityFactory1 = link.getType().getDecor1()
.getExtremityFactoryComplete(skinParam.getBackgroundColor());
if (extremityFactory1 == null)
return;
final XPoint2D p0 = dotPath.getEndPoint();
final double endAngle = dotPath.getEndAngle();
try {
final Extremity extremity1 = (Extremity) extremityFactory1.createUDrawable(p0, endAngle, null);
if (extremity1 != null) {
if (getLinkStrategy() == LinkStrategy.SIMPLIER) {
final double decorationLength = extremity1.getDecorationLength();
dotPath.moveEndPoint(new UTranslate(decorationLength, 0).rotate(endAngle - Math.PI));
}
extremity1.drawU(ug);
}
} catch (UnsupportedOperationException e) {
e.printStackTrace();
System.err.println("CANNOT DRAW printExtremityAtEnd");
}
}
// private void printDebug(UGraphic ug) {
// ug = ug.apply(HColors.BLUE).apply(HColors.BLUE.bg());
// final ST_splines splines = getSplines(edge);
// final ST_bezier beziers = splines.list.get__(0);
// for (int i = 0; i < beziers.size; i++) {
// XPoint2D pt = getPoint(splines, i);
// if (ymirror != null)
// pt = ymirror.getMirrored(pt);
//
// ug.apply(UTranslate.point(pt).compose(new UTranslate(-1, -1))).draw(UEllipse.build(3, 3));
// }
// if (getLabelRectangleTranslate("label") != null && getLabelURectangle() != null) {
// ug = ug.apply(HColors.BLUE).apply(HColors.none().bg());
// ug.apply(getLabelRectangleTranslate("label")).draw(getLabelURectangle());
// }
//
// }
private URectangle getLabelURectangle() {
final ST_Agedgeinfo_t data = (ST_Agedgeinfo_t) edge.data.castTo(ST_Agedgeinfo_t.class);
ST_textlabel_t label = (ST_textlabel_t) data.label;
if (label == null)
return null;
final BoxInfo boxInfo = BoxInfo.fromTextlabel(label);
return URectangle.build(boxInfo.getDimension());
}
private UTranslate getLabelRectangleTranslate(String fieldName) {
// final String fieldName = "label";
final ST_Agedgeinfo_t data = (ST_Agedgeinfo_t) edge.data;
ST_textlabel_t label = null;
if (fieldName.equals("label"))
label = data.label;
else if (fieldName.equals("head_label"))
label = data.head_label;
else if (fieldName.equals("tail_label"))
label = data.tail_label;
if (label == null)
return null;
final BoxInfo boxInfo = BoxInfo.fromTextlabel(label);
if (ymirror == null)
return new UTranslate(boxInfo.getLowerLeft().getX(), boxInfo.getUpperRight().getY());
return ymirror.getMirrored(UTranslate.point(boxInfo.getLowerLeft()));
}
// private DotPath dotPath;
private DotPath getDotPathInternal() {
// if (dotPath != null)
// return dotPath;
final ST_Agedgeinfo_t data = (ST_Agedgeinfo_t) edge.data;
final ST_splines splines = data.spl;
DotPath dotPath = new DotPath();
final ST_bezier beziers = (ST_bezier) splines.list.get__(0);
final XPoint2D pt1 = getPoint(splines, 0);
final XPoint2D pt2 = getPoint(splines, 1);
final XPoint2D pt3 = getPoint(splines, 2);
final XPoint2D pt4 = getPoint(splines, 3);
dotPath = dotPath.addCurve(pt1, pt2, pt3, pt4);
final int n = beziers.size;
for (int i = 4; i < n; i += 3) {
final XPoint2D ppt2 = getPoint(splines, i);
final XPoint2D ppt3 = getPoint(splines, i + 1);
final XPoint2D ppt4 = getPoint(splines, i + 2);
dotPath = dotPath.addCurve(ppt2, ppt3, ppt4);
}
return dotPath;
}
private XPoint2D getPoint(ST_splines splines, int i) {
final ST_bezier beziers = (ST_bezier) splines.list.get__(0);
final ST_pointf pt = beziers.list.get__(i);
return new XPoint2D(pt.x, pt.y);
}
@Override
public void moveDelta(double deltaX, double deltaY) {
throw new UnsupportedOperationException("refactor in progress");
}
@Override
public boolean isHidden() {
throw new UnsupportedOperationException("refactor in progress");
}
@Override
public Direction getArrowDirection() {
throw new UnsupportedOperationException("refactor in progress");
}
@Override
public double getArrowDirectionInRadian() {
throw new UnsupportedOperationException("refactor in progress");
}
}