jellyfin/MediaBrowser.Controller/MediaInfo/FFMpegManager.cs

217 lines
8.1 KiB
C#
Raw Normal View History

2013-09-04 17:02:19 +00:00
using MediaBrowser.Common.IO;
using MediaBrowser.Common.MediaInfo;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.IO;
2013-06-18 19:16:27 +00:00
using MediaBrowser.Controller.Persistence;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging;
2013-02-21 01:33:05 +00:00
using System;
2013-09-04 17:02:19 +00:00
using System.Collections.Generic;
2013-02-21 01:33:05 +00:00
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
namespace MediaBrowser.Controller.MediaInfo
{
/// <summary>
/// Class FFMpegManager
/// </summary>
public class FFMpegManager
2013-02-21 01:33:05 +00:00
{
/// <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 subtitle cache.
/// </summary>
/// <value>The subtitle cache.</value>
internal FileSystemRepository SubtitleCache { get; set; }
2013-02-21 05:00:56 +00:00
2013-03-04 05:43:06 +00:00
private readonly IServerApplicationPaths _appPaths;
private readonly IMediaEncoder _encoder;
2013-04-27 22:52:41 +00:00
private readonly ILogger _logger;
2013-06-18 19:16:27 +00:00
private readonly IItemRepository _itemRepo;
2013-04-27 22:52:41 +00:00
private readonly IFileSystem _fileSystem;
2013-02-21 01:33:05 +00:00
/// <summary>
/// Initializes a new instance of the <see cref="FFMpegManager" /> class.
/// </summary>
/// <param name="appPaths">The app paths.</param>
/// <param name="encoder">The encoder.</param>
2013-04-27 22:52:41 +00:00
/// <param name="logger">The logger.</param>
2013-06-18 19:16:27 +00:00
/// <param name="itemRepo">The item repo.</param>
2013-02-21 05:00:56 +00:00
/// <exception cref="System.ArgumentNullException">zipClient</exception>
public FFMpegManager(IServerApplicationPaths appPaths, IMediaEncoder encoder, ILogger logger, IItemRepository itemRepo, IFileSystem fileSystem)
2013-02-21 01:33:05 +00:00
{
2013-03-04 05:43:06 +00:00
_appPaths = appPaths;
_encoder = encoder;
2013-04-27 22:52:41 +00:00
_logger = logger;
2013-06-18 19:16:27 +00:00
_itemRepo = itemRepo;
_fileSystem = fileSystem;
2013-02-21 01:33:05 +00:00
VideoImageCache = new FileSystemRepository(VideoImagesDataPath);
SubtitleCache = new FileSystemRepository(SubtitleCachePath);
}
/// <summary>
/// Gets the video images data path.
/// </summary>
/// <value>The video images data path.</value>
public string VideoImagesDataPath
{
get
{
2013-06-04 02:02:49 +00:00
return Path.Combine(_appPaths.DataPath, "extracted-video-images");
2013-02-21 01:33:05 +00:00
}
}
/// <summary>
/// Gets the audio images data path.
/// </summary>
/// <value>The audio images data path.</value>
public string AudioImagesDataPath
{
get
{
2013-06-04 02:02:49 +00:00
return Path.Combine(_appPaths.DataPath, "extracted-audio-images");
2013-02-21 01:33:05 +00:00
}
}
/// <summary>
/// Gets the subtitle cache path.
/// </summary>
/// <value>The subtitle cache path.</value>
public string SubtitleCachePath
{
get
{
2013-06-04 02:02:49 +00:00
return Path.Combine(_appPaths.CachePath, "subtitles");
2013-02-21 01:33:05 +00:00
}
}
2013-04-27 22:52:41 +00:00
2013-02-21 01:33:05 +00:00
/// <summary>
/// The first chapter ticks
/// </summary>
private static readonly long FirstChapterTicks = TimeSpan.FromSeconds(15).Ticks;
2013-02-21 01:33:05 +00:00
/// <summary>
/// Extracts the chapter images.
/// </summary>
/// <param name="video">The video.</param>
2013-06-18 19:16:27 +00:00
/// <param name="chapters">The chapters.</param>
2013-02-21 01:33:05 +00:00
/// <param name="extractImages">if set to <c>true</c> [extract images].</param>
2013-06-18 19:16:27 +00:00
/// <param name="saveChapters">if set to <c>true</c> [save chapters].</param>
/// <param name="cancellationToken">The cancellation token.</param>
2013-02-21 01:33:05 +00:00
/// <returns>Task.</returns>
/// <exception cref="System.ArgumentNullException"></exception>
2013-06-18 19:16:27 +00:00
public async Task<bool> PopulateChapterImages(Video video, List<ChapterInfo> chapters, bool extractImages, bool saveChapters, CancellationToken cancellationToken)
2013-02-21 01:33:05 +00:00
{
// Can't extract images if there are no video streams
if (video.MediaStreams == null || video.MediaStreams.All(m => m.Type != MediaStreamType.Video))
{
return true;
}
var success = true;
2013-02-21 01:33:05 +00:00
var changesMade = false;
var runtimeTicks = video.RunTimeTicks ?? 0;
2013-06-18 19:16:27 +00:00
foreach (var chapter in chapters)
2013-02-21 01:33:05 +00:00
{
if (chapter.StartPositionTicks >= runtimeTicks)
{
_logger.Info("Stopping chapter extraction for {0} because a chapter was found with a position greater than the runtime.", video.Name);
break;
}
var filename = video.Path + "_" + video.DateModified.Ticks + "_" + chapter.StartPositionTicks;
2013-02-21 01:33:05 +00:00
var path = VideoImageCache.GetResourcePath(filename, ".jpg");
2013-06-04 02:02:49 +00:00
if (!File.Exists(path))
2013-02-21 01:33:05 +00:00
{
if (extractImages)
{
if (video.VideoType == VideoType.HdDvd || video.VideoType == VideoType.Iso)
2013-02-21 01:33:05 +00:00
{
continue;
}
if (video.VideoType == VideoType.BluRay)
{
// Can only extract reliably on single file blurays
if (video.PlayableStreamFileNames == null || video.PlayableStreamFileNames.Count != 1)
{
continue;
}
}
2013-02-21 01:33:05 +00:00
// 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);
InputType type;
2013-02-21 01:33:05 +00:00
var inputPath = MediaEncoderHelpers.GetInputArgument(video, null, out type);
try
2013-02-21 01:33:05 +00:00
{
2013-06-04 02:02:49 +00:00
var parentPath = Path.GetDirectoryName(path);
Directory.CreateDirectory(parentPath);
2013-06-04 02:02:49 +00:00
await _encoder.ExtractImage(inputPath, type, video.Video3DFormat, time, path, cancellationToken).ConfigureAwait(false);
2013-02-21 01:33:05 +00:00
chapter.ImagePath = path;
changesMade = true;
}
catch
{
success = false;
break;
}
2013-02-21 01:33:05 +00:00
}
}
else if (!string.Equals(path, chapter.ImagePath, StringComparison.OrdinalIgnoreCase))
{
chapter.ImagePath = path;
changesMade = true;
}
}
2013-06-18 19:16:27 +00:00
if (saveChapters && changesMade)
2013-02-21 01:33:05 +00:00
{
2013-06-18 19:16:27 +00:00
await _itemRepo.SaveChapters(video.Id, chapters, cancellationToken).ConfigureAwait(false);
2013-02-21 01:33:05 +00:00
}
return success;
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Gets the subtitle cache path.
/// </summary>
/// <param name="input">The input.</param>
/// <param name="subtitleStreamIndex">Index of the subtitle stream.</param>
2013-04-24 00:25:49 +00:00
/// <param name="offset">The offset.</param>
2013-02-21 01:33:05 +00:00
/// <param name="outputExtension">The output extension.</param>
/// <returns>System.String.</returns>
2013-04-24 00:25:49 +00:00
public string GetSubtitleCachePath(Video input, int subtitleStreamIndex, TimeSpan? offset, string outputExtension)
2013-02-21 01:33:05 +00:00
{
2013-04-24 00:25:49 +00:00
var ticksParam = offset.HasValue ? "_" + offset.Value.Ticks : "";
var stream = input.MediaStreams[subtitleStreamIndex];
if (stream.IsExternal)
{
ticksParam += _fileSystem.GetLastWriteTimeUtc(stream.Path).Ticks;
}
2013-04-24 00:25:49 +00:00
return SubtitleCache.GetResourcePath(input.Id + "_" + subtitleStreamIndex + "_" + input.DateModified.Ticks + ticksParam, outputExtension);
2013-02-21 01:33:05 +00:00
}
}
}