source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibCommon/TComPicYuv.h @ 1348

Last change on this file since 1348 was 1289, checked in by seregin, 9 years ago

port rev 4323 (g_uiMaxCUWidth, g_uiMaxCUHeight)
  • Property svn:eol-style set to native
File size: 10.3 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-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
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"
43#include "TComRom.h"
44#include "TComChromaFormat.h"
45#include "SEI.h"
46#if SVC_EXTENSION
47#include "TComSlice.h"
48#endif
49//! \ingroup TLibCommon
50//! \{
51
52// ====================================================================================================================
53// Class definition
54// ====================================================================================================================
55
56/// picture YUV buffer class
57class TComPicYuv
58{
59private:
60
61  // ------------------------------------------------------------------------------------------------
62  //  YUV buffer
63  // ------------------------------------------------------------------------------------------------
64
65  Pel*  m_apiPicBuf[MAX_NUM_COMPONENT];             ///< Buffer (including margin)
66
67  Pel*  m_piPicOrg[MAX_NUM_COMPONENT];              ///< m_apiPicBufY + m_iMarginLuma*getStride() + m_iMarginLuma
68
69  // ------------------------------------------------------------------------------------------------
70  //  Parameter for general YUV buffer usage
71  // ------------------------------------------------------------------------------------------------
72
73  Int   m_iPicWidth;                                ///< Width of picture in pixels
74  Int   m_iPicHeight;                               ///< Height of picture in pixels
75  ChromaFormat m_chromaFormatIDC;                   ///< Chroma Format
76
77  Int*  m_ctuOffsetInBuffer[MAX_NUM_CHANNEL_TYPE];  ///< Gives an offset in the buffer for a given CTU (and channel)
78  Int*  m_subCuOffsetInBuffer[MAX_NUM_CHANNEL_TYPE];///< Gives an offset in the buffer for a given sub-CU (and channel), relative to start of CTU
79
80  Int   m_iMarginX;                                 ///< margin of Luma channel (chroma's may be smaller, depending on ratio)
81  Int   m_iMarginY;                                 ///< margin of Luma channel (chroma's may be smaller, depending on ratio)
82
83  Bool  m_bIsBorderExtended;
84
85#if SVC_EXTENSION
86  Window  m_conformanceWindow;
87  Bool    m_isReconstructed;
88#endif
89 
90public:
91               TComPicYuv         ();
92  virtual     ~TComPicYuv         ();
93
94  // ------------------------------------------------------------------------------------------------
95  //  Memory management
96  // ------------------------------------------------------------------------------------------------
97
98  Void          create            (const Int iPicWidth,
99                                   const Int iPicHeight,
100                                   const ChromaFormat chromaFormatIDC,
101                                   const UInt uiMaxCUWidth,  ///< used for generating offsets to CUs. Can use iPicWidth if no offsets are required
102                                   const UInt uiMaxCUHeight, ///< used for generating offsets to CUs. Can use iPicHeight if no offsets are required
103                                   const UInt uiMaxCUDepth,  ///< used for generating offsets to CUs. Can use 0 if no offsets are required
104#if SVC_EXTENSION
105                                   const Bool bUseMargin,    ///< if true, then a margin of uiMaxCUWidth+16 and uiMaxCUHeight+16 is created around the image.
106                                   const Window* conformanceWindow = NULL);
107#else
108                                   const Bool bUseMargin);   ///< if true, then a margin of uiMaxCUWidth+16 and uiMaxCUHeight+16 is created around the image.
109#endif
110 
111  Void          destroy           ();
112
113  // The following have been removed - Use CHROMA_400 in the above function call.
114  //Void  createLuma  ( Int iPicWidth, Int iPicHeight, UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uhMaxCUDepth );
115  //Void  destroyLuma ();
116
117  // ------------------------------------------------------------------------------------------------
118  //  Get information of picture
119  // ------------------------------------------------------------------------------------------------
120
121  Int           getWidth          (const ComponentID id) const { return  m_iPicWidth >> getComponentScaleX(id);   }
122  Int           getHeight         (const ComponentID id) const { return  m_iPicHeight >> getComponentScaleY(id);  }
123  ChromaFormat  getChromaFormat   ()                     const { return m_chromaFormatIDC; }
124  UInt          getNumberValidComponents() const { return ::getNumberValidComponents(m_chromaFormatIDC); }
125
126  Int           getStride         (const ComponentID id) const { return ((m_iPicWidth     ) + (m_iMarginX  <<1)) >> getComponentScaleX(id); }
127  Int           getTotalHeight    (const ComponentID id) const { return ((m_iPicHeight    ) + (m_iMarginY  <<1)) >> getComponentScaleY(id); }
128
129  Int           getMarginX        (const ComponentID id) const { return m_iMarginX >> getComponentScaleX(id);  }
130  Int           getMarginY        (const ComponentID id) const { return m_iMarginY >> getComponentScaleY(id);  }
131
132  // ------------------------------------------------------------------------------------------------
133  //  Access function for picture buffer
134  // ------------------------------------------------------------------------------------------------
135
136  //  Access starting position of picture buffer with margin
137  Pel*          getBuf            (const ComponentID ch)       { return  m_apiPicBuf[ch];   }
138
139  //  Access starting position of original picture
140  Pel*          getAddr           (const ComponentID ch)       { return  m_piPicOrg[ch];   }
141  const Pel*    getAddr           (const ComponentID ch) const { return  m_piPicOrg[ch];   }
142
143  //  Access starting position of original picture for specific coding unit (CU) or partition unit (PU)
144  Pel*          getAddr           (const ComponentID ch, const Int ctuRSAddr )       { return m_piPicOrg[ch] + m_ctuOffsetInBuffer[ch==0?0:1][ ctuRSAddr ]; }
145  const Pel*    getAddr           (const ComponentID ch, const Int ctuRSAddr ) const { return m_piPicOrg[ch] + m_ctuOffsetInBuffer[ch==0?0:1][ ctuRSAddr ]; }
146  Pel*          getAddr           (const ComponentID ch, const Int ctuRSAddr, const Int uiAbsZorderIdx )
147                                     { return m_piPicOrg[ch] + m_ctuOffsetInBuffer[ch==0?0:1][ctuRSAddr] + m_subCuOffsetInBuffer[ch==0?0:1][g_auiZscanToRaster[uiAbsZorderIdx]]; }
148  const Pel*    getAddr           (const ComponentID ch, const Int ctuRSAddr, const Int uiAbsZorderIdx ) const
149                                     { return m_piPicOrg[ch] + m_ctuOffsetInBuffer[ch==0?0:1][ctuRSAddr] + m_subCuOffsetInBuffer[ch==0?0:1][g_auiZscanToRaster[uiAbsZorderIdx]]; }
150
151  UInt          getComponentScaleX(const ComponentID id) const { return ::getComponentScaleX(id, m_chromaFormatIDC); }
152  UInt          getComponentScaleY(const ComponentID id) const { return ::getComponentScaleY(id, m_chromaFormatIDC); }
153
154  // ------------------------------------------------------------------------------------------------
155  //  Miscellaneous
156  // ------------------------------------------------------------------------------------------------
157
158  //  Copy function to picture
159  Void          copyToPic         ( TComPicYuv*  pcPicYuvDst ) const ;
160
161  //  Extend function of picture buffer
162  Void          extendPicBorder   ();
163
164  //  Dump picture
165  Void          dump              (const Char* pFileName, const BitDepths &bitDepths, Bool bAdd = false) const ;
166
167  // Set border extension flag
168  Void          setBorderExtension(Bool b) { m_bIsBorderExtended = b; }
169 
170#if SVC_EXTENSION
171  Void          setHeight   ( Int iPicHeight )                   { m_iPicHeight = iPicHeight;               }
172  Window&       getConformanceWindow()                           { return  m_conformanceWindow;             }
173  Void          setConformanceWindow(Window& conformanceWindow ) { m_conformanceWindow = conformanceWindow; }
174  Void          setReconstructed(Bool x)                         { m_isReconstructed = x;                   }
175  Bool          isReconstructed()                                { return m_isReconstructed;                }
176#if AUXILIARY_PICTURES 
177  Void          convertToMonochrome(Int bitDepthChroma);
178#endif
179#endif
180
181  Void          dump( Char* pFileName, Bool bAdd, Int bitDepth );
182
183};// END CLASS DEFINITION TComPicYuv
184
185
186// These functions now return the length of the digest strings.
187UInt calcChecksum(const TComPicYuv& pic, TComPictureHash &digest, const BitDepths &bitDepths);
188UInt calcCRC     (const TComPicYuv& pic, TComPictureHash &digest, const BitDepths &bitDepths);
189UInt calcMD5     (const TComPicYuv& pic, TComPictureHash &digest, const BitDepths &bitDepths);
190std::string hashToString(const TComPictureHash &digest, Int numChar);
191//! \}
192
193#endif // __TCOMPICYUV__
Note: See TracBrowser for help on using the repository browser.