source: 3DVCSoftware/branches/HTM-4.0-Orange/source/Lib/TLibEncoder/TEncSlice.cpp @ 111

Last change on this file since 111 was 110, checked in by lg, 12 years ago

LGE_WVSO_A0119 integration (non-CTC)
  • Property svn:eol-style set to native
File size: 65.5 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license. 
5 *
6 * Copyright (c) 2010-2012, 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     TEncSlice.cpp
35    \brief    slice encoder class
36*/
37
38#include "TEncTop.h"
39#include "TEncSlice.h"
40#if HHI_VSO_SPEEDUP_A0033
41#include "../../App/TAppEncoder/TAppEncTop.h"
42#endif
43#include <math.h>
44
45//! \ingroup TLibEncoder
46//! \{
47
48// ====================================================================================================================
49// Constructor / destructor / create / destroy
50// ====================================================================================================================
51
52TEncSlice::TEncSlice()
53{
54  m_apcPicYuvPred = NULL;
55  m_apcPicYuvResi = NULL;
56 
57  m_pdRdPicLambda = NULL;
58  m_pdRdPicQp     = NULL;
59  m_piRdPicQp     = NULL;
60  m_pcBufferSbacCoders    = NULL;
61  m_pcBufferBinCoderCABACs  = NULL;
62  m_pcBufferLowLatSbacCoders    = NULL;
63  m_pcBufferLowLatBinCoderCABACs  = NULL;
64}
65
66TEncSlice::~TEncSlice()
67{
68}
69
70Void TEncSlice::create( Int iWidth, Int iHeight, UInt iMaxCUWidth, UInt iMaxCUHeight, UChar uhTotalDepth )
71{
72  // create prediction picture
73  if ( m_apcPicYuvPred == NULL )
74  {
75    m_apcPicYuvPred  = new TComPicYuv;
76    m_apcPicYuvPred->create( iWidth, iHeight, iMaxCUWidth, iMaxCUHeight, uhTotalDepth );
77  }
78 
79  // create residual picture
80  if( m_apcPicYuvResi == NULL )
81  {
82    m_apcPicYuvResi  = new TComPicYuv;
83    m_apcPicYuvResi->create( iWidth, iHeight, iMaxCUWidth, iMaxCUHeight, uhTotalDepth );
84  }
85}
86
87Void TEncSlice::destroy()
88{
89  // destroy prediction picture
90  if ( m_apcPicYuvPred )
91  {
92    m_apcPicYuvPred->destroy();
93    delete m_apcPicYuvPred;
94    m_apcPicYuvPred  = NULL;
95  }
96 
97  // destroy residual picture
98  if ( m_apcPicYuvResi )
99  {
100    m_apcPicYuvResi->destroy();
101    delete m_apcPicYuvResi;
102    m_apcPicYuvResi  = NULL;
103  }
104 
105  // free lambda and QP arrays
106  if ( m_pdRdPicLambda ) { xFree( m_pdRdPicLambda ); m_pdRdPicLambda = NULL; }
107  if ( m_pdRdPicQp     ) { xFree( m_pdRdPicQp     ); m_pdRdPicQp     = NULL; }
108  if ( m_piRdPicQp     ) { xFree( m_piRdPicQp     ); m_piRdPicQp     = NULL; }
109
110  if ( m_pcBufferSbacCoders )
111  {
112    delete[] m_pcBufferSbacCoders;
113  }
114  if ( m_pcBufferBinCoderCABACs )
115  {
116    delete[] m_pcBufferBinCoderCABACs;
117  }
118  if ( m_pcBufferLowLatSbacCoders )
119    delete[] m_pcBufferLowLatSbacCoders;
120  if ( m_pcBufferLowLatBinCoderCABACs )
121    delete[] m_pcBufferLowLatBinCoderCABACs;
122}
123
124Void TEncSlice::init( TEncTop* pcEncTop )
125{
126  m_pcCfg             = pcEncTop;
127  m_pcListPic         = pcEncTop->getListPic();
128 
129  m_pcGOPEncoder      = pcEncTop->getGOPEncoder();
130  m_pcCuEncoder       = pcEncTop->getCuEncoder();
131  m_pcPredSearch      = pcEncTop->getPredSearch();
132 
133  m_pcEntropyCoder    = pcEncTop->getEntropyCoder();
134  m_pcCavlcCoder      = pcEncTop->getCavlcCoder();
135  m_pcSbacCoder       = pcEncTop->getSbacCoder();
136  m_pcBinCABAC        = pcEncTop->getBinCABAC();
137  m_pcTrQuant         = pcEncTop->getTrQuant();
138 
139  m_pcBitCounter      = pcEncTop->getBitCounter();
140  m_pcRdCost          = pcEncTop->getRdCost();
141  m_pppcRDSbacCoder   = pcEncTop->getRDSbacCoder();
142  m_pcRDGoOnSbacCoder = pcEncTop->getRDGoOnSbacCoder();
143 
144  // create lambda and QP arrays
145  m_pdRdPicLambda     = (Double*)xMalloc( Double, m_pcCfg->getDeltaQpRD() * 2 + 1 );
146  m_pdRdPicQp         = (Double*)xMalloc( Double, m_pcCfg->getDeltaQpRD() * 2 + 1 );
147  m_piRdPicQp         = (Int*   )xMalloc( Int,    m_pcCfg->getDeltaQpRD() * 2 + 1 );
148}
149
150/**
151 - non-referenced frame marking
152 - QP computation based on temporal structure
153 - lambda computation based on QP
154 - set temporal layer ID and the parameter sets
155 .
156 \param pcPic         picture class
157 \param iPOCLast      POC of last picture
158 \param uiPOCCurr     current POC
159 \param iNumPicRcvd   number of received pictures
160 \param iTimeOffset   POC offset for hierarchical structure
161 \param iDepth        temporal layer depth
162 \param rpcSlice      slice header class
163 \param pSPS          SPS associated with the slice
164 \param pPPS          PPS associated with the slice
165 */
166#if VIDYO_VPS_INTEGRATION
167Void TEncSlice::initEncSlice( TComPic* pcPic, Int iPOCLast, UInt uiPOCCurr, Int iNumPicRcvd, Int iGOPid, TComSlice*& rpcSlice, TComVPS * pVPS, TComSPS* pSPS, TComPPS *pPPS )
168#else
169Void TEncSlice::initEncSlice( TComPic* pcPic, Int iPOCLast, UInt uiPOCCurr, Int iNumPicRcvd, Int iGOPid, TComSlice*& rpcSlice, TComSPS* pSPS, TComPPS *pPPS )
170#endif
171{
172  Double dQP;
173  Double dLambda;
174 
175  rpcSlice = pcPic->getSlice(0);
176#if VIDYO_VPS_INTEGRATION
177  rpcSlice->setVPS( pVPS );
178#endif
179  rpcSlice->setSPS( pSPS );
180  rpcSlice->setPPS( pPPS );
181  rpcSlice->setSliceBits(0);
182  rpcSlice->setPic( pcPic );
183  rpcSlice->initSlice();
184  rpcSlice->initTiles();
185#if H0388
186  rpcSlice->setPicOutputFlag( true );
187#endif
188  rpcSlice->setPOC( uiPOCCurr );
189 
190#if SONY_COLPIC_AVAILABILITY
191  rpcSlice->setViewOrderIdx(m_pcCfg->getViewOrderIdx());
192#endif
193
194  // set mutliview parameters
195  rpcSlice->initMultiviewSlice( pcPic->getCodedScale(), pcPic->getCodedOffset() );
196
197  // depth computation based on GOP size
198  int iDepth;
199  {
200    Int i, j;
201    Int iPOC = rpcSlice->getPOC()%m_pcCfg->getGOPSize();
202    if ( iPOC == 0 )
203    {
204      iDepth = 0;
205    }
206    else
207    {
208      Int iStep = m_pcCfg->getGOPSize();
209      iDepth    = 0;
210      for( i=iStep>>1; i>=1; i>>=1 )
211      {
212        for ( j=i; j<m_pcCfg->getGOPSize(); j+=iStep )
213        {
214          if ( j == iPOC )
215          {
216            i=0;
217            break;
218          }
219        }
220        iStep>>=1;
221        iDepth++;
222      }
223    }
224  }
225 
226  // slice type
227  SliceType eSliceTypeBaseView;
228  if( iPOCLast == 0 || uiPOCCurr % m_pcCfg->getIntraPeriod() == 0 || m_pcGOPEncoder->getGOPSize() == 0 )
229  {
230    eSliceTypeBaseView = I_SLICE;
231  }
232  else
233  {
234    eSliceTypeBaseView = B_SLICE;
235  }
236  SliceType eSliceType = eSliceTypeBaseView;
237  if( eSliceTypeBaseView == I_SLICE && m_pcCfg->getGOPEntry(MAX_GOP).m_POC == 0 && m_pcCfg->getGOPEntry(MAX_GOP).m_sliceType != 'I' )
238  {
239    eSliceType = B_SLICE; 
240  }
241  rpcSlice->setSliceType( eSliceType );
242 
243  // ------------------------------------------------------------------------------------------------------------------
244  // Non-referenced frame marking
245  // ------------------------------------------------------------------------------------------------------------------
246  rpcSlice->setReferenced( m_pcCfg->getGOPEntry(iGOPid).m_refPic );
247  if( eSliceTypeBaseView == I_SLICE )
248  {
249    rpcSlice->setReferenced(true);
250  }
251 
252  // ------------------------------------------------------------------------------------------------------------------
253  // QP setting
254  // ------------------------------------------------------------------------------------------------------------------
255 
256  dQP = m_pcCfg->getQP();
257  if( eSliceType != I_SLICE )
258  {
259#if LOSSLESS_CODING
260#if H0736_AVC_STYLE_QP_RANGE
261    if (!(( m_pcCfg->getMaxDeltaQP() == 0 ) && (dQP == -rpcSlice->getSPS()->getQpBDOffsetY() ) && (rpcSlice->getSPS()->getUseLossless()))) 
262#else
263    if (!(( m_pcCfg->getMaxDeltaQP() == 0 ) && (dQP == 0 ) && (rpcSlice->getSPS()->getUseLossless())))
264#endif
265#endif
266    {
267    dQP += m_pcCfg->getGOPEntry( (eSliceTypeBaseView == I_SLICE) ? MAX_GOP : iGOPid ).m_QPOffset;
268    }
269  }
270 
271  // modify QP
272  Int* pdQPs = m_pcCfg->getdQPs();
273  if ( pdQPs )
274  {
275    dQP += pdQPs[ rpcSlice->getPOC() ];
276  }
277 
278  // ------------------------------------------------------------------------------------------------------------------
279  // Lambda computation
280  // ------------------------------------------------------------------------------------------------------------------
281 
282  Int iQP;
283  Double dOrigQP = dQP;
284
285  // pre-compute lambda and QP values for all possible QP candidates
286  for ( Int iDQpIdx = 0; iDQpIdx < 2 * m_pcCfg->getDeltaQpRD() + 1; iDQpIdx++ )
287  {
288    // compute QP value
289    dQP = dOrigQP + ((iDQpIdx+1)>>1)*(iDQpIdx%2 ? -1 : 1);
290   
291    // compute lambda value
292    Int    NumberBFrames = ( m_pcCfg->getGOPSize() - 1 );
293    Int    SHIFT_QP = 12;
294    Double dLambda_scale = 1.0 - Clip3( 0.0, 0.5, 0.05*(Double)NumberBFrames );
295#if FULL_NBIT
296    Int    bitdepth_luma_qp_scale = 6 * (g_uiBitDepth - 8);
297#else
298    Int    bitdepth_luma_qp_scale = 0;
299#endif
300    Double qp_temp = (Double) dQP + bitdepth_luma_qp_scale - SHIFT_QP;
301#if FULL_NBIT
302    Double qp_temp_orig = (Double) dQP - SHIFT_QP;
303#endif
304    // Case #1: I or P-slices (key-frame)
305    Double dQPFactor;
306    if( eSliceType != I_SLICE ) 
307    {
308      dQPFactor = m_pcCfg->getGOPEntry( (eSliceTypeBaseView == I_SLICE) ? MAX_GOP : iGOPid ).m_QPFactor;
309    }
310    else
311    {
312      dQPFactor = 0.57 * dLambda_scale;
313    }
314
315    dLambda = dQPFactor*pow( 2.0, qp_temp/3.0 );
316
317    if ( iDepth>0 )
318    {
319#if FULL_NBIT
320        dLambda *= Clip3( 2.00, 4.00, (qp_temp_orig / 6.0) ); // (j == B_SLICE && p_cur_frm->layer != 0 )
321#else
322        dLambda *= Clip3( 2.00, 4.00, (qp_temp / 6.0) ); // (j == B_SLICE && p_cur_frm->layer != 0 )
323#endif
324    }
325   
326    // if hadamard is used in ME process
327    if ( !m_pcCfg->getUseHADME() )
328    {
329      dLambda *= 0.95;
330    }
331   
332#if H0736_AVC_STYLE_QP_RANGE
333    iQP = max( -pSPS->getQpBDOffsetY(), min( MAX_QP, (Int) floor( dQP + 0.5 ) ) );
334#else
335    iQP = max( MIN_QP, min( MAX_QP, (Int)floor( dQP + 0.5 ) ) );
336#endif
337
338    m_pdRdPicLambda[iDQpIdx] = dLambda;
339    m_pdRdPicQp    [iDQpIdx] = dQP;
340    m_piRdPicQp    [iDQpIdx] = iQP;
341  }
342 
343  // obtain dQP = 0 case
344  dLambda = m_pdRdPicLambda[0];
345  dQP     = m_pdRdPicQp    [0];
346  iQP     = m_piRdPicQp    [0];
347 
348  if( rpcSlice->getSliceType( ) != I_SLICE )
349  {
350    dLambda *= m_pcCfg->getLambdaModifier( iDepth );
351  }
352
353  // store lambda
354  m_pcRdCost ->setLambda( dLambda );
355#if WEIGHTED_CHROMA_DISTORTION
356// for RDO
357  // in RdCost there is only one lambda because the luma and chroma bits are not separated, instead we weight the distortion of chroma.
358#if H0736_AVC_STYLE_QP_RANGE
359  Double weight = 1.0;
360  if(iQP >= 0)
361  {
362    weight = pow( 2.0, (iQP-g_aucChromaScale[iQP])/3.0 );  // takes into account of the chroma qp mapping without chroma qp Offset
363  }
364#else
365  Double weight = pow( 2.0, (iQP-g_aucChromaScale[iQP])/3.0 );  // takes into account of the chroma qp mapping without chroma qp Offset
366#endif
367  m_pcRdCost ->setChromaDistortionWeight( weight );     
368#endif
369
370#if HHI_VSO
371  m_pcRdCost->setUseLambdaScaleVSO  ( (m_pcCfg->getUseVSO() ||  m_pcCfg->getForceLambdaScaleVSO()) && m_pcCfg->isDepthCoder()  );
372  m_pcRdCost->setLambdaVSO( dLambda * m_pcCfg->getLambdaScaleVSO() );
373#endif
374
375#if SAIT_VSO_EST_A0033
376  m_pcRdCost->setDisparityCoeff( m_pcCfg->getDispCoeff() );
377#endif
378#if LGE_WVSO_A0119
379  if( m_pcCfg->getWVSO() && m_pcCfg->isDepthCoder() )
380  {
381
382    Int iDWeight, iVSOWeight, iVSDWeight;
383    iDWeight = m_pcCfg->getDWeight();
384    iVSOWeight = m_pcCfg->getVSOWeight();
385    iVSDWeight = m_pcCfg->getVSDWeight();
386    //    if( rpcSlice->getViewId() )  iDWeight = ( iDWeight >> 1 );
387    //    if( rpcSlice->getPOC() % 8 != 0 )  iDWeight = ( iDWeight >> 1 );
388    //    if( rpcSlice->getPOC() % 4 != 0 )  iDWeight = ( iDWeight >> 1 );
389    //    if( rpcSlice->getPOC() % 2 != 0 )  iDWeight = ( iDWeight >> 1 );
390
391    m_pcRdCost->setDWeight( iDWeight );
392    m_pcRdCost->setVSOWeight( iVSOWeight );
393    m_pcRdCost->setVSDWeight( iVSDWeight );
394
395  }
396#endif
397#if RDOQ_CHROMA_LAMBDA
398// for RDOQ
399  m_pcTrQuant->setLambda( dLambda, dLambda / weight );   
400#else
401  m_pcTrQuant->setLambda( dLambda );
402#endif
403
404#if ALF_CHROMA_LAMBDA || SAO_CHROMA_LAMBDA
405// For ALF or SAO
406  rpcSlice   ->setLambda( dLambda, dLambda / weight ); 
407#else
408  rpcSlice   ->setLambda( dLambda );
409#endif
410 
411#if HHI_INTER_VIEW_MOTION_PRED
412  m_pcRdCost ->setLambdaMVReg ( dLambda * m_pcCfg->getMultiviewMvRegLambdaScale() );
413#endif
414 
415#if HB_LAMBDA_FOR_LDC
416  // restore original slice type
417  eSliceType = eSliceTypeBaseView;
418  if( eSliceTypeBaseView == I_SLICE && m_pcCfg->getGOPEntry(MAX_GOP).m_POC == 0 && m_pcCfg->getGOPEntry(MAX_GOP).m_sliceType != 'I' )
419  {
420    eSliceType = B_SLICE;
421  }
422  rpcSlice->setSliceType( eSliceType );
423#endif
424 
425  rpcSlice->setSliceQp          ( iQP );
426#if ADAPTIVE_QP_SELECTION
427  rpcSlice->setSliceQpBase      ( iQP );
428#endif
429  rpcSlice->setSliceQpDelta     ( 0 );
430  rpcSlice->setNumRefIdx(REF_PIC_LIST_0,m_pcCfg->getGOPEntry( (eSliceTypeBaseView == I_SLICE) ? MAX_GOP : iGOPid ).m_numRefPicsActive);
431  rpcSlice->setNumRefIdx(REF_PIC_LIST_1,m_pcCfg->getGOPEntry( (eSliceTypeBaseView == I_SLICE) ? MAX_GOP : iGOPid ).m_numRefPicsActive);
432 
433  rpcSlice->setLoopFilterOffsetInAPS( m_pcCfg->getLoopFilterOffsetInAPS() );
434#if DBL_CONTROL
435  if (rpcSlice->getPPS()->getDeblockingFilterControlPresent())
436  {
437#endif
438    rpcSlice->setInheritDblParamFromAPS( m_pcCfg->getLoopFilterOffsetInAPS() ? 1 : 0 );
439    rpcSlice->setLoopFilterDisable( m_pcCfg->getLoopFilterDisable() );
440    if ( !rpcSlice->getLoopFilterDisable())
441    {
442      rpcSlice->setLoopFilterBetaOffset( m_pcCfg->getLoopFilterBetaOffset() );
443      rpcSlice->setLoopFilterTcOffset( m_pcCfg->getLoopFilterTcOffset() );
444    }
445#if DBL_CONTROL
446  }
447#endif
448
449  rpcSlice->setDepth            ( iDepth );
450 
451  pcPic->setTLayer( m_pcCfg->getGOPEntry( (eSliceTypeBaseView == I_SLICE) ? MAX_GOP : iGOPid ).m_temporalId );
452  if( eSliceType == I_SLICE )
453  {
454    pcPic->setTLayer(0);
455  }
456  rpcSlice->setTLayer( pcPic->getTLayer() );
457#if !H0566_TLA
458  rpcSlice->setTLayerSwitchingFlag( pPPS->getTLayerSwitchingFlag( pcPic->getTLayer() ) );
459#endif
460
461  assert( m_apcPicYuvPred );
462  assert( m_apcPicYuvResi );
463 
464  pcPic->setPicYuvPred( m_apcPicYuvPred );
465  pcPic->setPicYuvResi( m_apcPicYuvResi );
466  rpcSlice->setSliceMode            ( m_pcCfg->getSliceMode()            );
467  rpcSlice->setSliceArgument        ( m_pcCfg->getSliceArgument()        );
468  rpcSlice->setEntropySliceMode     ( m_pcCfg->getEntropySliceMode()     );
469  rpcSlice->setEntropySliceArgument ( m_pcCfg->getEntropySliceArgument() );
470
471#if ( HHI_MPI || HHI_INTER_VIEW_MOTION_PRED )
472  #if ( HHI_MPI && HHI_INTER_VIEW_MOTION_PRED )
473  const int iExtraMergeCandidates = ( pSPS->getUseMVI() || pSPS->getMultiviewMvPredMode() ) ? 1 : 0;
474  #elif HHI_MPI
475  const int iExtraMergeCandidates = pSPS->getUseMVI() ? 1 : 0;
476  #else
477  const int iExtraMergeCandidates = pSPS->getMultiviewMvPredMode() ? 1 : 0;
478  #endif
479  rpcSlice->setMaxNumMergeCand      (MRG_MAX_NUM_CANDS_SIGNALED+iExtraMergeCandidates);
480#else
481  rpcSlice->setMaxNumMergeCand      (MRG_MAX_NUM_CANDS_SIGNALED);
482#endif
483  xStoreWPparam( pPPS->getUseWP(), pPPS->getWPBiPredIdc() );
484}
485
486// ====================================================================================================================
487// Public member functions
488// ====================================================================================================================
489
490Void TEncSlice::setSearchRange( TComSlice* pcSlice )
491{
492  Int iCurrPOC = pcSlice->getPOC();
493  Int iRefPOC;
494  Int iGOPSize = m_pcCfg->getGOPSize();
495  Int iOffset = (iGOPSize >> 1);
496  Int iMaxSR = m_pcCfg->getSearchRange();
497  Int iNumPredDir = pcSlice->isInterP() ? 1 : 2;
498 
499  for (Int iDir = 0; iDir <= iNumPredDir; iDir++)
500  {
501    RefPicList e = (RefPicList)iDir;
502    for (Int iRefIdx = 0; iRefIdx < pcSlice->getNumRefIdx(e); iRefIdx++)
503    {
504      iRefPOC = pcSlice->getRefPic(e, iRefIdx)->getPOC();
505      Int iNewSR = Clip3(8, iMaxSR, (iMaxSR*ADAPT_SR_SCALE*abs(iCurrPOC - iRefPOC)+iOffset)/iGOPSize);
506      m_pcPredSearch->setAdaptiveSearchRange(iDir, iRefIdx, iNewSR);
507    }
508  }
509}
510
511/**
512 - multi-loop slice encoding for different slice QP
513 .
514 \param rpcPic    picture class
515 */
516Void TEncSlice::precompressSlice( TComPic*& rpcPic )
517{
518  // if deltaQP RD is not used, simply return
519  if ( m_pcCfg->getDeltaQpRD() == 0 )
520  {
521    return;
522  }
523 
524  TComSlice* pcSlice        = rpcPic->getSlice(getSliceIdx());
525  Double     dPicRdCostBest = MAX_DOUBLE;
526  UInt       uiQpIdxBest = 0;
527 
528  Double dFrameLambda;
529#if FULL_NBIT
530  Int    SHIFT_QP = 12 + 6 * (g_uiBitDepth - 8);
531#else
532  Int    SHIFT_QP = 12;
533#endif
534 
535  // set frame lambda
536  if (m_pcCfg->getGOPSize() > 1)
537  {
538    dFrameLambda = 0.68 * pow (2, (m_piRdPicQp[0]  - SHIFT_QP) / 3.0) * (pcSlice->isInterB()? 2 : 1);
539  }
540  else
541  {
542    dFrameLambda = 0.68 * pow (2, (m_piRdPicQp[0] - SHIFT_QP) / 3.0);
543  }
544  m_pcRdCost      ->setFrameLambda(dFrameLambda);
545 
546  // for each QP candidate
547  for ( UInt uiQpIdx = 0; uiQpIdx < 2 * m_pcCfg->getDeltaQpRD() + 1; uiQpIdx++ )
548  {
549    pcSlice       ->setSliceQp             ( m_piRdPicQp    [uiQpIdx] );
550#if ADAPTIVE_QP_SELECTION
551    pcSlice       ->setSliceQpBase         ( m_piRdPicQp    [uiQpIdx] );
552#endif
553    m_pcRdCost    ->setLambda              ( m_pdRdPicLambda[uiQpIdx] );
554#if WEIGHTED_CHROMA_DISTORTION
555    // for RDO
556    // in RdCost there is only one lambda because the luma and chroma bits are not separated, instead we weight the distortion of chroma.
557    int iQP = m_piRdPicQp    [uiQpIdx];
558#if H0736_AVC_STYLE_QP_RANGE
559    Double weight = 1.0;
560    if(iQP >= 0)
561    {
562      weight = pow( 2.0, (iQP-g_aucChromaScale[iQP])/3.0 );  // takes into account of the chroma qp mapping without chroma qp Offset
563    }
564#else
565    Double weight = pow( 2.0, (iQP-g_aucChromaScale[iQP])/3.0 );  // takes into account of the chroma qp mapping without chroma qp Offset
566#endif
567    m_pcRdCost    ->setChromaDistortionWeight( weight );     
568#endif
569
570#if RDOQ_CHROMA_LAMBDA
571    // for RDOQ
572    m_pcTrQuant   ->setLambda( m_pdRdPicLambda[uiQpIdx], m_pdRdPicLambda[uiQpIdx] / weight );
573#else
574    m_pcTrQuant   ->setLambda              ( m_pdRdPicLambda[uiQpIdx] );
575#endif
576#if ALF_CHROMA_LAMBDA || SAO_CHROMA_LAMBDA
577    // For ALF or SAO
578    pcSlice       ->setLambda              ( m_pdRdPicLambda[uiQpIdx], m_pdRdPicLambda[uiQpIdx] / weight ); 
579#else
580    pcSlice       ->setLambda              ( m_pdRdPicLambda[uiQpIdx] );
581#endif
582#if HHI_INTER_VIEW_MOTION_PRED
583    m_pcRdCost    ->setLambdaMVReg         ( m_pdRdPicLambda[uiQpIdx] * m_pcCfg->getMultiviewMvRegLambdaScale() );
584#endif
585   
586    // try compress
587    compressSlice   ( rpcPic );
588   
589    Double dPicRdCost;
590    UInt64 uiPicDist        = m_uiPicDist;
591    UInt64 uiALFBits        = 0;
592   
593    m_pcGOPEncoder->preLoopFilterPicAll( rpcPic, uiPicDist, uiALFBits );
594   
595    // compute RD cost and choose the best
596    dPicRdCost = m_pcRdCost->calcRdCost64( m_uiPicTotalBits + uiALFBits, uiPicDist, true, DF_SSE_FRAME);
597   
598    if ( dPicRdCost < dPicRdCostBest )
599    {
600      uiQpIdxBest    = uiQpIdx;
601      dPicRdCostBest = dPicRdCost;
602    }
603  }
604 
605  // set best values
606  pcSlice       ->setSliceQp             ( m_piRdPicQp    [uiQpIdxBest] );
607#if ADAPTIVE_QP_SELECTION
608  pcSlice       ->setSliceQpBase         ( m_piRdPicQp    [uiQpIdxBest] );
609#endif
610  m_pcRdCost    ->setLambda              ( m_pdRdPicLambda[uiQpIdxBest] );
611#if WEIGHTED_CHROMA_DISTORTION
612  // in RdCost there is only one lambda because the luma and chroma bits are not separated, instead we weight the distortion of chroma.
613  int iQP = m_piRdPicQp    [uiQpIdxBest];
614#if H0736_AVC_STYLE_QP_RANGE
615  Double weight = 1.0;
616  if(iQP >= 0)
617  {
618    weight = pow( 2.0, (iQP-g_aucChromaScale[iQP])/3.0 );  // takes into account of the chroma qp mapping without chroma qp Offset
619  }
620#else
621  Double weight = pow( 2.0, (iQP-g_aucChromaScale[iQP])/3.0 );  // takes into account of the chroma qp mapping without chroma qp Offset
622#endif
623  m_pcRdCost ->setChromaDistortionWeight( weight );     
624#endif
625
626#if RDOQ_CHROMA_LAMBDA
627  // for RDOQ
628  m_pcTrQuant   ->setLambda( m_pdRdPicLambda[uiQpIdxBest], m_pdRdPicLambda[uiQpIdxBest] / weight ); 
629#else
630  m_pcTrQuant   ->setLambda              ( m_pdRdPicLambda[uiQpIdxBest] );
631#endif
632#if ALF_CHROMA_LAMBDA || SAO_CHROMA_LAMBDA
633  // For ALF or SAO
634  pcSlice       ->setLambda              ( m_pdRdPicLambda[uiQpIdxBest], m_pdRdPicLambda[uiQpIdxBest] / weight ); 
635#else
636  pcSlice       ->setLambda              ( m_pdRdPicLambda[uiQpIdxBest] );
637#endif
638#if HHI_INTER_VIEW_MOTION_PRED
639  m_pcRdCost    ->setLambdaMVReg         ( m_pdRdPicLambda[uiQpIdxBest] * m_pcCfg->getMultiviewMvRegLambdaScale() );
640#endif
641}
642
643/** \param rpcPic   picture class
644 */
645Void TEncSlice::compressSlice( TComPic*& rpcPic )
646{
647  UInt  uiCUAddr;
648  UInt   uiStartCUAddr;
649  UInt   uiBoundingCUAddr;
650  rpcPic->getSlice(getSliceIdx())->setEntropySliceCounter(0);
651  TEncBinCABAC* pppcRDSbacCoder = NULL;
652  TComSlice* pcSlice            = rpcPic->getSlice(getSliceIdx());
653  xDetermineStartAndBoundingCUAddr ( uiStartCUAddr, uiBoundingCUAddr, rpcPic, false );
654#if LG_ZEROINTRADEPTHRESI_M26039
655  rpcPic->setIntraPeriod(this->m_pcCfg->getIntraPeriod());
656#endif
657 
658  // initialize cost values
659  m_uiPicTotalBits  = 0;
660  m_dPicRdCost      = 0;
661  m_uiPicDist       = 0;
662 
663#if CABAC_INIT_FLAG && POZNAN_CABAC_INIT_FLAG_FIX
664  Bool bReset =(pcSlice->getPOC() == 0) || 
665    (pcSlice->getPOC() % m_pcCfg->getIntraPeriod() == 0) ||
666    (pcSlice->getPPS()->getEncPrevPOC() % m_pcCfg->getIntraPeriod() == 0) ||
667    (pcSlice->getPOC()/m_pcCfg->getIntraPeriod() > pcSlice->getPPS()->getEncPrevPOC()/m_pcCfg->getIntraPeriod()) ||
668    (m_pcGOPEncoder->getGOPSize() == 0);
669
670  if ( bReset && pcSlice->getPPS()->getCabacInitPresentFlag())
671  {
672    pcSlice->getPPS()->setEncCABACTableIdx(pcSlice->getSliceType()); // reset cabac initialization table index
673  };
674#endif
675
676  // set entropy coder
677  if( m_pcCfg->getUseSBACRD() )
678  {
679    m_pcSbacCoder->init( m_pcBinCABAC );
680    m_pcEntropyCoder->setEntropyCoder   ( m_pcSbacCoder, pcSlice );
681    m_pcEntropyCoder->resetEntropy      ();
682    m_pppcRDSbacCoder[0][CI_CURR_BEST]->load(m_pcSbacCoder);
683    pppcRDSbacCoder = (TEncBinCABAC *) m_pppcRDSbacCoder[0][CI_CURR_BEST]->getEncBinIf();
684    pppcRDSbacCoder->setBinCountingEnableFlag( false );
685    pppcRDSbacCoder->setBinsCoded( 0 );
686  }
687  else
688  {
689    m_pcEntropyCoder->setEntropyCoder ( m_pcCavlcCoder, pcSlice );
690    m_pcEntropyCoder->resetEntropy      ();
691    m_pcEntropyCoder->setBitstream    ( m_pcBitCounter );
692  }
693 
694  //------------------------------------------------------------------------------
695  //  Weighted Prediction parameters estimation.
696  //------------------------------------------------------------------------------
697  // calculate AC/DC values for current picture
698  if( pcSlice->getPPS()->getUseWP() || pcSlice->getPPS()->getWPBiPredIdc() )
699  {
700    xCalcACDCParamSlice(pcSlice);
701  }
702
703  Bool bWp_explicit = (pcSlice->getSliceType()==P_SLICE && pcSlice->getPPS()->getUseWP()) || (pcSlice->getSliceType()==B_SLICE && pcSlice->getPPS()->getWPBiPredIdc()==1);
704  Bool bWp_implicit = (pcSlice->getSliceType()==B_SLICE && pcSlice->getPPS()->getWPBiPredIdc()==2);
705
706  if ( bWp_explicit || bWp_implicit )
707  {
708    //------------------------------------------------------------------------------
709    //  Weighted Prediction implemented at Slice level. SliceMode=2 is not supported yet.
710    //------------------------------------------------------------------------------
711    if ( pcSlice->getSliceMode()==2 || pcSlice->getEntropySliceMode()==2 )
712    {
713      printf("Weighted Prediction is not supported with slice mode determined by max number of bins.\n"); exit(0);
714    }
715
716    if( bWp_explicit )
717    {
718      xEstimateWPParamSlice( pcSlice );
719    }
720   
721    pcSlice->initWpScaling();
722
723    // check WP on/off
724    if( bWp_explicit )
725    {
726      xCheckWPEnable( pcSlice );
727    }
728  }
729
730#if ADAPTIVE_QP_SELECTION
731  if( m_pcCfg->getUseAdaptQpSelect() )
732  {
733    m_pcTrQuant->clearSliceARLCnt();
734    if(pcSlice->getSliceType()!=I_SLICE)
735    {
736      Int qpBase = pcSlice->getSliceQpBase();
737      pcSlice->setSliceQp(qpBase + m_pcTrQuant->getQpDelta(qpBase));
738    }
739  }
740#endif
741  // initialize ALF parameters
742  m_pcEntropyCoder->setAlfCtrl(false);
743  m_pcEntropyCoder->setMaxAlfCtrlDepth(0); //unnecessary
744 
745#if SAIT_VSO_EST_A0033
746 if( m_pcCfg->getUseVSO() )
747 {
748
749   Int frameWidth = m_pcCfg->getSourceWidth();
750   Pel* pVideoRec = m_pcRdCost->getVideoRecPicYuv()->getLumaAddr();
751   Int iVideoRecStride = m_pcRdCost->getVideoRecPicYuv()->getStride();
752
753   Pel* pDepthOrg = m_pcRdCost->getDepthPicYuv()->getLumaAddr();
754   Int iDepthOrgStride = m_pcRdCost->getDepthPicYuv()->getStride();
755
756   for( Int y = 0 ; y < m_pcCfg->getSourceHeight() ; y++ )
757   {
758     pVideoRec[-1] = pVideoRec[0];
759     pVideoRec[frameWidth] = pVideoRec[frameWidth-1];
760     pDepthOrg[-1] = pDepthOrg[0];
761     pDepthOrg[frameWidth] = pDepthOrg[frameWidth-1];
762
763     pVideoRec += iVideoRecStride;
764     pDepthOrg += iDepthOrgStride;
765   }
766 }
767#endif
768
769  TEncTop* pcEncTop = (TEncTop*) m_pcCfg;
770  TEncSbac**** ppppcRDSbacCoders    = pcEncTop->getRDSbacCoders();
771  TComBitCounter* pcBitCounters     = pcEncTop->getBitCounters();
772  Int  iNumSubstreams = 1;
773  UInt uiTilesAcross  = 0;
774
775  if( m_pcCfg->getUseSBACRD() )
776  {
777    iNumSubstreams = pcSlice->getPPS()->getNumSubstreams();
778    uiTilesAcross = rpcPic->getPicSym()->getNumColumnsMinus1()+1;
779    delete[] m_pcBufferSbacCoders;
780    delete[] m_pcBufferBinCoderCABACs;
781    m_pcBufferSbacCoders     = new TEncSbac    [uiTilesAcross];
782    m_pcBufferBinCoderCABACs = new TEncBinCABAC[uiTilesAcross];
783    for (int ui = 0; ui < uiTilesAcross; ui++)
784    {
785      m_pcBufferSbacCoders[ui].init( &m_pcBufferBinCoderCABACs[ui] );
786    }
787    for (UInt ui = 0; ui < uiTilesAcross; ui++)
788    {
789      m_pcBufferSbacCoders[ui].load(m_pppcRDSbacCoder[0][CI_CURR_BEST]);  //init. state
790    }
791
792    for ( UInt ui = 0 ; ui < iNumSubstreams ; ui++ ) //init all sbac coders for RD optimization
793    {
794      ppppcRDSbacCoders[ui][0][CI_CURR_BEST]->load(m_pppcRDSbacCoder[0][CI_CURR_BEST]);
795    }
796  }
797  //if( m_pcCfg->getUseSBACRD() )
798  {
799    delete[] m_pcBufferLowLatSbacCoders;
800    delete[] m_pcBufferLowLatBinCoderCABACs;
801    m_pcBufferLowLatSbacCoders     = new TEncSbac    [uiTilesAcross];
802    m_pcBufferLowLatBinCoderCABACs = new TEncBinCABAC[uiTilesAcross];
803    for (int ui = 0; ui < uiTilesAcross; ui++)
804    {
805      m_pcBufferLowLatSbacCoders[ui].init( &m_pcBufferLowLatBinCoderCABACs[ui] );
806    }
807    for (UInt ui = 0; ui < uiTilesAcross; ui++)
808      m_pcBufferLowLatSbacCoders[ui].load(m_pppcRDSbacCoder[0][CI_CURR_BEST]);  //init. state
809  }
810  UInt uiWidthInLCUs  = rpcPic->getPicSym()->getFrameWidthInCU();
811  //UInt uiHeightInLCUs = rpcPic->getPicSym()->getFrameHeightInCU();
812  UInt uiCol=0, uiLin=0, uiSubStrm=0;
813#if !REMOVE_TILE_DEPENDENCE
814  Int  iBreakDep      = 0;
815#endif
816  UInt uiTileCol      = 0;
817  UInt uiTileStartLCU = 0;
818  UInt uiTileLCUX     = 0;
819
820#if HHI_VSO_SPEEDUP_A0033
821  Int iLastPosY = -1;
822#endif
823
824  // for every CU in slice
825  UInt uiEncCUOrder;
826  uiCUAddr = rpcPic->getPicSym()->getCUOrderMap( uiStartCUAddr /rpcPic->getNumPartInCU()); 
827  for( uiEncCUOrder = uiStartCUAddr/rpcPic->getNumPartInCU();
828       uiEncCUOrder < (uiBoundingCUAddr+(rpcPic->getNumPartInCU()-1))/rpcPic->getNumPartInCU();
829       uiCUAddr = rpcPic->getPicSym()->getCUOrderMap(++uiEncCUOrder) )
830  {
831    // initialize CU encoder
832    TComDataCU*& pcCU = rpcPic->getCU( uiCUAddr );
833    pcCU->initCU( rpcPic, uiCUAddr );
834
835#if HHI_VSO_SPEEDUP_A0033
836    if ( m_pcRdCost->getUseRenModel() )
837    {
838      // updated renderer model if necessary
839      Int iCurPosX;
840      Int iCurPosY; 
841      pcCU->getPosInPic(0, iCurPosX, iCurPosY );
842      if ( iCurPosY != iLastPosY )
843      {
844        iLastPosY = iCurPosY; 
845       
846        m_pcGOPEncoder->getEncTop()->getEncTop()->setupRenModel( rpcPic->getCurrSlice()->getPOC() , rpcPic->getCurrSlice()->getSPS()->getViewId(), rpcPic->getCurrSlice()->getSPS()->isDepth() ? 1 : 0, iCurPosY );
847      }
848    }   
849#endif
850
851
852    // inherit from TR if necessary, select substream to use.
853    if( m_pcCfg->getUseSBACRD() )
854    {
855#if !REMOVE_TILE_DEPENDENCE
856      iBreakDep = rpcPic->getPicSym()->getTileBoundaryIndependenceIdr();
857#endif
858      uiTileCol = rpcPic->getPicSym()->getTileIdxMap(uiCUAddr) % (rpcPic->getPicSym()->getNumColumnsMinus1()+1); // what column of tiles are we in?
859      uiTileStartLCU = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr();
860      uiTileLCUX = uiTileStartLCU % uiWidthInLCUs;
861      //UInt uiSliceStartLCU = pcSlice->getSliceCurStartCUAddr();
862      uiCol     = uiCUAddr % uiWidthInLCUs;
863      uiLin     = uiCUAddr / uiWidthInLCUs;
864#if !REMOVE_TILE_DEPENDENCE
865#if WPP_SIMPLIFICATION
866      if (iBreakDep && pcSlice->getPPS()->getNumSubstreams() > 1)
867#else
868      if (iBreakDep && pcSlice->getPPS()->getEntropyCodingSynchro())
869#endif
870#else
871#if WPP_SIMPLIFICATION
872      if (pcSlice->getPPS()->getNumSubstreams() > 1)
873#else
874      if (pcSlice->getPPS()->getEntropyCodingSynchro())
875#endif
876#endif
877      {
878        // independent tiles => substreams are "per tile".  iNumSubstreams has already been multiplied.
879        Int iNumSubstreamsPerTile = iNumSubstreams/rpcPic->getPicSym()->getNumTiles();
880        uiSubStrm = rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)*iNumSubstreamsPerTile
881                      + uiLin%iNumSubstreamsPerTile;
882      }
883      else
884      {
885        // dependent tiles => substreams are "per frame".
886        uiSubStrm = uiLin % iNumSubstreams;
887      }
888#if WPP_SIMPLIFICATION
889      if ( pcSlice->getPPS()->getNumSubstreams() > 1 && (uiCol == uiTileLCUX) )
890#else
891      if ( pcSlice->getPPS()->getEntropyCodingSynchro() && (uiCol == uiTileLCUX) )
892#endif
893      {
894        // We'll sync if the TR is available.
895        TComDataCU *pcCUUp = pcCU->getCUAbove();
896        UInt uiWidthInCU = rpcPic->getFrameWidthInCU();
897        UInt uiMaxParts = 1<<(pcSlice->getSPS()->getMaxCUDepth()<<1);
898        TComDataCU *pcCUTR = NULL;
899#if WPP_SIMPLIFICATION
900        if ( pcCUUp && ((uiCUAddr%uiWidthInCU+1) < uiWidthInCU)  )
901        {
902          pcCUTR = rpcPic->getCU( uiCUAddr - uiWidthInCU + 1 );
903        }
904#else
905        if ( pcCUUp && ((uiCUAddr%uiWidthInCU+pcSlice->getPPS()->getEntropyCodingSynchro()) < uiWidthInCU)  )
906        {
907          pcCUTR = rpcPic->getCU( uiCUAddr - uiWidthInCU + pcSlice->getPPS()->getEntropyCodingSynchro() );
908        }
909#endif
910        if ( ((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) || 
911             (pcCUTR->getSCUAddr()+uiMaxParts-1 < pcSlice->getSliceCurStartCUAddr()) ||
912#if !REMOVE_TILE_DEPENDENCE
913             (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr() && (rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
914#else
915             ((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
916#endif
917             )||
918             ((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) || 
919             (pcCUTR->getSCUAddr()+uiMaxParts-1 < pcSlice->getEntropySliceCurStartCUAddr()) ||
920#if !REMOVE_TILE_DEPENDENCE
921             (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr() && (rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
922#else
923             ((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
924#endif
925             )
926           )
927        {
928          // TR not available.
929        }
930        else
931        {
932          // TR is available, we use it.
933          ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST]->loadContexts( &m_pcBufferSbacCoders[uiTileCol] );
934        }
935      }
936      m_pppcRDSbacCoder[0][CI_CURR_BEST]->load( ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST] ); //this load is used to simplify the code
937    }
938
939    // reset the entropy coder
940    if( uiCUAddr == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr() &&                                   // must be first CU of tile
941        uiCUAddr!=0 &&                                                                                                                                    // cannot be first CU of picture
942        uiCUAddr!=rpcPic->getPicSym()->getPicSCUAddr(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceCurStartCUAddr())/rpcPic->getNumPartInCU())     // cannot be first CU of slice
943    {
944#if CABAC_INIT_FLAG
945      SliceType sliceType = pcSlice->getSliceType();
946      if (!pcSlice->isIntra() && pcSlice->getPPS()->getCabacInitPresentFlag() && pcSlice->getPPS()->getEncCABACTableIdx()!=0)
947      {
948        sliceType = (SliceType) pcSlice->getPPS()->getEncCABACTableIdx();
949      }
950      m_pcEntropyCoder->updateContextTables ( sliceType, pcSlice->getSliceQp(), false );
951      m_pcEntropyCoder->setEntropyCoder     ( m_pppcRDSbacCoder[0][CI_CURR_BEST], pcSlice );
952      m_pcEntropyCoder->updateContextTables ( sliceType, pcSlice->getSliceQp() );
953      m_pcEntropyCoder->setEntropyCoder     ( m_pcSbacCoder, pcSlice );
954#else
955      m_pcEntropyCoder->updateContextTables ( pcSlice->getSliceType(), pcSlice->getSliceQp(), false );
956      m_pcEntropyCoder->setEntropyCoder     ( m_pppcRDSbacCoder[0][CI_CURR_BEST], pcSlice );
957      m_pcEntropyCoder->updateContextTables ( pcSlice->getSliceType(), pcSlice->getSliceQp() );
958      m_pcEntropyCoder->setEntropyCoder     ( m_pcSbacCoder, pcSlice );
959#endif
960    }
961#if !REMOVE_TILE_DEPENDENCE
962    if( (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr()==0) && (rpcPic->getPicSym()->getNumColumnsMinus1()!=0) )
963    {
964      // Synchronize cabac probabilities with LCU among Tiles
965      if( (uiTileLCUX != 0) &&
966          (uiCUAddr == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr()) )
967      { 
968        TComDataCU *pcCULeft = pcCU->getCULeft();
969        UInt uiMaxParts = 1<<(pcSlice->getSPS()->getMaxCUDepth()<<1);
970
971        if ( (true/*bEnforceSliceRestriction*/ &&
972              ((pcCULeft==NULL) || (pcCULeft->getSlice()==NULL) || 
973               ((pcCULeft->getSCUAddr()+uiMaxParts-1) < pcSlice->getSliceCurStartCUAddr()) 
974              )
975             )||
976             (true/*bEnforceEntropySliceRestriction*/ &&
977              ((pcCULeft==NULL) || (pcCULeft->getSlice()==NULL) || 
978               ((pcCULeft->getSCUAddr()+uiMaxParts-1) < pcSlice->getEntropySliceCurStartCUAddr())
979              )
980             )
981           )
982        {
983          // Left not available.
984        }
985        else
986        {
987          // Left is available, we use it.
988          ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST]->loadContexts( &m_pcBufferLowLatSbacCoders[uiTileCol-1] );
989          m_pppcRDSbacCoder[0][CI_CURR_BEST]->loadContexts( ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST] ); //this load is used to simplify the code
990        }
991      }
992    }
993#endif
994    // if RD based on SBAC is used
995    if( m_pcCfg->getUseSBACRD() )
996    {
997      // set go-on entropy coder
998      m_pcEntropyCoder->setEntropyCoder ( m_pcRDGoOnSbacCoder, pcSlice );
999      m_pcEntropyCoder->setBitstream( &pcBitCounters[uiSubStrm] );
1000     
1001      ((TEncBinCABAC*)m_pcRDGoOnSbacCoder->getEncBinIf())->setBinCountingEnableFlag(true);
1002      // run CU encoder
1003      m_pcCuEncoder->compressCU( pcCU );
1004     
1005      // restore entropy coder to an initial stage
1006      m_pcEntropyCoder->setEntropyCoder ( m_pppcRDSbacCoder[0][CI_CURR_BEST], pcSlice );
1007      m_pcEntropyCoder->setBitstream( &pcBitCounters[uiSubStrm] );
1008      m_pcCuEncoder->setBitCounter( &pcBitCounters[uiSubStrm] );
1009      m_pcBitCounter = &pcBitCounters[uiSubStrm];
1010      pppcRDSbacCoder->setBinCountingEnableFlag( true );
1011      m_pcBitCounter->resetBits();
1012      pppcRDSbacCoder->setBinsCoded( 0 );
1013      m_pcCuEncoder->encodeCU( pcCU );
1014
1015      pppcRDSbacCoder->setBinCountingEnableFlag( false );
1016      if (m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_BYTES_IN_SLICE && ( ( pcSlice->getSliceBits() + m_pcEntropyCoder->getNumberOfWrittenBits() ) ) > m_pcCfg->getSliceArgument()<<3)
1017      {
1018        pcSlice->setNextSlice( true );
1019        break;
1020      }
1021      if (m_pcCfg->getEntropySliceMode()==SHARP_MULTIPLE_CONSTRAINT_BASED_ENTROPY_SLICE && pcSlice->getEntropySliceCounter()+pppcRDSbacCoder->getBinsCoded() > m_pcCfg->getEntropySliceArgument()&&pcSlice->getSliceCurEndCUAddr()!=pcSlice->getEntropySliceCurEndCUAddr())
1022      {
1023        pcSlice->setNextEntropySlice( true );
1024        break;
1025      }
1026      if( m_pcCfg->getUseSBACRD() )
1027      {
1028         ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST]->load( m_pppcRDSbacCoder[0][CI_CURR_BEST] );
1029       
1030         //Store probabilties of second LCU in line into buffer
1031#if WPP_SIMPLIFICATION
1032        if (pcSlice->getPPS()->getNumSubstreams() > 1 && uiCol == uiTileLCUX+1)
1033#else
1034        if (pcSlice->getPPS()->getEntropyCodingSynchro() && uiCol == uiTileLCUX+pcSlice->getPPS()->getEntropyCodingSynchro())
1035#endif
1036        {
1037          m_pcBufferSbacCoders[uiTileCol].loadContexts(ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST]);
1038        }
1039      }
1040#if !REMOVE_TILE_DEPENDENCE
1041      if( (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr()==0) && (rpcPic->getPicSym()->getNumColumnsMinus1()!=0) )
1042      {
1043         //Store probabilties for next tile
1044        if( (uiLin == (rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr() / uiWidthInLCUs )) && 
1045            (uiCol == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getRightEdgePosInCU()) )
1046        {
1047          m_pcBufferLowLatSbacCoders[uiTileCol].loadContexts(ppppcRDSbacCoders[uiSubStrm][0][CI_CURR_BEST]);
1048        }
1049      }
1050#endif
1051    }
1052    // other case: encodeCU is not called
1053    else
1054    {
1055      m_pcCuEncoder->compressCU( pcCU );
1056      m_pcCuEncoder->encodeCU( pcCU );
1057      if (m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_BYTES_IN_SLICE && ( ( pcSlice->getSliceBits()+ m_pcEntropyCoder->getNumberOfWrittenBits() ) ) > m_pcCfg->getSliceArgument()<<3)
1058      {
1059        pcSlice->setNextSlice( true );
1060        break;
1061      }
1062      if (m_pcCfg->getEntropySliceMode()==SHARP_MULTIPLE_CONSTRAINT_BASED_ENTROPY_SLICE && pcSlice->getEntropySliceCounter()+ m_pcEntropyCoder->getNumberOfWrittenBits()> m_pcCfg->getEntropySliceArgument()&&pcSlice->getSliceCurEndCUAddr()!=pcSlice->getEntropySliceCurEndCUAddr())
1063      {
1064        pcSlice->setNextEntropySlice( true );
1065        break;
1066      }
1067    }
1068   
1069    m_uiPicTotalBits += pcCU->getTotalBits();
1070    m_dPicRdCost     += pcCU->getTotalCost();
1071    m_uiPicDist      += pcCU->getTotalDistortion();
1072  }
1073  xRestoreWPparam( pcSlice );
1074}
1075
1076/**
1077 \param  rpcPic        picture class
1078 \retval rpcBitstream  bitstream class
1079 */
1080Void TEncSlice::encodeSlice   ( TComPic*& rpcPic, TComOutputBitstream* pcBitstream, TComOutputBitstream* pcSubstreams )
1081{
1082  UInt       uiCUAddr;
1083  UInt       uiStartCUAddr;
1084  UInt       uiBoundingCUAddr;
1085  TComSlice* pcSlice = rpcPic->getSlice(getSliceIdx());
1086
1087  uiStartCUAddr=pcSlice->getEntropySliceCurStartCUAddr();
1088  uiBoundingCUAddr=pcSlice->getEntropySliceCurEndCUAddr();
1089  // choose entropy coder
1090  {
1091    m_pcSbacCoder->init( (TEncBinIf*)m_pcBinCABAC );
1092    m_pcEntropyCoder->setEntropyCoder ( m_pcSbacCoder, pcSlice );
1093  }
1094 
1095  m_pcCuEncoder->setBitCounter( NULL );
1096  m_pcBitCounter = NULL;
1097  // Appropriate substream bitstream is switched later.
1098  // for every CU
1099#if ENC_DEC_TRACE
1100  g_bJustDoIt = g_bEncDecTraceEnable;
1101#endif
1102  DTRACE_CABAC_VL( g_nSymbolCounter++ );
1103  DTRACE_CABAC_T( "\tPOC: " );
1104  DTRACE_CABAC_V( rpcPic->getPOC() );
1105  DTRACE_CABAC_T( "\n" );
1106#if ENC_DEC_TRACE
1107  g_bJustDoIt = g_bEncDecTraceDisable;
1108#endif
1109
1110  TEncTop* pcEncTop = (TEncTop*) m_pcCfg;
1111  TEncSbac* pcSbacCoders = pcEncTop->getSbacCoders(); //coder for each substream
1112  Int iNumSubstreams = pcSlice->getPPS()->getNumSubstreams();
1113  UInt uiBitsOriginallyInSubstreams = 0;
1114  {
1115    UInt uiTilesAcross = rpcPic->getPicSym()->getNumColumnsMinus1()+1;
1116    for (UInt ui = 0; ui < uiTilesAcross; ui++)
1117    {
1118      m_pcBufferSbacCoders[ui].load(m_pcSbacCoder); //init. state
1119    }
1120   
1121    for (Int iSubstrmIdx=0; iSubstrmIdx < iNumSubstreams; iSubstrmIdx++)
1122    {
1123      uiBitsOriginallyInSubstreams += pcSubstreams[iSubstrmIdx].getNumberOfWrittenBits();
1124    }
1125
1126    for (UInt ui = 0; ui < uiTilesAcross; ui++)
1127    {
1128      m_pcBufferLowLatSbacCoders[ui].load(m_pcSbacCoder);  //init. state
1129    }
1130  }
1131
1132  UInt uiWidthInLCUs  = rpcPic->getPicSym()->getFrameWidthInCU();
1133  UInt uiCol=0, uiLin=0, uiSubStrm=0;
1134#if !REMOVE_TILE_DEPENDENCE
1135  Int  iBreakDep      = 0;
1136#endif
1137  UInt uiTileCol      = 0;
1138  UInt uiTileStartLCU = 0;
1139  UInt uiTileLCUX     = 0;
1140
1141  UInt uiEncCUOrder;
1142  uiCUAddr = rpcPic->getPicSym()->getCUOrderMap( uiStartCUAddr /rpcPic->getNumPartInCU());  /*for tiles, uiStartCUAddr is NOT the real raster scan address, it is actually
1143                                                                                              an encoding order index, so we need to convert the index (uiStartCUAddr)
1144                                                                                              into the real raster scan address (uiCUAddr) via the CUOrderMap*/
1145  for( uiEncCUOrder = uiStartCUAddr /rpcPic->getNumPartInCU();
1146       uiEncCUOrder < (uiBoundingCUAddr+rpcPic->getNumPartInCU()-1)/rpcPic->getNumPartInCU();
1147       uiCUAddr = rpcPic->getPicSym()->getCUOrderMap(++uiEncCUOrder) )
1148  {
1149    if( m_pcCfg->getUseSBACRD() )
1150    {
1151#if !REMOVE_TILE_DEPENDENCE
1152      iBreakDep = rpcPic->getPicSym()->getTileBoundaryIndependenceIdr();
1153#endif
1154      uiTileCol = rpcPic->getPicSym()->getTileIdxMap(uiCUAddr) % (rpcPic->getPicSym()->getNumColumnsMinus1()+1); // what column of tiles are we in?
1155      uiTileStartLCU = rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr();
1156      uiTileLCUX = uiTileStartLCU % uiWidthInLCUs;
1157      //UInt uiSliceStartLCU = pcSlice->getSliceCurStartCUAddr();
1158      uiCol     = uiCUAddr % uiWidthInLCUs;
1159      uiLin     = uiCUAddr / uiWidthInLCUs;
1160#if !REMOVE_TILE_DEPENDENCE
1161#if WPP_SIMPLIFICATION
1162      if (iBreakDep && pcSlice->getPPS()->getNumSubstreams() > 1)
1163#else
1164      if (iBreakDep && pcSlice->getPPS()->getEntropyCodingSynchro())
1165#endif
1166#else
1167#if WPP_SIMPLIFICATION
1168      if (pcSlice->getPPS()->getNumSubstreams() > 1)
1169#else
1170      if (pcSlice->getPPS()->getEntropyCodingSynchro())
1171#endif
1172#endif
1173      {
1174        // independent tiles => substreams are "per tile".  iNumSubstreams has already been multiplied.
1175        Int iNumSubstreamsPerTile = iNumSubstreams/rpcPic->getPicSym()->getNumTiles();
1176        uiSubStrm = rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)*iNumSubstreamsPerTile
1177                      + uiLin%iNumSubstreamsPerTile;
1178      }
1179      else
1180      {
1181        // dependent tiles => substreams are "per frame".
1182        uiSubStrm = uiLin % iNumSubstreams;
1183      }
1184
1185      m_pcEntropyCoder->setBitstream( &pcSubstreams[uiSubStrm] );
1186
1187      // Synchronize cabac probabilities with upper-right LCU if it's available and we're at the start of a line.
1188#if WPP_SIMPLIFICATION
1189      if (pcSlice->getPPS()->getNumSubstreams() > 1 && (uiCol == uiTileLCUX))
1190#else
1191      if (pcSlice->getPPS()->getEntropyCodingSynchro() && (uiCol == uiTileLCUX))
1192#endif
1193      {
1194        // We'll sync if the TR is available.
1195        TComDataCU *pcCUUp = rpcPic->getCU( uiCUAddr )->getCUAbove();
1196        UInt uiWidthInCU = rpcPic->getFrameWidthInCU();
1197        UInt uiMaxParts = 1<<(pcSlice->getSPS()->getMaxCUDepth()<<1);
1198        TComDataCU *pcCUTR = NULL;
1199#if WPP_SIMPLIFICATION
1200        if ( pcCUUp && ((uiCUAddr%uiWidthInCU+1) < uiWidthInCU)  )
1201        {
1202          pcCUTR = rpcPic->getCU( uiCUAddr - uiWidthInCU + 1 );
1203        }
1204#else
1205        if ( pcCUUp && ((uiCUAddr%uiWidthInCU+pcSlice->getPPS()->getEntropyCodingSynchro()) < uiWidthInCU)  )
1206        {
1207          pcCUTR = rpcPic->getCU( uiCUAddr - uiWidthInCU + pcSlice->getPPS()->getEntropyCodingSynchro() );
1208        }
1209#endif
1210        if ( (true/*bEnforceSliceRestriction*/ &&
1211             ((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) || 
1212             (pcCUTR->getSCUAddr()+uiMaxParts-1 < pcSlice->getSliceCurStartCUAddr()) ||
1213#if !REMOVE_TILE_DEPENDENCE
1214             (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr() && (rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
1215#else
1216             ((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
1217#endif
1218             ))||
1219             (true/*bEnforceEntropySliceRestriction*/ &&
1220             ((pcCUTR==NULL) || (pcCUTR->getSlice()==NULL) || 
1221             (pcCUTR->getSCUAddr()+uiMaxParts-1 < pcSlice->getEntropySliceCurStartCUAddr()) ||
1222#if !REMOVE_TILE_DEPENDENCE
1223             (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr() && (rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
1224#else
1225             ((rpcPic->getPicSym()->getTileIdxMap( pcCUTR->getAddr() ) != rpcPic->getPicSym()->getTileIdxMap(uiCUAddr)))
1226#endif
1227             ))
1228           )
1229        {
1230          // TR not available.
1231        }
1232        else
1233        {
1234          // TR is available, we use it.
1235          pcSbacCoders[uiSubStrm].loadContexts( &m_pcBufferSbacCoders[uiTileCol] );
1236        }
1237      }
1238      m_pcSbacCoder->load(&pcSbacCoders[uiSubStrm]);  //this load is used to simplify the code (avoid to change all the call to m_pcSbacCoder)
1239    }
1240    // reset the entropy coder
1241    if( uiCUAddr == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr() &&                                   // must be first CU of tile
1242        uiCUAddr!=0 &&                                                                                                                                    // cannot be first CU of picture
1243        uiCUAddr!=rpcPic->getPicSym()->getPicSCUAddr(rpcPic->getSlice(rpcPic->getCurrSliceIdx())->getSliceCurStartCUAddr())/rpcPic->getNumPartInCU())     // cannot be first CU of slice
1244    {
1245      Int iTileIdx            = rpcPic->getPicSym()->getTileIdxMap(uiCUAddr);
1246      Bool bWriteTileMarker   = false;
1247      // check if current iTileIdx should have a marker
1248      for (Int iEntryIdx=0; iEntryIdx<m_pcCfg->getMaxTileMarkerEntryPoints()-1; iEntryIdx++)
1249      {
1250        bWriteTileMarker = ( (((Int)((iEntryIdx+1)*m_pcCfg->getMaxTileMarkerOffset()+0.5)) == iTileIdx ) && iEntryIdx < (m_pcCfg->getMaxTileMarkerEntryPoints()-1)) ? true : false;
1251        if (bWriteTileMarker)
1252        {
1253          break;
1254        }
1255      }
1256      {
1257        // We're crossing into another tile, tiles are independent.
1258        // When tiles are independent, we have "substreams per tile".  Each substream has already been terminated, and we no longer
1259        // have to perform it here.
1260#if WPP_SIMPLIFICATION
1261        if (pcSlice->getPPS()->getNumSubstreams() > 1)
1262#else
1263        if (pcSlice->getPPS()->getEntropyCodingSynchro())
1264#endif
1265        {
1266          ; // do nothing.
1267        }
1268        else
1269        {
1270#if CABAC_INIT_FLAG
1271          SliceType sliceType  = pcSlice->getSliceType();
1272          if (!pcSlice->isIntra() && pcSlice->getPPS()->getCabacInitPresentFlag() && pcSlice->getPPS()->getEncCABACTableIdx()!=0)
1273          {
1274            sliceType = (SliceType) pcSlice->getPPS()->getEncCABACTableIdx();
1275          }
1276          m_pcEntropyCoder->updateContextTables( sliceType, pcSlice->getSliceQp() );
1277#else
1278          m_pcEntropyCoder->updateContextTables( pcSlice->getSliceType(), pcSlice->getSliceQp() );
1279#endif
1280          pcSubstreams[uiSubStrm].write( 1, 1 );
1281          pcSubstreams[uiSubStrm].writeAlignZero();
1282        }
1283      }
1284      {
1285        // Write TileMarker into the appropriate substream (nothing has been written to it yet).
1286        if (m_pcCfg->getTileMarkerFlag() && bWriteTileMarker)
1287        {
1288          // Log locations where tile markers are to be inserted during emulation prevention
1289          UInt uiMarkerCount = pcSubstreams[uiSubStrm].getTileMarkerLocationCount();
1290          pcSubstreams[uiSubStrm].setTileMarkerLocation     ( uiMarkerCount, pcSubstreams[uiSubStrm].getNumberOfWrittenBits() >> 3 );
1291          pcSubstreams[uiSubStrm].setTileMarkerLocationCount( uiMarkerCount + 1 );
1292          // Write tile index
1293          m_pcEntropyCoder->writeTileMarker(iTileIdx, rpcPic->getPicSym()->getBitsUsedByTileIdx()); // Tile index
1294        }
1295
1296       
1297        UInt uiAccumulatedSubstreamLength = 0;
1298        for (Int iSubstrmIdx=0; iSubstrmIdx < iNumSubstreams; iSubstrmIdx++)
1299        {
1300          uiAccumulatedSubstreamLength += pcSubstreams[iSubstrmIdx].getNumberOfWrittenBits();
1301        }
1302        UInt uiLocationCount = pcSlice->getTileLocationCount();
1303        // add bits coded in previous entropy slices + bits coded so far
1304        pcSlice->setTileLocation( uiLocationCount, (pcSlice->getTileOffstForMultES() + uiAccumulatedSubstreamLength - uiBitsOriginallyInSubstreams) >> 3 ); 
1305        pcSlice->setTileLocationCount( uiLocationCount + 1 );
1306      }
1307    }
1308
1309    TComDataCU*& pcCU = rpcPic->getCU( uiCUAddr );   
1310#if !REMOVE_TILE_DEPENDENCE
1311    if( (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr()==0) && (rpcPic->getPicSym()->getNumColumnsMinus1()!=0) )
1312    {   
1313      // Synchronize cabac probabilities with LCU among Tiles
1314      if( (uiTileLCUX != 0) &&
1315          (uiCUAddr == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr()) )
1316      {
1317        TComDataCU *pcCULeft = pcCU->getCULeft();
1318        UInt uiMaxParts = 1<<(pcSlice->getSPS()->getMaxCUDepth()<<1);
1319
1320        if ( (true/*bEnforceSliceRestriction*/ &&
1321              ((pcCULeft==NULL) || (pcCULeft->getSlice()==NULL) || 
1322               ((pcCULeft->getSCUAddr()+uiMaxParts-1) < pcSlice->getSliceCurStartCUAddr()) 
1323              )
1324             )||
1325             (true/*bEnforceEntropySliceRestriction*/ &&
1326              ((pcCULeft==NULL) || (pcCULeft->getSlice()==NULL) || 
1327               ((pcCULeft->getSCUAddr()+uiMaxParts-1) < pcSlice->getEntropySliceCurStartCUAddr())
1328              )
1329             )
1330           )
1331        {
1332          // Left not available.
1333        }
1334        else
1335        {
1336          // Left is available, we use it.
1337          pcSbacCoders[uiSubStrm].loadContexts( &m_pcBufferLowLatSbacCoders[uiTileCol-1] );
1338          m_pcSbacCoder->loadContexts(&pcSbacCoders[uiSubStrm]);  //this load is used to simplify the code (avoid to change all the call to m_pcSbacCoder)
1339        }
1340      }
1341    }
1342#endif
1343
1344#if SAO_UNIT_INTERLEAVING
1345    if ( pcSlice->getSPS()->getUseSAO() && pcSlice->getAPS()->getSaoInterleavingFlag() && pcSlice->getSaoEnabledFlag() )
1346    {
1347      Int iNumCuInWidth     = pcSlice->getAPS()->getSaoParam()->numCuInWidth;
1348      Int iCUAddrInSlice    = uiCUAddr - (pcSlice->getSliceCurStartCUAddr() /rpcPic->getNumPartInCU());
1349      Int iCUAddrUpInSlice  = iCUAddrInSlice - iNumCuInWidth;
1350      Int rx = uiCUAddr % iNumCuInWidth;
1351      Int ry = uiCUAddr / iNumCuInWidth;
1352      m_pcEntropyCoder->encodeSaoUnitInterleaving( rx, ry, pcSlice->getAPS()->getSaoParam(),pcCU, iCUAddrInSlice, iCUAddrUpInSlice, pcSlice->getSPS()->getLFCrossSliceBoundaryFlag());
1353    }
1354#endif
1355#if ENC_DEC_TRACE
1356    g_bJustDoIt = g_bEncDecTraceEnable;
1357#endif
1358    if ( (m_pcCfg->getSliceMode()!=0 || m_pcCfg->getEntropySliceMode()!=0) && 
1359      uiCUAddr == rpcPic->getPicSym()->getCUOrderMap((uiBoundingCUAddr+rpcPic->getNumPartInCU()-1)/rpcPic->getNumPartInCU()-1) )
1360    {
1361      m_pcCuEncoder->encodeCU( pcCU, true );
1362    }
1363    else
1364    {
1365      m_pcCuEncoder->encodeCU( pcCU );
1366    }
1367#if ENC_DEC_TRACE
1368    g_bJustDoIt = g_bEncDecTraceDisable;
1369#endif   
1370    if( m_pcCfg->getUseSBACRD() )
1371    {
1372       pcSbacCoders[uiSubStrm].load(m_pcSbacCoder);   //load back status of the entropy coder after encoding the LCU into relevant bitstream entropy coder
1373       
1374
1375       //Store probabilties of second LCU in line into buffer
1376#if WPP_SIMPLIFICATION
1377      if (pcSlice->getPPS()->getNumSubstreams() > 1 && (uiCol == uiTileLCUX+1))
1378#else
1379      if (pcSlice->getPPS()->getEntropyCodingSynchro() && (uiCol == uiTileLCUX+pcSlice->getPPS()->getEntropyCodingSynchro()))
1380#endif
1381      {
1382        m_pcBufferSbacCoders[uiTileCol].loadContexts( &pcSbacCoders[uiSubStrm] );
1383      }
1384    }
1385#if !REMOVE_TILE_DEPENDENCE
1386    if( (rpcPic->getPicSym()->getTileBoundaryIndependenceIdr()==0) && (rpcPic->getPicSym()->getNumColumnsMinus1()!=0) )
1387    {
1388      pcSbacCoders[uiSubStrm].load(m_pcSbacCoder);   //load back status of the entropy coder after encoding the LCU into relevant bitstream entropy coder
1389       //Store probabilties for next tile
1390      if( (uiLin == (rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getFirstCUAddr() / uiWidthInLCUs )) && 
1391          (uiCol == rpcPic->getPicSym()->getTComTile(rpcPic->getPicSym()->getTileIdxMap(uiCUAddr))->getRightEdgePosInCU()) )
1392      {
1393        m_pcBufferLowLatSbacCoders[uiTileCol].loadContexts( &pcSbacCoders[uiSubStrm] );
1394      }
1395    }
1396#endif
1397  }
1398
1399#if ADAPTIVE_QP_SELECTION
1400  if( m_pcCfg->getUseAdaptQpSelect() )
1401  {
1402    m_pcTrQuant->storeSliceQpNext(pcSlice);
1403  }
1404#endif
1405#if CABAC_INIT_FLAG
1406  if (pcSlice->getPPS()->getCabacInitPresentFlag())
1407  {
1408    m_pcEntropyCoder->determineCabacInitIdx();
1409  }
1410#endif
1411}
1412
1413/** Determines the starting and bounding LCU address of current slice / entropy slice
1414 * \param bEncodeSlice Identifies if the calling function is compressSlice() [false] or encodeSlice() [true]
1415 * \returns Updates uiStartCUAddr, uiBoundingCUAddr with appropriate LCU address
1416 */
1417Void TEncSlice::xDetermineStartAndBoundingCUAddr  ( UInt& uiStartCUAddr, UInt& uiBoundingCUAddr, TComPic*& rpcPic, Bool bEncodeSlice )
1418{
1419  TComSlice* pcSlice = rpcPic->getSlice(getSliceIdx());
1420  UInt uiStartCUAddrSlice, uiBoundingCUAddrSlice;
1421#if FIXED_NUMBER_OF_TILES_SLICE_MODE
1422  UInt tileIdxIncrement;
1423  UInt tileIdx;
1424  UInt tileWidthInLcu;
1425  UInt tileHeightInLcu;
1426  UInt tileTotalCount;
1427#endif
1428
1429  uiStartCUAddrSlice        = pcSlice->getSliceCurStartCUAddr();
1430  UInt uiNumberOfCUsInFrame = rpcPic->getNumCUsInFrame();
1431  uiBoundingCUAddrSlice     = uiNumberOfCUsInFrame;
1432  if (bEncodeSlice) 
1433  {
1434    UInt uiCUAddrIncrement;
1435    switch (m_pcCfg->getSliceMode())
1436    {
1437    case AD_HOC_SLICES_FIXED_NUMBER_OF_LCU_IN_SLICE:
1438      uiCUAddrIncrement        = m_pcCfg->getSliceArgument();
1439      uiBoundingCUAddrSlice    = ((uiStartCUAddrSlice + uiCUAddrIncrement) < uiNumberOfCUsInFrame*rpcPic->getNumPartInCU()) ? (uiStartCUAddrSlice + uiCUAddrIncrement) : uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1440      break;
1441    case AD_HOC_SLICES_FIXED_NUMBER_OF_BYTES_IN_SLICE:
1442      uiCUAddrIncrement        = rpcPic->getNumCUsInFrame();
1443      uiBoundingCUAddrSlice    = pcSlice->getSliceCurEndCUAddr();
1444      break;
1445#if FIXED_NUMBER_OF_TILES_SLICE_MODE
1446    case AD_HOC_SLICES_FIXED_NUMBER_OF_TILES_IN_SLICE:
1447      tileIdx                = rpcPic->getPicSym()->getTileIdxMap(
1448        rpcPic->getPicSym()->getCUOrderMap(uiStartCUAddrSlice/rpcPic->getNumPartInCU())
1449        );
1450      uiCUAddrIncrement        = 0;
1451      tileTotalCount         = (rpcPic->getPicSym()->getNumColumnsMinus1()+1) * (rpcPic->getPicSym()->getNumRowsMinus1()+1);
1452
1453      for(tileIdxIncrement = 0; tileIdxIncrement < m_pcCfg->getSliceArgument(); tileIdxIncrement++)
1454      {
1455        if((tileIdx + tileIdxIncrement) < tileTotalCount)
1456        {
1457          tileWidthInLcu   = rpcPic->getPicSym()->getTComTile(tileIdx + tileIdxIncrement)->getTileWidth();
1458          tileHeightInLcu  = rpcPic->getPicSym()->getTComTile(tileIdx + tileIdxIncrement)->getTileHeight();
1459          uiCUAddrIncrement += (tileWidthInLcu * tileHeightInLcu * rpcPic->getNumPartInCU()) >> (m_pcCfg->getSliceGranularity() << 1);
1460        }
1461      }
1462
1463      uiBoundingCUAddrSlice    = ((uiStartCUAddrSlice + uiCUAddrIncrement) < uiNumberOfCUsInFrame*rpcPic->getNumPartInCU()) ? (uiStartCUAddrSlice + uiCUAddrIncrement) : uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1464      break;
1465#endif
1466    default:
1467      uiCUAddrIncrement        = rpcPic->getNumCUsInFrame();
1468      uiBoundingCUAddrSlice    = uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1469      break;
1470    } 
1471    pcSlice->setSliceCurEndCUAddr( uiBoundingCUAddrSlice );
1472  }
1473  else
1474  {
1475    UInt uiCUAddrIncrement     ;
1476    switch (m_pcCfg->getSliceMode())
1477    {
1478    case AD_HOC_SLICES_FIXED_NUMBER_OF_LCU_IN_SLICE:
1479      uiCUAddrIncrement        = m_pcCfg->getSliceArgument();
1480      uiBoundingCUAddrSlice    = ((uiStartCUAddrSlice + uiCUAddrIncrement) < uiNumberOfCUsInFrame*rpcPic->getNumPartInCU()) ? (uiStartCUAddrSlice + uiCUAddrIncrement) : uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1481      break;
1482#if FIXED_NUMBER_OF_TILES_SLICE_MODE
1483    case AD_HOC_SLICES_FIXED_NUMBER_OF_TILES_IN_SLICE:
1484      tileIdx                = rpcPic->getPicSym()->getTileIdxMap(
1485        rpcPic->getPicSym()->getCUOrderMap(uiStartCUAddrSlice/rpcPic->getNumPartInCU())
1486        );
1487      uiCUAddrIncrement        = 0;
1488      tileTotalCount         = (rpcPic->getPicSym()->getNumColumnsMinus1()+1) * (rpcPic->getPicSym()->getNumRowsMinus1()+1);
1489
1490      for(tileIdxIncrement = 0; tileIdxIncrement < m_pcCfg->getSliceArgument(); tileIdxIncrement++)
1491      {
1492        if((tileIdx + tileIdxIncrement) < tileTotalCount)
1493        {
1494          tileWidthInLcu   = rpcPic->getPicSym()->getTComTile(tileIdx + tileIdxIncrement)->getTileWidth();
1495          tileHeightInLcu  = rpcPic->getPicSym()->getTComTile(tileIdx + tileIdxIncrement)->getTileHeight();
1496          uiCUAddrIncrement += (tileWidthInLcu * tileHeightInLcu * rpcPic->getNumPartInCU()) >> (m_pcCfg->getSliceGranularity() << 1);
1497        }
1498      }
1499
1500      uiBoundingCUAddrSlice    = ((uiStartCUAddrSlice + uiCUAddrIncrement) < uiNumberOfCUsInFrame*rpcPic->getNumPartInCU()) ? (uiStartCUAddrSlice + uiCUAddrIncrement) : uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1501      break;
1502#endif
1503    default:
1504      uiCUAddrIncrement        = rpcPic->getNumCUsInFrame();
1505      uiBoundingCUAddrSlice    = uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1506      break;
1507    } 
1508    pcSlice->setSliceCurEndCUAddr( uiBoundingCUAddrSlice );
1509  }
1510
1511#if COMPLETE_SLICES_IN_TILE
1512  Bool tileBoundary = false;
1513  if ((m_pcCfg->getSliceMode() == AD_HOC_SLICES_FIXED_NUMBER_OF_LCU_IN_SLICE || m_pcCfg->getSliceMode() == AD_HOC_SLICES_FIXED_NUMBER_OF_BYTES_IN_SLICE) && 
1514      (m_pcCfg->getNumRowsMinus1() > 0 || m_pcCfg->getNumColumnsMinus1() > 0))
1515  {
1516    UInt lcuEncAddr = (uiStartCUAddrSlice+rpcPic->getNumPartInCU()-1)/rpcPic->getNumPartInCU();
1517    UInt lcuAddr = rpcPic->getPicSym()->getCUOrderMap(lcuEncAddr);
1518    UInt startTileIdx = rpcPic->getPicSym()->getTileIdxMap(lcuAddr);
1519    UInt tileBoundingCUAddrSlice = 0;
1520    while (lcuEncAddr < uiNumberOfCUsInFrame && rpcPic->getPicSym()->getTileIdxMap(lcuAddr) == startTileIdx)
1521    {
1522      lcuEncAddr++;
1523      lcuAddr = rpcPic->getPicSym()->getCUOrderMap(lcuEncAddr);
1524    }
1525    tileBoundingCUAddrSlice = lcuEncAddr*rpcPic->getNumPartInCU();
1526   
1527    if (tileBoundingCUAddrSlice < uiBoundingCUAddrSlice)
1528    {
1529      uiBoundingCUAddrSlice = tileBoundingCUAddrSlice;
1530      pcSlice->setSliceCurEndCUAddr( uiBoundingCUAddrSlice );
1531      tileBoundary = true;
1532    }
1533  }
1534#endif
1535
1536  // Entropy slice
1537  UInt uiStartCUAddrEntropySlice, uiBoundingCUAddrEntropySlice;
1538  uiStartCUAddrEntropySlice    = pcSlice->getEntropySliceCurStartCUAddr();
1539  uiBoundingCUAddrEntropySlice = uiNumberOfCUsInFrame;
1540  if (bEncodeSlice) 
1541  {
1542    UInt uiCUAddrIncrement;
1543    switch (m_pcCfg->getEntropySliceMode())
1544    {
1545    case SHARP_FIXED_NUMBER_OF_LCU_IN_ENTROPY_SLICE:
1546      uiCUAddrIncrement               = m_pcCfg->getEntropySliceArgument();
1547      uiBoundingCUAddrEntropySlice    = ((uiStartCUAddrEntropySlice + uiCUAddrIncrement) < uiNumberOfCUsInFrame*rpcPic->getNumPartInCU() ) ? (uiStartCUAddrEntropySlice + uiCUAddrIncrement) : uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1548      break;
1549    case SHARP_MULTIPLE_CONSTRAINT_BASED_ENTROPY_SLICE:
1550      uiCUAddrIncrement               = rpcPic->getNumCUsInFrame();
1551      uiBoundingCUAddrEntropySlice    = pcSlice->getEntropySliceCurEndCUAddr();
1552      break;
1553    default:
1554      uiCUAddrIncrement               = rpcPic->getNumCUsInFrame();
1555      uiBoundingCUAddrEntropySlice    = uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1556      break;
1557    } 
1558    pcSlice->setEntropySliceCurEndCUAddr( uiBoundingCUAddrEntropySlice );
1559  }
1560  else
1561  {
1562    UInt uiCUAddrIncrement;
1563    switch (m_pcCfg->getEntropySliceMode())
1564    {
1565    case SHARP_FIXED_NUMBER_OF_LCU_IN_ENTROPY_SLICE:
1566      uiCUAddrIncrement               = m_pcCfg->getEntropySliceArgument();
1567      uiBoundingCUAddrEntropySlice    = ((uiStartCUAddrEntropySlice + uiCUAddrIncrement) < uiNumberOfCUsInFrame*rpcPic->getNumPartInCU() ) ? (uiStartCUAddrEntropySlice + uiCUAddrIncrement) : uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1568      break;
1569    default:
1570      uiCUAddrIncrement               = rpcPic->getNumCUsInFrame();
1571      uiBoundingCUAddrEntropySlice    = uiNumberOfCUsInFrame*rpcPic->getNumPartInCU();
1572      break;
1573    } 
1574    pcSlice->setEntropySliceCurEndCUAddr( uiBoundingCUAddrEntropySlice );
1575  }
1576  if(uiBoundingCUAddrEntropySlice>uiBoundingCUAddrSlice)
1577  {
1578    uiBoundingCUAddrEntropySlice = uiBoundingCUAddrSlice;
1579    pcSlice->setEntropySliceCurEndCUAddr(uiBoundingCUAddrSlice);
1580  }
1581  //calculate real entropy slice start address
1582  UInt uiInternalAddress = rpcPic->getPicSym()->getPicSCUAddr(pcSlice->getEntropySliceCurStartCUAddr()) % rpcPic->getNumPartInCU();
1583  UInt uiExternalAddress = rpcPic->getPicSym()->getPicSCUAddr(pcSlice->getEntropySliceCurStartCUAddr()) / rpcPic->getNumPartInCU();
1584  UInt uiPosX = ( uiExternalAddress % rpcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
1585  UInt uiPosY = ( uiExternalAddress / rpcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
1586  UInt uiWidth = pcSlice->getSPS()->getPicWidthInLumaSamples();
1587  UInt uiHeight = pcSlice->getSPS()->getPicHeightInLumaSamples();
1588  while((uiPosX>=uiWidth||uiPosY>=uiHeight)&&!(uiPosX>=uiWidth&&uiPosY>=uiHeight))
1589  {
1590    uiInternalAddress++;
1591    if(uiInternalAddress>=rpcPic->getNumPartInCU())
1592    {
1593      uiInternalAddress=0;
1594      uiExternalAddress = rpcPic->getPicSym()->getCUOrderMap(rpcPic->getPicSym()->getInverseCUOrderMap(uiExternalAddress)+1);
1595    }
1596    uiPosX = ( uiExternalAddress % rpcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
1597    uiPosY = ( uiExternalAddress / rpcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
1598  }
1599  UInt uiRealStartAddress = rpcPic->getPicSym()->getPicSCUEncOrder(uiExternalAddress*rpcPic->getNumPartInCU()+uiInternalAddress);
1600 
1601  pcSlice->setEntropySliceCurStartCUAddr(uiRealStartAddress);
1602  uiStartCUAddrEntropySlice=uiRealStartAddress;
1603 
1604  //calculate real slice start address
1605  uiInternalAddress = rpcPic->getPicSym()->getPicSCUAddr(pcSlice->getSliceCurStartCUAddr()) % rpcPic->getNumPartInCU();
1606  uiExternalAddress = rpcPic->getPicSym()->getPicSCUAddr(pcSlice->getSliceCurStartCUAddr()) / rpcPic->getNumPartInCU();
1607  uiPosX = ( uiExternalAddress % rpcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
1608  uiPosY = ( uiExternalAddress / rpcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
1609  uiWidth = pcSlice->getSPS()->getPicWidthInLumaSamples();
1610  uiHeight = pcSlice->getSPS()->getPicHeightInLumaSamples();
1611  while((uiPosX>=uiWidth||uiPosY>=uiHeight)&&!(uiPosX>=uiWidth&&uiPosY>=uiHeight))
1612  {
1613    uiInternalAddress++;
1614    if(uiInternalAddress>=rpcPic->getNumPartInCU())
1615    {
1616      uiInternalAddress=0;
1617      uiExternalAddress = rpcPic->getPicSym()->getCUOrderMap(rpcPic->getPicSym()->getInverseCUOrderMap(uiExternalAddress)+1);
1618    }
1619    uiPosX = ( uiExternalAddress % rpcPic->getFrameWidthInCU() ) * g_uiMaxCUWidth+ g_auiRasterToPelX[ g_auiZscanToRaster[uiInternalAddress] ];
1620    uiPosY = ( uiExternalAddress / rpcPic->getFrameWidthInCU() ) * g_uiMaxCUHeight+ g_auiRasterToPelY[ g_auiZscanToRaster[uiInternalAddress] ];
1621  }
1622  uiRealStartAddress = rpcPic->getPicSym()->getPicSCUEncOrder(uiExternalAddress*rpcPic->getNumPartInCU()+uiInternalAddress);
1623 
1624  pcSlice->setSliceCurStartCUAddr(uiRealStartAddress);
1625  uiStartCUAddrSlice=uiRealStartAddress;
1626 
1627  // Make a joint decision based on reconstruction and entropy slice bounds
1628  uiStartCUAddr    = max(uiStartCUAddrSlice   , uiStartCUAddrEntropySlice   );
1629  uiBoundingCUAddr = min(uiBoundingCUAddrSlice, uiBoundingCUAddrEntropySlice);
1630
1631
1632  if (!bEncodeSlice)
1633  {
1634    // For fixed number of LCU within an entropy and reconstruction slice we already know whether we will encounter end of entropy and/or reconstruction slice
1635    // first. Set the flags accordingly.
1636    if ( (m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_LCU_IN_SLICE && m_pcCfg->getEntropySliceMode()==SHARP_FIXED_NUMBER_OF_LCU_IN_ENTROPY_SLICE)
1637      || (m_pcCfg->getSliceMode()==0 && m_pcCfg->getEntropySliceMode()==SHARP_FIXED_NUMBER_OF_LCU_IN_ENTROPY_SLICE)
1638      || (m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_LCU_IN_SLICE && m_pcCfg->getEntropySliceMode()==0) 
1639#if FIXED_NUMBER_OF_TILES_SLICE_MODE
1640      || (m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_TILES_IN_SLICE && m_pcCfg->getEntropySliceMode()==SHARP_FIXED_NUMBER_OF_LCU_IN_ENTROPY_SLICE)
1641      || (m_pcCfg->getSliceMode()==AD_HOC_SLICES_FIXED_NUMBER_OF_TILES_IN_SLICE && m_pcCfg->getEntropySliceMode()==0) 
1642#endif
1643#if COMPLETE_SLICES_IN_TILE
1644      || tileBoundary
1645#endif
1646)
1647    {
1648      if (uiBoundingCUAddrSlice < uiBoundingCUAddrEntropySlice)
1649      {
1650        pcSlice->setNextSlice       ( true );
1651        pcSlice->setNextEntropySlice( false );
1652      }
1653      else if (uiBoundingCUAddrSlice > uiBoundingCUAddrEntropySlice)
1654      {
1655        pcSlice->setNextSlice       ( false );
1656        pcSlice->setNextEntropySlice( true );
1657      }
1658      else
1659      {
1660        pcSlice->setNextSlice       ( true );
1661        pcSlice->setNextEntropySlice( true );
1662      }
1663    }
1664    else
1665    {
1666      pcSlice->setNextSlice       ( false );
1667      pcSlice->setNextEntropySlice( false );
1668    }
1669  }
1670}
1671//! \}
Note: See TracBrowser for help on using the repository browser.