HEVC Test Model (HM)  HM-16.3
TComPrediction.h
Go to the documentation of this file.
1 /* The copyright in this software is being made available under the BSD
2  * License, included below. This software may be subject to other third party
3  * and contributor rights, including patent rights, and no such rights are
4  * granted under this license.
5  *
6  * Copyright (c) 2010-2015, ITU/ISO/IEC
7  * All rights reserved.
8  *
9  * Redistribution and use in source and binary forms, with or without
10  * modification, are permitted provided that the following conditions are met:
11  *
12  * * Redistributions of source code must retain the above copyright notice,
13  * this list of conditions and the following disclaimer.
14  * * Redistributions in binary form must reproduce the above copyright notice,
15  * this list of conditions and the following disclaimer in the documentation
16  * and/or other materials provided with the distribution.
17  * * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18  * be used to endorse or promote products derived from this software without
19  * specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31  * THE POSSIBILITY OF SUCH DAMAGE.
32  */
33 
38 #ifndef __TCOMPREDICTION__
39 #define __TCOMPREDICTION__
40 
41 
42 // Include files
43 #include "TComYuv.h"
45 #include "TComWeightPrediction.h"
46 
47 // forward declaration
48 class TComMv;
49 class TComTU;
50 
53 
54 // ====================================================================================================================
55 // Class definition
56 // ====================================================================================================================
57 
59 typedef enum PRED_BUF_E
60 {
64 } PRED_BUF;
65 
67 
69 {
70 private:
72 
73 protected:
76 
81 
83 
86 
87  Void xPredIntraAng ( Int bitDepth, const Pel* pSrc, Int srcStride, Pel* pDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format, UInt dirMode, Bool blkAboveAvailable, Bool blkLeftAvailable, const Bool bEnableEdgeFilters );
88  Void xPredIntraPlanar ( const Pel* pSrc, Int srcStride, Pel* rpDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format );
89 
90  // motion compensation functions
91  Void xPredInterUni ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv* pcYuvPred, Bool bi=false );
92  Void xPredInterBi ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv* pcYuvPred );
93  Void xPredInterBlk(const ComponentID compID, TComDataCU *cu, TComPicYuv *refPic, UInt partAddr, TComMv *mv, Int width, Int height, TComYuv *dstPic, Bool bi );
94  Void xWeightedAverage ( TComYuv* pcYuvSrc0, TComYuv* pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv* pcYuvDst );
95 
96  Void xGetLLSPrediction ( const Pel* pSrc0, Int iSrcStride, Pel* pDst0, Int iDstStride, UInt uiWidth, UInt uiHeight, UInt uiExt0, const ChromaFormat chFmt DEBUG_STRING_FN_DECLARE(sDebug) );
97 
98  Void xDCPredFiltering( const Pel* pSrc, Int iSrcStride, Pel* pDst, Int iDstStride, Int iWidth, Int iHeight, ChannelType channelType );
99  Bool xCheckIdenticalMotion ( TComDataCU* pcCU, UInt PartAddr);
100  Void destroy();
101 
102 public:
103  TComPrediction();
104  virtual ~TComPrediction();
105 
106  Void initTempBuff(ChromaFormat chromaFormatIDC);
107 
108  ChromaFormat getChromaFormat() const { return m_cYuvPredTemp.getChromaFormat(); }
109 
110  // inter
111  Void motionCompensation ( TComDataCU* pcCU, TComYuv* pcYuvPred, RefPicList eRefPicList = REF_PIC_LIST_X, Int iPartIdx = -1 );
112 
113  // motion vector prediction
114  Void getMvPredAMVP ( TComDataCU* pcCU, UInt uiPartIdx, UInt uiPartAddr, RefPicList eRefPicList, TComMv& rcMvPred );
115 
116  // Angular Intra
117  Void predIntraAng ( const ComponentID compID, UInt uiDirMode, Pel *piOrg /* Will be null for decoding */, UInt uiOrgStride, Pel* piPred, UInt uiStride, TComTU &rTu, Bool bAbove, Bool bLeft, const Bool bUseFilteredPredSamples, const Bool bUseLosslessDPCM = false );
118 
119  Pel predIntraGetPredValDC ( const Pel* pSrc, Int iSrcStride, UInt iWidth, UInt iHeight, ChannelType channelType, ChromaFormat format, Bool bAbove, Bool bLeft );
120 
121  Pel* getPredictorPtr ( const ComponentID compID, const Bool bUseFilteredPredictions )
122  {
123  return m_piYuvExt[compID][bUseFilteredPredictions?PRED_BUF_FILTERED:PRED_BUF_UNFILTERED];
124  }
125 
126  // This function is actually still in TComPattern.cpp
129  Bool& bAbove,
130  Bool& bLeft,
131  const ComponentID compID, const Bool bFilterRefSamples
133  );
134 
135  static Bool filteringIntraReferenceSamples(const ComponentID compID, UInt uiDirMode, UInt uiTuChWidth, UInt uiTuChHeight, const ChromaFormat chFmt, const Bool intraReferenceSmoothingDisabled);
136 
137  static Bool UseDPCMForFirstPassIntraEstimation(TComTU &rTu, const UInt uiDirMode);
138 };
139 
141 
142 #endif // __TCOMPREDICTION__
picture YUV buffer class
Definition: TComPicYuv.h:55
CU data structure class.
Definition: TComDataCU.h:64
ChannelType
Definition: TypeDef.h:361
weighting prediction class
void Void
Definition: TypeDef.h:285
Void xPredIntraPlanar(const Pel *pSrc, Int srcStride, Pel *rpDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format)
Void getMvPredAMVP(TComDataCU *pcCU, UInt uiPartIdx, UInt uiPartAddr, RefPicList eRefPicList, TComMv &rcMvPred)
static const UInt MAX_INTRA_FILTER_DEPTHS
TComYuv m_filteredBlock[4][4]
Declaration of TComInterpolationFilter class.
general YUV buffer class (header)
Interpolation filter class.
Void motionCompensation(TComDataCU *pcCU, TComYuv *pcYuvPred, RefPicList eRefPicList=REF_PIC_LIST_X, Int iPartIdx=-1)
unsigned int UInt
Definition: TypeDef.h:297
Void xGetLLSPrediction(const Pel *pSrc0, Int iSrcStride, Pel *pDst0, Int iDstStride, UInt uiWidth, UInt uiHeight, UInt uiExt0, const ChromaFormat chFmt)
TComYuv m_acYuvPred[NUM_REF_PIC_LIST_01]
Short Pel
pixel type
Definition: TypeDef.h:692
Void initTempBuff(ChromaFormat chromaFormatIDC)
static Bool filteringIntraReferenceSamples(const ComponentID compID, UInt uiDirMode, UInt uiTuChWidth, UInt uiTuChHeight, const ChromaFormat chFmt, const Bool intraReferenceSmoothingDisabled)
static const UChar m_aucIntraFilter[MAX_NUM_CHANNEL_TYPE][MAX_INTRA_FILTER_DEPTHS]
Void predIntraAng(const ComponentID compID, UInt uiDirMode, Pel *piOrg, UInt uiOrgStride, Pel *piPred, UInt uiStride, TComTU &rTu, Bool bAbove, Bool bLeft, const Bool bUseFilteredPredSamples, const Bool bUseLosslessDPCM=false)
#define LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS
Definition: TypeDef.h:212
TComYuv m_cYuvPredTemp
ChromaFormat getChromaFormat() const
general YUV buffer class
Definition: TComYuv.h:54
bool Bool
Definition: TypeDef.h:286
PRED_BUF_E
prediction class
Void xPredIntraAng(Int bitDepth, const Pel *pSrc, Int srcStride, Pel *pDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format, UInt dirMode, Bool blkAboveAvailable, Bool blkLeftAvailable, const Bool bEnableEdgeFilters)
Int m_iLumaRecStride
stride of m_pLumaRecBuffer array
RefPicList
reference list index
Definition: TypeDef.h:415
ChromaFormat
chroma formats (according to semantics of chroma_format_idc)
Definition: TypeDef.h:352
unsigned char UChar
Definition: TypeDef.h:293
TComYuv m_filteredBlockTmp[4]
Void xDCPredFiltering(const Pel *pSrc, Int iSrcStride, Pel *pDst, Int iDstStride, Int iWidth, Int iHeight, ChannelType channelType)
Void xWeightedAverage(TComYuv *pcYuvSrc0, TComYuv *pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv *pcYuvDst)
#define DEBUG_STRING_FN_DECLARE(name)
Definition: TypeDef.h:71
Definition: TComTU.h:48
ChromaFormat getChromaFormat() const
Definition: TComYuv.h:200
Pel * m_pLumaRecBuffer
array for downsampled reconstructed luma sample
TComInterpolationFilter m_if
enum PRED_BUF_E PRED_BUF
prediction class
Pel predIntraGetPredValDC(const Pel *pSrc, Int iSrcStride, UInt iWidth, UInt iHeight, ChannelType channelType, ChromaFormat format, Bool bAbove, Bool bLeft)
Void initAdiPatternChType(TComTU &rTu, Bool &bAbove, Bool &bLeft, const ComponentID compID, const Bool bFilterRefSamples)
set parameters from CU data for accessing ADI data
Pel * getPredictorPtr(const ComponentID compID, const Bool bUseFilteredPredictions)
int Int
Definition: TypeDef.h:296
basic motion vector class
Definition: TComMv.h:51
Bool xCheckIdenticalMotion(TComDataCU *pcCU, UInt PartAddr)
virtual ~TComPrediction()
ComponentID
Definition: TypeDef.h:368
Void xPredInterBi(TComDataCU *pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv *pcYuvPred)
weighting prediction class (header)
Void xPredInterBlk(const ComponentID compID, TComDataCU *cu, TComPicYuv *refPic, UInt partAddr, TComMv *mv, Int width, Int height, TComYuv *dstPic, Bool bi)
Generate motion-compensated block.
special mark
Definition: TypeDef.h:420
Void xPredInterUni(TComDataCU *pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv *pcYuvPred, Bool bi=false)
Pel * m_piYuvExt[MAX_NUM_COMPONENT][NUM_PRED_BUF]
static Bool UseDPCMForFirstPassIntraEstimation(TComTU &rTu, const UInt uiDirMode)