source: 3DVCSoftware/branches/HTM-16.0-MV-draft-5/source/Lib/TLibEncoder/SEIwrite.cpp

Last change on this file was 1390, checked in by tech, 9 years ago

Removed 3D.

  • Property svn:eol-style set to native
File size: 51.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
56Void SEIWriter::xWriteSEIpayloadData(TComBitIf& bs, const SEI& sei, const TComSPS *sps)
57{
58  switch (sei.payloadType())
59  {
60  case SEI::USER_DATA_UNREGISTERED:
61    xWriteSEIuserDataUnregistered(*static_cast<const SEIuserDataUnregistered*>(&sei));
62    break;
63  case SEI::ACTIVE_PARAMETER_SETS:
64    xWriteSEIActiveParameterSets(*static_cast<const SEIActiveParameterSets*>(& sei));
65    break;
66  case SEI::DECODING_UNIT_INFO:
67    xWriteSEIDecodingUnitInfo(*static_cast<const SEIDecodingUnitInfo*>(& sei), sps);
68    break;
69  case SEI::DECODED_PICTURE_HASH:
70    xWriteSEIDecodedPictureHash(*static_cast<const SEIDecodedPictureHash*>(&sei));
71    break;
72  case SEI::BUFFERING_PERIOD:
73    xWriteSEIBufferingPeriod(*static_cast<const SEIBufferingPeriod*>(&sei), sps);
74    break;
75  case SEI::PICTURE_TIMING:
76    xWriteSEIPictureTiming(*static_cast<const SEIPictureTiming*>(&sei), sps);
77    break;
78  case SEI::RECOVERY_POINT:
79    xWriteSEIRecoveryPoint(*static_cast<const SEIRecoveryPoint*>(&sei));
80    break;
81  case SEI::FRAME_PACKING:
82    xWriteSEIFramePacking(*static_cast<const SEIFramePacking*>(&sei));
83    break;
84  case SEI::SEGM_RECT_FRAME_PACKING:
85    xWriteSEISegmentedRectFramePacking(*static_cast<const SEISegmentedRectFramePacking*>(&sei));
86    break;
87  case SEI::DISPLAY_ORIENTATION:
88    xWriteSEIDisplayOrientation(*static_cast<const SEIDisplayOrientation*>(&sei));
89    break;
90  case SEI::TEMPORAL_LEVEL0_INDEX:
91    xWriteSEITemporalLevel0Index(*static_cast<const SEITemporalLevel0Index*>(&sei));
92    break;
93  case SEI::REGION_REFRESH_INFO:
94    xWriteSEIGradualDecodingRefreshInfo(*static_cast<const SEIGradualDecodingRefreshInfo*>(&sei));
95    break;
96  case SEI::NO_DISPLAY:
97    xWriteSEINoDisplay(*static_cast<const SEINoDisplay*>(&sei));
98    break;
99  case SEI::TONE_MAPPING_INFO:
100    xWriteSEIToneMappingInfo(*static_cast<const SEIToneMappingInfo*>(&sei));
101    break;
102  case SEI::SOP_DESCRIPTION:
103    xWriteSEISOPDescription(*static_cast<const SEISOPDescription*>(&sei));
104    break;
105  case SEI::SCALABLE_NESTING:
106    xWriteSEIScalableNesting(bs, *static_cast<const SEIScalableNesting*>(&sei), sps);
107    break;
108  case SEI::CHROMA_RESAMPLING_FILTER_HINT:
109    xWriteSEIChromaResamplingFilterHint(*static_cast<const SEIChromaResamplingFilterHint*>(&sei));
110    break;
111  case SEI::TEMP_MOTION_CONSTRAINED_TILE_SETS:
112    xWriteSEITempMotionConstrainedTileSets(*static_cast<const SEITempMotionConstrainedTileSets*>(&sei));
113    break;
114  case SEI::TIME_CODE:
115    xWriteSEITimeCode(*static_cast<const SEITimeCode*>(&sei));
116    break;
117  case SEI::KNEE_FUNCTION_INFO:
118    xWriteSEIKneeFunctionInfo(*static_cast<const SEIKneeFunctionInfo*>(&sei));
119    break;
120  case SEI::COLOUR_REMAPPING_INFO:
121    xWriteSEIColourRemappingInfo(*static_cast<const SEIColourRemappingInfo*>(&sei));
122    break;
123  case SEI::MASTERING_DISPLAY_COLOUR_VOLUME:
124    xWriteSEIMasteringDisplayColourVolume(*static_cast<const SEIMasteringDisplayColourVolume*>(&sei));
125    break;
126#if NH_MV
127   case SEI::LAYERS_NOT_PRESENT:
128       xWriteSEILayersNotPresent(*static_cast<const SEILayersNotPresent*>(&sei));
129     break;
130   case SEI::INTER_LAYER_CONSTRAINED_TILE_SETS:
131     xWriteSEIInterLayerConstrainedTileSets(*static_cast<const SEIInterLayerConstrainedTileSets*>(&sei));
132     break;
133#if NH_MV_SEI_TBD
134   case SEI::BSP_NESTING:
135     xWriteSEIBspNesting(*static_cast<const SEIBspNesting*>(&sei));
136     break;
137   case SEI::BSP_INITIAL_ARRIVAL_TIME:
138     xWriteSEIBspInitialArrivalTime(*static_cast<const SEIBspInitialArrivalTime*>(&sei));
139     break;
140#endif
141   case SEI::SUB_BITSTREAM_PROPERTY:
142     xWriteSEISubBitstreamProperty(*static_cast<const SEISubBitstreamProperty*>(&sei));
143     break;
144   case SEI::ALPHA_CHANNEL_INFO:
145     xWriteSEIAlphaChannelInfo(*static_cast<const SEIAlphaChannelInfo*>(&sei));
146     break;
147   case SEI::OVERLAY_INFO:
148     xWriteSEIOverlayInfo(*static_cast<const SEIOverlayInfo*>(&sei));
149     break;
150   case SEI::TEMPORAL_MV_PREDICTION_CONSTRAINTS:
151     xWriteSEITemporalMvPredictionConstraints(*static_cast<const SEITemporalMvPredictionConstraints*>(&sei));
152     break;
153#if NH_MV_SEI_TBD
154   case SEI::FRAME_FIELD_INFO:
155     xWriteSEIFrameFieldInfo(*static_cast<const SEIFrameFieldInfo*>(&sei));
156     break;
157#endif
158   case SEI::THREE_DIMENSIONAL_REFERENCE_DISPLAYS_INFO:
159     xWriteSEIThreeDimensionalReferenceDisplaysInfo(*static_cast<const SEIThreeDimensionalReferenceDisplaysInfo*>(&sei));
160     break;
161   case SEI::DEPTH_REPRESENTATION_INFO:
162       xWriteSEIDepthRepresentationInfo(*static_cast<const SEIDepthRepresentationInfo*>(&sei));
163       break;
164   case SEI::MULTIVIEW_SCENE_INFO:
165     xWriteSEIMultiviewSceneInfo(*static_cast<const SEIMultiviewSceneInfo*>(&sei));
166     break;
167   case SEI::MULTIVIEW_ACQUISITION_INFO:
168     xWriteSEIMultiviewAcquisitionInfo(*static_cast<const SEIMultiviewAcquisitionInfo*>(&sei));
169     break;
170
171   case SEI::MULTIVIEW_VIEW_POSITION:
172     xWriteSEIMultiviewViewPosition(*static_cast<const SEIMultiviewViewPosition*>(&sei));
173     break;
174#endif
175
176
177  default:
178    assert(!"Trying to write unhandled SEI message");
179    break;
180  }
181  xWriteByteAlign();
182}
183
184/**
185 * marshal all SEI messages in provided list into one bitstream bs
186 */
187Void SEIWriter::writeSEImessages(TComBitIf& bs, const SEIMessages &seiList, const TComSPS *sps, Bool isNested)
188{
189#if ENC_DEC_TRACE
190  if (g_HLSTraceEnable)
191    xTraceSEIHeader();
192#endif
193
194  TComBitCounter bs_count;
195
196  for (SEIMessages::const_iterator sei=seiList.begin(); sei!=seiList.end(); sei++)
197  {
198    // calculate how large the payload data is
199    // TODO: this would be far nicer if it used vectored buffers
200    bs_count.resetBits();
201    setBitstream(&bs_count);
202
203#if ENC_DEC_TRACE
204    Bool traceEnable = g_HLSTraceEnable;
205    g_HLSTraceEnable = false;
206#endif
207    xWriteSEIpayloadData(bs_count, **sei, sps);
208#if ENC_DEC_TRACE
209    g_HLSTraceEnable = traceEnable;
210#endif
211    UInt payload_data_num_bits = bs_count.getNumberOfWrittenBits();
212    assert(0 == payload_data_num_bits % 8);
213
214    setBitstream(&bs);
215    UInt payloadType = (*sei)->payloadType();
216    for (; payloadType >= 0xff; payloadType -= 0xff)
217    {
218      WRITE_CODE(0xff, 8, "payload_type");
219    }
220    WRITE_CODE(payloadType, 8, "payload_type");
221
222    UInt payloadSize = payload_data_num_bits/8;
223    for (; payloadSize >= 0xff; payloadSize -= 0xff)
224    {
225      WRITE_CODE(0xff, 8, "payload_size");
226    }
227    WRITE_CODE(payloadSize, 8, "payload_size");
228
229    /* payloadData */
230#if ENC_DEC_TRACE
231    if (g_HLSTraceEnable)
232      xTraceSEIMessageType((*sei)->payloadType());
233#endif
234
235    xWriteSEIpayloadData(bs, **sei, sps);
236  }
237  if (!isNested)
238  {
239    xWriteRbspTrailingBits();
240  }
241}
242
243/**
244 * marshal a user_data_unregistered SEI message sei, storing the marshalled
245 * representation in bitstream bs.
246 */
247Void SEIWriter::xWriteSEIuserDataUnregistered(const SEIuserDataUnregistered &sei)
248{
249  for (UInt i = 0; i < ISO_IEC_11578_LEN; i++)
250  {
251    WRITE_CODE(sei.uuid_iso_iec_11578[i], 8 , "sei.uuid_iso_iec_11578[i]");
252  }
253
254  for (UInt i = 0; i < sei.userDataLength; i++)
255  {
256    WRITE_CODE(sei.userData[i], 8 , "user_data");
257  }
258}
259
260/**
261 * marshal a decoded picture hash SEI message, storing the marshalled
262 * representation in bitstream bs.
263 */
264Void SEIWriter::xWriteSEIDecodedPictureHash(const SEIDecodedPictureHash& sei)
265{
266  const TChar *traceString="\0";
267  switch (sei.method)
268  {
269    case HASHTYPE_MD5: traceString="picture_md5"; break;
270    case HASHTYPE_CRC: traceString="picture_crc"; break;
271    case HASHTYPE_CHECKSUM: traceString="picture_checksum"; break;
272    default: assert(false); break;
273  }
274
275  if (traceString != 0) //use of this variable is needed to avoid a compiler error with G++ 4.6.1
276  {
277    WRITE_CODE(sei.method, 8, "hash_type");
278    for(UInt i=0; i<UInt(sei.m_pictureHash.hash.size()); i++)
279    {
280      WRITE_CODE(sei.m_pictureHash.hash[i], 8, traceString);
281    }
282  }
283}
284
285Void SEIWriter::xWriteSEIActiveParameterSets(const SEIActiveParameterSets& sei)
286{
287  WRITE_CODE(sei.activeVPSId,     4,         "active_video_parameter_set_id");
288  WRITE_FLAG(sei.m_selfContainedCvsFlag,     "self_contained_cvs_flag");
289  WRITE_FLAG(sei.m_noParameterSetUpdateFlag, "no_parameter_set_update_flag");
290  WRITE_UVLC(sei.numSpsIdsMinus1,            "num_sps_ids_minus1");
291
292  assert (sei.activeSeqParameterSetId.size() == (sei.numSpsIdsMinus1 + 1));
293
294  for (Int i = 0; i < sei.activeSeqParameterSetId.size(); i++)
295  {
296    WRITE_UVLC(sei.activeSeqParameterSetId[i], "active_seq_parameter_set_id");
297  }
298}
299
300Void SEIWriter::xWriteSEIDecodingUnitInfo(const SEIDecodingUnitInfo& sei, const TComSPS *sps)
301{
302  const TComVUI *vui = sps->getVuiParameters();
303  WRITE_UVLC(sei.m_decodingUnitIdx, "decoding_unit_idx");
304  if(vui->getHrdParameters()->getSubPicCpbParamsInPicTimingSEIFlag())
305  {
306    WRITE_CODE( sei.m_duSptCpbRemovalDelay, (vui->getHrdParameters()->getDuCpbRemovalDelayLengthMinus1() + 1), "du_spt_cpb_removal_delay_increment");
307  }
308  WRITE_FLAG( sei.m_dpbOutputDuDelayPresentFlag, "dpb_output_du_delay_present_flag");
309  if(sei.m_dpbOutputDuDelayPresentFlag)
310  {
311    WRITE_CODE(sei.m_picSptDpbOutputDuDelay, vui->getHrdParameters()->getDpbOutputDelayDuLengthMinus1() + 1, "pic_spt_dpb_output_du_delay");
312  }
313}
314
315Void SEIWriter::xWriteSEIBufferingPeriod(const SEIBufferingPeriod& sei, const TComSPS *sps)
316{
317  Int i, nalOrVcl;
318  const TComVUI *vui = sps->getVuiParameters();
319  const TComHRD *hrd = vui->getHrdParameters();
320
321  WRITE_UVLC( sei.m_bpSeqParameterSetId, "bp_seq_parameter_set_id" );
322  if( !hrd->getSubPicCpbParamsPresentFlag() )
323  {
324    WRITE_FLAG( sei.m_rapCpbParamsPresentFlag, "irap_cpb_params_present_flag" );
325  }
326  if( sei.m_rapCpbParamsPresentFlag )
327  {
328    WRITE_CODE( sei.m_cpbDelayOffset, hrd->getCpbRemovalDelayLengthMinus1() + 1, "cpb_delay_offset" );
329    WRITE_CODE( sei.m_dpbDelayOffset, hrd->getDpbOutputDelayLengthMinus1()  + 1, "dpb_delay_offset" );
330  }
331  WRITE_FLAG( sei.m_concatenationFlag, "concatenation_flag");
332  WRITE_CODE( sei.m_auCpbRemovalDelayDelta - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ), "au_cpb_removal_delay_delta_minus1" );
333  for( nalOrVcl = 0; nalOrVcl < 2; nalOrVcl ++ )
334  {
335    if( ( ( nalOrVcl == 0 ) && ( hrd->getNalHrdParametersPresentFlag() ) ) ||
336        ( ( nalOrVcl == 1 ) && ( hrd->getVclHrdParametersPresentFlag() ) ) )
337    {
338      for( i = 0; i < ( hrd->getCpbCntMinus1( 0 ) + 1 ); i ++ )
339      {
340        WRITE_CODE( sei.m_initialCpbRemovalDelay[i][nalOrVcl],( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ) ,           "initial_cpb_removal_delay" );
341        WRITE_CODE( sei.m_initialCpbRemovalDelayOffset[i][nalOrVcl],( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ),      "initial_cpb_removal_delay_offset" );
342        if( hrd->getSubPicCpbParamsPresentFlag() || sei.m_rapCpbParamsPresentFlag )
343        {
344          WRITE_CODE( sei.m_initialAltCpbRemovalDelay[i][nalOrVcl], ( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ) ,     "initial_alt_cpb_removal_delay" );
345          WRITE_CODE( sei.m_initialAltCpbRemovalDelayOffset[i][nalOrVcl], ( hrd->getInitialCpbRemovalDelayLengthMinus1() + 1 ),"initial_alt_cpb_removal_delay_offset" );
346        }
347      }
348    }
349  }
350}
351Void SEIWriter::xWriteSEIPictureTiming(const SEIPictureTiming& sei, const TComSPS *sps)
352{
353  Int i;
354  const TComVUI *vui = sps->getVuiParameters();
355  const TComHRD *hrd = vui->getHrdParameters();
356
357  if( vui->getFrameFieldInfoPresentFlag() )
358  {
359    WRITE_CODE( sei.m_picStruct, 4,              "pic_struct" );
360    WRITE_CODE( sei.m_sourceScanType, 2,         "source_scan_type" );
361    WRITE_FLAG( sei.m_duplicateFlag ? 1 : 0,     "duplicate_flag" );
362  }
363
364  if( hrd->getCpbDpbDelaysPresentFlag() )
365  {
366    WRITE_CODE( sei.m_auCpbRemovalDelay - 1, ( hrd->getCpbRemovalDelayLengthMinus1() + 1 ),                                         "au_cpb_removal_delay_minus1" );
367    WRITE_CODE( sei.m_picDpbOutputDelay, ( hrd->getDpbOutputDelayLengthMinus1() + 1 ),                                          "pic_dpb_output_delay" );
368    if(hrd->getSubPicCpbParamsPresentFlag())
369    {
370      WRITE_CODE(sei.m_picDpbOutputDuDelay, hrd->getDpbOutputDelayDuLengthMinus1()+1, "pic_dpb_output_du_delay" );
371    }
372    if( hrd->getSubPicCpbParamsPresentFlag() && hrd->getSubPicCpbParamsInPicTimingSEIFlag() )
373    {
374      WRITE_UVLC( sei.m_numDecodingUnitsMinus1,     "num_decoding_units_minus1" );
375      WRITE_FLAG( sei.m_duCommonCpbRemovalDelayFlag, "du_common_cpb_removal_delay_flag" );
376      if( sei.m_duCommonCpbRemovalDelayFlag )
377      {
378        WRITE_CODE( sei.m_duCommonCpbRemovalDelayMinus1, ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ),                       "du_common_cpb_removal_delay_minus1" );
379      }
380      for( i = 0; i <= sei.m_numDecodingUnitsMinus1; i ++ )
381      {
382        WRITE_UVLC( sei.m_numNalusInDuMinus1[ i ],  "num_nalus_in_du_minus1");
383        if( ( !sei.m_duCommonCpbRemovalDelayFlag ) && ( i < sei.m_numDecodingUnitsMinus1 ) )
384        {
385          WRITE_CODE( sei.m_duCpbRemovalDelayMinus1[ i ], ( hrd->getDuCpbRemovalDelayLengthMinus1() + 1 ),                        "du_cpb_removal_delay_minus1" );
386        }
387      }
388    }
389  }
390}
391Void SEIWriter::xWriteSEIRecoveryPoint(const SEIRecoveryPoint& sei)
392{
393  WRITE_SVLC( sei.m_recoveryPocCnt,    "recovery_poc_cnt"    );
394  WRITE_FLAG( sei.m_exactMatchingFlag, "exact_matching_flag" );
395  WRITE_FLAG( sei.m_brokenLinkFlag,    "broken_link_flag"    );
396}
397Void SEIWriter::xWriteSEIFramePacking(const SEIFramePacking& sei)
398{
399  WRITE_UVLC( sei.m_arrangementId,                  "frame_packing_arrangement_id" );
400  WRITE_FLAG( sei.m_arrangementCancelFlag,          "frame_packing_arrangement_cancel_flag" );
401
402  if( sei.m_arrangementCancelFlag == 0 )
403  {
404    WRITE_CODE( sei.m_arrangementType, 7,           "frame_packing_arrangement_type" );
405
406    WRITE_FLAG( sei.m_quincunxSamplingFlag,         "quincunx_sampling_flag" );
407    WRITE_CODE( sei.m_contentInterpretationType, 6, "content_interpretation_type" );
408    WRITE_FLAG( sei.m_spatialFlippingFlag,          "spatial_flipping_flag" );
409    WRITE_FLAG( sei.m_frame0FlippedFlag,            "frame0_flipped_flag" );
410    WRITE_FLAG( sei.m_fieldViewsFlag,               "field_views_flag" );
411    WRITE_FLAG( sei.m_currentFrameIsFrame0Flag,     "current_frame_is_frame0_flag" );
412
413    WRITE_FLAG( sei.m_frame0SelfContainedFlag,      "frame0_self_contained_flag" );
414    WRITE_FLAG( sei.m_frame1SelfContainedFlag,      "frame1_self_contained_flag" );
415
416    if(sei.m_quincunxSamplingFlag == 0 && sei.m_arrangementType != 5)
417    {
418      WRITE_CODE( sei.m_frame0GridPositionX, 4,     "frame0_grid_position_x" );
419      WRITE_CODE( sei.m_frame0GridPositionY, 4,     "frame0_grid_position_y" );
420      WRITE_CODE( sei.m_frame1GridPositionX, 4,     "frame1_grid_position_x" );
421      WRITE_CODE( sei.m_frame1GridPositionY, 4,     "frame1_grid_position_y" );
422    }
423
424    WRITE_CODE( sei.m_arrangementReservedByte, 8,   "frame_packing_arrangement_reserved_byte" );
425    WRITE_FLAG( sei.m_arrangementPersistenceFlag,   "frame_packing_arrangement_persistence_flag" );
426  }
427
428  WRITE_FLAG( sei.m_upsampledAspectRatio,           "upsampled_aspect_ratio" );
429}
430
431Void SEIWriter::xWriteSEISegmentedRectFramePacking(const SEISegmentedRectFramePacking& sei)
432{
433  WRITE_FLAG( sei.m_arrangementCancelFlag,          "segmented_rect_frame_packing_arrangement_cancel_flag" );
434  if( sei.m_arrangementCancelFlag == 0 )
435  {
436    WRITE_CODE( sei.m_contentInterpretationType, 2, "segmented_rect_content_interpretation_type" );
437    WRITE_FLAG( sei.m_arrangementPersistenceFlag,   "segmented_rect_frame_packing_arrangement_persistence" );
438  }
439}
440
441Void SEIWriter::xWriteSEIToneMappingInfo(const SEIToneMappingInfo& sei)
442{
443  Int i;
444  WRITE_UVLC( sei.m_toneMapId,                    "tone_map_id" );
445  WRITE_FLAG( sei.m_toneMapCancelFlag,            "tone_map_cancel_flag" );
446  if( !sei.m_toneMapCancelFlag )
447  {
448    WRITE_FLAG( sei.m_toneMapPersistenceFlag,     "tone_map_persistence_flag" );
449    WRITE_CODE( sei.m_codedDataBitDepth,    8,    "coded_data_bit_depth" );
450    WRITE_CODE( sei.m_targetBitDepth,       8,    "target_bit_depth" );
451    WRITE_UVLC( sei.m_modelId,                    "model_id" );
452    switch(sei.m_modelId)
453    {
454    case 0:
455      {
456        WRITE_CODE( sei.m_minValue,  32,        "min_value" );
457        WRITE_CODE( sei.m_maxValue, 32,         "max_value" );
458        break;
459      }
460    case 1:
461      {
462        WRITE_CODE( sei.m_sigmoidMidpoint, 32,  "sigmoid_midpoint" );
463        WRITE_CODE( sei.m_sigmoidWidth,    32,  "sigmoid_width"    );
464        break;
465      }
466    case 2:
467      {
468        UInt num = 1u << sei.m_targetBitDepth;
469        for(i = 0; i < num; i++)
470        {
471          WRITE_CODE( sei.m_startOfCodedInterval[i], (( sei.m_codedDataBitDepth + 7 ) >> 3 ) << 3,  "start_of_coded_interval" );
472        }
473        break;
474      }
475    case 3:
476      {
477        WRITE_CODE( sei.m_numPivots, 16,          "num_pivots" );
478        for(i = 0; i < sei.m_numPivots; i++ )
479        {
480          WRITE_CODE( sei.m_codedPivotValue[i], (( sei.m_codedDataBitDepth + 7 ) >> 3 ) << 3,       "coded_pivot_value" );
481          WRITE_CODE( sei.m_targetPivotValue[i], (( sei.m_targetBitDepth + 7 ) >> 3 ) << 3,         "target_pivot_value");
482        }
483        break;
484      }
485    case 4:
486      {
487        WRITE_CODE( sei.m_cameraIsoSpeedIdc,    8,    "camera_iso_speed_idc" );
488        if( sei.m_cameraIsoSpeedIdc == 255) //Extended_ISO
489        {
490          WRITE_CODE( sei.m_cameraIsoSpeedValue,    32,    "camera_iso_speed_value" );
491        }
492        WRITE_CODE( sei.m_exposureIndexIdc,     8,    "exposure_index_idc" );
493        if( sei.m_exposureIndexIdc == 255) //Extended_ISO
494        {
495          WRITE_CODE( sei.m_exposureIndexValue,     32,    "exposure_index_value" );
496        }
497        WRITE_FLAG( sei.m_exposureCompensationValueSignFlag,           "exposure_compensation_value_sign_flag" );
498        WRITE_CODE( sei.m_exposureCompensationValueNumerator,     16,  "exposure_compensation_value_numerator" );
499        WRITE_CODE( sei.m_exposureCompensationValueDenomIdc,      16,  "exposure_compensation_value_denom_idc" );
500        WRITE_CODE( sei.m_refScreenLuminanceWhite,                32,  "ref_screen_luminance_white" );
501        WRITE_CODE( sei.m_extendedRangeWhiteLevel,                32,  "extended_range_white_level" );
502        WRITE_CODE( sei.m_nominalBlackLevelLumaCodeValue,         16,  "nominal_black_level_luma_code_value" );
503        WRITE_CODE( sei.m_nominalWhiteLevelLumaCodeValue,         16,  "nominal_white_level_luma_code_value" );
504        WRITE_CODE( sei.m_extendedWhiteLevelLumaCodeValue,        16,  "extended_white_level_luma_code_value" );
505        break;
506      }
507    default:
508      {
509        assert(!"Undefined SEIToneMapModelId");
510        break;
511      }
512    }//switch m_modelId
513  }//if(!sei.m_toneMapCancelFlag)
514}
515
516Void SEIWriter::xWriteSEIDisplayOrientation(const SEIDisplayOrientation &sei)
517{
518  WRITE_FLAG( sei.cancelFlag,           "display_orientation_cancel_flag" );
519  if( !sei.cancelFlag )
520  {
521    WRITE_FLAG( sei.horFlip,                   "hor_flip" );
522    WRITE_FLAG( sei.verFlip,                   "ver_flip" );
523    WRITE_CODE( sei.anticlockwiseRotation, 16, "anticlockwise_rotation" );
524    WRITE_FLAG( sei.persistenceFlag,          "display_orientation_persistence_flag" );
525  }
526}
527
528Void SEIWriter::xWriteSEITemporalLevel0Index(const SEITemporalLevel0Index &sei)
529{
530  WRITE_CODE( sei.tl0Idx, 8 , "tl0_idx" );
531  WRITE_CODE( sei.rapIdx, 8 , "rap_idx" );
532}
533
534Void SEIWriter::xWriteSEIGradualDecodingRefreshInfo(const SEIGradualDecodingRefreshInfo &sei)
535{
536  WRITE_FLAG( sei.m_gdrForegroundFlag, "gdr_foreground_flag");
537}
538
539Void SEIWriter::xWriteSEINoDisplay(const SEINoDisplay& /*sei*/)
540{
541}
542
543Void SEIWriter::xWriteSEISOPDescription(const SEISOPDescription& sei)
544{
545  WRITE_UVLC( sei.m_sopSeqParameterSetId,           "sop_seq_parameter_set_id"               );
546  WRITE_UVLC( sei.m_numPicsInSopMinus1,             "num_pics_in_sop_minus1"               );
547  for (UInt i = 0; i <= sei.m_numPicsInSopMinus1; i++)
548  {
549    WRITE_CODE( sei.m_sopDescVclNaluType[i], 6, "sop_desc_vcl_nalu_type" );
550    WRITE_CODE( sei.m_sopDescTemporalId[i],  3, "sop_desc_temporal_id" );
551    if (sei.m_sopDescVclNaluType[i] != NAL_UNIT_CODED_SLICE_IDR_W_RADL && sei.m_sopDescVclNaluType[i] != NAL_UNIT_CODED_SLICE_IDR_N_LP)
552    {
553      WRITE_UVLC( sei.m_sopDescStRpsIdx[i],           "sop_desc_st_rps_idx"               );
554    }
555    if (i > 0)
556    {
557      WRITE_SVLC( sei.m_sopDescPocDelta[i],           "sop_desc_poc_delta"               );
558    }
559  }
560}
561
562Void SEIWriter::xWriteSEIScalableNesting(TComBitIf& bs, const SEIScalableNesting& sei, const TComSPS *sps)
563{
564  WRITE_FLAG( sei.m_bitStreamSubsetFlag,             "bitstream_subset_flag"         );
565  WRITE_FLAG( sei.m_nestingOpFlag,                   "nesting_op_flag      "         );
566  if (sei.m_nestingOpFlag)
567  {
568    WRITE_FLAG( sei.m_defaultOpFlag,                 "default_op_flag"               );
569    WRITE_UVLC( sei.m_nestingNumOpsMinus1,           "nesting_num_ops_minus1"        );
570    for (UInt i = (sei.m_defaultOpFlag ? 1 : 0); i <= sei.m_nestingNumOpsMinus1; i++)
571    {
572      WRITE_CODE( sei.m_nestingMaxTemporalIdPlus1[i], 3,  "nesting_max_temporal_id_plus1" );
573      WRITE_UVLC( sei.m_nestingOpIdx[i],                  "nesting_op_idx"                );
574    }
575  }
576  else
577  {
578    WRITE_FLAG( sei.m_allLayersFlag,                      "all_layers_flag"               );
579    if (!sei.m_allLayersFlag)
580    {
581      WRITE_CODE( sei.m_nestingNoOpMaxTemporalIdPlus1, 3, "nesting_no_op_max_temporal_id_plus1" );
582      WRITE_UVLC( sei.m_nestingNumLayersMinus1,           "nesting_num_layers"                  );
583      for (UInt i = 0; i <= sei.m_nestingNumLayersMinus1; i++)
584      {
585        WRITE_CODE( sei.m_nestingLayerId[i], 6,           "nesting_layer_id"              );
586      }
587    }
588  }
589
590  // byte alignment
591  while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
592  {
593    WRITE_FLAG( 0, "nesting_zero_bit" );
594  }
595
596  // write nested SEI messages
597  writeSEImessages(bs, sei.m_nestedSEIs, sps, true);
598}
599
600Void SEIWriter::xWriteSEITempMotionConstrainedTileSets(const SEITempMotionConstrainedTileSets& sei)
601{
602  //UInt code;
603  WRITE_FLAG((sei.m_mc_all_tiles_exact_sample_value_match_flag ? 1 : 0), "mc_all_tiles_exact_sample_value_match_flag");
604  WRITE_FLAG((sei.m_each_tile_one_tile_set_flag                ? 1 : 0), "each_tile_one_tile_set_flag"               );
605
606  if(!sei.m_each_tile_one_tile_set_flag)
607  {
608    WRITE_FLAG((sei.m_limited_tile_set_display_flag ? 1 : 0), "limited_tile_set_display_flag");
609    WRITE_UVLC((sei.getNumberOfTileSets() - 1),               "num_sets_in_message_minus1"   );
610
611    if(sei.getNumberOfTileSets() > 0)
612    {
613      for(Int i = 0; i < sei.getNumberOfTileSets(); i++)
614      {
615        WRITE_UVLC(sei.tileSetData(i).m_mcts_id, "mcts_id");
616
617        if(sei.m_limited_tile_set_display_flag)
618        {
619          WRITE_FLAG((sei.tileSetData(i).m_display_tile_set_flag ? 1 : 0), "display_tile_set_flag");
620        }
621
622        WRITE_UVLC((sei.tileSetData(i).getNumberOfTileRects() - 1), "num_tile_rects_in_set_minus1");
623
624        for(Int j = 0; j < sei.tileSetData(i).getNumberOfTileRects(); j++)
625        {
626          WRITE_UVLC(sei.tileSetData(i).topLeftTileIndex    (j), "top_left_tile_index");
627          WRITE_UVLC(sei.tileSetData(i).bottomRightTileIndex(j), "bottom_right_tile_index");
628        }
629
630        if(!sei.m_mc_all_tiles_exact_sample_value_match_flag)
631        {
632          WRITE_FLAG((sei.tileSetData(i).m_exact_sample_value_match_flag ? 1 : 0), "exact_sample_value_match_flag");
633        }
634
635        WRITE_FLAG((sei.tileSetData(i).m_mcts_tier_level_idc_present_flag ? 1 : 0), "mcts_tier_level_idc_present_flag");
636
637        if(sei.tileSetData(i).m_mcts_tier_level_idc_present_flag)
638        {
639          WRITE_FLAG((sei.tileSetData(i).m_mcts_tier_flag ? 1 : 0), "mcts_tier_flag");
640          WRITE_CODE( sei.tileSetData(i).m_mcts_level_idc, 8,       "mcts_level_idc");
641        }
642      }
643    }
644  }
645  else
646  {
647    WRITE_FLAG((sei.m_max_mcs_tier_level_idc_present_flag ? 1 : 0), "max_mcs_tier_level_idc_present_flag");
648
649    if(sei.m_max_mcs_tier_level_idc_present_flag)
650    {
651      WRITE_FLAG((sei.m_max_mcts_tier_flag ? 1 : 0), "max_mcts_tier_flag");
652      WRITE_CODE( sei.m_max_mcts_level_idc, 8,       "max_mcts_level_idc");
653    }
654  }
655}
656
657Void SEIWriter::xWriteSEITimeCode(const SEITimeCode& sei)
658{
659  WRITE_CODE(sei.numClockTs, 2, "num_clock_ts");
660  for(Int i = 0; i < sei.numClockTs; i++)
661  {
662    const TComSEITimeSet &currentTimeSet = sei.timeSetArray[i];
663    WRITE_FLAG(currentTimeSet.clockTimeStampFlag, "clock_time_stamp_flag");
664    if(currentTimeSet.clockTimeStampFlag)
665    {
666      WRITE_FLAG(currentTimeSet.numUnitFieldBasedFlag, "units_field_based_flag");
667      WRITE_CODE(currentTimeSet.countingType, 5, "counting_type");
668      WRITE_FLAG(currentTimeSet.fullTimeStampFlag, "full_timestamp_flag");
669      WRITE_FLAG(currentTimeSet.discontinuityFlag, "discontinuity_flag");
670      WRITE_FLAG(currentTimeSet.cntDroppedFlag, "cnt_dropped_flag");
671      WRITE_CODE(currentTimeSet.numberOfFrames, 9, "n_frames");
672      if(currentTimeSet.fullTimeStampFlag)
673      {
674        WRITE_CODE(currentTimeSet.secondsValue, 6, "seconds_value");
675        WRITE_CODE(currentTimeSet.minutesValue, 6, "minutes_value");
676        WRITE_CODE(currentTimeSet.hoursValue, 5, "hours_value");
677      }
678      else
679      {
680        WRITE_FLAG(currentTimeSet.secondsFlag, "seconds_flag");
681        if(currentTimeSet.secondsFlag)
682        {
683          WRITE_CODE(currentTimeSet.secondsValue, 6, "seconds_value");
684          WRITE_FLAG(currentTimeSet.minutesFlag, "minutes_flag");
685          if(currentTimeSet.minutesFlag)
686          {
687            WRITE_CODE(currentTimeSet.minutesValue, 6, "minutes_value");
688            WRITE_FLAG(currentTimeSet.hoursFlag, "hours_flag");
689            if(currentTimeSet.hoursFlag)
690            {
691              WRITE_CODE(currentTimeSet.hoursValue, 5, "hours_value");
692            }
693          }
694        }
695      }
696      WRITE_CODE(currentTimeSet.timeOffsetLength, 5, "time_offset_length");
697      if(currentTimeSet.timeOffsetLength > 0)
698      {
699        if(currentTimeSet.timeOffsetValue >= 0)
700        {
701          WRITE_CODE((UInt)currentTimeSet.timeOffsetValue, currentTimeSet.timeOffsetLength, "time_offset_value");
702        }
703        else
704        {
705          //  Two's complement conversion
706          UInt offsetValue = ~(currentTimeSet.timeOffsetValue) + 1;
707          offsetValue |= (1 << (currentTimeSet.timeOffsetLength-1));
708          WRITE_CODE(offsetValue, currentTimeSet.timeOffsetLength, "time_offset_value");
709        }
710      }
711    }
712  }
713}
714
715Void SEIWriter::xWriteSEIChromaResamplingFilterHint(const SEIChromaResamplingFilterHint &sei)
716{
717  WRITE_CODE(sei.m_verChromaFilterIdc, 8, "ver_chroma_filter_idc");
718  WRITE_CODE(sei.m_horChromaFilterIdc, 8, "hor_chroma_filter_idc");
719  WRITE_FLAG(sei.m_verFilteringFieldProcessingFlag, "ver_filtering_field_processing_flag");
720  if(sei.m_verChromaFilterIdc == 1 || sei.m_horChromaFilterIdc == 1)
721  {
722    WRITE_UVLC(sei.m_targetFormatIdc, "target_format_idc");
723    if(sei.m_verChromaFilterIdc == 1)
724    {
725      const Int numVerticalFilter = (Int)sei.m_verFilterCoeff.size();
726      WRITE_UVLC(numVerticalFilter, "num_vertical_filters");
727      if(numVerticalFilter > 0)
728      {
729        for(Int i = 0; i < numVerticalFilter; i ++)
730        {
731          const Int verTapLengthMinus1 = (Int) sei.m_verFilterCoeff[i].size() - 1;
732          WRITE_UVLC(verTapLengthMinus1, "ver_tap_length_minus_1");
733          for(Int j = 0; j < (verTapLengthMinus1 + 1); j ++)
734          {
735            WRITE_SVLC(sei.m_verFilterCoeff[i][j], "ver_filter_coeff");
736          }
737        }
738      }
739    }
740    if(sei.m_horChromaFilterIdc == 1)
741    {
742      const Int numHorizontalFilter = (Int) sei.m_horFilterCoeff.size();
743      WRITE_UVLC(numHorizontalFilter, "num_horizontal_filters");
744      if(numHorizontalFilter > 0)
745      {
746        for(Int i = 0; i < numHorizontalFilter; i ++)
747        {
748          const Int horTapLengthMinus1 = (Int) sei.m_horFilterCoeff[i].size() - 1;
749          WRITE_UVLC(horTapLengthMinus1, "hor_tap_length_minus_1");
750          for(Int j = 0; j < (horTapLengthMinus1 + 1); j ++)
751          {
752            WRITE_SVLC(sei.m_horFilterCoeff[i][j], "hor_filter_coeff");
753          }
754        }
755      }
756    }
757  }
758}
759
760Void SEIWriter::xWriteSEIKneeFunctionInfo(const SEIKneeFunctionInfo &sei)
761{
762  WRITE_UVLC( sei.m_kneeId, "knee_function_id" );
763  WRITE_FLAG( sei.m_kneeCancelFlag, "knee_function_cancel_flag" );
764  if ( !sei.m_kneeCancelFlag )
765  {
766    WRITE_FLAG( sei.m_kneePersistenceFlag, "knee_function_persistence_flag" );
767    WRITE_CODE( (UInt)sei.m_kneeInputDrange , 32,  "input_d_range" );
768    WRITE_CODE( (UInt)sei.m_kneeInputDispLuminance, 32,  "input_disp_luminance" );
769    WRITE_CODE( (UInt)sei.m_kneeOutputDrange, 32,  "output_d_range" );
770    WRITE_CODE( (UInt)sei.m_kneeOutputDispLuminance, 32,  "output_disp_luminance" );
771    WRITE_UVLC( sei.m_kneeNumKneePointsMinus1, "num_knee_points_minus1" );
772    for(Int i = 0; i <= sei.m_kneeNumKneePointsMinus1; i++ )
773    {
774      WRITE_CODE( (UInt)sei.m_kneeInputKneePoint[i], 10,"input_knee_point" );
775      WRITE_CODE( (UInt)sei.m_kneeOutputKneePoint[i], 10, "output_knee_point" );
776    }
777  }
778}
779
780Void SEIWriter::xWriteSEIColourRemappingInfo(const SEIColourRemappingInfo& sei)
781{
782  WRITE_UVLC( sei.m_colourRemapId,                             "colour_remap_id" );
783  WRITE_FLAG( sei.m_colourRemapCancelFlag,                     "colour_remap_cancel_flag" );
784  if( !sei.m_colourRemapCancelFlag ) 
785  {
786    WRITE_FLAG( sei.m_colourRemapPersistenceFlag,              "colour_remap_persistence_flag" );
787    WRITE_FLAG( sei.m_colourRemapVideoSignalInfoPresentFlag,   "colour_remap_video_signal_info_present_flag" );
788    if ( sei.m_colourRemapVideoSignalInfoPresentFlag )
789    {
790      WRITE_FLAG( sei.m_colourRemapFullRangeFlag,              "colour_remap_full_range_flag" );
791      WRITE_CODE( sei.m_colourRemapPrimaries,               8, "colour_remap_primaries" );
792      WRITE_CODE( sei.m_colourRemapTransferFunction,        8, "colour_remap_transfer_function" );
793      WRITE_CODE( sei.m_colourRemapMatrixCoefficients,      8, "colour_remap_matrix_coefficients" );
794    }
795    WRITE_CODE( sei.m_colourRemapInputBitDepth,             8, "colour_remap_input_bit_depth" );
796    WRITE_CODE( sei.m_colourRemapBitDepth,                  8, "colour_remap_bit_depth" );
797    for( Int c=0 ; c<3 ; c++ )
798    {
799      WRITE_CODE( sei.m_preLutNumValMinus1[c],              8, "pre_lut_num_val_minus1[c]" );
800      if( sei.m_preLutNumValMinus1[c]>0 )
801      {
802        for( Int i=0 ; i<=sei.m_preLutNumValMinus1[c] ; i++ )
803        {
804          WRITE_CODE( sei.m_preLut[c][i].codedValue,  (( sei.m_colourRemapInputBitDepth + 7 ) >> 3 ) << 3, "pre_lut_coded_value[c][i]" );
805          WRITE_CODE( sei.m_preLut[c][i].targetValue, (( sei.m_colourRemapBitDepth      + 7 ) >> 3 ) << 3, "pre_lut_target_value[c][i]" );
806        }
807      }
808    }
809    WRITE_FLAG( sei.m_colourRemapMatrixPresentFlag,            "colour_remap_matrix_present_flag" );
810    if( sei.m_colourRemapMatrixPresentFlag )
811    {
812      WRITE_CODE( sei.m_log2MatrixDenom,                    4, "log2_matrix_denom" );
813      for( Int c=0 ; c<3 ; c++ )
814      {
815        for( Int i=0 ; i<3 ; i++ )
816        {
817          WRITE_SVLC( sei.m_colourRemapCoeffs[c][i],           "colour_remap_coeffs[c][i]" );
818        }
819      }
820    }
821
822    for( Int c=0 ; c<3 ; c++ )
823    {
824      WRITE_CODE( sei.m_postLutNumValMinus1[c],             8, "m_postLutNumValMinus1[c]" );
825      if( sei.m_postLutNumValMinus1[c]>0 )
826      {
827        for( Int i=0 ; i<=sei.m_postLutNumValMinus1[c] ; i++ )
828        {
829          WRITE_CODE( sei.m_postLut[c][i].codedValue, (( sei.m_colourRemapBitDepth + 7 ) >> 3 ) << 3, "post_lut_coded_value[c][i]" );
830          WRITE_CODE( sei.m_postLut[c][i].targetValue, (( sei.m_colourRemapBitDepth + 7 ) >> 3 ) << 3, "post_lut_target_value[c][i]" );
831        }
832      }
833    }
834  }
835}
836
837Void SEIWriter::xWriteSEIMasteringDisplayColourVolume(const SEIMasteringDisplayColourVolume& sei)
838{
839  WRITE_CODE( sei.values.primaries[0][0],  16,  "display_primaries_x[0]" );
840  WRITE_CODE( sei.values.primaries[0][1],  16,  "display_primaries_y[0]" );
841
842  WRITE_CODE( sei.values.primaries[1][0],  16,  "display_primaries_x[1]" );
843  WRITE_CODE( sei.values.primaries[1][1],  16,  "display_primaries_y[1]" );
844
845  WRITE_CODE( sei.values.primaries[2][0],  16,  "display_primaries_x[2]" );
846  WRITE_CODE( sei.values.primaries[2][1],  16,  "display_primaries_y[2]" );
847
848  WRITE_CODE( sei.values.whitePoint[0],    16,  "white_point_x" );
849  WRITE_CODE( sei.values.whitePoint[1],    16,  "white_point_y" );
850   
851  WRITE_CODE( sei.values.maxLuminance,     32,  "max_display_mastering_luminance" );
852  WRITE_CODE( sei.values.minLuminance,     32,  "min_display_mastering_luminance" );
853}
854
855
856Void SEIWriter::xWriteByteAlign()
857{
858  if( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0)
859  {
860    WRITE_FLAG( 1, "payload_bit_equal_to_one" );
861    while( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
862    {
863      WRITE_FLAG( 0, "payload_bit_equal_to_zero" );
864    }
865  }
866}
867
868#if NH_MV
869Void SEIWriter::xWriteSEILayersNotPresent(const SEILayersNotPresent& sei)
870{
871  WRITE_CODE( sei.m_lnpSeiActiveVpsId, 4, "lnp_sei_active_vps_id" );
872  for( Int i = 0; i  <  sei.m_lnpSeiMaxLayers; i++ )
873  {
874    WRITE_FLAG( ( sei.m_layerNotPresentFlag[i] ? 1 : 0 ), "layer_not_present_flag" );
875  }
876};
877
878Void SEIWriter::xWriteSEIInterLayerConstrainedTileSets( const SEIInterLayerConstrainedTileSets& sei)
879{
880  WRITE_FLAG( ( sei.m_ilAllTilesExactSampleValueMatchFlag ? 1 : 0 ), "il_all_tiles_exact_sample_value_match_flag" );
881  WRITE_FLAG( ( sei.m_ilOneTilePerTileSetFlag ? 1 : 0 ),             "il_one_tile_per_tile_set_flag" );
882  if( !sei.m_ilOneTilePerTileSetFlag )
883  {
884    WRITE_UVLC( sei.m_ilNumSetsInMessageMinus1, "il_num_sets_in_message_minus1" );
885    if( sei.m_ilNumSetsInMessageMinus1 )
886    {
887      WRITE_FLAG( ( sei.m_skippedTileSetPresentFlag ? 1 : 0 ), "skipped_tile_set_present_flag" );
888    }
889    Int numSignificantSets = sei.m_ilNumSetsInMessageMinus1 - sei.m_skippedTileSetPresentFlag + 1;
890    for( Int i = 0; i < numSignificantSets; i++ )
891    {
892      WRITE_UVLC( sei.m_ilctsId[i],                   "ilcts_id" );
893      WRITE_UVLC( sei.m_ilNumTileRectsInSetMinus1[i], "il_num_tile_rects_in_set_minus1" );
894      for( Int j = 0; j  <= sei.m_ilNumTileRectsInSetMinus1[ i ]; j++ )
895      {
896        WRITE_UVLC( sei.m_ilTopLeftTileIndex[i][j],     "il_top_left_tile_index" );
897        WRITE_UVLC( sei.m_ilBottomRightTileIndex[i][j], "il_bottom_right_tile_index" );
898      }
899      WRITE_CODE( sei.m_ilcIdc[i], 2, "ilc_idc" );
900      if ( !sei.m_ilAllTilesExactSampleValueMatchFlag )
901      {
902        WRITE_FLAG( ( sei.m_ilExactSampleValueMatchFlag[i] ? 1 : 0 ), "il_exact_sample_value_match_flag" );
903      }
904    }
905  }
906  else
907  {
908    WRITE_CODE( sei.m_allTilesIlcIdc, 2, "all_tiles_ilc_idc" );
909  }
910};
911
912#if NH_MV_SEI_TBD
913Void SEIWriter::xWriteSEIBspNesting( const SEIBspNesting& sei)
914{
915  WRITE_UVLC( sei.m_seiOlsIdx, "sei_ols_idx" );
916  WRITE_UVLC( sei.m_seiPartitioningSchemeIdx, "sei_partitioning_scheme_idx" );
917  WRITE_UVLC( sei.m_bspIdx, "bsp_idx" );
918  while( !ByteaLigned(() ) );
919  {
920    WRITE_CODE( sei.m_bspNestingZeroBit, *equalto0*/u1, "bsp_nesting_zero_bit" );
921  }
922  WRITE_UVLC( sei.m_numSeisInBspMinus1, "num_seis_in_bsp_minus1" );
923  for( Int i = 0; i  <=  NumSeisInBspMinus1( ); i++ )
924  {
925    SeiMessage(() );
926  }
927};
928
929Void SEIWriter::xWriteSEIBspInitialArrivalTime( const SEIBspInitialArrivalTime& sei)
930{
931  psIdx = SeiPartitioningSchemeIdx();
932  if( nalInitialArrivalDelayPresent )
933  {
934    for( Int i = 0; i < BspSchedCnt( SeiOlsIdx(), psIdx, MaxTemporalId( 0 ) ); i++ )
935    {
936      WRITE_CODE( sei.m_nalInitialArrivalDelay[i], getNalInitialArrivalDelayLen ), "nal_initial_arrival_delay" );
937    }
938  }
939  if( vclInitialArrivalDelayPresent )
940  {
941    for( Int i = 0; i < BspSchedCnt( SeiOlsIdx(), psIdx, MaxTemporalId( 0 ) ); i++ )
942    {
943      WRITE_CODE( sei.m_vclInitialArrivalDelay[i], getVclInitialArrivalDelayLen ), "vcl_initial_arrival_delay" );
944    }
945  }
946};
947#endif
948Void SEIWriter::xWriteSEISubBitstreamProperty( const SEISubBitstreamProperty& sei)
949{
950  WRITE_CODE( sei.m_sbPropertyActiveVpsId, 4,      "sb_property_active_vps_id"         );
951  WRITE_UVLC( sei.m_numAdditionalSubStreamsMinus1, "num_additional_sub_streams_minus1" );
952  for( Int i = 0; i  <=  sei.m_numAdditionalSubStreamsMinus1; i++ )
953  {
954    WRITE_CODE( sei.m_subBitstreamMode    [i], 2,  "sub_bitstream_mode"       );
955    WRITE_UVLC( sei.m_olsIdxToVps         [i],     "ols_idx_to_vps"           );
956    WRITE_CODE( sei.m_highestSublayerId   [i], 3,  "highest_sublayer_id"      );
957    WRITE_CODE( sei.m_avgSbPropertyBitRate[i], 16, "avg_sb_property_bit_rate" );
958    WRITE_CODE( sei.m_maxSbPropertyBitRate[i], 16, "max_sb_property_bit_rate" );
959  }
960};
961
962Void SEIWriter::xWriteSEIAlphaChannelInfo( const SEIAlphaChannelInfo& sei)
963{
964  WRITE_FLAG( ( sei.m_alphaChannelCancelFlag ? 1 : 0 ), "alpha_channel_cancel_flag" );
965  if( !sei.m_alphaChannelCancelFlag )
966  {
967    WRITE_CODE( sei.m_alphaChannelUseIdc, 3, "alpha_channel_use_idc" );
968    WRITE_CODE( sei.m_alphaChannelBitDepthMinus8, 3, "alpha_channel_bit_depth_minus8" );
969    WRITE_CODE( sei.m_alphaTransparentValue, sei.m_alphaChannelBitDepthMinus8+9, "alpha_transparent_value" );
970    WRITE_CODE( sei.m_alphaOpaqueValue, sei.m_alphaChannelBitDepthMinus8+9, "alpha_opaque_value" );
971    WRITE_FLAG( ( sei.m_alphaChannelIncrFlag ? 1 : 0 ), "alpha_channel_incr_flag" );
972    WRITE_FLAG( ( sei.m_alphaChannelClipFlag ? 1 : 0 ), "alpha_channel_clip_flag" );
973    if( sei.m_alphaChannelClipFlag )
974    {
975      WRITE_FLAG( ( sei.m_alphaChannelClipTypeFlag ? 1 : 0 ), "alpha_channel_clip_type_flag" );
976    }
977  }
978};
979
980Void SEIWriter::xWriteSEIOverlayInfo( const SEIOverlayInfo& sei)
981{
982  WRITE_FLAG( ( sei.m_overlayInfoCancelFlag ? 1 : 0 ), "overlay_info_cancel_flag" );
983  if( !sei.m_overlayInfoCancelFlag )
984  {
985    WRITE_UVLC( sei.m_overlayContentAuxIdMinus128, "overlay_content_aux_id_minus128" );
986    WRITE_UVLC( sei.m_overlayLabelAuxIdMinus128, "overlay_label_aux_id_minus128" );
987    WRITE_UVLC( sei.m_overlayAlphaAuxIdMinus128, "overlay_alpha_aux_id_minus128" );
988    WRITE_UVLC( sei.m_overlayElementLabelValueLengthMinus8, "overlay_element_label_value_length_minus8" );
989    WRITE_UVLC( sei.m_numOverlaysMinus1, "num_overlays_minus1" );
990    for( Int i = 0; i  <=  sei.m_numOverlaysMinus1 ; i++ )
991    {
992      WRITE_UVLC( sei.m_overlayIdx[i], "overlay_idx" );
993      WRITE_FLAG( ( sei.m_languageOverlayPresentFlag[i] ? 1 : 0 ), "language_overlay_present_flag" );
994      WRITE_CODE( sei.m_overlayContentLayerId[i], 6, "overlay_content_layer_id" );
995      WRITE_FLAG( ( sei.m_overlayLabelPresentFlag[i] ? 1 : 0 ), "overlay_label_present_flag" );
996      if( sei.m_overlayLabelPresentFlag[i] )
997      {
998        WRITE_CODE( sei.m_overlayLabelLayerId[i], 6, "overlay_label_layer_id" );
999      }
1000      WRITE_FLAG( ( sei.m_overlayAlphaPresentFlag[i] ? 1 : 0 ), "overlay_alpha_present_flag" );
1001      if( sei.m_overlayAlphaPresentFlag[i] )
1002      {
1003        WRITE_CODE( sei.m_overlayAlphaLayerId[i], 6, "overlay_alpha_layer_id" );
1004      }
1005      if( sei.m_overlayLabelPresentFlag[i] )
1006      {
1007        WRITE_UVLC( sei.m_numOverlayElementsMinus1[i], "num_overlay_elements_minus1" );
1008        for( Int j = 0; j  <=  sei.m_numOverlayElementsMinus1[i]; j++ )
1009        {
1010          WRITE_CODE( sei.m_overlayElementLabelMin[i][j], (sei.m_overlayElementLabelValueLengthMinus8 + 8), "overlay_element_label_min" );
1011          WRITE_CODE( sei.m_overlayElementLabelMax[i][j], (sei.m_overlayElementLabelValueLengthMinus8 + 8), "overlay_element_label_max" );
1012        }
1013      }
1014    }
1015
1016    // byte alignment
1017    while ( m_pcBitIf->getNumberOfWrittenBits() % 8 != 0 )
1018    {
1019      WRITE_FLAG( 0, "overlay_zero_bit" );
1020    }
1021
1022    UChar* stmp;
1023    UInt ilength;
1024    for( Int i = 0; i  <=  sei.m_numOverlaysMinus1; i++ )
1025    {
1026      if( sei.m_languageOverlayPresentFlag[i] )
1027      {
1028        stmp = (UChar*) strdup( sei.m_overlayLanguage[i].c_str() );
1029        ilength = (UInt) sei.m_overlayLanguage[i].size();
1030        WRITE_STRING( stmp, ilength, "overlay_language" );
1031        free(stmp);
1032      }
1033      stmp = (UChar*) strdup( sei.m_overlayName[i].c_str() );
1034      ilength = (UInt) sei.m_overlayName[i].size();
1035      WRITE_STRING( stmp, ilength, "overlay_name" );
1036      free(stmp);
1037      if( sei.m_overlayLabelPresentFlag[i] )
1038      {
1039        for( Int j = 0; j  <=  sei.m_numOverlayElementsMinus1[i]; j++ )
1040        {
1041          stmp = (UChar*) strdup( sei.m_overlayElementName[i][j].c_str() );
1042          ilength = (UInt) sei.m_overlayElementName[i][j].size();
1043          WRITE_STRING( stmp, ilength, "overlay_element_name" );
1044          free(stmp);
1045        }
1046      }
1047    }
1048    WRITE_FLAG( ( sei.m_overlayInfoPersistenceFlag ? 1 : 0 ), "overlay_info_persistence_flag" );
1049  }
1050};
1051
1052
1053Void SEIWriter::xWriteSEITemporalMvPredictionConstraints( const SEITemporalMvPredictionConstraints& sei)
1054{
1055  WRITE_FLAG( ( sei.m_prevPicsNotUsedFlag    ? 1 : 0 ), "prev_pics_not_used_flag"     );
1056  WRITE_FLAG( ( sei.m_noIntraLayerColPicFlag ? 1 : 0 ), "no_intra_layer_col_pic_flag" );
1057};
1058
1059#if NH_MV_SEI_TBD
1060Void SEIWriter::xWriteSEIFrameFieldInfo( const SEIFrameFieldInfo& sei)
1061{
1062  WRITE_CODE( sei.m_ffinfoPicStruct, 4, "ffinfo_pic_struct" );
1063  WRITE_CODE( sei.m_ffinfoSourceScanType, 2, "ffinfo_source_scan_type" );
1064  WRITE_FLAG( ( sei.m_ffinfoDuplicateFlag ? 1 : 0 ), "ffinfo_duplicate_flag" );
1065};
1066#endif
1067
1068Void SEIWriter::xWriteSEIThreeDimensionalReferenceDisplaysInfo( const SEIThreeDimensionalReferenceDisplaysInfo& sei)
1069{
1070  WRITE_UVLC( sei.m_precRefDisplayWidth, "prec_ref_display_width" );
1071  WRITE_FLAG( ( sei.m_refViewingDistanceFlag ? 1 : 0 ), "ref_viewing_distance_flag" );
1072  if( sei.m_refViewingDistanceFlag )
1073  {
1074    WRITE_UVLC( sei.m_precRefViewingDist, "prec_ref_viewing_dist" );
1075  }
1076  WRITE_UVLC( sei.m_numRefDisplaysMinus1, "num_ref_displays_minus1" );
1077  for( Int i = 0; i  <=  sei.getNumRefDisplaysMinus1( ); i++ )
1078  {
1079    WRITE_UVLC( sei.m_leftViewId[i], "left_view_id" );
1080    WRITE_UVLC( sei.m_rightViewId[i], "right_view_id" );
1081    WRITE_CODE( sei.m_exponentRefDisplayWidth[i], 6, "exponent_ref_display_width" );
1082    WRITE_CODE( sei.m_mantissaRefDisplayWidth[i], sei.getMantissaReferenceDisplayWidthLen(i) , "mantissa_ref_display_width" );
1083    if( sei.m_refViewingDistanceFlag )
1084    {
1085      WRITE_CODE( sei.m_exponentRefViewingDistance[i], 6, "exponent_ref_viewing_distance" );
1086      WRITE_CODE( sei.m_mantissaRefViewingDistance[i], sei.getMantissaReferenceViewingDistanceLen(i), "mantissa_ref_viewing_distance" );
1087    }
1088    WRITE_FLAG( ( sei.m_additionalShiftPresentFlag[i] ? 1 : 0 ), "additional_shift_present_flag" );
1089    if( sei.m_additionalShiftPresentFlag[i] )
1090    {
1091      WRITE_CODE( sei.m_numSampleShiftPlus512[i], 10, "num_sample_shift_plus512" );
1092    }
1093  }
1094  WRITE_FLAG( ( sei.m_threeDimensionalReferenceDisplaysExtensionFlag ? 1 : 0 ), "three_dimensional_reference_displays_extension_flag" );
1095};
1096
1097Void SEIWriter::xWriteSEIDepthRepresentationInfo( const SEIDepthRepresentationInfo& sei)
1098{
1099
1100    assert(sei.m_currLayerID>=0);
1101
1102    WRITE_FLAG( ( sei.m_zNearFlag[sei.m_currLayerID] ? 1 : 0 ), "z_near_flag" );
1103    WRITE_FLAG( ( sei.m_zFarFlag[sei.m_currLayerID] ? 1 : 0 ), "z_far_flag" );
1104    WRITE_FLAG( ( sei.m_dMinFlag[sei.m_currLayerID] ? 1 : 0 ), "d_min_flag" );
1105    WRITE_FLAG( ( sei.m_dMaxFlag[sei.m_currLayerID] ? 1 : 0 ), "d_max_flag" );
1106    WRITE_UVLC( sei.m_depthRepresentationType[sei.m_currLayerID][0], "depth_representation_type" );
1107    if( sei.m_dMinFlag[sei.m_currLayerID]  ||  sei.m_dMaxFlag[sei.m_currLayerID] )
1108    {
1109        WRITE_UVLC( sei.m_disparityRefViewId[sei.m_currLayerID][0], "disparity_ref_view_id" );
1110    }
1111    if( sei.m_zNearFlag[sei.m_currLayerID] )
1112    {
1113        xWriteSEIDepthRepInfoElement(sei.m_zNear[sei.m_currLayerID][0]);
1114    }
1115    if( sei.m_zFarFlag[sei.m_currLayerID] )
1116    {
1117        xWriteSEIDepthRepInfoElement(sei.m_zFar[sei.m_currLayerID][0]);
1118    }
1119    if( sei.m_dMinFlag[sei.m_currLayerID] )
1120    {
1121        xWriteSEIDepthRepInfoElement(sei.m_dMin[sei.m_currLayerID][0]);
1122    }
1123    if( sei.m_dMaxFlag[sei.m_currLayerID] )
1124    {
1125        xWriteSEIDepthRepInfoElement(sei.m_dMax[sei.m_currLayerID][0]);
1126    }
1127
1128    if (sei.m_depthRepresentationType[sei.m_currLayerID][0] == 3)
1129    {
1130        WRITE_UVLC( sei.m_depthNonlinearRepresentationNumMinus1[sei.m_currLayerID][0], "depth_nonlinear_representation_num_minus1" );
1131        for( Int i = 1; i  <=  sei.m_depthNonlinearRepresentationNumMinus1[sei.m_currLayerID][0] + 1; i++ )
1132        {
1133            WRITE_UVLC(sei.m_depth_nonlinear_representation_model[sei.m_currLayerID][i-1],"depth_nonlinear_representation_model[ i ]");
1134        }
1135    }
1136}
1137
1138Void SEIWriter::xWriteSEIDepthRepInfoElement( double f)
1139{
1140    UInt x_sign, x_exp, x_mantissa,x_mantissa_len;
1141    if (f < 0)
1142    {
1143        f = f * (-1);
1144        x_sign = 1;
1145    }
1146    else
1147    {
1148        x_sign = 0;
1149    }
1150    int exponent=0;
1151    if(f >= 1)
1152    {
1153        while(f>=2)
1154        {
1155            exponent++;
1156            f = f/2;
1157        }
1158    }
1159    else
1160    {
1161        while (f<1)
1162        {
1163            exponent++;
1164            f = f*2;
1165        }
1166        exponent=-exponent;
1167    }
1168
1169    int i;
1170    f = f -1;
1171    double s = 1;
1172    char s_mantissa[32];
1173    double thr=1.0/(4.0*(1<<30));
1174
1175    if (f>=thr)
1176    {
1177        for(i=0;i<32;i++)
1178        {
1179            s /= 2;
1180            if(f>=s)
1181            {
1182                f = f-s;
1183                s_mantissa[i]=1;
1184
1185                if (f<thr)
1186                    break;
1187            }
1188            else
1189            {
1190                s_mantissa[i]=0;
1191            }
1192        }
1193
1194        if (i<32)
1195            x_mantissa_len=i+1;
1196        else
1197            x_mantissa_len=32;
1198
1199        x_mantissa=0;
1200
1201        for(i=0;i<x_mantissa_len;i++)
1202        {
1203            if (s_mantissa[i]==1)
1204                x_mantissa += (1u)<<(x_mantissa_len-1-i) ;
1205        }
1206
1207    }
1208    else
1209    {
1210        x_mantissa=0;
1211        x_mantissa_len=1;
1212    }
1213
1214    assert(exponent>=-31 && exponent<= (1<<7)-32);
1215    x_exp=exponent+31;
1216
1217    WRITE_FLAG(  x_sign,                         "da_sign_flag" );
1218    WRITE_CODE(  x_exp, 7 ,                      "da_exponent" );
1219    WRITE_CODE( x_mantissa_len-1, 5 ,            "da_mantissa_len_minus1" );
1220    WRITE_CODE( x_mantissa, x_mantissa_len ,     "da_mantissa" );
1221
1222};
1223
1224Void SEIWriter::xWriteSEIMultiviewSceneInfo( const SEIMultiviewSceneInfo& sei)
1225{
1226  WRITE_SVLC( sei.m_minDisparity     , "min_disparity"       );
1227  WRITE_UVLC( sei.m_maxDisparityRange, "max_disparity_range" );
1228};
1229
1230
1231Void SEIWriter::xWriteSEIMultiviewAcquisitionInfo( const SEIMultiviewAcquisitionInfo& sei)
1232{
1233  WRITE_FLAG( ( sei.m_intrinsicParamFlag ? 1 : 0 ), "intrinsic_param_flag" );
1234  WRITE_FLAG( ( sei.m_extrinsicParamFlag ? 1 : 0 ), "extrinsic_param_flag" );
1235  if( sei.m_intrinsicParamFlag )
1236  {
1237    WRITE_FLAG( ( sei.m_intrinsicParamsEqualFlag ? 1 : 0 ), "intrinsic_params_equal_flag" );
1238    WRITE_UVLC(   sei.m_precFocalLength                   , "prec_focal_length"           );
1239    WRITE_UVLC(   sei.m_precPrincipalPoint                , "prec_principal_point"        );
1240    WRITE_UVLC(   sei.m_precSkewFactor                    , "prec_skew_factor"            );
1241
1242    for( Int i = 0; i  <=  ( sei.m_intrinsicParamsEqualFlag ? 0 : sei.getNumViewsMinus1() ); i++ )
1243    {
1244      WRITE_FLAG( ( sei.m_signFocalLengthX       [i] ? 1 : 0 ),                                         "sign_focal_length_x"        );
1245      WRITE_CODE(   sei.m_exponentFocalLengthX   [i]          , 6                                  ,    "exponent_focal_length_x"    );
1246      WRITE_CODE(   sei.m_mantissaFocalLengthX   [i]          , sei.getMantissaFocalLengthXLen( i ),    "mantissa_focal_length_x"    );
1247      WRITE_FLAG( ( sei.m_signFocalLengthY       [i] ? 1 : 0 ),                                         "sign_focal_length_y"        );
1248      WRITE_CODE(   sei.m_exponentFocalLengthY   [i]          , 6                                  ,    "exponent_focal_length_y"    );
1249      WRITE_CODE(   sei.m_mantissaFocalLengthY   [i]          , sei.getMantissaFocalLengthYLen( i ),    "mantissa_focal_length_y"    );
1250      WRITE_FLAG( ( sei.m_signPrincipalPointX    [i] ? 1 : 0 ),                                         "sign_principal_point_x"     );
1251      WRITE_CODE(   sei.m_exponentPrincipalPointX[i]          , 6,                                      "exponent_principal_point_x" );
1252      WRITE_CODE(   sei.m_mantissaPrincipalPointX[i]          , sei.getMantissaPrincipalPointXLen( i ), "mantissa_principal_point_x" );
1253      WRITE_FLAG( ( sei.m_signPrincipalPointY    [i] ? 1 : 0 ),                                         "sign_principal_point_y"     );
1254      WRITE_CODE(   sei.m_exponentPrincipalPointY[i]          , 6,                                      "exponent_principal_point_y" );
1255      WRITE_CODE(   sei.m_mantissaPrincipalPointY[i]          , sei.getMantissaPrincipalPointYLen( i ), "mantissa_principal_point_y" );
1256      WRITE_FLAG( ( sei.m_signSkewFactor         [i] ? 1 : 0 ),                                         "sign_skew_factor"           );
1257      WRITE_CODE(   sei.m_exponentSkewFactor     [i]          , 6,                                      "exponent_skew_factor"       );
1258      WRITE_CODE(   sei.m_mantissaSkewFactor     [i]          , sei.getMantissaSkewFactorLen( i )  ,    "mantissa_skew_factor"       );
1259    }
1260  }
1261  if( sei.m_extrinsicParamFlag )
1262  {
1263    WRITE_UVLC( sei.m_precRotationParam   , "prec_rotation_param"    );
1264    WRITE_UVLC( sei.m_precTranslationParam, "prec_translation_param" );
1265    for( Int i = 0; i  <=  sei.getNumViewsMinus1(); i++ )
1266    {
1267      for( Int j = 0; j  <=  2; j++ )  /* row */
1268      {
1269        for( Int k = 0; k  <=  2; k++ )  /* column */
1270        {
1271          WRITE_FLAG( ( sei.m_signR    [i][j][k] ? 1 : 0 ),                                "sign_r"     );
1272          WRITE_CODE(   sei.m_exponentR[i][j][k]          , 6,                             "exponent_r" );
1273          WRITE_CODE(   sei.m_mantissaR[i][j][k]          , sei.getMantissaRLen( i,j,k ) , "mantissa_r" );
1274        }
1275        WRITE_FLAG( ( sei.m_signT    [i][j] ? 1 : 0 ),                          "sign_t"     );
1276        WRITE_CODE(   sei.m_exponentT[i][j]          , 6,                       "exponent_t" );
1277        WRITE_CODE(   sei.m_mantissaT[i][j]          , sei.getMantissaTLen( i,j ),"mantissa_t" );
1278      }
1279    }
1280  }
1281};
1282
1283Void SEIWriter::xWriteSEIMultiviewViewPosition( const SEIMultiviewViewPosition& sei)
1284{
1285  WRITE_UVLC( sei.m_numViewsMinus1, "num_views_minus1" );
1286  for( Int i = 0; i  <=  sei.m_numViewsMinus1; i++ )
1287  {
1288    WRITE_UVLC( sei.m_viewPosition[i], "view_position" );
1289  }
1290};
1291
1292#endif
1293
1294//! \}
Note: See TracBrowser for help on using the repository browser.