HEVC码率控制浅析——HM代码阅读之一

最新推荐文章于 2023-02-23 10:50:21 发布

置顶

HEVC_CJL

最新推荐文章于 2023-02-23 10:50:21 发布

阅读量1.4w

点赞数 5

分类专栏： HEVC 文章标签： hevc HM 码率控制 rate control RC

本文链接：https://blog.csdn.net/hevc_cjl/article/details/10982699

版权

本文基于HM12.0探讨HEVC码率控制，重点关注K0103提案。分析了HM中码率控制的总体框架，包括在TEncTop、TEncGOP、TEncSlice和TEncCu类中的初始化过程及参数更新步骤，并阐述了TEncRateCtrl及其子类在不同级别的RC管理职责。

摘要由CSDN通过智能技术生成

HM的码率控制提案主要参考如下三篇：K0103，M0036，M0257。本文及后续文章将基于HM12.0进行讨论，且首先仅讨论K0103对应的代码，之后再陆续补充M0036，M0257对应的代码分析，这么做可能会使得剧情不会显得那么地跳跃，分析起来能够更好地被接受。

按照我的个人习惯，还是先分析HM中码率控制部分（以后简称RC）的总体框架吧。

跟RC有关的头文件和源文件为工程TLibEncoder中的TEncRateCtrl.h和TEncRateCtrl.cpp，其余的地方都是调用这两个文件中定义的函数或者变量。

在最顶层，TEncTop这个类定义了成员变量TEncRateCtrl m_cRateCtrl，类TEncGOP，TEncSlice以及TEncCu也分别定义了成员变量TEncRateCtrl *m_pcRateCtrl，但是请注意，这三个类的m_pcRateCtrl实际上都是指向TEncTop这个类的m_cRateCtrl，即它们本质上是同一个。这个从这三个类的成员函数init中对各成员变量的初始化可以看出来。

（1）初始化：

首先，在TEncTop::create()中，对整个序列都要用到的相关参数进行初始化

#if RATE_CONTROL_LAMBDA_DOMAIN
  if ( m_RCEnableRateControl )
  {
    m_cRateCtrl.init( m_framesToBeEncoded, m_RCTargetBitrate, m_iFrameRate, m_iGOPSize, m_iSourceWidth, m_iSourceHeight,
                      g_uiMaxCUWidth, g_uiMaxCUHeight, m_RCKeepHierarchicalBit, m_RCUseLCUSeparateModel, m_GOPList );
  }
#else
  m_cRateCtrl.create(getIntraPeriod(), getGOPSize(), getFrameRate(), getTargetBitrate(), getQP(), getNumLCUInUnit(), getSourceWidth(), getSourceHeight(), g_uiMaxCUWidth, g_uiMaxCUHeight);
#endif

其次，在TEncTop::encode()中，对整个GOP需要用到的相关参数进行初始化

#if RATE_CONTROL_LAMBDA_DOMAIN
  if ( m_RCEnableRateControl )
  {
    m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
  }
#endif

接着，在TEncGOP::compressGOP()中，对每一幅picture需要用到的相关参数进行初始化

#if RATE_CONTROL_LAMBDA_DOMAIN
    Double lambda            = 0.0;
    Int actualHeadBits       = 0;
    Int actualTotalBits      = 0;
    Int estimatedBits        = 0;
    Int tmpBitsBeforeWriting = 0;
    if ( m_pcCfg->getUseRateCtrl() )
    {
      Int frameLevel = m_pcRateCtrl->getRCSeq()->getGOPID2Level( iGOPid );
      if ( pcPic->getSlice(0)->getSliceType() == I_SLIC