You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
jellyfin/MediaBrowser.Controller/MediaEncoding/EncodingJobInfo.cs

697 lines
20 KiB

#nullable disable
#pragma warning disable CS1591, SA1401
using System;
6 years ago
using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Text.Json.Serialization;
using Jellyfin.Data.Entities;
6 years ago
using MediaBrowser.Model.Dlna;
using MediaBrowser.Model.Drawing;
6 years ago
using MediaBrowser.Model.Dto;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.MediaInfo;
using MediaBrowser.Model.Net;
using MediaBrowser.Model.Session;
6 years ago
namespace MediaBrowser.Controller.MediaEncoding
{
// For now, a common base class until the API and MediaEncoding classes are unified
public class EncodingJobInfo
6 years ago
{
public int? OutputAudioBitrate;
public int? OutputAudioChannels;
private TranscodeReason? _transcodeReasons = null;
public EncodingJobInfo(TranscodingJobType jobType)
{
TranscodingType = jobType;
RemoteHttpHeaders = new Dictionary<string, string>(StringComparer.OrdinalIgnoreCase);
SupportedAudioCodecs = Array.Empty<string>();
SupportedVideoCodecs = Array.Empty<string>();
SupportedSubtitleCodecs = Array.Empty<string>();
}
public TranscodeReason[] TranscodeReasons { get => TranscodeReason.ToArray(); }
[JsonIgnore]
public TranscodeReason TranscodeReason
{
get
{
if (!_transcodeReasons.HasValue)
{
if (BaseRequest.TranscodeReasons == null)
{
_transcodeReasons = TranscodeReason.None;
return TranscodeReason.None;
}
TranscodeReason reason = TranscodeReason.None;
Enum.TryParse<TranscodeReason>(BaseRequest.TranscodeReasons, out reason);
_transcodeReasons = reason;
}
return _transcodeReasons.Value;
}
}
public IProgress<double> Progress { get; set; }
6 years ago
public MediaStream VideoStream { get; set; }
6 years ago
public VideoType VideoType { get; set; }
6 years ago
public Dictionary<string, string> RemoteHttpHeaders { get; set; }
6 years ago
public string OutputVideoCodec { get; set; }
6 years ago
public MediaProtocol InputProtocol { get; set; }
6 years ago
public string MediaPath { get; set; }
6 years ago
public bool IsInputVideo { get; set; }
6 years ago
public string OutputAudioCodec { get; set; }
6 years ago
public int? OutputVideoBitrate { get; set; }
6 years ago
public MediaStream SubtitleStream { get; set; }
6 years ago
public SubtitleDeliveryMethod SubtitleDeliveryMethod { get; set; }
6 years ago
public string[] SupportedSubtitleCodecs { get; set; }
public int InternalSubtitleStreamOffset { get; set; }
6 years ago
public MediaSourceInfo MediaSource { get; set; }
public User User { get; set; }
6 years ago
public long? RunTimeTicks { get; set; }
public bool ReadInputAtNativeFramerate { get; set; }
public string OutputFilePath { get; set; }
public string MimeType { get; set; }
public bool IgnoreInputDts => MediaSource.IgnoreDts;
6 years ago
public bool IgnoreInputIndex => MediaSource.IgnoreIndex;
6 years ago
public bool GenPtsInput => MediaSource.GenPtsInput;
6 years ago
public bool DiscardCorruptFramesInput => false;
6 years ago
public bool EnableFastSeekInput => false;
6 years ago
public bool GenPtsOutput => false;
6 years ago
public string OutputContainer { get; set; }
public string OutputVideoSync { get; set; }
6 years ago
public string AlbumCoverPath { get; set; }
public string InputAudioSync { get; set; }
6 years ago
public string InputVideoSync { get; set; }
6 years ago
public TransportStreamTimestamp InputTimestamp { get; set; }
public MediaStream AudioStream { get; set; }
6 years ago
public string[] SupportedAudioCodecs { get; set; }
6 years ago
public string[] SupportedVideoCodecs { get; set; }
6 years ago
public string InputContainer { get; set; }
6 years ago
public IsoType? IsoType { get; set; }
public BaseEncodingJobOptions BaseRequest { get; set; }
public bool IsVideoRequest { get; set; }
6 years ago
public TranscodingJobType TranscodingType { get; set; }
public long? StartTimeTicks => BaseRequest.StartTimeTicks;
public bool CopyTimestamps => BaseRequest.CopyTimestamps;
6 years ago
public bool IsSegmentedLiveStream
=> TranscodingType != TranscodingJobType.Progressive && !RunTimeTicks.HasValue;
6 years ago
public int? TotalOutputBitrate => (OutputAudioBitrate ?? 0) + (OutputVideoBitrate ?? 0);
6 years ago
public int? OutputWidth
{
get
{
if (VideoStream != null && VideoStream.Width.HasValue && VideoStream.Height.HasValue)
{
var size = new ImageDimensions(VideoStream.Width.Value, VideoStream.Height.Value);
6 years ago
var newSize = DrawingUtils.Resize(
size,
6 years ago
BaseRequest.Width ?? 0,
BaseRequest.Height ?? 0,
BaseRequest.MaxWidth ?? 0,
BaseRequest.MaxHeight ?? 0);
return newSize.Width;
6 years ago
}
if (!IsVideoRequest)
{
return null;
}
return BaseRequest.MaxWidth ?? BaseRequest.Width;
}
}
public int? OutputHeight
{
get
{
if (VideoStream != null && VideoStream.Width.HasValue && VideoStream.Height.HasValue)
{
var size = new ImageDimensions(VideoStream.Width.Value, VideoStream.Height.Value);
6 years ago
var newSize = DrawingUtils.Resize(
size,
6 years ago
BaseRequest.Width ?? 0,
BaseRequest.Height ?? 0,
BaseRequest.MaxWidth ?? 0,
BaseRequest.MaxHeight ?? 0);
return newSize.Height;
6 years ago
}
if (!IsVideoRequest)
{
return null;
}
return BaseRequest.MaxHeight ?? BaseRequest.Height;
}
}
public int? OutputAudioSampleRate
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputAudioCodec))
6 years ago
{
if (AudioStream != null)
{
return AudioStream.SampleRate;
}
}
else if (BaseRequest.AudioSampleRate.HasValue)
{
// Don't exceed what the encoder supports
// Seeing issues of attempting to encode to 88200
return BaseRequest.AudioSampleRate.Value;
6 years ago
}
return null;
}
}
public int? OutputAudioBitDepth
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputAudioCodec))
6 years ago
{
if (AudioStream != null)
{
return AudioStream.BitDepth;
}
}
return null;
}
}
/// <summary>
/// Gets the target video level.
6 years ago
/// </summary>
public double? TargetVideoLevel
{
get
{
if (BaseRequest.Static || EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.Level;
6 years ago
}
var level = GetRequestedLevel(ActualOutputVideoCodec);
if (!string.IsNullOrEmpty(level)
&& double.TryParse(level, NumberStyles.Any, CultureInfo.InvariantCulture, out var result))
6 years ago
{
return result;
}
return null;
}
}
/// <summary>
/// Gets the target video bit depth.
6 years ago
/// </summary>
public int? TargetVideoBitDepth
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.BitDepth;
6 years ago
}
return null;
}
}
/// <summary>
/// Gets the target reference frames.
/// </summary>
/// <value>The target reference frames.</value>
public int? TargetRefFrames
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.RefFrames;
6 years ago
}
return null;
}
}
/// <summary>
/// Gets the target framerate.
6 years ago
/// </summary>
public float? TargetFramerate
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream == null ? null : (VideoStream.AverageFrameRate ?? VideoStream.RealFrameRate);
}
return BaseRequest.MaxFramerate ?? BaseRequest.Framerate;
}
}
public TransportStreamTimestamp TargetTimestamp
{
get
{
if (BaseRequest.Static)
{
return InputTimestamp;
}
return string.Equals(OutputContainer, "m2ts", StringComparison.OrdinalIgnoreCase) ?
6 years ago
TransportStreamTimestamp.Valid :
TransportStreamTimestamp.None;
}
}
/// <summary>
/// Gets the target packet length.
6 years ago
/// </summary>
public int? TargetPacketLength
{
get
{
if (BaseRequest.Static || EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.PacketLength;
6 years ago
}
return null;
}
}
/// <summary>
/// Gets the target video profile.
6 years ago
/// </summary>
public string TargetVideoProfile
{
get
{
if (BaseRequest.Static || EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.Profile;
6 years ago
}
var requestedProfile = GetRequestedProfiles(ActualOutputVideoCodec).FirstOrDefault();
if (!string.IsNullOrEmpty(requestedProfile))
{
return requestedProfile;
}
return null;
}
}
public string TargetVideoCodecTag
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.CodecTag;
6 years ago
}
return null;
}
}
public bool? IsTargetAnamorphic
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.IsAnamorphic;
6 years ago
}
return false;
}
}
public string ActualOutputVideoCodec
{
get
{
if (VideoStream == null)
{
return null;
}
if (EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream.Codec;
6 years ago
}
return OutputVideoCodec;
6 years ago
}
}
public string ActualOutputAudioCodec
{
get
{
if (AudioStream == null)
{
return null;
}
if (EncodingHelper.IsCopyCodec(OutputAudioCodec))
{
return AudioStream.Codec;
}
return OutputAudioCodec;
}
}
6 years ago
public bool? IsTargetInterlaced
{
get
{
if (BaseRequest.Static
|| EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.IsInterlaced;
6 years ago
}
if (DeInterlace(ActualOutputVideoCodec, true))
{
return false;
}
return VideoStream?.IsInterlaced;
6 years ago
}
}
public bool? IsTargetAVC
{
get
{
if (BaseRequest.Static || EncodingHelper.IsCopyCodec(OutputVideoCodec))
6 years ago
{
return VideoStream?.IsAVC;
6 years ago
}
return false;
}
}
public int? TargetVideoStreamCount
{
get
{
if (BaseRequest.Static)
{
return GetMediaStreamCount(MediaStreamType.Video, int.MaxValue);
}
6 years ago
return GetMediaStreamCount(MediaStreamType.Video, 1);
}
}
public int? TargetAudioStreamCount
{
get
{
if (BaseRequest.Static)
{
return GetMediaStreamCount(MediaStreamType.Audio, int.MaxValue);
}
6 years ago
return GetMediaStreamCount(MediaStreamType.Audio, 1);
}
}
public int HlsListSize => 0;
public bool EnableBreakOnNonKeyFrames(string videoCodec)
{
if (TranscodingType != TranscodingJobType.Progressive)
{
if (IsSegmentedLiveStream)
{
return false;
}
return BaseRequest.BreakOnNonKeyFrames && EncodingHelper.IsCopyCodec(videoCodec);
}
return false;
}
6 years ago
private int? GetMediaStreamCount(MediaStreamType type, int limit)
{
var count = MediaSource.GetStreamCount(type);
if (count.HasValue)
{
count = Math.Min(count.Value, limit);
}
return count;
}
public string GetMimeType(string outputPath, bool enableStreamDefault = true)
{
if (!string.IsNullOrEmpty(MimeType))
{
return MimeType;
}
4 years ago
if (enableStreamDefault)
{
return MimeTypes.GetMimeType(outputPath);
}
return MimeTypes.GetMimeType(outputPath, null);
}
public bool DeInterlace(string videoCodec, bool forceDeinterlaceIfSourceIsInterlaced)
{
var videoStream = VideoStream;
var isInputInterlaced = videoStream != null && videoStream.IsInterlaced;
if (!isInputInterlaced)
{
return false;
}
// Support general param
if (BaseRequest.DeInterlace)
{
return true;
}
if (!string.IsNullOrEmpty(videoCodec))
{
if (string.Equals(BaseRequest.GetOption(videoCodec, "deinterlace"), "true", StringComparison.OrdinalIgnoreCase))
{
return true;
}
}
return forceDeinterlaceIfSourceIsInterlaced;
}
public string[] GetRequestedProfiles(string codec)
{
if (!string.IsNullOrEmpty(BaseRequest.Profile))
{
return BaseRequest.Profile.Split(new[] { '|', ',' }, StringSplitOptions.RemoveEmptyEntries);
}
if (!string.IsNullOrEmpty(codec))
{
var profile = BaseRequest.GetOption(codec, "profile");
if (!string.IsNullOrEmpty(profile))
{
return profile.Split(new[] { '|', ',' }, StringSplitOptions.RemoveEmptyEntries);
}
}
return Array.Empty<string>();
}
public string GetRequestedLevel(string codec)
{
if (!string.IsNullOrEmpty(BaseRequest.Level))
{
return BaseRequest.Level;
}
if (!string.IsNullOrEmpty(codec))
{
return BaseRequest.GetOption(codec, "level");
}
return null;
}
public int? GetRequestedMaxRefFrames(string codec)
{
if (BaseRequest.MaxRefFrames.HasValue)
{
return BaseRequest.MaxRefFrames;
}
if (!string.IsNullOrEmpty(codec))
{
var value = BaseRequest.GetOption(codec, "maxrefframes");
if (!string.IsNullOrEmpty(value)
&& int.TryParse(value, NumberStyles.Any, CultureInfo.InvariantCulture, out var result))
{
return result;
}
}
return null;
}
public int? GetRequestedVideoBitDepth(string codec)
{
if (BaseRequest.MaxVideoBitDepth.HasValue)
{
return BaseRequest.MaxVideoBitDepth;
}
if (!string.IsNullOrEmpty(codec))
{
var value = BaseRequest.GetOption(codec, "videobitdepth");
if (!string.IsNullOrEmpty(value)
&& int.TryParse(value, NumberStyles.Any, CultureInfo.InvariantCulture, out var result))
{
return result;
}
}
return null;
}
public int? GetRequestedAudioBitDepth(string codec)
{
if (BaseRequest.MaxAudioBitDepth.HasValue)
{
return BaseRequest.MaxAudioBitDepth;
}
if (!string.IsNullOrEmpty(codec))
{
var value = BaseRequest.GetOption(codec, "audiobitdepth");
if (!string.IsNullOrEmpty(value)
&& int.TryParse(value, NumberStyles.Any, CultureInfo.InvariantCulture, out var result))
{
return result;
}
}
return null;
}
public int? GetRequestedAudioChannels(string codec)
{
if (!string.IsNullOrEmpty(codec))
{
var value = BaseRequest.GetOption(codec, "audiochannels");
if (!string.IsNullOrEmpty(value)
&& int.TryParse(value, NumberStyles.Any, CultureInfo.InvariantCulture, out var result))
{
return result;
}
}
if (BaseRequest.MaxAudioChannels.HasValue)
{
return BaseRequest.MaxAudioChannels;
}
if (BaseRequest.AudioChannels.HasValue)
{
return BaseRequest.AudioChannels;
}
if (BaseRequest.TranscodingMaxAudioChannels.HasValue)
{
return BaseRequest.TranscodingMaxAudioChannels;
}
return null;
}
public virtual void ReportTranscodingProgress(TimeSpan? transcodingPosition, float? framerate, double? percentComplete, long? bytesTranscoded, int? bitRate)
{
6 years ago
Progress.Report(percentComplete.Value);
}
}
}