source: 3DVCSoftware/trunk/source/Lib/TLibDecoder/TDecSbac.cpp @ 1095

Last change on this file since 1095 was 1084, checked in by tech, 10 years ago

Merged branches/HTM-12.1-dev0@1083.

  • Property svn:eol-style set to native
File size: 66.4 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 *
[872]6* Copyright (c) 2010-2014, 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     TDecSbac.cpp
35    \brief    Context-adaptive entropy decoder class
36*/
37
38#include "TDecSbac.h"
39
[56]40//! \ingroup TLibDecoder
41//! \{
42
[2]43//////////////////////////////////////////////////////////////////////
44// Construction/Destruction
45//////////////////////////////////////////////////////////////////////
46
47TDecSbac::TDecSbac() 
48// new structure here
49: m_pcBitstream               ( 0 )
50, m_pcTDecBinIf               ( NULL )
[56]51, m_numContextModels          ( 0 )
52, m_cCUSplitFlagSCModel       ( 1,             1,               NUM_SPLIT_FLAG_CTX            , m_contextModels + m_numContextModels, m_numContextModels )
53, m_cCUSkipFlagSCModel        ( 1,             1,               NUM_SKIP_FLAG_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
[1084]54#if H_3D_SINGLE_DEPTH
[1039]55, m_cCUSingleDepthFlagSCModel        ( 1,             1,               NUM_SINGLEDEPTH_FLAG_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
56, m_cSingleDepthValueSCModel         ( 1,             1,               NUM_SINGLE_DEPTH_VALUE_DATA_CTX      , m_contextModels + m_numContextModels, m_numContextModels)
57#endif
[56]58, m_cCUMergeFlagExtSCModel    ( 1,             1,               NUM_MERGE_FLAG_EXT_CTX        , m_contextModels + m_numContextModels, m_numContextModels)
59, m_cCUMergeIdxExtSCModel     ( 1,             1,               NUM_MERGE_IDX_EXT_CTX         , m_contextModels + m_numContextModels, m_numContextModels)
[608]60#if H_3D_ARP
61, m_cCUPUARPWSCModel          ( 1,             1,               NUM_ARPW_CTX                  , m_contextModels + m_numContextModels, m_numContextModels)
[2]62#endif
[608]63#if H_3D_IC
64, m_cCUICFlagSCModel          ( 1,             1,               NUM_IC_FLAG_CTX               , m_contextModels + m_numContextModels, m_numContextModels)
[443]65#endif
[56]66, m_cCUPartSizeSCModel        ( 1,             1,               NUM_PART_SIZE_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
67, m_cCUPredModeSCModel        ( 1,             1,               NUM_PRED_MODE_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
68, m_cCUIntraPredSCModel       ( 1,             1,               NUM_ADI_CTX                   , m_contextModels + m_numContextModels, m_numContextModels)
69, m_cCUChromaPredSCModel      ( 1,             1,               NUM_CHROMA_PRED_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
70, m_cCUDeltaQpSCModel         ( 1,             1,               NUM_DELTA_QP_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
71, m_cCUInterDirSCModel        ( 1,             1,               NUM_INTER_DIR_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
72, m_cCURefPicSCModel          ( 1,             1,               NUM_REF_NO_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
73, m_cCUMvdSCModel             ( 1,             1,               NUM_MV_RES_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
74, m_cCUQtCbfSCModel           ( 1,             2,               NUM_QT_CBF_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
75, m_cCUTransSubdivFlagSCModel ( 1,             1,               NUM_TRANS_SUBDIV_FLAG_CTX     , m_contextModels + m_numContextModels, m_numContextModels)
76, m_cCUQtRootCbfSCModel       ( 1,             1,               NUM_QT_ROOT_CBF_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
77, m_cCUSigCoeffGroupSCModel   ( 1,             2,               NUM_SIG_CG_FLAG_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
78, m_cCUSigSCModel             ( 1,             1,               NUM_SIG_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
79, m_cCuCtxLastX               ( 1,             2,               NUM_CTX_LAST_FLAG_XY          , m_contextModels + m_numContextModels, m_numContextModels)
80, m_cCuCtxLastY               ( 1,             2,               NUM_CTX_LAST_FLAG_XY          , m_contextModels + m_numContextModels, m_numContextModels)
81, m_cCUOneSCModel             ( 1,             1,               NUM_ONE_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
82, m_cCUAbsSCModel             ( 1,             1,               NUM_ABS_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
83, m_cMVPIdxSCModel            ( 1,             1,               NUM_MVP_IDX_CTX               , m_contextModels + m_numContextModels, m_numContextModels)
[608]84, m_cSaoMergeSCModel      ( 1,             1,               NUM_SAO_MERGE_FLAG_CTX   , m_contextModels + m_numContextModels, m_numContextModels)
[443]85, m_cSaoTypeIdxSCModel        ( 1,             1,               NUM_SAO_TYPE_IDX_CTX          , m_contextModels + m_numContextModels, m_numContextModels)
[608]86, m_cTransformSkipSCModel     ( 1,             2,               NUM_TRANSFORMSKIP_FLAG_CTX    , m_contextModels + m_numContextModels, m_numContextModels)
87, m_CUTransquantBypassFlagSCModel( 1,          1,               NUM_CU_TRANSQUANT_BYPASS_FLAG_CTX, m_contextModels + m_numContextModels, m_numContextModels)
88#if H_3D_DIM
89, m_cDepthIntraModeSCModel    ( 1,             1,               NUM_DEPTH_INTRA_MODE_CTX      , m_contextModels + m_numContextModels, m_numContextModels)
90, m_cDdcFlagSCModel           ( 1,             1,               NUM_DDC_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
91, m_cDdcDataSCModel           ( 1,             1,               NUM_DDC_DATA_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
[833]92, m_cAngleFlagSCModel         ( 1,             1,               NUM_ANGLE_FLAG_CTX            , m_contextModels + m_numContextModels, m_numContextModels)
[608]93#if H_3D_DIM_SDC
94, m_cSDCResidualFlagSCModel     ( 1,             1,             SDC_NUM_RESIDUAL_FLAG_CTX     , m_contextModels + m_numContextModels, m_numContextModels)
95, m_cSDCResidualSCModel         ( 1,             1,             SDC_NUM_RESIDUAL_CTX          , m_contextModels + m_numContextModels, m_numContextModels)
[100]96#endif
97#endif
[884]98#if H_3D_DIM_SDC
[833]99, m_cSDCFlagSCModel                  ( 1,             1,  NUM_SDC_FLAG_CTX                 , m_contextModels + m_numContextModels, m_numContextModels)
100#endif
101#if H_3D_DBBP
102, m_cDBBPFlagSCModel             ( 1,             1,                 DBBP_NUM_FLAG_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
103#endif
[2]104{
[56]105  assert( m_numContextModels <= MAX_NUM_CTX_MOD );
[2]106}
107
108TDecSbac::~TDecSbac()
109{
110}
111
112// ====================================================================================================================
113// Public member functions
114// ====================================================================================================================
115
[56]116Void TDecSbac::resetEntropy(TComSlice* pSlice)
[2]117{
[56]118  SliceType sliceType  = pSlice->getSliceType();
119  Int       qp         = pSlice->getSliceQp();
120
121  if (pSlice->getPPS()->getCabacInitPresentFlag() && pSlice->getCabacInitFlag())
122  {
123    switch (sliceType)
124    {
125    case P_SLICE:           // change initialization table to B_SLICE initialization
126      sliceType = B_SLICE; 
127      break;
128    case B_SLICE:           // change initialization table to P_SLICE initialization
129      sliceType = P_SLICE; 
130      break;
131    default     :           // should not occur
132      assert(0);
133    }
134  }
135
136  m_cCUSplitFlagSCModel.initBuffer       ( sliceType, qp, (UChar*)INIT_SPLIT_FLAG );
137  m_cCUSkipFlagSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_SKIP_FLAG );
[1084]138#if H_3D_SINGLE_DEPTH
[1039]139  m_cCUSingleDepthFlagSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_SINGLEDEPTH_FLAG );
140  m_cSingleDepthValueSCModel.initBuffer         ( sliceType, qp, (UChar*)INIT_SINGLE_DEPTH_VALUE_DATA );
141#endif
[56]142  m_cCUMergeFlagExtSCModel.initBuffer    ( sliceType, qp, (UChar*)INIT_MERGE_FLAG_EXT );
143  m_cCUMergeIdxExtSCModel.initBuffer     ( sliceType, qp, (UChar*)INIT_MERGE_IDX_EXT );
[608]144#if H_3D_ARP
145  m_cCUPUARPWSCModel.initBuffer          ( sliceType, qp, (UChar*)INIT_ARPW );
[2]146#endif
[608]147#if H_3D_IC
148  m_cCUICFlagSCModel.initBuffer          ( sliceType, qp, (UChar*)INIT_IC_FLAG );
[443]149#endif
[56]150  m_cCUPartSizeSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_PART_SIZE );
151  m_cCUPredModeSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_PRED_MODE );
152  m_cCUIntraPredSCModel.initBuffer       ( sliceType, qp, (UChar*)INIT_INTRA_PRED_MODE );
153  m_cCUChromaPredSCModel.initBuffer      ( sliceType, qp, (UChar*)INIT_CHROMA_PRED_MODE );
154  m_cCUInterDirSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_INTER_DIR );
155  m_cCUMvdSCModel.initBuffer             ( sliceType, qp, (UChar*)INIT_MVD );
156  m_cCURefPicSCModel.initBuffer          ( sliceType, qp, (UChar*)INIT_REF_PIC );
157  m_cCUDeltaQpSCModel.initBuffer         ( sliceType, qp, (UChar*)INIT_DQP );
158  m_cCUQtCbfSCModel.initBuffer           ( sliceType, qp, (UChar*)INIT_QT_CBF );
159  m_cCUQtRootCbfSCModel.initBuffer       ( sliceType, qp, (UChar*)INIT_QT_ROOT_CBF );
160  m_cCUSigCoeffGroupSCModel.initBuffer   ( sliceType, qp, (UChar*)INIT_SIG_CG_FLAG );
161  m_cCUSigSCModel.initBuffer             ( sliceType, qp, (UChar*)INIT_SIG_FLAG );
162  m_cCuCtxLastX.initBuffer               ( sliceType, qp, (UChar*)INIT_LAST );
163  m_cCuCtxLastY.initBuffer               ( sliceType, qp, (UChar*)INIT_LAST );
164  m_cCUOneSCModel.initBuffer             ( sliceType, qp, (UChar*)INIT_ONE_FLAG );
165  m_cCUAbsSCModel.initBuffer             ( sliceType, qp, (UChar*)INIT_ABS_FLAG );
166  m_cMVPIdxSCModel.initBuffer            ( sliceType, qp, (UChar*)INIT_MVP_IDX );
[608]167  m_cSaoMergeSCModel.initBuffer      ( sliceType, qp, (UChar*)INIT_SAO_MERGE_FLAG );
[443]168  m_cSaoTypeIdxSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_SAO_TYPE_IDX );
[56]169
170  m_cCUTransSubdivFlagSCModel.initBuffer ( sliceType, qp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
[608]171  m_cTransformSkipSCModel.initBuffer     ( sliceType, qp, (UChar*)INIT_TRANSFORMSKIP_FLAG );
172  m_CUTransquantBypassFlagSCModel.initBuffer( sliceType, qp, (UChar*)INIT_CU_TRANSQUANT_BYPASS_FLAG );
173#if H_3D_DIM
174  m_cDepthIntraModeSCModel.initBuffer    ( sliceType, qp, (UChar*)INIT_DEPTH_INTRA_MODE );
175  m_cDdcFlagSCModel.initBuffer           ( sliceType, qp, (UChar*)INIT_DDC_FLAG );
176  m_cDdcDataSCModel.initBuffer           ( sliceType, qp, (UChar*)INIT_DDC_DATA );
[833]177  m_cAngleFlagSCModel.initBuffer         ( sliceType, qp, (UChar*)INIT_ANGLE_FLAG );
[608]178#if H_3D_DIM_SDC
179  m_cSDCResidualFlagSCModel.initBuffer    ( sliceType, qp, (UChar*)INIT_SDC_RESIDUAL_FLAG );
180  m_cSDCResidualSCModel.initBuffer        ( sliceType, qp, (UChar*)INIT_SDC_RESIDUAL );
[2]181#endif
[443]182#endif
[884]183#if H_3D_DIM_SDC
[833]184  m_cSDCFlagSCModel.initBuffer            ( sliceType, qp, (UChar*)INIT_SDC_FLAG );
185#endif
186#if H_3D_DBBP
187  m_cDBBPFlagSCModel.initBuffer              ( sliceType, qp, (UChar*)INIT_DBBP_FLAG );
188#endif
[608]189  m_uiLastDQpNonZero  = 0;
[2]190 
191  // new structure
[56]192  m_uiLastQp          = qp;
[2]193 
194  m_pcTDecBinIf->start();
195}
196
[56]197/** The function does the following: Read out terminate bit. Flush CABAC. Byte-align for next tile.
198 *  Intialize CABAC states. Start CABAC.
199 */
200Void TDecSbac::updateContextTables( SliceType eSliceType, Int iQp )
201{
202  UInt uiBit;
203  m_pcTDecBinIf->decodeBinTrm(uiBit);
[608]204  assert(uiBit); // end_of_sub_stream_one_bit must be equal to 1
[56]205  m_pcTDecBinIf->finish(); 
206  m_pcBitstream->readOutTrailingBits();
207  m_cCUSplitFlagSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_SPLIT_FLAG );
208  m_cCUSkipFlagSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SKIP_FLAG );
[1084]209#if H_3D_SINGLE_DEPTH
[1039]210  m_cCUSingleDepthFlagSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SINGLEDEPTH_FLAG );
211  m_cSingleDepthValueSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_SINGLE_DEPTH_VALUE_DATA );
212#endif
[56]213  m_cCUMergeFlagExtSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_MERGE_FLAG_EXT );
214  m_cCUMergeIdxExtSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_MERGE_IDX_EXT );
[608]215#if H_3D_ARP
216  m_cCUPUARPWSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_ARPW );
[56]217#endif
[608]218#if H_3D_IC
219  m_cCUICFlagSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_IC_FLAG );
[443]220#endif
[56]221  m_cCUPartSizeSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_PART_SIZE );
222  m_cCUPredModeSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_PRED_MODE );
223  m_cCUIntraPredSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_INTRA_PRED_MODE );
224  m_cCUChromaPredSCModel.initBuffer      ( eSliceType, iQp, (UChar*)INIT_CHROMA_PRED_MODE );
225  m_cCUInterDirSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_INTER_DIR );
226  m_cCUMvdSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_MVD );
227  m_cCURefPicSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_REF_PIC );
228  m_cCUDeltaQpSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_DQP );
229  m_cCUQtCbfSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_QT_CBF );
230  m_cCUQtRootCbfSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_QT_ROOT_CBF );
231  m_cCUSigCoeffGroupSCModel.initBuffer   ( eSliceType, iQp, (UChar*)INIT_SIG_CG_FLAG );
232  m_cCUSigSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_SIG_FLAG );
233  m_cCuCtxLastX.initBuffer               ( eSliceType, iQp, (UChar*)INIT_LAST );
234  m_cCuCtxLastY.initBuffer               ( eSliceType, iQp, (UChar*)INIT_LAST );
235  m_cCUOneSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_ONE_FLAG );
236  m_cCUAbsSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_ABS_FLAG );
237  m_cMVPIdxSCModel.initBuffer            ( eSliceType, iQp, (UChar*)INIT_MVP_IDX );
[608]238  m_cSaoMergeSCModel.initBuffer      ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_FLAG );
[443]239  m_cSaoTypeIdxSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SAO_TYPE_IDX );
[56]240  m_cCUTransSubdivFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
[608]241  m_cTransformSkipSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_TRANSFORMSKIP_FLAG );
242  m_CUTransquantBypassFlagSCModel.initBuffer( eSliceType, iQp, (UChar*)INIT_CU_TRANSQUANT_BYPASS_FLAG );
243
244#if H_3D_DIM
245  m_cDepthIntraModeSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_DEPTH_INTRA_MODE );
246  m_cDdcFlagSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_DDC_FLAG );
247  m_cDdcDataSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_DDC_DATA );
[833]248  m_cAngleFlagSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_ANGLE_FLAG );
[608]249#if H_3D_DIM_SDC
250  m_cSDCResidualFlagSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL_FLAG );
251  m_cSDCResidualSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL );
[443]252#endif
[189]253#endif
[884]254#if H_3D_DIM_SDC
[833]255  m_cSDCFlagSCModel.initBuffer            ( eSliceType, iQp, (UChar*)INIT_SDC_FLAG );
256#endif
257#if H_3D_DBBP
258  m_cDBBPFlagSCModel.initBuffer              ( eSliceType, iQp, (UChar*)INIT_DBBP_FLAG );
259#endif
[56]260  m_pcTDecBinIf->start();
261}
262
[608]263Void TDecSbac::parseTerminatingBit( UInt& ruiBit )
[56]264{
[608]265  m_pcTDecBinIf->decodeBinTrm( ruiBit );
266  if ( ruiBit )
[56]267  {
[608]268    m_pcTDecBinIf->finish();
[56]269  }
270}
271
[2]272
273Void TDecSbac::xReadUnaryMaxSymbol( UInt& ruiSymbol, ContextModel* pcSCModel, Int iOffset, UInt uiMaxSymbol )
274{
275  if (uiMaxSymbol == 0)
276  {
277    ruiSymbol = 0;
278    return;
279  }
280 
281  m_pcTDecBinIf->decodeBin( ruiSymbol, pcSCModel[0] );
282 
283  if( ruiSymbol == 0 || uiMaxSymbol == 1 )
284  {
285    return;
286  }
287 
288  UInt uiSymbol = 0;
289  UInt uiCont;
290 
291  do
292  {
293    m_pcTDecBinIf->decodeBin( uiCont, pcSCModel[ iOffset ] );
294    uiSymbol++;
295  }
296  while( uiCont && ( uiSymbol < uiMaxSymbol - 1 ) );
297 
298  if( uiCont && ( uiSymbol == uiMaxSymbol - 1 ) )
299  {
300    uiSymbol++;
301  }
302 
303  ruiSymbol = uiSymbol;
304}
305
306Void TDecSbac::xReadEpExGolomb( UInt& ruiSymbol, UInt uiCount )
307{
308  UInt uiSymbol = 0;
309  UInt uiBit = 1;
310 
311  while( uiBit )
312  {
313    m_pcTDecBinIf->decodeBinEP( uiBit );
314    uiSymbol += uiBit << uiCount++;
315  }
316 
[56]317  if ( --uiCount )
[2]318  {
[56]319    UInt bins;
320    m_pcTDecBinIf->decodeBinsEP( bins, uiCount );
321    uiSymbol += bins;
[2]322  }
323 
324  ruiSymbol = uiSymbol;
325}
326
327Void TDecSbac::xReadUnarySymbol( UInt& ruiSymbol, ContextModel* pcSCModel, Int iOffset )
328{
329  m_pcTDecBinIf->decodeBin( ruiSymbol, pcSCModel[0] );
330 
331  if( !ruiSymbol )
332  {
333    return;
334  }
335 
336  UInt uiSymbol = 0;
337  UInt uiCont;
338 
339  do
340  {
341    m_pcTDecBinIf->decodeBin( uiCont, pcSCModel[ iOffset ] );
342    uiSymbol++;
343  }
344  while( uiCont );
345 
346  ruiSymbol = uiSymbol;
347}
348
[608]349
350/** Parsing of coeff_abs_level_remaing
351 * \param ruiSymbol reference to coeff_abs_level_remaing
352 * \param ruiParam reference to parameter
[2]353 * \returns Void
354 */
[608]355Void TDecSbac::xReadCoefRemainExGolomb ( UInt &rSymbol, UInt &rParam )
[2]356{
357
[608]358  UInt prefix   = 0;
359  UInt codeWord = 0;
[2]360  do
361  {
[608]362    prefix++;
363    m_pcTDecBinIf->decodeBinEP( codeWord );
[2]364  }
[608]365  while( codeWord);
366  codeWord  = 1 - codeWord;
367  prefix -= codeWord;
368  codeWord=0;
369  if (prefix < COEF_REMAIN_BIN_REDUCTION )
370  {
371    m_pcTDecBinIf->decodeBinsEP(codeWord,rParam);
372    rSymbol = (prefix<<rParam) + codeWord;
373  }
374  else
375  {
376    m_pcTDecBinIf->decodeBinsEP(codeWord,prefix-COEF_REMAIN_BIN_REDUCTION+rParam);
377    rSymbol = (((1<<(prefix-COEF_REMAIN_BIN_REDUCTION))+COEF_REMAIN_BIN_REDUCTION-1)<<rParam)+codeWord;
378  }
379}
[2]380
[608]381#if H_3D_DIM
382Void TDecSbac::xReadExGolombLevel( UInt& ruiSymbol, ContextModel& rcSCModel  )
383{
384  UInt uiSymbol;
385  UInt uiCount = 0;
386  do
387  {
388    m_pcTDecBinIf->decodeBin( uiSymbol, rcSCModel );
389    uiCount++;
390  }
[950]391  while( uiSymbol && ( uiCount != 3 ) );
[608]392  ruiSymbol = uiCount - 1;
393
394  if( uiSymbol )
[2]395  {
[608]396    xReadEpExGolomb( uiSymbol, 0 );
397    ruiSymbol += uiSymbol + 1;
[2]398  }
399
[608]400  return;
401}
[2]402
[724]403Void TDecSbac::xParseDimDeltaDC( Pel& rValDeltaDC, UInt uiNumSeg )
404{
405  UInt absValDeltaDC = 0;
406  xReadExGolombLevel( absValDeltaDC, m_cDdcDataSCModel.get(0, 0, 0) );
407  rValDeltaDC = (Pel)absValDeltaDC + ( uiNumSeg > 1 ? 0 : 1 );
408
409  if( rValDeltaDC != 0 )
410  {
411    UInt uiSign;
412    m_pcTDecBinIf->decodeBinEP( uiSign );
413    if ( uiSign )
414    {
415      rValDeltaDC = -rValDeltaDC;
416    }
417  }
418}
[608]419
[724]420
[608]421#if H_3D_DIM_DMM
422Void TDecSbac::xParseDmm1WedgeIdx( UInt& ruiTabIdx, Int iNumBit )
423{
424  UInt uiSymbol, uiIdx = 0;
425  for( Int i = 0; i < iNumBit; i++ )
426  {
[976]427    m_pcTDecBinIf->decodeBinEP( uiSymbol );
[608]428    uiIdx += uiSymbol << i;
429  }
430  ruiTabIdx = uiIdx;
431}
432#endif
[2]433
[1084]434
[608]435#endif
[56]436/** Parse I_PCM information.
437 * \param pcCU
438 * \param uiAbsPartIdx
439 * \param uiDepth
440 * \returns Void
441 *
442 * If I_PCM flag indicates that the CU is I_PCM, parse its PCM alignment bits and codes.
443 */
444Void TDecSbac::parseIPCMInfo ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
[2]445{
446  UInt uiSymbol;
[56]447
448    m_pcTDecBinIf->decodeBinTrm(uiSymbol);
[2]449
[608]450#if H_MV_ENC_DEC_TRAC
451      DTRACE_CU("pcm_flag", uiSymbol)
452#endif
[56]453    if (uiSymbol)
454    {
455    Bool bIpcmFlag = true;
456
457    pcCU->setPartSizeSubParts  ( SIZE_2Nx2N, uiAbsPartIdx, uiDepth );
458    pcCU->setSizeSubParts      ( g_uiMaxCUWidth>>uiDepth, g_uiMaxCUHeight>>uiDepth, uiAbsPartIdx, uiDepth );
[608]459    pcCU->setTrIdxSubParts     ( 0, uiAbsPartIdx, uiDepth );
[56]460    pcCU->setIPCMFlagSubParts  ( bIpcmFlag, uiAbsPartIdx, uiDepth );
[2]461
[56]462    UInt uiMinCoeffSize = pcCU->getPic()->getMinCUWidth()*pcCU->getPic()->getMinCUHeight();
463    UInt uiLumaOffset   = uiMinCoeffSize*uiAbsPartIdx;
464    UInt uiChromaOffset = uiLumaOffset>>2;
465
466    Pel* piPCMSample;
467    UInt uiWidth;
468    UInt uiHeight;
469    UInt uiSampleBits;
470    UInt uiX, uiY;
471
472    piPCMSample = pcCU->getPCMSampleY() + uiLumaOffset;
473    uiWidth = pcCU->getWidth(uiAbsPartIdx);
474    uiHeight = pcCU->getHeight(uiAbsPartIdx);
475    uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthLuma();
476
477    for(uiY = 0; uiY < uiHeight; uiY++)
478    {
479      for(uiX = 0; uiX < uiWidth; uiX++)
480      {
481        UInt uiSample;
482        m_pcTDecBinIf->xReadPCMCode(uiSampleBits, uiSample);
483        piPCMSample[uiX] = uiSample;
484      }
485      piPCMSample += uiWidth;
486    }
487
488    piPCMSample = pcCU->getPCMSampleCb() + uiChromaOffset;
489    uiWidth = pcCU->getWidth(uiAbsPartIdx)/2;
490    uiHeight = pcCU->getHeight(uiAbsPartIdx)/2;
491    uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
492
493    for(uiY = 0; uiY < uiHeight; uiY++)
494    {
495      for(uiX = 0; uiX < uiWidth; uiX++)
496      {
497        UInt uiSample;
498        m_pcTDecBinIf->xReadPCMCode(uiSampleBits, uiSample);
499        piPCMSample[uiX] = uiSample;
500      }
501      piPCMSample += uiWidth;
502    }
503
504    piPCMSample = pcCU->getPCMSampleCr() + uiChromaOffset;
505    uiWidth = pcCU->getWidth(uiAbsPartIdx)/2;
506    uiHeight = pcCU->getHeight(uiAbsPartIdx)/2;
507    uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
508
509    for(uiY = 0; uiY < uiHeight; uiY++)
510    {
511      for(uiX = 0; uiX < uiWidth; uiX++)
512      {
513        UInt uiSample;
514        m_pcTDecBinIf->xReadPCMCode(uiSampleBits, uiSample);
515        piPCMSample[uiX] = uiSample;
516      }
517      piPCMSample += uiWidth;
518    }
519
[608]520    m_pcTDecBinIf->start();
[2]521  }
522}
523
[608]524Void TDecSbac::parseCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
525{
526  UInt uiSymbol;
527  m_pcTDecBinIf->decodeBin( uiSymbol, m_CUTransquantBypassFlagSCModel.get( 0, 0, 0 ) );
528#if H_MV_ENC_DEC_TRAC
529  DTRACE_CU("cu_transquant_bypass_flag", uiSymbol); 
530#endif
531  pcCU->setCUTransquantBypassSubParts(uiSymbol ? true : false, uiAbsPartIdx, uiDepth);
532}
533
[2]534/** parse skip flag
535 * \param pcCU
536 * \param uiAbsPartIdx
537 * \param uiDepth
538 * \returns Void
539 */
540Void TDecSbac::parseSkipFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
541{
542  if( pcCU->getSlice()->isIntra() )
543  {
544    return;
545  }
546 
547  UInt uiSymbol = 0;
548  UInt uiCtxSkip = pcCU->getCtxSkipFlag( uiAbsPartIdx );
549  m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUSkipFlagSCModel.get( 0, 0, uiCtxSkip ) );
[608]550#if !H_MV_ENC_DEC_TRAC
[56]551  DTRACE_CABAC_VL( g_nSymbolCounter++ );
[2]552  DTRACE_CABAC_T( "\tSkipFlag" );
553  DTRACE_CABAC_T( "\tuiCtxSkip: ");
554  DTRACE_CABAC_V( uiCtxSkip );
555  DTRACE_CABAC_T( "\tuiSymbol: ");
556  DTRACE_CABAC_V( uiSymbol );
557  DTRACE_CABAC_T( "\n");
[608]558#endif
[2]559 
560  if( uiSymbol )
561  {
[608]562    pcCU->setSkipFlagSubParts( true,        uiAbsPartIdx, uiDepth );
563    pcCU->setPredModeSubParts( MODE_INTER,  uiAbsPartIdx, uiDepth );
[2]564    pcCU->setPartSizeSubParts( SIZE_2Nx2N, uiAbsPartIdx, uiDepth );
565    pcCU->setSizeSubParts( g_uiMaxCUWidth>>uiDepth, g_uiMaxCUHeight>>uiDepth, uiAbsPartIdx, uiDepth );
566    pcCU->setMergeFlagSubParts( true , uiAbsPartIdx, 0, uiDepth );
567  }
[608]568#if H_MV_ENC_DEC_TRAC
569  DTRACE_CU("cu_skip_flag", uiSymbol); 
570#endif
[2]571}
[1084]572#if H_3D_SINGLE_DEPTH
[1039]573Void TDecSbac::parseSingleDepthMode( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
574{
575  pcCU->setSingleDepthFlagSubParts( false,        uiAbsPartIdx, uiDepth );
576  UInt uiSymbol = 0;
577  m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUSingleDepthFlagSCModel.get( 0, 0, 0 ) );
578  if( uiSymbol )
579  {
580    pcCU->setSingleDepthFlagSubParts( true,        uiAbsPartIdx, uiDepth );
581    pcCU->setSkipFlagSubParts( false,        uiAbsPartIdx, uiDepth );
582    pcCU->setSDCFlagSubParts( false,        uiAbsPartIdx, uiDepth );
583    pcCU->setPredModeSubParts( MODE_INTRA,  uiAbsPartIdx, uiDepth );
584    pcCU->setPartSizeSubParts( SIZE_2Nx2N, uiAbsPartIdx, uiDepth );
585    pcCU->setLumaIntraDirSubParts (DC_IDX, uiAbsPartIdx, uiDepth );
586    pcCU->setSizeSubParts( g_uiMaxCUWidth>>uiDepth, g_uiMaxCUHeight>>uiDepth, uiAbsPartIdx, uiDepth );
587    pcCU->setMergeFlagSubParts( false , uiAbsPartIdx, 0, uiDepth );
588    pcCU->setTrIdxSubParts(0, uiAbsPartIdx, uiDepth);
589    pcCU->setCbfSubParts(0, 1, 1, uiAbsPartIdx, uiDepth);
[2]590
[1039]591    UInt absValDeltaDC = 0;
592
593    UInt uiUnaryIdx = 0;
[1084]594    UInt uiNumCand = SINGLE_DEPTH_MODE_CAND_LIST_SIZE;
[1039]595    if ( uiNumCand > 1 )
596    {
597      for( ; uiUnaryIdx < uiNumCand - 1; ++uiUnaryIdx )
598      {
599        UInt uiSymbol2 = 0;
600        if ( uiUnaryIdx==0 )
601        {
602          m_pcTDecBinIf->decodeBin( uiSymbol2, m_cSingleDepthValueSCModel.get( 0, 0, 0 ) );
603        }
604        else
605        {
606          m_pcTDecBinIf->decodeBinEP( uiSymbol2);
607        }
608        if( uiSymbol2 == 0 )
609        {
610          break;
611        }
612      }
613    }
614    absValDeltaDC = uiUnaryIdx;
615    pcCU->setSingleDepthValueSubParts((Pel)absValDeltaDC,uiAbsPartIdx, 0, uiDepth);
616  }
617}
618
619#endif
[2]620/** parse merge flag
621 * \param pcCU
622 * \param uiAbsPartIdx
623 * \param uiDepth
624 * \param uiPUIdx
625 * \returns Void
626 */
627Void TDecSbac::parseMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiPUIdx )
628{
629  UInt uiSymbol;
[56]630  m_pcTDecBinIf->decodeBin( uiSymbol, *m_cCUMergeFlagExtSCModel.get( 0 ) );
[608]631#if H_MV_ENC_DEC_TRAC
632  DTRACE_PU("merge_flag", uiSymbol)
633#endif
[2]634  pcCU->setMergeFlagSubParts( uiSymbol ? true : false, uiAbsPartIdx, uiPUIdx, uiDepth );
635
[608]636#if !H_MV_ENC_DEC_TRAC
[56]637  DTRACE_CABAC_VL( g_nSymbolCounter++ );
[2]638  DTRACE_CABAC_T( "\tMergeFlag: " );
639  DTRACE_CABAC_V( uiSymbol );
640  DTRACE_CABAC_T( "\tAddress: " );
641  DTRACE_CABAC_V( pcCU->getAddr() );
642  DTRACE_CABAC_T( "\tuiAbsPartIdx: " );
643  DTRACE_CABAC_V( uiAbsPartIdx );
644  DTRACE_CABAC_T( "\n" );
[608]645#endif
[2]646}
647
[608]648Void TDecSbac::parseMergeIndex ( TComDataCU* pcCU, UInt& ruiMergeIndex )
[2]649{
[56]650  UInt uiUnaryIdx = 0;
[608]651  UInt uiNumCand = pcCU->getSlice()->getMaxNumMergeCand();
[56]652  if ( uiNumCand > 1 )
[2]653  {
[56]654    for( ; uiUnaryIdx < uiNumCand - 1; ++uiUnaryIdx )
[2]655    {
[56]656      UInt uiSymbol = 0;
657      if ( uiUnaryIdx==0 )
[2]658      {
[56]659        m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUMergeIdxExtSCModel.get( 0, 0, 0 ) );
[2]660      }
[56]661      else
662      {
663        m_pcTDecBinIf->decodeBinEP( uiSymbol );
664      }
665      if( uiSymbol == 0 )
[2]666      {
[56]667        break;
[2]668      }
669    }
[608]670#if H_MV_ENC_DEC_TRAC
671    DTRACE_PU("merge_idx", uiUnaryIdx)
672#endif
[2]673  }
674  ruiMergeIndex = uiUnaryIdx;
675
[608]676#if !H_MV_ENC_DEC_TRAC
[56]677  DTRACE_CABAC_VL( g_nSymbolCounter++ )
678  DTRACE_CABAC_T( "\tparseMergeIndex()" )
679  DTRACE_CABAC_T( "\tuiMRGIdx= " )
680  DTRACE_CABAC_V( ruiMergeIndex )
681  DTRACE_CABAC_T( "\n" )
[2]682#endif
683}
684
[56]685Void TDecSbac::parseMVPIdx      ( Int& riMVPIdx )
686{
687  UInt uiSymbol;
688  xReadUnaryMaxSymbol(uiSymbol, m_cMVPIdxSCModel.get(0), 1, AMVP_MAX_NUM_CANDS-1);
689  riMVPIdx = uiSymbol;
690}
[2]691
692Void TDecSbac::parseSplitFlag     ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
693{
694  if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth )
695  {
696    pcCU->setDepthSubParts( uiDepth, uiAbsPartIdx );
697    return;
698  }
[608]699
[2]700  UInt uiSymbol;
[608]701#if H_3D_QTLPC
[189]702  Bool bParseSplitFlag    = true;
703
[1039]704  TComVPS *vps           = pcCU->getPic()->getSlice(0)->getVPS();
705  Bool    bLimQtPredFlag = vps->getLimQtPredFlag(pcCU->getPic()->getSlice(0)->getLayerId());
[189]706  TComPic *pcTexture      = pcCU->getSlice()->getTexturePic();
707  Bool bDepthMapDetect    = (pcTexture != NULL);
708  Bool bIntraSliceDetect  = (pcCU->getSlice()->getSliceType() == I_SLICE);
709
[608]710  Bool rapPic = (pcCU->getSlice()->getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_W_RADL || pcCU->getSlice()->getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_N_LP || pcCU->getSlice()->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA);
711
[1084]712#if H_3D_FCO
[1066]713  if(bDepthMapDetect && !bIntraSliceDetect && !rapPic && bLimQtPredFlag && pcTexture->getReconMark())
714#else
[1039]715  if(bDepthMapDetect && !bIntraSliceDetect && !rapPic && bLimQtPredFlag)
[1066]716#endif
[189]717  {
718    TComDataCU *pcTextureCU = pcTexture->getCU(pcCU->getAddr());
719    assert(pcTextureCU->getDepth(uiAbsPartIdx) >= uiDepth);
720    bParseSplitFlag         = (pcTextureCU->getDepth(uiAbsPartIdx) > uiDepth);
721  }
722
723  if(bParseSplitFlag)
724  {
725#endif
726    m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUSplitFlagSCModel.get( 0, 0, pcCU->getCtxSplitFlag( uiAbsPartIdx, uiDepth ) ) );
[608]727#if H_MV_ENC_DEC_TRAC
728    DTRACE_CU("split_cu_flag", uiSymbol); 
729#else
[189]730    DTRACE_CABAC_VL( g_nSymbolCounter++ )
731    DTRACE_CABAC_T( "\tSplitFlag\n" )
[608]732#endif
733#if H_3D_QTLPC
[189]734  }
735  else
[608]736  {
[189]737    uiSymbol = 0;
[608]738  }
[189]739#endif
[2]740  pcCU->setDepthSubParts( uiDepth + uiSymbol, uiAbsPartIdx );
741 
742  return;
743}
744
745/** parse partition size
746 * \param pcCU
747 * \param uiAbsPartIdx
748 * \param uiDepth
749 * \returns Void
750 */
751Void TDecSbac::parsePartSize( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
752{
753  UInt uiSymbol, uiMode = 0;
754  PartSize eMode;
[189]755
[608]756#if H_3D_QTLPC
[189]757  Bool bParsePartSize    = true;
[1039]758  TComVPS *vps           = pcCU->getPic()->getSlice(0)->getVPS();
759  Bool    bLimQtPredFlag = vps->getLimQtPredFlag(pcCU->getPic()->getSlice(0)->getLayerId());
[189]760  TComPic *pcTexture     = pcCU->getSlice()->getTexturePic();
761  Bool bDepthMapDetect   = (pcTexture != NULL);
762  Bool bIntraSliceDetect = (pcCU->getSlice()->getSliceType() == I_SLICE);
763
[608]764  Bool rapPic     = (pcCU->getSlice()->getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_W_RADL || pcCU->getSlice()->getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR_N_LP || pcCU->getSlice()->getNalUnitType() == NAL_UNIT_CODED_SLICE_CRA);
765
[833]766  Bool depthDependent = false;
767  UInt uiTexturePart = uiMode;
[1084]768
769#if H_3D_FCO
[1066]770  if(bDepthMapDetect && !bIntraSliceDetect && !rapPic && bLimQtPredFlag && pcTexture->getReconMark())
771#else
[1039]772  if(bDepthMapDetect && !bIntraSliceDetect && !rapPic && bLimQtPredFlag )
[1066]773#endif
[189]774  {
775    TComDataCU *pcTextureCU = pcTexture->getCU(pcCU->getAddr());
776    assert(pcTextureCU->getDepth(uiAbsPartIdx) >= uiDepth);
[833]777    if(pcTextureCU->getDepth(uiAbsPartIdx) == uiDepth )
[189]778    {
[833]779      depthDependent = true;
780      uiTexturePart = pcTextureCU->getPartitionSize( uiAbsPartIdx );
781    }
782    if (pcTextureCU->getDepth(uiAbsPartIdx) == uiDepth && pcTextureCU->getPartitionSize( uiAbsPartIdx ) == SIZE_2Nx2N)
783    {
[189]784      bParsePartSize = false;
785      eMode          = SIZE_2Nx2N;
786    }
787  }
788#endif
[608]789
[2]790 
791  if ( pcCU->isIntra( uiAbsPartIdx ) )
792  {
[608]793#if H_3D_QTLPC
[189]794    if(bParsePartSize)
[2]795    {
[189]796#endif
[608]797     uiSymbol = 1;
[189]798      if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth )
799      {
800        m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUPartSizeSCModel.get( 0, 0, 0) );
[608]801#if H_MV_ENC_DEC_TRAC         
802        DTRACE_CU("part_mode", uiSymbol)
803#endif       
[189]804      }
805      eMode = uiSymbol ? SIZE_2Nx2N : SIZE_NxN;
[608]806#if H_3D_QTLPC
[2]807    }
[189]808#endif
[56]809    UInt uiTrLevel = 0;   
810    UInt uiWidthInBit  = g_aucConvertToBit[pcCU->getWidth(uiAbsPartIdx)]+2;
811    UInt uiTrSizeInBit = g_aucConvertToBit[pcCU->getSlice()->getSPS()->getMaxTrSize()]+2;
812    uiTrLevel          = uiWidthInBit >= uiTrSizeInBit ? uiWidthInBit - uiTrSizeInBit : 0;
813    if( eMode == SIZE_NxN )
814    {
815      pcCU->setTrIdxSubParts( 1+uiTrLevel, uiAbsPartIdx, uiDepth );
816    }
817    else
818    {
819      pcCU->setTrIdxSubParts( uiTrLevel, uiAbsPartIdx, uiDepth );
820    }
[2]821  }
822  else
823  {
[608]824#if H_3D_QTLPC
[189]825    if(bParsePartSize)
[2]826    {
[833]827      if (depthDependent==false || uiTexturePart == SIZE_NxN|| uiTexturePart == SIZE_2Nx2N)
828      {
829#endif
[189]830      UInt uiMaxNumBits = 2;
[608]831      if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth && !( (g_uiMaxCUWidth>>uiDepth) == 8 && (g_uiMaxCUHeight>>uiDepth) == 8 ) )
[2]832      {
[189]833        uiMaxNumBits ++;
[2]834      }
[189]835      for ( UInt ui = 0; ui < uiMaxNumBits; ui++ )
[56]836      {
[189]837        m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUPartSizeSCModel.get( 0, 0, ui) );
838        if ( uiSymbol )
839        {
840          break;
841        }
842        uiMode++;
843      }
844      eMode = (PartSize) uiMode;
845      if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
846      {
847        if (eMode == SIZE_2NxN)
848        {
[872]849        m_pcTDecBinIf->decodeBin(uiSymbol, m_cCUPartSizeSCModel.get( 0, 0, 3 ));
[189]850          if (uiSymbol == 0)
851          {
852            m_pcTDecBinIf->decodeBinEP(uiSymbol);
853            eMode = (uiSymbol == 0? SIZE_2NxnU : SIZE_2NxnD);
854          }
[56]855        }
[189]856        else if (eMode == SIZE_Nx2N)
857        {
[872]858        m_pcTDecBinIf->decodeBin(uiSymbol, m_cCUPartSizeSCModel.get( 0, 0, 3 ));
[189]859          if (uiSymbol == 0)
860          {
861            m_pcTDecBinIf->decodeBinEP(uiSymbol);
862            eMode = (uiSymbol == 0? SIZE_nLx2N : SIZE_nRx2N);
863          }
[2]864        }
865      }
[884]866#if H_3D_QTLPC
[833]867      }
868      else if(uiTexturePart == SIZE_2NxN || uiTexturePart == SIZE_2NxnU || uiTexturePart == SIZE_2NxnD)
869      {
870        UInt uiMaxNumBits = 1;
871        if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
872        {
873          uiMaxNumBits ++;
874        }
875        for ( UInt ui = 0; ui < uiMaxNumBits; ui++ )
876        {
877          m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUPartSizeSCModel.get( 0, 0, ui) );
878          if ( uiSymbol )
879          {
880            break;
881          }
882          uiMode++;
883        }
884        eMode = (PartSize) uiMode;
885        if(uiMode && pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) && uiSymbol==1 )
886        {
887          eMode = SIZE_2NxN;
888        }
889        else if (uiMode && pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth )==0  && uiSymbol==0)
890        {
891          eMode = SIZE_2NxN;
892        }
893        else if (uiMode && pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) && uiSymbol==0)
894        {
895          m_pcTDecBinIf->decodeBinEP(uiSymbol);
896          eMode = (uiSymbol == 0? SIZE_2NxnU : SIZE_2NxnD);
897        }
898      }
899      else if(uiTexturePart == SIZE_Nx2N|| uiTexturePart==SIZE_nLx2N || uiTexturePart==SIZE_nRx2N)
900      {
901        UInt uiMaxNumBits = 1;
902        if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
903        {
904          uiMaxNumBits ++;
905        }
906        for ( UInt ui = 0; ui < uiMaxNumBits; ui++ )
907        {
908          m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUPartSizeSCModel.get( 0, 0, ui) );
909          if ( uiSymbol )
910          {
911            break;
912          }
913          uiMode++;
914        }
915        eMode = (PartSize) uiMode;
916        if(uiMode && pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) && uiSymbol==1 )
917        {
918          eMode = SIZE_Nx2N;
919        }
920        else if (uiMode && pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth )==0  && uiSymbol==0)
921        {
922          eMode = SIZE_Nx2N;
923        }
924        else if (uiMode && pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) && uiSymbol==0)
925        {
926          m_pcTDecBinIf->decodeBinEP(uiSymbol);
927          eMode = (uiSymbol == 0? SIZE_nLx2N : SIZE_nRx2N);
928        }
929      }
930      else
[884]931      {
[833]932        assert(0);
[884]933      }
[964]934#endif
[608]935#if H_MV_ENC_DEC_TRAC         
936      DTRACE_CU("part_mode", eMode )
937#endif
[964]938#if H_3D_QTLPC
[2]939    }
[189]940#endif
[2]941  }
942  pcCU->setPartSizeSubParts( eMode, uiAbsPartIdx, uiDepth );
943  pcCU->setSizeSubParts( g_uiMaxCUWidth>>uiDepth, g_uiMaxCUHeight>>uiDepth, uiAbsPartIdx, uiDepth );
944}
945
946/** parse prediction mode
947 * \param pcCU
948 * \param uiAbsPartIdx
949 * \param uiDepth
950 * \returns Void
951 */
952Void TDecSbac::parsePredMode( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
953{
954  if( pcCU->getSlice()->isIntra() )
955  {
956    pcCU->setPredModeSubParts( MODE_INTRA, uiAbsPartIdx, uiDepth );
957    return;
958  }
959 
960  UInt uiSymbol;
961  Int  iPredMode = MODE_INTER;
[56]962  m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUPredModeSCModel.get( 0, 0, 0 ) );
[2]963  iPredMode += uiSymbol;
[608]964#if H_MV_ENC_DEC_TRAC         
965  DTRACE_CU("pred_mode_flag", uiSymbol)
966#endif       
[2]967  pcCU->setPredModeSubParts( (PredMode)iPredMode, uiAbsPartIdx, uiDepth );
968}
[608]969
970Void TDecSbac::parseIntraDirLumaAng  ( TComDataCU* pcCU, UInt absPartIdx, UInt depth )
[443]971{
[608]972  PartSize mode = pcCU->getPartitionSize( absPartIdx );
973  UInt partNum = mode==SIZE_NxN?4:1;
974  UInt partOffset = ( pcCU->getPic()->getNumPartInCU() >> ( pcCU->getDepth(absPartIdx) << 1 ) ) >> 2;
975  UInt mpmPred[4],symbol;
976  Int j,intraPredMode;   
977  if (mode==SIZE_NxN)
[443]978  {
[608]979    depth++;
[443]980  }
[608]981  for (j=0;j<partNum;j++)
[443]982  {
[608]983#if H_3D_DIM
[1039]984    if( pcCU->getSlice()->getVpsDepthModesFlag() || pcCU->getSlice()->getIVPFlag() )
[443]985    {
[608]986      parseIntraDepth( pcCU, absPartIdx+partOffset*j, depth );
[443]987    }
[608]988    if( pcCU->getLumaIntraDir( absPartIdx+partOffset*j ) < NUM_INTRA_MODE )
989    {
990#endif
991    m_pcTDecBinIf->decodeBin( symbol, m_cCUIntraPredSCModel.get( 0, 0, 0) );
992    mpmPred[j] = symbol;
993#if H_MV_ENC_DEC_TRAC         
994    DTRACE_CU("prev_intra_luma_pred_flag", symbol)
995#endif
996#if H_3D_DIM
997    }
998#endif
[443]999  }
[608]1000  for (j=0;j<partNum;j++)
[443]1001  {
[608]1002#if H_3D_DIM
1003    if( pcCU->getLumaIntraDir( absPartIdx+partOffset*j ) < NUM_INTRA_MODE )
[443]1004    {
[608]1005#endif
1006    Int preds[3] = {-1, -1, -1};
1007    Int predNum = pcCU->getIntraDirLumaPredictor(absPartIdx+partOffset*j, preds); 
1008    if (mpmPred[j])
1009    {
1010      m_pcTDecBinIf->decodeBinEP( symbol );
1011      if (symbol)
1012      {
1013        m_pcTDecBinIf->decodeBinEP( symbol );
1014        symbol++;
[443]1015      }
[608]1016#if H_MV_ENC_DEC_TRAC         
1017      DTRACE_CU("mpm_idx", symbol)
1018#endif
1019      intraPredMode = preds[symbol];
[443]1020    }
1021    else
1022    {
[608]1023      m_pcTDecBinIf->decodeBinsEP( symbol, 5 );
1024      intraPredMode = symbol;
1025#if H_MV_ENC_DEC_TRAC         
1026      DTRACE_CU("rem_intra_luma_pred_mode", symbol)
1027#endif       
1028      //postponed sorting of MPMs (only in remaining branch)
1029      if (preds[0] > preds[1])
[443]1030      { 
[608]1031        std::swap(preds[0], preds[1]); 
[443]1032      }
[608]1033      if (preds[0] > preds[2])
[443]1034      {
[608]1035        std::swap(preds[0], preds[2]);
[443]1036      }
[608]1037      if (preds[1] > preds[2])
1038      {
1039        std::swap(preds[1], preds[2]);
1040      }
1041      for ( Int i = 0; i < predNum; i++ )
1042      {
1043        intraPredMode += ( intraPredMode >= preds[i] );
1044      }
[443]1045    }
[608]1046    pcCU->setLumaIntraDirSubParts( (UChar)intraPredMode, absPartIdx+partOffset*j, depth );
1047#if H_3D_DIM
1048    }
1049#endif
[443]1050  }
1051}
[608]1052
1053Void TDecSbac::parseIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
[443]1054{
[608]1055  UInt uiSymbol;
1056
1057  m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUChromaPredSCModel.get( 0, 0, 0 ) );
1058
1059  if( uiSymbol == 0 )
[443]1060  {
[608]1061#if H_MV_ENC_DEC_TRAC         
1062    DTRACE_CU("intra_chroma_pred_mode", uiSymbol )
1063#endif       
1064    uiSymbol = DM_CHROMA_IDX;
[443]1065  } 
[608]1066  else 
[443]1067  {
[608]1068    {
1069      UInt uiIPredMode;
1070      m_pcTDecBinIf->decodeBinsEP( uiIPredMode, 2 );
1071#if H_MV_ENC_DEC_TRAC         
1072      DTRACE_CU("intra_chroma_pred_mode", uiIPredMode )
1073#endif       
1074      UInt uiAllowedChromaDir[ NUM_CHROMA_MODE ];
1075      pcCU->getAllowedChromaDir( uiAbsPartIdx, uiAllowedChromaDir );
1076      uiSymbol = uiAllowedChromaDir[ uiIPredMode ];
1077    }
[443]1078  }
[608]1079  pcCU->setChromIntraDirSubParts( uiSymbol, uiAbsPartIdx, uiDepth );
1080  return;
1081}
1082
1083#if H_3D_DIM
1084Void TDecSbac::parseIntraDepth( TComDataCU* pcCU, UInt absPartIdx, UInt depth )
1085{
1086  parseIntraDepthMode( pcCU, absPartIdx, depth );
1087
1088  UInt dir     = pcCU->getLumaIntraDir( absPartIdx );
1089  UInt dimType = getDimType( dir );
1090
1091  switch( dimType )
[443]1092  {
[608]1093#if H_3D_DIM_DMM
1094  case( DMM1_IDX ):
[443]1095    {
[608]1096      UInt uiTabIdx = 0;
1097      xParseDmm1WedgeIdx( uiTabIdx, g_dmm1TabIdxBits[pcCU->getIntraSizeIdx(absPartIdx)] );
1098      pcCU->setDmmWedgeTabIdxSubParts( uiTabIdx, dimType, absPartIdx, depth );
1099    } break;
1100  case( DMM4_IDX ): break;
1101#endif
1102  default: break;
1103  }
[443]1104
[608]1105  pcCU->setLumaIntraDirSubParts( (UChar)dir, absPartIdx, depth );
1106}
[443]1107
[608]1108Void TDecSbac::parseIntraDepthMode( TComDataCU* pcCU, UInt absPartIdx, UInt depth )
[2]1109{
[833]1110  UInt uiSymbol, uiIsDimMode;
1111
1112  if( ( pcCU->getSlice()->getSPS()->getMaxCUWidth() >> pcCU->getDepth( absPartIdx ) ) < 64 ) //DMM and HEVC intra modes are both allowed
1113  {
[950]1114    m_pcTDecBinIf->decodeBin( uiSymbol, m_cAngleFlagSCModel.get( 0, 0, 0 ) );
[833]1115  }
1116  else
1117  {
1118    uiSymbol = 1;
1119  }
1120  uiIsDimMode = uiSymbol ? 0 : 1;
1121  pcCU->setLumaIntraDirSubParts( 0, absPartIdx, depth );
1122
1123  //decode DMM index
1124  if( uiIsDimMode )
1125  {
[1039]1126    if( pcCU->getSlice()->getVpsDepthModesFlag() && pcCU->getSlice()->getIVPFlag() )
1127    {
1128      m_pcTDecBinIf->decodeBin( uiSymbol, m_cDepthIntraModeSCModel.get( 0, 0, 0 ) );
1129      if( !uiSymbol )
1130      {
1131        pcCU->setLumaIntraDirSubParts( DIM_OFFSET, absPartIdx, depth );
1132      }
1133      else
1134      {
1135        pcCU->setLumaIntraDirSubParts( ( 1+ DIM_OFFSET ), absPartIdx, depth );
1136      }
1137    }
1138    else if ( pcCU->getSlice()->getVpsDepthModesFlag() )
1139    {
1140      pcCU->setLumaIntraDirSubParts( DIM_OFFSET, absPartIdx, depth );
1141    }
1142    else if( pcCU->getSlice()->getIVPFlag() )
1143    {
1144      pcCU->setLumaIntraDirSubParts( ( 1+ DIM_OFFSET ), absPartIdx, depth );
1145    }
[833]1146  }
1147}
[2]1148#endif
1149
[608]1150Void TDecSbac::parseInterDir( TComDataCU* pcCU, UInt& ruiInterDir, UInt uiAbsPartIdx )
[2]1151{
1152  UInt uiSymbol;
[608]1153  const UInt uiCtx = pcCU->getCtxInterDir( uiAbsPartIdx );
1154  ContextModel *pCtx = m_cCUInterDirSCModel.get( 0 );
1155  uiSymbol = 0;
1156  if (pcCU->getPartitionSize(uiAbsPartIdx) == SIZE_2Nx2N || pcCU->getHeight(uiAbsPartIdx) != 8 )
[2]1157  {
[608]1158    m_pcTDecBinIf->decodeBin( uiSymbol, *( pCtx + uiCtx ) );
[2]1159  }
1160
[56]1161  if( uiSymbol )
[2]1162  {
1163    uiSymbol = 2;
1164  }
[608]1165  else
1166  {
1167    m_pcTDecBinIf->decodeBin( uiSymbol, *( pCtx + 4 ) );
1168    assert(uiSymbol == 0 || uiSymbol == 1);
1169  }
[56]1170
[2]1171  uiSymbol++;
1172  ruiInterDir = uiSymbol;
[608]1173#if H_MV_ENC_DEC_TRAC
1174    DTRACE_PU("inter_pred_idc", ruiInterDir - 1 )   
1175#endif
1176
[2]1177  return;
1178}
1179
[608]1180Void TDecSbac::parseRefFrmIdx( TComDataCU* pcCU, Int& riRefFrmIdx, RefPicList eRefList )
[2]1181{
1182  UInt uiSymbol;
1183  {
[56]1184    ContextModel *pCtx = m_cCURefPicSCModel.get( 0 );
1185    m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
[2]1186
[56]1187    if( uiSymbol )
[2]1188    {
[608]1189      UInt uiRefNum = pcCU->getSlice()->getNumRefIdx( eRefList ) - 2;
1190      pCtx++;
1191      UInt ui;
1192      for( ui = 0; ui < uiRefNum; ++ui )
1193      {
1194        if( ui == 0 )
1195        {
1196          m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
1197        }
1198        else
1199        {
1200          m_pcTDecBinIf->decodeBinEP( uiSymbol );
1201        }
1202        if( uiSymbol == 0 )
1203        {
1204          break;
1205        }
1206      }
1207      uiSymbol = ui + 1;
[2]1208    }
1209    riRefFrmIdx = uiSymbol;
1210  }
[608]1211
1212#if H_MV_ENC_DEC_TRAC
1213#if ENC_DEC_TRACE
1214  if ( eRefList == REF_PIC_LIST_0 )
1215  {
1216    DTRACE_PU("ref_idx_l0", uiSymbol)
1217  }
[2]1218  else
1219  {
[608]1220    DTRACE_PU("ref_idx_l1", uiSymbol)
[2]1221  }
[608]1222#endif
1223#endif
[2]1224  return;
1225}
1226
[56]1227Void TDecSbac::parseMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiPartIdx, UInt uiDepth, RefPicList eRefList )
[2]1228{
1229  UInt uiSymbol;
[56]1230  UInt uiHorAbs;
1231  UInt uiVerAbs;
1232  UInt uiHorSign = 0;
1233  UInt uiVerSign = 0;
1234  ContextModel *pCtx = m_cCUMvdSCModel.get( 0 );
[2]1235
[56]1236  if(pcCU->getSlice()->getMvdL1ZeroFlag() && eRefList == REF_PIC_LIST_1 && pcCU->getInterDir(uiAbsPartIdx)==3)
1237  {
1238    uiHorAbs=0;
1239    uiVerAbs=0;
1240  }
1241  else
1242  {
1243    m_pcTDecBinIf->decodeBin( uiHorAbs, *pCtx );
1244    m_pcTDecBinIf->decodeBin( uiVerAbs, *pCtx );
[2]1245
[56]1246    const Bool bHorAbsGr0 = uiHorAbs != 0;
1247    const Bool bVerAbsGr0 = uiVerAbs != 0;
1248    pCtx++;
1249
1250    if( bHorAbsGr0 )
1251    {
1252      m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
1253      uiHorAbs += uiSymbol;
1254    }
1255
1256    if( bVerAbsGr0 )
1257    {
1258      m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
1259      uiVerAbs += uiSymbol;
1260    }
1261
1262    if( bHorAbsGr0 )
1263    {
1264      if( 2 == uiHorAbs )
1265      {
1266        xReadEpExGolomb( uiSymbol, 1 );
1267        uiHorAbs += uiSymbol;
1268      }
1269
1270      m_pcTDecBinIf->decodeBinEP( uiHorSign );
1271    }
1272
1273    if( bVerAbsGr0 )
1274    {
1275      if( 2 == uiVerAbs )
1276      {
1277        xReadEpExGolomb( uiSymbol, 1 );
1278        uiVerAbs += uiSymbol;
1279      }
1280
1281      m_pcTDecBinIf->decodeBinEP( uiVerSign );
1282    }
1283
1284  }
[2]1285
[56]1286  const TComMv cMv( uiHorSign ? -Int( uiHorAbs ): uiHorAbs, uiVerSign ? -Int( uiVerAbs ) : uiVerAbs );
1287  pcCU->getCUMvField( eRefList )->setAllMvd( cMv, pcCU->getPartitionSize( uiAbsPartIdx ), uiAbsPartIdx, uiDepth, uiPartIdx );
[2]1288  return;
1289}
1290
1291
1292Void TDecSbac::parseTransformSubdivFlag( UInt& ruiSubdivFlag, UInt uiLog2TransformBlockSize )
1293{
1294  m_pcTDecBinIf->decodeBin( ruiSubdivFlag, m_cCUTransSubdivFlagSCModel.get( 0, 0, uiLog2TransformBlockSize ) );
[608]1295#if !H_MV_ENC_DEC_TRAC
[56]1296  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1297  DTRACE_CABAC_T( "\tparseTransformSubdivFlag()" )
1298  DTRACE_CABAC_T( "\tsymbol=" )
1299  DTRACE_CABAC_V( ruiSubdivFlag )
1300  DTRACE_CABAC_T( "\tctx=" )
1301  DTRACE_CABAC_V( uiLog2TransformBlockSize )
1302  DTRACE_CABAC_T( "\n" )
[608]1303#endif
[2]1304}
1305
[608]1306Void TDecSbac::parseQtRootCbf( UInt uiAbsPartIdx, UInt& uiQtRootCbf )
[2]1307{
1308  UInt uiSymbol;
[56]1309  const UInt uiCtx = 0;
[2]1310  m_pcTDecBinIf->decodeBin( uiSymbol , m_cCUQtRootCbfSCModel.get( 0, 0, uiCtx ) );
[608]1311#if !H_MV_ENC_DEC_TRAC
[56]1312  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1313  DTRACE_CABAC_T( "\tparseQtRootCbf()" )
1314  DTRACE_CABAC_T( "\tsymbol=" )
1315  DTRACE_CABAC_V( uiSymbol )
1316  DTRACE_CABAC_T( "\tctx=" )
1317  DTRACE_CABAC_V( uiCtx )
1318  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1319  DTRACE_CABAC_V( uiAbsPartIdx )
1320  DTRACE_CABAC_T( "\n" )
[608]1321#else
1322  DTRACE_CU( "rqt_root_cbf", uiSymbol )
1323#endif
[2]1324 
1325  uiQtRootCbf = uiSymbol;
1326}
1327
1328Void TDecSbac::parseDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
1329{
[56]1330  Int qp;
[2]1331  UInt uiDQp;
1332  Int  iDQp;
1333 
[608]1334  UInt uiSymbol;
1335
1336  xReadUnaryMaxSymbol (uiDQp,  &m_cCUDeltaQpSCModel.get( 0, 0, 0 ), 1, CU_DQP_TU_CMAX);
1337
1338  if( uiDQp >= CU_DQP_TU_CMAX)
[2]1339  {
[608]1340    xReadEpExGolomb( uiSymbol, CU_DQP_EG_k );
1341    uiDQp+=uiSymbol;
[2]1342  }
[608]1343
1344  if ( uiDQp > 0 )
[2]1345  {
[56]1346    UInt uiSign;
1347    Int qpBdOffsetY = pcCU->getSlice()->getSPS()->getQpBDOffsetY();
1348    m_pcTDecBinIf->decodeBinEP(uiSign);
[608]1349    iDQp = uiDQp;
[56]1350    if(uiSign)
[2]1351    {
1352      iDQp = -iDQp;
1353    }
[56]1354    qp = (((Int) pcCU->getRefQP( uiAbsPartIdx ) + iDQp + 52 + 2*qpBdOffsetY )%(52+qpBdOffsetY)) - qpBdOffsetY;
[2]1355  }
[608]1356  else 
1357  {
1358    qp = pcCU->getRefQP(uiAbsPartIdx);
1359  }
1360  pcCU->setQPSubParts(qp, uiAbsPartIdx, uiDepth); 
1361  pcCU->setCodedQP(qp);
[2]1362}
1363
1364Void TDecSbac::parseQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth, UInt uiDepth )
1365{
1366  UInt uiSymbol;
[608]1367  const UInt uiCtx = pcCU->getCtxQtCbf( eType, uiTrDepth );
[56]1368  m_pcTDecBinIf->decodeBin( uiSymbol , m_cCUQtCbfSCModel.get( 0, eType ? TEXT_CHROMA: eType, uiCtx ) );
[608]1369#if !H_MV_ENC_DEC_TRAC 
[56]1370  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1371  DTRACE_CABAC_T( "\tparseQtCbf()" )
1372  DTRACE_CABAC_T( "\tsymbol=" )
1373  DTRACE_CABAC_V( uiSymbol )
1374  DTRACE_CABAC_T( "\tctx=" )
1375  DTRACE_CABAC_V( uiCtx )
1376  DTRACE_CABAC_T( "\tetype=" )
1377  DTRACE_CABAC_V( eType )
1378  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1379  DTRACE_CABAC_V( uiAbsPartIdx )
1380  DTRACE_CABAC_T( "\n" )
[608]1381#endif
[2]1382 
1383  pcCU->setCbfSubParts( uiSymbol << uiTrDepth, eType, uiAbsPartIdx, uiDepth );
1384}
1385
[608]1386void TDecSbac::parseTransformSkipFlags (TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, UInt uiDepth, TextType eTType)
1387{
1388  if (pcCU->getCUTransquantBypass(uiAbsPartIdx))
1389  {
1390    return;
1391  }
1392  if(width != 4 || height != 4)
1393  {
1394    return;
1395  }
1396 
1397  UInt useTransformSkip;
1398  m_pcTDecBinIf->decodeBin( useTransformSkip , m_cTransformSkipSCModel.get( 0, eTType? TEXT_CHROMA: TEXT_LUMA, 0 ) );
1399  if(eTType!= TEXT_LUMA)
1400  {
1401    const UInt uiLog2TrafoSize = g_aucConvertToBit[pcCU->getSlice()->getSPS()->getMaxCUWidth()] + 2 - uiDepth;
1402    if(uiLog2TrafoSize == 2) 
1403    { 
1404      uiDepth --;
1405    }
1406  }
1407#if !H_MV_ENC_DEC_TRAC
1408  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1409  DTRACE_CABAC_T("\tparseTransformSkip()");
1410  DTRACE_CABAC_T( "\tsymbol=" )
1411  DTRACE_CABAC_V( useTransformSkip )
1412  DTRACE_CABAC_T( "\tAddr=" )
1413  DTRACE_CABAC_V( pcCU->getAddr() )
1414  DTRACE_CABAC_T( "\tetype=" )
1415  DTRACE_CABAC_V( eTType )
1416  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1417  DTRACE_CABAC_V( uiAbsPartIdx )
1418  DTRACE_CABAC_T( "\n" )
1419#endif
1420
1421  pcCU->setTransformSkipSubParts( useTransformSkip, eTType, uiAbsPartIdx, uiDepth);
1422}
1423
[2]1424/** Parse (X,Y) position of the last significant coefficient
1425 * \param uiPosLastX reference to X component of last coefficient
1426 * \param uiPosLastY reference to Y component of last coefficient
[56]1427 * \param width  Block width
1428 * \param height Block height
[2]1429 * \param eTType plane type / luminance or chrominance
1430 * \param uiScanIdx scan type (zig-zag, hor, ver)
[56]1431 *
[2]1432 * This method decodes the X and Y component within a block of the last significant coefficient.
1433 */
[56]1434Void TDecSbac::parseLastSignificantXY( UInt& uiPosLastX, UInt& uiPosLastY, Int width, Int height, TextType eTType, UInt uiScanIdx )
[2]1435{
1436  UInt uiLast;
[56]1437  ContextModel *pCtxX = m_cCuCtxLastX.get( 0, eTType );
1438  ContextModel *pCtxY = m_cCuCtxLastY.get( 0, eTType );
[2]1439
[608]1440  Int blkSizeOffsetX, blkSizeOffsetY, shiftX, shiftY;
1441  blkSizeOffsetX = eTType ? 0: (g_aucConvertToBit[ width ] *3 + ((g_aucConvertToBit[ width ] +1)>>2));
1442  blkSizeOffsetY = eTType ? 0: (g_aucConvertToBit[ height ]*3 + ((g_aucConvertToBit[ height ]+1)>>2));
1443  shiftX= eTType ? g_aucConvertToBit[ width  ] :((g_aucConvertToBit[ width  ]+3)>>2);
1444  shiftY= eTType ? g_aucConvertToBit[ height ] :((g_aucConvertToBit[ height ]+3)>>2);
[56]1445  // posX
1446  for( uiPosLastX = 0; uiPosLastX < g_uiGroupIdx[ width - 1 ]; uiPosLastX++ )
[2]1447  {
[608]1448    m_pcTDecBinIf->decodeBin( uiLast, *( pCtxX + blkSizeOffsetX + (uiPosLastX >>shiftX) ) );
[56]1449    if( !uiLast )
1450    {
[2]1451      break;
1452    }
1453  }
1454
[56]1455  // posY
1456  for( uiPosLastY = 0; uiPosLastY < g_uiGroupIdx[ height - 1 ]; uiPosLastY++ )
[2]1457  {
[608]1458    m_pcTDecBinIf->decodeBin( uiLast, *( pCtxY + blkSizeOffsetY + (uiPosLastY >>shiftY)) );
[56]1459    if( !uiLast )
1460    {
[2]1461      break;
1462    }
1463  }
[56]1464  if ( uiPosLastX > 3 )
1465  {
1466    UInt uiTemp  = 0;
1467    UInt uiCount = ( uiPosLastX - 2 ) >> 1;
1468    for ( Int i = uiCount - 1; i >= 0; i-- )
1469    {
1470      m_pcTDecBinIf->decodeBinEP( uiLast );
1471      uiTemp += uiLast << i;
1472    }
1473    uiPosLastX = g_uiMinInGroup[ uiPosLastX ] + uiTemp;
1474  }
1475  if ( uiPosLastY > 3 )
1476  {
1477    UInt uiTemp  = 0;
1478    UInt uiCount = ( uiPosLastY - 2 ) >> 1;
1479    for ( Int i = uiCount - 1; i >= 0; i-- )
1480    {
1481      m_pcTDecBinIf->decodeBinEP( uiLast );
1482      uiTemp += uiLast << i;
1483    }
1484    uiPosLastY = g_uiMinInGroup[ uiPosLastY ] + uiTemp;
1485  }
1486 
[2]1487  if( uiScanIdx == SCAN_VER )
1488  {
1489    swap( uiPosLastX, uiPosLastY );
1490  }
1491}
1492
1493Void TDecSbac::parseCoeffNxN( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
1494{
[608]1495#if !H_MV_ENC_DEC_TRAC
[56]1496  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1497  DTRACE_CABAC_T( "\tparseCoeffNxN()\teType=" )
1498  DTRACE_CABAC_V( eTType )
1499  DTRACE_CABAC_T( "\twidth=" )
1500  DTRACE_CABAC_V( uiWidth )
1501  DTRACE_CABAC_T( "\theight=" )
1502  DTRACE_CABAC_V( uiHeight )
1503  DTRACE_CABAC_T( "\tdepth=" )
1504  DTRACE_CABAC_V( uiDepth )
1505  DTRACE_CABAC_T( "\tabspartidx=" )
1506  DTRACE_CABAC_V( uiAbsPartIdx )
1507  DTRACE_CABAC_T( "\ttoCU-X=" )
1508  DTRACE_CABAC_V( pcCU->getCUPelX() )
1509  DTRACE_CABAC_T( "\ttoCU-Y=" )
1510  DTRACE_CABAC_V( pcCU->getCUPelY() )
1511  DTRACE_CABAC_T( "\tCU-addr=" )
1512  DTRACE_CABAC_V(  pcCU->getAddr() )
1513  DTRACE_CABAC_T( "\tinCU-X=" )
1514  DTRACE_CABAC_V( g_auiRasterToPelX[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1515  DTRACE_CABAC_T( "\tinCU-Y=" )
1516  DTRACE_CABAC_V( g_auiRasterToPelY[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1517  DTRACE_CABAC_T( "\tpredmode=" )
1518  DTRACE_CABAC_V(  pcCU->getPredictionMode( uiAbsPartIdx ) )
1519  DTRACE_CABAC_T( "\n" )
[608]1520#endif
[2]1521 
1522  if( uiWidth > pcCU->getSlice()->getSPS()->getMaxTrSize() )
1523  {
1524    uiWidth  = pcCU->getSlice()->getSPS()->getMaxTrSize();
1525    uiHeight = pcCU->getSlice()->getSPS()->getMaxTrSize();
1526  }
[608]1527  if(pcCU->getSlice()->getPPS()->getUseTransformSkip())
1528  {
1529    parseTransformSkipFlags( pcCU, uiAbsPartIdx, uiWidth, uiHeight, uiDepth, eTType);
1530  }
1531
[2]1532  eTType = eTType == TEXT_LUMA ? TEXT_LUMA : ( eTType == TEXT_NONE ? TEXT_NONE : TEXT_CHROMA );
1533 
1534  //----- parse significance map -----
1535  const UInt  uiLog2BlockSize   = g_aucConvertToBit[ uiWidth ] + 2;
[56]1536  const UInt  uiMaxNumCoeff     = uiWidth * uiHeight;
[2]1537  const UInt  uiMaxNumCoeffM1   = uiMaxNumCoeff - 1;
[56]1538  UInt uiScanIdx = pcCU->getCoefScanIdx(uiAbsPartIdx, uiWidth, eTType==TEXT_LUMA, pcCU->isIntra(uiAbsPartIdx));
[2]1539 
[56]1540  //===== decode last significant =====
1541  UInt uiPosLastX, uiPosLastY;
1542  parseLastSignificantXY( uiPosLastX, uiPosLastY, uiWidth, uiHeight, eTType, uiScanIdx );
1543  UInt uiBlkPosLast      = uiPosLastX + (uiPosLastY<<uiLog2BlockSize);
1544  pcCoef[ uiBlkPosLast ] = 1;
[2]1545
[56]1546  //===== decode significance flags =====
[608]1547  UInt uiScanPosLast;
1548  const UInt *scan   = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize-1 ];
[56]1549  for( uiScanPosLast = 0; uiScanPosLast < uiMaxNumCoeffM1; uiScanPosLast++ )
1550  {
1551    UInt uiBlkPos = scan[ uiScanPosLast ];
1552    if( uiBlkPosLast == uiBlkPos )
[2]1553    {
[56]1554      break;
[2]1555    }
[56]1556  }
[2]1557
[56]1558  ContextModel * const baseCoeffGroupCtx = m_cCUSigCoeffGroupSCModel.get( 0, eTType );
1559  ContextModel * const baseCtx = (eTType==TEXT_LUMA) ? m_cCUSigSCModel.get( 0, 0 ) : m_cCUSigSCModel.get( 0, 0 ) + NUM_SIG_FLAG_CTX_LUMA;
1560
1561  const Int  iLastScanSet      = uiScanPosLast >> LOG2_SCAN_SET_SIZE;
[608]1562  UInt c1 = 1;
[56]1563  UInt uiGoRiceParam           = 0;
1564
1565  Bool beValid; 
[608]1566  if (pcCU->getCUTransquantBypass(uiAbsPartIdx))
[2]1567  {
[56]1568    beValid = false;
1569  }
1570  else 
1571  {
1572    beValid = pcCU->getSlice()->getPPS()->getSignHideFlag() > 0;
1573  }
1574  UInt absSum = 0;
1575
1576  UInt uiSigCoeffGroupFlag[ MLS_GRP_NUM ];
1577  ::memset( uiSigCoeffGroupFlag, 0, sizeof(UInt) * MLS_GRP_NUM );
1578  const UInt uiNumBlkSide = uiWidth >> (MLS_CG_SIZE >> 1);
1579  const UInt * scanCG;
1580  {
1581    scanCG = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize > 3 ? uiLog2BlockSize-2-1 : 0  ];   
1582    if( uiLog2BlockSize == 3 )
[2]1583    {
[56]1584      scanCG = g_sigLastScan8x8[ uiScanIdx ];
[2]1585    }
[56]1586    else if( uiLog2BlockSize == 5 )
1587    {
1588      scanCG = g_sigLastScanCG32x32;
1589    }
[2]1590  }
[56]1591  Int  iScanPosSig             = (Int) uiScanPosLast;
1592  for( Int iSubSet = iLastScanSet; iSubSet >= 0; iSubSet-- )
1593  {
1594    Int  iSubPos     = iSubSet << LOG2_SCAN_SET_SIZE;
1595    uiGoRiceParam    = 0;
1596    Int numNonZero = 0;
1597   
1598    Int lastNZPosInCG = -1, firstNZPosInCG = SCAN_SET_SIZE;
1599
1600    Int pos[SCAN_SET_SIZE];
1601    if( iScanPosSig == (Int) uiScanPosLast )
1602    {
1603      lastNZPosInCG  = iScanPosSig;
1604      firstNZPosInCG = iScanPosSig;
1605      iScanPosSig--;
1606      pos[ numNonZero ] = uiBlkPosLast;
1607      numNonZero = 1;
1608    }
[2]1609
[608]1610    // decode significant_coeffgroup_flag
1611    Int iCGBlkPos = scanCG[ iSubSet ];
1612    Int iCGPosY   = iCGBlkPos / uiNumBlkSide;
1613    Int iCGPosX   = iCGBlkPos - (iCGPosY * uiNumBlkSide);
1614    if( iSubSet == iLastScanSet || iSubSet == 0)
1615    {
1616      uiSigCoeffGroupFlag[ iCGBlkPos ] = 1;
1617    }
1618    else
1619    {
1620      UInt uiSigCoeffGroup;
1621      UInt uiCtxSig  = TComTrQuant::getSigCoeffGroupCtxInc( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1622      m_pcTDecBinIf->decodeBin( uiSigCoeffGroup, baseCoeffGroupCtx[ uiCtxSig ] );
1623      uiSigCoeffGroupFlag[ iCGBlkPos ] = uiSigCoeffGroup;
1624    }
[56]1625
[608]1626    // decode significant_coeff_flag
1627    Int patternSigCtx = TComTrQuant::calcPatternSigCtx( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1628    UInt uiBlkPos, uiPosY, uiPosX, uiSig, uiCtxSig;
1629    for( ; iScanPosSig >= iSubPos; iScanPosSig-- )
1630    {
1631      uiBlkPos  = scan[ iScanPosSig ];
1632      uiPosY    = uiBlkPos >> uiLog2BlockSize;
1633      uiPosX    = uiBlkPos - ( uiPosY << uiLog2BlockSize );
1634      uiSig     = 0;
1635     
1636      if( uiSigCoeffGroupFlag[ iCGBlkPos ] )
[56]1637      {
[608]1638        if( iScanPosSig > iSubPos || iSubSet == 0  || numNonZero )
[2]1639        {
[608]1640          uiCtxSig  = TComTrQuant::getSigCtxInc( patternSigCtx, uiScanIdx, uiPosX, uiPosY, uiLog2BlockSize, eTType );
1641          m_pcTDecBinIf->decodeBin( uiSig, baseCtx[ uiCtxSig ] );
[2]1642        }
[608]1643        else
[2]1644        {
[608]1645          uiSig = 1;
[2]1646        }
[56]1647      }
[608]1648      pcCoef[ uiBlkPos ] = uiSig;
1649      if( uiSig )
1650      {
1651        pos[ numNonZero ] = uiBlkPos;
1652        numNonZero ++;
1653        if( lastNZPosInCG == -1 )
1654        {
1655          lastNZPosInCG = iScanPosSig;
1656        }
1657        firstNZPosInCG = iScanPosSig;
1658      }
1659    }
[2]1660   
[56]1661    if( numNonZero )
[2]1662    {
[608]1663      Bool signHidden = ( lastNZPosInCG - firstNZPosInCG >= SBH_THRESHOLD );
[56]1664      absSum = 0;
1665      UInt uiCtxSet    = (iSubSet > 0 && eTType==TEXT_LUMA) ? 2 : 0;
1666      UInt uiBin;
[608]1667      if( c1 == 0 )
[2]1668      {
[56]1669        uiCtxSet++;
1670      }
[608]1671      c1 = 1;
[56]1672      ContextModel *baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUOneSCModel.get( 0, 0 ) + 4 * uiCtxSet : m_cCUOneSCModel.get( 0, 0 ) + NUM_ONE_FLAG_CTX_LUMA + 4 * uiCtxSet;
1673      Int absCoeff[SCAN_SET_SIZE];
1674
1675      for ( Int i = 0; i < numNonZero; i++) absCoeff[i] = 1;   
1676      Int numC1Flag = min(numNonZero, C1FLAG_NUMBER);
1677      Int firstC2FlagIdx = -1;
1678
1679      for( Int idx = 0; idx < numC1Flag; idx++ )
1680      {
1681        m_pcTDecBinIf->decodeBin( uiBin, baseCtxMod[c1] );
1682        if( uiBin == 1 )
[2]1683        {
[56]1684          c1 = 0;
1685          if (firstC2FlagIdx == -1)
1686          {
1687            firstC2FlagIdx = idx;
1688          }
1689        }
1690        else if( (c1 < 3) && (c1 > 0) )
1691        {
[2]1692          c1++;
1693        }
[56]1694        absCoeff[ idx ] = uiBin + 1;
1695      }
1696     
1697      if (c1 == 0)
1698      {
1699        baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUAbsSCModel.get( 0, 0 ) + uiCtxSet : m_cCUAbsSCModel.get( 0, 0 ) + NUM_ABS_FLAG_CTX_LUMA + uiCtxSet;
1700        if ( firstC2FlagIdx != -1)
[2]1701        {
[56]1702          m_pcTDecBinIf->decodeBin( uiBin, baseCtxMod[0] ); 
1703          absCoeff[ firstC2FlagIdx ] = uiBin + 2;
[2]1704        }
1705      }
[56]1706
1707      UInt coeffSigns;
1708      if ( signHidden && beValid )
1709      {
1710        m_pcTDecBinIf->decodeBinsEP( coeffSigns, numNonZero-1 );
1711        coeffSigns <<= 32 - (numNonZero-1);
1712      }
1713      else
1714      {
1715        m_pcTDecBinIf->decodeBinsEP( coeffSigns, numNonZero );
1716        coeffSigns <<= 32 - numNonZero;
1717      }
[2]1718     
[56]1719      Int iFirstCoeff2 = 1;   
1720      if (c1 == 0 || numNonZero > C1FLAG_NUMBER)
[2]1721      {
[56]1722        for( Int idx = 0; idx < numNonZero; idx++ )
[2]1723        {
[56]1724          UInt baseLevel  = (idx < C1FLAG_NUMBER)? (2 + iFirstCoeff2) : 1;
[2]1725
[56]1726          if( absCoeff[ idx ] == baseLevel)
[2]1727          {
[56]1728            UInt uiLevel;
[608]1729            xReadCoefRemainExGolomb( uiLevel, uiGoRiceParam );
[56]1730            absCoeff[ idx ] = uiLevel + baseLevel;
[608]1731            if(absCoeff[idx]>3*(1<<uiGoRiceParam))
1732            {
1733              uiGoRiceParam = min<UInt>(uiGoRiceParam+ 1, 4);
1734            }
[2]1735          }
[56]1736
1737          if(absCoeff[ idx ] >= 2) 
[2]1738          {
[56]1739            iFirstCoeff2 = 0;
[2]1740          }
1741        }
1742      }
[56]1743
1744      for( Int idx = 0; idx < numNonZero; idx++ )
[2]1745      {
[56]1746        Int blkPos = pos[ idx ];
1747        // Signs applied later.
1748        pcCoef[ blkPos ] = absCoeff[ idx ];
1749        absSum += absCoeff[ idx ];
1750
1751        if ( idx == numNonZero-1 && signHidden && beValid )
1752        {
1753          // Infer sign of 1st element.
1754          if (absSum&0x1)
[608]1755          {
[56]1756            pcCoef[ blkPos ] = -pcCoef[ blkPos ];
[608]1757          }
[56]1758        }
1759        else
1760        {
1761          Int sign = static_cast<Int>( coeffSigns ) >> 31;
1762          pcCoef[ blkPos ] = ( pcCoef[ blkPos ] ^ sign ) - sign;
1763          coeffSigns <<= 1;
1764        }
[2]1765      }
1766    }
1767  }
1768 
1769  return;
1770}
1771
[608]1772
[443]1773Void TDecSbac::parseSaoMaxUvlc ( UInt& val, UInt maxSymbol )
1774{
[608]1775  if (maxSymbol == 0)
[2]1776  {
[608]1777    val = 0;
[2]1778    return;
1779  }
[56]1780
[608]1781  UInt code;
[2]1782  Int  i;
[608]1783  m_pcTDecBinIf->decodeBinEP( code );
1784  if ( code == 0 )
[2]1785  {
[608]1786    val = 0;
[2]1787    return;
1788  }
[56]1789
[2]1790  i=1;
1791  while (1)
1792  {
[608]1793    m_pcTDecBinIf->decodeBinEP( code );
1794    if ( code == 0 )
1795    {
1796      break;
1797    }
[2]1798    i++;
[608]1799    if (i == maxSymbol) 
[56]1800    {
[608]1801      break;
[56]1802    }
1803  }
[608]1804
1805  val = i;
[2]1806}
[608]1807Void TDecSbac::parseSaoUflc (UInt uiLength, UInt&  riVal)
[56]1808{
[608]1809  m_pcTDecBinIf->decodeBinsEP ( riVal, uiLength );
[56]1810}
[608]1811Void TDecSbac::parseSaoMerge (UInt&  ruiVal)
[2]1812{
1813  UInt uiCode;
[608]1814  m_pcTDecBinIf->decodeBin( uiCode, m_cSaoMergeSCModel.get( 0, 0, 0 ) );
[56]1815  ruiVal = (Int)uiCode;
1816}
1817Void TDecSbac::parseSaoTypeIdx (UInt&  ruiVal)
1818{
1819  UInt uiCode;
1820  m_pcTDecBinIf->decodeBin( uiCode, m_cSaoTypeIdxSCModel.get( 0, 0, 0 ) );
[608]1821  if (uiCode == 0) 
[2]1822  {
1823    ruiVal = 0;
1824  }
[608]1825  else
[2]1826  {
[608]1827    m_pcTDecBinIf->decodeBinEP( uiCode ); 
1828    if (uiCode == 0)
1829    {
[872]1830      ruiVal = 1;
[608]1831    }
1832    else
1833    {
[872]1834      ruiVal = 2;
[608]1835    }
[2]1836  }
1837}
1838
[872]1839Void TDecSbac::parseSaoSign(UInt& val)
[2]1840{
[872]1841  m_pcTDecBinIf->decodeBinEP ( val ); 
[56]1842}
[608]1843
[872]1844Void TDecSbac::parseSAOBlkParam (SAOBlkParam& saoBlkParam
1845                                , Bool* sliceEnabled
1846                                , Bool leftMergeAvail
1847                                , Bool aboveMergeAvail
1848                                )
[56]1849{
1850  UInt uiSymbol;
[2]1851
[872]1852  Bool isLeftMerge = false;
1853  Bool isAboveMerge= false;
1854
1855  if(leftMergeAvail)
[608]1856  {
[872]1857    parseSaoMerge(uiSymbol); //sao_merge_left_flag
1858    isLeftMerge = (uiSymbol?true:false);
[608]1859  }
[872]1860
1861  if( aboveMergeAvail && !isLeftMerge)
[608]1862  {
[872]1863    parseSaoMerge(uiSymbol); //sao_merge_up_flag
1864    isAboveMerge = (uiSymbol?true:false);
[608]1865  }
1866
[872]1867  if(isLeftMerge || isAboveMerge) //merge mode
[56]1868      {
[872]1869    saoBlkParam[SAO_Y].modeIdc = saoBlkParam[SAO_Cb].modeIdc = saoBlkParam[SAO_Cr].modeIdc = SAO_MODE_MERGE;
1870    saoBlkParam[SAO_Y].typeIdc = saoBlkParam[SAO_Cb].typeIdc = saoBlkParam[SAO_Cr].typeIdc = (isLeftMerge)?SAO_MERGE_LEFT:SAO_MERGE_ABOVE;
[56]1871      }   
[872]1872  else //new or off mode
[608]1873      {
[872]1874    for(Int compIdx=0; compIdx < NUM_SAO_COMPONENTS; compIdx++)
[608]1875        {
[872]1876      SAOOffset& ctbParam = saoBlkParam[compIdx];
1877
1878      if(!sliceEnabled[compIdx])
[608]1879          {
[872]1880        //off
1881        ctbParam.modeIdc = SAO_MODE_OFF;
1882        continue;
[608]1883        }
[872]1884
1885      //type
1886      if(compIdx == SAO_Y || compIdx == SAO_Cb)
[56]1887    {
[872]1888        parseSaoTypeIdx(uiSymbol); //sao_type_idx_luma or sao_type_idx_chroma
1889
1890        assert(uiSymbol ==0 || uiSymbol ==1 || uiSymbol ==2);
1891
1892        if(uiSymbol ==0) //OFF
[608]1893     {
[872]1894          ctbParam.modeIdc = SAO_MODE_OFF;
[608]1895   }
[872]1896        else if(uiSymbol == 1) //BO
1897        {
1898          ctbParam.modeIdc = SAO_MODE_NEW;
1899          ctbParam.typeIdc = SAO_TYPE_START_BO;
[56]1900  }
[872]1901        else //2, EO
[56]1902  {
[872]1903          ctbParam.modeIdc = SAO_MODE_NEW;
1904          ctbParam.typeIdc = SAO_TYPE_START_EO;
1905        }
1906
[56]1907  }
[872]1908      else //Cr, follow Cb SAO type
1909      {
1910        ctbParam.modeIdc = saoBlkParam[SAO_Cb].modeIdc;
1911        ctbParam.typeIdc = saoBlkParam[SAO_Cb].typeIdc;
[56]1912}
[2]1913
[872]1914      if(ctbParam.modeIdc == SAO_MODE_NEW)
[56]1915{
[872]1916        Int offset[4];
1917        for(Int i=0; i< 4; i++)
[2]1918  {
[872]1919          parseSaoMaxUvlc(uiSymbol,  g_saoMaxOffsetQVal[compIdx] ); //sao_offset_abs
1920          offset[i] = (Int)uiSymbol;
1921  }
[56]1922
[872]1923        if(ctbParam.typeIdc == SAO_TYPE_START_BO)
[608]1924  {
[872]1925          for(Int i=0; i< 4; i++)
[56]1926    {
[872]1927            if(offset[i] != 0)
[608]1928    {
[872]1929              parseSaoSign(uiSymbol); //sao_offset_sign
1930              if(uiSymbol)
[56]1931      {
[872]1932                offset[i] = -offset[i];
[56]1933      }
[608]1934    }
1935  }
[872]1936          parseSaoUflc(NUM_SAO_BO_CLASSES_LOG2, uiSymbol ); //sao_band_position
1937          ctbParam.typeAuxInfo = uiSymbol;
[608]1938
[872]1939          for(Int i=0; i<4; i++)
[608]1940      {
[872]1941            ctbParam.offset[(ctbParam.typeAuxInfo+i)%MAX_NUM_SAO_CLASSES] = offset[i];
[608]1942      }
[56]1943
1944        }
[872]1945        else //EO
[56]1946        {
[872]1947          ctbParam.typeAuxInfo = 0;
1948
1949          if(compIdx == SAO_Y || compIdx == SAO_Cb)
[56]1950        {
[872]1951            parseSaoUflc(NUM_SAO_EO_TYPES_LOG2, uiSymbol ); //sao_eo_class_luma or sao_eo_class_chroma
1952            ctbParam.typeIdc += uiSymbol;
[56]1953        }
1954        else
1955        {
[872]1956            ctbParam.typeIdc = saoBlkParam[SAO_Cb].typeIdc;
[56]1957        }
[872]1958          ctbParam.offset[SAO_CLASS_EO_FULL_VALLEY] = offset[0];
1959          ctbParam.offset[SAO_CLASS_EO_HALF_VALLEY] = offset[1];
1960          ctbParam.offset[SAO_CLASS_EO_PLAIN      ] = 0;
1961          ctbParam.offset[SAO_CLASS_EO_HALF_PEAK  ] = -offset[2];
1962          ctbParam.offset[SAO_CLASS_EO_FULL_PEAK  ] = -offset[3];
[56]1963      }
1964      }
1965    }
[2]1966  }
[56]1967}
[608]1968
[56]1969/**
1970 - Initialize our contexts from the nominated source.
1971 .
1972 \param pSrc Contexts to be copied.
1973 */
1974Void TDecSbac::xCopyContextsFrom( TDecSbac* pSrc )
1975{
1976  memcpy(m_contextModels, pSrc->m_contextModels, m_numContextModels*sizeof(m_contextModels[0]));
1977}
[2]1978
[56]1979Void TDecSbac::xCopyFrom( TDecSbac* pSrc )
1980{
1981  m_pcTDecBinIf->copyState( pSrc->m_pcTDecBinIf );
[2]1982
[56]1983  m_uiLastQp           = pSrc->m_uiLastQp;
1984  xCopyContextsFrom( pSrc );
1985
1986}
1987
1988Void TDecSbac::load ( TDecSbac* pScr )
1989{
1990  xCopyFrom(pScr);
1991}
1992
1993Void TDecSbac::loadContexts ( TDecSbac* pScr )
1994{
1995  xCopyContextsFrom(pScr);
1996}
1997
[608]1998#if H_3D_ARP
1999Void TDecSbac::parseARPW( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
[56]2000{
[608]2001  UInt uiMaxW = pcCU->getSlice()->getARPStepNum() - 1;
2002  UInt uiW = 0;
2003  UInt uiOffset = pcCU->getCTXARPWFlag(uiAbsPartIdx);
2004  UInt uiCode = 0;
[56]2005
[608]2006  assert ( uiMaxW > 0 );
[56]2007
[608]2008  m_pcTDecBinIf->decodeBin( uiCode , m_cCUPUARPWSCModel.get( 0, 0, 0 + uiOffset ) );
[2]2009
[608]2010  uiW = uiCode;
2011  if( 1 == uiW )   
[56]2012  {
[833]2013    m_pcTDecBinIf->decodeBin( uiCode , m_cCUPUARPWSCModel.get( 0, 0, 2 ) );
[608]2014    uiW += ( 1 == uiCode ? 1 : 0 );
[56]2015  }
[608]2016#if H_MV_ENC_DEC_TRAC
2017  DTRACE_CU("iv_res_pred_weight_idx", uiW )
2018#endif
2019  pcCU->setARPWSubParts( ( UChar )( uiW ) , uiAbsPartIdx, uiDepth ); 
[2]2020}
2021#endif
[56]2022
[608]2023#if H_3D_IC
2024/** parse illumination compensation flag
2025 * \param pcCU
2026 * \param uiAbsPartIdx
2027 * \param uiDepth
2028 * \returns Void
2029 */
2030Void TDecSbac::parseICFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2031{ 
2032  UInt uiSymbol = 0;
[833]2033  m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUICFlagSCModel.get( 0, 0, 0 ) );
[608]2034#if !H_MV_ENC_DEC_TRAC
2035  DTRACE_CABAC_VL( g_nSymbolCounter++ );
2036  DTRACE_CABAC_T( "\tICFlag" );
2037  DTRACE_CABAC_T( "\tuiCtxIC: ");
2038  DTRACE_CABAC_V( uiCtxIC );
2039  DTRACE_CABAC_T( "\tuiSymbol: ");
2040  DTRACE_CABAC_V( uiSymbol );
2041  DTRACE_CABAC_T( "\n");
2042#else
2043  DTRACE_CU("ic_flag", uiSymbol)
[56]2044#endif
[608]2045 
2046  pcCU->setICFlagSubParts( uiSymbol ? true : false , uiAbsPartIdx, 0, uiDepth );
[296]2047}
2048#endif
2049
[655]2050#if H_3D_INTER_SDC
[833]2051Void TDecSbac::parseDeltaDC( TComDataCU* pcCU, UInt absPartIdx, UInt depth )
2052{
2053  if( ! ( pcCU->getSDCFlag( absPartIdx ) || ( pcCU->isIntra( absPartIdx ) && getDimType( pcCU->getLumaIntraDir( absPartIdx ) ) < DIM_NUM_TYPE ) ) )
2054  {
2055    assert( 0 );
2056  }
2057
[1039]2058  UInt symbol = 1;
2059  UInt uiNumSegments = isDimMode( pcCU->getLumaIntraDir( absPartIdx ) ) ? 2 : 1;
[833]2060
[1039]2061  if( pcCU->isIntra( absPartIdx ) && pcCU->getSDCFlag( absPartIdx ))
2062  {
[1084]2063
[950]2064    m_pcTDecBinIf->decodeBin( symbol, m_cDdcFlagSCModel.get( 0, 0, 0 ) );
[833]2065      assert( pcCU->getPartitionSize( absPartIdx ) == SIZE_2Nx2N );
2066      pcCU->setTrIdxSubParts( 0, absPartIdx, depth );
2067      pcCU->setCbfSubParts( 1, 1, 1, absPartIdx, depth );
2068    }
2069
2070  for( UInt segment = 0; segment < uiNumSegments; segment++ )
2071  {
2072    Pel valDeltaDC = 0;
2073    if( symbol )
2074    {
2075      xParseDimDeltaDC( valDeltaDC, uiNumSegments );
2076    }
2077
2078    if( pcCU->isIntra( absPartIdx ) )
2079    {
2080      UInt dir     = pcCU->getLumaIntraDir( absPartIdx );
2081
2082      if( pcCU->getSDCFlag( absPartIdx ) )
2083      {
2084        pcCU->setSDCSegmentDCOffset( valDeltaDC, segment, absPartIdx );
2085      }
2086      else
2087      {
2088        pcCU->setDimDeltaDC( getDimType( dir ), segment, absPartIdx, valDeltaDC );
2089      }
2090    }
2091    else
2092    {
2093      pcCU->setSDCSegmentDCOffset( valDeltaDC, segment, absPartIdx );
2094    }
2095  }
2096}
2097
2098Void TDecSbac::parseSDCFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2099{
2100  UInt uiSymbol = 0;
2101  UInt uiCtxSDCFlag = pcCU->getCtxSDCFlag( uiAbsPartIdx );
2102
2103  m_pcTDecBinIf->decodeBin( uiSymbol, m_cSDCFlagSCModel.get( 0, 0, uiCtxSDCFlag ) );
2104
2105  if( uiSymbol )
2106  {
2107    pcCU->setSDCFlagSubParts( true, uiAbsPartIdx, uiDepth );
2108    pcCU->setTrIdxSubParts( 0, uiAbsPartIdx, uiDepth );
2109    pcCU->setCbfSubParts( 1, 1, 1, uiAbsPartIdx, uiDepth );
2110  }
2111  else
2112  {
2113    pcCU->setSDCFlagSubParts( false, uiAbsPartIdx, uiDepth );
2114  }
2115}
[100]2116
[189]2117#endif
2118
[833]2119#if H_3D_DBBP
2120Void TDecSbac::parseDBBPFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2121{
2122  AOF( pcCU->getSlice()->getVPS()->getUseDBBP(pcCU->getSlice()->getLayerIdInVps()) );
2123  AOF( !pcCU->getSlice()->getIsDepth() );
2124 
2125  UInt uiSymbol = 0;
2126 
2127  m_pcTDecBinIf->decodeBin( uiSymbol, m_cDBBPFlagSCModel.get( 0, 0, 0 ) );
2128 
[1039]2129  PartSize ePartSize = pcCU->getPartitionSize( uiAbsPartIdx );
2130  AOF( ePartSize == SIZE_2NxN || ePartSize == SIZE_Nx2N );
2131  UInt uiPUOffset = ( g_auiPUOffset[UInt( ePartSize )] << ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() - uiDepth ) << 1 ) ) >> 4;
2132  pcCU->setDBBPFlagSubParts(uiSymbol, uiAbsPartIdx, 0, uiDepth);
2133  pcCU->setDBBPFlagSubParts(uiSymbol, uiAbsPartIdx+uiPUOffset, 1, uiDepth);
[833]2134}
2135#endif
2136
[872]2137
2138
[56]2139//! \}
Note: See TracBrowser for help on using the repository browser.