mirror of https://github.com/Radarr/Radarr
766 lines
29 KiB
C#
766 lines
29 KiB
C#
using System;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text.RegularExpressions;
|
|
using NLog;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Instrumentation;
|
|
using NzbDrone.Core.MediaFiles;
|
|
using NzbDrone.Core.Qualities;
|
|
|
|
namespace NzbDrone.Core.Parser
|
|
{
|
|
public class QualityParser
|
|
{
|
|
private static readonly Logger Logger = NzbDroneLogger.GetLogger(typeof(QualityParser));
|
|
|
|
private static readonly Regex SourceRegex = new Regex(@"\b(?:
|
|
(?<bluray>M?BluRay|Blu-Ray|HD.?DVD|BD(?!$)|UHDBD|BDISO|BDMux|BD25|BD50|BR.?DISK)|
|
|
(?<webdl>WEB[-_. ]DL(?:mux)?|WEBDL|AmazonHD|iTunesHD|MaxdomeHD|NetflixU?HD|WebHD|[. ]WEB[. ](?:[xh]26[45]|DDP?5[. ]1)|[. ](?-i:WEB)$|(?:\d{3,4}0p)[-. ]WEB[-. ]|[-. ]WEB[-. ]\d{3,4}0p|\b\s\/\sWEB\s\/\s\b|AMZN[. -]WEB[. -]|NF[. ]WEB[. ])|
|
|
(?<webrip>WebRip|Web-Rip|WEBMux)|
|
|
(?<hdtv>HDTV)|
|
|
(?<bdrip>BDRip|BDLight)|
|
|
(?<brrip>BRRip)|
|
|
(?<dvdr>DVD-R|DVDR|DVD5|DVD9)|
|
|
(?<dvd>DVD(?!-R)|DVDRip|xvidvd)|
|
|
(?<dsr>WS[-_. ]DSR|DSR)|
|
|
(?<regional>R[0-9]{1}|REGIONAL)|
|
|
(?<scr>SCR|SCREENER|DVDSCR|DVDSCREENER)|
|
|
(?<ts>TS[-_. ]|TELESYNC|HD-TS|HDTS|PDVD|TSRip|HDTSRip)|
|
|
(?<tc>TC|TELECINE|HD-TC|HDTC)|
|
|
(?<cam>CAMRIP|CAM|HDCAM|HD-CAM)|
|
|
(?<wp>WORKPRINT|WP)|
|
|
(?<pdtv>PDTV)|
|
|
(?<sdtv>SDTV)|
|
|
(?<tvrip>TVRip)
|
|
)(?:\b|$|[ .])",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase | RegexOptions.IgnorePatternWhitespace);
|
|
|
|
private static readonly Regex RawHDRegex = new Regex(@"\b(?<rawhd>RawHD|Raw[-_. ]HD)\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex MPEG2Regex = new Regex(@"\b(?<mpeg2>MPEG[-_. ]?2)\b");
|
|
|
|
private static readonly Regex BRDISKRegex = new Regex(@"\b(COMPLETE|ISO|BDISO|BD25|BD50|BR.?DISK)\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex ProperRegex = new Regex(@"\b(?<proper>proper)\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex RepackRegex = new Regex(@"\b(?<repack>repack|rerip)\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex VersionRegex = new Regex(@"\dv(?<version>\d)\b|\[v(?<version>\d)\]",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex RealRegex = new Regex(@"\b(?<real>REAL)\b",
|
|
RegexOptions.Compiled);
|
|
|
|
private static readonly Regex ResolutionRegex = new Regex(@"\b(?:(?<R360p>360p)|(?<R480p>480p|640x480|848x480)|(?<R540p>540p)|(?<R576p>576p)|(?<R720p>720p|1280x720|960p)|(?<R1080p>1080p|1920x1080|1440p|FHD|1080i|4kto1080p)|(?<R2160p>2160p|3840x2160|4k[-_. ](?:UHD|HEVC|BD|H265)|(?:UHD|HEVC|BD|H265)[-_. ]4k))\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
//Handle cases where no resolution is in the release name; assume if UHD then 4k
|
|
private static readonly Regex ImpliedResolutionRegex = new Regex(@"\b(?<R2160p>UHD)\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex CodecRegex = new Regex(@"\b(?:(?<x264>x264)|(?<h264>h264)|(?<xvidhd>XvidHD)|(?<xvid>X-?vid)|(?<divx>divx))\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex OtherSourceRegex = new Regex(@"(?<hdtv>HD[-_. ]TV)|(?<sdtv>SD[-_. ]TV)", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex AnimeBlurayRegex = new Regex(@"bd(?:720|1080|2160)|(?<=[-_. (\[])bd(?=[-_. )\]])", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
private static readonly Regex AnimeWebDlRegex = new Regex(@"\[WEB\]", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex HighDefPdtvRegex = new Regex(@"hr[-_. ]ws", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex RemuxRegex = new Regex(@"(?:[_. ]|\d{4}p-)(?<remux>(?:(BD|UHD)[-_. ]?)?Remux)\b", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex HDShitQualityRegex = new Regex(@"(HD-TS|HDTS|HDTSRip|HD-TC|HDTC|HDCAM|HD-CAM)", RegexOptions.Compiled | RegexOptions.IgnoreCase);
|
|
|
|
private static readonly Regex HardcodedSubsRegex = new Regex(@"\b(?<hcsub>(\w+(?<!(SOFT|HORRIBLE))SUBS?)\b)|(?<hc>(HC|SUBBED))\b",
|
|
RegexOptions.Compiled | RegexOptions.IgnoreCase | RegexOptions.IgnorePatternWhitespace);
|
|
|
|
public static QualityModel ParseQuality(string name)
|
|
{
|
|
Logger.Debug("Trying to parse quality for {0}", name);
|
|
|
|
name = name.Trim();
|
|
|
|
var result = ParseQualityName(name);
|
|
|
|
// Based on extension
|
|
if (result.Quality == Quality.Unknown && !name.ContainsInvalidPathChars())
|
|
{
|
|
try
|
|
{
|
|
result.Quality = MediaFileExtensions.GetQualityForExtension(Path.GetExtension(name));
|
|
result.SourceDetectionSource = QualityDetectionSource.Extension;
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Extension;
|
|
result.ModifierDetectionSource = QualityDetectionSource.Extension;
|
|
}
|
|
catch (ArgumentException)
|
|
{
|
|
// Swallow exception for cases where string contains illegal
|
|
// path characters.
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
public static QualityModel ParseQualityName(string name)
|
|
{
|
|
var normalizedName = name.Replace('_', ' ').Trim();
|
|
var result = ParseQualityModifiers(name, normalizedName);
|
|
var subMatch = HardcodedSubsRegex.Matches(normalizedName).OfType<Match>().LastOrDefault();
|
|
|
|
if (subMatch != null && subMatch.Success)
|
|
{
|
|
if (subMatch.Groups["hcsub"].Success)
|
|
{
|
|
result.HardcodedSubs = subMatch.Groups["hcsub"].Value;
|
|
}
|
|
else if (subMatch.Groups["hc"].Success)
|
|
{
|
|
result.HardcodedSubs = "Generic Hardcoded Subs";
|
|
}
|
|
}
|
|
|
|
var sourceMatches = SourceRegex.Matches(normalizedName);
|
|
var sourceMatch = sourceMatches.OfType<Match>().LastOrDefault();
|
|
var resolution = ParseResolution(normalizedName);
|
|
var codecRegex = CodecRegex.Match(normalizedName);
|
|
var remuxMatch = RemuxRegex.IsMatch(normalizedName);
|
|
var brDiskMatch = BRDISKRegex.IsMatch(normalizedName);
|
|
|
|
if (RawHDRegex.IsMatch(normalizedName) && !brDiskMatch)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.RAWHD;
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution != Resolution.Unknown)
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
}
|
|
|
|
if (sourceMatch != null && sourceMatch.Success)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
|
|
if (sourceMatch.Groups["bluray"].Success)
|
|
{
|
|
if (brDiskMatch)
|
|
{
|
|
result.Quality = Quality.BRDISK;
|
|
return result;
|
|
}
|
|
|
|
if (codecRegex.Groups["xvid"].Success || codecRegex.Groups["divx"].Success)
|
|
{
|
|
result.Quality = Quality.Bluray480p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R2160p)
|
|
{
|
|
result.Quality = remuxMatch ? Quality.Remux2160p : Quality.Bluray2160p;
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p)
|
|
{
|
|
result.Quality = remuxMatch ? Quality.Remux1080p : Quality.Bluray1080p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R576p)
|
|
{
|
|
result.Quality = Quality.Bluray576p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R360p || resolution == Resolution.R480p ||
|
|
resolution == Resolution.R540p)
|
|
{
|
|
result.Quality = Quality.Bluray480p;
|
|
return result;
|
|
}
|
|
|
|
// Treat a remux without a source as 1080p, not 720p.
|
|
if (remuxMatch)
|
|
{
|
|
result.Quality = Quality.Remux1080p;
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.Bluray720p;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["webdl"].Success)
|
|
{
|
|
if (resolution == Resolution.R2160p)
|
|
{
|
|
result.Quality = Quality.WEBDL2160p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p)
|
|
{
|
|
result.Quality = Quality.WEBDL1080p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R720p)
|
|
{
|
|
result.Quality = Quality.WEBDL720p;
|
|
return result;
|
|
}
|
|
|
|
if (name.Contains("[WEBDL]"))
|
|
{
|
|
result.Quality = Quality.WEBDL720p;
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.WEBDL480p;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["webrip"].Success)
|
|
{
|
|
if (resolution == Resolution.R2160p)
|
|
{
|
|
result.Quality = Quality.WEBRip2160p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p)
|
|
{
|
|
result.Quality = Quality.WEBRip1080p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R720p)
|
|
{
|
|
result.Quality = Quality.WEBRip720p;
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.WEBRip480p;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["scr"].Success)
|
|
{
|
|
result.Quality = Quality.DVDSCR;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["cam"].Success)
|
|
{
|
|
result.Quality = Quality.CAM;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["ts"].Success)
|
|
{
|
|
result.Quality = Quality.TELESYNC;
|
|
result.Quality.Resolution = (int)resolution;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["tc"].Success)
|
|
{
|
|
result.Quality = Quality.TELECINE;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["wp"].Success)
|
|
{
|
|
result.Quality = Quality.WORKPRINT;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["regional"].Success)
|
|
{
|
|
result.Quality = Quality.REGIONAL;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["hdtv"].Success)
|
|
{
|
|
if (MPEG2Regex.IsMatch(normalizedName))
|
|
{
|
|
result.Quality = Quality.RAWHD;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R2160p)
|
|
{
|
|
result.Quality = Quality.HDTV2160p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p)
|
|
{
|
|
result.Quality = Quality.HDTV1080p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R720p)
|
|
{
|
|
result.Quality = Quality.HDTV720p;
|
|
return result;
|
|
}
|
|
|
|
if (name.Contains("[HDTV]"))
|
|
{
|
|
result.Quality = Quality.HDTV720p;
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.SDTV;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["bdrip"].Success ||
|
|
sourceMatch.Groups["brrip"].Success)
|
|
{
|
|
switch (resolution)
|
|
{
|
|
case Resolution.R720p:
|
|
result.Quality = Quality.Bluray720p;
|
|
return result;
|
|
case Resolution.R1080p:
|
|
result.Quality = Quality.Bluray1080p;
|
|
return result;
|
|
case Resolution.R2160p:
|
|
result.Quality = Quality.Bluray2160p;
|
|
return result;
|
|
case Resolution.R576p:
|
|
result.Quality = Quality.Bluray576p;
|
|
return result;
|
|
default:
|
|
result.Quality = Quality.Bluray480p;
|
|
return result;
|
|
}
|
|
}
|
|
|
|
if (sourceMatch.Groups["dvdr"].Success)
|
|
{
|
|
result.Quality = Quality.DVDR;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["dvd"].Success)
|
|
{
|
|
result.Quality = Quality.DVD;
|
|
return result;
|
|
}
|
|
|
|
if (sourceMatch.Groups["pdtv"].Success ||
|
|
sourceMatch.Groups["sdtv"].Success ||
|
|
sourceMatch.Groups["dsr"].Success ||
|
|
sourceMatch.Groups["tvrip"].Success)
|
|
{
|
|
if (resolution == Resolution.R1080p || normalizedName.ContainsIgnoreCase("1080p"))
|
|
{
|
|
result.Quality = Quality.HDTV1080p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R720p || normalizedName.ContainsIgnoreCase("720p"))
|
|
{
|
|
result.Quality = Quality.HDTV720p;
|
|
return result;
|
|
}
|
|
|
|
if (HighDefPdtvRegex.IsMatch(normalizedName))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.HDTV720p;
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.SDTV;
|
|
return result;
|
|
}
|
|
}
|
|
|
|
// Anime Bluray matching
|
|
if (AnimeBlurayRegex.Match(normalizedName).Success)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
|
|
if (resolution == Resolution.R360p || resolution == Resolution.R480p ||
|
|
resolution == Resolution.R540p || resolution == Resolution.R576p ||
|
|
normalizedName.ContainsIgnoreCase("480p"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.DVD;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p || normalizedName.ContainsIgnoreCase("1080p"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = remuxMatch ? Quality.Remux1080p : Quality.Bluray1080p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R2160p || normalizedName.ContainsIgnoreCase("2160p"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = remuxMatch ? Quality.Remux2160p : Quality.Bluray2160p;
|
|
return result;
|
|
}
|
|
|
|
// Treat a remux without a source as 1080p, not 720p.
|
|
if (remuxMatch)
|
|
{
|
|
result.Quality = Quality.Bluray1080p;
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.Bluray720p;
|
|
return result;
|
|
}
|
|
|
|
if (AnimeWebDlRegex.Match(normalizedName).Success)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
|
|
if (resolution == Resolution.R360p || resolution == Resolution.R480p ||
|
|
resolution == Resolution.R540p || resolution == Resolution.R576p ||
|
|
normalizedName.ContainsIgnoreCase("480p"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.WEBDL480p;
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p || normalizedName.ContainsIgnoreCase("1080p"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.WEBDL1080p;
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R2160p || normalizedName.ContainsIgnoreCase("2160p"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.WEBDL2160p;
|
|
|
|
return result;
|
|
}
|
|
|
|
result.Quality = Quality.WEBDL720p;
|
|
return result;
|
|
}
|
|
|
|
if (resolution != Resolution.Unknown)
|
|
{
|
|
var source = Source.UNKNOWN;
|
|
var modifier = Modifier.NONE;
|
|
|
|
if (remuxMatch)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
source = Source.BLURAY;
|
|
modifier = Modifier.REMUX;
|
|
}
|
|
else
|
|
{
|
|
try
|
|
{
|
|
var quality = MediaFileExtensions.GetQualityForExtension(name.GetPathExtension());
|
|
|
|
if (quality != Quality.Unknown)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Extension;
|
|
source = quality.Source;
|
|
}
|
|
}
|
|
catch (ArgumentException ex)
|
|
{
|
|
Logger.Debug(ex, "Unable to parse quality from extension");
|
|
}
|
|
}
|
|
|
|
if (resolution == Resolution.R2160p)
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
result.Quality = source == Source.UNKNOWN
|
|
? Quality.HDTV2160p
|
|
: QualityFinder.FindBySourceAndResolution(source, 2160, modifier);
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R1080p)
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
result.Quality = source == Source.UNKNOWN
|
|
? Quality.HDTV1080p
|
|
: QualityFinder.FindBySourceAndResolution(source, 1080, modifier);
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R720p)
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
result.Quality = source == Source.UNKNOWN
|
|
? Quality.HDTV720p
|
|
: QualityFinder.FindBySourceAndResolution(source, 720, modifier);
|
|
|
|
return result;
|
|
}
|
|
|
|
if (resolution == Resolution.R360p || resolution == Resolution.R480p ||
|
|
resolution == Resolution.R540p || resolution == Resolution.R576p)
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
result.Quality = source == Source.UNKNOWN
|
|
? Quality.SDTV
|
|
: QualityFinder.FindBySourceAndResolution(source, 480, modifier);
|
|
|
|
return result;
|
|
}
|
|
}
|
|
|
|
if (codecRegex.Groups["x264"].Success)
|
|
{
|
|
result.Quality = Quality.SDTV;
|
|
return result;
|
|
}
|
|
|
|
if (normalizedName.ContainsIgnoreCase("848x480"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
if (normalizedName.Contains("dvd"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.DVD;
|
|
}
|
|
else if (normalizedName.ContainsIgnoreCase("bluray"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.Bluray480p;
|
|
}
|
|
else
|
|
{
|
|
result.Quality = Quality.SDTV;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
if (normalizedName.ContainsIgnoreCase("1280x720"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
if (normalizedName.ContainsIgnoreCase("bluray"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.Bluray720p;
|
|
}
|
|
else
|
|
{
|
|
result.Quality = Quality.HDTV720p;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
if (normalizedName.ContainsIgnoreCase("1920x1080"))
|
|
{
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
|
|
if (normalizedName.ContainsIgnoreCase("bluray"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.Bluray1080p;
|
|
}
|
|
else
|
|
{
|
|
result.Quality = Quality.HDTV1080p;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
if (normalizedName.ContainsIgnoreCase("bluray720p"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.Bluray720p;
|
|
|
|
return result;
|
|
}
|
|
|
|
if (normalizedName.ContainsIgnoreCase("bluray1080p"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.Bluray1080p;
|
|
|
|
return result;
|
|
}
|
|
|
|
if (normalizedName.ContainsIgnoreCase("bluray2160p"))
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.ResolutionDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = Quality.Bluray2160p;
|
|
|
|
return result;
|
|
}
|
|
|
|
var otherSourceMatch = OtherSourceMatch(normalizedName);
|
|
|
|
if (otherSourceMatch != Quality.Unknown)
|
|
{
|
|
result.SourceDetectionSource = QualityDetectionSource.Name;
|
|
result.Quality = otherSourceMatch;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
private static Resolution ParseResolution(string name)
|
|
{
|
|
var match = ResolutionRegex.Match(name);
|
|
|
|
var matchimplied = ImpliedResolutionRegex.Match(name);
|
|
|
|
if (!match.Success & !matchimplied.Success)
|
|
{
|
|
return Resolution.Unknown;
|
|
}
|
|
|
|
if (match.Groups["R360p"].Success)
|
|
{
|
|
return Resolution.R360p;
|
|
}
|
|
|
|
if (match.Groups["R480p"].Success)
|
|
{
|
|
return Resolution.R480p;
|
|
}
|
|
|
|
if (match.Groups["R540p"].Success)
|
|
{
|
|
return Resolution.R540p;
|
|
}
|
|
|
|
if (match.Groups["R576p"].Success)
|
|
{
|
|
return Resolution.R576p;
|
|
}
|
|
|
|
if (match.Groups["R720p"].Success)
|
|
{
|
|
return Resolution.R720p;
|
|
}
|
|
|
|
if (match.Groups["R1080p"].Success)
|
|
{
|
|
return Resolution.R1080p;
|
|
}
|
|
|
|
if (match.Groups["R2160p"].Success || matchimplied.Groups["R2160p"].Success)
|
|
{
|
|
return Resolution.R2160p;
|
|
}
|
|
|
|
return Resolution.Unknown;
|
|
}
|
|
|
|
private static Quality OtherSourceMatch(string name)
|
|
{
|
|
var match = OtherSourceRegex.Match(name);
|
|
|
|
if (!match.Success)
|
|
{
|
|
return Quality.Unknown;
|
|
}
|
|
|
|
if (match.Groups["sdtv"].Success)
|
|
{
|
|
return Quality.SDTV;
|
|
}
|
|
|
|
if (match.Groups["hdtv"].Success)
|
|
{
|
|
return Quality.HDTV720p;
|
|
}
|
|
|
|
return Quality.Unknown;
|
|
}
|
|
|
|
private static QualityModel ParseQualityModifiers(string name, string normalizedName)
|
|
{
|
|
var result = new QualityModel { Quality = Quality.Unknown };
|
|
|
|
if (ProperRegex.IsMatch(normalizedName))
|
|
{
|
|
result.Revision.Version = 2;
|
|
result.RevisionDetectionSource = QualityDetectionSource.Name;
|
|
}
|
|
|
|
if (RepackRegex.IsMatch(normalizedName))
|
|
{
|
|
result.Revision.Version = 2;
|
|
result.Revision.IsRepack = true;
|
|
result.RevisionDetectionSource = QualityDetectionSource.Name;
|
|
}
|
|
|
|
var versionRegexResult = VersionRegex.Match(normalizedName);
|
|
|
|
if (versionRegexResult.Success)
|
|
{
|
|
result.Revision.Version = Convert.ToInt32(versionRegexResult.Groups["version"].Value);
|
|
result.RevisionDetectionSource = QualityDetectionSource.Name;
|
|
}
|
|
|
|
// TODO: re-enable this when we have a reliable way to determine real
|
|
// TODO: Only treat it as a real if it comes AFTER the season/episode number
|
|
var realRegexResult = RealRegex.Matches(name);
|
|
|
|
if (realRegexResult.Count > 0)
|
|
{
|
|
result.Revision.Real = realRegexResult.Count;
|
|
result.RevisionDetectionSource = QualityDetectionSource.Name;
|
|
}
|
|
|
|
return result;
|
|
}
|
|
}
|
|
|
|
public enum Resolution
|
|
{
|
|
Unknown,
|
|
R360p = 360,
|
|
R480p = 480,
|
|
R540p = 540,
|
|
R576p = 576,
|
|
R720p = 720,
|
|
R1080p = 1080,
|
|
R2160p = 2160
|
|
}
|
|
}
|