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

Last change on this file since 608 was 608, checked in by tech, 11 years ago

Merged DEV-2.0-dev0@604.

  • Property svn:eol-style set to native
File size: 67.5 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license. 
5 *
6 * Copyright (c) 2010-2013, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 *  * Redistributions of source code must retain the above copyright notice,
13 *    this list of conditions and the following disclaimer.
14 *  * Redistributions in binary form must reproduce the above copyright notice,
15 *    this list of conditions and the following disclaimer in the documentation
16 *    and/or other materials provided with the distribution.
17 *  * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 *    be used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34/** \file     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
48Void  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
57Void  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
66Void  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
79TEncCavlc::TEncCavlc()
80{
81  m_pcBitIf           = NULL;
82  m_uiCoeffCost       = 0;
83}
84
85TEncCavlc::~TEncCavlc()
86{
87}
88
89
90// ====================================================================================================================
91// Public member functions
92// ====================================================================================================================
93
94Void TEncCavlc::resetEntropy()
95{
96}
97
98
99Void TEncCavlc::codeDFFlag(UInt uiCode, const Char *pSymbolName)
100{
101  WRITE_FLAG(uiCode, pSymbolName);
102}
103Void TEncCavlc::codeDFSvlc(Int iCode, const Char *pSymbolName)
104{
105  WRITE_SVLC(iCode, pSymbolName);
106}
107
108Void 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
165Void 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  WRITE_UVLC( pcPPS->getNumRefIdxL0DefaultActive()-1,     "num_ref_idx_l0_default_active_minus1");
179  WRITE_UVLC( pcPPS->getNumRefIdxL1DefaultActive()-1,     "num_ref_idx_l1_default_active_minus1");
180
181  WRITE_SVLC( pcPPS->getPicInitQPMinus26(),                  "init_qp_minus26");
182  WRITE_FLAG( pcPPS->getConstrainedIntraPred() ? 1 : 0,      "constrained_intra_pred_flag" );
183  WRITE_FLAG( pcPPS->getUseTransformSkip() ? 1 : 0,  "transform_skip_enabled_flag" ); 
184  WRITE_FLAG( pcPPS->getUseDQP() ? 1 : 0, "cu_qp_delta_enabled_flag" );
185  if ( pcPPS->getUseDQP() )
186  {
187    WRITE_UVLC( pcPPS->getMaxCuDQPDepth(), "diff_cu_qp_delta_depth" );
188  }
189  WRITE_SVLC( pcPPS->getChromaCbQpOffset(),                   "pps_cb_qp_offset" );
190  WRITE_SVLC( pcPPS->getChromaCrQpOffset(),                   "pps_cr_qp_offset" );
191  WRITE_FLAG( pcPPS->getSliceChromaQpFlag() ? 1 : 0,          "pps_slice_chroma_qp_offsets_present_flag" );
192
193  WRITE_FLAG( pcPPS->getUseWP() ? 1 : 0,  "weighted_pred_flag" );   // Use of Weighting Prediction (P_SLICE)
194  WRITE_FLAG( pcPPS->getWPBiPred() ? 1 : 0, "weighted_bipred_flag" );  // Use of Weighting Bi-Prediction (B_SLICE)
195  WRITE_FLAG( pcPPS->getTransquantBypassEnableFlag() ? 1 : 0, "transquant_bypass_enable_flag" );
196  WRITE_FLAG( pcPPS->getTilesEnabledFlag()             ? 1 : 0, "tiles_enabled_flag" );
197  WRITE_FLAG( pcPPS->getEntropyCodingSyncEnabledFlag() ? 1 : 0, "entropy_coding_sync_enabled_flag" );
198  if( pcPPS->getTilesEnabledFlag() )
199  {
200    WRITE_UVLC( pcPPS->getNumColumnsMinus1(),                                    "num_tile_columns_minus1" );
201    WRITE_UVLC( pcPPS->getNumRowsMinus1(),                                       "num_tile_rows_minus1" );
202    WRITE_FLAG( pcPPS->getUniformSpacingFlag(),                                  "uniform_spacing_flag" );
203    if( pcPPS->getUniformSpacingFlag() == 0 )
204    {
205      for(UInt i=0; i<pcPPS->getNumColumnsMinus1(); i++)
206      {
207        WRITE_UVLC( pcPPS->getColumnWidth(i)-1,                                  "column_width_minus1" );
208      }
209      for(UInt i=0; i<pcPPS->getNumRowsMinus1(); i++)
210      {
211        WRITE_UVLC( pcPPS->getRowHeight(i)-1,                                    "row_height_minus1" );
212      }
213    }
214    if(pcPPS->getNumColumnsMinus1() !=0 || pcPPS->getNumRowsMinus1() !=0)
215    {
216      WRITE_FLAG( pcPPS->getLoopFilterAcrossTilesEnabledFlag()?1 : 0,          "loop_filter_across_tiles_enabled_flag");
217    }
218  }
219  WRITE_FLAG( pcPPS->getLoopFilterAcrossSlicesEnabledFlag()?1 : 0,        "loop_filter_across_slices_enabled_flag");
220  WRITE_FLAG( pcPPS->getDeblockingFilterControlPresentFlag()?1 : 0,       "deblocking_filter_control_present_flag");
221  if(pcPPS->getDeblockingFilterControlPresentFlag())
222  {
223    WRITE_FLAG( pcPPS->getDeblockingFilterOverrideEnabledFlag() ? 1 : 0,  "deblocking_filter_override_enabled_flag" ); 
224    WRITE_FLAG( pcPPS->getPicDisableDeblockingFilterFlag() ? 1 : 0,       "pps_disable_deblocking_filter_flag" );
225    if(!pcPPS->getPicDisableDeblockingFilterFlag())
226    {
227      WRITE_SVLC( pcPPS->getDeblockingFilterBetaOffsetDiv2(),             "pps_beta_offset_div2" );
228      WRITE_SVLC( pcPPS->getDeblockingFilterTcOffsetDiv2(),               "pps_tc_offset_div2" );
229    }
230  }
231  WRITE_FLAG( pcPPS->getScalingListPresentFlag() ? 1 : 0,                          "pps_scaling_list_data_present_flag" ); 
232  if( pcPPS->getScalingListPresentFlag() )
233  {
234#if SCALING_LIST_OUTPUT_RESULT
235    printf("PPS\n");
236#endif
237    codeScalingList( m_pcSlice->getScalingList() );
238  }
239  WRITE_FLAG( pcPPS->getListsModificationPresentFlag(), "lists_modification_present_flag");
240  WRITE_UVLC( pcPPS->getLog2ParallelMergeLevelMinus2(), "log2_parallel_merge_level_minus2");
241  WRITE_FLAG( pcPPS->getSliceHeaderExtensionPresentFlag() ? 1 : 0, "slice_segment_header_extension_present_flag");
242  WRITE_FLAG( 0, "pps_extension_flag" );
243}
244
245Void TEncCavlc::codeVUI( TComVUI *pcVUI, TComSPS* pcSPS )
246{
247#if ENC_DEC_TRACE
248  fprintf( g_hTrace, "----------- vui_parameters -----------\n");
249#endif
250  WRITE_FLAG(pcVUI->getAspectRatioInfoPresentFlag(),            "aspect_ratio_info_present_flag");
251  if (pcVUI->getAspectRatioInfoPresentFlag())
252  {
253    WRITE_CODE(pcVUI->getAspectRatioIdc(), 8,                   "aspect_ratio_idc" );
254    if (pcVUI->getAspectRatioIdc() == 255)
255    {
256      WRITE_CODE(pcVUI->getSarWidth(), 16,                      "sar_width");
257      WRITE_CODE(pcVUI->getSarHeight(), 16,                     "sar_height");
258    }
259  }
260  WRITE_FLAG(pcVUI->getOverscanInfoPresentFlag(),               "overscan_info_present_flag");
261  if (pcVUI->getOverscanInfoPresentFlag())
262  {
263    WRITE_FLAG(pcVUI->getOverscanAppropriateFlag(),             "overscan_appropriate_flag");
264  }
265  WRITE_FLAG(pcVUI->getVideoSignalTypePresentFlag(),            "video_signal_type_present_flag");
266  if (pcVUI->getVideoSignalTypePresentFlag())
267  {
268    WRITE_CODE(pcVUI->getVideoFormat(), 3,                      "video_format");
269    WRITE_FLAG(pcVUI->getVideoFullRangeFlag(),                  "video_full_range_flag");
270    WRITE_FLAG(pcVUI->getColourDescriptionPresentFlag(),        "colour_description_present_flag");
271    if (pcVUI->getColourDescriptionPresentFlag())
272    {
273      WRITE_CODE(pcVUI->getColourPrimaries(), 8,                "colour_primaries");
274      WRITE_CODE(pcVUI->getTransferCharacteristics(), 8,        "transfer_characteristics");
275      WRITE_CODE(pcVUI->getMatrixCoefficients(), 8,             "matrix_coefficients");
276    }
277  }
278
279  WRITE_FLAG(pcVUI->getChromaLocInfoPresentFlag(),              "chroma_loc_info_present_flag");
280  if (pcVUI->getChromaLocInfoPresentFlag())
281  {
282    WRITE_UVLC(pcVUI->getChromaSampleLocTypeTopField(),         "chroma_sample_loc_type_top_field");
283    WRITE_UVLC(pcVUI->getChromaSampleLocTypeBottomField(),      "chroma_sample_loc_type_bottom_field");
284  }
285
286  WRITE_FLAG(pcVUI->getNeutralChromaIndicationFlag(),           "neutral_chroma_indication_flag");
287  WRITE_FLAG(pcVUI->getFieldSeqFlag(),                          "field_seq_flag");
288  WRITE_FLAG(pcVUI->getFrameFieldInfoPresentFlag(),             "frame_field_info_present_flag");
289
290  Window defaultDisplayWindow = pcVUI->getDefaultDisplayWindow();
291  WRITE_FLAG(defaultDisplayWindow.getWindowEnabledFlag(),       "default_display_window_flag");
292  if( defaultDisplayWindow.getWindowEnabledFlag() )
293  {
294    WRITE_UVLC(defaultDisplayWindow.getWindowLeftOffset(),      "def_disp_win_left_offset");
295    WRITE_UVLC(defaultDisplayWindow.getWindowRightOffset(),     "def_disp_win_right_offset");
296    WRITE_UVLC(defaultDisplayWindow.getWindowTopOffset(),       "def_disp_win_top_offset");
297    WRITE_UVLC(defaultDisplayWindow.getWindowBottomOffset(),    "def_disp_win_bottom_offset");
298  }
299  TimingInfo *timingInfo = pcVUI->getTimingInfo();
300  WRITE_FLAG(timingInfo->getTimingInfoPresentFlag(),          "vui_timing_info_present_flag");
301  if(timingInfo->getTimingInfoPresentFlag())
302  {
303    WRITE_CODE(timingInfo->getNumUnitsInTick(), 32,           "vui_num_units_in_tick");
304    WRITE_CODE(timingInfo->getTimeScale(),      32,           "vui_time_scale");
305    WRITE_FLAG(timingInfo->getPocProportionalToTimingFlag(),  "vui_poc_proportional_to_timing_flag");
306    if(timingInfo->getPocProportionalToTimingFlag())
307    {
308      WRITE_UVLC(timingInfo->getNumTicksPocDiffOneMinus1(),   "vui_num_ticks_poc_diff_one_minus1");
309    }
310  WRITE_FLAG(pcVUI->getHrdParametersPresentFlag(),              "hrd_parameters_present_flag");
311  if( pcVUI->getHrdParametersPresentFlag() )
312  {
313    codeHrdParameters(pcVUI->getHrdParameters(), 1, pcSPS->getMaxTLayers() - 1 );
314  }
315  }
316
317  WRITE_FLAG(pcVUI->getBitstreamRestrictionFlag(),              "bitstream_restriction_flag");
318  if (pcVUI->getBitstreamRestrictionFlag())
319  {
320    WRITE_FLAG(pcVUI->getTilesFixedStructureFlag(),             "tiles_fixed_structure_flag");
321#if H_MV
322    if ( pcSPS->getLayerId() > 0 )
323    {
324      WRITE_FLAG( pcVUI->getTileBoundariesAlignedFlag( ) ? 1 : 0 , "tile_boundaries_aligned_flag" );
325    }
326#endif
327    WRITE_FLAG(pcVUI->getMotionVectorsOverPicBoundariesFlag(),  "motion_vectors_over_pic_boundaries_flag");
328    WRITE_FLAG(pcVUI->getRestrictedRefPicListsFlag(),           "restricted_ref_pic_lists_flag");
329    WRITE_UVLC(pcVUI->getMinSpatialSegmentationIdc(),           "min_spatial_segmentation_idc");
330    WRITE_UVLC(pcVUI->getMaxBytesPerPicDenom(),                 "max_bytes_per_pic_denom");
331    WRITE_UVLC(pcVUI->getMaxBitsPerMinCuDenom(),                "max_bits_per_mincu_denom");
332    WRITE_UVLC(pcVUI->getLog2MaxMvLengthHorizontal(),           "log2_max_mv_length_horizontal");
333    WRITE_UVLC(pcVUI->getLog2MaxMvLengthVertical(),             "log2_max_mv_length_vertical");
334  }
335}
336
337Void TEncCavlc::codeHrdParameters( TComHRD *hrd, Bool commonInfPresentFlag, UInt maxNumSubLayersMinus1 )
338{
339  if( commonInfPresentFlag )
340  {
341    WRITE_FLAG( hrd->getNalHrdParametersPresentFlag() ? 1 : 0 ,  "nal_hrd_parameters_present_flag" );
342    WRITE_FLAG( hrd->getVclHrdParametersPresentFlag() ? 1 : 0 ,  "vcl_hrd_parameters_present_flag" );
343    if( hrd->getNalHrdParametersPresentFlag() || hrd->getVclHrdParametersPresentFlag() )
344    {
345      WRITE_FLAG( hrd->getSubPicCpbParamsPresentFlag() ? 1 : 0,  "sub_pic_cpb_params_present_flag" );
346      if( hrd->getSubPicCpbParamsPresentFlag() )
347      {
348        WRITE_CODE( hrd->getTickDivisorMinus2(), 8,              "tick_divisor_minus2" );
349        WRITE_CODE( hrd->getDuCpbRemovalDelayLengthMinus1(), 5,  "du_cpb_removal_delay_length_minus1" );
350        WRITE_FLAG( hrd->getSubPicCpbParamsInPicTimingSEIFlag() ? 1 : 0, "sub_pic_cpb_params_in_pic_timing_sei_flag" );
351        WRITE_CODE( hrd->getDpbOutputDelayDuLengthMinus1(), 5,   "dpb_output_delay_du_length_minus1"  );
352      }
353      WRITE_CODE( hrd->getBitRateScale(), 4,                     "bit_rate_scale" );
354      WRITE_CODE( hrd->getCpbSizeScale(), 4,                     "cpb_size_scale" );
355      if( hrd->getSubPicCpbParamsPresentFlag() )
356      {
357        WRITE_CODE( hrd->getDuCpbSizeScale(), 4,                "du_cpb_size_scale" ); 
358      }
359      WRITE_CODE( hrd->getInitialCpbRemovalDelayLengthMinus1(), 5, "initial_cpb_removal_delay_length_minus1" );
360      WRITE_CODE( hrd->getCpbRemovalDelayLengthMinus1(),        5, "au_cpb_removal_delay_length_minus1" );
361      WRITE_CODE( hrd->getDpbOutputDelayLengthMinus1(),         5, "dpb_output_delay_length_minus1" );
362    }
363  }
364  Int i, j, nalOrVcl;
365  for( i = 0; i <= maxNumSubLayersMinus1; i ++ )
366  {
367    WRITE_FLAG( hrd->getFixedPicRateFlag( i ) ? 1 : 0,          "fixed_pic_rate_general_flag");
368    if( !hrd->getFixedPicRateFlag( i ) )
369    {
370      WRITE_FLAG( hrd->getFixedPicRateWithinCvsFlag( i ) ? 1 : 0, "fixed_pic_rate_within_cvs_flag");
371    }
372    else
373    {
374      hrd->setFixedPicRateWithinCvsFlag( i, true );
375    }
376    if( hrd->getFixedPicRateWithinCvsFlag( i ) )
377    {
378      WRITE_UVLC( hrd->getPicDurationInTcMinus1( i ),           "elemental_duration_in_tc_minus1");
379    }
380    else
381    {
382      WRITE_FLAG( hrd->getLowDelayHrdFlag( i ) ? 1 : 0,           "low_delay_hrd_flag");
383    }
384    if (!hrd->getLowDelayHrdFlag( i ))
385    {
386      WRITE_UVLC( hrd->getCpbCntMinus1( i ),                      "cpb_cnt_minus1");
387    }
388   
389    for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
390    {
391      if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) ||
392          ( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
393      {
394        for( j = 0; j <= ( hrd->getCpbCntMinus1( i ) ); j ++ )
395        {
396          WRITE_UVLC( hrd->getBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_value_minus1");
397          WRITE_UVLC( hrd->getCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_value_minus1");
398          if( hrd->getSubPicCpbParamsPresentFlag() )
399          {
400            WRITE_UVLC( hrd->getDuCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_du_value_minus1"); 
401            WRITE_UVLC( hrd->getDuBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_du_value_minus1");
402          }
403          WRITE_FLAG( hrd->getCbrFlag( i, j, nalOrVcl ) ? 1 : 0, "cbr_flag");
404        }
405      }
406    }
407  }
408}
409
410#if H_3D
411Void TEncCavlc::codeSPS( TComSPS* pcSPS, Int viewIndex, Bool depthFlag )
412#else
413Void TEncCavlc::codeSPS( TComSPS* pcSPS )
414#endif
415{
416#if ENC_DEC_TRACE 
417  xTraceSPSHeader (pcSPS);
418#endif
419  WRITE_CODE( pcSPS->getVPSId (),          4,       "sps_video_parameter_set_id" );
420#if H_MV
421  if ( pcSPS->getLayerId() == 0 )
422  {
423#endif
424  WRITE_CODE( pcSPS->getMaxTLayers() - 1,  3,       "sps_max_sub_layers_minus1" );
425  WRITE_FLAG( pcSPS->getTemporalIdNestingFlag() ? 1 : 0,                             "sps_temporal_id_nesting_flag" );
426  codePTL(pcSPS->getPTL(), 1, pcSPS->getMaxTLayers() - 1);
427#if H_MV
428}
429#endif
430  WRITE_UVLC( pcSPS->getSPSId (),                   "sps_seq_parameter_set_id" );
431  WRITE_UVLC( pcSPS->getChromaFormatIdc (),         "chroma_format_idc" );
432  assert(pcSPS->getChromaFormatIdc () == 1);
433  // in the first version chroma_format_idc can only be equal to 1 (4:2:0)
434  if( pcSPS->getChromaFormatIdc () == 3 )
435  {
436    WRITE_FLAG( 0,                                  "separate_colour_plane_flag");
437  }
438
439  WRITE_UVLC( pcSPS->getPicWidthInLumaSamples (),   "pic_width_in_luma_samples" );
440  WRITE_UVLC( pcSPS->getPicHeightInLumaSamples(),   "pic_height_in_luma_samples" );
441  Window conf = pcSPS->getConformanceWindow();
442
443  WRITE_FLAG( conf.getWindowEnabledFlag(),          "conformance_window_flag" );
444  if (conf.getWindowEnabledFlag())
445  {
446    WRITE_UVLC( conf.getWindowLeftOffset()   / TComSPS::getWinUnitX(pcSPS->getChromaFormatIdc() ), "conf_win_left_offset" );
447    WRITE_UVLC( conf.getWindowRightOffset()  / TComSPS::getWinUnitX(pcSPS->getChromaFormatIdc() ), "conf_win_right_offset" );
448    WRITE_UVLC( conf.getWindowTopOffset()    / TComSPS::getWinUnitY(pcSPS->getChromaFormatIdc() ), "conf_win_top_offset" );
449    WRITE_UVLC( conf.getWindowBottomOffset() / TComSPS::getWinUnitY(pcSPS->getChromaFormatIdc() ), "conf_win_bottom_offset" );
450  }
451
452  WRITE_UVLC( pcSPS->getBitDepthY() - 8,             "bit_depth_luma_minus8" );
453  WRITE_UVLC( pcSPS->getBitDepthC() - 8,             "bit_depth_chroma_minus8" );
454
455  WRITE_UVLC( pcSPS->getBitsForPOC()-4,                 "log2_max_pic_order_cnt_lsb_minus4" );
456
457  const Bool subLayerOrderingInfoPresentFlag = 1;
458  WRITE_FLAG(subLayerOrderingInfoPresentFlag,       "sps_sub_layer_ordering_info_present_flag");
459  for(UInt i=0; i <= pcSPS->getMaxTLayers()-1; i++)
460  {
461    WRITE_UVLC( pcSPS->getMaxDecPicBuffering(i) - 1,       "sps_max_dec_pic_buffering_minus1[i]" );
462    WRITE_UVLC( pcSPS->getNumReorderPics(i),               "sps_num_reorder_pics[i]" );
463    WRITE_UVLC( pcSPS->getMaxLatencyIncrease(i),           "sps_max_latency_increase_plus1[i]" );
464    if (!subLayerOrderingInfoPresentFlag)
465    {
466      break;
467    }
468  }
469  assert( pcSPS->getMaxCUWidth() == pcSPS->getMaxCUHeight() );
470 
471  WRITE_UVLC( pcSPS->getLog2MinCodingBlockSize() - 3,                                "log2_min_coding_block_size_minus3" );
472  WRITE_UVLC( pcSPS->getLog2DiffMaxMinCodingBlockSize(),                             "log2_diff_max_min_coding_block_size" );
473  WRITE_UVLC( pcSPS->getQuadtreeTULog2MinSize() - 2,                                 "log2_min_transform_block_size_minus2" );
474  WRITE_UVLC( pcSPS->getQuadtreeTULog2MaxSize() - pcSPS->getQuadtreeTULog2MinSize(), "log2_diff_max_min_transform_block_size" );
475  WRITE_UVLC( pcSPS->getQuadtreeTUMaxDepthInter() - 1,                               "max_transform_hierarchy_depth_inter" );
476  WRITE_UVLC( pcSPS->getQuadtreeTUMaxDepthIntra() - 1,                               "max_transform_hierarchy_depth_intra" );
477  WRITE_FLAG( pcSPS->getScalingListFlag() ? 1 : 0,                                   "scaling_list_enabled_flag" ); 
478  if(pcSPS->getScalingListFlag())
479  {
480    WRITE_FLAG( pcSPS->getScalingListPresentFlag() ? 1 : 0,                          "sps_scaling_list_data_present_flag" ); 
481    if(pcSPS->getScalingListPresentFlag())
482    {
483#if SCALING_LIST_OUTPUT_RESULT
484    printf("SPS\n");
485#endif
486      codeScalingList( m_pcSlice->getScalingList() );
487    }
488  }
489  WRITE_FLAG( pcSPS->getUseAMP() ? 1 : 0,                                            "amp_enabled_flag" );
490  WRITE_FLAG( pcSPS->getUseSAO() ? 1 : 0,                                            "sample_adaptive_offset_enabled_flag");
491
492  WRITE_FLAG( pcSPS->getUsePCM() ? 1 : 0,                                            "pcm_enabled_flag");
493  if( pcSPS->getUsePCM() )
494  {
495    WRITE_CODE( pcSPS->getPCMBitDepthLuma() - 1, 4,                                  "pcm_sample_bit_depth_luma_minus1" );
496    WRITE_CODE( pcSPS->getPCMBitDepthChroma() - 1, 4,                                "pcm_sample_bit_depth_chroma_minus1" );
497    WRITE_UVLC( pcSPS->getPCMLog2MinSize() - 3,                                      "log2_min_pcm_luma_coding_block_size_minus3" );
498    WRITE_UVLC( pcSPS->getPCMLog2MaxSize() - pcSPS->getPCMLog2MinSize(),             "log2_diff_max_min_pcm_luma_coding_block_size" );
499    WRITE_FLAG( pcSPS->getPCMFilterDisableFlag()?1 : 0,                              "pcm_loop_filter_disable_flag");
500  }
501
502  assert( pcSPS->getMaxTLayers() > 0 );         
503
504  TComRPSList* rpsList = pcSPS->getRPSList();
505  TComReferencePictureSet*      rps;
506
507  WRITE_UVLC(rpsList->getNumberOfReferencePictureSets(), "num_short_term_ref_pic_sets" );
508  for(Int i=0; i < rpsList->getNumberOfReferencePictureSets(); i++)
509  {
510    rps = rpsList->getReferencePictureSet(i);
511    codeShortTermRefPicSet(pcSPS,rps,false, i);
512  }
513  WRITE_FLAG( pcSPS->getLongTermRefsPresent() ? 1 : 0,         "long_term_ref_pics_present_flag" );
514  if (pcSPS->getLongTermRefsPresent()) 
515  {
516    WRITE_UVLC(pcSPS->getNumLongTermRefPicSPS(), "num_long_term_ref_pic_sps" );
517    for (UInt k = 0; k < pcSPS->getNumLongTermRefPicSPS(); k++)
518    {
519      WRITE_CODE( pcSPS->getLtRefPicPocLsbSps(k), pcSPS->getBitsForPOC(), "lt_ref_pic_poc_lsb_sps");
520      WRITE_FLAG( pcSPS->getUsedByCurrPicLtSPSFlag(k), "used_by_curr_pic_lt_sps_flag");
521    }
522  }
523  WRITE_FLAG( pcSPS->getTMVPFlagsPresent()  ? 1 : 0,           "sps_temporal_mvp_enable_flag" );
524
525  WRITE_FLAG( pcSPS->getUseStrongIntraSmoothing(),             "sps_strong_intra_smoothing_enable_flag" );
526
527  WRITE_FLAG( pcSPS->getVuiParametersPresentFlag(),             "vui_parameters_present_flag" );
528  if (pcSPS->getVuiParametersPresentFlag())
529  {
530      codeVUI(pcSPS->getVuiParameters(), pcSPS);
531  }
532
533#if H_MV
534  WRITE_FLAG( 1, "sps_extension_flag" );
535  WRITE_FLAG( pcSPS->getInterViewMvVertConstraintFlag() ? 1 : 0, "inter_view_mv_vert_constraint_flag" );
536  ////   sps_extension_vui_parameters( )
537  if( pcSPS->getVuiParameters()->getBitstreamRestrictionFlag() )
538  { 
539    WRITE_UVLC( pcSPS->getNumIlpRestrictedRefLayers( ),           "num_ilp_restricted_ref_layers" ); 
540    for( Int i = 0; i < pcSPS->getNumIlpRestrictedRefLayers( ); i++ ) 
541    { 
542      WRITE_UVLC( pcSPS->getMinSpatialSegmentOffsetPlus1( i ),    "min_spatial_segment_offset_plus1" ); 
543      if( pcSPS->getMinSpatialSegmentOffsetPlus1( i ) > 0 ) 
544      { 
545        WRITE_FLAG( pcSPS->getCtuBasedOffsetEnabledFlag( i ),      "ctu_based_offset_enabled_flag[ i ]"); 
546        if( pcSPS->getCtuBasedOffsetEnabledFlag( i ) ) 
547        {
548          WRITE_UVLC( pcSPS->getMinHorizontalCtuOffsetPlus1( i ), "min_horizontal_ctu_offset_plus1[ i ]"); 
549        }
550      } 
551    } 
552  } 
553#if H_3D_QTLPC
554  if( depthFlag )
555  {
556    WRITE_FLAG( pcSPS->getUseQTL() ? 1 : 0, "use_qtl_flag");
557    WRITE_FLAG( pcSPS->getUsePC()  ? 1 : 0, "use_pc_flag");
558  }
559#endif
560  ////   sps_extension_vui_parameters( ) END
561  WRITE_UVLC( 0, "sps_shvc_reserved_zero_idc" ); 
562#if !H_3D
563  WRITE_FLAG( 0, "sps_extension2_flag" );
564#else
565  WRITE_FLAG( 1, "sps_extension2_flag"  );
566  if (!depthFlag )
567  {
568    WRITE_UVLC( pcSPS->getCamParPrecision(), "cp_precision" );
569    WRITE_FLAG( pcSPS->hasCamParInSliceHeader() ? 1 : 0, "cp_in_slice_header_flag" );
570    if( !pcSPS->hasCamParInSliceHeader() )
571    {
572      for( UInt uiIndex = 0; uiIndex < viewIndex; uiIndex++ )
573      {
574        WRITE_SVLC( pcSPS->getCodedScale    ()[ uiIndex ],                                      "cp_scale" );
575        WRITE_SVLC( pcSPS->getCodedOffset   ()[ uiIndex ],                                      "cp_off" );
576        WRITE_SVLC( pcSPS->getInvCodedScale ()[ uiIndex ] + pcSPS->getCodedScale ()[ uiIndex ], "cp_inv_scale_plus_scale" );
577        WRITE_SVLC( pcSPS->getInvCodedOffset()[ uiIndex ] + pcSPS->getCodedOffset()[ uiIndex ], "cp_inv_off_plus_off" );
578      }
579    }
580  }
581
582  WRITE_FLAG( 0, "sps_extension3_flag" );
583#endif
584#else
585  WRITE_FLAG( 0, "sps_extension_flag" );
586#endif
587}
588
589Void TEncCavlc::codeVPS( TComVPS* pcVPS )
590{
591  WRITE_CODE( pcVPS->getVPSId(),                    4,        "vps_video_parameter_set_id" );
592  WRITE_CODE( 3,                                    2,        "vps_reserved_three_2bits" );
593#if H_MV
594  WRITE_CODE( pcVPS->getMaxLayers() - 1,            6,        "vps_max_layers_minus1" );
595#else
596  WRITE_CODE( 0,                                    6,        "vps_reserved_zero_6bits" );
597#endif
598  WRITE_CODE( pcVPS->getMaxTLayers() - 1,           3,        "vps_max_sub_layers_minus1" );
599  WRITE_FLAG( pcVPS->getTemporalNestingFlag(),                "vps_temporal_id_nesting_flag" );
600  assert (pcVPS->getMaxTLayers()>1||pcVPS->getTemporalNestingFlag());
601#if H_MV
602  WRITE_CODE( 0xffff,                              16,        "vps_extension_offset" );
603#else
604  WRITE_CODE( 0xffff,                              16,        "vps_reserved_ffff_16bits" );
605#endif
606  codePTL( pcVPS->getPTL(), true, pcVPS->getMaxTLayers() - 1 );
607  const Bool subLayerOrderingInfoPresentFlag = 1;
608  WRITE_FLAG(subLayerOrderingInfoPresentFlag,              "vps_sub_layer_ordering_info_present_flag");
609  for(UInt i=0; i <= pcVPS->getMaxTLayers()-1; i++)
610  {
611    WRITE_UVLC( pcVPS->getMaxDecPicBuffering(i) - 1,       "vps_max_dec_pic_buffering_minus1[i]" );
612    WRITE_UVLC( pcVPS->getNumReorderPics(i),               "vps_num_reorder_pics[i]" );
613    WRITE_UVLC( pcVPS->getMaxLatencyIncrease(i),           "vps_max_latency_increase_plus1[i]" );
614    if (!subLayerOrderingInfoPresentFlag)
615    {
616      break;
617    }
618  }
619
620  assert( pcVPS->getNumHrdParameters() <= MAX_VPS_NUM_HRD_PARAMETERS );
621#if H_MV
622  assert( pcVPS->getVpsMaxLayerId() < MAX_VPS_NUH_LAYER_ID_PLUS1 );
623  WRITE_CODE( pcVPS->getVpsMaxLayerId(), 6,                 "vps_max_layer_id" ); 
624 
625  WRITE_UVLC( pcVPS->getVpsNumLayerSetsMinus1(),  "vps_max_num_layer_sets_minus1" );
626  for( UInt opsIdx = 1; opsIdx <= pcVPS->getVpsNumLayerSetsMinus1(); opsIdx ++ )
627  {
628    // Operation point set
629    for( UInt i = 0; i <= pcVPS->getVpsMaxLayerId(); i ++ )
630    {
631#else
632  assert( pcVPS->getMaxNuhReservedZeroLayerId() < MAX_VPS_NUH_RESERVED_ZERO_LAYER_ID_PLUS1 );
633  WRITE_CODE( pcVPS->getMaxNuhReservedZeroLayerId(), 6,     "vps_max_nuh_reserved_zero_layer_id" );
634  pcVPS->setMaxOpSets(1);
635  WRITE_UVLC( pcVPS->getMaxOpSets() - 1,                    "vps_max_op_sets_minus1" );
636  for( UInt opsIdx = 1; opsIdx <= ( pcVPS->getMaxOpSets() - 1 ); opsIdx ++ )
637  {
638    // Operation point set
639    for( UInt i = 0; i <= pcVPS->getMaxNuhReservedZeroLayerId(); i ++ )
640    {
641      // Only applicable for version 1
642      pcVPS->setLayerIdIncludedFlag( true, opsIdx, i );
643#endif
644      WRITE_FLAG( pcVPS->getLayerIdIncludedFlag( opsIdx, i ) ? 1 : 0, "layer_id_included_flag[opsIdx][i]" );
645    }
646  }
647  TimingInfo *timingInfo = pcVPS->getTimingInfo();
648  WRITE_FLAG(timingInfo->getTimingInfoPresentFlag(),          "vps_timing_info_present_flag");
649  if(timingInfo->getTimingInfoPresentFlag())
650  {
651    WRITE_CODE(timingInfo->getNumUnitsInTick(), 32,           "vps_num_units_in_tick");
652    WRITE_CODE(timingInfo->getTimeScale(),      32,           "vps_time_scale");
653    WRITE_FLAG(timingInfo->getPocProportionalToTimingFlag(),  "vps_poc_proportional_to_timing_flag");
654    if(timingInfo->getPocProportionalToTimingFlag())
655    {
656      WRITE_UVLC(timingInfo->getNumTicksPocDiffOneMinus1(),   "vps_num_ticks_poc_diff_one_minus1");
657    }
658    pcVPS->setNumHrdParameters( 0 );
659    WRITE_UVLC( pcVPS->getNumHrdParameters(),                 "vps_num_hrd_parameters" );
660
661    if( pcVPS->getNumHrdParameters() > 0 )
662    {
663      pcVPS->createHrdParamBuffer();
664    }
665    for( UInt i = 0; i < pcVPS->getNumHrdParameters(); i ++ )
666    {
667      // Only applicable for version 1
668      pcVPS->setHrdOpSetIdx( 0, i );
669      WRITE_UVLC( pcVPS->getHrdOpSetIdx( i ),                "hrd_op_set_idx" );
670      if( i > 0 )
671      {
672        WRITE_FLAG( pcVPS->getCprmsPresentFlag( i ) ? 1 : 0, "cprms_present_flag[i]" );
673      }
674      codeHrdParameters(pcVPS->getHrdParameters(i), pcVPS->getCprmsPresentFlag( i ), pcVPS->getMaxTLayers() - 1);
675    }
676  }
677#if H_MV
678  WRITE_FLAG( 1,                                             "vps_extension_flag" );
679
680  m_pcBitIf->writeAlignOne();                       
681
682  WRITE_FLAG( pcVPS->getAvcBaseLayerFlag() ? 1 : 0,          "avc_base_layer_flag" );
683  WRITE_FLAG( pcVPS->getSplittingFlag() ? 1 : 0,             "splitting_flag" );
684 
685  for( Int type = 0; type < MAX_NUM_SCALABILITY_TYPES; type++ )
686  {
687    WRITE_FLAG( pcVPS->getScalabilityMask( type ) ? 1 : 0,   "scalability_mask[i]" );
688  }
689
690  for( Int sIdx = 0; sIdx < pcVPS->getNumScalabilityTypes( ) - ( pcVPS->getSplittingFlag() ? 1 : 0 ); sIdx++ )
691  {
692    WRITE_CODE( pcVPS->getDimensionIdLen( sIdx ) - 1 , 3,    "dimension_id_len_minus1[j]");   
693  }
694
695  if ( pcVPS->getSplittingFlag() )
696  { // Ignore old dimension id length
697    pcVPS->setDimensionIdLen( pcVPS->getNumScalabilityTypes( ) - 1 ,pcVPS->inferLastDimsionIdLenMinus1() + 1 );       
698  }   
699
700
701  WRITE_FLAG( pcVPS->getVpsNuhLayerIdPresentFlag() ? 1 : 0,  "vps_nuh_layer_id_present_flag");
702
703  for( Int i = 0; i <= pcVPS->getMaxLayers() - 1; i++ )
704  {
705    if ( pcVPS->getVpsNuhLayerIdPresentFlag() && ( i > 0 ) )
706    {     
707      WRITE_CODE( pcVPS->getLayerIdInNuh( i ), 6,          "layer_id_in_nuh[i]");
708  }
709    else
710    {
711      assert( pcVPS->getLayerIdInNuh( i ) == i ); 
712  }
713
714    assert(  pcVPS->getLayerIdInVps( pcVPS->getLayerIdInNuh( i ) ) == i ); 
715
716    for( Int j = 0; j < pcVPS->getNumScalabilityTypes() ; j++ )
717  {
718      if ( !pcVPS->getSplittingFlag() )
719    {
720        WRITE_CODE( pcVPS->getDimensionId( i, j ), pcVPS->getDimensionIdLen( j ), "dimension_id[i][j]");     
721      }
722      else
723      {
724        assert( pcVPS->getDimensionId( i, j ) ==  pcVPS->inferDimensionId( i, j )  );
725      }
726    }
727  }
728
729  for( Int i = 1; i <= pcVPS->getMaxLayers() - 1; i++ )
730  {
731    for( Int j = 0; j < i; j++ )
732    {
733      WRITE_FLAG( pcVPS->getDirectDependencyFlag( i, j ),    "direct_dependency_flag[i][j]" );
734    }
735  }
736
737  for( Int i = 0; i < pcVPS->getMaxLayers() - 1; i++ )
738  {
739    WRITE_CODE( pcVPS->getMaxTidIlRefPicPlus1( i ), 3,       "max_tid_il_ref_pics_plus1[i]" );
740  }
741
742  WRITE_CODE( pcVPS->getVpsNumberLayerSetsMinus1( )    , 10,    "vps_number_layer_sets_minus1"      );
743  WRITE_CODE( pcVPS->getVpsNumProfileTierLevelMinus1( ), 6,     "vps_num_profile_tier_level_minus1" );
744
745  for( Int i = 1; i <= pcVPS->getVpsNumProfileTierLevelMinus1(); i++ )
746  {
747    WRITE_FLAG( pcVPS->getVpsProfilePresentFlag( i ) ? 1 : 0, "vps_profile_present_flag[i]" );
748    if( !pcVPS->getVpsProfilePresentFlag( i ) )
749    {   
750      WRITE_CODE( pcVPS->getProfileRefMinus1( i ), 6, "profile_ref_minus1[i]" );
751    }
752    codePTL( pcVPS->getPTL( i ), pcVPS->getVpsProfilePresentFlag( i ), pcVPS->getMaxTLayers() - 1 );
753  }
754
755  Int numOutputLayerSets = pcVPS->getVpsNumberLayerSetsMinus1( ) + 1; 
756
757  WRITE_FLAG( pcVPS->getMoreOutputLayerSetsThanDefaultFlag( ) ? 1 : 0, "more_output_layer_sets_than_default_flag" );
758
759  if ( pcVPS->getMoreOutputLayerSetsThanDefaultFlag( ) )
760  {
761    WRITE_CODE( pcVPS->getNumAddOutputLayerSetsMinus1( )    , 10,    "num_add_output_layer_sets_minus1"      );
762    numOutputLayerSets += ( pcVPS->getNumAddOutputLayerSetsMinus1( ) + 1 ); 
763  }
764
765  if( numOutputLayerSets > 1)
766  {
767    WRITE_FLAG( pcVPS->getDefaultOneTargetOutputLayerFlag( ) ? 1 : 0, "default_one_target_output_layer_flag" );
768  } 
769
770  for( Int i = 1; i < numOutputLayerSets; i++ )
771  {
772    if( i > pcVPS->getVpsNumberLayerSetsMinus1( ) )
773    {     
774      WRITE_UVLC( pcVPS->getOutputLayerSetIdxMinus1( i ),      "output_layer_set_idx_minus1[i]" );
775      for( Int j = 0; j < pcVPS->getNumLayersInIdList( j ) - 1 ; j++ )
776      {
777        WRITE_FLAG( pcVPS->getOutputLayerFlag( i, j) ? 1 : 0, "output_layer_flag" );
778      }     
779    }
780    if ( pcVPS->getProfileLevelTierIdxLen()  > 0 )
781    {     
782      WRITE_CODE( pcVPS->getProfileLevelTierIdx( i ), pcVPS->getProfileLevelTierIdxLen() ,"profile_level_tier_idx[ i ]" );   
783    }
784  }
785
786  WRITE_FLAG( pcVPS->getMaxOneActiveRefLayerFlag( ) ? 1 : 0, "max_one_active_ref_layer_flag" );
787  WRITE_UVLC( pcVPS->getDirectDepTypeLenMinus2 ( ),         "direct_dep_type_len_minus2"); 
788
789    for( Int i = 1; i <= pcVPS->getMaxLayers() - 1; i++ )
790    {
791      for( Int j = 0; j < i; j++ )
792      {
793        if (pcVPS->getDirectDependencyFlag( i, j) )
794        {       
795          assert ( pcVPS->getDirectDependencyType( i, j ) != -1 ); 
796          WRITE_CODE( pcVPS->getDirectDependencyType( i, j ),pcVPS->getDirectDepTypeLenMinus2( ) + 2,  "direct_dependency_type[i][j]" );
797        }
798      }
799    }
800
801    WRITE_FLAG ( 0,                    "vps_shvc_reserved_zero_flag" ); 
802
803#if H_3D
804  WRITE_FLAG( 1,                                             "vps_extension2_flag" );
805  m_pcBitIf->writeAlignOne();     
806  for( Int i = 0; i <= pcVPS->getMaxLayers() - 1; i++ )
807  {
808    if (i!= 0)
809    {
810      if ( !( pcVPS->getDepthId( i ) == 1 ) )
811      {
812#if H_3D_IV_MERGE
813        WRITE_FLAG( pcVPS->getIvMvPredFlag         ( i ) ? 1 : 0 , "iv_mv_pred_flag[i]");
814#endif
815#if H_3D_ARP
816        WRITE_FLAG( pcVPS->getUseAdvRP             ( i ) ? 1 : 0,  "iv_res_pred_flag[i]"  );
817#endif
818#if H_3D_NBDV_REF
819        WRITE_FLAG( pcVPS->getDepthRefinementFlag  ( i ) ? 1 : 0 , "depth_refinement_flag[i]");
820#endif
821#if H_3D_VSP
822        WRITE_FLAG( pcVPS->getViewSynthesisPredFlag( i ) ? 1 : 0 , "view_synthesis_pred_flag[i]");
823#endif
824      }         
825      else
826      {
827        WRITE_FLAG( pcVPS->getVpsDepthModesFlag( i ) ? 1 : 0 ,          "vps_depth_modes_flag[i]" );
828        //WRITE_FLAG( pcVPS->getLimQtPredFlag    ( i ) ? 1 : 0 ,          "lim_qt_pred_flag[i]"     );
829#if H_3D_DIM_DLT
830        if( pcVPS->getVpsDepthModesFlag( i ) )
831        {
832          WRITE_FLAG( pcVPS->getUseDLTFlag( i ) ? 1 : 0, "dlt_flag[i]" );
833        }
834        if( pcVPS->getUseDLTFlag( i ) )
835        {
836          // code mapping
837          WRITE_UVLC(pcVPS->getNumDepthValues(i), "num_depth_values_in_dlt[i]");
838          for(Int d=0; d<pcVPS->getNumDepthValues(i); d++)
839          {
840            WRITE_UVLC( pcVPS->idx2DepthValue(i, d), "dlt_depth_value[i][d]" );
841          }
842        }       
843#endif
844#if LGE_INTER_SDC_E0156
845        WRITE_FLAG( pcVPS->getInterSDCFlag( i ) ? 1 : 0, "depth_inter_SDC_flag" );
846#endif
847      }
848    } 
849  }
850#if H_3D_TMVP
851  WRITE_FLAG( pcVPS->getIvMvScalingFlag( ) ? 1 : 0 ,          "iv_mv_scaling_flag" );
852#endif
853#else
854  WRITE_FLAG( 0,                                             "vps_extension2_flag" );
855#endif
856#else
857  WRITE_FLAG( 0,                     "vps_extension_flag" );
858#endif
859 
860  //future extensions here..
861 
862  return;
863}
864
865Void TEncCavlc::codeSliceHeader         ( TComSlice* pcSlice )
866{
867#if H_MV
868  TComVPS* vps = pcSlice->getVPS(); 
869#endif
870#if ENC_DEC_TRACE 
871  xTraceSliceHeader (pcSlice);
872#endif
873
874  //calculate number of bits required for slice address
875  Int maxSliceSegmentAddress = pcSlice->getPic()->getNumCUsInFrame();
876  Int bitsSliceSegmentAddress = 0;
877  while(maxSliceSegmentAddress>(1<<bitsSliceSegmentAddress)) 
878  {
879    bitsSliceSegmentAddress++;
880  }
881  Int ctuAddress;
882  if (pcSlice->isNextSlice())
883  {
884    // Calculate slice address
885    ctuAddress = (pcSlice->getSliceCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU());
886  }
887  else
888  {
889    // Calculate slice address
890    ctuAddress = (pcSlice->getSliceSegmentCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU());
891  }
892  //write slice address
893  Int sliceSegmentAddress = pcSlice->getPic()->getPicSym()->getCUOrderMap(ctuAddress);
894
895  WRITE_FLAG( sliceSegmentAddress==0, "first_slice_segment_in_pic_flag" );
896  if ( pcSlice->getRapPicFlag() )
897  {
898    WRITE_FLAG( 0, "no_output_of_prior_pics_flag" );
899  }
900  WRITE_UVLC( pcSlice->getPPS()->getPPSId(), "slice_pic_parameter_set_id" );
901  pcSlice->setDependentSliceSegmentFlag(!pcSlice->isNextSlice());
902  if ( pcSlice->getPPS()->getDependentSliceSegmentsEnabledFlag() && (sliceSegmentAddress!=0) )
903  {
904    WRITE_FLAG( pcSlice->getDependentSliceSegmentFlag() ? 1 : 0, "dependent_slice_segment_flag" );
905  }
906  if(sliceSegmentAddress>0)
907  {
908    WRITE_CODE( sliceSegmentAddress, bitsSliceSegmentAddress, "slice_segment_address" );
909  }
910  if ( !pcSlice->getDependentSliceSegmentFlag() )
911  {
912#if H_MV   
913    if ( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > 0 )
914    {
915      WRITE_FLAG( pcSlice->getDiscardableFlag( ) ? 1 : 0 , "discardable_flag" );
916    }
917
918    for (Int i = 1; i < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); i++)   
919#else
920    for (Int i = 0; i < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); i++)
921#endif
922    {
923      assert(!!"slice_reserved_undetermined_flag[]");
924      WRITE_FLAG(0, "slice_reserved_undetermined_flag[]");
925    }
926
927    WRITE_UVLC( pcSlice->getSliceType(),       "slice_type" );
928
929    if( pcSlice->getPPS()->getOutputFlagPresentFlag() )
930    {
931      WRITE_FLAG( pcSlice->getPicOutputFlag() ? 1 : 0, "pic_output_flag" );
932    }
933
934    // in the first version chroma_format_idc is equal to one, thus colour_plane_id will not be present
935    assert (pcSlice->getSPS()->getChromaFormatIdc() == 1 );
936    // if( separate_colour_plane_flag  ==  1 )
937    //   colour_plane_id                                      u(2)
938
939    if( !pcSlice->getIdrPicFlag() )
940    {
941      Int picOrderCntLSB = (pcSlice->getPOC()-pcSlice->getLastIDR()+(1<<pcSlice->getSPS()->getBitsForPOC()))%(1<<pcSlice->getSPS()->getBitsForPOC());
942      WRITE_CODE( picOrderCntLSB, pcSlice->getSPS()->getBitsForPOC(), "pic_order_cnt_lsb");
943      TComReferencePictureSet* rps = pcSlice->getRPS();
944     
945#if FIX1071
946      // check for bitstream restriction stating that:
947      // If the current picture is a BLA or CRA picture, the value of NumPocTotalCurr shall be equal to 0.
948      // Ideally this process should not be repeated for each slice in a picture
949      if (pcSlice->isIRAP())
950      {
951          for (Int picIdx = 0; picIdx < rps->getNumberOfPictures(); picIdx++)
952          {
953          assert (!rps->getUsed(picIdx));
954          }
955        }
956#endif
957
958      if(pcSlice->getRPSidx() < 0)
959      {
960        WRITE_FLAG( 0, "short_term_ref_pic_set_sps_flag");
961        codeShortTermRefPicSet(pcSlice->getSPS(), rps, true, pcSlice->getSPS()->getRPSList()->getNumberOfReferencePictureSets());
962      }
963      else
964      {
965        WRITE_FLAG( 1, "short_term_ref_pic_set_sps_flag");
966        Int numBits = 0;
967        while ((1 << numBits) < pcSlice->getSPS()->getRPSList()->getNumberOfReferencePictureSets())
968        {
969          numBits++;
970        }
971        if (numBits > 0)
972        {
973          WRITE_CODE( pcSlice->getRPSidx(), numBits, "short_term_ref_pic_set_idx" );         
974        }
975      }
976      if(pcSlice->getSPS()->getLongTermRefsPresent())
977      {
978        Int numLtrpInSH = rps->getNumberOfLongtermPictures();
979        Int ltrpInSPS[MAX_NUM_REF_PICS];
980        Int numLtrpInSPS = 0;
981        UInt ltrpIndex;
982        Int counter = 0;
983        for(Int k = rps->getNumberOfPictures()-1; k > rps->getNumberOfPictures()-rps->getNumberOfLongtermPictures()-1; k--) 
984        {
985          if (findMatchingLTRP(pcSlice, &ltrpIndex, rps->getPOC(k), rps->getUsed(k))) 
986          {
987            ltrpInSPS[numLtrpInSPS] = ltrpIndex;
988            numLtrpInSPS++;
989          }
990          else
991          {
992            counter++;
993          }
994        }
995        numLtrpInSH -= numLtrpInSPS;
996
997        Int bitsForLtrpInSPS = 0;
998        while (pcSlice->getSPS()->getNumLongTermRefPicSPS() > (1 << bitsForLtrpInSPS))
999        {
1000          bitsForLtrpInSPS++;
1001        }
1002        if (pcSlice->getSPS()->getNumLongTermRefPicSPS() > 0) 
1003        {
1004          WRITE_UVLC( numLtrpInSPS, "num_long_term_sps");
1005        }
1006        WRITE_UVLC( numLtrpInSH, "num_long_term_pics");
1007        // Note that the LSBs of the LT ref. pic. POCs must be sorted before.
1008        // Not sorted here because LT ref indices will be used in setRefPicList()
1009        Int prevDeltaMSB = 0, prevLSB = 0;
1010        Int offset = rps->getNumberOfNegativePictures() + rps->getNumberOfPositivePictures();
1011        for(Int i=rps->getNumberOfPictures()-1 ; i > offset-1; i--)
1012        {
1013          if (counter < numLtrpInSPS)
1014          {
1015            if (bitsForLtrpInSPS > 0)
1016            {
1017              WRITE_CODE( ltrpInSPS[counter], bitsForLtrpInSPS, "lt_idx_sps[i]");             
1018            }
1019          }
1020          else 
1021          {
1022            WRITE_CODE( rps->getPocLSBLT(i), pcSlice->getSPS()->getBitsForPOC(), "poc_lsb_lt");
1023            WRITE_FLAG( rps->getUsed(i), "used_by_curr_pic_lt_flag"); 
1024          }
1025          WRITE_FLAG( rps->getDeltaPocMSBPresentFlag(i), "delta_poc_msb_present_flag");
1026
1027          if(rps->getDeltaPocMSBPresentFlag(i))
1028          {
1029            Bool deltaFlag = false;
1030            //  First LTRP from SPS                 ||  First LTRP from SH                              || curr LSB            != prev LSB
1031            if( (i == rps->getNumberOfPictures()-1) || (i == rps->getNumberOfPictures()-1-numLtrpInSPS) || (rps->getPocLSBLT(i) != prevLSB) )
1032            {
1033              deltaFlag = true;
1034            }
1035            if(deltaFlag)
1036            {
1037              WRITE_UVLC( rps->getDeltaPocMSBCycleLT(i), "delta_poc_msb_cycle_lt[i]" );
1038            }
1039            else
1040            {             
1041              Int differenceInDeltaMSB = rps->getDeltaPocMSBCycleLT(i) - prevDeltaMSB;
1042              assert(differenceInDeltaMSB >= 0);
1043              WRITE_UVLC( differenceInDeltaMSB, "delta_poc_msb_cycle_lt[i]" );
1044            }
1045            prevLSB = rps->getPocLSBLT(i);
1046            prevDeltaMSB = rps->getDeltaPocMSBCycleLT(i);
1047          }
1048        }
1049      }
1050      if (pcSlice->getSPS()->getTMVPFlagsPresent())
1051      {
1052        WRITE_FLAG( pcSlice->getEnableTMVPFlag() ? 1 : 0, "slice_temporal_mvp_enable_flag" );
1053      }
1054    }
1055#if H_MV
1056    Int layerIdInVps       = pcSlice->getLayerIdInVps(); 
1057    if( pcSlice->getLayerId() > 0 && vps->getNumDirectRefLayers( layerIdInVps ) > 0 )
1058    {   
1059      WRITE_FLAG( pcSlice->getInterLayerPredEnabledFlag( ) ? 1 : 0 , "inter_layer_pred_enabled_flag" );
1060      if( pcSlice->getInterLayerPredEnabledFlag() && vps->getNumDirectRefLayers( layerIdInVps ) > 1 )
1061      {           
1062        if( !vps->getMaxOneActiveRefLayerFlag()) 
1063        {
1064          WRITE_CODE( pcSlice->getNumInterLayerRefPicsMinus1( ), pcSlice->getNumInterLayerRefPicsMinus1Len( ), "num_inter_layer_ref_pics_minus1" );
1065        }
1066        for( Int i = 0; i < pcSlice->getNumActiveRefLayerPics(); i++ )   
1067        {
1068          WRITE_CODE( pcSlice->getInterLayerPredLayerIdc( i ), pcSlice->getInterLayerPredLayerIdcLen( ), "inter_layer_pred_layer_idc" );
1069        }
1070      } 
1071    }
1072
1073    if( vps->getNumSamplePredRefLayers( layerIdInVps ) > 0  &&  pcSlice->getNumActiveRefLayerPics() > 0 )
1074    {
1075      WRITE_FLAG( pcSlice->getInterLayerSamplePredOnlyFlag( ) ? 1 : 0 , "inter_layer_sample_pred_only_flag" );
1076    }
1077
1078#endif
1079    if(pcSlice->getSPS()->getUseSAO())
1080    {
1081      if (pcSlice->getSPS()->getUseSAO())
1082      {
1083         WRITE_FLAG( pcSlice->getSaoEnabledFlag(), "slice_sao_luma_flag" );
1084         {
1085           SAOParam *saoParam = pcSlice->getPic()->getPicSym()->getSaoParam();
1086          WRITE_FLAG( saoParam->bSaoFlag[1], "slice_sao_chroma_flag" );
1087         }
1088      }
1089    }
1090
1091    //check if numrefidxes match the defaults. If not, override
1092
1093    if (!pcSlice->isIntra())
1094    {
1095      Bool overrideFlag = (pcSlice->getNumRefIdx( REF_PIC_LIST_0 )!=pcSlice->getPPS()->getNumRefIdxL0DefaultActive()||(pcSlice->isInterB()&&pcSlice->getNumRefIdx( REF_PIC_LIST_1 )!=pcSlice->getPPS()->getNumRefIdxL1DefaultActive()));
1096      WRITE_FLAG( overrideFlag ? 1 : 0,                               "num_ref_idx_active_override_flag");
1097      if (overrideFlag) 
1098      {
1099        WRITE_UVLC( pcSlice->getNumRefIdx( REF_PIC_LIST_0 ) - 1,      "num_ref_idx_l0_active_minus1" );
1100        if (pcSlice->isInterB())
1101        {
1102          WRITE_UVLC( pcSlice->getNumRefIdx( REF_PIC_LIST_1 ) - 1,    "num_ref_idx_l1_active_minus1" );
1103        }
1104        else
1105        {
1106          pcSlice->setNumRefIdx(REF_PIC_LIST_1, 0);
1107        }
1108      }
1109    }
1110    else
1111    {
1112      pcSlice->setNumRefIdx(REF_PIC_LIST_0, 0);
1113      pcSlice->setNumRefIdx(REF_PIC_LIST_1, 0);
1114    }
1115
1116    if( pcSlice->getPPS()->getListsModificationPresentFlag() && pcSlice->getNumRpsCurrTempList() > 1)
1117    {
1118      TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
1119      if(!pcSlice->isIntra())
1120      {
1121        WRITE_FLAG(pcSlice->getRefPicListModification()->getRefPicListModificationFlagL0() ? 1 : 0,       "ref_pic_list_modification_flag_l0" );
1122        if (pcSlice->getRefPicListModification()->getRefPicListModificationFlagL0())
1123        {
1124          Int numRpsCurrTempList0 = pcSlice->getNumRpsCurrTempList();
1125          if (numRpsCurrTempList0 > 1)
1126          {
1127            Int length = 1;
1128            numRpsCurrTempList0 --;
1129            while ( numRpsCurrTempList0 >>= 1) 
1130            {
1131              length ++;
1132            }
1133            for(Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_0 ); i++)
1134            {
1135              WRITE_CODE( refPicListModification->getRefPicSetIdxL0(i), length, "list_entry_l0");
1136            }
1137          }
1138        }
1139      }
1140      if(pcSlice->isInterB())
1141      {   
1142        WRITE_FLAG(pcSlice->getRefPicListModification()->getRefPicListModificationFlagL1() ? 1 : 0,       "ref_pic_list_modification_flag_l1" );
1143        if (pcSlice->getRefPicListModification()->getRefPicListModificationFlagL1())
1144        {
1145          Int numRpsCurrTempList1 = pcSlice->getNumRpsCurrTempList();
1146          if ( numRpsCurrTempList1 > 1 )
1147          {
1148            Int length = 1;
1149            numRpsCurrTempList1 --;
1150            while ( numRpsCurrTempList1 >>= 1)
1151            {
1152              length ++;
1153            }
1154            for(Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_1 ); i++)
1155            {
1156              WRITE_CODE( refPicListModification->getRefPicSetIdxL1(i), length, "list_entry_l1");
1157            }
1158          }
1159        }
1160      }
1161    }
1162   
1163    if (pcSlice->isInterB())
1164    {
1165      WRITE_FLAG( pcSlice->getMvdL1ZeroFlag() ? 1 : 0,   "mvd_l1_zero_flag");
1166    }
1167
1168    if(!pcSlice->isIntra())
1169    {
1170      if (!pcSlice->isIntra() && pcSlice->getPPS()->getCabacInitPresentFlag())
1171      {
1172        SliceType sliceType   = pcSlice->getSliceType();
1173        Int  encCABACTableIdx = pcSlice->getPPS()->getEncCABACTableIdx();
1174        Bool encCabacInitFlag = (sliceType!=encCABACTableIdx && encCABACTableIdx!=I_SLICE) ? true : false;
1175        pcSlice->setCabacInitFlag( encCabacInitFlag );
1176        WRITE_FLAG( encCabacInitFlag?1:0, "cabac_init_flag" );
1177      }
1178    }
1179
1180    if ( pcSlice->getEnableTMVPFlag() )
1181    {
1182#if H_MV
1183      if( pcSlice->getLayerId() > 0 && pcSlice->getNumActiveMotionPredRefLayers() > 0 )
1184      {
1185        WRITE_FLAG( pcSlice->getAltCollocatedIndicationFlag( ) ? 1 : 0 , "alt_collocated_indication_flag" );
1186      }
1187      if( pcSlice->getAltCollocatedIndicationFlag() && pcSlice->getNumActiveMotionPredRefLayers() > 1 ) 
1188      {         
1189        WRITE_UVLC( pcSlice->getCollocatedRefLayerIdx( ), "collocated_ref_layer_idx" );
1190      }     
1191      else 
1192      {
1193#endif
1194      if ( pcSlice->getSliceType() == B_SLICE )
1195      {
1196        WRITE_FLAG( pcSlice->getColFromL0Flag(), "collocated_from_l0_flag" );
1197      }
1198
1199      if ( pcSlice->getSliceType() != I_SLICE &&
1200        ((pcSlice->getColFromL0Flag()==1 && pcSlice->getNumRefIdx(REF_PIC_LIST_0)>1)||
1201        (pcSlice->getColFromL0Flag()==0  && pcSlice->getNumRefIdx(REF_PIC_LIST_1)>1)))
1202      {
1203        WRITE_UVLC( pcSlice->getColRefIdx(), "collocated_ref_idx" );
1204      }
1205    }
1206#if H_MV
1207    }
1208#endif
1209    if ( (pcSlice->getPPS()->getUseWP() && pcSlice->getSliceType()==P_SLICE) || (pcSlice->getPPS()->getWPBiPred() && pcSlice->getSliceType()==B_SLICE) )
1210    {
1211      xCodePredWeightTable( pcSlice );
1212    }
1213#if H_3D_IC
1214    else if( pcSlice->getViewIndex() && ( pcSlice->getSliceType() == P_SLICE || pcSlice->getSliceType() == B_SLICE ) )
1215    {
1216      WRITE_FLAG( pcSlice->getApplyIC() ? 1 : 0, "slice_ic_enable_flag" );
1217      if( pcSlice->getApplyIC() )
1218      {
1219        WRITE_FLAG( pcSlice->getIcSkipParseFlag() ? 1 : 0, "ic_skip_mergeidx0" );
1220      }
1221    }
1222#endif
1223
1224#if H_3D_IV_MERGE
1225    assert(pcSlice->getMaxNumMergeCand()<=MRG_MAX_NUM_CANDS_MEM);
1226#else
1227    assert(pcSlice->getMaxNumMergeCand()<=MRG_MAX_NUM_CANDS);
1228#endif
1229    if (!pcSlice->isIntra())
1230    {
1231#if H_3D_IV_MERGE
1232      Bool ivMvPredFlag = pcSlice->getVPS()->getIvMvPredFlag( pcSlice->getLayerIdInVps() ) ;
1233      WRITE_UVLC( ( ivMvPredFlag ? MRG_MAX_NUM_CANDS_MEM : MRG_MAX_NUM_CANDS ) - pcSlice->getMaxNumMergeCand(), "five_minus_max_num_merge_cand");
1234#else
1235      WRITE_UVLC(MRG_MAX_NUM_CANDS - pcSlice->getMaxNumMergeCand(), "five_minus_max_num_merge_cand");
1236#endif
1237    }
1238    Int iCode = pcSlice->getSliceQp() - ( pcSlice->getPPS()->getPicInitQPMinus26() + 26 );
1239    WRITE_SVLC( iCode, "slice_qp_delta" ); 
1240    if (pcSlice->getPPS()->getSliceChromaQpFlag())
1241    {
1242      iCode = pcSlice->getSliceQpDeltaCb();
1243      WRITE_SVLC( iCode, "slice_qp_delta_cb" );
1244      iCode = pcSlice->getSliceQpDeltaCr();
1245      WRITE_SVLC( iCode, "slice_qp_delta_cr" );
1246    }
1247    if (pcSlice->getPPS()->getDeblockingFilterControlPresentFlag())
1248    {
1249      if (pcSlice->getPPS()->getDeblockingFilterOverrideEnabledFlag() )
1250      {
1251        WRITE_FLAG(pcSlice->getDeblockingFilterOverrideFlag(), "deblocking_filter_override_flag");
1252      }
1253      if (pcSlice->getDeblockingFilterOverrideFlag())
1254      {
1255        WRITE_FLAG(pcSlice->getDeblockingFilterDisable(), "slice_disable_deblocking_filter_flag");
1256        if(!pcSlice->getDeblockingFilterDisable())
1257        {
1258          WRITE_SVLC (pcSlice->getDeblockingFilterBetaOffsetDiv2(), "slice_beta_offset_div2");
1259          WRITE_SVLC (pcSlice->getDeblockingFilterTcOffsetDiv2(),   "slice_tc_offset_div2");
1260        }
1261      }
1262    }
1263
1264    Bool isSAOEnabled = (!pcSlice->getSPS()->getUseSAO())?(false):(pcSlice->getSaoEnabledFlag()||pcSlice->getSaoEnabledFlagChroma());
1265    Bool isDBFEnabled = (!pcSlice->getDeblockingFilterDisable());
1266
1267    if(pcSlice->getPPS()->getLoopFilterAcrossSlicesEnabledFlag() && ( isSAOEnabled || isDBFEnabled ))
1268    {
1269      WRITE_FLAG(pcSlice->getLFCrossSliceBoundaryFlag()?1:0, "slice_loop_filter_across_slices_enabled_flag");
1270    }
1271  }
1272  if(pcSlice->getPPS()->getSliceHeaderExtensionPresentFlag())
1273  {
1274#if !H_3D
1275    WRITE_UVLC(0,"slice_header_extension_length");
1276#else
1277    WRITE_UVLC(0,"slice_header_extension_length"); //<- this element needs to be set to the correct value!!
1278
1279    if( pcSlice->getSPS()->hasCamParInSliceHeader() )
1280    {
1281      for( UInt uiId = 0; uiId < pcSlice->getViewIndex(); uiId++ )
1282      {
1283        WRITE_SVLC( pcSlice->getCodedScale    ()[ uiId ],                                     "cp_scale" );
1284        WRITE_SVLC( pcSlice->getCodedOffset   ()[ uiId ],                                     "cp_off" );
1285        WRITE_SVLC( pcSlice->getInvCodedScale ()[ uiId ] + pcSlice->getCodedScale ()[ uiId ], "cp_inv_scale_plus_scale" );
1286        WRITE_SVLC( pcSlice->getInvCodedOffset()[ uiId ] + pcSlice->getCodedOffset()[ uiId ], "cp_inv_off_plus_off" );
1287      }
1288    }
1289
1290    Bool sliceSegmentHeaderExtension2Flag = false; 
1291    WRITE_FLAG( sliceSegmentHeaderExtension2Flag ? 1 : 0 , "slice_segment_header_extension2_flag" ); 
1292    if ( sliceSegmentHeaderExtension2Flag )
1293    {
1294      WRITE_UVLC(0,"slice_header_extension2_length");
1295    }
1296#endif
1297  }
1298}
1299
1300Void TEncCavlc::codePTL( TComPTL* pcPTL, Bool profilePresentFlag, Int maxNumSubLayersMinus1)
1301{
1302  if(profilePresentFlag)
1303  {
1304    codeProfileTier(pcPTL->getGeneralPTL());    // general_...
1305  }
1306  WRITE_CODE( pcPTL->getGeneralPTL()->getLevelIdc(), 8, "general_level_idc" );
1307
1308  for (Int i = 0; i < maxNumSubLayersMinus1; i++)
1309  {
1310#if !H_MV
1311    if(profilePresentFlag)
1312    {
1313#endif
1314      WRITE_FLAG( pcPTL->getSubLayerProfilePresentFlag(i), "sub_layer_profile_present_flag[i]" );
1315#if !H_MV
1316    }
1317#endif
1318   
1319    WRITE_FLAG( pcPTL->getSubLayerLevelPresentFlag(i),   "sub_layer_level_present_flag[i]" );
1320  }
1321 
1322  if (maxNumSubLayersMinus1 > 0)
1323  {
1324    for (Int i = maxNumSubLayersMinus1; i < 8; i++)
1325    {
1326      WRITE_CODE(0, 2, "reserved_zero_2bits");
1327    }
1328  }
1329 
1330  for(Int i = 0; i < maxNumSubLayersMinus1; i++)
1331  {
1332    if( profilePresentFlag && pcPTL->getSubLayerProfilePresentFlag(i) )
1333    {
1334      codeProfileTier(pcPTL->getSubLayerPTL(i));  // sub_layer_...
1335    }
1336    if( pcPTL->getSubLayerLevelPresentFlag(i) )
1337    {
1338      WRITE_CODE( pcPTL->getSubLayerPTL(i)->getLevelIdc(), 8, "sub_layer_level_idc[i]" );
1339    }
1340  }
1341}
1342Void TEncCavlc::codeProfileTier( ProfileTierLevel* ptl )
1343{
1344  WRITE_CODE( ptl->getProfileSpace(), 2 ,     "XXX_profile_space[]");
1345  WRITE_FLAG( ptl->getTierFlag    (),         "XXX_tier_flag[]"    );
1346  WRITE_CODE( ptl->getProfileIdc  (), 5 ,     "XXX_profile_idc[]"  );   
1347  for(Int j = 0; j < 32; j++)
1348  {
1349    WRITE_FLAG( ptl->getProfileCompatibilityFlag(j), "XXX_profile_compatibility_flag[][j]");   
1350  }
1351
1352  WRITE_FLAG(ptl->getProgressiveSourceFlag(),   "general_progressive_source_flag");
1353  WRITE_FLAG(ptl->getInterlacedSourceFlag(),    "general_interlaced_source_flag");
1354  WRITE_FLAG(ptl->getNonPackedConstraintFlag(), "general_non_packed_constraint_flag");
1355  WRITE_FLAG(ptl->getFrameOnlyConstraintFlag(), "general_frame_only_constraint_flag");
1356 
1357  WRITE_CODE(0 , 16, "XXX_reserved_zero_44bits[0..15]");
1358  WRITE_CODE(0 , 16, "XXX_reserved_zero_44bits[16..31]");
1359  WRITE_CODE(0 , 12, "XXX_reserved_zero_44bits[32..43]");
1360    }
1361
1362/**
1363 - write wavefront substreams sizes for the slice header.
1364 .
1365 \param pcSlice Where we find the substream size information.
1366 */
1367Void  TEncCavlc::codeTilesWPPEntryPoint( TComSlice* pSlice )
1368{
1369  if (!pSlice->getPPS()->getTilesEnabledFlag() && !pSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
1370  {
1371    return;
1372  }
1373  UInt numEntryPointOffsets = 0, offsetLenMinus1 = 0, maxOffset = 0;
1374  Int  numZeroSubstreamsAtStartOfSlice  = 0;
1375  UInt *entryPointOffset = NULL;
1376  if ( pSlice->getPPS()->getTilesEnabledFlag() )
1377  {
1378    numEntryPointOffsets = pSlice->getTileLocationCount();
1379    entryPointOffset     = new UInt[numEntryPointOffsets];
1380    for (Int idx=0; idx<pSlice->getTileLocationCount(); idx++)
1381    {
1382      if ( idx == 0 )
1383      {
1384        entryPointOffset [ idx ] = pSlice->getTileLocation( 0 );
1385      }
1386      else
1387      {
1388        entryPointOffset [ idx ] = pSlice->getTileLocation( idx ) - pSlice->getTileLocation( idx-1 );
1389      }
1390
1391      if ( entryPointOffset[ idx ] > maxOffset )
1392      {
1393        maxOffset = entryPointOffset[ idx ];
1394      }
1395    }
1396  }
1397  else if ( pSlice->getPPS()->getEntropyCodingSyncEnabledFlag() )
1398  {
1399    UInt* pSubstreamSizes               = pSlice->getSubstreamSizes();
1400    Int maxNumParts                       = pSlice->getPic()->getNumPartInCU();
1401    numZeroSubstreamsAtStartOfSlice       = pSlice->getSliceSegmentCurStartCUAddr()/maxNumParts/pSlice->getPic()->getFrameWidthInCU();
1402    Int  numZeroSubstreamsAtEndOfSlice    = pSlice->getPic()->getFrameHeightInCU()-1 - ((pSlice->getSliceSegmentCurEndCUAddr()-1)/maxNumParts/pSlice->getPic()->getFrameWidthInCU());
1403    numEntryPointOffsets                  = pSlice->getPPS()->getNumSubstreams() - numZeroSubstreamsAtStartOfSlice - numZeroSubstreamsAtEndOfSlice - 1;
1404    pSlice->setNumEntryPointOffsets(numEntryPointOffsets);
1405    entryPointOffset           = new UInt[numEntryPointOffsets];
1406    for (Int idx=0; idx<numEntryPointOffsets; idx++)
1407    {
1408      entryPointOffset[ idx ] = ( pSubstreamSizes[ idx+numZeroSubstreamsAtStartOfSlice ] >> 3 ) ;
1409      if ( entryPointOffset[ idx ] > maxOffset )
1410      {
1411        maxOffset = entryPointOffset[ idx ];
1412      }
1413    }
1414  }
1415  // Determine number of bits "offsetLenMinus1+1" required for entry point information
1416  offsetLenMinus1 = 0;
1417  while (maxOffset >= (1u << (offsetLenMinus1 + 1)))
1418  {
1419    offsetLenMinus1++;
1420    assert(offsetLenMinus1 + 1 < 32);   
1421  }
1422
1423  WRITE_UVLC(numEntryPointOffsets, "num_entry_point_offsets");
1424  if (numEntryPointOffsets>0)
1425  {
1426    WRITE_UVLC(offsetLenMinus1, "offset_len_minus1");
1427  }
1428
1429  for (UInt idx=0; idx<numEntryPointOffsets; idx++)
1430  {
1431    WRITE_CODE(entryPointOffset[ idx ]-1, offsetLenMinus1+1, "entry_point_offset_minus1");
1432  }
1433
1434  delete [] entryPointOffset;
1435}
1436
1437Void TEncCavlc::codeTerminatingBit      ( UInt uilsLast )
1438{
1439}
1440
1441Void TEncCavlc::codeSliceFinish ()
1442{
1443}
1444
1445Void TEncCavlc::codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1446{
1447  assert(0);
1448}
1449
1450Void TEncCavlc::codePartSize( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
1451{
1452  assert(0);
1453}
1454
1455Void TEncCavlc::codePredMode( TComDataCU* pcCU, UInt uiAbsPartIdx )
1456{
1457  assert(0);
1458}
1459
1460Void TEncCavlc::codeMergeFlag    ( TComDataCU* pcCU, UInt uiAbsPartIdx )
1461{
1462  assert(0);
1463}
1464
1465Void TEncCavlc::codeMergeIndex    ( TComDataCU* pcCU, UInt uiAbsPartIdx )
1466{
1467  assert(0);
1468}
1469
1470#if H_3D_ARP
1471Void TEncCavlc::codeARPW( TComDataCU* pcCU, UInt uiAbsPartIdx )
1472{
1473  assert(0);
1474}
1475#endif
1476
1477#if H_3D_IC
1478Void TEncCavlc::codeICFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
1479{
1480  assert(0);
1481}
1482#endif
1483
1484Void TEncCavlc::codeInterModeFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiEncMode )
1485{
1486  assert(0);
1487}
1488
1489Void TEncCavlc::codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
1490{
1491  assert(0);
1492}
1493
1494Void TEncCavlc::codeSkipFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
1495{
1496  assert(0);
1497}
1498
1499Void TEncCavlc::codeSplitFlag   ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
1500{
1501  assert(0);
1502}
1503
1504Void TEncCavlc::codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx )
1505{
1506  assert(0);
1507}
1508
1509Void TEncCavlc::codeQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth )
1510{
1511  assert(0);
1512}
1513
1514Void TEncCavlc::codeQtRootCbf( TComDataCU* pcCU, UInt uiAbsPartIdx )
1515{
1516  assert(0);
1517}
1518
1519Void TEncCavlc::codeQtCbfZero( TComDataCU* pcCU, TextType eType, UInt uiTrDepth )
1520{
1521  assert(0);
1522}
1523Void TEncCavlc::codeQtRootCbfZero( TComDataCU* pcCU )
1524{
1525  assert(0);
1526}
1527
1528Void TEncCavlc::codeTransformSkipFlags (TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, TextType eTType )
1529{
1530  assert(0);
1531}
1532
1533/** Code I_PCM information.
1534 * \param pcCU pointer to CU
1535 * \param uiAbsPartIdx CU index
1536 * \returns Void
1537 */
1538Void TEncCavlc::codeIPCMInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
1539{
1540  assert(0);
1541}
1542
1543Void TEncCavlc::codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool isMultiple)
1544{
1545  assert(0);
1546}
1547
1548Void TEncCavlc::codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx )
1549{
1550  assert(0);
1551}
1552
1553Void TEncCavlc::codeInterDir( TComDataCU* pcCU, UInt uiAbsPartIdx )
1554{
1555  assert(0);
1556}
1557
1558Void TEncCavlc::codeRefFrmIdx( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1559{
1560  assert(0);
1561}
1562
1563Void TEncCavlc::codeMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
1564{
1565  assert(0);
1566}
1567
1568Void TEncCavlc::codeDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx )
1569{
1570  Int iDQp  = pcCU->getQP( uiAbsPartIdx ) - pcCU->getRefQP( uiAbsPartIdx );
1571
1572  Int qpBdOffsetY =  pcCU->getSlice()->getSPS()->getQpBDOffsetY();
1573  iDQp = (iDQp + 78 + qpBdOffsetY + (qpBdOffsetY/2)) % (52 + qpBdOffsetY) - 26 - (qpBdOffsetY/2);
1574
1575  xWriteSvlc( iDQp );
1576 
1577  return;
1578}
1579
1580Void TEncCavlc::codeCoeffNxN    ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
1581{
1582  assert(0);
1583}
1584
1585Void TEncCavlc::estBit( estBitsSbacStruct* pcEstBitsCabac, Int width, Int height, TextType eTType )
1586{
1587  // printf("error : no VLC mode support in this version\n");
1588  return;
1589}
1590
1591// ====================================================================================================================
1592// Protected member functions
1593// ====================================================================================================================
1594
1595/** code explicit wp tables
1596 * \param TComSlice* pcSlice
1597 * \returns Void
1598 */
1599Void TEncCavlc::xCodePredWeightTable( TComSlice* pcSlice )
1600{
1601  wpScalingParam  *wp;
1602  Bool            bChroma     = true; // color always present in HEVC ?
1603  Int             iNbRef       = (pcSlice->getSliceType() == B_SLICE ) ? (2) : (1);
1604  Bool            bDenomCoded  = false;
1605  UInt            uiMode = 0;
1606  UInt            uiTotalSignalledWeightFlags = 0;
1607  if ( (pcSlice->getSliceType()==P_SLICE && pcSlice->getPPS()->getUseWP()) || (pcSlice->getSliceType()==B_SLICE && pcSlice->getPPS()->getWPBiPred()) )
1608  {
1609    uiMode = 1; // explicit
1610  }
1611  if(uiMode == 1)
1612  {
1613
1614    for ( Int iNumRef=0 ; iNumRef<iNbRef ; iNumRef++ ) 
1615    {
1616      RefPicList  eRefPicList = ( iNumRef ? REF_PIC_LIST_1 : REF_PIC_LIST_0 );
1617
1618      for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ ) 
1619      {
1620        pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
1621        if ( !bDenomCoded ) 
1622        {
1623          Int iDeltaDenom;
1624          WRITE_UVLC( wp[0].uiLog2WeightDenom, "luma_log2_weight_denom" );     // ue(v): luma_log2_weight_denom
1625
1626          if( bChroma )
1627          {
1628            iDeltaDenom = (wp[1].uiLog2WeightDenom - wp[0].uiLog2WeightDenom);
1629            WRITE_SVLC( iDeltaDenom, "delta_chroma_log2_weight_denom" );       // se(v): delta_chroma_log2_weight_denom
1630          }
1631          bDenomCoded = true;
1632        }
1633        WRITE_FLAG( wp[0].bPresentFlag, "luma_weight_lX_flag" );               // u(1): luma_weight_lX_flag
1634        uiTotalSignalledWeightFlags += wp[0].bPresentFlag;
1635      }
1636      if (bChroma) 
1637      {
1638        for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ ) 
1639        {
1640          pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
1641          WRITE_FLAG( wp[1].bPresentFlag, "chroma_weight_lX_flag" );           // u(1): chroma_weight_lX_flag
1642          uiTotalSignalledWeightFlags += 2*wp[1].bPresentFlag;
1643        }
1644      }
1645
1646      for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ ) 
1647      {
1648        pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
1649        if ( wp[0].bPresentFlag ) 
1650        {
1651          Int iDeltaWeight = (wp[0].iWeight - (1<<wp[0].uiLog2WeightDenom));
1652          WRITE_SVLC( iDeltaWeight, "delta_luma_weight_lX" );                  // se(v): delta_luma_weight_lX
1653          WRITE_SVLC( wp[0].iOffset, "luma_offset_lX" );                       // se(v): luma_offset_lX
1654        }
1655
1656        if ( bChroma ) 
1657        {
1658          if ( wp[1].bPresentFlag )
1659          {
1660            for ( Int j=1 ; j<3 ; j++ ) 
1661            {
1662              Int iDeltaWeight = (wp[j].iWeight - (1<<wp[1].uiLog2WeightDenom));
1663              WRITE_SVLC( iDeltaWeight, "delta_chroma_weight_lX" );            // se(v): delta_chroma_weight_lX
1664
1665              Int pred = ( 128 - ( ( 128*wp[j].iWeight)>>(wp[j].uiLog2WeightDenom) ) );
1666              Int iDeltaChroma = (wp[j].iOffset - pred);
1667              WRITE_SVLC( iDeltaChroma, "delta_chroma_offset_lX" );            // se(v): delta_chroma_offset_lX
1668            }
1669          }
1670        }
1671      }
1672    }
1673    assert(uiTotalSignalledWeightFlags<=24);
1674  }
1675}
1676
1677/** code quantization matrix
1678 *  \param scalingList quantization matrix information
1679 */
1680Void TEncCavlc::codeScalingList( TComScalingList* scalingList )
1681{
1682  UInt listId,sizeId;
1683  Bool scalingListPredModeFlag;
1684
1685#if SCALING_LIST_OUTPUT_RESULT
1686  Int startBit;
1687  Int startTotalBit;
1688  startBit = m_pcBitIf->getNumberOfWrittenBits();
1689  startTotalBit = m_pcBitIf->getNumberOfWrittenBits();
1690#endif
1691
1692    //for each size
1693    for(sizeId = 0; sizeId < SCALING_LIST_SIZE_NUM; sizeId++)
1694    {
1695      for(listId = 0; listId < g_scalingListNum[sizeId]; listId++)
1696      {
1697#if SCALING_LIST_OUTPUT_RESULT
1698        startBit = m_pcBitIf->getNumberOfWrittenBits();
1699#endif
1700        scalingListPredModeFlag = scalingList->checkPredMode( sizeId, listId );
1701        WRITE_FLAG( scalingListPredModeFlag, "scaling_list_pred_mode_flag" );
1702        if(!scalingListPredModeFlag)// Copy Mode
1703        {
1704          WRITE_UVLC( (Int)listId - (Int)scalingList->getRefMatrixId (sizeId,listId), "scaling_list_pred_matrix_id_delta");
1705        }
1706        else// DPCM Mode
1707        {
1708          xCodeScalingList(scalingList, sizeId, listId);
1709        }
1710#if SCALING_LIST_OUTPUT_RESULT
1711        printf("Matrix [%d][%d] Bit %d\n",sizeId,listId,m_pcBitIf->getNumberOfWrittenBits() - startBit);
1712#endif
1713      }
1714    }
1715#if SCALING_LIST_OUTPUT_RESULT
1716  printf("Total Bit %d\n",m_pcBitIf->getNumberOfWrittenBits()-startTotalBit);
1717#endif
1718  return;
1719}
1720/** code DPCM
1721 * \param scalingList quantization matrix information
1722 * \param sizeIdc size index
1723 * \param listIdc list index
1724 */
1725Void TEncCavlc::xCodeScalingList(TComScalingList* scalingList, UInt sizeId, UInt listId)
1726{
1727  Int coefNum = min(MAX_MATRIX_COEF_NUM,(Int)g_scalingListSize[sizeId]);
1728  UInt* scan  = (sizeId == 0) ? g_auiSigLastScan [ SCAN_DIAG ] [ 1 ] :  g_sigLastScanCG32x32;
1729  Int nextCoef = SCALING_LIST_START_VALUE;
1730  Int data;
1731  Int *src = scalingList->getScalingListAddress(sizeId, listId);
1732    if( sizeId > SCALING_LIST_8x8 )
1733    {
1734      WRITE_SVLC( scalingList->getScalingListDC(sizeId,listId) - 8, "scaling_list_dc_coef_minus8");
1735      nextCoef = scalingList->getScalingListDC(sizeId,listId);
1736    }
1737    for(Int i=0;i<coefNum;i++)
1738    {
1739      data = src[scan[i]] - nextCoef;
1740      nextCoef = src[scan[i]];
1741      if(data > 127)
1742      {
1743        data = data - 256;
1744      }
1745      if(data < -128)
1746      {
1747        data = data + 256;
1748      }
1749
1750      WRITE_SVLC( data,  "scaling_list_delta_coef");
1751    }
1752}
1753Bool TEncCavlc::findMatchingLTRP ( TComSlice* pcSlice, UInt *ltrpsIndex, Int ltrpPOC, Bool usedFlag )
1754{
1755  // Bool state = true, state2 = false;
1756  Int lsb = ltrpPOC % (1<<pcSlice->getSPS()->getBitsForPOC());
1757  for (Int k = 0; k < pcSlice->getSPS()->getNumLongTermRefPicSPS(); k++)
1758  {
1759    if ( (lsb == pcSlice->getSPS()->getLtRefPicPocLsbSps(k)) && (usedFlag == pcSlice->getSPS()->getUsedByCurrPicLtSPSFlag(k)) )
1760    {
1761      *ltrpsIndex = k;
1762      return true;
1763    }
1764  }
1765  return false;
1766}
1767Bool TComScalingList::checkPredMode(UInt sizeId, UInt listId)
1768{
1769  for(Int predListIdx = (Int)listId ; predListIdx >= 0; predListIdx--)
1770  {
1771    if( !memcmp(getScalingListAddress(sizeId,listId),((listId == predListIdx) ?
1772      getScalingListDefaultAddress(sizeId, predListIdx): getScalingListAddress(sizeId, predListIdx)),sizeof(Int)*min(MAX_MATRIX_COEF_NUM,(Int)g_scalingListSize[sizeId])) // check value of matrix
1773     && ((sizeId < SCALING_LIST_16x16) || (getScalingListDC(sizeId,listId) == getScalingListDC(sizeId,predListIdx)))) // check DC value
1774    {
1775      setRefMatrixId(sizeId, listId, predListIdx);
1776      return false;
1777    }
1778  }
1779  return true;
1780}
1781
1782#if LGE_INTER_SDC_E0156
1783Void TEncCavlc::codeInterSDCFlag ( TComDataCU* pcCU, UInt uiAbsPartIdx )
1784{
1785  assert(0);
1786}
1787
1788Void TEncCavlc::codeInterSDCResidualData  ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiSegment )
1789{
1790  assert(0);
1791}
1792#endif
1793//! \}
Note: See TracBrowser for help on using the repository browser.