X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Futils%2FImage.java;h=64f7b5a18f583ca2df5ebb055f3ac081f36ea9eb;hb=76b51de96af0b8dfa9615db37823fffd093fcfe3;hp=6edc1051ab7b910c0b47733407c99d804a08a8fe;hpb=82fcfcde9722e9e3b3eb713cf1f2bcb35fefaa7e;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/utils/Image.java b/src/be/nikiroo/utils/Image.java
index 6edc105..64f7b5a 100644
--- a/src/be/nikiroo/utils/Image.java
+++ b/src/be/nikiroo/utils/Image.java
@@ -3,16 +3,24 @@ package be.nikiroo.utils;
 import java.io.ByteArrayInputStream;
 import java.io.Closeable;
 import java.io.File;
-import java.io.FileInputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
+import java.io.OutputStream;
+import java.io.Serializable;
+
+import be.nikiroo.utils.streams.Base64InputStream;
+import be.nikiroo.utils.streams.MarkableFileInputStream;
 
 /**
  * This class represents an image data.
  * 
  * @author niki
  */
-public class Image implements Closeable {
+public class Image implements Closeable, Serializable {
+	static private final long serialVersionUID = 1L;
+
 	static private File tempRoot;
 	static private TempFiles tmpRepository;
 	static private long count = 0;
@@ -50,16 +58,21 @@ public class Image implements Closeable {
 	}
 
 	/**
-	 * Create a new {@link Image} from its Base64 representation.
+	 * Create an image from Base64 encoded data.
+	 * 
+	 * @deprecated Please use {@link Image#Image(InputStream)} instead, with a
+	 *             {@link Base64InputStream}
 	 * 
-	 * @param base64
-	 *            the {@link Image} in Base64 format
+	 * @param base64EncodedData
+	 *            the Base64 encoded data as a String
 	 * 
 	 * @throws IOException
-	 *             in case of I/O error
+	 *             in case of I/O error or badly formated Base64
 	 */
-	public Image(String base64) throws IOException {
-		this(Base64.decode(base64));
+	@Deprecated
+	public Image(String base64EncodedData) throws IOException {
+		this(new Base64InputStream(new ByteArrayInputStream(
+				StringUtils.getBytes(base64EncodedData)), false));
 	}
 
 	/**
@@ -76,14 +89,37 @@ public class Image implements Closeable {
 		IOUtils.write(in, data);
 	}
 
+	/**
+	 * Generate an {@link InputStream} that you can {@link InputStream#reset()}
+	 * for this {@link Image}.
+	 * <p>
+	 * This {@link InputStream} will (always) be a new one, and <b>you</b> are
+	 * responsible for it.
+	 * <p>
+	 * Note: take care that the {@link InputStream} <b>must not</b> live past
+	 * the {@link Image} life time!
+	 * 
+	 * @return the stream
+	 * 
+	 * @throws IOException
+	 *             in case of I/O error
+	 */
+	public InputStream newInputStream() throws IOException {
+		return new MarkableFileInputStream(data);
+	}
+
 	/**
 	 * <b>Read</b> the actual image data, as a byte array.
 	 * 
+	 * @deprecated if possible, prefer the {@link Image#newInputStream()}
+	 *             method, as it can be more efficient
+	 * 
 	 * @return the image data
 	 */
+	@Deprecated
 	public byte[] getData() {
 		try {
-			FileInputStream in = new FileInputStream(data);
+			InputStream in = newInputStream();
 			try {
 				return IOUtils.toByteArray(in);
 			} finally {
@@ -98,10 +134,24 @@ public class Image implements Closeable {
 	 * Convert the given {@link Image} object into a Base64 representation of
 	 * the same {@link Image} object.
 	 * 
+	 * @deprecated Please use {@link Image#newInputStream()} instead, it is more
+	 *             efficient
+	 * 
 	 * @return the Base64 representation
 	 */
+	@Deprecated
 	public String toBase64() {
-		return Base64.encodeBytes(getData());
+		try {
+			Base64InputStream stream = new Base64InputStream(newInputStream(),
+					true);
+			try {
+				return IOUtils.readSmallStream(stream);
+			} finally {
+				stream.close();
+			}
+		} catch (IOException e) {
+			return null;
+		}
 	}
 
 	/**
@@ -154,6 +204,43 @@ public class Image implements Closeable {
 		}
 	}
 
+	/**
+	 * Write this {@link Image} for serialization purposes; that is, write the
+	 * content of the backing temporary file.
+	 * 
+	 * @param out
+	 *            the {@link OutputStream} to write to
+	 * 
+	 * @throws IOException
+	 *             in case of I/O error
+	 */
+	private void writeObject(ObjectOutputStream out) throws IOException {
+		InputStream in = newInputStream();
+		try {
+			IOUtils.write(in, out);
+		} finally {
+			in.close();
+		}
+	}
+
+	/**
+	 * Read an {@link Image} written by
+	 * {@link Image#writeObject(java.io.ObjectOutputStream)}; that is, create a
+	 * new temporary file with the saved content.
+	 * 
+	 * @param in
+	 *            the {@link InputStream} to read from
+	 * @throws IOException
+	 *             in case of I/O error
+	 * @throws ClassNotFoundException
+	 *             will not be thrown by this method
+	 */
+	private void readObject(ObjectInputStream in) throws IOException,
+			ClassNotFoundException {
+		data = getTemporaryFile();
+		IOUtils.write(in, data);
+	}
+
 	/**
 	 * Change the temporary root directory used by the program.
 	 * <p>