/* The copyright in this software is being made available under the BSD * License, included below. This software may be subject to other third party * and contributor rights, including patent rights, and no such rights are * granted under this license. * * Copyright (c) 2010-2014, ITU/ISO/IEC * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may * be used to endorse or promote products derived from this software without * specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF * THE POSSIBILITY OF SUCH DAMAGE. */ /** \file TAppDecTop.cpp \brief Decoder application class */ #include #include #include #include #include #include #include "TAppDecTop.h" #include "TLibDecoder/AnnexBread.h" #include "TLibDecoder/NALread.h" //! \ingroup TAppDecoder //! \{ // ==================================================================================================================== // Constructor / destructor / initialization / destroy // ==================================================================================================================== #if SVC_EXTENSION TAppDecTop::TAppDecTop() { for(UInt layer=0; layer < MAX_LAYERS; layer++) { m_aiPOCLastDisplay[layer] = -MAX_INT; m_apcTDecTop[layer] = &m_acTDecTop[layer]; } } #else TAppDecTop::TAppDecTop() : m_iPOCLastDisplay(-MAX_INT) { } #endif Void TAppDecTop::create() { } Void TAppDecTop::destroy() { if (m_pchBitstreamFile) { free (m_pchBitstreamFile); m_pchBitstreamFile = NULL; } #if SVC_EXTENSION for( Int i = 0; i <= m_tgtLayerId; i++ ) { if( m_pchReconFile[i] ) { free ( m_pchReconFile[i] ); m_pchReconFile[i] = NULL; } } #if AVC_BASE if( m_pchBLReconFile ) { free ( m_pchBLReconFile ); m_pchBLReconFile = NULL; } #endif #else if (m_pchReconFile) { free (m_pchReconFile); m_pchReconFile = NULL; } #endif #if AVC_SYNTAX || SYNTAX_OUTPUT if( m_pchBLSyntaxFile ) { free ( m_pchBLSyntaxFile ); m_pchBLSyntaxFile = NULL; } #endif } // ==================================================================================================================== // Public member functions // ==================================================================================================================== /** - create internal class - initialize internal class - until the end of the bitstream, call decoding function in TDecTop class - delete allocated buffers - destroy internal class . */ #if SVC_EXTENSION Void TAppDecTop::decode() { Int poc; TComList* pcListPic = NULL; ifstream bitstreamFile(m_pchBitstreamFile, ifstream::in | ifstream::binary); if (!bitstreamFile) { fprintf(stderr, "\nfailed to open bitstream file `%s' for reading\n", m_pchBitstreamFile); exit(EXIT_FAILURE); } InputByteStream bytestream(bitstreamFile); // create & initialize internal classes xCreateDecLib(); xInitDecLib (); // main decoder loop Bool openedReconFile[MAX_LAYERS]; // reconstruction file not yet opened. (must be performed after SPS is seen) Bool loopFiltered[MAX_LAYERS]; memset( loopFiltered, false, sizeof( loopFiltered ) ); for(UInt layer=0; layer<=m_tgtLayerId; layer++) { openedReconFile[layer] = false; m_aiPOCLastDisplay[layer] += m_iSkipFrame; // set the last displayed POC correctly for skip forward. } UInt curLayerId = 0; // current layer to be reconstructed #if AVC_BASE TComPic pcBLPic; fstream streamYUV; if( m_pchBLReconFile ) { streamYUV.open( m_pchBLReconFile, fstream::in | fstream::binary ); } TComList *cListPic = m_acTDecTop[0].getListPic(); m_acTDecTop[0].setBLsize( m_iBLSourceWidth, m_iBLSourceHeight ); m_acTDecTop[0].setBLReconFile( &streamYUV ); pcBLPic.setLayerId( 0 ); cListPic->pushBack( &pcBLPic ); #if AVC_SYNTAX fstream streamSyntaxFile; if( m_pchBLSyntaxFile ) { streamSyntaxFile.open( m_pchBLSyntaxFile, fstream::in | fstream::binary ); } m_acTDecTop[0].setBLSyntaxFile( &streamSyntaxFile ); #endif #endif while (!!bitstreamFile) { /* location serves to work around a design fault in the decoder, whereby * the process of reading a new slice that is the first slice of a new frame * requires the TDecTop::decode() method to be called again with the same * nal unit. */ streampos location = bitstreamFile.tellg(); AnnexBStats stats = AnnexBStats(); vector nalUnit; InputNALUnit nalu; byteStreamNALUnit(bytestream, nalUnit, stats); // call actual decoding function Bool bNewPicture = false; Bool bNewPOC = false; if (nalUnit.empty()) { /* this can happen if the following occur: * - empty input file * - two back-to-back start_code_prefixes * - start_code_prefix immediately followed by EOF */ fprintf(stderr, "Warning: Attempt to decode an empty NAL unit\n"); } else { read(nalu, nalUnit); if( (m_iMaxTemporalLayer >= 0 && nalu.m_temporalId > m_iMaxTemporalLayer) || !isNaluWithinTargetDecLayerIdSet(&nalu) || (nalu.m_layerId > m_tgtLayerId) ) { bNewPicture = false; } else { bNewPicture = m_acTDecTop[nalu.m_layerId].decode(nalu, m_iSkipFrame, m_aiPOCLastDisplay[nalu.m_layerId], curLayerId, bNewPOC); if (bNewPicture) { bitstreamFile.clear(); /* location points to the current nalunit payload[1] due to the * need for the annexB parser to read three extra bytes. * [1] except for the first NAL unit in the file * (but bNewPicture doesn't happen then) */ bitstreamFile.seekg(location-streamoff(3)); bytestream.reset(); } } } if (bNewPicture || !bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS) { #if O0194_DIFFERENT_BITDEPTH_EL_BL //Bug fix: The bit depth was not set correctly for each layer when doing DBF g_bitDepthY = g_bitDepthYLayer[curLayerId]; g_bitDepthC = g_bitDepthCLayer[curLayerId]; #endif if (!loopFiltered[curLayerId] || bitstreamFile) { m_acTDecTop[curLayerId].executeLoopFilters(poc, pcListPic); } loopFiltered[curLayerId] = (nalu.m_nalUnitType == NAL_UNIT_EOS); #if EARLY_REF_PIC_MARKING m_acTDecTop[curLayerId].earlyPicMarking(m_iMaxTemporalLayer, m_targetDecLayerIdSet); #endif } if( pcListPic ) { if ( m_pchReconFile[curLayerId] && !openedReconFile[curLayerId] ) { if (!m_outputBitDepthY) { m_outputBitDepthY = g_bitDepthY; } if (!m_outputBitDepthC) { m_outputBitDepthC = g_bitDepthC; } m_acTVideoIOYuvReconFile[curLayerId].open( m_pchReconFile[curLayerId], true, m_outputBitDepthY, m_outputBitDepthC, g_bitDepthY, g_bitDepthC ); // write mode openedReconFile[curLayerId] = true; } #if ALIGNED_BUMPING Bool outputPicturesFlag = true; #if NO_OUTPUT_OF_PRIOR_PICS if( m_acTDecTop[nalu.m_layerId].getNoOutputOfPriorPicsFlags() ) { outputPicturesFlag = false; } #endif if (nalu.m_nalUnitType == NAL_UNIT_EOS) // End of sequence { flushAllPictures( nalu.m_layerId, outputPicturesFlag ); } if( bNewPicture ) // New picture, slice header parsed but picture not decoded { #if NO_OUTPUT_OF_PRIOR_PICS if( #else if ( bNewPOC && #endif ( nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_W_RADL || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_N_LP || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_N_LP || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_W_RADL || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_W_LP ) ) { flushAllPictures( nalu.m_layerId, outputPicturesFlag ); } else { this->checkOutputBeforeDecoding( nalu.m_layerId ); } } /* The following code has to be executed when the last DU of the picture is decoded TODO: Need code to identify end of decoding a picture { this->checkOutputAfterDecoding( ); } */ #else if ( bNewPicture && bNewPOC && ( nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_W_RADL || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_N_LP || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_N_LP || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_W_RADL || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_W_LP ) ) { xFlushOutput( pcListPic, curLayerId ); } if (nalu.m_nalUnitType == NAL_UNIT_EOS) { xFlushOutput( pcListPic, curLayerId ); } // write reconstruction to file if(bNewPicture) { xWriteOutput( pcListPic, curLayerId, nalu.m_temporalId ); } #endif } } #if ALIGNED_BUMPING flushAllPictures( true ); #else for(UInt layer = 0; layer <= m_tgtLayerId; layer++) { xFlushOutput( m_acTDecTop[layer].getListPic(), layer ); } #endif // delete buffers #if AVC_BASE UInt layerIdmin = m_acTDecTop[0].getBLReconFile()->is_open() ? 1 : 0; if( streamYUV.is_open() ) { streamYUV.close(); } #if AVC_SYNTAX if( streamSyntaxFile.is_open() ) { streamSyntaxFile.close(); } #endif pcBLPic.destroy(); for(UInt layer = layerIdmin; layer <= m_tgtLayerId; layer++) #else for(UInt layer = 0; layer <= m_tgtLayerId; layer++) #endif { m_acTDecTop[layer].deletePicBuffer(); } // destroy internal classes xDestroyDecLib(); } #else Void TAppDecTop::decode() { Int poc; TComList* pcListPic = NULL; ifstream bitstreamFile(m_pchBitstreamFile, ifstream::in | ifstream::binary); if (!bitstreamFile) { fprintf(stderr, "\nfailed to open bitstream file `%s' for reading\n", m_pchBitstreamFile); exit(EXIT_FAILURE); } InputByteStream bytestream(bitstreamFile); // create & initialize internal classes xCreateDecLib(); xInitDecLib (); m_iPOCLastDisplay += m_iSkipFrame; // set the last displayed POC correctly for skip forward. // main decoder loop Bool openedReconFile = false; // reconstruction file not yet opened. (must be performed after SPS is seen) Bool loopFiltered = false; #if SYNTAX_OUTPUT if( !m_pchBLSyntaxFile ) { printf( "Wrong base layer syntax file\n" ); exit(EXIT_FAILURE); } fstream streamSyntaxFile( m_pchBLSyntaxFile, fstream::out | fstream::binary ); if( !streamSyntaxFile.good() ) { printf( "Base layer syntax input reading error\n" ); exit(EXIT_FAILURE); } m_cTDecTop.setBLSyntaxFile( &streamSyntaxFile ); for( Int i = m_iBLFrames * m_iBLSourceWidth * m_iBLSourceHeight * SYNTAX_BYTES / 16; i >= 0; i-- ) { streamSyntaxFile.put( 0 ); } streamSyntaxFile.seekp( 0 ); #endif while (!!bitstreamFile) { /* location serves to work around a design fault in the decoder, whereby * the process of reading a new slice that is the first slice of a new frame * requires the TDecTop::decode() method to be called again with the same * nal unit. */ streampos location = bitstreamFile.tellg(); AnnexBStats stats = AnnexBStats(); vector nalUnit; InputNALUnit nalu; byteStreamNALUnit(bytestream, nalUnit, stats); // call actual decoding function Bool bNewPicture = false; if (nalUnit.empty()) { /* this can happen if the following occur: * - empty input file * - two back-to-back start_code_prefixes * - start_code_prefix immediately followed by EOF */ fprintf(stderr, "Warning: Attempt to decode an empty NAL unit\n"); } else { read(nalu, nalUnit); if( (m_iMaxTemporalLayer >= 0 && nalu.m_temporalId > m_iMaxTemporalLayer) || !isNaluWithinTargetDecLayerIdSet(&nalu) ) { bNewPicture = false; } else { bNewPicture = m_cTDecTop.decode(nalu, m_iSkipFrame, m_iPOCLastDisplay); if (bNewPicture) { bitstreamFile.clear(); /* location points to the current nalunit payload[1] due to the * need for the annexB parser to read three extra bytes. * [1] except for the first NAL unit in the file * (but bNewPicture doesn't happen then) */ bitstreamFile.seekg(location-streamoff(3)); bytestream.reset(); } } } if (bNewPicture || !bitstreamFile || nalu.m_nalUnitType == NAL_UNIT_EOS) { if (!loopFiltered || bitstreamFile) { m_cTDecTop.executeLoopFilters(poc, pcListPic); } loopFiltered = (nalu.m_nalUnitType == NAL_UNIT_EOS); } if( pcListPic ) { if ( m_pchReconFile && !openedReconFile ) { if (!m_outputBitDepthY) { m_outputBitDepthY = g_bitDepthY; } if (!m_outputBitDepthC) { m_outputBitDepthC = g_bitDepthC; } m_cTVideoIOYuvReconFile.open( m_pchReconFile, true, m_outputBitDepthY, m_outputBitDepthC, g_bitDepthY, g_bitDepthC ); // write mode openedReconFile = true; } if ( bNewPicture && ( nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_W_RADL || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_IDR_N_LP || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_N_LP || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_W_RADL || nalu.m_nalUnitType == NAL_UNIT_CODED_SLICE_BLA_W_LP ) ) { xFlushOutput( pcListPic ); } if (nalu.m_nalUnitType == NAL_UNIT_EOS) { xFlushOutput( pcListPic ); } // write reconstruction to file if(bNewPicture) { xWriteOutput( pcListPic, nalu.m_temporalId ); } } } #if SYNTAX_OUTPUT if( streamSyntaxFile.is_open() ) { streamSyntaxFile.close(); } #endif xFlushOutput( pcListPic ); // delete buffers m_cTDecTop.deletePicBuffer(); // destroy internal classes xDestroyDecLib(); } #endif // ==================================================================================================================== // Protected member functions // ==================================================================================================================== Void TAppDecTop::xCreateDecLib() { #if SVC_EXTENSION // initialize global variables initROM(); for(UInt layer = 0; layer <= m_tgtLayerId; layer++) { // set layer ID m_acTDecTop[layer].setLayerId ( layer ); // create decoder class m_acTDecTop[layer].create(); m_acTDecTop[layer].setLayerDec(m_apcTDecTop); } #else // create decoder class m_cTDecTop.create(); #endif } Void TAppDecTop::xDestroyDecLib() { #if SVC_EXTENSION // destroy ROM destroyROM(); for(UInt layer = 0; layer <= m_tgtLayerId; layer++) { if ( m_pchReconFile[layer] ) { m_acTVideoIOYuvReconFile[layer].close(); } // destroy decoder class m_acTDecTop[layer].destroy(); } #else if ( m_pchReconFile ) { m_cTVideoIOYuvReconFile. close(); } // destroy decoder class m_cTDecTop.destroy(); #endif } Void TAppDecTop::xInitDecLib() { // initialize decoder class #if SVC_EXTENSION for(UInt layer = 0; layer <= m_tgtLayerId; layer++) { m_acTDecTop[layer].init(); m_acTDecTop[layer].setDecodedPictureHashSEIEnabled(m_decodedPictureHashSEIEnabled); m_acTDecTop[layer].setNumLayer( m_tgtLayerId + 1 ); #if OUTPUT_LAYER_SET_INDEX m_acTDecTop[layer].setCommonDecoderParams( this->getCommonDecoderParams() ); #endif } #else m_cTDecTop.init(); m_cTDecTop.setDecodedPictureHashSEIEnabled(m_decodedPictureHashSEIEnabled); #endif } /** \param pcListPic list of pictures to be written to file \todo DYN_REF_FREE should be revised */ #if SVC_EXTENSION Void TAppDecTop::xWriteOutput( TComList* pcListPic, UInt layerId, UInt tId ) #else Void TAppDecTop::xWriteOutput( TComList* pcListPic, UInt tId ) #endif { if (pcListPic->empty()) { return; } TComList::iterator iterPic = pcListPic->begin(); Int numPicsNotYetDisplayed = 0; while (iterPic != pcListPic->end()) { TComPic* pcPic = *(iterPic); #if SVC_EXTENSION if(pcPic->getOutputMark() && pcPic->getPOC() > m_aiPOCLastDisplay[layerId]) #else if(pcPic->getOutputMark() && pcPic->getPOC() > m_iPOCLastDisplay) #endif { numPicsNotYetDisplayed++; } iterPic++; } iterPic = pcListPic->begin(); if (numPicsNotYetDisplayed>2) { iterPic++; } TComPic* pcPic = *(iterPic); if (numPicsNotYetDisplayed>2 && pcPic->isField()) //Field Decoding { TComList::iterator endPic = pcListPic->end(); endPic--; iterPic = pcListPic->begin(); while (iterPic != endPic) { TComPic* pcPicTop = *(iterPic); iterPic++; TComPic* pcPicBottom = *(iterPic); #if SVC_EXTENSION if ( pcPicTop->getOutputMark() && (numPicsNotYetDisplayed > pcPicTop->getNumReorderPics(tId) && !(pcPicTop->getPOC()%2) && pcPicBottom->getPOC() == pcPicTop->getPOC()+1) && pcPicBottom->getOutputMark() && (numPicsNotYetDisplayed > pcPicBottom->getNumReorderPics(tId) && (pcPicTop->getPOC() == m_aiPOCLastDisplay[layerId]+1 || m_aiPOCLastDisplay[layerId]<0))) #else if ( pcPicTop->getOutputMark() && (numPicsNotYetDisplayed > pcPicTop->getNumReorderPics(tId) && !(pcPicTop->getPOC()%2) && pcPicBottom->getPOC() == pcPicTop->getPOC()+1) && pcPicBottom->getOutputMark() && (numPicsNotYetDisplayed > pcPicBottom->getNumReorderPics(tId) && (pcPicTop->getPOC() == m_iPOCLastDisplay+1 || m_iPOCLastDisplay<0))) #endif { // write to file numPicsNotYetDisplayed = numPicsNotYetDisplayed-2; #if SVC_EXTENSION if ( m_pchReconFile[layerId] ) { const Window &conf = pcPicTop->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPicTop->getDefDisplayWindow() : Window(); const Bool isTff = pcPicTop->isTopField(); #if REPN_FORMAT_IN_VPS UInt chromaFormatIdc = pcPic->getSlice(0)->getChromaFormatIdc(); Int xScal = TComSPS::getWinUnitX( chromaFormatIdc ), yScal = TComSPS::getWinUnitY( chromaFormatIdc ); m_acTVideoIOYuvReconFile[layerId].write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), conf.getWindowLeftOffset() * xScal + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() * xScal + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() * yScal + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset()* yScal + defDisp.getWindowBottomOffset(), isTff ); #else #if O0194_DIFFERENT_BITDEPTH_EL_BL // Compile time bug-fix m_acTVideoIOYuvReconFile[layerId].write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), #else m_cTVideoIOYuvReconFile.write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), #endif conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset(), isTff ); #endif } // update POC of display order m_aiPOCLastDisplay[layerId] = pcPicBottom->getPOC(); #else if ( m_pchReconFile ) { const Window &conf = pcPicTop->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPicTop->getDefDisplayWindow() : Window(); const Bool isTff = pcPicTop->isTopField(); m_cTVideoIOYuvReconFile.write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset(), isTff ); } // update POC of display order m_iPOCLastDisplay = pcPicBottom->getPOC(); #endif // erase non-referenced picture in the reference picture list after display if ( !pcPicTop->getSlice(0)->isReferenced() && pcPicTop->getReconMark() == true ) { #if !DYN_REF_FREE pcPicTop->setReconMark(false); // mark it should be extended later pcPicTop->getPicYuvRec()->setBorderExtension( false ); #else pcPicTop->destroy(); pcListPic->erase( iterPic ); iterPic = pcListPic->begin(); // to the beginning, non-efficient way, have to be revised! continue; #endif } if ( !pcPicBottom->getSlice(0)->isReferenced() && pcPicBottom->getReconMark() == true ) { #if !DYN_REF_FREE pcPicBottom->setReconMark(false); // mark it should be extended later pcPicBottom->getPicYuvRec()->setBorderExtension( false ); #else pcPicBottom->destroy(); pcListPic->erase( iterPic ); iterPic = pcListPic->begin(); // to the beginning, non-efficient way, have to be revised! continue; #endif } pcPicTop->setOutputMark(false); pcPicBottom->setOutputMark(false); } } } else if (!pcPic->isField()) //Frame Decoding { iterPic = pcListPic->begin(); while (iterPic != pcListPic->end()) { pcPic = *(iterPic); #if SVC_EXTENSION if ( pcPic->getOutputMark() && (numPicsNotYetDisplayed > pcPic->getNumReorderPics(tId) && pcPic->getPOC() > m_aiPOCLastDisplay[layerId])) #else if ( pcPic->getOutputMark() && (numPicsNotYetDisplayed > pcPic->getNumReorderPics(tId) && pcPic->getPOC() > m_iPOCLastDisplay)) #endif { // write to file numPicsNotYetDisplayed--; #if SVC_EXTENSION if ( m_pchReconFile[layerId] ) { const Window &conf = pcPic->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPic->getDefDisplayWindow() : Window(); #if REPN_FORMAT_IN_VPS UInt chromaFormatIdc = pcPic->getSlice(0)->getChromaFormatIdc(); Int xScal = TComSPS::getWinUnitX( chromaFormatIdc ), yScal = TComSPS::getWinUnitY( chromaFormatIdc ); m_acTVideoIOYuvReconFile[layerId].write( pcPic->getPicYuvRec(), conf.getWindowLeftOffset() * xScal + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() * xScal + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() * yScal + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset()* yScal + defDisp.getWindowBottomOffset() ); #else m_acTVideoIOYuvReconFile[layerId].write( pcPic->getPicYuvRec(), conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset() ); #endif } // update POC of display order m_aiPOCLastDisplay[layerId] = pcPic->getPOC(); #else if ( m_pchReconFile ) { #if SYNTAX_OUTPUT const Window &conf = pcPic->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPic->getDefDisplayWindow() : Window(); m_cTVideoIOYuvReconFile.write( pcPic->getPicYuvRec(), conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset() ); #endif } // update POC of display order m_iPOCLastDisplay = pcPic->getPOC(); #endif // erase non-referenced picture in the reference picture list after display if ( !pcPic->getSlice(0)->isReferenced() && pcPic->getReconMark() == true ) { #if !DYN_REF_FREE pcPic->setReconMark(false); // mark it should be extended later pcPic->getPicYuvRec()->setBorderExtension( false ); #else pcPic->destroy(); pcListPic->erase( iterPic ); iterPic = pcListPic->begin(); // to the beginning, non-efficient way, have to be revised! continue; #endif } pcPic->setOutputMark(false); } iterPic++; } } } /** \param pcListPic list of pictures to be written to file \todo DYN_REF_FREE should be revised */ #if SVC_EXTENSION Void TAppDecTop::xFlushOutput( TComList* pcListPic, UInt layerId ) #else Void TAppDecTop::xFlushOutput( TComList* pcListPic ) #endif { if(!pcListPic || pcListPic->empty()) { return; } TComList::iterator iterPic = pcListPic->begin(); iterPic = pcListPic->begin(); TComPic* pcPic = *(iterPic); if (pcPic->isField()) //Field Decoding { TComList::iterator endPic = pcListPic->end(); endPic--; TComPic *pcPicTop, *pcPicBottom = NULL; while (iterPic != endPic) { pcPicTop = *(iterPic); iterPic++; pcPicBottom = *(iterPic); if ( pcPicTop->getOutputMark() && pcPicBottom->getOutputMark() && !(pcPicTop->getPOC()%2) && (pcPicBottom->getPOC() == pcPicTop->getPOC()+1) ) { // write to file #if SVC_EXTENSION if ( m_pchReconFile[layerId] ) { const Window &conf = pcPicTop->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPicTop->getDefDisplayWindow() : Window(); const Bool isTff = pcPicTop->isTopField(); #if REPN_FORMAT_IN_VPS UInt chromaFormatIdc = pcPic->getSlice(0)->getChromaFormatIdc(); Int xScal = TComSPS::getWinUnitX( chromaFormatIdc ), yScal = TComSPS::getWinUnitY( chromaFormatIdc ); m_acTVideoIOYuvReconFile[layerId].write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), conf.getWindowLeftOffset() *xScal + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() *xScal + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() *yScal + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset()*yScal + defDisp.getWindowBottomOffset(), isTff ); #else #if O0194_DIFFERENT_BITDEPTH_EL_BL // Compile time bug-fix m_acTVideoIOYuvReconFile[layerId].write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), #else m_cTVideoIOYuvReconFile[layerId].write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), #endif conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset(), isTff ); #endif } // update POC of display order m_aiPOCLastDisplay[layerId] = pcPicBottom->getPOC(); #else if ( m_pchReconFile ) { const Window &conf = pcPicTop->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPicTop->getDefDisplayWindow() : Window(); const Bool isTff = pcPicTop->isTopField(); m_cTVideoIOYuvReconFile.write( pcPicTop->getPicYuvRec(), pcPicBottom->getPicYuvRec(), conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset(), isTff ); } // update POC of display order m_iPOCLastDisplay = pcPicBottom->getPOC(); #endif // erase non-referenced picture in the reference picture list after display if ( !pcPicTop->getSlice(0)->isReferenced() && pcPicTop->getReconMark() == true ) { #if !DYN_REF_FREE pcPicTop->setReconMark(false); // mark it should be extended later pcPicTop->getPicYuvRec()->setBorderExtension( false ); #else pcPicTop->destroy(); pcListPic->erase( iterPic ); iterPic = pcListPic->begin(); // to the beginning, non-efficient way, have to be revised! continue; #endif } if ( !pcPicBottom->getSlice(0)->isReferenced() && pcPicBottom->getReconMark() == true ) { #if !DYN_REF_FREE pcPicBottom->setReconMark(false); // mark it should be extended later pcPicBottom->getPicYuvRec()->setBorderExtension( false ); #else pcPicBottom->destroy(); pcListPic->erase( iterPic ); iterPic = pcListPic->begin(); // to the beginning, non-efficient way, have to be revised! continue; #endif } pcPicTop->setOutputMark(false); pcPicBottom->setOutputMark(false); #if !DYN_REF_FREE if(pcPicTop) { pcPicTop->destroy(); delete pcPicTop; pcPicTop = NULL; } #endif } } if(pcPicBottom) { pcPicBottom->destroy(); delete pcPicBottom; pcPicBottom = NULL; } } else //Frame decoding { while (iterPic != pcListPic->end()) { pcPic = *(iterPic); if ( pcPic->getOutputMark() ) { // write to file #if SVC_EXTENSION if ( m_pchReconFile[layerId] ) { const Window &conf = pcPic->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPic->getDefDisplayWindow() : Window(); #if REPN_FORMAT_IN_VPS UInt chromaFormatIdc = pcPic->getSlice(0)->getChromaFormatIdc(); Int xScal = TComSPS::getWinUnitX( chromaFormatIdc ), yScal = TComSPS::getWinUnitY( chromaFormatIdc ); m_acTVideoIOYuvReconFile[layerId].write( pcPic->getPicYuvRec(), conf.getWindowLeftOffset() *xScal + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() *xScal + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() *yScal + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset()*yScal + defDisp.getWindowBottomOffset() ); #else m_acTVideoIOYuvReconFile[layerId].write( pcPic->getPicYuvRec(), conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset() ); #endif } // update POC of display order m_aiPOCLastDisplay[layerId] = pcPic->getPOC(); #else if ( m_pchReconFile ) { const Window &conf = pcPic->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pcPic->getDefDisplayWindow() : Window(); m_cTVideoIOYuvReconFile.write( pcPic->getPicYuvRec(), conf.getWindowLeftOffset() + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset() + defDisp.getWindowBottomOffset() ); } // update POC of display order m_iPOCLastDisplay = pcPic->getPOC(); #endif // erase non-referenced picture in the reference picture list after display if ( !pcPic->getSlice(0)->isReferenced() && pcPic->getReconMark() == true ) { #if !DYN_REF_FREE pcPic->setReconMark(false); // mark it should be extended later pcPic->getPicYuvRec()->setBorderExtension( false ); #else pcPic->destroy(); pcListPic->erase( iterPic ); iterPic = pcListPic->begin(); // to the beginning, non-efficient way, have to be revised! continue; #endif } pcPic->setOutputMark(false); } #if !SVC_EXTENSION #if !DYN_REF_FREE if(pcPic) { pcPic->destroy(); delete pcPic; pcPic = NULL; } #endif #endif iterPic++; } } #if SVC_EXTENSION m_aiPOCLastDisplay[layerId] = -MAX_INT; #else pcListPic->clear(); m_iPOCLastDisplay = -MAX_INT; #endif } /** \param nalu Input nalu to check whether its LayerId is within targetDecLayerIdSet */ Bool TAppDecTop::isNaluWithinTargetDecLayerIdSet( InputNALUnit* nalu ) { if ( m_targetDecLayerIdSet.size() == 0 ) // By default, the set is empty, meaning all LayerIds are allowed { return true; } for (std::vector::iterator it = m_targetDecLayerIdSet.begin(); it != m_targetDecLayerIdSet.end(); it++) { if ( nalu->m_reservedZero6Bits == (*it) ) { return true; } } return false; } #if ALIGNED_BUMPING // Function outputs a picture, and marks it as not needed for output. Void TAppDecTop::xOutputAndMarkPic( TComPic *pic, const Char *reconFile, const Int layerIdx, Int &pocLastDisplay, DpbStatus &dpbStatus ) { if ( reconFile ) { const Window &conf = pic->getConformanceWindow(); const Window &defDisp = m_respectDefDispWindow ? pic->getDefDisplayWindow() : Window(); Int xScal = 1, yScal = 1; #if REPN_FORMAT_IN_VPS UInt chromaFormatIdc = pic->getSlice(0)->getChromaFormatIdc(); xScal = TComSPS::getWinUnitX( chromaFormatIdc ); yScal = TComSPS::getWinUnitY( chromaFormatIdc ); #endif m_acTVideoIOYuvReconFile[layerIdx].write( pic->getPicYuvRec(), conf.getWindowLeftOffset() * xScal + defDisp.getWindowLeftOffset(), conf.getWindowRightOffset() * xScal + defDisp.getWindowRightOffset(), conf.getWindowTopOffset() * yScal + defDisp.getWindowTopOffset(), conf.getWindowBottomOffset()* yScal + defDisp.getWindowBottomOffset() ); } // update POC of display order pocLastDisplay = pic->getPOC(); // Mark as not needed for output pic->setOutputMark(false); // "erase" non-referenced picture in the reference picture list after display if ( !pic->getSlice(0)->isReferenced() && pic->getReconMark() == true ) { pic->setReconMark(false); // mark it should be extended later pic->getPicYuvRec()->setBorderExtension( false ); dpbStatus.m_numPicsInLayer[layerIdx]--; } } Void TAppDecTop::flushAllPictures(Int layerId, Bool outputPictures) { // First "empty" all pictures that are not used for reference and not needed for output emptyUnusedPicturesNotNeededForOutput(); if( outputPictures ) // All pictures in the DPB in that layer are to be output; this means other pictures would also be output { std::vector listOfPocs; std::vector listOfPocsInEachLayer[MAX_LAYERS]; std::vector listOfPocsPositionInEachLayer[MAX_LAYERS]; DpbStatus dpbStatus; // Find the status of the DPB xFindDPBStatus(listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus); if( listOfPocs.size() ) { while( listOfPocsInEachLayer[layerId].size() ) // As long as there picture in the layer to be output { bumpingProcess( listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus ); } } } // Now remove all pictures from the layer DPB? markAllPicturesAsErased(layerId); } Void TAppDecTop::flushAllPictures(Bool outputPictures) { // First "empty" all pictures that are not used for reference and not needed for output emptyUnusedPicturesNotNeededForOutput(); if( outputPictures ) // All pictures in the DPB are to be output { std::vector listOfPocs; std::vector listOfPocsInEachLayer[MAX_LAYERS]; std::vector listOfPocsPositionInEachLayer[MAX_LAYERS]; DpbStatus dpbStatus; // Find the status of the DPB xFindDPBStatus(listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus); while( dpbStatus.m_numAUsNotDisplayed ) { bumpingProcess( listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus ); } } // Now remove all pictures from the DPB? markAllPicturesAsErased(); } Void TAppDecTop::markAllPicturesAsErased() { for(Int i = 0; i < MAX_LAYERS; i++) { m_acTDecTop[i].getListPic()->clear(); } } Void TAppDecTop::markAllPicturesAsErased(Int layerIdx) { m_acTDecTop[layerIdx].getListPic()->clear(); } Void TAppDecTop::checkOutputBeforeDecoding(Int layerIdx) { std::vector listOfPocs; std::vector listOfPocsInEachLayer[MAX_LAYERS]; std::vector listOfPocsPositionInEachLayer[MAX_LAYERS]; DpbStatus dpbStatus; // First "empty" all pictures that are not used for reference and not needed for output emptyUnusedPicturesNotNeededForOutput(); // Find the status of the DPB xFindDPBStatus(listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus); // If not picture to be output, return if( listOfPocs.size() == 0 ) { return; } // Find DPB-information from the VPS DpbStatus maxDpbLimit; Int targetLsIdx, subDpbIdx; TComVPS *vps = findDpbParametersFromVps(listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, maxDpbLimit); if( getCommonDecoderParams()->getTargetOutputLayerSetIdx() == 0 ) { targetLsIdx = 0; subDpbIdx = 0; } else { targetLsIdx = vps->getOutputLayerSetIdx( getCommonDecoderParams()->getTargetOutputLayerSetIdx() ); subDpbIdx = vps->getSubDpbAssigned( targetLsIdx, layerIdx ); } // Assume that listOfPocs is sorted in increasing order - if not have to sort it. while( ifInvokeBumpingBeforeDecoding(dpbStatus, maxDpbLimit, layerIdx, subDpbIdx) ) { bumpingProcess( listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus ); } } Void TAppDecTop::checkOutputAfterDecoding() { std::vector listOfPocs; std::vector listOfPocsInEachLayer[MAX_LAYERS]; std::vector listOfPocsPositionInEachLayer[MAX_LAYERS]; DpbStatus dpbStatus; // First "empty" all pictures that are not used for reference and not needed for output emptyUnusedPicturesNotNeededForOutput(); // Find the status of the DPB xFindDPBStatus(listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus); // If not picture to be output, return if( listOfPocs.size() == 0 ) { return; } // Find DPB-information from the VPS DpbStatus maxDpbLimit; findDpbParametersFromVps(listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, maxDpbLimit); // Assume that listOfPocs is sorted in increasing order - if not have to sort it. while( ifInvokeBumpingAfterDecoding(dpbStatus, maxDpbLimit) ) { bumpingProcess( listOfPocs, listOfPocsInEachLayer, listOfPocsPositionInEachLayer, dpbStatus ); } } Void TAppDecTop::bumpingProcess(std::vector &listOfPocs, std::vector *listOfPocsInEachLayer, std::vector *listOfPocsPositionInEachLayer, DpbStatus &dpbStatus) { // Choose the smallest POC value Int pocValue = *(listOfPocs.begin()); std::vector::iterator it; TComList::iterator iterPic; for( Int layerIdx = 0; layerIdx < dpbStatus.m_numLayers; layerIdx++) { // Check if picture with pocValue is present. it = find( listOfPocsInEachLayer[layerIdx].begin(), listOfPocsInEachLayer[layerIdx].end(), pocValue ); if( it != listOfPocsInEachLayer[layerIdx].end() ) // picture found. { Int picPosition = std::distance( listOfPocsInEachLayer[layerIdx].begin(), it ); Int j; for(j = 0, iterPic = m_acTDecTop[layerIdx].getListPic()->begin(); j < listOfPocsPositionInEachLayer[layerIdx][picPosition]; j++) // Picture to be output { iterPic++; } TComPic *pic = *iterPic; xOutputAndMarkPic( pic, m_pchReconFile[layerIdx], layerIdx, m_aiPOCLastDisplay[layerIdx], dpbStatus ); listOfPocsInEachLayer[layerIdx].erase( it ); listOfPocsPositionInEachLayer[layerIdx].erase( listOfPocsPositionInEachLayer[layerIdx].begin() + picPosition ); } } // Update sub-DPB status for( Int subDpbIdx = 0; subDpbIdx < dpbStatus.m_numSubDpbs; subDpbIdx++) { dpbStatus.m_numPicsInSubDpb[subDpbIdx]--; } dpbStatus.m_numAUsNotDisplayed--; // Remove the picture from the listOfPocs listOfPocs.erase( listOfPocs.begin() ); } TComVPS *TAppDecTop::findDpbParametersFromVps(std::vector const &listOfPocs, std::vector const *listOfPocsInEachLayer, std::vector const *listOfPocsPositionInEachLayer, DpbStatus &maxDpbLimit) { Int targetOutputLsIdx = getCommonDecoderParams()->getTargetOutputLayerSetIdx(); TComVPS *vps = NULL; if( targetOutputLsIdx == 0 ) // Only base layer is output { TComSPS *sps = NULL; assert( listOfPocsInEachLayer[0].size() != 0 ); TComList::iterator iterPic; Int j; for(j = 0, iterPic = m_acTDecTop[0].getListPic()->begin(); j < listOfPocsPositionInEachLayer[0][0]; j++) // Picture to be output { iterPic++; } TComPic *pic = *iterPic; sps = pic->getSlice(0)->getSPS(); assert( sps->getLayerId() == 0 ); vps = pic->getSlice(0)->getVPS(); Int highestTId = sps->getMaxTLayers() - 1; maxDpbLimit.m_numAUsNotDisplayed = sps->getNumReorderPics( highestTId ); // m_numAUsNotDisplayed is only variable name - stores reorderpics maxDpbLimit.m_maxLatencyIncrease = sps->getMaxLatencyIncrease( highestTId ) > 0; if( maxDpbLimit.m_maxLatencyIncrease ) { maxDpbLimit.m_maxLatencyPictures = sps->getMaxLatencyIncrease( highestTId ) + sps->getNumReorderPics( highestTId ) - 1; } maxDpbLimit.m_numPicsInLayer[0] = sps->getMaxDecPicBuffering( highestTId ); maxDpbLimit.m_numPicsInSubDpb[0] = sps->getMaxDecPicBuffering( highestTId ); } else { // ------------------------------------- // Find the VPS used for the pictures // ------------------------------------- for(Int i = 0; i < MAX_LAYERS; i++) { if( m_acTDecTop[i].getListPic()->empty() ) { assert( listOfPocsInEachLayer[i].size() == 0 ); continue; } std::vector::const_iterator it; it = find( listOfPocsInEachLayer[i].begin(), listOfPocsInEachLayer[i].end(), listOfPocs[0] ); TComList::iterator iterPic; if( it != listOfPocsInEachLayer[i].end() ) { Int picPosition = std::distance( listOfPocsInEachLayer[i].begin(), it ); Int j; for(j = 0, iterPic = m_acTDecTop[i].getListPic()->begin(); j < listOfPocsPositionInEachLayer[i][picPosition]; j++) // Picture to be output { iterPic++; } TComPic *pic = *iterPic; vps = pic->getSlice(0)->getVPS(); break; } } Int targetLsIdx = vps->getOutputLayerSetIdx( getCommonDecoderParams()->getTargetOutputLayerSetIdx() ); Int highestTId = vps->getMaxTLayers() - 1; maxDpbLimit.m_numAUsNotDisplayed = vps->getMaxVpsNumReorderPics( targetOutputLsIdx, highestTId ); // m_numAUsNotDisplayed is only variable name - stores reorderpics maxDpbLimit.m_maxLatencyIncrease = vps->getMaxVpsLatencyIncreasePlus1(targetOutputLsIdx, highestTId ) > 0; if( maxDpbLimit.m_maxLatencyIncrease ) { maxDpbLimit.m_maxLatencyPictures = vps->getMaxVpsNumReorderPics( targetOutputLsIdx, highestTId ) + vps->getMaxVpsLatencyIncreasePlus1(targetOutputLsIdx, highestTId ) - 1; } for(Int i = 0; i < vps->getNumLayersInIdList( targetLsIdx ); i++) { maxDpbLimit.m_numPicsInLayer[i] = vps->getMaxVpsLayerDecPicBuffMinus1( targetOutputLsIdx, i, highestTId ) + 1; maxDpbLimit.m_numPicsInSubDpb[vps->getSubDpbAssigned( targetLsIdx, i )] = vps->getMaxVpsDecPicBufferingMinus1( targetOutputLsIdx, vps->getSubDpbAssigned( targetLsIdx, i ), highestTId) + 1; } // ------------------------------------- } return vps; } Void TAppDecTop::emptyUnusedPicturesNotNeededForOutput() { for(Int layerIdx = 0; layerIdx < MAX_LAYERS; layerIdx++) { TComList *pcListPic = m_acTDecTop[layerIdx].getListPic(); TComList::iterator iterPic = pcListPic->begin(); while ( iterPic != pcListPic->end() ) { TComPic *pic = *iterPic; if( !pic->getSlice(0)->isReferenced() && !pic->getOutputMark() ) { // Emtpy the picture buffer pic->setReconMark( false ); } iterPic++; } } } Bool TAppDecTop::ifInvokeBumpingBeforeDecoding( const DpbStatus &dpbStatus, const DpbStatus &dpbLimit, const Int layerIdx, const Int subDpbIdx ) { Bool retVal = false; // Number of reorder picutres retVal |= ( dpbStatus.m_numAUsNotDisplayed > dpbLimit.m_numAUsNotDisplayed ); // Number of pictures in each sub-DPB retVal |= ( dpbStatus.m_numPicsInSubDpb[subDpbIdx] >= dpbLimit.m_numPicsInSubDpb[subDpbIdx] ); // Number of pictures in each layer retVal |= ( dpbStatus.m_numPicsInLayer[layerIdx] >= dpbLimit.m_numPicsInLayer[layerIdx]); return retVal; } Bool TAppDecTop::ifInvokeBumpingAfterDecoding( const DpbStatus &dpbStatus, const DpbStatus &dpbLimit ) { Bool retVal = false; // Number of reorder picutres retVal |= ( dpbStatus.m_numAUsNotDisplayed > dpbLimit.m_numAUsNotDisplayed ); return retVal; } Void TAppDecTop::xFindDPBStatus( std::vector &listOfPocs , std::vector *listOfPocsInEachLayer , std::vector *listOfPocsPositionInEachLayer , DpbStatus &dpbStatus ) { TComVPS *vps = NULL; dpbStatus.init(); for( Int i = 0; i < MAX_LAYERS; i++ ) { if( m_acTDecTop[i].getListPic()->empty() ) { continue; } // To check # AUs that have at least one picture not output, // For each layer, populate listOfPOcs if not already present TComList::iterator iterPic = m_acTDecTop[i].getListPic()->begin(); Int picPositionInList = 0; while (iterPic != m_acTDecTop[i].getListPic()->end()) { TComPic* pic = *(iterPic); if( pic->getReconMark() ) { if( vps == NULL ) { vps = pic->getSlice(0)->getVPS(); } std::vector::iterator it; if( pic->getOutputMark() ) // && pic->getPOC() > m_aiPOCLastDisplay[i]) { it = find( listOfPocs.begin(), listOfPocs.end(), pic->getPOC() ); // Check if already included if( it == listOfPocs.end() ) // New POC value - i.e. new AU - add to the list { listOfPocs.push_back( pic->getPOC() ); } listOfPocsInEachLayer [i].push_back( pic->getPOC() ); // POC to be output in each layer listOfPocsPositionInEachLayer [i].push_back( picPositionInList ); // For ease of access } if( pic->getSlice(0)->isReferenced() || pic->getOutputMark() ) { dpbStatus.m_numPicsInLayer[i]++; // Count pictures that are "used for reference" or "needed for output" } } iterPic++; picPositionInList++; } } assert( vps != NULL ); // No picture in any DPB? std::sort( listOfPocs.begin(), listOfPocs.end() ); // Sort in increasing order of POC Int targetLsIdx = vps->getOutputLayerSetIdx( getCommonDecoderParams()->getTargetOutputLayerSetIdx() ); // Update status dpbStatus.m_numAUsNotDisplayed = listOfPocs.size(); // Number of AUs not displayed dpbStatus.m_numLayers = vps->getNumLayersInIdList( targetLsIdx ); dpbStatus.m_numSubDpbs = vps->getNumSubDpbs( vps->getOutputLayerSetIdx( this->getCommonDecoderParams()->getTargetOutputLayerSetIdx() ) ); for(Int i = 0; i < dpbStatus.m_numLayers; i++) { dpbStatus.m_numPicsNotDisplayedInLayer[i] = listOfPocsInEachLayer[i].size(); dpbStatus.m_numPicsInSubDpb[vps->getSubDpbAssigned(targetLsIdx,i)] += dpbStatus.m_numPicsInLayer[i]; } assert( dpbStatus.m_numAUsNotDisplayed != -1 ); } #endif //! \}