/* 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 TComSlice.h \brief slice header and SPS class (header) */ #ifndef __TCOMSLICE__ #define __TCOMSLICE__ #include #include #include #include "CommonDef.h" #include "TComRom.h" #include "TComList.h" #include "TComChromaFormat.h" //! \ingroup TLibCommon //! \{ class TComPic; class TComTrQuant; // ==================================================================================================================== // Constants // ==================================================================================================================== static const UInt REF_PIC_LIST_NUM_IDX=32; // ==================================================================================================================== // Class definition // ==================================================================================================================== /// Reference Picture Set class class TComReferencePictureSet { private: Int m_numberOfPictures; Int m_numberOfNegativePictures; Int m_numberOfPositivePictures; Int m_numberOfLongtermPictures; Int m_deltaPOC[MAX_NUM_REF_PICS]; Int m_POC[MAX_NUM_REF_PICS]; Bool m_used[MAX_NUM_REF_PICS]; Bool m_interRPSPrediction; Int m_deltaRIdxMinus1; Int m_deltaRPS; Int m_numRefIdc; Int m_refIdc[MAX_NUM_REF_PICS+1]; Bool m_bCheckLTMSB[MAX_NUM_REF_PICS]; Int m_pocLSBLT[MAX_NUM_REF_PICS]; Int m_deltaPOCMSBCycleLT[MAX_NUM_REF_PICS]; Bool m_deltaPocMSBPresentFlag[MAX_NUM_REF_PICS]; public: TComReferencePictureSet(); virtual ~TComReferencePictureSet(); Int getPocLSBLT(Int i) { return m_pocLSBLT[i]; } Void setPocLSBLT(Int i, Int x) { m_pocLSBLT[i] = x; } Int getDeltaPocMSBCycleLT(Int i) { return m_deltaPOCMSBCycleLT[i]; } Void setDeltaPocMSBCycleLT(Int i, Int x) { m_deltaPOCMSBCycleLT[i] = x; } Bool getDeltaPocMSBPresentFlag(Int i) { return m_deltaPocMSBPresentFlag[i]; } Void setDeltaPocMSBPresentFlag(Int i, Bool x) { m_deltaPocMSBPresentFlag[i] = x; } Void setUsed(Int bufferNum, Bool used); Void setDeltaPOC(Int bufferNum, Int deltaPOC); Void setPOC(Int bufferNum, Int deltaPOC); Void setNumberOfPictures(Int numberOfPictures); Void setCheckLTMSBPresent(Int bufferNum, Bool b ); Bool getCheckLTMSBPresent(Int bufferNum); Int getUsed(Int bufferNum); Int getDeltaPOC(Int bufferNum); Int getPOC(Int bufferNum); Int getNumberOfPictures(); Void setNumberOfNegativePictures(Int number) { m_numberOfNegativePictures = number; } Int getNumberOfNegativePictures() { return m_numberOfNegativePictures; } Void setNumberOfPositivePictures(Int number) { m_numberOfPositivePictures = number; } Int getNumberOfPositivePictures() { return m_numberOfPositivePictures; } Void setNumberOfLongtermPictures(Int number) { m_numberOfLongtermPictures = number; } Int getNumberOfLongtermPictures() { return m_numberOfLongtermPictures; } Void setInterRPSPrediction(Bool flag) { m_interRPSPrediction = flag; } Bool getInterRPSPrediction() { return m_interRPSPrediction; } Void setDeltaRIdxMinus1(Int x) { m_deltaRIdxMinus1 = x; } Int getDeltaRIdxMinus1() { return m_deltaRIdxMinus1; } Void setDeltaRPS(Int x) { m_deltaRPS = x; } Int getDeltaRPS() { return m_deltaRPS; } Void setNumRefIdc(Int x) { m_numRefIdc = x; } Int getNumRefIdc() { return m_numRefIdc; } Void setRefIdc(Int bufferNum, Int refIdc); Int getRefIdc(Int bufferNum); Void sortDeltaPOC(); Void printDeltaPOC(); }; /// Reference Picture Set set class class TComRPSList { private: Int m_numberOfReferencePictureSets; TComReferencePictureSet* m_referencePictureSets; public: TComRPSList(); virtual ~TComRPSList(); Void create (Int numberOfEntries); Void destroy (); TComReferencePictureSet* getReferencePictureSet(Int referencePictureSetNum); Int getNumberOfReferencePictureSets(); Void setNumberOfReferencePictureSets(Int numberOfReferencePictureSets); }; /// SCALING_LIST class class TComScalingList { public: TComScalingList(); virtual ~TComScalingList(); Void setScalingListPresentFlag (Bool b) { m_scalingListPresentFlag = b; } Bool getScalingListPresentFlag () { return m_scalingListPresentFlag; } Int* getScalingListAddress (UInt sizeId, UInt listId) { return m_scalingListCoef[sizeId][listId]; } //!< get matrix coefficient const Int* getScalingListAddress (UInt sizeId, UInt listId) const { return m_scalingListCoef[sizeId][listId]; } //!< get matrix coefficient Bool checkPredMode (UInt sizeId, UInt listId); Void setRefMatrixId (UInt sizeId, UInt listId, UInt u) { m_refMatrixId[sizeId][listId] = u; } //!< set reference matrix ID UInt getRefMatrixId (UInt sizeId, UInt listId) { return m_refMatrixId[sizeId][listId]; } //!< get reference matrix ID Int* getScalingListDefaultAddress (UInt sizeId, UInt listId); //!< get default matrix coefficient Void processDefaultMatrix (UInt sizeId, UInt listId); Void setScalingListDC (UInt sizeId, UInt listId, UInt u) { m_scalingListDC[sizeId][listId] = u; } //!< set DC value Int getScalingListDC (UInt sizeId, UInt listId) const { return m_scalingListDC[sizeId][listId]; } //!< get DC value Void checkDcOfMatrix (); Void processRefMatrix (UInt sizeId, UInt listId , UInt refListId ); Bool xParseScalingList (Char* pchFile); private: Void init (); Void destroy (); Void outputScalingLists(std::ostream &os) const; Int m_scalingListDC [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; //!< the DC value of the matrix coefficient for 16x16 Bool m_useDefaultScalingMatrixFlag [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; //!< UseDefaultScalingMatrixFlag UInt m_refMatrixId [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; //!< RefMatrixID Bool m_scalingListPresentFlag; //!< flag for using default matrix UInt m_predMatrixId [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; //!< reference list index Int *m_scalingListCoef [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM]; //!< quantization matrix }; class ProfileTierLevel { Int m_profileSpace; Level::Tier m_tierFlag; Profile::Name m_profileIdc; Bool m_profileCompatibilityFlag[32]; Level::Name m_levelIdc; Bool m_progressiveSourceFlag; Bool m_interlacedSourceFlag; Bool m_nonPackedConstraintFlag; Bool m_frameOnlyConstraintFlag; UInt m_bitDepthConstraintValue; ChromaFormat m_chromaFormatConstraintValue; Bool m_intraConstraintFlag; Bool m_lowerBitRateConstraintFlag; public: ProfileTierLevel(); Int getProfileSpace() const { return m_profileSpace; } Void setProfileSpace(Int x) { m_profileSpace = x; } Level::Tier getTierFlag() const { return m_tierFlag; } Void setTierFlag(Level::Tier x) { m_tierFlag = x; } Profile::Name getProfileIdc() const { return m_profileIdc; } Void setProfileIdc(Profile::Name x) { m_profileIdc = x; } Bool getProfileCompatibilityFlag(Int i) const { return m_profileCompatibilityFlag[i]; } Void setProfileCompatibilityFlag(Int i, Bool x) { m_profileCompatibilityFlag[i] = x; } Level::Name getLevelIdc() const { return m_levelIdc; } Void setLevelIdc(Level::Name x) { m_levelIdc = x; } Bool getProgressiveSourceFlag() const { return m_progressiveSourceFlag; } Void setProgressiveSourceFlag(Bool b) { m_progressiveSourceFlag = b; } Bool getInterlacedSourceFlag() const { return m_interlacedSourceFlag; } Void setInterlacedSourceFlag(Bool b) { m_interlacedSourceFlag = b; } Bool getNonPackedConstraintFlag() const { return m_nonPackedConstraintFlag; } Void setNonPackedConstraintFlag(Bool b) { m_nonPackedConstraintFlag = b; } Bool getFrameOnlyConstraintFlag() const { return m_frameOnlyConstraintFlag; } Void setFrameOnlyConstraintFlag(Bool b) { m_frameOnlyConstraintFlag = b; } UInt getBitDepthConstraint() const { return m_bitDepthConstraintValue; } Void setBitDepthConstraint(UInt bitDepth) { m_bitDepthConstraintValue=bitDepth; } ChromaFormat getChromaFormatConstraint() const { return m_chromaFormatConstraintValue; } Void setChromaFormatConstraint(ChromaFormat fmt) { m_chromaFormatConstraintValue=fmt; } Bool getIntraConstraintFlag() const { return m_intraConstraintFlag; } Void setIntraConstraintFlag(Bool b) { m_intraConstraintFlag = b; } Bool getLowerBitRateConstraintFlag() const { return m_lowerBitRateConstraintFlag; } Void setLowerBitRateConstraintFlag(Bool b) { m_lowerBitRateConstraintFlag = b; } }; class TComPTL { ProfileTierLevel m_generalPTL; ProfileTierLevel m_subLayerPTL [MAX_TLAYER-1]; // max. value of max_sub_layers_minus1 is MAX_TLAYER-1 (= 6) Bool m_subLayerProfilePresentFlag [MAX_TLAYER-1]; Bool m_subLayerLevelPresentFlag [MAX_TLAYER-1]; public: TComPTL(); Bool getSubLayerProfilePresentFlag(Int i) const { return m_subLayerProfilePresentFlag[i]; } Void setSubLayerProfilePresentFlag(Int i, Bool x) { m_subLayerProfilePresentFlag[i] = x; } Bool getSubLayerLevelPresentFlag(Int i) const { return m_subLayerLevelPresentFlag[i]; } Void setSubLayerLevelPresentFlag(Int i, Bool x) { m_subLayerLevelPresentFlag[i] = x; } ProfileTierLevel* getGeneralPTL() { return &m_generalPTL; } ProfileTierLevel* getSubLayerPTL(Int i) { return &m_subLayerPTL[i]; } }; /// VPS class struct HrdSubLayerInfo { Bool fixedPicRateFlag; Bool fixedPicRateWithinCvsFlag; UInt picDurationInTcMinus1; Bool lowDelayHrdFlag; UInt cpbCntMinus1; UInt bitRateValueMinus1[MAX_CPB_CNT][2]; UInt cpbSizeValue [MAX_CPB_CNT][2]; UInt ducpbSizeValue [MAX_CPB_CNT][2]; UInt cbrFlag [MAX_CPB_CNT][2]; UInt duBitRateValue [MAX_CPB_CNT][2]; }; class TComHRD { private: Bool m_nalHrdParametersPresentFlag; Bool m_vclHrdParametersPresentFlag; Bool m_subPicCpbParamsPresentFlag; UInt m_tickDivisorMinus2; UInt m_duCpbRemovalDelayLengthMinus1; Bool m_subPicCpbParamsInPicTimingSEIFlag; UInt m_dpbOutputDelayDuLengthMinus1; UInt m_bitRateScale; UInt m_cpbSizeScale; UInt m_ducpbSizeScale; UInt m_initialCpbRemovalDelayLengthMinus1; UInt m_cpbRemovalDelayLengthMinus1; UInt m_dpbOutputDelayLengthMinus1; UInt m_numDU; HrdSubLayerInfo m_HRD[MAX_TLAYER]; public: TComHRD() :m_nalHrdParametersPresentFlag(0) ,m_vclHrdParametersPresentFlag(0) ,m_subPicCpbParamsPresentFlag(false) ,m_tickDivisorMinus2(0) ,m_duCpbRemovalDelayLengthMinus1(0) ,m_subPicCpbParamsInPicTimingSEIFlag(false) ,m_dpbOutputDelayDuLengthMinus1(0) ,m_bitRateScale(0) ,m_cpbSizeScale(0) ,m_initialCpbRemovalDelayLengthMinus1(23) ,m_cpbRemovalDelayLengthMinus1(23) ,m_dpbOutputDelayLengthMinus1(23) {} virtual ~TComHRD() {} Void setNalHrdParametersPresentFlag ( Bool flag ) { m_nalHrdParametersPresentFlag = flag; } Bool getNalHrdParametersPresentFlag ( ) { return m_nalHrdParametersPresentFlag; } Void setVclHrdParametersPresentFlag ( Bool flag ) { m_vclHrdParametersPresentFlag = flag; } Bool getVclHrdParametersPresentFlag ( ) { return m_vclHrdParametersPresentFlag; } Void setSubPicCpbParamsPresentFlag ( Bool flag ) { m_subPicCpbParamsPresentFlag = flag; } Bool getSubPicCpbParamsPresentFlag ( ) { return m_subPicCpbParamsPresentFlag; } Void setTickDivisorMinus2 ( UInt value ) { m_tickDivisorMinus2 = value; } UInt getTickDivisorMinus2 ( ) { return m_tickDivisorMinus2; } Void setDuCpbRemovalDelayLengthMinus1 ( UInt value ) { m_duCpbRemovalDelayLengthMinus1 = value; } UInt getDuCpbRemovalDelayLengthMinus1 ( ) { return m_duCpbRemovalDelayLengthMinus1; } Void setSubPicCpbParamsInPicTimingSEIFlag ( Bool flag) { m_subPicCpbParamsInPicTimingSEIFlag = flag; } Bool getSubPicCpbParamsInPicTimingSEIFlag () { return m_subPicCpbParamsInPicTimingSEIFlag; } Void setDpbOutputDelayDuLengthMinus1 (UInt value ) { m_dpbOutputDelayDuLengthMinus1 = value; } UInt getDpbOutputDelayDuLengthMinus1 () { return m_dpbOutputDelayDuLengthMinus1; } Void setBitRateScale ( UInt value ) { m_bitRateScale = value; } UInt getBitRateScale ( ) { return m_bitRateScale; } Void setCpbSizeScale ( UInt value ) { m_cpbSizeScale = value; } UInt getCpbSizeScale ( ) { return m_cpbSizeScale; } Void setDuCpbSizeScale ( UInt value ) { m_ducpbSizeScale = value; } UInt getDuCpbSizeScale ( ) { return m_ducpbSizeScale; } Void setInitialCpbRemovalDelayLengthMinus1( UInt value ) { m_initialCpbRemovalDelayLengthMinus1 = value; } UInt getInitialCpbRemovalDelayLengthMinus1( ) { return m_initialCpbRemovalDelayLengthMinus1; } Void setCpbRemovalDelayLengthMinus1 ( UInt value ) { m_cpbRemovalDelayLengthMinus1 = value; } UInt getCpbRemovalDelayLengthMinus1 ( ) { return m_cpbRemovalDelayLengthMinus1; } Void setDpbOutputDelayLengthMinus1 ( UInt value ) { m_dpbOutputDelayLengthMinus1 = value; } UInt getDpbOutputDelayLengthMinus1 ( ) { return m_dpbOutputDelayLengthMinus1; } Void setFixedPicRateFlag ( Int layer, Bool flag ) { m_HRD[layer].fixedPicRateFlag = flag; } Bool getFixedPicRateFlag ( Int layer ) { return m_HRD[layer].fixedPicRateFlag; } Void setFixedPicRateWithinCvsFlag ( Int layer, Bool flag ) { m_HRD[layer].fixedPicRateWithinCvsFlag = flag; } Bool getFixedPicRateWithinCvsFlag ( Int layer ) { return m_HRD[layer].fixedPicRateWithinCvsFlag; } Void setPicDurationInTcMinus1 ( Int layer, UInt value ) { m_HRD[layer].picDurationInTcMinus1 = value; } UInt getPicDurationInTcMinus1 ( Int layer ) { return m_HRD[layer].picDurationInTcMinus1; } Void setLowDelayHrdFlag ( Int layer, Bool flag ) { m_HRD[layer].lowDelayHrdFlag = flag; } Bool getLowDelayHrdFlag ( Int layer ) { return m_HRD[layer].lowDelayHrdFlag; } Void setCpbCntMinus1 ( Int layer, UInt value ) { m_HRD[layer].cpbCntMinus1 = value; } UInt getCpbCntMinus1 ( Int layer ) { return m_HRD[layer].cpbCntMinus1; } Void setBitRateValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl, UInt value ) { m_HRD[layer].bitRateValueMinus1[cpbcnt][nalOrVcl] = value; } UInt getBitRateValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl ) { return m_HRD[layer].bitRateValueMinus1[cpbcnt][nalOrVcl]; } Void setCpbSizeValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl, UInt value ) { m_HRD[layer].cpbSizeValue[cpbcnt][nalOrVcl] = value; } UInt getCpbSizeValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl ) { return m_HRD[layer].cpbSizeValue[cpbcnt][nalOrVcl]; } Void setDuCpbSizeValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl, UInt value ) { m_HRD[layer].ducpbSizeValue[cpbcnt][nalOrVcl] = value; } UInt getDuCpbSizeValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl ) { return m_HRD[layer].ducpbSizeValue[cpbcnt][nalOrVcl]; } Void setDuBitRateValueMinus1 ( Int layer, Int cpbcnt, Int nalOrVcl, UInt value ) { m_HRD[layer].duBitRateValue[cpbcnt][nalOrVcl] = value; } UInt getDuBitRateValueMinus1 (Int layer, Int cpbcnt, Int nalOrVcl ) { return m_HRD[layer].duBitRateValue[cpbcnt][nalOrVcl]; } Void setCbrFlag ( Int layer, Int cpbcnt, Int nalOrVcl, UInt value ) { m_HRD[layer].cbrFlag[cpbcnt][nalOrVcl] = value; } Bool getCbrFlag ( Int layer, Int cpbcnt, Int nalOrVcl ) { return m_HRD[layer].cbrFlag[cpbcnt][nalOrVcl]; } Void setNumDU ( UInt value ) { m_numDU = value; } UInt getNumDU ( ) { return m_numDU; } Bool getCpbDpbDelaysPresentFlag() { return getNalHrdParametersPresentFlag() || getVclHrdParametersPresentFlag(); } }; class TimingInfo { Bool m_timingInfoPresentFlag; UInt m_numUnitsInTick; UInt m_timeScale; Bool m_pocProportionalToTimingFlag; Int m_numTicksPocDiffOneMinus1; public: TimingInfo() : m_timingInfoPresentFlag(false) , m_numUnitsInTick(1001) , m_timeScale(60000) , m_pocProportionalToTimingFlag(false) , m_numTicksPocDiffOneMinus1(0) {} Void setTimingInfoPresentFlag ( Bool flag ) { m_timingInfoPresentFlag = flag; } Bool getTimingInfoPresentFlag ( ) { return m_timingInfoPresentFlag; } Void setNumUnitsInTick ( UInt value ) { m_numUnitsInTick = value; } UInt getNumUnitsInTick ( ) { return m_numUnitsInTick; } Void setTimeScale ( UInt value ) { m_timeScale = value; } UInt getTimeScale ( ) { return m_timeScale; } Bool getPocProportionalToTimingFlag ( ) { return m_pocProportionalToTimingFlag; } Void setPocProportionalToTimingFlag (Bool x ) { m_pocProportionalToTimingFlag = x; } Int getNumTicksPocDiffOneMinus1 ( ) { return m_numTicksPocDiffOneMinus1; } Void setNumTicksPocDiffOneMinus1 (Int x ) { m_numTicksPocDiffOneMinus1 = x; } }; struct ChromaQpAdj { union { struct { Int CbOffset; Int CrOffset; } comp; Int offset[2]; /* two chroma components */ } u; }; class TComVPS { private: Int m_VPSId; UInt m_uiMaxTLayers; UInt m_uiMaxLayers; Bool m_bTemporalIdNestingFlag; UInt m_numReorderPics[MAX_TLAYER]; UInt m_uiMaxDecPicBuffering[MAX_TLAYER]; UInt m_uiMaxLatencyIncrease[MAX_TLAYER]; // Really max latency increase plus 1 (value 0 expresses no limit) UInt m_numHrdParameters; UInt m_maxNuhReservedZeroLayerId; TComHRD* m_hrdParameters; UInt* m_hrdOpSetIdx; Bool* m_cprmsPresentFlag; UInt m_numOpSets; Bool m_layerIdIncludedFlag[MAX_VPS_OP_SETS_PLUS1][MAX_VPS_NUH_RESERVED_ZERO_LAYER_ID_PLUS1]; TComPTL m_pcPTL; TimingInfo m_timingInfo; public: TComVPS(); virtual ~TComVPS(); Void createHrdParamBuffer() { m_hrdParameters = new TComHRD[ getNumHrdParameters() ]; m_hrdOpSetIdx = new UInt [ getNumHrdParameters() ]; m_cprmsPresentFlag = new Bool [ getNumHrdParameters() ]; } TComHRD* getHrdParameters ( UInt i ) { return &m_hrdParameters[ i ]; } UInt getHrdOpSetIdx ( UInt i ) { return m_hrdOpSetIdx[ i ]; } Void setHrdOpSetIdx ( UInt val, UInt i ) { m_hrdOpSetIdx[ i ] = val; } Bool getCprmsPresentFlag ( UInt i ) { return m_cprmsPresentFlag[ i ]; } Void setCprmsPresentFlag ( Bool val, UInt i ) { m_cprmsPresentFlag[ i ] = val; } Int getVPSId () { return m_VPSId; } Void setVPSId (Int i) { m_VPSId = i; } UInt getMaxTLayers () { return m_uiMaxTLayers; } Void setMaxTLayers (UInt t) { m_uiMaxTLayers = t; } UInt getMaxLayers () { return m_uiMaxLayers; } Void setMaxLayers (UInt l) { m_uiMaxLayers = l; } Bool getTemporalNestingFlag () { return m_bTemporalIdNestingFlag; } Void setTemporalNestingFlag (Bool t) { m_bTemporalIdNestingFlag = t; } Void setNumReorderPics(UInt v, UInt tLayer) { m_numReorderPics[tLayer] = v; } UInt getNumReorderPics(UInt tLayer) { return m_numReorderPics[tLayer]; } Void setMaxDecPicBuffering(UInt v, UInt tLayer) { assert(tLayer < MAX_TLAYER); m_uiMaxDecPicBuffering[tLayer] = v; } UInt getMaxDecPicBuffering(UInt tLayer) { return m_uiMaxDecPicBuffering[tLayer]; } Void setMaxLatencyIncrease(UInt v, UInt tLayer) { m_uiMaxLatencyIncrease[tLayer] = v; } UInt getMaxLatencyIncrease(UInt tLayer) { return m_uiMaxLatencyIncrease[tLayer]; } UInt getNumHrdParameters() { return m_numHrdParameters; } Void setNumHrdParameters(UInt v) { m_numHrdParameters = v; } UInt getMaxNuhReservedZeroLayerId() { return m_maxNuhReservedZeroLayerId; } Void setMaxNuhReservedZeroLayerId(UInt v) { m_maxNuhReservedZeroLayerId = v; } UInt getMaxOpSets() { return m_numOpSets; } Void setMaxOpSets(UInt v) { m_numOpSets = v; } Bool getLayerIdIncludedFlag(UInt opsIdx, UInt id) { return m_layerIdIncludedFlag[opsIdx][id]; } Void setLayerIdIncludedFlag(Bool v, UInt opsIdx, UInt id) { m_layerIdIncludedFlag[opsIdx][id] = v; } TComPTL* getPTL() { return &m_pcPTL; } TimingInfo* getTimingInfo() { return &m_timingInfo; } }; class Window { private: Bool m_enabledFlag; Int m_winLeftOffset; Int m_winRightOffset; Int m_winTopOffset; Int m_winBottomOffset; public: Window() : m_enabledFlag (false) , m_winLeftOffset (0) , m_winRightOffset (0) , m_winTopOffset (0) , m_winBottomOffset (0) { } Bool getWindowEnabledFlag() const { return m_enabledFlag; } Void resetWindow() { m_enabledFlag = false; m_winLeftOffset = m_winRightOffset = m_winTopOffset = m_winBottomOffset = 0; } Int getWindowLeftOffset() const { return m_enabledFlag ? m_winLeftOffset : 0; } Void setWindowLeftOffset(Int val) { m_winLeftOffset = val; m_enabledFlag = true; } Int getWindowRightOffset() const { return m_enabledFlag ? m_winRightOffset : 0; } Void setWindowRightOffset(Int val) { m_winRightOffset = val; m_enabledFlag = true; } Int getWindowTopOffset() const { return m_enabledFlag ? m_winTopOffset : 0; } Void setWindowTopOffset(Int val) { m_winTopOffset = val; m_enabledFlag = true; } Int getWindowBottomOffset() const { return m_enabledFlag ? m_winBottomOffset: 0; } Void setWindowBottomOffset(Int val) { m_winBottomOffset = val; m_enabledFlag = true; } Void setWindow(Int offsetLeft, Int offsetLRight, Int offsetLTop, Int offsetLBottom) { m_enabledFlag = true; m_winLeftOffset = offsetLeft; m_winRightOffset = offsetLRight; m_winTopOffset = offsetLTop; m_winBottomOffset = offsetLBottom; } }; class TComVUI { private: Bool m_aspectRatioInfoPresentFlag; Int m_aspectRatioIdc; Int m_sarWidth; Int m_sarHeight; Bool m_overscanInfoPresentFlag; Bool m_overscanAppropriateFlag; Bool m_videoSignalTypePresentFlag; Int m_videoFormat; Bool m_videoFullRangeFlag; Bool m_colourDescriptionPresentFlag; Int m_colourPrimaries; Int m_transferCharacteristics; Int m_matrixCoefficients; Bool m_chromaLocInfoPresentFlag; Int m_chromaSampleLocTypeTopField; Int m_chromaSampleLocTypeBottomField; Bool m_neutralChromaIndicationFlag; Bool m_fieldSeqFlag; Window m_defaultDisplayWindow; Bool m_frameFieldInfoPresentFlag; Bool m_hrdParametersPresentFlag; Bool m_bitstreamRestrictionFlag; Bool m_tilesFixedStructureFlag; Bool m_motionVectorsOverPicBoundariesFlag; Bool m_restrictedRefPicListsFlag; Int m_minSpatialSegmentationIdc; Int m_maxBytesPerPicDenom; Int m_maxBitsPerMinCuDenom; Int m_log2MaxMvLengthHorizontal; Int m_log2MaxMvLengthVertical; TComHRD m_hrdParameters; TimingInfo m_timingInfo; public: TComVUI() :m_aspectRatioInfoPresentFlag(false) //TODO: This initialiser list contains magic numbers ,m_aspectRatioIdc(0) ,m_sarWidth(0) ,m_sarHeight(0) ,m_overscanInfoPresentFlag(false) ,m_overscanAppropriateFlag(false) ,m_videoSignalTypePresentFlag(false) ,m_videoFormat(5) ,m_videoFullRangeFlag(false) ,m_colourDescriptionPresentFlag(false) ,m_colourPrimaries(2) ,m_transferCharacteristics(2) ,m_matrixCoefficients(2) ,m_chromaLocInfoPresentFlag(false) ,m_chromaSampleLocTypeTopField(0) ,m_chromaSampleLocTypeBottomField(0) ,m_neutralChromaIndicationFlag(false) ,m_fieldSeqFlag(false) ,m_frameFieldInfoPresentFlag(false) ,m_hrdParametersPresentFlag(false) ,m_bitstreamRestrictionFlag(false) ,m_tilesFixedStructureFlag(false) ,m_motionVectorsOverPicBoundariesFlag(true) ,m_restrictedRefPicListsFlag(1) ,m_minSpatialSegmentationIdc(0) ,m_maxBytesPerPicDenom(2) ,m_maxBitsPerMinCuDenom(1) ,m_log2MaxMvLengthHorizontal(15) ,m_log2MaxMvLengthVertical(15) {} virtual ~TComVUI() {} Bool getAspectRatioInfoPresentFlag() { return m_aspectRatioInfoPresentFlag; } Void setAspectRatioInfoPresentFlag(Bool i) { m_aspectRatioInfoPresentFlag = i; } Int getAspectRatioIdc() { return m_aspectRatioIdc; } Void setAspectRatioIdc(Int i) { m_aspectRatioIdc = i; } Int getSarWidth() { return m_sarWidth; } Void setSarWidth(Int i) { m_sarWidth = i; } Int getSarHeight() { return m_sarHeight; } Void setSarHeight(Int i) { m_sarHeight = i; } Bool getOverscanInfoPresentFlag() { return m_overscanInfoPresentFlag; } Void setOverscanInfoPresentFlag(Bool i) { m_overscanInfoPresentFlag = i; } Bool getOverscanAppropriateFlag() { return m_overscanAppropriateFlag; } Void setOverscanAppropriateFlag(Bool i) { m_overscanAppropriateFlag = i; } Bool getVideoSignalTypePresentFlag() { return m_videoSignalTypePresentFlag; } Void setVideoSignalTypePresentFlag(Bool i) { m_videoSignalTypePresentFlag = i; } Int getVideoFormat() { return m_videoFormat; } Void setVideoFormat(Int i) { m_videoFormat = i; } Bool getVideoFullRangeFlag() { return m_videoFullRangeFlag; } Void setVideoFullRangeFlag(Bool i) { m_videoFullRangeFlag = i; } Bool getColourDescriptionPresentFlag() { return m_colourDescriptionPresentFlag; } Void setColourDescriptionPresentFlag(Bool i) { m_colourDescriptionPresentFlag = i; } Int getColourPrimaries() { return m_colourPrimaries; } Void setColourPrimaries(Int i) { m_colourPrimaries = i; } Int getTransferCharacteristics() { return m_transferCharacteristics; } Void setTransferCharacteristics(Int i) { m_transferCharacteristics = i; } Int getMatrixCoefficients() { return m_matrixCoefficients; } Void setMatrixCoefficients(Int i) { m_matrixCoefficients = i; } Bool getChromaLocInfoPresentFlag() { return m_chromaLocInfoPresentFlag; } Void setChromaLocInfoPresentFlag(Bool i) { m_chromaLocInfoPresentFlag = i; } Int getChromaSampleLocTypeTopField() { return m_chromaSampleLocTypeTopField; } Void setChromaSampleLocTypeTopField(Int i) { m_chromaSampleLocTypeTopField = i; } Int getChromaSampleLocTypeBottomField() { return m_chromaSampleLocTypeBottomField; } Void setChromaSampleLocTypeBottomField(Int i) { m_chromaSampleLocTypeBottomField = i; } Bool getNeutralChromaIndicationFlag() { return m_neutralChromaIndicationFlag; } Void setNeutralChromaIndicationFlag(Bool i) { m_neutralChromaIndicationFlag = i; } Bool getFieldSeqFlag() { return m_fieldSeqFlag; } Void setFieldSeqFlag(Bool i) { m_fieldSeqFlag = i; } Bool getFrameFieldInfoPresentFlag() { return m_frameFieldInfoPresentFlag; } Void setFrameFieldInfoPresentFlag(Bool i) { m_frameFieldInfoPresentFlag = i; } Window& getDefaultDisplayWindow() { return m_defaultDisplayWindow; } Void setDefaultDisplayWindow(Window& defaultDisplayWindow ) { m_defaultDisplayWindow = defaultDisplayWindow; } Bool getHrdParametersPresentFlag() { return m_hrdParametersPresentFlag; } Void setHrdParametersPresentFlag(Bool i) { m_hrdParametersPresentFlag = i; } Bool getBitstreamRestrictionFlag() { return m_bitstreamRestrictionFlag; } Void setBitstreamRestrictionFlag(Bool i) { m_bitstreamRestrictionFlag = i; } Bool getTilesFixedStructureFlag() { return m_tilesFixedStructureFlag; } Void setTilesFixedStructureFlag(Bool i) { m_tilesFixedStructureFlag = i; } Bool getMotionVectorsOverPicBoundariesFlag() { return m_motionVectorsOverPicBoundariesFlag; } Void setMotionVectorsOverPicBoundariesFlag(Bool i) { m_motionVectorsOverPicBoundariesFlag = i; } Bool getRestrictedRefPicListsFlag() { return m_restrictedRefPicListsFlag; } Void setRestrictedRefPicListsFlag(Bool b) { m_restrictedRefPicListsFlag = b; } Int getMinSpatialSegmentationIdc() { return m_minSpatialSegmentationIdc; } Void setMinSpatialSegmentationIdc(Int i) { m_minSpatialSegmentationIdc = i; } Int getMaxBytesPerPicDenom() { return m_maxBytesPerPicDenom; } Void setMaxBytesPerPicDenom(Int i) { m_maxBytesPerPicDenom = i; } Int getMaxBitsPerMinCuDenom() { return m_maxBitsPerMinCuDenom; } Void setMaxBitsPerMinCuDenom(Int i) { m_maxBitsPerMinCuDenom = i; } Int getLog2MaxMvLengthHorizontal() { return m_log2MaxMvLengthHorizontal; } Void setLog2MaxMvLengthHorizontal(Int i) { m_log2MaxMvLengthHorizontal = i; } Int getLog2MaxMvLengthVertical() { return m_log2MaxMvLengthVertical; } Void setLog2MaxMvLengthVertical(Int i) { m_log2MaxMvLengthVertical = i; } TComHRD* getHrdParameters () { return &m_hrdParameters; } TimingInfo* getTimingInfo() { return &m_timingInfo; } }; /// SPS class class TComSPS { private: Int m_SPSId; Int m_VPSId; ChromaFormat m_chromaFormatIdc; UInt m_uiMaxTLayers; // maximum number of temporal layers // Structure UInt m_picWidthInLumaSamples; UInt m_picHeightInLumaSamples; Int m_log2MinCodingBlockSize; Int m_log2DiffMaxMinCodingBlockSize; UInt m_uiMaxCUWidth; UInt m_uiMaxCUHeight; UInt m_uiMaxCUDepth; Window m_conformanceWindow; TComRPSList m_RPSList; Bool m_bLongTermRefsPresent; Bool m_TMVPFlagsPresent; Int m_numReorderPics[MAX_TLAYER]; // Tool list UInt m_uiQuadtreeTULog2MaxSize; UInt m_uiQuadtreeTULog2MinSize; UInt m_uiQuadtreeTUMaxDepthInter; UInt m_uiQuadtreeTUMaxDepthIntra; Bool m_usePCM; UInt m_pcmLog2MaxSize; UInt m_uiPCMLog2MinSize; Bool m_useAMP; // Parameter Int m_uiBitDepth[MAX_NUM_CHANNEL_TYPE]; Int m_qpBDOffset[MAX_NUM_CHANNEL_TYPE]; Bool m_useExtendedPrecision; Bool m_useHighPrecisionPredictionWeighting; Bool m_useResidualRotation; Bool m_useSingleSignificanceMapContext; Bool m_useGolombRiceParameterAdaptation; Bool m_alignCABACBeforeBypass; Bool m_useResidualDPCM[NUMBER_OF_RDPCM_SIGNALLING_MODES]; UInt m_uiPCMBitDepth[MAX_NUM_CHANNEL_TYPE]; Bool m_bPCMFilterDisableFlag; Bool m_disableIntraReferenceSmoothing; UInt m_uiBitsForPOC; UInt m_numLongTermRefPicSPS; UInt m_ltRefPicPocLsbSps[MAX_NUM_LONG_TERM_REF_PICS]; Bool m_usedByCurrPicLtSPSFlag[MAX_NUM_LONG_TERM_REF_PICS]; // Max physical transform size UInt m_uiMaxTrSize; Int m_iAMPAcc[MAX_CU_DEPTH]; Bool m_bUseSAO; Bool m_bTemporalIdNestingFlag; // temporal_id_nesting_flag Bool m_scalingListEnabledFlag; Bool m_scalingListPresentFlag; TComScalingList* m_scalingList; //!< ScalingList class pointer UInt m_uiMaxDecPicBuffering[MAX_TLAYER]; UInt m_uiMaxLatencyIncrease[MAX_TLAYER]; // Really max latency increase plus 1 (value 0 expresses no limit) Bool m_useDF; Bool m_useStrongIntraSmoothing; Bool m_vuiParametersPresentFlag; TComVUI m_vuiParameters; static const Int m_winUnitX[MAX_CHROMA_FORMAT_IDC+1]; static const Int m_winUnitY[MAX_CHROMA_FORMAT_IDC+1]; TComPTL m_pcPTL; #if O0043_BEST_EFFORT_DECODING UInt m_forceDecodeBitDepth; // 0 = do not force the decoder's bit depth, other = force the decoder's bit depth to this value (best effort decoding) #endif public: TComSPS(); virtual ~TComSPS(); #if O0043_BEST_EFFORT_DECODING Void setForceDecodeBitDepth(UInt bitDepth) { m_forceDecodeBitDepth = bitDepth; } UInt getForceDecodeBitDepth() const { return m_forceDecodeBitDepth; } #endif Int getVPSId () { return m_VPSId; } Void setVPSId (Int i) { m_VPSId = i; } Int getSPSId () { return m_SPSId; } Void setSPSId (Int i) { m_SPSId = i; } ChromaFormat getChromaFormatIdc () { return m_chromaFormatIdc; } Void setChromaFormatIdc (ChromaFormat i) { m_chromaFormatIdc = i; } static Int getWinUnitX (Int chromaFormatIdc) { assert (chromaFormatIdc >= 0 && chromaFormatIdc <= MAX_CHROMA_FORMAT_IDC); return m_winUnitX[chromaFormatIdc]; } static Int getWinUnitY (Int chromaFormatIdc) { assert (chromaFormatIdc >= 0 && chromaFormatIdc <= MAX_CHROMA_FORMAT_IDC); return m_winUnitY[chromaFormatIdc]; } // structure Void setPicWidthInLumaSamples ( UInt u ) { m_picWidthInLumaSamples = u; } UInt getPicWidthInLumaSamples () { return m_picWidthInLumaSamples; } Void setPicHeightInLumaSamples ( UInt u ) { m_picHeightInLumaSamples = u; } UInt getPicHeightInLumaSamples () { return m_picHeightInLumaSamples; } Window& getConformanceWindow() { return m_conformanceWindow; } Void setConformanceWindow(Window& conformanceWindow ) { m_conformanceWindow = conformanceWindow; } UInt getNumLongTermRefPicSPS() { return m_numLongTermRefPicSPS; } Void setNumLongTermRefPicSPS(UInt val) { m_numLongTermRefPicSPS = val; } UInt getLtRefPicPocLsbSps(UInt index) { assert( index < MAX_NUM_LONG_TERM_REF_PICS ); return m_ltRefPicPocLsbSps[index]; } Void setLtRefPicPocLsbSps(UInt index, UInt val) { assert( index < MAX_NUM_LONG_TERM_REF_PICS ); m_ltRefPicPocLsbSps[index] = val; } Bool getUsedByCurrPicLtSPSFlag(Int i) { assert( i < MAX_NUM_LONG_TERM_REF_PICS ); return m_usedByCurrPicLtSPSFlag[i];} Void setUsedByCurrPicLtSPSFlag(Int i, Bool x) { assert( i < MAX_NUM_LONG_TERM_REF_PICS ); m_usedByCurrPicLtSPSFlag[i] = x;} Int getLog2MinCodingBlockSize() const { return m_log2MinCodingBlockSize; } Void setLog2MinCodingBlockSize(Int val) { m_log2MinCodingBlockSize = val; } Int getLog2DiffMaxMinCodingBlockSize() const { return m_log2DiffMaxMinCodingBlockSize; } Void setLog2DiffMaxMinCodingBlockSize(Int val) { m_log2DiffMaxMinCodingBlockSize = val; } Void setMaxCUWidth ( UInt u ) { m_uiMaxCUWidth = u; } UInt getMaxCUWidth () { return m_uiMaxCUWidth; } Void setMaxCUHeight ( UInt u ) { m_uiMaxCUHeight = u; } UInt getMaxCUHeight () { return m_uiMaxCUHeight; } Void setMaxCUDepth ( UInt u ) { m_uiMaxCUDepth = u; } UInt getMaxCUDepth () { return m_uiMaxCUDepth; } Void setUsePCM ( Bool b ) { m_usePCM = b; } Bool getUsePCM () { return m_usePCM; } Void setPCMLog2MaxSize ( UInt u ) { m_pcmLog2MaxSize = u; } UInt getPCMLog2MaxSize () { return m_pcmLog2MaxSize; } Void setPCMLog2MinSize ( UInt u ) { m_uiPCMLog2MinSize = u; } UInt getPCMLog2MinSize () { return m_uiPCMLog2MinSize; } Void setBitsForPOC ( UInt u ) { m_uiBitsForPOC = u; } UInt getBitsForPOC () { return m_uiBitsForPOC; } Bool getUseAMP() { return m_useAMP; } Void setUseAMP( Bool b ) { m_useAMP = b; } Void setQuadtreeTULog2MaxSize( UInt u ) { m_uiQuadtreeTULog2MaxSize = u; } UInt getQuadtreeTULog2MaxSize() { return m_uiQuadtreeTULog2MaxSize; } Void setQuadtreeTULog2MinSize( UInt u ) { m_uiQuadtreeTULog2MinSize = u; } UInt getQuadtreeTULog2MinSize() { return m_uiQuadtreeTULog2MinSize; } Void setQuadtreeTUMaxDepthInter( UInt u ) { m_uiQuadtreeTUMaxDepthInter = u; } Void setQuadtreeTUMaxDepthIntra( UInt u ) { m_uiQuadtreeTUMaxDepthIntra = u; } UInt getQuadtreeTUMaxDepthInter() { return m_uiQuadtreeTUMaxDepthInter; } UInt getQuadtreeTUMaxDepthIntra() { return m_uiQuadtreeTUMaxDepthIntra; } Void setNumReorderPics(Int i, UInt tlayer) { m_numReorderPics[tlayer] = i; } Int getNumReorderPics(UInt tlayer) { return m_numReorderPics[tlayer]; } Void createRPSList( Int numRPS ); TComRPSList* getRPSList() { return &m_RPSList; } Bool getLongTermRefsPresent() { return m_bLongTermRefsPresent; } Void setLongTermRefsPresent(Bool b) { m_bLongTermRefsPresent=b; } Bool getTMVPFlagsPresent() { return m_TMVPFlagsPresent; } Void setTMVPFlagsPresent(Bool b) { m_TMVPFlagsPresent=b; } // physical transform Void setMaxTrSize ( UInt u ) { m_uiMaxTrSize = u; } UInt getMaxTrSize () { return m_uiMaxTrSize; } // AMP accuracy Int getAMPAcc ( UInt uiDepth ) { return m_iAMPAcc[uiDepth]; } Void setAMPAcc ( UInt uiDepth, Int iAccu ) { assert( uiDepth < g_uiMaxCUDepth); m_iAMPAcc[uiDepth] = iAccu; } // Bit-depth Int getBitDepth (ChannelType type) { return m_uiBitDepth[type]; } Void setBitDepth (ChannelType type, Int u ) { m_uiBitDepth[type] = u; } Int getDifferentialLumaChromaBitDepth() const { return Int(m_uiBitDepth[CHANNEL_TYPE_LUMA]) - Int(m_uiBitDepth[CHANNEL_TYPE_CHROMA]); } Int getQpBDOffset (ChannelType type) const { return m_qpBDOffset[type]; } Void setQpBDOffset (ChannelType type, Int i) { m_qpBDOffset[type] = i; } Bool getUseExtendedPrecision() const { return m_useExtendedPrecision; } Void setUseExtendedPrecision(Bool value) { m_useExtendedPrecision = value; } Bool getUseHighPrecisionPredictionWeighting() const { return m_useHighPrecisionPredictionWeighting; } Void setUseHighPrecisionPredictionWeighting(Bool value) { m_useHighPrecisionPredictionWeighting = value; } Void setUseSAO (Bool bVal) {m_bUseSAO = bVal;} Bool getUseSAO () {return m_bUseSAO;} Bool getUseResidualRotation () const { return m_useResidualRotation; } Void setUseResidualRotation (const Bool value) { m_useResidualRotation = value; } Bool getUseSingleSignificanceMapContext() const { return m_useSingleSignificanceMapContext; } Void setUseSingleSignificanceMapContext(const Bool value) { m_useSingleSignificanceMapContext = value; } Bool getUseGolombRiceParameterAdaptation() const { return m_useGolombRiceParameterAdaptation; } Void setUseGolombRiceParameterAdaptation(const Bool value) { m_useGolombRiceParameterAdaptation = value; } Bool getAlignCABACBeforeBypass () const { return m_alignCABACBeforeBypass; } Void setAlignCABACBeforeBypass (const Bool value) { m_alignCABACBeforeBypass = value; } Bool getUseResidualDPCM (const RDPCMSignallingMode signallingMode) const { return m_useResidualDPCM[signallingMode]; } Void setUseResidualDPCM (const RDPCMSignallingMode signallingMode, const Bool value) { m_useResidualDPCM[signallingMode] = value; } UInt getMaxTLayers() { return m_uiMaxTLayers; } Void setMaxTLayers( UInt uiMaxTLayers ) { assert( uiMaxTLayers <= MAX_TLAYER ); m_uiMaxTLayers = uiMaxTLayers; } Bool getTemporalIdNestingFlag() { return m_bTemporalIdNestingFlag; } Void setTemporalIdNestingFlag( Bool bValue ) { m_bTemporalIdNestingFlag = bValue; } UInt getPCMBitDepth (ChannelType type) const { return m_uiPCMBitDepth[type]; } Void setPCMBitDepth (ChannelType type, UInt u) { m_uiPCMBitDepth[type] = u; } Void setPCMFilterDisableFlag ( Bool bValue ) { m_bPCMFilterDisableFlag = bValue; } Bool getPCMFilterDisableFlag () { return m_bPCMFilterDisableFlag; } Void setDisableIntraReferenceSmoothing (Bool bValue) { m_disableIntraReferenceSmoothing=bValue; } Bool getDisableIntraReferenceSmoothing () const { return m_disableIntraReferenceSmoothing; } Bool getScalingListFlag () { return m_scalingListEnabledFlag; } Void setScalingListFlag ( Bool b ) { m_scalingListEnabledFlag = b; } Bool getScalingListPresentFlag() { return m_scalingListPresentFlag; } Void setScalingListPresentFlag( Bool b ) { m_scalingListPresentFlag = b; } Void setScalingList ( TComScalingList *scalingList); TComScalingList* getScalingList () { return m_scalingList; } //!< get ScalingList class pointer in SPS UInt getMaxDecPicBuffering (UInt tlayer) { return m_uiMaxDecPicBuffering[tlayer]; } Void setMaxDecPicBuffering ( UInt ui, UInt tlayer ) { assert(tlayer < MAX_TLAYER); m_uiMaxDecPicBuffering[tlayer] = ui; } UInt getMaxLatencyIncrease (UInt tlayer) { return m_uiMaxLatencyIncrease[tlayer]; } Void setMaxLatencyIncrease ( UInt ui , UInt tlayer) { m_uiMaxLatencyIncrease[tlayer] = ui; } Void setUseStrongIntraSmoothing (Bool bVal) {m_useStrongIntraSmoothing = bVal;} Bool getUseStrongIntraSmoothing () {return m_useStrongIntraSmoothing;} Bool getVuiParametersPresentFlag() { return m_vuiParametersPresentFlag; } Void setVuiParametersPresentFlag(Bool b) { m_vuiParametersPresentFlag = b; } TComVUI* getVuiParameters() { return &m_vuiParameters; } Void setHrdParameters( UInt frameRate, UInt numDU, UInt bitRate, Bool randomAccess ); TComPTL* getPTL() { return &m_pcPTL; } }; /// Reference Picture Lists class class TComRefPicListModification { private: UInt m_bRefPicListModificationFlagL0; UInt m_bRefPicListModificationFlagL1; UInt m_RefPicSetIdxL0[REF_PIC_LIST_NUM_IDX]; UInt m_RefPicSetIdxL1[REF_PIC_LIST_NUM_IDX]; public: TComRefPicListModification(); virtual ~TComRefPicListModification(); Void create (); Void destroy (); Bool getRefPicListModificationFlagL0() { return m_bRefPicListModificationFlagL0; } Void setRefPicListModificationFlagL0(Bool flag) { m_bRefPicListModificationFlagL0 = flag; } Bool getRefPicListModificationFlagL1() { return m_bRefPicListModificationFlagL1; } Void setRefPicListModificationFlagL1(Bool flag) { m_bRefPicListModificationFlagL1 = flag; } Void setRefPicSetIdxL0(UInt idx, UInt refPicSetIdx) { assert(idx m_tileColumnWidth; std::vector m_tileRowHeight; Int m_numSubstreams; Int m_signHideFlag; Bool m_cabacInitPresentFlag; UInt m_encCABACTableIdx; // Used to transmit table selection across slices Bool m_sliceHeaderExtensionPresentFlag; Bool m_loopFilterAcrossSlicesEnabledFlag; Bool m_deblockingFilterControlPresentFlag; Bool m_deblockingFilterOverrideEnabledFlag; Bool m_picDisableDeblockingFilterFlag; Int m_deblockingFilterBetaOffsetDiv2; //< beta offset for deblocking filter Int m_deblockingFilterTcOffsetDiv2; //< tc offset for deblocking filter Bool m_scalingListPresentFlag; TComScalingList* m_scalingList; //!< ScalingList class pointer Bool m_listsModificationPresentFlag; UInt m_log2ParallelMergeLevelMinus2; Int m_numExtraSliceHeaderBits; public: TComPPS(); virtual ~TComPPS(); Int getPPSId () { return m_PPSId; } Void setPPSId (Int i) { m_PPSId = i; } Int getSPSId () { return m_SPSId; } Void setSPSId (Int i) { m_SPSId = i; } Int getPicInitQPMinus26 () { return m_picInitQPMinus26; } Void setPicInitQPMinus26 ( Int i ) { m_picInitQPMinus26 = i; } Bool getUseDQP () { return m_useDQP; } Void setUseDQP ( Bool b ) { m_useDQP = b; } Bool getConstrainedIntraPred () { return m_bConstrainedIntraPred; } Void setConstrainedIntraPred ( Bool b ) { m_bConstrainedIntraPred = b; } Bool getSliceChromaQpFlag () { return m_bSliceChromaQpFlag; } Void setSliceChromaQpFlag ( Bool b ) { m_bSliceChromaQpFlag = b; } Void setSPS ( TComSPS* pcSPS ) { m_pcSPS = pcSPS; } TComSPS* getSPS () { return m_pcSPS; } Void setMaxCuDQPDepth ( UInt u ) { m_uiMaxCuDQPDepth = u; } UInt getMaxCuDQPDepth () { return m_uiMaxCuDQPDepth;} Void setMinCuDQPSize ( UInt u ) { m_uiMinCuDQPSize = u; } UInt getMinCuDQPSize () { return m_uiMinCuDQPSize; } Void setQpOffset(ComponentID compID, Int i ) { if (compID==COMPONENT_Cb) m_chromaCbQpOffset = i; else if (compID==COMPONENT_Cr) m_chromaCrQpOffset = i; else assert(0); } Int getQpOffset(ComponentID compID) const { return (compID==COMPONENT_Y) ? 0 : (compID==COMPONENT_Cb ? m_chromaCbQpOffset : m_chromaCrQpOffset ); } Void setMaxCuChromaQpAdjDepth ( UInt u ) { m_MaxCuChromaQpAdjDepth = u; } UInt getMaxCuChromaQpAdjDepth () { return m_MaxCuChromaQpAdjDepth; } Void setMinCuChromaQpAdjSize ( UInt u ) { m_MinCuChromaQpAdjSize = u; } UInt getMinCuChromaQpAdjSize () { return m_MinCuChromaQpAdjSize; } Void clearChromaQpAdjTable() { m_ChromaQpAdjTableSize = 0; } Int getChromaQpAdjTableSize() { return m_ChromaQpAdjTableSize; } const ChromaQpAdj& getChromaQpAdjTableAt( Int idx ) const { return m_ChromaQpAdjTable[idx]; } Void setChromaQpAdjTableAt( Int idx, Int cbOffset, Int crOffset ) { m_ChromaQpAdjTable[idx].u.comp.CbOffset = cbOffset; m_ChromaQpAdjTable[idx].u.comp.CrOffset = crOffset; m_ChromaQpAdjTableSize = max(m_ChromaQpAdjTableSize, idx); } Void setNumRefIdxL0DefaultActive(UInt ui) { m_numRefIdxL0DefaultActive=ui; } UInt getNumRefIdxL0DefaultActive() { return m_numRefIdxL0DefaultActive; } Void setNumRefIdxL1DefaultActive(UInt ui) { m_numRefIdxL1DefaultActive=ui; } UInt getNumRefIdxL1DefaultActive() { return m_numRefIdxL1DefaultActive; } Bool getUseWP () { return m_bUseWeightPred; } Bool getWPBiPred () { return m_useWeightedBiPred; } Void setUseWP ( Bool b ) { m_bUseWeightPred = b; } Void setWPBiPred ( Bool b ) { m_useWeightedBiPred = b; } Bool getUseCrossComponentPrediction() const { return m_useCrossComponentPrediction; } Void setUseCrossComponentPrediction(Bool value) { m_useCrossComponentPrediction = value; } UInt getSaoOffsetBitShift(ChannelType type) const { return m_saoOffsetBitShift[type]; } Void setSaoOffsetBitShift(ChannelType type, UInt uiBitShift) { m_saoOffsetBitShift[type] = uiBitShift; } Void setOutputFlagPresentFlag( Bool b ) { m_OutputFlagPresentFlag = b; } Bool getOutputFlagPresentFlag() { return m_OutputFlagPresentFlag; } Void setTransquantBypassEnableFlag( Bool b ) { m_TransquantBypassEnableFlag = b; } Bool getTransquantBypassEnableFlag() { return m_TransquantBypassEnableFlag; } Bool getUseTransformSkip () { return m_useTransformSkip; } Void setUseTransformSkip ( Bool b ) { m_useTransformSkip = b; } UInt getTransformSkipLog2MaxSize () { return m_transformSkipLog2MaxSize; } Void setTransformSkipLog2MaxSize ( UInt u ) { m_transformSkipLog2MaxSize = u; } Void setLoopFilterAcrossTilesEnabledFlag (Bool b) { m_loopFilterAcrossTilesEnabledFlag = b; } Bool getLoopFilterAcrossTilesEnabledFlag () { return m_loopFilterAcrossTilesEnabledFlag; } Bool getDependentSliceSegmentsEnabledFlag() const { return m_dependentSliceSegmentsEnabledFlag; } Void setDependentSliceSegmentsEnabledFlag(Bool val) { m_dependentSliceSegmentsEnabledFlag = val; } Bool getEntropyCodingSyncEnabledFlag() const { return m_entropyCodingSyncEnabledFlag; } Void setEntropyCodingSyncEnabledFlag(Bool val) { m_entropyCodingSyncEnabledFlag = val; } Void setTilesEnabledFlag (Bool val) { m_tilesEnabledFlag = val; } Bool getTilesEnabledFlag () const { return m_tilesEnabledFlag; } Void setTileUniformSpacingFlag (Bool b) { m_uniformSpacingFlag = b; } Bool getTileUniformSpacingFlag () const { return m_uniformSpacingFlag; } Void setNumTileColumnsMinus1 (Int i) { m_numTileColumnsMinus1 = i; } Int getNumTileColumnsMinus1 () const { return m_numTileColumnsMinus1; } Void setTileColumnWidth (const std::vector& columnWidth ) { m_tileColumnWidth = columnWidth; } UInt getTileColumnWidth (UInt columnIdx) const { return m_tileColumnWidth[columnIdx]; } Void setNumTileRowsMinus1 (Int i) { m_numTileRowsMinus1 = i; } Int getNumTileRowsMinus1 () const { return m_numTileRowsMinus1; } Void setTileRowHeight (const std::vector& rowHeight) { m_tileRowHeight = rowHeight; } UInt getTileRowHeight (UInt rowIdx) const { return m_tileRowHeight[rowIdx]; } Void setNumSubstreams (Int numSubstreams) { m_numSubstreams = numSubstreams; } Int getNumSubstreams () { return m_numSubstreams; } Void setSignHideFlag( Int signHideFlag ) { m_signHideFlag = signHideFlag; } Int getSignHideFlag() { return m_signHideFlag; } Void setCabacInitPresentFlag( Bool flag ) { m_cabacInitPresentFlag = flag; } Void setEncCABACTableIdx( Int idx ) { m_encCABACTableIdx = idx; } Bool getCabacInitPresentFlag() { return m_cabacInitPresentFlag; } UInt getEncCABACTableIdx() { return m_encCABACTableIdx; } Void setDeblockingFilterControlPresentFlag( Bool val ) { m_deblockingFilterControlPresentFlag = val; } Bool getDeblockingFilterControlPresentFlag() { return m_deblockingFilterControlPresentFlag; } Void setDeblockingFilterOverrideEnabledFlag( Bool val ) { m_deblockingFilterOverrideEnabledFlag = val; } Bool getDeblockingFilterOverrideEnabledFlag() { return m_deblockingFilterOverrideEnabledFlag; } Void setPicDisableDeblockingFilterFlag(Bool val) { m_picDisableDeblockingFilterFlag = val; } //!< set offset for deblocking filter disabled Bool getPicDisableDeblockingFilterFlag() { return m_picDisableDeblockingFilterFlag; } //!< get offset for deblocking filter disabled Void setDeblockingFilterBetaOffsetDiv2(Int val) { m_deblockingFilterBetaOffsetDiv2 = val; } //!< set beta offset for deblocking filter Int getDeblockingFilterBetaOffsetDiv2() { return m_deblockingFilterBetaOffsetDiv2; } //!< get beta offset for deblocking filter Void setDeblockingFilterTcOffsetDiv2(Int val) { m_deblockingFilterTcOffsetDiv2 = val; } //!< set tc offset for deblocking filter Int getDeblockingFilterTcOffsetDiv2() { return m_deblockingFilterTcOffsetDiv2; } //!< get tc offset for deblocking filter Bool getScalingListPresentFlag() { return m_scalingListPresentFlag; } Void setScalingListPresentFlag( Bool b ) { m_scalingListPresentFlag = b; } Void setScalingList ( TComScalingList *scalingList); TComScalingList* getScalingList () { return m_scalingList; } //!< get ScalingList class pointer in PPS Bool getListsModificationPresentFlag () { return m_listsModificationPresentFlag; } Void setListsModificationPresentFlag ( Bool b ) { m_listsModificationPresentFlag = b; } UInt getLog2ParallelMergeLevelMinus2 () { return m_log2ParallelMergeLevelMinus2; } Void setLog2ParallelMergeLevelMinus2 (UInt mrgLevel) { m_log2ParallelMergeLevelMinus2 = mrgLevel; } Int getNumExtraSliceHeaderBits() { return m_numExtraSliceHeaderBits; } Void setNumExtraSliceHeaderBits(Int i) { m_numExtraSliceHeaderBits = i; } Void setLoopFilterAcrossSlicesEnabledFlag ( Bool bValue ) { m_loopFilterAcrossSlicesEnabledFlag = bValue; } Bool getLoopFilterAcrossSlicesEnabledFlag () { return m_loopFilterAcrossSlicesEnabledFlag; } Bool getSliceHeaderExtensionPresentFlag () { return m_sliceHeaderExtensionPresentFlag; } Void setSliceHeaderExtensionPresentFlag (Bool val) { m_sliceHeaderExtensionPresentFlag = val; } }; struct WPScalingParam { // Explicit weighted prediction parameters parsed in slice header, // or Implicit weighted prediction parameters (8 bits depth values). Bool bPresentFlag; UInt uiLog2WeightDenom; Int iWeight; Int iOffset; // Weighted prediction scaling values built from above parameters (bitdepth scaled): Int w; Int o; Int offset; Int shift; Int round; }; struct WPACDCParam { Int64 iAC; Int64 iDC; }; /// slice header class class TComSlice { private: // Bitstream writing Bool m_saoEnabledFlag[MAX_NUM_CHANNEL_TYPE]; Int m_iPPSId; ///< picture parameter set ID Bool m_PicOutputFlag; ///< pic_output_flag Int m_iPOC; Int m_iLastIDR; Int m_iAssociatedIRAP; NalUnitType m_iAssociatedIRAPType; static Int m_prevTid0POC; TComReferencePictureSet *m_pcRPS; TComReferencePictureSet m_LocalRPS; Int m_iBDidx; TComRefPicListModification m_RefPicListModification; NalUnitType m_eNalUnitType; ///< Nal unit type for the slice SliceType m_eSliceType; Int m_iSliceQp; Bool m_dependentSliceSegmentFlag; #if ADAPTIVE_QP_SELECTION Int m_iSliceQpBase; #endif Bool m_ChromaQpAdjEnabled; Bool m_deblockingFilterDisable; Bool m_deblockingFilterOverrideFlag; //< offsets for deblocking filter inherit from PPS Int m_deblockingFilterBetaOffsetDiv2; //< beta offset for deblocking filter Int m_deblockingFilterTcOffsetDiv2; //< tc offset for deblocking filter Int m_list1IdxToList0Idx[MAX_NUM_REF]; Int m_aiNumRefIdx [NUM_REF_PIC_LIST_01]; // for multiple reference of current slice Bool m_bCheckLDC; // Data Int m_iSliceQpDelta; Int m_iSliceChromaQpDelta[MAX_NUM_COMPONENT]; TComPic* m_apcRefPicList [NUM_REF_PIC_LIST_01][MAX_NUM_REF+1]; Int m_aiRefPOCList [NUM_REF_PIC_LIST_01][MAX_NUM_REF+1]; Bool m_bIsUsedAsLongTerm[NUM_REF_PIC_LIST_01][MAX_NUM_REF+1]; Int m_iDepth; // referenced slice? Bool m_bRefenced; // access channel TComVPS* m_pcVPS; TComSPS* m_pcSPS; TComPPS* m_pcPPS; TComPic* m_pcPic; #if ADAPTIVE_QP_SELECTION TComTrQuant* m_pcTrQuant; #endif UInt m_colFromL0Flag; // collocated picture from List0 flag Bool m_noOutputPriorPicsFlag; Bool m_noRaslOutputFlag; Bool m_handleCraAsBlaFlag; UInt m_colRefIdx; UInt m_maxNumMergeCand; Double m_lambdas[MAX_NUM_COMPONENT]; Bool m_abEqualRef [NUM_REF_PIC_LIST_01][MAX_NUM_REF][MAX_NUM_REF]; UInt m_uiTLayer; Bool m_bTLayerSwitchingFlag; SliceConstraint m_sliceMode; UInt m_sliceArgument; UInt m_sliceCurStartCtuTsAddr; UInt m_sliceCurEndCtuTsAddr; UInt m_sliceIdx; SliceConstraint m_sliceSegmentMode; UInt m_sliceSegmentArgument; UInt m_sliceSegmentCurStartCtuTsAddr; UInt m_sliceSegmentCurEndCtuTsAddr; Bool m_nextSlice; Bool m_nextSliceSegment; UInt m_sliceBits; UInt m_sliceSegmentBits; Bool m_bFinalized; WPScalingParam m_weightPredTable[NUM_REF_PIC_LIST_01][MAX_NUM_REF][MAX_NUM_COMPONENT]; // [REF_PIC_LIST_0 or REF_PIC_LIST_1][refIdx][0:Y, 1:U, 2:V] WPACDCParam m_weightACDCParam[MAX_NUM_COMPONENT]; std::vector m_substreamSizes; TComScalingList* m_scalingList; //!< pointer of quantization matrix Bool m_cabacInitFlag; Bool m_bLMvdL1Zero; Bool m_temporalLayerNonReferenceFlag; Bool m_LFCrossSliceBoundaryFlag; Bool m_enableTMVPFlag; public: TComSlice(); virtual ~TComSlice(); Void initSlice (); Void initTiles(); Void setVPS ( TComVPS* pcVPS ) { m_pcVPS = pcVPS; } TComVPS* getVPS () { return m_pcVPS; } Void setSPS ( TComSPS* pcSPS ) { m_pcSPS = pcSPS; } TComSPS* getSPS () { return m_pcSPS; } const TComSPS* getSPS() const { return m_pcSPS; } Void setPPS ( TComPPS* pcPPS ) { assert(pcPPS!=NULL); m_pcPPS = pcPPS; m_iPPSId = pcPPS->getPPSId(); } TComPPS* getPPS () { return m_pcPPS; } const TComPPS* getPPS() const { return m_pcPPS; } #if ADAPTIVE_QP_SELECTION Void setTrQuant ( TComTrQuant* pcTrQuant ) { m_pcTrQuant = pcTrQuant; } TComTrQuant* getTrQuant () { return m_pcTrQuant; } #endif Void setPPSId ( Int PPSId ) { m_iPPSId = PPSId; } Int getPPSId () { return m_iPPSId; } Void setPicOutputFlag( Bool b ) { m_PicOutputFlag = b; } Bool getPicOutputFlag() { return m_PicOutputFlag; } Void setSaoEnabledFlag(ChannelType chType, Bool s) {m_saoEnabledFlag[chType] =s; } Bool getSaoEnabledFlag(ChannelType chType) { return m_saoEnabledFlag[chType]; } Void setRPS ( TComReferencePictureSet *pcRPS ) { m_pcRPS = pcRPS; } TComReferencePictureSet* getRPS () { return m_pcRPS; } TComReferencePictureSet* getLocalRPS () { return &m_LocalRPS; } Void setRPSidx ( Int iBDidx ) { m_iBDidx = iBDidx; } Int getRPSidx () { return m_iBDidx; } Int getPrevTid0POC () { return m_prevTid0POC; } TComRefPicListModification* getRefPicListModification() { return &m_RefPicListModification; } Void setLastIDR(Int iIDRPOC) { m_iLastIDR = iIDRPOC; } Int getLastIDR() { return m_iLastIDR; } Void setAssociatedIRAPPOC(Int iAssociatedIRAPPOC) { m_iAssociatedIRAP = iAssociatedIRAPPOC; } Int getAssociatedIRAPPOC() { return m_iAssociatedIRAP; } Void setAssociatedIRAPType(NalUnitType associatedIRAPType) { m_iAssociatedIRAPType = associatedIRAPType; } NalUnitType getAssociatedIRAPType() { return m_iAssociatedIRAPType; } SliceType getSliceType () { return m_eSliceType; } Int getPOC () { return m_iPOC; } Int getSliceQp () { return m_iSliceQp; } Bool getDependentSliceSegmentFlag() const { return m_dependentSliceSegmentFlag; } Void setDependentSliceSegmentFlag(Bool val) { m_dependentSliceSegmentFlag = val; } #if ADAPTIVE_QP_SELECTION Int getSliceQpBase () const { return m_iSliceQpBase; } #endif Int getSliceQpDelta () const { return m_iSliceQpDelta; } Int getSliceChromaQpDelta (ComponentID compID) const { return isLuma(compID) ? 0 : m_iSliceChromaQpDelta[compID]; } Bool getUseChromaQpAdj() const { return m_ChromaQpAdjEnabled; } Bool getDeblockingFilterDisable() const { return m_deblockingFilterDisable; } Bool getDeblockingFilterOverrideFlag() const { return m_deblockingFilterOverrideFlag; } Int getDeblockingFilterBetaOffsetDiv2() const { return m_deblockingFilterBetaOffsetDiv2; } Int getDeblockingFilterTcOffsetDiv2() const { return m_deblockingFilterTcOffsetDiv2; } Int getNumRefIdx ( RefPicList e ) const { return m_aiNumRefIdx[e]; } TComPic* getPic () { return m_pcPic; } TComPic* getRefPic ( RefPicList e, Int iRefIdx) { return m_apcRefPicList[e][iRefIdx]; } Int getRefPOC ( RefPicList e, Int iRefIdx) { return m_aiRefPOCList[e][iRefIdx]; } Int getDepth () { return m_iDepth; } UInt getColFromL0Flag () { return m_colFromL0Flag; } UInt getColRefIdx () { return m_colRefIdx; } Void checkColRefIdx (UInt curSliceIdx, TComPic* pic); Bool getIsUsedAsLongTerm (Int i, Int j) { return m_bIsUsedAsLongTerm[i][j]; } Void setIsUsedAsLongTerm (Int i, Int j, Bool value) { m_bIsUsedAsLongTerm[i][j] = value; } Bool getCheckLDC () { return m_bCheckLDC; } Bool getMvdL1ZeroFlag () { return m_bLMvdL1Zero; } Int getNumRpsCurrTempList(); Int getList1IdxToList0Idx ( Int list1Idx ) { return m_list1IdxToList0Idx[list1Idx]; } Void setReferenced(Bool b) { m_bRefenced = b; } Bool isReferenced() { return m_bRefenced; } Bool isReferenceNalu() { return ((getNalUnitType() <= NAL_UNIT_RESERVED_VCL_R15) && (getNalUnitType()%2 != 0)) || ((getNalUnitType() >= NAL_UNIT_CODED_SLICE_BLA_W_LP) && (getNalUnitType() <= NAL_UNIT_RESERVED_IRAP_VCL23) ); } Void setPOC ( Int i ) { m_iPOC = i; if ((getTLayer()==0) && (isReferenceNalu() && (getNalUnitType()!=NAL_UNIT_CODED_SLICE_RASL_R)&& (getNalUnitType()!=NAL_UNIT_CODED_SLICE_RADL_R))) {m_prevTid0POC=i;} } Void setNalUnitType ( NalUnitType e ) { m_eNalUnitType = e; } NalUnitType getNalUnitType () const { return m_eNalUnitType; } Bool getRapPicFlag (); Bool getIdrPicFlag () { return getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_W_RADL || getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_N_LP; } Bool isIRAP () const { return (getNalUnitType() >= 16) && (getNalUnitType() <= 23); } Void checkCRA(TComReferencePictureSet *pReferencePictureSet, Int& pocCRA, NalUnitType& associatedIRAPType, TComList& rcListPic); Void decodingRefreshMarking(Int& pocCRA, Bool& bRefreshPending, TComList& rcListPic); Void setSliceType ( SliceType e ) { m_eSliceType = e; } Void setSliceQp ( Int i ) { m_iSliceQp = i; } #if ADAPTIVE_QP_SELECTION Void setSliceQpBase ( Int i ) { m_iSliceQpBase = i; } #endif Void setSliceQpDelta ( Int i ) { m_iSliceQpDelta = i; } Void setSliceChromaQpDelta( ComponentID compID, Int i ) { m_iSliceChromaQpDelta[compID] = isLuma(compID) ? 0 : i; } Void setUseChromaQpAdj ( Bool b ) { m_ChromaQpAdjEnabled = b; } Void setDeblockingFilterDisable( Bool b ) { m_deblockingFilterDisable= b; } Void setDeblockingFilterOverrideFlag( Bool b ) { m_deblockingFilterOverrideFlag = b; } Void setDeblockingFilterBetaOffsetDiv2( Int i ) { m_deblockingFilterBetaOffsetDiv2 = i; } Void setDeblockingFilterTcOffsetDiv2( Int i ) { m_deblockingFilterTcOffsetDiv2 = i; } Void setRefPic ( TComPic* p, RefPicList e, Int iRefIdx ) { m_apcRefPicList[e][iRefIdx] = p; } Void setRefPOC ( Int i, RefPicList e, Int iRefIdx ) { m_aiRefPOCList[e][iRefIdx] = i; } Void setNumRefIdx ( RefPicList e, Int i ) { m_aiNumRefIdx[e] = i; } Void setPic ( TComPic* p ) { m_pcPic = p; } Void setDepth ( Int iDepth ) { m_iDepth = iDepth; } Void setRefPicList ( TComList& rcListPic, Bool checkNumPocTotalCurr = false ); Void setRefPOCList (); Void setColFromL0Flag ( UInt colFromL0 ) { m_colFromL0Flag = colFromL0; } Void setColRefIdx ( UInt refIdx) { m_colRefIdx = refIdx; } Void setCheckLDC ( Bool b ) { m_bCheckLDC = b; } Void setMvdL1ZeroFlag ( Bool b) { m_bLMvdL1Zero = b; } Bool isIntra () { return m_eSliceType == I_SLICE; } Bool isInterB () { return m_eSliceType == B_SLICE; } Bool isInterP () { return m_eSliceType == P_SLICE; } Void setLambdas ( const Double lambdas[MAX_NUM_COMPONENT] ) { for (Int component = 0; component < MAX_NUM_COMPONENT; component++) m_lambdas[component] = lambdas[component]; } const Double* getLambdas() const { return m_lambdas; } Void initEqualRef(); Bool isEqualRef ( RefPicList e, Int iRefIdx1, Int iRefIdx2 ) { assert(e& rcListPic ); Void setList1IdxToList0Idx(); UInt getTLayer () { return m_uiTLayer; } Void setTLayer ( UInt uiTLayer ) { m_uiTLayer = uiTLayer; } Void setTLayerInfo( UInt uiTLayer ); Void decodingMarking( TComList& rcListPic, Int iGOPSIze, Int& iMaxRefPicNum ); Void checkLeadingPictureRestrictions( TComList& rcListPic ); Void applyReferencePictureSet( TComList& rcListPic, TComReferencePictureSet *RPSList); Bool isTemporalLayerSwitchingPoint( TComList& rcListPic ); Bool isStepwiseTemporalLayerSwitchingPointCandidate( TComList& rcListPic ); #if ALLOW_RECOVERY_POINT_AS_RAP Int checkThatAllRefPicsAreAvailable( TComList& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool printErrors, Int pocRandomAccess = 0, Bool bUseRecoveryPoint = false); Void createExplicitReferencePictureSetFromReference( TComList& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool isRAP, Int pocRandomAccess = 0, Bool bUseRecoveryPoint = false); #else Int checkThatAllRefPicsAreAvailable( TComList& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool printErrors, Int pocRandomAccess = 0); Void createExplicitReferencePictureSetFromReference( TComList& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool isRAP); #endif Void setMaxNumMergeCand (UInt val ) { m_maxNumMergeCand = val; } UInt getMaxNumMergeCand () { return m_maxNumMergeCand; } Void setNoOutputPriorPicsFlag ( Bool val ) { m_noOutputPriorPicsFlag = val; } Bool getNoOutputPriorPicsFlag () { return m_noOutputPriorPicsFlag; } Void setNoRaslOutputFlag ( Bool val ) { m_noRaslOutputFlag = val; } Bool getNoRaslOutputFlag () { return m_noRaslOutputFlag; } Void setHandleCraAsBlaFlag ( Bool val ) { m_handleCraAsBlaFlag = val; } Bool getHandleCraAsBlaFlag () { return m_handleCraAsBlaFlag; } Void setSliceMode ( SliceConstraint mode ) { m_sliceMode = mode; } SliceConstraint getSliceMode () const { return m_sliceMode; } Void setSliceArgument ( UInt uiArgument ) { m_sliceArgument = uiArgument; } UInt getSliceArgument () { return m_sliceArgument; } Void setSliceCurStartCtuTsAddr ( UInt ctuTsAddr ) { m_sliceCurStartCtuTsAddr = ctuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) UInt getSliceCurStartCtuTsAddr () const { return m_sliceCurStartCtuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) Void setSliceCurEndCtuTsAddr ( UInt ctuTsAddr ) { m_sliceCurEndCtuTsAddr = ctuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) UInt getSliceCurEndCtuTsAddr () const { return m_sliceCurEndCtuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) Void setSliceIdx ( UInt i) { m_sliceIdx = i; } UInt getSliceIdx () { return m_sliceIdx; } Void copySliceInfo (TComSlice *pcSliceSrc); Void setSliceSegmentMode ( SliceConstraint mode ) { m_sliceSegmentMode = mode; } SliceConstraint getSliceSegmentMode () const { return m_sliceSegmentMode; } Void setSliceSegmentArgument ( UInt uiArgument ) { m_sliceSegmentArgument = uiArgument; } UInt getSliceSegmentArgument () { return m_sliceSegmentArgument; } Void setSliceSegmentCurStartCtuTsAddr ( UInt ctuTsAddr ) { m_sliceSegmentCurStartCtuTsAddr = ctuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) UInt getSliceSegmentCurStartCtuTsAddr () const { return m_sliceSegmentCurStartCtuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) Void setSliceSegmentCurEndCtuTsAddr ( UInt ctuTsAddr ) { m_sliceSegmentCurEndCtuTsAddr = ctuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) UInt getSliceSegmentCurEndCtuTsAddr () const { return m_sliceSegmentCurEndCtuTsAddr; } // CTU Tile-scan address (as opposed to raster-scan) Void setSliceBits ( UInt uiVal ) { m_sliceBits = uiVal; } UInt getSliceBits () { return m_sliceBits; } Void setSliceSegmentBits ( UInt uiVal ) { m_sliceSegmentBits = uiVal; } UInt getSliceSegmentBits () { return m_sliceSegmentBits; } Void setFinalized ( Bool uiVal ) { m_bFinalized = uiVal; } Bool getFinalized () { return m_bFinalized; } Void setWpScaling ( WPScalingParam wp[NUM_REF_PIC_LIST_01][MAX_NUM_REF][MAX_NUM_COMPONENT] ) { memcpy(m_weightPredTable, wp, sizeof(WPScalingParam)*NUM_REF_PIC_LIST_01*MAX_NUM_REF*MAX_NUM_COMPONENT); } Void getWpScaling ( RefPicList e, Int iRefIdx, WPScalingParam *&wp); Void resetWpScaling (); Void initWpScaling (); inline Bool applyWP () { return( (m_eSliceType==P_SLICE && m_pcPPS->getUseWP()) || (m_eSliceType==B_SLICE && m_pcPPS->getWPBiPred()) ); } Void setWpAcDcParam ( WPACDCParam wp[MAX_NUM_COMPONENT] ) { memcpy(m_weightACDCParam, wp, sizeof(WPACDCParam)*MAX_NUM_COMPONENT); } Void getWpAcDcParam ( WPACDCParam *&wp ); Void initWpAcDcParam (); Void clearSubstreamSizes ( ) { return m_substreamSizes.clear(); } UInt getNumberOfSubstreamSizes ( ) { return (UInt) m_substreamSizes.size(); } Void addSubstreamSize ( UInt size ) { m_substreamSizes.push_back(size); } UInt getSubstreamSize ( Int idx ) { assert(idx& rcListPic, Int poc); TComPic* xGetLongTermRefPic(TComList& rcListPic, Int poc, Bool pocHasMsb); };// END CLASS DEFINITION TComSlice template class ParameterSetMap { public: ParameterSetMap(Int maxId) :m_maxId (maxId) {} ~ParameterSetMap() { for (typename std::map::iterator i = m_paramsetMap.begin(); i!= m_paramsetMap.end(); i++) { delete (*i).second; } } Void storePS(Int psId, T *ps) { assert ( psId < m_maxId ); if ( m_paramsetMap.find(psId) != m_paramsetMap.end() ) { delete m_paramsetMap[psId]; } m_paramsetMap[psId] = ps; } Void mergePSList(ParameterSetMap &rPsList) { for (typename std::map::iterator i = rPsList.m_paramsetMap.begin(); i!= rPsList.m_paramsetMap.end(); i++) { storePS(i->first, i->second); } rPsList.m_paramsetMap.clear(); } T* getPS(Int psId) { return ( m_paramsetMap.find(psId) == m_paramsetMap.end() ) ? NULL : m_paramsetMap[psId]; } T* getFirstPS() { return (m_paramsetMap.begin() == m_paramsetMap.end() ) ? NULL : m_paramsetMap.begin()->second; } private: std::map m_paramsetMap; Int m_maxId; }; class ParameterSetManager { public: ParameterSetManager(); virtual ~ParameterSetManager(); //! store sequence parameter set and take ownership of it Void storeVPS(TComVPS *vps) { m_vpsMap.storePS( vps->getVPSId(), vps); }; //! get pointer to existing video parameter set TComVPS* getVPS(Int vpsId) { return m_vpsMap.getPS(vpsId); }; TComVPS* getFirstVPS() { return m_vpsMap.getFirstPS(); }; //! store sequence parameter set and take ownership of it Void storeSPS(TComSPS *sps) { m_spsMap.storePS( sps->getSPSId(), sps); }; //! get pointer to existing sequence parameter set TComSPS* getSPS(Int spsId) { return m_spsMap.getPS(spsId); }; TComSPS* getFirstSPS() { return m_spsMap.getFirstPS(); }; //! store picture parameter set and take ownership of it Void storePPS(TComPPS *pps) { m_ppsMap.storePS( pps->getPPSId(), pps); }; //! get pointer to existing picture parameter set TComPPS* getPPS(Int ppsId) { return m_ppsMap.getPS(ppsId); }; TComPPS* getFirstPPS() { return m_ppsMap.getFirstPS(); }; //! activate a SPS from a active parameter sets SEI message //! \returns true, if activation is successful Bool activateSPSWithSEI(Int SPSId); //! activate a PPS and depending on isIDR parameter also SPS and VPS //! \returns true, if activation is successful Bool activatePPS(Int ppsId, Bool isIRAP); TComVPS* getActiveVPS(){ return m_vpsMap.getPS(m_activeVPSId); }; TComSPS* getActiveSPS(){ return m_spsMap.getPS(m_activeSPSId); }; TComPPS* getActivePPS(){ return m_ppsMap.getPS(m_activePPSId); }; protected: ParameterSetMap m_vpsMap; ParameterSetMap m_spsMap; ParameterSetMap m_ppsMap; Int m_activeVPSId; Int m_activeSPSId; Int m_activePPSId; }; //! \} #endif // __TCOMSLICE__