mirror of https://github.com/octoleo/plantuml.git
120 lines
3.6 KiB
Java
120 lines
3.6 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009-2017, 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.dot;
|
|
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.LinkedHashSet;
|
|
import java.util.List;
|
|
|
|
import net.sourceforge.plantuml.cucadiagram.CucaDiagram;
|
|
import net.sourceforge.plantuml.cucadiagram.GroupRoot;
|
|
import net.sourceforge.plantuml.cucadiagram.GroupType;
|
|
import net.sourceforge.plantuml.cucadiagram.IGroup;
|
|
import net.sourceforge.plantuml.cucadiagram.LeafType;
|
|
import net.sourceforge.plantuml.graphic.StringBounder;
|
|
import net.sourceforge.plantuml.svek.GroupPngMakerState;
|
|
import net.sourceforge.plantuml.svek.IEntityImage;
|
|
|
|
public final class CucaDiagramSimplifierState {
|
|
|
|
private final CucaDiagram diagram;
|
|
private final StringBounder stringBounder;
|
|
|
|
public CucaDiagramSimplifierState(CucaDiagram diagram, List<String> dotStrings, StringBounder stringBounder)
|
|
throws IOException, InterruptedException {
|
|
this.diagram = diagram;
|
|
this.stringBounder = stringBounder;
|
|
boolean changed;
|
|
do {
|
|
changed = false;
|
|
final Collection<IGroup> groups = getOrdered(diagram.getRootGroup());
|
|
for (IGroup g : groups) {
|
|
if (diagram.isAutarkic(g)) {
|
|
final IEntityImage img = computeImage(g);
|
|
g.overrideImage(img, g.getGroupType() == GroupType.CONCURRENT_STATE ? LeafType.STATE_CONCURRENT
|
|
: LeafType.STATE);
|
|
|
|
changed = true;
|
|
}
|
|
}
|
|
} while (changed);
|
|
}
|
|
|
|
private Collection<IGroup> getOrdered(IGroup root) {
|
|
final Collection<IGroup> ordered = new LinkedHashSet<IGroup>();
|
|
ordered.add(root);
|
|
int size = 1;
|
|
while (true) {
|
|
size = ordered.size();
|
|
addOneLevel(ordered);
|
|
if (size == ordered.size()) {
|
|
break;
|
|
}
|
|
}
|
|
final List<IGroup> result = new ArrayList<IGroup>();
|
|
for (IGroup g : ordered) {
|
|
if (g instanceof GroupRoot == false) {
|
|
result.add(0, g);
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
private void addOneLevel(Collection<IGroup> currents) {
|
|
for (IGroup g : new ArrayList<IGroup>(currents)) {
|
|
for (IGroup child : reverse(g.getChildren())) {
|
|
currents.add(child);
|
|
}
|
|
}
|
|
}
|
|
|
|
private List<IGroup> reverse(Collection<IGroup> source) {
|
|
final List<IGroup> result = new ArrayList<IGroup>();
|
|
for (IGroup g : source) {
|
|
result.add(0, g);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
private IEntityImage computeImage(IGroup g) throws IOException, InterruptedException {
|
|
final GroupPngMakerState maker = new GroupPngMakerState(diagram, g, stringBounder);
|
|
return maker.getImage();
|
|
}
|
|
|
|
}
|