mirror of
https://github.com/octoleo/plantuml.git
synced 2024-05-29 06:30:48 +00:00
196 lines
6.5 KiB
Java
196 lines
6.5 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.project.draw;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
|
|
import net.sourceforge.plantuml.sequencediagram.graphic.Segment;
|
|
import net.sourceforge.plantuml.ugraphic.UGraphic;
|
|
import net.sourceforge.plantuml.ugraphic.ULine;
|
|
import net.sourceforge.plantuml.ugraphic.URectangle;
|
|
import net.sourceforge.plantuml.ugraphic.UStroke;
|
|
import net.sourceforge.plantuml.ugraphic.UTranslate;
|
|
import net.sourceforge.plantuml.ugraphic.color.HColor;
|
|
import net.sourceforge.plantuml.ugraphic.color.HColorNone;
|
|
|
|
public class RectangleTask {
|
|
|
|
private final List<Segment> segments;
|
|
private final double round;
|
|
private final int completion;
|
|
|
|
public RectangleTask(double startPos, double endPos, double round, int completion, Collection<Segment> paused) {
|
|
this.round = round;
|
|
this.completion = completion;
|
|
if (startPos < endPos) {
|
|
this.segments = new ArrayList<Segment>(new Segment(startPos, endPos).cutSegmentIfNeed(paused));
|
|
} else {
|
|
this.segments = Collections.singletonList(new Segment(startPos, startPos + 1));
|
|
}
|
|
}
|
|
|
|
private void draw2hlines(UGraphic ug, double height, ULine hline) {
|
|
ug.draw(hline);
|
|
ug.apply(UTranslate.dy(height)).draw(hline);
|
|
}
|
|
|
|
private void drawRect(UGraphic ug, int completion, HColor documentBackground, double width, double height) {
|
|
if (completion == 100 || completion == 0) {
|
|
if (completion == 0)
|
|
ug = ug.apply(documentBackground.bg());
|
|
final URectangle rect = new URectangle(width, height);
|
|
ug.draw(rect);
|
|
} else {
|
|
final URectangle rect1 = new URectangle(width * completion / 100, height);
|
|
ug.draw(rect1);
|
|
final URectangle rect2 = new URectangle(width * (100 - completion) / 100, height);
|
|
ug.apply(documentBackground.bg()).apply(UTranslate.dx(width * completion / 100)).draw(rect2);
|
|
}
|
|
}
|
|
|
|
public void draw(UGraphic ug, double height, HColor documentBackground, boolean oddStart, boolean oddEnd) {
|
|
|
|
if (round == 0) {
|
|
drawWithoutRound(ug, height, documentBackground, oddStart, oddEnd);
|
|
return;
|
|
}
|
|
|
|
if (segments.size() != 1) {
|
|
drawWithRound(ug, height, documentBackground);
|
|
return;
|
|
}
|
|
|
|
assert segments.size() == 1;
|
|
assert round > 0;
|
|
final Segment segment = segments.get(0);
|
|
|
|
final double width = segment.getLength();
|
|
final URectangle partial = new URectangle(width, height).rounded(round);
|
|
if (completion == 100 || completion == 0) {
|
|
if (completion == 0)
|
|
ug = ug.apply(documentBackground.bg());
|
|
if (oddStart && !oddEnd)
|
|
ug.apply(UTranslate.dx(segment.getPos1())).draw(PathUtils.UtoRight(width, height, round));
|
|
else if (!oddStart && oddEnd)
|
|
ug.apply(UTranslate.dx(segment.getPos1())).draw(PathUtils.UtoLeft(width, height, round));
|
|
else
|
|
ug.apply(UTranslate.dx(segment.getPos1())).draw(partial);
|
|
} else {
|
|
final double x1 = width * completion / 100;
|
|
ug.apply(new HColorNone()).apply(UTranslate.dx(segment.getPos1()))
|
|
.draw(PathUtils.UtoLeft(x1, height, round));
|
|
ug.apply(documentBackground.bg()).apply(new HColorNone()).apply(UTranslate.dx(segment.getPos1() + x1))
|
|
.draw(PathUtils.UtoRight(width * (100 - completion) / 100, height, round));
|
|
ug.apply(new HColorNone().bg()).apply(UTranslate.dx(segment.getPos1())).draw(partial);
|
|
}
|
|
|
|
}
|
|
|
|
private void drawWithRound(UGraphic ug, double height, HColor documentBackground) {
|
|
|
|
final Segment first = segments.get(0);
|
|
ug.apply(UTranslate.dx(first.getPos1())).draw(PathUtils.UtoLeft(first.getLength(), height, round));
|
|
|
|
for (int i = 1; i < segments.size() - 1; i++) {
|
|
final Segment segment = segments.get(i);
|
|
drawPartly(ug, segment, height, documentBackground, i);
|
|
}
|
|
|
|
final Segment last = segments.get(segments.size() - 1);
|
|
ug.apply(UTranslate.dx(last.getPos1())).draw(PathUtils.UtoRight(last.getLength(), height, round));
|
|
|
|
drawIntermediateDotted(ug, height);
|
|
}
|
|
|
|
private void drawWithoutRound(UGraphic ug, double height, HColor documentBackground, boolean oddStart,
|
|
boolean oddEnd) {
|
|
final ULine vline = ULine.vline(height);
|
|
|
|
for (int i = 0; i < segments.size(); i++) {
|
|
final Segment segment = segments.get(i);
|
|
drawPartly(ug, segment, height, documentBackground, i);
|
|
|
|
if (!oddStart && i == 0) {
|
|
ug.apply(UTranslate.dx(segment.getPos1())).draw(vline);
|
|
}
|
|
if (!oddEnd && i == segments.size() - 1) {
|
|
ug.apply(UTranslate.dx(segment.getPos2())).draw(vline);
|
|
}
|
|
}
|
|
drawIntermediateDotted(ug, height);
|
|
}
|
|
|
|
private void drawIntermediateDotted(UGraphic ug, double height) {
|
|
ug = ug.apply(new UStroke(2, 3, 1));
|
|
for (int i = 0; i < segments.size() - 1; i++) {
|
|
final double v1 = segments.get(i).getPos2() + 3;
|
|
final double v2 = segments.get(i + 1).getPos1() - 3;
|
|
if (v2 > v1) {
|
|
draw2hlines(ug.apply(UTranslate.dx(v1)), height, ULine.hline(v2 - v1));
|
|
}
|
|
}
|
|
}
|
|
|
|
private void drawPartly(UGraphic ug, final Segment segment, double height, HColor documentBackground, int i) {
|
|
double width = segment.getLength();
|
|
if (i != segments.size() - 1) {
|
|
width++;
|
|
}
|
|
if (width > 0) {
|
|
drawRect(ug.apply(new HColorNone()).apply(UTranslate.dx(segment.getPos1())), completion, documentBackground,
|
|
width, height);
|
|
}
|
|
|
|
double pos1 = segment.getPos1();
|
|
double len = segment.getLength();
|
|
if (i == 0) {
|
|
if (segments.size() > 1) {
|
|
len--;
|
|
}
|
|
} else {
|
|
pos1++;
|
|
len--;
|
|
}
|
|
if (len > 0) {
|
|
draw2hlines(ug.apply(UTranslate.dx(pos1)), height, ULine.hline(len));
|
|
}
|
|
}
|
|
|
|
}
|