source: 3DVCSoftware/branches/HTM-8.2-dev3-Samsung/source/Lib/TLibEncoder/TEncSbac.cpp @ 697

Last change on this file since 697 was 697, checked in by samsung-htm, 11 years ago

Integration of F0147: DMM simplification and signalling

  • Property svn:eol-style set to native
File size: 77.0 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license. 
5 *
6 * Copyright (c) 2010-2013, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 *  * Redistributions of source code must retain the above copyright notice,
13 *    this list of conditions and the following disclaimer.
14 *  * Redistributions in binary form must reproduce the above copyright notice,
15 *    this list of conditions and the following disclaimer in the documentation
16 *    and/or other materials provided with the distribution.
17 *  * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 *    be used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34/** \file     TEncSbac.cpp
35    \brief    SBAC encoder class
36*/
37
38#include "TEncTop.h"
39#include "TEncSbac.h"
40
41#include <map>
42#include <algorithm>
43
44//! \ingroup TLibEncoder
45//! \{
46
47// ====================================================================================================================
48// Constructor / destructor / create / destroy
49// ====================================================================================================================
50
51TEncSbac::TEncSbac()
52// new structure here
53: m_pcBitIf                   ( NULL )
54, m_pcSlice                   ( NULL )
55, m_pcBinIf                   ( NULL )
56, m_uiCoeffCost               ( 0 )
57, m_numContextModels          ( 0 )
58, m_cCUSplitFlagSCModel       ( 1,             1,               NUM_SPLIT_FLAG_CTX            , m_contextModels + m_numContextModels, m_numContextModels )
59, m_cCUSkipFlagSCModel        ( 1,             1,               NUM_SKIP_FLAG_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
60, m_cCUMergeFlagExtSCModel    ( 1,             1,               NUM_MERGE_FLAG_EXT_CTX        , m_contextModels + m_numContextModels, m_numContextModels)
61, m_cCUMergeIdxExtSCModel     ( 1,             1,               NUM_MERGE_IDX_EXT_CTX         , m_contextModels + m_numContextModels, m_numContextModels)
62#if H_3D_ARP
63, m_cCUPUARPWSCModel          ( 1,             1,               NUM_ARPW_CTX                  , m_contextModels + m_numContextModels, m_numContextModels)
64#endif
65#if H_3D_IC
66, m_cCUICFlagSCModel          ( 1,             1,               NUM_IC_FLAG_CTX               , m_contextModels + m_numContextModels, m_numContextModels)
67#endif
68, m_cCUPartSizeSCModel        ( 1,             1,               NUM_PART_SIZE_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
69, m_cCUPredModeSCModel        ( 1,             1,               NUM_PRED_MODE_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
70, m_cCUIntraPredSCModel       ( 1,             1,               NUM_ADI_CTX                   , m_contextModels + m_numContextModels, m_numContextModels)
71, m_cCUChromaPredSCModel      ( 1,             1,               NUM_CHROMA_PRED_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
72, m_cCUDeltaQpSCModel         ( 1,             1,               NUM_DELTA_QP_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
73, m_cCUInterDirSCModel        ( 1,             1,               NUM_INTER_DIR_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
74, m_cCURefPicSCModel          ( 1,             1,               NUM_REF_NO_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
75, m_cCUMvdSCModel             ( 1,             1,               NUM_MV_RES_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
76, m_cCUQtCbfSCModel           ( 1,             2,               NUM_QT_CBF_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
77, m_cCUTransSubdivFlagSCModel ( 1,             1,               NUM_TRANS_SUBDIV_FLAG_CTX     , m_contextModels + m_numContextModels, m_numContextModels)
78, m_cCUQtRootCbfSCModel       ( 1,             1,               NUM_QT_ROOT_CBF_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
79, m_cCUSigCoeffGroupSCModel   ( 1,             2,               NUM_SIG_CG_FLAG_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
80, m_cCUSigSCModel             ( 1,             1,               NUM_SIG_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
81, m_cCuCtxLastX               ( 1,             2,               NUM_CTX_LAST_FLAG_XY          , m_contextModels + m_numContextModels, m_numContextModels)
82, m_cCuCtxLastY               ( 1,             2,               NUM_CTX_LAST_FLAG_XY          , m_contextModels + m_numContextModels, m_numContextModels)
83, m_cCUOneSCModel             ( 1,             1,               NUM_ONE_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
84, m_cCUAbsSCModel             ( 1,             1,               NUM_ABS_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
85, m_cMVPIdxSCModel            ( 1,             1,               NUM_MVP_IDX_CTX               , m_contextModels + m_numContextModels, m_numContextModels)
86, m_cCUAMPSCModel             ( 1,             1,               NUM_CU_AMP_CTX                , m_contextModels + m_numContextModels, m_numContextModels)
87, m_cSaoMergeSCModel          ( 1,             1,               NUM_SAO_MERGE_FLAG_CTX   , m_contextModels + m_numContextModels, m_numContextModels)
88, m_cSaoTypeIdxSCModel        ( 1,             1,               NUM_SAO_TYPE_IDX_CTX          , m_contextModels + m_numContextModels, m_numContextModels)
89, m_cTransformSkipSCModel     ( 1,             2,               NUM_TRANSFORMSKIP_FLAG_CTX    , m_contextModels + m_numContextModels, m_numContextModels)
90, m_CUTransquantBypassFlagSCModel( 1,          1,               NUM_CU_TRANSQUANT_BYPASS_FLAG_CTX, m_contextModels + m_numContextModels, m_numContextModels)
91#if H_3D_DIM
92, m_cDepthIntraModeSCModel    ( 1,             1,               NUM_DEPTH_INTRA_MODE_CTX      , m_contextModels + m_numContextModels, m_numContextModels)
93, m_cDdcFlagSCModel           ( 1,             1,               NUM_DDC_FLAG_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
94, m_cDdcDataSCModel           ( 1,             1,               NUM_DDC_DATA_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
95#if H_3D_DIM_DMM
96, m_cDmm1DataSCModel          ( 1,             1,               NUM_DMM1_DATA_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
97#if !SEC_DMM3_RBC_F0147
98, m_cDmm3DataSCModel          ( 1,             1,               NUM_DMM3_DATA_CTX             , m_contextModels + m_numContextModels, m_numContextModels)
99#endif
100#endif
101#if H_3D_DIM_RBC
102, m_cRbcDataSCModel           ( 1,             1,               NUM_RBC_DATA_CTX              , m_contextModels + m_numContextModels, m_numContextModels)
103#endif
104#if H_3D_DIM_SDC
105, m_cSDCResidualFlagSCModel   ( 1,             1,               SDC_NUM_RESIDUAL_FLAG_CTX     , m_contextModels + m_numContextModels, m_numContextModels)
106, m_cSDCResidualSCModel       ( 1,             1,               SDC_NUM_RESIDUAL_CTX          , m_contextModels + m_numContextModels, m_numContextModels)
107#endif
108#endif
109#if H_3D_INTER_SDC
110, m_cInterSDCFlagSCModel             ( 1,             1,  NUM_INTER_SDC_FLAG_CTX           , m_contextModels + m_numContextModels, m_numContextModels)
111, m_cInterSDCResidualSCModel         ( 1,             1,  NUM_INTER_SDC_RESIDUAL_CTX       , m_contextModels + m_numContextModels, m_numContextModels)
112, m_cInterSDCResidualSignFlagSCModel ( 1,             1,  NUM_INTER_SDC_SIGN_FLAG_CTX      , m_contextModels + m_numContextModels, m_numContextModels)
113#endif
114{
115  assert( m_numContextModels <= MAX_NUM_CTX_MOD );
116}
117
118TEncSbac::~TEncSbac()
119{
120}
121
122// ====================================================================================================================
123// Public member functions
124// ====================================================================================================================
125
126Void TEncSbac::resetEntropy           ()
127{
128  Int  iQp              = m_pcSlice->getSliceQp();
129  SliceType eSliceType  = m_pcSlice->getSliceType();
130 
131  Int  encCABACTableIdx = m_pcSlice->getPPS()->getEncCABACTableIdx();
132  if (!m_pcSlice->isIntra() && (encCABACTableIdx==B_SLICE || encCABACTableIdx==P_SLICE) && m_pcSlice->getPPS()->getCabacInitPresentFlag())
133  {
134    eSliceType = (SliceType) encCABACTableIdx;
135  }
136
137  m_cCUSplitFlagSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_SPLIT_FLAG );
138 
139  m_cCUSkipFlagSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SKIP_FLAG );
140  m_cCUMergeFlagExtSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_MERGE_FLAG_EXT);
141  m_cCUMergeIdxExtSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_MERGE_IDX_EXT);
142#if H_3D_ARP
143  m_cCUPUARPWSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_ARPW );
144#endif
145#if H_3D_IC
146  m_cCUICFlagSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_IC_FLAG );
147#endif
148  m_cCUPartSizeSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_PART_SIZE );
149  m_cCUAMPSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_CU_AMP_POS );
150  m_cCUPredModeSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_PRED_MODE );
151  m_cCUIntraPredSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_INTRA_PRED_MODE );
152  m_cCUChromaPredSCModel.initBuffer      ( eSliceType, iQp, (UChar*)INIT_CHROMA_PRED_MODE );
153  m_cCUInterDirSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_INTER_DIR );
154  m_cCUMvdSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_MVD );
155  m_cCURefPicSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_REF_PIC );
156  m_cCUDeltaQpSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_DQP );
157  m_cCUQtCbfSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_QT_CBF );
158  m_cCUQtRootCbfSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_QT_ROOT_CBF );
159  m_cCUSigCoeffGroupSCModel.initBuffer   ( eSliceType, iQp, (UChar*)INIT_SIG_CG_FLAG );
160  m_cCUSigSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_SIG_FLAG );
161  m_cCuCtxLastX.initBuffer               ( eSliceType, iQp, (UChar*)INIT_LAST );
162  m_cCuCtxLastY.initBuffer               ( eSliceType, iQp, (UChar*)INIT_LAST );
163  m_cCUOneSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_ONE_FLAG );
164  m_cCUAbsSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_ABS_FLAG );
165  m_cMVPIdxSCModel.initBuffer            ( eSliceType, iQp, (UChar*)INIT_MVP_IDX );
166  m_cCUTransSubdivFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
167  m_cSaoMergeSCModel.initBuffer      ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_FLAG );
168  m_cSaoTypeIdxSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SAO_TYPE_IDX );
169  m_cTransformSkipSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_TRANSFORMSKIP_FLAG );
170  m_CUTransquantBypassFlagSCModel.initBuffer( eSliceType, iQp, (UChar*)INIT_CU_TRANSQUANT_BYPASS_FLAG );
171
172#if H_3D_DIM
173  m_cDepthIntraModeSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_DEPTH_INTRA_MODE );
174  m_cDdcFlagSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_DDC_FLAG );
175  m_cDdcDataSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_DDC_DATA );
176#if H_3D_DIM_DMM
177  m_cDmm1DataSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_DMM1_DATA );
178#if !SEC_DMM3_RBC_F0147
179  m_cDmm3DataSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_DMM3_DATA );
180#endif
181#endif
182#if H_3D_DIM_RBC
183  m_cRbcDataSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_RBC_DATA );
184#endif
185#if H_3D_DIM_SDC
186  m_cSDCResidualFlagSCModel.initBuffer   ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL_FLAG );
187  m_cSDCResidualSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL );
188#endif
189#endif
190#if H_3D_INTER_SDC
191  m_cInterSDCFlagSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_INTER_SDC_FLAG );
192  m_cInterSDCResidualSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_INTER_SDC_RESIDUAL );
193  m_cInterSDCResidualSignFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_INTER_SDC_SIGN_FLAG );
194#endif
195  // new structure
196  m_uiLastQp = iQp;
197 
198  m_pcBinIf->start();
199 
200  return;
201}
202
203/** The function does the following:
204 * If current slice type is P/B then it determines the distance of initialisation type 1 and 2 from the current CABAC states and
205 * stores the index of the closest table.  This index is used for the next P/B slice when cabac_init_present_flag is true.
206 */
207Void TEncSbac::determineCabacInitIdx()
208{
209  Int  qp              = m_pcSlice->getSliceQp();
210
211  if (!m_pcSlice->isIntra())
212  {
213    SliceType aSliceTypeChoices[] = {B_SLICE, P_SLICE};
214
215    UInt bestCost             = MAX_UINT;
216    SliceType bestSliceType   = aSliceTypeChoices[0];
217    for (UInt idx=0; idx<2; idx++)
218    {
219      UInt curCost          = 0;
220      SliceType curSliceType  = aSliceTypeChoices[idx];
221
222      curCost  = m_cCUSplitFlagSCModel.calcCost       ( curSliceType, qp, (UChar*)INIT_SPLIT_FLAG );
223      curCost += m_cCUSkipFlagSCModel.calcCost        ( curSliceType, qp, (UChar*)INIT_SKIP_FLAG );
224      curCost += m_cCUMergeFlagExtSCModel.calcCost    ( curSliceType, qp, (UChar*)INIT_MERGE_FLAG_EXT);
225      curCost += m_cCUMergeIdxExtSCModel.calcCost     ( curSliceType, qp, (UChar*)INIT_MERGE_IDX_EXT);
226#if H_3D_ARP
227      curCost += m_cCUPUARPWSCModel.calcCost          ( curSliceType, qp, (UChar*)INIT_ARPW );
228#endif
229#if H_3D_IC
230      curCost += m_cCUICFlagSCModel.calcCost          ( curSliceType, qp, (UChar*)INIT_IC_FLAG );
231#endif
232#if H_3D_INTER_SDC
233      curCost += m_cInterSDCFlagSCModel.calcCost      ( curSliceType, qp, (UChar*)INIT_INTER_SDC_FLAG );
234      curCost += m_cInterSDCResidualSCModel.calcCost  ( curSliceType, qp, (UChar*)INIT_INTER_SDC_RESIDUAL );
235      curCost += m_cInterSDCResidualSignFlagSCModel.calcCost( curSliceType, qp, (UChar*)INIT_INTER_SDC_SIGN_FLAG );
236#endif
237      curCost += m_cCUPartSizeSCModel.calcCost        ( curSliceType, qp, (UChar*)INIT_PART_SIZE );
238      curCost += m_cCUAMPSCModel.calcCost             ( curSliceType, qp, (UChar*)INIT_CU_AMP_POS );
239      curCost += m_cCUPredModeSCModel.calcCost        ( curSliceType, qp, (UChar*)INIT_PRED_MODE );
240      curCost += m_cCUIntraPredSCModel.calcCost       ( curSliceType, qp, (UChar*)INIT_INTRA_PRED_MODE );
241      curCost += m_cCUChromaPredSCModel.calcCost      ( curSliceType, qp, (UChar*)INIT_CHROMA_PRED_MODE );
242      curCost += m_cCUInterDirSCModel.calcCost        ( curSliceType, qp, (UChar*)INIT_INTER_DIR );
243      curCost += m_cCUMvdSCModel.calcCost             ( curSliceType, qp, (UChar*)INIT_MVD );
244      curCost += m_cCURefPicSCModel.calcCost          ( curSliceType, qp, (UChar*)INIT_REF_PIC );
245      curCost += m_cCUDeltaQpSCModel.calcCost         ( curSliceType, qp, (UChar*)INIT_DQP );
246      curCost += m_cCUQtCbfSCModel.calcCost           ( curSliceType, qp, (UChar*)INIT_QT_CBF );
247      curCost += m_cCUQtRootCbfSCModel.calcCost       ( curSliceType, qp, (UChar*)INIT_QT_ROOT_CBF );
248      curCost += m_cCUSigCoeffGroupSCModel.calcCost   ( curSliceType, qp, (UChar*)INIT_SIG_CG_FLAG );
249      curCost += m_cCUSigSCModel.calcCost             ( curSliceType, qp, (UChar*)INIT_SIG_FLAG );
250      curCost += m_cCuCtxLastX.calcCost               ( curSliceType, qp, (UChar*)INIT_LAST );
251      curCost += m_cCuCtxLastY.calcCost               ( curSliceType, qp, (UChar*)INIT_LAST );
252      curCost += m_cCUOneSCModel.calcCost             ( curSliceType, qp, (UChar*)INIT_ONE_FLAG );
253      curCost += m_cCUAbsSCModel.calcCost             ( curSliceType, qp, (UChar*)INIT_ABS_FLAG );
254      curCost += m_cMVPIdxSCModel.calcCost            ( curSliceType, qp, (UChar*)INIT_MVP_IDX );
255      curCost += m_cCUTransSubdivFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
256      curCost += m_cSaoMergeSCModel.calcCost      ( curSliceType, qp, (UChar*)INIT_SAO_MERGE_FLAG );
257      curCost += m_cSaoTypeIdxSCModel.calcCost        ( curSliceType, qp, (UChar*)INIT_SAO_TYPE_IDX );
258      curCost += m_cTransformSkipSCModel.calcCost     ( curSliceType, qp, (UChar*)INIT_TRANSFORMSKIP_FLAG );
259      curCost += m_CUTransquantBypassFlagSCModel.calcCost( curSliceType, qp, (UChar*)INIT_CU_TRANSQUANT_BYPASS_FLAG );
260#if H_3D_DIM
261    if( m_pcSlice->getVpsDepthModesFlag() )
262    {
263      curCost += m_cDepthIntraModeSCModel.calcCost    ( curSliceType, qp, (UChar*)INIT_DEPTH_INTRA_MODE );
264      curCost += m_cDdcFlagSCModel.calcCost           ( curSliceType, qp, (UChar*)INIT_DDC_FLAG );
265      curCost += m_cDdcDataSCModel.calcCost           ( curSliceType, qp, (UChar*)INIT_DDC_DATA );
266#if H_3D_DIM_DMM
267      curCost += m_cDmm1DataSCModel.calcCost          ( curSliceType, qp, (UChar*)INIT_DMM1_DATA );
268#if !SEC_DMM3_RBC_F0147
269      curCost += m_cDmm3DataSCModel.calcCost          ( curSliceType, qp, (UChar*)INIT_DMM3_DATA );
270#endif
271#endif
272#if H_3D_DIM_RBC
273      curCost += m_cRbcDataSCModel.calcCost           ( curSliceType, qp, (UChar*)INIT_RBC_DATA );
274#endif
275    }
276#endif
277      if (curCost < bestCost)
278      {
279        bestSliceType = curSliceType;
280        bestCost      = curCost;
281      }
282    }
283    m_pcSlice->getPPS()->setEncCABACTableIdx( bestSliceType );
284  }
285  else
286  {
287    m_pcSlice->getPPS()->setEncCABACTableIdx( I_SLICE );
288  } 
289}
290
291
292/** The function does the followng: Write out terminate bit. Flush CABAC. Intialize CABAC states. Start CABAC.
293 */
294Void TEncSbac::updateContextTables( SliceType eSliceType, Int iQp, Bool bExecuteFinish )
295{
296  m_pcBinIf->encodeBinTrm(1);
297  if (bExecuteFinish) m_pcBinIf->finish();
298  m_cCUSplitFlagSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_SPLIT_FLAG );
299 
300  m_cCUSkipFlagSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SKIP_FLAG );
301  m_cCUMergeFlagExtSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_MERGE_FLAG_EXT);
302  m_cCUMergeIdxExtSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_MERGE_IDX_EXT);
303#if H_3D_ARP
304  m_cCUPUARPWSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_ARPW );
305#endif
306#if H_3D_IC
307  m_cCUICFlagSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_IC_FLAG );
308#endif
309  m_cCUPartSizeSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_PART_SIZE );
310  m_cCUAMPSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_CU_AMP_POS );
311  m_cCUPredModeSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_PRED_MODE );
312  m_cCUIntraPredSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_INTRA_PRED_MODE );
313  m_cCUChromaPredSCModel.initBuffer      ( eSliceType, iQp, (UChar*)INIT_CHROMA_PRED_MODE );
314  m_cCUInterDirSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_INTER_DIR );
315  m_cCUMvdSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_MVD );
316  m_cCURefPicSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_REF_PIC );
317  m_cCUDeltaQpSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_DQP );
318  m_cCUQtCbfSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_QT_CBF );
319  m_cCUQtRootCbfSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_QT_ROOT_CBF );
320  m_cCUSigCoeffGroupSCModel.initBuffer   ( eSliceType, iQp, (UChar*)INIT_SIG_CG_FLAG );
321  m_cCUSigSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_SIG_FLAG );
322  m_cCuCtxLastX.initBuffer               ( eSliceType, iQp, (UChar*)INIT_LAST );
323  m_cCuCtxLastY.initBuffer               ( eSliceType, iQp, (UChar*)INIT_LAST );
324  m_cCUOneSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_ONE_FLAG );
325  m_cCUAbsSCModel.initBuffer             ( eSliceType, iQp, (UChar*)INIT_ABS_FLAG );
326  m_cMVPIdxSCModel.initBuffer            ( eSliceType, iQp, (UChar*)INIT_MVP_IDX );
327  m_cCUTransSubdivFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
328  m_cSaoMergeSCModel.initBuffer      ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_FLAG );
329  m_cSaoTypeIdxSCModel.initBuffer        ( eSliceType, iQp, (UChar*)INIT_SAO_TYPE_IDX );
330  m_cTransformSkipSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_TRANSFORMSKIP_FLAG );
331  m_CUTransquantBypassFlagSCModel.initBuffer( eSliceType, iQp, (UChar*)INIT_CU_TRANSQUANT_BYPASS_FLAG );
332
333#if H_3D_DIM
334  m_cDepthIntraModeSCModel.initBuffer    ( eSliceType, iQp, (UChar*)INIT_DEPTH_INTRA_MODE );
335  m_cDdcFlagSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_DDC_FLAG );
336  m_cDdcDataSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_DDC_DATA );
337#if H_3D_DIM_DMM
338  m_cDmm1DataSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_DMM1_DATA );
339#if !SEC_DMM3_RBC_F0147
340  m_cDmm3DataSCModel.initBuffer          ( eSliceType, iQp, (UChar*)INIT_DMM3_DATA );
341#endif
342#endif
343#if H_3D_DIM_RBC
344  m_cRbcDataSCModel.initBuffer           ( eSliceType, iQp, (UChar*)INIT_RBC_DATA );
345#endif
346#if H_3D_DIM_SDC
347  m_cSDCResidualFlagSCModel.initBuffer   ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL_FLAG );
348  m_cSDCResidualSCModel.initBuffer       ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL );
349#endif
350#endif
351#if H_3D_INTER_SDC
352  m_cInterSDCFlagSCModel.initBuffer         ( eSliceType, iQp, (UChar*)INIT_INTER_SDC_FLAG );
353  m_cInterSDCResidualSCModel.initBuffer     ( eSliceType, iQp, (UChar*)INIT_INTER_SDC_RESIDUAL );
354  m_cInterSDCResidualSignFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_INTER_SDC_SIGN_FLAG );
355#endif
356  m_pcBinIf->start();
357}
358
359Void TEncSbac::codeVPS( TComVPS* pcVPS )
360{
361  assert (0);
362  return;
363}
364
365#if H_3D
366Void TEncSbac::codeSPS( TComSPS* pcSPS, Int viewIndex, Bool depthFlag )
367#else
368Void TEncSbac::codeSPS( TComSPS* pcSPS )
369#endif
370{
371  assert (0);
372  return;
373}
374
375Void TEncSbac::codePPS( TComPPS* pcPPS )
376{
377  assert (0);
378  return;
379}
380
381Void TEncSbac::codeSliceHeader( TComSlice* pcSlice )
382{
383  assert (0);
384  return;
385}
386
387Void TEncSbac::codeTilesWPPEntryPoint( TComSlice* pSlice )
388{
389  assert (0);
390  return;
391}
392
393Void TEncSbac::codeTerminatingBit( UInt uilsLast )
394{
395  m_pcBinIf->encodeBinTrm( uilsLast );
396}
397
398Void TEncSbac::codeSliceFinish()
399{
400  m_pcBinIf->finish();
401}
402
403Void TEncSbac::xWriteUnarySymbol( UInt uiSymbol, ContextModel* pcSCModel, Int iOffset )
404{
405  m_pcBinIf->encodeBin( uiSymbol ? 1 : 0, pcSCModel[0] );
406 
407  if( 0 == uiSymbol)
408  {
409    return;
410  }
411 
412  while( uiSymbol-- )
413  {
414    m_pcBinIf->encodeBin( uiSymbol ? 1 : 0, pcSCModel[ iOffset ] );
415  }
416 
417  return;
418}
419
420Void TEncSbac::xWriteUnaryMaxSymbol( UInt uiSymbol, ContextModel* pcSCModel, Int iOffset, UInt uiMaxSymbol )
421{
422  if (uiMaxSymbol == 0)
423  {
424    return;
425  }
426 
427  m_pcBinIf->encodeBin( uiSymbol ? 1 : 0, pcSCModel[ 0 ] );
428 
429  if ( uiSymbol == 0 )
430  {
431    return;
432  }
433 
434  Bool bCodeLast = ( uiMaxSymbol > uiSymbol );
435 
436  while( --uiSymbol )
437  {
438    m_pcBinIf->encodeBin( 1, pcSCModel[ iOffset ] );
439  }
440  if( bCodeLast )
441  {
442    m_pcBinIf->encodeBin( 0, pcSCModel[ iOffset ] );
443  }
444 
445  return;
446}
447
448Void TEncSbac::xWriteEpExGolomb( UInt uiSymbol, UInt uiCount )
449{
450  UInt bins = 0;
451  Int numBins = 0;
452 
453  while( uiSymbol >= (UInt)(1<<uiCount) )
454  {
455    bins = 2 * bins + 1;
456    numBins++;
457    uiSymbol -= 1 << uiCount;
458    uiCount  ++;
459  }
460  bins = 2 * bins + 0;
461  numBins++;
462 
463  bins = (bins << uiCount) | uiSymbol;
464  numBins += uiCount;
465 
466  assert( numBins <= 32 );
467  m_pcBinIf->encodeBinsEP( bins, numBins );
468}
469
470/** Coding of coeff_abs_level_minus3
471 * \param uiSymbol value of coeff_abs_level_minus3
472 * \param ruiGoRiceParam reference to Rice parameter
473 * \returns Void
474 */
475Void TEncSbac::xWriteCoefRemainExGolomb ( UInt symbol, UInt &rParam )
476{
477  Int codeNumber  = (Int)symbol;
478  UInt length;
479  if (codeNumber < (COEF_REMAIN_BIN_REDUCTION << rParam))
480  {
481    length = codeNumber>>rParam;
482    m_pcBinIf->encodeBinsEP( (1<<(length+1))-2 , length+1);
483    m_pcBinIf->encodeBinsEP((codeNumber%(1<<rParam)),rParam);
484  }
485  else
486  {
487    length = rParam;
488    codeNumber  = codeNumber - ( COEF_REMAIN_BIN_REDUCTION << rParam);
489    while (codeNumber >= (1<<length))
490    {
491      codeNumber -=  (1<<(length++));   
492    }
493    m_pcBinIf->encodeBinsEP((1<<(COEF_REMAIN_BIN_REDUCTION+length+1-rParam))-2,COEF_REMAIN_BIN_REDUCTION+length+1-rParam);
494    m_pcBinIf->encodeBinsEP(codeNumber,length);
495  }
496}
497
498#if H_3D_DIM
499Void TEncSbac::xWriteExGolombLevel( UInt uiSymbol, ContextModel& rcSCModel  )
500{
501  if( uiSymbol )
502  {
503    m_pcBinIf->encodeBin( 1, rcSCModel );
504    UInt uiCount = 0;
505    Bool bNoExGo = (uiSymbol < 13);
506
507    while( --uiSymbol && ++uiCount < 13 )
508    {
509      m_pcBinIf->encodeBin( 1, rcSCModel );
510    }
511    if( bNoExGo )
512    {
513      m_pcBinIf->encodeBin( 0, rcSCModel );
514    }
515    else
516    {
517      xWriteEpExGolomb( uiSymbol, 0 );
518    }
519  }
520  else
521  {
522    m_pcBinIf->encodeBin( 0, rcSCModel );
523  }
524
525  return;
526}
527
528#if QC_DIM_DELTADC_UNIFY_F0132
529Void TEncSbac::xCodeDimDeltaDC( Pel valDeltaDC, UInt uiNumSeg )
530{
531  xWriteExGolombLevel( UInt( abs( valDeltaDC ) - ( uiNumSeg > 1 ? 0 : 1 ) ), m_cDdcDataSCModel.get(0, 0, 0) );
532  if( valDeltaDC != 0 )
533  {
534    UInt uiSign = valDeltaDC > 0 ? 0 : 1;
535    m_pcBinIf->encodeBinEP( uiSign );
536  }
537}
538#else
539Void TEncSbac::xCodeDimDeltaDC( Pel valDeltaDC, UInt dimType )
540{
541  xWriteExGolombLevel( UInt( abs( valDeltaDC ) ), m_cDdcDataSCModel.get(0, 0, (RBC_IDX == dimType) ? 1 : 0) );
542  if( valDeltaDC != 0 )
543  {
544    UInt uiSign = valDeltaDC > 0 ? 0 : 1;
545    m_pcBinIf->encodeBinEP( uiSign );
546  }
547}
548#endif
549
550#if H_3D_DIM_DMM
551Void TEncSbac::xCodeDmm1WedgeIdx( UInt uiTabIdx, Int iNumBit )
552{
553  for ( Int i = 0; i < iNumBit; i++ )
554  {
555    m_pcBinIf->encodeBin( ( uiTabIdx >> i ) & 1, m_cDmm1DataSCModel.get(0, 0, 0) );
556  }
557}
558
559#if !SEC_DMM3_RBC_F0147
560Void TEncSbac::xCodeDmm3WedgeIdx( UInt uiIntraIdx, Int iNumBit )
561{
562  for( Int i = 0; i < iNumBit; i++ )
563  {
564    m_pcBinIf->encodeBin( ( uiIntraIdx >> i ) & 1, m_cDmm3DataSCModel.get(0, 0, 0) );
565  }
566}
567#endif
568#endif
569#if H_3D_DIM_RBC
570Void TEncSbac::xCodeRbcEdge( TComDataCU* pcCU, UInt uiAbsPartIdx )
571{
572  UInt   uiDepth        = pcCU->getDepth( uiAbsPartIdx ) + (pcCU->getPartitionSize( uiAbsPartIdx ) == SIZE_NxN ? 1 : 0);
573  UChar* pucSymbolList  = pcCU->getEdgeCode     ( uiAbsPartIdx );
574  UChar  ucEdgeNumber   = pcCU->getEdgeNumber   ( uiAbsPartIdx );
575  Bool   bLeft          = pcCU->getEdgeLeftFirst( uiAbsPartIdx );
576  UChar  ucStart        = pcCU->getEdgeStartPos ( uiAbsPartIdx );
577  UInt   uiSymbol;
578
579  // 1. Top(0) or Left(1)
580  uiSymbol = (bLeft == false) ? 0 : 1;
581  m_pcBinIf->encodeBinEP( uiSymbol );
582
583  // 2. Start position (lowest bit first)
584  uiSymbol = ucStart;
585  for( UInt ui = 6; ui > uiDepth; ui-- ) // 64(0)->6, 32(1)->5, 16(2)->4, 8(3)->3, 4(4)->2
586  {
587    m_pcBinIf->encodeBinEP( uiSymbol & 0x1 );
588    uiSymbol >>= 1;
589  }
590
591  // 3. Number of edges
592  uiSymbol = ucEdgeNumber > 0 ? ucEdgeNumber - 1 : 0;
593  for( UInt ui = 7; ui > uiDepth; ui-- ) // 64(0)->7, 32(1)->6, 16(2)->5, 8(3)->4, 4(4)->3
594  {
595    m_pcBinIf->encodeBinEP( uiSymbol & 0x1 );
596    uiSymbol >>= 1;
597  }
598
599  if(uiSymbol != 0)
600  {
601    printf(" ucEdgeNumber %d at depth %d\n",ucEdgeNumber, uiDepth);
602    assert(false);
603  }
604
605  // 4. Edges
606  for( Int iPtr2 = 0; iPtr2 < ucEdgeNumber; iPtr2++ )
607  {
608    UInt uiReorderSymbol = pucSymbolList[iPtr2];
609    for( UInt ui = 0; ui < uiReorderSymbol; ui++ )
610    {
611      m_pcBinIf->encodeBin( 1, m_cRbcDataSCModel.get( 0, 0, 0 ) );
612    }
613
614    if( uiReorderSymbol != 6 )
615      m_pcBinIf->encodeBin( 0, m_cRbcDataSCModel.get( 0, 0, 0 ) );
616  }
617}
618#endif
619#if H_3D_DIM_SDC
620Void TEncSbac::xCodeSDCResidualData ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiSegment )
621{
622  assert( pcCU->getSlice()->getIsDepth() );
623  assert( pcCU->getPartitionSize(uiAbsPartIdx) == SIZE_2Nx2N );
624  assert( pcCU->getSDCFlag(uiAbsPartIdx) );
625  assert( uiSegment < 2 );
626 
627  Pel segmentDCOffset = pcCU->getSDCSegmentDCOffset(uiSegment, uiAbsPartIdx);
628 
629  UInt uiResidual = segmentDCOffset == 0 ? 0 : 1;
630  UInt uiSign     = segmentDCOffset < 0 ? 1 : 0;
631  UInt uiAbsIdx   = abs(segmentDCOffset);
632
633#if H_3D_DIM_DLT
634  UInt uiMaxResidualBits = pcCU->getSlice()->getVPS()->getBitsPerDepthValue( pcCU->getSlice()->getLayerIdInVps() );
635#else
636  UInt uiMaxResidualBits = g_bitDepthY;
637#endif
638  assert( uiMaxResidualBits <= g_bitDepthY );
639 
640  // residual flag
641  m_pcBinIf->encodeBin( uiResidual, m_cSDCResidualFlagSCModel.get( 0, 0, 0 ) );
642#if H_MV_ENC_DEC_TRAC
643  DTRACE_CU("sdc_residual_flag[i]", uiResidual)
644#endif 
645 
646  if (uiResidual)
647  {
648    // encode sign bit of residual
649    m_pcBinIf->encodeBinEP( uiSign );
650#if H_MV_ENC_DEC_TRAC
651    DTRACE_CU("sdc_residual_sign_flag[i]", uiSign)
652#endif
653   
654#if H_3D_DIM_DLT
655    UInt uiNumDepthValues = pcCU->getSlice()->getVPS()->getNumDepthValues( pcCU->getSlice()->getLayerIdInVps() );
656#else
657    UInt uiNumDepthValues = ((1 << g_bitDepthY)-1);
658#endif
659    assert(uiAbsIdx <= uiNumDepthValues);
660   
661    // encode residual magnitude
662    uiAbsIdx -= 1;
663
664#if H_MV_ENC_DEC_TRAC
665    DTRACE_CU("sdc_residual_abs_minus1[i]", uiAbsIdx)
666#endif
667
668    // prefix part
669    if ( uiAbsIdx == 0 )
670    {
671      m_pcBinIf->encodeBin( 0, m_cSDCResidualSCModel.get(0, 0, 0) );
672    }
673    else
674    {
675      UInt l = uiAbsIdx;
676      UInt k = 0;
677      UInt uiPrefixThreshold = ((uiNumDepthValues * 3) >> 2);
678      while ( l > 0 && k < uiPrefixThreshold )
679      {
680        m_pcBinIf->encodeBin( 1, m_cSDCResidualSCModel.get(0, 0, 0) );
681        l--;
682        k++;
683      }
684      if ( uiAbsIdx < uiPrefixThreshold )
685      {
686        m_pcBinIf->encodeBin( 0, m_cSDCResidualSCModel.get(0, 0, 0) );
687      }     
688      else
689      { // suffix part
690        uiAbsIdx -= uiPrefixThreshold;
691        UInt uiSuffixLength = numBitsForValue(uiNumDepthValues - uiPrefixThreshold);
692        UInt uiBitInfo = 0;
693        for ( Int i = 0; i < uiSuffixLength; i++)
694        {
695          uiBitInfo = ( uiAbsIdx & ( 1 << i ) ) >> i;
696          m_pcBinIf->encodeBinEP( uiBitInfo);
697        }
698      }
699    }
700   
701  }
702}
703#endif
704#endif
705
706// SBAC RD
707Void  TEncSbac::load ( TEncSbac* pSrc)
708{
709  this->xCopyFrom(pSrc);
710}
711
712Void  TEncSbac::loadIntraDirModeLuma( TEncSbac* pSrc)
713{
714  m_pcBinIf->copyState( pSrc->m_pcBinIf );
715 
716  this->m_cCUIntraPredSCModel      .copyFrom( &pSrc->m_cCUIntraPredSCModel       );
717}
718#if H_3D_DIM
719Void TEncSbac::loadIntraDepthMode( TEncSbac* pSrc)
720{
721  m_pcBinIf->copyState( pSrc->m_pcBinIf );
722
723  this->m_cDepthIntraModeSCModel .copyFrom( &pSrc->m_cDepthIntraModeSCModel );
724  this->m_cDdcFlagSCModel        .copyFrom( &pSrc->m_cDdcFlagSCModel );
725}
726#endif
727
728Void  TEncSbac::store( TEncSbac* pDest)
729{
730  pDest->xCopyFrom( this );
731}
732
733
734Void TEncSbac::xCopyFrom( TEncSbac* pSrc )
735{
736  m_pcBinIf->copyState( pSrc->m_pcBinIf );
737 
738  this->m_uiCoeffCost = pSrc->m_uiCoeffCost;
739  this->m_uiLastQp    = pSrc->m_uiLastQp;
740 
741  memcpy( m_contextModels, pSrc->m_contextModels, m_numContextModels * sizeof( ContextModel ) );
742}
743
744Void TEncSbac::codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
745{
746  Int iSymbol = pcCU->getMVPIdx(eRefList, uiAbsPartIdx);
747  Int iNum = AMVP_MAX_NUM_CANDS;
748
749  xWriteUnaryMaxSymbol(iSymbol, m_cMVPIdxSCModel.get(0), 1, iNum-1);
750#if H_MV_ENC_DEC_TRAC
751#if ENC_DEC_TRACE
752  if ( eRefList == REF_PIC_LIST_0 )
753  {
754    DTRACE_PU("mvp_l0_flag", iSymbol)
755  }
756  else
757  {
758    DTRACE_PU("mvp_l1_flag", iSymbol)
759  }
760#endif
761#endif
762}
763
764Void TEncSbac::codePartSize( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
765{
766  PartSize eSize         = pcCU->getPartitionSize( uiAbsPartIdx );
767#if H_3D_QTLPC
768  TComSPS *sps           = pcCU->getPic()->getSlice(0)->getSPS();
769  TComPic *pcTexture     = pcCU->getSlice()->getTexturePic();
770  Bool bDepthMapDetect   = (pcTexture != NULL);
771  Bool bIntraSliceDetect = (pcCU->getSlice()->getSliceType() == I_SLICE);
772 
773  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);
774
775  if(bDepthMapDetect && !bIntraSliceDetect && !rapPic && pcCU->getPic()->getReduceBitsFlag() && sps->getUseQTL() && sps->getUsePC() )
776  {
777    TComDataCU *pcTextureCU = pcTexture->getCU(pcCU->getAddr());
778    UInt uiCUIdx            = (pcCU->getZorderIdxInCU() == 0) ? uiAbsPartIdx : pcCU->getZorderIdxInCU();
779    assert(pcTextureCU->getDepth(uiCUIdx) >= uiDepth);
780    if (pcTextureCU->getDepth(uiCUIdx) == uiDepth && pcTextureCU->getPartitionSize( uiCUIdx ) != SIZE_NxN)
781    {
782      assert( eSize == SIZE_2Nx2N );
783      return;
784    }
785  }
786#endif
787  if ( pcCU->isIntra( uiAbsPartIdx ) )
788  {
789    if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth )
790    {
791      m_pcBinIf->encodeBin( eSize == SIZE_2Nx2N? 1 : 0, m_cCUPartSizeSCModel.get( 0, 0, 0 ) );
792#if H_MV_ENC_DEC_TRAC
793      DTRACE_CU("part_mode", eSize == SIZE_2Nx2N? 1 : 0)
794#endif       
795    }
796    return;
797  }
798 
799#if H_MV_ENC_DEC_TRAC         
800  DTRACE_CU("part_mode", eSize )
801#endif       
802  switch(eSize)
803  {
804    case SIZE_2Nx2N:
805    {
806      m_pcBinIf->encodeBin( 1, m_cCUPartSizeSCModel.get( 0, 0, 0) );
807      break;
808    }
809    case SIZE_2NxN:
810    case SIZE_2NxnU:
811    case SIZE_2NxnD:
812    {
813      m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 0) );
814      m_pcBinIf->encodeBin( 1, m_cCUPartSizeSCModel.get( 0, 0, 1) );
815      if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
816      {
817        if (eSize == SIZE_2NxN)
818        {
819          m_pcBinIf->encodeBin(1, m_cCUAMPSCModel.get( 0, 0, 0 ));
820        }
821        else
822        {
823          m_pcBinIf->encodeBin(0, m_cCUAMPSCModel.get( 0, 0, 0 ));         
824          m_pcBinIf->encodeBinEP((eSize == SIZE_2NxnU? 0: 1));
825        }
826      }
827      break;
828    }
829    case SIZE_Nx2N:
830    case SIZE_nLx2N:
831    case SIZE_nRx2N:
832    {
833      m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 0) );
834      m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 1) );
835      if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth && !( pcCU->getWidth(uiAbsPartIdx) == 8 && pcCU->getHeight(uiAbsPartIdx) == 8 ) )
836      {
837        m_pcBinIf->encodeBin( 1, m_cCUPartSizeSCModel.get( 0, 0, 2) );
838      }
839      if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
840      {
841        if (eSize == SIZE_Nx2N)
842        {
843          m_pcBinIf->encodeBin(1, m_cCUAMPSCModel.get( 0, 0, 0 ));
844        }
845        else
846        {
847          m_pcBinIf->encodeBin(0, m_cCUAMPSCModel.get( 0, 0, 0 ));
848          m_pcBinIf->encodeBinEP((eSize == SIZE_nLx2N? 0: 1));
849        }
850      }
851      break;
852    }
853    case SIZE_NxN:
854    {
855      if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth && !( pcCU->getWidth(uiAbsPartIdx) == 8 && pcCU->getHeight(uiAbsPartIdx) == 8 ) )
856      {
857        m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 0) );
858        m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 1) );
859        m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 2) );
860      }
861      break;
862    }
863    default:
864    {
865      assert(0);
866    }
867  }
868}
869
870/** code prediction mode
871 * \param pcCU
872 * \param uiAbsPartIdx 
873 * \returns Void
874 */
875Void TEncSbac::codePredMode( TComDataCU* pcCU, UInt uiAbsPartIdx )
876{
877#if H_3D_DIM_SDC
878  if ( pcCU->getSlice()->isIntra() )
879  {
880    assert( pcCU->isIntra(uiAbsPartIdx) );
881    return;
882  }
883#endif
884 
885  // get context function is here
886  Int iPredMode = pcCU->getPredictionMode( uiAbsPartIdx );
887  m_pcBinIf->encodeBin( iPredMode == MODE_INTER ? 0 : 1, m_cCUPredModeSCModel.get( 0, 0, 0 ) );
888#if H_MV_ENC_DEC_TRAC
889  DTRACE_CU("pred_mode_flag", iPredMode == MODE_INTER ? 0 : 1); 
890#endif
891}
892
893Void TEncSbac::codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
894{
895  UInt uiSymbol = pcCU->getCUTransquantBypass(uiAbsPartIdx);
896  m_pcBinIf->encodeBin( uiSymbol, m_CUTransquantBypassFlagSCModel.get( 0, 0, 0 ) );
897#if H_MV_ENC_DEC_TRAC
898  DTRACE_CU("cu_transquant_bypass_flag", uiSymbol); 
899#endif
900}
901
902/** code skip flag
903 * \param pcCU
904 * \param uiAbsPartIdx
905 * \returns Void
906 */
907Void TEncSbac::codeSkipFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
908{
909  // get context function is here
910  UInt uiSymbol = pcCU->isSkipped( uiAbsPartIdx ) ? 1 : 0;
911  UInt uiCtxSkip = pcCU->getCtxSkipFlag( uiAbsPartIdx ) ;
912  m_pcBinIf->encodeBin( uiSymbol, m_cCUSkipFlagSCModel.get( 0, 0, uiCtxSkip ) );
913#if !H_MV_ENC_DEC_TRAC
914  DTRACE_CABAC_VL( g_nSymbolCounter++ );
915  DTRACE_CABAC_T( "\tSkipFlag" );
916  DTRACE_CABAC_T( "\tuiCtxSkip: ");
917  DTRACE_CABAC_V( uiCtxSkip );
918  DTRACE_CABAC_T( "\tuiSymbol: ");
919  DTRACE_CABAC_V( uiSymbol );
920  DTRACE_CABAC_T( "\n");
921#else
922  DTRACE_CU("cu_skip_flag", uiSymbol); 
923#endif
924}
925
926/** code merge flag
927 * \param pcCU
928 * \param uiAbsPartIdx
929 * \returns Void
930 */
931Void TEncSbac::codeMergeFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
932{
933  const UInt uiSymbol = pcCU->getMergeFlag( uiAbsPartIdx ) ? 1 : 0;
934  m_pcBinIf->encodeBin( uiSymbol, *m_cCUMergeFlagExtSCModel.get( 0 ) );
935
936#if H_MV_ENC_DEC_TRAC
937  DTRACE_PU("merge_flag", uiSymbol); 
938#else
939  DTRACE_CABAC_VL( g_nSymbolCounter++ );
940  DTRACE_CABAC_T( "\tMergeFlag: " );
941  DTRACE_CABAC_V( uiSymbol );
942  DTRACE_CABAC_T( "\tAddress: " );
943  DTRACE_CABAC_V( pcCU->getAddr() );
944  DTRACE_CABAC_T( "\tuiAbsPartIdx: " );
945  DTRACE_CABAC_V( uiAbsPartIdx );
946  DTRACE_CABAC_T( "\n" );
947#endif
948}
949
950/** code merge index
951 * \param pcCU
952 * \param uiAbsPartIdx
953 * \returns Void
954 */
955Void TEncSbac::codeMergeIndex( TComDataCU* pcCU, UInt uiAbsPartIdx )
956{
957  UInt uiUnaryIdx = pcCU->getMergeIndex( uiAbsPartIdx );
958  UInt uiNumCand = pcCU->getSlice()->getMaxNumMergeCand();
959  if ( uiNumCand > 1 )
960  {
961    for( UInt ui = 0; ui < uiNumCand - 1; ++ui )
962    {
963      const UInt uiSymbol = ui == uiUnaryIdx ? 0 : 1;
964      if ( ui==0 )
965      {
966        m_pcBinIf->encodeBin( uiSymbol, m_cCUMergeIdxExtSCModel.get( 0, 0, 0 ) );
967      }
968      else
969      {
970        m_pcBinIf->encodeBinEP( uiSymbol );
971      }
972      if( uiSymbol == 0 )
973      {
974        break;
975      }
976    }
977#if H_MV_ENC_DEC_TRAC
978    DTRACE_PU("merge_idx", uiUnaryIdx); 
979#endif
980  }
981#if !H_MV_ENC_DEC_TRAC
982  DTRACE_CABAC_VL( g_nSymbolCounter++ );
983  DTRACE_CABAC_T( "\tparseMergeIndex()" );
984  DTRACE_CABAC_T( "\tuiMRGIdx= " );
985  DTRACE_CABAC_V( pcCU->getMergeIndex( uiAbsPartIdx ) );
986  DTRACE_CABAC_T( "\n" );
987#endif
988}
989
990#if H_3D_ARP
991Void TEncSbac::codeARPW( TComDataCU* pcCU, UInt uiAbsPartIdx )
992{
993  Int  iW = (Int)pcCU->getARPW( uiAbsPartIdx );
994  Int  iMaxW = pcCU->getSlice()->getARPStepNum() - 1;
995  assert( iMaxW > 0);
996
997  Int nOffset = pcCU->getCTXARPWFlag(uiAbsPartIdx);
998  Int nBinNum = iW + ( iW != iMaxW );
999  m_pcBinIf->encodeBin( iW ? 1 : 0 , m_cCUPUARPWSCModel.get( 0, 0, 0 + nOffset ) );
1000  if( nBinNum > 1 )
1001  {
1002     m_pcBinIf->encodeBin( ( iW == iMaxW ) ? 1 : 0, m_cCUPUARPWSCModel.get( 0, 0, 3 ) );
1003  }
1004#if H_MV_ENC_DEC_TRAC
1005  DTRACE_CU("iv_res_pred_weight_idx", iW); 
1006#endif
1007}
1008#endif
1009
1010#if H_3D_IC
1011/** code Illumination Compensation flag
1012 * \param pcCU
1013 * \param uiAbsPartIdx
1014 * \returns Void
1015 */
1016Void TEncSbac::codeICFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
1017{
1018  // get context function is here
1019  UInt uiSymbol = pcCU->getICFlag( uiAbsPartIdx ) ? 1 : 0;
1020  UInt uiCtxIC  = pcCU->getCtxICFlag( uiAbsPartIdx ) ;
1021  m_pcBinIf->encodeBin( uiSymbol, m_cCUICFlagSCModel.get( 0, 0, uiCtxIC ) );
1022#if !H_MV_ENC_DEC_TRAC
1023  DTRACE_CABAC_VL( g_nSymbolCounter++ );
1024  DTRACE_CABAC_T( "\tICFlag" );
1025  DTRACE_CABAC_T( "\tuiCtxIC: ");
1026  DTRACE_CABAC_V( uiCtxIC );
1027  DTRACE_CABAC_T( "\tuiSymbol: ");
1028  DTRACE_CABAC_V( uiSymbol );
1029  DTRACE_CABAC_T( "\n");
1030#else
1031  DTRACE_CU("ic_flag", uiSymbol ); 
1032#endif
1033}
1034#endif
1035
1036
1037Void TEncSbac::codeSplitFlag   ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
1038{
1039  if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth )
1040    return;
1041 
1042  UInt uiCtx           = pcCU->getCtxSplitFlag( uiAbsPartIdx, uiDepth );
1043  UInt uiCurrSplitFlag = ( pcCU->getDepth( uiAbsPartIdx ) > uiDepth ) ? 1 : 0;
1044 
1045  assert( uiCtx < 3 );
1046#if H_3D_QTLPC
1047  Bool bCodeSplitFlag    = true;
1048
1049  TComSPS *sps           = pcCU->getPic()->getSlice(0)->getSPS();
1050  TComPic *pcTexture     = pcCU->getSlice()->getTexturePic();
1051  Bool bDepthMapDetect   = (pcTexture != NULL);
1052  Bool bIntraSliceDetect = (pcCU->getSlice()->getSliceType() == I_SLICE);
1053
1054  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);
1055
1056  if(bDepthMapDetect && !bIntraSliceDetect && !rapPic && pcCU->getPic()->getReduceBitsFlag() && sps->getUseQTL() && sps->getUsePC())
1057  {
1058    TComDataCU *pcTextureCU = pcTexture->getCU(pcCU->getAddr());
1059    UInt uiCUIdx            = (pcCU->getZorderIdxInCU() == 0) ? uiAbsPartIdx : pcCU->getZorderIdxInCU();
1060    assert(pcTextureCU->getDepth(uiCUIdx) >= uiDepth);
1061    bCodeSplitFlag          = (pcTextureCU->getDepth(uiCUIdx) > uiDepth);
1062  }
1063
1064  if(!bCodeSplitFlag)
1065  {
1066    assert(uiCurrSplitFlag == 0);
1067    return;
1068  }
1069#endif
1070  m_pcBinIf->encodeBin( uiCurrSplitFlag, m_cCUSplitFlagSCModel.get( 0, 0, uiCtx ) );
1071#if !H_MV_ENC_DEC_TRAC
1072  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1073  DTRACE_CABAC_T( "\tSplitFlag\n" )
1074#else
1075  DTRACE_CU("split_cu_flag", uiCurrSplitFlag ); 
1076#endif
1077  return;
1078}
1079
1080Void TEncSbac::codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx )
1081{
1082  m_pcBinIf->encodeBin( uiSymbol, m_cCUTransSubdivFlagSCModel.get( 0, 0, uiCtx ) );
1083#if !H_MV_ENC_DEC_TRAC
1084  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1085  DTRACE_CABAC_T( "\tparseTransformSubdivFlag()" )
1086  DTRACE_CABAC_T( "\tsymbol=" )
1087  DTRACE_CABAC_V( uiSymbol )
1088  DTRACE_CABAC_T( "\tctx=" )
1089  DTRACE_CABAC_V( uiCtx )
1090  DTRACE_CABAC_T( "\n" )
1091#endif
1092}
1093
1094Void TEncSbac::codeIntraDirLumaAng( TComDataCU* pcCU, UInt absPartIdx, Bool isMultiple)
1095{
1096  UInt dir[4],j;
1097  Int preds[4][3] = {{-1, -1, -1},{-1, -1, -1},{-1, -1, -1},{-1, -1, -1}};
1098  Int predNum[4], predIdx[4] ={ -1,-1,-1,-1};
1099  PartSize mode = pcCU->getPartitionSize( absPartIdx );
1100  UInt partNum = isMultiple?(mode==SIZE_NxN?4:1):1;
1101  UInt partOffset = ( pcCU->getPic()->getNumPartInCU() >> ( pcCU->getDepth(absPartIdx) << 1 ) ) >> 2;
1102  for (j=0;j<partNum;j++)
1103  {
1104    dir[j] = pcCU->getLumaIntraDir( absPartIdx+partOffset*j );
1105#if H_3D_DIM
1106    if( pcCU->getSlice()->getVpsDepthModesFlag() )
1107    {
1108      codeIntraDepth( pcCU, absPartIdx+partOffset*j );
1109    }
1110    if( pcCU->getLumaIntraDir( absPartIdx+partOffset*j ) < NUM_INTRA_MODE )
1111#if H_3D_DIM_SDC
1112      if( !pcCU->getSDCFlag( absPartIdx+partOffset*j ) )
1113#endif
1114    {
1115#endif
1116    predNum[j] = pcCU->getIntraDirLumaPredictor(absPartIdx+partOffset*j, preds[j]); 
1117    for(UInt i = 0; i < predNum[j]; i++)
1118    {
1119      if(dir[j] == preds[j][i])
1120      {
1121        predIdx[j] = i;
1122      }
1123    }
1124    m_pcBinIf->encodeBin((predIdx[j] != -1)? 1 : 0, m_cCUIntraPredSCModel.get( 0, 0, 0 ) );
1125#if H_MV_ENC_DEC_TRAC
1126    DTRACE_CU("prev_intra_luma_pred_flag", (predIdx[j] != -1)? 1 : 0); 
1127#endif
1128#if H_3D_DIM
1129    }
1130#endif
1131
1132  } 
1133  for (j=0;j<partNum;j++)
1134  {
1135#if H_3D_DIM
1136    if( pcCU->getLumaIntraDir( absPartIdx+partOffset*j ) < NUM_INTRA_MODE )
1137#if H_3D_DIM_SDC
1138      if( !pcCU->getSDCFlag( absPartIdx+partOffset*j ) )
1139#endif
1140    {
1141#endif
1142    if(predIdx[j] != -1)
1143    {
1144      m_pcBinIf->encodeBinEP( predIdx[j] ? 1 : 0 );
1145      if (predIdx[j])
1146      {
1147        m_pcBinIf->encodeBinEP( predIdx[j]-1 );
1148      }
1149#if H_MV_ENC_DEC_TRAC
1150      DTRACE_CU("mpm_idx", predIdx[j] ); 
1151#endif
1152    }
1153    else
1154    {
1155      if (preds[j][0] > preds[j][1])
1156      { 
1157        std::swap(preds[j][0], preds[j][1]); 
1158      }
1159      if (preds[j][0] > preds[j][2])
1160      {
1161        std::swap(preds[j][0], preds[j][2]);
1162      }
1163      if (preds[j][1] > preds[j][2])
1164      {
1165        std::swap(preds[j][1], preds[j][2]);
1166      }
1167      for(Int i = (predNum[j] - 1); i >= 0; i--)
1168      {
1169        dir[j] = dir[j] > preds[j][i] ? dir[j] - 1 : dir[j];
1170      }
1171      m_pcBinIf->encodeBinsEP( dir[j], 5 );
1172#if H_MV_ENC_DEC_TRAC
1173      DTRACE_CU("rem_intra_luma_pred_mode", dir[j] ); 
1174#endif
1175    }
1176#if H_3D_DIM
1177    }
1178#endif
1179  }
1180  return;
1181}
1182
1183Void TEncSbac::codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx )
1184{
1185  UInt uiIntraDirChroma = pcCU->getChromaIntraDir( uiAbsPartIdx );
1186
1187  if( uiIntraDirChroma == DM_CHROMA_IDX ) 
1188  {
1189    m_pcBinIf->encodeBin( 0, m_cCUChromaPredSCModel.get( 0, 0, 0 ) );
1190#if H_MV_ENC_DEC_TRAC
1191    DTRACE_CU("intra_chroma_pred_mode", 0 ); 
1192#endif
1193  }
1194  else
1195  { 
1196    UInt uiAllowedChromaDir[ NUM_CHROMA_MODE ];
1197    pcCU->getAllowedChromaDir( uiAbsPartIdx, uiAllowedChromaDir );
1198
1199    for( Int i = 0; i < NUM_CHROMA_MODE - 1; i++ )
1200    {
1201      if( uiIntraDirChroma == uiAllowedChromaDir[i] )
1202      {
1203        uiIntraDirChroma = i;
1204        break;
1205      }
1206    }
1207    m_pcBinIf->encodeBin( 1, m_cCUChromaPredSCModel.get( 0, 0, 0 ) );
1208
1209    m_pcBinIf->encodeBinsEP( uiIntraDirChroma, 2 );
1210#if H_MV_ENC_DEC_TRAC
1211    DTRACE_CU("intra_chroma_pred_mode", uiIntraDirChroma ); 
1212#endif
1213
1214  }
1215  return;
1216}
1217
1218#if H_3D_DIM
1219Void TEncSbac::codeIntraDepth( TComDataCU* pcCU, UInt absPartIdx )
1220{
1221  codeIntraDepthMode( pcCU, absPartIdx );
1222
1223  UInt dir     = pcCU->getLumaIntraDir( absPartIdx );
1224  UInt dimType = getDimType( dir );
1225
1226  switch( dimType )
1227  {
1228#if H_3D_DIM_DMM
1229  case( DMM1_IDX ):
1230    {
1231      xCodeDmm1WedgeIdx( pcCU->getDmmWedgeTabIdx( dimType, absPartIdx ), g_dmm1TabIdxBits[pcCU->getIntraSizeIdx(absPartIdx)] );
1232    } break;
1233#if !SEC_DMM3_RBC_F0147
1234  case( DMM3_IDX ):
1235    {
1236      xCodeDmm3WedgeIdx( pcCU->getDmm3IntraTabIdx( absPartIdx ), g_dmm3IntraTabIdxBits[pcCU->getIntraSizeIdx(absPartIdx)] );
1237    } break;
1238#endif
1239  case( DMM4_IDX ): break;
1240#endif
1241#if H_3D_DIM_RBC
1242  case( RBC_IDX ):
1243    {
1244      assert( pcCU->getWidth( absPartIdx ) < 64 );
1245      xCodeRbcEdge( pcCU, absPartIdx );
1246    } break;
1247#endif
1248  default: break;
1249  }
1250
1251#if H_3D_DIM_SDC && !QC_DIM_DELTADC_UNIFY_F0132
1252  if( pcCU->getSDCFlag( absPartIdx ) )
1253  {
1254    assert(pcCU->getPartitionSize(absPartIdx)==SIZE_2Nx2N);
1255    UInt uiNumSegments = ( dir == DC_IDX || dir == PLANAR_IDX )? 1 : 2;
1256    for(UInt uiSeg=0; uiSeg<uiNumSegments; uiSeg++)
1257    {
1258      xCodeSDCResidualData(pcCU, absPartIdx, uiSeg);
1259    }
1260  }
1261  else
1262  {
1263#endif
1264#if QC_DIM_DELTADC_UNIFY_F0132
1265    if( dimType < DIM_NUM_TYPE || pcCU->getSDCFlag( absPartIdx ) )
1266#else
1267    if( dimType < DIM_NUM_TYPE )
1268#endif
1269    {
1270#if QC_DIM_DELTADC_UNIFY_F0132
1271      UInt dimDeltaDC;
1272      Pel  deltaDC;
1273      UInt uiNumSegments = ( dir == PLANAR_IDX ) ? 1 : 2;
1274      if( pcCU->getSDCFlag( absPartIdx ) )
1275      {
1276        if( uiNumSegments==1 )
1277        {
1278          dimDeltaDC = pcCU->getSDCSegmentDCOffset(0, absPartIdx) ? 1 : 0;
1279        }
1280        else
1281        {
1282          dimDeltaDC = ( pcCU->getSDCSegmentDCOffset(0, absPartIdx) || pcCU->getSDCSegmentDCOffset(1, absPartIdx) ) ? 1 : 0;
1283        }
1284      }
1285      else
1286      {
1287        dimDeltaDC = isDimDeltaDC( dir );
1288      }
1289
1290      m_pcBinIf->encodeBin( dimDeltaDC, m_cDdcFlagSCModel.get(0, 0, uiNumSegments-1) );
1291
1292      if( dimDeltaDC ) 
1293      {
1294        for( UInt segment = 0; segment < uiNumSegments; segment++ )
1295        {
1296          deltaDC = pcCU->getSDCFlag( absPartIdx ) ? pcCU->getSDCSegmentDCOffset(segment, absPartIdx) : pcCU->getDimDeltaDC( dimType, segment, absPartIdx );
1297          xCodeDimDeltaDC( deltaDC, uiNumSegments );
1298        }
1299      }
1300#else
1301      UInt dimDeltaDC = isDimDeltaDC( dir );
1302      m_pcBinIf->encodeBin( dimDeltaDC, m_cDdcFlagSCModel.get(0, 0, (RBC_IDX == dimType) ? 1 : 0) );
1303      if( dimDeltaDC ) 
1304      {
1305        for( UInt segment = 0; segment < 2; segment++ )
1306        {
1307          xCodeDimDeltaDC( pcCU->getDimDeltaDC( dimType, segment, absPartIdx ), dimType );
1308        }
1309      }
1310#endif
1311    }
1312#if H_3D_DIM_SDC && !QC_DIM_DELTADC_UNIFY_F0132
1313  }
1314#endif
1315}
1316
1317Void TEncSbac::codeIntraDepthMode( TComDataCU* pcCU, UInt absPartIdx )
1318{
1319#if SEC_DMM3_RBC_F0147
1320  UInt codeWordTable[3][7] =    {{0, 0, 0, 1, 0, 0, 0},{0, 2, 6, 14, 15, 0, 0},{0, 1, 0, 0, 0, 0, 0}};
1321  UInt codeWordLenTable[3][7] = {{0, 1, 0, 1, 0, 0, 0},{1, 2, 3,  4,  4, 0, 0},{1, 1, 0, 0, 0, 0, 0}};
1322#else
1323    UInt codeWordTable[3][7] =    {{0, 0, 0, 2, 0,6, 7},{0, 0, 2, 7, 3, 6, 2},{0, 1, 0, 0, 0, 0, 0}};
1324    UInt codeWordLenTable[3][7] = {{0, 1, 0, 2, 0,3, 3},{1, 1, 2, 3, 2, 3, 2},{1, 1, 0, 0, 0, 0, 0}};
1325#endif
1326  UInt dir = pcCU->getLumaIntraDir( absPartIdx );
1327  UInt puIdx = (pcCU->getWidth(absPartIdx) == 64) ? 2 : ( (pcCU->getPartitionSize(absPartIdx) == SIZE_NxN && pcCU->getWidth(absPartIdx) == 8) ? 0 : 1 );
1328  UInt codeIdx = 0;
1329
1330  if( dir < NUM_INTRA_MODE )
1331  { 
1332    codeIdx = 1; 
1333  }
1334  if( isDimMode( dir ) )
1335  {
1336    switch( getDimType( dir ) )
1337    {
1338    case DMM1_IDX: codeIdx = 3; break;
1339    case DMM4_IDX: codeIdx = 4; break;
1340#if !SEC_DMM3_RBC_F0147
1341    case DMM3_IDX: codeIdx = 5; break;
1342    case  RBC_IDX: codeIdx = 6; break;
1343#endif
1344    default:                    break;
1345    }
1346  }
1347
1348#if H_3D_DIM_SDC
1349  if( pcCU->getSDCFlag( absPartIdx ) )
1350  {
1351    switch( dir )
1352    {
1353      case PLANAR_IDX:  codeIdx = 0; break;
1354      default:          codeIdx = 2; break;
1355    }
1356  }
1357#endif
1358#if !SEC_DMM3_RBC_F0147
1359  if( puIdx==1 )
1360  {
1361      if( codeIdx==1 || codeIdx==2 || codeIdx==4 )
1362      {
1363          m_pcBinIf->encodeBinEP( 0 );
1364      }
1365      else
1366      {
1367          m_pcBinIf->encodeBinEP( 1 );
1368      }
1369  }
1370#endif
1371  //mode coding
1372  for( UInt i = 0; i < codeWordLenTable[puIdx][codeIdx]; i++ )
1373  {
1374    UInt bit = codeWordTable[puIdx][codeIdx] & ( 1<<(codeWordLenTable[puIdx][codeIdx] - i - 1) );
1375    UInt ctxDepthMode = puIdx*3 + ( (i >= 2) ? 2 : i );
1376    m_pcBinIf->encodeBin( bit!=0 , m_cDepthIntraModeSCModel.get(0, 0, ctxDepthMode) );
1377  }
1378}
1379#endif
1380
1381
1382Void TEncSbac::codeInterDir( TComDataCU* pcCU, UInt uiAbsPartIdx )
1383{
1384  const UInt uiInterDir = pcCU->getInterDir( uiAbsPartIdx ) - 1;
1385  const UInt uiCtx      = pcCU->getCtxInterDir( uiAbsPartIdx );
1386  ContextModel *pCtx    = m_cCUInterDirSCModel.get( 0 );
1387  if (pcCU->getPartitionSize(uiAbsPartIdx) == SIZE_2Nx2N || pcCU->getHeight(uiAbsPartIdx) != 8 )
1388  {
1389    m_pcBinIf->encodeBin( uiInterDir == 2 ? 1 : 0, *( pCtx + uiCtx ) );
1390  }
1391  if (uiInterDir < 2)
1392  {
1393    m_pcBinIf->encodeBin( uiInterDir, *( pCtx + 4 ) );
1394  }
1395#if H_MV_ENC_DEC_TRAC
1396  DTRACE_PU("inter_pred_idc", uiInterDir ); 
1397#endif
1398
1399  return;
1400}
1401
1402Void TEncSbac::codeRefFrmIdx( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1403{
1404  {
1405    Int iRefFrame = pcCU->getCUMvField( eRefList )->getRefIdx( uiAbsPartIdx );
1406    ContextModel *pCtx = m_cCURefPicSCModel.get( 0 );
1407    m_pcBinIf->encodeBin( ( iRefFrame == 0 ? 0 : 1 ), *pCtx );
1408   
1409    if( iRefFrame > 0 )
1410    {
1411      UInt uiRefNum = pcCU->getSlice()->getNumRefIdx( eRefList ) - 2;
1412      pCtx++;
1413      iRefFrame--;
1414      for( UInt ui = 0; ui < uiRefNum; ++ui )
1415      {
1416        const UInt uiSymbol = ui == iRefFrame ? 0 : 1;
1417        if( ui == 0 )
1418        {
1419          m_pcBinIf->encodeBin( uiSymbol, *pCtx );       
1420        }
1421        else
1422        {
1423          m_pcBinIf->encodeBinEP( uiSymbol );
1424        }
1425        if( uiSymbol == 0 )
1426        {
1427          break;
1428        }
1429      }
1430    }
1431#if H_MV_ENC_DEC_TRAC
1432#if ENC_DEC_TRACE
1433    iRefFrame = pcCU->getCUMvField( eRefList )->getRefIdx( uiAbsPartIdx ); 
1434    if ( eRefList == REF_PIC_LIST_0 )
1435    {
1436      DTRACE_PU("ref_idx_l0", iRefFrame)
1437    }
1438    else
1439    {
1440      DTRACE_PU("ref_idx_l1", iRefFrame)
1441    }
1442#endif
1443#endif
1444  }
1445  return;
1446}
1447
1448Void TEncSbac::codeMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1449{
1450  if(pcCU->getSlice()->getMvdL1ZeroFlag() && eRefList == REF_PIC_LIST_1 && pcCU->getInterDir(uiAbsPartIdx)==3)
1451  {
1452    return;
1453  }
1454
1455  const TComCUMvField* pcCUMvField = pcCU->getCUMvField( eRefList );
1456  const Int iHor = pcCUMvField->getMvd( uiAbsPartIdx ).getHor();
1457  const Int iVer = pcCUMvField->getMvd( uiAbsPartIdx ).getVer();
1458  ContextModel* pCtx = m_cCUMvdSCModel.get( 0 );
1459
1460  m_pcBinIf->encodeBin( iHor != 0 ? 1 : 0, *pCtx );
1461  m_pcBinIf->encodeBin( iVer != 0 ? 1 : 0, *pCtx );
1462
1463  const Bool bHorAbsGr0 = iHor != 0;
1464  const Bool bVerAbsGr0 = iVer != 0;
1465  const UInt uiHorAbs   = 0 > iHor ? -iHor : iHor;
1466  const UInt uiVerAbs   = 0 > iVer ? -iVer : iVer;
1467  pCtx++;
1468
1469  if( bHorAbsGr0 )
1470  {
1471    m_pcBinIf->encodeBin( uiHorAbs > 1 ? 1 : 0, *pCtx );
1472  }
1473
1474  if( bVerAbsGr0 )
1475  {
1476    m_pcBinIf->encodeBin( uiVerAbs > 1 ? 1 : 0, *pCtx );
1477  }
1478
1479  if( bHorAbsGr0 )
1480  {
1481    if( uiHorAbs > 1 )
1482    {
1483      xWriteEpExGolomb( uiHorAbs-2, 1 );
1484    }
1485
1486    m_pcBinIf->encodeBinEP( 0 > iHor ? 1 : 0 );
1487  }
1488
1489  if( bVerAbsGr0 )
1490  {
1491    if( uiVerAbs > 1 )
1492    {
1493      xWriteEpExGolomb( uiVerAbs-2, 1 );
1494    }
1495
1496    m_pcBinIf->encodeBinEP( 0 > iVer ? 1 : 0 );
1497  }
1498 
1499  return;
1500}
1501
1502Void TEncSbac::codeDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx )
1503{
1504  Int iDQp  = pcCU->getQP( uiAbsPartIdx ) - pcCU->getRefQP( uiAbsPartIdx );
1505 
1506  Int qpBdOffsetY =  pcCU->getSlice()->getSPS()->getQpBDOffsetY();
1507  iDQp = (iDQp + 78 + qpBdOffsetY + (qpBdOffsetY/2)) % (52 + qpBdOffsetY) - 26 - (qpBdOffsetY/2);
1508
1509  UInt uiAbsDQp = (UInt)((iDQp > 0)? iDQp  : (-iDQp));
1510  UInt TUValue = min((Int)uiAbsDQp, CU_DQP_TU_CMAX);
1511  xWriteUnaryMaxSymbol( TUValue, &m_cCUDeltaQpSCModel.get( 0, 0, 0 ), 1, CU_DQP_TU_CMAX);
1512  if( uiAbsDQp >= CU_DQP_TU_CMAX )
1513  {
1514    xWriteEpExGolomb( uiAbsDQp - CU_DQP_TU_CMAX, CU_DQP_EG_k );
1515  }
1516
1517  if ( uiAbsDQp > 0)
1518  {
1519    UInt uiSign = (iDQp > 0 ? 0 : 1);
1520    m_pcBinIf->encodeBinEP(uiSign);
1521  }
1522
1523  return;
1524}
1525
1526Void TEncSbac::codeQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth )
1527{
1528  UInt uiCbf = pcCU->getCbf     ( uiAbsPartIdx, eType, uiTrDepth );
1529  UInt uiCtx = pcCU->getCtxQtCbf( eType, uiTrDepth );
1530  m_pcBinIf->encodeBin( uiCbf , m_cCUQtCbfSCModel.get( 0, eType ? TEXT_CHROMA : eType, uiCtx ) );
1531#if !H_MV_ENC_DEC_TRAC
1532  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1533  DTRACE_CABAC_T( "\tparseQtCbf()" )
1534  DTRACE_CABAC_T( "\tsymbol=" )
1535  DTRACE_CABAC_V( uiCbf )
1536  DTRACE_CABAC_T( "\tctx=" )
1537  DTRACE_CABAC_V( uiCtx )
1538  DTRACE_CABAC_T( "\tetype=" )
1539  DTRACE_CABAC_V( eType )
1540  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1541  DTRACE_CABAC_V( uiAbsPartIdx )
1542  DTRACE_CABAC_T( "\n" )
1543#endif
1544}
1545
1546void TEncSbac::codeTransformSkipFlags (TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, TextType eTType )
1547{
1548  if (pcCU->getCUTransquantBypass(uiAbsPartIdx))
1549  {
1550    return;
1551  }
1552  if(width != 4 || height != 4)
1553  {
1554    return;
1555  }
1556
1557  UInt useTransformSkip = pcCU->getTransformSkip( uiAbsPartIdx,eTType);
1558  m_pcBinIf->encodeBin( useTransformSkip, m_cTransformSkipSCModel.get( 0, eTType? TEXT_CHROMA: TEXT_LUMA, 0 ) );
1559#if !H_MV_ENC_DEC_TRAC
1560  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1561  DTRACE_CABAC_T("\tparseTransformSkip()");
1562  DTRACE_CABAC_T( "\tsymbol=" )
1563  DTRACE_CABAC_V( useTransformSkip )
1564  DTRACE_CABAC_T( "\tAddr=" )
1565  DTRACE_CABAC_V( pcCU->getAddr() )
1566  DTRACE_CABAC_T( "\tetype=" )
1567  DTRACE_CABAC_V( eTType )
1568  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1569  DTRACE_CABAC_V( uiAbsPartIdx )
1570  DTRACE_CABAC_T( "\n" )
1571#endif
1572}
1573
1574/** Code I_PCM information.
1575 * \param pcCU pointer to CU
1576 * \param uiAbsPartIdx CU index
1577 * \returns Void
1578 */
1579Void TEncSbac::codeIPCMInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
1580{
1581  UInt uiIPCM = (pcCU->getIPCMFlag(uiAbsPartIdx) == true)? 1 : 0;
1582
1583  Bool writePCMSampleFlag = pcCU->getIPCMFlag(uiAbsPartIdx);
1584
1585  m_pcBinIf->encodeBinTrm (uiIPCM);
1586#if H_MV_ENC_DEC_TRAC         
1587  DTRACE_CU("pcm_flag", uiIPCM)
1588#endif
1589  if (writePCMSampleFlag)
1590  {
1591    m_pcBinIf->encodePCMAlignBits();
1592
1593    UInt uiMinCoeffSize = pcCU->getPic()->getMinCUWidth()*pcCU->getPic()->getMinCUHeight();
1594    UInt uiLumaOffset   = uiMinCoeffSize*uiAbsPartIdx;
1595    UInt uiChromaOffset = uiLumaOffset>>2;
1596    Pel* piPCMSample;
1597    UInt uiWidth;
1598    UInt uiHeight;
1599    UInt uiSampleBits;
1600    UInt uiX, uiY;
1601
1602    piPCMSample = pcCU->getPCMSampleY() + uiLumaOffset;
1603    uiWidth = pcCU->getWidth(uiAbsPartIdx);
1604    uiHeight = pcCU->getHeight(uiAbsPartIdx);
1605    uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthLuma();
1606
1607    for(uiY = 0; uiY < uiHeight; uiY++)
1608    {
1609      for(uiX = 0; uiX < uiWidth; uiX++)
1610      {
1611        UInt uiSample = piPCMSample[uiX];
1612
1613        m_pcBinIf->xWritePCMCode(uiSample, uiSampleBits);
1614      }
1615      piPCMSample += uiWidth;
1616    }
1617
1618    piPCMSample = pcCU->getPCMSampleCb() + uiChromaOffset;
1619    uiWidth = pcCU->getWidth(uiAbsPartIdx)/2;
1620    uiHeight = pcCU->getHeight(uiAbsPartIdx)/2;
1621    uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
1622
1623    for(uiY = 0; uiY < uiHeight; uiY++)
1624    {
1625      for(uiX = 0; uiX < uiWidth; uiX++)
1626      {
1627        UInt uiSample = piPCMSample[uiX];
1628
1629        m_pcBinIf->xWritePCMCode(uiSample, uiSampleBits);
1630      }
1631      piPCMSample += uiWidth;
1632    }
1633
1634    piPCMSample = pcCU->getPCMSampleCr() + uiChromaOffset;
1635    uiWidth = pcCU->getWidth(uiAbsPartIdx)/2;
1636    uiHeight = pcCU->getHeight(uiAbsPartIdx)/2;
1637    uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
1638
1639    for(uiY = 0; uiY < uiHeight; uiY++)
1640    {
1641      for(uiX = 0; uiX < uiWidth; uiX++)
1642      {
1643        UInt uiSample = piPCMSample[uiX];
1644
1645        m_pcBinIf->xWritePCMCode(uiSample, uiSampleBits);
1646      }
1647      piPCMSample += uiWidth;
1648    }
1649    m_pcBinIf->resetBac();
1650  }
1651}
1652
1653Void TEncSbac::codeQtRootCbf( TComDataCU* pcCU, UInt uiAbsPartIdx )
1654{
1655  UInt uiCbf = pcCU->getQtRootCbf( uiAbsPartIdx );
1656  UInt uiCtx = 0;
1657  m_pcBinIf->encodeBin( uiCbf , m_cCUQtRootCbfSCModel.get( 0, 0, uiCtx ) );
1658#if !H_MV_ENC_DEC_TRAC
1659  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1660  DTRACE_CABAC_T( "\tparseQtRootCbf()" )
1661  DTRACE_CABAC_T( "\tsymbol=" )
1662  DTRACE_CABAC_V( uiCbf )
1663  DTRACE_CABAC_T( "\tctx=" )
1664  DTRACE_CABAC_V( uiCtx )
1665  DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1666  DTRACE_CABAC_V( uiAbsPartIdx )
1667  DTRACE_CABAC_T( "\n" )
1668#else
1669  DTRACE_CU( "rqt_root_cbf", uiCbf )
1670#endif
1671}
1672
1673Void TEncSbac::codeQtCbfZero( TComDataCU* pcCU, TextType eType, UInt uiTrDepth )
1674{
1675  // this function is only used to estimate the bits when cbf is 0
1676  // and will never be called when writing the bistream. do not need to write log
1677  UInt uiCbf = 0;
1678  UInt uiCtx = pcCU->getCtxQtCbf( eType, uiTrDepth );
1679  m_pcBinIf->encodeBin( uiCbf , m_cCUQtCbfSCModel.get( 0, eType ? TEXT_CHROMA : eType, uiCtx ) );
1680}
1681
1682Void TEncSbac::codeQtRootCbfZero( TComDataCU* pcCU )
1683{
1684  // this function is only used to estimate the bits when cbf is 0
1685  // and will never be called when writing the bistream. do not need to write log
1686  UInt uiCbf = 0;
1687  UInt uiCtx = 0;
1688  m_pcBinIf->encodeBin( uiCbf , m_cCUQtRootCbfSCModel.get( 0, 0, uiCtx ) );
1689}
1690
1691/** Encode (X,Y) position of the last significant coefficient
1692 * \param uiPosX X component of last coefficient
1693 * \param uiPosY Y component of last coefficient
1694 * \param width  Block width
1695 * \param height Block height
1696 * \param eTType plane type / luminance or chrominance
1697 * \param uiScanIdx scan type (zig-zag, hor, ver)
1698 * This method encodes the X and Y component within a block of the last significant coefficient.
1699 */
1700Void TEncSbac::codeLastSignificantXY( UInt uiPosX, UInt uiPosY, Int width, Int height, TextType eTType, UInt uiScanIdx )
1701{ 
1702  // swap
1703  if( uiScanIdx == SCAN_VER )
1704  {
1705    swap( uiPosX, uiPosY );
1706  }
1707
1708  UInt uiCtxLast;
1709  ContextModel *pCtxX = m_cCuCtxLastX.get( 0, eTType );
1710  ContextModel *pCtxY = m_cCuCtxLastY.get( 0, eTType );
1711  UInt uiGroupIdxX    = g_uiGroupIdx[ uiPosX ];
1712  UInt uiGroupIdxY    = g_uiGroupIdx[ uiPosY ];
1713
1714
1715  Int blkSizeOffsetX, blkSizeOffsetY, shiftX, shiftY;
1716  blkSizeOffsetX = eTType ? 0: (g_aucConvertToBit[ width ] *3 + ((g_aucConvertToBit[ width ] +1)>>2));
1717  blkSizeOffsetY = eTType ? 0: (g_aucConvertToBit[ height ]*3 + ((g_aucConvertToBit[ height ]+1)>>2));
1718  shiftX= eTType ? g_aucConvertToBit[ width  ] :((g_aucConvertToBit[ width  ]+3)>>2);
1719  shiftY= eTType ? g_aucConvertToBit[ height ] :((g_aucConvertToBit[ height ]+3)>>2);
1720  // posX
1721  for( uiCtxLast = 0; uiCtxLast < uiGroupIdxX; uiCtxLast++ )
1722  {
1723      m_pcBinIf->encodeBin( 1, *( pCtxX + blkSizeOffsetX + (uiCtxLast >>shiftX) ) );
1724  }
1725  if( uiGroupIdxX < g_uiGroupIdx[ width - 1 ])
1726  {
1727      m_pcBinIf->encodeBin( 0, *( pCtxX + blkSizeOffsetX + (uiCtxLast >>shiftX) ) );
1728  }
1729
1730  // posY
1731  for( uiCtxLast = 0; uiCtxLast < uiGroupIdxY; uiCtxLast++ )
1732  {
1733    m_pcBinIf->encodeBin( 1, *( pCtxY + blkSizeOffsetY + (uiCtxLast >>shiftY) ) );
1734  }
1735  if( uiGroupIdxY < g_uiGroupIdx[ height - 1 ])
1736  {
1737    m_pcBinIf->encodeBin( 0, *( pCtxY + blkSizeOffsetY + (uiCtxLast >>shiftY) ) );
1738  }
1739  if ( uiGroupIdxX > 3 )
1740  {     
1741    UInt uiCount = ( uiGroupIdxX - 2 ) >> 1;
1742    uiPosX       = uiPosX - g_uiMinInGroup[ uiGroupIdxX ];
1743    for (Int i = uiCount - 1 ; i >= 0; i-- )
1744    {
1745      m_pcBinIf->encodeBinEP( ( uiPosX >> i ) & 1 );
1746    }
1747  }
1748  if ( uiGroupIdxY > 3 )
1749  {     
1750    UInt uiCount = ( uiGroupIdxY - 2 ) >> 1;
1751    uiPosY       = uiPosY - g_uiMinInGroup[ uiGroupIdxY ];
1752    for ( Int i = uiCount - 1 ; i >= 0; i-- )
1753    {
1754      m_pcBinIf->encodeBinEP( ( uiPosY >> i ) & 1 );
1755    }
1756  }
1757}
1758
1759Void TEncSbac::codeCoeffNxN( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
1760{
1761#if !H_MV_ENC_DEC_TRAC
1762  DTRACE_CABAC_VL( g_nSymbolCounter++ )
1763  DTRACE_CABAC_T( "\tparseCoeffNxN()\teType=" )
1764  DTRACE_CABAC_V( eTType )
1765  DTRACE_CABAC_T( "\twidth=" )
1766  DTRACE_CABAC_V( uiWidth )
1767  DTRACE_CABAC_T( "\theight=" )
1768  DTRACE_CABAC_V( uiHeight )
1769  DTRACE_CABAC_T( "\tdepth=" )
1770  DTRACE_CABAC_V( uiDepth )
1771  DTRACE_CABAC_T( "\tabspartidx=" )
1772  DTRACE_CABAC_V( uiAbsPartIdx )
1773  DTRACE_CABAC_T( "\ttoCU-X=" )
1774  DTRACE_CABAC_V( pcCU->getCUPelX() )
1775  DTRACE_CABAC_T( "\ttoCU-Y=" )
1776  DTRACE_CABAC_V( pcCU->getCUPelY() )
1777  DTRACE_CABAC_T( "\tCU-addr=" )
1778  DTRACE_CABAC_V(  pcCU->getAddr() )
1779  DTRACE_CABAC_T( "\tinCU-X=" )
1780  DTRACE_CABAC_V( g_auiRasterToPelX[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1781  DTRACE_CABAC_T( "\tinCU-Y=" )
1782  DTRACE_CABAC_V( g_auiRasterToPelY[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1783  DTRACE_CABAC_T( "\tpredmode=" )
1784  DTRACE_CABAC_V(  pcCU->getPredictionMode( uiAbsPartIdx ) )
1785  DTRACE_CABAC_T( "\n" )
1786#endif
1787  if( uiWidth > m_pcSlice->getSPS()->getMaxTrSize() )
1788  {
1789    uiWidth  = m_pcSlice->getSPS()->getMaxTrSize();
1790    uiHeight = m_pcSlice->getSPS()->getMaxTrSize();
1791  }
1792 
1793  UInt uiNumSig = 0;
1794 
1795  // compute number of significant coefficients
1796  uiNumSig = TEncEntropy::countNonZeroCoeffs(pcCoef, uiWidth * uiHeight);
1797 
1798  if ( uiNumSig == 0 )
1799    return;
1800  if(pcCU->getSlice()->getPPS()->getUseTransformSkip())
1801  {
1802    codeTransformSkipFlags( pcCU,uiAbsPartIdx, uiWidth, uiHeight, eTType );
1803  }
1804  eTType = eTType == TEXT_LUMA ? TEXT_LUMA : ( eTType == TEXT_NONE ? TEXT_NONE : TEXT_CHROMA );
1805 
1806  //----- encode significance map -----
1807  const UInt   uiLog2BlockSize = g_aucConvertToBit[ uiWidth ] + 2;
1808  UInt uiScanIdx = pcCU->getCoefScanIdx(uiAbsPartIdx, uiWidth, eTType==TEXT_LUMA, pcCU->isIntra(uiAbsPartIdx));
1809  const UInt *scan = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize - 1 ];
1810 
1811  Bool beValid;
1812  if (pcCU->getCUTransquantBypass(uiAbsPartIdx))
1813  {
1814    beValid = false;
1815  }
1816  else 
1817  {
1818    beValid = pcCU->getSlice()->getPPS()->getSignHideFlag() > 0;
1819  }
1820
1821  // Find position of last coefficient
1822  Int scanPosLast = -1;
1823  Int posLast;
1824
1825  const UInt * scanCG;
1826  {
1827    scanCG = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize > 3 ? uiLog2BlockSize-2-1 : 0 ];
1828    if( uiLog2BlockSize == 3 )
1829    {
1830      scanCG = g_sigLastScan8x8[ uiScanIdx ];
1831    }
1832    else if( uiLog2BlockSize == 5 )
1833    {
1834      scanCG = g_sigLastScanCG32x32;
1835    }
1836  }
1837  UInt uiSigCoeffGroupFlag[ MLS_GRP_NUM ];
1838  static const UInt uiShift = MLS_CG_SIZE >> 1;
1839  const UInt uiNumBlkSide = uiWidth >> uiShift;
1840
1841    ::memset( uiSigCoeffGroupFlag, 0, sizeof(UInt) * MLS_GRP_NUM );
1842
1843    do
1844    {
1845      posLast = scan[ ++scanPosLast ];
1846
1847      // get L1 sig map
1848      UInt uiPosY    = posLast >> uiLog2BlockSize;
1849      UInt uiPosX    = posLast - ( uiPosY << uiLog2BlockSize );
1850      UInt uiBlkIdx  = uiNumBlkSide * (uiPosY >> uiShift) + (uiPosX >> uiShift);
1851      if( pcCoef[ posLast ] )
1852      {
1853        uiSigCoeffGroupFlag[ uiBlkIdx ] = 1;
1854      }
1855
1856      uiNumSig -= ( pcCoef[ posLast ] != 0 );
1857    }
1858    while ( uiNumSig > 0 );
1859
1860  // Code position of last coefficient
1861  Int posLastY = posLast >> uiLog2BlockSize;
1862  Int posLastX = posLast - ( posLastY << uiLog2BlockSize );
1863  codeLastSignificantXY(posLastX, posLastY, uiWidth, uiHeight, eTType, uiScanIdx);
1864 
1865  //===== code significance flag =====
1866  ContextModel * const baseCoeffGroupCtx = m_cCUSigCoeffGroupSCModel.get( 0, eTType );
1867  ContextModel * const baseCtx = (eTType==TEXT_LUMA) ? m_cCUSigSCModel.get( 0, 0 ) : m_cCUSigSCModel.get( 0, 0 ) + NUM_SIG_FLAG_CTX_LUMA;
1868
1869
1870  const Int  iLastScanSet      = scanPosLast >> LOG2_SCAN_SET_SIZE;
1871  UInt c1 = 1;
1872  UInt uiGoRiceParam           = 0;
1873  Int  iScanPosSig             = scanPosLast;
1874
1875  for( Int iSubSet = iLastScanSet; iSubSet >= 0; iSubSet-- )
1876  {
1877    Int numNonZero = 0;
1878    Int  iSubPos     = iSubSet << LOG2_SCAN_SET_SIZE;
1879    uiGoRiceParam    = 0;
1880    Int absCoeff[16];
1881    UInt coeffSigns = 0;
1882
1883    Int lastNZPosInCG = -1, firstNZPosInCG = SCAN_SET_SIZE;
1884
1885    if( iScanPosSig == scanPosLast )
1886    {
1887      absCoeff[ 0 ] = abs( pcCoef[ posLast ] );
1888      coeffSigns    = ( pcCoef[ posLast ] < 0 );
1889      numNonZero    = 1;
1890      lastNZPosInCG  = iScanPosSig;
1891      firstNZPosInCG = iScanPosSig;
1892      iScanPosSig--;
1893    }
1894
1895      // encode significant_coeffgroup_flag
1896      Int iCGBlkPos = scanCG[ iSubSet ];
1897      Int iCGPosY   = iCGBlkPos / uiNumBlkSide;
1898      Int iCGPosX   = iCGBlkPos - (iCGPosY * uiNumBlkSide);
1899      if( iSubSet == iLastScanSet || iSubSet == 0)
1900      {
1901        uiSigCoeffGroupFlag[ iCGBlkPos ] = 1;
1902      }
1903      else
1904      {
1905          UInt uiSigCoeffGroup   = (uiSigCoeffGroupFlag[ iCGBlkPos ] != 0);
1906          UInt uiCtxSig  = TComTrQuant::getSigCoeffGroupCtxInc( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1907          m_pcBinIf->encodeBin( uiSigCoeffGroup, baseCoeffGroupCtx[ uiCtxSig ] );
1908      }
1909     
1910      // encode significant_coeff_flag
1911      if( uiSigCoeffGroupFlag[ iCGBlkPos ] )
1912      {
1913        Int patternSigCtx = TComTrQuant::calcPatternSigCtx( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1914        UInt uiBlkPos, uiPosY, uiPosX, uiSig, uiCtxSig;
1915        for( ; iScanPosSig >= iSubPos; iScanPosSig-- )
1916        {
1917          uiBlkPos  = scan[ iScanPosSig ]; 
1918          uiPosY    = uiBlkPos >> uiLog2BlockSize;
1919          uiPosX    = uiBlkPos - ( uiPosY << uiLog2BlockSize );
1920          uiSig     = (pcCoef[ uiBlkPos ] != 0);
1921          if( iScanPosSig > iSubPos || iSubSet == 0 || numNonZero )
1922          {
1923            uiCtxSig  = TComTrQuant::getSigCtxInc( patternSigCtx, uiScanIdx, uiPosX, uiPosY, uiLog2BlockSize, eTType );
1924            m_pcBinIf->encodeBin( uiSig, baseCtx[ uiCtxSig ] );
1925          }
1926          if( uiSig )
1927          {
1928            absCoeff[ numNonZero ] = abs( pcCoef[ uiBlkPos ] );
1929            coeffSigns = 2 * coeffSigns + ( pcCoef[ uiBlkPos ] < 0 );
1930            numNonZero++;
1931            if( lastNZPosInCG == -1 )
1932            {
1933              lastNZPosInCG = iScanPosSig;
1934            }
1935            firstNZPosInCG = iScanPosSig;
1936          }
1937        }
1938      }
1939      else
1940      {
1941        iScanPosSig = iSubPos - 1;
1942      }
1943
1944    if( numNonZero > 0 )
1945    {
1946      Bool signHidden = ( lastNZPosInCG - firstNZPosInCG >= SBH_THRESHOLD );
1947      UInt uiCtxSet = (iSubSet > 0 && eTType==TEXT_LUMA) ? 2 : 0;
1948     
1949      if( c1 == 0 )
1950      {
1951        uiCtxSet++;
1952      }
1953      c1 = 1;
1954      ContextModel *baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUOneSCModel.get( 0, 0 ) + 4 * uiCtxSet : m_cCUOneSCModel.get( 0, 0 ) + NUM_ONE_FLAG_CTX_LUMA + 4 * uiCtxSet;
1955     
1956      Int numC1Flag = min(numNonZero, C1FLAG_NUMBER);
1957      Int firstC2FlagIdx = -1;
1958      for( Int idx = 0; idx < numC1Flag; idx++ )
1959      {
1960        UInt uiSymbol = absCoeff[ idx ] > 1;
1961        m_pcBinIf->encodeBin( uiSymbol, baseCtxMod[c1] );
1962        if( uiSymbol )
1963        {
1964          c1 = 0;
1965
1966          if (firstC2FlagIdx == -1)
1967          {
1968            firstC2FlagIdx = idx;
1969          }
1970        }
1971        else if( (c1 < 3) && (c1 > 0) )
1972        {
1973          c1++;
1974        }
1975      }
1976     
1977      if (c1 == 0)
1978      {
1979
1980        baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUAbsSCModel.get( 0, 0 ) + uiCtxSet : m_cCUAbsSCModel.get( 0, 0 ) + NUM_ABS_FLAG_CTX_LUMA + uiCtxSet;
1981        if ( firstC2FlagIdx != -1)
1982        {
1983          UInt symbol = absCoeff[ firstC2FlagIdx ] > 2;
1984          m_pcBinIf->encodeBin( symbol, baseCtxMod[0] );
1985        }
1986      }
1987     
1988      if( beValid && signHidden )
1989      {
1990        m_pcBinIf->encodeBinsEP( (coeffSigns >> 1), numNonZero-1 );
1991      }
1992      else
1993      {
1994        m_pcBinIf->encodeBinsEP( coeffSigns, numNonZero );
1995      }
1996     
1997      Int iFirstCoeff2 = 1;   
1998      if (c1 == 0 || numNonZero > C1FLAG_NUMBER)
1999      {
2000        for ( Int idx = 0; idx < numNonZero; idx++ )
2001        {
2002          UInt baseLevel  = (idx < C1FLAG_NUMBER)? (2 + iFirstCoeff2 ) : 1;
2003
2004          if( absCoeff[ idx ] >= baseLevel)
2005          {
2006            xWriteCoefRemainExGolomb( absCoeff[ idx ] - baseLevel, uiGoRiceParam );
2007            if(absCoeff[idx] > 3*(1<<uiGoRiceParam))
2008            {
2009               uiGoRiceParam = min<UInt>(uiGoRiceParam+ 1, 4);
2010            }
2011          }
2012          if(absCoeff[ idx ] >= 2) 
2013          {
2014            iFirstCoeff2 = 0;
2015          }
2016        }       
2017      }
2018    }
2019  }
2020
2021  return;
2022}
2023
2024/** code SAO offset sign
2025 * \param code sign value
2026 */
2027Void TEncSbac::codeSAOSign( UInt code )
2028{
2029  m_pcBinIf->encodeBinEP( code );
2030}
2031
2032Void TEncSbac::codeSaoMaxUvlc    ( UInt code, UInt maxSymbol )
2033{
2034  if (maxSymbol == 0)
2035  {
2036    return;
2037  }
2038
2039  Int i;
2040  Bool bCodeLast = ( maxSymbol > code );
2041
2042  if ( code == 0 )
2043  {
2044    m_pcBinIf->encodeBinEP( 0 );
2045  }
2046  else
2047  {
2048    m_pcBinIf->encodeBinEP( 1 );
2049    for ( i=0; i<code-1; i++ )
2050    {
2051      m_pcBinIf->encodeBinEP( 1 );
2052    }
2053    if( bCodeLast )
2054    {
2055      m_pcBinIf->encodeBinEP( 0 );
2056    }
2057  }
2058}
2059
2060
2061/** Code SAO EO class or BO band position
2062 * \param uiLength
2063 * \param uiCode
2064 */
2065Void TEncSbac::codeSaoUflc       ( UInt uiLength, UInt uiCode )
2066{
2067   m_pcBinIf->encodeBinsEP ( uiCode, uiLength );
2068}
2069/** Code SAO merge flags
2070 * \param uiCode
2071 * \param uiCompIdx
2072 */
2073Void TEncSbac::codeSaoMerge       ( UInt uiCode )
2074{
2075  if (uiCode == 0)
2076  {
2077    m_pcBinIf->encodeBin(0,  m_cSaoMergeSCModel.get( 0, 0, 0 ));
2078  }
2079  else
2080  {
2081    m_pcBinIf->encodeBin(1,  m_cSaoMergeSCModel.get( 0, 0, 0 ));
2082  }
2083}
2084/** Code SAO type index
2085 * \param uiCode
2086 */
2087Void TEncSbac::codeSaoTypeIdx       ( UInt uiCode)
2088{
2089  if (uiCode == 0)
2090  {
2091    m_pcBinIf->encodeBin( 0, m_cSaoTypeIdxSCModel.get( 0, 0, 0 ) );
2092  }
2093  else
2094  {
2095    m_pcBinIf->encodeBin( 1, m_cSaoTypeIdxSCModel.get( 0, 0, 0 ) );
2096    m_pcBinIf->encodeBinEP( uiCode <= 4 ? 1 : 0 );
2097  }
2098}
2099/*!
2100 ****************************************************************************
2101 * \brief
2102 *   estimate bit cost for CBP, significant map and significant coefficients
2103 ****************************************************************************
2104 */
2105Void TEncSbac::estBit( estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType )
2106{
2107  estCBFBit( pcEstBitsSbac );
2108
2109  estSignificantCoeffGroupMapBit( pcEstBitsSbac, eTType );
2110 
2111  // encode significance map
2112  estSignificantMapBit( pcEstBitsSbac, width, height, eTType );
2113 
2114  // encode significant coefficients
2115  estSignificantCoefficientsBit( pcEstBitsSbac, eTType );
2116}
2117
2118/*!
2119 ****************************************************************************
2120 * \brief
2121 *    estimate bit cost for each CBP bit
2122 ****************************************************************************
2123 */
2124Void TEncSbac::estCBFBit( estBitsSbacStruct* pcEstBitsSbac )
2125{
2126  ContextModel *pCtx = m_cCUQtCbfSCModel.get( 0 );
2127
2128  for( UInt uiCtxInc = 0; uiCtxInc < 3*NUM_QT_CBF_CTX; uiCtxInc++ )
2129  {
2130    pcEstBitsSbac->blockCbpBits[ uiCtxInc ][ 0 ] = pCtx[ uiCtxInc ].getEntropyBits( 0 );
2131    pcEstBitsSbac->blockCbpBits[ uiCtxInc ][ 1 ] = pCtx[ uiCtxInc ].getEntropyBits( 1 );
2132  }
2133
2134  pCtx = m_cCUQtRootCbfSCModel.get( 0 );
2135 
2136  for( UInt uiCtxInc = 0; uiCtxInc < 4; uiCtxInc++ )
2137  {
2138    pcEstBitsSbac->blockRootCbpBits[ uiCtxInc ][ 0 ] = pCtx[ uiCtxInc ].getEntropyBits( 0 );
2139    pcEstBitsSbac->blockRootCbpBits[ uiCtxInc ][ 1 ] = pCtx[ uiCtxInc ].getEntropyBits( 1 );
2140  }
2141}
2142
2143
2144/*!
2145 ****************************************************************************
2146 * \brief
2147 *    estimate SAMBAC bit cost for significant coefficient group map
2148 ****************************************************************************
2149 */
2150Void TEncSbac::estSignificantCoeffGroupMapBit( estBitsSbacStruct* pcEstBitsSbac, TextType eTType )
2151{
2152  Int firstCtx = 0, numCtx = NUM_SIG_CG_FLAG_CTX;
2153
2154  for ( Int ctxIdx = firstCtx; ctxIdx < firstCtx + numCtx; ctxIdx++ )
2155  {
2156    for( UInt uiBin = 0; uiBin < 2; uiBin++ )
2157    {
2158      pcEstBitsSbac->significantCoeffGroupBits[ ctxIdx ][ uiBin ] = m_cCUSigCoeffGroupSCModel.get(  0, eTType, ctxIdx ).getEntropyBits( uiBin );
2159    }
2160  }
2161}
2162
2163
2164/*!
2165 ****************************************************************************
2166 * \brief
2167 *    estimate SAMBAC bit cost for significant coefficient map
2168 ****************************************************************************
2169 */
2170Void TEncSbac::estSignificantMapBit( estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType )
2171{
2172  Int firstCtx = 1, numCtx = 8;
2173  if (max(width, height) >= 16)
2174  {
2175    firstCtx = (eTType == TEXT_LUMA) ? 21 : 12;
2176    numCtx = (eTType == TEXT_LUMA) ? 6 : 3;
2177  }
2178  else if (width == 8)
2179  {
2180    firstCtx = 9;
2181    numCtx = (eTType == TEXT_LUMA) ? 12 : 3;
2182  }
2183 
2184  if (eTType == TEXT_LUMA )
2185  {
2186    for( UInt bin = 0; bin < 2; bin++ )
2187    {
2188      pcEstBitsSbac->significantBits[ 0 ][ bin ] = m_cCUSigSCModel.get(  0, 0, 0 ).getEntropyBits( bin );
2189    }
2190
2191    for ( Int ctxIdx = firstCtx; ctxIdx < firstCtx + numCtx; ctxIdx++ )
2192    {
2193      for( UInt uiBin = 0; uiBin < 2; uiBin++ )
2194      {
2195        pcEstBitsSbac->significantBits[ ctxIdx ][ uiBin ] = m_cCUSigSCModel.get(  0, 0, ctxIdx ).getEntropyBits( uiBin );
2196      }
2197    }
2198  }
2199  else
2200  {
2201    for( UInt bin = 0; bin < 2; bin++ )
2202    {
2203      pcEstBitsSbac->significantBits[ 0 ][ bin ] = m_cCUSigSCModel.get(  0, 0, NUM_SIG_FLAG_CTX_LUMA + 0 ).getEntropyBits( bin );
2204    }
2205    for ( Int ctxIdx = firstCtx; ctxIdx < firstCtx + numCtx; ctxIdx++ )
2206    {
2207      for( UInt uiBin = 0; uiBin < 2; uiBin++ )
2208      {
2209        pcEstBitsSbac->significantBits[ ctxIdx ][ uiBin ] = m_cCUSigSCModel.get(  0, 0, NUM_SIG_FLAG_CTX_LUMA + ctxIdx ).getEntropyBits( uiBin );
2210      }
2211    }
2212  }
2213  Int iBitsX = 0, iBitsY = 0;
2214  Int blkSizeOffsetX, blkSizeOffsetY, shiftX, shiftY;
2215
2216  blkSizeOffsetX = eTType ? 0: (g_aucConvertToBit[ width ] *3 + ((g_aucConvertToBit[ width ] +1)>>2));
2217  blkSizeOffsetY = eTType ? 0: (g_aucConvertToBit[ height ]*3 + ((g_aucConvertToBit[ height ]+1)>>2));
2218  shiftX = eTType ? g_aucConvertToBit[ width  ] :((g_aucConvertToBit[ width  ]+3)>>2);
2219  shiftY = eTType ? g_aucConvertToBit[ height ] :((g_aucConvertToBit[ height ]+3)>>2);
2220
2221  Int ctx;
2222  ContextModel *pCtxX      = m_cCuCtxLastX.get( 0, eTType );
2223  for (ctx = 0; ctx < g_uiGroupIdx[ width - 1 ]; ctx++)
2224  {
2225    Int ctxOffset = blkSizeOffsetX + (ctx >>shiftX);
2226    pcEstBitsSbac->lastXBits[ ctx ] = iBitsX + pCtxX[ ctxOffset ].getEntropyBits( 0 );
2227    iBitsX += pCtxX[ ctxOffset ].getEntropyBits( 1 );
2228  }
2229  pcEstBitsSbac->lastXBits[ctx] = iBitsX;
2230  ContextModel *pCtxY      = m_cCuCtxLastY.get( 0, eTType );
2231  for (ctx = 0; ctx < g_uiGroupIdx[ height - 1 ]; ctx++)
2232  {
2233    Int ctxOffset = blkSizeOffsetY + (ctx >>shiftY);
2234    pcEstBitsSbac->lastYBits[ ctx ] = iBitsY + pCtxY[ ctxOffset ].getEntropyBits( 0 );
2235    iBitsY += pCtxY[ ctxOffset ].getEntropyBits( 1 );
2236  }
2237  pcEstBitsSbac->lastYBits[ctx] = iBitsY;
2238}
2239
2240/*!
2241 ****************************************************************************
2242 * \brief
2243 *    estimate bit cost of significant coefficient
2244 ****************************************************************************
2245 */
2246Void TEncSbac::estSignificantCoefficientsBit( estBitsSbacStruct* pcEstBitsSbac, TextType eTType )
2247{
2248  if (eTType==TEXT_LUMA)
2249  {
2250    ContextModel *ctxOne = m_cCUOneSCModel.get(0, 0);
2251    ContextModel *ctxAbs = m_cCUAbsSCModel.get(0, 0);
2252
2253    for (Int ctxIdx = 0; ctxIdx < NUM_ONE_FLAG_CTX_LUMA; ctxIdx++)
2254    {
2255      pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 0 ] = ctxOne[ ctxIdx ].getEntropyBits( 0 );
2256      pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 1 ] = ctxOne[ ctxIdx ].getEntropyBits( 1 );   
2257    }
2258
2259    for (Int ctxIdx = 0; ctxIdx < NUM_ABS_FLAG_CTX_LUMA; ctxIdx++)
2260    {
2261      pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 0 ] = ctxAbs[ ctxIdx ].getEntropyBits( 0 );
2262      pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 1 ] = ctxAbs[ ctxIdx ].getEntropyBits( 1 );   
2263    }
2264  }
2265  else
2266  {
2267    ContextModel *ctxOne = m_cCUOneSCModel.get(0, 0) + NUM_ONE_FLAG_CTX_LUMA;
2268    ContextModel *ctxAbs = m_cCUAbsSCModel.get(0, 0) + NUM_ABS_FLAG_CTX_LUMA;
2269
2270    for (Int ctxIdx = 0; ctxIdx < NUM_ONE_FLAG_CTX_CHROMA; ctxIdx++)
2271    {
2272      pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 0 ] = ctxOne[ ctxIdx ].getEntropyBits( 0 );
2273      pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 1 ] = ctxOne[ ctxIdx ].getEntropyBits( 1 );   
2274    }
2275
2276    for (Int ctxIdx = 0; ctxIdx < NUM_ABS_FLAG_CTX_CHROMA; ctxIdx++)
2277    {
2278      pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 0 ] = ctxAbs[ ctxIdx ].getEntropyBits( 0 );
2279      pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 1 ] = ctxAbs[ ctxIdx ].getEntropyBits( 1 );   
2280    }
2281  }
2282}
2283
2284/**
2285 - Initialize our context information from the nominated source.
2286 .
2287 \param pSrc From where to copy context information.
2288 */
2289Void TEncSbac::xCopyContextsFrom( TEncSbac* pSrc )
2290{ 
2291  memcpy(m_contextModels, pSrc->m_contextModels, m_numContextModels*sizeof(m_contextModels[0]));
2292}
2293
2294Void  TEncSbac::loadContexts ( TEncSbac* pScr)
2295{
2296  this->xCopyContextsFrom(pScr);
2297}
2298
2299#if H_3D_INTER_SDC
2300Void TEncSbac::codeInterSDCFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2301{
2302  UInt uiSymbol = pcCU->getInterSDCFlag( uiAbsPartIdx ) ? 1 : 0;
2303  UInt uiCtxInterSDCFlag = pcCU->getCtxInterSDCFlag( uiAbsPartIdx );
2304
2305  m_pcBinIf->encodeBin( uiSymbol, m_cInterSDCFlagSCModel.get( 0, 0, uiCtxInterSDCFlag ) );
2306}
2307
2308Void TEncSbac::codeInterSDCResidualData ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiSegment )
2309{
2310  Pel segmentDCOffset = pcCU->getInterSDCSegmentDCOffset( uiSegment, uiAbsPartIdx );
2311
2312  UInt uiSign     = segmentDCOffset < 0 ? 1 : 0;
2313  UInt uiAbsIdx   = abs( segmentDCOffset );
2314
2315  assert( uiAbsIdx > 0 );
2316  uiAbsIdx--;
2317  xWriteExGolombLevel( uiAbsIdx, m_cInterSDCResidualSCModel.get( 0, 0, 0 ) );
2318  m_pcBinIf->encodeBin( uiSign, m_cInterSDCResidualSignFlagSCModel.get( 0, 0, 0 ) );
2319}
2320#endif
2321//! \}
Note: See TracBrowser for help on using the repository browser.