Context navigation

TEncSearch.cpp @ 446

Visit:

Last change on this file since 446 was 446, checked in by tech, 11 years ago
Added missing parts.
Property svn:eol-style set to `native`
File size: 246.3 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.cpp
35	\brief encoder search class
36	*/
37
38	#include "TLibCommon/TypeDef.h"
39	#include "TLibCommon/TComRom.h"
40	#include "TLibCommon/TComMotionInfo.h"
41	#include "TEncSearch.h"
42	#include <math.h>
43
44	//! \ingroup TLibEncoder
45	//! \{
46
47	static const TComMv s_acMvRefineH[9] =
48	{
49	TComMv( 0, 0 ), // 0
50	TComMv( 0, -1 ), // 1
51	TComMv( 0, 1 ), // 2
52	TComMv( -1, 0 ), // 3
53	TComMv( 1, 0 ), // 4
54	TComMv( -1, -1 ), // 5
55	TComMv( 1, -1 ), // 6
56	TComMv( -1, 1 ), // 7
57	TComMv( 1, 1 ) // 8
58	};
59
60	static const TComMv s_acMvRefineQ[9] =
61	{
62	TComMv( 0, 0 ), // 0
63	TComMv( 0, -1 ), // 1
64	TComMv( 0, 1 ), // 2
65	TComMv( -1, -1 ), // 5
66	TComMv( 1, -1 ), // 6
67	TComMv( -1, 0 ), // 3
68	TComMv( 1, 0 ), // 4
69	TComMv( -1, 1 ), // 7
70	TComMv( 1, 1 ) // 8
71	};
72
73	static const UInt s_auiDFilter[9] =
74	{
75	0, 1, 0,
76	2, 3, 2,
77	0, 1, 0
78	};
79
80	TEncSearch::TEncSearch()
81	{
82	m_ppcQTTempCoeffY = NULL;
83	m_ppcQTTempCoeffCb = NULL;
84	m_ppcQTTempCoeffCr = NULL;
85	m_pcQTTempCoeffY = NULL;
86	m_pcQTTempCoeffCb = NULL;
87	m_pcQTTempCoeffCr = NULL;
88	#if ADAPTIVE_QP_SELECTION
89	m_ppcQTTempArlCoeffY = NULL;
90	m_ppcQTTempArlCoeffCb = NULL;
91	m_ppcQTTempArlCoeffCr = NULL;
92	m_pcQTTempArlCoeffY = NULL;
93	m_pcQTTempArlCoeffCb = NULL;
94	m_pcQTTempArlCoeffCr = NULL;
95	#endif
96	m_puhQTTempTrIdx = NULL;
97	m_puhQTTempCbf[0] = m_puhQTTempCbf[1] = m_puhQTTempCbf[2] = NULL;
98	m_pcQTTempTComYuv = NULL;
99	m_pcEncCfg = NULL;
100	m_pcEntropyCoder = NULL;
101	m_pTempPel = NULL;
102	m_pSharedPredTransformSkip[0] = m_pSharedPredTransformSkip[1] = m_pSharedPredTransformSkip[2] = NULL;
103	m_pcQTTempTUCoeffY = NULL;
104	m_pcQTTempTUCoeffCb = NULL;
105	m_pcQTTempTUCoeffCr = NULL;
106	#if ADAPTIVE_QP_SELECTION
107	m_ppcQTTempTUArlCoeffY = NULL;
108	m_ppcQTTempTUArlCoeffCb = NULL;
109	m_ppcQTTempTUArlCoeffCr = NULL;
110	#endif
111	m_puhQTTempTransformSkipFlag[0] = NULL;
112	m_puhQTTempTransformSkipFlag[1] = NULL;
113	m_puhQTTempTransformSkipFlag[2] = NULL;
114	setWpScalingDistParam( NULL, -1, REF_PIC_LIST_X );
115	}
116
117	TEncSearch::~TEncSearch()
118	{
119	if ( m_pTempPel )
120	{
121	delete [] m_pTempPel;
122	m_pTempPel = NULL;
123	}
124
125	if ( m_pcEncCfg )
126	{
127	const UInt uiNumLayersAllocated = m_pcEncCfg->getQuadtreeTULog2MaxSize()-m_pcEncCfg->getQuadtreeTULog2MinSize()+1;
128	for( UInt ui = 0; ui < uiNumLayersAllocated; ++ui )
129	{
130	delete[] m_ppcQTTempCoeffY[ui];
131	delete[] m_ppcQTTempCoeffCb[ui];
132	delete[] m_ppcQTTempCoeffCr[ui];
133	#if ADAPTIVE_QP_SELECTION
134	delete[] m_ppcQTTempArlCoeffY[ui];
135	delete[] m_ppcQTTempArlCoeffCb[ui];
136	delete[] m_ppcQTTempArlCoeffCr[ui];
137	#endif
138	m_pcQTTempTComYuv[ui].destroy();
139	}
140	}
141	delete[] m_ppcQTTempCoeffY;
142	delete[] m_ppcQTTempCoeffCb;
143	delete[] m_ppcQTTempCoeffCr;
144	delete[] m_pcQTTempCoeffY;
145	delete[] m_pcQTTempCoeffCb;
146	delete[] m_pcQTTempCoeffCr;
147	#if ADAPTIVE_QP_SELECTION
148	delete[] m_ppcQTTempArlCoeffY;
149	delete[] m_ppcQTTempArlCoeffCb;
150	delete[] m_ppcQTTempArlCoeffCr;
151	delete[] m_pcQTTempArlCoeffY;
152	delete[] m_pcQTTempArlCoeffCb;
153	delete[] m_pcQTTempArlCoeffCr;
154	#endif
155	delete[] m_puhQTTempTrIdx;
156	delete[] m_puhQTTempCbf[0];
157	delete[] m_puhQTTempCbf[1];
158	delete[] m_puhQTTempCbf[2];
159	delete[] m_pcQTTempTComYuv;
160	delete[] m_pSharedPredTransformSkip[0];
161	delete[] m_pSharedPredTransformSkip[1];
162	delete[] m_pSharedPredTransformSkip[2];
163	delete[] m_pcQTTempTUCoeffY;
164	delete[] m_pcQTTempTUCoeffCb;
165	delete[] m_pcQTTempTUCoeffCr;
166	#if ADAPTIVE_QP_SELECTION
167	delete[] m_ppcQTTempTUArlCoeffY;
168	delete[] m_ppcQTTempTUArlCoeffCb;
169	delete[] m_ppcQTTempTUArlCoeffCr;
170	#endif
171	delete[] m_puhQTTempTransformSkipFlag[0];
172	delete[] m_puhQTTempTransformSkipFlag[1];
173	delete[] m_puhQTTempTransformSkipFlag[2];
174	m_pcQTTempTransformSkipTComYuv.destroy();
175	m_tmpYuvPred.destroy();
176	}
177
178	void TEncSearch::init(TEncCfg* pcEncCfg,
179	TComTrQuant* pcTrQuant,
180	Int iSearchRange,
181	Int bipredSearchRange,
182	Int iFastSearch,
183	Int iMaxDeltaQP,
184	TEncEntropy* pcEntropyCoder,
185	TComRdCost* pcRdCost,
186	TEncSbac*** pppcRDSbacCoder,
187	TEncSbac* pcRDGoOnSbacCoder
188	)
189	{
190	m_pcEncCfg = pcEncCfg;
191	m_pcTrQuant = pcTrQuant;
192	m_iSearchRange = iSearchRange;
193	m_bipredSearchRange = bipredSearchRange;
194	m_iFastSearch = iFastSearch;
195	m_iMaxDeltaQP = iMaxDeltaQP;
196	m_pcEntropyCoder = pcEntropyCoder;
197	m_pcRdCost = pcRdCost;
198
199	m_pppcRDSbacCoder = pppcRDSbacCoder;
200	m_pcRDGoOnSbacCoder = pcRDGoOnSbacCoder;
201
202	m_bUseSBACRD = pppcRDSbacCoder ? true : false;
203
204	for (Int iDir = 0; iDir < 2; iDir++)
205	{
206	for (Int iRefIdx = 0; iRefIdx < 33; iRefIdx++)
207	{
208	m_aaiAdaptSR[iDir][iRefIdx] = iSearchRange;
209	}
210	}
211
212	m_puiDFilter = s_auiDFilter + 4;
213
214	// initialize motion cost
215	#if !FIX203
216	m_pcRdCost->initRateDistortionModel( m_iSearchRange << 2 );
217	#endif
218
219	for( Int iNum = 0; iNum < AMVP_MAX_NUM_CANDS+1; iNum++)
220	{
221	for( Int iIdx = 0; iIdx < AMVP_MAX_NUM_CANDS; iIdx++)
222	{
223	if (iIdx < iNum)
224	m_auiMVPIdxCost[iIdx][iNum] = xGetMvpIdxBits(iIdx, iNum);
225	else
226	m_auiMVPIdxCost[iIdx][iNum] = MAX_INT;
227	}
228	}
229
230	initTempBuff();
231
232	m_pTempPel = new Pel[g_uiMaxCUWidth*g_uiMaxCUHeight];
233
234	const UInt uiNumLayersToAllocate = pcEncCfg->getQuadtreeTULog2MaxSize()-pcEncCfg->getQuadtreeTULog2MinSize()+1;
235	m_ppcQTTempCoeffY = new TCoeff*[uiNumLayersToAllocate];
236	m_ppcQTTempCoeffCb = new TCoeff*[uiNumLayersToAllocate];
237	m_ppcQTTempCoeffCr = new TCoeff*[uiNumLayersToAllocate];
238	m_pcQTTempCoeffY = new TCoeff [g_uiMaxCUWidth*g_uiMaxCUHeight ];
239	m_pcQTTempCoeffCb = new TCoeff [g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
240	m_pcQTTempCoeffCr = new TCoeff [g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
241	#if ADAPTIVE_QP_SELECTION
242	m_ppcQTTempArlCoeffY = new Int*[uiNumLayersToAllocate];
243	m_ppcQTTempArlCoeffCb = new Int*[uiNumLayersToAllocate];
244	m_ppcQTTempArlCoeffCr = new Int*[uiNumLayersToAllocate];
245	m_pcQTTempArlCoeffY = new Int [g_uiMaxCUWidth*g_uiMaxCUHeight ];
246	m_pcQTTempArlCoeffCb = new Int [g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
247	m_pcQTTempArlCoeffCr = new Int [g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
248	#endif
249
250	const UInt uiNumPartitions = 1<<(g_uiMaxCUDepth<<1);
251	m_puhQTTempTrIdx = new UChar [uiNumPartitions];
252	m_puhQTTempCbf[0] = new UChar [uiNumPartitions];
253	m_puhQTTempCbf[1] = new UChar [uiNumPartitions];
254	m_puhQTTempCbf[2] = new UChar [uiNumPartitions];
255	m_pcQTTempTComYuv = new TComYuv[uiNumLayersToAllocate];
256	for( UInt ui = 0; ui < uiNumLayersToAllocate; ++ui )
257	{
258	m_ppcQTTempCoeffY[ui] = new TCoeff[g_uiMaxCUWidth*g_uiMaxCUHeight ];
259	m_ppcQTTempCoeffCb[ui] = new TCoeff[g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
260	m_ppcQTTempCoeffCr[ui] = new TCoeff[g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
261	#if ADAPTIVE_QP_SELECTION
262	m_ppcQTTempArlCoeffY[ui] = new Int[g_uiMaxCUWidth*g_uiMaxCUHeight ];
263	m_ppcQTTempArlCoeffCb[ui] = new Int[g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
264	m_ppcQTTempArlCoeffCr[ui] = new Int[g_uiMaxCUWidth*g_uiMaxCUHeight>>2];
265	#endif
266	m_pcQTTempTComYuv[ui].create( g_uiMaxCUWidth, g_uiMaxCUHeight );
267	}
268	m_pSharedPredTransformSkip[0] = new Pel[MAX_TS_WIDTH*MAX_TS_HEIGHT];
269	m_pSharedPredTransformSkip[1] = new Pel[MAX_TS_WIDTH*MAX_TS_HEIGHT];
270	m_pSharedPredTransformSkip[2] = new Pel[MAX_TS_WIDTH*MAX_TS_HEIGHT];
271	m_pcQTTempTUCoeffY = new TCoeff[MAX_TS_WIDTH*MAX_TS_HEIGHT];
272	m_pcQTTempTUCoeffCb = new TCoeff[MAX_TS_WIDTH*MAX_TS_HEIGHT];
273	m_pcQTTempTUCoeffCr = new TCoeff[MAX_TS_WIDTH*MAX_TS_HEIGHT];
274	#if ADAPTIVE_QP_SELECTION
275	m_ppcQTTempTUArlCoeffY = new Int[MAX_TS_WIDTH*MAX_TS_HEIGHT];
276	m_ppcQTTempTUArlCoeffCb = new Int[MAX_TS_WIDTH*MAX_TS_HEIGHT];
277	m_ppcQTTempTUArlCoeffCr = new Int[MAX_TS_WIDTH*MAX_TS_HEIGHT];
278	#endif
279	m_pcQTTempTransformSkipTComYuv.create( g_uiMaxCUWidth, g_uiMaxCUHeight );
280
281	m_puhQTTempTransformSkipFlag[0] = new UChar [uiNumPartitions];
282	m_puhQTTempTransformSkipFlag[1] = new UChar [uiNumPartitions];
283	m_puhQTTempTransformSkipFlag[2] = new UChar [uiNumPartitions];
284	m_tmpYuvPred.create(MAX_CU_SIZE, MAX_CU_SIZE);
285	}
286
287	#if FASTME_SMOOTHER_MV
288	#define FIRSTSEARCHSTOP 1
289	#else
290	#define FIRSTSEARCHSTOP 0
291	#endif
292
293	#define TZ_SEARCH_CONFIGURATION \
294	const Int iRaster = 5; /* TZ soll von aussen ?ergeben werden */ \
295	const Bool bTestOtherPredictedMV = 0; \
296	const Bool bTestZeroVector = 1; \
297	const Bool bTestZeroVectorStart = 0; \
298	const Bool bTestZeroVectorStop = 0; \
299	const Bool bFirstSearchDiamond = 1; /* 1 = xTZ8PointDiamondSearch 0 = xTZ8PointSquareSearch */ \
300	const Bool bFirstSearchStop = FIRSTSEARCHSTOP; \
301	const UInt uiFirstSearchRounds = 3; /* first search stop X rounds after best match (must be >=1) */ \
302	const Bool bEnableRasterSearch = 1; \
303	const Bool bAlwaysRasterSearch = 0; /* ===== 1: BETTER but factor 2 slower ===== */ \
304	const Bool bRasterRefinementEnable = 0; /* enable either raster refinement or star refinement */ \
305	const Bool bRasterRefinementDiamond = 0; /* 1 = xTZ8PointDiamondSearch 0 = xTZ8PointSquareSearch */ \
306	const Bool bStarRefinementEnable = 1; /* enable either star refinement or raster refinement */ \
307	const Bool bStarRefinementDiamond = 1; /* 1 = xTZ8PointDiamondSearch 0 = xTZ8PointSquareSearch */ \
308	const Bool bStarRefinementStop = 0; \
309	const UInt uiStarRefinementRounds = 2; /* star refinement stop X rounds after best match (must be >=1) */ \
310
311
312	__inline Void TEncSearch::xTZSearchHelp( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, const Int iSearchX, const Int iSearchY, const UChar ucPointNr, const UInt uiDistance )
313	{
314	UInt uiSad;
315
316	Pel* piRefSrch;
317
318	piRefSrch = rcStruct.piRefY + iSearchY * rcStruct.iYStride + iSearchX;
319
320	//-- jclee for using the SAD function pointer
321	m_pcRdCost->setDistParam( pcPatternKey, piRefSrch, rcStruct.iYStride, m_cDistParam );
322
323	// fast encoder decision: use subsampled SAD when rows > 8 for integer ME
324	if ( m_pcEncCfg->getUseFastEnc() )
325	{
326	if ( m_cDistParam.iRows > 8 )
327	{
328	m_cDistParam.iSubShift = 1;
329	}
330	}
331
332	setDistParamComp(0); // Y component
333
334	// distortion
335	m_cDistParam.bitDepth = g_bitDepthY;
336	uiSad = m_cDistParam.DistFunc( &m_cDistParam );
337
338	// motion cost
339	uiSad += m_pcRdCost->getCost( iSearchX, iSearchY );
340
341	if( uiSad < rcStruct.uiBestSad )
342	{
343	rcStruct.uiBestSad = uiSad;
344	rcStruct.iBestX = iSearchX;
345	rcStruct.iBestY = iSearchY;
346	rcStruct.uiBestDistance = uiDistance;
347	rcStruct.uiBestRound = 0;
348	rcStruct.ucPointNr = ucPointNr;
349	}
350	}
351
352	__inline Void TEncSearch::xTZ2PointSearch( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB )
353	{
354	Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();
355	Int iSrchRngHorRight = pcMvSrchRngRB->getHor();
356	Int iSrchRngVerTop = pcMvSrchRngLT->getVer();
357	Int iSrchRngVerBottom = pcMvSrchRngRB->getVer();
358
359	// 2 point search, // 1 2 3
360	// check only the 2 untested points // 4 0 5
361	// around the start point // 6 7 8
362	Int iStartX = rcStruct.iBestX;
363	Int iStartY = rcStruct.iBestY;
364	switch( rcStruct.ucPointNr )
365	{
366	case 1:
367	{
368	if ( (iStartX - 1) >= iSrchRngHorLeft )
369	{
370	xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY, 0, 2 );
371	}
372	if ( (iStartY - 1) >= iSrchRngVerTop )
373	{
374	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iStartY - 1, 0, 2 );
375	}
376	}
377	break;
378	case 2:
379	{
380	if ( (iStartY - 1) >= iSrchRngVerTop )
381	{
382	if ( (iStartX - 1) >= iSrchRngHorLeft )
383	{
384	xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY - 1, 0, 2 );
385	}
386	if ( (iStartX + 1) <= iSrchRngHorRight )
387	{
388	xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY - 1, 0, 2 );
389	}
390	}
391	}
392	break;
393	case 3:
394	{
395	if ( (iStartY - 1) >= iSrchRngVerTop )
396	{
397	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iStartY - 1, 0, 2 );
398	}
399	if ( (iStartX + 1) <= iSrchRngHorRight )
400	{
401	xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY, 0, 2 );
402	}
403	}
404	break;
405	case 4:
406	{
407	if ( (iStartX - 1) >= iSrchRngHorLeft )
408	{
409	if ( (iStartY + 1) <= iSrchRngVerBottom )
410	{
411	xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY + 1, 0, 2 );
412	}
413	if ( (iStartY - 1) >= iSrchRngVerTop )
414	{
415	xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY - 1, 0, 2 );
416	}
417	}
418	}
419	break;
420	case 5:
421	{
422	if ( (iStartX + 1) <= iSrchRngHorRight )
423	{
424	if ( (iStartY - 1) >= iSrchRngVerTop )
425	{
426	xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY - 1, 0, 2 );
427	}
428	if ( (iStartY + 1) <= iSrchRngVerBottom )
429	{
430	xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY + 1, 0, 2 );
431	}
432	}
433	}
434	break;
435	case 6:
436	{
437	if ( (iStartX - 1) >= iSrchRngHorLeft )
438	{
439	xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY , 0, 2 );
440	}
441	if ( (iStartY + 1) <= iSrchRngVerBottom )
442	{
443	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iStartY + 1, 0, 2 );
444	}
445	}
446	break;
447	case 7:
448	{
449	if ( (iStartY + 1) <= iSrchRngVerBottom )
450	{
451	if ( (iStartX - 1) >= iSrchRngHorLeft )
452	{
453	xTZSearchHelp( pcPatternKey, rcStruct, iStartX - 1, iStartY + 1, 0, 2 );
454	}
455	if ( (iStartX + 1) <= iSrchRngHorRight )
456	{
457	xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY + 1, 0, 2 );
458	}
459	}
460	}
461	break;
462	case 8:
463	{
464	if ( (iStartX + 1) <= iSrchRngHorRight )
465	{
466	xTZSearchHelp( pcPatternKey, rcStruct, iStartX + 1, iStartY, 0, 2 );
467	}
468	if ( (iStartY + 1) <= iSrchRngVerBottom )
469	{
470	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iStartY + 1, 0, 2 );
471	}
472	}
473	break;
474	default:
475	{
476	assert( false );
477	}
478	break;
479	} // switch( rcStruct.ucPointNr )
480	}
481
482	__inline Void TEncSearch::xTZ8PointSquareSearch( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist )
483	{
484	Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();
485	Int iSrchRngHorRight = pcMvSrchRngRB->getHor();
486	Int iSrchRngVerTop = pcMvSrchRngLT->getVer();
487	Int iSrchRngVerBottom = pcMvSrchRngRB->getVer();
488
489	// 8 point search, // 1 2 3
490	// search around the start point // 4 0 5
491	// with the required distance // 6 7 8
492	assert( iDist != 0 );
493	const Int iTop = iStartY - iDist;
494	const Int iBottom = iStartY + iDist;
495	const Int iLeft = iStartX - iDist;
496	const Int iRight = iStartX + iDist;
497	rcStruct.uiBestRound += 1;
498
499	if ( iTop >= iSrchRngVerTop ) // check top
500	{
501	if ( iLeft >= iSrchRngHorLeft ) // check top left
502	{
503	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iTop, 1, iDist );
504	}
505	// top middle
506	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iTop, 2, iDist );
507
508	if ( iRight <= iSrchRngHorRight ) // check top right
509	{
510	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iTop, 3, iDist );
511	}
512	} // check top
513	if ( iLeft >= iSrchRngHorLeft ) // check middle left
514	{
515	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iStartY, 4, iDist );
516	}
517	if ( iRight <= iSrchRngHorRight ) // check middle right
518	{
519	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iStartY, 5, iDist );
520	}
521	if ( iBottom <= iSrchRngVerBottom ) // check bottom
522	{
523	if ( iLeft >= iSrchRngHorLeft ) // check bottom left
524	{
525	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iBottom, 6, iDist );
526	}
527	// check bottom middle
528	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iBottom, 7, iDist );
529
530	if ( iRight <= iSrchRngHorRight ) // check bottom right
531	{
532	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iBottom, 8, iDist );
533	}
534	} // check bottom
535	}
536
537	__inline Void TEncSearch::xTZ8PointDiamondSearch( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist )
538	{
539	Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();
540	Int iSrchRngHorRight = pcMvSrchRngRB->getHor();
541	Int iSrchRngVerTop = pcMvSrchRngLT->getVer();
542	Int iSrchRngVerBottom = pcMvSrchRngRB->getVer();
543
544	// 8 point search, // 1 2 3
545	// search around the start point // 4 0 5
546	// with the required distance // 6 7 8
547	assert ( iDist != 0 );
548	const Int iTop = iStartY - iDist;
549	const Int iBottom = iStartY + iDist;
550	const Int iLeft = iStartX - iDist;
551	const Int iRight = iStartX + iDist;
552	rcStruct.uiBestRound += 1;
553
554	if ( iDist == 1 ) // iDist == 1
555	{
556	if ( iTop >= iSrchRngVerTop ) // check top
557	{
558	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iTop, 2, iDist );
559	}
560	if ( iLeft >= iSrchRngHorLeft ) // check middle left
561	{
562	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iStartY, 4, iDist );
563	}
564	if ( iRight <= iSrchRngHorRight ) // check middle right
565	{
566	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iStartY, 5, iDist );
567	}
568	if ( iBottom <= iSrchRngVerBottom ) // check bottom
569	{
570	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iBottom, 7, iDist );
571	}
572	}
573	else // if (iDist != 1)
574	{
575	if ( iDist <= 8 )
576	{
577	const Int iTop_2 = iStartY - (iDist>>1);
578	const Int iBottom_2 = iStartY + (iDist>>1);
579	const Int iLeft_2 = iStartX - (iDist>>1);
580	const Int iRight_2 = iStartX + (iDist>>1);
581
582	if ( iTop >= iSrchRngVerTop && iLeft >= iSrchRngHorLeft &&
583	iRight <= iSrchRngHorRight && iBottom <= iSrchRngVerBottom ) // check border
584	{
585	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iTop, 2, iDist );
586	xTZSearchHelp( pcPatternKey, rcStruct, iLeft_2, iTop_2, 1, iDist>>1 );
587	xTZSearchHelp( pcPatternKey, rcStruct, iRight_2, iTop_2, 3, iDist>>1 );
588	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iStartY, 4, iDist );
589	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iStartY, 5, iDist );
590	xTZSearchHelp( pcPatternKey, rcStruct, iLeft_2, iBottom_2, 6, iDist>>1 );
591	xTZSearchHelp( pcPatternKey, rcStruct, iRight_2, iBottom_2, 8, iDist>>1 );
592	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iBottom, 7, iDist );
593	}
594	else // check border
595	{
596	if ( iTop >= iSrchRngVerTop ) // check top
597	{
598	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iTop, 2, iDist );
599	}
600	if ( iTop_2 >= iSrchRngVerTop ) // check half top
601	{
602	if ( iLeft_2 >= iSrchRngHorLeft ) // check half left
603	{
604	xTZSearchHelp( pcPatternKey, rcStruct, iLeft_2, iTop_2, 1, (iDist>>1) );
605	}
606	if ( iRight_2 <= iSrchRngHorRight ) // check half right
607	{
608	xTZSearchHelp( pcPatternKey, rcStruct, iRight_2, iTop_2, 3, (iDist>>1) );
609	}
610	} // check half top
611	if ( iLeft >= iSrchRngHorLeft ) // check left
612	{
613	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iStartY, 4, iDist );
614	}
615	if ( iRight <= iSrchRngHorRight ) // check right
616	{
617	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iStartY, 5, iDist );
618	}
619	if ( iBottom_2 <= iSrchRngVerBottom ) // check half bottom
620	{
621	if ( iLeft_2 >= iSrchRngHorLeft ) // check half left
622	{
623	xTZSearchHelp( pcPatternKey, rcStruct, iLeft_2, iBottom_2, 6, (iDist>>1) );
624	}
625	if ( iRight_2 <= iSrchRngHorRight ) // check half right
626	{
627	xTZSearchHelp( pcPatternKey, rcStruct, iRight_2, iBottom_2, 8, (iDist>>1) );
628	}
629	} // check half bottom
630	if ( iBottom <= iSrchRngVerBottom ) // check bottom
631	{
632	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iBottom, 7, iDist );
633	}
634	} // check border
635	}
636	else // iDist > 8
637	{
638	if ( iTop >= iSrchRngVerTop && iLeft >= iSrchRngHorLeft &&
639	iRight <= iSrchRngHorRight && iBottom <= iSrchRngVerBottom ) // check border
640	{
641	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iTop, 0, iDist );
642	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iStartY, 0, iDist );
643	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iStartY, 0, iDist );
644	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iBottom, 0, iDist );
645	for ( Int index = 1; index < 4; index++ )
646	{
647	Int iPosYT = iTop + ((iDist>>2) * index);
648	Int iPosYB = iBottom - ((iDist>>2) * index);
649	Int iPosXL = iStartX - ((iDist>>2) * index);
650	Int iPosXR = iStartX + ((iDist>>2) * index);
651	xTZSearchHelp( pcPatternKey, rcStruct, iPosXL, iPosYT, 0, iDist );
652	xTZSearchHelp( pcPatternKey, rcStruct, iPosXR, iPosYT, 0, iDist );
653	xTZSearchHelp( pcPatternKey, rcStruct, iPosXL, iPosYB, 0, iDist );
654	xTZSearchHelp( pcPatternKey, rcStruct, iPosXR, iPosYB, 0, iDist );
655	}
656	}
657	else // check border
658	{
659	if ( iTop >= iSrchRngVerTop ) // check top
660	{
661	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iTop, 0, iDist );
662	}
663	if ( iLeft >= iSrchRngHorLeft ) // check left
664	{
665	xTZSearchHelp( pcPatternKey, rcStruct, iLeft, iStartY, 0, iDist );
666	}
667	if ( iRight <= iSrchRngHorRight ) // check right
668	{
669	xTZSearchHelp( pcPatternKey, rcStruct, iRight, iStartY, 0, iDist );
670	}
671	if ( iBottom <= iSrchRngVerBottom ) // check bottom
672	{
673	xTZSearchHelp( pcPatternKey, rcStruct, iStartX, iBottom, 0, iDist );
674	}
675	for ( Int index = 1; index < 4; index++ )
676	{
677	Int iPosYT = iTop + ((iDist>>2) * index);
678	Int iPosYB = iBottom - ((iDist>>2) * index);
679	Int iPosXL = iStartX - ((iDist>>2) * index);
680	Int iPosXR = iStartX + ((iDist>>2) * index);
681
682	if ( iPosYT >= iSrchRngVerTop ) // check top
683	{
684	if ( iPosXL >= iSrchRngHorLeft ) // check left
685	{
686	xTZSearchHelp( pcPatternKey, rcStruct, iPosXL, iPosYT, 0, iDist );
687	}
688	if ( iPosXR <= iSrchRngHorRight ) // check right
689	{
690	xTZSearchHelp( pcPatternKey, rcStruct, iPosXR, iPosYT, 0, iDist );
691	}
692	} // check top
693	if ( iPosYB <= iSrchRngVerBottom ) // check bottom
694	{
695	if ( iPosXL >= iSrchRngHorLeft ) // check left
696	{
697	xTZSearchHelp( pcPatternKey, rcStruct, iPosXL, iPosYB, 0, iDist );
698	}
699	if ( iPosXR <= iSrchRngHorRight ) // check right
700	{
701	xTZSearchHelp( pcPatternKey, rcStruct, iPosXR, iPosYB, 0, iDist );
702	}
703	} // check bottom
704	} // for ...
705	} // check border
706	} // iDist <= 8
707	} // iDist == 1
708	}
709
710	//<--
711
712	UInt TEncSearch::xPatternRefinement( TComPattern* pcPatternKey,
713	TComMv baseRefMv,
714	Int iFrac, TComMv& rcMvFrac )
715	{
716	UInt uiDist;
717	UInt uiDistBest = MAX_UINT;
718	UInt uiDirecBest = 0;
719
720	Pel* piRefPos;
721	Int iRefStride = m_filteredBlock[0][0].getStride();
722	#if NS_HAD
723	m_pcRdCost->setDistParam( pcPatternKey, m_filteredBlock[0][0].getLumaAddr(), iRefStride, 1, m_cDistParam, m_pcEncCfg->getUseHADME(), m_pcEncCfg->getUseNSQT() );
724	#else
725	m_pcRdCost->setDistParam( pcPatternKey, m_filteredBlock[0][0].getLumaAddr(), iRefStride, 1, m_cDistParam, m_pcEncCfg->getUseHADME() );
726	#endif
727
728	const TComMv* pcMvRefine = (iFrac == 2 ? s_acMvRefineH : s_acMvRefineQ);
729
730	for (UInt i = 0; i < 9; i++)
731	{
732	TComMv cMvTest = pcMvRefine[i];
733	cMvTest += baseRefMv;
734
735	Int horVal = cMvTest.getHor() * iFrac;
736	Int verVal = cMvTest.getVer() * iFrac;
737	piRefPos = m_filteredBlock[ verVal & 3 ][ horVal & 3 ].getLumaAddr();
738	if ( horVal == 2 && ( verVal & 1 ) == 0 )
739	piRefPos += 1;
740	if ( ( horVal & 1 ) == 0 && verVal == 2 )
741	piRefPos += iRefStride;
742	cMvTest = pcMvRefine[i];
743	cMvTest += rcMvFrac;
744
745	setDistParamComp(0); // Y component
746
747	m_cDistParam.pCur = piRefPos;
748	m_cDistParam.bitDepth = g_bitDepthY;
749	uiDist = m_cDistParam.DistFunc( &m_cDistParam );
750	uiDist += m_pcRdCost->getCost( cMvTest.getHor(), cMvTest.getVer() );
751
752	if ( uiDist < uiDistBest )
753	{
754	uiDistBest = uiDist;
755	uiDirecBest = i;
756	}
757	}
758
759	rcMvFrac = pcMvRefine[uiDirecBest];
760
761	return uiDistBest;
762	}
763
764	Void
765	TEncSearch::xEncSubdivCbfQT( TComDataCU* pcCU,
766	UInt uiTrDepth,
767	UInt uiAbsPartIdx,
768	Bool bLuma,
769	Bool bChroma )
770	{
771	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
772	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
773	UInt uiSubdiv = ( uiTrMode > uiTrDepth ? 1 : 0 );
774	UInt uiLog2TrafoSize = g_aucConvertToBit[pcCU->getSlice()->getSPS()->getMaxCUWidth()] + 2 - uiFullDepth;
775
776	{
777	if( pcCU->getPredictionMode(0) == MODE_INTRA && pcCU->getPartitionSize(0) == SIZE_NxN && uiTrDepth == 0 )
778	{
779	assert( uiSubdiv );
780	}
781	else if( uiLog2TrafoSize > pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() )
782	{
783	assert( uiSubdiv );
784	}
785	else if( uiLog2TrafoSize == pcCU->getSlice()->getSPS()->getQuadtreeTULog2MinSize() )
786	{
787	assert( !uiSubdiv );
788	}
789	else if( uiLog2TrafoSize == pcCU->getQuadtreeTULog2MinSizeInCU(uiAbsPartIdx) )
790	{
791	assert( !uiSubdiv );
792	}
793	else
794	{
795	assert( uiLog2TrafoSize > pcCU->getQuadtreeTULog2MinSizeInCU(uiAbsPartIdx) );
796	if( bLuma )
797	{
798	m_pcEntropyCoder->encodeTransformSubdivFlag( uiSubdiv, 5 - uiLog2TrafoSize );
799	}
800	}
801	}
802
803	if ( bChroma )
804	{
805	if( uiLog2TrafoSize > 2 )
806	{
807	if( uiTrDepth==0 \|\| pcCU->getCbf( uiAbsPartIdx, TEXT_CHROMA_U, uiTrDepth-1 ) )
808	m_pcEntropyCoder->encodeQtCbf( pcCU, uiAbsPartIdx, TEXT_CHROMA_U, uiTrDepth );
809	if( uiTrDepth==0 \|\| pcCU->getCbf( uiAbsPartIdx, TEXT_CHROMA_V, uiTrDepth-1 ) )
810	m_pcEntropyCoder->encodeQtCbf( pcCU, uiAbsPartIdx, TEXT_CHROMA_V, uiTrDepth );
811	}
812	}
813
814	if( uiSubdiv )
815	{
816	UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> ( ( uiFullDepth + 1 ) << 1 );
817	for( UInt uiPart = 0; uiPart < 4; uiPart++ )
818	{
819	xEncSubdivCbfQT( pcCU, uiTrDepth + 1, uiAbsPartIdx + uiPart * uiQPartNum, bLuma, bChroma );
820	}
821	return;
822	}
823
824	{
825	//===== Cbfs =====
826	if( bLuma )
827	{
828	m_pcEntropyCoder->encodeQtCbf( pcCU, uiAbsPartIdx, TEXT_LUMA, uiTrMode );
829	}
830	}
831	}
832
833	Void
834	TEncSearch::xEncCoeffQT( TComDataCU* pcCU,
835	UInt uiTrDepth,
836	UInt uiAbsPartIdx,
837	TextType eTextType,
838	Bool bRealCoeff )
839	{
840	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
841	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
842	UInt uiSubdiv = ( uiTrMode > uiTrDepth ? 1 : 0 );
843	UInt uiLog2TrafoSize = g_aucConvertToBit[pcCU->getSlice()->getSPS()->getMaxCUWidth()] + 2 - uiFullDepth;
844	UInt uiChroma = ( eTextType != TEXT_LUMA ? 1 : 0 );
845
846	if( uiSubdiv )
847	{
848	UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> ( ( uiFullDepth + 1 ) << 1 );
849	for( UInt uiPart = 0; uiPart < 4; uiPart++ )
850	{
851	xEncCoeffQT( pcCU, uiTrDepth + 1, uiAbsPartIdx + uiPart * uiQPartNum, eTextType, bRealCoeff );
852	}
853	return;
854	}
855
856	if( eTextType != TEXT_LUMA && uiLog2TrafoSize == 2 )
857	{
858	assert( uiTrDepth > 0 );
859	uiTrDepth--;
860	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth ) << 1 );
861	Bool bFirstQ = ( ( uiAbsPartIdx % uiQPDiv ) == 0 );
862	if( !bFirstQ )
863	{
864	return;
865	}
866	}
867
868	//===== coefficients =====
869	UInt uiWidth = pcCU->getWidth ( 0 ) >> ( uiTrDepth + uiChroma );
870	UInt uiHeight = pcCU->getHeight ( 0 ) >> ( uiTrDepth + uiChroma );
871	UInt uiCoeffOffset = ( pcCU->getPic()->getMinCUWidth() * pcCU->getPic()->getMinCUHeight() * uiAbsPartIdx ) >> ( uiChroma << 1 );
872	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrafoSize;
873	TCoeff* pcCoeff = 0;
874	switch( eTextType )
875	{
876	case TEXT_LUMA: pcCoeff = ( bRealCoeff ? pcCU->getCoeffY () : m_ppcQTTempCoeffY [uiQTLayer] ); break;
877	case TEXT_CHROMA_U: pcCoeff = ( bRealCoeff ? pcCU->getCoeffCb() : m_ppcQTTempCoeffCb[uiQTLayer] ); break;
878	case TEXT_CHROMA_V: pcCoeff = ( bRealCoeff ? pcCU->getCoeffCr() : m_ppcQTTempCoeffCr[uiQTLayer] ); break;
879	default: assert(0);
880	}
881	pcCoeff += uiCoeffOffset;
882
883	m_pcEntropyCoder->encodeCoeffNxN( pcCU, pcCoeff, uiAbsPartIdx, uiWidth, uiHeight, uiFullDepth, eTextType );
884	}
885
886
887	Void
888	TEncSearch::xEncIntraHeader( TComDataCU* pcCU,
889	UInt uiTrDepth,
890	UInt uiAbsPartIdx,
891	Bool bLuma,
892	Bool bChroma )
893	{
894	if( bLuma )
895	{
896	// CU header
897	if( uiAbsPartIdx == 0 )
898	{
899	if( !pcCU->getSlice()->isIntra() )
900	{
901	if (pcCU->getSlice()->getPPS()->getTransquantBypassEnableFlag())
902	{
903	m_pcEntropyCoder->encodeCUTransquantBypassFlag( pcCU, 0, true );
904	}
905	m_pcEntropyCoder->encodeSkipFlag( pcCU, 0, true );
906	m_pcEntropyCoder->encodePredMode( pcCU, 0, true );
907	}
908
909	m_pcEntropyCoder ->encodePartSize( pcCU, 0, pcCU->getDepth(0), true );
910
911	if (pcCU->isIntra(0) && pcCU->getPartitionSize(0) == SIZE_2Nx2N )
912	{
913	m_pcEntropyCoder->encodeIPCMInfo( pcCU, 0, true );
914
915	if ( pcCU->getIPCMFlag (0))
916	{
917	return;
918	}
919	}
920	}
921	// luma prediction mode
922	if( pcCU->getPartitionSize(0) == SIZE_2Nx2N )
923	{
924	if( uiAbsPartIdx == 0 )
925	{
926	m_pcEntropyCoder->encodeIntraDirModeLuma ( pcCU, 0 );
927	}
928	}
929	else
930	{
931	UInt uiQNumParts = pcCU->getTotalNumPart() >> 2;
932	if( uiTrDepth == 0 )
933	{
934	assert( uiAbsPartIdx == 0 );
935	for( UInt uiPart = 0; uiPart < 4; uiPart++ )
936	{
937	m_pcEntropyCoder->encodeIntraDirModeLuma ( pcCU, uiPart * uiQNumParts );
938	}
939	}
940	else if( ( uiAbsPartIdx % uiQNumParts ) == 0 )
941	{
942	m_pcEntropyCoder->encodeIntraDirModeLuma ( pcCU, uiAbsPartIdx );
943	}
944	}
945	}
946	if( bChroma )
947	{
948	// chroma prediction mode
949	if( uiAbsPartIdx == 0 )
950	{
951	m_pcEntropyCoder->encodeIntraDirModeChroma( pcCU, 0, true );
952	}
953	}
954	}
955
956
957	UInt
958	TEncSearch::xGetIntraBitsQT( TComDataCU* pcCU,
959	UInt uiTrDepth,
960	UInt uiAbsPartIdx,
961	Bool bLuma,
962	Bool bChroma,
963	Bool bRealCoeff /* just for test */ )
964	{
965	m_pcEntropyCoder->resetBits();
966	xEncIntraHeader ( pcCU, uiTrDepth, uiAbsPartIdx, bLuma, bChroma );
967	xEncSubdivCbfQT ( pcCU, uiTrDepth, uiAbsPartIdx, bLuma, bChroma );
968
969	if( bLuma )
970	{
971	xEncCoeffQT ( pcCU, uiTrDepth, uiAbsPartIdx, TEXT_LUMA, bRealCoeff );
972	}
973	if( bChroma )
974	{
975	xEncCoeffQT ( pcCU, uiTrDepth, uiAbsPartIdx, TEXT_CHROMA_U, bRealCoeff );
976	xEncCoeffQT ( pcCU, uiTrDepth, uiAbsPartIdx, TEXT_CHROMA_V, bRealCoeff );
977	}
978	UInt uiBits = m_pcEntropyCoder->getNumberOfWrittenBits();
979	return uiBits;
980	}
981
982	UInt
983	TEncSearch::xGetIntraBitsQTChroma( TComDataCU* pcCU,
984	UInt uiTrDepth,
985	UInt uiAbsPartIdx,
986	UInt uiChromaId,
987	Bool bRealCoeff /* just for test */ )
988	{
989	m_pcEntropyCoder->resetBits();
990	if( uiChromaId == TEXT_CHROMA_U)
991	{
992	xEncCoeffQT ( pcCU, uiTrDepth, uiAbsPartIdx, TEXT_CHROMA_U, bRealCoeff );
993	}
994	else if(uiChromaId == TEXT_CHROMA_V)
995	{
996	xEncCoeffQT ( pcCU, uiTrDepth, uiAbsPartIdx, TEXT_CHROMA_V, bRealCoeff );
997	}
998
999	UInt uiBits = m_pcEntropyCoder->getNumberOfWrittenBits();
1000	return uiBits;
1001	}
1002
1003	Void
1004	TEncSearch::xIntraCodingLumaBlk( TComDataCU* pcCU,
1005	UInt uiTrDepth,
1006	UInt uiAbsPartIdx,
1007	TComYuv* pcOrgYuv,
1008	TComYuv* pcPredYuv,
1009	TComYuv* pcResiYuv,
1010	#if H_3D_VSO
1011	Dist& ruiDist,
1012	#else
1013	UInt& ruiDist,
1014	#endif
1015	Int default0Save1Load2 )
1016	{
1017	UInt uiLumaPredMode = pcCU ->getLumaIntraDir ( uiAbsPartIdx );
1018	UInt uiFullDepth = pcCU ->getDepth ( 0 ) + uiTrDepth;
1019	UInt uiWidth = pcCU ->getWidth ( 0 ) >> uiTrDepth;
1020	UInt uiHeight = pcCU ->getHeight ( 0 ) >> uiTrDepth;
1021	UInt uiStride = pcOrgYuv ->getStride ();
1022	Pel* piOrg = pcOrgYuv ->getLumaAddr( uiAbsPartIdx );
1023	Pel* piPred = pcPredYuv->getLumaAddr( uiAbsPartIdx );
1024	Pel* piResi = pcResiYuv->getLumaAddr( uiAbsPartIdx );
1025	Pel* piReco = pcPredYuv->getLumaAddr( uiAbsPartIdx );
1026
1027	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
1028	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1029	UInt uiNumCoeffPerInc = pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() >> ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 );
1030	TCoeff* pcCoeff = m_ppcQTTempCoeffY[ uiQTLayer ] + uiNumCoeffPerInc * uiAbsPartIdx;
1031	#if ADAPTIVE_QP_SELECTION
1032	Int* pcArlCoeff = m_ppcQTTempArlCoeffY[ uiQTLayer ] + uiNumCoeffPerInc * uiAbsPartIdx;
1033	#endif
1034	Pel* piRecQt = m_pcQTTempTComYuv[ uiQTLayer ].getLumaAddr( uiAbsPartIdx );
1035	UInt uiRecQtStride = m_pcQTTempTComYuv[ uiQTLayer ].getStride ();
1036
1037	UInt uiZOrder = pcCU->getZorderIdxInCU() + uiAbsPartIdx;
1038	Pel* piRecIPred = pcCU->getPic()->getPicYuvRec()->getLumaAddr( pcCU->getAddr(), uiZOrder );
1039	UInt uiRecIPredStride = pcCU->getPic()->getPicYuvRec()->getStride ();
1040	Bool useTransformSkip = pcCU->getTransformSkip(uiAbsPartIdx, TEXT_LUMA);
1041	//===== init availability pattern =====
1042	Bool bAboveAvail = false;
1043	Bool bLeftAvail = false;
1044	if(default0Save1Load2 != 2)
1045	{
1046	pcCU->getPattern()->initPattern ( pcCU, uiTrDepth, uiAbsPartIdx );
1047	pcCU->getPattern()->initAdiPattern( pcCU, uiAbsPartIdx, uiTrDepth, m_piYuvExt, m_iYuvExtStride, m_iYuvExtHeight, bAboveAvail, bLeftAvail );
1048	//===== get prediction signal =====
1049	predIntraLumaAng( pcCU->getPattern(), uiLumaPredMode, piPred, uiStride, uiWidth, uiHeight, bAboveAvail, bLeftAvail );
1050	// save prediction
1051	if(default0Save1Load2 == 1)
1052	{
1053	Pel* pPred = piPred;
1054	Pel* pPredBuf = m_pSharedPredTransformSkip[0];
1055	Int k = 0;
1056	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1057	{
1058	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1059	{
1060	pPredBuf[ k ++ ] = pPred[ uiX ];
1061	}
1062	pPred += uiStride;
1063	}
1064	}
1065	}
1066	else
1067	{
1068	// load prediction
1069	Pel* pPred = piPred;
1070	Pel* pPredBuf = m_pSharedPredTransformSkip[0];
1071	Int k = 0;
1072	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1073	{
1074	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1075	{
1076	pPred[ uiX ] = pPredBuf[ k ++ ];
1077	}
1078	pPred += uiStride;
1079	}
1080	}
1081	//===== get residual signal =====
1082	{
1083	// get residual
1084	Pel* pOrg = piOrg;
1085	Pel* pPred = piPred;
1086	Pel* pResi = piResi;
1087	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1088	{
1089	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1090	{
1091	pResi[ uiX ] = pOrg[ uiX ] - pPred[ uiX ];
1092	}
1093	pOrg += uiStride;
1094	pResi += uiStride;
1095	pPred += uiStride;
1096	}
1097	}
1098
1099	//===== transform and quantization =====
1100	//--- init rate estimation arrays for RDOQ ---
1101	if( useTransformSkip? m_pcEncCfg->getUseRDOQTS():m_pcEncCfg->getUseRDOQ())
1102	{
1103	m_pcEntropyCoder->estimateBit( m_pcTrQuant->m_pcEstBitsSbac, uiWidth, uiWidth, TEXT_LUMA );
1104	}
1105	//--- transform and quantization ---
1106	UInt uiAbsSum = 0;
1107	pcCU ->setTrIdxSubParts ( uiTrDepth, uiAbsPartIdx, uiFullDepth );
1108
1109	m_pcTrQuant->setQPforQuant ( pcCU->getQP( 0 ), TEXT_LUMA, pcCU->getSlice()->getSPS()->getQpBDOffsetY(), 0 );
1110
1111	#if RDOQ_CHROMA_LAMBDA
1112	m_pcTrQuant->selectLambda (TEXT_LUMA);
1113	#endif
1114
1115	m_pcTrQuant->transformNxN ( pcCU, piResi, uiStride, pcCoeff,
1116	#if ADAPTIVE_QP_SELECTION
1117	pcArlCoeff,
1118	#endif
1119	uiWidth, uiHeight, uiAbsSum, TEXT_LUMA, uiAbsPartIdx,useTransformSkip );
1120
1121	//--- set coded block flag ---
1122	pcCU->setCbfSubParts ( ( uiAbsSum ? 1 : 0 ) << uiTrDepth, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1123	//--- inverse transform ---
1124	if( uiAbsSum )
1125	{
1126	Int scalingListType = 0 + g_eTTable[(Int)TEXT_LUMA];
1127	assert(scalingListType < 6);
1128	m_pcTrQuant->invtransformNxN( pcCU->getCUTransquantBypass(uiAbsPartIdx), TEXT_LUMA,pcCU->getLumaIntraDir( uiAbsPartIdx ), piResi, uiStride, pcCoeff, uiWidth, uiHeight, scalingListType, useTransformSkip );
1129	}
1130	else
1131	{
1132	Pel* pResi = piResi;
1133	memset( pcCoeff, 0, sizeof( TCoeff ) * uiWidth * uiHeight );
1134	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1135	{
1136	memset( pResi, 0, sizeof( Pel ) * uiWidth );
1137	pResi += uiStride;
1138	}
1139	}
1140
1141	//===== reconstruction =====
1142	{
1143	Pel* pPred = piPred;
1144	Pel* pResi = piResi;
1145	Pel* pReco = piReco;
1146	Pel* pRecQt = piRecQt;
1147	Pel* pRecIPred = piRecIPred;
1148	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1149	{
1150	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1151	{
1152	pReco [ uiX ] = ClipY( pPred[ uiX ] + pResi[ uiX ] );
1153	pRecQt [ uiX ] = pReco[ uiX ];
1154	pRecIPred[ uiX ] = pReco[ uiX ];
1155	}
1156	pPred += uiStride;
1157	pResi += uiStride;
1158	pReco += uiStride;
1159	pRecQt += uiRecQtStride;
1160	pRecIPred += uiRecIPredStride;
1161	}
1162	}
1163
1164	//===== update distortion =====
1165	#if H_3D_VSO // M39
1166	if ( m_pcRdCost->getUseVSO() )
1167	ruiDist += m_pcRdCost->getDistPartVSO ( pcCU, uiAbsPartIdx, piReco, uiStride, piOrg, uiStride, uiWidth, uiHeight, false );
1168	else
1169	#endif
1170	ruiDist += m_pcRdCost->getDistPart(g_bitDepthY, piReco, uiStride, piOrg, uiStride, uiWidth, uiHeight );
1171	}
1172
1173	Void
1174	TEncSearch::xIntraCodingChromaBlk( TComDataCU* pcCU,
1175	UInt uiTrDepth,
1176	UInt uiAbsPartIdx,
1177	TComYuv* pcOrgYuv,
1178	TComYuv* pcPredYuv,
1179	TComYuv* pcResiYuv,
1180	UInt& ruiDist,
1181	UInt uiChromaId,
1182	Int default0Save1Load2 )
1183	{
1184	UInt uiOrgTrDepth = uiTrDepth;
1185	UInt uiFullDepth = pcCU->getDepth( 0 ) + uiTrDepth;
1186	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
1187	if( uiLog2TrSize == 2 )
1188	{
1189	assert( uiTrDepth > 0 );
1190	uiTrDepth--;
1191	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth ) << 1 );
1192	Bool bFirstQ = ( ( uiAbsPartIdx % uiQPDiv ) == 0 );
1193	if( !bFirstQ )
1194	{
1195	return;
1196	}
1197	}
1198
1199	TextType eText = ( uiChromaId > 0 ? TEXT_CHROMA_V : TEXT_CHROMA_U );
1200	UInt uiChromaPredMode = pcCU ->getChromaIntraDir( uiAbsPartIdx );
1201	UInt uiWidth = pcCU ->getWidth ( 0 ) >> ( uiTrDepth + 1 );
1202	UInt uiHeight = pcCU ->getHeight ( 0 ) >> ( uiTrDepth + 1 );
1203	UInt uiStride = pcOrgYuv ->getCStride ();
1204	Pel* piOrg = ( uiChromaId > 0 ? pcOrgYuv ->getCrAddr( uiAbsPartIdx ) : pcOrgYuv ->getCbAddr( uiAbsPartIdx ) );
1205	Pel* piPred = ( uiChromaId > 0 ? pcPredYuv->getCrAddr( uiAbsPartIdx ) : pcPredYuv->getCbAddr( uiAbsPartIdx ) );
1206	Pel* piResi = ( uiChromaId > 0 ? pcResiYuv->getCrAddr( uiAbsPartIdx ) : pcResiYuv->getCbAddr( uiAbsPartIdx ) );
1207	Pel* piReco = ( uiChromaId > 0 ? pcPredYuv->getCrAddr( uiAbsPartIdx ) : pcPredYuv->getCbAddr( uiAbsPartIdx ) );
1208
1209	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1210	UInt uiNumCoeffPerInc = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() >> ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 ) ) >> 2;
1211	TCoeff* pcCoeff = ( uiChromaId > 0 ? m_ppcQTTempCoeffCr[ uiQTLayer ] : m_ppcQTTempCoeffCb[ uiQTLayer ] ) + uiNumCoeffPerInc * uiAbsPartIdx;
1212	#if ADAPTIVE_QP_SELECTION
1213	Int* pcArlCoeff = ( uiChromaId > 0 ? m_ppcQTTempArlCoeffCr[ uiQTLayer ] : m_ppcQTTempArlCoeffCb[ uiQTLayer ] ) + uiNumCoeffPerInc * uiAbsPartIdx;
1214	#endif
1215	Pel* piRecQt = ( uiChromaId > 0 ? m_pcQTTempTComYuv[ uiQTLayer ].getCrAddr( uiAbsPartIdx ) : m_pcQTTempTComYuv[ uiQTLayer ].getCbAddr( uiAbsPartIdx ) );
1216	UInt uiRecQtStride = m_pcQTTempTComYuv[ uiQTLayer ].getCStride();
1217
1218	UInt uiZOrder = pcCU->getZorderIdxInCU() + uiAbsPartIdx;
1219	Pel* piRecIPred = ( uiChromaId > 0 ? pcCU->getPic()->getPicYuvRec()->getCrAddr( pcCU->getAddr(), uiZOrder ) : pcCU->getPic()->getPicYuvRec()->getCbAddr( pcCU->getAddr(), uiZOrder ) );
1220	UInt uiRecIPredStride = pcCU->getPic()->getPicYuvRec()->getCStride();
1221	Bool useTransformSkipChroma = pcCU->getTransformSkip(uiAbsPartIdx, eText);
1222	//===== update chroma mode =====
1223	if( uiChromaPredMode == DM_CHROMA_IDX )
1224	{
1225	uiChromaPredMode = pcCU->getLumaIntraDir( 0 );
1226	}
1227
1228	//===== init availability pattern =====
1229	Bool bAboveAvail = false;
1230	Bool bLeftAvail = false;
1231	if( default0Save1Load2 != 2 )
1232	{
1233	pcCU->getPattern()->initPattern ( pcCU, uiTrDepth, uiAbsPartIdx );
1234
1235	pcCU->getPattern()->initAdiPatternChroma( pcCU, uiAbsPartIdx, uiTrDepth, m_piYuvExt, m_iYuvExtStride, m_iYuvExtHeight, bAboveAvail, bLeftAvail );
1236	Int* pPatChroma = ( uiChromaId > 0 ? pcCU->getPattern()->getAdiCrBuf( uiWidth, uiHeight, m_piYuvExt ) : pcCU->getPattern()->getAdiCbBuf( uiWidth, uiHeight, m_piYuvExt ) );
1237
1238	//===== get prediction signal =====
1239	{
1240	predIntraChromaAng( pPatChroma, uiChromaPredMode, piPred, uiStride, uiWidth, uiHeight, bAboveAvail, bLeftAvail );
1241	}
1242	// save prediction
1243	if( default0Save1Load2 == 1 )
1244	{
1245	Pel* pPred = piPred;
1246	Pel* pPredBuf = m_pSharedPredTransformSkip[1 + uiChromaId];
1247	Int k = 0;
1248	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1249	{
1250	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1251	{
1252	pPredBuf[ k ++ ] = pPred[ uiX ];
1253	}
1254	pPred += uiStride;
1255	}
1256	}
1257	}
1258	else
1259	{
1260	// load prediction
1261	Pel* pPred = piPred;
1262	Pel* pPredBuf = m_pSharedPredTransformSkip[1 + uiChromaId];
1263	Int k = 0;
1264	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1265	{
1266	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1267	{
1268	pPred[ uiX ] = pPredBuf[ k ++ ];
1269	}
1270	pPred += uiStride;
1271	}
1272	}
1273	//===== get residual signal =====
1274	{
1275	// get residual
1276	Pel* pOrg = piOrg;
1277	Pel* pPred = piPred;
1278	Pel* pResi = piResi;
1279	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1280	{
1281	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1282	{
1283	pResi[ uiX ] = pOrg[ uiX ] - pPred[ uiX ];
1284	}
1285	pOrg += uiStride;
1286	pResi += uiStride;
1287	pPred += uiStride;
1288	}
1289	}
1290
1291	//===== transform and quantization =====
1292	{
1293	//--- init rate estimation arrays for RDOQ ---
1294	if( useTransformSkipChroma? m_pcEncCfg->getUseRDOQTS():m_pcEncCfg->getUseRDOQ())
1295	{
1296	m_pcEntropyCoder->estimateBit( m_pcTrQuant->m_pcEstBitsSbac, uiWidth, uiWidth, eText );
1297	}
1298	//--- transform and quantization ---
1299	UInt uiAbsSum = 0;
1300
1301	Int curChromaQpOffset;
1302	if(eText == TEXT_CHROMA_U)
1303	{
1304	curChromaQpOffset = pcCU->getSlice()->getPPS()->getChromaCbQpOffset() + pcCU->getSlice()->getSliceQpDeltaCb();
1305	}
1306	else
1307	{
1308	curChromaQpOffset = pcCU->getSlice()->getPPS()->getChromaCrQpOffset() + pcCU->getSlice()->getSliceQpDeltaCr();
1309	}
1310	m_pcTrQuant->setQPforQuant ( pcCU->getQP( 0 ), TEXT_CHROMA, pcCU->getSlice()->getSPS()->getQpBDOffsetC(), curChromaQpOffset );
1311
1312	#if RDOQ_CHROMA_LAMBDA
1313	m_pcTrQuant->selectLambda (TEXT_CHROMA);
1314	#endif
1315	m_pcTrQuant->transformNxN ( pcCU, piResi, uiStride, pcCoeff,
1316	#if ADAPTIVE_QP_SELECTION
1317	pcArlCoeff,
1318	#endif
1319	uiWidth, uiHeight, uiAbsSum, eText, uiAbsPartIdx, useTransformSkipChroma );
1320	//--- set coded block flag ---
1321	pcCU->setCbfSubParts ( ( uiAbsSum ? 1 : 0 ) << uiOrgTrDepth, eText, uiAbsPartIdx, pcCU->getDepth(0) + uiTrDepth );
1322	//--- inverse transform ---
1323	if( uiAbsSum )
1324	{
1325	Int scalingListType = 0 + g_eTTable[(Int)eText];
1326	assert(scalingListType < 6);
1327	m_pcTrQuant->invtransformNxN( pcCU->getCUTransquantBypass(uiAbsPartIdx), TEXT_CHROMA, REG_DCT, piResi, uiStride, pcCoeff, uiWidth, uiHeight, scalingListType, useTransformSkipChroma );
1328	}
1329	else
1330	{
1331	Pel* pResi = piResi;
1332	memset( pcCoeff, 0, sizeof( TCoeff ) * uiWidth * uiHeight );
1333	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1334	{
1335	memset( pResi, 0, sizeof( Pel ) * uiWidth );
1336	pResi += uiStride;
1337	}
1338	}
1339	}
1340
1341	//===== reconstruction =====
1342	{
1343	Pel* pPred = piPred;
1344	Pel* pResi = piResi;
1345	Pel* pReco = piReco;
1346	Pel* pRecQt = piRecQt;
1347	Pel* pRecIPred = piRecIPred;
1348	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
1349	{
1350	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1351	{
1352	pReco [ uiX ] = ClipC( pPred[ uiX ] + pResi[ uiX ] );
1353	pRecQt [ uiX ] = pReco[ uiX ];
1354	pRecIPred[ uiX ] = pReco[ uiX ];
1355	}
1356	pPred += uiStride;
1357	pResi += uiStride;
1358	pReco += uiStride;
1359	pRecQt += uiRecQtStride;
1360	pRecIPred += uiRecIPredStride;
1361	}
1362	}
1363
1364	//===== update distortion =====
1365	#if WEIGHTED_CHROMA_DISTORTION
1366	ruiDist += m_pcRdCost->getDistPart(g_bitDepthC, piReco, uiStride, piOrg, uiStride, uiWidth, uiHeight, eText );
1367	#else
1368	ruiDist += m_pcRdCost->getDistPart(g_bitDepthC, piReco, uiStride, piOrg, uiStride, uiWidth, uiHeight );
1369	#endif
1370	}
1371
1372
1373
1374	Void
1375	TEncSearch::xRecurIntraCodingQT( TComDataCU* pcCU,
1376	UInt uiTrDepth,
1377	UInt uiAbsPartIdx,
1378	Bool bLumaOnly,
1379	TComYuv* pcOrgYuv,
1380	TComYuv* pcPredYuv,
1381	TComYuv* pcResiYuv,
1382	#if H_3D_VSO
1383	Dist& ruiDistY,
1384	#else
1385	UInt& ruiDistY,
1386	#endif
1387	UInt& ruiDistC,
1388	#if HHI_RQT_INTRA_SPEEDUP
1389	Bool bCheckFirst,
1390	#endif
1391	Double& dRDCost )
1392	{
1393	UInt uiFullDepth = pcCU->getDepth( 0 ) + uiTrDepth;
1394	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
1395	Bool bCheckFull = ( uiLog2TrSize <= pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() );
1396	Bool bCheckSplit = ( uiLog2TrSize > pcCU->getQuadtreeTULog2MinSizeInCU(uiAbsPartIdx) );
1397
1398	#if HHI_RQT_INTRA_SPEEDUP
1399	#if L0232_RD_PENALTY
1400	Int maxTuSize = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize();
1401	Int isIntraSlice = (pcCU->getSlice()->getSliceType() == I_SLICE);
1402	// don't check split if TU size is less or equal to max TU size
1403	Bool noSplitIntraMaxTuSize = bCheckFull;
1404	if(m_pcEncCfg->getRDpenalty() && ! isIntraSlice)
1405	{
1406	// in addition don't check split if TU size is less or equal to 16x16 TU size for non-intra slice
1407	noSplitIntraMaxTuSize = ( uiLog2TrSize <= min(maxTuSize,4) );
1408
1409	// if maximum RD-penalty don't check TU size 32x32
1410	if(m_pcEncCfg->getRDpenalty()==2)
1411	{
1412	bCheckFull = ( uiLog2TrSize <= min(maxTuSize,4));
1413	}
1414	}
1415	if( bCheckFirst && noSplitIntraMaxTuSize )
1416	#else
1417	if( bCheckFirst && bCheckFull )
1418	#endif
1419	{
1420	bCheckSplit = false;
1421	}
1422	#else
1423	#if L0232_RD_PENALTY
1424	Int maxTuSize = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize();
1425	Int isIntraSlice = (pcCU->getSlice()->getSliceType() == I_SLICE);
1426	// if maximum RD-penalty don't check TU size 32x32
1427	if((m_pcEncCfg->getRDpenalty()==2) && !isIntraSlice)
1428	{
1429	bCheckFull = ( uiLog2TrSize <= min(maxTuSize,4));
1430	}
1431	#endif
1432	#endif
1433	Double dSingleCost = MAX_DOUBLE;
1434	#if H_3D_VSO
1435	Dist uiSingleDistY = 0;
1436	#else
1437	UInt uiSingleDistY = 0;
1438	#endif
1439	UInt uiSingleDistC = 0;
1440	UInt uiSingleCbfY = 0;
1441	UInt uiSingleCbfU = 0;
1442	UInt uiSingleCbfV = 0;
1443	Bool checkTransformSkip = pcCU->getSlice()->getPPS()->getUseTransformSkip();
1444	UInt widthTransformSkip = pcCU->getWidth ( 0 ) >> uiTrDepth;
1445	UInt heightTransformSkip = pcCU->getHeight( 0 ) >> uiTrDepth;
1446	Int bestModeId = 0;
1447	Int bestModeIdUV[2] = {0, 0};
1448	checkTransformSkip &= (widthTransformSkip == 4 && heightTransformSkip == 4);
1449	checkTransformSkip &= (!pcCU->getCUTransquantBypass(0));
1450	checkTransformSkip &= (!((pcCU->getQP( 0 ) == 0) && (pcCU->getSlice()->getSPS()->getUseLossless())));
1451	if ( m_pcEncCfg->getUseTransformSkipFast() )
1452	{
1453	checkTransformSkip &= (pcCU->getPartitionSize(uiAbsPartIdx)==SIZE_NxN);
1454	}
1455	if( bCheckFull )
1456	{
1457	if(checkTransformSkip == true)
1458	{
1459	//----- store original entropy coding status -----
1460	if( m_bUseSBACRD)
1461	{
1462	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
1463	}
1464	#if H_3D_VSO
1465	Dist singleDistYTmp = 0;
1466	#else
1467	UInt singleDistYTmp = 0;
1468	#endif
1469	UInt singleDistCTmp = 0;
1470	UInt singleCbfYTmp = 0;
1471	UInt singleCbfUTmp = 0;
1472	UInt singleCbfVTmp = 0;
1473	Double singleCostTmp = 0;
1474	Int default0Save1Load2 = 0;
1475	Int firstCheckId = 0;
1476
1477	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + (uiTrDepth - 1) ) << 1 );
1478	Bool bFirstQ = ( ( uiAbsPartIdx % uiQPDiv ) == 0 );
1479
1480	for(Int modeId = firstCheckId; modeId < 2; modeId ++)
1481	{
1482	singleDistYTmp = 0;
1483	singleDistCTmp = 0;
1484	pcCU ->setTransformSkipSubParts ( modeId, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1485	if(modeId == firstCheckId)
1486	{
1487	default0Save1Load2 = 1;
1488	}
1489	else
1490	{
1491	default0Save1Load2 = 2;
1492	}
1493	//----- code luma block with given intra prediction mode and store Cbf-----
1494	xIntraCodingLumaBlk( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, singleDistYTmp,default0Save1Load2);
1495	singleCbfYTmp = pcCU->getCbf( uiAbsPartIdx, TEXT_LUMA, uiTrDepth );
1496	//----- code chroma blocks with given intra prediction mode and store Cbf-----
1497	if( !bLumaOnly )
1498	{
1499	if(bFirstQ)
1500	{
1501	pcCU ->setTransformSkipSubParts ( modeId, TEXT_CHROMA_U, uiAbsPartIdx, uiFullDepth);
1502	pcCU ->setTransformSkipSubParts ( modeId, TEXT_CHROMA_V, uiAbsPartIdx, uiFullDepth);
1503	}
1504	xIntraCodingChromaBlk ( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, singleDistCTmp, 0, default0Save1Load2);
1505	xIntraCodingChromaBlk ( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, singleDistCTmp, 1, default0Save1Load2);
1506	singleCbfUTmp = pcCU->getCbf( uiAbsPartIdx, TEXT_CHROMA_U, uiTrDepth );
1507	singleCbfVTmp = pcCU->getCbf( uiAbsPartIdx, TEXT_CHROMA_V, uiTrDepth );
1508	}
1509	//----- determine rate and r-d cost -----
1510	if(modeId == 1 && singleCbfYTmp == 0)
1511	{
1512	//In order not to code TS flag when cbf is zero, the case for TS with cbf being zero is forbidden.
1513	singleCostTmp = MAX_DOUBLE;
1514	}
1515	else
1516	{
1517	UInt uiSingleBits = xGetIntraBitsQT( pcCU, uiTrDepth, uiAbsPartIdx, true, !bLumaOnly, false );
1518	#if H_3D_VSO // M NEW
1519	if ( m_pcRdCost->getUseRenModel() )
1520	singleCostTmp = m_pcRdCost->calcRdCostVSO( uiSingleBits, singleDistYTmp + singleDistCTmp );
1521	else
1522	#endif
1523	singleCostTmp = m_pcRdCost->calcRdCost( uiSingleBits, singleDistYTmp + singleDistCTmp );
1524	}
1525
1526	if(singleCostTmp < dSingleCost)
1527	{
1528	dSingleCost = singleCostTmp;
1529	uiSingleDistY = singleDistYTmp;
1530	uiSingleDistC = singleDistCTmp;
1531	uiSingleCbfY = singleCbfYTmp;
1532	uiSingleCbfU = singleCbfUTmp;
1533	uiSingleCbfV = singleCbfVTmp;
1534	bestModeId = modeId;
1535	if(bestModeId == firstCheckId)
1536	{
1537	xStoreIntraResultQT(pcCU, uiTrDepth, uiAbsPartIdx,bLumaOnly );
1538	if( m_bUseSBACRD)
1539	{
1540	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ uiFullDepth ][ CI_TEMP_BEST ] );
1541	}
1542	}
1543	}
1544	if(modeId == firstCheckId)
1545	{
1546	m_pcRDGoOnSbacCoder->load ( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
1547	}
1548	}
1549
1550	pcCU ->setTransformSkipSubParts ( bestModeId, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1551
1552	if(bestModeId == firstCheckId)
1553	{
1554	xLoadIntraResultQT(pcCU, uiTrDepth, uiAbsPartIdx,bLumaOnly );
1555	pcCU->setCbfSubParts ( uiSingleCbfY << uiTrDepth, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1556	if( !bLumaOnly )
1557	{
1558	if(bFirstQ)
1559	{
1560	pcCU->setCbfSubParts( uiSingleCbfU << uiTrDepth, TEXT_CHROMA_U, uiAbsPartIdx, pcCU->getDepth( 0 ) + uiTrDepth - 1 );
1561	pcCU->setCbfSubParts( uiSingleCbfV << uiTrDepth, TEXT_CHROMA_V, uiAbsPartIdx, pcCU->getDepth( 0 ) + uiTrDepth - 1 );
1562	}
1563	}
1564	if(m_bUseSBACRD)
1565	{
1566	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[ uiFullDepth ][ CI_TEMP_BEST ] );
1567	}
1568	}
1569
1570	if( !bLumaOnly )
1571	{
1572	bestModeIdUV[0] = bestModeIdUV[1] = bestModeId;
1573	if(bFirstQ && bestModeId == 1)
1574	{
1575	//In order not to code TS flag when cbf is zero, the case for TS with cbf being zero is forbidden.
1576	if(uiSingleCbfU == 0)
1577	{
1578	pcCU ->setTransformSkipSubParts ( 0, TEXT_CHROMA_U, uiAbsPartIdx, uiFullDepth);
1579	bestModeIdUV[0] = 0;
1580	}
1581	if(uiSingleCbfV == 0)
1582	{
1583	pcCU ->setTransformSkipSubParts ( 0, TEXT_CHROMA_V, uiAbsPartIdx, uiFullDepth);
1584	bestModeIdUV[1] = 0;
1585	}
1586	}
1587	}
1588	}
1589	else
1590	{
1591	pcCU ->setTransformSkipSubParts ( 0, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1592	//----- store original entropy coding status -----
1593	if( m_bUseSBACRD && bCheckSplit )
1594	{
1595	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
1596	}
1597	//----- code luma block with given intra prediction mode and store Cbf-----
1598	dSingleCost = 0.0;
1599	xIntraCodingLumaBlk( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, uiSingleDistY );
1600	if( bCheckSplit )
1601	{
1602	uiSingleCbfY = pcCU->getCbf( uiAbsPartIdx, TEXT_LUMA, uiTrDepth );
1603	}
1604	//----- code chroma blocks with given intra prediction mode and store Cbf-----
1605	if( !bLumaOnly )
1606	{
1607	pcCU ->setTransformSkipSubParts ( 0, TEXT_CHROMA_U, uiAbsPartIdx, uiFullDepth );
1608	pcCU ->setTransformSkipSubParts ( 0, TEXT_CHROMA_V, uiAbsPartIdx, uiFullDepth );
1609	xIntraCodingChromaBlk ( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, uiSingleDistC, 0 );
1610	xIntraCodingChromaBlk ( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, uiSingleDistC, 1 );
1611	if( bCheckSplit )
1612	{
1613	uiSingleCbfU = pcCU->getCbf( uiAbsPartIdx, TEXT_CHROMA_U, uiTrDepth );
1614	uiSingleCbfV = pcCU->getCbf( uiAbsPartIdx, TEXT_CHROMA_V, uiTrDepth );
1615	}
1616	}
1617	//----- determine rate and r-d cost -----
1618	UInt uiSingleBits = xGetIntraBitsQT( pcCU, uiTrDepth, uiAbsPartIdx, true, !bLumaOnly, false );
1619	#if L0232_RD_PENALTY
1620	if(m_pcEncCfg->getRDpenalty() && (uiLog2TrSize==5) && !isIntraSlice)
1621	{
1622	uiSingleBits=uiSingleBits*4;
1623	}
1624	#endif
1625	#if H_3D_VSO // M40
1626	if ( m_pcRdCost->getUseLambdaScaleVSO())
1627	dSingleCost = m_pcRdCost->calcRdCostVSO( uiSingleBits, uiSingleDistY + uiSingleDistC );
1628	else
1629	#endif
1630	dSingleCost = m_pcRdCost->calcRdCost( uiSingleBits, uiSingleDistY + uiSingleDistC );
1631	}
1632	}
1633
1634	if( bCheckSplit )
1635	{
1636	//----- store full entropy coding status, load original entropy coding status -----
1637	if( m_bUseSBACRD )
1638	{
1639	if( bCheckFull )
1640	{
1641	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_TEST ] );
1642	m_pcRDGoOnSbacCoder->load ( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
1643	}
1644	else
1645	{
1646	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
1647	}
1648	}
1649	//----- code splitted block -----
1650	Double dSplitCost = 0.0;
1651	#if H_3D_VSO
1652	Dist uiSplitDistY = 0;
1653	#else
1654	UInt uiSplitDistY = 0;
1655	#endif
1656	UInt uiSplitDistC = 0;
1657	UInt uiQPartsDiv = pcCU->getPic()->getNumPartInCU() >> ( ( uiFullDepth + 1 ) << 1 );
1658	UInt uiAbsPartIdxSub = uiAbsPartIdx;
1659
1660	UInt uiSplitCbfY = 0;
1661	UInt uiSplitCbfU = 0;
1662	UInt uiSplitCbfV = 0;
1663
1664	for( UInt uiPart = 0; uiPart < 4; uiPart++, uiAbsPartIdxSub += uiQPartsDiv )
1665	{
1666	#if HHI_RQT_INTRA_SPEEDUP
1667	xRecurIntraCodingQT( pcCU, uiTrDepth + 1, uiAbsPartIdxSub, bLumaOnly, pcOrgYuv, pcPredYuv, pcResiYuv, uiSplitDistY, uiSplitDistC, bCheckFirst, dSplitCost );
1668	#else
1669	xRecurIntraCodingQT( pcCU, uiTrDepth + 1, uiAbsPartIdxSub, bLumaOnly, pcOrgYuv, pcPredYuv, pcResiYuv, uiSplitDistY, uiSplitDistC, dSplitCost );
1670	#endif
1671
1672	uiSplitCbfY \|= pcCU->getCbf( uiAbsPartIdxSub, TEXT_LUMA, uiTrDepth + 1 );
1673	if(!bLumaOnly)
1674	{
1675	uiSplitCbfU \|= pcCU->getCbf( uiAbsPartIdxSub, TEXT_CHROMA_U, uiTrDepth + 1 );
1676	uiSplitCbfV \|= pcCU->getCbf( uiAbsPartIdxSub, TEXT_CHROMA_V, uiTrDepth + 1 );
1677	}
1678	}
1679
1680	for( UInt uiOffs = 0; uiOffs < 4 * uiQPartsDiv; uiOffs++ )
1681	{
1682	pcCU->getCbf( TEXT_LUMA )[ uiAbsPartIdx + uiOffs ] \|= ( uiSplitCbfY << uiTrDepth );
1683	}
1684	if( !bLumaOnly )
1685	{
1686	for( UInt uiOffs = 0; uiOffs < 4 * uiQPartsDiv; uiOffs++ )
1687	{
1688	pcCU->getCbf( TEXT_CHROMA_U )[ uiAbsPartIdx + uiOffs ] \|= ( uiSplitCbfU << uiTrDepth );
1689	pcCU->getCbf( TEXT_CHROMA_V )[ uiAbsPartIdx + uiOffs ] \|= ( uiSplitCbfV << uiTrDepth );
1690	}
1691	}
1692	//----- restore context states -----
1693	if( m_bUseSBACRD )
1694	{
1695	m_pcRDGoOnSbacCoder->load ( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
1696	}
1697	//----- determine rate and r-d cost -----
1698	UInt uiSplitBits = xGetIntraBitsQT( pcCU, uiTrDepth, uiAbsPartIdx, true, !bLumaOnly, false );
1699	#if H_3D_VSO // M41
1700	if( m_pcRdCost->getUseLambdaScaleVSO() )
1701	dSplitCost = m_pcRdCost->calcRdCostVSO( uiSplitBits, uiSplitDistY + uiSplitDistC );
1702	else
1703	#endif
1704	dSplitCost = m_pcRdCost->calcRdCost( uiSplitBits, uiSplitDistY + uiSplitDistC );
1705
1706	//===== compare and set best =====
1707	if( dSplitCost < dSingleCost )
1708	{
1709	//--- update cost ---
1710	ruiDistY += uiSplitDistY;
1711	ruiDistC += uiSplitDistC;
1712	dRDCost += dSplitCost;
1713	return;
1714	}
1715	//----- set entropy coding status -----
1716	if( m_bUseSBACRD )
1717	{
1718	m_pcRDGoOnSbacCoder->load ( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_TEST ] );
1719	}
1720
1721	//--- set transform index and Cbf values ---
1722	pcCU->setTrIdxSubParts( uiTrDepth, uiAbsPartIdx, uiFullDepth );
1723	pcCU->setCbfSubParts ( uiSingleCbfY << uiTrDepth, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1724	pcCU ->setTransformSkipSubParts ( bestModeId, TEXT_LUMA, uiAbsPartIdx, uiFullDepth );
1725	if( !bLumaOnly )
1726	{
1727	pcCU->setCbfSubParts( uiSingleCbfU << uiTrDepth, TEXT_CHROMA_U, uiAbsPartIdx, uiFullDepth );
1728	pcCU->setCbfSubParts( uiSingleCbfV << uiTrDepth, TEXT_CHROMA_V, uiAbsPartIdx, uiFullDepth );
1729	pcCU->setTransformSkipSubParts ( bestModeIdUV[0], TEXT_CHROMA_U, uiAbsPartIdx, uiFullDepth);
1730	pcCU->setTransformSkipSubParts ( bestModeIdUV[1], TEXT_CHROMA_V, uiAbsPartIdx, uiFullDepth);
1731	}
1732
1733	//--- set reconstruction for next intra prediction blocks ---
1734	UInt uiWidth = pcCU->getWidth ( 0 ) >> uiTrDepth;
1735	UInt uiHeight = pcCU->getHeight( 0 ) >> uiTrDepth;
1736	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1737	UInt uiZOrder = pcCU->getZorderIdxInCU() + uiAbsPartIdx;
1738	Pel* piSrc = m_pcQTTempTComYuv[ uiQTLayer ].getLumaAddr( uiAbsPartIdx );
1739	UInt uiSrcStride = m_pcQTTempTComYuv[ uiQTLayer ].getStride ();
1740	Pel* piDes = pcCU->getPic()->getPicYuvRec()->getLumaAddr( pcCU->getAddr(), uiZOrder );
1741	UInt uiDesStride = pcCU->getPic()->getPicYuvRec()->getStride ();
1742	for( UInt uiY = 0; uiY < uiHeight; uiY++, piSrc += uiSrcStride, piDes += uiDesStride )
1743	{
1744	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1745	{
1746	piDes[ uiX ] = piSrc[ uiX ];
1747	}
1748	}
1749	if( !bLumaOnly )
1750	{
1751	uiWidth >>= 1;
1752	uiHeight >>= 1;
1753	piSrc = m_pcQTTempTComYuv[ uiQTLayer ].getCbAddr ( uiAbsPartIdx );
1754	uiSrcStride = m_pcQTTempTComYuv[ uiQTLayer ].getCStride ();
1755	piDes = pcCU->getPic()->getPicYuvRec()->getCbAddr ( pcCU->getAddr(), uiZOrder );
1756	uiDesStride = pcCU->getPic()->getPicYuvRec()->getCStride();
1757	for( UInt uiY = 0; uiY < uiHeight; uiY++, piSrc += uiSrcStride, piDes += uiDesStride )
1758	{
1759	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1760	{
1761	piDes[ uiX ] = piSrc[ uiX ];
1762	}
1763	}
1764	piSrc = m_pcQTTempTComYuv[ uiQTLayer ].getCrAddr ( uiAbsPartIdx );
1765	piDes = pcCU->getPic()->getPicYuvRec()->getCrAddr ( pcCU->getAddr(), uiZOrder );
1766	for( UInt uiY = 0; uiY < uiHeight; uiY++, piSrc += uiSrcStride, piDes += uiDesStride )
1767	{
1768	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
1769	{
1770	piDes[ uiX ] = piSrc[ uiX ];
1771	}
1772	}
1773	}
1774	}
1775
1776	#if H_3D_VSO // M42
1777	if ( m_pcRdCost->getUseRenModel() && bCheckFull )
1778	{
1779	UInt uiWidth = pcCU->getWidth ( 0 ) >> uiTrDepth;
1780	UInt uiHeight = pcCU->getHeight( 0 ) >> uiTrDepth;
1781	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1782	Pel* piSrc = m_pcQTTempTComYuv[ uiQTLayer ].getLumaAddr( uiAbsPartIdx );
1783	UInt uiSrcStride = m_pcQTTempTComYuv[ uiQTLayer ].getStride ();
1784
1785	m_pcRdCost->setRenModelData( pcCU, uiAbsPartIdx, piSrc, (Int) uiSrcStride, (Int) uiWidth, (Int) uiHeight );
1786	}
1787	#endif
1788
1789	ruiDistY += uiSingleDistY;
1790	ruiDistC += uiSingleDistC;
1791	dRDCost += dSingleCost;
1792	}
1793
1794
1795	Void
1796	TEncSearch::xSetIntraResultQT( TComDataCU* pcCU,
1797	UInt uiTrDepth,
1798	UInt uiAbsPartIdx,
1799	Bool bLumaOnly,
1800	TComYuv* pcRecoYuv )
1801	{
1802	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
1803	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
1804	if( uiTrMode == uiTrDepth )
1805	{
1806	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
1807	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1808
1809	Bool bSkipChroma = false;
1810	Bool bChromaSame = false;
1811	if( !bLumaOnly && uiLog2TrSize == 2 )
1812	{
1813	assert( uiTrDepth > 0 );
1814	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth - 1 ) << 1 );
1815	bSkipChroma = ( ( uiAbsPartIdx % uiQPDiv ) != 0 );
1816	bChromaSame = true;
1817	}
1818
1819	//===== copy transform coefficients =====
1820	UInt uiNumCoeffY = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( uiFullDepth << 1 );
1821	UInt uiNumCoeffIncY = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 );
1822	TCoeff* pcCoeffSrcY = m_ppcQTTempCoeffY [ uiQTLayer ] + ( uiNumCoeffIncY * uiAbsPartIdx );
1823	TCoeff* pcCoeffDstY = pcCU->getCoeffY () + ( uiNumCoeffIncY * uiAbsPartIdx );
1824	::memcpy( pcCoeffDstY, pcCoeffSrcY, sizeof( TCoeff ) * uiNumCoeffY );
1825	#if ADAPTIVE_QP_SELECTION
1826	Int* pcArlCoeffSrcY = m_ppcQTTempArlCoeffY [ uiQTLayer ] + ( uiNumCoeffIncY * uiAbsPartIdx );
1827	Int* pcArlCoeffDstY = pcCU->getArlCoeffY () + ( uiNumCoeffIncY * uiAbsPartIdx );
1828	::memcpy( pcArlCoeffDstY, pcArlCoeffSrcY, sizeof( Int ) * uiNumCoeffY );
1829	#endif
1830	if( !bLumaOnly && !bSkipChroma )
1831	{
1832	UInt uiNumCoeffC = ( bChromaSame ? uiNumCoeffY : uiNumCoeffY >> 2 );
1833	UInt uiNumCoeffIncC = uiNumCoeffIncY >> 2;
1834	TCoeff* pcCoeffSrcU = m_ppcQTTempCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1835	TCoeff* pcCoeffSrcV = m_ppcQTTempCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1836	TCoeff* pcCoeffDstU = pcCU->getCoeffCb() + ( uiNumCoeffIncC * uiAbsPartIdx );
1837	TCoeff* pcCoeffDstV = pcCU->getCoeffCr() + ( uiNumCoeffIncC * uiAbsPartIdx );
1838	::memcpy( pcCoeffDstU, pcCoeffSrcU, sizeof( TCoeff ) * uiNumCoeffC );
1839	::memcpy( pcCoeffDstV, pcCoeffSrcV, sizeof( TCoeff ) * uiNumCoeffC );
1840	#if ADAPTIVE_QP_SELECTION
1841	Int* pcArlCoeffSrcU = m_ppcQTTempArlCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1842	Int* pcArlCoeffSrcV = m_ppcQTTempArlCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1843	Int* pcArlCoeffDstU = pcCU->getArlCoeffCb() + ( uiNumCoeffIncC * uiAbsPartIdx );
1844	Int* pcArlCoeffDstV = pcCU->getArlCoeffCr() + ( uiNumCoeffIncC * uiAbsPartIdx );
1845	::memcpy( pcArlCoeffDstU, pcArlCoeffSrcU, sizeof( Int ) * uiNumCoeffC );
1846	::memcpy( pcArlCoeffDstV, pcArlCoeffSrcV, sizeof( Int ) * uiNumCoeffC );
1847	#endif
1848	}
1849
1850	//===== copy reconstruction =====
1851	m_pcQTTempTComYuv[ uiQTLayer ].copyPartToPartLuma( pcRecoYuv, uiAbsPartIdx, 1 << uiLog2TrSize, 1 << uiLog2TrSize );
1852	if( !bLumaOnly && !bSkipChroma )
1853	{
1854	UInt uiLog2TrSizeChroma = ( bChromaSame ? uiLog2TrSize : uiLog2TrSize - 1 );
1855	m_pcQTTempTComYuv[ uiQTLayer ].copyPartToPartChroma( pcRecoYuv, uiAbsPartIdx, 1 << uiLog2TrSizeChroma, 1 << uiLog2TrSizeChroma );
1856	}
1857	}
1858	else
1859	{
1860	UInt uiNumQPart = pcCU->getPic()->getNumPartInCU() >> ( ( uiFullDepth + 1 ) << 1 );
1861	for( UInt uiPart = 0; uiPart < 4; uiPart++ )
1862	{
1863	xSetIntraResultQT( pcCU, uiTrDepth + 1, uiAbsPartIdx + uiPart * uiNumQPart, bLumaOnly, pcRecoYuv );
1864	}
1865	}
1866	}
1867
1868	Void
1869	TEncSearch::xStoreIntraResultQT( TComDataCU* pcCU,
1870	UInt uiTrDepth,
1871	UInt uiAbsPartIdx,
1872	Bool bLumaOnly )
1873	{
1874	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
1875	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
1876	assert( uiTrMode == uiTrDepth );
1877	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
1878	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1879
1880	Bool bSkipChroma = false;
1881	Bool bChromaSame = false;
1882	if( !bLumaOnly && uiLog2TrSize == 2 )
1883	{
1884	assert( uiTrDepth > 0 );
1885	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth - 1 ) << 1 );
1886	bSkipChroma = ( ( uiAbsPartIdx % uiQPDiv ) != 0 );
1887	bChromaSame = true;
1888	}
1889
1890	//===== copy transform coefficients =====
1891	UInt uiNumCoeffY = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( uiFullDepth << 1 );
1892	UInt uiNumCoeffIncY = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 );
1893	TCoeff* pcCoeffSrcY = m_ppcQTTempCoeffY [ uiQTLayer ] + ( uiNumCoeffIncY * uiAbsPartIdx );
1894	TCoeff* pcCoeffDstY = m_pcQTTempTUCoeffY;
1895
1896	::memcpy( pcCoeffDstY, pcCoeffSrcY, sizeof( TCoeff ) * uiNumCoeffY );
1897	#if ADAPTIVE_QP_SELECTION
1898	Int* pcArlCoeffSrcY = m_ppcQTTempArlCoeffY [ uiQTLayer ] + ( uiNumCoeffIncY * uiAbsPartIdx );
1899	Int* pcArlCoeffDstY = m_ppcQTTempTUArlCoeffY;
1900	::memcpy( pcArlCoeffDstY, pcArlCoeffSrcY, sizeof( Int ) * uiNumCoeffY );
1901	#endif
1902	if( !bLumaOnly && !bSkipChroma )
1903	{
1904	UInt uiNumCoeffC = ( bChromaSame ? uiNumCoeffY : uiNumCoeffY >> 2 );
1905	UInt uiNumCoeffIncC = uiNumCoeffIncY >> 2;
1906	TCoeff* pcCoeffSrcU = m_ppcQTTempCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1907	TCoeff* pcCoeffSrcV = m_ppcQTTempCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1908	TCoeff* pcCoeffDstU = m_pcQTTempTUCoeffCb;
1909	TCoeff* pcCoeffDstV = m_pcQTTempTUCoeffCr;
1910	::memcpy( pcCoeffDstU, pcCoeffSrcU, sizeof( TCoeff ) * uiNumCoeffC );
1911	::memcpy( pcCoeffDstV, pcCoeffSrcV, sizeof( TCoeff ) * uiNumCoeffC );
1912	#if ADAPTIVE_QP_SELECTION
1913	Int* pcArlCoeffSrcU = m_ppcQTTempArlCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1914	Int* pcArlCoeffSrcV = m_ppcQTTempArlCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1915	Int* pcArlCoeffDstU = m_ppcQTTempTUArlCoeffCb;
1916	Int* pcArlCoeffDstV = m_ppcQTTempTUArlCoeffCr;
1917	::memcpy( pcArlCoeffDstU, pcArlCoeffSrcU, sizeof( Int ) * uiNumCoeffC );
1918	::memcpy( pcArlCoeffDstV, pcArlCoeffSrcV, sizeof( Int ) * uiNumCoeffC );
1919	#endif
1920	}
1921
1922	//===== copy reconstruction =====
1923	m_pcQTTempTComYuv[ uiQTLayer ].copyPartToPartLuma( &m_pcQTTempTransformSkipTComYuv, uiAbsPartIdx, 1 << uiLog2TrSize, 1 << uiLog2TrSize );
1924
1925	if( !bLumaOnly && !bSkipChroma )
1926	{
1927	UInt uiLog2TrSizeChroma = ( bChromaSame ? uiLog2TrSize : uiLog2TrSize - 1 );
1928	m_pcQTTempTComYuv[ uiQTLayer ].copyPartToPartChroma( &m_pcQTTempTransformSkipTComYuv, uiAbsPartIdx, 1 << uiLog2TrSizeChroma, 1 << uiLog2TrSizeChroma );
1929	}
1930	}
1931
1932	Void
1933	TEncSearch::xLoadIntraResultQT( TComDataCU* pcCU,
1934	UInt uiTrDepth,
1935	UInt uiAbsPartIdx,
1936	Bool bLumaOnly )
1937	{
1938	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
1939	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
1940	assert( uiTrMode == uiTrDepth );
1941	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
1942	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
1943
1944	Bool bSkipChroma = false;
1945	Bool bChromaSame = false;
1946	if( !bLumaOnly && uiLog2TrSize == 2 )
1947	{
1948	assert( uiTrDepth > 0 );
1949	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth - 1 ) << 1 );
1950	bSkipChroma = ( ( uiAbsPartIdx % uiQPDiv ) != 0 );
1951	bChromaSame = true;
1952	}
1953
1954	//===== copy transform coefficients =====
1955	UInt uiNumCoeffY = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( uiFullDepth << 1 );
1956	UInt uiNumCoeffIncY = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 );
1957	TCoeff* pcCoeffDstY = m_ppcQTTempCoeffY [ uiQTLayer ] + ( uiNumCoeffIncY * uiAbsPartIdx );
1958	TCoeff* pcCoeffSrcY = m_pcQTTempTUCoeffY;
1959
1960	::memcpy( pcCoeffDstY, pcCoeffSrcY, sizeof( TCoeff ) * uiNumCoeffY );
1961	#if ADAPTIVE_QP_SELECTION
1962	Int* pcArlCoeffDstY = m_ppcQTTempArlCoeffY [ uiQTLayer ] + ( uiNumCoeffIncY * uiAbsPartIdx );
1963	Int* pcArlCoeffSrcY = m_ppcQTTempTUArlCoeffY;
1964	::memcpy( pcArlCoeffDstY, pcArlCoeffSrcY, sizeof( Int ) * uiNumCoeffY );
1965	#endif
1966	if( !bLumaOnly && !bSkipChroma )
1967	{
1968	UInt uiNumCoeffC = ( bChromaSame ? uiNumCoeffY : uiNumCoeffY >> 2 );
1969	UInt uiNumCoeffIncC = uiNumCoeffIncY >> 2;
1970	TCoeff* pcCoeffDstU = m_ppcQTTempCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1971	TCoeff* pcCoeffDstV = m_ppcQTTempCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1972	TCoeff* pcCoeffSrcU = m_pcQTTempTUCoeffCb;
1973	TCoeff* pcCoeffSrcV = m_pcQTTempTUCoeffCr;
1974	::memcpy( pcCoeffDstU, pcCoeffSrcU, sizeof( TCoeff ) * uiNumCoeffC );
1975	::memcpy( pcCoeffDstV, pcCoeffSrcV, sizeof( TCoeff ) * uiNumCoeffC );
1976	#if ADAPTIVE_QP_SELECTION
1977	Int* pcArlCoeffDstU = m_ppcQTTempArlCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1978	Int* pcArlCoeffDstV = m_ppcQTTempArlCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
1979	Int* pcArlCoeffSrcU = m_ppcQTTempTUArlCoeffCb;
1980	Int* pcArlCoeffSrcV = m_ppcQTTempTUArlCoeffCr;
1981	::memcpy( pcArlCoeffDstU, pcArlCoeffSrcU, sizeof( Int ) * uiNumCoeffC );
1982	::memcpy( pcArlCoeffDstV, pcArlCoeffSrcV, sizeof( Int ) * uiNumCoeffC );
1983	#endif
1984	}
1985
1986	//===== copy reconstruction =====
1987	m_pcQTTempTransformSkipTComYuv.copyPartToPartLuma( &m_pcQTTempTComYuv[ uiQTLayer ] , uiAbsPartIdx, 1 << uiLog2TrSize, 1 << uiLog2TrSize );
1988
1989	if( !bLumaOnly && !bSkipChroma )
1990	{
1991	UInt uiLog2TrSizeChroma = ( bChromaSame ? uiLog2TrSize : uiLog2TrSize - 1 );
1992	m_pcQTTempTransformSkipTComYuv.copyPartToPartChroma( &m_pcQTTempTComYuv[ uiQTLayer ], uiAbsPartIdx, 1 << uiLog2TrSizeChroma, 1 << uiLog2TrSizeChroma );
1993	}
1994
1995	UInt uiZOrder = pcCU->getZorderIdxInCU() + uiAbsPartIdx;
1996	Pel* piRecIPred = pcCU->getPic()->getPicYuvRec()->getLumaAddr( pcCU->getAddr(), uiZOrder );
1997	UInt uiRecIPredStride = pcCU->getPic()->getPicYuvRec()->getStride ();
1998	Pel* piRecQt = m_pcQTTempTComYuv[ uiQTLayer ].getLumaAddr( uiAbsPartIdx );
1999	UInt uiRecQtStride = m_pcQTTempTComYuv[ uiQTLayer ].getStride ();
2000	UInt uiWidth = pcCU ->getWidth ( 0 ) >> uiTrDepth;
2001	UInt uiHeight = pcCU ->getHeight ( 0 ) >> uiTrDepth;
2002	Pel* pRecQt = piRecQt;
2003	Pel* pRecIPred = piRecIPred;
2004	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
2005	{
2006	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
2007	{
2008	pRecIPred[ uiX ] = pRecQt [ uiX ];
2009	}
2010	pRecQt += uiRecQtStride;
2011	pRecIPred += uiRecIPredStride;
2012	}
2013
2014	if( !bLumaOnly && !bSkipChroma )
2015	{
2016	piRecIPred = pcCU->getPic()->getPicYuvRec()->getCbAddr( pcCU->getAddr(), uiZOrder );
2017	piRecQt = m_pcQTTempTComYuv[ uiQTLayer ].getCbAddr( uiAbsPartIdx );
2018	pRecQt = piRecQt;
2019	pRecIPred = piRecIPred;
2020	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
2021	{
2022	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
2023	{
2024	pRecIPred[ uiX ] = pRecQt[ uiX ];
2025	}
2026	pRecQt += uiRecQtStride;
2027	pRecIPred += uiRecIPredStride;
2028	}
2029
2030	piRecIPred = pcCU->getPic()->getPicYuvRec()->getCrAddr( pcCU->getAddr(), uiZOrder );
2031	piRecQt = m_pcQTTempTComYuv[ uiQTLayer ].getCrAddr( uiAbsPartIdx );
2032	pRecQt = piRecQt;
2033	pRecIPred = piRecIPred;
2034	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
2035	{
2036	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
2037	{
2038	pRecIPred[ uiX ] = pRecQt[ uiX ];
2039	}
2040	pRecQt += uiRecQtStride;
2041	pRecIPred += uiRecIPredStride;
2042	}
2043	}
2044	}
2045
2046	Void
2047	TEncSearch::xStoreIntraResultChromaQT( TComDataCU* pcCU,
2048	UInt uiTrDepth,
2049	UInt uiAbsPartIdx,
2050	UInt stateU0V1Both2 )
2051	{
2052	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
2053	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
2054	if( uiTrMode == uiTrDepth )
2055	{
2056	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
2057	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
2058
2059	Bool bChromaSame = false;
2060	if( uiLog2TrSize == 2 )
2061	{
2062	assert( uiTrDepth > 0 );
2063	uiTrDepth --;
2064	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth) << 1 );
2065	if( ( uiAbsPartIdx % uiQPDiv ) != 0 )
2066	{
2067	return;
2068	}
2069	bChromaSame = true;
2070	}
2071
2072	//===== copy transform coefficients =====
2073	UInt uiNumCoeffC = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( uiFullDepth << 1 );
2074	if( !bChromaSame )
2075	{
2076	uiNumCoeffC >>= 2;
2077	}
2078	UInt uiNumCoeffIncC = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 ) + 2 );
2079	if(stateU0V1Both2 == 0 \|\| stateU0V1Both2 == 2)
2080	{
2081	TCoeff* pcCoeffSrcU = m_ppcQTTempCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2082	TCoeff* pcCoeffDstU = m_pcQTTempTUCoeffCb;
2083	::memcpy( pcCoeffDstU, pcCoeffSrcU, sizeof( TCoeff ) * uiNumCoeffC );
2084
2085	#if ADAPTIVE_QP_SELECTION
2086	Int* pcArlCoeffSrcU = m_ppcQTTempArlCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2087	Int* pcArlCoeffDstU = m_ppcQTTempTUArlCoeffCb;
2088	::memcpy( pcArlCoeffDstU, pcArlCoeffSrcU, sizeof( Int ) * uiNumCoeffC );
2089	#endif
2090	}
2091	if(stateU0V1Both2 == 1 \|\| stateU0V1Both2 == 2)
2092	{
2093	TCoeff* pcCoeffSrcV = m_ppcQTTempCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2094	TCoeff* pcCoeffDstV = m_pcQTTempTUCoeffCr;
2095	::memcpy( pcCoeffDstV, pcCoeffSrcV, sizeof( TCoeff ) * uiNumCoeffC );
2096	#if ADAPTIVE_QP_SELECTION
2097	Int* pcArlCoeffSrcV = m_ppcQTTempArlCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2098	Int* pcArlCoeffDstV = m_ppcQTTempTUArlCoeffCr;
2099	::memcpy( pcArlCoeffDstV, pcArlCoeffSrcV, sizeof( Int ) * uiNumCoeffC );
2100	#endif
2101	}
2102
2103	//===== copy reconstruction =====
2104	UInt uiLog2TrSizeChroma = ( bChromaSame ? uiLog2TrSize : uiLog2TrSize - 1 );
2105	m_pcQTTempTComYuv[ uiQTLayer ].copyPartToPartChroma(&m_pcQTTempTransformSkipTComYuv, uiAbsPartIdx, 1 << uiLog2TrSizeChroma, 1 << uiLog2TrSizeChroma, stateU0V1Both2 );
2106	}
2107	}
2108
2109
2110	Void
2111	TEncSearch::xLoadIntraResultChromaQT( TComDataCU* pcCU,
2112	UInt uiTrDepth,
2113	UInt uiAbsPartIdx,
2114	UInt stateU0V1Both2 )
2115	{
2116	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
2117	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
2118	if( uiTrMode == uiTrDepth )
2119	{
2120	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
2121	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
2122
2123	Bool bChromaSame = false;
2124	if( uiLog2TrSize == 2 )
2125	{
2126	assert( uiTrDepth > 0 );
2127	uiTrDepth --;
2128	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth ) << 1 );
2129	if( ( uiAbsPartIdx % uiQPDiv ) != 0 )
2130	{
2131	return;
2132	}
2133	bChromaSame = true;
2134	}
2135
2136	//===== copy transform coefficients =====
2137	UInt uiNumCoeffC = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( uiFullDepth << 1 );
2138	if( !bChromaSame )
2139	{
2140	uiNumCoeffC >>= 2;
2141	}
2142	UInt uiNumCoeffIncC = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 ) + 2 );
2143
2144	if(stateU0V1Both2 ==0 \|\| stateU0V1Both2 == 2)
2145	{
2146	TCoeff* pcCoeffDstU = m_ppcQTTempCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2147	TCoeff* pcCoeffSrcU = m_pcQTTempTUCoeffCb;
2148	::memcpy( pcCoeffDstU, pcCoeffSrcU, sizeof( TCoeff ) * uiNumCoeffC );
2149	#if ADAPTIVE_QP_SELECTION
2150	Int* pcArlCoeffDstU = m_ppcQTTempArlCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2151	Int* pcArlCoeffSrcU = m_ppcQTTempTUArlCoeffCb;
2152	::memcpy( pcArlCoeffDstU, pcArlCoeffSrcU, sizeof( Int ) * uiNumCoeffC );
2153	#endif
2154	}
2155	if(stateU0V1Both2 ==1 \|\| stateU0V1Both2 == 2)
2156	{
2157	TCoeff* pcCoeffDstV = m_ppcQTTempCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2158	TCoeff* pcCoeffSrcV = m_pcQTTempTUCoeffCr;
2159	::memcpy( pcCoeffDstV, pcCoeffSrcV, sizeof( TCoeff ) * uiNumCoeffC );
2160	#if ADAPTIVE_QP_SELECTION
2161	Int* pcArlCoeffDstV = m_ppcQTTempArlCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2162	Int* pcArlCoeffSrcV = m_ppcQTTempTUArlCoeffCr;
2163	::memcpy( pcArlCoeffDstV, pcArlCoeffSrcV, sizeof( Int ) * uiNumCoeffC );
2164	#endif
2165	}
2166
2167	//===== copy reconstruction =====
2168	UInt uiLog2TrSizeChroma = ( bChromaSame ? uiLog2TrSize : uiLog2TrSize - 1 );
2169	m_pcQTTempTransformSkipTComYuv.copyPartToPartChroma( &m_pcQTTempTComYuv[ uiQTLayer ], uiAbsPartIdx, 1 << uiLog2TrSizeChroma, 1 << uiLog2TrSizeChroma, stateU0V1Both2);
2170
2171	UInt uiZOrder = pcCU->getZorderIdxInCU() + uiAbsPartIdx;
2172	UInt uiWidth = pcCU->getWidth ( 0 ) >> (uiTrDepth + 1);
2173	UInt uiHeight = pcCU->getHeight ( 0 ) >> (uiTrDepth + 1);
2174	UInt uiRecQtStride = m_pcQTTempTComYuv[ uiQTLayer ].getCStride ();
2175	UInt uiRecIPredStride = pcCU->getPic()->getPicYuvRec()->getCStride ();
2176
2177	if(stateU0V1Both2 ==0 \|\| stateU0V1Both2 == 2)
2178	{
2179	Pel* piRecIPred = pcCU->getPic()->getPicYuvRec()->getCbAddr( pcCU->getAddr(), uiZOrder );
2180	Pel* piRecQt = m_pcQTTempTComYuv[ uiQTLayer ].getCbAddr( uiAbsPartIdx );
2181	Pel* pRecQt = piRecQt;
2182	Pel* pRecIPred = piRecIPred;
2183	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
2184	{
2185	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
2186	{
2187	pRecIPred[ uiX ] = pRecQt[ uiX ];
2188	}
2189	pRecQt += uiRecQtStride;
2190	pRecIPred += uiRecIPredStride;
2191	}
2192	}
2193	if(stateU0V1Both2 == 1 \|\| stateU0V1Both2 == 2)
2194	{
2195	Pel* piRecIPred = pcCU->getPic()->getPicYuvRec()->getCrAddr( pcCU->getAddr(), uiZOrder );
2196	Pel* piRecQt = m_pcQTTempTComYuv[ uiQTLayer ].getCrAddr( uiAbsPartIdx );
2197	Pel* pRecQt = piRecQt;
2198	Pel* pRecIPred = piRecIPred;
2199	for( UInt uiY = 0; uiY < uiHeight; uiY++ )
2200	{
2201	for( UInt uiX = 0; uiX < uiWidth; uiX++ )
2202	{
2203	pRecIPred[ uiX ] = pRecQt[ uiX ];
2204	}
2205	pRecQt += uiRecQtStride;
2206	pRecIPred += uiRecIPredStride;
2207	}
2208	}
2209	}
2210	}
2211
2212	Void
2213	TEncSearch::xRecurIntraChromaCodingQT( TComDataCU* pcCU,
2214	UInt uiTrDepth,
2215	UInt uiAbsPartIdx,
2216	TComYuv* pcOrgYuv,
2217	TComYuv* pcPredYuv,
2218	TComYuv* pcResiYuv,
2219	UInt& ruiDist )
2220	{
2221	UInt uiFullDepth = pcCU->getDepth( 0 ) + uiTrDepth;
2222	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
2223	if( uiTrMode == uiTrDepth )
2224	{
2225	Bool checkTransformSkip = pcCU->getSlice()->getPPS()->getUseTransformSkip();
2226	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
2227
2228	UInt actualTrDepth = uiTrDepth;
2229	if( uiLog2TrSize == 2 )
2230	{
2231	assert( uiTrDepth > 0 );
2232	actualTrDepth--;
2233	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + actualTrDepth) << 1 );
2234	Bool bFirstQ = ( ( uiAbsPartIdx % uiQPDiv ) == 0 );
2235	if( !bFirstQ )
2236	{
2237	return;
2238	}
2239	}
2240
2241	checkTransformSkip &= (uiLog2TrSize <= 3);
2242	if ( m_pcEncCfg->getUseTransformSkipFast() )
2243	{
2244	checkTransformSkip &= (uiLog2TrSize < 3);
2245	if (checkTransformSkip)
2246	{
2247	Int nbLumaSkip = 0;
2248	for(UInt absPartIdxSub = uiAbsPartIdx; absPartIdxSub < uiAbsPartIdx + 4; absPartIdxSub ++)
2249	{
2250	nbLumaSkip += pcCU->getTransformSkip(absPartIdxSub, TEXT_LUMA);
2251	}
2252	checkTransformSkip &= (nbLumaSkip > 0);
2253	}
2254	}
2255
2256	if(checkTransformSkip)
2257	{
2258	//use RDO to decide whether Cr/Cb takes TS
2259	if( m_bUseSBACRD )
2260	{
2261	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[uiFullDepth][CI_QT_TRAFO_ROOT] );
2262	}
2263
2264	for(Int chromaId = 0; chromaId < 2; chromaId ++)
2265	{
2266	Double dSingleCost = MAX_DOUBLE;
2267	Int bestModeId = 0;
2268	UInt singleDistC = 0;
2269	UInt singleCbfC = 0;
2270	UInt singleDistCTmp = 0;
2271	Double singleCostTmp = 0;
2272	UInt singleCbfCTmp = 0;
2273
2274	Int default0Save1Load2 = 0;
2275	Int firstCheckId = 0;
2276
2277	for(Int chromaModeId = firstCheckId; chromaModeId < 2; chromaModeId ++)
2278	{
2279	pcCU->setTransformSkipSubParts ( chromaModeId, (TextType)(chromaId + 2), uiAbsPartIdx, pcCU->getDepth( 0 ) + actualTrDepth);
2280	if(chromaModeId == firstCheckId)
2281	{
2282	default0Save1Load2 = 1;
2283	}
2284	else
2285	{
2286	default0Save1Load2 = 2;
2287	}
2288	singleDistCTmp = 0;
2289	xIntraCodingChromaBlk( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, singleDistCTmp, chromaId ,default0Save1Load2);
2290	singleCbfCTmp = pcCU->getCbf( uiAbsPartIdx, (TextType)(chromaId + 2), uiTrDepth);
2291
2292	if(chromaModeId == 1 && singleCbfCTmp == 0)
2293	{
2294	//In order not to code TS flag when cbf is zero, the case for TS with cbf being zero is forbidden.
2295	singleCostTmp = MAX_DOUBLE;
2296	}
2297	else
2298	{
2299	UInt bitsTmp = xGetIntraBitsQTChroma( pcCU,uiTrDepth, uiAbsPartIdx,chromaId + 2, false );
2300	singleCostTmp = m_pcRdCost->calcRdCost( bitsTmp, singleDistCTmp);
2301	}
2302
2303	if(singleCostTmp < dSingleCost)
2304	{
2305	dSingleCost = singleCostTmp;
2306	singleDistC = singleDistCTmp;
2307	bestModeId = chromaModeId;
2308	singleCbfC = singleCbfCTmp;
2309
2310	if(bestModeId == firstCheckId)
2311	{
2312	xStoreIntraResultChromaQT(pcCU, uiTrDepth, uiAbsPartIdx,chromaId);
2313	if( m_bUseSBACRD)
2314	{
2315	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ uiFullDepth ][ CI_TEMP_BEST ] );
2316	}
2317	}
2318	}
2319	if(chromaModeId == firstCheckId)
2320	{
2321	m_pcRDGoOnSbacCoder->load ( m_pppcRDSbacCoder[ uiFullDepth ][ CI_QT_TRAFO_ROOT ] );
2322	}
2323	}
2324
2325	if(bestModeId == firstCheckId)
2326	{
2327	xLoadIntraResultChromaQT(pcCU, uiTrDepth, uiAbsPartIdx,chromaId);
2328	pcCU->setCbfSubParts ( singleCbfC << uiTrDepth, (TextType)(chromaId + 2), uiAbsPartIdx, pcCU->getDepth(0) + actualTrDepth );
2329	if(m_bUseSBACRD)
2330	{
2331	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[ uiFullDepth ][ CI_TEMP_BEST ] );
2332	}
2333	}
2334	pcCU ->setTransformSkipSubParts( bestModeId, (TextType)(chromaId + 2), uiAbsPartIdx, pcCU->getDepth( 0 ) + actualTrDepth );
2335	ruiDist += singleDistC;
2336
2337	if(chromaId == 0)
2338	{
2339	if( m_bUseSBACRD )
2340	{
2341	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[uiFullDepth][CI_QT_TRAFO_ROOT] );
2342	}
2343	}
2344	}
2345	}
2346	else
2347	{
2348	pcCU ->setTransformSkipSubParts( 0, TEXT_CHROMA_U, uiAbsPartIdx, pcCU->getDepth( 0 ) + actualTrDepth );
2349	pcCU ->setTransformSkipSubParts( 0, TEXT_CHROMA_V, uiAbsPartIdx, pcCU->getDepth( 0 ) + actualTrDepth );
2350	xIntraCodingChromaBlk( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, ruiDist, 0 );
2351	xIntraCodingChromaBlk( pcCU, uiTrDepth, uiAbsPartIdx, pcOrgYuv, pcPredYuv, pcResiYuv, ruiDist, 1 );
2352	}
2353	}
2354	else
2355	{
2356	UInt uiSplitCbfU = 0;
2357	UInt uiSplitCbfV = 0;
2358	UInt uiQPartsDiv = pcCU->getPic()->getNumPartInCU() >> ( ( uiFullDepth + 1 ) << 1 );
2359	UInt uiAbsPartIdxSub = uiAbsPartIdx;
2360	for( UInt uiPart = 0; uiPart < 4; uiPart++, uiAbsPartIdxSub += uiQPartsDiv )
2361	{
2362	xRecurIntraChromaCodingQT( pcCU, uiTrDepth + 1, uiAbsPartIdxSub, pcOrgYuv, pcPredYuv, pcResiYuv, ruiDist );
2363	uiSplitCbfU \|= pcCU->getCbf( uiAbsPartIdxSub, TEXT_CHROMA_U, uiTrDepth + 1 );
2364	uiSplitCbfV \|= pcCU->getCbf( uiAbsPartIdxSub, TEXT_CHROMA_V, uiTrDepth + 1 );
2365	}
2366	for( UInt uiOffs = 0; uiOffs < 4 * uiQPartsDiv; uiOffs++ )
2367	{
2368	pcCU->getCbf( TEXT_CHROMA_U )[ uiAbsPartIdx + uiOffs ] \|= ( uiSplitCbfU << uiTrDepth );
2369	pcCU->getCbf( TEXT_CHROMA_V )[ uiAbsPartIdx + uiOffs ] \|= ( uiSplitCbfV << uiTrDepth );
2370	}
2371	}
2372	}
2373
2374	Void
2375	TEncSearch::xSetIntraResultChromaQT( TComDataCU* pcCU,
2376	UInt uiTrDepth,
2377	UInt uiAbsPartIdx,
2378	TComYuv* pcRecoYuv )
2379	{
2380	UInt uiFullDepth = pcCU->getDepth(0) + uiTrDepth;
2381	UInt uiTrMode = pcCU->getTransformIdx( uiAbsPartIdx );
2382	if( uiTrMode == uiTrDepth )
2383	{
2384	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiFullDepth ] + 2;
2385	UInt uiQTLayer = pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() - uiLog2TrSize;
2386
2387	Bool bChromaSame = false;
2388	if( uiLog2TrSize == 2 )
2389	{
2390	assert( uiTrDepth > 0 );
2391	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrDepth - 1 ) << 1 );
2392	if( ( uiAbsPartIdx % uiQPDiv ) != 0 )
2393	{
2394	return;
2395	}
2396	bChromaSame = true;
2397	}
2398
2399	//===== copy transform coefficients =====
2400	UInt uiNumCoeffC = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( uiFullDepth << 1 );
2401	if( !bChromaSame )
2402	{
2403	uiNumCoeffC >>= 2;
2404	}
2405	UInt uiNumCoeffIncC = ( pcCU->getSlice()->getSPS()->getMaxCUWidth() * pcCU->getSlice()->getSPS()->getMaxCUHeight() ) >> ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() << 1 ) + 2 );
2406	TCoeff* pcCoeffSrcU = m_ppcQTTempCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2407	TCoeff* pcCoeffSrcV = m_ppcQTTempCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2408	TCoeff* pcCoeffDstU = pcCU->getCoeffCb() + ( uiNumCoeffIncC * uiAbsPartIdx );
2409	TCoeff* pcCoeffDstV = pcCU->getCoeffCr() + ( uiNumCoeffIncC * uiAbsPartIdx );
2410	::memcpy( pcCoeffDstU, pcCoeffSrcU, sizeof( TCoeff ) * uiNumCoeffC );
2411	::memcpy( pcCoeffDstV, pcCoeffSrcV, sizeof( TCoeff ) * uiNumCoeffC );
2412	#if ADAPTIVE_QP_SELECTION
2413	Int* pcArlCoeffSrcU = m_ppcQTTempArlCoeffCb[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2414	Int* pcArlCoeffSrcV = m_ppcQTTempArlCoeffCr[ uiQTLayer ] + ( uiNumCoeffIncC * uiAbsPartIdx );
2415	Int* pcArlCoeffDstU = pcCU->getArlCoeffCb() + ( uiNumCoeffIncC * uiAbsPartIdx );
2416	Int* pcArlCoeffDstV = pcCU->getArlCoeffCr() + ( uiNumCoeffIncC * uiAbsPartIdx );
2417	::memcpy( pcArlCoeffDstU, pcArlCoeffSrcU, sizeof( Int ) * uiNumCoeffC );
2418	::memcpy( pcArlCoeffDstV, pcArlCoeffSrcV, sizeof( Int ) * uiNumCoeffC );
2419	#endif
2420
2421	//===== copy reconstruction =====
2422	UInt uiLog2TrSizeChroma = ( bChromaSame ? uiLog2TrSize : uiLog2TrSize - 1 );
2423	m_pcQTTempTComYuv[ uiQTLayer ].copyPartToPartChroma( pcRecoYuv, uiAbsPartIdx, 1 << uiLog2TrSizeChroma, 1 << uiLog2TrSizeChroma );
2424	}
2425	else
2426	{
2427	UInt uiNumQPart = pcCU->getPic()->getNumPartInCU() >> ( ( uiFullDepth + 1 ) << 1 );
2428	for( UInt uiPart = 0; uiPart < 4; uiPart++ )
2429	{
2430	xSetIntraResultChromaQT( pcCU, uiTrDepth + 1, uiAbsPartIdx + uiPart * uiNumQPart, pcRecoYuv );
2431	}
2432	}
2433	}
2434
2435
2436	Void
2437	TEncSearch::preestChromaPredMode( TComDataCU* pcCU,
2438	TComYuv* pcOrgYuv,
2439	TComYuv* pcPredYuv )
2440	{
2441	UInt uiWidth = pcCU->getWidth ( 0 ) >> 1;
2442	UInt uiHeight = pcCU->getHeight( 0 ) >> 1;
2443	UInt uiStride = pcOrgYuv ->getCStride();
2444	Pel* piOrgU = pcOrgYuv ->getCbAddr ( 0 );
2445	Pel* piOrgV = pcOrgYuv ->getCrAddr ( 0 );
2446	Pel* piPredU = pcPredYuv->getCbAddr ( 0 );
2447	Pel* piPredV = pcPredYuv->getCrAddr ( 0 );
2448
2449	//===== init pattern =====
2450	Bool bAboveAvail = false;
2451	Bool bLeftAvail = false;
2452	pcCU->getPattern()->initPattern ( pcCU, 0, 0 );
2453	pcCU->getPattern()->initAdiPatternChroma( pcCU, 0, 0, m_piYuvExt, m_iYuvExtStride, m_iYuvExtHeight, bAboveAvail, bLeftAvail );
2454	Int* pPatChromaU = pcCU->getPattern()->getAdiCbBuf( uiWidth, uiHeight, m_piYuvExt );
2455	Int* pPatChromaV = pcCU->getPattern()->getAdiCrBuf( uiWidth, uiHeight, m_piYuvExt );
2456
2457	//===== get best prediction modes (using SAD) =====
2458	UInt uiMinMode = 0;
2459	UInt uiMaxMode = 4;
2460	UInt uiBestMode = MAX_UINT;
2461	UInt uiMinSAD = MAX_UINT;
2462	for( UInt uiMode = uiMinMode; uiMode < uiMaxMode; uiMode++ )
2463	{
2464	//--- get prediction ---
2465	predIntraChromaAng( pPatChromaU, uiMode, piPredU, uiStride, uiWidth, uiHeight, bAboveAvail, bLeftAvail );
2466	predIntraChromaAng( pPatChromaV, uiMode, piPredV, uiStride, uiWidth, uiHeight, bAboveAvail, bLeftAvail );
2467
2468	//--- get SAD ---
2469	UInt uiSAD = m_pcRdCost->calcHAD(g_bitDepthC, piOrgU, uiStride, piPredU, uiStride, uiWidth, uiHeight );
2470	uiSAD += m_pcRdCost->calcHAD(g_bitDepthC, piOrgV, uiStride, piPredV, uiStride, uiWidth, uiHeight );
2471	//--- check ---
2472	if( uiSAD < uiMinSAD )
2473	{
2474	uiMinSAD = uiSAD;
2475	uiBestMode = uiMode;
2476	}
2477	}
2478
2479	//===== set chroma pred mode =====
2480	pcCU->setChromIntraDirSubParts( uiBestMode, 0, pcCU->getDepth( 0 ) );
2481	}
2482
2483	Void
2484	TEncSearch::estIntraPredQT( TComDataCU* pcCU,
2485	TComYuv* pcOrgYuv,
2486	TComYuv* pcPredYuv,
2487	TComYuv* pcResiYuv,
2488	TComYuv* pcRecoYuv,
2489	UInt& ruiDistC,
2490	Bool bLumaOnly )
2491	{
2492	UInt uiDepth = pcCU->getDepth(0);
2493	UInt uiNumPU = pcCU->getNumPartInter();
2494	UInt uiInitTrDepth = pcCU->getPartitionSize(0) == SIZE_2Nx2N ? 0 : 1;
2495	UInt uiWidth = pcCU->getWidth (0) >> uiInitTrDepth;
2496	UInt uiHeight = pcCU->getHeight(0) >> uiInitTrDepth;
2497	UInt uiQNumParts = pcCU->getTotalNumPart() >> 2;
2498	UInt uiWidthBit = pcCU->getIntraSizeIdx(0);
2499	#if H_3D_VSO
2500	Dist uiOverallDistY = 0;
2501	#else
2502	UInt uiOverallDistY = 0;
2503	#endif
2504	UInt uiOverallDistC = 0;
2505	UInt CandNum;
2506	Double CandCostList[ FAST_UDI_MAX_RDMODE_NUM ];
2507
2508	//===== set QP and clear Cbf =====
2509	if ( pcCU->getSlice()->getPPS()->getUseDQP() == true)
2510	{
2511	pcCU->setQPSubParts( pcCU->getQP(0), 0, uiDepth );
2512	}
2513	else
2514	{
2515	pcCU->setQPSubParts( pcCU->getSlice()->getSliceQp(), 0, uiDepth );
2516	}
2517
2518	//===== loop over partitions =====
2519	UInt uiPartOffset = 0;
2520	for( UInt uiPU = 0; uiPU < uiNumPU; uiPU++, uiPartOffset += uiQNumParts )
2521	{
2522	//===== init pattern for luma prediction =====
2523	Bool bAboveAvail = false;
2524	Bool bLeftAvail = false;
2525	pcCU->getPattern()->initPattern ( pcCU, uiInitTrDepth, uiPartOffset );
2526	pcCU->getPattern()->initAdiPattern( pcCU, uiPartOffset, uiInitTrDepth, m_piYuvExt, m_iYuvExtStride, m_iYuvExtHeight, bAboveAvail, bLeftAvail );
2527
2528	//===== determine set of modes to be tested (using prediction signal only) =====
2529	Int numModesAvailable = 35; //total number of Intra modes
2530	Pel* piOrg = pcOrgYuv ->getLumaAddr( uiPU, uiWidth );
2531	Pel* piPred = pcPredYuv->getLumaAddr( uiPU, uiWidth );
2532	UInt uiStride = pcPredYuv->getStride();
2533	UInt uiRdModeList[FAST_UDI_MAX_RDMODE_NUM];
2534	Int numModesForFullRD = g_aucIntraModeNumFast[ uiWidthBit ];
2535
2536	Bool doFastSearch = (numModesForFullRD != numModesAvailable);
2537	if (doFastSearch)
2538	{
2539	assert(numModesForFullRD < numModesAvailable);
2540
2541	for( Int i=0; i < numModesForFullRD; i++ )
2542	{
2543	CandCostList[ i ] = MAX_DOUBLE;
2544	}
2545	CandNum = 0;
2546
2547	for( Int modeIdx = 0; modeIdx < numModesAvailable; modeIdx++ )
2548	{
2549	UInt uiMode = modeIdx;
2550
2551	predIntraLumaAng( pcCU->getPattern(), uiMode, piPred, uiStride, uiWidth, uiHeight, bAboveAvail, bLeftAvail );
2552	#if H_3D_VSO // M34
2553	Dist uiSad;
2554	if ( m_pcRdCost->getUseVSO() )
2555	{
2556	if ( m_pcRdCost->getUseEstimatedVSD() )
2557	{
2558	uiSad = (Dist) ( m_pcRdCost->getDistPartVSD( pcCU, uiPartOffset, piPred, uiStride, piOrg, uiStride, uiWidth, uiHeight, true ) );
2559	}
2560	else
2561	{
2562	uiSad = m_pcRdCost->getDistPartVSO(pcCU, uiPartOffset, piPred, uiStride, piOrg, uiStride, uiWidth, uiHeight, true );
2563	}
2564	}
2565	else
2566	{
2567	uiSad = m_pcRdCost->calcHAD(g_bitDepthY, piOrg, uiStride, piPred, uiStride, uiWidth, uiHeight );
2568	}
2569	#else
2570	// use hadamard transform here
2571	UInt uiSad = m_pcRdCost->calcHAD(g_bitDepthY, piOrg, uiStride, piPred, uiStride, uiWidth, uiHeight );
2572	#endif
2573
2574	UInt iModeBits = xModeBitsIntra( pcCU, uiMode, uiPU, uiPartOffset, uiDepth, uiInitTrDepth );
2575
2576
2577	#if H_3D_VSO // M35
2578	Double dLambda;
2579	if ( m_pcRdCost->getUseLambdaScaleVSO() )
2580	dLambda = m_pcRdCost->getUseRenModel() ? m_pcRdCost->getLambdaVSO() : m_pcRdCost->getSqrtLambdaVSO();
2581	else
2582	dLambda = m_pcRdCost->getSqrtLambda();
2583
2584	Double cost = (Double)uiSad + (Double)iModeBits * m_pcRdCost->getSqrtLambda();
2585	#else
2586	Double cost = (Double)uiSad + (Double)iModeBits * m_pcRdCost->getSqrtLambda();
2587	#endif
2588
2589	CandNum += xUpdateCandList( uiMode, cost, numModesForFullRD, uiRdModeList, CandCostList );
2590	}
2591
2592	#if FAST_UDI_USE_MPM
2593	Int uiPreds[3] = {-1, -1, -1};
2594	Int iMode = -1;
2595	Int numCand = pcCU->getIntraDirLumaPredictor( uiPartOffset, uiPreds, &iMode );
2596	if( iMode >= 0 )
2597	{
2598	numCand = iMode;
2599	}
2600
2601	for( Int j=0; j < numCand; j++)
2602
2603	{
2604	Bool mostProbableModeIncluded = false;
2605	Int mostProbableMode = uiPreds[j];
2606
2607	for( Int i=0; i < numModesForFullRD; i++)
2608	{
2609	mostProbableModeIncluded \|= (mostProbableMode == uiRdModeList[i]);
2610	}
2611	if (!mostProbableModeIncluded)
2612	{
2613	uiRdModeList[numModesForFullRD++] = mostProbableMode;
2614	}
2615	}
2616	#endif // FAST_UDI_USE_MPM
2617	}
2618	else
2619	{
2620	for( Int i=0; i < numModesForFullRD; i++)
2621	{
2622	uiRdModeList[i] = i;
2623	}
2624	}
2625
2626	//===== check modes (using r-d costs) =====
2627	#if HHI_RQT_INTRA_SPEEDUP_MOD
2628	UInt uiSecondBestMode = MAX_UINT;
2629	Double dSecondBestPUCost = MAX_DOUBLE;
2630	#endif
2631
2632	UInt uiBestPUMode = 0;
2633	#if H_3D_VSO
2634	Dist uiBestPUDistY = 0;
2635	#else
2636	UInt uiBestPUDistY = 0;
2637	#endif
2638	UInt uiBestPUDistC = 0;
2639	Double dBestPUCost = MAX_DOUBLE;
2640	for( UInt uiMode = 0; uiMode < numModesForFullRD; uiMode++ )
2641	{
2642	// set luma prediction mode
2643	UInt uiOrgMode = uiRdModeList[uiMode];
2644
2645	pcCU->setLumaIntraDirSubParts ( uiOrgMode, uiPartOffset, uiDepth + uiInitTrDepth );
2646
2647	// set context models
2648	if( m_bUseSBACRD )
2649	{
2650	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST] );
2651	}
2652
2653	// determine residual for partition
2654	#if H_3D_VSO
2655	Dist uiPUDistY = 0;
2656	#else
2657	UInt uiPUDistY = 0;
2658	#endif
2659	UInt uiPUDistC = 0;
2660	Double dPUCost = 0.0;
2661	#if H_3D_VSO // M36
2662	if( m_pcRdCost->getUseRenModel() )
2663	{
2664	m_pcRdCost->setRenModelData( pcCU, uiPartOffset, piOrg, uiStride, uiWidth, uiHeight );
2665	}
2666	#endif
2667	#if HHI_RQT_INTRA_SPEEDUP
2668	xRecurIntraCodingQT( pcCU, uiInitTrDepth, uiPartOffset, bLumaOnly, pcOrgYuv, pcPredYuv, pcResiYuv, uiPUDistY, uiPUDistC, true, dPUCost );
2669	#else
2670	xRecurIntraCodingQT( pcCU, uiInitTrDepth, uiPartOffset, bLumaOnly, pcOrgYuv, pcPredYuv, pcResiYuv, uiPUDistY, uiPUDistC, dPUCost );
2671	#endif
2672
2673	// check r-d cost
2674	if( dPUCost < dBestPUCost )
2675	{
2676	#if HHI_RQT_INTRA_SPEEDUP_MOD
2677	uiSecondBestMode = uiBestPUMode;
2678	dSecondBestPUCost = dBestPUCost;
2679	#endif
2680	uiBestPUMode = uiOrgMode;
2681	uiBestPUDistY = uiPUDistY;
2682	uiBestPUDistC = uiPUDistC;
2683	dBestPUCost = dPUCost;
2684
2685	xSetIntraResultQT( pcCU, uiInitTrDepth, uiPartOffset, bLumaOnly, pcRecoYuv );
2686
2687	UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth(0) + uiInitTrDepth ) << 1 );
2688	::memcpy( m_puhQTTempTrIdx, pcCU->getTransformIdx() + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2689	::memcpy( m_puhQTTempCbf[0], pcCU->getCbf( TEXT_LUMA ) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2690	::memcpy( m_puhQTTempCbf[1], pcCU->getCbf( TEXT_CHROMA_U ) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2691	::memcpy( m_puhQTTempCbf[2], pcCU->getCbf( TEXT_CHROMA_V ) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2692	::memcpy( m_puhQTTempTransformSkipFlag[0], pcCU->getTransformSkip(TEXT_LUMA) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2693	::memcpy( m_puhQTTempTransformSkipFlag[1], pcCU->getTransformSkip(TEXT_CHROMA_U) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2694	::memcpy( m_puhQTTempTransformSkipFlag[2], pcCU->getTransformSkip(TEXT_CHROMA_V) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2695	}
2696	#if HHI_RQT_INTRA_SPEEDUP_MOD
2697	else if( dPUCost < dSecondBestPUCost )
2698	{
2699	uiSecondBestMode = uiOrgMode;
2700	dSecondBestPUCost = dPUCost;
2701	}
2702	#endif
2703	} // Mode loop
2704
2705	#if HHI_RQT_INTRA_SPEEDUP
2706	#if HHI_RQT_INTRA_SPEEDUP_MOD
2707	for( UInt ui =0; ui < 2; ++ui )
2708	#endif
2709	{
2710	#if HHI_RQT_INTRA_SPEEDUP_MOD
2711	UInt uiOrgMode = ui ? uiSecondBestMode : uiBestPUMode;
2712	if( uiOrgMode == MAX_UINT )
2713	{
2714	break;
2715	}
2716	#else
2717	UInt uiOrgMode = uiBestPUMode;
2718	#endif
2719
2720	pcCU->setLumaIntraDirSubParts ( uiOrgMode, uiPartOffset, uiDepth + uiInitTrDepth );
2721
2722	// set context models
2723	if( m_bUseSBACRD )
2724	{
2725	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST] );
2726	}
2727
2728	// determine residual for partition
2729	#if H_3D_VSO
2730	Dist uiPUDistY = 0;
2731	#else
2732	UInt uiPUDistY = 0;
2733	#endif
2734	UInt uiPUDistC = 0;
2735	Double dPUCost = 0.0;
2736
2737	#if H_3D_VSO // M37
2738	// reset Model
2739	if( m_pcRdCost->getUseRenModel() )
2740	{
2741	m_pcRdCost->setRenModelData( pcCU, uiPartOffset, piOrg, uiStride, uiWidth, uiHeight );
2742	}
2743	#endif
2744	xRecurIntraCodingQT( pcCU, uiInitTrDepth, uiPartOffset, bLumaOnly, pcOrgYuv, pcPredYuv, pcResiYuv, uiPUDistY, uiPUDistC, false, dPUCost );
2745
2746	// check r-d cost
2747	if( dPUCost < dBestPUCost )
2748	{
2749	uiBestPUMode = uiOrgMode;
2750	uiBestPUDistY = uiPUDistY;
2751	uiBestPUDistC = uiPUDistC;
2752	dBestPUCost = dPUCost;
2753
2754	xSetIntraResultQT( pcCU, uiInitTrDepth, uiPartOffset, bLumaOnly, pcRecoYuv );
2755
2756	UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth(0) + uiInitTrDepth ) << 1 );
2757	::memcpy( m_puhQTTempTrIdx, pcCU->getTransformIdx() + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2758	::memcpy( m_puhQTTempCbf[0], pcCU->getCbf( TEXT_LUMA ) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2759	::memcpy( m_puhQTTempCbf[1], pcCU->getCbf( TEXT_CHROMA_U ) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2760	::memcpy( m_puhQTTempCbf[2], pcCU->getCbf( TEXT_CHROMA_V ) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2761	::memcpy( m_puhQTTempTransformSkipFlag[0], pcCU->getTransformSkip(TEXT_LUMA) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2762	::memcpy( m_puhQTTempTransformSkipFlag[1], pcCU->getTransformSkip(TEXT_CHROMA_U) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2763	::memcpy( m_puhQTTempTransformSkipFlag[2], pcCU->getTransformSkip(TEXT_CHROMA_V) + uiPartOffset, uiQPartNum * sizeof( UChar ) );
2764	}
2765	} // Mode loop
2766	#endif
2767
2768	//--- update overall distortion ---
2769	uiOverallDistY += uiBestPUDistY;
2770	uiOverallDistC += uiBestPUDistC;
2771
2772	//--- update transform index and cbf ---
2773	UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth(0) + uiInitTrDepth ) << 1 );
2774	::memcpy( pcCU->getTransformIdx() + uiPartOffset, m_puhQTTempTrIdx, uiQPartNum * sizeof( UChar ) );
2775	::memcpy( pcCU->getCbf( TEXT_LUMA ) + uiPartOffset, m_puhQTTempCbf[0], uiQPartNum * sizeof( UChar ) );
2776	::memcpy( pcCU->getCbf( TEXT_CHROMA_U ) + uiPartOffset, m_puhQTTempCbf[1], uiQPartNum * sizeof( UChar ) );
2777	::memcpy( pcCU->getCbf( TEXT_CHROMA_V ) + uiPartOffset, m_puhQTTempCbf[2], uiQPartNum * sizeof( UChar ) );
2778	::memcpy( pcCU->getTransformSkip(TEXT_LUMA) + uiPartOffset, m_puhQTTempTransformSkipFlag[0], uiQPartNum * sizeof( UChar ) );
2779	::memcpy( pcCU->getTransformSkip(TEXT_CHROMA_U) + uiPartOffset, m_puhQTTempTransformSkipFlag[1], uiQPartNum * sizeof( UChar ) );
2780	::memcpy( pcCU->getTransformSkip(TEXT_CHROMA_V) + uiPartOffset, m_puhQTTempTransformSkipFlag[2], uiQPartNum * sizeof( UChar ) );
2781	//--- set reconstruction for next intra prediction blocks ---
2782	if( uiPU != uiNumPU - 1 )
2783	{
2784	Bool bSkipChroma = false;
2785	Bool bChromaSame = false;
2786	UInt uiLog2TrSize = g_aucConvertToBit[ pcCU->getSlice()->getSPS()->getMaxCUWidth() >> ( pcCU->getDepth(0) + uiInitTrDepth ) ] + 2;
2787	if( !bLumaOnly && uiLog2TrSize == 2 )
2788	{
2789	assert( uiInitTrDepth > 0 );
2790	bSkipChroma = ( uiPU != 0 );
2791	bChromaSame = true;
2792	}
2793
2794	UInt uiCompWidth = pcCU->getWidth ( 0 ) >> uiInitTrDepth;
2795	UInt uiCompHeight = pcCU->getHeight( 0 ) >> uiInitTrDepth;
2796	UInt uiZOrder = pcCU->getZorderIdxInCU() + uiPartOffset;
2797	Pel* piDes = pcCU->getPic()->getPicYuvRec()->getLumaAddr( pcCU->getAddr(), uiZOrder );
2798	UInt uiDesStride = pcCU->getPic()->getPicYuvRec()->getStride();
2799	Pel* piSrc = pcRecoYuv->getLumaAddr( uiPartOffset );
2800	UInt uiSrcStride = pcRecoYuv->getStride();
2801	for( UInt uiY = 0; uiY < uiCompHeight; uiY++, piSrc += uiSrcStride, piDes += uiDesStride )
2802	{
2803	for( UInt uiX = 0; uiX < uiCompWidth; uiX++ )
2804	{
2805	piDes[ uiX ] = piSrc[ uiX ];
2806	}
2807	}
2808	#if H_3D_VSO // M38
2809	// set model
2810	if( m_pcRdCost->getUseRenModel() )
2811	{
2812	piSrc = pcRecoYuv->getLumaAddr( uiPartOffset );
2813	m_pcRdCost->setRenModelData( pcCU, uiPartOffset, piSrc, uiSrcStride, uiCompWidth, uiCompHeight);
2814	}
2815	#endif
2816	if( !bLumaOnly && !bSkipChroma )
2817	{
2818	if( !bChromaSame )
2819	{
2820	uiCompWidth >>= 1;
2821	uiCompHeight >>= 1;
2822	}
2823	piDes = pcCU->getPic()->getPicYuvRec()->getCbAddr( pcCU->getAddr(), uiZOrder );
2824	uiDesStride = pcCU->getPic()->getPicYuvRec()->getCStride();
2825	piSrc = pcRecoYuv->getCbAddr( uiPartOffset );
2826	uiSrcStride = pcRecoYuv->getCStride();
2827	for( UInt uiY = 0; uiY < uiCompHeight; uiY++, piSrc += uiSrcStride, piDes += uiDesStride )
2828	{
2829	for( UInt uiX = 0; uiX < uiCompWidth; uiX++ )
2830	{
2831	piDes[ uiX ] = piSrc[ uiX ];
2832	}
2833	}
2834	piDes = pcCU->getPic()->getPicYuvRec()->getCrAddr( pcCU->getAddr(), uiZOrder );
2835	piSrc = pcRecoYuv->getCrAddr( uiPartOffset );
2836	for( UInt uiY = 0; uiY < uiCompHeight; uiY++, piSrc += uiSrcStride, piDes += uiDesStride )
2837	{
2838	for( UInt uiX = 0; uiX < uiCompWidth; uiX++ )
2839	{
2840	piDes[ uiX ] = piSrc[ uiX ];
2841	}
2842	}
2843	}
2844	}
2845
2846	//=== update PU data ====
2847	pcCU->setLumaIntraDirSubParts ( uiBestPUMode, uiPartOffset, uiDepth + uiInitTrDepth );
2848	pcCU->copyToPic ( uiDepth, uiPU, uiInitTrDepth );
2849	} // PU loop
2850
2851
2852	if( uiNumPU > 1 )
2853	{ // set Cbf for all blocks
2854	UInt uiCombCbfY = 0;
2855	UInt uiCombCbfU = 0;
2856	UInt uiCombCbfV = 0;
2857	UInt uiPartIdx = 0;
2858	for( UInt uiPart = 0; uiPart < 4; uiPart++, uiPartIdx += uiQNumParts )
2859	{
2860	uiCombCbfY \|= pcCU->getCbf( uiPartIdx, TEXT_LUMA, 1 );
2861	uiCombCbfU \|= pcCU->getCbf( uiPartIdx, TEXT_CHROMA_U, 1 );
2862	uiCombCbfV \|= pcCU->getCbf( uiPartIdx, TEXT_CHROMA_V, 1 );
2863	}
2864	for( UInt uiOffs = 0; uiOffs < 4 * uiQNumParts; uiOffs++ )
2865	{
2866	pcCU->getCbf( TEXT_LUMA )[ uiOffs ] \|= uiCombCbfY;
2867	pcCU->getCbf( TEXT_CHROMA_U )[ uiOffs ] \|= uiCombCbfU;
2868	pcCU->getCbf( TEXT_CHROMA_V )[ uiOffs ] \|= uiCombCbfV;
2869	}
2870	}
2871
2872	//===== reset context models =====
2873	if(m_bUseSBACRD)
2874	{
2875	m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST]);
2876	}
2877
2878	//===== set distortion (rate and r-d costs are determined later) =====
2879	ruiDistC = uiOverallDistC;
2880	pcCU->getTotalDistortion() = uiOverallDistY + uiOverallDistC;
2881	}
2882
2883
2884
2885	Void
2886	TEncSearch::estIntraPredChromaQT( TComDataCU* pcCU,
2887	TComYuv* pcOrgYuv,
2888	TComYuv* pcPredYuv,
2889	TComYuv* pcResiYuv,
2890	TComYuv* pcRecoYuv,
2891	UInt uiPreCalcDistC )
2892	{
2893	UInt uiDepth = pcCU->getDepth(0);
2894	UInt uiBestMode = 0;
2895	UInt uiBestDist = 0;
2896	Double dBestCost = MAX_DOUBLE;
2897
2898	//----- init mode list -----
2899	UInt uiMinMode = 0;
2900	UInt uiModeList[ NUM_CHROMA_MODE ];
2901	pcCU->getAllowedChromaDir( 0, uiModeList );
2902	UInt uiMaxMode = NUM_CHROMA_MODE;
2903
2904	//----- check chroma modes -----
2905	for( UInt uiMode = uiMinMode; uiMode < uiMaxMode; uiMode++ )
2906	{
2907	//----- restore context models -----
2908	if( m_bUseSBACRD )
2909	{
2910	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST] );
2911	}
2912
2913	//----- chroma coding -----
2914	UInt uiDist = 0;
2915	pcCU->setChromIntraDirSubParts ( uiModeList[uiMode], 0, uiDepth );
2916	xRecurIntraChromaCodingQT ( pcCU, 0, 0, pcOrgYuv, pcPredYuv, pcResiYuv, uiDist );
2917	if( m_bUseSBACRD && pcCU->getSlice()->getPPS()->getUseTransformSkip() )
2918	{
2919	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST] );
2920	}
2921	UInt uiBits = xGetIntraBitsQT( pcCU, 0, 0, false, true, false );
2922	Double dCost = m_pcRdCost->calcRdCost( uiBits, uiDist );
2923
2924	//----- compare -----
2925	if( dCost < dBestCost )
2926	{
2927	dBestCost = dCost;
2928	uiBestDist = uiDist;
2929	uiBestMode = uiModeList[uiMode];
2930	UInt uiQPN = pcCU->getPic()->getNumPartInCU() >> ( uiDepth << 1 );
2931	xSetIntraResultChromaQT( pcCU, 0, 0, pcRecoYuv );
2932	::memcpy( m_puhQTTempCbf[1], pcCU->getCbf( TEXT_CHROMA_U ), uiQPN * sizeof( UChar ) );
2933	::memcpy( m_puhQTTempCbf[2], pcCU->getCbf( TEXT_CHROMA_V ), uiQPN * sizeof( UChar ) );
2934	::memcpy( m_puhQTTempTransformSkipFlag[1], pcCU->getTransformSkip( TEXT_CHROMA_U ), uiQPN * sizeof( UChar ) );
2935	::memcpy( m_puhQTTempTransformSkipFlag[2], pcCU->getTransformSkip( TEXT_CHROMA_V ), uiQPN * sizeof( UChar ) );
2936	}
2937	}
2938
2939	//----- set data -----
2940	UInt uiQPN = pcCU->getPic()->getNumPartInCU() >> ( uiDepth << 1 );
2941	::memcpy( pcCU->getCbf( TEXT_CHROMA_U ), m_puhQTTempCbf[1], uiQPN * sizeof( UChar ) );
2942	::memcpy( pcCU->getCbf( TEXT_CHROMA_V ), m_puhQTTempCbf[2], uiQPN * sizeof( UChar ) );
2943	::memcpy( pcCU->getTransformSkip( TEXT_CHROMA_U ), m_puhQTTempTransformSkipFlag[1], uiQPN * sizeof( UChar ) );
2944	::memcpy( pcCU->getTransformSkip( TEXT_CHROMA_V ), m_puhQTTempTransformSkipFlag[2], uiQPN * sizeof( UChar ) );
2945	pcCU->setChromIntraDirSubParts( uiBestMode, 0, uiDepth );
2946	pcCU->getTotalDistortion () += uiBestDist - uiPreCalcDistC;
2947
2948	//----- restore context models -----
2949	if( m_bUseSBACRD )
2950	{
2951	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST] );
2952	}
2953	}
2954
2955	/** Function for encoding and reconstructing luma/chroma samples of a PCM mode CU.
2956	* \param pcCU pointer to current CU
2957	* \param uiAbsPartIdx part index
2958	* \param piOrg pointer to original sample arrays
2959	* \param piPCM pointer to PCM code arrays
2960	* \param piPred pointer to prediction signal arrays
2961	* \param piResi pointer to residual signal arrays
2962	* \param piReco pointer to reconstructed sample arrays
2963	* \param uiStride stride of the original/prediction/residual sample arrays
2964	* \param uiWidth block width
2965	* \param uiHeight block height
2966	* \param ttText texture component type
2967	* \returns Void
2968	*/
2969	Void TEncSearch::xEncPCM (TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piOrg, Pel* piPCM, Pel* piPred, Pel* piResi, Pel* piReco, UInt uiStride, UInt uiWidth, UInt uiHeight, TextType eText )
2970	{
2971	UInt uiX, uiY;
2972	UInt uiReconStride;
2973	Pel* pOrg = piOrg;
2974	Pel* pPCM = piPCM;
2975	Pel* pPred = piPred;
2976	Pel* pResi = piResi;
2977	Pel* pReco = piReco;
2978	Pel* pRecoPic;
2979	Int shiftPcm;
2980
2981	if( eText == TEXT_LUMA)
2982	{
2983	uiReconStride = pcCU->getPic()->getPicYuvRec()->getStride();
2984	pRecoPic = pcCU->getPic()->getPicYuvRec()->getLumaAddr(pcCU->getAddr(), pcCU->getZorderIdxInCU()+uiAbsPartIdx);
2985	shiftPcm = g_bitDepthY - pcCU->getSlice()->getSPS()->getPCMBitDepthLuma();
2986	}
2987	else
2988	{
2989	uiReconStride = pcCU->getPic()->getPicYuvRec()->getCStride();
2990
2991	if( eText == TEXT_CHROMA_U )
2992	{
2993	pRecoPic = pcCU->getPic()->getPicYuvRec()->getCbAddr(pcCU->getAddr(), pcCU->getZorderIdxInCU()+uiAbsPartIdx);
2994	}
2995	else
2996	{
2997	pRecoPic = pcCU->getPic()->getPicYuvRec()->getCrAddr(pcCU->getAddr(), pcCU->getZorderIdxInCU()+uiAbsPartIdx);
2998	}
2999	shiftPcm = g_bitDepthC - pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
3000	}
3001
3002	// Reset pred and residual
3003	for( uiY = 0; uiY < uiHeight; uiY++ )
3004	{
3005	for( uiX = 0; uiX < uiWidth; uiX++ )
3006	{
3007	pPred[uiX] = 0;
3008	pResi[uiX] = 0;
3009	}
3010	pPred += uiStride;
3011	pResi += uiStride;
3012	}
3013
3014	// Encode
3015	for( uiY = 0; uiY < uiHeight; uiY++ )
3016	{
3017	for( uiX = 0; uiX < uiWidth; uiX++ )
3018	{
3019	pPCM[uiX] = pOrg[uiX]>> shiftPcm;
3020	}
3021	pPCM += uiWidth;
3022	pOrg += uiStride;
3023	}
3024
3025	pPCM = piPCM;
3026
3027	// Reconstruction
3028	for( uiY = 0; uiY < uiHeight; uiY++ )
3029	{
3030	for( uiX = 0; uiX < uiWidth; uiX++ )
3031	{
3032	pReco [uiX] = pPCM[uiX]<< shiftPcm;
3033	pRecoPic[uiX] = pReco[uiX];
3034	}
3035	pPCM += uiWidth;
3036	pReco += uiStride;
3037	pRecoPic += uiReconStride;
3038	}
3039	}
3040
3041	/** Function for PCM mode estimation.
3042	* \param pcCU
3043	* \param pcOrgYuv
3044	* \param rpcPredYuv
3045	* \param rpcResiYuv
3046	* \param rpcRecoYuv
3047	* \returns Void
3048	*/
3049	Void TEncSearch::IPCMSearch( TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv& rpcPredYuv, TComYuv& rpcResiYuv, TComYuv*& rpcRecoYuv )
3050	{
3051	UInt uiDepth = pcCU->getDepth(0);
3052	UInt uiWidth = pcCU->getWidth(0);
3053	UInt uiHeight = pcCU->getHeight(0);
3054	UInt uiStride = rpcPredYuv->getStride();
3055	UInt uiStrideC = rpcPredYuv->getCStride();
3056	UInt uiWidthC = uiWidth >> 1;
3057	UInt uiHeightC = uiHeight >> 1;
3058	UInt uiDistortion = 0;
3059	UInt uiBits;
3060
3061	Double dCost;
3062
3063	Pel* pOrig;
3064	Pel* pResi;
3065	Pel* pReco;
3066	Pel* pPred;
3067	Pel* pPCM;
3068
3069	UInt uiAbsPartIdx = 0;
3070
3071	UInt uiMinCoeffSize = pcCU->getPic()->getMinCUWidth()*pcCU->getPic()->getMinCUHeight();
3072	UInt uiLumaOffset = uiMinCoeffSize*uiAbsPartIdx;
3073	UInt uiChromaOffset = uiLumaOffset>>2;
3074
3075	// Luminance
3076	pOrig = pcOrgYuv->getLumaAddr(0, uiWidth);
3077	pResi = rpcResiYuv->getLumaAddr(0, uiWidth);
3078	pPred = rpcPredYuv->getLumaAddr(0, uiWidth);
3079	pReco = rpcRecoYuv->getLumaAddr(0, uiWidth);
3080	pPCM = pcCU->getPCMSampleY() + uiLumaOffset;
3081
3082	xEncPCM ( pcCU, 0, pOrig, pPCM, pPred, pResi, pReco, uiStride, uiWidth, uiHeight, TEXT_LUMA );
3083
3084	// Chroma U
3085	pOrig = pcOrgYuv->getCbAddr();
3086	pResi = rpcResiYuv->getCbAddr();
3087	pPred = rpcPredYuv->getCbAddr();
3088	pReco = rpcRecoYuv->getCbAddr();
3089	pPCM = pcCU->getPCMSampleCb() + uiChromaOffset;
3090
3091	xEncPCM ( pcCU, 0, pOrig, pPCM, pPred, pResi, pReco, uiStrideC, uiWidthC, uiHeightC, TEXT_CHROMA_U );
3092
3093	// Chroma V
3094	pOrig = pcOrgYuv->getCrAddr();
3095	pResi = rpcResiYuv->getCrAddr();
3096	pPred = rpcPredYuv->getCrAddr();
3097	pReco = rpcRecoYuv->getCrAddr();
3098	pPCM = pcCU->getPCMSampleCr() + uiChromaOffset;
3099
3100	xEncPCM ( pcCU, 0, pOrig, pPCM, pPred, pResi, pReco, uiStrideC, uiWidthC, uiHeightC, TEXT_CHROMA_V );
3101
3102	m_pcEntropyCoder->resetBits();
3103	xEncIntraHeader ( pcCU, uiDepth, uiAbsPartIdx, true, false);
3104	uiBits = m_pcEntropyCoder->getNumberOfWrittenBits();
3105
3106	#if H_3D_VSO // M43
3107	if( m_pcRdCost->getUseLambdaScaleVSO() )
3108	dCost = m_pcRdCost->calcRdCostVSO( uiBits, uiDistortion );
3109	else
3110	#endif
3111	dCost = m_pcRdCost->calcRdCost( uiBits, uiDistortion );
3112
3113	if(m_bUseSBACRD)
3114	{
3115	m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[uiDepth][CI_CURR_BEST]);
3116	}
3117
3118	pcCU->getTotalBits() = uiBits;
3119	pcCU->getTotalCost() = dCost;
3120	pcCU->getTotalDistortion() = uiDistortion;
3121
3122	pcCU->copyToPic(uiDepth, 0, 0);
3123	}
3124
3125	Void TEncSearch::xGetInterPredictionError( TComDataCU* pcCU, TComYuv* pcYuvOrg, Int iPartIdx, UInt& ruiErr, Bool bHadamard )
3126	{
3127	motionCompensation( pcCU, &m_tmpYuvPred, REF_PIC_LIST_X, iPartIdx );
3128
3129	UInt uiAbsPartIdx = 0;
3130	Int iWidth = 0;
3131	Int iHeight = 0;
3132	pcCU->getPartIndexAndSize( iPartIdx, uiAbsPartIdx, iWidth, iHeight );
3133
3134	DistParam cDistParam;
3135
3136	cDistParam.bApplyWeight = false;
3137
3138	m_pcRdCost->setDistParam( cDistParam, g_bitDepthY,
3139	pcYuvOrg->getLumaAddr( uiAbsPartIdx ), pcYuvOrg->getStride(),
3140	m_tmpYuvPred .getLumaAddr( uiAbsPartIdx ), m_tmpYuvPred .getStride(),
3141	#if NS_HAD
3142	iWidth, iHeight, m_pcEncCfg->getUseHADME(), m_pcEncCfg->getUseNSQT() );
3143	#else
3144	iWidth, iHeight, m_pcEncCfg->getUseHADME() );
3145	#endif
3146	ruiErr = cDistParam.DistFunc( &cDistParam );
3147	}
3148
3149	/** estimation of best merge coding
3150	* \param pcCU
3151	* \param pcYuvOrg
3152	* \param iPUIdx
3153	* \param uiInterDir
3154	* \param pacMvField
3155	* \param uiMergeIndex
3156	* \param ruiCost
3157	* \param ruiBits
3158	* \param puhNeighCands
3159	* \param bValid
3160	* \returns Void
3161	*/
3162	Void TEncSearch::xMergeEstimation( TComDataCU* pcCU, TComYuv* pcYuvOrg, Int iPUIdx, UInt& uiInterDir, TComMvField* pacMvField, UInt& uiMergeIndex, UInt& ruiCost, TComMvField* cMvFieldNeighbours, UChar* uhInterDirNeighbours, Int& numValidMergeCand )
3163	{
3164	UInt uiAbsPartIdx = 0;
3165	Int iWidth = 0;
3166	Int iHeight = 0;
3167
3168	pcCU->getPartIndexAndSize( iPUIdx, uiAbsPartIdx, iWidth, iHeight );
3169	UInt uiDepth = pcCU->getDepth( uiAbsPartIdx );
3170	PartSize partSize = pcCU->getPartitionSize( 0 );
3171	if ( pcCU->getSlice()->getPPS()->getLog2ParallelMergeLevelMinus2() && partSize != SIZE_2Nx2N && pcCU->getWidth( 0 ) <= 8 )
3172	{
3173	pcCU->setPartSizeSubParts( SIZE_2Nx2N, 0, uiDepth );
3174	if ( iPUIdx == 0 )
3175	{
3176	pcCU->getInterMergeCandidates( 0, 0, cMvFieldNeighbours,uhInterDirNeighbours, numValidMergeCand );
3177	}
3178	pcCU->setPartSizeSubParts( partSize, 0, uiDepth );
3179	}
3180	else
3181	{
3182	pcCU->getInterMergeCandidates( uiAbsPartIdx, iPUIdx, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand );
3183	}
3184	xRestrictBipredMergeCand( pcCU, iPUIdx, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand );
3185
3186	ruiCost = MAX_UINT;
3187	for( UInt uiMergeCand = 0; uiMergeCand < numValidMergeCand; ++uiMergeCand )
3188	{
3189	{
3190	UInt uiCostCand = MAX_UINT;
3191	UInt uiBitsCand = 0;
3192
3193	PartSize ePartSize = pcCU->getPartitionSize( 0 );
3194
3195	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvField( cMvFieldNeighbours[0 + 2*uiMergeCand], ePartSize, uiAbsPartIdx, 0, iPUIdx );
3196	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvField( cMvFieldNeighbours[1 + 2*uiMergeCand], ePartSize, uiAbsPartIdx, 0, iPUIdx );
3197
3198	xGetInterPredictionError( pcCU, pcYuvOrg, iPUIdx, uiCostCand, m_pcEncCfg->getUseHADME() );
3199	uiBitsCand = uiMergeCand + 1;
3200	if (uiMergeCand == m_pcEncCfg->getMaxNumMergeCand() -1)
3201	{
3202	uiBitsCand--;
3203	}
3204	uiCostCand = uiCostCand + m_pcRdCost->getCost( uiBitsCand );
3205	if ( uiCostCand < ruiCost )
3206	{
3207	ruiCost = uiCostCand;
3208	pacMvField[0] = cMvFieldNeighbours[0 + 2*uiMergeCand];
3209	pacMvField[1] = cMvFieldNeighbours[1 + 2*uiMergeCand];
3210	uiInterDir = uhInterDirNeighbours[uiMergeCand];
3211	uiMergeIndex = uiMergeCand;
3212	}
3213	}
3214	}
3215	}
3216
3217	/** convert bi-pred merge candidates to uni-pred
3218	* \param pcCU
3219	* \param puIdx
3220	* \param mvFieldNeighbours
3221	* \param interDirNeighbours
3222	* \param numValidMergeCand
3223	* \returns Void
3224	*/
3225	Void TEncSearch::xRestrictBipredMergeCand( TComDataCU* pcCU, UInt puIdx, TComMvField* mvFieldNeighbours, UChar* interDirNeighbours, Int numValidMergeCand )
3226	{
3227	if ( pcCU->isBipredRestriction(puIdx) )
3228	{
3229	for( UInt mergeCand = 0; mergeCand < numValidMergeCand; ++mergeCand )
3230	{
3231	if ( interDirNeighbours[mergeCand] == 3 )
3232	{
3233	interDirNeighbours[mergeCand] = 1;
3234	mvFieldNeighbours[(mergeCand << 1) + 1].setMvField(TComMv(0,0), -1);
3235	}
3236	}
3237	}
3238	}
3239
3240	/** search of the best candidate for inter prediction
3241	* \param pcCU
3242	* \param pcOrgYuv
3243	* \param rpcPredYuv
3244	* \param rpcResiYuv
3245	* \param rpcRecoYuv
3246	* \param bUseRes
3247	* \returns Void
3248	*/
3249	#if AMP_MRG
3250	Void TEncSearch::predInterSearch( TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv& rpcPredYuv, TComYuv& rpcResiYuv, TComYuv*& rpcRecoYuv, Bool bUseRes, Bool bUseMRG )
3251	#else
3252	Void TEncSearch::predInterSearch( TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv& rpcPredYuv, TComYuv& rpcResiYuv, TComYuv*& rpcRecoYuv, Bool bUseRes )
3253	#endif
3254	{
3255	m_acYuvPred[0].clear();
3256	m_acYuvPred[1].clear();
3257	m_cYuvPredTemp.clear();
3258	rpcPredYuv->clear();
3259
3260	if ( !bUseRes )
3261	{
3262	rpcResiYuv->clear();
3263	}
3264
3265	rpcRecoYuv->clear();
3266
3267	TComMv cMvSrchRngLT;
3268	TComMv cMvSrchRngRB;
3269
3270	TComMv cMvZero;
3271	TComMv TempMv; //kolya
3272
3273	TComMv cMv[2];
3274	TComMv cMvBi[2];
3275	TComMv cMvTemp[2][33];
3276
3277	Int iNumPart = pcCU->getNumPartInter();
3278	Int iNumPredDir = pcCU->getSlice()->isInterP() ? 1 : 2;
3279
3280	TComMv cMvPred[2][33];
3281
3282	TComMv cMvPredBi[2][33];
3283	Int aaiMvpIdxBi[2][33];
3284
3285	Int aaiMvpIdx[2][33];
3286	Int aaiMvpNum[2][33];
3287
3288	AMVPInfo aacAMVPInfo[2][33];
3289
3290	Int iRefIdx[2]={0,0}; //If un-initialized, may cause SEGV in bi-directional prediction iterative stage.
3291	Int iRefIdxBi[2];
3292
3293	UInt uiPartAddr;
3294	Int iRoiWidth, iRoiHeight;
3295
3296	UInt uiMbBits[3] = {1, 1, 0};
3297
3298	UInt uiLastMode = 0;
3299	Int iRefStart, iRefEnd;
3300
3301	PartSize ePartSize = pcCU->getPartitionSize( 0 );
3302
3303	Int bestBiPRefIdxL1 = 0;
3304	Int bestBiPMvpL1 = 0;
3305	UInt biPDistTemp = MAX_INT;
3306
3307	#if ZERO_MVD_EST
3308	Int aiZeroMvdMvpIdx[2] = {-1, -1};
3309	Int aiZeroMvdRefIdx[2] = {0, 0};
3310	Int iZeroMvdDir = -1;
3311	#endif
3312
3313	TComMvField cMvFieldNeighbours[MRG_MAX_NUM_CANDS << 1]; // double length for mv of both lists
3314	UChar uhInterDirNeighbours[MRG_MAX_NUM_CANDS];
3315	Int numValidMergeCand = 0 ;
3316
3317	for ( Int iPartIdx = 0; iPartIdx < iNumPart; iPartIdx++ )
3318	{
3319	UInt uiCost[2] = { MAX_UINT, MAX_UINT };
3320	UInt uiCostBi = MAX_UINT;
3321	UInt uiCostTemp;
3322
3323	UInt uiBits[3];
3324	UInt uiBitsTemp;
3325	#if ZERO_MVD_EST
3326	UInt uiZeroMvdCost = MAX_UINT;
3327	UInt uiZeroMvdCostTemp;
3328	UInt uiZeroMvdBitsTemp;
3329	UInt uiZeroMvdDistTemp = MAX_UINT;
3330	UInt auiZeroMvdBits[3];
3331	#endif
3332	UInt bestBiPDist = MAX_INT;
3333
3334	UInt uiCostTempL0[MAX_NUM_REF];
3335	for (Int iNumRef=0; iNumRef < MAX_NUM_REF; iNumRef++) uiCostTempL0[iNumRef] = MAX_UINT;
3336	UInt uiBitsTempL0[MAX_NUM_REF];
3337
3338	#if L0034_COMBINED_LIST_CLEANUP
3339	TComMv mvValidList1;
3340	Int refIdxValidList1 = 0;
3341	UInt bitsValidList1 = MAX_UINT;
3342	UInt costValidList1 = MAX_UINT;
3343	#endif
3344
3345	xGetBlkBits( ePartSize, pcCU->getSlice()->isInterP(), iPartIdx, uiLastMode, uiMbBits);
3346
3347	pcCU->getPartIndexAndSize( iPartIdx, uiPartAddr, iRoiWidth, iRoiHeight );
3348
3349	#if AMP_MRG
3350	Bool bTestNormalMC = true;
3351
3352	if ( bUseMRG && pcCU->getWidth( 0 ) > 8 && iNumPart == 2 )
3353	{
3354	bTestNormalMC = false;
3355	}
3356
3357	if (bTestNormalMC)
3358	{
3359	#endif
3360
3361	// Uni-directional prediction
3362	for ( Int iRefList = 0; iRefList < iNumPredDir; iRefList++ )
3363	{
3364	RefPicList eRefPicList = ( iRefList ? REF_PIC_LIST_1 : REF_PIC_LIST_0 );
3365
3366	for ( Int iRefIdxTemp = 0; iRefIdxTemp < pcCU->getSlice()->getNumRefIdx(eRefPicList); iRefIdxTemp++ )
3367	{
3368	uiBitsTemp = uiMbBits[iRefList];
3369	if ( pcCU->getSlice()->getNumRefIdx(eRefPicList) > 1 )
3370	{
3371	uiBitsTemp += iRefIdxTemp+1;
3372	if ( iRefIdxTemp == pcCU->getSlice()->getNumRefIdx(eRefPicList)-1 ) uiBitsTemp--;
3373	}
3374	#if ZERO_MVD_EST
3375	xEstimateMvPredAMVP( pcCU, pcOrgYuv, iPartIdx, eRefPicList, iRefIdxTemp, cMvPred[iRefList][iRefIdxTemp], false, &biPDistTemp, &uiZeroMvdDistTemp);
3376	#else
3377	xEstimateMvPredAMVP( pcCU, pcOrgYuv, iPartIdx, eRefPicList, iRefIdxTemp, cMvPred[iRefList][iRefIdxTemp], false, &biPDistTemp);
3378	#endif
3379	aaiMvpIdx[iRefList][iRefIdxTemp] = pcCU->getMVPIdx(eRefPicList, uiPartAddr);
3380	aaiMvpNum[iRefList][iRefIdxTemp] = pcCU->getMVPNum(eRefPicList, uiPartAddr);
3381
3382	if(pcCU->getSlice()->getMvdL1ZeroFlag() && iRefList==1 && biPDistTemp < bestBiPDist)
3383	{
3384	bestBiPDist = biPDistTemp;
3385	bestBiPMvpL1 = aaiMvpIdx[iRefList][iRefIdxTemp];
3386	bestBiPRefIdxL1 = iRefIdxTemp;
3387	}
3388
3389	uiBitsTemp += m_auiMVPIdxCost[aaiMvpIdx[iRefList][iRefIdxTemp]][AMVP_MAX_NUM_CANDS];
3390	#if ZERO_MVD_EST
3391	#if L0034_COMBINED_LIST_CLEANUP
3392	if ( iRefList == 0 \|\| pcCU->getSlice()->getList1IdxToList0Idx( iRefIdxTemp ) < 0 )
3393	#else
3394	if ((iRefList != 1 \|\| !pcCU->getSlice()->getNoBackPredFlag()) &&
3395	(pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) <= 0 \|\| pcCU->getSlice()->getRefIdxOfLC(eRefPicList, iRefIdxTemp)>=0))
3396	#endif
3397	{
3398	uiZeroMvdBitsTemp = uiBitsTemp;
3399	uiZeroMvdBitsTemp += 2; //zero mvd bits
3400
3401	m_pcRdCost->getMotionCost( 1, 0 );
3402	uiZeroMvdCostTemp = uiZeroMvdDistTemp + m_pcRdCost->getCost(uiZeroMvdBitsTemp);
3403
3404	if (uiZeroMvdCostTemp < uiZeroMvdCost)
3405	{
3406	uiZeroMvdCost = uiZeroMvdCostTemp;
3407	iZeroMvdDir = iRefList + 1;
3408	aiZeroMvdRefIdx[iRefList] = iRefIdxTemp;
3409	aiZeroMvdMvpIdx[iRefList] = aaiMvpIdx[iRefList][iRefIdxTemp];
3410	auiZeroMvdBits[iRefList] = uiZeroMvdBitsTemp;
3411	}
3412	}
3413	#endif
3414
3415	#if GPB_SIMPLE_UNI
3416	#if L0034_COMBINED_LIST_CLEANUP
3417	if ( iRefList == 1 ) // list 1
3418	{
3419	if ( pcCU->getSlice()->getList1IdxToList0Idx( iRefIdxTemp ) >= 0 )
3420	{
3421	cMvTemp[1][iRefIdxTemp] = cMvTemp[0][pcCU->getSlice()->getList1IdxToList0Idx( iRefIdxTemp )];
3422	uiCostTemp = uiCostTempL0[pcCU->getSlice()->getList1IdxToList0Idx( iRefIdxTemp )];
3423	/first subtract the bit-rate part of the cost of the other list/
3424	uiCostTemp -= m_pcRdCost->getCost( uiBitsTempL0[pcCU->getSlice()->getList1IdxToList0Idx( iRefIdxTemp )] );
3425	/correct the bit-rate part of the current ref/
3426	m_pcRdCost->setPredictor ( cMvPred[iRefList][iRefIdxTemp] );
3427	uiBitsTemp += m_pcRdCost->getBits( cMvTemp[1][iRefIdxTemp].getHor(), cMvTemp[1][iRefIdxTemp].getVer() );
3428	/calculate the correct cost/
3429	uiCostTemp += m_pcRdCost->getCost( uiBitsTemp );
3430	}
3431	else
3432	{
3433	xMotionEstimation ( pcCU, pcOrgYuv, iPartIdx, eRefPicList, &cMvPred[iRefList][iRefIdxTemp], iRefIdxTemp, cMvTemp[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp );
3434	}
3435	}
3436	else
3437	{
3438	xMotionEstimation ( pcCU, pcOrgYuv, iPartIdx, eRefPicList, &cMvPred[iRefList][iRefIdxTemp], iRefIdxTemp, cMvTemp[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp );
3439	}
3440	#else
3441	if ( pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0)
3442	{
3443	if ( iRefList && ( pcCU->getSlice()->getNoBackPredFlag() \|\| (pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0 && !pcCU->getSlice()->getNoBackPredFlag() && pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(iRefIdxTemp)>=0 ) ) )
3444	{
3445	if ( pcCU->getSlice()->getNoBackPredFlag() )
3446	{
3447	cMvTemp[1][iRefIdxTemp] = cMvTemp[0][iRefIdxTemp];
3448	uiCostTemp = uiCostTempL0[iRefIdxTemp];
3449	/first subtract the bit-rate part of the cost of the other list/
3450	uiCostTemp -= m_pcRdCost->getCost( uiBitsTempL0[iRefIdxTemp] );
3451	}
3452	else
3453	{
3454	cMvTemp[1][iRefIdxTemp] = cMvTemp[0][pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(iRefIdxTemp)];
3455	uiCostTemp = uiCostTempL0[pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(iRefIdxTemp)];
3456	/first subtract the bit-rate part of the cost of the other list/
3457	uiCostTemp -= m_pcRdCost->getCost( uiBitsTempL0[pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(iRefIdxTemp)] );
3458	}
3459	/correct the bit-rate part of the current ref/
3460	m_pcRdCost->setPredictor ( cMvPred[iRefList][iRefIdxTemp] );
3461	uiBitsTemp += m_pcRdCost->getBits( cMvTemp[1][iRefIdxTemp].getHor(), cMvTemp[1][iRefIdxTemp].getVer() );
3462	/calculate the correct cost/
3463	uiCostTemp += m_pcRdCost->getCost( uiBitsTemp );
3464	}
3465	else
3466	{
3467	xMotionEstimation ( pcCU, pcOrgYuv, iPartIdx, eRefPicList, &cMvPred[iRefList][iRefIdxTemp], iRefIdxTemp, cMvTemp[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp );
3468	}
3469	}
3470	else
3471	{
3472	if (iRefList && pcCU->getSlice()->getNoBackPredFlag())
3473	{
3474	uiCostTemp = MAX_UINT;
3475	cMvTemp[1][iRefIdxTemp] = cMvTemp[0][iRefIdxTemp];
3476	}
3477	else
3478	{
3479	xMotionEstimation ( pcCU, pcOrgYuv, iPartIdx, eRefPicList, &cMvPred[iRefList][iRefIdxTemp], iRefIdxTemp, cMvTemp[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp );
3480	}
3481	}
3482	#endif
3483	#else
3484	xMotionEstimation ( pcCU, pcOrgYuv, iPartIdx, eRefPicList, &cMvPred[iRefList][iRefIdxTemp], iRefIdxTemp, cMvTemp[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp );
3485	#endif
3486	xCopyAMVPInfo(pcCU->getCUMvField(eRefPicList)->getAMVPInfo(), &aacAMVPInfo[iRefList][iRefIdxTemp]); // must always be done ( also when AMVP_MODE = AM_NONE )
3487	xCheckBestMVP(pcCU, eRefPicList, cMvTemp[iRefList][iRefIdxTemp], cMvPred[iRefList][iRefIdxTemp], aaiMvpIdx[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp);
3488
3489	#if L0034_COMBINED_LIST_CLEANUP
3490	if ( iRefList == 0 )
3491	{
3492	uiCostTempL0[iRefIdxTemp] = uiCostTemp;
3493	uiBitsTempL0[iRefIdxTemp] = uiBitsTemp;
3494	}
3495	if ( uiCostTemp < uiCost[iRefList] )
3496	{
3497	uiCost[iRefList] = uiCostTemp;
3498	uiBits[iRefList] = uiBitsTemp; // storing for bi-prediction
3499
3500	// set motion
3501	cMv[iRefList] = cMvTemp[iRefList][iRefIdxTemp];
3502	iRefIdx[iRefList] = iRefIdxTemp;
3503	}
3504
3505	if ( iRefList == 1 && uiCostTemp < costValidList1 && pcCU->getSlice()->getList1IdxToList0Idx( iRefIdxTemp ) < 0 )
3506	{
3507	costValidList1 = uiCostTemp;
3508	bitsValidList1 = uiBitsTemp;
3509
3510	// set motion
3511	mvValidList1 = cMvTemp[iRefList][iRefIdxTemp];
3512	refIdxValidList1 = iRefIdxTemp;
3513	}
3514	#else
3515	if(pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0 && !pcCU->getSlice()->getNoBackPredFlag())
3516	{
3517	if(iRefList==REF_PIC_LIST_0)
3518	{
3519	uiCostTempL0[iRefIdxTemp] = uiCostTemp;
3520	uiBitsTempL0[iRefIdxTemp] = uiBitsTemp;
3521	if(pcCU->getSlice()->getRefIdxOfLC(REF_PIC_LIST_0, iRefIdxTemp)<0)
3522	{
3523	uiCostTemp = MAX_UINT;
3524	}
3525	}
3526	else
3527	{
3528	if(pcCU->getSlice()->getRefIdxOfLC(REF_PIC_LIST_1, iRefIdxTemp)<0)
3529	{
3530	uiCostTemp = MAX_UINT;
3531	}
3532	}
3533	}
3534
3535	if ( ( iRefList == 0 && uiCostTemp < uiCost[iRefList] ) \|\|
3536	( iRefList == 1 && pcCU->getSlice()->getNoBackPredFlag() && iRefIdxTemp == iRefIdx[0] ) \|\|
3537	( iRefList == 1 && (pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0) && (iRefIdxTemp==0 \|\| iRefIdxTemp == iRefIdx[0]) && !pcCU->getSlice()->getNoBackPredFlag() && (iRefIdxTemp == pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(iRefIdxTemp)) ) \|\|
3538	( iRefList == 1 && !pcCU->getSlice()->getNoBackPredFlag() && uiCostTemp < uiCost[iRefList] ) )
3539	{
3540	uiCost[iRefList] = uiCostTemp;
3541	uiBits[iRefList] = uiBitsTemp; // storing for bi-prediction
3542
3543	// set motion
3544	cMv[iRefList] = cMvTemp[iRefList][iRefIdxTemp];
3545	iRefIdx[iRefList] = iRefIdxTemp;
3546	pcCU->getCUMvField(eRefPicList)->setAllMv( cMv[iRefList], ePartSize, uiPartAddr, 0, iPartIdx );
3547	pcCU->getCUMvField(eRefPicList)->setAllRefIdx( iRefIdx[iRefList], ePartSize, uiPartAddr, 0, iPartIdx );
3548
3549	if(!pcCU->getSlice()->getMvdL1ZeroFlag())
3550	{
3551	// storing list 1 prediction signal for iterative bi-directional prediction
3552	if ( eRefPicList == REF_PIC_LIST_1 )
3553	{
3554	TComYuv* pcYuvPred = &m_acYuvPred[iRefList];
3555	motionCompensation ( pcCU, pcYuvPred, eRefPicList, iPartIdx );
3556	}
3557	if ( (pcCU->getSlice()->getNoBackPredFlag() \|\| (pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0 && pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(0)==0 )) && eRefPicList == REF_PIC_LIST_0 )
3558	{
3559	TComYuv* pcYuvPred = &m_acYuvPred[iRefList];
3560	motionCompensation ( pcCU, pcYuvPred, eRefPicList, iPartIdx );
3561	}
3562	}
3563	}
3564	#endif
3565	}
3566	}
3567	// Bi-directional prediction
3568	if ( (pcCU->getSlice()->isInterB()) && (pcCU->isBipredRestriction(iPartIdx) == false) )
3569	{
3570
3571	cMvBi[0] = cMv[0]; cMvBi[1] = cMv[1];
3572	iRefIdxBi[0] = iRefIdx[0]; iRefIdxBi[1] = iRefIdx[1];
3573
3574	::memcpy(cMvPredBi, cMvPred, sizeof(cMvPred));
3575	::memcpy(aaiMvpIdxBi, aaiMvpIdx, sizeof(aaiMvpIdx));
3576
3577	UInt uiMotBits[2];
3578
3579	if(pcCU->getSlice()->getMvdL1ZeroFlag())
3580	{
3581	xCopyAMVPInfo(&aacAMVPInfo[1][bestBiPRefIdxL1], pcCU->getCUMvField(REF_PIC_LIST_1)->getAMVPInfo());
3582	pcCU->setMVPIdxSubParts( bestBiPMvpL1, REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3583	aaiMvpIdxBi[1][bestBiPRefIdxL1] = bestBiPMvpL1;
3584	cMvPredBi[1][bestBiPRefIdxL1] = pcCU->getCUMvField(REF_PIC_LIST_1)->getAMVPInfo()->m_acMvCand[bestBiPMvpL1];
3585
3586	cMvBi[1] = cMvPredBi[1][bestBiPRefIdxL1];
3587	iRefIdxBi[1] = bestBiPRefIdxL1;
3588	pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllMv( cMvBi[1], ePartSize, uiPartAddr, 0, iPartIdx );
3589	pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllRefIdx( iRefIdxBi[1], ePartSize, uiPartAddr, 0, iPartIdx );
3590	TComYuv* pcYuvPred = &m_acYuvPred[1];
3591	motionCompensation( pcCU, pcYuvPred, REF_PIC_LIST_1, iPartIdx );
3592
3593	uiMotBits[0] = uiBits[0] - uiMbBits[0];
3594	uiMotBits[1] = uiMbBits[1];
3595
3596	if ( pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_1) > 1 )
3597	{
3598	uiMotBits[1] += bestBiPRefIdxL1+1;
3599	if ( bestBiPRefIdxL1 == pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_1)-1 ) uiMotBits[1]--;
3600	}
3601
3602	uiMotBits[1] += m_auiMVPIdxCost[aaiMvpIdxBi[1][bestBiPRefIdxL1]][AMVP_MAX_NUM_CANDS];
3603
3604	uiBits[2] = uiMbBits[2] + uiMotBits[0] + uiMotBits[1];
3605
3606	cMvTemp[1][bestBiPRefIdxL1] = cMvBi[1];
3607	}
3608	else
3609	{
3610	uiMotBits[0] = uiBits[0] - uiMbBits[0];
3611	uiMotBits[1] = uiBits[1] - uiMbBits[1];
3612	uiBits[2] = uiMbBits[2] + uiMotBits[0] + uiMotBits[1];
3613	}
3614
3615	// 4-times iteration (default)
3616	Int iNumIter = 4;
3617
3618	// fast encoder setting: only one iteration
3619	if ( m_pcEncCfg->getUseFastEnc() \|\| pcCU->getSlice()->getMvdL1ZeroFlag())
3620	{
3621	iNumIter = 1;
3622	}
3623
3624	for ( Int iIter = 0; iIter < iNumIter; iIter++ )
3625	{
3626
3627	Int iRefList = iIter % 2;
3628	#if L0034_COMBINED_LIST_CLEANUP
3629	if ( m_pcEncCfg->getUseFastEnc() )
3630	{
3631	if( uiCost[0] <= uiCost[1] )
3632	{
3633	iRefList = 1;
3634	}
3635	else
3636	{
3637	iRefList = 0;
3638	}
3639	}
3640	else if ( iIter == 0 )
3641	{
3642	iRefList = 0;
3643	}
3644	if ( iIter == 0 && !pcCU->getSlice()->getMvdL1ZeroFlag())
3645	{
3646	pcCU->getCUMvField(RefPicList(1-iRefList))->setAllMv( cMv[1-iRefList], ePartSize, uiPartAddr, 0, iPartIdx );
3647	pcCU->getCUMvField(RefPicList(1-iRefList))->setAllRefIdx( iRefIdx[1-iRefList], ePartSize, uiPartAddr, 0, iPartIdx );
3648	TComYuv* pcYuvPred = &m_acYuvPred[1-iRefList];
3649	motionCompensation ( pcCU, pcYuvPred, RefPicList(1-iRefList), iPartIdx );
3650	}
3651	#else
3652	if ( m_pcEncCfg->getUseFastEnc() && (pcCU->getSlice()->getNoBackPredFlag() \|\| (pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0 && pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(0)==0 )) )
3653	{
3654	iRefList = 1;
3655	}
3656	#endif
3657	RefPicList eRefPicList = ( iRefList ? REF_PIC_LIST_1 : REF_PIC_LIST_0 );
3658
3659	if(pcCU->getSlice()->getMvdL1ZeroFlag())
3660	{
3661	iRefList = 0;
3662	eRefPicList = REF_PIC_LIST_0;
3663	}
3664
3665	Bool bChanged = false;
3666
3667	iRefStart = 0;
3668	iRefEnd = pcCU->getSlice()->getNumRefIdx(eRefPicList)-1;
3669
3670	for ( Int iRefIdxTemp = iRefStart; iRefIdxTemp <= iRefEnd; iRefIdxTemp++ )
3671	{
3672	uiBitsTemp = uiMbBits[2] + uiMotBits[1-iRefList];
3673	if ( pcCU->getSlice()->getNumRefIdx(eRefPicList) > 1 )
3674	{
3675	uiBitsTemp += iRefIdxTemp+1;
3676	if ( iRefIdxTemp == pcCU->getSlice()->getNumRefIdx(eRefPicList)-1 ) uiBitsTemp--;
3677	}
3678	uiBitsTemp += m_auiMVPIdxCost[aaiMvpIdxBi[iRefList][iRefIdxTemp]][AMVP_MAX_NUM_CANDS];
3679	// call ME
3680	xMotionEstimation ( pcCU, pcOrgYuv, iPartIdx, eRefPicList, &cMvPredBi[iRefList][iRefIdxTemp], iRefIdxTemp, cMvTemp[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp, true );
3681	xCopyAMVPInfo(&aacAMVPInfo[iRefList][iRefIdxTemp], pcCU->getCUMvField(eRefPicList)->getAMVPInfo());
3682	xCheckBestMVP(pcCU, eRefPicList, cMvTemp[iRefList][iRefIdxTemp], cMvPredBi[iRefList][iRefIdxTemp], aaiMvpIdxBi[iRefList][iRefIdxTemp], uiBitsTemp, uiCostTemp);
3683
3684	if ( uiCostTemp < uiCostBi )
3685	{
3686	bChanged = true;
3687
3688	cMvBi[iRefList] = cMvTemp[iRefList][iRefIdxTemp];
3689	iRefIdxBi[iRefList] = iRefIdxTemp;
3690
3691	uiCostBi = uiCostTemp;
3692	uiMotBits[iRefList] = uiBitsTemp - uiMbBits[2] - uiMotBits[1-iRefList];
3693	uiBits[2] = uiBitsTemp;
3694
3695	if(iNumIter!=1)
3696	{
3697	// Set motion
3698	pcCU->getCUMvField( eRefPicList )->setAllMv( cMvBi[iRefList], ePartSize, uiPartAddr, 0, iPartIdx );
3699	pcCU->getCUMvField( eRefPicList )->setAllRefIdx( iRefIdxBi[iRefList], ePartSize, uiPartAddr, 0, iPartIdx );
3700
3701	TComYuv* pcYuvPred = &m_acYuvPred[iRefList];
3702	motionCompensation( pcCU, pcYuvPred, eRefPicList, iPartIdx );
3703	}
3704	}
3705	} // for loop-iRefIdxTemp
3706
3707	if ( !bChanged )
3708	{
3709	if ( uiCostBi <= uiCost[0] && uiCostBi <= uiCost[1] )
3710	{
3711	xCopyAMVPInfo(&aacAMVPInfo[0][iRefIdxBi[0]], pcCU->getCUMvField(REF_PIC_LIST_0)->getAMVPInfo());
3712	xCheckBestMVP(pcCU, REF_PIC_LIST_0, cMvBi[0], cMvPredBi[0][iRefIdxBi[0]], aaiMvpIdxBi[0][iRefIdxBi[0]], uiBits[2], uiCostBi);
3713	if(!pcCU->getSlice()->getMvdL1ZeroFlag())
3714	{
3715	xCopyAMVPInfo(&aacAMVPInfo[1][iRefIdxBi[1]], pcCU->getCUMvField(REF_PIC_LIST_1)->getAMVPInfo());
3716	xCheckBestMVP(pcCU, REF_PIC_LIST_1, cMvBi[1], cMvPredBi[1][iRefIdxBi[1]], aaiMvpIdxBi[1][iRefIdxBi[1]], uiBits[2], uiCostBi);
3717	}
3718	}
3719	break;
3720	}
3721	} // for loop-iter
3722	} // if (B_SLICE)
3723	#if ZERO_MVD_EST
3724	if ( (pcCU->getSlice()->isInterB()) && (pcCU->isBipredRestriction(iPartIdx) == false) )
3725	{
3726	m_pcRdCost->getMotionCost( 1, 0 );
3727
3728	for ( Int iL0RefIdxTemp = 0; iL0RefIdxTemp <= pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_0)-1; iL0RefIdxTemp++ )
3729	for ( Int iL1RefIdxTemp = 0; iL1RefIdxTemp <= pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_1)-1; iL1RefIdxTemp++ )
3730	{
3731	UInt uiRefIdxBitsTemp = 0;
3732	if ( pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_0) > 1 )
3733	{
3734	uiRefIdxBitsTemp += iL0RefIdxTemp+1;
3735	if ( iL0RefIdxTemp == pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_0)-1 ) uiRefIdxBitsTemp--;
3736	}
3737	if ( pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_1) > 1 )
3738	{
3739	uiRefIdxBitsTemp += iL1RefIdxTemp+1;
3740	if ( iL1RefIdxTemp == pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_1)-1 ) uiRefIdxBitsTemp--;
3741	}
3742
3743	Int iL0MVPIdx = 0;
3744	Int iL1MVPIdx = 0;
3745
3746	for (iL0MVPIdx = 0; iL0MVPIdx < aaiMvpNum[0][iL0RefIdxTemp]; iL0MVPIdx++)
3747	{
3748	for (iL1MVPIdx = 0; iL1MVPIdx < aaiMvpNum[1][iL1RefIdxTemp]; iL1MVPIdx++)
3749	{
3750	uiZeroMvdBitsTemp = uiRefIdxBitsTemp;
3751	uiZeroMvdBitsTemp += uiMbBits[2];
3752	uiZeroMvdBitsTemp += m_auiMVPIdxCost[iL0MVPIdx][aaiMvpNum[0][iL0RefIdxTemp]] + m_auiMVPIdxCost[iL1MVPIdx][aaiMvpNum[1][iL1RefIdxTemp]];
3753	uiZeroMvdBitsTemp += 4; //zero mvd for both directions
3754	pcCU->getCUMvField( REF_PIC_LIST_0 )->setAllMvField( aacAMVPInfo[0][iL0RefIdxTemp].m_acMvCand[iL0MVPIdx], iL0RefIdxTemp, ePartSize, uiPartAddr, iPartIdx, 0 );
3755	pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllMvField( aacAMVPInfo[1][iL1RefIdxTemp].m_acMvCand[iL1MVPIdx], iL1RefIdxTemp, ePartSize, uiPartAddr, iPartIdx, 0 );
3756
3757	xGetInterPredictionError( pcCU, pcOrgYuv, iPartIdx, uiZeroMvdDistTemp, m_pcEncCfg->getUseHADME() );
3758	uiZeroMvdCostTemp = uiZeroMvdDistTemp + m_pcRdCost->getCost( uiZeroMvdBitsTemp );
3759	if (uiZeroMvdCostTemp < uiZeroMvdCost)
3760	{
3761	uiZeroMvdCost = uiZeroMvdCostTemp;
3762	iZeroMvdDir = 3;
3763	aiZeroMvdMvpIdx[0] = iL0MVPIdx;
3764	aiZeroMvdMvpIdx[1] = iL1MVPIdx;
3765	aiZeroMvdRefIdx[0] = iL0RefIdxTemp;
3766	aiZeroMvdRefIdx[1] = iL1RefIdxTemp;
3767	auiZeroMvdBits[2] = uiZeroMvdBitsTemp;
3768	}
3769	}
3770	}
3771	}
3772	}
3773	#endif
3774
3775	#if AMP_MRG
3776	} //end if bTestNormalMC
3777	#endif
3778	// Clear Motion Field
3779	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvField( TComMvField(), ePartSize, uiPartAddr, 0, iPartIdx );
3780	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvField( TComMvField(), ePartSize, uiPartAddr, 0, iPartIdx );
3781	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvd ( cMvZero, ePartSize, uiPartAddr, 0, iPartIdx );
3782	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvd ( cMvZero, ePartSize, uiPartAddr, 0, iPartIdx );
3783
3784	pcCU->setMVPIdxSubParts( -1, REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3785	pcCU->setMVPNumSubParts( -1, REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3786	pcCU->setMVPIdxSubParts( -1, REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3787	pcCU->setMVPNumSubParts( -1, REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3788
3789	UInt uiMEBits = 0;
3790	// Set Motion Field_
3791	#if L0034_COMBINED_LIST_CLEANUP
3792	cMv[1] = mvValidList1;
3793	iRefIdx[1] = refIdxValidList1;
3794	uiBits[1] = bitsValidList1;
3795	uiCost[1] = costValidList1;
3796	#else
3797	if ( pcCU->getSlice()->getNoBackPredFlag() \|\| (pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0 && pcCU->getSlice()->getRefIdxOfL0FromRefIdxOfL1(0)==0 ) )
3798	{
3799	uiCost[1] = MAX_UINT;
3800	}
3801	#endif
3802	#if AMP_MRG
3803	if (bTestNormalMC)
3804	{
3805	#endif
3806	#if ZERO_MVD_EST
3807	if (uiZeroMvdCost <= uiCostBi && uiZeroMvdCost <= uiCost[0] && uiZeroMvdCost <= uiCost[1])
3808	{
3809	if (iZeroMvdDir == 3)
3810	{
3811	uiLastMode = 2;
3812
3813	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvField( aacAMVPInfo[0][aiZeroMvdRefIdx[0]].m_acMvCand[aiZeroMvdMvpIdx[0]], aiZeroMvdRefIdx[0], ePartSize, uiPartAddr, iPartIdx, 0 );
3814	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvField( aacAMVPInfo[1][aiZeroMvdRefIdx[1]].m_acMvCand[aiZeroMvdMvpIdx[1]], aiZeroMvdRefIdx[1], ePartSize, uiPartAddr, iPartIdx, 0 );
3815
3816	pcCU->setInterDirSubParts( 3, uiPartAddr, iPartIdx, pcCU->getDepth(0) );
3817
3818	pcCU->setMVPIdxSubParts( aiZeroMvdMvpIdx[0], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3819	pcCU->setMVPNumSubParts( aaiMvpNum[0][aiZeroMvdRefIdx[0]], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3820	pcCU->setMVPIdxSubParts( aiZeroMvdMvpIdx[1], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3821	pcCU->setMVPNumSubParts( aaiMvpNum[1][aiZeroMvdRefIdx[1]], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3822	uiMEBits = auiZeroMvdBits[2];
3823	}
3824	else if (iZeroMvdDir == 1)
3825	{
3826	uiLastMode = 0;
3827
3828	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvField( aacAMVPInfo[0][aiZeroMvdRefIdx[0]].m_acMvCand[aiZeroMvdMvpIdx[0]], aiZeroMvdRefIdx[0], ePartSize, uiPartAddr, iPartIdx, 0 );
3829
3830	pcCU->setInterDirSubParts( 1, uiPartAddr, iPartIdx, pcCU->getDepth(0) );
3831
3832	pcCU->setMVPIdxSubParts( aiZeroMvdMvpIdx[0], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3833	pcCU->setMVPNumSubParts( aaiMvpNum[0][aiZeroMvdRefIdx[0]], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3834	uiMEBits = auiZeroMvdBits[0];
3835	}
3836	else if (iZeroMvdDir == 2)
3837	{
3838	uiLastMode = 1;
3839
3840	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvField( aacAMVPInfo[1][aiZeroMvdRefIdx[1]].m_acMvCand[aiZeroMvdMvpIdx[1]], aiZeroMvdRefIdx[1], ePartSize, uiPartAddr, iPartIdx, 0 );
3841
3842	pcCU->setInterDirSubParts( 2, uiPartAddr, iPartIdx, pcCU->getDepth(0) );
3843
3844	pcCU->setMVPIdxSubParts( aiZeroMvdMvpIdx[1], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3845	pcCU->setMVPNumSubParts( aaiMvpNum[1][aiZeroMvdRefIdx[1]], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3846	uiMEBits = auiZeroMvdBits[1];
3847	}
3848	else
3849	{
3850	assert(0);
3851	}
3852	}
3853	else
3854	#endif
3855	if ( uiCostBi <= uiCost[0] && uiCostBi <= uiCost[1])
3856	{
3857	uiLastMode = 2;
3858	{
3859	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMv( cMvBi[0], ePartSize, uiPartAddr, 0, iPartIdx );
3860	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllRefIdx( iRefIdxBi[0], ePartSize, uiPartAddr, 0, iPartIdx );
3861	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMv( cMvBi[1], ePartSize, uiPartAddr, 0, iPartIdx );
3862	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllRefIdx( iRefIdxBi[1], ePartSize, uiPartAddr, 0, iPartIdx );
3863	}
3864	{
3865	TempMv = cMvBi[0] - cMvPredBi[0][iRefIdxBi[0]];
3866	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvd ( TempMv, ePartSize, uiPartAddr, 0, iPartIdx );
3867	}
3868	{
3869	TempMv = cMvBi[1] - cMvPredBi[1][iRefIdxBi[1]];
3870	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvd ( TempMv, ePartSize, uiPartAddr, 0, iPartIdx );
3871	}
3872
3873	pcCU->setInterDirSubParts( 3, uiPartAddr, iPartIdx, pcCU->getDepth(0) );
3874
3875	pcCU->setMVPIdxSubParts( aaiMvpIdxBi[0][iRefIdxBi[0]], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3876	pcCU->setMVPNumSubParts( aaiMvpNum[0][iRefIdxBi[0]], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3877	pcCU->setMVPIdxSubParts( aaiMvpIdxBi[1][iRefIdxBi[1]], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3878	pcCU->setMVPNumSubParts( aaiMvpNum[1][iRefIdxBi[1]], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3879
3880	uiMEBits = uiBits[2];
3881	}
3882	else if ( uiCost[0] <= uiCost[1] )
3883	{
3884	uiLastMode = 0;
3885	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMv( cMv[0], ePartSize, uiPartAddr, 0, iPartIdx );
3886	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllRefIdx( iRefIdx[0], ePartSize, uiPartAddr, 0, iPartIdx );
3887	{
3888	TempMv = cMv[0] - cMvPred[0][iRefIdx[0]];
3889	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvd ( TempMv, ePartSize, uiPartAddr, 0, iPartIdx );
3890	}
3891	pcCU->setInterDirSubParts( 1, uiPartAddr, iPartIdx, pcCU->getDepth(0) );
3892
3893	pcCU->setMVPIdxSubParts( aaiMvpIdx[0][iRefIdx[0]], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3894	pcCU->setMVPNumSubParts( aaiMvpNum[0][iRefIdx[0]], REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3895
3896	uiMEBits = uiBits[0];
3897	}
3898	else
3899	{
3900	uiLastMode = 1;
3901	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMv( cMv[1], ePartSize, uiPartAddr, 0, iPartIdx );
3902	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllRefIdx( iRefIdx[1], ePartSize, uiPartAddr, 0, iPartIdx );
3903	{
3904	TempMv = cMv[1] - cMvPred[1][iRefIdx[1]];
3905	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvd ( TempMv, ePartSize, uiPartAddr, 0, iPartIdx );
3906	}
3907	pcCU->setInterDirSubParts( 2, uiPartAddr, iPartIdx, pcCU->getDepth(0) );
3908
3909	pcCU->setMVPIdxSubParts( aaiMvpIdx[1][iRefIdx[1]], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3910	pcCU->setMVPNumSubParts( aaiMvpNum[1][iRefIdx[1]], REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3911
3912	uiMEBits = uiBits[1];
3913	}
3914	#if AMP_MRG
3915	} // end if bTestNormalMC
3916	#endif
3917
3918	if ( pcCU->getPartitionSize( uiPartAddr ) != SIZE_2Nx2N )
3919	{
3920	UInt uiMRGInterDir = 0;
3921	TComMvField cMRGMvField[2];
3922	UInt uiMRGIndex = 0;
3923
3924	UInt uiMEInterDir = 0;
3925	TComMvField cMEMvField[2];
3926
3927	m_pcRdCost->getMotionCost( 1, 0 );
3928	#if AMP_MRG
3929	// calculate ME cost
3930	UInt uiMEError = MAX_UINT;
3931	UInt uiMECost = MAX_UINT;
3932
3933	if (bTestNormalMC)
3934	{
3935	xGetInterPredictionError( pcCU, pcOrgYuv, iPartIdx, uiMEError, m_pcEncCfg->getUseHADME() );
3936	uiMECost = uiMEError + m_pcRdCost->getCost( uiMEBits );
3937	}
3938	#else
3939	// calculate ME cost
3940	UInt uiMEError = MAX_UINT;
3941	xGetInterPredictionError( pcCU, pcOrgYuv, iPartIdx, uiMEError, m_pcEncCfg->getUseHADME() );
3942	UInt uiMECost = uiMEError + m_pcRdCost->getCost( uiMEBits );
3943	#endif
3944	// save ME result.
3945	uiMEInterDir = pcCU->getInterDir( uiPartAddr );
3946	pcCU->getMvField( pcCU, uiPartAddr, REF_PIC_LIST_0, cMEMvField[0] );
3947	pcCU->getMvField( pcCU, uiPartAddr, REF_PIC_LIST_1, cMEMvField[1] );
3948
3949	// find Merge result
3950	UInt uiMRGCost = MAX_UINT;
3951	xMergeEstimation( pcCU, pcOrgYuv, iPartIdx, uiMRGInterDir, cMRGMvField, uiMRGIndex, uiMRGCost, cMvFieldNeighbours, uhInterDirNeighbours, numValidMergeCand);
3952	if ( uiMRGCost < uiMECost )
3953	{
3954	// set Merge result
3955	pcCU->setMergeFlagSubParts ( true, uiPartAddr, iPartIdx, pcCU->getDepth( uiPartAddr ) );
3956	pcCU->setMergeIndexSubParts( uiMRGIndex, uiPartAddr, iPartIdx, pcCU->getDepth( uiPartAddr ) );
3957	pcCU->setInterDirSubParts ( uiMRGInterDir, uiPartAddr, iPartIdx, pcCU->getDepth( uiPartAddr ) );
3958	{
3959	pcCU->getCUMvField( REF_PIC_LIST_0 )->setAllMvField( cMRGMvField[0], ePartSize, uiPartAddr, 0, iPartIdx );
3960	pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllMvField( cMRGMvField[1], ePartSize, uiPartAddr, 0, iPartIdx );
3961	}
3962
3963	pcCU->getCUMvField(REF_PIC_LIST_0)->setAllMvd ( cMvZero, ePartSize, uiPartAddr, 0, iPartIdx );
3964	pcCU->getCUMvField(REF_PIC_LIST_1)->setAllMvd ( cMvZero, ePartSize, uiPartAddr, 0, iPartIdx );
3965
3966	pcCU->setMVPIdxSubParts( -1, REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3967	pcCU->setMVPNumSubParts( -1, REF_PIC_LIST_0, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3968	pcCU->setMVPIdxSubParts( -1, REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3969	pcCU->setMVPNumSubParts( -1, REF_PIC_LIST_1, uiPartAddr, iPartIdx, pcCU->getDepth(uiPartAddr));
3970	}
3971	else
3972	{
3973	// set ME result
3974	pcCU->setMergeFlagSubParts( false, uiPartAddr, iPartIdx, pcCU->getDepth( uiPartAddr ) );
3975	pcCU->setInterDirSubParts ( uiMEInterDir, uiPartAddr, iPartIdx, pcCU->getDepth( uiPartAddr ) );
3976	{
3977	pcCU->getCUMvField( REF_PIC_LIST_0 )->setAllMvField( cMEMvField[0], ePartSize, uiPartAddr, 0, iPartIdx );
3978	pcCU->getCUMvField( REF_PIC_LIST_1 )->setAllMvField( cMEMvField[1], ePartSize, uiPartAddr, 0, iPartIdx );
3979	}
3980	}
3981	}
3982
3983	// MC
3984	motionCompensation ( pcCU, rpcPredYuv, REF_PIC_LIST_X, iPartIdx );
3985
3986	} // end of for ( Int iPartIdx = 0; iPartIdx < iNumPart; iPartIdx++ )
3987
3988	setWpScalingDistParam( pcCU, -1, REF_PIC_LIST_X );
3989
3990	return;
3991	}
3992
3993	// AMVP
3994	#if ZERO_MVD_EST
3995	Void TEncSearch::xEstimateMvPredAMVP( TComDataCU* pcCU, TComYuv* pcOrgYuv, UInt uiPartIdx, RefPicList eRefPicList, Int iRefIdx, TComMv& rcMvPred, Bool bFilled, UInt* puiDistBiP, UInt* puiDist )
3996	#else
3997	Void TEncSearch::xEstimateMvPredAMVP( TComDataCU* pcCU, TComYuv* pcOrgYuv, UInt uiPartIdx, RefPicList eRefPicList, Int iRefIdx, TComMv& rcMvPred, Bool bFilled, UInt* puiDistBiP )
3998	#endif
3999	{
4000	AMVPInfo* pcAMVPInfo = pcCU->getCUMvField(eRefPicList)->getAMVPInfo();
4001
4002	TComMv cBestMv;
4003	Int iBestIdx = 0;
4004	TComMv cZeroMv;
4005	TComMv cMvPred;
4006	UInt uiBestCost = MAX_INT;
4007	UInt uiPartAddr = 0;
4008	Int iRoiWidth, iRoiHeight;
4009	Int i;
4010
4011	pcCU->getPartIndexAndSize( uiPartIdx, uiPartAddr, iRoiWidth, iRoiHeight );
4012	// Fill the MV Candidates
4013	if (!bFilled)
4014	{
4015	pcCU->fillMvpCand( uiPartIdx, uiPartAddr, eRefPicList, iRefIdx, pcAMVPInfo );
4016	}
4017
4018	// initialize Mvp index & Mvp
4019	iBestIdx = 0;
4020	cBestMv = pcAMVPInfo->m_acMvCand[0];
4021	#if !ZERO_MVD_EST
4022	if (pcAMVPInfo->iN <= 1)
4023	{
4024	rcMvPred = cBestMv;
4025
4026	pcCU->setMVPIdxSubParts( iBestIdx, eRefPicList, uiPartAddr, uiPartIdx, pcCU->getDepth(uiPartAddr));
4027	pcCU->setMVPNumSubParts( pcAMVPInfo->iN, eRefPicList, uiPartAddr, uiPartIdx, pcCU->getDepth(uiPartAddr));
4028
4029	if(pcCU->getSlice()->getMvdL1ZeroFlag() && eRefPicList==REF_PIC_LIST_1)
4030	{
4031	#if ZERO_MVD_EST
4032	(*puiDistBiP) = xGetTemplateCost( pcCU, uiPartIdx, uiPartAddr, pcOrgYuv, &m_cYuvPredTemp, rcMvPred, 0, AMVP_MAX_NUM_CANDS, eRefPicList, iRefIdx, iRoiWidth, iRoiHeight, uiDist );
4033	#else
4034	(*puiDistBiP) = xGetTemplateCost( pcCU, uiPartIdx, uiPartAddr, pcOrgYuv, &m_cYuvPredTemp, rcMvPred, 0, AMVP_MAX_NUM_CANDS, eRefPicList, iRefIdx, iRoiWidth, iRoiHeight);
4035	#endif
4036	}
4037	return;
4038	}
4039	#endif
4040	if (bFilled)
4041	{
4042	assert(pcCU->getMVPIdx(eRefPicList,uiPartAddr) >= 0);
4043	rcMvPred = pcAMVPInfo->m_acMvCand[pcCU->getMVPIdx(eRefPicList,uiPartAddr)];
4044	return;
4045	}
4046
4047	m_cYuvPredTemp.clear();
4048	#if ZERO_MVD_EST
4049	UInt uiDist;
4050	#endif
4051	//-- Check Minimum Cost.
4052	for ( i = 0 ; i < pcAMVPInfo->iN; i++)
4053	{
4054	UInt uiTmpCost;
4055	#if ZERO_MVD_EST
4056	uiTmpCost = xGetTemplateCost( pcCU, uiPartIdx, uiPartAddr, pcOrgYuv, &m_cYuvPredTemp, pcAMVPInfo->m_acMvCand[i], i, AMVP_MAX_NUM_CANDS, eRefPicList, iRefIdx, iRoiWidth, iRoiHeight, uiDist );
4057	#else
4058	uiTmpCost = xGetTemplateCost( pcCU, uiPartIdx, uiPartAddr, pcOrgYuv, &m_cYuvPredTemp, pcAMVPInfo->m_acMvCand[i], i, AMVP_MAX_NUM_CANDS, eRefPicList, iRefIdx, iRoiWidth, iRoiHeight);
4059	#endif
4060	if ( uiBestCost > uiTmpCost )
4061	{
4062	uiBestCost = uiTmpCost;
4063	cBestMv = pcAMVPInfo->m_acMvCand[i];
4064	iBestIdx = i;
4065	(*puiDistBiP) = uiTmpCost;
4066	#if ZERO_MVD_EST
4067	(*puiDist) = uiDist;
4068	#endif
4069	}
4070	}
4071
4072	m_cYuvPredTemp.clear();
4073
4074	// Setting Best MVP
4075	rcMvPred = cBestMv;
4076	pcCU->setMVPIdxSubParts( iBestIdx, eRefPicList, uiPartAddr, uiPartIdx, pcCU->getDepth(uiPartAddr));
4077	pcCU->setMVPNumSubParts( pcAMVPInfo->iN, eRefPicList, uiPartAddr, uiPartIdx, pcCU->getDepth(uiPartAddr));
4078	return;
4079	}
4080
4081	UInt TEncSearch::xGetMvpIdxBits(Int iIdx, Int iNum)
4082	{
4083	assert(iIdx >= 0 && iNum >= 0 && iIdx < iNum);
4084
4085	if (iNum == 1)
4086	return 0;
4087
4088	UInt uiLength = 1;
4089	Int iTemp = iIdx;
4090	if ( iTemp == 0 )
4091	{
4092	return uiLength;
4093	}
4094
4095	Bool bCodeLast = ( iNum-1 > iTemp );
4096
4097	uiLength += (iTemp-1);
4098
4099	if( bCodeLast )
4100	{
4101	uiLength++;
4102	}
4103
4104	return uiLength;
4105	}
4106
4107	Void TEncSearch::xGetBlkBits( PartSize eCUMode, Bool bPSlice, Int iPartIdx, UInt uiLastMode, UInt uiBlkBit[3])
4108	{
4109	if ( eCUMode == SIZE_2Nx2N )
4110	{
4111	uiBlkBit[0] = (! bPSlice) ? 3 : 1;
4112	uiBlkBit[1] = 3;
4113	uiBlkBit[2] = 5;
4114	}
4115	else if ( (eCUMode == SIZE_2NxN \|\| eCUMode == SIZE_2NxnU) \|\| eCUMode == SIZE_2NxnD )
4116	{
4117	UInt aauiMbBits[2][3][3] = { { {0,0,3}, {0,0,0}, {0,0,0} } , { {5,7,7}, {7,5,7}, {9-3,9-3,9-3} } };
4118	if ( bPSlice )
4119	{
4120	uiBlkBit[0] = 3;
4121	uiBlkBit[1] = 0;
4122	uiBlkBit[2] = 0;
4123	}
4124	else
4125	{
4126	::memcpy( uiBlkBit, aauiMbBits[iPartIdx][uiLastMode], 3*sizeof(UInt) );
4127	}
4128	}
4129	else if ( (eCUMode == SIZE_Nx2N \|\| eCUMode == SIZE_nLx2N) \|\| eCUMode == SIZE_nRx2N )
4130	{
4131	UInt aauiMbBits[2][3][3] = { { {0,2,3}, {0,0,0}, {0,0,0} } , { {5,7,7}, {7-2,7-2,9-2}, {9-3,9-3,9-3} } };
4132	if ( bPSlice )
4133	{
4134	uiBlkBit[0] = 3;
4135	uiBlkBit[1] = 0;
4136	uiBlkBit[2] = 0;
4137	}
4138	else
4139	{
4140	::memcpy( uiBlkBit, aauiMbBits[iPartIdx][uiLastMode], 3*sizeof(UInt) );
4141	}
4142	}
4143	else if ( eCUMode == SIZE_NxN )
4144	{
4145	uiBlkBit[0] = (! bPSlice) ? 3 : 1;
4146	uiBlkBit[1] = 3;
4147	uiBlkBit[2] = 5;
4148	}
4149	else
4150	{
4151	printf("Wrong!\n");
4152	assert( 0 );
4153	}
4154	}
4155
4156	Void TEncSearch::xCopyAMVPInfo (AMVPInfo* pSrc, AMVPInfo* pDst)
4157	{
4158	pDst->iN = pSrc->iN;
4159	for (Int i = 0; i < pSrc->iN; i++)
4160	{
4161	pDst->m_acMvCand[i] = pSrc->m_acMvCand[i];
4162	}
4163	}
4164
4165	Void TEncSearch::xCheckBestMVP ( TComDataCU* pcCU, RefPicList eRefPicList, TComMv cMv, TComMv& rcMvPred, Int& riMVPIdx, UInt& ruiBits, UInt& ruiCost )
4166	{
4167	AMVPInfo* pcAMVPInfo = pcCU->getCUMvField(eRefPicList)->getAMVPInfo();
4168
4169	assert(pcAMVPInfo->m_acMvCand[riMVPIdx] == rcMvPred);
4170
4171	if (pcAMVPInfo->iN < 2) return;
4172
4173	m_pcRdCost->getMotionCost( 1, 0 );
4174	m_pcRdCost->setCostScale ( 0 );
4175
4176	Int iBestMVPIdx = riMVPIdx;
4177
4178	m_pcRdCost->setPredictor( rcMvPred );
4179	Int iOrgMvBits = m_pcRdCost->getBits(cMv.getHor(), cMv.getVer());
4180	iOrgMvBits += m_auiMVPIdxCost[riMVPIdx][AMVP_MAX_NUM_CANDS];
4181	Int iBestMvBits = iOrgMvBits;
4182
4183	for (Int iMVPIdx = 0; iMVPIdx < pcAMVPInfo->iN; iMVPIdx++)
4184	{
4185	if (iMVPIdx == riMVPIdx) continue;
4186
4187	m_pcRdCost->setPredictor( pcAMVPInfo->m_acMvCand[iMVPIdx] );
4188
4189	Int iMvBits = m_pcRdCost->getBits(cMv.getHor(), cMv.getVer());
4190	iMvBits += m_auiMVPIdxCost[iMVPIdx][AMVP_MAX_NUM_CANDS];
4191
4192	if (iMvBits < iBestMvBits)
4193	{
4194	iBestMvBits = iMvBits;
4195	iBestMVPIdx = iMVPIdx;
4196	}
4197	}
4198
4199	if (iBestMVPIdx != riMVPIdx) //if changed
4200	{
4201	rcMvPred = pcAMVPInfo->m_acMvCand[iBestMVPIdx];
4202
4203	riMVPIdx = iBestMVPIdx;
4204	UInt uiOrgBits = ruiBits;
4205	ruiBits = uiOrgBits - iOrgMvBits + iBestMvBits;
4206	ruiCost = (ruiCost - m_pcRdCost->getCost( uiOrgBits )) + m_pcRdCost->getCost( ruiBits );
4207	}
4208	}
4209
4210	UInt TEncSearch::xGetTemplateCost( TComDataCU* pcCU,
4211	UInt uiPartIdx,
4212	UInt uiPartAddr,
4213	TComYuv* pcOrgYuv,
4214	TComYuv* pcTemplateCand,
4215	TComMv cMvCand,
4216	Int iMVPIdx,
4217	Int iMVPNum,
4218	RefPicList eRefPicList,
4219	Int iRefIdx,
4220	Int iSizeX,
4221	Int iSizeY
4222	#if ZERO_MVD_EST
4223	, UInt& ruiDist
4224	#endif
4225	)
4226	{
4227	UInt uiCost = MAX_INT;
4228
4229	TComPicYuv* pcPicYuvRef = pcCU->getSlice()->getRefPic( eRefPicList, iRefIdx )->getPicYuvRec();
4230
4231	pcCU->clipMv( cMvCand );
4232
4233	// prediction pattern
4234	if ( pcCU->getSlice()->getPPS()->getUseWP() && pcCU->getSlice()->getSliceType()==P_SLICE )
4235	{
4236	xPredInterLumaBlk( pcCU, pcPicYuvRef, uiPartAddr, &cMvCand, iSizeX, iSizeY, pcTemplateCand, true );
4237	}
4238	else
4239	{
4240	xPredInterLumaBlk( pcCU, pcPicYuvRef, uiPartAddr, &cMvCand, iSizeX, iSizeY, pcTemplateCand, false );
4241	}
4242
4243	if ( pcCU->getSlice()->getPPS()->getUseWP() && pcCU->getSlice()->getSliceType()==P_SLICE )
4244	{
4245	xWeightedPredictionUni( pcCU, pcTemplateCand, uiPartAddr, iSizeX, iSizeY, eRefPicList, pcTemplateCand, iRefIdx );
4246	}
4247
4248	// calc distortion
4249	#if ZERO_MVD_EST
4250	m_pcRdCost->getMotionCost( 1, 0 );
4251	DistParam cDistParam;
4252	m_pcRdCost->setDistParam( cDistParam, g_bitDepthY,
4253	pcOrgYuv->getLumaAddr(uiPartAddr), pcOrgYuv->getStride(),
4254	pcTemplateCand->getLumaAddr(uiPartAddr), pcTemplateCand->getStride(),
4255	#if NS_HAD
4256	iSizeX, iSizeY, m_pcEncCfg->getUseHADME(), m_pcEncCfg->getUseNSQT() );
4257	#else
4258	iSizeX, iSizeY, m_pcEncCfg->getUseHADME() );
4259	#endif
4260	ruiDist = cDistParam.DistFunc( &cDistParam );
4261	uiCost = ruiDist + m_pcRdCost->getCost( m_auiMVPIdxCost[iMVPIdx][iMVPNum] );
4262	#else
4263	#if WEIGHTED_CHROMA_DISTORTION
4264	uiCost = m_pcRdCost->getDistPart(g_bitDepthY, pcTemplateCand->getLumaAddr(uiPartAddr), pcTemplateCand->getStride(), pcOrgYuv->getLumaAddr(uiPartAddr), pcOrgYuv->getStride(), iSizeX, iSizeY, TEXT_LUMA, DF_SAD );
4265	#else
4266	uiCost = m_pcRdCost->getDistPart(g_bitDepthY, pcTemplateCand->getLumaAddr(uiPartAddr), pcTemplateCand->getStride(), pcOrgYuv->getLumaAddr(uiPartAddr), pcOrgYuv->getStride(), iSizeX, iSizeY, DF_SAD );
4267	#endif
4268	uiCost = (UInt) m_pcRdCost->calcRdCost( m_auiMVPIdxCost[iMVPIdx][iMVPNum], uiCost, false, DF_SAD );
4269	#endif
4270	return uiCost;
4271	}
4272
4273	Void TEncSearch::xMotionEstimation( TComDataCU* pcCU, TComYuv* pcYuvOrg, Int iPartIdx, RefPicList eRefPicList, TComMv* pcMvPred, Int iRefIdxPred, TComMv& rcMv, UInt& ruiBits, UInt& ruiCost, Bool bBi )
4274	{
4275	UInt uiPartAddr;
4276	Int iRoiWidth;
4277	Int iRoiHeight;
4278
4279	TComMv cMvHalf, cMvQter;
4280	TComMv cMvSrchRngLT;
4281	TComMv cMvSrchRngRB;
4282
4283	TComYuv* pcYuv = pcYuvOrg;
4284	m_iSearchRange = m_aaiAdaptSR[eRefPicList][iRefIdxPred];
4285
4286	Int iSrchRng = ( bBi ? m_bipredSearchRange : m_iSearchRange );
4287	TComPattern* pcPatternKey = pcCU->getPattern ();
4288
4289	Double fWeight = 1.0;
4290
4291	pcCU->getPartIndexAndSize( iPartIdx, uiPartAddr, iRoiWidth, iRoiHeight );
4292
4293	if ( bBi )
4294	{
4295	TComYuv* pcYuvOther = &m_acYuvPred[1-(Int)eRefPicList];
4296	pcYuv = &m_cYuvPredTemp;
4297
4298	pcYuvOrg->copyPartToPartYuv( pcYuv, uiPartAddr, iRoiWidth, iRoiHeight );
4299
4300	pcYuv->removeHighFreq( pcYuvOther, uiPartAddr, iRoiWidth, iRoiHeight );
4301
4302	fWeight = 0.5;
4303	}
4304
4305	// Search key pattern initialization
4306	pcPatternKey->initPattern( pcYuv->getLumaAddr( uiPartAddr ),
4307	pcYuv->getCbAddr ( uiPartAddr ),
4308	pcYuv->getCrAddr ( uiPartAddr ),
4309	iRoiWidth,
4310	iRoiHeight,
4311	pcYuv->getStride(),
4312	0, 0 );
4313
4314	Pel* piRefY = pcCU->getSlice()->getRefPic( eRefPicList, iRefIdxPred )->getPicYuvRec()->getLumaAddr( pcCU->getAddr(), pcCU->getZorderIdxInCU() + uiPartAddr );
4315	Int iRefStride = pcCU->getSlice()->getRefPic( eRefPicList, iRefIdxPred )->getPicYuvRec()->getStride();
4316
4317	TComMv cMvPred = *pcMvPred;
4318
4319	if ( bBi ) xSetSearchRange ( pcCU, rcMv , iSrchRng, cMvSrchRngLT, cMvSrchRngRB );
4320	else xSetSearchRange ( pcCU, cMvPred, iSrchRng, cMvSrchRngLT, cMvSrchRngRB );
4321
4322	m_pcRdCost->getMotionCost ( 1, 0 );
4323
4324	m_pcRdCost->setPredictor ( *pcMvPred );
4325	m_pcRdCost->setCostScale ( 2 );
4326
4327	setWpScalingDistParam( pcCU, iRefIdxPred, eRefPicList );
4328	// Do integer search
4329	if ( !m_iFastSearch \|\| bBi )
4330	{
4331	xPatternSearch ( pcPatternKey, piRefY, iRefStride, &cMvSrchRngLT, &cMvSrchRngRB, rcMv, ruiCost );
4332	}
4333	else
4334	{
4335	rcMv = *pcMvPred;
4336	xPatternSearchFast ( pcCU, pcPatternKey, piRefY, iRefStride, &cMvSrchRngLT, &cMvSrchRngRB, rcMv, ruiCost );
4337	}
4338
4339	m_pcRdCost->getMotionCost( 1, 0 );
4340	m_pcRdCost->setCostScale ( 1 );
4341
4342	{
4343	xPatternSearchFracDIF( pcCU, pcPatternKey, piRefY, iRefStride, &rcMv, cMvHalf, cMvQter, ruiCost
4344	,bBi
4345	);
4346	}
4347
4348
4349
4350	m_pcRdCost->setCostScale( 0 );
4351	rcMv <<= 2;
4352	rcMv += (cMvHalf <<= 1);
4353	rcMv += cMvQter;
4354
4355	UInt uiMvBits = m_pcRdCost->getBits( rcMv.getHor(), rcMv.getVer() );
4356
4357	ruiBits += uiMvBits;
4358	ruiCost = (UInt)( floor( fWeight * ( (Double)ruiCost - (Double)m_pcRdCost->getCost( uiMvBits ) ) ) + (Double)m_pcRdCost->getCost( ruiBits ) );
4359	}
4360
4361
4362	Void TEncSearch::xSetSearchRange ( TComDataCU* pcCU, TComMv& cMvPred, Int iSrchRng, TComMv& rcMvSrchRngLT, TComMv& rcMvSrchRngRB )
4363	{
4364	Int iMvShift = 2;
4365	TComMv cTmpMvPred = cMvPred;
4366	pcCU->clipMv( cTmpMvPred );
4367
4368	rcMvSrchRngLT.setHor( cTmpMvPred.getHor() - (iSrchRng << iMvShift) );
4369	rcMvSrchRngLT.setVer( cTmpMvPred.getVer() - (iSrchRng << iMvShift) );
4370
4371	rcMvSrchRngRB.setHor( cTmpMvPred.getHor() + (iSrchRng << iMvShift) );
4372	rcMvSrchRngRB.setVer( cTmpMvPred.getVer() + (iSrchRng << iMvShift) );
4373	pcCU->clipMv ( rcMvSrchRngLT );
4374	pcCU->clipMv ( rcMvSrchRngRB );
4375
4376	rcMvSrchRngLT >>= iMvShift;
4377	rcMvSrchRngRB >>= iMvShift;
4378	}
4379
4380	Void TEncSearch::xPatternSearch( TComPattern* pcPatternKey, Pel* piRefY, Int iRefStride, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, TComMv& rcMv, UInt& ruiSAD )
4381	{
4382	Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();
4383	Int iSrchRngHorRight = pcMvSrchRngRB->getHor();
4384	Int iSrchRngVerTop = pcMvSrchRngLT->getVer();
4385	Int iSrchRngVerBottom = pcMvSrchRngRB->getVer();
4386
4387	UInt uiSad;
4388	UInt uiSadBest = MAX_UINT;
4389	Int iBestX = 0;
4390	Int iBestY = 0;
4391
4392	Pel* piRefSrch;
4393
4394	//-- jclee for using the SAD function pointer
4395	m_pcRdCost->setDistParam( pcPatternKey, piRefY, iRefStride, m_cDistParam );
4396
4397	// fast encoder decision: use subsampled SAD for integer ME
4398	if ( m_pcEncCfg->getUseFastEnc() )
4399	{
4400	if ( m_cDistParam.iRows > 8 )
4401	{
4402	m_cDistParam.iSubShift = 1;
4403	}
4404	}
4405
4406	piRefY += (iSrchRngVerTop * iRefStride);
4407	for ( Int y = iSrchRngVerTop; y <= iSrchRngVerBottom; y++ )
4408	{
4409	for ( Int x = iSrchRngHorLeft; x <= iSrchRngHorRight; x++ )
4410	{
4411	// find min. distortion position
4412	piRefSrch = piRefY + x;
4413	m_cDistParam.pCur = piRefSrch;
4414
4415	setDistParamComp(0);
4416
4417	m_cDistParam.bitDepth = g_bitDepthY;
4418	uiSad = m_cDistParam.DistFunc( &m_cDistParam );
4419
4420	// motion cost
4421	uiSad += m_pcRdCost->getCost( x, y );
4422
4423	if ( uiSad < uiSadBest )
4424	{
4425	uiSadBest = uiSad;
4426	iBestX = x;
4427	iBestY = y;
4428	}
4429	}
4430	piRefY += iRefStride;
4431	}
4432
4433	rcMv.set( iBestX, iBestY );
4434
4435	ruiSAD = uiSadBest - m_pcRdCost->getCost( iBestX, iBestY );
4436	return;
4437	}
4438
4439	Void TEncSearch::xPatternSearchFast( TComDataCU* pcCU, TComPattern* pcPatternKey, Pel* piRefY, Int iRefStride, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, TComMv& rcMv, UInt& ruiSAD )
4440	{
4441	pcCU->getMvPredLeft ( m_acMvPredictors[0] );
4442	pcCU->getMvPredAbove ( m_acMvPredictors[1] );
4443	pcCU->getMvPredAboveRight ( m_acMvPredictors[2] );
4444
4445	switch ( m_iFastSearch )
4446	{
4447	case 1:
4448	xTZSearch( pcCU, pcPatternKey, piRefY, iRefStride, pcMvSrchRngLT, pcMvSrchRngRB, rcMv, ruiSAD );
4449	break;
4450
4451	default:
4452	break;
4453	}
4454	}
4455
4456	Void TEncSearch::xTZSearch( TComDataCU* pcCU, TComPattern* pcPatternKey, Pel* piRefY, Int iRefStride, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, TComMv& rcMv, UInt& ruiSAD )
4457	{
4458	Int iSrchRngHorLeft = pcMvSrchRngLT->getHor();
4459	Int iSrchRngHorRight = pcMvSrchRngRB->getHor();
4460	Int iSrchRngVerTop = pcMvSrchRngLT->getVer();
4461	Int iSrchRngVerBottom = pcMvSrchRngRB->getVer();
4462
4463	TZ_SEARCH_CONFIGURATION
4464
4465	UInt uiSearchRange = m_iSearchRange;
4466	pcCU->clipMv( rcMv );
4467	rcMv >>= 2;
4468	// init TZSearchStruct
4469	IntTZSearchStruct cStruct;
4470	cStruct.iYStride = iRefStride;
4471	cStruct.piRefY = piRefY;
4472	cStruct.uiBestSad = MAX_UINT;
4473
4474	// set rcMv (Median predictor) as start point and as best point
4475	xTZSearchHelp( pcPatternKey, cStruct, rcMv.getHor(), rcMv.getVer(), 0, 0 );
4476
4477	// test whether one of PRED_A, PRED_B, PRED_C MV is better start point than Median predictor
4478	if ( bTestOtherPredictedMV )
4479	{
4480	for ( UInt index = 0; index < 3; index++ )
4481	{
4482	TComMv cMv = m_acMvPredictors[index];
4483	pcCU->clipMv( cMv );
4484	cMv >>= 2;
4485	xTZSearchHelp( pcPatternKey, cStruct, cMv.getHor(), cMv.getVer(), 0, 0 );
4486	}
4487	}
4488
4489	// test whether zero Mv is better start point than Median predictor
4490	if ( bTestZeroVector )
4491	{
4492	xTZSearchHelp( pcPatternKey, cStruct, 0, 0, 0, 0 );
4493	}
4494
4495	// start search
4496	Int iDist = 0;
4497	Int iStartX = cStruct.iBestX;
4498	Int iStartY = cStruct.iBestY;
4499
4500	// first search
4501	for ( iDist = 1; iDist <= (Int)uiSearchRange; iDist*=2 )
4502	{
4503	if ( bFirstSearchDiamond == 1 )
4504	{
4505	xTZ8PointDiamondSearch ( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, iStartX, iStartY, iDist );
4506	}
4507	else
4508	{
4509	xTZ8PointSquareSearch ( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, iStartX, iStartY, iDist );
4510	}
4511
4512	if ( bFirstSearchStop && ( cStruct.uiBestRound >= uiFirstSearchRounds ) ) // stop criterion
4513	{
4514	break;
4515	}
4516	}
4517
4518	// test whether zero Mv is a better start point than Median predictor
4519	if ( bTestZeroVectorStart && ((cStruct.iBestX != 0) \|\| (cStruct.iBestY != 0)) )
4520	{
4521	xTZSearchHelp( pcPatternKey, cStruct, 0, 0, 0, 0 );
4522	if ( (cStruct.iBestX == 0) && (cStruct.iBestY == 0) )
4523	{
4524	// test its neighborhood
4525	for ( iDist = 1; iDist <= (Int)uiSearchRange; iDist*=2 )
4526	{
4527	xTZ8PointDiamondSearch( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, 0, 0, iDist );
4528	if ( bTestZeroVectorStop && (cStruct.uiBestRound > 0) ) // stop criterion
4529	{
4530	break;
4531	}
4532	}
4533	}
4534	}
4535
4536	// calculate only 2 missing points instead 8 points if cStruct.uiBestDistance == 1
4537	if ( cStruct.uiBestDistance == 1 )
4538	{
4539	cStruct.uiBestDistance = 0;
4540	xTZ2PointSearch( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB );
4541	}
4542
4543	// raster search if distance is too big
4544	if ( bEnableRasterSearch && ( ((Int)(cStruct.uiBestDistance) > iRaster) \|\| bAlwaysRasterSearch ) )
4545	{
4546	cStruct.uiBestDistance = iRaster;
4547	for ( iStartY = iSrchRngVerTop; iStartY <= iSrchRngVerBottom; iStartY += iRaster )
4548	{
4549	for ( iStartX = iSrchRngHorLeft; iStartX <= iSrchRngHorRight; iStartX += iRaster )
4550	{
4551	xTZSearchHelp( pcPatternKey, cStruct, iStartX, iStartY, 0, iRaster );
4552	}
4553	}
4554	}
4555
4556	// raster refinement
4557	if ( bRasterRefinementEnable && cStruct.uiBestDistance > 0 )
4558	{
4559	while ( cStruct.uiBestDistance > 0 )
4560	{
4561	iStartX = cStruct.iBestX;
4562	iStartY = cStruct.iBestY;
4563	if ( cStruct.uiBestDistance > 1 )
4564	{
4565	iDist = cStruct.uiBestDistance >>= 1;
4566	if ( bRasterRefinementDiamond == 1 )
4567	{
4568	xTZ8PointDiamondSearch ( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, iStartX, iStartY, iDist );
4569	}
4570	else
4571	{
4572	xTZ8PointSquareSearch ( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, iStartX, iStartY, iDist );
4573	}
4574	}
4575
4576	// calculate only 2 missing points instead 8 points if cStruct.uiBestDistance == 1
4577	if ( cStruct.uiBestDistance == 1 )
4578	{
4579	cStruct.uiBestDistance = 0;
4580	if ( cStruct.ucPointNr != 0 )
4581	{
4582	xTZ2PointSearch( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB );
4583	}
4584	}
4585	}
4586	}
4587
4588	// start refinement
4589	if ( bStarRefinementEnable && cStruct.uiBestDistance > 0 )
4590	{
4591	while ( cStruct.uiBestDistance > 0 )
4592	{
4593	iStartX = cStruct.iBestX;
4594	iStartY = cStruct.iBestY;
4595	cStruct.uiBestDistance = 0;
4596	cStruct.ucPointNr = 0;
4597	for ( iDist = 1; iDist < (Int)uiSearchRange + 1; iDist*=2 )
4598	{
4599	if ( bStarRefinementDiamond == 1 )
4600	{
4601	xTZ8PointDiamondSearch ( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, iStartX, iStartY, iDist );
4602	}
4603	else
4604	{
4605	xTZ8PointSquareSearch ( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB, iStartX, iStartY, iDist );
4606	}
4607	if ( bStarRefinementStop && (cStruct.uiBestRound >= uiStarRefinementRounds) ) // stop criterion
4608	{
4609	break;
4610	}
4611	}
4612
4613	// calculate only 2 missing points instead 8 points if cStrukt.uiBestDistance == 1
4614	if ( cStruct.uiBestDistance == 1 )
4615	{
4616	cStruct.uiBestDistance = 0;
4617	if ( cStruct.ucPointNr != 0 )
4618	{
4619	xTZ2PointSearch( pcPatternKey, cStruct, pcMvSrchRngLT, pcMvSrchRngRB );
4620	}
4621	}
4622	}
4623	}
4624
4625	// write out best match
4626	rcMv.set( cStruct.iBestX, cStruct.iBestY );
4627	ruiSAD = cStruct.uiBestSad - m_pcRdCost->getCost( cStruct.iBestX, cStruct.iBestY );
4628	}
4629
4630	Void TEncSearch::xPatternSearchFracDIF(TComDataCU* pcCU,
4631	TComPattern* pcPatternKey,
4632	Pel* piRefY,
4633	Int iRefStride,
4634	TComMv* pcMvInt,
4635	TComMv& rcMvHalf,
4636	TComMv& rcMvQter,
4637	UInt& ruiCost
4638	,Bool biPred
4639	)
4640	{
4641	// Reference pattern initialization (integer scale)
4642	TComPattern cPatternRoi;
4643	Int iOffset = pcMvInt->getHor() + pcMvInt->getVer() * iRefStride;
4644	cPatternRoi.initPattern( piRefY + iOffset,
4645	NULL,
4646	NULL,
4647	pcPatternKey->getROIYWidth(),
4648	pcPatternKey->getROIYHeight(),
4649	iRefStride,
4650	0, 0 );
4651
4652	// Half-pel refinement
4653	xExtDIFUpSamplingH ( &cPatternRoi, biPred );
4654
4655	rcMvHalf = *pcMvInt; rcMvHalf <<= 1; // for mv-cost
4656	TComMv baseRefMv(0, 0);
4657	ruiCost = xPatternRefinement( pcPatternKey, baseRefMv, 2, rcMvHalf );
4658
4659	m_pcRdCost->setCostScale( 0 );
4660
4661	xExtDIFUpSamplingQ ( &cPatternRoi, rcMvHalf, biPred );
4662	baseRefMv = rcMvHalf;
4663	baseRefMv <<= 1;
4664
4665	rcMvQter = *pcMvInt; rcMvQter <<= 1; // for mv-cost
4666	rcMvQter += rcMvHalf; rcMvQter <<= 1;
4667	ruiCost = xPatternRefinement( pcPatternKey, baseRefMv, 1, rcMvQter );
4668	}
4669
4670	/** encode residual and calculate rate-distortion for a CU block
4671	* \param pcCU
4672	* \param pcYuvOrg
4673	* \param pcYuvPred
4674	* \param rpcYuvResi
4675	* \param rpcYuvResiBest
4676	* \param rpcYuvRec
4677	* \param bSkipRes
4678	* \returns Void
4679	*/
4680	Void TEncSearch::encodeResAndCalcRdInterCU( TComDataCU* pcCU, TComYuv* pcYuvOrg, TComYuv* pcYuvPred, TComYuv& rpcYuvResi, TComYuv& rpcYuvResiBest, TComYuv*& rpcYuvRec, Bool bSkipRes )
4681	{
4682	if ( pcCU->isIntra(0) )
4683	{
4684	return;
4685	}
4686
4687	Bool bHighPass = pcCU->getSlice()->getDepth() ? true : false;
4688	UInt uiBits = 0, uiBitsBest = 0;
4689	#if H_3D_VSO
4690	Dist uiDistortion = 0, uiDistortionBest = 0;
4691	#else
4692	UInt uiDistortion = 0, uiDistortionBest = 0;
4693	#endif
4694
4695	UInt uiWidth = pcCU->getWidth ( 0 );
4696	UInt uiHeight = pcCU->getHeight( 0 );
4697
4698	// No residual coding : SKIP mode
4699	if ( bSkipRes )
4700	{
4701	pcCU->setSkipFlagSubParts( true, 0, pcCU->getDepth(0) );
4702
4703	rpcYuvResi->clear();
4704
4705	pcYuvPred->copyToPartYuv( rpcYuvRec, 0 );
4706
4707	#if H_3D_VSO // M13
4708	if ( m_pcRdCost->getUseVSO() )
4709	{
4710	uiDistortion = m_pcRdCost->getDistPartVSO( pcCU, 0, rpcYuvRec->getLumaAddr(), rpcYuvRec->getStride(), pcYuvOrg->getLumaAddr(), pcYuvOrg->getStride(), uiWidth, uiHeight , false );
4711	}
4712	else
4713	{
4714	#endif
4715	#if WEIGHTED_CHROMA_DISTORTION
4716	uiDistortion = m_pcRdCost->getDistPart(g_bitDepthY, rpcYuvRec->getLumaAddr(), rpcYuvRec->getStride(), pcYuvOrg->getLumaAddr(), pcYuvOrg->getStride(), uiWidth, uiHeight )
4717	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCbAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCbAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1, TEXT_CHROMA_U )
4718	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCrAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCrAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1, TEXT_CHROMA_V );
4719	#else
4720	uiDistortion = m_pcRdCost->getDistPart(g_bitDepthY, rpcYuvRec->getLumaAddr(), rpcYuvRec->getStride(), pcYuvOrg->getLumaAddr(), pcYuvOrg->getStride(), uiWidth, uiHeight )
4721	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCbAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCbAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1 )
4722	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCrAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCrAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1 );
4723	#endif
4724
4725	#if H_3D_VSO // MIgnore
4726	}
4727	#endif
4728
4729	if( m_bUseSBACRD )
4730	m_pcRDGoOnSbacCoder->load(m_pppcRDSbacCoder[pcCU->getDepth(0)][CI_CURR_BEST]);
4731
4732	m_pcEntropyCoder->resetBits();
4733	if (pcCU->getSlice()->getPPS()->getTransquantBypassEnableFlag())
4734	{
4735	m_pcEntropyCoder->encodeCUTransquantBypassFlag(pcCU, 0, true);
4736	}
4737	m_pcEntropyCoder->encodeSkipFlag(pcCU, 0, true);
4738	m_pcEntropyCoder->encodeMergeIndex( pcCU, 0, true );
4739
4740	uiBits = m_pcEntropyCoder->getNumberOfWrittenBits();
4741	pcCU->getTotalBits() = uiBits;
4742	pcCU->getTotalDistortion() = uiDistortion;
4743	#if H_3D_VSO //M 14
4744	if ( m_pcRdCost->getUseLambdaScaleVSO() )
4745	pcCU->getTotalCost() = m_pcRdCost->calcRdCostVSO( uiBits, uiDistortion );
4746	else
4747	#endif
4748	pcCU->getTotalCost() = m_pcRdCost->calcRdCost( uiBits, uiDistortion );
4749
4750	if( m_bUseSBACRD )
4751	m_pcRDGoOnSbacCoder->store(m_pppcRDSbacCoder[pcCU->getDepth(0)][CI_TEMP_BEST]);
4752
4753	pcCU->setCbfSubParts( 0, 0, 0, 0, pcCU->getDepth( 0 ) );
4754	pcCU->setTrIdxSubParts( 0, 0, pcCU->getDepth(0) );
4755
4756	#if H_3D_VSO // necessary? // M15
4757	// set Model
4758	if( !m_pcRdCost->getUseEstimatedVSD()&& m_pcRdCost->getUseRenModel() )
4759	{
4760	Pel* piSrc = rpcYuvRec->getLumaAddr();
4761	UInt uiSrcStride = rpcYuvRec->getStride();
4762	m_pcRdCost->setRenModelData( pcCU, 0, piSrc, uiSrcStride, uiWidth, uiHeight );
4763	}
4764	#endif
4765
4766	return;
4767	}
4768
4769	// Residual coding.
4770	Int qp, qpBest = 0, qpMin, qpMax;
4771	Double dCost, dCostBest = MAX_DOUBLE;
4772
4773	UInt uiTrLevel = 0;
4774	if( (pcCU->getWidth(0) > pcCU->getSlice()->getSPS()->getMaxTrSize()) )
4775	{
4776	while( pcCU->getWidth(0) > (pcCU->getSlice()->getSPS()->getMaxTrSize()<<uiTrLevel) ) uiTrLevel++;
4777	}
4778	UInt uiMaxTrMode = 1 + uiTrLevel;
4779
4780	while((uiWidth>>uiMaxTrMode) < (g_uiMaxCUWidth>>g_uiMaxCUDepth)) uiMaxTrMode--;
4781
4782	qpMin = bHighPass ? Clip3( -pcCU->getSlice()->getSPS()->getQpBDOffsetY(), MAX_QP, pcCU->getQP(0) - m_iMaxDeltaQP ) : pcCU->getQP( 0 );
4783	qpMax = bHighPass ? Clip3( -pcCU->getSlice()->getSPS()->getQpBDOffsetY(), MAX_QP, pcCU->getQP(0) + m_iMaxDeltaQP ) : pcCU->getQP( 0 );
4784
4785	rpcYuvResi->subtract( pcYuvOrg, pcYuvPred, 0, uiWidth );
4786
4787	for ( qp = qpMin; qp <= qpMax; qp++ )
4788	{
4789	dCost = 0.;
4790	uiBits = 0;
4791	uiDistortion = 0;
4792	if( m_bUseSBACRD )
4793	{
4794	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[ pcCU->getDepth( 0 ) ][ CI_CURR_BEST ] );
4795	}
4796
4797	#if H_3D_VSO // M16 // M18
4798	Dist uiZeroDistortion = 0;
4799	if ( m_pcRdCost->getUseVSO() ) // This creating and destroying need to be fixed.
4800	{
4801	m_cYuvRecTemp.create( pcYuvPred->getWidth(), pcYuvPred->getHeight() );
4802	}
4803
4804	xEstimateResidualQT( pcCU, 0, 0, 0, pcYuvOrg, pcYuvPred, rpcYuvResi, pcCU->getDepth(0), dCost, uiBits, uiDistortion, &uiZeroDistortion );
4805
4806	if ( m_pcRdCost->getUseVSO() )
4807	{
4808	m_cYuvRecTemp.destroy();
4809	}
4810	#else
4811	UInt uiZeroDistortion = 0;
4812	xEstimateResidualQT( pcCU, 0, 0, 0, rpcYuvResi, pcCU->getDepth(0), dCost, uiBits, uiDistortion, &uiZeroDistortion );
4813	#endif
4814
4815	m_pcEntropyCoder->resetBits();
4816	m_pcEntropyCoder->encodeQtRootCbfZero( pcCU );
4817	UInt zeroResiBits = m_pcEntropyCoder->getNumberOfWrittenBits();
4818
4819	#if H_3D_VSO // M19
4820	Double dZeroCost;
4821	if( m_pcRdCost->getUseLambdaScaleVSO() )
4822	dZeroCost = m_pcRdCost->calcRdCostVSO( 0, uiZeroDistortion );
4823	else
4824	dZeroCost = m_pcRdCost->calcRdCost( zeroResiBits, uiZeroDistortion );
4825	#else
4826	Double dZeroCost = m_pcRdCost->calcRdCost( zeroResiBits, uiZeroDistortion );
4827	#endif
4828	if(pcCU->isLosslessCoded( 0 ))
4829	{
4830	dZeroCost = dCost + 1;
4831	}
4832	if ( dZeroCost < dCost )
4833	{
4834	dCost = dZeroCost;
4835	uiBits = 0;
4836	uiDistortion = uiZeroDistortion;
4837
4838	const UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> (pcCU->getDepth(0) << 1);
4839	::memset( pcCU->getTransformIdx() , 0, uiQPartNum * sizeof(UChar) );
4840	::memset( pcCU->getCbf( TEXT_LUMA ) , 0, uiQPartNum * sizeof(UChar) );
4841	::memset( pcCU->getCbf( TEXT_CHROMA_U ), 0, uiQPartNum * sizeof(UChar) );
4842	::memset( pcCU->getCbf( TEXT_CHROMA_V ), 0, uiQPartNum * sizeof(UChar) );
4843	::memset( pcCU->getCoeffY() , 0, uiWidth * uiHeight * sizeof( TCoeff ) );
4844	::memset( pcCU->getCoeffCb() , 0, uiWidth * uiHeight * sizeof( TCoeff ) >> 2 );
4845	::memset( pcCU->getCoeffCr() , 0, uiWidth * uiHeight * sizeof( TCoeff ) >> 2 );
4846	pcCU->setTransformSkipSubParts ( 0, 0, 0, 0, pcCU->getDepth(0) );
4847	}
4848	else
4849	{
4850	xSetResidualQTData( pcCU, 0, 0, 0, NULL, pcCU->getDepth(0), false );
4851	}
4852
4853	if( m_bUseSBACRD )
4854	{
4855	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[pcCU->getDepth(0)][CI_CURR_BEST] );
4856	}
4857	#if 0 // check
4858	{
4859	m_pcEntropyCoder->resetBits();
4860	m_pcEntropyCoder->encodeCoeff( pcCU, 0, pcCU->getDepth(0), pcCU->getWidth(0), pcCU->getHeight(0) );
4861	const UInt uiBitsForCoeff = m_pcEntropyCoder->getNumberOfWrittenBits();
4862	if( m_bUseSBACRD )
4863	{
4864	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[pcCU->getDepth(0)][CI_CURR_BEST] );
4865	}
4866	if( uiBitsForCoeff != uiBits )
4867	assert( 0 );
4868	}
4869	#endif
4870	uiBits = 0;
4871	{
4872	TComYuv *pDummy = NULL;
4873	xAddSymbolBitsInter( pcCU, 0, 0, uiBits, pDummy, NULL, pDummy );
4874	}
4875
4876	#if H_3D_VSO // M20
4877	Double dExactCost;
4878	if( m_pcRdCost->getUseLambdaScaleVSO() )
4879	dExactCost = m_pcRdCost->calcRdCostVSO( uiBits, uiDistortion );
4880	else
4881	dExactCost = m_pcRdCost->calcRdCost ( uiBits, uiDistortion );
4882	#else
4883	Double dExactCost = m_pcRdCost->calcRdCost( uiBits, uiDistortion );
4884	#endif
4885	dCost = dExactCost;
4886
4887	if ( dCost < dCostBest )
4888	{
4889	if ( !pcCU->getQtRootCbf( 0 ) )
4890	{
4891	rpcYuvResiBest->clear();
4892	}
4893	else
4894	{
4895	xSetResidualQTData( pcCU, 0, 0, 0, rpcYuvResiBest, pcCU->getDepth(0), true );
4896	}
4897
4898	if( qpMin != qpMax && qp != qpMax )
4899	{
4900	const UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> (pcCU->getDepth(0) << 1);
4901	::memcpy( m_puhQTTempTrIdx, pcCU->getTransformIdx(), uiQPartNum * sizeof(UChar) );
4902	::memcpy( m_puhQTTempCbf[0], pcCU->getCbf( TEXT_LUMA ), uiQPartNum * sizeof(UChar) );
4903	::memcpy( m_puhQTTempCbf[1], pcCU->getCbf( TEXT_CHROMA_U ), uiQPartNum * sizeof(UChar) );
4904	::memcpy( m_puhQTTempCbf[2], pcCU->getCbf( TEXT_CHROMA_V ), uiQPartNum * sizeof(UChar) );
4905	::memcpy( m_pcQTTempCoeffY, pcCU->getCoeffY(), uiWidth * uiHeight * sizeof( TCoeff ) );
4906	::memcpy( m_pcQTTempCoeffCb, pcCU->getCoeffCb(), uiWidth * uiHeight * sizeof( TCoeff ) >> 2 );
4907	::memcpy( m_pcQTTempCoeffCr, pcCU->getCoeffCr(), uiWidth * uiHeight * sizeof( TCoeff ) >> 2 );
4908	#if ADAPTIVE_QP_SELECTION
4909	::memcpy( m_pcQTTempArlCoeffY, pcCU->getArlCoeffY(), uiWidth * uiHeight * sizeof( Int ) );
4910	::memcpy( m_pcQTTempArlCoeffCb, pcCU->getArlCoeffCb(), uiWidth * uiHeight * sizeof( Int ) >> 2 );
4911	::memcpy( m_pcQTTempArlCoeffCr, pcCU->getArlCoeffCr(), uiWidth * uiHeight * sizeof( Int ) >> 2 );
4912	#endif
4913	::memcpy( m_puhQTTempTransformSkipFlag[0], pcCU->getTransformSkip(TEXT_LUMA), uiQPartNum * sizeof( UChar ) );
4914	::memcpy( m_puhQTTempTransformSkipFlag[1], pcCU->getTransformSkip(TEXT_CHROMA_U), uiQPartNum * sizeof( UChar ) );
4915	::memcpy( m_puhQTTempTransformSkipFlag[2], pcCU->getTransformSkip(TEXT_CHROMA_V), uiQPartNum * sizeof( UChar ) );
4916	}
4917	uiBitsBest = uiBits;
4918	uiDistortionBest = uiDistortion;
4919	dCostBest = dCost;
4920	qpBest = qp;
4921	if( m_bUseSBACRD )
4922	{
4923	m_pcRDGoOnSbacCoder->store( m_pppcRDSbacCoder[ pcCU->getDepth( 0 ) ][ CI_TEMP_BEST ] );
4924	}
4925	}
4926	#if H_3D_VSO // M21
4927	if( m_pcRdCost->getUseRenModel() && !m_pcRdCost->getUseEstimatedVSD() )
4928	{
4929	Pel* piSrc = pcYuvOrg->getLumaAddr();
4930	UInt uiSrcStride = pcYuvOrg->getStride();
4931	m_pcRdCost->setRenModelData( pcCU, 0, piSrc, uiSrcStride, uiWidth, uiHeight );
4932	}
4933	#endif
4934	}
4935
4936	assert ( dCostBest != MAX_DOUBLE );
4937
4938	if( qpMin != qpMax && qpBest != qpMax )
4939	{
4940	if( m_bUseSBACRD )
4941	{
4942	assert( 0 ); // check
4943	m_pcRDGoOnSbacCoder->load( m_pppcRDSbacCoder[ pcCU->getDepth( 0 ) ][ CI_TEMP_BEST ] );
4944	}
4945	// copy best cbf and trIdx to pcCU
4946	const UInt uiQPartNum = pcCU->getPic()->getNumPartInCU() >> (pcCU->getDepth(0) << 1);
4947	::memcpy( pcCU->getTransformIdx(), m_puhQTTempTrIdx, uiQPartNum * sizeof(UChar) );
4948	::memcpy( pcCU->getCbf( TEXT_LUMA ), m_puhQTTempCbf[0], uiQPartNum * sizeof(UChar) );
4949	::memcpy( pcCU->getCbf( TEXT_CHROMA_U ), m_puhQTTempCbf[1], uiQPartNum * sizeof(UChar) );
4950	::memcpy( pcCU->getCbf( TEXT_CHROMA_V ), m_puhQTTempCbf[2], uiQPartNum * sizeof(UChar) );
4951	::memcpy( pcCU->getCoeffY(), m_pcQTTempCoeffY, uiWidth * uiHeight * sizeof( TCoeff ) );
4952	::memcpy( pcCU->getCoeffCb(), m_pcQTTempCoeffCb, uiWidth * uiHeight * sizeof( TCoeff ) >> 2 );
4953	::memcpy( pcCU->getCoeffCr(), m_pcQTTempCoeffCr, uiWidth * uiHeight * sizeof( TCoeff ) >> 2 );
4954	#if ADAPTIVE_QP_SELECTION
4955	::memcpy( pcCU->getArlCoeffY(), m_pcQTTempArlCoeffY, uiWidth * uiHeight * sizeof( Int ) );
4956	::memcpy( pcCU->getArlCoeffCb(), m_pcQTTempArlCoeffCb, uiWidth * uiHeight * sizeof( Int ) >> 2 );
4957	::memcpy( pcCU->getArlCoeffCr(), m_pcQTTempArlCoeffCr, uiWidth * uiHeight * sizeof( Int ) >> 2 );
4958	#endif
4959	::memcpy( pcCU->getTransformSkip(TEXT_LUMA), m_puhQTTempTransformSkipFlag[0], uiQPartNum * sizeof( UChar ) );
4960	::memcpy( pcCU->getTransformSkip(TEXT_CHROMA_U), m_puhQTTempTransformSkipFlag[1], uiQPartNum * sizeof( UChar ) );
4961	::memcpy( pcCU->getTransformSkip(TEXT_CHROMA_V), m_puhQTTempTransformSkipFlag[2], uiQPartNum * sizeof( UChar ) );
4962	}
4963	rpcYuvRec->addClip ( pcYuvPred, rpcYuvResiBest, 0, uiWidth );
4964
4965	#if H_3D_VSO // M22 // GT: might be removed since VSO already provided clipped distortion
4966	if ( m_pcRdCost->getUseVSO() )
4967	{
4968	uiDistortionBest = m_pcRdCost->getDistPartVSO ( pcCU, 0, rpcYuvRec->getLumaAddr(), rpcYuvRec->getStride(), pcYuvOrg->getLumaAddr(), pcYuvOrg->getStride(), uiWidth, uiHeight, false );
4969	}
4970	else
4971	{
4972	#endif
4973	// update with clipped distortion and cost (qp estimation loop uses unclipped values)
4974	#if WEIGHTED_CHROMA_DISTORTION
4975	uiDistortionBest = m_pcRdCost->getDistPart(g_bitDepthY, rpcYuvRec->getLumaAddr(), rpcYuvRec->getStride(), pcYuvOrg->getLumaAddr(), pcYuvOrg->getStride(), uiWidth, uiHeight )
4976	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCbAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCbAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1, TEXT_CHROMA_U )
4977	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCrAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCrAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1, TEXT_CHROMA_V );
4978	#else
4979	uiDistortionBest = m_pcRdCost->getDistPart(g_bitDepthY, rpcYuvRec->getLumaAddr(), rpcYuvRec->getStride(), pcYuvOrg->getLumaAddr(), pcYuvOrg->getStride(), uiWidth, uiHeight )
4980	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCbAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCbAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1 )
4981	+ m_pcRdCost->getDistPart(g_bitDepthC, rpcYuvRec->getCrAddr(), rpcYuvRec->getCStride(), pcYuvOrg->getCrAddr(), pcYuvOrg->getCStride(), uiWidth >> 1, uiHeight >> 1 );
4982	#endif
4983	#if H_3D_VSO // M23
4984	}
4985	if ( m_pcRdCost->getUseLambdaScaleVSO() )
4986	dCostBest = m_pcRdCost->calcRdCostVSO( uiBitsBest, uiDistortionBest );
4987	else
4988	#endif
4989	dCostBest = m_pcRdCost->calcRdCost( uiBitsBest, uiDistortionBest );
4990
4991	pcCU->getTotalBits() = uiBitsBest;
4992	pcCU->getTotalDistortion() = uiDistortionBest;
4993	pcCU->getTotalCost() = dCostBest;
4994
4995	if ( pcCU->isSkipped(0) )
4996	{
4997	pcCU->setCbfSubParts( 0, 0, 0, 0, pcCU->getDepth( 0 ) );
4998	}
4999
5000	pcCU->setQPSubParts( qpBest, 0, pcCU->getDepth(0) );
5001
5002	#if H_3D_VSO // M24 // necessary??
5003	if( m_pcRdCost->getUseRenModel() && !m_pcRdCost->getUseEstimatedVSD() )
5004	{
5005	Pel* piSrc = rpcYuvRec->getLumaAddr();
5006	UInt uiSrcStride = rpcYuvRec->getStride();
5007	m_pcRdCost->setRenModelData( pcCU, 0, piSrc, uiSrcStride, uiWidth, uiHeight );
5008	}
5009	#endif
5010	}
5011
5012	#if H_3D_VSO // M25
5013	Void TEncSearch::xEstimateResidualQT( TComDataCU* pcCU, UInt uiQuadrant, UInt uiAbsPartIdx, UInt absTUPartIdx, TComYuv* pcOrg, TComYuv* pcPred, TComYuv* pcResi, const UInt uiDepth, Double &rdCost, UInt &ruiBits, Dist &ruiDist, Dist *puiZeroDist )
5014	#else
5015	Void TEncSearch::xEstimateResidualQT( TComDataCU* pcCU, UInt uiQuadrant, UInt uiAbsPartIdx, UInt absTUPartIdx, TComYuv* pcResi, const UInt uiDepth, Double &rdCost, UInt &ruiBits, UInt &ruiDist, UInt *puiZeroDist )
5016	#endif
5017	{
5018	const UInt uiTrMode = uiDepth - pcCU->getDepth( 0 );
5019
5020	assert( pcCU->getDepth( 0 ) == pcCU->getDepth( uiAbsPartIdx ) );
5021	const UInt uiLog2TrSize = g_aucConvertToBit[pcCU->getSlice()->getSPS()->getMaxCUWidth() >> uiDepth]+2;
5022
5023	UInt SplitFlag = ((pcCU->getSlice()->getSPS()->getQuadtreeTUMaxDepthInter() == 1) && pcCU->getPredictionMode(uiAbsPartIdx) == MODE_INTER && ( pcCU->getPartitionSize(uiAbsPartIdx) != SIZE_2Nx2N ));
5024	Bool bCheckFull;
5025	if ( SplitFlag && uiDepth == pcCU->getDepth(uiAbsPartIdx) && ( uiLog2TrSize > pcCU->getQuadtreeTULog2MinSizeInCU(uiAbsPartIdx) ) )
5026	bCheckFull = false;
5027	else
5028	bCheckFull = ( uiLog2TrSize <= pcCU->getSlice()->getSPS()->getQuadtreeTULog2MaxSize() );
5029
5030	const Bool bCheckSplit = ( uiLog2TrSize > pcCU->getQuadtreeTULog2MinSizeInCU(uiAbsPartIdx) );
5031
5032	assert( bCheckFull \|\| bCheckSplit );
5033
5034	Bool bCodeChroma = true;
5035	UInt uiTrModeC = uiTrMode;
5036	UInt uiLog2TrSizeC = uiLog2TrSize-1;
5037	if( uiLog2TrSize == 2 )
5038	{
5039	uiLog2TrSizeC++;
5040	uiTrModeC --;
5041	UInt uiQPDiv = pcCU->getPic()->getNumPartInCU() >> ( ( pcCU->getDepth( 0 ) + uiTrModeC ) << 1 );
5042	bCodeChroma = ( ( uiAbsPartIdx % uiQPDiv ) == 0 );
5043	}
5044
5045	const UInt uiSetCbf = 1 << uiTrMode;
5046	// code full block
5047	Double dSingleCost = MAX_DOUBLE;
5048	UInt uiSing

JCT-3V 3D-HEVC

Context navigation

source: 3DVCSoftware/branches/HTM-DEV-0.2-dev/source/Lib/TLibEncoder/TEncSearch.cpp @ 446