2013-12-15 13:29:34 -05:00
|
|
|
|
using MediaBrowser.Common.Extensions;
|
2013-12-15 11:53:32 -05:00
|
|
|
|
using MediaBrowser.Common.IO;
|
2013-04-07 16:55:05 -04:00
|
|
|
|
using MediaBrowser.Common.MediaInfo;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
using MediaBrowser.Controller.Entities;
|
2013-06-18 15:16:27 -04:00
|
|
|
|
using MediaBrowser.Controller.Persistence;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
using MediaBrowser.Model.Entities;
|
2013-05-03 00:10:11 -04:00
|
|
|
|
using MediaBrowser.Model.Logging;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
using System;
|
2013-09-04 13:02:19 -04:00
|
|
|
|
using System.Collections.Generic;
|
2013-12-15 13:29:34 -05:00
|
|
|
|
using System.Globalization;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
using System.IO;
|
|
|
|
|
using System.Linq;
|
|
|
|
|
using System.Threading;
|
|
|
|
|
using System.Threading.Tasks;
|
|
|
|
|
|
|
|
|
|
namespace MediaBrowser.Controller.MediaInfo
|
|
|
|
|
{
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Class FFMpegManager
|
|
|
|
|
/// </summary>
|
2013-04-07 16:55:05 -04:00
|
|
|
|
public class FFMpegManager
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-03-04 00:43:06 -05:00
|
|
|
|
private readonly IServerApplicationPaths _appPaths;
|
2013-04-07 16:55:05 -04:00
|
|
|
|
private readonly IMediaEncoder _encoder;
|
2013-04-27 18:52:41 -04:00
|
|
|
|
private readonly ILogger _logger;
|
2013-06-18 15:16:27 -04:00
|
|
|
|
private readonly IItemRepository _itemRepo;
|
2013-04-27 18:52:41 -04:00
|
|
|
|
|
2013-10-31 10:03:23 -04:00
|
|
|
|
private readonly IFileSystem _fileSystem;
|
|
|
|
|
|
2013-12-15 13:29:34 -05:00
|
|
|
|
public static FFMpegManager Instance { get; private set; }
|
|
|
|
|
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// Initializes a new instance of the <see cref="FFMpegManager" /> class.
|
|
|
|
|
/// </summary>
|
2013-04-07 16:55:05 -04:00
|
|
|
|
/// <param name="appPaths">The app paths.</param>
|
|
|
|
|
/// <param name="encoder">The encoder.</param>
|
2013-04-27 18:52:41 -04:00
|
|
|
|
/// <param name="logger">The logger.</param>
|
2013-06-18 15:16:27 -04:00
|
|
|
|
/// <param name="itemRepo">The item repo.</param>
|
2013-02-21 00:00:56 -05:00
|
|
|
|
/// <exception cref="System.ArgumentNullException">zipClient</exception>
|
2013-10-31 10:03:23 -04:00
|
|
|
|
public FFMpegManager(IServerApplicationPaths appPaths, IMediaEncoder encoder, ILogger logger, IItemRepository itemRepo, IFileSystem fileSystem)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-03-04 00:43:06 -05:00
|
|
|
|
_appPaths = appPaths;
|
2013-04-07 16:55:05 -04:00
|
|
|
|
_encoder = encoder;
|
2013-04-27 18:52:41 -04:00
|
|
|
|
_logger = logger;
|
2013-06-18 15:16:27 -04:00
|
|
|
|
_itemRepo = itemRepo;
|
2013-10-31 10:03:23 -04:00
|
|
|
|
_fileSystem = fileSystem;
|
2013-12-15 13:29:34 -05:00
|
|
|
|
|
|
|
|
|
// TODO: Remove this static instance
|
|
|
|
|
Instance = this;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
2013-12-15 11:53:32 -05:00
|
|
|
|
/// Gets the chapter images data path.
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// </summary>
|
2013-12-15 11:53:32 -05:00
|
|
|
|
/// <value>The chapter images data path.</value>
|
|
|
|
|
public string ChapterImagesPath
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2013-12-15 11:53:32 -05:00
|
|
|
|
return Path.Combine(_appPaths.DataPath, "chapter-images");
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Gets the subtitle cache path.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <value>The subtitle cache path.</value>
|
2013-12-15 11:53:32 -05:00
|
|
|
|
private string SubtitleCachePath
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
|
|
|
|
get
|
|
|
|
|
{
|
2013-06-03 22:02:49 -04:00
|
|
|
|
return Path.Combine(_appPaths.CachePath, "subtitles");
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
}
|
2013-12-05 22:39:44 -05:00
|
|
|
|
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// The first chapter ticks
|
|
|
|
|
/// </summary>
|
2013-04-07 16:55:05 -04:00
|
|
|
|
private static readonly long FirstChapterTicks = TimeSpan.FromSeconds(15).Ticks;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
|
|
|
|
/// <summary>
|
|
|
|
|
/// Extracts the chapter images.
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="video">The video.</param>
|
2013-06-18 15:16:27 -04:00
|
|
|
|
/// <param name="chapters">The chapters.</param>
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// <param name="extractImages">if set to <c>true</c> [extract images].</param>
|
2013-06-18 15:16:27 -04:00
|
|
|
|
/// <param name="saveChapters">if set to <c>true</c> [save chapters].</param>
|
|
|
|
|
/// <param name="cancellationToken">The cancellation token.</param>
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// <returns>Task.</returns>
|
|
|
|
|
/// <exception cref="System.ArgumentNullException"></exception>
|
2013-06-18 15:16:27 -04:00
|
|
|
|
public async Task<bool> PopulateChapterImages(Video video, List<ChapterInfo> chapters, bool extractImages, bool saveChapters, CancellationToken cancellationToken)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-03-13 21:24:43 -04:00
|
|
|
|
// Can't extract images if there are no video streams
|
2013-12-05 22:39:44 -05:00
|
|
|
|
if (!video.DefaultVideoStreamIndex.HasValue)
|
2013-03-13 21:24:43 -04:00
|
|
|
|
{
|
2013-05-25 01:17:32 -04:00
|
|
|
|
return true;
|
2013-03-13 21:24:43 -04:00
|
|
|
|
}
|
|
|
|
|
|
2013-05-25 01:17:32 -04:00
|
|
|
|
var success = true;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
var changesMade = false;
|
|
|
|
|
|
2013-06-01 09:56:48 -04:00
|
|
|
|
var runtimeTicks = video.RunTimeTicks ?? 0;
|
|
|
|
|
|
2013-12-15 11:53:32 -05:00
|
|
|
|
var currentImages = GetSavedChapterImages(video);
|
|
|
|
|
|
2013-06-18 15:16:27 -04:00
|
|
|
|
foreach (var chapter in chapters)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-06-01 09:56:48 -04: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;
|
|
|
|
|
}
|
|
|
|
|
|
2013-12-15 11:53:32 -05:00
|
|
|
|
var path = GetChapterImagePath(video, chapter.StartPositionTicks);
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
2013-12-15 11:53:32 -05:00
|
|
|
|
if (!currentImages.Contains(path, StringComparer.OrdinalIgnoreCase))
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
|
|
|
|
if (extractImages)
|
|
|
|
|
{
|
2013-04-09 22:44:58 -04:00
|
|
|
|
if (video.VideoType == VideoType.HdDvd || video.VideoType == VideoType.Iso)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
2013-04-09 22:44:58 -04:00
|
|
|
|
if (video.VideoType == VideoType.BluRay)
|
|
|
|
|
{
|
|
|
|
|
// Can only extract reliably on single file blurays
|
|
|
|
|
if (video.PlayableStreamFileNames == null || video.PlayableStreamFileNames.Count != 1)
|
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-02-20 20:33:05 -05: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);
|
|
|
|
|
|
2013-04-07 16:55:05 -04:00
|
|
|
|
InputType type;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
2013-04-07 16:55:05 -04:00
|
|
|
|
var inputPath = MediaEncoderHelpers.GetInputArgument(video, null, out type);
|
|
|
|
|
|
|
|
|
|
try
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-06-03 22:02:49 -04:00
|
|
|
|
var parentPath = Path.GetDirectoryName(path);
|
|
|
|
|
|
2013-10-01 14:24:27 -04:00
|
|
|
|
Directory.CreateDirectory(parentPath);
|
2013-12-05 22:39:44 -05:00
|
|
|
|
|
2013-06-27 11:59:32 -04:00
|
|
|
|
await _encoder.ExtractImage(inputPath, type, video.Video3DFormat, time, path, cancellationToken).ConfigureAwait(false);
|
2013-02-20 20:33:05 -05:00
|
|
|
|
chapter.ImagePath = path;
|
|
|
|
|
changesMade = true;
|
|
|
|
|
}
|
2013-04-07 16:55:05 -04:00
|
|
|
|
catch
|
2013-04-06 20:27:33 -04:00
|
|
|
|
{
|
2013-05-25 01:17:32 -04:00
|
|
|
|
success = false;
|
2013-04-06 20:27:33 -04:00
|
|
|
|
break;
|
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else if (!string.Equals(path, chapter.ImagePath, StringComparison.OrdinalIgnoreCase))
|
|
|
|
|
{
|
|
|
|
|
chapter.ImagePath = path;
|
|
|
|
|
changesMade = true;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-06-18 15:16:27 -04:00
|
|
|
|
if (saveChapters && changesMade)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-06-18 15:16:27 -04:00
|
|
|
|
await _itemRepo.SaveChapters(video.Id, chapters, cancellationToken).ConfigureAwait(false);
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
2013-05-25 01:17:32 -04:00
|
|
|
|
|
2013-12-15 11:53:32 -05:00
|
|
|
|
DeleteDeadImages(currentImages, chapters);
|
|
|
|
|
|
2013-05-25 01:17:32 -04:00
|
|
|
|
return success;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
|
2013-12-15 11:53:32 -05:00
|
|
|
|
private void DeleteDeadImages(IEnumerable<string> images, IEnumerable<ChapterInfo> chapters)
|
|
|
|
|
{
|
|
|
|
|
var deadImages = images
|
2013-12-15 12:01:56 -05:00
|
|
|
|
.Except(chapters.Select(i => i.ImagePath).Where(i => !string.IsNullOrEmpty(i)), StringComparer.OrdinalIgnoreCase)
|
2013-12-15 11:53:32 -05:00
|
|
|
|
.Where(i => BaseItem.SupportedImageExtensions.Contains(Path.GetExtension(i), StringComparer.OrdinalIgnoreCase))
|
|
|
|
|
.ToList();
|
|
|
|
|
|
|
|
|
|
foreach (var image in deadImages)
|
|
|
|
|
{
|
|
|
|
|
_logger.Debug("Deleting dead chapter image {0}", image);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
File.Delete(image);
|
|
|
|
|
}
|
|
|
|
|
catch (IOException ex)
|
|
|
|
|
{
|
|
|
|
|
_logger.ErrorException("Error deleting {0}.", ex, image);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-12-15 12:01:56 -05:00
|
|
|
|
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
|
2013-12-15 11:53:32 -05:00
|
|
|
|
|
2013-02-20 20:33:05 -05: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-23 20:25:49 -04:00
|
|
|
|
/// <param name="offset">The offset.</param>
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// <param name="outputExtension">The output extension.</param>
|
|
|
|
|
/// <returns>System.String.</returns>
|
2013-04-23 20:25:49 -04:00
|
|
|
|
public string GetSubtitleCachePath(Video input, int subtitleStreamIndex, TimeSpan? offset, string outputExtension)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2013-04-23 20:25:49 -04:00
|
|
|
|
var ticksParam = offset.HasValue ? "_" + offset.Value.Ticks : "";
|
|
|
|
|
|
2013-12-05 22:39:44 -05:00
|
|
|
|
var stream = _itemRepo.GetMediaStreams(new MediaStreamQuery
|
|
|
|
|
{
|
|
|
|
|
ItemId = input.Id,
|
|
|
|
|
Index = subtitleStreamIndex
|
|
|
|
|
|
|
|
|
|
}).FirstOrDefault();
|
|
|
|
|
|
|
|
|
|
if (stream == null)
|
|
|
|
|
{
|
|
|
|
|
return null;
|
|
|
|
|
}
|
2013-05-09 10:47:28 -04:00
|
|
|
|
|
|
|
|
|
if (stream.IsExternal)
|
|
|
|
|
{
|
2013-10-31 10:03:23 -04:00
|
|
|
|
ticksParam += _fileSystem.GetLastWriteTimeUtc(stream.Path).Ticks;
|
2013-05-09 10:47:28 -04:00
|
|
|
|
}
|
|
|
|
|
|
2013-12-15 12:01:56 -05:00
|
|
|
|
var filename = (input.Id + "_" + subtitleStreamIndex.ToString(_usCulture) + "_" + input.DateModified.Ticks.ToString(_usCulture) + ticksParam).GetMD5() + outputExtension;
|
2013-12-15 11:53:32 -05:00
|
|
|
|
|
|
|
|
|
var prefix = filename.Substring(0, 1);
|
|
|
|
|
|
|
|
|
|
return Path.Combine(SubtitleCachePath, prefix, filename);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public string GetChapterImagePath(Video video, long chapterPositionTicks)
|
|
|
|
|
{
|
2013-12-15 12:01:56 -05:00
|
|
|
|
var filename = video.DateModified.Ticks.ToString(_usCulture) + "_" + chapterPositionTicks.ToString(_usCulture) + ".jpg";
|
2013-12-15 11:53:32 -05:00
|
|
|
|
|
|
|
|
|
var videoId = video.Id.ToString();
|
|
|
|
|
var prefix = videoId.Substring(0, 1);
|
|
|
|
|
|
|
|
|
|
return Path.Combine(ChapterImagesPath, prefix, videoId, filename);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public List<string> GetSavedChapterImages(Video video)
|
|
|
|
|
{
|
|
|
|
|
var videoId = video.Id.ToString();
|
|
|
|
|
var prefix = videoId.Substring(0, 1);
|
|
|
|
|
|
|
|
|
|
var path = Path.Combine(ChapterImagesPath, prefix, videoId);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
return Directory.EnumerateFiles(path)
|
|
|
|
|
.ToList();
|
|
|
|
|
}
|
|
|
|
|
catch (DirectoryNotFoundException)
|
|
|
|
|
{
|
|
|
|
|
return new List<string>();
|
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|