source: SHVCSoftware/branches/SHM-5.0-dev/source/Lib/TLibEncoder/TEncCavlc.cpp @ 684

Last change on this file since 684 was 593, checked in by seregin, 11 years ago

update to HM-13.0

  • Property svn:eol-style set to native
File size: 84.7 KB
RevLine 
[313]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 *
[593]6 * Copyright (c) 2010-2014, ITU/ISO/IEC
[313]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
42//! \ingroup TLibEncoder
43//! \{
44
45#if ENC_DEC_TRACE
46
47Void  xTraceSPSHeader (TComSPS *pSPS)
48{
49  fprintf( g_hTrace, "=========== Sequence Parameter Set ID: %d ===========\n", pSPS->getSPSId() );
50}
51
52Void  xTracePPSHeader (TComPPS *pPPS)
53{
54  fprintf( g_hTrace, "=========== Picture Parameter Set ID: %d ===========\n", pPPS->getPPSId() );
55}
56
57Void  xTraceSliceHeader (TComSlice *pSlice)
58{
59  fprintf( g_hTrace, "=========== Slice ===========\n");
60}
61
62#endif
63
64
65
66// ====================================================================================================================
67// Constructor / destructor / create / destroy
68// ====================================================================================================================
69
70TEncCavlc::TEncCavlc()
71{
72  m_pcBitIf           = NULL;
73  m_uiCoeffCost       = 0;
74}
75
76TEncCavlc::~TEncCavlc()
77{
78}
79
80
81// ====================================================================================================================
82// Public member functions
83// ====================================================================================================================
84
85Void TEncCavlc::resetEntropy()
86{
87}
88
89
90Void TEncCavlc::codeDFFlag(UInt uiCode, const Char *pSymbolName)
91{
92  WRITE_FLAG(uiCode, pSymbolName);
93}
94Void TEncCavlc::codeDFSvlc(Int iCode, const Char *pSymbolName)
95{
96  WRITE_SVLC(iCode, pSymbolName);
97}
98
99Void TEncCavlc::codeShortTermRefPicSet( TComSPS* pcSPS, TComReferencePictureSet* rps, Bool calledFromSliceHeader, Int idx)
100{
101#if PRINT_RPS_INFO
102  Int lastBits = getNumberOfWrittenBits();
103#endif
104  if (idx > 0)
105  {
106  WRITE_FLAG( rps->getInterRPSPrediction(), "inter_ref_pic_set_prediction_flag" ); // inter_RPS_prediction_flag
107  }
108  if (rps->getInterRPSPrediction())
109  {
110    Int deltaRPS = rps->getDeltaRPS();
111    if(calledFromSliceHeader)
112    {
113      WRITE_UVLC( rps->getDeltaRIdxMinus1(), "delta_idx_minus1" ); // delta index of the Reference Picture Set used for prediction minus 1
114    }
115
116    WRITE_CODE( (deltaRPS >=0 ? 0: 1), 1, "delta_rps_sign" ); //delta_rps_sign
117    WRITE_UVLC( abs(deltaRPS) - 1, "abs_delta_rps_minus1"); // absolute delta RPS minus 1
118
119    for(Int j=0; j < rps->getNumRefIdc(); j++)
120    {
121      Int refIdc = rps->getRefIdc(j);
122      WRITE_CODE( (refIdc==1? 1: 0), 1, "used_by_curr_pic_flag" ); //first bit is "1" if Idc is 1
123      if (refIdc != 1) 
124      {
125        WRITE_CODE( refIdc>>1, 1, "use_delta_flag" ); //second bit is "1" if Idc is 2, "0" otherwise.
126      }
127    }
128  }
129  else
130  {
131    WRITE_UVLC( rps->getNumberOfNegativePictures(), "num_negative_pics" );
132    WRITE_UVLC( rps->getNumberOfPositivePictures(), "num_positive_pics" );
133    Int prev = 0;
134    for(Int j=0 ; j < rps->getNumberOfNegativePictures(); j++)
135    {
136      WRITE_UVLC( prev-rps->getDeltaPOC(j)-1, "delta_poc_s0_minus1" );
137      prev = rps->getDeltaPOC(j);
138      WRITE_FLAG( rps->getUsed(j), "used_by_curr_pic_s0_flag"); 
139    }
140    prev = 0;
141    for(Int j=rps->getNumberOfNegativePictures(); j < rps->getNumberOfNegativePictures()+rps->getNumberOfPositivePictures(); j++)
142    {
143      WRITE_UVLC( rps->getDeltaPOC(j)-prev-1, "delta_poc_s1_minus1" );
144      prev = rps->getDeltaPOC(j);
145      WRITE_FLAG( rps->getUsed(j), "used_by_curr_pic_s1_flag" ); 
146    }
147  }
148
149#if PRINT_RPS_INFO
150  printf("irps=%d (%2d bits) ", rps->getInterRPSPrediction(), getNumberOfWrittenBits() - lastBits);
151  rps->printDeltaPOC();
152#endif
153}
154
155
156Void TEncCavlc::codePPS( TComPPS* pcPPS )
157{
158#if ENC_DEC_TRACE 
159  xTracePPSHeader (pcPPS);
160#endif
161 
162  WRITE_UVLC( pcPPS->getPPSId(),                             "pps_pic_parameter_set_id" );
163  WRITE_UVLC( pcPPS->getSPSId(),                             "pps_seq_parameter_set_id" );
164  WRITE_FLAG( pcPPS->getDependentSliceSegmentsEnabledFlag()    ? 1 : 0, "dependent_slice_segments_enabled_flag" );
165  WRITE_FLAG( pcPPS->getOutputFlagPresentFlag() ? 1 : 0,     "output_flag_present_flag" );
166  WRITE_CODE( pcPPS->getNumExtraSliceHeaderBits(), 3,        "num_extra_slice_header_bits");
167  WRITE_FLAG( pcPPS->getSignHideFlag(), "sign_data_hiding_flag" );
168  WRITE_FLAG( pcPPS->getCabacInitPresentFlag() ? 1 : 0,   "cabac_init_present_flag" );
169  WRITE_UVLC( pcPPS->getNumRefIdxL0DefaultActive()-1,     "num_ref_idx_l0_default_active_minus1");
170  WRITE_UVLC( pcPPS->getNumRefIdxL1DefaultActive()-1,     "num_ref_idx_l1_default_active_minus1");
171
172  WRITE_SVLC( pcPPS->getPicInitQPMinus26(),                  "init_qp_minus26");
173  WRITE_FLAG( pcPPS->getConstrainedIntraPred() ? 1 : 0,      "constrained_intra_pred_flag" );
174  WRITE_FLAG( pcPPS->getUseTransformSkip() ? 1 : 0,  "transform_skip_enabled_flag" ); 
175  WRITE_FLAG( pcPPS->getUseDQP() ? 1 : 0, "cu_qp_delta_enabled_flag" );
176  if ( pcPPS->getUseDQP() )
177  {
178    WRITE_UVLC( pcPPS->getMaxCuDQPDepth(), "diff_cu_qp_delta_depth" );
179  }
180  WRITE_SVLC( pcPPS->getChromaCbQpOffset(),                   "pps_cb_qp_offset" );
181  WRITE_SVLC( pcPPS->getChromaCrQpOffset(),                   "pps_cr_qp_offset" );
182  WRITE_FLAG( pcPPS->getSliceChromaQpFlag() ? 1 : 0,          "pps_slice_chroma_qp_offsets_present_flag" );
183
184  WRITE_FLAG( pcPPS->getUseWP() ? 1 : 0,  "weighted_pred_flag" );   // Use of Weighting Prediction (P_SLICE)
185  WRITE_FLAG( pcPPS->getWPBiPred() ? 1 : 0, "weighted_bipred_flag" );  // Use of Weighting Bi-Prediction (B_SLICE)
186  WRITE_FLAG( pcPPS->getTransquantBypassEnableFlag() ? 1 : 0, "transquant_bypass_enable_flag" );
187  WRITE_FLAG( pcPPS->getTilesEnabledFlag()             ? 1 : 0, "tiles_enabled_flag" );
188  WRITE_FLAG( pcPPS->getEntropyCodingSyncEnabledFlag() ? 1 : 0, "entropy_coding_sync_enabled_flag" );
189  if( pcPPS->getTilesEnabledFlag() )
190  {
191    WRITE_UVLC( pcPPS->getNumColumnsMinus1(),                                    "num_tile_columns_minus1" );
192    WRITE_UVLC( pcPPS->getNumRowsMinus1(),                                       "num_tile_rows_minus1" );
193    WRITE_FLAG( pcPPS->getUniformSpacingFlag(),                                  "uniform_spacing_flag" );
194    if( pcPPS->getUniformSpacingFlag() == 0 )
195    {
196      for(UInt i=0; i<pcPPS->getNumColumnsMinus1(); i++)
197      {
198        WRITE_UVLC( pcPPS->getColumnWidth(i)-1,                                  "column_width_minus1" );
199      }
200      for(UInt i=0; i<pcPPS->getNumRowsMinus1(); i++)
201      {
202        WRITE_UVLC( pcPPS->getRowHeight(i)-1,                                    "row_height_minus1" );
203      }
204    }
205    if(pcPPS->getNumColumnsMinus1() !=0 || pcPPS->getNumRowsMinus1() !=0)
206    {
207      WRITE_FLAG( pcPPS->getLoopFilterAcrossTilesEnabledFlag()?1 : 0,          "loop_filter_across_tiles_enabled_flag");
208    }
209  }
210  WRITE_FLAG( pcPPS->getLoopFilterAcrossSlicesEnabledFlag()?1 : 0,        "loop_filter_across_slices_enabled_flag");
211  WRITE_FLAG( pcPPS->getDeblockingFilterControlPresentFlag()?1 : 0,       "deblocking_filter_control_present_flag");
212  if(pcPPS->getDeblockingFilterControlPresentFlag())
213  {
214    WRITE_FLAG( pcPPS->getDeblockingFilterOverrideEnabledFlag() ? 1 : 0,  "deblocking_filter_override_enabled_flag" ); 
215    WRITE_FLAG( pcPPS->getPicDisableDeblockingFilterFlag() ? 1 : 0,       "pps_disable_deblocking_filter_flag" );
216    if(!pcPPS->getPicDisableDeblockingFilterFlag())
217    {
218      WRITE_SVLC( pcPPS->getDeblockingFilterBetaOffsetDiv2(),             "pps_beta_offset_div2" );
219      WRITE_SVLC( pcPPS->getDeblockingFilterTcOffsetDiv2(),               "pps_tc_offset_div2" );
220    }
221  }
[442]222
[540]223#if SCALINGLIST_INFERRING
224  if( pcPPS->getLayerId() > 0 )
225  {
226    WRITE_FLAG( pcPPS->getInferScalingListFlag() ? 1 : 0, "pps_infer_scaling_list_flag" );
227  }
228
229  if( pcPPS->getInferScalingListFlag() )
230  {
231    // The value of pps_scaling_list_ref_layer_id shall be in the range of 0 to 62, inclusive
232    assert( pcPPS->getScalingListRefLayerId() <= 62 );
233
234    WRITE_UVLC( pcPPS->getScalingListRefLayerId(), "pps_scaling_list_ref_layer_id" );
235  }
236  else
237  {
[442]238#endif
239
[540]240  WRITE_FLAG( pcPPS->getScalingListPresentFlag() ? 1 : 0,                          "pps_scaling_list_data_present_flag" ); 
[313]241  if( pcPPS->getScalingListPresentFlag() )
242  {
[540]243    codeScalingList( m_pcSlice->getScalingList() );
244  }
[442]245
[540]246#if SCALINGLIST_INFERRING
247  }
[442]248#endif
249
[313]250  WRITE_FLAG( pcPPS->getListsModificationPresentFlag(), "lists_modification_present_flag");
251  WRITE_UVLC( pcPPS->getLog2ParallelMergeLevelMinus2(), "log2_parallel_merge_level_minus2");
252  WRITE_FLAG( pcPPS->getSliceHeaderExtensionPresentFlag() ? 1 : 0, "slice_segment_header_extension_present_flag");
253  WRITE_FLAG( 0, "pps_extension_flag" );
254}
255
256Void TEncCavlc::codeVUI( TComVUI *pcVUI, TComSPS* pcSPS )
257{
258#if ENC_DEC_TRACE
259  fprintf( g_hTrace, "----------- vui_parameters -----------\n");
260#endif
261  WRITE_FLAG(pcVUI->getAspectRatioInfoPresentFlag(),            "aspect_ratio_info_present_flag");
262  if (pcVUI->getAspectRatioInfoPresentFlag())
263  {
264    WRITE_CODE(pcVUI->getAspectRatioIdc(), 8,                   "aspect_ratio_idc" );
265    if (pcVUI->getAspectRatioIdc() == 255)
266    {
267      WRITE_CODE(pcVUI->getSarWidth(), 16,                      "sar_width");
268      WRITE_CODE(pcVUI->getSarHeight(), 16,                     "sar_height");
269    }
270  }
271  WRITE_FLAG(pcVUI->getOverscanInfoPresentFlag(),               "overscan_info_present_flag");
272  if (pcVUI->getOverscanInfoPresentFlag())
273  {
274    WRITE_FLAG(pcVUI->getOverscanAppropriateFlag(),             "overscan_appropriate_flag");
275  }
276  WRITE_FLAG(pcVUI->getVideoSignalTypePresentFlag(),            "video_signal_type_present_flag");
277  if (pcVUI->getVideoSignalTypePresentFlag())
278  {
279    WRITE_CODE(pcVUI->getVideoFormat(), 3,                      "video_format");
280    WRITE_FLAG(pcVUI->getVideoFullRangeFlag(),                  "video_full_range_flag");
281    WRITE_FLAG(pcVUI->getColourDescriptionPresentFlag(),        "colour_description_present_flag");
282    if (pcVUI->getColourDescriptionPresentFlag())
283    {
284      WRITE_CODE(pcVUI->getColourPrimaries(), 8,                "colour_primaries");
285      WRITE_CODE(pcVUI->getTransferCharacteristics(), 8,        "transfer_characteristics");
286      WRITE_CODE(pcVUI->getMatrixCoefficients(), 8,             "matrix_coefficients");
287    }
288  }
289
290  WRITE_FLAG(pcVUI->getChromaLocInfoPresentFlag(),              "chroma_loc_info_present_flag");
291  if (pcVUI->getChromaLocInfoPresentFlag())
292  {
293    WRITE_UVLC(pcVUI->getChromaSampleLocTypeTopField(),         "chroma_sample_loc_type_top_field");
294    WRITE_UVLC(pcVUI->getChromaSampleLocTypeBottomField(),      "chroma_sample_loc_type_bottom_field");
295  }
296
297  WRITE_FLAG(pcVUI->getNeutralChromaIndicationFlag(),           "neutral_chroma_indication_flag");
298  WRITE_FLAG(pcVUI->getFieldSeqFlag(),                          "field_seq_flag");
299  WRITE_FLAG(pcVUI->getFrameFieldInfoPresentFlag(),             "frame_field_info_present_flag");
300
301  Window defaultDisplayWindow = pcVUI->getDefaultDisplayWindow();
302  WRITE_FLAG(defaultDisplayWindow.getWindowEnabledFlag(),       "default_display_window_flag");
303  if( defaultDisplayWindow.getWindowEnabledFlag() )
304  {
305    WRITE_UVLC(defaultDisplayWindow.getWindowLeftOffset(),      "def_disp_win_left_offset");
306    WRITE_UVLC(defaultDisplayWindow.getWindowRightOffset(),     "def_disp_win_right_offset");
307    WRITE_UVLC(defaultDisplayWindow.getWindowTopOffset(),       "def_disp_win_top_offset");
308    WRITE_UVLC(defaultDisplayWindow.getWindowBottomOffset(),    "def_disp_win_bottom_offset");
309  }
310  TimingInfo *timingInfo = pcVUI->getTimingInfo();
311  WRITE_FLAG(timingInfo->getTimingInfoPresentFlag(),          "vui_timing_info_present_flag");
312  if(timingInfo->getTimingInfoPresentFlag())
313  {
314    WRITE_CODE(timingInfo->getNumUnitsInTick(), 32,           "vui_num_units_in_tick");
315    WRITE_CODE(timingInfo->getTimeScale(),      32,           "vui_time_scale");
316    WRITE_FLAG(timingInfo->getPocProportionalToTimingFlag(),  "vui_poc_proportional_to_timing_flag");
317    if(timingInfo->getPocProportionalToTimingFlag())
318    {
319      WRITE_UVLC(timingInfo->getNumTicksPocDiffOneMinus1(),   "vui_num_ticks_poc_diff_one_minus1");
320    }
321  WRITE_FLAG(pcVUI->getHrdParametersPresentFlag(),              "hrd_parameters_present_flag");
322  if( pcVUI->getHrdParametersPresentFlag() )
323  {
324    codeHrdParameters(pcVUI->getHrdParameters(), 1, pcSPS->getMaxTLayers() - 1 );
325  }
326  }
327
328  WRITE_FLAG(pcVUI->getBitstreamRestrictionFlag(),              "bitstream_restriction_flag");
329  if (pcVUI->getBitstreamRestrictionFlag())
330  {
331    WRITE_FLAG(pcVUI->getTilesFixedStructureFlag(),             "tiles_fixed_structure_flag");
332    WRITE_FLAG(pcVUI->getMotionVectorsOverPicBoundariesFlag(),  "motion_vectors_over_pic_boundaries_flag");
333    WRITE_FLAG(pcVUI->getRestrictedRefPicListsFlag(),           "restricted_ref_pic_lists_flag");
334    WRITE_UVLC(pcVUI->getMinSpatialSegmentationIdc(),           "min_spatial_segmentation_idc");
335    WRITE_UVLC(pcVUI->getMaxBytesPerPicDenom(),                 "max_bytes_per_pic_denom");
336    WRITE_UVLC(pcVUI->getMaxBitsPerMinCuDenom(),                "max_bits_per_mincu_denom");
337    WRITE_UVLC(pcVUI->getLog2MaxMvLengthHorizontal(),           "log2_max_mv_length_horizontal");
338    WRITE_UVLC(pcVUI->getLog2MaxMvLengthVertical(),             "log2_max_mv_length_vertical");
339  }
340}
341
342Void TEncCavlc::codeHrdParameters( TComHRD *hrd, Bool commonInfPresentFlag, UInt maxNumSubLayersMinus1 )
343{
344  if( commonInfPresentFlag )
345  {
346    WRITE_FLAG( hrd->getNalHrdParametersPresentFlag() ? 1 : 0 ,  "nal_hrd_parameters_present_flag" );
347    WRITE_FLAG( hrd->getVclHrdParametersPresentFlag() ? 1 : 0 ,  "vcl_hrd_parameters_present_flag" );
348    if( hrd->getNalHrdParametersPresentFlag() || hrd->getVclHrdParametersPresentFlag() )
349    {
350      WRITE_FLAG( hrd->getSubPicCpbParamsPresentFlag() ? 1 : 0,  "sub_pic_cpb_params_present_flag" );
351      if( hrd->getSubPicCpbParamsPresentFlag() )
352      {
353        WRITE_CODE( hrd->getTickDivisorMinus2(), 8,              "tick_divisor_minus2" );
354        WRITE_CODE( hrd->getDuCpbRemovalDelayLengthMinus1(), 5,  "du_cpb_removal_delay_length_minus1" );
355        WRITE_FLAG( hrd->getSubPicCpbParamsInPicTimingSEIFlag() ? 1 : 0, "sub_pic_cpb_params_in_pic_timing_sei_flag" );
356        WRITE_CODE( hrd->getDpbOutputDelayDuLengthMinus1(), 5,   "dpb_output_delay_du_length_minus1"  );
357      }
358      WRITE_CODE( hrd->getBitRateScale(), 4,                     "bit_rate_scale" );
359      WRITE_CODE( hrd->getCpbSizeScale(), 4,                     "cpb_size_scale" );
360      if( hrd->getSubPicCpbParamsPresentFlag() )
361      {
362        WRITE_CODE( hrd->getDuCpbSizeScale(), 4,                "du_cpb_size_scale" ); 
363      }
364      WRITE_CODE( hrd->getInitialCpbRemovalDelayLengthMinus1(), 5, "initial_cpb_removal_delay_length_minus1" );
365      WRITE_CODE( hrd->getCpbRemovalDelayLengthMinus1(),        5, "au_cpb_removal_delay_length_minus1" );
366      WRITE_CODE( hrd->getDpbOutputDelayLengthMinus1(),         5, "dpb_output_delay_length_minus1" );
367    }
368  }
369  Int i, j, nalOrVcl;
370  for( i = 0; i <= maxNumSubLayersMinus1; i ++ )
371  {
372    WRITE_FLAG( hrd->getFixedPicRateFlag( i ) ? 1 : 0,          "fixed_pic_rate_general_flag");
373    if( !hrd->getFixedPicRateFlag( i ) )
374    {
375      WRITE_FLAG( hrd->getFixedPicRateWithinCvsFlag( i ) ? 1 : 0, "fixed_pic_rate_within_cvs_flag");
376    }
377    else
378    {
379      hrd->setFixedPicRateWithinCvsFlag( i, true );
380    }
381    if( hrd->getFixedPicRateWithinCvsFlag( i ) )
382    {
383      WRITE_UVLC( hrd->getPicDurationInTcMinus1( i ),           "elemental_duration_in_tc_minus1");
384    }
385    else
386    {
387      WRITE_FLAG( hrd->getLowDelayHrdFlag( i ) ? 1 : 0,           "low_delay_hrd_flag");
388    }
389    if (!hrd->getLowDelayHrdFlag( i ))
390    {
391      WRITE_UVLC( hrd->getCpbCntMinus1( i ),                      "cpb_cnt_minus1");
392    }
393   
394    for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
395    {
396      if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) ||
397          ( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
398      {
399        for( j = 0; j <= ( hrd->getCpbCntMinus1( i ) ); j ++ )
400        {
401          WRITE_UVLC( hrd->getBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_value_minus1");
402          WRITE_UVLC( hrd->getCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_value_minus1");
403          if( hrd->getSubPicCpbParamsPresentFlag() )
404          {
405            WRITE_UVLC( hrd->getDuCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_du_value_minus1"); 
406            WRITE_UVLC( hrd->getDuBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_du_value_minus1");
407          }
408          WRITE_FLAG( hrd->getCbrFlag( i, j, nalOrVcl ) ? 1 : 0, "cbr_flag");
409        }
410      }
411    }
412  }
413}
414
415Void TEncCavlc::codeSPS( TComSPS* pcSPS )
416{
417#if ENC_DEC_TRACE 
418  xTraceSPSHeader (pcSPS);
419#endif
420  WRITE_CODE( pcSPS->getVPSId (),          4,       "sps_video_parameter_set_id" );
[494]421#if SVC_EXTENSION
[313]422  if(pcSPS->getLayerId() == 0)
423  {
424#endif
425    WRITE_CODE( pcSPS->getMaxTLayers() - 1,  3,       "sps_max_sub_layers_minus1" );
426    WRITE_FLAG( pcSPS->getTemporalIdNestingFlag() ? 1 : 0,                             "sps_temporal_id_nesting_flag" );
[494]427#if SVC_EXTENSION
[313]428  }
429#endif
[442]430#ifdef SPS_PTL_FIX
431  if (pcSPS->getLayerId() == 0)
432  {
433    codePTL(pcSPS->getPTL(), 1, pcSPS->getMaxTLayers() - 1);
434  }
435#else
[313]436  codePTL(pcSPS->getPTL(), 1, pcSPS->getMaxTLayers() - 1);
[442]437#endif
[313]438  WRITE_UVLC( pcSPS->getSPSId (),                   "sps_seq_parameter_set_id" );
[442]439#if REPN_FORMAT_IN_VPS
440  if( pcSPS->getLayerId() > 0 )
[313]441  {
[442]442    WRITE_FLAG( pcSPS->getUpdateRepFormatFlag(), "update_rep_format_flag" );
[313]443  }
[540]444#if O0096_REP_FORMAT_INDEX
445  if( pcSPS->getLayerId() == 0 ) 
446#else
[442]447  if( pcSPS->getLayerId() == 0 || pcSPS->getUpdateRepFormatFlag() ) 
[540]448#endif
[442]449  {
450#endif
451    WRITE_UVLC( pcSPS->getChromaFormatIdc (),         "chroma_format_idc" );
452    assert(pcSPS->getChromaFormatIdc () == 1);
453    // in the first version chroma_format_idc can only be equal to 1 (4:2:0)
454    if( pcSPS->getChromaFormatIdc () == 3 )
455    {
456      WRITE_FLAG( 0,                                  "separate_colour_plane_flag");
457    }
[313]458
[442]459    WRITE_UVLC( pcSPS->getPicWidthInLumaSamples (),   "pic_width_in_luma_samples" );
460    WRITE_UVLC( pcSPS->getPicHeightInLumaSamples(),   "pic_height_in_luma_samples" );
461#if REPN_FORMAT_IN_VPS
462  }
[540]463#if O0096_REP_FORMAT_INDEX
464  else if (pcSPS->getUpdateRepFormatFlag())
465  {
466    WRITE_CODE( pcSPS->getUpdateRepFormatIndex(), 8,   "update_rep_format_index");
467  }
[442]468#endif
[540]469#endif
[313]470  Window conf = pcSPS->getConformanceWindow();
471
472  WRITE_FLAG( conf.getWindowEnabledFlag(),          "conformance_window_flag" );
473  if (conf.getWindowEnabledFlag())
474  {
475    WRITE_UVLC( conf.getWindowLeftOffset()   / TComSPS::getWinUnitX(pcSPS->getChromaFormatIdc() ), "conf_win_left_offset" );
476    WRITE_UVLC( conf.getWindowRightOffset()  / TComSPS::getWinUnitX(pcSPS->getChromaFormatIdc() ), "conf_win_right_offset" );
477    WRITE_UVLC( conf.getWindowTopOffset()    / TComSPS::getWinUnitY(pcSPS->getChromaFormatIdc() ), "conf_win_top_offset" );
478    WRITE_UVLC( conf.getWindowBottomOffset() / TComSPS::getWinUnitY(pcSPS->getChromaFormatIdc() ), "conf_win_bottom_offset" );
479  }
480
[442]481#if REPN_FORMAT_IN_VPS
[540]482#if O0096_REP_FORMAT_INDEX
483  if( pcSPS->getLayerId() == 0 ) 
484#else
[442]485  if( pcSPS->getLayerId() == 0 || pcSPS->getUpdateRepFormatFlag() ) 
[540]486#endif 
[442]487  {
488    assert( pcSPS->getBitDepthY() >= 8 );
489    assert( pcSPS->getBitDepthC() >= 8 );
490#endif
491    WRITE_UVLC( pcSPS->getBitDepthY() - 8,             "bit_depth_luma_minus8" );
492    WRITE_UVLC( pcSPS->getBitDepthC() - 8,             "bit_depth_chroma_minus8" );
493#if REPN_FORMAT_IN_VPS
494  }
495#endif
[313]496  WRITE_UVLC( pcSPS->getBitsForPOC()-4,                 "log2_max_pic_order_cnt_lsb_minus4" );
497
498  const Bool subLayerOrderingInfoPresentFlag = 1;
499  WRITE_FLAG(subLayerOrderingInfoPresentFlag,       "sps_sub_layer_ordering_info_present_flag");
500  for(UInt i=0; i <= pcSPS->getMaxTLayers()-1; i++)
501  {
502    WRITE_UVLC( pcSPS->getMaxDecPicBuffering(i) - 1,       "sps_max_dec_pic_buffering_minus1[i]" );
503    WRITE_UVLC( pcSPS->getNumReorderPics(i),               "sps_num_reorder_pics[i]" );
504    WRITE_UVLC( pcSPS->getMaxLatencyIncrease(i),           "sps_max_latency_increase_plus1[i]" );
505    if (!subLayerOrderingInfoPresentFlag)
506    {
507      break;
508    }
509  }
510  assert( pcSPS->getMaxCUWidth() == pcSPS->getMaxCUHeight() );
511 
512  WRITE_UVLC( pcSPS->getLog2MinCodingBlockSize() - 3,                                "log2_min_coding_block_size_minus3" );
513  WRITE_UVLC( pcSPS->getLog2DiffMaxMinCodingBlockSize(),                             "log2_diff_max_min_coding_block_size" );
514  WRITE_UVLC( pcSPS->getQuadtreeTULog2MinSize() - 2,                                 "log2_min_transform_block_size_minus2" );
515  WRITE_UVLC( pcSPS->getQuadtreeTULog2MaxSize() - pcSPS->getQuadtreeTULog2MinSize(), "log2_diff_max_min_transform_block_size" );
516  WRITE_UVLC( pcSPS->getQuadtreeTUMaxDepthInter() - 1,                               "max_transform_hierarchy_depth_inter" );
517  WRITE_UVLC( pcSPS->getQuadtreeTUMaxDepthIntra() - 1,                               "max_transform_hierarchy_depth_intra" );
518  WRITE_FLAG( pcSPS->getScalingListFlag() ? 1 : 0,                                   "scaling_list_enabled_flag" ); 
519  if(pcSPS->getScalingListFlag())
520  {
[540]521#if SCALINGLIST_INFERRING
522    if( pcSPS->getLayerId() > 0 )
[313]523    {
[540]524      WRITE_FLAG( pcSPS->getInferScalingListFlag() ? 1 : 0, "sps_infer_scaling_list_flag" );
525    }
[442]526
[540]527    if( pcSPS->getInferScalingListFlag() )
[442]528    {
[540]529      // The value of pps_scaling_list_ref_layer_id shall be in the range of 0 to 62, inclusive
530      assert( pcSPS->getScalingListRefLayerId() <= 62 );
[442]531
[540]532      WRITE_UVLC( pcSPS->getScalingListRefLayerId(), "sps_scaling_list_ref_layer_id" );
[442]533    }
534    else
535    {
[540]536#endif
537    WRITE_FLAG( pcSPS->getScalingListPresentFlag() ? 1 : 0,                          "sps_scaling_list_data_present_flag" ); 
538    if(pcSPS->getScalingListPresentFlag())
539    {
[313]540      codeScalingList( m_pcSlice->getScalingList() );
541    }
[540]542#if SCALINGLIST_INFERRING
543    }
[442]544#endif
[313]545  }
546  WRITE_FLAG( pcSPS->getUseAMP() ? 1 : 0,                                            "amp_enabled_flag" );
547  WRITE_FLAG( pcSPS->getUseSAO() ? 1 : 0,                                            "sample_adaptive_offset_enabled_flag");
548
549  WRITE_FLAG( pcSPS->getUsePCM() ? 1 : 0,                                            "pcm_enabled_flag");
550  if( pcSPS->getUsePCM() )
551  {
552    WRITE_CODE( pcSPS->getPCMBitDepthLuma() - 1, 4,                                  "pcm_sample_bit_depth_luma_minus1" );
553    WRITE_CODE( pcSPS->getPCMBitDepthChroma() - 1, 4,                                "pcm_sample_bit_depth_chroma_minus1" );
554    WRITE_UVLC( pcSPS->getPCMLog2MinSize() - 3,                                      "log2_min_pcm_luma_coding_block_size_minus3" );
555    WRITE_UVLC( pcSPS->getPCMLog2MaxSize() - pcSPS->getPCMLog2MinSize(),             "log2_diff_max_min_pcm_luma_coding_block_size" );
556    WRITE_FLAG( pcSPS->getPCMFilterDisableFlag()?1 : 0,                              "pcm_loop_filter_disable_flag");
557  }
558
559  assert( pcSPS->getMaxTLayers() > 0 );         
560
561  TComRPSList* rpsList = pcSPS->getRPSList();
562  TComReferencePictureSet*      rps;
[442]563 
[313]564  WRITE_UVLC(rpsList->getNumberOfReferencePictureSets(), "num_short_term_ref_pic_sets" );
565  for(Int i=0; i < rpsList->getNumberOfReferencePictureSets(); i++)
566  {
567    rps = rpsList->getReferencePictureSet(i);
568    codeShortTermRefPicSet(pcSPS,rps,false, i);
569  }
570  WRITE_FLAG( pcSPS->getLongTermRefsPresent() ? 1 : 0,         "long_term_ref_pics_present_flag" );
571  if (pcSPS->getLongTermRefsPresent()) 
572  {
573    WRITE_UVLC(pcSPS->getNumLongTermRefPicSPS(), "num_long_term_ref_pic_sps" );
574    for (UInt k = 0; k < pcSPS->getNumLongTermRefPicSPS(); k++)
575    {
576      WRITE_CODE( pcSPS->getLtRefPicPocLsbSps(k), pcSPS->getBitsForPOC(), "lt_ref_pic_poc_lsb_sps");
577      WRITE_FLAG( pcSPS->getUsedByCurrPicLtSPSFlag(k), "used_by_curr_pic_lt_sps_flag");
578    }
579  }
580  WRITE_FLAG( pcSPS->getTMVPFlagsPresent()  ? 1 : 0,           "sps_temporal_mvp_enable_flag" );
[593]581
[313]582  WRITE_FLAG( pcSPS->getUseStrongIntraSmoothing(),             "sps_strong_intra_smoothing_enable_flag" );
583
584  WRITE_FLAG( pcSPS->getVuiParametersPresentFlag(),             "vui_parameters_present_flag" );
585  if (pcSPS->getVuiParametersPresentFlag())
586  {
587      codeVUI(pcSPS->getVuiParameters(), pcSPS);
588  }
589
590#if SPS_EXTENSION
591  WRITE_FLAG( 1, "sps_extension_flag" );
592  if( 1 )   // if( sps_extension_flag )
593  {
[540]594#if O0142_CONDITIONAL_SPS_EXTENSION
595    UInt spsExtensionTypeFlag[8] = { 0, 1, 0, 0, 0, 0, 0, 0 };
596    for (UInt i = 0; i < 8; i++)
597    {
598      WRITE_FLAG( spsExtensionTypeFlag[i], "sps_extension_type_flag" );
599    }
600    if (spsExtensionTypeFlag[1])
601    {
602      codeSPSExtension( pcSPS );
603    }
604#else
[313]605    codeSPSExtension( pcSPS );
606    WRITE_FLAG( 0, "sps_extension2_flag" );
[540]607#endif
[313]608  }
609#else
610  WRITE_FLAG( 0, "sps_extension_flag" );
611#endif
612}
613#if SPS_EXTENSION
614Void TEncCavlc::codeSPSExtension( TComSPS* pcSPS )
615{
616  // more syntax elements to be written here
[442]617
618  // Vertical MV component restriction is not used in SHVC CTC
619  WRITE_FLAG( 0, "inter_view_mv_vert_constraint_flag" );
[566]620
[313]621  if( pcSPS->getLayerId() > 0 )
622  {
623    WRITE_UVLC( pcSPS->getNumScaledRefLayerOffsets(),      "num_scaled_ref_layer_offsets" );
624    for(Int i = 0; i < pcSPS->getNumScaledRefLayerOffsets(); i++)
625    {
626      Window scaledWindow = pcSPS->getScaledRefLayerWindow(i);
[540]627#if O0098_SCALED_REF_LAYER_ID
628      WRITE_CODE( pcSPS->getScaledRefLayerId(i), 6,          "scaled_ref_layer_id" );
629#endif
[313]630      WRITE_SVLC( scaledWindow.getWindowLeftOffset()   >> 1, "scaled_ref_layer_left_offset" );
631      WRITE_SVLC( scaledWindow.getWindowTopOffset()    >> 1, "scaled_ref_layer_top_offset" );
632      WRITE_SVLC( scaledWindow.getWindowRightOffset()  >> 1, "scaled_ref_layer_right_offset" );
633      WRITE_SVLC( scaledWindow.getWindowBottomOffset() >> 1, "scaled_ref_layer_bottom_offset" );
634    }
635  }
636}
637#endif
638Void TEncCavlc::codeVPS( TComVPS* pcVPS )
639{
[568]640#if !P0125_REVERT_VPS_EXTN_OFFSET_TO_RESERVED
[494]641#if VPS_EXTN_OFFSET_CALC
642  UInt numBytesInVps = this->m_pcBitIf->getNumberOfWrittenBits();
643#endif
[568]644#endif
[556]645#if !P0307_REMOVE_VPS_VUI_OFFSET
[550]646#if VPS_VUI_OFFSET
647   m_vpsVuiCounter = this->m_pcBitIf->getNumberOfWrittenBits();
648#endif
[556]649#endif
[313]650  WRITE_CODE( pcVPS->getVPSId(),                    4,        "vps_video_parameter_set_id" );
651  WRITE_CODE( 3,                                    2,        "vps_reserved_three_2bits" );
652#if VPS_RENAME
653  WRITE_CODE( pcVPS->getMaxLayers() - 1,            6,        "vps_max_layers_minus1" );           
654#else
655  WRITE_CODE( 0,                                    6,        "vps_reserved_zero_6bits" );
656#endif
657  WRITE_CODE( pcVPS->getMaxTLayers() - 1,           3,        "vps_max_sub_layers_minus1" );
658  WRITE_FLAG( pcVPS->getTemporalNestingFlag(),                "vps_temporal_id_nesting_flag" );
659  assert (pcVPS->getMaxTLayers()>1||pcVPS->getTemporalNestingFlag());
[568]660#if !P0125_REVERT_VPS_EXTN_OFFSET_TO_RESERVED
[442]661#if VPS_EXTN_OFFSET
662  WRITE_CODE( pcVPS->getExtensionOffset(),         16,        "vps_extension_offset" );
663#else
[313]664  WRITE_CODE( 0xffff,                              16,        "vps_reserved_ffff_16bits" );
[442]665#endif
[568]666#else
667  WRITE_CODE( 0xffff,                              16,        "vps_reserved_ffff_16bits" );
668#endif
[313]669  codePTL( pcVPS->getPTL(), true, pcVPS->getMaxTLayers() - 1 );
670  const Bool subLayerOrderingInfoPresentFlag = 1;
671  WRITE_FLAG(subLayerOrderingInfoPresentFlag,              "vps_sub_layer_ordering_info_present_flag");
672  for(UInt i=0; i <= pcVPS->getMaxTLayers()-1; i++)
673  {
674    WRITE_UVLC( pcVPS->getMaxDecPicBuffering(i) - 1,       "vps_max_dec_pic_buffering_minus1[i]" );
675    WRITE_UVLC( pcVPS->getNumReorderPics(i),               "vps_num_reorder_pics[i]" );
676    WRITE_UVLC( pcVPS->getMaxLatencyIncrease(i),           "vps_max_latency_increase_plus1[i]" );
677    if (!subLayerOrderingInfoPresentFlag)
678    {
679      break;
680    }
681  }
682
683#if VPS_RENAME
684  assert( pcVPS->getNumHrdParameters() <= MAX_VPS_LAYER_SETS_PLUS1 );
685  assert( pcVPS->getMaxLayerId() < MAX_VPS_LAYER_ID_PLUS1 );
686#if !VPS_EXTN_OP_LAYER_SETS     // num layer sets set in TAppEncTop.cpp
687  pcVPS->setNumLayerSets(1);
688#endif
689  WRITE_CODE( pcVPS->getMaxLayerId(), 6,                       "vps_max_layer_id" );
690  WRITE_UVLC( pcVPS->getNumLayerSets() - 1,                 "vps_num_layer_sets_minus1" );
691  for( UInt opsIdx = 1; opsIdx <= ( pcVPS->getNumLayerSets() - 1 ); opsIdx ++ )
692  {
693    // Operation point set
694    for( UInt i = 0; i <= pcVPS->getMaxLayerId(); i ++ )
695#else
696  assert( pcVPS->getNumHrdParameters() <= MAX_VPS_NUM_HRD_PARAMETERS );
697  assert( pcVPS->getMaxNuhReservedZeroLayerId() < MAX_VPS_NUH_RESERVED_ZERO_LAYER_ID_PLUS1 );
698  WRITE_CODE( pcVPS->getMaxNuhReservedZeroLayerId(), 6,     "vps_max_nuh_reserved_zero_layer_id" );
699  pcVPS->setMaxOpSets(1);
700  WRITE_UVLC( pcVPS->getMaxOpSets() - 1,                    "vps_max_op_sets_minus1" );
701  for( UInt opsIdx = 1; opsIdx <= ( pcVPS->getMaxOpSets() - 1 ); opsIdx ++ )
702  {
703    // Operation point set
704    for( UInt i = 0; i <= pcVPS->getMaxNuhReservedZeroLayerId(); i ++ )
705#endif
706    {
707#if !VPS_EXTN_OP_LAYER_SETS     // layer Id include flag set in TAppEncTop.cpp
708      // Only applicable for version 1
709      pcVPS->setLayerIdIncludedFlag( true, opsIdx, i );
710#endif
711      WRITE_FLAG( pcVPS->getLayerIdIncludedFlag( opsIdx, i ) ? 1 : 0, "layer_id_included_flag[opsIdx][i]" );
712    }
713  }
714#if DERIVE_LAYER_ID_LIST_VARIABLES
715  pcVPS->deriveLayerIdListVariables();
716#endif
717  TimingInfo *timingInfo = pcVPS->getTimingInfo();
718  WRITE_FLAG(timingInfo->getTimingInfoPresentFlag(),          "vps_timing_info_present_flag");
719  if(timingInfo->getTimingInfoPresentFlag())
720  {
721    WRITE_CODE(timingInfo->getNumUnitsInTick(), 32,           "vps_num_units_in_tick");
722    WRITE_CODE(timingInfo->getTimeScale(),      32,           "vps_time_scale");
723    WRITE_FLAG(timingInfo->getPocProportionalToTimingFlag(),  "vps_poc_proportional_to_timing_flag");
724    if(timingInfo->getPocProportionalToTimingFlag())
725    {
726      WRITE_UVLC(timingInfo->getNumTicksPocDiffOneMinus1(),   "vps_num_ticks_poc_diff_one_minus1");
727    }
728    pcVPS->setNumHrdParameters( 0 );
729    WRITE_UVLC( pcVPS->getNumHrdParameters(),                 "vps_num_hrd_parameters" );
730
731    if( pcVPS->getNumHrdParameters() > 0 )
732    {
733      pcVPS->createHrdParamBuffer();
734    }
735    for( UInt i = 0; i < pcVPS->getNumHrdParameters(); i ++ )
736    {
737      // Only applicable for version 1
738      pcVPS->setHrdOpSetIdx( 0, i );
739      WRITE_UVLC( pcVPS->getHrdOpSetIdx( i ),                "hrd_op_set_idx" );
740      if( i > 0 )
741      {
742        WRITE_FLAG( pcVPS->getCprmsPresentFlag( i ) ? 1 : 0, "cprms_present_flag[i]" );
743      }
744      codeHrdParameters(pcVPS->getHrdParameters(i), pcVPS->getCprmsPresentFlag( i ), pcVPS->getMaxTLayers() - 1);
745    }
746  }
747#if !VPS_EXTNS
748  WRITE_FLAG( 0,                     "vps_extension_flag" );
749#else
750  WRITE_FLAG( 1,                     "vps_extension_flag" );
[442]751  if(1) // Should be conditioned on the value of vps_extension_flag
752  {
753    while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
754    {
755      WRITE_FLAG(1,                  "vps_extension_alignment_bit_equal_to_one");
756    }
[568]757#if !P0125_REVERT_VPS_EXTN_OFFSET_TO_RESERVED
[494]758#if VPS_EXTN_OFFSET_CALC
759    Int vpsExntOffsetValueInBits = this->m_pcBitIf->getNumberOfWrittenBits() - numBytesInVps + 16; // 2 bytes for NUH
760    assert( vpsExntOffsetValueInBits % 8 == 0 );
761    pcVPS->setExtensionOffset( vpsExntOffsetValueInBits >> 3 );
762#endif
[568]763#endif
[442]764    codeVPSExtension(pcVPS);
765    WRITE_FLAG( 0,                     "vps_extension2_flag" );   // Flag value of 1 reserved
766  }
[313]767#endif 
768  //future extensions here..
769 
770  return;
771}
772
[494]773#if SVC_EXTENSION
[313]774#if VPS_EXTNS
775Void TEncCavlc::codeVPSExtension (TComVPS *vps)
776{
777  // ... More syntax elements to be written here
778#if VPS_EXTN_MASK_AND_DIM_INFO
779  UInt i = 0, j = 0;
780
781  WRITE_FLAG( vps->getAvcBaseLayerFlag(),              "avc_base_layer_flag" );
[556]782#if !P0307_REMOVE_VPS_VUI_OFFSET
[550]783#if O0109_MOVE_VPS_VUI_FLAG
784#if !VPS_VUI
785  WRITE_FLAG( 0,                     "vps_vui_present_flag" );
786  vps->setVpsVuiPresentFlag(false);
787#else
788  WRITE_FLAG( 1,                     "vps_vui_present_flag" );
789  vps->setVpsVuiPresentFlag(true);
790#endif
791  if ( vps->getVpsVuiPresentFlag() ) 
792  {
793#if VPS_VUI_OFFSET
794    WRITE_CODE( vps->getVpsVuiOffset(  ), 16,             "vps_vui_offset" );
795#endif
796    WRITE_FLAG( vps->getSplittingFlag(),                 "splitting_flag" );
797  }
798#else
799#if VPS_VUI_OFFSET
800  WRITE_CODE( vps->getVpsVuiOffset(  ), 16,             "vps_vui_offset" ); 
801#endif
[313]802  WRITE_FLAG( vps->getSplittingFlag(),                 "splitting_flag" );
[550]803#endif // O0109_MOVE_VPS_VUI_FLAG
[556]804#endif
805  WRITE_FLAG( vps->getSplittingFlag(),                 "splitting_flag" );
[313]806
807  for(i = 0; i < MAX_VPS_NUM_SCALABILITY_TYPES; i++)
808  {
809    WRITE_FLAG( vps->getScalabilityMask(i),            "scalability_mask[i]" );
810  }
811
812  for(j = 0; j < vps->getNumScalabilityTypes() - vps->getSplittingFlag(); j++)
813  {
814    WRITE_CODE( vps->getDimensionIdLen(j) - 1, 3,      "dimension_id_len_minus1[j]" );
815  }
816
[345]817#if SPL_FLG_CHK
818  if(vps->getSplittingFlag())
819  {
820    UInt splDimSum=0;
821    for(j = 0; j < vps->getNumScalabilityTypes(); j++)
822    {
823      splDimSum+=(vps->getDimensionIdLen(j));
824    }
825    assert(splDimSum<=6);
826  }
827#endif
828
[313]829  WRITE_FLAG( vps->getNuhLayerIdPresentFlag(),         "vps_nuh_layer_id_present_flag" );
830  for(i = 1; i < vps->getMaxLayers(); i++)
831  {
832    if( vps->getNuhLayerIdPresentFlag() )
833    {
834      WRITE_CODE( vps->getLayerIdInNuh(i),     6,      "layer_id_in_nuh[i]" );
835    }
[494]836
837    if( !vps->getSplittingFlag() )
838    {
[313]839    for(j = 0; j < vps->getNumScalabilityTypes(); j++)
840    {
841      UInt bits = vps->getDimensionIdLen(j);
842      WRITE_CODE( vps->getDimensionId(i, j),   bits,   "dimension_id[i][j]" );
843    }
844  }
[494]845  }
[313]846#endif
[442]847#if VIEW_ID_RELATED_SIGNALING
848  // if ( pcVPS->getNumViews() > 1 ) 
849  //   However, this is a bug in the text since, view_id_len_minus1 is needed to parse view_id_val.
850  {
[550]851#if O0109_VIEW_ID_LEN
852    WRITE_CODE( vps->getViewIdLen( ), 4, "view_id_len" );
853    assert ( vps->getNumViews() >= (1<<vps->getViewIdLen()) );
854#else
[442]855    WRITE_CODE( vps->getViewIdLenMinus1( ), 4, "view_id_len_minus1" );
[550]856#endif
[442]857  }
858
[550]859#if O0109_VIEW_ID_LEN
860  if ( vps->getViewIdLen() > 0 )
861  {
862#endif
[442]863  for(  i = 0; i < vps->getNumViews(); i++ )
864  {
[550]865#if O0109_VIEW_ID_LEN
866    WRITE_CODE( vps->getViewIdVal( i ), vps->getViewIdLen( ), "view_id_val[i]" );
867#else
[442]868    WRITE_CODE( vps->getViewIdVal( i ), vps->getViewIdLenMinus1( ) + 1, "view_id_val[i]" );
[550]869#endif
[442]870  }
[550]871#if O0109_VIEW_ID_LEN
872  }
[442]873#endif
[550]874#endif // VIEW_ID_RELATED_SIGNALING
875
[313]876#if VPS_EXTN_DIRECT_REF_LAYERS
877  for( Int layerCtr = 1; layerCtr <= vps->getMaxLayers() - 1; layerCtr++)
878  {
879    for( Int refLayerCtr = 0; refLayerCtr < layerCtr; refLayerCtr++)
880    {
881      WRITE_FLAG(vps->getDirectDependencyFlag(layerCtr, refLayerCtr), "direct_dependency_flag[i][j]" );
882    }
883  }
884#endif
[540]885#if VPS_TSLAYERS
886    WRITE_FLAG( vps->getMaxTSLayersPresentFlag(), "vps_sub_layers_max_minus1_present_flag");
887    if (vps->getMaxTSLayersPresentFlag())
888    {
889        for( i = 0; i < vps->getMaxLayers() - 1; i++)
890        {
891            WRITE_CODE(vps->getMaxTSLayersMinus1(i), 3, "sub_layers_vps_max_minus1[i]" );
892        }
893    }
894#endif
[345]895#if N0120_MAX_TID_REF_PRESENT_FLAG
[442]896   WRITE_FLAG( vps->getMaxTidRefPresentFlag(), "max_tid_ref_present_flag");
897   if (vps->getMaxTidRefPresentFlag())
[345]898   {
899     for( i = 0; i < vps->getMaxLayers() - 1; i++)
900     {
[494]901#if O0225_MAX_TID_FOR_REF_LAYERS
902       for( j = i+1; j <= vps->getMaxLayers() - 1; j++)
903       {
904         if(vps->getDirectDependencyFlag(j, i))
905         {
906           WRITE_CODE(vps->getMaxTidIlRefPicsPlus1(i,j), 3, "max_tid_il_ref_pics_plus1[i][j]" );
907         }
908       }
909#else
[442]910       WRITE_CODE(vps->getMaxTidIlRefPicsPlus1(i), 3, "max_tid_il_ref_pics_plus1[i]" );
[494]911#endif
[345]912     }
913   }
914#else
[313]915  for( i = 0; i < vps->getMaxLayers() - 1; i++)
916  {
[494]917#if O0225_MAX_TID_FOR_REF_LAYERS
918       for( j = i+1; j <= vps->getMaxLayers() - 1; j++)
919       {
920         if(vps->getDirectDependencyFlag(j, i))
921         {
922           WRITE_CODE(vps->getMaxTidIlRefPicsPlus1(i,j), 3, "max_tid_il_ref_pics_plus1[i][j]" );
923         }
924       }
925#else
[442]926    WRITE_CODE(vps->getMaxTidIlRefPicsPlus1(i), 3, "max_tid_il_ref_pics_plus1[i]" );
[494]927#endif
[313]928  }
929#endif
[345]930#if ILP_SSH_SIG
931    WRITE_FLAG( vps->getIlpSshSignalingEnabledFlag(), "all_ref_layers_active_flag" );
932#endif
[313]933#if VPS_EXTN_PROFILE_INFO
934  // Profile-tier-level signalling
[576]935#if !VPS_EXTN_UEV_CODING
[313]936  WRITE_CODE( vps->getNumLayerSets() - 1   , 10, "vps_number_layer_sets_minus1" );     
937  WRITE_CODE( vps->getNumProfileTierLevel() - 1,  6, "vps_num_profile_tier_level_minus1"); 
[576]938#else
939  WRITE_UVLC( vps->getNumProfileTierLevel() - 1, "vps_num_profile_tier_level_minus1"); 
940#endif
[313]941  for(Int idx = 1; idx <= vps->getNumProfileTierLevel() - 1; idx++)
942  {
943    WRITE_FLAG( vps->getProfilePresentFlag(idx),       "vps_profile_present_flag[i]" );
[552]944#if !P0048_REMOVE_PROFILE_REF
[313]945    if( !vps->getProfilePresentFlag(idx) )
946    {
947      WRITE_CODE( vps->getProfileLayerSetRef(idx) - 1, 6, "profile_ref_minus1[i]" );
948    }
[552]949#endif
[313]950    codePTL( vps->getPTLForExtn(idx), vps->getProfilePresentFlag(idx), vps->getMaxTLayers() - 1 );
951  }
952#endif
953
[576]954#if !VPS_EXTN_UEV_CODING
[313]955  Int numOutputLayerSets = vps->getNumOutputLayerSets() ;
956  WRITE_FLAG(  (numOutputLayerSets > vps->getNumLayerSets()), "more_output_layer_sets_than_default_flag" ); 
957  if(numOutputLayerSets > vps->getNumLayerSets())
958  {
959    WRITE_CODE( numOutputLayerSets - vps->getNumLayerSets(), 10, "num_add_output_layer_sets" );
960  }
[576]961#else
962  Int numOutputLayerSets = vps->getNumOutputLayerSets() ;
[581]963  assert( numOutputLayerSets - (Int)vps->getNumLayerSets() >= 0 );
[576]964  WRITE_UVLC( numOutputLayerSets - vps->getNumLayerSets(), "num_add_output_layer_sets" );
965#endif
[313]966  if( numOutputLayerSets > 1 )
967  {
[577]968#if P0295_DEFAULT_OUT_LAYER_IDC
969    WRITE_CODE( vps->getDefaultTargetOutputLayerIdc(), 2, "default_target_output_layer_idc" );   
970#else
[550]971#if O0109_DEFAULT_ONE_OUT_LAYER_IDC
972    WRITE_CODE( vps->getDefaultOneTargetOutputLayerIdc(), 2, "default_one_target_output_layer_idc" );   
973#else
[313]974    WRITE_FLAG( vps->getDefaultOneTargetOutputLayerFlag(), "default_one_target_output_layer_flag" );   
[550]975#endif
[577]976#endif
[313]977  }
978
979  for(i = 1; i < numOutputLayerSets; i++)
980  {
981    if( i > (vps->getNumLayerSets() - 1) )
982    {
983      Int numBits = 1;
984      while ((1 << numBits) < (vps->getNumLayerSets() - 1))
985      {
986        numBits++;
987      }
[577]988      WRITE_CODE( vps->getOutputLayerSetIdx(i) - 1, numBits, "output_layer_set_idx_minus1"); 
989#if P0295_DEFAULT_OUT_LAYER_IDC
990    }
991    if ( i > (vps->getNumLayerSets() - 1) || vps->getDefaultTargetOutputLayerIdc() >= 2 ) //Instead of == 2, >= 2 is used to follow the agreement that value 3 should be interpreted as 2
992    {
993#endif
[313]994      Int lsIdx = vps->getOutputLayerSetIdx(i);
995      for(j = 0; j < vps->getNumLayersInIdList(lsIdx) - 1; j++)
996      {
997        WRITE_FLAG( vps->getOutputLayerFlag(i,j), "output_layer_flag[i][j]");
998      }
999    }
1000    Int numBits = 1;
1001    while ((1 << numBits) < (vps->getNumProfileTierLevel()))
1002    {
1003      numBits++;
1004    }
1005    WRITE_CODE( vps->getProfileLevelTierIdx(i), numBits, "profile_level_tier_idx[i]" );     
1006  }
[442]1007
[540]1008#if O0153_ALT_OUTPUT_LAYER_FLAG
1009  if( vps->getMaxLayers() > 1 )
1010  {
1011    WRITE_FLAG( vps->getAltOuputLayerFlag(), "alt_output_layer_flag" );   
1012  }
1013#endif
1014
[442]1015#if REPN_FORMAT_IN_VPS
1016  WRITE_FLAG( vps->getRepFormatIdxPresentFlag(), "rep_format_idx_present_flag"); 
1017
1018  if( vps->getRepFormatIdxPresentFlag() )
1019  {
[540]1020#if O0096_REP_FORMAT_INDEX
[576]1021#if !VPS_EXTN_UEV_CODING
[540]1022    WRITE_CODE( vps->getVpsNumRepFormats() - 1, 8, "vps_num_rep_formats_minus1" );
1023#else
[576]1024    WRITE_UVLC( vps->getVpsNumRepFormats() - 1, "vps_num_rep_formats_minus1" );
1025#endif
1026#else
[442]1027    WRITE_CODE( vps->getVpsNumRepFormats() - 1, 4, "vps_num_rep_formats_minus1" );
[540]1028#endif
[442]1029  }
1030  for(i = 0; i < vps->getVpsNumRepFormats(); i++)
1031  {
1032    // Read rep_format_structures
1033    codeRepFormat( vps->getVpsRepFormat(i) );
1034  }
1035 
1036  if( vps->getRepFormatIdxPresentFlag() )
1037  {
1038    for(i = 1; i < vps->getMaxLayers(); i++)
1039    {
1040      if( vps->getVpsNumRepFormats() > 1 )
1041      {
[540]1042#if O0096_REP_FORMAT_INDEX
[576]1043#if !VPS_EXTN_UEV_CODING
[540]1044        WRITE_CODE( vps->getVpsRepFormatIdx(i), 8, "vps_rep_format_idx[i]" );
1045#else
[576]1046        Int numBits = 1;
1047        while ((1 << numBits) < (vps->getVpsNumRepFormats()))
1048        {
1049          numBits++;
1050        }
1051        WRITE_CODE( vps->getVpsRepFormatIdx(i), numBits, "vps_rep_format_idx[i]" );
1052#endif
1053#else
[442]1054        WRITE_CODE( vps->getVpsRepFormatIdx(i), 4, "vps_rep_format_idx[i]" );
[540]1055#endif
[442]1056      }
1057    }
1058  }
1059#endif
1060
[564]1061  WRITE_FLAG(vps->getMaxOneActiveRefLayerFlag(), "max_one_active_ref_layer_flag");
[494]1062#if O0062_POC_LSB_NOT_PRESENT_FLAG
1063  for(i = 1; i< vps->getMaxLayers(); i++)
[313]1064  {
[494]1065    if( vps->getNumDirectRefLayers( vps->getLayerIdInNuh(i) ) == 0  )
[313]1066    {
[494]1067      WRITE_FLAG(vps->getPocLsbNotPresentFlag(i), "poc_lsb_not_present_flag[i]");
[313]1068    }
1069  }
1070#endif
[494]1071#if O0215_PHASE_ALIGNMENT
1072  WRITE_FLAG(vps->getPhaseAlignFlag(), "cross_layer_phase_alignment_flag" );
[313]1073#endif
[494]1074#if N0147_IRAP_ALIGN_FLAG && !IRAP_ALIGN_FLAG_IN_VPS_VUI
1075  WRITE_FLAG(vps->getCrossLayerIrapAlignFlag(), "cross_layer_irap_aligned_flag");
1076#endif
[540]1077#if VPS_DPB_SIZE_TABLE
[578]1078  codeVpsDpbSizeTable(vps);
[540]1079#endif
[560]1080#if VPS_EXTN_DIRECT_REF_LAYERS
[313]1081  WRITE_UVLC( vps->getDirectDepTypeLen()-2,                           "direct_dep_type_len_minus2");
[540]1082#if O0096_DEFAULT_DEPENDENCY_TYPE
1083  WRITE_FLAG(vps->getDefaultDirectDependencyTypeFlag(), "default_direct_dependency_flag");
1084  if (vps->getDefaultDirectDependencyTypeFlag())
[313]1085  {
[540]1086    WRITE_CODE( vps->getDefaultDirectDependencyType(), vps->getDirectDepTypeLen(), "default_direct_dependency_type" );
1087  }
1088  else
1089  {
1090    for(i = 1; i < vps->getMaxLayers(); i++)
[313]1091    {
[540]1092      for(j = 0; j < i; j++)
[313]1093      {
[540]1094        if (vps->getDirectDependencyFlag(i, j))
1095        {
1096          WRITE_CODE( vps->getDirectDependencyType(i, j), vps->getDirectDepTypeLen(), "direct_dependency_type[i][j]" );
1097        }
[313]1098      }
1099    }
1100  }
[540]1101#else
[442]1102  for(i = 1; i < vps->getMaxLayers(); i++)
1103  {
1104    for(j = 0; j < i; j++)
1105    {
[540]1106      if (vps->getDirectDependencyFlag(i, j))
1107      {
1108        WRITE_CODE( vps->getDirectDependencyType(i, j), vps->getDirectDepTypeLen(), "direct_dependency_type[i][j]" );
1109      }
[442]1110    }
1111  }
[313]1112#endif
[442]1113#endif
[550]1114
1115#if !O0109_O0199_FLAGS_TO_VUI
[313]1116#if M0040_ADAPTIVE_RESOLUTION_CHANGE
1117  WRITE_FLAG(vps->getSingleLayerForNonIrapFlag(), "single_layer_for_non_irap_flag" );
1118#endif
[540]1119#if HIGHER_LAYER_IRAP_SKIP_FLAG
1120  WRITE_FLAG(vps->getHigherLayerIrapSkipFlag(), "higher_layer_irap_skip_flag" );
1121#endif
[550]1122#endif
[442]1123
[556]1124#if P0307_VPS_NON_VUI_EXTENSION
1125  WRITE_UVLC( vps->getVpsNonVuiExtLength(), "vps_non_vui_extension_length" );
1126  if ( vps->getVpsNonVuiExtLength() > 0 )
1127  {
1128    printf("\n\nUp to the current spec, the value of vps_non_vui_extension_length is supposed to be 0\n");
1129  }
1130#endif
1131
[550]1132#if !O0109_MOVE_VPS_VUI_FLAG
[442]1133#if !VPS_VUI
1134  WRITE_FLAG( 0,                     "vps_vui_present_flag" );
1135#else
1136  WRITE_FLAG( 1,                     "vps_vui_present_flag" );
1137  if(1)   // Should be conditioned on the value of vps_vui_present_flag
1138  {
1139    while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
1140    {
1141      WRITE_FLAG(1,                  "vps_vui_alignment_bit_equal_to_one");
1142    }
[550]1143#if VPS_VUI_OFFSET
1144    Int vpsVuiOffsetValeInBits = this->m_pcBitIf->getNumberOfWrittenBits() - m_vpsVuiCounter + 16; // 2 bytes for NUH
1145    assert( vpsVuiOffsetValeInBits % 8 == 0 );
1146    vps->setVpsVuiOffset( vpsVuiOffsetValeInBits >> 3 );
1147#endif
[442]1148    codeVPSVUI(vps); 
1149  }
1150#endif
[550]1151#else
[556]1152#if P0307_REMOVE_VPS_VUI_OFFSET
1153  WRITE_FLAG( 1,                     "vps_vui_present_flag" );
1154  vps->setVpsVuiPresentFlag(true);
1155#endif
[550]1156  if(vps->getVpsVuiPresentFlag())   // Should be conditioned on the value of vps_vui_present_flag
1157  {
1158    while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
1159    {
1160      WRITE_FLAG(1,                  "vps_vui_alignment_bit_equal_to_one");
1161    }
[556]1162#if !P0307_REMOVE_VPS_VUI_OFFSET
[550]1163#if VPS_VUI_OFFSET
1164    Int vpsVuiOffsetValeInBits = this->m_pcBitIf->getNumberOfWrittenBits() - m_vpsVuiCounter + 16; // 2 bytes for NUH
1165    assert( vpsVuiOffsetValeInBits % 8 == 0 );
1166    vps->setVpsVuiOffset( vpsVuiOffsetValeInBits >> 3 );
1167#endif
[556]1168#endif
[550]1169    codeVPSVUI(vps); 
1170  }
1171#endif // 0109_MOVE_VPS_FLAG
[313]1172}
1173#endif
[442]1174#if REPN_FORMAT_IN_VPS
1175Void  TEncCavlc::codeRepFormat      ( RepFormat *repFormat )
1176{
[540]1177#if REPN_FORMAT_CONTROL_FLAG
1178   WRITE_FLAG ( repFormat->getChromaAndBitDepthVpsPresentFlag(), "chroma_and_bit_depth_vps_presenet_flag"); 
1179
1180   WRITE_CODE ( repFormat->getPicWidthVpsInLumaSamples (), 16, "pic_width_in_luma_samples" );   
1181   WRITE_CODE ( repFormat->getPicHeightVpsInLumaSamples(), 16, "pic_height_in_luma_samples" ); 
1182
1183   if ( repFormat->getChromaAndBitDepthVpsPresentFlag() )
1184   {
1185     WRITE_CODE( repFormat->getChromaFormatVpsIdc(), 2, "chroma_format_idc" );   
1186
1187     if( repFormat->getChromaFormatVpsIdc() == 3 )
1188     {
1189       WRITE_FLAG( repFormat->getSeparateColourPlaneVpsFlag(), "separate_colour_plane_flag");     
1190     }
1191
1192     assert( repFormat->getBitDepthVpsLuma() >= 8 );
1193     assert( repFormat->getBitDepthVpsChroma() >= 8 );
1194     WRITE_CODE( repFormat->getBitDepthVpsLuma() - 8,   4, "bit_depth_luma_minus8" );           
1195     WRITE_CODE( repFormat->getBitDepthVpsChroma() - 8, 4, "bit_depth_chroma_minus8" );
1196   }
1197#else
[442]1198  WRITE_CODE( repFormat->getChromaFormatVpsIdc(), 2, "chroma_format_idc" );   
1199 
1200  if( repFormat->getChromaFormatVpsIdc() == 3 )
1201  {
1202    WRITE_FLAG( repFormat->getSeparateColourPlaneVpsFlag(), "separate_colour_plane_flag");     
1203  }
[313]1204
[442]1205  WRITE_CODE ( repFormat->getPicWidthVpsInLumaSamples (), 16, "pic_width_in_luma_samples" );   
1206  WRITE_CODE ( repFormat->getPicHeightVpsInLumaSamples(), 16, "pic_height_in_luma_samples" );   
1207 
1208  assert( repFormat->getBitDepthVpsLuma() >= 8 );
1209  assert( repFormat->getBitDepthVpsChroma() >= 8 );
1210  WRITE_CODE( repFormat->getBitDepthVpsLuma() - 8,   4, "bit_depth_luma_minus8" );           
[540]1211  WRITE_CODE( repFormat->getBitDepthVpsChroma() - 8, 4, "bit_depth_chroma_minus8" );
1212#endif
[442]1213
1214}
1215#endif
[578]1216#if VPS_DPB_SIZE_TABLE
1217Void TEncCavlc::codeVpsDpbSizeTable(TComVPS *vps)
1218{
1219  for(Int i = 1; i < vps->getNumOutputLayerSets(); i++)
1220  {
[579]1221#if CHANGE_NUMSUBDPB_IDX
1222    Int layerSetIdxForOutputLayerSet = vps->getOutputLayerSetIdx( i );
1223#endif
[578]1224    WRITE_FLAG( vps->getSubLayerFlagInfoPresentFlag( i ), "sub_layer_flag_info_present_flag[i]"); 
1225    for(Int j = 0; j < vps->getMaxTLayers(); j++)
1226    {
1227      if( j > 0 && vps->getSubLayerFlagInfoPresentFlag(i) )
1228      {
1229        WRITE_FLAG( vps->getSubLayerDpbInfoPresentFlag( i, j), "sub_layer_dpb_info_present_flag[i]"); 
1230      }
1231      if( vps->getSubLayerDpbInfoPresentFlag(i, j) )
1232      {
[579]1233#if CHANGE_NUMSUBDPB_IDX
1234        for(Int k = 0; k < vps->getNumSubDpbs(layerSetIdxForOutputLayerSet); k++)
1235#else
[578]1236        for(Int k = 0; k < vps->getNumSubDpbs(i); k++)
[579]1237#endif
[578]1238        {
1239          WRITE_UVLC( vps->getMaxVpsDecPicBufferingMinus1( i, k, j), "max_vps_dec_pic_buffering_minus1[i][k][j]" ); 
1240        }
1241        WRITE_UVLC( vps->getMaxVpsNumReorderPics( i, j), "max_vps_num_reorder_pics[i][j]" );             
[582]1242#if RESOLUTION_BASED_DPB
1243        if( vps->getNumSubDpbs(layerSetIdxForOutputLayerSet) != vps->getNumLayersInIdList( layerSetIdxForOutputLayerSet ) )  // NumSubDpbs
1244        {
1245          for(Int k = 0; k < vps->getNumLayersInIdList( layerSetIdxForOutputLayerSet ); k++)
1246          {
1247            WRITE_UVLC( vps->getMaxVpsLayerDecPicBuffMinus1( i, k, j), "max_vps_layer_dec_pic_buff_minus1[i][k][j]" );
1248          }
1249        }
1250#endif
[578]1251        WRITE_UVLC( vps->getMaxVpsLatencyIncreasePlus1( i, j), "max_vps_latency_increase_plus1[i][j]" );       
1252      }
1253    }
1254  }
1255}
1256#endif
[442]1257#if VPS_VUI
1258Void TEncCavlc::codeVPSVUI (TComVPS *vps)
1259{
1260  Int i,j;
[540]1261#if O0223_PICTURE_TYPES_ALIGN_FLAG
1262  WRITE_FLAG(vps->getCrossLayerPictureTypeAlignFlag(), "cross_layer_pic_type_aligned_flag");
1263  if (!vps->getCrossLayerPictureTypeAlignFlag())
1264  {
1265#endif
[494]1266#if IRAP_ALIGN_FLAG_IN_VPS_VUI
[540]1267    WRITE_FLAG(vps->getCrossLayerIrapAlignFlag(), "cross_layer_irap_aligned_flag");
[494]1268#endif
[540]1269#if O0223_PICTURE_TYPES_ALIGN_FLAG
1270  }
1271#endif
[442]1272#if VPS_VUI_BITRATE_PICRATE
1273  WRITE_FLAG( vps->getBitRatePresentVpsFlag(),        "bit_rate_present_vps_flag" );
1274  WRITE_FLAG( vps->getPicRatePresentVpsFlag(),        "pic_rate_present_vps_flag" );
1275
1276  if( vps->getBitRatePresentVpsFlag() || vps->getPicRatePresentVpsFlag() )
1277  {
1278    for( i = 0; i < vps->getNumLayerSets(); i++ )
1279    {
1280      for( j = 0; j < vps->getMaxTLayers(); j++ )
1281      {
1282        if( vps->getBitRatePresentVpsFlag() )
1283        {
1284          WRITE_FLAG( vps->getBitRatePresentFlag( i, j),        "bit_rate_present_vps_flag[i][j]" );
1285        }
1286        if( vps->getPicRatePresentVpsFlag() )
1287        {
1288          WRITE_FLAG( vps->getPicRatePresentFlag( i, j),        "pic_rate_present_vps_flag[i][j]" );
1289        }
1290        if( vps->getBitRatePresentFlag(i, j) )
1291        {
1292          WRITE_CODE( vps->getAvgBitRate( i, j ), 16, "avg_bit_rate[i][j]" );
1293          WRITE_CODE( vps->getAvgBitRate( i, j ), 16, "max_bit_rate[i][j]" );
1294        }
1295        if( vps->getPicRatePresentFlag(i, j) )
1296        {
1297          WRITE_CODE( vps->getConstPicRateIdc( i, j), 2 , "constant_pic_rate_idc[i][j]" ); 
1298          WRITE_CODE( vps->getConstPicRateIdc( i, j), 16, "avg_pic_rate[i][j]"          ); 
1299        }
1300      }
1301    }
1302  }
1303#endif
[570]1304#if VPS_VUI_VIDEO_SIGNAL_MOVE
1305  WRITE_FLAG( vps->getVideoSigPresentVpsFlag(), "video_signal_info_idx_present_flag" );
1306  if (vps->getVideoSigPresentVpsFlag())
1307  {
1308    WRITE_CODE(vps->getNumVideoSignalInfo()-1, 4, "vps_num_video_signal_info_minus1" );
1309  }
1310
1311  for(i = 0; i < vps->getNumVideoSignalInfo(); i++)
1312  {
1313    WRITE_CODE(vps->getVideoVPSFormat(i), 3, "video_vps_format" );
1314    WRITE_FLAG(vps->getVideoFullRangeVpsFlag(i), "video_full_range_vps_flag" );
1315    WRITE_CODE(vps->getColorPrimaries(i), 8, "color_primaries_vps" );
1316    WRITE_CODE(vps->getTransCharacter(i), 8, "transfer_characteristics_vps" );
1317    WRITE_CODE(vps->getMaxtrixCoeff(i), 8, "matrix_coeffs_vps" );
1318  }
1319
1320  if (vps->getVideoSigPresentVpsFlag() && vps->getNumVideoSignalInfo() > 1 )
1321  {
1322    for (i=1; i < vps->getMaxLayers(); i++)
1323      WRITE_CODE(vps->getVideoSignalInfoIdx(i), 4, "vps_video_signal_info_idx" );
1324  }
1325#endif
[540]1326#if VPS_VUI_TILES_NOT_IN_USE__FLAG
1327  UInt layerIdx;
1328  WRITE_FLAG( vps->getTilesNotInUseFlag() ? 1 : 0 , "tiles_not_in_use_flag" );
1329  if (!vps->getTilesNotInUseFlag())
1330  {
1331    for(i = 0; i < vps->getMaxLayers(); i++)
1332    {
1333      WRITE_FLAG( vps->getTilesInUseFlag(i) ? 1 : 0 , "tiles_in_use_flag[ i ]" );
1334      if (vps->getTilesInUseFlag(i))
1335      {
1336        WRITE_FLAG( vps->getLoopFilterNotAcrossTilesFlag(i) ? 1 : 0 , "loop_filter_not_across_tiles_flag[ i ]" );
1337      }
1338    }
1339#endif
[442]1340#if TILE_BOUNDARY_ALIGNED_FLAG
[540]1341    for(i = 1; i < vps->getMaxLayers(); i++)
1342    {
1343      for(j = 0; j < vps->getNumDirectRefLayers(vps->getLayerIdInNuh(i)); j++)
1344      {
1345#if VPS_VUI_TILES_NOT_IN_USE__FLAG
1346        layerIdx = vps->getLayerIdInVps(vps->getRefLayerId(vps->getLayerIdInNuh(i), j));
1347        if (vps->getTilesInUseFlag(i) && vps->getTilesInUseFlag(layerIdx)) {
1348          WRITE_FLAG( vps->getTileBoundariesAlignedFlag(i,j) ? 1 : 0 , "tile_boundaries_aligned_flag[i][j]" );
1349        }
1350#else
1351        WRITE_FLAG( vps->getTileBoundariesAlignedFlag(i,j) ? 1 : 0 , "tile_boundaries_aligned_flag[i][j]" );
1352#endif
1353      }
1354    } 
1355#endif
1356#if VPS_VUI_TILES_NOT_IN_USE__FLAG
1357  }
1358#endif
1359#if VPS_VUI_WPP_NOT_IN_USE__FLAG
1360  WRITE_FLAG( vps->getWppNotInUseFlag() ? 1 : 0 , "wpp_not_in_use_flag" );
1361  if (!vps->getWppNotInUseFlag())
[442]1362  {
[540]1363    for(i = 0; i < vps->getMaxLayers(); i++)
[442]1364    {
[540]1365      WRITE_FLAG( vps->getWppInUseFlag(i) ? 1 : 0 , "wpp_in_use_flag[ i ]" );
[442]1366    }
[540]1367  }
1368#endif
[550]1369
1370#if O0109_O0199_FLAGS_TO_VUI
1371#if M0040_ADAPTIVE_RESOLUTION_CHANGE
1372  WRITE_FLAG(vps->getSingleLayerForNonIrapFlag(), "single_layer_for_non_irap_flag" );
1373#endif
1374#if HIGHER_LAYER_IRAP_SKIP_FLAG
1375  WRITE_FLAG(vps->getHigherLayerIrapSkipFlag(), "higher_layer_irap_skip_flag" );
1376#endif
1377#endif
[442]1378#if N0160_VUI_EXT_ILP_REF
[551]1379  WRITE_FLAG( vps->getIlpRestrictedRefLayersFlag() ? 1 : 0 , "ilp_restricted_ref_layers_flag" );   
1380  if( vps->getIlpRestrictedRefLayersFlag())
[442]1381  {
1382    for(i = 1; i < vps->getMaxLayers(); i++)
1383    {
1384      for(j = 0; j < vps->getNumDirectRefLayers(vps->getLayerIdInNuh(i)); j++)
1385      {       
1386        WRITE_UVLC(vps->getMinSpatialSegmentOffsetPlus1( i, j),    "min_spatial_segment_offset_plus1[i][j]");
1387       
1388        if( vps->getMinSpatialSegmentOffsetPlus1(i,j ) > 0 ) 
1389        { 
1390          WRITE_FLAG( vps->getCtuBasedOffsetEnabledFlag( i, j) ? 1 : 0 , "ctu_based_offset_enabled_flag[i][j]" );   
1391         
1392          if(vps->getCtuBasedOffsetEnabledFlag(i,j)) 
1393          {
1394            WRITE_UVLC(vps->getMinHorizontalCtuOffsetPlus1( i, j),    "min_horizontal_ctu_offset_plus1[i][j]");           
1395          }
1396        } 
1397      } 
1398    }
1399  }
1400#endif
[540]1401#if VPS_VUI_VIDEO_SIGNAL
[570]1402#if VPS_VUI_VIDEO_SIGNAL_MOVE
1403#else
[540]1404    WRITE_FLAG( vps->getVideoSigPresentVpsFlag(), "video_signal_info_idx_present_flag" );
1405    if (vps->getVideoSigPresentVpsFlag())
1406    {
1407        WRITE_CODE(vps->getNumVideoSignalInfo()-1, 4, "vps_num_video_signal_info_minus1" );
1408    }
1409   
1410    for(i = 0; i < vps->getNumVideoSignalInfo(); i++)
1411    {
1412        WRITE_CODE(vps->getVideoVPSFormat(i), 3, "video_vps_format" );
1413        WRITE_FLAG(vps->getVideoFullRangeVpsFlag(i), "video_full_range_vps_flag" );
1414        WRITE_CODE(vps->getColorPrimaries(i), 8, "color_primaries_vps" );
1415        WRITE_CODE(vps->getTransCharacter(i), 8, "transfer_characteristics_vps" );
1416        WRITE_CODE(vps->getMaxtrixCoeff(i), 8, "matrix_coeffs_vps" );
1417    }
1418   
1419    if (vps->getVideoSigPresentVpsFlag() && vps->getNumVideoSignalInfo() > 1 )
1420    {
1421        for (i=1; i < vps->getMaxLayers(); i++)
1422            WRITE_CODE(vps->getVideoSignalInfoIdx(i), 4, "vps_video_signal_info_idx" );
1423    }
[570]1424#endif
[540]1425#endif
[442]1426}
1427#endif
[494]1428#endif //SVC_EXTENSION
[442]1429
[313]1430Void TEncCavlc::codeSliceHeader         ( TComSlice* pcSlice )
1431{
1432#if ENC_DEC_TRACE 
1433  xTraceSliceHeader (pcSlice);
1434#endif
1435
1436  //calculate number of bits required for slice address
1437  Int maxSliceSegmentAddress = pcSlice->getPic()->getNumCUsInFrame();
1438  Int bitsSliceSegmentAddress = 0;
1439  while(maxSliceSegmentAddress>(1<<bitsSliceSegmentAddress)) 
1440  {
1441    bitsSliceSegmentAddress++;
1442  }
1443  Int ctuAddress;
1444  if (pcSlice->isNextSlice())
1445  {
1446    // Calculate slice address
1447    ctuAddress = (pcSlice->getSliceCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU());
1448  }
1449  else
1450  {
1451    // Calculate slice address
1452    ctuAddress = (pcSlice->getSliceSegmentCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU());
1453  }
1454  //write slice address
1455  Int sliceSegmentAddress = pcSlice->getPic()->getPicSym()->getCUOrderMap(ctuAddress);
1456
1457  WRITE_FLAG( sliceSegmentAddress==0, "first_slice_segment_in_pic_flag" );
1458  if ( pcSlice->getRapPicFlag() )
1459  {
1460    WRITE_FLAG( 0, "no_output_of_prior_pics_flag" );
1461  }
1462  WRITE_UVLC( pcSlice->getPPS()->getPPSId(), "slice_pic_parameter_set_id" );
1463  pcSlice->setDependentSliceSegmentFlag(!pcSlice->isNextSlice());
1464  if ( pcSlice->getPPS()->getDependentSliceSegmentsEnabledFlag() && (sliceSegmentAddress!=0) )
1465  {
1466    WRITE_FLAG( pcSlice->getDependentSliceSegmentFlag() ? 1 : 0, "dependent_slice_segment_flag" );
1467  }
1468  if(sliceSegmentAddress>0)
1469  {
1470    WRITE_CODE( sliceSegmentAddress, bitsSliceSegmentAddress, "slice_segment_address" );
1471  }
1472  if ( !pcSlice->getDependentSliceSegmentFlag() )
1473  {
[494]1474#if SVC_EXTENSION
[442]1475#if POC_RESET_FLAG
1476    Int iBits = 0;
1477    if( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > iBits )
1478    {
1479      WRITE_FLAG( pcSlice->getPocResetFlag(), "poc_reset_flag" );
1480      iBits++;
1481    }
1482    if( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > iBits )
1483    {
1484      assert(!!"discardable_flag");
1485      WRITE_FLAG(pcSlice->getDiscardableFlag(), "discardable_flag");
1486      iBits++;
1487    }
[540]1488#if O0149_CROSS_LAYER_BLA_FLAG
1489    if( pcSlice->getPPS()->getNumExtraSliceHeaderBits() > iBits )
1490    {
1491      assert(!!"cross_layer_bla_flag");
1492      WRITE_FLAG(pcSlice->getCrossLayerBLAFlag(), "cross_layer_bla_flag");
1493      iBits++;
1494    }
1495#endif
[442]1496    for ( ; iBits < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); iBits++)
1497    {
1498      assert(!!"slice_reserved_undetermined_flag[]");
1499      WRITE_FLAG(0, "slice_reserved_undetermined_flag[]");
1500    }
1501#else
[313]1502    if (pcSlice->getPPS()->getNumExtraSliceHeaderBits()>0)
1503    {
1504      assert(!!"discardable_flag");
1505      WRITE_FLAG(pcSlice->getDiscardableFlag(), "discardable_flag");
1506    }
1507    for (Int i = 1; i < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); i++)
1508    {
1509      assert(!!"slice_reserved_undetermined_flag[]");
1510      WRITE_FLAG(0, "slice_reserved_undetermined_flag[]");
1511    }
[494]1512#endif
1513#else //SVC_EXTENSION
[313]1514    for (Int i = 0; i < pcSlice->getPPS()->getNumExtraSliceHeaderBits(); i++)
1515    {
1516      assert(!!"slice_reserved_undetermined_flag[]");
1517      WRITE_FLAG(0, "slice_reserved_undetermined_flag[]");
1518    }
[494]1519#endif //SVC_EXTENSION
[313]1520
1521    WRITE_UVLC( pcSlice->getSliceType(),       "slice_type" );
1522
1523    if( pcSlice->getPPS()->getOutputFlagPresentFlag() )
1524    {
1525      WRITE_FLAG( pcSlice->getPicOutputFlag() ? 1 : 0, "pic_output_flag" );
1526    }
1527
[494]1528#if !AUXILIARY_PICTURES
[442]1529#if REPN_FORMAT_IN_VPS
[313]1530    // in the first version chroma_format_idc is equal to one, thus colour_plane_id will not be present
[442]1531    assert( pcSlice->getChromaFormatIdc() == 1 );
1532#else
1533    // in the first version chroma_format_idc is equal to one, thus colour_plane_id will not be present
[313]1534    assert (pcSlice->getSPS()->getChromaFormatIdc() == 1 );
[442]1535#endif
[494]1536#endif
[313]1537    // if( separate_colour_plane_flag  ==  1 )
1538    //   colour_plane_id                                      u(2)
1539
[442]1540#if N0065_LAYER_POC_ALIGNMENT
[494]1541#if O0062_POC_LSB_NOT_PRESENT_FLAG
1542    if( (pcSlice->getLayerId() > 0 && !pcSlice->getVPS()->getPocLsbNotPresentFlag( pcSlice->getVPS()->getLayerIdInVps(pcSlice->getLayerId())) ) || !pcSlice->getIdrPicFlag())
1543#else
[442]1544    if( pcSlice->getLayerId() > 0 || !pcSlice->getIdrPicFlag() )
[494]1545#endif
[442]1546#else
[313]1547    if( !pcSlice->getIdrPicFlag() )
[442]1548#endif
[313]1549    {
[442]1550#if POC_RESET_FLAG
1551      Int picOrderCntLSB;
1552      if( !pcSlice->getPocResetFlag() )
1553      {
1554        picOrderCntLSB = (pcSlice->getPOC()-pcSlice->getLastIDR()+(1<<pcSlice->getSPS()->getBitsForPOC())) & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
1555      }
1556      else
1557      {
1558        picOrderCntLSB = (pcSlice->getPocValueBeforeReset()-pcSlice->getLastIDR()+(1<<pcSlice->getSPS()->getBitsForPOC())) & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
1559      }
1560#else
1561      Int picOrderCntLSB = (pcSlice->getPOC()-pcSlice->getLastIDR()+(1<<pcSlice->getSPS()->getBitsForPOC())) & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
1562#endif
[313]1563      WRITE_CODE( picOrderCntLSB, pcSlice->getSPS()->getBitsForPOC(), "pic_order_cnt_lsb");
[442]1564
1565#if N0065_LAYER_POC_ALIGNMENT
[494]1566#if SHM_FIX7
1567    }
1568#endif
[442]1569      if( !pcSlice->getIdrPicFlag() )
1570      {
1571#endif
[313]1572      TComReferencePictureSet* rps = pcSlice->getRPS();
1573     
1574      // check for bitstream restriction stating that:
1575      // If the current picture is a BLA or CRA picture, the value of NumPocTotalCurr shall be equal to 0.
1576      // Ideally this process should not be repeated for each slice in a picture
1577#if SVC_EXTENSION
1578      if( pcSlice->getLayerId() == 0 )
1579#endif
1580      if (pcSlice->isIRAP())
1581      {
1582        for (Int picIdx = 0; picIdx < rps->getNumberOfPictures(); picIdx++)
1583        {
1584          assert (!rps->getUsed(picIdx));
1585        }
1586      }
1587
1588      if(pcSlice->getRPSidx() < 0)
1589      {
1590        WRITE_FLAG( 0, "short_term_ref_pic_set_sps_flag");
1591        codeShortTermRefPicSet(pcSlice->getSPS(), rps, true, pcSlice->getSPS()->getRPSList()->getNumberOfReferencePictureSets());
1592      }
1593      else
1594      {
1595        WRITE_FLAG( 1, "short_term_ref_pic_set_sps_flag");
1596        Int numBits = 0;
1597        while ((1 << numBits) < pcSlice->getSPS()->getRPSList()->getNumberOfReferencePictureSets())
1598        {
1599          numBits++;
1600        }
1601        if (numBits > 0)
1602        {
1603          WRITE_CODE( pcSlice->getRPSidx(), numBits, "short_term_ref_pic_set_idx" );         
1604        }
1605      }
1606      if(pcSlice->getSPS()->getLongTermRefsPresent())
1607      {
1608        Int numLtrpInSH = rps->getNumberOfLongtermPictures();
1609        Int ltrpInSPS[MAX_NUM_REF_PICS];
1610        Int numLtrpInSPS = 0;
1611        UInt ltrpIndex;
1612        Int counter = 0;
1613        for(Int k = rps->getNumberOfPictures()-1; k > rps->getNumberOfPictures()-rps->getNumberOfLongtermPictures()-1; k--) 
1614        {
1615          if (findMatchingLTRP(pcSlice, &ltrpIndex, rps->getPOC(k), rps->getUsed(k))) 
1616          {
1617            ltrpInSPS[numLtrpInSPS] = ltrpIndex;
1618            numLtrpInSPS++;
1619          }
1620          else
1621          {
1622            counter++;
1623          }
1624        }
1625        numLtrpInSH -= numLtrpInSPS;
1626
1627        Int bitsForLtrpInSPS = 0;
1628        while (pcSlice->getSPS()->getNumLongTermRefPicSPS() > (1 << bitsForLtrpInSPS))
1629        {
1630          bitsForLtrpInSPS++;
1631        }
1632        if (pcSlice->getSPS()->getNumLongTermRefPicSPS() > 0) 
1633        {
1634          WRITE_UVLC( numLtrpInSPS, "num_long_term_sps");
1635        }
1636        WRITE_UVLC( numLtrpInSH, "num_long_term_pics");
1637        // Note that the LSBs of the LT ref. pic. POCs must be sorted before.
1638        // Not sorted here because LT ref indices will be used in setRefPicList()
1639        Int prevDeltaMSB = 0, prevLSB = 0;
1640        Int offset = rps->getNumberOfNegativePictures() + rps->getNumberOfPositivePictures();
1641        for(Int i=rps->getNumberOfPictures()-1 ; i > offset-1; i--)
1642        {
1643          if (counter < numLtrpInSPS)
1644          {
1645            if (bitsForLtrpInSPS > 0)
1646            {
1647              WRITE_CODE( ltrpInSPS[counter], bitsForLtrpInSPS, "lt_idx_sps[i]");             
1648            }
1649          }
1650          else 
1651          {
1652            WRITE_CODE( rps->getPocLSBLT(i), pcSlice->getSPS()->getBitsForPOC(), "poc_lsb_lt");
1653            WRITE_FLAG( rps->getUsed(i), "used_by_curr_pic_lt_flag"); 
1654          }
1655          WRITE_FLAG( rps->getDeltaPocMSBPresentFlag(i), "delta_poc_msb_present_flag");
1656
1657          if(rps->getDeltaPocMSBPresentFlag(i))
1658          {
1659            Bool deltaFlag = false;
1660            //  First LTRP from SPS                 ||  First LTRP from SH                              || curr LSB            != prev LSB
1661            if( (i == rps->getNumberOfPictures()-1) || (i == rps->getNumberOfPictures()-1-numLtrpInSPS) || (rps->getPocLSBLT(i) != prevLSB) )
1662            {
1663              deltaFlag = true;
1664            }
1665            if(deltaFlag)
1666            {
1667              WRITE_UVLC( rps->getDeltaPocMSBCycleLT(i), "delta_poc_msb_cycle_lt[i]" );
1668            }
1669            else
1670            {             
1671              Int differenceInDeltaMSB = rps->getDeltaPocMSBCycleLT(i) - prevDeltaMSB;
1672              assert(differenceInDeltaMSB >= 0);
1673              WRITE_UVLC( differenceInDeltaMSB, "delta_poc_msb_cycle_lt[i]" );
1674            }
1675            prevLSB = rps->getPocLSBLT(i);
1676            prevDeltaMSB = rps->getDeltaPocMSBCycleLT(i);
1677          }
1678        }
1679      }
1680      if (pcSlice->getSPS()->getTMVPFlagsPresent())
1681      {
1682        WRITE_FLAG( pcSlice->getEnableTMVPFlag() ? 1 : 0, "slice_temporal_mvp_enable_flag" );
1683      }
[494]1684#if N0065_LAYER_POC_ALIGNMENT && !SHM_FIX7
[442]1685      }
1686#endif
[313]1687    }
1688
[564]1689#if SVC_EXTENSION
[345]1690#if ILP_SSH_SIG
[494]1691#if ILP_SSH_SIG_FIX
1692    if((pcSlice->getSPS()->getLayerId() > 0) && !(pcSlice->getVPS()->getIlpSshSignalingEnabledFlag()) && (pcSlice->getNumILRRefIdx() > 0) )
1693#else
[345]1694    if((pcSlice->getSPS()->getLayerId() > 0) && pcSlice->getVPS()->getIlpSshSignalingEnabledFlag() && (pcSlice->getNumILRRefIdx() > 0) )
[494]1695#endif
[345]1696#else
1697    if((pcSlice->getSPS()->getLayerId() > 0)  &&  (pcSlice->getNumILRRefIdx() > 0) )
1698#endif
[313]1699    {
1700      WRITE_FLAG(pcSlice->getInterLayerPredEnabledFlag(),"inter_layer_pred_enabled_flag");
1701      if( pcSlice->getInterLayerPredEnabledFlag())
1702      {
1703        if(pcSlice->getNumILRRefIdx() > 1)
1704        {
1705          Int numBits = 1;
1706          while ((1 << numBits) < pcSlice->getNumILRRefIdx())
1707          {
1708            numBits++;
1709          }
1710          if( !pcSlice->getVPS()->getMaxOneActiveRefLayerFlag()) 
1711          {
1712            WRITE_CODE(pcSlice->getActiveNumILRRefIdx() - 1, numBits,"num_inter_layer_ref_pics_minus1");
1713          }       
[345]1714#if ILP_NUM_REF_CHK
1715          if( pcSlice->getNumILRRefIdx() != pcSlice->getActiveNumILRRefIdx() )
1716          {
1717#endif
[313]1718          for(Int i = 0; i < pcSlice->getActiveNumILRRefIdx(); i++ )
1719          {
1720            WRITE_CODE(pcSlice->getInterLayerPredLayerIdc(i),numBits,"inter_layer_pred_layer_idc[i]");   
1721          }
[345]1722#if ILP_NUM_REF_CHK
1723          }
1724#endif
[313]1725        }
1726      }
1727    }     
[564]1728#endif //SVC_EXTENSION
[313]1729
1730    if(pcSlice->getSPS()->getUseSAO())
1731    {
1732      if (pcSlice->getSPS()->getUseSAO())
1733      {
[540]1734         WRITE_FLAG( pcSlice->getSaoEnabledFlag(), "slice_sao_luma_flag" );         
1735#if AUXILIARY_PICTURES
[593]1736         if (pcSlice->getChromaFormatIdc() != CHROMA_400)         
[494]1737#endif
[540]1738         WRITE_FLAG( pcSlice->getSaoEnabledFlagChroma(), "slice_sao_chroma_flag" );
[313]1739      }
[540]1740    }   
[313]1741
1742    //check if numrefidxes match the defaults. If not, override
1743
1744    if (!pcSlice->isIntra())
1745    {
1746      Bool overrideFlag = (pcSlice->getNumRefIdx( REF_PIC_LIST_0 )!=pcSlice->getPPS()->getNumRefIdxL0DefaultActive()||(pcSlice->isInterB()&&pcSlice->getNumRefIdx( REF_PIC_LIST_1 )!=pcSlice->getPPS()->getNumRefIdxL1DefaultActive()));
1747      WRITE_FLAG( overrideFlag ? 1 : 0,                               "num_ref_idx_active_override_flag");
1748      if (overrideFlag) 
1749      {
1750        WRITE_UVLC( pcSlice->getNumRefIdx( REF_PIC_LIST_0 ) - 1,      "num_ref_idx_l0_active_minus1" );
1751        if (pcSlice->isInterB())
1752        {
1753          WRITE_UVLC( pcSlice->getNumRefIdx( REF_PIC_LIST_1 ) - 1,    "num_ref_idx_l1_active_minus1" );
1754        }
1755        else
1756        {
1757          pcSlice->setNumRefIdx(REF_PIC_LIST_1, 0);
1758        }
1759      }
1760    }
1761    else
1762    {
1763      pcSlice->setNumRefIdx(REF_PIC_LIST_0, 0);
1764      pcSlice->setNumRefIdx(REF_PIC_LIST_1, 0);
1765    }
1766
1767    if( pcSlice->getPPS()->getListsModificationPresentFlag() && pcSlice->getNumRpsCurrTempList() > 1)
1768    {
1769      TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
1770      if(!pcSlice->isIntra())
1771      {
1772        WRITE_FLAG(pcSlice->getRefPicListModification()->getRefPicListModificationFlagL0() ? 1 : 0,       "ref_pic_list_modification_flag_l0" );
1773        if (pcSlice->getRefPicListModification()->getRefPicListModificationFlagL0())
1774        {
1775          Int numRpsCurrTempList0 = pcSlice->getNumRpsCurrTempList();
1776          if (numRpsCurrTempList0 > 1)
1777          {
1778            Int length = 1;
1779            numRpsCurrTempList0 --;
1780            while ( numRpsCurrTempList0 >>= 1) 
1781            {
1782              length ++;
1783            }
1784            for(Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_0 ); i++)
1785            {
1786              WRITE_CODE( refPicListModification->getRefPicSetIdxL0(i), length, "list_entry_l0");
1787            }
1788          }
1789        }
1790      }
1791      if(pcSlice->isInterB())
1792      {   
1793        WRITE_FLAG(pcSlice->getRefPicListModification()->getRefPicListModificationFlagL1() ? 1 : 0,       "ref_pic_list_modification_flag_l1" );
1794        if (pcSlice->getRefPicListModification()->getRefPicListModificationFlagL1())
1795        {
1796          Int numRpsCurrTempList1 = pcSlice->getNumRpsCurrTempList();
1797          if ( numRpsCurrTempList1 > 1 )
1798          {
1799            Int length = 1;
1800            numRpsCurrTempList1 --;
1801            while ( numRpsCurrTempList1 >>= 1)
1802            {
1803              length ++;
1804            }
1805            for(Int i = 0; i < pcSlice->getNumRefIdx( REF_PIC_LIST_1 ); i++)
1806            {
1807              WRITE_CODE( refPicListModification->getRefPicSetIdxL1(i), length, "list_entry_l1");
1808            }
1809          }
1810        }
1811      }
1812    }
1813   
1814    if (pcSlice->isInterB())
1815    {
1816      WRITE_FLAG( pcSlice->getMvdL1ZeroFlag() ? 1 : 0,   "mvd_l1_zero_flag");
1817    }
1818
1819    if(!pcSlice->isIntra())
1820    {
1821      if (!pcSlice->isIntra() && pcSlice->getPPS()->getCabacInitPresentFlag())
1822      {
1823        SliceType sliceType   = pcSlice->getSliceType();
1824        Int  encCABACTableIdx = pcSlice->getPPS()->getEncCABACTableIdx();
1825        Bool encCabacInitFlag = (sliceType!=encCABACTableIdx && encCABACTableIdx!=I_SLICE) ? true : false;
1826        pcSlice->setCabacInitFlag( encCabacInitFlag );
1827        WRITE_FLAG( encCabacInitFlag?1:0, "cabac_init_flag" );
1828      }
1829    }
1830
1831    if ( pcSlice->getEnableTMVPFlag() )
1832    {
1833      if ( pcSlice->getSliceType() == B_SLICE )
1834      {
1835        WRITE_FLAG( pcSlice->getColFromL0Flag(), "collocated_from_l0_flag" );
1836      }
1837
1838      if ( pcSlice->getSliceType() != I_SLICE &&
1839        ((pcSlice->getColFromL0Flag()==1 && pcSlice->getNumRefIdx(REF_PIC_LIST_0)>1)||
1840        (pcSlice->getColFromL0Flag()==0  && pcSlice->getNumRefIdx(REF_PIC_LIST_1)>1)))
1841      {
1842        WRITE_UVLC( pcSlice->getColRefIdx(), "collocated_ref_idx" );
1843      }
1844    }
1845    if ( (pcSlice->getPPS()->getUseWP() && pcSlice->getSliceType()==P_SLICE) || (pcSlice->getPPS()->getWPBiPred() && pcSlice->getSliceType()==B_SLICE) )
1846    {
1847      xCodePredWeightTable( pcSlice );
1848    }
1849    assert(pcSlice->getMaxNumMergeCand()<=MRG_MAX_NUM_CANDS);
1850    if (!pcSlice->isIntra())
1851    {
1852      WRITE_UVLC(MRG_MAX_NUM_CANDS - pcSlice->getMaxNumMergeCand(), "five_minus_max_num_merge_cand");
1853    }
1854    Int iCode = pcSlice->getSliceQp() - ( pcSlice->getPPS()->getPicInitQPMinus26() + 26 );
1855    WRITE_SVLC( iCode, "slice_qp_delta" ); 
1856    if (pcSlice->getPPS()->getSliceChromaQpFlag())
1857    {
1858      iCode = pcSlice->getSliceQpDeltaCb();
1859      WRITE_SVLC( iCode, "slice_qp_delta_cb" );
1860      iCode = pcSlice->getSliceQpDeltaCr();
1861      WRITE_SVLC( iCode, "slice_qp_delta_cr" );
1862    }
1863    if (pcSlice->getPPS()->getDeblockingFilterControlPresentFlag())
1864    {
1865      if (pcSlice->getPPS()->getDeblockingFilterOverrideEnabledFlag() )
1866      {
1867        WRITE_FLAG(pcSlice->getDeblockingFilterOverrideFlag(), "deblocking_filter_override_flag");
1868      }
1869      if (pcSlice->getDeblockingFilterOverrideFlag())
1870      {
1871        WRITE_FLAG(pcSlice->getDeblockingFilterDisable(), "slice_disable_deblocking_filter_flag");
1872        if(!pcSlice->getDeblockingFilterDisable())
1873        {
1874          WRITE_SVLC (pcSlice->getDeblockingFilterBetaOffsetDiv2(), "slice_beta_offset_div2");
1875          WRITE_SVLC (pcSlice->getDeblockingFilterTcOffsetDiv2(),   "slice_tc_offset_div2");
1876        }
1877      }
1878    }
1879
1880    Bool isSAOEnabled = (!pcSlice->getSPS()->getUseSAO())?(false):(pcSlice->getSaoEnabledFlag()||pcSlice->getSaoEnabledFlagChroma());
1881    Bool isDBFEnabled = (!pcSlice->getDeblockingFilterDisable());
1882
1883    if(pcSlice->getPPS()->getLoopFilterAcrossSlicesEnabledFlag() && ( isSAOEnabled || isDBFEnabled ))
1884    {
1885      WRITE_FLAG(pcSlice->getLFCrossSliceBoundaryFlag()?1:0, "slice_loop_filter_across_slices_enabled_flag");
1886    }
1887  }
1888  if(pcSlice->getPPS()->getSliceHeaderExtensionPresentFlag())
1889  {
1890    WRITE_UVLC(0,"slice_header_extension_length");
1891  }
1892}
1893
1894Void TEncCavlc::codePTL( TComPTL* pcPTL, Bool profilePresentFlag, Int maxNumSubLayersMinus1)
1895{
1896  if(profilePresentFlag)
1897  {
1898    codeProfileTier(pcPTL->getGeneralPTL());    // general_...
1899  }
1900  WRITE_CODE( pcPTL->getGeneralPTL()->getLevelIdc(), 8, "general_level_idc" );
1901
1902  for (Int i = 0; i < maxNumSubLayersMinus1; i++)
1903  {
1904    if(profilePresentFlag)
1905    {
1906      WRITE_FLAG( pcPTL->getSubLayerProfilePresentFlag(i), "sub_layer_profile_present_flag[i]" );
1907    }
1908   
1909    WRITE_FLAG( pcPTL->getSubLayerLevelPresentFlag(i),   "sub_layer_level_present_flag[i]" );
1910  }
1911 
1912  if (maxNumSubLayersMinus1 > 0)
1913  {
1914    for (Int i = maxNumSubLayersMinus1; i < 8; i++)
1915    {
1916      WRITE_CODE(0, 2, "reserved_zero_2bits");
1917    }
1918  }
1919 
1920  for(Int i = 0; i < maxNumSubLayersMinus1; i++)
1921  {
1922    if( profilePresentFlag && pcPTL->getSubLayerProfilePresentFlag(i) )
1923    {
1924      codeProfileTier(pcPTL->getSubLayerPTL(i));  // sub_layer_...
1925    }
1926    if( pcPTL->getSubLayerLevelPresentFlag(i) )
1927    {
1928      WRITE_CODE( pcPTL->getSubLayerPTL(i)->getLevelIdc(), 8, "sub_layer_level_idc[i]" );
1929    }
1930  }
1931}
1932Void TEncCavlc::codeProfileTier( ProfileTierLevel* ptl )
1933{
1934  WRITE_CODE( ptl->getProfileSpace(), 2 ,     "XXX_profile_space[]");
1935  WRITE_FLAG( ptl->getTierFlag    (),         "XXX_tier_flag[]"    );
1936  WRITE_CODE( ptl->getProfileIdc  (), 5 ,     "XXX_profile_idc[]"  );   
1937  for(Int j = 0; j < 32; j++)
1938  {
1939    WRITE_FLAG( ptl->getProfileCompatibilityFlag(j), "XXX_profile_compatibility_flag[][j]");   
1940  }
1941
1942  WRITE_FLAG(ptl->getProgressiveSourceFlag(),   "general_progressive_source_flag");
1943  WRITE_FLAG(ptl->getInterlacedSourceFlag(),    "general_interlaced_source_flag");
1944  WRITE_FLAG(ptl->getNonPackedConstraintFlag(), "general_non_packed_constraint_flag");
1945  WRITE_FLAG(ptl->getFrameOnlyConstraintFlag(), "general_frame_only_constraint_flag");
1946 
1947  WRITE_CODE(0 , 16, "XXX_reserved_zero_44bits[0..15]");
1948  WRITE_CODE(0 , 16, "XXX_reserved_zero_44bits[16..31]");
1949  WRITE_CODE(0 , 12, "XXX_reserved_zero_44bits[32..43]");
1950}
1951
1952/**
1953 - write wavefront substreams sizes for the slice header.
1954 .
1955 \param pcSlice Where we find the substream size information.
1956 */
1957Void  TEncCavlc::codeTilesWPPEntryPoint( TComSlice* pSlice )
1958{
1959  if (!pSlice->getPPS()->getTilesEnabledFlag() && !pSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
1960  {
1961    return;
1962  }
1963  UInt numEntryPointOffsets = 0, offsetLenMinus1 = 0, maxOffset = 0;
1964  Int  numZeroSubstreamsAtStartOfSlice  = 0;
1965  UInt *entryPointOffset = NULL;
1966  if ( pSlice->getPPS()->getTilesEnabledFlag() )
1967  {
1968    numEntryPointOffsets = pSlice->getTileLocationCount();
1969    entryPointOffset     = new UInt[numEntryPointOffsets];
1970    for (Int idx=0; idx<pSlice->getTileLocationCount(); idx++)
1971    {
1972      if ( idx == 0 )
1973      {
1974        entryPointOffset [ idx ] = pSlice->getTileLocation( 0 );
1975      }
1976      else
1977      {
1978        entryPointOffset [ idx ] = pSlice->getTileLocation( idx ) - pSlice->getTileLocation( idx-1 );
1979      }
1980
1981      if ( entryPointOffset[ idx ] > maxOffset )
1982      {
1983        maxOffset = entryPointOffset[ idx ];
1984      }
1985    }
1986  }
1987  else if ( pSlice->getPPS()->getEntropyCodingSyncEnabledFlag() )
1988  {
1989    UInt* pSubstreamSizes               = pSlice->getSubstreamSizes();
1990    Int maxNumParts                       = pSlice->getPic()->getNumPartInCU();
1991    numZeroSubstreamsAtStartOfSlice       = pSlice->getSliceSegmentCurStartCUAddr()/maxNumParts/pSlice->getPic()->getFrameWidthInCU();
1992    Int  numZeroSubstreamsAtEndOfSlice    = pSlice->getPic()->getFrameHeightInCU()-1 - ((pSlice->getSliceSegmentCurEndCUAddr()-1)/maxNumParts/pSlice->getPic()->getFrameWidthInCU());
1993    numEntryPointOffsets                  = pSlice->getPPS()->getNumSubstreams() - numZeroSubstreamsAtStartOfSlice - numZeroSubstreamsAtEndOfSlice - 1;
1994    pSlice->setNumEntryPointOffsets(numEntryPointOffsets);
1995    entryPointOffset           = new UInt[numEntryPointOffsets];
1996    for (Int idx=0; idx<numEntryPointOffsets; idx++)
1997    {
1998      entryPointOffset[ idx ] = ( pSubstreamSizes[ idx+numZeroSubstreamsAtStartOfSlice ] >> 3 ) ;
1999      if ( entryPointOffset[ idx ] > maxOffset )
2000      {
2001        maxOffset = entryPointOffset[ idx ];
2002      }
2003    }
2004  }
2005  // Determine number of bits "offsetLenMinus1+1" required for entry point information
2006  offsetLenMinus1 = 0;
2007  while (maxOffset >= (1u << (offsetLenMinus1 + 1)))
2008  {
2009    offsetLenMinus1++;
2010    assert(offsetLenMinus1 + 1 < 32);   
2011  }
2012
2013  WRITE_UVLC(numEntryPointOffsets, "num_entry_point_offsets");
2014  if (numEntryPointOffsets>0)
2015  {
2016    WRITE_UVLC(offsetLenMinus1, "offset_len_minus1");
2017  }
2018
2019  for (UInt idx=0; idx<numEntryPointOffsets; idx++)
2020  {
2021    WRITE_CODE(entryPointOffset[ idx ]-1, offsetLenMinus1+1, "entry_point_offset_minus1");
2022  }
2023
2024  delete [] entryPointOffset;
2025}
2026
2027Void TEncCavlc::codeTerminatingBit      ( UInt uilsLast )
2028{
2029}
2030
2031Void TEncCavlc::codeSliceFinish ()
2032{
2033}
2034
2035Void TEncCavlc::codeMVPIdx ( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
2036{
2037  assert(0);
2038}
2039
2040Void TEncCavlc::codePartSize( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2041{
2042  assert(0);
2043}
2044
2045Void TEncCavlc::codePredMode( TComDataCU* pcCU, UInt uiAbsPartIdx )
2046{
2047  assert(0);
2048}
2049
2050Void TEncCavlc::codeMergeFlag    ( TComDataCU* pcCU, UInt uiAbsPartIdx )
2051{
2052  assert(0);
2053}
2054
2055Void TEncCavlc::codeMergeIndex    ( TComDataCU* pcCU, UInt uiAbsPartIdx )
2056{
2057  assert(0);
2058}
2059
2060Void TEncCavlc::codeInterModeFlag( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth, UInt uiEncMode )
2061{
2062  assert(0);
2063}
2064
2065Void TEncCavlc::codeCUTransquantBypassFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2066{
2067  assert(0);
2068}
2069
2070Void TEncCavlc::codeSkipFlag( TComDataCU* pcCU, UInt uiAbsPartIdx )
2071{
2072  assert(0);
2073}
2074
2075Void TEncCavlc::codeSplitFlag   ( TComDataCU* pcCU, UInt uiAbsPartIdx, UInt uiDepth )
2076{
2077  assert(0);
2078}
2079
2080Void TEncCavlc::codeTransformSubdivFlag( UInt uiSymbol, UInt uiCtx )
2081{
2082  assert(0);
2083}
2084
2085Void TEncCavlc::codeQtCbf( TComDataCU* pcCU, UInt uiAbsPartIdx, TextType eType, UInt uiTrDepth )
2086{
2087  assert(0);
2088}
2089
2090Void TEncCavlc::codeQtRootCbf( TComDataCU* pcCU, UInt uiAbsPartIdx )
2091{
2092  assert(0);
2093}
2094
2095Void TEncCavlc::codeQtCbfZero( TComDataCU* pcCU, TextType eType, UInt uiTrDepth )
2096{
2097  assert(0);
2098}
2099Void TEncCavlc::codeQtRootCbfZero( TComDataCU* pcCU )
2100{
2101  assert(0);
2102}
2103
2104Void TEncCavlc::codeTransformSkipFlags (TComDataCU* pcCU, UInt uiAbsPartIdx, UInt width, UInt height, TextType eTType )
2105{
2106  assert(0);
2107}
2108
2109/** Code I_PCM information.
2110 * \param pcCU pointer to CU
2111 * \param uiAbsPartIdx CU index
2112 * \returns Void
2113 */
2114Void TEncCavlc::codeIPCMInfo( TComDataCU* pcCU, UInt uiAbsPartIdx )
2115{
2116  assert(0);
2117}
2118
2119Void TEncCavlc::codeIntraDirLumaAng( TComDataCU* pcCU, UInt uiAbsPartIdx, Bool isMultiple)
2120{
2121  assert(0);
2122}
2123
2124Void TEncCavlc::codeIntraDirChroma( TComDataCU* pcCU, UInt uiAbsPartIdx )
2125{
2126  assert(0);
2127}
2128
2129Void TEncCavlc::codeInterDir( TComDataCU* pcCU, UInt uiAbsPartIdx )
2130{
2131  assert(0);
2132}
2133
2134Void TEncCavlc::codeRefFrmIdx( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
2135{
2136  assert(0);
2137}
2138
2139Void TEncCavlc::codeMvd( TComDataCU* pcCU, UInt uiAbsPartIdx, RefPicList eRefList )
2140{
2141  assert(0);
2142}
2143
2144Void TEncCavlc::codeDeltaQP( TComDataCU* pcCU, UInt uiAbsPartIdx )
2145{
2146  Int iDQp  = pcCU->getQP( uiAbsPartIdx ) - pcCU->getRefQP( uiAbsPartIdx );
2147
[442]2148#if REPN_FORMAT_IN_VPS
2149  Int qpBdOffsetY =  pcCU->getSlice()->getQpBDOffsetY();
2150#else
[313]2151  Int qpBdOffsetY =  pcCU->getSlice()->getSPS()->getQpBDOffsetY();
[442]2152#endif
[313]2153  iDQp = (iDQp + 78 + qpBdOffsetY + (qpBdOffsetY/2)) % (52 + qpBdOffsetY) - 26 - (qpBdOffsetY/2);
2154
2155  xWriteSvlc( iDQp );
2156 
2157  return;
2158}
2159
2160Void TEncCavlc::codeCoeffNxN    ( TComDataCU* pcCU, TCoeff* pcCoef, UInt uiAbsPartIdx, UInt uiWidth, UInt uiHeight, UInt uiDepth, TextType eTType )
2161{
2162  assert(0);
2163}
2164
2165Void TEncCavlc::estBit( estBitsSbacStruct* pcEstBitsCabac, Int width, Int height, TextType eTType )
2166{
2167  // printf("error : no VLC mode support in this version\n");
2168  return;
2169}
2170
2171// ====================================================================================================================
2172// Protected member functions
2173// ====================================================================================================================
2174
2175/** code explicit wp tables
2176 * \param TComSlice* pcSlice
2177 * \returns Void
2178 */
2179Void TEncCavlc::xCodePredWeightTable( TComSlice* pcSlice )
2180{
2181  wpScalingParam  *wp;
2182  Bool            bChroma     = true; // color always present in HEVC ?
2183  Int             iNbRef       = (pcSlice->getSliceType() == B_SLICE ) ? (2) : (1);
2184  Bool            bDenomCoded  = false;
2185  UInt            uiMode = 0;
2186  UInt            uiTotalSignalledWeightFlags = 0;
[494]2187#if AUXILIARY_PICTURES
2188  if (pcSlice->getChromaFormatIdc() == CHROMA_400)
2189  {
2190    bChroma = false;
2191  }
2192#endif
[313]2193  if ( (pcSlice->getSliceType()==P_SLICE && pcSlice->getPPS()->getUseWP()) || (pcSlice->getSliceType()==B_SLICE && pcSlice->getPPS()->getWPBiPred()) )
2194  {
2195    uiMode = 1; // explicit
2196  }
2197  if(uiMode == 1)
2198  {
2199
2200    for ( Int iNumRef=0 ; iNumRef<iNbRef ; iNumRef++ ) 
2201    {
2202      RefPicList  eRefPicList = ( iNumRef ? REF_PIC_LIST_1 : REF_PIC_LIST_0 );
2203
2204      for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ ) 
2205      {
2206        pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
2207        if ( !bDenomCoded ) 
2208        {
2209          Int iDeltaDenom;
2210          WRITE_UVLC( wp[0].uiLog2WeightDenom, "luma_log2_weight_denom" );     // ue(v): luma_log2_weight_denom
2211
2212          if( bChroma )
2213          {
2214            iDeltaDenom = (wp[1].uiLog2WeightDenom - wp[0].uiLog2WeightDenom);
2215            WRITE_SVLC( iDeltaDenom, "delta_chroma_log2_weight_denom" );       // se(v): delta_chroma_log2_weight_denom
2216          }
2217          bDenomCoded = true;
2218        }
2219        WRITE_FLAG( wp[0].bPresentFlag, "luma_weight_lX_flag" );               // u(1): luma_weight_lX_flag
2220        uiTotalSignalledWeightFlags += wp[0].bPresentFlag;
2221      }
2222      if (bChroma) 
2223      {
2224        for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ ) 
2225        {
2226          pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
2227          WRITE_FLAG( wp[1].bPresentFlag, "chroma_weight_lX_flag" );           // u(1): chroma_weight_lX_flag
2228          uiTotalSignalledWeightFlags += 2*wp[1].bPresentFlag;
2229        }
2230      }
2231
2232      for ( Int iRefIdx=0 ; iRefIdx<pcSlice->getNumRefIdx(eRefPicList) ; iRefIdx++ ) 
2233      {
2234        pcSlice->getWpScaling(eRefPicList, iRefIdx, wp);
2235        if ( wp[0].bPresentFlag ) 
2236        {
2237          Int iDeltaWeight = (wp[0].iWeight - (1<<wp[0].uiLog2WeightDenom));
2238          WRITE_SVLC( iDeltaWeight, "delta_luma_weight_lX" );                  // se(v): delta_luma_weight_lX
2239          WRITE_SVLC( wp[0].iOffset, "luma_offset_lX" );                       // se(v): luma_offset_lX
2240        }
2241
2242        if ( bChroma ) 
2243        {
2244          if ( wp[1].bPresentFlag )
2245          {
2246            for ( Int j=1 ; j<3 ; j++ ) 
2247            {
2248              Int iDeltaWeight = (wp[j].iWeight - (1<<wp[1].uiLog2WeightDenom));
2249              WRITE_SVLC( iDeltaWeight, "delta_chroma_weight_lX" );            // se(v): delta_chroma_weight_lX
2250
2251              Int pred = ( 128 - ( ( 128*wp[j].iWeight)>>(wp[j].uiLog2WeightDenom) ) );
2252              Int iDeltaChroma = (wp[j].iOffset - pred);
2253              WRITE_SVLC( iDeltaChroma, "delta_chroma_offset_lX" );            // se(v): delta_chroma_offset_lX
2254            }
2255          }
2256        }
2257      }
2258    }
2259    assert(uiTotalSignalledWeightFlags<=24);
2260  }
2261}
2262
2263/** code quantization matrix
2264 *  \param scalingList quantization matrix information
2265 */
2266Void TEncCavlc::codeScalingList( TComScalingList* scalingList )
2267{
2268  UInt listId,sizeId;
2269  Bool scalingListPredModeFlag;
2270
2271    //for each size
2272    for(sizeId = 0; sizeId < SCALING_LIST_SIZE_NUM; sizeId++)
2273    {
2274      for(listId = 0; listId < g_scalingListNum[sizeId]; listId++)
2275      {
2276        scalingListPredModeFlag = scalingList->checkPredMode( sizeId, listId );
2277        WRITE_FLAG( scalingListPredModeFlag, "scaling_list_pred_mode_flag" );
2278        if(!scalingListPredModeFlag)// Copy Mode
2279        {
2280          WRITE_UVLC( (Int)listId - (Int)scalingList->getRefMatrixId (sizeId,listId), "scaling_list_pred_matrix_id_delta");
2281        }
2282        else// DPCM Mode
2283        {
2284          xCodeScalingList(scalingList, sizeId, listId);
2285        }
2286      }
2287    }
2288  return;
2289}
2290/** code DPCM
2291 * \param scalingList quantization matrix information
2292 * \param sizeIdc size index
2293 * \param listIdc list index
2294 */
2295Void TEncCavlc::xCodeScalingList(TComScalingList* scalingList, UInt sizeId, UInt listId)
2296{
2297  Int coefNum = min(MAX_MATRIX_COEF_NUM,(Int)g_scalingListSize[sizeId]);
2298  UInt* scan  = (sizeId == 0) ? g_auiSigLastScan [ SCAN_DIAG ] [ 1 ] :  g_sigLastScanCG32x32;
2299  Int nextCoef = SCALING_LIST_START_VALUE;
2300  Int data;
2301  Int *src = scalingList->getScalingListAddress(sizeId, listId);
[540]2302  if( sizeId > SCALING_LIST_8x8 )
2303  {
2304    WRITE_SVLC( scalingList->getScalingListDC(sizeId,listId) - 8, "scaling_list_dc_coef_minus8");
2305    nextCoef = scalingList->getScalingListDC(sizeId,listId);
2306  }
2307  for(Int i=0;i<coefNum;i++)
2308  {
2309    data = src[scan[i]] - nextCoef;
2310    nextCoef = src[scan[i]];
2311    if(data > 127)
[313]2312    {
[540]2313      data = data - 256;
[313]2314    }
[540]2315    if(data < -128)
[313]2316    {
[540]2317      data = data + 256;
2318    }
[442]2319
[540]2320    WRITE_SVLC( data,  "scaling_list_delta_coef");
2321  }
[313]2322}
2323Bool TEncCavlc::findMatchingLTRP ( TComSlice* pcSlice, UInt *ltrpsIndex, Int ltrpPOC, Bool usedFlag )
2324{
2325  // Bool state = true, state2 = false;
[442]2326  Int lsb = ltrpPOC & ((1<<pcSlice->getSPS()->getBitsForPOC())-1);
[313]2327  for (Int k = 0; k < pcSlice->getSPS()->getNumLongTermRefPicSPS(); k++)
2328  {
2329    if ( (lsb == pcSlice->getSPS()->getLtRefPicPocLsbSps(k)) && (usedFlag == pcSlice->getSPS()->getUsedByCurrPicLtSPSFlag(k)) )
2330    {
2331      *ltrpsIndex = k;
2332      return true;
2333    }
2334  }
2335  return false;
2336}
2337Bool TComScalingList::checkPredMode(UInt sizeId, UInt listId)
2338{
2339  for(Int predListIdx = (Int)listId ; predListIdx >= 0; predListIdx--)
2340  {
2341    if( !memcmp(getScalingListAddress(sizeId,listId),((listId == predListIdx) ?
2342      getScalingListDefaultAddress(sizeId, predListIdx): getScalingListAddress(sizeId, predListIdx)),sizeof(Int)*min(MAX_MATRIX_COEF_NUM,(Int)g_scalingListSize[sizeId])) // check value of matrix
2343     && ((sizeId < SCALING_LIST_16x16) || (getScalingListDC(sizeId,listId) == getScalingListDC(sizeId,predListIdx)))) // check DC value
2344    {
2345      setRefMatrixId(sizeId, listId, predListIdx);
2346      return false;
2347    }
2348  }
2349  return true;
2350}
2351//! \}
Note: See TracBrowser for help on using the repository browser.