/* 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-2012, 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 TDecTop.h \brief decoder class (header) */ #ifndef __TDECTOP__ #define __TDECTOP__ #include "TLibCommon/CommonDef.h" #include "TLibCommon/TComList.h" #include "TLibCommon/TComPicYuv.h" #include "TLibCommon/TComPic.h" #include "TLibCommon/TComTrQuant.h" #include "TLibCommon/SEI.h" #include "TDecGop.h" #include "TDecEntropy.h" #include "TDecSbac.h" #include "TDecCAVLC.h" #include "SEIread.h" struct InputNALUnit; //! \ingroup TLibDecoder //! \{ // ==================================================================================================================== // Class definition // ==================================================================================================================== /// decoder class class TDecTop { private: Int m_iGopSize; Bool m_bGopSizeSet; Int m_iMaxRefPicNum; Bool m_bRefreshPending; ///< refresh pending flag Int m_pocCRA; ///< POC number of the latest CRA picture Bool m_prevRAPisBLA; ///< true if the previous RAP (CRA/CRANT/BLA/BLANT/IDR) picture is a BLA/BLANT picture Int m_pocRandomAccess; ///< POC number of the random access point (the first IDR or CRA picture) TComList m_cListPic; // Dynamic buffer #if SVC_EXTENSION static ParameterSetManagerDecoder m_parameterSetManagerDecoder; // storage for parameter sets #else ParameterSetManagerDecoder m_parameterSetManagerDecoder; // storage for parameter sets #endif #if REF_IDX_MFM TComSPS* m_pcSPS; Bool m_bMFMEnabledFlag; #endif TComSlice* m_apcSlicePilot; SEImessages *m_SEIs; ///< "all" SEI messages. If not NULL, we own the object. // functional classes TComPrediction m_cPrediction; TComTrQuant m_cTrQuant; TDecGop m_cGopDecoder; TDecSlice m_cSliceDecoder; TDecCu m_cCuDecoder; TDecEntropy m_cEntropyDecoder; TDecCavlc m_cCavlcDecoder; TDecSbac m_cSbacDecoder; TDecBinCABAC m_cBinCABAC; SEIReader m_seiReader; TComLoopFilter m_cLoopFilter; #if !REMOVE_ALF TComAdaptiveLoopFilter m_cAdaptiveLoopFilter; #endif TComSampleAdaptiveOffset m_cSAO; Bool isSkipPictureForBLA(Int& iPOCLastDisplay); Bool isRandomAccessSkipPicture(Int& iSkipFrame, Int& iPOCLastDisplay); TComPic* m_pcPic; UInt m_uiSliceIdx; #if !SVC_EXTENSION Int m_prevPOC; #endif Bool m_bFirstSliceInPicture; #if !SVC_EXTENSION Bool m_bFirstSliceInSequence; #endif #if SVC_EXTENSION static UInt m_prevPOC; // POC of the previous slice static UInt m_uiPrevLayerId; // LayerId of the previous slice static Bool m_bFirstSliceInSequence; UInt m_layerId; UInt m_numLayer; TDecTop** m_ppcTDecTop; #if AVC_BASE fstream* m_pBLReconFile; Int m_iBLSourceWidth; Int m_iBLSourceHeight; #endif #endif #if AVC_SYNTAX || SYNTAX_OUTPUT fstream* m_pBLSyntaxFile; #endif #if REF_IDX_FRAMEWORK TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture #endif public: TDecTop(); virtual ~TDecTop(); Void create (); Void destroy (); void setDecodedPictureHashSEIEnabled(Int enabled) { m_cGopDecoder.setDecodedPictureHashSEIEnabled(enabled); } Void init(); #if SVC_EXTENSION Bool decode(InputNALUnit& nalu, Int& iSkipFrame, Int& iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC); #else Bool decode(InputNALUnit& nalu, Int& iSkipFrame, Int& iPOCLastDisplay); #endif Void deletePicBuffer(); Void executeDeblockAndAlf(UInt& ruiPOC, TComList*& rpcListPic, Int& iSkipFrame, Int& iPOCLastDisplay); #if SVC_EXTENSION UInt getLayerId () { return m_layerId; } Void setLayerId (UInt layer) { m_layerId = layer; } UInt getNumLayer () { return m_numLayer; } Void setNumLayer (UInt uiNum) { m_numLayer = uiNum; } TComList* getListPic() { return &m_cListPic; } Void setLayerDec(TDecTop **p) { m_ppcTDecTop = p; } TDecTop* getLayerDec(UInt layer) { return m_ppcTDecTop[layer]; } #if AVC_BASE Void setBLReconFile( fstream* pFile ) { m_pBLReconFile = pFile; } fstream* getBLReconFile() { return m_pBLReconFile; } Void setBLsize( Int iWidth, Int iHeight ) { m_iBLSourceWidth = iWidth; m_iBLSourceHeight = iHeight; } Int getBLWidth() { return m_iBLSourceWidth; } Int getBLHeight() { return m_iBLSourceHeight; } #endif #endif #if AVC_SYNTAX || SYNTAX_OUTPUT Void setBLSyntaxFile( fstream* pFile ) { m_pBLSyntaxFile = pFile; } fstream* getBLSyntaxFile() { return m_pBLSyntaxFile; } #endif #if REF_IDX_FRAMEWORK Void xInitILRP(TComSPS *pcSPS); Void setILRPic(TComPic *pcPic); #endif #if REF_IDX_MFM TComSPS* getSPS() {return m_pcSPS;} Void setMFMEnabledFlag(Bool flag) {m_bMFMEnabledFlag = flag;} Bool getMFMEnabledFlag() {return m_bMFMEnabledFlag;} #endif protected: Void xGetNewPicBuffer (TComSlice* pcSlice, TComPic*& rpcPic); Void xUpdateGopSize (TComSlice* pcSlice); Void xCreateLostPicture (Int iLostPOC); #if !REMOVE_APS Void decodeAPS( TComAPS* cAPS) { m_cEntropyDecoder.decodeAPS(cAPS); }; #endif Void xActivateParameterSets(); #if SVC_EXTENSION Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC); #else Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay); #endif Void xDecodeVPS(); Void xDecodeSPS(); Void xDecodePPS(); #if !REMOVE_APS Void xDecodeAPS(); #endif Void xDecodeSEI( TComInputBitstream* bs ); #if !REMOVE_APS Void allocAPS (TComAPS* pAPS); //!< memory allocation for APS #endif };// END CLASS DEFINITION TDecTop //! \} #endif // __TDECTOP__