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

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

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