X-Git-Url: http://git.nikiroo.be/?a=blobdiff_plain;f=src%2Fbe%2Fnikiroo%2Ffanfix%2Fsupported%2FCbz.java;h=8ab2a5202044ec850bcb29b979bdec0f0483fd17;hb=0ec7855764b78c0956c552566fe52587c2b43f4e;hp=1080ad2755adebb174ad7a3e04f4053c9e0b35f4;hpb=d3c84ac3317780016a5a6fff263ac8ff46220db9;p=nikiroo-utils.git

diff --git a/src/be/nikiroo/fanfix/supported/Cbz.java b/src/be/nikiroo/fanfix/supported/Cbz.java
index 1080ad2..8ab2a52 100644
--- a/src/be/nikiroo/fanfix/supported/Cbz.java
+++ b/src/be/nikiroo/fanfix/supported/Cbz.java
@@ -1,12 +1,12 @@
 package be.nikiroo.fanfix.supported;
 
-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.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.zip.ZipEntry;
 import java.util.zip.ZipInputStream;
 
@@ -14,6 +14,8 @@ 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.Image;
+import be.nikiroo.utils.Progress;
 
 /**
  * Support class for CBZ files (works better with CBZ created with this program,
@@ -43,26 +45,35 @@ class Cbz extends Epub {
 	}
 
 	@Override
-	protected boolean getCover() {
-		return false;
+	protected boolean isImagesDocumentByDefault() {
+		return true;
 	}
 
 	@Override
-	protected void preprocess(URL source, InputStream in) throws IOException {
-		super.preprocess(source, in);
-		meta.setImageDocument(true);
+	protected boolean getCover() {
+		return false;
 	}
 
 	@Override
-	public Story process(URL url) throws IOException {
-		Story story = processMeta(url, false, true);
+	public Story process(URL url, Progress pg) throws IOException {
+		if (pg == null) {
+			pg = new Progress();
+		} else {
+			pg.setMinMax(0, 100);
+		}
+
+		Progress pgMeta = new Progress();
+		pg.addProgress(pgMeta, 10);
+		Story story = processMeta(url, false, true, pgMeta);
+		pgMeta.done(); // 10%
+
 		story.setChapters(new ArrayList<Chapter>());
 		Chapter chap = new Chapter(1, null);
 		story.getChapters().add(chap);
 
 		ZipInputStream zipIn = new ZipInputStream(getInput());
 
-		List<String> images = new ArrayList<String>();
+		Map<String, Image> images = new HashMap<String, Image>();
 		for (ZipEntry entry = zipIn.getNextEntry(); entry != null; entry = zipIn
 				.getNextEntry()) {
 			if (!entry.isDirectory()
@@ -77,28 +88,41 @@ class Cbz extends Epub {
 
 				if (imageEntry) {
 					String uuid = meta.getUuid() + "_" + entry.getName();
-					images.add(uuid);
 					try {
-						Instance.getCache().addToCache(zipIn, uuid);
+						images.put(uuid, new Image(zipIn));
 					} catch (Exception e) {
-						Instance.syserr(e);
+						Instance.getTraceHandler().error(e);
+					}
+
+					if (pg.getProgress() < 85) {
+						pg.add(1);
 					}
 				}
 			}
 		}
 
-		// ZIP order is not sure
-		Collections.sort(images);
+		pg.setProgress(85);
+
+		// ZIP order is not correct for us
+		List<String> imagesList = new ArrayList<String>(images.keySet());
+		Collections.sort(imagesList);
 
-		for (String uuid : images) {
+		pg.setProgress(90);
+
+		for (String uuid : imagesList) {
 			try {
-				chap.getParagraphs().add(
-						new Paragraph(new File(uuid).toURI().toURL()));
+				chap.getParagraphs().add(new Paragraph(images.get(uuid)));
 			} catch (Exception e) {
-				Instance.syserr(e);
+				Instance.getTraceHandler().error(e);
 			}
 		}
 
+		if (meta.getCover() == null && !images.isEmpty()) {
+			meta.setCover(images.get(imagesList.get(0)));
+			meta.setFakeCover(true);
+		}
+
+		pg.setProgress(100);
 		return story;
 	}
 }