Context navigation

TEncSearch.h @ 577

Visit:

Last change on this file since 577 was 74, checked in by seregin, 12 years ago
remove REF_IDX_ME_AROUND_ZEROMV
File size: 24.2 KB

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

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

JCT-VC SHVC

Context navigation

source: SHVCSoftware/branches/HM-10.0-dev-SHM/source/Lib/TLibEncoder/TEncSearch.h @ 577

Download in other formats: