source: 3DVCSoftware/branches/HTM-DEV-0.3-dev0/source/Lib/TLibEncoder/TEncGOP.cpp

Last change on this file was 500, checked in by tech, 12 years ago

Further fixes.

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