gopdecoder.cpp
来自「JVT-S203 contains the JSVM 6 reference s」· C++ 代码 · 共 2,153 行 · 第 1/5 页
CPP
2,153 行
{
if((*iter)->getPoc()==uiPoc)
{
pNext = (*iter);
}
}
rcList1.add( pNext->getFrame() );
}
return Err::m_nOK;
}
//TMM_EC }}
ErrVal
DecodedPicBuffer::setPrdRefLists( DPBUnit* pcCurrDPBUnit )
{
ROF( m_pcCurrDPBUnit == pcCurrDPBUnit );
ROF( m_pcCurrDPBUnit->getCtrlData().getSliceHeader() );
RefFrameList& rcList0 = m_pcCurrDPBUnit->getCtrlData().getPrdFrameList( LIST_0 );
RefFrameList& rcList1 = m_pcCurrDPBUnit->getCtrlData().getPrdFrameList( LIST_1 );
rcList0.reset();
rcList1.reset();
ROTRS( m_pcCurrDPBUnit->getCtrlData().getSliceHeader()->isIntra(), Err::m_nOK );
if( m_pcCurrDPBUnit->getCtrlData().getSliceHeader()->isInterP() )
{
RNOK( xInitPrdListPSlice( rcList0 ) );
RNOK( xPrdListRemapping ( rcList0, LIST_0, m_pcCurrDPBUnit->getCtrlData().getSliceHeader() ) );
RNOK( xDumpRefList( LIST_0, rcList0 ) );
}
else
{
RNOK( xInitPrdListsBSlice( rcList0, rcList1 ) );
RNOK( xPrdListRemapping ( rcList0, LIST_0, m_pcCurrDPBUnit->getCtrlData().getSliceHeader() ) );
RNOK( xPrdListRemapping ( rcList1, LIST_1, m_pcCurrDPBUnit->getCtrlData().getSliceHeader() ) );
RNOK( xDumpRefList( LIST_0, rcList0 ) );
RNOK( xDumpRefList( LIST_1, rcList1 ) );
}
return Err::m_nOK;
}
ErrVal
DecodedPicBuffer::store( DPBUnit*& rpcDPBUnit,
PicBufferList& rcOutputList,
PicBufferList& rcUnusedList,
IntFrame* pcFrameBaseRep )
{
RNOK( xStorePicture( rpcDPBUnit, rcOutputList, rcUnusedList,
rpcDPBUnit->getCtrlData().getSliceHeader()->isIdrNalUnit() ) );
if( rpcDPBUnit->isNeededForRef() )
{
m_uiLastRefFrameNum = rpcDPBUnit->getFrameNum();
}
ROFRS( pcFrameBaseRep, Err::m_nOK );
//===== store base representation =====
//--- get DPB unit ---
if( m_cFreeDPBUnitList.empty() )
{
// not sure whether this always works ...
RNOK( xOutput( rcOutputList, rcUnusedList ) );
}
DPBUnit* pcBaseRep = m_cFreeDPBUnitList.popFront();
//--- init unit and extend picture ---
RNOK( pcBaseRep->initBase( *rpcDPBUnit, pcFrameBaseRep ) );
RNOK( m_pcYuvBufferCtrl->initMb() );
RNOK( pcBaseRep->getFrame()->extendFrame( NULL ) );
//--- store just before normal representation of the same picture
DPBUnitList::iterator iter = m_cUsedDPBUnitList.begin();
DPBUnitList::iterator end = m_cUsedDPBUnitList.end ();
for( ; iter != end; iter++ )
{
if( (*iter) == rpcDPBUnit )
{
break;
}
}
ROT( iter == end );
m_cUsedDPBUnitList.insert( iter, pcBaseRep );
RNOK( xDumpDPB() );
//===== reset DPB unit =====
rpcDPBUnit = 0;
return Err::m_nOK;
}
ErrVal
DecodedPicBuffer::update( DPBUnit* pcDPBUnit )
{
ROF( pcDPBUnit );
//---- fill border ----
RNOK( m_pcYuvBufferCtrl->initMb() );
RNOK( pcDPBUnit->getFrame()->extendFrame( NULL ) );
return Err::m_nOK;
}
//////////////////////////////////////////////////////////////////////////
// MCTF DECODER
//////////////////////////////////////////////////////////////////////////
MCTFDecoder::MCTFDecoder()
: m_pcH264AVCDecoder ( 0 )
, m_pcSliceReader ( 0 )
, m_pcSliceDecoder ( 0 )
, m_pcNalUnitParser ( 0 )
, m_pcControlMng ( 0 )
, m_pcLoopFilter ( 0 )
, m_pcHeaderSymbolReadIf ( 0 )
, m_pcParameterSetMng ( 0 )
, m_pcPocCalculator ( 0 )
, m_pcYuvFullPelBufferCtrl ( 0 )
, m_pcDecodedPictureBuffer ( 0 )
, m_pcMotionCompensation ( 0 )
, m_pcQuarterPelFilter ( 0 )
, m_bInitDone ( false )
, m_bCreateDone ( false )
, m_bWaitForIdr ( true )
, m_bReconstructAll ( false )
, m_uiFrameWidthInMb ( 0 )
, m_uiFrameHeightInMb ( 0 )
, m_uiMbNumber ( 0 )
, m_pcResidual ( 0 )
, m_pcILPrediction ( 0 )
, m_pcPredSignal ( 0 )
, m_pcBaseLayerResidual ( 0 )
, m_pcBaseLayerFrame ( 0 )
, m_pcBaseLayerCtrl ( 0 )
, m_pcCurrDPBUnit ( 0 )
, m_uiLayerId ( 0 )
, m_bActive ( false )
, m_uiQualityLevelForPrediction ( 3 )
#if MULTIPLE_LOOP_DECODING
, m_bCompletelyDecodeLayer ( false )
#endif
, m_pcResizeParameter ( 0 ) //TMM_ESS
, m_iMbProcessed (-1) //--ICU/ETRI FMO Implementation
, m_bIsNewPic (true)
{
::memset( m_apcFrameTemp, 0x00, sizeof( m_apcFrameTemp ) );
//TMM_EC {{
m_pcVeryFirstSliceHeader = NULL;
m_bBaseLayerLost = false;
//TMM_EC }}
}
MCTFDecoder::~MCTFDecoder()
{
}
ErrVal
MCTFDecoder::create( MCTFDecoder*& rpcMCTFDecoder )
{
rpcMCTFDecoder = new MCTFDecoder;
ROT( NULL == rpcMCTFDecoder );
return Err::m_nOK;
}
ErrVal
MCTFDecoder::destroy()
{
ROT( m_bInitDone );
delete this;
return Err::m_nOK;
}
ErrVal
MCTFDecoder::init( H264AVCDecoder* pcH264AVCDecoder,
SliceReader* pcSliceReader,
SliceDecoder* pcSliceDecoder,
RQFGSDecoder* pcRQFGSDecoder,
NalUnitParser* pcNalUnitParser,
ControlMngIf* pcControlMng,
LoopFilter* pcLoopFilter,
HeaderSymbolReadIf* pcHeaderSymbolReadIf,
ParameterSetMng* pcParameterSetMng,
PocCalculator* pcPocCalculator,
YuvBufferCtrl* pcYuvFullPelBufferCtrl,
DecodedPicBuffer* pcDecodedPictureBuffer,
MotionCompensation* pcMotionCompensation,
QuarterPelFilter* pcQuarterPelFilter )
{
ROT( NULL == pcH264AVCDecoder );
ROT( NULL == pcSliceReader );
ROT( NULL == pcSliceDecoder );
ROT( NULL == pcNalUnitParser );
ROT( NULL == pcControlMng );
ROT( NULL == pcLoopFilter );
ROT( NULL == pcHeaderSymbolReadIf );
ROT( NULL == pcParameterSetMng );
ROT( NULL == pcPocCalculator );
ROT( NULL == pcQuarterPelFilter );
ROT( NULL == pcYuvFullPelBufferCtrl );
ROT( NULL == pcDecodedPictureBuffer );
ROT( NULL == pcMotionCompensation );
ROT( NULL == pcRQFGSDecoder );
m_pcH264AVCDecoder = pcH264AVCDecoder;
m_pcSliceReader = pcSliceReader;
m_pcSliceDecoder = pcSliceDecoder ;
m_pcNalUnitParser = pcNalUnitParser;
m_pcControlMng = pcControlMng;
m_pcLoopFilter = pcLoopFilter;
m_pcHeaderSymbolReadIf = pcHeaderSymbolReadIf;
m_pcParameterSetMng = pcParameterSetMng;
m_pcPocCalculator = pcPocCalculator;
m_pcYuvFullPelBufferCtrl = pcYuvFullPelBufferCtrl;
m_pcDecodedPictureBuffer = pcDecodedPictureBuffer;
m_pcMotionCompensation = pcMotionCompensation;
m_pcQuarterPelFilter = pcQuarterPelFilter;
m_pcRQFGSDecoder = pcRQFGSDecoder;
m_bInitDone = true;
m_bCreateDone = false;
m_bWaitForIdr = true;
m_bActive = false;
m_uiFrameWidthInMb = 0;
m_uiFrameHeightInMb = 0;
m_uiMbNumber = 0;
m_pcResidual = 0;
m_pcILPrediction = 0;
m_pcBaseLayerFrame = 0;
m_pcBaseLayerResidual = 0;
m_pcPredSignal = 0;
m_pcBaseLayerCtrl = 0;
m_pcCurrDPBUnit = 0;
m_uiLayerId = 0;
m_iMbProcessed = -1;
m_bIsNewPic = true;
return Err::m_nOK;
}
ErrVal
MCTFDecoder::uninit()
{
m_pcSliceReader = NULL;
m_pcSliceDecoder = NULL;
m_pcNalUnitParser = NULL;
m_pcControlMng = NULL;
m_pcLoopFilter = NULL;
m_pcHeaderSymbolReadIf = NULL;
m_pcParameterSetMng = NULL;
m_pcPocCalculator = NULL;
m_pcYuvFullPelBufferCtrl = NULL;
m_pcMotionCompensation = NULL;
m_uiFrameWidthInMb = 0;
m_uiFrameHeightInMb = 0;
// {{ TMM_EC
delete m_pcVeryFirstSliceHeader;
m_pcVeryFirstSliceHeader = NULL;
// }}TMM_EC
RNOK( xDeleteData() );
m_bInitDone = false;
return Err::m_nOK;
}
ErrVal
MCTFDecoder::initSlice0( SliceHeader* rcSH )
{
ROTRS( m_bActive, Err::m_nOK );
//===== get and set relevant parameters =====
m_uiLayerId = rcSH->getLayerId();
m_uiFrameWidthInMb = rcSH->getSPS().getFrameWidthInMbs();
m_uiFrameHeightInMb = rcSH->getSPS().getFrameHeightInMbs();
m_uiMbNumber = m_uiFrameWidthInMb * m_uiFrameHeightInMb;
//===== re-allocate dynamic memory =====
RNOK( xDeleteData() );
RNOK( xCreateData( rcSH->getSPS() ) );
//===== initialize DPB =====
RNOK( m_pcDecodedPictureBuffer->initSPS( rcSH->getSPS() ) );
//===== initialize some parameters =====
m_bActive = true;
m_bInitDone = true;
return Err::m_nOK;
}
ErrVal
MCTFDecoder::process( SliceHeader*& rpcSliceHeader,
PicBuffer* pcPicBuffer,
PicBufferList& rcPicBufferOutputList,
PicBufferList& rcPicBufferUnusedList,
Bool bReconstructionLayer )
{
ROF ( m_bInitDone );
ROTRS( m_bWaitForIdr && !rpcSliceHeader->isIdrNalUnit(), Err::m_nOK );
m_bWaitForIdr = false;
//TMM_EC {{
if( NULL == m_pcVeryFirstSliceHeader )
{
m_pcVeryFirstSliceHeader = new SliceHeader( rpcSliceHeader->getSPS(), rpcSliceHeader->getPPS() );
}
//TMM_EC }}
//===== decoding =====
if( rpcSliceHeader->getSliceType() == F_SLICE )
{
RNOK( xDecodeFGSRefinement( rpcSliceHeader ) );
}
else
{
RNOK( xDecodeBaseRepresentation( rpcSliceHeader, pcPicBuffer, rcPicBufferOutputList, rcPicBufferUnusedList, bReconstructionLayer ) );
}
//===== clear unused pic buffer ====
if( pcPicBuffer )
{
rcPicBufferUnusedList.push_back( pcPicBuffer );
}
//===== delete slice header (if not stored) =====
delete rpcSliceHeader;
rpcSliceHeader = NULL;
return Err::m_nOK;
}
ErrVal
MCTFDecoder::finishProcess( PicBufferList& rcPicBufferOutputList,
PicBufferList& rcPicBufferUnusedList,
Int& riMaxPoc )
{
RNOK( m_pcDecodedPictureBuffer->clear( rcPicBufferOutputList, rcPicBufferUnusedList, riMaxPoc ) );
return Err::m_nOK;
}
ErrVal
MCTFDecoder::getBaseLayerPWTable( SliceHeader::PredWeightTable*& rpcPredWeightTable,
ListIdx eListIdx,
Int iPoc )
{
DPBUnit* pcBaseDPBUnit = m_pcDecodedPictureBuffer->getDPBUnit( iPoc );
ROF( pcBaseDPBUnit );
SliceHeader* pcSliceHeader = pcBaseDPBUnit->getCtrlData().getSliceHeader();
ROF( pcSliceHeader );
rpcPredWeightTable = &pcSliceHeader->getPredWeightTable( eListIdx );
return Err::m_nOK;
}
ErrVal
MCTFDecoder::getBaseLayerData ( IntFrame*& pcFrame,
IntFrame*& pcResidual,
MbDataCtrl*& pcMbDataCtrl,
Bool& rbConstrainedIPred,
Bool bSpatialScalability,
Int iPoc )
{
pcFrame = 0;
pcResidual = 0;
pcMbDataCtrl = 0;
SliceHeader* pcSliceHeader = 0;
DPBUnit* pcBaseDPBUnit = m_pcDecodedPictureBuffer->getDPBUnit( iPoc );
ROF( pcBaseDPBUnit );
pcFrame = m_pcILPrediction;
pcResidual = m_pcResidual;
pcMbDataCtrl = pcBaseDPBUnit->getCtrlData().getMbDataCtrl ();
pcSliceHeader = pcBaseDPBUnit->getCtrlData().getSliceHeader ();
rbConstrainedIPred = pcBaseDPBUnit->isConstrIPred();
if( bSpatialScalability )
{
RNOK( m_apcFrameTemp[0]->copy( pcFrame ) );
pcFrame = m_apcFrameTemp[0];
#if MULTIPLE_LOOP_DECODING
if( pcSliceHeader->getPPS().getConstrainedIntraPredFlag() && !m_bCompletelyDecodeLayer )
#else
if( pcSliceHeader->getPPS().getConstrainedIntraPredFlag() )
#endif
{
m_pcLoopFilter->setFilterMode( LoopFilter::LFMode( LoopFilter::LFM_NO_INTER_FILTER + LoopFilter::LFM_EXTEND_INTRA_SUR ) );
RNOK( m_pcLoopFilter->process(*pcSliceHeader,
pcFrame,
pcMbDataCtrl,
pcMbDataCtrl,
m_uiFrameWidthInMb,
NULL,
NULL,
false,
⌨️ 快捷键说明
复制代码Ctrl + C
搜索代码Ctrl + F
全屏模式F11
增大字号Ctrl + =
减小字号Ctrl + -
显示快捷键?