source: 3DVCSoftware/trunk/source/Lib/TLibEncoder/TEncTop.cpp @ 665

Last change on this file since 665 was 655, checked in by tech, 11 years ago

Merged 8.1-Cleanup@654

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