/* 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 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" #if Q0048_CGS_3D_ASYMLUT #include "TLibCommon/TCom3DAsymLUT.h" #endif #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_iMaxRefPicNum; NalUnitType m_associatedIRAPType; ///< NAL unit type of the associated IRAP picture Int m_pocCRA; ///< POC number of the latest CRA picture Int m_pocRandomAccess; ///< POC number of the random access point (the first IDR or CRA picture) TComList m_cListPic; // Dynamic buffer ParameterSetManagerDecoder m_parameterSetManagerDecoder; // storage for parameter sets TComSlice* m_apcSlicePilot; SEIMessages m_SEIs; ///< List of SEI messages that have been received before the first slice and between slices // functional classes TComPrediction m_cPrediction; #if Q0048_CGS_3D_ASYMLUT TCom3DAsymLUT m_c3DAsymLUTPPS; TComPicYuv* m_pColorMappedPic; #endif 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; 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 Bool m_prevSliceSkipped; Int m_skippedPOC; Bool m_bFirstSliceInBitstream; Int m_lastPOCNoOutputPriorPics; Bool m_isNoOutputPriorPics; Bool m_craNoRaslOutputFlag; //value of variable NoRaslOutputFlag of the last CRA pic #if O0043_BEST_EFFORT_DECODING UInt m_forceDecodeBitDepth; #endif std::ostream *m_pDecodedSEIOutputStream; #if SVC_EXTENSION #if Q0177_EOS_CHECKS Bool m_isLastNALWasEos; #endif Bool m_lastPicHasEos; 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; UInt m_smallestLayerId; #if P0297_VPS_POC_LSB_ALIGNED_FLAG Bool m_pocResettingFlag; Bool m_pocDecrementedInDPBFlag; #endif #if AVC_BASE fstream* m_pBLReconFile; #if !REPN_FORMAT_IN_VPS Int m_iBLSourceWidth; Int m_iBLSourceHeight; #endif #endif #if VPS_EXTN_DIRECT_REF_LAYERS Int m_numDirectRefLayers; Int m_refLayerId[MAX_VPS_LAYER_IDX_PLUS1]; Int m_numSamplePredRefLayers; Int m_numMotionPredRefLayers; #endif TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture #if OUTPUT_LAYER_SET_INDEX CommonDecoderParams* m_commonDecoderParams; #endif #if NO_CLRAS_OUTPUT_FLAG Bool m_noClrasOutputFlag; Bool m_layerInitializedFlag; Bool m_firstPicInLayerDecodedFlag; #endif #if POC_RESET_IDC_DECODER Int m_parseIdc; Int m_lastPocPeriodId; Int m_prevPicOrderCnt; #endif #if CONFORMANCE_BITSTREAM_MODE Bool m_confModeFlag; std::vector m_confListPic; // Dynamic buffer for storing pictures for conformance purposes #endif Bool m_isOutputLayerFlag; #endif //SVC_EXTENSION public: #if SVC_EXTENSION #if POC_RESET_RESTRICTIONS static Bool m_checkPocRestrictionsForCurrAu; static Int m_pocResetIdcOrCurrAu; static Bool m_baseLayerIdrFlag; static Bool m_baseLayerPicPresentFlag; static Bool m_baseLayerIrapFlag; static Bool m_nonBaseIdrPresentFlag; static Int m_nonBaseIdrType; static Bool m_picNonIdrWithRadlPresentFlag; static Bool m_picNonIdrNoLpPresentFlag; #endif #if POC_RESET_VALUE_RESTRICTION static Int m_crossLayerPocResetPeriodId; static Int m_crossLayerPocResetIdc; #endif #endif //SVC_EXTENSION 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(); TComSPS* getActiveSPS() { return m_parameterSetManagerDecoder.getActiveSPS(); } Void executeLoopFilters(Int& poc, TComList*& rpcListPic); Void checkNoOutputPriorPics (TComList* rpcListPic); Bool getNoOutputPriorPicsFlag () { return m_isNoOutputPriorPics; } Void setNoOutputPriorPicsFlag (Bool val) { m_isNoOutputPriorPics = val; } Void setFirstSliceInPicture (bool val) { m_bFirstSliceInPicture = val; } Bool getFirstSliceInSequence () { return m_bFirstSliceInSequence; } Void setFirstSliceInSequence (bool val) { m_bFirstSliceInSequence = val; } #if O0043_BEST_EFFORT_DECODING Void setForceDecodeBitDepth(UInt bitDepth) { m_forceDecodeBitDepth = bitDepth; } #endif Void setDecodedSEIMessageOutputStream(std::ostream *pOpStream) { m_pDecodedSEIOutputStream = pOpStream; } #if SVC_EXTENSION #if POC_RESET_IDC_DECODER Int getParseIdc () { return m_parseIdc; } Void setParseIdc (Int x) { m_parseIdc = x; } Void markAllPicsAsNoCurrAu (TComVPS *vps); Int getLastPocPeriodId () { return m_lastPocPeriodId; } Void setLastPocPeriodId (Int x) { m_lastPocPeriodId = x; } Int getPrevPicOrderCnt () { return m_prevPicOrderCnt; } Void setPrevPicOrderCnt (Int const x) { m_prevPicOrderCnt = x; } #endif 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 layerId) { return m_ppcTDecTop[layerId]; } Void xDeriveSmallestLayerId(TComVPS* vps); #if VPS_EXTN_DIRECT_REF_LAYERS TDecTop* getRefLayerDec (UInt refLayerIdx); Int getNumDirectRefLayers () { return m_numDirectRefLayers; } Void setNumDirectRefLayers (Int num) { m_numDirectRefLayers = num; } Int getRefLayerId (Int i) { return m_refLayerId[i]; } Void setRefLayerId (Int i, Int refLayerId) { m_refLayerId[i] = refLayerId; } Int getNumSamplePredRefLayers () { return m_numSamplePredRefLayers; } Void setNumSamplePredRefLayers (Int num) { m_numSamplePredRefLayers = num; } Int getNumMotionPredRefLayers () { return m_numMotionPredRefLayers; } Void setNumMotionPredRefLayers (Int num) { m_numMotionPredRefLayers = num; } Void setRefLayerParams( TComVPS* vps ); #endif #if AVC_BASE Void setBLReconFile( fstream* pFile ) { m_pBLReconFile = pFile; } fstream* getBLReconFile() { return m_pBLReconFile; } #if !REPN_FORMAT_IN_VPS 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 REPN_FORMAT_IN_VPS Void xInitILRP(TComSlice *slice); #else Void xInitILRP(TComSPS *pcSPS); #endif #if OUTPUT_LAYER_SET_INDEX CommonDecoderParams* getCommonDecoderParams() { return m_commonDecoderParams; } Void setCommonDecoderParams(CommonDecoderParams* x) { m_commonDecoderParams = x; } Void checkValueOfTargetOutputLayerSetIdx(TComVPS *vps); #endif #if SCALINGLIST_INFERRING ParameterSetManagerDecoder* getParameterSetManager() { return &m_parameterSetManagerDecoder; } #endif #if CONFORMANCE_BITSTREAM_MODE std::vector* getConfListPic() {return &m_confListPic; } Bool const getConfModeFlag() { return m_confModeFlag; } Void setConfModeFlag(Bool x) { m_confModeFlag = x; } #endif #endif //SVC_EXTENSION protected: Void xGetNewPicBuffer (TComSlice* pcSlice, TComPic*& rpcPic); Void xCreateLostPicture (Int iLostPOC); Void xActivateParameterSets(); #if SVC_EXTENSION Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC); Void xSetSpatialEnhLayerFlag(TComSlice* slice, TComPic* pic); #else Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay); #endif Void xDecodeVPS(); Void xDecodeSPS(); Void xDecodePPS( #if Q0048_CGS_3D_ASYMLUT TCom3DAsymLUT * pc3DAsymLUT #endif ); Void xDecodeSEI( TComInputBitstream* bs, const NalUnitType nalUnitType ); #if SVC_EXTENSION #if NO_CLRAS_OUTPUT_FLAG Int getNoClrasOutputFlag() { return m_noClrasOutputFlag;} Void setNoClrasOutputFlag(Bool x) { m_noClrasOutputFlag = x; } Int getLayerInitializedFlag() { return m_layerInitializedFlag;} Void setLayerInitializedFlag(Bool x) { m_layerInitializedFlag = x; } Int getFirstPicInLayerDecodedFlag() { return m_firstPicInLayerDecodedFlag;} Void setFirstPicInLayerDecodedFlag(Bool x) { m_firstPicInLayerDecodedFlag = x; } #endif #if Q0048_CGS_3D_ASYMLUT Void initAsymLut(TComSlice *pcSlice); #endif #if POC_RESET_RESTRICTIONS Void resetPocRestrictionCheckParameters(); #endif Void xCheckLayerReset(); Void xSetNoRaslOutputFlag(); Void xSetLayerInitializedFlag(); #endif };// END CLASS DEFINITION TDecTop //! \} #endif // __TDECTOP__