source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibEncoder/SEIEncoder.cpp @ 1590

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

port rev 4752
File size: 41.2 KB
RevLine 
[1273]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 *
[1549]6 * Copyright (c) 2010-2016, ITU/ISO/IEC
[1273]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/CommonDef.h"
35#include "TLibCommon/SEI.h"
36#include "TEncGOP.h"
37#include "TEncTop.h"
38
39//! \ingroup TLibEncoder
40//! \{
41
42Void SEIEncoder::initSEIActiveParameterSets (SEIActiveParameterSets *seiActiveParameterSets, const TComVPS *vps, const TComSPS *sps)
43{
44  assert (m_isInitialized);
45  assert (seiActiveParameterSets!=NULL);
46  assert (vps!=NULL);
47  assert (sps!=NULL);
48
49  seiActiveParameterSets->activeVPSId = vps->getVPSId(); 
50  seiActiveParameterSets->m_selfContainedCvsFlag = false;
51  seiActiveParameterSets->m_noParameterSetUpdateFlag = false;
52#if R0247_SEI_ACTIVE
53  seiActiveParameterSets->numSpsIdsMinus1 = m_pcCfg->getNumLayer()-1;
54  seiActiveParameterSets->activeSeqParameterSetId.resize(seiActiveParameterSets->numSpsIdsMinus1 + 1);
55  seiActiveParameterSets->layerSpsIdx.resize(seiActiveParameterSets->numSpsIdsMinus1+ 1); 
56  for (Int c=0; c <= seiActiveParameterSets->numSpsIdsMinus1; c++)
57  {
58     seiActiveParameterSets->activeSeqParameterSetId[c] = c;
59  }
60  for (Int c=1; c <= seiActiveParameterSets->numSpsIdsMinus1; c++)
61  {
62     seiActiveParameterSets->layerSpsIdx[c] = c;
63  }
64#else
65  seiActiveParameterSets->numSpsIdsMinus1 = 0;
66  seiActiveParameterSets->activeSeqParameterSetId.resize(seiActiveParameterSets->numSpsIdsMinus1 + 1);
67  seiActiveParameterSets->activeSeqParameterSetId[0] = sps->getSPSId();
68#endif
69}
70
71Void SEIEncoder::initSEIFramePacking(SEIFramePacking *seiFramePacking, Int currPicNum)
72{
73  assert (m_isInitialized);
74  assert (seiFramePacking!=NULL);
75
76  seiFramePacking->m_arrangementId = m_pcCfg->getFramePackingArrangementSEIId();
77  seiFramePacking->m_arrangementCancelFlag = 0;
78  seiFramePacking->m_arrangementType = m_pcCfg->getFramePackingArrangementSEIType();
79  assert((seiFramePacking->m_arrangementType > 2) && (seiFramePacking->m_arrangementType < 6) );
80  seiFramePacking->m_quincunxSamplingFlag = m_pcCfg->getFramePackingArrangementSEIQuincunx();
81  seiFramePacking->m_contentInterpretationType = m_pcCfg->getFramePackingArrangementSEIInterpretation();
82  seiFramePacking->m_spatialFlippingFlag = 0;
83  seiFramePacking->m_frame0FlippedFlag = 0;
84  seiFramePacking->m_fieldViewsFlag = (seiFramePacking->m_arrangementType == 2);
85  seiFramePacking->m_currentFrameIsFrame0Flag = ((seiFramePacking->m_arrangementType == 5) && (currPicNum&1) );
86  seiFramePacking->m_frame0SelfContainedFlag = 0;
87  seiFramePacking->m_frame1SelfContainedFlag = 0;
88  seiFramePacking->m_frame0GridPositionX = 0;
89  seiFramePacking->m_frame0GridPositionY = 0;
90  seiFramePacking->m_frame1GridPositionX = 0;
91  seiFramePacking->m_frame1GridPositionY = 0;
92  seiFramePacking->m_arrangementReservedByte = 0;
93  seiFramePacking->m_arrangementPersistenceFlag = true;
94  seiFramePacking->m_upsampledAspectRatio = 0;
95}
96
97Void SEIEncoder::initSEISegmentedRectFramePacking(SEISegmentedRectFramePacking *seiSegmentedRectFramePacking)
98{
99  assert (m_isInitialized);
100  assert (seiSegmentedRectFramePacking!=NULL);
101
102  seiSegmentedRectFramePacking->m_arrangementCancelFlag = m_pcCfg->getSegmentedRectFramePackingArrangementSEICancel();
103  seiSegmentedRectFramePacking->m_contentInterpretationType = m_pcCfg->getSegmentedRectFramePackingArrangementSEIType();
104  seiSegmentedRectFramePacking->m_arrangementPersistenceFlag = m_pcCfg->getSegmentedRectFramePackingArrangementSEIPersistence();
105}
106
107Void SEIEncoder::initSEIDisplayOrientation(SEIDisplayOrientation* seiDisplayOrientation)
108{
109  assert (m_isInitialized);
110  assert (seiDisplayOrientation!=NULL);
111
112  seiDisplayOrientation->cancelFlag = false;
113  seiDisplayOrientation->horFlip = false;
114  seiDisplayOrientation->verFlip = false;
115  seiDisplayOrientation->anticlockwiseRotation = m_pcCfg->getDisplayOrientationSEIAngle();
116}
117
118Void SEIEncoder::initSEIToneMappingInfo(SEIToneMappingInfo *seiToneMappingInfo)
119{
120  assert (m_isInitialized);
121  assert (seiToneMappingInfo!=NULL);
122
123  seiToneMappingInfo->m_toneMapId = m_pcCfg->getTMISEIToneMapId();
124  seiToneMappingInfo->m_toneMapCancelFlag = m_pcCfg->getTMISEIToneMapCancelFlag();
125  seiToneMappingInfo->m_toneMapPersistenceFlag = m_pcCfg->getTMISEIToneMapPersistenceFlag();
126
127  seiToneMappingInfo->m_codedDataBitDepth = m_pcCfg->getTMISEICodedDataBitDepth();
128  assert(seiToneMappingInfo->m_codedDataBitDepth >= 8 && seiToneMappingInfo->m_codedDataBitDepth <= 14);
129  seiToneMappingInfo->m_targetBitDepth = m_pcCfg->getTMISEITargetBitDepth();
130  assert(seiToneMappingInfo->m_targetBitDepth >= 1 && seiToneMappingInfo->m_targetBitDepth <= 17);
131  seiToneMappingInfo->m_modelId = m_pcCfg->getTMISEIModelID();
132  assert(seiToneMappingInfo->m_modelId >=0 &&seiToneMappingInfo->m_modelId<=4);
133
134  switch( seiToneMappingInfo->m_modelId)
135  {
136  case 0:
137    {
138      seiToneMappingInfo->m_minValue = m_pcCfg->getTMISEIMinValue();
139      seiToneMappingInfo->m_maxValue = m_pcCfg->getTMISEIMaxValue();
140      break;
141    }
142  case 1:
143    {
144      seiToneMappingInfo->m_sigmoidMidpoint = m_pcCfg->getTMISEISigmoidMidpoint();
145      seiToneMappingInfo->m_sigmoidWidth = m_pcCfg->getTMISEISigmoidWidth();
146      break;
147    }
148  case 2:
149    {
150      UInt num = 1u<<(seiToneMappingInfo->m_targetBitDepth);
151      seiToneMappingInfo->m_startOfCodedInterval.resize(num);
152      Int* ptmp = m_pcCfg->getTMISEIStartOfCodedInterva();
153      if(ptmp)
154      {
155        for(Int i=0; i<num;i++)
156        {
157          seiToneMappingInfo->m_startOfCodedInterval[i] = ptmp[i];
158        }
159      }
160      break;
161    }
162  case 3:
163    {
164      seiToneMappingInfo->m_numPivots = m_pcCfg->getTMISEINumPivots();
165      seiToneMappingInfo->m_codedPivotValue.resize(seiToneMappingInfo->m_numPivots);
166      seiToneMappingInfo->m_targetPivotValue.resize(seiToneMappingInfo->m_numPivots);
167      Int* ptmpcoded = m_pcCfg->getTMISEICodedPivotValue();
168      Int* ptmptarget = m_pcCfg->getTMISEITargetPivotValue();
169      if(ptmpcoded&&ptmptarget)
170      {
171        for(Int i=0; i<(seiToneMappingInfo->m_numPivots);i++)
172        {
173          seiToneMappingInfo->m_codedPivotValue[i]=ptmpcoded[i];
174          seiToneMappingInfo->m_targetPivotValue[i]=ptmptarget[i];
175        }
176      }
177      break;
178    }
179  case 4:
180    {
181      seiToneMappingInfo->m_cameraIsoSpeedIdc = m_pcCfg->getTMISEICameraIsoSpeedIdc();
182      seiToneMappingInfo->m_cameraIsoSpeedValue = m_pcCfg->getTMISEICameraIsoSpeedValue();
183      assert( seiToneMappingInfo->m_cameraIsoSpeedValue !=0 );
184      seiToneMappingInfo->m_exposureIndexIdc = m_pcCfg->getTMISEIExposurIndexIdc();
185      seiToneMappingInfo->m_exposureIndexValue = m_pcCfg->getTMISEIExposurIndexValue();
186      assert( seiToneMappingInfo->m_exposureIndexValue !=0 );
187      seiToneMappingInfo->m_exposureCompensationValueSignFlag = m_pcCfg->getTMISEIExposureCompensationValueSignFlag();
188      seiToneMappingInfo->m_exposureCompensationValueNumerator = m_pcCfg->getTMISEIExposureCompensationValueNumerator();
189      seiToneMappingInfo->m_exposureCompensationValueDenomIdc = m_pcCfg->getTMISEIExposureCompensationValueDenomIdc();
190      seiToneMappingInfo->m_refScreenLuminanceWhite = m_pcCfg->getTMISEIRefScreenLuminanceWhite();
191      seiToneMappingInfo->m_extendedRangeWhiteLevel = m_pcCfg->getTMISEIExtendedRangeWhiteLevel();
192      assert( seiToneMappingInfo->m_extendedRangeWhiteLevel >= 100 );
193      seiToneMappingInfo->m_nominalBlackLevelLumaCodeValue = m_pcCfg->getTMISEINominalBlackLevelLumaCodeValue();
194      seiToneMappingInfo->m_nominalWhiteLevelLumaCodeValue = m_pcCfg->getTMISEINominalWhiteLevelLumaCodeValue();
195      assert( seiToneMappingInfo->m_nominalWhiteLevelLumaCodeValue > seiToneMappingInfo->m_nominalBlackLevelLumaCodeValue );
196      seiToneMappingInfo->m_extendedWhiteLevelLumaCodeValue = m_pcCfg->getTMISEIExtendedWhiteLevelLumaCodeValue();
197      assert( seiToneMappingInfo->m_extendedWhiteLevelLumaCodeValue >= seiToneMappingInfo->m_nominalWhiteLevelLumaCodeValue );
198      break;
199    }
200  default:
201    {
202      assert(!"Undefined SEIToneMapModelId");
203      break;
204    }
205  }
206}
207
208Void SEIEncoder::initSEISOPDescription(SEISOPDescription *sopDescriptionSEI, TComSlice *slice, Int picInGOP, Int lastIdr, Int currGOPSize)
209{
210  assert (m_isInitialized);
211  assert (sopDescriptionSEI != NULL);
212  assert (slice != NULL);
213
214  Int sopCurrPOC = slice->getPOC();
215  sopDescriptionSEI->m_sopSeqParameterSetId = slice->getSPS()->getSPSId();
216
217  Int i = 0;
218  Int prevEntryId = picInGOP;
219  for (Int j = picInGOP; j < currGOPSize; j++)
220  {
221    Int deltaPOC = m_pcCfg->getGOPEntry(j).m_POC - m_pcCfg->getGOPEntry(prevEntryId).m_POC;
222    if ((sopCurrPOC + deltaPOC) < m_pcCfg->getFramesToBeEncoded())
223    {
224      sopCurrPOC += deltaPOC;
225      sopDescriptionSEI->m_sopDescVclNaluType[i] = m_pcEncGOP->getNalUnitType(sopCurrPOC, lastIdr, slice->getPic()->isField());
226      sopDescriptionSEI->m_sopDescTemporalId[i] = m_pcCfg->getGOPEntry(j).m_temporalId;
[1307]227      sopDescriptionSEI->m_sopDescStRpsIdx[i] = m_pcEncTop->getReferencePictureSetIdxForSOP(sopCurrPOC, j);
[1273]228      sopDescriptionSEI->m_sopDescPocDelta[i] = deltaPOC;
229
230      prevEntryId = j;
231      i++;
232    }
233  }
234
235  sopDescriptionSEI->m_numPicsInSopMinus1 = i - 1;
236}
237
238Void SEIEncoder::initSEIBufferingPeriod(SEIBufferingPeriod *bufferingPeriodSEI, TComSlice *slice)
239{
240  assert (m_isInitialized);
241  assert (bufferingPeriodSEI != NULL);
242  assert (slice != NULL);
243
244  UInt uiInitialCpbRemovalDelay = (90000/2);                      // 0.5 sec
245  bufferingPeriodSEI->m_initialCpbRemovalDelay      [0][0]     = uiInitialCpbRemovalDelay;
246  bufferingPeriodSEI->m_initialCpbRemovalDelayOffset[0][0]     = uiInitialCpbRemovalDelay;
247  bufferingPeriodSEI->m_initialCpbRemovalDelay      [0][1]     = uiInitialCpbRemovalDelay;
248  bufferingPeriodSEI->m_initialCpbRemovalDelayOffset[0][1]     = uiInitialCpbRemovalDelay;
249
250  Double dTmp = (Double)slice->getSPS()->getVuiParameters()->getTimingInfo()->getNumUnitsInTick() / (Double)slice->getSPS()->getVuiParameters()->getTimingInfo()->getTimeScale();
251
252  UInt uiTmp = (UInt)( dTmp * 90000.0 );
253  uiInitialCpbRemovalDelay -= uiTmp;
254  uiInitialCpbRemovalDelay -= uiTmp / ( slice->getSPS()->getVuiParameters()->getHrdParameters()->getTickDivisorMinus2() + 2 );
255  bufferingPeriodSEI->m_initialAltCpbRemovalDelay      [0][0]  = uiInitialCpbRemovalDelay;
256  bufferingPeriodSEI->m_initialAltCpbRemovalDelayOffset[0][0]  = uiInitialCpbRemovalDelay;
257  bufferingPeriodSEI->m_initialAltCpbRemovalDelay      [0][1]  = uiInitialCpbRemovalDelay;
258  bufferingPeriodSEI->m_initialAltCpbRemovalDelayOffset[0][1]  = uiInitialCpbRemovalDelay;
259
260  bufferingPeriodSEI->m_rapCpbParamsPresentFlag = 0;
261  //for the concatenation, it can be set to one during splicing.
262  bufferingPeriodSEI->m_concatenationFlag = 0;
263  //since the temporal layer HRD is not ready, we assumed it is fixed
264  bufferingPeriodSEI->m_auCpbRemovalDelayDelta = 1;
265  bufferingPeriodSEI->m_cpbDelayOffset = 0;
266  bufferingPeriodSEI->m_dpbDelayOffset = 0;
267}
268
[1302]269//! initialize scalable nesting SEI message.
270//! Note: The SEI message structures input into this function will become part of the scalable nesting SEI and will be
271//!       automatically freed, when the nesting SEI is disposed.
[1273]272Void SEIEncoder::initSEIScalableNesting(SEIScalableNesting *scalableNestingSEI, SEIMessages &nestedSEIs)
273{
274  assert (m_isInitialized);
275  assert (scalableNestingSEI != NULL);
276
277  scalableNestingSEI->m_bitStreamSubsetFlag           = 1;      // If the nested SEI messages are picture buffering SEI messages, picture timing SEI messages or sub-picture timing SEI messages, bitstream_subset_flag shall be equal to 1
278  scalableNestingSEI->m_nestingOpFlag                 = 0;
279  scalableNestingSEI->m_nestingNumOpsMinus1           = 0;      //nesting_num_ops_minus1
280  scalableNestingSEI->m_allLayersFlag                 = 0;
281  scalableNestingSEI->m_nestingNoOpMaxTemporalIdPlus1 = 6 + 1;  //nesting_no_op_max_temporal_id_plus1
282  scalableNestingSEI->m_nestingNumLayersMinus1        = 1 - 1;  //nesting_num_layers_minus1
283  scalableNestingSEI->m_nestingLayerId[0]             = 0;
284
285  scalableNestingSEI->m_nestedSEIs.clear();
286  for (SEIMessages::iterator it=nestedSEIs.begin(); it!=nestedSEIs.end(); it++)
287  {
288    scalableNestingSEI->m_nestedSEIs.push_back((*it));
289  }
290}
291
292Void SEIEncoder::initSEIRecoveryPoint(SEIRecoveryPoint *recoveryPointSEI, TComSlice *slice)
293{
294  assert (m_isInitialized);
295  assert (recoveryPointSEI != NULL);
296  assert (slice != NULL);
297
298  recoveryPointSEI->m_recoveryPocCnt    = 0;
299#if SVC_EXTENSION
300  recoveryPointSEI->m_exactMatchingFlag = ( slice->getPocValueBeforeReset() == 0 ) ? (true) : (false);
301#else 
302  recoveryPointSEI->m_exactMatchingFlag = ( slice->getPOC() == 0 ) ? (true) : (false);
303#endif
304  recoveryPointSEI->m_brokenLinkFlag    = false;
305}
306
307//! calculate hashes for entire reconstructed picture
[1287]308Void SEIEncoder::initDecodedPictureHashSEI(SEIDecodedPictureHash *decodedPictureHashSEI, TComPic *pcPic, std::string &rHashString, const BitDepths &bitDepths)
[1273]309{
310  assert (m_isInitialized);
311  assert (decodedPictureHashSEI!=NULL);
312  assert (pcPic!=NULL);
313
[1459]314  decodedPictureHashSEI->method = m_pcCfg->getDecodedPictureHashSEIType();
315  switch (m_pcCfg->getDecodedPictureHashSEIType())
[1273]316  {
[1459]317    case HASHTYPE_MD5:
318      {
319        UInt numChar=calcMD5(*pcPic->getPicYuvRec(), decodedPictureHashSEI->m_pictureHash, bitDepths);
320        rHashString = hashToString(decodedPictureHashSEI->m_pictureHash, numChar);
321      }
322      break;
323    case HASHTYPE_CRC:
324      {
325        UInt numChar=calcCRC(*pcPic->getPicYuvRec(), decodedPictureHashSEI->m_pictureHash, bitDepths);
326        rHashString = hashToString(decodedPictureHashSEI->m_pictureHash, numChar);
327      }
328      break;
329    case HASHTYPE_CHECKSUM:
330    default:
331      {
332        UInt numChar=calcChecksum(*pcPic->getPicYuvRec(), decodedPictureHashSEI->m_pictureHash, bitDepths);
333        rHashString = hashToString(decodedPictureHashSEI->m_pictureHash, numChar);
334      }
335      break;
[1273]336  }
337}
338
339Void SEIEncoder::initTemporalLevel0IndexSEI(SEITemporalLevel0Index *temporalLevel0IndexSEI, TComSlice *slice)
340{
341  assert (m_isInitialized);
342  assert (temporalLevel0IndexSEI!=NULL);
343  assert (slice!=NULL);
344
345  if (slice->getRapPicFlag())
346  {
347    m_tl0Idx = 0;
348    m_rapIdx = (m_rapIdx + 1) & 0xFF;
349  }
350  else
351  {
352    m_tl0Idx = (m_tl0Idx + (slice->getTLayer() ? 0 : 1)) & 0xFF;
353  }
354  temporalLevel0IndexSEI->tl0Idx = m_tl0Idx;
355  temporalLevel0IndexSEI->rapIdx = m_rapIdx;
356}
357
358Void SEIEncoder::initSEITempMotionConstrainedTileSets (SEITempMotionConstrainedTileSets *sei, const TComPPS *pps)
359{
360  assert (m_isInitialized);
361  assert (sei!=NULL);
362  assert (pps!=NULL);
363
364  if(pps->getTilesEnabledFlag())
365  {
366    sei->m_mc_all_tiles_exact_sample_value_match_flag = false;
367    sei->m_each_tile_one_tile_set_flag                = false;
368    sei->m_limited_tile_set_display_flag              = false;
369    sei->setNumberOfTileSets((pps->getNumTileColumnsMinus1() + 1) * (pps->getNumTileRowsMinus1() + 1));
370
371    for(Int i=0; i < sei->getNumberOfTileSets(); i++)
372    {
373      sei->tileSetData(i).m_mcts_id = i;  //depends the application;
374      sei->tileSetData(i).setNumberOfTileRects(1);
375
376      for(Int j=0; j<sei->tileSetData(i).getNumberOfTileRects(); j++)
377      {
378        sei->tileSetData(i).topLeftTileIndex(j)     = i+j;
379        sei->tileSetData(i).bottomRightTileIndex(j) = i+j;
380      }
381
382      sei->tileSetData(i).m_exact_sample_value_match_flag    = false;
383      sei->tileSetData(i).m_mcts_tier_level_idc_present_flag = false;
384    }
385  }
386  else
387  {
388    assert(!"Tile is not enabled");
389  }
390}
391
392Void SEIEncoder::initSEIKneeFunctionInfo(SEIKneeFunctionInfo *seiKneeFunctionInfo)
393{
394  assert (m_isInitialized);
395  assert (seiKneeFunctionInfo!=NULL);
396
397  seiKneeFunctionInfo->m_kneeId = m_pcCfg->getKneeSEIId();
398  seiKneeFunctionInfo->m_kneeCancelFlag = m_pcCfg->getKneeSEICancelFlag();
399  if ( !seiKneeFunctionInfo->m_kneeCancelFlag )
400  {
401    seiKneeFunctionInfo->m_kneePersistenceFlag = m_pcCfg->getKneeSEIPersistenceFlag();
402    seiKneeFunctionInfo->m_kneeInputDrange = m_pcCfg->getKneeSEIInputDrange();
403    seiKneeFunctionInfo->m_kneeInputDispLuminance = m_pcCfg->getKneeSEIInputDispLuminance();
404    seiKneeFunctionInfo->m_kneeOutputDrange = m_pcCfg->getKneeSEIOutputDrange();
405    seiKneeFunctionInfo->m_kneeOutputDispLuminance = m_pcCfg->getKneeSEIOutputDispLuminance();
406
407    seiKneeFunctionInfo->m_kneeNumKneePointsMinus1 = m_pcCfg->getKneeSEINumKneePointsMinus1();
408    Int* piInputKneePoint  = m_pcCfg->getKneeSEIInputKneePoint();
409    Int* piOutputKneePoint = m_pcCfg->getKneeSEIOutputKneePoint();
410    if(piInputKneePoint&&piOutputKneePoint)
411    {
412      seiKneeFunctionInfo->m_kneeInputKneePoint.resize(seiKneeFunctionInfo->m_kneeNumKneePointsMinus1+1);
413      seiKneeFunctionInfo->m_kneeOutputKneePoint.resize(seiKneeFunctionInfo->m_kneeNumKneePointsMinus1+1);
414      for(Int i=0; i<=seiKneeFunctionInfo->m_kneeNumKneePointsMinus1; i++)
415      {
416        seiKneeFunctionInfo->m_kneeInputKneePoint[i] = piInputKneePoint[i];
417        seiKneeFunctionInfo->m_kneeOutputKneePoint[i] = piOutputKneePoint[i];
418      }
419    }
420  }
421}
422
[1460]423template <typename T>
424static Void readTokenValue(T            &returnedValue, /// value returned
425                           Bool         &failed,        /// used and updated
426                           std::istream &is,            /// stream to read token from
427                           const TChar  *pToken)        /// token string
428{
429  returnedValue=T();
430  if (failed)
431  {
432    return;
433  }
434
435  Int c;
436  // Ignore any whitespace
437  while ((c=is.get())!=EOF && isspace(c));
438  // test for comment mark
439  while (c=='#')
440  {
441    // Ignore to the end of the line
442    while ((c=is.get())!=EOF && (c!=10 && c!=13));
443    // Ignore any white space at the start of the next line
444    while ((c=is.get())!=EOF && isspace(c));
445  }
446  // test first character of token
447  failed=(c!=pToken[0]);
448  // test remaining characters of token
449  Int pos;
450  for(pos=1;!failed && pToken[pos]!=0 && is.get()==pToken[pos]; pos++);
451  failed|=(pToken[pos]!=0);
452  // Ignore any whitespace before the ':'
453  while (!failed && (c=is.get())!=EOF && isspace(c));
454  failed|=(c!=':');
455  // Now read the value associated with the token:
456  if (!failed)
457  {
458    is >> returnedValue;
459    failed=!is.good();
460    if (!failed)
461    {
462      c=is.get();
463      failed=(c!=EOF && !isspace(c));
464    }
465  }
466  if (failed)
467  {
468    std::cerr << "Unable to read token '" << pToken << "'\n";
469  }
470}
471
472template <typename T>
473static Void readTokenValueAndValidate(T            &returnedValue, /// value returned
474                                      Bool         &failed,        /// used and updated
475                                      std::istream &is,            /// stream to read token from
476                                      const TChar  *pToken,        /// token string
477                                      const T      &minInclusive,  /// minimum value allowed, inclusive
478                                      const T      &maxInclusive)  /// maximum value allowed, inclusive
479{
480  readTokenValue(returnedValue, failed, is, pToken);
481  if (!failed)
482  {
483    if (returnedValue<minInclusive || returnedValue>maxInclusive)
484    {
485      failed=true;
486      std::cerr << "Value for token " << pToken << " must be in the range " << minInclusive << " to " << maxInclusive << " (inclusive); value read: " << returnedValue << std::endl;
487    }
488  }
489}
490
491// Bool version does not have maximum and minimum values.
492static Void readTokenValueAndValidate(Bool         &returnedValue, /// value returned
493                                      Bool         &failed,        /// used and updated
494                                      std::istream &is,            /// stream to read token from
495                                      const TChar  *pToken)        /// token string
496{
497  readTokenValue(returnedValue, failed, is, pToken);
498}
499
500Bool SEIEncoder::initSEIColourRemappingInfo(SEIColourRemappingInfo* seiColourRemappingInfo, Int currPOC) // returns true on success, false on failure.
501{
502  assert (m_isInitialized);
503  assert (seiColourRemappingInfo!=NULL);
504
505  // reading external Colour Remapping Information SEI message parameters from file
506  if( !m_pcCfg->getColourRemapInfoSEIFileRoot().empty())
507  {
508    Bool failed=false;
509
510    // building the CRI file name with poc num in prefix "_poc.txt"
511    std::string colourRemapSEIFileWithPoc(m_pcCfg->getColourRemapInfoSEIFileRoot());
512    {
513      std::stringstream suffix;
514      suffix << "_" << currPOC << ".txt";
515      colourRemapSEIFileWithPoc+=suffix.str();
516    }
517
518    std::ifstream fic(colourRemapSEIFileWithPoc.c_str());
519    if (!fic.good() || !fic.is_open())
520    {
521      std::cerr <<  "No Colour Remapping Information SEI parameters file " << colourRemapSEIFileWithPoc << " for POC " << currPOC << std::endl;
522      return false;
523    }
524
525    // TODO: identify and remove duplication with decoder parsing through abstraction.
526
527    readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapId,         failed, fic, "colour_remap_id",        UInt(0), UInt(0x7fffffff) );
528    readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapCancelFlag, failed, fic, "colour_remap_cancel_flag" );
529    if( !seiColourRemappingInfo->m_colourRemapCancelFlag )
530    {
531      readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapPersistenceFlag,            failed, fic, "colour_remap_persistence_flag" );
532      readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapVideoSignalInfoPresentFlag, failed, fic, "colour_remap_video_signal_info_present_flag");
533      if( seiColourRemappingInfo->m_colourRemapVideoSignalInfoPresentFlag )
534      {
535        readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapFullRangeFlag,      failed, fic, "colour_remap_full_range_flag" );
536        readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapPrimaries,          failed, fic, "colour_remap_primaries",           Int(0), Int(255) );
537        readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapTransferFunction,   failed, fic, "colour_remap_transfer_function",   Int(0), Int(255) );
538        readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapMatrixCoefficients, failed, fic, "colour_remap_matrix_coefficients", Int(0), Int(255) );
539      }
540      readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapInputBitDepth, failed, fic, "colour_remap_input_bit_depth",            Int(8), Int(16) );
541      readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapBitDepth,      failed, fic, "colour_remap_bit_depth",                  Int(8), Int(16) );
542
[1479]543      const Int maximumInputValue    = (1 << (((seiColourRemappingInfo->m_colourRemapInputBitDepth + 7) >> 3) << 3)) - 1;
544      const Int maximumRemappedValue = (1 << (((seiColourRemappingInfo->m_colourRemapBitDepth      + 7) >> 3) << 3)) - 1;
[1460]545
546      for( Int c=0 ; c<3 ; c++ )
547      {
548        readTokenValueAndValidate(seiColourRemappingInfo->m_preLutNumValMinus1[c],         failed, fic, "pre_lut_num_val_minus1[c]",        Int(0), Int(32) );
549        if( seiColourRemappingInfo->m_preLutNumValMinus1[c]>0 )
550        {
551          seiColourRemappingInfo->m_preLut[c].resize(seiColourRemappingInfo->m_preLutNumValMinus1[c]+1);
552          for( Int i=0 ; i<=seiColourRemappingInfo->m_preLutNumValMinus1[c] ; i++ )
553          {
554            readTokenValueAndValidate(seiColourRemappingInfo->m_preLut[c][i].codedValue,   failed, fic, "pre_lut_coded_value[c][i]",  Int(0), maximumInputValue    );
555            readTokenValueAndValidate(seiColourRemappingInfo->m_preLut[c][i].targetValue,  failed, fic, "pre_lut_target_value[c][i]", Int(0), maximumRemappedValue );
556          }
557        }
558      }
559      readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapMatrixPresentFlag, failed, fic, "colour_remap_matrix_present_flag" );
560      if( seiColourRemappingInfo->m_colourRemapMatrixPresentFlag )
561      {
562        readTokenValueAndValidate(seiColourRemappingInfo->m_log2MatrixDenom, failed, fic, "log2_matrix_denom", Int(0), Int(15) );
563        for( Int c=0 ; c<3 ; c++ )
564        {
565          for( Int i=0 ; i<3 ; i++ )
566          {
567            readTokenValueAndValidate(seiColourRemappingInfo->m_colourRemapCoeffs[c][i], failed, fic, "colour_remap_coeffs[c][i]", -32768, 32767 );
568          }
569        }
570      }
571      for( Int c=0 ; c<3 ; c++ )
572      {
573        readTokenValueAndValidate(seiColourRemappingInfo->m_postLutNumValMinus1[c], failed, fic, "post_lut_num_val_minus1[c]", Int(0), Int(32) );
574        if( seiColourRemappingInfo->m_postLutNumValMinus1[c]>0 )
575        {
576          seiColourRemappingInfo->m_postLut[c].resize(seiColourRemappingInfo->m_postLutNumValMinus1[c]+1);
577          for( Int i=0 ; i<=seiColourRemappingInfo->m_postLutNumValMinus1[c] ; i++ )
578          {
579            readTokenValueAndValidate(seiColourRemappingInfo->m_postLut[c][i].codedValue,  failed, fic, "post_lut_coded_value[c][i]",  Int(0), maximumRemappedValue );
580            readTokenValueAndValidate(seiColourRemappingInfo->m_postLut[c][i].targetValue, failed, fic, "post_lut_target_value[c][i]", Int(0), maximumRemappedValue );
581          }
582        }
583      }
584    }
585
586    if( failed )
587    {
588      std::cerr << "Error while reading Colour Remapping Information SEI parameters file '" << colourRemapSEIFileWithPoc << "'" << std::endl;
589      exit(EXIT_FAILURE);
590    }
591  }
592  return true;
593}
594
[1434]595Void SEIEncoder::initSEIChromaResamplingFilterHint(SEIChromaResamplingFilterHint *seiChromaResamplingFilterHint, Int iHorFilterIndex, Int iVerFilterIndex)
[1273]596{
597  assert (m_isInitialized);
[1434]598  assert (seiChromaResamplingFilterHint!=NULL);
[1273]599
[1434]600  seiChromaResamplingFilterHint->m_verChromaFilterIdc = iVerFilterIndex;
601  seiChromaResamplingFilterHint->m_horChromaFilterIdc = iHorFilterIndex;
602  seiChromaResamplingFilterHint->m_verFilteringFieldProcessingFlag = 1;
603  seiChromaResamplingFilterHint->m_targetFormatIdc = 3;
604  seiChromaResamplingFilterHint->m_perfectReconstructionFlag = false;
605
606  // this creates some example filter values, if explicit filter definition is selected
607  if (seiChromaResamplingFilterHint->m_verChromaFilterIdc == 1)
[1273]608  {
[1434]609    const Int numVerticalFilters = 3;
610    const Int verTapLengthMinus1[] = {5,3,3};
611
612    seiChromaResamplingFilterHint->m_verFilterCoeff.resize(numVerticalFilters);
613    for(Int i = 0; i < numVerticalFilters; i ++)
[1273]614    {
[1434]615      seiChromaResamplingFilterHint->m_verFilterCoeff[i].resize(verTapLengthMinus1[i]+1);
[1273]616    }
[1434]617    // Note: C++11 -> seiChromaResamplingFilterHint->m_verFilterCoeff[0] = {-3,13,31,23,3,-3};
618    seiChromaResamplingFilterHint->m_verFilterCoeff[0][0] = -3;
619    seiChromaResamplingFilterHint->m_verFilterCoeff[0][1] = 13;
620    seiChromaResamplingFilterHint->m_verFilterCoeff[0][2] = 31;
621    seiChromaResamplingFilterHint->m_verFilterCoeff[0][3] = 23;
622    seiChromaResamplingFilterHint->m_verFilterCoeff[0][4] = 3;
623    seiChromaResamplingFilterHint->m_verFilterCoeff[0][5] = -3;
624
625    seiChromaResamplingFilterHint->m_verFilterCoeff[1][0] = -1;
626    seiChromaResamplingFilterHint->m_verFilterCoeff[1][1] = 25;
627    seiChromaResamplingFilterHint->m_verFilterCoeff[1][2] = 247;
628    seiChromaResamplingFilterHint->m_verFilterCoeff[1][3] = -15;
629
630    seiChromaResamplingFilterHint->m_verFilterCoeff[2][0] = -20;
631    seiChromaResamplingFilterHint->m_verFilterCoeff[2][1] = 186;
632    seiChromaResamplingFilterHint->m_verFilterCoeff[2][2] = 100;
633    seiChromaResamplingFilterHint->m_verFilterCoeff[2][3] = -10;
[1273]634  }
635  else
636  {
[1434]637    seiChromaResamplingFilterHint->m_verFilterCoeff.resize(0);
[1273]638  }
[1434]639
640  if (seiChromaResamplingFilterHint->m_horChromaFilterIdc == 1)
[1273]641  {
[1434]642    Int const numHorizontalFilters = 1;
643    const Int horTapLengthMinus1[] = {3};
644
645    seiChromaResamplingFilterHint->m_horFilterCoeff.resize(numHorizontalFilters);
646    for(Int i = 0; i < numHorizontalFilters; i ++)
[1273]647    {
[1434]648      seiChromaResamplingFilterHint->m_horFilterCoeff[i].resize(horTapLengthMinus1[i]+1);
[1273]649    }
[1434]650    seiChromaResamplingFilterHint->m_horFilterCoeff[0][0] = 1;
651    seiChromaResamplingFilterHint->m_horFilterCoeff[0][1] = 6;
652    seiChromaResamplingFilterHint->m_horFilterCoeff[0][2] = 1;
[1273]653  }
654  else
655  {
[1434]656    seiChromaResamplingFilterHint->m_horFilterCoeff.resize(0);
[1273]657  }
658}
659
660Void SEIEncoder::initSEITimeCode(SEITimeCode *seiTimeCode)
661{
662  assert (m_isInitialized);
663  assert (seiTimeCode!=NULL);
664  //  Set data as per command line options
665  seiTimeCode->numClockTs = m_pcCfg->getNumberOfTimesets();
666  for(Int i = 0; i < seiTimeCode->numClockTs; i++)
667  {
668    seiTimeCode->timeSetArray[i] = m_pcCfg->getTimeSet(i);
669  }
670}
671
[1538]672#if U0033_ALTERNATIVE_TRANSFER_CHARACTERISTICS_SEI
673Void SEIEncoder::initSEIAlternativeTransferCharacteristics(SEIAlternativeTransferCharacteristics *seiAltTransCharacteristics)
674{
675  assert (m_isInitialized);
676  assert (seiAltTransCharacteristics!=NULL);
677  //  Set SEI message parameters read from command line options
678  seiAltTransCharacteristics->m_preferredTransferCharacteristics = m_pcCfg->getSEIPreferredTransferCharacteristics();
679}
680#endif
681
[1572]682Void SEIEncoder::initSEIGreenMetadataInfo(SEIGreenMetadataInfo *seiGreenMetadataInfo, UInt u)
683{
684    assert (m_isInitialized);
685    assert (seiGreenMetadataInfo!=NULL);
686
687    seiGreenMetadataInfo->m_greenMetadataType = m_pcCfg->getSEIGreenMetadataType();
688    seiGreenMetadataInfo->m_xsdMetricType = m_pcCfg->getSEIXSDMetricType();
689    seiGreenMetadataInfo->m_xsdMetricValue = u;
690}
691
[1287]692#if LAYERS_NOT_PRESENT_SEI
693Void SEIEncoder::initSEILayersNotPresent(SEILayersNotPresent *seiLayersNotPresent)
694{
695  UInt i = 0;
696  seiLayersNotPresent->m_activeVpsId = m_pcCfg->getVPS()->getVPSId(); 
697  seiLayersNotPresent->m_vpsMaxLayers = m_pcCfg->getVPS()->getMaxLayers();
698  for ( ; i < seiLayersNotPresent->m_vpsMaxLayers; i++)
699  {
700    seiLayersNotPresent->m_layerNotPresentFlag[i] = true; 
701  }
702  for ( ; i < MAX_LAYERS; i++)
703  {
704    seiLayersNotPresent->m_layerNotPresentFlag[i] = false; 
705  }
706}
[1273]707#endif
708
[1287]709#if N0383_IL_CONSTRAINED_TILE_SETS_SEI
710Void SEIEncoder::initSEIInterLayerConstrainedTileSets(SEIInterLayerConstrainedTileSets *seiInterLayerConstrainedTileSets)
711{
712  seiInterLayerConstrainedTileSets->m_ilAllTilesExactSampleValueMatchFlag = false;
713  seiInterLayerConstrainedTileSets->m_ilOneTilePerTileSetFlag = false;
714  if (!seiInterLayerConstrainedTileSets->m_ilOneTilePerTileSetFlag)
715  {
716    seiInterLayerConstrainedTileSets->m_ilNumSetsInMessageMinus1 = m_pcCfg->getIlNumSetsInMessage() - 1;
717    if (seiInterLayerConstrainedTileSets->m_ilNumSetsInMessageMinus1)
718    {
719      seiInterLayerConstrainedTileSets->m_skippedTileSetPresentFlag = m_pcCfg->getSkippedTileSetPresentFlag();
720    }
721    else
722    {
723      seiInterLayerConstrainedTileSets->m_skippedTileSetPresentFlag = false;
724    }
725    seiInterLayerConstrainedTileSets->m_ilNumSetsInMessageMinus1 += seiInterLayerConstrainedTileSets->m_skippedTileSetPresentFlag ? 1 : 0;
726    for (UInt i = 0; i < m_pcCfg->getIlNumSetsInMessage(); i++)
727    {
728      seiInterLayerConstrainedTileSets->m_ilctsId[i] = i;
729      seiInterLayerConstrainedTileSets->m_ilNumTileRectsInSetMinus1[i] = 0;
730      for( UInt j = 0; j <= seiInterLayerConstrainedTileSets->m_ilNumTileRectsInSetMinus1[i]; j++)
731      {
732        seiInterLayerConstrainedTileSets->m_ilTopLeftTileIndex[i][j]     = m_pcCfg->getTopLeftTileIndex(i);
733        seiInterLayerConstrainedTileSets->m_ilBottomRightTileIndex[i][j] = m_pcCfg->getBottomRightTileIndex(i);
734      }
735      seiInterLayerConstrainedTileSets->m_ilcIdc[i] = m_pcCfg->getIlcIdc(i);
736      if (seiInterLayerConstrainedTileSets->m_ilAllTilesExactSampleValueMatchFlag)
737      {
738        seiInterLayerConstrainedTileSets->m_ilExactSampleValueMatchFlag[i] = false;
739      }
740    }
741  }
742}
[1273]743#endif
744
[1287]745#if P0123_ALPHA_CHANNEL_SEI
746Void SEIEncoder::initSEIAlphaChannelInfo(SEIAlphaChannelInfo *seiAlphaChannelInfo)
747{
748  seiAlphaChannelInfo->m_alphaChannelCancelFlag = m_pcCfg->getAlphaCancelFlag();
749  if(!seiAlphaChannelInfo->m_alphaChannelCancelFlag)
750  {
751    seiAlphaChannelInfo->m_alphaChannelUseIdc = m_pcCfg->getAlphaUseIdc();
752    seiAlphaChannelInfo->m_alphaChannelBitDepthMinus8 = m_pcCfg->getAlphaBitDepthMinus8();
753    seiAlphaChannelInfo->m_alphaTransparentValue = m_pcCfg->getAlphaTransparentValue();
754    seiAlphaChannelInfo->m_alphaOpaqueValue = m_pcCfg->getAlphaOpaqueValue();
755    seiAlphaChannelInfo->m_alphaChannelIncrFlag = m_pcCfg->getAlphaIncrementFlag();
756    seiAlphaChannelInfo->m_alphaChannelClipFlag = m_pcCfg->getAlphaClipFlag();
757    seiAlphaChannelInfo->m_alphaChannelClipTypeFlag = m_pcCfg->getAlphaClipTypeFlag();
758  }
759}
[1273]760#endif
761
[1287]762#if Q0096_OVERLAY_SEI
763Void SEIEncoder::initSEIOverlayInfo(SEIOverlayInfo *seiOverlayInfo)
764{ 
765  seiOverlayInfo->m_overlayInfoCancelFlag = m_pcCfg->getOverlaySEICancelFlag();   
766  if ( !seiOverlayInfo->m_overlayInfoCancelFlag )
767  {
768    seiOverlayInfo->m_overlayContentAuxIdMinus128          = m_pcCfg->getOverlaySEIContentAuxIdMinus128(); 
769    seiOverlayInfo->m_overlayLabelAuxIdMinus128            = m_pcCfg->getOverlaySEILabelAuxIdMinus128(); 
770    seiOverlayInfo->m_overlayAlphaAuxIdMinus128            = m_pcCfg->getOverlaySEIAlphaAuxIdMinus128(); 
771    seiOverlayInfo->m_overlayElementLabelValueLengthMinus8 = m_pcCfg->getOverlaySEIElementLabelValueLengthMinus8(); 
772    seiOverlayInfo->m_numOverlaysMinus1                    = m_pcCfg->getOverlaySEINumOverlaysMinus1();     
773    seiOverlayInfo->m_overlayIdx                           = m_pcCfg->getOverlaySEIIdx();           
774    seiOverlayInfo->m_languageOverlayPresentFlag           = m_pcCfg->getOverlaySEILanguagePresentFlag();
775    seiOverlayInfo->m_overlayContentLayerId                = m_pcCfg->getOverlaySEIContentLayerId();   
776    seiOverlayInfo->m_overlayLabelPresentFlag              = m_pcCfg->getOverlaySEILabelPresentFlag();   
777    seiOverlayInfo->m_overlayLabelLayerId                  = m_pcCfg->getOverlaySEILabelLayerId();   
778    seiOverlayInfo->m_overlayAlphaPresentFlag              = m_pcCfg->getOverlaySEIAlphaPresentFlag();   
779    seiOverlayInfo->m_overlayAlphaLayerId                  = m_pcCfg->getOverlaySEIAlphaLayerId();   
780    seiOverlayInfo->m_numOverlayElementsMinus1             = m_pcCfg->getOverlaySEINumElementsMinus1();
781    seiOverlayInfo->m_overlayElementLabelMin               = m_pcCfg->getOverlaySEIElementLabelMin();
782    seiOverlayInfo->m_overlayElementLabelMax               = m_pcCfg->getOverlaySEIElementLabelMax();   
783    seiOverlayInfo->m_overlayLanguage.resize               ( seiOverlayInfo->m_numOverlaysMinus1+1, NULL );
784    seiOverlayInfo->m_overlayLanguageLength.resize         ( seiOverlayInfo->m_numOverlaysMinus1+1 );
785    seiOverlayInfo->m_overlayName.resize                   ( seiOverlayInfo->m_numOverlaysMinus1+1, NULL );
786    seiOverlayInfo->m_overlayNameLength.resize             ( seiOverlayInfo->m_numOverlaysMinus1+1 );
787    seiOverlayInfo->m_overlayElementName.resize            ( seiOverlayInfo->m_numOverlaysMinus1+1 );
788    seiOverlayInfo->m_overlayElementNameLength.resize      ( seiOverlayInfo->m_numOverlaysMinus1+1 ); 
789
790    Int i,j;
791    string strTmp;
792    Int nBytes;
793    assert( m_pcCfg->getOverlaySEILanguage().size()    == seiOverlayInfo->m_numOverlaysMinus1+1 );
794    assert( m_pcCfg->getOverlaySEIName().size()        == seiOverlayInfo->m_numOverlaysMinus1+1 );
795    assert( m_pcCfg->getOverlaySEIElementName().size() == seiOverlayInfo->m_numOverlaysMinus1+1 );
796   
797    for ( i=0 ; i<=seiOverlayInfo->m_numOverlaysMinus1; i++ )
798    {     
799      //language tag
800      if ( seiOverlayInfo->m_languageOverlayPresentFlag[i] )
801      {               
802        strTmp = m_pcCfg->getOverlaySEILanguage()[i];
803        nBytes = (Int)m_pcCfg->getOverlaySEILanguage()[i].size();       
804        assert( nBytes>0 );
805        seiOverlayInfo->m_overlayLanguage[i] = new UChar[nBytes];
806        memcpy(seiOverlayInfo->m_overlayLanguage[i], strTmp.c_str(), nBytes);       
807        seiOverlayInfo->m_overlayLanguageLength[i] = nBytes;       
808      }
809
810      //overlay name
811      strTmp = m_pcCfg->getOverlaySEIName()[i];
812      nBytes = (Int)m_pcCfg->getOverlaySEIName()[i].size();       
813      assert( nBytes>0 );
814      seiOverlayInfo->m_overlayName[i] = new UChar[nBytes];     
815      memcpy(seiOverlayInfo->m_overlayName[i], strTmp.c_str(), nBytes);       
816      seiOverlayInfo->m_overlayNameLength[i] = nBytes;
817
818      //overlay element names
819      if ( seiOverlayInfo->m_overlayLabelPresentFlag[i] )
820      {       
821        seiOverlayInfo->m_overlayElementName[i].resize( seiOverlayInfo->m_numOverlayElementsMinus1[i]+1, NULL );
822        seiOverlayInfo->m_overlayElementNameLength[i].resize( seiOverlayInfo->m_numOverlayElementsMinus1[i]+1 );
823        assert( m_pcCfg->getOverlaySEIElementName()[i].size() == seiOverlayInfo->m_numOverlayElementsMinus1[i]+1 );       
824        for ( j=0 ; j<=seiOverlayInfo->m_numOverlayElementsMinus1[i] ; j++)
825        {
826          strTmp = m_pcCfg->getOverlaySEIElementName()[i][j];
827          nBytes = (Int)m_pcCfg->getOverlaySEIElementName()[i][j].size();       
828          assert( nBytes>0 );
829          seiOverlayInfo->m_overlayElementName[i][j] = new UChar[nBytes];
830          memcpy(seiOverlayInfo->m_overlayElementName[i][j], strTmp.c_str(), nBytes);       
831          seiOverlayInfo->m_overlayElementNameLength[i][j] = nBytes;
832        }
833      }
834    }
835  seiOverlayInfo->m_overlayInfoPersistenceFlag = true;
836  }
837}
838#endif
839
840#if O0164_MULTI_LAYER_HRD
841Void SEIEncoder::initBspNestingSEI(SEIScalableNesting *seiScalableNesting, const TComVPS *vps, const TComSPS *sps, Int olsIdx, Int partitioningSchemeIdx, Int bspIdx)
842{
843  SEIBspInitialArrivalTime *seiBspInitialArrivalTime = new SEIBspInitialArrivalTime();
844  SEIBspNesting *seiBspNesting = new SEIBspNesting();
845  SEIBufferingPeriod *seiBufferingPeriod = new SEIBufferingPeriod();
846
847  // Scalable nesting SEI
848
849  seiScalableNesting->m_bitStreamSubsetFlag           = 1;      // If the nested SEI messages are picture buffereing SEI mesages, picure timing SEI messages or sub-picture timing SEI messages, bitstream_subset_flag shall be equal to 1
850  seiScalableNesting->m_nestingOpFlag                 = 1;
851  seiScalableNesting->m_defaultOpFlag                 = 0;
852  seiScalableNesting->m_nestingNumOpsMinus1           = 0;      //nesting_num_ops_minus1
853  seiScalableNesting->m_nestingOpIdx[0]               = vps->getOutputLayerSetIdx(olsIdx);
854  seiScalableNesting->m_nestingMaxTemporalIdPlus1[0]  = 6 + 1;
855  seiScalableNesting->m_allLayersFlag                 = 0;
856  seiScalableNesting->m_nestingNoOpMaxTemporalIdPlus1 = 6 + 1;  //nesting_no_op_max_temporal_id_plus1
857  seiScalableNesting->m_nestingNumLayersMinus1        = 1 - 1;  //nesting_num_layers_minus1
858  seiScalableNesting->m_nestingLayerId[0]             = 0;
859
860  // Bitstream partition nesting SEI
861  seiBspNesting->m_bspIdx = 0;
862
863  // Buffering period SEI
864
865  UInt uiInitialCpbRemovalDelay = (90000/2);                      // 0.5 sec
866  seiBufferingPeriod->m_initialCpbRemovalDelay      [0][0]     = uiInitialCpbRemovalDelay;
867  seiBufferingPeriod->m_initialCpbRemovalDelayOffset[0][0]     = uiInitialCpbRemovalDelay;
868  seiBufferingPeriod->m_initialCpbRemovalDelay      [0][1]     = uiInitialCpbRemovalDelay;
869  seiBufferingPeriod->m_initialCpbRemovalDelayOffset[0][1]     = uiInitialCpbRemovalDelay;
870
871  Double dTmp = (Double)sps->getVuiParameters()->getTimingInfo()->getNumUnitsInTick() / (Double)sps->getVuiParameters()->getTimingInfo()->getTimeScale();
872
873  UInt uiTmp = (UInt)( dTmp * 90000.0 ); 
874  uiInitialCpbRemovalDelay -= uiTmp;
875  uiInitialCpbRemovalDelay -= uiTmp / ( sps->getVuiParameters()->getHrdParameters()->getTickDivisorMinus2() + 2 );
876  seiBufferingPeriod->m_initialAltCpbRemovalDelay      [0][0]  = uiInitialCpbRemovalDelay;
877  seiBufferingPeriod->m_initialAltCpbRemovalDelayOffset[0][0]  = uiInitialCpbRemovalDelay;
878  seiBufferingPeriod->m_initialAltCpbRemovalDelay      [0][1]  = uiInitialCpbRemovalDelay;
879  seiBufferingPeriod->m_initialAltCpbRemovalDelayOffset[0][1]  = uiInitialCpbRemovalDelay;
880
881  seiBufferingPeriod->m_rapCpbParamsPresentFlag              = 0;
882  //for the concatenation, it can be set to one during splicing.
883  seiBufferingPeriod->m_concatenationFlag = 0;
884  //since the temporal layer HRD is not ready, we assumed it is fixed
885  seiBufferingPeriod->m_auCpbRemovalDelayDelta = 1;
886  seiBufferingPeriod->m_cpbDelayOffset = 0;
887  seiBufferingPeriod->m_dpbDelayOffset = 0;
888
889  // Intial arrival time SEI message
890
891  seiBspInitialArrivalTime->m_nalInitialArrivalDelay[0] = 0;
892  seiBspInitialArrivalTime->m_vclInitialArrivalDelay[0] = 0;
893
894
895  seiBspNesting->m_nestedSEIs.push_back(seiBufferingPeriod);
896  seiBspNesting->m_nestedSEIs.push_back(seiBspInitialArrivalTime);
897  seiBspNesting->m_bspIdx = bspIdx;
898  seiBspNesting->m_seiOlsIdx = olsIdx;
899  seiBspNesting->m_seiPartitioningSchemeIdx = partitioningSchemeIdx;
900  seiScalableNesting->m_nestedSEIs.push_back(seiBspNesting); // BSP nesting SEI is contained in scalable nesting SEI
901}
902#endif
903
[1273]904//! \}
Note: See TracBrowser for help on using the repository browser.