1
0
mirror of https://github.com/octoleo/plantuml.git synced 2024-06-09 11:42:43 +00:00
plantuml/src/net/sourceforge/plantuml/ugraphic/g2d/DriverShadowedG2d.java
2020-06-14 22:35:42 +02:00

193 lines
6.7 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.ugraphic.g2d;
import java.awt.Color;
import java.awt.Graphics2D;
import java.awt.Shape;
import java.awt.geom.AffineTransform;
import java.awt.geom.Area;
import java.awt.geom.Line2D;
import java.awt.geom.Rectangle2D;
import java.awt.image.BufferedImage;
import java.awt.image.ConvolveOp;
import java.awt.image.Kernel;
import net.sourceforge.plantuml.Log;
public class DriverShadowedG2d {
private ConvolveOp getConvolveOp(int blurRadius, double dpiFactor) {
blurRadius = (int) (blurRadius * dpiFactor);
final int blurRadius2 = blurRadius * blurRadius;
final float blurRadius2F = blurRadius2;
// final float weight = (float) (1.0 / blurRadius2F / dpiFactor);
final float weight = (float) (1.0 / blurRadius2F);
final float[] elements = new float[blurRadius2];
for (int k = 0; k < blurRadius2; k++) {
elements[k] = weight;
}
final Kernel myKernel = new Kernel(blurRadius, blurRadius, elements);
// if EDGE_NO_OP is not selected, EDGE_ZERO_FILL is the default which
// creates a black border
return new ConvolveOp(myKernel, ConvolveOp.EDGE_NO_OP, null);
}
private final Color color = new Color(170, 170, 170);
private final Color colorLine = new Color(30, 30, 30);
protected void drawShadow(Graphics2D g2d, Shape shape, double deltaShadow, double dpiFactor) {
if (dpiFactor < 1) {
dpiFactor = 1;
}
final Rectangle2D bounds = shape.getBounds2D();
final double ww = bounds.getMaxX() - bounds.getMinX();
final double hh = bounds.getMaxY() - bounds.getMinY();
final double w = (ww + deltaShadow * 2 + 6) * dpiFactor;
final double h = (hh + deltaShadow * 2 + 6) * dpiFactor;
BufferedImage destination = null;
try {
destination = new BufferedImage((int) w, (int) h, BufferedImage.TYPE_INT_ARGB);
final Graphics2D gg = destination.createGraphics();
gg.scale(dpiFactor, dpiFactor);
gg.translate(deltaShadow - bounds.getMinX(), deltaShadow - bounds.getMinY());
final boolean isLine = shape instanceof Line2D.Double;
if (isLine) {
gg.setColor(colorLine);
gg.draw(shape);
} else {
gg.setColor(color);
gg.fill(shape);
}
gg.dispose();
final ConvolveOp simpleBlur = getConvolveOp(6, dpiFactor);
destination = simpleBlur.filter(destination, null);
} catch (OutOfMemoryError error) {
Log.info("Warning: Cannot draw shadow, image too big.");
} catch (Exception e) {
Log.info("Warning: Cannot draw shadow: " + e);
}
if (destination != null) {
final AffineTransform at = g2d.getTransform();
g2d.scale(1 / dpiFactor, 1 / dpiFactor);
g2d.drawImage(destination, (int) (bounds.getMinX() * dpiFactor), (int) (bounds.getMinY() * dpiFactor),
null);
g2d.setTransform(at);
}
}
protected void drawOnlyLineShadow(Graphics2D g2d, Shape shape, double deltaShadow, double dpiFactor) {
if (dpiFactor < 1) {
dpiFactor = 1;
}
final Rectangle2D bounds = shape.getBounds2D();
final double ww = bounds.getMaxX() - bounds.getMinX();
final double hh = bounds.getMaxY() - bounds.getMinY();
final double w = (ww + deltaShadow * 2 + 6) * dpiFactor;
final double h = (hh + deltaShadow * 2 + 6) * dpiFactor;
BufferedImage destination = null;
try {
destination = new BufferedImage((int) w, (int) h, BufferedImage.TYPE_INT_ARGB);
final Graphics2D gg = destination.createGraphics();
gg.scale(dpiFactor, dpiFactor);
gg.translate(deltaShadow - bounds.getMinX(), deltaShadow - bounds.getMinY());
gg.draw(shape);
gg.dispose();
final ConvolveOp simpleBlur = getConvolveOp(6, dpiFactor);
destination = simpleBlur.filter(destination, null);
} catch (OutOfMemoryError error) {
Log.info("Warning: Cannot draw shadow, image too big.");
} catch (Exception e) {
Log.info("Warning: Cannot draw shadow: " + e);
}
if (destination != null) {
final AffineTransform at = g2d.getTransform();
g2d.scale(1 / dpiFactor, 1 / dpiFactor);
g2d.drawImage(destination, (int) (bounds.getMinX() * dpiFactor), (int) (bounds.getMinY() * dpiFactor),
null);
g2d.setTransform(at);
}
}
protected void drawOnlyLineShadowSpecial(Graphics2D g2d, Shape shape, double deltaShadow, double dpiFactor) {
if (dpiFactor < 1) {
dpiFactor = 1;
}
final Rectangle2D bounds = shape.getBounds2D();
final double ww = bounds.getMaxX() - bounds.getMinX();
final double hh = bounds.getMaxY() - bounds.getMinY();
final double w = (ww + deltaShadow * 2 + 6) * dpiFactor;
final double h = (hh + deltaShadow * 2 + 6) * dpiFactor;
BufferedImage destination = null;
try {
destination = new BufferedImage((int) w, (int) h, BufferedImage.TYPE_INT_ARGB);
final Graphics2D gg = destination.createGraphics();
gg.scale(dpiFactor, dpiFactor);
gg.translate(deltaShadow - bounds.getMinX(), deltaShadow - bounds.getMinY());
gg.draw(shape);
gg.dispose();
final ConvolveOp simpleBlur = getConvolveOp(6, dpiFactor);
destination = simpleBlur.filter(destination, null);
} catch (OutOfMemoryError error) {
Log.info("Warning: Cannot draw shadow, image too big.");
} catch (Exception e) {
Log.info("Warning: Cannot draw shadow: " + e);
}
if (destination != null) {
final AffineTransform at = g2d.getTransform();
g2d.scale(1 / dpiFactor, 1 / dpiFactor);
final Shape sav = g2d.getClip();
Area full = new Area(new Rectangle2D.Double(0, 0, bounds.getMaxX() + deltaShadow * 2 + 6,
bounds.getMaxY() + deltaShadow * 2 + 6));
full.subtract(new Area(shape));
g2d.setClip(full);
g2d.drawImage(destination, (int) (bounds.getMinX() * dpiFactor), (int) (bounds.getMinY() * dpiFactor),
null);
g2d.setClip(sav);
g2d.setTransform(at);
}
}
}