Disallow invalid samplerates

Author: WyattBlue

src/render/format.nim

@@ -25,6 +25,25 @@ proc initPriority(index: float64, frame: ptr AVFrame, stream: ptr AVStream): Pri
 
 proc `<`(a, b: Priority): bool = a.index < b.index
 
+
+proc checkAudioEncoder(encoder: ptr AVCodec, rate: cint) =
+  if encoder.sample_fmts == nil:
+    error &"{encoder.name}: No known audio formats avail."
+
+  if encoder.supported_samplerates == nil:
+    return
+
+  var allowed: seq[cint]
+  var i: cint = 0
+  let samplerates = cast[ptr UncheckedArray[cint]](encoder.supported_samplerates)
+  while samplerates[i] != 0:
+    allowed.add samplerates[i]
+    inc i
+
+  if rate notin allowed:
+    let allowedStr = allowed.join(" ")
+    error &"samplerate '{rate}' not allowed for {encoder.name} codec.\nAllowed: {allowedStr}"
+
 proc makeMedia*(args: mainArgs, tl: v3, outputPath: string, rules: Rules, bar: Bar) =
   var options: Table[string, string]
   var movFlags: seq[string] = @[]
@@ -68,9 +87,7 @@ proc makeMedia*(args: mainArgs, tl: v3, outputPath: string, rules: Rules, bar: B
       var (aOutStream, aEncCtx) = output.addStream(args.audioCodec, rate = rate,
           layout = tl.layout, metadata = {"language": "und"}.toTable)
       let encoder = aEncCtx.codec
-      if encoder.sample_fmts == nil:
-        error &"{encoder.name}: No known audio formats avail."
-
+      checkAudioEncoder(encoder, tl.sr)
       aEncCtx.open()
 
       # Update stream parameters after opening encoder for formats like AAC in MKV
@@ -98,9 +115,7 @@ proc makeMedia*(args: mainArgs, tl: v3, outputPath: string, rules: Rules, bar: B
         var (aOutStream, aEncCtx) = output.addStream(args.audioCodec, rate = rate,
             layout = tl.layout, metadata = {"language": tl.a[i].lang}.toTable)
         let encoder = aEncCtx.codec
-        if encoder.sample_fmts == nil:
-          error &"{encoder.name}: No known audio formats avail."
-
+        checkAudioEncoder(encoder, tl.sr)
         aEncCtx.open()
 
         # Update stream parameters after opening encoder for formats like AAC in MKV

src/util/rules.nim

@@ -3,23 +3,23 @@ import std/strformat
 import ../ffmpeg
 import ../log
 
-proc defaultVideoCodec*(self: ptr AVOutputFormat): string =
+func defaultVideoCodec*(self: ptr AVOutputFormat): string =
   let codecId = self.video_codec
   if codecId != AV_CODEC_ID_NONE:
     let codecName = avcodec_get_name(codecId)
     if codecName != nil:
       return $codecName
   return "none"
 
-proc defaultAudioCodec*(self: ptr AVOutputFormat): string =
+func defaultAudioCodec*(self: ptr AVOutputFormat): string =
   let codecId = self.audio_codec
   if codecId != AV_CODEC_ID_NONE:
     let codecName = avcodec_get_name(codecId)
     if codecName != nil:
       return $codecName
   return "none"
 
-proc defaultSubtitleCodec*(self: ptr AVOutputFormat): string =
+func defaultSubtitleCodec*(self: ptr AVOutputFormat): string =
   let codecId = self.subtitle_codec
   if codecId != AV_CODEC_ID_NONE:
     let codecName = avcodec_get_name(codecId)