Av1an: Automatic HDR args for aomenc

Author: n00mkrad

ff-utils-winforms/Data/Codecs/Video/AomAv1.cs

@@ -1,5 +1,6 @@
 using Nmkoder.Extensions;
 using Nmkoder.IO;
+using Nmkoder.Utils;
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -39,12 +40,24 @@ public CodecArgs GetArgs(Dictionary<string, string> encArgs = null, MediaFile me
             string denoise = encArgs.ContainsKey("grainSynthDenoise") ? (encArgs["grainSynthDenoise"].GetBool() ? "1" : "0") : "0";
             string tiles = CodecUtils.GetTilingArgs(mediaFile.VideoStreams.FirstOrDefault().Resolution, "--tile-rows=", "--tile-columns=");
             string cust = encArgs.ContainsKey("custom") ? encArgs["custom"] : "";
+            string colors = "";
+
+            if(mediaFile != null && mediaFile.ColorData != null)
+            {
+                string prims = ColorDataUtils.GetColorPrimariesString(mediaFile.ColorData.ColorPrimaries);
+                string transfer = ColorDataUtils.GetColorTransferString(mediaFile.ColorData.ColorTransfer);
+                string matrixCoeffs = ColorDataUtils.GetColorMatrixCoeffsString(mediaFile.ColorData.ColorMatrixCoeffs);
+                colors = $"{(prims != "" ? $"--color-primaries={prims}" : "")} {(transfer != "" ? $"--transfer-characteristics={transfer}" : "")} {(matrixCoeffs != "" ? $"--matrix-coefficients={matrixCoeffs}" : "")}";
+
+                if (mediaFile.ColorData.ColorPrimaries == 9)
+                    colors += " --deltaq-mode=5 --enable-chroma-deltaq=1";
+            }
 
             return new CodecArgs($" -e aom -v \" " +
                 $"--end-usage=q --cpu-used={preset} --cq-level={q} " +
-                $"--kf-min-dist=12 --kf-max-dist={g} " +
+                $"--disable-kf --kf-min-dist=12 --kf-max-dist={g} " +
                 $"--enable-dnl-denoising={denoise} --denoise-noise-level={grain} " +
-                $"--enable-keyframe-filtering=0 --threads={thr} {tiles} {cust} \" --pix-format {pixFmt}");
+                $"--enable-keyframe-filtering=0 {colors} --threads={thr} {tiles} {cust} \" --pix-format {pixFmt}");
         }
     }
 }

ff-utils-winforms/Data/MediaFile.cs

@@ -37,6 +37,7 @@ public class MediaFile
         public List<SubtitleStream> SubtitleStreams = new List<SubtitleStream>();
         public List<DataStream> DataStreams = new List<DataStream>();
         public List<AttachmentStream> AttachmentStreams = new List<AttachmentStream>();
+        public VideoColorData ColorData = null;
         public long CreationTime;
         public bool Initialized = false;
 

ff-utils-winforms/UI/Tasks/Av1an.cs

@@ -12,7 +12,9 @@
 using Nmkoder.Extensions;
 using Nmkoder.Forms;
 using Nmkoder.IO;
+using Nmkoder.Main;
 using Nmkoder.Media;
+using Nmkoder.Utils;
 using static Nmkoder.UI.Tasks.Av1anUi;
 
 namespace Nmkoder.UI.Tasks
@@ -48,6 +50,12 @@ public static async Task RunResumeWithNewArgs(string sourceFile, string override
 
         public static async Task Run(bool resume = false, string overrideTempDir = "", string overrideArgs = "")
         {
+            if (TrackList.current.File.IsDirectory)
+            {
+                RunTask.Cancel("Av1an cannot use image sequence inputs!");
+                return;
+            }
+
             Program.mainForm.SetWorking(true);
             string args = "";
             string inPath = "";
@@ -66,6 +74,7 @@ public static async Task Run(bool resume = false, string overrideTempDir = "", s
                     outPath = GetOutPath();
                     string cust = Program.mainForm.av1anCustomArgsBox.Text.Trim();
                     string custEnc = Program.mainForm.av1anCustomEncArgsBox.Text.Trim();
+                    TrackList.current.File.ColorData = await ColorDataUtils.GetColorData(TrackList.current.File.SourcePath);
                     CodecArgs codecArgs = CodecUtils.GetCodec(vCodec).GetArgs(GetVideoArgsFromUi(), TrackList.current.File, Data.Codecs.Pass.OneOfOne);
                     string v = codecArgs.Arguments;
                     string vf = await GetVideoFilterArgs(codecArgs);

ff-utils-winforms/Utils/ColorDataUtils.cs

@@ -258,6 +258,82 @@ public static int GetColorRange(string s) // Defined by the "Matrix coefficients
             return 0; // Fallback: Unspecified
         }
 
+        #region Get string from int
+
+        public static string GetColorPrimariesString(int n)
+        {
+            switch (n)
+            {
+                case 1: return "bt709";
+                case 4: return "bt470m";
+                case 5: return "bt470bg";
+                case 6: return "bt610";
+                case 7: return "smpte240m";
+                case 8: return "film";
+                case 9: return "bt2020";
+                case 10: return "smpte428";
+                case 11: return "smpte431";
+                case 12: return "smpte432";
+            }
+
+            return "";
+        }
+
+        public static string GetColorTransferString(int n)
+        {
+            switch (n)
+            {
+                case 1: return "bt709";
+                case 4: return "gamma22"; // "bt470m"
+                case 5: return "gamma28"; // "bt470bg"
+                case 6: return "bt610"; // "smpte170m"
+                case 7: return "smpte240m";
+                case 8: return "linear";
+                case 11: return "iec61966-2-4";
+                case 12: return "bt1361";
+                case 13: return "srgb";
+                case 14: return "bt2020-10";
+                case 15: return "bt2020-12";
+                case 16: return "smpte2084";
+                case 17: return "smpte428";
+                case 18: return "bt2100";
+            }
+
+            return "";
+        }
+
+        public static string GetColorMatrixCoeffsString(int n)
+        {
+            switch (n)
+            {
+                case 1: return "bt709";
+                case 4: return "fcc";
+                case 5: return "bt470bg";
+                case 6: return "bt610";
+                case 7: return "smpte240m";
+                case 8: return "ycgco";
+                case 9: return "bt2020ncl";
+                case 10: return "bt2020";
+            }
+
+            return "";
+        }
+
+        public static string GetColorRangeString(int n)
+        {
+            switch (n)
+            {
+                case 1: return "tv";
+                case 2: return "pc";
+            }
+
+            return "";
+        }
+
+        #endregion
+
+        #region Get friendly name from int
+
         public static string GetColorPrimariesName(int n)
         {
             switch (n)
@@ -338,5 +414,7 @@ public static string GetColorRangeName(int n)
 
             return "Unknown";
         }
+
+        #endregion
     }
 }