moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-08-13 17:03:45 -04:00
Code Issues Releases Wiki Activity

* fix ranking issues

Browse Source
This commit is contained in:
Reinhard Pointner 2014-09-15 23:34:43 +00:00
parent dd17ec1556
commit d228f6d758
1 changed files with 10 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
source/net/filebot/media/MediaDetection.java
View File

@ -39,6 +39,7 @@ import java.util.logging.Level;
import java.util.logging.Logger; import java.util.logging.Logger;
import java.util.regex.Matcher; import java.util.regex.Matcher;
import java.util.regex.Pattern; import java.util.regex.Pattern;
import java.util.stream.Collectors;
import net.filebot.WebServices; import net.filebot.WebServices;
import net.filebot.archive.Archive; import net.filebot.archive.Archive;
@ -554,7 +555,7 @@ public class MediaDetection {
} }
public static List<Movie> detectMovie(File movieFile, MovieIdentificationService hashLookupService, MovieIdentificationService queryLookupService, Locale locale, boolean strict) throws Exception { public static List<Movie> detectMovie(File movieFile, MovieIdentificationService hashLookupService, MovieIdentificationService queryLookupService, Locale locale, boolean strict) throws Exception {
Set<Movie> options = new LinkedHashSet<Movie>(); List<Movie> options = new ArrayList<Movie>();
// try xattr metadata if enabled // try xattr metadata if enabled
Object metaObject = readMetaInfo(movieFile); Object metaObject = readMetaInfo(movieFile);
@ -643,12 +644,9 @@ public class MediaDetection {
} }
} }
// add local index matches before online search (and retain additional alias titles available in the local index)
options.addAll(movieNameMatches);
// query by file / folder name // query by file / folder name
if (queryLookupService != null) { if (queryLookupService != null) {
Collection<Movie> results = queryMovieByFileName(terms, queryLookupService, locale); List<Movie> results = queryMovieByFileName(terms, queryLookupService, locale);
// try query without year as it sometimes messes up results if years don't match properly (movie release years vs dvd release year, etc) // try query without year as it sometimes messes up results if years don't match properly (movie release years vs dvd release year, etc)
if (results.isEmpty() && !strict) { if (results.isEmpty() && !strict) {
@ -668,9 +666,13 @@ public class MediaDetection {
} }
} }
// online results have better ranking so add them first
options.addAll(results); options.addAll(results);
} }
// consider potential local index matches second
options.addAll(movieNameMatches);
// sort by relevance // sort by relevance
return sortBySimilarity(options, terms, getMovieMatchMetric(), true); return sortBySimilarity(options, terms, getMovieMatchMetric(), true);
} }
@ -732,9 +734,8 @@ public class MediaDetection {
} }
}; };
// sort output array // sort by ranking and remvove duplicate entries
List<T> result = new ArrayList<T>(options); List<T> result = options.stream().sorted(comparator).distinct().collect(Collectors.toList());
sort(result, comparator);
// DEBUG // DEBUG
// System.out.format("sortBySimilarity %s => %s%n", terms, result); // System.out.format("sortBySimilarity %s => %s%n", terms, result);
@ -937,7 +938,7 @@ public class MediaDetection {
return new ArrayList<Movie>(movies); return new ArrayList<Movie>(movies);
} }
private static Collection<Movie> queryMovieByFileName(Collection<String> files, MovieIdentificationService queryLookupService, Locale locale) throws Exception { private static List<Movie> queryMovieByFileName(Collection<String> files, MovieIdentificationService queryLookupService, Locale locale) throws Exception {
// remove blacklisted terms // remove blacklisted terms
List<String> querySet = new ArrayList<String>(); List<String> querySet = new ArrayList<String>();
querySet.addAll(stripReleaseInfo(files, true)); querySet.addAll(stripReleaseInfo(files, true));