source: 3DVCSoftware/trunk/source/Lib/TLibCommon/TComPicYuv.h @ 737

Last change on this file since 737 was 608, checked in by tech, 11 years ago

Merged DEV-2.0-dev0@604.

  • Property svn:eol-style set to native
File size: 7.9 KB
RevLine 
[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 *
[608]6 * Copyright (c) 2010-2013, 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     TComPicYuv.h
35    \brief    picture YUV buffer class (header)
36*/
37
38#ifndef __TCOMPICYUV__
39#define __TCOMPICYUV__
40
41#include <stdio.h>
42#include "CommonDef.h"
[56]43#include "TComRom.h"
[2]44
[56]45//! \ingroup TLibCommon
46//! \{
47
[2]48// ====================================================================================================================
49// Class definition
50// ====================================================================================================================
51
52/// picture YUV buffer class
53class TComPicYuv
54{
55private:
56 
57  // ------------------------------------------------------------------------------------------------
58  //  YUV buffer
59  // ------------------------------------------------------------------------------------------------
60 
61  Pel*  m_apiPicBufY;           ///< Buffer (including margin)
62  Pel*  m_apiPicBufU;
63  Pel*  m_apiPicBufV;
64 
65  Pel*  m_piPicOrgY;            ///< m_apiPicBufY + m_iMarginLuma*getStride() + m_iMarginLuma
66  Pel*  m_piPicOrgU;
67  Pel*  m_piPicOrgV;
68 
69  // ------------------------------------------------------------------------------------------------
70  //  Parameter for general YUV buffer usage
71  // ------------------------------------------------------------------------------------------------
72 
73  Int   m_iPicWidth;            ///< Width of picture
74  Int   m_iPicHeight;           ///< Height of picture
75 
76  Int   m_iCuWidth;             ///< Width of Coding Unit (CU)
77  Int   m_iCuHeight;            ///< Height of Coding Unit (CU)
[56]78  Int*  m_cuOffsetY;
79  Int*  m_cuOffsetC;
80  Int*  m_buOffsetY;
81  Int*  m_buOffsetC;
82 
[2]83  Int   m_iLumaMarginX;
84  Int   m_iLumaMarginY;
85  Int   m_iChromaMarginX;
86  Int   m_iChromaMarginY;
87 
88  Bool  m_bIsBorderExtended;
89 
[608]90#if H_3D_IV_MERGE
91  Int   m_iBaseUnitWidth;       ///< Width of Base Unit (with maximum depth or minimum size, m_iCuWidth >> Max. Depth)
92  Int   m_iBaseUnitHeight;      ///< Height of Base Unit (with maximum depth or minimum size, m_iCuHeight >> Max. Depth)
93  Int   m_iNumCuInWidth;
94#endif
[2]95protected:
96  Void  xExtendPicCompBorder (Pel* piTxt, Int iStride, Int iWidth, Int iHeight, Int iMarginX, Int iMarginY);
[608]97#if H_3D
[2]98  Void  xSetPels( Pel* piPelSource , Int iSourceStride, Int iWidth, Int iHeight, Pel iVal );
[608]99#endif
[2]100 
101public:
102  TComPicYuv         ();
103  virtual ~TComPicYuv();
104 
105  // ------------------------------------------------------------------------------------------------
106  //  Memory management
107  // ------------------------------------------------------------------------------------------------
108 
109  Void  create      ( Int iPicWidth, Int iPicHeight, UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uiMaxCUDepth );
110  Void  destroy     ();
111 
112  Void  createLuma  ( Int iPicWidth, Int iPicHeight, UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uhMaxCUDepth );
113  Void  destroyLuma ();
114 
115  // ------------------------------------------------------------------------------------------------
116  //  Get information of picture
117  // ------------------------------------------------------------------------------------------------
118 
119  Int   getWidth    ()     { return  m_iPicWidth;    }
120  Int   getHeight   ()     { return  m_iPicHeight;   }
[56]121 
[2]122  Int   getStride   ()     { return (m_iPicWidth     ) + (m_iLumaMarginX  <<1); }
123  Int   getCStride  ()     { return (m_iPicWidth >> 1) + (m_iChromaMarginX<<1); }
124 
125  Int   getLumaMargin   () { return m_iLumaMarginX;  }
126  Int   getChromaMargin () { return m_iChromaMarginX;}
127 
128  // ------------------------------------------------------------------------------------------------
129  //  Access function for picture buffer
130  // ------------------------------------------------------------------------------------------------
131 
132  //  Access starting position of picture buffer with margin
133  Pel*  getBufY     ()     { return  m_apiPicBufY;   }
134  Pel*  getBufU     ()     { return  m_apiPicBufU;   }
135  Pel*  getBufV     ()     { return  m_apiPicBufV;   }
136 
137  //  Access starting position of original picture
138  Pel*  getLumaAddr ()     { return  m_piPicOrgY;    }
139  Pel*  getCbAddr   ()     { return  m_piPicOrgU;    }
140  Pel*  getCrAddr   ()     { return  m_piPicOrgV;    }
141 
142  //  Access starting position of original picture for specific coding unit (CU) or partition unit (PU)
[56]143  Pel*  getLumaAddr ( Int iCuAddr ) { return m_piPicOrgY + m_cuOffsetY[ iCuAddr ]; }
144  Pel*  getCbAddr   ( Int iCuAddr ) { return m_piPicOrgU + m_cuOffsetC[ iCuAddr ]; }
145  Pel*  getCrAddr   ( Int iCuAddr ) { return m_piPicOrgV + m_cuOffsetC[ iCuAddr ]; }
146  Pel*  getLumaAddr ( Int iCuAddr, Int uiAbsZorderIdx ) { return m_piPicOrgY + m_cuOffsetY[iCuAddr] + m_buOffsetY[g_auiZscanToRaster[uiAbsZorderIdx]]; }
147  Pel*  getCbAddr   ( Int iCuAddr, Int uiAbsZorderIdx ) { return m_piPicOrgU + m_cuOffsetC[iCuAddr] + m_buOffsetC[g_auiZscanToRaster[uiAbsZorderIdx]]; }
148  Pel*  getCrAddr   ( Int iCuAddr, Int uiAbsZorderIdx ) { return m_piPicOrgV + m_cuOffsetC[iCuAddr] + m_buOffsetC[g_auiZscanToRaster[uiAbsZorderIdx]]; }
[2]149 
150  // ------------------------------------------------------------------------------------------------
151  //  Miscellaneous
152  // ------------------------------------------------------------------------------------------------
[56]153 
[2]154  //  Copy function to picture
155  Void  copyToPic       ( TComPicYuv*  pcPicYuvDst );
156  Void  copyToPicLuma   ( TComPicYuv*  pcPicYuvDst );
157  Void  copyToPicCb     ( TComPicYuv*  pcPicYuvDst );
158  Void  copyToPicCr     ( TComPicYuv*  pcPicYuvDst );
159 
160  //  Extend function of picture buffer
161  Void  extendPicBorder      ();
162 
163  //  Dump picture
[608]164  Void  dump (Char* pFileName, Bool bAdd = false);
[2]165 
166  // Set border extension flag
167  Void  setBorderExtension(Bool b) { m_bIsBorderExtended = b; }
[608]168#if H_3D
[2]169  // Set Function
170  Void  setLumaTo    ( Pel pVal ); 
171  Void  setChromaTo  ( Pel pVal ); 
[608]172#if H_3D_IV_MERGE
173  // sample to block and block to sample conversion
174  Void  getTopLeftSamplePos( Int iCuAddr, Int iAbsZorderIdx, Int& riX, Int& riY );
175  Void  getCUAddrAndPartIdx( Int iX, Int iY, Int& riCuAddr, Int& riAbsZorderIdx );
[443]176#endif
[608]177#endif
[2]178};// END CLASS DEFINITION TComPicYuv
179
[608]180void calcChecksum(TComPicYuv& pic, UChar digest[3][16]);
181void calcCRC(TComPicYuv& pic, UChar digest[3][16]);
182void calcMD5(TComPicYuv& pic, UChar digest[3][16]);
[56]183//! \}
184
[2]185#endif // __TCOMPICYUV__
Note: See TracBrowser for help on using the repository browser.