Context navigation

source: SHVCSoftware/trunk/source/Lib/TLibEncoder/TEncSearch.h @ 225

Visit:

Last change on this file since 225 was 191, checked in by seregin, 12 years ago
unix2dos for .cpp and .h
File size: 24.7 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-2013, 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 TEncSearch.h
35	\brief encoder search class (header)
36	*/
37
38	#ifndef __TENCSEARCH__
39	#define __TENCSEARCH__
40
41	// Include files
42	#include "TLibCommon/TComYuv.h"
43	#include "TLibCommon/TComMotionInfo.h"
44	#include "TLibCommon/TComPattern.h"
45	#include "TLibCommon/TComPrediction.h"
46	#include "TLibCommon/TComTrQuant.h"
47	#include "TLibCommon/TComPic.h"
48	#include "TEncEntropy.h"
49	#include "TEncSbac.h"
50	#include "TEncCfg.h"
51
52	//! \ingroup TLibEncoder
53	//! \{
54
55	class TEncCu;
56
57	// ====================================================================================================================
58	// Class definition
59	// ====================================================================================================================
60
61	/// encoder search class
62	class TEncSearch : public TComPrediction
63	{
64	private:
65	TCoeff** m_ppcQTTempCoeffY;
66	TCoeff** m_ppcQTTempCoeffCb;
67	TCoeff** m_ppcQTTempCoeffCr;
68	TCoeff* m_pcQTTempCoeffY;
69	TCoeff* m_pcQTTempCoeffCb;
70	TCoeff* m_pcQTTempCoeffCr;
71	#if ADAPTIVE_QP_SELECTION
72	Int** m_ppcQTTempArlCoeffY;
73	Int** m_ppcQTTempArlCoeffCb;
74	Int** m_ppcQTTempArlCoeffCr;
75	Int* m_pcQTTempArlCoeffY;
76	Int* m_pcQTTempArlCoeffCb;
77	Int* m_pcQTTempArlCoeffCr;
78	#endif
79	UChar* m_puhQTTempTrIdx;
80	UChar* m_puhQTTempCbf[3];
81
82	TComYuv* m_pcQTTempTComYuv;
83	TComYuv m_tmpYuvPred; // To be used in xGetInterPredictionError() to avoid constant memory allocation/deallocation
84	Pel* m_pSharedPredTransformSkip[3];
85	TCoeff* m_pcQTTempTUCoeffY;
86	TCoeff* m_pcQTTempTUCoeffCb;
87	TCoeff* m_pcQTTempTUCoeffCr;
88	UChar* m_puhQTTempTransformSkipFlag[3];
89	TComYuv m_pcQTTempTransformSkipTComYuv;
90	#if ADAPTIVE_QP_SELECTION
91	Int* m_ppcQTTempTUArlCoeffY;
92	Int* m_ppcQTTempTUArlCoeffCb;
93	Int* m_ppcQTTempTUArlCoeffCr;
94	#endif
95	protected:
96	// interface to option
97	TEncCfg* m_pcEncCfg;
98
99	#if SVC_EXTENSION
100	TEncTop** m_ppcTEncTop;
101	#if INTRA_BL
102	TComPicYuv* m_pcPicYuvRecBase; ///< reconstructed base layer
103	#endif
104	#endif
105
106	// interface to classes
107	TComTrQuant* m_pcTrQuant;
108	TComRdCost* m_pcRdCost;
109	TEncEntropy* m_pcEntropyCoder;
110
111	// ME parameters
112	Int m_iSearchRange;
113	Int m_bipredSearchRange; // Search range for bi-prediction
114	Int m_iFastSearch;
115	Int m_aaiAdaptSR[2][33];
116	TComMv m_cSrchRngLT;
117	TComMv m_cSrchRngRB;
118	TComMv m_acMvPredictors[3];
119
120	// RD computation
121	TEncSbac*** m_pppcRDSbacCoder;
122	TEncSbac* m_pcRDGoOnSbacCoder;
123	Bool m_bUseSBACRD;
124	DistParam m_cDistParam;
125
126	// Misc.
127	Pel* m_pTempPel;
128	const UInt* m_puiDFilter;
129	Int m_iMaxDeltaQP;
130
131	// AMVP cost computation
132	// UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+1][AMVP_MAX_NUM_CANDS];
133	UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+1][AMVP_MAX_NUM_CANDS+1]; //th array bounds
134
135	public:
136	TEncSearch();
137	virtual ~TEncSearch();
138
139	Void init( TEncCfg* pcEncCfg,
140	TComTrQuant* pcTrQuant,
141	Int iSearchRange,
142	Int bipredSearchRange,
143	Int iFastSearch,
144	Int iMaxDeltaQP,
145	TEncEntropy* pcEntropyCoder,
146	TComRdCost* pcRdCost,
147	TEncSbac*** pppcRDSbacCoder,
148	TEncSbac* pcRDGoOnSbacCoder );
149
150	protected:
151
152	/// sub-function for motion vector refinement used in fractional-pel accuracy
153	UInt xPatternRefinement( TComPattern* pcPatternKey,
154	TComMv baseRefMv,
155	Int iFrac, TComMv& rcMvFrac );
156
157	typedef struct
158	{
159	Pel* piRefY;
160	Int iYStride;
161	Int iBestX;
162	Int iBestY;
163	UInt uiBestRound;
164	UInt uiBestDistance;
165	UInt uiBestSad;
166	UChar ucPointNr;
167	} IntTZSearchStruct;
168
169	// sub-functions for ME
170	__inline Void xTZSearchHelp ( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, const Int iSearchX, const Int iSearchY, const UChar ucPointNr, const UInt uiDistance );
171	__inline Void xTZ2PointSearch ( TComPattern* pcPatternKey, IntTZSearchStruct& rcStrukt, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB );
172	__inline Void xTZ8PointSquareSearch ( TComPattern* pcPatternKey, IntTZSearchStruct& rcStrukt, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist );
173	__inline Void xTZ8PointDiamondSearch( TComPattern* pcPatternKey, IntTZSearchStruct& rcStrukt, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist );
174
175	Void xGetInterPredictionError( TComDataCU* pcCU, TComYuv* pcYuvOrg, Int iPartIdx, UInt& ruiSAD, Bool Hadamard );
176
177	public:
178	Void preestChromaPredMode ( TComDataCU* pcCU,
179	TComYuv* pcOrgYuv,
180	TComYuv* pcPredYuv );
181	Void estIntraPredQT ( TComDataCU* pcCU,
182	TComYuv* pcOrgYuv,
183	TComYuv* pcPredYuv,
184	TComYuv* pcResiYuv,
185	TComYuv* pcRecoYuv,
186	UInt& ruiDistC,
187	Bool bLumaOnly );
188	Void estIntraPredChromaQT ( TComDataCU* pcCU,
189	TComYuv* pcOrgYuv,
190	TComYuv* pcPredYuv,
191	TComYuv* pcResiYuv,
192	TComYuv* pcRecoYuv,
193	UInt uiPreCalcDistC );
194
195	#if INTRA_BL
196	Void setBaseRecPic ( TComPicYuv* pcPicYuvRecBase ) { m_pcPicYuvRecBase = pcPicYuvRecBase; }
197	TComPicYuv* getBaseRecPic () { return m_pcPicYuvRecBase; }
198	Void estIntraBLPredQT ( TComDataCU* pcCU,
199	TComYuv* pcOrgYuv,
200	TComYuv* pcPredYuv,
201	TComYuv* pcResiYuv,
202	TComYuv* pcRecoYuv );
203	#endif
204
205	/// encoder estimation - inter prediction (non-skip)
206	Void predInterSearch ( TComDataCU* pcCU,
207	TComYuv* pcOrgYuv,
208	TComYuv*& rpcPredYuv,
209	TComYuv*& rpcResiYuv,
210	TComYuv*& rpcRecoYuv,
211	Bool bUseRes = false
212	#if AMP_MRG
213	,Bool bUseMRG = false
214	#endif
215	);
216
217	#if (ENCODER_FAST_MODE)
218	Bool predInterSearchILRUni ( TComDataCU* pcCU,
219	TComYuv* pcOrgYuv,
220	TComYuv*& rpcPredYuv,
221	TComYuv*& rpcResiYuv,
222	TComYuv*& rpcRecoYuv
223	);
224
225	#endif
226
227	/// encode residual and compute rd-cost for inter mode
228	Void encodeResAndCalcRdInterCU( TComDataCU* pcCU,
229	TComYuv* pcYuvOrg,
230	TComYuv* pcYuvPred,
231	TComYuv*& rpcYuvResi,
232	TComYuv*& rpcYuvResiBest,
233	TComYuv*& rpcYuvRec,
234	Bool bSkipRes );
235
236	/// set ME search range
237	Void setAdaptiveSearchRange ( Int iDir, Int iRefIdx, Int iSearchRange) { m_aaiAdaptSR[iDir][iRefIdx] = iSearchRange; }
238
239	Void xEncPCM (TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piOrg, Pel* piPCM, Pel* piPred, Pel* piResi, Pel* piReco, UInt uiStride, UInt uiWidth, UInt uiHeight, TextType eText);
240	Void IPCMSearch (TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv& rpcPredYuv, TComYuv& rpcResiYuv, TComYuv*& rpcRecoYuv );
241	protected:
242
243	// -------------------------------------------------------------------------------------------------------------------
244	// Intra search
245	// -------------------------------------------------------------------------------------------------------------------
246
247	Void xEncSubdivCbfQT ( TComDataCU* pcCU,
248	UInt uiTrDepth,
249	UInt uiAbsPartIdx,
250	Bool bLuma,
251	Bool bChroma );
252
253	Void xEncCoeffQT ( TComDataCU* pcCU,
254	UInt uiTrDepth,
255	UInt uiAbsPartIdx,
256	TextType eTextType,
257	Bool bRealCoeff );
258	Void xEncIntraHeader ( TComDataCU* pcCU,
259	UInt uiTrDepth,
260	UInt uiAbsPartIdx,
261	Bool bLuma,
262	Bool bChroma );
263	UInt xGetIntraBitsQT ( TComDataCU* pcCU,
264	UInt uiTrDepth,
265	UInt uiAbsPartIdx,
266	Bool bLuma,
267	Bool bChroma,
268	Bool bRealCoeff );
269	UInt xGetIntraBitsQTChroma ( TComDataCU* pcCU,
270	UInt uiTrDepth,
271	UInt uiAbsPartIdx,
272	UInt uiChromaId,
273	Bool bRealCoeff );
274
275	Void xIntraCodingLumaBlk ( TComDataCU* pcCU,
276	UInt uiTrDepth,
277	UInt uiAbsPartIdx,
278	TComYuv* pcOrgYuv,
279	TComYuv* pcPredYuv,
280	TComYuv* pcResiYuv,
281	UInt& ruiDist,
282	Int default0Save1Load2 = 0);
283	Void xIntraCodingChromaBlk ( TComDataCU* pcCU,
284	UInt uiTrDepth,
285	UInt uiAbsPartIdx,
286	TComYuv* pcOrgYuv,
287	TComYuv* pcPredYuv,
288	TComYuv* pcResiYuv,
289	UInt& ruiDist,
290	UInt uiChromaId,
291	Int default0Save1Load2 = 0 );
292
293	Void xRecurIntraCodingQT ( TComDataCU* pcCU,
294	UInt uiTrDepth,
295	UInt uiAbsPartIdx,
296	Bool bLumaOnly,
297	TComYuv* pcOrgYuv,
298	TComYuv* pcPredYuv,
299	TComYuv* pcResiYuv,
300	UInt& ruiDistY,
301	UInt& ruiDistC,
302	#if HHI_RQT_INTRA_SPEEDUP
303	Bool bCheckFirst,
304	#endif
305	Double& dRDCost );
306
307	Void xSetIntraResultQT ( TComDataCU* pcCU,
308	UInt uiTrDepth,
309	UInt uiAbsPartIdx,
310	Bool bLumaOnly,
311	TComYuv* pcRecoYuv );
312
313	Void xRecurIntraChromaCodingQT ( TComDataCU* pcCU,
314	UInt uiTrDepth,
315	UInt uiAbsPartIdx,
316	TComYuv* pcOrgYuv,
317	TComYuv* pcPredYuv,
318	TComYuv* pcResiYuv,
319	UInt& ruiDist );
320	Void xSetIntraResultChromaQT ( TComDataCU* pcCU,
321	UInt uiTrDepth,
322	UInt uiAbsPartIdx,
323	TComYuv* pcRecoYuv );
324
325	Void xStoreIntraResultQT ( TComDataCU* pcCU,
326	UInt uiTrDepth,
327	UInt uiAbsPartIdx,
328	Bool bLumaOnly );
329	Void xLoadIntraResultQT ( TComDataCU* pcCU,
330	UInt uiTrDepth,
331	UInt uiAbsPartIdx,
332	Bool bLumaOnly );
333	Void xStoreIntraResultChromaQT ( TComDataCU* pcCU,
334	UInt uiTrDepth,
335	UInt uiAbsPartIdx,
336	UInt stateU0V1Both2 );
337	Void xLoadIntraResultChromaQT ( TComDataCU* pcCU,
338	UInt uiTrDepth,
339	UInt uiAbsPartIdx,
340	UInt stateU0V1Both2 );
341
342	// -------------------------------------------------------------------------------------------------------------------
343	// Inter search (AMP)
344	// -------------------------------------------------------------------------------------------------------------------
345
346	Void xEstimateMvPredAMVP ( TComDataCU* pcCU,
347	TComYuv* pcOrgYuv,
348	UInt uiPartIdx,
349	RefPicList eRefPicList,
350	Int iRefIdx,
351	TComMv& rcMvPred,
352	Bool bFilled = false
353	, UInt* puiDistBiP = NULL
354	#if ZERO_MVD_EST
355	, UInt* puiDist = NULL
356	#endif
357	);
358
359	Void xCheckBestMVP ( TComDataCU* pcCU,
360	RefPicList eRefPicList,
361	TComMv cMv,
362	TComMv& rcMvPred,
363	Int& riMVPIdx,
364	UInt& ruiBits,
365	UInt& ruiCost );
366
367	UInt xGetTemplateCost ( TComDataCU* pcCU,
368	UInt uiPartIdx,
369	UInt uiPartAddr,
370	TComYuv* pcOrgYuv,
371	TComYuv* pcTemplateCand,
372	TComMv cMvCand,
373	Int iMVPIdx,
374	Int iMVPNum,
375	RefPicList eRefPicList,
376	Int iRefIdx,
377	Int iSizeX,
378	Int iSizeY
379	#if ZERO_MVD_EST
380	, UInt& ruiDist
381	#endif
382	);
383
384
385	Void xCopyAMVPInfo ( AMVPInfo* pSrc, AMVPInfo* pDst );
386	UInt xGetMvpIdxBits ( Int iIdx, Int iNum );
387	Void xGetBlkBits ( PartSize eCUMode, Bool bPSlice, Int iPartIdx, UInt uiLastMode, UInt uiBlkBit[3]);
388
389	Void xMergeEstimation ( TComDataCU* pcCU,
390	TComYuv* pcYuvOrg,
391	Int iPartIdx,
392	UInt& uiInterDir,
393	TComMvField* pacMvField,
394	UInt& uiMergeIndex,
395	UInt& ruiCost
396	, TComMvField* cMvFieldNeighbours,
397	UChar* uhInterDirNeighbours,
398	Int& numValidMergeCand
399	);
400
401	Void xRestrictBipredMergeCand ( TComDataCU* pcCU,
402	UInt puIdx,
403	TComMvField* mvFieldNeighbours,
404	UChar* interDirNeighbours,
405	Int numValidMergeCand );
406
407	// -------------------------------------------------------------------------------------------------------------------
408	// motion estimation
409	// -------------------------------------------------------------------------------------------------------------------
410
411	Void xMotionEstimation ( TComDataCU* pcCU,
412	TComYuv* pcYuvOrg,
413	Int iPartIdx,
414	RefPicList eRefPicList,
415	TComMv* pcMvPred,
416	Int iRefIdxPred,
417	TComMv& rcMv,
418	UInt& ruiBits,
419	UInt& ruiCost,
420	Bool bBi = false );
421
422	Void xTZSearch ( TComDataCU* pcCU,
423	TComPattern* pcPatternKey,
424	Pel* piRefY,
425	Int iRefStride,
426	TComMv* pcMvSrchRngLT,
427	TComMv* pcMvSrchRngRB,
428	TComMv& rcMv,
429	UInt& ruiSAD );
430
431	Void xSetSearchRange ( TComDataCU* pcCU,
432	TComMv& cMvPred,
433	Int iSrchRng,
434	TComMv& rcMvSrchRngLT,
435	TComMv& rcMvSrchRngRB );
436
437	Void xPatternSearchFast ( TComDataCU* pcCU,
438	TComPattern* pcPatternKey,
439	Pel* piRefY,
440	Int iRefStride,
441	TComMv* pcMvSrchRngLT,
442	TComMv* pcMvSrchRngRB,
443	TComMv& rcMv,
444	UInt& ruiSAD );
445
446	Void xPatternSearch ( TComPattern* pcPatternKey,
447	Pel* piRefY,
448	Int iRefStride,
449	TComMv* pcMvSrchRngLT,
450	TComMv* pcMvSrchRngRB,
451	TComMv& rcMv,
452	UInt& ruiSAD );
453
454	Void xPatternSearchFracDIF ( TComDataCU* pcCU,
455	TComPattern* pcPatternKey,
456	Pel* piRefY,
457	Int iRefStride,
458	TComMv* pcMvInt,
459	TComMv& rcMvHalf,
460	TComMv& rcMvQter,
461	UInt& ruiCost
462	,Bool biPred
463	);
464	#if REF_IDX_ME_ZEROMV
465	Void xPatternSearchFracDIFMv0 ( TComDataCU* pcCU,
466	TComPattern* pcPatternKey,
467	Pel* piRefY,
468	Int iRefStride,
469	TComMv* pcMvInt,
470	TComMv& rcMvHalf,
471	TComMv& rcMvQter,
472	UInt& ruiCost,
473	Bool biPred );
474	#endif
475	Void xExtDIFUpSamplingH( TComPattern* pcPattern, Bool biPred );
476	Void xExtDIFUpSamplingQ( TComPattern* pcPatternKey, TComMv halfPelRef, Bool biPred );
477
478	// -------------------------------------------------------------------------------------------------------------------
479	// T & Q & Q-1 & T-1
480	// -------------------------------------------------------------------------------------------------------------------
481
482	Void xEncodeResidualQT( TComDataCU* pcCU, UInt uiAbsPartIdx, const UInt uiDepth, Bool bSubdivAndCbf, TextType eType );
483	Void xEstimateResidualQT( TComDataCU* pcCU, UInt uiQuadrant, UInt uiAbsPartIdx, UInt absTUPartIdx,TComYuv* pcResi, const UInt uiDepth, Double &rdCost, UInt &ruiBits, UInt &ruiDist, UInt *puiZeroDist );
484	Void xSetResidualQTData( TComDataCU* pcCU, UInt uiQuadrant, UInt uiAbsPartIdx,UInt absTUPartIdx, TComYuv* pcResi, UInt uiDepth, Bool bSpatial );
485
486	UInt xModeBitsIntra ( TComDataCU* pcCU, UInt uiMode, UInt uiPU, UInt uiPartOffset, UInt uiDepth, UInt uiInitTrDepth );
487	UInt xUpdateCandList( UInt uiMode, Double uiCost, UInt uiFastCandNum, UInt * CandModeList, Double * CandCostList );
488
489	// -------------------------------------------------------------------------------------------------------------------
490	// compute symbol bits
491	// -------------------------------------------------------------------------------------------------------------------
492
493	Void xAddSymbolBitsInter ( TComDataCU* pcCU,
494	UInt uiQp,
495	UInt uiTrMode,
496	UInt& ruiBits,
497	TComYuv*& rpcYuvRec,
498	TComYuv* pcYuvPred,
499	TComYuv*& rpcYuvResi );
500
501	Void setWpScalingDistParam( TComDataCU* pcCU, Int iRefIdx, RefPicList eRefPicListCur );
502	inline Void setDistParamComp( UInt uiComp ) { m_cDistParam.uiComp = uiComp; }
503
504	};// END CLASS DEFINITION TEncSearch
505
506	//! \}
507
508	#endif // __TENCSEARCH__

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

Download in other formats: