Context navigation

source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibEncoder/TEncSearch.h @ 1502

Visit:

Last change on this file since 1502 was 1487, checked in by fujitsu, 10 years ago
Added support of scalable range extension profiles. Also corrected few issues such as the confWindow wrong initialization. Code is controlled by the macro SCALABLE_REXT.
Property svn:eol-style set to `native`
File size: 23.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-2015, 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 "TLibCommon/TComRectangle.h"
49	#include "TEncEntropy.h"
50	#include "TEncSbac.h"
51	#include "TEncCfg.h"
52
53
54	//! \ingroup TLibEncoder
55	//! \{
56
57	class TEncCu;
58
59	// ====================================================================================================================
60	// Class definition
61	// ====================================================================================================================
62
63	static const UInt MAX_NUM_REF_LIST_ADAPT_SR=2;
64	static const UInt MAX_IDX_ADAPT_SR=33;
65	static const UInt NUM_MV_PREDICTORS=3;
66
67	/// encoder search class
68	class TEncSearch : public TComPrediction
69	{
70	private:
71	TCoeff** m_ppcQTTempCoeff[MAX_NUM_COMPONENT /* 0->Y, 1->Cb, 2->Cr*/];
72	#if ADAPTIVE_QP_SELECTION
73	TCoeff** m_ppcQTTempArlCoeff[MAX_NUM_COMPONENT];
74	#endif
75	UChar* m_puhQTTempTrIdx;
76	UChar* m_puhQTTempCbf[MAX_NUM_COMPONENT];
77
78	TComYuv* m_pcQTTempTComYuv;
79	TComYuv m_tmpYuvPred; // To be used in xGetInterPredictionError() to avoid constant memory allocation/deallocation
80
81	SChar* m_phQTTempCrossComponentPredictionAlpha[MAX_NUM_COMPONENT];
82	Pel* m_pSharedPredTransformSkip[MAX_NUM_COMPONENT];
83	TCoeff* m_pcQTTempTUCoeff[MAX_NUM_COMPONENT];
84	UChar* m_puhQTTempTransformSkipFlag[MAX_NUM_COMPONENT];
85	TComYuv m_pcQTTempTransformSkipTComYuv;
86	#if ADAPTIVE_QP_SELECTION
87	TCoeff* m_ppcQTTempTUArlCoeff[MAX_NUM_COMPONENT];
88	#endif
89
90	protected:
91	// interface to option
92	TEncCfg* m_pcEncCfg;
93
94	// interface to classes
95	TComTrQuant* m_pcTrQuant;
96	TComRdCost* m_pcRdCost;
97	TEncEntropy* m_pcEntropyCoder;
98
99	// ME parameters
100	Int m_iSearchRange;
101	Int m_bipredSearchRange; // Search range for bi-prediction
102	MESearchMethod m_motionEstimationSearchMethod;
103	Int m_aaiAdaptSR[MAX_NUM_REF_LIST_ADAPT_SR][MAX_IDX_ADAPT_SR];
104	TComMv m_acMvPredictors[NUM_MV_PREDICTORS]; // Left, Above, AboveRight. enum MVP_DIR first NUM_MV_PREDICTORS entries are suitable for accessing.
105
106	// RD computation
107	TEncSbac*** m_pppcRDSbacCoder;
108	TEncSbac* m_pcRDGoOnSbacCoder;
109	DistParam m_cDistParam;
110
111	// Misc.
112	Pel* m_pTempPel;
113
114	// AMVP cost computation
115	// UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+1][AMVP_MAX_NUM_CANDS];
116	UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+1][AMVP_MAX_NUM_CANDS+1]; //th array bounds
117
118	TComMv m_integerMv2Nx2N[NUM_REF_PIC_LIST_01][MAX_NUM_REF];
119
120	Bool m_isInitialized;
121
122	#if SVC_EXTENSION && N0383_IL_CONSTRAINED_TILE_SETS_SEI
123	Bool m_disableILP;
124	#endif
125
126	public:
127	TEncSearch();
128	virtual ~TEncSearch();
129
130	Void init(TEncCfg* pcEncCfg,
131	TComTrQuant* pcTrQuant,
132	Int iSearchRange,
133	Int bipredSearchRange,
134	MESearchMethod motionEstimationSearchMethod,
135	const UInt maxCUWidth,
136	const UInt maxCUHeight,
137	const UInt maxTotalCUDepth,
138	TEncEntropy* pcEntropyCoder,
139	TComRdCost* pcRdCost,
140	TEncSbac*** pppcRDSbacCoder,
141	TEncSbac* pcRDGoOnSbacCoder );
142
143	Void destroy();
144
145	protected:
146
147	/// sub-function for motion vector refinement used in fractional-pel accuracy
148	Distortion xPatternRefinement( TComPattern* pcPatternKey,
149	TComMv baseRefMv,
150	Int iFrac, TComMv& rcMvFrac, Bool bAllowUseOfHadamard
151	);
152
153	typedef struct
154	{
155	const Pel* piRefY;
156	Int iYStride;
157	Int iBestX;
158	Int iBestY;
159	UInt uiBestRound;
160	UInt uiBestDistance;
161	Distortion uiBestSad;
162	UChar ucPointNr;
163	} IntTZSearchStruct;
164
165	// sub-functions for ME
166	__inline Void xTZSearchHelp ( const TComPattern* const pcPatternKey, IntTZSearchStruct& rcStruct, const Int iSearchX, const Int iSearchY, const UChar ucPointNr, const UInt uiDistance );
167	__inline Void xTZ2PointSearch ( const TComPattern* const pcPatternKey, IntTZSearchStruct& rcStruct, const TComMv* const pcMvSrchRngLT, const TComMv* const pcMvSrchRngRB );
168	__inline Void xTZ8PointSquareSearch ( const TComPattern* const pcPatternKey, IntTZSearchStruct& rcStruct, const TComMv* const pcMvSrchRngLT, const TComMv* const pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist );
169	__inline Void xTZ8PointDiamondSearch( const TComPattern* const pcPatternKey, IntTZSearchStruct& rcStruct, const TComMv* const pcMvSrchRngLT, const TComMv* const pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist, const Bool bCheckCornersAtDist1 );
170
171	Void xGetInterPredictionError( TComDataCU* pcCU, TComYuv* pcYuvOrg, Int iPartIdx, Distortion& ruiSAD, Bool Hadamard );
172
173	public:
174	Void estIntraPredLumaQT ( TComDataCU* pcCU,
175	TComYuv* pcOrgYuv,
176	TComYuv* pcPredYuv,
177	TComYuv* pcResiYuv,
178	TComYuv* pcRecoYuv,
179	Pel resiLuma[NUMBER_OF_STORED_RESIDUAL_TYPES][MAX_CU_SIZE * MAX_CU_SIZE]
180	DEBUG_STRING_FN_DECLARE(sDebug));
181
182	Void estIntraPredChromaQT ( TComDataCU* pcCU,
183	TComYuv* pcOrgYuv,
184	TComYuv* pcPredYuv,
185	TComYuv* pcResiYuv,
186	TComYuv* pcRecoYuv,
187	Pel resiLuma[NUMBER_OF_STORED_RESIDUAL_TYPES][MAX_CU_SIZE * MAX_CU_SIZE]
188	DEBUG_STRING_FN_DECLARE(sDebug));
189
190	/// encoder estimation - inter prediction (non-skip)
191	#if SVC_EXTENSION
192	Bool predInterSearch ( TComDataCU* pcCU,
193	#else
194	Void predInterSearch ( TComDataCU* pcCU,
195	#endif
196	TComYuv* pcOrgYuv,
197	TComYuv* pcPredYuv,
198	TComYuv* pcResiYuv,
199	TComYuv* pcRecoYuv
200	DEBUG_STRING_FN_DECLARE(sDebug),
201	Bool bUseRes = false
202	#if AMP_MRG
203	,Bool bUseMRG = false
204	#endif
205	);
206
207	/// encode residual and compute rd-cost for inter mode
208	Void encodeResAndCalcRdInterCU( TComDataCU* pcCU,
209	TComYuv* pcYuvOrg,
210	TComYuv* pcYuvPred,
211	TComYuv* pcYuvResi,
212	TComYuv* pcYuvResiBest,
213	TComYuv* pcYuvRec,
214	Bool bSkipResidual
215	DEBUG_STRING_FN_DECLARE(sDebug) );
216
217	/// set ME search range
218	Void setAdaptiveSearchRange ( Int iDir, Int iRefIdx, Int iSearchRange) { assert(iDir < MAX_NUM_REF_LIST_ADAPT_SR && iRefIdx<Int(MAX_IDX_ADAPT_SR)); m_aaiAdaptSR[iDir][iRefIdx] = iSearchRange; }
219
220	Void xEncPCM (TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piOrg, Pel* piPCM, Pel* piPred, Pel* piResi, Pel* piReco, UInt uiStride, UInt uiWidth, UInt uiHeight, const ComponentID compID );
221	Void IPCMSearch (TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv* rpcPredYuv, TComYuv* rpcResiYuv, TComYuv* rpcRecoYuv );
222
223	#if SVC_EXTENSION
224	#if ENCODER_FAST_MODE
225	Bool predInterSearchILRUni ( TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv& rpcPredYuv, TComYuv& rpcResiYuv, TComYuv*& rpcRecoYuv, UInt refLayerId );
226	#endif
227	#if N0383_IL_CONSTRAINED_TILE_SETS_SEI
228	Void setDisableILP(Bool a) {m_disableILP = a;}
229	#endif
230	#endif //SVC_EXTENSION
231
232	protected:
233
234	// -------------------------------------------------------------------------------------------------------------------
235	// Intra search
236	// -------------------------------------------------------------------------------------------------------------------
237
238	Void xEncSubdivCbfQT ( TComTU &rTu,
239	Bool bLuma,
240	Bool bChroma );
241
242	Void xEncCoeffQT ( TComTU &rTu,
243	ComponentID component,
244	Bool bRealCoeff );
245	Void xEncIntraHeader ( TComDataCU* pcCU,
246	UInt uiTrDepth,
247	UInt uiAbsPartIdx,
248	Bool bLuma,
249	Bool bChroma );
250	UInt xGetIntraBitsQT ( TComTU &rTu,
251	Bool bLuma,
252	Bool bChroma,
253	Bool bRealCoeff );
254
255	UInt xGetIntraBitsQTChroma ( TComTU &rTu,
256	ComponentID compID,
257	Bool bRealCoeff );
258
259	Void xIntraCodingTUBlock ( TComYuv* pcOrgYuv,
260	TComYuv* pcPredYuv,
261	TComYuv* pcResiYuv,
262	Pel resiLuma[NUMBER_OF_STORED_RESIDUAL_TYPES][MAX_CU_SIZE * MAX_CU_SIZE],
263	const Bool checkCrossCPrediction,
264	Distortion& ruiDist,
265	const ComponentID compID,
266	TComTU &rTu
267	DEBUG_STRING_FN_DECLARE(sTest)
268	,Int default0Save1Load2 = 0
269	);
270
271	Void xRecurIntraCodingLumaQT ( TComYuv* pcOrgYuv,
272	TComYuv* pcPredYuv,
273	TComYuv* pcResiYuv,
274	Pel resiLuma[NUMBER_OF_STORED_RESIDUAL_TYPES][MAX_CU_SIZE * MAX_CU_SIZE],
275	Distortion& ruiDistY,
276	#if HHI_RQT_INTRA_SPEEDUP
277	Bool bCheckFirst,
278	#endif
279	Double& dRDCost,
280	TComTU &rTu
281	DEBUG_STRING_FN_DECLARE(sDebug));
282
283	Void xSetIntraResultLumaQT ( TComYuv* pcRecoYuv,
284	TComTU &rTu);
285
286	Void xStoreCrossComponentPredictionResult ( Pel *pResiLuma,
287	const Pel *pBestLuma,
288	TComTU &rTu,
289	const Int xOffset,
290	const Int yOffset,
291	const Int strideResi,
292	const Int strideBest );
293
294	SChar xCalcCrossComponentPredictionAlpha ( TComTU &rTu,
295	const ComponentID compID,
296	const Pel* piResiL,
297	const Pel* piResiC,
298	const Int width,
299	const Int height,
300	const Int strideL,
301	const Int strideC );
302
303	Void xRecurIntraChromaCodingQT ( TComYuv* pcOrgYuv,
304	TComYuv* pcPredYuv,
305	TComYuv* pcResiYuv,
306	Pel resiLuma[NUMBER_OF_STORED_RESIDUAL_TYPES][MAX_CU_SIZE * MAX_CU_SIZE],
307	Distortion& ruiDist,
308	TComTU &rTu
309	DEBUG_STRING_FN_DECLARE(sDebug));
310
311	Void xSetIntraResultChromaQT ( TComYuv* pcRecoYuv, TComTU &rTu);
312
313	Void xStoreIntraResultQT ( const ComponentID compID, TComTU &rTu);
314	Void xLoadIntraResultQT ( const ComponentID compID, TComTU &rTu);
315
316
317	// -------------------------------------------------------------------------------------------------------------------
318	// Inter search (AMP)
319	// -------------------------------------------------------------------------------------------------------------------
320
321	Void xEstimateMvPredAMVP ( TComDataCU* pcCU,
322	TComYuv* pcOrgYuv,
323	UInt uiPartIdx,
324	RefPicList eRefPicList,
325	Int iRefIdx,
326	TComMv& rcMvPred,
327	Bool bFilled = false
328	, Distortion* puiDistBiP = NULL
329	);
330
331	Void xCheckBestMVP ( TComDataCU* pcCU,
332	RefPicList eRefPicList,
333	TComMv cMv,
334	TComMv& rcMvPred,
335	Int& riMVPIdx,
336	UInt& ruiBits,
337	Distortion& ruiCost );
338
339	Distortion xGetTemplateCost ( TComDataCU* pcCU,
340	UInt uiPartAddr,
341	TComYuv* pcOrgYuv,
342	TComYuv* pcTemplateCand,
343	TComMv cMvCand,
344	Int iMVPIdx,
345	Int iMVPNum,
346	RefPicList eRefPicList,
347	Int iRefIdx,
348	Int iSizeX,
349	Int iSizeY
350	);
351
352
353	Void xCopyAMVPInfo ( AMVPInfo* pSrc, AMVPInfo* pDst );
354	UInt xGetMvpIdxBits ( Int iIdx, Int iNum );
355	Void xGetBlkBits ( PartSize eCUMode, Bool bPSlice, Int iPartIdx, UInt uiLastMode, UInt uiBlkBit[3]);
356
357	Void xMergeEstimation ( TComDataCU* pcCU,
358	TComYuv* pcYuvOrg,
359	Int iPartIdx,
360	UInt& uiInterDir,
361	TComMvField* pacMvField,
362	UInt& uiMergeIndex,
363	Distortion& ruiCost,
364	TComMvField* cMvFieldNeighbours,
365	UChar* uhInterDirNeighbours,
366	Int& numValidMergeCand
367	);
368
369	Void xRestrictBipredMergeCand ( TComDataCU* pcCU,
370	UInt puIdx,
371	TComMvField* mvFieldNeighbours,
372	UChar* interDirNeighbours,
373	Int numValidMergeCand );
374
375
376	// -------------------------------------------------------------------------------------------------------------------
377	// motion estimation
378	// -------------------------------------------------------------------------------------------------------------------
379
380	Void xMotionEstimation ( TComDataCU* pcCU,
381	TComYuv* pcYuvOrg,
382	Int iPartIdx,
383	RefPicList eRefPicList,
384	TComMv* pcMvPred,
385	Int iRefIdxPred,
386	TComMv& rcMv,
387	UInt& ruiBits,
388	Distortion& ruiCost,
389	Bool bBi = false );
390
391	Void xTZSearch ( const TComDataCU* const pcCU,
392	const TComPattern* const pcPatternKey,
393	const Pel* const piRefY,
394	const Int iRefStride,
395	const TComMv* const pcMvSrchRngLT,
396	const TComMv* const pcMvSrchRngRB,
397	TComMv& rcMv,
398	Distortion& ruiSAD,
399	const TComMv* const pIntegerMv2Nx2NPred,
400	const Bool bExtendedSettings
401	);
402
403	Void xTZSearchSelective ( const TComDataCU* const pcCU,
404	const TComPattern* const pcPatternKey,
405	const Pel* const piRefY,
406	const Int iRefStride,
407	const TComMv* const pcMvSrchRngLT,
408	const TComMv* const pcMvSrchRngRB,
409	TComMv& rcMv,
410	Distortion& ruiSAD,
411	const TComMv* const pIntegerMv2Nx2NPred
412	);
413
414	Void xSetSearchRange ( const TComDataCU* const pcCU,
415	const TComMv& cMvPred,
416	const Int iSrchRng,
417	TComMv& rcMvSrchRngLT,
418	TComMv& rcMvSrchRngRB );
419
420	Void xPatternSearchFast ( const TComDataCU* const pcCU,
421	const TComPattern* const pcPatternKey,
422	const Pel* const piRefY,
423	const Int iRefStride,
424	const TComMv* const pcMvSrchRngLT,
425	const TComMv* const pcMvSrchRngRB,
426	TComMv& rcMv,
427	Distortion& ruiSAD,
428	const TComMv* const pIntegerMv2Nx2NPred
429	);
430
431	Void xPatternSearch ( const TComPattern* const pcPatternKey,
432	const Pel* piRefY,
433	const Int iRefStride,
434	const TComMv* const pcMvSrchRngLT,
435	const TComMv* const pcMvSrchRngRB,
436	TComMv& rcMv,
437	Distortion& ruiSAD );
438
439	Void xPatternSearchFracDIF (
440	Bool bIsLosslessCoded,
441	TComPattern* pcPatternKey,
442	Pel* piRefY,
443	Int iRefStride,
444	TComMv* pcMvInt,
445	TComMv& rcMvHalf,
446	TComMv& rcMvQter,
447	Distortion& ruiCost
448	);
449
450	Void xExtDIFUpSamplingH( TComPattern* pcPattern );
451	Void xExtDIFUpSamplingQ( TComPattern* pcPatternKey, TComMv halfPelRef );
452
453	// -------------------------------------------------------------------------------------------------------------------
454	// T & Q & Q-1 & T-1
455	// -------------------------------------------------------------------------------------------------------------------
456
457
458	Void xEncodeInterResidualQT( const ComponentID compID, TComTU &rTu );
459	Void xEstimateInterResidualQT( TComYuv* pcResi, Double &rdCost, UInt &ruiBits, Distortion &ruiDist, Distortion *puiZeroDist, TComTU &rTu DEBUG_STRING_FN_DECLARE(sDebug) );
460	Void xSetInterResidualQTData( TComYuv* pcResi, Bool bSpatial, TComTU &rTu );
461
462	UInt xModeBitsIntra ( TComDataCU* pcCU, UInt uiMode, UInt uiPartOffset, UInt uiDepth, const ChannelType compID );
463	UInt xUpdateCandList( UInt uiMode, Double uiCost, UInt uiFastCandNum, UInt * CandModeList, Double * CandCostList );
464
465	// -------------------------------------------------------------------------------------------------------------------
466	// compute symbol bits
467	// -------------------------------------------------------------------------------------------------------------------
468
469	Void xAddSymbolBitsInter ( TComDataCU* pcCU,
470	UInt& ruiBits);
471
472	Void setWpScalingDistParam( TComDataCU* pcCU, Int iRefIdx, RefPicList eRefPicListCur );
473	inline Void setDistParamComp( ComponentID compIdx ) { m_cDistParam.compIdx = compIdx; }
474
475	#if SVC_EXTENSION && REF_IDX_ME_ZEROMV
476	Void xPatternSearchFracDIFMv0 ( TComPattern* pcPatternKey,
477	Pel* piRefY,
478	Int iRefStride,
479	TComMv* pcMvInt,
480	TComMv& rcMvHalf,
481	TComMv& rcMvQter,
482	#if SCALABLE_REXT
483	Distortion& ruiCost );
484	#else
485	UInt& ruiCost );
486	#endif
487	#endif //SVC_EXTENSION
488
489	};// END CLASS DEFINITION TEncSearch
490
491	//! \}
492
493	#endif // __TENCSEARCH__

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

Download in other formats: