/* 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 #if R0071_IRAP_EOS_CROSS_LAYER_IMPACTS Bool m_lastPicHasEos; #endif 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 R0235_SMALLEST_LAYER_ID UInt m_smallestLayerId; #endif #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_ID_PLUS1]; Int m_numSamplePredRefLayers; Int m_samplePredRefLayerId[MAX_VPS_LAYER_ID_PLUS1]; Int m_numMotionPredRefLayers; Int m_motionPredRefLayerId[MAX_VPS_LAYER_ID_PLUS1]; Bool m_samplePredEnabledFlag[MAX_VPS_LAYER_ID_PLUS1]; Bool m_motionPredEnabledFlag[MAX_VPS_LAYER_ID_PLUS1]; #endif TComPic* m_cIlpPic[MAX_NUM_REF]; ///< Inter layer Prediction picture = upsampled picture #endif #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 RESOLUTION_BASED_DPB Int m_subDpbIdx; // Index to the sub-DPB that the layer belongs to. // When new VPS is activated, this should be re-initialized to -1 #endif #if CONFORMANCE_BITSTREAM_MODE Bool m_confModeFlag; std::vector m_confListPic; // Dynamic buffer for storing pictures for conformance purposes #endif #if FIX_NON_OUTPUT_LAYER Bool m_isOutputLayerFlag; #endif public: #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 TDecTop(); virtual ~TDecTop(); Void create (); Void destroy (); Void setDecodedPictureHashSEIEnabled(Int enabled) { m_cGopDecoder.setDecodedPictureHashSEIEnabled(enabled); } #if Q0074_COLOUR_REMAPPING_SEI void setColourRemappingInfoSEIEnabled(Bool enabled) { m_cGopDecoder.setColourRemappingInfoSEIEnabled(enabled); } #endif 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 EARLY_REF_PIC_MARKING Void earlyPicMarking(Int maxTemporalLayer, std::vector& targetDecLayerIdList); #endif #if POC_RESET_IDC_DECODER Int getParseIdc() { return m_parseIdc;} Void setParseIdc(Int x) { m_parseIdc = x;} Void markAllPicsAsNoCurrAu(); 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 layer) { return m_ppcTDecTop[layer]; } #if R0235_SMALLEST_LAYER_ID Void xDeriveSmallestLayerId(TComVPS* vps); #endif #if VPS_EXTN_DIRECT_REF_LAYERS TDecTop* getRefLayerDec(UInt refLayerIdc); 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 getSamplePredRefLayerId (Int i) { return m_samplePredRefLayerId[i]; } Void setSamplePredRefLayerId (Int i, Int refLayerId) { m_samplePredRefLayerId[i] = refLayerId; } Int getNumMotionPredRefLayers () { return m_numMotionPredRefLayers; } Void setNumMotionPredRefLayers (Int num) { m_numMotionPredRefLayers = num; } Int getMotionPredRefLayerId (Int i) { return m_motionPredRefLayerId[i]; } Void setMotionPredRefLayerId (Int i, Int refLayerId) { m_motionPredRefLayerId[i] = refLayerId; } Bool getSamplePredEnabledFlag (Int i) { return m_samplePredEnabledFlag[i]; } Void setSamplePredEnabledFlag (Int i,Bool flag) { m_samplePredEnabledFlag[i] = flag; } Bool getMotionPredEnabledFlag (Int i) { return m_motionPredEnabledFlag[i]; } Void setMotionPredEnabledFlag (Int i,Bool flag) { m_motionPredEnabledFlag[i] = flag; } TDecTop* getSamplePredRefLayerDec ( UInt layerId ); TDecTop* getMotionPredRefLayerDec ( UInt layerId ); 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 RESOLUTION_BASED_DPB Void setSubDpbIdx(Int idx) { m_subDpbIdx = idx; } Int getSubDpbIdx() { return m_subDpbIdx; } Void assignSubDpbs(TComVPS *vps); #endif #if CONFORMANCE_BITSTREAM_MODE std::vector* getConfListPic() {return &m_confListPic; } // std::string const getDecodedYuvLayerFileName(Int layerId) { return m_decodedYuvLayerFileName[layerId]; } 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 #if POC_RESET_FLAG Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int &iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC); #else Bool xDecodeSlice(InputNALUnit &nalu, Int &iSkipFrame, Int iPOCLastDisplay, UInt& curLayerId, Bool& bNewPOC); #endif #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 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 #if R0071_IRAP_EOS_CROSS_LAYER_IMPACTS Void xCheckLayerReset(); Void xSetNoRaslOutputFlag(); Void xSetLayerInitializedFlag(); #endif };// END CLASS DEFINITION TDecTop //! \} #endif // __TDECTOP__