source: 3DVCSoftware/branches/0.3-poznan-univ/source/Lib/TLibCommon/TComMP.h @ 28

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

Poznan Tools

  • Encoding only disoccluded CUs in depended views
  • Depth based motion prediction
  • Texture QP adjustment based on depth data
  • Nonlinear depth representation
File size: 7.3 KB
Line 
1
2
3#ifndef __TCOMMP__
4#define __TCOMMP__
5
6class TComMP;
7
8#include "../TLibCommon/TypeDef.h"
9#include "../TLibCommon/CommonDef.h"
10
11#if POZNAN_MP_USE_DEPTH_MAP_GENERATION
12class TComDepthMapGenerator;
13//#include "../TLibCommon/TComDepthMapGenerator.h"
14#else
15class TAppComCamPara;
16//#include "../../App/TAppCommon/TAppComCamPara.h"
17#endif
18
19#include "../TLibCommon/TComList.h"
20#include "../TLibCommon/TComSlice.h"
21#include "../TLibCommon/TComPic.h"
22#include "../TLibCommon/TComDataCU.h"
23#include "../TLibCommon/TComMotionInfo.h"
24
25#include <map>
26//#include <cstdlib>
27//using namespace std;
28
29#if POZNAN_MP
30
31// ====================================================================================================================
32// Class definition
33// ====================================================================================================================
34
35
36/// MP class
37class TComMP
38{
39//typedef Int PUT_MP_ARRAY_TYPE;
40typedef Short PUT_MP_ARRAY_TYPE;
41typedef std::map<UInt,PUT_MP_ARRAY_TYPE**> TComMapppi;
42
43static const int OCCLUSION = -1;
44
45private:
46
47  Bool m_bInit;
48  Bool m_bEnabled;
49  Bool m_bDBMPEnabled;
50
51  UInt m_uiHeight;
52  UInt m_uiWidth;
53
54  UInt m_uiPOC;
55  UInt m_uiView;
56  Bool m_bIsDepth;
57
58#if POZNAN_MP_USE_DEPTH_MAP_GENERATION
59  TComDepthMapGenerator* m_pcDepthMapGenerator;
60#else
61  //TAppComCamPara* m_pcCameraData;
62  Int**** m_aiBaseViewShiftLUT;
63#endif
64  std::vector<TComPic*>* m_pcRefPicsList;
65  std::vector<TComPic*>* m_pcDepthRefPicsList;
66
67  Short** m_ppiMvPtCorrX;
68  Short** m_ppiMvPtCorrY;
69  Short** m_ppiMvPtCorrZ;
70  Short** m_ppiMvPtCorrRefViewIdx;
71
72  TComDataCU*** m_pppcRefCU; //XY array of pointers to reference CU for each point of picture
73  UShort** m_ppuicRefPartAddr; //XY array of part addresses for each reference CU for each point of picture
74
75#if POZNAN_DBMP & !POZNAN_DBMP_COMPRESS_ME_DATA
76  TComMapppi m_ppiL0RefPOC;
77  TComMapppi m_ppiL0MvX;
78  TComMapppi m_ppiL0MvY;
79  TComMapppi m_ppiL1RefPOC;
80  TComMapppi m_ppiL1MvX;
81  TComMapppi m_ppiL1MvY;
82#endif
83
84#if POZNAN_DBMP_CALC_PRED_DATA
85  Int* m_piTempL0RefIdx;
86  Int* m_piTempL0MvX;
87  Int* m_piTempL0MvY;
88  Int* m_piTempL1RefIdx;
89  Int* m_piTempL1MvX;
90  Int* m_piTempL1MvY;
91
92  Int*  m_piTempBuff;
93  Int*  m_piResBuff;
94#endif
95
96#if POZNAN_DBMP_CALC_PRED_DATA
97  TComCUMvField* apcDBMPPredMVField[2];
98#endif
99
100#if POZNAN_DBMP & !POZNAN_DBMP_COMPRESS_ME_DATA
101  Void initArrayMap(UInt uiViewId, Bool bIsDepth);
102#endif
103
104public:
105
106#if POZNAN_MP_USE_DEPTH_MAP_GENERATION
107  TComMP(UInt uiHeight=0, UInt uiWidth=0);
108#else
109  //TComMP(UInt uiHeight=0, UInt uiWidth=0, TAppComCamPara* pcCameraData=NULL);
110  TComMP(UInt uiHeight=0, UInt uiWidth=0, Int**** aiBaseViewShiftLUT=NULL);
111#endif
112  ~TComMP();
113
114#if POZNAN_MP_USE_DEPTH_MAP_GENERATION
115  Void                  init(UInt uiHeight, UInt uiWidth);
116#else
117  //Void                        init(UInt uiHeight, UInt uiWidth, TAppComCamPara* pcCameraData);
118  Void                  init(UInt uiHeight, UInt uiWidth, Int**** aiBaseViewShiftLUT);
119#endif
120  Void                  uninit();
121  Void                  clear();
122  Void                  disable();
123
124  Bool                  isInit();
125  Bool                  isEnabled();
126  Bool                  isDBMPEnabled();
127
128#if POZNAN_MP_USE_DEPTH_MAP_GENERATION
129  Void                  setDepthMapGenerator(TComDepthMapGenerator*  pcDepthMapGenerator){m_pcDepthMapGenerator = pcDepthMapGenerator;}
130#endif 
131  Void                  setRefPicsList(std::vector<TComPic*>* pcListPic){m_pcRefPicsList = pcListPic;}
132  Void                  setDepthRefPicsList(std::vector<TComPic*>* pcListPic){m_pcDepthRefPicsList = pcListPic;}
133  Bool                  pairMultiview(TComPic* pcPic);
134
135  Short**               getMPCx() {return m_ppiMvPtCorrX;}
136  Short**               getMPCy() {return m_ppiMvPtCorrY;}
137  Short**               getMPCz() {return m_ppiMvPtCorrZ;}
138  Short**               getMPCRefViewIdx() {return m_ppiMvPtCorrRefViewIdx;}
139  TComDataCU*** getMPCRefCU() {return m_pppcRefCU;}
140  UShort**              getMPCRefPartAddr() {return m_ppuicRefPartAddr;}
141
142#if POZNAN_MP_FILL
143  Void                  fillMultiview(UInt uiViewId);
144#endif
145
146#if POZNAN_DBMP
147#if !POZNAN_DBMP_COMPRESS_ME_DATA
148  PUT_MP_ARRAY_TYPE**                   getL0RefPOC(UInt uiViewID, Bool bIsDepth) {return m_ppiL0RefPOC[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID];}
149  PUT_MP_ARRAY_TYPE**                   getL0MvX(UInt uiViewID, Bool bIsDepth) {return m_ppiL0MvX[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID];}
150  PUT_MP_ARRAY_TYPE**                   getL0MvY(UInt uiViewID, Bool bIsDepth) {return m_ppiL0MvY[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID];}
151  PUT_MP_ARRAY_TYPE**                   getL1RefPOC(UInt uiViewID, Bool bIsDepth) {return m_ppiL1RefPOC[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID];}
152  PUT_MP_ARRAY_TYPE**                   getL1MvX(UInt uiViewID, Bool bIsDepth) {return m_ppiL1MvX[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID];}
153  PUT_MP_ARRAY_TYPE**                   getL1MvY(UInt uiViewID, Bool bIsDepth) {return m_ppiL1MvY[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID];}
154
155  Void                  setL0RefPOC(UInt uiViewID, Bool bIsDepth, UInt uiX, UInt uiY, PUT_MP_ARRAY_TYPE iVal) {m_ppiL0RefPOC[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID][uiY][uiX]=iVal;}
156  Void                  setL0MvX(UInt uiViewID, Bool bIsDepth, UInt uiX, UInt uiY, PUT_MP_ARRAY_TYPE iVal) {m_ppiL0MvX[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID][uiY][uiX]=iVal;}
157  Void                  setL0MvY(UInt uiViewID, Bool bIsDepth, UInt uiX, UInt uiY, PUT_MP_ARRAY_TYPE iVal) {m_ppiL0MvY[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID][uiY][uiX]=iVal;}
158  Void                  setL1RefPOC(UInt uiViewID, Bool bIsDepth, UInt uiX, UInt uiY, PUT_MP_ARRAY_TYPE iVal) {m_ppiL1RefPOC[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID][uiY][uiX]=iVal;}
159  Void                  setL1MvX(UInt uiViewID, Bool bIsDepth, UInt uiX, UInt uiY, PUT_MP_ARRAY_TYPE iVal) {m_ppiL1MvX[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID][uiY][uiX]=iVal;}
160  Void                  setL1MvY(UInt uiViewID, Bool bIsDepth, UInt uiX, UInt uiY, PUT_MP_ARRAY_TYPE iVal) {m_ppiL1MvY[(bIsDepth)? (uiViewID+MAX_NUMBER_VIEWS) : uiViewID][uiY][uiX]=iVal;}
161
162  Void saveDBMPData(TComDataCU* pcCU);
163#endif
164
165  inline Void getDBMPPredData(TComDataCU* pcCU, Int x, Int y, 
166                                                          Int &ref_frame0, Int &ref_frame0_idx, TComMv &mv0, Int ref_frame0_idx_2nd, TComMv mv0_2nd,
167                                                          Int &ref_frame1, Int &ref_frame1_idx, TComMv &mv1, Int ref_frame1_idx_2nd, TComMv mv1_2nd);
168#endif
169
170#if POZNAN_DBMP_CALC_PRED_DATA
171  Int*                  getTempL0RefIdx() {return m_piTempL0RefIdx;}
172  Int*                  getTempL0MvX() {return m_piTempL0MvX;}
173  Int*                  getTempL0MvY() {return m_piTempL0MvY;}
174  Int*                  getTempL1RefIdx() {return m_piTempL1RefIdx;}
175  Int*                  getTempL1MvX() {return m_piTempL1MvX;}
176  Int*                  getTempL1MvY() {return m_piTempL1MvY;}
177 
178  inline Void xCalcDBMPPredData(UInt uiCnt, Int &ref_frame0_idx, TComMv &mv0, Int &ref_frame1_idx, TComMv &mv1);
179#endif
180
181#if POZNAN_DBMP_CALC_PRED_DATA
182  Void calcDBMPPredData(TComDataCU* pcCU, UInt uiAbsPartIdx, Int &ref_frame0_idx,  TComMv &mv0, Int &ref_frame1_idx,  TComMv &mv1);
183  TComCUMvField* getDBMPPredMVField(RefPicList eRefListIdx){return apcDBMPPredMVField[eRefListIdx];}
184  Void setDBMPPredMVField(RefPicList eRefListIdx, TComDataCU* pcCU);
185  Void setDBMPPredMVField(RefPicList eRefListIdx, TComDataCU* pcCU, Int iNumPartSrc, Int iPartAddrDst);
186#endif
187
188  TComPic*              GetPicFromList(std::vector<TComPic*>* rcListPic, UInt uiPOC, UInt uiView);
189  Int                   GetRepresentativeVal(UInt &uiCntMax, Int* puiIn, UInt uiLen, Int* puiTemp, Bool bUseRestrictedVal=false, Int uiRestrictedVal=0);
190
191};
192
193
194
195
196//#endif
197
198#endif // __TCOMMP__
199
200
201#endif // POZNAN_MP
Note: See TracBrowser for help on using the repository browser.