/* 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 TEncSbac.h \brief Context-adaptive entropy encoder class (header) */ #ifndef __TENCSBAC__ #define __TENCSBAC__ #if _MSC_VER > 1000 #pragma once #endif // _MSC_VER > 1000 #include "../TLibCommon/TComBitStream.h" #include "../TLibCommon/ContextTables.h" #include "../TLibCommon/ContextModel.h" #include "../TLibCommon/ContextModel3DBuffer.h" #include "TEncEntropy.h" #include "TEncBinCoder.h" #include "TEncBinCoderCABAC.h" class TEncTop; // ==================================================================================================================== // Class definition // ==================================================================================================================== /// SBAC encoder class class TEncSbac : public TEncEntropyIf { public: TEncSbac(); virtual ~TEncSbac(); Void init ( TEncBinIf* p ) { m_pcBinIf = p; } Void uninit () { m_pcBinIf = 0; } // Virtual list Void resetEntropy (); Void setBitstream ( TComBitIf* p ) { m_pcBitIf = p; m_pcBinIf->init( p ); } Void setSlice ( TComSlice* p ) { m_pcSlice = p; } Bool getAlfCtrl () { return m_bAlfCtrl; } UInt getMaxAlfCtrlDepth () { return m_uiMaxAlfCtrlDepth; } Void setAlfCtrl ( Bool bAlfCtrl ) { m_bAlfCtrl = bAlfCtrl; } Void setMaxAlfCtrlDepth ( UInt uiMaxAlfCtrlDepth ) { m_uiMaxAlfCtrlDepth = uiMaxAlfCtrlDepth; } // SBAC RD Void resetCoeffCost () { m_uiCoeffCost = 0; } UInt getCoeffCost () { return m_uiCoeffCost; } Void load ( TEncSbac* pScr ); Void store ( TEncSbac* pDest ); Void resetBits () { m_pcBinIf->resetBits(); m_pcBitIf->resetBits(); } UInt getNumberOfWrittenBits () { return m_pcBinIf->getNumWrittenBits(); } //--SBAC RD #if BITSTREAM_EXTRACTION Void codeNALUnitHeader ( NalUnitType eNalUnitType, NalRefIdc eNalRefIdc, UInt TemporalId = 0, UInt uiLayerId = 0 ); #else Void codeNALUnitHeader ( NalUnitType eNalUnitType, NalRefIdc eNalRefIdc, UInt TemporalId = 0, Bool bOutputFlag = true ); #endif Void codeSPS ( TComSPS* pcSPS ); Void codePPS ( TComPPS* pcPPS ); void codeSEI(const SEI&); Void codeSliceHeader ( TComSlice* pcSlice ); Void codeTerminatingBit ( UInt uilsLast ); Void codeSliceFinish (); Void codeAlfFlag ( UInt uiCode ); Void codeAlfUvlc ( UInt uiCode ); Void codeAlfSvlc ( Int uiCode ); Void codeAlfCtrlDepth (); #if TSB_ALF_HEADER Void codeAlfFlagNum ( UInt uiCode, UInt minValue ); Void codeAlfCtrlFlag ( UInt uiSymbol ); #endif #if MTK_SAO Void codeAoFlag ( UInt uiCode ); Void codeAoUvlc ( UInt uiCode ); Void codeAoSvlc ( Int uiCode ); #endif private: Void xWriteUnarySymbol ( UInt uiSymbol, ContextModel* pcSCModel, Int iOffset ); Void xWriteUnaryMaxSymbol ( UInt uiSymbol, ContextModel* pcSCModel, Int iOffset, UInt uiMaxSymbol ); Void xWriteEpExGolomb ( UInt uiSymbol, UInt uiCount ); #if E253 Void xWriteGoRiceExGolomb ( UInt uiSymbol, UInt &ruiGoRiceParam ); #if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX Void xWriteExGolombLevel ( UInt uiSymbol, ContextModel& rcSCModel ); #endif #else Void xWriteExGolombLevel ( UInt uiSymbol, ContextModel& rcSCModel ); #endif Void xWriteTerminatingBit ( UInt uiBit ); Void xCheckCoeff( TCoeff* pcCoef, UInt uiSize, UInt uiDepth, UInt& uiNumofCoeff, UInt& uiPart ); #if MVD_CTX Void xWriteMvd ( Int iMvd, UInt uiAbsSumL, UInt uiAbsSumA, UInt uiCtx ); #else Void xWriteMvd ( Int iMvd, UInt uiAbsSum, UInt uiCtx ); #endif Void xWriteExGolombMvd ( UInt uiSymbol, ContextModel* pcSCModel, UInt uiMaxBin ); Void xCopyFrom ( TEncSbac* pSrc ); #if HHI_DMM_WEDGE_INTRA Void xCodeWedgeFullInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void xCodeWedgeFullDeltaInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void xCodeWedgePredDirInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void xCodeWedgePredDirDeltaInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ); #endif #if HHI_DMM_PRED_TEX Void xCodeWedgePredTexDeltaInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void xCodeContourPredTexDeltaInfo( TComDataCU* pcCU, UInt uiAbsPartIdx ); #endif protected: TComBitIf* m_pcBitIf; TComSlice* m_pcSlice; TEncBinIf* m_pcBinIf; Bool m_bAlfCtrl; //SBAC RD UInt m_uiCoeffCost; // Adaptive loop filter UInt m_uiMaxAlfCtrlDepth; //--Adaptive loop filter public: Void codeAlfCtrlFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx ); #if HHI_INTER_VIEW_MOTION_PRED || HHI_MPI Void codeMergeIndexMV ( TComDataCU* pcCU, UInt uiAbsPartIdx ); #endif #if HHI_INTER_VIEW_RESIDUAL_PRED Void codeResPredFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ); #endif Void codeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ); Void codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ); Void codePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ); Void codePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeTransformSubdivFlag ( UInt uiSymbol, UInt uiCtx ); Void codeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ); Void codeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeIntraDirLumaAng ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeIntraDirChroma ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeInterDir ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeRefFrmIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ); Void codeMvd ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ); Void codeViewIdx (Int iViewIdx); Void codeDeltaQP ( TComDataCU* pcCU, UInt uiAbsPartIdx ); Void codeCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ) {} Void codeBlockCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth, UInt uiQPartNum, Bool bRD = false ) {} #if CAVLC_RQT_CBP Void codeCbfTrdiv ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) {} UInt xGetFlagPattern ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) { return 0; } #endif #if PCP_SIGMAP_SIMPLE_LAST __inline Void codeLastSignificantXY ( UInt uiPosX, UInt uiPosY, const UInt uiWidth, const TextType eTType, const UInt uiCTXIdx, const UInt uiScanIdx ); #endif Void codeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType, Bool bRD = false ); // ------------------------------------------------------------------------------------------------------------------- // for RD-optimizatioon // ------------------------------------------------------------------------------------------------------------------- Void estBit ( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType ); Void estCBFBit ( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType ); Void estSignificantMapBit ( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType ); Void estSignificantCoefficientsBit ( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType ); __inline Int biari_no_bits ( Short symbol, ContextModel& rcSCModel ); TEncBinIf* getEncBinIf() { return m_pcBinIf; } private: UInt m_uiLastQp; ContextModel3DBuffer m_cCUSplitFlagSCModel; ContextModel3DBuffer m_cCUSkipFlagSCModel; ContextModel3DBuffer m_cCUMergeFlagExtSCModel; ContextModel3DBuffer m_cCUMergeIdxExtSCModel; ContextModel3DBuffer m_cCUMVMergeIdxExtSCModel; ContextModel3DBuffer m_cResPredFlagSCModel; ContextModel3DBuffer m_cCUPartSizeSCModel; ContextModel3DBuffer m_cCUPredModeSCModel; ContextModel3DBuffer m_cCUAlfCtrlFlagSCModel; ContextModel3DBuffer m_cCUIntraPredSCModel; #if ADD_PLANAR_MODE ContextModel3DBuffer m_cPlanarFlagSCModel; #endif ContextModel3DBuffer m_cCUChromaPredSCModel; ContextModel3DBuffer m_cCUDeltaQpSCModel; ContextModel3DBuffer m_cCUInterDirSCModel; ContextModel3DBuffer m_cCURefPicSCModel; ContextModel3DBuffer m_cCUMvdSCModel; ContextModel3DBuffer m_cCUQtCbfSCModel; ContextModel3DBuffer m_cCUTransSubdivFlagSCModel; ContextModel3DBuffer m_cCUQtRootCbfSCModel; ContextModel3DBuffer m_cCUSigSCModel; #if PCP_SIGMAP_SIMPLE_LAST ContextModel3DBuffer m_cCuCtxLastX; ContextModel3DBuffer m_cCuCtxLastY; #else ContextModel3DBuffer m_cCULastSCModel; #endif ContextModel3DBuffer m_cCUOneSCModel; ContextModel3DBuffer m_cCUAbsSCModel; ContextModel3DBuffer m_cMVPIdxSCModel; ContextModel3DBuffer m_cALFFlagSCModel; ContextModel3DBuffer m_cALFUvlcSCModel; ContextModel3DBuffer m_cALFSvlcSCModel; #if MTK_SAO ContextModel3DBuffer m_cAOFlagSCModel; ContextModel3DBuffer m_cAOUvlcSCModel; ContextModel3DBuffer m_cAOSvlcSCModel; #endif ContextModel3DBuffer m_cViewIdxSCModel; #if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX ContextModel3DBuffer m_cIntraDMMSCModel; ContextModel3DBuffer m_cIntraWedgeSCModel; #endif }; #endif // !defined(AFX_TENCSBAC_H__DDA7CDC4_EDE3_4015_9D32_2156249C82AA__INCLUDED_)