mirror/FileFlowsPlugins
1
0
Fork 0
mirror of https://github.com/revenz/FileFlowsPlugins.git synced 2026-02-18 06:48:32 -06:00
Code Issues Packages Projects Releases Wiki Activity

FF-1096 - added hardware encoding for AV1 using NVIDIA

Browse Source
This commit is contained in:
John Andrews
2023-08-28 16:34:23 +12:00
parent 80cfb8c6db
commit f24bcabade
4 changed files with 118 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
VideoNodes/FfmpegBuilderNodes/Video/FfmpegBuilderVideoEncode/AV1.cs Normal file
View File

@@ -0,0 +1,65 @@
namespace FileFlows.VideoNodes.FfmpegBuilderNodes;
public partial class FfmpegBuilderVideoEncode
{
/// <summary>
/// AV1 CPU encoding
/// </summary>
/// <param name="quality">the quality</param>
/// <param name="speed">the speed</param>
/// <returns>the encoding parameters</returns>
private static IEnumerable<string> AV1_CPU(int quality, string speed)
{
string preset = "4";
switch (speed)
{
case "ultrafast": preset = "13"; break;
case "superfast": preset = "11"; break;
case "veryfast": preset = "10"; break;
case "faster": preset = "9"; break;
case "fast": preset = "8"; break;
case "medium": preset = "6"; break;
case "slow": preset = "4"; break;
case "slower": preset = "2"; break;
case "veryslow": preset = "1"; break;
}
return new []
{
//"libaom-av1",
"libsvtav1",
"-preset", preset,
"-crf", quality.ToString()
};
}
/// <summary>
/// AV1 NVIDIA encoding
/// </summary>
/// <param name="quality">the quality</param>
/// <param name="speed">the speed</param>
/// <returns>the encoding parameters</returns>
private static IEnumerable<string> AV1_Nvidia(int quality, string speed)
{
switch (speed)
{
case "ultrafast": speed = "p1"; break;
case "superfast": speed = "p1"; break;
case "veryfast": speed = "p1"; break;
case "faster": speed = "p2"; break;
case "fast": speed = "p3"; break;
case "medium": speed = "p4"; break;
case "slow": speed = "p5"; break;
case "slower": speed = "p6"; break;
case "veryslow": speed = "p7"; break;
default: speed = "p4"; break; // unexpected
}
return new []
{
"av1_nvenc",
"-rc", "constqp",
"-qp", quality.ToString(),
"-preset", speed,
"-spatial-aq", "1"
};
}
}

41
VideoNodes/FfmpegBuilderNodes/Video/FfmpegBuilderVideoEncode/FfmpegBuilderVideoEncode.cs
View File

@@ -168,7 +168,7 @@ public partial class FfmpegBuilderVideoEncode:FfmpegBuilderNode
else if (Codec == CODEC_H265 || Codec == CODEC_H265_10BIT)
stream.EncodingParameters.AddRange(H265(args, Codec == CODEC_H265_10BIT, Quality, encoder, stream.Stream.FramesPerSecond, Speed));
else if (Codec == CODEC_AV1 || Codec == CODEC_AV1_10BIT)
stream.EncodingParameters.AddRange(AV1(args, Codec == CODEC_AV1_10BIT, Quality, Speed));
stream.EncodingParameters.AddRange(AV1(args, Codec == CODEC_AV1_10BIT, Quality, encoder, Speed));
else if(Codec == CODEC_VP9)
stream.EncodingParameters.AddRange(VP9(args, Quality, encoder, Speed));
else
@@ -188,7 +188,7 @@ public partial class FfmpegBuilderVideoEncode:FfmpegBuilderNode
if (codec == CODEC_H265 || codec == CODEC_H265_10BIT)
return H265(args, codec == CODEC_H265_10BIT, quality, encoder, fps, speed).Select(x => x.Replace("{index}", "0"));
if(codec == CODEC_AV1)
return AV1(args, codec == CODEC_AV1_10BIT, quality, speed).Select(x => x.Replace("{index}", "0"));
return AV1(args, codec == CODEC_AV1_10BIT, quality, encoder, speed).Select(x => x.Replace("{index}", "0"));
throw new Exception("Unsupported codec: " + codec);
}
@@ -270,11 +270,21 @@ public partial class FfmpegBuilderVideoEncode:FfmpegBuilderNode
}
private static IEnumerable<string> AV1(NodeParameters args, bool tenBit, int quality, string speed)
private static IEnumerable<string> AV1(NodeParameters args, bool tenBit, int quality, string encoder, string speed)
{
// hevc_qsv -load_plugin hevc_hw -pix_fmt p010le -profile:v main10 -global_quality 21 -g 24 -look_ahead 1 -look_ahead_depth 60
List<string> parameters = new List<string>();
parameters.AddRange(AV1_CPU(quality, speed));
if (encoder == ENCODER_CPU)
parameters.AddRange(AV1_CPU(quality, speed));
else if(encoder == ENCODER_NVIDIA)
parameters.AddRange(AV1_Nvidia(quality, speed));
else if (CanUseHardwareEncoding.CanProcess_Nvidia_AV1(args))
parameters.AddRange(AV1_Nvidia(quality, speed));
else
parameters.AddRange(AV1_CPU(quality, speed));
if (tenBit)
parameters.AddRange(new [] { "-pix_fmt:v:{index}", "yuv420p10le" });
return parameters;
@@ -294,29 +304,6 @@ public partial class FfmpegBuilderVideoEncode:FfmpegBuilderNode
return parameters;
}
private static IEnumerable<string> AV1_CPU(int quality, string speed)
{
string preset = "4";
switch (speed)
{
case "ultrafast": preset = "13"; break;
case "superfast": preset = "11"; break;
case "veryfast": preset = "10"; break;
case "faster": preset = "9"; break;
case "fast": preset = "8"; break;
case "medium": preset = "6"; break;
case "slow": preset = "4"; break;
case "slower": preset = "2"; break;
case "veryslow": preset = "1"; break;
}
return new []
{
//"libaom-av1",
"libsvtav1",
"-preset", preset,
"-crf", quality.ToString()
};
}
/// <summary>

7
VideoNodes/LogicalNodes/CanUseHardwareEncoding.cs
View File

@@ -99,6 +99,13 @@ public class CanUseHardwareEncoding:Node
/// <param name="args">the node parameters</param>
/// <returns>true if can use it, otherwise false</returns>
internal static bool CanProcess_Nvidia_Hevc(NodeParameters args) => CanProcess(args, "hevc_nvenc");
/// <summary>
/// Checks if this flow runner can use NVIDIA AV1 encoder
/// </summary>
/// <param name="args">the node parameters</param>
/// <returns>true if can use it, otherwise false</returns>
internal static bool CanProcess_Nvidia_AV1(NodeParameters args) => CanProcess(args, "av1_nvenc");
/// <summary>
/// Checks if this flow runner can use NVIDIA H.264 encoder

32
VideoNodes/Tests/FfmpegBuilderTests/FfmpegBuilder_VideoEncodeTests.cs
View File

@@ -167,6 +167,38 @@ public class FfmpegBuilder_VideoEncode_VideoEncodeTests: TestBase
string log = logger.ToString();
Assert.AreEqual(1, result);
}
[TestMethod]
public void FfmpegBuilder_VideoEncode_Av1()
{
var logger = new TestLogger();
string ffmpeg = FfmpegPath;
var vi = new VideoInfoHelper(ffmpeg, logger);
var vii = vi.Read(TestFile_BasicMkv);
var args = new NodeParameters(TestFile_BasicMkv, logger, false, string.Empty);
args.GetToolPathActual = (string tool) => ffmpeg;
args.TempPath = TempPath;
args.Parameters.Add("VideoInfo", vii);
FfmpegBuilderStart ffStart = new();
ffStart.PreExecute(args);
Assert.AreEqual(1, ffStart.Execute(args));
FfmpegBuilderVideoEncode ffEncode = new();
ffEncode.Quality = 28;
ffEncode.Speed = "veryslow";
//ffEncode.Encoder = "Nvid;
ffEncode.Codec = "av1 10BIT";
ffEncode.PreExecute(args);
ffEncode.Execute(args);
FfmpegBuilderExecutor ffExecutor = new();
ffExecutor.PreExecute(args);
int result = ffExecutor.Execute(args);
string log = logger.ToString();
Assert.AreEqual(1, result);
}
}
#endif