net.sourceforge.plantuml.cucadiagram.BodyEnhanced2 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: 7637 $
*
*/
package net.sourceforge.plantuml.cucadiagram;
import java.awt.geom.Dimension2D;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import net.sourceforge.plantuml.FontParam;
import net.sourceforge.plantuml.ISkinParam;
import net.sourceforge.plantuml.SpriteContainer;
import net.sourceforge.plantuml.StringUtils;
import net.sourceforge.plantuml.Url;
import net.sourceforge.plantuml.graphic.FontConfiguration;
import net.sourceforge.plantuml.graphic.HorizontalAlignement;
import net.sourceforge.plantuml.graphic.StringBounder;
import net.sourceforge.plantuml.graphic.TextBlock;
import net.sourceforge.plantuml.graphic.TextBlockLineBefore2;
import net.sourceforge.plantuml.graphic.TextBlockUtils;
import net.sourceforge.plantuml.graphic.TextBlockVertical2;
import net.sourceforge.plantuml.skin.rose.Rose;
import net.sourceforge.plantuml.ugraphic.UGraphic;
public class BodyEnhanced2 implements TextBlock {
private TextBlock area2;
private final FontConfiguration titleConfig;
private final List rawBody;
private final FontParam fontParam;
private final ISkinParam skinParam;
private final boolean lineFirst;
private final HorizontalAlignement align;
private final boolean manageHorizontalLine;
private final boolean manageModifier;
private final List urls = new ArrayList();
public BodyEnhanced2(List rawBody, FontParam fontParam, ISkinParam skinParam, boolean manageModifier) {
this.rawBody = new ArrayList(rawBody);
this.fontParam = fontParam;
this.skinParam = skinParam;
this.titleConfig = new FontConfiguration(skinParam.getFont(fontParam, null), new Rose().getFontColor(skinParam,
fontParam));
this.lineFirst = true;
this.align = HorizontalAlignement.LEFT;
this.manageHorizontalLine = true;
this.manageModifier = manageModifier;
}
public BodyEnhanced2(Display display, FontParam fontParam, ISkinParam skinParam, HorizontalAlignement align,
Stereotype stereotype, boolean manageHorizontalLine, boolean manageModifier) {
this.rawBody = new ArrayList();
for (CharSequence s : display) {
this.rawBody.add(s.toString());
}
this.fontParam = fontParam;
this.skinParam = skinParam;
this.titleConfig = new FontConfiguration(skinParam, fontParam, stereotype);
this.lineFirst = false;
this.align = align;
this.manageHorizontalLine = manageHorizontalLine;
this.manageModifier = manageModifier;
}
private TextBlock decorate(StringBounder stringBounder, TextBlock b, char separator, TextBlock title) {
if (separator == 0) {
return b;
}
if (title == null) {
return new TextBlockLineBefore2(TextBlockUtils.withMargin(b, 6, 4), separator);
}
final Dimension2D dimTitle = title.calculateDimension(stringBounder);
final TextBlock raw = new TextBlockLineBefore2(TextBlockUtils.withMargin(b, 6, 6, dimTitle.getHeight() / 2, 4),
separator, title);
return TextBlockUtils.withMargin(raw, 0, 0, dimTitle.getHeight() / 2, 0);
}
public Dimension2D calculateDimension(StringBounder stringBounder) {
return getArea(stringBounder).calculateDimension(stringBounder);
}
private TextBlock getArea(StringBounder stringBounder) {
if (area2 != null) {
return area2;
}
urls.clear();
final List blocks = new ArrayList();
char separator = lineFirst ? '_' : 0;
TextBlock title = null;
List members = new ArrayList();
for (String s : rawBody) {
if (manageHorizontalLine && isBlockSeparator(s)) {
blocks.add(decorate(stringBounder, new MethodsOrFieldsArea(members, fontParam, skinParam, align),
separator, title));
separator = s.charAt(0);
title = getTitle(s, skinParam);
members = new ArrayList();
} else {
final Member m = new MemberImpl(s, StringUtils.isMethod(s), manageModifier);
members.add(m);
if (m.getUrl() != null) {
urls.add(m.getUrl());
}
}
}
blocks.add(decorate(stringBounder, new MethodsOrFieldsArea(members, fontParam, skinParam, align), separator,
title));
if (blocks.size() == 1) {
this.area2 = blocks.get(0);
} else {
this.area2 = new TextBlockVertical2(blocks, align);
}
return area2;
}
public static boolean isBlockSeparator(String s) {
if (s.startsWith("--") && s.endsWith("--")) {
return true;
}
if (s.startsWith("==") && s.endsWith("==")) {
return true;
}
if (s.startsWith("..") && s.endsWith("..")) {
return true;
}
if (s.startsWith("__") && s.endsWith("__")) {
return true;
}
return false;
}
private TextBlock getTitle(String s, SpriteContainer spriteContainer) {
if (s.length() <= 4) {
return null;
}
s = s.substring(2, s.length() - 2).trim();
return TextBlockUtils.create(Display.getWithNewlines(s), titleConfig, HorizontalAlignement.LEFT,
spriteContainer);
}
public void drawU(UGraphic ug, double x, double y) {
getArea(ug.getStringBounder()).drawU(ug, x, y);
}
public List getUrls() {
return Collections.unmodifiableList(urls);
}
}