X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FCbz.java;h=7113248ba485fb4b91ad06f5b614cd49f1b7a3c7;hb=a9eb3f46dd83115e7a549e96e800b932162b68ad;hp=f9eee08d51428cab98f2c3fbe9b6561cd3087f56;hpb=68686a37a591a767f6d1af428ea0d5f3d3a1ddc1;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/fanfix/supported/Cbz.java b/src/be/nikiroo/fanfix/supported/Cbz.java
index f9eee08..7113248 100644
--- a/src/be/nikiroo/fanfix/supported/Cbz.java
+++ b/src/be/nikiroo/fanfix/supported/Cbz.java
@@ -1,17 +1,23 @@
 package be.nikiroo.fanfix.supported;
 
+import java.awt.image.BufferedImage;
 import java.io.File;
 import java.io.IOException;
 import java.io.InputStream;
 import java.net.URL;
 import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipInputStream;
 
+import javax.imageio.ImageIO;
+
 import be.nikiroo.fanfix.Instance;
 import be.nikiroo.fanfix.data.Chapter;
 import be.nikiroo.fanfix.data.Paragraph;
 import be.nikiroo.fanfix.data.Story;
+import be.nikiroo.utils.Progress;
 
 /**
  * Support class for CBZ files (works better with CBZ created with this program,
@@ -52,7 +58,13 @@ class Cbz extends Epub {
 	}
 
 	@Override
-	public Story process(URL url) throws IOException {
+	public Story process(URL url, Progress pg) throws IOException {
+		if (pg == null) {
+			pg = new Progress();
+		} else {
+			pg.setMinMax(0, 100);
+		}
+
 		Story story = processMeta(url, false, true);
 		story.setChapters(new ArrayList<Chapter>());
 		Chapter chap = new Chapter(1, null);
@@ -60,6 +72,8 @@ class Cbz extends Epub {
 
 		ZipInputStream zipIn = new ZipInputStream(getInput());
 
+		pg.setProgress(10);
+		List<String> images = new ArrayList<String>();
 		for (ZipEntry entry = zipIn.getNextEntry(); entry != null; entry = zipIn
 				.getNextEntry()) {
 			if (!entry.isDirectory()
@@ -73,12 +87,10 @@ class Cbz extends Epub {
 				}
 
 				if (imageEntry) {
+					String uuid = meta.getUuid() + "_" + entry.getName();
 					try {
-						String uuid = meta.getUuid() + "_" + entry.getName();
-
-						Instance.getCache().addToCache(zipIn, uuid);
-						chap.getParagraphs().add(
-								new Paragraph(new File(uuid).toURI().toURL()));
+						File tmp = Instance.getCache().addToCache(zipIn, uuid);
+						images.add(tmp.toURI().toURL().toString());
 					} catch (Exception e) {
 						Instance.syserr(e);
 					}
@@ -86,6 +98,33 @@ class Cbz extends Epub {
 			}
 		}
 
+		pg.setProgress(80);
+
+		// ZIP order is not correct for us
+		Collections.sort(images);
+		pg.setProgress(90);
+
+		for (String uuid : images) {
+			try {
+				chap.getParagraphs().add(new Paragraph(new URL(uuid)));
+			} catch (Exception e) {
+				Instance.syserr(e);
+			}
+		}
+
+		if (meta.getCover() == null && !images.isEmpty()) {
+			InputStream in = Instance.getCache().open(new URL(images.get(0)),
+					this, true);
+			try {
+				BufferedImage fcover = ImageIO.read(in);
+				meta.setCover(fcover);
+				meta.setFakeCover(true);
+			} finally {
+				in.close();
+			}
+		}
+
+		pg.setProgress(100);
 		return story;
 	}
 }