HEVC参考帧列表管理代码部分

这篇文章大部分内容转载于http://blog.csdn.net/yangxiao_xiang/article/details/8872173,作者根据自己的理解,加入了一些自己理解的代码部分。如有不正确的地方还请大家给与指正。

对于帧间预测而言,一定会涉及到参考图像的问题,但是搞清楚HM中参考图像是如何配置,对理解其编码结构尤为关键。顾名思义,参考图像应该是属于picture级别的,因此可以在compressGOP函数中找到相应的函数。主要包括以下三个函数:

1、arrangeLongtermPicturesInRPS(pcSlice, rcListPic);

2、pcSlice->setRefPicList ( rcListPic );

3、pcSlice->setRefPOCList();

首先说第一个函数,主要是在RPS(Reference Picture Set)句法中设置LongTermPicture,那么什么是LT呢?参考图像主要有以下三种:long-term,short-term before curr和short-term after curr。第一个函数就是用来提供long-term 参考图像的,具体实现可以查看函数,这里不做重点叙述。

HEVC参考帧列表管理代码部分_第1张图片


重点是第二个函数setRefPicList,上面我们了解到参考图像主要有三种,但是最后参考图像列表中的参考图像是通过下面的方式确定的。那些在参考列表中用到的参考图像,其标志为1,相反,则为0;故setRefPicList函数的实现过程是先通过标志位判断(if(m_pcRPS->getUsed(i)))确定三种参考图像的实际参考帧(如下面第一部分),然后再通过ref_pic_list_init过程,再组成list0和list1列表,实现过程详见程序。


HEVC参考帧列表管理代码部分_第2张图片

[cpp]  view plain copy
  1. Void TComSlice::setRefPicList( TComList& rcListPic )  
  2. {  
  3.   if (m_eSliceType == I_SLICE)  
  4.   {//==对于I帧而言,直接返回==//  
  5.     ::memset( m_apcRefPicList, 0, sizeof (m_apcRefPicList));  
  6.     ::memset( m_aiNumRefIdx,   0, sizeof ( m_aiNumRefIdx ));  
  7.       
  8.     return;  
  9.   }  
  10.     
  11.   m_aiNumRefIdx[0] = getNumRefIdx(REF_PIC_LIST_0); //==list0参考图像数目==//  
  12.   m_aiNumRefIdx[1] = getNumRefIdx(REF_PIC_LIST_1); "font-family: Arial, Helvetica, sans-serif;">//==list0参考图像数目==//  
  13.   
  14.   TComPic*  pcRefPic= NULL;  
  15.   TComPic*  RefPicSetStCurr0[16];  
  16.   TComPic*  RefPicSetStCurr1[16];  
  17.   TComPic*  RefPicSetLtCurr[16];  
  18.   UInt NumPocStCurr0 = 0;  
  19.   UInt NumPocStCurr1 = 0;  
  20.   UInt NumPocLtCurr = 0;  
  21.   Int i;  
  22.   
  23.   for(i=0; i < m_pcRPS->getNumberOfNegativePictures(); i++)  
  24.   {// 设置short term negative(即前向参考图像)  
  25.     if(m_pcRPS->getUsed(i))  
  26.     {  
  27.       pcRefPic = xGetRefPic(rcListPic, getPOC()+m_pcRPS->getDeltaPOC(i));  
  28.       pcRefPic->setIsLongTerm(0);  
  29.       pcRefPic->setIsUsedAsLongTerm(0);  
  30.       pcRefPic->getPicYuvRec()->extendPicBorder();  
  31.       RefPicSetStCurr0[NumPocStCurr0 ] = pcRefPic;  
  32.       NumPocStCurr0++;  
  33.       pcRefPic->setCheckLTMSBPresent(false);    
  34.     }  
  35.   }  
  36.   for(; i < m_pcRPS->getNumberOfNegativePictures()+m_pcRPS->getNumberOfPositivePictures(); i++)  
  37.   {// 设置short term positive(即后向参考图像)  
  38.     if(m_pcRPS->getUsed(i))  
  39.     {  
  40.       pcRefPic = xGetRefPic(rcListPic, getPOC()+m_pcRPS->getDeltaPOC(i));  
  41.       pcRefPic->setIsLongTerm(0);  
  42.       pcRefPic->setIsUsedAsLongTerm(0);  
  43.       pcRefPic->getPicYuvRec()->extendPicBorder();  
  44.       RefPicSetStCurr1[NumPocStCurr1] = pcRefPic;  
  45.       NumPocStCurr1++;  
  46.       pcRefPic->setCheckLTMSBPresent(false);    
  47.     }  
  48.   }  
  49.   for(i = m_pcRPS->getNumberOfNegativePictures()+m_pcRPS->getNumberOfPositivePictures()+m_pcRPS->getNumberOfLongtermPictures()-1; i > m_pcRPS->getNumberOfNegativePictures()+m_pcRPS->getNumberOfPositivePictures()-1 ; i--)  
  50.   {//==设置long term参考图像==//  
  51.     if(m_pcRPS->getUsed(i))  
  52.     {  
  53.       pcRefPic = xGetLongTermRefPic(rcListPic, m_pcRPS->getPOC(i));  
  54.       pcRefPic->setIsLongTerm(1);  
  55.       pcRefPic->setIsUsedAsLongTerm(1);  
  56.       pcRefPic->getPicYuvRec()->extendPicBorder();  
  57.       RefPicSetLtCurr[NumPocLtCurr] = pcRefPic;  
  58.       NumPocLtCurr++;  
  59.     }  
  60.     if(pcRefPic==NULL)   
  61.     {  
  62.       pcRefPic = xGetLongTermRefPic(rcListPic, m_pcRPS->getPOC(i));  
  63.     }  
  64.     pcRefPic->setCheckLTMSBPresent(m_pcRPS->getCheckLTMSBPresent(i));    
  65.   }  
  66.   // ref_pic_list_init  
  67. #if RPL_INIT_FIX  
  68.   TComPic*  rpsCurrList0[MAX_NUM_REF+1];  
  69.   TComPic*  rpsCurrList1[MAX_NUM_REF+1];  
  70.   Int numPocTotalCurr = NumPocStCurr0 + NumPocStCurr1 + NumPocLtCurr;  
  71.   
  72.   {  
  73.     Int cIdx = 0;  
  74.     for ( i=0; i
  75.     {//==(0<=cIdx  
  76.       rpsCurrList0[cIdx] = RefPicSetStCurr0[i];  
  77.     }  
  78.     for ( i=0; i
  79.     {//==(NumPocStCurr0<=cIdx  
  80.       rpsCurrList0[cIdx] = RefPicSetStCurr1[i];  
  81.     }  
  82.     for ( i=0; i
  83.     {//==(NumPocStCurr1<=cIdx  
  84.       rpsCurrList0[cIdx] = RefPicSetLtCurr[i];  
  85.     }  
  86.   }  
  87.   
  88.   if (m_eSliceType==B_SLICE)  
  89.   {  
  90.     Int cIdx = 0;  
  91.     for ( i=0; i
  92.     {//==(0<=cIdx  
  93.       rpsCurrList1[cIdx] = RefPicSetStCurr1[i];  
  94.     }  
  95.     for ( i=0; i
  96.     {//==(NumPocStCurr0<=cIdx  
  97.       rpsCurrList1[cIdx] = RefPicSetStCurr0[i];  
  98.     }  
  99.     for ( i=0; i
  100.     {//==(NumPocStCurr1+NumPocStCurr0<=cIdx  
  101.       rpsCurrList1[cIdx] = RefPicSetLtCurr[i];  
  102.     }  
  103.   }  
  104.      
  105.   ///=====Unification of reference picture list modification processes(参考JCTVC-H0138)=====///  
  106.   for (Int rIdx = 0; rIdx <= (m_aiNumRefIdx[0]-1); rIdx ++)  
  107.   {//===修正list0列表===//  
  108.     m_apcRefPicList[0][rIdx] = m_RefPicListModification.getRefPicListModificationFlagL0() ? rpsCurrList0[ m_RefPicListModification.getRefPicSetIdxL0(rIdx) ] : rpsCurrList0[rIdx % numPocTotalCurr];  
  109.   }  
  110.   if ( m_eSliceType == P_SLICE )  
  111.   {//////////////////==若为P帧,则list1为零==/////////////////////////////  
  112.     m_aiNumRefIdx[1] = 0;  
  113.     ::memset( m_apcRefPicList[1], 0, sizeof(m_apcRefPicList[1]));  
  114.   }  
  115.   else  
  116.   {//===修正list1列表===//  
  117.     for (Int rIdx = 0; rIdx <= (m_aiNumRefIdx[1]-1); rIdx ++)  
  118.     {  
  119.       m_apcRefPicList[1][rIdx] = m_RefPicListModification.getRefPicListModificationFlagL1() ? rpsCurrList1[ m_RefPicListModification.getRefPicSetIdxL1(rIdx) ] : rpsCurrList1[rIdx % numPocTotalCurr];  
  120.     }  
  121.   }  

而最后一个函数setRefPOCList就是实现将参考图像转化成其相应的POC列表。

[cpp]  view plain copy
  1. Void TComSlice::setRefPOCList       ()  
  2. {//===将参考图像列表转化成相应的POC===//  
  3.   for (Int iDir = 0; iDir < 2; iDir++)  
  4.   {  
  5.     for (Int iNumRefIdx = 0; iNumRefIdx < m_aiNumRefIdx[iDir]; iNumRefIdx++)  
  6.     {  
  7.       m_aiRefPOCList[iDir][iNumRefIdx] = m_apcRefPicList[iDir][iNumRefIdx]->getPOC();  
  8.     }  
  9.   }  
  10. }  


另外在第二个函数中有涉及到ListModification(列表修正)的问题,由于篇幅有限,将和 List Combination(参考列表联合)一起留到后续再讨论!

你可能感兴趣的:(HEVC)