Context navigation

source: 3DVCSoftware/trunk/source/Lib/TLibEncoder/TEncCavlc.cpp @ 950

Visit:

Last change on this file since 950 was 950, checked in by tech, 10 years ago
Merged 10.2-dev0@949.
Property svn:eol-style set to `native`
File size: 96.6 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-2014, 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 TEncCavlc.cpp
35	\brief CAVLC encoder class
36	*/
37
38	#include "../TLibCommon/CommonDef.h"
39	#include "TEncCavlc.h"
40	#include "SEIwrite.h"
41	#include "../TLibCommon/TypeDef.h"
42
43	//! \ingroup TLibEncoder
44	//! \{
45
46	#if ENC_DEC_TRACE
47
48	Void xTraceSPSHeader (TComSPS *pSPS)
49	{
50	#if H_MV_ENC_DEC_TRAC
51	fprintf( g_hTrace, "=========== Sequence Parameter Set ===========\n" );
52	#else
53	fprintf( g_hTrace, "=========== Sequence Parameter Set ID: %d ===========\n", pSPS->getSPSId() );
54	#endif
55	}
56
57	Void xTracePPSHeader (TComPPS *pPPS)
58	{
59	#if H_MV_ENC_DEC_TRAC
60	fprintf( g_hTrace, "=========== Picture Parameter Set ===========\n" );
61	#else
62	fprintf( g_hTrace, "=========== Picture Parameter Set ID: %d ===========\n", pPPS->getPPSId() );
63	#endif
64	}
65
66	Void xTraceSliceHeader (TComSlice *pSlice)
67	{
68	fprintf( g_hTrace, "=========== Slice ===========\n");
69	}
70
71	#endif
72
73
74
75	// ====================================================================================================================
76	// Constructor / destructor / create / destroy
77	// ====================================================================================================================
78
79	TEncCavlc::TEncCavlc()
80	{
81	m_pcBitIf = NULL;
82	m_uiCoeffCost = 0;
83	}
84
85	TEncCavlc::~TEncCavlc()
86	{
87	}
88
89
90	// ====================================================================================================================
91	// Public member functions
92	// ====================================================================================================================
93
94	Void TEncCavlc::resetEntropy()
95	{
96	}
97
98
99	Void TEncCavlc::codeDFFlag(UInt uiCode, const Char *pSymbolName)
100	{
101	WRITE_FLAG(uiCode, pSymbolName);
102	}
103	Void TEncCavlc::codeDFSvlc(Int iCode, const Char *pSymbolName)
104	{
105	WRITE_SVLC(iCode, pSymbolName);
106	}
107
108	Void TEncCavlc::codeShortTermRefPicSet( TComSPS* pcSPS, TComReferencePictureSet* rps, Bool calledFromSliceHeader, Int idx)
109	{
110	#if PRINT_RPS_INFO
111	Int lastBits = getNumberOfWrittenBits();
112	#endif
113	if (idx > 0)
114	{
115	WRITE_FLAG( rps->getInterRPSPrediction(), "inter_ref_pic_set_prediction_flag" ); // inter_RPS_prediction_flag
116	}
117	if (rps->getInterRPSPrediction())
118	{
119	Int deltaRPS = rps->getDeltaRPS();
120	if(calledFromSliceHeader)
121	{
122	WRITE_UVLC( rps->getDeltaRIdxMinus1(), "delta_idx_minus1" ); // delta index of the Reference Picture Set used for prediction minus 1
123	}
124
125	WRITE_CODE( (deltaRPS >=0 ? 0: 1), 1, "delta_rps_sign" ); //delta_rps_sign
126	WRITE_UVLC( abs(deltaRPS) - 1, "abs_delta_rps_minus1"); // absolute delta RPS minus 1
127
128	for(Int j=0; j < rps->getNumRefIdc(); j++)
129	{
130	Int refIdc = rps->getRefIdc(j);
131	WRITE_CODE( (refIdc==1? 1: 0), 1, "used_by_curr_pic_flag" ); //first bit is "1" if Idc is 1
132	if (refIdc != 1)
133	{
134	WRITE_CODE( refIdc>>1, 1, "use_delta_flag" ); //second bit is "1" if Idc is 2, "0" otherwise.
135	}
136	}
137	}
138	else
139	{
140	WRITE_UVLC( rps->getNumberOfNegativePictures(), "num_negative_pics" );
141	WRITE_UVLC( rps->getNumberOfPositivePictures(), "num_positive_pics" );
142	Int prev = 0;
143	for(Int j=0 ; j < rps->getNumberOfNegativePictures(); j++)
144	{
145	WRITE_UVLC( prev-rps->getDeltaPOC(j)-1, "delta_poc_s0_minus1" );
146	prev = rps->getDeltaPOC(j);
147	WRITE_FLAG( rps->getUsed(j), "used_by_curr_pic_s0_flag");
148	}
149	prev = 0;
150	for(Int j=rps->getNumberOfNegativePictures(); j < rps->getNumberOfNegativePictures()+rps->getNumberOfPositivePictures(); j++)
151	{
152	WRITE_UVLC( rps->getDeltaPOC(j)-prev-1, "delta_poc_s1_minus1" );
153	prev = rps->getDeltaPOC(j);
154	WRITE_FLAG( rps->getUsed(j), "used_by_curr_pic_s1_flag" );
155	}
156	}
157
158	#if PRINT_RPS_INFO
159	printf("irps=%d (%2d bits) ", rps->getInterRPSPrediction(), getNumberOfWrittenBits() - lastBits);
160	rps->printDeltaPOC();
161	#endif
162	}
163
164
165	Void TEncCavlc::codePPS( TComPPS* pcPPS )
166	{
167	#if ENC_DEC_TRACE
168	xTracePPSHeader (pcPPS);
169	#endif
170
171	WRITE_UVLC( pcPPS->getPPSId(), "pps_pic_parameter_set_id" );
172	WRITE_UVLC( pcPPS->getSPSId(), "pps_seq_parameter_set_id" );
173	WRITE_FLAG( pcPPS->getDependentSliceSegmentsEnabledFlag() ? 1 : 0, "dependent_slice_segments_enabled_flag" );
174	WRITE_FLAG( pcPPS->getOutputFlagPresentFlag() ? 1 : 0, "output_flag_present_flag" );
175	WRITE_CODE( pcPPS->getNumExtraSliceHeaderBits(), 3, "num_extra_slice_header_bits");
176	WRITE_FLAG( pcPPS->getSignHideFlag(), "sign_data_hiding_flag" );
177	WRITE_FLAG( pcPPS->getCabacInitPresentFlag() ? 1 : 0, "cabac_init_present_flag" );
178	#if PPS_FIX_DEPTH
179	if( pcPPS->getSPS()->getVPS()->getDepthId(pcPPS->getSPS()->getLayerId()) )
180	{
181	WRITE_UVLC( pcPPS->getNumRefIdxL0DefaultActive(), "num_ref_idx_l0_default_active_minus1");
182	WRITE_UVLC( pcPPS->getNumRefIdxL1DefaultActive(), "num_ref_idx_l1_default_active_minus1");
183	}
184	else
185	{
186	#endif
187	WRITE_UVLC( pcPPS->getNumRefIdxL0DefaultActive()-1, "num_ref_idx_l0_default_active_minus1");
188	WRITE_UVLC( pcPPS->getNumRefIdxL1DefaultActive()-1, "num_ref_idx_l1_default_active_minus1");
189	#if PPS_FIX_DEPTH
190	}
191	#endif
192	WRITE_SVLC( pcPPS->getPicInitQPMinus26(), "init_qp_minus26");
193	WRITE_FLAG( pcPPS->getConstrainedIntraPred() ? 1 : 0, "constrained_intra_pred_flag" );
194	WRITE_FLAG( pcPPS->getUseTransformSkip() ? 1 : 0, "transform_skip_enabled_flag" );
195	WRITE_FLAG( pcPPS->getUseDQP() ? 1 : 0, "cu_qp_delta_enabled_flag" );
196	if ( pcPPS->getUseDQP() )
197	{
198	WRITE_UVLC( pcPPS->getMaxCuDQPDepth(), "diff_cu_qp_delta_depth" );
199	}
200	WRITE_SVLC( pcPPS->getChromaCbQpOffset(), "pps_cb_qp_offset" );
201	WRITE_SVLC( pcPPS->getChromaCrQpOffset(), "pps_cr_qp_offset" );
202	WRITE_FLAG( pcPPS->getSliceChromaQpFlag() ? 1 : 0, "pps_slice_chroma_qp_offsets_present_flag" );
203
204	WRITE_FLAG( pcPPS->getUseWP() ? 1 : 0, "weighted_pred_flag" ); // Use of Weighting Prediction (P_SLICE)
205	WRITE_FLAG( pcPPS->getWPBiPred() ? 1 : 0, "weighted_bipred_flag" ); // Use of Weighting Bi-Prediction (B_SLICE)
206	WRITE_FLAG( pcPPS->getTransquantBypassEnableFlag() ? 1 : 0, "transquant_bypass_enable_flag" );
207	WRITE_FLAG( pcPPS->getTilesEnabledFlag() ? 1 : 0, "tiles_enabled_flag" );
208	WRITE_FLAG( pcPPS->getEntropyCodingSyncEnabledFlag() ? 1 : 0, "entropy_coding_sync_enabled_flag" );
209	if( pcPPS->getTilesEnabledFlag() )
210	{
211	WRITE_UVLC( pcPPS->getNumColumnsMinus1(), "num_tile_columns_minus1" );
212	WRITE_UVLC( pcPPS->getNumRowsMinus1(), "num_tile_rows_minus1" );
213	WRITE_FLAG( pcPPS->getUniformSpacingFlag(), "uniform_spacing_flag" );
214	if( pcPPS->getUniformSpacingFlag() == 0 )
215	{
216	for(UInt i=0; i<pcPPS->getNumColumnsMinus1(); i++)
217	{
218	WRITE_UVLC( pcPPS->getColumnWidth(i)-1, "column_width_minus1" );
219	}
220	for(UInt i=0; i<pcPPS->getNumRowsMinus1(); i++)
221	{
222	WRITE_UVLC( pcPPS->getRowHeight(i)-1, "row_height_minus1" );
223	}
224	}
225	if(pcPPS->getNumColumnsMinus1() !=0 \|\| pcPPS->getNumRowsMinus1() !=0)
226	{
227	WRITE_FLAG( pcPPS->getLoopFilterAcrossTilesEnabledFlag()?1 : 0, "loop_filter_across_tiles_enabled_flag");
228	}
229	}
230	WRITE_FLAG( pcPPS->getLoopFilterAcrossSlicesEnabledFlag()?1 : 0, "loop_filter_across_slices_enabled_flag");
231	WRITE_FLAG( pcPPS->getDeblockingFilterControlPresentFlag()?1 : 0, "deblocking_filter_control_present_flag");
232	if(pcPPS->getDeblockingFilterControlPresentFlag())
233	{
234	WRITE_FLAG( pcPPS->getDeblockingFilterOverrideEnabledFlag() ? 1 : 0, "deblocking_filter_override_enabled_flag" );
235	WRITE_FLAG( pcPPS->getPicDisableDeblockingFilterFlag() ? 1 : 0, "pps_disable_deblocking_filter_flag" );
236	if(!pcPPS->getPicDisableDeblockingFilterFlag())
237	{
238	WRITE_SVLC( pcPPS->getDeblockingFilterBetaOffsetDiv2(), "pps_beta_offset_div2" );
239	WRITE_SVLC( pcPPS->getDeblockingFilterTcOffsetDiv2(), "pps_tc_offset_div2" );
240	}
241	}
242	#if H_MV
243	if ( pcPPS->getLayerId() > 0 )
244	{
245	WRITE_FLAG( pcPPS->getPpsInferScalingListFlag( ) ? 1 : 0 , "pps_infer_scaling_list_flag" );
246	}
247
248	if( pcPPS->getPpsInferScalingListFlag( ) )
249	{
250	WRITE_CODE( pcPPS->getPpsScalingListRefLayerId( ), 6, "pps_scaling_list_ref_layer_id" );
251	}
252	else
253	{
254	#endif
255	WRITE_FLAG( pcPPS->getScalingListPresentFlag() ? 1 : 0, "pps_scaling_list_data_present_flag" );
256	if( pcPPS->getScalingListPresentFlag() )
257	{
258	codeScalingList( m_pcSlice->getScalingList() );
259	}
260	#if H_MV
261	}
262	#endif
263	#if PPS_FIX_DEPTH
264	if( pcPPS->getSPS()->getVPS()->getDepthId(pcPPS->getSPS()->getLayerId()) )
265	{
266	WRITE_FLAG( 1, "lists_modification_present_flag" );
267	}
268	else
269	#endif
270	WRITE_FLAG( pcPPS->getListsModificationPresentFlag(), "lists_modification_present_flag");
271	WRITE_UVLC( pcPPS->getLog2ParallelMergeLevelMinus2(), "log2_parallel_merge_level_minus2");
272	WRITE_FLAG( pcPPS->getSliceHeaderExtensionPresentFlag() ? 1 : 0, "slice_segment_header_extension_present_flag");
273
274	#if MTK_DLT_CODING_FIX_H0091
275	if(( !pcPPS->getSPS()->getVPS()->getDepthId( pcPPS->getSPS()->getLayerId() ) )\|\|
276	pcPPS->getLayerId() != 1 )
277	{
278	pcPPS->setPpsExtensionTypeFlag( PPS_EX_T_3D, false );
279	}
280	#endif
281
282	#if !H_MV
283	WRITE_FLAG( 0, "pps_extension_flag" );
284	#else
285	WRITE_FLAG( 1, "pps_extension_flag" );
286
287	for ( Int i = 0; i < PPS_EX_T_MAX_NUM; i++ )
288	{
289	WRITE_FLAG( pcPPS->getPpsExtensionTypeFlag( i ) ? 1 : 0 , "pps_extension_type_flag" );
290	#if H_3D
291	assert( !pcPPS->getPpsExtensionTypeFlag( i ) \|\| i == PPS_EX_T_MV \|\| i == PPS_EX_T_3D \|\| i == PPS_EX_T_ESC );
292	#else
293	assert( !pcPPS->getPpsExtensionTypeFlag( i ) \|\| i == PPS_EX_T_MV \|\| i == PPS_EX_T_ESC );
294	#endif
295	}
296
297	if( pcPPS->getPpsExtensionTypeFlag( PPS_EX_T_MV ) )
298	{
299	WRITE_FLAG( pcPPS->getPocResetInfoPresentFlag( ) ? 1 : 0 , "poc_reset_info_present_flag" );
300	}
301
302	#if H_3D
303	if( pcPPS->getPpsExtensionTypeFlag( PPS_EX_T_3D ) ) // This probably needs to be aligned with Rext and SHVC
304	{
305	codePPSExtension( pcPPS );
306	}
307	#endif
308
309	#endif
310
311	}
312
313	#if H_3D
314	Void TEncCavlc::codePPSExtension ( TComPPS* pcPPS )
315	{
316	// Assuming that all PPS indirectly refer to the same VPS via different SPS
317	// There is no parsing dependency in decoding DLT in PPS.
318	// The VPS information passed to decodePPS() is used to arrange the decoded DLT tables to their corresponding layers.
319	// This is equivalent to the process of
320	// Step 1) decoding DLT tables based on the number of depth layers, and
321	// Step 2) mapping DLT tables to the depth layers
322	// as described in the 3D-HEVC WD.
323	TComVPS* pcVPS = pcPPS->getSPS()->getVPS();
324
325	TComDLT* pcDLT = pcPPS->getDLT();
326
327	WRITE_FLAG( pcDLT->getDltPresentFlag() ? 1 : 0, "dlt_present_flag" );
328
329	if ( pcDLT->getDltPresentFlag() )
330	{
331	WRITE_CODE(pcDLT->getNumDepthViews(), 6, "pps_depth_layers_minus1");
332	WRITE_CODE((pcDLT->getDepthViewBitDepth() - 8), 4, "pps_bit_depth_for_depth_views_minus8");
333
334	for( Int i = 0; i <= pcVPS->getMaxLayersMinus1(); i++ )
335	{
336	if ( i != 0 )
337	{
338	if ( pcVPS->getDepthId( i ) == 1 )
339	{
340	WRITE_FLAG( pcDLT->getUseDLTFlag( i ) ? 1 : 0, "dlt_flag[i]" );
341
342	if ( pcDLT->getUseDLTFlag( i ) )
343	{
344	WRITE_FLAG( pcDLT->getInterViewDltPredEnableFlag( i ) ? 1 : 0, "inter_view_dlt_pred_enable_flag[ i ]");
345
346	// ----------------------------- determine whether to use bit-map -----------------------------
347	Bool bDltBitMapRepFlag = false;
348	UInt uiNumBitsNonBitMap = 0;
349	UInt uiNumBitsBitMap = 0;
350
351	UInt uiMaxDiff = 0;
352	UInt uiMinDiff = 0xffffffff;
353	UInt uiLengthMinDiff = 0;
354	UInt uiLengthDltDiffMinusMin = 0;
355
356	UInt* puiDltDiffValues = NULL;
357
358	Int aiIdx2DepthValue_coded[256];
359	UInt uiNumDepthValues_coded = 0;
360
361	uiNumDepthValues_coded = pcDLT->getNumDepthValues(i);
362	for( UInt ui = 0; ui<uiNumDepthValues_coded; ui++ )
363	{
364	aiIdx2DepthValue_coded[ui] = pcDLT->idx2DepthValue(i, ui);
365	}
366
367	#if H_3D_DELTA_DLT
368	if( pcDLT->getInterViewDltPredEnableFlag( i ) )
369	{
370	AOF( pcVPS->getDepthId( 1 ) == 1 );
371	AOF( i > 1 );
372	// assumes ref layer id to be 1
373	Int* piRefDLT = pcDLT->idx2DepthValue( 1 );
374	UInt uiRefNum = pcDLT->getNumDepthValues( 1 );
375	pcDLT->getDeltaDLT(i, piRefDLT, uiRefNum, aiIdx2DepthValue_coded, &uiNumDepthValues_coded);
376	}
377	#endif
378
379	if ( NULL == (puiDltDiffValues = (UInt *)calloc(uiNumDepthValues_coded, sizeof(UInt))) )
380	{
381	exit(-1);
382	}
383
384	for (UInt d = 1; d < uiNumDepthValues_coded; d++)
385	{
386	puiDltDiffValues[d] = aiIdx2DepthValue_coded[d] - aiIdx2DepthValue_coded[d-1];
387
388	if ( uiMaxDiff < puiDltDiffValues[d] )
389	{
390	uiMaxDiff = puiDltDiffValues[d];
391	}
392
393	if ( uiMinDiff > puiDltDiffValues[d] )
394	{
395	uiMinDiff = puiDltDiffValues[d];
396	}
397	}
398
399	// counting bits
400	// diff coding branch
401	uiNumBitsNonBitMap += 8; // u(v) bits for num_depth_values_in_dlt[layerId] (i.e. num_entry[ layerId ])
402
403	if ( uiNumDepthValues_coded > 1 )
404	{
405	uiNumBitsNonBitMap += 8; // u(v) bits for max_diff[ layerId ]
406	}
407
408	if ( uiNumDepthValues_coded > 2 )
409	{
410	uiLengthMinDiff = (UInt) ceil(Log2(uiMaxDiff + 1));
411	uiNumBitsNonBitMap += uiLengthMinDiff; // u(v) bits for min_diff[ layerId ]
412	}
413
414	uiNumBitsNonBitMap += 8; // u(v) bits for dlt_depth_value0[ layerId ]
415
416	if (uiMaxDiff > uiMinDiff)
417	{
418	uiLengthDltDiffMinusMin = (UInt) ceil(Log2(uiMaxDiff - uiMinDiff + 1));
419	uiNumBitsNonBitMap += uiLengthDltDiffMinusMin * (uiNumDepthValues_coded - 1); // u(v) bits for dlt_depth_value_diff_minus_min[ layerId ][ j ]
420	}
421
422	// bit map branch
423	uiNumBitsBitMap = 256; // uiNumBitsBitMap = 1 << pcDLT->getDepthViewBitDepth();
424
425	// determine bDltBitMapFlag
426	bDltBitMapRepFlag = (uiNumBitsBitMap > uiNumBitsNonBitMap) ? false : true;
427
428	// ----------------------------- Actual coding -----------------------------
429	if ( pcDLT->getInterViewDltPredEnableFlag( i ) == false )
430	{
431	WRITE_FLAG( bDltBitMapRepFlag ? 1 : 0, "dlt_bit_map_rep_flag[ layerId ]" );
432	}
433	else
434	{
435	bDltBitMapRepFlag = false;
436	}
437
438	// bit map coding
439	if ( bDltBitMapRepFlag )
440	{
441	UInt uiDltArrayIndex = 0;
442	for (UInt d=0; d < 256; d++)
443	{
444	if ( d == aiIdx2DepthValue_coded[uiDltArrayIndex] )
445	{
446	WRITE_FLAG(1, "dlt_bit_map_flag[ layerId ][ j ]");
447	uiDltArrayIndex++;
448	}
449	else
450	{
451	WRITE_FLAG(0, "dlt_bit_map_flag[ layerId ][ j ]");
452	}
453	}
454	}
455	// Diff Coding
456	else
457	{
458	WRITE_CODE(uiNumDepthValues_coded, 8, "num_depth_values_in_dlt[i]"); // num_entry
459
460	#if !H_3D_DELTA_DLT
461	if ( pcDLT->getInterViewDltPredEnableFlag( i ) == false ) // Single-view DLT Diff Coding
462	#endif
463	{
464	// The condition if( uiNumDepthValues_coded > 0 ) is always true since for Single-view Diff Coding, there is at least one depth value in depth component.
465	if ( uiNumDepthValues_coded > 1 )
466	{
467	WRITE_CODE(uiMaxDiff, 8, "max_diff[ layerId ]"); // max_diff
468	}
469
470	if ( uiNumDepthValues_coded > 2 )
471	{
472	WRITE_CODE((uiMinDiff - 1), uiLengthMinDiff, "min_diff_minus1[ layerId ]"); // min_diff_minus1
473	}
474
475	WRITE_CODE(aiIdx2DepthValue_coded[0], 8, "dlt_depth_value0[layerId]"); // entry0
476
477	if (uiMaxDiff > uiMinDiff)
478	{
479	for (UInt d=1; d < uiNumDepthValues_coded; d++)
480	{
481	WRITE_CODE( (puiDltDiffValues[d] - uiMinDiff), uiLengthDltDiffMinusMin, "dlt_depth_value_diff_minus_min[ layerId ][ j ]"); // entry_value_diff_minus_min[ k ]
482	}
483	}
484	}
485	}
486
487	free(puiDltDiffValues);
488	}
489	}
490	}
491	}
492	}
493	}
494	#endif
495
496	Void TEncCavlc::codeVUI( TComVUI pcVUI, TComSPS pcSPS )
497	{
498	#if ENC_DEC_TRACE
499	fprintf( g_hTrace, "----------- vui_parameters -----------\n");
500	#endif
501	WRITE_FLAG(pcVUI->getAspectRatioInfoPresentFlag(), "aspect_ratio_info_present_flag");
502	if (pcVUI->getAspectRatioInfoPresentFlag())
503	{
504	WRITE_CODE(pcVUI->getAspectRatioIdc(), 8, "aspect_ratio_idc" );
505	if (pcVUI->getAspectRatioIdc() == 255)
506	{
507	WRITE_CODE(pcVUI->getSarWidth(), 16, "sar_width");
508	WRITE_CODE(pcVUI->getSarHeight(), 16, "sar_height");
509	}
510	}
511	WRITE_FLAG(pcVUI->getOverscanInfoPresentFlag(), "overscan_info_present_flag");
512	if (pcVUI->getOverscanInfoPresentFlag())
513	{
514	WRITE_FLAG(pcVUI->getOverscanAppropriateFlag(), "overscan_appropriate_flag");
515	}
516	WRITE_FLAG(pcVUI->getVideoSignalTypePresentFlag(), "video_signal_type_present_flag");
517	#if H_MV
518	assert( pcSPS->getLayerId() == 0 \|\| !pcVUI->getVideoSignalTypePresentFlag() );
519	#endif
520	if (pcVUI->getVideoSignalTypePresentFlag())
521	{
522	WRITE_CODE(pcVUI->getVideoFormat(), 3, "video_format");
523	WRITE_FLAG(pcVUI->getVideoFullRangeFlag(), "video_full_range_flag");
524	WRITE_FLAG(pcVUI->getColourDescriptionPresentFlag(), "colour_description_present_flag");
525	if (pcVUI->getColourDescriptionPresentFlag())
526	{
527	WRITE_CODE(pcVUI->getColourPrimaries(), 8, "colour_primaries");
528	WRITE_CODE(pcVUI->getTransferCharacteristics(), 8, "transfer_characteristics");
529	WRITE_CODE(pcVUI->getMatrixCoefficients(), 8, "matrix_coefficients");
530	}
531	}
532
533	WRITE_FLAG(pcVUI->getChromaLocInfoPresentFlag(), "chroma_loc_info_present_flag");
534	if (pcVUI->getChromaLocInfoPresentFlag())
535	{
536	WRITE_UVLC(pcVUI->getChromaSampleLocTypeTopField(), "chroma_sample_loc_type_top_field");
537	WRITE_UVLC(pcVUI->getChromaSampleLocTypeBottomField(), "chroma_sample_loc_type_bottom_field");
538	}
539
540	WRITE_FLAG(pcVUI->getNeutralChromaIndicationFlag(), "neutral_chroma_indication_flag");
541	WRITE_FLAG(pcVUI->getFieldSeqFlag(), "field_seq_flag");
542	WRITE_FLAG(pcVUI->getFrameFieldInfoPresentFlag(), "frame_field_info_present_flag");
543
544	Window defaultDisplayWindow = pcVUI->getDefaultDisplayWindow();
545	WRITE_FLAG(defaultDisplayWindow.getWindowEnabledFlag(), "default_display_window_flag");
546	if( defaultDisplayWindow.getWindowEnabledFlag() )
547	{
548	WRITE_UVLC(defaultDisplayWindow.getWindowLeftOffset(), "def_disp_win_left_offset");
549	WRITE_UVLC(defaultDisplayWindow.getWindowRightOffset(), "def_disp_win_right_offset");
550	WRITE_UVLC(defaultDisplayWindow.getWindowTopOffset(), "def_disp_win_top_offset");
551	WRITE_UVLC(defaultDisplayWindow.getWindowBottomOffset(), "def_disp_win_bottom_offset");
552	}
553	TimingInfo *timingInfo = pcVUI->getTimingInfo();
554	WRITE_FLAG(timingInfo->getTimingInfoPresentFlag(), "vui_timing_info_present_flag");
555	if(timingInfo->getTimingInfoPresentFlag())
556	{
557	WRITE_CODE(timingInfo->getNumUnitsInTick(), 32, "vui_num_units_in_tick");
558	WRITE_CODE(timingInfo->getTimeScale(), 32, "vui_time_scale");
559	WRITE_FLAG(timingInfo->getPocProportionalToTimingFlag(), "vui_poc_proportional_to_timing_flag");
560	if(timingInfo->getPocProportionalToTimingFlag())
561	{
562	WRITE_UVLC(timingInfo->getNumTicksPocDiffOneMinus1(), "vui_num_ticks_poc_diff_one_minus1");
563	}
564	WRITE_FLAG(pcVUI->getHrdParametersPresentFlag(), "hrd_parameters_present_flag");
565	if( pcVUI->getHrdParametersPresentFlag() )
566	{
567	codeHrdParameters(pcVUI->getHrdParameters(), 1, pcSPS->getMaxTLayers() - 1 );
568	}
569	}
570
571	WRITE_FLAG(pcVUI->getBitstreamRestrictionFlag(), "bitstream_restriction_flag");
572	if (pcVUI->getBitstreamRestrictionFlag())
573	{
574	WRITE_FLAG(pcVUI->getTilesFixedStructureFlag(), "tiles_fixed_structure_flag");
575	WRITE_FLAG(pcVUI->getMotionVectorsOverPicBoundariesFlag(), "motion_vectors_over_pic_boundaries_flag");
576	WRITE_FLAG(pcVUI->getRestrictedRefPicListsFlag(), "restricted_ref_pic_lists_flag");
577	WRITE_UVLC(pcVUI->getMinSpatialSegmentationIdc(), "min_spatial_segmentation_idc");
578	WRITE_UVLC(pcVUI->getMaxBytesPerPicDenom(), "max_bytes_per_pic_denom");
579	WRITE_UVLC(pcVUI->getMaxBitsPerMinCuDenom(), "max_bits_per_mincu_denom");
580	WRITE_UVLC(pcVUI->getLog2MaxMvLengthHorizontal(), "log2_max_mv_length_horizontal");
581	WRITE_UVLC(pcVUI->getLog2MaxMvLengthVertical(), "log2_max_mv_length_vertical");
582	}
583	}
584
585	Void TEncCavlc::codeHrdParameters( TComHRD *hrd, Bool commonInfPresentFlag, UInt maxNumSubLayersMinus1 )
586	{
587	if( commonInfPresentFlag )
588	{
589	WRITE_FLAG( hrd->getNalHrdParametersPresentFlag() ? 1 : 0 , "nal_hrd_parameters_present_flag" );
590	WRITE_FLAG( hrd->getVclHrdParametersPresentFlag() ? 1 : 0 , "vcl_hrd_parameters_present_flag" );
591	if( hrd->getNalHrdParametersPresentFlag() \|\| hrd->getVclHrdParametersPresentFlag() )
592	{
593	WRITE_FLAG( hrd->getSubPicCpbParamsPresentFlag() ? 1 : 0, "sub_pic_cpb_params_present_flag" );
594	if( hrd->getSubPicCpbParamsPresentFlag() )
595	{
596	WRITE_CODE( hrd->getTickDivisorMinus2(), 8, "tick_divisor_minus2" );
597	WRITE_CODE( hrd->getDuCpbRemovalDelayLengthMinus1(), 5, "du_cpb_removal_delay_length_minus1" );
598	WRITE_FLAG( hrd->getSubPicCpbParamsInPicTimingSEIFlag() ? 1 : 0, "sub_pic_cpb_params_in_pic_timing_sei_flag" );
599	WRITE_CODE( hrd->getDpbOutputDelayDuLengthMinus1(), 5, "dpb_output_delay_du_length_minus1" );
600	}
601	WRITE_CODE( hrd->getBitRateScale(), 4, "bit_rate_scale" );
602	WRITE_CODE( hrd->getCpbSizeScale(), 4, "cpb_size_scale" );
603	if( hrd->getSubPicCpbParamsPresentFlag() )
604	{
605	WRITE_CODE( hrd->getDuCpbSizeScale(), 4, "du_cpb_size_scale" );
606	}
607	WRITE_CODE( hrd->getInitialCpbRemovalDelayLengthMinus1(), 5, "initial_cpb_removal_delay_length_minus1" );
608	WRITE_CODE( hrd->getCpbRemovalDelayLengthMinus1(), 5, "au_cpb_removal_delay_length_minus1" );
609	WRITE_CODE( hrd->getDpbOutputDelayLengthMinus1(), 5, "dpb_output_delay_length_minus1" );
610	}
611	}
612	Int i, j, nalOrVcl;
613	for( i = 0; i <= maxNumSubLayersMinus1; i ++ )
614	{
615	WRITE_FLAG( hrd->getFixedPicRateFlag( i ) ? 1 : 0, "fixed_pic_rate_general_flag");
616	if( !hrd->getFixedPicRateFlag( i ) )
617	{
618	WRITE_FLAG( hrd->getFixedPicRateWithinCvsFlag( i ) ? 1 : 0, "fixed_pic_rate_within_cvs_flag");
619	}
620	else
621	{
622	hrd->setFixedPicRateWithinCvsFlag( i, true );
623	}
624	if( hrd->getFixedPicRateWithinCvsFlag( i ) )
625	{
626	WRITE_UVLC( hrd->getPicDurationInTcMinus1( i ), "elemental_duration_in_tc_minus1");
627	}
628	else
629	{
630	WRITE_FLAG( hrd->getLowDelayHrdFlag( i ) ? 1 : 0, "low_delay_hrd_flag");
631	}
632	if (!hrd->getLowDelayHrdFlag( i ))
633	{
634	WRITE_UVLC( hrd->getCpbCntMinus1( i ), "cpb_cnt_minus1");
635	}
636
637	for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
638	{
639	if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) \|\|
640	( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
641	{
642	for( j = 0; j <= ( hrd->getCpbCntMinus1( i ) ); j ++ )
643	{
644	WRITE_UVLC( hrd->getBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_value_minus1");
645	WRITE_UVLC( hrd->getCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_value_minus1");
646	if( hrd->getSubPicCpbParamsPresentFlag() )
647	{
648	WRITE_UVLC( hrd->getDuCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_du_value_minus1");
649	WRITE_UVLC( hrd->getDuBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_du_value_minus1");
650	}
651	WRITE_FLAG( hrd->getCbrFlag( i, j, nalOrVcl ) ? 1 : 0, "cbr_flag");
652	}
653	}
654	}
655	}
656	}
657
658	#if H_3D
659	Void TEncCavlc::codeSPS( TComSPS* pcSPS, Int viewIndex, Bool depthFlag )
660	#else
661	Void TEncCavlc::codeSPS( TComSPS* pcSPS )
662	#endif
663	{
664	#if ENC_DEC_TRACE
665	xTraceSPSHeader (pcSPS);
666	#endif
667	WRITE_CODE( pcSPS->getVPSId (), 4, "sps_video_parameter_set_id" );
668	#if H_MV
669	if ( pcSPS->getLayerId() == 0 )
670	{
671	#endif
672	WRITE_CODE( pcSPS->getMaxTLayers() - 1, 3, "sps_max_sub_layers_minus1" );
673	WRITE_FLAG( pcSPS->getTemporalIdNestingFlag() ? 1 : 0, "sps_temporal_id_nesting_flag" );
674	codePTL(pcSPS->getPTL(), 1, pcSPS->getMaxTLayers() - 1);
675	#if H_MV
676	}
677	#endif
678	WRITE_UVLC( pcSPS->getSPSId (), "sps_seq_parameter_set_id" );
679	#if H_MV
680	if ( pcSPS->getLayerId() > 0 )
681	{
682	WRITE_FLAG( pcSPS->getUpdateRepFormatFlag( ) ? 1 : 0 , "update_rep_format_flag" );
683	if ( pcSPS->getUpdateRepFormatFlag() )
684	{
685	WRITE_CODE( pcSPS->getSpsRepFormatIdx( ), 8, "sps_rep_format_idx" );
686	}
687	}
688	else
689	{
690	#endif
691	WRITE_UVLC( pcSPS->getChromaFormatIdc (), "chroma_format_idc" );
692	assert(pcSPS->getChromaFormatIdc () == 1);
693	// in the first version chroma_format_idc can only be equal to 1 (4:2:0)
694	if( pcSPS->getChromaFormatIdc () == 3 )
695	{
696	WRITE_FLAG( 0, "separate_colour_plane_flag");
697	}
698
699	WRITE_UVLC( pcSPS->getPicWidthInLumaSamples (), "pic_width_in_luma_samples" );
700	WRITE_UVLC( pcSPS->getPicHeightInLumaSamples(), "pic_height_in_luma_samples" );
701	#if H_MV
702	}
703	#endif
704	Window conf = pcSPS->getConformanceWindow();
705
706	WRITE_FLAG( conf.getWindowEnabledFlag(), "conformance_window_flag" );
707	if (conf.getWindowEnabledFlag())
708	{
709	WRITE_UVLC( conf.getWindowLeftOffset() / TComSPS::getWinUnitX(pcSPS->getChromaFormatIdc() ), "conf_win_left_offset" );
710	WRITE_UVLC( conf.getWindowRightOffset() / TComSPS::getWinUnitX(pcSPS->getChromaFormatIdc() ), "conf_win_right_offset" );
711	WRITE_UVLC( conf.getWindowTopOffset() / TComSPS::getWinUnitY(pcSPS->getChromaFormatIdc() ), "conf_win_top_offset" );
712	WRITE_UVLC( conf.getWindowBottomOffset() / TComSPS::getWinUnitY(pcSPS->getChromaFormatIdc() ), "conf_win_bottom_offset" );
713	}
714	#if H_MV
715	if ( pcSPS->getLayerId() == 0 )
716	{
717	#endif
718	WRITE_UVLC( pcSPS->getBitDepthY() - 8, "bit_depth_luma_minus8" );
719	WRITE_UVLC( pcSPS->getBitDepthC() - 8, "bit_depth_chroma_minus8" );
720	#if H_MV
721	}
722	#endif
723	WRITE_UVLC( pcSPS->getBitsForPOC()-4, "log2_max_pic_order_cnt_lsb_minus4" );
724	#if H_MV
725	if ( pcSPS->getLayerId() == 0 )
726	{
727	#endif
728	const Bool subLayerOrderingInfoPresentFlag = 1;
729	WRITE_FLAG(subLayerOrderingInfoPresentFlag, "sps_sub_layer_ordering_info_present_flag");
730	for(UInt i=0; i <= pcSPS->getMaxTLayers()-1; i++)
731	{
732	WRITE_UVLC( pcSPS->getMaxDecPicBuffering(i) - 1, "sps_max_dec_pic_buffering_minus1[i]" );
733	WRITE_UVLC( pcSPS->getNumReorderPics(i), "sps_num_reorder_pics[i]" );
734	WRITE_UVLC( pcSPS->getMaxLatencyIncrease(i), "sps_max_latency_increase_plus1[i]" );
735	if (!subLayerOrderingInfoPresentFlag)
736	{
737	break;
738	}
739	}
740	#if H_MV
741	}
742	#endif
743
744	assert( pcSPS->getMaxCUWidth() == pcSPS->getMaxCUHeight() );
745
746	WRITE_UVLC( pcSPS->getLog2MinCodingBlockSize() - 3, "log2_min_coding_block_size_minus3" );
747	WRITE_UVLC( pcSPS->getLog2DiffMaxMinCodingBlockSize(), "log2_diff_max_min_coding_block_size" );
748	WRITE_UVLC( pcSPS->getQuadtreeTULog2MinSize() - 2, "log2_min_transform_block_size_minus2" );
749	WRITE_UVLC( pcSPS->getQuadtreeTULog2MaxSize() - pcSPS->getQuadtreeTULog2MinSize(), "log2_diff_max_min_transform_block_size" );
750	WRITE_UVLC( pcSPS->getQuadtreeTUMaxDepthInter() - 1, "max_transform_hierarchy_depth_inter" );
751	WRITE_UVLC( pcSPS->getQuadtreeTUMaxDepthIntra() - 1, "max_transform_hierarchy_depth_intra" );
752	WRITE_FLAG( pcSPS->getScalingListFlag() ? 1 : 0, "scaling_list_enabled_flag" );
753	if(pcSPS->getScalingListFlag())
754	{
755	#if H_MV
756	if ( pcSPS->getLayerId() > 0 )
757	{
758	WRITE_FLAG( pcSPS->getSpsInferScalingListFlag( ) ? 1 : 0 , "sps_infer_scaling_list_flag" );
759	}
760
761	if ( pcSPS->getSpsInferScalingListFlag() )
762	{
763	WRITE_CODE( pcSPS->getSpsScalingListRefLayerId( ), 6, "sps_scaling_list_ref_layer_id" );
764	}
765	else
766	{
767	#endif
768	WRITE_FLAG( pcSPS->getScalingListPresentFlag() ? 1 : 0, "sps_scaling_list_data_present_flag" );
769	if(pcSPS->getScalingListPresentFlag())
770	{
771	codeScalingList( m_pcSlice->getScalingList() );
772	}
773	#if H_MV
774	}
775	#endif
776	}
777	WRITE_FLAG( pcSPS->getUseAMP() ? 1 : 0, "amp_enabled_flag" );
778	WRITE_FLAG( pcSPS->getUseSAO() ? 1 : 0, "sample_adaptive_offset_enabled_flag");
779
780	WRITE_FLAG( pcSPS->getUsePCM() ? 1 : 0, "pcm_enabled_flag");
781	if( pcSPS->getUsePCM() )
782	{
783	WRITE_CODE( pcSPS->getPCMBitDepthLuma() - 1, 4, "pcm_sample_bit_depth_luma_minus1" );
784	WRITE_CODE( pcSPS->getPCMBitDepthChroma() - 1, 4, "pcm_sample_bit_depth_chroma_minus1" );
785	WRITE_UVLC( pcSPS->getPCMLog2MinSize() - 3, "log2_min_pcm_luma_coding_block_size_minus3" );
786	WRITE_UVLC( pcSPS->getPCMLog2MaxSize() - pcSPS->getPCMLog2MinSize(), "log2_diff_max_min_pcm_luma_coding_block_size" );
787	WRITE_FLAG( pcSPS->getPCMFilterDisableFlag()?1 : 0, "pcm_loop_filter_disable_flag");
788	}
789
790	assert( pcSPS->getMaxTLayers() > 0 );
791
792	TComRPSList* rpsList = pcSPS->getRPSList();
793	TComReferencePictureSet* rps;
794
795	WRITE_UVLC(rpsList->getNumberOfReferencePictureSets(), "num_short_term_ref_pic_sets" );
796	for(Int i=0; i < rpsList->getNumberOfReferencePictureSets(); i++)
797	{
798	rps = rpsList->getReferencePictureSet(i);
799	codeShortTermRefPicSet(pcSPS,rps,false, i);
800	}
801	WRITE_FLAG( pcSPS->getLongTermRefsPresent() ? 1 : 0, "long_term_ref_pics_present_flag" );
802	if (pcSPS->getLongTermRefsPresent())
803	{
804	WRITE_UVLC(pcSPS->getNumLongTermRefPicSPS(), "num_long_term_ref_pic_sps" );
805	for (UInt k = 0; k < pcSPS->getNumLongTermRefPicSPS(); k++)
806	{
807	WRITE_CODE( pcSPS->getLtRefPicPocLsbSps(k), pcSPS->getBitsForPOC(), "lt_ref_pic_poc_lsb_sps");
808	WRITE_FLAG( pcSPS->getUsedByCurrPicLtSPSFlag(k), "used_by_curr_pic_lt_sps_flag");
809	}
810	}
811	WRITE_FLAG( pcSPS->getTMVPFlagsPresent() ? 1 : 0, "sps_temporal_mvp_enable_flag" );
812
813	WRITE_FLAG( pcSPS->getUseStrongIntraSmoothing(), "sps_strong_intra_smoothing_enable_flag" );
814
815	WRITE_FLAG( pcSPS->getVuiParametersPresentFlag(), "vui_parameters_present_flag" );
816	if (pcSPS->getVuiParametersPresentFlag())
817	{
818	codeVUI(pcSPS->getVuiParameters(), pcSPS);
819	}
820
821	#if !H_MV
822	WRITE_FLAG( 0, "sps_extension_flag" );
823	#else
824	WRITE_FLAG( pcSPS->getSpsExtensionFlag(), "sps_extension_flag" );
825
826	if ( pcSPS->getSpsExtensionFlag() )
827	{
828	for (Int i = 0; i < PS_EX_T_MAX_NUM; i++)
829	{
830	WRITE_FLAG( pcSPS->getSpsExtensionTypeFlag( i ) ? 1 : 0 , "sps_extension_type_flag" );
831	#if H_3D
832	assert( !pcSPS->getSpsExtensionTypeFlag( i ) \|\| i == PS_EX_T_MV \|\| i == PS_EX_T_3D );
833	#else
834	assert( !pcSPS->getSpsExtensionTypeFlag( i ) \|\| i == PS_EX_T_MV );
835	#endif
836	}
837
838	if( pcSPS->getSpsExtensionTypeFlag( PS_EX_T_MV ))
839	{
840	codeSPSExtension( pcSPS );
841	}
842
843	#if H_3D
844	if( pcSPS->getSpsExtensionTypeFlag( PS_EX_T_3D ))
845	{
846	codeSPSExtension2( pcSPS, viewIndex, depthFlag );
847	}
848	#endif
849	}
850	#endif
851	}
852
853	#if H_MV
854	Void TEncCavlc::codeSPSExtension( TComSPS* pcSPS )
855	{
856	WRITE_FLAG( pcSPS->getInterViewMvVertConstraintFlag() ? 1 : 0, "inter_view_mv_vert_constraint_flag" );
857
858	WRITE_UVLC( pcSPS->getNumScaledRefLayerOffsets( ), "num_scaled_ref_layer_offsets" );
859
860	for( Int i = 0; i < pcSPS->getNumScaledRefLayerOffsets( ); i++)
861	{
862	WRITE_CODE( pcSPS->getScaledRefLayerId( i ), 6, "scaled_ref_layer_id" );
863
864	Int j = pcSPS->getScaledRefLayerId( i );
865
866	WRITE_SVLC( pcSPS->getScaledRefLayerLeftOffset( j ), "scaled_ref_layer_left_offset" );
867	WRITE_SVLC( pcSPS->getScaledRefLayerTopOffset( j ), "scaled_ref_layer_top_offset" );
868	WRITE_SVLC( pcSPS->getScaledRefLayerRightOffset( j ), "scaled_ref_layer_right_offset" );
869	WRITE_SVLC( pcSPS->getScaledRefLayerBottomOffset( j ), "scaled_ref_layer_bottom_offset" );
870	WRITE_FLAG( 0, "sps_multilayer_ext_reserved_zero_flag[ j ]" );
871	}
872	}
873	#endif
874
875	#if H_3D
876	Void TEncCavlc::codeSPSExtension2( TComSPS* pcSPS, Int viewIndex, Bool depthFlag )
877	{
878	#if H_3D_QTLPC
879	//GT: This has to go to VPS
880	if( depthFlag )
881	{
882	WRITE_FLAG( pcSPS->getUseQTL() ? 1 : 0, "use_qtl_flag");
883	WRITE_FLAG( pcSPS->getUsePC() ? 1 : 0, "use_pc_flag");
884	}
885	#endif
886	}
887	#endif
888
889	Void TEncCavlc::codeVPS( TComVPS* pcVPS )
890	{
891	WRITE_CODE( pcVPS->getVPSId(), 4, "vps_video_parameter_set_id" );
892	WRITE_CODE( 3, 2, "vps_reserved_three_2bits" );
893	#if H_MV
894	WRITE_CODE( pcVPS->getMaxLayersMinus1(), 6, "vps_max_layers_minus1" );
895	#else
896	WRITE_CODE( 0, 6, "vps_reserved_zero_6bits" );
897	#endif
898	WRITE_CODE( pcVPS->getMaxTLayers() - 1, 3, "vps_max_sub_layers_minus1" );
899	WRITE_FLAG( pcVPS->getTemporalNestingFlag(), "vps_temporal_id_nesting_flag" );
900	assert (pcVPS->getMaxTLayers()>1\|\|pcVPS->getTemporalNestingFlag());
901	WRITE_CODE( 0xffff, 16, "vps_reserved_ffff_16bits" );
902	codePTL( pcVPS->getPTL(), true, pcVPS->getMaxTLayers() - 1 );
903	const Bool subLayerOrderingInfoPresentFlag = 1;
904	WRITE_FLAG(subLayerOrderingInfoPresentFlag, "vps_sub_layer_ordering_info_present_flag");
905	for(UInt i=0; i <= pcVPS->getMaxTLayers()-1; i++)
906	{
907	WRITE_UVLC( pcVPS->getMaxDecPicBuffering(i) - 1, "vps_max_dec_pic_buffering_minus1[i]" );
908	WRITE_UVLC( pcVPS->getNumReorderPics(i), "vps_num_reorder_pics[i]" );
909	WRITE_UVLC( pcVPS->getMaxLatencyIncrease(i), "vps_max_latency_increase_plus1[i]" );
910	if (!subLayerOrderingInfoPresentFlag)
911	{
912	break;
913	}
914	}
915
916	assert( pcVPS->getNumHrdParameters() <= MAX_VPS_NUM_HRD_PARAMETERS );
917	#if H_MV
918	assert( pcVPS->getVpsMaxLayerId() < MAX_VPS_NUH_LAYER_ID_PLUS1 );
919	WRITE_CODE( pcVPS->getVpsMaxLayerId(), 6, "vps_max_layer_id" );
920
921	WRITE_UVLC( pcVPS->getVpsNumLayerSetsMinus1(), "vps_max_num_layer_sets_minus1" );
922	for( UInt opsIdx = 1; opsIdx <= pcVPS->getVpsNumLayerSetsMinus1(); opsIdx ++ )
923	{
924	// Operation point set
925	for( UInt i = 0; i <= pcVPS->getVpsMaxLayerId(); i ++ )
926	{
927	#else
928	assert( pcVPS->getMaxNuhReservedZeroLayerId() < MAX_VPS_NUH_RESERVED_ZERO_LAYER_ID_PLUS1 );
929	WRITE_CODE( pcVPS->getMaxNuhReservedZeroLayerId(), 6, "vps_max_nuh_reserved_zero_layer_id" );
930	pcVPS->setMaxOpSets(1);
931	WRITE_UVLC( pcVPS->getMaxOpSets() - 1, "vps_max_op_sets_minus1" );
932	for( UInt opsIdx = 1; opsIdx <= ( pcVPS->getMaxOpSets() - 1 ); opsIdx ++ )
933	{
934	// Operation point set
935	for( UInt i = 0; i <= pcVPS->getMaxNuhReservedZeroLayerId(); i ++ )
936	{
937	// Only applicable for version 1
938	pcVPS->setLayerIdIncludedFlag( true, opsIdx, i );
939	#endif
940	WRITE_FLAG( pcVPS->getLayerIdIncludedFlag( opsIdx, i ) ? 1 : 0, "layer_id_included_flag[opsIdx][i]" );
941	}
942	}
943	TimingInfo *timingInfo = pcVPS->getTimingInfo();
944	WRITE_FLAG(timingInfo->getTimingInfoPresentFlag(), "vps_timing_info_present_flag");
945	if(timingInfo->getTimingInfoPresentFlag())
946	{
947	WRITE_CODE(timingInfo->getNumUnitsInTick(), 32, "vps_num_units_in_tick");
948	WRITE_CODE(timingInfo->getTimeScale(), 32, "vps_time_scale");
949	WRITE_FLAG(timingInfo->getPocProportionalToTimingFlag(), "vps_poc_proportional_to_timing_flag");
950	if(timingInfo->getPocProportionalToTimingFlag())
951	{
952	WRITE_UVLC(timingInfo->getNumTicksPocDiffOneMinus1(), "vps_num_ticks_poc_diff_one_minus1");
953	}
954	pcVPS->setNumHrdParameters( 0 );
955	WRITE_UVLC( pcVPS->getNumHrdParameters(), "vps_num_hrd_parameters" );
956
957	if( pcVPS->getNumHrdParameters() > 0 )
958	{
959	pcVPS->createHrdParamBuffer();
960	}
961	for( UInt i = 0; i < pcVPS->getNumHrdParameters(); i ++ )
962	{
963	// Only applicable for version 1
964	pcVPS->setHrdOpSetIdx( 0, i );
965	WRITE_UVLC( pcVPS->getHrdOpSetIdx( i ), "hrd_op_set_idx" );
966	if( i > 0 )
967	{
968	WRITE_FLAG( pcVPS->getCprmsPresentFlag( i ) ? 1 : 0, "cprms_present_flag[i]" );
969	}
970	codeHrdParameters(pcVPS->getHrdParameters(i), pcVPS->getCprmsPresentFlag( i ), pcVPS->getMaxTLayers() - 1);
971	}
972	}
973	#if H_MV
974	WRITE_FLAG( 1, "vps_extension_flag" );
975	m_pcBitIf->writeAlignOne();
976	codeVPSExtension( pcVPS );
977	#if H_3D
978	WRITE_FLAG( 1, "vps_extension2_flag" );
979	m_pcBitIf->writeAlignOne();
980	codeVPSExtension2( pcVPS );
981	WRITE_FLAG( 0, "vps_extension3_flag" );
982	#else
983	WRITE_FLAG( 0, "vps_extension2_flag" );
984	#endif
985	#else
986	WRITE_FLAG( 0, "vps_extension_flag" );
987	#endif
988	//future extensions here..
989
990	return;
991	}
992
993
994
995	#if H_MV
996	Void TEncCavlc::codeVPSExtension( TComVPS *pcVPS )
997	{
998	WRITE_FLAG( pcVPS->getAvcBaseLayerFlag() ? 1 : 0, "avc_base_layer_flag" );
999	WRITE_FLAG( pcVPS->getSplittingFlag() ? 1 : 0, "splitting_flag" );
1000
1001	for( Int type = 0; type < MAX_NUM_SCALABILITY_TYPES; type++ )
1002	{
1003	WRITE_FLAG( pcVPS->getScalabilityMaskFlag( type ) ? 1 : 0, "scalability_mask_flag[i]" );
1004	}
1005
1006	for( Int sIdx = 0; sIdx < pcVPS->getNumScalabilityTypes( ) - ( pcVPS->getSplittingFlag() ? 1 : 0 ); sIdx++ )
1007	{
1008	WRITE_CODE( pcVPS->getDimensionIdLen( sIdx ) - 1 , 3, "dimension_id_len_minus1[j]");
1009	}
1010
1011	if ( pcVPS->getSplittingFlag() )
1012	{ // Ignore old dimension id length
1013	pcVPS->setDimensionIdLen( pcVPS->getNumScalabilityTypes( ) - 1 ,pcVPS->inferLastDimsionIdLenMinus1() + 1 );
1014	}
1015
1016	WRITE_FLAG( pcVPS->getVpsNuhLayerIdPresentFlag() ? 1 : 0, "vps_nuh_layer_id_present_flag");
1017
1018	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1019	{
1020	if ( pcVPS->getVpsNuhLayerIdPresentFlag() )
1021	{
1022	WRITE_CODE( pcVPS->getLayerIdInNuh( i ), 6, "layer_id_in_nuh[i]");
1023	}
1024	else
1025	{
1026	assert( pcVPS->getLayerIdInNuh( i ) == i );
1027	}
1028
1029	assert( pcVPS->getLayerIdInVps( pcVPS->getLayerIdInNuh( i ) ) == i );
1030
1031	for( Int j = 0; j < pcVPS->getNumScalabilityTypes() ; j++ )
1032	{
1033	if ( !pcVPS->getSplittingFlag() )
1034	{
1035	WRITE_CODE( pcVPS->getDimensionId( i, j ), pcVPS->getDimensionIdLen( j ), "dimension_id[i][j]");
1036	}
1037	else
1038	{
1039	assert( pcVPS->getDimensionId( i, j ) == pcVPS->inferDimensionId( i, j ) );
1040	}
1041	}
1042	}
1043
1044
1045	WRITE_CODE( pcVPS->getViewIdLen( ), 4, "view_id_len" );
1046
1047	if ( pcVPS->getViewIdLen( ) > 0 )
1048	{
1049	for( Int i = 0; i < pcVPS->getNumViews(); i++ )
1050	{
1051	WRITE_CODE( pcVPS->getViewIdVal( i ), pcVPS->getViewIdLen( ), "view_id_val[i]" );
1052	}
1053	}
1054	else
1055	{
1056	for( Int i = 0; i < pcVPS->getNumViews(); i++ )
1057	{
1058	assert( pcVPS->getViewIdVal( i ) == 0 );
1059	}
1060	}
1061
1062
1063	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1064	{
1065	for( Int j = 0; j < i; j++ )
1066	{
1067	WRITE_FLAG( pcVPS->getDirectDependencyFlag( i, j ), "direct_dependency_flag[i][j]" );
1068	}
1069	}
1070	WRITE_FLAG( pcVPS->getVpsSubLayersMaxMinus1PresentFlag( ) ? 1 : 0 , "vps_sub_layers_max_minus1_present_flag" );
1071	if ( pcVPS->getVpsSubLayersMaxMinus1PresentFlag() )
1072	{
1073	for (Int i = 0; i <= pcVPS->getMaxLayersMinus1(); i++ )
1074	{
1075	WRITE_CODE( pcVPS->getSubLayersVpsMaxMinus1( i ), 3, "sub_layers_vps_max_minus1" );
1076	pcVPS->checkSubLayersVpsMaxMinus1( i );
1077	}
1078	}
1079	else
1080	{
1081	for (Int i = 0; i <= pcVPS->getMaxLayersMinus1(); i++ )
1082	{
1083	assert( pcVPS->getSubLayersVpsMaxMinus1( i ) + 1 == pcVPS->getMaxTLayers( ) );
1084	}
1085	}
1086	WRITE_FLAG( pcVPS->getMaxTidRefPresentFlag( ) ? 1 : 0 , "max_tid_ref_present_flag" );
1087
1088	if ( pcVPS->getMaxTidRefPresentFlag() )
1089	{
1090	for( Int i = 0; i < pcVPS->getMaxLayersMinus1(); i++ )
1091	{
1092	for( Int j = i + 1; j <= pcVPS->getMaxLayersMinus1(); j++ )
1093	{
1094	if ( pcVPS->getDirectDependencyFlag(j,i) )
1095	{
1096	WRITE_CODE( pcVPS->getMaxTidIlRefPicsPlus1( i, j ), 3, "max_tid_il_ref_pics_plus1" );
1097	}
1098	}
1099	}
1100	}
1101
1102	WRITE_FLAG( pcVPS->getAllRefLayersActiveFlag( ) ? 1 : 0 , "all_ref_layers_active_flag" );
1103	WRITE_UVLC( pcVPS->getVpsNumProfileTierLevelMinus1( ), "vps_num_profile_tier_level_minus1" );
1104
1105	for( Int i = 1; i <= pcVPS->getVpsNumProfileTierLevelMinus1(); i++ )
1106	{
1107	WRITE_FLAG( pcVPS->getVpsProfilePresentFlag( i ) ? 1 : 0, "vps_profile_present_flag[i]" );
1108	codePTL( pcVPS->getPTL( i ), pcVPS->getVpsProfilePresentFlag( i ), pcVPS->getMaxTLayers() - 1 );
1109	}
1110
1111	WRITE_UVLC( pcVPS->getNumAddOutputLayerSets( ), "num_add_output_layer_sets" );
1112
1113	if( pcVPS->getNumOutputLayerSets() > 1)
1114	{
1115	WRITE_CODE( pcVPS->getDefaultTargetOutputLayerIdc( ), 2, "default_target_output_layer_idc" );
1116	}
1117
1118
1119	assert( pcVPS->getOutputLayerFlag(0, 0) == pcVPS->inferOutputLayerFlag( 0, 0 ));
1120	assert( pcVPS->getOutputLayerSetIdxMinus1( 0 ) == -1 );
1121
1122	for( Int i = 1; i < pcVPS->getNumOutputLayerSets( ); i++ )
1123	{
1124	if( i > pcVPS->getVpsNumLayerSetsMinus1( ) )
1125	{
1126	WRITE_UVLC( pcVPS->getOutputLayerSetIdxMinus1( i ), "output_layer_set_idx_minus1[i]" );
1127	}
1128
1129	if ( i > pcVPS->getVpsNumLayerSetsMinus1() \|\| pcVPS->getDefaultTargetOutputLayerIdc() == 2 )
1130	{
1131	for( Int j = 0; j < pcVPS->getNumLayersInIdList( pcVPS->getLayerSetIdxForOutputLayerSet( i ) ); j++ )
1132	{
1133	WRITE_FLAG( pcVPS->getOutputLayerFlag( i, j) ? 1 : 0, "output_layer_flag" );
1134	}
1135	}
1136	else
1137	{
1138	for( Int j = 0; j < pcVPS->getNumLayersInIdList( pcVPS->getLayerSetIdxForOutputLayerSet( i ) ) - 1; j++ )
1139	{
1140	assert( pcVPS->getOutputLayerFlag( i , j ) == pcVPS->inferOutputLayerFlag( i, j ));
1141	}
1142	}
1143
1144	if ( pcVPS->getProfileLevelTierIdxLen() > 0 )
1145	{
1146	WRITE_CODE( pcVPS->getProfileLevelTierIdx( i ), pcVPS->getProfileLevelTierIdxLen() ,"profile_level_tier_idx[ i ]" );
1147	}
1148	if( pcVPS->getNumOutputLayersInOutputLayerSet( i ) == 1 && pcVPS->getNumDirectRefLayers( pcVPS->getOlsHighestOutputLayerId( i ) ) > 0 )
1149	{
1150	WRITE_FLAG( pcVPS->getAltOutputLayerFlag( i ) ? 1 : 0 , "alt_output_layer_flag[ i ]" );
1151	}
1152	}
1153
1154	WRITE_FLAG( pcVPS->getRepFormatIdxPresentFlag( ) ? 1 : 0 , "rep_format_idx_present_flag" );
1155	if ( pcVPS->getRepFormatIdxPresentFlag() )
1156	{
1157	WRITE_UVLC( pcVPS->getVpsNumRepFormatsMinus1( ), "vps_num_rep_formats_minus1" );
1158	}
1159
1160	for (Int i = 0; i <= pcVPS->getVpsNumRepFormatsMinus1(); i++ )
1161	{
1162	TComRepFormat* curRepFormat = pcVPS->getRepFormat(i);
1163	TComRepFormat* prevRepFormat = i > 0 ? pcVPS->getRepFormat( i - 1) : NULL;
1164	codeRepFormat( i, curRepFormat , prevRepFormat);
1165	}
1166
1167	if( pcVPS->getRepFormatIdxPresentFlag() )
1168	{
1169	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1170	{
1171	if( pcVPS->getVpsNumRepFormatsMinus1() > 0 )
1172	{
1173	WRITE_CODE( pcVPS->getVpsRepFormatIdx(i), pcVPS->getVpsRepFormatIdxLen(), "vps_rep_format_idx[i]" );
1174	}
1175	}
1176	}
1177
1178	WRITE_FLAG( pcVPS->getMaxOneActiveRefLayerFlag( ) ? 1 : 0, "max_one_active_ref_layer_flag" );
1179	#if H_MV_HLS7_GEN
1180	WRITE_FLAG( pcVPS->getVpsPocLsbAlignedFlag( ) ? 1 : 0 , "vps_poc_lsb_aligned_flag" );
1181	#endif
1182	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1183	{
1184	if( pcVPS->getNumDirectRefLayers( pcVPS->getLayerIdInNuh( i ) ) == 0 )
1185	{
1186	WRITE_FLAG( pcVPS->getPocLsbNotPresentFlag( i ) ? 1 : 0 , "poc_lsb_not_present_flag" );
1187	}
1188	}
1189	WRITE_FLAG( 0, "vps_reserved_zero_flag" );
1190	codeDpbSize( pcVPS );
1191
1192	WRITE_UVLC( pcVPS->getDirectDepTypeLenMinus2 ( ), "direct_dep_type_len_minus2");
1193
1194	WRITE_FLAG( pcVPS->getDefaultDirectDependencyFlag( ) ? 1 : 0 , "default_direct_dependency_flag" );
1195
1196	if ( pcVPS->getDefaultDirectDependencyFlag( ) )
1197	{
1198	WRITE_CODE( pcVPS->getDefaultDirectDependencyType( ), pcVPS->getDirectDepTypeLenMinus2( ) + 2 , "default_direct_dependency_type" );
1199	}
1200
1201	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1202	{
1203	for( Int j = 0; j < i; j++ )
1204	{
1205	if (pcVPS->getDirectDependencyFlag( i, j) )
1206	{
1207	if ( pcVPS->getDefaultDirectDependencyFlag( ) )
1208	{
1209	assert( pcVPS->getDirectDependencyType( i, j ) == pcVPS->getDefaultDirectDependencyType( ) );
1210	}
1211	else
1212	{
1213	assert ( pcVPS->getDirectDependencyType( i, j ) != -1 );
1214	WRITE_CODE( pcVPS->getDirectDependencyType( i, j ),pcVPS->getDirectDepTypeLenMinus2( ) + 2, "direct_dependency_type[i][j]" );
1215	}
1216	}
1217	}
1218	}
1219	WRITE_UVLC( 0, "vps_non_vui_extension_length" );
1220	WRITE_FLAG( pcVPS->getVpsVuiPresentFlag() ? 1 : 0 , "vps_vui_present_flag" );
1221	if( pcVPS->getVpsVuiPresentFlag() )
1222	{
1223	m_pcBitIf->writeAlignOne(); // vps_vui_alignment_bit_equal_to_one
1224	codeVPSVUI( pcVPS );
1225	}
1226	{
1227	TComVPSVUI* pcVPSVUI = pcVPS->getVPSVUI( );
1228	assert( pcVPSVUI );
1229	pcVPSVUI->inferVpsVui( true );
1230	}
1231	}
1232
1233	Void TEncCavlc::codeVideoSignalInfo( TComVideoSignalInfo* pcVideoSignalInfo )
1234	{
1235	assert( pcVideoSignalInfo );
1236	WRITE_CODE( pcVideoSignalInfo->getVideoVpsFormat( ), 3, "video_vps_format" );
1237	WRITE_FLAG( pcVideoSignalInfo->getVideoFullRangeVpsFlag( ) ? 1 : 0 , "video_full_range_vps_flag" );
1238	WRITE_CODE( pcVideoSignalInfo->getColourPrimariesVps( ), 8, "colour_primaries_vps" );
1239	WRITE_CODE( pcVideoSignalInfo->getTransferCharacteristicsVps( ), 8, "transfer_characteristics_vps" );
1240	WRITE_CODE( pcVideoSignalInfo->getMatrixCoeffsVps( ), 8, "matrix_coeffs_vps" );
1241	}
1242
1243	Void TEncCavlc::codeDpbSize( TComVPS* vps )
1244	{
1245	TComDpbSize* dpbSize = vps->getDpbSize();
1246	assert ( dpbSize != 0 );
1247
1248	for( Int i = 1; i < vps->getNumOutputLayerSets(); i++ )
1249	{
1250	WRITE_FLAG( dpbSize->getSubLayerFlagInfoPresentFlag( i ) ? 1 : 0 , "sub_layer_flag_info_present_flag" );
1251	for( Int j = 0; j <= vps->getMaxSubLayersInLayerSetMinus1( i ); j++ )
1252	{
1253	if( j > 0 && dpbSize->getSubLayerDpbInfoPresentFlag( i, j ) )
1254	{
1255	WRITE_FLAG( dpbSize->getSubLayerDpbInfoPresentFlag( i, j ) ? 1 : 0 , "sub_layer_dpb_info_present_flag" );
1256	}
1257	if( dpbSize->getSubLayerDpbInfoPresentFlag( i, j ) )
1258	{
1259	for( Int k = 0; k < vps->getNumSubDpbs( vps->getLayerSetIdxForOutputLayerSet( i )); k++ )
1260	{
1261	WRITE_UVLC( dpbSize->getMaxVpsDecPicBufferingMinus1( i, k, j ), "max_vps_dec_pic_buffering_minus1" );
1262	}
1263	WRITE_UVLC( dpbSize->getMaxVpsNumReorderPics( i, j ), "max_vps_num_reorder_pics" );
1264	#if H_MV_HLS7_GEN
1265	if( vps->getNumSubDpbs( vps->getLayerSetIdxForOutputLayerSet( i ) ) != vps->getNumLayersInIdList( vps->getLayerSetIdxForOutputLayerSet( i ) ) )
1266	{
1267	for( Int k = 0; k < vps->getNumLayersInIdList( vps->getLayerSetIdxForOutputLayerSet( i ) ); k++ )
1268	{
1269	WRITE_UVLC( dpbSize->getMaxVpsLayerDecPicBuffMinus1( i, k, j ), "max_vps_layer_dec_pic_buff_minus1" );
1270	}
1271	}
1272	#endif
1273	WRITE_UVLC( dpbSize->getMaxVpsLatencyIncreasePlus1( i, j ), "max_vps_latency_increase_plus1" );
1274	}
1275	else
1276	{
1277	if ( j > 0 )
1278	{
1279	for( Int k = 0; k < vps->getNumSubDpbs( vps->getLayerSetIdxForOutputLayerSet( i ) ); k++ )
1280	{
1281	assert( dpbSize->getMaxVpsDecPicBufferingMinus1( i, k, j ) == dpbSize->getMaxVpsDecPicBufferingMinus1( i,k, j - 1 ) );
1282	}
1283	assert( dpbSize->getMaxVpsNumReorderPics ( i, j ) == dpbSize->getMaxVpsNumReorderPics ( i, j - 1 ) );
1284	assert( dpbSize->getMaxVpsLatencyIncreasePlus1( i, j ) == dpbSize->getMaxVpsLatencyIncreasePlus1( i, j - 1 ) );
1285	}
1286	}
1287	}
1288	}
1289	}
1290
1291	Void TEncCavlc::codeRepFormat( Int i, TComRepFormat* pcRepFormat, TComRepFormat* pcPrevRepFormat )
1292	{
1293	assert( pcRepFormat );
1294
1295	WRITE_CODE( pcRepFormat->getPicWidthVpsInLumaSamples( ), 16, "pic_width_vps_in_luma_samples" );
1296	WRITE_CODE( pcRepFormat->getPicHeightVpsInLumaSamples( ), 16, "pic_height_vps_in_luma_samples" );
1297	WRITE_FLAG( pcRepFormat->getChromaAndBitDepthVpsPresentFlag( ) ? 1 : 0 , "chroma_and_bit_depth_vps_present_flag" );
1298
1299	pcRepFormat->checkChromaAndBitDepthVpsPresentFlag( i );
1300
1301	if ( pcRepFormat->getChromaAndBitDepthVpsPresentFlag() )
1302	{
1303	WRITE_CODE( pcRepFormat->getChromaFormatVpsIdc( ), 2, "chroma_format_vps_idc" );
1304
1305	if ( pcRepFormat->getChromaFormatVpsIdc() == 3 )
1306	{
1307	WRITE_FLAG( pcRepFormat->getSeparateColourPlaneVpsFlag( ) ? 1 : 0 , "separate_colour_plane_vps_flag" );
1308	}
1309	WRITE_CODE( pcRepFormat->getBitDepthVpsLumaMinus8( ), 4, "bit_depth_vps_luma_minus8" );
1310	WRITE_CODE( pcRepFormat->getBitDepthVpsChromaMinus8( ), 4, "bit_depth_vps_chroma_minus8" );
1311	}
1312	else
1313	{
1314	pcRepFormat->inferChromaAndBitDepth(pcPrevRepFormat, true );
1315	}
1316	}
1317
1318	Void TEncCavlc::codeVPSVUI( TComVPS* pcVPS )
1319	{
1320	assert( pcVPS );
1321
1322	TComVPSVUI* pcVPSVUI = pcVPS->getVPSVUI( );
1323
1324	assert( pcVPSVUI );
1325
1326	WRITE_FLAG( pcVPSVUI->getCrossLayerPicTypeAlignedFlag( ) ? 1 : 0 , "cross_layer_pic_type_aligned_flag" );
1327	if ( !pcVPSVUI->getCrossLayerPicTypeAlignedFlag() )
1328	{
1329	WRITE_FLAG( pcVPSVUI->getCrossLayerIrapAlignedFlag( ) ? 1 : 0 , "cross_layer_irap_aligned_flag" );
1330	}
1331	if( pcVPSVUI->getCrossLayerIrapAlignedFlag( ) )
1332	{
1333	WRITE_FLAG( pcVPSVUI->getAllLayersIdrAlignedFlag( ) ? 1 : 0 , "all_layers_idr_aligned_flag" );
1334	}
1335	WRITE_FLAG( pcVPSVUI->getBitRatePresentVpsFlag( ) ? 1 : 0 , "bit_rate_present_vps_flag" );
1336	WRITE_FLAG( pcVPSVUI->getPicRatePresentVpsFlag( ) ? 1 : 0 , "pic_rate_present_vps_flag" );
1337	if( pcVPSVUI->getBitRatePresentVpsFlag( ) \|\| pcVPSVUI->getPicRatePresentVpsFlag( ) )
1338	{
1339	for( Int i = 0; i <= pcVPS->getVpsNumLayerSetsMinus1(); i++ )
1340	{
1341	for( Int j = 0; j <= pcVPS->getMaxTLayers(); j++ )
1342	{
1343	if( pcVPSVUI->getBitRatePresentVpsFlag( ) )
1344	{
1345	WRITE_FLAG( pcVPSVUI->getBitRatePresentFlag( i, j ) ? 1 : 0 , "bit_rate_present_flag" );
1346	}
1347	if( pcVPSVUI->getBitRatePresentVpsFlag( ) )
1348	{
1349	WRITE_FLAG( pcVPSVUI->getPicRatePresentFlag( i, j ) ? 1 : 0 , "pic_rate_present_flag" );
1350	}
1351	if( pcVPSVUI->getBitRatePresentFlag( i, j ) )
1352	{
1353	WRITE_CODE( pcVPSVUI->getAvgBitRate( i, j ), 16, "avg_bit_rate" );
1354	WRITE_CODE( pcVPSVUI->getMaxBitRate( i, j ), 16, "max_bit_rate" );
1355	}
1356	if( pcVPSVUI->getPicRatePresentFlag( i, j ) )
1357	{
1358	WRITE_CODE( pcVPSVUI->getConstantPicRateIdc( i, j ), 2, "constant_pic_rate_idc" );
1359	WRITE_CODE( pcVPSVUI->getAvgPicRate( i, j ), 16, "avg_pic_rate" );
1360	}
1361	}
1362	}
1363	}
1364
1365	WRITE_FLAG( pcVPSVUI->getVideoSignalInfoIdxPresentFlag( ) ? 1 : 0 , "video_signal_info_idx_present_flag" );
1366	if( pcVPSVUI->getVideoSignalInfoIdxPresentFlag() )
1367	{
1368	WRITE_CODE( pcVPSVUI->getVpsNumVideoSignalInfoMinus1( ), 4, "vps_num_video_signal_info_minus1" );
1369	}
1370	else
1371	{
1372	pcVPSVUI->setVpsNumVideoSignalInfoMinus1( pcVPS->getMaxLayersMinus1() );
1373	}
1374
1375	for( Int i = 0; i <= pcVPSVUI->getVpsNumVideoSignalInfoMinus1(); i++ )
1376	{
1377	assert( pcVPSVUI->getVideoSignalInfo( i ) != NULL );
1378	TComVideoSignalInfo* curVideoSignalInfo = pcVPSVUI->getVideoSignalInfo( i );
1379	codeVideoSignalInfo( curVideoSignalInfo );
1380
1381	}
1382
1383	if( pcVPSVUI->getVideoSignalInfoIdxPresentFlag() && pcVPSVUI->getVpsNumVideoSignalInfoMinus1() > 0 )
1384	{
1385	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1386	{
1387	WRITE_CODE( pcVPSVUI->getVpsVideoSignalInfoIdx( i ), 4, "vps_video_signal_info_idx" );
1388	assert( pcVPSVUI->getVpsVideoSignalInfoIdx( i ) >= 0 && pcVPSVUI->getVpsVideoSignalInfoIdx( i ) <= pcVPSVUI->getVpsNumVideoSignalInfoMinus1() );
1389	}
1390	}
1391	else
1392	{
1393	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1394	{
1395	assert( pcVPSVUI->getVpsVideoSignalInfoIdx( i ) == ( pcVPSVUI->getVideoSignalInfoIdxPresentFlag() ? 0 : i ) );
1396	}
1397	}
1398	WRITE_FLAG( pcVPSVUI->getTilesNotInUseFlag( ) ? 1 : 0 , "tiles_not_in_use_flag" );
1399	if( !pcVPSVUI->getTilesNotInUseFlag() )
1400	{
1401	for( Int i = 0; i <= pcVPS->getMaxLayersMinus1(); i++ )
1402	{
1403	WRITE_FLAG( pcVPSVUI->getTilesInUseFlag( i ) ? 1 : 0 , "tiles_in_use_flag[i]" );
1404	if( pcVPSVUI->getTilesInUseFlag( i ) )
1405	{
1406	WRITE_FLAG( pcVPSVUI->getLoopFilterNotAcrossTilesFlag( i ) ? 1 : 0, "loop_filter_not_across_tiles_flag[i]" );
1407	}
1408	}
1409
1410	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1411	{
1412	for( Int j = 0; j < pcVPS->getNumDirectRefLayers( pcVPS->getLayerIdInNuh( i ) ) ; j++ )
1413	{
1414	Int layerIdx = pcVPS->getLayerIdInVps(pcVPS->getRefLayerId(pcVPS->getLayerIdInNuh( i ) , j ));
1415	if( pcVPSVUI->getTilesInUseFlag( i ) && pcVPSVUI->getTilesInUseFlag( layerIdx ) )
1416	{
1417	WRITE_FLAG( pcVPSVUI->getTileBoundariesAlignedFlag( i, j ) ? 1 : 0 , "tile_boundaries_aligned_flag[i][j]" );
1418	}
1419	}
1420	}
1421	}
1422
1423	WRITE_FLAG( pcVPSVUI->getWppNotInUseFlag( ) ? 1 : 0 , "wpp_not_in_use_flag" );
1424
1425	if( !pcVPSVUI->getWppNotInUseFlag( ) )
1426	{
1427	for( Int i = 0; i <= pcVPS->getMaxLayersMinus1(); i++ )
1428	{
1429	WRITE_FLAG( pcVPSVUI->getWppInUseFlag( i ) ? 1 : 0 , "wpp_in_use_flag[i]" );
1430	}
1431	}
1432	WRITE_CODE( 0, 3, "vps_vui_reserved_zero_3bits" );
1433	WRITE_FLAG( pcVPSVUI->getIlpRestrictedRefLayersFlag( ) ? 1 : 0 , "ilp_restricted_ref_layers_flag" );
1434
1435	if( pcVPSVUI->getIlpRestrictedRefLayersFlag( ) )
1436	{
1437	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1438	{
1439	for( Int j = 0; j < pcVPS->getNumDirectRefLayers( pcVPS->getLayerIdInNuh( i ) ); j++ )
1440	{
1441	WRITE_UVLC( pcVPSVUI->getMinSpatialSegmentOffsetPlus1( i, j ), "min_spatial_segment_offset_plus1" );
1442	if( pcVPSVUI->getMinSpatialSegmentOffsetPlus1( i, j ) > 0 )
1443	{
1444	WRITE_FLAG( pcVPSVUI->getCtuBasedOffsetEnabledFlag( i, j ) ? 1 : 0 , "ctu_based_offset_enabled_flag" );
1445	if( pcVPSVUI->getCtuBasedOffsetEnabledFlag( i, j ) )
1446	{
1447	WRITE_UVLC( pcVPSVUI->getMinHorizontalCtuOffsetPlus1( i, j ), "min_horizontal_ctu_offset_plus1" );
1448	}
1449	}
1450	}
1451	}
1452	}
1453
1454
1455	WRITE_FLAG( pcVPSVUI->getVpsVuiBspHrdPresentFlag( ) ? 1 : 0 , "vps_vui_bsp_hrd_present_flag" );
1456	if ( pcVPSVUI->getVpsVuiBspHrdPresentFlag( ) )
1457	{
1458	codeVpsVuiBspHrdParameters( pcVPS );
1459	}
1460	for( Int i = 1; i <= pcVPS->getMaxLayersMinus1(); i++ )
1461	{
1462	if( pcVPS->getNumDirectRefLayers( pcVPS->getLayerIdInNuh( i )) == 0 )
1463	{
1464	WRITE_FLAG( pcVPSVUI->getBaseLayerParameterSetCompatibilityFlag( i ) ? 1 : 0 , "base_layer_parameter_set_compatibility_flag" );
1465	}
1466	}
1467	}
1468
1469	Void TEncCavlc::codeVpsVuiBspHrdParameters( TComVPS* pcVPS )
1470	{
1471	assert( pcVPS );
1472
1473	TComVPSVUI* pcVPSVUI = pcVPS->getVPSVUI( );
1474
1475	assert( pcVPSVUI );
1476
1477	TComVpsVuiBspHrdParameters* vpsVuiBspHrdP = pcVPSVUI->getVpsVuiBspHrdParameters();
1478
1479	assert ( vpsVuiBspHrdP );
1480
1481
1482	WRITE_UVLC( vpsVuiBspHrdP->getVpsNumBspHrdParametersMinus1( ), "vps_num_bsp_hrd_parameters_minus1" );
1483	for( Int i = 0; i <= vpsVuiBspHrdP->getVpsNumBspHrdParametersMinus1( ); i++ )
1484	{
1485	if( i > 0 )
1486	{
1487	WRITE_FLAG( vpsVuiBspHrdP->getBspCprmsPresentFlag( i ) ? 1 : 0 , "bsp_cprms_present_flag" );
1488	}
1489	TComHRD* hrdParameters = vpsVuiBspHrdP->getHrdParametermeters( i );
1490	codeHrdParameters( hrdParameters, vpsVuiBspHrdP->getBspCprmsPresentFlag( i ), pcVPS->getMaxSubLayersMinus1() );
1491	}
1492	for( Int h = 1; h <= pcVPS->getVpsNumLayerSetsMinus1(); h++ )
1493	{
1494	WRITE_UVLC( vpsVuiBspHrdP->getNumBitstreamPartitions( h ), "num_bitstream_partitions" );
1495	for( Int i = 0; i < vpsVuiBspHrdP->getNumBitstreamPartitions( h ); i++ )
1496	{
1497	for( Int j = 0; j <= pcVPS->getMaxLayersMinus1(); j++ )
1498	{
1499	if( pcVPS->getLayerIdIncludedFlag( h ,j ) )
1500	{
1501	WRITE_FLAG( vpsVuiBspHrdP->getLayerInBspFlag( h, i, j ) ? 1 : 0 , "layer_in_bsp_flag" );
1502	}
1503	else
1504	{
1505	vpsVuiBspHrdP->setLayerInBspFlag( h, i, j, false ); // This inference seems to be missing in spec
1506	}
1507	}
1508	}
1509	vpsVuiBspHrdP->checkLayerInBspFlag( pcVPS, h );
1510
1511	if( vpsVuiBspHrdP->getNumBitstreamPartitions( h ) )
1512	{
1513	WRITE_UVLC( vpsVuiBspHrdP->getNumBspSchedCombinations( h ), "num_bsp_sched_combinations" );
1514	for( Int i = 0; i < vpsVuiBspHrdP->getNumBspSchedCombinations( h ); i++ )
1515	{
1516	for( Int j = 0; j < vpsVuiBspHrdP->getNumBitstreamPartitions( h ); j++ )
1517	{
1518	WRITE_UVLC( vpsVuiBspHrdP->getBspCombHrdIdx( h, i, j ), "bsp_comb_hrd_idx" );
1519	WRITE_UVLC( vpsVuiBspHrdP->getBspCombSchedIdx( h, i, j ), "bsp_comb_sched_idx" );
1520	}
1521	}
1522	}
1523	}
1524	}
1525
1526	#endif
1527
1528	#if H_3D
1529	Void TEncCavlc::codeVPSExtension2( TComVPS* pcVPS )
1530	{
1531	for( Int i = 0; i <= pcVPS->getMaxLayersMinus1(); i++ )
1532	{
1533	if (i!= 0)
1534	{
1535	if ( !( pcVPS->getDepthId( i ) == 1 ) )
1536	{
1537	#if H_3D_IV_MERGE
1538	WRITE_FLAG( pcVPS->getIvMvPredFlag ( i ) ? 1 : 0 , "iv_mv_pred_flag[i]");
1539	#if H_3D_SPIVMP
1540	WRITE_UVLC( pcVPS->getSubPULog2Size(i)-3, "log2_sub_PU_size_minus3[i]");
1541	#endif
1542	#endif
1543	#if H_3D_ARP
1544	WRITE_FLAG( pcVPS->getUseAdvRP ( i ) ? 1 : 0, "iv_res_pred_flag[i]" );
1545	#endif
1546	#if H_3D_NBDV_REF
1547	WRITE_FLAG( pcVPS->getDepthRefinementFlag ( i ) ? 1 : 0 , "depth_refinement_flag[i]");
1548	#endif
1549	#if H_3D_VSP
1550	WRITE_FLAG( pcVPS->getViewSynthesisPredFlag( i ) ? 1 : 0 , "view_synthesis_pred_flag[i]");
1551	#endif
1552	#if H_3D_DBBP
1553	WRITE_FLAG( pcVPS->getUseDBBP( i ) ? 1 : 0, "use_dbbp_flag[i]" );
1554	#endif
1555	}
1556	else
1557	{
1558	if(i!=1)
1559	{
1560	WRITE_FLAG( pcVPS->getIvMvPredFlag ( i ) ? 1 : 0 , "iv_mv_pred_flag[i]");
1561	}
1562	#if H_3D_SPIVMP
1563	if (i!=1)
1564	{
1565	WRITE_UVLC( pcVPS->getSubPULog2Size(i)-3, "log2_sub_PU_size_minus3[i]");
1566	}
1567	#endif
1568	#if H_3D_IV_MERGE
1569	WRITE_FLAG( pcVPS->getMPIFlag( i ) ? 1 : 0 , "mpi_flag[i]" );
1570	#endif
1571	WRITE_FLAG( pcVPS->getVpsDepthModesFlag( i ) ? 1 : 0 , "vps_depth_modes_flag[i]" );
1572	//WRITE_FLAG( pcVPS->getLimQtPredFlag ( i ) ? 1 : 0 , "lim_qt_pred_flag[i]" );
1573	#if H_3D_INTER_SDC
1574	WRITE_FLAG( pcVPS->getInterSDCFlag( i ) ? 1 : 0, "depth_inter_SDC_flag" );
1575	#endif
1576	}
1577	}
1578	}
1579	WRITE_UVLC( pcVPS->getCamParPrecision(), "cp_precision" );
1580	for (UInt viewIndex=0; viewIndex<pcVPS->getNumViews(); viewIndex++)
1581	{
1582	WRITE_FLAG( pcVPS->getCamParPresent(viewIndex) ? 1 : 0, "cp_present_flag[i]" );
1583	if ( pcVPS->getCamParPresent(viewIndex) )
1584	{
1585	WRITE_FLAG( pcVPS->hasCamParInSliceHeader(viewIndex) ? 1 : 0, "cp_in_slice_segment_header_flag[i]" );
1586	if ( !pcVPS->hasCamParInSliceHeader(viewIndex) )
1587	{
1588	for( UInt uiIndex = 0; uiIndex < viewIndex; uiIndex++ )
1589	{
1590	WRITE_SVLC( pcVPS->getCodedScale (viewIndex)[ uiIndex ], "vps_cp_scale" );
1591	WRITE_SVLC( pcVPS->getCodedOffset (viewIndex)[ uiIndex ], "vps_cp_off" );
1592	WRITE_SVLC( pcVPS->getInvCodedScale (viewIndex)[ uiIndex ] + pcVPS->getCodedScale (viewIndex)[ uiIndex ], "vps_cp_inv_scale_plus_scale" );
1593	WRITE_SVLC( pcVPS->getInvCodedOffset(viewIndex)[ uiIndex ] + pcVPS->getCodedOffset(viewIndex)[ uiIndex ], "vps_cp_inv_off_plus_off" );
1594	}
1595	}
1596	}
1597	}
1598	WRITE_UVLC( pcVPS->getSubPUMPILog2Size( ) - 3, "log2_sub_PU_MPI_size_minus3");
1599	#if H_3D_TMVP
1600	WRITE_FLAG( pcVPS->getIvMvScalingFlag( ) ? 1 : 0 , "iv_mv_scaling_flag" );
1601	#endif
1602	}
1603	#endif
1604
1605	Void TEncCavlc::codeSliceHeader ( TComSlice* pcSlice )
1606	{
1607	#if H_MV
1608	TComVPS* vps = pcSlice->getVPS();
1609	#endif
1610	#if ENC_DEC_TRACE
1611	xTraceSliceHeader (pcSlice);
1612	#endif
1613
1614	//calculate number of bits required for slice address
1615	Int maxSliceSegmentAddress = pcSlice->getPic()->getNumCUsInFrame();
1616	Int bitsSliceSegmentAddress = 0;
1617	while(maxSliceSegmentAddress>(1<<bitsSliceSegmentAddress))
1618	{
1619	bitsSliceSegmentAddress++;
1620	}
1621	Int ctuAddress;
1622	if (pcSlice->isNextSlice())
1623	{
1624	// Calculate slice address
1625	ctuAddress = (pcSlice->getSliceCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU());
1626	}
1627	else
1628	{
1629	// Calculate slice address
1630	ctuAddress = (pcSlice->getSliceSegmentCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU());
1631	}
1632	//write slice address
1633	Int sliceSegmentAddress = pcSlice->getPic()->getPicSym()->getCUOrderMap(ctuAddress);
1634
1635	WRITE_FLAG( sliceSegmentAddress==0, "first_slice_segment_in_pic_flag" );
1636	if ( pcSlice->getRapPicFlag() )
1637	{
1638	WRITE_FLAG( 0, "no_output_of_prior_pics_flag" );
1639	}
1640	#if PPS_FIX_DEPTH
1641	if( pcSlice->getIsDepth() )
1642	{
1643	WRITE_UVLC( 1, "slice_pic_parameter_set_id" );
1644	}
1645	else
1646	#endif
1647	WRITE_UVLC( pcSlice->getPPS()->getPPSId(), "slice_pic_parameter_set_id" );
1648	pcSlice->setDependentSliceSegmentFlag(!pcSlice->isNextSlice());
1649	if ( pcSlice->getPPS()->getDependentSliceSegmentsEnabledFlag() && (sliceSegmentAddress!=0) )
1650	{
1651	WRITE_FLAG( pcSlice->getDependentSliceSegmentFlag() ? 1 : 0, "dependent_slice_segment_flag" );
1652	}
1653	if(sliceSegmentAddress>0)
1654	{
1655	WRITE_CODE( sliceSegmentAddress, bitsSliceSegmentAddress, "slice_segment_address" );
1656	}
1657	if ( !pcSlice->getDependentSliceSegmentFlag() )
1658	{
1659	#if H_MV
1660	Int esb = 0; //Don't use i, otherwise will shadow something below
1661
1662	if ( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > esb )
1663	{
1664	esb++;
1665	WRITE_FLAG( pcSlice->getDiscardableFlag( ) ? 1 : 0 , "discardable_flag" );
1666	}
1667
1668	if ( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > esb )
1669	{
1670	esb++;
1671	WRITE_FLAG( pcSlice->getCrossLayerBlaFlag( ) ? 1 : 0 , "cross_layer_bla_flag" );
1672	}
1673	pcSlice->checkCrossLayerBlaFlag( );
1674	#if !H_MV_HLS7_GEN
1675	if ( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > esb )
1676	{
1677	esb++;
1678	WRITE_FLAG( pcSlice->getPocResetFlag( ) ? 1 : 0 , "poc_reset_flag" );
1679	}
1680	#endif
1681
1682
1683
1684	for (; esb < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); esb++)
1685	#else
1686	for (Int i = 0; i < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); i++)
1687	#endif
1688	{
1689	assert(!!"slice_reserved_undetermined_flag[]");
1690	WRITE_FLAG(0, "slice_reserved_undetermined_flag[]");
1691	}
1692
1693	WRITE_UVLC( pcSlice->getSliceType(), "slice_type" );
1694
1695	if( pcSlice->getPPS()->getOutputFlagPresentFlag() )
1696	{
1697	WRITE_FLAG( pcSlice->getPicOutputFlag() ? 1 : 0, "pic_output_flag" );
1698	}
1699
1700	// in the first version chroma_format_idc is equal to one, thus colour_plane_id will not be present
1701	assert (pcSlice->getSPS()->getChromaFormatIdc() == 1 );
1702	// if( separate_colour_plane_flag == 1 )
1703	// colour_plane_id u(2)
1704	#if H_MV
1705	if ( (pcSlice->getLayerId() > 0 && !vps->getPocLsbNotPresentFlag( pcSlice->getLayerIdInVps())) \|\| !pcSlice->getIdrPicFlag() )
1706	{
1707	Int picOrderCntLSB = (pcSlice->getPOC()-pcSlice->getLastIDR()+(1<<pcSlice->getSPS()->getBitsForPOC())) & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
1708	WRITE_CODE( picOrderCntLSB, pcSlice->getSPS()->getBitsForPOC(), "slice_pic_order_cnt_lsb");
1709	}
1710	#endif
1711	if( !pcSlice->getIdrPicFlag() )
1712	{
1713	#if !H_MV
1714	Int picOrderCntLSB = (pcSlice->getPOC()-pcSlice->getLastIDR()+(1<<pcSlice->getSPS()->getBitsForPOC())) & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
1715	WRITE_CODE( picOrderCntLSB, pcSlice->getSPS()->getBitsForPOC(), "pic_order_cnt_lsb");
1716	#endif
1717	TComReferencePictureSet* rps = pcSlice->getRPS();
1718
1719	// check for bitstream restriction stating that:
1720	// If the current picture is a BLA or CRA picture, the value of NumPocTotalCurr shall be equal to 0.
1721	// Ideally this process should not be repeated for each slice in a picture
1722	if (pcSlice->isIRAP())
1723	{
1724	for (Int picIdx = 0; picIdx < rps->getNumberOfPictures(); picIdx++)
1725	{
1726	assert (!rps->getUsed(picIdx));
1727	}
1728	}
1729
1730	if(pcSlice->getRPSidx() < 0)
1731	{
1732	WRITE_FLAG( 0, "short_term_ref_pic_set_sps_flag");
1733	codeShortTermRefPicSet(pcSlice->getSPS(), rps, true, pcSlice->getSPS()->getRPSList()->getNumberOfReferencePictureSets());
1734	}
1735	else
1736	{
1737	WRITE_FLAG( 1, "short_term_ref_pic_set_sps_flag");
1738	Int numBits = 0;
1739	while ((1 << numBits) < pcSlice->getSPS()->getRPSList()->getNumberOfReferencePictureSets())
1740	{
1741	numBits++;
1742	}
1743	if (numBits > 0)
1744	{
1745	WRITE_CODE( pcSlice->getRPSidx(), numBits, "short_term_ref_pic_set_idx" );
1746	}
1747	}
1748	if(pcSlice->getSPS()->getLongTermRefsPresent())
1749	{
1750	Int numLtrpInSH = rps->getNumberOfLongtermPictures();
1751	Int ltrpInSPS[MAX_NUM_REF_PICS];
1752	Int numLtrpInSPS = 0;
1753	UInt ltrpIndex;
1754	Int counter = 0;
1755	for(Int k = rps->getNumberOfPictures()-1; k > rps->getNumberOfPictures()-rps->getNumberOfLongtermPictures()-1; k--)
1756	{
1757	if (findMatchingLTRP(pcSlice, &ltrpIndex, rps->getPOC(k), rps->getUsed(k)))
1758	{
1759	ltrpInSPS[numLtrpInSPS] = ltrpIndex;
1760	numLtrpInSPS++;
1761	}
1762	else
1763	{
1764	counter++;
1765	}
1766	}
1767	numLtrpInSH -= numLtrpInSPS;
1768
1769	Int bitsForLtrpInSPS = 0;
1770	while (pcSlice->getSPS()->getNumLongTermRefPicSPS() > (1 << bitsForLtrpInSPS))
1771	{
1772	bitsForLtrpInSPS++;
1773	}
1774	if (pcSlice->getSPS()->getNumLongTermRefPicSPS() > 0)
1775	{
1776	WRITE_UVLC( numLtrpInSPS, "num_long_term_sps");
1777	}
1778	WRITE_UVLC( numLtrpInSH, "num_long_term_pics");
1779	// Note that the LSBs of the LT ref. pic. POCs must be sorted before.
1780	// Not sorted here because LT ref indices will be used in setRefPicList()
1781	Int prevDeltaMSB = 0, prevLSB = 0;
1782	Int offset = rps->getNumberOfNegativePictures() + rps->getNumberOfPositivePictures();
1783	for(Int i=rps->getNumberOfPictures()-1 ; i > offset-1; i--)
1784	{
1785	if (counter < numLtrpInSPS)
1786	{
1787	if (bitsForLtrpInSPS > 0)
1788	{
1789	WRITE_CODE( ltrpInSPS[counter], bitsForLtrpInSPS, "lt_idx_sps[i]");
1790	}
1791	}
1792	else
1793	{
1794	WRITE_CODE( rps->getPocLSBLT(i), pcSlice->getSPS()->getBitsForPOC(), "poc_lsb_lt");
1795	WRITE_FLAG( rps->getUsed(i), "used_by_curr_pic_lt_flag");
1796	}
1797	WRITE_FLAG( rps->getDeltaPocMSBPresentFlag(i), "delta_poc_msb_present_flag");
1798
1799	if(rps->getDeltaPocMSBPresentFlag(i))
1800	{
1801	Bool deltaFlag = false;
1802	// First LTRP from SPS \|\| First LTRP from SH \|\| curr LSB != prev LSB
1803	if( (i == rps->getNumberOfPictures()-1) \|\| (i == rps->getNumberOfPictures()-1-numLtrpInSPS) \|\| (rps->getPocLSBLT(i) != prevLSB) )
1804	{
1805	deltaFlag = true;
1806	}
1807	if(deltaFlag)
1808	{
1809	WRITE_UVLC( rps->getDeltaPocMSBCycleLT(i), "delta_poc_msb_cycle_lt[i]" );
1810	}
1811	else
1812	{
1813	Int differenceInDeltaMSB = rps->getDeltaPocMSBCycleLT(i) - prevDeltaMSB;
1814	assert(differenceInDeltaMSB >= 0);
1815	WRITE_UVLC( differenceInDeltaMSB, "delta_poc_msb_cycle_lt[i]" );
1816	}
1817	prevLSB = rps->getPocLSBLT(i);
1818	prevDeltaMSB = rps->getDeltaPocMSBCycleLT(i);
1819	}
1820	}
1821	}
1822	if (pcSlice->getSPS()->getTMVPFlagsPresent())
1823	{
1824	WRITE_FLAG( pcSlice->getEnableTMVPFlag() ? 1 : 0, "slice_temporal_mvp_enable_flag" );
1825	}
1826	}
1827	#if H_MV
1828	Bool interLayerPredLayerIdcPresentFlag = false;
1829	Int layerId = pcSlice->getLayerId();
1830	if( pcSlice->getLayerId() > 0 && !vps->getAllRefLayersActiveFlag() && vps->getNumDirectRefLayers( layerId ) > 0 )
1831	{
1832	WRITE_FLAG( pcSlice->getInterLayerPredEnabledFlag( ) ? 1 : 0 , "inter_layer_pred_enabled_flag" );
1833	if( pcSlice->getInterLayerPredEnabledFlag() && vps->getNumDirectRefLayers( layerId ) > 1 )
1834	{
1835	if( !vps->getMaxOneActiveRefLayerFlag())
1836	{
1837	WRITE_CODE( pcSlice->getNumInterLayerRefPicsMinus1( ), pcSlice->getNumInterLayerRefPicsMinus1Len( ), "num_inter_layer_ref_pics_minus1" );
1838	}
1839	if ( pcSlice->getNumActiveRefLayerPics() != vps->getNumDirectRefLayers( layerId ) )
1840	{
1841	interLayerPredLayerIdcPresentFlag = true;
1842	for( Int idx = 0; idx < pcSlice->getNumActiveRefLayerPics(); idx++ )
1843	{
1844	WRITE_CODE( pcSlice->getInterLayerPredLayerIdc( idx ), pcSlice->getInterLayerPredLayerIdcLen( ), "inter_layer_pred_layer_idc" );
1845	}
1846	}
1847	}
1848	}
1849	if ( !interLayerPredLayerIdcPresentFlag )
1850	{
1851	for( Int i = 0; i < pcSlice->getNumActiveRefLayerPics(); i++ )
1852	{
1853	assert( pcSlice->getInterLayerPredLayerIdc( i ) == pcSlice->getRefLayerPicIdc( i ) );
1854	}
1855	}
1856	#endif
1857	if(pcSlice->getSPS()->getUseSAO())
1858	{
1859	if (pcSlice->getSPS()->getUseSAO())
1860	{
1861	WRITE_FLAG( pcSlice->getSaoEnabledFlag(), "slice_sao_luma_flag" );
1862	WRITE_FLAG( pcSlice->getSaoEnabledFlagChroma(), "slice_sao_chroma_flag" );
1863	}
1864	}
1865
1866	//check if numrefidxes match the defaults. If not, override
1867
1868	if (!pcSlice->isIntra())
1869	{
1870	Bool overrideFlag = (pcSlice->getNumRefIdx( REF_PIC_LIST_0 )!=pcSlice->getPPS()->getNumRefIdxL0DefaultActive()\|\|(pcSlice->isInterB()&&pcSlice->getNumRefIdx( REF_PIC_LIST_1 )!=pcSlice->getPPS()->getNumRefIdxL1DefaultActive()));
1871	#if PPS_FIX_DEPTH
1872	overrideFlag \|= (pcSlice->getIsDepth() && !pcSlice->getViewIndex());
1873	#endif
1874	WRITE_FLAG( overrideFlag ? 1 : 0, "num_ref_idx_active_override_flag");
1875	if (overrideFlag)
1876	{
1877	WRITE_UVLC( pcSlice->getNumRefIdx( REF_PIC_LIST_0 ) - 1, "num_ref_idx_l0_active_minus1" );
1878	if (pcSlice->isInterB())
1879	{
1880	WRITE_UVLC( pcSlice->getNumRefIdx( REF_PIC_LIST_1 ) - 1, "num_ref_idx_l1_active_minus1" );
1881	}
1882	else
1883	{
1884	pcSlice->setNumRefIdx(REF_PIC_LIST_1, 0);
1885	}
1886	}
1887	}
1888	else
1889	{
1890	pcSlice->setNumRefIdx(REF_PIC_LIST_0, 0);
1891	pcSlice->setNumRefIdx(REF_PIC_LIST_1, 0);
1892	}
1893	#if PPS_FIX_DEPTH
1894	if( (pcSlice->getPPS()->getListsModificationPresentFlag() \|\| (pcSlice->getIsDepth() && !pcSlice->getViewIndex())) && pcSlice->getNumRpsCurrTempList() > 1)
1895	#else
1896	if( pcSlice->getPPS()->getListsModificationPresentFlag() && pcSlice->getNumRpsCurrTempList() > 1)
1897	#endif
1898	{
1899	TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
1900	if(!pcSlice->isIntra())
1901	{
1902	WRITE_FLAG(pcSlice->getRefPicListModification()->getRefPicListModificationFlagL0() ? 1 : 0, "ref_pic_list_modification_flag_l0" );
1903	if (pcSlice->getRefPicListModification()->getRefPicListModificationFlagL0())
1904	{
1905	Int numRpsCurrTempList0 = pcSlice->getNumRpsCurrTempList();
1906	if (numRpsCurrTempList0 > 1)
1907	{
1908	Int length = 1;
1909	numRpsCurrTempList0 --;
1910	while ( numRpsCurrTempList0 >>= 1)
1911	{
1912	length ++;
1913	}
1914	for(Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_0 ); i++)
1915	{
1916	WRITE_CODE( refPicListModification->getRefPicSetIdxL0(i), length, "list_entry_l0");
1917	}
1918	}
1919	}
1920	}
1921	if(pcSlice->isInterB())
1922	{
1923	WRITE_FLAG(pcSlice->getRefPicListModification()->getRefPicListModificationFlagL1() ? 1 : 0, "ref_pic_list_modification_flag_l1" );
1924	if (pcSlice->getRefPicListModification()->getRefPicListModificationFlagL1())
1925	{
1926	Int numRpsCurrTempList1 = pcSlice->getNumRpsCurrTempList();
1927	if ( numRpsCurrTempList1 > 1 )
1928	{
1929	Int length = 1;
1930	numRpsCurrTempList1 --;
1931	while ( numRpsCurrTempList1 >>= 1)
1932	{
1933	length ++;
1934	}
1935	for(Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_1 ); i++)
1936	{
1937	WRITE_CODE( refPicListModification->getRefPicSetIdxL1(i), length, "list_entry_l1");
1938	}
1939	}
1940	}
1941	}
1942	}
1943
1944	if (pcSlice->isInterB())
1945	{
1946	WRITE_FLAG( pcSlice->getMvdL1ZeroFlag() ? 1 : 0, "mvd_l1_zero_flag");
1947	}
1948
1949	if(!pcSlice->isIntra())
1950	{
1951	if (!pcSlice->isIntra() && pcSlice->getPPS()->getCabacInitPresentFlag())
1952	{
1953	SliceType sliceType = pcSlice->getSliceType();
1954	Int encCABACTableIdx = pcSlice->getPPS()->getEncCABACTableIdx();
1955	Bool encCabacInitFlag = (sliceType!=encCABACTableIdx && encCABACTableIdx!=I_SLICE) ? true : false;
1956	pcSlice->setCabacInitFlag( encCabacInitFlag );
1957	WRITE_FLAG( encCabacInitFlag?1:0, "cabac_init_flag" );
1958	}
1959	}
1960
1961	if ( pcSlice->getEnableTMVPFlag() )
1962	{
1963	if ( pcSlice->getSliceType() == B_SLICE )
1964	{
1965	WRITE_FLAG( pcSlice->getColFromL0Flag(), "collocated_from_l0_flag" );
1966	}
1967
1968	if ( pcSlice->getSliceType() != I_SLICE &&
1969	((pcSlice->getColFromL0Flag()==1 && pcSlice->getNumRefIdx(REF_PIC_LIST_0)>1)\|\|
1970	(pcSlice->getColFromL0Flag()==0 && pcSlice->getNumRefIdx(REF_PIC_LIST_1)>1)))
1971	{
1972	WRITE_UVLC( pcSlice->getColRefIdx(), "collocated_ref_idx" );
1973	}
1974	}
1975	if ( (pcSlice->getPPS()->getUseWP() && pcSlice->getSliceType()==P_SLICE) \|\| (pcSlice->getPPS()->getWPBiPred() && pcSlice->getSliceType()==B_SLICE) )
1976	{
1977	xCodePredWeightTable( pcSlice );
1978	}
1979	#if H_3D_IC
1980	else if( pcSlice->getViewIndex() && ( pcSlice->getSliceType() == P_SLICE \|\| pcSlice->getSliceType() == B_SLICE ) && !pcSlice->getIsDepth())
1981	{
1982	WRITE_FLAG( pcSlice->getApplyIC() ? 1 : 0, "slice_ic_enable_flag" );
1983	if( pcSlice->getApplyIC() )
1984	{
1985	WRITE_FLAG( pcSlice->getIcSkipParseFlag() ? 1 : 0, "ic_skip_mergeidx0" );
1986	}
1987	}
1988	#endif
1989
1990	#if H_3D_IV_MERGE
1991	assert(pcSlice->getMaxNumMergeCand()<=MRG_MAX_NUM_CANDS_MEM);
1992	#else
1993	assert(pcSlice->getMaxNumMergeCand()<=MRG_MAX_NUM_CANDS);
1994	#endif
1995	if (!pcSlice->isIntra())
1996	{
1997	#if H_3D_IV_MERGE
1998	if(pcSlice->getIsDepth())
1999	{
2000	Bool bMPIFlag = pcSlice->getVPS()->getMPIFlag( pcSlice->getLayerIdInVps() ) ;
2001	Bool ivMvPredFlag = pcSlice->getVPS()->getIvMvPredFlag( pcSlice->getLayerIdInVps() ) ;
2002	WRITE_UVLC( ( ( bMPIFlag \|\| ivMvPredFlag ) ? MRG_MAX_NUM_CANDS_MEM : MRG_MAX_NUM_CANDS ) - pcSlice->getMaxNumMergeCand(), "five_minus_max_num_merge_cand");
2003	}
2004	else
2005	{
2006	Bool ivMvPredFlag = pcSlice->getVPS()->getIvMvPredFlag( pcSlice->getLayerIdInVps() ) ;
2007	WRITE_UVLC( ( ivMvPredFlag ? MRG_MAX_NUM_CANDS_MEM : MRG_MAX_NUM_CANDS ) - pcSlice->getMaxNumMergeCand(), "five_minus_max_num_merge_cand");
2008	}
2009	#else
2010	WRITE_UVLC(MRG_MAX_NUM_CANDS - pcSlice->getMaxNumMergeCand(), "five_minus_max_num_merge_cand");
2011	#endif
2012	}
2013	Int iCode = pcSlice->getSliceQp() - ( pcSlice->getPPS()->getPicInitQPMinus26() + 26 );
2014	WRITE_SVLC( iCode, "slice_qp_delta" );
2015	if (pcSlice->getPPS()->getSliceChromaQpFlag())
2016	{
2017	iCode = pcSlice->getSliceQpDeltaCb();
2018	WRITE_SVLC( iCode, "slice_qp_delta_cb" );
2019	iCode = pcSlice->getSliceQpDeltaCr();
2020	WRITE_SVLC( iCode, "slice_qp_delta_cr" );
2021	}
2022	if (pcSlice->getPPS()->getDeblockingFilterControlPresentFlag())
2023	{
2024	if (pcSlice->getPPS()->getDeblockingFilterOverrideEnabledFlag() )
2025	{
2026	WRITE_FLAG(pcSlice->getDeblockingFilterOverrideFlag(), "deblocking_filter_override_flag");
2027	}
2028	if (pcSlice->getDeblockingFilterOverrideFlag())
2029	{
2030	WRITE_FLAG(pcSlice->getDeblockingFilterDisable(), "slice_disable_deblocking_filter_flag");
2031	if(!pcSlice->getDeblockingFilterDisable())
2032	{
2033	WRITE_SVLC (pcSlice->getDeblockingFilterBetaOffsetDiv2(), "slice_beta_offset_div2");
2034	WRITE_SVLC (pcSlice->getDeblockingFilterTcOffsetDiv2(), "slice_tc_offset_div2");
2035	}
2036	}
2037	}
2038
2039	Bool isSAOEnabled = (!pcSlice->getSPS()->getUseSAO())?(false):(pcSlice->getSaoEnabledFlag()\|\|pcSlice->getSaoEnabledFlagChroma());
2040	Bool isDBFEnabled = (!pcSlice->getDeblockingFilterDisable());
2041
2042	if(pcSlice->getPPS()->getLoopFilterAcrossSlicesEnabledFlag() && ( isSAOEnabled \|\| isDBFEnabled ))
2043	{
2044	WRITE_FLAG(pcSlice->getLFCrossSliceBoundaryFlag()?1:0, "slice_loop_filter_across_slices_enabled_flag");
2045	}
2046	}
2047	#if H_3D
2048	if( pcSlice->getVPS()->hasCamParInSliceHeader( pcSlice->getViewIndex() ) && !pcSlice->getIsDepth() )
2049	{
2050	for( UInt uiId = 0; uiId < pcSlice->getViewIndex(); uiId++ )
2051	{
2052	WRITE_SVLC( pcSlice->getCodedScale ()[ uiId ], "cp_scale" );
2053	WRITE_SVLC( pcSlice->getCodedOffset ()[ uiId ], "cp_off" );
2054	WRITE_SVLC( pcSlice->getInvCodedScale ()[ uiId ] + pcSlice->getCodedScale ()[ uiId ], "cp_inv_scale_plus_scale" );
2055	WRITE_SVLC( pcSlice->getInvCodedOffset()[ uiId ] + pcSlice->getCodedOffset()[ uiId ], "cp_inv_off_plus_off" );
2056	}
2057	}
2058	#endif
2059
2060
2061	#if !H_MV
2062	if(pcSlice->getPPS()->getSliceHeaderExtensionPresentFlag())
2063	{
2064	WRITE_UVLC(0,"slice_header_extension_length");
2065	}
2066	#else
2067	if(pcSlice->getPPS()->getSliceHeaderExtensionPresentFlag())
2068	{
2069	// Derive the value of PocMsbValRequiredFlag
2070	pcSlice->setPocMsbValRequiredFlag( pcSlice->getCraPicFlag() \|\| pcSlice->getBlaPicFlag()
2071	/* \|\| related to vps_poc_lsb_aligned_flag */
2072	);
2073
2074	// Determine value of SH extension length.
2075	Int shExtnLengthInBit = 0;
2076	if (pcSlice->getPPS()->getPocResetInfoPresentFlag())
2077	{
2078	shExtnLengthInBit += 2;
2079	}
2080	if (pcSlice->getPocResetIdc() > 0)
2081	{
2082	shExtnLengthInBit += 6;
2083	}
2084	if (pcSlice->getPocResetIdc() == 3)
2085	{
2086	shExtnLengthInBit += (pcSlice->getSPS()->getBitsForPOC() + 1);
2087	}
2088
2089
2090	if( !pcSlice->getPocMsbValRequiredFlag() /* TODO && pcSlice->getVPS()->getVpsPocLsbAlignedFlag() */ )
2091	{
2092	shExtnLengthInBit++; // For poc_msb_val_present_flag
2093	}
2094	else
2095	{
2096	if( pcSlice->getPocMsbValRequiredFlag() )
2097	{
2098	pcSlice->setPocMsbValPresentFlag( true );
2099	}
2100	else
2101	{
2102	pcSlice->setPocMsbValPresentFlag( false );
2103	}
2104	}
2105
2106	if( pcSlice->getPocMsbValPresentFlag() )
2107	{
2108	// Int iMaxPOClsb = 1<< pcSlice->getSPS()->getBitsForPOC(); currently unused
2109
2110	UInt lengthVal = 1;
2111	UInt tempVal = pcSlice->getPocMsbVal() + 1;
2112	assert ( tempVal );
2113	while( 1 != tempVal )
2114	{
2115	tempVal >>= 1;
2116	lengthVal += 2;
2117	}
2118	shExtnLengthInBit += lengthVal;
2119	}
2120	Int shExtnAdditionalBits = 0;
2121	if(shExtnLengthInBit % 8 != 0)
2122	{
2123	shExtnAdditionalBits = 8 - (shExtnLengthInBit % 8);
2124	}
2125	pcSlice->setSliceSegmentHeaderExtensionLength((shExtnLengthInBit + shExtnAdditionalBits) / 8);
2126
2127
2128	WRITE_UVLC( pcSlice->getSliceSegmentHeaderExtensionLength( ), "slice_segment_header_extension_length" );
2129	UInt posFollSliceSegHeaderExtLen = m_pcBitIf->getNumberOfWrittenBits();
2130	if( pcSlice->getPPS()->getPocResetInfoPresentFlag() )
2131	{
2132	WRITE_CODE( pcSlice->getPocResetIdc( ), 2, "poc_reset_idc" );
2133	}
2134	else
2135	{
2136	assert( pcSlice->getPocResetIdc( ) == 0 );
2137	}
2138
2139	if( pcSlice->getPocResetIdc() != 0 )
2140	{
2141	WRITE_CODE( pcSlice->getPocResetPeriodId( ), 6, "poc_reset_period_id" );
2142	}
2143
2144	if( pcSlice->getPocResetIdc() == 3 )
2145	{
2146	WRITE_FLAG( pcSlice->getFullPocResetFlag( ) ? 1 : 0 , "full_poc_reset_flag" );
2147	WRITE_CODE( pcSlice->getPocLsbVal( ), pcSlice->getPocLsbValLen() , "poc_lsb_val" );
2148	}
2149
2150	if( !pcSlice->getPocMsbValRequiredFlag() /* TODO && pcSlice->getVPS()->getVpsPocLsbAlignedFlag() */ )
2151	{
2152	WRITE_FLAG( pcSlice->getPocMsbValPresentFlag( ) ? 1 : 0 , "poc_msb_val_present_flag" );
2153	}
2154
2155	if( pcSlice->getPocMsbValPresentFlag() )
2156	{
2157	WRITE_UVLC( pcSlice->getPocMsbVal( ), "poc_msb_val" );
2158	}
2159
2160	while( ( m_pcBitIf->getNumberOfWrittenBits() - posFollSliceSegHeaderExtLen ) < pcSlice->getSliceSegmentHeaderExtensionLength() * 8 )
2161	{
2162	WRITE_FLAG( 0, "slice_segment_header_extension_data_bit" );
2163	}
2164
2165	assert( ( m_pcBitIf->getNumberOfWrittenBits() - posFollSliceSegHeaderExtLen ) == pcSlice->getSliceSegmentHeaderExtensionLength() * 8 );
2166	}
2167	#endif
2168	}
2169
2170	Void TEncCavlc::codePTL( TComPTL* pcPTL, Bool profilePresentFlag, Int maxNumSubLayersMinus1)
2171	{
2172	if(profilePresentFlag)
2173	{
2174	codeProfileTier(pcPTL->getGeneralPTL()); // general_...
2175	}
2176	WRITE_CODE( pcPTL->getGeneralPTL()->getLevelIdc(), 8, "general_level_idc" );
2177
2178	for (Int i = 0; i < maxNumSubLayersMinus1; i++)
2179	{
2180	#if !H_MV
2181	if(profilePresentFlag)
2182	{
2183	#endif
2184	WRITE_FLAG( pcPTL->getSubLayerProfilePresentFlag(i), "sub_layer_profile_present_flag[i]" );
2185	#if !H_MV
2186	}
2187	#endif
2188
2189	WRITE_FLAG( pcPTL->getSubLayerLevelPresentFlag(i), "sub_layer_level_present_flag[i]" );
2190	}
2191
2192	if (maxNumSubLayersMinus1 > 0)
2193	{
2194	for (Int i = maxNumSubLayersMinus1; i < 8; i++)
2195	{
2196	WRITE_CODE(0, 2, "reserved_zero_2bits");
2197	}
2198	}
2199
2200	for(Int i = 0; i < maxNumSubLayersMinus1; i++)
2201	{
2202	if( profilePresentFlag && pcPTL->getSubLayerProfilePresentFlag(i) )
2203	{
2204	codeProfileTier(pcPTL->getSubLayerPTL(i)); // sub_layer_...
2205	}
2206	if( pcPTL->getSubLayerLevelPresentFlag(i) )
2207	{
2208	WRITE_CODE( pcPTL->getSubLayerPTL(i)->getLevelIdc(), 8, "sub_layer_level_idc[i]" );
2209	}
2210	}
2211	}
2212	Void TEncCavlc::codeProfileTier( ProfileTierLevel* ptl )
2213	{
2214	WRITE_CODE( ptl->getProfileSpace(), 2 , "XXX_profile_space[]");
2215	WRITE_FLAG( ptl->getTierFlag (), "XXX_tier_flag[]" );
2216	WRITE_CODE( ptl->getProfileIdc (), 5 , "XXX_profile_idc[]" );
2217	for(Int j = 0; j < 32; j++)
2218	{
2219	WRITE_FLAG( ptl->getProfileCompatibilityFlag(j), "XXX_profile_compatibility_flag[][j]");
2220	}
2221
2222	WRITE_FLAG(ptl->getProgressiveSourceFlag(), "general_progressive_source_flag");
2223	WRITE_FLAG(ptl->getInterlacedSourceFlag(), "general_interlaced_source_flag");
2224	WRITE_FLAG(ptl->getNonPackedConstraintFlag(), "general_non_packed_constraint_flag");
2225	WRITE_FLAG(ptl->getFrameOnlyConstraintFlag(), "general_frame_only_constraint_flag");
2226
2227	WRITE_CODE(0 , 16, "XXX_reserved_zero_44bits[0..15]");
2228	WRITE_CODE(0 , 16, "XXX_reserved_zero_44bits[16..31]");
2229	WRITE_CODE(0 , 12, "XXX_reserved_zero_44bits[32..43]");
2230	}
2231
2232	/**
2233	- write wavefront substreams sizes for the slice header.
2234	.
2235	\param pcSlice Where we find the substream size information.
2236	*/
2237	Void TEncCavlc::codeTilesWPPEntryPoint( TComSlice* pSlice )
2238	{
2239	if (!pSlice->getPPS()->getTilesEnabledFlag() && !pSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
2240	{
2241	return;
2242	}
2243	UInt numEntryPointOffsets = 0, offsetLenMinus1 = 0, maxOffset = 0;
2244	Int numZeroSubstreamsAtStartOfSlice = 0;
2245	UInt *entryPointOffset = NULL;
2246	if ( pSlice->getPPS()->getTilesEnabledFlag() )
2247	{
2248	numEntryPointOffsets = pSlice->getTileLocationCount();
2249	entryPointOffset = new UInt[numEntryPointOffsets];
2250	for (Int idx=0; idx<pSlice->getTileLocationCount(); idx++)
2251	{
2252	if ( idx == 0 )
2253	{
2254	entryPointOffset [ idx ] = pSlice->getTileLocation( 0 );
2255	}
2256	else
2257	{
2258	entryPointOffset [ idx ] = pSlice->getTileLocation( idx ) - pSlice->getTileLocation( idx-1 );
2259	}
2260
2261	if ( entryPointOffset[ idx ] > maxOffset )
2262	{
2263	maxOffset = entryPointOffset[ idx ];
2264	}
2265	}
2266	}
2267	else if ( pSlice->getPPS()->getEntropyCodingSyncEnabledFlag() )
2268	{
2269	UInt* pSubstreamSizes = pSlice->getSubstreamSizes();
2270	Int maxNumParts = pSlice->getPic()->getNumPartInCU();
2271	numZeroSubstreamsAtStartOfSlice = pSlice->getSliceSegmentCurStartCUAddr()/maxNumParts/pSlice->getPic()->getFrameWidthInCU();
2272	Int numZeroSubstreamsAtEndOfSlice = pSlice->getPic()->getFrameHeightInCU()-1 - ((pSlice->getSliceSegmentCurEndCUAddr()-1)/maxNumParts/pSlice->getPic()->getFrameWidthInCU());
2273	numEntryPointOffsets = pSlice->getPPS()->getNumSubstreams() - numZeroSubstreamsAtStartOfSlice - numZeroSubstreamsAtEndOfSlice - 1;
2274	pSlice->setNumEntryPointOffsets(numEntryPointOffsets);
2275	entryPointOffset = new UInt[numEntryPointOffsets];
2276	for (Int idx=0; idx<numEntryPointOffsets; idx++)
2277	{
2278	entryPointOffset[ idx ] = ( pSubstreamSizes[ idx+numZeroSubstreamsAtStartOfSlice ] >> 3 ) ;
2279	if ( entryPointOffset[ idx ] > maxOffset )
2280	{
2281	maxOffset = entryPointOffset[ idx ];
2282	}
2283	}
2284	}
2285	// Determine number of bits "offsetLenMinus1+1" required for entry point information
2286	offsetLenMinus1 = 0;
2287	while (maxOffset >= (1u << (offsetLenMinus1 + 1)))
2288	{
2289	offsetLenMinus1++;
2290	assert(offsetLenMinus1 + 1 < 32);
2291	}
2292
2293	WRITE_UVLC(numEntryPointOffsets, "num_entry_point_offsets");
2294	if (numEntryPointOffsets>0)
2295	{
2296	WRITE_UVLC(offsetLenMinus1, "offset_len_minus1");
2297	}
2298
2299	for (UInt idx=0; idx<numEntryPointOffsets; idx++)
2300	{
2301	WRITE_CODE(entryPointOffset[ idx ]-1, offsetLenMinus1+1, "entry_point_offset_minus1");
2302	}
2303
2304	delete [] entryPointOffset;
2305	}
2306
2307	Void TEncCavlc::codeTerminatingBit ( UInt uilsLast )
2308	{
2309	}
2310
2311	Void TEncCavlc::codeSliceFinish ()
2312	{
2313	}
2314
2315	Void TEncCavlc::codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
2316	{
2317	assert(0);
2318	}
2319
2320	Void TEncCavlc::codePartSize( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2321	{
2322	assert(0);
2323	}
2324
2325	Void TEncCavlc::codePredMode( TComDataCU* pcCU, UInt uiAbsPartIdx )
2326	{
2327	assert(0);
2328	}
2329
2330	Void TEncCavlc::codeMergeFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx )
2331	{
2332	assert(0);
2333	}
2334
2335	Void TEncCavlc::codeMergeIndex ( TComDataCU* pcCU, UInt uiAbsPartIdx )
2336	{
2337	assert(0);
2338	}
2339
2340	#if H_3D_ARP
2341	Void TEncCavlc::codeARPW( TComDataCU* pcCU, UInt uiAbsPartIdx )
2342	{
2343	assert(0);
2344	}
2345	#endif
2346
2347	#if H_3D_IC
2348	Void TEncCavlc::codeICFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2349	{
2350	assert(0);
2351	}
2352	#endif
2353
2354	Void TEncCavlc::codeInterModeFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiEncMode )
2355	{
2356	assert(0);
2357	}
2358
2359	Void TEncCavlc::codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2360	{
2361	assert(0);
2362	}
2363
2364	Void TEncCavlc::codeSkipFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2365	{
2366	assert(0);
2367	}
2368
2369	Void TEncCavlc::codeSplitFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2370	{
2371	assert(0);
2372	}
2373
2374	Void TEncCavlc::codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx )
2375	{
2376	assert(0);
2377	}
2378
2379	Void TEncCavlc::codeQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth )
2380	{
2381	assert(0);
2382	}
2383
2384	Void TEncCavlc::codeQtRootCbf( TComDataCU* pcCU, UInt uiAbsPartIdx )
2385	{
2386	assert(0);
2387	}
2388
2389	Void TEncCavlc::codeQtCbfZero( TComDataCU* pcCU, TextType eType, UInt uiTrDepth )
2390	{
2391	assert(0);
2392	}
2393	Void TEncCavlc::codeQtRootCbfZero( TComDataCU* pcCU )
2394	{
2395	assert(0);
2396	}
2397
2398	Void TEncCavlc::codeTransformSkipFlags (TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, TextType eTType )
2399	{
2400	assert(0);
2401	}
2402
2403	/** Code I_PCM information.
2404	* \param pcCU pointer to CU
2405	* \param uiAbsPartIdx CU index
2406	* \returns Void
2407	*/
2408	Void TEncCavlc::codeIPCMInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2409	{
2410	assert(0);
2411	}
2412
2413	Void TEncCavlc::codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool isMultiple)
2414	{
2415	assert(0);
2416	}
2417
2418	Void TEncCavlc::codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx )
2419	{
2420	assert(0);
2421	}
2422
2423	Void TEncCavlc::codeInterDir( TComDataCU* pcCU, UInt uiAbsPartIdx )
2424	{
2425	assert(0);
2426	}
2427
2428	Void TEncCavlc::codeRefFrmIdx( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
2429	{
2430	assert(0);
2431	}
2432
2433	Void TEncCavlc::codeMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
2434	{
2435	assert(0);
2436	}
2437
2438	Void TEncCavlc::codeDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx )
2439	{
2440	Int iDQp = pcCU->getQP( uiAbsPartIdx ) - pcCU->getRefQP( uiAbsPartIdx );
2441
2442	Int qpBdOffsetY = pcCU->getSlice()->getSPS()->getQpBDOffsetY();
2443	iDQp = (iDQp + 78 + qpBdOffsetY + (qpBdOffsetY/2)) % (52 + qpBdOffsetY) - 26 - (qpBdOffsetY/2);
2444
2445	xWriteSvlc( iDQp );
2446
2447	return;
2448	}
2449
2450	Void TEncCavlc::codeCoeffNxN ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
2451	{
2452	assert(0);
2453	}
2454
2455	Void TEncCavlc::estBit( estBitsSbacStruct* pcEstBitsCabac, Int width, Int height, TextType eTType )
2456	{
2457	// printf("error : no VLC mode support in this version\n");
2458	return;
2459	}
2460
2461	// ====================================================================================================================
2462	// Protected member functions
2463	// ====================================================================================================================
2464
2465	/** code explicit wp tables
2466	* \param TComSlice* pcSlice
2467	* \returns Void
2468	*/
2469	Void TEncCavlc::xCodePredWeightTable( TComSlice* pcSlice )
2470	{
2471	wpScalingParam *wp;
2472	Bool bChroma = true; // color always present in HEVC ?
2473	Int iNbRef = (pcSlice->getSliceType() == B_SLICE ) ? (2) : (1);
2474	Bool bDenomCoded = false;
2475	UInt uiMode = 0;
2476	UInt uiTotalSignalledWeightFlags = 0;
2477	if ( (pcSlice->getSliceType()==P_SLICE && pcSlice->getPPS()->getUseWP()) \|\| (pcSlice->getSliceType()==B_SLICE && pcSlice->getPPS()->getWPBiPred()) )
2478	{
2479	uiMode = 1; // explicit
2480	}
2481	if(uiMode == 1)
2482	{
2483
2484	for ( Int iNumRef=0 ; iNumRef<iNbRef ; iNumRef++ )
2485	{
2486	RefPicList eRefPicList = ( iNumRef ? REF_PIC_LIST_1 : REF_PIC_LIST_0 );
2487
2488	for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ )
2489	{
2490	pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
2491	if ( !bDenomCoded )
2492	{
2493	Int iDeltaDenom;
2494	WRITE_UVLC( wp[0].uiLog2WeightDenom, "luma_log2_weight_denom" ); // ue(v): luma_log2_weight_denom
2495
2496	if( bChroma )
2497	{
2498	iDeltaDenom = (wp[1].uiLog2WeightDenom - wp[0].uiLog2WeightDenom);
2499	WRITE_SVLC( iDeltaDenom, "delta_chroma_log2_weight_denom" ); // se(v): delta_chroma_log2_weight_denom
2500	}
2501	bDenomCoded = true;
2502	}
2503	WRITE_FLAG( wp[0].bPresentFlag, "luma_weight_lX_flag" ); // u(1): luma_weight_lX_flag
2504	uiTotalSignalledWeightFlags += wp[0].bPresentFlag;
2505	}
2506	if (bChroma)
2507	{
2508	for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ )
2509	{
2510	pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
2511	WRITE_FLAG( wp[1].bPresentFlag, "chroma_weight_lX_flag" ); // u(1): chroma_weight_lX_flag
2512	uiTotalSignalledWeightFlags += 2*wp[1].bPresentFlag;
2513	}
2514	}
2515
2516	for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ )
2517	{
2518	pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
2519	if ( wp[0].bPresentFlag )
2520	{
2521	Int iDeltaWeight = (wp[0].iWeight - (1<<wp[0].uiLog2WeightDenom));
2522	WRITE_SVLC( iDeltaWeight, "delta_luma_weight_lX" ); // se(v): delta_luma_weight_lX
2523	WRITE_SVLC( wp[0].iOffset, "luma_offset_lX" ); // se(v): luma_offset_lX
2524	}
2525
2526	if ( bChroma )
2527	{
2528	if ( wp[1].bPresentFlag )
2529	{
2530	for ( Int j=1 ; j<3 ; j++ )
2531	{
2532	Int iDeltaWeight = (wp[j].iWeight - (1<<wp[1].uiLog2WeightDenom));
2533	WRITE_SVLC( iDeltaWeight, "delta_chroma_weight_lX" ); // se(v): delta_chroma_weight_lX
2534
2535	Int pred = ( 128 - ( ( 128*wp[j].iWeight)>>(wp[j].uiLog2WeightDenom) ) );
2536	Int iDeltaChroma = (wp[j].iOffset - pred);
2537	WRITE_SVLC( iDeltaChroma, "delta_chroma_offset_lX" ); // se(v): delta_chroma_offset_lX
2538	}
2539	}
2540	}
2541	}
2542	}
2543	assert(uiTotalSignalledWeightFlags<=24);
2544	}
2545	}
2546
2547	/** code quantization matrix
2548	* \param scalingList quantization matrix information
2549	*/
2550	Void TEncCavlc::codeScalingList( TComScalingList* scalingList )
2551	{
2552	UInt listId,sizeId;
2553	Bool scalingListPredModeFlag;
2554
2555	//for each size
2556	for(sizeId = 0; sizeId < SCALING_LIST_SIZE_NUM; sizeId++)
2557	{
2558	for(listId = 0; listId < g_scalingListNum[sizeId]; listId++)
2559	{
2560	scalingListPredModeFlag = scalingList->checkPredMode( sizeId, listId );
2561	WRITE_FLAG( scalingListPredModeFlag, "scaling_list_pred_mode_flag" );
2562	if(!scalingListPredModeFlag)// Copy Mode
2563	{
2564	WRITE_UVLC( (Int)listId - (Int)scalingList->getRefMatrixId (sizeId,listId), "scaling_list_pred_matrix_id_delta");
2565	}
2566	else// DPCM Mode
2567	{
2568	xCodeScalingList(scalingList, sizeId, listId);
2569	}
2570	}
2571	}
2572	return;
2573	}
2574	/** code DPCM
2575	* \param scalingList quantization matrix information
2576	* \param sizeIdc size index
2577	* \param listIdc list index
2578	*/
2579	Void TEncCavlc::xCodeScalingList(TComScalingList* scalingList, UInt sizeId, UInt listId)
2580	{
2581	Int coefNum = min(MAX_MATRIX_COEF_NUM,(Int)g_scalingListSize[sizeId]);
2582	UInt* scan = (sizeId == 0) ? g_auiSigLastScan [ SCAN_DIAG ] [ 1 ] : g_sigLastScanCG32x32;
2583	Int nextCoef = SCALING_LIST_START_VALUE;
2584	Int data;
2585	Int *src = scalingList->getScalingListAddress(sizeId, listId);
2586	if( sizeId > SCALING_LIST_8x8 )
2587	{
2588	WRITE_SVLC( scalingList->getScalingListDC(sizeId,listId) - 8, "scaling_list_dc_coef_minus8");
2589	nextCoef = scalingList->getScalingListDC(sizeId,listId);
2590	}
2591	for(Int i=0;i<coefNum;i++)
2592	{
2593	data = src[scan[i]] - nextCoef;
2594	nextCoef = src[scan[i]];
2595	if(data > 127)
2596	{
2597	data = data - 256;
2598	}
2599	if(data < -128)
2600	{
2601	data = data + 256;
2602	}
2603
2604	WRITE_SVLC( data, "scaling_list_delta_coef");
2605	}
2606	}
2607	Bool TEncCavlc::findMatchingLTRP ( TComSlice* pcSlice, UInt *ltrpsIndex, Int ltrpPOC, Bool usedFlag )
2608	{
2609	// Bool state = true, state2 = false;
2610	Int lsb = ltrpPOC & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
2611	for (Int k = 0; k < pcSlice->getSPS()->getNumLongTermRefPicSPS(); k++)
2612	{
2613	if ( (lsb == pcSlice->getSPS()->getLtRefPicPocLsbSps(k)) && (usedFlag == pcSlice->getSPS()->getUsedByCurrPicLtSPSFlag(k)) )
2614	{
2615	*ltrpsIndex = k;
2616	return true;
2617	}
2618	}
2619	return false;
2620	}
2621	Bool TComScalingList::checkPredMode(UInt sizeId, UInt listId)
2622	{
2623	for(Int predListIdx = (Int)listId ; predListIdx >= 0; predListIdx--)
2624	{
2625	if( !memcmp(getScalingListAddress(sizeId,listId),((listId == predListIdx) ?
2626	getScalingListDefaultAddress(sizeId, predListIdx): getScalingListAddress(sizeId, predListIdx)),sizeof(Int)*min(MAX_MATRIX_COEF_NUM,(Int)g_scalingListSize[sizeId])) // check value of matrix
2627	&& ((sizeId < SCALING_LIST_16x16) \|\| (getScalingListDC(sizeId,listId) == getScalingListDC(sizeId,predListIdx)))) // check DC value
2628	{
2629	setRefMatrixId(sizeId, listId, predListIdx);
2630	return false;
2631	}
2632	}
2633	return true;
2634	}
2635
2636	#if H_3D_INTER_SDC
2637	Void TEncCavlc::codeDeltaDC( TComDataCU* pcCU, UInt uiAbsPartIdx )
2638	{
2639	assert(0);
2640	}
2641
2642	Void TEncCavlc::codeSDCFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx )
2643	{
2644	assert(0);
2645	}
2646
2647	#endif
2648
2649	#if H_3D_DBBP
2650	Void TEncCavlc::codeDBBPFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx )
2651	{
2652	assert(0);
2653	}
2654	#endif
2655	//! \}

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

Download in other formats: