H.266/VVC代码学习14：色度列表及DM模式代码（getFinalIntraMode、getIntraChromaCandModes）

1、DM模式：

DM即derived mode，是一种色度预测特有的模式。在预测色度的时候，亮度已经预测完成，此时预测当前块的色度模式可以借用已经完成的亮度模式。经过事实的验证，当前色度块对应中心亮度块的预测模式更符合当前色度块的预测模式，预测效果更好。下图为DM模式使用亮度块的位置：

2、色度列表：

然而此时出现一个问题，传统的四种模式已经填写在色度预测列表中，如果对应亮度预测模式是传统四种模式中的一种，那么后续的RD代价的检验就多进行了一次，且是无用的一次。所以色度列表由于DM的存在进行微调，将四种传统模式与DM预测出的模式冲突的位置换成66，用以占位，即可迎刃而解。获取列表不单单在预测过程中会启用，在CABAC编解码中也会启用，用于编码色度信息。色度列表如下图：（温馨提示，此表竖着看，更直观的请看：H.266/VVC代码学习1：帧内预测框架）

3、代码相关：

1 获取色度列表入口函数：H.266/VVC代码学习8：xRecurIntraChromaCodingQT函数。
在进行预测前，列表就已经写好了。所以获取列表后才会调用下面那个函数：
2 DM模式预测入口函数：H.266/VVC代码学习9：xIntraCodingTUBlock函数
这两个函数比较相似，具体代码如下：

void PU::getIntraChromaCandModes( const PredictionUnit &pu, unsigned modeList[NUM_CHROMA_MODE] )//进行预测前和编解码时用的
{
  {
    modeList[  0 ] = PLANAR_IDX;
    modeList[  1 ] = VER_IDX;
    modeList[  2 ] = HOR_IDX;
    modeList[  3 ] = DC_IDX;
    modeList[4] = LM_CHROMA_IDX;
    modeList[5] = MDLM_L_IDX;
    modeList[6] = MDLM_T_IDX;
	modeList[7] = DM_CHROMA_IDX;

    Position topLeftPos = pu.blocks[pu.chType].lumaPos();//左上角pu位置
    Position refPos = topLeftPos.offset( pu.blocks[pu.chType].lumaSize().width >> 1, pu.blocks[pu.chType].lumaSize().height >> 1 );//中间块pu位置
    const PredictionUnit *lumaPU = CS::isDualITree( *pu.cs ) ? pu.cs->picture->cs->getPU( refPos, CHANNEL_TYPE_LUMA ) : &pu;//若是亮度：亮度块pu；若是色度：色度块对应亮度块pu
    const uint32_t lumaMode = lumaPU->intraDir[CHANNEL_TYPE_LUMA];//亮度块预测模式
    for( int i = 0; i < 4; i++ )//当DM预测的模式与四种传统的色度预测模式相同时
    {
      if( lumaMode == modeList[i] )
      {
        modeList[i] = VDIA_IDX;//使用斜对角，即模式66替代传统的色度预测模式，起到占位作用，选中几率很小
        break;
      }
    }
  }
}

uint32_t PU::getFinalIntraMode( const PredictionUnit &pu, const ChannelType &chType )
{
  uint32_t uiIntraMode = pu.intraDir[chType];

  if( uiIntraMode == DM_CHROMA_IDX && !isLuma( chType ) )
  {
    Position topLeftPos = pu.blocks[pu.chType].lumaPos();//左上角pu位置
    Position refPos = topLeftPos.offset( pu.blocks[pu.chType].lumaSize().width >> 1, pu.blocks[pu.chType].lumaSize().height >> 1 );//中间块pu位置
    const PredictionUnit &lumaPU = CS::isDualITree( *pu.cs ) ? *pu.cs->picture->cs->getPU( refPos, CHANNEL_TYPE_LUMA ) : *pu.cs->getPU( topLeftPos, CHANNEL_TYPE_LUMA );//取色度块对应中心亮度块pu

    uiIntraMode = lumaPU.intraDir[0];//获得中心亮度块的亮度模式
  }
  if( pu.chromaFormat == CHROMA_422 && !isLuma( chType ) )
  {
    uiIntraMode = g_chroma422IntraAngleMappingTable[uiIntraMode];
  }
  return uiIntraMode;//得到了当前pu的预测方向
}