Context navigation

TEncSbac.cpp @ 213

Visit:

Last change on this file since 213 was 213, checked in by mitsubishi-htm, 12 years ago

A final release, as planned

Migrate to HTM 5.1
For VC project files, only VC9 file is updated
To be used as an additional anchor for CE1.h for 3rd JCTVC meeting at Geneva

Property svn:eol-style set to native

File size: 92.5 KB

Line
1	/* The copyright in this software is being made available under the BSD
2	* License, included below. This software may be subject to other third party
3	* and contributor rights, including patent rights, and no such rights are
4	* granted under this license.
5	*
6	* Copyright (c) 2010-2012, ITU/ISO/IEC
7	* All rights reserved.
8	*
9	* Redistribution and use in source and binary forms, with or without
10	* modification, are permitted provided that the following conditions are met:
11	*
12	* * Redistributions of source code must retain the above copyright notice,
13	* this list of conditions and the following disclaimer.
14	* * Redistributions in binary form must reproduce the above copyright notice,
15	* this list of conditions and the following disclaimer in the documentation
16	* and/or other materials provided with the distribution.
17	* * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18	* be used to endorse or promote products derived from this software without
19	* specific prior written permission.
20	*
21	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24	* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25	* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31	* THE POSSIBILITY OF SUCH DAMAGE.
32	*/
33
34	/** \file 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	#if RWTH_SDC_DLT_B0036
45	#define GetNumDepthValues() (pcCU->getSlice()->getSPS()->getNumDepthValues())
46	#define GetBitsPerDepthValue() (pcCU->getSlice()->getSPS()->getBitsPerDepthValue())
47	#endif
48
49	//! \ingroup TLibEncoder
50	//! \{
51
52	// ====================================================================================================================
53	// Constructor / destructor / create / destroy
54	// ====================================================================================================================
55
56	TEncSbac::TEncSbac()
57	// new structure here
58	: m_pcBitIf ( NULL )
59	, m_pcSlice ( NULL )
60	, m_pcBinIf ( NULL )
61	, m_bAlfCtrl ( false )
62	, m_uiCoeffCost ( 0 )
63	, m_uiMaxAlfCtrlDepth ( 0 )
64	, m_numContextModels ( 0 )
65	, m_cCUSplitFlagSCModel ( 1, 1, NUM_SPLIT_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels )
66	, m_cCUSkipFlagSCModel ( 1, 1, NUM_SKIP_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
67	#if LGE_ILLUCOMP_B0045
68	, m_cCUICFlagSCModel ( 1, 1, NUM_IC_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
69	#endif
70	, m_cCUMergeFlagExtSCModel ( 1, 1, NUM_MERGE_FLAG_EXT_CTX , m_contextModels + m_numContextModels, m_numContextModels)
71	, m_cCUMergeIdxExtSCModel ( 1, 1, NUM_MERGE_IDX_EXT_CTX , m_contextModels + m_numContextModels, m_numContextModels)
72	#if HHI_INTER_VIEW_RESIDUAL_PRED
73	, m_cResPredFlagSCModel ( 1, 1, NUM_RES_PRED_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
74	#endif
75	, m_cCUPartSizeSCModel ( 1, 1, NUM_PART_SIZE_CTX , m_contextModels + m_numContextModels, m_numContextModels)
76	, m_cCUPredModeSCModel ( 1, 1, NUM_PRED_MODE_CTX , m_contextModels + m_numContextModels, m_numContextModels)
77	, m_cCUAlfCtrlFlagSCModel ( 1, 1, NUM_ALF_CTRL_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
78	, m_cCUIntraPredSCModel ( 1, 1, NUM_ADI_CTX , m_contextModels + m_numContextModels, m_numContextModels)
79	, m_cCUChromaPredSCModel ( 1, 1, NUM_CHROMA_PRED_CTX , m_contextModels + m_numContextModels, m_numContextModels)
80	, m_cCUDeltaQpSCModel ( 1, 1, NUM_DELTA_QP_CTX , m_contextModels + m_numContextModels, m_numContextModels)
81	, m_cCUInterDirSCModel ( 1, 1, NUM_INTER_DIR_CTX , m_contextModels + m_numContextModels, m_numContextModels)
82	, m_cCURefPicSCModel ( 1, 1, NUM_REF_NO_CTX , m_contextModels + m_numContextModels, m_numContextModels)
83	, m_cCUMvdSCModel ( 1, 1, NUM_MV_RES_CTX , m_contextModels + m_numContextModels, m_numContextModels)
84	, m_cCUQtCbfSCModel ( 1, 2, NUM_QT_CBF_CTX , m_contextModels + m_numContextModels, m_numContextModels)
85	, m_cCUTransSubdivFlagSCModel ( 1, 1, NUM_TRANS_SUBDIV_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
86	, m_cCUQtRootCbfSCModel ( 1, 1, NUM_QT_ROOT_CBF_CTX , m_contextModels + m_numContextModels, m_numContextModels)
87	, m_cCUSigCoeffGroupSCModel ( 1, 2, NUM_SIG_CG_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
88	, m_cCUSigSCModel ( 1, 1, NUM_SIG_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
89	, m_cCuCtxLastX ( 1, 2, NUM_CTX_LAST_FLAG_XY , m_contextModels + m_numContextModels, m_numContextModels)
90	, m_cCuCtxLastY ( 1, 2, NUM_CTX_LAST_FLAG_XY , m_contextModels + m_numContextModels, m_numContextModels)
91	, m_cCUOneSCModel ( 1, 1, NUM_ONE_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
92	, m_cCUAbsSCModel ( 1, 1, NUM_ABS_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
93	, m_cMVPIdxSCModel ( 1, 1, NUM_MVP_IDX_CTX , m_contextModels + m_numContextModels, m_numContextModels)
94	, m_cALFFlagSCModel ( 1, 1, NUM_ALF_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
95	, m_cALFUvlcSCModel ( 1, 1, NUM_ALF_UVLC_CTX , m_contextModels + m_numContextModels, m_numContextModels)
96	, m_cALFSvlcSCModel ( 1, 1, NUM_ALF_SVLC_CTX , m_contextModels + m_numContextModels, m_numContextModels)
97	#if AMP_CTX
98	, m_cCUAMPSCModel ( 1, 1, NUM_CU_AMP_CTX , m_contextModels + m_numContextModels, m_numContextModels)
99	#else
100	, m_cCUXPosiSCModel ( 1, 1, NUM_CU_X_POS_CTX , m_contextModels + m_numContextModels, m_numContextModels)
101	, m_cCUYPosiSCModel ( 1, 1, NUM_CU_Y_POS_CTX , m_contextModels + m_numContextModels, m_numContextModels)
102	#endif
103	, m_cSaoFlagSCModel ( 1, 1, NUM_SAO_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
104	, m_cSaoUvlcSCModel ( 1, 1, NUM_SAO_UVLC_CTX , m_contextModels + m_numContextModels, m_numContextModels)
105	, m_cSaoSvlcSCModel ( 1, 1, NUM_SAO_SVLC_CTX , m_contextModels + m_numContextModels, m_numContextModels)
106	#if SAO_UNIT_INTERLEAVING
107	, m_cSaoMergeLeftSCModel ( 1, 1, NUM_SAO_MERGE_LEFT_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
108	, m_cSaoMergeUpSCModel ( 1, 1, NUM_SAO_MERGE_UP_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
109	, m_cSaoTypeIdxSCModel ( 1, 1, NUM_SAO_TYPE_IDX_CTX , m_contextModels + m_numContextModels, m_numContextModels)
110	#endif
111	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
112	, m_cDmmFlagSCModel ( 1, 1, NUM_DMM_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
113	, m_cDmmModeSCModel ( 1, 1, NUM_DMM_MODE_CTX , m_contextModels + m_numContextModels, m_numContextModels)
114	, m_cDmmDataSCModel ( 1, 1, NUM_DMM_DATA_CTX , m_contextModels + m_numContextModels, m_numContextModels)
115	#endif
116	#if LGE_EDGE_INTRA_A0070
117	, m_cEdgeIntraSCModel ( 1, 1, NUM_EDGE_INTRA_CTX , m_contextModels + m_numContextModels, m_numContextModels)
118	#if LGE_EDGE_INTRA_DELTA_DC
119	, m_cEdgeIntraDeltaDCSCModel ( 1, 1, NUM_EDGE_INTRA_DELTA_DC_CTX , m_contextModels + m_numContextModels, m_numContextModels)
120	#endif
121	#endif
122	#if RWTH_SDC_DLT_B0036
123	, m_cSDCFlagSCModel ( 1, 1, SDC_NUM_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
124	, m_cSDCResidualFlagSCModel ( 1, 2, SDC_NUM_RESIDUAL_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
125	, m_cSDCResidualSignFlagSCModel ( 1, 2, SDC_NUM_SIGN_FLAG_CTX , m_contextModels + m_numContextModels, m_numContextModels)
126	, m_cSDCResidualSCModel ( 1, 2, SDC_NUM_RESIDUAL_CTX , m_contextModels + m_numContextModels, m_numContextModels)
127	, m_cSDCPredModeSCModel ( 1, 3, SDC_NUM_PRED_MODE_CTX , m_contextModels + m_numContextModels, m_numContextModels)
128	#endif
129	{
130	assert( m_numContextModels <= MAX_NUM_CTX_MOD );
131	m_iSliceGranularity = 0;
132	}
133
134	TEncSbac::~TEncSbac()
135	{
136	}
137
138	// ====================================================================================================================
139	// Public member functions
140	// ====================================================================================================================
141
142	Void TEncSbac::resetEntropy ()
143	{
144	Int iQp = m_pcSlice->getSliceQp();
145	SliceType eSliceType = m_pcSlice->getSliceType();
146
147	#if CABAC_INIT_FLAG
148	Int encCABACTableIdx = m_pcSlice->getPPS()->getEncCABACTableIdx();
149	if (!m_pcSlice->isIntra() && (encCABACTableIdx==B_SLICE \|\| encCABACTableIdx==P_SLICE) && m_pcSlice->getPPS()->getCabacInitPresentFlag())
150	{
151	eSliceType = (SliceType) encCABACTableIdx;
152	}
153	#endif
154
155	m_cCUSplitFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SPLIT_FLAG );
156
157	m_cCUSkipFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SKIP_FLAG );
158	#if LGE_ILLUCOMP_B0045
159	m_cCUICFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_IC_FLAG );
160	#endif
161	m_cCUAlfCtrlFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_CTRL_FLAG );
162	m_cCUMergeFlagExtSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MERGE_FLAG_EXT);
163	m_cCUMergeIdxExtSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MERGE_IDX_EXT);
164	#if HHI_INTER_VIEW_RESIDUAL_PRED
165	m_cResPredFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_RES_PRED_FLAG );
166	#endif
167	m_cCUPartSizeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_PART_SIZE );
168	#if AMP_CTX
169	m_cCUAMPSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CU_AMP_POS );
170	#else
171	m_cCUXPosiSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CU_X_POS );
172	m_cCUYPosiSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CU_Y_POS );
173	#endif
174	m_cCUPredModeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_PRED_MODE );
175	m_cCUIntraPredSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_INTRA_PRED_MODE );
176	m_cCUChromaPredSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CHROMA_PRED_MODE );
177	m_cCUInterDirSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_INTER_DIR );
178	m_cCUMvdSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MVD );
179	m_cCURefPicSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_REF_PIC );
180	m_cCUDeltaQpSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DQP );
181	m_cCUQtCbfSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_QT_CBF );
182	m_cCUQtRootCbfSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_QT_ROOT_CBF );
183	m_cCUSigCoeffGroupSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SIG_CG_FLAG );
184	m_cCUSigSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SIG_FLAG );
185	m_cCuCtxLastX.initBuffer ( eSliceType, iQp, (UChar*)INIT_LAST );
186	m_cCuCtxLastY.initBuffer ( eSliceType, iQp, (UChar*)INIT_LAST );
187	m_cCUOneSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ONE_FLAG );
188	m_cCUAbsSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ABS_FLAG );
189	m_cMVPIdxSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MVP_IDX );
190	m_cALFFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_FLAG );
191	m_cALFUvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_UVLC );
192	m_cALFSvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_SVLC );
193	m_cCUTransSubdivFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
194	m_cSaoFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_FLAG );
195	m_cSaoUvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_UVLC );
196	m_cSaoSvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_SVLC );
197	#if SAO_UNIT_INTERLEAVING
198	m_cSaoMergeLeftSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_LEFT_FLAG );
199	m_cSaoMergeUpSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_UP_FLAG );
200	m_cSaoTypeIdxSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_TYPE_IDX );
201	#endif
202	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
203	m_cDmmFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DMM_FLAG );
204	m_cDmmModeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DMM_MODE );
205	m_cDmmDataSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DMM_DATA );
206	#endif
207	#if LGE_EDGE_INTRA_A0070
208	m_cEdgeIntraSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_EDGE_INTRA );
209	#if LGE_EDGE_INTRA_DELTA_DC
210	m_cEdgeIntraDeltaDCSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_EDGE_INTRA_DELTA_DC );
211	#endif
212	#endif
213	#if RWTH_SDC_DLT_B0036
214	m_cSDCFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_FLAG );
215	m_cSDCResidualFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL_FLAG );
216	m_cSDCResidualSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL );
217	m_cSDCResidualSignFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_SIGN_FLAG );
218	m_cSDCPredModeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_PRED_MODE );
219	#endif
220
221	// new structure
222	m_uiLastQp = iQp;
223
224	m_pcBinIf->start();
225
226	return;
227	}
228
229	#if CABAC_INIT_FLAG
230	/** The function does the following:
231	* If current slice type is P/B then it determines the distance of initialisation type 1 and 2 from the current CABAC states and
232	* stores the index of the closest table. This index is used for the next P/B slice when cabac_init_present_flag is true.
233	*/
234	Void TEncSbac::determineCabacInitIdx()
235	{
236	Int qp = m_pcSlice->getSliceQp();
237
238	if (!m_pcSlice->isIntra())
239	{
240	SliceType aSliceTypeChoices[] = {B_SLICE, P_SLICE};
241
242	UInt bestCost = MAX_UINT;
243	SliceType bestSliceType = aSliceTypeChoices[0];
244	for (UInt idx=0; idx<2; idx++)
245	{
246	UInt curCost = 0;
247	SliceType curSliceType = aSliceTypeChoices[idx];
248
249	curCost = m_cCUSplitFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SPLIT_FLAG );
250	curCost += m_cCUSkipFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SKIP_FLAG );
251	#if LGE_ILLUCOMP_B0045
252	curCost += m_cCUICFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_IC_FLAG );
253	#endif
254	curCost += m_cCUAlfCtrlFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_ALF_CTRL_FLAG );
255	curCost += m_cCUMergeFlagExtSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_MERGE_FLAG_EXT);
256	curCost += m_cCUMergeIdxExtSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_MERGE_IDX_EXT);
257	#if HHI_INTER_VIEW_RESIDUAL_PRED
258	curCost += m_cResPredFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_RES_PRED_FLAG);
259	#endif
260	curCost += m_cCUPartSizeSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_PART_SIZE );
261	#if AMP_CTX
262	curCost += m_cCUAMPSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_CU_AMP_POS );
263	#else
264	curCost += m_cCUXPosiSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_CU_X_POS );
265	curCost += m_cCUYPosiSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_CU_Y_POS );
266	#endif
267	curCost += m_cCUPredModeSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_PRED_MODE );
268	curCost += m_cCUIntraPredSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_INTRA_PRED_MODE );
269	curCost += m_cCUChromaPredSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_CHROMA_PRED_MODE );
270	curCost += m_cCUInterDirSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_INTER_DIR );
271	curCost += m_cCUMvdSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_MVD );
272	curCost += m_cCURefPicSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_REF_PIC );
273	curCost += m_cCUDeltaQpSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_DQP );
274	curCost += m_cCUQtCbfSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_QT_CBF );
275	curCost += m_cCUQtRootCbfSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_QT_ROOT_CBF );
276	curCost += m_cCUSigCoeffGroupSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SIG_CG_FLAG );
277	curCost += m_cCUSigSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SIG_FLAG );
278	curCost += m_cCuCtxLastX.calcCost ( curSliceType, qp, (UChar*)INIT_LAST );
279	curCost += m_cCuCtxLastY.calcCost ( curSliceType, qp, (UChar*)INIT_LAST );
280	curCost += m_cCUOneSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_ONE_FLAG );
281	curCost += m_cCUAbsSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_ABS_FLAG );
282	curCost += m_cMVPIdxSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_MVP_IDX );
283	curCost += m_cALFFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_ALF_FLAG );
284	curCost += m_cALFUvlcSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_ALF_UVLC );
285	curCost += m_cALFSvlcSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_ALF_SVLC );
286	curCost += m_cCUTransSubdivFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
287	curCost += m_cSaoFlagSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SAO_FLAG );
288	curCost += m_cSaoUvlcSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SAO_UVLC );
289	curCost += m_cSaoSvlcSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SAO_SVLC );
290	#if SAO_UNIT_INTERLEAVING
291	curCost += m_cSaoMergeLeftSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SAO_MERGE_LEFT_FLAG );
292	curCost += m_cSaoMergeUpSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SAO_MERGE_UP_FLAG );
293	curCost += m_cSaoTypeIdxSCModel.calcCost ( curSliceType, qp, (UChar*)INIT_SAO_TYPE_IDX );
294	#endif
295
296	if (curCost < bestCost)
297	{
298	bestSliceType = curSliceType;
299	bestCost = curCost;
300	}
301	}
302	m_pcSlice->getPPS()->setEncCABACTableIdx( bestSliceType );
303	}
304	else
305	{
306	m_pcSlice->getPPS()->setEncCABACTableIdx( I_SLICE );
307	}
308
309	#if CABAC_INIT_FLAG && FIX_POZNAN_CABAC_INIT_FLAG
310	m_pcSlice->getPPS()->setEncPrevPOC( m_pcSlice->getPOC() );
311	#endif
312	}
313	#endif
314
315
316	/** The function does the followng: Write out terminate bit. Flush CABAC. Intialize CABAC states. Start CABAC.
317	*/
318	Void TEncSbac::updateContextTables( SliceType eSliceType, Int iQp, Bool bExecuteFinish )
319	{
320	m_pcBinIf->encodeBinTrm(1);
321	if (bExecuteFinish) m_pcBinIf->finish();
322	m_cCUSplitFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SPLIT_FLAG );
323
324	m_cCUSkipFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SKIP_FLAG );
325	#if LGE_ILLUCOMP_B0045
326	m_cCUICFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_IC_FLAG );
327	#endif
328	m_cCUAlfCtrlFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_CTRL_FLAG );
329	m_cCUMergeFlagExtSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MERGE_FLAG_EXT);
330	m_cCUMergeIdxExtSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MERGE_IDX_EXT);
331	#if HHI_INTER_VIEW_RESIDUAL_PRED
332	m_cResPredFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_RES_PRED_FLAG );
333	#endif
334	m_cCUPartSizeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_PART_SIZE );
335	#if AMP_CTX
336	m_cCUAMPSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CU_AMP_POS );
337	#else
338	m_cCUXPosiSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CU_X_POS );
339	m_cCUYPosiSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CU_Y_POS );
340	#endif
341	m_cCUPredModeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_PRED_MODE );
342	m_cCUIntraPredSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_INTRA_PRED_MODE );
343	m_cCUChromaPredSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_CHROMA_PRED_MODE );
344	m_cCUInterDirSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_INTER_DIR );
345	m_cCUMvdSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MVD );
346	m_cCURefPicSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_REF_PIC );
347	m_cCUDeltaQpSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DQP );
348	m_cCUQtCbfSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_QT_CBF );
349	m_cCUQtRootCbfSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_QT_ROOT_CBF );
350	m_cCUSigCoeffGroupSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SIG_CG_FLAG );
351	m_cCUSigSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SIG_FLAG );
352	m_cCuCtxLastX.initBuffer ( eSliceType, iQp, (UChar*)INIT_LAST );
353	m_cCuCtxLastY.initBuffer ( eSliceType, iQp, (UChar*)INIT_LAST );
354	m_cCUOneSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ONE_FLAG );
355	m_cCUAbsSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ABS_FLAG );
356	m_cMVPIdxSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_MVP_IDX );
357	m_cALFFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_FLAG );
358	m_cALFUvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_UVLC );
359	m_cALFSvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_ALF_SVLC );
360	m_cCUTransSubdivFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_TRANS_SUBDIV_FLAG );
361	m_cSaoFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_FLAG );
362	m_cSaoUvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_UVLC );
363	m_cSaoSvlcSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_SVLC );
364	#if SAO_UNIT_INTERLEAVING
365	m_cSaoMergeLeftSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_LEFT_FLAG );
366	m_cSaoMergeUpSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_MERGE_UP_FLAG );
367	m_cSaoTypeIdxSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SAO_TYPE_IDX );
368	#endif
369	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
370	m_cDmmFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DMM_FLAG );
371	m_cDmmModeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DMM_MODE );
372	m_cDmmDataSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_DMM_DATA );
373	#endif
374	#if RWTH_SDC_DLT_B0036
375	m_cSDCFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_FLAG );
376	m_cSDCResidualFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL_FLAG );
377	m_cSDCResidualSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_RESIDUAL );
378	m_cSDCResidualSignFlagSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_SIGN_FLAG );
379	m_cSDCPredModeSCModel.initBuffer ( eSliceType, iQp, (UChar*)INIT_SDC_PRED_MODE );
380	#endif
381
382	m_pcBinIf->start();
383	}
384
385	Void TEncSbac::writeTileMarker( UInt uiTileIdx, UInt uiBitsUsed )
386	{
387	for (Int iShift=uiBitsUsed-1; iShift>=0; iShift--)
388	{
389	m_pcBinIf->encodeBinEP ( (uiTileIdx & (1 << iShift)) >> iShift );
390	}
391	}
392
393	void TEncSbac::codeSEI(const SEI&)
394	{
395	assert(0);
396	}
397
398	#if VIDYO_VPS_INTEGRATION\|QC_MVHEVC_B0046
399	Void TEncSbac::codeVPS( TComVPS* pcVPS )
400	{
401	assert (0);
402	return;
403	}
404	#endif
405
406	#if HHI_MPI
407	Void TEncSbac::codeSPS( TComSPS* pcSPS, Bool bIsDepth )
408	#else
409	Void TEncSbac::codeSPS( TComSPS* pcSPS )
410	#endif
411	{
412	assert (0);
413	return;
414	}
415
416	Void TEncSbac::codePPS( TComPPS* pcPPS )
417	{
418	assert (0);
419	return;
420	}
421
422	Void TEncSbac::codeSliceHeader( TComSlice* pcSlice )
423	{
424	assert (0);
425	return;
426	}
427
428	#if TILES_WPP_ENTRY_POINT_SIGNALLING
429	Void TEncSbac::codeTilesWPPEntryPoint( TComSlice* pSlice )
430	{
431	assert (0);
432	return;
433	}
434	#else
435	Void TEncSbac::codeSliceHeaderSubstreamTable( TComSlice* pcSlice )
436	{
437	assert (0);
438	}
439	#endif
440
441	Void TEncSbac::codeTerminatingBit( UInt uilsLast )
442	{
443	m_pcBinIf->encodeBinTrm( uilsLast );
444	}
445
446	Void TEncSbac::codeSliceFinish()
447	{
448	m_pcBinIf->finish();
449	}
450
451	#if OL_FLUSH
452	Void TEncSbac::codeFlush()
453	{
454	m_pcBinIf->flush();
455	}
456
457	Void TEncSbac::encodeStart()
458	{
459	m_pcBinIf->start();
460	}
461	#endif
462
463	Void TEncSbac::xWriteUnarySymbol( UInt uiSymbol, ContextModel* pcSCModel, Int iOffset )
464	{
465	m_pcBinIf->encodeBin( uiSymbol ? 1 : 0, pcSCModel[0] );
466
467	if( 0 == uiSymbol)
468	{
469	return;
470	}
471
472	while( uiSymbol-- )
473	{
474	m_pcBinIf->encodeBin( uiSymbol ? 1 : 0, pcSCModel[ iOffset ] );
475	}
476
477	return;
478	}
479
480	Void TEncSbac::xWriteUnaryMaxSymbol( UInt uiSymbol, ContextModel* pcSCModel, Int iOffset, UInt uiMaxSymbol )
481	{
482	if (uiMaxSymbol == 0)
483	{
484	return;
485	}
486
487	m_pcBinIf->encodeBin( uiSymbol ? 1 : 0, pcSCModel[ 0 ] );
488
489	if ( uiSymbol == 0 )
490	{
491	return;
492	}
493
494	Bool bCodeLast = ( uiMaxSymbol > uiSymbol );
495
496	while( --uiSymbol )
497	{
498	m_pcBinIf->encodeBin( 1, pcSCModel[ iOffset ] );
499	}
500	if( bCodeLast )
501	{
502	m_pcBinIf->encodeBin( 0, pcSCModel[ iOffset ] );
503	}
504
505	return;
506	}
507
508	Void TEncSbac::xWriteEpExGolomb( UInt uiSymbol, UInt uiCount )
509	{
510	UInt bins = 0;
511	Int numBins = 0;
512
513	while( uiSymbol >= (UInt)(1<<uiCount) )
514	{
515	bins = 2 * bins + 1;
516	numBins++;
517	uiSymbol -= 1 << uiCount;
518	uiCount ++;
519	}
520	bins = 2 * bins + 0;
521	numBins++;
522
523	bins = (bins << uiCount) \| uiSymbol;
524	numBins += uiCount;
525
526	assert( numBins <= 32 );
527	m_pcBinIf->encodeBinsEP( bins, numBins );
528	}
529
530	/** Coding of coeff_abs_level_minus3
531	* \param uiSymbol value of coeff_abs_level_minus3
532	* \param ruiGoRiceParam reference to Rice parameter
533	* \returns Void
534	*/
535	Void TEncSbac::xWriteGoRiceExGolomb( UInt uiSymbol, UInt &ruiGoRiceParam )
536	{
537	UInt uiMaxVlc = g_auiGoRiceRange[ ruiGoRiceParam ];
538	Bool bExGolomb = ( uiSymbol > uiMaxVlc );
539	UInt uiCodeWord = min<UInt>( uiSymbol, ( uiMaxVlc + 1 ) );
540	UInt uiQuotient = uiCodeWord >> ruiGoRiceParam;
541	UInt uiMaxPreLen = g_auiGoRicePrefixLen[ ruiGoRiceParam ];
542
543	UInt binValues;
544	Int numBins;
545
546	if ( uiQuotient >= uiMaxPreLen )
547	{
548	numBins = uiMaxPreLen;
549	binValues = ( 1 << numBins ) - 1;
550	}
551	else
552	{
553	numBins = uiQuotient + 1;
554	binValues = ( 1 << numBins ) - 2;
555	}
556
557	m_pcBinIf->encodeBinsEP( ( binValues << ruiGoRiceParam ) + uiCodeWord - ( uiQuotient << ruiGoRiceParam ), numBins + ruiGoRiceParam );
558
559	#if EIGHT_BITS_RICE_CODE
560	ruiGoRiceParam = g_aauiGoRiceUpdate[ ruiGoRiceParam ][ min<UInt>( uiSymbol, 23 ) ];
561	#else
562	ruiGoRiceParam = g_aauiGoRiceUpdate[ ruiGoRiceParam ][ min<UInt>( uiSymbol, 15 ) ];
563	#endif
564
565	if( bExGolomb )
566	{
567	uiSymbol -= uiMaxVlc + 1;
568	xWriteEpExGolomb( uiSymbol, 0 );
569	}
570	}
571
572	// SBAC RD
573	Void TEncSbac::load ( TEncSbac* pSrc)
574	{
575	this->xCopyFrom(pSrc);
576	}
577
578	Void TEncSbac::loadIntraDirModeLuma( TEncSbac* pSrc)
579	{
580	m_pcBinIf->copyState( pSrc->m_pcBinIf );
581
582	this->m_cCUIntraPredSCModel .copyFrom( &pSrc->m_cCUIntraPredSCModel );
583	}
584
585
586	Void TEncSbac::store( TEncSbac* pDest)
587	{
588	pDest->xCopyFrom( this );
589	}
590
591
592	Void TEncSbac::xCopyFrom( TEncSbac* pSrc )
593	{
594	m_pcBinIf->copyState( pSrc->m_pcBinIf );
595
596	this->m_uiCoeffCost = pSrc->m_uiCoeffCost;
597	this->m_uiLastQp = pSrc->m_uiLastQp;
598
599	memcpy( m_contextModels, pSrc->m_contextModels, m_numContextModels * sizeof( ContextModel ) );
600	}
601
602	#if HHI_INTER_VIEW_MOTION_PRED
603	Void TEncSbac::codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList, Int iNum )
604	#else
605	Void TEncSbac::codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
606	#endif
607	{
608	Int iSymbol = pcCU->getMVPIdx(eRefList, uiAbsPartIdx);
609	#if HHI_INTER_VIEW_MOTION_PRED
610	#else
611	Int iNum = AMVP_MAX_NUM_CANDS;
612	#endif
613
614	xWriteUnaryMaxSymbol(iSymbol, m_cMVPIdxSCModel.get(0), 1, iNum-1);
615	}
616
617	Void TEncSbac::codePartSize( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
618	{
619	PartSize eSize = pcCU->getPartitionSize( uiAbsPartIdx );
620
621	#if OL_QTLIMIT_PREDCODING_B0068
622	TComSPS *sps = pcCU->getPic()->getSlice(0)->getSPS();
623	TComPic *pcTexture = pcCU->getSlice()->getTexturePic();
624	Bool bDepthMapDetect = (pcTexture != NULL);
625	Bool bIntraSliceDetect = (pcCU->getSlice()->getSliceType() == I_SLICE);
626
627	if(bDepthMapDetect && !bIntraSliceDetect && sps->getUseQTLPC() && pcCU->getPic()->getReduceBitsFlag())
628	{
629	TComDataCU *pcTextureCU = pcTexture->getCU(pcCU->getAddr());
630	UInt uiCUIdx = (pcCU->getZorderIdxInCU() == 0) ? uiAbsPartIdx : pcCU->getZorderIdxInCU();
631	assert(pcTextureCU->getDepth(uiCUIdx) >= uiDepth);
632	if (pcTextureCU->getDepth(uiCUIdx) == uiDepth && pcTextureCU->getPartitionSize( uiCUIdx ) != SIZE_NxN)
633	{
634	assert( eSize == SIZE_2Nx2N );
635	return;
636	}
637	}
638	#endif
639
640	if ( pcCU->isIntra( uiAbsPartIdx ) )
641	{
642	if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth )
643	{
644	m_pcBinIf->encodeBin( eSize == SIZE_2Nx2N? 1 : 0, m_cCUPartSizeSCModel.get( 0, 0, 0 ) );
645	}
646	return;
647	}
648
649	switch(eSize)
650	{
651	case SIZE_2Nx2N:
652	{
653	m_pcBinIf->encodeBin( 1, m_cCUPartSizeSCModel.get( 0, 0, 0) );
654	break;
655	}
656	case SIZE_2NxN:
657	case SIZE_2NxnU:
658	case SIZE_2NxnD:
659	{
660	m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 0) );
661	m_pcBinIf->encodeBin( 1, m_cCUPartSizeSCModel.get( 0, 0, 1) );
662	if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
663	{
664	if (eSize == SIZE_2NxN)
665	{
666	#if AMP_CTX
667	m_pcBinIf->encodeBin(1, m_cCUAMPSCModel.get( 0, 0, 0 ));
668	#else
669	m_pcBinIf->encodeBin(1, m_cCUYPosiSCModel.get( 0, 0, 0 ));
670	#endif
671	}
672	else
673	{
674	#if AMP_CTX
675	m_pcBinIf->encodeBin(0, m_cCUAMPSCModel.get( 0, 0, 0 ));
676	m_pcBinIf->encodeBinEP((eSize == SIZE_2NxnU? 0: 1));
677	#else
678	m_pcBinIf->encodeBin(0, m_cCUYPosiSCModel.get( 0, 0, 0 ));
679	m_pcBinIf->encodeBin((eSize == SIZE_2NxnU? 0: 1), m_cCUYPosiSCModel.get( 0, 0, 1 ));
680	#endif
681	}
682	}
683	break;
684	}
685	case SIZE_Nx2N:
686	case SIZE_nLx2N:
687	case SIZE_nRx2N:
688	{
689	m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 0) );
690	m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 1) );
691	if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth && !( pcCU->getSlice()->getSPS()->getDisInter4x4() && pcCU->getWidth(uiAbsPartIdx) == 8 && pcCU->getHeight(uiAbsPartIdx) == 8 ) )
692	{
693	m_pcBinIf->encodeBin( 1, m_cCUPartSizeSCModel.get( 0, 0, 2) );
694	}
695	if ( pcCU->getSlice()->getSPS()->getAMPAcc( uiDepth ) )
696	{
697	if (eSize == SIZE_Nx2N)
698	{
699	#if AMP_CTX
700	m_pcBinIf->encodeBin(1, m_cCUAMPSCModel.get( 0, 0, 0 ));
701	#else
702	m_pcBinIf->encodeBin(1, m_cCUXPosiSCModel.get( 0, 0, 0 ));
703	#endif
704	}
705	else
706	{
707	#if AMP_CTX
708	m_pcBinIf->encodeBin(0, m_cCUAMPSCModel.get( 0, 0, 0 ));
709	m_pcBinIf->encodeBinEP((eSize == SIZE_nLx2N? 0: 1));
710	#else
711	m_pcBinIf->encodeBin(0, m_cCUXPosiSCModel.get( 0, 0, 0 ));
712	m_pcBinIf->encodeBin((eSize == SIZE_nLx2N? 0: 1), m_cCUXPosiSCModel.get( 0, 0, 1 ));
713	#endif
714	}
715	}
716	break;
717	}
718	case SIZE_NxN:
719	{
720	if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth && !( pcCU->getSlice()->getSPS()->getDisInter4x4() && pcCU->getWidth(uiAbsPartIdx) == 8 && pcCU->getHeight(uiAbsPartIdx) == 8 ) )
721	{
722	m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 0) );
723	m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 1) );
724	m_pcBinIf->encodeBin( 0, m_cCUPartSizeSCModel.get( 0, 0, 2) );
725	}
726	break;
727	}
728	default:
729	{
730	assert(0);
731	}
732	}
733	}
734
735	/** code prediction mode
736	* \param pcCU
737	* \param uiAbsPartIdx
738	* \returns Void
739	*/
740	Void TEncSbac::codePredMode( TComDataCU* pcCU, UInt uiAbsPartIdx )
741	{
742	#if RWTH_SDC_DLT_B0036
743	if ( pcCU->getSlice()->isIntra() )
744	{
745	assert( pcCU->isIntra(uiAbsPartIdx) );
746	return;
747	}
748	#endif
749
750	// get context function is here
751	Int iPredMode = pcCU->getPredictionMode( uiAbsPartIdx );
752	m_pcBinIf->encodeBin( iPredMode == MODE_INTER ? 0 : 1, m_cCUPredModeSCModel.get( 0, 0, 0 ) );
753	}
754
755	Void TEncSbac::codeAlfCtrlFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
756	{
757	if (!m_bAlfCtrl)
758	return;
759
760	if( pcCU->getDepth(uiAbsPartIdx) > m_uiMaxAlfCtrlDepth && !pcCU->isFirstAbsZorderIdxInDepth(uiAbsPartIdx, m_uiMaxAlfCtrlDepth))
761	{
762	return;
763	}
764
765	const UInt uiSymbol = pcCU->getAlfCtrlFlag( uiAbsPartIdx ) ? 1 : 0;
766	m_pcBinIf->encodeBin( uiSymbol, *m_cCUAlfCtrlFlagSCModel.get( 0 ) );
767	}
768
769	Void TEncSbac::codeAlfCtrlDepth()
770	{
771	if (!m_bAlfCtrl)
772	return;
773
774	UInt uiDepth = m_uiMaxAlfCtrlDepth;
775	xWriteUnaryMaxSymbol(uiDepth, m_cALFUvlcSCModel.get(0), 1, g_uiMaxCUDepth-1);
776	}
777
778	/** code skip flag
779	* \param pcCU
780	* \param uiAbsPartIdx
781	* \returns Void
782	*/
783	Void TEncSbac::codeSkipFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
784	{
785	// get context function is here
786	#if FORCE_REF_VSP==1
787	UInt uiSymbol = (pcCU->isSkipped( uiAbsPartIdx ) \|\| pcCU->isVspMode( uiAbsPartIdx )) ? 1 : 0;
788	#else
789	UInt uiSymbol = pcCU->isSkipped( uiAbsPartIdx ) ? 1 : 0;
790	#endif
791	UInt uiCtxSkip = pcCU->getCtxSkipFlag( uiAbsPartIdx ) ;
792	m_pcBinIf->encodeBin( uiSymbol, m_cCUSkipFlagSCModel.get( 0, 0, uiCtxSkip ) );
793	DTRACE_CABAC_VL( g_nSymbolCounter++ );
794	DTRACE_CABAC_T( "\tSkipFlag" );
795	DTRACE_CABAC_T( "\tuiCtxSkip: ");
796	DTRACE_CABAC_V( uiCtxSkip );
797	DTRACE_CABAC_T( "\tuiSymbol: ");
798	DTRACE_CABAC_V( uiSymbol );
799	DTRACE_CABAC_T( "\n");
800	}
801
802	#if LGE_ILLUCOMP_B0045
803	/** code Illumination Compensation flag
804	* \param pcCU
805	* \param uiAbsPartIdx
806	* \returns Void
807	*/
808	Void TEncSbac::codeICFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
809	{
810	// get context function is here
811	UInt uiSymbol = pcCU->getICFlag( uiAbsPartIdx ) ? 1 : 0;
812	UInt uiCtxIC = pcCU->getCtxICFlag( uiAbsPartIdx ) ;
813	m_pcBinIf->encodeBin( uiSymbol, m_cCUICFlagSCModel.get( 0, 0, uiCtxIC ) );
814	DTRACE_CABAC_VL( g_nSymbolCounter++ );
815	DTRACE_CABAC_T( "\tICFlag" );
816	DTRACE_CABAC_T( "\tuiCtxIC: ");
817	DTRACE_CABAC_V( uiCtxIC );
818	DTRACE_CABAC_T( "\tuiSymbol: ");
819	DTRACE_CABAC_V( uiSymbol );
820	DTRACE_CABAC_T( "\n");
821	}
822	#endif
823
824	#if FORCE_REF_VSP==1
825	Void TEncSbac::codeVspFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
826	{
827	// get context function is here
828	UInt uiSymbol = (UInt)pcCU->isVspMode( uiAbsPartIdx );
829	UInt uiCtxSkip = pcCU->getCtxSkipFlag( uiAbsPartIdx ) ;
830	m_pcBinIf->encodeBin( uiSymbol, m_cCUSkipFlagSCModel.get( 0, 0, uiCtxSkip ) );
831	DTRACE_CABAC_VL( g_nSymbolCounter++ );
832	DTRACE_CABAC_T( "\tVspFlag" );
833	DTRACE_CABAC_T( "\tuiCtxSkip: ");
834	DTRACE_CABAC_V( uiCtxSkip );
835	DTRACE_CABAC_T( "\tuiSymbol: ");
836	DTRACE_CABAC_V( uiSymbol );
837	DTRACE_CABAC_T( "\n");
838	}
839	#endif
840
841	/** code merge flag
842	* \param pcCU
843	* \param uiAbsPartIdx
844	* \returns Void
845	*/
846	Void TEncSbac::codeMergeFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
847	{
848	const UInt uiSymbol = pcCU->getMergeFlag( uiAbsPartIdx ) ? 1 : 0;
849	m_pcBinIf->encodeBin( uiSymbol, *m_cCUMergeFlagExtSCModel.get( 0 ) );
850
851	DTRACE_CABAC_VL( g_nSymbolCounter++ );
852	DTRACE_CABAC_T( "\tMergeFlag: " );
853	DTRACE_CABAC_V( uiSymbol );
854	DTRACE_CABAC_T( "\tAddress: " );
855	DTRACE_CABAC_V( pcCU->getAddr() );
856	DTRACE_CABAC_T( "\tuiAbsPartIdx: " );
857	DTRACE_CABAC_V( uiAbsPartIdx );
858	DTRACE_CABAC_T( "\n" );
859	}
860
861	/** code merge index
862	* \param pcCU
863	* \param uiAbsPartIdx
864	* \returns Void
865	*/
866	Void TEncSbac::codeMergeIndex( TComDataCU* pcCU, UInt uiAbsPartIdx )
867	{
868	UInt uiNumCand = MRG_MAX_NUM_CANDS;
869	#if !MRG_IDX_CTX_RED
870	UInt auiCtx[4] = { 0, 1, 2, 3 };
871	#endif
872	UInt uiUnaryIdx = pcCU->getMergeIndex( uiAbsPartIdx );
873	uiNumCand = pcCU->getSlice()->getMaxNumMergeCand();
874	#if HHI_MPI
875	#if VSP_N & !MTK_UNCONSTRAINED_MVI_B0083
876	TComDataCU* pcTextCU = pcCU->getSlice()->getSPS()->getUseMVI() ? pcCU->getSlice()->getTexturePic()->getCU( pcCU->getAddr() ) : NULL;
877	Int aiRefIdxVsp[2] = { pcTextCU ? pcTextCU->getCUMvField( RefPicList(0) )->getRefIdx( uiAbsPartIdx ) : -1,
878	pcTextCU ? pcTextCU->getCUMvField( RefPicList(1) )->getRefIdx( uiAbsPartIdx ) : -1 };
879	#if VSP_CFG
880	const Bool bMVIAvailable = (pcCU->getSlice()->getSPS()->getVspDepthPresentFlag() \|\| pcCU->getSlice()->getViewId()==0)
881	? pcCU->getSlice()->getSPS()->getUseMVI() && pcCU->getSlice()->getSliceType() != I_SLICE
882	: pcCU->getSlice()->getSPS()->getUseMVI() && pcCU->getSlice()->getSliceType() != I_SLICE
883	&& !( aiRefIdxVsp[0] >= pcCU->getSlice()->getNumRefIdx( RefPicList(0) ) )
884	&& !( aiRefIdxVsp[1] >= pcCU->getSlice()->getNumRefIdx( RefPicList(1) ) );
885	#else
886	const Bool bMVIAvailable = pcCU->getSlice()->getVspDepthDisableFlag() ? pcCU->getSlice()->getSPS()->getUseMVI() && pcCU->getSlice()->getSliceType() != I_SLICE
887	&& !( aiRefIdxVsp[0] >= 0 && pcTextCU->isVspRef( RefPicList(0), aiRefIdxVsp[0] ) )
888	&& !( aiRefIdxVsp[1] >= 0 && pcTextCU->isVspRef( RefPicList(1), aiRefIdxVsp[1] ) )
889	: pcCU->getSlice()->getSPS()->getUseMVI() && pcCU->getSlice()->getSliceType() != I_SLICE;
890	#endif
891	#else
892	const Bool bMVIAvailable = pcCU->getSlice()->getSPS()->getUseMVI() && pcCU->getSlice()->getSliceType() != I_SLICE;
893	#endif
894	if( bMVIAvailable )
895	{
896	const Bool bUseMVI = pcCU->getTextureModeDepth( uiAbsPartIdx ) != -1;
897	if( bUseMVI )
898	{
899	uiUnaryIdx = (UInt)HHI_MPI_MERGE_POS;
900	}
901	else if( (Int)uiUnaryIdx >= (Int)HHI_MPI_MERGE_POS )
902	{
903	uiUnaryIdx++;
904	}
905	}
906	#endif
907	if ( uiNumCand > 1 )
908	{
909	for( UInt ui = 0; ui < uiNumCand - 1; ++ui )
910	{
911	const UInt uiSymbol = ui == uiUnaryIdx ? 0 : 1;
912	#if MRG_IDX_CTX_RED
913	if ( ui==0 )
914	{
915	m_pcBinIf->encodeBin( uiSymbol, m_cCUMergeIdxExtSCModel.get( 0, 0, 0 ) );
916	}
917	else
918	{
919	m_pcBinIf->encodeBinEP( uiSymbol );
920	}
921	#else
922	m_pcBinIf->encodeBin( uiSymbol, m_cCUMergeIdxExtSCModel.get( 0, 0, auiCtx[ui] ) );
923	#endif
924	if( uiSymbol == 0 )
925	{
926	break;
927	}
928	}
929	}
930	DTRACE_CABAC_VL( g_nSymbolCounter++ );
931	DTRACE_CABAC_T( "\tparseMergeIndex()" );
932	DTRACE_CABAC_T( "\tuiMRGIdx= " );
933	DTRACE_CABAC_V( pcCU->getMergeIndex( uiAbsPartIdx ) );
934	DTRACE_CABAC_T( "\n" );
935	}
936
937	#if HHI_INTER_VIEW_RESIDUAL_PRED
938	Void
939	TEncSbac::codeResPredFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
940	{
941	UInt uiCtx = pcCU->getCtxResPredFlag( uiAbsPartIdx );
942	UInt uiSymbol = ( pcCU->getResPredFlag( uiAbsPartIdx ) ? 1 : 0 );
943	m_pcBinIf->encodeBin( uiSymbol, m_cResPredFlagSCModel.get( 0, 0, uiCtx ) );
944	}
945	#endif
946
947	Void TEncSbac::codeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
948	{
949	if( uiDepth == g_uiMaxCUDepth - g_uiAddCUDepth )
950	return;
951
952	UInt uiCtx = pcCU->getCtxSplitFlag( uiAbsPartIdx, uiDepth );
953	UInt uiCurrSplitFlag = ( pcCU->getDepth( uiAbsPartIdx ) > uiDepth ) ? 1 : 0;
954
955	assert( uiCtx < 3 );
956
957	#if OL_QTLIMIT_PREDCODING_B0068
958	Bool bCodeSplitFlag = true;
959
960	TComSPS *sps = pcCU->getPic()->getSlice(0)->getSPS();
961	TComPic *pcTexture = pcCU->getSlice()->getTexturePic();
962	Bool bDepthMapDetect = (pcTexture != NULL);
963	Bool bIntraSliceDetect = (pcCU->getSlice()->getSliceType() == I_SLICE);
964
965	if(bDepthMapDetect && !bIntraSliceDetect && sps->getUseQTLPC() && pcCU->getPic()->getReduceBitsFlag())
966	{
967	TComDataCU *pcTextureCU = pcTexture->getCU(pcCU->getAddr());
968	UInt uiCUIdx = (pcCU->getZorderIdxInCU() == 0) ? uiAbsPartIdx : pcCU->getZorderIdxInCU();
969	assert(pcTextureCU->getDepth(uiCUIdx) >= uiDepth);
970	bCodeSplitFlag = (pcTextureCU->getDepth(uiCUIdx) > uiDepth);
971	}
972
973	if(!bCodeSplitFlag)
974	{
975	assert(uiCurrSplitFlag == 0);
976	return;
977	}
978	#endif
979
980	m_pcBinIf->encodeBin( uiCurrSplitFlag, m_cCUSplitFlagSCModel.get( 0, 0, uiCtx ) );
981	DTRACE_CABAC_VL( g_nSymbolCounter++ )
982	DTRACE_CABAC_T( "\tSplitFlag\n" )
983	return;
984	}
985
986	Void TEncSbac::codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx )
987	{
988	m_pcBinIf->encodeBin( uiSymbol, m_cCUTransSubdivFlagSCModel.get( 0, 0, uiCtx ) );
989	DTRACE_CABAC_VL( g_nSymbolCounter++ )
990	DTRACE_CABAC_T( "\tparseTransformSubdivFlag()" )
991	DTRACE_CABAC_T( "\tsymbol=" )
992	DTRACE_CABAC_V( uiSymbol )
993	DTRACE_CABAC_T( "\tctx=" )
994	DTRACE_CABAC_V( uiCtx )
995	DTRACE_CABAC_T( "\n" )
996	}
997
998	#if LGE_EDGE_INTRA_A0070
999	Void TEncSbac::xCodeEdgeIntraInfo( TComDataCU* pcCU, UInt uiPartIdx )
1000	{
1001	UInt uiDepth = pcCU->getDepth( uiPartIdx ) + (pcCU->getPartitionSize( uiPartIdx ) == SIZE_NxN ? 1 : 0);
1002	UInt uiCtxEdgeIntra = pcCU->getCtxEdgeIntra( uiPartIdx );
1003	UChar* pucSymbolList = pcCU->getEdgeCode( uiPartIdx );
1004	UChar ucEdgeNumber = pcCU->getEdgeNumber( uiPartIdx );
1005	Bool bLeft = pcCU->getEdgeLeftFirst( uiPartIdx );
1006	UChar ucStart = pcCU->getEdgeStartPos( uiPartIdx );
1007	UInt uiSymbol;
1008
1009	// 1. Top(0) or Left(1)
1010	uiSymbol = (bLeft == false) ? 0 : 1;
1011	m_pcBinIf->encodeBinEP( uiSymbol );
1012
1013	// 2. Start position (lowest bit first)
1014	uiSymbol = ucStart;
1015	for( UInt ui = 6; ui > uiDepth; ui-- ) // 64(0)->6, 32(1)->5, 16(2)->4, 8(3)->3, 4(4)->2
1016	{
1017	m_pcBinIf->encodeBinEP( uiSymbol & 0x1 );
1018	uiSymbol >>= 1;
1019	}
1020
1021	// 3. Number of edges
1022	uiSymbol = ucEdgeNumber > 0 ? ucEdgeNumber - 1 : 0;
1023	for( UInt ui = 7; ui > uiDepth; ui-- ) // 64(0)->7, 32(1)->6, 16(2)->5, 8(3)->4, 4(4)->3
1024	{
1025	m_pcBinIf->encodeBinEP( uiSymbol & 0x1 );
1026	uiSymbol >>= 1;
1027	}
1028
1029	if(uiSymbol != 0)
1030	{
1031	printf(" ucEdgeNumber %d at depth %d\n",ucEdgeNumber, uiDepth);
1032	assert(false);
1033	}
1034
1035	// 4. Edges
1036	for( Int iPtr2 = 0; iPtr2 < ucEdgeNumber; iPtr2++ )
1037	{
1038	UInt uiReorderSymbol = pucSymbolList[iPtr2];
1039
1040	//printf ("Ptr = %d, Symbol = %d\n", iPtr2, uiSymbol);
1041
1042	// Left-friendly direction (Top start)
1043	// 0 ( 0deg) => 0
1044	// 1 ( 45deg) => 10
1045	// 2 ( -45deg) => 110
1046	// 3 ( 90deg) => 1110
1047	// 4 ( -90deg) => 11110
1048	// 5 ( 135deg) => 111110
1049	// 6 (-135deg) => 111111
1050	// Right-friendly direction (Left start)
1051	// 0 ( 0deg) => 0
1052	// 2 ( -45deg) => 10
1053	// 1 ( 45deg) => 110
1054	// 4 ( -90deg) => 1110
1055	// 3 ( 90deg) => 11110
1056	// 6 (-135deg) => 111110
1057	// 5 ( 135deg) => 111111
1058
1059	// refer to a paper "An efficient chain code with Huffman coding"
1060
1061	for( UInt ui = 0; ui < uiReorderSymbol; ui++ )
1062	{
1063	m_pcBinIf->encodeBin( 1, m_cEdgeIntraSCModel.get( 0, 0, uiCtxEdgeIntra ) );
1064	}
1065
1066	if( uiReorderSymbol != 6 )
1067	m_pcBinIf->encodeBin( 0, m_cEdgeIntraSCModel.get( 0, 0, uiCtxEdgeIntra ) );
1068	}
1069	}
1070	#endif
1071
1072	Void TEncSbac::codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx )
1073	{
1074	UInt uiDir = pcCU->getLumaIntraDir( uiAbsPartIdx );
1075
1076	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
1077	if( pcCU->getSlice()->getSPS()->getUseDMM() && pcCU->getWidth( uiAbsPartIdx ) <= DMM_WEDGEMODEL_MAX_SIZE )
1078	{
1079	#if LGE_EDGE_INTRA_A0070
1080	m_pcBinIf->encodeBin( uiDir >= NUM_INTRA_MODE && uiDir < EDGE_INTRA_IDX, m_cDmmFlagSCModel.get(0, 0, 0) );
1081	#else
1082	m_pcBinIf->encodeBin( uiDir >= NUM_INTRA_MODE, m_cDmmFlagSCModel.get(0, 0, 0) );
1083	#endif
1084	}
1085	#if LGE_EDGE_INTRA_A0070
1086	if( uiDir >= NUM_INTRA_MODE && uiDir < EDGE_INTRA_IDX )
1087	#else
1088	if( uiDir >= NUM_INTRA_MODE )
1089	#endif
1090	{
1091	assert( pcCU->getWidth( uiAbsPartIdx ) <= DMM_WEDGEMODEL_MAX_SIZE );
1092	UInt uiDMMode = uiDir - NUM_INTRA_MODE;
1093
1094	#if HHI_DMM_WEDGE_INTRA && HHI_DMM_PRED_TEX
1095	m_pcBinIf->encodeBin( (uiDMMode & 0x01), m_cDmmModeSCModel.get(0, 0, 0) );
1096	m_pcBinIf->encodeBin( (uiDMMode & 0x02) >> 1, m_cDmmModeSCModel.get(0, 0, 0) );
1097
1098	if( pcCU->getPartitionSize( uiAbsPartIdx ) != SIZE_NxN && pcCU->getWidth( uiAbsPartIdx ) > 4 )
1099	{
1100	m_pcBinIf->encodeBin( (uiDMMode & 0x04) >> 2, m_cDmmModeSCModel.get(0, 0, 0) );
1101	}
1102	#else
1103	m_pcBinIf->encodeBin( (uiDMMode & 0x01), m_cDmmModeSCModel.get(0, 0, 0) );
1104
1105	if( pcCU->getPartitionSize( uiAbsPartIdx ) != SIZE_NxN && pcCU->getWidth( uiAbsPartIdx ) > 4 )
1106	{
1107	m_pcBinIf->encodeBin( (uiDMMode & 0x02) >> 1, m_cDmmModeSCModel.get(0, 0, 0) );
1108	}
1109	#endif
1110	#if HHI_DMM_WEDGE_INTRA
1111	if( uiDir == DMM_WEDGE_FULL_IDX ) { xCodeWedgeFullInfo ( pcCU, uiAbsPartIdx ); }
1112	if( uiDir == DMM_WEDGE_FULL_D_IDX ) { xCodeWedgeFullDeltaInfo ( pcCU, uiAbsPartIdx ); }
1113	if( uiDir == DMM_WEDGE_PREDDIR_IDX ) { xCodeWedgePredDirInfo ( pcCU, uiAbsPartIdx ); }
1114	if( uiDir == DMM_WEDGE_PREDDIR_D_IDX ) { xCodeWedgePredDirDeltaInfo ( pcCU, uiAbsPartIdx ); }
1115	#endif
1116	#if HHI_DMM_PRED_TEX
1117
1118	#if FLEX_CODING_ORDER_M23723
1119	if ( !pcCU->getSlice()->getSPS()->getUseDMM34() )
1120	{
1121	assert( uiDir != DMM_WEDGE_PREDTEX_D_IDX );
1122	assert( uiDir != DMM_CONTOUR_PREDTEX_D_IDX );
1123	}
1124	#endif
1125
1126	if( uiDir == DMM_WEDGE_PREDTEX_D_IDX ) { xCodeWedgePredTexDeltaInfo ( pcCU, uiAbsPartIdx ); }
1127	if( uiDir == DMM_CONTOUR_PREDTEX_D_IDX ) { xCodeContourPredTexDeltaInfo( pcCU, uiAbsPartIdx ); }
1128	#endif
1129	}
1130	else
1131	#if LGE_EDGE_INTRA_A0070
1132	if ( uiDir >= EDGE_INTRA_IDX)
1133	{
1134	m_pcBinIf->encodeBin( 0, m_cCUIntraPredSCModel.get( 0, 0, 0 ) );
1135	m_pcBinIf->encodeBinsEP( 63, 6 );
1136	xCodeEdgeIntraInfo( pcCU, uiAbsPartIdx );
1137	#if LGE_EDGE_INTRA_DELTA_DC
1138	m_pcBinIf->encodeBin( (uiDir == EDGE_INTRA_DELTA_IDX), m_cEdgeIntraDeltaDCSCModel.get(0, 0, 0) );
1139	if( uiDir == EDGE_INTRA_DELTA_IDX )
1140	{
1141	Int iDeltaDC0 = pcCU->getEdgeDeltaDC0( uiAbsPartIdx );
1142	Int iDeltaDC1 = pcCU->getEdgeDeltaDC1( uiAbsPartIdx );
1143
1144	xWriteExGolombLevel( UInt( abs( iDeltaDC0 ) ), m_cEdgeIntraDeltaDCSCModel.get(0, 0, 1) );
1145	if ( iDeltaDC0 != 0 )
1146	{
1147	UInt uiSign = iDeltaDC0 > 0 ? 0 : 1;
1148	m_pcBinIf->encodeBinEP( uiSign );
1149	}
1150	xWriteExGolombLevel( UInt( abs( iDeltaDC1 ) ), m_cEdgeIntraDeltaDCSCModel.get(0, 0, 1) );
1151	if ( iDeltaDC1 != 0 )
1152	{
1153	UInt uiSign = iDeltaDC1 > 0 ? 0 : 1;
1154	m_pcBinIf->encodeBinEP( uiSign );
1155	}
1156	}
1157	#endif
1158	}
1159	else
1160	#endif // LGE_EDGE_INTRA
1161	{
1162	#endif
1163	#if !LOGI_INTRA_NAME_3MPM
1164	Int iIntraIdx = pcCU->getIntraSizeIdx(uiAbsPartIdx);
1165	#endif
1166	#if LGE_EDGE_INTRA_A0070
1167	Bool bCodeEdgeIntra = false;
1168	if( pcCU->getSlice()->getSPS()->isDepth() )
1169	{
1170	UInt uiPUWidth = pcCU->getWidth( uiAbsPartIdx ) >> (pcCU->getPartitionSize( uiAbsPartIdx ) == SIZE_NxN ? 1 : 0);
1171	if( uiPUWidth <= LGE_EDGE_INTRA_MAX_SIZE && uiPUWidth >= LGE_EDGE_INTRA_MIN_SIZE )
1172	bCodeEdgeIntra = true;
1173	}
1174	#endif
1175
1176	#if LOGI_INTRA_NAME_3MPM
1177	Int uiPreds[3] = {-1, -1, -1};
1178	#else
1179	Int uiPreds[2] = {-1, -1};
1180	#endif
1181	Int uiPredNum = pcCU->getIntraDirLumaPredictor(uiAbsPartIdx, uiPreds);
1182
1183	Int uiPredIdx = -1;
1184
1185	for(UInt i = 0; i < uiPredNum; i++)
1186	{
1187	if(uiDir == uiPreds[i])
1188	{
1189	uiPredIdx = i;
1190	}
1191	}
1192
1193	if(uiPredIdx != -1)
1194	{
1195	m_pcBinIf->encodeBin( 1, m_cCUIntraPredSCModel.get( 0, 0, 0 ) );
1196	#if LOGI_INTRA_NAME_3MPM
1197	m_pcBinIf->encodeBinEP( uiPredIdx ? 1 : 0 );
1198	if (uiPredIdx)
1199	{
1200	m_pcBinIf->encodeBinEP( uiPredIdx-1 );
1201	}
1202	#else
1203	m_pcBinIf->encodeBinEP( uiPredIdx );
1204	#endif
1205	}
1206	else
1207	{
1208	m_pcBinIf->encodeBin( 0, m_cCUIntraPredSCModel.get( 0, 0, 0 ) );
1209
1210	#if LOGI_INTRA_NAME_3MPM
1211	if (uiPreds[0] > uiPreds[1])
1212	{
1213	std::swap(uiPreds[0], uiPreds[1]);
1214	}
1215	if (uiPreds[0] > uiPreds[2])
1216	{
1217	std::swap(uiPreds[0], uiPreds[2]);
1218	}
1219	if (uiPreds[1] > uiPreds[2])
1220	{
1221	std::swap(uiPreds[1], uiPreds[2]);
1222	}
1223	#endif
1224
1225	for(Int i = (uiPredNum - 1); i >= 0; i--)
1226	{
1227	uiDir = uiDir > uiPreds[i] ? uiDir - 1 : uiDir;
1228	}
1229
1230	#if LOGI_INTRA_NAME_3MPM
1231	m_pcBinIf->encodeBinsEP( uiDir, 5 );
1232	#if LGE_EDGE_INTRA_A0070
1233	if (bCodeEdgeIntra)
1234	if (uiDir == 31) m_pcBinIf->encodeBinsEP(0,1);
1235	#endif
1236	#else
1237	if ( uiDir < 31 )
1238	{
1239	m_pcBinIf->encodeBinsEP( uiDir, g_aucIntraModeBitsAng[ iIntraIdx ] - 1 );
1240	}
1241	else
1242	{
1243	m_pcBinIf->encodeBinsEP( 31, 5 );
1244	m_pcBinIf->encodeBinEP( uiDir - 31 );
1245	}
1246	#endif
1247	}
1248	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX
1249	}
1250	#endif
1251	return;
1252	}
1253
1254	Void TEncSbac::codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx )
1255	{
1256	UInt uiIntraDirChroma = pcCU->getChromaIntraDir( uiAbsPartIdx );
1257
1258	if( uiIntraDirChroma == DM_CHROMA_IDX )
1259	{
1260	m_pcBinIf->encodeBin( 0, m_cCUChromaPredSCModel.get( 0, 0, 0 ) );
1261	}
1262	else if( uiIntraDirChroma == LM_CHROMA_IDX )
1263	{
1264	m_pcBinIf->encodeBin( 1, m_cCUChromaPredSCModel.get( 0, 0, 0 ) );
1265	m_pcBinIf->encodeBin( 0, m_cCUChromaPredSCModel.get( 0, 0, 1 ) );
1266	}
1267	else
1268	{
1269	UInt uiAllowedChromaDir[ NUM_CHROMA_MODE ];
1270	pcCU->getAllowedChromaDir( uiAbsPartIdx, uiAllowedChromaDir );
1271
1272	for( Int i = 0; i < NUM_CHROMA_MODE - 2; i++ )
1273	{
1274	if( uiIntraDirChroma == uiAllowedChromaDir[i] )
1275	{
1276	uiIntraDirChroma = i;
1277	break;
1278	}
1279	}
1280	m_pcBinIf->encodeBin( 1, m_cCUChromaPredSCModel.get( 0, 0, 0 ) );
1281
1282	if (pcCU->getSlice()->getSPS()->getUseLMChroma())
1283	{
1284	m_pcBinIf->encodeBin( 1, m_cCUChromaPredSCModel.get( 0, 0, 1 ));
1285	}
1286	#if CHROMA_MODE_CODING
1287	m_pcBinIf->encodeBinsEP( uiIntraDirChroma, 2 );
1288	#else
1289	xWriteUnaryMaxSymbol( uiIntraDirChroma, m_cCUChromaPredSCModel.get( 0, 0 ) + 1, 0, 3 );
1290	#endif
1291	}
1292	return;
1293	}
1294
1295	Void TEncSbac::codeInterDir( TComDataCU* pcCU, UInt uiAbsPartIdx )
1296	{
1297	const UInt uiInterDir = pcCU->getInterDir( uiAbsPartIdx ) - 1;
1298	const UInt uiCtx = pcCU->getCtxInterDir( uiAbsPartIdx );
1299	ContextModel *pCtx = m_cCUInterDirSCModel.get( 0 );
1300	m_pcBinIf->encodeBin( uiInterDir == 2 ? 1 : 0, *( pCtx + uiCtx ) );
1301
1302	return;
1303	}
1304
1305	Void TEncSbac::codeRefFrmIdx( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1306	{
1307	if ( pcCU->getSlice()->getNumRefIdx(REF_PIC_LIST_C) > 0 && pcCU->getInterDir( uiAbsPartIdx ) != 3)
1308	{
1309	Int iRefFrame = pcCU->getSlice()->getRefIdxOfLC(eRefList, pcCU->getCUMvField( eRefList )->getRefIdx( uiAbsPartIdx ));
1310
1311	ContextModel *pCtx = m_cCURefPicSCModel.get( 0 );
1312	m_pcBinIf->encodeBin( ( iRefFrame == 0 ? 0 : 1 ), *pCtx );
1313
1314	if( iRefFrame > 0 )
1315	{
1316	xWriteUnaryMaxSymbol( iRefFrame - 1, pCtx + 1, 1, pcCU->getSlice()->getNumRefIdx( REF_PIC_LIST_C )-2 );
1317	}
1318	}
1319	else
1320	{
1321	Int iRefFrame = pcCU->getCUMvField( eRefList )->getRefIdx( uiAbsPartIdx );
1322	ContextModel *pCtx = m_cCURefPicSCModel.get( 0 );
1323	m_pcBinIf->encodeBin( ( iRefFrame == 0 ? 0 : 1 ), *pCtx );
1324
1325	if( iRefFrame > 0 )
1326	{
1327	xWriteUnaryMaxSymbol( iRefFrame - 1, pCtx + 1, 1, pcCU->getSlice()->getNumRefIdx( eRefList )-2 );
1328	}
1329	}
1330	return;
1331	}
1332
1333	Void TEncSbac::codeMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1334	{
1335	#if H0111_MVD_L1_ZERO
1336	if(pcCU->getSlice()->getMvdL1ZeroFlag() && eRefList == REF_PIC_LIST_1 && pcCU->getInterDir(uiAbsPartIdx)==3)
1337	{
1338	return;
1339	}
1340	#endif
1341
1342	const TComCUMvField* pcCUMvField = pcCU->getCUMvField( eRefList );
1343	const Int iHor = pcCUMvField->getMvd( uiAbsPartIdx ).getHor();
1344	const Int iVer = pcCUMvField->getMvd( uiAbsPartIdx ).getVer();
1345	ContextModel* pCtx = m_cCUMvdSCModel.get( 0 );
1346
1347	m_pcBinIf->encodeBin( iHor != 0 ? 1 : 0, *pCtx );
1348	m_pcBinIf->encodeBin( iVer != 0 ? 1 : 0, *pCtx );
1349
1350	const Bool bHorAbsGr0 = iHor != 0;
1351	const Bool bVerAbsGr0 = iVer != 0;
1352	const UInt uiHorAbs = 0 > iHor ? -iHor : iHor;
1353	const UInt uiVerAbs = 0 > iVer ? -iVer : iVer;
1354	pCtx++;
1355
1356	if( bHorAbsGr0 )
1357	{
1358	m_pcBinIf->encodeBin( uiHorAbs > 1 ? 1 : 0, *pCtx );
1359	}
1360
1361	if( bVerAbsGr0 )
1362	{
1363	m_pcBinIf->encodeBin( uiVerAbs > 1 ? 1 : 0, *pCtx );
1364	}
1365
1366	if( bHorAbsGr0 )
1367	{
1368	if( uiHorAbs > 1 )
1369	{
1370	xWriteEpExGolomb( uiHorAbs-2, 1 );
1371	}
1372
1373	m_pcBinIf->encodeBinEP( 0 > iHor ? 1 : 0 );
1374	}
1375
1376	if( bVerAbsGr0 )
1377	{
1378	if( uiVerAbs > 1 )
1379	{
1380	xWriteEpExGolomb( uiVerAbs-2, 1 );
1381	}
1382
1383	m_pcBinIf->encodeBinEP( 0 > iVer ? 1 : 0 );
1384	}
1385
1386	return;
1387	}
1388
1389	Void TEncSbac::codeDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx )
1390	{
1391	Int iDQp = pcCU->getQP( uiAbsPartIdx ) - pcCU->getRefQP( uiAbsPartIdx );
1392
1393	#if H0736_AVC_STYLE_QP_RANGE
1394	Int qpBdOffsetY = pcCU->getSlice()->getSPS()->getQpBDOffsetY();
1395	iDQp = (iDQp + 78 + qpBdOffsetY + (qpBdOffsetY/2)) % (52 + qpBdOffsetY) - 26 - (qpBdOffsetY/2);
1396	#else
1397	#if LOSSLESS_CODING
1398	if(pcCU->getSlice()->getSPS()->getUseLossless())
1399	{
1400	if(iDQp > 25)
1401	{
1402	iDQp = iDQp - 52;
1403	}
1404	if(iDQp < -26)
1405	{
1406	iDQp = iDQp + 52;
1407	}
1408	}
1409	#endif
1410	#endif
1411
1412	if ( iDQp == 0 )
1413	{
1414	m_pcBinIf->encodeBin( 0, m_cCUDeltaQpSCModel.get( 0, 0, 0 ) );
1415	}
1416	else
1417	{
1418	m_pcBinIf->encodeBin( 1, m_cCUDeltaQpSCModel.get( 0, 0, 0 ) );
1419
1420	UInt uiSign = (iDQp > 0 ? 0 : 1);
1421	#if !H0736_AVC_STYLE_QP_RANGE
1422	UInt uiQpBdOffsetY = 6*(g_uiBitIncrement + g_uiBitDepth - 8);
1423	#endif
1424
1425	m_pcBinIf->encodeBinEP(uiSign);
1426
1427	#if H0736_AVC_STYLE_QP_RANGE
1428	assert(iDQp >= -(26+(qpBdOffsetY/2)));
1429	assert(iDQp <= (25+(qpBdOffsetY/2)));
1430
1431	UInt uiMaxAbsDQpMinus1 = 24 + (qpBdOffsetY/2) + (uiSign);
1432	#else
1433	assert(iDQp >= -(26+(Int)(uiQpBdOffsetY/2)));
1434	assert(iDQp <= (25+(Int)(uiQpBdOffsetY/2)));
1435
1436	UInt uiMaxAbsDQpMinus1 = 24 + (uiQpBdOffsetY/2) + (uiSign);
1437	#endif
1438	UInt uiAbsDQpMinus1 = (UInt)((iDQp > 0)? iDQp : (-iDQp)) - 1;
1439	xWriteUnaryMaxSymbol( uiAbsDQpMinus1, &m_cCUDeltaQpSCModel.get( 0, 0, 1 ), 1, uiMaxAbsDQpMinus1);
1440	}
1441
1442	return;
1443	}
1444
1445	Void TEncSbac::codeQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth )
1446	{
1447	UInt uiCbf = pcCU->getCbf ( uiAbsPartIdx, eType, uiTrDepth );
1448	UInt uiCtx = pcCU->getCtxQtCbf( uiAbsPartIdx, eType, uiTrDepth );
1449	m_pcBinIf->encodeBin( uiCbf , m_cCUQtCbfSCModel.get( 0, eType ? TEXT_CHROMA : eType, uiCtx ) );
1450	DTRACE_CABAC_VL( g_nSymbolCounter++ )
1451	DTRACE_CABAC_T( "\tparseQtCbf()" )
1452	DTRACE_CABAC_T( "\tsymbol=" )
1453	DTRACE_CABAC_V( uiCbf )
1454	DTRACE_CABAC_T( "\tctx=" )
1455	DTRACE_CABAC_V( uiCtx )
1456	DTRACE_CABAC_T( "\tetype=" )
1457	DTRACE_CABAC_V( eType )
1458	DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1459	DTRACE_CABAC_V( uiAbsPartIdx )
1460	DTRACE_CABAC_T( "\n" )
1461	}
1462
1463	#if BURST_IPCM
1464	/** Code I_PCM information.
1465	* \param pcCU pointer to CU
1466	* \param uiAbsPartIdx CU index
1467	* \param numIPCM the number of succesive IPCM blocks with the same size
1468	* \param firstIPCMFlag
1469	* \returns Void
1470	*/
1471	Void TEncSbac::codeIPCMInfo( TComDataCU* pcCU, UInt uiAbsPartIdx, Int numIPCM, Bool firstIPCMFlag)
1472	#else
1473	/** Code I_PCM information.
1474	* \param pcCU pointer to CU
1475	* \param uiAbsPartIdx CU index
1476	* \returns Void
1477	*
1478	* If I_PCM flag indicates that the CU is I_PCM, code its PCM alignment bits and codes.
1479	*/
1480	Void TEncSbac::codeIPCMInfo( TComDataCU* pcCU, UInt uiAbsPartIdx)
1481	#endif
1482	{
1483	UInt uiIPCM = (pcCU->getIPCMFlag(uiAbsPartIdx) == true)? 1 : 0;
1484
1485	#if BURST_IPCM
1486	Bool writePCMSampleFlag = pcCU->getIPCMFlag(uiAbsPartIdx);
1487
1488	if( uiIPCM == 0 \|\| firstIPCMFlag)
1489	{
1490	m_pcBinIf->encodeBinTrm (uiIPCM);
1491
1492	if ( firstIPCMFlag )
1493	{
1494	m_pcBinIf->encodeNumSubseqIPCM( numIPCM - 1 );
1495	m_pcBinIf->encodePCMAlignBits();
1496	}
1497	}
1498	#else
1499	m_pcBinIf->encodeBinTrm (uiIPCM);
1500	#endif
1501
1502	#if BURST_IPCM
1503	if (writePCMSampleFlag)
1504	#else
1505	if (uiIPCM)
1506	#endif
1507	{
1508	#if !BURST_IPCM
1509	m_pcBinIf->encodePCMAlignBits();
1510	#endif
1511	UInt uiMinCoeffSize = pcCU->getPic()->getMinCUWidth()*pcCU->getPic()->getMinCUHeight();
1512	UInt uiLumaOffset = uiMinCoeffSize*uiAbsPartIdx;
1513	UInt uiChromaOffset = uiLumaOffset>>2;
1514	Pel* piPCMSample;
1515	UInt uiWidth;
1516	UInt uiHeight;
1517	UInt uiSampleBits;
1518	UInt uiX, uiY;
1519
1520	piPCMSample = pcCU->getPCMSampleY() + uiLumaOffset;
1521	uiWidth = pcCU->getWidth(uiAbsPartIdx);
1522	uiHeight = pcCU->getHeight(uiAbsPartIdx);
1523	uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthLuma();
1524
1525	for(uiY = 0; uiY < uiHeight; uiY++)
1526	{
1527	for(uiX = 0; uiX < uiWidth; uiX++)
1528	{
1529	UInt uiSample = piPCMSample[uiX];
1530
1531	m_pcBinIf->xWritePCMCode(uiSample, uiSampleBits);
1532	}
1533	piPCMSample += uiWidth;
1534	}
1535
1536	piPCMSample = pcCU->getPCMSampleCb() + uiChromaOffset;
1537	uiWidth = pcCU->getWidth(uiAbsPartIdx)/2;
1538	uiHeight = pcCU->getHeight(uiAbsPartIdx)/2;
1539	uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
1540
1541	for(uiY = 0; uiY < uiHeight; uiY++)
1542	{
1543	for(uiX = 0; uiX < uiWidth; uiX++)
1544	{
1545	UInt uiSample = piPCMSample[uiX];
1546
1547	m_pcBinIf->xWritePCMCode(uiSample, uiSampleBits);
1548	}
1549	piPCMSample += uiWidth;
1550	}
1551
1552	piPCMSample = pcCU->getPCMSampleCr() + uiChromaOffset;
1553	uiWidth = pcCU->getWidth(uiAbsPartIdx)/2;
1554	uiHeight = pcCU->getHeight(uiAbsPartIdx)/2;
1555	uiSampleBits = pcCU->getSlice()->getSPS()->getPCMBitDepthChroma();
1556
1557	for(uiY = 0; uiY < uiHeight; uiY++)
1558	{
1559	for(uiX = 0; uiX < uiWidth; uiX++)
1560	{
1561	UInt uiSample = piPCMSample[uiX];
1562
1563	m_pcBinIf->xWritePCMCode(uiSample, uiSampleBits);
1564	}
1565	piPCMSample += uiWidth;
1566	}
1567	#if BURST_IPCM
1568	numIPCM--;
1569	if(numIPCM == 0)
1570	{
1571	m_pcBinIf->resetBac();
1572	}
1573	#else
1574	m_pcBinIf->resetBac();
1575	#endif
1576	}
1577	}
1578
1579	Void TEncSbac::codeQtRootCbf( TComDataCU* pcCU, UInt uiAbsPartIdx )
1580	{
1581	UInt uiCbf = pcCU->getQtRootCbf( uiAbsPartIdx );
1582	UInt uiCtx = 0;
1583	m_pcBinIf->encodeBin( uiCbf , m_cCUQtRootCbfSCModel.get( 0, 0, uiCtx ) );
1584	DTRACE_CABAC_VL( g_nSymbolCounter++ )
1585	DTRACE_CABAC_T( "\tparseQtRootCbf()" )
1586	DTRACE_CABAC_T( "\tsymbol=" )
1587	DTRACE_CABAC_V( uiCbf )
1588	DTRACE_CABAC_T( "\tctx=" )
1589	DTRACE_CABAC_V( uiCtx )
1590	DTRACE_CABAC_T( "\tuiAbsPartIdx=" )
1591	DTRACE_CABAC_V( uiAbsPartIdx )
1592	DTRACE_CABAC_T( "\n" )
1593	}
1594
1595	/** Encode (X,Y) position of the last significant coefficient
1596	* \param uiPosX X component of last coefficient
1597	* \param uiPosY Y component of last coefficient
1598	* \param width Block width
1599	* \param height Block height
1600	* \param eTType plane type / luminance or chrominance
1601	* \param uiScanIdx scan type (zig-zag, hor, ver)
1602	* This method encodes the X and Y component within a block of the last significant coefficient.
1603	*/
1604	Void TEncSbac::codeLastSignificantXY( UInt uiPosX, UInt uiPosY, Int width, Int height, TextType eTType, UInt uiScanIdx )
1605	{
1606	// swap
1607	if( uiScanIdx == SCAN_VER )
1608	{
1609	swap( uiPosX, uiPosY );
1610	}
1611
1612	UInt uiCtxLast;
1613	ContextModel *pCtxX = m_cCuCtxLastX.get( 0, eTType );
1614	ContextModel *pCtxY = m_cCuCtxLastY.get( 0, eTType );
1615	UInt uiGroupIdxX = g_uiGroupIdx[ uiPosX ];
1616	UInt uiGroupIdxY = g_uiGroupIdx[ uiPosY ];
1617
1618	// posX
1619	#if LAST_CTX_REDUCTION
1620	Int widthCtx = eTType? 4: width;
1621	const UInt puiCtxIdxX = g_uiLastCtx + ( g_aucConvertToBit[ widthCtx ] ( g_aucConvertToBit[ widthCtx ] + 3 ) );
1622	#else
1623	const UInt puiCtxIdxX = g_uiLastCtx + ( g_aucConvertToBit[ width ] ( g_aucConvertToBit[ width ] + 3 ) );
1624	#endif
1625	for( uiCtxLast = 0; uiCtxLast < uiGroupIdxX; uiCtxLast++ )
1626	{
1627	#if LAST_CTX_REDUCTION
1628	if (eTType)
1629	{
1630	m_pcBinIf->encodeBin( 1, *( pCtxX + (uiCtxLast>>g_aucConvertToBit[ width ]) ) );
1631	}
1632	else
1633	{
1634	#endif
1635	m_pcBinIf->encodeBin( 1, *( pCtxX + puiCtxIdxX[ uiCtxLast ] ) );
1636	#if LAST_CTX_REDUCTION
1637	}
1638	#endif
1639	}
1640	if( uiGroupIdxX < g_uiGroupIdx[ width - 1 ])
1641	{
1642	#if LAST_CTX_REDUCTION
1643	if ( eTType )
1644	{
1645	m_pcBinIf->encodeBin( 0, *( pCtxX + (uiCtxLast>>g_aucConvertToBit[ width ]) ) );
1646	}
1647	else
1648	{
1649	#endif
1650	m_pcBinIf->encodeBin( 0, *( pCtxX + puiCtxIdxX[ uiCtxLast ] ) );
1651	#if LAST_CTX_REDUCTION
1652	}
1653	#endif
1654	}
1655
1656	// posY
1657	#if LAST_CTX_REDUCTION
1658	Int heightCtx = eTType? 4: height;
1659	const UInt puiCtxIdxY = g_uiLastCtx + ( g_aucConvertToBit[ heightCtx ] ( g_aucConvertToBit[ heightCtx ] + 3 ) );
1660	#else
1661	const UInt puiCtxIdxY = g_uiLastCtx + ( g_aucConvertToBit[ height ] ( g_aucConvertToBit[ height ] + 3 ) );
1662	#endif
1663	for( uiCtxLast = 0; uiCtxLast < uiGroupIdxY; uiCtxLast++ )
1664	{
1665	#if LAST_CTX_REDUCTION
1666	if (eTType)
1667	{
1668	m_pcBinIf->encodeBin( 1, *( pCtxY + (uiCtxLast>>g_aucConvertToBit[ height ])));
1669	}
1670	else
1671	{
1672	#endif
1673	m_pcBinIf->encodeBin( 1, *( pCtxY + puiCtxIdxY[ uiCtxLast ] ) );
1674	#if LAST_CTX_REDUCTION
1675	}
1676	#endif
1677	}
1678	if( uiGroupIdxY < g_uiGroupIdx[ height - 1 ])
1679	{
1680	#if LAST_CTX_REDUCTION
1681	if (eTType)
1682	{
1683	m_pcBinIf->encodeBin( 0, *( pCtxY + (uiCtxLast>>g_aucConvertToBit[ height ]) ) );
1684	}
1685	else
1686	{
1687	#endif
1688	m_pcBinIf->encodeBin( 0, *( pCtxY + puiCtxIdxY[ uiCtxLast ] ) );
1689	#if LAST_CTX_REDUCTION
1690	}
1691	#endif
1692	}
1693	if ( uiGroupIdxX > 3 )
1694	{
1695	UInt uiCount = ( uiGroupIdxX - 2 ) >> 1;
1696	uiPosX = uiPosX - g_uiMinInGroup[ uiGroupIdxX ];
1697	for (Int i = uiCount - 1 ; i >= 0; i-- )
1698	{
1699	m_pcBinIf->encodeBinEP( ( uiPosX >> i ) & 1 );
1700	}
1701	}
1702	if ( uiGroupIdxY > 3 )
1703	{
1704	UInt uiCount = ( uiGroupIdxY - 2 ) >> 1;
1705	uiPosY = uiPosY - g_uiMinInGroup[ uiGroupIdxY ];
1706	for ( Int i = uiCount - 1 ; i >= 0; i-- )
1707	{
1708	m_pcBinIf->encodeBinEP( ( uiPosY >> i ) & 1 );
1709	}
1710	}
1711	}
1712
1713	Void TEncSbac::codeCoeffNxN( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
1714	{
1715	DTRACE_CABAC_VL( g_nSymbolCounter++ )
1716	DTRACE_CABAC_T( "\tparseCoeffNxN()\teType=" )
1717	DTRACE_CABAC_V( eTType )
1718	DTRACE_CABAC_T( "\twidth=" )
1719	DTRACE_CABAC_V( uiWidth )
1720	DTRACE_CABAC_T( "\theight=" )
1721	DTRACE_CABAC_V( uiHeight )
1722	DTRACE_CABAC_T( "\tdepth=" )
1723	DTRACE_CABAC_V( uiDepth )
1724	DTRACE_CABAC_T( "\tabspartidx=" )
1725	DTRACE_CABAC_V( uiAbsPartIdx )
1726	DTRACE_CABAC_T( "\ttoCU-X=" )
1727	DTRACE_CABAC_V( pcCU->getCUPelX() )
1728	DTRACE_CABAC_T( "\ttoCU-Y=" )
1729	DTRACE_CABAC_V( pcCU->getCUPelY() )
1730	DTRACE_CABAC_T( "\tCU-addr=" )
1731	DTRACE_CABAC_V( pcCU->getAddr() )
1732	DTRACE_CABAC_T( "\tinCU-X=" )
1733	DTRACE_CABAC_V( g_auiRasterToPelX[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1734	DTRACE_CABAC_T( "\tinCU-Y=" )
1735	DTRACE_CABAC_V( g_auiRasterToPelY[ g_auiZscanToRaster[uiAbsPartIdx] ] )
1736	DTRACE_CABAC_T( "\tpredmode=" )
1737	DTRACE_CABAC_V( pcCU->getPredictionMode( uiAbsPartIdx ) )
1738	DTRACE_CABAC_T( "\n" )
1739
1740	if( uiWidth > m_pcSlice->getSPS()->getMaxTrSize() )
1741	{
1742	uiWidth = m_pcSlice->getSPS()->getMaxTrSize();
1743	uiHeight = m_pcSlice->getSPS()->getMaxTrSize();
1744	}
1745
1746	UInt uiNumSig = 0;
1747
1748	// compute number of significant coefficients
1749	uiNumSig = TEncEntropy::countNonZeroCoeffs(pcCoef, uiWidth * uiHeight);
1750
1751	if ( uiNumSig == 0 )
1752	return;
1753
1754	eTType = eTType == TEXT_LUMA ? TEXT_LUMA : ( eTType == TEXT_NONE ? TEXT_NONE : TEXT_CHROMA );
1755
1756	//----- encode significance map -----
1757	const UInt uiLog2BlockSize = g_aucConvertToBit[ uiWidth ] + 2;
1758	UInt uiScanIdx = pcCU->getCoefScanIdx(uiAbsPartIdx, uiWidth, eTType==TEXT_LUMA, pcCU->isIntra(uiAbsPartIdx));
1759	if (uiScanIdx == SCAN_ZIGZAG)
1760	{
1761	// Map zigzag to diagonal scan
1762	uiScanIdx = SCAN_DIAG;
1763	}
1764	Int blockType = uiLog2BlockSize;
1765	if (uiWidth != uiHeight)
1766	{
1767	uiScanIdx = SCAN_DIAG;
1768	blockType = 4;
1769	}
1770
1771	const UInt * scan;
1772	if (uiWidth == uiHeight)
1773	{
1774	scan = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize - 1 ];
1775	}
1776	else
1777	{
1778	scan = g_sigScanNSQT[ uiLog2BlockSize - 2 ];
1779	}
1780
1781	#if MULTIBITS_DATA_HIDING
1782	UInt const tsig = pcCU->getSlice()->getPPS()->getTSIG();
1783	#if LOSSLESS_CODING
1784	Bool beValid;
1785	if (pcCU->isLosslessCoded(uiAbsPartIdx))
1786	{
1787	beValid = false;
1788	}
1789	else
1790	{
1791	beValid = pcCU->getSlice()->getPPS()->getSignHideFlag() > 0;
1792	}
1793	#else
1794	Bool beValid = pcCU->getSlice()->getPPS()->getSignHideFlag() > 0;
1795	#endif
1796	#endif
1797
1798	// Find position of last coefficient
1799	Int scanPosLast = -1;
1800	Int posLast;
1801
1802	const UInt * scanCG;
1803	if (uiWidth == uiHeight)
1804	{
1805	scanCG = g_auiSigLastScan[ uiScanIdx ][ uiLog2BlockSize > 3 ? uiLog2BlockSize-2-1 : 0 ];
1806	#if MULTILEVEL_SIGMAP_EXT
1807	if( uiLog2BlockSize == 3 )
1808	{
1809	scanCG = g_sigLastScan8x8[ uiScanIdx ];
1810	}
1811	else if( uiLog2BlockSize == 5 )
1812	{
1813	scanCG = g_sigLastScanCG32x32;
1814	}
1815	#endif
1816	}
1817	else
1818	{
1819	scanCG = g_sigCGScanNSQT[ uiLog2BlockSize - 2 ];
1820	}
1821	UInt uiSigCoeffGroupFlag[ MLS_GRP_NUM ];
1822	static const UInt uiShift = MLS_CG_SIZE >> 1;
1823	const UInt uiNumBlkSide = uiWidth >> uiShift;
1824
1825	#if !MULTILEVEL_SIGMAP_EXT
1826	if( blockType > 3 )
1827	{
1828	#endif
1829	::memset( uiSigCoeffGroupFlag, 0, sizeof(UInt) * MLS_GRP_NUM );
1830
1831	do
1832	{
1833	posLast = scan[ ++scanPosLast ];
1834
1835	// get L1 sig map
1836	UInt uiPosY = posLast >> uiLog2BlockSize;
1837	UInt uiPosX = posLast - ( uiPosY << uiLog2BlockSize );
1838	UInt uiBlkIdx = uiNumBlkSide * (uiPosY >> uiShift) + (uiPosX >> uiShift);
1839	#if MULTILEVEL_SIGMAP_EXT
1840	if( uiWidth == 8 && uiHeight == 8 && (uiScanIdx == SCAN_HOR \|\| uiScanIdx == SCAN_VER) )
1841	{
1842	if( uiScanIdx == SCAN_HOR )
1843	{
1844	uiBlkIdx = uiPosY >> 1;
1845	}
1846	else if( uiScanIdx == SCAN_VER )
1847	{
1848	uiBlkIdx = uiPosX >> 1;
1849	}
1850	}
1851	#endif
1852	if( pcCoef[ posLast ] )
1853	{
1854	uiSigCoeffGroupFlag[ uiBlkIdx ] = 1;
1855	}
1856
1857	uiNumSig -= ( pcCoef[ posLast ] != 0 );
1858	}
1859	while ( uiNumSig > 0 );
1860	#if !MULTILEVEL_SIGMAP_EXT
1861	}
1862	else
1863	{
1864
1865	do
1866	{
1867	posLast = scan[ ++scanPosLast ];
1868	uiNumSig -= ( pcCoef[ posLast ] != 0 );
1869	}
1870	while ( uiNumSig > 0 );
1871
1872	}
1873	#endif
1874
1875	// Code position of last coefficient
1876	Int posLastY = posLast >> uiLog2BlockSize;
1877	Int posLastX = posLast - ( posLastY << uiLog2BlockSize );
1878	codeLastSignificantXY(posLastX, posLastY, uiWidth, uiHeight, eTType, uiScanIdx);
1879
1880	//===== code significance flag =====
1881	ContextModel * const baseCoeffGroupCtx = m_cCUSigCoeffGroupSCModel.get( 0, eTType );
1882	ContextModel * const baseCtx = (eTType==TEXT_LUMA) ? m_cCUSigSCModel.get( 0, 0 ) : m_cCUSigSCModel.get( 0, 0 ) + NUM_SIG_FLAG_CTX_LUMA;
1883
1884
1885	const Int iLastScanSet = scanPosLast >> LOG2_SCAN_SET_SIZE;
1886	UInt uiNumOne = 0;
1887	UInt uiGoRiceParam = 0;
1888	Int iScanPosSig = scanPosLast;
1889
1890	for( Int iSubSet = iLastScanSet; iSubSet >= 0; iSubSet-- )
1891	{
1892	Int numNonZero = 0;
1893	Int iSubPos = iSubSet << LOG2_SCAN_SET_SIZE;
1894	uiGoRiceParam = 0;
1895	Int absCoeff[16];
1896	UInt coeffSigns = 0;
1897
1898	#if MULTIBITS_DATA_HIDING
1899	Int lastNZPosInCG = -1, firstNZPosInCG = SCAN_SET_SIZE;
1900	#endif
1901
1902	if( iScanPosSig == scanPosLast )
1903	{
1904	absCoeff[ 0 ] = abs( pcCoef[ posLast ] );
1905	coeffSigns = ( pcCoef[ posLast ] < 0 );
1906	numNonZero = 1;
1907	#if MULTIBITS_DATA_HIDING
1908	lastNZPosInCG = iScanPosSig;
1909	firstNZPosInCG = iScanPosSig;
1910	#endif
1911	iScanPosSig--;
1912	}
1913
1914	#if !MULTILEVEL_SIGMAP_EXT
1915	if( blockType > 3 )
1916	{
1917	#endif
1918	// encode significant_coeffgroup_flag
1919	Int iCGBlkPos = scanCG[ iSubSet ];
1920	Int iCGPosY = iCGBlkPos / uiNumBlkSide;
1921	Int iCGPosX = iCGBlkPos - (iCGPosY * uiNumBlkSide);
1922	#if MULTILEVEL_SIGMAP_EXT
1923	if( uiWidth == 8 && uiHeight == 8 && (uiScanIdx == SCAN_HOR \|\| uiScanIdx == SCAN_VER) )
1924	{
1925	iCGPosY = (uiScanIdx == SCAN_HOR ? iCGBlkPos : 0);
1926	iCGPosX = (uiScanIdx == SCAN_VER ? iCGBlkPos : 0);
1927	}
1928	#endif
1929	#if !REMOVE_INFER_SIGGRP
1930	Bool bInferredCGFlag = false;
1931	#endif
1932	#if REMOVE_INFER_SIGGRP
1933	if( iSubSet == iLastScanSet \|\| iSubSet == 0)
1934	#else
1935	if( iSubSet == iLastScanSet )
1936	#endif
1937	{
1938	uiSigCoeffGroupFlag[ iCGBlkPos ] = 1;
1939	}
1940	else
1941	{
1942	#if !REMOVE_INFER_SIGGRP
1943	#if MULTILEVEL_SIGMAP_EXT
1944	if( !TComTrQuant::bothCGNeighboursOne( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiScanIdx, uiWidth, uiHeight ) && ( iSubSet ) )
1945	#else
1946	if( !TComTrQuant::bothCGNeighboursOne( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight ) && ( iSubSet ) )
1947	#endif
1948	{
1949	#endif
1950	UInt uiSigCoeffGroup = (uiSigCoeffGroupFlag[ iCGBlkPos ] != 0);
1951	#if MULTILEVEL_SIGMAP_EXT
1952	UInt uiCtxSig = TComTrQuant::getSigCoeffGroupCtxInc( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiScanIdx, uiWidth, uiHeight );
1953	#else
1954	UInt uiCtxSig = TComTrQuant::getSigCoeffGroupCtxInc( uiSigCoeffGroupFlag, iCGPosX, iCGPosY, uiWidth, uiHeight );
1955	#endif
1956	m_pcBinIf->encodeBin( uiSigCoeffGroup, baseCoeffGroupCtx[ uiCtxSig ] );
1957	#if !REMOVE_INFER_SIGGRP
1958	}
1959	else
1960	{
1961	uiSigCoeffGroupFlag[ iCGBlkPos ] = 1;
1962	bInferredCGFlag = true;
1963	}
1964	#endif
1965	}
1966
1967	// encode significant_coeff_flag
1968	if( uiSigCoeffGroupFlag[ iCGBlkPos ] )
1969	{
1970	UInt uiBlkPos, uiPosY, uiPosX, uiSig, uiCtxSig;
1971	for( ; iScanPosSig >= iSubPos; iScanPosSig-- )
1972	{
1973	uiBlkPos = scan[ iScanPosSig ];
1974	uiPosY = uiBlkPos >> uiLog2BlockSize;
1975	uiPosX = uiBlkPos - ( uiPosY << uiLog2BlockSize );
1976	uiSig = (pcCoef[ uiBlkPos ] != 0);
1977	#if REMOVE_INFER_SIGGRP
1978	if( iScanPosSig > iSubPos \|\| iSubSet == 0 \|\| numNonZero )
1979	#else
1980	if( iScanPosSig > iSubPos \|\| bInferredCGFlag \|\| numNonZero )
1981	#endif
1982	{
1983	uiCtxSig = TComTrQuant::getSigCtxInc( pcCoef, uiPosX, uiPosY, blockType, uiWidth, uiHeight, eTType );
1984	m_pcBinIf->encodeBin( uiSig, baseCtx[ uiCtxSig ] );
1985	}
1986	if( uiSig )
1987	{
1988	absCoeff[ numNonZero ] = abs( pcCoef[ uiBlkPos ] );
1989	coeffSigns = 2 * coeffSigns + ( pcCoef[ uiBlkPos ] < 0 );
1990	numNonZero++;
1991	#if MULTIBITS_DATA_HIDING
1992	if( lastNZPosInCG == -1 )
1993	{
1994	lastNZPosInCG = iScanPosSig;
1995	}
1996	firstNZPosInCG = iScanPosSig;
1997	#endif
1998	}
1999	}
2000	}
2001	else
2002	{
2003	iScanPosSig = iSubPos - 1;
2004	}
2005	#if !MULTILEVEL_SIGMAP_EXT
2006	}
2007	else
2008	{
2009
2010	for( ; iScanPosSig >= iSubPos; iScanPosSig-- )
2011	{
2012	UInt uiBlkPos = scan[ iScanPosSig ];
2013	UInt uiPosY = uiBlkPos >> uiLog2BlockSize;
2014	UInt uiPosX = uiBlkPos - ( uiPosY << uiLog2BlockSize );
2015	UInt uiSig = 0;
2016	if( pcCoef[ uiBlkPos ] != 0 )
2017	{
2018	uiSig = 1;
2019	absCoeff[ numNonZero ] = abs( pcCoef[ uiBlkPos ] );
2020	coeffSigns = 2 * coeffSigns + ( pcCoef[ uiBlkPos ] < 0 );
2021	numNonZero++;
2022	#if MULTIBITS_DATA_HIDING
2023	if( lastNZPosInCG == -1 )
2024	{
2025	lastNZPosInCG = iScanPosSig;
2026	}
2027	firstNZPosInCG = iScanPosSig;
2028	#endif
2029	}
2030	UInt uiCtxSig = TComTrQuant::getSigCtxInc( pcCoef, uiPosX, uiPosY, blockType, uiWidth, uiHeight, eTType );
2031	m_pcBinIf->encodeBin( uiSig, baseCtx[ uiCtxSig ] );
2032	}
2033
2034	}
2035	#endif
2036
2037	if( numNonZero > 0 )
2038	{
2039	#if MULTIBITS_DATA_HIDING
2040	Bool signHidden = ( lastNZPosInCG - firstNZPosInCG >= (Int)tsig );
2041	#endif // MULTIBITS_DATA_HIDING
2042
2043	UInt c1 = 1;
2044	#if !RESTRICT_GR1GR2FLAG_NUMBER
2045	UInt c2 = 0;
2046	#endif
2047	#if LEVEL_CTX_LUMA_RED
2048	UInt uiCtxSet = (iSubSet > 0 && eTType==TEXT_LUMA) ? 2 : 0;
2049	#else
2050	UInt uiCtxSet = (iSubSet > 0 && eTType==TEXT_LUMA) ? 3 : 0;
2051	#endif
2052
2053	if( uiNumOne > 0 )
2054	{
2055	uiCtxSet++;
2056	#if !LEVEL_CTX_LUMA_RED
2057	if( uiNumOne > 3 && eTType==TEXT_LUMA)
2058	{
2059	uiCtxSet++;
2060	}
2061	#endif
2062	}
2063
2064	uiNumOne >>= 1;
2065	ContextModel baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUOneSCModel.get( 0, 0 ) + 4 uiCtxSet : m_cCUOneSCModel.get( 0, 0 ) + NUM_ONE_FLAG_CTX_LUMA + 4 * uiCtxSet;
2066
2067	#if RESTRICT_GR1GR2FLAG_NUMBER
2068	Int numC1Flag = min(numNonZero, C1FLAG_NUMBER);
2069	Int firstC2FlagIdx = -1;
2070	for( Int idx = 0; idx < numC1Flag; idx++ )
2071	#else
2072	for ( Int idx = 0; idx < numNonZero; idx++ )
2073	#endif
2074	{
2075	UInt uiSymbol = absCoeff[ idx ] > 1;
2076	m_pcBinIf->encodeBin( uiSymbol, baseCtxMod[c1] );
2077	if( uiSymbol )
2078	{
2079	c1 = 0;
2080
2081	#if RESTRICT_GR1GR2FLAG_NUMBER
2082	if (firstC2FlagIdx == -1)
2083	{
2084	firstC2FlagIdx = idx;
2085	}
2086	#endif
2087	}
2088	else if( (c1 < 3) && (c1 > 0) )
2089	{
2090	c1++;
2091	}
2092	}
2093
2094	if (c1 == 0)
2095	{
2096
2097	#if RESTRICT_GR1GR2FLAG_NUMBER
2098	baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUAbsSCModel.get( 0, 0 ) + uiCtxSet : m_cCUAbsSCModel.get( 0, 0 ) + NUM_ABS_FLAG_CTX_LUMA + uiCtxSet;
2099	if ( firstC2FlagIdx != -1)
2100	{
2101	UInt symbol = absCoeff[ firstC2FlagIdx ] > 2;
2102	m_pcBinIf->encodeBin( symbol, baseCtxMod[0] );
2103	}
2104	#else
2105	baseCtxMod = ( eTType==TEXT_LUMA ) ? m_cCUAbsSCModel.get( 0, 0 ) + 3 * uiCtxSet : m_cCUAbsSCModel.get( 0, 0 ) + NUM_ABS_FLAG_CTX_LUMA + 3 * uiCtxSet;
2106	for ( Int idx = 0; idx < numNonZero; idx++ )
2107	{
2108	if( absCoeff[ idx ] > 1 )
2109	{
2110	UInt symbol = absCoeff[ idx ] > 2;
2111	m_pcBinIf->encodeBin( symbol, baseCtxMod[c2] );
2112	c2 += (c2 < 2);
2113	uiNumOne++;
2114	}
2115	}
2116	#endif
2117	}
2118
2119	#if MULTIBITS_DATA_HIDING
2120	if( beValid && signHidden )
2121	{
2122	m_pcBinIf->encodeBinsEP( (coeffSigns >> 1), numNonZero-1 );
2123	}
2124	else
2125	{
2126	m_pcBinIf->encodeBinsEP( coeffSigns, numNonZero );
2127	}
2128	#else
2129	m_pcBinIf->encodeBinsEP( coeffSigns, numNonZero );
2130	#endif
2131
2132	#if RESTRICT_GR1GR2FLAG_NUMBER
2133	Int iFirstCoeff2 = 1;
2134	if (c1 == 0 \|\| numNonZero > C1FLAG_NUMBER)
2135	#else
2136	if (c1 == 0)
2137	#endif
2138	{
2139	for ( Int idx = 0; idx < numNonZero; idx++ )
2140	{
2141	#if RESTRICT_GR1GR2FLAG_NUMBER
2142	UInt baseLevel = (idx < C1FLAG_NUMBER)? (2 + iFirstCoeff2 ) : 1;
2143
2144	if( absCoeff[ idx ] >= baseLevel)
2145	{
2146	xWriteGoRiceExGolomb( absCoeff[ idx ] - baseLevel, uiGoRiceParam );
2147	}
2148	if(absCoeff[ idx ] >= 2)
2149	{
2150	iFirstCoeff2 = 0;
2151	uiNumOne++;
2152	}
2153	#else
2154	if ( absCoeff[ idx ] > 2 )
2155	{
2156	xWriteGoRiceExGolomb( absCoeff[ idx ] - 3, uiGoRiceParam );
2157	}
2158	#endif
2159	}
2160	}
2161	}
2162	else
2163	{
2164	uiNumOne >>= 1;
2165	}
2166	}
2167
2168	return;
2169	}
2170
2171	Void TEncSbac::codeAlfFlag ( UInt uiCode )
2172	{
2173	UInt uiSymbol = ( ( uiCode == 0 ) ? 0 : 1 );
2174	m_pcBinIf->encodeBin( uiSymbol, m_cALFFlagSCModel.get( 0, 0, 0 ) );
2175	}
2176
2177	Void TEncSbac::codeAlfCtrlFlag( UInt uiSymbol )
2178	{
2179	m_pcBinIf->encodeBin( uiSymbol, m_cCUAlfCtrlFlagSCModel.get( 0, 0, 0) );
2180	}
2181
2182	Void TEncSbac::codeAlfUvlc ( UInt uiCode )
2183	{
2184	Int i;
2185
2186	if ( uiCode == 0 )
2187	{
2188	m_pcBinIf->encodeBin( 0, m_cALFUvlcSCModel.get( 0, 0, 0 ) );
2189	}
2190	else
2191	{
2192	m_pcBinIf->encodeBin( 1, m_cALFUvlcSCModel.get( 0, 0, 0 ) );
2193	for ( i=0; i<uiCode-1; i++ )
2194	{
2195	m_pcBinIf->encodeBin( 1, m_cALFUvlcSCModel.get( 0, 0, 1 ) );
2196	}
2197	m_pcBinIf->encodeBin( 0, m_cALFUvlcSCModel.get( 0, 0, 1 ) );
2198	}
2199	}
2200
2201	Void TEncSbac::codeAlfSvlc ( Int iCode )
2202	{
2203	Int i;
2204
2205	if ( iCode == 0 )
2206	{
2207	m_pcBinIf->encodeBin( 0, m_cALFSvlcSCModel.get( 0, 0, 0 ) );
2208	}
2209	else
2210	{
2211	m_pcBinIf->encodeBin( 1, m_cALFSvlcSCModel.get( 0, 0, 0 ) );
2212
2213	// write sign
2214	if ( iCode > 0 )
2215	{
2216	m_pcBinIf->encodeBin( 0, m_cALFSvlcSCModel.get( 0, 0, 1 ) );
2217	}
2218	else
2219	{
2220	m_pcBinIf->encodeBin( 1, m_cALFSvlcSCModel.get( 0, 0, 1 ) );
2221	iCode = -iCode;
2222	}
2223
2224	// write magnitude
2225	for ( i=0; i<iCode-1; i++ )
2226	{
2227	m_pcBinIf->encodeBin( 1, m_cALFSvlcSCModel.get( 0, 0, 2 ) );
2228	}
2229	m_pcBinIf->encodeBin( 0, m_cALFSvlcSCModel.get( 0, 0, 2 ) );
2230	}
2231	}
2232
2233	Void TEncSbac::codeSaoFlag ( UInt uiCode )
2234	{
2235	UInt uiSymbol = ( ( uiCode == 0 ) ? 0 : 1 );
2236	m_pcBinIf->encodeBin( uiSymbol, m_cSaoFlagSCModel.get( 0, 0, 0 ) );
2237	}
2238
2239	Void TEncSbac::codeSaoUvlc ( UInt uiCode )
2240	{
2241	Int i;
2242
2243	if ( uiCode == 0 )
2244	{
2245	m_pcBinIf->encodeBin( 0, m_cSaoUvlcSCModel.get( 0, 0, 0 ) );
2246	}
2247	else
2248	{
2249	m_pcBinIf->encodeBin( 1, m_cSaoUvlcSCModel.get( 0, 0, 0 ) );
2250	for ( i=0; i<uiCode-1; i++ )
2251	{
2252	m_pcBinIf->encodeBin( 1, m_cSaoUvlcSCModel.get( 0, 0, 1 ) );
2253	}
2254	m_pcBinIf->encodeBin( 0, m_cSaoUvlcSCModel.get( 0, 0, 1 ) );
2255	}
2256	}
2257
2258	Void TEncSbac::codeSaoSvlc ( Int iCode )
2259	{
2260	Int i;
2261
2262	if ( iCode == 0 )
2263	{
2264	m_pcBinIf->encodeBin( 0, m_cSaoSvlcSCModel.get( 0, 0, 0 ) );
2265	}
2266	else
2267	{
2268	m_pcBinIf->encodeBin( 1, m_cSaoSvlcSCModel.get( 0, 0, 0 ) );
2269
2270	// write sign
2271	if ( iCode > 0 )
2272	{
2273	m_pcBinIf->encodeBin( 0, m_cSaoSvlcSCModel.get( 0, 0, 1 ) );
2274	}
2275	else
2276	{
2277	m_pcBinIf->encodeBin( 1, m_cSaoSvlcSCModel.get( 0, 0, 1 ) );
2278	iCode = -iCode;
2279	}
2280
2281	// write magnitude
2282	for ( i=0; i<iCode-1; i++ )
2283	{
2284	m_pcBinIf->encodeBin( 1, m_cSaoSvlcSCModel.get( 0, 0, 2 ) );
2285	}
2286	m_pcBinIf->encodeBin( 0, m_cSaoSvlcSCModel.get( 0, 0, 2 ) );
2287	}
2288	}
2289	#if SAO_UNIT_INTERLEAVING
2290	/** Code SAO band position
2291	* \param uiCode
2292	*/
2293	Void TEncSbac::codeSaoUflc ( UInt uiCode )
2294	{
2295	for (Int i=0;i<5;i++)
2296	{
2297	m_pcBinIf->encodeBinEP ( (uiCode>>i) &0x01 );
2298	}
2299	}
2300	/** Code SAO merge left flag
2301	* \param uiCode
2302	* \param uiCompIdx
2303	*/
2304	Void TEncSbac::codeSaoMergeLeft ( UInt uiCode, UInt uiCompIdx )
2305	{
2306	if (uiCode == 0)
2307	{
2308	m_pcBinIf->encodeBin(0, m_cSaoMergeLeftSCModel.get( 0, 0, uiCompIdx ));
2309	}
2310	else
2311	{
2312	m_pcBinIf->encodeBin(1, m_cSaoMergeLeftSCModel.get( 0, 0, uiCompIdx ));
2313	}
2314	}
2315	/** Code SAO merge up flag
2316	* \param uiCode
2317	*/
2318	Void TEncSbac::codeSaoMergeUp ( UInt uiCode)
2319	{
2320	if (uiCode == 0)
2321	{
2322	m_pcBinIf->encodeBin(0, m_cSaoMergeUpSCModel.get( 0, 0, 0 ));
2323	}
2324	else
2325	{
2326	m_pcBinIf->encodeBin(1, m_cSaoMergeUpSCModel.get( 0, 0, 0 ));
2327	}
2328	}
2329	/** Code SAO type index
2330	* \param uiCode
2331	*/
2332	Void TEncSbac::codeSaoTypeIdx ( UInt uiCode)
2333	{
2334	Int i;
2335	if ( uiCode == 0 )
2336	{
2337	m_pcBinIf->encodeBin( 0, m_cSaoTypeIdxSCModel.get( 0, 0, 0 ) );
2338	}
2339	else
2340	{
2341	m_pcBinIf->encodeBin( 1, m_cSaoTypeIdxSCModel.get( 0, 0, 0 ) );
2342	for ( i=0; i<uiCode-1; i++ )
2343	{
2344	m_pcBinIf->encodeBin( 1, m_cSaoTypeIdxSCModel.get( 0, 0, 1 ) );
2345	}
2346	m_pcBinIf->encodeBin( 0, m_cSaoTypeIdxSCModel.get( 0, 0, 1 ) );
2347	}
2348	}
2349	#endif
2350	/*!
2351	****************************************************************************
2352	* \brief
2353	* estimate bit cost for CBP, significant map and significant coefficients
2354	****************************************************************************
2355	*/
2356	Void TEncSbac::estBit( estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType )
2357	{
2358	estCBFBit( pcEstBitsSbac, 0, eTType );
2359
2360	estSignificantCoeffGroupMapBit( pcEstBitsSbac, 0, eTType );
2361
2362	// encode significance map
2363	estSignificantMapBit( pcEstBitsSbac, width, height, eTType );
2364
2365	// encode significant coefficients
2366	estSignificantCoefficientsBit( pcEstBitsSbac, 0, eTType );
2367	}
2368
2369	/*!
2370	****************************************************************************
2371	* \brief
2372	* estimate bit cost for each CBP bit
2373	****************************************************************************
2374	*/
2375	Void TEncSbac::estCBFBit( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType )
2376	{
2377	ContextModel *pCtx = m_cCUQtCbfSCModel.get( 0 );
2378
2379	for( UInt uiCtxInc = 0; uiCtxInc < 3*NUM_QT_CBF_CTX; uiCtxInc++ )
2380	{
2381	pcEstBitsSbac->blockCbpBits[ uiCtxInc ][ 0 ] = pCtx[ uiCtxInc ].getEntropyBits( 0 );
2382	pcEstBitsSbac->blockCbpBits[ uiCtxInc ][ 1 ] = pCtx[ uiCtxInc ].getEntropyBits( 1 );
2383	}
2384
2385	pCtx = m_cCUQtRootCbfSCModel.get( 0 );
2386
2387	for( UInt uiCtxInc = 0; uiCtxInc < 4; uiCtxInc++ )
2388	{
2389	pcEstBitsSbac->blockRootCbpBits[ uiCtxInc ][ 0 ] = pCtx[ uiCtxInc ].getEntropyBits( 0 );
2390	pcEstBitsSbac->blockRootCbpBits[ uiCtxInc ][ 1 ] = pCtx[ uiCtxInc ].getEntropyBits( 1 );
2391	}
2392	}
2393
2394
2395	/*!
2396	****************************************************************************
2397	* \brief
2398	* estimate SAMBAC bit cost for significant coefficient group map
2399	****************************************************************************
2400	*/
2401	Void TEncSbac::estSignificantCoeffGroupMapBit( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType )
2402	{
2403	Int firstCtx = 0, numCtx = NUM_SIG_CG_FLAG_CTX;
2404
2405	for ( Int ctxIdx = firstCtx; ctxIdx < firstCtx + numCtx; ctxIdx++ )
2406	{
2407	for( UInt uiBin = 0; uiBin < 2; uiBin++ )
2408	{
2409	pcEstBitsSbac->significantCoeffGroupBits[ ctxIdx ][ uiBin ] = m_cCUSigCoeffGroupSCModel.get( 0, eTType, ctxIdx ).getEntropyBits( uiBin );
2410	}
2411	}
2412	}
2413
2414
2415	/*!
2416	****************************************************************************
2417	* \brief
2418	* estimate SAMBAC bit cost for significant coefficient map
2419	****************************************************************************
2420	*/
2421	Void TEncSbac::estSignificantMapBit( estBitsSbacStruct* pcEstBitsSbac, Int width, Int height, TextType eTType )
2422	{
2423	Int firstCtx = 0, numCtx = (eTType == TEXT_LUMA) ? 9 : 6;
2424	if (std::max(width, height) >= 16)
2425	{
2426	firstCtx = (eTType == TEXT_LUMA) ? 20 : 17;
2427	numCtx = (eTType == TEXT_LUMA) ? 7 : 4;
2428	}
2429	else if (width == 8)
2430	{
2431	firstCtx = (eTType == TEXT_LUMA) ? 9 : 6;
2432	numCtx = 11;
2433	}
2434
2435	if (eTType == TEXT_LUMA )
2436	{
2437	for ( Int ctxIdx = firstCtx; ctxIdx < firstCtx + numCtx; ctxIdx++ )
2438	{
2439	for( UInt uiBin = 0; uiBin < 2; uiBin++ )
2440	{
2441	pcEstBitsSbac->significantBits[ ctxIdx ][ uiBin ] = m_cCUSigSCModel.get( 0, 0, ctxIdx ).getEntropyBits( uiBin );
2442	}
2443	}
2444	}
2445	else
2446	{
2447	for ( Int ctxIdx = firstCtx; ctxIdx < firstCtx + numCtx; ctxIdx++ )
2448	{
2449	for( UInt uiBin = 0; uiBin < 2; uiBin++ )
2450	{
2451	pcEstBitsSbac->significantBits[ ctxIdx ][ uiBin ] = m_cCUSigSCModel.get( 0, 0, NUM_SIG_FLAG_CTX_LUMA + ctxIdx ).getEntropyBits( uiBin );
2452	}
2453	}
2454	}
2455	Int iBitsX = 0, iBitsY = 0;
2456	const UInt *puiCtxIdx;
2457	Int ctx;
2458	#if LAST_CTX_REDUCTION
2459	Int widthCtx = eTType? 4 : width;
2460	puiCtxIdx = g_uiLastCtx + (g_aucConvertToBit[ widthCtx ]*(g_aucConvertToBit[ widthCtx ]+3));
2461	#else
2462	puiCtxIdx = g_uiLastCtx + (g_aucConvertToBit[ width ]*(g_aucConvertToBit[ width ]+3));
2463	#endif
2464	ContextModel *pCtxX = m_cCuCtxLastX.get( 0, eTType );
2465	for (ctx = 0; ctx < g_uiGroupIdx[ width - 1 ]; ctx++)
2466	{
2467	Int ctxOffset = puiCtxIdx[ ctx ];
2468	#if LAST_CTX_REDUCTION
2469	if (eTType)
2470	{
2471	Int ctxOffsetC = ctx>>g_aucConvertToBit[ width ];
2472	pcEstBitsSbac->lastXBits[ ctx ] = iBitsX + pCtxX[ ctxOffsetC ].getEntropyBits( 0 );
2473	iBitsX += pCtxX[ ctxOffsetC].getEntropyBits( 1 );
2474	}
2475	else
2476	{
2477	#endif
2478	pcEstBitsSbac->lastXBits[ ctx ] = iBitsX + pCtxX[ ctxOffset ].getEntropyBits( 0 );
2479	iBitsX += pCtxX[ ctxOffset ].getEntropyBits( 1 );
2480	#if LAST_CTX_REDUCTION
2481	}
2482	#endif
2483	}
2484	pcEstBitsSbac->lastXBits[ctx] = iBitsX;
2485
2486	#if LAST_CTX_REDUCTION
2487	Int heightCtx = eTType? 4 : height;
2488	puiCtxIdx = g_uiLastCtx + (g_aucConvertToBit[ heightCtx ]*(g_aucConvertToBit[ heightCtx ]+3));
2489	#else
2490	puiCtxIdx = g_uiLastCtx + (g_aucConvertToBit[ height ]*(g_aucConvertToBit[ height ]+3));
2491	#endif
2492	ContextModel *pCtxY = m_cCuCtxLastY.get( 0, eTType );
2493	for (ctx = 0; ctx < g_uiGroupIdx[ height - 1 ]; ctx++)
2494	{
2495	Int ctxOffset = puiCtxIdx[ ctx ];
2496	#if LAST_CTX_REDUCTION
2497	if (eTType)
2498	{
2499	Int ctxOffsetC = ctx>>g_aucConvertToBit[ height ];
2500	pcEstBitsSbac->lastYBits[ ctx ] = iBitsY + pCtxY[ ctxOffsetC ].getEntropyBits( 0 );
2501	iBitsY += pCtxY[ctxOffsetC].getEntropyBits( 1 );
2502	}
2503	else
2504	{
2505	#endif
2506	pcEstBitsSbac->lastYBits[ ctx ] = iBitsY + pCtxY[ ctxOffset ].getEntropyBits( 0 );
2507	iBitsY += pCtxY[ ctxOffset ].getEntropyBits( 1 );
2508	#if LAST_CTX_REDUCTION
2509	}
2510	#endif
2511	}
2512	pcEstBitsSbac->lastYBits[ctx] = iBitsY;
2513	}
2514
2515	/*!
2516	****************************************************************************
2517	* \brief
2518	* estimate bit cost of significant coefficient
2519	****************************************************************************
2520	*/
2521	Void TEncSbac::estSignificantCoefficientsBit( estBitsSbacStruct* pcEstBitsSbac, UInt uiCTXIdx, TextType eTType )
2522	{
2523	if (eTType==TEXT_LUMA)
2524	{
2525	ContextModel *ctxOne = m_cCUOneSCModel.get(0, 0);
2526	ContextModel *ctxAbs = m_cCUAbsSCModel.get(0, 0);
2527
2528	for (Int ctxIdx = 0; ctxIdx < NUM_ONE_FLAG_CTX_LUMA; ctxIdx++)
2529	{
2530	pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 0 ] = ctxOne[ ctxIdx ].getEntropyBits( 0 );
2531	pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 1 ] = ctxOne[ ctxIdx ].getEntropyBits( 1 );
2532	}
2533
2534	for (Int ctxIdx = 0; ctxIdx < NUM_ABS_FLAG_CTX_LUMA; ctxIdx++)
2535	{
2536	pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 0 ] = ctxAbs[ ctxIdx ].getEntropyBits( 0 );
2537	pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 1 ] = ctxAbs[ ctxIdx ].getEntropyBits( 1 );
2538	}
2539	}
2540	else
2541	{
2542	ContextModel *ctxOne = m_cCUOneSCModel.get(0, 0) + NUM_ONE_FLAG_CTX_LUMA;
2543	ContextModel *ctxAbs = m_cCUAbsSCModel.get(0, 0) + NUM_ABS_FLAG_CTX_LUMA;
2544
2545	for (Int ctxIdx = 0; ctxIdx < NUM_ONE_FLAG_CTX_CHROMA; ctxIdx++)
2546	{
2547	pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 0 ] = ctxOne[ ctxIdx ].getEntropyBits( 0 );
2548	pcEstBitsSbac->m_greaterOneBits[ ctxIdx ][ 1 ] = ctxOne[ ctxIdx ].getEntropyBits( 1 );
2549	}
2550
2551	for (Int ctxIdx = 0; ctxIdx < NUM_ABS_FLAG_CTX_CHROMA; ctxIdx++)
2552	{
2553	pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 0 ] = ctxAbs[ ctxIdx ].getEntropyBits( 0 );
2554	pcEstBitsSbac->m_levelAbsBits[ ctxIdx ][ 1 ] = ctxAbs[ ctxIdx ].getEntropyBits( 1 );
2555	}
2556	}
2557	}
2558
2559	/**
2560	- Initialize our context information from the nominated source.
2561	.
2562	\param pSrc From where to copy context information.
2563	*/
2564	Void TEncSbac::xCopyContextsFrom( TEncSbac* pSrc )
2565	{
2566	memcpy(m_contextModels, pSrc->m_contextModels, m_numContextModels*sizeof(m_contextModels[0]));
2567	}
2568
2569	Void TEncSbac::loadContexts ( TEncSbac* pScr)
2570	{
2571	this->xCopyContextsFrom(pScr);
2572	}
2573
2574	#if HHI_DMM_WEDGE_INTRA \|\| HHI_DMM_PRED_TEX \|\| (LGE_EDGE_INTRA_A0070 && LGE_EDGE_INTRA_DELTA_DC)
2575	Void TEncSbac::xWriteExGolombLevel( UInt uiSymbol, ContextModel& rcSCModel )
2576	{
2577	if( uiSymbol )
2578	{
2579	m_pcBinIf->encodeBin( 1, rcSCModel );
2580	UInt uiCount = 0;
2581	Bool bNoExGo = (uiSymbol < 13);
2582
2583	while( --uiSymbol && ++uiCount < 13 )
2584	{
2585	m_pcBinIf->encodeBin( 1, rcSCModel );
2586	}
2587	if( bNoExGo )
2588	{
2589	m_pcBinIf->encodeBin( 0, rcSCModel );
2590	}
2591	else
2592	{
2593	xWriteEpExGolomb( uiSymbol, 0 );
2594	}
2595	}
2596	else
2597	{
2598	m_pcBinIf->encodeBin( 0, rcSCModel );
2599	}
2600
2601	return;
2602	}
2603	#endif
2604	#if HHI_DMM_WEDGE_INTRA
2605	Void TEncSbac::xCodeWedgeFullInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2606	{
2607	Int iIntraIdx = pcCU->getIntraSizeIdx(uiAbsPartIdx);
2608	Int iBits = g_aucWedgeFullBitsListIdx[iIntraIdx];
2609
2610	UInt uiTabIdx = pcCU->getWedgeFullTabIdx( uiAbsPartIdx );
2611
2612	for ( Int i = 0; i < iBits; i++ )
2613	{
2614	m_pcBinIf->encodeBin( ( uiTabIdx >> i ) & 1, m_cDmmDataSCModel.get(0, 0, 0) );
2615	}
2616	}
2617
2618	Void TEncSbac::xCodeWedgeFullDeltaInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2619	{
2620	Int iIntraIdx = pcCU->getIntraSizeIdx(uiAbsPartIdx);
2621	Int iBits = g_aucWedgeFullBitsListIdx[iIntraIdx];
2622
2623	UInt uiTabIdx = pcCU->getWedgeFullTabIdx( uiAbsPartIdx );
2624
2625	for ( Int i = 0; i < iBits; i++ )
2626	{
2627	m_pcBinIf->encodeBin( ( uiTabIdx >> i ) & 1, m_cDmmDataSCModel.get(0, 0, 0) );
2628	}
2629
2630	Int iDeltaDC1 = pcCU->getWedgeFullDeltaDC1( uiAbsPartIdx );
2631	Int iDeltaDC2 = pcCU->getWedgeFullDeltaDC2( uiAbsPartIdx );
2632
2633	xWriteExGolombLevel( UInt( abs( iDeltaDC1 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2634	if ( iDeltaDC1 != 0 )
2635	{
2636	UInt uiSign = iDeltaDC1 > 0 ? 0 : 1;
2637	m_pcBinIf->encodeBinEP( uiSign );
2638	}
2639	xWriteExGolombLevel( UInt( abs( iDeltaDC2 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2640	if ( iDeltaDC2 != 0 )
2641	{
2642	UInt uiSign = iDeltaDC2 > 0 ? 0 : 1;
2643	m_pcBinIf->encodeBinEP( uiSign );
2644	}
2645	}
2646
2647	Void TEncSbac::xCodeWedgePredDirInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2648	{
2649	if( DMM_WEDGE_PREDDIR_DELTAEND_MAX > 0 )
2650	{
2651	Int iDeltaEnd = pcCU->getWedgePredDirDeltaEnd( uiAbsPartIdx );
2652	m_pcBinIf->encodeBin( (iDeltaEnd!=0), m_cDmmDataSCModel.get(0, 0, 2) );
2653
2654	if( iDeltaEnd != 0 )
2655	{
2656	UInt uiAbsValMinus1 = abs(iDeltaEnd)-1;
2657	m_pcBinIf->encodeBin( (uiAbsValMinus1 & 0x01), m_cDmmDataSCModel.get(0, 0, 2) );
2658	m_pcBinIf->encodeBin( (uiAbsValMinus1 & 0x02) >> 1, m_cDmmDataSCModel.get(0, 0, 2) );
2659
2660	UInt uiSign = iDeltaEnd > 0 ? 0 : 1;
2661	m_pcBinIf->encodeBinEP( uiSign );
2662	}
2663	}
2664	}
2665
2666	Void TEncSbac::xCodeWedgePredDirDeltaInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2667	{
2668	if( DMM_WEDGE_PREDDIR_DELTAEND_MAX > 0 )
2669	{
2670	Int iDeltaEnd = pcCU->getWedgePredDirDeltaEnd( uiAbsPartIdx );
2671	m_pcBinIf->encodeBin( (iDeltaEnd!=0), m_cDmmDataSCModel.get(0, 0, 2) );
2672
2673	if( iDeltaEnd != 0 )
2674	{
2675	UInt uiAbsValMinus1 = abs(iDeltaEnd)-1;
2676	m_pcBinIf->encodeBin( (uiAbsValMinus1 & 0x01), m_cDmmDataSCModel.get(0, 0, 2) );
2677	m_pcBinIf->encodeBin( (uiAbsValMinus1 & 0x02) >> 1, m_cDmmDataSCModel.get(0, 0, 2) );
2678
2679	UInt uiSign = iDeltaEnd > 0 ? 0 : 1;
2680	m_pcBinIf->encodeBinEP( uiSign );
2681	}
2682	}
2683
2684	Int iDeltaDC1 = pcCU->getWedgePredDirDeltaDC1( uiAbsPartIdx );
2685	Int iDeltaDC2 = pcCU->getWedgePredDirDeltaDC2( uiAbsPartIdx );
2686
2687	xWriteExGolombLevel( UInt( abs( iDeltaDC1 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2688	if ( iDeltaDC1 != 0 )
2689	{
2690	UInt uiSign = iDeltaDC1 > 0 ? 0 : 1;
2691	m_pcBinIf->encodeBinEP( uiSign );
2692	}
2693	xWriteExGolombLevel( UInt( abs( iDeltaDC2 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2694	if ( iDeltaDC2 != 0 )
2695	{
2696	UInt uiSign = iDeltaDC2 > 0 ? 0 : 1;
2697	m_pcBinIf->encodeBinEP( uiSign );
2698	}
2699	}
2700	#endif
2701	#if HHI_DMM_PRED_TEX
2702	Void TEncSbac::xCodeWedgePredTexDeltaInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2703	{
2704	Int iDeltaDC1 = pcCU->getWedgePredTexDeltaDC1( uiAbsPartIdx );
2705	Int iDeltaDC2 = pcCU->getWedgePredTexDeltaDC2( uiAbsPartIdx );
2706
2707	xWriteExGolombLevel( UInt( abs( iDeltaDC1 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2708	if ( iDeltaDC1 != 0 )
2709	{
2710	UInt uiSign = iDeltaDC1 > 0 ? 0 : 1;
2711	m_pcBinIf->encodeBinEP( uiSign );
2712	}
2713	xWriteExGolombLevel( UInt( abs( iDeltaDC2 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2714	if ( iDeltaDC2 != 0 )
2715	{
2716	UInt uiSign = iDeltaDC2 > 0 ? 0 : 1;
2717	m_pcBinIf->encodeBinEP( uiSign );
2718	}
2719	}
2720
2721	Void TEncSbac::xCodeContourPredTexDeltaInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2722	{
2723	Int iDeltaDC1 = pcCU->getContourPredTexDeltaDC1( uiAbsPartIdx );
2724	Int iDeltaDC2 = pcCU->getContourPredTexDeltaDC2( uiAbsPartIdx );
2725
2726	xWriteExGolombLevel( UInt( abs( iDeltaDC1 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2727	if ( iDeltaDC1 != 0 )
2728	{
2729	UInt uiSign = iDeltaDC1 > 0 ? 0 : 1;
2730	m_pcBinIf->encodeBinEP( uiSign );
2731	}
2732	xWriteExGolombLevel( UInt( abs( iDeltaDC2 ) ), m_cDmmDataSCModel.get(0, 0, 1) );
2733	if ( iDeltaDC2 != 0 )
2734	{
2735	UInt uiSign = iDeltaDC2 > 0 ? 0 : 1;
2736	m_pcBinIf->encodeBinEP( uiSign );
2737	}
2738	}
2739	#endif
2740
2741	#if RWTH_SDC_DLT_B0036
2742	Void TEncSbac::codeSDCPredMode( TComDataCU* pcCU, UInt uiAbsPartIdx )
2743	{
2744	assert( pcCU->getSlice()->getSPS()->isDepth() );
2745	assert( pcCU->getPartitionSize(uiAbsPartIdx) == SIZE_2Nx2N );
2746	assert( pcCU->getSDCFlag(uiAbsPartIdx) );
2747
2748	UInt uiPredMode = pcCU->getLumaIntraDir(uiAbsPartIdx);
2749	UInt uiCtx = 0;
2750
2751	UInt uiMPModeIdx = 0;
2752
2753	for(Int i=0; i<RWTH_SDC_NUM_PRED_MODES-1; i++)
2754	{
2755	UInt uiBit = (uiPredMode == g_auiSDCPredModes[uiMPModeIdx]) ? 1 : 0;
2756	m_pcBinIf->encodeBin( uiBit, m_cSDCPredModeSCModel.get( 0, i, uiCtx ) );
2757
2758	// if mode is most probable mode, we are done here
2759	if ( uiBit == 1 )
2760	break;
2761
2762	// else: get next most probable pred mode
2763	uiMPModeIdx = (uiMPModeIdx+1)%RWTH_SDC_NUM_PRED_MODES;
2764	}
2765
2766	#if HHI_DMM_WEDGE_INTRA
2767	if( uiPredMode == DMM_WEDGE_FULL_IDX ) { xCodeWedgeFullInfo ( pcCU, uiAbsPartIdx ); }
2768	if( uiPredMode == DMM_WEDGE_PREDDIR_IDX ) { xCodeWedgePredDirInfo ( pcCU, uiAbsPartIdx ); }
2769	#endif
2770
2771	AOF(uiPredMode == g_auiSDCPredModes[uiMPModeIdx]);
2772	}
2773
2774	Void TEncSbac::codeSDCFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2775	{
2776	// get context function is here
2777	UInt uiSymbol = pcCU->getSDCFlag( uiAbsPartIdx ) ? 1 : 0;
2778	UInt uiCtxSDCFlag = pcCU->getCtxSDCFlag( uiAbsPartIdx );
2779	m_pcBinIf->encodeBin( uiSymbol, m_cSDCFlagSCModel.get( 0, 0, uiCtxSDCFlag ) );
2780
2781	}
2782
2783	Void TEncSbac::codeSDCResidualData ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiSegment )
2784	{
2785	assert( pcCU->getSlice()->getSPS()->isDepth() );
2786	assert( pcCU->getPartitionSize(uiAbsPartIdx) == SIZE_2Nx2N );
2787	assert( pcCU->getSDCFlag(uiAbsPartIdx) );
2788	assert( uiSegment < 2 );
2789
2790	Pel segmentDCOffset = pcCU->getSDCSegmentDCOffset(uiSegment, uiAbsPartIdx);
2791
2792	UInt uiResidual = segmentDCOffset == 0 ? 0 : 1;
2793	UInt uiSign = segmentDCOffset < 0 ? 1 : 0;
2794	UInt uiAbsIdx = abs(segmentDCOffset);
2795	UInt uiBit = 0;
2796
2797	UInt uiMaxResidualBits = GetBitsPerDepthValue();
2798	assert( uiMaxResidualBits <= g_uiBitDepth );
2799
2800	// residual flag
2801	m_pcBinIf->encodeBin( uiResidual, m_cSDCResidualFlagSCModel.get( 0, uiSegment, 0 ) ); //TODO depthmap: more sophisticated context selection
2802
2803	if (uiResidual)
2804	{
2805	// encode sign bit of residual
2806	m_pcBinIf->encodeBin( uiSign, m_cSDCResidualSignFlagSCModel.get( 0, uiSegment, 0 ) ); //TODO depthmap: more sophisticated context selection
2807
2808	assert(uiAbsIdx < GetNumDepthValues());
2809
2810	// encode residual magnitude
2811	uiAbsIdx -= 1;
2812	for (Int i=0; i<uiMaxResidualBits; i++)
2813	{
2814	uiBit = (uiAbsIdx & (1<<i))>>i;
2815
2816	m_pcBinIf->encodeBin( uiBit, m_cSDCResidualSCModel.get( 0, uiSegment, i ) ); //TODO depthmap: more sophisticated context selection
2817	}
2818
2819	}
2820	}
2821	#endif
2822	//! \}

Note: See TracBrowser for help on using the repository browser.

JCT-3V 3D-HEVC

Context navigation

source: 3DVCSoftware/branches/HTM-4.0.1-VSP-dev0/source/Lib/TLibEncoder/TEncSbac.cpp @ 213

Download in other formats: