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

* better performance for SxE/Airdate matching

Browse Source
This commit is contained in:
Reinhard Pointner 2011-11-13 18:29:25 +00:00
parent cfee1cbb51
commit 9c55decf67
4 changed files with 39 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
source/net/sourceforge/filebot/cli/CmdlineOperations.java
View File

@ -150,7 +150,7 @@ public class CmdlineOperations implements CmdlineInterface {
if (strict) {
sequence = new SimilarityMetric[] { StrictMetric.EpisodeIdentifier, StrictMetric.Title, StrictMetric.Name }; // use SEI for matching and SN for excluding false positives
} else {
sequence = MatchSimilarityMetric.defaultSequence(); // same as in GUI
sequence = MatchSimilarityMetric.defaultSequence(false); // same as in GUI
}
List<Match<File, Episode>> matches = new ArrayList<Match<File, Episode>>();

2
source/net/sourceforge/filebot/ui/rename/EpisodeListMatcher.java
View File

@ -192,7 +192,7 @@ class EpisodeListMatcher implements AutoCompleteMatcher {
// group by subtitles first and then by files in general
for (List<File> filesPerType : mapByExtension(mediaFiles).values()) {
Matcher<File, Episode> matcher = new Matcher<File, Episode>(filesPerType, episodes, false, MatchSimilarityMetric.defaultSequence());
Matcher<File, Episode> matcher = new Matcher<File, Episode>(filesPerType, episodes, false, MatchSimilarityMetric.defaultSequence(false));
matches.addAll(matcher.match());
}

2
source/net/sourceforge/filebot/ui/rename/MatchAction.java
View File

@ -49,7 +49,7 @@ class MatchAction extends AbstractAction {
Window window = getWindow(evt.getSource());
window.setCursor(Cursor.getPredefinedCursor(Cursor.WAIT_CURSOR));
BackgroundMatcher backgroundMatcher = new BackgroundMatcher(model, MatchSimilarityMetric.defaultSequence());
BackgroundMatcher backgroundMatcher = new BackgroundMatcher(model, MatchSimilarityMetric.defaultSequence(true));
backgroundMatcher.execute();
try {

46
source/net/sourceforge/filebot/ui/rename/MatchSimilarityMetric.java
View File

@ -3,12 +3,14 @@ package net.sourceforge.filebot.ui.rename;
import static java.lang.Math.*;
import static java.util.Arrays.*;
import static java.util.Collections.*;
import static net.sourceforge.tuned.FileUtilities.*;
import java.io.File;
import java.util.Arrays;
import java.util.Collection;
import java.util.Collections;
import java.util.Map;
import java.util.WeakHashMap;
import net.sourceforge.filebot.similarity.DateMetric;
import net.sourceforge.filebot.similarity.FileSizeMetric;
@ -65,8 +67,16 @@ public enum MatchSimilarityMetric implements SimilarityMetric {
// Match by season / episode numbers
SeasonEpisode(new SeasonEpisodeMetric() {
private final Map<Object, Collection<SxE>> matchCache = synchronizedMap(new WeakHashMap<Object, Collection<SxE>>(64, 4));
@Override
protected Collection<SxE> parse(Object object) {
Collection<SxE> result = matchCache.get(object);
if (result != null) {
return result;
}
if (object instanceof Episode) {
Episode episode = (Episode) object;
@ -74,26 +84,39 @@ public enum MatchSimilarityMetric implements SimilarityMetric {
SxE seasonEpisode = new SxE(episode.getSeason(), episode.getEpisode());
SxE absoluteEpisode = new SxE(null, episode.getAbsolute());
return seasonEpisode.equals(absoluteEpisode) ? Collections.singleton(absoluteEpisode) : Arrays.asList(seasonEpisode, absoluteEpisode);
result = seasonEpisode.equals(absoluteEpisode) ? singleton(absoluteEpisode) : asList(seasonEpisode, absoluteEpisode);
} else {
result = super.parse(object);
}
return super.parse(object);
matchCache.put(object, result);
return result;
}
}),
// Match episode airdate
AirDate(new DateMetric() {
private final Map<Object, Date> matchCache = synchronizedMap(new WeakHashMap<Object, Date>(64, 4));
@Override
protected Date parse(Object object) {
if (object instanceof Episode) {
Episode episode = (Episode) object;
// create SxE from episode
// use airdate from episode
return episode.airdate();
}
return super.parse(object);
Date result = matchCache.get(object);
if (result != null) {
return result;
}
result = super.parse(object);
matchCache.put(object, result);
return result;
}
}),
@ -152,7 +175,6 @@ public enum MatchSimilarityMetric implements SimilarityMetric {
return new Object[] { object };
}
}),
// Match by generic name similarity
@ -220,13 +242,17 @@ public enum MatchSimilarityMetric implements SimilarityMetric {
}
public static SimilarityMetric[] defaultSequence() {
public static SimilarityMetric[] defaultSequence(boolean includeFileMetrics) {
// 1. pass: match by file length (fast, but only works when matching torrents or files)
// 2. pass: match by season / episode numbers
// 3. pass: match by checking series/episode title against the file path
// 3. pass: match by checking series / episode title against the file path
// 4. pass: match by generic name similarity (slow, but most matches will have been determined in second pass)
// 5. pass: match by generic numeric similarity
return new SimilarityMetric[] { FileSize, EpisodeIdentifier, Title, Name, Numeric };
if (includeFileMetrics) {
return new SimilarityMetric[] { FileSize, EpisodeIdentifier, Title, Name, Numeric };
} else {
return new SimilarityMetric[] { EpisodeIdentifier, Title, Name, Numeric };
}
}
}