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

Prefer exact series name as query if known. Otherwise perform the usual query normalization.

Browse Source
This commit is contained in:
Reinhard Pointner 2016-02-22 06:24:12 +00:00
parent 290d9363aa
commit 331b271d2b
1 changed files with 27 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -297,25 +297,26 @@ public class MediaDetection {
}
public static List<String> detectSeriesNames(Collection<File> files, List<IndexEntry<SearchResult>> index, Locale locale) throws Exception {
List<String> names = new ArrayList<String>();
// known series names
List<String> unids = new ArrayList<String>();
// try xattr metadata if enabled
for (File it : files) {
Object metaObject = readMetaInfo(it);
if (metaObject instanceof Episode) {
names.add(((Episode) metaObject).getSeriesName());
unids.add(((Episode) metaObject).getSeriesName());
}
}
// completely trust xattr metadata if all files are tagged
if (names.size() == files.size()) {
return getUniqueQuerySet(names);
if (unids.size() == files.size()) {
return getUniqueQuerySet(unids, emptySet());
}
// try to detect series name via nfo files
try {
for (SearchResult it : lookupSeriesNameByInfoFile(files, locale)) {
names.add(it.getName());
unids.add(it.getName());
}
} catch (Exception e) {
Logger.getLogger(MediaDetection.class.getClass().getName()).log(Level.WARNING, "Failed to lookup info by id: " + e);
@ -323,11 +324,14 @@ public class MediaDetection {
// try to detect series name via known patterns
try {
names.addAll(matchSeriesByDirectMapping(files));
unids.addAll(matchSeriesByDirectMapping(files));
} catch (Exception e) {
Logger.getLogger(MediaDetection.class.getClass().getName()).log(Level.WARNING, "Failed to match direct mappings: " + e);
}
// guessed queries
List<String> names = new ArrayList<String>();
// strict series name matcher for recognizing 1x01 patterns
SeriesNameMatcher strictSeriesNameMatcher = getSeriesNameMatcher(true);
@ -438,7 +442,7 @@ public class MediaDetection {
names.addAll(matches);
// don't allow duplicates
return getUniqueQuerySet(names);
return getUniqueQuerySet(unids, names);
}
public static List<String> matchSeriesByDirectMapping(Collection<File> files) throws Exception {
@ -933,7 +937,7 @@ public class MediaDetection {
querySet.addAll(stripReleaseInfo(files, false));
// remove duplicates
querySet = getUniqueQuerySet(stripBlacklistedTerms(querySet));
querySet = getUniqueQuerySet(emptySet(), stripBlacklistedTerms(querySet));
// DEBUG
// System.out.format("Query %s: %s%n", queryLookupService.getName(), querySet);
@ -959,14 +963,24 @@ public class MediaDetection {
return results;
}
private static List<String> getUniqueQuerySet(Collection<String> terms) {
Map<String, String> unique = new LinkedHashMap<String, String>();
private static List<String> getUniqueQuerySet(Collection<String> exactMatches, Collection<String> guessMatches) {
Map<String, String> uniqueMap = new LinkedHashMap<String, String>();
// unique key function (case-insensitive ignore-punctuation)
Function<String, String> normalize = (s) -> normalizePunctuation(s).toLowerCase();
addUniqueQuerySet(exactMatches, normalize, Function.identity(), uniqueMap);
addUniqueQuerySet(guessMatches, normalize, normalize, uniqueMap);
return new ArrayList<String>(uniqueMap.values());
}
private static void addUniqueQuerySet(Collection<String> terms, Function<String, String> keyFunction, Function<String, String> valueFunction, Map<String, String> uniqueMap) {
for (String it : terms) {
if (it.length() > 0) {
unique.put(normalizePunctuation(it).toLowerCase(), it);
String key = keyFunction.apply(it);
if (key.length() > 0 && !uniqueMap.containsKey(key)) {
uniqueMap.put(key, valueFunction.apply(it));
}
}
return new ArrayList<String>(unique.values());
}
public static String stripReleaseInfo(String name) {