/* 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-2011, 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 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 TEncEntropy.h \brief entropy encoder class (header) */ #ifndef __TENCENTROPY__ #define __TENCENTROPY__ #include "../TLibCommon/TComSlice.h" #include "../TLibCommon/TComDataCU.h" #include "../TLibCommon/TComBitStream.h" #include "../TLibCommon/ContextModel.h" #include "../TLibCommon/TComPic.h" #include "../TLibCommon/TComTrQuant.h" class TEncSbac; class TEncCavlc; class SEI; // ==================================================================================================================== // Class definition // ==================================================================================================================== /// entropy encoder pure class class TEncEntropyIf { public: virtual Bool getAlfCtrl() = 0; virtual UInt getMaxAlfCtrlDepth() = 0; virtual Void setAlfCtrl(Bool bAlfCtrl) = 0; virtual Void setMaxAlfCtrlDepth(UInt uiMaxAlfCtrlDepth) = 0; virtual Void resetEntropy () = 0; virtual Void setBitstream ( TComBitIf* p ) = 0; virtual Void setSlice ( TComSlice* p ) = 0; virtual Void resetBits () = 0; virtual Void resetCoeffCost () = 0; virtual UInt getNumberOfWrittenBits() = 0; virtual UInt getCoeffCost () = 0; #if BITSTREAM_EXTRACTION virtual Void codeNALUnitHeader ( NalUnitType eNalUnitType, NalRefIdc eNalRefIdc, UInt TemporalId = 0, UInt uiLayerId = 0 ) = 0; #else virtual Void codeNALUnitHeader ( NalUnitType eNalUnitType, NalRefIdc eNalRefIdc, UInt TemporalId = 0, Bool bOutputFlag = true ) = 0; #endif virtual Void codeSPS ( TComSPS* pcSPS ) = 0; virtual Void codePPS ( TComPPS* pcPPS ) = 0; virtual void codeSEI(const SEI&) = 0; virtual Void codeSliceHeader ( TComSlice* pcSlice ) = 0; virtual Void codeTerminatingBit ( UInt uilsLast ) = 0; virtual Void codeSliceFinish () = 0; virtual Void codeAlfCtrlDepth() = 0; virtual Void codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0; virtual Void codeViewIdx( Int iViewIdx ) = 0; public: virtual Void codeAlfCtrlFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; #if HHI_INTER_VIEW_RESIDUAL_PRED virtual Void codeResPredFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; #endif virtual Void codeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0; virtual Void codePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0; virtual Void codePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx ) = 0; virtual Void codeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ) = 0; virtual Void codeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeInterDir ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeRefFrmIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0; virtual Void codeMvd ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0; virtual Void codeDeltaQP ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0; virtual Void codeCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ) = 0; virtual Void codeBlockCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth, UInt uiQPartNum, Bool bRD = false) = 0; #if CAVLC_RQT_CBP virtual Void codeCbfTrdiv ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0; virtual UInt xGetFlagPattern ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0; #endif virtual Void codeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType, Bool bRD = false ) = 0; virtual Void codeAlfFlag ( UInt uiCode ) = 0; virtual Void codeAlfUvlc ( UInt uiCode ) = 0; virtual Void codeAlfSvlc ( Int iCode ) = 0; #if TSB_ALF_HEADER virtual Void codeAlfFlagNum ( UInt uiCode, UInt minValue ) = 0; virtual Void codeAlfCtrlFlag ( UInt uiSymbol ) = 0; #endif #if MTK_SAO virtual Void codeAoFlag ( UInt uiCode ) = 0; virtual Void codeAoUvlc ( UInt uiCode ) = 0; virtual Void codeAoSvlc ( Int iCode ) = 0; #endif virtual Void estBit (estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType) = 0; virtual ~TEncEntropyIf() {} }; /// entropy encoder class class TEncEntropy { public: Void setEntropyCoder ( TEncEntropyIf* e, TComSlice* pcSlice ); Void setBitstream ( TComBitIf* p ) { m_pcEntropyCoderIf->setBitstream(p); } Void resetBits () { m_pcEntropyCoderIf->resetBits(); } Void resetCoeffCost () { m_pcEntropyCoderIf->resetCoeffCost(); } UInt getNumberOfWrittenBits () { return m_pcEntropyCoderIf->getNumberOfWrittenBits(); } UInt getCoeffCost () { return m_pcEntropyCoderIf->getCoeffCost(); } Void resetEntropy () { m_pcEntropyCoderIf->resetEntropy(); } Void encodeSliceHeader ( TComSlice* pcSlice ); Void encodeTerminatingBit ( UInt uiIsLast ); Void encodeSliceFinish (); Void encodeAlfParam(ALFParam* pAlfParam); Void encodeMVPIdx( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList, Bool bRD = false ); Void encodeViewIdx(Int iViewIdx ); TEncEntropyIf* m_pcEntropyCoderIf; public: // SPS Void encodeSPS ( TComSPS* pcSPS ); Void encodePPS ( TComPPS* pcPPS ); void encodeSEI(const SEI&); Bool getAlfCtrl() {return m_pcEntropyCoderIf->getAlfCtrl();} UInt getMaxAlfCtrlDepth() {return m_pcEntropyCoderIf->getMaxAlfCtrlDepth();} Void setAlfCtrl(Bool bAlfCtrl) {m_pcEntropyCoderIf->setAlfCtrl(bAlfCtrl);} Void setMaxAlfCtrlDepth(UInt uiMaxAlfCtrlDepth) {m_pcEntropyCoderIf->setMaxAlfCtrlDepth(uiMaxAlfCtrlDepth);} Void encodeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false ); Void encodeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); Void encodePUWise ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); Void encodeInterDirPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx ); Void encodeRefFrmIdxPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList ); Void encodeMvdPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList ); Void encodeMVPIdxPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList ); Void encodeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiPUIdx ); #if HHI_MRG_SKIP Void encodeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiPUIdx, Bool bRD = false ); #else Void encodeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiPUIdx ); #endif #if HHI_INTER_VIEW_RESIDUAL_PRED Void encodeResPredFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiPUIdx, Bool bRD = false ); #endif Void encodeAlfCtrlFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); #if TSB_ALF_HEADER Void encodeAlfCtrlParam ( ALFParam *pAlfParam ); #endif Void encodePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); Void encodePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false ); Void encodePredInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); Void encodeIntraDirModeLuma ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void encodeIntraDirModeChroma( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); Void encodeInterDir ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); Void encodeRefFrmIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList, Bool bRD = false ); Void encodeMvd ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList, Bool bRD = false ); Void encodeTransformIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false ); Void encodeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx ); Void encodeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ); Void encodeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void encodeQP ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false ); private: Void xEncodeTransformSubdiv ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiInnerQuadIdx, UInt& uiYCbfFront3, UInt& uiUCbfFront3, UInt& uiVCbfFront3 ); Void xEncodeCoeff ( TComDataCU* pcCU, TCoeff* pcCoeff, UInt uiAbsPartIdx, UInt uiDepth, UInt uiWidth, UInt uiHeight, UInt uiTrIdx, UInt uiCurrTrIdx, TextType eType, Bool bRD = false ); public: Void encodeCoeff ( TComDataCU* pcCU , UInt uiAbsPartIdx, UInt uiDepth, UInt uiWidth, UInt uiHeight ); Void encodeCoeff ( TComDataCU* pcCU, TCoeff* pCoeff , UInt uiAbsPartIdx, UInt uiDepth, UInt uiWidth, UInt uiHeight, UInt uiMaxTrMode, UInt uiTrMode, TextType eType, Bool bRD = false ); Void encodeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoeff, UInt uiAbsPartIdx, UInt uiTrWidth, UInt uiTrHeight, UInt uiDepth, TextType eType, Bool bRD = false ); Void estimateBit ( estBitsSbacStruct* pcEstBitsSbac, UInt uiWidth, TextType eTType); // ALF-related Void codeAuxCountBit(ALFParam* pAlfParam, Int64* ruiRate); Void codeFiltCountBit(ALFParam* pAlfParam, Int64* ruiRate); Void codeAux (ALFParam* pAlfParam); Void codeFilt (ALFParam* pAlfParam); Int codeFilterCoeff(ALFParam* ALFp); Int writeFilterCodingParams(int minKStart, int maxScanVal, int kMinTab[]); Int writeFilterCoeffs(int sqrFiltLength, int filters_per_group, int pDepthInt[], int **FilterCoeff, int kMinTab[]); Int golombEncode(int coeff, int k); Int lengthGolomb(int coeffVal, int k); #if MTK_SAO Void encodeQAOOnePart(SAOParam* pQaoParam, Int part_idx); Void encodeQuadTreeSplitFlag(SAOParam* pQaoParam, Int part_idx); Void encodeSaoParam(SAOParam* pQaoParam) ; #endif };// END CLASS DEFINITION TEncEntropy #endif // __TENCENTROPY__