Void TComSlice::sortPicList (TComList<TComPic*>& rcListPic)
{
TComPic* pcPicExtract;
TComPic* pcPicInsert;
TComList<TComPic*>::iterator iterPicExtract;
TComList<TComPic*>::iterator iterPicExtract_1;
TComList<TComPic*>::iterator iterPicInsert;
for (Int i = 1; i < (Int)(rcListPic.size()); i++)
{
iterPicExtract = rcListPic.begin();
for (Int j = 0; j < i; j++) iterPicExtract++;
pcPicExtract = *(iterPicExtract);
pcPicExtract->setCurrSliceIdx(0);
iterPicInsert = rcListPic.begin();
while (iterPicInsert != iterPicExtract)
{
pcPicInsert = *(iterPicInsert);
pcPicInsert->setCurrSliceIdx(0);
if (pcPicInsert->getPOC() >= pcPicExtract->getPOC())
{
break;
}
iterPicInsert++;
}
iterPicExtract_1 = iterPicExtract; iterPicExtract_1++;
// swap iterPicExtract and iterPicInsert, iterPicExtract = curr. / iterPicInsert = insertion position
rcListPic.insert (iterPicInsert, iterPicExtract, iterPicExtract_1);
rcListPic.erase (iterPicExtract);
}
}
Void TComSlice::setRefPicList ( TComList<TComPic*>& rcListPic )
{
if (m_eSliceType == I_SLICE)
{
::memset( m_apcRefPicList, 0, sizeof (m_apcRefPicList));
::memset( m_aiNumRefIdx, 0, sizeof ( m_aiNumRefIdx ));
return;
}
m_aiNumRefIdx[0] = min ( m_aiNumRefIdx[0], (Int)(rcListPic.size())-1 );
m_aiNumRefIdx[1] = min ( m_aiNumRefIdx[1], (Int)(rcListPic.size())-1 );
sortPicList(rcListPic);
TComPic* pcRefPic;
for (Int n = 0; n < 2; n++)
{
RefPicList eRefPicList = (RefPicList)n;
UInt uiOrderDRB = 1;
UInt uiOrderERB = 1;
Int iRefIdx = 0;
UInt uiActualListSize = 0;
if ( m_eSliceType == P_SLICE && eRefPicList == REF_PIC_LIST_1 )
{
m_aiNumRefIdx[eRefPicList] = 0;
::memset( m_apcRefPicList[eRefPicList], 0, sizeof(m_apcRefPicList[eRefPicList]));
break;
}
// First DRB
pcRefPic = xGetRefPic(rcListPic, true, ERB_NONE, m_iPOC, eRefPicList, uiOrderDRB, m_uiTLayer);
if (pcRefPic != NULL)
{
m_apcRefPicList[eRefPicList][iRefIdx] = pcRefPic;
pcRefPic->getPicYuvRec()->extendPicBorder();
iRefIdx++;
uiOrderDRB++;
uiActualListSize++;
}
if ( (Int)uiActualListSize >= m_aiNumRefIdx[eRefPicList] )
{
m_aiNumRefIdx[eRefPicList] = uiActualListSize;
continue;
}
// Long term reference
// Should be enabled to support long term refernce
//*
// First ERB
pcRefPic = xGetRefPic(rcListPic, false, ERB_LTR, m_iPOC, eRefPicList, uiOrderERB, m_uiTLayer);
if (pcRefPic != NULL)
{
Bool bChangeDrbErb = false;
if (iRefIdx > 0 && eRefPicList == REF_PIC_LIST_0 && pcRefPic->getPOC() > m_apcRefPicList[eRefPicList][iRefIdx-1]->getPOC())
{
bChangeDrbErb = true;
}
else if (iRefIdx > 0 && eRefPicList == REF_PIC_LIST_1 && pcRefPic->getPOC() < m_apcRefPicList[eRefPicList][iRefIdx-1]->getPOC())
{
bChangeDrbErb = true;
}
if ( bChangeDrbErb)
{
m_apcRefPicList[eRefPicList][iRefIdx] = m_apcRefPicList[eRefPicList][iRefIdx-1];
m_apcRefPicList[eRefPicList][iRefIdx-1] = pcRefPic;
}
else
{
m_apcRefPicList[eRefPicList][iRefIdx] = pcRefPic;
}
pcRefPic->getPicYuvRec()->extendPicBorder();
iRefIdx++;
uiOrderERB++;
uiActualListSize++;
}
//*/
// Short term reference
// Residue DRB
UInt uiBreakCount = 17 - iRefIdx;
while (1)
{
uiBreakCount--;
if ( (Int)uiActualListSize >= m_aiNumRefIdx[eRefPicList] || uiBreakCount == 0 )
{
break;
}
pcRefPic = xGetRefPic(rcListPic, true, ERB_NONE, m_iPOC, eRefPicList, uiOrderDRB, m_uiTLayer);
if (pcRefPic != NULL)
{
uiOrderDRB++;
}
if (pcRefPic != NULL)
{
m_apcRefPicList[eRefPicList][iRefIdx] = pcRefPic;
pcRefPic->getPicYuvRec()->extendPicBorder();
iRefIdx++;
uiActualListSize++;
}
}
m_aiNumRefIdx[eRefPicList] = uiActualListSize;
}
}
Void TEncTop::encode(Bool flush, TComPicYuv* pcPicYuvOrg, TComPicYuv* pcPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Bool isTff)
{
iNumEncoded = 0;
for (Int fieldNum=0; fieldNum<2; fieldNum++)
{
if (pcPicYuvOrg)
{
/* -- field initialization -- */
const Bool isTopField=isTff==(fieldNum==0);
TComPic *pcField;
xGetNewPicBuffer( pcField );
pcField->setReconMark (false); // where is this normally?
if (fieldNum==1) // where is this normally?
{
TComPicYuv* rpcPicYuvRec;
// org. buffer
if ( rcListPicYuvRecOut.size() >= (UInt)m_iGOPSize+1 ) // need to maintain field 0 in list of RecOuts while processing field 1. Hence +1 on m_iGOPSize.
{
rpcPicYuvRec = rcListPicYuvRecOut.popFront();
}
else
{
rpcPicYuvRec = new TComPicYuv;
rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, true);
}
rcListPicYuvRecOut.pushBack( rpcPicYuvRec );
}
pcField->getSlice(0)->setPOC( m_iPOCLast ); // superfluous?
pcField->getPicYuvRec()->setBorderExtension(false);// where is this normally?
pcField->setTopField(isTopField); // interlaced requirement
for (UInt componentIndex = 0; componentIndex < pcPicYuvOrg->getNumberValidComponents(); componentIndex++)
{
const ComponentID component = ComponentID(componentIndex);
const UInt stride = pcPicYuvOrg->getStride(component);
separateFields((pcPicYuvOrg->getBuf(component) + pcPicYuvOrg->getMarginX(component) + (pcPicYuvOrg->getMarginY(component) * stride)),
pcField->getPicYuvOrg()->getAddr(component),
pcPicYuvOrg->getStride(component),
pcPicYuvOrg->getWidth(component),
pcPicYuvOrg->getHeight(component),
isTopField);
separateFields((pcPicYuvTrueOrg->getBuf(component) + pcPicYuvTrueOrg->getMarginX(component) + (pcPicYuvTrueOrg->getMarginY(component) * stride)),
pcField->getPicYuvTrueOrg()->getAddr(component),
pcPicYuvTrueOrg->getStride(component),
pcPicYuvTrueOrg->getWidth(component),
pcPicYuvTrueOrg->getHeight(component),
isTopField);
}
// compute image characteristics
if ( getUseAdaptiveQP() )
{
m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcField ) );
}
}
if ( m_iNumPicRcvd && ((flush&&fieldNum==1) || (m_iPOCLast/2)==0 || m_iNumPicRcvd==m_iGOPSize ) )
{
// compress GOP
m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff, snrCSC, m_printFrameMSE,&m_cSearch);
iNumEncoded += m_iNumPicRcvd;
m_uiNumAllPicCoded += m_iNumPicRcvd;
m_iNumPicRcvd = 0;
}
}
}