1 files changed, 106 insertions, 0 deletions
diff --git a/src/net/sourceforge/plantuml/cucadiagram/LinkConstraint.java b/src/net/sourceforge/plantuml/cucadiagram/LinkConstraint.java
new file mode 100644
index 0000000..fc5dba1
--- /dev/null
+++ b/src/net/sourceforge/plantuml/cucadiagram/LinkConstraint.java
@@ -0,0 +1,106 @@
+/* ========================================================================
+ * 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.cucadiagram;
+
+import java.awt.geom.Dimension2D;
+import java.awt.geom.Point2D;
+
+import net.sourceforge.plantuml.FontParam;
+import net.sourceforge.plantuml.ISkinParam;
+import net.sourceforge.plantuml.graphic.FontConfiguration;
+import net.sourceforge.plantuml.graphic.HorizontalAlignment;
+import net.sourceforge.plantuml.graphic.HtmlColorUtils;
+import net.sourceforge.plantuml.graphic.TextBlock;
+import net.sourceforge.plantuml.ugraphic.UChangeColor;
+import net.sourceforge.plantuml.ugraphic.UGraphic;
+import net.sourceforge.plantuml.ugraphic.ULine;
+import net.sourceforge.plantuml.ugraphic.UStroke;
+import net.sourceforge.plantuml.ugraphic.UTranslate;
+
+public class LinkConstraint {
+
+	private final Link link1;
+	private final Link link2;
+	private final Display display;
+
+	private double x1;
+	private double y1;
+	private double x2;
+	private double y2;
+
+	public LinkConstraint(Link link1, Link link2, Display display) {
+		this.link1 = link1;
+		this.link2 = link2;
+		this.display = display;
+	}
+
+	public void setPosition(Link link, Point2D pt) {
+		if (link == link1) {
+			x1 = pt.getX();
+			y1 = pt.getY();
+		} else if (link == link2) {
+			x2 = pt.getX();
+			y2 = pt.getY();
+		} else {
+			throw new IllegalArgumentException();
+		}
+	}
+
+	public void drawMe(UGraphic ug, ISkinParam skinParam) {
+		if (x1 == 0 && y1 == 0) {
+			return;
+		}
+		if (x2 == 0 && y2 == 0) {
+			return;
+		}
+		ug = ug.apply(new UChangeColor(HtmlColorUtils.BLACK));
+//		ug.apply(new UTranslate(x1, y1)).draw(new URectangle(10, 10));
+//		ug.apply(new UTranslate(x2, y2)).draw(new URectangle(10, 10));
+
+		final ULine line = new ULine(x2 - x1, y2 - y1);
+		ug = ug.apply(new UStroke(3, 3, 1));
+		ug.apply(new UTranslate(x1, y1)).draw(line);
+
+		final TextBlock label = display.create(new FontConfiguration(skinParam, FontParam.ARROW, null),
+				HorizontalAlignment.CENTER, skinParam);
+		final Dimension2D dimLabel = label.calculateDimension(ug.getStringBounder());
+		final double x = (x1 + x2) / 2 - dimLabel.getWidth() / 2;
+		final double y = (y1 + y2) / 2 - dimLabel.getHeight() / 2;
+		label.drawU(ug.apply(new UTranslate(x, y)));
+
+	}
+
+}