source: 3DVCSoftware/branches/HTM-5.1-dev1-LG-Fix/source/Lib/TLibCommon/TComPattern.h @ 368

Last change on this file since 368 was 189, checked in by tech, 12 years ago

Reintegrated branch 4.1-dev0 Rev. 188.

  • Property svn:eol-style set to native
File size: 9.0 KB
Line 
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-2012, 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     TComPattern.h
35    \brief    neighbouring pixel access classes (header)
36*/
37
38#ifndef __TCOMPATTERN__
39#define __TCOMPATTERN__
40
41// Include files
42#include <stdio.h>
43#include "CommonDef.h"
44
45//! \ingroup TLibCommon
46//! \{
47
48// ====================================================================================================================
49// Class definition
50// ====================================================================================================================
51
52class TComDataCU;
53
54/// neighbouring pixel access class for one component
55class TComPatternParam
56{
57private:
58  Int   m_iOffsetLeft;
59  Int   m_iOffsetRight;
60  Int   m_iOffsetAbove;
61  Int   m_iOffsetBottom;
62  Pel*  m_piPatternOrigin;
63 
64public:
65  Int   m_iROIWidth;
66  Int   m_iROIHeight;
67  Int   m_iPatternStride;
68 
69  /// return starting position of buffer
70  Pel*  getPatternOrigin()        { return  m_piPatternOrigin; }
71 
72  /// return starting position of ROI (ROI = &pattern[AboveOffset][LeftOffset])
73  __inline Pel*  getROIOrigin()
74  {
75    return  m_piPatternOrigin + m_iPatternStride * m_iOffsetAbove + m_iOffsetLeft;
76  }
77 
78  /// set parameters from Pel buffer for accessing neighbouring pixels
79  Void setPatternParamPel ( Pel*        piTexture,
80                           Int         iRoiWidth,
81                           Int         iRoiHeight,
82                           Int         iStride,
83                           Int         iOffsetLeft,
84                           Int         iOffsetRight,
85                           Int         iOffsetAbove,
86                           Int         iOffsetBottom );
87 
88  /// set parameters of one color component from CU data for accessing neighbouring pixels
89  Void setPatternParamCU  ( TComDataCU* pcCU,
90                           UChar       iComp,
91                           UChar       iRoiWidth,
92                           UChar       iRoiHeight,
93                           Int         iOffsetLeft,
94                           Int         iOffsetRight,
95                           Int         iOffsetAbove,
96                           Int         iOffsetBottom,
97                           UInt        uiPartDepth,
98                           UInt        uiAbsZorderIdx
99#if DEPTH_MAP_GENERATION
100                           ,Bool        bPrdDepthMap = false 
101#endif 
102                           );
103};
104
105/// neighbouring pixel access class for all components
106class TComPattern
107{
108private:
109  TComPatternParam  m_cPatternY;
110  TComPatternParam  m_cPatternCb;
111  TComPatternParam  m_cPatternCr;
112 
113#if LGE_ILLUCOMP_B0045
114  Bool              m_bICFlag;
115#endif
116
117#if LOGI_INTRA_NAME_3MPM
118  static const UChar m_aucIntraFilter[5];
119#else
120  static const UChar m_aucIntraFilter[5][NUM_INTRA_MODE];
121#endif
122 
123public:
124 
125  // ROI & pattern information, (ROI = &pattern[AboveOffset][LeftOffset])
126  Pel*  getROIY()                 { return m_cPatternY.getROIOrigin();    }
127  Int   getROIYWidth()            { return m_cPatternY.m_iROIWidth;       }
128  Int   getROIYHeight()           { return m_cPatternY.m_iROIHeight;      }
129  Int   getPatternLStride()       { return m_cPatternY.m_iPatternStride;  }
130
131#if LGE_ILLUCOMP_B0045
132  Bool  getICFlag()               { return m_bICFlag; }
133  Void  setICFlag(Bool bICFlag)   { m_bICFlag = bICFlag; }
134#endif
135
136  // access functions of ADI buffers
137  Int*  getAdiOrgBuf              ( Int iCuWidth, Int iCuHeight, Int* piAdiBuf );
138  Int*  getAdiCbBuf               ( Int iCuWidth, Int iCuHeight, Int* piAdiBuf );
139  Int*  getAdiCrBuf               ( Int iCuWidth, Int iCuHeight, Int* piAdiBuf );
140 
141  Int*  getPredictorPtr           ( UInt uiDirMode, UInt uiWidthBits, Int* piAdiBuf );
142  // -------------------------------------------------------------------------------------------------------------------
143  // initialization functions
144  // -------------------------------------------------------------------------------------------------------------------
145 
146  /// set parameters from Pel buffers for accessing neighbouring pixels
147  Void initPattern            ( Pel*        piY,
148                               Pel*        piCb,
149                               Pel*        piCr,
150                               Int         iRoiWidth,
151                               Int         iRoiHeight,
152                               Int         iStride,
153                               Int         iOffsetLeft,
154                               Int         iOffsetRight,
155                               Int         iOffsetAbove,
156                               Int         iOffsetBottom );
157 
158  /// set parameters from CU data for accessing neighbouring pixels
159  Void  initPattern           ( TComDataCU* pcCU,
160                               UInt        uiPartDepth,
161                               UInt        uiAbsPartIdx
162#if DEPTH_MAP_GENERATION
163                               ,Bool        bPrdDepthMap = false 
164#endif
165                               );
166 
167  /// set luma parameters from CU data for accessing ADI data
168  Void  initAdiPattern        ( TComDataCU* pcCU,
169                               UInt        uiZorderIdxInPart,
170                               UInt        uiPartDepth,
171                               Int*        piAdiBuf,
172                               Int         iOrgBufStride,
173                               Int         iOrgBufHeight,
174                               Bool&       bAbove,
175                               Bool&       bLeft
176                              ,Bool        bLMmode = false // using for LM chroma or not
177#if DEPTH_MAP_GENERATION
178                              ,
179                              Bool         bPrdDepthMap = false,
180                              UInt         uiSubSampExpX = 0,
181                              UInt         uiSubSampExpY = 0
182#endif
183                               );
184 
185  /// set chroma parameters from CU data for accessing ADI data
186  Void  initAdiPatternChroma  ( TComDataCU* pcCU,
187                               UInt        uiZorderIdxInPart,
188                               UInt        uiPartDepth,
189                               Int*        piAdiBuf,
190                               Int         iOrgBufStride,
191                               Int         iOrgBufHeight,
192                               Bool&       bAbove,
193                               Bool&       bLeft );
194
195private:
196
197  /// padding of unavailable reference samples for intra prediction
198  Void  fillReferenceSamples        ( TComDataCU* pcCU, Pel* piRoiOrigin, Int* piAdiTemp, Bool* bNeighborFlags, Int iNumIntraNeighbor, Int iUnitSize, Int iNumUnitsInCu, Int iTotalUnits, UInt uiCuWidth, UInt uiCuHeight, UInt uiWidth, UInt uiHeight, Int iPicStride, Bool bLMmode = false
199#if DEPTH_MAP_GENERATION
200                                    , Bool bPrdDepthMap = false 
201#endif
202                                    );
203 
204
205  /// constrained intra prediction
206  Bool  isAboveLeftAvailable  ( TComDataCU* pcCU, UInt uiPartIdxLT );
207  Int   isAboveAvailable      ( TComDataCU* pcCU, UInt uiPartIdxLT, UInt uiPartIdxRT, Bool* bValidFlags );
208  Int   isLeftAvailable       ( TComDataCU* pcCU, UInt uiPartIdxLT, UInt uiPartIdxLB, Bool* bValidFlags );
209  Int   isAboveRightAvailable ( TComDataCU* pcCU, UInt uiPartIdxLT, UInt uiPartIdxRT, Bool* bValidFlags );
210  Int   isBelowLeftAvailable  ( TComDataCU* pcCU, UInt uiPartIdxLT, UInt uiPartIdxLB, Bool* bValidFlags );
211
212};
213
214//! \}
215
216#endif // __TCOMPATTERN__
Note: See TracBrowser for help on using the repository browser.