Mercurial > hg > forks > libbpg

comparison jctvc/TLibCommon/TComPrediction.h @ 0:772086c29cc7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Initial import.
author Matti Hamalainen <ccr@tnsp.org>
date Wed, 16 Nov 2016 11:16:33 +0200
parents
children
comparison
equal deleted inserted replaced
-1:000000000000 0:772086c29cc7
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__