net.sourceforge.plantuml.classdiagram.ClassDiagramFactory1317 Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml Show documentation
Show all versions of plantuml Show documentation
PlantUML is a component that allows to quickly write :
* sequence diagram,
* use case diagram,
* class diagram,
* activity diagram,
* component diagram,
* state diagram
* object diagram
/* ========================================================================
* PlantUML : a free UML diagram generator
* ========================================================================
*
* (C) Copyright 2009-2013, Arnaud Roques
*
* Project Info: http://plantuml.sourceforge.net
*
* 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.
*
* [Java is a trademark or registered trademark of Sun Microsystems, Inc.
* in the United States and other countries.]
*
* Original Author: Arnaud Roques
*
* Revision $Revision: 10006 $
*
*/
package net.sourceforge.plantuml.classdiagram;
import java.util.ArrayList;
import java.util.List;
import net.sourceforge.plantuml.AbstractPSystem;
import net.sourceforge.plantuml.UmlDiagramType;
import net.sourceforge.plantuml.classdiagram.command.CommandAddMethod;
import net.sourceforge.plantuml.classdiagram.command.CommandCreateClass;
import net.sourceforge.plantuml.classdiagram.command.CommandCreateClassMultilines;
import net.sourceforge.plantuml.classdiagram.command.CommandDiamondAssociation;
import net.sourceforge.plantuml.classdiagram.command.CommandHideShow;
import net.sourceforge.plantuml.classdiagram.command.CommandHideShow3;
import net.sourceforge.plantuml.classdiagram.command.CommandHideShow4;
import net.sourceforge.plantuml.classdiagram.command.CommandImport;
import net.sourceforge.plantuml.classdiagram.command.CommandLinkClass;
import net.sourceforge.plantuml.classdiagram.command.CommandLinkLollipop;
import net.sourceforge.plantuml.classdiagram.command.CommandMouseOver;
import net.sourceforge.plantuml.classdiagram.command.CommandNamespaceSeparator;
import net.sourceforge.plantuml.classdiagram.command.CommandStereotype;
import net.sourceforge.plantuml.classdiagram.command.CommandUrl;
import net.sourceforge.plantuml.command.Command;
import net.sourceforge.plantuml.command.CommandEndNamespace;
import net.sourceforge.plantuml.command.CommandEndPackage;
import net.sourceforge.plantuml.command.CommandNamespace;
import net.sourceforge.plantuml.command.CommandPackage;
import net.sourceforge.plantuml.command.CommandPackageEmpty;
import net.sourceforge.plantuml.command.CommandPage;
import net.sourceforge.plantuml.command.CommandRankDir;
import net.sourceforge.plantuml.command.UmlDiagramFactory1317;
import net.sourceforge.plantuml.command.note.FactoryNoteCommand;
import net.sourceforge.plantuml.command.note.FactoryNoteOnEntityCommand;
import net.sourceforge.plantuml.command.note.FactoryNoteOnLinkCommand;
import net.sourceforge.plantuml.command.regex.RegexLeaf;
import net.sourceforge.plantuml.cucadiagram.IEntity;
import net.sourceforge.plantuml.cucadiagram.IGroup;
import net.sourceforge.plantuml.cucadiagram.ILeaf;
import net.sourceforge.plantuml.cucadiagram.Link;
import net.sourceforge.plantuml.cucadiagram.LinkDecor;
import net.sourceforge.plantuml.cucadiagram.LinkType;
public class ClassDiagramFactory1317 extends UmlDiagramFactory1317 {
@Override
public ClassDiagram createEmptyDiagram() {
return new ClassDiagram();
}
@Override
protected List createCommands() {
final List cmds = new ArrayList();
addCommonCommands(cmds);
cmds.add(new CommandRankDir());
cmds.add(new CommandPage());
cmds.add(new CommandAddMethod());
cmds.add(new CommandCreateClass());
final FactoryNoteCommand factoryNoteCommand = new FactoryNoteCommand();
cmds.add(factoryNoteCommand.createSingleLine());
cmds.add(new CommandPackage());
cmds.add(new CommandEndPackage());
cmds.add(new CommandPackageEmpty());
cmds.add(new CommandNamespace());
cmds.add(new CommandEndNamespace());
cmds.add(new CommandStereotype());
cmds.add(new CommandLinkClass(UmlDiagramType.CLASS));
cmds.add(new CommandLinkLollipop(UmlDiagramType.CLASS));
cmds.add(new CommandImport());
final FactoryNoteOnEntityCommand factoryNoteOnEntityCommand = new FactoryNoteOnEntityCommand(new RegexLeaf(
"ENTITY", "(" + CommandCreateClass.CODE + "|\"[^\"]+\")"));
cmds.add(factoryNoteOnEntityCommand.createSingleLine());
cmds.add(new CommandUrl());
cmds.add(factoryNoteOnEntityCommand.createMultiLine());
cmds.add(factoryNoteCommand.createMultiLine());
cmds.add(new CommandCreateClassMultilines());
final FactoryNoteOnLinkCommand factoryNoteOnLinkCommand = new FactoryNoteOnLinkCommand();
cmds.add(factoryNoteOnLinkCommand.createSingleLine());
cmds.add(factoryNoteOnLinkCommand.createMultiLine());
cmds.add(new CommandDiamondAssociation());
cmds.add(new CommandMouseOver());
cmds.add(new CommandHideShow3());
cmds.add(new CommandHideShow());
cmds.add(new CommandHideShow4());
cmds.add(new CommandNamespaceSeparator());
return cmds;
}
@Override
public String checkFinalError(AbstractPSystem sys) {
final ClassDiagram system = (ClassDiagram) sys;
for (Link link : system.getLinks()) {
final int len = link.getLength();
if (len == 1) {
for (Link link2 : system.getLinks()) {
if (link2.sameConnections(link) && link2.getLength() != 1) {
link2.setLength(1);
}
}
}
}
for (IGroup g : system.getGroups(true)) {
final List standalones = new ArrayList();
for (ILeaf ent : g.getLeafsDirect()) {
if (system.isStandalone(ent)) {
standalones.add(ent);
}
}
if (standalones.size() < 3) {
continue;
}
putInSquare(system, standalones);
}
return super.checkFinalError(system);
}
private void putInSquare(ClassDiagram system, List standalones) {
final LinkType linkType = new LinkType(LinkDecor.NONE, LinkDecor.NONE).getInvisible();
final int branch = computeBranch(standalones.size());
int headBranch = 0;
for (int i = 1; i < standalones.size(); i++) {
final int dist = i - headBranch;
final IEntity ent2 = standalones.get(i);
final Link link;
if (dist == branch) {
final IEntity ent1 = standalones.get(headBranch);
link = new Link(ent1, ent2, linkType, null, 2);
headBranch = i;
} else {
final IEntity ent1 = standalones.get(i - 1);
link = new Link(ent1, ent2, linkType, null, 1);
}
system.addLink(link);
}
}
static int computeBranch(int size) {
final double sqrt = Math.sqrt(size);
final int r = (int) sqrt;
if (r * r == size) {
return r;
}
return r + 1;
}
}