diff options
Diffstat (limited to 'src/net/sourceforge/plantuml/classdiagram/command/CommandLinkLollipop.java')
| -rw-r--r-- | src/net/sourceforge/plantuml/classdiagram/command/CommandLinkLollipop.java | 224 |
1 files changed, 224 insertions, 0 deletions
diff --git a/src/net/sourceforge/plantuml/classdiagram/command/CommandLinkLollipop.java b/src/net/sourceforge/plantuml/classdiagram/command/CommandLinkLollipop.java new file mode 100644 index 0000000..ee926e3 --- /dev/null +++ b/src/net/sourceforge/plantuml/classdiagram/command/CommandLinkLollipop.java @@ -0,0 +1,224 @@ +/* ========================================================================
+ * PlantUML : a free UML diagram generator
+ * ========================================================================
+ *
+ * (C) Copyright 2009-2014, Arnaud Roques
+ *
+ * Project Info: http://plantuml.sourceforge.net
+ *
+ * This file is part of PlantUML.
+ *
+ * Licensed under The MIT License (Massachusetts Institute of Technology 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.
+ *
+ *
+ * Original Author: Arnaud Roques
+ */
+package net.sourceforge.plantuml.classdiagram.command;
+
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import net.sourceforge.plantuml.UmlDiagramType;
+import net.sourceforge.plantuml.command.CommandExecutionResult;
+import net.sourceforge.plantuml.command.SingleLineCommand2;
+import net.sourceforge.plantuml.command.regex.MyPattern;
+import net.sourceforge.plantuml.command.regex.RegexConcat;
+import net.sourceforge.plantuml.command.regex.RegexLeaf;
+import net.sourceforge.plantuml.command.regex.RegexOr;
+import net.sourceforge.plantuml.command.regex.RegexResult;
+import net.sourceforge.plantuml.cucadiagram.Code;
+import net.sourceforge.plantuml.cucadiagram.Display;
+import net.sourceforge.plantuml.cucadiagram.IEntity;
+import net.sourceforge.plantuml.cucadiagram.LeafType;
+import net.sourceforge.plantuml.cucadiagram.Link;
+import net.sourceforge.plantuml.cucadiagram.LinkDecor;
+import net.sourceforge.plantuml.cucadiagram.LinkType;
+import net.sourceforge.plantuml.objectdiagram.AbstractClassOrObjectDiagram;
+import net.sourceforge.plantuml.StringUtils;
+import net.sourceforge.plantuml.utils.UniqueSequence;
+
+final public class CommandLinkLollipop extends SingleLineCommand2<AbstractClassOrObjectDiagram> {
+
+ public CommandLinkLollipop(UmlDiagramType umlDiagramType) {
+ super(getRegexConcat(umlDiagramType));
+ }
+
+ static RegexConcat getRegexConcat(UmlDiagramType umlDiagramType) {
+ return new RegexConcat(new RegexLeaf("HEADER", "^(?:@([\\d.]+)[%s]+)?"), //
+ new RegexLeaf("ENT1", "(?:" + optionalKeywords(umlDiagramType) + "[%s]+)?"
+ + "(\\.?[\\p{L}0-9_]+(?:\\.[\\p{L}0-9_]+)*|[%g][^%g]+[%g])[%s]*(\\<\\<.*\\>\\>)?"), //
+ new RegexLeaf("[%s]*"), //
+ new RegexLeaf("FIRST_LABEL", "(?:[%g]([^%g]+)[%g])?"), //
+ new RegexLeaf("[%s]*"), //
+ new RegexOr(new RegexLeaf("LOL_THEN_ENT", "\\(\\)([-=.]+)"), //
+ new RegexLeaf("ENT_THEN_LOL", "([-=.]+)\\(\\)")), //
+ new RegexLeaf("[%s]*"), //
+ new RegexLeaf("SECOND_LABEL", "(?:[%g]([^%g]+)[%g])?"), //
+ new RegexLeaf("[%s]*"), //
+ new RegexLeaf("ENT2", "(?:" + optionalKeywords(umlDiagramType) + "[%s]+)?"
+ + "(\\.?[\\p{L}0-9_]+(?:\\.[\\p{L}0-9_]+)*|[%g][^%g]+[%g])[%s]*(\\<\\<.*\\>\\>)?"), //
+ new RegexLeaf("[%s]*"), //
+ new RegexLeaf("LABEL_LINK", "(?::[%s]*(.+))?"), //
+ new RegexLeaf("$"));
+ }
+
+ private static String optionalKeywords(UmlDiagramType type) {
+ if (type == UmlDiagramType.CLASS) {
+ return "(interface|enum|annotation|abstract[%s]+class|abstract|class)";
+ }
+ if (type == UmlDiagramType.OBJECT) {
+ return "(object)";
+ }
+ throw new IllegalArgumentException();
+ }
+
+ @Override
+ protected CommandExecutionResult executeArg(AbstractClassOrObjectDiagram diagram, RegexResult arg) {
+
+ final Code ent1 = Code.of(arg.get("ENT1", 1));
+ final Code ent2 = Code.of(arg.get("ENT2", 1));
+
+ final IEntity cl1;
+ final IEntity cl2;
+ final IEntity normalEntity;
+
+ final String suffix = "lol" + UniqueSequence.getValue();
+ if (arg.get("LOL_THEN_ENT", 0) == null) {
+ assert arg.get("ENT_THEN_LOL", 0) != null;
+ cl1 = diagram.getOrCreateLeaf(ent1, null, null);
+ cl2 = diagram.createLeaf(cl1.getCode().addSuffix(suffix), Display.getWithNewlines(ent2), LeafType.LOLLIPOP, null);
+ normalEntity = cl1;
+ } else {
+ cl2 = diagram.getOrCreateLeaf(ent2, null, null);
+ cl1 = diagram.createLeaf(cl2.getCode().addSuffix(suffix), Display.getWithNewlines(ent1), LeafType.LOLLIPOP, null);
+ normalEntity = cl2;
+ }
+
+ final LinkType linkType = getLinkType(arg);
+ final String queue = getQueue(arg);
+
+ int length = queue.length();
+ if (length == 1 && diagram.getNbOfHozizontalLollipop(normalEntity) > 1) {
+ length++;
+ }
+
+ String firstLabel = arg.get("FIRST_LABEL", 0);
+ String secondLabel = arg.get("SECOND_LABEL", 0);
+
+ String labelLink = null;
+
+ if (arg.get("LABEL_LINK", 0) != null) {
+ labelLink = arg.get("LABEL_LINK", 0);
+ if (firstLabel == null && secondLabel == null) {
+ final Pattern p1 = MyPattern.cmpile("^\"([^\"]+)\"([^\"]+)\"([^\"]+)\"$");
+ final Matcher m1 = p1.matcher(labelLink);
+ if (m1.matches()) {
+ firstLabel = m1.group(1);
+ labelLink = StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(m1.group(2).trim()).trim();
+ secondLabel = m1.group(3);
+ } else {
+ final Pattern p2 = MyPattern.cmpile("^\"([^\"]+)\"([^\"]+)$");
+ final Matcher m2 = p2.matcher(labelLink);
+ if (m2.matches()) {
+ firstLabel = m2.group(1);
+ labelLink = StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(m2.group(2).trim()).trim();
+ secondLabel = null;
+ } else {
+ final Pattern p3 = MyPattern.cmpile("^([^\"]+)\"([^\"]+)\"$");
+ final Matcher m3 = p3.matcher(labelLink);
+ if (m3.matches()) {
+ firstLabel = null;
+ labelLink = StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(m3.group(1).trim())
+ .trim();
+ secondLabel = m3.group(2);
+ }
+ }
+ }
+ }
+ labelLink = StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(labelLink);
+ } /*
+ * else if (arg.get("LABEL_LINK_XT").get(0) != null || arg.get("LABEL_LINK_XT").get(1) != null ||
+ * arg.get("LABEL_LINK_XT").get(2) != null) { labelLink = arg.get("LABEL_LINK_XT").get(1); firstLabel =
+ * merge(firstLabel, arg.get("LABEL_LINK_XT").get(0)); secondLabel = merge(arg.get("LABEL_LINK_XT").get(2),
+ * secondLabel); }
+ */
+
+ final Link link = new Link(cl1, cl2, linkType, Display.getWithNewlines(labelLink), length, firstLabel,
+ secondLabel, diagram.getLabeldistance(), diagram.getLabelangle());
+ diagram.resetPragmaLabel();
+ addLink(diagram, link, arg.get("HEADER", 0));
+
+ return CommandExecutionResult.ok();
+ }
+
+ // private String merge(String a, String b) {
+ // if (a == null && b == null) {
+ // return null;
+ // }
+ // if (a == null && b != null) {
+ // return StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(b);
+ // }
+ // if (b == null && a != null) {
+ // return StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(a);
+ // }
+ // return StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(a) +
+ // "\\n"
+ // + StringUtils.eventuallyRemoveStartingAndEndingDoubleQuote(b);
+ // }
+
+ private void addLink(AbstractClassOrObjectDiagram diagram, Link link, String weight) {
+ diagram.addLink(link);
+ if (weight == null) {
+ // final LinkType type = link.getType();
+ // --|> highest
+ // --*, -->, --o normal
+ // ..*, ..>, ..o lowest
+ // if (type.isDashed() == false) {
+ // if (type.contains(LinkDecor.EXTENDS)) {
+ // link.setWeight(3);
+ // }
+ // if (type.contains(LinkDecor.ARROW) ||
+ // type.contains(LinkDecor.COMPOSITION)
+ // || type.contains(LinkDecor.AGREGATION)) {
+ // link.setWeight(2);
+ // }
+ // }
+ } else {
+ link.setWeight(Double.parseDouble(weight));
+ }
+ }
+
+ private LinkType getLinkType(RegexResult arg) {
+ return new LinkType(LinkDecor.NONE, LinkDecor.NONE);
+ }
+
+ private String getQueue(RegexResult arg) {
+ if (arg.get("LOL_THEN_ENT", 0) != null) {
+ return arg.get("LOL_THEN_ENT", 0).trim();
+ }
+ if (arg.get("ENT_THEN_LOL", 0) != null) {
+ return arg.get("ENT_THEN_LOL", 0).trim();
+ }
+ throw new IllegalArgumentException();
+ }
+
+}
|