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

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

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