| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090 |
- using MediaBrowser.Common.Extensions;
- using MediaBrowser.Common.IO;
- using MediaBrowser.Common.Kernel;
- using MediaBrowser.Common.Serialization;
- using MediaBrowser.Controller.Entities;
- using MediaBrowser.Controller.Entities.Audio;
- using MediaBrowser.Controller.Library;
- using MediaBrowser.Model.Entities;
- using MediaBrowser.Model.IO;
- using System;
- using System.Collections.Generic;
- using System.ComponentModel;
- using System.Diagnostics;
- using System.IO;
- using System.Linq;
- using System.Reflection;
- using System.Runtime.InteropServices;
- using System.Threading;
- using System.Threading.Tasks;
- namespace MediaBrowser.Controller.MediaInfo
- {
- /// <summary>
- /// Class FFMpegManager
- /// </summary>
- public class FFMpegManager : BaseManager<Kernel>
- {
- /// <summary>
- /// Holds the list of new items to generate chapter image for when the NewItemTimer expires
- /// </summary>
- private readonly List<Video> _newlyAddedItems = new List<Video>();
- /// <summary>
- /// The amount of time to wait before generating chapter images
- /// </summary>
- private const int NewItemDelay = 300000;
- /// <summary>
- /// The current new item timer
- /// </summary>
- /// <value>The new item timer.</value>
- private Timer NewItemTimer { get; set; }
- /// <summary>
- /// Gets or sets the video image cache.
- /// </summary>
- /// <value>The video image cache.</value>
- internal FileSystemRepository VideoImageCache { get; set; }
- /// <summary>
- /// Gets or sets the image cache.
- /// </summary>
- /// <value>The image cache.</value>
- internal FileSystemRepository AudioImageCache { get; set; }
- /// <summary>
- /// Gets or sets the subtitle cache.
- /// </summary>
- /// <value>The subtitle cache.</value>
- internal FileSystemRepository SubtitleCache { get; set; }
- /// <summary>
- /// Gets or sets the zip client.
- /// </summary>
- /// <value>The zip client.</value>
- private IZipClient ZipClient { get; set; }
- /// <summary>
- /// Initializes a new instance of the <see cref="FFMpegManager" /> class.
- /// </summary>
- /// <param name="kernel">The kernel.</param>
- /// <param name="zipClient">The zip client.</param>
- /// <exception cref="System.ArgumentNullException">zipClient</exception>
- public FFMpegManager(Kernel kernel, IZipClient zipClient)
- : base(kernel)
- {
- if (zipClient == null)
- {
- throw new ArgumentNullException("zipClient");
- }
- ZipClient = zipClient;
- // Not crazy about this but it's the only way to suppress ffmpeg crash dialog boxes
- SetErrorMode(ErrorModes.SEM_FAILCRITICALERRORS | ErrorModes.SEM_NOALIGNMENTFAULTEXCEPT | ErrorModes.SEM_NOGPFAULTERRORBOX | ErrorModes.SEM_NOOPENFILEERRORBOX);
- VideoImageCache = new FileSystemRepository(VideoImagesDataPath);
- AudioImageCache = new FileSystemRepository(AudioImagesDataPath);
- SubtitleCache = new FileSystemRepository(SubtitleCachePath);
- Kernel.LibraryManager.LibraryChanged += LibraryManager_LibraryChanged;
- Task.Run(() => VersionedDirectoryPath = GetVersionedDirectoryPath());
- }
- /// <summary>
- /// Handles the LibraryChanged event of the LibraryManager control.
- /// </summary>
- /// <param name="sender">The source of the event.</param>
- /// <param name="e">The <see cref="ChildrenChangedEventArgs" /> instance containing the event data.</param>
- void LibraryManager_LibraryChanged(object sender, ChildrenChangedEventArgs e)
- {
- var videos = e.ItemsAdded.OfType<Video>().ToList();
- // Use a timer to prevent lots of these notifications from showing in a short period of time
- if (videos.Count > 0)
- {
- lock (_newlyAddedItems)
- {
- _newlyAddedItems.AddRange(videos);
- if (NewItemTimer == null)
- {
- NewItemTimer = new Timer(NewItemTimerCallback, null, NewItemDelay, Timeout.Infinite);
- }
- else
- {
- NewItemTimer.Change(NewItemDelay, Timeout.Infinite);
- }
- }
- }
- }
- /// <summary>
- /// Called when the new item timer expires
- /// </summary>
- /// <param name="state">The state.</param>
- private async void NewItemTimerCallback(object state)
- {
- List<Video> newItems;
- // Lock the list and release all resources
- lock (_newlyAddedItems)
- {
- newItems = _newlyAddedItems.ToList();
- _newlyAddedItems.Clear();
- NewItemTimer.Dispose();
- NewItemTimer = null;
- }
- // Limit the number of videos we generate images for
- // The idea is to catch new items that are added here and there
- // Mass image generation can be left to the scheduled task
- foreach (var video in newItems.Where(c => c.Chapters != null).Take(3))
- {
- try
- {
- await PopulateChapterImages(video, CancellationToken.None, true, true).ConfigureAwait(false);
- }
- catch (Exception ex)
- {
- Logger.ErrorException("Error creating chapter images for {0}", ex, video.Name);
- }
- }
- }
- /// <summary>
- /// Releases unmanaged and - optionally - managed resources.
- /// </summary>
- /// <param name="dispose"><c>true</c> to release both managed and unmanaged resources; <c>false</c> to release only unmanaged resources.</param>
- protected override void Dispose(bool dispose)
- {
- if (dispose)
- {
- if (NewItemTimer != null)
- {
- NewItemTimer.Dispose();
- }
- SetErrorMode(ErrorModes.SYSTEM_DEFAULT);
- Kernel.LibraryManager.LibraryChanged -= LibraryManager_LibraryChanged;
- AudioImageCache.Dispose();
- VideoImageCache.Dispose();
- }
- base.Dispose(dispose);
- }
- /// <summary>
- /// The FF probe resource pool count
- /// </summary>
- private const int FFProbeResourcePoolCount = 4;
- /// <summary>
- /// The audio image resource pool count
- /// </summary>
- private const int AudioImageResourcePoolCount = 4;
- /// <summary>
- /// The video image resource pool count
- /// </summary>
- private const int VideoImageResourcePoolCount = 2;
- /// <summary>
- /// The FF probe resource pool
- /// </summary>
- private readonly SemaphoreSlim FFProbeResourcePool = new SemaphoreSlim(FFProbeResourcePoolCount, FFProbeResourcePoolCount);
- /// <summary>
- /// The audio image resource pool
- /// </summary>
- private readonly SemaphoreSlim AudioImageResourcePool = new SemaphoreSlim(AudioImageResourcePoolCount, AudioImageResourcePoolCount);
- /// <summary>
- /// The video image resource pool
- /// </summary>
- private readonly SemaphoreSlim VideoImageResourcePool = new SemaphoreSlim(VideoImageResourcePoolCount, VideoImageResourcePoolCount);
- /// <summary>
- /// Gets or sets the versioned directory path.
- /// </summary>
- /// <value>The versioned directory path.</value>
- private string VersionedDirectoryPath { get; set; }
- /// <summary>
- /// Gets the FFMPEG version.
- /// </summary>
- /// <value>The FFMPEG version.</value>
- public string FFMpegVersion
- {
- get { return Path.GetFileNameWithoutExtension(VersionedDirectoryPath); }
- }
- /// <summary>
- /// The _ FF MPEG path
- /// </summary>
- private string _FFMpegPath;
- /// <summary>
- /// Gets the path to ffmpeg.exe
- /// </summary>
- /// <value>The FF MPEG path.</value>
- public string FFMpegPath
- {
- get
- {
- return _FFMpegPath ?? (_FFMpegPath = Path.Combine(VersionedDirectoryPath, "ffmpeg.exe"));
- }
- }
- /// <summary>
- /// The _ FF probe path
- /// </summary>
- private string _FFProbePath;
- /// <summary>
- /// Gets the path to ffprobe.exe
- /// </summary>
- /// <value>The FF probe path.</value>
- public string FFProbePath
- {
- get
- {
- return _FFProbePath ?? (_FFProbePath = Path.Combine(VersionedDirectoryPath, "ffprobe.exe"));
- }
- }
- /// <summary>
- /// The _video images data path
- /// </summary>
- private string _videoImagesDataPath;
- /// <summary>
- /// Gets the video images data path.
- /// </summary>
- /// <value>The video images data path.</value>
- public string VideoImagesDataPath
- {
- get
- {
- if (_videoImagesDataPath == null)
- {
- _videoImagesDataPath = Path.Combine(Kernel.ApplicationPaths.DataPath, "ffmpeg-video-images");
- if (!Directory.Exists(_videoImagesDataPath))
- {
- Directory.CreateDirectory(_videoImagesDataPath);
- }
- }
- return _videoImagesDataPath;
- }
- }
- /// <summary>
- /// The _audio images data path
- /// </summary>
- private string _audioImagesDataPath;
- /// <summary>
- /// Gets the audio images data path.
- /// </summary>
- /// <value>The audio images data path.</value>
- public string AudioImagesDataPath
- {
- get
- {
- if (_audioImagesDataPath == null)
- {
- _audioImagesDataPath = Path.Combine(Kernel.ApplicationPaths.DataPath, "ffmpeg-audio-images");
- if (!Directory.Exists(_audioImagesDataPath))
- {
- Directory.CreateDirectory(_audioImagesDataPath);
- }
- }
- return _audioImagesDataPath;
- }
- }
- /// <summary>
- /// The _subtitle cache path
- /// </summary>
- private string _subtitleCachePath;
- /// <summary>
- /// Gets the subtitle cache path.
- /// </summary>
- /// <value>The subtitle cache path.</value>
- public string SubtitleCachePath
- {
- get
- {
- if (_subtitleCachePath == null)
- {
- _subtitleCachePath = Path.Combine(Kernel.ApplicationPaths.CachePath, "ffmpeg-subtitles");
- if (!Directory.Exists(_subtitleCachePath))
- {
- Directory.CreateDirectory(_subtitleCachePath);
- }
- }
- return _subtitleCachePath;
- }
- }
- /// <summary>
- /// Gets the versioned directory path.
- /// </summary>
- /// <returns>System.String.</returns>
- private string GetVersionedDirectoryPath()
- {
- var assembly = GetType().Assembly;
- const string prefix = "MediaBrowser.Controller.MediaInfo.";
- const string srch = prefix + "ffmpeg";
- var resource = assembly.GetManifestResourceNames().First(r => r.StartsWith(srch));
- var filename = resource.Substring(resource.IndexOf(prefix, StringComparison.OrdinalIgnoreCase) + prefix.Length);
- var versionedDirectoryPath = Path.Combine(Kernel.ApplicationPaths.MediaToolsPath, Path.GetFileNameWithoutExtension(filename));
- if (!Directory.Exists(versionedDirectoryPath))
- {
- Directory.CreateDirectory(versionedDirectoryPath);
- }
- ExtractTools(assembly, resource, versionedDirectoryPath);
- return versionedDirectoryPath;
- }
- /// <summary>
- /// Extracts the tools.
- /// </summary>
- /// <param name="assembly">The assembly.</param>
- /// <param name="zipFileResourcePath">The zip file resource path.</param>
- /// <param name="targetPath">The target path.</param>
- private void ExtractTools(Assembly assembly, string zipFileResourcePath, string targetPath)
- {
- using (var resourceStream = assembly.GetManifestResourceStream(zipFileResourcePath))
- {
- ZipClient.ExtractAll(resourceStream, targetPath, false);
- }
- }
- /// <summary>
- /// Gets the probe size argument.
- /// </summary>
- /// <param name="item">The item.</param>
- /// <returns>System.String.</returns>
- public string GetProbeSizeArgument(BaseItem item)
- {
- var video = item as Video;
- return video != null ? GetProbeSizeArgument(video.VideoType, video.IsoType) : string.Empty;
- }
- /// <summary>
- /// Gets the probe size argument.
- /// </summary>
- /// <param name="videoType">Type of the video.</param>
- /// <param name="isoType">Type of the iso.</param>
- /// <returns>System.String.</returns>
- public string GetProbeSizeArgument(VideoType videoType, IsoType? isoType)
- {
- if (videoType == VideoType.Dvd || (isoType.HasValue && isoType.Value == IsoType.Dvd))
- {
- return "-probesize 1G -analyzeduration 200M";
- }
- return string.Empty;
- }
- /// <summary>
- /// Runs FFProbe against a BaseItem
- /// </summary>
- /// <param name="item">The item.</param>
- /// <param name="inputPath">The input path.</param>
- /// <param name="lastDateModified">The last date modified.</param>
- /// <param name="cache">The cache.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task{FFProbeResult}.</returns>
- /// <exception cref="System.ArgumentNullException">item</exception>
- public Task<FFProbeResult> RunFFProbe(BaseItem item, string inputPath, DateTime lastDateModified, FileSystemRepository cache, CancellationToken cancellationToken)
- {
- if (string.IsNullOrEmpty(inputPath))
- {
- throw new ArgumentNullException("inputPath");
- }
- if (cache == null)
- {
- throw new ArgumentNullException("cache");
- }
- // Put the ffmpeg version into the cache name so that it's unique per-version
- // We don't want to try and deserialize data based on an old version, which could potentially fail
- var resourceName = item.Id + "_" + lastDateModified.Ticks + "_" + FFMpegVersion;
- // Forumulate the cache file path
- var cacheFilePath = cache.GetResourcePath(resourceName, ".pb");
- cancellationToken.ThrowIfCancellationRequested();
- // Avoid File.Exists by just trying to deserialize
- try
- {
- return Task.FromResult(Kernel.ProtobufSerializer.DeserializeFromFile<FFProbeResult>(cacheFilePath));
- }
- catch (FileNotFoundException)
- {
- var extractChapters = false;
- var video = item as Video;
- var probeSizeArgument = string.Empty;
- if (video != null)
- {
- extractChapters = true;
- probeSizeArgument = GetProbeSizeArgument(video.VideoType, video.IsoType);
- }
- return RunFFProbeInternal(inputPath, extractChapters, cacheFilePath, probeSizeArgument, cancellationToken);
- }
- }
- /// <summary>
- /// Runs FFProbe against a BaseItem
- /// </summary>
- /// <param name="inputPath">The input path.</param>
- /// <param name="extractChapters">if set to <c>true</c> [extract chapters].</param>
- /// <param name="cacheFile">The cache file.</param>
- /// <param name="probeSizeArgument">The probe size argument.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task{FFProbeResult}.</returns>
- /// <exception cref="System.ApplicationException"></exception>
- private async Task<FFProbeResult> RunFFProbeInternal(string inputPath, bool extractChapters, string cacheFile, string probeSizeArgument, CancellationToken cancellationToken)
- {
- var process = new Process
- {
- StartInfo = new ProcessStartInfo
- {
- CreateNoWindow = true,
- UseShellExecute = false,
- // Must consume both or ffmpeg may hang due to deadlocks. See comments below.
- RedirectStandardOutput = true,
- RedirectStandardError = true,
- FileName = FFProbePath,
- Arguments = string.Format("{0} -i {1} -threads 0 -v info -print_format json -show_streams -show_format", probeSizeArgument, inputPath).Trim(),
- WindowStyle = ProcessWindowStyle.Hidden,
- ErrorDialog = false
- },
- EnableRaisingEvents = true
- };
- Logger.Debug("{0} {1}", process.StartInfo.FileName, process.StartInfo.Arguments);
- process.Exited += ProcessExited;
- await FFProbeResourcePool.WaitAsync(cancellationToken).ConfigureAwait(false);
- FFProbeResult result;
- string standardError = null;
- try
- {
- process.Start();
- Task<string> standardErrorReadTask = null;
- // MUST read both stdout and stderr asynchronously or a deadlock may occurr
- if (extractChapters)
- {
- standardErrorReadTask = process.StandardError.ReadToEndAsync();
- }
- else
- {
- process.BeginErrorReadLine();
- }
- result = JsonSerializer.DeserializeFromStream<FFProbeResult>(process.StandardOutput.BaseStream);
- if (extractChapters)
- {
- standardError = await standardErrorReadTask.ConfigureAwait(false);
- }
- }
- catch
- {
- // Hate having to do this
- try
- {
- process.Kill();
- }
- catch (InvalidOperationException ex1)
- {
- Logger.ErrorException("Error killing ffprobe", ex1);
- }
- catch (Win32Exception ex1)
- {
- Logger.ErrorException("Error killing ffprobe", ex1);
- }
- throw;
- }
- finally
- {
- FFProbeResourcePool.Release();
- }
- if (result == null)
- {
- throw new ApplicationException(string.Format("FFProbe failed for {0}", inputPath));
- }
- cancellationToken.ThrowIfCancellationRequested();
- if (extractChapters && !string.IsNullOrEmpty(standardError))
- {
- AddChapters(result, standardError);
- }
- Kernel.ProtobufSerializer.SerializeToFile(result, cacheFile);
- return result;
- }
- /// <summary>
- /// Adds the chapters.
- /// </summary>
- /// <param name="result">The result.</param>
- /// <param name="standardError">The standard error.</param>
- private void AddChapters(FFProbeResult result, string standardError)
- {
- var lines = standardError.Split('\n').Select(l => l.TrimStart());
- var chapters = new List<ChapterInfo> { };
- ChapterInfo lastChapter = null;
- foreach (var line in lines)
- {
- if (line.StartsWith("Chapter", StringComparison.OrdinalIgnoreCase))
- {
- // Example:
- // Chapter #0.2: start 400.534, end 4565.435
- const string srch = "start ";
- var start = line.IndexOf(srch, StringComparison.OrdinalIgnoreCase);
- if (start == -1)
- {
- continue;
- }
- var subString = line.Substring(start + srch.Length);
- subString = subString.Substring(0, subString.IndexOf(','));
- double seconds;
- if (double.TryParse(subString, out seconds))
- {
- lastChapter = new ChapterInfo
- {
- StartPositionTicks = TimeSpan.FromSeconds(seconds).Ticks
- };
- chapters.Add(lastChapter);
- }
- }
- else if (line.StartsWith("title", StringComparison.OrdinalIgnoreCase))
- {
- if (lastChapter != null && string.IsNullOrEmpty(lastChapter.Name))
- {
- var index = line.IndexOf(':');
- if (index != -1)
- {
- lastChapter.Name = line.Substring(index + 1).Trim().TrimEnd('\r');
- }
- }
- }
- }
- result.Chapters = chapters;
- }
- /// <summary>
- /// The first chapter ticks
- /// </summary>
- private static long FirstChapterTicks = TimeSpan.FromSeconds(15).Ticks;
- /// <summary>
- /// Extracts the chapter images.
- /// </summary>
- /// <param name="video">The video.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <param name="extractImages">if set to <c>true</c> [extract images].</param>
- /// <param name="saveItem">if set to <c>true</c> [save item].</param>
- /// <returns>Task.</returns>
- /// <exception cref="System.ArgumentNullException"></exception>
- public async Task PopulateChapterImages(Video video, CancellationToken cancellationToken, bool extractImages, bool saveItem)
- {
- if (video.Chapters == null)
- {
- throw new ArgumentNullException();
- }
- var changesMade = false;
- foreach (var chapter in video.Chapters)
- {
- var filename = video.Id + "_" + video.DateModified.Ticks + "_" + chapter.StartPositionTicks;
- var path = VideoImageCache.GetResourcePath(filename, ".jpg");
- if (!VideoImageCache.ContainsFilePath(path))
- {
- if (extractImages)
- {
- // Disable for now on folder rips
- if (video.VideoType != VideoType.VideoFile)
- {
- continue;
- }
- // Add some time for the first chapter to make sure we don't end up with a black image
- var time = chapter.StartPositionTicks == 0 ? TimeSpan.FromTicks(Math.Min(FirstChapterTicks, video.RunTimeTicks ?? 0)) : TimeSpan.FromTicks(chapter.StartPositionTicks);
- var success = await ExtractImage(GetInputArgument(video), time, path, cancellationToken).ConfigureAwait(false);
- if (success)
- {
- chapter.ImagePath = path;
- changesMade = true;
- }
- }
- }
- else if (!string.Equals(path, chapter.ImagePath, StringComparison.OrdinalIgnoreCase))
- {
- chapter.ImagePath = path;
- changesMade = true;
- }
- }
- if (saveItem && changesMade)
- {
- await Kernel.ItemRepository.SaveItem(video, CancellationToken.None).ConfigureAwait(false);
- }
- }
- /// <summary>
- /// Extracts an image from an Audio file and returns a Task whose result indicates whether it was successful or not
- /// </summary>
- /// <param name="input">The input.</param>
- /// <param name="outputPath">The output path.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task{System.Boolean}.</returns>
- /// <exception cref="System.ArgumentNullException">input</exception>
- public async Task<bool> ExtractImage(Audio input, string outputPath, CancellationToken cancellationToken)
- {
- if (input == null)
- {
- throw new ArgumentNullException("input");
- }
- if (string.IsNullOrEmpty(outputPath))
- {
- throw new ArgumentNullException("outputPath");
- }
- var process = new Process
- {
- StartInfo = new ProcessStartInfo
- {
- CreateNoWindow = true,
- UseShellExecute = false,
- FileName = FFMpegPath,
- Arguments = string.Format("-i {0} -threads 0 -v quiet -f image2 \"{1}\"", GetInputArgument(input), outputPath),
- WindowStyle = ProcessWindowStyle.Hidden,
- ErrorDialog = false
- }
- };
- await AudioImageResourcePool.WaitAsync(cancellationToken).ConfigureAwait(false);
- await process.RunAsync().ConfigureAwait(false);
- AudioImageResourcePool.Release();
- var exitCode = process.ExitCode;
- process.Dispose();
- if (exitCode != -1 && File.Exists(outputPath))
- {
- return true;
- }
- Logger.Error("ffmpeg audio image extraction failed for {0}", input.Path);
- return false;
- }
- /// <summary>
- /// Determines whether [is subtitle cached] [the specified input].
- /// </summary>
- /// <param name="input">The input.</param>
- /// <param name="subtitleStreamIndex">Index of the subtitle stream.</param>
- /// <param name="outputExtension">The output extension.</param>
- /// <returns><c>true</c> if [is subtitle cached] [the specified input]; otherwise, <c>false</c>.</returns>
- public bool IsSubtitleCached(Video input, int subtitleStreamIndex, string outputExtension)
- {
- return SubtitleCache.ContainsFilePath(GetSubtitleCachePath(input, subtitleStreamIndex, outputExtension));
- }
- /// <summary>
- /// Gets the subtitle cache path.
- /// </summary>
- /// <param name="input">The input.</param>
- /// <param name="subtitleStreamIndex">Index of the subtitle stream.</param>
- /// <param name="outputExtension">The output extension.</param>
- /// <returns>System.String.</returns>
- public string GetSubtitleCachePath(Video input, int subtitleStreamIndex, string outputExtension)
- {
- return SubtitleCache.GetResourcePath(input.Id + "_" + subtitleStreamIndex + "_" + input.DateModified.Ticks, outputExtension);
- }
- /// <summary>
- /// Extracts the text subtitle.
- /// </summary>
- /// <param name="input">The input.</param>
- /// <param name="subtitleStreamIndex">Index of the subtitle stream.</param>
- /// <param name="outputPath">The output path.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task{System.Boolean}.</returns>
- /// <exception cref="System.ArgumentNullException">input</exception>
- public async Task<bool> ExtractTextSubtitle(Video input, int subtitleStreamIndex, string outputPath, CancellationToken cancellationToken)
- {
- if (input == null)
- {
- throw new ArgumentNullException("input");
- }
- if (cancellationToken == null)
- {
- throw new ArgumentNullException("cancellationToken");
- }
- var process = new Process
- {
- StartInfo = new ProcessStartInfo
- {
- CreateNoWindow = true,
- UseShellExecute = false,
- FileName = FFMpegPath,
- Arguments = string.Format("-i {0} -map 0:{1} -an -vn -c:s ass \"{2}\"", GetInputArgument(input), subtitleStreamIndex, outputPath),
- WindowStyle = ProcessWindowStyle.Hidden,
- ErrorDialog = false
- }
- };
- Logger.Debug("{0} {1}", process.StartInfo.FileName, process.StartInfo.Arguments);
- await AudioImageResourcePool.WaitAsync(cancellationToken).ConfigureAwait(false);
- await process.RunAsync().ConfigureAwait(false);
- AudioImageResourcePool.Release();
- var exitCode = process.ExitCode;
- process.Dispose();
- if (exitCode != -1 && File.Exists(outputPath))
- {
- return true;
- }
- Logger.Error("ffmpeg subtitle extraction failed for {0}", input.Path);
- return false;
- }
- /// <summary>
- /// Converts the text subtitle.
- /// </summary>
- /// <param name="mediaStream">The media stream.</param>
- /// <param name="outputPath">The output path.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task{System.Boolean}.</returns>
- /// <exception cref="System.ArgumentNullException">mediaStream</exception>
- /// <exception cref="System.ArgumentException">The given MediaStream is not an external subtitle stream</exception>
- public async Task<bool> ConvertTextSubtitle(MediaStream mediaStream, string outputPath, CancellationToken cancellationToken)
- {
- if (mediaStream == null)
- {
- throw new ArgumentNullException("mediaStream");
- }
- if (!mediaStream.IsExternal || string.IsNullOrEmpty(mediaStream.Path))
- {
- throw new ArgumentException("The given MediaStream is not an external subtitle stream");
- }
- var process = new Process
- {
- StartInfo = new ProcessStartInfo
- {
- CreateNoWindow = true,
- UseShellExecute = false,
- FileName = FFMpegPath,
- Arguments = string.Format("-i \"{0}\" \"{1}\"", mediaStream.Path, outputPath),
- WindowStyle = ProcessWindowStyle.Hidden,
- ErrorDialog = false
- }
- };
- Logger.Debug("{0} {1}", process.StartInfo.FileName, process.StartInfo.Arguments);
- await AudioImageResourcePool.WaitAsync(cancellationToken).ConfigureAwait(false);
- await process.RunAsync().ConfigureAwait(false);
- AudioImageResourcePool.Release();
- var exitCode = process.ExitCode;
- process.Dispose();
- if (exitCode != -1 && File.Exists(outputPath))
- {
- return true;
- }
- Logger.Error("ffmpeg subtitle conversion failed for {0}", mediaStream.Path);
- return false;
- }
- /// <summary>
- /// Extracts an image from a Video and returns a Task whose result indicates whether it was successful or not
- /// </summary>
- /// <param name="inputPath">The input path.</param>
- /// <param name="offset">The offset.</param>
- /// <param name="outputPath">The output path.</param>
- /// <param name="cancellationToken">The cancellation token.</param>
- /// <returns>Task{System.Boolean}.</returns>
- /// <exception cref="System.ArgumentNullException">video</exception>
- public async Task<bool> ExtractImage(string inputPath, TimeSpan offset, string outputPath, CancellationToken cancellationToken)
- {
- if (string.IsNullOrEmpty(inputPath))
- {
- throw new ArgumentNullException("inputPath");
- }
- if (string.IsNullOrEmpty(outputPath))
- {
- throw new ArgumentNullException("outputPath");
- }
- var process = new Process
- {
- StartInfo = new ProcessStartInfo
- {
- CreateNoWindow = true,
- UseShellExecute = false,
- FileName = FFMpegPath,
- Arguments = string.Format("-ss {0} -i {1} -threads 0 -v quiet -t 1 -f image2 \"{2}\"", Convert.ToInt32(offset.TotalSeconds), inputPath, outputPath),
- WindowStyle = ProcessWindowStyle.Hidden,
- ErrorDialog = false
- }
- };
- await VideoImageResourcePool.WaitAsync(cancellationToken).ConfigureAwait(false);
- process.Start();
- var ranToCompletion = process.WaitForExit(10000);
- if (!ranToCompletion)
- {
- try
- {
- Logger.Info("Killing ffmpeg process");
- process.Kill();
- process.WaitForExit(1000);
- }
- catch (Win32Exception ex)
- {
- Logger.ErrorException("Error killing process", ex);
- }
- catch (InvalidOperationException ex)
- {
- Logger.ErrorException("Error killing process", ex);
- }
- catch (NotSupportedException ex)
- {
- Logger.ErrorException("Error killing process", ex);
- }
- }
- VideoImageResourcePool.Release();
- var exitCode = ranToCompletion ? process.ExitCode : -1;
- process.Dispose();
- if (exitCode == -1)
- {
- if (File.Exists(outputPath))
- {
- try
- {
- Logger.Info("Deleting extracted image due to failure: ", outputPath);
- File.Delete(outputPath);
- }
- catch (IOException ex)
- {
- Logger.ErrorException("Error deleting extracted image {0}", ex, outputPath);
- }
- }
- }
- else
- {
- if (File.Exists(outputPath))
- {
- return true;
- }
- }
- Logger.Error("ffmpeg video image extraction failed for {0}", inputPath);
- return false;
- }
- /// <summary>
- /// Gets the input argument.
- /// </summary>
- /// <param name="item">The item.</param>
- /// <returns>System.String.</returns>
- public string GetInputArgument(BaseItem item)
- {
- var video = item as Video;
- if (video != null)
- {
- if (video.VideoType == VideoType.BluRay)
- {
- return GetBlurayInputArgument(video.Path);
- }
- if (video.VideoType == VideoType.Dvd)
- {
- return GetDvdInputArgument(video.GetPlayableStreamFiles());
- }
- }
- return string.Format("file:\"{0}\"", item.Path);
- }
- /// <summary>
- /// Gets the input argument.
- /// </summary>
- /// <param name="item">The item.</param>
- /// <param name="mount">The mount.</param>
- /// <returns>System.String.</returns>
- public string GetInputArgument(Video item, IIsoMount mount)
- {
- if (item.VideoType == VideoType.Iso && item.IsoType.HasValue)
- {
- if (item.IsoType.Value == IsoType.BluRay)
- {
- return GetBlurayInputArgument(mount.MountedPath);
- }
- if (item.IsoType.Value == IsoType.Dvd)
- {
- return GetDvdInputArgument(item.GetPlayableStreamFiles(mount.MountedPath));
- }
- }
- return GetInputArgument(item);
- }
- /// <summary>
- /// Gets the bluray input argument.
- /// </summary>
- /// <param name="blurayRoot">The bluray root.</param>
- /// <returns>System.String.</returns>
- public string GetBlurayInputArgument(string blurayRoot)
- {
- return string.Format("bluray:\"{0}\"", blurayRoot);
- }
- /// <summary>
- /// Gets the DVD input argument.
- /// </summary>
- /// <param name="playableStreamFiles">The playable stream files.</param>
- /// <returns>System.String.</returns>
- public string GetDvdInputArgument(IEnumerable<string> playableStreamFiles)
- {
- // Get all streams
- var streamFilePaths = (playableStreamFiles ?? new string[] { }).ToArray();
- // If there's more than one we'll need to use the concat command
- if (streamFilePaths.Length > 1)
- {
- var files = string.Join("|", streamFilePaths);
- return string.Format("concat:\"{0}\"", files);
- }
- // Determine the input path for video files
- return string.Format("file:\"{0}\"", streamFilePaths[0]);
- }
- /// <summary>
- /// Processes the exited.
- /// </summary>
- /// <param name="sender">The sender.</param>
- /// <param name="e">The <see cref="EventArgs" /> instance containing the event data.</param>
- void ProcessExited(object sender, EventArgs e)
- {
- ((Process)sender).Dispose();
- }
- /// <summary>
- /// Sets the error mode.
- /// </summary>
- /// <param name="uMode">The u mode.</param>
- /// <returns>ErrorModes.</returns>
- [DllImport("kernel32.dll")]
- static extern ErrorModes SetErrorMode(ErrorModes uMode);
- /// <summary>
- /// Enum ErrorModes
- /// </summary>
- [Flags]
- public enum ErrorModes : uint
- {
- /// <summary>
- /// The SYSTE m_ DEFAULT
- /// </summary>
- SYSTEM_DEFAULT = 0x0,
- /// <summary>
- /// The SE m_ FAILCRITICALERRORS
- /// </summary>
- SEM_FAILCRITICALERRORS = 0x0001,
- /// <summary>
- /// The SE m_ NOALIGNMENTFAULTEXCEPT
- /// </summary>
- SEM_NOALIGNMENTFAULTEXCEPT = 0x0004,
- /// <summary>
- /// The SE m_ NOGPFAULTERRORBOX
- /// </summary>
- SEM_NOGPFAULTERRORBOX = 0x0002,
- /// <summary>
- /// The SE m_ NOOPENFILEERRORBOX
- /// </summary>
- SEM_NOOPENFILEERRORBOX = 0x8000
- }
- }
- }
|