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

Last change on this file since 783 was 778, checked in by nokia, 11 years ago

JCTVC-Q0078/R0042: Additional layer sets and independent non-base layer rewriting tool (utils/BLRewrite)

  • Property svn:eol-style set to native
File size: 55.7 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license. 
5 *
6 * Copyright (c) 2010-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     TEncTop.cpp
35    \brief    encoder class
36*/
37
38#include "TLibCommon/CommonDef.h"
39#include "TEncTop.h"
40#include "TEncPic.h"
41#if FAST_BIT_EST
42#include "TLibCommon/ContextModel.h"
43#endif
44
45//! \ingroup TLibEncoder
46//! \{
47#if SVC_EXTENSION 
48Int TEncTop::m_iSPSIdCnt = 0;
49Int TEncTop::m_iPPSIdCnt = 0;
50TComVPS TEncCfg::m_cVPS;
51#endif
52
53// ====================================================================================================================
54// Constructor / destructor / create / destroy
55// ====================================================================================================================
56
57TEncTop::TEncTop()
58{
59  m_iPOCLast          = -1;
60  m_iNumPicRcvd       =  0;
61  m_uiNumAllPicCoded  =  0;
62  m_pppcRDSbacCoder   =  NULL;
63  m_pppcBinCoderCABAC =  NULL;
64  m_cRDGoOnSbacCoder.init( &m_cRDGoOnBinCoderCABAC );
65#if ENC_DEC_TRACE
66  g_hTrace = fopen( "TraceEnc.txt", "wb" );
67  g_bJustDoIt = g_bEncDecTraceDisable;
68  g_nSymbolCounter = 0;
69#endif
70
71  m_iMaxRefPicNum     = 0;
72
73#if FAST_BIT_EST
74  ContextModel::buildNextStateTable();
75#endif
76
77  m_pcSbacCoders           = NULL;
78  m_pcBinCoderCABACs       = NULL;
79  m_ppppcRDSbacCoders      = NULL;
80  m_ppppcBinCodersCABAC    = NULL;
81  m_pcRDGoOnSbacCoders     = NULL;
82  m_pcRDGoOnBinCodersCABAC = NULL;
83  m_pcBitCounters          = NULL;
84  m_pcRdCosts              = NULL;
85#if SVC_EXTENSION
86  memset(m_cIlpPic, 0, sizeof(m_cIlpPic));
87#if REF_IDX_MFM
88  m_bMFMEnabledFlag = false;
89#endif
90  m_numScaledRefLayerOffsets = 0;
91#if POC_RESET_FLAG
92  m_pocAdjustmentValue     = 0;
93#endif
94#if NO_CLRAS_OUTPUT_FLAG
95  m_noClrasOutputFlag          = false;
96  m_layerInitializedFlag       = false;
97  m_firstPicInLayerDecodedFlag = false;
98  m_noOutputOfPriorPicsFlags   = false;
99#endif
100#endif //SVC_EXTENSION
101}
102
103TEncTop::~TEncTop()
104{
105#if ENC_DEC_TRACE
106  fclose( g_hTrace );
107#endif
108}
109
110Void TEncTop::create ()
111{
112#if !SVC_EXTENSION
113  // initialize global variables
114  initROM();
115#endif
116
117  // create processing unit classes
118#if SVC_EXTENSION
119  m_cGOPEncoder.        create( m_layerId );
120#else
121  m_cGOPEncoder.        create();
122#endif
123#if AUXILIARY_PICTURES
124  m_cSliceEncoder.      create( getSourceWidth(), getSourceHeight(), m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
125#else
126  m_cSliceEncoder.      create( getSourceWidth(), getSourceHeight(), g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
127#endif
128  m_cCuEncoder.         create( g_uiMaxCUDepth, g_uiMaxCUWidth, g_uiMaxCUHeight );
129  if (m_bUseSAO)
130  {
131#if AUXILIARY_PICTURES
132    m_cEncSAO.create( getSourceWidth(), getSourceHeight(), m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
133#else
134    m_cEncSAO.create( getSourceWidth(), getSourceHeight(), g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
135#endif
136#if SAO_ENCODE_ALLOW_USE_PREDEBLOCK
137    m_cEncSAO.createEncData(getSaoLcuBoundary());
138#else
139    m_cEncSAO.createEncData();
140#endif
141  }
142#if ADAPTIVE_QP_SELECTION
143  if (m_bUseAdaptQpSelect)
144  {
145    m_cTrQuant.initSliceQpDelta();
146  }
147#endif
148  m_cLoopFilter.        create( g_uiMaxCUDepth );
149 
150  if ( m_RCEnableRateControl )
151  {
152    m_cRateCtrl.init( m_framesToBeEncoded, m_RCTargetBitrate, m_iFrameRate, m_iGOPSize, m_iSourceWidth, m_iSourceHeight,
153                      g_uiMaxCUWidth, g_uiMaxCUHeight, m_RCKeepHierarchicalBit, m_RCUseLCUSeparateModel, m_GOPList );
154  }
155
156  m_pppcRDSbacCoder = new TEncSbac** [g_uiMaxCUDepth+1];
157#if FAST_BIT_EST
158  m_pppcBinCoderCABAC = new TEncBinCABACCounter** [g_uiMaxCUDepth+1];
159#else
160  m_pppcBinCoderCABAC = new TEncBinCABAC** [g_uiMaxCUDepth+1];
161#endif
162
163  for ( Int iDepth = 0; iDepth < g_uiMaxCUDepth+1; iDepth++ )
164  {
165    m_pppcRDSbacCoder[iDepth] = new TEncSbac* [CI_NUM];
166#if FAST_BIT_EST
167    m_pppcBinCoderCABAC[iDepth] = new TEncBinCABACCounter* [CI_NUM];
168#else
169    m_pppcBinCoderCABAC[iDepth] = new TEncBinCABAC* [CI_NUM];
170#endif
171
172    for (Int iCIIdx = 0; iCIIdx < CI_NUM; iCIIdx ++ )
173    {
174      m_pppcRDSbacCoder[iDepth][iCIIdx] = new TEncSbac;
175#if FAST_BIT_EST
176      m_pppcBinCoderCABAC [iDepth][iCIIdx] = new TEncBinCABACCounter;
177#else
178      m_pppcBinCoderCABAC [iDepth][iCIIdx] = new TEncBinCABAC;
179#endif
180      m_pppcRDSbacCoder   [iDepth][iCIIdx]->init( m_pppcBinCoderCABAC [iDepth][iCIIdx] );
181    }
182  }
183
184#if LAYER_CTB
185  memcpy(g_auiLayerZscanToRaster[m_layerId], g_auiZscanToRaster, sizeof( g_auiZscanToRaster ) );
186  memcpy(g_auiLayerRasterToZscan[m_layerId], g_auiRasterToZscan, sizeof( g_auiRasterToZscan ) );
187  memcpy(g_auiLayerRasterToPelX[m_layerId],  g_auiRasterToPelX,  sizeof( g_auiRasterToPelX ) );
188  memcpy(g_auiLayerRasterToPelY[m_layerId],  g_auiRasterToPelY,  sizeof( g_auiRasterToPelY ) );
189#endif
190}
191
192/**
193 - Allocate coders required for wavefront for the nominated number of substreams.
194 .
195 \param iNumSubstreams Determines how much information to allocate.
196 */
197Void TEncTop::createWPPCoders(Int iNumSubstreams)
198{
199  if (m_pcSbacCoders != NULL)
200  {
201    return; // already generated.
202  }
203
204  m_iNumSubstreams         = iNumSubstreams;
205  m_pcSbacCoders           = new TEncSbac       [iNumSubstreams];
206  m_pcBinCoderCABACs       = new TEncBinCABAC   [iNumSubstreams];
207  m_pcRDGoOnSbacCoders     = new TEncSbac       [iNumSubstreams];
208  m_pcRDGoOnBinCodersCABAC = new TEncBinCABAC   [iNumSubstreams];
209  m_pcBitCounters          = new TComBitCounter [iNumSubstreams];
210  m_pcRdCosts              = new TComRdCost     [iNumSubstreams];
211
212  for ( UInt ui = 0 ; ui < iNumSubstreams; ui++ )
213  {
214    m_pcRDGoOnSbacCoders[ui].init( &m_pcRDGoOnBinCodersCABAC[ui] );
215    m_pcSbacCoders[ui].init( &m_pcBinCoderCABACs[ui] );
216  }
217
218  m_ppppcRDSbacCoders      = new TEncSbac***    [iNumSubstreams];
219  m_ppppcBinCodersCABAC    = new TEncBinCABAC***[iNumSubstreams];
220  for ( UInt ui = 0 ; ui < iNumSubstreams ; ui++ )
221  {
222    m_ppppcRDSbacCoders[ui]  = new TEncSbac** [g_uiMaxCUDepth+1];
223    m_ppppcBinCodersCABAC[ui]= new TEncBinCABAC** [g_uiMaxCUDepth+1];
224
225    for ( Int iDepth = 0; iDepth < g_uiMaxCUDepth+1; iDepth++ )
226    {
227      m_ppppcRDSbacCoders[ui][iDepth]  = new TEncSbac*     [CI_NUM];
228      m_ppppcBinCodersCABAC[ui][iDepth]= new TEncBinCABAC* [CI_NUM];
229
230      for (Int iCIIdx = 0; iCIIdx < CI_NUM; iCIIdx ++ )
231      {
232        m_ppppcRDSbacCoders  [ui][iDepth][iCIIdx] = new TEncSbac;
233        m_ppppcBinCodersCABAC[ui][iDepth][iCIIdx] = new TEncBinCABAC;
234        m_ppppcRDSbacCoders  [ui][iDepth][iCIIdx]->init( m_ppppcBinCodersCABAC[ui][iDepth][iCIIdx] );
235      }
236    }
237  }
238}
239
240Void TEncTop::destroy ()
241{
242  // destroy processing unit classes
243  m_cGOPEncoder.        destroy();
244  m_cSliceEncoder.      destroy();
245  m_cCuEncoder.         destroy();
246  if (m_cSPS.getUseSAO())
247  {
248    m_cEncSAO.destroyEncData();
249    m_cEncSAO.destroy();
250  }
251  m_cLoopFilter.        destroy();
252  m_cRateCtrl.          destroy();
253
254  Int iDepth;
255  for ( iDepth = 0; iDepth < g_uiMaxCUDepth+1; iDepth++ )
256  {
257    for (Int iCIIdx = 0; iCIIdx < CI_NUM; iCIIdx ++ )
258    {
259      delete m_pppcRDSbacCoder[iDepth][iCIIdx];
260      delete m_pppcBinCoderCABAC[iDepth][iCIIdx];
261    }
262  }
263
264  for ( iDepth = 0; iDepth < g_uiMaxCUDepth+1; iDepth++ )
265  {
266    delete [] m_pppcRDSbacCoder[iDepth];
267    delete [] m_pppcBinCoderCABAC[iDepth];
268  }
269
270  delete [] m_pppcRDSbacCoder;
271  delete [] m_pppcBinCoderCABAC;
272
273  for ( UInt ui = 0; ui < m_iNumSubstreams; ui++ )
274  {
275    for ( iDepth = 0; iDepth < g_uiMaxCUDepth+1; iDepth++ )
276    {
277      for (Int iCIIdx = 0; iCIIdx < CI_NUM; iCIIdx ++ )
278      {
279        delete m_ppppcRDSbacCoders  [ui][iDepth][iCIIdx];
280        delete m_ppppcBinCodersCABAC[ui][iDepth][iCIIdx];
281      }
282    }
283
284    for ( iDepth = 0; iDepth < g_uiMaxCUDepth+1; iDepth++ )
285    {
286      delete [] m_ppppcRDSbacCoders  [ui][iDepth];
287      delete [] m_ppppcBinCodersCABAC[ui][iDepth];
288    }
289    delete[] m_ppppcRDSbacCoders  [ui];
290    delete[] m_ppppcBinCodersCABAC[ui];
291  }
292  delete[] m_ppppcRDSbacCoders;
293  delete[] m_ppppcBinCodersCABAC;
294  delete[] m_pcSbacCoders;
295  delete[] m_pcBinCoderCABACs;
296  delete[] m_pcRDGoOnSbacCoders; 
297  delete[] m_pcRDGoOnBinCodersCABAC;
298  delete[] m_pcBitCounters;
299  delete[] m_pcRdCosts;
300 
301#if SVC_EXTENSION
302  for(Int i=0; i<MAX_NUM_REF; i++)
303  {
304    if(m_cIlpPic[i])
305    {
306      m_cIlpPic[i]->destroy();
307      delete m_cIlpPic[i];
308      m_cIlpPic[i] = NULL;
309    }
310  }
311#else
312  // destroy ROM
313  destroyROM();
314#endif
315  return;
316}
317
318Void TEncTop::init(Bool isFieldCoding)
319{
320  // initialize SPS
321  xInitSPS();
322 
323  /* set the VPS profile information */
324  *m_cVPS.getPTL() = *m_cSPS.getPTL();
325  m_cVPS.getTimingInfo()->setTimingInfoPresentFlag       ( false );
326  // initialize PPS
327  m_cPPS.setSPS(&m_cSPS);
328  xInitPPS();
329  xInitRPS(isFieldCoding);
330
331  xInitPPSforTiles();
332
333  // initialize processing unit classes
334  m_cGOPEncoder.  init( this );
335  m_cSliceEncoder.init( this );
336  m_cCuEncoder.   init( this );
337 
338  // initialize transform & quantization class
339  m_pcCavlcCoder = getCavlcCoder();
340 
341  m_cTrQuant.init( 1 << m_uiQuadtreeTULog2MaxSize,
342                  m_useRDOQ, 
343                  m_useRDOQTS,
344                  true 
345                  ,m_useTransformSkipFast
346#if ADAPTIVE_QP_SELECTION                 
347                  , m_bUseAdaptQpSelect
348#endif
349                  );
350 
351  // initialize encoder search class
352  m_cSearch.init( this, &m_cTrQuant, m_iSearchRange, m_bipredSearchRange, m_iFastSearch, 0, &m_cEntropyCoder, &m_cRdCost, getRDSbacCoder(), getRDGoOnSbacCoder() );
353
354  m_iMaxRefPicNum = 0;
355#if SVC_EXTENSION
356  m_iSPSIdCnt ++;
357  m_iPPSIdCnt ++;
358  xInitILRP();
359#endif
360}
361
362// ====================================================================================================================
363// Public member functions
364// ====================================================================================================================
365
366Void TEncTop::deletePicBuffer()
367{
368  TComList<TComPic*>::iterator iterPic = m_cListPic.begin();
369  Int iSize = Int( m_cListPic.size() );
370 
371  for ( Int i = 0; i < iSize; i++ )
372  {
373    TComPic* pcPic = *(iterPic++);
374   
375    pcPic->destroy();
376    delete pcPic;
377    pcPic = NULL;
378  }
379}
380
381/**
382 - Application has picture buffer list with size of GOP + 1
383 - Picture buffer list acts like as ring buffer
384 - End of the list has the latest picture
385 .
386 \param   flush               cause encoder to encode a partial GOP
387 \param   pcPicYuvOrg         original YUV picture
388 \retval  rcListPicYuvRecOut  list of reconstruction YUV pictures
389 \retval  rcListBitstreamOut  list of output bitstreams
390 \retval  iNumEncoded         number of encoded pictures
391 */
392#if SVC_EXTENSION
393Void TEncTop::encode( TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int iPicIdInGOP )
394{
395  // compress GOP
396#if !RC_SHVC_HARMONIZATION
397  if ( m_RCEnableRateControl )
398  {
399    m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
400  }
401#endif
402
403  // compress GOP
404  m_cGOPEncoder.compressGOP(iPicIdInGOP, m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, false, false);
405
406#if !RC_SHVC_HARMONIZATION
407  if ( m_RCEnableRateControl )
408  {
409    m_cRateCtrl.destroyRCGOP();
410  }
411#endif
412 
413  m_uiNumAllPicCoded ++;
414}
415
416Void TEncTop::encodePrep( TComPicYuv* pcPicYuvOrg )
417{
418  if (pcPicYuvOrg) 
419  {
420    // get original YUV
421    TComPic* pcPicCurr = NULL;
422    xGetNewPicBuffer( pcPicCurr );
423    pcPicYuvOrg->copyToPic( pcPicCurr->getPicYuvOrg() );
424
425    // compute image characteristics
426    if ( getUseAdaptiveQP() )
427    {
428      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcPicCurr ) );
429    }
430  }
431}
432#else
433Void TEncTop::encode(Bool flush, TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded )
434{
435  if (pcPicYuvOrg) {
436    // get original YUV
437    TComPic* pcPicCurr = NULL;
438    xGetNewPicBuffer( pcPicCurr );
439    pcPicYuvOrg->copyToPic( pcPicCurr->getPicYuvOrg() );
440
441    // compute image characteristics
442    if ( getUseAdaptiveQP() )
443    {
444      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcPicCurr ) );
445    }
446  }
447 
448  if (!m_iNumPicRcvd || (!flush && m_iPOCLast != 0 && m_iNumPicRcvd != m_iGOPSize && m_iGOPSize))
449  {
450    iNumEncoded = 0;
451    return;
452  }
453 
454  if ( m_RCEnableRateControl )
455  {
456    m_cRateCtrl.initRCGOP( m_iNumPicRcvd );
457  }
458
459  // compress GOP
460
461  m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, false, false);
462
463  if ( m_RCEnableRateControl )
464  {
465    m_cRateCtrl.destroyRCGOP();
466  }
467 
468  iNumEncoded         = m_iNumPicRcvd;
469  m_iNumPicRcvd       = 0;
470  m_uiNumAllPicCoded += iNumEncoded;
471}
472#endif
473
474/**------------------------------------------------
475 Separate interlaced frame into two fields
476 -------------------------------------------------**/
477void separateFields(Pel* org, Pel* dstField, UInt stride, UInt width, UInt height, Bool isTop)
478{
479  if (!isTop)
480  {
481    org += stride;
482  }
483  for (Int y = 0; y < height>>1; y++)
484  {
485    for (Int x = 0; x < width; x++)
486    {
487      dstField[x] = org[x];
488    }
489   
490    dstField += stride;
491    org += stride*2;
492  }
493 
494}
495
496#if SVC_EXTENSION
497Void TEncTop::encodePrep( TComPicYuv* pcPicYuvOrg, Bool isTff )
498{
499  if (pcPicYuvOrg)
500  {
501    /* -- TOP FIELD -- */
502    /* -- Top field initialization -- */
503   
504    TComPic *pcTopField;
505    xGetNewPicBuffer( pcTopField );
506    pcTopField->setReconMark (false);
507   
508    pcTopField->getSlice(0)->setPOC( m_iPOCLast );
509    pcTopField->getPicYuvRec()->setBorderExtension(false);
510    pcTopField->setTopField(isTff);
511   
512    Int nHeight = pcPicYuvOrg->getHeight();
513    Int nWidth = pcPicYuvOrg->getWidth();
514    Int nStride = pcPicYuvOrg->getStride();
515    Int nPadLuma = pcPicYuvOrg->getLumaMargin();
516    Int nPadChroma = pcPicYuvOrg->getChromaMargin();
517   
518    // Get pointers
519    Pel * PicBufY = pcPicYuvOrg->getBufY();
520    Pel * PicBufU = pcPicYuvOrg->getBufU();
521    Pel * PicBufV = pcPicYuvOrg->getBufV();
522   
523    Pel * pcTopFieldY =  pcTopField->getPicYuvOrg()->getLumaAddr();
524    Pel * pcTopFieldU =  pcTopField->getPicYuvOrg()->getCbAddr();
525    Pel * pcTopFieldV =  pcTopField->getPicYuvOrg()->getCrAddr();
526
527    // compute image characteristics
528    if ( getUseAdaptiveQP() )
529    {
530      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcTopField ) );
531    }   
532
533    /* -- Defield -- */
534   
535    Bool isTop = isTff;
536   
537    separateFields(PicBufY + nPadLuma + nStride*nPadLuma, pcTopFieldY, nStride, nWidth, nHeight, isTop);
538    separateFields(PicBufU + nPadChroma + (nStride >> 1)*nPadChroma, pcTopFieldU, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
539    separateFields(PicBufV + nPadChroma + (nStride >> 1)*nPadChroma, pcTopFieldV, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
540
541    /* -- BOTTOM FIELD -- */
542    /* -- Bottom field initialization -- */
543   
544    TComPic* pcBottomField;
545    xGetNewPicBuffer( pcBottomField );
546    pcBottomField->setReconMark (false);
547   
548    pcBottomField->getSlice(0)->setPOC( m_iPOCLast);
549    pcBottomField->getPicYuvRec()->setBorderExtension(false);
550    pcBottomField->setTopField(!isTff);
551   
552    nHeight = pcPicYuvOrg->getHeight();
553    nWidth = pcPicYuvOrg->getWidth();
554    nStride = pcPicYuvOrg->getStride();
555    nPadLuma = pcPicYuvOrg->getLumaMargin();
556    nPadChroma = pcPicYuvOrg->getChromaMargin();
557   
558    // Get pointers
559    PicBufY = pcPicYuvOrg->getBufY();
560    PicBufU = pcPicYuvOrg->getBufU();
561    PicBufV = pcPicYuvOrg->getBufV();
562   
563    Pel * pcBottomFieldY =  pcBottomField->getPicYuvOrg()->getLumaAddr();
564    Pel * pcBottomFieldU =  pcBottomField->getPicYuvOrg()->getCbAddr();
565    Pel * pcBottomFieldV =  pcBottomField->getPicYuvOrg()->getCrAddr();
566
567    // compute image characteristics
568    if ( getUseAdaptiveQP() )
569    {
570      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcBottomField ) );
571    }       
572
573    /* -- Defield -- */
574   
575    isTop = !isTff;
576   
577    separateFields(PicBufY + nPadLuma + nStride*nPadLuma, pcBottomFieldY, nStride, nWidth, nHeight, isTop);
578    separateFields(PicBufU + nPadChroma + (nStride >> 1)*nPadChroma, pcBottomFieldU, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
579    separateFields(PicBufV + nPadChroma + (nStride >> 1)*nPadChroma, pcBottomFieldV, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
580   
581  }
582}
583
584Void TEncTop::encode( TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int iPicIdInGOP, Bool isTff )
585{
586  // compress GOP
587  if (m_iPOCLast == 0) // compress field 0
588  {
589    m_cGOPEncoder.compressGOP(iPicIdInGOP, m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff);
590  }
591
592  if (pcPicYuvOrg)
593  {
594    TComPicYuv* rpcPicYuvRec = new TComPicYuv;
595    if ( rcListPicYuvRecOut.size() == (UInt)m_iGOPSize )
596    {
597      rpcPicYuvRec = rcListPicYuvRecOut.popFront();
598    }
599    else
600    {
601#if AUXILIARY_PICTURES
602      rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
603#else
604      rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
605#endif
606    }
607    rcListPicYuvRecOut.pushBack( rpcPicYuvRec );
608  }
609 
610  // compress GOP
611  m_cGOPEncoder.compressGOP(iPicIdInGOP, m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff);
612
613  m_uiNumAllPicCoded ++;
614}
615#else
616Void TEncTop::encode(Bool flush, TComPicYuv* pcPicYuvOrg, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsOut, Int& iNumEncoded, Bool isTff)
617{
618  /* -- TOP FIELD -- */
619 
620  if (pcPicYuvOrg)
621  {
622   
623    /* -- Top field initialization -- */
624   
625    TComPic *pcTopField;
626    xGetNewPicBuffer( pcTopField );
627    pcTopField->setReconMark (false);
628   
629    pcTopField->getSlice(0)->setPOC( m_iPOCLast );
630    pcTopField->getPicYuvRec()->setBorderExtension(false);
631    pcTopField->setTopField(isTff);
632   
633    Int nHeight = pcPicYuvOrg->getHeight();
634    Int nWidth = pcPicYuvOrg->getWidth();
635    Int nStride = pcPicYuvOrg->getStride();
636    Int nPadLuma = pcPicYuvOrg->getLumaMargin();
637    Int nPadChroma = pcPicYuvOrg->getChromaMargin();
638   
639    // Get pointers
640    Pel * PicBufY = pcPicYuvOrg->getBufY();
641    Pel * PicBufU = pcPicYuvOrg->getBufU();
642    Pel * PicBufV = pcPicYuvOrg->getBufV();
643   
644    Pel * pcTopFieldY =  pcTopField->getPicYuvOrg()->getLumaAddr();
645    Pel * pcTopFieldU =  pcTopField->getPicYuvOrg()->getCbAddr();
646    Pel * pcTopFieldV =  pcTopField->getPicYuvOrg()->getCrAddr();
647   
648    // compute image characteristics
649    if ( getUseAdaptiveQP() )
650    {
651      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcTopField ) );
652    }
653   
654    /* -- Defield -- */
655   
656    Bool isTop = isTff;
657   
658    separateFields(PicBufY + nPadLuma + nStride*nPadLuma, pcTopFieldY, nStride, nWidth, nHeight, isTop);
659    separateFields(PicBufU + nPadChroma + (nStride >> 1)*nPadChroma, pcTopFieldU, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
660    separateFields(PicBufV + nPadChroma + (nStride >> 1)*nPadChroma, pcTopFieldV, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
661   
662  }
663 
664  if (m_iPOCLast == 0) // compress field 0
665  {
666    m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff);
667  }
668 
669  /* -- BOTTOM FIELD -- */
670 
671  if (pcPicYuvOrg)
672  {
673   
674    /* -- Bottom field initialization -- */
675   
676    TComPic* pcBottomField;
677    xGetNewPicBuffer( pcBottomField );
678    pcBottomField->setReconMark (false);
679   
680    TComPicYuv* rpcPicYuvRec;
681    if ( rcListPicYuvRecOut.size() == (UInt)m_iGOPSize )
682    {
683      rpcPicYuvRec = rcListPicYuvRecOut.popFront();
684    }
685    else
686    {
687      rpcPicYuvRec = new TComPicYuv;
688      rpcPicYuvRec->create( m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth );
689    }
690    rcListPicYuvRecOut.pushBack( rpcPicYuvRec );
691   
692    pcBottomField->getSlice(0)->setPOC( m_iPOCLast);
693    pcBottomField->getPicYuvRec()->setBorderExtension(false);
694    pcBottomField->setTopField(!isTff);
695   
696    int nHeight = pcPicYuvOrg->getHeight();
697    int nWidth = pcPicYuvOrg->getWidth();
698    int nStride = pcPicYuvOrg->getStride();
699    int nPadLuma = pcPicYuvOrg->getLumaMargin();
700    int nPadChroma = pcPicYuvOrg->getChromaMargin();
701   
702    // Get pointers
703    Pel * PicBufY = pcPicYuvOrg->getBufY();
704    Pel * PicBufU = pcPicYuvOrg->getBufU();
705    Pel * PicBufV = pcPicYuvOrg->getBufV();
706   
707    Pel * pcBottomFieldY =  pcBottomField->getPicYuvOrg()->getLumaAddr();
708    Pel * pcBottomFieldU =  pcBottomField->getPicYuvOrg()->getCbAddr();
709    Pel * pcBottomFieldV =  pcBottomField->getPicYuvOrg()->getCrAddr();
710   
711    // Compute image characteristics
712    if ( getUseAdaptiveQP() )
713    {
714      m_cPreanalyzer.xPreanalyze( dynamic_cast<TEncPic*>( pcBottomField ) );
715    }
716   
717    /* -- Defield -- */
718   
719    Bool isTop = !isTff;
720   
721    separateFields(PicBufY + nPadLuma + nStride*nPadLuma, pcBottomFieldY, nStride, nWidth, nHeight, isTop);
722    separateFields(PicBufU + nPadChroma + (nStride >> 1)*nPadChroma, pcBottomFieldU, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
723    separateFields(PicBufV + nPadChroma + (nStride >> 1)*nPadChroma, pcBottomFieldV, nStride >> 1, nWidth >> 1, nHeight >> 1, isTop);
724   
725  }
726 
727  if ( ( !(m_iNumPicRcvd) || (!flush && m_iPOCLast != 1 && m_iNumPicRcvd != m_iGOPSize && m_iGOPSize)) )
728  {
729    iNumEncoded = 0;
730    return;
731  }
732 
733  // compress GOP
734  m_cGOPEncoder.compressGOP(m_iPOCLast, m_iNumPicRcvd, m_cListPic, rcListPicYuvRecOut, accessUnitsOut, true, isTff);
735 
736  iNumEncoded = m_iNumPicRcvd;
737  m_iNumPicRcvd = 0;
738  m_uiNumAllPicCoded += iNumEncoded;
739}
740#endif
741
742// ====================================================================================================================
743// Protected member functions
744// ====================================================================================================================
745
746/**
747 - Application has picture buffer list with size of GOP + 1
748 - Picture buffer list acts like as ring buffer
749 - End of the list has the latest picture
750 .
751 \retval rpcPic obtained picture buffer
752 */
753Void TEncTop::xGetNewPicBuffer ( TComPic*& rpcPic )
754{
755  TComSlice::sortPicList(m_cListPic);
756 
757  if (m_cListPic.size() >= (UInt)(m_iGOPSize + getMaxDecPicBuffering(MAX_TLAYER-1) + 2) )
758  {
759    TComList<TComPic*>::iterator iterPic  = m_cListPic.begin();
760    Int iSize = Int( m_cListPic.size() );
761    for ( Int i = 0; i < iSize; i++ )
762    {
763      rpcPic = *(iterPic++);
764      if(rpcPic->getSlice(0)->isReferenced() == false)
765      {
766        break;
767      }
768    }
769  }
770  else
771  {
772    if ( getUseAdaptiveQP() )
773    {
774      TEncPic* pcEPic = new TEncPic;
775
776#if SVC_EXTENSION //Temporal solution, should be modified
777      if(m_layerId > 0)
778      {
779        for(UInt i = 0; i < m_cVPS.getNumDirectRefLayers( m_layerId ); i++ )
780        {
781#if O0098_SCALED_REF_LAYER_ID
782          const Window scalEL = getSPS()->getScaledRefLayerWindowForLayer(m_cVPS.getRefLayerId(m_layerId, i));
783#else
784          const Window scalEL = getSPS()->getScaledRefLayerWindow(i);
785#endif
786          Bool zeroOffsets = ( scalEL.getWindowLeftOffset() == 0 && scalEL.getWindowRightOffset() == 0 && scalEL.getWindowTopOffset() == 0 && scalEL.getWindowBottomOffset() == 0 );
787
788#if VPS_EXTN_DIRECT_REF_LAYERS
789          TEncTop *pcEncTopBase = (TEncTop *)getRefLayerEnc( i );
790#else
791          TEncTop *pcEncTopBase = (TEncTop *)getLayerEnc( m_layerId-1 );
792#endif
793#if O0194_DIFFERENT_BITDEPTH_EL_BL
794          UInt refLayerId = m_cVPS.getRefLayerId(m_layerId, i);
795          Bool sameBitDepths = ( g_bitDepthYLayer[m_layerId] == g_bitDepthYLayer[refLayerId] ) && ( g_bitDepthCLayer[m_layerId] == g_bitDepthCLayer[refLayerId] );
796
797          if( m_iSourceWidth != pcEncTopBase->getSourceWidth() || m_iSourceHeight != pcEncTopBase->getSourceHeight() || !zeroOffsets || !sameBitDepths
798#if Q0048_CGS_3D_ASYMLUT
799            || m_cPPS.getCGSFlag() > 0
800#endif
801#if LAYER_CTB
802            || pcEncTopBase->getSPS()->getMaxCUWidth() != m_cSPS.getMaxCUWidth() || pcEncTopBase->getSPS()->getMaxCUHeight() != m_cSPS.getMaxCUHeight() || pcEncTopBase->getSPS()->getMaxCUDepth() != m_cSPS.getMaxCUDepth()
803#endif
804            )
805#else
806          if(m_iSourceWidth != pcEncTopBase->getSourceWidth() || m_iSourceHeight != pcEncTopBase->getSourceHeight() || !zeroOffsets )
807#endif
808          {
809            pcEPic->setSpatialEnhLayerFlag( i, true );
810
811            //only for scalable extension
812#if SCALABILITY_MASK_E0104
813            assert( m_cVPS.getScalabilityMask(2) == true );
814#else
815            assert( m_cVPS.getScalabilityMask(1) == true );
816#endif
817          }
818        }
819      }
820#endif
821
822#if SVC_EXTENSION
823#if AUXILIARY_PICTURES
824      pcEPic->create( m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, m_cPPS.getMaxCuDQPDepth()+1 ,
825                      m_conformanceWindow, m_defaultDisplayWindow, m_numReorderPics, &m_cSPS);
826#else
827      pcEPic->create( m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, m_cPPS.getMaxCuDQPDepth()+1 ,
828                      m_conformanceWindow, m_defaultDisplayWindow, m_numReorderPics, &m_cSPS);
829#endif
830#else  //SVC_EXTENSION
831      pcEPic->create( m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, m_cPPS.getMaxCuDQPDepth()+1 ,
832                      m_conformanceWindow, m_defaultDisplayWindow, m_numReorderPics);
833#endif //SVC_EXTENSION
834      rpcPic = pcEPic;
835    }
836    else
837    {
838      rpcPic = new TComPic;
839
840#if SVC_EXTENSION //Temporal solution, should be modified
841      if(m_layerId > 0)
842      {
843        for(UInt i = 0; i < m_cVPS.getNumDirectRefLayers( m_layerId ); i++ )
844        {
845#if O0098_SCALED_REF_LAYER_ID
846          const Window scalEL = getSPS()->getScaledRefLayerWindowForLayer(m_cVPS.getRefLayerId(m_layerId, i));
847#else
848          const Window scalEL = getSPS()->getScaledRefLayerWindow(i);
849#endif
850          Bool zeroOffsets = ( scalEL.getWindowLeftOffset() == 0 && scalEL.getWindowRightOffset() == 0 && scalEL.getWindowTopOffset() == 0 && scalEL.getWindowBottomOffset() == 0 );
851
852#if VPS_EXTN_DIRECT_REF_LAYERS
853          TEncTop *pcEncTopBase = (TEncTop *)getRefLayerEnc( i );
854#else
855          TEncTop *pcEncTopBase = (TEncTop *)getLayerEnc( m_layerId-1 );
856#endif
857#if O0194_DIFFERENT_BITDEPTH_EL_BL
858          UInt refLayerId = m_cVPS.getRefLayerId(m_layerId, i);
859          Bool sameBitDepths = ( g_bitDepthYLayer[m_layerId] == g_bitDepthYLayer[refLayerId] ) && ( g_bitDepthCLayer[m_layerId] == g_bitDepthCLayer[refLayerId] );
860
861          if( m_iSourceWidth != pcEncTopBase->getSourceWidth() || m_iSourceHeight != pcEncTopBase->getSourceHeight() || !zeroOffsets || !sameBitDepths
862#if Q0048_CGS_3D_ASYMLUT
863            || m_cPPS.getCGSFlag() > 0
864#endif
865#if LAYER_CTB
866            || pcEncTopBase->getSPS()->getMaxCUWidth() != m_cSPS.getMaxCUWidth() || pcEncTopBase->getSPS()->getMaxCUHeight() != m_cSPS.getMaxCUHeight() || pcEncTopBase->getSPS()->getMaxCUDepth() != m_cSPS.getMaxCUDepth()
867#endif
868)
869#else
870          if(m_iSourceWidth != pcEncTopBase->getSourceWidth() || m_iSourceHeight != pcEncTopBase->getSourceHeight() || !zeroOffsets )
871#endif
872          {
873            rpcPic->setSpatialEnhLayerFlag( i, true );
874
875            //only for scalable extension
876#if SCALABILITY_MASK_E0104
877            assert( m_cVPS.getScalabilityMask(2) == true );
878#else
879            assert( m_cVPS.getScalabilityMask(1) == true );
880#endif
881          }
882        }
883      }
884#endif
885
886#if SVC_EXTENSION
887#if AUXILIARY_PICTURES
888      rpcPic->create( m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, 
889                      m_conformanceWindow, m_defaultDisplayWindow, m_numReorderPics, &m_cSPS);
890#else
891      rpcPic->create( m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, 
892                      m_conformanceWindow, m_defaultDisplayWindow, m_numReorderPics, &m_cSPS);
893#endif
894#else  //SVC_EXTENSION
895      rpcPic->create( m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, 
896                      m_conformanceWindow, m_defaultDisplayWindow, m_numReorderPics);
897#endif //SVC_EXTENSION
898    }
899    m_cListPic.pushBack( rpcPic );
900  }
901  rpcPic->setReconMark (false);
902 
903  m_iPOCLast++;
904  m_iNumPicRcvd++;
905 
906  rpcPic->getSlice(0)->setPOC( m_iPOCLast );
907  // mark it should be extended
908  rpcPic->getPicYuvRec()->setBorderExtension(false);
909}
910
911Void TEncTop::xInitSPS()
912{
913#if SVC_EXTENSION
914#if Q0078_ADD_LAYER_SETS
915  if (getNumDirectRefLayers() == 0 && getNumAddLayerSets() > 0)
916  {
917    m_cSPS.setLayerId(0); // layer ID 0 for independent layers
918  }
919  else
920  {
921    m_cSPS.setLayerId(m_layerId);
922  }
923#else
924  m_cSPS.setLayerId(m_layerId);
925#endif
926  m_cSPS.setNumScaledRefLayerOffsets(m_numScaledRefLayerOffsets);
927  for(Int i = 0; i < m_cSPS.getNumScaledRefLayerOffsets(); i++)
928  {
929#if O0098_SCALED_REF_LAYER_ID
930    m_cSPS.setScaledRefLayerId(i, m_scaledRefLayerId[i]);
931#endif
932    m_cSPS.getScaledRefLayerWindow(i) = m_scaledRefLayerWindow[i];
933#if P0312_VERT_PHASE_ADJ
934    m_cSPS.setVertPhasePositionEnableFlag( m_scaledRefLayerId[i], m_scaledRefLayerWindow[i].getVertPhasePositionEnableFlag() );
935#endif
936  }
937#endif //SVC_EXTENSION
938  ProfileTierLevel& profileTierLevel = *m_cSPS.getPTL()->getGeneralPTL();
939  profileTierLevel.setLevelIdc(m_level);
940  profileTierLevel.setTierFlag(m_levelTier);
941  profileTierLevel.setProfileIdc(m_profile);
942  profileTierLevel.setProfileCompatibilityFlag(m_profile, 1);
943  profileTierLevel.setProgressiveSourceFlag(m_progressiveSourceFlag);
944  profileTierLevel.setInterlacedSourceFlag(m_interlacedSourceFlag);
945  profileTierLevel.setNonPackedConstraintFlag(m_nonPackedConstraintFlag);
946  profileTierLevel.setFrameOnlyConstraintFlag(m_frameOnlyConstraintFlag);
947 
948  if (m_profile == Profile::MAIN10 && g_bitDepthY == 8 && g_bitDepthC == 8)
949  {
950    /* The above constraint is equal to Profile::MAIN */
951    profileTierLevel.setProfileCompatibilityFlag(Profile::MAIN, 1);
952  }
953  if (m_profile == Profile::MAIN)
954  {
955    /* A Profile::MAIN10 decoder can always decode Profile::MAIN */
956    profileTierLevel.setProfileCompatibilityFlag(Profile::MAIN10, 1);
957  }
958  /* XXX: should Main be marked as compatible with still picture? */
959  /* XXX: may be a good idea to refactor the above into a function
960   * that chooses the actual compatibility based upon options */
961
962  m_cSPS.setPicWidthInLumaSamples         ( m_iSourceWidth      );
963  m_cSPS.setPicHeightInLumaSamples        ( m_iSourceHeight     );
964  m_cSPS.setConformanceWindow             ( m_conformanceWindow );
965  m_cSPS.setMaxCUWidth    ( g_uiMaxCUWidth      );
966  m_cSPS.setMaxCUHeight   ( g_uiMaxCUHeight     );
967  m_cSPS.setMaxCUDepth    ( g_uiMaxCUDepth      );
968#if AUXILIARY_PICTURES
969  m_cSPS.setChromaFormatIdc( m_chromaFormatIDC);
970#endif
971
972  Int minCUSize = m_cSPS.getMaxCUWidth() >> ( m_cSPS.getMaxCUDepth()-g_uiAddCUDepth );
973  Int log2MinCUSize = 0;
974  while(minCUSize > 1)
975  {
976    minCUSize >>= 1;
977    log2MinCUSize++;
978  }
979
980  m_cSPS.setLog2MinCodingBlockSize(log2MinCUSize);
981  m_cSPS.setLog2DiffMaxMinCodingBlockSize(m_cSPS.getMaxCUDepth()-g_uiAddCUDepth);
982#if SVC_EXTENSION
983  m_cSPS.setSPSId         ( m_iSPSIdCnt       );
984#endif
985 
986  m_cSPS.setPCMLog2MinSize (m_uiPCMLog2MinSize);
987  m_cSPS.setUsePCM        ( m_usePCM           );
988  m_cSPS.setPCMLog2MaxSize( m_pcmLog2MaxSize  );
989
990  m_cSPS.setQuadtreeTULog2MaxSize( m_uiQuadtreeTULog2MaxSize );
991  m_cSPS.setQuadtreeTULog2MinSize( m_uiQuadtreeTULog2MinSize );
992  m_cSPS.setQuadtreeTUMaxDepthInter( m_uiQuadtreeTUMaxDepthInter    );
993  m_cSPS.setQuadtreeTUMaxDepthIntra( m_uiQuadtreeTUMaxDepthIntra    );
994 
995  m_cSPS.setTMVPFlagsPresent(false);
996
997  m_cSPS.setMaxTrSize   ( 1 << m_uiQuadtreeTULog2MaxSize );
998 
999  Int i;
1000 
1001  for (i = 0; i < g_uiMaxCUDepth-g_uiAddCUDepth; i++ )
1002  {
1003    m_cSPS.setAMPAcc( i, m_useAMP );
1004    //m_cSPS.setAMPAcc( i, 1 );
1005  }
1006
1007  m_cSPS.setUseAMP ( m_useAMP );
1008
1009  for (i = g_uiMaxCUDepth-g_uiAddCUDepth; i < g_uiMaxCUDepth; i++ )
1010  {
1011    m_cSPS.setAMPAcc(i, 0);
1012  }
1013
1014#if REPN_FORMAT_IN_VPS
1015  m_cSPS.setBitDepthY( m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_layerId ) )->getBitDepthVpsLuma() );
1016  m_cSPS.setBitDepthC( m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_layerId ) )->getBitDepthVpsChroma()  );
1017
1018  m_cSPS.setQpBDOffsetY ( 6*(m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_layerId ) )->getBitDepthVpsLuma()  - 8) );
1019  m_cSPS.setQpBDOffsetC ( 6*(m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_layerId ) )->getBitDepthVpsChroma()  - 8) );
1020#else
1021  m_cSPS.setBitDepthY( g_bitDepthY );
1022  m_cSPS.setBitDepthC( g_bitDepthC );
1023
1024  m_cSPS.setQpBDOffsetY ( 6*(g_bitDepthY - 8) );
1025  m_cSPS.setQpBDOffsetC ( 6*(g_bitDepthC - 8) );
1026#endif
1027
1028  m_cSPS.setUseSAO( m_bUseSAO );
1029
1030  m_cSPS.setMaxTLayers( m_maxTempLayer );
1031  m_cSPS.setTemporalIdNestingFlag( ( m_maxTempLayer == 1 ) ? true : false );
1032  for ( i = 0; i < min(m_cSPS.getMaxTLayers(),(UInt) MAX_TLAYER); i++ )
1033  {
1034    m_cSPS.setMaxDecPicBuffering(m_maxDecPicBuffering[i], i);
1035    m_cSPS.setNumReorderPics(m_numReorderPics[i], i);
1036  }
1037  m_cSPS.setPCMBitDepthLuma (g_uiPCMBitDepthLuma);
1038  m_cSPS.setPCMBitDepthChroma (g_uiPCMBitDepthChroma);
1039  m_cSPS.setPCMFilterDisableFlag  ( m_bPCMFilterDisableFlag );
1040
1041  m_cSPS.setScalingListFlag ( (m_useScalingListId == 0) ? 0 : 1 );
1042
1043  m_cSPS.setUseStrongIntraSmoothing( m_useStrongIntraSmoothing );
1044
1045  m_cSPS.setVuiParametersPresentFlag(getVuiParametersPresentFlag());
1046  if (m_cSPS.getVuiParametersPresentFlag())
1047  {
1048    TComVUI* pcVUI = m_cSPS.getVuiParameters();
1049    pcVUI->setAspectRatioInfoPresentFlag(getAspectRatioIdc() != -1);
1050    pcVUI->setAspectRatioIdc(getAspectRatioIdc());
1051    pcVUI->setSarWidth(getSarWidth());
1052    pcVUI->setSarHeight(getSarHeight());
1053    pcVUI->setOverscanInfoPresentFlag(getOverscanInfoPresentFlag());
1054    pcVUI->setOverscanAppropriateFlag(getOverscanAppropriateFlag());
1055    pcVUI->setVideoSignalTypePresentFlag(getVideoSignalTypePresentFlag());
1056    pcVUI->setVideoFormat(getVideoFormat());
1057    pcVUI->setVideoFullRangeFlag(getVideoFullRangeFlag());
1058    pcVUI->setColourDescriptionPresentFlag(getColourDescriptionPresentFlag());
1059    pcVUI->setColourPrimaries(getColourPrimaries());
1060    pcVUI->setTransferCharacteristics(getTransferCharacteristics());
1061    pcVUI->setMatrixCoefficients(getMatrixCoefficients());
1062    pcVUI->setChromaLocInfoPresentFlag(getChromaLocInfoPresentFlag());
1063    pcVUI->setChromaSampleLocTypeTopField(getChromaSampleLocTypeTopField());
1064    pcVUI->setChromaSampleLocTypeBottomField(getChromaSampleLocTypeBottomField());
1065    pcVUI->setNeutralChromaIndicationFlag(getNeutralChromaIndicationFlag());
1066    pcVUI->setDefaultDisplayWindow(getDefaultDisplayWindow());
1067    pcVUI->setFrameFieldInfoPresentFlag(getFrameFieldInfoPresentFlag());
1068    pcVUI->setFieldSeqFlag(false);
1069    pcVUI->setHrdParametersPresentFlag(false);
1070    pcVUI->getTimingInfo()->setPocProportionalToTimingFlag(getPocProportionalToTimingFlag());
1071    pcVUI->getTimingInfo()->setNumTicksPocDiffOneMinus1   (getNumTicksPocDiffOneMinus1()   );
1072    pcVUI->setBitstreamRestrictionFlag(getBitstreamRestrictionFlag());
1073    pcVUI->setTilesFixedStructureFlag(getTilesFixedStructureFlag());
1074    pcVUI->setMotionVectorsOverPicBoundariesFlag(getMotionVectorsOverPicBoundariesFlag());
1075    pcVUI->setMinSpatialSegmentationIdc(getMinSpatialSegmentationIdc());
1076    pcVUI->setMaxBytesPerPicDenom(getMaxBytesPerPicDenom());
1077    pcVUI->setMaxBitsPerMinCuDenom(getMaxBitsPerMinCuDenom());
1078    pcVUI->setLog2MaxMvLengthHorizontal(getLog2MaxMvLengthHorizontal());
1079    pcVUI->setLog2MaxMvLengthVertical(getLog2MaxMvLengthVertical());
1080  }
1081}
1082
1083Void TEncTop::xInitPPS()
1084{
1085  m_cPPS.setConstrainedIntraPred( m_bUseConstrainedIntraPred );
1086  Bool bUseDQP = (getMaxCuDQPDepth() > 0)? true : false;
1087
1088  if((getMaxDeltaQP() != 0 )|| getUseAdaptiveQP())
1089  {
1090    bUseDQP = true;
1091  }
1092
1093  if(bUseDQP)
1094  {
1095    m_cPPS.setUseDQP(true);
1096    m_cPPS.setMaxCuDQPDepth( m_iMaxCuDQPDepth );
1097    m_cPPS.setMinCuDQPSize( m_cPPS.getSPS()->getMaxCUWidth() >> ( m_cPPS.getMaxCuDQPDepth()) );
1098  }
1099  else
1100  {
1101    m_cPPS.setUseDQP(false);
1102    m_cPPS.setMaxCuDQPDepth( 0 );
1103    m_cPPS.setMinCuDQPSize( m_cPPS.getSPS()->getMaxCUWidth() >> ( m_cPPS.getMaxCuDQPDepth()) );
1104  }
1105
1106  if ( m_RCEnableRateControl )
1107  {
1108    m_cPPS.setUseDQP(true);
1109    m_cPPS.setMaxCuDQPDepth( 0 );
1110    m_cPPS.setMinCuDQPSize( m_cPPS.getSPS()->getMaxCUWidth() >> ( m_cPPS.getMaxCuDQPDepth()) );
1111  } 
1112
1113  m_cPPS.setChromaCbQpOffset( m_chromaCbQpOffset );
1114  m_cPPS.setChromaCrQpOffset( m_chromaCrQpOffset );
1115
1116  m_cPPS.setNumSubstreams(m_iWaveFrontSubstreams);
1117  m_cPPS.setEntropyCodingSyncEnabledFlag( m_iWaveFrontSynchro > 0 );
1118  m_cPPS.setTilesEnabledFlag( (m_iNumColumnsMinus1 > 0 || m_iNumRowsMinus1 > 0) );
1119  m_cPPS.setUseWP( m_useWeightedPred );
1120  m_cPPS.setWPBiPred( m_useWeightedBiPred );
1121  m_cPPS.setOutputFlagPresentFlag( false );
1122  m_cPPS.setSignHideFlag(getSignHideFlag());
1123  if ( getDeblockingFilterMetric() )
1124  {
1125    m_cPPS.setDeblockingFilterControlPresentFlag (true);
1126    m_cPPS.setDeblockingFilterOverrideEnabledFlag(true);
1127    m_cPPS.setPicDisableDeblockingFilterFlag(false);
1128    m_cPPS.setDeblockingFilterBetaOffsetDiv2(0);
1129    m_cPPS.setDeblockingFilterTcOffsetDiv2(0);
1130  } 
1131  else
1132  {
1133  m_cPPS.setDeblockingFilterControlPresentFlag (m_DeblockingFilterControlPresent );
1134  }
1135  m_cPPS.setLog2ParallelMergeLevelMinus2   (m_log2ParallelMergeLevelMinus2 );
1136  m_cPPS.setCabacInitPresentFlag(CABAC_INIT_PRESENT_FLAG);
1137  m_cPPS.setLoopFilterAcrossSlicesEnabledFlag( m_bLFCrossSliceBoundaryFlag );
1138  Int histogram[MAX_NUM_REF + 1];
1139  for( Int i = 0; i <= MAX_NUM_REF; i++ )
1140  {
1141    histogram[i]=0;
1142  }
1143  for( Int i = 0; i < getGOPSize(); i++ )
1144  {
1145    assert(getGOPEntry(i).m_numRefPicsActive >= 0 && getGOPEntry(i).m_numRefPicsActive <= MAX_NUM_REF);
1146    histogram[getGOPEntry(i).m_numRefPicsActive]++;
1147  }
1148  Int maxHist=-1;
1149  Int bestPos=0;
1150  for( Int i = 0; i <= MAX_NUM_REF; i++ )
1151  {
1152    if(histogram[i]>maxHist)
1153    {
1154      maxHist=histogram[i];
1155      bestPos=i;
1156    }
1157  }
1158  assert(bestPos <= 15);
1159  m_cPPS.setNumRefIdxL0DefaultActive(bestPos);
1160  m_cPPS.setNumRefIdxL1DefaultActive(bestPos);
1161  m_cPPS.setTransquantBypassEnableFlag(getTransquantBypassEnableFlag());
1162  m_cPPS.setUseTransformSkip( m_useTransformSkip );
1163  if (m_sliceSegmentMode)
1164  {
1165    m_cPPS.setDependentSliceSegmentsEnabledFlag( true );
1166  }
1167  if( m_cPPS.getDependentSliceSegmentsEnabledFlag() )
1168  {
1169    Int NumCtx = m_cPPS.getEntropyCodingSyncEnabledFlag()?2:1;
1170    m_cSliceEncoder.initCtxMem( NumCtx );
1171    for ( UInt st = 0; st < NumCtx; st++ )
1172    {
1173      TEncSbac* ctx = NULL;
1174      ctx = new TEncSbac;
1175      ctx->init( &m_cBinCoderCABAC );
1176      m_cSliceEncoder.setCtxMem( ctx, st );
1177    }
1178  }
1179#if SVC_EXTENSION
1180#if SCALINGLIST_INFERRING
1181  m_cPPS.setLayerId( m_layerId );
1182#endif
1183
1184#if Q0078_ADD_LAYER_SETS
1185  if (getNumDirectRefLayers() == 0 && getNumAddLayerSets() > 0)
1186  {
1187    m_cPPS.setLayerId(0); // layer ID 0 for independent layers
1188  }
1189#endif
1190
1191  if( m_layerId > 0 )
1192  {
1193    m_cPPS.setListsModificationPresentFlag(true);
1194    m_cPPS.setExtensionFlag(true);
1195  }
1196  else
1197  {
1198    m_cPPS.setListsModificationPresentFlag(false);
1199    m_cPPS.setExtensionFlag(false);
1200  }
1201
1202  m_cPPS.setPPSId( m_iPPSIdCnt );
1203  m_cPPS.setSPSId( m_iSPSIdCnt );
1204#if POC_RESET_FLAG
1205  m_cPPS.setNumExtraSliceHeaderBits( 2 );
1206#endif
1207#if O0149_CROSS_LAYER_BLA_FLAG
1208  if (m_crossLayerBLAFlag)
1209  {
1210    m_cPPS.setNumExtraSliceHeaderBits( 3 );
1211  }
1212#endif
1213#if Q0048_CGS_3D_ASYMLUT
1214  m_cPPS.setCGSFlag( m_nCGSFlag );
1215#endif
1216#endif //SVC_EXTENSION
1217}
1218
1219//Function for initializing m_RPSList, a list of TComReferencePictureSet, based on the GOPEntry objects read from the config file.
1220Void TEncTop::xInitRPS(Bool isFieldCoding)
1221{
1222  TComReferencePictureSet*      rps;
1223 
1224  m_cSPS.createRPSList(getGOPSize()+m_extraRPSs+1);
1225  TComRPSList* rpsList = m_cSPS.getRPSList();
1226
1227  for( Int i = 0; i < getGOPSize()+m_extraRPSs; i++) 
1228  {
1229    GOPEntry ge = getGOPEntry(i);
1230    rps = rpsList->getReferencePictureSet(i);
1231    rps->setNumberOfPictures(ge.m_numRefPics);
1232    rps->setNumRefIdc(ge.m_numRefIdc);
1233    Int numNeg = 0;
1234    Int numPos = 0;
1235    for( Int j = 0; j < ge.m_numRefPics; j++)
1236    {
1237      rps->setDeltaPOC(j,ge.m_referencePics[j]);
1238      rps->setUsed(j,ge.m_usedByCurrPic[j]);
1239      if(ge.m_referencePics[j]>0)
1240      {
1241        numPos++;
1242      }
1243      else
1244      {
1245        numNeg++;
1246      }
1247    }
1248    rps->setNumberOfNegativePictures(numNeg);
1249    rps->setNumberOfPositivePictures(numPos);
1250
1251    // handle inter RPS intialization from the config file.
1252#if AUTO_INTER_RPS
1253    rps->setInterRPSPrediction(ge.m_interRPSPrediction > 0);  // not very clean, converting anything > 0 to true.
1254    rps->setDeltaRIdxMinus1(0);                               // index to the Reference RPS is always the previous one.
1255    TComReferencePictureSet*     RPSRef = rpsList->getReferencePictureSet(i-1);  // get the reference RPS
1256
1257    if (ge.m_interRPSPrediction == 2)  // Automatic generation of the inter RPS idc based on the RIdx provided.
1258    {
1259      Int deltaRPS = getGOPEntry(i-1).m_POC - ge.m_POC;  // the ref POC - current POC
1260      Int numRefDeltaPOC = RPSRef->getNumberOfPictures();
1261
1262      rps->setDeltaRPS(deltaRPS);           // set delta RPS
1263      rps->setNumRefIdc(numRefDeltaPOC+1);  // set the numRefIdc to the number of pictures in the reference RPS + 1.
1264      Int count=0;
1265      for (Int j = 0; j <= numRefDeltaPOC; j++ ) // cycle through pics in reference RPS.
1266      {
1267        Int RefDeltaPOC = (j<numRefDeltaPOC)? RPSRef->getDeltaPOC(j): 0;  // if it is the last decoded picture, set RefDeltaPOC = 0
1268        rps->setRefIdc(j, 0);
1269        for (Int k = 0; k < rps->getNumberOfPictures(); k++ )  // cycle through pics in current RPS.
1270        {
1271          if (rps->getDeltaPOC(k) == ( RefDeltaPOC + deltaRPS))  // if the current RPS has a same picture as the reference RPS.
1272          {
1273              rps->setRefIdc(j, (rps->getUsed(k)?1:2));
1274              count++;
1275              break;
1276          }
1277        }
1278      }
1279      if (count != rps->getNumberOfPictures())
1280      {
1281        printf("Warning: Unable fully predict all delta POCs using the reference RPS index given in the config file.  Setting Inter RPS to false for this RPS.\n");
1282        rps->setInterRPSPrediction(0);
1283      }
1284    }
1285    else if (ge.m_interRPSPrediction == 1)  // inter RPS idc based on the RefIdc values provided in config file.
1286    {
1287      rps->setDeltaRPS(ge.m_deltaRPS);
1288      rps->setNumRefIdc(ge.m_numRefIdc);
1289      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1290      {
1291        rps->setRefIdc(j, ge.m_refIdc[j]);
1292      }
1293#if WRITE_BACK
1294      // the folowing code overwrite the deltaPOC and Used by current values read from the config file with the ones
1295      // computed from the RefIdc.  A warning is printed if they are not identical.
1296      numNeg = 0;
1297      numPos = 0;
1298      TComReferencePictureSet      RPSTemp;  // temporary variable
1299
1300      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1301      {
1302        if (ge.m_refIdc[j])
1303        {
1304          Int deltaPOC = ge.m_deltaRPS + ((j < RPSRef->getNumberOfPictures())? RPSRef->getDeltaPOC(j) : 0);
1305          RPSTemp.setDeltaPOC((numNeg+numPos),deltaPOC);
1306          RPSTemp.setUsed((numNeg+numPos),ge.m_refIdc[j]==1?1:0);
1307          if (deltaPOC<0)
1308          {
1309            numNeg++;
1310          }
1311          else
1312          {
1313            numPos++;
1314          }
1315        }
1316      }
1317      if (numNeg != rps->getNumberOfNegativePictures())
1318      {
1319        printf("Warning: number of negative pictures in RPS is different between intra and inter RPS specified in the config file.\n");
1320        rps->setNumberOfNegativePictures(numNeg);
1321        rps->setNumberOfPictures(numNeg+numPos);
1322      }
1323      if (numPos != rps->getNumberOfPositivePictures())
1324      {
1325        printf("Warning: number of positive pictures in RPS is different between intra and inter RPS specified in the config file.\n");
1326        rps->setNumberOfPositivePictures(numPos);
1327        rps->setNumberOfPictures(numNeg+numPos);
1328      }
1329      RPSTemp.setNumberOfPictures(numNeg+numPos);
1330      RPSTemp.setNumberOfNegativePictures(numNeg);
1331      RPSTemp.sortDeltaPOC();     // sort the created delta POC before comparing
1332      // check if Delta POC and Used are the same
1333      // print warning if they are not.
1334      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1335      {
1336        if (RPSTemp.getDeltaPOC(j) != rps->getDeltaPOC(j))
1337        {
1338          printf("Warning: delta POC is different between intra RPS and inter RPS specified in the config file.\n");
1339          rps->setDeltaPOC(j,RPSTemp.getDeltaPOC(j));
1340        }
1341        if (RPSTemp.getUsed(j) != rps->getUsed(j))
1342        {
1343          printf("Warning: Used by Current in RPS is different between intra and inter RPS specified in the config file.\n");
1344          rps->setUsed(j,RPSTemp.getUsed(j));
1345        }
1346      }
1347#endif
1348    }
1349#else
1350    rps->setInterRPSPrediction(ge.m_interRPSPrediction);
1351    if (ge.m_interRPSPrediction)
1352    {
1353      rps->setDeltaRIdxMinus1(0);
1354      rps->setDeltaRPS(ge.m_deltaRPS);
1355      rps->setNumRefIdc(ge.m_numRefIdc);
1356      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1357      {
1358        rps->setRefIdc(j, ge.m_refIdc[j]);
1359      }
1360#if WRITE_BACK
1361      // the folowing code overwrite the deltaPOC and Used by current values read from the config file with the ones
1362      // computed from the RefIdc.  This is not necessary if both are identical. Currently there is no check to see if they are identical.
1363      numNeg = 0;
1364      numPos = 0;
1365      TComReferencePictureSet*     RPSRef = m_RPSList.getReferencePictureSet(i-1);
1366
1367      for (Int j = 0; j < ge.m_numRefIdc; j++ )
1368      {
1369        if (ge.m_refIdc[j])
1370        {
1371          Int deltaPOC = ge.m_deltaRPS + ((j < RPSRef->getNumberOfPictures())? RPSRef->getDeltaPOC(j) : 0);
1372          rps->setDeltaPOC((numNeg+numPos),deltaPOC);
1373          rps->setUsed((numNeg+numPos),ge.m_refIdc[j]==1?1:0);
1374          if (deltaPOC<0)
1375          {
1376            numNeg++;
1377          }
1378          else
1379          {
1380            numPos++;
1381          }
1382        }
1383      }
1384      rps->setNumberOfNegativePictures(numNeg);
1385      rps->setNumberOfPositivePictures(numPos);
1386      rps->sortDeltaPOC();
1387#endif
1388    }
1389#endif //INTER_RPS_AUTO
1390  }
1391  //In case of field coding, we need to set special parameters for the first bottom field of the sequence, since it is not specified in the cfg file.
1392  //The position = GOPSize + extraRPSs which is (a priori) unused is reserved for this field in the RPS.
1393  if (isFieldCoding) 
1394  {
1395    rps = rpsList->getReferencePictureSet(getGOPSize()+m_extraRPSs);
1396    rps->setNumberOfPictures(1);
1397    rps->setNumberOfNegativePictures(1);
1398    rps->setNumberOfPositivePictures(0);
1399    rps->setNumberOfLongtermPictures(0);
1400    rps->setDeltaPOC(0,-1);
1401    rps->setPOC(0,0);
1402    rps->setUsed(0,true);
1403    rps->setInterRPSPrediction(false);
1404    rps->setDeltaRIdxMinus1(0);
1405    rps->setDeltaRPS(0);
1406    rps->setNumRefIdc(0);
1407}
1408}
1409
1410   // This is a function that
1411   // determines what Reference Picture Set to use
1412   // for a specific slice (with POC = POCCurr)
1413Void TEncTop::selectReferencePictureSet(TComSlice* slice, Int POCCurr, Int GOPid )
1414{
1415  slice->setRPSidx(GOPid);
1416  for(Int extraNum=m_iGOPSize; extraNum<m_extraRPSs+m_iGOPSize; extraNum++)
1417  {   
1418    if(m_uiIntraPeriod > 0 && getDecodingRefreshType() > 0)
1419    {
1420      Int POCIndex = POCCurr%m_uiIntraPeriod;
1421      if(POCIndex == 0)
1422      {
1423        POCIndex = m_uiIntraPeriod;
1424      }
1425      if(POCIndex == m_GOPList[extraNum].m_POC)
1426      {
1427        slice->setRPSidx(extraNum);
1428      }
1429    }
1430    else
1431    {
1432      if(POCCurr==m_GOPList[extraNum].m_POC)
1433      {
1434        slice->setRPSidx(extraNum);
1435      }
1436    }
1437  }
1438
1439  if(POCCurr == 1 && slice->getPic()->isField())
1440  {
1441    slice->setRPSidx(m_iGOPSize+m_extraRPSs);
1442  }
1443
1444  slice->setRPS(getSPS()->getRPSList()->getReferencePictureSet(slice->getRPSidx()));
1445  slice->getRPS()->setNumberOfPictures(slice->getRPS()->getNumberOfNegativePictures()+slice->getRPS()->getNumberOfPositivePictures());
1446}
1447
1448Int TEncTop::getReferencePictureSetIdxForSOP(TComSlice* slice, Int POCCurr, Int GOPid )
1449{
1450  int rpsIdx = GOPid;
1451
1452  for(Int extraNum=m_iGOPSize; extraNum<m_extraRPSs+m_iGOPSize; extraNum++)
1453  {   
1454    if(m_uiIntraPeriod > 0 && getDecodingRefreshType() > 0)
1455    {
1456      Int POCIndex = POCCurr%m_uiIntraPeriod;
1457      if(POCIndex == 0)
1458      {
1459        POCIndex = m_uiIntraPeriod;
1460      }
1461      if(POCIndex == m_GOPList[extraNum].m_POC)
1462      {
1463        rpsIdx = extraNum;
1464      }
1465    }
1466    else
1467    {
1468      if(POCCurr==m_GOPList[extraNum].m_POC)
1469      {
1470        rpsIdx = extraNum;
1471      }
1472    }
1473  }
1474
1475  return rpsIdx;
1476}
1477
1478Void  TEncTop::xInitPPSforTiles()
1479{
1480  m_cPPS.setUniformSpacingFlag( m_iUniformSpacingIdr );
1481  m_cPPS.setNumColumnsMinus1( m_iNumColumnsMinus1 );
1482  m_cPPS.setNumRowsMinus1( m_iNumRowsMinus1 );
1483  if( m_iUniformSpacingIdr == 0 )
1484  {
1485    m_cPPS.setColumnWidth( m_puiColumnWidth );
1486    m_cPPS.setRowHeight( m_puiRowHeight );
1487  }
1488  m_cPPS.setLoopFilterAcrossTilesEnabledFlag( m_loopFilterAcrossTilesEnabledFlag );
1489
1490  // # substreams is "per tile" when tiles are independent.
1491  if (m_iWaveFrontSynchro
1492    )
1493  {
1494    m_cPPS.setNumSubstreams(m_iWaveFrontSubstreams * (m_iNumColumnsMinus1+1));
1495  }
1496}
1497
1498Void  TEncCfg::xCheckGSParameters()
1499{
1500  Int   iWidthInCU = ( m_iSourceWidth%g_uiMaxCUWidth ) ? m_iSourceWidth/g_uiMaxCUWidth + 1 : m_iSourceWidth/g_uiMaxCUWidth;
1501  Int   iHeightInCU = ( m_iSourceHeight%g_uiMaxCUHeight ) ? m_iSourceHeight/g_uiMaxCUHeight + 1 : m_iSourceHeight/g_uiMaxCUHeight;
1502  UInt  uiCummulativeColumnWidth = 0;
1503  UInt  uiCummulativeRowHeight = 0;
1504
1505  //check the column relative parameters
1506  if( m_iNumColumnsMinus1 >= (1<<(LOG2_MAX_NUM_COLUMNS_MINUS1+1)) )
1507  {
1508    printf( "The number of columns is larger than the maximum allowed number of columns.\n" );
1509    exit( EXIT_FAILURE );
1510  }
1511
1512  if( m_iNumColumnsMinus1 >= iWidthInCU )
1513  {
1514    printf( "The current picture can not have so many columns.\n" );
1515    exit( EXIT_FAILURE );
1516  }
1517
1518  if( m_iNumColumnsMinus1 && m_iUniformSpacingIdr==0 )
1519  {
1520    for(Int i=0; i<m_iNumColumnsMinus1; i++)
1521    {
1522      uiCummulativeColumnWidth += m_puiColumnWidth[i];
1523    }
1524
1525    if( uiCummulativeColumnWidth >= iWidthInCU )
1526    {
1527      printf( "The width of the column is too large.\n" );
1528      exit( EXIT_FAILURE );
1529    }
1530  }
1531
1532  //check the row relative parameters
1533  if( m_iNumRowsMinus1 >= (1<<(LOG2_MAX_NUM_ROWS_MINUS1+1)) )
1534  {
1535    printf( "The number of rows is larger than the maximum allowed number of rows.\n" );
1536    exit( EXIT_FAILURE );
1537  }
1538
1539  if( m_iNumRowsMinus1 >= iHeightInCU )
1540  {
1541    printf( "The current picture can not have so many rows.\n" );
1542    exit( EXIT_FAILURE );
1543  }
1544
1545  if( m_iNumRowsMinus1 && m_iUniformSpacingIdr==0 )
1546  {
1547    for(Int i=0; i<m_iNumRowsMinus1; i++)
1548    {
1549      uiCummulativeRowHeight += m_puiRowHeight[i];
1550    }
1551
1552    if( uiCummulativeRowHeight >= iHeightInCU )
1553    {
1554      printf( "The height of the row is too large.\n" );
1555      exit( EXIT_FAILURE );
1556    }
1557  }
1558}
1559
1560#if SVC_EXTENSION
1561#if VPS_EXTN_DIRECT_REF_LAYERS
1562TEncTop* TEncTop::getRefLayerEnc( UInt refLayerIdc )
1563{
1564  if( m_ppcTEncTop[m_layerId]->getNumDirectRefLayers() <= 0 )
1565  {
1566    return (TEncTop *)getLayerEnc( 0 );
1567  }
1568
1569  return (TEncTop *)getLayerEnc( m_cVPS.getRefLayerId( m_layerId, refLayerIdc ) );
1570}
1571#endif
1572
1573#if !REPN_FORMAT_IN_VPS
1574Void TEncTop::xInitILRP()
1575{
1576  if(m_layerId>0)
1577  {
1578    g_bitDepthY     = m_cSPS.getBitDepthY();
1579    g_bitDepthC     = m_cSPS.getBitDepthC();
1580    g_uiMaxCUWidth  = m_cSPS.getMaxCUWidth();
1581    g_uiMaxCUHeight = m_cSPS.getMaxCUHeight();
1582    g_uiMaxCUDepth  = m_cSPS.getMaxCUDepth();
1583    g_uiAddCUDepth  = max (0, m_cSPS.getLog2MinCodingBlockSize() - (Int)m_cSPS.getQuadtreeTULog2MinSize() );
1584
1585    Int  numReorderPics[MAX_TLAYER];
1586    Window &conformanceWindow = m_cSPS.getConformanceWindow();
1587    Window defaultDisplayWindow = m_cSPS.getVuiParametersPresentFlag() ? m_cSPS.getVuiParameters()->getDefaultDisplayWindow() : Window();
1588
1589    if (m_cIlpPic[0] == NULL)
1590    {
1591      for (Int j=0; j < m_numLayer; j++) // consider to set to NumDirectRefLayers[LayerIdInVps[nuh_layer_id]]
1592      {
1593        m_cIlpPic[j] = new  TComPic;
1594#if AUXILIARY_PICTURES
1595        m_cIlpPic[j]->create(m_iSourceWidth, m_iSourceHeight, m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, conformanceWindow, defaultDisplayWindow, numReorderPics, &m_cSPS, true);
1596#else
1597        m_cIlpPic[j]->create(m_iSourceWidth, m_iSourceHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, conformanceWindow, defaultDisplayWindow, numReorderPics, &m_cSPS, true);
1598#endif
1599        for (Int i=0; i<m_cIlpPic[j]->getPicSym()->getNumberOfCUsInFrame(); i++)
1600        {
1601          m_cIlpPic[j]->getPicSym()->getCU(i)->initCU(m_cIlpPic[j], i);
1602        }
1603      }
1604    }
1605  }
1606}
1607#else
1608Void TEncTop::xInitILRP()
1609{
1610#if O0096_REP_FORMAT_INDEX
1611  RepFormat *repFormat = m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_cSPS.getUpdateRepFormatFlag() ? m_cSPS.getUpdateRepFormatIndex() : m_layerId ) );
1612#else
1613  RepFormat *repFormat = m_cVPS.getVpsRepFormat( m_cVPS.getVpsRepFormatIdx( m_layerId ) );
1614#endif
1615  Int bitDepthY,bitDepthC,picWidth,picHeight;
1616
1617#if O0096_REP_FORMAT_INDEX
1618  bitDepthY   = repFormat->getBitDepthVpsLuma();
1619  bitDepthC   = repFormat->getBitDepthVpsChroma();
1620  picWidth    = repFormat->getPicWidthVpsInLumaSamples();
1621  picHeight   = repFormat->getPicHeightVpsInLumaSamples();
1622#else
1623  if( m_cSPS.getUpdateRepFormatFlag() )
1624  {
1625    bitDepthY   = m_cSPS.getBitDepthY();
1626    bitDepthC   = m_cSPS.getBitDepthC();
1627    picWidth    = m_cSPS.getPicWidthInLumaSamples();
1628    picHeight   = m_cSPS.getPicHeightInLumaSamples();
1629  }
1630  else
1631  {
1632    bitDepthY   = repFormat->getBitDepthVpsLuma();
1633    bitDepthC   = repFormat->getBitDepthVpsChroma();
1634    picWidth    = repFormat->getPicWidthVpsInLumaSamples();
1635    picHeight   = repFormat->getPicHeightVpsInLumaSamples();
1636  }
1637#endif
1638 
1639  if(m_layerId > 0)
1640  {
1641    g_bitDepthY     = bitDepthY;
1642    g_bitDepthC     = bitDepthC;
1643    g_uiMaxCUWidth  = m_cSPS.getMaxCUWidth();
1644    g_uiMaxCUHeight = m_cSPS.getMaxCUHeight();
1645    g_uiMaxCUDepth  = m_cSPS.getMaxCUDepth();
1646    g_uiAddCUDepth  = max (0, m_cSPS.getLog2MinCodingBlockSize() - (Int)m_cSPS.getQuadtreeTULog2MinSize() );
1647
1648    Int  numReorderPics[MAX_TLAYER];
1649    Window &conformanceWindow = m_cSPS.getConformanceWindow();
1650    Window defaultDisplayWindow = m_cSPS.getVuiParametersPresentFlag() ? m_cSPS.getVuiParameters()->getDefaultDisplayWindow() : Window();
1651
1652    if (m_cIlpPic[0] == NULL)
1653    {
1654      for (Int j=0; j < m_numDirectRefLayers; j++)
1655      {
1656        m_cIlpPic[j] = new  TComPic;
1657#if AUXILIARY_PICTURES
1658        m_cIlpPic[j]->create(picWidth, picHeight, m_chromaFormatIDC, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, conformanceWindow, defaultDisplayWindow, numReorderPics, &m_cSPS, true);
1659#else
1660        m_cIlpPic[j]->create(picWidth, picHeight, g_uiMaxCUWidth, g_uiMaxCUHeight, g_uiMaxCUDepth, conformanceWindow, defaultDisplayWindow, numReorderPics, &m_cSPS, true);
1661#endif
1662        for (Int i=0; i<m_cIlpPic[j]->getPicSym()->getNumberOfCUsInFrame(); i++)
1663        {
1664          m_cIlpPic[j]->getPicSym()->getCU(i)->initCU(m_cIlpPic[j], i);
1665        }
1666      }
1667    }
1668  }
1669}
1670#endif
1671
1672#if O0098_SCALED_REF_LAYER_ID
1673Window& TEncTop::getScaledRefLayerWindowForLayer(Int layerId)
1674{
1675  static Window win;
1676
1677  for (Int i = 0; i < m_numScaledRefLayerOffsets; i++)
1678  {
1679    if (layerId == m_scaledRefLayerId[i])
1680    {
1681      return m_scaledRefLayerWindow[i];
1682    }
1683  }
1684
1685  win.resetWindow();  // scaled reference layer offsets are inferred to be zero when not present
1686  return win;
1687}
1688#endif
1689
1690#endif //SVC_EXTENSION
1691//! \}
Note: See TracBrowser for help on using the repository browser.