Context navigation

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

Visit:

Last change on this file since 777 was 655, checked in by tech, 11 years ago
Merged 8.1-Cleanup@654
Property svn:eol-style set to `native`
File size: 12.1 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	*
[608]	6	* Copyright (c) 2010-2013, 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;
[608]	71	#if !H_3D
	72	virtual Void codeSPS ( TComSPS* pcSPS ) = 0;
[42]	73	#else
[608]	74	virtual Void codeSPS ( TComSPS* pcSPS, Int viewIndex, Bool depthFlag ) = 0;
[42]	75	#endif
[2]	76	virtual Void codePPS ( TComPPS* pcPPS ) = 0;
	77	virtual Void codeSliceHeader ( TComSlice* pcSlice ) = 0;
[56]	78
	79	virtual Void codeTilesWPPEntryPoint ( TComSlice* pSlice ) = 0;
[2]	80	virtual Void codeTerminatingBit ( UInt uilsLast ) = 0;
	81	virtual Void codeSliceFinish () = 0;
	82	virtual Void codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0;
[56]	83	virtual Void codeScalingList ( TComScalingList* scalingList ) = 0;
[2]	84
	85	public:
[608]	86	virtual Void codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[2]	87	virtual Void codeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	88	virtual Void codeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	89	virtual Void codeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[608]	90	#if H_3D_ARP
	91	virtual Void codeARPW ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[5]	92	#endif
[608]	93	#if H_3D_IC
	94	virtual Void codeICFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[443]	95	#endif
[655]	96	#if H_3D_INTER_SDC
[608]	97	virtual Void codeInterSDCFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	98	virtual Void codeInterSDCResidualData ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiSegment ) = 0;
	99	#endif
[2]	100	virtual Void codeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0;
	101
	102	virtual Void codePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth ) = 0;
	103	virtual Void codePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	104
[608]	105	virtual Void codeIPCMInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[56]	106
[2]	107	virtual Void codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx ) = 0;
	108	virtual Void codeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth ) = 0;
	109	virtual Void codeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[608]	110	virtual Void codeQtCbfZero ( TComDataCU* pcCU, TextType eType, UInt uiTrDepth ) = 0;
	111	virtual Void codeQtRootCbfZero ( TComDataCU* pcCU ) = 0;
	112	virtual Void codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool isMultiplePU ) = 0;
[2]	113
	114	virtual Void codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	115	virtual Void codeInterDir ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
	116	virtual Void codeRefFrmIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0;
	117	virtual Void codeMvd ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList ) = 0;
	118	virtual Void codeDeltaQP ( TComDataCU* pcCU, UInt uiAbsPartIdx ) = 0;
[56]	119	virtual Void codeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType ) = 0;
[608]	120	virtual Void codeTransformSkipFlags ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, TextType eTType ) = 0;
[443]	121	virtual Void codeSAOSign ( UInt code ) = 0;
	122	virtual Void codeSaoMaxUvlc ( UInt code, UInt maxSymbol ) = 0;
[608]	123	virtual Void codeSaoMerge ( UInt uiCode ) = 0;
[56]	124	virtual Void codeSaoTypeIdx ( UInt uiCode) = 0;
[608]	125	virtual Void codeSaoUflc ( UInt uiLength, UInt uiCode ) = 0;
[56]	126	virtual Void estBit (estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType) = 0;
[2]	127
[56]	128	virtual Void updateContextTables ( SliceType eSliceType, Int iQp, Bool bExecuteFinish ) = 0;
	129	virtual Void updateContextTables ( SliceType eSliceType, Int iQp ) = 0;
	130
	131	virtual Void codeDFFlag (UInt uiCode, const Char *pSymbolName) = 0;
	132	virtual Void codeDFSvlc (Int iCode, const Char *pSymbolName) = 0;
	133
[2]	134	virtual ~TEncEntropyIf() {}
[56]	135
[2]	136	};
	137
	138	/// entropy encoder class
	139	class TEncEntropy
	140	{
[56]	141	private:
	142	UInt m_uiBakAbsPartIdx;
	143	UInt m_uiBakChromaOffset;
	144	UInt m_bakAbsPartIdxCU;
	145
[2]	146	public:
	147	Void setEntropyCoder ( TEncEntropyIf* e, TComSlice* pcSlice );
	148	Void setBitstream ( TComBitIf* p ) { m_pcEntropyCoderIf->setBitstream(p); }
	149	Void resetBits () { m_pcEntropyCoderIf->resetBits(); }
	150	Void resetCoeffCost () { m_pcEntropyCoderIf->resetCoeffCost(); }
	151	UInt getNumberOfWrittenBits () { return m_pcEntropyCoderIf->getNumberOfWrittenBits(); }
	152	UInt getCoeffCost () { return m_pcEntropyCoderIf->getCoeffCost(); }
	153	Void resetEntropy () { m_pcEntropyCoderIf->resetEntropy(); }
[56]	154	Void determineCabacInitIdx () { m_pcEntropyCoderIf->determineCabacInitIdx(); }
[2]	155
	156	Void encodeSliceHeader ( TComSlice* pcSlice );
[56]	157	Void encodeTilesWPPEntryPoint( TComSlice* pSlice );
[2]	158	Void encodeTerminatingBit ( UInt uiIsLast );
	159	Void encodeSliceFinish ();
	160	TEncEntropyIf* m_pcEntropyCoderIf;
	161
	162	public:
[100]	163	Void encodeVPS ( TComVPS* pcVPS);
[2]	164	// SPS
[608]	165	#if H_3D
	166	Void encodeSPS ( TComSPS* pcSPS, Int viewIndex, Bool depthFlag );
[56]	167	#else
[2]	168	Void encodeSPS ( TComSPS* pcSPS );
[56]	169	#endif
[2]	170	Void encodePPS ( TComPPS* pcPPS );
	171	Void encodeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false );
[608]	172	Void encodeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[2]	173	Void encodeSkipFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	174	Void encodePUWise ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	175	Void encodeInterDirPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx );
	176	Void encodeRefFrmIdxPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList );
	177	Void encodeMvdPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList );
	178	Void encodeMVPIdxPU ( TComDataCU* pcSubCU, UInt uiAbsPartIdx, RefPicList eRefList );
[608]	179	Void encodeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx );
	180	Void encodeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	181	#if H_3D_ARP
	182	Void encodeARPW ( TComDataCU* pcCU, UInt uiAbspartIdx );
[443]	183	#endif
[608]	184	#if H_3D_IC
	185	Void encodeICFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	186	#endif
[655]	187	#if H_3D_INTER_SDC
[608]	188	Void encodeInterSDCFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	189	Void encodeInterSDCResidualData( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD );
	190	#endif
[2]	191	Void encodePredMode ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	192	Void encodePartSize ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, Bool bRD = false );
[56]	193	Void encodeIPCMInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[608]	194	Void encodePredInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	195	Void encodeIntraDirModeLuma ( TComDataCU* pcCU, UInt absPartIdx, Bool isMultiplePU = false );
[2]	196
	197	Void encodeIntraDirModeChroma( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
	198
	199	Void encodeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx );
	200	Void encodeQtCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth );
[608]	201	Void encodeQtCbfZero ( TComDataCU* pcCU, TextType eType, UInt uiTrDepth );
	202	Void encodeQtRootCbfZero ( TComDataCU* pcCU );
[2]	203	Void encodeQtRootCbf ( TComDataCU* pcCU, UInt uiAbsPartIdx );
	204	Void encodeQP ( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool bRD = false );
[56]	205	Void updateContextTables ( SliceType eSliceType, Int iQp, Bool bExecuteFinish ) { m_pcEntropyCoderIf->updateContextTables( eSliceType, iQp, bExecuteFinish ); }
	206	Void updateContextTables ( SliceType eSliceType, Int iQp ) { m_pcEntropyCoderIf->updateContextTables( eSliceType, iQp, true ); }
[608]	207
[56]	208	Void encodeScalingList ( TComScalingList* scalingList );
	209
[2]	210	private:
[608]	211	Void xEncodeTransform ( TComDataCU* pcCU,UInt offsetLumaOffset, UInt offsetChroma, UInt uiAbsPartIdx, UInt uiDepth, UInt width, UInt height, UInt uiTrIdx, Bool& bCodeDQP );
[2]	212	public:
[56]	213	Void encodeCoeff ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiWidth, UInt uiHeight, Bool& bCodeDQP );
[2]	214
[56]	215	Void encodeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoeff, UInt uiAbsPartIdx, UInt uiTrWidth, UInt uiTrHeight, UInt uiDepth, TextType eType );
[2]	216
[56]	217	Void estimateBit ( estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType);
[443]	218	Void encodeSaoOffset(SaoLcuParam* saoLcuParam, UInt compIdx);
	219	Void encodeSaoUnitInterleaving(Int compIdx, Bool saoFlag, Int rx, Int ry, SaoLcuParam* saoLcuParam, Int cuAddrInSlice, Int cuAddrUpInSlice, Int allowMergeLeft, Int allowMergeUp);
[56]	220	static Int countNonZeroCoeffs( TCoeff* pcCoef, UInt uiSize );
	221
[2]	222	};// END CLASS DEFINITION TEncEntropy
	223
[56]	224	//! \}
[2]	225
	226	#endif // __TENCENTROPY__
	227

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

Download in other formats: