More development

spav1an.ps1

@@ -3,13 +3,12 @@ param (
   [String] $FFmpegPath = "ffmpeg.exe",
   [String] $FFprobePath = "ffprobe.exe",
   [int] $EncoderType = 1, # Encoder type. 1 = aomenc, 2 = SVT-AV1
-  [int] $Renderer = 1, # Frame type, 1 = avisynth
+  [int] $Renderer = 1, # Frame type, 1 = ffmpeg native, 2 = avisynth
   [int] $MinChunkSec = 10,
   [int] $Threads = 0,
-  [String] $Framerate = "24000/1001",
+  [String] $Framerate = "auto",
-  [String] $EncOptions = "",
+  [String] $AnalyzeVidSize = "848x480",
-  [String] $AnalyzeVideoSize = "848x480",
+  [String] $AnalyzeVidEnc = "-hide_banner -loglevel info -i {0} -map 0:0 -vf zscale={1}:f=spline36 -vcodec libx264 -preset veryfast -crf 40 -x264-params keyint={2}:min-keyint=5:scenecut=40 -y -pix_fmt yuv420p {3}",
-  [String] $AnalyzeVideoFFmpeg = "-hide_banner -loglevel info -i {0} -map 0:0 -vf zscale={1}:f=spline36 -vcodec libx264 -preset veryfast -crf 40 -x264-params keyint={2}:min-keyint=5:scenecut=40 -y -pix_fmt yuv420p {3}",
   [String] $AvisynthScript = "LSMASHVideoSource({0})",
   [String] $EncoderOptions = "--cpu-used=8 --passes=1 --cq-level=10 --rt -o {0} -",
   [parameter(Position=0)] [String] $InputFile = ""
@@ -43,18 +42,39 @@ Write-Host ""
 function Write-OurHelp {
   Write-Host ""
   Write-Host "S. Powershell variant of av1an"
-  Write-Host "Usage: .\spav1an.ps1 [options] [INPUT_FILE]"
+  Write-Host "Usage: .\spav1an.ps1 [options] INPUT_FILE"
   Write-Host ""
   Write-Host "Main options:"
+  Write-Host "  -EncoderOptions <str>      The encoding options for the encoder"
   Write-Host "  -EncoderType    1,2        Select which encoder type we are using:       "
   Write-Host "                             1 = aomenc encoder"
   Write-Host "                             2 = SVT-AV1 encoder"
-  Write-Host "  -Renderer    1       Select which frame renderer we are gonna be using."
+  Write-Host "  -Renderer       1,2        Select which frame renderer we are gonna be"
-  Write-Host "                        1 = avisynth"
+  Write-Host "                             using for splitting the file:"
-  Write-Host "  -Framerate rate/scale Frame rate of the video"
+  Write-Host "                             1 = ffmpeg (uses -ss)"
+  Write-Host "                             2 = avisynth"
+  Write-Host "  -MinChunkSec    10         The minimum amount of seconds each chunk"
+  Write-Host "                             encode shouldbe at least."
+  Write-Host "  -Threads        0          Total amount of workers to use. Default 0"
+  Write-Host "                             will use the total of logical processors"
+  Write-Host "  -Framerate      auto       Frame rate of the video. Auto will use ffprobe"
+  Write-Host "                             to autodect. Otherwise you can specify it using"
+  Write-Host "                             rate/scale"
   Write-Host ""
+  Write-Host "Analyze options:"
+  Write-Host "The analyze file will be used to generate scenecut changes as well as other"
+  Write-Host "keyframe data that will be used for splitting the video into chunks."
+  Write-Host "  -AnalyzeVidSize widxhei    The width and height of the analyze video"
+  Write-Host "                             Smaller size might be more innacurate but"
+  Write-Host "                             will be faster to encode"
+  Write-Host "  -AnalyzeVidEnc  <str>      The encoding options for the analyze video."
+  Write-Host "                             By default this is x264 veryfast crf 40"
 }
-
+<#
+-------------------------------------------------------------------
+--------------- Verify all the parameters look okay  --------------
+-------------------------------------------------------------------
+#>
 function Assert-OurParameters {
   try {
     $gobble = Start-ThreadJob -ScriptBlock {}
@@ -66,6 +86,8 @@ function Assert-OurParameters {
   $writehelp = $false
 
   if ($inputfile -eq "") {
+    Write-Host "Error: Inputfile is missing"
+    Write-Host ""
     $writehelp = $true
   }
   if ($InputFile -notmatch '[^.]+\..+') {
@@ -73,21 +95,31 @@ function Assert-OurParameters {
     Write-Host ""
     $writehelp = $true
   }
-  if ($Framerate -notmatch '\d+/\d+') {
+  if ($Framerate -notmatch '\d+/\d+' -and $Framerate -ne 'auto') {
-    Write-Host "Error: -Framerate has to be in a format of <rate>/<scale>"
+    Write-Host "Error: -Framerate has to be in a format of <rate>/<scale> or auto"
     Write-Host "       Example: -Framerate 24000/1001"
     Write-Host "       Example: -Framerate 25/1"
+    Write-Host "       Example: -Framerate auto"
     Write-Host ""
     $writehelp = $true
   }
-  if ($AnalyzeVideoSize -notmatch '\d+x\d+') {
+  if ($AnalyzeVidSize -notmatch '\d+x\d+') {
     Write-Host "Error: -AnalyzeVideoSize has to be in a format of <width>x<height>"
     Write-Host "       Example: -AnalyzeVideoSize 1280x720"
     Write-Host "       Example: -AnalyzeVideoSize 848x480"
     Write-Host ""
     $writehelp = $true
   }
-  if (($EncoderType -ne 1 -and $EncoderType -ne 2) -or $Renderer -ne 1) {
+  if ($EncoderType -ne 1 -and $EncoderType -ne 2) {
+    Write-Host "Error: Unknown encoder type of $EncoderType."
+    Write-Host "Only values of 1 and 2 are supported yet."
+    Write-Host ""
+    $writehelp = $true
+  }
+  if ($Renderer -ne 1 -and $Renderer -ne 2) {
+    Write-Host "Error: Unknown renderer type of $Renderer."
+    Write-Host "Only values of 1 and 2 are supported yet."
+    Write-Host ""
     $writehelp = $true
   }
 
@@ -97,40 +129,84 @@ function Assert-OurParameters {
   }
 }
 
-function Write-AnalyzeFile {
+function Rename-ArrayIdentifiers([String[]] $InputArr, [String] $Identifier1, [String] $ReplaceWith1, [String] $Identifier2, [String] $ReplaceWith2) {
+  for ($i = 0; $i -le $InputArr.Length; $i++) {
+    if ($InputArr[$i] -match $Identifier1) {
+      $InputArr[$i] = $InputArr[$i] -replace $Identifier1, $ReplaceWith1
+    } elseif ($Identifier2 -and $InputArr[$i] -match $Identifier2) {
+      $InputArr[$i] = $InputArr[$i] -replace $Identifier2, $ReplaceWith2
+    }
+  }
+  return $InputArr
+}
+
+<#
+-------------------------------------------------------------------
+------- Create our project files  --------
+-------------------------------------------------------------------
+#>
+function Write-ProjectFile {
+  # identifier for text replacement 1
   $identifier1 = '2Nhd6VgWZr69B4kzAStydVGCcVugprCpJR'
+  # identifier for text replacement 2
   $identifier2 = 'jh2Acz3nR3XjKrzqPiE3ZBANYYRtEHeqr4'
 
-  $splitsize = $AnalyzeVideoSize.Split('x')
+  Write-Host "Project file was not found. Creating a project file."
-  $vfsize = "w=" + $splitsize[0] + ":h=" + $splitsize[1]
+  $analyzefilename = $workproject + '_analyze.mkv'
-  $AnalyzeVideoFFmpeg = $AnalyzeVideoFFmpeg -f $identifier1, $vfsize, $maxkeyint, $identifier2
-  $analyzeffmpegSplitted = $AnalyzeVideoFFmpeg.Split(' ')
 
-  for ($i = 0; $i -le $analyzeffmpegSplitted.Length; $i++) {
+  # Check if framerate is auto and grab it from the source file
-    if ($analyzeffmpegSplitted[$i] -match $identifier1) {
+  if ($Framerate -eq 'auto') {
-      $analyzeffmpegSplitted[$i] = $analyzeffmpegSplitted[$i] -replace $identifier1, $InputFile
+    $fpsPropeArgs = "-v error -select_streams v -of default=noprint_wrappers=1:nokey=1 -show_entries stream=r_frame_rate $identifier1".Split(' ')
-    } elseif ($analyzeffmpegSplitted[$i] -match $identifier2) {
+    $fpsPropeArgs = Rename-ArrayIdentifiers $fpsPropeArgs $identifier1 $InputFile
-      $analyzeffmpegSplitted[$i] = $analyzeffmpegSplitted[$i] -replace $identifier2, $analyzefilename
+    $Framerate = Invoke-NativeCommand -FilePath $FFprobePath -ArgumentList $fpsPropeArgs | Receive-RawPipeline
-    }
   }
 
+  # Split the framerate (rate/scale) to their respective values and calculate
+  # the max keyint from that
+  $FramerateSplit = $Framerate.Split('/')
+  $fps = [convert]::ToDouble($FramerateSplit[0]) / [convert]::ToDouble($FramerateSplit[1])
+  $maxkeyint = [int][Math]::Ceiling($fps * 10)
+
+  # Split the video size string of 'widthXheight'
+  $splitsize = $AnalyzeVidSize.Split('x')
+  $vfsize = "w=" + $splitsize[0] + ":h=" + $splitsize[1]
+
+  # Create our command line
+  # Reason to use identifier isntead of using filename directly is cause we
+  # wanna split by space to get array argument list and that might split
+  # the file pathname.
+  $AnalyzeVidEnc = $AnalyzeVidEnc -f $identifier1, $vfsize, $maxkeyint, $identifier2
+  
+  $analyzeffmpegSplitted = Rename-ArrayIdentifiers $AnalyzeVidEnc.Split(' ') $identifier1 $InputFile $identifier2 $analyzefilename
+
   Write-Host "Creating $analyzefilename to generate keyframe split data from"
   Write-Host ""
   Write-Host "$FFmpegPath $analyzeffmpegSplitted"
-
-  Invoke-NativeCommand -FilePath $FFmpegPath -ArgumentList $analyzeffmpegSplitted | Receive-RawPipeline
   Write-Host ""
-}
 
-function Write-ProjectFile {
+  # Create our analyze file
+  # Reason we use Invoke-NativeCommand is "ffmpeg.exe" if located in the same
+  # folder as script will make the Invoke-Expression fail as it requires .\ in
+  # front of it. Invoke-NativeCommand allows us to specify filepath and it will
+  # run regardless
+  # Invoke-NativeCommand -FilePath $FFmpegPath -ArgumentList $analyzeffmpegSplitted | Receive-RawPipeline
+  Write-Host ""
+
   Write-Host "Reading keyframe data from $analyzefilename"
   Write-Host ""
   $analyzekeyframefilename = $workproject + '_kf.txt'
-  Invoke-Expression ".\$FFprobePath -hide_banner -show_entries packet=pts_time,flags -select_streams v:0 -of csv=print_section=0 $analyzefilename > $analyzekeyframefilename"
+  
-  Write-Host ""
+  # Start probing and analyzing our analyze file
-  $videodata = Get-Content $analyzekeyframefilename
+  $analyzePropeArgs = "-hide_banner -show_entries packet=pts_time,flags -select_streams v:0 -of csv=print_section=0 $identifier1".Split(' ')
+  $analyzePropeArgs = Rename-ArrayIdentifiers $analyzePropeArgs $identifier1 $InputFile
+  $videodata = Invoke-NativeCommand -FilePath $FFprobePath -ArgumentList $analyzePropeArgs | Receive-RawPipeline
+  
+  # Invoke-Expression ".\$FFprobePath -hide_banner -show_entries packet=pts_time,flags -select_streams v:0 -of csv=print_section=0 $analyzefilename > $analyzekeyframefilename"
+  # Write-Host ""
+  
   $project = New-Object 'System.Collections.Generic.List[KeyFrameData]'
   $keyframes = New-Object 'System.Collections.Generic.List[KeyFrameData]'
+  
   for ($i = 0; $i -lt $videodata.Length; $i++) {
     $split = $videodata[$i].Split(',')
     if ($split.Count -gt 1 -and $split[1][0] -eq 'K') {
@@ -138,6 +214,7 @@ function Write-ProjectFile {
     }
   }
 
+  # If there are too few frames, the user might as well just encode directly
   if ($keyframes.Count -lt 2) {
     Write-Host "Too few keyframes found, exiting"
     exit
@@ -170,15 +247,23 @@ function Write-ProjectFile {
   $start.end = $videodata.Length - 1
   $project.Add($start)
   
-  $projectoutput = '['
+  $projectoutput = '{'
+  $projectoutput += "`n  `"fps`":`"" + $Framerate + "`","
+  $projectoutput += "`n  `"chunks`": ["
   $projectoutput += "`n    [" + $project[0].start + ", " + $project[0].end + ", " + $project[0].startstamp + "]"
   for ($i = 1; $i -lt $project.Count; $i++) {
     $projectoutput += ",`n    [" + $project[$i].start + ", " + $project[$i].end + ", " + $project[$i].startstamp + "]"
   }
-  $projectoutput += "`n]"
+  $projectoutput += "`n  ]"
+  $projectoutput += "`n}"
   Set-Content -Path $projectfile $projectoutput
 }
 
+<#
+-------------------------------------------------------------------
+----------------------- Our encoding worker -----------------------
+-------------------------------------------------------------------
+#>
 function Start-OurEncoding {
   Write-Host "Starting our chunk encoder with maximum $Threads workers"
   Write-Host ""
@@ -217,24 +302,21 @@ function Start-OurEncoding {
       if ($null -eq $activeworkers[$x].job) {
         $chunkfile = $encodefile + '_' + $project[$i].start + '-' + $project[$i].end + '.ivf'
 
-        # Old avisynth version
+        if ($Renderer -eq 2) {
-        <#
           $trimcontent = $AvisynthScript -f ('"' + $InputFile + '"')
-        if ($Renderer -eq 1) {
           $trimcontent += "`nTrim(" + $project[$i].start + ", " + $project[$i].end + ")"
-        }
           $chunkavs = $encodefile + '_' + $project[$i].start + '-' + $project[$i].end + '.avs'
           Set-Content -Path $chunkavs $trimcontent
   
           $ffmpegreadchunkparameters = @('-hide_banner', '-loglevel', 'warning', '-i', $chunkavs, '-map', '0:0', '-strict', '-1', '-f', 'yuv4mpegpipe', '-')
-        #>
+        } elseif ($Renderer -eq 1) {
-        
           $sshours = [Math]::Floor($project[$i].startstamp / 3600)
           $ssminutes = [Math]::Floor(($project[$i].startstamp % 3600) / 60)
           $ssseconds = $project[$i].startstamp % 60
           $ssstart = "{0:00}:{1:00}:{2:00.000}" -f $sshours, $ssminutes, $ssseconds
 
           $ffmpegreadchunkparameters = @('-hide_banner', '-loglevel', 'warning', '-ss', $ssstart, '-i', $InputFile, '-map', '0:0', '-vframes', ($project[$i].end - $project[$i].start + 1), '-strict', '-1', '-f', 'yuv4mpegpipe', '-')
+        }
 
         $identifier1 = '5BdMoB52CQyrNYEof3gARsbLgsTjZCdqF9'
         $EncoderParameters = $EncoderOptions -f $identifier1, ($project[$i].end - $project[$i].start + 1)
@@ -319,27 +401,29 @@ function Start-OurEncoding {
   Write-Host "Finished"
 }
 
-Assert-OurParameters
+try {
+  Assert-OurParameters
 
-if ($Threads -lt 1) {
+  if ($Threads -lt 1) {
     $Threads = (Get-CimInstance Win32_ComputerSystem).NumberOfLogicalProcessors
-}
+  }
 
-$FramerateSplit = $Framerate.Split('/')
+  $workproject = [io.path]::GetFileNameWithoutExtension($InputFile)
-$fps = [convert]::ToDouble($FramerateSplit[0]) / [convert]::ToDouble($FramerateSplit[1])
+  $projectfile = $workproject + '_spav1an.json'
-$workproject = [io.path]::GetFileNameWithoutExtension($InputFile)
-$projectfile = $workproject + '_spav1an.json'
 
-if (!(Test-Path $projectfile)) {
+  if (!(Test-Path $projectfile)) {
-  Write-Host "Project file was not found. Creating a project file."
-  $analyzefilename = $workproject + '_analyze.mkv'
-  $maxkeyint = [int][Math]::Ceiling($fps * 10)
-
-  # Write-AnalyzeFile
     Write-ProjectFile
+  }
+}
+catch {
+  Write-Host ""
+  Write-Host "FATAL EXCEPTION OCCURED:"
+  Write-Host ""
+  Write-Host $_
+  Write-Host $_.ScriptStackTrace
 }
 
-Start-OurEncoding
+# Start-OurEncoding
 
 # .\ffmpeg.exe -hide_banner -i gi_joe_720p_275.webm -map 0:0 -vf "zscale=w=1280:h=720:f=spline36" -pix_fmt yuv420p -strict -1 -f yuv4mpegpipe - | .\aomenc-3.0.0-335.exe --cpu-used=6 --cq-level=20 --bit-depth=8 -o gi_joe_test.webm -