source: SHVCSoftware/branches/SHM-dev/source/Lib/TLibEncoder/TEncTop.cpp @ 1503

Last change on this file since 1503 was 1503, checked in by seregin, 8 years ago

remove VPS member from picture

  • Property svn:eol-style set to native
File size: 59.7 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license.
5 *
6 * Copyright (c) 2010-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/** \file     TEncTop.cpp
35    \brief    encoder class
36*/
37
38#include "TLibCommon/CommonDef.h"
39#include "TEncTop.h"
40#include "TEncPic.h"
41#include "TLibCommon/TComChromaFormat.h"
42#if FAST_BIT_EST
43#include "TLibCommon/ContextModel.h"
44#endif
45
46//! \ingroup TLibEncoder
47//! \{
48#if SVC_EXTENSION 
49Int TEncTop::m_iSPSIdCnt = 0;
50Int TEncTop::m_iPPSIdCnt = 0;
51TComVPS TEncCfg::m_cVPS;
52#endif
53
54// ====================================================================================================================
55// Constructor / destructor / create / destroy
56// ====================================================================================================================
57
58TEncTop::TEncTop()
59{
60  m_iPOCLast          = -1;
61  m_iNumPicRcvd       =  0;
62  m_uiNumAllPicCoded  =  0;
63  m_pppcRDSbacCoder   =  NULL;
64  m_pppcBinCoderCABAC =  NULL;
65  m_cRDGoOnSbacCoder.init( &m_cRDGoOnBinCoderCABAC );
66#if ENC_DEC_TRACE
67  if (g_hTrace == NULL)
68  {
69    g_hTrace = fopen( "TraceEnc.txt", "wb" );
70  }
71  g_bJustDoIt = g_bEncDecTraceDisable;
72  g_nSymbolCounter = 0;
73#endif
74
75  m_iMaxRefPicNum     = 0;
76
77#if FAST_BIT_EST
78  ContextModel::buildNextStateTable();
79#endif
80
81#if SVC_EXTENSION
82  memset( m_cIlpPic, 0, sizeof(m_cIlpPic) );
83  m_bMFMEnabledFlag = false;
84  m_numRefLayerLocationOffsets = 0;
85  m_pocAdjustmentValue     = 0;
86#if NO_CLRAS_OUTPUT_FLAG
87  m_noClrasOutputFlag          = false;
88  m_layerInitializedFlag       = false;
89  m_firstPicInLayerDecodedFlag = false;
90  m_noOutputOfPriorPicsFlags   = false;
91#endif
92  m_pocDecrementedInDPBFlag    = false;
93#endif //SVC_EXTENSION
94}
95
96TEncTop::~TEncTop()
97{
98#if ENC_DEC_TRACE
99  if (g_hTrace != stdout)
100  {
101    fclose( g_hTrace );
102  }
103#endif
104}
105
106Void TEncTop::create ()
107{
108#if !SVC_EXTENSION
109  // initialize global variables
110  initROM();
111#endif
112
113  // create processing unit classes
114#if SVC_EXTENSION
115  m_cGOPEncoder.        create( m_layerId );
116#else
117  m_cGOPEncoder.        create( );
118#endif
119  m_cSliceEncoder.      create( getSourceWidth(), getSourceHeight(), m_chromaFormatIDC, m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth );
120  m_cCuEncoder.         create( m_maxTotalCUDepth, m_maxCUWidth, m_maxCUHeight, m_chromaFormatIDC );
121  if (m_bUseSAO)
122  {
123    m_cEncSAO.create( getSourceWidth(), getSourceHeight(), m_chromaFormatIDC, m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, m_log2SaoOffsetScale[CHANNEL_TYPE_LUMA], m_log2SaoOffsetScale[CHANNEL_TYPE_CHROMA] );
124    m_cEncSAO.createEncData(getSaoCtuBoundary());
125  }
126#if ADAPTIVE_QP_SELECTION
127  if (m_bUseAdaptQpSelect)
128  {
129    m_cTrQuant.initSliceQpDelta();
130  }
131#endif
132
133  m_cLoopFilter.create( m_maxTotalCUDepth );
134
135  if ( m_RCEnableRateControl )
136  {
137    m_cRateCtrl.init( m_framesToBeEncoded, m_RCTargetBitrate, m_iFrameRate, m_iGOPSize, m_iSourceWidth, m_iSourceHeight,
138        m_maxCUWidth, m_maxCUHeight, m_RCKeepHierarchicalBit, m_RCUseLCUSeparateModel, m_GOPList );
139  }
140
141  m_pppcRDSbacCoder = new TEncSbac** [m_maxTotalCUDepth+1];
142#if FAST_BIT_EST
143  m_pppcBinCoderCABAC = new TEncBinCABACCounter** [m_maxTotalCUDepth+1];
144#else
145  m_pppcBinCoderCABAC = new TEncBinCABAC** [m_maxTotalCUDepth+1];
146#endif
147
148  for ( Int iDepth = 0; iDepth < m_maxTotalCUDepth+1; iDepth++ )
149  {
150    m_pppcRDSbacCoder[iDepth] = new TEncSbac* [CI_NUM];
151#if FAST_BIT_EST
152    m_pppcBinCoderCABAC[iDepth] = new TEncBinCABACCounter* [CI_NUM];
153#else
154    m_pppcBinCoderCABAC[iDepth] = new TEncBinCABAC* [CI_NUM];
155#endif
156
157    for (Int iCIIdx = 0; iCIIdx < CI_NUM; iCIIdx ++ )
158    {
159      m_pppcRDSbacCoder[iDepth][iCIIdx] = new TEncSbac;
160#if FAST_BIT_EST
161      m_pppcBinCoderCABAC [iDepth][iCIIdx] = new TEncBinCABACCounter;
162#else
163      m_pppcBinCoderCABAC [iDepth][iCIIdx] = new TEncBinCABAC;
164#endif
165      m_pppcRDSbacCoder   [iDepth][iCIIdx]->init( m_pppcBinCoderCABAC [iDepth][iCIIdx] );
166    }
167  }
168
169#if LAYER_CTB
170  memcpy(g_auiLayerZscanToRaster[m_layerId], g_auiZscanToRaster, sizeof( g_auiZscanToRaster ) );
171  memcpy(g_auiLayerRasterToZscan[m_layerId], g_auiRasterToZscan, sizeof( g_auiRasterToZscan ) );
172  memcpy(g_auiLayerRasterToPelX[m_layerId],  g_auiRasterToPelX,  sizeof( g_auiRasterToPelX ) );
173  memcpy(g_auiLayerRasterToPelY[m_layerId],  g_auiRasterToPelY,  sizeof( g_auiRasterToPelY ) );
174#endif
175}
176
177Void TEncTop::destroy ()
178{
179  // destroy processing unit classes
180  m_cGOPEncoder.        destroy();
181  m_cSliceEncoder.      destroy();
182  m_cCuEncoder.         destroy();
183  m_cEncSAO.            destroyEncData();
184  m_cEncSAO.            destroy();
185  m_cLoopFilter.        destroy();
186  m_cRateCtrl.          destroy();
187  m_cSearch.            destroy();
188  Int iDepth;
189  for ( iDepth = 0; iDepth < m_maxTotalCUDepth+1; iDepth++ )
190  {
191    for (Int iCIIdx = 0; iCIIdx < CI_NUM; iCIIdx ++ )
192    {
193      delete m_pppcRDSbacCoder[iDepth][iCIIdx];
194      delete m_pppcBinCoderCABAC[iDepth][iCIIdx];
195    }
196  }
197
198  for ( iDepth = 0; iDepth < m_maxTotalCUDepth+1; iDepth++ )
199  {
200    delete [] m_pppcRDSbacCoder[iDepth];
201    delete [] m_pppcBinCoderCABAC[iDepth];
202  }
203
204  delete [] m_pppcRDSbacCoder;
205  delete [] m_pppcBinCoderCABAC;
206
207#if SVC_EXTENSION
208  for(Int i=0; i<MAX_NUM_REF; i++)
209  {
210    if(m_cIlpPic[i])
211    {
212      m_cIlpPic[i]->destroy();
213      delete m_cIlpPic[i];
214      m_cIlpPic[i] = NULL;
215    }
216  }
217#else
218  // destroy ROM
219  destroyROM();
220#endif
221  return;
222}
223
224Void TEncTop::init(Bool isFieldCoding)
225{
226  // initialize SPS
227  xInitSPS();
228  xInitVPS();
229
230#if U0132_TARGET_BITS_SATURATION
231  if (m_RCCpbSaturationEnabled)
232  {
233    m_cRateCtrl.initHrdParam(m_cSPS.getVuiParameters()->getHrdParameters(), m_iFrameRate, m_RCInitialCpbFullness);
234  }
235#endif
236  m_cRdCost.setCostMode(m_costMode);
237
238  // initialize PPS
239  xInitPPS();
240  xInitRPS(isFieldCoding);
241
242  xInitPPSforTiles();
243
244  // initialize processing unit classes
245  m_cGOPEncoder.  init( this );
246  m_cSliceEncoder.init( this );
247  m_cCuEncoder.   init( this );
248
249  // initialize transform & quantization class
250  m_pcCavlcCoder = getCavlcCoder();
251
252  m_cTrQuant.init( 1 << m_uiQuadtreeTULog2MaxSize,
253                   m_useRDOQ,
254                   m_useRDOQTS,
255#if T0196_SELECTIVE_RDOQ
256                   m_useSelectiveRDOQ,
257#endif
258                   true
259                  ,m_useTransformSkipFast
260#if ADAPTIVE_QP_SELECTION
261                  ,m_bUseAdaptQpSelect
262#endif
263                  );
264
265  // initialize encoder search class
266  m_cSearch.init( this, &m_cTrQuant, m_iSearchRange, m_bipredSearchRange, m_motionEstimationSearchMethod, m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, &m_cEntropyCoder, &m_cRdCost, getRDSbacCoder(), getRDGoOnSbacCoder() );
267
268  m_iMaxRefPicNum = 0;
269
270  xInitScalingLists();
271
272#if SVC_EXTENSION
273  m_iSPSIdCnt ++;
274  m_iPPSIdCnt ++;
275  xInitILRP();
276#endif
277}
278
279Void TEncTop::xInitScalingLists()
280{
281  // Initialise scaling lists
282  // The encoder will only use the SPS scaling lists. The PPS will never be marked present.
283  const Int maxLog2TrDynamicRange[MAX_NUM_CHANNEL_TYPE] =
284  {
285      m_cSPS.getMaxLog2TrDynamicRange(CHANNEL_TYPE_LUMA),
286      m_cSPS.getMaxLog2TrDynamicRange(CHANNEL_TYPE_CHROMA)
287  };
288  if(getUseScalingListId() == SCALING_LIST_OFF)
289  {
290    getTrQuant()->setFlatScalingList(maxLog2TrDynamicRange, m_cSPS.getBitDepths());
291    getTrQuant()->setUseScalingList(false);
292    m_cSPS.setScalingListPresentFlag(false);
293    m_cPPS.setScalingListPresentFlag(false);
294  }
295  else if(getUseScalingListId() == SCALING_LIST_DEFAULT)
296  {
297#if SVC_EXTENSION
298    // inferring of the scaling list can be moved to the config file
299    UInt refLayerId = 0;
300    if( m_layerId > 0 && !m_cVPS.getNonHEVCBaseLayerFlag() && m_cVPS.getRecursiveRefLayerFlag( m_layerId, refLayerId ) )
301    {
302      m_cSPS.setInferScalingListFlag( true );
303      m_cSPS.setScalingListRefLayerId( refLayerId );
304      m_cSPS.setScalingListPresentFlag( false );
305      m_cPPS.setInferScalingListFlag( false );
306      m_cPPS.setScalingListPresentFlag( false );
307
308      // infer the scaling list from the reference layer
309      getTrQuant()->setScalingList( &m_ppcTEncTop[m_cVPS.getLayerIdxInVps(refLayerId)]->getSPS()->getScalingList(), maxLog2TrDynamicRange, m_ppcTEncTop[m_cVPS.getLayerIdxInVps(refLayerId)]->getSPS()->getBitDepths() );
310    }
311    else
312    {
313#endif
314    m_cSPS.getScalingList().setDefaultScalingList ();
315    m_cSPS.setScalingListPresentFlag(false);
316    m_cPPS.setScalingListPresentFlag(false);
317
318    getTrQuant()->setScalingList(&(m_cSPS.getScalingList()), maxLog2TrDynamicRange, m_cSPS.getBitDepths());
319#if SVC_EXTENSION
320    }
321#endif
322    getTrQuant()->setUseScalingList(true);
323  }
324  else if(getUseScalingListId() == SCALING_LIST_FILE_READ)
325  {
326#if SVC_EXTENSION
327    // inferring of the scaling list can be moved to the config file
328    UInt refLayerId = 0;
329    if( m_layerId > 0 && !m_cVPS.getNonHEVCBaseLayerFlag() && m_cVPS.getRecursiveRefLayerFlag( m_layerId, refLayerId ) )
330    {
331      m_cSPS.setInferScalingListFlag( true );
332      m_cSPS.setScalingListRefLayerId( refLayerId );
333      m_cSPS.setScalingListPresentFlag( false );
334      m_cPPS.setInferScalingListFlag( false );
335      m_cPPS.setScalingListPresentFlag( false );
336
337      // infer the scaling list from the reference layer
338      getTrQuant()->setScalingList( &m_ppcTEncTop[m_cVPS.getLayerIdxInVps(refLayerId)]->getSPS()->getScalingList(), maxLog2TrDynamicRange, m_ppcTEncTop[m_cVPS.getLayerIdxInVps(refLayerId)]->getSPS()->getBitDepths() );
339    }
340    else
341    {
342#endif
343    m_cSPS.getScalingList().setDefaultScalingList ();
344    if(m_cSPS.getScalingList().xParseScalingList(getScalingListFileName()))
345    {
346      Bool bParsedScalingList=false; // Use of boolean so that assertion outputs useful string
347      assert(bParsedScalingList);
348      exit(1);
349    }
350    m_cSPS.getScalingList().checkDcOfMatrix();
351    m_cSPS.setScalingListPresentFlag(m_cSPS.getScalingList().checkDefaultScalingList());
352    m_cPPS.setScalingListPresentFlag(false);
353    getTrQuant()->setScalingList(&(m_cSPS.getScalingList()), maxLog2TrDynamicRange, m_cSPS.getBitDepths());
354#if SVC_EXTENSION
355    }
356#endif
357    getTrQuant()->setUseScalingList(true);
358  }
359  else
360  {
361    printf("error : ScalingList == %d not supported\n",getUseScalingListId());
362    assert(0);
363  }
364
365  if (getUseScalingListId() != SCALING_LIST_OFF)
366  {
367    // Prepare delta's:
368    for(UInt sizeId = 0; sizeId < SCALING_LIST_SIZE_NUM; sizeId++)
369    {
370      const Int predListStep = (sizeId == SCALING_LIST_32x32? (SCALING_LIST_NUM/NUMBER_OF_PREDICTION_MODES) : 1); // if 32x32, skip over chroma entries.
371
372      for(UInt listId = 0; listId < SCALING_LIST_NUM; listId+=predListStep)
373      {
374        m_cSPS.getScalingList().checkPredMode( sizeId, listId );
375      }
376    }
377  }
378}
379
380// ====================================================================================================================
381// Public member functions
382// ====================================================================================================================
383
384Void TEncTop::deletePicBuffer()
385{
386  TComList<TComPic*>::iterator iterPic = m_cListPic.begin();
387  Int iSize = Int( m_cListPic.size() );
388
389  for ( Int i = 0; i < iSize; i++ )
390  {
391    TComPic* pcPic = *(iterPic++);
392
393    pcPic->destroy();
394    delete pcPic;
395    pcPic = NULL;
396  }
397}
398
399/**
400 - Application has picture buffer list with size of GOP + 1
401 - Picture buffer list acts like as ring buffer
402 - End of the list has the latest picture
403 .
404 \param   flush               cause encoder to encode a partial GOP
405 \param   pcPicYuvOrg         original YUV picture
406 \param   pcPicYuvTrueOrg     
407 \param   snrCSC
408 \retval  rcListPicYuvRecOut  list of reconstruction YUV pictures
409 \retval  accessUnitsOut      list of output access units
410 \retval  iNumEncoded         number of encoded pictures
411 */
412#if SVC_EXTENSION
413Void TEncTop::encode( TComPicYuv* pcPicYuvOrg, const InputColourSpaceConversion snrCSC, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int iPicIdInGOP )
414{
415  // compress GOP
416#if !RC_SHVC_HARMONIZATION
417  if ( m_RCEnableRateControl )
418  {
419    m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
420  }
421#endif
422
423  // compress GOP
424  m_cGOPEncoder.compressGOP(iPicIdInGOP, m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, false, false, snrCSC, m_printFrameMSE);
425
426#if !RC_SHVC_HARMONIZATION
427  if ( m_RCEnableRateControl )
428  {
429    m_cRateCtrl.destroyRCGOP();
430  }
431#endif
432 
433  m_uiNumAllPicCoded ++;
434}
435
436Void TEncTop::encodePrep( TComPicYuv* pcPicYuvOrg, TComPicYuv* pcPicYuvTrueOrg )
437{
438  if (pcPicYuvOrg != NULL)
439  {
440    // get original YUV
441    TComPic* pcPicCurr = NULL;
442    xGetNewPicBuffer( pcPicCurr );
443    pcPicYuvOrg->copyToPic( pcPicCurr->getPicYuvOrg() );
444    pcPicYuvTrueOrg->copyToPic( pcPicCurr->getPicYuvTrueOrg() );
445
446    // compute image characteristics
447    if ( getUseAdaptiveQP() )
448    {
449      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcPicCurr ) );
450    }
451  }
452}
453#else
454Void TEncTop::encode( Bool flush, TComPicYuv* pcPicYuvOrg, TComPicYuv* pcPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded )
455{
456  if (pcPicYuvOrg != NULL)
457  {
458    // get original YUV
459    TComPic* pcPicCurr = NULL;
460
461    xGetNewPicBuffer( pcPicCurr );
462    pcPicYuvOrg->copyToPic( pcPicCurr->getPicYuvOrg() );
463    pcPicYuvTrueOrg->copyToPic( pcPicCurr->getPicYuvTrueOrg() );
464
465    // compute image characteristics
466    if ( getUseAdaptiveQP() )
467    {
468      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcPicCurr ) );
469    }
470  }
471
472  if ((m_iNumPicRcvd == 0) || (!flush && (m_iPOCLast != 0) && (m_iNumPicRcvd != m_iGOPSize) && (m_iGOPSize != 0)))
473  {
474    iNumEncoded = 0;
475    return;
476  }
477
478  if ( m_RCEnableRateControl )
479  {
480    m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
481  }
482
483  // compress GOP
484  m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, false, false, snrCSC, m_printFrameMSE);
485
486  if ( m_RCEnableRateControl )
487  {
488    m_cRateCtrl.destroyRCGOP();
489  }
490
491  iNumEncoded         = m_iNumPicRcvd;
492  m_iNumPicRcvd       = 0;
493  m_uiNumAllPicCoded += iNumEncoded;
494}
495#endif
496
497/**------------------------------------------------
498 Separate interlaced frame into two fields
499 -------------------------------------------------**/
500Void separateFields(Pel* org, Pel* dstField, UInt stride, UInt width, UInt height, Bool isTop)
501{
502  if (!isTop)
503  {
504    org += stride;
505  }
506  for (Int y = 0; y < height>>1; y++)
507  {
508    for (Int x = 0; x < width; x++)
509    {
510      dstField[x] = org[x];
511    }
512
513    dstField += stride;
514    org += stride*2;
515  }
516
517}
518
519#if SVC_EXTENSION
520Void TEncTop::encodePrep( TComPicYuv* pcPicYuvOrg, TComPicYuv* pcPicYuvTrueOrg, Bool isTff )
521{
522  for (Int fieldNum=0; fieldNum<2; fieldNum++)
523  {
524    if (pcPicYuvOrg)
525    {
526      /* -- field initialization -- */
527      const Bool isTopField=isTff==(fieldNum==0);
528
529      TComPic *pcField;
530      xGetNewPicBuffer( pcField );
531      pcField->setReconMark (false);                     // where is this normally?
532           
533      pcField->getSlice(0)->setPOC( m_iPOCLast );        // superfluous?
534      pcField->getPicYuvRec()->setBorderExtension(false);// where is this normally?
535
536      pcField->setTopField(isTopField);                  // interlaced requirement
537
538      for (UInt componentIndex = 0; componentIndex < pcPicYuvOrg->getNumberValidComponents(); componentIndex++)
539      {
540        const ComponentID component = ComponentID(componentIndex);
541        const UInt stride = pcPicYuvOrg->getStride(component);
542
543        separateFields((pcPicYuvOrg->getBuf(component) + pcPicYuvOrg->getMarginX(component) + (pcPicYuvOrg->getMarginY(component) * stride)),
544                       pcField->getPicYuvOrg()->getAddr(component),
545                       pcPicYuvOrg->getStride(component),
546                       pcPicYuvOrg->getWidth(component),
547                       pcPicYuvOrg->getHeight(component),
548                       isTopField);
549
550        separateFields((pcPicYuvTrueOrg->getBuf(component) + pcPicYuvTrueOrg->getMarginX(component) + (pcPicYuvTrueOrg->getMarginY(component) * stride)),
551                       pcField->getPicYuvTrueOrg()->getAddr(component),
552                       pcPicYuvTrueOrg->getStride(component),
553                       pcPicYuvTrueOrg->getWidth(component),
554                       pcPicYuvTrueOrg->getHeight(component),
555                       isTopField);
556      }
557
558      // compute image characteristics
559      if ( getUseAdaptiveQP() )
560      {
561        m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcField ) );
562      }
563    }
564  }
565}
566
567Void TEncTop::encode( TComPicYuv* pcPicYuvOrg, const InputColourSpaceConversion snrCSC, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int iPicIdInGOP, Bool isTff )
568{
569  for (Int fieldNum=0; fieldNum<2; fieldNum++)
570  {
571    if (pcPicYuvOrg)
572    {
573      if (fieldNum==1)                                   // where is this normally?
574      {
575        TComPicYuv* rpcPicYuvRec;
576
577        // org. buffer
578        if ( rcListPicYuvRecOut.size() >= (UInt)m_iGOPSize+1 ) // need to maintain field 0 in list of RecOuts while processing field 1. Hence +1 on m_iGOPSize.
579        {
580          rpcPicYuvRec = rcListPicYuvRecOut.popFront();
581        }
582        else
583        {
584          rpcPicYuvRec = new TComPicYuv;
585          rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, true);
586        }
587        rcListPicYuvRecOut.pushBack( rpcPicYuvRec );
588      }
589    }
590
591    // compress GOP
592    m_cGOPEncoder.compressGOP(iPicIdInGOP, m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff, snrCSC, m_printFrameMSE);
593  }
594
595  m_uiNumAllPicCoded ++;
596}
597#else
598Void TEncTop::encode(Bool flush, TComPicYuv* pcPicYuvOrg, TComPicYuv* pcPicYuvTrueOrg, const InputColourSpaceConversion snrCSC, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Bool isTff)
599{
600  iNumEncoded = 0;
601
602  for (Int fieldNum=0; fieldNum<2; fieldNum++)
603  {
604    if (pcPicYuvOrg)
605    {
606
607      /* -- field initialization -- */
608      const Bool isTopField=isTff==(fieldNum==0);
609
610      TComPic *pcField;
611      xGetNewPicBuffer( pcField );
612      pcField->setReconMark (false);                     // where is this normally?
613
614      if (fieldNum==1)                                   // where is this normally?
615      {
616        TComPicYuv* rpcPicYuvRec;
617
618        // org. buffer
619        if ( rcListPicYuvRecOut.size() >= (UInt)m_iGOPSize+1 ) // need to maintain field 0 in list of RecOuts while processing field 1. Hence +1 on m_iGOPSize.
620        {
621          rpcPicYuvRec = rcListPicYuvRecOut.popFront();
622        }
623        else
624        {
625          rpcPicYuvRec = new TComPicYuv;
626          rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, m_maxCUWidth, m_maxCUHeight, m_maxTotalCUDepth, true);
627        }
628        rcListPicYuvRecOut.pushBack( rpcPicYuvRec );
629      }
630
631      pcField->getSlice(0)->setPOC( m_iPOCLast );        // superfluous?
632      pcField->getPicYuvRec()->setBorderExtension(false);// where is this normally?
633
634      pcField->setTopField(isTopField);                  // interlaced requirement
635
636      for (UInt componentIndex = 0; componentIndex < pcPicYuvOrg->getNumberValidComponents(); componentIndex++)
637      {
638        const ComponentID component = ComponentID(componentIndex);
639        const UInt stride = pcPicYuvOrg->getStride(component);
640
641        separateFields((pcPicYuvOrg->getBuf(component) + pcPicYuvOrg->getMarginX(component) + (pcPicYuvOrg->getMarginY(component) * stride)),
642                       pcField->getPicYuvOrg()->getAddr(component),
643                       pcPicYuvOrg->getStride(component),
644                       pcPicYuvOrg->getWidth(component),
645                       pcPicYuvOrg->getHeight(component),
646                       isTopField);
647
648        separateFields((pcPicYuvTrueOrg->getBuf(component) + pcPicYuvTrueOrg->getMarginX(component) + (pcPicYuvTrueOrg->getMarginY(component) * stride)),
649                       pcField->getPicYuvTrueOrg()->getAddr(component),
650                       pcPicYuvTrueOrg->getStride(component),
651                       pcPicYuvTrueOrg->getWidth(component),
652                       pcPicYuvTrueOrg->getHeight(component),
653                       isTopField);
654      }
655
656      // compute image characteristics
657      if ( getUseAdaptiveQP() )
658      {
659        m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcField ) );
660      }
661    }
662
663    if ( m_iNumPicRcvd && ((flush&&fieldNum==1) || (m_iPOCLast/2)==0 || m_iNumPicRcvd==m_iGOPSize ) )
664    {
665      // compress GOP
666      m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff, snrCSC, m_printFrameMSE);
667
668      iNumEncoded += m_iNumPicRcvd;
669      m_uiNumAllPicCoded += m_iNumPicRcvd;
670      m_iNumPicRcvd = 0;
671    }
672  }
673}
674#endif
675
676// ====================================================================================================================
677// Protected member functions
678// ====================================================================================================================
679
680/**
681 - Application has picture buffer list with size of GOP + 1
682 - Picture buffer list acts like as ring buffer
683 - End of the list has the latest picture
684 .
685 \retval rpcPic obtained picture buffer
686 */
687Void TEncTop::xGetNewPicBuffer ( TComPic*& rpcPic )
688{
689  // At this point, the SPS and PPS can be considered activated - they are copied to the new TComPic.
690
691  TComSlice::sortPicList(m_cListPic);
692
693
694  if (m_cListPic.size() >= (UInt)(m_iGOPSize + getMaxDecPicBuffering(MAX_TLAYER-1) + 2) )
695  {
696    TComList<TComPic*>::iterator iterPic  = m_cListPic.begin();
697    Int iSize = Int( m_cListPic.size() );
698    for ( Int i = 0; i < iSize; i++ )
699    {
700      rpcPic = *(iterPic++);
701      if(rpcPic->getSlice(0)->isReferenced() == false)
702      {
703        break;
704      }
705    }
706  }
707  else
708  {
709    if ( getUseAdaptiveQP() )
710    {
711      TEncPic* pcEPic = new TEncPic;
712
713#if SVC_EXTENSION //Temporal solution, should be modified
714      if( m_layerId > 0 )
715      {
716        for( UInt i = 0; i < m_cVPS.getNumDirectRefLayers( m_layerId ); i++ )
717        {
718          const Window scalEL = m_cPPS.getScaledRefLayerWindowForLayer(m_cVPS.getRefLayerId(m_layerId, i));
719          const Window altRL = m_cPPS.getRefLayerWindowForLayer(m_cVPS.getRefLayerId(m_layerId, i));
720          Bool equalOffsets = scalEL.hasEqualOffset(altRL);
721          Bool zeroPhase = m_cPPS.hasZeroResamplingPhase(m_cVPS.getRefLayerId(m_layerId, i));
722
723          TEncTop *pcEncTopBase = (TEncTop *)getRefLayerEnc( i );
724
725          const Int bitDepthLuma = m_cSPS.getBitDepth(CHANNEL_TYPE_LUMA);
726          const Int bitDepthChroma = m_cSPS.getBitDepth(CHANNEL_TYPE_CHROMA);
727          const Int refBitDepthLuma = pcEncTopBase->getSPS()->getBitDepth(CHANNEL_TYPE_LUMA);
728          const Int refBitDepthChroma = pcEncTopBase->getSPS()->getBitDepth(CHANNEL_TYPE_CHROMA);
729
730          Bool sameBitDepths = ( bitDepthLuma == refBitDepthLuma ) && ( bitDepthChroma == refBitDepthChroma );
731
732          if( m_iSourceWidth == pcEncTopBase->getSourceWidth() && m_iSourceHeight == pcEncTopBase->getSourceHeight() && equalOffsets && zeroPhase )
733          {
734            pcEPic->setEqualPictureSizeAndOffsetFlag( i, true );
735          }
736
737          if( !pcEPic->equalPictureSizeAndOffsetFlag(i) || !sameBitDepths
738#if CGS_3D_ASYMLUT
739            || m_cPPS.getCGSFlag() > 0
740#endif
741            )
742          {
743            pcEPic->setSpatialEnhLayerFlag( i, true );
744
745            //only for scalable extension
746            assert( m_cVPS.getScalabilityMask( SCALABILITY_ID ) == true );
747          }
748        }
749      }
750
751      pcEPic->create( m_cVPS, m_cSPS, m_cPPS, m_cPPS.getMaxCuDQPDepth()+1, false, m_layerId);
752#else  //SVC_EXTENSION
753      pcEPic->create( m_cSPS, m_cPPS, m_cPPS.getMaxCuDQPDepth()+1, false);
754#endif //SVC_EXTENSION
755      rpcPic = pcEPic;
756    }
757    else
758    {
759      rpcPic = new TComPic;
760
761#if SVC_EXTENSION //Temporal solution, should be modified
762      if( m_layerId > 0 )
763      {
764        for( UInt i = 0; i < m_cVPS.getNumDirectRefLayers( m_layerId ); i++ )
765        {
766          const Window scalEL = m_cPPS.getScaledRefLayerWindowForLayer(m_cVPS.getRefLayerId(m_layerId, i));
767          const Window altRL = m_cPPS.getRefLayerWindowForLayer(m_cVPS.getRefLayerId(m_layerId, i));
768          Bool equalOffsets = scalEL.hasEqualOffset(altRL);
769          Bool zeroPhase = m_cPPS.hasZeroResamplingPhase(m_cVPS.getRefLayerId(m_layerId, i));
770
771          TEncTop *pcEncTopBase = (TEncTop *)getRefLayerEnc( i );
772
773          const Int bitDepthLuma = m_cSPS.getBitDepth(CHANNEL_TYPE_LUMA);
774          const Int bitDepthChroma = m_cSPS.getBitDepth(CHANNEL_TYPE_CHROMA);
775          const Int refBitDepthLuma = m_cSPS.getBitDepth(CHANNEL_TYPE_LUMA);
776          const Int refBitDepthChroma = m_cSPS.getBitDepth(CHANNEL_TYPE_CHROMA);
777
778          Bool sameBitDepths = ( bitDepthLuma == refBitDepthLuma ) && ( bitDepthChroma == refBitDepthChroma );
779
780          if( m_iSourceWidth != pcEncTopBase->getSourceWidth() || m_iSourceHeight != pcEncTopBase->getSourceHeight() || !sameBitDepths
781            || !equalOffsets
782            || !zeroPhase
783#if CGS_3D_ASYMLUT
784            || m_cPPS.getCGSFlag() > 0
785#endif
786            )
787          {
788            rpcPic->setSpatialEnhLayerFlag( i, true );
789
790            //only for scalable extension
791            assert( m_cVPS.getScalabilityMask( SCALABILITY_ID ) == true );
792          }
793        }
794      }
795
796      rpcPic->create( m_cSPS, m_cPPS, false, m_layerId );
797#else  //SVC_EXTENSION
798      rpcPic->create( m_cSPS, m_cPPS, false );
799#endif //SVC_EXTENSION
800    }
801
802    m_cListPic.pushBack( rpcPic );
803  }
804  rpcPic->setReconMark (false);
805
806  m_iPOCLast++;
807  m_iNumPicRcvd++;
808
809  rpcPic->getSlice(0)->setPOC( m_iPOCLast );
810  // mark it should be extended
811  rpcPic->getPicYuvRec()->setBorderExtension(false);
812}
813
814Void TEncTop::xInitVPS()
815{
816  // The SPS must have already been set up.
817  // set the VPS profile information.
818#if SVC_EXTENSION
819  m_cVPS.setVpsVuiPresentFlag( true );
820#else
821  *m_cVPS.getPTL() = *m_cSPS.getPTL();
822  m_cVPS.setMaxOpSets(1);
823#endif
824  m_cVPS.getTimingInfo()->setTimingInfoPresentFlag       ( false );
825  m_cVPS.setNumHrdParameters( 0 );
826
827  m_cVPS.createHrdParamBuffer();
828  for( UInt i = 0; i < m_cVPS.getNumHrdParameters(); i ++ )
829  {
830    m_cVPS.setHrdOpSetIdx( 0, i );
831    m_cVPS.setCprmsPresentFlag( false, i );
832    // Set up HrdParameters here.
833  }
834}
835
836Void TEncTop::xInitSPS()
837{
838#if SVC_EXTENSION
839  m_cSPS.setExtensionFlag( m_layerId > 0 ? true : false );
840  m_cSPS.setNumDirectRefLayers(m_numDirectRefLayers);
841
842  if( !m_numDirectRefLayers && m_numAddLayerSets )
843  {
844    m_cSPS.setLayerId(0); // layer ID 0 for independent layers
845  }
846  else
847  {
848    m_cSPS.setLayerId(m_layerId);
849  }
850#endif //SVC_EXTENSION
851
852  ProfileTierLevel& profileTierLevel = *m_cSPS.getPTL()->getGeneralPTL();
853  profileTierLevel.setLevelIdc(m_level);
854  profileTierLevel.setTierFlag(m_levelTier);
855  profileTierLevel.setProfileIdc(m_profile);
856  profileTierLevel.setProfileCompatibilityFlag(m_profile, 1);
857  profileTierLevel.setProgressiveSourceFlag(m_progressiveSourceFlag);
858  profileTierLevel.setInterlacedSourceFlag(m_interlacedSourceFlag);
859  profileTierLevel.setNonPackedConstraintFlag(m_nonPackedConstraintFlag);
860  profileTierLevel.setFrameOnlyConstraintFlag(m_frameOnlyConstraintFlag);
861  profileTierLevel.setBitDepthConstraint(m_bitDepthConstraintValue);
862  profileTierLevel.setChromaFormatConstraint(m_chromaFormatConstraintValue);
863  profileTierLevel.setIntraConstraintFlag(m_intraConstraintFlag);
864  profileTierLevel.setOnePictureOnlyConstraintFlag(m_onePictureOnlyConstraintFlag);
865  profileTierLevel.setLowerBitRateConstraintFlag(m_lowerBitRateConstraintFlag);
866
867  if ((m_profile == Profile::MAIN10) && (m_bitDepth[CHANNEL_TYPE_LUMA] == 8) && (m_bitDepth[CHANNEL_TYPE_CHROMA] == 8))
868  {
869    /* The above constraint is equal to Profile::MAIN */
870    profileTierLevel.setProfileCompatibilityFlag(Profile::MAIN, 1);
871  }
872  if (m_profile == Profile::MAIN)
873  {
874    /* A Profile::MAIN10 decoder can always decode Profile::MAIN */
875    profileTierLevel.setProfileCompatibilityFlag(Profile::MAIN10, 1);
876  }
877  /* XXX: should Main be marked as compatible with still picture? */
878  /* XXX: may be a good idea to refactor the above into a function
879   * that chooses the actual compatibility based upon options */
880
881  m_cSPS.setPicWidthInLumaSamples  ( m_iSourceWidth      );
882  m_cSPS.setPicHeightInLumaSamples ( m_iSourceHeight     );
883  m_cSPS.setConformanceWindow      ( m_conformanceWindow );
884  m_cSPS.setMaxCUWidth             ( m_maxCUWidth        );
885  m_cSPS.setMaxCUHeight            ( m_maxCUHeight       );
886  m_cSPS.setMaxTotalCUDepth        ( m_maxTotalCUDepth   );
887  m_cSPS.setChromaFormatIdc( m_chromaFormatIDC);
888  m_cSPS.setLog2DiffMaxMinCodingBlockSize(m_log2DiffMaxMinCodingBlockSize);
889
890  Int minCUSize = m_cSPS.getMaxCUWidth() >> ( m_cSPS.getLog2DiffMaxMinCodingBlockSize() );
891  Int log2MinCUSize = 0;
892  while(minCUSize > 1)
893  {
894    minCUSize >>= 1;
895    log2MinCUSize++;
896  }
897
898  m_cSPS.setLog2MinCodingBlockSize(log2MinCUSize);
899
900#if SVC_EXTENSION
901  m_cSPS.setSPSId         ( m_iSPSIdCnt       );
902#endif
903
904  m_cSPS.setPCMLog2MinSize (m_uiPCMLog2MinSize);
905  m_cSPS.setUsePCM        ( m_usePCM           );
906  m_cSPS.setPCMLog2MaxSize( m_pcmLog2MaxSize  );
907
908  m_cSPS.setQuadtreeTULog2MaxSize( m_uiQuadtreeTULog2MaxSize );
909  m_cSPS.setQuadtreeTULog2MinSize( m_uiQuadtreeTULog2MinSize );
910  m_cSPS.setQuadtreeTUMaxDepthInter( m_uiQuadtreeTUMaxDepthInter    );
911  m_cSPS.setQuadtreeTUMaxDepthIntra( m_uiQuadtreeTUMaxDepthIntra    );
912
913  m_cSPS.setTMVPFlagsPresent((getTMVPModeId() == 2 || getTMVPModeId() == 1));
914
915  m_cSPS.setMaxTrSize   ( 1 << m_uiQuadtreeTULog2MaxSize );
916
917  m_cSPS.setUseAMP ( m_useAMP );
918
919  for (UInt channelType = 0; channelType < MAX_NUM_CHANNEL_TYPE; channelType++)
920  {
921#if SVC_EXTENSION
922    m_cSPS.setBitDepth    (ChannelType(channelType), m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_cVPS.getLayerIdxInVps( m_layerId ) ) )->getBitDepthVps(ChannelType(channelType))            );   
923#if O0043_BEST_EFFORT_DECODING
924    m_cSPS.setStreamBitDepth(ChannelType(channelType), m_bitDepth[channelType] );
925#endif
926    m_cSPS.setQpBDOffset  (ChannelType(channelType), (6 * (m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_cVPS.getLayerIdxInVps( m_layerId ) ) )->getBitDepthVps(ChannelType(channelType)) - 8)));
927#else
928    m_cSPS.setBitDepth      (ChannelType(channelType), m_bitDepth[channelType] );
929#if O0043_BEST_EFFORT_DECODING
930    m_cSPS.setStreamBitDepth(ChannelType(channelType), m_bitDepth[channelType] );
931#endif
932    m_cSPS.setQpBDOffset  (ChannelType(channelType), (6 * (m_bitDepth[channelType] - 8)));
933#endif
934    m_cSPS.setPCMBitDepth (ChannelType(channelType), m_PCMBitDepth[channelType]         );
935  }
936
937  m_cSPS.setUseSAO( m_bUseSAO );
938
939  m_cSPS.setMaxTLayers( m_maxTempLayer );
940  m_cSPS.setTemporalIdNestingFlag( ( m_maxTempLayer == 1 ) ? true : false );
941
942  for (Int i = 0; i < min(m_cSPS.getMaxTLayers(),(UInt) MAX_TLAYER); i++ )
943  {
944#if SVC_EXTENSION
945    assert( i < MAX_TLAYER );
946#endif
947    m_cSPS.setMaxDecPicBuffering(m_maxDecPicBuffering[i], i);
948    m_cSPS.setNumReorderPics(m_numReorderPics[i], i);
949  }
950
951  m_cSPS.setPCMFilterDisableFlag  ( m_bPCMFilterDisableFlag );
952  m_cSPS.setScalingListFlag ( (m_useScalingListId == SCALING_LIST_OFF) ? 0 : 1 );
953  m_cSPS.setUseStrongIntraSmoothing( m_useStrongIntraSmoothing );
954  m_cSPS.setVuiParametersPresentFlag(getVuiParametersPresentFlag());
955
956  if (m_cSPS.getVuiParametersPresentFlag())
957  {
958    TComVUI* pcVUI = m_cSPS.getVuiParameters();
959    pcVUI->setAspectRatioInfoPresentFlag(getAspectRatioInfoPresentFlag());
960    pcVUI->setAspectRatioIdc(getAspectRatioIdc());
961    pcVUI->setSarWidth(getSarWidth());
962    pcVUI->setSarHeight(getSarHeight());
963    pcVUI->setOverscanInfoPresentFlag(getOverscanInfoPresentFlag());
964    pcVUI->setOverscanAppropriateFlag(getOverscanAppropriateFlag());
965    pcVUI->setVideoSignalTypePresentFlag(getVideoSignalTypePresentFlag());
966    pcVUI->setVideoFormat(getVideoFormat());
967    pcVUI->setVideoFullRangeFlag(getVideoFullRangeFlag());
968    pcVUI->setColourDescriptionPresentFlag(getColourDescriptionPresentFlag());
969    pcVUI->setColourPrimaries(getColourPrimaries());
970    pcVUI->setTransferCharacteristics(getTransferCharacteristics());
971    pcVUI->setMatrixCoefficients(getMatrixCoefficients());
972    pcVUI->setChromaLocInfoPresentFlag(getChromaLocInfoPresentFlag());
973    pcVUI->setChromaSampleLocTypeTopField(getChromaSampleLocTypeTopField());
974    pcVUI->setChromaSampleLocTypeBottomField(getChromaSampleLocTypeBottomField());
975    pcVUI->setNeutralChromaIndicationFlag(getNeutralChromaIndicationFlag());
976    pcVUI->setDefaultDisplayWindow(getDefaultDisplayWindow());
977    pcVUI->setFrameFieldInfoPresentFlag(getFrameFieldInfoPresentFlag());
978    pcVUI->setFieldSeqFlag(false);
979    pcVUI->setHrdParametersPresentFlag(false);
980    pcVUI->getTimingInfo()->setPocProportionalToTimingFlag(getPocProportionalToTimingFlag());
981    pcVUI->getTimingInfo()->setNumTicksPocDiffOneMinus1   (getNumTicksPocDiffOneMinus1()   );
982    pcVUI->setBitstreamRestrictionFlag(getBitstreamRestrictionFlag());
983    pcVUI->setTilesFixedStructureFlag(getTilesFixedStructureFlag());
984    pcVUI->setMotionVectorsOverPicBoundariesFlag(getMotionVectorsOverPicBoundariesFlag());
985    pcVUI->setMinSpatialSegmentationIdc(getMinSpatialSegmentationIdc());
986    pcVUI->setMaxBytesPerPicDenom(getMaxBytesPerPicDenom());
987    pcVUI->setMaxBitsPerMinCuDenom(getMaxBitsPerMinCuDenom());
988    pcVUI->setLog2MaxMvLengthHorizontal(getLog2MaxMvLengthHorizontal());
989    pcVUI->setLog2MaxMvLengthVertical(getLog2MaxMvLengthVertical());
990  }
991  m_cSPS.setNumLongTermRefPicSPS(NUM_LONG_TERM_REF_PIC_SPS);
992  assert (NUM_LONG_TERM_REF_PIC_SPS <= MAX_NUM_LONG_TERM_REF_PICS);
993  for (Int k = 0; k < NUM_LONG_TERM_REF_PIC_SPS; k++)
994  {
995    m_cSPS.setLtRefPicPocLsbSps(k, 0);
996    m_cSPS.setUsedByCurrPicLtSPSFlag(k, 0);
997  }
998
999#if U0132_TARGET_BITS_SATURATION
1000  if( getPictureTimingSEIEnabled() || getDecodingUnitInfoSEIEnabled() || getCpbSaturationEnabled() )
1001#else
1002  if( getPictureTimingSEIEnabled() || getDecodingUnitInfoSEIEnabled() )
1003#endif
1004  {
1005    xInitHrdParameters();
1006  }
1007  if( getBufferingPeriodSEIEnabled() || getPictureTimingSEIEnabled() || getDecodingUnitInfoSEIEnabled() )
1008  {
1009    m_cSPS.getVuiParameters()->setHrdParametersPresentFlag( true );
1010  }
1011
1012  // Set up SPS range extension settings
1013  m_cSPS.getSpsRangeExtension().setTransformSkipRotationEnabledFlag(m_transformSkipRotationEnabledFlag);
1014  m_cSPS.getSpsRangeExtension().setTransformSkipContextEnabledFlag(m_transformSkipContextEnabledFlag);
1015  for (UInt signallingModeIndex = 0; signallingModeIndex < NUMBER_OF_RDPCM_SIGNALLING_MODES; signallingModeIndex++)
1016  {
1017    m_cSPS.getSpsRangeExtension().setRdpcmEnabledFlag(RDPCMSignallingMode(signallingModeIndex), m_rdpcmEnabledFlag[signallingModeIndex]);
1018  }
1019  m_cSPS.getSpsRangeExtension().setExtendedPrecisionProcessingFlag(m_extendedPrecisionProcessingFlag);
1020  m_cSPS.getSpsRangeExtension().setIntraSmoothingDisabledFlag( m_intraSmoothingDisabledFlag );
1021  m_cSPS.getSpsRangeExtension().setHighPrecisionOffsetsEnabledFlag(m_highPrecisionOffsetsEnabledFlag);
1022  m_cSPS.getSpsRangeExtension().setPersistentRiceAdaptationEnabledFlag(m_persistentRiceAdaptationEnabledFlag);
1023  m_cSPS.getSpsRangeExtension().setCabacBypassAlignmentEnabledFlag(m_cabacBypassAlignmentEnabledFlag);
1024
1025#if SVC_EXTENSION
1026  m_cSPS.inferSPS( m_layerId, &m_cVPS );
1027#endif
1028}
1029
1030#if U0132_TARGET_BITS_SATURATION
1031// calculate scale value of bitrate and initial delay
1032Int calcScale(Int x)
1033{
1034  UInt iMask = 0xffffffff;
1035  Int ScaleValue = 32;
1036
1037  while ((x&iMask) != 0)
1038  {
1039    ScaleValue--;
1040    iMask = (iMask >> 1);
1041  }
1042
1043  return ScaleValue;
1044}
1045#endif
1046Void TEncTop::xInitHrdParameters()
1047{
1048  Bool useSubCpbParams = (getSliceMode() > 0) || (getSliceSegmentMode() > 0);
1049  Int  bitRate         = getTargetBitrate();
1050  Bool isRandomAccess  = getIntraPeriod() > 0;
1051# if U0132_TARGET_BITS_SATURATION
1052  Int cpbSize          = getCpbSize();
1053
1054  if( !getVuiParametersPresentFlag() && !getCpbSaturationEnabled() )
1055#else
1056  if( !getVuiParametersPresentFlag() )
1057#endif
1058  {
1059    return;
1060  }
1061
1062  TComVUI *vui = m_cSPS.getVuiParameters();
1063  TComHRD *hrd = vui->getHrdParameters();
1064
1065  TimingInfo *timingInfo = vui->getTimingInfo();
1066
1067#if SVC_EXTENSION
1068  if( m_layerId > 0 )
1069  {
1070    timingInfo->setTimingInfoPresentFlag( false );
1071    return;
1072  }
1073#endif
1074
1075  timingInfo->setTimingInfoPresentFlag( true );
1076  switch( getFrameRate() )
1077  {
1078  case 24:
1079    timingInfo->setNumUnitsInTick( 1125000 );    timingInfo->setTimeScale    ( 27000000 );
1080    break;
1081  case 25:
1082    timingInfo->setNumUnitsInTick( 1080000 );    timingInfo->setTimeScale    ( 27000000 );
1083    break;
1084  case 30:
1085    timingInfo->setNumUnitsInTick( 900900 );     timingInfo->setTimeScale    ( 27000000 );
1086    break;
1087  case 50:
1088    timingInfo->setNumUnitsInTick( 540000 );     timingInfo->setTimeScale    ( 27000000 );
1089    break;
1090  case 60:
1091    timingInfo->setNumUnitsInTick( 450450 );     timingInfo->setTimeScale    ( 27000000 );
1092    break;
1093  default:
1094    timingInfo->setNumUnitsInTick( 1001 );       timingInfo->setTimeScale    ( 60000 );
1095    break;
1096  }
1097
1098  Bool rateCnt = ( bitRate > 0 );
1099  hrd->setNalHrdParametersPresentFlag( rateCnt );
1100  hrd->setVclHrdParametersPresentFlag( rateCnt );
1101  hrd->setSubPicCpbParamsPresentFlag( useSubCpbParams );
1102
1103  if( hrd->getSubPicCpbParamsPresentFlag() )
1104  {
1105    hrd->setTickDivisorMinus2( 100 - 2 );                          //
1106    hrd->setDuCpbRemovalDelayLengthMinus1( 7 );                    // 8-bit precision ( plus 1 for last DU in AU )
1107    hrd->setSubPicCpbParamsInPicTimingSEIFlag( true );
1108    hrd->setDpbOutputDelayDuLengthMinus1( 5 + 7 );                 // With sub-clock tick factor of 100, at least 7 bits to have the same value as AU dpb delay
1109  }
1110  else
1111  {
1112    hrd->setSubPicCpbParamsInPicTimingSEIFlag( false );
1113  }
1114
1115#if U0132_TARGET_BITS_SATURATION
1116  if (calcScale(bitRate) <= 6)
1117  {
1118    hrd->setBitRateScale(0);
1119  }
1120  else
1121  {
1122    hrd->setBitRateScale(calcScale(bitRate) - 6);
1123  }
1124
1125  if (calcScale(cpbSize) <= 4)
1126  {
1127    hrd->setCpbSizeScale(0);
1128  }
1129  else
1130  {
1131    hrd->setCpbSizeScale(calcScale(cpbSize) - 4);
1132  }
1133#else
1134  hrd->setBitRateScale( 4 );                                       // in units of 2^( 6 + 4 ) = 1,024 bps
1135  hrd->setCpbSizeScale( 6 );                                       // in units of 2^( 4 + 6 ) = 1,024 bit
1136#endif
1137
1138  hrd->setDuCpbSizeScale( 6 );                                     // in units of 2^( 4 + 6 ) = 1,024 bit
1139
1140  hrd->setInitialCpbRemovalDelayLengthMinus1(15);                  // assuming 0.5 sec, log2( 90,000 * 0.5 ) = 16-bit
1141  if( isRandomAccess )
1142  {
1143    hrd->setCpbRemovalDelayLengthMinus1(5);                        // 32 = 2^5 (plus 1)
1144    hrd->setDpbOutputDelayLengthMinus1 (5);                        // 32 + 3 = 2^6
1145  }
1146  else
1147  {
1148    hrd->setCpbRemovalDelayLengthMinus1(9);                        // max. 2^10
1149    hrd->setDpbOutputDelayLengthMinus1 (9);                        // max. 2^10
1150  }
1151
1152  // Note: parameters for all temporal layers are initialized with the same values
1153  Int i, j;
1154  UInt bitrateValue, cpbSizeValue;
1155  UInt duCpbSizeValue;
1156  UInt duBitRateValue = 0;
1157
1158  for( i = 0; i < MAX_TLAYER; i ++ )
1159  {
1160    hrd->setFixedPicRateFlag( i, 1 );
1161    hrd->setPicDurationInTcMinus1( i, 0 );
1162    hrd->setLowDelayHrdFlag( i, 0 );
1163    hrd->setCpbCntMinus1( i, 0 );
1164
1165    //! \todo check for possible PTL violations
1166    // BitRate[ i ] = ( bit_rate_value_minus1[ i ] + 1 ) * 2^( 6 + bit_rate_scale )
1167    bitrateValue = bitRate / (1 << (6 + hrd->getBitRateScale()) );      // bitRate is in bits, so it needs to be scaled down
1168    // CpbSize[ i ] = ( cpb_size_value_minus1[ i ] + 1 ) * 2^( 4 + cpb_size_scale )
1169#if U0132_TARGET_BITS_SATURATION
1170    cpbSizeValue = cpbSize / (1 << (4 + hrd->getCpbSizeScale()) );      // using bitRate results in 1 second CPB size
1171#else
1172    cpbSizeValue = bitRate / (1 << (4 + hrd->getCpbSizeScale()) );      // using bitRate results in 1 second CPB size
1173#endif
1174
1175
1176    // DU CPB size could be smaller (i.e. bitrateValue / number of DUs), but we don't know
1177    // in how many DUs the slice segment settings will result
1178    duCpbSizeValue = bitrateValue;
1179    duBitRateValue = cpbSizeValue;
1180
1181    for( j = 0; j < ( hrd->getCpbCntMinus1( i ) + 1 ); j ++ )
1182    {
1183      hrd->setBitRateValueMinus1( i, j, 0, ( bitrateValue - 1 ) );
1184      hrd->setCpbSizeValueMinus1( i, j, 0, ( cpbSizeValue - 1 ) );
1185      hrd->setDuCpbSizeValueMinus1( i, j, 0, ( duCpbSizeValue - 1 ) );
1186      hrd->setDuBitRateValueMinus1( i, j, 0, ( duBitRateValue - 1 ) );
1187      hrd->setCbrFlag( i, j, 0, false );
1188
1189      hrd->setBitRateValueMinus1( i, j, 1, ( bitrateValue - 1) );
1190      hrd->setCpbSizeValueMinus1( i, j, 1, ( cpbSizeValue - 1 ) );
1191      hrd->setDuCpbSizeValueMinus1( i, j, 1, ( duCpbSizeValue - 1 ) );
1192      hrd->setDuBitRateValueMinus1( i, j, 1, ( duBitRateValue - 1 ) );
1193      hrd->setCbrFlag( i, j, 1, false );
1194    }
1195  }
1196}
1197
1198Void TEncTop::xInitPPS()
1199{
1200  m_cPPS.setConstrainedIntraPred( m_bUseConstrainedIntraPred );
1201  Bool bUseDQP = (getMaxCuDQPDepth() > 0)? true : false;
1202
1203  if((getMaxDeltaQP() != 0 )|| getUseAdaptiveQP())
1204  {
1205    bUseDQP = true;
1206  }
1207
1208  if (m_costMode==COST_SEQUENCE_LEVEL_LOSSLESS || m_costMode==COST_LOSSLESS_CODING)
1209  {
1210    bUseDQP=false;
1211  }
1212
1213
1214  if ( m_RCEnableRateControl )
1215  {
1216    m_cPPS.setUseDQP(true);
1217    m_cPPS.setMaxCuDQPDepth( 0 );
1218  }
1219  else if(bUseDQP)
1220  {
1221    m_cPPS.setUseDQP(true);
1222    m_cPPS.setMaxCuDQPDepth( m_iMaxCuDQPDepth );
1223  }
1224  else
1225  {
1226    m_cPPS.setUseDQP(false);
1227    m_cPPS.setMaxCuDQPDepth( 0 );
1228  }
1229
1230  if ( m_diffCuChromaQpOffsetDepth >= 0 )
1231  {
1232    m_cPPS.getPpsRangeExtension().setDiffCuChromaQpOffsetDepth(m_diffCuChromaQpOffsetDepth);
1233    m_cPPS.getPpsRangeExtension().clearChromaQpOffsetList();
1234    m_cPPS.getPpsRangeExtension().setChromaQpOffsetListEntry(1, 6, 6);
1235    /* todo, insert table entries from command line (NB, 0 should not be touched) */
1236  }
1237  else
1238  {
1239    m_cPPS.getPpsRangeExtension().setDiffCuChromaQpOffsetDepth(0);
1240    m_cPPS.getPpsRangeExtension().clearChromaQpOffsetList();
1241  }
1242  m_cPPS.getPpsRangeExtension().setCrossComponentPredictionEnabledFlag(m_crossComponentPredictionEnabledFlag);
1243  m_cPPS.getPpsRangeExtension().setLog2SaoOffsetScale(CHANNEL_TYPE_LUMA,   m_log2SaoOffsetScale[CHANNEL_TYPE_LUMA  ]);
1244  m_cPPS.getPpsRangeExtension().setLog2SaoOffsetScale(CHANNEL_TYPE_CHROMA, m_log2SaoOffsetScale[CHANNEL_TYPE_CHROMA]);
1245
1246  m_cPPS.setQpOffset(COMPONENT_Cb, m_chromaCbQpOffset );
1247  m_cPPS.setQpOffset(COMPONENT_Cr, m_chromaCrQpOffset );
1248
1249  m_cPPS.setEntropyCodingSyncEnabledFlag( m_entropyCodingSyncEnabledFlag );
1250  m_cPPS.setTilesEnabledFlag( (m_iNumColumnsMinus1 > 0 || m_iNumRowsMinus1 > 0) );
1251  m_cPPS.setUseWP( m_useWeightedPred );
1252  m_cPPS.setWPBiPred( m_useWeightedBiPred );
1253  m_cPPS.setOutputFlagPresentFlag( false );
1254  m_cPPS.setSignHideFlag(getSignHideFlag());
1255
1256  if ( getDeblockingFilterMetric() )
1257  {
1258    m_cPPS.setDeblockingFilterOverrideEnabledFlag(true);
1259    m_cPPS.setPicDisableDeblockingFilterFlag(false);
1260  }
1261  else
1262  {
1263    m_cPPS.setDeblockingFilterOverrideEnabledFlag( !getLoopFilterOffsetInPPS() );
1264    m_cPPS.setPicDisableDeblockingFilterFlag( getLoopFilterDisable() );
1265  }
1266
1267  if (! m_cPPS.getPicDisableDeblockingFilterFlag())
1268  {
1269    m_cPPS.setDeblockingFilterBetaOffsetDiv2( getLoopFilterBetaOffset() );
1270    m_cPPS.setDeblockingFilterTcOffsetDiv2( getLoopFilterTcOffset() );
1271  }
1272  else
1273  {
1274    m_cPPS.setDeblockingFilterBetaOffsetDiv2(0);
1275    m_cPPS.setDeblockingFilterTcOffsetDiv2(0);
1276  }
1277
1278  // deblockingFilterControlPresentFlag is true if any of the settings differ from the inferred values:
1279  const Bool deblockingFilterControlPresentFlag = m_cPPS.getDeblockingFilterOverrideEnabledFlag() ||
1280                                                  m_cPPS.getPicDisableDeblockingFilterFlag()      ||
1281                                                  m_cPPS.getDeblockingFilterBetaOffsetDiv2() != 0 ||
1282                                                  m_cPPS.getDeblockingFilterTcOffsetDiv2() != 0;
1283
1284  m_cPPS.setDeblockingFilterControlPresentFlag(deblockingFilterControlPresentFlag);
1285
1286  m_cPPS.setLog2ParallelMergeLevelMinus2   (m_log2ParallelMergeLevelMinus2 );
1287  m_cPPS.setCabacInitPresentFlag(CABAC_INIT_PRESENT_FLAG);
1288  m_cPPS.setLoopFilterAcrossSlicesEnabledFlag( m_bLFCrossSliceBoundaryFlag );
1289
1290
1291  Int histogram[MAX_NUM_REF + 1];
1292  for( Int i = 0; i <= MAX_NUM_REF; i++ )
1293  {
1294    histogram[i]=0;
1295  }
1296  for( Int i = 0; i < getGOPSize(); i++)
1297  {
1298    assert(getGOPEntry(i).m_numRefPicsActive >= 0 && getGOPEntry(i).m_numRefPicsActive <= MAX_NUM_REF);
1299    histogram[getGOPEntry(i).m_numRefPicsActive]++;
1300  }
1301
1302  Int maxHist=-1;
1303  Int bestPos=0;
1304  for( Int i = 0; i <= MAX_NUM_REF; i++ )
1305  {
1306    if(histogram[i]>maxHist)
1307    {
1308      maxHist=histogram[i];
1309      bestPos=i;
1310    }
1311  }
1312  assert(bestPos <= 15);
1313  m_cPPS.setNumRefIdxL0DefaultActive(bestPos);
1314  m_cPPS.setNumRefIdxL1DefaultActive(bestPos);
1315  m_cPPS.setTransquantBypassEnableFlag(getTransquantBypassEnableFlag());
1316  m_cPPS.setUseTransformSkip( m_useTransformSkip );
1317  m_cPPS.getPpsRangeExtension().setLog2MaxTransformSkipBlockSize( m_log2MaxTransformSkipBlockSize  );
1318
1319  if (m_sliceSegmentMode != NO_SLICES)
1320  {
1321    m_cPPS.setDependentSliceSegmentsEnabledFlag( true );
1322  }
1323 
1324#if SVC_EXTENSION
1325  m_cPPS.setLayerId( m_layerId );
1326
1327  if( !m_numDirectRefLayers && m_numAddLayerSets )
1328  {
1329    m_cPPS.setLayerId(0); // layer ID 0 for independent layers
1330  }
1331
1332  if( m_layerId > 0 )
1333  {
1334    m_cPPS.setListsModificationPresentFlag(true);
1335    m_cPPS.setExtensionFlag(true);
1336  }
1337  else
1338  {
1339    m_cPPS.setListsModificationPresentFlag(false);
1340    m_cPPS.setExtensionFlag(false);
1341  }
1342
1343  m_cPPS.setPPSId( m_iPPSIdCnt );
1344  m_cPPS.setSPSId( m_iSPSIdCnt );
1345
1346  if( m_crossLayerBLAFlag )
1347  {
1348    m_cPPS.setNumExtraSliceHeaderBits( 3 );
1349  }
1350
1351  m_cPPS.setNumRefLayerLocationOffsets(m_numRefLayerLocationOffsets);
1352  for(Int i = 0; i < m_cPPS.getNumRefLayerLocationOffsets(); i++)
1353  {
1354    m_cPPS.setRefLocationOffsetLayerId(i, m_refLocationOffsetLayerId[i]);
1355    m_cPPS.getScaledRefLayerWindow(i) = m_scaledRefLayerWindow[i];
1356    m_cPPS.getRefLayerWindow(i) = m_refLayerWindow[i];
1357    m_cPPS.setScaledRefLayerOffsetPresentFlag( i, m_scaledRefLayerOffsetPresentFlag[i] );
1358    m_cPPS.setRefRegionOffsetPresentFlag( i, m_refRegionOffsetPresentFlag[i] );
1359    m_cPPS.setResamplePhaseSetPresentFlag( i, m_resamplePhaseSetPresentFlag[i] );
1360    m_cPPS.setPhaseHorLuma( m_refLocationOffsetLayerId[i], m_phaseHorLuma[i] );
1361    m_cPPS.setPhaseVerLuma( m_refLocationOffsetLayerId[i], m_phaseVerLuma[i] );
1362    m_cPPS.setPhaseHorChroma( m_refLocationOffsetLayerId[i], m_phaseHorChroma[i] );
1363    m_cPPS.setPhaseVerChroma( m_refLocationOffsetLayerId[i], m_phaseVerChroma[i] );
1364  }
1365#if CGS_3D_ASYMLUT
1366  m_cPPS.setCGSFlag( m_nCGSFlag );
1367#endif
1368  m_cPPS.setPocResetInfoPresentFlag( true );
1369  m_cPPS.setExtensionFlag( true );
1370  m_cPPS.setSliceHeaderExtensionPresentFlag( true );
1371#endif //SVC_EXTENSION
1372}
1373
1374//Function for initializing m_RPSList, a list of TComReferencePictureSet, based on the GOPEntry objects read from the config file.
1375Void TEncTop::xInitRPS(Bool isFieldCoding)
1376{
1377  TComReferencePictureSet*      rps;
1378
1379  m_cSPS.createRPSList(getGOPSize() + m_extraRPSs + 1);
1380  TComRPSList* rpsList = m_cSPS.getRPSList();
1381
1382  for( Int i = 0; i < getGOPSize()+m_extraRPSs; i++)
1383  {
1384    GOPEntry ge = getGOPEntry(i);
1385    rps = rpsList->getReferencePictureSet(i);
1386    rps->setNumberOfPictures(ge.m_numRefPics);
1387    rps->setNumRefIdc(ge.m_numRefIdc);
1388    Int numNeg = 0;
1389    Int numPos = 0;
1390    for( Int j = 0; j < ge.m_numRefPics; j++)
1391    {
1392      rps->setDeltaPOC(j,ge.m_referencePics[j]);
1393      rps->setUsed(j,ge.m_usedByCurrPic[j]);
1394      if(ge.m_referencePics[j]>0)
1395      {
1396        numPos++;
1397      }
1398      else
1399      {
1400        numNeg++;
1401      }
1402    }
1403    rps->setNumberOfNegativePictures(numNeg);
1404    rps->setNumberOfPositivePictures(numPos);
1405
1406    // handle inter RPS intialization from the config file.
1407    rps->setInterRPSPrediction(ge.m_interRPSPrediction > 0);  // not very clean, converting anything > 0 to true.
1408    rps->setDeltaRIdxMinus1(0);                               // index to the Reference RPS is always the previous one.
1409    TComReferencePictureSet*     RPSRef = i>0 ? rpsList->getReferencePictureSet(i-1): NULL;  // get the reference RPS
1410
1411    if (ge.m_interRPSPrediction == 2)  // Automatic generation of the inter RPS idc based on the RIdx provided.
1412    {
1413      assert (RPSRef!=NULL);
1414      Int deltaRPS = getGOPEntry(i-1).m_POC - ge.m_POC;  // the ref POC - current POC
1415      Int numRefDeltaPOC = RPSRef->getNumberOfPictures();
1416
1417      rps->setDeltaRPS(deltaRPS);           // set delta RPS
1418      rps->setNumRefIdc(numRefDeltaPOC+1);  // set the numRefIdc to the number of pictures in the reference RPS + 1.
1419      Int count=0;
1420      for (Int j = 0; j <= numRefDeltaPOC; j++ ) // cycle through pics in reference RPS.
1421      {
1422        Int RefDeltaPOC = (j<numRefDeltaPOC)? RPSRef->getDeltaPOC(j): 0;  // if it is the last decoded picture, set RefDeltaPOC = 0
1423        rps->setRefIdc(j, 0);
1424        for (Int k = 0; k < rps->getNumberOfPictures(); k++ )  // cycle through pics in current RPS.
1425        {
1426          if (rps->getDeltaPOC(k) == ( RefDeltaPOC + deltaRPS))  // if the current RPS has a same picture as the reference RPS.
1427          {
1428              rps->setRefIdc(j, (rps->getUsed(k)?1:2));
1429              count++;
1430              break;
1431          }
1432        }
1433      }
1434      if (count != rps->getNumberOfPictures())
1435      {
1436        printf("Warning: Unable fully predict all delta POCs using the reference RPS index given in the config file.  Setting Inter RPS to false for this RPS.\n");
1437        rps->setInterRPSPrediction(0);
1438      }
1439    }
1440    else if (ge.m_interRPSPrediction == 1)  // inter RPS idc based on the RefIdc values provided in config file.
1441    {
1442      assert (RPSRef!=NULL);
1443      rps->setDeltaRPS(ge.m_deltaRPS);
1444      rps->setNumRefIdc(ge.m_numRefIdc);
1445      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1446      {
1447        rps->setRefIdc(j, ge.m_refIdc[j]);
1448      }
1449      // the following code overwrite the deltaPOC and Used by current values read from the config file with the ones
1450      // computed from the RefIdc.  A warning is printed if they are not identical.
1451      numNeg = 0;
1452      numPos = 0;
1453      TComReferencePictureSet      RPSTemp;  // temporary variable
1454
1455      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1456      {
1457        if (ge.m_refIdc[j])
1458        {
1459          Int deltaPOC = ge.m_deltaRPS + ((j < RPSRef->getNumberOfPictures())? RPSRef->getDeltaPOC(j) : 0);
1460          RPSTemp.setDeltaPOC((numNeg+numPos),deltaPOC);
1461          RPSTemp.setUsed((numNeg+numPos),ge.m_refIdc[j]==1?1:0);
1462          if (deltaPOC<0)
1463          {
1464            numNeg++;
1465          }
1466          else
1467          {
1468            numPos++;
1469          }
1470        }
1471      }
1472      if (numNeg != rps->getNumberOfNegativePictures())
1473      {
1474        printf("Warning: number of negative pictures in RPS is different between intra and inter RPS specified in the config file.\n");
1475        rps->setNumberOfNegativePictures(numNeg);
1476        rps->setNumberOfPictures(numNeg+numPos);
1477      }
1478      if (numPos != rps->getNumberOfPositivePictures())
1479      {
1480        printf("Warning: number of positive pictures in RPS is different between intra and inter RPS specified in the config file.\n");
1481        rps->setNumberOfPositivePictures(numPos);
1482        rps->setNumberOfPictures(numNeg+numPos);
1483      }
1484      RPSTemp.setNumberOfPictures(numNeg+numPos);
1485      RPSTemp.setNumberOfNegativePictures(numNeg);
1486      RPSTemp.sortDeltaPOC();     // sort the created delta POC before comparing
1487      // check if Delta POC and Used are the same
1488      // print warning if they are not.
1489      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1490      {
1491        if (RPSTemp.getDeltaPOC(j) != rps->getDeltaPOC(j))
1492        {
1493          printf("Warning: delta POC is different between intra RPS and inter RPS specified in the config file.\n");
1494          rps->setDeltaPOC(j,RPSTemp.getDeltaPOC(j));
1495        }
1496        if (RPSTemp.getUsed(j) != rps->getUsed(j))
1497        {
1498          printf("Warning: Used by Current in RPS is different between intra and inter RPS specified in the config file.\n");
1499          rps->setUsed(j,RPSTemp.getUsed(j));
1500        }
1501      }
1502    }
1503  }
1504  //In case of field coding, we need to set special parameters for the first bottom field of the sequence, since it is not specified in the cfg file.
1505  //The position = GOPSize + extraRPSs which is (a priori) unused is reserved for this field in the RPS.
1506  if (isFieldCoding)
1507  {
1508    rps = rpsList->getReferencePictureSet(getGOPSize()+m_extraRPSs);
1509    rps->setNumberOfPictures(1);
1510    rps->setNumberOfNegativePictures(1);
1511    rps->setNumberOfPositivePictures(0);
1512    rps->setNumberOfLongtermPictures(0);
1513    rps->setDeltaPOC(0,-1);
1514    rps->setPOC(0,0);
1515    rps->setUsed(0,true);
1516    rps->setInterRPSPrediction(false);
1517    rps->setDeltaRIdxMinus1(0);
1518    rps->setDeltaRPS(0);
1519    rps->setNumRefIdc(0);
1520  }
1521}
1522
1523   // This is a function that
1524   // determines what Reference Picture Set to use
1525   // for a specific slice (with POC = POCCurr)
1526Void TEncTop::selectReferencePictureSet(TComSlice* slice, Int POCCurr, Int GOPid )
1527{
1528  slice->setRPSidx(GOPid);
1529
1530  for(Int extraNum=m_iGOPSize; extraNum<m_extraRPSs+m_iGOPSize; extraNum++)
1531  {
1532    if(m_uiIntraPeriod > 0 && getDecodingRefreshType() > 0)
1533    {
1534      Int POCIndex = POCCurr%m_uiIntraPeriod;
1535      if(POCIndex == 0)
1536      {
1537        POCIndex = m_uiIntraPeriod;
1538      }
1539      if(POCIndex == m_GOPList[extraNum].m_POC)
1540      {
1541        slice->setRPSidx(extraNum);
1542      }
1543    }
1544    else
1545    {
1546      if(POCCurr==m_GOPList[extraNum].m_POC)
1547      {
1548        slice->setRPSidx(extraNum);
1549      }
1550    }
1551  }
1552
1553  if(POCCurr == 1 && slice->getPic()->isField())
1554  {
1555    slice->setRPSidx(m_iGOPSize+m_extraRPSs);
1556  }
1557
1558  const TComReferencePictureSet *rps = (slice->getSPS()->getRPSList()->getReferencePictureSet(slice->getRPSidx()));
1559  slice->setRPS(rps);
1560}
1561
1562Int TEncTop::getReferencePictureSetIdxForSOP(Int POCCurr, Int GOPid )
1563{
1564  Int rpsIdx = GOPid;
1565
1566  for(Int extraNum=m_iGOPSize; extraNum<m_extraRPSs+m_iGOPSize; extraNum++)
1567  {
1568    if(m_uiIntraPeriod > 0 && getDecodingRefreshType() > 0)
1569    {
1570      Int POCIndex = POCCurr%m_uiIntraPeriod;
1571      if(POCIndex == 0)
1572      {
1573        POCIndex = m_uiIntraPeriod;
1574      }
1575      if(POCIndex == m_GOPList[extraNum].m_POC)
1576      {
1577        rpsIdx = extraNum;
1578      }
1579    }
1580    else
1581    {
1582      if(POCCurr==m_GOPList[extraNum].m_POC)
1583      {
1584        rpsIdx = extraNum;
1585      }
1586    }
1587  }
1588
1589  return rpsIdx;
1590}
1591
1592Void  TEncTop::xInitPPSforTiles()
1593{
1594  m_cPPS.setTileUniformSpacingFlag( m_tileUniformSpacingFlag );
1595  m_cPPS.setNumTileColumnsMinus1( m_iNumColumnsMinus1 );
1596  m_cPPS.setNumTileRowsMinus1( m_iNumRowsMinus1 );
1597  if( !m_tileUniformSpacingFlag )
1598  {
1599    m_cPPS.setTileColumnWidth( m_tileColumnWidth );
1600    m_cPPS.setTileRowHeight( m_tileRowHeight );
1601  }
1602  m_cPPS.setLoopFilterAcrossTilesEnabledFlag( m_loopFilterAcrossTilesEnabledFlag );
1603
1604  // # substreams is "per tile" when tiles are independent.
1605}
1606
1607Void  TEncCfg::xCheckGSParameters()
1608{
1609  Int   iWidthInCU = ( m_iSourceWidth%m_maxCUWidth ) ? m_iSourceWidth/m_maxCUWidth + 1 : m_iSourceWidth/m_maxCUWidth;
1610  Int   iHeightInCU = ( m_iSourceHeight%m_maxCUHeight ) ? m_iSourceHeight/m_maxCUHeight + 1 : m_iSourceHeight/m_maxCUHeight;
1611  UInt  uiCummulativeColumnWidth = 0;
1612  UInt  uiCummulativeRowHeight = 0;
1613
1614  //check the column relative parameters
1615  if( m_iNumColumnsMinus1 >= (1<<(LOG2_MAX_NUM_COLUMNS_MINUS1+1)) )
1616  {
1617    printf( "The number of columns is larger than the maximum allowed number of columns.\n" );
1618    exit( EXIT_FAILURE );
1619  }
1620
1621  if( m_iNumColumnsMinus1 >= iWidthInCU )
1622  {
1623    printf( "The current picture can not have so many columns.\n" );
1624    exit( EXIT_FAILURE );
1625  }
1626
1627  if( m_iNumColumnsMinus1 && !m_tileUniformSpacingFlag )
1628  {
1629    for(Int i=0; i<m_iNumColumnsMinus1; i++)
1630    {
1631      uiCummulativeColumnWidth += m_tileColumnWidth[i];
1632    }
1633
1634    if( uiCummulativeColumnWidth >= iWidthInCU )
1635    {
1636      printf( "The width of the column is too large.\n" );
1637      exit( EXIT_FAILURE );
1638    }
1639  }
1640
1641  //check the row relative parameters
1642  if( m_iNumRowsMinus1 >= (1<<(LOG2_MAX_NUM_ROWS_MINUS1+1)) )
1643  {
1644    printf( "The number of rows is larger than the maximum allowed number of rows.\n" );
1645    exit( EXIT_FAILURE );
1646  }
1647
1648  if( m_iNumRowsMinus1 >= iHeightInCU )
1649  {
1650    printf( "The current picture can not have so many rows.\n" );
1651    exit( EXIT_FAILURE );
1652  }
1653
1654  if( m_iNumRowsMinus1 && !m_tileUniformSpacingFlag )
1655  {
1656    for(Int i=0; i<m_iNumRowsMinus1; i++)
1657    {
1658      uiCummulativeRowHeight += m_tileRowHeight[i];
1659    }
1660
1661    if( uiCummulativeRowHeight >= iHeightInCU )
1662    {
1663      printf( "The height of the row is too large.\n" );
1664      exit( EXIT_FAILURE );
1665    }
1666  }
1667}
1668
1669#if SVC_EXTENSION
1670TEncTop* TEncTop::getRefLayerEnc( UInt refLayerIdx )
1671{
1672  if( m_ppcTEncTop[m_cVPS.getLayerIdxInVps(m_layerId)]->getNumDirectRefLayers() <= 0 )
1673  {
1674    return (TEncTop *)getLayerEnc( 0 );
1675  }
1676
1677  return (TEncTop *)getLayerEnc( m_cVPS.getLayerIdxInVps(m_cVPS.getRefLayerId( m_layerId, refLayerIdx )) );
1678}
1679
1680Void TEncTop::xInitILRP()
1681{
1682  if( m_layerId > 0 )
1683  {
1684    if( m_cIlpPic[0] == NULL )
1685    {
1686      for (Int j=0; j < m_numDirectRefLayers; j++)
1687      {
1688        m_cIlpPic[j] = new TComPic;
1689        m_cIlpPic[j]->create(m_cSPS, m_cPPS, true, m_layerId);
1690        for (Int i=0; i<m_cIlpPic[j]->getPicSym()->getNumberOfCtusInFrame(); i++)
1691        {
1692          m_cIlpPic[j]->getPicSym()->getCtu(i)->initCtu(m_cIlpPic[j], i);
1693        }
1694      }
1695    }
1696
1697    if( m_cVPS.getNumRefLayers( m_layerId ) == 0 )
1698    {
1699      UInt layerIdx = m_cVPS.getLayerIdxInVps( m_layerId );
1700      RepFormat* repFmt = m_cVPS.getVpsRepFormat(m_cVPS.getVpsRepFormatIdx(layerIdx));
1701     
1702      if( m_cPPS.getLayerId() == 0 && 
1703          m_cSPS.getLayerId() == 0 &&
1704          repFmt->getChromaFormatVpsIdc() == m_cSPS.getChromaFormatIdc() &&
1705          repFmt->getSeparateColourPlaneVpsFlag() == 0 &&
1706          repFmt->getPicHeightVpsInLumaSamples() == m_cSPS.getPicHeightInLumaSamples() &&
1707          repFmt->getPicWidthVpsInLumaSamples()  == m_cSPS.getPicWidthInLumaSamples() &&
1708          repFmt->getBitDepthVpsLuma()   == m_cSPS.getBitDepth(CHANNEL_TYPE_LUMA) &&
1709          repFmt->getBitDepthVpsChroma() == m_cSPS.getBitDepth(CHANNEL_TYPE_CHROMA) &&
1710          repFmt->getConformanceWindowVps().getWindowLeftOffset()   == m_cSPS.getConformanceWindow().getWindowLeftOffset() &&
1711          repFmt->getConformanceWindowVps().getWindowRightOffset()  == m_cSPS.getConformanceWindow().getWindowRightOffset() &&
1712          repFmt->getConformanceWindowVps().getWindowTopOffset()    == m_cSPS.getConformanceWindow().getWindowTopOffset() &&
1713          repFmt->getConformanceWindowVps().getWindowBottomOffset() == m_cSPS.getConformanceWindow().getWindowBottomOffset() )
1714      {
1715        m_cVPS.setBaseLayerPSCompatibilityFlag(layerIdx, 1);
1716      }
1717      else
1718      {
1719        m_cVPS.setBaseLayerPSCompatibilityFlag(layerIdx, 0);
1720      }
1721    }
1722  }
1723}
1724
1725Window TEncTop::getScaledRefLayerWindowForLayer(Int layerId)
1726{
1727  for (Int i = 0; i < m_numRefLayerLocationOffsets; i++)
1728  {
1729    if (layerId == m_refLocationOffsetLayerId[i])
1730    {
1731      return m_scaledRefLayerWindow[i];
1732    }
1733  }
1734
1735  return Window();
1736}
1737
1738Window TEncTop::getRefLayerWindowForLayer(Int layerId)
1739{
1740  for (Int i = 0; i < m_numRefLayerLocationOffsets; i++)
1741  {
1742    if (layerId == m_refLayerId[i])
1743    {
1744      return m_refLayerWindow[i];
1745    }
1746  }
1747
1748  return Window();
1749}
1750#endif //SVC_EXTENSION
1751//! \}
Note: See TracBrowser for help on using the repository browser.