0
|
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-2014, 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 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"
|
|
47 #include "TComInterpolationFilter.h"
|
|
48 #include "TComWeightPrediction.h"
|
|
49
|
|
50 class TComTU; // forward declaration
|
|
51
|
|
52 //! \ingroup TLibCommon
|
|
53 //! \{
|
|
54
|
|
55 // ====================================================================================================================
|
|
56 // Class definition
|
|
57 // ====================================================================================================================
|
|
58
|
|
59 /// prediction class
|
|
60 typedef enum PRED_BUF_E
|
|
61 {
|
|
62 PRED_BUF_UNFILTERED=0,
|
|
63 PRED_BUF_FILTERED=1,
|
|
64 NUM_PRED_BUF=2
|
|
65 } PRED_BUF;
|
|
66
|
|
67 static const UInt MAX_INTRA_FILTER_DEPTHS=5;
|
|
68
|
|
69 class TComPrediction : public TComWeightPrediction
|
|
70 {
|
|
71 private:
|
|
72 static const UChar m_aucIntraFilter[MAX_NUM_CHANNEL_TYPE][MAX_INTRA_FILTER_DEPTHS];
|
|
73
|
|
74 protected:
|
|
75 Pel* m_piYuvExt[MAX_NUM_COMPONENT][NUM_PRED_BUF];
|
|
76 Int m_iYuvExtSize;
|
|
77
|
|
78 TComYuv m_acYuvPred[NUM_REF_PIC_LIST_01];
|
|
79 TComYuv m_cYuvPredTemp;
|
|
80 TComYuv m_filteredBlock[LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS][LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS];
|
|
81 TComYuv m_filteredBlockTmp[LUMA_INTERPOLATION_FILTER_SUB_SAMPLE_POSITIONS];
|
|
82
|
|
83 TComInterpolationFilter m_if;
|
|
84
|
|
85 Pel* m_pLumaRecBuffer; ///< array for downsampled reconstructed luma sample
|
|
86 Int m_iLumaRecStride; ///< stride of #m_pLumaRecBuffer array
|
|
87
|
|
88 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 );
|
|
89 Void xPredIntraPlanar ( const Pel* pSrc, Int srcStride, Pel* rpDst, Int dstStride, UInt width, UInt height, ChannelType channelType, ChromaFormat format );
|
|
90
|
|
91 // motion compensation functions
|
|
92 Void xPredInterUni ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, RefPicList eRefPicList, TComYuv* pcYuvPred, Bool bi=false );
|
|
93 Void xPredInterBi ( TComDataCU* pcCU, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv* pcYuvPred );
|
|
94 Void xPredInterBlk(const ComponentID compID, TComDataCU *cu, TComPicYuv *refPic, UInt partAddr, TComMv *mv, Int width, Int height, TComYuv *dstPic, Bool bi );
|
|
95 Void xWeightedAverage ( TComYuv* pcYuvSrc0, TComYuv* pcYuvSrc1, Int iRefIdx0, Int iRefIdx1, UInt uiPartAddr, Int iWidth, Int iHeight, TComYuv* pcYuvDst );
|
|
96
|
|
97 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) );
|
|
98
|
|
99 Void xDCPredFiltering( const Pel* pSrc, Int iSrcStride, Pel* pDst, Int iDstStride, Int iWidth, Int iHeight, ChannelType channelType );
|
|
100 Bool xCheckIdenticalMotion ( TComDataCU* pcCU, UInt PartAddr);
|
|
101 Void destroy();
|
|
102
|
|
103 public:
|
|
104 TComPrediction();
|
|
105 virtual ~TComPrediction();
|
|
106
|
|
107 Void initTempBuff(ChromaFormat chromaFormatIDC);
|
|
108
|
|
109 ChromaFormat getChromaFormat() const { return m_cYuvPredTemp.getChromaFormat(); }
|
|
110
|
|
111 // inter
|
|
112 Void motionCompensation ( TComDataCU* pcCU, TComYuv* pcYuvPred, RefPicList eRefPicList = REF_PIC_LIST_X, Int iPartIdx = -1 );
|
|
113
|
|
114 // motion vector prediction
|
|
115 Void getMvPredAMVP ( TComDataCU* pcCU, UInt uiPartIdx, UInt uiPartAddr, RefPicList eRefPicList, TComMv& rcMvPred );
|
|
116
|
|
117 // Angular Intra
|
|
118 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 );
|
|
119
|
|
120 Pel predIntraGetPredValDC ( const Pel* pSrc, Int iSrcStride, UInt iWidth, UInt iHeight, ChannelType channelType, ChromaFormat format, Bool bAbove, Bool bLeft );
|
|
121
|
|
122 Pel* getPredictorPtr ( const ComponentID compID, const Bool bUseFilteredPredictions )
|
|
123 {
|
|
124 return m_piYuvExt[compID][bUseFilteredPredictions?PRED_BUF_FILTERED:PRED_BUF_UNFILTERED];
|
|
125 }
|
|
126
|
|
127 // This function is actually still in TComPattern.cpp
|
|
128 /// set parameters from CU data for accessing ADI data
|
|
129 Void initAdiPatternChType ( TComTU &rTu,
|
|
130 Bool& bAbove,
|
|
131 Bool& bLeft,
|
|
132 const ComponentID compID, const Bool bFilterRefSamples
|
|
133 DEBUG_STRING_FN_DECLARE(sDebug)
|
|
134 );
|
|
135
|
|
136 static Bool filteringIntraReferenceSamples(const ComponentID compID, UInt uiDirMode, UInt uiTuChWidth, UInt uiTuChHeight, const ChromaFormat chFmt, const Bool intraReferenceSmoothingDisabled);
|
|
137
|
|
138 static Bool UseDPCMForFirstPassIntraEstimation(TComTU &rTu, const UInt uiDirMode);
|
|
139 };
|
|
140
|
|
141 //! \}
|
|
142
|
|
143 #endif // __TCOMPREDICTION__
|