plantuml/src/net/sourceforge/plantuml/timingdiagram/TimingRuler.java

293 lines
8.8 KiB
Java

/* ========================================================================
* PlantUML : a free UML diagram generator
* ========================================================================
*
* (C) Copyright 2009-2023, 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.timingdiagram;
import java.math.BigDecimal;
import java.util.Collection;
import java.util.Comparator;
import java.util.Set;
import java.util.SortedSet;
import java.util.TreeSet;
import net.sourceforge.plantuml.FontParam;
import net.sourceforge.plantuml.ISkinParam;
import net.sourceforge.plantuml.UseStyle;
import net.sourceforge.plantuml.api.ThemeStyle;
import net.sourceforge.plantuml.awt.geom.Dimension2D;
import net.sourceforge.plantuml.cucadiagram.Display;
import net.sourceforge.plantuml.graphic.FontConfiguration;
import net.sourceforge.plantuml.graphic.HorizontalAlignment;
import net.sourceforge.plantuml.graphic.StringBounder;
import net.sourceforge.plantuml.graphic.TextBlock;
import net.sourceforge.plantuml.style.PName;
import net.sourceforge.plantuml.style.SName;
import net.sourceforge.plantuml.style.Style;
import net.sourceforge.plantuml.style.StyleSignatureBasic;
import net.sourceforge.plantuml.ugraphic.UGraphic;
import net.sourceforge.plantuml.ugraphic.ULine;
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.HColorSet;
import net.sourceforge.plantuml.ugraphic.color.HColorUtils;
public class TimingRuler {
private final SortedSet<TimeTick> times = new TreeSet<>();
private final ISkinParam skinParam;
private long tickIntervalInPixels = 50;
private long tickUnitary;
private TimingFormat format = TimingFormat.DECIMAL;
static UGraphic applyForVLines(UGraphic ug, Style style, ISkinParam skinParam) {
final UStroke stroke = new UStroke(3, 5, 0.5);
final HColor color;
if (UseStyle.useBetaStyle())
color = style.value(PName.LineColor).asColor(skinParam.getThemeStyle(), skinParam.getIHtmlColorSet());
else
color = HColorSet.instance().getColorOrWhite(ThemeStyle.LIGHT_REGULAR, "#AAA");
return ug.apply(stroke).apply(color);
}
public void ensureNotEmpty() {
if (times.size() == 0)
this.times.add(new TimeTick(BigDecimal.ZERO, TimingFormat.DECIMAL));
if (getMax().getTime().signum() > 0 && getMin().getTime().signum() < 0)
this.times.add(new TimeTick(BigDecimal.ZERO, TimingFormat.DECIMAL));
}
public TimingRuler(ISkinParam skinParam) {
this.skinParam = skinParam;
}
public void scaleInPixels(long tick, long pixel) {
if (pixel <= 0 || tick <= 0)
throw new IllegalArgumentException();
this.tickIntervalInPixels = pixel;
this.tickUnitary = tick;
}
private long tickUnitary() {
if (tickUnitary == 0)
return highestCommonFactor();
return tickUnitary;
}
private long highestCommonFactorInternal = -1;
private long highestCommonFactor() {
if (highestCommonFactorInternal == -1) {
for (long tick : getAbsolutesTicks()) {
if (highestCommonFactorInternal == -1) {
highestCommonFactorInternal = tick;
} else {
final long candidate = computeHighestCommonFactor(highestCommonFactorInternal, tick);
final double size = (getMax().getTime().doubleValue() - getMin().getTime().doubleValue())
/ candidate;
if (size > 200)
return highestCommonFactorInternal;
highestCommonFactorInternal = candidate;
}
}
}
return highestCommonFactorInternal;
}
private Set<Long> getAbsolutesTicks() {
final Set<Long> result = new TreeSet<>(new Comparator<Long>() {
public int compare(Long o1, Long o2) {
return o2.compareTo(o1);
}
});
for (TimeTick time : times) {
final long value = Math.abs(time.getTime().longValue());
if (value > 0)
result.add(value);
}
return result;
}
private int getNbTick() {
if (times.size() == 0)
return 1;
final long delta = getMax().getTime().longValue() - getMin().getTime().longValue();
return Math.min(1000, (int) (1 + delta / tickUnitary()));
}
public double getWidth() {
final double delta = getMax().getTime().doubleValue() - getMin().getTime().doubleValue();
return (delta / tickUnitary() + 1) * tickIntervalInPixels;
}
public final double getPosInPixel(TimeTick when) {
return getPosInPixelInternal(when.getTime().doubleValue());
}
private double getPosInPixelInternal(double time) {
time -= getMin().getTime().doubleValue();
return time / tickUnitary() * tickIntervalInPixels;
}
private long tickToTime(int i) {
return tickUnitary * i + getMin().getTime().longValue();
}
public void addTime(TimeTick time) {
this.highestCommonFactorInternal = -1;
times.add(time);
if (time.getFormat() != TimingFormat.DECIMAL)
this.format = time.getFormat();
}
private FontConfiguration getFontConfiguration() {
if (UseStyle.useBetaStyle() == false)
return FontConfiguration.create(skinParam, FontParam.TIMING, null);
return FontConfiguration.create(skinParam, getStyle());
}
private Style getStyle() {
return StyleSignatureBasic.of(SName.root, SName.element, SName.timingDiagram, SName.timeline)
.getMergedStyle(skinParam.getCurrentStyleBuilder());
}
private TextBlock getTimeTextBlock(long time) {
final Display display = Display.getWithNewlines(format.formatTime(time));
return display.create(getFontConfiguration(), HorizontalAlignment.LEFT, skinParam);
}
public void drawTimeAxis(UGraphic ug) {
ug = ug.apply(new UStroke(2.0)).apply(black());
final double tickHeight = 5;
final ULine line = ULine.vline(tickHeight);
final double firstTickPosition = getPosInPixelInternal(getFirstPositiveOrZeroValue().doubleValue());
int nb = 0;
while (firstTickPosition + nb * tickIntervalInPixels <= getWidth()) {
ug.apply(UTranslate.dx(firstTickPosition + nb * tickIntervalInPixels)).draw(line);
nb++;
}
ug.apply(UTranslate.dx(firstTickPosition)).draw(ULine.hline((nb - 1) * tickIntervalInPixels));
for (long round : roundValues()) {
final TextBlock text = getTimeTextBlock(round);
final Dimension2D dim = text.calculateDimension(ug.getStringBounder());
text.drawU(ug.apply(new UTranslate(getPosInPixelInternal(round) - dim.getWidth() / 2, tickHeight + 1)));
}
}
private HColor black() {
if (UseStyle.useBetaStyle() == false)
return HColorUtils.BLACK;
final Style style = StyleSignatureBasic.of(SName.root, SName.element, SName.timingDiagram)
.getMergedStyle(skinParam.getCurrentStyleBuilder());
return style.value(PName.LineColor).asColor(skinParam.getThemeStyle(), skinParam.getIHtmlColorSet());
}
private BigDecimal getFirstPositiveOrZeroValue() {
for (TimeTick time : times)
if (time.getTime().signum() >= 0)
return time.getTime();
throw new IllegalStateException();
}
private Collection<Long> roundValues() {
final SortedSet<Long> result = new TreeSet<>();
if (tickUnitary == 0) {
for (TimeTick tick : times) {
final long round = tick.getTime().longValue();
result.add(round);
}
} else {
final int nb = getNbTick();
for (int i = 0; i <= nb; i++) {
final long round = tickToTime(i);
result.add(round);
}
}
if (result.first() < 0 && result.last() > 0)
result.add(0L);
return result;
}
public void drawVlines(UGraphic ug, double height) {
ug = applyForVLines(ug, getStyle(), skinParam);
final ULine line = ULine.vline(height);
final int nb = getNbTick();
for (int i = 0; i <= nb; i++)
ug.apply(UTranslate.dx(tickIntervalInPixels * i)).draw(line);
}
public double getHeight(StringBounder stringBounder) {
return getTimeTextBlock(0).calculateDimension(stringBounder).getHeight();
}
private TimeTick getMax() {
return times.last();
}
private TimeTick getMin() {
return times.first();
}
private static long computeHighestCommonFactor(long a, long b) {
long r = a;
while (r != 0) {
r = a % b;
a = b;
b = r;
}
return (Math.abs(a));
}
}