X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2FDataLoader.java;h=25300130116214d8202bc88e1acc83de98addf3d;hb=48587cad70aa1fee408d5336d1fdd15065a402d2;hp=0f3cad00de0b7a19d11e9ffd95b6421202f7b64a;hpb=c8faa52a7993d29944e505b517619de44ac58279;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/fanfix/DataLoader.java b/src/be/nikiroo/fanfix/DataLoader.java
index 0f3cad0..2530013 100644
--- a/src/be/nikiroo/fanfix/DataLoader.java
+++ b/src/be/nikiroo/fanfix/DataLoader.java
@@ -1,5 +1,6 @@
 package be.nikiroo.fanfix;
 
+import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
@@ -99,8 +100,8 @@ public class DataLoader {
 		// MUST NOT return null
 		try {
 			InputStream in = cache.load(originalUrl, false, stable);
-			Instance.trace("Cache " + (in != null ? "hit" : "miss") + ": "
-					+ url);
+			Instance.getTraceHandler().trace(
+					"Cache " + (in != null ? "hit" : "miss") + ": " + url);
 
 			if (in == null) {
 				try {
@@ -228,16 +229,48 @@ public class DataLoader {
 	public void saveAsImage(URL url, File target) throws IOException {
 		InputStream in = open(url, null, true);
 		try {
-			ImageIO.write(ImageUtils.fromStream(in), Instance.getConfig()
-					.getString(Config.IMAGE_FORMAT_CONTENT).toLowerCase(),
-					target);
-		} catch (IOException e) {
-			throw new IOException("Cannot write image " + url, e);
+			saveAsImage(ImageUtils.fromStream(in), target);
 		} finally {
 			in.close();
 		}
 	}
 
+	/**
+	 * Save the given resource as an image on disk using the default image
+	 * format for content.
+	 * 
+	 * @param image
+	 *            the resource
+	 * @param target
+	 *            the target file
+	 * 
+	 * @throws IOException
+	 *             in case of I/O error
+	 */
+	public void saveAsImage(BufferedImage image, File target)
+			throws IOException {
+		try {
+			String format = Instance.getConfig()
+					.getString(Config.IMAGE_FORMAT_CONTENT).toLowerCase();
+			boolean ok = ImageIO.write(image, format, target);
+			if (!ok) {
+				// Some formats are not reliable
+				// Second change: PNG
+				if (!format.equals("png")) {
+					ok = ImageIO.write(image, "png", target);
+				}
+
+				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);
+		}
+	}
+
 	/**
 	 * Manually add this item to the cache.
 	 * 
@@ -268,6 +301,18 @@ public class DataLoader {
 		return cache.load(uniqueID, true, true);
 	}
 
+	/**
+	 * Remove the given resource from the cache.
+	 * 
+	 * @param uniqueID
+	 *            a unique ID used to locate the cached resource
+	 * 
+	 * @return TRUE if it was removed
+	 */
+	public boolean removeFromCache(String uniqueID) {
+		return cache.remove(uniqueID);
+	}
+
 	/**
 	 * Clean the cache (delete the cached items).
 	 *