forked from mirror/libbpg
1596 lines
82 KiB
C
1596 lines
82 KiB
C
|
/* 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__
|