source: 3DVCSoftware/branches/HTM-3.1-MediaTek/source/Lib/TLibCommon/TComRom.h @ 145

Last change on this file since 145 was 56, checked in by hschwarz, 13 years ago

updated trunk (move to HM6.1)

  • Property svn:eol-style set to native
File size: 14.8 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     TComRom.h
35    \brief    global variables & functions (header)
36*/
37
38#ifndef __TCOMROM__
39#define __TCOMROM__
40
41#include "CommonDef.h"
42
43#include<stdio.h>
44#include<iostream>
45
46#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
47#include "TComWedgelet.h"
48#endif
49
50//! \ingroup TLibCommon
51//! \{
52
53// ====================================================================================================================
54// Macros
55// ====================================================================================================================
56
57#define     MAX_CU_DEPTH            7                           // log2(LCUSize)
58#define     MAX_CU_SIZE             (1<<(MAX_CU_DEPTH))         // maximum allowable size of CU
59#define     MIN_PU_SIZE             4
60#define     MAX_NUM_SPU_W           (MAX_CU_SIZE/MIN_PU_SIZE)   // maximum number of SPU in horizontal line
61
62// ====================================================================================================================
63// Initialize / destroy functions
64// ====================================================================================================================
65
66Void         initROM();
67Void         destroyROM();
68Void         initFrameScanXY( UInt* pBuff, UInt* pBuffX, UInt* pBuffY, Int iWidth, Int iHeight );
69Void         initSigLastScan(UInt* pBuffZ, UInt* pBuffH, UInt* pBuffV, UInt* pBuffD, Int iWidth, Int iHeight, Int iDepth);
70Void         initNonSquareSigLastScan(UInt* pBuffZ, UInt uiWidth, UInt uiHeight);
71// ====================================================================================================================
72// Data structure related table & variable
73// ====================================================================================================================
74
75// flexible conversion from relative to absolute index
76extern       UInt   g_auiZscanToRaster[ MAX_NUM_SPU_W*MAX_NUM_SPU_W ];
77extern       UInt   g_auiRasterToZscan[ MAX_NUM_SPU_W*MAX_NUM_SPU_W ];
78extern       UInt   g_motionRefer[ MAX_NUM_SPU_W*MAX_NUM_SPU_W ];
79
80Void         initZscanToRaster ( Int iMaxDepth, Int iDepth, UInt uiStartVal, UInt*& rpuiCurrIdx );
81Void         initRasterToZscan ( UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uiMaxDepth         );
82
83Void          initMotionReferIdx ( UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uiMaxDepth );
84
85// conversion of partition index to picture pel position
86extern       UInt   g_auiRasterToPelX[ MAX_NUM_SPU_W*MAX_NUM_SPU_W ];
87extern       UInt   g_auiRasterToPelY[ MAX_NUM_SPU_W*MAX_NUM_SPU_W ];
88
89Void         initRasterToPelXY ( UInt uiMaxCUWidth, UInt uiMaxCUHeight, UInt uiMaxDepth );
90
91// global variable (LCU width/height, max. CU depth)
92extern       UInt g_uiMaxCUWidth;
93extern       UInt g_uiMaxCUHeight;
94extern       UInt g_uiMaxCUDepth;
95extern       UInt g_uiAddCUDepth;
96
97extern       UInt g_auiPUOffset[8];
98
99#define QUANT_IQUANT_SHIFT    20 // Q(QP%6) * IQ(QP%6) = 2^20
100#define QUANT_SHIFT           14 // Q(4) = 2^14
101#define SCALE_BITS            15 // Inherited from TMuC, pressumably for fractional bit estimates in RDOQ
102#define MAX_TR_DYNAMIC_RANGE  15 // Maximum transform dynamic range (excluding sign bit)
103
104#define SHIFT_INV_1ST          7 // Shift after first inverse transform stage
105#define SHIFT_INV_2ND         12 // Shift after second inverse transform stage
106
107extern Int g_quantScales[6];             // Q(QP%6) 
108extern Int g_invQuantScales[6];          // IQ(QP%6)
109extern const short g_aiT4[4][4];
110extern const short g_aiT8[8][8];
111extern const short g_aiT16[16][16];
112extern const short g_aiT32[32][32];
113
114// ====================================================================================================================
115// Luma QP to Chroma QP mapping
116// ====================================================================================================================
117
118extern const UChar  g_aucChromaScale      [52];
119
120// ====================================================================================================================
121// Scanning order & context mapping table
122// ====================================================================================================================
123
124extern       UInt*  g_auiFrameScanXY[ MAX_CU_DEPTH  ];    // raster index     from scanning index
125extern       UInt*  g_auiFrameScanX [ MAX_CU_DEPTH  ];    // raster index (x) from scanning index
126extern       UInt*  g_auiFrameScanY [ MAX_CU_DEPTH  ];    // raster index (y) from scanning index
127extern       UInt*  g_auiSigLastScan[4][ MAX_CU_DEPTH ];  // raster index from scanning index (zigzag, hor, ver, diag)
128extern UInt *g_sigScanNSQT[ 4 ]; // scan for non-square partitions
129extern UInt g_sigCGScanNSQT[ 4 ][ 16 ]; // coarse-grain scan for non-square partitions
130
131extern       UInt*  g_auiNonSquareSigLastScan[ 4 ];      // raster index from scanning index (zigzag)
132
133extern const UInt   g_uiGroupIdx[ 32 ];
134extern const UInt   g_uiMinInGroup[ 10 ];
135extern const UInt   g_uiLastCtx[ 28 ];
136
137#if EIGHT_BITS_RICE_CODE
138extern const UInt   g_auiGoRiceRange[5];                  //!< maximum value coded with Rice codes
139extern const UInt   g_auiGoRicePrefixLen[5];              //!< prefix length for each maximum value
140extern const UInt   g_aauiGoRiceUpdate[5][24];            //!< parameter update rules for Rice codes
141#else
142extern const UInt   g_auiGoRiceRange[4];                  //!< maximum value coded with Rice codes
143extern const UInt   g_auiGoRicePrefixLen[4];              //!< prefix length for each maximum value
144extern const UInt   g_aauiGoRiceUpdate[4][16];            //!< parameter update rules for Rice codes
145#endif
146 
147#if MULTILEVEL_SIGMAP_EXT
148extern const UInt   g_sigLastScan8x8[ 4 ][ 4 ];           //!< coefficient group scan order for 8x8 TUs
149extern       UInt   g_sigLastScanCG32x32[ 64 ];
150#endif
151
152// ====================================================================================================================
153// ADI table
154// ====================================================================================================================
155
156extern const UChar  g_aucIntraModeNumFast[7];
157
158// ====================================================================================================================
159// Angular Intra table
160// ====================================================================================================================
161
162extern const UChar g_aucIntraModeNumAng[7];
163extern const UChar g_aucIntraModeBitsAng[7];
164extern const UChar g_aucAngIntraModeOrder[NUM_INTRA_MODE];
165
166// ====================================================================================================================
167// Bit-depth
168// ====================================================================================================================
169
170extern       UInt g_uiBitDepth;
171extern       UInt g_uiBitIncrement;
172extern       UInt g_uiIBDI_MAX;
173extern       UInt g_uiBASE_MAX;
174extern       UInt g_uiPCMBitDepthLuma;
175extern       UInt g_uiPCMBitDepthChroma;
176
177// ====================================================================================================================
178// Texture type to integer mapping
179// ====================================================================================================================
180
181extern const UChar g_aucConvertTxtTypeToIdx[4];
182
183// ====================================================================================================================
184// Depth model modes
185// ====================================================================================================================
186#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
187extern       Int             g_iDeltaDCsQuantOffset;
188extern const WedgeResolution g_aeWedgeResolutionList[5];
189extern const UChar           g_aucWedgeFullBitsListIdx[7];
190extern const UChar           g_aucIntraSizeIdxToWedgeSize[7];
191
192extern       std::vector< std::vector<TComWedgelet> >  g_aacWedgeLists;
193extern       std::vector< std::vector<TComWedgeRef> >  g_aacWedgeRefLists;
194
195Void initWedgeLists();
196Void createWedgeList( UInt uiWidth, UInt uiHeight, std::vector<TComWedgelet> &racWedgeList, std::vector<TComWedgeRef> &racWedgeRefList, WedgeResolution eWedgeRes );
197Void addWedgeletToList( TComWedgelet cWedgelet, std::vector<TComWedgelet> &racWedgeList, std::vector<TComWedgeRef> &racWedgeRefList );
198
199#if DEFAULT_DC
200__inline Void mapDMMtoIntraMode(  UInt& dirMode ) { dirMode = (dirMode >= NUM_INTRA_MODE) ? DC_IDX : dirMode; }
201__inline Void mapDMMtoIntraMode(   Int& dirMode ) { dirMode = (dirMode >= NUM_INTRA_MODE) ? DC_IDX : dirMode; }
202#else
203__inline Void mapDMMtoIntraMode(  UInt& dirMode ) { dirMode = (dirMode >= NUM_INTRA_MODE) ? PLANAR_IDX : dirMode; }
204__inline Void mapDMMtoIntraMode(   Int& dirMode ) { dirMode = (dirMode >= NUM_INTRA_MODE) ? PLANAR_IDX : dirMode; }
205#endif
206#endif
207
208// ==========================================
209// Mode-Dependent DST Matrices
210extern const short g_as_DST_MAT_4 [4][4];
211#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
212extern const UChar g_aucDCTDSTMode_Vert[NUM_INTRA_MODE+NUM_DMM_MODE];
213extern const UChar g_aucDCTDSTMode_Hor [NUM_INTRA_MODE+NUM_DMM_MODE];
214#else
215extern const UChar g_aucDCTDSTMode_Vert[NUM_INTRA_MODE];
216extern const UChar g_aucDCTDSTMode_Hor[NUM_INTRA_MODE];
217#endif
218// ==========================================
219
220// ====================================================================================================================
221// Misc.
222// ====================================================================================================================
223
224extern       Char   g_aucConvertToBit  [ MAX_CU_SIZE+1 ];   // from width to log2(width)-2
225
226#define ENC_DEC_TRACE 0
227
228
229#if ENC_DEC_TRACE
230extern FILE*  g_hTrace;
231extern Bool   g_bJustDoIt;
232extern const Bool g_bEncDecTraceEnable;
233extern const Bool g_bEncDecTraceDisable;
234extern UInt64 g_nSymbolCounter;
235
236#define COUNTER_START    1
237#define COUNTER_END      0 //( UInt64(1) << 63 )
238
239#define DTRACE_CABAC_F(x)     if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, "%f", x );
240#define DTRACE_CABAC_V(x)     if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, "%d", x );
241#define DTRACE_CABAC_VL(x)    if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, "%lld", x );
242#define DTRACE_CABAC_T(x)     if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, "%s", x );
243#define DTRACE_CABAC_X(x)     if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, "%x", x );
244#define DTRACE_CABAC_R( x,y ) if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, x,    y );
245#define DTRACE_CABAC_N        if ( ( g_nSymbolCounter >= COUNTER_START && g_nSymbolCounter <= COUNTER_END )|| g_bJustDoIt ) fprintf( g_hTrace, "\n"    );
246
247#else
248
249#define DTRACE_CABAC_F(x)
250#define DTRACE_CABAC_V(x)
251#define DTRACE_CABAC_VL(x)
252#define DTRACE_CABAC_T(x)
253#define DTRACE_CABAC_X(x)
254#define DTRACE_CABAC_R( x,y )
255#define DTRACE_CABAC_N
256
257#endif
258
259
260#define SCALING_LIST_NUM 6         ///< list number for quantization matrix
261#define SCALING_LIST_NUM_32x32 2   ///< list number for quantization matrix 32x32
262#define SCALING_LIST_REM_NUM 6     ///< remainder of QP/6
263#define SCALING_LIST_START_VALUE 8 ///< start value for dpcm mode
264#define MAX_MATRIX_COEF_NUM 64     ///< max coefficient number for quantization matrix
265#define MAX_MATRIX_SIZE_NUM 8      ///< max size number for quantization matrix
266#if SCALING_LIST
267#define SCALING_LIST_DC 16         ///< default DC value
268#endif
269enum ScalingListDIR
270{
271  SCALING_LIST_SQT = 0,
272  SCALING_LIST_VER,
273  SCALING_LIST_HOR,
274  SCALING_LIST_DIR_NUM
275};
276enum ScalingListSize
277{
278  SCALING_LIST_4x4 = 0,
279  SCALING_LIST_8x8,
280  SCALING_LIST_16x16,
281  SCALING_LIST_32x32,
282  SCALING_LIST_SIZE_NUM
283};
284static const char MatrixType[4][6][20] =
285{
286  {
287  "INTRA4X4_LUMA",
288  "INTRA4X4_CHROMAU",
289  "INTRA4X4_CHROMAV",
290  "INTER4X4_LUMA",
291  "INTER4X4_CHROMAU",
292  "INTER4X4_CHROMAV"
293  },
294  {
295  "INTRA8X8_LUMA",
296  "INTRA8X8_CHROMAU", 
297  "INTRA8X8_CHROMAV", 
298  "INTER8X8_LUMA",
299  "INTER8X8_CHROMAU", 
300  "INTER8X8_CHROMAV" 
301  },
302  {
303  "INTRA16X16_LUMA",
304  "INTRA16X16_CHROMAU", 
305  "INTRA16X16_CHROMAV", 
306  "INTER16X16_LUMA",
307  "INTER16X16_CHROMAU", 
308  "INTER16X16_CHROMAV" 
309  },
310  {
311  "INTRA32X32_LUMA",
312  "INTER32X32_LUMA",
313  },
314};
315#if SCALING_LIST
316static const char MatrixType_DC[4][12][22] =
317{
318  {
319  },
320  {
321  },
322  {
323  "INTRA16X16_LUMA_DC",
324  "INTRA16X16_CHROMAU_DC", 
325  "INTRA16X16_CHROMAV_DC", 
326  "INTER16X16_LUMA_DC",
327  "INTER16X16_CHROMAU_DC", 
328  "INTER16X16_CHROMAV_DC" 
329  },
330  {
331  "INTRA32X32_LUMA_DC",
332  "INTER32X32_LUMA_DC",
333  },
334};
335#endif
336extern Int g_quantIntraDefault4x4[16];
337extern Int g_quantIntraDefault8x8[64];
338extern Int g_quantIntraDefault16x16[256];
339extern Int g_quantIntraDefault32x32[1024];
340extern Int g_quantInterDefault4x4[16];
341extern Int g_quantInterDefault8x8[64];
342extern Int g_quantInterDefault16x16[256];
343extern Int g_quantInterDefault32x32[1024];
344extern UInt g_scalingListSize [SCALING_LIST_SIZE_NUM];
345extern UInt g_scalingListSizeX[SCALING_LIST_SIZE_NUM];
346extern UInt g_scalingListNum  [SCALING_LIST_SIZE_NUM];
347extern Int  g_eTTable[4];
348//! \}
349
350#endif  //__TCOMROM__
351
Note: See TracBrowser for help on using the repository browser.