moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-12-24 16:58:51 -05:00
Code Issues Releases Wiki Activity

Fix formatMultiEpisode preview format

Browse Source
This commit is contained in:
Reinhard Pointner 2016-02-26 17:42:21 +00:00
parent 07037b34ce
commit b301f445a5
1 changed files with 11 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
source/net/filebot/web/EpisodeFormat.java
View File

@ -2,22 +2,21 @@ package net.filebot.web;
import static java.util.stream.Collectors.*; import static java.util.stream.Collectors.*;
import static net.filebot.similarity.Normalization.*; import static net.filebot.similarity.Normalization.*;
import static net.filebot.util.StringUtilities.*;
import java.text.FieldPosition; import java.text.FieldPosition;
import java.text.Format; import java.text.Format;
import java.text.ParseException; import java.text.ParseException;
import java.text.ParsePosition; import java.text.ParsePosition;
import java.util.LinkedHashSet;
import java.util.List; import java.util.List;
import java.util.Set;
import java.util.SortedMap; import java.util.SortedMap;
import java.util.SortedSet; import java.util.SortedSet;
import java.util.TreeMap; import java.util.TreeMap;
import java.util.TreeSet; import java.util.TreeSet;
import java.util.function.Function;
import java.util.regex.Matcher; import java.util.regex.Matcher;
import java.util.regex.Pattern; import java.util.regex.Pattern;
import java.util.stream.IntStream; import java.util.stream.IntStream;
import java.util.stream.Stream;
public class EpisodeFormat extends Format { public class EpisodeFormat extends Format {
@ -60,15 +59,13 @@ public class EpisodeFormat extends Format {
} }
public String formatMultiEpisode(List<Episode> episodes) { public String formatMultiEpisode(List<Episode> episodes) {
Set<String> name = new LinkedHashSet<String>(2); Function<Episode, String> seriesName = it -> it.getSeriesName();
Set<String> sxe = new LinkedHashSet<String>(2); Function<Episode, String> episodeNumber = it -> formatSxE(it);
Set<String> title = new LinkedHashSet<String>(2); Function<Episode, String> episodeTitle = it -> removeTrailingBrackets(it.getTitle());
for (Episode it : episodes) {
name.add(it.getSeriesName()); return Stream.of(seriesName, episodeNumber, episodeTitle).map(f -> {
sxe.add(formatSxE(it)); return episodes.stream().map(f::apply).distinct().collect(joining(" & "));
title.add(removeTrailingBrackets(it.getTitle())); }).collect(joining(" - "));
}
return String.format("%s - %s - %s", join(name, " & "), join(" & ", sxe), join(" & ", title));
} }
public String formatSxE(Episode episode) { public String formatSxE(Episode episode) {
@ -102,7 +99,7 @@ public class EpisodeFormat extends Format {
} }
public String formatMultiTitle(List<Episode> episodes) { public String formatMultiTitle(List<Episode> episodes) {
return episodes.stream().map(e -> removeTrailingBrackets(e.getTitle())).distinct().collect(joining(" & ")); return episodes.stream().map(it -> removeTrailingBrackets(it.getTitle())).distinct().collect(joining(" & "));
} }
public String formatMultiRangeSxE(List<Episode> episodes) { public String formatMultiRangeSxE(List<Episode> episodes) {
@ -116,8 +113,7 @@ public class EpisodeFormat extends Format {
public String formatMultiRangeNumbers(List<Episode> episodes, String seasonFormat, String episodeFormat) { public String formatMultiRangeNumbers(List<Episode> episodes, String seasonFormat, String episodeFormat) {
return getSeasonEpisodeNumbers(episodes).entrySet().stream().map(it -> { return getSeasonEpisodeNumbers(episodes).entrySet().stream().map(it -> {
String s = it.getKey() >= 0 ? String.format(seasonFormat, it.getKey()) : ""; String s = it.getKey() >= 0 ? String.format(seasonFormat, it.getKey()) : "";
String e = IntStream.of(it.getValue().first(), it.getValue().last()).distinct().mapToObj(i -> String.format(episodeFormat, i)).collect(joining("-")); return IntStream.of(it.getValue().first(), it.getValue().last()).distinct().mapToObj(i -> String.format(episodeFormat, i)).collect(joining("-", s, ""));
return s + e;
}).collect(joining(" - ")); }).collect(joining(" - "));
} }
@ -126,7 +122,6 @@ public class EpisodeFormat extends Format {
for (Episode it : episodes) { for (Episode it : episodes) {
Integer s = it.getSeason() == null || it.getSpecial() != null ? it.getSpecial() == null ? -1 : 0 : it.getSeason(); Integer s = it.getSeason() == null || it.getSpecial() != null ? it.getSpecial() == null ? -1 : 0 : it.getSeason();
Integer e = it.getEpisode() == null ? it.getSpecial() == null ? -1 : it.getSpecial() : it.getEpisode(); Integer e = it.getEpisode() == null ? it.getSpecial() == null ? -1 : it.getSpecial() : it.getEpisode();
n.computeIfAbsent(s, key -> new TreeSet<Integer>()).add(e); n.computeIfAbsent(s, key -> new TreeSet<Integer>()).add(e);
} }
return n; return n;