Perf improvement: covers
[fanfix.git] / src / be / nikiroo / fanfix / Library.java
index eb9c9a39f3d4ae76298fd636c0e3e6ad43a962cc..7819eff65d15e665a97b124ac412260e79821927 100644 (file)
@@ -1,9 +1,12 @@
 package be.nikiroo.fanfix;
 
+import java.awt.image.BufferedImage;
 import java.io.File;
+import java.io.FileFilter;
 import java.io.IOException;
 import java.net.URL;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -25,6 +28,9 @@ import be.nikiroo.utils.Progress;
  * <p>
  * Each {@link Story} object will be associated with a (local to the library)
  * unique ID, the LUID, which will be used to identify the {@link Story}.
+ * <p>
+ * Most of the {@link Library} functions work on either the LUID or a partial
+ * (cover not included) {@link MetaData} object.
  * 
  * @author niki
  */
@@ -55,6 +61,16 @@ public class Library {
                dir.mkdirs();
        }
 
+       /**
+        * Refresh the {@link Library}, that is, make sure all stories are loaded.
+        * 
+        * @param pg
+        *            the optional progress reporter
+        */
+       public void refresh(Progress pg) {
+               getStories(pg);
+       }
+
        /**
         * List all the known types of stories.
         * 
@@ -62,39 +78,86 @@ public class Library {
         */
        public synchronized List<String> getTypes() {
                List<String> list = new ArrayList<String>();
-               for (Entry<MetaData, File> entry : getStories().entrySet()) {
-                       String storyType = entry.getValue().getParentFile().getName();
+               for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
+                       String storyType = entry.getKey().getSource();
                        if (!list.contains(storyType)) {
                                list.add(storyType);
                        }
                }
 
+               Collections.sort(list);
+               return list;
+       }
+
+       /**
+        * List all the known authors of stories.
+        * 
+        * @return the authors
+        */
+       public synchronized List<String> getAuthors() {
+               List<String> list = new ArrayList<String>();
+               for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
+                       String storyAuthor = entry.getKey().getAuthor();
+                       if (!list.contains(storyAuthor)) {
+                               list.add(storyAuthor);
+                       }
+               }
+
+               Collections.sort(list);
+               return list;
+       }
+
+       /**
+        * List all the stories of the given author in the {@link Library}, or all
+        * the stories if NULL is passed as an author.
+        * <p>
+        * Cover images not included.
+        * 
+        * @param author
+        *            the author of the stories to retrieve, or NULL for all
+        * 
+        * @return the stories
+        */
+       public synchronized List<MetaData> getListByAuthor(String author) {
+               List<MetaData> list = new ArrayList<MetaData>();
+               for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
+                       String storyAuthor = entry.getKey().getAuthor();
+                       if (author == null || author.equalsIgnoreCase(storyAuthor)) {
+                               list.add(entry.getKey());
+                       }
+               }
+
+               Collections.sort(list);
                return list;
        }
 
        /**
         * List all the stories of the given source type in the {@link Library}, or
         * all the stories if NULL is passed as a type.
+        * <p>
+        * Cover images not included.
         * 
         * @param type
         *            the type of story to retrieve, or NULL for all
         * 
         * @return the stories
         */
-       public synchronized List<MetaData> getList(String type) {
+       public synchronized List<MetaData> getListByType(String type) {
                List<MetaData> list = new ArrayList<MetaData>();
-               for (Entry<MetaData, File> entry : getStories().entrySet()) {
+               for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
                        String storyType = entry.getValue().getParentFile().getName();
                        if (type == null || type.equalsIgnoreCase(storyType)) {
                                list.add(entry.getKey());
                        }
                }
 
+               Collections.sort(list);
                return list;
        }
 
        /**
-        * Retrieve a {@link File} corresponding to the given {@link Story}.
+        * Retrieve a {@link File} corresponding to the given {@link Story}, cover
+        * image not included.
         * 
         * @param luid
         *            the Library UID of the story
@@ -103,7 +166,7 @@ public class Library {
         */
        public synchronized MetaData getInfo(String luid) {
                if (luid != null) {
-                       for (Entry<MetaData, File> entry : getStories().entrySet()) {
+                       for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
                                if (luid.equals(entry.getKey().getLuid())) {
                                        return entry.getKey();
                                }
@@ -123,7 +186,7 @@ public class Library {
         */
        public synchronized File getFile(String luid) {
                if (luid != null) {
-                       for (Entry<MetaData, File> entry : getStories().entrySet()) {
+                       for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
                                if (luid.equals(entry.getKey().getLuid())) {
                                        return entry.getValue();
                                }
@@ -133,6 +196,29 @@ public class Library {
                return null;
        }
 
+       /**
+        * Return the cover image associated to this story.
+        * 
+        * @param luid
+        *            the Library UID of the story
+        * 
+        * @return the cover image
+        */
+       public synchronized BufferedImage getCover(String luid) {
+               MetaData meta = getInfo(luid);
+               if (meta != null) {
+                       try {
+                               File infoFile = new File(getFile(meta).getPath() + ".info");
+                               meta = readMeta(infoFile, true).getKey();
+                               return meta.getCover();
+                       } catch (IOException e) {
+                               Instance.syserr(e);
+                       }
+               }
+
+               return null;
+       }
+
        /**
         * Retrieve a specific {@link Story}.
         * 
@@ -145,7 +231,7 @@ public class Library {
         */
        public synchronized Story getStory(String luid, Progress pg) {
                if (luid != null) {
-                       for (Entry<MetaData, File> entry : getStories().entrySet()) {
+                       for (Entry<MetaData, File> entry : getStories(null).entrySet()) {
                                if (luid.equals(entry.getKey().getLuid())) {
                                        try {
                                                SupportType type = SupportType.valueOfAllOkUC(entry
@@ -280,7 +366,7 @@ public class Library {
                story.setMeta(key);
 
                if (luid == null || luid.isEmpty()) {
-                       getStories(); // refresh lastId if needed
+                       getStories(null); // refresh lastId if needed
                        key.setLuid(String.format("%03d", (++lastId)));
                } else {
                        key.setLuid(luid);
@@ -319,7 +405,7 @@ public class Library {
                boolean ok = false;
 
                MetaData meta = getInfo(luid);
-               File file = getStories().get(meta);
+               File file = getStories(null).get(meta);
 
                if (file != null) {
                        if (file.delete()) {
@@ -388,68 +474,120 @@ public class Library {
         * @return the target
         */
        private File getFile(MetaData key) {
-               String title = key.getTitle().replaceAll("[^a-zA-Z0-9._+-]", "_");
+               String title = key.getTitle();
+               if (title == null) {
+                       title = "";
+               }
+               title = title.replaceAll("[^a-zA-Z0-9._+-]", "_");
                return new File(getDir(key), key.getLuid() + "_" + title);
        }
 
        /**
         * Return all the known stories in this {@link Library} object.
         * 
+        * @param pg
+        *            the optional progress reporter
+        * 
         * @return the stories
         */
-       private synchronized Map<MetaData, File> getStories() {
+       private synchronized Map<MetaData, File> getStories(Progress pg) {
+               if (pg == null) {
+                       pg = new Progress();
+               } else {
+                       pg.setMinMax(0, 100);
+               }
+
                if (stories.isEmpty()) {
                        lastId = 0;
 
-                       String ext = ".info";
-                       for (File dir : baseDir.listFiles()) {
-                               if (dir.isDirectory()) {
-                                       for (File file : dir.listFiles()) {
+                       File[] dirs = baseDir.listFiles(new FileFilter() {
+                               public boolean accept(File file) {
+                                       return file != null && file.isDirectory();
+                               }
+                       });
+
+                       Progress pgDirs = new Progress(0, 100 * dirs.length);
+                       pg.addProgress(pgDirs, 100);
+
+                       for (File dir : dirs) {
+                               File[] files = dir.listFiles(new FileFilter() {
+                                       public boolean accept(File file) {
+                                               return file != null
+                                                               && file.getPath().toLowerCase()
+                                                                               .endsWith(".info");
+                                       }
+                               });
+
+                               Progress pgFiles = new Progress(0, files.length);
+                               pgDirs.addProgress(pgFiles, 100);
+                               pgDirs.setName("Loading from: " + dir.getName());
+
+                               for (File file : files) {
+                                       pgFiles.setName(file.getName());
+                                       try {
+                                               Entry<MetaData, File> entry = readMeta(file, false);
                                                try {
-                                                       if (file.getPath().toLowerCase().endsWith(ext)) {
-                                                               MetaData meta = InfoReader.readMeta(file);
-                                                               try {
-                                                                       int id = Integer.parseInt(meta.getLuid());
-                                                                       if (id > lastId) {
-                                                                               lastId = id;
-                                                                       }
-
-                                                                       // Replace .info with whatever is needed:
-                                                                       String path = file.getPath();
-                                                                       path = path.substring(0, path.length()
-                                                                                       - ext.length());
-
-                                                                       String newExt = getOutputType(meta)
-                                                                                       .getDefaultExtension(true);
-
-                                                                       file = new File(path + newExt);
-                                                                       //
-
-                                                                       stories.put(meta, file);
-
-                                                               } catch (Exception e) {
-                                                                       // not normal!!
-                                                                       Instance.syserr(new IOException(
-                                                                                       "Cannot understand the LUID of "
-                                                                                                       + file.getPath() + ": "
-                                                                                                       + meta.getLuid(), e));
-                                                               }
+                                                       int id = Integer.parseInt(entry.getKey().getLuid());
+                                                       if (id > lastId) {
+                                                               lastId = id;
                                                        }
-                                               } catch (IOException e) {
-                                                       // We should not have not-supported files in the
-                                                       // library
-                                                       Instance.syserr(new IOException(
-                                                                       "Cannot load file from library: "
-                                                                                       + file.getPath(), e));
+
+                                                       stories.put(entry.getKey(), entry.getValue());
+                                               } catch (Exception e) {
+                                                       // not normal!!
+                                                       throw new IOException(
+                                                                       "Cannot understand the LUID of "
+                                                                                       + file.getPath() + ": "
+                                                                                       + entry.getKey().getLuid(), e);
                                                }
+                                       } catch (IOException e) {
+                                               // We should not have not-supported files in the
+                                               // library
+                                               Instance.syserr(new IOException(
+                                                               "Cannot load file from library: "
+                                                                               + file.getPath(), e));
                                        }
+                                       pgFiles.add(1);
                                }
+
+                               pgFiles.setName(null);
                        }
+
+                       pgDirs.setName("Loading directories");
                }
 
                return stories;
        }
 
+       private Entry<MetaData, File> readMeta(File infoFile, boolean withCover)
+                       throws IOException {
+
+               final MetaData meta = InfoReader.readMeta(infoFile, withCover);
+
+               // Replace .info with whatever is needed:
+               String path = infoFile.getPath();
+               path = path.substring(0, path.length() - ".info".length());
+
+               String newExt = getOutputType(meta).getDefaultExtension(true);
+
+               File targetFile = new File(path + newExt);
+
+               final File ffile = targetFile;
+               return new Entry<MetaData, File>() {
+                       public File setValue(File value) {
+                               return null;
+                       }
+
+                       public File getValue() {
+                               return ffile;
+                       }
+
+                       public MetaData getKey() {
+                               return meta;
+                       }
+               };
+       }
+
        /**
         * Return the {@link OutputType} for this {@link Story}.
         *