jellyfin/Emby.Server.Implementations/MediaEncoder/EncodingManager.cs

240 lines
8.6 KiB
C#
Raw Normal View History

2013-02-21 01:33:05 +00:00
using System;
2013-09-04 17:02:19 +00:00
using System.Collections.Generic;
2013-12-15 18:29:34 +00:00
using System.Globalization;
2013-02-21 01:33:05 +00:00
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using MediaBrowser.Controller.Chapters;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.MediaEncoding;
using MediaBrowser.Controller.Providers;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.IO;
using MediaBrowser.Model.MediaInfo;
using Microsoft.Extensions.Logging;
2013-02-21 01:33:05 +00:00
namespace Emby.Server.Implementations.MediaEncoder
2013-02-21 01:33:05 +00:00
{
2014-02-20 16:37:41 +00:00
public class EncodingManager : IEncodingManager
2013-02-21 01:33:05 +00:00
{
2014-02-20 16:37:41 +00:00
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
private readonly IFileSystem _fileSystem;
2013-04-27 22:52:41 +00:00
private readonly ILogger _logger;
2014-02-20 16:37:41 +00:00
private readonly IMediaEncoder _encoder;
2014-06-10 17:36:06 +00:00
private readonly IChapterManager _chapterManager;
private readonly ILibraryManager _libraryManager;
2013-04-27 22:52:41 +00:00
public EncodingManager(
IFileSystem fileSystem,
ILoggerFactory loggerFactory,
2017-08-05 19:02:33 +00:00
IMediaEncoder encoder,
2016-08-30 06:06:24 +00:00
IChapterManager chapterManager, ILibraryManager libraryManager)
2013-02-21 01:33:05 +00:00
{
2014-02-20 16:37:41 +00:00
_fileSystem = fileSystem;
_logger = loggerFactory.CreateLogger(nameof(EncodingManager));
2014-02-20 16:37:41 +00:00
_encoder = encoder;
2014-06-10 17:36:06 +00:00
_chapterManager = chapterManager;
2016-08-30 06:06:24 +00:00
_libraryManager = libraryManager;
2014-02-20 16:37:41 +00:00
}
2013-02-21 01:33:05 +00:00
/// <summary>
2014-02-20 16:37:41 +00:00
/// Gets the chapter images data path.
2013-02-21 01:33:05 +00:00
/// </summary>
2014-02-20 16:37:41 +00:00
/// <value>The chapter images data path.</value>
private static string GetChapterImagesPath(BaseItem item)
2013-02-21 01:33:05 +00:00
{
2014-09-28 15:27:26 +00:00
return Path.Combine(item.GetInternalMetadataPath(), "chapters");
2013-02-21 01:33:05 +00:00
}
2013-12-06 03:39:44 +00:00
2013-12-18 05:44:46 +00: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)
{
if (video.IsPlaceHolder)
{
return false;
}
var libraryOptions = _libraryManager.GetLibraryOptions(video);
if (libraryOptions != null)
2013-12-18 05:44:46 +00:00
{
if (!libraryOptions.EnableChapterImageExtraction)
2013-12-18 05:44:46 +00:00
{
return false;
}
}
else
2013-12-18 05:44:46 +00:00
{
return false;
2013-12-18 05:44:46 +00:00
}
if (video.VideoType == VideoType.Iso)
{
return false;
}
if (video.VideoType == VideoType.BluRay || video.VideoType == VideoType.Dvd)
{
return false;
}
if (video.IsShortcut)
{
return false;
}
2017-08-26 19:50:02 +00:00
if (!video.IsCompleteMedia)
{
return false;
}
2013-12-18 05:44:46 +00:00
// Can't extract images if there are no video streams
return video.DefaultVideoStreamIndex.HasValue;
}
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
public async Task<bool> RefreshChapterImages(Video video, IDirectoryService directoryService, List<ChapterInfo> chapters, bool extractImages, bool saveChapters, CancellationToken cancellationToken)
2013-02-21 01:33:05 +00:00
{
2013-12-18 05:44:46 +00:00
if (!IsEligibleForChapterImageExtraction(video))
{
2014-01-31 04:50:09 +00:00
extractImages = false;
}
var success = true;
2013-02-21 01:33:05 +00:00
var changesMade = false;
var runtimeTicks = video.RunTimeTicks ?? 0;
var currentImages = GetSavedChapterImages(video, directoryService);
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.LogInformation("Stopping chapter extraction for {0} because a chapter was found with a position greater than the runtime.", video.Name);
break;
}
var path = GetChapterImagePath(video, chapter.StartPositionTicks);
2013-02-21 01:33:05 +00:00
if (!currentImages.Contains(path, StringComparer.OrdinalIgnoreCase))
2013-02-21 01:33:05 +00:00
{
if (extractImages)
{
2018-09-12 17:26:21 +00:00
cancellationToken.ThrowIfCancellationRequested();
2016-09-05 20:07:36 +00: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-21 01:33:05 +00:00
2016-09-05 20:07:36 +00:00
var protocol = MediaProtocol.File;
2013-02-21 01:33:05 +00:00
2018-09-12 17:26:21 +00:00
var inputPath = MediaEncoderHelpers.GetInputArgument(_fileSystem, video.Path, protocol, null, Array.Empty<string>());
Directory.CreateDirectory(Path.GetDirectoryName(path));
2013-12-06 03:39:44 +00:00
2016-09-30 18:43:59 +00:00
var container = video.Container;
var tempFile = await _encoder.ExtractVideoImage(inputPath, container, protocol, video.GetDefaultVideoStream(), video.Video3DFormat, time, cancellationToken).ConfigureAwait(false);
2019-01-26 21:31:59 +00:00
File.Copy(tempFile, path, true);
try
{
_fileSystem.DeleteFile(tempFile);
}
catch
{
2017-08-05 19:02:33 +00:00
}
2013-02-21 01:33:05 +00:00
chapter.ImagePath = path;
2016-07-06 17:44:44 +00:00
chapter.ImageDateModified = _fileSystem.GetLastWriteTimeUtc(path);
2013-02-21 01:33:05 +00:00
changesMade = true;
}
2015-04-11 17:59:55 +00:00
catch (Exception ex)
{
2018-12-20 12:11:26 +00:00
_logger.LogError(ex, "Error extracting chapter images for {0}", string.Join(",", video.Path));
success = false;
break;
}
2013-02-21 01:33:05 +00:00
}
2014-01-31 04:50:09 +00:00
else if (!string.IsNullOrEmpty(chapter.ImagePath))
{
chapter.ImagePath = null;
changesMade = true;
}
2013-02-21 01:33:05 +00:00
}
else if (!string.Equals(path, chapter.ImagePath, StringComparison.OrdinalIgnoreCase))
{
chapter.ImagePath = path;
2016-07-06 17:44:44 +00:00
chapter.ImageDateModified = _fileSystem.GetLastWriteTimeUtc(path);
2013-02-21 01:33:05 +00:00
changesMade = true;
}
}
2013-06-18 19:16:27 +00:00
if (saveChapters && changesMade)
2013-02-21 01:33:05 +00:00
{
_chapterManager.SaveChapters(video.Id.ToString(), chapters);
2013-02-21 01:33:05 +00:00
}
DeleteDeadImages(currentImages, chapters);
return success;
2013-02-21 01:33:05 +00:00
}
2014-02-20 16:37:41 +00:00
private string GetChapterImagePath(Video video, long chapterPositionTicks)
{
var filename = video.DateModified.Ticks.ToString(_usCulture) + "_" + chapterPositionTicks.ToString(_usCulture) + ".jpg";
2014-09-28 15:27:26 +00:00
return Path.Combine(GetChapterImagesPath(video), filename);
2014-02-20 16:37:41 +00:00
}
private static List<string> GetSavedChapterImages(Video video, IDirectoryService directoryService)
2014-02-20 16:37:41 +00:00
{
2014-09-28 15:27:26 +00:00
var path = GetChapterImagesPath(video);
2014-02-20 16:37:41 +00:00
try
{
return directoryService.GetFilePaths(path)
2014-02-20 16:37:41 +00:00
.ToList();
}
catch (IOException)
2014-02-20 16:37:41 +00:00
{
return new List<string>();
}
}
private void DeleteDeadImages(IEnumerable<string> images, IEnumerable<ChapterInfo> chapters)
{
var deadImages = images
2013-12-15 17:01:56 +00:00
.Except(chapters.Select(i => i.ImagePath).Where(i => !string.IsNullOrEmpty(i)), StringComparer.OrdinalIgnoreCase)
.Where(i => BaseItem.SupportedImageExtensions.Contains(Path.GetExtension(i), StringComparer.OrdinalIgnoreCase))
.ToList();
foreach (var image in deadImages)
{
2018-12-20 12:11:26 +00:00
_logger.LogDebug("Deleting dead chapter image {path}", image);
try
{
_fileSystem.DeleteFile(image);
}
catch (IOException ex)
{
2018-12-20 12:11:26 +00:00
_logger.LogError(ex, "Error deleting {path}.", image);
}
}
}
2013-02-21 01:33:05 +00:00
}
}