| | 1 | | #nullable disable |
| | 2 | |
|
| | 3 | | #pragma warning disable CS1591 |
| | 4 | | // We need lowercase normalized string for ffmpeg |
| | 5 | | #pragma warning disable CA1308 |
| | 6 | |
|
| | 7 | | using System; |
| | 8 | | using System.Collections.Generic; |
| | 9 | | using System.Globalization; |
| | 10 | | using System.IO; |
| | 11 | | using System.Linq; |
| | 12 | | using System.Runtime.InteropServices; |
| | 13 | | using System.Text; |
| | 14 | | using System.Text.RegularExpressions; |
| | 15 | | using System.Threading; |
| | 16 | | using Jellyfin.Data.Enums; |
| | 17 | | using Jellyfin.Extensions; |
| | 18 | | using MediaBrowser.Common.Configuration; |
| | 19 | | using MediaBrowser.Controller.Extensions; |
| | 20 | | using MediaBrowser.Model.Configuration; |
| | 21 | | using MediaBrowser.Model.Dlna; |
| | 22 | | using MediaBrowser.Model.Dto; |
| | 23 | | using MediaBrowser.Model.Entities; |
| | 24 | | using MediaBrowser.Model.MediaInfo; |
| | 25 | | using Microsoft.Extensions.Configuration; |
| | 26 | | using IConfigurationManager = MediaBrowser.Common.Configuration.IConfigurationManager; |
| | 27 | |
|
| | 28 | | namespace MediaBrowser.Controller.MediaEncoding |
| | 29 | | { |
| | 30 | | public partial class EncodingHelper |
| | 31 | | { |
| | 32 | | /// <summary> |
| | 33 | | /// The codec validation regex. |
| | 34 | | /// This regular expression matches strings that consist of alphanumeric characters, hyphens, |
| | 35 | | /// periods, underscores, commas, and vertical bars, with a length between 0 and 40 characters. |
| | 36 | | /// This should matches all common valid codecs. |
| | 37 | | /// </summary> |
| | 38 | | public const string ValidationRegex = @"^[a-zA-Z0-9\-\._,|]{0,40}$"; |
| | 39 | |
|
| | 40 | | private const string _defaultMjpegEncoder = "mjpeg"; |
| | 41 | |
|
| | 42 | | private const string QsvAlias = "qs"; |
| | 43 | | private const string VaapiAlias = "va"; |
| | 44 | | private const string D3d11vaAlias = "dx11"; |
| | 45 | | private const string VideotoolboxAlias = "vt"; |
| | 46 | | private const string RkmppAlias = "rk"; |
| | 47 | | private const string OpenclAlias = "ocl"; |
| | 48 | | private const string CudaAlias = "cu"; |
| | 49 | | private const string DrmAlias = "dr"; |
| | 50 | | private const string VulkanAlias = "vk"; |
| | 51 | | private readonly IApplicationPaths _appPaths; |
| | 52 | | private readonly IMediaEncoder _mediaEncoder; |
| | 53 | | private readonly ISubtitleEncoder _subtitleEncoder; |
| | 54 | | private readonly IConfiguration _config; |
| | 55 | | private readonly IConfigurationManager _configurationManager; |
| | 56 | |
|
| | 57 | | // i915 hang was fixed by linux 6.2 (3f882f2) |
| 22 | 58 | | private readonly Version _minKerneli915Hang = new Version(5, 18); |
| 22 | 59 | | private readonly Version _maxKerneli915Hang = new Version(6, 1, 3); |
| 22 | 60 | | private readonly Version _minFixedKernel60i915Hang = new Version(6, 0, 18); |
| 22 | 61 | | private readonly Version _minKernelVersionAmdVkFmtModifier = new Version(5, 15); |
| | 62 | |
|
| 22 | 63 | | private readonly Version _minFFmpegImplictHwaccel = new Version(6, 0); |
| 22 | 64 | | private readonly Version _minFFmpegHwaUnsafeOutput = new Version(6, 0); |
| 22 | 65 | | private readonly Version _minFFmpegOclCuTonemapMode = new Version(5, 1, 3); |
| 22 | 66 | | private readonly Version _minFFmpegSvtAv1Params = new Version(5, 1); |
| 22 | 67 | | private readonly Version _minFFmpegVaapiH26xEncA53CcSei = new Version(6, 0); |
| 22 | 68 | | private readonly Version _minFFmpegReadrateOption = new Version(5, 0); |
| 22 | 69 | | private readonly Version _minFFmpegWorkingVtHwSurface = new Version(7, 0, 1); |
| 22 | 70 | | private readonly Version _minFFmpegDisplayRotationOption = new Version(6, 0); |
| 22 | 71 | | private readonly Version _minFFmpegAdvancedTonemapMode = new Version(7, 0, 1); |
| 22 | 72 | | private readonly Version _minFFmpegAlteredVaVkInterop = new Version(7, 0, 1); |
| 22 | 73 | | private readonly Version _minFFmpegQsvVppTonemapOption = new Version(7, 0, 1); |
| | 74 | |
|
| 0 | 75 | | private static readonly Regex _validationRegex = new(ValidationRegex, RegexOptions.Compiled); |
| | 76 | |
|
| 0 | 77 | | private static readonly string[] _videoProfilesH264 = |
| 0 | 78 | | [ |
| 0 | 79 | | "ConstrainedBaseline", |
| 0 | 80 | | "Baseline", |
| 0 | 81 | | "Extended", |
| 0 | 82 | | "Main", |
| 0 | 83 | | "High", |
| 0 | 84 | | "ProgressiveHigh", |
| 0 | 85 | | "ConstrainedHigh", |
| 0 | 86 | | "High10" |
| 0 | 87 | | ]; |
| | 88 | |
|
| 0 | 89 | | private static readonly string[] _videoProfilesH265 = |
| 0 | 90 | | [ |
| 0 | 91 | | "Main", |
| 0 | 92 | | "Main10" |
| 0 | 93 | | ]; |
| | 94 | |
|
| 0 | 95 | | private static readonly string[] _videoProfilesAv1 = |
| 0 | 96 | | [ |
| 0 | 97 | | "Main", |
| 0 | 98 | | "High", |
| 0 | 99 | | "Professional", |
| 0 | 100 | | ]; |
| | 101 | |
|
| 0 | 102 | | private static readonly HashSet<string> _mp4ContainerNames = new(StringComparer.OrdinalIgnoreCase) |
| 0 | 103 | | { |
| 0 | 104 | | "mp4", |
| 0 | 105 | | "m4a", |
| 0 | 106 | | "m4p", |
| 0 | 107 | | "m4b", |
| 0 | 108 | | "m4r", |
| 0 | 109 | | "m4v", |
| 0 | 110 | | }; |
| | 111 | |
|
| 0 | 112 | | private static readonly TonemappingMode[] _legacyTonemapModes = [TonemappingMode.max, TonemappingMode.rgb]; |
| 0 | 113 | | private static readonly TonemappingMode[] _advancedTonemapModes = [TonemappingMode.lum, TonemappingMode.itp]; |
| | 114 | |
|
| | 115 | | // Set max transcoding channels for encoders that can't handle more than a set amount of channels |
| | 116 | | // AAC, FLAC, ALAC, libopus, libvorbis encoders all support at least 8 channels |
| 0 | 117 | | private static readonly Dictionary<string, int> _audioTranscodeChannelLookup = new(StringComparer.OrdinalIgnoreC |
| 0 | 118 | | { |
| 0 | 119 | | { "libmp3lame", 2 }, |
| 0 | 120 | | { "libfdk_aac", 6 }, |
| 0 | 121 | | { "ac3", 6 }, |
| 0 | 122 | | { "eac3", 6 }, |
| 0 | 123 | | { "dca", 6 }, |
| 0 | 124 | | { "mlp", 6 }, |
| 0 | 125 | | { "truehd", 6 }, |
| 0 | 126 | | }; |
| | 127 | |
|
| 0 | 128 | | private static readonly Dictionary<HardwareAccelerationType, string> _mjpegCodecMap = new() |
| 0 | 129 | | { |
| 0 | 130 | | { HardwareAccelerationType.vaapi, _defaultMjpegEncoder + "_vaapi" }, |
| 0 | 131 | | { HardwareAccelerationType.qsv, _defaultMjpegEncoder + "_qsv" }, |
| 0 | 132 | | { HardwareAccelerationType.videotoolbox, _defaultMjpegEncoder + "_videotoolbox" }, |
| 0 | 133 | | { HardwareAccelerationType.rkmpp, _defaultMjpegEncoder + "_rkmpp" } |
| 0 | 134 | | }; |
| | 135 | |
|
| 0 | 136 | | public static readonly string[] LosslessAudioCodecs = |
| 0 | 137 | | [ |
| 0 | 138 | | "alac", |
| 0 | 139 | | "ape", |
| 0 | 140 | | "flac", |
| 0 | 141 | | "mlp", |
| 0 | 142 | | "truehd", |
| 0 | 143 | | "wavpack" |
| 0 | 144 | | ]; |
| | 145 | |
|
| | 146 | | public EncodingHelper( |
| | 147 | | IApplicationPaths appPaths, |
| | 148 | | IMediaEncoder mediaEncoder, |
| | 149 | | ISubtitleEncoder subtitleEncoder, |
| | 150 | | IConfiguration config, |
| | 151 | | IConfigurationManager configurationManager) |
| | 152 | | { |
| 22 | 153 | | _appPaths = appPaths; |
| 22 | 154 | | _mediaEncoder = mediaEncoder; |
| 22 | 155 | | _subtitleEncoder = subtitleEncoder; |
| 22 | 156 | | _config = config; |
| 22 | 157 | | _configurationManager = configurationManager; |
| 22 | 158 | | } |
| | 159 | |
|
| | 160 | | [GeneratedRegex(@"\s+")] |
| | 161 | | private static partial Regex WhiteSpaceRegex(); |
| | 162 | |
|
| | 163 | | public string GetH264Encoder(EncodingJobInfo state, EncodingOptions encodingOptions) |
| 0 | 164 | | => GetH26xOrAv1Encoder("libx264", "h264", state, encodingOptions); |
| | 165 | |
|
| | 166 | | public string GetH265Encoder(EncodingJobInfo state, EncodingOptions encodingOptions) |
| 0 | 167 | | => GetH26xOrAv1Encoder("libx265", "hevc", state, encodingOptions); |
| | 168 | |
|
| | 169 | | public string GetAv1Encoder(EncodingJobInfo state, EncodingOptions encodingOptions) |
| 0 | 170 | | => GetH26xOrAv1Encoder("libsvtav1", "av1", state, encodingOptions); |
| | 171 | |
|
| | 172 | | private string GetH26xOrAv1Encoder(string defaultEncoder, string hwEncoder, EncodingJobInfo state, EncodingOptio |
| | 173 | | { |
| | 174 | | // Only use alternative encoders for video files. |
| | 175 | | // When using concat with folder rips, if the mfx session fails to initialize, ffmpeg will be stuck retrying |
| | 176 | | // Since transcoding of folder rips is experimental anyway, it's not worth adding additional variables such |
| 0 | 177 | | if (state.VideoType == VideoType.VideoFile) |
| | 178 | | { |
| 0 | 179 | | var hwType = encodingOptions.HardwareAccelerationType; |
| | 180 | |
|
| 0 | 181 | | var codecMap = new Dictionary<HardwareAccelerationType, string>() |
| 0 | 182 | | { |
| 0 | 183 | | { HardwareAccelerationType.amf, hwEncoder + "_amf" }, |
| 0 | 184 | | { HardwareAccelerationType.nvenc, hwEncoder + "_nvenc" }, |
| 0 | 185 | | { HardwareAccelerationType.qsv, hwEncoder + "_qsv" }, |
| 0 | 186 | | { HardwareAccelerationType.vaapi, hwEncoder + "_vaapi" }, |
| 0 | 187 | | { HardwareAccelerationType.videotoolbox, hwEncoder + "_videotoolbox" }, |
| 0 | 188 | | { HardwareAccelerationType.v4l2m2m, hwEncoder + "_v4l2m2m" }, |
| 0 | 189 | | { HardwareAccelerationType.rkmpp, hwEncoder + "_rkmpp" }, |
| 0 | 190 | | }; |
| | 191 | |
|
| 0 | 192 | | if (hwType != HardwareAccelerationType.none |
| 0 | 193 | | && encodingOptions.EnableHardwareEncoding |
| 0 | 194 | | && codecMap.TryGetValue(hwType, out var preferredEncoder) |
| 0 | 195 | | && _mediaEncoder.SupportsEncoder(preferredEncoder)) |
| | 196 | | { |
| 0 | 197 | | return preferredEncoder; |
| | 198 | | } |
| | 199 | | } |
| | 200 | |
|
| 0 | 201 | | return defaultEncoder; |
| | 202 | | } |
| | 203 | |
|
| | 204 | | private string GetMjpegEncoder(EncodingJobInfo state, EncodingOptions encodingOptions) |
| | 205 | | { |
| 0 | 206 | | if (state.VideoType == VideoType.VideoFile) |
| | 207 | | { |
| 0 | 208 | | var hwType = encodingOptions.HardwareAccelerationType; |
| | 209 | |
|
| 0 | 210 | | if (hwType != HardwareAccelerationType.none |
| 0 | 211 | | && encodingOptions.EnableHardwareEncoding |
| 0 | 212 | | && _mjpegCodecMap.TryGetValue(hwType, out var preferredEncoder) |
| 0 | 213 | | && _mediaEncoder.SupportsEncoder(preferredEncoder)) |
| | 214 | | { |
| 0 | 215 | | return preferredEncoder; |
| | 216 | | } |
| | 217 | | } |
| | 218 | |
|
| 0 | 219 | | return _defaultMjpegEncoder; |
| | 220 | | } |
| | 221 | |
|
| | 222 | | private bool IsVaapiSupported(EncodingJobInfo state) |
| | 223 | | { |
| | 224 | | // vaapi will throw an error with this input |
| | 225 | | // [vaapi @ 0x7faed8000960] No VAAPI support for codec mpeg4 profile -99. |
| 0 | 226 | | if (string.Equals(state.VideoStream?.Codec, "mpeg4", StringComparison.OrdinalIgnoreCase)) |
| | 227 | | { |
| 0 | 228 | | return false; |
| | 229 | | } |
| | 230 | |
|
| 0 | 231 | | return _mediaEncoder.SupportsHwaccel("vaapi"); |
| | 232 | | } |
| | 233 | |
|
| | 234 | | private bool IsVaapiFullSupported() |
| | 235 | | { |
| 0 | 236 | | return _mediaEncoder.SupportsHwaccel("drm") |
| 0 | 237 | | && _mediaEncoder.SupportsHwaccel("vaapi") |
| 0 | 238 | | && _mediaEncoder.SupportsFilter("scale_vaapi") |
| 0 | 239 | | && _mediaEncoder.SupportsFilter("deinterlace_vaapi") |
| 0 | 240 | | && _mediaEncoder.SupportsFilter("tonemap_vaapi") |
| 0 | 241 | | && _mediaEncoder.SupportsFilter("procamp_vaapi") |
| 0 | 242 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.OverlayVaapiFrameSync) |
| 0 | 243 | | && _mediaEncoder.SupportsFilter("transpose_vaapi") |
| 0 | 244 | | && _mediaEncoder.SupportsFilter("hwupload_vaapi"); |
| | 245 | | } |
| | 246 | |
|
| | 247 | | private bool IsRkmppFullSupported() |
| | 248 | | { |
| 0 | 249 | | return _mediaEncoder.SupportsHwaccel("rkmpp") |
| 0 | 250 | | && _mediaEncoder.SupportsFilter("scale_rkrga") |
| 0 | 251 | | && _mediaEncoder.SupportsFilter("vpp_rkrga") |
| 0 | 252 | | && _mediaEncoder.SupportsFilter("overlay_rkrga"); |
| | 253 | | } |
| | 254 | |
|
| | 255 | | private bool IsOpenclFullSupported() |
| | 256 | | { |
| 0 | 257 | | return _mediaEncoder.SupportsHwaccel("opencl") |
| 0 | 258 | | && _mediaEncoder.SupportsFilter("scale_opencl") |
| 0 | 259 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.TonemapOpenclBt2390) |
| 0 | 260 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.OverlayOpenclFrameSync); |
| | 261 | |
|
| | 262 | | // Let transpose_opencl optional for the time being. |
| | 263 | | } |
| | 264 | |
|
| | 265 | | private bool IsCudaFullSupported() |
| | 266 | | { |
| 0 | 267 | | return _mediaEncoder.SupportsHwaccel("cuda") |
| 0 | 268 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.ScaleCudaFormat) |
| 0 | 269 | | && _mediaEncoder.SupportsFilter("yadif_cuda") |
| 0 | 270 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.TonemapCudaName) |
| 0 | 271 | | && _mediaEncoder.SupportsFilter("overlay_cuda") |
| 0 | 272 | | && _mediaEncoder.SupportsFilter("hwupload_cuda"); |
| | 273 | |
|
| | 274 | | // Let transpose_cuda optional for the time being. |
| | 275 | | } |
| | 276 | |
|
| | 277 | | private bool IsVulkanFullSupported() |
| | 278 | | { |
| 0 | 279 | | return _mediaEncoder.SupportsHwaccel("vulkan") |
| 0 | 280 | | && _mediaEncoder.SupportsFilter("libplacebo") |
| 0 | 281 | | && _mediaEncoder.SupportsFilter("scale_vulkan") |
| 0 | 282 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.OverlayVulkanFrameSync) |
| 0 | 283 | | && _mediaEncoder.SupportsFilter("transpose_vulkan") |
| 0 | 284 | | && _mediaEncoder.SupportsFilter("flip_vulkan"); |
| | 285 | | } |
| | 286 | |
|
| | 287 | | private bool IsVideoToolboxFullSupported() |
| | 288 | | { |
| 0 | 289 | | return _mediaEncoder.SupportsHwaccel("videotoolbox") |
| 0 | 290 | | && _mediaEncoder.SupportsFilter("yadif_videotoolbox") |
| 0 | 291 | | && _mediaEncoder.SupportsFilter("overlay_videotoolbox") |
| 0 | 292 | | && _mediaEncoder.SupportsFilter("tonemap_videotoolbox") |
| 0 | 293 | | && _mediaEncoder.SupportsFilter("scale_vt"); |
| | 294 | |
|
| | 295 | | // Let transpose_vt optional for the time being. |
| | 296 | | } |
| | 297 | |
|
| | 298 | | private bool IsSwTonemapAvailable(EncodingJobInfo state, EncodingOptions options) |
| | 299 | | { |
| 0 | 300 | | if (state.VideoStream is null |
| 0 | 301 | | || !options.EnableTonemapping |
| 0 | 302 | | || GetVideoColorBitDepth(state) != 10 |
| 0 | 303 | | || !_mediaEncoder.SupportsFilter("tonemapx")) |
| | 304 | | { |
| 0 | 305 | | return false; |
| | 306 | | } |
| | 307 | |
|
| 0 | 308 | | return state.VideoStream.VideoRange == VideoRange.HDR; |
| | 309 | | } |
| | 310 | |
|
| | 311 | | private bool IsHwTonemapAvailable(EncodingJobInfo state, EncodingOptions options) |
| | 312 | | { |
| 0 | 313 | | if (state.VideoStream is null |
| 0 | 314 | | || !options.EnableTonemapping |
| 0 | 315 | | || GetVideoColorBitDepth(state) != 10) |
| | 316 | | { |
| 0 | 317 | | return false; |
| | 318 | | } |
| | 319 | |
|
| 0 | 320 | | if (state.VideoStream.VideoRange == VideoRange.HDR |
| 0 | 321 | | && state.VideoStream.VideoRangeType == VideoRangeType.DOVI) |
| | 322 | | { |
| | 323 | | // Only native SW decoder and HW accelerator can parse dovi rpu. |
| 0 | 324 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 325 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 326 | | var isNvdecDecoder = vidDecoder.Contains("cuda", StringComparison.OrdinalIgnoreCase); |
| 0 | 327 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 328 | | var isD3d11vaDecoder = vidDecoder.Contains("d3d11va", StringComparison.OrdinalIgnoreCase); |
| 0 | 329 | | var isVideoToolBoxDecoder = vidDecoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| 0 | 330 | | return isSwDecoder || isNvdecDecoder || isVaapiDecoder || isD3d11vaDecoder || isVideoToolBoxDecoder; |
| | 331 | | } |
| | 332 | |
|
| 0 | 333 | | return state.VideoStream.VideoRange == VideoRange.HDR |
| 0 | 334 | | && (state.VideoStream.VideoRangeType == VideoRangeType.HDR10 |
| 0 | 335 | | || state.VideoStream.VideoRangeType == VideoRangeType.HLG |
| 0 | 336 | | || state.VideoStream.VideoRangeType == VideoRangeType.DOVIWithHDR10 |
| 0 | 337 | | || state.VideoStream.VideoRangeType == VideoRangeType.DOVIWithHLG); |
| | 338 | | } |
| | 339 | |
|
| | 340 | | private bool IsVulkanHwTonemapAvailable(EncodingJobInfo state, EncodingOptions options) |
| | 341 | | { |
| 0 | 342 | | if (state.VideoStream is null) |
| | 343 | | { |
| 0 | 344 | | return false; |
| | 345 | | } |
| | 346 | |
|
| | 347 | | // libplacebo has partial Dolby Vision to SDR tonemapping support. |
| 0 | 348 | | return options.EnableTonemapping |
| 0 | 349 | | && state.VideoStream.VideoRange == VideoRange.HDR |
| 0 | 350 | | && GetVideoColorBitDepth(state) == 10; |
| | 351 | | } |
| | 352 | |
|
| | 353 | | private bool IsIntelVppTonemapAvailable(EncodingJobInfo state, EncodingOptions options) |
| | 354 | | { |
| 0 | 355 | | if (state.VideoStream is null |
| 0 | 356 | | || !options.EnableVppTonemapping |
| 0 | 357 | | || GetVideoColorBitDepth(state) != 10) |
| | 358 | | { |
| 0 | 359 | | return false; |
| | 360 | | } |
| | 361 | |
|
| | 362 | | // prefer 'tonemap_vaapi' over 'vpp_qsv' on Linux for supporting Gen9/KBLx. |
| | 363 | | // 'vpp_qsv' requires VPL, which is only supported on Gen12/TGLx and newer. |
| 0 | 364 | | if (OperatingSystem.IsWindows() |
| 0 | 365 | | && options.HardwareAccelerationType == HardwareAccelerationType.qsv |
| 0 | 366 | | && _mediaEncoder.EncoderVersion < _minFFmpegQsvVppTonemapOption) |
| | 367 | | { |
| 0 | 368 | | return false; |
| | 369 | | } |
| | 370 | |
|
| 0 | 371 | | return state.VideoStream.VideoRange == VideoRange.HDR |
| 0 | 372 | | && (state.VideoStream.VideoRangeType == VideoRangeType.HDR10 |
| 0 | 373 | | || state.VideoStream.VideoRangeType == VideoRangeType.DOVIWithHDR10); |
| | 374 | | } |
| | 375 | |
|
| | 376 | | private bool IsVideoToolboxTonemapAvailable(EncodingJobInfo state, EncodingOptions options) |
| | 377 | | { |
| 0 | 378 | | if (state.VideoStream is null |
| 0 | 379 | | || !options.EnableVideoToolboxTonemapping |
| 0 | 380 | | || GetVideoColorBitDepth(state) != 10) |
| | 381 | | { |
| 0 | 382 | | return false; |
| | 383 | | } |
| | 384 | |
|
| | 385 | | // Certain DV profile 5 video works in Safari with direct playing, but the VideoToolBox does not produce cor |
| | 386 | | // All other HDR formats working. |
| 0 | 387 | | return state.VideoStream.VideoRange == VideoRange.HDR |
| 0 | 388 | | && state.VideoStream.VideoRangeType is VideoRangeType.HDR10 or VideoRangeType.HLG or VideoRangeType.H |
| | 389 | | } |
| | 390 | |
|
| | 391 | | /// <summary> |
| | 392 | | /// Gets the name of the output video codec. |
| | 393 | | /// </summary> |
| | 394 | | /// <param name="state">Encoding state.</param> |
| | 395 | | /// <param name="encodingOptions">Encoding options.</param> |
| | 396 | | /// <returns>Encoder string.</returns> |
| | 397 | | public string GetVideoEncoder(EncodingJobInfo state, EncodingOptions encodingOptions) |
| | 398 | | { |
| 0 | 399 | | var codec = state.OutputVideoCodec; |
| | 400 | |
|
| 0 | 401 | | if (!string.IsNullOrEmpty(codec)) |
| | 402 | | { |
| 0 | 403 | | if (string.Equals(codec, "av1", StringComparison.OrdinalIgnoreCase)) |
| | 404 | | { |
| 0 | 405 | | return GetAv1Encoder(state, encodingOptions); |
| | 406 | | } |
| | 407 | |
|
| 0 | 408 | | if (string.Equals(codec, "h265", StringComparison.OrdinalIgnoreCase) |
| 0 | 409 | | || string.Equals(codec, "hevc", StringComparison.OrdinalIgnoreCase)) |
| | 410 | | { |
| 0 | 411 | | return GetH265Encoder(state, encodingOptions); |
| | 412 | | } |
| | 413 | |
|
| 0 | 414 | | if (string.Equals(codec, "h264", StringComparison.OrdinalIgnoreCase)) |
| | 415 | | { |
| 0 | 416 | | return GetH264Encoder(state, encodingOptions); |
| | 417 | | } |
| | 418 | |
|
| 0 | 419 | | if (string.Equals(codec, "mjpeg", StringComparison.OrdinalIgnoreCase)) |
| | 420 | | { |
| 0 | 421 | | return GetMjpegEncoder(state, encodingOptions); |
| | 422 | | } |
| | 423 | |
|
| 0 | 424 | | if (_validationRegex.IsMatch(codec)) |
| | 425 | | { |
| 0 | 426 | | return codec.ToLowerInvariant(); |
| | 427 | | } |
| | 428 | | } |
| | 429 | |
|
| 0 | 430 | | return "copy"; |
| | 431 | | } |
| | 432 | |
|
| | 433 | | /// <summary> |
| | 434 | | /// Gets the user agent param. |
| | 435 | | /// </summary> |
| | 436 | | /// <param name="state">The state.</param> |
| | 437 | | /// <returns>System.String.</returns> |
| | 438 | | public string GetUserAgentParam(EncodingJobInfo state) |
| | 439 | | { |
| 0 | 440 | | if (state.RemoteHttpHeaders.TryGetValue("User-Agent", out string useragent)) |
| | 441 | | { |
| 0 | 442 | | return "-user_agent \"" + useragent + "\""; |
| | 443 | | } |
| | 444 | |
|
| 0 | 445 | | return string.Empty; |
| | 446 | | } |
| | 447 | |
|
| | 448 | | /// <summary> |
| | 449 | | /// Gets the referer param. |
| | 450 | | /// </summary> |
| | 451 | | /// <param name="state">The state.</param> |
| | 452 | | /// <returns>System.String.</returns> |
| | 453 | | public string GetRefererParam(EncodingJobInfo state) |
| | 454 | | { |
| 0 | 455 | | if (state.RemoteHttpHeaders.TryGetValue("Referer", out string referer)) |
| | 456 | | { |
| 0 | 457 | | return "-referer \"" + referer + "\""; |
| | 458 | | } |
| | 459 | |
|
| 0 | 460 | | return string.Empty; |
| | 461 | | } |
| | 462 | |
|
| | 463 | | public static string GetInputFormat(string container) |
| | 464 | | { |
| 0 | 465 | | if (string.IsNullOrEmpty(container) || !_validationRegex.IsMatch(container)) |
| | 466 | | { |
| 0 | 467 | | return null; |
| | 468 | | } |
| | 469 | |
|
| 0 | 470 | | container = container.Replace("mkv", "matroska", StringComparison.OrdinalIgnoreCase); |
| | 471 | |
|
| 0 | 472 | | if (string.Equals(container, "ts", StringComparison.OrdinalIgnoreCase)) |
| | 473 | | { |
| 0 | 474 | | return "mpegts"; |
| | 475 | | } |
| | 476 | |
|
| | 477 | | // For these need to find out the ffmpeg names |
| 0 | 478 | | if (string.Equals(container, "m2ts", StringComparison.OrdinalIgnoreCase)) |
| | 479 | | { |
| 0 | 480 | | return null; |
| | 481 | | } |
| | 482 | |
|
| 0 | 483 | | if (string.Equals(container, "wmv", StringComparison.OrdinalIgnoreCase)) |
| | 484 | | { |
| 0 | 485 | | return null; |
| | 486 | | } |
| | 487 | |
|
| 0 | 488 | | if (string.Equals(container, "mts", StringComparison.OrdinalIgnoreCase)) |
| | 489 | | { |
| 0 | 490 | | return null; |
| | 491 | | } |
| | 492 | |
|
| 0 | 493 | | if (string.Equals(container, "vob", StringComparison.OrdinalIgnoreCase)) |
| | 494 | | { |
| 0 | 495 | | return null; |
| | 496 | | } |
| | 497 | |
|
| 0 | 498 | | if (string.Equals(container, "mpg", StringComparison.OrdinalIgnoreCase)) |
| | 499 | | { |
| 0 | 500 | | return null; |
| | 501 | | } |
| | 502 | |
|
| 0 | 503 | | if (string.Equals(container, "mpeg", StringComparison.OrdinalIgnoreCase)) |
| | 504 | | { |
| 0 | 505 | | return null; |
| | 506 | | } |
| | 507 | |
|
| 0 | 508 | | if (string.Equals(container, "rec", StringComparison.OrdinalIgnoreCase)) |
| | 509 | | { |
| 0 | 510 | | return null; |
| | 511 | | } |
| | 512 | |
|
| 0 | 513 | | if (string.Equals(container, "dvr-ms", StringComparison.OrdinalIgnoreCase)) |
| | 514 | | { |
| 0 | 515 | | return null; |
| | 516 | | } |
| | 517 | |
|
| 0 | 518 | | if (string.Equals(container, "ogm", StringComparison.OrdinalIgnoreCase)) |
| | 519 | | { |
| 0 | 520 | | return null; |
| | 521 | | } |
| | 522 | |
|
| 0 | 523 | | if (string.Equals(container, "divx", StringComparison.OrdinalIgnoreCase)) |
| | 524 | | { |
| 0 | 525 | | return null; |
| | 526 | | } |
| | 527 | |
|
| 0 | 528 | | if (string.Equals(container, "tp", StringComparison.OrdinalIgnoreCase)) |
| | 529 | | { |
| 0 | 530 | | return null; |
| | 531 | | } |
| | 532 | |
|
| 0 | 533 | | if (string.Equals(container, "rmvb", StringComparison.OrdinalIgnoreCase)) |
| | 534 | | { |
| 0 | 535 | | return null; |
| | 536 | | } |
| | 537 | |
|
| 0 | 538 | | if (string.Equals(container, "rtp", StringComparison.OrdinalIgnoreCase)) |
| | 539 | | { |
| 0 | 540 | | return null; |
| | 541 | | } |
| | 542 | |
|
| | 543 | | // Seeing reported failures here, not sure yet if this is related to specifying input format |
| 0 | 544 | | if (string.Equals(container, "m4v", StringComparison.OrdinalIgnoreCase)) |
| | 545 | | { |
| 0 | 546 | | return null; |
| | 547 | | } |
| | 548 | |
|
| | 549 | | // obviously don't do this for strm files |
| 0 | 550 | | if (string.Equals(container, "strm", StringComparison.OrdinalIgnoreCase)) |
| | 551 | | { |
| 0 | 552 | | return null; |
| | 553 | | } |
| | 554 | |
|
| | 555 | | // ISO files don't have an ffmpeg format |
| 0 | 556 | | if (string.Equals(container, "iso", StringComparison.OrdinalIgnoreCase)) |
| | 557 | | { |
| 0 | 558 | | return null; |
| | 559 | | } |
| | 560 | |
|
| 0 | 561 | | return container; |
| | 562 | | } |
| | 563 | |
|
| | 564 | | /// <summary> |
| | 565 | | /// Gets decoder from a codec. |
| | 566 | | /// </summary> |
| | 567 | | /// <param name="codec">Codec to use.</param> |
| | 568 | | /// <returns>Decoder string.</returns> |
| | 569 | | public string GetDecoderFromCodec(string codec) |
| | 570 | | { |
| | 571 | | // For these need to find out the ffmpeg names |
| 0 | 572 | | if (string.Equals(codec, "mp2", StringComparison.OrdinalIgnoreCase)) |
| | 573 | | { |
| 0 | 574 | | return null; |
| | 575 | | } |
| | 576 | |
|
| 0 | 577 | | if (string.Equals(codec, "aac_latm", StringComparison.OrdinalIgnoreCase)) |
| | 578 | | { |
| 0 | 579 | | return null; |
| | 580 | | } |
| | 581 | |
|
| 0 | 582 | | if (string.Equals(codec, "eac3", StringComparison.OrdinalIgnoreCase)) |
| | 583 | | { |
| 0 | 584 | | return null; |
| | 585 | | } |
| | 586 | |
|
| 0 | 587 | | if (_mediaEncoder.SupportsDecoder(codec)) |
| | 588 | | { |
| 0 | 589 | | return codec; |
| | 590 | | } |
| | 591 | |
|
| 0 | 592 | | return null; |
| | 593 | | } |
| | 594 | |
|
| | 595 | | /// <summary> |
| | 596 | | /// Infers the audio codec based on the url. |
| | 597 | | /// </summary> |
| | 598 | | /// <param name="container">Container to use.</param> |
| | 599 | | /// <returns>Codec string.</returns> |
| | 600 | | public string InferAudioCodec(string container) |
| | 601 | | { |
| 0 | 602 | | var ext = "." + (container ?? string.Empty); |
| | 603 | |
|
| 0 | 604 | | if (string.Equals(ext, ".mp3", StringComparison.OrdinalIgnoreCase)) |
| | 605 | | { |
| 0 | 606 | | return "mp3"; |
| | 607 | | } |
| | 608 | |
|
| 0 | 609 | | if (string.Equals(ext, ".aac", StringComparison.OrdinalIgnoreCase)) |
| | 610 | | { |
| 0 | 611 | | return "aac"; |
| | 612 | | } |
| | 613 | |
|
| 0 | 614 | | if (string.Equals(ext, ".wma", StringComparison.OrdinalIgnoreCase)) |
| | 615 | | { |
| 0 | 616 | | return "wma"; |
| | 617 | | } |
| | 618 | |
|
| 0 | 619 | | if (string.Equals(ext, ".ogg", StringComparison.OrdinalIgnoreCase)) |
| | 620 | | { |
| 0 | 621 | | return "vorbis"; |
| | 622 | | } |
| | 623 | |
|
| 0 | 624 | | if (string.Equals(ext, ".oga", StringComparison.OrdinalIgnoreCase)) |
| | 625 | | { |
| 0 | 626 | | return "vorbis"; |
| | 627 | | } |
| | 628 | |
|
| 0 | 629 | | if (string.Equals(ext, ".ogv", StringComparison.OrdinalIgnoreCase)) |
| | 630 | | { |
| 0 | 631 | | return "vorbis"; |
| | 632 | | } |
| | 633 | |
|
| 0 | 634 | | if (string.Equals(ext, ".webm", StringComparison.OrdinalIgnoreCase)) |
| | 635 | | { |
| 0 | 636 | | return "vorbis"; |
| | 637 | | } |
| | 638 | |
|
| 0 | 639 | | if (string.Equals(ext, ".webma", StringComparison.OrdinalIgnoreCase)) |
| | 640 | | { |
| 0 | 641 | | return "vorbis"; |
| | 642 | | } |
| | 643 | |
|
| 0 | 644 | | return "copy"; |
| | 645 | | } |
| | 646 | |
|
| | 647 | | /// <summary> |
| | 648 | | /// Infers the video codec. |
| | 649 | | /// </summary> |
| | 650 | | /// <param name="url">The URL.</param> |
| | 651 | | /// <returns>System.Nullable{VideoCodecs}.</returns> |
| | 652 | | public string InferVideoCodec(string url) |
| | 653 | | { |
| 0 | 654 | | var ext = Path.GetExtension(url.AsSpan()); |
| | 655 | |
|
| 0 | 656 | | if (ext.Equals(".asf", StringComparison.OrdinalIgnoreCase)) |
| | 657 | | { |
| 0 | 658 | | return "wmv"; |
| | 659 | | } |
| | 660 | |
|
| 0 | 661 | | if (ext.Equals(".webm", StringComparison.OrdinalIgnoreCase)) |
| | 662 | | { |
| | 663 | | // TODO: this may not always mean VP8, as the codec ages |
| 0 | 664 | | return "vp8"; |
| | 665 | | } |
| | 666 | |
|
| 0 | 667 | | if (ext.Equals(".ogg", StringComparison.OrdinalIgnoreCase) || ext.Equals(".ogv", StringComparison.OrdinalIgn |
| | 668 | | { |
| 0 | 669 | | return "theora"; |
| | 670 | | } |
| | 671 | |
|
| 0 | 672 | | if (ext.Equals(".m3u8", StringComparison.OrdinalIgnoreCase) || ext.Equals(".ts", StringComparison.OrdinalIgn |
| | 673 | | { |
| 0 | 674 | | return "h264"; |
| | 675 | | } |
| | 676 | |
|
| 0 | 677 | | return "copy"; |
| | 678 | | } |
| | 679 | |
|
| | 680 | | public int GetVideoProfileScore(string videoCodec, string videoProfile) |
| | 681 | | { |
| | 682 | | // strip spaces because they may be stripped out on the query string |
| 0 | 683 | | string profile = videoProfile.Replace(" ", string.Empty, StringComparison.Ordinal); |
| 0 | 684 | | if (string.Equals("h264", videoCodec, StringComparison.OrdinalIgnoreCase)) |
| | 685 | | { |
| 0 | 686 | | return Array.FindIndex(_videoProfilesH264, x => string.Equals(x, profile, StringComparison.OrdinalIgnore |
| | 687 | | } |
| | 688 | |
|
| 0 | 689 | | if (string.Equals("hevc", videoCodec, StringComparison.OrdinalIgnoreCase)) |
| | 690 | | { |
| 0 | 691 | | return Array.FindIndex(_videoProfilesH265, x => string.Equals(x, profile, StringComparison.OrdinalIgnore |
| | 692 | | } |
| | 693 | |
|
| 0 | 694 | | if (string.Equals("av1", videoCodec, StringComparison.OrdinalIgnoreCase)) |
| | 695 | | { |
| 0 | 696 | | return Array.FindIndex(_videoProfilesAv1, x => string.Equals(x, profile, StringComparison.OrdinalIgnoreC |
| | 697 | | } |
| | 698 | |
|
| 0 | 699 | | return -1; |
| | 700 | | } |
| | 701 | |
|
| | 702 | | /// <summary> |
| | 703 | | /// Gets the audio encoder. |
| | 704 | | /// </summary> |
| | 705 | | /// <param name="state">The state.</param> |
| | 706 | | /// <returns>System.String.</returns> |
| | 707 | | public string GetAudioEncoder(EncodingJobInfo state) |
| | 708 | | { |
| 0 | 709 | | var codec = state.OutputAudioCodec; |
| | 710 | |
|
| 0 | 711 | | if (!_validationRegex.IsMatch(codec)) |
| | 712 | | { |
| 0 | 713 | | codec = "aac"; |
| | 714 | | } |
| | 715 | |
|
| 0 | 716 | | if (string.Equals(codec, "aac", StringComparison.OrdinalIgnoreCase)) |
| | 717 | | { |
| | 718 | | // Use Apple's aac encoder if available as it provides best audio quality |
| 0 | 719 | | if (_mediaEncoder.SupportsEncoder("aac_at")) |
| | 720 | | { |
| 0 | 721 | | return "aac_at"; |
| | 722 | | } |
| | 723 | |
|
| | 724 | | // Use libfdk_aac for better audio quality if using custom build of FFmpeg which has fdk_aac support |
| 0 | 725 | | if (_mediaEncoder.SupportsEncoder("libfdk_aac")) |
| | 726 | | { |
| 0 | 727 | | return "libfdk_aac"; |
| | 728 | | } |
| | 729 | |
|
| 0 | 730 | | return "aac"; |
| | 731 | | } |
| | 732 | |
|
| 0 | 733 | | if (string.Equals(codec, "mp3", StringComparison.OrdinalIgnoreCase)) |
| | 734 | | { |
| 0 | 735 | | return "libmp3lame"; |
| | 736 | | } |
| | 737 | |
|
| 0 | 738 | | if (string.Equals(codec, "vorbis", StringComparison.OrdinalIgnoreCase)) |
| | 739 | | { |
| 0 | 740 | | return "libvorbis"; |
| | 741 | | } |
| | 742 | |
|
| 0 | 743 | | if (string.Equals(codec, "opus", StringComparison.OrdinalIgnoreCase)) |
| | 744 | | { |
| 0 | 745 | | return "libopus"; |
| | 746 | | } |
| | 747 | |
|
| 0 | 748 | | if (string.Equals(codec, "flac", StringComparison.OrdinalIgnoreCase)) |
| | 749 | | { |
| 0 | 750 | | return "flac"; |
| | 751 | | } |
| | 752 | |
|
| 0 | 753 | | if (string.Equals(codec, "dts", StringComparison.OrdinalIgnoreCase)) |
| | 754 | | { |
| 0 | 755 | | return "dca"; |
| | 756 | | } |
| | 757 | |
|
| 0 | 758 | | if (string.Equals(codec, "alac", StringComparison.OrdinalIgnoreCase)) |
| | 759 | | { |
| | 760 | | // The ffmpeg upstream breaks the AudioToolbox ALAC encoder in version 6.1 but fixes it in version 7.0. |
| | 761 | | // Since ALAC is lossless in quality and the AudioToolbox encoder is not faster, |
| | 762 | | // its only benefit is a smaller file size. |
| | 763 | | // To prevent problems, use the ffmpeg native encoder instead. |
| 0 | 764 | | return "alac"; |
| | 765 | | } |
| | 766 | |
|
| 0 | 767 | | return codec.ToLowerInvariant(); |
| | 768 | | } |
| | 769 | |
|
| | 770 | | private string GetRkmppDeviceArgs(string alias) |
| | 771 | | { |
| 0 | 772 | | alias ??= RkmppAlias; |
| | 773 | |
|
| | 774 | | // device selection in rk is not supported. |
| 0 | 775 | | return " -init_hw_device rkmpp=" + alias; |
| | 776 | | } |
| | 777 | |
|
| | 778 | | private string GetVideoToolboxDeviceArgs(string alias) |
| | 779 | | { |
| 0 | 780 | | alias ??= VideotoolboxAlias; |
| | 781 | |
|
| | 782 | | // device selection in vt is not supported. |
| 0 | 783 | | return " -init_hw_device videotoolbox=" + alias; |
| | 784 | | } |
| | 785 | |
|
| | 786 | | private string GetCudaDeviceArgs(int deviceIndex, string alias) |
| | 787 | | { |
| 0 | 788 | | alias ??= CudaAlias; |
| 0 | 789 | | deviceIndex = deviceIndex >= 0 |
| 0 | 790 | | ? deviceIndex |
| 0 | 791 | | : 0; |
| | 792 | |
|
| 0 | 793 | | return string.Format( |
| 0 | 794 | | CultureInfo.InvariantCulture, |
| 0 | 795 | | " -init_hw_device cuda={0}:{1}", |
| 0 | 796 | | alias, |
| 0 | 797 | | deviceIndex); |
| | 798 | | } |
| | 799 | |
|
| | 800 | | private string GetVulkanDeviceArgs(int deviceIndex, string deviceName, string srcDeviceAlias, string alias) |
| | 801 | | { |
| 0 | 802 | | alias ??= VulkanAlias; |
| 0 | 803 | | deviceIndex = deviceIndex >= 0 |
| 0 | 804 | | ? deviceIndex |
| 0 | 805 | | : 0; |
| 0 | 806 | | var vendorOpts = string.IsNullOrEmpty(deviceName) |
| 0 | 807 | | ? ":" + deviceIndex |
| 0 | 808 | | : ":" + "\"" + deviceName + "\""; |
| 0 | 809 | | var options = string.IsNullOrEmpty(srcDeviceAlias) |
| 0 | 810 | | ? vendorOpts |
| 0 | 811 | | : "@" + srcDeviceAlias; |
| | 812 | |
|
| 0 | 813 | | return string.Format( |
| 0 | 814 | | CultureInfo.InvariantCulture, |
| 0 | 815 | | " -init_hw_device vulkan={0}{1}", |
| 0 | 816 | | alias, |
| 0 | 817 | | options); |
| | 818 | | } |
| | 819 | |
|
| | 820 | | private string GetOpenclDeviceArgs(int deviceIndex, string deviceVendorName, string srcDeviceAlias, string alias |
| | 821 | | { |
| 0 | 822 | | alias ??= OpenclAlias; |
| 0 | 823 | | deviceIndex = deviceIndex >= 0 |
| 0 | 824 | | ? deviceIndex |
| 0 | 825 | | : 0; |
| 0 | 826 | | var vendorOpts = string.IsNullOrEmpty(deviceVendorName) |
| 0 | 827 | | ? ":0.0" |
| 0 | 828 | | : ":." + deviceIndex + ",device_vendor=\"" + deviceVendorName + "\""; |
| 0 | 829 | | var options = string.IsNullOrEmpty(srcDeviceAlias) |
| 0 | 830 | | ? vendorOpts |
| 0 | 831 | | : "@" + srcDeviceAlias; |
| | 832 | |
|
| 0 | 833 | | return string.Format( |
| 0 | 834 | | CultureInfo.InvariantCulture, |
| 0 | 835 | | " -init_hw_device opencl={0}{1}", |
| 0 | 836 | | alias, |
| 0 | 837 | | options); |
| | 838 | | } |
| | 839 | |
|
| | 840 | | private string GetD3d11vaDeviceArgs(int deviceIndex, string deviceVendorId, string alias) |
| | 841 | | { |
| 0 | 842 | | alias ??= D3d11vaAlias; |
| 0 | 843 | | deviceIndex = deviceIndex >= 0 ? deviceIndex : 0; |
| 0 | 844 | | var options = string.IsNullOrEmpty(deviceVendorId) |
| 0 | 845 | | ? deviceIndex.ToString(CultureInfo.InvariantCulture) |
| 0 | 846 | | : ",vendor=" + deviceVendorId; |
| | 847 | |
|
| 0 | 848 | | return string.Format( |
| 0 | 849 | | CultureInfo.InvariantCulture, |
| 0 | 850 | | " -init_hw_device d3d11va={0}:{1}", |
| 0 | 851 | | alias, |
| 0 | 852 | | options); |
| | 853 | | } |
| | 854 | |
|
| | 855 | | private string GetVaapiDeviceArgs(string renderNodePath, string driver, string kernelDriver, string srcDeviceAli |
| | 856 | | { |
| 0 | 857 | | alias ??= VaapiAlias; |
| | 858 | |
|
| | 859 | | // 'renderNodePath' has higher priority than 'kernelDriver' |
| 0 | 860 | | var driverOpts = string.IsNullOrEmpty(renderNodePath) |
| 0 | 861 | | ? (string.IsNullOrEmpty(kernelDriver) ? string.Empty : ",kernel_driver=" + kernelDriver) |
| 0 | 862 | | : renderNodePath; |
| | 863 | |
|
| | 864 | | // 'driver' behaves similarly to env LIBVA_DRIVER_NAME |
| 0 | 865 | | driverOpts += string.IsNullOrEmpty(driver) ? string.Empty : ",driver=" + driver; |
| | 866 | |
|
| 0 | 867 | | var options = string.IsNullOrEmpty(srcDeviceAlias) |
| 0 | 868 | | ? (string.IsNullOrEmpty(driverOpts) ? string.Empty : ":" + driverOpts) |
| 0 | 869 | | : "@" + srcDeviceAlias; |
| | 870 | |
|
| 0 | 871 | | return string.Format( |
| 0 | 872 | | CultureInfo.InvariantCulture, |
| 0 | 873 | | " -init_hw_device vaapi={0}{1}", |
| 0 | 874 | | alias, |
| 0 | 875 | | options); |
| | 876 | | } |
| | 877 | |
|
| | 878 | | private string GetDrmDeviceArgs(string renderNodePath, string alias) |
| | 879 | | { |
| 0 | 880 | | alias ??= DrmAlias; |
| 0 | 881 | | renderNodePath = renderNodePath ?? "/dev/dri/renderD128"; |
| | 882 | |
|
| 0 | 883 | | return string.Format( |
| 0 | 884 | | CultureInfo.InvariantCulture, |
| 0 | 885 | | " -init_hw_device drm={0}:{1}", |
| 0 | 886 | | alias, |
| 0 | 887 | | renderNodePath); |
| | 888 | | } |
| | 889 | |
|
| | 890 | | private string GetQsvDeviceArgs(string renderNodePath, string alias) |
| | 891 | | { |
| 0 | 892 | | var arg = " -init_hw_device qsv=" + (alias ?? QsvAlias); |
| 0 | 893 | | if (OperatingSystem.IsLinux()) |
| | 894 | | { |
| | 895 | | // derive qsv from vaapi device |
| 0 | 896 | | return GetVaapiDeviceArgs(renderNodePath, "iHD", "i915", null, VaapiAlias) + arg + "@" + VaapiAlias; |
| | 897 | | } |
| | 898 | |
|
| 0 | 899 | | if (OperatingSystem.IsWindows()) |
| | 900 | | { |
| | 901 | | // on Windows, the deviceIndex is an int |
| 0 | 902 | | if (int.TryParse(renderNodePath, NumberStyles.Integer, CultureInfo.InvariantCulture, out int deviceIndex |
| | 903 | | { |
| 0 | 904 | | return GetD3d11vaDeviceArgs(deviceIndex, string.Empty, D3d11vaAlias) + arg + "@" + D3d11vaAlias; |
| | 905 | | } |
| | 906 | |
|
| | 907 | | // derive qsv from d3d11va device |
| 0 | 908 | | return GetD3d11vaDeviceArgs(0, "0x8086", D3d11vaAlias) + arg + "@" + D3d11vaAlias; |
| | 909 | | } |
| | 910 | |
|
| 0 | 911 | | return null; |
| | 912 | | } |
| | 913 | |
|
| | 914 | | private string GetFilterHwDeviceArgs(string alias) |
| | 915 | | { |
| 0 | 916 | | return string.IsNullOrEmpty(alias) |
| 0 | 917 | | ? string.Empty |
| 0 | 918 | | : " -filter_hw_device " + alias; |
| | 919 | | } |
| | 920 | |
|
| | 921 | | public string GetGraphicalSubCanvasSize(EncodingJobInfo state) |
| | 922 | | { |
| | 923 | | // DVBSUB uses the fixed canvas size 720x576 |
| 0 | 924 | | if (state.SubtitleStream is not null |
| 0 | 925 | | && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode |
| 0 | 926 | | && !state.SubtitleStream.IsTextSubtitleStream |
| 0 | 927 | | && !string.Equals(state.SubtitleStream.Codec, "DVBSUB", StringComparison.OrdinalIgnoreCase)) |
| | 928 | | { |
| 0 | 929 | | var subtitleWidth = state.SubtitleStream?.Width; |
| 0 | 930 | | var subtitleHeight = state.SubtitleStream?.Height; |
| | 931 | |
|
| 0 | 932 | | if (subtitleWidth.HasValue |
| 0 | 933 | | && subtitleHeight.HasValue |
| 0 | 934 | | && subtitleWidth.Value > 0 |
| 0 | 935 | | && subtitleHeight.Value > 0) |
| | 936 | | { |
| 0 | 937 | | return string.Format( |
| 0 | 938 | | CultureInfo.InvariantCulture, |
| 0 | 939 | | " -canvas_size {0}x{1}", |
| 0 | 940 | | subtitleWidth.Value, |
| 0 | 941 | | subtitleHeight.Value); |
| | 942 | | } |
| | 943 | | } |
| | 944 | |
|
| 0 | 945 | | return string.Empty; |
| | 946 | | } |
| | 947 | |
|
| | 948 | | /// <summary> |
| | 949 | | /// Gets the input video hwaccel argument. |
| | 950 | | /// </summary> |
| | 951 | | /// <param name="state">Encoding state.</param> |
| | 952 | | /// <param name="options">Encoding options.</param> |
| | 953 | | /// <returns>Input video hwaccel arguments.</returns> |
| | 954 | | public string GetInputVideoHwaccelArgs(EncodingJobInfo state, EncodingOptions options) |
| | 955 | | { |
| 0 | 956 | | if (!state.IsVideoRequest) |
| | 957 | | { |
| 0 | 958 | | return string.Empty; |
| | 959 | | } |
| | 960 | |
|
| 0 | 961 | | var vidEncoder = GetVideoEncoder(state, options) ?? string.Empty; |
| 0 | 962 | | if (IsCopyCodec(vidEncoder)) |
| | 963 | | { |
| 0 | 964 | | return string.Empty; |
| | 965 | | } |
| | 966 | |
|
| 0 | 967 | | var args = new StringBuilder(); |
| 0 | 968 | | var isWindows = OperatingSystem.IsWindows(); |
| 0 | 969 | | var isLinux = OperatingSystem.IsLinux(); |
| 0 | 970 | | var isMacOS = OperatingSystem.IsMacOS(); |
| 0 | 971 | | var optHwaccelType = options.HardwareAccelerationType; |
| 0 | 972 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 973 | | var isHwTonemapAvailable = IsHwTonemapAvailable(state, options); |
| | 974 | |
|
| 0 | 975 | | if (optHwaccelType == HardwareAccelerationType.vaapi) |
| | 976 | | { |
| 0 | 977 | | if (!isLinux || !_mediaEncoder.SupportsHwaccel("vaapi")) |
| | 978 | | { |
| 0 | 979 | | return string.Empty; |
| | 980 | | } |
| | 981 | |
|
| 0 | 982 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 983 | | var isVaapiEncoder = vidEncoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 984 | | if (!isVaapiDecoder && !isVaapiEncoder) |
| | 985 | | { |
| 0 | 986 | | return string.Empty; |
| | 987 | | } |
| | 988 | |
|
| 0 | 989 | | if (_mediaEncoder.IsVaapiDeviceInteliHD) |
| | 990 | | { |
| 0 | 991 | | args.Append(GetVaapiDeviceArgs(options.VaapiDevice, "iHD", null, null, VaapiAlias)); |
| | 992 | | } |
| 0 | 993 | | else if (_mediaEncoder.IsVaapiDeviceInteli965) |
| | 994 | | { |
| | 995 | | // Only override i965 since it has lower priority than iHD in libva lookup. |
| 0 | 996 | | Environment.SetEnvironmentVariable("LIBVA_DRIVER_NAME", "i965"); |
| 0 | 997 | | Environment.SetEnvironmentVariable("LIBVA_DRIVER_NAME_JELLYFIN", "i965"); |
| 0 | 998 | | args.Append(GetVaapiDeviceArgs(options.VaapiDevice, "i965", null, null, VaapiAlias)); |
| | 999 | | } |
| | 1000 | |
|
| 0 | 1001 | | var filterDevArgs = string.Empty; |
| 0 | 1002 | | var doOclTonemap = isHwTonemapAvailable && IsOpenclFullSupported(); |
| | 1003 | |
|
| 0 | 1004 | | if (_mediaEncoder.IsVaapiDeviceInteliHD || _mediaEncoder.IsVaapiDeviceInteli965) |
| | 1005 | | { |
| 0 | 1006 | | if (doOclTonemap && !isVaapiDecoder) |
| | 1007 | | { |
| 0 | 1008 | | args.Append(GetOpenclDeviceArgs(0, null, VaapiAlias, OpenclAlias)); |
| 0 | 1009 | | filterDevArgs = GetFilterHwDeviceArgs(OpenclAlias); |
| | 1010 | | } |
| | 1011 | | } |
| 0 | 1012 | | else if (_mediaEncoder.IsVaapiDeviceAmd) |
| | 1013 | | { |
| | 1014 | | // Disable AMD EFC feature since it's still unstable in upstream Mesa. |
| 0 | 1015 | | Environment.SetEnvironmentVariable("AMD_DEBUG", "noefc"); |
| | 1016 | |
|
| 0 | 1017 | | if (IsVulkanFullSupported() |
| 0 | 1018 | | && _mediaEncoder.IsVaapiDeviceSupportVulkanDrmInterop |
| 0 | 1019 | | && Environment.OSVersion.Version >= _minKernelVersionAmdVkFmtModifier) |
| | 1020 | | { |
| 0 | 1021 | | args.Append(GetDrmDeviceArgs(options.VaapiDevice, DrmAlias)); |
| 0 | 1022 | | args.Append(GetVaapiDeviceArgs(null, null, null, DrmAlias, VaapiAlias)); |
| 0 | 1023 | | args.Append(GetVulkanDeviceArgs(0, null, DrmAlias, VulkanAlias)); |
| | 1024 | |
|
| | 1025 | | // libplacebo wants an explicitly set vulkan filter device. |
| 0 | 1026 | | filterDevArgs = GetFilterHwDeviceArgs(VulkanAlias); |
| | 1027 | | } |
| | 1028 | | else |
| | 1029 | | { |
| 0 | 1030 | | args.Append(GetVaapiDeviceArgs(options.VaapiDevice, null, null, null, VaapiAlias)); |
| 0 | 1031 | | filterDevArgs = GetFilterHwDeviceArgs(VaapiAlias); |
| | 1032 | |
|
| 0 | 1033 | | if (doOclTonemap) |
| | 1034 | | { |
| | 1035 | | // ROCm/ROCr OpenCL runtime |
| 0 | 1036 | | args.Append(GetOpenclDeviceArgs(0, "Advanced Micro Devices", null, OpenclAlias)); |
| 0 | 1037 | | filterDevArgs = GetFilterHwDeviceArgs(OpenclAlias); |
| | 1038 | | } |
| | 1039 | | } |
| | 1040 | | } |
| 0 | 1041 | | else if (doOclTonemap) |
| | 1042 | | { |
| 0 | 1043 | | args.Append(GetOpenclDeviceArgs(0, null, null, OpenclAlias)); |
| 0 | 1044 | | filterDevArgs = GetFilterHwDeviceArgs(OpenclAlias); |
| | 1045 | | } |
| | 1046 | |
|
| 0 | 1047 | | args.Append(filterDevArgs); |
| | 1048 | | } |
| 0 | 1049 | | else if (optHwaccelType == HardwareAccelerationType.qsv) |
| | 1050 | | { |
| 0 | 1051 | | if ((!isLinux && !isWindows) || !_mediaEncoder.SupportsHwaccel("qsv")) |
| | 1052 | | { |
| 0 | 1053 | | return string.Empty; |
| | 1054 | | } |
| | 1055 | |
|
| 0 | 1056 | | var isD3d11vaDecoder = vidDecoder.Contains("d3d11va", StringComparison.OrdinalIgnoreCase); |
| 0 | 1057 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 1058 | | var isQsvDecoder = vidDecoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 1059 | | var isQsvEncoder = vidEncoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 1060 | | var isHwDecoder = isQsvDecoder || isVaapiDecoder || isD3d11vaDecoder; |
| 0 | 1061 | | if (!isHwDecoder && !isQsvEncoder) |
| | 1062 | | { |
| 0 | 1063 | | return string.Empty; |
| | 1064 | | } |
| | 1065 | |
|
| 0 | 1066 | | args.Append(GetQsvDeviceArgs(options.QsvDevice, QsvAlias)); |
| 0 | 1067 | | var filterDevArgs = GetFilterHwDeviceArgs(QsvAlias); |
| | 1068 | | // child device used by qsv. |
| 0 | 1069 | | if (_mediaEncoder.SupportsHwaccel("vaapi") || _mediaEncoder.SupportsHwaccel("d3d11va")) |
| | 1070 | | { |
| 0 | 1071 | | if (isHwTonemapAvailable && IsOpenclFullSupported()) |
| | 1072 | | { |
| 0 | 1073 | | var srcAlias = isLinux ? VaapiAlias : D3d11vaAlias; |
| 0 | 1074 | | args.Append(GetOpenclDeviceArgs(0, null, srcAlias, OpenclAlias)); |
| 0 | 1075 | | if (!isHwDecoder) |
| | 1076 | | { |
| 0 | 1077 | | filterDevArgs = GetFilterHwDeviceArgs(OpenclAlias); |
| | 1078 | | } |
| | 1079 | | } |
| | 1080 | | } |
| | 1081 | |
|
| 0 | 1082 | | args.Append(filterDevArgs); |
| | 1083 | | } |
| 0 | 1084 | | else if (optHwaccelType == HardwareAccelerationType.nvenc) |
| | 1085 | | { |
| 0 | 1086 | | if ((!isLinux && !isWindows) || !IsCudaFullSupported()) |
| | 1087 | | { |
| 0 | 1088 | | return string.Empty; |
| | 1089 | | } |
| | 1090 | |
|
| 0 | 1091 | | var isCuvidDecoder = vidDecoder.Contains("cuvid", StringComparison.OrdinalIgnoreCase); |
| 0 | 1092 | | var isNvdecDecoder = vidDecoder.Contains("cuda", StringComparison.OrdinalIgnoreCase); |
| 0 | 1093 | | var isNvencEncoder = vidEncoder.Contains("nvenc", StringComparison.OrdinalIgnoreCase); |
| 0 | 1094 | | var isHwDecoder = isNvdecDecoder || isCuvidDecoder; |
| 0 | 1095 | | if (!isHwDecoder && !isNvencEncoder) |
| | 1096 | | { |
| 0 | 1097 | | return string.Empty; |
| | 1098 | | } |
| | 1099 | |
|
| 0 | 1100 | | args.Append(GetCudaDeviceArgs(0, CudaAlias)) |
| 0 | 1101 | | .Append(GetFilterHwDeviceArgs(CudaAlias)); |
| | 1102 | | } |
| 0 | 1103 | | else if (optHwaccelType == HardwareAccelerationType.amf) |
| | 1104 | | { |
| 0 | 1105 | | if (!isWindows || !_mediaEncoder.SupportsHwaccel("d3d11va")) |
| | 1106 | | { |
| 0 | 1107 | | return string.Empty; |
| | 1108 | | } |
| | 1109 | |
|
| 0 | 1110 | | var isD3d11vaDecoder = vidDecoder.Contains("d3d11va", StringComparison.OrdinalIgnoreCase); |
| 0 | 1111 | | var isAmfEncoder = vidEncoder.Contains("amf", StringComparison.OrdinalIgnoreCase); |
| 0 | 1112 | | if (!isD3d11vaDecoder && !isAmfEncoder) |
| | 1113 | | { |
| 0 | 1114 | | return string.Empty; |
| | 1115 | | } |
| | 1116 | |
|
| | 1117 | | // no dxva video processor hw filter. |
| 0 | 1118 | | args.Append(GetD3d11vaDeviceArgs(0, "0x1002", D3d11vaAlias)); |
| 0 | 1119 | | var filterDevArgs = string.Empty; |
| 0 | 1120 | | if (IsOpenclFullSupported()) |
| | 1121 | | { |
| 0 | 1122 | | args.Append(GetOpenclDeviceArgs(0, null, D3d11vaAlias, OpenclAlias)); |
| 0 | 1123 | | filterDevArgs = GetFilterHwDeviceArgs(OpenclAlias); |
| | 1124 | | } |
| | 1125 | |
|
| 0 | 1126 | | args.Append(filterDevArgs); |
| | 1127 | | } |
| 0 | 1128 | | else if (optHwaccelType == HardwareAccelerationType.videotoolbox) |
| | 1129 | | { |
| 0 | 1130 | | if (!isMacOS || !_mediaEncoder.SupportsHwaccel("videotoolbox")) |
| | 1131 | | { |
| 0 | 1132 | | return string.Empty; |
| | 1133 | | } |
| | 1134 | |
|
| 0 | 1135 | | var isVideotoolboxDecoder = vidDecoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| 0 | 1136 | | var isVideotoolboxEncoder = vidEncoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| 0 | 1137 | | if (!isVideotoolboxDecoder && !isVideotoolboxEncoder) |
| | 1138 | | { |
| 0 | 1139 | | return string.Empty; |
| | 1140 | | } |
| | 1141 | |
|
| | 1142 | | // videotoolbox hw filter does not require device selection |
| 0 | 1143 | | args.Append(GetVideoToolboxDeviceArgs(VideotoolboxAlias)); |
| | 1144 | | } |
| 0 | 1145 | | else if (optHwaccelType == HardwareAccelerationType.rkmpp) |
| | 1146 | | { |
| 0 | 1147 | | if (!isLinux || !_mediaEncoder.SupportsHwaccel("rkmpp")) |
| | 1148 | | { |
| 0 | 1149 | | return string.Empty; |
| | 1150 | | } |
| | 1151 | |
|
| 0 | 1152 | | var isRkmppDecoder = vidDecoder.Contains("rkmpp", StringComparison.OrdinalIgnoreCase); |
| 0 | 1153 | | var isRkmppEncoder = vidEncoder.Contains("rkmpp", StringComparison.OrdinalIgnoreCase); |
| 0 | 1154 | | if (!isRkmppDecoder && !isRkmppEncoder) |
| | 1155 | | { |
| 0 | 1156 | | return string.Empty; |
| | 1157 | | } |
| | 1158 | |
|
| 0 | 1159 | | args.Append(GetRkmppDeviceArgs(RkmppAlias)); |
| | 1160 | |
|
| 0 | 1161 | | var filterDevArgs = string.Empty; |
| 0 | 1162 | | var doOclTonemap = isHwTonemapAvailable && IsOpenclFullSupported(); |
| | 1163 | |
|
| 0 | 1164 | | if (doOclTonemap && !isRkmppDecoder) |
| | 1165 | | { |
| 0 | 1166 | | args.Append(GetOpenclDeviceArgs(0, null, RkmppAlias, OpenclAlias)); |
| 0 | 1167 | | filterDevArgs = GetFilterHwDeviceArgs(OpenclAlias); |
| | 1168 | | } |
| | 1169 | |
|
| 0 | 1170 | | args.Append(filterDevArgs); |
| | 1171 | | } |
| | 1172 | |
|
| 0 | 1173 | | if (!string.IsNullOrEmpty(vidDecoder)) |
| | 1174 | | { |
| 0 | 1175 | | args.Append(vidDecoder); |
| | 1176 | | } |
| | 1177 | |
|
| 0 | 1178 | | return args.ToString().Trim(); |
| | 1179 | | } |
| | 1180 | |
|
| | 1181 | | /// <summary> |
| | 1182 | | /// Gets the input argument. |
| | 1183 | | /// </summary> |
| | 1184 | | /// <param name="state">Encoding state.</param> |
| | 1185 | | /// <param name="options">Encoding options.</param> |
| | 1186 | | /// <param name="segmentContainer">Segment Container.</param> |
| | 1187 | | /// <returns>Input arguments.</returns> |
| | 1188 | | public string GetInputArgument(EncodingJobInfo state, EncodingOptions options, string segmentContainer) |
| | 1189 | | { |
| 0 | 1190 | | var arg = new StringBuilder(); |
| 0 | 1191 | | var inputVidHwaccelArgs = GetInputVideoHwaccelArgs(state, options); |
| | 1192 | |
|
| 0 | 1193 | | if (!string.IsNullOrEmpty(inputVidHwaccelArgs)) |
| | 1194 | | { |
| 0 | 1195 | | arg.Append(inputVidHwaccelArgs); |
| | 1196 | | } |
| | 1197 | |
|
| 0 | 1198 | | var canvasArgs = GetGraphicalSubCanvasSize(state); |
| 0 | 1199 | | if (!string.IsNullOrEmpty(canvasArgs)) |
| | 1200 | | { |
| 0 | 1201 | | arg.Append(canvasArgs); |
| | 1202 | | } |
| | 1203 | |
|
| 0 | 1204 | | if (state.MediaSource.VideoType == VideoType.Dvd || state.MediaSource.VideoType == VideoType.BluRay) |
| | 1205 | | { |
| 0 | 1206 | | var concatFilePath = Path.Join(_configurationManager.CommonApplicationPaths.CachePath, "concat", state.M |
| 0 | 1207 | | if (!File.Exists(concatFilePath)) |
| | 1208 | | { |
| 0 | 1209 | | _mediaEncoder.GenerateConcatConfig(state.MediaSource, concatFilePath); |
| | 1210 | | } |
| | 1211 | |
|
| 0 | 1212 | | arg.Append(" -f concat -safe 0 -i \"") |
| 0 | 1213 | | .Append(concatFilePath) |
| 0 | 1214 | | .Append("\" "); |
| | 1215 | | } |
| | 1216 | | else |
| | 1217 | | { |
| 0 | 1218 | | arg.Append(" -i ") |
| 0 | 1219 | | .Append(_mediaEncoder.GetInputPathArgument(state)); |
| | 1220 | | } |
| | 1221 | |
|
| | 1222 | | // sub2video for external graphical subtitles |
| 0 | 1223 | | if (state.SubtitleStream is not null |
| 0 | 1224 | | && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode |
| 0 | 1225 | | && !state.SubtitleStream.IsTextSubtitleStream |
| 0 | 1226 | | && state.SubtitleStream.IsExternal) |
| | 1227 | | { |
| 0 | 1228 | | var subtitlePath = state.SubtitleStream.Path; |
| 0 | 1229 | | var subtitleExtension = Path.GetExtension(subtitlePath.AsSpan()); |
| | 1230 | |
|
| | 1231 | | // dvdsub/vobsub graphical subtitles use .sub+.idx pairs |
| 0 | 1232 | | if (subtitleExtension.Equals(".sub", StringComparison.OrdinalIgnoreCase)) |
| | 1233 | | { |
| 0 | 1234 | | var idxFile = Path.ChangeExtension(subtitlePath, ".idx"); |
| 0 | 1235 | | if (File.Exists(idxFile)) |
| | 1236 | | { |
| 0 | 1237 | | subtitlePath = idxFile; |
| | 1238 | | } |
| | 1239 | | } |
| | 1240 | |
|
| | 1241 | | // Also seek the external subtitles stream. |
| 0 | 1242 | | var seekSubParam = GetFastSeekCommandLineParameter(state, options, segmentContainer); |
| 0 | 1243 | | if (!string.IsNullOrEmpty(seekSubParam)) |
| | 1244 | | { |
| 0 | 1245 | | arg.Append(' ').Append(seekSubParam); |
| | 1246 | | } |
| | 1247 | |
|
| 0 | 1248 | | if (!string.IsNullOrEmpty(canvasArgs)) |
| | 1249 | | { |
| 0 | 1250 | | arg.Append(canvasArgs); |
| | 1251 | | } |
| | 1252 | |
|
| 0 | 1253 | | arg.Append(" -i file:\"").Append(subtitlePath).Append('\"'); |
| | 1254 | | } |
| | 1255 | |
|
| 0 | 1256 | | if (state.AudioStream is not null && state.AudioStream.IsExternal) |
| | 1257 | | { |
| | 1258 | | // Also seek the external audio stream. |
| 0 | 1259 | | var seekAudioParam = GetFastSeekCommandLineParameter(state, options, segmentContainer); |
| 0 | 1260 | | if (!string.IsNullOrEmpty(seekAudioParam)) |
| | 1261 | | { |
| 0 | 1262 | | arg.Append(' ').Append(seekAudioParam); |
| | 1263 | | } |
| | 1264 | |
|
| 0 | 1265 | | arg.Append(" -i \"").Append(state.AudioStream.Path).Append('"'); |
| | 1266 | | } |
| | 1267 | |
|
| | 1268 | | // Disable auto inserted SW scaler for HW decoders in case of changed resolution. |
| 0 | 1269 | | var isSwDecoder = string.IsNullOrEmpty(GetHardwareVideoDecoder(state, options)); |
| 0 | 1270 | | if (!isSwDecoder) |
| | 1271 | | { |
| 0 | 1272 | | arg.Append(" -noautoscale"); |
| | 1273 | | } |
| | 1274 | |
|
| 0 | 1275 | | return arg.ToString(); |
| | 1276 | | } |
| | 1277 | |
|
| | 1278 | | /// <summary> |
| | 1279 | | /// Determines whether the specified stream is H264. |
| | 1280 | | /// </summary> |
| | 1281 | | /// <param name="stream">The stream.</param> |
| | 1282 | | /// <returns><c>true</c> if the specified stream is H264; otherwise, <c>false</c>.</returns> |
| | 1283 | | public static bool IsH264(MediaStream stream) |
| | 1284 | | { |
| 0 | 1285 | | var codec = stream.Codec ?? string.Empty; |
| | 1286 | |
|
| 0 | 1287 | | return codec.Contains("264", StringComparison.OrdinalIgnoreCase) |
| 0 | 1288 | | || codec.Contains("avc", StringComparison.OrdinalIgnoreCase); |
| | 1289 | | } |
| | 1290 | |
|
| | 1291 | | public static bool IsH265(MediaStream stream) |
| | 1292 | | { |
| 0 | 1293 | | var codec = stream.Codec ?? string.Empty; |
| | 1294 | |
|
| 0 | 1295 | | return codec.Contains("265", StringComparison.OrdinalIgnoreCase) |
| 0 | 1296 | | || codec.Contains("hevc", StringComparison.OrdinalIgnoreCase); |
| | 1297 | | } |
| | 1298 | |
|
| | 1299 | | public static bool IsAAC(MediaStream stream) |
| | 1300 | | { |
| 0 | 1301 | | var codec = stream.Codec ?? string.Empty; |
| | 1302 | |
|
| 0 | 1303 | | return codec.Contains("aac", StringComparison.OrdinalIgnoreCase); |
| | 1304 | | } |
| | 1305 | |
|
| | 1306 | | public static string GetBitStreamArgs(MediaStream stream) |
| | 1307 | | { |
| | 1308 | | // TODO This is auto inserted into the mpegts mux so it might not be needed. |
| | 1309 | | // https://www.ffmpeg.org/ffmpeg-bitstream-filters.html#h264_005fmp4toannexb |
| 0 | 1310 | | if (IsH264(stream)) |
| | 1311 | | { |
| 0 | 1312 | | return "-bsf:v h264_mp4toannexb"; |
| | 1313 | | } |
| | 1314 | |
|
| 0 | 1315 | | if (IsH265(stream)) |
| | 1316 | | { |
| 0 | 1317 | | return "-bsf:v hevc_mp4toannexb"; |
| | 1318 | | } |
| | 1319 | |
|
| 0 | 1320 | | if (IsAAC(stream)) |
| | 1321 | | { |
| | 1322 | | // Convert adts header(mpegts) to asc header(mp4). |
| 0 | 1323 | | return "-bsf:a aac_adtstoasc"; |
| | 1324 | | } |
| | 1325 | |
|
| 0 | 1326 | | return null; |
| | 1327 | | } |
| | 1328 | |
|
| | 1329 | | public static string GetAudioBitStreamArguments(EncodingJobInfo state, string segmentContainer, string mediaSour |
| | 1330 | | { |
| 0 | 1331 | | var bitStreamArgs = string.Empty; |
| 0 | 1332 | | var segmentFormat = GetSegmentFileExtension(segmentContainer).TrimStart('.'); |
| | 1333 | |
|
| | 1334 | | // Apply aac_adtstoasc bitstream filter when media source is in mpegts. |
| 0 | 1335 | | if (string.Equals(segmentFormat, "mp4", StringComparison.OrdinalIgnoreCase) |
| 0 | 1336 | | && (string.Equals(mediaSourceContainer, "ts", StringComparison.OrdinalIgnoreCase) |
| 0 | 1337 | | || string.Equals(mediaSourceContainer, "aac", StringComparison.OrdinalIgnoreCase) |
| 0 | 1338 | | || string.Equals(mediaSourceContainer, "hls", StringComparison.OrdinalIgnoreCase))) |
| | 1339 | | { |
| 0 | 1340 | | bitStreamArgs = GetBitStreamArgs(state.AudioStream); |
| 0 | 1341 | | bitStreamArgs = string.IsNullOrEmpty(bitStreamArgs) ? string.Empty : " " + bitStreamArgs; |
| | 1342 | | } |
| | 1343 | |
|
| 0 | 1344 | | return bitStreamArgs; |
| | 1345 | | } |
| | 1346 | |
|
| | 1347 | | public static string GetSegmentFileExtension(string segmentContainer) |
| | 1348 | | { |
| 0 | 1349 | | if (!string.IsNullOrWhiteSpace(segmentContainer)) |
| | 1350 | | { |
| 0 | 1351 | | return "." + segmentContainer; |
| | 1352 | | } |
| | 1353 | |
|
| 0 | 1354 | | return ".ts"; |
| | 1355 | | } |
| | 1356 | |
|
| | 1357 | | private string GetVideoBitrateParam(EncodingJobInfo state, string videoCodec) |
| | 1358 | | { |
| 0 | 1359 | | if (state.OutputVideoBitrate is null) |
| | 1360 | | { |
| 0 | 1361 | | return string.Empty; |
| | 1362 | | } |
| | 1363 | |
|
| 0 | 1364 | | int bitrate = state.OutputVideoBitrate.Value; |
| | 1365 | |
|
| | 1366 | | // Bit rate under 1000k is not allowed in h264_qsv |
| 0 | 1367 | | if (string.Equals(videoCodec, "h264_qsv", StringComparison.OrdinalIgnoreCase)) |
| | 1368 | | { |
| 0 | 1369 | | bitrate = Math.Max(bitrate, 1000); |
| | 1370 | | } |
| | 1371 | |
|
| | 1372 | | // Currently use the same buffer size for all encoders |
| 0 | 1373 | | int bufsize = bitrate * 2; |
| | 1374 | |
|
| 0 | 1375 | | if (string.Equals(videoCodec, "libsvtav1", StringComparison.OrdinalIgnoreCase)) |
| | 1376 | | { |
| 0 | 1377 | | return FormattableString.Invariant($" -b:v {bitrate} -bufsize {bufsize}"); |
| | 1378 | | } |
| | 1379 | |
|
| 0 | 1380 | | if (string.Equals(videoCodec, "libx264", StringComparison.OrdinalIgnoreCase) |
| 0 | 1381 | | || string.Equals(videoCodec, "libx265", StringComparison.OrdinalIgnoreCase)) |
| | 1382 | | { |
| 0 | 1383 | | return FormattableString.Invariant($" -maxrate {bitrate} -bufsize {bufsize}"); |
| | 1384 | | } |
| | 1385 | |
|
| 0 | 1386 | | if (string.Equals(videoCodec, "h264_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1387 | | || string.Equals(videoCodec, "hevc_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1388 | | || string.Equals(videoCodec, "av1_amf", StringComparison.OrdinalIgnoreCase)) |
| | 1389 | | { |
| | 1390 | | // Override the too high default qmin 18 in transcoding preset |
| 0 | 1391 | | return FormattableString.Invariant($" -rc cbr -qmin 0 -qmax 32 -b:v {bitrate} -maxrate {bitrate} -bufsiz |
| | 1392 | | } |
| | 1393 | |
|
| 0 | 1394 | | if (string.Equals(videoCodec, "h264_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1395 | | || string.Equals(videoCodec, "hevc_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1396 | | || string.Equals(videoCodec, "av1_vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 1397 | | { |
| | 1398 | | // VBR in i965 driver may result in pixelated output. |
| 0 | 1399 | | if (_mediaEncoder.IsVaapiDeviceInteli965) |
| | 1400 | | { |
| 0 | 1401 | | return FormattableString.Invariant($" -rc_mode CBR -b:v {bitrate} -maxrate {bitrate} -bufsize {bufsi |
| | 1402 | | } |
| | 1403 | |
|
| 0 | 1404 | | return FormattableString.Invariant($" -rc_mode VBR -b:v {bitrate} -maxrate {bitrate} -bufsize {bufsize}" |
| | 1405 | | } |
| | 1406 | |
|
| 0 | 1407 | | if (string.Equals(videoCodec, "h264_videotoolbox", StringComparison.OrdinalIgnoreCase) |
| 0 | 1408 | | || string.Equals(videoCodec, "hevc_videotoolbox", StringComparison.OrdinalIgnoreCase)) |
| | 1409 | | { |
| | 1410 | | // The `maxrate` and `bufsize` options can potentially lead to performance regression |
| | 1411 | | // and even encoder hangs, especially when the value is very high. |
| 0 | 1412 | | return FormattableString.Invariant($" -b:v {bitrate} -qmin -1 -qmax -1"); |
| | 1413 | | } |
| | 1414 | |
|
| 0 | 1415 | | return FormattableString.Invariant($" -b:v {bitrate} -maxrate {bitrate} -bufsize {bufsize}"); |
| | 1416 | | } |
| | 1417 | |
|
| | 1418 | | private string GetEncoderParam(EncoderPreset? preset, EncoderPreset defaultPreset, EncodingOptions encodingOptio |
| | 1419 | | { |
| 0 | 1420 | | var param = string.Empty; |
| 0 | 1421 | | var encoderPreset = preset ?? defaultPreset; |
| 0 | 1422 | | if (string.Equals(videoEncoder, "libx264", StringComparison.OrdinalIgnoreCase) || isLibX265) |
| | 1423 | | { |
| 0 | 1424 | | param += " -preset " + encoderPreset.ToString().ToLowerInvariant(); |
| | 1425 | |
|
| 0 | 1426 | | int encodeCrf = encodingOptions.H264Crf; |
| 0 | 1427 | | if (isLibX265) |
| | 1428 | | { |
| 0 | 1429 | | encodeCrf = encodingOptions.H265Crf; |
| | 1430 | | } |
| | 1431 | |
|
| 0 | 1432 | | if (encodeCrf >= 0 && encodeCrf <= 51) |
| | 1433 | | { |
| 0 | 1434 | | param += " -crf " + encodeCrf.ToString(CultureInfo.InvariantCulture); |
| | 1435 | | } |
| | 1436 | | else |
| | 1437 | | { |
| 0 | 1438 | | string defaultCrf = "23"; |
| 0 | 1439 | | if (isLibX265) |
| | 1440 | | { |
| 0 | 1441 | | defaultCrf = "28"; |
| | 1442 | | } |
| | 1443 | |
|
| 0 | 1444 | | param += " -crf " + defaultCrf; |
| | 1445 | | } |
| | 1446 | | } |
| 0 | 1447 | | else if (string.Equals(videoEncoder, "libsvtav1", StringComparison.OrdinalIgnoreCase)) |
| | 1448 | | { |
| | 1449 | | // Default to use the recommended preset 10. |
| | 1450 | | // Omit presets < 5, which are too slow for on the fly encoding. |
| | 1451 | | // https://gitlab.com/AOMediaCodec/SVT-AV1/-/blob/master/Docs/Ffmpeg.md |
| 0 | 1452 | | param += encoderPreset switch |
| 0 | 1453 | | { |
| 0 | 1454 | | EncoderPreset.veryslow => " -preset 5", |
| 0 | 1455 | | EncoderPreset.slower => " -preset 6", |
| 0 | 1456 | | EncoderPreset.slow => " -preset 7", |
| 0 | 1457 | | EncoderPreset.medium => " -preset 8", |
| 0 | 1458 | | EncoderPreset.fast => " -preset 9", |
| 0 | 1459 | | EncoderPreset.faster => " -preset 10", |
| 0 | 1460 | | EncoderPreset.veryfast => " -preset 11", |
| 0 | 1461 | | EncoderPreset.superfast => " -preset 12", |
| 0 | 1462 | | EncoderPreset.ultrafast => " -preset 13", |
| 0 | 1463 | | _ => " -preset 10" |
| 0 | 1464 | | }; |
| | 1465 | | } |
| 0 | 1466 | | else if (string.Equals(videoEncoder, "h264_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1467 | | || string.Equals(videoEncoder, "hevc_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1468 | | || string.Equals(videoEncoder, "av1_vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 1469 | | { |
| | 1470 | | // -compression_level is not reliable on AMD. |
| 0 | 1471 | | if (_mediaEncoder.IsVaapiDeviceInteliHD) |
| | 1472 | | { |
| 0 | 1473 | | param += encoderPreset switch |
| 0 | 1474 | | { |
| 0 | 1475 | | EncoderPreset.veryslow => " -compression_level 1", |
| 0 | 1476 | | EncoderPreset.slower => " -compression_level 2", |
| 0 | 1477 | | EncoderPreset.slow => " -compression_level 3", |
| 0 | 1478 | | EncoderPreset.medium => " -compression_level 4", |
| 0 | 1479 | | EncoderPreset.fast => " -compression_level 5", |
| 0 | 1480 | | EncoderPreset.faster => " -compression_level 6", |
| 0 | 1481 | | EncoderPreset.veryfast => " -compression_level 7", |
| 0 | 1482 | | EncoderPreset.superfast => " -compression_level 7", |
| 0 | 1483 | | EncoderPreset.ultrafast => " -compression_level 7", |
| 0 | 1484 | | _ => string.Empty |
| 0 | 1485 | | }; |
| | 1486 | | } |
| | 1487 | | } |
| 0 | 1488 | | else if (string.Equals(videoEncoder, "h264_qsv", StringComparison.OrdinalIgnoreCase) // h264 (h264_qsv) |
| 0 | 1489 | | || string.Equals(videoEncoder, "hevc_qsv", StringComparison.OrdinalIgnoreCase) // hevc (hevc_qsv) |
| 0 | 1490 | | || string.Equals(videoEncoder, "av1_qsv", StringComparison.OrdinalIgnoreCase)) // av1 (av1_qsv) |
| | 1491 | | { |
| 0 | 1492 | | EncoderPreset[] valid_presets = [EncoderPreset.veryslow, EncoderPreset.slower, EncoderPreset.slow, Encod |
| | 1493 | |
|
| 0 | 1494 | | param += " -preset " + (valid_presets.Contains(encoderPreset) ? encoderPreset : EncoderPreset.veryfast). |
| | 1495 | | } |
| 0 | 1496 | | else if (string.Equals(videoEncoder, "h264_nvenc", StringComparison.OrdinalIgnoreCase) // h264 (h264_nvenc) |
| 0 | 1497 | | || string.Equals(videoEncoder, "hevc_nvenc", StringComparison.OrdinalIgnoreCase) // hevc (hevc_n |
| 0 | 1498 | | || string.Equals(videoEncoder, "av1_nvenc", StringComparison.OrdinalIgnoreCase) // av1 (av1_nven |
| 0 | 1499 | | ) |
| | 1500 | | { |
| 0 | 1501 | | param += encoderPreset switch |
| 0 | 1502 | | { |
| 0 | 1503 | | EncoderPreset.veryslow => " -preset p7", |
| 0 | 1504 | | EncoderPreset.slower => " -preset p6", |
| 0 | 1505 | | EncoderPreset.slow => " -preset p5", |
| 0 | 1506 | | EncoderPreset.medium => " -preset p4", |
| 0 | 1507 | | EncoderPreset.fast => " -preset p3", |
| 0 | 1508 | | EncoderPreset.faster => " -preset p2", |
| 0 | 1509 | | _ => " -preset p1" |
| 0 | 1510 | | }; |
| | 1511 | | } |
| 0 | 1512 | | else if (string.Equals(videoEncoder, "h264_amf", StringComparison.OrdinalIgnoreCase) // h264 (h264_amf) |
| 0 | 1513 | | || string.Equals(videoEncoder, "hevc_amf", StringComparison.OrdinalIgnoreCase) // hevc (hevc_amf |
| 0 | 1514 | | || string.Equals(videoEncoder, "av1_amf", StringComparison.OrdinalIgnoreCase) // av1 (av1_amf) |
| 0 | 1515 | | ) |
| | 1516 | | { |
| 0 | 1517 | | param += encoderPreset switch |
| 0 | 1518 | | { |
| 0 | 1519 | | EncoderPreset.veryslow => " -quality quality", |
| 0 | 1520 | | EncoderPreset.slower => " -quality quality", |
| 0 | 1521 | | EncoderPreset.slow => " -quality quality", |
| 0 | 1522 | | EncoderPreset.medium => " -quality balanced", |
| 0 | 1523 | | _ => " -quality speed" |
| 0 | 1524 | | }; |
| | 1525 | |
|
| 0 | 1526 | | if (string.Equals(videoEncoder, "hevc_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1527 | | || string.Equals(videoEncoder, "av1_amf", StringComparison.OrdinalIgnoreCase)) |
| | 1528 | | { |
| 0 | 1529 | | param += " -header_insertion_mode gop"; |
| | 1530 | | } |
| | 1531 | |
|
| 0 | 1532 | | if (string.Equals(videoEncoder, "hevc_amf", StringComparison.OrdinalIgnoreCase)) |
| | 1533 | | { |
| 0 | 1534 | | param += " -gops_per_idr 1"; |
| | 1535 | | } |
| | 1536 | | } |
| 0 | 1537 | | else if (string.Equals(videoEncoder, "h264_videotoolbox", StringComparison.OrdinalIgnoreCase) // h264 (h264_ |
| 0 | 1538 | | || string.Equals(videoEncoder, "hevc_videotoolbox", StringComparison.OrdinalIgnoreCase) // hevc |
| 0 | 1539 | | ) |
| | 1540 | | { |
| 0 | 1541 | | param += encoderPreset switch |
| 0 | 1542 | | { |
| 0 | 1543 | | EncoderPreset.veryslow => " -prio_speed 0", |
| 0 | 1544 | | EncoderPreset.slower => " -prio_speed 0", |
| 0 | 1545 | | EncoderPreset.slow => " -prio_speed 0", |
| 0 | 1546 | | EncoderPreset.medium => " -prio_speed 0", |
| 0 | 1547 | | _ => " -prio_speed 1" |
| 0 | 1548 | | }; |
| | 1549 | | } |
| | 1550 | |
|
| 0 | 1551 | | return param; |
| | 1552 | | } |
| | 1553 | |
|
| | 1554 | | public static string NormalizeTranscodingLevel(EncodingJobInfo state, string level) |
| | 1555 | | { |
| 0 | 1556 | | if (double.TryParse(level, CultureInfo.InvariantCulture, out double requestLevel)) |
| | 1557 | | { |
| 0 | 1558 | | if (string.Equals(state.ActualOutputVideoCodec, "av1", StringComparison.OrdinalIgnoreCase)) |
| | 1559 | | { |
| | 1560 | | // Transcode to level 5.3 (15) and lower for maximum compatibility. |
| | 1561 | | // https://en.wikipedia.org/wiki/AV1#Levels |
| 0 | 1562 | | if (requestLevel < 0 || requestLevel >= 15) |
| | 1563 | | { |
| 0 | 1564 | | return "15"; |
| | 1565 | | } |
| | 1566 | | } |
| 0 | 1567 | | else if (string.Equals(state.ActualOutputVideoCodec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1568 | | || string.Equals(state.ActualOutputVideoCodec, "h265", StringComparison.OrdinalIgnoreCase)) |
| | 1569 | | { |
| | 1570 | | // Transcode to level 5.0 and lower for maximum compatibility. |
| | 1571 | | // Level 5.0 is suitable for up to 4k 30fps hevc encoding, otherwise let the encoder to handle it. |
| | 1572 | | // https://en.wikipedia.org/wiki/High_Efficiency_Video_Coding_tiers_and_levels |
| | 1573 | | // MaxLumaSampleRate = 3840*2160*30 = 248832000 < 267386880. |
| 0 | 1574 | | if (requestLevel < 0 || requestLevel >= 150) |
| | 1575 | | { |
| 0 | 1576 | | return "150"; |
| | 1577 | | } |
| | 1578 | | } |
| 0 | 1579 | | else if (string.Equals(state.ActualOutputVideoCodec, "h264", StringComparison.OrdinalIgnoreCase)) |
| | 1580 | | { |
| | 1581 | | // Transcode to level 5.1 and lower for maximum compatibility. |
| | 1582 | | // h264 4k 30fps requires at least level 5.1 otherwise it will break on safari fmp4. |
| | 1583 | | // https://en.wikipedia.org/wiki/Advanced_Video_Coding#Levels |
| 0 | 1584 | | if (requestLevel < 0 || requestLevel >= 51) |
| | 1585 | | { |
| 0 | 1586 | | return "51"; |
| | 1587 | | } |
| | 1588 | | } |
| | 1589 | | } |
| | 1590 | |
|
| 0 | 1591 | | return level; |
| | 1592 | | } |
| | 1593 | |
|
| | 1594 | | /// <summary> |
| | 1595 | | /// Gets the text subtitle param. |
| | 1596 | | /// </summary> |
| | 1597 | | /// <param name="state">The state.</param> |
| | 1598 | | /// <param name="enableAlpha">Enable alpha processing.</param> |
| | 1599 | | /// <param name="enableSub2video">Enable sub2video mode.</param> |
| | 1600 | | /// <returns>System.String.</returns> |
| | 1601 | | public string GetTextSubtitlesFilter(EncodingJobInfo state, bool enableAlpha, bool enableSub2video) |
| | 1602 | | { |
| 0 | 1603 | | var seconds = Math.Round(TimeSpan.FromTicks(state.StartTimeTicks ?? 0).TotalSeconds); |
| | 1604 | |
|
| | 1605 | | // hls always copies timestamps |
| 0 | 1606 | | var setPtsParam = state.CopyTimestamps || state.TranscodingType != TranscodingJobType.Progressive |
| 0 | 1607 | | ? string.Empty |
| 0 | 1608 | | : string.Format(CultureInfo.InvariantCulture, ",setpts=PTS -{0}/TB", seconds); |
| | 1609 | |
|
| 0 | 1610 | | var alphaParam = enableAlpha ? ":alpha=1" : string.Empty; |
| 0 | 1611 | | var sub2videoParam = enableSub2video ? ":sub2video=1" : string.Empty; |
| | 1612 | |
|
| 0 | 1613 | | var fontPath = Path.Combine(_appPaths.CachePath, "attachments", state.MediaSource.Id); |
| 0 | 1614 | | var fontParam = string.Format( |
| 0 | 1615 | | CultureInfo.InvariantCulture, |
| 0 | 1616 | | ":fontsdir='{0}'", |
| 0 | 1617 | | _mediaEncoder.EscapeSubtitleFilterPath(fontPath)); |
| | 1618 | |
|
| 0 | 1619 | | if (state.SubtitleStream.IsExternal) |
| | 1620 | | { |
| 0 | 1621 | | var charsetParam = string.Empty; |
| | 1622 | |
|
| 0 | 1623 | | if (!string.IsNullOrEmpty(state.SubtitleStream.Language)) |
| | 1624 | | { |
| 0 | 1625 | | var charenc = _subtitleEncoder.GetSubtitleFileCharacterSet( |
| 0 | 1626 | | state.SubtitleStream, |
| 0 | 1627 | | state.SubtitleStream.Language, |
| 0 | 1628 | | state.MediaSource, |
| 0 | 1629 | | CancellationToken.None).GetAwaiter().GetResult(); |
| | 1630 | |
|
| 0 | 1631 | | if (!string.IsNullOrEmpty(charenc)) |
| | 1632 | | { |
| 0 | 1633 | | charsetParam = ":charenc=" + charenc; |
| | 1634 | | } |
| | 1635 | | } |
| | 1636 | |
|
| 0 | 1637 | | return string.Format( |
| 0 | 1638 | | CultureInfo.InvariantCulture, |
| 0 | 1639 | | "subtitles=f='{0}'{1}{2}{3}{4}{5}", |
| 0 | 1640 | | _mediaEncoder.EscapeSubtitleFilterPath(state.SubtitleStream.Path), |
| 0 | 1641 | | charsetParam, |
| 0 | 1642 | | alphaParam, |
| 0 | 1643 | | sub2videoParam, |
| 0 | 1644 | | fontParam, |
| 0 | 1645 | | setPtsParam); |
| | 1646 | | } |
| | 1647 | |
|
| 0 | 1648 | | var subtitlePath = _subtitleEncoder.GetSubtitleFilePath( |
| 0 | 1649 | | state.SubtitleStream, |
| 0 | 1650 | | state.MediaSource, |
| 0 | 1651 | | CancellationToken.None).GetAwaiter().GetResult(); |
| | 1652 | |
|
| 0 | 1653 | | return string.Format( |
| 0 | 1654 | | CultureInfo.InvariantCulture, |
| 0 | 1655 | | "subtitles=f='{0}'{1}{2}{3}{4}", |
| 0 | 1656 | | _mediaEncoder.EscapeSubtitleFilterPath(subtitlePath), |
| 0 | 1657 | | alphaParam, |
| 0 | 1658 | | sub2videoParam, |
| 0 | 1659 | | fontParam, |
| 0 | 1660 | | setPtsParam); |
| | 1661 | | } |
| | 1662 | |
|
| | 1663 | | public double? GetFramerateParam(EncodingJobInfo state) |
| | 1664 | | { |
| 0 | 1665 | | var request = state.BaseRequest; |
| | 1666 | |
|
| 0 | 1667 | | if (request.Framerate.HasValue) |
| | 1668 | | { |
| 0 | 1669 | | return request.Framerate.Value; |
| | 1670 | | } |
| | 1671 | |
|
| 0 | 1672 | | var maxrate = request.MaxFramerate; |
| | 1673 | |
|
| 0 | 1674 | | if (maxrate.HasValue && state.VideoStream is not null) |
| | 1675 | | { |
| 0 | 1676 | | var contentRate = state.VideoStream.ReferenceFrameRate; |
| | 1677 | |
|
| 0 | 1678 | | if (contentRate.HasValue && contentRate.Value > maxrate.Value) |
| | 1679 | | { |
| 0 | 1680 | | return maxrate; |
| | 1681 | | } |
| | 1682 | | } |
| | 1683 | |
|
| 0 | 1684 | | return null; |
| | 1685 | | } |
| | 1686 | |
|
| | 1687 | | public string GetHlsVideoKeyFrameArguments( |
| | 1688 | | EncodingJobInfo state, |
| | 1689 | | string codec, |
| | 1690 | | int segmentLength, |
| | 1691 | | bool isEventPlaylist, |
| | 1692 | | int? startNumber) |
| | 1693 | | { |
| 0 | 1694 | | var args = string.Empty; |
| 0 | 1695 | | var gopArg = string.Empty; |
| | 1696 | |
|
| 0 | 1697 | | var keyFrameArg = string.Format( |
| 0 | 1698 | | CultureInfo.InvariantCulture, |
| 0 | 1699 | | " -force_key_frames:0 \"expr:gte(t,n_forced*{0})\"", |
| 0 | 1700 | | segmentLength); |
| | 1701 | |
|
| 0 | 1702 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 1703 | | if (framerate.HasValue) |
| | 1704 | | { |
| | 1705 | | // This is to make sure keyframe interval is limited to our segment, |
| | 1706 | | // as forcing keyframes is not enough. |
| | 1707 | | // Example: we encoded half of desired length, then codec detected |
| | 1708 | | // scene cut and inserted a keyframe; next forced keyframe would |
| | 1709 | | // be created outside of segment, which breaks seeking. |
| 0 | 1710 | | gopArg = string.Format( |
| 0 | 1711 | | CultureInfo.InvariantCulture, |
| 0 | 1712 | | " -g:v:0 {0} -keyint_min:v:0 {0}", |
| 0 | 1713 | | Math.Ceiling(segmentLength * framerate.Value)); |
| | 1714 | | } |
| | 1715 | |
|
| | 1716 | | // Unable to force key frames using these encoders, set key frames by GOP. |
| 0 | 1717 | | if (string.Equals(codec, "h264_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1718 | | || string.Equals(codec, "h264_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1719 | | || string.Equals(codec, "h264_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1720 | | || string.Equals(codec, "h264_rkmpp", StringComparison.OrdinalIgnoreCase) |
| 0 | 1721 | | || string.Equals(codec, "hevc_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1722 | | || string.Equals(codec, "hevc_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1723 | | || string.Equals(codec, "hevc_rkmpp", StringComparison.OrdinalIgnoreCase) |
| 0 | 1724 | | || string.Equals(codec, "av1_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1725 | | || string.Equals(codec, "av1_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1726 | | || string.Equals(codec, "av1_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1727 | | || string.Equals(codec, "libsvtav1", StringComparison.OrdinalIgnoreCase)) |
| | 1728 | | { |
| 0 | 1729 | | args += gopArg; |
| | 1730 | | } |
| 0 | 1731 | | else if (string.Equals(codec, "libx264", StringComparison.OrdinalIgnoreCase) |
| 0 | 1732 | | || string.Equals(codec, "libx265", StringComparison.OrdinalIgnoreCase) |
| 0 | 1733 | | || string.Equals(codec, "h264_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1734 | | || string.Equals(codec, "hevc_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1735 | | || string.Equals(codec, "av1_vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 1736 | | { |
| 0 | 1737 | | args += keyFrameArg; |
| | 1738 | |
|
| | 1739 | | // prevent the libx264 from post processing to break the set keyframe. |
| 0 | 1740 | | if (string.Equals(codec, "libx264", StringComparison.OrdinalIgnoreCase)) |
| | 1741 | | { |
| 0 | 1742 | | args += " -sc_threshold:v:0 0"; |
| | 1743 | | } |
| | 1744 | | } |
| | 1745 | | else |
| | 1746 | | { |
| 0 | 1747 | | args += keyFrameArg + gopArg; |
| | 1748 | | } |
| | 1749 | |
|
| | 1750 | | // global_header produced by AMD HEVC VA-API encoder causes non-playable fMP4 on iOS |
| 0 | 1751 | | if (string.Equals(codec, "hevc_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1752 | | && _mediaEncoder.IsVaapiDeviceAmd) |
| | 1753 | | { |
| 0 | 1754 | | args += " -flags:v -global_header"; |
| | 1755 | | } |
| | 1756 | |
|
| 0 | 1757 | | return args; |
| | 1758 | | } |
| | 1759 | |
|
| | 1760 | | /// <summary> |
| | 1761 | | /// Gets the video bitrate to specify on the command line. |
| | 1762 | | /// </summary> |
| | 1763 | | /// <param name="state">Encoding state.</param> |
| | 1764 | | /// <param name="videoEncoder">Video encoder to use.</param> |
| | 1765 | | /// <param name="encodingOptions">Encoding options.</param> |
| | 1766 | | /// <param name="defaultPreset">Default present to use for encoding.</param> |
| | 1767 | | /// <returns>Video bitrate.</returns> |
| | 1768 | | public string GetVideoQualityParam(EncodingJobInfo state, string videoEncoder, EncodingOptions encodingOptions, |
| | 1769 | | { |
| 0 | 1770 | | var param = string.Empty; |
| | 1771 | |
|
| | 1772 | | // Tutorials: Enable Intel GuC / HuC firmware loading for Low Power Encoding. |
| | 1773 | | // https://01.org/group/43/downloads/firmware |
| | 1774 | | // https://wiki.archlinux.org/title/intel_graphics#Enable_GuC_/_HuC_firmware_loading |
| | 1775 | | // Intel Low Power Encoding can save unnecessary CPU-GPU synchronization, |
| | 1776 | | // which will reduce overhead in performance intensive tasks such as 4k transcoding and tonemapping. |
| 0 | 1777 | | var intelLowPowerHwEncoding = false; |
| | 1778 | |
|
| | 1779 | | // Workaround for linux 5.18 to 6.1.3 i915 hang at cost of performance. |
| | 1780 | | // https://github.com/intel/media-driver/issues/1456 |
| 0 | 1781 | | var enableWaFori915Hang = false; |
| | 1782 | |
|
| 0 | 1783 | | var hardwareAccelerationType = encodingOptions.HardwareAccelerationType; |
| | 1784 | |
|
| 0 | 1785 | | if (hardwareAccelerationType == HardwareAccelerationType.vaapi) |
| | 1786 | | { |
| 0 | 1787 | | var isIntelVaapiDriver = _mediaEncoder.IsVaapiDeviceInteliHD || _mediaEncoder.IsVaapiDeviceInteli965; |
| | 1788 | |
|
| 0 | 1789 | | if (string.Equals(videoEncoder, "h264_vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 1790 | | { |
| 0 | 1791 | | intelLowPowerHwEncoding = encodingOptions.EnableIntelLowPowerH264HwEncoder && isIntelVaapiDriver; |
| | 1792 | | } |
| 0 | 1793 | | else if (string.Equals(videoEncoder, "hevc_vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 1794 | | { |
| 0 | 1795 | | intelLowPowerHwEncoding = encodingOptions.EnableIntelLowPowerHevcHwEncoder && isIntelVaapiDriver; |
| | 1796 | | } |
| | 1797 | | } |
| 0 | 1798 | | else if (hardwareAccelerationType == HardwareAccelerationType.qsv) |
| | 1799 | | { |
| 0 | 1800 | | if (OperatingSystem.IsLinux()) |
| | 1801 | | { |
| 0 | 1802 | | var ver = Environment.OSVersion.Version; |
| 0 | 1803 | | var isFixedKernel60 = ver.Major == 6 && ver.Minor == 0 && ver >= _minFixedKernel60i915Hang; |
| 0 | 1804 | | var isUnaffectedKernel = ver < _minKerneli915Hang || ver > _maxKerneli915Hang; |
| | 1805 | |
|
| 0 | 1806 | | if (!(isUnaffectedKernel || isFixedKernel60)) |
| | 1807 | | { |
| 0 | 1808 | | var vidDecoder = GetHardwareVideoDecoder(state, encodingOptions) ?? string.Empty; |
| 0 | 1809 | | var isIntelDecoder = vidDecoder.Contains("qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1810 | | || vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 1811 | | var doOclTonemap = _mediaEncoder.SupportsHwaccel("qsv") |
| 0 | 1812 | | && IsVaapiSupported(state) |
| 0 | 1813 | | && IsOpenclFullSupported() |
| 0 | 1814 | | && !IsIntelVppTonemapAvailable(state, encodingOptions) |
| 0 | 1815 | | && IsHwTonemapAvailable(state, encodingOptions); |
| | 1816 | |
|
| 0 | 1817 | | enableWaFori915Hang = isIntelDecoder && doOclTonemap; |
| | 1818 | | } |
| | 1819 | | } |
| | 1820 | |
|
| 0 | 1821 | | if (string.Equals(videoEncoder, "h264_qsv", StringComparison.OrdinalIgnoreCase)) |
| | 1822 | | { |
| 0 | 1823 | | intelLowPowerHwEncoding = encodingOptions.EnableIntelLowPowerH264HwEncoder; |
| | 1824 | | } |
| 0 | 1825 | | else if (string.Equals(videoEncoder, "hevc_qsv", StringComparison.OrdinalIgnoreCase)) |
| | 1826 | | { |
| 0 | 1827 | | intelLowPowerHwEncoding = encodingOptions.EnableIntelLowPowerHevcHwEncoder; |
| | 1828 | | } |
| | 1829 | | else |
| | 1830 | | { |
| 0 | 1831 | | enableWaFori915Hang = false; |
| | 1832 | | } |
| | 1833 | | } |
| | 1834 | |
|
| 0 | 1835 | | if (intelLowPowerHwEncoding) |
| | 1836 | | { |
| 0 | 1837 | | param += " -low_power 1"; |
| | 1838 | | } |
| | 1839 | |
|
| 0 | 1840 | | if (enableWaFori915Hang) |
| | 1841 | | { |
| 0 | 1842 | | param += " -async_depth 1"; |
| | 1843 | | } |
| | 1844 | |
|
| 0 | 1845 | | var isLibX265 = string.Equals(videoEncoder, "libx265", StringComparison.OrdinalIgnoreCase); |
| 0 | 1846 | | var encodingPreset = encodingOptions.EncoderPreset; |
| | 1847 | |
|
| 0 | 1848 | | param += GetEncoderParam(encodingPreset, defaultPreset, encodingOptions, videoEncoder, isLibX265); |
| 0 | 1849 | | param += GetVideoBitrateParam(state, videoEncoder); |
| | 1850 | |
|
| 0 | 1851 | | var framerate = GetFramerateParam(state); |
| 0 | 1852 | | if (framerate.HasValue) |
| | 1853 | | { |
| 0 | 1854 | | param += string.Format(CultureInfo.InvariantCulture, " -r {0}", framerate.Value.ToString(CultureInfo.Inv |
| | 1855 | | } |
| | 1856 | |
|
| 0 | 1857 | | var targetVideoCodec = state.ActualOutputVideoCodec; |
| 0 | 1858 | | if (string.Equals(targetVideoCodec, "h265", StringComparison.OrdinalIgnoreCase) |
| 0 | 1859 | | || string.Equals(targetVideoCodec, "hevc", StringComparison.OrdinalIgnoreCase)) |
| | 1860 | | { |
| 0 | 1861 | | targetVideoCodec = "hevc"; |
| | 1862 | | } |
| | 1863 | |
|
| 0 | 1864 | | var profile = state.GetRequestedProfiles(targetVideoCodec).FirstOrDefault() ?? string.Empty; |
| 0 | 1865 | | profile = WhiteSpaceRegex().Replace(profile, string.Empty).ToLowerInvariant(); |
| | 1866 | |
|
| 0 | 1867 | | var videoProfiles = Array.Empty<string>(); |
| 0 | 1868 | | if (string.Equals("h264", targetVideoCodec, StringComparison.OrdinalIgnoreCase)) |
| | 1869 | | { |
| 0 | 1870 | | videoProfiles = _videoProfilesH264; |
| | 1871 | | } |
| 0 | 1872 | | else if (string.Equals("hevc", targetVideoCodec, StringComparison.OrdinalIgnoreCase)) |
| | 1873 | | { |
| 0 | 1874 | | videoProfiles = _videoProfilesH265; |
| | 1875 | | } |
| 0 | 1876 | | else if (string.Equals("av1", targetVideoCodec, StringComparison.OrdinalIgnoreCase)) |
| | 1877 | | { |
| 0 | 1878 | | videoProfiles = _videoProfilesAv1; |
| | 1879 | | } |
| | 1880 | |
|
| 0 | 1881 | | if (!videoProfiles.Contains(profile, StringComparison.OrdinalIgnoreCase)) |
| | 1882 | | { |
| 0 | 1883 | | profile = string.Empty; |
| | 1884 | | } |
| | 1885 | |
|
| | 1886 | | // We only transcode to HEVC 8-bit for now, force Main Profile. |
| 0 | 1887 | | if (profile.Contains("main10", StringComparison.OrdinalIgnoreCase) |
| 0 | 1888 | | || profile.Contains("mainstill", StringComparison.OrdinalIgnoreCase)) |
| | 1889 | | { |
| 0 | 1890 | | profile = "main"; |
| | 1891 | | } |
| | 1892 | |
|
| | 1893 | | // Extended Profile is not supported by any known h264 encoders, force Main Profile. |
| 0 | 1894 | | if (profile.Contains("extended", StringComparison.OrdinalIgnoreCase)) |
| | 1895 | | { |
| 0 | 1896 | | profile = "main"; |
| | 1897 | | } |
| | 1898 | |
|
| | 1899 | | // Only libx264 support encoding H264 High 10 Profile, otherwise force High Profile. |
| 0 | 1900 | | if (!string.Equals(videoEncoder, "libx264", StringComparison.OrdinalIgnoreCase) |
| 0 | 1901 | | && profile.Contains("high10", StringComparison.OrdinalIgnoreCase)) |
| | 1902 | | { |
| 0 | 1903 | | profile = "high"; |
| | 1904 | | } |
| | 1905 | |
|
| | 1906 | | // We only need Main profile of AV1 encoders. |
| 0 | 1907 | | if (videoEncoder.Contains("av1", StringComparison.OrdinalIgnoreCase) |
| 0 | 1908 | | && (profile.Contains("high", StringComparison.OrdinalIgnoreCase) |
| 0 | 1909 | | || profile.Contains("professional", StringComparison.OrdinalIgnoreCase))) |
| | 1910 | | { |
| 0 | 1911 | | profile = "main"; |
| | 1912 | | } |
| | 1913 | |
|
| | 1914 | | // h264_vaapi does not support Baseline profile, force Constrained Baseline in this case, |
| | 1915 | | // which is compatible (and ugly). |
| 0 | 1916 | | if (string.Equals(videoEncoder, "h264_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1917 | | && profile.Contains("baseline", StringComparison.OrdinalIgnoreCase)) |
| | 1918 | | { |
| 0 | 1919 | | profile = "constrained_baseline"; |
| | 1920 | | } |
| | 1921 | |
|
| | 1922 | | // libx264, h264_{qsv,nvenc,rkmpp} does not support Constrained Baseline profile, force Baseline in this cas |
| 0 | 1923 | | if ((string.Equals(videoEncoder, "libx264", StringComparison.OrdinalIgnoreCase) |
| 0 | 1924 | | || string.Equals(videoEncoder, "h264_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1925 | | || string.Equals(videoEncoder, "h264_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1926 | | || string.Equals(videoEncoder, "h264_rkmpp", StringComparison.OrdinalIgnoreCase)) |
| 0 | 1927 | | && profile.Contains("baseline", StringComparison.OrdinalIgnoreCase)) |
| | 1928 | | { |
| 0 | 1929 | | profile = "baseline"; |
| | 1930 | | } |
| | 1931 | |
|
| | 1932 | | // libx264, h264_{qsv,nvenc,vaapi,rkmpp} does not support Constrained High profile, force High in this case. |
| 0 | 1933 | | if ((string.Equals(videoEncoder, "libx264", StringComparison.OrdinalIgnoreCase) |
| 0 | 1934 | | || string.Equals(videoEncoder, "h264_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1935 | | || string.Equals(videoEncoder, "h264_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1936 | | || string.Equals(videoEncoder, "h264_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 1937 | | || string.Equals(videoEncoder, "h264_rkmpp", StringComparison.OrdinalIgnoreCase)) |
| 0 | 1938 | | && profile.Contains("high", StringComparison.OrdinalIgnoreCase)) |
| | 1939 | | { |
| 0 | 1940 | | profile = "high"; |
| | 1941 | | } |
| | 1942 | |
|
| 0 | 1943 | | if (string.Equals(videoEncoder, "h264_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1944 | | && profile.Contains("baseline", StringComparison.OrdinalIgnoreCase)) |
| | 1945 | | { |
| 0 | 1946 | | profile = "constrained_baseline"; |
| | 1947 | | } |
| | 1948 | |
|
| 0 | 1949 | | if (string.Equals(videoEncoder, "h264_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 1950 | | && profile.Contains("constrainedhigh", StringComparison.OrdinalIgnoreCase)) |
| | 1951 | | { |
| 0 | 1952 | | profile = "constrained_high"; |
| | 1953 | | } |
| | 1954 | |
|
| 0 | 1955 | | if (string.Equals(videoEncoder, "h264_videotoolbox", StringComparison.OrdinalIgnoreCase) |
| 0 | 1956 | | && profile.Contains("constrainedbaseline", StringComparison.OrdinalIgnoreCase)) |
| | 1957 | | { |
| 0 | 1958 | | profile = "constrained_baseline"; |
| | 1959 | | } |
| | 1960 | |
|
| 0 | 1961 | | if (string.Equals(videoEncoder, "h264_videotoolbox", StringComparison.OrdinalIgnoreCase) |
| 0 | 1962 | | && profile.Contains("constrainedhigh", StringComparison.OrdinalIgnoreCase)) |
| | 1963 | | { |
| 0 | 1964 | | profile = "constrained_high"; |
| | 1965 | | } |
| | 1966 | |
|
| 0 | 1967 | | if (!string.IsNullOrEmpty(profile)) |
| | 1968 | | { |
| | 1969 | | // Currently there's no profile option in av1_nvenc encoder |
| 0 | 1970 | | if (!(string.Equals(videoEncoder, "av1_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 1971 | | || string.Equals(videoEncoder, "h264_v4l2m2m", StringComparison.OrdinalIgnoreCase))) |
| | 1972 | | { |
| 0 | 1973 | | param += " -profile:v:0 " + profile; |
| | 1974 | | } |
| | 1975 | | } |
| | 1976 | |
|
| 0 | 1977 | | var level = state.GetRequestedLevel(targetVideoCodec); |
| | 1978 | |
|
| 0 | 1979 | | if (!string.IsNullOrEmpty(level)) |
| | 1980 | | { |
| 0 | 1981 | | level = NormalizeTranscodingLevel(state, level); |
| | 1982 | |
|
| | 1983 | | // libx264, QSV, AMF can adjust the given level to match the output. |
| 0 | 1984 | | if (string.Equals(videoEncoder, "h264_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1985 | | || string.Equals(videoEncoder, "libx264", StringComparison.OrdinalIgnoreCase)) |
| | 1986 | | { |
| 0 | 1987 | | param += " -level " + level; |
| | 1988 | | } |
| 0 | 1989 | | else if (string.Equals(videoEncoder, "hevc_qsv", StringComparison.OrdinalIgnoreCase)) |
| | 1990 | | { |
| | 1991 | | // hevc_qsv use -level 51 instead of -level 153. |
| 0 | 1992 | | if (double.TryParse(level, CultureInfo.InvariantCulture, out double hevcLevel)) |
| | 1993 | | { |
| 0 | 1994 | | param += " -level " + (hevcLevel / 3); |
| | 1995 | | } |
| | 1996 | | } |
| 0 | 1997 | | else if (string.Equals(videoEncoder, "av1_qsv", StringComparison.OrdinalIgnoreCase) |
| 0 | 1998 | | || string.Equals(videoEncoder, "libsvtav1", StringComparison.OrdinalIgnoreCase)) |
| | 1999 | | { |
| | 2000 | | // libsvtav1 and av1_qsv use -level 60 instead of -level 16 |
| | 2001 | | // https://aomedia.org/av1/specification/annex-a/ |
| 0 | 2002 | | if (int.TryParse(level, NumberStyles.Any, CultureInfo.InvariantCulture, out int av1Level)) |
| | 2003 | | { |
| 0 | 2004 | | var x = 2 + (av1Level >> 2); |
| 0 | 2005 | | var y = av1Level & 3; |
| 0 | 2006 | | var res = (x * 10) + y; |
| 0 | 2007 | | param += " -level " + res; |
| | 2008 | | } |
| | 2009 | | } |
| 0 | 2010 | | else if (string.Equals(videoEncoder, "h264_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 2011 | | || string.Equals(videoEncoder, "hevc_amf", StringComparison.OrdinalIgnoreCase) |
| 0 | 2012 | | || string.Equals(videoEncoder, "av1_amf", StringComparison.OrdinalIgnoreCase)) |
| | 2013 | | { |
| 0 | 2014 | | param += " -level " + level; |
| | 2015 | | } |
| 0 | 2016 | | else if (string.Equals(videoEncoder, "h264_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 2017 | | || string.Equals(videoEncoder, "hevc_nvenc", StringComparison.OrdinalIgnoreCase) |
| 0 | 2018 | | || string.Equals(videoEncoder, "av1_nvenc", StringComparison.OrdinalIgnoreCase)) |
| | 2019 | | { |
| | 2020 | | // level option may cause NVENC to fail. |
| | 2021 | | // NVENC cannot adjust the given level, just throw an error. |
| | 2022 | | } |
| 0 | 2023 | | else if (string.Equals(videoEncoder, "h264_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 2024 | | || string.Equals(videoEncoder, "hevc_vaapi", StringComparison.OrdinalIgnoreCase) |
| 0 | 2025 | | || string.Equals(videoEncoder, "av1_vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 2026 | | { |
| | 2027 | | // level option may cause corrupted frames on AMD VAAPI. |
| 0 | 2028 | | if (_mediaEncoder.IsVaapiDeviceInteliHD || _mediaEncoder.IsVaapiDeviceInteli965) |
| | 2029 | | { |
| 0 | 2030 | | param += " -level " + level; |
| | 2031 | | } |
| | 2032 | | } |
| 0 | 2033 | | else if (string.Equals(videoEncoder, "h264_rkmpp", StringComparison.OrdinalIgnoreCase) |
| 0 | 2034 | | || string.Equals(videoEncoder, "hevc_rkmpp", StringComparison.OrdinalIgnoreCase)) |
| | 2035 | | { |
| 0 | 2036 | | param += " -level " + level; |
| | 2037 | | } |
| 0 | 2038 | | else if (!string.Equals(videoEncoder, "libx265", StringComparison.OrdinalIgnoreCase)) |
| | 2039 | | { |
| 0 | 2040 | | param += " -level " + level; |
| | 2041 | | } |
| | 2042 | | } |
| | 2043 | |
|
| 0 | 2044 | | if (string.Equals(videoEncoder, "libx264", StringComparison.OrdinalIgnoreCase)) |
| | 2045 | | { |
| 0 | 2046 | | param += " -x264opts:0 subme=0:me_range=16:rc_lookahead=10:me=hex:open_gop=0"; |
| | 2047 | | } |
| | 2048 | |
|
| 0 | 2049 | | if (string.Equals(videoEncoder, "libx265", StringComparison.OrdinalIgnoreCase)) |
| | 2050 | | { |
| | 2051 | | // libx265 only accept level option in -x265-params. |
| | 2052 | | // level option may cause libx265 to fail. |
| | 2053 | | // libx265 cannot adjust the given level, just throw an error. |
| 0 | 2054 | | param += " -x265-params:0 subme=3:merange=25:rc-lookahead=10:me=star:ctu=32:max-tu-size=32:min-cu-size=1 |
| | 2055 | | } |
| | 2056 | |
|
| 0 | 2057 | | if (string.Equals(videoEncoder, "libsvtav1", StringComparison.OrdinalIgnoreCase) |
| 0 | 2058 | | && _mediaEncoder.EncoderVersion >= _minFFmpegSvtAv1Params) |
| | 2059 | | { |
| 0 | 2060 | | param += " -svtav1-params:0 rc=1:tune=0:film-grain=0:enable-overlays=1:enable-tf=0"; |
| | 2061 | | } |
| | 2062 | |
|
| | 2063 | | /* Access unit too large: 8192 < 20880 error */ |
| 0 | 2064 | | if ((string.Equals(videoEncoder, "h264_vaapi", StringComparison.OrdinalIgnoreCase) || |
| 0 | 2065 | | string.Equals(videoEncoder, "hevc_vaapi", StringComparison.OrdinalIgnoreCase)) && |
| 0 | 2066 | | _mediaEncoder.EncoderVersion >= _minFFmpegVaapiH26xEncA53CcSei) |
| | 2067 | | { |
| 0 | 2068 | | param += " -sei -a53_cc"; |
| | 2069 | | } |
| | 2070 | |
|
| 0 | 2071 | | return param; |
| | 2072 | | } |
| | 2073 | |
|
| | 2074 | | public bool CanStreamCopyVideo(EncodingJobInfo state, MediaStream videoStream) |
| | 2075 | | { |
| 0 | 2076 | | var request = state.BaseRequest; |
| | 2077 | |
|
| 0 | 2078 | | if (!request.AllowVideoStreamCopy) |
| | 2079 | | { |
| 0 | 2080 | | return false; |
| | 2081 | | } |
| | 2082 | |
|
| 0 | 2083 | | if (videoStream.IsInterlaced |
| 0 | 2084 | | && state.DeInterlace(videoStream.Codec, false)) |
| | 2085 | | { |
| 0 | 2086 | | return false; |
| | 2087 | | } |
| | 2088 | |
|
| 0 | 2089 | | if (videoStream.IsAnamorphic ?? false) |
| | 2090 | | { |
| 0 | 2091 | | if (request.RequireNonAnamorphic) |
| | 2092 | | { |
| 0 | 2093 | | return false; |
| | 2094 | | } |
| | 2095 | | } |
| | 2096 | |
|
| | 2097 | | // Can't stream copy if we're burning in subtitles |
| 0 | 2098 | | if (request.SubtitleStreamIndex.HasValue |
| 0 | 2099 | | && request.SubtitleStreamIndex.Value >= 0 |
| 0 | 2100 | | && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode) |
| | 2101 | | { |
| 0 | 2102 | | return false; |
| | 2103 | | } |
| | 2104 | |
|
| 0 | 2105 | | if (string.Equals("h264", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 2106 | | && videoStream.IsAVC.HasValue |
| 0 | 2107 | | && !videoStream.IsAVC.Value |
| 0 | 2108 | | && request.RequireAvc) |
| | 2109 | | { |
| 0 | 2110 | | return false; |
| | 2111 | | } |
| | 2112 | |
|
| | 2113 | | // Source and target codecs must match |
| 0 | 2114 | | if (string.IsNullOrEmpty(videoStream.Codec) |
| 0 | 2115 | | || (state.SupportedVideoCodecs.Length != 0 |
| 0 | 2116 | | && !state.SupportedVideoCodecs.Contains(videoStream.Codec, StringComparison.OrdinalIgnoreCase))) |
| | 2117 | | { |
| 0 | 2118 | | return false; |
| | 2119 | | } |
| | 2120 | |
|
| 0 | 2121 | | var requestedProfiles = state.GetRequestedProfiles(videoStream.Codec); |
| | 2122 | |
|
| | 2123 | | // If client is requesting a specific video profile, it must match the source |
| 0 | 2124 | | if (requestedProfiles.Length > 0) |
| | 2125 | | { |
| 0 | 2126 | | if (string.IsNullOrEmpty(videoStream.Profile)) |
| | 2127 | | { |
| | 2128 | | // return false; |
| | 2129 | | } |
| | 2130 | |
|
| 0 | 2131 | | var requestedProfile = requestedProfiles[0]; |
| | 2132 | | // strip spaces because they may be stripped out on the query string as well |
| 0 | 2133 | | if (!string.IsNullOrEmpty(videoStream.Profile) |
| 0 | 2134 | | && !requestedProfiles.Contains(videoStream.Profile.Replace(" ", string.Empty, StringComparison.Ordin |
| | 2135 | | { |
| 0 | 2136 | | var currentScore = GetVideoProfileScore(videoStream.Codec, videoStream.Profile); |
| 0 | 2137 | | var requestedScore = GetVideoProfileScore(videoStream.Codec, requestedProfile); |
| | 2138 | |
|
| 0 | 2139 | | if (currentScore == -1 || currentScore > requestedScore) |
| | 2140 | | { |
| 0 | 2141 | | return false; |
| | 2142 | | } |
| | 2143 | | } |
| | 2144 | | } |
| | 2145 | |
|
| 0 | 2146 | | var requestedRangeTypes = state.GetRequestedRangeTypes(videoStream.Codec); |
| 0 | 2147 | | if (requestedRangeTypes.Length > 0) |
| | 2148 | | { |
| 0 | 2149 | | if (videoStream.VideoRangeType == VideoRangeType.Unknown) |
| | 2150 | | { |
| 0 | 2151 | | return false; |
| | 2152 | | } |
| | 2153 | |
|
| | 2154 | | // DOVIWithHDR10 should be compatible with HDR10 supporting players. Same goes with HLG and of course SD |
| | 2155 | |
|
| 0 | 2156 | | var requestHasHDR10 = requestedRangeTypes.Contains(VideoRangeType.HDR10.ToString(), StringComparison.Ord |
| 0 | 2157 | | var requestHasHLG = requestedRangeTypes.Contains(VideoRangeType.HLG.ToString(), StringComparison.Ordinal |
| 0 | 2158 | | var requestHasSDR = requestedRangeTypes.Contains(VideoRangeType.SDR.ToString(), StringComparison.Ordinal |
| | 2159 | |
|
| 0 | 2160 | | if (!requestedRangeTypes.Contains(videoStream.VideoRangeType.ToString(), StringComparison.OrdinalIgnoreC |
| 0 | 2161 | | && !((requestHasHDR10 && videoStream.VideoRangeType == VideoRangeType.DOVIWithHDR10) |
| 0 | 2162 | | || (requestHasHLG && videoStream.VideoRangeType == VideoRangeType.DOVIWithHLG) |
| 0 | 2163 | | || (requestHasSDR && videoStream.VideoRangeType == VideoRangeType.DOVIWithSDR))) |
| | 2164 | | { |
| 0 | 2165 | | return false; |
| | 2166 | | } |
| | 2167 | | } |
| | 2168 | |
|
| | 2169 | | // Video width must fall within requested value |
| 0 | 2170 | | if (request.MaxWidth.HasValue |
| 0 | 2171 | | && (!videoStream.Width.HasValue || videoStream.Width.Value > request.MaxWidth.Value)) |
| | 2172 | | { |
| 0 | 2173 | | return false; |
| | 2174 | | } |
| | 2175 | |
|
| | 2176 | | // Video height must fall within requested value |
| 0 | 2177 | | if (request.MaxHeight.HasValue |
| 0 | 2178 | | && (!videoStream.Height.HasValue || videoStream.Height.Value > request.MaxHeight.Value)) |
| | 2179 | | { |
| 0 | 2180 | | return false; |
| | 2181 | | } |
| | 2182 | |
|
| | 2183 | | // Video framerate must fall within requested value |
| 0 | 2184 | | var requestedFramerate = request.MaxFramerate ?? request.Framerate; |
| 0 | 2185 | | if (requestedFramerate.HasValue) |
| | 2186 | | { |
| 0 | 2187 | | var videoFrameRate = videoStream.ReferenceFrameRate; |
| | 2188 | |
|
| 0 | 2189 | | if (!videoFrameRate.HasValue || videoFrameRate.Value > requestedFramerate.Value) |
| | 2190 | | { |
| 0 | 2191 | | return false; |
| | 2192 | | } |
| | 2193 | | } |
| | 2194 | |
|
| | 2195 | | // Video bitrate must fall within requested value |
| 0 | 2196 | | if (request.VideoBitRate.HasValue |
| 0 | 2197 | | && (!videoStream.BitRate.HasValue || videoStream.BitRate.Value > request.VideoBitRate.Value)) |
| | 2198 | | { |
| | 2199 | | // For LiveTV that has no bitrate, let's try copy if other conditions are met |
| 0 | 2200 | | if (string.IsNullOrWhiteSpace(request.LiveStreamId) || videoStream.BitRate.HasValue) |
| | 2201 | | { |
| 0 | 2202 | | return false; |
| | 2203 | | } |
| | 2204 | | } |
| | 2205 | |
|
| 0 | 2206 | | var maxBitDepth = state.GetRequestedVideoBitDepth(videoStream.Codec); |
| 0 | 2207 | | if (maxBitDepth.HasValue) |
| | 2208 | | { |
| 0 | 2209 | | if (videoStream.BitDepth.HasValue && videoStream.BitDepth.Value > maxBitDepth.Value) |
| | 2210 | | { |
| 0 | 2211 | | return false; |
| | 2212 | | } |
| | 2213 | | } |
| | 2214 | |
|
| 0 | 2215 | | var maxRefFrames = state.GetRequestedMaxRefFrames(videoStream.Codec); |
| 0 | 2216 | | if (maxRefFrames.HasValue |
| 0 | 2217 | | && videoStream.RefFrames.HasValue && videoStream.RefFrames.Value > maxRefFrames.Value) |
| | 2218 | | { |
| 0 | 2219 | | return false; |
| | 2220 | | } |
| | 2221 | |
|
| | 2222 | | // If a specific level was requested, the source must match or be less than |
| 0 | 2223 | | var level = state.GetRequestedLevel(videoStream.Codec); |
| 0 | 2224 | | if (double.TryParse(level, CultureInfo.InvariantCulture, out var requestLevel)) |
| | 2225 | | { |
| 0 | 2226 | | if (!videoStream.Level.HasValue) |
| | 2227 | | { |
| | 2228 | | // return false; |
| | 2229 | | } |
| | 2230 | |
|
| 0 | 2231 | | if (videoStream.Level.HasValue && videoStream.Level.Value > requestLevel) |
| | 2232 | | { |
| 0 | 2233 | | return false; |
| | 2234 | | } |
| | 2235 | | } |
| | 2236 | |
|
| 0 | 2237 | | if (string.Equals(state.InputContainer, "avi", StringComparison.OrdinalIgnoreCase) |
| 0 | 2238 | | && string.Equals(videoStream.Codec, "h264", StringComparison.OrdinalIgnoreCase) |
| 0 | 2239 | | && !(videoStream.IsAVC ?? false)) |
| | 2240 | | { |
| | 2241 | | // see Coach S01E01 - Kelly and the Professor(0).avi |
| 0 | 2242 | | return false; |
| | 2243 | | } |
| | 2244 | |
|
| 0 | 2245 | | return true; |
| | 2246 | | } |
| | 2247 | |
|
| | 2248 | | public bool CanStreamCopyAudio(EncodingJobInfo state, MediaStream audioStream, IEnumerable<string> supportedAudi |
| | 2249 | | { |
| 0 | 2250 | | var request = state.BaseRequest; |
| | 2251 | |
|
| 0 | 2252 | | if (!request.AllowAudioStreamCopy) |
| | 2253 | | { |
| 0 | 2254 | | return false; |
| | 2255 | | } |
| | 2256 | |
|
| 0 | 2257 | | var maxBitDepth = state.GetRequestedAudioBitDepth(audioStream.Codec); |
| 0 | 2258 | | if (maxBitDepth.HasValue |
| 0 | 2259 | | && audioStream.BitDepth.HasValue |
| 0 | 2260 | | && audioStream.BitDepth.Value > maxBitDepth.Value) |
| | 2261 | | { |
| 0 | 2262 | | return false; |
| | 2263 | | } |
| | 2264 | |
|
| | 2265 | | // Source and target codecs must match |
| 0 | 2266 | | if (string.IsNullOrEmpty(audioStream.Codec) |
| 0 | 2267 | | || !supportedAudioCodecs.Contains(audioStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 2268 | | { |
| 0 | 2269 | | return false; |
| | 2270 | | } |
| | 2271 | |
|
| | 2272 | | // Channels must fall within requested value |
| 0 | 2273 | | var channels = state.GetRequestedAudioChannels(audioStream.Codec); |
| 0 | 2274 | | if (channels.HasValue) |
| | 2275 | | { |
| 0 | 2276 | | if (!audioStream.Channels.HasValue || audioStream.Channels.Value <= 0) |
| | 2277 | | { |
| 0 | 2278 | | return false; |
| | 2279 | | } |
| | 2280 | |
|
| 0 | 2281 | | if (audioStream.Channels.Value > channels.Value) |
| | 2282 | | { |
| 0 | 2283 | | return false; |
| | 2284 | | } |
| | 2285 | | } |
| | 2286 | |
|
| | 2287 | | // Sample rate must fall within requested value |
| 0 | 2288 | | if (request.AudioSampleRate.HasValue) |
| | 2289 | | { |
| 0 | 2290 | | if (!audioStream.SampleRate.HasValue || audioStream.SampleRate.Value <= 0) |
| | 2291 | | { |
| 0 | 2292 | | return false; |
| | 2293 | | } |
| | 2294 | |
|
| 0 | 2295 | | if (audioStream.SampleRate.Value > request.AudioSampleRate.Value) |
| | 2296 | | { |
| 0 | 2297 | | return false; |
| | 2298 | | } |
| | 2299 | | } |
| | 2300 | |
|
| | 2301 | | // Audio bitrate must fall within requested value |
| 0 | 2302 | | if (request.AudioBitRate.HasValue |
| 0 | 2303 | | && audioStream.BitRate.HasValue |
| 0 | 2304 | | && audioStream.BitRate.Value > request.AudioBitRate.Value) |
| | 2305 | | { |
| 0 | 2306 | | return false; |
| | 2307 | | } |
| | 2308 | |
|
| 0 | 2309 | | return request.EnableAutoStreamCopy; |
| | 2310 | | } |
| | 2311 | |
|
| | 2312 | | public int GetVideoBitrateParamValue(BaseEncodingJobOptions request, MediaStream videoStream, string outputVideo |
| | 2313 | | { |
| 0 | 2314 | | var bitrate = request.VideoBitRate; |
| | 2315 | |
|
| 0 | 2316 | | if (videoStream is not null) |
| | 2317 | | { |
| 0 | 2318 | | var isUpscaling = request.Height.HasValue |
| 0 | 2319 | | && videoStream.Height.HasValue |
| 0 | 2320 | | && request.Height.Value > videoStream.Height.Value |
| 0 | 2321 | | && request.Width.HasValue |
| 0 | 2322 | | && videoStream.Width.HasValue |
| 0 | 2323 | | && request.Width.Value > videoStream.Width.Value; |
| | 2324 | |
|
| | 2325 | | // Don't allow bitrate increases unless upscaling |
| 0 | 2326 | | if (!isUpscaling && bitrate.HasValue && videoStream.BitRate.HasValue) |
| | 2327 | | { |
| 0 | 2328 | | bitrate = GetMinBitrate(videoStream.BitRate.Value, bitrate.Value); |
| | 2329 | | } |
| | 2330 | |
|
| 0 | 2331 | | if (bitrate.HasValue) |
| | 2332 | | { |
| 0 | 2333 | | var inputVideoCodec = videoStream.Codec; |
| 0 | 2334 | | bitrate = ScaleBitrate(bitrate.Value, inputVideoCodec, outputVideoCodec); |
| | 2335 | |
|
| | 2336 | | // If a max bitrate was requested, don't let the scaled bitrate exceed it |
| 0 | 2337 | | if (request.VideoBitRate.HasValue) |
| | 2338 | | { |
| 0 | 2339 | | bitrate = Math.Min(bitrate.Value, request.VideoBitRate.Value); |
| | 2340 | | } |
| | 2341 | | } |
| | 2342 | | } |
| | 2343 | |
|
| | 2344 | | // Cap the max target bitrate to intMax/2 to satisfy the bufsize=bitrate*2. |
| 0 | 2345 | | return Math.Min(bitrate ?? 0, int.MaxValue / 2); |
| | 2346 | | } |
| | 2347 | |
|
| | 2348 | | private int GetMinBitrate(int sourceBitrate, int requestedBitrate) |
| | 2349 | | { |
| | 2350 | | // these values were chosen from testing to improve low bitrate streams |
| 0 | 2351 | | if (sourceBitrate <= 2000000) |
| | 2352 | | { |
| 0 | 2353 | | sourceBitrate = Convert.ToInt32(sourceBitrate * 2.5); |
| | 2354 | | } |
| 0 | 2355 | | else if (sourceBitrate <= 3000000) |
| | 2356 | | { |
| 0 | 2357 | | sourceBitrate *= 2; |
| | 2358 | | } |
| | 2359 | |
|
| 0 | 2360 | | var bitrate = Math.Min(sourceBitrate, requestedBitrate); |
| | 2361 | |
|
| 0 | 2362 | | return bitrate; |
| | 2363 | | } |
| | 2364 | |
|
| | 2365 | | private static double GetVideoBitrateScaleFactor(string codec) |
| | 2366 | | { |
| | 2367 | | // hevc & vp9 - 40% more efficient than h.264 |
| 0 | 2368 | | if (string.Equals(codec, "h265", StringComparison.OrdinalIgnoreCase) |
| 0 | 2369 | | || string.Equals(codec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 2370 | | || string.Equals(codec, "vp9", StringComparison.OrdinalIgnoreCase)) |
| | 2371 | | { |
| 0 | 2372 | | return .6; |
| | 2373 | | } |
| | 2374 | |
|
| | 2375 | | // av1 - 50% more efficient than h.264 |
| 0 | 2376 | | if (string.Equals(codec, "av1", StringComparison.OrdinalIgnoreCase)) |
| | 2377 | | { |
| 0 | 2378 | | return .5; |
| | 2379 | | } |
| | 2380 | |
|
| 0 | 2381 | | return 1; |
| | 2382 | | } |
| | 2383 | |
|
| | 2384 | | private static int ScaleBitrate(int bitrate, string inputVideoCodec, string outputVideoCodec) |
| | 2385 | | { |
| 0 | 2386 | | var inputScaleFactor = GetVideoBitrateScaleFactor(inputVideoCodec); |
| 0 | 2387 | | var outputScaleFactor = GetVideoBitrateScaleFactor(outputVideoCodec); |
| | 2388 | |
|
| | 2389 | | // Don't scale the real bitrate lower than the requested bitrate |
| 0 | 2390 | | var scaleFactor = Math.Max(outputScaleFactor / inputScaleFactor, 1); |
| | 2391 | |
|
| 0 | 2392 | | if (bitrate <= 500000) |
| | 2393 | | { |
| 0 | 2394 | | scaleFactor = Math.Max(scaleFactor, 4); |
| | 2395 | | } |
| 0 | 2396 | | else if (bitrate <= 1000000) |
| | 2397 | | { |
| 0 | 2398 | | scaleFactor = Math.Max(scaleFactor, 3); |
| | 2399 | | } |
| 0 | 2400 | | else if (bitrate <= 2000000) |
| | 2401 | | { |
| 0 | 2402 | | scaleFactor = Math.Max(scaleFactor, 2.5); |
| | 2403 | | } |
| 0 | 2404 | | else if (bitrate <= 3000000) |
| | 2405 | | { |
| 0 | 2406 | | scaleFactor = Math.Max(scaleFactor, 2); |
| | 2407 | | } |
| | 2408 | |
|
| 0 | 2409 | | return Convert.ToInt32(scaleFactor * bitrate); |
| | 2410 | | } |
| | 2411 | |
|
| | 2412 | | public int? GetAudioBitrateParam(BaseEncodingJobOptions request, MediaStream audioStream, int? outputAudioChanne |
| | 2413 | | { |
| 0 | 2414 | | return GetAudioBitrateParam(request.AudioBitRate, request.AudioCodec, audioStream, outputAudioChannels); |
| | 2415 | | } |
| | 2416 | |
|
| | 2417 | | public int? GetAudioBitrateParam(int? audioBitRate, string audioCodec, MediaStream audioStream, int? outputAudio |
| | 2418 | | { |
| 0 | 2419 | | if (audioStream is null) |
| | 2420 | | { |
| 0 | 2421 | | return null; |
| | 2422 | | } |
| | 2423 | |
|
| 0 | 2424 | | var inputChannels = audioStream.Channels ?? 0; |
| 0 | 2425 | | var outputChannels = outputAudioChannels ?? 0; |
| 0 | 2426 | | var bitrate = audioBitRate ?? int.MaxValue; |
| | 2427 | |
|
| 0 | 2428 | | if (string.IsNullOrEmpty(audioCodec) |
| 0 | 2429 | | || string.Equals(audioCodec, "aac", StringComparison.OrdinalIgnoreCase) |
| 0 | 2430 | | || string.Equals(audioCodec, "mp3", StringComparison.OrdinalIgnoreCase) |
| 0 | 2431 | | || string.Equals(audioCodec, "opus", StringComparison.OrdinalIgnoreCase) |
| 0 | 2432 | | || string.Equals(audioCodec, "vorbis", StringComparison.OrdinalIgnoreCase) |
| 0 | 2433 | | || string.Equals(audioCodec, "ac3", StringComparison.OrdinalIgnoreCase) |
| 0 | 2434 | | || string.Equals(audioCodec, "eac3", StringComparison.OrdinalIgnoreCase)) |
| | 2435 | | { |
| 0 | 2436 | | return (inputChannels, outputChannels) switch |
| 0 | 2437 | | { |
| 0 | 2438 | | (>= 6, >= 6 or 0) => Math.Min(640000, bitrate), |
| 0 | 2439 | | (> 0, > 0) => Math.Min(outputChannels * 128000, bitrate), |
| 0 | 2440 | | (> 0, _) => Math.Min(inputChannels * 128000, bitrate), |
| 0 | 2441 | | (_, _) => Math.Min(384000, bitrate) |
| 0 | 2442 | | }; |
| | 2443 | | } |
| | 2444 | |
|
| 0 | 2445 | | if (string.Equals(audioCodec, "dts", StringComparison.OrdinalIgnoreCase) |
| 0 | 2446 | | || string.Equals(audioCodec, "dca", StringComparison.OrdinalIgnoreCase)) |
| | 2447 | | { |
| 0 | 2448 | | return (inputChannels, outputChannels) switch |
| 0 | 2449 | | { |
| 0 | 2450 | | (>= 6, >= 6 or 0) => Math.Min(768000, bitrate), |
| 0 | 2451 | | (> 0, > 0) => Math.Min(outputChannels * 136000, bitrate), |
| 0 | 2452 | | (> 0, _) => Math.Min(inputChannels * 136000, bitrate), |
| 0 | 2453 | | (_, _) => Math.Min(672000, bitrate) |
| 0 | 2454 | | }; |
| | 2455 | | } |
| | 2456 | |
|
| | 2457 | | // Empty bitrate area is not allow on iOS |
| | 2458 | | // Default audio bitrate to 128K per channel if we don't have codec specific defaults |
| | 2459 | | // https://ffmpeg.org/ffmpeg-codecs.html#toc-Codec-Options |
| 0 | 2460 | | return 128000 * (outputAudioChannels ?? audioStream.Channels ?? 2); |
| | 2461 | | } |
| | 2462 | |
|
| | 2463 | | public string GetAudioVbrModeParam(string encoder, int bitrate, int channels) |
| | 2464 | | { |
| 0 | 2465 | | var bitratePerChannel = bitrate / Math.Max(channels, 1); |
| 0 | 2466 | | if (string.Equals(encoder, "libfdk_aac", StringComparison.OrdinalIgnoreCase)) |
| | 2467 | | { |
| 0 | 2468 | | return " -vbr:a " + bitratePerChannel switch |
| 0 | 2469 | | { |
| 0 | 2470 | | < 32000 => "1", |
| 0 | 2471 | | < 48000 => "2", |
| 0 | 2472 | | < 64000 => "3", |
| 0 | 2473 | | < 96000 => "4", |
| 0 | 2474 | | _ => "5" |
| 0 | 2475 | | }; |
| | 2476 | | } |
| | 2477 | |
|
| 0 | 2478 | | if (string.Equals(encoder, "libmp3lame", StringComparison.OrdinalIgnoreCase)) |
| | 2479 | | { |
| | 2480 | | // lame's VBR is only good for a certain bitrate range |
| | 2481 | | // For very low and very high bitrate, use abr mode |
| 0 | 2482 | | if (bitratePerChannel is < 122500 and > 48000) |
| | 2483 | | { |
| 0 | 2484 | | return " -qscale:a " + bitratePerChannel switch |
| 0 | 2485 | | { |
| 0 | 2486 | | < 64000 => "6", |
| 0 | 2487 | | < 88000 => "4", |
| 0 | 2488 | | < 112000 => "2", |
| 0 | 2489 | | _ => "0" |
| 0 | 2490 | | }; |
| | 2491 | | } |
| | 2492 | |
|
| 0 | 2493 | | return " -abr:a 1" + " -b:a " + bitrate; |
| | 2494 | | } |
| | 2495 | |
|
| 0 | 2496 | | if (string.Equals(encoder, "aac_at", StringComparison.OrdinalIgnoreCase)) |
| | 2497 | | { |
| | 2498 | | // aac_at's CVBR mode |
| 0 | 2499 | | return " -aac_at_mode:a 2" + " -b:a " + bitrate; |
| | 2500 | | } |
| | 2501 | |
|
| 0 | 2502 | | if (string.Equals(encoder, "libvorbis", StringComparison.OrdinalIgnoreCase)) |
| | 2503 | | { |
| 0 | 2504 | | return " -qscale:a " + bitratePerChannel switch |
| 0 | 2505 | | { |
| 0 | 2506 | | < 40000 => "0", |
| 0 | 2507 | | < 56000 => "2", |
| 0 | 2508 | | < 80000 => "4", |
| 0 | 2509 | | < 112000 => "6", |
| 0 | 2510 | | _ => "8" |
| 0 | 2511 | | }; |
| | 2512 | | } |
| | 2513 | |
|
| 0 | 2514 | | return null; |
| | 2515 | | } |
| | 2516 | |
|
| | 2517 | | public string GetAudioFilterParam(EncodingJobInfo state, EncodingOptions encodingOptions) |
| | 2518 | | { |
| 0 | 2519 | | var channels = state.OutputAudioChannels; |
| | 2520 | |
|
| 0 | 2521 | | var filters = new List<string>(); |
| | 2522 | |
|
| 0 | 2523 | | if (channels is 2 && state.AudioStream?.Channels is > 2) |
| | 2524 | | { |
| 0 | 2525 | | var hasDownMixFilter = DownMixAlgorithmsHelper.AlgorithmFilterStrings.TryGetValue((encodingOptions.DownM |
| 0 | 2526 | | if (hasDownMixFilter) |
| | 2527 | | { |
| 0 | 2528 | | filters.Add(downMixFilterString); |
| | 2529 | | } |
| | 2530 | |
|
| 0 | 2531 | | if (!encodingOptions.DownMixAudioBoost.Equals(1)) |
| | 2532 | | { |
| 0 | 2533 | | filters.Add("volume=" + encodingOptions.DownMixAudioBoost.ToString(CultureInfo.InvariantCulture)); |
| | 2534 | | } |
| | 2535 | | } |
| | 2536 | |
|
| 0 | 2537 | | var isCopyingTimestamps = state.CopyTimestamps || state.TranscodingType != TranscodingJobType.Progressive; |
| 0 | 2538 | | if (state.SubtitleStream is not null && state.SubtitleStream.IsTextSubtitleStream && state.SubtitleDeliveryM |
| | 2539 | | { |
| 0 | 2540 | | var seconds = TimeSpan.FromTicks(state.StartTimeTicks ?? 0).TotalSeconds; |
| | 2541 | |
|
| 0 | 2542 | | filters.Add( |
| 0 | 2543 | | string.Format( |
| 0 | 2544 | | CultureInfo.InvariantCulture, |
| 0 | 2545 | | "asetpts=PTS-{0}/TB", |
| 0 | 2546 | | Math.Round(seconds))); |
| | 2547 | | } |
| | 2548 | |
|
| 0 | 2549 | | if (filters.Count > 0) |
| | 2550 | | { |
| 0 | 2551 | | return " -af \"" + string.Join(',', filters) + "\""; |
| | 2552 | | } |
| | 2553 | |
|
| 0 | 2554 | | return string.Empty; |
| | 2555 | | } |
| | 2556 | |
|
| | 2557 | | /// <summary> |
| | 2558 | | /// Gets the number of audio channels to specify on the command line. |
| | 2559 | | /// </summary> |
| | 2560 | | /// <param name="state">The state.</param> |
| | 2561 | | /// <param name="audioStream">The audio stream.</param> |
| | 2562 | | /// <param name="outputAudioCodec">The output audio codec.</param> |
| | 2563 | | /// <returns>System.Nullable{System.Int32}.</returns> |
| | 2564 | | public int? GetNumAudioChannelsParam(EncodingJobInfo state, MediaStream audioStream, string outputAudioCodec) |
| | 2565 | | { |
| 0 | 2566 | | if (audioStream is null) |
| | 2567 | | { |
| 0 | 2568 | | return null; |
| | 2569 | | } |
| | 2570 | |
|
| 0 | 2571 | | var request = state.BaseRequest; |
| | 2572 | |
|
| 0 | 2573 | | var codec = outputAudioCodec ?? string.Empty; |
| | 2574 | |
|
| 0 | 2575 | | int? resultChannels = state.GetRequestedAudioChannels(codec); |
| | 2576 | |
|
| 0 | 2577 | | var inputChannels = audioStream.Channels; |
| | 2578 | |
|
| 0 | 2579 | | if (inputChannels > 0) |
| | 2580 | | { |
| 0 | 2581 | | resultChannels = inputChannels < resultChannels ? inputChannels : resultChannels ?? inputChannels; |
| | 2582 | | } |
| | 2583 | |
|
| 0 | 2584 | | var isTranscodingAudio = !IsCopyCodec(codec); |
| | 2585 | |
|
| 0 | 2586 | | if (isTranscodingAudio) |
| | 2587 | | { |
| 0 | 2588 | | var audioEncoder = GetAudioEncoder(state); |
| 0 | 2589 | | if (!_audioTranscodeChannelLookup.TryGetValue(audioEncoder, out var transcoderChannelLimit)) |
| | 2590 | | { |
| | 2591 | | // Set default max transcoding channels to 8 to prevent encoding errors due to asking for too many c |
| 0 | 2592 | | transcoderChannelLimit = 8; |
| | 2593 | | } |
| | 2594 | |
|
| | 2595 | | // Set resultChannels to minimum between resultChannels, TranscodingMaxAudioChannels, transcoderChannelL |
| 0 | 2596 | | resultChannels = transcoderChannelLimit < resultChannels ? transcoderChannelLimit : resultChannels ?? tr |
| | 2597 | |
|
| 0 | 2598 | | if (request.TranscodingMaxAudioChannels < resultChannels) |
| | 2599 | | { |
| 0 | 2600 | | resultChannels = request.TranscodingMaxAudioChannels; |
| | 2601 | | } |
| | 2602 | |
|
| | 2603 | | // Avoid transcoding to audio channels other than 1ch, 2ch, 6ch (5.1 layout) and 8ch (7.1 layout). |
| | 2604 | | // https://developer.apple.com/documentation/http_live_streaming/hls_authoring_specification_for_apple_d |
| 0 | 2605 | | if (state.TranscodingType != TranscodingJobType.Progressive |
| 0 | 2606 | | && ((resultChannels > 2 && resultChannels < 6) || resultChannels == 7)) |
| | 2607 | | { |
| | 2608 | | // We can let FFMpeg supply an extra LFE channel for 5ch and 7ch to make them 5.1 and 7.1 |
| 0 | 2609 | | if (resultChannels == 5) |
| | 2610 | | { |
| 0 | 2611 | | resultChannels = 6; |
| | 2612 | | } |
| 0 | 2613 | | else if (resultChannels == 7) |
| | 2614 | | { |
| 0 | 2615 | | resultChannels = 8; |
| | 2616 | | } |
| | 2617 | | else |
| | 2618 | | { |
| | 2619 | | // For other weird layout, just downmix to stereo for compatibility |
| 0 | 2620 | | resultChannels = 2; |
| | 2621 | | } |
| | 2622 | | } |
| | 2623 | | } |
| | 2624 | |
|
| 0 | 2625 | | return resultChannels; |
| | 2626 | | } |
| | 2627 | |
|
| | 2628 | | /// <summary> |
| | 2629 | | /// Enforces the resolution limit. |
| | 2630 | | /// </summary> |
| | 2631 | | /// <param name="state">The state.</param> |
| | 2632 | | public void EnforceResolutionLimit(EncodingJobInfo state) |
| | 2633 | | { |
| 0 | 2634 | | var videoRequest = state.BaseRequest; |
| | 2635 | |
|
| | 2636 | | // Switch the incoming params to be ceilings rather than fixed values |
| 0 | 2637 | | videoRequest.MaxWidth = videoRequest.MaxWidth ?? videoRequest.Width; |
| 0 | 2638 | | videoRequest.MaxHeight = videoRequest.MaxHeight ?? videoRequest.Height; |
| | 2639 | |
|
| 0 | 2640 | | videoRequest.Width = null; |
| 0 | 2641 | | videoRequest.Height = null; |
| 0 | 2642 | | } |
| | 2643 | |
|
| | 2644 | | /// <summary> |
| | 2645 | | /// Gets the fast seek command line parameter. |
| | 2646 | | /// </summary> |
| | 2647 | | /// <param name="state">The state.</param> |
| | 2648 | | /// <param name="options">The options.</param> |
| | 2649 | | /// <param name="segmentContainer">Segment Container.</param> |
| | 2650 | | /// <returns>System.String.</returns> |
| | 2651 | | /// <value>The fast seek command line parameter.</value> |
| | 2652 | | public string GetFastSeekCommandLineParameter(EncodingJobInfo state, EncodingOptions options, string segmentCont |
| | 2653 | | { |
| 0 | 2654 | | var time = state.BaseRequest.StartTimeTicks ?? 0; |
| 0 | 2655 | | var seekParam = string.Empty; |
| | 2656 | |
|
| 0 | 2657 | | if (time > 0) |
| | 2658 | | { |
| | 2659 | | // For direct streaming/remuxing, we seek at the exact position of the keyframe |
| | 2660 | | // However, ffmpeg will seek to previous keyframe when the exact time is the input |
| | 2661 | | // Workaround this by adding 0.5s offset to the seeking time to get the exact keyframe on most videos. |
| | 2662 | | // This will help subtitle syncing. |
| 0 | 2663 | | var isHlsRemuxing = state.IsVideoRequest && state.TranscodingType is TranscodingJobType.Hls && IsCopyCod |
| 0 | 2664 | | var seekTick = isHlsRemuxing ? time + 5000000L : time; |
| 0 | 2665 | | seekParam += string.Format(CultureInfo.InvariantCulture, "-ss {0}", _mediaEncoder.GetTimeParameter(seekT |
| | 2666 | |
|
| 0 | 2667 | | if (state.IsVideoRequest) |
| | 2668 | | { |
| 0 | 2669 | | var outputVideoCodec = GetVideoEncoder(state, options); |
| 0 | 2670 | | var segmentFormat = GetSegmentFileExtension(segmentContainer).TrimStart('.'); |
| | 2671 | |
|
| | 2672 | | // Important: If this is ever re-enabled, make sure not to use it with wtv because it breaks seeking |
| | 2673 | | // Disable -noaccurate_seek on mpegts container due to the timestamps issue on some clients, |
| | 2674 | | // but it's still required for fMP4 container otherwise the audio can't be synced to the video. |
| 0 | 2675 | | if (!string.Equals(state.InputContainer, "wtv", StringComparison.OrdinalIgnoreCase) |
| 0 | 2676 | | && !string.Equals(segmentFormat, "ts", StringComparison.OrdinalIgnoreCase) |
| 0 | 2677 | | && state.TranscodingType != TranscodingJobType.Progressive |
| 0 | 2678 | | && !state.EnableBreakOnNonKeyFrames(outputVideoCodec) |
| 0 | 2679 | | && (state.BaseRequest.StartTimeTicks ?? 0) > 0) |
| | 2680 | | { |
| 0 | 2681 | | seekParam += " -noaccurate_seek"; |
| | 2682 | | } |
| | 2683 | | } |
| | 2684 | | } |
| | 2685 | |
|
| 0 | 2686 | | return seekParam; |
| | 2687 | | } |
| | 2688 | |
|
| | 2689 | | /// <summary> |
| | 2690 | | /// Gets the map args. |
| | 2691 | | /// </summary> |
| | 2692 | | /// <param name="state">The state.</param> |
| | 2693 | | /// <returns>System.String.</returns> |
| | 2694 | | public string GetMapArgs(EncodingJobInfo state) |
| | 2695 | | { |
| | 2696 | | // If we don't have known media info |
| | 2697 | | // If input is video, use -sn to drop subtitles |
| | 2698 | | // Otherwise just return empty |
| 0 | 2699 | | if (state.VideoStream is null && state.AudioStream is null) |
| | 2700 | | { |
| 0 | 2701 | | return state.IsInputVideo ? "-sn" : string.Empty; |
| | 2702 | | } |
| | 2703 | |
|
| | 2704 | | // We have media info, but we don't know the stream index |
| 0 | 2705 | | if (state.VideoStream is not null && state.VideoStream.Index == -1) |
| | 2706 | | { |
| 0 | 2707 | | return "-sn"; |
| | 2708 | | } |
| | 2709 | |
|
| | 2710 | | // We have media info, but we don't know the stream index |
| 0 | 2711 | | if (state.AudioStream is not null && state.AudioStream.Index == -1) |
| | 2712 | | { |
| 0 | 2713 | | return state.IsInputVideo ? "-sn" : string.Empty; |
| | 2714 | | } |
| | 2715 | |
|
| 0 | 2716 | | var args = string.Empty; |
| | 2717 | |
|
| 0 | 2718 | | if (state.VideoStream is not null) |
| | 2719 | | { |
| 0 | 2720 | | int videoStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.VideoStream); |
| | 2721 | |
|
| 0 | 2722 | | args += string.Format( |
| 0 | 2723 | | CultureInfo.InvariantCulture, |
| 0 | 2724 | | "-map 0:{0}", |
| 0 | 2725 | | videoStreamIndex); |
| | 2726 | | } |
| | 2727 | | else |
| | 2728 | | { |
| | 2729 | | // No known video stream |
| 0 | 2730 | | args += "-vn"; |
| | 2731 | | } |
| | 2732 | |
|
| 0 | 2733 | | if (state.AudioStream is not null) |
| | 2734 | | { |
| 0 | 2735 | | int audioStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.AudioStream); |
| 0 | 2736 | | if (state.AudioStream.IsExternal) |
| | 2737 | | { |
| 0 | 2738 | | bool hasExternalGraphicsSubs = state.SubtitleStream is not null |
| 0 | 2739 | | && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode |
| 0 | 2740 | | && state.SubtitleStream.IsExternal |
| 0 | 2741 | | && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 2742 | | int externalAudioMapIndex = hasExternalGraphicsSubs ? 2 : 1; |
| | 2743 | |
|
| 0 | 2744 | | args += string.Format( |
| 0 | 2745 | | CultureInfo.InvariantCulture, |
| 0 | 2746 | | " -map {0}:{1}", |
| 0 | 2747 | | externalAudioMapIndex, |
| 0 | 2748 | | audioStreamIndex); |
| | 2749 | | } |
| | 2750 | | else |
| | 2751 | | { |
| 0 | 2752 | | args += string.Format( |
| 0 | 2753 | | CultureInfo.InvariantCulture, |
| 0 | 2754 | | " -map 0:{0}", |
| 0 | 2755 | | audioStreamIndex); |
| | 2756 | | } |
| | 2757 | | } |
| | 2758 | | else |
| | 2759 | | { |
| 0 | 2760 | | args += " -map -0:a"; |
| | 2761 | | } |
| | 2762 | |
|
| 0 | 2763 | | var subtitleMethod = state.SubtitleDeliveryMethod; |
| 0 | 2764 | | if (state.SubtitleStream is null || subtitleMethod == SubtitleDeliveryMethod.Hls) |
| | 2765 | | { |
| 0 | 2766 | | args += " -map -0:s"; |
| | 2767 | | } |
| 0 | 2768 | | else if (subtitleMethod == SubtitleDeliveryMethod.Embed) |
| | 2769 | | { |
| 0 | 2770 | | int subtitleStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.SubtitleStream); |
| | 2771 | |
|
| 0 | 2772 | | args += string.Format( |
| 0 | 2773 | | CultureInfo.InvariantCulture, |
| 0 | 2774 | | " -map 0:{0}", |
| 0 | 2775 | | subtitleStreamIndex); |
| | 2776 | | } |
| 0 | 2777 | | else if (state.SubtitleStream.IsExternal && !state.SubtitleStream.IsTextSubtitleStream) |
| | 2778 | | { |
| 0 | 2779 | | int externalSubtitleStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.SubtitleStream); |
| | 2780 | |
|
| 0 | 2781 | | args += string.Format( |
| 0 | 2782 | | CultureInfo.InvariantCulture, |
| 0 | 2783 | | " -map 1:{0} -sn", |
| 0 | 2784 | | externalSubtitleStreamIndex); |
| | 2785 | | } |
| | 2786 | |
|
| 0 | 2787 | | return args; |
| | 2788 | | } |
| | 2789 | |
|
| | 2790 | | /// <summary> |
| | 2791 | | /// Gets the negative map args by filters. |
| | 2792 | | /// </summary> |
| | 2793 | | /// <param name="state">The state.</param> |
| | 2794 | | /// <param name="videoProcessFilters">The videoProcessFilters.</param> |
| | 2795 | | /// <returns>System.String.</returns> |
| | 2796 | | public string GetNegativeMapArgsByFilters(EncodingJobInfo state, string videoProcessFilters) |
| | 2797 | | { |
| 0 | 2798 | | string args = string.Empty; |
| | 2799 | |
|
| | 2800 | | // http://ffmpeg.org/ffmpeg-all.html#toc-Complex-filtergraphs-1 |
| 0 | 2801 | | if (state.VideoStream is not null && videoProcessFilters.Contains("-filter_complex", StringComparison.Ordina |
| | 2802 | | { |
| 0 | 2803 | | int videoStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.VideoStream); |
| | 2804 | |
|
| 0 | 2805 | | args += string.Format( |
| 0 | 2806 | | CultureInfo.InvariantCulture, |
| 0 | 2807 | | "-map -0:{0} ", |
| 0 | 2808 | | videoStreamIndex); |
| | 2809 | | } |
| | 2810 | |
|
| 0 | 2811 | | return args; |
| | 2812 | | } |
| | 2813 | |
|
| | 2814 | | /// <summary> |
| | 2815 | | /// Determines which stream will be used for playback. |
| | 2816 | | /// </summary> |
| | 2817 | | /// <param name="allStream">All stream.</param> |
| | 2818 | | /// <param name="desiredIndex">Index of the desired.</param> |
| | 2819 | | /// <param name="type">The type.</param> |
| | 2820 | | /// <param name="returnFirstIfNoIndex">if set to <c>true</c> [return first if no index].</param> |
| | 2821 | | /// <returns>MediaStream.</returns> |
| | 2822 | | public MediaStream GetMediaStream(IEnumerable<MediaStream> allStream, int? desiredIndex, MediaStreamType type, b |
| | 2823 | | { |
| 0 | 2824 | | var streams = allStream.Where(s => s.Type == type).OrderBy(i => i.Index).ToList(); |
| | 2825 | |
|
| 0 | 2826 | | if (desiredIndex.HasValue) |
| | 2827 | | { |
| 0 | 2828 | | var stream = streams.FirstOrDefault(s => s.Index == desiredIndex.Value); |
| | 2829 | |
|
| 0 | 2830 | | if (stream is not null) |
| | 2831 | | { |
| 0 | 2832 | | return stream; |
| | 2833 | | } |
| | 2834 | | } |
| | 2835 | |
|
| 0 | 2836 | | if (returnFirstIfNoIndex && type == MediaStreamType.Audio) |
| | 2837 | | { |
| 0 | 2838 | | return streams.FirstOrDefault(i => i.Channels.HasValue && i.Channels.Value > 0) ?? |
| 0 | 2839 | | streams.FirstOrDefault(); |
| | 2840 | | } |
| | 2841 | |
|
| | 2842 | | // Just return the first one |
| 0 | 2843 | | return returnFirstIfNoIndex ? streams.FirstOrDefault() : null; |
| | 2844 | | } |
| | 2845 | |
|
| | 2846 | | public static (int? Width, int? Height) GetFixedOutputSize( |
| | 2847 | | int? videoWidth, |
| | 2848 | | int? videoHeight, |
| | 2849 | | int? requestedWidth, |
| | 2850 | | int? requestedHeight, |
| | 2851 | | int? requestedMaxWidth, |
| | 2852 | | int? requestedMaxHeight) |
| | 2853 | | { |
| 0 | 2854 | | if (!videoWidth.HasValue && !requestedWidth.HasValue) |
| | 2855 | | { |
| 0 | 2856 | | return (null, null); |
| | 2857 | | } |
| | 2858 | |
|
| 0 | 2859 | | if (!videoHeight.HasValue && !requestedHeight.HasValue) |
| | 2860 | | { |
| 0 | 2861 | | return (null, null); |
| | 2862 | | } |
| | 2863 | |
|
| 0 | 2864 | | int inputWidth = Convert.ToInt32(videoWidth ?? requestedWidth, CultureInfo.InvariantCulture); |
| 0 | 2865 | | int inputHeight = Convert.ToInt32(videoHeight ?? requestedHeight, CultureInfo.InvariantCulture); |
| 0 | 2866 | | int outputWidth = requestedWidth ?? inputWidth; |
| 0 | 2867 | | int outputHeight = requestedHeight ?? inputHeight; |
| | 2868 | |
|
| | 2869 | | // Don't transcode video to bigger than 4k when using HW. |
| 0 | 2870 | | int maximumWidth = Math.Min(requestedMaxWidth ?? outputWidth, 4096); |
| 0 | 2871 | | int maximumHeight = Math.Min(requestedMaxHeight ?? outputHeight, 4096); |
| | 2872 | |
|
| 0 | 2873 | | if (outputWidth > maximumWidth || outputHeight > maximumHeight) |
| | 2874 | | { |
| 0 | 2875 | | var scaleW = (double)maximumWidth / outputWidth; |
| 0 | 2876 | | var scaleH = (double)maximumHeight / outputHeight; |
| 0 | 2877 | | var scale = Math.Min(scaleW, scaleH); |
| 0 | 2878 | | outputWidth = Math.Min(maximumWidth, Convert.ToInt32(outputWidth * scale)); |
| 0 | 2879 | | outputHeight = Math.Min(maximumHeight, Convert.ToInt32(outputHeight * scale)); |
| | 2880 | | } |
| | 2881 | |
|
| 0 | 2882 | | outputWidth = 2 * (outputWidth / 2); |
| 0 | 2883 | | outputHeight = 2 * (outputHeight / 2); |
| | 2884 | |
|
| 0 | 2885 | | return (outputWidth, outputHeight); |
| | 2886 | | } |
| | 2887 | |
|
| | 2888 | | public static bool IsScaleRatioSupported( |
| | 2889 | | int? videoWidth, |
| | 2890 | | int? videoHeight, |
| | 2891 | | int? requestedWidth, |
| | 2892 | | int? requestedHeight, |
| | 2893 | | int? requestedMaxWidth, |
| | 2894 | | int? requestedMaxHeight, |
| | 2895 | | double? maxScaleRatio) |
| | 2896 | | { |
| 0 | 2897 | | var (outWidth, outHeight) = GetFixedOutputSize( |
| 0 | 2898 | | videoWidth, |
| 0 | 2899 | | videoHeight, |
| 0 | 2900 | | requestedWidth, |
| 0 | 2901 | | requestedHeight, |
| 0 | 2902 | | requestedMaxWidth, |
| 0 | 2903 | | requestedMaxHeight); |
| | 2904 | |
|
| 0 | 2905 | | if (!videoWidth.HasValue |
| 0 | 2906 | | || !videoHeight.HasValue |
| 0 | 2907 | | || !outWidth.HasValue |
| 0 | 2908 | | || !outHeight.HasValue |
| 0 | 2909 | | || !maxScaleRatio.HasValue |
| 0 | 2910 | | || (maxScaleRatio.Value < 1.0f)) |
| | 2911 | | { |
| 0 | 2912 | | return false; |
| | 2913 | | } |
| | 2914 | |
|
| 0 | 2915 | | var minScaleRatio = 1.0f / maxScaleRatio; |
| 0 | 2916 | | var scaleRatioW = (double)outWidth / (double)videoWidth; |
| 0 | 2917 | | var scaleRatioH = (double)outHeight / (double)videoHeight; |
| | 2918 | |
|
| 0 | 2919 | | if (scaleRatioW < minScaleRatio |
| 0 | 2920 | | || scaleRatioW > maxScaleRatio |
| 0 | 2921 | | || scaleRatioH < minScaleRatio |
| 0 | 2922 | | || scaleRatioH > maxScaleRatio) |
| | 2923 | | { |
| 0 | 2924 | | return false; |
| | 2925 | | } |
| | 2926 | |
|
| 0 | 2927 | | return true; |
| | 2928 | | } |
| | 2929 | |
|
| | 2930 | | public static string GetHwScaleFilter( |
| | 2931 | | string hwScalePrefix, |
| | 2932 | | string hwScaleSuffix, |
| | 2933 | | string videoFormat, |
| | 2934 | | bool swapOutputWandH, |
| | 2935 | | int? videoWidth, |
| | 2936 | | int? videoHeight, |
| | 2937 | | int? requestedWidth, |
| | 2938 | | int? requestedHeight, |
| | 2939 | | int? requestedMaxWidth, |
| | 2940 | | int? requestedMaxHeight) |
| | 2941 | | { |
| 0 | 2942 | | var (outWidth, outHeight) = GetFixedOutputSize( |
| 0 | 2943 | | videoWidth, |
| 0 | 2944 | | videoHeight, |
| 0 | 2945 | | requestedWidth, |
| 0 | 2946 | | requestedHeight, |
| 0 | 2947 | | requestedMaxWidth, |
| 0 | 2948 | | requestedMaxHeight); |
| | 2949 | |
|
| 0 | 2950 | | var isFormatFixed = !string.IsNullOrEmpty(videoFormat); |
| 0 | 2951 | | var isSizeFixed = !videoWidth.HasValue |
| 0 | 2952 | | || outWidth.Value != videoWidth.Value |
| 0 | 2953 | | || !videoHeight.HasValue |
| 0 | 2954 | | || outHeight.Value != videoHeight.Value; |
| | 2955 | |
|
| 0 | 2956 | | var swpOutW = swapOutputWandH ? outHeight.Value : outWidth.Value; |
| 0 | 2957 | | var swpOutH = swapOutputWandH ? outWidth.Value : outHeight.Value; |
| | 2958 | |
|
| 0 | 2959 | | var arg1 = isSizeFixed ? $"=w={swpOutW}:h={swpOutH}" : string.Empty; |
| 0 | 2960 | | var arg2 = isFormatFixed ? $"format={videoFormat}" : string.Empty; |
| 0 | 2961 | | if (isFormatFixed) |
| | 2962 | | { |
| 0 | 2963 | | arg2 = (isSizeFixed ? ':' : '=') + arg2; |
| | 2964 | | } |
| | 2965 | |
|
| 0 | 2966 | | if (!string.IsNullOrEmpty(hwScaleSuffix) && (isSizeFixed || isFormatFixed)) |
| | 2967 | | { |
| 0 | 2968 | | return string.Format( |
| 0 | 2969 | | CultureInfo.InvariantCulture, |
| 0 | 2970 | | "{0}_{1}{2}{3}", |
| 0 | 2971 | | hwScalePrefix ?? "scale", |
| 0 | 2972 | | hwScaleSuffix, |
| 0 | 2973 | | arg1, |
| 0 | 2974 | | arg2); |
| | 2975 | | } |
| | 2976 | |
|
| 0 | 2977 | | return string.Empty; |
| | 2978 | | } |
| | 2979 | |
|
| | 2980 | | public static string GetGraphicalSubPreProcessFilters( |
| | 2981 | | int? videoWidth, |
| | 2982 | | int? videoHeight, |
| | 2983 | | int? subtitleWidth, |
| | 2984 | | int? subtitleHeight, |
| | 2985 | | int? requestedWidth, |
| | 2986 | | int? requestedHeight, |
| | 2987 | | int? requestedMaxWidth, |
| | 2988 | | int? requestedMaxHeight) |
| | 2989 | | { |
| 0 | 2990 | | var (outWidth, outHeight) = GetFixedOutputSize( |
| 0 | 2991 | | videoWidth, |
| 0 | 2992 | | videoHeight, |
| 0 | 2993 | | requestedWidth, |
| 0 | 2994 | | requestedHeight, |
| 0 | 2995 | | requestedMaxWidth, |
| 0 | 2996 | | requestedMaxHeight); |
| | 2997 | |
|
| 0 | 2998 | | if (!outWidth.HasValue |
| 0 | 2999 | | || !outHeight.HasValue |
| 0 | 3000 | | || outWidth.Value <= 0 |
| 0 | 3001 | | || outHeight.Value <= 0) |
| | 3002 | | { |
| 0 | 3003 | | return string.Empty; |
| | 3004 | | } |
| | 3005 | |
|
| | 3006 | | // Automatically add padding based on subtitle input |
| 0 | 3007 | | var filters = @"scale,scale=-1:{1}:fast_bilinear,crop,pad=max({0}\,iw):max({1}\,ih):(ow-iw)/2:(oh-ih)/2:blac |
| | 3008 | |
|
| 0 | 3009 | | if (subtitleWidth.HasValue |
| 0 | 3010 | | && subtitleHeight.HasValue |
| 0 | 3011 | | && subtitleWidth.Value > 0 |
| 0 | 3012 | | && subtitleHeight.Value > 0) |
| | 3013 | | { |
| 0 | 3014 | | var videoDar = (double)outWidth.Value / outHeight.Value; |
| 0 | 3015 | | var subtitleDar = (double)subtitleWidth.Value / subtitleHeight.Value; |
| | 3016 | |
|
| | 3017 | | // No need to add padding when DAR is the same -> 1080p PGSSUB on 2160p video |
| 0 | 3018 | | if (Math.Abs(videoDar - subtitleDar) < 0.01f) |
| | 3019 | | { |
| 0 | 3020 | | filters = @"scale,scale={0}:{1}:fast_bilinear"; |
| | 3021 | | } |
| | 3022 | | } |
| | 3023 | |
|
| 0 | 3024 | | return string.Format( |
| 0 | 3025 | | CultureInfo.InvariantCulture, |
| 0 | 3026 | | filters, |
| 0 | 3027 | | outWidth.Value, |
| 0 | 3028 | | outHeight.Value); |
| | 3029 | | } |
| | 3030 | |
|
| | 3031 | | public static string GetAlphaSrcFilter( |
| | 3032 | | EncodingJobInfo state, |
| | 3033 | | int? videoWidth, |
| | 3034 | | int? videoHeight, |
| | 3035 | | int? requestedWidth, |
| | 3036 | | int? requestedHeight, |
| | 3037 | | int? requestedMaxWidth, |
| | 3038 | | int? requestedMaxHeight, |
| | 3039 | | float? framerate) |
| | 3040 | | { |
| 0 | 3041 | | var reqTicks = state.BaseRequest.StartTimeTicks ?? 0; |
| 0 | 3042 | | var startTime = TimeSpan.FromTicks(reqTicks).ToString(@"hh\\\:mm\\\:ss\\\.fff", CultureInfo.InvariantCulture |
| 0 | 3043 | | var (outWidth, outHeight) = GetFixedOutputSize( |
| 0 | 3044 | | videoWidth, |
| 0 | 3045 | | videoHeight, |
| 0 | 3046 | | requestedWidth, |
| 0 | 3047 | | requestedHeight, |
| 0 | 3048 | | requestedMaxWidth, |
| 0 | 3049 | | requestedMaxHeight); |
| | 3050 | |
|
| 0 | 3051 | | if (outWidth.HasValue && outHeight.HasValue) |
| | 3052 | | { |
| 0 | 3053 | | return string.Format( |
| 0 | 3054 | | CultureInfo.InvariantCulture, |
| 0 | 3055 | | "alphasrc=s={0}x{1}:r={2}:start='{3}'", |
| 0 | 3056 | | outWidth.Value, |
| 0 | 3057 | | outHeight.Value, |
| 0 | 3058 | | framerate ?? 25, |
| 0 | 3059 | | reqTicks > 0 ? startTime : 0); |
| | 3060 | | } |
| | 3061 | |
|
| 0 | 3062 | | return string.Empty; |
| | 3063 | | } |
| | 3064 | |
|
| | 3065 | | public static string GetSwScaleFilter( |
| | 3066 | | EncodingJobInfo state, |
| | 3067 | | EncodingOptions options, |
| | 3068 | | string videoEncoder, |
| | 3069 | | int? videoWidth, |
| | 3070 | | int? videoHeight, |
| | 3071 | | Video3DFormat? threedFormat, |
| | 3072 | | int? requestedWidth, |
| | 3073 | | int? requestedHeight, |
| | 3074 | | int? requestedMaxWidth, |
| | 3075 | | int? requestedMaxHeight) |
| | 3076 | | { |
| 0 | 3077 | | var isV4l2 = string.Equals(videoEncoder, "h264_v4l2m2m", StringComparison.OrdinalIgnoreCase); |
| 0 | 3078 | | var isMjpeg = videoEncoder is not null && videoEncoder.Contains("mjpeg", StringComparison.OrdinalIgnoreCase) |
| 0 | 3079 | | var scaleVal = isV4l2 ? 64 : 2; |
| 0 | 3080 | | var targetAr = isMjpeg ? "(a*sar)" : "a"; // manually calculate AR when using mjpeg encoder |
| | 3081 | |
|
| | 3082 | | // If fixed dimensions were supplied |
| 0 | 3083 | | if (requestedWidth.HasValue && requestedHeight.HasValue) |
| | 3084 | | { |
| 0 | 3085 | | if (isV4l2) |
| | 3086 | | { |
| 0 | 3087 | | var widthParam = requestedWidth.Value.ToString(CultureInfo.InvariantCulture); |
| 0 | 3088 | | var heightParam = requestedHeight.Value.ToString(CultureInfo.InvariantCulture); |
| | 3089 | |
|
| 0 | 3090 | | return string.Format( |
| 0 | 3091 | | CultureInfo.InvariantCulture, |
| 0 | 3092 | | "scale=trunc({0}/64)*64:trunc({1}/2)*2", |
| 0 | 3093 | | widthParam, |
| 0 | 3094 | | heightParam); |
| | 3095 | | } |
| | 3096 | |
|
| 0 | 3097 | | return GetFixedSwScaleFilter(threedFormat, requestedWidth.Value, requestedHeight.Value); |
| | 3098 | | } |
| | 3099 | |
|
| | 3100 | | // If Max dimensions were supplied, for width selects lowest even number between input width and width req s |
| | 3101 | |
|
| 0 | 3102 | | if (requestedMaxWidth.HasValue && requestedMaxHeight.HasValue) |
| | 3103 | | { |
| 0 | 3104 | | var maxWidthParam = requestedMaxWidth.Value.ToString(CultureInfo.InvariantCulture); |
| 0 | 3105 | | var maxHeightParam = requestedMaxHeight.Value.ToString(CultureInfo.InvariantCulture); |
| | 3106 | |
|
| 0 | 3107 | | return string.Format( |
| 0 | 3108 | | CultureInfo.InvariantCulture, |
| 0 | 3109 | | @"scale=trunc(min(max(iw\,ih*{3})\,min({0}\,{1}*{3}))/{2})*{2}:trunc(min(max(iw/{3}\,ih)\,min({0}/{3 |
| 0 | 3110 | | maxWidthParam, |
| 0 | 3111 | | maxHeightParam, |
| 0 | 3112 | | scaleVal, |
| 0 | 3113 | | targetAr); |
| | 3114 | | } |
| | 3115 | |
|
| | 3116 | | // If a fixed width was requested |
| 0 | 3117 | | if (requestedWidth.HasValue) |
| | 3118 | | { |
| 0 | 3119 | | if (threedFormat.HasValue) |
| | 3120 | | { |
| | 3121 | | // This method can handle 0 being passed in for the requested height |
| 0 | 3122 | | return GetFixedSwScaleFilter(threedFormat, requestedWidth.Value, 0); |
| | 3123 | | } |
| | 3124 | |
|
| 0 | 3125 | | var widthParam = requestedWidth.Value.ToString(CultureInfo.InvariantCulture); |
| | 3126 | |
|
| 0 | 3127 | | return string.Format( |
| 0 | 3128 | | CultureInfo.InvariantCulture, |
| 0 | 3129 | | "scale={0}:trunc(ow/{1}/2)*2", |
| 0 | 3130 | | widthParam, |
| 0 | 3131 | | targetAr); |
| | 3132 | | } |
| | 3133 | |
|
| | 3134 | | // If a fixed height was requested |
| 0 | 3135 | | if (requestedHeight.HasValue) |
| | 3136 | | { |
| 0 | 3137 | | var heightParam = requestedHeight.Value.ToString(CultureInfo.InvariantCulture); |
| | 3138 | |
|
| 0 | 3139 | | return string.Format( |
| 0 | 3140 | | CultureInfo.InvariantCulture, |
| 0 | 3141 | | "scale=trunc(oh*{2}/{1})*{1}:{0}", |
| 0 | 3142 | | heightParam, |
| 0 | 3143 | | scaleVal, |
| 0 | 3144 | | targetAr); |
| | 3145 | | } |
| | 3146 | |
|
| | 3147 | | // If a max width was requested |
| 0 | 3148 | | if (requestedMaxWidth.HasValue) |
| | 3149 | | { |
| 0 | 3150 | | var maxWidthParam = requestedMaxWidth.Value.ToString(CultureInfo.InvariantCulture); |
| | 3151 | |
|
| 0 | 3152 | | return string.Format( |
| 0 | 3153 | | CultureInfo.InvariantCulture, |
| 0 | 3154 | | @"scale=trunc(min(max(iw\,ih*{2})\,{0})/{1})*{1}:trunc(ow/{2}/2)*2", |
| 0 | 3155 | | maxWidthParam, |
| 0 | 3156 | | scaleVal, |
| 0 | 3157 | | targetAr); |
| | 3158 | | } |
| | 3159 | |
|
| | 3160 | | // If a max height was requested |
| 0 | 3161 | | if (requestedMaxHeight.HasValue) |
| | 3162 | | { |
| 0 | 3163 | | var maxHeightParam = requestedMaxHeight.Value.ToString(CultureInfo.InvariantCulture); |
| | 3164 | |
|
| 0 | 3165 | | return string.Format( |
| 0 | 3166 | | CultureInfo.InvariantCulture, |
| 0 | 3167 | | @"scale=trunc(oh*{2}/{1})*{1}:min(max(iw/{2}\,ih)\,{0})", |
| 0 | 3168 | | maxHeightParam, |
| 0 | 3169 | | scaleVal, |
| 0 | 3170 | | targetAr); |
| | 3171 | | } |
| | 3172 | |
|
| 0 | 3173 | | return string.Empty; |
| | 3174 | | } |
| | 3175 | |
|
| | 3176 | | private static string GetFixedSwScaleFilter(Video3DFormat? threedFormat, int requestedWidth, int requestedHeight |
| | 3177 | | { |
| 0 | 3178 | | var widthParam = requestedWidth.ToString(CultureInfo.InvariantCulture); |
| 0 | 3179 | | var heightParam = requestedHeight.ToString(CultureInfo.InvariantCulture); |
| | 3180 | |
|
| 0 | 3181 | | string filter = null; |
| | 3182 | |
|
| 0 | 3183 | | if (threedFormat.HasValue) |
| | 3184 | | { |
| 0 | 3185 | | switch (threedFormat.Value) |
| | 3186 | | { |
| | 3187 | | case Video3DFormat.HalfSideBySide: |
| 0 | 3188 | | filter = @"crop=iw/2:ih:0:0,scale=(iw*2):ih,setdar=dar=a,crop=min(iw\,ih*dar):min(ih\,iw/dar):(i |
| | 3189 | | // hsbs crop width in half,scale to correct size, set the display aspect,crop out any black bars |
| 0 | 3190 | | break; |
| | 3191 | | case Video3DFormat.FullSideBySide: |
| 0 | 3192 | | filter = @"crop=iw/2:ih:0:0,setdar=dar=a,crop=min(iw\,ih*dar):min(ih\,iw/dar):(iw-min(iw\,iw*sar |
| | 3193 | | // fsbs crop width in half,set the display aspect,crop out any black bars we may have made the s |
| 0 | 3194 | | break; |
| | 3195 | | case Video3DFormat.HalfTopAndBottom: |
| 0 | 3196 | | filter = @"crop=iw:ih/2:0:0,scale=(iw*2):ih),setdar=dar=a,crop=min(iw\,ih*dar):min(ih\,iw/dar):( |
| | 3197 | | // htab crop height in half,scale to correct size, set the display aspect,crop out any black bar |
| 0 | 3198 | | break; |
| | 3199 | | case Video3DFormat.FullTopAndBottom: |
| 0 | 3200 | | filter = @"crop=iw:ih/2:0:0,setdar=dar=a,crop=min(iw\,ih*dar):min(ih\,iw/dar):(iw-min(iw\,iw*sar |
| | 3201 | | // ftab crop height in half, set the display aspect,crop out any black bars we may have made the |
| | 3202 | | break; |
| | 3203 | | default: |
| | 3204 | | break; |
| | 3205 | | } |
| | 3206 | | } |
| | 3207 | |
|
| | 3208 | | // default |
| 0 | 3209 | | if (filter is null) |
| | 3210 | | { |
| 0 | 3211 | | if (requestedHeight > 0) |
| | 3212 | | { |
| 0 | 3213 | | filter = "scale=trunc({0}/2)*2:trunc({1}/2)*2"; |
| | 3214 | | } |
| | 3215 | | else |
| | 3216 | | { |
| 0 | 3217 | | filter = "scale={0}:trunc({0}/a/2)*2"; |
| | 3218 | | } |
| | 3219 | | } |
| | 3220 | |
|
| 0 | 3221 | | return string.Format(CultureInfo.InvariantCulture, filter, widthParam, heightParam); |
| | 3222 | | } |
| | 3223 | |
|
| | 3224 | | public static string GetSwDeinterlaceFilter(EncodingJobInfo state, EncodingOptions options) |
| | 3225 | | { |
| 0 | 3226 | | var doubleRateDeint = options.DeinterlaceDoubleRate && state.VideoStream?.ReferenceFrameRate <= 30; |
| 0 | 3227 | | return string.Format( |
| 0 | 3228 | | CultureInfo.InvariantCulture, |
| 0 | 3229 | | "{0}={1}:-1:0", |
| 0 | 3230 | | options.DeinterlaceMethod.ToString().ToLowerInvariant(), |
| 0 | 3231 | | doubleRateDeint ? "1" : "0"); |
| | 3232 | | } |
| | 3233 | |
|
| | 3234 | | public string GetHwDeinterlaceFilter(EncodingJobInfo state, EncodingOptions options, string hwDeintSuffix) |
| | 3235 | | { |
| 0 | 3236 | | var doubleRateDeint = options.DeinterlaceDoubleRate && (state.VideoStream?.ReferenceFrameRate ?? 60) <= 30; |
| 0 | 3237 | | if (hwDeintSuffix.Contains("cuda", StringComparison.OrdinalIgnoreCase)) |
| | 3238 | | { |
| 0 | 3239 | | var useBwdif = options.DeinterlaceMethod == DeinterlaceMethod.bwdif && _mediaEncoder.SupportsFilter("bwd |
| | 3240 | |
|
| 0 | 3241 | | return string.Format( |
| 0 | 3242 | | CultureInfo.InvariantCulture, |
| 0 | 3243 | | "{0}_cuda={1}:-1:0", |
| 0 | 3244 | | useBwdif ? "bwdif" : "yadif", |
| 0 | 3245 | | doubleRateDeint ? "1" : "0"); |
| | 3246 | | } |
| | 3247 | |
|
| 0 | 3248 | | if (hwDeintSuffix.Contains("vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 3249 | | { |
| 0 | 3250 | | return string.Format( |
| 0 | 3251 | | CultureInfo.InvariantCulture, |
| 0 | 3252 | | "deinterlace_vaapi=rate={0}", |
| 0 | 3253 | | doubleRateDeint ? "field" : "frame"); |
| | 3254 | | } |
| | 3255 | |
|
| 0 | 3256 | | if (hwDeintSuffix.Contains("qsv", StringComparison.OrdinalIgnoreCase)) |
| | 3257 | | { |
| 0 | 3258 | | return "deinterlace_qsv=mode=2"; |
| | 3259 | | } |
| | 3260 | |
|
| 0 | 3261 | | if (hwDeintSuffix.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase)) |
| | 3262 | | { |
| 0 | 3263 | | var useBwdif = options.DeinterlaceMethod == DeinterlaceMethod.bwdif && _mediaEncoder.SupportsFilter("bwd |
| | 3264 | |
|
| 0 | 3265 | | return string.Format( |
| 0 | 3266 | | CultureInfo.InvariantCulture, |
| 0 | 3267 | | "{0}_videotoolbox={1}:-1:0", |
| 0 | 3268 | | useBwdif ? "bwdif" : "yadif", |
| 0 | 3269 | | doubleRateDeint ? "1" : "0"); |
| | 3270 | | } |
| | 3271 | |
|
| 0 | 3272 | | return string.Empty; |
| | 3273 | | } |
| | 3274 | |
|
| | 3275 | | public string GetHwTonemapFilter(EncodingOptions options, string hwTonemapSuffix, string videoFormat) |
| | 3276 | | { |
| 0 | 3277 | | if (string.IsNullOrEmpty(hwTonemapSuffix)) |
| | 3278 | | { |
| 0 | 3279 | | return string.Empty; |
| | 3280 | | } |
| | 3281 | |
|
| 0 | 3282 | | var args = string.Empty; |
| 0 | 3283 | | var algorithm = options.TonemappingAlgorithm.ToString().ToLowerInvariant(); |
| 0 | 3284 | | var mode = options.TonemappingMode.ToString().ToLowerInvariant(); |
| 0 | 3285 | | var range = options.TonemappingRange; |
| 0 | 3286 | | var rangeString = range.ToString().ToLowerInvariant(); |
| | 3287 | |
|
| 0 | 3288 | | if (string.Equals(hwTonemapSuffix, "vaapi", StringComparison.OrdinalIgnoreCase)) |
| | 3289 | | { |
| 0 | 3290 | | var doVaVppProcamp = false; |
| 0 | 3291 | | var procampParams = string.Empty; |
| 0 | 3292 | | if (options.VppTonemappingBrightness != 0 |
| 0 | 3293 | | && options.VppTonemappingBrightness >= -100 |
| 0 | 3294 | | && options.VppTonemappingBrightness <= 100) |
| | 3295 | | { |
| 0 | 3296 | | procampParams += $"=b={options.VppTonemappingBrightness}"; |
| 0 | 3297 | | doVaVppProcamp = true; |
| | 3298 | | } |
| | 3299 | |
|
| 0 | 3300 | | if (options.VppTonemappingContrast > 1 |
| 0 | 3301 | | && options.VppTonemappingContrast <= 10) |
| | 3302 | | { |
| 0 | 3303 | | procampParams += doVaVppProcamp ? ":" : "="; |
| 0 | 3304 | | procampParams += $"c={options.VppTonemappingContrast}"; |
| 0 | 3305 | | doVaVppProcamp = true; |
| | 3306 | | } |
| | 3307 | |
|
| 0 | 3308 | | args = "{0}tonemap_vaapi=format={1}:p=bt709:t=bt709:m=bt709:extra_hw_frames=32"; |
| | 3309 | |
|
| 0 | 3310 | | return string.Format( |
| 0 | 3311 | | CultureInfo.InvariantCulture, |
| 0 | 3312 | | args, |
| 0 | 3313 | | doVaVppProcamp ? $"procamp_vaapi{procampParams}," : string.Empty, |
| 0 | 3314 | | videoFormat ?? "nv12"); |
| | 3315 | | } |
| | 3316 | | else |
| | 3317 | | { |
| 0 | 3318 | | args = "tonemap_{0}=format={1}:p=bt709:t=bt709:m=bt709:tonemap={2}:peak={3}:desat={4}"; |
| | 3319 | |
|
| 0 | 3320 | | var useLegacyTonemapModes = _mediaEncoder.EncoderVersion >= _minFFmpegOclCuTonemapMode |
| 0 | 3321 | | && _legacyTonemapModes.Contains(options.TonemappingMode); |
| | 3322 | |
|
| 0 | 3323 | | var useAdvancedTonemapModes = _mediaEncoder.EncoderVersion >= _minFFmpegAdvancedTonemapMode |
| 0 | 3324 | | && _advancedTonemapModes.Contains(options.TonemappingMode); |
| | 3325 | |
|
| 0 | 3326 | | if (useLegacyTonemapModes || useAdvancedTonemapModes) |
| | 3327 | | { |
| 0 | 3328 | | args += ":tonemap_mode={5}"; |
| | 3329 | | } |
| | 3330 | |
|
| 0 | 3331 | | if (options.TonemappingParam != 0) |
| | 3332 | | { |
| 0 | 3333 | | args += ":param={6}"; |
| | 3334 | | } |
| | 3335 | |
|
| 0 | 3336 | | if (range == TonemappingRange.tv || range == TonemappingRange.pc) |
| | 3337 | | { |
| 0 | 3338 | | args += ":range={7}"; |
| | 3339 | | } |
| | 3340 | | } |
| | 3341 | |
|
| 0 | 3342 | | return string.Format( |
| 0 | 3343 | | CultureInfo.InvariantCulture, |
| 0 | 3344 | | args, |
| 0 | 3345 | | hwTonemapSuffix, |
| 0 | 3346 | | videoFormat ?? "nv12", |
| 0 | 3347 | | algorithm, |
| 0 | 3348 | | options.TonemappingPeak, |
| 0 | 3349 | | options.TonemappingDesat, |
| 0 | 3350 | | mode, |
| 0 | 3351 | | options.TonemappingParam, |
| 0 | 3352 | | rangeString); |
| | 3353 | | } |
| | 3354 | |
|
| | 3355 | | public string GetLibplaceboFilter( |
| | 3356 | | EncodingOptions options, |
| | 3357 | | string videoFormat, |
| | 3358 | | bool doTonemap, |
| | 3359 | | int? videoWidth, |
| | 3360 | | int? videoHeight, |
| | 3361 | | int? requestedWidth, |
| | 3362 | | int? requestedHeight, |
| | 3363 | | int? requestedMaxWidth, |
| | 3364 | | int? requestedMaxHeight) |
| | 3365 | | { |
| 0 | 3366 | | var (outWidth, outHeight) = GetFixedOutputSize( |
| 0 | 3367 | | videoWidth, |
| 0 | 3368 | | videoHeight, |
| 0 | 3369 | | requestedWidth, |
| 0 | 3370 | | requestedHeight, |
| 0 | 3371 | | requestedMaxWidth, |
| 0 | 3372 | | requestedMaxHeight); |
| | 3373 | |
|
| 0 | 3374 | | var isFormatFixed = !string.IsNullOrEmpty(videoFormat); |
| 0 | 3375 | | var isSizeFixed = !videoWidth.HasValue |
| 0 | 3376 | | || outWidth.Value != videoWidth.Value |
| 0 | 3377 | | || !videoHeight.HasValue |
| 0 | 3378 | | || outHeight.Value != videoHeight.Value; |
| | 3379 | |
|
| 0 | 3380 | | var sizeArg = isSizeFixed ? (":w=" + outWidth.Value + ":h=" + outHeight.Value) : string.Empty; |
| 0 | 3381 | | var formatArg = isFormatFixed ? (":format=" + videoFormat) : string.Empty; |
| 0 | 3382 | | var tonemapArg = string.Empty; |
| | 3383 | |
|
| 0 | 3384 | | if (doTonemap) |
| | 3385 | | { |
| 0 | 3386 | | var algorithm = options.TonemappingAlgorithm; |
| | 3387 | | var algorithmString = "clip"; |
| 0 | 3388 | | var mode = options.TonemappingMode; |
| 0 | 3389 | | var range = options.TonemappingRange; |
| | 3390 | |
|
| 0 | 3391 | | if (algorithm == TonemappingAlgorithm.bt2390) |
| | 3392 | | { |
| | 3393 | | algorithmString = "bt.2390"; |
| | 3394 | | } |
| 0 | 3395 | | else if (algorithm != TonemappingAlgorithm.none) |
| | 3396 | | { |
| 0 | 3397 | | algorithmString = algorithm.ToString().ToLowerInvariant(); |
| | 3398 | | } |
| | 3399 | |
|
| 0 | 3400 | | tonemapArg = ":tonemapping=" + algorithm + ":peak_detect=0:color_primaries=bt709:color_trc=bt709:colorsp |
| | 3401 | |
|
| 0 | 3402 | | if (range == TonemappingRange.tv || range == TonemappingRange.pc) |
| | 3403 | | { |
| 0 | 3404 | | tonemapArg += ":range=" + range.ToString().ToLowerInvariant(); |
| | 3405 | | } |
| | 3406 | | } |
| | 3407 | |
|
| 0 | 3408 | | return string.Format( |
| 0 | 3409 | | CultureInfo.InvariantCulture, |
| 0 | 3410 | | "libplacebo=upscaler=none:downscaler=none{0}{1}{2}", |
| 0 | 3411 | | sizeArg, |
| 0 | 3412 | | formatArg, |
| 0 | 3413 | | tonemapArg); |
| | 3414 | | } |
| | 3415 | |
|
| | 3416 | | public string GetVideoTransposeDirection(EncodingJobInfo state) |
| | 3417 | | { |
| 0 | 3418 | | return (state.VideoStream?.Rotation ?? 0) switch |
| 0 | 3419 | | { |
| 0 | 3420 | | 90 => "cclock", |
| 0 | 3421 | | 180 => "reversal", |
| 0 | 3422 | | -90 => "clock", |
| 0 | 3423 | | -180 => "reversal", |
| 0 | 3424 | | _ => string.Empty |
| 0 | 3425 | | }; |
| | 3426 | | } |
| | 3427 | |
|
| | 3428 | | /// <summary> |
| | 3429 | | /// Gets the parameter of software filter chain. |
| | 3430 | | /// </summary> |
| | 3431 | | /// <param name="state">Encoding state.</param> |
| | 3432 | | /// <param name="options">Encoding options.</param> |
| | 3433 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 3434 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 3435 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetSwVidFilterChain( |
| | 3436 | | EncodingJobInfo state, |
| | 3437 | | EncodingOptions options, |
| | 3438 | | string vidEncoder) |
| | 3439 | | { |
| 0 | 3440 | | var inW = state.VideoStream?.Width; |
| 0 | 3441 | | var inH = state.VideoStream?.Height; |
| 0 | 3442 | | var reqW = state.BaseRequest.Width; |
| 0 | 3443 | | var reqH = state.BaseRequest.Height; |
| 0 | 3444 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 3445 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 3446 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 3447 | |
|
| 0 | 3448 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 3449 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 3450 | | var isVaapiEncoder = vidEncoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 3451 | | var isV4l2Encoder = vidEncoder.Contains("h264_v4l2m2m", StringComparison.OrdinalIgnoreCase); |
| | 3452 | |
|
| 0 | 3453 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 3454 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 3455 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 3456 | | var doToneMap = IsSwTonemapAvailable(state, options); |
| 0 | 3457 | | var requireDoviReshaping = doToneMap && state.VideoStream.VideoRangeType == VideoRangeType.DOVI; |
| | 3458 | |
|
| 0 | 3459 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 3460 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 3461 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| | 3462 | |
|
| 0 | 3463 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 3464 | | var swapWAndH = Math.Abs(rotation) == 90; |
| 0 | 3465 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 3466 | | var swpInH = swapWAndH ? inW : inH; |
| | 3467 | |
|
| | 3468 | | /* Make main filters for video stream */ |
| 0 | 3469 | | var mainFilters = new List<string>(); |
| | 3470 | |
|
| 0 | 3471 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doToneMap)); |
| | 3472 | |
|
| | 3473 | | // INPUT sw surface(memory/copy-back from vram) |
| | 3474 | | // sw deint |
| 0 | 3475 | | if (doDeintH2645) |
| | 3476 | | { |
| 0 | 3477 | | var deintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 3478 | | mainFilters.Add(deintFilter); |
| | 3479 | | } |
| | 3480 | |
|
| 0 | 3481 | | var outFormat = isSwDecoder ? "yuv420p" : "nv12"; |
| 0 | 3482 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, reqH, r |
| 0 | 3483 | | if (isVaapiEncoder) |
| | 3484 | | { |
| 0 | 3485 | | outFormat = "nv12"; |
| | 3486 | | } |
| 0 | 3487 | | else if (isV4l2Encoder) |
| | 3488 | | { |
| 0 | 3489 | | outFormat = "yuv420p"; |
| | 3490 | | } |
| | 3491 | |
|
| | 3492 | | // sw scale |
| 0 | 3493 | | mainFilters.Add(swScaleFilter); |
| | 3494 | |
|
| | 3495 | | // sw tonemap |
| 0 | 3496 | | if (doToneMap) |
| | 3497 | | { |
| | 3498 | | // tonemapx requires yuv420p10 input for dovi reshaping, let ffmpeg convert the frame when necessary |
| 0 | 3499 | | var tonemapFormat = requireDoviReshaping ? "yuv420p" : outFormat; |
| | 3500 | |
|
| 0 | 3501 | | var tonemapArgs = $"tonemapx=tonemap={options.TonemappingAlgorithm}:desat={options.TonemappingDesat}:pea |
| | 3502 | |
|
| 0 | 3503 | | if (options.TonemappingParam != 0) |
| | 3504 | | { |
| 0 | 3505 | | tonemapArgs += $":param={options.TonemappingParam}"; |
| | 3506 | | } |
| | 3507 | |
|
| 0 | 3508 | | var range = options.TonemappingRange; |
| 0 | 3509 | | if (range == TonemappingRange.tv || range == TonemappingRange.pc) |
| | 3510 | | { |
| 0 | 3511 | | tonemapArgs += $":range={options.TonemappingRange}"; |
| | 3512 | | } |
| | 3513 | |
|
| 0 | 3514 | | mainFilters.Add(tonemapArgs); |
| | 3515 | | } |
| | 3516 | | else |
| | 3517 | | { |
| | 3518 | | // OUTPUT yuv420p/nv12 surface(memory) |
| 0 | 3519 | | mainFilters.Add("format=" + outFormat); |
| | 3520 | | } |
| | 3521 | |
|
| | 3522 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 3523 | | var subFilters = new List<string>(); |
| 0 | 3524 | | var overlayFilters = new List<string>(); |
| 0 | 3525 | | if (hasTextSubs) |
| | 3526 | | { |
| | 3527 | | // subtitles=f='*.ass':alpha=0 |
| 0 | 3528 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 3529 | | mainFilters.Add(textSubtitlesFilter); |
| | 3530 | | } |
| 0 | 3531 | | else if (hasGraphicalSubs) |
| | 3532 | | { |
| 0 | 3533 | | var subW = state.SubtitleStream?.Width; |
| 0 | 3534 | | var subH = state.SubtitleStream?.Height; |
| 0 | 3535 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, reqMaxW |
| 0 | 3536 | | subFilters.Add(subPreProcFilters); |
| 0 | 3537 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 3538 | | } |
| | 3539 | |
|
| 0 | 3540 | | return (mainFilters, subFilters, overlayFilters); |
| | 3541 | | } |
| | 3542 | |
|
| | 3543 | | /// <summary> |
| | 3544 | | /// Gets the parameter of Nvidia NVENC filter chain. |
| | 3545 | | /// </summary> |
| | 3546 | | /// <param name="state">Encoding state.</param> |
| | 3547 | | /// <param name="options">Encoding options.</param> |
| | 3548 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 3549 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 3550 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetNvidiaVidFilterChain( |
| | 3551 | | EncodingJobInfo state, |
| | 3552 | | EncodingOptions options, |
| | 3553 | | string vidEncoder) |
| | 3554 | | { |
| 0 | 3555 | | if (options.HardwareAccelerationType != HardwareAccelerationType.nvenc) |
| | 3556 | | { |
| 0 | 3557 | | return (null, null, null); |
| | 3558 | | } |
| | 3559 | |
|
| 0 | 3560 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 3561 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 3562 | | var isSwEncoder = !vidEncoder.Contains("nvenc", StringComparison.OrdinalIgnoreCase); |
| | 3563 | |
|
| | 3564 | | // legacy cuvid pipeline(copy-back) |
| 0 | 3565 | | if ((isSwDecoder && isSwEncoder) |
| 0 | 3566 | | || !IsCudaFullSupported() |
| 0 | 3567 | | || !_mediaEncoder.SupportsFilter("alphasrc")) |
| | 3568 | | { |
| 0 | 3569 | | return GetSwVidFilterChain(state, options, vidEncoder); |
| | 3570 | | } |
| | 3571 | |
|
| | 3572 | | // prefered nvdec/cuvid + cuda filters + nvenc pipeline |
| 0 | 3573 | | return GetNvidiaVidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 3574 | | } |
| | 3575 | |
|
| | 3576 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetNvidiaVidFiltersPrefe |
| | 3577 | | EncodingJobInfo state, |
| | 3578 | | EncodingOptions options, |
| | 3579 | | string vidDecoder, |
| | 3580 | | string vidEncoder) |
| | 3581 | | { |
| 0 | 3582 | | var inW = state.VideoStream?.Width; |
| 0 | 3583 | | var inH = state.VideoStream?.Height; |
| 0 | 3584 | | var reqW = state.BaseRequest.Width; |
| 0 | 3585 | | var reqH = state.BaseRequest.Height; |
| 0 | 3586 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 3587 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 3588 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 3589 | |
|
| 0 | 3590 | | var isNvDecoder = vidDecoder.Contains("cuda", StringComparison.OrdinalIgnoreCase); |
| 0 | 3591 | | var isNvencEncoder = vidEncoder.Contains("nvenc", StringComparison.OrdinalIgnoreCase); |
| 0 | 3592 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 3593 | | var isSwEncoder = !isNvencEncoder; |
| 0 | 3594 | | var isCuInCuOut = isNvDecoder && isNvencEncoder; |
| | 3595 | |
|
| 0 | 3596 | | var doubleRateDeint = options.DeinterlaceDoubleRate && (state.VideoStream?.ReferenceFrameRate ?? 60) <= 30; |
| 0 | 3597 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 3598 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 3599 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 3600 | | var doCuTonemap = IsHwTonemapAvailable(state, options); |
| | 3601 | |
|
| 0 | 3602 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 3603 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 3604 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 3605 | | var hasAssSubs = hasSubs |
| 0 | 3606 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 3607 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| 0 | 3608 | | var subW = state.SubtitleStream?.Width; |
| 0 | 3609 | | var subH = state.SubtitleStream?.Height; |
| | 3610 | |
|
| 0 | 3611 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 3612 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 3613 | | var doCuTranspose = !string.IsNullOrEmpty(tranposeDir) && _mediaEncoder.SupportsFilter("transpose_cuda"); |
| 0 | 3614 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || (isNvDecoder && doCuTranspose)); |
| 0 | 3615 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 3616 | | var swpInH = swapWAndH ? inW : inH; |
| | 3617 | |
|
| | 3618 | | /* Make main filters for video stream */ |
| 0 | 3619 | | var mainFilters = new List<string>(); |
| | 3620 | |
|
| 0 | 3621 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doCuTonemap)); |
| | 3622 | |
|
| 0 | 3623 | | if (isSwDecoder) |
| | 3624 | | { |
| | 3625 | | // INPUT sw surface(memory) |
| | 3626 | | // sw deint |
| 0 | 3627 | | if (doDeintH2645) |
| | 3628 | | { |
| 0 | 3629 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 3630 | | mainFilters.Add(swDeintFilter); |
| | 3631 | | } |
| | 3632 | |
|
| 0 | 3633 | | var outFormat = doCuTonemap ? "yuv420p10le" : "yuv420p"; |
| 0 | 3634 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| | 3635 | | // sw scale |
| 0 | 3636 | | mainFilters.Add(swScaleFilter); |
| 0 | 3637 | | mainFilters.Add($"format={outFormat}"); |
| | 3638 | |
|
| | 3639 | | // sw => hw |
| 0 | 3640 | | if (doCuTonemap) |
| | 3641 | | { |
| 0 | 3642 | | mainFilters.Add("hwupload=derive_device=cuda"); |
| | 3643 | | } |
| | 3644 | | } |
| | 3645 | |
|
| 0 | 3646 | | if (isNvDecoder) |
| | 3647 | | { |
| | 3648 | | // INPUT cuda surface(vram) |
| | 3649 | | // hw deint |
| 0 | 3650 | | if (doDeintH2645) |
| | 3651 | | { |
| 0 | 3652 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "cuda"); |
| 0 | 3653 | | mainFilters.Add(deintFilter); |
| | 3654 | | } |
| | 3655 | |
|
| | 3656 | | // hw transpose |
| 0 | 3657 | | if (doCuTranspose) |
| | 3658 | | { |
| 0 | 3659 | | mainFilters.Add($"transpose_cuda=dir={tranposeDir}"); |
| | 3660 | | } |
| | 3661 | |
|
| 0 | 3662 | | var outFormat = doCuTonemap ? string.Empty : "yuv420p"; |
| 0 | 3663 | | var hwScaleFilter = GetHwScaleFilter("scale", "cuda", outFormat, false, swpInW, swpInH, reqW, reqH, reqM |
| | 3664 | | // hw scale |
| 0 | 3665 | | mainFilters.Add(hwScaleFilter); |
| | 3666 | | } |
| | 3667 | |
|
| | 3668 | | // hw tonemap |
| 0 | 3669 | | if (doCuTonemap) |
| | 3670 | | { |
| 0 | 3671 | | var tonemapFilter = GetHwTonemapFilter(options, "cuda", "yuv420p"); |
| 0 | 3672 | | mainFilters.Add(tonemapFilter); |
| | 3673 | | } |
| | 3674 | |
|
| 0 | 3675 | | var memoryOutput = false; |
| 0 | 3676 | | var isUploadForCuTonemap = isSwDecoder && doCuTonemap; |
| 0 | 3677 | | if ((isNvDecoder && isSwEncoder) || (isUploadForCuTonemap && hasSubs)) |
| | 3678 | | { |
| 0 | 3679 | | memoryOutput = true; |
| | 3680 | |
|
| | 3681 | | // OUTPUT yuv420p surface(memory) |
| 0 | 3682 | | mainFilters.Add("hwdownload"); |
| 0 | 3683 | | mainFilters.Add("format=yuv420p"); |
| | 3684 | | } |
| | 3685 | |
|
| | 3686 | | // OUTPUT yuv420p surface(memory) |
| 0 | 3687 | | if (isSwDecoder && isNvencEncoder && !isUploadForCuTonemap) |
| | 3688 | | { |
| 0 | 3689 | | memoryOutput = true; |
| | 3690 | | } |
| | 3691 | |
|
| 0 | 3692 | | if (memoryOutput) |
| | 3693 | | { |
| | 3694 | | // text subtitles |
| 0 | 3695 | | if (hasTextSubs) |
| | 3696 | | { |
| 0 | 3697 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 3698 | | mainFilters.Add(textSubtitlesFilter); |
| | 3699 | | } |
| | 3700 | | } |
| | 3701 | |
|
| | 3702 | | // OUTPUT cuda(yuv420p) surface(vram) |
| | 3703 | |
|
| | 3704 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 3705 | | var subFilters = new List<string>(); |
| 0 | 3706 | | var overlayFilters = new List<string>(); |
| 0 | 3707 | | if (isCuInCuOut) |
| | 3708 | | { |
| 0 | 3709 | | if (hasSubs) |
| | 3710 | | { |
| 0 | 3711 | | if (hasGraphicalSubs) |
| | 3712 | | { |
| 0 | 3713 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, |
| 0 | 3714 | | subFilters.Add(subPreProcFilters); |
| 0 | 3715 | | subFilters.Add("format=yuva420p"); |
| | 3716 | | } |
| 0 | 3717 | | else if (hasTextSubs) |
| | 3718 | | { |
| 0 | 3719 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 3720 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 3721 | |
|
| | 3722 | | // alphasrc=s=1280x720:r=10:start=0,format=yuva420p,subtitles,hwupload |
| 0 | 3723 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH, subF |
| 0 | 3724 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 3725 | | subFilters.Add(alphaSrcFilter); |
| 0 | 3726 | | subFilters.Add("format=yuva420p"); |
| 0 | 3727 | | subFilters.Add(subTextSubtitlesFilter); |
| | 3728 | | } |
| | 3729 | |
|
| 0 | 3730 | | subFilters.Add("hwupload=derive_device=cuda"); |
| 0 | 3731 | | overlayFilters.Add("overlay_cuda=eof_action=pass:repeatlast=0"); |
| | 3732 | | } |
| | 3733 | | } |
| | 3734 | | else |
| | 3735 | | { |
| 0 | 3736 | | if (hasGraphicalSubs) |
| | 3737 | | { |
| 0 | 3738 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 3739 | | subFilters.Add(subPreProcFilters); |
| 0 | 3740 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 3741 | | } |
| | 3742 | | } |
| | 3743 | |
|
| 0 | 3744 | | return (mainFilters, subFilters, overlayFilters); |
| | 3745 | | } |
| | 3746 | |
|
| | 3747 | | /// <summary> |
| | 3748 | | /// Gets the parameter of AMD AMF filter chain. |
| | 3749 | | /// </summary> |
| | 3750 | | /// <param name="state">Encoding state.</param> |
| | 3751 | | /// <param name="options">Encoding options.</param> |
| | 3752 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 3753 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 3754 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetAmdVidFilterChain( |
| | 3755 | | EncodingJobInfo state, |
| | 3756 | | EncodingOptions options, |
| | 3757 | | string vidEncoder) |
| | 3758 | | { |
| 0 | 3759 | | if (options.HardwareAccelerationType != HardwareAccelerationType.amf) |
| | 3760 | | { |
| 0 | 3761 | | return (null, null, null); |
| | 3762 | | } |
| | 3763 | |
|
| 0 | 3764 | | var isWindows = OperatingSystem.IsWindows(); |
| 0 | 3765 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 3766 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 3767 | | var isSwEncoder = !vidEncoder.Contains("amf", StringComparison.OrdinalIgnoreCase); |
| 0 | 3768 | | var isAmfDx11OclSupported = isWindows && _mediaEncoder.SupportsHwaccel("d3d11va") && IsOpenclFullSupported() |
| | 3769 | |
|
| | 3770 | | // legacy d3d11va pipeline(copy-back) |
| 0 | 3771 | | if ((isSwDecoder && isSwEncoder) |
| 0 | 3772 | | || !isAmfDx11OclSupported |
| 0 | 3773 | | || !_mediaEncoder.SupportsFilter("alphasrc")) |
| | 3774 | | { |
| 0 | 3775 | | return GetSwVidFilterChain(state, options, vidEncoder); |
| | 3776 | | } |
| | 3777 | |
|
| | 3778 | | // prefered d3d11va + opencl filters + amf pipeline |
| 0 | 3779 | | return GetAmdDx11VidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 3780 | | } |
| | 3781 | |
|
| | 3782 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetAmdDx11VidFiltersPref |
| | 3783 | | EncodingJobInfo state, |
| | 3784 | | EncodingOptions options, |
| | 3785 | | string vidDecoder, |
| | 3786 | | string vidEncoder) |
| | 3787 | | { |
| 0 | 3788 | | var inW = state.VideoStream?.Width; |
| 0 | 3789 | | var inH = state.VideoStream?.Height; |
| 0 | 3790 | | var reqW = state.BaseRequest.Width; |
| 0 | 3791 | | var reqH = state.BaseRequest.Height; |
| 0 | 3792 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 3793 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 3794 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 3795 | |
|
| 0 | 3796 | | var isD3d11vaDecoder = vidDecoder.Contains("d3d11va", StringComparison.OrdinalIgnoreCase); |
| 0 | 3797 | | var isAmfEncoder = vidEncoder.Contains("amf", StringComparison.OrdinalIgnoreCase); |
| 0 | 3798 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 3799 | | var isSwEncoder = !isAmfEncoder; |
| 0 | 3800 | | var isDxInDxOut = isD3d11vaDecoder && isAmfEncoder; |
| | 3801 | |
|
| 0 | 3802 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 3803 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 3804 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 3805 | | var doOclTonemap = IsHwTonemapAvailable(state, options); |
| | 3806 | |
|
| 0 | 3807 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 3808 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 3809 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 3810 | | var hasAssSubs = hasSubs |
| 0 | 3811 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 3812 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| 0 | 3813 | | var subW = state.SubtitleStream?.Width; |
| 0 | 3814 | | var subH = state.SubtitleStream?.Height; |
| | 3815 | |
|
| 0 | 3816 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 3817 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 3818 | | var doOclTranspose = !string.IsNullOrEmpty(tranposeDir) |
| 0 | 3819 | | && _mediaEncoder.SupportsFilterWithOption(FilterOptionType.TransposeOpenclReversal); |
| 0 | 3820 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || (isD3d11vaDecoder && doOclTranspose)); |
| 0 | 3821 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 3822 | | var swpInH = swapWAndH ? inW : inH; |
| | 3823 | |
|
| | 3824 | | /* Make main filters for video stream */ |
| 0 | 3825 | | var mainFilters = new List<string>(); |
| | 3826 | |
|
| 0 | 3827 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doOclTonemap)); |
| | 3828 | |
|
| 0 | 3829 | | if (isSwDecoder) |
| | 3830 | | { |
| | 3831 | | // INPUT sw surface(memory) |
| | 3832 | | // sw deint |
| 0 | 3833 | | if (doDeintH2645) |
| | 3834 | | { |
| 0 | 3835 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 3836 | | mainFilters.Add(swDeintFilter); |
| | 3837 | | } |
| | 3838 | |
|
| 0 | 3839 | | var outFormat = doOclTonemap ? "yuv420p10le" : "yuv420p"; |
| 0 | 3840 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| | 3841 | | // sw scale |
| 0 | 3842 | | mainFilters.Add(swScaleFilter); |
| 0 | 3843 | | mainFilters.Add($"format={outFormat}"); |
| | 3844 | |
|
| | 3845 | | // keep video at memory except ocl tonemap, |
| | 3846 | | // since the overhead caused by hwupload >>> using sw filter. |
| | 3847 | | // sw => hw |
| 0 | 3848 | | if (doOclTonemap) |
| | 3849 | | { |
| 0 | 3850 | | mainFilters.Add("hwupload=derive_device=d3d11va:extra_hw_frames=24"); |
| 0 | 3851 | | mainFilters.Add("format=d3d11"); |
| 0 | 3852 | | mainFilters.Add("hwmap=derive_device=opencl:mode=read"); |
| | 3853 | | } |
| | 3854 | | } |
| | 3855 | |
|
| 0 | 3856 | | if (isD3d11vaDecoder) |
| | 3857 | | { |
| | 3858 | | // INPUT d3d11 surface(vram) |
| | 3859 | | // map from d3d11va to opencl via d3d11-opencl interop. |
| 0 | 3860 | | mainFilters.Add("hwmap=derive_device=opencl:mode=read"); |
| | 3861 | |
|
| | 3862 | | // hw deint <= TODO: finsh the 'yadif_opencl' filter |
| | 3863 | |
|
| | 3864 | | // hw transpose |
| 0 | 3865 | | if (doOclTranspose) |
| | 3866 | | { |
| 0 | 3867 | | mainFilters.Add($"transpose_opencl=dir={tranposeDir}"); |
| | 3868 | | } |
| | 3869 | |
|
| 0 | 3870 | | var outFormat = doOclTonemap ? string.Empty : "nv12"; |
| 0 | 3871 | | var hwScaleFilter = GetHwScaleFilter("scale", "opencl", outFormat, false, swpInW, swpInH, reqW, reqH, re |
| | 3872 | | // hw scale |
| 0 | 3873 | | mainFilters.Add(hwScaleFilter); |
| | 3874 | | } |
| | 3875 | |
|
| | 3876 | | // hw tonemap |
| 0 | 3877 | | if (doOclTonemap) |
| | 3878 | | { |
| 0 | 3879 | | var tonemapFilter = GetHwTonemapFilter(options, "opencl", "nv12"); |
| 0 | 3880 | | mainFilters.Add(tonemapFilter); |
| | 3881 | | } |
| | 3882 | |
|
| 0 | 3883 | | var memoryOutput = false; |
| 0 | 3884 | | var isUploadForOclTonemap = isSwDecoder && doOclTonemap; |
| 0 | 3885 | | if (isD3d11vaDecoder && isSwEncoder) |
| | 3886 | | { |
| 0 | 3887 | | memoryOutput = true; |
| | 3888 | |
|
| | 3889 | | // OUTPUT nv12 surface(memory) |
| | 3890 | | // prefer hwmap to hwdownload on opencl. |
| 0 | 3891 | | var hwTransferFilter = hasGraphicalSubs ? "hwdownload" : "hwmap=mode=read"; |
| 0 | 3892 | | mainFilters.Add(hwTransferFilter); |
| 0 | 3893 | | mainFilters.Add("format=nv12"); |
| | 3894 | | } |
| | 3895 | |
|
| | 3896 | | // OUTPUT yuv420p surface |
| 0 | 3897 | | if (isSwDecoder && isAmfEncoder && !isUploadForOclTonemap) |
| | 3898 | | { |
| 0 | 3899 | | memoryOutput = true; |
| | 3900 | | } |
| | 3901 | |
|
| 0 | 3902 | | if (memoryOutput) |
| | 3903 | | { |
| | 3904 | | // text subtitles |
| 0 | 3905 | | if (hasTextSubs) |
| | 3906 | | { |
| 0 | 3907 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 3908 | | mainFilters.Add(textSubtitlesFilter); |
| | 3909 | | } |
| | 3910 | | } |
| | 3911 | |
|
| 0 | 3912 | | if ((isDxInDxOut || isUploadForOclTonemap) && !hasSubs) |
| | 3913 | | { |
| | 3914 | | // OUTPUT d3d11(nv12) surface(vram) |
| | 3915 | | // reverse-mapping via d3d11-opencl interop. |
| 0 | 3916 | | mainFilters.Add("hwmap=derive_device=d3d11va:mode=write:reverse=1"); |
| 0 | 3917 | | mainFilters.Add("format=d3d11"); |
| | 3918 | | } |
| | 3919 | |
|
| | 3920 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 3921 | | var subFilters = new List<string>(); |
| 0 | 3922 | | var overlayFilters = new List<string>(); |
| 0 | 3923 | | if (isDxInDxOut || isUploadForOclTonemap) |
| | 3924 | | { |
| 0 | 3925 | | if (hasSubs) |
| | 3926 | | { |
| 0 | 3927 | | if (hasGraphicalSubs) |
| | 3928 | | { |
| 0 | 3929 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, |
| 0 | 3930 | | subFilters.Add(subPreProcFilters); |
| 0 | 3931 | | subFilters.Add("format=yuva420p"); |
| | 3932 | | } |
| 0 | 3933 | | else if (hasTextSubs) |
| | 3934 | | { |
| 0 | 3935 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 3936 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 3937 | |
|
| | 3938 | | // alphasrc=s=1280x720:r=10:start=0,format=yuva420p,subtitles,hwupload |
| 0 | 3939 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH, subF |
| 0 | 3940 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 3941 | | subFilters.Add(alphaSrcFilter); |
| 0 | 3942 | | subFilters.Add("format=yuva420p"); |
| 0 | 3943 | | subFilters.Add(subTextSubtitlesFilter); |
| | 3944 | | } |
| | 3945 | |
|
| 0 | 3946 | | subFilters.Add("hwupload=derive_device=opencl"); |
| 0 | 3947 | | overlayFilters.Add("overlay_opencl=eof_action=pass:repeatlast=0"); |
| 0 | 3948 | | overlayFilters.Add("hwmap=derive_device=d3d11va:mode=write:reverse=1"); |
| 0 | 3949 | | overlayFilters.Add("format=d3d11"); |
| | 3950 | | } |
| | 3951 | | } |
| 0 | 3952 | | else if (memoryOutput) |
| | 3953 | | { |
| 0 | 3954 | | if (hasGraphicalSubs) |
| | 3955 | | { |
| 0 | 3956 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 3957 | | subFilters.Add(subPreProcFilters); |
| 0 | 3958 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 3959 | | } |
| | 3960 | | } |
| | 3961 | |
|
| 0 | 3962 | | return (mainFilters, subFilters, overlayFilters); |
| | 3963 | | } |
| | 3964 | |
|
| | 3965 | | /// <summary> |
| | 3966 | | /// Gets the parameter of Intel QSV filter chain. |
| | 3967 | | /// </summary> |
| | 3968 | | /// <param name="state">Encoding state.</param> |
| | 3969 | | /// <param name="options">Encoding options.</param> |
| | 3970 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 3971 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 3972 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetIntelVidFilterChain( |
| | 3973 | | EncodingJobInfo state, |
| | 3974 | | EncodingOptions options, |
| | 3975 | | string vidEncoder) |
| | 3976 | | { |
| 0 | 3977 | | if (options.HardwareAccelerationType != HardwareAccelerationType.qsv) |
| | 3978 | | { |
| 0 | 3979 | | return (null, null, null); |
| | 3980 | | } |
| | 3981 | |
|
| 0 | 3982 | | var isWindows = OperatingSystem.IsWindows(); |
| 0 | 3983 | | var isLinux = OperatingSystem.IsLinux(); |
| 0 | 3984 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 3985 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 3986 | | var isSwEncoder = !vidEncoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 3987 | | var isQsvOclSupported = _mediaEncoder.SupportsHwaccel("qsv") && IsOpenclFullSupported(); |
| 0 | 3988 | | var isIntelDx11OclSupported = isWindows |
| 0 | 3989 | | && _mediaEncoder.SupportsHwaccel("d3d11va") |
| 0 | 3990 | | && isQsvOclSupported; |
| 0 | 3991 | | var isIntelVaapiOclSupported = isLinux |
| 0 | 3992 | | && IsVaapiSupported(state) |
| 0 | 3993 | | && isQsvOclSupported; |
| | 3994 | |
|
| | 3995 | | // legacy qsv pipeline(copy-back) |
| 0 | 3996 | | if ((isSwDecoder && isSwEncoder) |
| 0 | 3997 | | || (!isIntelVaapiOclSupported && !isIntelDx11OclSupported) |
| 0 | 3998 | | || !_mediaEncoder.SupportsFilter("alphasrc")) |
| | 3999 | | { |
| 0 | 4000 | | return GetSwVidFilterChain(state, options, vidEncoder); |
| | 4001 | | } |
| | 4002 | |
|
| | 4003 | | // prefered qsv(vaapi) + opencl filters pipeline |
| 0 | 4004 | | if (isIntelVaapiOclSupported) |
| | 4005 | | { |
| 0 | 4006 | | return GetIntelQsvVaapiVidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 4007 | | } |
| | 4008 | |
|
| | 4009 | | // prefered qsv(d3d11) + opencl filters pipeline |
| 0 | 4010 | | if (isIntelDx11OclSupported) |
| | 4011 | | { |
| 0 | 4012 | | return GetIntelQsvDx11VidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 4013 | | } |
| | 4014 | |
|
| 0 | 4015 | | return (null, null, null); |
| | 4016 | | } |
| | 4017 | |
|
| | 4018 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetIntelQsvDx11VidFilter |
| | 4019 | | EncodingJobInfo state, |
| | 4020 | | EncodingOptions options, |
| | 4021 | | string vidDecoder, |
| | 4022 | | string vidEncoder) |
| | 4023 | | { |
| 0 | 4024 | | var inW = state.VideoStream?.Width; |
| 0 | 4025 | | var inH = state.VideoStream?.Height; |
| 0 | 4026 | | var reqW = state.BaseRequest.Width; |
| 0 | 4027 | | var reqH = state.BaseRequest.Height; |
| 0 | 4028 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 4029 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 4030 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 4031 | |
|
| 0 | 4032 | | var isD3d11vaDecoder = vidDecoder.Contains("d3d11va", StringComparison.OrdinalIgnoreCase); |
| 0 | 4033 | | var isQsvDecoder = vidDecoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 4034 | | var isQsvEncoder = vidEncoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 4035 | | var isHwDecoder = isD3d11vaDecoder || isQsvDecoder; |
| 0 | 4036 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 4037 | | var isSwEncoder = !isQsvEncoder; |
| 0 | 4038 | | var isQsvInQsvOut = isHwDecoder && isQsvEncoder; |
| | 4039 | |
|
| 0 | 4040 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 4041 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 4042 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 4043 | | var doVppTonemap = IsIntelVppTonemapAvailable(state, options); |
| 0 | 4044 | | var doOclTonemap = !doVppTonemap && IsHwTonemapAvailable(state, options); |
| 0 | 4045 | | var doTonemap = doVppTonemap || doOclTonemap; |
| | 4046 | |
|
| 0 | 4047 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 4048 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4049 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4050 | | var hasAssSubs = hasSubs |
| 0 | 4051 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 4052 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| 0 | 4053 | | var subW = state.SubtitleStream?.Width; |
| 0 | 4054 | | var subH = state.SubtitleStream?.Height; |
| | 4055 | |
|
| 0 | 4056 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 4057 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 4058 | | var doVppTranspose = !string.IsNullOrEmpty(tranposeDir); |
| 0 | 4059 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || ((isD3d11vaDecoder || isQsvDecoder) && doVppTran |
| 0 | 4060 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 4061 | | var swpInH = swapWAndH ? inW : inH; |
| | 4062 | |
|
| | 4063 | | /* Make main filters for video stream */ |
| 0 | 4064 | | var mainFilters = new List<string>(); |
| | 4065 | |
|
| 0 | 4066 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doTonemap)); |
| | 4067 | |
|
| 0 | 4068 | | if (isSwDecoder) |
| | 4069 | | { |
| | 4070 | | // INPUT sw surface(memory) |
| | 4071 | | // sw deint |
| 0 | 4072 | | if (doDeintH2645) |
| | 4073 | | { |
| 0 | 4074 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 4075 | | mainFilters.Add(swDeintFilter); |
| | 4076 | | } |
| | 4077 | |
|
| 0 | 4078 | | var outFormat = doOclTonemap ? "yuv420p10le" : (hasGraphicalSubs ? "yuv420p" : "nv12"); |
| 0 | 4079 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| | 4080 | | // sw scale |
| 0 | 4081 | | mainFilters.Add(swScaleFilter); |
| 0 | 4082 | | mainFilters.Add($"format={outFormat}"); |
| | 4083 | |
|
| | 4084 | | // keep video at memory except ocl tonemap, |
| | 4085 | | // since the overhead caused by hwupload >>> using sw filter. |
| | 4086 | | // sw => hw |
| 0 | 4087 | | if (doOclTonemap) |
| | 4088 | | { |
| 0 | 4089 | | mainFilters.Add("hwupload=derive_device=opencl"); |
| | 4090 | | } |
| | 4091 | | } |
| 0 | 4092 | | else if (isD3d11vaDecoder || isQsvDecoder) |
| | 4093 | | { |
| 0 | 4094 | | var doVppProcamp = false; |
| 0 | 4095 | | var procampParams = string.Empty; |
| 0 | 4096 | | if (doVppTonemap) |
| | 4097 | | { |
| 0 | 4098 | | if (options.VppTonemappingBrightness != 0 |
| 0 | 4099 | | && options.VppTonemappingBrightness >= -100 |
| 0 | 4100 | | && options.VppTonemappingBrightness <= 100) |
| | 4101 | | { |
| 0 | 4102 | | procampParams += $":brightness={options.VppTonemappingBrightness}"; |
| 0 | 4103 | | doVppProcamp = true; |
| | 4104 | | } |
| | 4105 | |
|
| 0 | 4106 | | if (options.VppTonemappingContrast > 1 |
| 0 | 4107 | | && options.VppTonemappingContrast <= 10) |
| | 4108 | | { |
| 0 | 4109 | | procampParams += $":contrast={options.VppTonemappingContrast}"; |
| 0 | 4110 | | doVppProcamp = true; |
| | 4111 | | } |
| | 4112 | |
|
| 0 | 4113 | | procampParams += doVppProcamp ? ":procamp=1:async_depth=2" : string.Empty; |
| | 4114 | | } |
| | 4115 | |
|
| 0 | 4116 | | var outFormat = doOclTonemap ? (doVppTranspose ? "p010" : string.Empty) : "nv12"; |
| 0 | 4117 | | outFormat = (doVppTonemap && doVppProcamp) ? "p010" : outFormat; |
| | 4118 | |
|
| 0 | 4119 | | var swapOutputWandH = doVppTranspose && swapWAndH; |
| 0 | 4120 | | var hwScalePrefix = (doVppTranspose || doVppTonemap) ? "vpp" : "scale"; |
| 0 | 4121 | | var hwScaleFilter = GetHwScaleFilter(hwScalePrefix, "qsv", outFormat, swapOutputWandH, swpInW, swpInH, r |
| | 4122 | |
|
| 0 | 4123 | | if (!string.IsNullOrEmpty(hwScaleFilter) && doVppTranspose) |
| | 4124 | | { |
| 0 | 4125 | | hwScaleFilter += $":transpose={tranposeDir}"; |
| | 4126 | | } |
| | 4127 | |
|
| 0 | 4128 | | if (!string.IsNullOrEmpty(hwScaleFilter) && doVppTonemap) |
| | 4129 | | { |
| 0 | 4130 | | hwScaleFilter += doVppProcamp ? procampParams : ":tonemap=1"; |
| | 4131 | | } |
| | 4132 | |
|
| 0 | 4133 | | if (isD3d11vaDecoder) |
| | 4134 | | { |
| 0 | 4135 | | if (!string.IsNullOrEmpty(hwScaleFilter) || doDeintH2645) |
| | 4136 | | { |
| | 4137 | | // INPUT d3d11 surface(vram) |
| | 4138 | | // map from d3d11va to qsv. |
| 0 | 4139 | | mainFilters.Add("hwmap=derive_device=qsv"); |
| | 4140 | | } |
| | 4141 | | } |
| | 4142 | |
|
| | 4143 | | // hw deint |
| 0 | 4144 | | if (doDeintH2645) |
| | 4145 | | { |
| 0 | 4146 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "qsv"); |
| 0 | 4147 | | mainFilters.Add(deintFilter); |
| | 4148 | | } |
| | 4149 | |
|
| | 4150 | | // hw transpose & scale & tonemap(w/o procamp) |
| 0 | 4151 | | mainFilters.Add(hwScaleFilter); |
| | 4152 | |
|
| | 4153 | | // hw tonemap(w/ procamp) |
| 0 | 4154 | | if (doVppTonemap && doVppProcamp) |
| | 4155 | | { |
| 0 | 4156 | | mainFilters.Add("vpp_qsv=tonemap=1:format=nv12:async_depth=2"); |
| | 4157 | | } |
| | 4158 | |
|
| | 4159 | | // force bt709 just in case vpp tonemap is not triggered or using MSDK instead of VPL. |
| 0 | 4160 | | if (doVppTonemap) |
| | 4161 | | { |
| 0 | 4162 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, false)); |
| | 4163 | | } |
| | 4164 | | } |
| | 4165 | |
|
| 0 | 4166 | | if (doOclTonemap && isHwDecoder) |
| | 4167 | | { |
| | 4168 | | // map from qsv to opencl via qsv(d3d11)-opencl interop. |
| 0 | 4169 | | mainFilters.Add("hwmap=derive_device=opencl:mode=read"); |
| | 4170 | | } |
| | 4171 | |
|
| | 4172 | | // hw tonemap |
| 0 | 4173 | | if (doOclTonemap) |
| | 4174 | | { |
| 0 | 4175 | | var tonemapFilter = GetHwTonemapFilter(options, "opencl", "nv12"); |
| 0 | 4176 | | mainFilters.Add(tonemapFilter); |
| | 4177 | | } |
| | 4178 | |
|
| 0 | 4179 | | var memoryOutput = false; |
| 0 | 4180 | | var isUploadForOclTonemap = isSwDecoder && doOclTonemap; |
| 0 | 4181 | | var isHwmapUsable = isSwEncoder && doOclTonemap; |
| 0 | 4182 | | if ((isHwDecoder && isSwEncoder) || isUploadForOclTonemap) |
| | 4183 | | { |
| 0 | 4184 | | memoryOutput = true; |
| | 4185 | |
|
| | 4186 | | // OUTPUT nv12 surface(memory) |
| | 4187 | | // prefer hwmap to hwdownload on opencl. |
| | 4188 | | // qsv hwmap is not fully implemented for the time being. |
| 0 | 4189 | | mainFilters.Add(isHwmapUsable ? "hwmap=mode=read" : "hwdownload"); |
| 0 | 4190 | | mainFilters.Add("format=nv12"); |
| | 4191 | | } |
| | 4192 | |
|
| | 4193 | | // OUTPUT nv12 surface(memory) |
| 0 | 4194 | | if (isSwDecoder && isQsvEncoder) |
| | 4195 | | { |
| 0 | 4196 | | memoryOutput = true; |
| | 4197 | | } |
| | 4198 | |
|
| 0 | 4199 | | if (memoryOutput) |
| | 4200 | | { |
| | 4201 | | // text subtitles |
| 0 | 4202 | | if (hasTextSubs) |
| | 4203 | | { |
| 0 | 4204 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 4205 | | mainFilters.Add(textSubtitlesFilter); |
| | 4206 | | } |
| | 4207 | | } |
| | 4208 | |
|
| 0 | 4209 | | if (isQsvInQsvOut && doOclTonemap) |
| | 4210 | | { |
| | 4211 | | // OUTPUT qsv(nv12) surface(vram) |
| | 4212 | | // reverse-mapping via qsv(d3d11)-opencl interop. |
| 0 | 4213 | | mainFilters.Add("hwmap=derive_device=qsv:mode=write:reverse=1"); |
| 0 | 4214 | | mainFilters.Add("format=qsv"); |
| | 4215 | | } |
| | 4216 | |
|
| | 4217 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 4218 | | var subFilters = new List<string>(); |
| 0 | 4219 | | var overlayFilters = new List<string>(); |
| 0 | 4220 | | if (isQsvInQsvOut) |
| | 4221 | | { |
| 0 | 4222 | | if (hasSubs) |
| | 4223 | | { |
| 0 | 4224 | | if (hasGraphicalSubs) |
| | 4225 | | { |
| | 4226 | | // overlay_qsv can handle overlay scaling, setup a smaller height to reduce transfer overhead |
| 0 | 4227 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, |
| 0 | 4228 | | subFilters.Add(subPreProcFilters); |
| 0 | 4229 | | subFilters.Add("format=bgra"); |
| | 4230 | | } |
| 0 | 4231 | | else if (hasTextSubs) |
| | 4232 | | { |
| 0 | 4233 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 4234 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 4235 | |
|
| | 4236 | | // alphasrc=s=1280x720:r=10:start=0,format=bgra,subtitles,hwupload |
| 0 | 4237 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, 1080, subFram |
| 0 | 4238 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 4239 | | subFilters.Add(alphaSrcFilter); |
| 0 | 4240 | | subFilters.Add("format=bgra"); |
| 0 | 4241 | | subFilters.Add(subTextSubtitlesFilter); |
| | 4242 | | } |
| | 4243 | |
|
| | 4244 | | // qsv requires a fixed pool size. |
| | 4245 | | // default to 64 otherwise it will fail on certain iGPU. |
| 0 | 4246 | | subFilters.Add("hwupload=derive_device=qsv:extra_hw_frames=64"); |
| | 4247 | |
|
| 0 | 4248 | | var (overlayW, overlayH) = GetFixedOutputSize(swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH); |
| 0 | 4249 | | var overlaySize = (overlayW.HasValue && overlayH.HasValue) |
| 0 | 4250 | | ? $":w={overlayW.Value}:h={overlayH.Value}" |
| 0 | 4251 | | : string.Empty; |
| 0 | 4252 | | var overlayQsvFilter = string.Format( |
| 0 | 4253 | | CultureInfo.InvariantCulture, |
| 0 | 4254 | | "overlay_qsv=eof_action=pass:repeatlast=0{0}", |
| 0 | 4255 | | overlaySize); |
| 0 | 4256 | | overlayFilters.Add(overlayQsvFilter); |
| | 4257 | | } |
| | 4258 | | } |
| 0 | 4259 | | else if (memoryOutput) |
| | 4260 | | { |
| 0 | 4261 | | if (hasGraphicalSubs) |
| | 4262 | | { |
| 0 | 4263 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 4264 | | subFilters.Add(subPreProcFilters); |
| 0 | 4265 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 4266 | | } |
| | 4267 | | } |
| | 4268 | |
|
| 0 | 4269 | | return (mainFilters, subFilters, overlayFilters); |
| | 4270 | | } |
| | 4271 | |
|
| | 4272 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetIntelQsvVaapiVidFilte |
| | 4273 | | EncodingJobInfo state, |
| | 4274 | | EncodingOptions options, |
| | 4275 | | string vidDecoder, |
| | 4276 | | string vidEncoder) |
| | 4277 | | { |
| 0 | 4278 | | var inW = state.VideoStream?.Width; |
| 0 | 4279 | | var inH = state.VideoStream?.Height; |
| 0 | 4280 | | var reqW = state.BaseRequest.Width; |
| 0 | 4281 | | var reqH = state.BaseRequest.Height; |
| 0 | 4282 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 4283 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 4284 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 4285 | |
|
| 0 | 4286 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 4287 | | var isQsvDecoder = vidDecoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 4288 | | var isQsvEncoder = vidEncoder.Contains("qsv", StringComparison.OrdinalIgnoreCase); |
| 0 | 4289 | | var isHwDecoder = isVaapiDecoder || isQsvDecoder; |
| 0 | 4290 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 4291 | | var isSwEncoder = !isQsvEncoder; |
| 0 | 4292 | | var isQsvInQsvOut = isHwDecoder && isQsvEncoder; |
| | 4293 | |
|
| 0 | 4294 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 4295 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 4296 | | var doVaVppTonemap = IsIntelVppTonemapAvailable(state, options); |
| 0 | 4297 | | var doOclTonemap = !doVaVppTonemap && IsHwTonemapAvailable(state, options); |
| 0 | 4298 | | var doTonemap = doVaVppTonemap || doOclTonemap; |
| 0 | 4299 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| | 4300 | |
|
| 0 | 4301 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 4302 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4303 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4304 | | var hasAssSubs = hasSubs |
| 0 | 4305 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 4306 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| 0 | 4307 | | var subW = state.SubtitleStream?.Width; |
| 0 | 4308 | | var subH = state.SubtitleStream?.Height; |
| | 4309 | |
|
| 0 | 4310 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 4311 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 4312 | | var doVppTranspose = !string.IsNullOrEmpty(tranposeDir); |
| 0 | 4313 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || ((isVaapiDecoder || isQsvDecoder) && doVppTransp |
| 0 | 4314 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 4315 | | var swpInH = swapWAndH ? inW : inH; |
| | 4316 | |
|
| | 4317 | | /* Make main filters for video stream */ |
| 0 | 4318 | | var mainFilters = new List<string>(); |
| | 4319 | |
|
| 0 | 4320 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doTonemap)); |
| | 4321 | |
|
| 0 | 4322 | | if (isSwDecoder) |
| | 4323 | | { |
| | 4324 | | // INPUT sw surface(memory) |
| | 4325 | | // sw deint |
| 0 | 4326 | | if (doDeintH2645) |
| | 4327 | | { |
| 0 | 4328 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 4329 | | mainFilters.Add(swDeintFilter); |
| | 4330 | | } |
| | 4331 | |
|
| 0 | 4332 | | var outFormat = doOclTonemap ? "yuv420p10le" : (hasGraphicalSubs ? "yuv420p" : "nv12"); |
| 0 | 4333 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| | 4334 | | // sw scale |
| 0 | 4335 | | mainFilters.Add(swScaleFilter); |
| 0 | 4336 | | mainFilters.Add($"format={outFormat}"); |
| | 4337 | |
|
| | 4338 | | // keep video at memory except ocl tonemap, |
| | 4339 | | // since the overhead caused by hwupload >>> using sw filter. |
| | 4340 | | // sw => hw |
| 0 | 4341 | | if (doOclTonemap) |
| | 4342 | | { |
| 0 | 4343 | | mainFilters.Add("hwupload=derive_device=opencl"); |
| | 4344 | | } |
| | 4345 | | } |
| 0 | 4346 | | else if (isVaapiDecoder || isQsvDecoder) |
| | 4347 | | { |
| 0 | 4348 | | var hwFilterSuffix = isVaapiDecoder ? "vaapi" : "qsv"; |
| | 4349 | |
|
| | 4350 | | // INPUT vaapi/qsv surface(vram) |
| | 4351 | | // hw deint |
| 0 | 4352 | | if (doDeintH2645) |
| | 4353 | | { |
| 0 | 4354 | | var deintFilter = GetHwDeinterlaceFilter(state, options, hwFilterSuffix); |
| 0 | 4355 | | mainFilters.Add(deintFilter); |
| | 4356 | | } |
| | 4357 | |
|
| | 4358 | | // hw transpose(vaapi vpp) |
| 0 | 4359 | | if (isVaapiDecoder && doVppTranspose) |
| | 4360 | | { |
| 0 | 4361 | | mainFilters.Add($"transpose_vaapi=dir={tranposeDir}"); |
| | 4362 | | } |
| | 4363 | |
|
| 0 | 4364 | | var outFormat = doOclTonemap ? ((isQsvDecoder && doVppTranspose) ? "p010" : string.Empty) : "nv12"; |
| 0 | 4365 | | var swapOutputWandH = isQsvDecoder && doVppTranspose && swapWAndH; |
| 0 | 4366 | | var hwScalePrefix = (isQsvDecoder && doVppTranspose) ? "vpp" : "scale"; |
| 0 | 4367 | | var hwScaleFilter = GetHwScaleFilter(hwScalePrefix, hwFilterSuffix, outFormat, swapOutputWandH, swpInW, |
| | 4368 | |
|
| 0 | 4369 | | if (!string.IsNullOrEmpty(hwScaleFilter) && isQsvDecoder && doVppTranspose) |
| | 4370 | | { |
| 0 | 4371 | | hwScaleFilter += $":transpose={tranposeDir}"; |
| | 4372 | | } |
| | 4373 | |
|
| | 4374 | | // allocate extra pool sizes for vaapi vpp scale |
| 0 | 4375 | | if (!string.IsNullOrEmpty(hwScaleFilter) && isVaapiDecoder) |
| | 4376 | | { |
| 0 | 4377 | | hwScaleFilter += ":extra_hw_frames=24"; |
| | 4378 | | } |
| | 4379 | |
|
| | 4380 | | // hw transpose(qsv vpp) & scale |
| 0 | 4381 | | mainFilters.Add(hwScaleFilter); |
| | 4382 | | } |
| | 4383 | |
|
| | 4384 | | // vaapi vpp tonemap |
| 0 | 4385 | | if (doVaVppTonemap && isHwDecoder) |
| | 4386 | | { |
| 0 | 4387 | | if (isQsvDecoder) |
| | 4388 | | { |
| | 4389 | | // map from qsv to vaapi. |
| 0 | 4390 | | mainFilters.Add("hwmap=derive_device=vaapi"); |
| 0 | 4391 | | mainFilters.Add("format=vaapi"); |
| | 4392 | | } |
| | 4393 | |
|
| 0 | 4394 | | var tonemapFilter = GetHwTonemapFilter(options, "vaapi", "nv12"); |
| 0 | 4395 | | mainFilters.Add(tonemapFilter); |
| | 4396 | |
|
| 0 | 4397 | | if (isQsvDecoder) |
| | 4398 | | { |
| | 4399 | | // map from vaapi to qsv. |
| 0 | 4400 | | mainFilters.Add("hwmap=derive_device=qsv"); |
| 0 | 4401 | | mainFilters.Add("format=qsv"); |
| | 4402 | | } |
| | 4403 | | } |
| | 4404 | |
|
| 0 | 4405 | | if (doOclTonemap && isHwDecoder) |
| | 4406 | | { |
| | 4407 | | // map from qsv to opencl via qsv(vaapi)-opencl interop. |
| 0 | 4408 | | mainFilters.Add("hwmap=derive_device=opencl:mode=read"); |
| | 4409 | | } |
| | 4410 | |
|
| | 4411 | | // ocl tonemap |
| 0 | 4412 | | if (doOclTonemap) |
| | 4413 | | { |
| 0 | 4414 | | var tonemapFilter = GetHwTonemapFilter(options, "opencl", "nv12"); |
| 0 | 4415 | | mainFilters.Add(tonemapFilter); |
| | 4416 | | } |
| | 4417 | |
|
| 0 | 4418 | | var memoryOutput = false; |
| 0 | 4419 | | var isUploadForOclTonemap = isSwDecoder && doOclTonemap; |
| 0 | 4420 | | var isHwmapUsable = isSwEncoder && (doOclTonemap || isVaapiDecoder); |
| 0 | 4421 | | if ((isHwDecoder && isSwEncoder) || isUploadForOclTonemap) |
| | 4422 | | { |
| 0 | 4423 | | memoryOutput = true; |
| | 4424 | |
|
| | 4425 | | // OUTPUT nv12 surface(memory) |
| | 4426 | | // prefer hwmap to hwdownload on opencl/vaapi. |
| | 4427 | | // qsv hwmap is not fully implemented for the time being. |
| 0 | 4428 | | mainFilters.Add(isHwmapUsable ? "hwmap=mode=read" : "hwdownload"); |
| 0 | 4429 | | mainFilters.Add("format=nv12"); |
| | 4430 | | } |
| | 4431 | |
|
| | 4432 | | // OUTPUT nv12 surface(memory) |
| 0 | 4433 | | if (isSwDecoder && isQsvEncoder) |
| | 4434 | | { |
| 0 | 4435 | | memoryOutput = true; |
| | 4436 | | } |
| | 4437 | |
|
| 0 | 4438 | | if (memoryOutput) |
| | 4439 | | { |
| | 4440 | | // text subtitles |
| 0 | 4441 | | if (hasTextSubs) |
| | 4442 | | { |
| 0 | 4443 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 4444 | | mainFilters.Add(textSubtitlesFilter); |
| | 4445 | | } |
| | 4446 | | } |
| | 4447 | |
|
| 0 | 4448 | | if (isQsvInQsvOut) |
| | 4449 | | { |
| 0 | 4450 | | if (doOclTonemap) |
| | 4451 | | { |
| | 4452 | | // OUTPUT qsv(nv12) surface(vram) |
| | 4453 | | // reverse-mapping via qsv(vaapi)-opencl interop. |
| | 4454 | | // add extra pool size to avoid the 'cannot allocate memory' error on hevc_qsv. |
| 0 | 4455 | | mainFilters.Add("hwmap=derive_device=qsv:mode=write:reverse=1:extra_hw_frames=16"); |
| 0 | 4456 | | mainFilters.Add("format=qsv"); |
| | 4457 | | } |
| 0 | 4458 | | else if (isVaapiDecoder) |
| | 4459 | | { |
| 0 | 4460 | | mainFilters.Add("hwmap=derive_device=qsv"); |
| 0 | 4461 | | mainFilters.Add("format=qsv"); |
| | 4462 | | } |
| | 4463 | | } |
| | 4464 | |
|
| | 4465 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 4466 | | var subFilters = new List<string>(); |
| 0 | 4467 | | var overlayFilters = new List<string>(); |
| 0 | 4468 | | if (isQsvInQsvOut) |
| | 4469 | | { |
| 0 | 4470 | | if (hasSubs) |
| | 4471 | | { |
| 0 | 4472 | | if (hasGraphicalSubs) |
| | 4473 | | { |
| | 4474 | | // overlay_qsv can handle overlay scaling, setup a smaller height to reduce transfer overhead |
| 0 | 4475 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, |
| 0 | 4476 | | subFilters.Add(subPreProcFilters); |
| 0 | 4477 | | subFilters.Add("format=bgra"); |
| | 4478 | | } |
| 0 | 4479 | | else if (hasTextSubs) |
| | 4480 | | { |
| 0 | 4481 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 4482 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 4483 | |
|
| 0 | 4484 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, 1080, subFram |
| 0 | 4485 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 4486 | | subFilters.Add(alphaSrcFilter); |
| 0 | 4487 | | subFilters.Add("format=bgra"); |
| 0 | 4488 | | subFilters.Add(subTextSubtitlesFilter); |
| | 4489 | | } |
| | 4490 | |
|
| | 4491 | | // qsv requires a fixed pool size. |
| | 4492 | | // default to 64 otherwise it will fail on certain iGPU. |
| 0 | 4493 | | subFilters.Add("hwupload=derive_device=qsv:extra_hw_frames=64"); |
| | 4494 | |
|
| 0 | 4495 | | var (overlayW, overlayH) = GetFixedOutputSize(swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH); |
| 0 | 4496 | | var overlaySize = (overlayW.HasValue && overlayH.HasValue) |
| 0 | 4497 | | ? $":w={overlayW.Value}:h={overlayH.Value}" |
| 0 | 4498 | | : string.Empty; |
| 0 | 4499 | | var overlayQsvFilter = string.Format( |
| 0 | 4500 | | CultureInfo.InvariantCulture, |
| 0 | 4501 | | "overlay_qsv=eof_action=pass:repeatlast=0{0}", |
| 0 | 4502 | | overlaySize); |
| 0 | 4503 | | overlayFilters.Add(overlayQsvFilter); |
| | 4504 | | } |
| | 4505 | | } |
| 0 | 4506 | | else if (memoryOutput) |
| | 4507 | | { |
| 0 | 4508 | | if (hasGraphicalSubs) |
| | 4509 | | { |
| 0 | 4510 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 4511 | | subFilters.Add(subPreProcFilters); |
| 0 | 4512 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 4513 | | } |
| | 4514 | | } |
| | 4515 | |
|
| 0 | 4516 | | return (mainFilters, subFilters, overlayFilters); |
| | 4517 | | } |
| | 4518 | |
|
| | 4519 | | /// <summary> |
| | 4520 | | /// Gets the parameter of Intel/AMD VAAPI filter chain. |
| | 4521 | | /// </summary> |
| | 4522 | | /// <param name="state">Encoding state.</param> |
| | 4523 | | /// <param name="options">Encoding options.</param> |
| | 4524 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 4525 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 4526 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetVaapiVidFilterChain( |
| | 4527 | | EncodingJobInfo state, |
| | 4528 | | EncodingOptions options, |
| | 4529 | | string vidEncoder) |
| | 4530 | | { |
| 0 | 4531 | | if (options.HardwareAccelerationType != HardwareAccelerationType.vaapi) |
| | 4532 | | { |
| 0 | 4533 | | return (null, null, null); |
| | 4534 | | } |
| | 4535 | |
|
| 0 | 4536 | | var isLinux = OperatingSystem.IsLinux(); |
| 0 | 4537 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 4538 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 4539 | | var isSwEncoder = !vidEncoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 4540 | | var isVaapiFullSupported = isLinux && IsVaapiSupported(state) && IsVaapiFullSupported(); |
| 0 | 4541 | | var isVaapiOclSupported = isVaapiFullSupported && IsOpenclFullSupported(); |
| 0 | 4542 | | var isVaapiVkSupported = isVaapiFullSupported && IsVulkanFullSupported(); |
| | 4543 | |
|
| | 4544 | | // legacy vaapi pipeline(copy-back) |
| 0 | 4545 | | if ((isSwDecoder && isSwEncoder) |
| 0 | 4546 | | || !isVaapiOclSupported |
| 0 | 4547 | | || !_mediaEncoder.SupportsFilter("alphasrc")) |
| | 4548 | | { |
| 0 | 4549 | | var swFilterChain = GetSwVidFilterChain(state, options, vidEncoder); |
| | 4550 | |
|
| 0 | 4551 | | if (!isSwEncoder) |
| | 4552 | | { |
| 0 | 4553 | | var newfilters = new List<string>(); |
| 0 | 4554 | | var noOverlay = swFilterChain.OverlayFilters.Count == 0; |
| 0 | 4555 | | newfilters.AddRange(noOverlay ? swFilterChain.MainFilters : swFilterChain.OverlayFilters); |
| 0 | 4556 | | newfilters.Add("hwupload=derive_device=vaapi"); |
| | 4557 | |
|
| 0 | 4558 | | var mainFilters = noOverlay ? newfilters : swFilterChain.MainFilters; |
| 0 | 4559 | | var overlayFilters = noOverlay ? swFilterChain.OverlayFilters : newfilters; |
| 0 | 4560 | | return (mainFilters, swFilterChain.SubFilters, overlayFilters); |
| | 4561 | | } |
| | 4562 | |
|
| 0 | 4563 | | return swFilterChain; |
| | 4564 | | } |
| | 4565 | |
|
| | 4566 | | // prefered vaapi + opencl filters pipeline |
| 0 | 4567 | | if (_mediaEncoder.IsVaapiDeviceInteliHD) |
| | 4568 | | { |
| | 4569 | | // Intel iHD path, with extra vpp tonemap and overlay support. |
| 0 | 4570 | | return GetIntelVaapiFullVidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 4571 | | } |
| | 4572 | |
|
| | 4573 | | // prefered vaapi + vulkan filters pipeline |
| 0 | 4574 | | if (_mediaEncoder.IsVaapiDeviceAmd |
| 0 | 4575 | | && isVaapiVkSupported |
| 0 | 4576 | | && _mediaEncoder.IsVaapiDeviceSupportVulkanDrmInterop |
| 0 | 4577 | | && Environment.OSVersion.Version >= _minKernelVersionAmdVkFmtModifier) |
| | 4578 | | { |
| | 4579 | | // AMD radeonsi path(targeting Polaris/gfx8+), with extra vulkan tonemap and overlay support. |
| 0 | 4580 | | return GetAmdVaapiFullVidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 4581 | | } |
| | 4582 | |
|
| | 4583 | | // Intel i965 and Amd legacy driver path, only featuring scale and deinterlace support. |
| 0 | 4584 | | return GetVaapiLimitedVidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 4585 | | } |
| | 4586 | |
|
| | 4587 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetIntelVaapiFullVidFilt |
| | 4588 | | EncodingJobInfo state, |
| | 4589 | | EncodingOptions options, |
| | 4590 | | string vidDecoder, |
| | 4591 | | string vidEncoder) |
| | 4592 | | { |
| 0 | 4593 | | var inW = state.VideoStream?.Width; |
| 0 | 4594 | | var inH = state.VideoStream?.Height; |
| 0 | 4595 | | var reqW = state.BaseRequest.Width; |
| 0 | 4596 | | var reqH = state.BaseRequest.Height; |
| 0 | 4597 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 4598 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 4599 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 4600 | |
|
| 0 | 4601 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 4602 | | var isVaapiEncoder = vidEncoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 4603 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 4604 | | var isSwEncoder = !isVaapiEncoder; |
| 0 | 4605 | | var isVaInVaOut = isVaapiDecoder && isVaapiEncoder; |
| | 4606 | |
|
| 0 | 4607 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 4608 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 4609 | | var doVaVppTonemap = isVaapiDecoder && IsIntelVppTonemapAvailable(state, options); |
| 0 | 4610 | | var doOclTonemap = !doVaVppTonemap && IsHwTonemapAvailable(state, options); |
| 0 | 4611 | | var doTonemap = doVaVppTonemap || doOclTonemap; |
| 0 | 4612 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| | 4613 | |
|
| 0 | 4614 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 4615 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4616 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4617 | | var hasAssSubs = hasSubs |
| 0 | 4618 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 4619 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| 0 | 4620 | | var subW = state.SubtitleStream?.Width; |
| 0 | 4621 | | var subH = state.SubtitleStream?.Height; |
| | 4622 | |
|
| 0 | 4623 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 4624 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 4625 | | var doVaVppTranspose = !string.IsNullOrEmpty(tranposeDir); |
| 0 | 4626 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || (isVaapiDecoder && doVaVppTranspose)); |
| 0 | 4627 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 4628 | | var swpInH = swapWAndH ? inW : inH; |
| | 4629 | |
|
| | 4630 | | /* Make main filters for video stream */ |
| 0 | 4631 | | var mainFilters = new List<string>(); |
| | 4632 | |
|
| 0 | 4633 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doTonemap)); |
| | 4634 | |
|
| 0 | 4635 | | if (isSwDecoder) |
| | 4636 | | { |
| | 4637 | | // INPUT sw surface(memory) |
| | 4638 | | // sw deint |
| 0 | 4639 | | if (doDeintH2645) |
| | 4640 | | { |
| 0 | 4641 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 4642 | | mainFilters.Add(swDeintFilter); |
| | 4643 | | } |
| | 4644 | |
|
| 0 | 4645 | | var outFormat = doOclTonemap ? "yuv420p10le" : "nv12"; |
| 0 | 4646 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| | 4647 | | // sw scale |
| 0 | 4648 | | mainFilters.Add(swScaleFilter); |
| 0 | 4649 | | mainFilters.Add($"format={outFormat}"); |
| | 4650 | |
|
| | 4651 | | // keep video at memory except ocl tonemap, |
| | 4652 | | // since the overhead caused by hwupload >>> using sw filter. |
| | 4653 | | // sw => hw |
| 0 | 4654 | | if (doOclTonemap) |
| | 4655 | | { |
| 0 | 4656 | | mainFilters.Add("hwupload=derive_device=opencl"); |
| | 4657 | | } |
| | 4658 | | } |
| 0 | 4659 | | else if (isVaapiDecoder) |
| | 4660 | | { |
| | 4661 | | // INPUT vaapi surface(vram) |
| | 4662 | | // hw deint |
| 0 | 4663 | | if (doDeintH2645) |
| | 4664 | | { |
| 0 | 4665 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "vaapi"); |
| 0 | 4666 | | mainFilters.Add(deintFilter); |
| | 4667 | | } |
| | 4668 | |
|
| | 4669 | | // hw transpose |
| 0 | 4670 | | if (doVaVppTranspose) |
| | 4671 | | { |
| 0 | 4672 | | mainFilters.Add($"transpose_vaapi=dir={tranposeDir}"); |
| | 4673 | | } |
| | 4674 | |
|
| 0 | 4675 | | var outFormat = doTonemap ? string.Empty : "nv12"; |
| 0 | 4676 | | var hwScaleFilter = GetHwScaleFilter("scale", "vaapi", outFormat, false, swpInW, swpInH, reqW, reqH, req |
| | 4677 | |
|
| | 4678 | | // allocate extra pool sizes for vaapi vpp |
| 0 | 4679 | | if (!string.IsNullOrEmpty(hwScaleFilter)) |
| | 4680 | | { |
| 0 | 4681 | | hwScaleFilter += ":extra_hw_frames=24"; |
| | 4682 | | } |
| | 4683 | |
|
| | 4684 | | // hw scale |
| 0 | 4685 | | mainFilters.Add(hwScaleFilter); |
| | 4686 | | } |
| | 4687 | |
|
| | 4688 | | // vaapi vpp tonemap |
| 0 | 4689 | | if (doVaVppTonemap && isVaapiDecoder) |
| | 4690 | | { |
| 0 | 4691 | | var tonemapFilter = GetHwTonemapFilter(options, "vaapi", "nv12"); |
| 0 | 4692 | | mainFilters.Add(tonemapFilter); |
| | 4693 | | } |
| | 4694 | |
|
| 0 | 4695 | | if (doOclTonemap && isVaapiDecoder) |
| | 4696 | | { |
| | 4697 | | // map from vaapi to opencl via vaapi-opencl interop(Intel only). |
| 0 | 4698 | | mainFilters.Add("hwmap=derive_device=opencl:mode=read"); |
| | 4699 | | } |
| | 4700 | |
|
| | 4701 | | // ocl tonemap |
| 0 | 4702 | | if (doOclTonemap) |
| | 4703 | | { |
| 0 | 4704 | | var tonemapFilter = GetHwTonemapFilter(options, "opencl", "nv12"); |
| 0 | 4705 | | mainFilters.Add(tonemapFilter); |
| | 4706 | | } |
| | 4707 | |
|
| 0 | 4708 | | if (doOclTonemap && isVaInVaOut) |
| | 4709 | | { |
| | 4710 | | // OUTPUT vaapi(nv12) surface(vram) |
| | 4711 | | // reverse-mapping via vaapi-opencl interop. |
| 0 | 4712 | | mainFilters.Add("hwmap=derive_device=vaapi:mode=write:reverse=1"); |
| 0 | 4713 | | mainFilters.Add("format=vaapi"); |
| | 4714 | | } |
| | 4715 | |
|
| 0 | 4716 | | var memoryOutput = false; |
| 0 | 4717 | | var isUploadForOclTonemap = isSwDecoder && doOclTonemap; |
| 0 | 4718 | | var isHwmapNotUsable = isUploadForOclTonemap && isVaapiEncoder; |
| 0 | 4719 | | if ((isVaapiDecoder && isSwEncoder) || isUploadForOclTonemap) |
| | 4720 | | { |
| 0 | 4721 | | memoryOutput = true; |
| | 4722 | |
|
| | 4723 | | // OUTPUT nv12 surface(memory) |
| | 4724 | | // prefer hwmap to hwdownload on opencl/vaapi. |
| 0 | 4725 | | mainFilters.Add(isHwmapNotUsable ? "hwdownload" : "hwmap=mode=read"); |
| 0 | 4726 | | mainFilters.Add("format=nv12"); |
| | 4727 | | } |
| | 4728 | |
|
| | 4729 | | // OUTPUT nv12 surface(memory) |
| 0 | 4730 | | if (isSwDecoder && isVaapiEncoder) |
| | 4731 | | { |
| 0 | 4732 | | memoryOutput = true; |
| | 4733 | | } |
| | 4734 | |
|
| 0 | 4735 | | if (memoryOutput) |
| | 4736 | | { |
| | 4737 | | // text subtitles |
| 0 | 4738 | | if (hasTextSubs) |
| | 4739 | | { |
| 0 | 4740 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 4741 | | mainFilters.Add(textSubtitlesFilter); |
| | 4742 | | } |
| | 4743 | | } |
| | 4744 | |
|
| 0 | 4745 | | if (memoryOutput && isVaapiEncoder) |
| | 4746 | | { |
| 0 | 4747 | | if (!hasGraphicalSubs) |
| | 4748 | | { |
| 0 | 4749 | | mainFilters.Add("hwupload_vaapi"); |
| | 4750 | | } |
| | 4751 | | } |
| | 4752 | |
|
| | 4753 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 4754 | | var subFilters = new List<string>(); |
| 0 | 4755 | | var overlayFilters = new List<string>(); |
| 0 | 4756 | | if (isVaInVaOut) |
| | 4757 | | { |
| 0 | 4758 | | if (hasSubs) |
| | 4759 | | { |
| 0 | 4760 | | if (hasGraphicalSubs) |
| | 4761 | | { |
| | 4762 | | // overlay_vaapi can handle overlay scaling, setup a smaller height to reduce transfer overhead |
| 0 | 4763 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, |
| 0 | 4764 | | subFilters.Add(subPreProcFilters); |
| 0 | 4765 | | subFilters.Add("format=bgra"); |
| | 4766 | | } |
| 0 | 4767 | | else if (hasTextSubs) |
| | 4768 | | { |
| 0 | 4769 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 4770 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 4771 | |
|
| 0 | 4772 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, 1080, subFram |
| 0 | 4773 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 4774 | | subFilters.Add(alphaSrcFilter); |
| 0 | 4775 | | subFilters.Add("format=bgra"); |
| 0 | 4776 | | subFilters.Add(subTextSubtitlesFilter); |
| | 4777 | | } |
| | 4778 | |
|
| 0 | 4779 | | subFilters.Add("hwupload=derive_device=vaapi"); |
| | 4780 | |
|
| 0 | 4781 | | var (overlayW, overlayH) = GetFixedOutputSize(swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH); |
| 0 | 4782 | | var overlaySize = (overlayW.HasValue && overlayH.HasValue) |
| 0 | 4783 | | ? $":w={overlayW.Value}:h={overlayH.Value}" |
| 0 | 4784 | | : string.Empty; |
| 0 | 4785 | | var overlayVaapiFilter = string.Format( |
| 0 | 4786 | | CultureInfo.InvariantCulture, |
| 0 | 4787 | | "overlay_vaapi=eof_action=pass:repeatlast=0{0}", |
| 0 | 4788 | | overlaySize); |
| 0 | 4789 | | overlayFilters.Add(overlayVaapiFilter); |
| | 4790 | | } |
| | 4791 | | } |
| 0 | 4792 | | else if (memoryOutput) |
| | 4793 | | { |
| 0 | 4794 | | if (hasGraphicalSubs) |
| | 4795 | | { |
| 0 | 4796 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 4797 | | subFilters.Add(subPreProcFilters); |
| 0 | 4798 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 4799 | |
|
| 0 | 4800 | | if (isVaapiEncoder) |
| | 4801 | | { |
| 0 | 4802 | | overlayFilters.Add("hwupload_vaapi"); |
| | 4803 | | } |
| | 4804 | | } |
| | 4805 | | } |
| | 4806 | |
|
| 0 | 4807 | | return (mainFilters, subFilters, overlayFilters); |
| | 4808 | | } |
| | 4809 | |
|
| | 4810 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetAmdVaapiFullVidFilter |
| | 4811 | | EncodingJobInfo state, |
| | 4812 | | EncodingOptions options, |
| | 4813 | | string vidDecoder, |
| | 4814 | | string vidEncoder) |
| | 4815 | | { |
| 0 | 4816 | | var inW = state.VideoStream?.Width; |
| 0 | 4817 | | var inH = state.VideoStream?.Height; |
| 0 | 4818 | | var reqW = state.BaseRequest.Width; |
| 0 | 4819 | | var reqH = state.BaseRequest.Height; |
| 0 | 4820 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 4821 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 4822 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 4823 | |
|
| 0 | 4824 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 4825 | | var isVaapiEncoder = vidEncoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 4826 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 4827 | | var isSwEncoder = !isVaapiEncoder; |
| | 4828 | |
|
| 0 | 4829 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 4830 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 4831 | | var doVkTonemap = IsVulkanHwTonemapAvailable(state, options); |
| 0 | 4832 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| | 4833 | |
|
| 0 | 4834 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 4835 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4836 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 4837 | | var hasAssSubs = hasSubs |
| 0 | 4838 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 4839 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| | 4840 | |
|
| 0 | 4841 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 4842 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 4843 | | var doVkTranspose = isVaapiDecoder && !string.IsNullOrEmpty(tranposeDir); |
| 0 | 4844 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || (isVaapiDecoder && doVkTranspose)); |
| 0 | 4845 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 4846 | | var swpInH = swapWAndH ? inW : inH; |
| | 4847 | |
|
| | 4848 | | /* Make main filters for video stream */ |
| 0 | 4849 | | var mainFilters = new List<string>(); |
| | 4850 | |
|
| 0 | 4851 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doVkTonemap)); |
| | 4852 | |
|
| 0 | 4853 | | if (isSwDecoder) |
| | 4854 | | { |
| | 4855 | | // INPUT sw surface(memory) |
| | 4856 | | // sw deint |
| 0 | 4857 | | if (doDeintH2645) |
| | 4858 | | { |
| 0 | 4859 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 4860 | | mainFilters.Add(swDeintFilter); |
| | 4861 | | } |
| | 4862 | |
|
| 0 | 4863 | | if (doVkTonemap || hasSubs) |
| | 4864 | | { |
| | 4865 | | // sw => hw |
| 0 | 4866 | | mainFilters.Add("hwupload=derive_device=vulkan"); |
| 0 | 4867 | | mainFilters.Add("format=vulkan"); |
| | 4868 | | } |
| | 4869 | | else |
| | 4870 | | { |
| | 4871 | | // sw scale |
| 0 | 4872 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, |
| 0 | 4873 | | mainFilters.Add(swScaleFilter); |
| 0 | 4874 | | mainFilters.Add("format=nv12"); |
| | 4875 | | } |
| | 4876 | | } |
| 0 | 4877 | | else if (isVaapiDecoder) |
| | 4878 | | { |
| | 4879 | | // INPUT vaapi surface(vram) |
| 0 | 4880 | | if (doVkTranspose || doVkTonemap || hasSubs) |
| | 4881 | | { |
| | 4882 | | // map from vaapi to vulkan/drm via interop (Polaris/gfx8+). |
| 0 | 4883 | | if (_mediaEncoder.EncoderVersion >= _minFFmpegAlteredVaVkInterop) |
| | 4884 | | { |
| 0 | 4885 | | if (doVkTranspose || !_mediaEncoder.IsVaapiDeviceSupportVulkanDrmModifier) |
| | 4886 | | { |
| | 4887 | | // disable the indirect va-drm-vk mapping since it's no longer reliable. |
| 0 | 4888 | | mainFilters.Add("hwmap=derive_device=drm"); |
| 0 | 4889 | | mainFilters.Add("format=drm_prime"); |
| 0 | 4890 | | mainFilters.Add("hwmap=derive_device=vulkan"); |
| 0 | 4891 | | mainFilters.Add("format=vulkan"); |
| | 4892 | |
|
| | 4893 | | // workaround for libplacebo using the imported vulkan frame on gfx8. |
| 0 | 4894 | | if (!_mediaEncoder.IsVaapiDeviceSupportVulkanDrmModifier) |
| | 4895 | | { |
| 0 | 4896 | | mainFilters.Add("scale_vulkan"); |
| | 4897 | | } |
| | 4898 | | } |
| 0 | 4899 | | else if (doVkTonemap || hasSubs) |
| | 4900 | | { |
| | 4901 | | // non ad-hoc libplacebo also accepts drm_prime direct input. |
| 0 | 4902 | | mainFilters.Add("hwmap=derive_device=drm"); |
| 0 | 4903 | | mainFilters.Add("format=drm_prime"); |
| | 4904 | | } |
| | 4905 | | } |
| | 4906 | | else // legacy va-vk mapping that works only in jellyfin-ffmpeg6 |
| | 4907 | | { |
| 0 | 4908 | | mainFilters.Add("hwmap=derive_device=vulkan"); |
| 0 | 4909 | | mainFilters.Add("format=vulkan"); |
| | 4910 | | } |
| | 4911 | | } |
| | 4912 | | else |
| | 4913 | | { |
| | 4914 | | // hw deint |
| 0 | 4915 | | if (doDeintH2645) |
| | 4916 | | { |
| 0 | 4917 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "vaapi"); |
| 0 | 4918 | | mainFilters.Add(deintFilter); |
| | 4919 | | } |
| | 4920 | |
|
| | 4921 | | // hw scale |
| 0 | 4922 | | var hwScaleFilter = GetHwScaleFilter("scale", "vaapi", "nv12", false, inW, inH, reqW, reqH, reqMaxW, |
| 0 | 4923 | | mainFilters.Add(hwScaleFilter); |
| | 4924 | | } |
| | 4925 | | } |
| | 4926 | |
|
| | 4927 | | // vk transpose |
| 0 | 4928 | | if (doVkTranspose) |
| | 4929 | | { |
| 0 | 4930 | | if (string.Equals(tranposeDir, "reversal", StringComparison.OrdinalIgnoreCase)) |
| | 4931 | | { |
| 0 | 4932 | | mainFilters.Add("flip_vulkan"); |
| | 4933 | | } |
| | 4934 | | else |
| | 4935 | | { |
| 0 | 4936 | | mainFilters.Add($"transpose_vulkan=dir={tranposeDir}"); |
| | 4937 | | } |
| | 4938 | | } |
| | 4939 | |
|
| | 4940 | | // vk libplacebo |
| 0 | 4941 | | if (doVkTonemap || hasSubs) |
| | 4942 | | { |
| 0 | 4943 | | var libplaceboFilter = GetLibplaceboFilter(options, "bgra", doVkTonemap, swpInW, swpInH, reqW, reqH, req |
| 0 | 4944 | | mainFilters.Add(libplaceboFilter); |
| 0 | 4945 | | mainFilters.Add("format=vulkan"); |
| | 4946 | | } |
| | 4947 | |
|
| 0 | 4948 | | if (doVkTonemap && !hasSubs) |
| | 4949 | | { |
| | 4950 | | // OUTPUT vaapi(nv12) surface(vram) |
| | 4951 | | // map from vulkan/drm to vaapi via interop (Polaris/gfx8+). |
| 0 | 4952 | | mainFilters.Add("hwmap=derive_device=vaapi"); |
| 0 | 4953 | | mainFilters.Add("format=vaapi"); |
| | 4954 | |
|
| | 4955 | | // clear the surf->meta_offset and output nv12 |
| 0 | 4956 | | mainFilters.Add("scale_vaapi=format=nv12"); |
| | 4957 | |
|
| | 4958 | | // hw deint |
| 0 | 4959 | | if (doDeintH2645) |
| | 4960 | | { |
| 0 | 4961 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "vaapi"); |
| 0 | 4962 | | mainFilters.Add(deintFilter); |
| | 4963 | | } |
| | 4964 | | } |
| | 4965 | |
|
| 0 | 4966 | | if (!hasSubs) |
| | 4967 | | { |
| | 4968 | | // OUTPUT nv12 surface(memory) |
| 0 | 4969 | | if (isSwEncoder && (doVkTonemap || isVaapiDecoder)) |
| | 4970 | | { |
| 0 | 4971 | | mainFilters.Add("hwdownload"); |
| 0 | 4972 | | mainFilters.Add("format=nv12"); |
| | 4973 | | } |
| | 4974 | |
|
| 0 | 4975 | | if (isSwDecoder && isVaapiEncoder && !doVkTonemap) |
| | 4976 | | { |
| 0 | 4977 | | mainFilters.Add("hwupload_vaapi"); |
| | 4978 | | } |
| | 4979 | | } |
| | 4980 | |
|
| | 4981 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 4982 | | var subFilters = new List<string>(); |
| 0 | 4983 | | var overlayFilters = new List<string>(); |
| 0 | 4984 | | if (hasSubs) |
| | 4985 | | { |
| 0 | 4986 | | if (hasGraphicalSubs) |
| | 4987 | | { |
| 0 | 4988 | | var subW = state.SubtitleStream?.Width; |
| 0 | 4989 | | var subH = state.SubtitleStream?.Height; |
| 0 | 4990 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 4991 | | subFilters.Add(subPreProcFilters); |
| 0 | 4992 | | subFilters.Add("format=bgra"); |
| | 4993 | | } |
| 0 | 4994 | | else if (hasTextSubs) |
| | 4995 | | { |
| 0 | 4996 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 4997 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 4998 | |
|
| 0 | 4999 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH, subFrame |
| 0 | 5000 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 5001 | | subFilters.Add(alphaSrcFilter); |
| 0 | 5002 | | subFilters.Add("format=bgra"); |
| 0 | 5003 | | subFilters.Add(subTextSubtitlesFilter); |
| | 5004 | | } |
| | 5005 | |
|
| 0 | 5006 | | subFilters.Add("hwupload=derive_device=vulkan"); |
| 0 | 5007 | | subFilters.Add("format=vulkan"); |
| | 5008 | |
|
| 0 | 5009 | | overlayFilters.Add("overlay_vulkan=eof_action=pass:repeatlast=0"); |
| | 5010 | |
|
| 0 | 5011 | | if (isSwEncoder) |
| | 5012 | | { |
| | 5013 | | // OUTPUT nv12 surface(memory) |
| 0 | 5014 | | overlayFilters.Add("scale_vulkan=format=nv12"); |
| 0 | 5015 | | overlayFilters.Add("hwdownload"); |
| 0 | 5016 | | overlayFilters.Add("format=nv12"); |
| | 5017 | | } |
| 0 | 5018 | | else if (isVaapiEncoder) |
| | 5019 | | { |
| | 5020 | | // OUTPUT vaapi(nv12) surface(vram) |
| | 5021 | | // map from vulkan/drm to vaapi via interop (Polaris/gfx8+). |
| 0 | 5022 | | overlayFilters.Add("hwmap=derive_device=vaapi"); |
| 0 | 5023 | | overlayFilters.Add("format=vaapi"); |
| | 5024 | |
|
| | 5025 | | // clear the surf->meta_offset and output nv12 |
| 0 | 5026 | | overlayFilters.Add("scale_vaapi=format=nv12"); |
| | 5027 | |
|
| | 5028 | | // hw deint |
| 0 | 5029 | | if (doDeintH2645) |
| | 5030 | | { |
| 0 | 5031 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "vaapi"); |
| 0 | 5032 | | overlayFilters.Add(deintFilter); |
| | 5033 | | } |
| | 5034 | | } |
| | 5035 | | } |
| | 5036 | |
|
| 0 | 5037 | | return (mainFilters, subFilters, overlayFilters); |
| | 5038 | | } |
| | 5039 | |
|
| | 5040 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetVaapiLimitedVidFilter |
| | 5041 | | EncodingJobInfo state, |
| | 5042 | | EncodingOptions options, |
| | 5043 | | string vidDecoder, |
| | 5044 | | string vidEncoder) |
| | 5045 | | { |
| 0 | 5046 | | var inW = state.VideoStream?.Width; |
| 0 | 5047 | | var inH = state.VideoStream?.Height; |
| 0 | 5048 | | var reqW = state.BaseRequest.Width; |
| 0 | 5049 | | var reqH = state.BaseRequest.Height; |
| 0 | 5050 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 5051 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 5052 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 5053 | |
|
| 0 | 5054 | | var isVaapiDecoder = vidDecoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 5055 | | var isVaapiEncoder = vidEncoder.Contains("vaapi", StringComparison.OrdinalIgnoreCase); |
| 0 | 5056 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 5057 | | var isSwEncoder = !isVaapiEncoder; |
| 0 | 5058 | | var isVaInVaOut = isVaapiDecoder && isVaapiEncoder; |
| 0 | 5059 | | var isi965Driver = _mediaEncoder.IsVaapiDeviceInteli965; |
| 0 | 5060 | | var isAmdDriver = _mediaEncoder.IsVaapiDeviceAmd; |
| | 5061 | |
|
| 0 | 5062 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 5063 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 5064 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 5065 | | var doOclTonemap = IsHwTonemapAvailable(state, options); |
| | 5066 | |
|
| 0 | 5067 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 5068 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 5069 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| | 5070 | |
|
| 0 | 5071 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 5072 | | var swapWAndH = Math.Abs(rotation) == 90 && isSwDecoder; |
| 0 | 5073 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 5074 | | var swpInH = swapWAndH ? inW : inH; |
| | 5075 | |
|
| | 5076 | | /* Make main filters for video stream */ |
| 0 | 5077 | | var mainFilters = new List<string>(); |
| | 5078 | |
|
| 0 | 5079 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doOclTonemap)); |
| | 5080 | |
|
| 0 | 5081 | | var outFormat = string.Empty; |
| 0 | 5082 | | if (isSwDecoder) |
| | 5083 | | { |
| | 5084 | | // INPUT sw surface(memory) |
| | 5085 | | // sw deint |
| 0 | 5086 | | if (doDeintH2645) |
| | 5087 | | { |
| 0 | 5088 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 5089 | | mainFilters.Add(swDeintFilter); |
| | 5090 | | } |
| | 5091 | |
|
| 0 | 5092 | | outFormat = doOclTonemap ? "yuv420p10le" : "nv12"; |
| 0 | 5093 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| | 5094 | | // sw scale |
| 0 | 5095 | | mainFilters.Add(swScaleFilter); |
| 0 | 5096 | | mainFilters.Add("format=" + outFormat); |
| | 5097 | |
|
| | 5098 | | // keep video at memory except ocl tonemap, |
| | 5099 | | // since the overhead caused by hwupload >>> using sw filter. |
| | 5100 | | // sw => hw |
| 0 | 5101 | | if (doOclTonemap) |
| | 5102 | | { |
| 0 | 5103 | | mainFilters.Add("hwupload=derive_device=opencl"); |
| | 5104 | | } |
| | 5105 | | } |
| 0 | 5106 | | else if (isVaapiDecoder) |
| | 5107 | | { |
| | 5108 | | // INPUT vaapi surface(vram) |
| | 5109 | | // hw deint |
| 0 | 5110 | | if (doDeintH2645) |
| | 5111 | | { |
| 0 | 5112 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "vaapi"); |
| 0 | 5113 | | mainFilters.Add(deintFilter); |
| | 5114 | | } |
| | 5115 | |
|
| 0 | 5116 | | outFormat = doOclTonemap ? string.Empty : "nv12"; |
| 0 | 5117 | | var hwScaleFilter = GetHwScaleFilter("scale", "vaapi", outFormat, false, inW, inH, reqW, reqH, reqMaxW, |
| | 5118 | |
|
| | 5119 | | // allocate extra pool sizes for vaapi vpp |
| 0 | 5120 | | if (!string.IsNullOrEmpty(hwScaleFilter)) |
| | 5121 | | { |
| 0 | 5122 | | hwScaleFilter += ":extra_hw_frames=24"; |
| | 5123 | | } |
| | 5124 | |
|
| | 5125 | | // hw scale |
| 0 | 5126 | | mainFilters.Add(hwScaleFilter); |
| | 5127 | | } |
| | 5128 | |
|
| 0 | 5129 | | if (doOclTonemap && isVaapiDecoder) |
| | 5130 | | { |
| 0 | 5131 | | if (isi965Driver) |
| | 5132 | | { |
| | 5133 | | // map from vaapi to opencl via vaapi-opencl interop(Intel only). |
| 0 | 5134 | | mainFilters.Add("hwmap=derive_device=opencl"); |
| | 5135 | | } |
| | 5136 | | else |
| | 5137 | | { |
| 0 | 5138 | | mainFilters.Add("hwdownload"); |
| 0 | 5139 | | mainFilters.Add("format=p010le"); |
| 0 | 5140 | | mainFilters.Add("hwupload=derive_device=opencl"); |
| | 5141 | | } |
| | 5142 | | } |
| | 5143 | |
|
| | 5144 | | // ocl tonemap |
| 0 | 5145 | | if (doOclTonemap) |
| | 5146 | | { |
| 0 | 5147 | | var tonemapFilter = GetHwTonemapFilter(options, "opencl", "nv12"); |
| 0 | 5148 | | mainFilters.Add(tonemapFilter); |
| | 5149 | | } |
| | 5150 | |
|
| 0 | 5151 | | if (doOclTonemap && isVaInVaOut) |
| | 5152 | | { |
| 0 | 5153 | | if (isi965Driver) |
| | 5154 | | { |
| | 5155 | | // OUTPUT vaapi(nv12) surface(vram) |
| | 5156 | | // reverse-mapping via vaapi-opencl interop. |
| 0 | 5157 | | mainFilters.Add("hwmap=derive_device=vaapi:reverse=1"); |
| 0 | 5158 | | mainFilters.Add("format=vaapi"); |
| | 5159 | | } |
| | 5160 | | } |
| | 5161 | |
|
| 0 | 5162 | | var memoryOutput = false; |
| 0 | 5163 | | var isUploadForOclTonemap = doOclTonemap && (isSwDecoder || (isVaapiDecoder && !isi965Driver)); |
| 0 | 5164 | | var isHwmapNotUsable = hasGraphicalSubs || isUploadForOclTonemap; |
| 0 | 5165 | | var isHwmapForSubs = hasSubs && isVaapiDecoder; |
| 0 | 5166 | | var isHwUnmapForTextSubs = hasTextSubs && isVaInVaOut && !isUploadForOclTonemap; |
| 0 | 5167 | | if ((isVaapiDecoder && isSwEncoder) || isUploadForOclTonemap || isHwmapForSubs) |
| | 5168 | | { |
| 0 | 5169 | | memoryOutput = true; |
| | 5170 | |
|
| | 5171 | | // OUTPUT nv12 surface(memory) |
| | 5172 | | // prefer hwmap to hwdownload on opencl/vaapi. |
| 0 | 5173 | | mainFilters.Add(isHwmapNotUsable ? "hwdownload" : "hwmap"); |
| 0 | 5174 | | mainFilters.Add("format=nv12"); |
| | 5175 | | } |
| | 5176 | |
|
| | 5177 | | // OUTPUT nv12 surface(memory) |
| 0 | 5178 | | if (isSwDecoder && isVaapiEncoder) |
| | 5179 | | { |
| 0 | 5180 | | memoryOutput = true; |
| | 5181 | | } |
| | 5182 | |
|
| 0 | 5183 | | if (memoryOutput) |
| | 5184 | | { |
| | 5185 | | // text subtitles |
| 0 | 5186 | | if (hasTextSubs) |
| | 5187 | | { |
| 0 | 5188 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 5189 | | mainFilters.Add(textSubtitlesFilter); |
| | 5190 | | } |
| | 5191 | | } |
| | 5192 | |
|
| 0 | 5193 | | if (isHwUnmapForTextSubs) |
| | 5194 | | { |
| 0 | 5195 | | mainFilters.Add("hwmap"); |
| 0 | 5196 | | mainFilters.Add("format=vaapi"); |
| | 5197 | | } |
| 0 | 5198 | | else if (memoryOutput && isVaapiEncoder) |
| | 5199 | | { |
| 0 | 5200 | | if (!hasGraphicalSubs) |
| | 5201 | | { |
| 0 | 5202 | | mainFilters.Add("hwupload_vaapi"); |
| | 5203 | | } |
| | 5204 | | } |
| | 5205 | |
|
| | 5206 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 5207 | | var subFilters = new List<string>(); |
| 0 | 5208 | | var overlayFilters = new List<string>(); |
| 0 | 5209 | | if (memoryOutput) |
| | 5210 | | { |
| 0 | 5211 | | if (hasGraphicalSubs) |
| | 5212 | | { |
| 0 | 5213 | | var subW = state.SubtitleStream?.Width; |
| 0 | 5214 | | var subH = state.SubtitleStream?.Height; |
| 0 | 5215 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 5216 | | subFilters.Add(subPreProcFilters); |
| 0 | 5217 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 5218 | |
|
| 0 | 5219 | | if (isVaapiEncoder) |
| | 5220 | | { |
| 0 | 5221 | | overlayFilters.Add("hwupload_vaapi"); |
| | 5222 | | } |
| | 5223 | | } |
| | 5224 | | } |
| | 5225 | |
|
| 0 | 5226 | | return (mainFilters, subFilters, overlayFilters); |
| | 5227 | | } |
| | 5228 | |
|
| | 5229 | | /// <summary> |
| | 5230 | | /// Gets the parameter of Apple VideoToolBox filter chain. |
| | 5231 | | /// </summary> |
| | 5232 | | /// <param name="state">Encoding state.</param> |
| | 5233 | | /// <param name="options">Encoding options.</param> |
| | 5234 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 5235 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 5236 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetAppleVidFilterChain( |
| | 5237 | | EncodingJobInfo state, |
| | 5238 | | EncodingOptions options, |
| | 5239 | | string vidEncoder) |
| | 5240 | | { |
| 0 | 5241 | | if (options.HardwareAccelerationType != HardwareAccelerationType.videotoolbox) |
| | 5242 | | { |
| 0 | 5243 | | return (null, null, null); |
| | 5244 | | } |
| | 5245 | |
|
| | 5246 | | // ReSharper disable once InconsistentNaming |
| 0 | 5247 | | var isMacOS = OperatingSystem.IsMacOS(); |
| 0 | 5248 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 5249 | | var isVtDecoder = vidDecoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| 0 | 5250 | | var isVtEncoder = vidEncoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| 0 | 5251 | | var isVtFullSupported = isMacOS && IsVideoToolboxFullSupported(); |
| | 5252 | |
|
| | 5253 | | // legacy videotoolbox pipeline (disable hw filters) |
| 0 | 5254 | | if (!(isVtEncoder || isVtDecoder) |
| 0 | 5255 | | || !isVtFullSupported |
| 0 | 5256 | | || !_mediaEncoder.SupportsFilter("alphasrc")) |
| | 5257 | | { |
| 0 | 5258 | | return GetSwVidFilterChain(state, options, vidEncoder); |
| | 5259 | | } |
| | 5260 | |
|
| | 5261 | | // preferred videotoolbox + metal filters pipeline |
| 0 | 5262 | | return GetAppleVidFiltersPreferred(state, options, vidDecoder, vidEncoder); |
| | 5263 | | } |
| | 5264 | |
|
| | 5265 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetAppleVidFiltersPrefer |
| | 5266 | | EncodingJobInfo state, |
| | 5267 | | EncodingOptions options, |
| | 5268 | | string vidDecoder, |
| | 5269 | | string vidEncoder) |
| | 5270 | | { |
| 0 | 5271 | | var isVtEncoder = vidEncoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| 0 | 5272 | | var isVtDecoder = vidDecoder.Contains("videotoolbox", StringComparison.OrdinalIgnoreCase); |
| | 5273 | |
|
| 0 | 5274 | | var inW = state.VideoStream?.Width; |
| 0 | 5275 | | var inH = state.VideoStream?.Height; |
| 0 | 5276 | | var reqW = state.BaseRequest.Width; |
| 0 | 5277 | | var reqH = state.BaseRequest.Height; |
| 0 | 5278 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 5279 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 5280 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 5281 | |
|
| 0 | 5282 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 5283 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 5284 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 5285 | | var doVtTonemap = IsVideoToolboxTonemapAvailable(state, options); |
| 0 | 5286 | | var doMetalTonemap = !doVtTonemap && IsHwTonemapAvailable(state, options); |
| 0 | 5287 | | var usingHwSurface = isVtDecoder && (_mediaEncoder.EncoderVersion >= _minFFmpegWorkingVtHwSurface); |
| | 5288 | |
|
| 0 | 5289 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 5290 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 5291 | | var doVtTranspose = !string.IsNullOrEmpty(tranposeDir) && _mediaEncoder.SupportsFilter("transpose_vt"); |
| 0 | 5292 | | var swapWAndH = Math.Abs(rotation) == 90 && doVtTranspose; |
| 0 | 5293 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 5294 | | var swpInH = swapWAndH ? inW : inH; |
| | 5295 | |
|
| 0 | 5296 | | var scaleFormat = string.Empty; |
| | 5297 | | // Use P010 for Metal tone mapping, otherwise force an 8bit output. |
| 0 | 5298 | | if (!string.Equals(state.VideoStream.PixelFormat, "yuv420p", StringComparison.OrdinalIgnoreCase)) |
| | 5299 | | { |
| 0 | 5300 | | if (doMetalTonemap) |
| | 5301 | | { |
| 0 | 5302 | | if (!string.Equals(state.VideoStream.PixelFormat, "yuv420p10le", StringComparison.OrdinalIgnoreCase) |
| | 5303 | | { |
| 0 | 5304 | | scaleFormat = "p010le"; |
| | 5305 | | } |
| | 5306 | | } |
| | 5307 | | else |
| | 5308 | | { |
| 0 | 5309 | | scaleFormat = "nv12"; |
| | 5310 | | } |
| | 5311 | | } |
| | 5312 | |
|
| 0 | 5313 | | var hwScaleFilter = GetHwScaleFilter("scale", "vt", scaleFormat, false, swpInW, swpInH, reqW, reqH, reqMaxW, |
| | 5314 | |
|
| 0 | 5315 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 5316 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 5317 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 5318 | | var hasAssSubs = hasSubs |
| 0 | 5319 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 5320 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| | 5321 | |
|
| | 5322 | | /* Make main filters for video stream */ |
| 0 | 5323 | | var mainFilters = new List<string>(); |
| | 5324 | |
|
| | 5325 | | // hw deint |
| 0 | 5326 | | if (doDeintH2645) |
| | 5327 | | { |
| 0 | 5328 | | var deintFilter = GetHwDeinterlaceFilter(state, options, "videotoolbox"); |
| 0 | 5329 | | mainFilters.Add(deintFilter); |
| | 5330 | | } |
| | 5331 | |
|
| | 5332 | | // hw transpose |
| 0 | 5333 | | if (doVtTranspose) |
| | 5334 | | { |
| 0 | 5335 | | mainFilters.Add($"transpose_vt=dir={tranposeDir}"); |
| | 5336 | | } |
| | 5337 | |
|
| 0 | 5338 | | if (doVtTonemap) |
| | 5339 | | { |
| | 5340 | | const string VtTonemapArgs = "color_matrix=bt709:color_primaries=bt709:color_transfer=bt709"; |
| | 5341 | |
|
| | 5342 | | // scale_vt can handle scaling & tonemapping in one shot, just like vpp_qsv. |
| 0 | 5343 | | hwScaleFilter = string.IsNullOrEmpty(hwScaleFilter) |
| 0 | 5344 | | ? "scale_vt=" + VtTonemapArgs |
| 0 | 5345 | | : hwScaleFilter + ":" + VtTonemapArgs; |
| | 5346 | | } |
| | 5347 | |
|
| | 5348 | | // hw scale & vt tonemap |
| 0 | 5349 | | mainFilters.Add(hwScaleFilter); |
| | 5350 | |
|
| | 5351 | | // Metal tonemap |
| 0 | 5352 | | if (doMetalTonemap) |
| | 5353 | | { |
| 0 | 5354 | | var tonemapFilter = GetHwTonemapFilter(options, "videotoolbox", "nv12"); |
| 0 | 5355 | | mainFilters.Add(tonemapFilter); |
| | 5356 | | } |
| | 5357 | |
|
| | 5358 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 5359 | | var subFilters = new List<string>(); |
| 0 | 5360 | | var overlayFilters = new List<string>(); |
| | 5361 | |
|
| 0 | 5362 | | if (hasSubs) |
| | 5363 | | { |
| 0 | 5364 | | if (hasGraphicalSubs) |
| | 5365 | | { |
| 0 | 5366 | | var subW = state.SubtitleStream?.Width; |
| 0 | 5367 | | var subH = state.SubtitleStream?.Height; |
| 0 | 5368 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 5369 | | subFilters.Add(subPreProcFilters); |
| 0 | 5370 | | subFilters.Add("format=bgra"); |
| | 5371 | | } |
| 0 | 5372 | | else if (hasTextSubs) |
| | 5373 | | { |
| 0 | 5374 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 5375 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 5376 | |
|
| 0 | 5377 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH, subFrame |
| 0 | 5378 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 5379 | | subFilters.Add(alphaSrcFilter); |
| 0 | 5380 | | subFilters.Add("format=bgra"); |
| 0 | 5381 | | subFilters.Add(subTextSubtitlesFilter); |
| | 5382 | | } |
| | 5383 | |
|
| 0 | 5384 | | subFilters.Add("hwupload"); |
| 0 | 5385 | | overlayFilters.Add("overlay_videotoolbox=eof_action=pass:repeatlast=0"); |
| | 5386 | | } |
| | 5387 | |
|
| 0 | 5388 | | if (usingHwSurface) |
| | 5389 | | { |
| 0 | 5390 | | if (!isVtEncoder) |
| | 5391 | | { |
| 0 | 5392 | | mainFilters.Add("hwdownload"); |
| 0 | 5393 | | mainFilters.Add("format=nv12"); |
| | 5394 | | } |
| | 5395 | |
|
| 0 | 5396 | | return (mainFilters, subFilters, overlayFilters); |
| | 5397 | | } |
| | 5398 | |
|
| | 5399 | | // For old jellyfin-ffmpeg that has broken hwsurface, add a hwupload |
| 0 | 5400 | | var needFiltering = mainFilters.Any(f => !string.IsNullOrEmpty(f)) || |
| 0 | 5401 | | subFilters.Any(f => !string.IsNullOrEmpty(f)) || |
| 0 | 5402 | | overlayFilters.Any(f => !string.IsNullOrEmpty(f)); |
| 0 | 5403 | | if (needFiltering) |
| | 5404 | | { |
| | 5405 | | // INPUT videotoolbox/memory surface(vram/uma) |
| | 5406 | | // this will pass-through automatically if in/out format matches. |
| 0 | 5407 | | mainFilters.Insert(0, "hwupload"); |
| 0 | 5408 | | mainFilters.Insert(0, "format=nv12|p010le|videotoolbox_vld"); |
| | 5409 | |
|
| 0 | 5410 | | if (!isVtEncoder) |
| | 5411 | | { |
| 0 | 5412 | | mainFilters.Add("hwdownload"); |
| 0 | 5413 | | mainFilters.Add("format=nv12"); |
| | 5414 | | } |
| | 5415 | | } |
| | 5416 | |
|
| 0 | 5417 | | return (mainFilters, subFilters, overlayFilters); |
| | 5418 | | } |
| | 5419 | |
|
| | 5420 | | /// <summary> |
| | 5421 | | /// Gets the parameter of Rockchip RKMPP/RKRGA filter chain. |
| | 5422 | | /// </summary> |
| | 5423 | | /// <param name="state">Encoding state.</param> |
| | 5424 | | /// <param name="options">Encoding options.</param> |
| | 5425 | | /// <param name="vidEncoder">Video encoder to use.</param> |
| | 5426 | | /// <returns>The tuple contains three lists: main, sub and overlay filters.</returns> |
| | 5427 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetRkmppVidFilterChain( |
| | 5428 | | EncodingJobInfo state, |
| | 5429 | | EncodingOptions options, |
| | 5430 | | string vidEncoder) |
| | 5431 | | { |
| 0 | 5432 | | if (options.HardwareAccelerationType != HardwareAccelerationType.rkmpp) |
| | 5433 | | { |
| 0 | 5434 | | return (null, null, null); |
| | 5435 | | } |
| | 5436 | |
|
| 0 | 5437 | | var isLinux = OperatingSystem.IsLinux(); |
| 0 | 5438 | | var vidDecoder = GetHardwareVideoDecoder(state, options) ?? string.Empty; |
| 0 | 5439 | | var isSwDecoder = string.IsNullOrEmpty(vidDecoder); |
| 0 | 5440 | | var isSwEncoder = !vidEncoder.Contains("rkmpp", StringComparison.OrdinalIgnoreCase); |
| 0 | 5441 | | var isRkmppOclSupported = isLinux && IsRkmppFullSupported() && IsOpenclFullSupported(); |
| | 5442 | |
|
| 0 | 5443 | | if ((isSwDecoder && isSwEncoder) |
| 0 | 5444 | | || !isRkmppOclSupported |
| 0 | 5445 | | || !_mediaEncoder.SupportsFilter("alphasrc")) |
| | 5446 | | { |
| 0 | 5447 | | return GetSwVidFilterChain(state, options, vidEncoder); |
| | 5448 | | } |
| | 5449 | |
|
| | 5450 | | // prefered rkmpp + rkrga + opencl filters pipeline |
| 0 | 5451 | | if (isRkmppOclSupported) |
| | 5452 | | { |
| 0 | 5453 | | return GetRkmppVidFiltersPrefered(state, options, vidDecoder, vidEncoder); |
| | 5454 | | } |
| | 5455 | |
|
| 0 | 5456 | | return (null, null, null); |
| | 5457 | | } |
| | 5458 | |
|
| | 5459 | | public (List<string> MainFilters, List<string> SubFilters, List<string> OverlayFilters) GetRkmppVidFiltersPrefer |
| | 5460 | | EncodingJobInfo state, |
| | 5461 | | EncodingOptions options, |
| | 5462 | | string vidDecoder, |
| | 5463 | | string vidEncoder) |
| | 5464 | | { |
| 0 | 5465 | | var inW = state.VideoStream?.Width; |
| 0 | 5466 | | var inH = state.VideoStream?.Height; |
| 0 | 5467 | | var reqW = state.BaseRequest.Width; |
| 0 | 5468 | | var reqH = state.BaseRequest.Height; |
| 0 | 5469 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 5470 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| 0 | 5471 | | var threeDFormat = state.MediaSource.Video3DFormat; |
| | 5472 | |
|
| 0 | 5473 | | var isRkmppDecoder = vidDecoder.Contains("rkmpp", StringComparison.OrdinalIgnoreCase); |
| 0 | 5474 | | var isRkmppEncoder = vidEncoder.Contains("rkmpp", StringComparison.OrdinalIgnoreCase); |
| 0 | 5475 | | var isSwDecoder = !isRkmppDecoder; |
| 0 | 5476 | | var isSwEncoder = !isRkmppEncoder; |
| 0 | 5477 | | var isDrmInDrmOut = isRkmppDecoder && isRkmppEncoder; |
| 0 | 5478 | | var isEncoderSupportAfbc = isRkmppEncoder |
| 0 | 5479 | | && (vidEncoder.Contains("h264", StringComparison.OrdinalIgnoreCase) |
| 0 | 5480 | | || vidEncoder.Contains("hevc", StringComparison.OrdinalIgnoreCase)); |
| | 5481 | |
|
| 0 | 5482 | | var doDeintH264 = state.DeInterlace("h264", true) || state.DeInterlace("avc", true); |
| 0 | 5483 | | var doDeintHevc = state.DeInterlace("h265", true) || state.DeInterlace("hevc", true); |
| 0 | 5484 | | var doDeintH2645 = doDeintH264 || doDeintHevc; |
| 0 | 5485 | | var doOclTonemap = IsHwTonemapAvailable(state, options); |
| | 5486 | |
|
| 0 | 5487 | | var hasSubs = state.SubtitleStream != null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Encode; |
| 0 | 5488 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 5489 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 5490 | | var hasAssSubs = hasSubs |
| 0 | 5491 | | && (string.Equals(state.SubtitleStream.Codec, "ass", StringComparison.OrdinalIgnoreCase) |
| 0 | 5492 | | || string.Equals(state.SubtitleStream.Codec, "ssa", StringComparison.OrdinalIgnoreCase)); |
| 0 | 5493 | | var subW = state.SubtitleStream?.Width; |
| 0 | 5494 | | var subH = state.SubtitleStream?.Height; |
| | 5495 | |
|
| 0 | 5496 | | var rotation = state.VideoStream?.Rotation ?? 0; |
| 0 | 5497 | | var tranposeDir = rotation == 0 ? string.Empty : GetVideoTransposeDirection(state); |
| 0 | 5498 | | var doRkVppTranspose = !string.IsNullOrEmpty(tranposeDir); |
| 0 | 5499 | | var swapWAndH = Math.Abs(rotation) == 90 && (isSwDecoder || (isRkmppDecoder && doRkVppTranspose)); |
| 0 | 5500 | | var swpInW = swapWAndH ? inH : inW; |
| 0 | 5501 | | var swpInH = swapWAndH ? inW : inH; |
| | 5502 | |
|
| | 5503 | | /* Make main filters for video stream */ |
| 0 | 5504 | | var mainFilters = new List<string>(); |
| | 5505 | |
|
| 0 | 5506 | | mainFilters.Add(GetOverwriteColorPropertiesParam(state, doOclTonemap)); |
| | 5507 | |
|
| 0 | 5508 | | if (isSwDecoder) |
| | 5509 | | { |
| | 5510 | | // INPUT sw surface(memory) |
| | 5511 | | // sw deint |
| 0 | 5512 | | if (doDeintH2645) |
| | 5513 | | { |
| 0 | 5514 | | var swDeintFilter = GetSwDeinterlaceFilter(state, options); |
| 0 | 5515 | | mainFilters.Add(swDeintFilter); |
| | 5516 | | } |
| | 5517 | |
|
| 0 | 5518 | | var outFormat = doOclTonemap ? "yuv420p10le" : (hasGraphicalSubs ? "yuv420p" : "nv12"); |
| 0 | 5519 | | var swScaleFilter = GetSwScaleFilter(state, options, vidEncoder, swpInW, swpInH, threeDFormat, reqW, req |
| 0 | 5520 | | if (!string.IsNullOrEmpty(swScaleFilter)) |
| | 5521 | | { |
| 0 | 5522 | | swScaleFilter += ":flags=fast_bilinear"; |
| | 5523 | | } |
| | 5524 | |
|
| | 5525 | | // sw scale |
| 0 | 5526 | | mainFilters.Add(swScaleFilter); |
| 0 | 5527 | | mainFilters.Add($"format={outFormat}"); |
| | 5528 | |
|
| | 5529 | | // keep video at memory except ocl tonemap, |
| | 5530 | | // since the overhead caused by hwupload >>> using sw filter. |
| | 5531 | | // sw => hw |
| 0 | 5532 | | if (doOclTonemap) |
| | 5533 | | { |
| 0 | 5534 | | mainFilters.Add("hwupload=derive_device=opencl"); |
| | 5535 | | } |
| | 5536 | | } |
| 0 | 5537 | | else if (isRkmppDecoder) |
| | 5538 | | { |
| | 5539 | | // INPUT rkmpp/drm surface(gem/dma-heap) |
| | 5540 | |
|
| 0 | 5541 | | var isFullAfbcPipeline = isEncoderSupportAfbc && isDrmInDrmOut && !doOclTonemap; |
| 0 | 5542 | | var swapOutputWandH = doRkVppTranspose && swapWAndH; |
| 0 | 5543 | | var outFormat = doOclTonemap ? "p010" : "nv12"; |
| 0 | 5544 | | var hwScalePrefix = doRkVppTranspose ? "vpp" : "scale"; |
| 0 | 5545 | | var hwScaleFilter = GetHwScaleFilter(hwScalePrefix, "rkrga", outFormat, swapOutputWandH, swpInW, swpInH, |
| 0 | 5546 | | var hwScaleFilter2 = GetHwScaleFilter(hwScalePrefix, "rkrga", string.Empty, swapOutputWandH, swpInW, swp |
| | 5547 | |
|
| 0 | 5548 | | if (!hasSubs |
| 0 | 5549 | | || doRkVppTranspose |
| 0 | 5550 | | || !isFullAfbcPipeline |
| 0 | 5551 | | || !string.IsNullOrEmpty(hwScaleFilter2)) |
| | 5552 | | { |
| 0 | 5553 | | if (!string.IsNullOrEmpty(hwScaleFilter) && doRkVppTranspose) |
| | 5554 | | { |
| 0 | 5555 | | hwScaleFilter += $":transpose={tranposeDir}"; |
| | 5556 | | } |
| | 5557 | |
|
| | 5558 | | // try enabling AFBC to save DDR bandwidth |
| 0 | 5559 | | if (!string.IsNullOrEmpty(hwScaleFilter) && isFullAfbcPipeline) |
| | 5560 | | { |
| 0 | 5561 | | hwScaleFilter += ":afbc=1"; |
| | 5562 | | } |
| | 5563 | |
|
| | 5564 | | // hw transpose & scale |
| 0 | 5565 | | mainFilters.Add(hwScaleFilter); |
| | 5566 | | } |
| | 5567 | | } |
| | 5568 | |
|
| 0 | 5569 | | if (doOclTonemap && isRkmppDecoder) |
| | 5570 | | { |
| | 5571 | | // map from rkmpp/drm to opencl via drm-opencl interop. |
| 0 | 5572 | | mainFilters.Add("hwmap=derive_device=opencl:mode=read"); |
| | 5573 | | } |
| | 5574 | |
|
| | 5575 | | // ocl tonemap |
| 0 | 5576 | | if (doOclTonemap) |
| | 5577 | | { |
| 0 | 5578 | | var tonemapFilter = GetHwTonemapFilter(options, "opencl", "nv12"); |
| 0 | 5579 | | mainFilters.Add(tonemapFilter); |
| | 5580 | | } |
| | 5581 | |
|
| 0 | 5582 | | var memoryOutput = false; |
| 0 | 5583 | | var isUploadForOclTonemap = isSwDecoder && doOclTonemap; |
| 0 | 5584 | | if ((isRkmppDecoder && isSwEncoder) || isUploadForOclTonemap) |
| | 5585 | | { |
| 0 | 5586 | | memoryOutput = true; |
| | 5587 | |
|
| | 5588 | | // OUTPUT nv12 surface(memory) |
| 0 | 5589 | | mainFilters.Add("hwdownload"); |
| 0 | 5590 | | mainFilters.Add("format=nv12"); |
| | 5591 | | } |
| | 5592 | |
|
| | 5593 | | // OUTPUT nv12 surface(memory) |
| 0 | 5594 | | if (isSwDecoder && isRkmppEncoder) |
| | 5595 | | { |
| 0 | 5596 | | memoryOutput = true; |
| | 5597 | | } |
| | 5598 | |
|
| 0 | 5599 | | if (memoryOutput) |
| | 5600 | | { |
| | 5601 | | // text subtitles |
| 0 | 5602 | | if (hasTextSubs) |
| | 5603 | | { |
| 0 | 5604 | | var textSubtitlesFilter = GetTextSubtitlesFilter(state, false, false); |
| 0 | 5605 | | mainFilters.Add(textSubtitlesFilter); |
| | 5606 | | } |
| | 5607 | | } |
| | 5608 | |
|
| 0 | 5609 | | if (isDrmInDrmOut) |
| | 5610 | | { |
| 0 | 5611 | | if (doOclTonemap) |
| | 5612 | | { |
| | 5613 | | // OUTPUT drm(nv12) surface(gem/dma-heap) |
| | 5614 | | // reverse-mapping via drm-opencl interop. |
| 0 | 5615 | | mainFilters.Add("hwmap=derive_device=rkmpp:mode=write:reverse=1"); |
| 0 | 5616 | | mainFilters.Add("format=drm_prime"); |
| | 5617 | | } |
| | 5618 | | } |
| | 5619 | |
|
| | 5620 | | /* Make sub and overlay filters for subtitle stream */ |
| 0 | 5621 | | var subFilters = new List<string>(); |
| 0 | 5622 | | var overlayFilters = new List<string>(); |
| 0 | 5623 | | if (isDrmInDrmOut) |
| | 5624 | | { |
| 0 | 5625 | | if (hasSubs) |
| | 5626 | | { |
| 0 | 5627 | | if (hasGraphicalSubs) |
| | 5628 | | { |
| 0 | 5629 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, |
| 0 | 5630 | | subFilters.Add(subPreProcFilters); |
| 0 | 5631 | | subFilters.Add("format=bgra"); |
| | 5632 | | } |
| 0 | 5633 | | else if (hasTextSubs) |
| | 5634 | | { |
| 0 | 5635 | | var framerate = state.VideoStream?.RealFrameRate; |
| 0 | 5636 | | var subFramerate = hasAssSubs ? Math.Min(framerate ?? 25, 60) : 10; |
| | 5637 | |
|
| | 5638 | | // alphasrc=s=1280x720:r=10:start=0,format=bgra,subtitles,hwupload |
| 0 | 5639 | | var alphaSrcFilter = GetAlphaSrcFilter(state, swpInW, swpInH, reqW, reqH, reqMaxW, reqMaxH, subF |
| 0 | 5640 | | var subTextSubtitlesFilter = GetTextSubtitlesFilter(state, true, true); |
| 0 | 5641 | | subFilters.Add(alphaSrcFilter); |
| 0 | 5642 | | subFilters.Add("format=bgra"); |
| 0 | 5643 | | subFilters.Add(subTextSubtitlesFilter); |
| | 5644 | | } |
| | 5645 | |
|
| 0 | 5646 | | subFilters.Add("hwupload=derive_device=rkmpp"); |
| | 5647 | |
|
| | 5648 | | // try enabling AFBC to save DDR bandwidth |
| 0 | 5649 | | var hwOverlayFilter = "overlay_rkrga=eof_action=pass:repeatlast=0:format=nv12"; |
| 0 | 5650 | | if (isEncoderSupportAfbc) |
| | 5651 | | { |
| 0 | 5652 | | hwOverlayFilter += ":afbc=1"; |
| | 5653 | | } |
| | 5654 | |
|
| 0 | 5655 | | overlayFilters.Add(hwOverlayFilter); |
| | 5656 | | } |
| | 5657 | | } |
| 0 | 5658 | | else if (memoryOutput) |
| | 5659 | | { |
| 0 | 5660 | | if (hasGraphicalSubs) |
| | 5661 | | { |
| 0 | 5662 | | var subPreProcFilters = GetGraphicalSubPreProcessFilters(swpInW, swpInH, subW, subH, reqW, reqH, req |
| 0 | 5663 | | subFilters.Add(subPreProcFilters); |
| 0 | 5664 | | overlayFilters.Add("overlay=eof_action=pass:repeatlast=0"); |
| | 5665 | | } |
| | 5666 | | } |
| | 5667 | |
|
| 0 | 5668 | | return (mainFilters, subFilters, overlayFilters); |
| | 5669 | | } |
| | 5670 | |
|
| | 5671 | | /// <summary> |
| | 5672 | | /// Gets the parameter of video processing filters. |
| | 5673 | | /// </summary> |
| | 5674 | | /// <param name="state">Encoding state.</param> |
| | 5675 | | /// <param name="options">Encoding options.</param> |
| | 5676 | | /// <param name="outputVideoCodec">Video codec to use.</param> |
| | 5677 | | /// <returns>The video processing filters parameter.</returns> |
| | 5678 | | public string GetVideoProcessingFilterParam( |
| | 5679 | | EncodingJobInfo state, |
| | 5680 | | EncodingOptions options, |
| | 5681 | | string outputVideoCodec) |
| | 5682 | | { |
| 0 | 5683 | | var videoStream = state.VideoStream; |
| 0 | 5684 | | if (videoStream is null) |
| | 5685 | | { |
| 0 | 5686 | | return string.Empty; |
| | 5687 | | } |
| | 5688 | |
|
| 0 | 5689 | | var hasSubs = state.SubtitleStream is not null && state.SubtitleDeliveryMethod == SubtitleDeliveryMethod.Enc |
| 0 | 5690 | | var hasTextSubs = hasSubs && state.SubtitleStream.IsTextSubtitleStream; |
| 0 | 5691 | | var hasGraphicalSubs = hasSubs && !state.SubtitleStream.IsTextSubtitleStream; |
| | 5692 | |
|
| | 5693 | | List<string> mainFilters; |
| | 5694 | | List<string> subFilters; |
| | 5695 | | List<string> overlayFilters; |
| | 5696 | |
|
| 0 | 5697 | | (mainFilters, subFilters, overlayFilters) = options.HardwareAccelerationType switch |
| 0 | 5698 | | { |
| 0 | 5699 | | HardwareAccelerationType.vaapi => GetVaapiVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5700 | | HardwareAccelerationType.amf => GetAmdVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5701 | | HardwareAccelerationType.qsv => GetIntelVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5702 | | HardwareAccelerationType.nvenc => GetNvidiaVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5703 | | HardwareAccelerationType.videotoolbox => GetAppleVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5704 | | HardwareAccelerationType.rkmpp => GetRkmppVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5705 | | _ => GetSwVidFilterChain(state, options, outputVideoCodec), |
| 0 | 5706 | | }; |
| | 5707 | |
|
| 0 | 5708 | | mainFilters?.RemoveAll(string.IsNullOrEmpty); |
| 0 | 5709 | | subFilters?.RemoveAll(string.IsNullOrEmpty); |
| 0 | 5710 | | overlayFilters?.RemoveAll(string.IsNullOrEmpty); |
| | 5711 | |
|
| 0 | 5712 | | var framerate = GetFramerateParam(state); |
| 0 | 5713 | | if (framerate.HasValue) |
| | 5714 | | { |
| 0 | 5715 | | mainFilters.Insert(0, string.Format( |
| 0 | 5716 | | CultureInfo.InvariantCulture, |
| 0 | 5717 | | "fps={0}", |
| 0 | 5718 | | framerate.Value)); |
| | 5719 | | } |
| | 5720 | |
|
| 0 | 5721 | | var mainStr = string.Empty; |
| 0 | 5722 | | if (mainFilters?.Count > 0) |
| | 5723 | | { |
| 0 | 5724 | | mainStr = string.Format( |
| 0 | 5725 | | CultureInfo.InvariantCulture, |
| 0 | 5726 | | "{0}", |
| 0 | 5727 | | string.Join(',', mainFilters)); |
| | 5728 | | } |
| | 5729 | |
|
| 0 | 5730 | | if (overlayFilters?.Count == 0) |
| | 5731 | | { |
| | 5732 | | // -vf "scale..." |
| 0 | 5733 | | return string.IsNullOrEmpty(mainStr) ? string.Empty : " -vf \"" + mainStr + "\""; |
| | 5734 | | } |
| | 5735 | |
|
| 0 | 5736 | | if (overlayFilters?.Count > 0 |
| 0 | 5737 | | && subFilters?.Count > 0 |
| 0 | 5738 | | && state.SubtitleStream is not null) |
| | 5739 | | { |
| | 5740 | | // overlay graphical/text subtitles |
| 0 | 5741 | | var subStr = string.Format( |
| 0 | 5742 | | CultureInfo.InvariantCulture, |
| 0 | 5743 | | "{0}", |
| 0 | 5744 | | string.Join(',', subFilters)); |
| | 5745 | |
|
| 0 | 5746 | | var overlayStr = string.Format( |
| 0 | 5747 | | CultureInfo.InvariantCulture, |
| 0 | 5748 | | "{0}", |
| 0 | 5749 | | string.Join(',', overlayFilters)); |
| | 5750 | |
|
| 0 | 5751 | | var mapPrefix = Convert.ToInt32(state.SubtitleStream.IsExternal); |
| 0 | 5752 | | var subtitleStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.SubtitleStream); |
| 0 | 5753 | | var videoStreamIndex = FindIndex(state.MediaSource.MediaStreams, state.VideoStream); |
| | 5754 | |
|
| 0 | 5755 | | if (hasSubs) |
| | 5756 | | { |
| | 5757 | | // -filter_complex "[0:s]scale=s[sub]..." |
| 0 | 5758 | | var filterStr = string.IsNullOrEmpty(mainStr) |
| 0 | 5759 | | ? " -filter_complex \"[{0}:{1}]{4}[sub];[0:{2}][sub]{5}\"" |
| 0 | 5760 | | : " -filter_complex \"[{0}:{1}]{4}[sub];[0:{2}]{3}[main];[main][sub]{5}\""; |
| | 5761 | |
|
| 0 | 5762 | | if (hasTextSubs) |
| | 5763 | | { |
| 0 | 5764 | | filterStr = string.IsNullOrEmpty(mainStr) |
| 0 | 5765 | | ? " -filter_complex \"{4}[sub];[0:{2}][sub]{5}\"" |
| 0 | 5766 | | : " -filter_complex \"{4}[sub];[0:{2}]{3}[main];[main][sub]{5}\""; |
| | 5767 | | } |
| | 5768 | |
|
| 0 | 5769 | | return string.Format( |
| 0 | 5770 | | CultureInfo.InvariantCulture, |
| 0 | 5771 | | filterStr, |
| 0 | 5772 | | mapPrefix, |
| 0 | 5773 | | subtitleStreamIndex, |
| 0 | 5774 | | videoStreamIndex, |
| 0 | 5775 | | mainStr, |
| 0 | 5776 | | subStr, |
| 0 | 5777 | | overlayStr); |
| | 5778 | | } |
| | 5779 | | } |
| | 5780 | |
|
| 0 | 5781 | | return string.Empty; |
| | 5782 | | } |
| | 5783 | |
|
| | 5784 | | public string GetOverwriteColorPropertiesParam(EncodingJobInfo state, bool isTonemapAvailable) |
| | 5785 | | { |
| 0 | 5786 | | if (isTonemapAvailable) |
| | 5787 | | { |
| 0 | 5788 | | return GetInputHdrParam(state.VideoStream?.ColorTransfer); |
| | 5789 | | } |
| | 5790 | |
|
| 0 | 5791 | | return GetOutputSdrParam(null); |
| | 5792 | | } |
| | 5793 | |
|
| | 5794 | | public string GetInputHdrParam(string colorTransfer) |
| | 5795 | | { |
| 0 | 5796 | | if (string.Equals(colorTransfer, "arib-std-b67", StringComparison.OrdinalIgnoreCase)) |
| | 5797 | | { |
| | 5798 | | // HLG |
| 0 | 5799 | | return "setparams=color_primaries=bt2020:color_trc=arib-std-b67:colorspace=bt2020nc"; |
| | 5800 | | } |
| | 5801 | |
|
| | 5802 | | // HDR10 |
| 0 | 5803 | | return "setparams=color_primaries=bt2020:color_trc=smpte2084:colorspace=bt2020nc"; |
| | 5804 | | } |
| | 5805 | |
|
| | 5806 | | public string GetOutputSdrParam(string tonemappingRange) |
| | 5807 | | { |
| | 5808 | | // SDR |
| 0 | 5809 | | if (string.Equals(tonemappingRange, "tv", StringComparison.OrdinalIgnoreCase)) |
| | 5810 | | { |
| 0 | 5811 | | return "setparams=color_primaries=bt709:color_trc=bt709:colorspace=bt709:range=tv"; |
| | 5812 | | } |
| | 5813 | |
|
| 0 | 5814 | | if (string.Equals(tonemappingRange, "pc", StringComparison.OrdinalIgnoreCase)) |
| | 5815 | | { |
| 0 | 5816 | | return "setparams=color_primaries=bt709:color_trc=bt709:colorspace=bt709:range=pc"; |
| | 5817 | | } |
| | 5818 | |
|
| 0 | 5819 | | return "setparams=color_primaries=bt709:color_trc=bt709:colorspace=bt709"; |
| | 5820 | | } |
| | 5821 | |
|
| | 5822 | | public static int GetVideoColorBitDepth(EncodingJobInfo state) |
| | 5823 | | { |
| 0 | 5824 | | var videoStream = state.VideoStream; |
| 0 | 5825 | | if (videoStream is not null) |
| | 5826 | | { |
| 0 | 5827 | | if (videoStream.BitDepth.HasValue) |
| | 5828 | | { |
| 0 | 5829 | | return videoStream.BitDepth.Value; |
| | 5830 | | } |
| | 5831 | |
|
| 0 | 5832 | | if (string.Equals(videoStream.PixelFormat, "yuv420p", StringComparison.OrdinalIgnoreCase) |
| 0 | 5833 | | || string.Equals(videoStream.PixelFormat, "yuvj420p", StringComparison.OrdinalIgnoreCase) |
| 0 | 5834 | | || string.Equals(videoStream.PixelFormat, "yuv422p", StringComparison.OrdinalIgnoreCase) |
| 0 | 5835 | | || string.Equals(videoStream.PixelFormat, "yuv444p", StringComparison.OrdinalIgnoreCase)) |
| | 5836 | | { |
| 0 | 5837 | | return 8; |
| | 5838 | | } |
| | 5839 | |
|
| 0 | 5840 | | if (string.Equals(videoStream.PixelFormat, "yuv420p10le", StringComparison.OrdinalIgnoreCase) |
| 0 | 5841 | | || string.Equals(videoStream.PixelFormat, "yuv422p10le", StringComparison.OrdinalIgnoreCase) |
| 0 | 5842 | | || string.Equals(videoStream.PixelFormat, "yuv444p10le", StringComparison.OrdinalIgnoreCase)) |
| | 5843 | | { |
| 0 | 5844 | | return 10; |
| | 5845 | | } |
| | 5846 | |
|
| 0 | 5847 | | if (string.Equals(videoStream.PixelFormat, "yuv420p12le", StringComparison.OrdinalIgnoreCase) |
| 0 | 5848 | | || string.Equals(videoStream.PixelFormat, "yuv422p12le", StringComparison.OrdinalIgnoreCase) |
| 0 | 5849 | | || string.Equals(videoStream.PixelFormat, "yuv444p12le", StringComparison.OrdinalIgnoreCase)) |
| | 5850 | | { |
| 0 | 5851 | | return 12; |
| | 5852 | | } |
| | 5853 | |
|
| 0 | 5854 | | return 8; |
| | 5855 | | } |
| | 5856 | |
|
| 0 | 5857 | | return 0; |
| | 5858 | | } |
| | 5859 | |
|
| | 5860 | | /// <summary> |
| | 5861 | | /// Gets the ffmpeg option string for the hardware accelerated video decoder. |
| | 5862 | | /// </summary> |
| | 5863 | | /// <param name="state">The encoding job info.</param> |
| | 5864 | | /// <param name="options">The encoding options.</param> |
| | 5865 | | /// <returns>The option string or null if none available.</returns> |
| | 5866 | | protected string GetHardwareVideoDecoder(EncodingJobInfo state, EncodingOptions options) |
| | 5867 | | { |
| 0 | 5868 | | var videoStream = state.VideoStream; |
| 0 | 5869 | | var mediaSource = state.MediaSource; |
| 0 | 5870 | | if (videoStream is null || mediaSource is null) |
| | 5871 | | { |
| 0 | 5872 | | return null; |
| | 5873 | | } |
| | 5874 | |
|
| | 5875 | | // HWA decoders can handle both video files and video folders. |
| 0 | 5876 | | var videoType = state.VideoType; |
| 0 | 5877 | | if (videoType != VideoType.VideoFile |
| 0 | 5878 | | && videoType != VideoType.Iso |
| 0 | 5879 | | && videoType != VideoType.Dvd |
| 0 | 5880 | | && videoType != VideoType.BluRay) |
| | 5881 | | { |
| 0 | 5882 | | return null; |
| | 5883 | | } |
| | 5884 | |
|
| 0 | 5885 | | if (IsCopyCodec(state.OutputVideoCodec)) |
| | 5886 | | { |
| 0 | 5887 | | return null; |
| | 5888 | | } |
| | 5889 | |
|
| 0 | 5890 | | var hardwareAccelerationType = options.HardwareAccelerationType; |
| | 5891 | |
|
| 0 | 5892 | | if (!string.IsNullOrEmpty(videoStream.Codec) && hardwareAccelerationType != HardwareAccelerationType.none) |
| | 5893 | | { |
| 0 | 5894 | | var bitDepth = GetVideoColorBitDepth(state); |
| | 5895 | |
|
| | 5896 | | // Only HEVC, VP9 and AV1 formats have 10-bit hardware decoder support for most platforms |
| 0 | 5897 | | if (bitDepth == 10 |
| 0 | 5898 | | && !(string.Equals(videoStream.Codec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 5899 | | || string.Equals(videoStream.Codec, "h265", StringComparison.OrdinalIgnoreCase) |
| 0 | 5900 | | || string.Equals(videoStream.Codec, "vp9", StringComparison.OrdinalIgnoreCase) |
| 0 | 5901 | | || string.Equals(videoStream.Codec, "av1", StringComparison.OrdinalIgnoreCase))) |
| | 5902 | | { |
| | 5903 | | // RKMPP has H.264 Hi10P decoder |
| 0 | 5904 | | bool hasHardwareHi10P = hardwareAccelerationType == HardwareAccelerationType.rkmpp; |
| | 5905 | |
|
| | 5906 | | // VideoToolbox on Apple Silicon has H.264 Hi10P mode enabled after macOS 14.6 |
| 0 | 5907 | | if (hardwareAccelerationType == HardwareAccelerationType.videotoolbox) |
| | 5908 | | { |
| 0 | 5909 | | var ver = Environment.OSVersion.Version; |
| 0 | 5910 | | var arch = RuntimeInformation.OSArchitecture; |
| 0 | 5911 | | if (arch.Equals(Architecture.Arm64) && ver >= new Version(14, 6)) |
| | 5912 | | { |
| 0 | 5913 | | hasHardwareHi10P = true; |
| | 5914 | | } |
| | 5915 | | } |
| | 5916 | |
|
| 0 | 5917 | | if (!hasHardwareHi10P |
| 0 | 5918 | | && string.Equals(videoStream.Codec, "h264", StringComparison.OrdinalIgnoreCase)) |
| | 5919 | | { |
| 0 | 5920 | | return null; |
| | 5921 | | } |
| | 5922 | | } |
| | 5923 | |
|
| 0 | 5924 | | var decoder = hardwareAccelerationType switch |
| 0 | 5925 | | { |
| 0 | 5926 | | HardwareAccelerationType.vaapi => GetVaapiVidDecoder(state, options, videoStream, bitDepth), |
| 0 | 5927 | | HardwareAccelerationType.amf => GetAmfVidDecoder(state, options, videoStream, bitDepth), |
| 0 | 5928 | | HardwareAccelerationType.qsv => GetQsvHwVidDecoder(state, options, videoStream, bitDepth), |
| 0 | 5929 | | HardwareAccelerationType.nvenc => GetNvdecVidDecoder(state, options, videoStream, bitDepth), |
| 0 | 5930 | | HardwareAccelerationType.videotoolbox => GetVideotoolboxVidDecoder(state, options, videoStream, bitD |
| 0 | 5931 | | HardwareAccelerationType.rkmpp => GetRkmppVidDecoder(state, options, videoStream, bitDepth), |
| 0 | 5932 | | _ => string.Empty |
| 0 | 5933 | | }; |
| | 5934 | |
|
| 0 | 5935 | | if (!string.IsNullOrEmpty(decoder)) |
| | 5936 | | { |
| 0 | 5937 | | return decoder; |
| | 5938 | | } |
| | 5939 | | } |
| | 5940 | |
|
| 0 | 5941 | | var whichCodec = videoStream.Codec; |
| 0 | 5942 | | if (string.Equals(whichCodec, "avc", StringComparison.OrdinalIgnoreCase)) |
| | 5943 | | { |
| 0 | 5944 | | whichCodec = "h264"; |
| | 5945 | | } |
| 0 | 5946 | | else if (string.Equals(whichCodec, "h265", StringComparison.OrdinalIgnoreCase)) |
| | 5947 | | { |
| 0 | 5948 | | whichCodec = "hevc"; |
| | 5949 | | } |
| | 5950 | |
|
| | 5951 | | // Avoid a second attempt if no hardware acceleration is being used |
| 0 | 5952 | | options.HardwareDecodingCodecs = options.HardwareDecodingCodecs.Where(c => !string.Equals(c, whichCodec, Str |
| | 5953 | |
|
| | 5954 | | // leave blank so ffmpeg will decide |
| 0 | 5955 | | return null; |
| | 5956 | | } |
| | 5957 | |
|
| | 5958 | | /// <summary> |
| | 5959 | | /// Gets a hw decoder name. |
| | 5960 | | /// </summary> |
| | 5961 | | /// <param name="options">Encoding options.</param> |
| | 5962 | | /// <param name="decoderPrefix">Decoder prefix.</param> |
| | 5963 | | /// <param name="decoderSuffix">Decoder suffix.</param> |
| | 5964 | | /// <param name="videoCodec">Video codec to use.</param> |
| | 5965 | | /// <param name="bitDepth">Video color bit depth.</param> |
| | 5966 | | /// <returns>Hardware decoder name.</returns> |
| | 5967 | | public string GetHwDecoderName(EncodingOptions options, string decoderPrefix, string decoderSuffix, string video |
| | 5968 | | { |
| 0 | 5969 | | if (string.IsNullOrEmpty(decoderPrefix) || string.IsNullOrEmpty(decoderSuffix)) |
| | 5970 | | { |
| 0 | 5971 | | return null; |
| | 5972 | | } |
| | 5973 | |
|
| 0 | 5974 | | var decoderName = decoderPrefix + '_' + decoderSuffix; |
| | 5975 | |
|
| 0 | 5976 | | var isCodecAvailable = _mediaEncoder.SupportsDecoder(decoderName) && options.HardwareDecodingCodecs.Contains |
| 0 | 5977 | | if (bitDepth == 10 && isCodecAvailable) |
| | 5978 | | { |
| 0 | 5979 | | if (string.Equals(videoCodec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 5980 | | && options.HardwareDecodingCodecs.Contains("hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 5981 | | && !options.EnableDecodingColorDepth10Hevc) |
| | 5982 | | { |
| 0 | 5983 | | return null; |
| | 5984 | | } |
| | 5985 | |
|
| 0 | 5986 | | if (string.Equals(videoCodec, "vp9", StringComparison.OrdinalIgnoreCase) |
| 0 | 5987 | | && options.HardwareDecodingCodecs.Contains("vp9", StringComparison.OrdinalIgnoreCase) |
| 0 | 5988 | | && !options.EnableDecodingColorDepth10Vp9) |
| | 5989 | | { |
| 0 | 5990 | | return null; |
| | 5991 | | } |
| | 5992 | | } |
| | 5993 | |
|
| 0 | 5994 | | if (string.Equals(decoderSuffix, "cuvid", StringComparison.OrdinalIgnoreCase) && options.EnableEnhancedNvdec |
| | 5995 | | { |
| 0 | 5996 | | return null; |
| | 5997 | | } |
| | 5998 | |
|
| 0 | 5999 | | if (string.Equals(decoderSuffix, "qsv", StringComparison.OrdinalIgnoreCase) && options.PreferSystemNativeHwD |
| | 6000 | | { |
| 0 | 6001 | | return null; |
| | 6002 | | } |
| | 6003 | |
|
| 0 | 6004 | | if (string.Equals(decoderSuffix, "rkmpp", StringComparison.OrdinalIgnoreCase)) |
| | 6005 | | { |
| 0 | 6006 | | return null; |
| | 6007 | | } |
| | 6008 | |
|
| 0 | 6009 | | return isCodecAvailable ? (" -c:v " + decoderName) : null; |
| | 6010 | | } |
| | 6011 | |
|
| | 6012 | | /// <summary> |
| | 6013 | | /// Gets a hwaccel type to use as a hardware decoder depending on the system. |
| | 6014 | | /// </summary> |
| | 6015 | | /// <param name="state">Encoding state.</param> |
| | 6016 | | /// <param name="options">Encoding options.</param> |
| | 6017 | | /// <param name="videoCodec">Video codec to use.</param> |
| | 6018 | | /// <param name="bitDepth">Video color bit depth.</param> |
| | 6019 | | /// <param name="outputHwSurface">Specifies if output hw surface.</param> |
| | 6020 | | /// <returns>Hardware accelerator type.</returns> |
| | 6021 | | public string GetHwaccelType(EncodingJobInfo state, EncodingOptions options, string videoCodec, int bitDepth, bo |
| | 6022 | | { |
| 0 | 6023 | | var isWindows = OperatingSystem.IsWindows(); |
| 0 | 6024 | | var isLinux = OperatingSystem.IsLinux(); |
| 0 | 6025 | | var isMacOS = OperatingSystem.IsMacOS(); |
| 0 | 6026 | | var isD3d11Supported = isWindows && _mediaEncoder.SupportsHwaccel("d3d11va"); |
| 0 | 6027 | | var isVaapiSupported = isLinux && IsVaapiSupported(state); |
| 0 | 6028 | | var isCudaSupported = (isLinux || isWindows) && IsCudaFullSupported(); |
| 0 | 6029 | | var isQsvSupported = (isLinux || isWindows) && _mediaEncoder.SupportsHwaccel("qsv"); |
| 0 | 6030 | | var isVideotoolboxSupported = isMacOS && _mediaEncoder.SupportsHwaccel("videotoolbox"); |
| 0 | 6031 | | var isRkmppSupported = isLinux && IsRkmppFullSupported(); |
| 0 | 6032 | | var isCodecAvailable = options.HardwareDecodingCodecs.Contains(videoCodec, StringComparison.OrdinalIgnoreCas |
| 0 | 6033 | | var hardwareAccelerationType = options.HardwareAccelerationType; |
| | 6034 | |
|
| 0 | 6035 | | var ffmpegVersion = _mediaEncoder.EncoderVersion; |
| | 6036 | |
|
| | 6037 | | // Set the av1 codec explicitly to trigger hw accelerator, otherwise libdav1d will be used. |
| 0 | 6038 | | var isAv1 = ffmpegVersion < _minFFmpegImplictHwaccel |
| 0 | 6039 | | && string.Equals(videoCodec, "av1", StringComparison.OrdinalIgnoreCase); |
| | 6040 | |
|
| | 6041 | | // Allow profile mismatch if decoding H.264 baseline with d3d11va and vaapi hwaccels. |
| 0 | 6042 | | var profileMismatch = string.Equals(videoCodec, "h264", StringComparison.OrdinalIgnoreCase) |
| 0 | 6043 | | && string.Equals(state.VideoStream?.Profile, "baseline", StringComparison.OrdinalIgnoreCase); |
| | 6044 | |
|
| | 6045 | | // Disable the extra internal copy in nvdec. We already handle it in filter chain. |
| 0 | 6046 | | var nvdecNoInternalCopy = ffmpegVersion >= _minFFmpegHwaUnsafeOutput; |
| | 6047 | |
|
| | 6048 | | // Strip the display rotation side data from the transposed fmp4 output stream. |
| 0 | 6049 | | var stripRotationData = (state.VideoStream?.Rotation ?? 0) != 0 |
| 0 | 6050 | | && ffmpegVersion >= _minFFmpegDisplayRotationOption; |
| 0 | 6051 | | var stripRotationDataArgs = stripRotationData ? " -display_rotation 0" : string.Empty; |
| | 6052 | |
|
| 0 | 6053 | | if (bitDepth == 10 && isCodecAvailable) |
| | 6054 | | { |
| 0 | 6055 | | if (string.Equals(videoCodec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 6056 | | && options.HardwareDecodingCodecs.Contains("hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 6057 | | && !options.EnableDecodingColorDepth10Hevc) |
| | 6058 | | { |
| 0 | 6059 | | return null; |
| | 6060 | | } |
| | 6061 | |
|
| 0 | 6062 | | if (string.Equals(videoCodec, "vp9", StringComparison.OrdinalIgnoreCase) |
| 0 | 6063 | | && options.HardwareDecodingCodecs.Contains("vp9", StringComparison.OrdinalIgnoreCase) |
| 0 | 6064 | | && !options.EnableDecodingColorDepth10Vp9) |
| | 6065 | | { |
| 0 | 6066 | | return null; |
| | 6067 | | } |
| | 6068 | | } |
| | 6069 | |
|
| | 6070 | | // Intel qsv/d3d11va/vaapi |
| 0 | 6071 | | if (hardwareAccelerationType == HardwareAccelerationType.qsv) |
| | 6072 | | { |
| 0 | 6073 | | if (options.PreferSystemNativeHwDecoder) |
| | 6074 | | { |
| 0 | 6075 | | if (isVaapiSupported && isCodecAvailable) |
| | 6076 | | { |
| 0 | 6077 | | return " -hwaccel vaapi" + (outputHwSurface ? " -hwaccel_output_format vaapi -noautorotate" + st |
| 0 | 6078 | | + (profileMismatch ? " -hwaccel_flags +allow_profile_mismatch" : string.Empty) + (isAv1 ? " |
| | 6079 | | } |
| | 6080 | |
|
| 0 | 6081 | | if (isD3d11Supported && isCodecAvailable) |
| | 6082 | | { |
| 0 | 6083 | | return " -hwaccel d3d11va" + (outputHwSurface ? " -hwaccel_output_format d3d11 -noautorotate" + |
| 0 | 6084 | | + (profileMismatch ? " -hwaccel_flags +allow_profile_mismatch" : string.Empty) + " -threads |
| | 6085 | | } |
| | 6086 | | } |
| | 6087 | | else |
| | 6088 | | { |
| 0 | 6089 | | if (isQsvSupported && isCodecAvailable) |
| | 6090 | | { |
| 0 | 6091 | | return " -hwaccel qsv" + (outputHwSurface ? " -hwaccel_output_format qsv -noautorotate" + stripR |
| | 6092 | | } |
| | 6093 | | } |
| | 6094 | | } |
| | 6095 | |
|
| | 6096 | | // Nvidia cuda |
| 0 | 6097 | | if (hardwareAccelerationType == HardwareAccelerationType.nvenc) |
| | 6098 | | { |
| 0 | 6099 | | if (isCudaSupported && isCodecAvailable) |
| | 6100 | | { |
| 0 | 6101 | | if (options.EnableEnhancedNvdecDecoder) |
| | 6102 | | { |
| | 6103 | | // set -threads 1 to nvdec decoder explicitly since it doesn't implement threading support. |
| 0 | 6104 | | return " -hwaccel cuda" + (outputHwSurface ? " -hwaccel_output_format cuda -noautorotate" + stri |
| 0 | 6105 | | + (nvdecNoInternalCopy ? " -hwaccel_flags +unsafe_output" : string.Empty) + " -threads 1" + |
| | 6106 | | } |
| | 6107 | |
|
| | 6108 | | // cuvid decoder doesn't have threading issue. |
| 0 | 6109 | | return " -hwaccel cuda" + (outputHwSurface ? " -hwaccel_output_format cuda -noautorotate" + stripRot |
| | 6110 | | } |
| | 6111 | | } |
| | 6112 | |
|
| | 6113 | | // Amd d3d11va |
| 0 | 6114 | | if (hardwareAccelerationType == HardwareAccelerationType.amf) |
| | 6115 | | { |
| 0 | 6116 | | if (isD3d11Supported && isCodecAvailable) |
| | 6117 | | { |
| 0 | 6118 | | return " -hwaccel d3d11va" + (outputHwSurface ? " -hwaccel_output_format d3d11 -noautorotate" + stri |
| 0 | 6119 | | + (profileMismatch ? " -hwaccel_flags +allow_profile_mismatch" : string.Empty) + (isAv1 ? " -c:v |
| | 6120 | | } |
| | 6121 | | } |
| | 6122 | |
|
| | 6123 | | // Vaapi |
| 0 | 6124 | | if (hardwareAccelerationType == HardwareAccelerationType.vaapi |
| 0 | 6125 | | && isVaapiSupported |
| 0 | 6126 | | && isCodecAvailable) |
| | 6127 | | { |
| 0 | 6128 | | return " -hwaccel vaapi" + (outputHwSurface ? " -hwaccel_output_format vaapi -noautorotate" + stripRotat |
| 0 | 6129 | | + (profileMismatch ? " -hwaccel_flags +allow_profile_mismatch" : string.Empty) + (isAv1 ? " -c:v av1 |
| | 6130 | | } |
| | 6131 | |
|
| | 6132 | | // Apple videotoolbox |
| 0 | 6133 | | if (hardwareAccelerationType == HardwareAccelerationType.videotoolbox |
| 0 | 6134 | | && isVideotoolboxSupported |
| 0 | 6135 | | && isCodecAvailable) |
| | 6136 | | { |
| 0 | 6137 | | return " -hwaccel videotoolbox" + (outputHwSurface ? " -hwaccel_output_format videotoolbox_vld" : string |
| | 6138 | | } |
| | 6139 | |
|
| | 6140 | | // Rockchip rkmpp |
| 0 | 6141 | | if (hardwareAccelerationType == HardwareAccelerationType.rkmpp |
| 0 | 6142 | | && isRkmppSupported |
| 0 | 6143 | | && isCodecAvailable) |
| | 6144 | | { |
| 0 | 6145 | | return " -hwaccel rkmpp" + (outputHwSurface ? " -hwaccel_output_format drm_prime -noautorotate" + stripR |
| | 6146 | | } |
| | 6147 | |
|
| 0 | 6148 | | return null; |
| | 6149 | | } |
| | 6150 | |
|
| | 6151 | | public string GetQsvHwVidDecoder(EncodingJobInfo state, EncodingOptions options, MediaStream videoStream, int bi |
| | 6152 | | { |
| 0 | 6153 | | var isWindows = OperatingSystem.IsWindows(); |
| 0 | 6154 | | var isLinux = OperatingSystem.IsLinux(); |
| | 6155 | |
|
| 0 | 6156 | | if ((!isWindows && !isLinux) |
| 0 | 6157 | | || options.HardwareAccelerationType != HardwareAccelerationType.qsv) |
| | 6158 | | { |
| 0 | 6159 | | return null; |
| | 6160 | | } |
| | 6161 | |
|
| 0 | 6162 | | var isQsvOclSupported = _mediaEncoder.SupportsHwaccel("qsv") && IsOpenclFullSupported(); |
| 0 | 6163 | | var isIntelDx11OclSupported = isWindows |
| 0 | 6164 | | && _mediaEncoder.SupportsHwaccel("d3d11va") |
| 0 | 6165 | | && isQsvOclSupported; |
| 0 | 6166 | | var isIntelVaapiOclSupported = isLinux |
| 0 | 6167 | | && IsVaapiSupported(state) |
| 0 | 6168 | | && isQsvOclSupported; |
| 0 | 6169 | | var hwSurface = (isIntelDx11OclSupported || isIntelVaapiOclSupported) |
| 0 | 6170 | | && _mediaEncoder.SupportsFilter("alphasrc"); |
| | 6171 | |
|
| 0 | 6172 | | var is8bitSwFormatsQsv = string.Equals("yuv420p", videoStream.PixelFormat, StringComparison.OrdinalIgnoreCas |
| 0 | 6173 | | || string.Equals("yuvj420p", videoStream.PixelFormat, StringComparison.OrdinalIgnor |
| 0 | 6174 | | var is8_10bitSwFormatsQsv = is8bitSwFormatsQsv || string.Equals("yuv420p10le", videoStream.PixelFormat, Stri |
| | 6175 | | // TODO: add more 8/10bit and 4:4:4 formats for Qsv after finishing the ffcheck tool |
| | 6176 | |
|
| 0 | 6177 | | if (is8bitSwFormatsQsv) |
| | 6178 | | { |
| 0 | 6179 | | if (string.Equals(videoStream.Codec, "avc", StringComparison.OrdinalIgnoreCase) |
| 0 | 6180 | | || string.Equals(videoStream.Codec, "h264", StringComparison.OrdinalIgnoreCase)) |
| | 6181 | | { |
| 0 | 6182 | | return GetHwaccelType(state, options, "h264", bitDepth, hwSurface) + GetHwDecoderName(options, "h264 |
| | 6183 | | } |
| | 6184 | |
|
| 0 | 6185 | | if (string.Equals(videoStream.Codec, "vc1", StringComparison.OrdinalIgnoreCase)) |
| | 6186 | | { |
| 0 | 6187 | | return GetHwaccelType(state, options, "vc1", bitDepth, hwSurface) + GetHwDecoderName(options, "vc1", |
| | 6188 | | } |
| | 6189 | |
|
| 0 | 6190 | | if (string.Equals(videoStream.Codec, "vp8", StringComparison.OrdinalIgnoreCase)) |
| | 6191 | | { |
| 0 | 6192 | | return GetHwaccelType(state, options, "vp8", bitDepth, hwSurface) + GetHwDecoderName(options, "vp8", |
| | 6193 | | } |
| | 6194 | |
|
| 0 | 6195 | | if (string.Equals(videoStream.Codec, "mpeg2video", StringComparison.OrdinalIgnoreCase)) |
| | 6196 | | { |
| 0 | 6197 | | return GetHwaccelType(state, options, "mpeg2video", bitDepth, hwSurface) + GetHwDecoderName(options, |
| | 6198 | | } |
| | 6199 | | } |
| | 6200 | |
|
| 0 | 6201 | | if (is8_10bitSwFormatsQsv) |
| | 6202 | | { |
| 0 | 6203 | | if (string.Equals(videoStream.Codec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 6204 | | || string.Equals(videoStream.Codec, "h265", StringComparison.OrdinalIgnoreCase)) |
| | 6205 | | { |
| 0 | 6206 | | return GetHwaccelType(state, options, "hevc", bitDepth, hwSurface) + GetHwDecoderName(options, "hevc |
| | 6207 | | } |
| | 6208 | |
|
| 0 | 6209 | | if (string.Equals(videoStream.Codec, "vp9", StringComparison.OrdinalIgnoreCase)) |
| | 6210 | | { |
| 0 | 6211 | | return GetHwaccelType(state, options, "vp9", bitDepth, hwSurface) + GetHwDecoderName(options, "vp9", |
| | 6212 | | } |
| | 6213 | |
|
| 0 | 6214 | | if (string.Equals(videoStream.Codec, "av1", StringComparison.OrdinalIgnoreCase)) |
| | 6215 | | { |
| 0 | 6216 | | return GetHwaccelType(state, options, "av1", bitDepth, hwSurface) + GetHwDecoderName(options, "av1", |
| | 6217 | | } |
| | 6218 | | } |
| | 6219 | |
|
| 0 | 6220 | | return null; |
| | 6221 | | } |
| | 6222 | |
|
| | 6223 | | public string GetNvdecVidDecoder(EncodingJobInfo state, EncodingOptions options, MediaStream videoStream, int bi |
| | 6224 | | { |
| 0 | 6225 | | if ((!OperatingSystem.IsWindows() && !OperatingSystem.IsLinux()) |
| 0 | 6226 | | || options.HardwareAccelerationType != HardwareAccelerationType.nvenc) |
| | 6227 | | { |
| 0 | 6228 | | return null; |
| | 6229 | | } |
| | 6230 | |
|
| 0 | 6231 | | var hwSurface = IsCudaFullSupported() && _mediaEncoder.SupportsFilter("alphasrc"); |
| 0 | 6232 | | var is8bitSwFormatsNvdec = string.Equals("yuv420p", videoStream.PixelFormat, StringComparison.OrdinalIgnoreC |
| 0 | 6233 | | || string.Equals("yuvj420p", videoStream.PixelFormat, StringComparison.OrdinalIgn |
| 0 | 6234 | | var is8_10bitSwFormatsNvdec = is8bitSwFormatsNvdec || string.Equals("yuv420p10le", videoStream.PixelFormat, |
| | 6235 | | // TODO: add more 8/10/12bit and 4:4:4 formats for Nvdec after finishing the ffcheck tool |
| | 6236 | |
|
| 0 | 6237 | | if (is8bitSwFormatsNvdec) |
| | 6238 | | { |
| 0 | 6239 | | if (string.Equals("avc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6240 | | || string.Equals("h264", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6241 | | { |
| 0 | 6242 | | return GetHwaccelType(state, options, "h264", bitDepth, hwSurface) + GetHwDecoderName(options, "h264 |
| | 6243 | | } |
| | 6244 | |
|
| 0 | 6245 | | if (string.Equals("mpeg2video", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6246 | | { |
| 0 | 6247 | | return GetHwaccelType(state, options, "mpeg2video", bitDepth, hwSurface) + GetHwDecoderName(options, |
| | 6248 | | } |
| | 6249 | |
|
| 0 | 6250 | | if (string.Equals("vc1", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6251 | | { |
| 0 | 6252 | | return GetHwaccelType(state, options, "vc1", bitDepth, hwSurface) + GetHwDecoderName(options, "vc1", |
| | 6253 | | } |
| | 6254 | |
|
| 0 | 6255 | | if (string.Equals("mpeg4", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6256 | | { |
| 0 | 6257 | | return GetHwaccelType(state, options, "mpeg4", bitDepth, hwSurface) + GetHwDecoderName(options, "mpe |
| | 6258 | | } |
| | 6259 | |
|
| 0 | 6260 | | if (string.Equals("vp8", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6261 | | { |
| 0 | 6262 | | return GetHwaccelType(state, options, "vp8", bitDepth, hwSurface) + GetHwDecoderName(options, "vp8", |
| | 6263 | | } |
| | 6264 | | } |
| | 6265 | |
|
| 0 | 6266 | | if (is8_10bitSwFormatsNvdec) |
| | 6267 | | { |
| 0 | 6268 | | if (string.Equals("hevc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6269 | | || string.Equals("h265", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6270 | | { |
| 0 | 6271 | | return GetHwaccelType(state, options, "hevc", bitDepth, hwSurface) + GetHwDecoderName(options, "hevc |
| | 6272 | | } |
| | 6273 | |
|
| 0 | 6274 | | if (string.Equals("vp9", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6275 | | { |
| 0 | 6276 | | return GetHwaccelType(state, options, "vp9", bitDepth, hwSurface) + GetHwDecoderName(options, "vp9", |
| | 6277 | | } |
| | 6278 | |
|
| 0 | 6279 | | if (string.Equals("av1", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6280 | | { |
| 0 | 6281 | | return GetHwaccelType(state, options, "av1", bitDepth, hwSurface) + GetHwDecoderName(options, "av1", |
| | 6282 | | } |
| | 6283 | | } |
| | 6284 | |
|
| 0 | 6285 | | return null; |
| | 6286 | | } |
| | 6287 | |
|
| | 6288 | | public string GetAmfVidDecoder(EncodingJobInfo state, EncodingOptions options, MediaStream videoStream, int bitD |
| | 6289 | | { |
| 0 | 6290 | | if (!OperatingSystem.IsWindows() |
| 0 | 6291 | | || options.HardwareAccelerationType != HardwareAccelerationType.amf) |
| | 6292 | | { |
| 0 | 6293 | | return null; |
| | 6294 | | } |
| | 6295 | |
|
| 0 | 6296 | | var hwSurface = _mediaEncoder.SupportsHwaccel("d3d11va") |
| 0 | 6297 | | && IsOpenclFullSupported() |
| 0 | 6298 | | && _mediaEncoder.SupportsFilter("alphasrc"); |
| 0 | 6299 | | var is8bitSwFormatsAmf = string.Equals("yuv420p", videoStream.PixelFormat, StringComparison.OrdinalIgnoreCas |
| 0 | 6300 | | || string.Equals("yuvj420p", videoStream.PixelFormat, StringComparison.OrdinalIgnor |
| 0 | 6301 | | var is8_10bitSwFormatsAmf = is8bitSwFormatsAmf || string.Equals("yuv420p10le", videoStream.PixelFormat, Stri |
| | 6302 | |
|
| 0 | 6303 | | if (is8bitSwFormatsAmf) |
| | 6304 | | { |
| 0 | 6305 | | if (string.Equals("avc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6306 | | || string.Equals("h264", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6307 | | { |
| 0 | 6308 | | return GetHwaccelType(state, options, "h264", bitDepth, hwSurface); |
| | 6309 | | } |
| | 6310 | |
|
| 0 | 6311 | | if (string.Equals("mpeg2video", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6312 | | { |
| 0 | 6313 | | return GetHwaccelType(state, options, "mpeg2video", bitDepth, hwSurface); |
| | 6314 | | } |
| | 6315 | |
|
| 0 | 6316 | | if (string.Equals("vc1", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6317 | | { |
| 0 | 6318 | | return GetHwaccelType(state, options, "vc1", bitDepth, hwSurface); |
| | 6319 | | } |
| | 6320 | | } |
| | 6321 | |
|
| 0 | 6322 | | if (is8_10bitSwFormatsAmf) |
| | 6323 | | { |
| 0 | 6324 | | if (string.Equals("hevc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6325 | | || string.Equals("h265", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6326 | | { |
| 0 | 6327 | | return GetHwaccelType(state, options, "hevc", bitDepth, hwSurface); |
| | 6328 | | } |
| | 6329 | |
|
| 0 | 6330 | | if (string.Equals("vp9", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6331 | | { |
| 0 | 6332 | | return GetHwaccelType(state, options, "vp9", bitDepth, hwSurface); |
| | 6333 | | } |
| | 6334 | |
|
| 0 | 6335 | | if (string.Equals("av1", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6336 | | { |
| 0 | 6337 | | return GetHwaccelType(state, options, "av1", bitDepth, hwSurface); |
| | 6338 | | } |
| | 6339 | | } |
| | 6340 | |
|
| 0 | 6341 | | return null; |
| | 6342 | | } |
| | 6343 | |
|
| | 6344 | | public string GetVaapiVidDecoder(EncodingJobInfo state, EncodingOptions options, MediaStream videoStream, int bi |
| | 6345 | | { |
| 0 | 6346 | | if (!OperatingSystem.IsLinux() |
| 0 | 6347 | | || options.HardwareAccelerationType != HardwareAccelerationType.vaapi) |
| | 6348 | | { |
| 0 | 6349 | | return null; |
| | 6350 | | } |
| | 6351 | |
|
| 0 | 6352 | | var hwSurface = IsVaapiSupported(state) |
| 0 | 6353 | | && IsVaapiFullSupported() |
| 0 | 6354 | | && IsOpenclFullSupported() |
| 0 | 6355 | | && _mediaEncoder.SupportsFilter("alphasrc"); |
| 0 | 6356 | | var is8bitSwFormatsVaapi = string.Equals("yuv420p", videoStream.PixelFormat, StringComparison.OrdinalIgnoreC |
| 0 | 6357 | | || string.Equals("yuvj420p", videoStream.PixelFormat, StringComparison.OrdinalIgn |
| 0 | 6358 | | var is8_10bitSwFormatsVaapi = is8bitSwFormatsVaapi || string.Equals("yuv420p10le", videoStream.PixelFormat, |
| | 6359 | |
|
| 0 | 6360 | | if (is8bitSwFormatsVaapi) |
| | 6361 | | { |
| 0 | 6362 | | if (string.Equals("avc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6363 | | || string.Equals("h264", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6364 | | { |
| 0 | 6365 | | return GetHwaccelType(state, options, "h264", bitDepth, hwSurface); |
| | 6366 | | } |
| | 6367 | |
|
| 0 | 6368 | | if (string.Equals("mpeg2video", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6369 | | { |
| 0 | 6370 | | return GetHwaccelType(state, options, "mpeg2video", bitDepth, hwSurface); |
| | 6371 | | } |
| | 6372 | |
|
| 0 | 6373 | | if (string.Equals("vc1", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6374 | | { |
| 0 | 6375 | | return GetHwaccelType(state, options, "vc1", bitDepth, hwSurface); |
| | 6376 | | } |
| | 6377 | |
|
| 0 | 6378 | | if (string.Equals("vp8", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6379 | | { |
| 0 | 6380 | | return GetHwaccelType(state, options, "vp8", bitDepth, hwSurface); |
| | 6381 | | } |
| | 6382 | | } |
| | 6383 | |
|
| 0 | 6384 | | if (is8_10bitSwFormatsVaapi) |
| | 6385 | | { |
| 0 | 6386 | | if (string.Equals("hevc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6387 | | || string.Equals("h265", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6388 | | { |
| 0 | 6389 | | return GetHwaccelType(state, options, "hevc", bitDepth, hwSurface); |
| | 6390 | | } |
| | 6391 | |
|
| 0 | 6392 | | if (string.Equals("vp9", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6393 | | { |
| 0 | 6394 | | return GetHwaccelType(state, options, "vp9", bitDepth, hwSurface); |
| | 6395 | | } |
| | 6396 | |
|
| 0 | 6397 | | if (string.Equals("av1", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6398 | | { |
| 0 | 6399 | | return GetHwaccelType(state, options, "av1", bitDepth, hwSurface); |
| | 6400 | | } |
| | 6401 | | } |
| | 6402 | |
|
| 0 | 6403 | | return null; |
| | 6404 | | } |
| | 6405 | |
|
| | 6406 | | public string GetVideotoolboxVidDecoder(EncodingJobInfo state, EncodingOptions options, MediaStream videoStream, |
| | 6407 | | { |
| 0 | 6408 | | if (!OperatingSystem.IsMacOS() |
| 0 | 6409 | | || options.HardwareAccelerationType != HardwareAccelerationType.videotoolbox) |
| | 6410 | | { |
| 0 | 6411 | | return null; |
| | 6412 | | } |
| | 6413 | |
|
| 0 | 6414 | | var is8bitSwFormatsVt = string.Equals("yuv420p", videoStream.PixelFormat, StringComparison.OrdinalIgnoreCase |
| 0 | 6415 | | || string.Equals("yuvj420p", videoStream.PixelFormat, StringComparison.OrdinalIgnore |
| 0 | 6416 | | var is8_10bitSwFormatsVt = is8bitSwFormatsVt || string.Equals("yuv420p10le", videoStream.PixelFormat, String |
| | 6417 | |
|
| | 6418 | | // The related patches make videotoolbox hardware surface working is only available in jellyfin-ffmpeg 7.0.1 |
| 0 | 6419 | | bool useHwSurface = (_mediaEncoder.EncoderVersion >= _minFFmpegWorkingVtHwSurface) && IsVideoToolboxFullSupp |
| | 6420 | |
|
| 0 | 6421 | | if (is8bitSwFormatsVt) |
| | 6422 | | { |
| 0 | 6423 | | if (string.Equals("vp8", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6424 | | { |
| 0 | 6425 | | return GetHwaccelType(state, options, "vp8", bitDepth, useHwSurface); |
| | 6426 | | } |
| | 6427 | | } |
| | 6428 | |
|
| 0 | 6429 | | if (is8_10bitSwFormatsVt) |
| | 6430 | | { |
| 0 | 6431 | | if (string.Equals("avc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6432 | | || string.Equals("h264", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6433 | | { |
| 0 | 6434 | | return GetHwaccelType(state, options, "h264", bitDepth, useHwSurface); |
| | 6435 | | } |
| | 6436 | |
|
| 0 | 6437 | | if (string.Equals("hevc", videoStream.Codec, StringComparison.OrdinalIgnoreCase) |
| 0 | 6438 | | || string.Equals("h265", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6439 | | { |
| 0 | 6440 | | return GetHwaccelType(state, options, "hevc", bitDepth, useHwSurface); |
| | 6441 | | } |
| | 6442 | |
|
| 0 | 6443 | | if (string.Equals("vp9", videoStream.Codec, StringComparison.OrdinalIgnoreCase)) |
| | 6444 | | { |
| 0 | 6445 | | return GetHwaccelType(state, options, "vp9", bitDepth, useHwSurface); |
| | 6446 | | } |
| | 6447 | | } |
| | 6448 | |
|
| 0 | 6449 | | return null; |
| | 6450 | | } |
| | 6451 | |
|
| | 6452 | | public string GetRkmppVidDecoder(EncodingJobInfo state, EncodingOptions options, MediaStream videoStream, int bi |
| | 6453 | | { |
| 0 | 6454 | | var isLinux = OperatingSystem.IsLinux(); |
| | 6455 | |
|
| 0 | 6456 | | if (!isLinux |
| 0 | 6457 | | || options.HardwareAccelerationType != HardwareAccelerationType.rkmpp) |
| | 6458 | | { |
| 0 | 6459 | | return null; |
| | 6460 | | } |
| | 6461 | |
|
| 0 | 6462 | | var inW = state.VideoStream?.Width; |
| 0 | 6463 | | var inH = state.VideoStream?.Height; |
| 0 | 6464 | | var reqW = state.BaseRequest.Width; |
| 0 | 6465 | | var reqH = state.BaseRequest.Height; |
| 0 | 6466 | | var reqMaxW = state.BaseRequest.MaxWidth; |
| 0 | 6467 | | var reqMaxH = state.BaseRequest.MaxHeight; |
| | 6468 | |
|
| | 6469 | | // rkrga RGA2e supports range from 1/16 to 16 |
| 0 | 6470 | | if (!IsScaleRatioSupported(inW, inH, reqW, reqH, reqMaxW, reqMaxH, 16.0f)) |
| | 6471 | | { |
| 0 | 6472 | | return null; |
| | 6473 | | } |
| | 6474 | |
|
| 0 | 6475 | | var isRkmppOclSupported = IsRkmppFullSupported() && IsOpenclFullSupported(); |
| 0 | 6476 | | var hwSurface = isRkmppOclSupported |
| 0 | 6477 | | && _mediaEncoder.SupportsFilter("alphasrc"); |
| | 6478 | |
|
| | 6479 | | // rkrga RGA3 supports range from 1/8 to 8 |
| 0 | 6480 | | var isAfbcSupported = hwSurface && IsScaleRatioSupported(inW, inH, reqW, reqH, reqMaxW, reqMaxH, 8.0f); |
| | 6481 | |
|
| | 6482 | | // TODO: add more 8/10bit and 4:2:2 formats for Rkmpp after finishing the ffcheck tool |
| 0 | 6483 | | var is8bitSwFormatsRkmpp = string.Equals("yuv420p", videoStream.PixelFormat, StringComparison.OrdinalIgnoreC |
| 0 | 6484 | | || string.Equals("yuvj420p", videoStream.PixelFormat, StringComparison.OrdinalIgn |
| 0 | 6485 | | var is10bitSwFormatsRkmpp = string.Equals("yuv420p10le", videoStream.PixelFormat, StringComparison.OrdinalIg |
| 0 | 6486 | | var is8_10bitSwFormatsRkmpp = is8bitSwFormatsRkmpp || is10bitSwFormatsRkmpp; |
| | 6487 | |
|
| | 6488 | | // nv15 and nv20 are bit-stream only formats |
| 0 | 6489 | | if (is10bitSwFormatsRkmpp && !hwSurface) |
| | 6490 | | { |
| 0 | 6491 | | return null; |
| | 6492 | | } |
| | 6493 | |
|
| 0 | 6494 | | if (is8bitSwFormatsRkmpp) |
| | 6495 | | { |
| 0 | 6496 | | if (string.Equals(videoStream.Codec, "mpeg1video", StringComparison.OrdinalIgnoreCase)) |
| | 6497 | | { |
| 0 | 6498 | | return GetHwaccelType(state, options, "mpeg1video", bitDepth, hwSurface); |
| | 6499 | | } |
| | 6500 | |
|
| 0 | 6501 | | if (string.Equals(videoStream.Codec, "mpeg2video", StringComparison.OrdinalIgnoreCase)) |
| | 6502 | | { |
| 0 | 6503 | | return GetHwaccelType(state, options, "mpeg2video", bitDepth, hwSurface); |
| | 6504 | | } |
| | 6505 | |
|
| 0 | 6506 | | if (string.Equals(videoStream.Codec, "mpeg4", StringComparison.OrdinalIgnoreCase)) |
| | 6507 | | { |
| 0 | 6508 | | return GetHwaccelType(state, options, "mpeg4", bitDepth, hwSurface); |
| | 6509 | | } |
| | 6510 | |
|
| 0 | 6511 | | if (string.Equals(videoStream.Codec, "vp8", StringComparison.OrdinalIgnoreCase)) |
| | 6512 | | { |
| 0 | 6513 | | return GetHwaccelType(state, options, "vp8", bitDepth, hwSurface); |
| | 6514 | | } |
| | 6515 | | } |
| | 6516 | |
|
| 0 | 6517 | | if (is8_10bitSwFormatsRkmpp) |
| | 6518 | | { |
| 0 | 6519 | | if (string.Equals(videoStream.Codec, "avc", StringComparison.OrdinalIgnoreCase) |
| 0 | 6520 | | || string.Equals(videoStream.Codec, "h264", StringComparison.OrdinalIgnoreCase)) |
| | 6521 | | { |
| 0 | 6522 | | var accelType = GetHwaccelType(state, options, "h264", bitDepth, hwSurface); |
| 0 | 6523 | | return accelType + ((!string.IsNullOrEmpty(accelType) && isAfbcSupported) ? " -afbc rga" : string.Em |
| | 6524 | | } |
| | 6525 | |
|
| 0 | 6526 | | if (string.Equals(videoStream.Codec, "hevc", StringComparison.OrdinalIgnoreCase) |
| 0 | 6527 | | || string.Equals(videoStream.Codec, "h265", StringComparison.OrdinalIgnoreCase)) |
| | 6528 | | { |
| 0 | 6529 | | var accelType = GetHwaccelType(state, options, "hevc", bitDepth, hwSurface); |
| 0 | 6530 | | return accelType + ((!string.IsNullOrEmpty(accelType) && isAfbcSupported) ? " -afbc rga" : string.Em |
| | 6531 | | } |
| | 6532 | |
|
| 0 | 6533 | | if (string.Equals(videoStream.Codec, "vp9", StringComparison.OrdinalIgnoreCase)) |
| | 6534 | | { |
| 0 | 6535 | | var accelType = GetHwaccelType(state, options, "vp9", bitDepth, hwSurface); |
| 0 | 6536 | | return accelType + ((!string.IsNullOrEmpty(accelType) && isAfbcSupported) ? " -afbc rga" : string.Em |
| | 6537 | | } |
| | 6538 | |
|
| 0 | 6539 | | if (string.Equals(videoStream.Codec, "av1", StringComparison.OrdinalIgnoreCase)) |
| | 6540 | | { |
| 0 | 6541 | | return GetHwaccelType(state, options, "av1", bitDepth, hwSurface); |
| | 6542 | | } |
| | 6543 | | } |
| | 6544 | |
|
| 0 | 6545 | | return null; |
| | 6546 | | } |
| | 6547 | |
|
| | 6548 | | /// <summary> |
| | 6549 | | /// Gets the number of threads. |
| | 6550 | | /// </summary> |
| | 6551 | | /// <param name="state">Encoding state.</param> |
| | 6552 | | /// <param name="encodingOptions">Encoding options.</param> |
| | 6553 | | /// <param name="outputVideoCodec">Video codec to use.</param> |
| | 6554 | | /// <returns>Number of threads.</returns> |
| | 6555 | | #nullable enable |
| | 6556 | | public static int GetNumberOfThreads(EncodingJobInfo? state, EncodingOptions encodingOptions, string? outputVide |
| | 6557 | | { |
| 0 | 6558 | | var threads = state?.BaseRequest.CpuCoreLimit ?? encodingOptions.EncodingThreadCount; |
| | 6559 | |
|
| 0 | 6560 | | if (threads <= 0) |
| | 6561 | | { |
| | 6562 | | // Automatically set thread count |
| 0 | 6563 | | return 0; |
| | 6564 | | } |
| | 6565 | |
|
| 0 | 6566 | | return Math.Min(threads, Environment.ProcessorCount); |
| | 6567 | | } |
| | 6568 | |
|
| | 6569 | | #nullable disable |
| | 6570 | | public void TryStreamCopy(EncodingJobInfo state) |
| | 6571 | | { |
| 0 | 6572 | | if (state.VideoStream is not null && CanStreamCopyVideo(state, state.VideoStream)) |
| | 6573 | | { |
| 0 | 6574 | | state.OutputVideoCodec = "copy"; |
| | 6575 | | } |
| | 6576 | | else |
| | 6577 | | { |
| 0 | 6578 | | var user = state.User; |
| | 6579 | |
|
| | 6580 | | // If the user doesn't have access to transcoding, then force stream copy, regardless of whether it will |
| 0 | 6581 | | if (user is not null && !user.HasPermission(PermissionKind.EnableVideoPlaybackTranscoding)) |
| | 6582 | | { |
| 0 | 6583 | | state.OutputVideoCodec = "copy"; |
| | 6584 | | } |
| | 6585 | | } |
| | 6586 | |
|
| 0 | 6587 | | if (state.AudioStream is not null |
| 0 | 6588 | | && CanStreamCopyAudio(state, state.AudioStream, state.SupportedAudioCodecs)) |
| | 6589 | | { |
| 0 | 6590 | | state.OutputAudioCodec = "copy"; |
| | 6591 | | } |
| | 6592 | | else |
| | 6593 | | { |
| 0 | 6594 | | var user = state.User; |
| | 6595 | |
|
| | 6596 | | // If the user doesn't have access to transcoding, then force stream copy, regardless of whether it will |
| 0 | 6597 | | if (user is not null && !user.HasPermission(PermissionKind.EnableAudioPlaybackTranscoding)) |
| | 6598 | | { |
| 0 | 6599 | | state.OutputAudioCodec = "copy"; |
| | 6600 | | } |
| | 6601 | | } |
| 0 | 6602 | | } |
| | 6603 | |
|
| | 6604 | | public string GetInputModifier(EncodingJobInfo state, EncodingOptions encodingOptions, string segmentContainer) |
| | 6605 | | { |
| 0 | 6606 | | var inputModifier = string.Empty; |
| 0 | 6607 | | var analyzeDurationArgument = string.Empty; |
| | 6608 | |
|
| | 6609 | | // Apply -analyzeduration as per the environment variable, |
| | 6610 | | // otherwise ffmpeg will break on certain files due to default value is 0. |
| 0 | 6611 | | var ffmpegAnalyzeDuration = _config.GetFFmpegAnalyzeDuration() ?? string.Empty; |
| | 6612 | |
|
| 0 | 6613 | | if (state.MediaSource.AnalyzeDurationMs > 0) |
| | 6614 | | { |
| 0 | 6615 | | analyzeDurationArgument = "-analyzeduration " + (state.MediaSource.AnalyzeDurationMs.Value * 1000).ToStr |
| | 6616 | | } |
| 0 | 6617 | | else if (!string.IsNullOrEmpty(ffmpegAnalyzeDuration)) |
| | 6618 | | { |
| 0 | 6619 | | analyzeDurationArgument = "-analyzeduration " + ffmpegAnalyzeDuration; |
| | 6620 | | } |
| | 6621 | |
|
| 0 | 6622 | | if (!string.IsNullOrEmpty(analyzeDurationArgument)) |
| | 6623 | | { |
| 0 | 6624 | | inputModifier += " " + analyzeDurationArgument; |
| | 6625 | | } |
| | 6626 | |
|
| 0 | 6627 | | inputModifier = inputModifier.Trim(); |
| | 6628 | |
|
| | 6629 | | // Apply -probesize if configured |
| 0 | 6630 | | var ffmpegProbeSize = _config.GetFFmpegProbeSize(); |
| | 6631 | |
|
| 0 | 6632 | | if (!string.IsNullOrEmpty(ffmpegProbeSize)) |
| | 6633 | | { |
| 0 | 6634 | | inputModifier += $" -probesize {ffmpegProbeSize}"; |
| | 6635 | | } |
| | 6636 | |
|
| 0 | 6637 | | var userAgentParam = GetUserAgentParam(state); |
| | 6638 | |
|
| 0 | 6639 | | if (!string.IsNullOrEmpty(userAgentParam)) |
| | 6640 | | { |
| 0 | 6641 | | inputModifier += " " + userAgentParam; |
| | 6642 | | } |
| | 6643 | |
|
| 0 | 6644 | | inputModifier = inputModifier.Trim(); |
| | 6645 | |
|
| 0 | 6646 | | var refererParam = GetRefererParam(state); |
| | 6647 | |
|
| 0 | 6648 | | if (!string.IsNullOrEmpty(refererParam)) |
| | 6649 | | { |
| 0 | 6650 | | inputModifier += " " + refererParam; |
| | 6651 | | } |
| | 6652 | |
|
| 0 | 6653 | | inputModifier = inputModifier.Trim(); |
| | 6654 | |
|
| 0 | 6655 | | inputModifier += " " + GetFastSeekCommandLineParameter(state, encodingOptions, segmentContainer); |
| 0 | 6656 | | inputModifier = inputModifier.Trim(); |
| | 6657 | |
|
| 0 | 6658 | | if (state.InputProtocol == MediaProtocol.Rtsp) |
| | 6659 | | { |
| 0 | 6660 | | inputModifier += " -rtsp_transport tcp+udp -rtsp_flags prefer_tcp"; |
| | 6661 | | } |
| | 6662 | |
|
| 0 | 6663 | | if (!string.IsNullOrEmpty(state.InputAudioSync)) |
| | 6664 | | { |
| 0 | 6665 | | inputModifier += " -async " + state.InputAudioSync; |
| | 6666 | | } |
| | 6667 | |
|
| 0 | 6668 | | if (!string.IsNullOrEmpty(state.InputVideoSync)) |
| | 6669 | | { |
| 0 | 6670 | | inputModifier += " -vsync " + state.InputVideoSync; |
| | 6671 | | } |
| | 6672 | |
|
| 0 | 6673 | | if (state.ReadInputAtNativeFramerate && state.InputProtocol != MediaProtocol.Rtsp) |
| | 6674 | | { |
| 0 | 6675 | | inputModifier += " -re"; |
| | 6676 | | } |
| 0 | 6677 | | else if (encodingOptions.EnableSegmentDeletion |
| 0 | 6678 | | && state.VideoStream is not null |
| 0 | 6679 | | && state.TranscodingType == TranscodingJobType.Hls |
| 0 | 6680 | | && IsCopyCodec(state.OutputVideoCodec) |
| 0 | 6681 | | && _mediaEncoder.EncoderVersion >= _minFFmpegReadrateOption) |
| | 6682 | | { |
| | 6683 | | // Set an input read rate limit 10x for using SegmentDeletion with stream-copy |
| | 6684 | | // to prevent ffmpeg from exiting prematurely (due to fast drive) |
| 0 | 6685 | | inputModifier += " -readrate 10"; |
| | 6686 | | } |
| | 6687 | |
|
| 0 | 6688 | | var flags = new List<string>(); |
| 0 | 6689 | | if (state.IgnoreInputDts) |
| | 6690 | | { |
| 0 | 6691 | | flags.Add("+igndts"); |
| | 6692 | | } |
| | 6693 | |
|
| 0 | 6694 | | if (state.IgnoreInputIndex) |
| | 6695 | | { |
| 0 | 6696 | | flags.Add("+ignidx"); |
| | 6697 | | } |
| | 6698 | |
|
| 0 | 6699 | | if (state.GenPtsInput || IsCopyCodec(state.OutputVideoCodec)) |
| | 6700 | | { |
| 0 | 6701 | | flags.Add("+genpts"); |
| | 6702 | | } |
| | 6703 | |
|
| 0 | 6704 | | if (state.DiscardCorruptFramesInput) |
| | 6705 | | { |
| 0 | 6706 | | flags.Add("+discardcorrupt"); |
| | 6707 | | } |
| | 6708 | |
|
| 0 | 6709 | | if (state.EnableFastSeekInput) |
| | 6710 | | { |
| 0 | 6711 | | flags.Add("+fastseek"); |
| | 6712 | | } |
| | 6713 | |
|
| 0 | 6714 | | if (flags.Count > 0) |
| | 6715 | | { |
| 0 | 6716 | | inputModifier += " -fflags " + string.Join(string.Empty, flags); |
| | 6717 | | } |
| | 6718 | |
|
| 0 | 6719 | | if (state.IsVideoRequest) |
| | 6720 | | { |
| 0 | 6721 | | if (!string.IsNullOrEmpty(state.InputContainer) && state.VideoType == VideoType.VideoFile && encodingOpt |
| | 6722 | | { |
| 0 | 6723 | | var inputFormat = GetInputFormat(state.InputContainer); |
| 0 | 6724 | | if (!string.IsNullOrEmpty(inputFormat)) |
| | 6725 | | { |
| 0 | 6726 | | inputModifier += " -f " + inputFormat; |
| | 6727 | | } |
| | 6728 | | } |
| | 6729 | | } |
| | 6730 | |
|
| 0 | 6731 | | if (state.MediaSource.RequiresLooping) |
| | 6732 | | { |
| 0 | 6733 | | inputModifier += " -stream_loop -1 -reconnect_at_eof 1 -reconnect_streamed 1 -reconnect_delay_max 2"; |
| | 6734 | | } |
| | 6735 | |
|
| 0 | 6736 | | return inputModifier; |
| | 6737 | | } |
| | 6738 | |
|
| | 6739 | | public void AttachMediaSourceInfo( |
| | 6740 | | EncodingJobInfo state, |
| | 6741 | | EncodingOptions encodingOptions, |
| | 6742 | | MediaSourceInfo mediaSource, |
| | 6743 | | string requestedUrl) |
| | 6744 | | { |
| 0 | 6745 | | ArgumentNullException.ThrowIfNull(state); |
| | 6746 | |
|
| 0 | 6747 | | ArgumentNullException.ThrowIfNull(mediaSource); |
| | 6748 | |
|
| 0 | 6749 | | var path = mediaSource.Path; |
| 0 | 6750 | | var protocol = mediaSource.Protocol; |
| | 6751 | |
|
| 0 | 6752 | | if (!string.IsNullOrEmpty(mediaSource.EncoderPath) && mediaSource.EncoderProtocol.HasValue) |
| | 6753 | | { |
| 0 | 6754 | | path = mediaSource.EncoderPath; |
| 0 | 6755 | | protocol = mediaSource.EncoderProtocol.Value; |
| | 6756 | | } |
| | 6757 | |
|
| 0 | 6758 | | state.MediaPath = path; |
| 0 | 6759 | | state.InputProtocol = protocol; |
| 0 | 6760 | | state.InputContainer = mediaSource.Container; |
| 0 | 6761 | | state.RunTimeTicks = mediaSource.RunTimeTicks; |
| 0 | 6762 | | state.RemoteHttpHeaders = mediaSource.RequiredHttpHeaders; |
| | 6763 | |
|
| 0 | 6764 | | state.IsoType = mediaSource.IsoType; |
| | 6765 | |
|
| 0 | 6766 | | if (mediaSource.Timestamp.HasValue) |
| | 6767 | | { |
| 0 | 6768 | | state.InputTimestamp = mediaSource.Timestamp.Value; |
| | 6769 | | } |
| | 6770 | |
|
| 0 | 6771 | | state.RunTimeTicks = mediaSource.RunTimeTicks; |
| 0 | 6772 | | state.RemoteHttpHeaders = mediaSource.RequiredHttpHeaders; |
| 0 | 6773 | | state.ReadInputAtNativeFramerate = mediaSource.ReadAtNativeFramerate; |
| | 6774 | |
|
| 0 | 6775 | | if (state.ReadInputAtNativeFramerate |
| 0 | 6776 | | || (mediaSource.Protocol == MediaProtocol.File |
| 0 | 6777 | | && string.Equals(mediaSource.Container, "wtv", StringComparison.OrdinalIgnoreCase))) |
| | 6778 | | { |
| 0 | 6779 | | state.InputVideoSync = "-1"; |
| 0 | 6780 | | state.InputAudioSync = "1"; |
| | 6781 | | } |
| | 6782 | |
|
| 0 | 6783 | | if (string.Equals(mediaSource.Container, "wma", StringComparison.OrdinalIgnoreCase) |
| 0 | 6784 | | || string.Equals(mediaSource.Container, "asf", StringComparison.OrdinalIgnoreCase)) |
| | 6785 | | { |
| | 6786 | | // Seeing some stuttering when transcoding wma to audio-only HLS |
| 0 | 6787 | | state.InputAudioSync = "1"; |
| | 6788 | | } |
| | 6789 | |
|
| 0 | 6790 | | var mediaStreams = mediaSource.MediaStreams; |
| | 6791 | |
|
| 0 | 6792 | | if (state.IsVideoRequest) |
| | 6793 | | { |
| 0 | 6794 | | var videoRequest = state.BaseRequest; |
| | 6795 | |
|
| 0 | 6796 | | if (string.IsNullOrEmpty(videoRequest.VideoCodec)) |
| | 6797 | | { |
| 0 | 6798 | | if (string.IsNullOrEmpty(requestedUrl)) |
| | 6799 | | { |
| 0 | 6800 | | requestedUrl = "test." + videoRequest.Container; |
| | 6801 | | } |
| | 6802 | |
|
| 0 | 6803 | | videoRequest.VideoCodec = InferVideoCodec(requestedUrl); |
| | 6804 | | } |
| | 6805 | |
|
| 0 | 6806 | | state.VideoStream = GetMediaStream(mediaStreams, videoRequest.VideoStreamIndex, MediaStreamType.Video); |
| 0 | 6807 | | state.SubtitleStream = GetMediaStream(mediaStreams, videoRequest.SubtitleStreamIndex, MediaStreamType.Su |
| 0 | 6808 | | state.SubtitleDeliveryMethod = videoRequest.SubtitleMethod; |
| 0 | 6809 | | state.AudioStream = GetMediaStream(mediaStreams, videoRequest.AudioStreamIndex, MediaStreamType.Audio); |
| | 6810 | |
|
| 0 | 6811 | | if (state.SubtitleStream is not null && !state.SubtitleStream.IsExternal) |
| | 6812 | | { |
| 0 | 6813 | | state.InternalSubtitleStreamOffset = mediaStreams.Where(i => i.Type == MediaStreamType.Subtitle && ! |
| | 6814 | | } |
| | 6815 | |
|
| 0 | 6816 | | EnforceResolutionLimit(state); |
| | 6817 | |
|
| 0 | 6818 | | NormalizeSubtitleEmbed(state); |
| | 6819 | | } |
| | 6820 | | else |
| | 6821 | | { |
| 0 | 6822 | | state.AudioStream = GetMediaStream(mediaStreams, null, MediaStreamType.Audio, true); |
| | 6823 | | } |
| | 6824 | |
|
| 0 | 6825 | | state.MediaSource = mediaSource; |
| | 6826 | |
|
| 0 | 6827 | | var request = state.BaseRequest; |
| 0 | 6828 | | var supportedAudioCodecs = state.SupportedAudioCodecs; |
| 0 | 6829 | | if (request is not null && supportedAudioCodecs is not null && supportedAudioCodecs.Length > 0) |
| | 6830 | | { |
| 0 | 6831 | | var supportedAudioCodecsList = supportedAudioCodecs.ToList(); |
| | 6832 | |
|
| 0 | 6833 | | ShiftAudioCodecsIfNeeded(supportedAudioCodecsList, state.AudioStream); |
| | 6834 | |
|
| 0 | 6835 | | state.SupportedAudioCodecs = supportedAudioCodecsList.ToArray(); |
| | 6836 | |
|
| 0 | 6837 | | request.AudioCodec = state.SupportedAudioCodecs.FirstOrDefault(_mediaEncoder.CanEncodeToAudioCodec) |
| 0 | 6838 | | ?? state.SupportedAudioCodecs.FirstOrDefault(); |
| | 6839 | | } |
| | 6840 | |
|
| 0 | 6841 | | var supportedVideoCodecs = state.SupportedVideoCodecs; |
| 0 | 6842 | | if (request is not null && supportedVideoCodecs is not null && supportedVideoCodecs.Length > 0) |
| | 6843 | | { |
| 0 | 6844 | | var supportedVideoCodecsList = supportedVideoCodecs.ToList(); |
| | 6845 | |
|
| 0 | 6846 | | ShiftVideoCodecsIfNeeded(supportedVideoCodecsList, encodingOptions); |
| | 6847 | |
|
| 0 | 6848 | | state.SupportedVideoCodecs = supportedVideoCodecsList.ToArray(); |
| | 6849 | |
|
| 0 | 6850 | | request.VideoCodec = state.SupportedVideoCodecs.FirstOrDefault(); |
| | 6851 | | } |
| 0 | 6852 | | } |
| | 6853 | |
|
| | 6854 | | private void ShiftAudioCodecsIfNeeded(List<string> audioCodecs, MediaStream audioStream) |
| | 6855 | | { |
| | 6856 | | // No need to shift if there is only one supported audio codec. |
| 0 | 6857 | | if (audioCodecs.Count < 2) |
| | 6858 | | { |
| 0 | 6859 | | return; |
| | 6860 | | } |
| | 6861 | |
|
| 0 | 6862 | | var inputChannels = audioStream is null ? 6 : audioStream.Channels ?? 6; |
| 0 | 6863 | | var shiftAudioCodecs = new List<string>(); |
| 0 | 6864 | | if (inputChannels >= 6) |
| | 6865 | | { |
| | 6866 | | // DTS and TrueHD are not supported by HLS |
| | 6867 | | // Keep them in the supported codecs list, but shift them to the end of the list so that if transcoding |
| 0 | 6868 | | shiftAudioCodecs.Add("dca"); |
| 0 | 6869 | | shiftAudioCodecs.Add("truehd"); |
| | 6870 | | } |
| | 6871 | | else |
| | 6872 | | { |
| | 6873 | | // Transcoding to 2ch ac3 or eac3 almost always causes a playback failure |
| | 6874 | | // Keep them in the supported codecs list, but shift them to the end of the list so that if transcoding |
| 0 | 6875 | | shiftAudioCodecs.Add("ac3"); |
| 0 | 6876 | | shiftAudioCodecs.Add("eac3"); |
| | 6877 | | } |
| | 6878 | |
|
| 0 | 6879 | | if (audioCodecs.All(i => shiftAudioCodecs.Contains(i, StringComparison.OrdinalIgnoreCase))) |
| | 6880 | | { |
| 0 | 6881 | | return; |
| | 6882 | | } |
| | 6883 | |
|
| 0 | 6884 | | while (shiftAudioCodecs.Contains(audioCodecs[0], StringComparison.OrdinalIgnoreCase)) |
| | 6885 | | { |
| 0 | 6886 | | var removed = audioCodecs[0]; |
| 0 | 6887 | | audioCodecs.RemoveAt(0); |
| 0 | 6888 | | audioCodecs.Add(removed); |
| | 6889 | | } |
| 0 | 6890 | | } |
| | 6891 | |
|
| | 6892 | | private void ShiftVideoCodecsIfNeeded(List<string> videoCodecs, EncodingOptions encodingOptions) |
| | 6893 | | { |
| | 6894 | | // No need to shift if there is only one supported video codec. |
| 0 | 6895 | | if (videoCodecs.Count < 2) |
| | 6896 | | { |
| 0 | 6897 | | return; |
| | 6898 | | } |
| | 6899 | |
|
| | 6900 | | // Shift codecs to the end of list if it's not allowed. |
| 0 | 6901 | | var shiftVideoCodecs = new List<string>(); |
| 0 | 6902 | | if (!encodingOptions.AllowHevcEncoding) |
| | 6903 | | { |
| 0 | 6904 | | shiftVideoCodecs.Add("hevc"); |
| 0 | 6905 | | shiftVideoCodecs.Add("h265"); |
| | 6906 | | } |
| | 6907 | |
|
| 0 | 6908 | | if (!encodingOptions.AllowAv1Encoding) |
| | 6909 | | { |
| 0 | 6910 | | shiftVideoCodecs.Add("av1"); |
| | 6911 | | } |
| | 6912 | |
|
| 0 | 6913 | | if (videoCodecs.All(i => shiftVideoCodecs.Contains(i, StringComparison.OrdinalIgnoreCase))) |
| | 6914 | | { |
| 0 | 6915 | | return; |
| | 6916 | | } |
| | 6917 | |
|
| 0 | 6918 | | while (shiftVideoCodecs.Contains(videoCodecs[0], StringComparison.OrdinalIgnoreCase)) |
| | 6919 | | { |
| 0 | 6920 | | var removed = videoCodecs[0]; |
| 0 | 6921 | | videoCodecs.RemoveAt(0); |
| 0 | 6922 | | videoCodecs.Add(removed); |
| | 6923 | | } |
| 0 | 6924 | | } |
| | 6925 | |
|
| | 6926 | | private void NormalizeSubtitleEmbed(EncodingJobInfo state) |
| | 6927 | | { |
| 0 | 6928 | | if (state.SubtitleStream is null || state.SubtitleDeliveryMethod != SubtitleDeliveryMethod.Embed) |
| | 6929 | | { |
| 0 | 6930 | | return; |
| | 6931 | | } |
| | 6932 | |
|
| | 6933 | | // This is tricky to remux in, after converting to dvdsub it's not positioned correctly |
| | 6934 | | // Therefore, let's just burn it in |
| 0 | 6935 | | if (string.Equals(state.SubtitleStream.Codec, "DVBSUB", StringComparison.OrdinalIgnoreCase)) |
| | 6936 | | { |
| 0 | 6937 | | state.SubtitleDeliveryMethod = SubtitleDeliveryMethod.Encode; |
| | 6938 | | } |
| 0 | 6939 | | } |
| | 6940 | |
|
| | 6941 | | public string GetSubtitleEmbedArguments(EncodingJobInfo state) |
| | 6942 | | { |
| 0 | 6943 | | if (state.SubtitleStream is null || state.SubtitleDeliveryMethod != SubtitleDeliveryMethod.Embed) |
| | 6944 | | { |
| 0 | 6945 | | return string.Empty; |
| | 6946 | | } |
| | 6947 | |
|
| 0 | 6948 | | var format = state.SupportedSubtitleCodecs.FirstOrDefault(); |
| | 6949 | | string codec; |
| | 6950 | |
|
| 0 | 6951 | | if (string.IsNullOrEmpty(format) || string.Equals(format, state.SubtitleStream.Codec, StringComparison.Ordin |
| | 6952 | | { |
| 0 | 6953 | | codec = "copy"; |
| | 6954 | | } |
| | 6955 | | else |
| | 6956 | | { |
| 0 | 6957 | | codec = format; |
| | 6958 | | } |
| | 6959 | |
|
| 0 | 6960 | | return " -codec:s:0 " + codec + " -disposition:s:0 default"; |
| | 6961 | | } |
| | 6962 | |
|
| | 6963 | | public string GetProgressiveVideoFullCommandLine(EncodingJobInfo state, EncodingOptions encodingOptions, Encoder |
| | 6964 | | { |
| | 6965 | | // Get the output codec name |
| 0 | 6966 | | var videoCodec = GetVideoEncoder(state, encodingOptions); |
| | 6967 | |
|
| 0 | 6968 | | var format = string.Empty; |
| 0 | 6969 | | var keyFrame = string.Empty; |
| 0 | 6970 | | var outputPath = state.OutputFilePath; |
| | 6971 | |
|
| 0 | 6972 | | if (Path.GetExtension(outputPath.AsSpan()).Equals(".mp4", StringComparison.OrdinalIgnoreCase) |
| 0 | 6973 | | && state.BaseRequest.Context == EncodingContext.Streaming) |
| | 6974 | | { |
| | 6975 | | // Comparison: https://github.com/jansmolders86/mediacenterjs/blob/master/lib/transcoding/desktop.js |
| 0 | 6976 | | format = " -f mp4 -movflags frag_keyframe+empty_moov+delay_moov"; |
| | 6977 | | } |
| | 6978 | |
|
| 0 | 6979 | | var threads = GetNumberOfThreads(state, encodingOptions, videoCodec); |
| | 6980 | |
|
| 0 | 6981 | | var inputModifier = GetInputModifier(state, encodingOptions, null); |
| | 6982 | |
|
| 0 | 6983 | | return string.Format( |
| 0 | 6984 | | CultureInfo.InvariantCulture, |
| 0 | 6985 | | "{0} {1}{2} {3} {4} -map_metadata -1 -map_chapters -1 -threads {5} {6}{7}{8} -y \"{9}\"", |
| 0 | 6986 | | inputModifier, |
| 0 | 6987 | | GetInputArgument(state, encodingOptions, null), |
| 0 | 6988 | | keyFrame, |
| 0 | 6989 | | GetMapArgs(state), |
| 0 | 6990 | | GetProgressiveVideoArguments(state, encodingOptions, videoCodec, defaultPreset), |
| 0 | 6991 | | threads, |
| 0 | 6992 | | GetProgressiveVideoAudioArguments(state, encodingOptions), |
| 0 | 6993 | | GetSubtitleEmbedArguments(state), |
| 0 | 6994 | | format, |
| 0 | 6995 | | outputPath).Trim(); |
| | 6996 | | } |
| | 6997 | |
|
| | 6998 | | public string GetOutputFFlags(EncodingJobInfo state) |
| | 6999 | | { |
| 0 | 7000 | | var flags = new List<string>(); |
| 0 | 7001 | | if (state.GenPtsOutput) |
| | 7002 | | { |
| 0 | 7003 | | flags.Add("+genpts"); |
| | 7004 | | } |
| | 7005 | |
|
| 0 | 7006 | | if (flags.Count > 0) |
| | 7007 | | { |
| 0 | 7008 | | return " -fflags " + string.Join(string.Empty, flags); |
| | 7009 | | } |
| | 7010 | |
|
| 0 | 7011 | | return string.Empty; |
| | 7012 | | } |
| | 7013 | |
|
| | 7014 | | public string GetProgressiveVideoArguments(EncodingJobInfo state, EncodingOptions encodingOptions, string videoC |
| | 7015 | | { |
| 0 | 7016 | | var args = "-codec:v:0 " + videoCodec; |
| | 7017 | |
|
| 0 | 7018 | | if (state.BaseRequest.EnableMpegtsM2TsMode) |
| | 7019 | | { |
| 0 | 7020 | | args += " -mpegts_m2ts_mode 1"; |
| | 7021 | | } |
| | 7022 | |
|
| 0 | 7023 | | if (IsCopyCodec(videoCodec)) |
| | 7024 | | { |
| 0 | 7025 | | if (state.VideoStream is not null |
| 0 | 7026 | | && string.Equals(state.OutputContainer, "ts", StringComparison.OrdinalIgnoreCase) |
| 0 | 7027 | | && !string.Equals(state.VideoStream.NalLengthSize, "0", StringComparison.OrdinalIgnoreCase)) |
| | 7028 | | { |
| 0 | 7029 | | string bitStreamArgs = GetBitStreamArgs(state.VideoStream); |
| 0 | 7030 | | if (!string.IsNullOrEmpty(bitStreamArgs)) |
| | 7031 | | { |
| 0 | 7032 | | args += " " + bitStreamArgs; |
| | 7033 | | } |
| | 7034 | | } |
| | 7035 | |
|
| 0 | 7036 | | if (state.RunTimeTicks.HasValue && state.BaseRequest.CopyTimestamps) |
| | 7037 | | { |
| 0 | 7038 | | args += " -copyts -avoid_negative_ts disabled -start_at_zero"; |
| | 7039 | | } |
| | 7040 | |
|
| 0 | 7041 | | if (!state.RunTimeTicks.HasValue) |
| | 7042 | | { |
| 0 | 7043 | | args += " -fflags +genpts"; |
| | 7044 | | } |
| | 7045 | | } |
| | 7046 | | else |
| | 7047 | | { |
| 0 | 7048 | | var keyFrameArg = string.Format( |
| 0 | 7049 | | CultureInfo.InvariantCulture, |
| 0 | 7050 | | " -force_key_frames \"expr:gte(t,n_forced*{0})\"", |
| 0 | 7051 | | 5); |
| | 7052 | |
|
| 0 | 7053 | | args += keyFrameArg; |
| | 7054 | |
|
| 0 | 7055 | | var hasGraphicalSubs = state.SubtitleStream is not null && !state.SubtitleStream.IsTextSubtitleStream && |
| | 7056 | |
|
| 0 | 7057 | | var hasCopyTs = false; |
| | 7058 | |
|
| | 7059 | | // video processing filters. |
| 0 | 7060 | | var videoProcessParam = GetVideoProcessingFilterParam(state, encodingOptions, videoCodec); |
| | 7061 | |
|
| 0 | 7062 | | var negativeMapArgs = GetNegativeMapArgsByFilters(state, videoProcessParam); |
| | 7063 | |
|
| 0 | 7064 | | args = negativeMapArgs + args + videoProcessParam; |
| | 7065 | |
|
| 0 | 7066 | | hasCopyTs = videoProcessParam.Contains("copyts", StringComparison.OrdinalIgnoreCase); |
| | 7067 | |
|
| 0 | 7068 | | if (state.RunTimeTicks.HasValue && state.BaseRequest.CopyTimestamps) |
| | 7069 | | { |
| 0 | 7070 | | if (!hasCopyTs) |
| | 7071 | | { |
| 0 | 7072 | | args += " -copyts"; |
| | 7073 | | } |
| | 7074 | |
|
| 0 | 7075 | | args += " -avoid_negative_ts disabled"; |
| | 7076 | |
|
| 0 | 7077 | | if (!(state.SubtitleStream is not null && state.SubtitleStream.IsExternal && !state.SubtitleStream.I |
| | 7078 | | { |
| 0 | 7079 | | args += " -start_at_zero"; |
| | 7080 | | } |
| | 7081 | | } |
| | 7082 | |
|
| 0 | 7083 | | var qualityParam = GetVideoQualityParam(state, videoCodec, encodingOptions, defaultPreset); |
| | 7084 | |
|
| 0 | 7085 | | if (!string.IsNullOrEmpty(qualityParam)) |
| | 7086 | | { |
| 0 | 7087 | | args += " " + qualityParam.Trim(); |
| | 7088 | | } |
| | 7089 | | } |
| | 7090 | |
|
| 0 | 7091 | | if (!string.IsNullOrEmpty(state.OutputVideoSync)) |
| | 7092 | | { |
| 0 | 7093 | | args += " -vsync " + state.OutputVideoSync; |
| | 7094 | | } |
| | 7095 | |
|
| 0 | 7096 | | args += GetOutputFFlags(state); |
| | 7097 | |
|
| 0 | 7098 | | return args; |
| | 7099 | | } |
| | 7100 | |
|
| | 7101 | | public string GetProgressiveVideoAudioArguments(EncodingJobInfo state, EncodingOptions encodingOptions) |
| | 7102 | | { |
| | 7103 | | // If the video doesn't have an audio stream, return a default. |
| 0 | 7104 | | if (state.AudioStream is null && state.VideoStream is not null) |
| | 7105 | | { |
| 0 | 7106 | | return string.Empty; |
| | 7107 | | } |
| | 7108 | |
|
| | 7109 | | // Get the output codec name |
| 0 | 7110 | | var codec = GetAudioEncoder(state); |
| | 7111 | |
|
| 0 | 7112 | | var args = "-codec:a:0 " + codec; |
| | 7113 | |
|
| 0 | 7114 | | if (IsCopyCodec(codec)) |
| | 7115 | | { |
| 0 | 7116 | | return args; |
| | 7117 | | } |
| | 7118 | |
|
| 0 | 7119 | | var channels = state.OutputAudioChannels; |
| | 7120 | |
|
| 0 | 7121 | | var useDownMixAlgorithm = state.AudioStream is not null |
| 0 | 7122 | | && DownMixAlgorithmsHelper.AlgorithmFilterStrings.ContainsKey((encodingOptions.Dow |
| | 7123 | |
|
| 0 | 7124 | | if (channels.HasValue && !useDownMixAlgorithm) |
| | 7125 | | { |
| 0 | 7126 | | args += " -ac " + channels.Value; |
| | 7127 | | } |
| | 7128 | |
|
| 0 | 7129 | | var bitrate = state.OutputAudioBitrate; |
| 0 | 7130 | | if (bitrate.HasValue && !LosslessAudioCodecs.Contains(codec, StringComparison.OrdinalIgnoreCase)) |
| | 7131 | | { |
| 0 | 7132 | | var vbrParam = GetAudioVbrModeParam(codec, bitrate.Value, channels ?? 2); |
| 0 | 7133 | | if (encodingOptions.EnableAudioVbr && state.EnableAudioVbrEncoding && vbrParam is not null) |
| | 7134 | | { |
| 0 | 7135 | | args += vbrParam; |
| | 7136 | | } |
| | 7137 | | else |
| | 7138 | | { |
| 0 | 7139 | | args += " -ab " + bitrate.Value.ToString(CultureInfo.InvariantCulture); |
| | 7140 | | } |
| | 7141 | | } |
| | 7142 | |
|
| 0 | 7143 | | if (state.OutputAudioSampleRate.HasValue) |
| | 7144 | | { |
| 0 | 7145 | | args += " -ar " + state.OutputAudioSampleRate.Value.ToString(CultureInfo.InvariantCulture); |
| | 7146 | | } |
| | 7147 | |
|
| 0 | 7148 | | args += GetAudioFilterParam(state, encodingOptions); |
| | 7149 | |
|
| 0 | 7150 | | return args; |
| | 7151 | | } |
| | 7152 | |
|
| | 7153 | | public string GetProgressiveAudioFullCommandLine(EncodingJobInfo state, EncodingOptions encodingOptions, string |
| | 7154 | | { |
| 0 | 7155 | | var audioTranscodeParams = new List<string>(); |
| | 7156 | |
|
| 0 | 7157 | | var bitrate = state.OutputAudioBitrate; |
| 0 | 7158 | | var channels = state.OutputAudioChannels; |
| 0 | 7159 | | var outputCodec = state.OutputAudioCodec; |
| | 7160 | |
|
| 0 | 7161 | | if (bitrate.HasValue && !LosslessAudioCodecs.Contains(outputCodec, StringComparison.OrdinalIgnoreCase)) |
| | 7162 | | { |
| 0 | 7163 | | var vbrParam = GetAudioVbrModeParam(GetAudioEncoder(state), bitrate.Value, channels ?? 2); |
| 0 | 7164 | | if (encodingOptions.EnableAudioVbr && state.EnableAudioVbrEncoding && vbrParam is not null) |
| | 7165 | | { |
| 0 | 7166 | | audioTranscodeParams.Add(vbrParam); |
| | 7167 | | } |
| | 7168 | | else |
| | 7169 | | { |
| 0 | 7170 | | audioTranscodeParams.Add("-ab " + bitrate.Value.ToString(CultureInfo.InvariantCulture)); |
| | 7171 | | } |
| | 7172 | | } |
| | 7173 | |
|
| 0 | 7174 | | if (channels.HasValue) |
| | 7175 | | { |
| 0 | 7176 | | audioTranscodeParams.Add("-ac " + state.OutputAudioChannels.Value.ToString(CultureInfo.InvariantCulture) |
| | 7177 | | } |
| | 7178 | |
|
| 0 | 7179 | | if (!string.IsNullOrEmpty(outputCodec)) |
| | 7180 | | { |
| 0 | 7181 | | audioTranscodeParams.Add("-acodec " + GetAudioEncoder(state)); |
| | 7182 | | } |
| | 7183 | |
|
| 0 | 7184 | | if (GetAudioEncoder(state).StartsWith("pcm_", StringComparison.Ordinal)) |
| | 7185 | | { |
| 0 | 7186 | | audioTranscodeParams.Add(string.Concat("-f ", GetAudioEncoder(state).AsSpan(4))); |
| 0 | 7187 | | audioTranscodeParams.Add("-ar " + state.BaseRequest.AudioBitRate); |
| | 7188 | | } |
| | 7189 | |
|
| 0 | 7190 | | if (!string.Equals(outputCodec, "opus", StringComparison.OrdinalIgnoreCase)) |
| | 7191 | | { |
| | 7192 | | // opus only supports specific sampling rates |
| 0 | 7193 | | var sampleRate = state.OutputAudioSampleRate; |
| 0 | 7194 | | if (sampleRate.HasValue) |
| | 7195 | | { |
| 0 | 7196 | | var sampleRateValue = sampleRate.Value switch |
| 0 | 7197 | | { |
| 0 | 7198 | | <= 8000 => 8000, |
| 0 | 7199 | | <= 12000 => 12000, |
| 0 | 7200 | | <= 16000 => 16000, |
| 0 | 7201 | | <= 24000 => 24000, |
| 0 | 7202 | | _ => 48000 |
| 0 | 7203 | | }; |
| | 7204 | |
|
| 0 | 7205 | | audioTranscodeParams.Add("-ar " + sampleRateValue.ToString(CultureInfo.InvariantCulture)); |
| | 7206 | | } |
| | 7207 | | } |
| | 7208 | |
|
| | 7209 | | // Copy the movflags from GetProgressiveVideoFullCommandLine |
| | 7210 | | // See #9248 and the associated PR for why this is needed |
| 0 | 7211 | | if (_mp4ContainerNames.Contains(state.OutputContainer)) |
| | 7212 | | { |
| 0 | 7213 | | audioTranscodeParams.Add("-movflags empty_moov+delay_moov"); |
| | 7214 | | } |
| | 7215 | |
|
| 0 | 7216 | | var threads = GetNumberOfThreads(state, encodingOptions, null); |
| | 7217 | |
|
| 0 | 7218 | | var inputModifier = GetInputModifier(state, encodingOptions, null); |
| | 7219 | |
|
| 0 | 7220 | | return string.Format( |
| 0 | 7221 | | CultureInfo.InvariantCulture, |
| 0 | 7222 | | "{0} {1}{7}{8} -threads {2}{3} {4} -id3v2_version 3 -write_id3v1 1{6} -y \"{5}\"", |
| 0 | 7223 | | inputModifier, |
| 0 | 7224 | | GetInputArgument(state, encodingOptions, null), |
| 0 | 7225 | | threads, |
| 0 | 7226 | | " -vn", |
| 0 | 7227 | | string.Join(' ', audioTranscodeParams), |
| 0 | 7228 | | outputPath, |
| 0 | 7229 | | string.Empty, |
| 0 | 7230 | | string.Empty, |
| 0 | 7231 | | string.Empty).Trim(); |
| | 7232 | | } |
| | 7233 | |
|
| | 7234 | | public static int FindIndex(IReadOnlyList<MediaStream> mediaStreams, MediaStream streamToFind) |
| | 7235 | | { |
| 0 | 7236 | | var index = 0; |
| 0 | 7237 | | var length = mediaStreams.Count; |
| | 7238 | |
|
| 0 | 7239 | | for (var i = 0; i < length; i++) |
| | 7240 | | { |
| 0 | 7241 | | var currentMediaStream = mediaStreams[i]; |
| 0 | 7242 | | if (currentMediaStream == streamToFind) |
| | 7243 | | { |
| 0 | 7244 | | return index; |
| | 7245 | | } |
| | 7246 | |
|
| 0 | 7247 | | if (string.Equals(currentMediaStream.Path, streamToFind.Path, StringComparison.Ordinal)) |
| | 7248 | | { |
| 0 | 7249 | | index++; |
| | 7250 | | } |
| | 7251 | | } |
| | 7252 | |
|
| 0 | 7253 | | return -1; |
| | 7254 | | } |
| | 7255 | |
|
| | 7256 | | public static bool IsCopyCodec(string codec) |
| | 7257 | | { |
| 0 | 7258 | | return string.Equals(codec, "copy", StringComparison.OrdinalIgnoreCase); |
| | 7259 | | } |
| | 7260 | | } |
| | 7261 | | } |