Context navigation

← Previous revision
Next revision →
Normal
Revision log

TEncSearch.h

Visit:

Last change on this file was 213, checked in by mitsubishi-htm, 12 years ago

A final release, as planned

Migrate to HTM 5.1
For VC project files, only VC9 file is updated
To be used as an additional anchor for CE1.h for 3rd JCTVC meeting at Geneva

Property svn:eol-style set to native

File size: 30.4 KB

Rev	Line
[5]	1	/* The copyright in this software is being made available under the BSD
	2	* License, included below. This software may be subject to other third party
	3	* and contributor rights, including patent rights, and no such rights are
[56]	4	* granted under this license.
[5]	5	*
[56]	6	* Copyright (c) 2010-2012, ITU/ISO/IEC
[5]	7	* All rights reserved.
	8	*
	9	* Redistribution and use in source and binary forms, with or without
	10	* modification, are permitted provided that the following conditions are met:
	11	*
	12	* * Redistributions of source code must retain the above copyright notice,
	13	* this list of conditions and the following disclaimer.
	14	* * Redistributions in binary form must reproduce the above copyright notice,
	15	* this list of conditions and the following disclaimer in the documentation
	16	* and/or other materials provided with the distribution.
[56]	17	* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
[5]	18	* be used to endorse or promote products derived from this software without
	19	* specific prior written permission.
	20	*
	21	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
	22	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	23	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	24	* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
	25	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	26	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	27	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	28	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	29	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	30	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
	31	* THE POSSIBILITY OF SUCH DAMAGE.
	32	*/
[2]	33
	34	/** \file TEncSearch.h
	35	\brief encoder search class (header)
	36	*/
	37
	38	#ifndef __TENCSEARCH__
	39	#define __TENCSEARCH__
	40
	41	// Include files
[56]	42	#include "TLibCommon/TComYuv.h"
	43	#include "TLibCommon/TComMotionInfo.h"
	44	#include "TLibCommon/TComPattern.h"
	45	#include "TLibCommon/TComPrediction.h"
	46	#include "TLibCommon/TComTrQuant.h"
	47	#include "TLibCommon/TComPic.h"
[2]	48	#include "TEncEntropy.h"
	49	#include "TEncSbac.h"
	50	#include "TEncCfg.h"
	51
[56]	52	//! \ingroup TLibEncoder
	53	//! \{
	54
[2]	55	class TEncCu;
	56
	57	// ====================================================================================================================
	58	// Class definition
	59	// ====================================================================================================================
	60
	61	/// encoder search class
	62	class TEncSearch : public TComPrediction
	63	{
	64	private:
	65	TCoeff** m_ppcQTTempCoeffY;
	66	TCoeff** m_ppcQTTempCoeffCb;
	67	TCoeff** m_ppcQTTempCoeffCr;
	68	TCoeff* m_pcQTTempCoeffY;
	69	TCoeff* m_pcQTTempCoeffCb;
	70	TCoeff* m_pcQTTempCoeffCr;
[56]	71	#if ADAPTIVE_QP_SELECTION
	72	Int** m_ppcQTTempArlCoeffY;
	73	Int** m_ppcQTTempArlCoeffCb;
	74	Int** m_ppcQTTempArlCoeffCr;
	75	Int* m_pcQTTempArlCoeffY;
	76	Int* m_pcQTTempArlCoeffCb;
	77	Int* m_pcQTTempArlCoeffCr;
	78	#endif
[2]	79	UChar* m_puhQTTempTrIdx;
	80	UChar* m_puhQTTempCbf[3];
	81
	82	TComYuv* m_pcQTTempTComYuv;
[56]	83	TComYuv m_tmpYuvPred; // To be used in xGetInterPredictionError() to avoid constant memory allocation/deallocation
[2]	84	protected:
	85	// interface to option
	86	TEncCfg* m_pcEncCfg;
	87
	88	// interface to classes
	89	TComTrQuant* m_pcTrQuant;
	90	TComRdCost* m_pcRdCost;
	91	TEncEntropy* m_pcEntropyCoder;
	92
	93	// ME parameters
	94	Int m_iSearchRange;
	95	Int m_bipredSearchRange; // Search range for bi-prediction
[213]	96	#if DV_V_RESTRICTION_B0037
	97	Bool m_bUseDisparitySearchRangeRestriction;
	98	Int m_iVerticalDisparitySearchRange;
	99	#endif
[2]	100	Int m_iFastSearch;
	101	Int m_aaiAdaptSR[2][33];
	102	TComMv m_cSrchRngLT;
	103	TComMv m_cSrchRngRB;
	104	TComMv m_acMvPredictors[3];
	105
	106	// RD computation
	107	TEncSbac*** m_pppcRDSbacCoder;
	108	TEncSbac* m_pcRDGoOnSbacCoder;
	109	Bool m_bUseSBACRD;
	110	DistParam m_cDistParam;
	111
	112	// Misc.
	113	Pel* m_pTempPel;
	114	UInt* m_puiDFilter;
	115	Int m_iMaxDeltaQP;
	116
	117
[5]	118	#if HHI_VSO
[2]	119	TComYuv m_cYuvRecTemp;
[5]	120	#endif
[2]	121
	122	// AMVP cost computation
[56]	123	#if HHI_INTER_VIEW_MOTION_PRED
	124	UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+2][AMVP_MAX_NUM_CANDS+2]; //th array bounds
	125	#else
[2]	126	// UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+1][AMVP_MAX_NUM_CANDS];
	127	UInt m_auiMVPIdxCost[AMVP_MAX_NUM_CANDS+1][AMVP_MAX_NUM_CANDS+1]; //th array bounds
[56]	128	#endif
[2]	129
	130	public:
	131	TEncSearch();
	132	virtual ~TEncSearch();
	133
	134	Void init( TEncCfg* pcEncCfg,
	135	TComTrQuant* pcTrQuant,
	136	Int iSearchRange,
	137	Int bipredSearchRange,
[213]	138	#if DV_V_RESTRICTION_B0037
	139	Bool bUseDisparitySearchRangeRestriction,
	140	Int iVerticalDisparitySearchRange,
	141	#endif
[2]	142	Int iFastSearch,
	143	Int iMaxDeltaQP,
	144	TEncEntropy* pcEntropyCoder,
	145	TComRdCost* pcRdCost,
	146	TEncSbac*** pppcRDSbacCoder,
	147	TEncSbac* pcRDGoOnSbacCoder );
	148
	149	protected:
	150
	151	/// sub-function for motion vector refinement used in fractional-pel accuracy
[56]	152	UInt xPatternRefinement( TComPattern* pcPatternKey,
	153	TComMv baseRefMv,
[166]	154	Int iFrac, TComMv& rcMvFrac
	155	#if VSP_MV_ZERO
	156	,Bool bIsVsp
	157	#endif
	158	);
[2]	159
	160	typedef struct
	161	{
	162	Pel* piRefY;
	163	Int iYStride;
	164	Int iBestX;
	165	Int iBestY;
	166	UInt uiBestRound;
	167	UInt uiBestDistance;
	168	UInt uiBestSad;
	169	UChar ucPointNr;
	170	} IntTZSearchStruct;
	171
	172	// sub-functions for ME
	173	__inline Void xTZSearchHelp ( TComPattern* pcPatternKey, IntTZSearchStruct& rcStruct, const Int iSearchX, const Int iSearchY, const UChar ucPointNr, const UInt uiDistance );
	174	__inline Void xTZ2PointSearch ( TComPattern* pcPatternKey, IntTZSearchStruct& rcStrukt, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB );
	175	__inline Void xTZ8PointSquareSearch ( TComPattern* pcPatternKey, IntTZSearchStruct& rcStrukt, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist );
	176	__inline Void xTZ8PointDiamondSearch( TComPattern* pcPatternKey, IntTZSearchStruct& rcStrukt, TComMv* pcMvSrchRngLT, TComMv* pcMvSrchRngRB, const Int iStartX, const Int iStartY, const Int iDist );
	177
	178	Void xGetInterPredictionError( TComDataCU* pcCU, TComYuv* pcYuvOrg, Int iPartIdx, UInt& ruiSAD, Bool Hadamard );
	179
	180	public:
	181	Void preestChromaPredMode ( TComDataCU* pcCU,
	182	TComYuv* pcOrgYuv,
	183	TComYuv* pcPredYuv );
	184	Void estIntraPredQT ( TComDataCU* pcCU,
	185	TComYuv* pcOrgYuv,
	186	TComYuv* pcPredYuv,
	187	TComYuv* pcResiYuv,
	188	TComYuv* pcRecoYuv,
	189	Dist& ruiDistC,
	190	Bool bLumaOnly );
	191	Void estIntraPredChromaQT ( TComDataCU* pcCU,
	192	TComYuv* pcOrgYuv,
	193	TComYuv* pcPredYuv,
	194	TComYuv* pcResiYuv,
	195	TComYuv* pcRecoYuv,
	196	Dist uiPreCalcDistC );
	197
	198
	199	/// encoder estimation - inter prediction (non-skip)
	200	Void predInterSearch ( TComDataCU* pcCU,
	201	TComYuv* pcOrgYuv,
[77]	202	#if LG_RESTRICTEDRESPRED_M24766
[100]	203	TComYuv* rpcResiPredYuv,
[77]	204	#endif
[2]	205	TComYuv*& rpcPredYuv,
	206	TComYuv*& rpcResiYuv,
	207	TComYuv*& rpcRecoYuv,
[56]	208	Bool bUseRes = false
	209	#if AMP_MRG
	210	,Bool bUseMRG = false
	211	#endif
[166]	212	#if FORCE_REF_VSP==1
	213	,Bool bForceRefVsp = false
	214	#endif
[56]	215	);
[2]	216
[56]	217	#if HHI_INTER_VIEW_RESIDUAL_PRED
[2]	218	/// encode residual and compute rd-cost for inter mode
	219	Void encodeResAndCalcRdInterCU( TComDataCU* pcCU,
	220	TComYuv* pcYuvOrg,
	221	TComYuv* pcYuvPred,
	222	TComYuv*& rpcYuvResi,
	223	TComYuv*& rpcYuvResiBest,
	224	TComYuv*& rpcYuvRec,
	225	TComYuv*& rpcYuvResPrd,
	226	Bool bSkipRes );
[56]	227	#else
	228	/// encode residual and compute rd-cost for inter mode
	229	Void encodeResAndCalcRdInterCU( TComDataCU* pcCU,
	230	TComYuv* pcYuvOrg,
	231	TComYuv* pcYuvPred,
	232	TComYuv*& rpcYuvResi,
	233	TComYuv*& rpcYuvResiBest,
	234	TComYuv*& rpcYuvRec,
	235	Bool bSkipRes );
	236	#endif
[2]	237	/// set ME search range
	238	Void setAdaptiveSearchRange ( Int iDir, Int iRefIdx, Int iSearchRange) { m_aaiAdaptSR[iDir][iRefIdx] = iSearchRange; }
	239
[56]	240	Void xEncPCM (TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piOrg, Pel* piPCM, Pel* piPred, Pel* piResi, Pel* piReco, UInt uiStride, UInt uiWidth, UInt uiHeight, TextType eText);
	241	Void IPCMSearch (TComDataCU* pcCU, TComYuv* pcOrgYuv, TComYuv& rpcPredYuv, TComYuv& rpcResiYuv, TComYuv*& rpcRecoYuv );
[2]	242	protected:
	243
	244	// -------------------------------------------------------------------------------------------------------------------
	245	// Intra search
	246	// -------------------------------------------------------------------------------------------------------------------
	247
	248	Void xEncSubdivCbfQT ( TComDataCU* pcCU,
	249	UInt uiTrDepth,
	250	UInt uiAbsPartIdx,
	251	Bool bLuma,
	252	Bool bChroma );
	253	Void xEncCoeffQT ( TComDataCU* pcCU,
	254	UInt uiTrDepth,
	255	UInt uiAbsPartIdx,
	256	TextType eTextType,
	257	Bool bRealCoeff );
	258	Void xEncIntraHeader ( TComDataCU* pcCU,
	259	UInt uiTrDepth,
	260	UInt uiAbsPartIdx,
	261	Bool bLuma,
	262	Bool bChroma );
	263	UInt xGetIntraBitsQT ( TComDataCU* pcCU,
	264	UInt uiTrDepth,
	265	UInt uiAbsPartIdx,
	266	Bool bLuma,
	267	Bool bChroma,
[56]	268	Bool bRealCoeff );
[2]	269
	270	Void xIntraCodingLumaBlk ( TComDataCU* pcCU,
	271	UInt uiTrDepth,
	272	UInt uiAbsPartIdx,
	273	TComYuv* pcOrgYuv,
	274	TComYuv* pcPredYuv,
	275	TComYuv* pcResiYuv,
[100]	276	Dist& ruiDist
[213]	277	#if LG_ZEROINTRADEPTHRESI_A0087
[100]	278	,Bool bZeroResi = false
	279	#endif
	280	);
[2]	281	Void xIntraCodingChromaBlk ( TComDataCU* pcCU,
	282	UInt uiTrDepth,
	283	UInt uiAbsPartIdx,
	284	TComYuv* pcOrgYuv,
	285	TComYuv* pcPredYuv,
	286	TComYuv* pcResiYuv,
	287	Dist& ruiDist,
	288	UInt uiChromaId );
	289	Void xRecurIntraCodingQT ( TComDataCU* pcCU,
	290	UInt uiTrDepth,
	291	UInt uiAbsPartIdx,
	292	Bool bLumaOnly,
	293	TComYuv* pcOrgYuv,
	294	TComYuv* pcPredYuv,
	295	TComYuv* pcResiYuv,
	296	Dist& ruiDistY,
	297	Dist& ruiDistC,
	298	#if HHI_RQT_INTRA_SPEEDUP
[100]	299	Bool bCheckFirst,
[2]	300	#endif
[100]	301	Double& dRDCost
[213]	302	#if LG_ZEROINTRADEPTHRESI_A0087
[100]	303	,Bool bZeroResi = false
	304	#endif
	305	);
[2]	306
	307	Void xSetIntraResultQT ( TComDataCU* pcCU,
	308	UInt uiTrDepth,
	309	UInt uiAbsPartIdx,
	310	Bool bLumaOnly,
	311	TComYuv* pcRecoYuv );
	312
	313	Void xRecurIntraChromaCodingQT ( TComDataCU* pcCU,
	314	UInt uiTrDepth,
	315	UInt uiAbsPartIdx,
	316	TComYuv* pcOrgYuv,
	317	TComYuv* pcPredYuv,
	318	TComYuv* pcResiYuv,
	319	Dist& ruiDist );
	320	Void xSetIntraResultChromaQT ( TComDataCU* pcCU,
	321	UInt uiTrDepth,
	322	UInt uiAbsPartIdx,
	323	TComYuv* pcRecoYuv );
	324
[213]	325	#if RWTH_SDC_DLT_B0036
	326	Void xAnalyzeSegmentsSDC ( Pel* pOrig,
	327	UInt uiStride,
	328	UInt uiSize,
	329	Pel* rpSegMeans,
	330	UInt uiNumSegments,
	331	Bool* pMask,
	332	UInt uiMaskStride );
	333
	334	Void xIntraCodingSDC ( TComDataCU* pcCU, UInt uiAbsPartIdx, TComYuv* pcOrgYuv, TComYuv* pcPredYuv, Dist& ruiDist, Double& dRDCost, Bool bResidual );
	335	#endif
	336
[2]	337	// -------------------------------------------------------------------------------------------------------------------
[56]	338	// DMM intra search
	339	// -------------------------------------------------------------------------------------------------------------------
	340
	341	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
	342	Bool predIntraLumaDMMAvailable ( UInt uiMode,
	343	UInt uiWidth,
[213]	344	#if HHI_DMM_PRED_TEX && FLEX_CODING_ORDER_M23723
	345	UInt uiHeight,
	346	Bool bDMMAvailable34 );
	347	#else
	348	UInt uiHeight );
	349	#endif
	350
[56]	351	Void xGetWedgeDeltaDCsMinDist ( TComWedgelet* pcWedgelet,
	352	TComDataCU* pcCU,
	353	UInt uiAbsPtIdx,
	354	Pel* piOrig,
	355	Pel* piPredic,
	356	UInt uiStride,
	357	UInt uiWidth,
	358	UInt uiHeight,
	359	Int& riDeltaDC1,
	360	Int& riDeltaDC2,
	361	Bool bAboveAvail,
	362	Bool bLeftAvail );
	363	#endif
[100]	364
[213]	365	#if LGE_EDGE_INTRA_A0070
[100]	366	Bool xEdgePartition ( TComDataCU* pcCU, UInt uiPartIdx, Bool bPU4x4 );
	367	Bool xCheckTerminatedEdge ( Bool* pbEdge, Int iX, Int iY, Int iWidth, Int iHeight );
	368	Bool xConstructChainCode ( TComDataCU* pcCU, UInt uiPartIdx, Bool bPU4x4 );
	369	#if LGE_EDGE_INTRA_DELTA_DC
	370	Void xAssignEdgeIntraDeltaDCs( TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piOrig, UInt uiStride, Pel* piPredic, UInt uiWidth, UInt uiHeight );
	371	#endif
	372	#endif
	373
[56]	374	#if HHI_DMM_WEDGE_INTRA
	375	Void findWedgeFullMinDist ( TComDataCU* pcCU,
	376	UInt uiAbsPtIdx,
	377	Pel* piOrig,
	378	Pel* piPredic,
	379	UInt uiStride,
	380	UInt uiWidth,
	381	UInt uiHeight,
	382	UInt& ruiTabIdx,
	383	Int& riDeltaDC1,
	384	Int& riDeltaDC2,
	385	Bool bAboveAvail,
	386	Bool bLeftAvail );
	387	Void findWedgePredDirMinDist ( TComDataCU* pcCU,
	388	UInt uiAbsPtIdx,
	389	Pel* piOrig,
	390	Pel* piPredic,
	391	UInt uiStride,
	392	UInt uiWidth,
	393	UInt uiHeight,
	394	UInt& ruiTabIdx,
	395	Int& riWedgeDeltaEnd,
	396	Int& riDeltaDC1,
	397	Int& riDeltaDC2,
	398	Bool bAboveAvail,
	399	Bool bLeftAvail );
	400	Void xSearchWedgeFullMinDist ( TComDataCU* pcCU,
	401	UInt uiAbsPtIdx,
	402	WedgeList* pacWedgeList,
	403	Pel* piRef,
	404	UInt uiRefStride,
	405	UInt uiWidth,
	406	UInt uiHeight,
	407	UInt& ruiTabIdx,
	408	Dist& riDist );
[213]	409	#if HHIQC_DMMFASTSEARCH_B0039
	410	Void xSearchWedgeFullMinDistFast( TComDataCU* pcCU,
	411	UInt uiAbsPtIdx,
	412	WedgeNodeList* pacWedgeNodeList,
	413	WedgeList* pacWedgeList,
	414	Pel* piRef,
	415	UInt uiRefStride,
	416	UInt uiWidth,
	417	UInt uiHeight,
	418	UInt& ruiTabIdx,
	419	Dist& riDist );
	420	#endif
[56]	421	Void xSearchWedgePredDirMinDist ( TComDataCU* pcCU,
	422	UInt uiAbsPtIdx,
	423	WedgeList* pacWedgeList,
	424	Pel* piRef,
	425	UInt uiRefStride,
	426	UInt uiWidth,
	427	UInt uiHeight,
	428	UInt& ruiTabIdx,
	429	Int& riWedgeDeltaEnd );
	430	#endif
	431	#if HHI_DMM_PRED_TEX
	432	Void findWedgeTexMinDist ( TComDataCU* pcCU,
	433	UInt uiAbsPtIdx,
	434	Pel* piOrig,
	435	Pel* piPredic,
	436	UInt uiStride,
	437	UInt uiWidth,
	438	UInt uiHeight,
	439	UInt& ruiTabIdx,
	440	Int& riDeltaDC1,
	441	Int& riDeltaDC2,
	442	Bool bAboveAvail,
	443	Bool bLeftAvail );
	444	Void findContourPredTex ( TComDataCU* pcCU,
	445	UInt uiAbsPtIdx,
	446	Pel* piOrig,
	447	Pel* piPredic,
	448	UInt uiStride,
	449	UInt uiWidth,
	450	UInt uiHeight,
	451	Int& riDeltaDC1,
	452	Int& riDeltaDC2,
	453	Bool bAboveAvail,
	454	Bool bLeftAvail );
	455	#endif
	456
	457	// -------------------------------------------------------------------------------------------------------------------
[2]	458	// Inter search (AMP)
	459	// -------------------------------------------------------------------------------------------------------------------
	460
	461	Void xEstimateMvPredAMVP ( TComDataCU* pcCU,
	462	TComYuv* pcOrgYuv,
	463	UInt uiPartIdx,
	464	RefPicList eRefPicList,
	465	Int iRefIdx,
	466	TComMv& rcMvPred,
	467	Bool bFilled = false
[56]	468	#if H0111_MVD_L1_ZERO
	469	, UInt* puiDistBiP = NULL
	470	#endif
[2]	471	#if ZERO_MVD_EST
	472	, UInt* puiDist = NULL
	473	#endif
	474	);
	475
	476	Void xCheckBestMVP ( TComDataCU* pcCU,
	477	RefPicList eRefPicList,
	478	TComMv cMv,
	479	TComMv& rcMvPred,
	480	Int& riMVPIdx,
	481	UInt& ruiBits,
	482	UInt& ruiCost );
	483
	484	UInt xGetTemplateCost ( TComDataCU* pcCU,
	485	UInt uiPartIdx,
	486	UInt uiPartAddr,
	487	TComYuv* pcOrgYuv,
	488	TComYuv* pcTemplateCand,
	489	TComMv cMvCand,
	490	Int iMVPIdx,
	491	Int iMVPNum,
	492	RefPicList eRefPicList,
	493	Int iRefIdx,
	494	Int iSizeX,
	495	Int iSizeY
	496	#if ZERO_MVD_EST
	497	, UInt& ruiDist
	498	#endif
	499	);
	500
	501
	502	Void xCopyAMVPInfo ( AMVPInfo* pSrc, AMVPInfo* pDst );
	503	UInt xGetMvpIdxBits ( Int iIdx, Int iNum );
	504	Void xGetBlkBits ( PartSize eCUMode, Bool bPSlice, Int iPartIdx, UInt uiLastMode, UInt uiBlkBit[3]);
	505
	506	Void xMergeEstimation ( TComDataCU* pcCU,
	507	TComYuv* pcYuvOrg,
[77]	508	#if LG_RESTRICTEDRESPRED_M24766
[100]	509	TComYuv* rpcResiPredYuv,
[77]	510	#endif
[2]	511	Int iPartIdx,
	512	UInt& uiInterDir,
	513	TComMvField* pacMvField,
	514	UInt& uiMergeIndex,
[56]	515	UInt& ruiCost
	516	#if CU_BASED_MRG_CAND_LIST
	517	, TComMvField* cMvFieldNeighbours,
	518	UChar* uhInterDirNeighbours,
	519	Int& numValidMergeCand
	520	#endif
	521	);
[2]	522	// -------------------------------------------------------------------------------------------------------------------
	523	// motion estimation
	524	// -------------------------------------------------------------------------------------------------------------------
	525
	526	Void xMotionEstimation ( TComDataCU* pcCU,
	527	TComYuv* pcYuvOrg,
	528	Int iPartIdx,
	529	RefPicList eRefPicList,
	530	TComMv* pcMvPred,
	531	Int iRefIdxPred,
	532	TComMv& rcMv,
[166]	533	#if VSP_MV_ZERO
	534	Bool bIsVsp,
	535	#endif
[2]	536	UInt& ruiBits,
	537	UInt& ruiCost,
	538	Bool bBi = false );
	539
	540	Void xTZSearch ( TComDataCU* pcCU,
	541	TComPattern* pcPatternKey,
	542	Pel* piRefY,
	543	Int iRefStride,
	544	TComMv* pcMvSrchRngLT,
	545	TComMv* pcMvSrchRngRB,
	546	TComMv& rcMv,
	547	UInt& ruiSAD );
	548
[213]	549	#if DV_V_RESTRICTION_B0037
[2]	550	Void xSetSearchRange ( TComDataCU* pcCU,
	551	TComMv& cMvPred,
	552	Int iSrchRng,
	553	TComMv& rcMvSrchRngLT,
[213]	554	TComMv& rcMvSrchRngRB,
	555	Bool bDispSrchRngRst,
	556	Int iDispVerSrchRng );
	557	#else
	558	Void xSetSearchRange ( TComDataCU* pcCU,
	559	TComMv& cMvPred,
	560	Int iSrchRng,
	561	TComMv& rcMvSrchRngLT,
[2]	562	TComMv& rcMvSrchRngRB );
[213]	563	#endif
[2]	564
	565	Void xPatternSearchFast ( TComDataCU* pcCU,
	566	TComPattern* pcPatternKey,
	567	Pel* piRefY,
	568	Int iRefStride,
	569	TComMv* pcMvSrchRngLT,
	570	TComMv* pcMvSrchRngRB,
	571	TComMv& rcMv,
	572	UInt& ruiSAD );
	573
	574	Void xPatternSearch ( TComPattern* pcPatternKey,
	575	Pel* piRefY,
	576	Int iRefStride,
	577	TComMv* pcMvSrchRngLT,
	578	TComMv* pcMvSrchRngRB,
	579	TComMv& rcMv,
	580	UInt& ruiSAD );
	581
	582	Void xPatternSearchFracDIF ( TComDataCU* pcCU,
	583	TComPattern* pcPatternKey,
	584	Pel* piRefY,
	585	Int iRefStride,
	586	TComMv* pcMvInt,
	587	TComMv& rcMvHalf,
	588	TComMv& rcMvQter,
	589	UInt& ruiCost
[56]	590	,Bool biPred
[166]	591	#if VSP_MV_ZERO
	592	,Bool bIsVsp
	593	#endif
[2]	594	);
	595
[56]	596	Void xExtDIFUpSamplingH( TComPattern* pcPattern, Bool biPred );
	597	Void xExtDIFUpSamplingQ( TComPattern* pcPatternKey, TComMv halfPelRef, Bool biPred );
[2]	598
	599	// -------------------------------------------------------------------------------------------------------------------
	600	// T & Q & Q-1 & T-1
	601	// -------------------------------------------------------------------------------------------------------------------
	602
	603	Void xEncodeResidualQT( TComDataCU* pcCU, UInt uiAbsPartIdx, const UInt uiDepth, Bool bSubdivAndCbf, TextType eType );
[56]	604	#if IBDI_DISTORTION \|\| HHI_VSO
	605	Void 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 );
	606	#else
	607	Void xEstimateResidualQT( TComDataCU* pcCU, UInt uiQuadrant, UInt uiAbsPartIdx, UInt absTUPartIdx , TComYuv* pcResi, const UInt uiDepth, Double &rdCost, UInt &ruiBits, Dist &ruiDist, Dist *puiZeroDist );
	608	#endif
	609	Void xSetResidualQTData( TComDataCU* pcCU, UInt uiQuadrant, UInt uiAbsPartIdx,UInt absTUPartIdx, TComYuv* pcResi, UInt uiDepth, Bool bSpatial );
[2]	610
	611	UInt xModeBitsIntra ( TComDataCU* pcCU, UInt uiMode, UInt uiPU, UInt uiPartOffset, UInt uiDepth, UInt uiInitTrDepth );
	612	UInt xUpdateCandList( UInt uiMode, Double uiCost, UInt uiFastCandNum, UInt * CandModeList, Double * CandCostList );
	613
	614	// -------------------------------------------------------------------------------------------------------------------
	615	// compute symbol bits
	616	// -------------------------------------------------------------------------------------------------------------------
	617
	618	Void xAddSymbolBitsInter ( TComDataCU* pcCU,
	619	UInt uiQp,
	620	UInt uiTrMode,
	621	UInt& ruiBits,
	622	TComYuv*& rpcYuvRec,
	623	TComYuv* pcYuvPred,
	624	TComYuv*& rpcYuvResi );
	625
[56]	626	Void setWpScalingDistParam( TComDataCU* pcCU, Int iRefIdx, RefPicList eRefPicListCur );
[2]	627	inline Void setDistParamComp( UInt uiComp ) { m_cDistParam.uiComp = uiComp; }
[56]	628
[2]	629	};// END CLASS DEFINITION TEncSearch
	630
[56]	631	//! \}
[2]	632
	633	#endif // __TENCSEARCH__

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

JCT-3V 3D-HEVC

Context navigation

source: 3DVCSoftware/branches/HTM-4.0.1-VSP-dev0/source/Lib/TLibEncoder/TEncSearch.h

Download in other formats: