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

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

Bug fixes related to not correctly initialized slice at encoder.

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