mirror/libbpg
1
0
Fork 1
Code Issues Pull requests Releases Wiki Activity
libbpg/jctvc/TLibCommon/TComSlice.h
King_DuckZ b21307932d libbpg-0.9.3
2015-01-16 13:46:18 +01:00

1595 lines
82 KiB
C++
Raw Permalink Blame History

/* 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 <cstring>
#include <map>
#include <vector>
#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<REF_PIC_LIST_NUM_IDX); m_RefPicSetIdxL0[idx] = refPicSetIdx; }
UInt getRefPicSetIdxL0(UInt idx) { assert(idx<REF_PIC_LIST_NUM_IDX); return m_RefPicSetIdxL0[idx]; }
Void setRefPicSetIdxL1(UInt idx, UInt refPicSetIdx) { assert(idx<REF_PIC_LIST_NUM_IDX); m_RefPicSetIdxL1[idx] = refPicSetIdx; }
UInt getRefPicSetIdxL1(UInt idx) { assert(idx<REF_PIC_LIST_NUM_IDX); return m_RefPicSetIdxL1[idx]; }
};
/// PPS class
class TComPPS
{
private:
Int m_PPSId; // pic_parameter_set_id
Int m_SPSId; // seq_parameter_set_id
Int m_picInitQPMinus26;
Bool m_useDQP;
Bool m_bConstrainedIntraPred; // constrained_intra_pred_flag
Bool m_bSliceChromaQpFlag; // slicelevel_chroma_qp_flag
// access channel
TComSPS* m_pcSPS;
UInt m_uiMaxCuDQPDepth;
UInt m_uiMinCuDQPSize;
/* Chroma QP Adjustments*/
Int m_MaxCuChromaQpAdjDepth;
Int m_MinCuChromaQpAdjSize;
Int m_ChromaQpAdjTableSize;
ChromaQpAdj m_ChromaQpAdjTable[7];
Int m_chromaCbQpOffset;
Int m_chromaCrQpOffset;
UInt m_numRefIdxL0DefaultActive;
UInt m_numRefIdxL1DefaultActive;
Bool m_bUseWeightPred; // Use of Weighting Prediction (P_SLICE)
Bool m_useWeightedBiPred; // Use of Weighting Bi-Prediction (B_SLICE)
Bool m_OutputFlagPresentFlag; // Indicates the presence of output_flag in slice header
Bool m_useCrossComponentPrediction;
UInt m_saoOffsetBitShift[MAX_NUM_CHANNEL_TYPE];
Bool m_TransquantBypassEnableFlag; // Indicates presence of cu_transquant_bypass_flag in CUs.
Bool m_useTransformSkip;
Int m_transformSkipLog2MaxSize;
Bool m_dependentSliceSegmentsEnabledFlag; //!< Indicates the presence of dependent slices
Bool m_tilesEnabledFlag; //!< Indicates the presence of tiles
Bool m_entropyCodingSyncEnabledFlag; //!< Indicates the presence of wavefronts
Bool m_loopFilterAcrossTilesEnabledFlag;
Bool m_uniformSpacingFlag;
Int m_numTileColumnsMinus1;
Int m_numTileRowsMinus1;
std::vector<Int> m_tileColumnWidth;
std::vector<Int> 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<Int>& 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<Int>& 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<UInt> 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<TComPic *>& rcListPic);
Void decodingRefreshMarking(Int& pocCRA, Bool& bRefreshPending, TComList<TComPic*>& 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<TComPic*>& 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<NUM_REF_PIC_LIST_01);
if (iRefIdx1 < 0 || iRefIdx2 < 0) return false;
return m_abEqualRef[e][iRefIdx1][iRefIdx2];
}
Void setEqualRef( RefPicList e, Int iRefIdx1, Int iRefIdx2, Bool b)
{
assert(e<NUM_REF_PIC_LIST_01);
m_abEqualRef[e][iRefIdx1][iRefIdx2] = m_abEqualRef[e][iRefIdx2][iRefIdx1] = b;
}
static Void sortPicList ( TComList<TComPic*>& rcListPic );
Void setList1IdxToList0Idx();
UInt getTLayer () { return m_uiTLayer; }
Void setTLayer ( UInt uiTLayer ) { m_uiTLayer = uiTLayer; }
Void setTLayerInfo( UInt uiTLayer );
Void decodingMarking( TComList<TComPic*>& rcListPic, Int iGOPSIze, Int& iMaxRefPicNum );
Void checkLeadingPictureRestrictions( TComList<TComPic*>& rcListPic );
Void applyReferencePictureSet( TComList<TComPic*>& rcListPic, TComReferencePictureSet *RPSList);
Bool isTemporalLayerSwitchingPoint( TComList<TComPic*>& rcListPic );
Bool isStepwiseTemporalLayerSwitchingPointCandidate( TComList<TComPic*>& rcListPic );
#if ALLOW_RECOVERY_POINT_AS_RAP
Int checkThatAllRefPicsAreAvailable( TComList<TComPic*>& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool printErrors, Int pocRandomAccess = 0, Bool bUseRecoveryPoint = false);
Void createExplicitReferencePictureSetFromReference( TComList<TComPic*>& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool isRAP, Int pocRandomAccess = 0, Bool bUseRecoveryPoint = false);
#else
Int checkThatAllRefPicsAreAvailable( TComList<TComPic*>& rcListPic, TComReferencePictureSet *pReferencePictureSet, Bool printErrors, Int pocRandomAccess = 0);
Void createExplicitReferencePictureSetFromReference( TComList<TComPic*>& 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<getNumberOfSubstreamSizes()); return m_substreamSizes[idx]; }
Void setScalingList ( TComScalingList* scalingList ) { m_scalingList = scalingList; }
TComScalingList* getScalingList () { return m_scalingList; }
Void setDefaultScalingList ();
Bool checkDefaultScalingList ();
Void setCabacInitFlag ( Bool val ) { m_cabacInitFlag = val; } //!< set CABAC initial flag
Bool getCabacInitFlag () { return m_cabacInitFlag; } //!< get CABAC initial flag
Bool getTemporalLayerNonReferenceFlag() { return m_temporalLayerNonReferenceFlag;}
Void setTemporalLayerNonReferenceFlag(Bool x) { m_temporalLayerNonReferenceFlag = x;}
Void setLFCrossSliceBoundaryFlag ( Bool val ) { m_LFCrossSliceBoundaryFlag = val; }
Bool getLFCrossSliceBoundaryFlag () { return m_LFCrossSliceBoundaryFlag;}
Void setEnableTMVPFlag ( Bool b ) { m_enableTMVPFlag = b; }
Bool getEnableTMVPFlag () { return m_enableTMVPFlag;}
protected:
TComPic* xGetRefPic (TComList<TComPic*>& rcListPic, Int poc);
TComPic* xGetLongTermRefPic(TComList<TComPic*>& rcListPic, Int poc, Bool pocHasMsb);
};// END CLASS DEFINITION TComSlice
template <class T> class ParameterSetMap
{
public:
ParameterSetMap(Int maxId)
:m_maxId (maxId)
{}
~ParameterSetMap()
{
for (typename std::map<Int,T *>::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<T> &rPsList)
{
for (typename std::map<Int,T *>::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<Int,T *> 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<TComVPS> m_vpsMap;
ParameterSetMap<TComSPS> m_spsMap;
ParameterSetMap<TComPPS> m_ppsMap;
Int m_activeVPSId;
Int m_activeSPSId;
Int m_activePPSId;
};
//! \}
#endif // __TCOMSLICE__
Reference in a new issue View git blame Copy permalink