source: SHVCSoftware/branches/SHM-6-dev/source/App/TAppEncoder/TAppEncTop.cpp @ 652

Last change on this file since 652 was 652, checked in by seregin, 11 years ago

update to HM-14.0

  • Property svn:eol-style set to native
File size: 89.4 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license.
5 *
6 * Copyright (c) 2010-2014, 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     TAppEncTop.cpp
35    \brief    Encoder application class
36*/
37
38#include <list>
39#include <fstream>
40#include <stdlib.h>
41#include <stdio.h>
42#include <fcntl.h>
43#include <assert.h>
44
45#include "TAppEncTop.h"
46#include "TLibEncoder/AnnexBwrite.h"
47
48using namespace std;
49
50//! \ingroup TAppEncoder
51//! \{
52
53// ====================================================================================================================
54// Constructor / destructor / initialization / destroy
55// ====================================================================================================================
56
57TAppEncTop::TAppEncTop()
58{
59  m_iFrameRcvd = 0;
60  m_totalBytes = 0;
61  m_essentialBytes = 0;
62#if SVC_EXTENSION
63  for(UInt layer=0; layer < MAX_LAYERS; layer++)
64  {
65    m_apcTEncTop[layer] = &m_acTEncTop[layer];
66  }
67#endif
68}
69
70TAppEncTop::~TAppEncTop()
71{
72}
73
74#if SVC_EXTENSION
75Void TAppEncTop::xInitLibCfg()
76{
77  TComVPS* vps = m_acTEncTop[0].getVPS();
78
79  vps->setMaxTLayers                       ( m_maxTempLayer );
80  if (m_maxTempLayer == 1)
81  {
82    vps->setTemporalNestingFlag(true);
83  }
84#if !VPS_RENAME
85  vps.setMaxLayers                        ( 1 );
86#endif
87  for(Int i = 0; i < MAX_TLAYER; i++)
88  {
89    vps->setNumReorderPics                 ( m_numReorderPics[i], i );
90    vps->setMaxDecPicBuffering             ( m_maxDecPicBuffering[i], i );
91  }
92
93#if REPN_FORMAT_IN_VPS
94  vps->setRepFormatIdxPresentFlag( true );   // Could be disabled to optimize in some cases.
95  Int maxRepFormatIdx = -1;
96  Int formatIdx = -1;
97  for(UInt layer=0; layer < m_numLayers; layer++)
98  {
99    // Auto generation of the format index
100    if( m_acLayerCfg[layer].getRepFormatIdx() == -1 )
101    {     
102      Bool found = false;
103      for( UInt idx = 0; idx < layer; idx++ )
104      {
105        if( m_acLayerCfg[layer].getSourceWidth() == m_acLayerCfg[idx].getSourceWidth() && m_acLayerCfg[layer].getSourceHeight() == m_acLayerCfg[idx].getSourceHeight()
106#if AUXILIARY_PICTURES
107          && m_acLayerCfg[layer].getChromaFormatIDC() == m_acLayerCfg[idx].getChromaFormatIDC()
108#endif
109#if O0194_DIFFERENT_BITDEPTH_EL_BL
110          && m_acLayerCfg[layer].m_internalBitDepthY == m_acLayerCfg[idx].m_internalBitDepthY && m_acLayerCfg[layer].m_internalBitDepthC == m_acLayerCfg[idx].m_internalBitDepthC
111#endif
112          )
113        {
114          found = true;
115          break;
116        }
117      }
118      if( !found )
119      {
120        formatIdx++;
121      }
122
123      m_acLayerCfg[layer].setRepFormatIdx( formatIdx );
124    }
125
126    assert( m_acLayerCfg[layer].getRepFormatIdx() != -1 && "RepFormatIdx not assigned for a layer" );
127    vps->setVpsRepFormatIdx( layer, m_acLayerCfg[layer].getRepFormatIdx() );
128    maxRepFormatIdx = std::max( m_acLayerCfg[layer].getRepFormatIdx(), maxRepFormatIdx );
129  }
130  assert( vps->getVpsRepFormatIdx( 0 ) == 0 );  // Base layer should point to the first one.
131  Int* mapIdxToLayer = new Int[maxRepFormatIdx + 1];
132  // Check that all the indices from 0 to maxRepFormatIdx are used in the VPS
133  for(Int i = 0; i <= maxRepFormatIdx; i++)
134  {
135    mapIdxToLayer[i] = -1;
136    UInt layer;
137    for(layer=0; layer < m_numLayers; layer++)
138    {
139      if( vps->getVpsRepFormatIdx(layer) == i )
140      {
141        mapIdxToLayer[i] = layer;
142        break;
143      }
144    }
145    assert( layer != m_numLayers );   // One of the VPS Rep format indices not set
146  }
147  vps->setVpsNumRepFormats( maxRepFormatIdx + 1 );
148  for(UInt idx=0; idx < vps->getVpsNumRepFormats(); idx++)
149  {
150    RepFormat *repFormat = vps->getVpsRepFormat( idx );
151#if REPN_FORMAT_CONTROL_FLAG
152    repFormat->setChromaAndBitDepthVpsPresentFlag( true ); 
153    if (idx==0)
154    {
155      assert(repFormat->getChromaAndBitDepthVpsPresentFlag() == true); 
156    }
157#endif
158    repFormat->setPicWidthVpsInLumaSamples  ( m_acLayerCfg[mapIdxToLayer[idx]].getSourceWidth()   );
159    repFormat->setPicHeightVpsInLumaSamples ( m_acLayerCfg[mapIdxToLayer[idx]].getSourceHeight()  );
160#if AUXILIARY_PICTURES
161    repFormat->setChromaFormatVpsIdc        ( m_acLayerCfg[mapIdxToLayer[idx]].getChromaFormatIDC() );
162#else
163    repFormat->setChromaFormatVpsIdc        ( 1                                             );  // Need modification to change for each layer - corresponds to 420
164#endif
165    repFormat->setSeparateColourPlaneVpsFlag( 0                                             );  // Need modification to change for each layer
166#if O0194_DIFFERENT_BITDEPTH_EL_BL
167    repFormat->setBitDepthVpsLuma           ( getInternalBitDepthY(mapIdxToLayer[idx])      );  // Need modification to change for each layer
168    repFormat->setBitDepthVpsChroma         ( getInternalBitDepthC(mapIdxToLayer[idx])      );  // Need modification to change for each layer
169#else
170    repFormat->setBitDepthVpsLuma           ( getInternalBitDepthY()                        );  // Need modification to change for each layer
171    repFormat->setBitDepthVpsChroma         ( getInternalBitDepthC()                        );  // Need modification to change for each layer
172#endif
173#if HIGHER_LAYER_IRAP_SKIP_FLAG
174    m_acTEncTop[mapIdxToLayer[idx]].setSkipPictureAtArcSwitch( m_skipPictureAtArcSwitch );
175#endif
176  }
177  delete [] mapIdxToLayer;
178#endif
179
180  for(UInt layer=0; layer<m_numLayers; layer++)
181  {
182#if O0194_DIFFERENT_BITDEPTH_EL_BL
183    //1
184    g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
185    g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
186
187    g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
188    g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
189
190    // Set this to be used in Upsampling filter in function "TComUpsampleFilter::upsampleBasePic"
191    g_bitDepthYLayer[layer] = g_bitDepthY;
192    g_bitDepthCLayer[layer] = g_bitDepthC;
193
194#if O0194_WEIGHTED_PREDICTION_CGS
195    m_acTEncTop[layer].setInterLayerWeightedPredFlag( m_useInterLayerWeightedPred );
196#endif
197#endif
198    //m_acTEncTop[layer].setVPS(&vps);
199    m_acTEncTop[layer].setFrameRate                    ( m_acLayerCfg[layer].getFrameRate() );
200    m_acTEncTop[layer].setFrameSkip                    ( m_FrameSkip );
201    m_acTEncTop[layer].setSourceWidth                  ( m_acLayerCfg[layer].getSourceWidth() );
202    m_acTEncTop[layer].setSourceHeight                 ( m_acLayerCfg[layer].getSourceHeight() );
203    m_acTEncTop[layer].setConformanceMode              ( m_acLayerCfg[layer].getConformanceMode() );
204    m_acTEncTop[layer].setConformanceWindow            ( m_acLayerCfg[layer].m_confLeft, m_acLayerCfg[layer].m_confRight, m_acLayerCfg[layer].m_confTop, m_acLayerCfg[layer].m_confBottom );
205    m_acTEncTop[layer].setFramesToBeEncoded            ( m_framesToBeEncoded );
206
207    m_acTEncTop[layer].setProfile(m_profile);
208    m_acTEncTop[layer].setLevel(m_levelTier, m_level);
209    m_acTEncTop[layer].setProgressiveSourceFlag(m_progressiveSourceFlag);
210    m_acTEncTop[layer].setInterlacedSourceFlag(m_interlacedSourceFlag);
211    m_acTEncTop[layer].setNonPackedConstraintFlag(m_nonPackedConstraintFlag);
212    m_acTEncTop[layer].setFrameOnlyConstraintFlag(m_frameOnlyConstraintFlag);
213
214#if REF_IDX_MFM
215#if AVC_BASE
216    m_acTEncTop[layer].setMFMEnabledFlag(layer == 0 ? false : ( m_avcBaseLayerFlag ? AVC_SYNTAX : true ) && m_acLayerCfg[layer].getNumMotionPredRefLayers());
217#else
218    m_acTEncTop[layer].setMFMEnabledFlag(layer == 0 ? false : ( m_acLayerCfg[layer].getNumMotionPredRefLayers() > 0 ) );
219#endif
220#endif
221    // set layer ID
222    m_acTEncTop[layer].setLayerId ( layer );
223    m_acTEncTop[layer].setNumLayer ( m_numLayers );
224    m_acTEncTop[layer].setLayerEnc(m_apcTEncTop);
225
226    //====== Coding Structure ========
227    m_acTEncTop[layer].setIntraPeriod                  ( m_acLayerCfg[layer].m_iIntraPeriod );
228    m_acTEncTop[layer].setDecodingRefreshType          ( m_iDecodingRefreshType );
229    m_acTEncTop[layer].setGOPSize                      ( m_iGOPSize );
230    m_acTEncTop[layer].setGopList                      ( m_GOPList );
231
232    m_acTEncTop[layer].setExtraRPSs                    ( m_extraRPSs );
233    for(Int i = 0; i < MAX_TLAYER; i++)
234    {
235      m_acTEncTop[layer].setNumReorderPics             ( m_numReorderPics[i], i );
236      m_acTEncTop[layer].setMaxDecPicBuffering         ( m_maxDecPicBuffering[i], i );
237    }
238    for( UInt uiLoop = 0; uiLoop < MAX_TLAYER; ++uiLoop )
239    {
240      m_acTEncTop[layer].setLambdaModifier( uiLoop, m_adLambdaModifier[ uiLoop ] );
241    }
242    m_acTEncTop[layer].setQP                           ( m_acLayerCfg[layer].getIntQP() );
243
244    m_acTEncTop[layer].setPad                          ( m_acLayerCfg[layer].getPad() );
245
246    m_acTEncTop[layer].setMaxTempLayer                 ( m_maxTempLayer );
247    m_acTEncTop[layer].setUseAMP( m_enableAMP );
248#if N0120_MAX_TID_REF_CFG
249    if (layer<m_numLayers-1)
250    {
251       m_acTEncTop[layer].setMaxTidIlRefPicsPlus1 ( m_acLayerCfg[layer].getMaxTidIlRefPicsPlus1());
252    }
253#endif
254#if VPS_EXTN_DIRECT_REF_LAYERS
255    if(layer)
256    {
257      for(Int i = 0; i < MAX_VPS_LAYER_ID_PLUS1; i++)
258      {
259        m_acTEncTop[layer].setSamplePredEnabledFlag(i, false);
260        m_acTEncTop[layer].setMotionPredEnabledFlag(i, false);
261      }
262      if(m_acLayerCfg[layer].getNumSamplePredRefLayers() == -1)
263      {
264        // Not included in the configuration file; assume that each layer depends on previous layer
265        m_acTEncTop[layer].setNumSamplePredRefLayers   (1);      // One sample pred ref. layer
266        m_acTEncTop[layer].setSamplePredRefLayerId     (0, layer - 1);   // Previous layer
267        m_acTEncTop[layer].setSamplePredEnabledFlag    (layer - 1, true);
268      }
269      else
270      {
271        m_acTEncTop[layer].setNumSamplePredRefLayers   ( m_acLayerCfg[layer].getNumSamplePredRefLayers() );
272        for(Int i = 0; i < m_acTEncTop[layer].getNumSamplePredRefLayers(); i++)
273        {
274          m_acTEncTop[layer].setSamplePredRefLayerId   ( i, m_acLayerCfg[layer].getSamplePredRefLayerId(i));
275          m_acTEncTop[layer].setSamplePredEnabledFlag  (m_acLayerCfg[layer].getSamplePredRefLayerId(i), true);
276        }
277      }
278      if(m_acLayerCfg[layer].getNumMotionPredRefLayers() == -1)
279      {
280        // Not included in the configuration file; assume that each layer depends on previous layer
281        m_acTEncTop[layer].setNumMotionPredRefLayers   (1);      // One motion pred ref. layer
282        m_acTEncTop[layer].setMotionPredRefLayerId     (0, layer - 1);   // Previous layer
283        m_acTEncTop[layer].setMotionPredEnabledFlag    (layer - 1, true);
284      }
285      else
286      {
287        m_acTEncTop[layer].setNumMotionPredRefLayers   ( m_acLayerCfg[layer].getNumMotionPredRefLayers() );
288        for(Int i = 0; i < m_acTEncTop[layer].getNumMotionPredRefLayers(); i++)
289        {
290          m_acTEncTop[layer].setMotionPredRefLayerId   ( i, m_acLayerCfg[layer].getMotionPredRefLayerId(i));
291          m_acTEncTop[layer].setMotionPredEnabledFlag  (m_acLayerCfg[layer].getMotionPredRefLayerId(i), true);
292        }
293      }
294      Int numDirectRefLayers = 0;
295      for (Int i = 0; i < layer; i++)
296      {
297        if (m_acTEncTop[layer].getSamplePredEnabledFlag(i) || m_acTEncTop[layer].getMotionPredEnabledFlag(i))
298        {
299          m_acTEncTop[layer].setRefLayerId(numDirectRefLayers, i);
300          numDirectRefLayers++;
301        }
302      }
303      m_acTEncTop[layer].setNumDirectRefLayers(numDirectRefLayers);
304
305      if(m_acLayerCfg[layer].getNumActiveRefLayers() == -1)
306      {
307        m_acTEncTop[layer].setNumActiveRefLayers( m_acTEncTop[layer].getNumDirectRefLayers() );
308        for( Int i = 0; i < m_acTEncTop[layer].getNumActiveRefLayers(); i++ )
309        {
310          m_acTEncTop[layer].setPredLayerId(i, i);
311        }
312      }
313      else
314      {
315        m_acTEncTop[layer].setNumActiveRefLayers       ( m_acLayerCfg[layer].getNumActiveRefLayers() );
316        for(Int i = 0; i < m_acTEncTop[layer].getNumActiveRefLayers(); i++)
317        {
318          m_acTEncTop[layer].setPredLayerId             ( i, m_acLayerCfg[layer].getPredLayerId(i));
319        }
320      }
321    }
322#endif //VPS_EXTN_DIRECT_REF_LAYERS
323    //===== Slice ========
324
325    //====== Loop/Deblock Filter ========
326    m_acTEncTop[layer].setLoopFilterDisable            ( m_bLoopFilterDisable       );
327    m_acTEncTop[layer].setLoopFilterOffsetInPPS        ( m_loopFilterOffsetInPPS );
328    m_acTEncTop[layer].setLoopFilterBetaOffset         ( m_loopFilterBetaOffsetDiv2  );
329    m_acTEncTop[layer].setLoopFilterTcOffset           ( m_loopFilterTcOffsetDiv2    );
330    m_acTEncTop[layer].setDeblockingFilterControlPresent( m_DeblockingFilterControlPresent);
331    m_acTEncTop[layer].setDeblockingFilterMetric       ( m_DeblockingFilterMetric );
332
333    //====== Motion search ========
334    m_acTEncTop[layer].setFastSearch                   ( m_iFastSearch  );
335    m_acTEncTop[layer].setSearchRange                  ( m_iSearchRange );
336    m_acTEncTop[layer].setBipredSearchRange            ( m_bipredSearchRange );
337
338    //====== Quality control ========
339    m_acTEncTop[layer].setMaxDeltaQP                   ( m_iMaxDeltaQP  );
340    m_acTEncTop[layer].setMaxCuDQPDepth                ( m_iMaxCuDQPDepth  );
341
342    m_acTEncTop[layer].setChromaCbQpOffset             ( m_cbQpOffset     );
343    m_acTEncTop[layer].setChromaCrQpOffset             ( m_crQpOffset  );
344
345#if ADAPTIVE_QP_SELECTION
346    m_acTEncTop[layer].setUseAdaptQpSelect             ( m_bUseAdaptQpSelect   );
347#endif
348   
349    m_acTEncTop[layer].setUseAdaptiveQP                ( m_bUseAdaptiveQP  );
350    m_acTEncTop[layer].setQPAdaptationRange            ( m_iQPAdaptationRange );
351
352    //====== Tool list ========   
353    m_acTEncTop[layer].setDeltaQpRD                    ( m_uiDeltaQpRD  );
354    m_acTEncTop[layer].setUseASR                       ( m_bUseASR      );
355    m_acTEncTop[layer].setUseHADME                     ( m_bUseHADME    );   
356    m_acTEncTop[layer].setdQPs                         ( m_acLayerCfg[layer].getdQPs() );
357    m_acTEncTop[layer].setUseRDOQ                      ( m_useRDOQ     );
358    m_acTEncTop[layer].setUseRDOQTS                    ( m_useRDOQTS   );
359    m_acTEncTop[layer].setRDpenalty                    ( m_rdPenalty );
360#if LAYER_CTB
361    m_acTEncTop[layer].setQuadtreeTULog2MaxSize        ( m_acLayerCfg[layer].m_uiQuadtreeTULog2MaxSize );
362    m_acTEncTop[layer].setQuadtreeTULog2MinSize        ( m_acLayerCfg[layer].m_uiQuadtreeTULog2MinSize );
363    m_acTEncTop[layer].setQuadtreeTUMaxDepthInter      ( m_acLayerCfg[layer].m_uiQuadtreeTUMaxDepthInter );
364    m_acTEncTop[layer].setQuadtreeTUMaxDepthIntra      ( m_acLayerCfg[layer].m_uiQuadtreeTUMaxDepthIntra );
365#else
366    m_acTEncTop[layer].setQuadtreeTULog2MaxSize        ( m_uiQuadtreeTULog2MaxSize );
367    m_acTEncTop[layer].setQuadtreeTULog2MinSize        ( m_uiQuadtreeTULog2MinSize );
368    m_acTEncTop[layer].setQuadtreeTUMaxDepthInter      ( m_uiQuadtreeTUMaxDepthInter );
369    m_acTEncTop[layer].setQuadtreeTUMaxDepthIntra      ( m_uiQuadtreeTUMaxDepthIntra );
370#endif
371    m_acTEncTop[layer].setUseFastEnc                   ( m_bUseFastEnc  );
372    m_acTEncTop[layer].setUseEarlyCU                   ( m_bUseEarlyCU  );
373    m_acTEncTop[layer].setUseFastDecisionForMerge      ( m_useFastDecisionForMerge  );
374    m_acTEncTop[layer].setUseCbfFastMode               ( m_bUseCbfFastMode  );
375    m_acTEncTop[layer].setUseEarlySkipDetection        ( m_useEarlySkipDetection );
376#if FAST_INTRA_SHVC
377    m_acTEncTop[layer].setUseFastIntraScalable         ( m_useFastIntraScalable );
378#endif
379
380    m_acTEncTop[layer].setUseTransformSkip             ( m_useTransformSkip      );
381    m_acTEncTop[layer].setUseTransformSkipFast         ( m_useTransformSkipFast  );
382    m_acTEncTop[layer].setUseConstrainedIntraPred      ( m_bUseConstrainedIntraPred );
383    m_acTEncTop[layer].setPCMLog2MinSize               ( m_uiPCMLog2MinSize);
384    m_acTEncTop[layer].setUsePCM                       ( m_usePCM );
385    m_acTEncTop[layer].setPCMLog2MaxSize               ( m_pcmLog2MaxSize);
386    m_acTEncTop[layer].setMaxNumMergeCand              ( m_maxNumMergeCand );
387
388
389    //====== Weighted Prediction ========
390    m_acTEncTop[layer].setUseWP                   ( m_useWeightedPred      );
391    m_acTEncTop[layer].setWPBiPred                ( m_useWeightedBiPred   );
392#if O0194_WEIGHTED_PREDICTION_CGS
393    if( layer != 0 && m_useInterLayerWeightedPred )
394    {
395      // Enable weighted prediction for enhancement layer
396      m_acTEncTop[layer].setUseWP                 ( true   );
397      m_acTEncTop[layer].setWPBiPred              ( true   );
398    }
399#endif
400    //====== Parallel Merge Estimation ========
401    m_acTEncTop[layer].setLog2ParallelMergeLevelMinus2 ( m_log2ParallelMergeLevel - 2 );
402
403    //====== Slice ========
404    m_acTEncTop[layer].setSliceMode               ( m_sliceMode                );
405    m_acTEncTop[layer].setSliceArgument           ( m_sliceArgument            );
406
407    //====== Dependent Slice ========
408    m_acTEncTop[layer].setSliceSegmentMode        ( m_sliceSegmentMode         );
409    m_acTEncTop[layer].setSliceSegmentArgument    ( m_sliceSegmentArgument     );
410#if LAYER_CTB
411    Int iNumPartInCU = 1<<(m_acLayerCfg[layer].m_uiMaxCUDepth<<1);
412#else
413    Int iNumPartInCU = 1<<(m_uiMaxCUDepth<<1);
414#endif
415    if(m_sliceSegmentMode==FIXED_NUMBER_OF_LCU)
416    {
417      m_acTEncTop[layer].setSliceSegmentArgument ( m_sliceSegmentArgument * iNumPartInCU );
418    }
419    if(m_sliceMode==FIXED_NUMBER_OF_LCU)
420    {
421      m_acTEncTop[layer].setSliceArgument ( m_sliceArgument * iNumPartInCU );
422    }
423    if(m_sliceMode==FIXED_NUMBER_OF_TILES)
424    {
425      m_acTEncTop[layer].setSliceArgument ( m_sliceArgument );
426    }
427
428    if(m_sliceMode == 0 )
429    {
430      m_bLFCrossSliceBoundaryFlag = true;
431    }
432    m_acTEncTop[layer].setLFCrossSliceBoundaryFlag( m_bLFCrossSliceBoundaryFlag );
433    m_acTEncTop[layer].setUseSAO ( m_bUseSAO );
434    m_acTEncTop[layer].setMaxNumOffsetsPerPic (m_maxNumOffsetsPerPic);
435
436    m_acTEncTop[layer].setSaoLcuBoundary (m_saoLcuBoundary);
437    m_acTEncTop[layer].setPCMInputBitDepthFlag  ( m_bPCMInputBitDepthFlag);
438    m_acTEncTop[layer].setPCMFilterDisableFlag  ( m_bPCMFilterDisableFlag);
439
440    m_acTEncTop[layer].setDecodedPictureHashSEIEnabled(m_decodedPictureHashSEIEnabled);
441    m_acTEncTop[layer].setRecoveryPointSEIEnabled( m_recoveryPointSEIEnabled );
442    m_acTEncTop[layer].setBufferingPeriodSEIEnabled( m_bufferingPeriodSEIEnabled );
443    m_acTEncTop[layer].setPictureTimingSEIEnabled( m_pictureTimingSEIEnabled );
444    m_acTEncTop[layer].setToneMappingInfoSEIEnabled                 ( m_toneMappingInfoSEIEnabled );
445    m_acTEncTop[layer].setTMISEIToneMapId                           ( m_toneMapId );
446    m_acTEncTop[layer].setTMISEIToneMapCancelFlag                   ( m_toneMapCancelFlag );
447    m_acTEncTop[layer].setTMISEIToneMapPersistenceFlag              ( m_toneMapPersistenceFlag );
448    m_acTEncTop[layer].setTMISEICodedDataBitDepth                   ( m_toneMapCodedDataBitDepth );
449    m_acTEncTop[layer].setTMISEITargetBitDepth                      ( m_toneMapTargetBitDepth );
450    m_acTEncTop[layer].setTMISEIModelID                             ( m_toneMapModelId );
451    m_acTEncTop[layer].setTMISEIMinValue                            ( m_toneMapMinValue );
452    m_acTEncTop[layer].setTMISEIMaxValue                            ( m_toneMapMaxValue );
453    m_acTEncTop[layer].setTMISEISigmoidMidpoint                     ( m_sigmoidMidpoint );
454    m_acTEncTop[layer].setTMISEISigmoidWidth                        ( m_sigmoidWidth );
455    m_acTEncTop[layer].setTMISEIStartOfCodedInterva                 ( m_startOfCodedInterval );
456    m_acTEncTop[layer].setTMISEINumPivots                           ( m_numPivots );
457    m_acTEncTop[layer].setTMISEICodedPivotValue                     ( m_codedPivotValue );
458    m_acTEncTop[layer].setTMISEITargetPivotValue                    ( m_targetPivotValue );
459    m_acTEncTop[layer].setTMISEICameraIsoSpeedIdc                   ( m_cameraIsoSpeedIdc );
460    m_acTEncTop[layer].setTMISEICameraIsoSpeedValue                 ( m_cameraIsoSpeedValue );
461    m_acTEncTop[layer].setTMISEIExposureCompensationValueSignFlag   ( m_exposureCompensationValueSignFlag );
462    m_acTEncTop[layer].setTMISEIExposureCompensationValueNumerator  ( m_exposureCompensationValueNumerator );
463    m_acTEncTop[layer].setTMISEIExposureCompensationValueDenomIdc   ( m_exposureCompensationValueDenomIdc );
464    m_acTEncTop[layer].setTMISEIRefScreenLuminanceWhite             ( m_refScreenLuminanceWhite );
465    m_acTEncTop[layer].setTMISEIExtendedRangeWhiteLevel             ( m_extendedRangeWhiteLevel );
466    m_acTEncTop[layer].setTMISEINominalBlackLevelLumaCodeValue      ( m_nominalBlackLevelLumaCodeValue );
467    m_acTEncTop[layer].setTMISEINominalWhiteLevelLumaCodeValue      ( m_nominalWhiteLevelLumaCodeValue );
468    m_acTEncTop[layer].setTMISEIExtendedWhiteLevelLumaCodeValue     ( m_extendedWhiteLevelLumaCodeValue );
469    m_acTEncTop[layer].setFramePackingArrangementSEIEnabled( m_framePackingSEIEnabled );
470    m_acTEncTop[layer].setFramePackingArrangementSEIType( m_framePackingSEIType );
471    m_acTEncTop[layer].setFramePackingArrangementSEIId( m_framePackingSEIId );
472    m_acTEncTop[layer].setFramePackingArrangementSEIQuincunx( m_framePackingSEIQuincunx );
473    m_acTEncTop[layer].setFramePackingArrangementSEIInterpretation( m_framePackingSEIInterpretation );
474    m_acTEncTop[layer].setDisplayOrientationSEIAngle( m_displayOrientationSEIAngle );
475    m_acTEncTop[layer].setTemporalLevel0IndexSEIEnabled( m_temporalLevel0IndexSEIEnabled );
476    m_acTEncTop[layer].setGradualDecodingRefreshInfoEnabled( m_gradualDecodingRefreshInfoEnabled );
477    m_acTEncTop[layer].setDecodingUnitInfoSEIEnabled( m_decodingUnitInfoSEIEnabled );
478#if LAYERS_NOT_PRESENT_SEI
479    m_acTEncTop[layer].setLayersNotPresentSEIEnabled( m_layersNotPresentSEIEnabled );
480#endif
481    m_acTEncTop[layer].setSOPDescriptionSEIEnabled( m_SOPDescriptionSEIEnabled );
482    m_acTEncTop[layer].setScalableNestingSEIEnabled( m_scalableNestingSEIEnabled );
483#if N0383_IL_CONSTRAINED_TILE_SETS_SEI
484    m_acTEncTop[layer].setInterLayerConstrainedTileSetsSEIEnabled( m_interLayerConstrainedTileSetsSEIEnabled );
485    m_acTEncTop[layer].setIlNumSetsInMessage( m_ilNumSetsInMessage );
486    m_acTEncTop[layer].setSkippedTileSetPresentFlag( m_skippedTileSetPresentFlag );
487    m_acTEncTop[layer].setTopLeftTileIndex( m_topLeftTileIndex );
488    m_acTEncTop[layer].setBottomRightTileIndex( m_bottomRightTileIndex );
489    m_acTEncTop[layer].setIlcIdc( m_ilcIdc );
490#endif
491    m_acTEncTop[layer].setUniformSpacingIdr          ( m_iUniformSpacingIdr );
492    m_acTEncTop[layer].setNumColumnsMinus1           ( m_iNumColumnsMinus1 );
493    m_acTEncTop[layer].setNumRowsMinus1              ( m_iNumRowsMinus1 );
494    if(m_iUniformSpacingIdr==0)
495    {
496      m_acTEncTop[layer].setColumnWidth              ( m_pColumnWidth );
497      m_acTEncTop[layer].setRowHeight                ( m_pRowHeight );
498    }
499    m_acTEncTop[layer].xCheckGSParameters();
500    Int uiTilesCount          = (m_iNumRowsMinus1+1) * (m_iNumColumnsMinus1+1);
501    if(uiTilesCount == 1)
502    {
503      m_bLFCrossTileBoundaryFlag = true;
504    }
505    m_acTEncTop[layer].setLFCrossTileBoundaryFlag( m_bLFCrossTileBoundaryFlag );
506    m_acTEncTop[layer].setWaveFrontSynchro           ( m_iWaveFrontSynchro );
507    m_acTEncTop[layer].setWaveFrontSubstreams        ( m_acLayerCfg[layer].m_iWaveFrontSubstreams );
508    m_acTEncTop[layer].setTMVPModeId ( m_TMVPModeId );
509    m_acTEncTop[layer].setUseScalingListId           ( m_useScalingListId  );
510    m_acTEncTop[layer].setScalingListFile            ( m_scalingListFile   );
511    m_acTEncTop[layer].setSignHideFlag(m_signHideFlag);
512#if RC_SHVC_HARMONIZATION
513    m_acTEncTop[layer].setUseRateCtrl     (m_acLayerCfg[layer].getRCEnableRateControl());
514    m_acTEncTop[layer].setTargetBitrate   (m_acLayerCfg[layer].getRCTargetBitrate());
515    m_acTEncTop[layer].setKeepHierBit     (m_acLayerCfg[layer].getRCKeepHierarchicalBit());
516    m_acTEncTop[layer].setLCULevelRC      (m_acLayerCfg[layer].getRCLCULevelRC());
517    m_acTEncTop[layer].setUseLCUSeparateModel (m_acLayerCfg[layer].getRCUseLCUSeparateModel());
518    m_acTEncTop[layer].setInitialQP           (m_acLayerCfg[layer].getRCInitialQP());
519    m_acTEncTop[layer].setForceIntraQP        (m_acLayerCfg[layer].getRCForceIntraQP());
520#else
521    m_acTEncTop[layer].setUseRateCtrl         ( m_RCEnableRateControl );
522    m_acTEncTop[layer].setTargetBitrate       ( m_RCTargetBitrate );
523    m_acTEncTop[layer].setKeepHierBit         ( m_RCKeepHierarchicalBit );
524    m_acTEncTop[layer].setLCULevelRC          ( m_RCLCULevelRC );
525    m_acTEncTop[layer].setUseLCUSeparateModel ( m_RCUseLCUSeparateModel );
526    m_acTEncTop[layer].setInitialQP           ( m_RCInitialQP );
527    m_acTEncTop[layer].setForceIntraQP        ( m_RCForceIntraQP );
528#endif
529    m_acTEncTop[layer].setTransquantBypassEnableFlag(m_TransquantBypassEnableFlag);
530    m_acTEncTop[layer].setCUTransquantBypassFlagForceValue(m_CUTransquantBypassFlagForce);
531    m_acTEncTop[layer].setUseRecalculateQPAccordingToLambda( m_recalculateQPAccordingToLambda );
532    m_acTEncTop[layer].setUseStrongIntraSmoothing( m_useStrongIntraSmoothing );
533    m_acTEncTop[layer].setActiveParameterSetsSEIEnabled ( m_activeParameterSetsSEIEnabled );
534    m_acTEncTop[layer].setVuiParametersPresentFlag( m_vuiParametersPresentFlag );
535    m_acTEncTop[layer].setAspectRatioIdc( m_aspectRatioIdc );
536    m_acTEncTop[layer].setSarWidth( m_sarWidth );
537    m_acTEncTop[layer].setSarHeight( m_sarHeight );
538    m_acTEncTop[layer].setOverscanInfoPresentFlag( m_overscanInfoPresentFlag );
539    m_acTEncTop[layer].setOverscanAppropriateFlag( m_overscanAppropriateFlag );
540    m_acTEncTop[layer].setVideoSignalTypePresentFlag( m_videoSignalTypePresentFlag );
541    m_acTEncTop[layer].setVideoFormat( m_videoFormat );
542    m_acTEncTop[layer].setVideoFullRangeFlag( m_videoFullRangeFlag );
543    m_acTEncTop[layer].setColourDescriptionPresentFlag( m_colourDescriptionPresentFlag );
544    m_acTEncTop[layer].setColourPrimaries( m_colourPrimaries );
545    m_acTEncTop[layer].setTransferCharacteristics( m_transferCharacteristics );
546    m_acTEncTop[layer].setMatrixCoefficients( m_matrixCoefficients );
547    m_acTEncTop[layer].setChromaLocInfoPresentFlag( m_chromaLocInfoPresentFlag );
548    m_acTEncTop[layer].setChromaSampleLocTypeTopField( m_chromaSampleLocTypeTopField );
549    m_acTEncTop[layer].setChromaSampleLocTypeBottomField( m_chromaSampleLocTypeBottomField );
550    m_acTEncTop[layer].setNeutralChromaIndicationFlag( m_neutralChromaIndicationFlag );
551    m_acTEncTop[layer].setDefaultDisplayWindow( m_defDispWinLeftOffset, m_defDispWinRightOffset, m_defDispWinTopOffset, m_defDispWinBottomOffset );
552    m_acTEncTop[layer].setFrameFieldInfoPresentFlag( m_frameFieldInfoPresentFlag );
553    m_acTEncTop[layer].setPocProportionalToTimingFlag( m_pocProportionalToTimingFlag );
554    m_acTEncTop[layer].setNumTicksPocDiffOneMinus1   ( m_numTicksPocDiffOneMinus1    );
555    m_acTEncTop[layer].setBitstreamRestrictionFlag( m_bitstreamRestrictionFlag );
556    m_acTEncTop[layer].setTilesFixedStructureFlag( m_tilesFixedStructureFlag );
557    m_acTEncTop[layer].setMotionVectorsOverPicBoundariesFlag( m_motionVectorsOverPicBoundariesFlag );
558    m_acTEncTop[layer].setMinSpatialSegmentationIdc( m_minSpatialSegmentationIdc );
559    m_acTEncTop[layer].setMaxBytesPerPicDenom( m_maxBytesPerPicDenom );
560    m_acTEncTop[layer].setMaxBitsPerMinCuDenom( m_maxBitsPerMinCuDenom );
561    m_acTEncTop[layer].setLog2MaxMvLengthHorizontal( m_log2MaxMvLengthHorizontal );
562    m_acTEncTop[layer].setLog2MaxMvLengthVertical( m_log2MaxMvLengthVertical );
563    m_acTEncTop[layer].setElRapSliceTypeB(layer == 0? 0 : m_elRapSliceBEnabled);
564    if( layer > 0 )
565    {
566      m_acTEncTop[layer].setNumScaledRefLayerOffsets( m_acLayerCfg[layer].m_numScaledRefLayerOffsets );
567      for(Int i = 0; i < m_acLayerCfg[layer].m_numScaledRefLayerOffsets; i++)
568      {
569#if O0098_SCALED_REF_LAYER_ID
570        m_acTEncTop[layer].setScaledRefLayerId(i, m_acLayerCfg[layer].m_scaledRefLayerId[i]);
571#endif
572#if P0312_VERT_PHASE_ADJ
573        m_acTEncTop[layer].setVertPhasePositionEnableFlag( i, m_acLayerCfg[layer].m_vertPhasePositionEnableFlag[i] );
574        m_acTEncTop[layer].getScaledRefLayerWindow(i).setWindow( 2*m_acLayerCfg[layer].m_scaledRefLayerLeftOffset[i], 2*m_acLayerCfg[layer].m_scaledRefLayerRightOffset[i],
575                                                  2*m_acLayerCfg[layer].m_scaledRefLayerTopOffset[i], 2*m_acLayerCfg[layer].m_scaledRefLayerBottomOffset[i], m_acLayerCfg[layer].m_vertPhasePositionEnableFlag[i] );
576#else
577        m_acTEncTop[layer].getScaledRefLayerWindow(i).setWindow( 2*m_acLayerCfg[layer].m_scaledRefLayerLeftOffset[i], 2*m_acLayerCfg[layer].m_scaledRefLayerRightOffset[i],
578                                                  2*m_acLayerCfg[layer].m_scaledRefLayerTopOffset[i], 2*m_acLayerCfg[layer].m_scaledRefLayerBottomOffset[i]);
579#endif
580      }
581    }
582#if M0040_ADAPTIVE_RESOLUTION_CHANGE
583    m_acTEncTop[layer].setAdaptiveResolutionChange( m_adaptiveResolutionChange );
584#endif
585#if AUXILIARY_PICTURES
586    m_acTEncTop[layer].setChromaFormatIDC( m_acLayerCfg[layer].m_chromaFormatIDC );
587#endif
588#if O0153_ALT_OUTPUT_LAYER_FLAG
589    m_acTEncTop[layer].setAltOuputLayerFlag( m_altOutputLayerFlag );
590#endif
591#if O0149_CROSS_LAYER_BLA_FLAG
592    m_acTEncTop[layer].setCrossLayerBLAFlag( m_crossLayerBLAFlag );
593#endif
594  }
595}
596#else //SVC_EXTENSION
597Void TAppEncTop::xInitLibCfg()
598{
599  TComVPS vps;
600
601  vps.setMaxTLayers                       ( m_maxTempLayer );
602  if (m_maxTempLayer == 1)
603  {
604    vps.setTemporalNestingFlag(true);
605  }
606  vps.setMaxLayers                        ( 1 );
607  for(Int i = 0; i < MAX_TLAYER; i++)
608  {
609    vps.setNumReorderPics                 ( m_numReorderPics[i], i );
610    vps.setMaxDecPicBuffering             ( m_maxDecPicBuffering[i], i );
611  }
612  m_cTEncTop.setVPS(&vps);
613
614  m_cTEncTop.setProfile(m_profile);
615  m_cTEncTop.setLevel(m_levelTier, m_level);
616  m_cTEncTop.setProgressiveSourceFlag(m_progressiveSourceFlag);
617  m_cTEncTop.setInterlacedSourceFlag(m_interlacedSourceFlag);
618  m_cTEncTop.setNonPackedConstraintFlag(m_nonPackedConstraintFlag);
619  m_cTEncTop.setFrameOnlyConstraintFlag(m_frameOnlyConstraintFlag);
620
621  m_cTEncTop.setFrameRate                    ( m_iFrameRate );
622  m_cTEncTop.setFrameSkip                    ( m_FrameSkip );
623  m_cTEncTop.setSourceWidth                  ( m_iSourceWidth );
624  m_cTEncTop.setSourceHeight                 ( m_iSourceHeight );
625  m_cTEncTop.setConformanceWindow            ( m_confLeft, m_confRight, m_confTop, m_confBottom );
626  m_cTEncTop.setFramesToBeEncoded            ( m_framesToBeEncoded );
627
628  //====== Coding Structure ========
629  m_cTEncTop.setIntraPeriod                  ( m_iIntraPeriod );
630  m_cTEncTop.setDecodingRefreshType          ( m_iDecodingRefreshType );
631  m_cTEncTop.setGOPSize                      ( m_iGOPSize );
632  m_cTEncTop.setGopList                      ( m_GOPList );
633  m_cTEncTop.setExtraRPSs                    ( m_extraRPSs );
634  for(Int i = 0; i < MAX_TLAYER; i++)
635  {
636    m_cTEncTop.setNumReorderPics             ( m_numReorderPics[i], i );
637    m_cTEncTop.setMaxDecPicBuffering         ( m_maxDecPicBuffering[i], i );
638  }
639  for( UInt uiLoop = 0; uiLoop < MAX_TLAYER; ++uiLoop )
640  {
641    m_cTEncTop.setLambdaModifier( uiLoop, m_adLambdaModifier[ uiLoop ] );
642  }
643  m_cTEncTop.setQP                           ( m_iQP );
644
645  m_cTEncTop.setPad                          ( m_aiPad );
646
647  m_cTEncTop.setMaxTempLayer                 ( m_maxTempLayer );
648  m_cTEncTop.setUseAMP( m_enableAMP );
649
650  //===== Slice ========
651
652  //====== Loop/Deblock Filter ========
653  m_cTEncTop.setLoopFilterDisable            ( m_bLoopFilterDisable       );
654  m_cTEncTop.setLoopFilterOffsetInPPS        ( m_loopFilterOffsetInPPS );
655  m_cTEncTop.setLoopFilterBetaOffset         ( m_loopFilterBetaOffsetDiv2  );
656  m_cTEncTop.setLoopFilterTcOffset           ( m_loopFilterTcOffsetDiv2    );
657  m_cTEncTop.setDeblockingFilterControlPresent( m_DeblockingFilterControlPresent);
658  m_cTEncTop.setDeblockingFilterMetric       ( m_DeblockingFilterMetric );
659
660  //====== Motion search ========
661  m_cTEncTop.setFastSearch                   ( m_iFastSearch  );
662  m_cTEncTop.setSearchRange                  ( m_iSearchRange );
663  m_cTEncTop.setBipredSearchRange            ( m_bipredSearchRange );
664
665  //====== Quality control ========
666  m_cTEncTop.setMaxDeltaQP                   ( m_iMaxDeltaQP  );
667  m_cTEncTop.setMaxCuDQPDepth                ( m_iMaxCuDQPDepth  );
668
669  m_cTEncTop.setChromaCbQpOffset               ( m_cbQpOffset     );
670  m_cTEncTop.setChromaCrQpOffset            ( m_crQpOffset  );
671
672#if ADAPTIVE_QP_SELECTION
673  m_cTEncTop.setUseAdaptQpSelect             ( m_bUseAdaptQpSelect   );
674#endif
675
676  m_cTEncTop.setUseAdaptiveQP                ( m_bUseAdaptiveQP  );
677  m_cTEncTop.setQPAdaptationRange            ( m_iQPAdaptationRange );
678
679  //====== Tool list ========
680  m_cTEncTop.setDeltaQpRD                    ( m_uiDeltaQpRD  );
681  m_cTEncTop.setUseASR                       ( m_bUseASR      );
682  m_cTEncTop.setUseHADME                     ( m_bUseHADME    );
683  m_cTEncTop.setdQPs                         ( m_aidQP        );
684  m_cTEncTop.setUseRDOQ                      ( m_useRDOQ     );
685  m_cTEncTop.setUseRDOQTS                    ( m_useRDOQTS   );
686  m_cTEncTop.setRDpenalty                 ( m_rdPenalty );
687  m_cTEncTop.setQuadtreeTULog2MaxSize        ( m_uiQuadtreeTULog2MaxSize );
688  m_cTEncTop.setQuadtreeTULog2MinSize        ( m_uiQuadtreeTULog2MinSize );
689  m_cTEncTop.setQuadtreeTUMaxDepthInter      ( m_uiQuadtreeTUMaxDepthInter );
690  m_cTEncTop.setQuadtreeTUMaxDepthIntra      ( m_uiQuadtreeTUMaxDepthIntra );
691  m_cTEncTop.setUseFastEnc                   ( m_bUseFastEnc  );
692  m_cTEncTop.setUseEarlyCU                   ( m_bUseEarlyCU  );
693  m_cTEncTop.setUseFastDecisionForMerge      ( m_useFastDecisionForMerge  );
694  m_cTEncTop.setUseCbfFastMode            ( m_bUseCbfFastMode  );
695  m_cTEncTop.setUseEarlySkipDetection            ( m_useEarlySkipDetection );
696#if FAST_INTRA_SHVC
697  m_cTEncTop.setUseFastIntraScalable            ( m_useFastIntraScalable );
698#endif
699
700  m_cTEncTop.setUseTransformSkip             ( m_useTransformSkip      );
701  m_cTEncTop.setUseTransformSkipFast         ( m_useTransformSkipFast  );
702  m_cTEncTop.setUseConstrainedIntraPred      ( m_bUseConstrainedIntraPred );
703  m_cTEncTop.setPCMLog2MinSize          ( m_uiPCMLog2MinSize);
704  m_cTEncTop.setUsePCM                       ( m_usePCM );
705  m_cTEncTop.setPCMLog2MaxSize               ( m_pcmLog2MaxSize);
706  m_cTEncTop.setMaxNumMergeCand              ( m_maxNumMergeCand );
707
708
709  //====== Weighted Prediction ========
710  m_cTEncTop.setUseWP                   ( m_useWeightedPred      );
711  m_cTEncTop.setWPBiPred                ( m_useWeightedBiPred   );
712  //====== Parallel Merge Estimation ========
713  m_cTEncTop.setLog2ParallelMergeLevelMinus2 ( m_log2ParallelMergeLevel - 2 );
714
715  //====== Slice ========
716  m_cTEncTop.setSliceMode               ( m_sliceMode                );
717  m_cTEncTop.setSliceArgument           ( m_sliceArgument            );
718
719  //====== Dependent Slice ========
720  m_cTEncTop.setSliceSegmentMode        ( m_sliceSegmentMode         );
721  m_cTEncTop.setSliceSegmentArgument    ( m_sliceSegmentArgument     );
722  Int iNumPartInCU = 1<<(m_uiMaxCUDepth<<1);
723  if(m_sliceSegmentMode==FIXED_NUMBER_OF_LCU)
724  {
725    m_cTEncTop.setSliceSegmentArgument ( m_sliceSegmentArgument * iNumPartInCU );
726  }
727  if(m_sliceMode==FIXED_NUMBER_OF_LCU)
728  {
729    m_cTEncTop.setSliceArgument ( m_sliceArgument * iNumPartInCU );
730  }
731  if(m_sliceMode==FIXED_NUMBER_OF_TILES)
732  {
733    m_cTEncTop.setSliceArgument ( m_sliceArgument );
734  }
735
736  if(m_sliceMode == 0 )
737  {
738    m_bLFCrossSliceBoundaryFlag = true;
739  }
740  m_cTEncTop.setLFCrossSliceBoundaryFlag( m_bLFCrossSliceBoundaryFlag );
741  m_cTEncTop.setUseSAO ( m_bUseSAO );
742  m_cTEncTop.setMaxNumOffsetsPerPic (m_maxNumOffsetsPerPic);
743
744  m_cTEncTop.setSaoLcuBoundary (m_saoLcuBoundary);
745  m_cTEncTop.setPCMInputBitDepthFlag  ( m_bPCMInputBitDepthFlag);
746  m_cTEncTop.setPCMFilterDisableFlag  ( m_bPCMFilterDisableFlag);
747
748  m_cTEncTop.setDecodedPictureHashSEIEnabled(m_decodedPictureHashSEIEnabled);
749  m_cTEncTop.setRecoveryPointSEIEnabled( m_recoveryPointSEIEnabled );
750  m_cTEncTop.setBufferingPeriodSEIEnabled( m_bufferingPeriodSEIEnabled );
751  m_cTEncTop.setPictureTimingSEIEnabled( m_pictureTimingSEIEnabled );
752  m_cTEncTop.setToneMappingInfoSEIEnabled                 ( m_toneMappingInfoSEIEnabled );
753  m_cTEncTop.setTMISEIToneMapId                           ( m_toneMapId );
754  m_cTEncTop.setTMISEIToneMapCancelFlag                   ( m_toneMapCancelFlag );
755  m_cTEncTop.setTMISEIToneMapPersistenceFlag              ( m_toneMapPersistenceFlag );
756  m_cTEncTop.setTMISEICodedDataBitDepth                   ( m_toneMapCodedDataBitDepth );
757  m_cTEncTop.setTMISEITargetBitDepth                      ( m_toneMapTargetBitDepth );
758  m_cTEncTop.setTMISEIModelID                             ( m_toneMapModelId );
759  m_cTEncTop.setTMISEIMinValue                            ( m_toneMapMinValue );
760  m_cTEncTop.setTMISEIMaxValue                            ( m_toneMapMaxValue );
761  m_cTEncTop.setTMISEISigmoidMidpoint                     ( m_sigmoidMidpoint );
762  m_cTEncTop.setTMISEISigmoidWidth                        ( m_sigmoidWidth );
763  m_cTEncTop.setTMISEIStartOfCodedInterva                 ( m_startOfCodedInterval );
764  m_cTEncTop.setTMISEINumPivots                           ( m_numPivots );
765  m_cTEncTop.setTMISEICodedPivotValue                     ( m_codedPivotValue );
766  m_cTEncTop.setTMISEITargetPivotValue                    ( m_targetPivotValue );
767  m_cTEncTop.setTMISEICameraIsoSpeedIdc                   ( m_cameraIsoSpeedIdc );
768  m_cTEncTop.setTMISEICameraIsoSpeedValue                 ( m_cameraIsoSpeedValue );
769  m_cTEncTop.setTMISEIExposureIndexIdc                    ( m_exposureIndexIdc );
770  m_cTEncTop.setTMISEIExposureIndexValue                  ( m_exposureIndexValue );
771  m_cTEncTop.setTMISEIExposureCompensationValueSignFlag   ( m_exposureCompensationValueSignFlag );
772  m_cTEncTop.setTMISEIExposureCompensationValueNumerator  ( m_exposureCompensationValueNumerator );
773  m_cTEncTop.setTMISEIExposureCompensationValueDenomIdc   ( m_exposureCompensationValueDenomIdc );
774  m_cTEncTop.setTMISEIRefScreenLuminanceWhite             ( m_refScreenLuminanceWhite );
775  m_cTEncTop.setTMISEIExtendedRangeWhiteLevel             ( m_extendedRangeWhiteLevel );
776  m_cTEncTop.setTMISEINominalBlackLevelLumaCodeValue      ( m_nominalBlackLevelLumaCodeValue );
777  m_cTEncTop.setTMISEINominalWhiteLevelLumaCodeValue      ( m_nominalWhiteLevelLumaCodeValue );
778  m_cTEncTop.setTMISEIExtendedWhiteLevelLumaCodeValue     ( m_extendedWhiteLevelLumaCodeValue );
779  m_cTEncTop.setFramePackingArrangementSEIEnabled( m_framePackingSEIEnabled );
780  m_cTEncTop.setFramePackingArrangementSEIType( m_framePackingSEIType );
781  m_cTEncTop.setFramePackingArrangementSEIId( m_framePackingSEIId );
782  m_cTEncTop.setFramePackingArrangementSEIQuincunx( m_framePackingSEIQuincunx );
783  m_cTEncTop.setFramePackingArrangementSEIInterpretation( m_framePackingSEIInterpretation );
784  m_cTEncTop.setDisplayOrientationSEIAngle( m_displayOrientationSEIAngle );
785  m_cTEncTop.setTemporalLevel0IndexSEIEnabled( m_temporalLevel0IndexSEIEnabled );
786  m_cTEncTop.setGradualDecodingRefreshInfoEnabled( m_gradualDecodingRefreshInfoEnabled );
787  m_cTEncTop.setDecodingUnitInfoSEIEnabled( m_decodingUnitInfoSEIEnabled );
788#if LAYERS_NOT_PRESENT_SEI
789  m_cTEncTop.setLayersNotPresentSEIEnabled( m_layersNotPresentSEIEnabled );
790#endif
791  m_cTEncTop.setSOPDescriptionSEIEnabled( m_SOPDescriptionSEIEnabled );
792  m_cTEncTop.setScalableNestingSEIEnabled( m_scalableNestingSEIEnabled );
793  m_cTEncTop.setUniformSpacingIdr          ( m_iUniformSpacingIdr );
794  m_cTEncTop.setNumColumnsMinus1           ( m_iNumColumnsMinus1 );
795  m_cTEncTop.setNumRowsMinus1              ( m_iNumRowsMinus1 );
796  if(m_iUniformSpacingIdr==0)
797  {
798    m_cTEncTop.setColumnWidth              ( m_pColumnWidth );
799    m_cTEncTop.setRowHeight                ( m_pRowHeight );
800  }
801  m_cTEncTop.xCheckGSParameters();
802  Int uiTilesCount          = (m_iNumRowsMinus1+1) * (m_iNumColumnsMinus1+1);
803  if(uiTilesCount == 1)
804  {
805    m_bLFCrossTileBoundaryFlag = true;
806  }
807  m_cTEncTop.setLFCrossTileBoundaryFlag( m_bLFCrossTileBoundaryFlag );
808  m_cTEncTop.setWaveFrontSynchro           ( m_iWaveFrontSynchro );
809  m_cTEncTop.setWaveFrontSubstreams        ( m_iWaveFrontSubstreams );
810  m_cTEncTop.setTMVPModeId ( m_TMVPModeId );
811  m_cTEncTop.setUseScalingListId           ( m_useScalingListId  );
812  m_cTEncTop.setScalingListFile            ( m_scalingListFile   );
813  m_cTEncTop.setSignHideFlag(m_signHideFlag);
814  m_cTEncTop.setUseRateCtrl         ( m_RCEnableRateControl );
815  m_cTEncTop.setTargetBitrate       ( m_RCTargetBitrate );
816  m_cTEncTop.setKeepHierBit         ( m_RCKeepHierarchicalBit );
817  m_cTEncTop.setLCULevelRC          ( m_RCLCULevelRC );
818  m_cTEncTop.setUseLCUSeparateModel ( m_RCUseLCUSeparateModel );
819  m_cTEncTop.setInitialQP           ( m_RCInitialQP );
820  m_cTEncTop.setForceIntraQP        ( m_RCForceIntraQP );
821  m_cTEncTop.setTransquantBypassEnableFlag(m_TransquantBypassEnableFlag);
822  m_cTEncTop.setCUTransquantBypassFlagForceValue(m_CUTransquantBypassFlagForce);
823  m_cTEncTop.setUseRecalculateQPAccordingToLambda( m_recalculateQPAccordingToLambda );
824  m_cTEncTop.setUseStrongIntraSmoothing( m_useStrongIntraSmoothing );
825  m_cTEncTop.setActiveParameterSetsSEIEnabled ( m_activeParameterSetsSEIEnabled );
826  m_cTEncTop.setVuiParametersPresentFlag( m_vuiParametersPresentFlag );
827  m_cTEncTop.setAspectRatioIdc( m_aspectRatioIdc );
828  m_cTEncTop.setSarWidth( m_sarWidth );
829  m_cTEncTop.setSarHeight( m_sarHeight );
830  m_cTEncTop.setOverscanInfoPresentFlag( m_overscanInfoPresentFlag );
831  m_cTEncTop.setOverscanAppropriateFlag( m_overscanAppropriateFlag );
832  m_cTEncTop.setVideoSignalTypePresentFlag( m_videoSignalTypePresentFlag );
833  m_cTEncTop.setVideoFormat( m_videoFormat );
834  m_cTEncTop.setVideoFullRangeFlag( m_videoFullRangeFlag );
835  m_cTEncTop.setColourDescriptionPresentFlag( m_colourDescriptionPresentFlag );
836  m_cTEncTop.setColourPrimaries( m_colourPrimaries );
837  m_cTEncTop.setTransferCharacteristics( m_transferCharacteristics );
838  m_cTEncTop.setMatrixCoefficients( m_matrixCoefficients );
839  m_cTEncTop.setChromaLocInfoPresentFlag( m_chromaLocInfoPresentFlag );
840  m_cTEncTop.setChromaSampleLocTypeTopField( m_chromaSampleLocTypeTopField );
841  m_cTEncTop.setChromaSampleLocTypeBottomField( m_chromaSampleLocTypeBottomField );
842  m_cTEncTop.setNeutralChromaIndicationFlag( m_neutralChromaIndicationFlag );
843  m_cTEncTop.setDefaultDisplayWindow( m_defDispWinLeftOffset, m_defDispWinRightOffset, m_defDispWinTopOffset, m_defDispWinBottomOffset );
844  m_cTEncTop.setFrameFieldInfoPresentFlag( m_frameFieldInfoPresentFlag );
845  m_cTEncTop.setPocProportionalToTimingFlag( m_pocProportionalToTimingFlag );
846  m_cTEncTop.setNumTicksPocDiffOneMinus1   ( m_numTicksPocDiffOneMinus1    );
847  m_cTEncTop.setBitstreamRestrictionFlag( m_bitstreamRestrictionFlag );
848  m_cTEncTop.setTilesFixedStructureFlag( m_tilesFixedStructureFlag );
849  m_cTEncTop.setMotionVectorsOverPicBoundariesFlag( m_motionVectorsOverPicBoundariesFlag );
850  m_cTEncTop.setMinSpatialSegmentationIdc( m_minSpatialSegmentationIdc );
851  m_cTEncTop.setMaxBytesPerPicDenom( m_maxBytesPerPicDenom );
852  m_cTEncTop.setMaxBitsPerMinCuDenom( m_maxBitsPerMinCuDenom );
853  m_cTEncTop.setLog2MaxMvLengthHorizontal( m_log2MaxMvLengthHorizontal );
854  m_cTEncTop.setLog2MaxMvLengthVertical( m_log2MaxMvLengthVertical );
855}
856#endif //SVC_EXTENSION
857
858Void TAppEncTop::xCreateLib()
859{
860  // Video I/O
861#if SVC_EXTENSION
862  // initialize global variables
863  initROM();
864
865  for(UInt layer=0; layer<m_numLayers; layer++)
866  {
867#if O0194_DIFFERENT_BITDEPTH_EL_BL
868    //2
869    g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
870    g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
871
872    g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
873    g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
874#endif
875#if LAYER_CTB
876    g_uiMaxCUWidth  = g_auiLayerMaxCUWidth[layer];
877    g_uiMaxCUHeight = g_auiLayerMaxCUHeight[layer];
878    g_uiMaxCUDepth  = g_auiLayerMaxCUDepth[layer];
879    g_uiAddCUDepth  = g_auiLayerAddCUDepth[layer];
880#endif
881#if O0194_DIFFERENT_BITDEPTH_EL_BL
882    m_acTVideoIOYuvInputFile[layer].open( (Char *)m_acLayerCfg[layer].getInputFile().c_str(),  false, m_acLayerCfg[layer].m_inputBitDepthY, m_acLayerCfg[layer].m_inputBitDepthC, m_acLayerCfg[layer].m_internalBitDepthY, m_acLayerCfg[layer].m_internalBitDepthC );  // read  mode
883#else
884    m_acTVideoIOYuvInputFile[layer].open( (Char *)m_acLayerCfg[layer].getInputFile().c_str(),  false, m_inputBitDepthY, m_inputBitDepthC, m_internalBitDepthY, m_internalBitDepthC );  // read  mode
885#endif
886    m_acTVideoIOYuvInputFile[layer].skipFrames(m_FrameSkip, m_acLayerCfg[layer].getSourceWidth() - m_acLayerCfg[layer].getPad()[0], m_acLayerCfg[layer].getSourceHeight() - m_acLayerCfg[layer].getPad()[1]);
887
888    if (!m_acLayerCfg[layer].getReconFile().empty())
889    {
890#if O0194_DIFFERENT_BITDEPTH_EL_BL
891      m_acTVideoIOYuvReconFile[layer].open((Char *)m_acLayerCfg[layer].getReconFile().c_str(), true, m_acLayerCfg[layer].m_outputBitDepthY, m_acLayerCfg[layer].m_outputBitDepthC, m_acLayerCfg[layer].m_internalBitDepthY, m_acLayerCfg[layer].m_internalBitDepthC );  // write mode
892#else
893      m_acTVideoIOYuvReconFile[layer].open((Char *)m_acLayerCfg[layer].getReconFile().c_str(), true, m_outputBitDepthY, m_outputBitDepthC, m_internalBitDepthY, m_internalBitDepthC );  // write mode
894#endif
895    }
896
897    m_acTEncTop[layer].create();
898  }
899#else //SVC_EXTENSION
900  m_cTVideoIOYuvInputFile.open( m_pchInputFile,     false, m_inputBitDepthY, m_inputBitDepthC, m_internalBitDepthY, m_internalBitDepthC );  // read  mode
901  m_cTVideoIOYuvInputFile.skipFrames(m_FrameSkip, m_iSourceWidth - m_aiPad[0], m_iSourceHeight - m_aiPad[1]);
902
903  if (m_pchReconFile)
904    m_cTVideoIOYuvReconFile.open(m_pchReconFile, true, m_outputBitDepthY, m_outputBitDepthC, m_internalBitDepthY, m_internalBitDepthC);  // write mode
905
906  // Neo Decoder
907  m_cTEncTop.create();
908#endif //SVC_EXTENSION
909}
910
911Void TAppEncTop::xDestroyLib()
912{
913  // Video I/O
914#if SVC_EXTENSION
915  // destroy ROM
916  destroyROM();
917
918  for(UInt layer=0; layer<m_numLayers; layer++)
919  {
920#if LAYER_CTB
921    g_uiMaxCUWidth  = g_auiLayerMaxCUWidth[layer];
922    g_uiMaxCUHeight = g_auiLayerMaxCUHeight[layer];
923    g_uiMaxCUDepth  = g_auiLayerMaxCUDepth[layer];
924    g_uiAddCUDepth  = g_auiLayerAddCUDepth[layer];
925#endif
926
927    m_acTVideoIOYuvInputFile[layer].close();
928    m_acTVideoIOYuvReconFile[layer].close();
929
930    m_acTEncTop[layer].destroy();
931  }
932#else //SVC_EXTENSION
933  m_cTVideoIOYuvInputFile.close();
934  m_cTVideoIOYuvReconFile.close();
935
936  // Neo Decoder
937  m_cTEncTop.destroy();
938#endif //SVC_EXTENSION
939}
940
941Void TAppEncTop::xInitLib(Bool isFieldCoding)
942{
943#if SVC_EXTENSION
944  for(UInt layer=0; layer<m_numLayers; layer++)
945  {
946#if O0194_DIFFERENT_BITDEPTH_EL_BL
947    //3
948    g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
949    g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
950
951    g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
952    g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
953#endif
954#if LAYER_CTB
955    g_uiMaxCUWidth  = g_auiLayerMaxCUWidth[layer];
956    g_uiMaxCUHeight = g_auiLayerMaxCUHeight[layer];
957    g_uiMaxCUDepth  = g_auiLayerMaxCUDepth[layer];
958    g_uiAddCUDepth  = g_auiLayerAddCUDepth[layer];
959
960    memcpy( g_auiZscanToRaster, g_auiLayerZscanToRaster[layer], sizeof( g_auiZscanToRaster ) );
961    memcpy( g_auiRasterToZscan, g_auiLayerRasterToZscan[layer], sizeof( g_auiRasterToZscan ) );
962    memcpy( g_auiRasterToPelX,  g_auiLayerRasterToPelX[layer],  sizeof( g_auiRasterToPelX ) );
963    memcpy( g_auiRasterToPelY,  g_auiLayerRasterToPelY[layer],  sizeof( g_auiRasterToPelY ) );
964#endif
965    m_acTEncTop[layer].init(isFieldCoding);
966#if P0182_VPS_VUI_PS_FLAG
967    m_acTEncTop[layer].getVPS()->setSPSId(layer, m_acTEncTop[layer].getSPS()->getSPSId());
968    m_acTEncTop[layer].getVPS()->setPPSId(layer, m_acTEncTop[layer].getPPS()->getPPSId());
969#endif
970  }
971#if VPS_RENAME
972  m_acTEncTop[0].getVPS()->setMaxLayers( m_numLayers );
973#endif
974#if VPS_EXTN_OP_LAYER_SETS
975  TComVPS* vps = m_acTEncTop[0].getVPS();
976  vps->setMaxLayerId(m_numLayers - 1);    // Set max-layer ID
977
978  vps->setNumLayerSets(m_numLayers);
979  for(Int setId = 1; setId < vps->getNumLayerSets(); setId++)
980  {
981    for(Int layerId = 0; layerId <= vps->getMaxLayerId(); layerId++)
982    {
983#if O0194_DIFFERENT_BITDEPTH_EL_BL
984      //4
985      g_bitDepthY = m_acLayerCfg[layerId].m_internalBitDepthY;
986      g_bitDepthC = m_acLayerCfg[layerId].m_internalBitDepthC;
987
988      g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layerId].m_inputBitDepthY : m_acLayerCfg[layerId].m_internalBitDepthY;
989      g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layerId].m_inputBitDepthC : m_acLayerCfg[layerId].m_internalBitDepthC;
990#endif
991      if( layerId <= setId )
992      {
993        vps->setLayerIdIncludedFlag(true, setId, layerId);
994      }
995      else
996      {
997        vps->setLayerIdIncludedFlag(false, setId, layerId);
998      }
999    }
1000  }
1001#if VPS_EXTN_MASK_AND_DIM_INFO
1002  UInt i = 0, dimIdLen = 0;
1003#if AVC_BASE
1004  vps->setAvcBaseLayerFlag(m_avcBaseLayerFlag);
1005#else
1006  vps->setAvcBaseLayerFlag(false);
1007#endif
1008  vps->setSplittingFlag(false);
1009  for(i = 0; i < MAX_VPS_NUM_SCALABILITY_TYPES; i++)
1010  {
1011    vps->setScalabilityMask(i, false);
1012  }
1013  if(m_numLayers > 1)
1014  {
1015    Int scalabilityTypes = 0;
1016    for(i = 0; i < MAX_VPS_NUM_SCALABILITY_TYPES; i++)
1017    {
1018      vps->setScalabilityMask(i, m_scalabilityMask[i]);
1019      scalabilityTypes += m_scalabilityMask[i];
1020    }
1021#if AUXILIARY_PICTURES
1022    assert( scalabilityTypes <= 2 );
1023#else
1024    assert( scalabilityTypes == 1 );
1025#endif
1026    vps->setNumScalabilityTypes(scalabilityTypes);
1027  }
1028  else
1029  {
1030    vps->setNumScalabilityTypes(0);
1031  }
1032  while((1 << dimIdLen) < m_numLayers)
1033  {
1034    dimIdLen++;
1035  }
1036  vps->setDimensionIdLen(0, dimIdLen);
1037  vps->setNuhLayerIdPresentFlag(false);
1038  vps->setLayerIdInNuh(0, 0);
1039  vps->setLayerIdInVps(0, 0);
1040  for(i = 1; i < vps->getMaxLayers(); i++)
1041  {
1042    vps->setLayerIdInNuh(i, i);
1043    vps->setLayerIdInVps(vps->getLayerIdInNuh(i), i);
1044    vps->setDimensionId(i, 0, i);
1045  }
1046#if AUXILIARY_PICTURES
1047  if (m_scalabilityMask[3])
1048  {
1049    UInt maxAuxId = 0;
1050    UInt auxDimIdLen = 1;
1051    for(i = 1; i < vps->getMaxLayers(); i++)
1052    {
1053      if (m_acLayerCfg[i].getAuxId() > maxAuxId)
1054      {
1055        maxAuxId = m_acLayerCfg[i].getAuxId();
1056      }
1057    }
1058    while((1 << auxDimIdLen) < (maxAuxId + 1))
1059    {
1060      auxDimIdLen++;
1061    }
1062    vps->setDimensionIdLen(1, auxDimIdLen);
1063    for(i = 1; i < vps->getMaxLayers(); i++)
1064    {
1065      vps->setDimensionId(i, 1, m_acLayerCfg[i].getAuxId());
1066    }
1067  }
1068#endif
1069#endif
1070#if VPS_TSLAYERS
1071    vps->setMaxTSLayersPresentFlag(true);
1072    for( i = 0; i < MAX_VPS_LAYER_ID_PLUS1 - 1; i++ )
1073    {
1074        vps->setMaxTSLayersMinus1(i, vps->getMaxTLayers()-1);
1075    }
1076#endif
1077#if N0120_MAX_TID_REF_PRESENT_FLAG
1078#if N0120_MAX_TID_REF_CFG
1079  vps->setMaxTidRefPresentFlag(m_maxTidRefPresentFlag);
1080#else
1081  vps->setMaxTidRefPresentFlag(true);
1082#endif
1083  if (vps->getMaxTidRefPresentFlag())
1084  {
1085    for( i = 0; i < MAX_VPS_LAYER_ID_PLUS1 - 1; i++ )
1086    {
1087#if N0120_MAX_TID_REF_CFG
1088#if O0225_MAX_TID_FOR_REF_LAYERS
1089      for( Int j = i+1; j <= MAX_VPS_LAYER_ID_PLUS1 - 1; j++)
1090      {
1091        vps->setMaxTidIlRefPicsPlus1(i, j, m_acTEncTop[i].getMaxTidIlRefPicsPlus1());
1092      }
1093#else
1094      vps->setMaxTidIlRefPicsPlus1(i, m_acTEncTop[i].getMaxTidIlRefPicsPlus1());
1095#endif
1096#else
1097#if O0225_MAX_TID_FOR_REF_LAYERS
1098      for( Int j = i+1; j <= MAX_VPS_LAYER_ID_PLUS1 - 1; j++)
1099      {
1100        vps->setMaxTidIlRefPicsPlus1(i, j, vps->getMaxTLayers()+1);
1101      }
1102#else
1103      vps->setMaxTidIlRefPicsPlus1(i, vps->getMaxTLayers()+1);
1104#endif
1105#endif
1106    }
1107  }
1108  else
1109  {
1110    for( i = 0; i < MAX_VPS_LAYER_ID_PLUS1 - 1; i++ )
1111    {
1112#if O0225_MAX_TID_FOR_REF_LAYERS
1113      for( Int j = i+1; j <= MAX_VPS_LAYER_ID_PLUS1 - 1; j++)
1114      {
1115        vps->setMaxTidIlRefPicsPlus1(i, j, 7);
1116      }
1117#else
1118      vps->setMaxTidIlRefPicsPlus1(i, 7);
1119#endif
1120    }
1121  }
1122#endif
1123#if ILP_SSH_SIG
1124#if ILP_SSH_SIG_FIX
1125    vps->setIlpSshSignalingEnabledFlag(false);
1126#else
1127    vps->setIlpSshSignalingEnabledFlag(true);
1128#endif
1129#endif
1130#if VPS_EXTN_PROFILE_INFO
1131  vps->getPTLForExtnPtr()->resize(vps->getNumLayerSets());
1132  for(Int setId = 1; setId < vps->getNumLayerSets(); setId++)
1133  {
1134    vps->setProfilePresentFlag(setId, true);
1135    // Note - may need to be changed for other layer structures.
1136    *(vps->getPTLForExtn(setId)) = *(m_acTEncTop[setId].getSPS()->getPTL());
1137  }
1138#endif
1139  // Target output layer
1140  vps->setNumOutputLayerSets(vps->getNumLayerSets());
1141  vps->setNumProfileTierLevel(vps->getNumLayerSets());
1142#if P0295_DEFAULT_OUT_LAYER_IDC
1143  vps->setDefaultTargetOutputLayerIdc(1);
1144#else
1145#if O0109_DEFAULT_ONE_OUT_LAYER_IDC
1146  vps->setDefaultOneTargetOutputLayerIdc(1);
1147#else
1148  vps->setDefaultOneTargetOutputLayerFlag(true);
1149#endif
1150#endif
1151  for(i = 1; i < vps->getNumLayerSets(); i++)
1152  {
1153    vps->setProfileLevelTierIdx(i, i);
1154    vps->setOutputLayerSetIdx(i, i);
1155  } 
1156#endif
1157 #if VPS_DPB_SIZE_TABLE
1158  // The Layer ID List variables can be derived here. 
1159#if DERIVE_LAYER_ID_LIST_VARIABLES
1160  vps->deriveLayerIdListVariables();
1161#endif
1162#if RESOLUTION_BASED_DPB
1163  vps->assignSubDpbIndices();
1164#else
1165  vps->deriveNumberOfSubDpbs();
1166#endif
1167
1168  // derive OutputLayerFlag[i][j]
1169  if( vps->getDefaultTargetOutputLayerIdc() == 1 )
1170  {
1171    // default_target_output_layer_idc equal to 1 specifies that only the layer with the highest value of nuh_layer_id such that nuh_layer_id equal to nuhLayerIdA and
1172    // AuxId[ nuhLayerIdA ] equal to 0 in each of the output layer sets with index in the range of 1 to vps_num_layer_sets_minus1, inclusive, is an output layer of its output layer set.
1173
1174    // Include the highest layer as output layer for each layer set
1175    for(Int lsIdx = 1; lsIdx < vps->getNumLayerSets(); lsIdx++)
1176    {
1177      for( UInt layer = 0; layer < vps->getNumLayersInIdList(lsIdx); layer++ )
1178      {
1179        if( vps->getLayerIdIncludedFlag(lsIdx, layer) )     
1180        {       
1181          vps->setOutputLayerFlag( lsIdx, layer, layer == vps->getNumLayersInIdList(lsIdx) - 1 );
1182        }
1183      }
1184    }
1185  }
1186  else
1187  {
1188    // cases when default_target_output_layer_idc is not equal to 1
1189    assert(!"default_target_output_layer_idc not equal to 1 is not yet supported");
1190  }
1191
1192  // Initialize dpb_size_table() for all ouput layer sets in the VPS extension
1193  for(i = 1; i < vps->getNumOutputLayerSets(); i++)
1194  {
1195#if CHANGE_NUMSUBDPB_IDX
1196    Int layerSetIdxForOutputLayerSet = vps->getOutputLayerSetIdx( i );
1197#endif
1198    Int layerSetId = vps->getOutputLayerSetIdx(i);
1199
1200    for(Int j = 0; j < vps->getMaxTLayers(); j++)
1201    {
1202
1203      Int maxNumReorderPics = -1;
1204#if CHANGE_NUMSUBDPB_IDX
1205#if RESOLUTION_BASED_DPB
1206      for(Int k = 0; k < vps->getNumLayersInIdList(layerSetIdxForOutputLayerSet); k++)
1207#else
1208      for(Int k = 0; k < vps->getNumSubDpbs(layerSetIdxForOutputLayerSet); k++)
1209#endif
1210#else
1211      for(Int k = 0; k < vps->getNumSubDpbs(i); k++)
1212#endif
1213      {
1214        Int layerId = vps->getLayerSetLayerIdList(layerSetId, k); // k-th layer in the output layer set
1215#if RESOLUTION_BASED_DPB
1216        vps->setMaxVpsLayerDecPicBuffMinus1( i, k, j, m_acTEncTop[layerId].getMaxDecPicBuffering(j) - 1 );
1217        // Add sub-DPB sizes of layers belonging to a sub-DPB. If a different sub-DPB size is calculated
1218        // at the encoder, modify below
1219        Int oldValue = vps->getMaxVpsDecPicBufferingMinus1( i, vps->getSubDpbAssigned( layerSetIdxForOutputLayerSet, k ), j );
1220        oldValue += vps->getMaxVpsLayerDecPicBuffMinus1( i, k, j ) + 1;
1221        vps->setMaxVpsDecPicBufferingMinus1( i, vps->getSubDpbAssigned( layerSetIdxForOutputLayerSet, k ), j, oldValue );
1222#else
1223        vps->setMaxVpsDecPicBufferingMinus1( i, k, j,  m_acTEncTop[layerId].getMaxDecPicBuffering(j) - 1 );
1224#endif
1225        maxNumReorderPics       = std::max( maxNumReorderPics, m_acTEncTop[layerId].getNumReorderPics(j));
1226      }
1227#if RESOLUTION_BASED_DPB
1228      for(Int k = 0; k < vps->getNumSubDpbs(i); k++)
1229      {
1230        // Decrement m_maxVpsDecPicBufferingMinus1
1231        Int oldValue = vps->getMaxVpsDecPicBufferingMinus1( i, vps->getSubDpbAssigned( layerSetIdxForOutputLayerSet, k ), j );
1232        vps->setMaxVpsDecPicBufferingMinus1( i, vps->getSubDpbAssigned( layerSetIdxForOutputLayerSet, k ), j, oldValue - 1 );
1233      }
1234#endif
1235      vps->setMaxVpsNumReorderPics(i, j, maxNumReorderPics);
1236      vps->determineSubDpbInfoFlags();
1237    }
1238  }
1239#endif
1240#if VPS_EXTN_DIRECT_REF_LAYERS
1241  // Direct reference layers
1242  UInt maxDirectRefLayers = 0;
1243#if O0096_DEFAULT_DEPENDENCY_TYPE
1244  Bool isDefaultDirectDependencyTypeSet = false;
1245#endif
1246  for(UInt layerCtr = 1;layerCtr <= vps->getMaxLayers() - 1; layerCtr++)
1247  {
1248    vps->setNumDirectRefLayers( layerCtr, m_acTEncTop[layerCtr].getNumDirectRefLayers() );
1249    maxDirectRefLayers = max<UInt>( maxDirectRefLayers, vps->getNumDirectRefLayers( layerCtr ) );
1250
1251    for(i = 0; i < vps->getNumDirectRefLayers(layerCtr); i++)
1252    {
1253      vps->setRefLayerId( layerCtr, i, m_acTEncTop[layerCtr].getRefLayerId(i) );
1254    }
1255    // Set direct dependency flag
1256    // Initialize flag to 0
1257    for(Int refLayerCtr = 0; refLayerCtr < layerCtr; refLayerCtr++)
1258    {
1259      vps->setDirectDependencyFlag( layerCtr, refLayerCtr, false );
1260    }
1261    for(i = 0; i < vps->getNumDirectRefLayers(layerCtr); i++)
1262    {
1263      vps->setDirectDependencyFlag( layerCtr, vps->getLayerIdInVps(m_acTEncTop[layerCtr].getRefLayerId(i)), true);
1264    }
1265    // prediction indications
1266    vps->setDirectDepTypeLen(2); // sample and motion types are encoded
1267    for(Int refLayerCtr = 0; refLayerCtr < layerCtr; refLayerCtr++)
1268    {
1269      if (vps->getDirectDependencyFlag( layerCtr, refLayerCtr))
1270      {
1271        assert(m_acTEncTop[layerCtr].getSamplePredEnabledFlag(refLayerCtr) || m_acTEncTop[layerCtr].getMotionPredEnabledFlag(refLayerCtr));
1272        vps->setDirectDependencyType( layerCtr, refLayerCtr, ((m_acTEncTop[layerCtr].getSamplePredEnabledFlag(refLayerCtr) ? 1 : 0) |
1273                                                              (m_acTEncTop[layerCtr].getMotionPredEnabledFlag(refLayerCtr) ? 2 : 0)) - 1);
1274#if O0096_DEFAULT_DEPENDENCY_TYPE
1275        if (!isDefaultDirectDependencyTypeSet)
1276        {
1277          vps->setDefaultDirectDependecyTypeFlag(1);
1278          vps->setDefaultDirectDependecyType(vps->getDirectDependencyType(layerCtr, refLayerCtr));
1279          isDefaultDirectDependencyTypeSet = true;
1280        }
1281        else if (vps->getDirectDependencyType(layerCtr, refLayerCtr) != vps->getDefaultDirectDependencyType())
1282        {
1283          vps->setDefaultDirectDependecyTypeFlag(0);
1284        }
1285#endif
1286      }
1287      else
1288      {
1289        vps->setDirectDependencyType( layerCtr, refLayerCtr, 0 );
1290      }
1291    }
1292  }
1293
1294#if O0092_0094_DEPENDENCY_CONSTRAINT
1295  for(UInt layerCtr = 1;layerCtr <= vps->getMaxLayers() - 1; layerCtr++)
1296  {
1297    vps->setNumRefLayers(vps->getLayerIdInNuh(layerCtr));   // identify the number of direct and indirect reference layers of current layer and set recursiveRefLayersFlags
1298  }
1299  if(vps->getMaxLayers() > MAX_REF_LAYERS)
1300  {
1301    for(UInt layerCtr = 1;layerCtr <= vps->getMaxLayers() - 1; layerCtr++)
1302    {
1303      assert( vps->getNumRefLayers(vps->getLayerIdInNuh(layerCtr)) <= MAX_REF_LAYERS);
1304    }
1305  }
1306#endif
1307#endif
1308    vps->setMaxOneActiveRefLayerFlag(maxDirectRefLayers > 1 ? false : true);
1309#if O0062_POC_LSB_NOT_PRESENT_FLAG
1310    for(i = 1; i< vps->getMaxLayers(); i++)
1311    {
1312      if( vps->getNumDirectRefLayers( vps->getLayerIdInNuh(i) ) == 0  )
1313      {
1314        vps->setPocLsbNotPresentFlag(i, false);
1315      }
1316    }
1317#endif
1318#if O0223_PICTURE_TYPES_ALIGN_FLAG
1319    vps->setCrossLayerPictureTypeAlignFlag( m_crossLayerPictureTypeAlignFlag );
1320#endif
1321#if P0068_CROSS_LAYER_ALIGNED_IDR_ONLY_FOR_IRAP_FLAG
1322    vps->setCrossLayerAlignedIdrOnlyFlag( m_crossLayerAlignedIdrOnlyFlag );
1323#endif
1324#if N0147_IRAP_ALIGN_FLAG
1325    vps->setCrossLayerIrapAlignFlag( m_crossLayerIrapAlignFlag );
1326    for(UInt layerCtr = 1;layerCtr <= vps->getMaxLayers() - 1; layerCtr++)
1327    {
1328      for(Int refLayerCtr = 0; refLayerCtr < layerCtr; refLayerCtr++)
1329      {
1330        if (vps->getDirectDependencyFlag( layerCtr, refLayerCtr))
1331        {
1332          if(m_acTEncTop[layerCtr].getIntraPeriod() !=  m_acTEncTop[refLayerCtr].getIntraPeriod())
1333          {
1334            vps->setCrossLayerIrapAlignFlag(false);
1335            break;
1336          }
1337        }
1338      }
1339    }
1340#endif
1341#if M0040_ADAPTIVE_RESOLUTION_CHANGE
1342  vps->setSingleLayerForNonIrapFlag(m_adaptiveResolutionChange > 0 ? true : false);
1343#endif
1344#if HIGHER_LAYER_IRAP_SKIP_FLAG
1345  vps->setHigherLayerIrapSkipFlag(m_skipPictureAtArcSwitch);
1346#endif
1347#if !P0125_REVERT_VPS_EXTN_OFFSET_TO_RESERVED
1348#if !VPS_EXTN_OFFSET_CALC
1349#if VPS_EXTN_OFFSET
1350  // to be updated according to the current semantics
1351  vps->setExtensionOffset( 0xffff );
1352#endif
1353#endif
1354#endif
1355
1356#if O0215_PHASE_ALIGNMENT
1357  vps->setPhaseAlignFlag( m_phaseAlignFlag );
1358#endif
1359
1360#if P0300_ALT_OUTPUT_LAYER_FLAG
1361  for (Int k = 0; k < MAX_VPS_LAYER_SETS_PLUS1; k++)
1362  {
1363    vps->setAltOuputLayerFlag( k, m_altOutputLayerFlag );
1364  }
1365#else
1366#if O0153_ALT_OUTPUT_LAYER_FLAG
1367  vps->setAltOuputLayerFlag( m_altOutputLayerFlag );
1368#endif
1369#endif
1370
1371#if P0312_VERT_PHASE_ADJ
1372  Bool vpsVuiVertPhaseInUseFlag = false;
1373  for( UInt layerId = 1; layerId < m_numLayers; layerId++ )
1374  {
1375    for( i = 0; i < m_acLayerCfg[layerId].m_numScaledRefLayerOffsets; i++ )
1376    {
1377      if( m_acTEncTop[layerId].getVertPhasePositionEnableFlag(i) )
1378      {
1379        vpsVuiVertPhaseInUseFlag = true;
1380        break;
1381      }
1382    }
1383  }
1384  vps->setVpsVuiVertPhaseInUseFlag( vpsVuiVertPhaseInUseFlag );
1385#endif
1386
1387#if O0164_MULTI_LAYER_HRD
1388  vps->setVpsVuiBspHrdPresentFlag(false);
1389  TEncTop *pcCfg = &m_acTEncTop[0];
1390  if( pcCfg->getBufferingPeriodSEIEnabled() )
1391  {
1392    vps->setVpsVuiBspHrdPresentFlag(true);
1393    vps->setVpsNumBspHrdParametersMinus1(vps->getNumLayerSets() - 2); 
1394    vps->createBspHrdParamBuffer(vps->getVpsNumBspHrdParametersMinus1() + 1);
1395    for ( i = 0; i <= vps->getVpsNumBspHrdParametersMinus1(); i++ )
1396    {
1397      vps->setBspCprmsPresentFlag(i, true);
1398
1399      UInt layerId = i + 1;
1400      TEncTop *pcCfgLayer = &m_acTEncTop[layerId];
1401
1402      Int iPicWidth         = pcCfgLayer->getSourceWidth();
1403      Int iPicHeight        = pcCfgLayer->getSourceHeight();
1404      UInt uiWidthInCU       = ( iPicWidth %m_uiMaxCUWidth  ) ? iPicWidth /m_uiMaxCUWidth  + 1 : iPicWidth /m_uiMaxCUWidth;
1405      UInt uiHeightInCU      = ( iPicHeight%m_uiMaxCUHeight ) ? iPicHeight/m_uiMaxCUHeight + 1 : iPicHeight/m_uiMaxCUHeight;
1406      UInt uiNumCUsInFrame   = uiWidthInCU * uiHeightInCU;
1407
1408      UInt maxCU = pcCfgLayer->getSliceArgument() >> ( m_uiMaxCUDepth << 1);
1409      UInt numDU = ( pcCfgLayer->getSliceMode() == 1 ) ? ( uiNumCUsInFrame / maxCU ) : ( 0 );
1410      if( uiNumCUsInFrame % maxCU != 0 || numDU == 0 )
1411      {
1412        numDU ++;
1413      }
1414      vps->getBspHrd(i)->setNumDU( numDU );
1415      vps->setBspHrdParameters( i, pcCfgLayer->getFrameRate(), numDU, pcCfgLayer->getTargetBitrate(), ( pcCfgLayer->getIntraPeriod() > 0 ) );
1416    }
1417    for(UInt h = 1; h <= (vps->getNumLayerSets()-1); h++)
1418    {
1419      vps->setNumBitstreamPartitions(h, 1);
1420      for( i = 0; i < vps->getNumBitstreamPartitions(h); i++ )
1421      {
1422        for( UInt j = 0; j <= (vps->getMaxLayers()-1); j++ )
1423        {
1424          if (vps->getLayerIdIncludedFlag(h, j) && h == j)
1425          {
1426            vps->setLayerInBspFlag(h, i, j, true);
1427          }
1428        }
1429      }
1430      vps->setNumBspSchedCombinations(h, 1);
1431      for( i = 0; i < vps->getNumBspSchedCombinations(h); i++ )
1432      {
1433        for( UInt j = 0; j < vps->getNumBitstreamPartitions(h); j++ )
1434        {
1435          vps->setBspCombHrdIdx(h, i, j, 0);
1436          vps->setBspCombSchedIdx(h, i, j, 0);
1437        }
1438      }
1439    }
1440  }
1441#endif
1442
1443#else //SVC_EXTENSION
1444  m_cTEncTop.init(isFieldCoding);
1445#endif //SVC_EXTENSION
1446}
1447
1448// ====================================================================================================================
1449// Public member functions
1450// ====================================================================================================================
1451
1452/**
1453 - create internal class
1454 - initialize internal variable
1455 - until the end of input YUV file, call encoding function in TEncTop class
1456 - delete allocated buffers
1457 - destroy internal class
1458 .
1459 */
1460#if SVC_EXTENSION
1461Void TAppEncTop::encode()
1462{
1463  fstream bitstreamFile(m_pBitstreamFile, fstream::binary | fstream::out);
1464  if (!bitstreamFile)
1465  {
1466    fprintf(stderr, "\nfailed to open bitstream file `%s' for writing\n", m_pBitstreamFile);
1467    exit(EXIT_FAILURE);
1468  }
1469
1470  TComPicYuv*       pcPicYuvOrg [MAX_LAYERS];
1471  TComPicYuv*       pcPicYuvRec = NULL;
1472
1473  // initialize internal class & member variables
1474  xInitLibCfg();
1475  xCreateLib();
1476  xInitLib(m_isField);
1477
1478  // main encoder loop
1479  Int   iNumEncoded = 0, iTotalNumEncoded = 0;
1480  Bool  bEos = false;
1481
1482  list<AccessUnit> outputAccessUnits; ///< list of access units to write out.  is populated by the encoding process
1483
1484  for(UInt layer=0; layer<m_numLayers; layer++)
1485  {
1486#if O0194_DIFFERENT_BITDEPTH_EL_BL
1487    //5
1488    g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
1489    g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
1490
1491    g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
1492    g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
1493#endif
1494    // allocate original YUV buffer
1495    pcPicYuvOrg[layer] = new TComPicYuv;
1496    if( m_isField )
1497    {
1498#if SVC_UPSAMPLING
1499#if LAYER_CTB
1500#if AUXILIARY_PICTURES
1501      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeightOrg(), m_acLayerCfg[layer].getChromaFormatIDC(), m_acLayerCfg[layer].m_uiMaxCUWidth, m_acLayerCfg[layer].m_uiMaxCUHeight, m_acLayerCfg[layer].m_uiMaxCUDepth, NULL );
1502#else
1503      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeightOrg(), m_acLayerCfg[layer].m_uiMaxCUWidth, m_acLayerCfg[layer].m_uiMaxCUHeight, m_acLayerCfg[layer].m_uiMaxCUDepth, NULL );
1504#endif
1505#else
1506#if AUXILIARY_PICTURES
1507      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeightOrg(), m_acLayerCfg[layer].getChromaFormatIDC(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth, NULL );
1508#else
1509      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeightOrg(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth, NULL );
1510#endif
1511#endif
1512#else
1513      pcPicYuvOrg->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeightOrg(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1514#endif
1515    }
1516    else
1517    {
1518#if SVC_UPSAMPLING
1519#if LAYER_CTB
1520#if AUXILIARY_PICTURES
1521      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_acLayerCfg[layer].getChromaFormatIDC(), m_acLayerCfg[layer].m_uiMaxCUWidth, m_acLayerCfg[layer].m_uiMaxCUHeight, m_acLayerCfg[layer].m_uiMaxCUDepth, NULL );
1522#else
1523      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_acLayerCfg[layer].m_uiMaxCUWidth, m_acLayerCfg[layer].m_uiMaxCUHeight, m_acLayerCfg[layer].m_uiMaxCUDepth, NULL );
1524#endif
1525#else
1526#if AUXILIARY_PICTURES
1527      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_acLayerCfg[layer].getChromaFormatIDC(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth, NULL );
1528#else
1529      pcPicYuvOrg[layer]->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth, NULL );
1530#endif
1531#endif
1532#else
1533      pcPicYuvOrg->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1534#endif
1535    }
1536  }
1537
1538#if AVC_SYNTAX
1539  fstream streamSyntaxFile;
1540  if( m_acTEncTop[0].getVPS()->getAvcBaseLayerFlag() )
1541  {
1542    if( !m_BLSyntaxFile )
1543    {
1544      printf( "Wrong base layer syntax input file\n" );
1545      exit(EXIT_FAILURE);
1546    }
1547    streamSyntaxFile.open( m_BLSyntaxFile, fstream::in | fstream::binary );
1548    if( !streamSyntaxFile.good() )
1549    {
1550      printf( "Base layer syntax input reading error\n" );
1551      exit(EXIT_FAILURE);
1552    }
1553    m_acTEncTop[0].setBLSyntaxFile( &streamSyntaxFile );
1554  }
1555#endif
1556
1557  Bool bFirstFrame = true;
1558  while ( !bEos )
1559  {
1560    // Read enough frames
1561    Bool bFramesReadyToCode = false;
1562    while(!bFramesReadyToCode)
1563    {
1564      for(UInt layer=0; layer<m_numLayers; layer++)
1565      {
1566#if O0194_DIFFERENT_BITDEPTH_EL_BL
1567        //6
1568        g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
1569        g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
1570
1571        g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
1572        g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
1573#endif
1574#if LAYER_CTB
1575        g_uiMaxCUWidth  = g_auiLayerMaxCUWidth[layer];
1576        g_uiMaxCUHeight = g_auiLayerMaxCUHeight[layer];
1577        g_uiMaxCUDepth  = g_auiLayerMaxCUDepth[layer];
1578        g_uiAddCUDepth  = g_auiLayerAddCUDepth[layer];
1579#endif
1580
1581        // get buffers
1582        xGetBuffer(pcPicYuvRec, layer);
1583
1584        // read input YUV file
1585        m_acTVideoIOYuvInputFile[layer].read( pcPicYuvOrg[layer], m_acLayerCfg[layer].getPad() );
1586
1587#if AUXILIARY_PICTURES
1588        if (m_acLayerCfg[layer].getChromaFormatIDC() == CHROMA_400)
1589        {
1590          pcPicYuvOrg[layer]->convertToMonochrome();
1591        }
1592#endif
1593
1594        if(layer == m_numLayers-1)
1595        {
1596          // increase number of received frames
1597          m_iFrameRcvd++;
1598          // check end of file
1599          bEos = (m_isField && (m_iFrameRcvd == (m_framesToBeEncoded >> 1) )) || ( !m_isField && (m_iFrameRcvd == m_framesToBeEncoded) );
1600        }
1601
1602        if ( m_isField )
1603        {
1604          m_acTEncTop[layer].encodePrep( pcPicYuvOrg[layer], m_isTopFieldFirst );
1605        }
1606        else
1607        {
1608          m_acTEncTop[layer].encodePrep( pcPicYuvOrg[layer] );
1609        }
1610      }
1611
1612      bFramesReadyToCode = !(!bFirstFrame && ( m_acTEncTop[m_numLayers-1].getNumPicRcvd() != m_iGOPSize && m_iGOPSize ) && !bEos );
1613    }
1614    Bool flush = 0;
1615    // if end of file (which is only detected on a read failure) flush the encoder of any queued pictures
1616    if (m_acTVideoIOYuvInputFile[m_numLayers-1].isEof())
1617    {
1618      flush = true;
1619      bEos = true;
1620      m_iFrameRcvd--;
1621      m_acTEncTop[m_numLayers-1].setFramesToBeEncoded(m_iFrameRcvd);
1622    }
1623
1624#if RC_SHVC_HARMONIZATION
1625    for(UInt layer=0; layer<m_numLayers; layer++)
1626    {
1627      if ( m_acTEncTop[layer].getUseRateCtrl() )
1628      {
1629        (m_acTEncTop[layer].getRateCtrl())->initRCGOP(m_acTEncTop[layer].getNumPicRcvd());
1630      }
1631    }
1632#endif
1633
1634#if M0040_ADAPTIVE_RESOLUTION_CHANGE
1635    if (m_adaptiveResolutionChange)
1636    {
1637      for(UInt layer = 0; layer < m_numLayers; layer++)
1638      {
1639        TComList<TComPicYuv*>::iterator iterPicYuvRec;
1640        for (iterPicYuvRec = m_acListPicYuvRec[layer].begin(); iterPicYuvRec != m_acListPicYuvRec[layer].end(); iterPicYuvRec++)
1641        {
1642          TComPicYuv* recPic = *(iterPicYuvRec);
1643          recPic->setReconstructed(false);
1644        }
1645      }
1646    }
1647#endif
1648
1649    // loop through frames in one GOP
1650    for ( UInt iPicIdInGOP=0; iPicIdInGOP < (bFirstFrame? 1:m_iGOPSize); iPicIdInGOP++ )
1651    {
1652      // layer by layer for each frame
1653      for(UInt layer=0; layer<m_numLayers; layer++)
1654      {
1655#if O0194_DIFFERENT_BITDEPTH_EL_BL
1656        //7
1657        g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
1658        g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
1659
1660        g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
1661        g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
1662#endif
1663#if LAYER_CTB
1664        g_uiMaxCUWidth  = g_auiLayerMaxCUWidth[layer];
1665        g_uiMaxCUHeight = g_auiLayerMaxCUHeight[layer];
1666        g_uiMaxCUDepth  = g_auiLayerMaxCUDepth[layer];
1667        g_uiAddCUDepth  = g_auiLayerAddCUDepth[layer];
1668
1669        memcpy( g_auiZscanToRaster, g_auiLayerZscanToRaster[layer], sizeof( g_auiZscanToRaster ) );
1670        memcpy( g_auiRasterToZscan, g_auiLayerRasterToZscan[layer], sizeof( g_auiRasterToZscan ) );
1671        memcpy( g_auiRasterToPelX,  g_auiLayerRasterToPelX[layer],  sizeof( g_auiRasterToPelX ) );
1672        memcpy( g_auiRasterToPelY,  g_auiLayerRasterToPelY[layer],  sizeof( g_auiRasterToPelY ) );
1673#endif
1674        // call encoding function for one frame
1675        if ( m_isField )
1676        {
1677          m_acTEncTop[layer].encode( flush ? 0 : pcPicYuvOrg[layer], m_acListPicYuvRec[layer], outputAccessUnits, iPicIdInGOP, m_isTopFieldFirst );
1678        }
1679        else
1680        {
1681          m_acTEncTop[layer].encode( flush ? 0 : pcPicYuvOrg[layer], m_acListPicYuvRec[layer], outputAccessUnits, iPicIdInGOP );
1682        }
1683      }
1684    }
1685
1686#if RC_SHVC_HARMONIZATION
1687    for(UInt layer=0; layer<m_numLayers; layer++)
1688    {
1689      if ( m_acTEncTop[layer].getUseRateCtrl() )
1690      {
1691        (m_acTEncTop[layer].getRateCtrl())->destroyRCGOP();
1692      }
1693    }
1694#endif
1695
1696    iTotalNumEncoded = 0;
1697    for(UInt layer=0; layer<m_numLayers; layer++)
1698    {
1699#if O0194_DIFFERENT_BITDEPTH_EL_BL
1700      //8
1701      g_bitDepthY = m_acLayerCfg[layer].m_internalBitDepthY;
1702      g_bitDepthC = m_acLayerCfg[layer].m_internalBitDepthC;
1703
1704      g_uiPCMBitDepthLuma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthY : m_acLayerCfg[layer].m_internalBitDepthY;
1705      g_uiPCMBitDepthChroma = m_bPCMInputBitDepthFlag ? m_acLayerCfg[layer].m_inputBitDepthC : m_acLayerCfg[layer].m_internalBitDepthC;
1706#endif
1707      // write bistream to file if necessary
1708      iNumEncoded = m_acTEncTop[layer].getNumPicRcvd();
1709      if ( iNumEncoded > 0 )
1710      {
1711        xWriteRecon(layer, iNumEncoded);
1712        iTotalNumEncoded += iNumEncoded;
1713      }
1714      m_acTEncTop[layer].setNumPicRcvd( 0 );
1715    }
1716
1717    // write bitstream out
1718    if(iTotalNumEncoded)
1719    {
1720#if P0130_EOB
1721      if( bEos )
1722      {
1723        OutputNALUnit nalu(NAL_UNIT_EOB);
1724        nalu.m_layerId = 0;
1725
1726        AccessUnit& accessUnit = outputAccessUnits.back();
1727        nalu.m_temporalId = accessUnit.front()->m_temporalId;
1728        accessUnit.push_back(new NALUnitEBSP(nalu));
1729      }
1730#endif
1731      xWriteStream(bitstreamFile, iTotalNumEncoded, outputAccessUnits);
1732      outputAccessUnits.clear();
1733    }
1734
1735    // print out summary
1736    if (bEos)
1737    {
1738      printOutSummary(m_isTopFieldFirst);
1739    }
1740
1741    bFirstFrame = false;
1742  }
1743  // delete original YUV buffer
1744  for(UInt layer=0; layer<m_numLayers; layer++)
1745  {
1746    pcPicYuvOrg[layer]->destroy();
1747    delete pcPicYuvOrg[layer];
1748    pcPicYuvOrg[layer] = NULL;
1749
1750    // delete used buffers in encoder class
1751    m_acTEncTop[layer].deletePicBuffer();
1752  }
1753
1754#if AVC_SYNTAX
1755  if( streamSyntaxFile.is_open() )
1756  {
1757    streamSyntaxFile.close();
1758  }
1759#endif
1760
1761  // delete buffers & classes
1762  xDeleteBuffer();
1763  xDestroyLib();
1764
1765  printRateSummary();
1766
1767  return;
1768}
1769
1770Void TAppEncTop::printOutSummary(Bool isField)
1771{
1772  UInt layer;
1773
1774  // set frame rate
1775  for(layer = 0; layer < m_numLayers; layer++)
1776  {
1777    if(isField)
1778    {
1779      m_gcAnalyzeAll[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() * 2);
1780      m_gcAnalyzeI[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() * 2 );
1781      m_gcAnalyzeP[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() * 2 );
1782      m_gcAnalyzeB[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() * 2 );
1783    }
1784    else
1785    {
1786      m_gcAnalyzeAll[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate());
1787      m_gcAnalyzeI[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() );
1788      m_gcAnalyzeP[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() );
1789      m_gcAnalyzeB[layer].setFrmRate( m_acLayerCfg[layer].getFrameRate() );
1790    }
1791  }
1792
1793  //-- all
1794  printf( "\n\nSUMMARY --------------------------------------------------------\n" );
1795  printf( "\tTotal Frames |  "   "Bitrate    "  "Y-PSNR    "  "U-PSNR    "  "V-PSNR \n" );
1796  for(layer = 0; layer < m_numLayers; layer++)
1797  {
1798    m_gcAnalyzeAll[layer].printOut('a', layer);
1799  }
1800
1801  printf( "\n\nI Slices--------------------------------------------------------\n" );
1802  printf( "\tTotal Frames |  "   "Bitrate    "  "Y-PSNR    "  "U-PSNR    "  "V-PSNR \n" );
1803  for(layer = 0; layer < m_numLayers; layer++)
1804  {
1805    m_gcAnalyzeI[layer].printOut('i', layer);
1806  }
1807
1808  printf( "\n\nP Slices--------------------------------------------------------\n" );
1809  printf( "\tTotal Frames |  "   "Bitrate    "  "Y-PSNR    "  "U-PSNR    "  "V-PSNR \n" );
1810  for(layer = 0; layer < m_numLayers; layer++)
1811  {
1812    m_gcAnalyzeP[layer].printOut('p', layer);
1813  }
1814
1815  printf( "\n\nB Slices--------------------------------------------------------\n" );
1816  printf( "\tTotal Frames |  "   "Bitrate    "  "Y-PSNR    "  "U-PSNR    "  "V-PSNR \n" );
1817  for(layer = 0; layer < m_numLayers; layer++)
1818  {
1819    m_gcAnalyzeB[layer].printOut('b', layer);
1820  }
1821
1822  if(isField)
1823  {
1824    for(layer = 0; layer < m_numLayers; layer++)
1825    {
1826      //-- interlaced summary
1827      m_gcAnalyzeAll_in.setFrmRate( m_acLayerCfg[layer].getFrameRate());
1828      printf( "\n\nSUMMARY INTERLACED ---------------------------------------------\n" );
1829      m_gcAnalyzeAll_in.printOutInterlaced('a',  m_gcAnalyzeAll[layer].getBits());
1830
1831#if _SUMMARY_OUT_
1832      m_gcAnalyzeAll_in.printSummaryOutInterlaced();
1833#endif
1834    }
1835  }
1836}
1837
1838#else
1839Void TAppEncTop::encode()
1840{
1841  fstream bitstreamFile(m_pchBitstreamFile, fstream::binary | fstream::out);
1842  if (!bitstreamFile)
1843  {
1844    fprintf(stderr, "\nfailed to open bitstream file `%s' for writing\n", m_pchBitstreamFile);
1845    exit(EXIT_FAILURE);
1846  }
1847
1848  TComPicYuv*       pcPicYuvOrg = new TComPicYuv;
1849  TComPicYuv*       pcPicYuvRec = NULL;
1850
1851  // initialize internal class & member variables
1852  xInitLibCfg();
1853  xCreateLib();
1854  xInitLib(m_isField);
1855
1856  // main encoder loop
1857  Int   iNumEncoded = 0;
1858  Bool  bEos = false;
1859
1860  list<AccessUnit> outputAccessUnits; ///< list of access units to write out.  is populated by the encoding process
1861
1862  // allocate original YUV buffer
1863  if( m_isField )
1864  {
1865    pcPicYuvOrg->create( m_iSourceWidth, m_iSourceHeightOrg, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1866  }
1867  else
1868  {
1869    pcPicYuvOrg->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1870  }
1871
1872  while ( !bEos )
1873  {
1874    // get buffers
1875    xGetBuffer(pcPicYuvRec);
1876
1877    // read input YUV file
1878    m_cTVideoIOYuvInputFile.read( pcPicYuvOrg, m_aiPad );
1879
1880    // increase number of received frames
1881    m_iFrameRcvd++;
1882
1883    bEos = (m_isField && (m_iFrameRcvd == (m_framesToBeEncoded >> 1) )) || ( !m_isField && (m_iFrameRcvd == m_framesToBeEncoded) );
1884    Bool flush = 0;
1885    // if end of file (which is only detected on a read failure) flush the encoder of any queued pictures
1886    if (m_cTVideoIOYuvInputFile.isEof())
1887    {
1888      flush = true;
1889      bEos = true;
1890      m_iFrameRcvd--;
1891      m_cTEncTop.setFramesToBeEncoded(m_iFrameRcvd);
1892    }
1893
1894    // call encoding function for one frame
1895    if ( m_isField )
1896    {
1897      m_cTEncTop.encode( bEos, flush ? 0 : pcPicYuvOrg, m_cListPicYuvRec, outputAccessUnits, iNumEncoded, m_isTopFieldFirst);
1898    }
1899    else
1900    {
1901    m_cTEncTop.encode( bEos, flush ? 0 : pcPicYuvOrg, m_cListPicYuvRec, outputAccessUnits, iNumEncoded );
1902    }
1903
1904    // write bistream to file if necessary
1905    if ( iNumEncoded > 0 )
1906    {
1907      xWriteOutput(bitstreamFile, iNumEncoded, outputAccessUnits);
1908      outputAccessUnits.clear();
1909    }
1910  }
1911
1912  m_cTEncTop.printSummary(m_isField);
1913
1914  // delete original YUV buffer
1915  pcPicYuvOrg->destroy();
1916  delete pcPicYuvOrg;
1917  pcPicYuvOrg = NULL;
1918
1919  // delete used buffers in encoder class
1920  m_cTEncTop.deletePicBuffer();
1921
1922  // delete buffers & classes
1923  xDeleteBuffer();
1924  xDestroyLib();
1925
1926  printRateSummary();
1927
1928  return;
1929}
1930#endif
1931
1932// ====================================================================================================================
1933// Protected member functions
1934// ====================================================================================================================
1935
1936/**
1937 - application has picture buffer list with size of GOP
1938 - picture buffer list acts as ring buffer
1939 - end of the list has the latest picture
1940 .
1941 */
1942#if SVC_EXTENSION
1943Void TAppEncTop::xGetBuffer( TComPicYuv*& rpcPicYuvRec, UInt layer)
1944{
1945  assert( m_iGOPSize > 0 );
1946
1947  // org. buffer
1948  if ( m_acListPicYuvRec[layer].size() == (UInt)m_iGOPSize )
1949  {
1950    rpcPicYuvRec = m_acListPicYuvRec[layer].popFront();
1951
1952  }
1953  else
1954  {
1955    rpcPicYuvRec = new TComPicYuv;
1956
1957#if SVC_UPSAMPLING
1958#if LAYER_CTB
1959#if AUXILIARY_PICTURES
1960    rpcPicYuvRec->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_acLayerCfg[layer].getChromaFormatIDC(), m_acLayerCfg[layer].m_uiMaxCUWidth, m_acLayerCfg[layer].m_uiMaxCUHeight, m_acLayerCfg[layer].m_uiMaxCUDepth, NULL );
1961#else
1962    rpcPicYuvRec->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_acLayerCfg[layer].m_uiMaxCUWidth, m_acLayerCfg[layer].m_uiMaxCUHeight, m_acLayerCfg[layer].m_uiMaxCUDepth, NULL );
1963#endif
1964#else
1965#if AUXILIARY_PICTURES
1966    rpcPicYuvRec->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_acLayerCfg[layer].getChromaFormatIDC(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth, NULL );
1967#else
1968    rpcPicYuvRec->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth, NULL );
1969#endif
1970#endif
1971#else
1972    rpcPicYuvRec->create( m_acLayerCfg[layer].getSourceWidth(), m_acLayerCfg[layer].getSourceHeight(), m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
1973#endif
1974
1975  }
1976  m_acListPicYuvRec[layer].pushBack( rpcPicYuvRec );
1977}
1978
1979Void TAppEncTop::xDeleteBuffer( )
1980{
1981  for(UInt layer=0; layer<m_numLayers; layer++)
1982  {
1983    TComList<TComPicYuv*>::iterator iterPicYuvRec  = m_acListPicYuvRec[layer].begin();
1984
1985    Int iSize = Int( m_acListPicYuvRec[layer].size() );
1986
1987    for ( Int i = 0; i < iSize; i++ )
1988    {
1989      TComPicYuv*  pcPicYuvRec  = *(iterPicYuvRec++);
1990      pcPicYuvRec->destroy();
1991      delete pcPicYuvRec; pcPicYuvRec = NULL;
1992    }
1993  }
1994}
1995
1996Void TAppEncTop::xWriteRecon(UInt layer, Int iNumEncoded)
1997{
1998  if (m_isField)
1999  {
2000    //Reinterlace fields
2001    Int i;
2002    TComList<TComPicYuv*>::iterator iterPicYuvRec = m_acListPicYuvRec[layer].end();
2003
2004    for ( i = 0; i < iNumEncoded; i++ )
2005    {
2006      --iterPicYuvRec;
2007    }
2008
2009    for ( i = 0; i < iNumEncoded/2; i++ )
2010    {
2011      TComPicYuv*  pcPicYuvRecTop  = *(iterPicYuvRec++);
2012      TComPicYuv*  pcPicYuvRecBottom  = *(iterPicYuvRec++);
2013
2014#if M0040_ADAPTIVE_RESOLUTION_CHANGE
2015      if (!m_acLayerCfg[layer].getReconFile().empty() && pcPicYuvRecTop->isReconstructed() && pcPicYuvRecBottom->isReconstructed())
2016#else
2017      if (!m_acLayerCfg[layer].getReconFile().empty())
2018#endif
2019      {
2020        m_acTVideoIOYuvReconFile[layer].write( pcPicYuvRecTop, pcPicYuvRecBottom, m_acLayerCfg[layer].getConfLeft(), m_acLayerCfg[layer].getConfRight(), m_acLayerCfg[layer].getConfTop(), m_acLayerCfg[layer].getConfBottom(), m_isTopFieldFirst );
2021      }
2022    }
2023  }
2024  else
2025  {
2026    Int i;
2027
2028    TComList<TComPicYuv*>::iterator iterPicYuvRec = m_acListPicYuvRec[layer].end();
2029
2030    for ( i = 0; i < iNumEncoded; i++ )
2031    {
2032      --iterPicYuvRec;
2033    }
2034
2035    for ( i = 0; i < iNumEncoded; i++ )
2036    {
2037      TComPicYuv*  pcPicYuvRec  = *(iterPicYuvRec++);
2038#if M0040_ADAPTIVE_RESOLUTION_CHANGE
2039      if (!m_acLayerCfg[layer].getReconFile().empty() && pcPicYuvRec->isReconstructed())
2040#else
2041      if (!m_acLayerCfg[layer].getReconFile().empty())
2042#endif
2043      {
2044        m_acTVideoIOYuvReconFile[layer].write( pcPicYuvRec, m_acLayerCfg[layer].getConfLeft(), m_acLayerCfg[layer].getConfRight(),
2045          m_acLayerCfg[layer].getConfTop(), m_acLayerCfg[layer].getConfBottom() );
2046      }
2047    }
2048  }
2049}
2050
2051Void TAppEncTop::xWriteStream(std::ostream& bitstreamFile, Int iNumEncoded, const std::list<AccessUnit>& accessUnits)
2052{
2053  if (m_isField)
2054  {
2055    //Reinterlace fields
2056    Int i;
2057    list<AccessUnit>::const_iterator iterBitstream = accessUnits.begin();
2058
2059#if M0040_ADAPTIVE_RESOLUTION_CHANGE
2060    for ( i = 0; i < iNumEncoded/2 && iterBitstream != accessUnits.end(); i++ )
2061#else
2062    for ( i = 0; i < iNumEncoded/2; i++ )
2063#endif
2064    {
2065      const AccessUnit& auTop = *(iterBitstream++);
2066      const vector<UInt>& statsTop = writeAnnexB(bitstreamFile, auTop);
2067      rateStatsAccum(auTop, statsTop);
2068
2069      const AccessUnit& auBottom = *(iterBitstream++);
2070      const vector<UInt>& statsBottom = writeAnnexB(bitstreamFile, auBottom);
2071      rateStatsAccum(auBottom, statsBottom);
2072    }
2073  }
2074  else
2075  {
2076    Int i;
2077
2078    list<AccessUnit>::const_iterator iterBitstream = accessUnits.begin();
2079
2080#if M0040_ADAPTIVE_RESOLUTION_CHANGE
2081    for ( i = 0; i < iNumEncoded && iterBitstream != accessUnits.end(); i++ )
2082#else
2083    for ( i = 0; i < iNumEncoded; i++ )
2084#endif
2085    {
2086      const AccessUnit& au = *(iterBitstream++);
2087      const vector<UInt>& stats = writeAnnexB(bitstreamFile, au);
2088      rateStatsAccum(au, stats);
2089    }
2090  }
2091}
2092
2093#else // SVC_EXTENSION
2094Void TAppEncTop::xGetBuffer( TComPicYuv*& rpcPicYuvRec)
2095{
2096  assert( m_iGOPSize > 0 );
2097
2098  // org. buffer
2099  if ( m_cListPicYuvRec.size() == (UInt)m_iGOPSize )
2100  {
2101    rpcPicYuvRec = m_cListPicYuvRec.popFront();
2102
2103  }
2104  else
2105  {
2106    rpcPicYuvRec = new TComPicYuv;
2107
2108    rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_uiMaxCUWidth, m_uiMaxCUHeight, m_uiMaxCUDepth );
2109
2110  }
2111  m_cListPicYuvRec.pushBack( rpcPicYuvRec );
2112}
2113
2114Void TAppEncTop::xDeleteBuffer( )
2115{
2116  TComList<TComPicYuv*>::iterator iterPicYuvRec  = m_cListPicYuvRec.begin();
2117
2118  Int iSize = Int( m_cListPicYuvRec.size() );
2119
2120  for ( Int i = 0; i < iSize; i++ )
2121  {
2122    TComPicYuv*  pcPicYuvRec  = *(iterPicYuvRec++);
2123    pcPicYuvRec->destroy();
2124    delete pcPicYuvRec; pcPicYuvRec = NULL;
2125  }
2126
2127}
2128
2129/** \param iNumEncoded  number of encoded frames
2130 */
2131Void TAppEncTop::xWriteOutput(std::ostream& bitstreamFile, Int iNumEncoded, const std::list<AccessUnit>& accessUnits)
2132{
2133  if (m_isField)
2134  {
2135    //Reinterlace fields
2136    Int i;
2137    TComList<TComPicYuv*>::iterator iterPicYuvRec = m_cListPicYuvRec.end();
2138    list<AccessUnit>::const_iterator iterBitstream = accessUnits.begin();
2139
2140    for ( i = 0; i < iNumEncoded; i++ )
2141    {
2142      --iterPicYuvRec;
2143    }
2144
2145    for ( i = 0; i < iNumEncoded/2; i++ )
2146    {
2147      TComPicYuv*  pcPicYuvRecTop  = *(iterPicYuvRec++);
2148      TComPicYuv*  pcPicYuvRecBottom  = *(iterPicYuvRec++);
2149
2150      if (m_pchReconFile)
2151      {
2152        m_cTVideoIOYuvReconFile.write( pcPicYuvRecTop, pcPicYuvRecBottom, m_confLeft, m_confRight, m_confTop, m_confBottom, m_isTopFieldFirst );
2153      }
2154
2155      const AccessUnit& auTop = *(iterBitstream++);
2156      const vector<UInt>& statsTop = writeAnnexB(bitstreamFile, auTop);
2157      rateStatsAccum(auTop, statsTop);
2158
2159      const AccessUnit& auBottom = *(iterBitstream++);
2160      const vector<UInt>& statsBottom = writeAnnexB(bitstreamFile, auBottom);
2161      rateStatsAccum(auBottom, statsBottom);
2162    }
2163  }
2164  else
2165  {
2166    Int i;
2167
2168    TComList<TComPicYuv*>::iterator iterPicYuvRec = m_cListPicYuvRec.end();
2169    list<AccessUnit>::const_iterator iterBitstream = accessUnits.begin();
2170
2171    for ( i = 0; i < iNumEncoded; i++ )
2172    {
2173      --iterPicYuvRec;
2174    }
2175
2176    for ( i = 0; i < iNumEncoded; i++ )
2177    {
2178      TComPicYuv*  pcPicYuvRec  = *(iterPicYuvRec++);
2179      if (m_pchReconFile)
2180      {
2181#if SYNTAX_OUTPUT
2182        m_cTVideoIOYuvReconFile.write( pcPicYuvRec, m_confLeft, m_confRight, m_confTop, m_confBottom );
2183#endif
2184      }
2185
2186      const AccessUnit& au = *(iterBitstream++);
2187      const vector<UInt>& stats = writeAnnexB(bitstreamFile, au);
2188      rateStatsAccum(au, stats);
2189    }
2190  }
2191}
2192#endif
2193
2194/**
2195 *
2196 */
2197void TAppEncTop::rateStatsAccum(const AccessUnit& au, const std::vector<UInt>& annexBsizes)
2198{
2199  AccessUnit::const_iterator it_au = au.begin();
2200  vector<UInt>::const_iterator it_stats = annexBsizes.begin();
2201
2202  for (; it_au != au.end(); it_au++, it_stats++)
2203  {
2204    switch ((*it_au)->m_nalUnitType)
2205    {
2206    case NAL_UNIT_CODED_SLICE_TRAIL_R:
2207    case NAL_UNIT_CODED_SLICE_TRAIL_N:
2208    case NAL_UNIT_CODED_SLICE_TSA_R:
2209    case NAL_UNIT_CODED_SLICE_TSA_N:
2210    case NAL_UNIT_CODED_SLICE_STSA_R:
2211    case NAL_UNIT_CODED_SLICE_STSA_N:
2212    case NAL_UNIT_CODED_SLICE_BLA_W_LP:
2213    case NAL_UNIT_CODED_SLICE_BLA_W_RADL:
2214    case NAL_UNIT_CODED_SLICE_BLA_N_LP:
2215    case NAL_UNIT_CODED_SLICE_IDR_W_RADL:
2216    case NAL_UNIT_CODED_SLICE_IDR_N_LP:
2217    case NAL_UNIT_CODED_SLICE_CRA:
2218    case NAL_UNIT_CODED_SLICE_RADL_N:
2219    case NAL_UNIT_CODED_SLICE_RADL_R:
2220    case NAL_UNIT_CODED_SLICE_RASL_N:
2221    case NAL_UNIT_CODED_SLICE_RASL_R:
2222    case NAL_UNIT_VPS:
2223    case NAL_UNIT_SPS:
2224    case NAL_UNIT_PPS:
2225      m_essentialBytes += *it_stats;
2226      break;
2227    default:
2228      break;
2229    }
2230
2231    m_totalBytes += *it_stats;
2232  }
2233}
2234
2235void TAppEncTop::printRateSummary()
2236{
2237#if SVC_EXTENSION
2238  Double time = (Double) m_iFrameRcvd / m_acLayerCfg[m_numLayers-1].getFrameRate();
2239#else
2240  Double time = (Double) m_iFrameRcvd / m_iFrameRate;
2241#endif
2242  printf("Bytes written to file: %u (%.3f kbps)\n", m_totalBytes, 0.008 * m_totalBytes / time);
2243#if VERBOSE_RATE
2244  printf("Bytes for SPS/PPS/Slice (Incl. Annex B): %u (%.3f kbps)\n", m_essentialBytes, 0.008 * m_essentialBytes / time);
2245#endif
2246}
2247
2248//! \}
Note: See TracBrowser for help on using the repository browser.