[ffmpeg-vaapi] add b_depth test options for h265 encode

Signed-off-by: Wang Hangjie <[email protected]>
Hangjie22Coder · Sep 6, 2024 · 4e88010 · 4e88010
1 parent a88d221
commit 4e88010
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/lib/parameters.py b/lib/parameters.py
@@ -694,6 +694,8 @@ def gen_hevc_seek_lp_parameters(spec, profiles):
 gen_hevc_cqp_parameters = gen_avc_cqp_parameters
 gen_hevc_cbr_parameters = gen_avc_cbr_parameters
 gen_hevc_vbr_parameters = gen_avc_vbr_parameters
+gen_hevc_bdepth_parameters = gen_avc_bdepth_parameters
+gen_hevc_bdepth_lp_parameters = gen_avc_bdepth_lp_parameters
 gen_hevc_roi_parameters = gen_avc_roi_parameters
 gen_hevc_roi_lp_parameters = gen_avc_roi_lp_parameters
 gen_hevc_forced_idr_parameters = gen_avc_forced_idr_parameters

diff --git a/test/ffmpeg-vaapi/encode/hevc.py b/test/ffmpeg-vaapi/encode/hevc.py
@@ -273,3 +273,45 @@ def init(self, tspec, case, gop, bframes, bitrate, maxrate, profile, rcmode):
   def test(self, case, gop, bframes, bitrate, maxrate, profile, rcmode):
     self.init(spec, case, gop, bframes, bitrate, maxrate, profile, rcmode)
     self.encode()
+
+class bdepth(HEVC8EncoderTest):
+  def init(self, tspec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, bdepth):
+    vars(self).update(tspec[case].copy())
+    vars(self).update(
+      rcmode    = rcmode,
+      bframes   = bframes,
+      bitrate   = bitrate,
+      qp        = qp,
+      case      = case,
+      maxrate   = maxrate,
+      minrate   = bitrate,
+      gop       = gop,
+      profile   = profile,
+      bdepth    = bdepth,
+    )
+
+  @slash.parametrize(*gen_hevc_bdepth_parameters(spec, ['high', 'main']))
+  def test(self, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, bdepth):
+    self.init(spec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, bdepth)
+    self.encode()
+
+class bdepth_lp(HEVC8EncoderLPTest):
+  def init(self, tspec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, bdepth):
+    vars(self).update(tspec[case].copy())
+    vars(self).update(
+      rcmode    = rcmode,
+      bframes   = bframes,
+      bitrate   = bitrate,
+      qp        = qp,
+      case      = case,
+      maxrate   = maxrate,
+      minrate   = bitrate,
+      gop       = gop,
+      profile   = profile,
+      bdepth    = bdepth,
+    )
+
+  @slash.parametrize(*gen_hevc_bdepth_lp_parameters(spec, ['high', 'main']))
+  def test(self, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, bdepth):
+    self.init(spec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, bdepth)
+    self.encode()