source: 3DVCSoftware/branches/HTM-DEV-0.3-dev2a/source/Lib/TLibEncoder/TEncGOP.cpp @ 489

Last change on this file since 489 was 489, checked in by zhang, 11 years ago

Use m_ivPicLists instead of TAppE/DencTop encapsulated with H_3D_ARP_FIX

  • Property svn:eol-style set to native
File size: 112.3 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-2013, 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     TEncGOP.cpp
35    \brief    GOP encoder class
36*/
37
38#include <list>
39#include <algorithm>
40#include <functional>
41
42#include "TEncTop.h"
43#include "TEncGOP.h"
44#include "TEncAnalyze.h"
45#include "libmd5/MD5.h"
46#include "TLibCommon/SEI.h"
47#include "TLibCommon/NAL.h"
48#include "NALwrite.h"
49#if H_3D_ARP
50#include "../../App/TAppEncoder/TAppEncTop.h"
51#endif
52#include <time.h>
53#include <math.h>
54
55using namespace std;
56//! \ingroup TLibEncoder
57//! \{
58
59// ====================================================================================================================
60// Constructor / destructor / initialization / destroy
61// ====================================================================================================================
62Int getLSB(Int poc, Int maxLSB)
63{
64  if (poc >= 0)
65  {
66    return poc % maxLSB;
67  }
68  else
69  {
70    return (maxLSB - ((-poc) % maxLSB)) % maxLSB;
71  }
72}
73
74TEncGOP::TEncGOP()
75{
76  m_iLastIDR            = 0;
77  m_iGopSize            = 0;
78  m_iNumPicCoded        = 0; //Niko
79  m_bFirst              = true;
80 
81  m_pcCfg               = NULL;
82  m_pcSliceEncoder      = NULL;
83  m_pcListPic           = NULL;
84 
85  m_pcEntropyCoder      = NULL;
86  m_pcCavlcCoder        = NULL;
87  m_pcSbacCoder         = NULL;
88  m_pcBinCABAC          = NULL;
89 
90  m_bSeqFirst           = true;
91 
92  m_bRefreshPending     = 0;
93  m_pocCRA            = 0;
94  m_numLongTermRefPicSPS = 0;
95  ::memset(m_ltRefPicPocLsbSps, 0, sizeof(m_ltRefPicPocLsbSps));
96  ::memset(m_ltRefPicUsedByCurrPicFlag, 0, sizeof(m_ltRefPicUsedByCurrPicFlag));
97  m_cpbRemovalDelay   = 0;
98  m_lastBPSEI         = 0;
99#if L0045_NON_NESTED_SEI_RESTRICTIONS
100  xResetNonNestedSEIPresentFlags();
101#if K0180_SCALABLE_NESTING_SEI
102  xResetNestedSEIPresentFlags();
103#endif
104#endif
105#if H_MV
106  m_layerId      = 0;
107  m_viewId       = 0;
108  m_pocLastCoded = -1; 
109#if H_3D
110  m_viewIndex  =   0; 
111  m_isDepth = false;
112#endif
113#endif
114  return;
115}
116
117TEncGOP::~TEncGOP()
118{
119}
120
121/** Create list to contain pointers to LCU start addresses of slice.
122 */
123Void  TEncGOP::create()
124{
125  m_bLongtermTestPictureHasBeenCoded = 0;
126  m_bLongtermTestPictureHasBeenCoded2 = 0;
127}
128
129Void  TEncGOP::destroy()
130{
131}
132
133Void TEncGOP::init ( TEncTop* pcTEncTop )
134{
135  m_pcEncTop     = pcTEncTop;
136  m_pcCfg                = pcTEncTop;
137  m_pcSliceEncoder       = pcTEncTop->getSliceEncoder();
138  m_pcListPic            = pcTEncTop->getListPic();
139 
140  m_pcEntropyCoder       = pcTEncTop->getEntropyCoder();
141  m_pcCavlcCoder         = pcTEncTop->getCavlcCoder();
142  m_pcSbacCoder          = pcTEncTop->getSbacCoder();
143  m_pcBinCABAC           = pcTEncTop->getBinCABAC();
144  m_pcLoopFilter         = pcTEncTop->getLoopFilter();
145  m_pcBitCounter         = pcTEncTop->getBitCounter();
146 
147  //--Adaptive Loop filter
148  m_pcSAO                = pcTEncTop->getSAO();
149  m_pcRateCtrl           = pcTEncTop->getRateCtrl();
150  m_lastBPSEI          = 0;
151  m_totalCoded         = 0;
152
153#if H_MV
154  m_ivPicLists           = pcTEncTop->getIvPicLists(); 
155  m_layerId              = pcTEncTop->getLayerId();
156  m_viewId               = pcTEncTop->getViewId();
157#if H_3D
158  m_viewIndex            = pcTEncTop->getViewIndex();
159  m_isDepth              = pcTEncTop->getIsDepth();
160#endif
161#endif
162}
163
164SEIActiveParameterSets* TEncGOP::xCreateSEIActiveParameterSets (TComSPS *sps)
165{
166  SEIActiveParameterSets *seiActiveParameterSets = new SEIActiveParameterSets(); 
167  seiActiveParameterSets->activeVPSId = m_pcCfg->getVPS()->getVPSId(); 
168#if L0047_APS_FLAGS
169  seiActiveParameterSets->m_fullRandomAccessFlag = false;
170  seiActiveParameterSets->m_noParamSetUpdateFlag = false;
171#endif
172  seiActiveParameterSets->numSpsIdsMinus1 = 0;
173  seiActiveParameterSets->activeSeqParamSetId.resize(seiActiveParameterSets->numSpsIdsMinus1 + 1); 
174  seiActiveParameterSets->activeSeqParamSetId[0] = sps->getSPSId();
175  return seiActiveParameterSets;
176}
177
178SEIFramePacking* TEncGOP::xCreateSEIFramePacking()
179{
180  SEIFramePacking *seiFramePacking = new SEIFramePacking();
181  seiFramePacking->m_arrangementId = m_pcCfg->getFramePackingArrangementSEIId();
182  seiFramePacking->m_arrangementCancelFlag = 0;
183  seiFramePacking->m_arrangementType = m_pcCfg->getFramePackingArrangementSEIType();
184#if L0444_FPA_TYPE
185  assert((seiFramePacking->m_arrangementType > 2) && (seiFramePacking->m_arrangementType < 6) );
186#endif
187  seiFramePacking->m_quincunxSamplingFlag = m_pcCfg->getFramePackingArrangementSEIQuincunx();
188  seiFramePacking->m_contentInterpretationType = m_pcCfg->getFramePackingArrangementSEIInterpretation();
189  seiFramePacking->m_spatialFlippingFlag = 0;
190  seiFramePacking->m_frame0FlippedFlag = 0;
191  seiFramePacking->m_fieldViewsFlag = (seiFramePacking->m_arrangementType == 2);
192  seiFramePacking->m_currentFrameIsFrame0Flag = ((seiFramePacking->m_arrangementType == 5) && m_iNumPicCoded&1);
193  seiFramePacking->m_frame0SelfContainedFlag = 0;
194  seiFramePacking->m_frame1SelfContainedFlag = 0;
195  seiFramePacking->m_frame0GridPositionX = 0;
196  seiFramePacking->m_frame0GridPositionY = 0;
197  seiFramePacking->m_frame1GridPositionX = 0;
198  seiFramePacking->m_frame1GridPositionY = 0;
199  seiFramePacking->m_arrangementReservedByte = 0;
200#if L0045_PERSISTENCE_FLAGS
201  seiFramePacking->m_arrangementPersistenceFlag = true;
202#else
203  seiFramePacking->m_arrangementRepetetionPeriod = 1;
204#endif
205  seiFramePacking->m_upsampledAspectRatio = 0;
206  return seiFramePacking;
207}
208
209SEIDisplayOrientation* TEncGOP::xCreateSEIDisplayOrientation()
210{
211  SEIDisplayOrientation *seiDisplayOrientation = new SEIDisplayOrientation();
212  seiDisplayOrientation->cancelFlag = false;
213  seiDisplayOrientation->horFlip = false;
214  seiDisplayOrientation->verFlip = false;
215  seiDisplayOrientation->anticlockwiseRotation = m_pcCfg->getDisplayOrientationSEIAngle();
216  return seiDisplayOrientation;
217}
218
219#if J0149_TONE_MAPPING_SEI
220SEIToneMappingInfo*  TEncGOP::xCreateSEIToneMappingInfo()
221{
222  SEIToneMappingInfo *seiToneMappingInfo = new SEIToneMappingInfo();
223  seiToneMappingInfo->m_toneMapId = m_pcCfg->getTMISEIToneMapId();
224  seiToneMappingInfo->m_toneMapCancelFlag = m_pcCfg->getTMISEIToneMapCancelFlag();
225  seiToneMappingInfo->m_toneMapPersistenceFlag = m_pcCfg->getTMISEIToneMapPersistenceFlag();
226
227  seiToneMappingInfo->m_codedDataBitDepth = m_pcCfg->getTMISEICodedDataBitDepth();
228  assert(seiToneMappingInfo->m_codedDataBitDepth >= 8 && seiToneMappingInfo->m_codedDataBitDepth <= 14);
229  seiToneMappingInfo->m_targetBitDepth = m_pcCfg->getTMISEITargetBitDepth();
230  assert( (seiToneMappingInfo->m_targetBitDepth >= 1 && seiToneMappingInfo->m_targetBitDepth <= 17) || (seiToneMappingInfo->m_targetBitDepth  == 255) );
231  seiToneMappingInfo->m_modelId = m_pcCfg->getTMISEIModelID();
232  assert(seiToneMappingInfo->m_modelId >=0 &&seiToneMappingInfo->m_modelId<=4);
233
234  switch( seiToneMappingInfo->m_modelId)
235  {
236  case 0:
237    {
238      seiToneMappingInfo->m_minValue = m_pcCfg->getTMISEIMinValue();
239      seiToneMappingInfo->m_maxValue = m_pcCfg->getTMISEIMaxValue();
240      break;
241    }
242  case 1:
243    {
244      seiToneMappingInfo->m_sigmoidMidpoint = m_pcCfg->getTMISEISigmoidMidpoint();
245      seiToneMappingInfo->m_sigmoidWidth = m_pcCfg->getTMISEISigmoidWidth();
246      break;
247    }
248  case 2:
249    {
250      UInt num = 1u<<(seiToneMappingInfo->m_targetBitDepth);
251      seiToneMappingInfo->m_startOfCodedInterval.resize(num);
252      Int* ptmp = m_pcCfg->getTMISEIStartOfCodedInterva();
253      if(ptmp)
254      {
255        for(int i=0; i<num;i++)
256        {
257          seiToneMappingInfo->m_startOfCodedInterval[i] = ptmp[i];
258        }
259      }
260      break;
261    }
262  case 3:
263    {
264      seiToneMappingInfo->m_numPivots = m_pcCfg->getTMISEINumPivots();
265      seiToneMappingInfo->m_codedPivotValue.resize(seiToneMappingInfo->m_numPivots);
266      seiToneMappingInfo->m_targetPivotValue.resize(seiToneMappingInfo->m_numPivots);
267      Int* ptmpcoded = m_pcCfg->getTMISEICodedPivotValue();
268      Int* ptmptarget = m_pcCfg->getTMISEITargetPivotValue();
269      if(ptmpcoded&&ptmptarget)
270      {
271        for(int i=0; i<(seiToneMappingInfo->m_numPivots);i++)
272        {
273          seiToneMappingInfo->m_codedPivotValue[i]=ptmpcoded[i];
274          seiToneMappingInfo->m_targetPivotValue[i]=ptmptarget[i];
275         }
276       }
277       break;
278     }
279  case 4:
280     {
281       seiToneMappingInfo->m_cameraIsoSpeedIdc = m_pcCfg->getTMISEICameraIsoSpeedIdc();
282       seiToneMappingInfo->m_cameraIsoSpeedValue = m_pcCfg->getTMISEICameraIsoSpeedValue();
283       assert( seiToneMappingInfo->m_cameraIsoSpeedValue !=0 );
284       seiToneMappingInfo->m_exposureCompensationValueSignFlag = m_pcCfg->getTMISEIExposureCompensationValueSignFlag();
285       seiToneMappingInfo->m_exposureCompensationValueNumerator = m_pcCfg->getTMISEIExposureCompensationValueNumerator();
286       seiToneMappingInfo->m_exposureCompensationValueDenomIdc = m_pcCfg->getTMISEIExposureCompensationValueDenomIdc();
287       seiToneMappingInfo->m_refScreenLuminanceWhite = m_pcCfg->getTMISEIRefScreenLuminanceWhite();
288       seiToneMappingInfo->m_extendedRangeWhiteLevel = m_pcCfg->getTMISEIExtendedRangeWhiteLevel();
289       assert( seiToneMappingInfo->m_extendedRangeWhiteLevel >= 100 );
290       seiToneMappingInfo->m_nominalBlackLevelLumaCodeValue = m_pcCfg->getTMISEINominalBlackLevelLumaCodeValue();
291       seiToneMappingInfo->m_nominalWhiteLevelLumaCodeValue = m_pcCfg->getTMISEINominalWhiteLevelLumaCodeValue();
292       assert( seiToneMappingInfo->m_nominalWhiteLevelLumaCodeValue > seiToneMappingInfo->m_nominalBlackLevelLumaCodeValue );
293       seiToneMappingInfo->m_extendedWhiteLevelLumaCodeValue = m_pcCfg->getTMISEIExtendedWhiteLevelLumaCodeValue();
294       assert( seiToneMappingInfo->m_extendedWhiteLevelLumaCodeValue >= seiToneMappingInfo->m_nominalWhiteLevelLumaCodeValue );
295       break;
296    }
297  default:
298    {
299      assert(!"Undefined SEIToneMapModelId");
300      break;
301    }
302  }
303  return seiToneMappingInfo;
304}
305#endif
306Void TEncGOP::xCreateLeadingSEIMessages (/*SEIMessages seiMessages,*/ AccessUnit &accessUnit, TComSPS *sps)
307{
308  OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI);
309
310  if(m_pcCfg->getActiveParameterSetsSEIEnabled())
311  {
312    SEIActiveParameterSets *sei = xCreateSEIActiveParameterSets (sps);
313
314    //nalu = NALUnit(NAL_UNIT_SEI);
315    m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
316    m_seiWriter.writeSEImessage(nalu.m_Bitstream, *sei, sps); 
317    writeRBSPTrailingBits(nalu.m_Bitstream);
318    accessUnit.push_back(new NALUnitEBSP(nalu));
319    delete sei;
320#if L0045_NON_NESTED_SEI_RESTRICTIONS
321    m_activeParameterSetSEIPresentInAU = true;
322#endif
323  }
324
325  if(m_pcCfg->getFramePackingArrangementSEIEnabled())
326  {
327    SEIFramePacking *sei = xCreateSEIFramePacking ();
328
329    nalu = NALUnit(NAL_UNIT_PREFIX_SEI);
330    m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
331    m_seiWriter.writeSEImessage(nalu.m_Bitstream, *sei, sps);
332    writeRBSPTrailingBits(nalu.m_Bitstream);
333    accessUnit.push_back(new NALUnitEBSP(nalu));
334    delete sei;
335  }
336  if (m_pcCfg->getDisplayOrientationSEIAngle())
337  {
338    SEIDisplayOrientation *sei = xCreateSEIDisplayOrientation();
339
340    nalu = NALUnit(NAL_UNIT_PREFIX_SEI); 
341    m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
342    m_seiWriter.writeSEImessage(nalu.m_Bitstream, *sei, sps); 
343    writeRBSPTrailingBits(nalu.m_Bitstream);
344    accessUnit.push_back(new NALUnitEBSP(nalu));
345    delete sei;
346  }
347#if J0149_TONE_MAPPING_SEI
348  if(m_pcCfg->getToneMappingInfoSEIEnabled())
349  {
350    SEIToneMappingInfo *sei = xCreateSEIToneMappingInfo ();
351     
352    nalu = NALUnit(NAL_UNIT_PREFIX_SEI); 
353    m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
354    m_seiWriter.writeSEImessage(nalu.m_Bitstream, *sei, sps); 
355    writeRBSPTrailingBits(nalu.m_Bitstream);
356    accessUnit.push_back(new NALUnitEBSP(nalu));
357    delete sei;
358  }
359#endif
360}
361
362// ====================================================================================================================
363// Public member functions
364// ====================================================================================================================
365#if H_MV
366Void TEncGOP::initGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic*>& rcListPic, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsInGOP)
367{
368  xInitGOP( iPOCLast, iNumPicRcvd, rcListPic, rcListPicYuvRecOut );
369  m_iNumPicCoded = 0;
370}
371#endif
372#if H_MV
373Void TEncGOP::compressPicInGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic*>& rcListPic, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsInGOP, Int iGOPid)
374#else
375Void TEncGOP::compressGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic*>& rcListPic, TComList<TComPicYuv*>& rcListPicYuvRecOut, std::list<AccessUnit>& accessUnitsInGOP)
376#endif
377{
378  TComPic*        pcPic;
379  TComPicYuv*     pcPicYuvRecOut;
380  TComSlice*      pcSlice;
381  TComOutputBitstream  *pcBitstreamRedirect;
382  pcBitstreamRedirect = new TComOutputBitstream;
383  AccessUnit::iterator  itLocationToPushSliceHeaderNALU; // used to store location where NALU containing slice header is to be inserted
384  UInt                  uiOneBitstreamPerSliceLength = 0;
385  TEncSbac* pcSbacCoders = NULL;
386  TComOutputBitstream* pcSubstreamsOut = NULL;
387
388#if !H_MV
389  xInitGOP( iPOCLast, iNumPicRcvd, rcListPic, rcListPicYuvRecOut );
390 
391  m_iNumPicCoded = 0;
392#endif
393  SEIPictureTiming pictureTimingSEI;
394#if L0208_SOP_DESCRIPTION_SEI
395  Bool writeSOP = m_pcCfg->getSOPDescriptionSEIEnabled();
396#endif
397#if K0180_SCALABLE_NESTING_SEI
398  // Initialize Scalable Nesting SEI with single layer values
399  SEIScalableNesting scalableNestingSEI;
400  scalableNestingSEI.m_bitStreamSubsetFlag           = 1;      // If the nested SEI messages are picture buffereing SEI mesages, picure timing SEI messages or sub-picture timing SEI messages, bitstream_subset_flag shall be equal to 1
401  scalableNestingSEI.m_nestingOpFlag                 = 0;
402  scalableNestingSEI.m_nestingNumOpsMinus1           = 0;      //nesting_num_ops_minus1
403  scalableNestingSEI.m_allLayersFlag                 = 0;
404  scalableNestingSEI.m_nestingNoOpMaxTemporalIdPlus1 = 6 + 1;  //nesting_no_op_max_temporal_id_plus1
405  scalableNestingSEI.m_nestingNumLayersMinus1        = 1 - 1;  //nesting_num_layers_minus1
406  scalableNestingSEI.m_nestingLayerId[0]             = 0;
407  scalableNestingSEI.m_callerOwnsSEIs                = true;
408#endif
409#if L0044_DU_DPB_OUTPUT_DELAY_HRD
410  Int picSptDpbOutputDuDelay = 0;
411#endif
412  UInt *accumBitsDU = NULL;
413  UInt *accumNalsDU = NULL;
414  SEIDecodingUnitInfo decodingUnitInfoSEI;
415#if !H_MV
416  for ( Int iGOPid=0; iGOPid < m_iGopSize; iGOPid++ )
417#endif
418  {
419    UInt uiColDir = 1;
420    //-- For time output for each slice
421    long iBeforeTime = clock();
422
423    //select uiColDir
424    Int iCloseLeft=1, iCloseRight=-1;
425    for(Int i = 0; i<m_pcCfg->getGOPEntry(iGOPid).m_numRefPics; i++) 
426    {
427      Int iRef = m_pcCfg->getGOPEntry(iGOPid).m_referencePics[i];
428      if(iRef>0&&(iRef<iCloseRight||iCloseRight==-1))
429      {
430        iCloseRight=iRef;
431      }
432      else if(iRef<0&&(iRef>iCloseLeft||iCloseLeft==1))
433      {
434        iCloseLeft=iRef;
435      }
436    }
437    if(iCloseRight>-1)
438    {
439      iCloseRight=iCloseRight+m_pcCfg->getGOPEntry(iGOPid).m_POC-1;
440    }
441    if(iCloseLeft<1) 
442    {
443      iCloseLeft=iCloseLeft+m_pcCfg->getGOPEntry(iGOPid).m_POC-1;
444      while(iCloseLeft<0)
445      {
446        iCloseLeft+=m_iGopSize;
447      }
448    }
449    Int iLeftQP=0, iRightQP=0;
450    for(Int i=0; i<m_iGopSize; i++)
451    {
452      if(m_pcCfg->getGOPEntry(i).m_POC==(iCloseLeft%m_iGopSize)+1)
453      {
454        iLeftQP= m_pcCfg->getGOPEntry(i).m_QPOffset;
455      }
456      if (m_pcCfg->getGOPEntry(i).m_POC==(iCloseRight%m_iGopSize)+1)
457      {
458        iRightQP=m_pcCfg->getGOPEntry(i).m_QPOffset;
459      }
460    }
461    if(iCloseRight>-1&&iRightQP<iLeftQP)
462    {
463      uiColDir=0;
464    }
465
466    /////////////////////////////////////////////////////////////////////////////////////////////////// Initial to start encoding
467    Int pocCurr = iPOCLast -iNumPicRcvd+ m_pcCfg->getGOPEntry(iGOPid).m_POC;
468    Int iTimeOffset = m_pcCfg->getGOPEntry(iGOPid).m_POC;
469    if(iPOCLast == 0)
470    {
471      pocCurr=0;
472      iTimeOffset = 1;
473    }
474    if(pocCurr>=m_pcCfg->getFramesToBeEncoded())
475    {
476#if H_MV
477      delete pcBitstreamRedirect;
478      return;
479#else
480      continue;
481#endif
482    }
483
484    if( getNalUnitType(pocCurr, m_iLastIDR) == NAL_UNIT_CODED_SLICE_IDR_W_RADL || getNalUnitType(pocCurr, m_iLastIDR) == NAL_UNIT_CODED_SLICE_IDR_N_LP )
485    {
486      m_iLastIDR = pocCurr;
487    }       
488    // start a new access unit: create an entry in the list of output access units
489    accessUnitsInGOP.push_back(AccessUnit());
490    AccessUnit& accessUnit = accessUnitsInGOP.back();
491    xGetBuffer( rcListPic, rcListPicYuvRecOut, iNumPicRcvd, iTimeOffset, pcPic, pcPicYuvRecOut, pocCurr );
492
493    //  Slice data initialization
494    pcPic->clearSliceBuffer();
495    assert(pcPic->getNumAllocatedSlice() == 1);
496    m_pcSliceEncoder->setSliceIdx(0);
497    pcPic->setCurrSliceIdx(0);
498
499    m_pcSliceEncoder->initEncSlice ( pcPic, iPOCLast, pocCurr, iNumPicRcvd, iGOPid, pcSlice, m_pcEncTop->getSPS(), m_pcEncTop->getPPS() );
500    pcSlice->setLastIDR(m_iLastIDR);
501    pcSlice->setSliceIdx(0);
502#if H_MV
503    pcPic  ->setLayerId     ( getLayerId()   );
504    pcPic  ->setViewId      ( getViewId()    );   
505    pcSlice->setLayerId     ( getLayerId() );
506    pcSlice->setViewId      ( getViewId()  );   
507    pcSlice->setVPS         ( m_pcEncTop->getVPS() );
508#if H_3D
509    pcPic  ->setViewIndex   ( getViewIndex() ); 
510    pcPic  ->setIsDepth( getIsDepth() );
511    pcSlice->setViewIndex   ( getViewIndex()  );
512    pcSlice->setIsDepth( getIsDepth() );   
513    pcSlice->setCamparaSlice( pcPic->getCodedScale(), pcPic->getCodedOffset() );
514#endif
515#endif
516    //set default slice level flag to the same as SPS level flag
517    pcSlice->setLFCrossSliceBoundaryFlag(  pcSlice->getPPS()->getLoopFilterAcrossSlicesEnabledFlag()  );
518    pcSlice->setScalingList ( m_pcEncTop->getScalingList()  );
519    pcSlice->getScalingList()->setUseTransformSkip(m_pcEncTop->getPPS()->getUseTransformSkip());
520    if(m_pcEncTop->getUseScalingListId() == SCALING_LIST_OFF)
521    {
522      m_pcEncTop->getTrQuant()->setFlatScalingList();
523      m_pcEncTop->getTrQuant()->setUseScalingList(false);
524      m_pcEncTop->getSPS()->setScalingListPresentFlag(false);
525      m_pcEncTop->getPPS()->setScalingListPresentFlag(false);
526    }
527    else if(m_pcEncTop->getUseScalingListId() == SCALING_LIST_DEFAULT)
528    {
529      pcSlice->setDefaultScalingList ();
530      m_pcEncTop->getSPS()->setScalingListPresentFlag(false);
531      m_pcEncTop->getPPS()->setScalingListPresentFlag(false);
532      m_pcEncTop->getTrQuant()->setScalingList(pcSlice->getScalingList());
533      m_pcEncTop->getTrQuant()->setUseScalingList(true);
534    }
535    else if(m_pcEncTop->getUseScalingListId() == SCALING_LIST_FILE_READ)
536    {
537      if(pcSlice->getScalingList()->xParseScalingList(m_pcCfg->getScalingListFile()))
538      {
539        pcSlice->setDefaultScalingList ();
540      }
541      pcSlice->getScalingList()->checkDcOfMatrix();
542      m_pcEncTop->getSPS()->setScalingListPresentFlag(pcSlice->checkDefaultScalingList());
543      m_pcEncTop->getPPS()->setScalingListPresentFlag(false);
544      m_pcEncTop->getTrQuant()->setScalingList(pcSlice->getScalingList());
545      m_pcEncTop->getTrQuant()->setUseScalingList(true);
546    }
547    else
548    {
549      printf("error : ScalingList == %d no support\n",m_pcEncTop->getUseScalingListId());
550      assert(0);
551    }
552
553#if H_MV
554    // Set the nal unit type
555    pcSlice->setNalUnitType(getNalUnitType(pocCurr, m_iLastIDR));
556    if( pcSlice->getSliceType() == B_SLICE )
557    {
558      if( m_pcCfg->getGOPEntry( ( pcSlice->getRapPicFlag() && getLayerId() > 0 ) ? MAX_GOP : iGOPid ).m_sliceType == 'P' ) 
559      { 
560        pcSlice->setSliceType( P_SLICE );
561      }
562    }
563#else
564    if(pcSlice->getSliceType()==B_SLICE&&m_pcCfg->getGOPEntry(iGOPid).m_sliceType=='P')
565    {
566      pcSlice->setSliceType(P_SLICE);
567    }
568    // Set the nal unit type
569    pcSlice->setNalUnitType(getNalUnitType(pocCurr, m_iLastIDR));
570#endif
571    if(pcSlice->getTemporalLayerNonReferenceFlag())
572    {
573      if(pcSlice->getNalUnitType()==NAL_UNIT_CODED_SLICE_TRAIL_R)
574      {
575        pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_TRAIL_N);
576      }
577      if(pcSlice->getNalUnitType()==NAL_UNIT_CODED_SLICE_RADL_R)
578      {
579        pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_RADL_N);
580      }
581      if(pcSlice->getNalUnitType()==NAL_UNIT_CODED_SLICE_RASL_R)
582      {
583        pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_RASL_N);
584      }
585    }
586
587    // Do decoding refresh marking if any
588    pcSlice->decodingRefreshMarking(m_pocCRA, m_bRefreshPending, rcListPic);
589    m_pcEncTop->selectReferencePictureSet(pcSlice, pocCurr, iGOPid);
590    pcSlice->getRPS()->setNumberOfLongtermPictures(0);
591
592    if(pcSlice->checkThatAllRefPicsAreAvailable(rcListPic, pcSlice->getRPS(), false) != 0)
593    {
594      pcSlice->createExplicitReferencePictureSetFromReference(rcListPic, pcSlice->getRPS());
595    }
596    pcSlice->applyReferencePictureSet(rcListPic, pcSlice->getRPS());
597
598    if(pcSlice->getTLayer() > 0)
599    {
600      if(pcSlice->isTemporalLayerSwitchingPoint(rcListPic) || pcSlice->getSPS()->getTemporalIdNestingFlag())
601      {
602        if(pcSlice->getTemporalLayerNonReferenceFlag())
603        {
604          pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_TSA_N);
605        }
606        else
607        {
608          pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_TLA_R);
609        }
610      }
611      else if(pcSlice->isStepwiseTemporalLayerSwitchingPointCandidate(rcListPic))
612      {
613        Bool isSTSA=true;
614        for(Int ii=iGOPid+1;(ii<m_pcCfg->getGOPSize() && isSTSA==true);ii++)
615        {
616          Int lTid= m_pcCfg->getGOPEntry(ii).m_temporalId;
617          if(lTid==pcSlice->getTLayer()) 
618          {
619            TComReferencePictureSet* nRPS = pcSlice->getSPS()->getRPSList()->getReferencePictureSet(ii);
620            for(Int jj=0;jj<nRPS->getNumberOfPictures();jj++)
621            {
622              if(nRPS->getUsed(jj)) 
623              {
624                Int tPoc=m_pcCfg->getGOPEntry(ii).m_POC+nRPS->getDeltaPOC(jj);
625                Int kk=0;
626                for(kk=0;kk<m_pcCfg->getGOPSize();kk++)
627                {
628                  if(m_pcCfg->getGOPEntry(kk).m_POC==tPoc)
629                    break;
630                }
631                Int tTid=m_pcCfg->getGOPEntry(kk).m_temporalId;
632                if(tTid >= pcSlice->getTLayer())
633                {
634                  isSTSA=false;
635                  break;
636                }
637              }
638            }
639          }
640        }
641        if(isSTSA==true)
642        {   
643          if(pcSlice->getTemporalLayerNonReferenceFlag())
644          {
645            pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_STSA_N);
646          }
647          else
648          {
649            pcSlice->setNalUnitType(NAL_UNIT_CODED_SLICE_STSA_R);
650          }
651        }
652      }
653    }
654    arrangeLongtermPicturesInRPS(pcSlice, rcListPic);
655    TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
656    refPicListModification->setRefPicListModificationFlagL0(0);
657    refPicListModification->setRefPicListModificationFlagL1(0);
658#if H_MV
659    pcSlice->createAndApplyIvReferencePictureSet( m_ivPicLists, m_refPicSetInterLayer ); 
660    pcSlice->setNumRefIdx(REF_PIC_LIST_0,min(m_pcCfg->getGOPEntry( (pcSlice->getRapPicFlag() && getLayerId() > 0) ? MAX_GOP : iGOPid ).m_numRefPicsActive,( pcSlice->getRPS()->getNumberOfPictures() + (Int) m_refPicSetInterLayer.size() ) ) );
661    pcSlice->setNumRefIdx(REF_PIC_LIST_1,min(m_pcCfg->getGOPEntry( (pcSlice->getRapPicFlag() && getLayerId() > 0) ? MAX_GOP : iGOPid ).m_numRefPicsActive,( pcSlice->getRPS()->getNumberOfPictures() + (Int) m_refPicSetInterLayer.size() ) ) );
662    xSetRefPicListModificationsMvc( pcSlice, pocCurr, iGOPid );   
663#else
664    pcSlice->setNumRefIdx(REF_PIC_LIST_0,min(m_pcCfg->getGOPEntry(iGOPid).m_numRefPicsActive,pcSlice->getRPS()->getNumberOfPictures()));
665    pcSlice->setNumRefIdx(REF_PIC_LIST_1,min(m_pcCfg->getGOPEntry(iGOPid).m_numRefPicsActive,pcSlice->getRPS()->getNumberOfPictures()));
666#endif
667
668#if ADAPTIVE_QP_SELECTION
669    pcSlice->setTrQuant( m_pcEncTop->getTrQuant() );
670#endif     
671
672    //  Set reference list
673#if H_MV   
674    pcSlice->setRefPicList( rcListPic, m_refPicSetInterLayer );
675#if H_3D_ARP
676    pcSlice->setARPStepNum();
677    if(pcSlice->getARPStepNum() > 1)
678    {
679      for(Int iLayerId = 0; iLayerId < getLayerId(); iLayerId ++ )
680      {
681        Int  iViewIdx =   pcSlice->getVPS()->getViewIndex(iLayerId);
682        Bool bIsDepth = ( pcSlice->getVPS()->getDepthId  ( iLayerId ) == 1 );
683        if( iViewIdx<getViewIndex() && !bIsDepth )
684        {
685#if H_3D_ARP_FIX
686          pcSlice->setBaseViewRefPicList( m_ivPicLists->getPicList( iLayerId ), iViewIdx );
687#else
688          //GT (IVREFS): m_tAppEncTop is not needed any more. Reference pictures can be obtained from m_ivPicLists. Should be fixed later.
689          pcSlice->setBaseViewRefPicList( m_pcEncTop->getTAppEncTop()->getTEncTop( iLayerId )->getListPic(), iViewIdx );
690#endif
691        }
692      }
693    }
694#endif
695#else
696    pcSlice->setRefPicList ( rcListPic );
697#endif
698
699    //  Slice info. refinement
700#if H_MV
701    if ( pcSlice->getSliceType() == B_SLICE )
702    {
703      if( m_pcCfg->getGOPEntry( ( pcSlice->getRapPicFlag() == true && getLayerId() > 0 ) ? MAX_GOP : iGOPid ).m_sliceType == 'P' ) 
704      { 
705        pcSlice->setSliceType( P_SLICE ); 
706      }
707    }
708#else
709    if ( (pcSlice->getSliceType() == B_SLICE) && (pcSlice->getNumRefIdx(REF_PIC_LIST_1) == 0) )
710    {
711      pcSlice->setSliceType ( P_SLICE );
712    }
713#endif
714#if !L0034_COMBINED_LIST_CLEANUP
715    if (pcSlice->getSliceType() != B_SLICE || !pcSlice->getSPS()->getUseLComb())
716    {
717      pcSlice->setNumRefIdx(REF_PIC_LIST_C, 0);
718      pcSlice->setRefPicListCombinationFlag(false);
719      pcSlice->setRefPicListModificationFlagLC(false);
720    }
721    else
722    {
723      pcSlice->setRefPicListCombinationFlag(pcSlice->getSPS()->getUseLComb());
724      pcSlice->setNumRefIdx(REF_PIC_LIST_C, pcSlice->getNumRefIdx(REF_PIC_LIST_0));
725    }
726#endif
727
728    if (pcSlice->getSliceType() == B_SLICE)
729    {
730      pcSlice->setColFromL0Flag(1-uiColDir);
731      Bool bLowDelay = true;
732      Int  iCurrPOC  = pcSlice->getPOC();
733      Int iRefIdx = 0;
734
735      for (iRefIdx = 0; iRefIdx < pcSlice->getNumRefIdx(REF_PIC_LIST_0) && bLowDelay; iRefIdx++)
736      {
737        if ( pcSlice->getRefPic(REF_PIC_LIST_0, iRefIdx)->getPOC() > iCurrPOC )
738        {
739          bLowDelay = false;
740        }
741      }
742      for (iRefIdx = 0; iRefIdx < pcSlice->getNumRefIdx(REF_PIC_LIST_1) && bLowDelay; iRefIdx++)
743      {
744        if ( pcSlice->getRefPic(REF_PIC_LIST_1, iRefIdx)->getPOC() > iCurrPOC )
745        {
746          bLowDelay = false;
747        }
748      }
749
750      pcSlice->setCheckLDC(bLowDelay); 
751    }
752    else
753    {
754      pcSlice->setCheckLDC(true); 
755    }
756
757    uiColDir = 1-uiColDir;
758
759    //-------------------------------------------------------------
760    pcSlice->setRefPOCList();
761
762#if L0034_COMBINED_LIST_CLEANUP
763    pcSlice->setList1IdxToList0Idx();
764#else
765    pcSlice->setNoBackPredFlag( false );
766    if ( pcSlice->getSliceType() == B_SLICE && !pcSlice->getRefPicListCombinationFlag())
767    {
768      if ( pcSlice->getNumRefIdx(RefPicList( 0 ) ) == pcSlice->getNumRefIdx(RefPicList( 1 ) ) )
769      {
770        pcSlice->setNoBackPredFlag( true );
771        Int i;
772        for ( i=0; i < pcSlice->getNumRefIdx(RefPicList( 1 ) ); i++ )
773        {
774          if ( pcSlice->getRefPOC(RefPicList(1), i) != pcSlice->getRefPOC(RefPicList(0), i) ) 
775          {
776            pcSlice->setNoBackPredFlag( false );
777            break;
778          }
779        }
780      }
781    }
782
783    if(pcSlice->getNoBackPredFlag())
784    {
785      pcSlice->setNumRefIdx(REF_PIC_LIST_C, 0);
786    }
787    pcSlice->generateCombinedList();
788#endif
789
790    if (m_pcEncTop->getTMVPModeId() == 2)
791    {
792      if (iGOPid == 0) // first picture in SOP (i.e. forward B)
793      {
794        pcSlice->setEnableTMVPFlag(0);
795      }
796      else
797      {
798        // Note: pcSlice->getColFromL0Flag() is assumed to be always 0 and getcolRefIdx() is always 0.
799        pcSlice->setEnableTMVPFlag(1);
800      }
801      pcSlice->getSPS()->setTMVPFlagsPresent(1);
802    }
803    else if (m_pcEncTop->getTMVPModeId() == 1)
804    {
805      pcSlice->getSPS()->setTMVPFlagsPresent(1);
806      pcSlice->setEnableTMVPFlag(1);
807    }
808    else
809    {
810      pcSlice->getSPS()->setTMVPFlagsPresent(0);
811      pcSlice->setEnableTMVPFlag(0);
812    }
813
814#if H_3D_VSO
815  // Should be moved to TEncTop !!!
816  Bool bUseVSO = m_pcEncTop->getUseVSO();
817 
818  TComRdCost* pcRdCost = m_pcEncTop->getRdCost();   
819
820  pcRdCost->setUseVSO( bUseVSO );
821
822  // SAIT_VSO_EST_A0033
823  pcRdCost->setUseEstimatedVSD( m_pcEncTop->getUseEstimatedVSD() );
824
825  if ( bUseVSO )
826  {
827    Int iVSOMode = m_pcEncTop->getVSOMode();
828    pcRdCost->setVSOMode( iVSOMode  );
829    pcRdCost->setAllowNegDist( m_pcEncTop->getAllowNegDist() );
830
831    // SAIT_VSO_EST_A0033
832    pcRdCost->setVideoRecPicYuv( m_pcEncTop->getIvPicLists()->getPicYuv( pcSlice->getViewIndex(), false , pcSlice->getPOC(), true ) );
833    pcRdCost->setDepthPicYuv   ( m_pcEncTop->getIvPicLists()->getPicYuv( pcSlice->getViewIndex(), true  , pcSlice->getPOC(), false ) );
834
835    // LGE_WVSO_A0119
836    Bool bUseWVSO  = m_pcEncTop->getUseWVSO();
837    pcRdCost->setUseWVSO( bUseWVSO );
838
839  }
840#endif
841    /////////////////////////////////////////////////////////////////////////////////////////////////// Compress a slice
842    //  Slice compression
843    if (m_pcCfg->getUseASR())
844    {
845      m_pcSliceEncoder->setSearchRange(pcSlice);
846    }
847
848    Bool bGPBcheck=false;
849    if ( pcSlice->getSliceType() == B_SLICE)
850    {
851      if ( pcSlice->getNumRefIdx(RefPicList( 0 ) ) == pcSlice->getNumRefIdx(RefPicList( 1 ) ) )
852      {
853        bGPBcheck=true;
854        Int i;
855        for ( i=0; i < pcSlice->getNumRefIdx(RefPicList( 1 ) ); i++ )
856        {
857          if ( pcSlice->getRefPOC(RefPicList(1), i) != pcSlice->getRefPOC(RefPicList(0), i) ) 
858          {
859            bGPBcheck=false;
860            break;
861          }
862        }
863      }
864    }
865    if(bGPBcheck)
866    {
867      pcSlice->setMvdL1ZeroFlag(true);
868    }
869    else
870    {
871      pcSlice->setMvdL1ZeroFlag(false);
872    }
873    pcPic->getSlice(pcSlice->getSliceIdx())->setMvdL1ZeroFlag(pcSlice->getMvdL1ZeroFlag());
874
875#if RATE_CONTROL_LAMBDA_DOMAIN
876    Int sliceQP              = pcSlice->getSliceQp();
877    Double lambda            = 0.0;
878    Int actualHeadBits       = 0;
879    Int actualTotalBits      = 0;
880    Int estimatedBits        = 0;
881    Int tmpBitsBeforeWriting = 0;
882    if ( m_pcCfg->getUseRateCtrl() )
883    {
884      Int frameLevel = m_pcRateCtrl->getRCSeq()->getGOPID2Level( iGOPid );
885      if ( pcPic->getSlice(0)->getSliceType() == I_SLICE )
886      {
887        frameLevel = 0;
888      }
889      m_pcRateCtrl->initRCPic( frameLevel );
890      estimatedBits = m_pcRateCtrl->getRCPic()->getTargetBits();
891
892      if ( ( pcSlice->getPOC() == 0 && m_pcCfg->getInitialQP() > 0 ) || ( frameLevel == 0 && m_pcCfg->getForceIntraQP() ) ) // QP is specified
893      {
894        sliceQP              = m_pcCfg->getInitialQP();
895        Int    NumberBFrames = ( m_pcCfg->getGOPSize() - 1 );
896        Double dLambda_scale = 1.0 - Clip3( 0.0, 0.5, 0.05*(Double)NumberBFrames );
897        Double dQPFactor     = 0.57*dLambda_scale;
898        Int    SHIFT_QP      = 12;
899        Int    bitdepth_luma_qp_scale = 0;
900        Double qp_temp = (Double) sliceQP + bitdepth_luma_qp_scale - SHIFT_QP;
901        lambda = dQPFactor*pow( 2.0, qp_temp/3.0 );
902      }
903      else if ( frameLevel == 0 )   // intra case, but use the model
904      {
905        if ( m_pcCfg->getIntraPeriod() != 1 )   // do not refine allocated bits for all intra case
906        {
907          Int bits = m_pcRateCtrl->getRCSeq()->getLeftAverageBits();
908          bits = m_pcRateCtrl->getRCSeq()->getRefineBitsForIntra( bits );
909          if ( bits < 200 )
910          {
911            bits = 200;
912          }
913          m_pcRateCtrl->getRCPic()->setTargetBits( bits );
914        }
915
916        list<TEncRCPic*> listPreviousPicture = m_pcRateCtrl->getPicList();
917        lambda  = m_pcRateCtrl->getRCPic()->estimatePicLambda( listPreviousPicture );
918        sliceQP = m_pcRateCtrl->getRCPic()->estimatePicQP( lambda, listPreviousPicture );
919      }
920      else    // normal case
921      {
922        list<TEncRCPic*> listPreviousPicture = m_pcRateCtrl->getPicList();
923        lambda  = m_pcRateCtrl->getRCPic()->estimatePicLambda( listPreviousPicture );
924        sliceQP = m_pcRateCtrl->getRCPic()->estimatePicQP( lambda, listPreviousPicture );
925      }
926
927      sliceQP = Clip3( -pcSlice->getSPS()->getQpBDOffsetY(), MAX_QP, sliceQP );
928      m_pcRateCtrl->getRCPic()->setPicEstQP( sliceQP );
929
930      m_pcSliceEncoder->resetQP( pcPic, sliceQP, lambda );
931    }
932#endif
933
934    UInt uiNumSlices = 1;
935
936    UInt uiInternalAddress = pcPic->getNumPartInCU()-4;
937    UInt uiExternalAddress = pcPic->getPicSym()->getNumberOfCUsInFrame()-1;
938    UInt uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
939    UInt uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
940    UInt uiWidth = pcSlice->getSPS()->getPicWidthInLumaSamples();
941    UInt uiHeight = pcSlice->getSPS()->getPicHeightInLumaSamples();
942    while(uiPosX>=uiWidth||uiPosY>=uiHeight) 
943    {
944      uiInternalAddress--;
945      uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
946      uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
947    }
948    uiInternalAddress++;
949    if(uiInternalAddress==pcPic->getNumPartInCU()) 
950    {
951      uiInternalAddress = 0;
952      uiExternalAddress++;
953    }
954    UInt uiRealEndAddress = uiExternalAddress*pcPic->getNumPartInCU()+uiInternalAddress;
955
956    UInt uiCummulativeTileWidth;
957    UInt uiCummulativeTileHeight;
958    Int  p, j;
959    UInt uiEncCUAddr;
960
961    //set NumColumnsMinus1 and NumRowsMinus1
962    pcPic->getPicSym()->setNumColumnsMinus1( pcSlice->getPPS()->getNumColumnsMinus1() );
963    pcPic->getPicSym()->setNumRowsMinus1( pcSlice->getPPS()->getNumRowsMinus1() );
964
965    //create the TComTileArray
966    pcPic->getPicSym()->xCreateTComTileArray();
967
968    if( pcSlice->getPPS()->getUniformSpacingFlag() == 1 )
969    {
970      //set the width for each tile
971      for(j=0; j < pcPic->getPicSym()->getNumRowsMinus1()+1; j++)
972      {
973        for(p=0; p < pcPic->getPicSym()->getNumColumnsMinus1()+1; p++)
974        {
975          pcPic->getPicSym()->getTComTile( j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p )->
976            setTileWidth( (p+1)*pcPic->getPicSym()->getFrameWidthInCU()/(pcPic->getPicSym()->getNumColumnsMinus1()+1) 
977            - (p*pcPic->getPicSym()->getFrameWidthInCU())/(pcPic->getPicSym()->getNumColumnsMinus1()+1) );
978        }
979      }
980
981      //set the height for each tile
982      for(j=0; j < pcPic->getPicSym()->getNumColumnsMinus1()+1; j++)
983      {
984        for(p=0; p < pcPic->getPicSym()->getNumRowsMinus1()+1; p++)
985        {
986          pcPic->getPicSym()->getTComTile( p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j )->
987            setTileHeight( (p+1)*pcPic->getPicSym()->getFrameHeightInCU()/(pcPic->getPicSym()->getNumRowsMinus1()+1) 
988            - (p*pcPic->getPicSym()->getFrameHeightInCU())/(pcPic->getPicSym()->getNumRowsMinus1()+1) );   
989        }
990      }
991    }
992    else
993    {
994      //set the width for each tile
995      for(j=0; j < pcPic->getPicSym()->getNumRowsMinus1()+1; j++)
996      {
997        uiCummulativeTileWidth = 0;
998        for(p=0; p < pcPic->getPicSym()->getNumColumnsMinus1(); p++)
999        {
1000          pcPic->getPicSym()->getTComTile( j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p )->setTileWidth( pcSlice->getPPS()->getColumnWidth(p) );
1001          uiCummulativeTileWidth += pcSlice->getPPS()->getColumnWidth(p);
1002        }
1003        pcPic->getPicSym()->getTComTile(j * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + p)->setTileWidth( pcPic->getPicSym()->getFrameWidthInCU()-uiCummulativeTileWidth );
1004      }
1005
1006      //set the height for each tile
1007      for(j=0; j < pcPic->getPicSym()->getNumColumnsMinus1()+1; j++)
1008      {
1009        uiCummulativeTileHeight = 0;
1010        for(p=0; p < pcPic->getPicSym()->getNumRowsMinus1(); p++)
1011        {
1012          pcPic->getPicSym()->getTComTile( p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j )->setTileHeight( pcSlice->getPPS()->getRowHeight(p) );
1013          uiCummulativeTileHeight += pcSlice->getPPS()->getRowHeight(p);
1014        }
1015        pcPic->getPicSym()->getTComTile(p * (pcPic->getPicSym()->getNumColumnsMinus1()+1) + j)->setTileHeight( pcPic->getPicSym()->getFrameHeightInCU()-uiCummulativeTileHeight );
1016      }
1017    }
1018    //intialize each tile of the current picture
1019    pcPic->getPicSym()->xInitTiles();
1020
1021    // Allocate some coders, now we know how many tiles there are.
1022    Int iNumSubstreams = pcSlice->getPPS()->getNumSubstreams();
1023
1024    //generate the Coding Order Map and Inverse Coding Order Map
1025    for(p=0, uiEncCUAddr=0; p<pcPic->getPicSym()->getNumberOfCUsInFrame(); p++, uiEncCUAddr = pcPic->getPicSym()->xCalculateNxtCUAddr(uiEncCUAddr))
1026    {
1027      pcPic->getPicSym()->setCUOrderMap(p, uiEncCUAddr);
1028      pcPic->getPicSym()->setInverseCUOrderMap(uiEncCUAddr, p);
1029    }
1030    pcPic->getPicSym()->setCUOrderMap(pcPic->getPicSym()->getNumberOfCUsInFrame(), pcPic->getPicSym()->getNumberOfCUsInFrame());   
1031    pcPic->getPicSym()->setInverseCUOrderMap(pcPic->getPicSym()->getNumberOfCUsInFrame(), pcPic->getPicSym()->getNumberOfCUsInFrame());
1032
1033    // Allocate some coders, now we know how many tiles there are.
1034    m_pcEncTop->createWPPCoders(iNumSubstreams);
1035    pcSbacCoders = m_pcEncTop->getSbacCoders();
1036    pcSubstreamsOut = new TComOutputBitstream[iNumSubstreams];
1037
1038    UInt startCUAddrSliceIdx = 0; // used to index "m_uiStoredStartCUAddrForEncodingSlice" containing locations of slice boundaries
1039    UInt startCUAddrSlice    = 0; // used to keep track of current slice's starting CU addr.
1040    pcSlice->setSliceCurStartCUAddr( startCUAddrSlice ); // Setting "start CU addr" for current slice
1041    m_storedStartCUAddrForEncodingSlice.clear();
1042
1043    UInt startCUAddrSliceSegmentIdx = 0; // used to index "m_uiStoredStartCUAddrForEntropyEncodingSlice" containing locations of slice boundaries
1044    UInt startCUAddrSliceSegment    = 0; // used to keep track of current Dependent slice's starting CU addr.
1045    pcSlice->setSliceSegmentCurStartCUAddr( startCUAddrSliceSegment ); // Setting "start CU addr" for current Dependent slice
1046
1047    m_storedStartCUAddrForEncodingSliceSegment.clear();
1048    UInt nextCUAddr = 0;
1049    m_storedStartCUAddrForEncodingSlice.push_back (nextCUAddr);
1050    startCUAddrSliceIdx++;
1051    m_storedStartCUAddrForEncodingSliceSegment.push_back(nextCUAddr);
1052    startCUAddrSliceSegmentIdx++;
1053#if H_3D_NBDV
1054      if(pcSlice->getViewIndex() && !pcSlice->getIsDepth()) //Notes from QC: this condition shall be changed once the configuration is completed, e.g. in pcSlice->getSPS()->getMultiviewMvPredMode() || ARP in prev. HTM. Remove this comment once it is done.
1055      {
1056        Int iColPoc = pcSlice->getRefPOC(RefPicList(1-pcSlice->getColFromL0Flag()), pcSlice->getColRefIdx());
1057        pcPic->setNumDdvCandPics(pcPic->getDisCandRefPictures(iColPoc));
1058      }
1059#endif
1060    while(nextCUAddr<uiRealEndAddress) // determine slice boundaries
1061    {
1062      pcSlice->setNextSlice       ( false );
1063      pcSlice->setNextSliceSegment( false );
1064      assert(pcPic->getNumAllocatedSlice() == startCUAddrSliceIdx);
1065      m_pcSliceEncoder->precompressSlice( pcPic );
1066      m_pcSliceEncoder->compressSlice   ( pcPic );
1067
1068      Bool bNoBinBitConstraintViolated = (!pcSlice->isNextSlice() && !pcSlice->isNextSliceSegment());
1069      if (pcSlice->isNextSlice() || (bNoBinBitConstraintViolated && m_pcCfg->getSliceMode()==FIXED_NUMBER_OF_LCU))
1070      {
1071        startCUAddrSlice = pcSlice->getSliceCurEndCUAddr();
1072        // Reconstruction slice
1073        m_storedStartCUAddrForEncodingSlice.push_back(startCUAddrSlice);
1074        startCUAddrSliceIdx++;
1075        // Dependent slice
1076        if (startCUAddrSliceSegmentIdx>0 && m_storedStartCUAddrForEncodingSliceSegment[startCUAddrSliceSegmentIdx-1] != startCUAddrSlice)
1077        {
1078          m_storedStartCUAddrForEncodingSliceSegment.push_back(startCUAddrSlice);
1079          startCUAddrSliceSegmentIdx++;
1080        }
1081
1082        if (startCUAddrSlice < uiRealEndAddress)
1083        {
1084          pcPic->allocateNewSlice();         
1085          pcPic->setCurrSliceIdx                  ( startCUAddrSliceIdx-1 );
1086          m_pcSliceEncoder->setSliceIdx           ( startCUAddrSliceIdx-1 );
1087          pcSlice = pcPic->getSlice               ( startCUAddrSliceIdx-1 );
1088          pcSlice->copySliceInfo                  ( pcPic->getSlice(0)      );
1089          pcSlice->setSliceIdx                    ( startCUAddrSliceIdx-1 );
1090          pcSlice->setSliceCurStartCUAddr         ( startCUAddrSlice      );
1091          pcSlice->setSliceSegmentCurStartCUAddr  ( startCUAddrSlice      );
1092          pcSlice->setSliceBits(0);
1093          uiNumSlices ++;
1094        }
1095      }
1096      else if (pcSlice->isNextSliceSegment() || (bNoBinBitConstraintViolated && m_pcCfg->getSliceSegmentMode()==FIXED_NUMBER_OF_LCU))
1097      {
1098        startCUAddrSliceSegment                                                     = pcSlice->getSliceSegmentCurEndCUAddr();
1099        m_storedStartCUAddrForEncodingSliceSegment.push_back(startCUAddrSliceSegment);
1100        startCUAddrSliceSegmentIdx++;
1101        pcSlice->setSliceSegmentCurStartCUAddr( startCUAddrSliceSegment );
1102      }
1103      else
1104      {
1105        startCUAddrSlice                                                            = pcSlice->getSliceCurEndCUAddr();
1106        startCUAddrSliceSegment                                                     = pcSlice->getSliceSegmentCurEndCUAddr();
1107      }       
1108
1109      nextCUAddr = (startCUAddrSlice > startCUAddrSliceSegment) ? startCUAddrSlice : startCUAddrSliceSegment;
1110    }
1111    m_storedStartCUAddrForEncodingSlice.push_back( pcSlice->getSliceCurEndCUAddr());
1112    startCUAddrSliceIdx++;
1113    m_storedStartCUAddrForEncodingSliceSegment.push_back(pcSlice->getSliceCurEndCUAddr());
1114    startCUAddrSliceSegmentIdx++;
1115
1116    pcSlice = pcPic->getSlice(0);
1117
1118    // SAO parameter estimation using non-deblocked pixels for LCU bottom and right boundary areas
1119    if( m_pcCfg->getSaoLcuBasedOptimization() && m_pcCfg->getSaoLcuBoundary() )
1120    {
1121      m_pcSAO->resetStats();
1122      m_pcSAO->calcSaoStatsCu_BeforeDblk( pcPic );
1123    }
1124
1125    //-- Loop filter
1126    Bool bLFCrossTileBoundary = pcSlice->getPPS()->getLoopFilterAcrossTilesEnabledFlag();
1127    m_pcLoopFilter->setCfg(bLFCrossTileBoundary);
1128#if L0386_DB_METRIC
1129    if ( m_pcCfg->getDeblockingFilterMetric() )
1130    {
1131      dblMetric(pcPic, uiNumSlices);
1132    }
1133#endif
1134    m_pcLoopFilter->loopFilterPic( pcPic );
1135
1136    pcSlice = pcPic->getSlice(0);
1137    if(pcSlice->getSPS()->getUseSAO())
1138    {
1139      std::vector<Bool> LFCrossSliceBoundaryFlag;
1140      for(Int s=0; s< uiNumSlices; s++)
1141      {
1142        LFCrossSliceBoundaryFlag.push_back(  ((uiNumSlices==1)?true:pcPic->getSlice(s)->getLFCrossSliceBoundaryFlag()) );
1143      }
1144      m_storedStartCUAddrForEncodingSlice.resize(uiNumSlices+1);
1145      pcPic->createNonDBFilterInfo(m_storedStartCUAddrForEncodingSlice, 0, &LFCrossSliceBoundaryFlag ,pcPic->getPicSym()->getNumTiles() ,bLFCrossTileBoundary);
1146    }
1147
1148
1149    pcSlice = pcPic->getSlice(0);
1150
1151    if(pcSlice->getSPS()->getUseSAO())
1152    {
1153      m_pcSAO->createPicSaoInfo(pcPic);
1154    }
1155
1156    /////////////////////////////////////////////////////////////////////////////////////////////////// File writing
1157    // Set entropy coder
1158    m_pcEntropyCoder->setEntropyCoder   ( m_pcCavlcCoder, pcSlice );
1159
1160    /* write various header sets. */
1161    if ( m_bSeqFirst )
1162    {
1163      OutputNALUnit nalu(NAL_UNIT_VPS);
1164#if H_MV
1165      if( getLayerId() == 0 )
1166      {
1167#endif
1168      m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1169      m_pcEntropyCoder->encodeVPS(m_pcEncTop->getVPS());
1170      writeRBSPTrailingBits(nalu.m_Bitstream);
1171      accessUnit.push_back(new NALUnitEBSP(nalu));
1172#if RATE_CONTROL_LAMBDA_DOMAIN
1173      actualTotalBits += UInt(accessUnit.back()->m_nalUnitData.str().size()) * 8;
1174#endif
1175
1176#if H_MV
1177      }
1178      nalu = NALUnit(NAL_UNIT_SPS, 0, getLayerId());
1179#else
1180      nalu = NALUnit(NAL_UNIT_SPS);
1181#endif
1182      m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1183      if (m_bSeqFirst)
1184      {
1185        pcSlice->getSPS()->setNumLongTermRefPicSPS(m_numLongTermRefPicSPS);
1186        for (Int k = 0; k < m_numLongTermRefPicSPS; k++)
1187        {
1188          pcSlice->getSPS()->setLtRefPicPocLsbSps(k, m_ltRefPicPocLsbSps[k]);
1189          pcSlice->getSPS()->setUsedByCurrPicLtSPSFlag(k, m_ltRefPicUsedByCurrPicFlag[k]);
1190        }
1191      }
1192      if( m_pcCfg->getPictureTimingSEIEnabled() || m_pcCfg->getDecodingUnitInfoSEIEnabled() )
1193      {
1194        UInt maxCU = m_pcCfg->getSliceArgument() >> ( pcSlice->getSPS()->getMaxCUDepth() << 1);
1195        UInt numDU = ( m_pcCfg->getSliceMode() == 1 ) ? ( pcPic->getNumCUsInFrame() / maxCU ) : ( 0 );
1196        if( pcPic->getNumCUsInFrame() % maxCU != 0 )
1197        {
1198          numDU ++;
1199        }
1200        pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->setNumDU( numDU );
1201        pcSlice->getSPS()->setHrdParameters( m_pcCfg->getFrameRate(), numDU, m_pcCfg->getTargetBitrate(), ( m_pcCfg->getIntraPeriod() > 0 ) );
1202      }
1203      if( m_pcCfg->getBufferingPeriodSEIEnabled() || m_pcCfg->getPictureTimingSEIEnabled() || m_pcCfg->getDecodingUnitInfoSEIEnabled() )
1204      {
1205        pcSlice->getSPS()->getVuiParameters()->setHrdParametersPresentFlag( true );
1206      }
1207#if !H_3D
1208      m_pcEntropyCoder->encodeSPS(pcSlice->getSPS());
1209#else
1210      m_pcEntropyCoder->encodeSPS(pcSlice->getSPS(), pcSlice->getViewIndex(), pcSlice->getIsDepth() );
1211#endif
1212      writeRBSPTrailingBits(nalu.m_Bitstream);
1213      accessUnit.push_back(new NALUnitEBSP(nalu));
1214#if RATE_CONTROL_LAMBDA_DOMAIN
1215      actualTotalBits += UInt(accessUnit.back()->m_nalUnitData.str().size()) * 8;
1216#endif
1217
1218#if H_MV
1219      nalu = NALUnit(NAL_UNIT_PPS, 0, getLayerId());
1220#else
1221      nalu = NALUnit(NAL_UNIT_PPS);
1222#endif
1223      m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1224      m_pcEntropyCoder->encodePPS(pcSlice->getPPS());
1225      writeRBSPTrailingBits(nalu.m_Bitstream);
1226      accessUnit.push_back(new NALUnitEBSP(nalu));
1227#if RATE_CONTROL_LAMBDA_DOMAIN
1228      actualTotalBits += UInt(accessUnit.back()->m_nalUnitData.str().size()) * 8;
1229#endif
1230
1231      xCreateLeadingSEIMessages(accessUnit, pcSlice->getSPS());
1232
1233      m_bSeqFirst = false;
1234    }
1235
1236#if L0208_SOP_DESCRIPTION_SEI
1237    if (writeSOP) // write SOP description SEI (if enabled) at the beginning of GOP
1238    {
1239      Int SOPcurrPOC = pocCurr;
1240
1241      OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI);
1242      m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1243      m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1244
1245      SEISOPDescription SOPDescriptionSEI;
1246      SOPDescriptionSEI.m_sopSeqParameterSetId = pcSlice->getSPS()->getSPSId();
1247
1248      UInt i = 0;
1249      UInt prevEntryId = iGOPid;
1250      for (j = iGOPid; j < m_iGopSize; j++)
1251      {
1252        Int deltaPOC = m_pcCfg->getGOPEntry(j).m_POC - m_pcCfg->getGOPEntry(prevEntryId).m_POC;
1253        if ((SOPcurrPOC + deltaPOC) < m_pcCfg->getFramesToBeEncoded())
1254        {
1255          SOPcurrPOC += deltaPOC;
1256          SOPDescriptionSEI.m_sopDescVclNaluType[i] = getNalUnitType(SOPcurrPOC, m_iLastIDR);
1257          SOPDescriptionSEI.m_sopDescTemporalId[i] = m_pcCfg->getGOPEntry(j).m_temporalId;
1258          SOPDescriptionSEI.m_sopDescStRpsIdx[i] = m_pcEncTop->getReferencePictureSetIdxForSOP(pcSlice, SOPcurrPOC, j);
1259          SOPDescriptionSEI.m_sopDescPocDelta[i] = deltaPOC;
1260
1261          prevEntryId = j;
1262          i++;
1263        }
1264      }
1265
1266      SOPDescriptionSEI.m_numPicsInSopMinus1 = i - 1;
1267
1268      m_seiWriter.writeSEImessage( nalu.m_Bitstream, SOPDescriptionSEI, pcSlice->getSPS());
1269      writeRBSPTrailingBits(nalu.m_Bitstream);
1270      accessUnit.push_back(new NALUnitEBSP(nalu));
1271
1272      writeSOP = false;
1273    }
1274#endif
1275
1276    if( ( m_pcCfg->getPictureTimingSEIEnabled() || m_pcCfg->getDecodingUnitInfoSEIEnabled() ) &&
1277        ( pcSlice->getSPS()->getVuiParametersPresentFlag() ) &&
1278        ( ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getNalHrdParametersPresentFlag() ) 
1279       || ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getVclHrdParametersPresentFlag() ) ) )
1280    {
1281      if( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getSubPicCpbParamsPresentFlag() )
1282      {
1283        UInt numDU = pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getNumDU();
1284        pictureTimingSEI.m_numDecodingUnitsMinus1     = ( numDU - 1 );
1285        pictureTimingSEI.m_duCommonCpbRemovalDelayFlag = false;
1286
1287        if( pictureTimingSEI.m_numNalusInDuMinus1 == NULL )
1288        {
1289          pictureTimingSEI.m_numNalusInDuMinus1       = new UInt[ numDU ];
1290        }
1291        if( pictureTimingSEI.m_duCpbRemovalDelayMinus1  == NULL )
1292        {
1293          pictureTimingSEI.m_duCpbRemovalDelayMinus1  = new UInt[ numDU ];
1294        }
1295        if( accumBitsDU == NULL )
1296        {
1297          accumBitsDU                                  = new UInt[ numDU ];
1298        }
1299        if( accumNalsDU == NULL )
1300        {
1301          accumNalsDU                                  = new UInt[ numDU ];
1302        }
1303      }
1304      pictureTimingSEI.m_auCpbRemovalDelay = std::max<Int>(1, m_totalCoded - m_lastBPSEI); // Syntax element signalled as minus, hence the .
1305      pictureTimingSEI.m_picDpbOutputDelay = pcSlice->getSPS()->getNumReorderPics(0) + pcSlice->getPOC() - m_totalCoded;
1306#if L0044_DU_DPB_OUTPUT_DELAY_HRD
1307      Int factor = pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getTickDivisorMinus2() + 2;
1308      pictureTimingSEI.m_picDpbOutputDuDelay = factor * pictureTimingSEI.m_picDpbOutputDelay;
1309      if( m_pcCfg->getDecodingUnitInfoSEIEnabled() )
1310      {
1311        picSptDpbOutputDuDelay = factor * pictureTimingSEI.m_picDpbOutputDelay;
1312      }
1313#endif
1314    }
1315
1316    if( ( m_pcCfg->getBufferingPeriodSEIEnabled() ) && ( pcSlice->getSliceType() == I_SLICE ) &&
1317        ( pcSlice->getSPS()->getVuiParametersPresentFlag() ) && 
1318        ( ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getNalHrdParametersPresentFlag() ) 
1319       || ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getVclHrdParametersPresentFlag() ) ) )
1320    {
1321      OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI);
1322      m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1323      m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1324
1325      SEIBufferingPeriod sei_buffering_period;
1326     
1327      UInt uiInitialCpbRemovalDelay = (90000/2);                      // 0.5 sec
1328      sei_buffering_period.m_initialCpbRemovalDelay      [0][0]     = uiInitialCpbRemovalDelay;
1329      sei_buffering_period.m_initialCpbRemovalDelayOffset[0][0]     = uiInitialCpbRemovalDelay;
1330      sei_buffering_period.m_initialCpbRemovalDelay      [0][1]     = uiInitialCpbRemovalDelay;
1331      sei_buffering_period.m_initialCpbRemovalDelayOffset[0][1]     = uiInitialCpbRemovalDelay;
1332
1333#if L0043_TIMING_INFO
1334      Double dTmp = (Double)pcSlice->getSPS()->getVuiParameters()->getTimingInfo()->getNumUnitsInTick() / (Double)pcSlice->getSPS()->getVuiParameters()->getTimingInfo()->getTimeScale();
1335#else
1336      Double dTmp = (Double)pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getNumUnitsInTick() / (Double)pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getTimeScale();
1337#endif
1338
1339      UInt uiTmp = (UInt)( dTmp * 90000.0 ); 
1340      uiInitialCpbRemovalDelay -= uiTmp;
1341      uiInitialCpbRemovalDelay -= uiTmp / ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getTickDivisorMinus2() + 2 );
1342      sei_buffering_period.m_initialAltCpbRemovalDelay      [0][0]  = uiInitialCpbRemovalDelay;
1343      sei_buffering_period.m_initialAltCpbRemovalDelayOffset[0][0]  = uiInitialCpbRemovalDelay;
1344      sei_buffering_period.m_initialAltCpbRemovalDelay      [0][1]  = uiInitialCpbRemovalDelay;
1345      sei_buffering_period.m_initialAltCpbRemovalDelayOffset[0][1]  = uiInitialCpbRemovalDelay;
1346
1347      sei_buffering_period.m_rapCpbParamsPresentFlag              = 0;
1348#if L0328_SPLICING
1349      //for the concatenation, it can be set to one during splicing.
1350      sei_buffering_period.m_concatenationFlag = 0;
1351      //since the temporal layer HRD is not ready, we assumed it is fixed
1352      sei_buffering_period.m_auCpbRemovalDelayDelta = 1;
1353#endif
1354#if L0044_CPB_DPB_DELAY_OFFSET
1355      sei_buffering_period.m_cpbDelayOffset = 0;
1356      sei_buffering_period.m_dpbDelayOffset = 0;
1357#endif
1358
1359      m_seiWriter.writeSEImessage( nalu.m_Bitstream, sei_buffering_period, pcSlice->getSPS());
1360      writeRBSPTrailingBits(nalu.m_Bitstream);
1361#if L0045_NON_NESTED_SEI_RESTRICTIONS
1362      {
1363      UInt seiPositionInAu = xGetFirstSeiLocation(accessUnit);
1364      UInt offsetPosition = m_activeParameterSetSEIPresentInAU;   // Insert BP SEI after APS SEI
1365      AccessUnit::iterator it;
1366      for(j = 0, it = accessUnit.begin(); j < seiPositionInAu + offsetPosition; j++)
1367      {
1368        it++;
1369      }
1370      accessUnit.insert(it, new NALUnitEBSP(nalu));
1371      m_bufferingPeriodSEIPresentInAU = true;
1372      }
1373#else
1374      accessUnit.push_back(new NALUnitEBSP(nalu));
1375#endif
1376
1377#if K0180_SCALABLE_NESTING_SEI
1378      if (m_pcCfg->getScalableNestingSEIEnabled())
1379      {
1380        OutputNALUnit naluTmp(NAL_UNIT_PREFIX_SEI);
1381        m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1382        m_pcEntropyCoder->setBitstream(&naluTmp.m_Bitstream);
1383        scalableNestingSEI.m_nestedSEIs.clear();
1384        scalableNestingSEI.m_nestedSEIs.push_back(&sei_buffering_period);
1385        m_seiWriter.writeSEImessage( naluTmp.m_Bitstream, scalableNestingSEI, pcSlice->getSPS());
1386        writeRBSPTrailingBits(naluTmp.m_Bitstream);
1387#if L0045_NON_NESTED_SEI_RESTRICTIONS
1388        UInt seiPositionInAu = xGetFirstSeiLocation(accessUnit);
1389        UInt offsetPosition = m_activeParameterSetSEIPresentInAU + m_bufferingPeriodSEIPresentInAU + m_pictureTimingSEIPresentInAU;   // Insert BP SEI after non-nested APS, BP and PT SEIs
1390        AccessUnit::iterator it;
1391        for(j = 0, it = accessUnit.begin(); j < seiPositionInAu + offsetPosition; j++)
1392        {
1393          it++;
1394        }
1395        accessUnit.insert(it, new NALUnitEBSP(naluTmp));
1396        m_nestedBufferingPeriodSEIPresentInAU = true;
1397#else
1398        accessUnit.push_back(new NALUnitEBSP(naluTmp));
1399#endif
1400      }
1401#endif
1402
1403      m_lastBPSEI = m_totalCoded;
1404      m_cpbRemovalDelay = 0;
1405    }
1406    m_cpbRemovalDelay ++;
1407    if( ( m_pcEncTop->getRecoveryPointSEIEnabled() ) && ( pcSlice->getSliceType() == I_SLICE ) )
1408    {
1409      if( m_pcEncTop->getGradualDecodingRefreshInfoEnabled() && !pcSlice->getRapPicFlag() )
1410      {
1411        // Gradual decoding refresh SEI
1412        OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI);
1413        m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1414        m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1415
1416        SEIGradualDecodingRefreshInfo seiGradualDecodingRefreshInfo;
1417        seiGradualDecodingRefreshInfo.m_gdrForegroundFlag = true; // Indicating all "foreground"
1418
1419        m_seiWriter.writeSEImessage( nalu.m_Bitstream, seiGradualDecodingRefreshInfo, pcSlice->getSPS() );
1420        writeRBSPTrailingBits(nalu.m_Bitstream);
1421        accessUnit.push_back(new NALUnitEBSP(nalu));
1422      }
1423    // Recovery point SEI
1424      OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI);
1425      m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1426      m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1427
1428      SEIRecoveryPoint sei_recovery_point;
1429      sei_recovery_point.m_recoveryPocCnt    = 0;
1430      sei_recovery_point.m_exactMatchingFlag = ( pcSlice->getPOC() == 0 ) ? (true) : (false);
1431      sei_recovery_point.m_brokenLinkFlag    = false;
1432
1433      m_seiWriter.writeSEImessage( nalu.m_Bitstream, sei_recovery_point, pcSlice->getSPS() );
1434      writeRBSPTrailingBits(nalu.m_Bitstream);
1435      accessUnit.push_back(new NALUnitEBSP(nalu));
1436    }
1437
1438    /* use the main bitstream buffer for storing the marshalled picture */
1439    m_pcEntropyCoder->setBitstream(NULL);
1440
1441    startCUAddrSliceIdx = 0;
1442    startCUAddrSlice    = 0; 
1443
1444    startCUAddrSliceSegmentIdx = 0;
1445    startCUAddrSliceSegment    = 0; 
1446    nextCUAddr                 = 0;
1447    pcSlice = pcPic->getSlice(startCUAddrSliceIdx);
1448
1449    Int processingState = (pcSlice->getSPS()->getUseSAO())?(EXECUTE_INLOOPFILTER):(ENCODE_SLICE);
1450    Bool skippedSlice=false;
1451    while (nextCUAddr < uiRealEndAddress) // Iterate over all slices
1452    {
1453      switch(processingState)
1454      {
1455      case ENCODE_SLICE:
1456        {
1457          pcSlice->setNextSlice       ( false );
1458          pcSlice->setNextSliceSegment( false );
1459          if (nextCUAddr == m_storedStartCUAddrForEncodingSlice[startCUAddrSliceIdx])
1460          {
1461            pcSlice = pcPic->getSlice(startCUAddrSliceIdx);
1462            if(startCUAddrSliceIdx > 0 && pcSlice->getSliceType()!= I_SLICE)
1463            {
1464              pcSlice->checkColRefIdx(startCUAddrSliceIdx, pcPic);
1465            }
1466            pcPic->setCurrSliceIdx(startCUAddrSliceIdx);
1467            m_pcSliceEncoder->setSliceIdx(startCUAddrSliceIdx);
1468            assert(startCUAddrSliceIdx == pcSlice->getSliceIdx());
1469            // Reconstruction slice
1470            pcSlice->setSliceCurStartCUAddr( nextCUAddr );  // to be used in encodeSlice() + context restriction
1471            pcSlice->setSliceCurEndCUAddr  ( m_storedStartCUAddrForEncodingSlice[startCUAddrSliceIdx+1 ] );
1472            // Dependent slice
1473            pcSlice->setSliceSegmentCurStartCUAddr( nextCUAddr );  // to be used in encodeSlice() + context restriction
1474            pcSlice->setSliceSegmentCurEndCUAddr  ( m_storedStartCUAddrForEncodingSliceSegment[startCUAddrSliceSegmentIdx+1 ] );
1475
1476            pcSlice->setNextSlice       ( true );
1477
1478            startCUAddrSliceIdx++;
1479            startCUAddrSliceSegmentIdx++;
1480          } 
1481          else if (nextCUAddr == m_storedStartCUAddrForEncodingSliceSegment[startCUAddrSliceSegmentIdx])
1482          {
1483            // Dependent slice
1484            pcSlice->setSliceSegmentCurStartCUAddr( nextCUAddr );  // to be used in encodeSlice() + context restriction
1485            pcSlice->setSliceSegmentCurEndCUAddr  ( m_storedStartCUAddrForEncodingSliceSegment[startCUAddrSliceSegmentIdx+1 ] );
1486
1487            pcSlice->setNextSliceSegment( true );
1488
1489            startCUAddrSliceSegmentIdx++;
1490          }
1491
1492          pcSlice->setRPS(pcPic->getSlice(0)->getRPS());
1493          pcSlice->setRPSidx(pcPic->getSlice(0)->getRPSidx());
1494          UInt uiDummyStartCUAddr;
1495          UInt uiDummyBoundingCUAddr;
1496          m_pcSliceEncoder->xDetermineStartAndBoundingCUAddr(uiDummyStartCUAddr,uiDummyBoundingCUAddr,pcPic,true);
1497
1498          uiInternalAddress = pcPic->getPicSym()->getPicSCUAddr(pcSlice->getSliceSegmentCurEndCUAddr()-1) % pcPic->getNumPartInCU();
1499          uiExternalAddress = pcPic->getPicSym()->getPicSCUAddr(pcSlice->getSliceSegmentCurEndCUAddr()-1) / pcPic->getNumPartInCU();
1500          uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
1501          uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
1502          uiWidth = pcSlice->getSPS()->getPicWidthInLumaSamples();
1503          uiHeight = pcSlice->getSPS()->getPicHeightInLumaSamples();
1504          while(uiPosX>=uiWidth||uiPosY>=uiHeight)
1505          {
1506            uiInternalAddress--;
1507            uiPosX = ( uiExternalAddress % pcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
1508            uiPosY = ( uiExternalAddress / pcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
1509          }
1510          uiInternalAddress++;
1511          if(uiInternalAddress==pcPic->getNumPartInCU())
1512          {
1513            uiInternalAddress = 0;
1514            uiExternalAddress = pcPic->getPicSym()->getCUOrderMap(pcPic->getPicSym()->getInverseCUOrderMap(uiExternalAddress)+1);
1515          }
1516          UInt endAddress = pcPic->getPicSym()->getPicSCUEncOrder(uiExternalAddress*pcPic->getNumPartInCU()+uiInternalAddress);
1517          if(endAddress<=pcSlice->getSliceSegmentCurStartCUAddr()) 
1518          {
1519            UInt boundingAddrSlice, boundingAddrSliceSegment;
1520            boundingAddrSlice          = m_storedStartCUAddrForEncodingSlice[startCUAddrSliceIdx];         
1521            boundingAddrSliceSegment = m_storedStartCUAddrForEncodingSliceSegment[startCUAddrSliceSegmentIdx];         
1522            nextCUAddr               = min(boundingAddrSlice, boundingAddrSliceSegment);
1523            if(pcSlice->isNextSlice())
1524            {
1525              skippedSlice=true;
1526            }
1527            continue;
1528          }
1529          if(skippedSlice) 
1530          {
1531            pcSlice->setNextSlice       ( true );
1532            pcSlice->setNextSliceSegment( false );
1533          }
1534          skippedSlice=false;
1535          pcSlice->allocSubstreamSizes( iNumSubstreams );
1536          for ( UInt ui = 0 ; ui < iNumSubstreams; ui++ )
1537          {
1538            pcSubstreamsOut[ui].clear();
1539          }
1540
1541          m_pcEntropyCoder->setEntropyCoder   ( m_pcCavlcCoder, pcSlice );
1542          m_pcEntropyCoder->resetEntropy      ();
1543          /* start slice NALunit */
1544#if H_MV
1545          OutputNALUnit nalu( pcSlice->getNalUnitType(), pcSlice->getTLayer(), getLayerId() );
1546#else
1547          OutputNALUnit nalu( pcSlice->getNalUnitType(), pcSlice->getTLayer() );
1548#endif
1549          Bool sliceSegment = (!pcSlice->isNextSlice());
1550          if (!sliceSegment)
1551          {
1552            uiOneBitstreamPerSliceLength = 0; // start of a new slice
1553          }
1554          m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1555#if RATE_CONTROL_LAMBDA_DOMAIN
1556          tmpBitsBeforeWriting = m_pcEntropyCoder->getNumberOfWrittenBits();
1557#endif
1558          m_pcEntropyCoder->encodeSliceHeader(pcSlice);
1559#if RATE_CONTROL_LAMBDA_DOMAIN
1560          actualHeadBits += ( m_pcEntropyCoder->getNumberOfWrittenBits() - tmpBitsBeforeWriting );
1561#endif
1562
1563          // is it needed?
1564          {
1565            if (!sliceSegment)
1566            {
1567              pcBitstreamRedirect->writeAlignOne();
1568            }
1569            else
1570            {
1571              // We've not completed our slice header info yet, do the alignment later.
1572            }
1573            m_pcSbacCoder->init( (TEncBinIf*)m_pcBinCABAC );
1574            m_pcEntropyCoder->setEntropyCoder ( m_pcSbacCoder, pcSlice );
1575            m_pcEntropyCoder->resetEntropy    ();
1576            for ( UInt ui = 0 ; ui < pcSlice->getPPS()->getNumSubstreams() ; ui++ )
1577            {
1578              m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[ui], pcSlice );
1579              m_pcEntropyCoder->resetEntropy    ();
1580            }
1581          }
1582
1583          if(pcSlice->isNextSlice())
1584          {
1585            // set entropy coder for writing
1586            m_pcSbacCoder->init( (TEncBinIf*)m_pcBinCABAC );
1587            {
1588              for ( UInt ui = 0 ; ui < pcSlice->getPPS()->getNumSubstreams() ; ui++ )
1589              {
1590                m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[ui], pcSlice );
1591                m_pcEntropyCoder->resetEntropy    ();
1592              }
1593              pcSbacCoders[0].load(m_pcSbacCoder);
1594              m_pcEntropyCoder->setEntropyCoder ( &pcSbacCoders[0], pcSlice );  //ALF is written in substream #0 with CABAC coder #0 (see ALF param encoding below)
1595            }
1596            m_pcEntropyCoder->resetEntropy    ();
1597            // File writing
1598            if (!sliceSegment)
1599            {
1600              m_pcEntropyCoder->setBitstream(pcBitstreamRedirect);
1601            }
1602            else
1603            {
1604              m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1605            }
1606            // for now, override the TILES_DECODER setting in order to write substreams.
1607            m_pcEntropyCoder->setBitstream    ( &pcSubstreamsOut[0] );
1608
1609          }
1610          pcSlice->setFinalized(true);
1611
1612          m_pcSbacCoder->load( &pcSbacCoders[0] );
1613
1614          pcSlice->setTileOffstForMultES( uiOneBitstreamPerSliceLength );
1615            pcSlice->setTileLocationCount ( 0 );
1616          m_pcSliceEncoder->encodeSlice(pcPic, pcSubstreamsOut);
1617
1618          {
1619            // Construct the final bitstream by flushing and concatenating substreams.
1620            // The final bitstream is either nalu.m_Bitstream or pcBitstreamRedirect;
1621            UInt* puiSubstreamSizes = pcSlice->getSubstreamSizes();
1622            UInt uiTotalCodedSize = 0; // for padding calcs.
1623            UInt uiNumSubstreamsPerTile = iNumSubstreams;
1624            if (iNumSubstreams > 1)
1625            {
1626              uiNumSubstreamsPerTile /= pcPic->getPicSym()->getNumTiles();
1627            }
1628            for ( UInt ui = 0 ; ui < iNumSubstreams; ui++ )
1629            {
1630              // Flush all substreams -- this includes empty ones.
1631              // Terminating bit and flush.
1632              m_pcEntropyCoder->setEntropyCoder   ( &pcSbacCoders[ui], pcSlice );
1633              m_pcEntropyCoder->setBitstream      (  &pcSubstreamsOut[ui] );
1634              m_pcEntropyCoder->encodeTerminatingBit( 1 );
1635              m_pcEntropyCoder->encodeSliceFinish();
1636
1637              pcSubstreamsOut[ui].writeByteAlignment();   // Byte-alignment in slice_data() at end of sub-stream
1638              // Byte alignment is necessary between tiles when tiles are independent.
1639              uiTotalCodedSize += pcSubstreamsOut[ui].getNumberOfWrittenBits();
1640
1641              Bool bNextSubstreamInNewTile = ((ui+1) < iNumSubstreams)&& ((ui+1)%uiNumSubstreamsPerTile == 0);
1642              if (bNextSubstreamInNewTile)
1643              {
1644                pcSlice->setTileLocation(ui/uiNumSubstreamsPerTile, pcSlice->getTileOffstForMultES()+(uiTotalCodedSize>>3));
1645              }
1646              if (ui+1 < pcSlice->getPPS()->getNumSubstreams())
1647              {
1648                puiSubstreamSizes[ui] = pcSubstreamsOut[ui].getNumberOfWrittenBits() + (pcSubstreamsOut[ui].countStartCodeEmulations()<<3);
1649              }
1650            }
1651
1652            // Complete the slice header info.
1653            m_pcEntropyCoder->setEntropyCoder   ( m_pcCavlcCoder, pcSlice );
1654            m_pcEntropyCoder->setBitstream(&nalu.m_Bitstream);
1655            m_pcEntropyCoder->encodeTilesWPPEntryPoint( pcSlice );
1656
1657            // Substreams...
1658            TComOutputBitstream *pcOut = pcBitstreamRedirect;
1659          Int offs = 0;
1660          Int nss = pcSlice->getPPS()->getNumSubstreams();
1661          if (pcSlice->getPPS()->getEntropyCodingSyncEnabledFlag())
1662          {
1663            // 1st line present for WPP.
1664            offs = pcSlice->getSliceSegmentCurStartCUAddr()/pcSlice->getPic()->getNumPartInCU()/pcSlice->getPic()->getFrameWidthInCU();
1665            nss  = pcSlice->getNumEntryPointOffsets()+1;
1666          }
1667          for ( UInt ui = 0 ; ui < nss; ui++ )
1668          {
1669            pcOut->addSubstream(&pcSubstreamsOut[ui+offs]);
1670            }
1671          }
1672
1673          UInt boundingAddrSlice, boundingAddrSliceSegment;
1674          boundingAddrSlice        = m_storedStartCUAddrForEncodingSlice[startCUAddrSliceIdx];         
1675          boundingAddrSliceSegment = m_storedStartCUAddrForEncodingSliceSegment[startCUAddrSliceSegmentIdx];         
1676          nextCUAddr               = min(boundingAddrSlice, boundingAddrSliceSegment);
1677          // If current NALU is the first NALU of slice (containing slice header) and more NALUs exist (due to multiple dependent slices) then buffer it.
1678          // If current NALU is the last NALU of slice and a NALU was buffered, then (a) Write current NALU (b) Update an write buffered NALU at approproate location in NALU list.
1679          Bool bNALUAlignedWrittenToList    = false; // used to ensure current NALU is not written more than once to the NALU list.
1680          xAttachSliceDataToNalUnit(nalu, pcBitstreamRedirect);
1681          accessUnit.push_back(new NALUnitEBSP(nalu));
1682#if RATE_CONTROL_LAMBDA_DOMAIN
1683          actualTotalBits += UInt(accessUnit.back()->m_nalUnitData.str().size()) * 8;
1684#endif
1685          bNALUAlignedWrittenToList = true; 
1686          uiOneBitstreamPerSliceLength += nalu.m_Bitstream.getNumberOfWrittenBits(); // length of bitstream after byte-alignment
1687
1688          if (!bNALUAlignedWrittenToList)
1689          {
1690            {
1691              nalu.m_Bitstream.writeAlignZero();
1692            }
1693            accessUnit.push_back(new NALUnitEBSP(nalu));
1694            uiOneBitstreamPerSliceLength += nalu.m_Bitstream.getNumberOfWrittenBits() + 24; // length of bitstream after byte-alignment + 3 byte startcode 0x000001
1695          }
1696
1697          if( ( m_pcCfg->getPictureTimingSEIEnabled() || m_pcCfg->getDecodingUnitInfoSEIEnabled() ) &&
1698              ( pcSlice->getSPS()->getVuiParametersPresentFlag() ) &&
1699              ( ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getNalHrdParametersPresentFlag() ) 
1700             || ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getVclHrdParametersPresentFlag() ) ) &&
1701              ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getSubPicCpbParamsPresentFlag() ) )
1702          {
1703              UInt numNalus = 0;
1704            UInt numRBSPBytes = 0;
1705            for (AccessUnit::const_iterator it = accessUnit.begin(); it != accessUnit.end(); it++)
1706            {
1707              UInt numRBSPBytes_nal = UInt((*it)->m_nalUnitData.str().size());
1708              if ((*it)->m_nalUnitType != NAL_UNIT_PREFIX_SEI && (*it)->m_nalUnitType != NAL_UNIT_SUFFIX_SEI)
1709              {
1710                numRBSPBytes += numRBSPBytes_nal;
1711                numNalus ++;
1712              }
1713            }
1714            accumBitsDU[ pcSlice->getSliceIdx() ] = ( numRBSPBytes << 3 );
1715            accumNalsDU[ pcSlice->getSliceIdx() ] = numNalus;   // SEI not counted for bit count; hence shouldn't be counted for # of NALUs - only for consistency
1716          }
1717          processingState = ENCODE_SLICE;
1718          }
1719          break;
1720        case EXECUTE_INLOOPFILTER:
1721          {
1722            // set entropy coder for RD
1723            m_pcEntropyCoder->setEntropyCoder ( m_pcSbacCoder, pcSlice );
1724            if ( pcSlice->getSPS()->getUseSAO() )
1725            {
1726              m_pcEntropyCoder->resetEntropy();
1727              m_pcEntropyCoder->setBitstream( m_pcBitCounter );
1728              m_pcSAO->startSaoEnc(pcPic, m_pcEntropyCoder, m_pcEncTop->getRDSbacCoder(), m_pcEncTop->getRDGoOnSbacCoder());
1729              SAOParam& cSaoParam = *pcSlice->getPic()->getPicSym()->getSaoParam();
1730
1731#if SAO_CHROMA_LAMBDA
1732#if SAO_ENCODING_CHOICE
1733              m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma(), pcPic->getSlice(0)->getDepth());
1734#else
1735              m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambdaLuma(), pcPic->getSlice(0)->getLambdaChroma());
1736#endif
1737#else
1738              m_pcSAO->SAOProcess(&cSaoParam, pcPic->getSlice(0)->getLambda());
1739#endif
1740              m_pcSAO->endSaoEnc();
1741              m_pcSAO->PCMLFDisableProcess(pcPic);
1742            }
1743#if SAO_RDO
1744            m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice );
1745#endif
1746            processingState = ENCODE_SLICE;
1747
1748            for(Int s=0; s< uiNumSlices; s++)
1749            {
1750              if (pcSlice->getSPS()->getUseSAO())
1751              {
1752                pcPic->getSlice(s)->setSaoEnabledFlag((pcSlice->getPic()->getPicSym()->getSaoParam()->bSaoFlag[0]==1)?true:false);
1753              }
1754            }
1755          }
1756          break;
1757        default:
1758          {
1759            printf("Not a supported encoding state\n");
1760            assert(0);
1761            exit(-1);
1762          }
1763        }
1764      } // end iteration over slices
1765
1766      if(pcSlice->getSPS()->getUseSAO())
1767      {
1768        if(pcSlice->getSPS()->getUseSAO())
1769        {
1770          m_pcSAO->destroyPicSaoInfo();
1771        }
1772        pcPic->destroyNonDBFilterInfo();
1773      }
1774
1775#if !H_3D
1776      pcPic->compressMotion(); 
1777#endif
1778#if H_MV
1779      m_pocLastCoded = pcPic->getPOC();
1780#endif
1781
1782      //-- For time output for each slice
1783      Double dEncTime = (Double)(clock()-iBeforeTime) / CLOCKS_PER_SEC;
1784
1785      const Char* digestStr = NULL;
1786      if (m_pcCfg->getDecodedPictureHashSEIEnabled())
1787      {
1788        /* calculate MD5sum for entire reconstructed picture */
1789        SEIDecodedPictureHash sei_recon_picture_digest;
1790        if(m_pcCfg->getDecodedPictureHashSEIEnabled() == 1)
1791        {
1792          sei_recon_picture_digest.method = SEIDecodedPictureHash::MD5;
1793          calcMD5(*pcPic->getPicYuvRec(), sei_recon_picture_digest.digest);
1794          digestStr = digestToString(sei_recon_picture_digest.digest, 16);
1795        }
1796        else if(m_pcCfg->getDecodedPictureHashSEIEnabled() == 2)
1797        {
1798          sei_recon_picture_digest.method = SEIDecodedPictureHash::CRC;
1799          calcCRC(*pcPic->getPicYuvRec(), sei_recon_picture_digest.digest);
1800          digestStr = digestToString(sei_recon_picture_digest.digest, 2);
1801        }
1802        else if(m_pcCfg->getDecodedPictureHashSEIEnabled() == 3)
1803        {
1804          sei_recon_picture_digest.method = SEIDecodedPictureHash::CHECKSUM;
1805          calcChecksum(*pcPic->getPicYuvRec(), sei_recon_picture_digest.digest);
1806          digestStr = digestToString(sei_recon_picture_digest.digest, 4);
1807        }
1808        OutputNALUnit nalu(NAL_UNIT_SUFFIX_SEI, pcSlice->getTLayer());
1809
1810        /* write the SEI messages */
1811        m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1812        m_seiWriter.writeSEImessage(nalu.m_Bitstream, sei_recon_picture_digest, pcSlice->getSPS());
1813        writeRBSPTrailingBits(nalu.m_Bitstream);
1814
1815        accessUnit.insert(accessUnit.end(), new NALUnitEBSP(nalu));
1816      }
1817      if (m_pcCfg->getTemporalLevel0IndexSEIEnabled())
1818      {
1819        SEITemporalLevel0Index sei_temporal_level0_index;
1820        if (pcSlice->getRapPicFlag())
1821        {
1822          m_tl0Idx = 0;
1823          m_rapIdx = (m_rapIdx + 1) & 0xFF;
1824        }
1825        else
1826        {
1827          m_tl0Idx = (m_tl0Idx + (pcSlice->getTLayer() ? 0 : 1)) & 0xFF;
1828        }
1829        sei_temporal_level0_index.tl0Idx = m_tl0Idx;
1830        sei_temporal_level0_index.rapIdx = m_rapIdx;
1831
1832        OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI); 
1833
1834        /* write the SEI messages */
1835        m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1836        m_seiWriter.writeSEImessage(nalu.m_Bitstream, sei_temporal_level0_index, pcSlice->getSPS());
1837        writeRBSPTrailingBits(nalu.m_Bitstream);
1838
1839        /* insert the SEI message NALUnit before any Slice NALUnits */
1840        AccessUnit::iterator it = find_if(accessUnit.begin(), accessUnit.end(), mem_fun(&NALUnit::isSlice));
1841        accessUnit.insert(it, new NALUnitEBSP(nalu));
1842      }
1843
1844      xCalculateAddPSNR( pcPic, pcPic->getPicYuvRec(), accessUnit, dEncTime );
1845
1846      if (digestStr)
1847      {
1848        if(m_pcCfg->getDecodedPictureHashSEIEnabled() == 1)
1849        {
1850          printf(" [MD5:%s]", digestStr);
1851        }
1852        else if(m_pcCfg->getDecodedPictureHashSEIEnabled() == 2)
1853        {
1854          printf(" [CRC:%s]", digestStr);
1855        }
1856        else if(m_pcCfg->getDecodedPictureHashSEIEnabled() == 3)
1857        {
1858          printf(" [Checksum:%s]", digestStr);
1859        }
1860      }
1861#if RATE_CONTROL_LAMBDA_DOMAIN
1862      if ( m_pcCfg->getUseRateCtrl() )
1863      {
1864        Double effectivePercentage = m_pcRateCtrl->getRCPic()->getEffectivePercentage();
1865        Double avgQP     = m_pcRateCtrl->getRCPic()->calAverageQP();
1866        Double avgLambda = m_pcRateCtrl->getRCPic()->calAverageLambda();
1867        if ( avgLambda < 0.0 )
1868        {
1869          avgLambda = lambda;
1870        }
1871        m_pcRateCtrl->getRCPic()->updateAfterPicture( actualHeadBits, actualTotalBits, avgQP, avgLambda, effectivePercentage );
1872        m_pcRateCtrl->getRCPic()->addToPictureLsit( m_pcRateCtrl->getPicList() );
1873
1874        m_pcRateCtrl->getRCSeq()->updateAfterPic( actualTotalBits );
1875        if ( pcSlice->getSliceType() != I_SLICE )
1876        {
1877          m_pcRateCtrl->getRCGOP()->updateAfterPicture( actualTotalBits );
1878        }
1879        else    // for intra picture, the estimated bits are used to update the current status in the GOP
1880        {
1881          m_pcRateCtrl->getRCGOP()->updateAfterPicture( estimatedBits );
1882        }
1883      }
1884#else
1885      if(m_pcCfg->getUseRateCtrl())
1886      {
1887        UInt  frameBits = m_vRVM_RP[m_vRVM_RP.size()-1];
1888        m_pcRateCtrl->updataRCFrameStatus((Int)frameBits, pcSlice->getSliceType());
1889      }
1890#endif
1891      if( ( m_pcCfg->getPictureTimingSEIEnabled() || m_pcCfg->getDecodingUnitInfoSEIEnabled() ) &&
1892          ( pcSlice->getSPS()->getVuiParametersPresentFlag() ) &&
1893          ( ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getNalHrdParametersPresentFlag() ) 
1894         || ( pcSlice->getSPS()->getVuiParameters()->getHrdParameters()->getVclHrdParametersPresentFlag() ) ) )
1895      {
1896        TComVUI *vui = pcSlice->getSPS()->getVuiParameters();
1897        TComHRD *hrd = vui->getHrdParameters();
1898
1899        if( hrd->getSubPicCpbParamsPresentFlag() )
1900        {
1901          Int i;
1902          UInt64 ui64Tmp;
1903          UInt uiPrev = 0;
1904          UInt numDU = ( pictureTimingSEI.m_numDecodingUnitsMinus1 + 1 );
1905          UInt *pCRD = &pictureTimingSEI.m_duCpbRemovalDelayMinus1[0];
1906          UInt maxDiff = ( hrd->getTickDivisorMinus2() + 2 ) - 1;
1907
1908          for( i = 0; i < numDU; i ++ )
1909          {
1910            pictureTimingSEI.m_numNalusInDuMinus1[ i ]       = ( i == 0 ) ? ( accumNalsDU[ i ] - 1 ) : ( accumNalsDU[ i ] - accumNalsDU[ i - 1] - 1 );
1911          }
1912
1913          if( numDU == 1 )
1914          {
1915            pCRD[ 0 ] = 0; /* don't care */
1916          }
1917          else
1918          {
1919            pCRD[ numDU - 1 ] = 0;/* by definition */
1920            UInt tmp = 0;
1921            UInt accum = 0;
1922
1923            for( i = ( numDU - 2 ); i >= 0; i -- )
1924            {
1925#if L0043_TIMING_INFO
1926              ui64Tmp = ( ( ( accumBitsDU[ numDU - 1 ]  - accumBitsDU[ i ] ) * ( vui->getTimingInfo()->getTimeScale() / vui->getTimingInfo()->getNumUnitsInTick() ) * ( hrd->getTickDivisorMinus2() + 2 ) ) / ( m_pcCfg->getTargetBitrate() ) );
1927#else
1928              ui64Tmp = ( ( ( accumBitsDU[ numDU - 1 ]  - accumBitsDU[ i ] ) * ( hrd->getTimeScale() / hrd->getNumUnitsInTick() ) * ( hrd->getTickDivisorMinus2() + 2 ) ) / ( m_pcCfg->getTargetBitrate() ) );
1929#endif
1930              if( (UInt)ui64Tmp > maxDiff )
1931              {
1932                tmp ++;
1933              }
1934            }
1935            uiPrev = 0;
1936
1937            UInt flag = 0;
1938            for( i = ( numDU - 2 ); i >= 0; i -- )
1939            {
1940              flag = 0;
1941#if L0043_TIMING_INFO
1942              ui64Tmp = ( ( ( accumBitsDU[ numDU - 1 ]  - accumBitsDU[ i ] ) * ( vui->getTimingInfo()->getTimeScale() / vui->getTimingInfo()->getNumUnitsInTick() ) * ( hrd->getTickDivisorMinus2() + 2 ) ) / ( m_pcCfg->getTargetBitrate() ) );
1943#else
1944              ui64Tmp = ( ( ( accumBitsDU[ numDU - 1 ]  - accumBitsDU[ i ] ) * ( hrd->getTimeScale() / hrd->getNumUnitsInTick() ) * ( hrd->getTickDivisorMinus2() + 2 ) ) / ( m_pcCfg->getTargetBitrate() ) );
1945#endif
1946
1947              if( (UInt)ui64Tmp > maxDiff )
1948              {
1949                if(uiPrev >= maxDiff - tmp)
1950                {
1951                  ui64Tmp = uiPrev + 1;
1952                  flag = 1;
1953                }
1954                else                            ui64Tmp = maxDiff - tmp + 1;
1955              }
1956              pCRD[ i ] = (UInt)ui64Tmp - uiPrev - 1;
1957              if( (Int)pCRD[ i ] < 0 )
1958              {
1959                pCRD[ i ] = 0;
1960              }
1961              else if (tmp > 0 && flag == 1) 
1962              {
1963                tmp --;
1964              }
1965              accum += pCRD[ i ] + 1;
1966              uiPrev = accum;
1967            }
1968          }
1969        }
1970        if( m_pcCfg->getPictureTimingSEIEnabled() )
1971        {
1972          {
1973            OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI, pcSlice->getTLayer());
1974          m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1975          m_seiWriter.writeSEImessage(nalu.m_Bitstream, pictureTimingSEI, pcSlice->getSPS());
1976          writeRBSPTrailingBits(nalu.m_Bitstream);
1977#if L0045_NON_NESTED_SEI_RESTRICTIONS
1978          UInt seiPositionInAu = xGetFirstSeiLocation(accessUnit);
1979          UInt offsetPosition = m_activeParameterSetSEIPresentInAU
1980                                    + m_bufferingPeriodSEIPresentInAU;    // Insert PT SEI after APS and BP SEI
1981          AccessUnit::iterator it;
1982          for(j = 0, it = accessUnit.begin(); j < seiPositionInAu + offsetPosition; j++)
1983          {
1984            it++;
1985          }
1986          accessUnit.insert(it, new NALUnitEBSP(nalu));
1987          m_pictureTimingSEIPresentInAU = true;
1988#else
1989          AccessUnit::iterator it = find_if(accessUnit.begin(), accessUnit.end(), mem_fun(&NALUnit::isSlice));
1990          accessUnit.insert(it, new NALUnitEBSP(nalu));
1991#endif
1992        }
1993#if K0180_SCALABLE_NESTING_SEI
1994          if ( m_pcCfg->getScalableNestingSEIEnabled() ) // put picture timing SEI into scalable nesting SEI
1995          {
1996            OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI, pcSlice->getTLayer());
1997            m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
1998            scalableNestingSEI.m_nestedSEIs.clear();
1999            scalableNestingSEI.m_nestedSEIs.push_back(&pictureTimingSEI);
2000            m_seiWriter.writeSEImessage(nalu.m_Bitstream, scalableNestingSEI, pcSlice->getSPS());
2001            writeRBSPTrailingBits(nalu.m_Bitstream);
2002#if L0045_NON_NESTED_SEI_RESTRICTIONS
2003            UInt seiPositionInAu = xGetFirstSeiLocation(accessUnit);
2004            UInt offsetPosition = m_activeParameterSetSEIPresentInAU
2005              + m_bufferingPeriodSEIPresentInAU + m_pictureTimingSEIPresentInAU + m_nestedBufferingPeriodSEIPresentInAU;    // Insert PT SEI after APS and BP SEI
2006            AccessUnit::iterator it;
2007            for(j = 0, it = accessUnit.begin(); j < seiPositionInAu + offsetPosition; j++)
2008            {
2009              it++;
2010            }
2011            accessUnit.insert(it, new NALUnitEBSP(nalu));
2012            m_nestedPictureTimingSEIPresentInAU = true;
2013#else
2014            AccessUnit::iterator it = find_if(accessUnit.begin(), accessUnit.end(), mem_fun(&NALUnit::isSlice));
2015            accessUnit.insert(it, new NALUnitEBSP(nalu));
2016#endif
2017          }
2018#endif
2019
2020        }
2021        if( m_pcCfg->getDecodingUnitInfoSEIEnabled() && hrd->getSubPicCpbParamsPresentFlag() )
2022        {             
2023          m_pcEntropyCoder->setEntropyCoder(m_pcCavlcCoder, pcSlice);
2024          for( Int i = 0; i < ( pictureTimingSEI.m_numDecodingUnitsMinus1 + 1 ); i ++ )
2025          {
2026            OutputNALUnit nalu(NAL_UNIT_PREFIX_SEI, pcSlice->getTLayer());
2027
2028            SEIDecodingUnitInfo tempSEI;
2029            tempSEI.m_decodingUnitIdx = i;
2030            tempSEI.m_duSptCpbRemovalDelay = pictureTimingSEI.m_duCpbRemovalDelayMinus1[i] + 1;
2031#if L0044_DU_DPB_OUTPUT_DELAY_HRD
2032            tempSEI.m_dpbOutputDuDelayPresentFlag = false;
2033            tempSEI.m_picSptDpbOutputDuDelay = picSptDpbOutputDuDelay;
2034#endif
2035
2036            AccessUnit::iterator it;
2037            // Insert the first one in the right location, before the first slice
2038            if(i == 0)
2039            {
2040              // Insert before the first slice.
2041              m_seiWriter.writeSEImessage(nalu.m_Bitstream, tempSEI, pcSlice->getSPS());
2042              writeRBSPTrailingBits(nalu.m_Bitstream);
2043
2044#if L0045_NON_NESTED_SEI_RESTRICTIONS
2045              UInt seiPositionInAu = xGetFirstSeiLocation(accessUnit);
2046              UInt offsetPosition = m_activeParameterSetSEIPresentInAU
2047                                    + m_bufferingPeriodSEIPresentInAU
2048                                    + m_pictureTimingSEIPresentInAU;  // Insert DU info SEI after APS, BP and PT SEI
2049              for(j = 0, it = accessUnit.begin(); j < seiPositionInAu + offsetPosition; j++)
2050              {
2051                it++;
2052              }
2053              accessUnit.insert(it, new NALUnitEBSP(nalu));
2054#else
2055              it = find_if(accessUnit.begin(), accessUnit.end(), mem_fun(&NALUnit::isSlice));
2056              accessUnit.insert(it, new NALUnitEBSP(nalu)); 
2057#endif
2058            }
2059            else
2060            {
2061              Int ctr;
2062              // For the second decoding unit onwards we know how many NALUs are present
2063              for (ctr = 0, it = accessUnit.begin(); it != accessUnit.end(); it++)
2064              {           
2065                if(ctr == accumNalsDU[ i - 1 ])
2066                {
2067                  // Insert before the first slice.
2068                  m_seiWriter.writeSEImessage(nalu.m_Bitstream, tempSEI, pcSlice->getSPS());
2069                  writeRBSPTrailingBits(nalu.m_Bitstream);
2070
2071                  accessUnit.insert(it, new NALUnitEBSP(nalu));
2072                  break;
2073                }
2074                if ((*it)->m_nalUnitType != NAL_UNIT_PREFIX_SEI && (*it)->m_nalUnitType != NAL_UNIT_SUFFIX_SEI)
2075                {
2076                  ctr++;
2077                }
2078              }
2079            }           
2080          }
2081        }
2082      }
2083#if L0045_NON_NESTED_SEI_RESTRICTIONS
2084      xResetNonNestedSEIPresentFlags();
2085#if K0180_SCALABLE_NESTING_SEI
2086      xResetNestedSEIPresentFlags();
2087#endif
2088#endif
2089      pcPic->getPicYuvRec()->copyToPic(pcPicYuvRecOut);
2090
2091      pcPic->setReconMark   ( true );
2092#if H_MV
2093      TComSlice::markIvRefPicsAsShortTerm( m_refPicSetInterLayer ); 
2094      std::vector<Int> temp; 
2095      TComSlice::markIvRefPicsAsUnused   ( m_ivPicLists, temp, pcPic->getSlice(0)->getVPS(), m_layerId, pcPic->getPOC() ); 
2096#endif
2097      m_bFirst = false;
2098      m_iNumPicCoded++;
2099      m_totalCoded ++;
2100      /* logging: insert a newline at end of picture period */
2101      printf("\n");
2102      fflush(stdout);
2103
2104      delete[] pcSubstreamsOut;
2105  }
2106#if !RATE_CONTROL_LAMBDA_DOMAIN
2107  if(m_pcCfg->getUseRateCtrl())
2108  {
2109    m_pcRateCtrl->updateRCGOPStatus();
2110  }
2111#endif
2112  delete pcBitstreamRedirect;
2113
2114  if( accumBitsDU != NULL) delete accumBitsDU;
2115  if( accumNalsDU != NULL) delete accumNalsDU;
2116
2117#if !H_MV
2118  assert ( m_iNumPicCoded == iNumPicRcvd );
2119#endif
2120}
2121
2122#if !H_MV
2123Void TEncGOP::printOutSummary(UInt uiNumAllPicCoded)
2124{
2125  assert (uiNumAllPicCoded == m_gcAnalyzeAll.getNumPic());
2126 
2127   
2128  //--CFG_KDY
2129  m_gcAnalyzeAll.setFrmRate( m_pcCfg->getFrameRate() );
2130  m_gcAnalyzeI.setFrmRate( m_pcCfg->getFrameRate() );
2131  m_gcAnalyzeP.setFrmRate( m_pcCfg->getFrameRate() );
2132  m_gcAnalyzeB.setFrmRate( m_pcCfg->getFrameRate() );
2133 
2134  //-- all
2135  printf( "\n\nSUMMARY --------------------------------------------------------\n" );
2136  m_gcAnalyzeAll.printOut('a');
2137 
2138  printf( "\n\nI Slices--------------------------------------------------------\n" );
2139  m_gcAnalyzeI.printOut('i');
2140 
2141  printf( "\n\nP Slices--------------------------------------------------------\n" );
2142  m_gcAnalyzeP.printOut('p');
2143 
2144  printf( "\n\nB Slices--------------------------------------------------------\n" );
2145  m_gcAnalyzeB.printOut('b');
2146 
2147#if _SUMMARY_OUT_
2148  m_gcAnalyzeAll.printSummaryOut();
2149#endif
2150#if _SUMMARY_PIC_
2151  m_gcAnalyzeI.printSummary('I');
2152  m_gcAnalyzeP.printSummary('P');
2153  m_gcAnalyzeB.printSummary('B');
2154#endif
2155
2156  printf("\nRVM: %.3lf\n" , xCalculateRVM());
2157}
2158#endif
2159#if H_3D_VSO
2160Void TEncGOP::preLoopFilterPicAll( TComPic* pcPic, Dist64& ruiDist, UInt64& ruiBits )
2161#else
2162Void TEncGOP::preLoopFilterPicAll( TComPic* pcPic, UInt64& ruiDist, UInt64& ruiBits )
2163#endif
2164{
2165  TComSlice* pcSlice = pcPic->getSlice(pcPic->getCurrSliceIdx());
2166  Bool bCalcDist = false;
2167  m_pcLoopFilter->setCfg(m_pcCfg->getLFCrossTileBoundaryFlag());
2168  m_pcLoopFilter->loopFilterPic( pcPic );
2169 
2170  m_pcEntropyCoder->setEntropyCoder ( m_pcEncTop->getRDGoOnSbacCoder(), pcSlice );
2171  m_pcEntropyCoder->resetEntropy    ();
2172  m_pcEntropyCoder->setBitstream    ( m_pcBitCounter );
2173  pcSlice = pcPic->getSlice(0);
2174  if(pcSlice->getSPS()->getUseSAO())
2175  {
2176    std::vector<Bool> LFCrossSliceBoundaryFlag(1, true);
2177    std::vector<Int>  sliceStartAddress;
2178    sliceStartAddress.push_back(0);
2179    sliceStartAddress.push_back(pcPic->getNumCUsInFrame()* pcPic->getNumPartInCU());
2180    pcPic->createNonDBFilterInfo(sliceStartAddress, 0, &LFCrossSliceBoundaryFlag);
2181  }
2182 
2183  if( pcSlice->getSPS()->getUseSAO())
2184  {
2185    pcPic->destroyNonDBFilterInfo();
2186  }
2187 
2188  m_pcEntropyCoder->resetEntropy    ();
2189  ruiBits += m_pcEntropyCoder->getNumberOfWrittenBits();
2190 
2191  if (!bCalcDist)
2192    ruiDist = xFindDistortionFrame(pcPic->getPicYuvOrg(), pcPic->getPicYuvRec());
2193}
2194
2195// ====================================================================================================================
2196// Protected member functions
2197// ====================================================================================================================
2198
2199Void TEncGOP::xInitGOP( Int iPOCLast, Int iNumPicRcvd, TComList<TComPic*>& rcListPic, TComList<TComPicYuv*>& rcListPicYuvRecOut )
2200{
2201  assert( iNumPicRcvd > 0 );
2202  //  Exception for the first frame
2203  if ( iPOCLast == 0 )
2204  {
2205    m_iGopSize    = 1;
2206  }
2207  else
2208    m_iGopSize    = m_pcCfg->getGOPSize();
2209 
2210  assert (m_iGopSize > 0); 
2211
2212  return;
2213}
2214
2215Void TEncGOP::xGetBuffer( TComList<TComPic*>&      rcListPic,
2216                         TComList<TComPicYuv*>&    rcListPicYuvRecOut,
2217                         Int                       iNumPicRcvd,
2218                         Int                       iTimeOffset,
2219                         TComPic*&                 rpcPic,
2220                         TComPicYuv*&              rpcPicYuvRecOut,
2221                         Int                       pocCurr )
2222{
2223  Int i;
2224  //  Rec. output
2225  TComList<TComPicYuv*>::iterator     iterPicYuvRec = rcListPicYuvRecOut.end();
2226  for ( i = 0; i < iNumPicRcvd - iTimeOffset + 1; i++ )
2227  {
2228    iterPicYuvRec--;
2229  }
2230 
2231  rpcPicYuvRecOut = *(iterPicYuvRec);
2232 
2233  //  Current pic.
2234  TComList<TComPic*>::iterator        iterPic       = rcListPic.begin();
2235  while (iterPic != rcListPic.end())
2236  {
2237    rpcPic = *(iterPic);
2238    rpcPic->setCurrSliceIdx(0);
2239    if (rpcPic->getPOC() == pocCurr)
2240    {
2241      break;
2242    }
2243    iterPic++;
2244  }
2245 
2246  assert (rpcPic->getPOC() == pocCurr);
2247 
2248  return;
2249}
2250
2251#if H_3D_VSO
2252Dist64 TEncGOP::xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1)
2253#else
2254UInt64 TEncGOP::xFindDistortionFrame (TComPicYuv* pcPic0, TComPicYuv* pcPic1)
2255#endif
2256{
2257  Int     x, y;
2258  Pel*  pSrc0   = pcPic0 ->getLumaAddr();
2259  Pel*  pSrc1   = pcPic1 ->getLumaAddr();
2260  UInt  uiShift = 2 * DISTORTION_PRECISION_ADJUSTMENT(g_bitDepthY-8);
2261  Int   iTemp;
2262 
2263  Int   iStride = pcPic0->getStride();
2264  Int   iWidth  = pcPic0->getWidth();
2265  Int   iHeight = pcPic0->getHeight();
2266 
2267#if H_3D_VSO
2268  Dist64  uiTotalDiff = 0;
2269#else
2270  UInt64  uiTotalDiff = 0;
2271#endif
2272 
2273  for( y = 0; y < iHeight; y++ )
2274  {
2275    for( x = 0; x < iWidth; x++ )
2276    {
2277      iTemp = pSrc0[x] - pSrc1[x]; uiTotalDiff += (iTemp*iTemp) >> uiShift;
2278    }
2279    pSrc0 += iStride;
2280    pSrc1 += iStride;
2281  }
2282 
2283  uiShift = 2 * DISTORTION_PRECISION_ADJUSTMENT(g_bitDepthC-8);
2284  iHeight >>= 1;
2285  iWidth  >>= 1;
2286  iStride >>= 1;
2287 
2288  pSrc0  = pcPic0->getCbAddr();
2289  pSrc1  = pcPic1->getCbAddr();
2290 
2291  for( y = 0; y < iHeight; y++ )
2292  {
2293    for( x = 0; x < iWidth; x++ )
2294    {
2295      iTemp = pSrc0[x] - pSrc1[x]; uiTotalDiff += (iTemp*iTemp) >> uiShift;
2296    }
2297    pSrc0 += iStride;
2298    pSrc1 += iStride;
2299  }
2300 
2301  pSrc0  = pcPic0->getCrAddr();
2302  pSrc1  = pcPic1->getCrAddr();
2303 
2304  for( y = 0; y < iHeight; y++ )
2305  {
2306    for( x = 0; x < iWidth; x++ )
2307    {
2308      iTemp = pSrc0[x] - pSrc1[x]; uiTotalDiff += (iTemp*iTemp) >> uiShift;
2309    }
2310    pSrc0 += iStride;
2311    pSrc1 += iStride;
2312  }
2313 
2314  return uiTotalDiff;
2315}
2316
2317#if VERBOSE_RATE
2318static const Char* nalUnitTypeToString(NalUnitType type)
2319{
2320  switch (type)
2321  {
2322    case NAL_UNIT_CODED_SLICE_TRAIL_R: return "TRAIL_R";
2323    case NAL_UNIT_CODED_SLICE_TRAIL_N: return "TRAIL_N";
2324    case NAL_UNIT_CODED_SLICE_TLA_R:      return "TLA_R";
2325    case NAL_UNIT_CODED_SLICE_TSA_N: return "TSA_N";
2326    case NAL_UNIT_CODED_SLICE_STSA_R: return "STSA_R";
2327    case NAL_UNIT_CODED_SLICE_STSA_N: return "STSA_N";
2328    case NAL_UNIT_CODED_SLICE_BLA_W_LP:   return "BLA_W_LP";
2329    case NAL_UNIT_CODED_SLICE_BLA_W_RADL: return "BLA_W_RADL";
2330    case NAL_UNIT_CODED_SLICE_BLA_N_LP: return "BLA_N_LP";
2331    case NAL_UNIT_CODED_SLICE_IDR_W_RADL: return "IDR_W_RADL";
2332    case NAL_UNIT_CODED_SLICE_IDR_N_LP: return "IDR_N_LP";
2333    case NAL_UNIT_CODED_SLICE_CRA: return "CRA";
2334    case NAL_UNIT_CODED_SLICE_RADL_R:     return "RADL_R";
2335    case NAL_UNIT_CODED_SLICE_RASL_R:     return "RASL_R";
2336    case NAL_UNIT_VPS: return "VPS";
2337    case NAL_UNIT_SPS: return "SPS";
2338    case NAL_UNIT_PPS: return "PPS";
2339    case NAL_UNIT_ACCESS_UNIT_DELIMITER: return "AUD";
2340    case NAL_UNIT_EOS: return "EOS";
2341    case NAL_UNIT_EOB: return "EOB";
2342    case NAL_UNIT_FILLER_DATA: return "FILLER";
2343    case NAL_UNIT_PREFIX_SEI:             return "SEI";
2344    case NAL_UNIT_SUFFIX_SEI:             return "SEI";
2345    default: return "UNK";
2346  }
2347}
2348#endif
2349
2350Void TEncGOP::xCalculateAddPSNR( TComPic* pcPic, TComPicYuv* pcPicD, const AccessUnit& accessUnit, Double dEncTime )
2351{
2352  Int     x, y;
2353  UInt64 uiSSDY  = 0;
2354  UInt64 uiSSDU  = 0;
2355  UInt64 uiSSDV  = 0;
2356 
2357  Double  dYPSNR  = 0.0;
2358  Double  dUPSNR  = 0.0;
2359  Double  dVPSNR  = 0.0;
2360 
2361  //===== calculate PSNR =====
2362  Pel*  pOrg    = pcPic ->getPicYuvOrg()->getLumaAddr();
2363  Pel*  pRec    = pcPicD->getLumaAddr();
2364  Int   iStride = pcPicD->getStride();
2365 
2366  Int   iWidth;
2367  Int   iHeight;
2368 
2369  iWidth  = pcPicD->getWidth () - m_pcEncTop->getPad(0);
2370  iHeight = pcPicD->getHeight() - m_pcEncTop->getPad(1);
2371 
2372  Int   iSize   = iWidth*iHeight;
2373 
2374  for( y = 0; y < iHeight; y++ )
2375  {
2376    for( x = 0; x < iWidth; x++ )
2377    {
2378      Int iDiff = (Int)( pOrg[x] - pRec[x] );
2379      uiSSDY   += iDiff * iDiff;
2380    }
2381    pOrg += iStride;
2382    pRec += iStride;
2383  }
2384 
2385#if H_3D_VSO
2386#if H_3D_VSO_SYNTH_DIST_OUT
2387  if ( m_pcRdCost->getUseRenModel() )
2388  {
2389    unsigned int maxval = 255 * (1<<(g_uiBitDepth + g_uiBitIncrement -8));
2390    Double fRefValueY = (double) maxval * maxval * iSize;
2391    Double fRefValueC = fRefValueY / 4.0;
2392    TRenModel*  pcRenModel = m_pcEncTop->getEncTop()->getRenModel();
2393    Int64 iDistVSOY, iDistVSOU, iDistVSOV;
2394    pcRenModel->getTotalSSE( iDistVSOY, iDistVSOU, iDistVSOV );
2395    dYPSNR = ( iDistVSOY ? 10.0 * log10( fRefValueY / (Double) iDistVSOY ) : 99.99 );
2396    dUPSNR = ( iDistVSOU ? 10.0 * log10( fRefValueC / (Double) iDistVSOU ) : 99.99 );
2397    dVPSNR = ( iDistVSOV ? 10.0 * log10( fRefValueC / (Double) iDistVSOV ) : 99.99 );
2398  }
2399  else
2400  {
2401#endif
2402#endif
2403    iHeight >>= 1;
2404  iWidth  >>= 1;
2405  iStride >>= 1;
2406  pOrg  = pcPic ->getPicYuvOrg()->getCbAddr();
2407  pRec  = pcPicD->getCbAddr();
2408 
2409  for( y = 0; y < iHeight; y++ )
2410  {
2411    for( x = 0; x < iWidth; x++ )
2412    {
2413      Int iDiff = (Int)( pOrg[x] - pRec[x] );
2414      uiSSDU   += iDiff * iDiff;
2415    }
2416    pOrg += iStride;
2417    pRec += iStride;
2418  }
2419 
2420  pOrg  = pcPic ->getPicYuvOrg()->getCrAddr();
2421  pRec  = pcPicD->getCrAddr();
2422 
2423  for( y = 0; y < iHeight; y++ )
2424  {
2425    for( x = 0; x < iWidth; x++ )
2426    {
2427      Int iDiff = (Int)( pOrg[x] - pRec[x] );
2428      uiSSDV   += iDiff * iDiff;
2429    }
2430    pOrg += iStride;
2431    pRec += iStride;
2432  }
2433 
2434  Int maxvalY = 255 << (g_bitDepthY-8);
2435  Int maxvalC = 255 << (g_bitDepthC-8);
2436  Double fRefValueY = (Double) maxvalY * maxvalY * iSize;
2437  Double fRefValueC = (Double) maxvalC * maxvalC * iSize / 4.0;
2438  dYPSNR            = ( uiSSDY ? 10.0 * log10( fRefValueY / (Double)uiSSDY ) : 99.99 );
2439  dUPSNR            = ( uiSSDU ? 10.0 * log10( fRefValueC / (Double)uiSSDU ) : 99.99 );
2440  dVPSNR            = ( uiSSDV ? 10.0 * log10( fRefValueC / (Double)uiSSDV ) : 99.99 );
2441#if H_3D_VSO
2442#if H_3D_VSO_SYNTH_DIST_OUT
2443}
2444#endif
2445#endif
2446  /* calculate the size of the access unit, excluding:
2447   *  - any AnnexB contributions (start_code_prefix, zero_byte, etc.,)
2448   *  - SEI NAL units
2449   */
2450  UInt numRBSPBytes = 0;
2451  for (AccessUnit::const_iterator it = accessUnit.begin(); it != accessUnit.end(); it++)
2452  {
2453    UInt numRBSPBytes_nal = UInt((*it)->m_nalUnitData.str().size());
2454#if VERBOSE_RATE
2455    printf("*** %6s numBytesInNALunit: %u\n", nalUnitTypeToString((*it)->m_nalUnitType), numRBSPBytes_nal);
2456#endif
2457    if ((*it)->m_nalUnitType != NAL_UNIT_PREFIX_SEI && (*it)->m_nalUnitType != NAL_UNIT_SUFFIX_SEI)
2458    {
2459      numRBSPBytes += numRBSPBytes_nal;
2460    }
2461  }
2462
2463  UInt uibits = numRBSPBytes * 8;
2464  m_vRVM_RP.push_back( uibits );
2465
2466  //===== add PSNR =====
2467#if H_MV
2468  m_pcEncTop->getAnalyzeAll()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2469#else
2470  m_gcAnalyzeAll.addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2471#endif
2472  TComSlice*  pcSlice = pcPic->getSlice(0);
2473  if (pcSlice->isIntra())
2474  {
2475#if H_MV
2476    m_pcEncTop->getAnalyzeI()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2477#else
2478    m_gcAnalyzeI.addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2479#endif
2480  }
2481  if (pcSlice->isInterP())
2482  {
2483#if H_MV
2484    m_pcEncTop->getAnalyzeP()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2485#else
2486    m_gcAnalyzeP.addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2487#endif
2488  }
2489  if (pcSlice->isInterB())
2490  {
2491#if H_MV
2492    m_pcEncTop->getAnalyzeB()->addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2493#else
2494    m_gcAnalyzeB.addResult (dYPSNR, dUPSNR, dVPSNR, (Double)uibits);
2495#endif
2496  }
2497
2498  Char c = (pcSlice->isIntra() ? 'I' : pcSlice->isInterP() ? 'P' : 'B');
2499  if (!pcSlice->isReferenced()) c += 32;
2500
2501#if ADAPTIVE_QP_SELECTION
2502#if H_MV
2503  printf("Layer %3d   POC %4d TId: %1d ( %c-SLICE, nQP %d QP %d ) %10d bits",
2504    pcSlice->getLayerId(),
2505    pcSlice->getPOC(),
2506    pcSlice->getTLayer(),
2507    c,
2508    pcSlice->getSliceQpBase(),
2509    pcSlice->getSliceQp(),
2510    uibits );
2511#else
2512  printf("POC %4d TId: %1d ( %c-SLICE, nQP %d QP %d ) %10d bits",
2513         pcSlice->getPOC(),
2514         pcSlice->getTLayer(),
2515         c,
2516         pcSlice->getSliceQpBase(),
2517         pcSlice->getSliceQp(),
2518         uibits );
2519#endif
2520#else
2521#if H_MV
2522  printf("Layer %3d   POC %4d TId: %1d ( %c-SLICE, QP %d ) %10d bits",
2523    pcSlice->getLayerId(),
2524    pcSlice->getPOC()-pcSlice->getLastIDR(),
2525    pcSlice->getTLayer(),
2526    c,
2527    pcSlice->getSliceQp(),
2528    uibits );
2529#else
2530  printf("POC %4d TId: %1d ( %c-SLICE, QP %d ) %10d bits",
2531         pcSlice->getPOC()-pcSlice->getLastIDR(),
2532         pcSlice->getTLayer(),
2533         c,
2534         pcSlice->getSliceQp(),
2535         uibits );
2536#endif
2537#endif
2538
2539  printf(" [Y %6.4lf dB    U %6.4lf dB    V %6.4lf dB]", dYPSNR, dUPSNR, dVPSNR );
2540  printf(" [ET %5.0f ]", dEncTime );
2541 
2542  for (Int iRefList = 0; iRefList < 2; iRefList++)
2543  {
2544    printf(" [L%d ", iRefList);
2545    for (Int iRefIndex = 0; iRefIndex < pcSlice->getNumRefIdx(RefPicList(iRefList)); iRefIndex++)
2546    {
2547#if H_MV
2548      if( pcSlice->getLayerId() != pcSlice->getRefLayerId( RefPicList(iRefList), iRefIndex ) )
2549      {
2550        printf( "V%d ", pcSlice->getRefLayerId( RefPicList(iRefList), iRefIndex ) );
2551      }
2552      else
2553      {
2554#endif
2555      printf ("%d ", pcSlice->getRefPOC(RefPicList(iRefList), iRefIndex)-pcSlice->getLastIDR());
2556#if H_MV
2557      }
2558#endif
2559    }
2560    printf("]");
2561  }
2562}
2563
2564/** Function for deciding the nal_unit_type.
2565 * \param pocCurr POC of the current picture
2566 * \returns the nal unit type of the picture
2567 * This function checks the configuration and returns the appropriate nal_unit_type for the picture.
2568 */
2569NalUnitType TEncGOP::getNalUnitType(Int pocCurr, Int lastIDR)
2570{
2571  if (pocCurr == 0)
2572  {
2573    return NAL_UNIT_CODED_SLICE_IDR_W_RADL;
2574  }
2575  if (pocCurr % m_pcCfg->getIntraPeriod() == 0)
2576  {
2577    if (m_pcCfg->getDecodingRefreshType() == 1)
2578    {
2579      return NAL_UNIT_CODED_SLICE_CRA;
2580    }
2581    else if (m_pcCfg->getDecodingRefreshType() == 2)
2582    {
2583      return NAL_UNIT_CODED_SLICE_IDR_W_RADL;
2584    }
2585  }
2586  if(m_pocCRA>0)
2587  {
2588    if(pocCurr<m_pocCRA)
2589    {
2590      // All leading pictures are being marked as TFD pictures here since current encoder uses all
2591      // reference pictures while encoding leading pictures. An encoder can ensure that a leading
2592      // picture can be still decodable when random accessing to a CRA/CRANT/BLA/BLANT picture by
2593      // controlling the reference pictures used for encoding that leading picture. Such a leading
2594      // picture need not be marked as a TFD picture.
2595      return NAL_UNIT_CODED_SLICE_RASL_R;
2596    }
2597  }
2598  if (lastIDR>0)
2599  {
2600    if (pocCurr < lastIDR)
2601    {
2602      return NAL_UNIT_CODED_SLICE_RADL_R;
2603    }
2604  }
2605  return NAL_UNIT_CODED_SLICE_TRAIL_R;
2606}
2607
2608Double TEncGOP::xCalculateRVM()
2609{
2610  Double dRVM = 0;
2611 
2612  if( m_pcCfg->getGOPSize() == 1 && m_pcCfg->getIntraPeriod() != 1 && m_pcCfg->getFramesToBeEncoded() > RVM_VCEGAM10_M * 2 )
2613  {
2614    // calculate RVM only for lowdelay configurations
2615    std::vector<Double> vRL , vB;
2616    size_t N = m_vRVM_RP.size();
2617    vRL.resize( N );
2618    vB.resize( N );
2619   
2620    Int i;
2621    Double dRavg = 0 , dBavg = 0;
2622    vB[RVM_VCEGAM10_M] = 0;
2623    for( i = RVM_VCEGAM10_M + 1 ; i < N - RVM_VCEGAM10_M + 1 ; i++ )
2624    {
2625      vRL[i] = 0;
2626      for( Int j = i - RVM_VCEGAM10_M ; j <= i + RVM_VCEGAM10_M - 1 ; j++ )
2627        vRL[i] += m_vRVM_RP[j];
2628      vRL[i] /= ( 2 * RVM_VCEGAM10_M );
2629      vB[i] = vB[i-1] + m_vRVM_RP[i] - vRL[i];
2630      dRavg += m_vRVM_RP[i];
2631      dBavg += vB[i];
2632    }
2633   
2634    dRavg /= ( N - 2 * RVM_VCEGAM10_M );
2635    dBavg /= ( N - 2 * RVM_VCEGAM10_M );
2636   
2637    Double dSigamB = 0;
2638    for( i = RVM_VCEGAM10_M + 1 ; i < N - RVM_VCEGAM10_M + 1 ; i++ )
2639    {
2640      Double tmp = vB[i] - dBavg;
2641      dSigamB += tmp * tmp;
2642    }
2643    dSigamB = sqrt( dSigamB / ( N - 2 * RVM_VCEGAM10_M ) );
2644   
2645    Double f = sqrt( 12.0 * ( RVM_VCEGAM10_M - 1 ) / ( RVM_VCEGAM10_M + 1 ) );
2646   
2647    dRVM = dSigamB / dRavg * f;
2648  }
2649 
2650  return( dRVM );
2651}
2652
2653/** Attaches the input bitstream to the stream in the output NAL unit
2654    Updates rNalu to contain concatenated bitstream. rpcBitstreamRedirect is cleared at the end of this function call.
2655 *  \param codedSliceData contains the coded slice data (bitstream) to be concatenated to rNalu
2656 *  \param rNalu          target NAL unit
2657 */
2658Void TEncGOP::xAttachSliceDataToNalUnit (OutputNALUnit& rNalu, TComOutputBitstream*& codedSliceData)
2659{
2660  // Byte-align
2661  rNalu.m_Bitstream.writeByteAlignment();   // Slice header byte-alignment
2662
2663  // Perform bitstream concatenation
2664  if (codedSliceData->getNumberOfWrittenBits() > 0)
2665    {
2666    rNalu.m_Bitstream.addSubstream(codedSliceData);
2667  }
2668
2669  m_pcEntropyCoder->setBitstream(&rNalu.m_Bitstream);
2670
2671  codedSliceData->clear();
2672}
2673
2674// Function will arrange the long-term pictures in the decreasing order of poc_lsb_lt,
2675// and among the pictures with the same lsb, it arranges them in increasing delta_poc_msb_cycle_lt value
2676Void TEncGOP::arrangeLongtermPicturesInRPS(TComSlice *pcSlice, TComList<TComPic*>& rcListPic)
2677{
2678  TComReferencePictureSet *rps = pcSlice->getRPS();
2679  if(!rps->getNumberOfLongtermPictures())
2680  {
2681    return;
2682  }
2683
2684  // Arrange long-term reference pictures in the correct order of LSB and MSB,
2685  // and assign values for pocLSBLT and MSB present flag
2686  Int longtermPicsPoc[MAX_NUM_REF_PICS], longtermPicsLSB[MAX_NUM_REF_PICS], indices[MAX_NUM_REF_PICS];
2687  Int longtermPicsMSB[MAX_NUM_REF_PICS];
2688  Bool mSBPresentFlag[MAX_NUM_REF_PICS];
2689  ::memset(longtermPicsPoc, 0, sizeof(longtermPicsPoc));    // Store POC values of LTRP
2690  ::memset(longtermPicsLSB, 0, sizeof(longtermPicsLSB));    // Store POC LSB values of LTRP
2691  ::memset(longtermPicsMSB, 0, sizeof(longtermPicsMSB));    // Store POC LSB values of LTRP
2692  ::memset(indices        , 0, sizeof(indices));            // Indices to aid in tracking sorted LTRPs
2693  ::memset(mSBPresentFlag , 0, sizeof(mSBPresentFlag));     // Indicate if MSB needs to be present
2694
2695  // Get the long-term reference pictures
2696  Int offset = rps->getNumberOfNegativePictures() + rps->getNumberOfPositivePictures();
2697  Int i, ctr = 0;
2698  Int maxPicOrderCntLSB = 1 << pcSlice->getSPS()->getBitsForPOC();
2699  for(i = rps->getNumberOfPictures() - 1; i >= offset; i--, ctr++)
2700  {
2701    longtermPicsPoc[ctr] = rps->getPOC(i);                                  // LTRP POC
2702    longtermPicsLSB[ctr] = getLSB(longtermPicsPoc[ctr], maxPicOrderCntLSB); // LTRP POC LSB
2703    indices[ctr]      = i; 
2704    longtermPicsMSB[ctr] = longtermPicsPoc[ctr] - longtermPicsLSB[ctr];
2705  }
2706  Int numLongPics = rps->getNumberOfLongtermPictures();
2707  assert(ctr == numLongPics);
2708
2709  // Arrange pictures in decreasing order of MSB;
2710  for(i = 0; i < numLongPics; i++)
2711  {
2712    for(Int j = 0; j < numLongPics - 1; j++)
2713    {
2714      if(longtermPicsMSB[j] < longtermPicsMSB[j+1])
2715      {
2716        std::swap(longtermPicsPoc[j], longtermPicsPoc[j+1]);
2717        std::swap(longtermPicsLSB[j], longtermPicsLSB[j+1]);
2718        std::swap(longtermPicsMSB[j], longtermPicsMSB[j+1]);
2719        std::swap(indices[j]        , indices[j+1]        );
2720      }
2721    }
2722  }
2723
2724  for(i = 0; i < numLongPics; i++)
2725  {
2726    // Check if MSB present flag should be enabled.
2727    // Check if the buffer contains any pictures that have the same LSB.
2728    TComList<TComPic*>::iterator  iterPic = rcListPic.begin(); 
2729    TComPic*                      pcPic;
2730    while ( iterPic != rcListPic.end() )
2731    {
2732      pcPic = *iterPic;
2733      if( (getLSB(pcPic->getPOC(), maxPicOrderCntLSB) == longtermPicsLSB[i])   &&     // Same LSB
2734                                      (pcPic->getSlice(0)->isReferenced())     &&    // Reference picture
2735                                        (pcPic->getPOC() != longtermPicsPoc[i])    )  // Not the LTRP itself
2736      {
2737        mSBPresentFlag[i] = true;
2738        break;
2739      }
2740      iterPic++;     
2741    }
2742  }
2743
2744  // tempArray for usedByCurr flag
2745  Bool tempArray[MAX_NUM_REF_PICS]; ::memset(tempArray, 0, sizeof(tempArray));
2746  for(i = 0; i < numLongPics; i++)
2747  {
2748    tempArray[i] = rps->getUsed(indices[i]);
2749  }
2750  // Now write the final values;
2751  ctr = 0;
2752  Int currMSB = 0, currLSB = 0;
2753  // currPicPoc = currMSB + currLSB
2754  currLSB = getLSB(pcSlice->getPOC(), maxPicOrderCntLSB); 
2755  currMSB = pcSlice->getPOC() - currLSB;
2756
2757  for(i = rps->getNumberOfPictures() - 1; i >= offset; i--, ctr++)
2758  {
2759    rps->setPOC                   (i, longtermPicsPoc[ctr]);
2760    rps->setDeltaPOC              (i, - pcSlice->getPOC() + longtermPicsPoc[ctr]);
2761    rps->setUsed                  (i, tempArray[ctr]);
2762    rps->setPocLSBLT              (i, longtermPicsLSB[ctr]);
2763    rps->setDeltaPocMSBCycleLT    (i, (currMSB - (longtermPicsPoc[ctr] - longtermPicsLSB[ctr])) / maxPicOrderCntLSB);
2764    rps->setDeltaPocMSBPresentFlag(i, mSBPresentFlag[ctr]);     
2765
2766    assert(rps->getDeltaPocMSBCycleLT(i) >= 0);   // Non-negative value
2767  }
2768  for(i = rps->getNumberOfPictures() - 1, ctr = 1; i >= offset; i--, ctr++)
2769  {
2770    for(Int j = rps->getNumberOfPictures() - 1 - ctr; j >= offset; j--)
2771    {
2772      // Here at the encoder we know that we have set the full POC value for the LTRPs, hence we
2773      // don't have to check the MSB present flag values for this constraint.
2774      assert( rps->getPOC(i) != rps->getPOC(j) ); // If assert fails, LTRP entry repeated in RPS!!!
2775    }
2776  }
2777}
2778
2779#if L0045_NON_NESTED_SEI_RESTRICTIONS
2780/** Function for finding the position to insert the first of APS and non-nested BP, PT, DU info SEI messages.
2781 * \param accessUnit Access Unit of the current picture
2782 * This function finds the position to insert the first of APS and non-nested BP, PT, DU info SEI messages.
2783 */
2784Int TEncGOP::xGetFirstSeiLocation(AccessUnit &accessUnit)
2785{
2786  // Find the location of the first SEI message
2787  AccessUnit::iterator it;
2788  Int seiStartPos = 0;
2789  for(it = accessUnit.begin(); it != accessUnit.end(); it++, seiStartPos++)
2790  {
2791     if ((*it)->isSei() || (*it)->isVcl())
2792     {
2793       break;
2794     }               
2795  }
2796//  assert(it != accessUnit.end());  // Triggers with some legit configurations
2797  return seiStartPos;
2798}
2799#endif
2800
2801#if L0386_DB_METRIC
2802Void TEncGOP::dblMetric( TComPic* pcPic, UInt uiNumSlices )
2803{
2804  TComPicYuv* pcPicYuvRec = pcPic->getPicYuvRec();
2805  Pel* Rec    = pcPicYuvRec->getLumaAddr( 0 );
2806  Pel* tempRec = Rec;
2807  Int  stride = pcPicYuvRec->getStride();
2808  UInt log2maxTB = pcPic->getSlice(0)->getSPS()->getQuadtreeTULog2MaxSize();
2809  UInt maxTBsize = (1<<log2maxTB);
2810  const UInt minBlockArtSize = 8;
2811  const UInt picWidth = pcPicYuvRec->getWidth();
2812  const UInt picHeight = pcPicYuvRec->getHeight();
2813  const UInt noCol = (picWidth>>log2maxTB);
2814  const UInt noRows = (picHeight>>log2maxTB);
2815  UInt64 *colSAD = (UInt64*)malloc(noCol*sizeof(UInt64));
2816  UInt64 *rowSAD = (UInt64*)malloc(noRows*sizeof(UInt64));
2817  UInt colIdx = 0;
2818  UInt rowIdx = 0;
2819  Pel p0, p1, p2, q0, q1, q2;
2820 
2821  Int qp = pcPic->getSlice(0)->getSliceQp();
2822  Int bitdepthScale = 1 << (g_bitDepthY-8);
2823  Int beta = TComLoopFilter::getBeta( qp ) * bitdepthScale;
2824  const Int thr2 = (beta>>2);
2825  const Int thr1 = 2*bitdepthScale;
2826  UInt a = 0;
2827 
2828  memset(colSAD, 0, noCol*sizeof(UInt64));
2829  memset(rowSAD, 0, noRows*sizeof(UInt64));
2830 
2831  if (maxTBsize > minBlockArtSize)
2832  {
2833    // Analyze vertical artifact edges
2834    for(Int c = maxTBsize; c < picWidth; c += maxTBsize)
2835    {
2836      for(Int r = 0; r < picHeight; r++)
2837      {
2838        p2 = Rec[c-3];
2839        p1 = Rec[c-2];
2840        p0 = Rec[c-1];
2841        q0 = Rec[c];
2842        q1 = Rec[c+1];
2843        q2 = Rec[c+2];
2844        a = ((abs(p2-(p1<<1)+p0)+abs(q0-(q1<<1)+q2))<<1);
2845        if ( thr1 < a && a < thr2)
2846        {
2847          colSAD[colIdx] += abs(p0 - q0);
2848        }
2849        Rec += stride;
2850      }
2851      colIdx++;
2852      Rec = tempRec;
2853    }
2854   
2855    // Analyze horizontal artifact edges
2856    for(Int r = maxTBsize; r < picHeight; r += maxTBsize)
2857    {
2858      for(Int c = 0; c < picWidth; c++)
2859      {
2860        p2 = Rec[c + (r-3)*stride];
2861        p1 = Rec[c + (r-2)*stride];
2862        p0 = Rec[c + (r-1)*stride];
2863        q0 = Rec[c + r*stride];
2864        q1 = Rec[c + (r+1)*stride];
2865        q2 = Rec[c + (r+2)*stride];
2866        a = ((abs(p2-(p1<<1)+p0)+abs(q0-(q1<<1)+q2))<<1);
2867        if (thr1 < a && a < thr2)
2868        {
2869          rowSAD[rowIdx] += abs(p0 - q0);
2870        }
2871      }
2872      rowIdx++;
2873    }
2874  }
2875 
2876  UInt64 colSADsum = 0;
2877  UInt64 rowSADsum = 0;
2878  for(Int c = 0; c < noCol-1; c++)
2879  {
2880    colSADsum += colSAD[c];
2881  }
2882  for(Int r = 0; r < noRows-1; r++)
2883  {
2884    rowSADsum += rowSAD[r];
2885  }
2886 
2887  colSADsum <<= 10;
2888  rowSADsum <<= 10;
2889  colSADsum /= (noCol-1);
2890  colSADsum /= picHeight;
2891  rowSADsum /= (noRows-1);
2892  rowSADsum /= picWidth;
2893 
2894  UInt64 avgSAD = ((colSADsum + rowSADsum)>>1);
2895  avgSAD >>= (g_bitDepthY-8);
2896 
2897  if ( avgSAD > 2048 )
2898  {
2899    avgSAD >>= 9;
2900    Int offset = Clip3(2,6,(Int)avgSAD);
2901    for (Int i=0; i<uiNumSlices; i++)
2902    {
2903      pcPic->getSlice(i)->setDeblockingFilterOverrideFlag(true);
2904      pcPic->getSlice(i)->setDeblockingFilterDisable(false);
2905      pcPic->getSlice(i)->setDeblockingFilterBetaOffsetDiv2( offset );
2906      pcPic->getSlice(i)->setDeblockingFilterTcOffsetDiv2( offset );
2907    }
2908  }
2909  else
2910  {
2911    for (Int i=0; i<uiNumSlices; i++)
2912    {
2913      pcPic->getSlice(i)->setDeblockingFilterOverrideFlag(false);
2914      pcPic->getSlice(i)->setDeblockingFilterDisable(        pcPic->getSlice(i)->getPPS()->getPicDisableDeblockingFilterFlag() );
2915      pcPic->getSlice(i)->setDeblockingFilterBetaOffsetDiv2( pcPic->getSlice(i)->getPPS()->getDeblockingFilterBetaOffsetDiv2() );
2916      pcPic->getSlice(i)->setDeblockingFilterTcOffsetDiv2(   pcPic->getSlice(i)->getPPS()->getDeblockingFilterTcOffsetDiv2()   );
2917    }
2918  }
2919 
2920  free(colSAD);
2921  free(rowSAD);
2922}
2923#endif
2924#if H_MV
2925Void TEncGOP::xSetRefPicListModificationsMvc( TComSlice* pcSlice, UInt uiPOCCurr, UInt iGOPid )
2926{ 
2927  TComVPS* vps = pcSlice->getVPS(); 
2928  Int layer    = pcSlice->getLayerIdInVps( ); 
2929 
2930  if( pcSlice->getSliceType() == I_SLICE || !(pcSlice->getPPS()->getListsModificationPresentFlag()) || vps->getNumDirectRefLayers( layer ) == 0 )
2931  {
2932    return;
2933  }
2934
2935  // analyze inter-view modifications
2936  GOPEntry ge = m_pcCfg->getGOPEntry( (pcSlice->getRapPicFlag() && ( layer > 0) ) ? MAX_GOP : iGOPid );
2937
2938  TComRefPicListModification* refPicListModification = pcSlice->getRefPicListModification();
2939 
2940  Int maxRefListSize  = pcSlice->getNumRpsCurrTempList();
2941  Int numTemporalRefs = maxRefListSize - vps->getNumDirectRefLayers( layer );
2942
2943
2944  for (Int li = 0; li < 2; li ++) // Loop over lists L0 and L1
2945  {
2946    Int numModifications = 0;
2947   
2948    for( Int k = 0; k < ge.m_numInterViewRefPics; k++ ) 
2949    {
2950      numModifications +=  ( ge.m_interViewRefPosL[li][k] >= 0 ) ? 1 : 0; 
2951    }
2952
2953    // set inter-view modifications
2954    Bool isModified = false;
2955      Int tempList[16];
2956      for( Int k = 0; k < 16; k++ ) { tempList[k] = -1; }
2957
2958    if( (maxRefListSize > 1) && (numModifications > 0) )
2959    {
2960      for( Int k = 0; k < ge.m_numInterViewRefPics; k++ )
2961      {
2962        if( ge.m_interViewRefPosL[li][k] >= 0 )
2963        {
2964          Int orgIdx    = numTemporalRefs;
2965          Int targetIdx = ge.m_interViewRefPosL[ li ][ k ];
2966          for( Int idx = 0; idx < vps->getNumDirectRefLayers( layer ); idx++ )
2967          {           
2968            Int refLayer  = vps->getLayerIdInVps( vps->getRefLayerId( layer, idx ) );         
2969            if( ( layer + ge.m_interViewRefs[ k ]) == refLayer )
2970            {
2971              tempList[ targetIdx ] = orgIdx;             
2972              isModified = ( targetIdx != orgIdx  );
2973            }
2974            orgIdx++;
2975          }
2976        }
2977      }
2978    }
2979
2980    refPicListModification->setRefPicListModificationFlagL( li, isModified ); 
2981
2982      if( isModified )
2983      {
2984        Int temporalRefIdx = 0;
2985        for( Int i = 0; i < pcSlice->getNumRefIdx( ( li == 0 ) ? REF_PIC_LIST_0 : REF_PIC_LIST_1 ); i++ )
2986        {
2987          if( tempList[i] >= 0 ) 
2988          {
2989            refPicListModification->setRefPicSetIdxL( li, i, tempList[i] );
2990          }
2991          else
2992          {
2993            refPicListModification->setRefPicSetIdxL( li, i, temporalRefIdx );
2994            temporalRefIdx++;
2995          }
2996        }
2997      }
2998  }
2999}
3000#endif
3001//! \}
Note: See TracBrowser for help on using the repository browser.