[ffmpeg-qsv] add encode/hevc playback_seek test case

Signed-off-by: Wang Hangjie <[email protected]>

squash_commit

Author: Hangjie22Coder

lib/ffmpeg/encoderbase.py

@@ -65,6 +65,7 @@ def hwformat(self):
   pict          = property(lambda s: s.ifprop("vpict", " -pic_timing_sei 0"))
   roi           = property(lambda s: s.ifprop("roi", ",addroi=0:0:{width}/2:{height}/2:-1/3"))
   strict        = property(lambda s: s.ifprop("strict", " -strict {strict}"))
+  seek          = property(lambda s: s.ifprop("seek", " -ss {seek}"))
   hwupload      = property(lambda s: ",hwupload")
 
   @property
@@ -109,7 +110,7 @@ def encparams(self):
       f"{self.bframes}{self.slices}{self.minrate}{self.maxrate}{self.refs}"
       f"{self.extbrc}{self.loopshp}{self.looplvl}{self.tilecols}{self.tilerows}"
       f"{self.level}{self.ladepth}{self.forced_idr}{self.intref}{self.lowpower}"
-      f"{self.maxframesize}{self.pict}{self.rqp}{self.strict}"
+      f"{self.maxframesize}{self.pict}{self.rqp}{self.strict}{self.seek}"
     )
 
   @property

lib/parameters.py

@@ -516,6 +516,70 @@ def gen_hevc_pict_lp_parameters(spec, profiles):
   params = gen_hevc_pict_lp_variants(spec, profiles)
   return keys, params
 
+def gen_hevc_seek_variants(spec, profiles):
+  for case, params in spec.items():
+    for variant in copy.deepcopy(params.get("variants", dict()).get("seek", [])):
+      uprofile = variant.get("profile", None)
+      cprofiles = [uprofile] if uprofile else profiles
+      gop     = variant.get("gop", None)
+      bframes = variant.get("bframes", None)
+      bitrate = variant.get("bitrate", None)
+      qp      = variant.get("qp", None)
+      fps     = variant.get("fps", 25)
+      rcmode  = variant["rcmode"]
+      seek    = variant["seek"]
+      if "cqp" == rcmode:
+        assert bitrate == None, "We shouldn't set a value to bitrate for CQP."
+        variant.update(maxrate = None)
+      else:
+        assert qp == None, "We shouldn't set a value to qp for CBR/VBR."
+        if "cbr" == rcmode:
+          variant.update(maxrate = bitrate)
+        elif "vbr" == rcmode:
+          variant.update(maxrate = bitrate * 2)
+      for profile in cprofiles:
+        yield [
+          case, gop, bframes, bitrate, qp, variant["maxrate"],
+          profile, rcmode, seek, fps,
+        ]
+
+def gen_hevc_seek_parameters(spec, profiles):
+  keys = ("case", "gop", "bframes", "bitrate", "qp", "maxrate", "profile", "rcmode", "seek", "fps")
+  params = gen_hevc_seek_variants(spec, profiles)
+  return keys, params
+
+def gen_hevc_seek_lp_variants(spec, profiles):
+  for case, params in spec.items():
+    for variant in copy.deepcopy(params.get("variants", dict()).get("seek_lp", [])):
+      uprofile = variant.get("profile", None)
+      cprofiles = [uprofile] if uprofile else profiles
+      gop     = variant.get("gop", None)
+      bframes = variant.get("bframes", None)
+      bitrate = variant.get("bitrate", None)
+      qp      = variant.get("qp", None)
+      fps     = variant.get("fps", 25)
+      rcmode  = variant["rcmode"]
+      seek    = variant["seek"]
+      if "cqp" == rcmode:
+        assert bitrate == None, "We shouldn't set a value to bitrate for CQP."
+        variant.update(maxrate = None)
+      else:
+        assert qp == None, "We shouldn't set a value to qp for CBR/VBR."
+        if "cbr" == rcmode:
+          variant.update(maxrate = bitrate)
+        elif "vbr" == rcmode:
+          variant.update(maxrate = bitrate * 2)
+      for profile in cprofiles:
+        yield [
+          case, gop, bframes, bitrate, qp, variant["maxrate"],
+          profile, rcmode, seek, fps,
+        ]
+
+def gen_hevc_seek_lp_parameters(spec, profiles):
+  keys = ("case", "gop", "bframes", "bitrate", "qp", "maxrate", "profile", "rcmode", "seek", "fps")
+  params = gen_hevc_seek_lp_variants(spec, profiles)
+  return keys, params
+
 gen_hevc_cqp_parameters = gen_avc_cqp_parameters
 gen_hevc_cbr_parameters = gen_avc_cbr_parameters
 gen_hevc_vbr_parameters = gen_avc_vbr_parameters

test/ffmpeg-qsv/encode/hevc.py

@@ -363,3 +363,46 @@ def test(self, case, gop, bframes, bitrate, maxrate, profile, rcmode):
     self.init(spec, case, gop, bframes, bitrate, maxrate, profile, rcmode)
     self.encode()
 
+class seek(HEVC8EncoderTest):
+  def init(self, tspec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, seek, fps):
+    vars(self).update(tspec[case].copy())
+    vars(self).update(
+      rcmode     = rcmode,
+      bframes    = bframes,
+      bitrate    = bitrate,
+      case       = case,
+      maxrate    = maxrate,
+      minrate    = bitrate,
+      profile    = profile,
+      qp         = qp,
+      gop        = gop,
+      seek       = seek,
+      fps        = fps,
+    )
+
+  @slash.parametrize(*gen_hevc_seek_parameters(spec, ['main']))
+  def test(self, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, seek, fps):
+    self.init(spec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, seek, fps)
+    self.encode()
+
+class seek_lp(HEVC8EncoderLPTest):
+  def init(self, tspec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, seek, fps):
+    vars(self).update(tspec[case].copy())
+    vars(self).update(
+      rcmode     = rcmode,
+      bframes    = bframes,
+      bitrate    = bitrate,
+      case       = case,
+      maxrate    = maxrate,
+      minrate    = bitrate,
+      profile    = profile,
+      qp         = qp,
+      gop        = gop,
+      seek       = seek,
+      fps        = fps,
+    )
+
+  @slash.parametrize(*gen_hevc_seek_lp_parameters(spec, ['main']))
+  def test(self, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, seek, fps):
+    self.init(spec, case, gop, bframes, bitrate, qp, maxrate, profile, rcmode, seek, fps)
+    self.encode()