X-Git-Url: http://git.nikiroo.be/?p=fanfix.git;a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2Fui%2FImageUtilsAwt.java;fp=src%2Fbe%2Fnikiroo%2Futils%2Fui%2FImageUtilsAwt.java;h=4cf12c04cd0573629d90ed6e376be7c5ca6e7155;hp=0000000000000000000000000000000000000000;hb=d46b7b96f94e88a776bcd2dfd756549ffb300cc9;hpb=c9994f27667bc421bcd448d39e55774fddf5c431

diff --git a/src/be/nikiroo/utils/ui/ImageUtilsAwt.java b/src/be/nikiroo/utils/ui/ImageUtilsAwt.java
new file mode 100644
index 0000000..4cf12c0
--- /dev/null
+++ b/src/be/nikiroo/utils/ui/ImageUtilsAwt.java
@@ -0,0 +1,180 @@
+package be.nikiroo.utils.ui;
+
+import java.awt.geom.AffineTransform;
+import java.awt.image.AffineTransformOp;
+import java.awt.image.BufferedImage;
+import java.io.File;
+import java.io.IOException;
+import java.io.InputStream;
+
+import javax.imageio.ImageIO;
+
+import be.nikiroo.utils.Image;
+import be.nikiroo.utils.ImageUtils;
+import be.nikiroo.utils.StringUtils;
+
+/**
+ * This class offer some utilities based around images and uses java.awt.
+ * 
+ * @author niki
+ */
+public class ImageUtilsAwt extends ImageUtils {
+	@Override
+	protected boolean check() {
+		// Will not work if ImageIO is not available
+		ImageIO.getCacheDirectory();
+		return true;
+	}
+
+	@Override
+	public void saveAsImage(Image img, File target, String format)
+			throws IOException {
+		try {
+			BufferedImage image = fromImage(img);
+
+			boolean ok = false;
+			try {
+
+				ok = ImageIO.write(image, format, target);
+			} catch (IOException e) {
+				ok = false;
+			}
+
+			// Some formats are not reliable
+			// Second chance: PNG
+			if (!ok && !format.equals("png")) {
+				try {
+					ok = ImageIO.write(image, "png", target);
+				} catch (IllegalArgumentException e) {
+					throw e;
+				} catch (Exception e) {
+					throw new IOException("Undocumented exception occured, "
+							+ "converting to IOException", e);
+				}
+			}
+
+			if (!ok) {
+				throw new IOException(
+						"Cannot find a writer for this image and format: "
+								+ format);
+			}
+		} catch (IOException e) {
+			throw new IOException("Cannot write image to " + target, e);
+		}
+	}
+
+	/**
+	 * Convert the given {@link Image} into a {@link BufferedImage} object,
+	 * respecting the EXIF transformations if any.
+	 * 
+	 * @param img
+	 *            the {@link Image}
+	 * 
+	 * @return the {@link Image} object
+	 * 
+	 * @throws IOException
+	 *             in case of IO error
+	 */
+	public static BufferedImage fromImage(Image img) throws IOException {
+		InputStream in = img.newInputStream();
+		BufferedImage image;
+		try {
+			int orientation;
+			try {
+				orientation = getExifTransorm(in);
+			} catch (Exception e) {
+				// no EXIF transform, ok
+				orientation = -1;
+			}
+
+			in.reset();
+
+			try {
+				image = ImageIO.read(in);
+			} catch (IllegalArgumentException e) {
+				throw e;
+			} catch (Exception e) {
+				throw new IOException("Undocumented exception occured, "
+						+ "converting to IOException", e);
+			}
+
+			if (image == null) {
+				String extra = "";
+				if (img.getSize() <= 2048) {
+					try {
+						extra = ", content: "
+								+ new String(img.getData(), "UTF-8");
+					} catch (Exception e) {
+						extra = ", content unavailable";
+					}
+				}
+				String ssize = StringUtils.formatNumber(img.getSize());
+				throw new IOException(
+						"Failed to convert input to image, size was: " + ssize
+								+ extra);
+			}
+
+			// Note: this code has been found on Internet;
+			// thank you anonymous coder.
+			int width = image.getWidth();
+			int height = image.getHeight();
+			AffineTransform affineTransform = new AffineTransform();
+
+			switch (orientation) {
+			case 1:
+				affineTransform = null;
+				break;
+			case 2: // Flip X
+				affineTransform.scale(-1.0, 1.0);
+				affineTransform.translate(-width, 0);
+				break;
+			case 3: // PI rotation
+				affineTransform.translate(width, height);
+				affineTransform.rotate(Math.PI);
+				break;
+			case 4: // Flip Y
+				affineTransform.scale(1.0, -1.0);
+				affineTransform.translate(0, -height);
+				break;
+			case 5: // - PI/2 and Flip X
+				affineTransform.rotate(-Math.PI / 2);
+				affineTransform.scale(-1.0, 1.0);
+				break;
+			case 6: // -PI/2 and -width
+				affineTransform.translate(height, 0);
+				affineTransform.rotate(Math.PI / 2);
+				break;
+			case 7: // PI/2 and Flip
+				affineTransform.scale(-1.0, 1.0);
+				affineTransform.translate(-height, 0);
+				affineTransform.translate(0, width);
+				affineTransform.rotate(3 * Math.PI / 2);
+				break;
+			case 8: // PI / 2
+				affineTransform.translate(0, width);
+				affineTransform.rotate(3 * Math.PI / 2);
+				break;
+			default:
+				affineTransform = null;
+				break;
+			}
+
+			if (affineTransform != null) {
+				AffineTransformOp affineTransformOp = new AffineTransformOp(
+						affineTransform, AffineTransformOp.TYPE_BILINEAR);
+
+				BufferedImage transformedImage = new BufferedImage(width,
+						height, image.getType());
+				transformedImage = affineTransformOp.filter(image,
+						transformedImage);
+
+				image = transformedImage;
+			}
+			//
+		} finally {
+			in.close();
+		}
+
+		return image;
+	}
+}