source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibEncoder/SEIwrite.cpp @ 1096

Last change on this file since 1096 was 1096, checked in by seregin, 9 years ago

cleanup macros: REMOVE_BSP_HRD_SEI

  • Property svn:eol-style set to native
File size: 58.7 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license.
5 *
6 * Copyright (c) 2010-2014, ITU/ISO/IEC
7 * All rights reserved.
8 *
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions are met:
11 *
12 *  * Redistributions of source code must retain the above copyright notice,
13 *    this list of conditions and the following disclaimer.
14 *  * Redistributions in binary form must reproduce the above copyright notice,
15 *    this list of conditions and the following disclaimer in the documentation
16 *    and/or other materials provided with the distribution.
17 *  * Neither the name of the ITU/ISO/IEC nor the names of its contributors may
18 *    be used to endorse or promote products derived from this software without
19 *    specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
22 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS
25 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
26 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
27 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
28 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
29 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
31 * THE POSSIBILITY OF SUCH DAMAGE.
32 */
33
34#include "TLibCommon/TComBitCounter.h"
35#include "TLibCommon/TComBitStream.h"
36#include "TLibCommon/SEI.h"
37#include "TLibCommon/TComSlice.h"
38#include "TLibCommon/TComPicYuv.h"
39#include "SEIwrite.h"
40
41//! \ingroup TLibEncoder
42//! \{
43
44#if ENC_DEC_TRACE
45Void  xTraceSEIHeader()
46{
47  fprintf( g_hTrace, "=========== SEI message ===========\n");
48}
49
50Void  xTraceSEIMessageType(SEI::PayloadType payloadType)
51{
52  fprintf( g_hTrace, "=========== %s SEI message ===========\n", SEI::getSEIMessageString(payloadType));
53}
54#endif
55
56#if O0164_MULTI_LAYER_HRD
57#if VPS_VUI_BSP_HRD_PARAMS
58Void SEIWriter::xWriteSEIpayloadData(TComBitIf& bs, const SEI& sei, TComVPS *vps, TComSPS *sps, const SEIScalableNesting* nestingSeiPtr, const SEIBspNesting* bspNestingSeiPtr)
59#else
60Void SEIWriter::xWriteSEIpayloadData(TComBitIf& bs, const SEI& sei, TComVPS *vps, TComSPS *sps, const SEIScalableNesting& nestingSei, const SEIBspNesting& bspNestingSei)
61#endif
62#else
63Void SEIWriter::xWriteSEIpayloadData(TComBitIf& bs, const SEI& sei, TComSPS *sps)
64#endif
65{
66#if VPS_VUI_BSP_HRD_PARAMS
67  SEIScalableNesting nestingSei;
68  SEIBspNesting      bspNestingSei;
69  if( nestingSeiPtr )
70  {
71    nestingSei = *nestingSeiPtr;
72  }
73  if( bspNestingSeiPtr )
74  {
75    bspNestingSei = *bspNestingSeiPtr;
76  }
77#endif
78  switch (sei.payloadType())
79  {
80  case SEI::USER_DATA_UNREGISTERED:
81    xWriteSEIuserDataUnregistered(*static_cast<const SEIuserDataUnregistered*>(&sei));
82    break;
83  case SEI::ACTIVE_PARAMETER_SETS:
84    xWriteSEIActiveParameterSets(*static_cast<const SEIActiveParameterSets*>(& sei));
85    break; 
86  case SEI::DECODED_PICTURE_HASH:
87    xWriteSEIDecodedPictureHash(*static_cast<const SEIDecodedPictureHash*>(&sei));
88    break;
89#if VPS_VUI_BSP_HRD_PARAMS
90  case SEI::DECODING_UNIT_INFO:
91    xWriteSEIDecodingUnitInfo(*static_cast<const SEIDecodingUnitInfo*>(& sei), sps, nestingSeiPtr, bspNestingSeiPtr, vps);
92    break;
93  case SEI::BUFFERING_PERIOD:
94    xWriteSEIBufferingPeriod(*static_cast<const SEIBufferingPeriod*>(&sei), sps, nestingSeiPtr, bspNestingSeiPtr, vps);
95    break;
96  case SEI::PICTURE_TIMING:
97    xWriteSEIPictureTiming(*static_cast<const SEIPictureTiming*>(&sei), sps, nestingSeiPtr, bspNestingSeiPtr, vps);
98    break;
99#else
100  case SEI::DECODING_UNIT_INFO:
101    xWriteSEIDecodingUnitInfo(*static_cast<const SEIDecodingUnitInfo*>(& sei), sps);
102    break;
103  case SEI::BUFFERING_PERIOD:
104    xWriteSEIBufferingPeriod(*static_cast<const SEIBufferingPeriod*>(&sei), sps);
105    break;
106  case SEI::PICTURE_TIMING:
107    xWriteSEIPictureTiming(*static_cast<const SEIPictureTiming*>(&sei), sps);
108    break;
109#endif
110  case SEI::RECOVERY_POINT:
111    xWriteSEIRecoveryPoint(*static_cast<const SEIRecoveryPoint*>(&sei));
112    break;
113  case SEI::FRAME_PACKING:
114    xWriteSEIFramePacking(*static_cast<const SEIFramePacking*>(&sei));
115    break;
116  case SEI::SEGM_RECT_FRAME_PACKING:
117    xWriteSEISegmentedRectFramePacking(*static_cast<const SEISegmentedRectFramePacking*>(&sei));
118    break;
119  case SEI::DISPLAY_ORIENTATION:
120    xWriteSEIDisplayOrientation(*static_cast<const SEIDisplayOrientation*>(&sei));
121    break;
122  case SEI::TEMPORAL_LEVEL0_INDEX:
123    xWriteSEITemporalLevel0Index(*static_cast<const SEITemporalLevel0Index*>(&sei));
124    break;
125  case SEI::REGION_REFRESH_INFO:
126    xWriteSEIGradualDecodingRefreshInfo(*static_cast<const SEIGradualDecodingRefreshInfo*>(&sei));
127    break;
128  case SEI::NO_DISPLAY:
129    xWriteSEINoDisplay(*static_cast<const SEINoDisplay*>(&sei));
130    break;
131  case SEI::TONE_MAPPING_INFO:
132    xWriteSEIToneMappingInfo(*static_cast<const SEIToneMappingInfo*>(&sei));
133    break;
134  case SEI::SOP_DESCRIPTION:
135    xWriteSEISOPDescription(*static_cast<const SEISOPDescription*>(&sei));
136    break;
137  case SEI::SCALABLE_NESTING:
138#if O0164_MULTI_LAYER_HRD
139    xWriteSEIScalableNesting(bs, *static_cast<const SEIScalableNesting*>(&sei), vps, sps);
140#else
141    xWriteSEIScalableNesting(bs, *static_cast<const SEIScalableNesting*>(&sei), sps);
142#endif
143    break;
144  case SEI::CHROMA_SAMPLING_FILTER_HINT:
145    xWriteSEIChromaSamplingFilterHint(*static_cast<const SEIChromaSamplingFilterHint*>(&sei)/*, sps*/);
146    break;
147  case SEI::TEMP_MOTION_CONSTRAINED_TILE_SETS:
148    xWriteSEITempMotionConstrainedTileSets(bs, *static_cast<const SEITempMotionConstrainedTileSets*>(&sei));
149    break;
150  case SEI::TIME_CODE:
151    xWriteSEITimeCode(*static_cast<const SEITimeCode*>(&sei));
152    break;
153  case SEI::KNEE_FUNCTION_INFO:
154    xWriteSEIKneeFunctionInfo(*static_cast<const SEIKneeFunctionInfo*>(&sei));
155    break;
156  case SEI::MASTERING_DISPLAY_COLOUR_VOLUME:
157    xWriteSEIMasteringDisplayColourVolume(*static_cast<const SEIMasteringDisplayColourVolume*>(&sei));
158    break;
159#if Q0074_COLOUR_REMAPPING_SEI
160  case SEI::COLOUR_REMAPPING_INFO:
161    xWriteSEIColourRemappingInfo(*static_cast<const SEIColourRemappingInfo*>(&sei));
162    break;
163#endif
164#if SVC_EXTENSION
165#if LAYERS_NOT_PRESENT_SEI
166  case SEI::LAYERS_NOT_PRESENT:
167    xWriteSEILayersNotPresent(*static_cast<const SEILayersNotPresent*>(&sei));
168    break;
169#endif
170#if N0383_IL_CONSTRAINED_TILE_SETS_SEI
171  case SEI::INTER_LAYER_CONSTRAINED_TILE_SETS:
172    xWriteSEIInterLayerConstrainedTileSets(*static_cast<const SEIInterLayerConstrainedTileSets*>(&sei));
173    break;
174#endif
175#if SUB_BITSTREAM_PROPERTY_SEI
176   case SEI::SUB_BITSTREAM_PROPERTY:
177     xWriteSEISubBitstreamProperty(*static_cast<const SEISubBitstreamProperty*>(&sei));
178     break;
179#endif
180#if O0164_MULTI_LAYER_HRD
181   case SEI::BSP_NESTING:
182     xWriteSEIBspNesting(bs, *static_cast<const SEIBspNesting*>(&sei), vps, sps, nestingSei);
183     break;
184   case SEI::BSP_INITIAL_ARRIVAL_TIME:
185     xWriteSEIBspInitialArrivalTime(*static_cast<const SEIBspInitialArrivalTime*>(&sei), vps, sps, nestingSei, bspNestingSei);
186     break;
187#endif
188#if Q0078_ADD_LAYER_SETS
189   case SEI::OUTPUT_LAYER_SET_NESTING:
190     xWriteSEIOutputLayerSetNesting(bs, *static_cast<const SEIOutputLayerSetNesting*>(&sei), vps, sps);
191     break;
192   case SEI::VPS_REWRITING:
193     xWriteSEIVPSRewriting(*static_cast<const SEIVPSRewriting*>(&sei));
194     break;
195#endif
196#if Q0189_TMVP_CONSTRAINTS
197   case SEI::TMVP_CONSTRAINTS:
198     xWriteSEITMVPConstraints(*static_cast<const SEITMVPConstrains*>(&sei));
199     break;
200#endif
201#if Q0247_FRAME_FIELD_INFO
202   case SEI::FRAME_FIELD_INFO:
203     xWriteSEIFrameFieldInfo(*static_cast<const SEIFrameFieldInfo*>(&sei));
204     break;
205#endif
206#if P0123_ALPHA_CHANNEL_SEI
207   case SEI::ALPHA_CHANNEL_INFO:
208     xWriteSEIAlphaChannelInfo(*static_cast<const SEIAlphaChannelInfo*>(&sei));
209     break;
210#endif
211#if Q0096_OVERLAY_SEI
212   case SEI::OVERLAY_INFO:
213     xWriteSEIOverlayInfo(*static_cast<const SEIOverlayInfo*>(&sei));
214     break;
215#endif
216#endif //SVC_EXTENSION
217  default:
218    assert(!"Unhandled SEI message");
219    break;
220  }
221  xWriteByteAlign();
222}
223
224/**
225 * marshal a single SEI message sei, storing the marshalled representation
226 * in bitstream bs.
227 */
228#if O0164_MULTI_LAYER_HRD
229Void SEIWriter::writeSEImessage(TComBitIf& bs, const SEI& sei, TComVPS *vps, TComSPS *sps, const SEIScalableNesting* nestingSei, const SEIBspNesting* bspNestingSei)
230#else
231Void SEIWriter::writeSEImessage(TComBitIf& bs, const SEI& sei, TComSPS *sps)
232#endif
233{
234  /* calculate how large the payload data is */
235  /* TODO: this would be far nicer if it used vectored buffers */
236  TComBitCounter bs_count;
237  bs_count.resetBits();
238  setBitstream(&bs_count);
239
240
241#if ENC_DEC_TRACE
242  Bool traceEnable = g_HLSTraceEnable;
243  g_HLSTraceEnable = false;
244#endif
245#if O0164_MULTI_LAYER_HRD
246#if VPS_VUI_BSP_HRD_PARAMS
247  xWriteSEIpayloadData(bs_count, sei, vps, sps, nestingSei, bspNestingSei);
248#else
249  xWriteSEIpayloadData(bs_count, sei, vps, sps, *nestingSei, *bspNestingSei);
250#endif
251#else
252  xWriteSEIpayloadData(bs_count, sei, sps);
253#endif
254#if ENC_DEC_TRACE
255  g_HLSTraceEnable = traceEnable;
256#endif
257
258  UInt payload_data_num_bits = bs_count.getNumberOfWrittenBits();
259  assert(0 == payload_data_num_bits % 8);
260
261  setBitstream(&bs);
262
263#if ENC_DEC_TRACE
264  if (g_HLSTraceEnable)
265  xTraceSEIHeader();
266#endif
267
268  UInt payloadType = sei.payloadType();
269  for (; payloadType >= 0xff; payloadType -= 0xff)
270  {
271    WRITE_CODE(0xff, 8, "payload_type");
272  }
273  WRITE_CODE(payloadType, 8, "payload_type");
274
275  UInt payloadSize = payload_data_num_bits/8;
276  for (; payloadSize >= 0xff; payloadSize -= 0xff)
277  {
278    WRITE_CODE(0xff, 8, "payload_size");
279  }
280  WRITE_CODE(payloadSize, 8, "payload_size");
281
282  /* payloadData */
283#if ENC_DEC_TRACE
284  if (g_HLSTraceEnable)
285  xTraceSEIMessageType(sei.payloadType());
286#endif
287
288#if O0164_MULTI_LAYER_HRD
289#if VPS_VUI_BSP_HRD_PARAMS
290  xWriteSEIpayloadData(bs, sei, vps, sps, nestingSei, bspNestingSei);
291#else
292  xWriteSEIpayloadData(bs, sei, vps, sps, *nestingSei, *bspNestingSei);
293#endif
294#else
295  xWriteSEIpayloadData(bs, sei, sps);
296#endif
297}
298
299/**
300 * marshal a user_data_unregistered SEI message sei, storing the marshalled
301 * representation in bitstream bs.
302 */
303Void SEIWriter::xWriteSEIuserDataUnregistered(const SEIuserDataUnregistered &sei)
304{
305  for (UInt i = 0; i < ISO_IEC_11578_LEN; i++)
306  {
307    WRITE_CODE(sei.uuid_iso_iec_11578[i], 8 , "sei.uuid_iso_iec_11578[i]");
308  }
309
310  for (UInt i = 0; i < sei.userDataLength; i++)
311  {
312    WRITE_CODE(sei.userData[i], 8 , "user_data");
313  }
314}
315
316/**
317 * marshal a decoded picture hash SEI message, storing the marshalled
318 * representation in bitstream bs.
319 */
320Void SEIWriter::xWriteSEIDecodedPictureHash(const SEIDecodedPictureHash& sei)
321{
322  const Char *traceString="\0";
323  switch (sei.method)
324  {
325    case SEIDecodedPictureHash::MD5: traceString="picture_md5"; break;
326    case SEIDecodedPictureHash::CRC: traceString="picture_crc"; break;
327    case SEIDecodedPictureHash::CHECKSUM: traceString="picture_checksum"; break;
328    default: assert(false); break;
329  }
330
331  if (traceString != 0) //use of this variable is needed to avoid a compiler error with G++ 4.6.1
332  {
333    WRITE_CODE(sei.method, 8, "hash_type");
334    for(UInt i=0; i<UInt(sei.m_digest.hash.size()); i++)
335    {
336      WRITE_CODE(sei.m_digest.hash[i], 8, traceString);
337    }
338  }
339}
340
341Void SEIWriter::xWriteSEIActiveParameterSets(const SEIActiveParameterSets& sei)
342{
343  WRITE_CODE(sei.activeVPSId,     4,         "active_video_parameter_set_id");
344  WRITE_FLAG(sei.m_selfContainedCvsFlag,     "self_contained_cvs_flag");
345  WRITE_FLAG(sei.m_noParameterSetUpdateFlag, "no_parameter_set_update_flag");
346  WRITE_UVLC(sei.numSpsIdsMinus1,            "num_sps_ids_minus1");
347
348  assert (sei.activeSeqParameterSetId.size() == (sei.numSpsIdsMinus1 + 1));
349
350  for (Int i = 0; i < sei.activeSeqParameterSetId.size(); i++)
351  {
352    WRITE_UVLC(sei.activeSeqParameterSetId[i], "active_seq_parameter_set_id"); 
353  }
354#if R0247_SEI_ACTIVE
355  for (Int i = 1; i < sei.activeSeqParameterSetId.size(); i++)
356  {
357    WRITE_UVLC(sei.layerSpsIdx[i], "layer_sps_idx"); 
358  }
359#endif
360}
361
362#if VPS_VUI_BSP_HRD_PARAMS
363Void SEIWriter::xWriteSEIDecodingUnitInfo(const SEIDecodingUnitInfo& sei, TComSPS *sps, const SEIScalableNesting* nestingSei, const SEIBspNesting* bspNestingSei, TComVPS *vps)
364#else
365Void SEIWriter::xWriteSEIDecodingUnitInfo(const SEIDecodingUnitInfo& sei, TComSPS *sps)
366#endif
367{
368#if VPS_VUI_BSP_HRD_PARAMS
369  TComHRD *hrd;
370  if( bspNestingSei )   // If DU info SEI contained inside a BSP nesting SEI message
371  {
372    assert( nestingSei );
373    Int psIdx = bspNestingSei->m_seiPartitioningSchemeIdx;
374    Int seiOlsIdx = bspNestingSei->m_seiOlsIdx;
375    Int maxTemporalId = nestingSei->m_nestingMaxTemporalIdPlus1[0] - 1;
376    Int maxValues = vps->getNumBspSchedulesMinus1(seiOlsIdx, psIdx, maxTemporalId) + 1;
377    std::vector<Int> hrdIdx(maxValues, 0);
378    std::vector<TComHRD *> hrdVec;
379    std::vector<Int> syntaxElemLen(maxValues, 0);
380    for(Int i = 0; i < maxValues; i++)
381    {
382      hrdIdx[i] = vps->getBspHrdIdx( seiOlsIdx, psIdx, maxTemporalId, i, bspNestingSei->m_bspIdx);
383      hrdVec.push_back(vps->getBspHrd(hrdIdx[i]));
384   
385      syntaxElemLen[i] = hrdVec[i]->getInitialCpbRemovalDelayLengthMinus1() + 1;
386      if ( !(hrdVec[i]->getNalHrdParametersPresentFlag() || hrdVec[i]->getVclHrdParametersPresentFlag()) )
387      {
388        assert( syntaxElemLen[i] == 24 ); // Default of value init_cpb_removal_delay_length_minus1 is 23
389      }
390      if( i > 0 )
391      {
392        assert( hrdVec[i]->getSubPicCpbParamsPresentFlag()    == hrdVec[i-1]->getSubPicCpbParamsPresentFlag() );
393        assert( hrdVec[i]->getSubPicCpbParamsInPicTimingSEIFlag()   == hrdVec[i-1]->getSubPicCpbParamsInPicTimingSEIFlag() );
394        assert( hrdVec[i]->getDpbOutputDelayDuLengthMinus1()  == hrdVec[i-1]->getDpbOutputDelayDuLengthMinus1() );
395        // To be done: Check CpbDpbDelaysPresentFlag
396      }
397    }
398    hrd = hrdVec[0];
399  }
400  else
401  {
402    TComVUI *vui = sps->getVuiParameters();
403    hrd = vui->getHrdParameters();
404  }
405
406  WRITE_UVLC(sei.m_decodingUnitIdx, "decoding_unit_idx");
407  if(hrd->getSubPicCpbParamsInPicTimingSEIFlag())
408  {
409    WRITE_CODE( sei.m_duSptCpbRemovalDelay, (hrd->getDuCpbRemovalDelayLengthMinus1() + 1), "du_spt_cpb_removal_delay");
410  }
411  WRITE_FLAG( sei.m_dpbOutputDuDelayPresentFlag, "dpb_output_du_delay_present_flag");
412  if(sei.m_dpbOutputDuDelayPresentFlag)
413  {
414    WRITE_CODE(sei.m_picSptDpbOutputDuDelay, hrd->getDpbOutputDelayDuLengthMinus1() + 1, "pic_spt_dpb_output_du_delay");
415  }
416#else
417  TComVUI *vui = sps->getVuiParameters();
418  WRITE_UVLC(sei.m_decodingUnitIdx, "decoding_unit_idx");
419  if(vui->getHrdParameters()->getSubPicCpbParamsInPicTimingSEIFlag())
420  {
421    WRITE_CODE( sei.m_duSptCpbRemovalDelay, (vui->getHrdParameters()->getDuCpbRemovalDelayLengthMinus1() + 1), "du_spt_cpb_removal_delay");
422  }
423  WRITE_FLAG( sei.m_dpbOutputDuDelayPresentFlag, "dpb_output_du_delay_present_flag");
424  if(sei.m_dpbOutputDuDelayPresentFlag)
425  {
426    WRITE_CODE(sei.m_picSptDpbOutputDuDelay, vui->getHrdParameters()->getDpbOutputDelayDuLengthMinus1() + 1, "pic_spt_dpb_output_du_delay");
427  }
428#endif
429}
430
431#if VPS_VUI_BSP_HRD_PARAMS
432Void SEIWriter::xWriteSEIBufferingPeriod(const SEIBufferingPeriod& sei, TComSPS *sps, const SEIScalableNesting* nestingSei, const SEIBspNesting* bspNestingSei, TComVPS *vps)
433#else
434Void SEIWriter::xWriteSEIBufferingPeriod(const SEIBufferingPeriod& sei, TComSPS *sps)
435#endif
436{
437  Int i, nalOrVcl;
438#if VPS_VUI_BSP_HRD_PARAMS
439  TComHRD *hrd;
440  if( bspNestingSei )   // If BP SEI contained inside a BSP nesting SEI message
441  {
442    assert( nestingSei );
443    Int psIdx = bspNestingSei->m_seiPartitioningSchemeIdx;
444    Int seiOlsIdx = bspNestingSei->m_seiOlsIdx;
445    Int maxTemporalId = nestingSei->m_nestingMaxTemporalIdPlus1[0] - 1;
446    Int maxValues = vps->getNumBspSchedulesMinus1(seiOlsIdx, psIdx, maxTemporalId) + 1;
447    std::vector<Int> hrdIdx(maxValues, 0);
448    std::vector<TComHRD *> hrdVec;
449    std::vector<Int> syntaxElemLen(maxValues, 0);
450    for(i = 0; i < maxValues; i++)
451    {
452      hrdIdx[i] = vps->getBspHrdIdx( seiOlsIdx, psIdx, maxTemporalId, i, bspNestingSei->m_bspIdx);
453      hrdVec.push_back(vps->getBspHrd(hrdIdx[i]));
454   
455      syntaxElemLen[i] = hrdVec[i]->getInitialCpbRemovalDelayLengthMinus1() + 1;
456      if ( !(hrdVec[i]->getNalHrdParametersPresentFlag() || hrdVec[i]->getVclHrdParametersPresentFlag()) )
457      {
458        assert( syntaxElemLen[i] == 24 ); // Default of value init_cpb_removal_delay_length_minus1 is 23
459      }
460      if( i > 0 )
461      {
462        assert( hrdVec[i]->getCpbRemovalDelayLengthMinus1()   == hrdVec[i-1]->getCpbRemovalDelayLengthMinus1() );
463        assert( hrdVec[i]->getDpbOutputDelayDuLengthMinus1()  == hrdVec[i-1]->getDpbOutputDelayDuLengthMinus1() );
464        assert( hrdVec[i]->getSubPicCpbParamsPresentFlag()    == hrdVec[i-1]->getSubPicCpbParamsPresentFlag() );
465      }
466    }
467    hrd = hrdVec[i];
468  }
469  else
470  {
471    TComVUI *vui = sps->getVuiParameters();
472    hrd = vui->getHrdParameters();
473  }
474  // To be done: When contained in an BSP HRD SEI message, the hrd structure is to be chosen differently.
475#else
476  TComVUI *vui = sps->getVuiParameters();
477  TComHRD *hrd = vui->getHrdParameters();
478#endif
479
480  WRITE_UVLC( sei.m_bpSeqParameterSetId, "bp_seq_parameter_set_id" );
481  if( !hrd->getSubPicCpbParamsPresentFlag() )
482  {
483    WRITE_FLAG( sei.m_rapCpbParamsPresentFlag, "irap_cpb_params_present_flag" );
484  }
485  if( sei.m_rapCpbParamsPresentFlag )
486  {
487    WRITE_CODE( sei.m_cpbDelayOffset, hrd->getCpbRemovalDelayLengthMinus1() + 1, "cpb_delay_offset" );
488    WRITE_CODE( sei.m_dpbDelayOffset, hrd->getDpbOutputDelayLengthMinus1()  + 1, "dpb_delay_offset" );
489  }
490  WRITE_FLAG( sei.m_concatenationFlag, "concatenation_flag");
491  WRITE_CODE( sei.m_auCpbRemovalDelayDelta - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay_delta_minus1" );
492  for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
493  {
494    if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) ||
495        ( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
496    {
497      for( i = 0; i < ( hrd->getCpbCntMinus1( 0 ) + 1 ); i ++ )
498      {
499        WRITE_CODE( sei.m_initialCpbRemovalDelay[i][nalOrVcl],( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ) ,           "initial_cpb_removal_delay" );
500        WRITE_CODE( sei.m_initialCpbRemovalDelayOffset[i][nalOrVcl],( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ),      "initial_cpb_removal_delay_offset" );
501        if( hrd->getSubPicCpbParamsPresentFlag() || sei.m_rapCpbParamsPresentFlag )
502        {
503          WRITE_CODE( sei.m_initialAltCpbRemovalDelay[i][nalOrVcl], ( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ) ,     "initial_alt_cpb_removal_delay" );
504          WRITE_CODE( sei.m_initialAltCpbRemovalDelayOffset[i][nalOrVcl], ( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ),"initial_alt_cpb_removal_delay_offset" );
505        }
506      }
507    }
508  }
509#if P0138_USE_ALT_CPB_PARAMS_FLAG
510  if (sei.m_useAltCpbParamsFlagPresent)
511  {
512    WRITE_FLAG( sei.m_useAltCpbParamsFlag, "use_alt_cpb_params_flag");
513  }
514#endif
515}
516#if VPS_VUI_BSP_HRD_PARAMS
517Void SEIWriter::xWriteSEIPictureTiming(const SEIPictureTiming& sei,  TComSPS *sps, const SEIScalableNesting* nestingSei, const SEIBspNesting* bspNestingSei, TComVPS *vps)
518#else
519Void SEIWriter::xWriteSEIPictureTiming(const SEIPictureTiming& sei,  TComSPS *sps)
520#endif
521{
522  Int i;
523#if VPS_VUI_BSP_HRD_PARAMS
524  TComHRD *hrd;   
525  TComVUI *vui = sps->getVuiParameters(); 
526  if( bspNestingSei )   // If BP SEI contained inside a BSP nesting SEI message
527  {
528    assert( nestingSei );
529    Int psIdx = bspNestingSei->m_seiPartitioningSchemeIdx;
530    Int seiOlsIdx = bspNestingSei->m_seiOlsIdx;
531    Int maxTemporalId = nestingSei->m_nestingMaxTemporalIdPlus1[0] - 1;
532    Int maxValues = vps->getNumBspSchedulesMinus1(seiOlsIdx, psIdx, maxTemporalId) + 1;
533    std::vector<Int> hrdIdx(maxValues, 0);
534    std::vector<TComHRD *> hrdVec;
535    std::vector<Int> syntaxElemLen(maxValues, 0);
536    for(i = 0; i < maxValues; i++)
537    {
538      hrdIdx[i] = vps->getBspHrdIdx( seiOlsIdx, psIdx, maxTemporalId, i, bspNestingSei->m_bspIdx);
539      hrdVec.push_back(vps->getBspHrd(hrdIdx[i]));
540   
541      syntaxElemLen[i] = hrdVec[i]->getInitialCpbRemovalDelayLengthMinus1() + 1;
542      if ( !(hrdVec[i]->getNalHrdParametersPresentFlag() || hrdVec[i]->getVclHrdParametersPresentFlag()) )
543      {
544        assert( syntaxElemLen[i] == 24 ); // Default of value init_cpb_removal_delay_length_minus1 is 23
545      }
546      if( i > 0 )
547      {
548        assert( hrdVec[i]->getSubPicCpbParamsPresentFlag()    == hrdVec[i-1]->getSubPicCpbParamsPresentFlag() );
549        assert( hrdVec[i]->getSubPicCpbParamsInPicTimingSEIFlag()   == hrdVec[i-1]->getSubPicCpbParamsInPicTimingSEIFlag() );
550        assert( hrdVec[i]->getCpbRemovalDelayLengthMinus1()  == hrdVec[i-1]->getCpbRemovalDelayLengthMinus1() );
551        assert( hrdVec[i]->getDpbOutputDelayLengthMinus1()  == hrdVec[i-1]->getDpbOutputDelayLengthMinus1() );
552        assert( hrdVec[i]->getDpbOutputDelayDuLengthMinus1()  == hrdVec[i-1]->getDpbOutputDelayDuLengthMinus1() );
553        assert( hrdVec[i]->getDuCpbRemovalDelayLengthMinus1()  == hrdVec[i-1]->getDuCpbRemovalDelayLengthMinus1() );
554        // To be done: Check CpbDpbDelaysPresentFlag
555      }
556    }
557    hrd = hrdVec[0];
558  }
559  else
560  {
561    hrd = vui->getHrdParameters();
562  }
563  // To be done: When contained in an BSP HRD SEI message, the hrd structure is to be chosen differently.
564#else
565  TComVUI *vui = sps->getVuiParameters();
566  TComHRD *hrd = vui->getHrdParameters();
567#endif
568
569  if( vui->getFrameFieldInfoPresentFlag() )
570  {
571    WRITE_CODE( sei.m_picStruct, 4,              "pic_struct" );
572    WRITE_CODE( sei.m_sourceScanType, 2,         "source_scan_type" );
573    WRITE_FLAG( sei.m_duplicateFlag ? 1 : 0,     "duplicate_flag" );
574  }
575
576  if( hrd->getCpbDpbDelaysPresentFlag() )
577  {
578    WRITE_CODE( sei.m_auCpbRemovalDelay - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ),                                         "au_cpb_removal_delay_minus1" );
579    WRITE_CODE( sei.m_picDpbOutputDelay, ( hrd->getDpbOutputDelayLengthMinus1() + 1 ),                                          "pic_dpb_output_delay" );
580    if(hrd->getSubPicCpbParamsPresentFlag())
581    {
582      WRITE_CODE(sei.m_picDpbOutputDuDelay, hrd->getDpbOutputDelayDuLengthMinus1()+1, "pic_dpb_output_du_delay" );
583    }
584    if( hrd->getSubPicCpbParamsPresentFlag() && hrd->getSubPicCpbParamsInPicTimingSEIFlag() )
585    {
586      WRITE_UVLC( sei.m_numDecodingUnitsMinus1,     "num_decoding_units_minus1" );
587      WRITE_FLAG( sei.m_duCommonCpbRemovalDelayFlag, "du_common_cpb_removal_delay_flag" );
588      if( sei.m_duCommonCpbRemovalDelayFlag )
589      {
590        WRITE_CODE( sei.m_duCommonCpbRemovalDelayMinus1, ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ),                       "du_common_cpb_removal_delay_minus1" );
591      }
592      for( i = 0; i <= sei.m_numDecodingUnitsMinus1; i ++ )
593      {
594        WRITE_UVLC( sei.m_numNalusInDuMinus1[ i ],  "num_nalus_in_du_minus1");
595        if( ( !sei.m_duCommonCpbRemovalDelayFlag ) && ( i < sei.m_numDecodingUnitsMinus1 ) )
596        {
597          WRITE_CODE( sei.m_duCpbRemovalDelayMinus1[ i ], ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ),                        "du_cpb_removal_delay_minus1" );
598        }
599      }
600    }
601  }
602}
603Void SEIWriter::xWriteSEIRecoveryPoint(const SEIRecoveryPoint& sei)
604{
605  WRITE_SVLC( sei.m_recoveryPocCnt,    "recovery_poc_cnt"    );
606  WRITE_FLAG( sei.m_exactMatchingFlag, "exact_matching_flag" );
607  WRITE_FLAG( sei.m_brokenLinkFlag,    "broken_link_flag"    );
608}
609Void SEIWriter::xWriteSEIFramePacking(const SEIFramePacking& sei)
610{
611  WRITE_UVLC( sei.m_arrangementId,                  "frame_packing_arrangement_id" );
612  WRITE_FLAG( sei.m_arrangementCancelFlag,          "frame_packing_arrangement_cancel_flag" );
613
614  if( sei.m_arrangementCancelFlag == 0 ) {
615    WRITE_CODE( sei.m_arrangementType, 7,           "frame_packing_arrangement_type" );
616
617    WRITE_FLAG( sei.m_quincunxSamplingFlag,         "quincunx_sampling_flag" );
618    WRITE_CODE( sei.m_contentInterpretationType, 6, "content_interpretation_type" );
619    WRITE_FLAG( sei.m_spatialFlippingFlag,          "spatial_flipping_flag" );
620    WRITE_FLAG( sei.m_frame0FlippedFlag,            "frame0_flipped_flag" );
621    WRITE_FLAG( sei.m_fieldViewsFlag,               "field_views_flag" );
622    WRITE_FLAG( sei.m_currentFrameIsFrame0Flag,     "current_frame_is_frame0_flag" );
623
624    WRITE_FLAG( sei.m_frame0SelfContainedFlag,      "frame0_self_contained_flag" );
625    WRITE_FLAG( sei.m_frame1SelfContainedFlag,      "frame1_self_contained_flag" );
626
627    if(sei.m_quincunxSamplingFlag == 0 && sei.m_arrangementType != 5)
628    {
629      WRITE_CODE( sei.m_frame0GridPositionX, 4,     "frame0_grid_position_x" );
630      WRITE_CODE( sei.m_frame0GridPositionY, 4,     "frame0_grid_position_y" );
631      WRITE_CODE( sei.m_frame1GridPositionX, 4,     "frame1_grid_position_x" );
632      WRITE_CODE( sei.m_frame1GridPositionY, 4,     "frame1_grid_position_y" );
633    }
634
635    WRITE_CODE( sei.m_arrangementReservedByte, 8,   "frame_packing_arrangement_reserved_byte" );
636    WRITE_FLAG( sei.m_arrangementPersistenceFlag,   "frame_packing_arrangement_persistence_flag" );
637  }
638
639  WRITE_FLAG( sei.m_upsampledAspectRatio,           "upsampled_aspect_ratio" );
640}
641
642Void SEIWriter::xWriteSEISegmentedRectFramePacking(const SEISegmentedRectFramePacking& sei)
643{
644  WRITE_FLAG( sei.m_arrangementCancelFlag,          "segmented_rect_frame_packing_arrangement_cancel_flag" );
645  if( sei.m_arrangementCancelFlag == 0 ) 
646  {
647    WRITE_CODE( sei.m_contentInterpretationType, 2, "segmented_rect_content_interpretation_type" );
648    WRITE_FLAG( sei.m_arrangementPersistenceFlag,   "segmented_rect_frame_packing_arrangement_persistence" );
649  }
650}
651
652Void SEIWriter::xWriteSEIToneMappingInfo(const SEIToneMappingInfo& sei)
653{
654  Int i;
655  WRITE_UVLC( sei.m_toneMapId,                    "tone_map_id" );
656  WRITE_FLAG( sei.m_toneMapCancelFlag,            "tone_map_cancel_flag" );
657  if( !sei.m_toneMapCancelFlag )
658  {
659    WRITE_FLAG( sei.m_toneMapPersistenceFlag,     "tone_map_persistence_flag" );
660    WRITE_CODE( sei.m_codedDataBitDepth,    8,    "coded_data_bit_depth" );
661    WRITE_CODE( sei.m_targetBitDepth,       8,    "target_bit_depth" );
662    WRITE_UVLC( sei.m_modelId,                    "model_id" );
663    switch(sei.m_modelId)
664    {
665    case 0:
666      {
667        WRITE_CODE( sei.m_minValue,  32,        "min_value" );
668        WRITE_CODE( sei.m_maxValue, 32,         "max_value" );
669        break;
670      }
671    case 1:
672      {
673        WRITE_CODE( sei.m_sigmoidMidpoint, 32,  "sigmoid_midpoint" );
674        WRITE_CODE( sei.m_sigmoidWidth,    32,  "sigmoid_width"    );
675        break;
676      }
677    case 2:
678      {
679        UInt num = 1u << sei.m_targetBitDepth;
680        for(i = 0; i < num; i++)
681        {
682          WRITE_CODE( sei.m_startOfCodedInterval[i], (( sei.m_codedDataBitDepth + 7 ) >> 3 ) << 3,  "start_of_coded_interval" );
683        }
684        break;
685      }
686    case 3:
687      {
688        WRITE_CODE( sei.m_numPivots, 16,          "num_pivots" );
689        for(i = 0; i < sei.m_numPivots; i++ )
690        {
691          WRITE_CODE( sei.m_codedPivotValue[i], (( sei.m_codedDataBitDepth + 7 ) >> 3 ) << 3,       "coded_pivot_value" );
692          WRITE_CODE( sei.m_targetPivotValue[i], (( sei.m_targetBitDepth + 7 ) >> 3 ) << 3,         "target_pivot_value");
693        }
694        break;
695      }
696    case 4:
697      {
698        WRITE_CODE( sei.m_cameraIsoSpeedIdc,    8,    "camera_iso_speed_idc" );
699        if( sei.m_cameraIsoSpeedIdc == 255) //Extended_ISO
700        {
701          WRITE_CODE( sei.m_cameraIsoSpeedValue,    32,    "camera_iso_speed_value" );
702        }
703        WRITE_CODE( sei.m_exposureIndexIdc,     8,    "exposure_index_idc" );
704        if( sei.m_exposureIndexIdc == 255) //Extended_ISO
705        {
706          WRITE_CODE( sei.m_exposureIndexValue,     32,    "exposure_index_value" );
707        }
708        WRITE_FLAG( sei.m_exposureCompensationValueSignFlag,           "exposure_compensation_value_sign_flag" );
709        WRITE_CODE( sei.m_exposureCompensationValueNumerator,     16,  "exposure_compensation_value_numerator" );
710        WRITE_CODE( sei.m_exposureCompensationValueDenomIdc,      16,  "exposure_compensation_value_denom_idc" );
711        WRITE_CODE( sei.m_refScreenLuminanceWhite,                32,  "ref_screen_luminance_white" );
712        WRITE_CODE( sei.m_extendedRangeWhiteLevel,                32,  "extended_range_white_level" );
713        WRITE_CODE( sei.m_nominalBlackLevelLumaCodeValue,         16,  "nominal_black_level_luma_code_value" );
714        WRITE_CODE( sei.m_nominalWhiteLevelLumaCodeValue,         16,  "nominal_white_level_luma_code_value" );
715        WRITE_CODE( sei.m_extendedWhiteLevelLumaCodeValue,        16,  "extended_white_level_luma_code_value" );
716        break;
717      }
718    default:
719      {
720        assert(!"Undefined SEIToneMapModelId");
721        break;
722      }
723    }//switch m_modelId
724  }//if(!sei.m_toneMapCancelFlag)
725}
726
727Void SEIWriter::xWriteSEIDisplayOrientation(const SEIDisplayOrientation &sei)
728{
729  WRITE_FLAG( sei.cancelFlag,           "display_orientation_cancel_flag" );
730  if( !sei.cancelFlag )
731  {
732    WRITE_FLAG( sei.horFlip,                   "hor_flip" );
733    WRITE_FLAG( sei.verFlip,                   "ver_flip" );
734    WRITE_CODE( sei.anticlockwiseRotation, 16, "anticlockwise_rotation" );
735    WRITE_FLAG( sei.persistenceFlag,          "display_orientation_persistence_flag" );
736  }
737}
738
739Void SEIWriter::xWriteSEITemporalLevel0Index(const SEITemporalLevel0Index &sei)
740{
741  WRITE_CODE( sei.tl0Idx, 8 , "tl0_idx" );
742  WRITE_CODE( sei.rapIdx, 8 , "rap_idx" );
743}
744
745Void SEIWriter::xWriteSEIGradualDecodingRefreshInfo(const SEIGradualDecodingRefreshInfo &sei)
746{
747  WRITE_FLAG( sei.m_gdrForegroundFlag, "gdr_foreground_flag");
748}
749
750Void SEIWriter::xWriteSEINoDisplay(const SEINoDisplay &sei)
751{
752}
753
754Void SEIWriter::xWriteSEISOPDescription(const SEISOPDescription& sei)
755{
756  WRITE_UVLC( sei.m_sopSeqParameterSetId,           "sop_seq_parameter_set_id"               );
757  WRITE_UVLC( sei.m_numPicsInSopMinus1,             "num_pics_in_sop_minus1"               );
758  for (UInt i = 0; i <= sei.m_numPicsInSopMinus1; i++)
759  {
760    WRITE_CODE( sei.m_sopDescVclNaluType[i], 6, "sop_desc_vcl_nalu_type" );
761    WRITE_CODE( sei.m_sopDescTemporalId[i],  3, "sop_desc_temporal_id" );
762    if (sei.m_sopDescVclNaluType[i] != NAL_UNIT_CODED_SLICE_IDR_W_RADL && sei.m_sopDescVclNaluType[i] != NAL_UNIT_CODED_SLICE_IDR_N_LP)
763    {
764      WRITE_UVLC( sei.m_sopDescStRpsIdx[i],           "sop_desc_st_rps_idx"               );
765    }
766    if (i > 0)
767    {
768      WRITE_SVLC( sei.m_sopDescPocDelta[i],           "sop_desc_poc_delta"               );
769    }
770  }
771}
772
773#if O0164_MULTI_LAYER_HRD
774Void SEIWriter::xWriteSEIScalableNesting(TComBitIf& bs, const SEIScalableNesting& sei, TComVPS *vps, TComSPS *sps)
775#else
776Void SEIWriter::xWriteSEIScalableNesting(TComBitIf& bs, const SEIScalableNesting& sei, TComSPS *sps)
777#endif
778{
779  WRITE_FLAG( sei.m_bitStreamSubsetFlag,             "bitstream_subset_flag"         );
780  WRITE_FLAG( sei.m_nestingOpFlag,                   "nesting_op_flag      "         );
781  if (sei.m_nestingOpFlag)
782  {
783    WRITE_FLAG( sei.m_defaultOpFlag,                 "default_op_flag"               );
784    WRITE_UVLC( sei.m_nestingNumOpsMinus1,           "nesting_num_ops_minus1"        );
785    for (UInt i = (sei.m_defaultOpFlag ? 1 : 0); i <= sei.m_nestingNumOpsMinus1; i++)
786    {
787      WRITE_CODE( sei.m_nestingMaxTemporalIdPlus1[i], 3,  "nesting_max_temporal_id_plus1" );
788      WRITE_UVLC( sei.m_nestingOpIdx[i],                  "nesting_op_idx"                );
789    }
790  }
791  else
792  {
793    WRITE_FLAG( sei.m_allLayersFlag,                      "all_layers_flag"               );
794    if (!sei.m_allLayersFlag)
795    {
796      WRITE_CODE( sei.m_nestingNoOpMaxTemporalIdPlus1, 3, "nesting_no_op_max_temporal_id_plus1" );
797      WRITE_UVLC( sei.m_nestingNumLayersMinus1,           "nesting_num_layers"                  );
798      for (UInt i = 0; i <= sei.m_nestingNumLayersMinus1; i++)
799      {
800        WRITE_CODE( sei.m_nestingLayerId[i], 6,           "nesting_layer_id"              );
801      }
802    }
803  }
804
805  // byte alignment
806  while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
807  {
808    WRITE_FLAG( 0, "nesting_zero_bit" );
809  }
810
811  // write nested SEI messages
812  for (SEIMessages::const_iterator it = sei.m_nestedSEIs.begin(); it != sei.m_nestedSEIs.end(); it++)
813  {
814#if O0164_MULTI_LAYER_HRD
815    writeSEImessage(bs, *(*it), vps, sps, &sei);
816#else
817    writeSEImessage(bs, *(*it), sps);
818#endif
819  }
820}
821
822Void SEIWriter::xWriteSEITempMotionConstrainedTileSets(TComBitIf& bs, const SEITempMotionConstrainedTileSets& sei)
823{
824  //UInt code;
825  WRITE_FLAG((sei.m_mc_all_tiles_exact_sample_value_match_flag ? 1 : 0), "mc_all_tiles_exact_sample_value_match_flag"); 
826  WRITE_FLAG((sei.m_each_tile_one_tile_set_flag                ? 1 : 0), "each_tile_one_tile_set_flag"               );
827
828  if(!sei.m_each_tile_one_tile_set_flag)
829  {
830    WRITE_FLAG((sei.m_limited_tile_set_display_flag ? 1 : 0), "limited_tile_set_display_flag");
831    WRITE_UVLC((sei.getNumberOfTileSets() - 1),               "num_sets_in_message_minus1"   );
832
833    if(sei.getNumberOfTileSets() > 0)
834    {
835      for(Int i = 0; i < sei.getNumberOfTileSets(); i++)
836      {
837        WRITE_UVLC(sei.tileSetData(i).m_mcts_id, "mcts_id");
838
839        if(sei.m_limited_tile_set_display_flag)
840        { 
841          WRITE_FLAG((sei.tileSetData(i).m_display_tile_set_flag ? 1 : 0), "display_tile_set_flag"); 
842        }
843
844        WRITE_UVLC((sei.tileSetData(i).getNumberOfTileRects() - 1), "num_tile_rects_in_set_minus1"); 
845       
846        for(Int j = 0; j < sei.tileSetData(i).getNumberOfTileRects(); j++)
847        {
848          WRITE_UVLC(sei.tileSetData(i).topLeftTileIndex    (j), "top_left_tile_index"); 
849          WRITE_UVLC(sei.tileSetData(i).bottomRightTileIndex(j), "bottom_right_tile_index"); 
850        }
851
852        if(!sei.m_mc_all_tiles_exact_sample_value_match_flag)
853        {
854          WRITE_FLAG((sei.tileSetData(i).m_exact_sample_value_match_flag ? 1 : 0), "exact_sample_value_match_flag"); 
855        }
856
857        WRITE_FLAG((sei.tileSetData(i).m_mcts_tier_level_idc_present_flag ? 1 : 0), "mcts_tier_level_idc_present_flag");
858
859        if(sei.tileSetData(i).m_mcts_tier_level_idc_present_flag)
860        {
861          WRITE_FLAG((sei.tileSetData(i).m_mcts_tier_flag ? 1 : 0), "mcts_tier_flag");
862          WRITE_CODE( sei.tileSetData(i).m_mcts_level_idc, 8,       "mcts_level_idc"); 
863        }
864      }
865    }
866  }
867  else
868  {
869    WRITE_FLAG((sei.m_max_mcs_tier_level_idc_present_flag ? 1 : 0), "max_mcs_tier_level_idc_present_flag");
870
871    if(sei.m_max_mcs_tier_level_idc_present_flag)
872    {
873      WRITE_FLAG((sei.m_max_mcts_tier_flag ? 1 : 0), "max_mcts_tier_flag"); 
874      WRITE_CODE( sei.m_max_mcts_level_idc, 8,       "max_mcts_level_idc"); 
875    }
876  }
877}
878
879Void SEIWriter::xWriteSEITimeCode(const SEITimeCode& sei)
880{
881  WRITE_CODE(sei.numClockTs, 2, "num_clock_ts");
882  for(Int i = 0; i < sei.numClockTs; i++)
883  {
884    const TComSEITimeSet &currentTimeSet = sei.timeSetArray[i];
885    WRITE_FLAG(currentTimeSet.clockTimeStampFlag, "clock_time_stamp_flag");
886    if(currentTimeSet.clockTimeStampFlag)
887    {
888      WRITE_FLAG(currentTimeSet.numUnitFieldBasedFlag, "units_field_based_flag");
889      WRITE_CODE(currentTimeSet.countingType, 5, "counting_type");
890      WRITE_FLAG(currentTimeSet.fullTimeStampFlag, "full_timestamp_flag");
891      WRITE_FLAG(currentTimeSet.discontinuityFlag, "discontinuity_flag");
892      WRITE_FLAG(currentTimeSet.cntDroppedFlag, "cnt_dropped_flag");
893      WRITE_CODE(currentTimeSet.numberOfFrames, 9, "n_frames");
894      if(currentTimeSet.fullTimeStampFlag)
895      {
896        WRITE_CODE(currentTimeSet.secondsValue, 6, "seconds_value");
897        WRITE_CODE(currentTimeSet.minutesValue, 6, "minutes_value");
898        WRITE_CODE(currentTimeSet.hoursValue, 5, "hours_value");
899      }
900      else
901      {
902        WRITE_FLAG(currentTimeSet.secondsFlag, "seconds_flag");
903        if(currentTimeSet.secondsFlag)
904        {
905          WRITE_CODE(currentTimeSet.secondsValue, 6, "seconds_value");
906          WRITE_FLAG(currentTimeSet.minutesFlag, "minutes_flag");
907          if(currentTimeSet.minutesFlag)
908          {
909            WRITE_CODE(currentTimeSet.minutesValue, 6, "minutes_value");
910            WRITE_FLAG(currentTimeSet.hoursFlag, "hours_flag");
911            if(currentTimeSet.hoursFlag)
912              WRITE_CODE(currentTimeSet.hoursValue, 5, "hours_value");
913          }
914        }
915      }
916      WRITE_CODE(currentTimeSet.timeOffsetLength, 5, "time_offset_length");
917      if(currentTimeSet.timeOffsetLength > 0)
918      {
919        if(currentTimeSet.timeOffsetValue >= 0)
920        {
921          WRITE_CODE((UInt)currentTimeSet.timeOffsetValue, currentTimeSet.timeOffsetLength, "time_offset_value");
922        }
923        else
924        {
925          //  Two's complement conversion
926          UInt offsetValue = ~(currentTimeSet.timeOffsetValue) + 1;
927          offsetValue |= (1 << (currentTimeSet.timeOffsetLength-1));
928          WRITE_CODE(offsetValue, currentTimeSet.timeOffsetLength, "time_offset_value");
929        }
930      }
931    }
932  }
933}
934
935Void SEIWriter::xWriteSEIChromaSamplingFilterHint(const SEIChromaSamplingFilterHint &sei/*, TComSPS* sps*/)
936{
937  WRITE_CODE(sei.m_verChromaFilterIdc, 8, "ver_chroma_filter_idc");
938  WRITE_CODE(sei.m_horChromaFilterIdc, 8, "hor_chroma_filter_idc");
939  WRITE_FLAG(sei.m_verFilteringProcessFlag, "ver_filtering_process_flag");
940  if(sei.m_verChromaFilterIdc == 1 || sei.m_horChromaFilterIdc == 1)
941  {
942    writeUserDefinedCoefficients(sei);
943  }
944}
945
946// write hardcoded chroma filter coefficients in the SEI messages
947Void SEIWriter::writeUserDefinedCoefficients(const SEIChromaSamplingFilterHint &sei)
948{
949  Int const iNumVerticalFilters = 3;
950  Int verticalTapLength_minus1[iNumVerticalFilters] = {5,3,3};
951  Int* userVerticalCoefficients[iNumVerticalFilters];
952  for(Int i = 0; i < iNumVerticalFilters; i ++)
953  {
954    userVerticalCoefficients[i] = (Int*)malloc( (verticalTapLength_minus1[i]+1) * sizeof(Int));
955  }
956  userVerticalCoefficients[0][0] = -3;
957  userVerticalCoefficients[0][1] = 13;
958  userVerticalCoefficients[0][2] = 31;
959  userVerticalCoefficients[0][3] = 23;
960  userVerticalCoefficients[0][4] = 3;
961  userVerticalCoefficients[0][5] = -3;
962
963  userVerticalCoefficients[1][0] = -1;
964  userVerticalCoefficients[1][1] = 25;
965  userVerticalCoefficients[1][2] = 247;
966  userVerticalCoefficients[1][3] = -15;
967
968  userVerticalCoefficients[2][0] = -20;
969  userVerticalCoefficients[2][1] = 186;
970  userVerticalCoefficients[2][2] = 100;
971  userVerticalCoefficients[2][3] = -10;
972 
973  Int const iNumHorizontalFilters = 1;
974  Int horizontalTapLength_minus1[iNumHorizontalFilters] = {3};
975  Int* userHorizontalCoefficients[iNumHorizontalFilters];
976  for(Int i = 0; i < iNumHorizontalFilters; i ++)
977  {
978    userHorizontalCoefficients[i] = (Int*)malloc( (horizontalTapLength_minus1[i]+1) * sizeof(Int));
979  }
980  userHorizontalCoefficients[0][0] = 1;
981  userHorizontalCoefficients[0][1] = 6;
982  userHorizontalCoefficients[0][2] = 1;
983
984  WRITE_UVLC(3, "target_format_idc");
985  if(sei.m_verChromaFilterIdc == 1)
986  {
987    WRITE_UVLC(iNumVerticalFilters, "num_vertical_filters");
988    if(iNumVerticalFilters > 0)
989    {
990      for(Int i = 0; i < iNumVerticalFilters; i ++)
991      {
992        WRITE_UVLC(verticalTapLength_minus1[i], "ver_tap_length_minus_1");
993        for(Int j = 0; j < verticalTapLength_minus1[i]; j ++)
994        {
995          WRITE_SVLC(userVerticalCoefficients[i][j], "ver_filter_coeff");
996        }
997      }
998    }
999  }
1000  if(sei.m_horChromaFilterIdc == 1)
1001  {
1002    WRITE_UVLC(iNumHorizontalFilters, "num_horizontal_filters");
1003    if(iNumHorizontalFilters > 0)
1004    {
1005      for(Int i = 0; i < iNumHorizontalFilters; i ++)
1006      {
1007        WRITE_UVLC(horizontalTapLength_minus1[i], "hor_tap_length_minus_1");
1008        for(Int j = 0; j < horizontalTapLength_minus1[i]; j ++)
1009        {
1010          WRITE_SVLC(userHorizontalCoefficients[i][j], "hor_filter_coeff");
1011        }
1012      }
1013    }
1014  }
1015}
1016
1017Void SEIWriter::xWriteSEIKneeFunctionInfo(const SEIKneeFunctionInfo &sei)
1018{
1019  WRITE_UVLC( sei.m_kneeId, "knee_function_id" );
1020  WRITE_FLAG( sei.m_kneeCancelFlag, "knee_function_cancel_flag" ); 
1021  if ( !sei.m_kneeCancelFlag )
1022  {
1023    WRITE_FLAG( sei.m_kneePersistenceFlag, "knee_function_persistence_flag" );
1024    WRITE_CODE( (UInt)sei.m_kneeInputDrange , 32,  "input_d_range" );
1025    WRITE_CODE( (UInt)sei.m_kneeInputDispLuminance, 32,  "input_disp_luminance" );
1026    WRITE_CODE( (UInt)sei.m_kneeOutputDrange, 32,  "output_d_range" );
1027    WRITE_CODE( (UInt)sei.m_kneeOutputDispLuminance, 32,  "output_disp_luminance" );
1028    WRITE_UVLC( sei.m_kneeNumKneePointsMinus1, "num_knee_points_minus1" );
1029    for(Int i = 0; i <= sei.m_kneeNumKneePointsMinus1; i++ )
1030    {
1031      WRITE_CODE( (UInt)sei.m_kneeInputKneePoint[i], 10,"input_knee_point" );
1032      WRITE_CODE( (UInt)sei.m_kneeOutputKneePoint[i], 10, "output_knee_point" );
1033    }
1034  }
1035}
1036
1037
1038Void SEIWriter::xWriteSEIMasteringDisplayColourVolume(const SEIMasteringDisplayColourVolume& sei)
1039{
1040  WRITE_CODE( sei.values.primaries[0][0],  16,  "display_primaries_x[0]" );
1041  WRITE_CODE( sei.values.primaries[0][1],  16,  "display_primaries_y[0]" );
1042
1043  WRITE_CODE( sei.values.primaries[1][0],  16,  "display_primaries_x[1]" );
1044  WRITE_CODE( sei.values.primaries[1][1],  16,  "display_primaries_y[1]" );
1045
1046  WRITE_CODE( sei.values.primaries[2][0],  16,  "display_primaries_x[2]" );
1047  WRITE_CODE( sei.values.primaries[2][1],  16,  "display_primaries_y[2]" );
1048
1049  WRITE_CODE( sei.values.whitePoint[0],    16,  "white_point_x" );
1050  WRITE_CODE( sei.values.whitePoint[1],    16,  "white_point_y" );
1051   
1052  WRITE_CODE( sei.values.maxLuminance,     32,  "max_display_mastering_luminance" );
1053  WRITE_CODE( sei.values.minLuminance,     32,  "min_display_mastering_luminance" );
1054}
1055
1056
1057Void SEIWriter::xWriteByteAlign()
1058{
1059  if( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0)
1060  {
1061    WRITE_FLAG( 1, "payload_bit_equal_to_one" );
1062    while( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
1063    {
1064      WRITE_FLAG( 0, "payload_bit_equal_to_zero" );
1065    }
1066  }
1067}
1068
1069#if SVC_EXTENSION
1070#if LAYERS_NOT_PRESENT_SEI
1071Void SEIWriter::xWriteSEILayersNotPresent(const SEILayersNotPresent& sei)
1072{
1073  WRITE_UVLC( sei.m_activeVpsId,           "lp_sei_active_vps_id" );
1074  for (UInt i = 0; i < sei.m_vpsMaxLayers; i++)
1075  {
1076    WRITE_FLAG( sei.m_layerNotPresentFlag[i], "layer_not_present_flag"   );
1077  } 
1078}
1079#endif
1080
1081#if N0383_IL_CONSTRAINED_TILE_SETS_SEI
1082Void SEIWriter::xWriteSEIInterLayerConstrainedTileSets(const SEIInterLayerConstrainedTileSets& sei)
1083{
1084  WRITE_FLAG( sei.m_ilAllTilesExactSampleValueMatchFlag,  "il_all_tiles_exact_sample_value_match_flag"   );
1085  WRITE_FLAG( sei.m_ilOneTilePerTileSetFlag,              "il_one_tile_per_tile_set_flag"                );
1086  if( !sei.m_ilOneTilePerTileSetFlag )
1087  {
1088    WRITE_UVLC( sei.m_ilNumSetsInMessageMinus1,             "il_num_sets_in_message_minus1"                );
1089    if( sei.m_ilNumSetsInMessageMinus1 )
1090    {
1091      WRITE_FLAG( sei.m_skippedTileSetPresentFlag,            "skipped_tile_set_present_flag"                );
1092    }
1093    UInt numSignificantSets = sei.m_ilNumSetsInMessageMinus1 - (sei.m_skippedTileSetPresentFlag ? 1 : 0) + 1;
1094    for( UInt i = 0; i < numSignificantSets; i++ )
1095    {
1096      WRITE_UVLC( sei.m_ilctsId[i],                           "ilcts_id"                                     );
1097      WRITE_UVLC( sei.m_ilNumTileRectsInSetMinus1[i],         "il_num_tile_rects_in_set_minus1"              );
1098      for( UInt j = 0; j <= sei.m_ilNumTileRectsInSetMinus1[i]; j++ )
1099      {
1100        WRITE_UVLC( sei.m_ilTopLeftTileIndex[i][j],             "il_top_left_tile_index"                       );
1101        WRITE_UVLC( sei.m_ilBottomRightTileIndex[i][j],         "il_bottom_right_tile_index"                   );
1102      }
1103      WRITE_CODE( sei.m_ilcIdc[i], 2,                         "ilc_idc"                                      );
1104      if( sei.m_ilAllTilesExactSampleValueMatchFlag )
1105      {
1106        WRITE_FLAG( sei.m_ilExactSampleValueMatchFlag[i],        "il_exact_sample_value_match_flag"            );
1107      }
1108    }
1109  }
1110  else
1111  {
1112    WRITE_CODE( sei.m_allTilesIlcIdc, 2,                    "all_tiles_ilc_idc"                          );
1113  }
1114}
1115#endif
1116#if SUB_BITSTREAM_PROPERTY_SEI
1117Void SEIWriter::xWriteSEISubBitstreamProperty(const SEISubBitstreamProperty &sei)
1118{
1119  WRITE_CODE( sei.m_activeVpsId, 4, "active_vps_id" );
1120  assert( sei.m_numAdditionalSubStreams >= 1 );
1121  WRITE_UVLC( sei.m_numAdditionalSubStreams - 1, "num_additional_sub_streams_minus1" );
1122
1123  for( Int i = 0; i < sei.m_numAdditionalSubStreams; i++ )
1124  {
1125    WRITE_CODE( sei.m_subBitstreamMode[i],       2, "sub_bitstream_mode[i]"           );
1126    WRITE_UVLC( sei.m_outputLayerSetIdxToVps[i],    "output_layer_set_idx_to_vps[i]"  );
1127    WRITE_CODE( sei.m_highestSublayerId[i],      3, "highest_sub_layer_id[i]"         );
1128    WRITE_CODE( sei.m_avgBitRate[i],            16, "avg_bit_rate[i]"                 );
1129    WRITE_CODE( sei.m_maxBitRate[i],            16, "max_bit_rate[i]"                 );
1130  } 
1131}
1132#endif
1133
1134#if Q0189_TMVP_CONSTRAINTS
1135Void SEIWriter::xWriteSEITMVPConstraints (const SEITMVPConstrains &sei)
1136{
1137  WRITE_UVLC( sei.prev_pics_not_used_flag ,    "prev_pics_not_used_flag"  );
1138  WRITE_UVLC( sei.no_intra_layer_col_pic_flag ,    "no_intra_layer_col_pic_flag"  ); 
1139}
1140#endif
1141
1142#if Q0247_FRAME_FIELD_INFO
1143Void SEIWriter::xWriteSEIFrameFieldInfo  (const SEIFrameFieldInfo &sei)
1144{
1145  WRITE_CODE( sei.m_ffinfo_picStruct , 4,             "ffinfo_pic_struct" );
1146  WRITE_CODE( sei.m_ffinfo_sourceScanType, 2,         "ffinfo_source_scan_type" );
1147  WRITE_FLAG( sei.m_ffinfo_duplicateFlag ? 1 : 0,     "ffinfo_duplicate_flag" );
1148}
1149#endif
1150
1151#if O0164_MULTI_LAYER_HRD
1152Void SEIWriter::xWriteSEIBspNesting(TComBitIf& bs, const SEIBspNesting &sei, TComVPS *vps, TComSPS *sps, const SEIScalableNesting &nestingSei)
1153{
1154  WRITE_UVLC( sei.m_bspIdx, "bsp_idx" );
1155
1156  while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
1157  {
1158    WRITE_FLAG( 0, "bsp_nesting_zero_bit" );
1159  }
1160#if NESTING_SEI_EXTENSIBILITY
1161  assert( sei.m_nestedSEIs.size() <= MAX_SEIS_IN_BSP_NESTING );
1162  WRITE_UVLC( (UInt)sei.m_nestedSEIs.size(), "num_seis_in_bsp_minus1" );
1163#endif
1164  // write nested SEI messages
1165  for (SEIMessages::const_iterator it = sei.m_nestedSEIs.begin(); it != sei.m_nestedSEIs.end(); it++)
1166  {
1167    writeSEImessage(bs, *(*it), vps, sps, &nestingSei, &sei);
1168  }
1169}
1170
1171Void SEIWriter::xWriteSEIBspInitialArrivalTime(const SEIBspInitialArrivalTime &sei, TComVPS *vps, TComSPS *sps, const SEIScalableNesting &nestingSei, const SEIBspNesting &bspNestingSei)
1172{
1173  assert(vps->getVpsVuiPresentFlag());
1174
1175#if VPS_VUI_BSP_HRD_PARAMS
1176  Int psIdx = bspNestingSei.m_seiPartitioningSchemeIdx;
1177  Int seiOlsIdx = bspNestingSei.m_seiOlsIdx;
1178  Int maxTemporalId = nestingSei.m_nestingMaxTemporalIdPlus1[0] - 1;
1179  Int maxValues = vps->getNumBspSchedulesMinus1(seiOlsIdx, psIdx, maxTemporalId) + 1;
1180  std::vector<Int> hrdIdx(maxValues, 0);
1181  std::vector<TComHRD *> hrdVec;
1182  std::vector<Int> syntaxElemLen(maxValues, 0);
1183  for(Int i = 0; i < maxValues; i++)
1184  {
1185    hrdIdx[i] = vps->getBspHrdIdx( seiOlsIdx, psIdx, maxTemporalId, i, bspNestingSei.m_bspIdx);
1186    hrdVec.push_back(vps->getBspHrd(hrdIdx[i]));
1187   
1188    syntaxElemLen[i] = hrdVec[i]->getInitialCpbRemovalDelayLengthMinus1() + 1;
1189    if ( !(hrdVec[i]->getNalHrdParametersPresentFlag() || hrdVec[i]->getVclHrdParametersPresentFlag()) )
1190    {
1191      assert( syntaxElemLen[i] == 24 ); // Default of value init_cpb_removal_delay_length_minus1 is 23
1192    }
1193    if( i > 0 )
1194    {
1195      assert( hrdVec[i]->getNalHrdParametersPresentFlag() == hrdVec[i-1]->getNalHrdParametersPresentFlag() );
1196      assert( hrdVec[i]->getVclHrdParametersPresentFlag() == hrdVec[i-1]->getVclHrdParametersPresentFlag() );
1197    }
1198  }
1199  if (hrdVec[0]->getNalHrdParametersPresentFlag())
1200  {
1201    for(UInt i = 0; i < maxValues; i++)
1202    {
1203      WRITE_CODE( sei.m_nalInitialArrivalDelay[i], syntaxElemLen[i], "nal_initial_arrival_delay[i]" );
1204    }
1205  }
1206  if( hrdVec[0]->getVclHrdParametersPresentFlag() )
1207  {
1208    for(UInt i = 0; i < maxValues; i++)
1209    {
1210      WRITE_CODE( sei.m_vclInitialArrivalDelay[i], syntaxElemLen[i], "vcl_initial_arrival_delay[i]" );
1211    }
1212  }
1213#else
1214  UInt schedCombCnt = vps->getNumBspSchedCombinations(nestingSei.m_nestingOpIdx[0]);
1215  UInt len;
1216  UInt hrdIdx;
1217
1218  if (schedCombCnt > 0)
1219  {
1220    hrdIdx = vps->getBspCombHrdIdx(nestingSei.m_nestingOpIdx[0], 0, bspNestingSei.m_bspIdx);
1221  }
1222  else
1223  {
1224    hrdIdx = 0;
1225  }
1226
1227  TComHRD *hrd = vps->getBspHrd(hrdIdx);
1228
1229  if (hrd->getNalHrdParametersPresentFlag() || hrd->getVclHrdParametersPresentFlag())
1230  {
1231    len = hrd->getInitialCpbRemovalDelayLengthMinus1() + 1;
1232  }
1233  else
1234  {
1235    len = 23 + 1;
1236  }
1237
1238  if (hrd->getNalHrdParametersPresentFlag())
1239  {
1240    for(UInt i = 0; i < schedCombCnt; i++)
1241    {
1242      WRITE_CODE( sei.m_nalInitialArrivalDelay[i], len, "nal_initial_arrival_delay" );
1243    }
1244  }
1245#if BSP_INIT_ARRIVAL_SEI
1246  if( hrd->getVclHrdParametersPresentFlag() )
1247#else
1248  else
1249#endif
1250  {
1251    for(UInt i = 0; i < schedCombCnt; i++)
1252    {
1253      WRITE_CODE( sei.m_vclInitialArrivalDelay[i], len, "vcl_initial_arrival_delay" );
1254    }
1255  }
1256#endif
1257}
1258
1259Void SEIWriter::xCodeHrdParameters( TComHRD *hrd, Bool commonInfPresentFlag, UInt maxNumSubLayersMinus1 )
1260{
1261  if( commonInfPresentFlag )
1262  {
1263    WRITE_FLAG( hrd->getNalHrdParametersPresentFlag() ? 1 : 0 ,  "nal_hrd_parameters_present_flag" );
1264    WRITE_FLAG( hrd->getVclHrdParametersPresentFlag() ? 1 : 0 ,  "vcl_hrd_parameters_present_flag" );
1265    if( hrd->getNalHrdParametersPresentFlag() || hrd->getVclHrdParametersPresentFlag() )
1266    {
1267      WRITE_FLAG( hrd->getSubPicCpbParamsPresentFlag() ? 1 : 0,  "sub_pic_cpb_params_present_flag" );
1268      if( hrd->getSubPicCpbParamsPresentFlag() )
1269      {
1270        WRITE_CODE( hrd->getTickDivisorMinus2(), 8,              "tick_divisor_minus2" );
1271        WRITE_CODE( hrd->getDuCpbRemovalDelayLengthMinus1(), 5,  "du_cpb_removal_delay_length_minus1" );
1272        WRITE_FLAG( hrd->getSubPicCpbParamsInPicTimingSEIFlag() ? 1 : 0, "sub_pic_cpb_params_in_pic_timing_sei_flag" );
1273        WRITE_CODE( hrd->getDpbOutputDelayDuLengthMinus1(), 5,   "dpb_output_delay_du_length_minus1"  );
1274      }
1275      WRITE_CODE( hrd->getBitRateScale(), 4,                     "bit_rate_scale" );
1276      WRITE_CODE( hrd->getCpbSizeScale(), 4,                     "cpb_size_scale" );
1277      if( hrd->getSubPicCpbParamsPresentFlag() )
1278      {
1279        WRITE_CODE( hrd->getDuCpbSizeScale(), 4,                "du_cpb_size_scale" ); 
1280      }
1281      WRITE_CODE( hrd->getInitialCpbRemovalDelayLengthMinus1(), 5, "initial_cpb_removal_delay_length_minus1" );
1282      WRITE_CODE( hrd->getCpbRemovalDelayLengthMinus1(),        5, "au_cpb_removal_delay_length_minus1" );
1283      WRITE_CODE( hrd->getDpbOutputDelayLengthMinus1(),         5, "dpb_output_delay_length_minus1" );
1284    }
1285  }
1286  Int i, j, nalOrVcl;
1287  for( i = 0; i <= maxNumSubLayersMinus1; i ++ )
1288  {
1289    WRITE_FLAG( hrd->getFixedPicRateFlag( i ) ? 1 : 0,          "fixed_pic_rate_general_flag");
1290    if( !hrd->getFixedPicRateFlag( i ) )
1291    {
1292      WRITE_FLAG( hrd->getFixedPicRateWithinCvsFlag( i ) ? 1 : 0, "fixed_pic_rate_within_cvs_flag");
1293    }
1294    else
1295    {
1296      hrd->setFixedPicRateWithinCvsFlag( i, true );
1297    }
1298    if( hrd->getFixedPicRateWithinCvsFlag( i ) )
1299    {
1300      WRITE_UVLC( hrd->getPicDurationInTcMinus1( i ),           "elemental_duration_in_tc_minus1");
1301    }
1302    else
1303    {
1304      WRITE_FLAG( hrd->getLowDelayHrdFlag( i ) ? 1 : 0,           "low_delay_hrd_flag");
1305    }
1306    if (!hrd->getLowDelayHrdFlag( i ))
1307    {
1308      WRITE_UVLC( hrd->getCpbCntMinus1( i ),                      "cpb_cnt_minus1");
1309    }
1310   
1311    for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
1312    {
1313      if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) ||
1314          ( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
1315      {
1316        for( j = 0; j <= ( hrd->getCpbCntMinus1( i ) ); j ++ )
1317        {
1318          WRITE_UVLC( hrd->getBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_value_minus1");
1319          WRITE_UVLC( hrd->getCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_value_minus1");
1320          if( hrd->getSubPicCpbParamsPresentFlag() )
1321          {
1322            WRITE_UVLC( hrd->getDuCpbSizeValueMinus1( i, j, nalOrVcl ), "cpb_size_du_value_minus1"); 
1323            WRITE_UVLC( hrd->getDuBitRateValueMinus1( i, j, nalOrVcl ), "bit_rate_du_value_minus1");
1324          }
1325          WRITE_FLAG( hrd->getCbrFlag( i, j, nalOrVcl ) ? 1 : 0, "cbr_flag");
1326        }
1327      }
1328    }
1329  }
1330}
1331
1332#endif
1333
1334#if Q0078_ADD_LAYER_SETS
1335
1336Void SEIWriter::xWriteSEIOutputLayerSetNesting(TComBitIf& bs, const SEIOutputLayerSetNesting &sei, TComVPS *vps, TComSPS *sps)
1337{
1338  WRITE_FLAG(sei.m_olsFlag, "ols_flag");
1339  WRITE_UVLC(sei.m_numOlsIndicesMinus1, "num_ols_indices_minus1");
1340
1341  for (Int i = 0; i <= sei.m_numOlsIndicesMinus1; i++)
1342  {
1343    WRITE_UVLC(sei.m_olsIdx[i], "ols_idx[i]");
1344  }
1345
1346  while (m_pcBitIf->getNumberOfWrittenBits() % 8 != 0)
1347  {
1348    WRITE_FLAG(0, "ols_nesting_zero_bit");
1349  }
1350
1351  // write nested SEI messages
1352  for (SEIMessages::const_iterator it = sei.m_nestedSEIs.begin(); it != sei.m_nestedSEIs.end(); it++)
1353  {
1354    writeSEImessage(bs, *(*it), vps, sps);
1355  }
1356}
1357
1358Void SEIWriter::xWriteSEIVPSRewriting(const SEIVPSRewriting &sei)
1359{
1360  //sei.nalu->
1361}
1362
1363#endif
1364
1365#if P0123_ALPHA_CHANNEL_SEI
1366Void SEIWriter::xWriteSEIAlphaChannelInfo(const SEIAlphaChannelInfo &sei)
1367{
1368  WRITE_FLAG(sei.m_alphaChannelCancelFlag, "alpha_channel_cancel_flag");
1369  if(!sei.m_alphaChannelCancelFlag)
1370  {
1371    WRITE_CODE(sei.m_alphaChannelUseIdc, 3, "alpha_channel_use_idc");
1372    WRITE_CODE(sei.m_alphaChannelBitDepthMinus8, 3, "alpha_channel_bit_depth_minus8");
1373    WRITE_CODE(sei.m_alphaTransparentValue, sei.m_alphaChannelBitDepthMinus8 + 9, "alpha_transparent_value");
1374    WRITE_CODE(sei.m_alphaOpaqueValue, sei.m_alphaChannelBitDepthMinus8 + 9, "alpha_opaque_value");
1375    WRITE_FLAG(sei.m_alphaChannelIncrFlag, "alpha_channel_incr_flag");
1376    WRITE_FLAG(sei.m_alphaChannelClipFlag, "alpha_channel_clip_flag");
1377    if(sei.m_alphaChannelClipFlag)
1378    {
1379      WRITE_FLAG(sei.m_alphaChannelClipTypeFlag, "alpha_channel_clip_type_flag");
1380    }
1381  }
1382  xWriteByteAlign();
1383}
1384#endif
1385
1386#if Q0096_OVERLAY_SEI
1387Void SEIWriter::xWriteSEIOverlayInfo(const SEIOverlayInfo &sei)
1388{
1389  Int i,j;
1390  WRITE_FLAG( sei.m_overlayInfoCancelFlag, "overlay_info_cancel_flag" );
1391  if ( !sei.m_overlayInfoCancelFlag )
1392  {
1393    WRITE_UVLC( sei.m_overlayContentAuxIdMinus128, "overlay_content_aux_id_minus128" );
1394    WRITE_UVLC( sei.m_overlayLabelAuxIdMinus128, "overlay_label_aux_id_minus128" );
1395    WRITE_UVLC( sei.m_overlayAlphaAuxIdMinus128, "overlay_alpha_aux_id_minus128" );
1396    WRITE_UVLC( sei.m_overlayElementLabelValueLengthMinus8, "overlay_element_label_value_length_minus8" );
1397    assert( sei.m_numOverlaysMinus1 < MAX_OVERLAYS );
1398    WRITE_UVLC( sei.m_numOverlaysMinus1, "num_overlays_minus1" );
1399    for (i=0 ; i<=sei.m_numOverlaysMinus1 ; i++)
1400    {
1401      WRITE_UVLC( sei.m_overlayIdx[i], "overlay_idx" );
1402      WRITE_FLAG( sei.m_languageOverlayPresentFlag[i], "language_overlay_present_flag" );
1403      WRITE_CODE( sei.m_overlayContentLayerId[i], 6, "overlay_content_layer_id");
1404      WRITE_FLAG( sei.m_overlayLabelPresentFlag[i], "overlay_label_present_flag" );
1405      if ( sei.m_overlayLabelPresentFlag[i] )
1406      {
1407        WRITE_CODE(  sei.m_overlayLabelLayerId[i], 6, "overlay_label_layer_id");
1408      }
1409      WRITE_FLAG( sei.m_overlayAlphaPresentFlag[i], "overlay_alpha_present_flag" );
1410      if ( sei.m_overlayAlphaPresentFlag[i] )
1411      {
1412        WRITE_CODE( sei.m_overlayAlphaLayerId[i], 6, "overlay_alpha_layer_id");
1413      }
1414      if ( sei.m_overlayLabelPresentFlag[i] )
1415      {
1416        assert( sei.m_numOverlayElementsMinus1[i] < MAX_OVERLAY_ELEMENTS );
1417        WRITE_UVLC( sei.m_numOverlayElementsMinus1[i], "num_overlay_elements_minus1");
1418        for ( j=0 ; j<=sei.m_numOverlayElementsMinus1[i] ; j++ )
1419        {
1420          WRITE_CODE(sei.m_overlayElementLabelMin[i][j], sei.m_overlayElementLabelValueLengthMinus8 + 8, "overlay_element_label_min");
1421          WRITE_CODE(sei.m_overlayElementLabelMax[i][j], sei.m_overlayElementLabelValueLengthMinus8 + 8, "overlay_element_label_max"); 
1422        }
1423      }
1424    }
1425
1426
1427    // byte alignment
1428    while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
1429    {
1430      WRITE_FLAG( 0, "overlay_zero_bit" );
1431    }
1432
1433    for ( i=0 ; i<=sei.m_numOverlaysMinus1 ; i++ )
1434    {
1435      if ( sei.m_languageOverlayPresentFlag[i] )
1436      {
1437        WRITE_STRING( sei.m_overlayLanguage[i], sei.m_overlayLanguageLength[i], "overlay_language" );    //WRITE_STRING adds zero-termination byte
1438      }
1439      WRITE_STRING( sei.m_overlayName[i], sei.m_overlayNameLength[i], "overlay_name" );
1440      if ( sei.m_overlayLabelPresentFlag[i] )
1441      {
1442        for ( j=0 ; j<=sei.m_numOverlayElementsMinus1[i] ; j++)
1443        {
1444          WRITE_STRING( sei.m_overlayElementName[i][j], sei.m_overlayElementNameLength[i][j], "overlay_element_name" );
1445        }
1446      }
1447    }
1448    WRITE_FLAG( sei.m_overlayInfoPersistenceFlag, "overlay_info_persistence_flag" );
1449  }
1450}
1451#endif
1452
1453#endif //SVC_EXTENSION
1454
1455#if Q0074_COLOUR_REMAPPING_SEI
1456Void SEIWriter::xWriteSEIColourRemappingInfo(const SEIColourRemappingInfo& sei)
1457{
1458  WRITE_UVLC( sei.m_colourRemapId,                             "colour_remap_id" );
1459  WRITE_FLAG( sei.m_colourRemapCancelFlag,                     "colour_remap_cancel_flag" );
1460  if( !sei.m_colourRemapCancelFlag ) 
1461  {
1462    WRITE_FLAG( sei.m_colourRemapPersistenceFlag,              "colour_remap_persistence_flag" );
1463    WRITE_FLAG( sei.m_colourRemapVideoSignalInfoPresentFlag,   "colour_remap_video_signal_info_present_flag" );
1464    if ( sei.m_colourRemapVideoSignalInfoPresentFlag )
1465    {
1466      WRITE_FLAG( sei.m_colourRemapFullRangeFlag,              "colour_remap_full_range_flag" );
1467      WRITE_CODE( sei.m_colourRemapPrimaries,               8, "colour_remap_primaries" );
1468      WRITE_CODE( sei.m_colourRemapTransferFunction,        8, "colour_remap_transfer_function" );
1469      WRITE_CODE( sei.m_colourRemapMatrixCoefficients,      8, "colour_remap_matrix_coefficients" );
1470    }
1471    WRITE_CODE( sei.m_colourRemapInputBitDepth,             8, "colour_remap_input_bit_depth" );
1472    WRITE_CODE( sei.m_colourRemapBitDepth,                  8, "colour_remap_bit_depth" );
1473    for( Int c=0 ; c<3 ; c++ )
1474    {
1475      WRITE_CODE( sei.m_preLutNumValMinus1[c],              8, "pre_lut_num_val_minus1[c]" );
1476      if( sei.m_preLutNumValMinus1[c]>0 )
1477        for( Int i=0 ; i<=sei.m_preLutNumValMinus1[c] ; i++ )
1478        {
1479          WRITE_CODE( sei.m_preLutCodedValue[c][i], (( sei.m_colourRemapInputBitDepth + 7 ) >> 3 ) << 3, "pre_lut_coded_value[c][i]" );
1480          WRITE_CODE( sei.m_preLutTargetValue[c][i], (( sei.m_colourRemapBitDepth + 7 ) >> 3 ) << 3, "pre_lut_target_value[c][i]" );
1481        }
1482    }
1483    WRITE_FLAG( sei.m_colourRemapMatrixPresentFlag,            "colour_remap_matrix_present_flag" );
1484    if( sei.m_colourRemapMatrixPresentFlag )
1485    {
1486      WRITE_CODE( sei.m_log2MatrixDenom,                    4, "log2_matrix_denom" );
1487      for( Int c=0 ; c<3 ; c++ )
1488        for( Int i=0 ; i<3 ; i++ )
1489          WRITE_SVLC( sei.m_colourRemapCoeffs[c][i],           "colour_remap_coeffs[c][i]" );
1490    }
1491
1492    for( Int c=0 ; c<3 ; c++ )
1493    {
1494      WRITE_CODE( sei.m_postLutNumValMinus1[c],             8, "m_postLutNumValMinus1[c]" );
1495      if( sei.m_postLutNumValMinus1[c]>0 )
1496        for( Int i=0 ; i<=sei.m_postLutNumValMinus1[c] ; i++ )
1497        {
1498          WRITE_CODE( sei.m_postLutCodedValue[c][i], (( sei.m_colourRemapBitDepth + 7 ) >> 3 ) << 3, "post_lut_coded_value[c][i]" );       
1499          WRITE_CODE( sei.m_postLutTargetValue[c][i], (( sei.m_colourRemapBitDepth + 7 ) >> 3 ) << 3, "post_lut_target_value[c][i]" );
1500        }
1501    }
1502  }
1503}
1504#endif
1505
1506//! \}
Note: See TracBrowser for help on using the repository browser.