Context navigation

source: 3DVCSoftware/trunk/source/Lib/TLibEncoder/TEncEntropy.h @ 1271

Visit:

Last change on this file since 1271 was 1196, checked in by tech, 10 years ago
Merged 14.0-dev0@1187.
Property svn:eol-style set to `native`
File size: 12.6 KB

Rev	Line
[5]	1	/* The copyright in this software is being made available under the BSD
	2	* License, included below. This software may be subject to other third party
	3	* and contributor rights, including patent rights, and no such rights are
[56]	4	* granted under this license.
[5]	5	*
[1179]	6	* Copyright (c) 2010-2015, ITU/ISO/IEC
[5]	7	* All rights reserved.
	8	*
	9	* Redistribution and use in source and binary forms, with or without
	10	* modification, are permitted provided that the following conditions are met:
	11	*
	12	* * Redistributions of source code must retain the above copyright notice,
	13	* this list of conditions and the following disclaimer.
	14	* * Redistributions in binary form must reproduce the above copyright notice,
	15	* this list of conditions and the following disclaimer in the documentation
	16	* and/or other materials provided with the distribution.
[56]	17	* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
[5]	18	* be used to endorse or promote products derived from this software without
	19	* specific prior written permission.
	20	*
	21	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
	22	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	23	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	24	* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
	25	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	26	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	27	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	28	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	29	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	30	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
	31	* THE POSSIBILITY OF SUCH DAMAGE.
	32	*/
[2]	33
	34	/** \file TEncEntropy.h
	35	\brief entropy encoder class (header)
	36	*/
	37
	38	#ifndef __TENCENTROPY__
	39	#define __TENCENTROPY__
	40
[56]	41	#include "TLibCommon/TComSlice.h"
	42	#include "TLibCommon/TComDataCU.h"
	43	#include "TLibCommon/TComBitStream.h"
	44	#include "TLibCommon/ContextModel.h"
	45	#include "TLibCommon/TComPic.h"
	46	#include "TLibCommon/TComTrQuant.h"
	47	#include "TLibCommon/TComSampleAdaptiveOffset.h"
	48
[2]	49	class TEncSbac;
	50	class TEncCavlc;
	51	class SEI;
	52
	53	// ====================================================================================================================
	54	// Class definition
	55	// ====================================================================================================================
	56
	57	/// entropy encoder pure class
	58	class TEncEntropyIf
	59	{
	60	public:
	61	virtual Void resetEntropy () = 0;
[56]	62	virtual Void determineCabacInitIdx () = 0;
[2]	63	virtual Void setBitstream ( TComBitIf* p ) = 0;
	64	virtual Void setSlice ( TComSlice* p ) = 0;
	65	virtual Void resetBits () = 0;
	66	virtual Void resetCoeffCost () = 0;
	67	virtual UInt getNumberOfWrittenBits() = 0;
	68	virtual UInt getCoeffCost () = 0;
[56]	69
[100]	70	virtual Void codeVPS ( TComVPS* pcVPS ) = 0;
[1124]	71	virtual Void codeSPS ( TComSPS* pcSPS ) = 0;
[2]	72	virtual Void codePPS ( TComPPS* pcPPS ) = 0;
	73	virtual Void codeSliceHeader ( TComSlice* pcSlice ) = 0;
[56]	74
	75	virtual Void codeTilesWPPEntryPoint ( TComSlice* pSlice ) = 0;
[2]	76	virtual Void codeTerminatingBit ( UInt uilsLast ) = 0;
	77	virtual Void codeSliceFinish () = 0;
	78	virtual Void codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0;
[56]	79	virtual Void codeScalingList ( TComScalingList* scalingList ) = 0;
[2]	80
	81	public:
[608]	82	virtual Void codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[2]	83	virtual Void codeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[1196]	84	#if H_3D
[1179]	85	virtual Void codeDIS ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[1039]	86	#endif
[2]	87	virtual Void codeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	88	virtual Void codeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[608]	89	#if H_3D_ARP
	90	virtual Void codeARPW ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[5]	91	#endif
[608]	92	#if H_3D_IC
	93	virtual Void codeICFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[443]	94	#endif
[655]	95	#if H_3D_INTER_SDC
[833]	96	virtual Void codeDeltaDC ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	97	virtual Void codeSDCFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[608]	98	#endif
[833]	99	#if H_3D_DBBP
	100	virtual Void codeDBBPFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	101	#endif
[2]	102	virtual Void codeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0;
	103
	104	virtual Void codePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0;
	105	virtual Void codePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	106
[608]	107	virtual Void codeIPCMInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[56]	108
[2]	109	virtual Void codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx ) = 0;
	110	virtual Void codeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ) = 0;
	111	virtual Void codeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[608]	112	virtual Void codeQtCbfZero ( TComDataCU* pcCU, TextType eType, UInt uiTrDepth ) = 0;
	113	virtual Void codeQtRootCbfZero ( TComDataCU* pcCU ) = 0;
	114	virtual Void codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool isMultiplePU ) = 0;
[2]	115
	116	virtual Void codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	117	virtual Void codeInterDir ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	118	virtual Void codeRefFrmIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0;
	119	virtual Void codeMvd ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0;
	120	virtual Void codeDeltaQP ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[56]	121	virtual Void codeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType ) = 0;
[608]	122	virtual Void codeTransformSkipFlags ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, TextType eTType ) = 0;
[872]	123	virtual Void codeSAOBlkParam(SAOBlkParam& saoBlkParam, Bool* sliceEnabled, Bool leftMergeAvail, Bool aboveMergeAvail, Bool onlyEstMergeInfo = false) =0;
[56]	124	virtual Void estBit (estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType) = 0;
[2]	125
[56]	126	virtual Void updateContextTables ( SliceType eSliceType, Int iQp, Bool bExecuteFinish ) = 0;
	127	virtual Void updateContextTables ( SliceType eSliceType, Int iQp ) = 0;
	128
	129	virtual Void codeDFFlag (UInt uiCode, const Char *pSymbolName) = 0;
	130	virtual Void codeDFSvlc (Int iCode, const Char *pSymbolName) = 0;
	131
[2]	132	virtual ~TEncEntropyIf() {}
[56]	133
[2]	134	};
	135
	136	/// entropy encoder class
	137	class TEncEntropy
	138	{
[56]	139	private:
	140	UInt m_uiBakAbsPartIdx;
	141	UInt m_uiBakChromaOffset;
	142	UInt m_bakAbsPartIdxCU;
	143
[2]	144	public:
	145	Void setEntropyCoder ( TEncEntropyIf* e, TComSlice* pcSlice );
	146	Void setBitstream ( TComBitIf* p ) { m_pcEntropyCoderIf->setBitstream(p); }
	147	Void resetBits () { m_pcEntropyCoderIf->resetBits(); }
	148	Void resetCoeffCost () { m_pcEntropyCoderIf->resetCoeffCost(); }
[1179]	149	UInt getNumberOfWrittenBits ()
	150	{
	151	#if ENC_DEC_TRACE && H_MV_ENC_DEC_TRAC
	152	Bool oldJustDoIt = g_bJustDoIt;
	153	g_bJustDoIt = true;
	154	writeToTraceFile( "NumberOfWrittenBits", m_pcEntropyCoderIf->getNumberOfWrittenBits(), g_encNumberOfWrittenBits );
	155	g_bJustDoIt = oldJustDoIt;
	156	#endif
	157	return m_pcEntropyCoderIf->getNumberOfWrittenBits();
	158	}
[2]	159	UInt getCoeffCost () { return m_pcEntropyCoderIf->getCoeffCost(); }
	160	Void resetEntropy () { m_pcEntropyCoderIf->resetEntropy(); }
[56]	161	Void determineCabacInitIdx () { m_pcEntropyCoderIf->determineCabacInitIdx(); }
[2]	162
	163	Void encodeSliceHeader ( TComSlice* pcSlice );
[56]	164	Void encodeTilesWPPEntryPoint( TComSlice* pSlice );
[2]	165	Void encodeTerminatingBit ( UInt uiIsLast );
	166	Void encodeSliceFinish ();
	167	TEncEntropyIf* m_pcEntropyCoderIf;
	168
	169	public:
[100]	170	Void encodeVPS ( TComVPS* pcVPS);
[2]	171	// SPS
[1124]	172	Void encodeSPS ( TComSPS* pcSPS );
[2]	173	Void encodePPS ( TComPPS* pcPPS );
	174	Void encodeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false );
[608]	175	Void encodeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[2]	176	Void encodeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[1196]	177	#if H_3D
[1179]	178	Void encodeDIS ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD=false );
[1039]	179	#endif
[2]	180	Void encodePUWise ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	181	Void encodeInterDirPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx );
	182	Void encodeRefFrmIdxPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList );
	183	Void encodeMvdPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList );
	184	Void encodeMVPIdxPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList );
[608]	185	Void encodeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx );
	186	Void encodeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	187	#if H_3D_ARP
	188	Void encodeARPW ( TComDataCU* pcCU, UInt uiAbspartIdx );
[443]	189	#endif
[608]	190	#if H_3D_IC
	191	Void encodeICFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	192	#endif
[655]	193	#if H_3D_INTER_SDC
[833]	194	Void encodeDeltaDC ( TComDataCU* pcCU, UInt absPartIdx );
	195	Void encodeSDCFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[608]	196	#endif
[833]	197	#if H_3D_DBBP
	198	Void encodeDBBPFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	199	#endif
[2]	200	Void encodePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	201	Void encodePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false );
[56]	202	Void encodeIPCMInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[608]	203	Void encodePredInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	204	Void encodeIntraDirModeLuma ( TComDataCU* pcCU, UInt absPartIdx, Bool isMultiplePU = false );
[2]	205
	206	Void encodeIntraDirModeChroma( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	207
	208	Void encodeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx );
	209	Void encodeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth );
[608]	210	Void encodeQtCbfZero ( TComDataCU* pcCU, TextType eType, UInt uiTrDepth );
	211	Void encodeQtRootCbfZero ( TComDataCU* pcCU );
[2]	212	Void encodeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx );
	213	Void encodeQP ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[56]	214	Void updateContextTables ( SliceType eSliceType, Int iQp, Bool bExecuteFinish ) { m_pcEntropyCoderIf->updateContextTables( eSliceType, iQp, bExecuteFinish ); }
	215	Void updateContextTables ( SliceType eSliceType, Int iQp ) { m_pcEntropyCoderIf->updateContextTables( eSliceType, iQp, true ); }
[608]	216
[56]	217	Void encodeScalingList ( TComScalingList* scalingList );
	218
[2]	219	private:
[1179]	220	#if H_3D_DISABLE_CHROMA
	221	Void xEncodeTransform ( TComDataCU* pcCU,UInt offsetLumaOffset, UInt offsetChroma, UInt uiAbsPartIdx, UInt uiDepth, UInt width, UInt height, UInt uiTrIdx, Bool& bCodeDQP, Bool rd );
	222	public:
	223	Void encodeCoeff ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiWidth, UInt uiHeight, Bool& bCodeDQP, Bool rd );
	224	#else
[608]	225	Void xEncodeTransform ( TComDataCU* pcCU,UInt offsetLumaOffset, UInt offsetChroma, UInt uiAbsPartIdx, UInt uiDepth, UInt width, UInt height, UInt uiTrIdx, Bool& bCodeDQP );
[2]	226	public:
[56]	227	Void encodeCoeff ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiWidth, UInt uiHeight, Bool& bCodeDQP );
[1179]	228	#endif
[2]	229
[56]	230	Void encodeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoeff, UInt uiAbsPartIdx, UInt uiTrWidth, UInt uiTrHeight, UInt uiDepth, TextType eType );
[2]	231
[56]	232	Void estimateBit ( estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType);
[872]	233	Void encodeSAOBlkParam(SAOBlkParam& saoBlkParam, Bool* sliceEnabled, Bool leftMergeAvail, Bool aboveMergeAvail){m_pcEntropyCoderIf->codeSAOBlkParam(saoBlkParam, sliceEnabled, leftMergeAvail, aboveMergeAvail, false);}
[56]	234	static Int countNonZeroCoeffs( TCoeff* pcCoef, UInt uiSize );
	235
[2]	236	};// END CLASS DEFINITION TEncEntropy
	237
[56]	238	//! \}
[2]	239
	240	#endif // __TENCENTROPY__
	241

Note: See TracBrowser for help on using the repository browser.

Download in other formats: