source: 3DVCSoftware/branches/0.2-poznan-univ/source/Lib/TLibCommon/TComSlice.cpp @ 10

Last change on this file since 10 was 5, checked in by hhi, 13 years ago

Clean version with cfg-files
  • Property svn:eol-style set to native
File size: 24.8 KB
Line 
1/* The copyright in this software is being made available under the BSD
2 * License, included below. This software may be subject to other third party
3 * and contributor rights, including patent rights, and no such rights are
4 * granted under this license.
5 *
6 * Copyright (c) 2010-2011, 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 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
35
36/** \file     TComSlice.cpp
37    \brief    slice header and SPS class
38*/
39
40#include "CommonDef.h"
41#include "TComSlice.h"
42#include "TComPic.h"
43
44TComSlice::TComSlice()
45{
46  m_uiPPSId             = 0;
47  m_iPOC                = 0;
48  m_eSliceType          = I_SLICE;
49  m_iSliceQp            = 0;
50  m_iSymbolMode         = 1;
51  m_aiNumRefIdx[0]      = 0;
52  m_aiNumRefIdx[1]      = 0;
53  m_bLoopFilterDisable  = false;
54 
55  m_iSliceQpDelta       = 0;
56 
57  m_iDepth              = 0;
58 
59  m_pcPic               = NULL;
60  m_bRefenced           = false;
61#ifdef ROUNDING_CONTROL_BIPRED
62  m_bRounding           = false;
63#endif
64  m_uiColDir = 0;
65 
66  m_iViewIdx = 0 ;
67
68  initEqualRef();
69  m_bNoBackPredFlag = false;
70#if MS_LCEC_LOOKUP_TABLE_EXCEPTION
71  m_bRefIdxCombineCoding = false;
72#endif
73#if DCM_COMB_LIST
74  m_bRefPicListCombinationFlag = false;
75  m_bRefPicListModificationFlagLC = false;
76#endif
77  m_uiSliceCurStartCUAddr        = 0;
78  m_uiEntropySliceCurStartCUAddr = 0;
79
80  ::memset( m_aaiCodedScale,  0x00, sizeof( m_aaiCodedScale  ) );
81  ::memset( m_aaiCodedOffset, 0x00, sizeof( m_aaiCodedOffset ) );
82  m_pcTexturePic = NULL;
83#ifdef WEIGHT_PRED
84  resetWpScaling(m_weightPredTable);
85  initWpAcDcParam();
86#endif
87}
88
89TComSlice::~TComSlice()
90{
91}
92
93
94Void TComSlice::initSlice()
95{
96  m_aiNumRefIdx[0]      = 0;
97  m_aiNumRefIdx[1]      = 0;
98 
99  m_uiColDir = 0;
100
101  ::memset( m_apcRefPicList, 0, sizeof (m_apcRefPicList));
102  ::memset( m_aiNumRefIdx,   0, sizeof ( m_aiNumRefIdx ));
103  m_pcTexturePic = NULL;
104 
105  initEqualRef();
106  m_bNoBackPredFlag = false;
107#if MS_LCEC_LOOKUP_TABLE_EXCEPTION
108  m_bRefIdxCombineCoding = false;
109#endif
110#if DCM_COMB_LIST
111  m_bRefPicListCombinationFlag = false;
112  m_bRefPicListModificationFlagLC = false;
113#endif
114}
115
116Void  TComSlice::sortPicList        (TComList<TComPic*>& rcListPic)
117{
118  TComPic*    pcPicExtract;
119  TComPic*    pcPicInsert;
120 
121  TComList<TComPic*>::iterator    iterPicExtract;
122  TComList<TComPic*>::iterator    iterPicExtract_1;
123  TComList<TComPic*>::iterator    iterPicInsert;
124 
125  for (Int i = 1; i < (Int)(rcListPic.size()); i++)
126  {
127    iterPicExtract = rcListPic.begin();
128    for (Int j = 0; j < i; j++) iterPicExtract++;
129    pcPicExtract = *(iterPicExtract);
130    pcPicExtract->setCurrSliceIdx(0);
131   
132    iterPicInsert = rcListPic.begin();
133    while (iterPicInsert != iterPicExtract)
134    {
135      pcPicInsert = *(iterPicInsert);
136      pcPicInsert->setCurrSliceIdx(0);
137      if (pcPicInsert->getPOC() >= pcPicExtract->getPOC())
138      {
139        break;
140      }
141     
142      iterPicInsert++;
143    }
144   
145    iterPicExtract_1 = iterPicExtract;    iterPicExtract_1++;
146   
147    //  swap iterPicExtract and iterPicInsert, iterPicExtract = curr. / iterPicInsert = insertion position
148    rcListPic.insert (iterPicInsert, iterPicExtract, iterPicExtract_1);
149    rcListPic.erase  (iterPicExtract);
150  }
151}
152
153Void TComSlice::setRefPOCList       ()
154{
155  for (Int iDir = 0; iDir < 2; iDir++)
156  {
157    for (Int iNumRefIdx = 0; iNumRefIdx < m_aiNumRefIdx[iDir]; iNumRefIdx++)
158    {
159      m_aiRefPOCList[iDir][iNumRefIdx] = m_apcRefPicList[iDir][iNumRefIdx]->getPOC();
160    }
161  }
162
163}
164
165#if DCM_COMB_LIST
166Void TComSlice::generateCombinedList()
167{
168  if(m_aiNumRefIdx[REF_PIC_LIST_C] > 0)
169  {
170    m_aiNumRefIdx[REF_PIC_LIST_C]=0;
171    for(Int iNumCount = 0; iNumCount < MAX_NUM_REF_LC; iNumCount++)
172    {
173      m_iRefIdxOfLC[REF_PIC_LIST_0][iNumCount]=-1;
174      m_iRefIdxOfLC[REF_PIC_LIST_1][iNumCount]=-1;
175      m_eListIdFromIdxOfLC[iNumCount]=0;
176      m_iRefIdxFromIdxOfLC[iNumCount]=0;
177      m_iRefIdxOfL0FromRefIdxOfL1[iNumCount] = -1;
178      m_iRefIdxOfL1FromRefIdxOfL0[iNumCount] = -1;
179    }
180
181    for (Int iNumRefIdx = 0; iNumRefIdx < MAX_NUM_REF; iNumRefIdx++)
182    {
183      if(iNumRefIdx < m_aiNumRefIdx[REF_PIC_LIST_0]){
184        Bool bTempRefIdxInL2 = true;
185        for ( Int iRefIdxLC = 0; iRefIdxLC < m_aiNumRefIdx[REF_PIC_LIST_C]; iRefIdxLC++ )
186        {
187          if (( m_apcRefPicList[REF_PIC_LIST_0][iNumRefIdx]->getPOC() == m_apcRefPicList[m_eListIdFromIdxOfLC[iRefIdxLC]][m_iRefIdxFromIdxOfLC[iRefIdxLC]]->getPOC() ) &&
188              ( m_apcRefPicList[REF_PIC_LIST_0][iNumRefIdx]->getViewIdx() == m_apcRefPicList[m_eListIdFromIdxOfLC[iRefIdxLC]][m_iRefIdxFromIdxOfLC[iRefIdxLC]]->getViewIdx() ))
189          {
190            m_iRefIdxOfL1FromRefIdxOfL0[iNumRefIdx] = m_iRefIdxFromIdxOfLC[iRefIdxLC];
191            m_iRefIdxOfL0FromRefIdxOfL1[m_iRefIdxFromIdxOfLC[iRefIdxLC]] = iNumRefIdx;
192            bTempRefIdxInL2 = false;
193            assert(m_eListIdFromIdxOfLC[iRefIdxLC]==REF_PIC_LIST_1);
194            break;
195          }
196        }
197
198        if(bTempRefIdxInL2 == true)
199        { 
200          m_eListIdFromIdxOfLC[m_aiNumRefIdx[REF_PIC_LIST_C]] = REF_PIC_LIST_0;
201          m_iRefIdxFromIdxOfLC[m_aiNumRefIdx[REF_PIC_LIST_C]] = iNumRefIdx;
202          m_iRefIdxOfLC[REF_PIC_LIST_0][iNumRefIdx] = m_aiNumRefIdx[REF_PIC_LIST_C]++;
203        }
204      }
205
206      if(iNumRefIdx < m_aiNumRefIdx[REF_PIC_LIST_1]){
207        Bool bTempRefIdxInL2 = true;
208        for ( Int iRefIdxLC = 0; iRefIdxLC < m_aiNumRefIdx[REF_PIC_LIST_C]; iRefIdxLC++ )
209        {
210          if (( m_apcRefPicList[REF_PIC_LIST_1][iNumRefIdx]->getPOC() == m_apcRefPicList[m_eListIdFromIdxOfLC[iRefIdxLC]][m_iRefIdxFromIdxOfLC[iRefIdxLC]]->getPOC() ) &&
211              ( m_apcRefPicList[REF_PIC_LIST_1][iNumRefIdx]->getViewIdx() == m_apcRefPicList[m_eListIdFromIdxOfLC[iRefIdxLC]][m_iRefIdxFromIdxOfLC[iRefIdxLC]]->getViewIdx() ))
212          {
213            m_iRefIdxOfL0FromRefIdxOfL1[iNumRefIdx] = m_iRefIdxFromIdxOfLC[iRefIdxLC];
214            m_iRefIdxOfL1FromRefIdxOfL0[m_iRefIdxFromIdxOfLC[iRefIdxLC]] = iNumRefIdx;
215            bTempRefIdxInL2 = false;
216            assert(m_eListIdFromIdxOfLC[iRefIdxLC]==REF_PIC_LIST_0);
217            break;
218          }
219        }
220        if(bTempRefIdxInL2 == true)
221        {
222          m_eListIdFromIdxOfLC[m_aiNumRefIdx[REF_PIC_LIST_C]] = REF_PIC_LIST_1;
223          m_iRefIdxFromIdxOfLC[m_aiNumRefIdx[REF_PIC_LIST_C]] = iNumRefIdx;
224          m_iRefIdxOfLC[REF_PIC_LIST_1][iNumRefIdx] = m_aiNumRefIdx[REF_PIC_LIST_C]++;
225        }
226      }
227    }
228  }
229}
230#endif
231
232Void TComSlice::setRefPicListFromGOPSTring( TComList<TComPic*>& rcListPic, std::vector<TComPic*>& rapcSpatRefPics )
233{
234
235  if (m_eSliceType == I_SLICE)
236  {
237    ::memset( m_apcRefPicList, 0, sizeof (m_apcRefPicList));
238    ::memset( m_aiNumRefIdx,   0, sizeof ( m_aiNumRefIdx ));
239
240    return;
241  }
242
243  sortPicList(rcListPic);
244  assert( m_eSliceType != P_SLICE || m_pcPic->getNumRefs( REF_PIC_LIST_1 ) == 0 );
245  for(Int iRefList = 0; iRefList<2; iRefList++ )
246  {
247    RefPicList eRefList = (RefPicList) iRefList;
248    m_aiNumRefIdx[eRefList] = m_pcPic->getNumRefs( eRefList );
249    for( Int i =0; i<m_pcPic->getNumRefs(eRefList); i++ )
250    {
251      const int iRefPoc = m_pcPic->getRefPOC( eRefList, i );
252      const int iRefViewIdx = m_pcPic->getRefViewIdx( eRefList, i );
253      m_aiRefPOCList[eRefList][i] = iRefPoc;
254      m_aiRefViewList[eRefList][i] = iRefViewIdx;
255
256      TComPic* pcRefPic = NULL;
257      if( iRefViewIdx == m_iViewIdx )
258      {
259        // temporal prediction from current view
260        for( TComList<TComPic*>::iterator it = rcListPic.begin(); it!=rcListPic.end(); it++)
261        {
262          if((*it)->getPOC() == iRefPoc)
263          {
264            pcRefPic = *it;
265            break;
266          }
267        }
268        assert( pcRefPic );
269        if( pcRefPic == NULL && iRefPoc > m_iPOC )
270        {
271          const int iAltRefPoc = m_iPOC-(iRefPoc-m_iPOC) ;
272          m_aiRefPOCList[eRefList][i] = iAltRefPoc ;
273          for( TComList<TComPic*>::iterator it = rcListPic.begin(); it!=rcListPic.end(); it++)
274          {
275            if((*it)->getPOC() == iAltRefPoc)
276            {
277              pcRefPic = *it ;
278              break;
279            }
280          }
281        }
282      }
283      else
284      {
285        // inter-view prediction
286        assert( iRefPoc == m_iPOC );
287        assert( 0 <= iRefViewIdx && iRefViewIdx < rapcSpatRefPics.size() );
288        pcRefPic = rapcSpatRefPics[iRefViewIdx];
289      }
290      if( pcRefPic )
291      {
292        m_apcRefPicList[eRefList][i] = pcRefPic;
293        pcRefPic->getPicYuvRec()->extendPicBorder();
294      }
295      else
296      {
297        printf("\n inconsistence in gop string!") ; // gop string inconsistent
298        assert(0) ;
299      }
300    }
301  }
302}
303Void TComSlice::setRefPicListExplicitlyDecoderSided( TComList<TComPic*>& rcListPic, std::vector<TComPic*>& rapcSpatRefPics )
304{
305  sortPicList(rcListPic);
306  TComPic*  pcRefPic = NULL;
307
308  for(Int iRefList = 0; iRefList<2; iRefList++)
309  {
310    RefPicList eRefList = (RefPicList) iRefList;
311    for( Int i=0; i<m_aiNumRefIdx[eRefList]; i++ )
312    {
313      const int iRefPoc = m_aiRefPOCList[ eRefList][i ];
314      const int iRefViewIdx = m_aiRefViewList[ eRefList][i ];
315      if( iRefViewIdx == m_iViewIdx )
316      {
317        for( TComList<TComPic*>::iterator it = rcListPic.begin(); it!=rcListPic.end(); it++)
318        {
319          if((*it)->getPOC() == iRefPoc)
320          {
321            pcRefPic = *it;
322            break;
323          }
324        }
325      }
326      else
327      {
328        // inter-view prediction
329        assert( iRefPoc == m_iPOC );
330        assert( 0 <= iRefViewIdx && iRefViewIdx < rapcSpatRefPics.size() );
331        pcRefPic = rapcSpatRefPics[iRefViewIdx];
332      }
333      if( pcRefPic )
334      {
335        m_apcRefPicList[eRefList][i] = pcRefPic;
336        pcRefPic->getPicYuvRec()->extendPicBorder();
337      }
338      else
339      {
340        printf("\n inconsistence in gop string!") ; // gop string inconsistent
341        assert(0) ;
342      }
343    }
344  }
345}
346
347Void TComSlice::initEqualRef()
348{
349  for (Int iDir = 0; iDir < 2; iDir++)
350  {
351    for (Int iRefIdx1 = 0; iRefIdx1 < MAX_NUM_REF; iRefIdx1++)
352    {
353      for (Int iRefIdx2 = iRefIdx1; iRefIdx2 < MAX_NUM_REF; iRefIdx2++)
354      {
355        m_abEqualRef[iDir][iRefIdx1][iRefIdx2] = m_abEqualRef[iDir][iRefIdx2][iRefIdx1] = (iRefIdx1 == iRefIdx2? true : false);
356      }
357    }
358  }
359}
360
361Void TComSlice::initMultiviewSlice( Int** aaiScale, Int** aaiOffset )
362{
363  if( m_pcSPS->hasCamParInSliceHeader() )
364  {
365    UInt uiViewId = m_pcSPS->getViewId();
366    for( UInt uiBaseViewId = 0; uiBaseViewId < uiViewId; uiBaseViewId++ )
367    {
368      m_aaiCodedScale [ 0 ][ uiBaseViewId ] = aaiScale [ uiBaseViewId ][     uiViewId ];
369      m_aaiCodedScale [ 1 ][ uiBaseViewId ] = aaiScale [     uiViewId ][ uiBaseViewId ];
370      m_aaiCodedOffset[ 0 ][ uiBaseViewId ] = aaiOffset[ uiBaseViewId ][     uiViewId ];
371      m_aaiCodedOffset[ 1 ][ uiBaseViewId ] = aaiOffset[     uiViewId ][ uiBaseViewId ];
372    }
373  }
374}
375
376
377#if DCM_DECODING_REFRESH
378/** Function for marking the reference pictures when an IDR and CDR is encountered.
379 * \param uiPOCCDR POC of the CDR picture
380 * \param bRefreshPending flag indicating if a deferred decoding refresh is pending
381 * \param rcListPic reference to the reference picture list
382 * This function marks the reference pictures as "unused for reference" in the following conditions.
383 * If the nal_unit_type is IDR all pictures in the reference picture list 
384 * is marked as "unused for reference"
385 * Otherwise do for the CDR case (non CDR case has no effect since both if conditions below will not be true)
386 *    If the bRefreshPending flag is true (a deferred decoding refresh is pending) and the current
387 *    temporal reference is greater than the temporal reference of the latest CDR picture (uiPOCCDR),
388 *    mark all reference pictures except the latest CDR picture as "unused for reference" and set
389 *    the bRefreshPending flag to false.
390 *    If the nal_unit_type is CDR, set the bRefreshPending flag to true and iPOCCDR to the temporal
391 *    reference of the current picture.
392 * Note that the current picture is already placed in the reference list and its marking is not changed.
393 * If the current picture has a nal_ref_idc that is not 0, it will remain marked as "used for reference".
394 */
395Void TComSlice::decodingRefreshMarking(UInt& uiPOCCDR, Bool& bRefreshPending, TComList<TComPic*>& rcListPic)
396{
397  TComPic*                 rpcPic;
398  UInt uiPOCCurr = getPOC(); 
399
400  if (getNalUnitType() == NAL_UNIT_CODED_SLICE_IDR)  // IDR
401  {
402    // mark all pictures as not used for reference
403    TComList<TComPic*>::iterator        iterPic       = rcListPic.begin();
404    while (iterPic != rcListPic.end())
405    {
406      rpcPic = *(iterPic);
407      rpcPic->setCurrSliceIdx(0);
408      if (rpcPic->getPOC() != uiPOCCurr) rpcPic->getSlice(0)->setReferenced(false);
409      iterPic++;
410    }
411  }
412  else // CDR or No DR
413  {
414    if (bRefreshPending==true && uiPOCCurr > uiPOCCDR) // CDR reference marking pending
415    {
416      TComList<TComPic*>::iterator        iterPic       = rcListPic.begin();
417      while (iterPic != rcListPic.end())
418      {
419        rpcPic = *(iterPic);
420        if (rpcPic->getPOC() != uiPOCCurr && rpcPic->getPOC() != uiPOCCDR) rpcPic->getSlice(0)->setReferenced(false);
421        iterPic++;
422      }
423      bRefreshPending = false; 
424    }
425    if (getNalUnitType() == NAL_UNIT_CODED_SLICE_CDR) // CDR picture found
426    {
427      bRefreshPending = true; 
428      uiPOCCDR = uiPOCCurr;
429    }
430  }
431}
432#endif
433
434Void TComSlice::copySliceInfo(TComSlice *pSrc)
435{
436  assert( pSrc != NULL );
437
438  Int i, j, k;
439
440  m_iPOC                 = pSrc->m_iPOC;
441  m_iViewIdx             = pSrc->m_iViewIdx;
442#if DCM_DECODING_REFRESH
443  m_eNalUnitType         = pSrc->m_eNalUnitType;
444#endif 
445  m_eSliceType           = pSrc->m_eSliceType;
446  m_iSliceQp             = pSrc->m_iSliceQp;
447  m_iSymbolMode          = pSrc->m_iSymbolMode;
448  m_bLoopFilterDisable   = pSrc->m_bLoopFilterDisable;
449 
450#if DCM_COMB_LIST 
451  for (i = 0; i < 3; i++)
452  {
453    m_aiNumRefIdx[i]     = pSrc->m_aiNumRefIdx[i];
454  }
455
456  for (i = 0; i < 2; i++)
457  {
458    for (j = 0; j < MAX_NUM_REF_LC; j++)
459    {
460       m_iRefIdxOfLC[i][j]  = pSrc->m_iRefIdxOfLC[i][j];
461    }
462  }
463  for (i = 0; i < MAX_NUM_REF_LC; i++)
464  {
465    m_eListIdFromIdxOfLC[i] = pSrc->m_eListIdFromIdxOfLC[i];
466    m_iRefIdxFromIdxOfLC[i] = pSrc->m_iRefIdxFromIdxOfLC[i];
467    m_iRefIdxOfL1FromRefIdxOfL0[i] = pSrc->m_iRefIdxOfL1FromRefIdxOfL0[i];
468    m_iRefIdxOfL0FromRefIdxOfL1[i] = pSrc->m_iRefIdxOfL0FromRefIdxOfL1[i];
469  }
470  m_bRefPicListModificationFlagLC = pSrc->m_bRefPicListModificationFlagLC;
471  m_bRefPicListCombinationFlag    = pSrc->m_bRefPicListCombinationFlag;
472#else
473  for (i = 0; i < 2; i++)
474  {
475    m_aiNumRefIdx[i]     = pSrc->m_aiNumRefIdx[i];
476  }
477#endif 
478
479  m_iSliceQpDelta        = pSrc->m_iSliceQpDelta;
480  for (i = 0; i < 2; i++)
481  {
482    for (j = 0; j < MAX_NUM_REF; j++)
483    {
484      m_apcRefPicList[i][j]  = pSrc->m_apcRefPicList[i][j];
485      m_aiRefPOCList[i][j]   = pSrc->m_aiRefPOCList[i][j];
486      m_aiRefViewList[i][j]  = pSrc->m_aiRefViewList[i][j];
487    }
488  } 
489  m_iDepth               = pSrc->m_iDepth;
490
491  // referenced slice
492  m_bRefenced            = pSrc->m_bRefenced;
493#ifdef ROUNDING_CONTROL_BIPRED
494  m_bRounding            = pSrc->m_bRounding;
495#endif
496
497  // access channel
498  m_pcSPS                = pSrc->m_pcSPS;
499  m_pcPPS                = pSrc->m_pcPPS;
500  m_uiPPSId              = pSrc->m_uiPPSId;
501  m_pcPic                = pSrc->m_pcPic;
502
503  m_uiColDir             = pSrc->m_uiColDir;
504  m_dLambda              = pSrc->m_dLambda;
505  for (i = 0; i < 2; i++)
506  {
507    for (j = 0; j < MAX_NUM_REF; j++)
508    {
509      for (k =0; k < MAX_NUM_REF; k++)
510      {
511        m_abEqualRef[i][j][k] = pSrc->m_abEqualRef[i][j][k];
512      }
513    }
514  }
515
516  m_bNoBackPredFlag      = pSrc->m_bNoBackPredFlag;
517#if MS_LCEC_LOOKUP_TABLE_EXCEPTION
518  m_bRefIdxCombineCoding = pSrc->m_bRefIdxCombineCoding;
519#endif
520  m_uiSliceMode                   = pSrc->m_uiSliceMode;
521  m_uiSliceArgument               = pSrc->m_uiSliceArgument;
522  m_uiSliceCurStartCUAddr         = pSrc->m_uiSliceCurStartCUAddr;
523  m_uiSliceCurEndCUAddr           = pSrc->m_uiSliceCurEndCUAddr;
524  m_uiSliceIdx                    = pSrc->m_uiSliceIdx;
525  m_uiEntropySliceMode            = pSrc->m_uiEntropySliceMode;
526  m_uiEntropySliceArgument        = pSrc->m_uiEntropySliceArgument; 
527  m_uiEntropySliceCurStartCUAddr  = pSrc->m_uiEntropySliceCurStartCUAddr;
528  m_uiEntropySliceCurEndCUAddr    = pSrc->m_uiEntropySliceCurEndCUAddr;
529  m_bNextSlice                    = pSrc->m_bNextSlice;
530  m_bNextEntropySlice             = pSrc->m_bNextEntropySlice;
531#ifdef WEIGHT_PRED
532  for ( int e=0 ; e<2 ; e++ )
533  for ( int n=0 ; n<MAX_NUM_REF ; n++ )
534    memcpy(m_weightPredTable[e][n], pSrc->m_weightPredTable[e][n], sizeof(wpScalingParam)*3 );
535#endif
536}
537
538#ifdef WEIGHT_PRED
539Void  TComSlice::getWpScaling( RefPicList e, Int iRefIdx, wpScalingParam *&wp )
540{
541  wp = m_weightPredTable[e][iRefIdx];
542}
543
544Void  TComSlice::displayWpScaling()
545{
546  Bool  bFound = false;
547  for ( int e=0 ; e<2 ; e++ ) {
548    for ( int i=0 ; i<MAX_NUM_REF ; i++ )
549    for ( int yuv=0 ; yuv<3 ; yuv++ )
550      if ( m_weightPredTable[e][i][yuv].bPresentFlag ) {
551        if ( !bFound ) {
552          printf("\tluma_log2_weight_denom = %d\n",   m_weightPredTable[0][0][0].uiLog2WeightDenom);
553          printf("\tchroma_log2_weight_denom = %d\n", m_weightPredTable[0][0][1].uiLog2WeightDenom);
554          bFound = true;
555        }
556        Double  weight = (Double)m_weightPredTable[e][i][yuv].iWeight / (Double)(1<<m_weightPredTable[0][0][0].uiLog2WeightDenom);
557        if ( yuv == 0 ) {
558          printf("\tluma_weight_l%d_flag = 1\n", e);
559          printf("\t luma_weight_l%d[%d] = %d => w = %g\n", e, i, m_weightPredTable[e][i][yuv].iWeight, weight);
560          printf("\t luma_offset_l%d[%d] = %d\n", e, i, m_weightPredTable[e][i][yuv].iOffset);
561        }
562        else {
563          if ( yuv == 1 ) printf("\tchroma_weight_l%d_flag = 1\n", e);
564          printf("\t chroma_weight_l%d[%d][%d] = %d => w = %g\n", e, i, yuv-1, m_weightPredTable[e][i][yuv].iWeight, weight);
565          printf("\t chroma_offset_l%d[%d][%d] = %d\n", e, i, yuv-1, m_weightPredTable[e][i][yuv].iOffset);
566        }
567      }
568  }
569}
570
571// Default WP values settings : no weight.
572Void  TComSlice::resetWpScaling(wpScalingParam  wp[2][MAX_NUM_REF][3])
573{
574  for ( int e=0 ; e<2 ; e++ ) {
575    for ( int i=0 ; i<MAX_NUM_REF ; i++ )
576      for ( int yuv=0 ; yuv<3 ; yuv++ ) {
577        wpScalingParam  *pwp = &(wp[e][i][yuv]);
578        pwp->bPresentFlag      = false;
579        pwp->uiLog2WeightDenom = 0;
580        pwp->uiLog2WeightDenom = 0;
581        pwp->iWeight           = 1;
582        pwp->iOffset           = 0;
583      }
584  }
585}
586
587Void  TComSlice::initWpScaling()
588{
589  initWpScaling(m_weightPredTable);
590}
591
592Void  TComSlice::initWpScaling(wpScalingParam  wp[2][MAX_NUM_REF][3])
593{
594  for ( int e=0 ; e<2 ; e++ ) {
595    for ( int i=0 ; i<MAX_NUM_REF ; i++ )
596      for ( int yuv=0 ; yuv<3 ; yuv++ ) {
597        wpScalingParam  *pwp = &(wp[e][i][yuv]);
598        if ( !pwp->bPresentFlag ) {
599          // Inferring values not present :
600          pwp->iWeight = (1 << pwp->uiLog2WeightDenom);
601          pwp->iOffset = 0;
602        }
603
604        pwp->w      = pwp->iWeight;
605        pwp->o      = pwp->iOffset * (1 << (g_uiBitDepth-8));
606        pwp->shift  = pwp->uiLog2WeightDenom;
607        pwp->round  = (pwp->uiLog2WeightDenom>=1) ? (1 << (pwp->uiLog2WeightDenom-1)) : (0);
608      }
609  }
610}
611#endif
612
613#ifdef WEIGHT_PRED
614Void  TComSlice::getWpAcDcParam(wpACDCParam *&wp)
615{
616  wp = m_weightACDCParam;
617}
618
619Void  TComSlice::initWpAcDcParam()
620{
621  for(Int iComp = 0; iComp < 3; iComp++ )
622  {
623    m_weightACDCParam[iComp].iAC = 0;
624    m_weightACDCParam[iComp].iDC = 0;
625  }
626}
627#endif
628
629// ------------------------------------------------------------------------------------------------
630// Sequence parameter set (SPS)
631// ------------------------------------------------------------------------------------------------
632
633TComSPS::TComSPS()
634{
635  // Structure
636  m_uiSPSId       = 0;
637  m_uiWidth       = 352;
638  m_uiHeight      = 288;
639  m_uiMaxCUWidth  = 32;
640  m_uiMaxCUHeight = 32;
641  m_uiMaxCUDepth  = 3;
642  m_uiMinTrDepth  = 0;
643  m_uiMaxTrDepth  = 1;
644  m_uiMaxTrSize   = 32;
645 
646  // Tool list
647  m_bUseALF       = false;
648  m_bUseDQP       = false;
649 
650  m_bUseMRG      = false; // SOPH:
651#if HHI_MPI
652  m_bUseMVI = false;
653#endif
654 
655  m_uiViewId              = 0;
656  m_iViewOrderIdx         = 0;
657  m_bDepth                = false;
658  m_uiCamParPrecision     = 0;
659  m_bCamParInSliceHeader  = false;
660  ::memset( m_aaiCodedScale,  0x00, sizeof( m_aaiCodedScale  ) );
661  ::memset( m_aaiCodedOffset, 0x00, sizeof( m_aaiCodedOffset ) );
662
663#if DEPTH_MAP_GENERATION
664  m_uiPredDepthMapGeneration = 0;
665  m_uiPdmPrecision           = 0;
666  ::memset( m_aiPdmScaleNomDelta, 0x00, sizeof( m_aiPdmScaleNomDelta  ) );
667  ::memset( m_aiPdmOffset,        0x00, sizeof( m_aiPdmOffset         ) );
668#endif
669#if HHI_INTER_VIEW_MOTION_PRED
670  m_uiMultiviewMvPredMode    = 0;
671#endif
672#if HHI_INTER_VIEW_RESIDUAL_PRED
673  m_uiMultiviewResPredMode   = 0;
674#endif
675
676  // AMVP parameter
677  ::memset( m_aeAMVPMode, 0, sizeof( m_aeAMVPMode ) );
678  m_bUseDMM = false;
679}
680
681TComSPS::~TComSPS()
682{
683}
684
685TComPPS::TComPPS()
686{
687#if CONSTRAINED_INTRA_PRED
688  m_bConstrainedIntraPred = false;
689#endif
690  m_uiSPSId = 0;
691  m_uiPPSId = 0;
692}
693
694TComPPS::~TComPPS()
695{
696}
697
698Void
699TComSPS::initMultiviewSPS( UInt uiViewId, Int iViewOrderIdx, UInt uiCamParPrecision, Bool bCamParSlice, Int** aaiScale, Int** aaiOffset )
700{
701  AOT( uiViewId == 0 && iViewOrderIdx != 0 );
702  AOT( uiViewId != 0 && iViewOrderIdx == 0 );
703  AOT( uiViewId != 0 && !bCamParSlice && ( aaiScale == 0 || aaiOffset == 0 ) );
704
705  m_uiViewId              = uiViewId;
706  m_iViewOrderIdx         = iViewOrderIdx;
707  m_bDepth                = false;
708  m_uiCamParPrecision     = ( m_uiViewId ? uiCamParPrecision : 0 );
709  m_bCamParInSliceHeader  = ( m_uiViewId ? bCamParSlice  : false );
710  ::memset( m_aaiCodedScale,  0x00, sizeof( m_aaiCodedScale  ) );
711  ::memset( m_aaiCodedOffset, 0x00, sizeof( m_aaiCodedOffset ) );
712  if( !m_bCamParInSliceHeader )
713  {
714    for( UInt uiBaseViewId = 0; uiBaseViewId < m_uiViewId; uiBaseViewId++ )
715    {
716      m_aaiCodedScale [ 0 ][ uiBaseViewId ] = aaiScale [ uiBaseViewId ][   m_uiViewId ];
717      m_aaiCodedScale [ 1 ][ uiBaseViewId ] = aaiScale [   m_uiViewId ][ uiBaseViewId ];
718      m_aaiCodedOffset[ 0 ][ uiBaseViewId ] = aaiOffset[ uiBaseViewId ][   m_uiViewId ];
719      m_aaiCodedOffset[ 1 ][ uiBaseViewId ] = aaiOffset[   m_uiViewId ][ uiBaseViewId ];
720    }
721  }
722}
723
724Void
725TComSPS::initMultiviewSPSDepth( UInt uiViewId, Int iViewOrderIdx )
726{
727  AOT( uiViewId == 0 && iViewOrderIdx != 0 );
728  AOT( uiViewId != 0 && iViewOrderIdx == 0 );
729
730  m_uiViewId              = uiViewId;
731  m_iViewOrderIdx         = iViewOrderIdx;
732  m_bDepth                = true;
733  m_uiCamParPrecision     = 0;
734  m_bCamParInSliceHeader  = false;
735  ::memset( m_aaiCodedScale,  0x00, sizeof( m_aaiCodedScale  ) );
736  ::memset( m_aaiCodedOffset, 0x00, sizeof( m_aaiCodedOffset ) );
737}
738
739
740#if DEPTH_MAP_GENERATION
741Void
742TComSPS::setPredDepthMapGeneration( UInt uiViewId, Bool bIsDepth, UInt uiPdmGenMode, UInt uiPdmMvPredMode, UInt uiPdmPrec, Int** aaiPdmScaleNomDelta, Int** aaiPdmOffset )
743{ 
744  AOF( m_uiViewId == uiViewId );
745  AOF( m_bDepth   == bIsDepth );
746  AOT( ( uiViewId == 0 || bIsDepth ) && uiPdmGenMode );
747  AOT( uiPdmGenMode && ( aaiPdmScaleNomDelta == 0 || aaiPdmOffset == 0 ) );
748  AOT( uiPdmMvPredMode && uiPdmGenMode == 0 );
749 
750  m_uiPredDepthMapGeneration = uiPdmGenMode;
751#if HHI_INTER_VIEW_MOTION_PRED
752  m_uiMultiviewMvPredMode    = uiPdmMvPredMode;
753#endif
754  m_uiPdmPrecision           = ( m_uiPredDepthMapGeneration ? uiPdmPrec : 0 );
755  ::memset( m_aiPdmScaleNomDelta, 0x00, sizeof( m_aiPdmScaleNomDelta  ) );
756  ::memset( m_aiPdmOffset,        0x00, sizeof( m_aiPdmOffset         ) );
757  if( m_uiPredDepthMapGeneration )
758  {
759    for( UInt uiBaseId = 0; uiBaseId < m_uiViewId; uiBaseId++ )
760    {
761      m_aiPdmScaleNomDelta[ uiBaseId ]  = aaiPdmScaleNomDelta[ m_uiViewId ][ uiBaseId ];
762      m_aiPdmOffset       [ uiBaseId ]  = aaiPdmOffset       [ m_uiViewId ][ uiBaseId ];
763    }
764  }
765}
766#endif
Note: See TracBrowser for help on using the repository browser.