新增H264的读取视频主要编码设置并应用到当前转码设置的功能

autodotua · May 28, 2023 · a479ca0 · a479ca0
1 parent 6de3014
commit a479ca0
Show file tree

Hide file tree

Showing 3 changed files with 82 additions and 39 deletions.
diff --git a/SimpleFFmpegGUI.Core/FFmpegLib/VideoCodec.cs b/SimpleFFmpegGUI.Core/FFmpegLib/VideoCodec.cs
@@ -31,7 +31,7 @@ public abstract class VideoCodec : CodecBase
         public static VideoCodec GetCodec(string name)
         {
             name2codec ??= VideoCodecs.ToDictionary(p => p.Name, p => p);
-            return name2codec.GetValueOrDefault(name);
+            return name == null ? null : name2codec.GetValueOrDefault(name);
         }
 
 

diff --git a/SimpleFFmpegGUI.Core/Manager/MediaInfoManager.cs b/SimpleFFmpegGUI.Core/Manager/MediaInfoManager.cs
@@ -2,6 +2,7 @@
 using Mapster;
 using Newtonsoft.Json.Linq;
 using SimpleFFmpegGUI.Dto;
+using SimpleFFmpegGUI.FFmpegLib;
 using SimpleFFmpegGUI.Model;
 using SimpleFFmpegGUI.Model.MediaInfo;
 using System;
@@ -142,19 +143,26 @@ private static MediaInfoGeneral ParseMediaInfoJSON(JObject json)
         public static VideoCodeArguments ConvertToVideoArguments(MediaInfoGeneral mediaInfo)
         {
             VideoCodeArguments arguments = new VideoCodeArguments();
+
             var tracks = JObject.Parse(mediaInfo.Raw)["media"]["track"] as JArray;
             if (mediaInfo.Videos.Count == 0)
             {
                 throw new Exception("源文件不含视频");
             }
             var video = mediaInfo.Videos[0];
 
+            arguments.Code = video.Format switch
+            {
+                "AVC" => VideoCodec.X264.Name,
+                "HEVC" => VideoCodec.X265.Name,
+                _ => throw new Exception("仅支持H264或H265")
+            };
+
             if (video.EncodingSettings != null && video.EncodingSettings.Count > 0)
             {
                 var settings = video.EncodingSettings.ToDictionary(p => p.Name, p => p.Value);
                 try
                 {
-                    arguments = new VideoCodeArguments();
                     if (settings["rc"].Equals("crf"))
                     {
                         if (settings.ContainsKey("crf"))
@@ -182,37 +190,76 @@ public static VideoCodeArguments ConvertToVideoArguments(MediaInfoGeneral mediaI
                 }
 
                 int preset = 0;
-                if (settings.ContainsKey("no-signhide"))
-                {
-                    preset = 8;
-                }
-                else if (settings.ContainsKey("no-sao"))
-                {
-                    preset = 7;
-                }
-                else if (settings["subme"].Equals(1))
-                {
-                    preset = 6;
-                }
-                else if (settings["ref"].Equals(2))
-                {
-                    preset = 5;
-                }
-                else if (settings["max-merge"].Equals(2))
-                {
-                    preset = 4;
-                }
-                else if (settings["ref"].Equals(3))
-                {
-                    preset = 3;
-                }
-                else if (settings["ref"].Equals(4))
+
+                if (arguments.Code == VideoCodec.X264.Name)
                 {
-                    preset = 2;
+                    if (settings["cabac"].Equals(0))
+                    {
+                        preset = 8;
+                    }
+                    else if (settings["subme"].Equals(1))
+                    {
+                        preset = 7;
+                    }
+                    else if (settings["subme"].Equals(2))
+                    {
+                        preset = 6;
+                    }
+                    else if (settings["subme"].Equals(4))
+                    {
+                        preset = 5;
+                    }
+                    else if (settings["subme"].Equals(6))
+                    {
+                        preset = 4;
+                    }
+                    else if (settings["subme"].Equals(7))
+                    {
+                        preset = 3;
+                    }
+                    else if (settings["subme"].Equals(8))
+                    {
+                        preset = 2;
+                    }
+                    else if (settings["subme"].Equals(9))
+                    {
+                        preset = 1;
+                    }
                 }
-                else if (settings["max-merge"].Equals(4))
+                else if (arguments.Code == VideoCodec.X265.Name)
                 {
-                    preset = 1;
+                    if (settings.ContainsKey("no-signhide"))
+                    {
+                        preset = 8;
+                    }
+                    else if (settings.ContainsKey("no-sao"))
+                    {
+                        preset = 7;
+                    }
+                    else if (settings["subme"].Equals(1))
+                    {
+                        preset = 6;
+                    }
+                    else if (settings["ref"].Equals(2))
+                    {
+                        preset = 5;
+                    }
+                    else if (settings["max-merge"].Equals(2))
+                    {
+                        preset = 4;
+                    }
+                    else if (settings["ref"].Equals(3))
+                    {
+                        preset = 3;
+                    }
+                    else if (settings["ref"].Equals(4))
+                    {
+                        preset = 2;
+                    }
+                    else if (settings["max-merge"].Equals(4))
+                    {
+                        preset = 1;
+                    }
                 }
                 arguments.Preset = preset;
             }
@@ -221,14 +268,6 @@ public static VideoCodeArguments ConvertToVideoArguments(MediaInfoGeneral mediaI
                 throw new Exception("源视频未提供编码设置信息，无法转换为输出参数");
             }
 
-            arguments.Code = video.Format switch
-            {
-                "AVC" => "H264",
-                "HEVC" => "H265",
-                _ => video.Format
-            };
-
-
             return arguments;
         }
     }

diff --git a/日志.md b/日志.md
@@ -647,4 +647,8 @@
 
 ## 20230524
 
-【Core】【WPF】基本实现了读取视频主要编码设置并应用到当前转码设置的功能：通过将文件拖放到视频编码设置上方。
+【Core】【WPF】基本实现了读取H265视频主要编码设置并应用到当前转码设置的功能：通过将文件拖放到视频编码设置上方。
+
+## 20230528
+
+【Core】新增H264的读取视频主要编码设置并应用到当前转码设置的功能