/** Create non-deblocked filter information
* \param pSliceStartAddress array for storing slice start addresses
* \param numSlices number of slices in picture
* \param sliceGranularityDepth slice granularity
* \param bNDBFilterCrossSliceBoundary cross-slice-boundary in-loop filtering; true for "cross".
* \param numTiles number of tiles in picture
* \param bNDBFilterCrossTileBoundary cross-tile-boundary in-loop filtering; true for "cross".
*/
Void TComPic::createNonDBFilterInfo(std::vector<Int> sliceStartAddress, Int sliceGranularityDepth
,std::vector<Bool>* LFCrossSliceBoundary
,Int numTiles
,Bool bNDBFilterCrossTileBoundary)
{
UInt maxNumSUInLCU = getNumPartInCU();
UInt numLCUInPic = getNumCUsInFrame();
UInt picWidth = getSlice(0)->getSPS()->getPicWidthInLumaSamples();
UInt picHeight = getSlice(0)->getSPS()->getPicHeightInLumaSamples();
Int numLCUsInPicWidth = getFrameWidthInCU();
Int numLCUsInPicHeight= getFrameHeightInCU();
UInt maxNumSUInLCUWidth = getNumPartInWidth();
UInt maxNumSUInLCUHeight= getNumPartInHeight();
Int numSlices = (Int) sliceStartAddress.size() - 1;
m_bIndependentSliceBoundaryForNDBFilter = false;
if(numSlices > 1)
{
for(Int s=0; s< numSlices; s++)
{
if((*LFCrossSliceBoundary)[s] == false)
{
m_bIndependentSliceBoundaryForNDBFilter = true;
}
}
}
m_sliceGranularityForNDBFilter = sliceGranularityDepth;
m_bIndependentTileBoundaryForNDBFilter = (bNDBFilterCrossTileBoundary)?(false) :((numTiles > 1)?(true):(false));
m_pbValidSlice = new Bool[numSlices];
for(Int s=0; s< numSlices; s++)
{
m_pbValidSlice[s] = true;
}
m_pSliceSUMap = new Int[maxNumSUInLCU * numLCUInPic];
//initialization
for(UInt i=0; i< (maxNumSUInLCU * numLCUInPic); i++ )
{
m_pSliceSUMap[i] = -1;
}
for( UInt CUAddr = 0; CUAddr < numLCUInPic ; CUAddr++ )
{
TComDataCU* pcCU = getCU( CUAddr );
pcCU->setSliceSUMap(m_pSliceSUMap + (CUAddr* maxNumSUInLCU));
pcCU->getNDBFilterBlocks()->clear();
}
m_vSliceCUDataLink.clear();
m_vSliceCUDataLink.resize(numSlices);
UInt startAddr, endAddr, firstCUInStartLCU, startLCU, endLCU, lastCUInEndLCU, uiAddr;
UInt LPelX, TPelY, LCUX, LCUY;
UInt currSU;
UInt startSU, endSU;
for(Int s=0; s< numSlices; s++)
{
//1st step: decide the real start address
startAddr = sliceStartAddress[s];
endAddr = sliceStartAddress[s+1] -1;
startLCU = startAddr / maxNumSUInLCU;
firstCUInStartLCU = startAddr % maxNumSUInLCU;
endLCU = endAddr / maxNumSUInLCU;
lastCUInEndLCU = endAddr % maxNumSUInLCU;
uiAddr = m_apcPicSym->getCUOrderMap(startLCU);
LCUX = getCU(uiAddr)->getCUPelX();
LCUY = getCU(uiAddr)->getCUPelY();
LPelX = LCUX + g_auiRasterToPelX[ g_auiZscanToRaster[firstCUInStartLCU] ];
TPelY = LCUY + g_auiRasterToPelY[ g_auiZscanToRaster[firstCUInStartLCU] ];
currSU = firstCUInStartLCU;
Bool bMoveToNextLCU = false;
Bool bSliceInOneLCU = (startLCU == endLCU);
while(!( LPelX < picWidth ) || !( TPelY < picHeight ))
{
currSU ++;
if(bSliceInOneLCU)
{
if(currSU > lastCUInEndLCU)
{
m_pbValidSlice[s] = false;
break;
}
}
if(currSU >= maxNumSUInLCU )
{
bMoveToNextLCU = true;
break;
}
LPelX = LCUX + g_auiRasterToPelX[ g_auiZscanToRaster[currSU] ];
TPelY = LCUY + g_auiRasterToPelY[ g_auiZscanToRaster[currSU] ];
}
if(!m_pbValidSlice[s])
{
continue;
}
if(currSU != firstCUInStartLCU)
{
if(!bMoveToNextLCU)
{
firstCUInStartLCU = currSU;
}
else
{
startLCU++;
firstCUInStartLCU = 0;
assert( startLCU < getNumCUsInFrame());
}
assert(startLCU*maxNumSUInLCU + firstCUInStartLCU < endAddr);
}
//2nd step: assign NonDBFilterInfo to each processing block
for(UInt i= startLCU; i <= endLCU; i++)
{
startSU = (i == startLCU)?(firstCUInStartLCU):(0);
endSU = (i == endLCU )?(lastCUInEndLCU ):(maxNumSUInLCU -1);
uiAddr = m_apcPicSym->getCUOrderMap(i);
Int iTileID= m_apcPicSym->getTileIdxMap(uiAddr);
TComDataCU* pcCU = getCU(uiAddr);
m_vSliceCUDataLink[s].push_back(pcCU);
createNonDBFilterInfoLCU(iTileID, s, pcCU, startSU, endSU, m_sliceGranularityForNDBFilter, picWidth, picHeight);
}
}
//step 3: border availability
for(Int s=0; s< numSlices; s++)
{
if(!m_pbValidSlice[s])
{
continue;
}
for(Int i=0; i< m_vSliceCUDataLink[s].size(); i++)
{
TComDataCU* pcCU = m_vSliceCUDataLink[s][i];
uiAddr = pcCU->getAddr();
if(pcCU->getPic()==0)
{
continue;
}
Int iTileID= m_apcPicSym->getTileIdxMap(uiAddr);
Bool bTopTileBoundary = false, bDownTileBoundary= false, bLeftTileBoundary= false, bRightTileBoundary= false;
if(m_bIndependentTileBoundaryForNDBFilter)
{
//left
if( uiAddr % numLCUsInPicWidth != 0)
{
bLeftTileBoundary = ( m_apcPicSym->getTileIdxMap(uiAddr -1) != iTileID )?true:false;
}
//right
if( (uiAddr % numLCUsInPicWidth) != (numLCUsInPicWidth -1) )
{
bRightTileBoundary = ( m_apcPicSym->getTileIdxMap(uiAddr +1) != iTileID)?true:false;
}
//top
if( uiAddr >= numLCUsInPicWidth)
{
bTopTileBoundary = (m_apcPicSym->getTileIdxMap(uiAddr - numLCUsInPicWidth) != iTileID )?true:false;
}
//down
if( uiAddr + numLCUsInPicWidth < numLCUInPic )
{
bDownTileBoundary = (m_apcPicSym->getTileIdxMap(uiAddr + numLCUsInPicWidth) != iTileID)?true:false;
}
}
pcCU->setNDBFilterBlockBorderAvailability(numLCUsInPicWidth, numLCUsInPicHeight, maxNumSUInLCUWidth, maxNumSUInLCUHeight,picWidth, picHeight
, *LFCrossSliceBoundary
,bTopTileBoundary, bDownTileBoundary, bLeftTileBoundary, bRightTileBoundary
,m_bIndependentTileBoundaryForNDBFilter);
}
}
if( m_bIndependentSliceBoundaryForNDBFilter || m_bIndependentTileBoundaryForNDBFilter)
{
m_pNDBFilterYuvTmp = new TComPicYuv();
m_pNDBFilterYuvTmp->create(picWidth, picHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth);
}
}
Void TDecSlice::decompressSlice(TComInputBitstream** ppcSubstreams, TComPic*& rpcPic, TDecSbac* pcSbacDecoder, TDecSbac* pcSbacDecoders)
{
TComDataCU* pcCU;
UInt uiIsLast = 0;
Int iStartCUEncOrder = max(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceCurStartCUAddr()/rpcPic->getNumPartInCU(), rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceSegmentCurStartCUAddr()/rpcPic->getNumPartInCU());
Int iStartCUAddr = rpcPic->getPicSym()->getCUOrderMap(iStartCUEncOrder);
// decoder don't need prediction & residual frame buffer
#if ENABLE_ANAYSIS_OUTPUT
#else
rpcPic->setPicYuvPred( 0 );
rpcPic->setPicYuvResi( 0 );
#endif
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceEnable;
#endif
DTRACE_CABAC_VL( g_nSymbolCounter++ );
DTRACE_CABAC_T( "\tPOC: " );
DTRACE_CABAC_V( rpcPic->getPOC() );
DTRACE_CABAC_T( "\n" );
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceDisable;
#endif
UInt uiTilesAcross = rpcPic->getPicSym()->getNumColumnsMinus1()+1;
TComSlice* pcSlice = rpcPic->getSlice(rpcPic->getCurrSliceIdx());
Int iNumSubstreams = pcSlice->getPPS()->getNumSubstreams();
// delete decoders if already allocated in previous slice
if (m_pcBufferSbacDecoders)
{
delete [] m_pcBufferSbacDecoders;
}
if (m_pcBufferBinCABACs)
{
delete [] m_pcBufferBinCABACs;
}
// allocate new decoders based on tile numbaer
m_pcBufferSbacDecoders = new TDecSbac [uiTilesAcross];
m_pcBufferBinCABACs = new TDecBinCABAC[uiTilesAcross];
for (UInt ui = 0; ui < uiTilesAcross; ui++)
{
m_pcBufferSbacDecoders[ui].init(&m_pcBufferBinCABACs[ui]);
}
//save init. state
for (UInt ui = 0; ui < uiTilesAcross; ui++)
{
m_pcBufferSbacDecoders[ui].load(pcSbacDecoder);
}
// free memory if already allocated in previous call
if (m_pcBufferLowLatSbacDecoders)
{
delete [] m_pcBufferLowLatSbacDecoders;
}
if (m_pcBufferLowLatBinCABACs)
{
delete [] m_pcBufferLowLatBinCABACs;
}
m_pcBufferLowLatSbacDecoders = new TDecSbac [uiTilesAcross];
m_pcBufferLowLatBinCABACs = new TDecBinCABAC[uiTilesAcross];
for (UInt ui = 0; ui < uiTilesAcross; ui++)
{
m_pcBufferLowLatSbacDecoders[ui].init(&m_pcBufferLowLatBinCABACs[ui]);
}
//save init. state
for (UInt ui = 0; ui < uiTilesAcross; ui++)
{
m_pcBufferLowLatSbacDecoders[ui].load(pcSbacDecoder);
}
UInt uiWidthInLCUs = rpcPic->getPicSym()->getFrameWidthInCU();
//UInt uiHeightInLCUs = rpcPic->getPicSym()->getFrameHeightInCU();
UInt uiCol=0, uiLin=0, uiSubStrm=0;
UInt uiTileCol;
UInt uiTileStartLCU;
UInt uiTileLCUX;
Int iNumSubstreamsPerTile = 1; // if independent.
Bool depSliceSegmentsEnabled = rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getPPS()->getDependentSliceSegmentsEnabledFlag();
uiTileStartLCU = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iStartCUAddr))->getFirstCUAddr();
if( depSliceSegmentsEnabled )
{
if( (!rpcPic->getSlice(rpcPic->getCurrSliceIdx())->isNextSlice()) &&
iStartCUAddr != rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iStartCUAddr))->getFirstCUAddr())
{
if(pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
{
uiTileCol = rpcPic->getPicSym()->getTileIdxMap(iStartCUAddr) % (rpcPic->getPicSym()->getNumColumnsMinus1()+1);
m_pcBufferSbacDecoders[uiTileCol].loadContexts( CTXMem[1] );//2.LCU
if ( (iStartCUAddr%uiWidthInLCUs+1) >= uiWidthInLCUs )
{
uiTileLCUX = uiTileStartLCU % uiWidthInLCUs;
uiCol = iStartCUAddr % uiWidthInLCUs;
if(uiCol==uiTileLCUX)
{
CTXMem[0]->loadContexts(pcSbacDecoder);
}
}
}
pcSbacDecoder->loadContexts(CTXMem[0] ); //end of depSlice-1
pcSbacDecoders[uiSubStrm].loadContexts(pcSbacDecoder);
}
else
{
if(pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
{
CTXMem[1]->loadContexts(pcSbacDecoder);
}
CTXMem[0]->loadContexts(pcSbacDecoder);
}
}
for( Int iCUAddr = iStartCUAddr; !uiIsLast && iCUAddr < rpcPic->getNumCUsInFrame(); iCUAddr = rpcPic->getPicSym()->xCalculateNxtCUAddr(iCUAddr) )
{
pcCU = rpcPic->getCU( iCUAddr );
pcCU->initCU( rpcPic, iCUAddr );
uiTileCol = rpcPic->getPicSym()->getTileIdxMap(iCUAddr) % (rpcPic->getPicSym()->getNumColumnsMinus1()+1); // what column of tiles are we in?
uiTileStartLCU = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iCUAddr))->getFirstCUAddr();
uiTileLCUX = uiTileStartLCU % uiWidthInLCUs;
uiCol = iCUAddr % uiWidthInLCUs;
// The 'line' is now relative to the 1st line in the slice, not the 1st line in the picture.
uiLin = (iCUAddr/uiWidthInLCUs)-(iStartCUAddr/uiWidthInLCUs);
// inherit from TR if necessary, select substream to use.
if( (pcSlice->getPPS()->getNumSubstreams() > 1) || ( depSliceSegmentsEnabled && (uiCol == uiTileLCUX)&&(pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag()) ))
{
// independent tiles => substreams are "per tile". iNumSubstreams has already been multiplied.
iNumSubstreamsPerTile = iNumSubstreams/rpcPic->getPicSym()->getNumTiles();
uiSubStrm = rpcPic->getPicSym()->getTileIdxMap(iCUAddr)*iNumSubstreamsPerTile
+ uiLin%iNumSubstreamsPerTile;
m_pcEntropyDecoder->setBitstream( ppcSubstreams[uiSubStrm] );
// Synchronize cabac probabilities with upper-right LCU if it's available and we're at the start of a line.
if (((pcSlice->getPPS()->getNumSubstreams() > 1) || depSliceSegmentsEnabled ) && (uiCol == uiTileLCUX)&&(pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag()))
{
// We'll sync if the TR is available.
TComDataCU *pcCUUp = pcCU->getCUAbove();
UInt uiWidthInCU = rpcPic->getFrameWidthInCU();
TComDataCU *pcCUTR = NULL;
if ( pcCUUp && ((iCUAddr%uiWidthInCU+1) < uiWidthInCU) )
{
pcCUTR = rpcPic->getCU( iCUAddr - uiWidthInCU + 1 );
}
UInt uiMaxParts = 1<<(pcSlice->getSPS()->getMaxCUDepth()<<1);
if ( (true/*bEnforceSliceRestriction*/ &&
((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) ||
((pcCUTR->getSCUAddr()+uiMaxParts-1) < pcSlice->getSliceCurStartCUAddr()) ||
((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(iCUAddr)))
))
)
{
// TR not available.
}
else
{
// TR is available, we use it.
pcSbacDecoders[uiSubStrm].loadContexts( &m_pcBufferSbacDecoders[uiTileCol] );
}
}
pcSbacDecoder->load(&pcSbacDecoders[uiSubStrm]); //this load is used to simplify the code (avoid to change all the call to pcSbacDecoders)
}
else if ( pcSlice->getPPS()->getNumSubstreams() <= 1 )
{
// Set variables to appropriate values to avoid later code change.
iNumSubstreamsPerTile = 1;
}
if ( (iCUAddr == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iCUAddr))->getFirstCUAddr()) && // 1st in tile.
(iCUAddr!=0) && (iCUAddr!=rpcPic->getPicSym()->getPicSCUAddr(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceCurStartCUAddr())/rpcPic->getNumPartInCU())
&& (iCUAddr!=rpcPic->getPicSym()->getPicSCUAddr(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceSegmentCurStartCUAddr())/rpcPic->getNumPartInCU())
) // !1st in frame && !1st in slice
{
if (pcSlice->getPPS()->getNumSubstreams() > 1)
{
// We're crossing into another tile, tiles are independent.
// When tiles are independent, we have "substreams per tile". Each substream has already been terminated, and we no longer
// have to perform it here.
// For TILES_DECODER, there can be a header at the start of the 1st substream in a tile. These are read when the substreams
// are extracted, not here.
}
else
{
SliceType sliceType = pcSlice->getSliceType();
if (pcSlice->getCabacInitFlag())
{
switch (sliceType)
{
case P_SLICE: // change initialization table to B_SLICE intialization
sliceType = B_SLICE;
break;
case B_SLICE: // change initialization table to P_SLICE intialization
sliceType = P_SLICE;
break;
default : // should not occur
assert(0);
}
}
m_pcEntropyDecoder->updateContextTables( sliceType, pcSlice->getSliceQp() );
}
}
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceEnable;
#endif
if ( pcSlice->getSPS()->getUseSAO() && (pcSlice->getSaoEnabledFlag()||pcSlice->getSaoEnabledFlagChroma()) )
{
SAOParam *saoParam = rpcPic->getPicSym()->getSaoParam();
saoParam->bSaoFlag[0] = pcSlice->getSaoEnabledFlag();
if (iCUAddr == iStartCUAddr)
{
saoParam->bSaoFlag[1] = pcSlice->getSaoEnabledFlagChroma();
}
Int numCuInWidth = saoParam->numCuInWidth;
Int cuAddrInSlice = iCUAddr - rpcPic->getPicSym()->getCUOrderMap(pcSlice->getSliceCurStartCUAddr()/rpcPic->getNumPartInCU());
Int cuAddrUpInSlice = cuAddrInSlice - numCuInWidth;
Int rx = iCUAddr % numCuInWidth;
Int ry = iCUAddr / numCuInWidth;
Int allowMergeLeft = 1;
Int allowMergeUp = 1;
if (rx!=0)
{
if (rpcPic->getPicSym()->getTileIdxMap(iCUAddr-1) != rpcPic->getPicSym()->getTileIdxMap(iCUAddr))
{
allowMergeLeft = 0;
}
}
if (ry!=0)
{
if (rpcPic->getPicSym()->getTileIdxMap(iCUAddr-numCuInWidth) != rpcPic->getPicSym()->getTileIdxMap(iCUAddr))
{
allowMergeUp = 0;
}
}
pcSbacDecoder->parseSaoOneLcuInterleaving(rx, ry, saoParam,pcCU, cuAddrInSlice, cuAddrUpInSlice, allowMergeLeft, allowMergeUp);
}
else if ( pcSlice->getSPS()->getUseSAO() )
{
Int addr = pcCU->getAddr();
SAOParam *saoParam = rpcPic->getPicSym()->getSaoParam();
for (Int cIdx=0; cIdx<3; cIdx++)
{
SaoLcuParam *saoLcuParam = &(saoParam->saoLcuParam[cIdx][addr]);
if ( ((cIdx == 0) && !pcSlice->getSaoEnabledFlag()) || ((cIdx == 1 || cIdx == 2) && !pcSlice->getSaoEnabledFlagChroma()))
{
saoLcuParam->mergeUpFlag = 0;
saoLcuParam->mergeLeftFlag = 0;
saoLcuParam->subTypeIdx = 0;
saoLcuParam->typeIdx = -1;
saoLcuParam->offset[0] = 0;
saoLcuParam->offset[1] = 0;
saoLcuParam->offset[2] = 0;
saoLcuParam->offset[3] = 0;
}
}
}
#if ENABLE_ANAYSIS_OUTPUT
UInt uiBefore = ppcSubstreams[uiSubStrm]->getByteLocation();
#endif
m_pcCuDecoder->decodeCU ( pcCU, uiIsLast );
#if ENABLE_ANAYSIS_OUTPUT
pcCU->getTotalBits() = ppcSubstreams[uiSubStrm]->getByteLocation() - uiBefore;
#endif
m_pcCuDecoder->decompressCU ( pcCU );
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceDisable;
#endif
pcSbacDecoders[uiSubStrm].load(pcSbacDecoder);
//Store probabilities of second LCU in line into buffer
if ( (uiCol == uiTileLCUX+1)&& (depSliceSegmentsEnabled || (pcSlice->getPPS()->getNumSubstreams() > 1)) && (pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag()) )
{
m_pcBufferSbacDecoders[uiTileCol].loadContexts( &pcSbacDecoders[uiSubStrm] );
}
if( uiIsLast && depSliceSegmentsEnabled )
{
if (pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
{
CTXMem[1]->loadContexts( &m_pcBufferSbacDecoders[uiTileCol] );//ctx 2.LCU
}
CTXMem[0]->loadContexts( pcSbacDecoder );//ctx end of dep.slice
return;
}
}
}
Void TComPic::createNonDBFilterInfo(UInt* puiSliceStartAddress, Int iNumSlices, Int iSliceGranularityDepth
,Bool bNDBFilterCrossSliceBoundary
,Int iNumTiles
,Bool bNDBFilterCrossTileBoundary)
{
UInt uiMaxNumSUInLCU = getNumPartInCU();
UInt uiNumLCUInPic = getNumCUsInFrame();
UInt uiPicWidth = getCU(0)->getSlice()->getSPS()->getWidth();
UInt uiPicHeight = getCU(0)->getSlice()->getSPS()->getHeight();
Int iNumLCUsInPicWidth = getFrameWidthInCU();
Int iNumLCUsInPicHeight= getFrameHeightInCU();
UInt uiMaxNumSUInLCUWidth = getNumPartInWidth();
UInt uiMAxNumSUInLCUHeight= getNumPartInHeight();
m_bIndependentSliceBoundaryForNDBFilter = (bNDBFilterCrossSliceBoundary)?(false):((iNumSlices > 1)?(true):(false)) ;
m_iSliceGranularityForNDBFilter = iSliceGranularityDepth;
m_bIndependentTileBoundaryForNDBFilter = (bNDBFilterCrossTileBoundary)?(false) :((iNumTiles > 1)?(true):(false));
m_pbValidSlice = new Bool[iNumSlices];
for(Int s=0; s< iNumSlices; s++)
{
m_pbValidSlice[s] = true;
}
if( puiSliceStartAddress == NULL || (iNumSlices == 1 && iNumTiles == 1) )
{
return;
}
m_piSliceSUMap = new Int[uiMaxNumSUInLCU * uiNumLCUInPic];
//initialization
for(UInt i=0; i< (uiMaxNumSUInLCU * uiNumLCUInPic); i++ )
{
m_piSliceSUMap[i] = -1;
}
for( UInt uiCUAddr = 0; uiCUAddr < uiNumLCUInPic ; uiCUAddr++ )
{
TComDataCU* pcCU = getCU( uiCUAddr );
pcCU->setSliceSUMap(m_piSliceSUMap + (uiCUAddr* uiMaxNumSUInLCU));
pcCU->getNDBFilterBlocks()->clear();
}
m_vSliceCUDataLink.clear();
m_vSliceCUDataLink.resize(iNumSlices);
UInt uiStartAddr, uiEndAddr, uiFirstCUInStartLCU, uiStartLCU, uiEndLCU, uiLastCUInEndLCU, uiAddr;
UInt uiLPelX, uiTPelY, uiLCUX, uiLCUY;
UInt uiCurrSU;
UInt uiStartSU, uiEndSU;
for(Int s=0; s< iNumSlices; s++)
{
//1st step: decide the real start address
#if FINE_GRANULARITY_SLICES
uiStartAddr = puiSliceStartAddress[s];
uiEndAddr = puiSliceStartAddress[s+1] -1;
#else
uiStartAddr = (puiSliceStartAddress[s]*uiMaxNumSUInLCU);
uiEndAddr = (puiSliceStartAddress[s+1]*uiMaxNumSUInLCU) -1;
#endif
uiStartLCU = uiStartAddr / uiMaxNumSUInLCU;
uiFirstCUInStartLCU = uiStartAddr % uiMaxNumSUInLCU;
uiEndLCU = uiEndAddr / uiMaxNumSUInLCU;
uiLastCUInEndLCU = uiEndAddr % uiMaxNumSUInLCU;
#if TILES
uiAddr = m_apcPicSym->getCUOrderMap(uiStartLCU);
#else
uiAddr = uiStartLCU;
#endif
uiLCUX = getCU(uiAddr)->getCUPelX();
uiLCUY = getCU(uiAddr)->getCUPelY();
uiLPelX = uiLCUX + g_auiRasterToPelX[ g_auiZscanToRaster[uiFirstCUInStartLCU] ];
uiTPelY = uiLCUY + g_auiRasterToPelY[ g_auiZscanToRaster[uiFirstCUInStartLCU] ];
uiCurrSU = uiFirstCUInStartLCU;
Bool bMoveToNextLCU = false;
Bool bSliceInOneLCU = (uiStartLCU == uiEndLCU);
while(!( uiLPelX < uiPicWidth ) || !( uiTPelY < uiPicHeight ))
{
uiCurrSU ++;
if(bSliceInOneLCU)
{
if(uiCurrSU > uiLastCUInEndLCU)
{
m_pbValidSlice[s] = false;
break;
}
}
if(uiCurrSU >= uiMaxNumSUInLCU )
{
bMoveToNextLCU = true;
break;
}
uiLPelX = uiLCUX + g_auiRasterToPelX[ g_auiZscanToRaster[uiCurrSU] ];
uiTPelY = uiLCUY + g_auiRasterToPelY[ g_auiZscanToRaster[uiCurrSU] ];
}
if(!m_pbValidSlice[s])
{
continue;
}
if(uiCurrSU != uiFirstCUInStartLCU)
{
if(!bMoveToNextLCU)
{
uiFirstCUInStartLCU = uiCurrSU;
}
else
{
uiStartLCU++;
uiFirstCUInStartLCU = 0;
assert( uiStartLCU < getNumCUsInFrame());
}
assert(uiStartLCU*uiMaxNumSUInLCU + uiFirstCUInStartLCU < uiEndAddr);
}
//2nd step: assign NonDBFilterInfo to each processing block
for(UInt i= uiStartLCU; i <= uiEndLCU; i++)
{
uiStartSU = (i == uiStartLCU)?(uiFirstCUInStartLCU):(0);
uiEndSU = (i == uiEndLCU )?(uiLastCUInEndLCU ):(uiMaxNumSUInLCU -1);
#if TILES
uiAddr = m_apcPicSym->getCUOrderMap(i);
Int iTileID= m_apcPicSym->getTileIdxMap(uiAddr);
#else
uiAddr = i;
#endif
TComDataCU* pcCU = getCU(uiAddr);
m_vSliceCUDataLink[s].push_back(pcCU);
#if TILES
createNonDBFilterInfoLCU(iTileID, s, pcCU, uiStartSU, uiEndSU, m_iSliceGranularityForNDBFilter, uiPicWidth, uiPicHeight);
#else
createNonDBFilterInfoLCU(s, pcCU, uiStartSU, uiEndSU, m_iSliceGranularityForNDBFilter, uiPicWidth, uiPicHeight);
#endif
}
}
//step 3: border availability
for(Int s=0; s< iNumSlices; s++)
{
if(!m_pbValidSlice[s])
{
continue;
}
for(Int i=0; i< m_vSliceCUDataLink[s].size(); i++)
{
TComDataCU* pcCU = m_vSliceCUDataLink[s][i];
uiAddr = pcCU->getAddr();
Int iTileID= m_apcPicSym->getTileIdxMap(uiAddr);
Bool bTopTileBoundary = false, bDownTileBoundary= false, bLeftTileBoundary= false, bRightTileBoundary= false;
if(m_bIndependentTileBoundaryForNDBFilter)
{
//left
if( uiAddr % iNumLCUsInPicWidth != 0)
{
bLeftTileBoundary = ( m_apcPicSym->getTileIdxMap(uiAddr -1) != iTileID )?true:false;
}
//right
if( (uiAddr % iNumLCUsInPicWidth) != (iNumLCUsInPicWidth -1) )
{
bRightTileBoundary = ( m_apcPicSym->getTileIdxMap(uiAddr +1) != iTileID)?true:false;
}
//top
if( uiAddr >= iNumLCUsInPicWidth)
{
bTopTileBoundary = (m_apcPicSym->getTileIdxMap(uiAddr - iNumLCUsInPicWidth) != iTileID )?true:false;
}
//down
if( uiAddr + iNumLCUsInPicWidth < uiNumLCUInPic )
{
bDownTileBoundary = (m_apcPicSym->getTileIdxMap(uiAddr + iNumLCUsInPicWidth) != iTileID)?true:false;
}
}
pcCU->setNDBFilterBlockBorderAvailability(iNumLCUsInPicWidth, iNumLCUsInPicHeight, uiMaxNumSUInLCUWidth, uiMAxNumSUInLCUHeight,uiPicWidth, uiPicHeight
,m_bIndependentSliceBoundaryForNDBFilter
,bTopTileBoundary, bDownTileBoundary, bLeftTileBoundary, bRightTileBoundary
,m_bIndependentTileBoundaryForNDBFilter);
}
}
if( m_bIndependentSliceBoundaryForNDBFilter || m_bIndependentTileBoundaryForNDBFilter)
{
m_pcNDBFilterYuvTmp = new TComPicYuv();
m_pcNDBFilterYuvTmp->create(uiPicWidth, uiPicHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth);
}
}
Void TDecSlice::decompressSlice(TComInputBitstream* pcBitstream, TComInputBitstream** ppcSubstreams, TComPic*& rpcPic, TDecSbac* pcSbacDecoder, TDecSbac* pcSbacDecoders)
{
TComDataCU* pcCU;
UInt uiIsLast = 0;
Int iStartCUEncOrder = max(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceCurStartCUAddr()/rpcPic->getNumPartInCU(), rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getEntropySliceCurStartCUAddr()/rpcPic->getNumPartInCU());
Int iStartCUAddr = rpcPic->getPicSym()->getCUOrderMap(iStartCUEncOrder);
// decoder don't need prediction & residual frame buffer
rpcPic->setPicYuvPred( 0 );
rpcPic->setPicYuvResi( 0 );
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceEnable;
#endif
DTRACE_CABAC_VL( g_nSymbolCounter++ );
DTRACE_CABAC_T( "\tPOC: " );
DTRACE_CABAC_V( rpcPic->getPOC() );
DTRACE_CABAC_T( "\n" );
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceDisable;
#endif
UInt uiTilesAcross = rpcPic->getPicSym()->getNumColumnsMinus1()+1;
TComSlice* pcSlice = rpcPic->getSlice(rpcPic->getCurrSliceIdx());
UInt iSymbolMode = pcSlice->getPPS()->getEntropyCodingMode();
Int iNumSubstreams = pcSlice->getPPS()->getNumSubstreams();
if( iSymbolMode )
{
m_pcBufferSbacDecoders = new TDecSbac [uiTilesAcross];
m_pcBufferBinCABACs = new TDecBinCABAC[uiTilesAcross];
for (UInt ui = 0; ui < uiTilesAcross; ui++)
{
m_pcBufferSbacDecoders[ui].init(&m_pcBufferBinCABACs[ui]);
}
//save init. state
for (UInt ui = 0; ui < uiTilesAcross; ui++)
{
m_pcBufferSbacDecoders[ui].load(pcSbacDecoder);
}
}
if( iSymbolMode )
{
m_pcBufferLowLatSbacDecoders = new TDecSbac [uiTilesAcross];
m_pcBufferLowLatBinCABACs = new TDecBinCABAC[uiTilesAcross];
for (UInt ui = 0; ui < uiTilesAcross; ui++)
m_pcBufferLowLatSbacDecoders[ui].init(&m_pcBufferLowLatBinCABACs[ui]);
//save init. state
for (UInt ui = 0; ui < uiTilesAcross; ui++)
m_pcBufferLowLatSbacDecoders[ui].load(pcSbacDecoder);
}
UInt uiWidthInLCUs = rpcPic->getPicSym()->getFrameWidthInCU();
//UInt uiHeightInLCUs = rpcPic->getPicSym()->getFrameHeightInCU();
UInt uiCol=0, uiLin=0, uiSubStrm=0;
UInt uiTileCol;
UInt uiTileStartLCU;
UInt uiTileLCUX;
UInt uiTileLCUY;
UInt uiTileWidth;
UInt uiTileHeight;
Int iNumSubstreamsPerTile = 1; // if independent.
for( Int iCUAddr = iStartCUAddr; !uiIsLast && iCUAddr < rpcPic->getNumCUsInFrame(); iCUAddr = rpcPic->getPicSym()->xCalculateNxtCUAddr(iCUAddr) )
{
pcCU = rpcPic->getCU( iCUAddr );
pcCU->initCU( rpcPic, iCUAddr );
uiTileCol = rpcPic->getPicSym()->getTileIdxMap(iCUAddr) % (rpcPic->getPicSym()->getNumColumnsMinus1()+1); // what column of tiles are we in?
uiTileStartLCU = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iCUAddr))->getFirstCUAddr();
uiTileLCUX = uiTileStartLCU % uiWidthInLCUs;
uiTileLCUY = uiTileStartLCU / uiWidthInLCUs;
uiTileWidth = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iCUAddr))->getTileWidth();
uiTileHeight = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iCUAddr))->getTileHeight();
uiCol = iCUAddr % uiWidthInLCUs;
uiLin = iCUAddr / uiWidthInLCUs;
// inherit from TR if necessary, select substream to use.
if( iSymbolMode && pcSlice->getPPS()->getNumSubstreams() > 1 )
{
if (pcSlice->getPPS()->getNumSubstreams() > 1)
{
// independent tiles => substreams are "per tile". iNumSubstreams has already been multiplied.
iNumSubstreamsPerTile = iNumSubstreams/rpcPic->getPicSym()->getNumTiles();
uiSubStrm = rpcPic->getPicSym()->getTileIdxMap(iCUAddr)*iNumSubstreamsPerTile
+ uiLin%iNumSubstreamsPerTile;
}
else
{
// dependent tiles => substreams are "per frame".
uiSubStrm = uiLin % iNumSubstreams;
}
m_pcEntropyDecoder->setBitstream( ppcSubstreams[uiSubStrm] );
// Synchronize cabac probabilities with upper-right LCU if it's available and we're at the start of a line.
if (pcSlice->getPPS()->getNumSubstreams() > 1 && uiCol == uiTileLCUX)
{
// We'll sync if the TR is available.
TComDataCU *pcCUUp = pcCU->getCUAbove();
UInt uiWidthInCU = rpcPic->getFrameWidthInCU();
TComDataCU *pcCUTR = NULL;
if ( pcCUUp && ((iCUAddr%uiWidthInCU+1) < uiWidthInCU) )
{
pcCUTR = rpcPic->getCU( iCUAddr - uiWidthInCU + 1 );
}
UInt uiMaxParts = 1<<(pcSlice->getSPS()->getMaxCUDepth()<<1);
if ( (true/*bEnforceSliceRestriction*/ &&
((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) ||
((pcCUTR->getSCUAddr()+uiMaxParts-1) < pcSlice->getSliceCurStartCUAddr()) ||
((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(iCUAddr)))
))||
(true/*bEnforceEntropySliceRestriction*/ &&
((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) ||
((pcCUTR->getSCUAddr()+uiMaxParts-1) < pcSlice->getEntropySliceCurStartCUAddr()) ||
((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(iCUAddr)))
))
)
{
// TR not available.
}
else
{
// TR is available, we use it.
pcSbacDecoders[uiSubStrm].loadContexts( &m_pcBufferSbacDecoders[uiTileCol] );
}
}
pcSbacDecoder->load(&pcSbacDecoders[uiSubStrm]); //this load is used to simplify the code (avoid to change all the call to pcSbacDecoders)
}
else if ( iSymbolMode && pcSlice->getPPS()->getNumSubstreams() <= 1 )
{
// Set variables to appropriate values to avoid later code change.
iNumSubstreamsPerTile = 1;
}
if ( (iCUAddr == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(iCUAddr))->getFirstCUAddr()) && // 1st in tile.
(iCUAddr!=0) && (iCUAddr!=rpcPic->getPicSym()->getPicSCUAddr(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceCurStartCUAddr())/rpcPic->getNumPartInCU())) // !1st in frame && !1st in slice
{
if (pcSlice->getPPS()->getNumSubstreams() > 1)
{
// We're crossing into another tile, tiles are independent.
// When tiles are independent, we have "substreams per tile". Each substream has already been terminated, and we no longer
// have to perform it here.
// For TILES_DECODER, there can be a header at the start of the 1st substream in a tile. These are read when the substreams
// are extracted, not here.
}
else
{
#if CABAC_INIT_FLAG
SliceType sliceType = pcSlice->getSliceType();
if (pcSlice->getCabacInitFlag())
{
switch (sliceType)
{
case P_SLICE: // change initialization table to B_SLICE intialization
sliceType = B_SLICE;
break;
case B_SLICE: // change initialization table to P_SLICE intialization
sliceType = P_SLICE;
break;
default : // should not occur
assert(0);
}
}
m_pcEntropyDecoder->updateContextTables( sliceType, pcSlice->getSliceQp() );
#else
m_pcEntropyDecoder->updateContextTables( pcSlice->getSliceType(), pcSlice->getSliceQp() );
#endif
}
Bool bTileMarkerFoundFlag = false;
TComInputBitstream *pcTmpPtr;
pcTmpPtr = ppcSubstreams[uiSubStrm]; // for CABAC
for (UInt uiIdx=0; uiIdx<pcTmpPtr->getTileMarkerLocationCount(); uiIdx++)
{
if ( pcTmpPtr->getByteLocation() == (pcTmpPtr->getTileMarkerLocation( uiIdx )+2) )
{
bTileMarkerFoundFlag = true;
break;
}
}
if (bTileMarkerFoundFlag)
{
UInt uiTileIdx;
// Read tile index
m_pcEntropyDecoder->readTileMarker( uiTileIdx, rpcPic->getPicSym()->getBitsUsedByTileIdx() );
}
}
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceEnable;
#endif
if ( pcSlice->getSPS()->getUseSAO() && pcSlice->getSaoInterleavingFlag() && pcSlice->getSaoEnabledFlag() )
{
pcSlice->getAPS()->getSaoParam()->bSaoFlag[0] = pcSlice->getSaoEnabledFlag();
if (iCUAddr == iStartCUAddr)
{
pcSlice->getAPS()->getSaoParam()->bSaoFlag[1] = pcSlice->getSaoEnabledFlagCb();
pcSlice->getAPS()->getSaoParam()->bSaoFlag[2] = pcSlice->getSaoEnabledFlagCr();
}
Int numCuInWidth = pcSlice->getAPS()->getSaoParam()->numCuInWidth;
Int cuAddrInSlice = iCUAddr - pcSlice->getSliceCurStartCUAddr()/rpcPic->getNumPartInCU();
Int cuAddrUpInSlice = cuAddrInSlice - numCuInWidth;
Int rx = iCUAddr % numCuInWidth;
Int ry = iCUAddr / numCuInWidth;
pcSbacDecoder->parseSaoOneLcuInterleaving(rx, ry, pcSlice->getAPS()->getSaoParam(),pcCU, cuAddrInSlice, cuAddrUpInSlice, pcSlice->getSPS()->getLFCrossSliceBoundaryFlag() );
}
m_pcCuDecoder->decodeCU ( pcCU, uiIsLast );
m_pcCuDecoder->decompressCU ( pcCU );
#if ENC_DEC_TRACE
g_bJustDoIt = g_bEncDecTraceDisable;
#endif
if( iSymbolMode )
{
/*If at the end of a LCU line but not at the end of a substream, perform CABAC flush*/
if (!uiIsLast && pcSlice->getPPS()->getNumSubstreams() > 1)
{
if ((uiCol == uiTileLCUX+uiTileWidth-1) && (uiLin+iNumSubstreamsPerTile < uiTileLCUY+uiTileHeight))
{
m_pcEntropyDecoder->decodeFlush();
}
}
pcSbacDecoders[uiSubStrm].load(pcSbacDecoder);
//Store probabilities of second LCU in line into buffer
if (pcSlice->getPPS()->getNumSubstreams() > 1 && (uiCol == uiTileLCUX+1))
{
m_pcBufferSbacDecoders[uiTileCol].loadContexts( &pcSbacDecoders[uiSubStrm] );
}
}
}
}
