mirror of https://github.com/octoleo/plantuml.git
112 lines
3.3 KiB
Java
112 lines
3.3 KiB
Java
/* ========================================================================
|
|
* 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.jsondiagram;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
|
|
import net.sourceforge.plantuml.ISkinParam;
|
|
import net.sourceforge.plantuml.StringLocated;
|
|
import net.sourceforge.plantuml.command.BlocLines;
|
|
import net.sourceforge.plantuml.style.Style;
|
|
import net.sourceforge.plantuml.style.StyleBuilder;
|
|
import net.sourceforge.plantuml.style.StyleLoader;
|
|
|
|
public class StyleExtractor {
|
|
|
|
private final List<String> list = new ArrayList<>();
|
|
private final List<StringLocated> style = new ArrayList<>();
|
|
private String title = null;
|
|
|
|
public StyleExtractor(Iterator<StringLocated> data) {
|
|
while (data.hasNext()) {
|
|
StringLocated line = data.next();
|
|
if (startStyle(line)) {
|
|
while (data.hasNext()) {
|
|
style.add(line);
|
|
if (endStyle(line))
|
|
break;
|
|
line = data.next();
|
|
}
|
|
} else if (line.getString().trim().startsWith("!assume ")) {
|
|
// Ignore
|
|
} else if (line.getString().trim().startsWith("!pragma ")) {
|
|
// Ignore
|
|
} else if (line.getString().trim().startsWith("title ")) {
|
|
this.title = line.getString().trim().substring("title ".length()).trim();
|
|
} else if (line.getString().trim().startsWith("skinparam ")) {
|
|
if (line.getString().trim().contains("{")) {
|
|
while (data.hasNext()) {
|
|
if (line.getString().trim().equals("}"))
|
|
break;
|
|
line = data.next();
|
|
}
|
|
}
|
|
} else {
|
|
list.add(line.getString());
|
|
}
|
|
}
|
|
}
|
|
|
|
private boolean startStyle(StringLocated line) {
|
|
return line.getString().trim().equals("<style>");
|
|
}
|
|
|
|
private boolean endStyle(StringLocated line) {
|
|
return line.getString().trim().equals("</style>");
|
|
}
|
|
|
|
public void applyStyles(ISkinParam skinParam) {
|
|
if (style.size() > 0) {
|
|
final StyleBuilder styleBuilder = skinParam.getCurrentStyleBuilder();
|
|
final BlocLines blocLines = BlocLines.from(style);
|
|
for (Style modifiedStyle : StyleLoader.getDeclaredStyles(blocLines.subExtract(1, 1), styleBuilder)) {
|
|
skinParam.muteStyle(modifiedStyle);
|
|
}
|
|
}
|
|
}
|
|
|
|
public Iterator<String> getIterator() {
|
|
return list.iterator();
|
|
}
|
|
|
|
public String getTitle() {
|
|
return title;
|
|
}
|
|
|
|
}
|