forked from vcb-s/OKEGui
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Signed-off-by: akarin <[email protected]>
- Loading branch information
Showing
8 changed files
with
219 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,143 @@ | ||
| using System; | ||
| using System.Diagnostics; | ||
| using System.IO; | ||
| using System.Text; | ||
| using System.Text.RegularExpressions; | ||
| using OKEGui.Utils; | ||
| using OKEGui.JobProcessor; | ||
| using System.Collections.Generic; | ||
|
|
||
| namespace OKEGui | ||
| { | ||
| public class SVTAV1Encoder : CommandlineVideoEncoder | ||
| { | ||
| private static readonly NLog.Logger Logger = NLog.LogManager.GetCurrentClassLogger(); | ||
| private readonly string svtav1Path = ""; | ||
| private readonly string vspipePath = ""; | ||
| private bool expectTotalFrames = false; | ||
|
|
||
| public SVTAV1Encoder(VideoJob job) : base() | ||
| { | ||
| this.job = job; | ||
| getInputProperties(job); | ||
|
|
||
| executable = Path.Combine(Environment.SystemDirectory, "cmd.exe"); | ||
|
|
||
| if (File.Exists(job.EncoderPath)) | ||
| { | ||
| this.svtav1Path = job.EncoderPath; | ||
| } | ||
|
|
||
| // 获取VSPipe路径 | ||
| this.vspipePath = Initializer.Config.vspipePath; | ||
|
|
||
| commandLine = BuildCommandline(job.EncodeParam, job.NumaNode, job.VspipeArgs); | ||
| } | ||
|
|
||
| public override void ProcessLine(string line, StreamType stream) | ||
| { | ||
| if (line.Contains("Svt[error]: ") || line.Contains("[SVT-Error]: ") || line.Contains("Error: ")) | ||
| { | ||
| Logger.Error(line); | ||
| OKETaskException ex = new OKETaskException(Constants.svtav1ErrorSmr); | ||
| ex.progress = 0.0; | ||
| ex.Data["SVTAV1_ERROR"] = line.Substring(line.IndexOf(':')+2); | ||
| throw ex; | ||
| } | ||
|
|
||
| if (line.Contains("Error: fwrite() call failed when writing frame: ")) | ||
| { | ||
| Logger.Error(line); | ||
| OKETaskException ex = new OKETaskException(Constants.svtav1CrashSmr); | ||
| throw ex; | ||
| } | ||
|
|
||
| if (line.ToLowerInvariant().Contains("all_done_encoding")) // svt-av1 must be built with -DLOG_ENC_DONE=1. | ||
| { | ||
| Logger.Debug(line); | ||
| Regex rf = new Regex("all_done_encoding *([0-9]+) frames"); | ||
|
|
||
| var result = rf.Split(line); | ||
|
|
||
| ulong reportedFrames = ulong.Parse(result[1]); | ||
|
|
||
| Debugger.Log(0, "EncodeFinish", result[1] + " frames\n"); | ||
|
|
||
| base.encodeFinish(reportedFrames); | ||
| } | ||
| if (line.StartsWith("Total Frames\t")) | ||
| { | ||
| Logger.Debug(line); | ||
| expectTotalFrames = true; | ||
| return; | ||
| } | ||
| if (expectTotalFrames) | ||
| { | ||
| Logger.Debug(line); | ||
| Regex rf = new Regex("[\t ]*([0-9]+)[\t ]"); | ||
| var result = rf.Split(line); | ||
| if (result.Length < 2) | ||
| return; | ||
| ulong reportedFrames = ulong.Parse(result[1]); | ||
| Debugger.Log(0, "EncodeFinish", result[1] + " frames\n"); | ||
| base.encodeFinish(reportedFrames); | ||
| expectTotalFrames = false; | ||
| } | ||
|
|
||
| Regex regOfficial = new Regex("Encoding frame *([0-9]+) *([0-9]+.[0-9]+) *kbps *([0-9]+.[0-9]+) *(fp[sm])", RegexOptions.IgnoreCase); | ||
|
|
||
| string[] status; | ||
|
|
||
| if (regOfficial.Split(line).Length >= 4) | ||
| { | ||
| status = regOfficial.Split(line); | ||
| } | ||
| else | ||
| { | ||
| Logger.Debug(line); | ||
| return; | ||
| } | ||
|
|
||
| if (!base.setFrameNumber(status[1], true)) | ||
| { | ||
| return; | ||
| } | ||
|
|
||
| base.setBitrate(status[2], "kb/s"); | ||
|
|
||
| if (!base.setSpeed(status[3], status[4])) | ||
| { | ||
| return; | ||
| } | ||
| } | ||
|
|
||
| private string BuildCommandline(string extractParam, int numaNode, List<string> vspipeArgs) | ||
| { | ||
| StringBuilder sb = new StringBuilder(); | ||
| sb.Append("/c \"start \"foo\" /b /wait "); | ||
| if (!Initializer.Config.singleNuma) | ||
| { | ||
| sb.Append("/affinity 0xFFFFFFFFFFFFFFFF /node "); | ||
| sb.Append(numaNode.ToString()); | ||
| } | ||
| // 构建vspipe参数 | ||
| sb.Append(" \"" + vspipePath + "\""); | ||
| sb.Append(" --y4m"); | ||
| foreach (string arg in vspipeArgs) | ||
| { | ||
| sb.Append($" --arg \"{arg}\""); | ||
| } | ||
| sb.Append(" \"" + job.Input + "\""); | ||
| sb.Append(" - |"); | ||
|
|
||
| // 构建svtav1参数 | ||
| sb.Append(" \"" + svtav1Path + "\""); | ||
| sb.Append(" --progress 2 " + extractParam + " -b"); | ||
| sb.Append(" \"" + job.Output + "\" -i -"); | ||
| sb.Append(" \""); // leave extra spaces for ApppendParameter. | ||
|
|
||
| return sb.ToString(); | ||
| } | ||
|
|
||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters