2016-03-27 16:35:25 -04:00
|
|
|
|
using MediaBrowser.Controller.Chapters;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
using MediaBrowser.Controller.Entities;
|
2013-12-18 00:44:46 -05:00
|
|
|
|
using MediaBrowser.Controller.Entities.Movies;
|
|
|
|
|
using MediaBrowser.Controller.Entities.TV;
|
2014-02-20 11:37:41 -05:00
|
|
|
|
using MediaBrowser.Controller.MediaEncoding;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
using MediaBrowser.Model.Entities;
|
2013-05-03 00:10:11 -04:00
|
|
|
|
using MediaBrowser.Model.Logging;
|
2014-06-29 13:35:05 -04:00
|
|
|
|
using MediaBrowser.Model.MediaInfo;
|
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;
|
2015-10-04 00:23:11 -04:00
|
|
|
|
using CommonIO;
|
2016-08-29 17:06:24 -04:00
|
|
|
|
using MediaBrowser.Controller.Library;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
2014-02-20 11:37:41 -05:00
|
|
|
|
namespace MediaBrowser.Server.Implementations.MediaEncoder
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2014-02-20 11:37:41 -05:00
|
|
|
|
public class EncodingManager : IEncodingManager
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2014-02-20 11:37:41 -05:00
|
|
|
|
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
|
|
|
|
|
private readonly IFileSystem _fileSystem;
|
2013-04-27 18:52:41 -04:00
|
|
|
|
private readonly ILogger _logger;
|
2014-02-20 11:37:41 -05:00
|
|
|
|
private readonly IMediaEncoder _encoder;
|
2014-06-10 13:36:06 -04:00
|
|
|
|
private readonly IChapterManager _chapterManager;
|
2016-08-29 17:06:24 -04:00
|
|
|
|
private readonly ILibraryManager _libraryManager;
|
2013-04-27 18:52:41 -04:00
|
|
|
|
|
2014-09-28 11:27:26 -04:00
|
|
|
|
public EncodingManager(IFileSystem fileSystem,
|
2014-06-11 10:42:03 -04:00
|
|
|
|
ILogger logger,
|
|
|
|
|
IMediaEncoder encoder,
|
2016-08-30 02:06:24 -04:00
|
|
|
|
IChapterManager chapterManager, ILibraryManager libraryManager)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2014-02-20 11:37:41 -05:00
|
|
|
|
_fileSystem = fileSystem;
|
2013-04-27 18:52:41 -04:00
|
|
|
|
_logger = logger;
|
2014-02-20 11:37:41 -05:00
|
|
|
|
_encoder = encoder;
|
2014-06-10 13:36:06 -04:00
|
|
|
|
_chapterManager = chapterManager;
|
2016-08-30 02:06:24 -04:00
|
|
|
|
_libraryManager = libraryManager;
|
2014-02-20 11:37:41 -05:00
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
|
|
|
|
/// <summary>
|
2014-02-20 11:37:41 -05:00
|
|
|
|
/// Gets the chapter images data path.
|
2013-02-20 20:33:05 -05:00
|
|
|
|
/// </summary>
|
2014-02-20 11:37:41 -05:00
|
|
|
|
/// <value>The chapter images data path.</value>
|
2014-09-28 11:27:26 -04:00
|
|
|
|
private string GetChapterImagesPath(IHasImages item)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2014-09-28 11:27:26 -04:00
|
|
|
|
return Path.Combine(item.GetInternalMetadataPath(), "chapters");
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
2013-12-05 22:39:44 -05:00
|
|
|
|
|
2013-12-18 00:44:46 -05:00
|
|
|
|
/// <summary>
|
|
|
|
|
/// Determines whether [is eligible for chapter image extraction] [the specified video].
|
|
|
|
|
/// </summary>
|
|
|
|
|
/// <param name="video">The video.</param>
|
|
|
|
|
/// <returns><c>true</c> if [is eligible for chapter image extraction] [the specified video]; otherwise, <c>false</c>.</returns>
|
|
|
|
|
private bool IsEligibleForChapterImageExtraction(Video video)
|
|
|
|
|
{
|
2014-03-03 00:11:03 -05:00
|
|
|
|
if (video.IsPlaceHolder)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
2016-08-29 17:06:24 -04:00
|
|
|
|
var libraryOptions = _libraryManager.GetLibraryOptions(video);
|
|
|
|
|
if (libraryOptions != null && libraryOptions.SchemaVersion >= 2)
|
2013-12-18 00:44:46 -05:00
|
|
|
|
{
|
2016-08-29 17:06:24 -04:00
|
|
|
|
if (!libraryOptions.EnableChapterImageExtraction)
|
2013-12-18 00:44:46 -05:00
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
2016-08-29 17:06:24 -04:00
|
|
|
|
else
|
2013-12-18 00:44:46 -05:00
|
|
|
|
{
|
2016-08-29 17:06:24 -04:00
|
|
|
|
var options = _chapterManager.GetConfiguration();
|
|
|
|
|
|
|
|
|
|
if (video is Movie)
|
2013-12-18 00:44:46 -05:00
|
|
|
|
{
|
2016-08-29 17:06:24 -04:00
|
|
|
|
if (!options.EnableMovieChapterImageExtraction)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2013-12-18 00:44:46 -05:00
|
|
|
|
}
|
2016-08-29 17:06:24 -04:00
|
|
|
|
else if (video is Episode)
|
2013-12-18 00:44:46 -05:00
|
|
|
|
{
|
2016-08-29 17:06:24 -04:00
|
|
|
|
if (!options.EnableEpisodeChapterImageExtraction)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
if (!options.EnableOtherVideoChapterImageExtraction)
|
|
|
|
|
{
|
|
|
|
|
return false;
|
|
|
|
|
}
|
2013-12-18 00:44:46 -05:00
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Can't extract images if there are no video streams
|
|
|
|
|
return video.DefaultVideoStreamIndex.HasValue;
|
|
|
|
|
}
|
|
|
|
|
|
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
|
|
|
|
|
2014-02-20 11:37:41 -05:00
|
|
|
|
public async Task<bool> RefreshChapterImages(ChapterImageRefreshOptions options, CancellationToken cancellationToken)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2014-02-20 11:37:41 -05:00
|
|
|
|
var extractImages = options.ExtractImages;
|
|
|
|
|
var video = options.Video;
|
|
|
|
|
var chapters = options.Chapters;
|
|
|
|
|
var saveChapters = options.SaveChapters;
|
|
|
|
|
|
2013-12-18 00:44:46 -05:00
|
|
|
|
if (!IsEligibleForChapterImageExtraction(video))
|
2013-03-13 21:24:43 -04:00
|
|
|
|
{
|
2014-01-30 23:50:09 -05:00
|
|
|
|
extractImages = false;
|
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)
|
|
|
|
|
{
|
2016-08-28 12:46:28 -04:00
|
|
|
|
if (video.VideoType == VideoType.HdDvd || video.VideoType == VideoType.Iso)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
2016-08-28 12:46:28 -04:00
|
|
|
|
if (video.VideoType == VideoType.BluRay || video.VideoType == VideoType.Dvd)
|
|
|
|
|
{
|
|
|
|
|
if (video.PlayableStreamFileNames.Count != 1)
|
|
|
|
|
{
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
2016-09-05 16:07:36 -04:00
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
// 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-02-20 20:33:05 -05:00
|
|
|
|
|
2016-09-05 16:07:36 -04:00
|
|
|
|
var protocol = MediaProtocol.File;
|
2013-02-20 20:33:05 -05:00
|
|
|
|
|
2016-09-05 16:07:36 -04:00
|
|
|
|
var inputPath = MediaEncoderHelpers.GetInputArgument(_fileSystem, video.Path, protocol, null, video.PlayableStreamFileNames);
|
2013-04-07 16:55:05 -04:00
|
|
|
|
|
2016-09-05 16:07:36 -04:00
|
|
|
|
_fileSystem.CreateDirectory(Path.GetDirectoryName(path));
|
2013-12-05 22:39:44 -05:00
|
|
|
|
|
2016-09-30 14:43:59 -04:00
|
|
|
|
var container = video.Container;
|
|
|
|
|
|
|
|
|
|
var tempFile = await _encoder.ExtractVideoImage(inputPath, container, protocol, video.Video3DFormat, time, cancellationToken).ConfigureAwait(false);
|
2016-06-30 22:35:18 -04:00
|
|
|
|
File.Copy(tempFile, path, true);
|
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
|
|
|
|
File.Delete(tempFile);
|
|
|
|
|
}
|
|
|
|
|
catch
|
2014-02-06 18:57:21 -05:00
|
|
|
|
{
|
2016-06-30 22:35:18 -04:00
|
|
|
|
|
2014-02-06 18:57:21 -05:00
|
|
|
|
}
|
|
|
|
|
|
2013-02-20 20:33:05 -05:00
|
|
|
|
chapter.ImagePath = path;
|
2016-07-06 13:44:44 -04:00
|
|
|
|
chapter.ImageDateModified = _fileSystem.GetLastWriteTimeUtc(path);
|
2013-02-20 20:33:05 -05:00
|
|
|
|
changesMade = true;
|
|
|
|
|
}
|
2015-04-11 13:59:55 -04:00
|
|
|
|
catch (Exception ex)
|
2013-04-06 20:27:33 -04:00
|
|
|
|
{
|
2016-09-05 16:07:36 -04:00
|
|
|
|
_logger.ErrorException("Error extracting chapter images for {0}", ex, string.Join(",", video.Path));
|
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
|
|
|
|
}
|
2014-01-30 23:50:09 -05:00
|
|
|
|
else if (!string.IsNullOrEmpty(chapter.ImagePath))
|
|
|
|
|
{
|
|
|
|
|
chapter.ImagePath = null;
|
|
|
|
|
changesMade = true;
|
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
else if (!string.Equals(path, chapter.ImagePath, StringComparison.OrdinalIgnoreCase))
|
|
|
|
|
{
|
|
|
|
|
chapter.ImagePath = path;
|
2016-07-06 13:44:44 -04:00
|
|
|
|
chapter.ImageDateModified = _fileSystem.GetLastWriteTimeUtc(path);
|
2013-02-20 20:33:05 -05:00
|
|
|
|
changesMade = true;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
2013-06-18 15:16:27 -04:00
|
|
|
|
if (saveChapters && changesMade)
|
2013-02-20 20:33:05 -05:00
|
|
|
|
{
|
2014-06-10 13:36:06 -04:00
|
|
|
|
await _chapterManager.SaveChapters(video.Id.ToString(), 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
|
|
|
|
}
|
|
|
|
|
|
2014-02-20 11:37:41 -05:00
|
|
|
|
private string GetChapterImagePath(Video video, long chapterPositionTicks)
|
|
|
|
|
{
|
|
|
|
|
var filename = video.DateModified.Ticks.ToString(_usCulture) + "_" + chapterPositionTicks.ToString(_usCulture) + ".jpg";
|
|
|
|
|
|
2014-09-28 11:27:26 -04:00
|
|
|
|
return Path.Combine(GetChapterImagesPath(video), filename);
|
2014-02-20 11:37:41 -05:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private List<string> GetSavedChapterImages(Video video)
|
|
|
|
|
{
|
2014-09-28 11:27:26 -04:00
|
|
|
|
var path = GetChapterImagesPath(video);
|
2014-02-20 11:37:41 -05:00
|
|
|
|
|
|
|
|
|
try
|
|
|
|
|
{
|
2015-09-24 13:50:49 -04:00
|
|
|
|
return _fileSystem.GetFilePaths(path)
|
2014-02-20 11:37:41 -05:00
|
|
|
|
.ToList();
|
|
|
|
|
}
|
|
|
|
|
catch (DirectoryNotFoundException)
|
|
|
|
|
{
|
|
|
|
|
return new List<string>();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
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
|
|
|
|
|
{
|
2015-01-12 22:46:44 -05:00
|
|
|
|
_fileSystem.DeleteFile(image);
|
2013-12-15 11:53:32 -05:00
|
|
|
|
}
|
|
|
|
|
catch (IOException ex)
|
|
|
|
|
{
|
|
|
|
|
_logger.ErrorException("Error deleting {0}.", ex, image);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
2013-02-20 20:33:05 -05:00
|
|
|
|
}
|
|
|
|
|
}
|