HM的码率控制.docVIP

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()中，对整个序列都要用到的相关参数进行初始化 [cpp]?view plaincopy #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需要用到的相关参数进行初始化 [cpp]?view plaincopy #if?RATE_CONTROL_LAMBDA_DOMAIN?? ??if?(?m_RCEnableRateControl?)?? ??{?? ????m_cRateCtrl.initRCGOP(?m_iNumPicRcvd?);?? ??}?? #endif?? ? 接着，在TEncGOP::compressGOP()中，对每一幅picture需要用到的相关参数进行初始化 [cpp]?view plaincopy #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_SLICE?)?? ??????{?? ????????frameLevel?=?0;?? ??????}?? ??????m_pcRateCtrl-initRCPic(?frameLevel?);?//!?picture?level?初始化?? ??????estimatedBits?=?m_pcRateCtrl-getRCPic()-getTargetBits();?? ???????? ??????Int?sliceQP?=?m_pcCfg-getInitialQP();?? ??????if?(?(?pcSlice-