source: 3DVCSoftware/branches/HTM-13.0-dev0/source/Lib/TLibDecoder/TDecSbac.cpp @ 1345

Last change on this file since 1345 was 1131, checked in by tech, 10 years ago

Cleanup.

  • Property svn:eol-style set to native
File size: 66.3 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
[1124]704 
705  Bool    bLimQtPredFlag = pcCU->getPic()->getSlice(0)->getQtPredFlag();
[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;
[1124]758 
759  Bool    bLimQtPredFlag = pcCU->getPic()->getSlice(0)->getQtPredFlag();
[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
[1124]984    if( pcCU->getSlice()->getIntraSdcWedgeFlag() || pcCU->getSlice()->getIntraContourFlag() )
[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)] );
[1124]1098
1099      assert( uiTabIdx < g_dmmWedgeLists[ g_aucConvertToBit[ pcCU->getDMM1BasePatternWidth( pcCU->getWidth( absPartIdx ) )]].size());
1100
[608]1101      pcCU->setDmmWedgeTabIdxSubParts( uiTabIdx, dimType, absPartIdx, depth );
1102    } break;
1103  case( DMM4_IDX ): break;
1104#endif
1105  default: break;
1106  }
[443]1107
[608]1108  pcCU->setLumaIntraDirSubParts( (UChar)dir, absPartIdx, depth );
1109}
[443]1110
[608]1111Void TDecSbac::parseIntraDepthMode( TComDataCU* pcCU, UInt absPartIdx, UInt depth )
[2]1112{
[833]1113  UInt uiSymbol, uiIsDimMode;
1114
1115  if( ( pcCU->getSlice()->getSPS()->getMaxCUWidth() >> pcCU->getDepth( absPartIdx ) ) < 64 ) //DMM and HEVC intra modes are both allowed
1116  {
[950]1117    m_pcTDecBinIf->decodeBin( uiSymbol, m_cAngleFlagSCModel.get( 0, 0, 0 ) );
[833]1118  }
1119  else
1120  {
1121    uiSymbol = 1;
1122  }
1123  uiIsDimMode = uiSymbol ? 0 : 1;
1124  pcCU->setLumaIntraDirSubParts( 0, absPartIdx, depth );
1125
1126  //decode DMM index
1127  if( uiIsDimMode )
1128  {
[1124]1129    if( pcCU->getSlice()->getIntraSdcWedgeFlag() && pcCU->getSlice()->getIntraContourFlag() )
[1039]1130    {
1131      m_pcTDecBinIf->decodeBin( uiSymbol, m_cDepthIntraModeSCModel.get( 0, 0, 0 ) );
1132      if( !uiSymbol )
1133      {
1134        pcCU->setLumaIntraDirSubParts( DIM_OFFSET, absPartIdx, depth );
1135      }
1136      else
1137      {
1138        pcCU->setLumaIntraDirSubParts( ( 1+ DIM_OFFSET ), absPartIdx, depth );
1139      }
1140    }
[1124]1141    else if ( pcCU->getSlice()->getIntraSdcWedgeFlag() )
[1039]1142    {
1143      pcCU->setLumaIntraDirSubParts( DIM_OFFSET, absPartIdx, depth );
1144    }
[1124]1145    else if( pcCU->getSlice()->getIntraContourFlag() )
[1039]1146    {
1147      pcCU->setLumaIntraDirSubParts( ( 1+ DIM_OFFSET ), absPartIdx, depth );
1148    }
[833]1149  }
1150}
[2]1151#endif
1152
[608]1153Void TDecSbac::parseInterDir( TComDataCU* pcCU, UInt& ruiInterDir, UInt uiAbsPartIdx )
[2]1154{
1155  UInt uiSymbol;
[608]1156  const UInt uiCtx = pcCU->getCtxInterDir( uiAbsPartIdx );
1157  ContextModel *pCtx = m_cCUInterDirSCModel.get( 0 );
1158  uiSymbol = 0;
1159  if (pcCU->getPartitionSize(uiAbsPartIdx) == SIZE_2Nx2N || pcCU->getHeight(uiAbsPartIdx) != 8 )
[2]1160  {
[608]1161    m_pcTDecBinIf->decodeBin( uiSymbol, *( pCtx + uiCtx ) );
[2]1162  }
1163
[56]1164  if( uiSymbol )
[2]1165  {
1166    uiSymbol = 2;
1167  }
[608]1168  else
1169  {
1170    m_pcTDecBinIf->decodeBin( uiSymbol, *( pCtx + 4 ) );
1171    assert(uiSymbol == 0 || uiSymbol == 1);
1172  }
[56]1173
[2]1174  uiSymbol++;
1175  ruiInterDir = uiSymbol;
[608]1176#if H_MV_ENC_DEC_TRAC
1177    DTRACE_PU("inter_pred_idc", ruiInterDir - 1 )   
1178#endif
1179
[2]1180  return;
1181}
1182
[608]1183Void TDecSbac::parseRefFrmIdx( TComDataCU* pcCU, Int& riRefFrmIdx, RefPicList eRefList )
[2]1184{
1185  UInt uiSymbol;
1186  {
[56]1187    ContextModel *pCtx = m_cCURefPicSCModel.get( 0 );
1188    m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
[2]1189
[56]1190    if( uiSymbol )
[2]1191    {
[608]1192      UInt uiRefNum = pcCU->getSlice()->getNumRefIdx( eRefList ) - 2;
1193      pCtx++;
1194      UInt ui;
1195      for( ui = 0; ui < uiRefNum; ++ui )
1196      {
1197        if( ui == 0 )
1198        {
1199          m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
1200        }
1201        else
1202        {
1203          m_pcTDecBinIf->decodeBinEP( uiSymbol );
1204        }
1205        if( uiSymbol == 0 )
1206        {
1207          break;
1208        }
1209      }
1210      uiSymbol = ui + 1;
[2]1211    }
1212    riRefFrmIdx = uiSymbol;
1213  }
[608]1214
1215#if H_MV_ENC_DEC_TRAC
1216#if ENC_DEC_TRACE
1217  if ( eRefList == REF_PIC_LIST_0 )
1218  {
1219    DTRACE_PU("ref_idx_l0", uiSymbol)
1220  }
[2]1221  else
1222  {
[608]1223    DTRACE_PU("ref_idx_l1", uiSymbol)
[2]1224  }
[608]1225#endif
1226#endif
[2]1227  return;
1228}
1229
[56]1230Void TDecSbac::parseMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiPartIdx, UInt uiDepth, RefPicList eRefList )
[2]1231{
1232  UInt uiSymbol;
[56]1233  UInt uiHorAbs;
1234  UInt uiVerAbs;
1235  UInt uiHorSign = 0;
1236  UInt uiVerSign = 0;
1237  ContextModel *pCtx = m_cCUMvdSCModel.get( 0 );
[2]1238
[56]1239  if(pcCU->getSlice()->getMvdL1ZeroFlag() && eRefList == REF_PIC_LIST_1 && pcCU->getInterDir(uiAbsPartIdx)==3)
1240  {
1241    uiHorAbs=0;
1242    uiVerAbs=0;
1243  }
1244  else
1245  {
1246    m_pcTDecBinIf->decodeBin( uiHorAbs, *pCtx );
1247    m_pcTDecBinIf->decodeBin( uiVerAbs, *pCtx );
[2]1248
[56]1249    const Bool bHorAbsGr0 = uiHorAbs != 0;
1250    const Bool bVerAbsGr0 = uiVerAbs != 0;
1251    pCtx++;
1252
1253    if( bHorAbsGr0 )
1254    {
1255      m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
1256      uiHorAbs += uiSymbol;
1257    }
1258
1259    if( bVerAbsGr0 )
1260    {
1261      m_pcTDecBinIf->decodeBin( uiSymbol, *pCtx );
1262      uiVerAbs += uiSymbol;
1263    }
1264
1265    if( bHorAbsGr0 )
1266    {
1267      if( 2 == uiHorAbs )
1268      {
1269        xReadEpExGolomb( uiSymbol, 1 );
1270        uiHorAbs += uiSymbol;
1271      }
1272
1273      m_pcTDecBinIf->decodeBinEP( uiHorSign );
1274    }
1275
1276    if( bVerAbsGr0 )
1277    {
1278      if( 2 == uiVerAbs )
1279      {
1280        xReadEpExGolomb( uiSymbol, 1 );
1281        uiVerAbs += uiSymbol;
1282      }
1283
1284      m_pcTDecBinIf->decodeBinEP( uiVerSign );
1285    }
1286
1287  }
[2]1288
[56]1289  const TComMv cMv( uiHorSign ? -Int( uiHorAbs ): uiHorAbs, uiVerSign ? -Int( uiVerAbs ) : uiVerAbs );
1290  pcCU->getCUMvField( eRefList )->setAllMvd( cMv, pcCU->getPartitionSize( uiAbsPartIdx ), uiAbsPartIdx, uiDepth, uiPartIdx );
[2]1291  return;
1292}
1293
1294
1295Void TDecSbac::parseTransformSubdivFlag( UInt& ruiSubdivFlag, UInt uiLog2TransformBlockSize )
1296{
1297  m_pcTDecBinIf->decodeBin( ruiSubdivFlag, m_cCUTransSubdivFlagSCModel.get( 0, 0, uiLog2TransformBlockSize ) );
[608]1298#if !H_MV_ENC_DEC_TRAC
[56]1299  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1300  DTRACE_CABAC_T( "\tparseTransformSubdivFlag()" )
1301  DTRACE_CABAC_T( "\tsymbol=" )
1302  DTRACE_CABAC_V( ruiSubdivFlag )
1303  DTRACE_CABAC_T( "\tctx=" )
1304  DTRACE_CABAC_V( uiLog2TransformBlockSize )
1305  DTRACE_CABAC_T( "\n" )
[608]1306#endif
[2]1307}
1308
[608]1309Void TDecSbac::parseQtRootCbf( UInt uiAbsPartIdx, UInt& uiQtRootCbf )
[2]1310{
1311  UInt uiSymbol;
[56]1312  const UInt uiCtx = 0;
[2]1313  m_pcTDecBinIf->decodeBin( uiSymbol , m_cCUQtRootCbfSCModel.get( 0, 0, uiCtx ) );
[608]1314#if !H_MV_ENC_DEC_TRAC
[56]1315  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1316  DTRACE_CABAC_T( "\tparseQtRootCbf()" )
1317  DTRACE_CABAC_T( "\tsymbol=" )
1318  DTRACE_CABAC_V( uiSymbol )
1319  DTRACE_CABAC_T( "\tctx=" )
1320  DTRACE_CABAC_V( uiCtx )
1321  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1322  DTRACE_CABAC_V( uiAbsPartIdx )
1323  DTRACE_CABAC_T( "\n" )
[608]1324#else
1325  DTRACE_CU( "rqt_root_cbf", uiSymbol )
1326#endif
[2]1327 
1328  uiQtRootCbf = uiSymbol;
1329}
1330
1331Void TDecSbac::parseDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
1332{
[56]1333  Int qp;
[2]1334  UInt uiDQp;
1335  Int  iDQp;
1336 
[608]1337  UInt uiSymbol;
1338
1339  xReadUnaryMaxSymbol (uiDQp,  &m_cCUDeltaQpSCModel.get( 0, 0, 0 ), 1, CU_DQP_TU_CMAX);
1340
1341  if( uiDQp >= CU_DQP_TU_CMAX)
[2]1342  {
[608]1343    xReadEpExGolomb( uiSymbol, CU_DQP_EG_k );
1344    uiDQp+=uiSymbol;
[2]1345  }
[608]1346
1347  if ( uiDQp > 0 )
[2]1348  {
[56]1349    UInt uiSign;
1350    Int qpBdOffsetY = pcCU->getSlice()->getSPS()->getQpBDOffsetY();
1351    m_pcTDecBinIf->decodeBinEP(uiSign);
[608]1352    iDQp = uiDQp;
[56]1353    if(uiSign)
[2]1354    {
1355      iDQp = -iDQp;
1356    }
[56]1357    qp = (((Int) pcCU->getRefQP( uiAbsPartIdx ) + iDQp + 52 + 2*qpBdOffsetY )%(52+qpBdOffsetY)) - qpBdOffsetY;
[2]1358  }
[608]1359  else 
1360  {
1361    qp = pcCU->getRefQP(uiAbsPartIdx);
1362  }
1363  pcCU->setQPSubParts(qp, uiAbsPartIdx, uiDepth); 
1364  pcCU->setCodedQP(qp);
[2]1365}
1366
1367Void TDecSbac::parseQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth, UInt uiDepth )
1368{
1369  UInt uiSymbol;
[608]1370  const UInt uiCtx = pcCU->getCtxQtCbf( eType, uiTrDepth );
[56]1371  m_pcTDecBinIf->decodeBin( uiSymbol , m_cCUQtCbfSCModel.get( 0, eType ? TEXT_CHROMA: eType, uiCtx ) );
[608]1372#if !H_MV_ENC_DEC_TRAC 
[56]1373  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1374  DTRACE_CABAC_T( "\tparseQtCbf()" )
1375  DTRACE_CABAC_T( "\tsymbol=" )
1376  DTRACE_CABAC_V( uiSymbol )
1377  DTRACE_CABAC_T( "\tctx=" )
1378  DTRACE_CABAC_V( uiCtx )
1379  DTRACE_CABAC_T( "\tetype=" )
1380  DTRACE_CABAC_V( eType )
1381  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1382  DTRACE_CABAC_V( uiAbsPartIdx )
1383  DTRACE_CABAC_T( "\n" )
[608]1384#endif
[2]1385 
1386  pcCU->setCbfSubParts( uiSymbol << uiTrDepth, eType, uiAbsPartIdx, uiDepth );
1387}
1388
[608]1389void TDecSbac::parseTransformSkipFlags (TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, UInt uiDepth, TextType eTType)
1390{
1391  if (pcCU->getCUTransquantBypass(uiAbsPartIdx))
1392  {
1393    return;
1394  }
1395  if(width != 4 || height != 4)
1396  {
1397    return;
1398  }
1399 
1400  UInt useTransformSkip;
1401  m_pcTDecBinIf->decodeBin( useTransformSkip , m_cTransformSkipSCModel.get( 0, eTType? TEXT_CHROMA: TEXT_LUMA, 0 ) );
1402  if(eTType!= TEXT_LUMA)
1403  {
1404    const UInt uiLog2TrafoSize = g_aucConvertToBit[pcCU->getSlice()->getSPS()->getMaxCUWidth()] + 2 - uiDepth;
1405    if(uiLog2TrafoSize == 2) 
1406    { 
1407      uiDepth --;
1408    }
1409  }
1410#if !H_MV_ENC_DEC_TRAC
1411  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1412  DTRACE_CABAC_T("\tparseTransformSkip()");
1413  DTRACE_CABAC_T( "\tsymbol=" )
1414  DTRACE_CABAC_V( useTransformSkip )
1415  DTRACE_CABAC_T( "\tAddr=" )
1416  DTRACE_CABAC_V( pcCU->getAddr() )
1417  DTRACE_CABAC_T( "\tetype=" )
1418  DTRACE_CABAC_V( eTType )
1419  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1420  DTRACE_CABAC_V( uiAbsPartIdx )
1421  DTRACE_CABAC_T( "\n" )
1422#endif
1423
1424  pcCU->setTransformSkipSubParts( useTransformSkip, eTType, uiAbsPartIdx, uiDepth);
1425}
1426
[2]1427/** Parse (X,Y) position of the last significant coefficient
1428 * \param uiPosLastX reference to X component of last coefficient
1429 * \param uiPosLastY reference to Y component of last coefficient
[56]1430 * \param width  Block width
1431 * \param height Block height
[2]1432 * \param eTType plane type / luminance or chrominance
1433 * \param uiScanIdx scan type (zig-zag, hor, ver)
[56]1434 *
[2]1435 * This method decodes the X and Y component within a block of the last significant coefficient.
1436 */
[56]1437Void TDecSbac::parseLastSignificantXY( UInt& uiPosLastX, UInt& uiPosLastY, Int width, Int height, TextType eTType, UInt uiScanIdx )
[2]1438{
1439  UInt uiLast;
[56]1440  ContextModel *pCtxX = m_cCuCtxLastX.get( 0, eTType );
1441  ContextModel *pCtxY = m_cCuCtxLastY.get( 0, eTType );
[2]1442
[608]1443  Int blkSizeOffsetX, blkSizeOffsetY, shiftX, shiftY;
1444  blkSizeOffsetX = eTType ? 0: (g_aucConvertToBit[ width ] *3 + ((g_aucConvertToBit[ width ] +1)>>2));
1445  blkSizeOffsetY = eTType ? 0: (g_aucConvertToBit[ height ]*3 + ((g_aucConvertToBit[ height ]+1)>>2));
1446  shiftX= eTType ? g_aucConvertToBit[ width  ] :((g_aucConvertToBit[ width  ]+3)>>2);
1447  shiftY= eTType ? g_aucConvertToBit[ height ] :((g_aucConvertToBit[ height ]+3)>>2);
[56]1448  // posX
1449  for( uiPosLastX = 0; uiPosLastX < g_uiGroupIdx[ width - 1 ]; uiPosLastX++ )
[2]1450  {
[608]1451    m_pcTDecBinIf->decodeBin( uiLast, *( pCtxX + blkSizeOffsetX + (uiPosLastX >>shiftX) ) );
[56]1452    if( !uiLast )
1453    {
[2]1454      break;
1455    }
1456  }
1457
[56]1458  // posY
1459  for( uiPosLastY = 0; uiPosLastY < g_uiGroupIdx[ height - 1 ]; uiPosLastY++ )
[2]1460  {
[608]1461    m_pcTDecBinIf->decodeBin( uiLast, *( pCtxY + blkSizeOffsetY + (uiPosLastY >>shiftY)) );
[56]1462    if( !uiLast )
1463    {
[2]1464      break;
1465    }
1466  }
[56]1467  if ( uiPosLastX > 3 )
1468  {
1469    UInt uiTemp  = 0;
1470    UInt uiCount = ( uiPosLastX - 2 ) >> 1;
1471    for ( Int i = uiCount - 1; i >= 0; i-- )
1472    {
1473      m_pcTDecBinIf->decodeBinEP( uiLast );
1474      uiTemp += uiLast << i;
1475    }
1476    uiPosLastX = g_uiMinInGroup[ uiPosLastX ] + uiTemp;
1477  }
1478  if ( uiPosLastY > 3 )
1479  {
1480    UInt uiTemp  = 0;
1481    UInt uiCount = ( uiPosLastY - 2 ) >> 1;
1482    for ( Int i = uiCount - 1; i >= 0; i-- )
1483    {
1484      m_pcTDecBinIf->decodeBinEP( uiLast );
1485      uiTemp += uiLast << i;
1486    }
1487    uiPosLastY = g_uiMinInGroup[ uiPosLastY ] + uiTemp;
1488  }
1489 
[2]1490  if( uiScanIdx == SCAN_VER )
1491  {
1492    swap( uiPosLastX, uiPosLastY );
1493  }
1494}
1495
1496Void TDecSbac::parseCoeffNxN( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
1497{
[608]1498#if !H_MV_ENC_DEC_TRAC
[56]1499  DTRACE_CABAC_VL( g_nSymbolCounter++ )
[2]1500  DTRACE_CABAC_T( "\tparseCoeffNxN()\teType=" )
1501  DTRACE_CABAC_V( eTType )
1502  DTRACE_CABAC_T( "\twidth=" )
1503  DTRACE_CABAC_V( uiWidth )
1504  DTRACE_CABAC_T( "\theight=" )
1505  DTRACE_CABAC_V( uiHeight )
1506  DTRACE_CABAC_T( "\tdepth=" )
1507  DTRACE_CABAC_V( uiDepth )
1508  DTRACE_CABAC_T( "\tabspartidx=" )
1509  DTRACE_CABAC_V( uiAbsPartIdx )
1510  DTRACE_CABAC_T( "\ttoCU-X=" )
1511  DTRACE_CABAC_V( pcCU->getCUPelX() )
1512  DTRACE_CABAC_T( "\ttoCU-Y=" )
1513  DTRACE_CABAC_V( pcCU->getCUPelY() )
1514  DTRACE_CABAC_T( "\tCU-addr=" )
1515  DTRACE_CABAC_V(  pcCU->getAddr() )
1516  DTRACE_CABAC_T( "\tinCU-X=" )
1517  DTRACE_CABAC_V( g_auiRasterToPelX[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1518  DTRACE_CABAC_T( "\tinCU-Y=" )
1519  DTRACE_CABAC_V( g_auiRasterToPelY[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1520  DTRACE_CABAC_T( "\tpredmode=" )
1521  DTRACE_CABAC_V(  pcCU->getPredictionMode( uiAbsPartIdx ) )
1522  DTRACE_CABAC_T( "\n" )
[608]1523#endif
[2]1524 
1525  if( uiWidth > pcCU->getSlice()->getSPS()->getMaxTrSize() )
1526  {
1527    uiWidth  = pcCU->getSlice()->getSPS()->getMaxTrSize();
1528    uiHeight = pcCU->getSlice()->getSPS()->getMaxTrSize();
1529  }
[608]1530  if(pcCU->getSlice()->getPPS()->getUseTransformSkip())
1531  {
1532    parseTransformSkipFlags( pcCU, uiAbsPartIdx, uiWidth, uiHeight, uiDepth, eTType);
1533  }
1534
[2]1535  eTType = eTType == TEXT_LUMA ? TEXT_LUMA : ( eTType == TEXT_NONE ? TEXT_NONE : TEXT_CHROMA );
1536 
1537  //----- parse significance map -----
1538  const UInt  uiLog2BlockSize   = g_aucConvertToBit[ uiWidth ] + 2;
[56]1539  const UInt  uiMaxNumCoeff     = uiWidth * uiHeight;
[2]1540  const UInt  uiMaxNumCoeffM1   = uiMaxNumCoeff - 1;
[56]1541  UInt uiScanIdx = pcCU->getCoefScanIdx(uiAbsPartIdx, uiWidth, eTType==TEXT_LUMA, pcCU->isIntra(uiAbsPartIdx));
[2]1542 
[56]1543  //===== decode last significant =====
1544  UInt uiPosLastX, uiPosLastY;
1545  parseLastSignificantXY( uiPosLastX, uiPosLastY, uiWidth, uiHeight, eTType, uiScanIdx );
1546  UInt uiBlkPosLast      = uiPosLastX + (uiPosLastY<<uiLog2BlockSize);
1547  pcCoef[ uiBlkPosLast ] = 1;
[2]1548
[56]1549  //===== decode significance flags =====
[608]1550  UInt uiScanPosLast;
1551  const UInt *scan   = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize-1 ];
[56]1552  for( uiScanPosLast = 0; uiScanPosLast < uiMaxNumCoeffM1; uiScanPosLast++ )
1553  {
1554    UInt uiBlkPos = scan[ uiScanPosLast ];
1555    if( uiBlkPosLast == uiBlkPos )
[2]1556    {
[56]1557      break;
[2]1558    }
[56]1559  }
[2]1560
[56]1561  ContextModel * const baseCoeffGroupCtx = m_cCUSigCoeffGroupSCModel.get( 0, eTType );
1562  ContextModel * const baseCtx = (eTType==TEXT_LUMA) ? m_cCUSigSCModel.get( 0, 0 ) : m_cCUSigSCModel.get( 0, 0 ) + NUM_SIG_FLAG_CTX_LUMA;
1563
1564  const Int  iLastScanSet      = uiScanPosLast >> LOG2_SCAN_SET_SIZE;
[608]1565  UInt c1 = 1;
[56]1566  UInt uiGoRiceParam           = 0;
1567
1568  Bool beValid; 
[608]1569  if (pcCU->getCUTransquantBypass(uiAbsPartIdx))
[2]1570  {
[56]1571    beValid = false;
1572  }
1573  else 
1574  {
1575    beValid = pcCU->getSlice()->getPPS()->getSignHideFlag() > 0;
1576  }
1577  UInt absSum = 0;
1578
1579  UInt uiSigCoeffGroupFlag[ MLS_GRP_NUM ];
1580  ::memset( uiSigCoeffGroupFlag, 0, sizeof(UInt) * MLS_GRP_NUM );
1581  const UInt uiNumBlkSide = uiWidth >> (MLS_CG_SIZE >> 1);
1582  const UInt * scanCG;
1583  {
1584    scanCG = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize > 3 ? uiLog2BlockSize-2-1 : 0  ];   
1585    if( uiLog2BlockSize == 3 )
[2]1586    {
[56]1587      scanCG = g_sigLastScan8x8[ uiScanIdx ];
[2]1588    }
[56]1589    else if( uiLog2BlockSize == 5 )
1590    {
1591      scanCG = g_sigLastScanCG32x32;
1592    }
[2]1593  }
[56]1594  Int  iScanPosSig             = (Int) uiScanPosLast;
1595  for( Int iSubSet = iLastScanSet; iSubSet >= 0; iSubSet-- )
1596  {
1597    Int  iSubPos     = iSubSet << LOG2_SCAN_SET_SIZE;
1598    uiGoRiceParam    = 0;
1599    Int numNonZero = 0;
1600   
1601    Int lastNZPosInCG = -1, firstNZPosInCG = SCAN_SET_SIZE;
1602
1603    Int pos[SCAN_SET_SIZE];
1604    if( iScanPosSig == (Int) uiScanPosLast )
1605    {
1606      lastNZPosInCG  = iScanPosSig;
1607      firstNZPosInCG = iScanPosSig;
1608      iScanPosSig--;
1609      pos[ numNonZero ] = uiBlkPosLast;
1610      numNonZero = 1;
1611    }
[2]1612
[608]1613    // decode significant_coeffgroup_flag
1614    Int iCGBlkPos = scanCG[ iSubSet ];
1615    Int iCGPosY   = iCGBlkPos / uiNumBlkSide;
1616    Int iCGPosX   = iCGBlkPos - (iCGPosY * uiNumBlkSide);
1617    if( iSubSet == iLastScanSet || iSubSet == 0)
1618    {
1619      uiSigCoeffGroupFlag[ iCGBlkPos ] = 1;
1620    }
1621    else
1622    {
1623      UInt uiSigCoeffGroup;
1624      UInt uiCtxSig  = TComTrQuant::getSigCoeffGroupCtxInc( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1625      m_pcTDecBinIf->decodeBin( uiSigCoeffGroup, baseCoeffGroupCtx[ uiCtxSig ] );
1626      uiSigCoeffGroupFlag[ iCGBlkPos ] = uiSigCoeffGroup;
1627    }
[56]1628
[608]1629    // decode significant_coeff_flag
1630    Int patternSigCtx = TComTrQuant::calcPatternSigCtx( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1631    UInt uiBlkPos, uiPosY, uiPosX, uiSig, uiCtxSig;
1632    for( ; iScanPosSig >= iSubPos; iScanPosSig-- )
1633    {
1634      uiBlkPos  = scan[ iScanPosSig ];
1635      uiPosY    = uiBlkPos >> uiLog2BlockSize;
1636      uiPosX    = uiBlkPos - ( uiPosY << uiLog2BlockSize );
1637      uiSig     = 0;
1638     
1639      if( uiSigCoeffGroupFlag[ iCGBlkPos ] )
[56]1640      {
[608]1641        if( iScanPosSig > iSubPos || iSubSet == 0  || numNonZero )
[2]1642        {
[608]1643          uiCtxSig  = TComTrQuant::getSigCtxInc( patternSigCtx, uiScanIdx, uiPosX, uiPosY, uiLog2BlockSize, eTType );
1644          m_pcTDecBinIf->decodeBin( uiSig, baseCtx[ uiCtxSig ] );
[2]1645        }
[608]1646        else
[2]1647        {
[608]1648          uiSig = 1;
[2]1649        }
[56]1650      }
[608]1651      pcCoef[ uiBlkPos ] = uiSig;
1652      if( uiSig )
1653      {
1654        pos[ numNonZero ] = uiBlkPos;
1655        numNonZero ++;
1656        if( lastNZPosInCG == -1 )
1657        {
1658          lastNZPosInCG = iScanPosSig;
1659        }
1660        firstNZPosInCG = iScanPosSig;
1661      }
1662    }
[2]1663   
[56]1664    if( numNonZero )
[2]1665    {
[608]1666      Bool signHidden = ( lastNZPosInCG - firstNZPosInCG >= SBH_THRESHOLD );
[56]1667      absSum = 0;
1668      UInt uiCtxSet    = (iSubSet > 0 && eTType==TEXT_LUMA) ? 2 : 0;
1669      UInt uiBin;
[608]1670      if( c1 == 0 )
[2]1671      {
[56]1672        uiCtxSet++;
1673      }
[608]1674      c1 = 1;
[56]1675      ContextModel *baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUOneSCModel.get( 0, 0 ) + 4 * uiCtxSet : m_cCUOneSCModel.get( 0, 0 ) + NUM_ONE_FLAG_CTX_LUMA + 4 * uiCtxSet;
1676      Int absCoeff[SCAN_SET_SIZE];
1677
1678      for ( Int i = 0; i < numNonZero; i++) absCoeff[i] = 1;   
1679      Int numC1Flag = min(numNonZero, C1FLAG_NUMBER);
1680      Int firstC2FlagIdx = -1;
1681
1682      for( Int idx = 0; idx < numC1Flag; idx++ )
1683      {
1684        m_pcTDecBinIf->decodeBin( uiBin, baseCtxMod[c1] );
1685        if( uiBin == 1 )
[2]1686        {
[56]1687          c1 = 0;
1688          if (firstC2FlagIdx == -1)
1689          {
1690            firstC2FlagIdx = idx;
1691          }
1692        }
1693        else if( (c1 < 3) && (c1 > 0) )
1694        {
[2]1695          c1++;
1696        }
[56]1697        absCoeff[ idx ] = uiBin + 1;
1698      }
1699     
1700      if (c1 == 0)
1701      {
1702        baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUAbsSCModel.get( 0, 0 ) + uiCtxSet : m_cCUAbsSCModel.get( 0, 0 ) + NUM_ABS_FLAG_CTX_LUMA + uiCtxSet;
1703        if ( firstC2FlagIdx != -1)
[2]1704        {
[56]1705          m_pcTDecBinIf->decodeBin( uiBin, baseCtxMod[0] ); 
1706          absCoeff[ firstC2FlagIdx ] = uiBin + 2;
[2]1707        }
1708      }
[56]1709
1710      UInt coeffSigns;
1711      if ( signHidden && beValid )
1712      {
1713        m_pcTDecBinIf->decodeBinsEP( coeffSigns, numNonZero-1 );
1714        coeffSigns <<= 32 - (numNonZero-1);
1715      }
1716      else
1717      {
1718        m_pcTDecBinIf->decodeBinsEP( coeffSigns, numNonZero );
1719        coeffSigns <<= 32 - numNonZero;
1720      }
[2]1721     
[56]1722      Int iFirstCoeff2 = 1;   
1723      if (c1 == 0 || numNonZero > C1FLAG_NUMBER)
[2]1724      {
[56]1725        for( Int idx = 0; idx < numNonZero; idx++ )
[2]1726        {
[56]1727          UInt baseLevel  = (idx < C1FLAG_NUMBER)? (2 + iFirstCoeff2) : 1;
[2]1728
[56]1729          if( absCoeff[ idx ] == baseLevel)
[2]1730          {
[56]1731            UInt uiLevel;
[608]1732            xReadCoefRemainExGolomb( uiLevel, uiGoRiceParam );
[56]1733            absCoeff[ idx ] = uiLevel + baseLevel;
[608]1734            if(absCoeff[idx]>3*(1<<uiGoRiceParam))
1735            {
1736              uiGoRiceParam = min<UInt>(uiGoRiceParam+ 1, 4);
1737            }
[2]1738          }
[56]1739
1740          if(absCoeff[ idx ] >= 2) 
[2]1741          {
[56]1742            iFirstCoeff2 = 0;
[2]1743          }
1744        }
1745      }
[56]1746
1747      for( Int idx = 0; idx < numNonZero; idx++ )
[2]1748      {
[56]1749        Int blkPos = pos[ idx ];
1750        // Signs applied later.
1751        pcCoef[ blkPos ] = absCoeff[ idx ];
1752        absSum += absCoeff[ idx ];
1753
1754        if ( idx == numNonZero-1 && signHidden && beValid )
1755        {
1756          // Infer sign of 1st element.
1757          if (absSum&0x1)
[608]1758          {
[56]1759            pcCoef[ blkPos ] = -pcCoef[ blkPos ];
[608]1760          }
[56]1761        }
1762        else
1763        {
1764          Int sign = static_cast<Int>( coeffSigns ) >> 31;
1765          pcCoef[ blkPos ] = ( pcCoef[ blkPos ] ^ sign ) - sign;
1766          coeffSigns <<= 1;
1767        }
[2]1768      }
1769    }
1770  }
1771 
1772  return;
1773}
1774
[608]1775
[443]1776Void TDecSbac::parseSaoMaxUvlc ( UInt& val, UInt maxSymbol )
1777{
[608]1778  if (maxSymbol == 0)
[2]1779  {
[608]1780    val = 0;
[2]1781    return;
1782  }
[56]1783
[608]1784  UInt code;
[2]1785  Int  i;
[608]1786  m_pcTDecBinIf->decodeBinEP( code );
1787  if ( code == 0 )
[2]1788  {
[608]1789    val = 0;
[2]1790    return;
1791  }
[56]1792
[2]1793  i=1;
1794  while (1)
1795  {
[608]1796    m_pcTDecBinIf->decodeBinEP( code );
1797    if ( code == 0 )
1798    {
1799      break;
1800    }
[2]1801    i++;
[608]1802    if (i == maxSymbol) 
[56]1803    {
[608]1804      break;
[56]1805    }
1806  }
[608]1807
1808  val = i;
[2]1809}
[608]1810Void TDecSbac::parseSaoUflc (UInt uiLength, UInt&  riVal)
[56]1811{
[608]1812  m_pcTDecBinIf->decodeBinsEP ( riVal, uiLength );
[56]1813}
[608]1814Void TDecSbac::parseSaoMerge (UInt&  ruiVal)
[2]1815{
1816  UInt uiCode;
[608]1817  m_pcTDecBinIf->decodeBin( uiCode, m_cSaoMergeSCModel.get( 0, 0, 0 ) );
[56]1818  ruiVal = (Int)uiCode;
1819}
1820Void TDecSbac::parseSaoTypeIdx (UInt&  ruiVal)
1821{
1822  UInt uiCode;
1823  m_pcTDecBinIf->decodeBin( uiCode, m_cSaoTypeIdxSCModel.get( 0, 0, 0 ) );
[608]1824  if (uiCode == 0) 
[2]1825  {
1826    ruiVal = 0;
1827  }
[608]1828  else
[2]1829  {
[608]1830    m_pcTDecBinIf->decodeBinEP( uiCode ); 
1831    if (uiCode == 0)
1832    {
[872]1833      ruiVal = 1;
[608]1834    }
1835    else
1836    {
[872]1837      ruiVal = 2;
[608]1838    }
[2]1839  }
1840}
1841
[872]1842Void TDecSbac::parseSaoSign(UInt& val)
[2]1843{
[872]1844  m_pcTDecBinIf->decodeBinEP ( val ); 
[56]1845}
[608]1846
[872]1847Void TDecSbac::parseSAOBlkParam (SAOBlkParam& saoBlkParam
1848                                , Bool* sliceEnabled
1849                                , Bool leftMergeAvail
1850                                , Bool aboveMergeAvail
1851                                )
[56]1852{
1853  UInt uiSymbol;
[2]1854
[872]1855  Bool isLeftMerge = false;
1856  Bool isAboveMerge= false;
1857
1858  if(leftMergeAvail)
[608]1859  {
[872]1860    parseSaoMerge(uiSymbol); //sao_merge_left_flag
1861    isLeftMerge = (uiSymbol?true:false);
[608]1862  }
[872]1863
1864  if( aboveMergeAvail && !isLeftMerge)
[608]1865  {
[872]1866    parseSaoMerge(uiSymbol); //sao_merge_up_flag
1867    isAboveMerge = (uiSymbol?true:false);
[608]1868  }
1869
[872]1870  if(isLeftMerge || isAboveMerge) //merge mode
[56]1871      {
[872]1872    saoBlkParam[SAO_Y].modeIdc = saoBlkParam[SAO_Cb].modeIdc = saoBlkParam[SAO_Cr].modeIdc = SAO_MODE_MERGE;
1873    saoBlkParam[SAO_Y].typeIdc = saoBlkParam[SAO_Cb].typeIdc = saoBlkParam[SAO_Cr].typeIdc = (isLeftMerge)?SAO_MERGE_LEFT:SAO_MERGE_ABOVE;
[56]1874      }   
[872]1875  else //new or off mode
[608]1876      {
[872]1877    for(Int compIdx=0; compIdx < NUM_SAO_COMPONENTS; compIdx++)
[608]1878        {
[872]1879      SAOOffset& ctbParam = saoBlkParam[compIdx];
1880
1881      if(!sliceEnabled[compIdx])
[608]1882          {
[872]1883        //off
1884        ctbParam.modeIdc = SAO_MODE_OFF;
1885        continue;
[608]1886        }
[872]1887
1888      //type
1889      if(compIdx == SAO_Y || compIdx == SAO_Cb)
[56]1890    {
[872]1891        parseSaoTypeIdx(uiSymbol); //sao_type_idx_luma or sao_type_idx_chroma
1892
1893        assert(uiSymbol ==0 || uiSymbol ==1 || uiSymbol ==2);
1894
1895        if(uiSymbol ==0) //OFF
[608]1896     {
[872]1897          ctbParam.modeIdc = SAO_MODE_OFF;
[608]1898   }
[872]1899        else if(uiSymbol == 1) //BO
1900        {
1901          ctbParam.modeIdc = SAO_MODE_NEW;
1902          ctbParam.typeIdc = SAO_TYPE_START_BO;
[56]1903  }
[872]1904        else //2, EO
[56]1905  {
[872]1906          ctbParam.modeIdc = SAO_MODE_NEW;
1907          ctbParam.typeIdc = SAO_TYPE_START_EO;
1908        }
1909
[56]1910  }
[872]1911      else //Cr, follow Cb SAO type
1912      {
1913        ctbParam.modeIdc = saoBlkParam[SAO_Cb].modeIdc;
1914        ctbParam.typeIdc = saoBlkParam[SAO_Cb].typeIdc;
[56]1915}
[2]1916
[872]1917      if(ctbParam.modeIdc == SAO_MODE_NEW)
[56]1918{
[872]1919        Int offset[4];
1920        for(Int i=0; i< 4; i++)
[2]1921  {
[872]1922          parseSaoMaxUvlc(uiSymbol,  g_saoMaxOffsetQVal[compIdx] ); //sao_offset_abs
1923          offset[i] = (Int)uiSymbol;
1924  }
[56]1925
[872]1926        if(ctbParam.typeIdc == SAO_TYPE_START_BO)
[608]1927  {
[872]1928          for(Int i=0; i< 4; i++)
[56]1929    {
[872]1930            if(offset[i] != 0)
[608]1931    {
[872]1932              parseSaoSign(uiSymbol); //sao_offset_sign
1933              if(uiSymbol)
[56]1934      {
[872]1935                offset[i] = -offset[i];
[56]1936      }
[608]1937    }
1938  }
[872]1939          parseSaoUflc(NUM_SAO_BO_CLASSES_LOG2, uiSymbol ); //sao_band_position
1940          ctbParam.typeAuxInfo = uiSymbol;
[608]1941
[872]1942          for(Int i=0; i<4; i++)
[608]1943      {
[872]1944            ctbParam.offset[(ctbParam.typeAuxInfo+i)%MAX_NUM_SAO_CLASSES] = offset[i];
[608]1945      }
[56]1946
1947        }
[872]1948        else //EO
[56]1949        {
[872]1950          ctbParam.typeAuxInfo = 0;
1951
1952          if(compIdx == SAO_Y || compIdx == SAO_Cb)
[56]1953        {
[872]1954            parseSaoUflc(NUM_SAO_EO_TYPES_LOG2, uiSymbol ); //sao_eo_class_luma or sao_eo_class_chroma
1955            ctbParam.typeIdc += uiSymbol;
[56]1956        }
1957        else
1958        {
[872]1959            ctbParam.typeIdc = saoBlkParam[SAO_Cb].typeIdc;
[56]1960        }
[872]1961          ctbParam.offset[SAO_CLASS_EO_FULL_VALLEY] = offset[0];
1962          ctbParam.offset[SAO_CLASS_EO_HALF_VALLEY] = offset[1];
1963          ctbParam.offset[SAO_CLASS_EO_PLAIN      ] = 0;
1964          ctbParam.offset[SAO_CLASS_EO_HALF_PEAK  ] = -offset[2];
1965          ctbParam.offset[SAO_CLASS_EO_FULL_PEAK  ] = -offset[3];
[56]1966      }
1967      }
1968    }
[2]1969  }
[56]1970}
[608]1971
[56]1972/**
1973 - Initialize our contexts from the nominated source.
1974 .
1975 \param pSrc Contexts to be copied.
1976 */
1977Void TDecSbac::xCopyContextsFrom( TDecSbac* pSrc )
1978{
1979  memcpy(m_contextModels, pSrc->m_contextModels, m_numContextModels*sizeof(m_contextModels[0]));
1980}
[2]1981
[56]1982Void TDecSbac::xCopyFrom( TDecSbac* pSrc )
1983{
1984  m_pcTDecBinIf->copyState( pSrc->m_pcTDecBinIf );
[2]1985
[56]1986  m_uiLastQp           = pSrc->m_uiLastQp;
1987  xCopyContextsFrom( pSrc );
1988
1989}
1990
1991Void TDecSbac::load ( TDecSbac* pScr )
1992{
1993  xCopyFrom(pScr);
1994}
1995
1996Void TDecSbac::loadContexts ( TDecSbac* pScr )
1997{
1998  xCopyContextsFrom(pScr);
1999}
2000
[608]2001#if H_3D_ARP
2002Void TDecSbac::parseARPW( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
[56]2003{
[608]2004  UInt uiMaxW = pcCU->getSlice()->getARPStepNum() - 1;
2005  UInt uiW = 0;
2006  UInt uiOffset = pcCU->getCTXARPWFlag(uiAbsPartIdx);
2007  UInt uiCode = 0;
[56]2008
[608]2009  assert ( uiMaxW > 0 );
[56]2010
[608]2011  m_pcTDecBinIf->decodeBin( uiCode , m_cCUPUARPWSCModel.get( 0, 0, 0 + uiOffset ) );
[2]2012
[608]2013  uiW = uiCode;
2014  if( 1 == uiW )   
[56]2015  {
[833]2016    m_pcTDecBinIf->decodeBin( uiCode , m_cCUPUARPWSCModel.get( 0, 0, 2 ) );
[608]2017    uiW += ( 1 == uiCode ? 1 : 0 );
[56]2018  }
[608]2019#if H_MV_ENC_DEC_TRAC
2020  DTRACE_CU("iv_res_pred_weight_idx", uiW )
2021#endif
2022  pcCU->setARPWSubParts( ( UChar )( uiW ) , uiAbsPartIdx, uiDepth ); 
[2]2023}
2024#endif
[56]2025
[608]2026#if H_3D_IC
2027/** parse illumination compensation flag
2028 * \param pcCU
2029 * \param uiAbsPartIdx
2030 * \param uiDepth
2031 * \returns Void
2032 */
2033Void TDecSbac::parseICFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2034{ 
2035  UInt uiSymbol = 0;
[833]2036  m_pcTDecBinIf->decodeBin( uiSymbol, m_cCUICFlagSCModel.get( 0, 0, 0 ) );
[608]2037#if !H_MV_ENC_DEC_TRAC
2038  DTRACE_CABAC_VL( g_nSymbolCounter++ );
2039  DTRACE_CABAC_T( "\tICFlag" );
2040  DTRACE_CABAC_T( "\tuiCtxIC: ");
2041  DTRACE_CABAC_V( uiCtxIC );
2042  DTRACE_CABAC_T( "\tuiSymbol: ");
2043  DTRACE_CABAC_V( uiSymbol );
2044  DTRACE_CABAC_T( "\n");
2045#else
2046  DTRACE_CU("ic_flag", uiSymbol)
[56]2047#endif
[608]2048 
2049  pcCU->setICFlagSubParts( uiSymbol ? true : false , uiAbsPartIdx, 0, uiDepth );
[296]2050}
2051#endif
2052
[655]2053#if H_3D_INTER_SDC
[833]2054Void TDecSbac::parseDeltaDC( TComDataCU* pcCU, UInt absPartIdx, UInt depth )
2055{
2056  if( ! ( pcCU->getSDCFlag( absPartIdx ) || ( pcCU->isIntra( absPartIdx ) && getDimType( pcCU->getLumaIntraDir( absPartIdx ) ) < DIM_NUM_TYPE ) ) )
2057  {
2058    assert( 0 );
2059  }
2060
[1039]2061  UInt symbol = 1;
2062  UInt uiNumSegments = isDimMode( pcCU->getLumaIntraDir( absPartIdx ) ) ? 2 : 1;
[833]2063
[1039]2064  if( pcCU->isIntra( absPartIdx ) && pcCU->getSDCFlag( absPartIdx ))
2065  {
[1084]2066
[950]2067    m_pcTDecBinIf->decodeBin( symbol, m_cDdcFlagSCModel.get( 0, 0, 0 ) );
[833]2068      assert( pcCU->getPartitionSize( absPartIdx ) == SIZE_2Nx2N );
2069      pcCU->setTrIdxSubParts( 0, absPartIdx, depth );
2070      pcCU->setCbfSubParts( 1, 1, 1, absPartIdx, depth );
2071    }
2072
2073  for( UInt segment = 0; segment < uiNumSegments; segment++ )
2074  {
2075    Pel valDeltaDC = 0;
2076    if( symbol )
2077    {
2078      xParseDimDeltaDC( valDeltaDC, uiNumSegments );
2079    }
2080
2081    if( pcCU->isIntra( absPartIdx ) )
2082    {
2083      UInt dir     = pcCU->getLumaIntraDir( absPartIdx );
2084
2085      if( pcCU->getSDCFlag( absPartIdx ) )
2086      {
2087        pcCU->setSDCSegmentDCOffset( valDeltaDC, segment, absPartIdx );
2088      }
2089      else
2090      {
2091        pcCU->setDimDeltaDC( getDimType( dir ), segment, absPartIdx, valDeltaDC );
2092      }
2093    }
2094    else
2095    {
2096      pcCU->setSDCSegmentDCOffset( valDeltaDC, segment, absPartIdx );
2097    }
2098  }
2099}
2100
2101Void TDecSbac::parseSDCFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2102{
2103  UInt uiSymbol = 0;
2104  UInt uiCtxSDCFlag = pcCU->getCtxSDCFlag( uiAbsPartIdx );
2105
2106  m_pcTDecBinIf->decodeBin( uiSymbol, m_cSDCFlagSCModel.get( 0, 0, uiCtxSDCFlag ) );
2107
2108  if( uiSymbol )
2109  {
2110    pcCU->setSDCFlagSubParts( true, uiAbsPartIdx, uiDepth );
2111    pcCU->setTrIdxSubParts( 0, uiAbsPartIdx, uiDepth );
2112    pcCU->setCbfSubParts( 1, 1, 1, uiAbsPartIdx, uiDepth );
2113  }
2114  else
2115  {
2116    pcCU->setSDCFlagSubParts( false, uiAbsPartIdx, uiDepth );
2117  }
2118}
[100]2119
[189]2120#endif
2121
[833]2122#if H_3D_DBBP
2123Void TDecSbac::parseDBBPFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2124{
[1124]2125  AOF( pcCU->getSlice()->getDepthBasedBlkPartFlag() );
[833]2126  AOF( !pcCU->getSlice()->getIsDepth() );
2127 
2128  UInt uiSymbol = 0;
2129 
2130  m_pcTDecBinIf->decodeBin( uiSymbol, m_cDBBPFlagSCModel.get( 0, 0, 0 ) );
2131 
[1039]2132  PartSize ePartSize = pcCU->getPartitionSize( uiAbsPartIdx );
2133  AOF( ePartSize == SIZE_2NxN || ePartSize == SIZE_Nx2N );
2134  UInt uiPUOffset = ( g_auiPUOffset[UInt( ePartSize )] << ( ( pcCU->getSlice()->getSPS()->getMaxCUDepth() - uiDepth ) << 1 ) ) >> 4;
2135  pcCU->setDBBPFlagSubParts(uiSymbol, uiAbsPartIdx, 0, uiDepth);
2136  pcCU->setDBBPFlagSubParts(uiSymbol, uiAbsPartIdx+uiPUOffset, 1, uiDepth);
[833]2137}
2138#endif
2139
[872]2140
2141
[56]2142//! \}
Note: See TracBrowser for help on using the repository browser.