mirror of
https://github.com/octoleo/plantuml.git
synced 2024-06-02 08:30:49 +00:00
226 lines
7.0 KiB
Java
226 lines
7.0 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009-2024, Arnaud Roques
|
|
*
|
|
* Project Info: https://plantuml.com
|
|
*
|
|
* If you like this project or if you find it useful, you can support us at:
|
|
*
|
|
* https://plantuml.com/patreon (only 1$ per month!)
|
|
* https://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.klimt.drawing.g2d;
|
|
|
|
import java.awt.Graphics2D;
|
|
import java.awt.Shape;
|
|
import java.awt.geom.Rectangle2D;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.IOException;
|
|
import java.io.OutputStream;
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Objects;
|
|
import java.util.Set;
|
|
|
|
import net.sourceforge.plantuml.FileFormat;
|
|
import net.sourceforge.plantuml.klimt.UAntiAliasing;
|
|
import net.sourceforge.plantuml.klimt.UChange;
|
|
import net.sourceforge.plantuml.klimt.UClip;
|
|
import net.sourceforge.plantuml.klimt.UPath;
|
|
import net.sourceforge.plantuml.klimt.color.ColorMapper;
|
|
import net.sourceforge.plantuml.klimt.color.HColor;
|
|
import net.sourceforge.plantuml.klimt.drawing.AbstractCommonUGraphic;
|
|
import net.sourceforge.plantuml.klimt.drawing.AbstractUGraphic;
|
|
import net.sourceforge.plantuml.klimt.drawing.UGraphic;
|
|
import net.sourceforge.plantuml.klimt.font.StringBounder;
|
|
import net.sourceforge.plantuml.klimt.geom.EnsureVisible;
|
|
import net.sourceforge.plantuml.klimt.shape.DotPath;
|
|
import net.sourceforge.plantuml.klimt.shape.UCenteredCharacter;
|
|
import net.sourceforge.plantuml.klimt.shape.UEllipse;
|
|
import net.sourceforge.plantuml.klimt.shape.UImage;
|
|
import net.sourceforge.plantuml.klimt.shape.UImageSvg;
|
|
import net.sourceforge.plantuml.klimt.shape.ULine;
|
|
import net.sourceforge.plantuml.klimt.shape.UPixel;
|
|
import net.sourceforge.plantuml.klimt.shape.UPolygon;
|
|
import net.sourceforge.plantuml.klimt.shape.URectangle;
|
|
import net.sourceforge.plantuml.klimt.shape.UText;
|
|
import net.sourceforge.plantuml.png.PngIO;
|
|
import net.sourceforge.plantuml.security.SecurityUtils;
|
|
import net.sourceforge.plantuml.url.Url;
|
|
|
|
public class UGraphicG2d extends AbstractUGraphic<Graphics2D> implements EnsureVisible {
|
|
|
|
private BufferedImage bufferedImage;
|
|
|
|
private final double dpiFactor;
|
|
private final FileFormat format;
|
|
|
|
private UAntiAliasing antiAliasing = UAntiAliasing.ANTI_ALIASING_ON;
|
|
|
|
private List<Url> urls = new ArrayList<>();
|
|
private Set<Url> allUrls = new HashSet<>();
|
|
|
|
public final Set<Url> getAllUrlsEncountered() {
|
|
return Collections.unmodifiableSet(allUrls);
|
|
}
|
|
|
|
@Override
|
|
public UGraphic apply(UChange change) {
|
|
final UGraphicG2d copy = (UGraphicG2d) super.apply(change);
|
|
if (change instanceof UAntiAliasing)
|
|
copy.antiAliasing = (UAntiAliasing) change;
|
|
|
|
return copy;
|
|
}
|
|
|
|
@Override
|
|
protected AbstractCommonUGraphic copyUGraphic() {
|
|
final UGraphicG2d result = new UGraphicG2d(this);
|
|
return result;
|
|
}
|
|
|
|
private UGraphicG2d(UGraphicG2d other) {
|
|
super(other.getStringBounder());
|
|
copy(other);
|
|
this.dpiFactor = other.dpiFactor;
|
|
this.bufferedImage = other.bufferedImage;
|
|
this.urls = other.urls;
|
|
this.allUrls = other.allUrls;
|
|
this.antiAliasing = other.antiAliasing;
|
|
this.format = other.format;
|
|
register(dpiFactor);
|
|
}
|
|
|
|
public UGraphicG2d(HColor defaultBackground, ColorMapper colorMapper, StringBounder stringBounder, Graphics2D g2d,
|
|
double dpiFactor, FileFormat format) {
|
|
this(defaultBackground, colorMapper, stringBounder, g2d, dpiFactor, 0, 0, format);
|
|
}
|
|
|
|
public UGraphicG2d(HColor defaultBackground, ColorMapper colorMapper, StringBounder stringBounder, Graphics2D g2d,
|
|
double dpiFactor, double dx, double dy, FileFormat format) {
|
|
super(stringBounder);
|
|
copy(defaultBackground, colorMapper, g2d);
|
|
this.format = format;
|
|
this.dpiFactor = dpiFactor;
|
|
if (dpiFactor != 1.0)
|
|
g2d.scale(dpiFactor, dpiFactor);
|
|
|
|
register(dpiFactor);
|
|
}
|
|
|
|
private void register(double dpiFactor) {
|
|
registerDriver(URectangle.class, new DriverRectangleG2d(dpiFactor, this));
|
|
// ::comment when __CORE__
|
|
if (dpiFactor != 1.0)
|
|
registerDriver(UText.class, new DriverTextAsPathG2d(this, getStringBounder()));
|
|
else
|
|
// ::done
|
|
registerDriver(UText.class, new DriverTextG2d(this, getStringBounder()));
|
|
|
|
registerDriver(ULine.class, new DriverLineG2d(dpiFactor));
|
|
registerDriver(UPixel.class, new DriverPixelG2d());
|
|
registerDriver(UPolygon.class, new DriverPolygonG2d(dpiFactor, this));
|
|
registerDriver(UEllipse.class, new DriverEllipseG2d(dpiFactor, this));
|
|
ignoreShape(UImageSvg.class);
|
|
registerDriver(UImage.class, new DriverImageG2d(dpiFactor, this));
|
|
registerDriver(DotPath.class, new DriverDotPathG2d(this));
|
|
registerDriver(UPath.class, new DriverPathG2d(dpiFactor));
|
|
registerDriver(UCenteredCharacter.class, new DriverCenteredCharacterG2d());
|
|
}
|
|
|
|
@Override
|
|
protected void beforeDraw() {
|
|
super.beforeDraw();
|
|
applyClip();
|
|
antiAliasing.apply(getGraphicObject());
|
|
}
|
|
|
|
private void applyClip() {
|
|
final UClip uclip = getClip();
|
|
if (uclip == null) {
|
|
getGraphicObject().setClip(null);
|
|
} else {
|
|
final Shape clip = new Rectangle2D.Double(uclip.getX(), uclip.getY(), uclip.getWidth(), uclip.getHeight());
|
|
getGraphicObject().setClip(clip);
|
|
}
|
|
}
|
|
|
|
protected final double getDpiFactor() {
|
|
return dpiFactor;
|
|
}
|
|
|
|
@Override
|
|
public void startUrl(Url url) {
|
|
Objects.requireNonNull(url);
|
|
// javascript: security issue
|
|
if (SecurityUtils.ignoreThisLink(url.getUrl())) {
|
|
urls.add(null);
|
|
} else {
|
|
urls.add(url);
|
|
allUrls.add(url);
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void closeUrl() {
|
|
urls.remove(urls.size() - 1);
|
|
}
|
|
|
|
public void ensureVisible(double x, double y) {
|
|
for (Url u : urls)
|
|
if (u != null && (getClip() == null || getClip().isInside(x, y)))
|
|
u.ensureVisible(x, y);
|
|
|
|
}
|
|
|
|
public BufferedImage getBufferedImage() {
|
|
return bufferedImage;
|
|
}
|
|
|
|
public void setBufferedImage(BufferedImage bufferedImage) {
|
|
this.bufferedImage = bufferedImage;
|
|
}
|
|
|
|
public Graphics2D getGraphics2D() {
|
|
return getGraphicObject();
|
|
}
|
|
|
|
@Override
|
|
public void writeToStream(OutputStream os, String metadata, int dpi) throws IOException {
|
|
final BufferedImage im = getBufferedImage();
|
|
PngIO.write(im, getColorMapper(), os, metadata, dpi);
|
|
}
|
|
|
|
@Override
|
|
public double dpiFactor() {
|
|
return dpiFactor;
|
|
}
|
|
|
|
}
|