/*******************************************************************************
* Copyright (c) 2001-2005 Sasa Markovic and Ciaran Treanor.
* Copyright (c) 2011 The OpenNMS Group, Inc.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser 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
*******************************************************************************/
package org.jrobin.graph;
import java.awt.Font;
import java.awt.Graphics2D;
import java.awt.Paint;
import java.awt.Rectangle;
import java.awt.RenderingHints;
import java.awt.Stroke;
import java.awt.TexturePaint;
import java.awt.font.LineMetrics;
import java.awt.geom.AffineTransform;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.RandomAccessFile;
import javax.imageio.ImageIO;
import com.sun.image.codec.jpeg.JPEGCodec;
import com.sun.image.codec.jpeg.JPEGEncodeParam;
import com.sun.image.codec.jpeg.JPEGImageEncoder;
@SuppressWarnings("restriction")
class ImageWorker {
private static final String DUMMY_TEXT = "Dummy";
private BufferedImage img;
private Graphics2D gd;
private int imgWidth, imgHeight;
private AffineTransform aftInitial;
ImageWorker(int width, int height) {
resize(width, height);
}
void resize(int width, int height) {
if (gd != null) {
gd.dispose();
}
this.imgWidth = width;
this.imgHeight = height;
this.img = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
this.gd = img.createGraphics();
this.aftInitial = gd.getTransform();
this.setAntiAliasing(false);
}
void clip(int x, int y, int width, int height) {
gd.setClip(x, y, width, height);
}
void transform(int x, int y, double angle) {
gd.translate(x, y);
gd.rotate(angle);
}
void reset() {
gd.setTransform(aftInitial);
gd.setClip(0, 0, imgWidth, imgHeight);
}
void fillRect(int x, int y, int width, int height, Paint paint) {
gd.setPaint(paint);
gd.fillRect(x, y, width, height);
}
void fillPolygon(int[] x, int[] y, Paint paint) {
gd.setPaint(paint);
gd.fillPolygon(x, y, x.length);
}
void fillPolygon(double[] x, double yBottom, double[] yTop, Paint paint) {
gd.setPaint(paint);
PathIterator path = new PathIterator(yTop);
for (int[] pos = path.getNextPath(); pos != null; pos = path.getNextPath()) {
int start = pos[0], end = pos[1], n = end - start;
int[] xDev = new int[n + 2], yDev = new int[n + 2];
for (int i = start; i < end; i++) {
xDev[i - start] = (int) x[i];
yDev[i - start] = (int) yTop[i];
}
xDev[n] = xDev[n - 1];
xDev[n + 1] = xDev[0];
yDev[n] = yDev[n + 1] = (int) yBottom;
gd.fillPolygon(xDev, yDev, xDev.length);
gd.drawPolygon(xDev, yDev, xDev.length);
}
}
void fillPolygon(double[] x, double[] yBottom, double[] yTop, Paint paint) {
gd.setPaint(paint);
PathIterator path = new PathIterator(yTop);
for (int[] pos = path.getNextPath(); pos != null; pos = path.getNextPath()) {
int start = pos[0], end = pos[1], n = end - start;
int[] xDev = new int[n * 2], yDev = new int[n * 2];
for (int i = start; i < end; i++) {
int ix1 = i - start, ix2 = n * 2 - 1 - i + start;
xDev[ix1] = xDev[ix2] = (int) x[i];
yDev[ix1] = (int) yTop[i];
yDev[ix2] = (int) yBottom[i];
}
gd.fillPolygon(xDev, yDev, xDev.length);
gd.drawPolygon(xDev, yDev, xDev.length);
}
}
void drawLine(int x1, int y1, int x2, int y2, Paint paint, Stroke stroke) {
gd.setStroke(stroke);
gd.setPaint(paint);
gd.drawLine(x1, y1, x2, y2);
}
void drawPolyline(int[] x, int[] y, Paint paint, Stroke stroke) {
gd.setStroke(stroke);
gd.setPaint(paint);
gd.drawPolyline(x, y, x.length);
}
void drawPolyline(double[] x, double[] y, Paint paint, Stroke stroke) {
gd.setPaint(paint);
gd.setStroke(stroke);
PathIterator path = new PathIterator(y);
for (int[] pos = path.getNextPath(); pos != null; pos = path.getNextPath()) {
int start = pos[0], end = pos[1];
int[] xDev = new int[end - start], yDev = new int[end - start];
for (int i = start; i < end; i++) {
xDev[i - start] = (int) x[i];
yDev[i - start] = (int) y[i];
}
gd.drawPolyline(xDev, yDev, xDev.length);
}
}
void drawString(String text, int x, int y, Font font, Paint paint) {
gd.setFont(font);
gd.setPaint(paint);
gd.drawString(text, x, y);
}
double getFontAscent(Font font) {
LineMetrics lm = font.getLineMetrics(DUMMY_TEXT, gd.getFontRenderContext());
return lm.getAscent();
}
double getFontHeight(Font font) {
LineMetrics lm = font.getLineMetrics(DUMMY_TEXT, gd.getFontRenderContext());
return lm.getAscent() + lm.getDescent();
}
double getStringWidth(String text, Font font) {
return font.getStringBounds(text, 0, text.length(), gd.getFontRenderContext()).getBounds().getWidth();
}
void setAntiAliasing(boolean enable) {
gd.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
enable ? RenderingHints.VALUE_ANTIALIAS_ON : RenderingHints.VALUE_ANTIALIAS_OFF);
gd.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING,RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
gd.setRenderingHint(RenderingHints.KEY_RENDERING,RenderingHints.VALUE_RENDER_QUALITY);
}
void dispose() {
gd.dispose();
}
void saveImage(OutputStream stream, String type, float quality) throws IOException {
if (type.equalsIgnoreCase("png")) {
ImageIO.write(img, "png", stream);
}
else if (type.equalsIgnoreCase("gif")) {
GifEncoder gifEncoder = new GifEncoder(img);
gifEncoder.encode(stream);
}
else if (type.equalsIgnoreCase("jpg") || type.equalsIgnoreCase("jpeg")) {
JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(stream);
JPEGEncodeParam param = encoder.getDefaultJPEGEncodeParam(img);
param.setQuality(quality, false);
encoder.setJPEGEncodeParam(param);
encoder.encode(img);
}
else {
throw new IOException("Unsupported image format: " + type);
}
stream.flush();
}
byte[] saveImage(String path, String type, float quality) throws IOException {
byte[] bytes = getImageBytes(type, quality);
RandomAccessFile f = new RandomAccessFile(path, "rw");
try {
f.write(bytes);
return bytes;
}
finally {
f.close();
}
}
byte[] getImageBytes(String type, float quality) throws IOException {
ByteArrayOutputStream stream = new ByteArrayOutputStream();
try {
saveImage(stream, type, quality);
return stream.toByteArray();
}
finally {
stream.close();
}
}
public void loadImage(String imageFile) throws IOException {
BufferedImage wpImage = ImageIO.read(new File(imageFile));
TexturePaint paint = new TexturePaint(wpImage, new Rectangle(0, 0, wpImage.getWidth(), wpImage.getHeight()));
gd.setPaint(paint);
gd.fillRect(0, 0, wpImage.getWidth(), wpImage.getHeight());
}
}