Trickplay generation, manager, storage

MediaBrowser.Controller/MediaEncoding/EncodingHelper.cs

@@ -149,6 +149,36 @@ namespace MediaBrowser.Controller.MediaEncoding
             return defaultEncoder;
         }
 
+        private string GetMjpegEncoder(EncodingJobInfo state, EncodingOptions encodingOptions)
+        {
+            var defaultEncoder = "mjpeg";
+
+            if (state.VideoType == VideoType.VideoFile)
+            {
+                var hwType = encodingOptions.HardwareAccelerationType;
+
+                var codecMap = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase)
+                {
+                    { "vaapi",                  defaultEncoder + "_vaapi" },
+                    { "qsv",                defaultEncoder + "_qsv" }
+                };
+
+                if (!string.IsNullOrEmpty(hwType)
+                    && encodingOptions.EnableHardwareEncoding
+                    && codecMap.ContainsKey(hwType))
+                {
+                    var preferredEncoder = codecMap[hwType];
+
+                    if (_mediaEncoder.SupportsEncoder(preferredEncoder))
+                    {
+                        return preferredEncoder;
+                    }
+                }
+            }
+
+            return defaultEncoder;
+        }
+
         private bool IsVaapiSupported(EncodingJobInfo state)
         {
             // vaapi will throw an error with this input
@@ -277,6 +307,11 @@ namespace MediaBrowser.Controller.MediaEncoding
                     return GetH264Encoder(state, encodingOptions);
                 }
 
+                if (string.Equals(codec, "mjpeg", StringComparison.OrdinalIgnoreCase))
+                {
+                    return GetMjpegEncoder(state, encodingOptions);
+                }
+
                 if (string.Equals(codec, "vp8", StringComparison.OrdinalIgnoreCase)
                     || string.Equals(codec, "vpx", StringComparison.OrdinalIgnoreCase))
                 {

MediaBrowser.Controller/MediaEncoding/IMediaEncoder.cs

@@ -6,6 +6,7 @@ using System;
 using System.Collections.Generic;
 using System.Threading;
 using System.Threading.Tasks;
+using MediaBrowser.Model.Configuration;
 using MediaBrowser.Model.Dlna;
 using MediaBrowser.Model.Drawing;
 using MediaBrowser.Model.Dto;
@@ -137,6 +138,32 @@ namespace MediaBrowser.Controller.MediaEncoding
         /// <returns>Location of video image.</returns>
         Task<string> ExtractVideoImage(string inputFile, string container, MediaSourceInfo mediaSource, MediaStream imageStream, int? imageStreamIndex, ImageFormat? targetFormat, CancellationToken cancellationToken);
 
+        /// <summary>
+        /// Extracts the video images on interval.
+        /// </summary>
+        /// <param name="inputFile">Input file.</param>
+        /// <param name="container">Video container type.</param>
+        /// <param name="mediaSource">Media source information.</param>
+        /// <param name="imageStream">Media stream information.</param>
+        /// <param name="interval">The interval.</param>
+        /// <param name="maxWidth">The maximum width.</param>
+        /// <param name="allowHwAccel">Allow for hardware acceleration.</param>
+        /// <param name="allowHwEncode">Allow for hardware encoding. allowHwAccel must also be true.</param>
+        /// <param name="encodingHelper">EncodingHelper instance.</param>
+        /// <param name="cancellationToken">The cancellation token.</param>
+        /// <returns>Directory where images where extracted. A given image made before another will always be named with a lower number.</returns>
+        Task<string> ExtractVideoImagesOnIntervalAccelerated(
+            string inputFile,
+            string container,
+            MediaSourceInfo mediaSource,
+            MediaStream imageStream,
+            TimeSpan interval,
+            int maxWidth,
+            bool allowHwAccel,
+            bool allowHwEncode,
+            EncodingHelper encodingHelper,
+            CancellationToken cancellationToken);
+
         /// <summary>
         /// Gets the media info.
         /// </summary>

MediaBrowser.Controller/Persistence/IItemRepository.cs

@@ -61,6 +61,27 @@ namespace MediaBrowser.Controller.Persistence
         /// <param name="chapters">The list of chapters to save.</param>
         void SaveChapters(Guid id, IReadOnlyList<ChapterInfo> chapters);
 
+        /// <summary>
+        /// Get available trickplay resolutions and corresponding info.
+        /// </summary>
+        /// <param name="itemId">The item.</param>
+        /// <returns>Map of width resolutions to trickplay tiles info.</returns>
+        Dictionary<int, TrickplayTilesInfo> GetTilesResolutions(Guid itemId);
+
+        /// <summary>
+        /// Saves trickplay tiles info.
+        /// </summary>
+        /// <param name="itemId">The item.</param>
+        /// <param name="tilesInfo">The trickplay tiles info.</param>
+        void SaveTilesInfo(Guid itemId, TrickplayTilesInfo tilesInfo);
+
+        /// <summary>
+        /// Gets trickplay data for an item.
+        /// </summary>
+        /// <param name="item">The item.</param>
+        /// <returns>A map of media source id to a map of tile width to tile info.</returns>
+        Dictionary<Guid, Dictionary<int, TrickplayTilesInfo>> GetTrickplayManifest(BaseItem item);
+
         /// <summary>
         /// Gets the media streams.
         /// </summary>

MediaBrowser.Controller/Trickplay/ITrickplayManager.cs

@@ -0,0 +1,54 @@
+using System;
+using System.Collections.Generic;
+using System.Threading;
+using System.Threading.Tasks;
+using MediaBrowser.Controller.Entities;
+using MediaBrowser.Model.Entities;
+
+namespace MediaBrowser.Controller.Trickplay
+{
+    /// <summary>
+    /// Interface ITrickplayManager.
+    /// </summary>
+    public interface ITrickplayManager
+    {
+        /// <summary>
+        /// Generate or replace trickplay data.
+        /// </summary>
+        /// <param name="video">The video.</param>
+        /// <param name="replace">Whether or not existing data should be replaced.</param>
+        /// <param name="cancellationToken">CancellationToken to use for operation.</param>
+        /// <returns>Task.</returns>
+        Task RefreshTrickplayData(Video video, bool replace, CancellationToken cancellationToken);
+
+        /// <summary>
+        /// Get available trickplay resolutions and corresponding info.
+        /// </summary>
+        /// <param name="itemId">The item.</param>
+        /// <returns>Map of width resolutions to trickplay tiles info.</returns>
+        Dictionary<int, TrickplayTilesInfo> GetTilesResolutions(Guid itemId);
+
+        /// <summary>
+        /// Saves trickplay tiles info.
+        /// </summary>
+        /// <param name="itemId">The item.</param>
+        /// <param name="tilesInfo">The trickplay tiles info.</param>
+        void SaveTilesInfo(Guid itemId, TrickplayTilesInfo tilesInfo);
+
+        /// <summary>
+        /// Gets the trickplay manifest.
+        /// </summary>
+        /// <param name="item">The item.</param>
+        /// <returns>A map of media source id to a map of tile width to tile info.</returns>
+        Dictionary<Guid, Dictionary<int, TrickplayTilesInfo>> GetTrickplayManifest(BaseItem item);
+
+        /// <summary>
+        /// Gets the path to a trickplay tiles image.
+        /// </summary>
+        /// <param name="item">The item.</param>
+        /// <param name="width">The width of a single tile.</param>
+        /// <param name="index">The tile grid's index.</param>
+        /// <returns>The absolute path.</returns>
+        string GetTrickplayTilePath(BaseItem item, int width, int index);
+    }
+}