source: 3DVCSoftware/branches/0.3-poznan-univ/source/Lib/TLibCommon/TComSlice.h @ 46

Last change on this file since 46 was 46, checked in by poznan-univ, 12 years ago

Added Ericsson Extraction Tool

  • Property svn:eol-style set to native
File size: 30.2 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.h
37    \brief    slice header and SPS class (header)
38*/
39
40#ifndef __TCOMSLICE__
41#define __TCOMSLICE__
42
43
44#include "CommonDef.h"
45#ifdef WEIGHT_PRED
46  #include <string.h>   // To avoid compilation failure with some g++ linux compilators...
47#endif
48#include "TComList.h"
49#include <math.h>
50
51class TComPic;
52#if DEPTH_MAP_GENERATION
53class TComDepthMapGenerator;
54#endif
55#if HHI_INTER_VIEW_RESIDUAL_PRED
56class TComResidualGenerator;
57#endif
58
59#if POZNAN_MP
60//#include "../TLibCommon/TComMP.h"
61class TComMP;
62#endif
63
64// ====================================================================================================================
65// Class definition
66// ====================================================================================================================
67
68/// SPS class
69class TComSPS
70{
71private:
72  // Structure
73  UInt        m_uiSPSId;
74  UInt        m_uiWidth;
75  UInt        m_uiHeight;
76  Int         m_aiPad[2];
77  UInt        m_uiMaxCUWidth;
78  UInt        m_uiMaxCUHeight;
79  UInt        m_uiMaxCUDepth;
80  UInt        m_uiMinTrDepth;
81  UInt        m_uiMaxTrDepth;
82
83#if BITSTREAM_EXTRACTION
84  UInt        m_uiLayerId;
85#endif
86  UInt        m_uiViewId;
87  Int         m_iViewOrderIdx;
88  Bool        m_bDepth;
89  UInt        m_uiCamParPrecision;
90  Bool        m_bCamParInSliceHeader;
91  Int         m_aaiCodedScale [2][MAX_NUMBER_VIEWS];
92  Int         m_aaiCodedOffset[2][MAX_NUMBER_VIEWS];
93
94  // Tool list
95  UInt        m_uiQuadtreeTULog2MaxSize;
96  UInt        m_uiQuadtreeTULog2MinSize;
97  UInt        m_uiQuadtreeTUMaxDepthInter;
98  UInt        m_uiQuadtreeTUMaxDepthIntra;
99  Bool        m_bUseALF;
100  Bool        m_bUseDQP;
101#if !HHI_NO_LowDelayCoding
102  Bool        m_bUseLDC;
103#endif
104  Bool        m_bUsePAD;
105  Bool        m_bUseMRG; // SOPH:
106
107#if LM_CHROMA
108  Bool        m_bUseLMChroma; // JL:
109#endif
110
111#if DCM_COMB_LIST
112  Bool        m_bUseLComb;
113  Bool        m_bLCMod;
114#endif
115
116#if HHI_RMP_SWITCH
117  Bool        m_bUseRMP;
118#endif
119  // Parameter
120  AMVP_MODE   m_aeAMVPMode[MAX_CU_DEPTH];
121  UInt        m_uiBitDepth;
122  UInt        m_uiBitIncrement;
123
124  // Max physical transform size
125  UInt        m_uiMaxTrSize;
126
127  Int m_iAMPAcc[MAX_CU_DEPTH];
128
129#if MTK_NONCROSS_INLOOP_FILTER
130  Bool        m_bLFCrossSliceBoundaryFlag;
131#endif
132#if MTK_SAO
133  Bool        m_bUseSAO;
134#endif
135#if HHI_MPI
136  Bool        m_bUseMVI;
137#endif
138
139  UInt m_uiCodedPictureBufferSize ;
140
141#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
142  Bool  m_bUseDMM;
143#endif
144#if HHI_DMM_PRED_TEX && FLEX_CODING_ORDER
145  Bool  m_bUseDMM34;
146#endif
147
148#if DEPTH_MAP_GENERATION
149  UInt  m_uiPredDepthMapGeneration;
150  UInt  m_uiPdmPrecision;
151  Int   m_aiPdmScaleNomDelta[MAX_NUMBER_VIEWS];
152  Int   m_aiPdmOffset       [MAX_NUMBER_VIEWS];
153#endif
154#if HHI_INTER_VIEW_MOTION_PRED
155  UInt  m_uiMultiviewMvPredMode;
156#endif
157#if HHI_INTER_VIEW_RESIDUAL_PRED
158  UInt  m_uiMultiviewResPredMode;
159#endif
160
161#if POZNAN_DBMP
162  UInt  m_uiDBMP;
163#endif
164
165#if POZNAN_ENCODE_ONLY_DISOCCLUDED_CU
166  UInt  m_uiUseCUSkip;
167#endif
168
169#if DEPTH_MAP_GENERATION
170  TComDepthMapGenerator* m_pcDepthMapGenerator;
171#endif
172#if HHI_INTER_VIEW_RESIDUAL_PRED
173  TComResidualGenerator* m_pcResidualGenerator;
174#endif
175
176#if POZNAN_NONLINEAR_DEPTH
177  TComNonlinearDepthModel m_cNonlinearDepthModel;   
178  Bool  m_bUseNonlinearDepth;
179#endif
180
181#if POZNAN_MP
182  TComMP* m_pcMP;
183#endif
184#if POZNAN_TEXTURE_TU_DELTA_QP_ACCORDING_TO_DEPTH
185  Bool   m_bUseTexDqpAccordingToDepth;
186#endif
187
188public:
189  TComSPS();
190  virtual ~TComSPS();
191
192  // structure
193  Void setSPSId       ( UInt u ) { m_uiSPSId = u;           }
194  UInt getSPSId       ()         { return m_uiSPSId;        }
195  Void setWidth       ( UInt u ) { m_uiWidth = u;           }
196  UInt getWidth       ()         { return  m_uiWidth;       }
197  Void setHeight      ( UInt u ) { m_uiHeight = u;          }
198  UInt getHeight      ()         { return  m_uiHeight;      }
199  Void setMaxCUWidth  ( UInt u ) { m_uiMaxCUWidth = u;      }
200  UInt getMaxCUWidth  ()         { return  m_uiMaxCUWidth;  }
201  Void setMaxCUHeight ( UInt u ) { m_uiMaxCUHeight = u;     }
202  UInt getMaxCUHeight ()         { return  m_uiMaxCUHeight; }
203  Void setMaxCUDepth  ( UInt u ) { m_uiMaxCUDepth = u;      }
204  UInt getMaxCUDepth  ()         { return  m_uiMaxCUDepth;  }
205  Void setMinTrDepth  ( UInt u ) { m_uiMinTrDepth = u;      }
206  UInt getMinTrDepth  ()         { return  m_uiMinTrDepth;  }
207  Void setMaxTrDepth  ( UInt u ) { m_uiMaxTrDepth = u;      }
208  UInt getMaxTrDepth  ()         { return  m_uiMaxTrDepth;  }
209  Void setQuadtreeTULog2MaxSize( UInt u ) { m_uiQuadtreeTULog2MaxSize = u;    }
210  UInt getQuadtreeTULog2MaxSize()         { return m_uiQuadtreeTULog2MaxSize; }
211  Void setQuadtreeTULog2MinSize( UInt u ) { m_uiQuadtreeTULog2MinSize = u;    }
212  UInt getQuadtreeTULog2MinSize()         { return m_uiQuadtreeTULog2MinSize; }
213  Void setQuadtreeTUMaxDepthInter( UInt u ) { m_uiQuadtreeTUMaxDepthInter = u;    }
214  Void setQuadtreeTUMaxDepthIntra( UInt u ) { m_uiQuadtreeTUMaxDepthIntra = u;    }
215  UInt getQuadtreeTUMaxDepthInter()         { return m_uiQuadtreeTUMaxDepthInter; }
216  UInt getQuadtreeTUMaxDepthIntra()         { return m_uiQuadtreeTUMaxDepthIntra; }
217  Void setPad         (Int iPad[2]) { m_aiPad[0] = iPad[0]; m_aiPad[1] = iPad[1]; }
218  Void setPadX        ( Int  u ) { m_aiPad[0] = u; }
219  Void setPadY        ( Int  u ) { m_aiPad[1] = u; }
220  Int  getPad         ( Int  u ) { assert(u < 2); return m_aiPad[u];}
221  Int* getPad         ( )        { return m_aiPad; }
222
223  Void        initMultiviewSPS      ( UInt uiViewId, Int iViewOrderIdx = 0, UInt uiCamParPrecision = 0, Bool bCamParSlice = false, Int** aaiScale = 0, Int** aaiOffset = 0 );
224  Void        initMultiviewSPSDepth ( UInt uiViewId, Int iViewOrderIdx );
225
226  UInt        getViewId             ()  { return m_uiViewId; }
227  Int         getViewOrderIdx       ()  { return m_iViewOrderIdx; }
228  Bool        isDepth               ()  { return m_bDepth; }
229  UInt        getCamParPrecision    ()  { return m_uiCamParPrecision; }
230  Bool        hasCamParInSliceHeader()  { return m_bCamParInSliceHeader; }
231  Int*        getCodedScale         ()  { return m_aaiCodedScale [0]; }
232  Int*        getCodedOffset        ()  { return m_aaiCodedOffset[0]; }
233  Int*        getInvCodedScale      ()  { return m_aaiCodedScale [1]; }
234  Int*        getInvCodedOffset     ()  { return m_aaiCodedOffset[1]; }
235
236  // physical transform
237  Void setMaxTrSize   ( UInt u ) { m_uiMaxTrSize = u;       }
238  UInt getMaxTrSize   ()         { return  m_uiMaxTrSize;   }
239
240  // Tool list
241#if POZNAN_ENCODE_ONLY_DISOCCLUDED_CU
242  Bool getUseALF      ()         { return (getViewId()>0)?false:m_bUseALF;}  //todo fix ALF
243#else
244  Bool getUseALF      ()         { return m_bUseALF;        }
245#endif
246  Bool getUseDQP      ()         { return m_bUseDQP;        }
247
248#if !HHI_NO_LowDelayCoding
249  Bool getUseLDC      ()         { return m_bUseLDC;        }
250#endif
251  Bool getUsePAD      ()         { return m_bUsePAD;        }
252  Bool getUseMRG      ()         { return m_bUseMRG;        } // SOPH:
253
254  Void setUseALF      ( Bool b ) { m_bUseALF  = b;          }
255  Void setUseDQP      ( Bool b ) { m_bUseDQP   = b;         }
256
257#if !HHI_NO_LowDelayCoding
258  Void setUseLDC      ( Bool b ) { m_bUseLDC   = b;         }
259#endif
260  Void setUsePAD      ( Bool b ) { m_bUsePAD   = b;         }
261  Void setUseMRG      ( Bool b ) { m_bUseMRG  = b;          } // SOPH:
262 
263#if HHI_DMM_WEDGE_INTRA || HHI_DMM_PRED_TEX
264  Bool getUseDMM()         { return m_bUseDMM; }
265  Void setUseDMM( Bool b ) { m_bUseDMM = b;    }
266#endif
267#if HHI_DMM_PRED_TEX && FLEX_CODING_ORDER
268  Bool getUseDMM34()         { return m_bUseDMM34; }
269  Void setUseDMM34( Bool b ) { m_bUseDMM34 = b;    }
270#endif
271
272#if DCM_COMB_LIST
273  Void setUseLComb    (Bool b)   { m_bUseLComb = b;         }
274  Bool getUseLComb    ()         { return m_bUseLComb;      }
275  Void setLCMod       (Bool b)   { m_bLCMod = b;     }
276  Bool getLCMod       ()         { return m_bLCMod;  }
277#endif
278
279#if HHI_RMP_SWITCH
280  Bool getUseRMP     ()         { return m_bUseRMP; }
281  Void setUseRMP     ( Bool b ) { m_bUseRMP = b;    }
282#endif
283
284#if LM_CHROMA
285  Bool getUseLMChroma ()         { return m_bUseLMChroma;        }
286  Void setUseLMChroma ( Bool b ) { m_bUseLMChroma  = b;          }
287#endif
288
289  // AMVP mode (for each depth)
290  AMVP_MODE getAMVPMode ( UInt uiDepth ) { assert(uiDepth < g_uiMaxCUDepth);  return m_aeAMVPMode[uiDepth]; }
291  Void      setAMVPMode ( UInt uiDepth, AMVP_MODE eMode) { assert(uiDepth < g_uiMaxCUDepth);  m_aeAMVPMode[uiDepth] = eMode; }
292
293  // Bit-depth
294  UInt      getBitDepth     ()         { return m_uiBitDepth;     }
295  Void      setBitDepth     ( UInt u ) { m_uiBitDepth = u;        }
296  UInt      getBitIncrement ()         { return m_uiBitIncrement; }
297  Void      setBitIncrement ( UInt u ) { m_uiBitIncrement = u;    }
298
299#if MTK_NONCROSS_INLOOP_FILTER
300  Void      setLFCrossSliceBoundaryFlag     ( Bool   bValue  )    { m_bLFCrossSliceBoundaryFlag = bValue; }
301  Bool      getLFCrossSliceBoundaryFlag     ()                    { return m_bLFCrossSliceBoundaryFlag;   }
302#endif
303
304#if MTK_SAO
305  Void setUseSAO                  (Bool bVal)  {m_bUseSAO = bVal;}
306#if POZNAN_ENCODE_ONLY_DISOCCLUDED_CU //todo fix SAO
307  Bool getUseSAO     ()           { return (getUseCUSkip() && getViewId()>0)?false:m_bUseSAO;}
308#else
309  Bool getUseSAO                  ()           {return m_bUseSAO;}
310#endif
311#endif
312
313#if HHI_MPI
314  Void setUseMVI                  (Bool bVal)  {m_bUseMVI = bVal;}
315  Bool getUseMVI                  ()           {return m_bUseMVI;}
316#endif
317
318  Void      setCodedPictureBufferSize( UInt u ) { m_uiCodedPictureBufferSize = u ;}
319  UInt      getCodedPictureBufferSize( )        { return m_uiCodedPictureBufferSize ;}
320
321#if DEPTH_MAP_GENERATION
322  Void  setPredDepthMapGeneration( UInt uiViewId, Bool bIsDepth, UInt uiPdmGenMode = 0, UInt uiPdmMvPredMode = 0, UInt uiPdmPrec = 0, Int** aaiPdmScaleNomDelta = 0, Int** aaiPdmOffset = 0 );
323#endif
324#if HHI_INTER_VIEW_RESIDUAL_PRED
325  Void  setMultiviewResPredMode  ( UInt uiResPrdMode ) { m_uiMultiviewResPredMode = uiResPrdMode; }
326#endif
327
328#if POZNAN_DBMP
329  Void  setDBMP                  ( UInt uiDBMP )       { m_uiDBMP = uiDBMP; }
330#endif
331
332#if POZNAN_ENCODE_ONLY_DISOCCLUDED_CU
333  Void  setUseCUSkip                ( UInt uiUseCUSkip )     { m_uiUseCUSkip = uiUseCUSkip; }
334#endif
335
336#if DEPTH_MAP_GENERATION
337  UInt  getPredDepthMapGeneration()          { return m_uiPredDepthMapGeneration; }
338  UInt  getPdmPrecision          ()          { return m_uiPdmPrecision;           }
339  Int*  getPdmScaleNomDelta      ()          { return m_aiPdmScaleNomDelta;       }
340  Int*  getPdmOffset             ()          { return m_aiPdmOffset;              }
341#endif
342#if HHI_INTER_VIEW_MOTION_PRED
343  UInt  getMultiviewMvPredMode   ()          { return m_uiMultiviewMvPredMode;    }
344#endif
345#if HHI_INTER_VIEW_RESIDUAL_PRED
346  UInt  getMultiviewResPredMode  ()          { return m_uiMultiviewResPredMode;   }
347#endif
348
349#if POZNAN_DBMP
350  UInt  getDBMP                  ()          { return m_uiDBMP;   }
351#endif
352
353#if POZNAN_ENCODE_ONLY_DISOCCLUDED_CU
354  UInt  getUseCUSkip                ()          { return m_uiUseCUSkip;   }
355#endif
356
357#if DEPTH_MAP_GENERATION
358  Void                    setDepthMapGenerator( TComDepthMapGenerator* pcDepthMapGenerator )  { m_pcDepthMapGenerator = pcDepthMapGenerator; }
359  TComDepthMapGenerator*  getDepthMapGenerator()                                              { return m_pcDepthMapGenerator; }
360#endif
361#if HHI_INTER_VIEW_RESIDUAL_PRED
362  Void                    setResidualGenerator( TComResidualGenerator* pcResidualGenerator )  { m_pcResidualGenerator = pcResidualGenerator; }
363  TComResidualGenerator*  getResidualGenerator()                                              { return m_pcResidualGenerator; }
364#endif
365#if POZNAN_NONLINEAR_DEPTH
366  inline Void                            setNonlinearDepthModel( TComNonlinearDepthModel &rp ){ m_cNonlinearDepthModel = rp; }
367  inline TComNonlinearDepthModel&        getNonlinearDepthModel()                             { return m_cNonlinearDepthModel; }
368  Bool                    getUseNonlinearDepth()                                              { return m_bUseNonlinearDepth; }
369  Void                    setUseNonlinearDepth( Bool bVal )                                   { m_bUseNonlinearDepth = bVal; }
370#endif
371#if POZNAN_TEXTURE_TU_DELTA_QP_ACCORDING_TO_DEPTH
372  Void setUseTexDqpAccordingToDepth    (Bool bVal)  {m_bUseTexDqpAccordingToDepth = bVal;}
373  Bool getUseTexDqpAccordingToDepth    ()           {return m_bUseTexDqpAccordingToDepth;}
374#endif
375#if BITSTREAM_EXTRACTION
376  Void      setLayerId              ( UInt u ) { m_uiLayerId = u; }
377  UInt      getLayerId              ()         { return m_uiLayerId; }
378#endif
379};
380
381/// PPS class
382class TComPPS
383{
384private:
385#if CONSTRAINED_INTRA_PRED
386  Bool        m_bConstrainedIntraPred;    //  constrained_intra_pred_flag
387#endif
388#ifdef WEIGHT_PRED
389  Bool        m_bUseWeightPred;           // Use of Weighting Prediction (P_SLICE)
390  UInt        m_uiBiPredIdc;              // Use of Weighting Bi-Prediction (B_SLICE)
391#endif
392  UInt        m_uiPPSId;
393  UInt        m_uiSPSId;
394#if BITSTREAM_EXTRACTION
395  UInt        m_uiLayerId;
396#endif
397
398public:
399  TComPPS();
400  virtual ~TComPPS();
401
402  Void      setPPSId                ( UInt u ) { m_uiPPSId = u; }
403  UInt      getPPSId                ()         { return m_uiPPSId; }
404  Void      setSPSId                ( UInt u ) { m_uiSPSId = u; }
405  UInt      getSPSId                ()         { return m_uiSPSId; }
406#if CONSTRAINED_INTRA_PRED
407  Bool      getConstrainedIntraPred ()         { return  m_bConstrainedIntraPred; }
408  Void      setConstrainedIntraPred ( Bool b ) { m_bConstrainedIntraPred = b;     }
409#endif
410
411#ifdef WEIGHT_PRED
412  Bool getUseWP                     ()          { return m_bUseWeightPred;      }
413  UInt getWPBiPredIdc               ()          { return m_uiBiPredIdc;         }
414
415  Void setUseWP                     ( Bool b )  { m_bUseWeightPred = b;       }
416  Void setWPBiPredIdc               ( UInt u )  { m_uiBiPredIdc = u;          }
417#endif
418
419#if BITSTREAM_EXTRACTION
420  Void      setLayerId              ( UInt u ) { m_uiLayerId = u; }
421  UInt      getLayerId              ()         { return m_uiLayerId; }
422#endif
423};
424
425#ifdef WEIGHT_PRED
426typedef struct {
427  // Explicit weighted prediction parameters parsed in slice header,
428  // or Implicit weighted prediction parameters (8 bits depth values).
429  Bool        bPresentFlag;
430  UInt        uiLog2WeightDenom;
431  Int         iWeight;
432  Int         iOffset;
433
434  // Weighted prediction scaling values built from above parameters (bitdepth scaled):
435  Int         w, o, offset, shift, round;
436} wpScalingParam;
437
438typedef struct {
439  Int64 iAC;
440  Int64 iDC;
441} wpACDCParam;
442#endif
443
444/// slice header class
445class TComSlice
446{
447
448private:
449  //  Bitstream writing
450  UInt        m_uiPPSId;
451  Int         m_iPOC;
452#if BITSTREAM_EXTRACTION
453  UInt        m_uiLayerId;
454#endif
455#if SONY_COLPIC_AVAILABILITY
456  Int         m_iViewOrderIdx;
457#endif
458#if DCM_DECODING_REFRESH
459  NalUnitType m_eNalUnitType;         ///< Nal unit type for the slice
460#endif
461  SliceType   m_eSliceType;
462  Int         m_iSliceQp;
463  Int         m_iSymbolMode;
464  Bool        m_bLoopFilterDisable;
465
466#if DCM_COMB_LIST
467  Int         m_aiNumRefIdx   [3];    //  for multiple reference of current slice
468
469  Int         m_iRefIdxOfLC[2][MAX_NUM_REF_LC];
470  Int         m_eListIdFromIdxOfLC[MAX_NUM_REF_LC];
471  Int         m_iRefIdxFromIdxOfLC[MAX_NUM_REF_LC];
472  Int         m_iRefIdxOfL1FromRefIdxOfL0[MAX_NUM_REF_LC];
473  Int         m_iRefIdxOfL0FromRefIdxOfL1[MAX_NUM_REF_LC];
474  Bool        m_bRefPicListModificationFlagLC;
475  Bool        m_bRefPicListCombinationFlag;
476#else
477  Int         m_aiNumRefIdx   [2];    //  for multiple reference of current slice
478#endif
479
480  //  Data
481  Int         m_iSliceQpDelta;
482  TComPic*    m_apcRefPicList [2][MAX_NUM_REF];
483  Int         m_aiRefPOCList  [2][MAX_NUM_REF];
484  Int         m_iDepth;
485  TComPic*    m_pcTexturePic;
486  TComPic*    m_pcDepthPic;
487
488  // referenced slice?
489  Bool        m_bRefenced;
490#ifdef ROUNDING_CONTROL_BIPRED
491  Bool        m_bRounding;
492#endif
493
494  // access channel
495  TComSPS*    m_pcSPS;
496  TComPPS*    m_pcPPS;
497  TComPic*    m_pcPic;
498
499  UInt        m_uiColDir;  // direction to get colocated CUs
500
501  Double      m_dLambda;
502
503  Bool        m_abEqualRef  [2][MAX_NUM_REF][MAX_NUM_REF];
504
505  Int         m_iViewIdx;
506  Int         m_aiRefViewList[2][MAX_INPUT_VIEW_NUM];
507
508  Bool        m_bNoBackPredFlag;
509#if MS_LCEC_LOOKUP_TABLE_EXCEPTION
510  Bool        m_bRefIdxCombineCoding;
511#endif
512
513  Int         m_aaiCodedScale [2][MAX_NUMBER_VIEWS];
514  Int         m_aaiCodedOffset[2][MAX_NUMBER_VIEWS];
515
516  UInt        m_uiSliceMode;
517  UInt        m_uiSliceArgument;
518  UInt        m_uiSliceCurStartCUAddr;
519  UInt        m_uiSliceCurEndCUAddr;
520  UInt        m_uiSliceIdx;
521  UInt        m_uiEntropySliceMode;
522  UInt        m_uiEntropySliceArgument;
523  UInt        m_uiEntropySliceCurStartCUAddr;
524  UInt        m_uiEntropySliceCurEndCUAddr;
525  Bool        m_bNextSlice;
526  Bool        m_bNextEntropySlice;
527  UInt        m_uiSliceBits;
528
529#ifdef WEIGHT_PRED
530  wpScalingParam  m_weightPredTable[2][MAX_NUM_REF][3]; // [REF_PIC_LIST_0 or REF_PIC_LIST_1][refIdx][0:Y, 1:U, 2:V]
531  wpACDCParam   m_weightACDCParam[3]; // [0:Y, 1:U, 2:V]
532#endif
533
534#if POZNAN_MP
535  TComMP* m_pcMP;
536#endif
537
538public:
539  TComSlice();
540#if POZNAN_MP
541  ~TComSlice();
542#else
543  virtual ~TComSlice();
544#endif
545
546  Void      initSlice       ();
547
548  Void      setSPS          ( TComSPS* pcSPS ) { m_pcSPS = pcSPS; }
549  TComSPS*  getSPS          () { return m_pcSPS; }
550
551  Void      setPPS          ( TComPPS* pcPPS ) { m_pcPPS = pcPPS; }
552  TComPPS*  getPPS          () { return m_pcPPS; }
553
554  UInt      getPPSId        ()                          { return  m_uiPPSId;            }
555  SliceType getSliceType    ()                          { return  m_eSliceType;         }
556  Int       getPOC          ()                          { return  m_iPOC;           }
557#if SONY_COLPIC_AVAILABILITY
558  Int       getViewOrderIdx ()                          { return  m_iViewOrderIdx;  }
559#endif
560  Int       getSliceQp      ()                          { return  m_iSliceQp;           }
561  Int       getSliceQpDelta ()                          { return  m_iSliceQpDelta;      }
562  Int       getSymbolMode   ()                          { return  m_iSymbolMode;        }
563  Bool      getLoopFilterDisable()                      { return  m_bLoopFilterDisable; }
564  Int       getNumRefIdx        ( RefPicList e )                { return  m_aiNumRefIdx[e];             }
565  TComPic*  getPic              ()                              { return  m_pcPic;                      }
566  TComPic*  getRefPic           ( RefPicList e, Int iRefIdx)    { return  m_apcRefPicList[e][iRefIdx];  }
567  Int       getRefPOC           ( RefPicList e, Int iRefIdx)    { return  m_aiRefPOCList[e][iRefIdx];   }
568  Int       getDepth            ()                              { return  m_iDepth;                     }
569  UInt      getColDir           ()                              { return  m_uiColDir;                   }
570
571#if DCM_COMB_LIST
572  Int       getRefIdxOfLC       (RefPicList e, Int iRefIdx)     { return m_iRefIdxOfLC[e][iRefIdx];           }
573  Int       getListIdFromIdxOfLC(Int iRefIdx)                   { return m_eListIdFromIdxOfLC[iRefIdx];       }
574  Int       getRefIdxFromIdxOfLC(Int iRefIdx)                   { return m_iRefIdxFromIdxOfLC[iRefIdx];       }
575  Int       getRefIdxOfL0FromRefIdxOfL1(Int iRefIdx)            { return m_iRefIdxOfL0FromRefIdxOfL1[iRefIdx];}
576  Int       getRefIdxOfL1FromRefIdxOfL0(Int iRefIdx)            { return m_iRefIdxOfL1FromRefIdxOfL0[iRefIdx];}
577  Bool      getRefPicListModificationFlagLC()                   {return m_bRefPicListModificationFlagLC;}
578  Void      setRefPicListModificationFlagLC(Bool bflag)         {m_bRefPicListModificationFlagLC=bflag;}
579  Bool      getRefPicListCombinationFlag()                      {return m_bRefPicListCombinationFlag;}
580  Void      setRefPicListCombinationFlag(Bool bflag)            {m_bRefPicListCombinationFlag=bflag;}
581  Void      setListIdFromIdxOfLC(Int  iRefIdx, UInt uiVal)      { m_eListIdFromIdxOfLC[iRefIdx]=uiVal; }
582  Void      setRefIdxFromIdxOfLC(Int  iRefIdx, UInt uiVal)      { m_iRefIdxFromIdxOfLC[iRefIdx]=uiVal; }
583  Void      setRefIdxOfLC       (RefPicList e, Int iRefIdx, Int RefIdxLC)     { m_iRefIdxOfLC[e][iRefIdx]=RefIdxLC;}
584#endif
585
586  Void      setReferenced(Bool b)                               { m_bRefenced = b; }
587  Bool      isReferenced()                                      { return m_bRefenced; }
588#ifdef ROUNDING_CONTROL_BIPRED
589  Void      setRounding(Bool bRound)                            { m_bRounding = bRound; }
590  Bool      isRounding()                                        { return m_bRounding; }
591#endif
592
593  Void      setPPSId            ( UInt u )                      { m_uiPPSId           = u;      }
594  Void      setPOC              ( Int i )                       { m_iPOC              = i;      }
595#if SONY_COLPIC_AVAILABILITY
596  Void      setViewOrderIdx     ( Int i )                       { m_iViewOrderIdx     = i;      }
597#endif
598#if DCM_DECODING_REFRESH
599  Void      setNalUnitType      ( NalUnitType e )               { m_eNalUnitType      = e;      }
600  NalUnitType getNalUnitType    ()                              { return m_eNalUnitType;        }
601  Void      decodingRefreshMarking(UInt& uiPOCCDR, Bool& bRefreshPending, TComList<TComPic*>& rcListPic);
602#endif
603  Void      setSliceType        ( SliceType e )                 { m_eSliceType        = e;      }
604  Void      setSliceQp          ( Int i )                       { m_iSliceQp          = i;      }
605  Void      setSliceQpDelta     ( Int i )                       { m_iSliceQpDelta     = i;      }
606  Void      setSymbolMode       ( Int b  )                      { m_iSymbolMode       = b;      }
607  Void      setLoopFilterDisable( Bool b )                      { m_bLoopFilterDisable= b;      }
608
609  Void      setRefPic           ( TComPic* p, RefPicList e, Int iRefIdx ) { m_apcRefPicList[e][iRefIdx] = p; }
610  Void      setRefPOC           ( Int i, RefPicList e, Int iRefIdx ) { m_aiRefPOCList[e][iRefIdx] = i; }
611  Void      setNumRefIdx        ( RefPicList e, Int i )         { m_aiNumRefIdx[e]    = i;      }
612  Void      setPic              ( TComPic* p )                  { m_pcPic             = p;      }
613  Void      setDepth            ( Int iDepth )                  { m_iDepth            = iDepth; }
614
615  Void      setRefPicList       ( TComList<TComPic*>& rcListPic );
616  Void      setRefPOCList       ();
617  Void      setColDir           ( UInt uiDir ) { m_uiColDir = uiDir; }
618
619  Void      setRefPicListFromGOPSTring( TComList<TComPic*>& rcListPic, std::vector<TComPic*>& rapcSpatRefPics );
620  Void      setRefPicListExplicitlyDecoderSided( TComList<TComPic*>& rcListPic, std::vector<TComPic*>& rapcSpatRefPics );
621
622  Bool      isIntra         ()                          { return  m_eSliceType == I_SLICE;  }
623  Bool      isInterB        ()                          { return  m_eSliceType == B_SLICE;  }
624  Bool      isInterP        ()                          { return  m_eSliceType == P_SLICE;  }
625
626  Void      setLambda( Double d ) { m_dLambda = d; }
627  Double    getLambda() { return m_dLambda;        }
628
629#if BITSTREAM_EXTRACTION
630  Void      setLayerId( UInt u )                        { m_uiLayerId = u; }
631  UInt      getLayerId()                                { return m_uiLayerId; }
632#endif
633
634  Void      setViewIdx(Int i)                           { m_iViewIdx = i; }
635  Int       getViewIdx()                                { return m_iViewIdx; }
636
637  Void      setRefViewIdx       ( Int i, RefPicList e, Int iRefIdx ) { m_aiRefViewList[e][iRefIdx] = i; }
638  Int       getRefViewIdx       ( RefPicList e, Int iRefIdx)    { return  m_aiRefViewList[e][iRefIdx]; }
639
640  Void      initEqualRef();
641  Bool      isEqualRef  ( RefPicList e, Int iRefIdx1, Int iRefIdx2 )
642  {
643    if (iRefIdx1 < 0 || iRefIdx2 < 0) return false;
644    return m_abEqualRef[e][iRefIdx1][iRefIdx2];
645  }
646
647  Void setEqualRef( RefPicList e, Int iRefIdx1, Int iRefIdx2, Bool b)
648  {
649    m_abEqualRef[e][iRefIdx1][iRefIdx2] = m_abEqualRef[e][iRefIdx2][iRefIdx1] = b;
650  }
651
652  static Void      sortPicList         ( TComList<TComPic*>& rcListPic );
653
654  Bool getNoBackPredFlag() { return m_bNoBackPredFlag; }
655  Void setNoBackPredFlag( Bool b ) { m_bNoBackPredFlag = b; }
656#if MS_LCEC_LOOKUP_TABLE_EXCEPTION
657  Bool getRefIdxCombineCoding() { return m_bRefIdxCombineCoding; }
658  Void setRefIdxCombineCoding( Bool b ) { m_bRefIdxCombineCoding = b; }
659#endif
660#if DCM_COMB_LIST
661  Void      generateCombinedList       ();
662#endif
663  Void setSliceMode                     ( UInt uiMode )     { m_uiSliceMode = uiMode;                     }
664  UInt getSliceMode                     ()                  { return m_uiSliceMode;                       }
665  Void setSliceArgument                 ( UInt uiArgument ) { m_uiSliceArgument = uiArgument;             }
666  UInt getSliceArgument                 ()                  { return m_uiSliceArgument;                   }
667  Void setSliceCurStartCUAddr           ( UInt uiAddr )     { m_uiSliceCurStartCUAddr = uiAddr;           }
668  UInt getSliceCurStartCUAddr           ()                  { return m_uiSliceCurStartCUAddr;             }
669  Void setSliceCurEndCUAddr             ( UInt uiAddr )     { m_uiSliceCurEndCUAddr = uiAddr;             }
670  UInt getSliceCurEndCUAddr             ()                  { return m_uiSliceCurEndCUAddr;               }
671  Void setSliceIdx                      ( UInt i)           { m_uiSliceIdx = i;                           }
672  UInt getSliceIdx                      ()                  { return  m_uiSliceIdx;                       }
673  Void copySliceInfo                    (TComSlice *pcSliceSrc);
674  Void setEntropySliceMode              ( UInt uiMode )     { m_uiEntropySliceMode = uiMode;              }
675  UInt getEntropySliceMode              ()                  { return m_uiEntropySliceMode;                }
676  Void setEntropySliceArgument          ( UInt uiArgument ) { m_uiEntropySliceArgument = uiArgument;      }
677  UInt getEntropySliceArgument          ()                  { return m_uiEntropySliceArgument;            }
678  Void setEntropySliceCurStartCUAddr    ( UInt uiAddr )     { m_uiEntropySliceCurStartCUAddr = uiAddr;    }
679  UInt getEntropySliceCurStartCUAddr    ()                  { return m_uiEntropySliceCurStartCUAddr;      }
680  Void setEntropySliceCurEndCUAddr      ( UInt uiAddr )     { m_uiEntropySliceCurEndCUAddr = uiAddr;      }
681  UInt getEntropySliceCurEndCUAddr      ()                  { return m_uiEntropySliceCurEndCUAddr;        }
682  Void setNextSlice                     ( Bool b )          { m_bNextSlice = b;                           }
683  Bool isNextSlice                      ()                  { return m_bNextSlice;                        }
684  Void setNextEntropySlice              ( Bool b )          { m_bNextEntropySlice = b;                    }
685  Bool isNextEntropySlice               ()                  { return m_bNextEntropySlice;                 }
686  Void setSliceBits                     ( UInt uiVal )      { m_uiSliceBits = uiVal;                      }
687  UInt getSliceBits                     ()                  { return m_uiSliceBits;                       }
688
689  Void      initMultiviewSlice    ( Int** aaiScale = 0, Int** aaiOffset = 0 );
690
691  Int*      getCodedScale         ()  { return m_aaiCodedScale [0]; }
692  Int*      getCodedOffset        ()  { return m_aaiCodedOffset[0]; }
693  Int*      getInvCodedScale      ()  { return m_aaiCodedScale [1]; }
694  Int*      getInvCodedOffset     ()  { return m_aaiCodedOffset[1]; }
695
696  Void setTexturePic( TComPic *pcTexturePic ) { m_pcTexturePic = pcTexturePic; }
697  Void setDepthPic  ( TComPic *pcDepthPic   ) { m_pcDepthPic   = pcDepthPic;   }
698  TComPic *getTexturePic() const { return m_pcTexturePic; }
699  TComPic *getDepthPic()   const { return m_pcDepthPic; }
700
701#ifdef WEIGHT_PRED
702  Void  setWpScaling( wpScalingParam  wp[2][MAX_NUM_REF][3] ) { memcpy(m_weightPredTable, wp, sizeof(wpScalingParam)*2*MAX_NUM_REF*3); }
703  Void  getWpScaling( RefPicList e, Int iRefIdx, wpScalingParam *&wp);
704  Void  displayWpScaling();
705  Void  resetWpScaling(wpScalingParam  wp[2][MAX_NUM_REF][3]);
706  Void  initWpScaling(wpScalingParam  wp[2][MAX_NUM_REF][3]);
707  Void  initWpScaling();
708  inline Bool applyWP() { return( (m_eSliceType==P_SLICE && m_pcPPS->getUseWP()) || (m_eSliceType==B_SLICE && m_pcPPS->getWPBiPredIdc()) ); }
709
710  Void  setWpAcDcParam ( wpACDCParam wp[3] ) { memcpy(m_weightACDCParam, wp, sizeof(wpACDCParam)*3); }
711  Void  getWpAcDcParam ( wpACDCParam *&wp );
712  Void  initWpAcDcParam();
713#endif
714
715#if POZNAN_MP
716  Void setMP(TComMP* pcMP) { m_pcMP = pcMP; }
717  TComMP* getMP() { return m_pcMP; }
718#endif
719
720protected:
721#if 0
722  TComPic*  xGetRefPic  (TComList<TComPic*>& rcListPic,
723                         Bool                bDRBFlag,
724                         ERBIndex            eERBIndex,
725                         UInt                uiPOCCurr,
726                         RefPicList          eRefPicList,
727                         UInt                uiNthRefPic );
728#endif
729
730};// END CLASS DEFINITION TComSlice
731
732
733#endif // __TCOMSLICE__
734
Note: See TracBrowser for help on using the repository browser.