FVC/JEM代码学习22：xCheckRDCostInterKLT

在JEM中新加入了KLT模式，暂时还没看出来哪里体现出了KLT，但是此函数是在执行完帧间，帧内，PCM模式后才会执行。与帧间，帧内，PCM也形成了竞争关系。#if VCEG_AZ08_INTER_KLT
#if VCEG_AZ08_USE_KLT
if (sps.getUseInterKLT())//JEM中新加的;
{
#endif
if (!rpcBestCU->isIntra(0) && rpcBestCU->getQtRootCbf(0) != 0)//如果最优CU的预测模式不是帧内且cbf不等于0
{
//Only check from the best modes for speeding up，为了速度，只check最好的模式;
g_bEnableCheck = true;
Int iQP = rpcBestCU->getQP(0);
#if JVET_C0024_QTBT
xCheckRDCostInterKLT(rpcBestCU, rpcTempCU, SIZE_2Nx2N);//JEM里新加的技术;
#else
PartSize eSize = rpcBestCU->getPartitionSize(0);
xCheckRDCostInterKLT(rpcBestCU, rpcTempCU, eSize);
#endif
rpcTempCU->initEstData(uiDepth, iQP, false);
}
#if VCEG_AZ08_USE_KLT
}
#endif
#endif

#if VCEG_AZ08_INTER_KLT
Void TEncCu::xCheckRDCostInterKLT(TComDataCU*& rpcBestCU, TComDataCU*& rpcTempCU, PartSize ePartSize)
{
DEBUG_STRING_NEW(sTest)

#if JVET_C0024_QTBT
if( rpcTempCU->getWidth(0) != rpcTempCU->getHeight(0) )//KLT只适用于正方形？？
{
return;
}
#endif

if(getFastDeltaQp())
{
const TComSPS &sps=*( rpcTempCU->getSlice()->getSPS());
#if JVET
b94b
_C0024_QTBT
const UInt fastDeltaQPCuMaxSize = Clip3(sps.getMinQTSize(rpcBestCU->getSlice()->getSliceType(), rpcBestCU->getTextType()), sps.getCTUSize(), 32u);
#else
const UInt fastDeltaQPCuMaxSize = Clip3(sps.getMaxCUHeight()>>(sps.getLog2DiffMaxMinCodingBlockSize()), sps.getMaxCUHeight(), 32u);
#endif
if(ePartSize != SIZE_2Nx2N || rpcTempCU->getWidth( 0 ) > fastDeltaQPCuMaxSize)
{
return; // only check necessary 2Nx2N Inter in fast deltaqp mode
}
}

// prior to this, rpcTempCU will have just been reset using rpcTempCU->initEstData( uiDepth, iQP, bIsLosslessMode );
UChar uhDepth = rpcTempCU->getDepth(0);

#if COM16_C806_LARGE_CTU
if (m_pcEncCfg->getUseFastLCTU())//如果使用快速LCTU
{
if (ePartSize != SIZE_2Nx2N && rpcTempCU->getWidth(0) > 64)//如果块不为2N*2N且块的宽大于64
{
rpcTempCU->getTotalCost() = MAX_DOUBLE / 4;//直接将TempCU的代价赋为MAX_DOUBLE / 4
rpcTempCU->getTotalDistortion() = MAX_INT;//直接将TempCU的失真赋为MAX_INT
xCheckBestMode(rpcBestCU, rpcTempCU, uhDepth);//然后将TempCU与BestCU进行比较
return;
}
}
#endif

Bool bSkipPossible = false;
rpcTempCU->copySameSizeCUFrom(rpcBestCU, 0, uhDepth);
UInt uiWidth = rpcTempCU->getWidth(0);
UInt uiHeigh = rpcTempCU->getHeight(0);
#if JVET_C0024_QTBT
UInt uiWIdx = g_aucConvertToBit[uiWidth];
UInt uiHIdx = g_aucConvertToBit[uiHeigh];
Pel *pYSrc = m_pppcPredYuvBest[uiWIdx][uiHIdx]->getAddr(COMPONENT_Y);
Pel *pYDst = m_pppcPredYuvTemp[uiWIdx][uiHIdx]->getAddr(COMPONENT_Y);
#else
Pel *pYSrc = m_ppcPredYuvBest[uhDepth]->getAddr(COMPONENT_Y);
Pel *pYDst = m_ppcPredYuvTemp[uhDepth]->getAddr(COMPONENT_Y);
#endif
memcpy(pYDst, pYSrc, sizeof(Pel)*uiWidth*uiHeigh);//将 m_pppcPredYuvBest的亮度分量复制给m_pppcPredYuvTemp
const UInt componentShiftCb = rpcTempCU->getPic()->getComponentScaleX(COMPONENT_Cb) + rpcTempCU->getPic()->getComponentScaleY(COMPONENT_Cb);
#if JVET_C0024_QTBT
pYSrc = m_pppcPredYuvBest[uiWIdx][uiHIdx]->getAddr(COMPONENT_Cb);
pYDst = m_pppcPredYuvTemp[uiWIdx][uiHIdx]->getAddr(COMPONENT_Cb);
#else
pYSrc = m_ppcPredYuvBest[uhDepth]->getAddr(COMPONENT_Cb);
pYDst = m_ppcPredYuvTemp[uhDepth]->getAddr(COMPONENT_Cb);
#endif
memcpy(pYDst, pYSrc, sizeof(Pel)* uiWidth * uiHeigh >> componentShiftCb);//将 m_pppcPredYuvBest的色度分量Cb复制给m_pppcPredYuvTemp
const UInt componentShiftCr = rpcTempCU->getPic()->getComponentScaleX(COMPONENT_Cb) + rpcTempCU->getPic()->getComponentScaleY(COMPONENT_Cb);
#if JVET_C0024_QTBT
pYSrc = m_pppcPredYuvBest[uiWIdx][uiHIdx]->getAddr(COMPONENT_Cr);
pYDst = m_pppcPredYuvTemp[uiWIdx][uiHIdx]->getAddr(COMPONENT_Cb);
#else
pYSrc = m_ppcPredYuvBest[uhDepth]->getAddr(COMPONENT_Cr);
pYDst = m_ppcPredYuvTemp[uhDepth]->getAddr(COMPONENT_Cb);
#endif
memcpy(pYDst, pYSrc, sizeof(Pel)*uiWidth*uiHeigh >> componentShiftCr);//将 m_pppcPredYuvBest的色度分量Cr复制给m_pppcPredYuvTemp
bSkipPossible = rpcBestCU->getSkipFlag(0);
#if AMP_MRG && !JVET_C0024_QTBT
if (!rpcTempCU->getMergeAMP())
{
return;
}
#endif

#if COM16_C806_OBMC //QC_OBMC
#if JVET_C0024_QTBT
m_pcPredSearch->motionCompensation(rpcTempCU, m_pppcPredYuvTemp[uiWIdx][uiHIdx]);//执行运动补偿
rpcTempCU->setOBMCFlagSubParts(true, 0, uhDepth);
m_pcPredSearch->subBlockOBMC(rpcTempCU, 0, m_pppcPredYuvTemp[uiWIdx][uiHIdx], m_pppcTmpYuv1[uiWIdx][uiHIdx], m_pppcTmpYuv2[uiWIdx][uiHIdx]);//重叠块运动补偿
#else
m_pcPredSearch->motionCompensation(rpcTempCU, m_ppcPredYuvTemp[uhDepth]);
rpcTempCU->setOBMCFlagSubParts(true, 0, uhDepth);
m_pcPredSearch->subBlockOBMC(rpcTempCU, 0, m_ppcPredYuvTemp[uhDepth], m_ppcTmpYuv1[uhDepth], m_ppcTmpYuv2[uhDepth]);
#endif
#endif

#if JVET_C0024_QTBT
m_pcPredSearch->encodeResAndCalcRdInterCU( rpcTempCU, m_pppcOrigYuv[uiWIdx][uiHIdx],
m_pppcPredYuvTemp[uiWIdx][uiHIdx], m_pppcResiYuvTemp[uiWIdx][uiHIdx], m_pppcResiYuvBest[uiWIdx][uiHIdx], m_pppcRecoYuvTemp[uiWIdx][uiHIdx], false
#else
m_pcPredSearch->encodeResAndCalcRdInterCU( rpcTempCU, m_ppcOrigYuv[uhDepth],
m_ppcPredYuvTemp[uhDepth], m_ppcResiYuvTemp[uhDepth], m_ppcResiYuvBest[uhDepth], m_ppcRecoYuvTemp[uhDepth], false
#endif
#if COM16_C806_EMT
, rpcBestCU->getTotalCost()
#endif
DEBUG_STRING_PASS_INTO(sTest)
);//编码残差并计算失真
if (bSkipPossible)
{
rpcTempCU->setSkipFlagSubParts(rpcTempCU->getQtRootCbf(0) == 0, 0, uhDepth);//如果Cbf等于0，则将SkipFlag设置为1，否则为0;
}
xCheckDQP(rpcTempCU);
xCheckBestMode(rpcBestCU, rpcTempCU, uhDepth);
}
#endif