source: 3DVCSoftware/branches/HTM-DEV-0.1-dev/source/Lib/TLibEncoder/TEncGOP.cpp @ 364

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

Further cleanups

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