1079 lines
38 KiB
C#
1079 lines
38 KiB
C#
|
using Ionic.Zip;
|
|||
|
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 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>
|
|||
|
/// Initializes a new instance of the <see cref="FFMpegManager" /> class.
|
|||
|
/// </summary>
|
|||
|
/// <param name="kernel">The kernel.</param>
|
|||
|
public FFMpegManager(Kernel kernel)
|
|||
|
: base(kernel)
|
|||
|
{
|
|||
|
// 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))
|
|||
|
{
|
|||
|
using (var zipFile = ZipFile.Read(resourceStream))
|
|||
|
{
|
|||
|
zipFile.ExtractAll(targetPath, ExtractExistingFileAction.DoNotOverwrite);
|
|||
|
}
|
|||
|
}
|
|||
|
}
|
|||
|
|
|||
|
/// <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
|
|||
|
}
|
|||
|
}
|
|||
|
}
|