From c22d867453de3371d8059bdb6603094e3cab1959 Mon Sep 17 00:00:00 2001 From: Reinhard Pointner Date: Tue, 29 Nov 2011 00:56:56 +0000 Subject: [PATCH] * ignore empty query --- .../filebot/web/OpenSubtitlesClient.java | 53 ++++++++++--------- .../filebot/web/OpenSubtitlesHasher.java | 2 +- 2 files changed, 30 insertions(+), 25 deletions(-) diff --git a/source/net/sourceforge/filebot/web/OpenSubtitlesClient.java b/source/net/sourceforge/filebot/web/OpenSubtitlesClient.java index c17014ea..2069816a 100644 --- a/source/net/sourceforge/filebot/web/OpenSubtitlesClient.java +++ b/source/net/sourceforge/filebot/web/OpenSubtitlesClient.java @@ -3,6 +3,7 @@ package net.sourceforge.filebot.web; import static java.util.Collections.*; +import static net.sourceforge.filebot.web.OpenSubtitlesHasher.*; import java.io.File; import java.math.BigInteger; @@ -106,26 +107,29 @@ public class OpenSubtitlesClient implements SubtitleProvider, VideoHashSubtitleS List queryList = new ArrayList(files.length); for (File file : files) { - String movieHash = OpenSubtitlesHasher.computeHash(file); - // add query - queryList.add(Query.forHash(movieHash, file.length(), languageFilter)); + if (file.length() > HASH_CHUNK_SIZE) { + String movieHash = computeHash(file); + queryList.add(Query.forHash(movieHash, file.length(), languageFilter)); + hashMap.put(movieHash, file); + } // prepare result map - hashMap.put(movieHash, file); resultMap.put(file, new LinkedList()); } - // require login - login(); - - // submit query and map results to given files - for (OpenSubtitlesSubtitleDescriptor subtitle : xmlrpc.searchSubtitles(queryList)) { - // get file for hash - File file = hashMap.get(subtitle.getMovieHash()); + if (queryList.size() > 0) { + // require login + login(); - // add subtitle - resultMap.get(file).add(subtitle); + // submit query and map results to given files + for (OpenSubtitlesSubtitleDescriptor subtitle : xmlrpc.searchSubtitles(queryList)) { + // get file for hash + File file = hashMap.get(subtitle.getMovieHash()); + + // add subtitle + resultMap.get(file).add(subtitle); + } } return resultMap; @@ -180,19 +184,20 @@ public class OpenSubtitlesClient implements SubtitleProvider, VideoHashSubtitleS Map indexMap = new HashMap(movieFiles.length); for (int i = 0; i < movieFiles.length; i++) { - String hash = OpenSubtitlesHasher.computeHash(movieFiles[i]); - - // remember original index - indexMap.put(hash, i); + if (movieFiles[i].length() > HASH_CHUNK_SIZE) { + indexMap.put(computeHash(movieFiles[i]), i); // remember original index + } } - // require login - login(); - - // dispatch single query for all hashes - for (Entry entry : xmlrpc.checkMovieHash(indexMap.keySet()).entrySet()) { - int index = indexMap.get(entry.getKey()); - result[index] = entry.getValue(); + if (indexMap.size() > 0) { + // require login + login(); + + // dispatch single query for all hashes + for (Entry entry : xmlrpc.checkMovieHash(indexMap.keySet()).entrySet()) { + int index = indexMap.get(entry.getKey()); + result[index] = entry.getValue(); + } } return result; diff --git a/source/net/sourceforge/filebot/web/OpenSubtitlesHasher.java b/source/net/sourceforge/filebot/web/OpenSubtitlesHasher.java index 43192789..900513a2 100644 --- a/source/net/sourceforge/filebot/web/OpenSubtitlesHasher.java +++ b/source/net/sourceforge/filebot/web/OpenSubtitlesHasher.java @@ -24,7 +24,7 @@ public final class OpenSubtitlesHasher { /** * Size of the chunks that will be hashed in bytes (64 KB) */ - private static final int HASH_CHUNK_SIZE = 64 * 1024; + public static final int HASH_CHUNK_SIZE = 64 * 1024; public static String computeHash(File file) throws IOException {