Context navigation

source: 3DVCSoftware/trunk/source/Lib/TLibCommon/TComPrediction.h @ 605

Visit:

Last change on this file since 605 was 443, checked in by tech, 12 years ago
Reintegrated branch 6.2-dev0 rev. 442. Changed version number. Added coding results.
Property svn:eol-style set to `native`
File size: 15.6 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 TComPrediction.h
	35	\brief prediction class (header)
	36	*/
	37
	38	#ifndef __TCOMPREDICTION__
	39	#define __TCOMPREDICTION__
	40
	41
	42	// Include files
	43	#include "TComPic.h"
	44	#include "TComMotionInfo.h"
	45	#include "TComPattern.h"
	46	#include "TComTrQuant.h"
[56]	47	#include "TComInterpolationFilter.h"
	48	#include "TComWeightPrediction.h"
	49
	50	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
	51	#include <math.h>
[2]	52	#endif
	53
[56]	54	//! \ingroup TLibCommon
	55	//! \{
	56
[2]	57	// ====================================================================================================================
	58	// Class definition
	59	// ====================================================================================================================
	60
	61	/// prediction class
[56]	62	class TComPrediction : public TComWeightPrediction
[2]	63	{
	64	protected:
	65	Int* m_piYuvExt;
	66	Int m_iYuvExtStride;
	67	Int m_iYuvExtHeight;
	68
	69	TComYuv m_acYuvPred[2];
	70	TComYuv m_cYuvPredTemp;
[443]	71	#if QC_ARP_D0177
	72	TComYuv m_acYuvPredBase[2];
	73	TComYuv m_acYuvDiff[2];
	74	#endif
[56]	75	TComYuv m_filteredBlock[4][4];
	76	TComYuv m_filteredBlockTmp[4];
[2]	77
[56]	78	TComInterpolationFilter m_if;
	79
	80	Pel* m_pLumaRecBuffer; ///< array for downsampled reconstructed luma sample
	81	Int m_iLumaRecStride; ///< stride of #m_pLumaRecBuffer array
	82	UInt m_uiaShift[ 63 ]; // Table for multiplication to substitue of division operation
[2]	83
[296]	84	#if MERL_VSP_C0152
	85	Int* m_pDepth; ///< Local variable, to store a depth block, just to prevent allocate memory every time
[443]	86
	87	#if MERL_CVSP_D0165
	88	static Int m_iRangeLuma[12];
	89	static Int m_iRangeChroma[12];
[296]	90	#endif
[443]	91	#endif
[296]	92
[56]	93	Void xPredIntraAng ( Int* pSrc, Int srcStride, Pel*& rpDst, Int dstStride, UInt width, UInt height, UInt dirMode, Bool blkAboveAvailable, Bool blkLeftAvailable, Bool bFilter );
	94	Void xPredIntraPlanar ( Int* pSrc, Int srcStride, Pel* rpDst, Int dstStride, UInt width, UInt height );
[2]	95
	96	// motion compensation functions
[443]	97	#if QC_ARP_D0177
	98	Void xPredInterUniARP ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv& rpcYuvPred, Int iPartIdx, Bool bi=false, TComMvField pNewMvFiled = NULL );
	99	#endif
[296]	100	#if MERL_VSP_C0152
	101	Void xPredInterUni ( TComDataCU* pcCU, UInt uiPartAddr, UInt uiAbsPartIdx, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv*& rpcYuvPred, Int iPartIdx, Bool bPrdDepthMap, UInt uiSubSampExpX = 0, UInt uiSubSampExpY = 0, Bool bi=false );
	102	#else
[21]	103	Void xPredInterUni ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv*& rpcYuvPred, Int iPartIdx, Bool bPrdDepthMap, UInt uiSubSampExpX = 0, UInt uiSubSampExpY = 0, Bool bi=false );
[296]	104	#endif
[56]	105
[296]	106	#if MERL_VSP_C0152
	107	Void xPredInterBi ( TComDataCU* pcCU, UInt uiPartAddr, UInt uiAbsPartIdx, Int iWidth, Int iHeight, UInt uiSubSampExpX, UInt uiSubSampExpY, TComYuv*& rpcYuvPred, Int iPartIdx, Bool bPrdDepthMap );
	108	#else
[21]	109	Void xPredInterBi ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, UInt uiSubSampExpX, UInt uiSubSampExpY, TComYuv*& rpcYuvPred, Int iPartIdx, Bool bPrdDepthMap );
[296]	110	#endif
	111	Void xPredInterPrdDepthMap ( TComDataCU* pcCU, TComPicYuv* pcPicYuvRef, UInt uiPartAddr, TComMv* pcMv, Int iWidth, Int iHeight, UInt uiSubSampExpX, UInt uiSubSampExpY, TComYuv*& rpcYuv, UInt uiRShift, UInt uiOffset
	112	#if LGE_ILLUCOMP_DEPTH_C0046
	113	, Bool bICFlag = false
	114	#endif
	115	);
[332]	116
[443]	117	#if !MERL_General_Fix
[296]	118	#if MERL_VSP_C0152
	119	Void xPredInterUniBWVSP ( TComDataCU* pcCU, UInt uiPartAddr, UInt uiAbsPartIdx, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv*& rpcYuvPred, Int iPartIdx, Bool bPrdDepthMap, UInt uiSubSampExpX, UInt uiSubSampExpY, Bool bi=false );
	120	Void xPredInterBiBWVSP ( TComDataCU* pcCU, UInt uiPartAddr, UInt uiAbsPartIdx, Int iWidth, Int iHeight, UInt uiSubSampExpX, UInt uiSubSampExpY, TComYuv*& rpcYuvPred, Int iPartIdx, Bool bPrdDepthMap );
	121	#endif
[443]	122	#endif
[56]	123
[21]	124	#if DEPTH_MAP_GENERATION
	125	Void xWeightedAveragePdm ( TComDataCU* pcCU, TComYuv* pcYuvSrc0, TComYuv* pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv*& rpcYuvDst, UInt uiSubSampExpX, UInt uiSubSampExpY );
	126	#endif
[56]	127
[189]	128	#if LGE_ILLUCOMP_B0045
[443]	129	Void xPredInterLumaBlk ( TComDataCU cu, TComPicYuv refPic, UInt partAddr, TComMv mv, Int width, Int height, TComYuv &dstPic, Bool bi, Bool bICFlag = false
	130	#if QC_ARP_D0177
	131	,
	132	Int filterType = 0
	133	#endif
	134	);
	135	Void xPredInterChromaBlk( TComDataCU cu, TComPicYuv refPic, UInt partAddr, TComMv mv, Int width, Int height, TComYuv &dstPic, Bool bi, Bool bICFlag = false
	136	#if QC_ARP_D0177
	137	,
	138	Int filterType = 0
	139	#endif
	140	);
[189]	141	#else
[443]	142	Void xPredInterLumaBlk ( TComDataCU cu, TComPicYuv refPic, UInt partAddr, TComMv mv, Int width, Int height, TComYuv &dstPic, Bool bi
	143	#if QC_ARP_D0177
	144	,
	145	Int filterType = 0
[189]	146	#endif
[443]	147	);
	148	Void xPredInterChromaBlk( TComDataCU cu, TComPicYuv refPic, UInt partAddr, TComMv mv, Int width, Int height, TComYuv &dstPic, Bool bi
	149	#if QC_ARP_D0177
	150	,
	151	Int filterType = 0
	152	#endif
	153	);
	154	#endif
	155
	156	#if MERL_Bi_VSP_D0166
	157	Void xWeightedAverage ( TComDataCU* pcCU, TComYuv* pcYuvSrc0, TComYuv* pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv*& rpcYuvDst, Int predDir);
	158	#else
[2]	159	Void xWeightedAverage ( TComDataCU* pcCU, TComYuv* pcYuvSrc0, TComYuv* pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv*& rpcYuvDst );
[443]	160	#endif
[296]	161
	162	#if MERL_VSP_C0152
[443]	163	#if !MERL_Bi_VSP_D0166
	164	Void xPredInterLumaBlkFromDM ( TComPicYuv refPic, TComPicYuv pPicBaseDepth, Int* pShiftLUT, Int iShiftPrec, TComMv *mv, UInt partAddr,Int posX, Int posY, Int sizeX, Int sizeY, Bool isDepth, Int vspIdx
[296]	165	, TComYuv *&dstPic );
[443]	166	Void xPredInterChromaBlkFromDM ( TComPicYuv refPic, TComPicYuv pPicBaseDepth, Int* pShiftLUT, Int iShiftPrec, TComMv *mv, UInt partAddr,Int posX, Int posY, Int sizeX, Int sizeY, Bool isDepth, Int vspIdx
[296]	167	, TComYuv *&dstPic );
[443]	168	#else
	169	TComPic* xGetVspRefTxt(TComDataCU* pcCU, UInt uiPartAddr, RefPicList eRefPicList);
	170	Void xPredInterLumaBlkFromDM ( TComPicYuv refPic, TComPicYuv pPicBaseDepth, Int* pShiftLUT, Int iShiftPrec, TComMv *mv, UInt partAddr,Int posX, Int posY, Int sizeX, Int sizeY, Bool isDepth,
	171	TComYuv *&dstPic, Bool bi);
	172	Void xPredInterChromaBlkFromDM ( TComPicYuv refPic, TComPicYuv pPicBaseDepth, Int* pShiftLUT, Int iShiftPrec, TComMv *mv, UInt partAddr,Int posX, Int posY, Int sizeX, Int sizeY, Bool isDepth,
	173	TComYuv *&dstPic, Bool bi);
[296]	174	#endif
[443]	175	#endif
	176
[56]	177	Void xGetLLSPrediction ( TComPattern* pcPattern, Int* pSrc0, Int iSrcStride, Pel* pDst0, Int iDstStride, UInt uiWidth, UInt uiHeight, UInt uiExt0 );
[189]	178	#if LGE_ILLUCOMP_B0045
	179	Void xGetLLSICPrediction(TComDataCU* pcCU, TComMv pMv, TComPicYuv pRefPic, Int &a, Int &b, Int &iShift);
	180	Void xGetLLSICPredictionChroma(TComDataCU* pcCU, TComMv pMv, TComPicYuv pRefPic, Int &a, Int &b, Int &iShift, Int iChromaId);
	181	#endif
[56]	182	Void xDCPredFiltering( Int* pSrc, Int iSrcStride, Pel*& rpDst, Int iDstStride, Int iWidth, Int iHeight );
	183	Bool xCheckIdenticalMotion ( TComDataCU* pcCU, UInt PartAddr);
[21]	184
[443]	185	#if MERL_General_Fix
	186	#if MERL_VSP_C0152
	187	Void xPredInterVSPBlk(TComDataCU* pcCU, UInt uiPartAddr, UInt uiAbsPartIdx, Int iWidth, Int iHeight, TComMv cMv, RefPicList eRefPicList, TComYuv*& rpcYuvPred, Bool bi
	188	#if !MERL_Bi_VSP_D0166
	189	, Int vspIdx
	190	#endif
	191	);
	192	#endif
	193	#endif
	194
[5]	195	#if HHI_DMM_WEDGE_INTRA
[2]	196	Void xPredIntraWedgeFull ( TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, Bool bAbove, Bool bLeft, Bool bEncoder, Bool bDelta, UInt uiTabIdx, Int iDeltaDC1 = 0, Int iDeltaDC2 = 0 );
	197	Void xPredIntraWedgeDir ( TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, Bool bAbove, Bool bLeft, Bool bEncoder, Bool bDelta, Int iWedgeDeltaEnd, Int iDeltaDC1 = 0, Int iDeltaDC2 = 0 );
	198	Void xGetBlockOffset ( TComDataCU* pcCU, UInt uiAbsPartIdx, TComDataCU* pcRefCU, UInt uiRefAbsPartIdx, UInt& ruiOffsetX, UInt& ruiOffsetY );
	199	Bool xGetWedgeIntraDirPredData ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiBlockSize, Int& riSlopeX, Int& riSlopeY, UInt& ruiStartPosX, UInt& ruiStartPosY );
	200	Void xGetWedgeIntraDirStartEnd ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiBlockSize, Int iDeltaX, Int iDeltaY, UInt uiPMSPosX, UInt uiPMSPosY, UChar& ruhXs, UChar& ruhYs, UChar& ruhXe, UChar& ruhYe, Int iDeltaEnd = 0 );
[5]	201	#endif
	202	#if HHI_DMM_PRED_TEX
	203	Void xPredIntraWedgeTex ( TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, Bool bAbove, Bool bLeft, Bool bEncoder, Bool bDelta, Int iDeltaDC1 = 0, Int iDeltaDC2 = 0 );
	204	Void xPredIntraContourTex ( TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, Bool bAbove, Bool bLeft, Bool bEncoder, Bool bDelta, Int iDeltaDC1 = 0, Int iDeltaDC2 = 0 );
	205	#endif
	206	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
[2]	207	Void xDeltaDCQuantScaleUp ( TComDataCU* pcCU, Int& riDeltaDC );
[56]	208	Void xDeltaDCQuantScaleDown ( TComDataCU* pcCU, Int& riDeltaDC );
[2]	209	#endif
	210
[189]	211	#if LGE_EDGE_INTRA_A0070
[100]	212	Pel xGetNearestNeighbor ( Int x, Int y, Int* pSrc, Int srcStride, Int iWidth, Int iHeight, Bool* bpRegion );
	213	Void xPredIntraEdge ( TComDataCU* pcCU, UInt uiAbsPartIdx, Int iWidth, Int iHeight, Int* pSrc, Int srcStride, Pel*& rpDst, Int dstStride, Bool bDelta = false );
	214	#endif
	215
[2]	216	public:
	217	TComPrediction();
	218	virtual ~TComPrediction();
	219
	220	Void initTempBuff();
	221
	222	// inter
[296]	223	#if MERL_VSP_C0152
	224	Void motionCompensation ( TComDataCU* pcCU, TComYuv* pcYuvPred, UInt uiAbsPartIdx, RefPicList eRefPicList = REF_PIC_LIST_X, Int iPartIdx = -1, Bool bPrdDepthMap = false, UInt uiSubSampExpX = 0, UInt uiSubSampExpY = 0 );
	225	#else
[21]	226	Void motionCompensation ( TComDataCU* pcCU, TComYuv* pcYuvPred, RefPicList eRefPicList = REF_PIC_LIST_X, Int iPartIdx = -1, Bool bPrdDepthMap = false, UInt uiSubSampExpX = 0, UInt uiSubSampExpY = 0 );
[296]	227	#endif
	228
[443]	229	#if H3D_IVRP & !QC_ARP_D0177
[296]	230	Void residualPrediction (TComDataCU* pcCU, TComYuv* pcYuvPred, TComYuv* pcYuvResPred);
	231	#endif
	232
[2]	233	// motion vector prediction
	234	Void getMvPredAMVP ( TComDataCU* pcCU, UInt uiPartIdx, UInt uiPartAddr, RefPicList eRefPicList, Int iRefIdx, TComMv& rcMvPred );
	235
	236	// Angular Intra
	237	Void predIntraLumaAng ( TComPattern* pcTComPattern, UInt uiDirMode, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, TComDataCU* pcCU, Bool bAbove, Bool bLeft );
	238	Void predIntraChromaAng ( TComPattern* pcTComPattern, Int* piSrc, UInt uiDirMode, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, TComDataCU* pcCU, Bool bAbove, Bool bLeft );
	239
[56]	240	Pel predIntraGetPredValDC ( Int* pSrc, Int iSrcStride, UInt iWidth, UInt iHeight, Bool bAbove, Bool bLeft );
	241
	242	Int* getPredicBuf() { return m_piYuvExt; }
	243	Int getPredicBufWidth() { return m_iYuvExtStride; }
	244	Int getPredicBufHeight() { return m_iYuvExtHeight; }
	245
	246	Void predLMIntraChroma( TComPattern* pcPattern, Int* piSrc, Pel* pPred, UInt uiPredStride, UInt uiCWidth, UInt uiCHeight, UInt uiChromaId );
	247	Void getLumaRecPixels ( TComPattern* pcPattern, UInt uiWidth0, UInt uiHeight0 );
	248
[5]	249	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
[56]	250	Void predIntraLumaDMM ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiMode, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight, Bool bAbove, Bool bLeft, Bool bEncoder );
[443]	251	#if QC_DC_PREDICTOR_D0183
	252	Void getPredDCs ( Bool* pbPattern, Int iPatternStride, Int* piMask, Int iMaskStride, Int& riPredDC1, Int& riPredDC2 );
	253	#else
[2]	254	Void getWedgePredDCs ( TComWedgelet* pcWedgelet, Int* piMask, Int iMaskStride, Int& riPredDC1, Int& riPredDC2, Bool bAbove, Bool bLeft );
[443]	255	#endif
[56]	256	Void calcWedgeDCs ( TComWedgelet* pcWedgelet, Pel* piOrig, UInt uiStride, Int& riDC1, Int& riDC2 );
	257	Void assignWedgeDCs2Pred ( TComWedgelet* pcWedgelet, Pel* piPred, UInt uiStride, Int iDC1, Int iDC2 );
[2]	258	#endif
[5]	259	#if HHI_DMM_PRED_TEX
[56]	260	Void getBestContourFromTex ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, TComWedgelet* pcContourWedge );
[296]	261	#if LGE_DMM3_SIMP_C0044
	262	UInt getBestWedgeFromTex ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, Pel * piOrgi, UInt uiStride, UInt &ruiIntraTabIdx);//encoder
	263	UInt getBestWedgeFromTex ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt IntraTabIdx);//decoder
	264	#else
[56]	265	UInt getBestWedgeFromTex ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight );
[296]	266	#endif
[56]	267	Void copyTextureLumaBlock ( TComDataCU* pcCU, UInt uiAbsPartIdx, Pel* piDestBlockY, UInt uiWidth, UInt uiHeight );
[5]	268	#endif
	269	#if HHI_DMM_WEDGE_INTRA
[56]	270	UInt getBestContinueWedge ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, Int iDeltaEnd = 0 );
	271	Bool getWedgePatternIdx ( WedgeRefList* pcWedgeRefList, UInt& ruiTabIdx, UChar uhXs, UChar uhYs, UChar uhXe, UChar uhYe );
[5]	272	#endif
[189]	273	#if LGE_EDGE_INTRA_A0070
[100]	274	Void predIntraLumaEdge ( TComDataCU* pcCU, TComPattern* pcTComPattern, UInt uiAbsPartIdx, Int iWidth, Int iHeight, Pel* piPred, UInt uiStride, Bool bDelta = false );
	275	#endif
[2]	276
	277	// simplified intra pred for "virtual" depth maps
	278	Void predIntraDepthAng ( TComPattern* pcTComPattern, UInt uiDirMode, Pel* piPred, UInt uiStride, Int iWidth, Int iHeight );
	279	Int xGetDCDepth ( Int* pSrc, Int iDelta, Int iBlkSize );
	280	Int xGetDCValDepth ( Int iVal1, Int iVal2, Int iVal3, Int iVal4 );
	281	Void xPredIntraAngDepth( Int* pSrc, Int srcStride, Pel* pDst, Int dstStride, UInt width, UInt height, UInt dirMode );
[56]	282
[2]	283	};
	284
[56]	285	//! \}
[2]	286
	287	#endif // __TCOMPREDICTION__

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

Download in other formats: