Context navigation

source: 3DVCSoftware/trunk/source/Lib/TLibCommon/TComTrQuant.h @ 399

Visit:

Last change on this file since 399 was 296, checked in by tech, 12 years ago
Reintegrated branch 5.1-dev0 rev. 295.
Property svn:eol-style set to `native`
File size: 14.4 KB

Line
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
4	* granted under this license.
5	*
6	* Copyright (c) 2010-2012, ITU/ISO/IEC
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.
17	* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
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	*/
33
34	/** \file TComTrQuant.h
35	\brief transform and quantization class (header)
36	*/
37
38	#ifndef __TCOMTRQUANT__
39	#define __TCOMTRQUANT__
40
41	#include "CommonDef.h"
42	#include "TComYuv.h"
43	#include "TComDataCU.h"
44	#include "ContextTables.h"
45
46	//! \ingroup TLibCommon
47	//! \{
48
49	// ====================================================================================================================
50	// Constants
51	// ====================================================================================================================
52
53	#define QP_BITS 15
54
55	// ====================================================================================================================
56	// Type definition
57	// ====================================================================================================================
58
59	typedef struct
60	{
61	Int significantCoeffGroupBits[NUM_SIG_CG_FLAG_CTX][2];
62	Int significantBits[NUM_SIG_FLAG_CTX][2];
63	Int lastXBits[32];
64	Int lastYBits[32];
65	Int m_greaterOneBits[NUM_ONE_FLAG_CTX][2];
66	Int m_levelAbsBits[NUM_ABS_FLAG_CTX][2];
67
68	Int blockCbpBits[3*NUM_QT_CBF_CTX][2];
69	Int blockRootCbpBits[4][2];
70	Int scanZigzag[2]; ///< flag for zigzag scan
71	Int scanNonZigzag[2]; ///< flag for non zigzag scan
72	} estBitsSbacStruct;
73
74	// ====================================================================================================================
75	// Class definition
76	// ====================================================================================================================
77
78	/// QP class
79	class QpParam
80	{
81	public:
82	QpParam();
83
84	Int m_iQP;
85	Int m_iPer;
86	Int m_iRem;
87
88	public:
89	Int m_iBits;
90
91	Void setQpParam( Int qpScaled, Bool bLowpass, SliceType eSliceType )
92	{
93	m_iQP = qpScaled;
94	m_iPer = qpScaled / 6;
95	m_iRem = qpScaled % 6;
96	m_iBits = QP_BITS + m_iPer;
97	}
98
99	Void clear()
100	{
101	m_iQP = 0;
102	m_iPer = 0;
103	m_iRem = 0;
104	m_iBits = 0;
105	}
106
107
108	Int per() const { return m_iPer; }
109	Int rem() const { return m_iRem; }
110	Int bits() const { return m_iBits; }
111
112	Int qp() {return m_iQP;}
113	}; // END CLASS DEFINITION QpParam
114
115	/// transform and quantization class
116	class TComTrQuant
117	{
118	public:
119	TComTrQuant();
120	~TComTrQuant();
121
122	// initialize class
123	Void init ( UInt uiMaxWidth, UInt uiMaxHeight, UInt uiMaxTrSize, Int iSymbolMode = 0, UInt aTable4 = NULL, UInt aTable8 = NULL, UInt *aTableLastPosVlcIndex=NULL, Bool bUseRDOQ = false, Bool bEnc = false
124	#if ADAPTIVE_QP_SELECTION
125	, Bool bUseAdaptQpSelect = false
126	#endif
127	);
128
129	// transform & inverse transform functions
130	Void transformNxN( TComDataCU* pcCU,
131	Pel* pcResidual,
132	UInt uiStride,
133	TCoeff* rpcCoeff,
134	#if ADAPTIVE_QP_SELECTION
135	Int*& rpcArlCoeff,
136	#endif
137	UInt uiWidth,
138	UInt uiHeight,
139	UInt& uiAbsSum,
140	TextType eTType,
141	UInt uiAbsPartIdx );
142	#if LOSSLESS_CODING
143	Void invtransformNxN( TComDataCU* pcCU, TextType eText, UInt uiMode,Pel* rpcResidual, UInt uiStride, TCoeff* pcCoeff, UInt uiWidth, UInt uiHeight, Int scalingListType);
144	#else
145	Void invtransformNxN( TextType eText, UInt uiMode,Pel& rpcResidual, UInt uiStride, TCoeff pcCoeff, UInt uiWidth, UInt uiHeight, Int scalingListType);
146	#endif
147	Void invRecurTransformNxN ( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eTxt, Pel* rpcResidual, UInt uiAddr, UInt uiStride, UInt uiWidth, UInt uiHeight,
148	UInt uiMaxTrMode, UInt uiTrMode, TCoeff* rpcCoeff );
149
150	// Misc functions
151	Void setQPforQuant( Int qpy, Bool bLowpass, SliceType eSliceType, TextType eTxtType, Int qpBdOffset, Int chromaQPOffset);
152
153	#if RDOQ_CHROMA_LAMBDA
154	Void setLambda(Double dLambdaLuma, Double dLambdaChroma) { m_dLambdaLuma = dLambdaLuma; m_dLambdaChroma = dLambdaChroma; }
155	Void selectLambda(TextType eTType) { m_dLambda = (eTType == TEXT_LUMA) ? m_dLambdaLuma : m_dLambdaChroma; }
156	#else
157	Void setLambda(Double dLambda) { m_dLambda = dLambda;}
158	#endif
159	Void setRDOQOffset( UInt uiRDOQOffset ) { m_uiRDOQOffset = uiRDOQOffset; }
160
161	estBitsSbacStruct* m_pcEstBitsSbac;
162
163	static Int getSigCtxInc ( TCoeff* pcCoeff,
164	Int posX,
165	Int posY,
166	Int blockType,
167	Int width
168	,Int height
169	,TextType textureType
170	);
171	static UInt getSigCoeffGroupCtxInc ( const UInt* uiSigCoeffGroupFlag,
172	const UInt uiCGPosX,
173	const UInt uiCGPosY,
174	const UInt scanIdx,
175	Int width, Int height);
176	Void initScalingList ();
177	Void destroyScalingList ();
178	Void setErrScaleCoeff ( UInt list, UInt size, UInt qp, UInt dir);
179	double* getErrScaleCoeff ( UInt list, UInt size, UInt qp, UInt dir) {return m_errScale[size][list][qp][dir];}; //!< get Error Scale Coefficent
180	Int* getQuantCoeff ( UInt list, UInt qp, UInt size, UInt dir) {return m_quantCoef[size][list][qp][dir];}; //!< get Quant Coefficent
181	Int* getDequantCoeff ( UInt list, UInt qp, UInt size, UInt dir) {return m_dequantCoef[size][list][qp][dir];}; //!< get DeQuant Coefficent
182	Void setUseScalingList ( Bool bUseScalingList){ m_scalingListEnabledFlag = bUseScalingList; };
183	Bool getUseScalingList (){ return m_scalingListEnabledFlag; };
184	Void setFlatScalingList ();
185	Void xsetFlatScalingList ( UInt list, UInt size, UInt qp);
186	Void xSetScalingListEnc ( TComScalingList *scalingList, UInt list, UInt size, UInt qp);
187	Void xSetScalingListDec ( TComScalingList *scalingList, UInt list, UInt size, UInt qp);
188	Void setScalingList ( TComScalingList *scalingList);
189	Void setScalingListDec ( TComScalingList *scalingList);
190	Void processScalingListEnc( Int coeff, Int quantcoeff, Int quantScales, UInt height, UInt width, UInt ratio, Int sizuNum, UInt dc);
191	Void processScalingListDec( Int coeff, Int dequantcoeff, Int invQuantScales, UInt height, UInt width, UInt ratio, Int sizuNum, UInt dc);
192	#if ADAPTIVE_QP_SELECTION
193	Void initSliceQpDelta() ;
194	Void storeSliceQpNext(TComSlice* pcSlice);
195	Void clearSliceARLCnt();
196	Int getQpDelta(Int qp) { return m_qpDelta[qp]; }
197	Int* getSliceNSamples(){ return m_sliceNsamples ;}
198	Double* getSliceSumC() { return m_sliceSumC; }
199	#endif
200	protected:
201	#if ADAPTIVE_QP_SELECTION
202	Int m_qpDelta[MAX_QP+1];
203	Int m_sliceNsamples[LEVEL_RANGE+1];
204	Double m_sliceSumC[LEVEL_RANGE+1] ;
205	#endif
206	Int* m_plTempCoeff;
207
208	QpParam m_cQP;
209	#if RDOQ_CHROMA_LAMBDA
210	Double m_dLambdaLuma;
211	Double m_dLambdaChroma;
212	#endif
213	Double m_dLambda;
214	UInt m_uiRDOQOffset;
215	UInt m_uiMaxTrSize;
216	Bool m_bEnc;
217	Bool m_bUseRDOQ;
218	#if ADAPTIVE_QP_SELECTION
219	Bool m_bUseAdaptQpSelect;
220	#endif
221
222	Bool m_scalingListEnabledFlag;
223	Int *m_quantCoef [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM][SCALING_LIST_REM_NUM][SCALING_LIST_DIR_NUM]; ///< array of quantization matrix coefficient 4x4
224	Int *m_dequantCoef [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM][SCALING_LIST_REM_NUM][SCALING_LIST_DIR_NUM]; ///< array of dequantization matrix coefficient 4x4
225	double *m_errScale [SCALING_LIST_SIZE_NUM][SCALING_LIST_NUM][SCALING_LIST_REM_NUM][SCALING_LIST_DIR_NUM]; ///< array of quantization matrix coefficient 4x4
226	private:
227	// forward Transform
228	Void xT ( UInt uiMode,Pel* pResidual, UInt uiStride, Int* plCoeff, Int iWidth, Int iHeight );
229
230	Void signBitHidingHDQ( TComDataCU* pcCU, TCoeff* pQCoef, TCoeff* pCoef, UInt const scan, Int deltaU, Int width, Int height );
231
232	// quantization
233	Void xQuant( TComDataCU* pcCU,
234	Int* pSrc,
235	TCoeff* pDes,
236	#if ADAPTIVE_QP_SELECTION
237	Int*& pArlDes,
238	#endif
239	Int iWidth,
240	Int iHeight,
241	UInt& uiAcSum,
242	TextType eTType,
243	UInt uiAbsPartIdx );
244
245	// RDOQ functions
246
247	Void xRateDistOptQuant ( TComDataCU* pcCU,
248	Int* plSrcCoeff,
249	TCoeff* piDstCoeff,
250	#if ADAPTIVE_QP_SELECTION
251	Int*& piArlDstCoeff,
252	#endif
253	UInt uiWidth,
254	UInt uiHeight,
255	UInt& uiAbsSum,
256	TextType eTType,
257	UInt uiAbsPartIdx );
258	__inline UInt xGetCodedLevel ( Double& rd64CodedCost,
259	Double& rd64CodedCost0,
260	Double& rd64CodedCostSig,
261	Int lLevelDouble,
262	UInt uiMaxAbsLevel,
263	UShort ui16CtxNumSig,
264	UShort ui16CtxNumOne,
265	UShort ui16CtxNumAbs,
266	UShort ui16AbsGoRice,
267	UInt c1Idx,
268	UInt c2Idx,
269	Int iQBits,
270	Double dTemp,
271	Bool bLast ) const;
272	__inline Double xGetICRateCost ( UInt uiAbsLevel,
273	UShort ui16CtxNumOne,
274	UShort ui16CtxNumAbs,
275	UShort ui16AbsGoRice
276	, UInt c1Idx,
277	UInt c2Idx
278	) const;
279	__inline Int xGetICRate ( UInt uiAbsLevel,
280	UShort ui16CtxNumOne,
281	UShort ui16CtxNumAbs,
282	UShort ui16AbsGoRice
283	, UInt c1Idx,
284	UInt c2Idx
285	) const;
286	__inline Double xGetRateLast ( const UInt uiPosX,
287	const UInt uiPosY,
288	const UInt uiBlkWdth ) const;
289	__inline Double xGetRateSigCoeffGroup ( UShort uiSignificanceCoeffGroup,
290	UShort ui16CtxNumSig ) const;
291	__inline Double xGetRateSigCoef ( UShort uiSignificance,
292	UShort ui16CtxNumSig ) const;
293	__inline Double xGetICost ( Double dRate ) const;
294	__inline Double xGetIEPRate ( ) const;
295
296
297	// dequantization
298	Void xDeQuant( const TCoeff* pSrc, Int* pDes, Int iWidth, Int iHeight, Int scalingListType );
299
300	// inverse transform
301	Void xIT ( UInt uiMode, Int* plCoef, Pel* pResidual, UInt uiStride, Int iWidth, Int iHeight );
302
303	};// END CLASS DEFINITION TComTrQuant
304
305	//! \}
306
307	#endif // __TCOMTRQUANT__

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

Download in other formats: