1 | |
---|
2 | |
---|
3 | #ifndef __TCOMMVDREFDATA__ |
---|
4 | #define __TCOMMVDREFDATA__ |
---|
5 | |
---|
6 | // Include files |
---|
7 | #include "CommonDef.h" |
---|
8 | #include "TComPicYuv.h" |
---|
9 | #include <vector> |
---|
10 | |
---|
11 | // ==================================================================================================================== |
---|
12 | // Class definition |
---|
13 | // ==================================================================================================================== |
---|
14 | |
---|
15 | |
---|
16 | class TComMVDRefData |
---|
17 | { |
---|
18 | private: |
---|
19 | // PicYuvs |
---|
20 | TComPicYuv* m_apcVideoOrg[3]; |
---|
21 | TComPicYuv* m_apcDepthOrg[3]; |
---|
22 | TComPicYuv* m_apcVideoRec[3]; |
---|
23 | TComPicYuv* m_apcDepthRec[3]; |
---|
24 | |
---|
25 | // LUTs |
---|
26 | Double** m_adBaseViewShiftLUT[3]; |
---|
27 | Int** m_aiBaseViewShiftLUT[3]; |
---|
28 | |
---|
29 | |
---|
30 | //GT VSO |
---|
31 | //PicYuvs |
---|
32 | std::vector<TComPicYuv*> m_apcExternalReferenceViews; |
---|
33 | |
---|
34 | // LUTs |
---|
35 | Double*** m_adERViewShiftLUT; |
---|
36 | Int*** m_aiERViewShiftLUT; |
---|
37 | |
---|
38 | // Indices |
---|
39 | std::vector<Int> m_aiBaseViewRefInd; |
---|
40 | std::vector<Int> m_aiExtViewRefInd; |
---|
41 | std::vector<Int> m_aiExtViewRefLUTInd; |
---|
42 | |
---|
43 | //GT VSO end |
---|
44 | |
---|
45 | public: |
---|
46 | TComMVDRefData(); |
---|
47 | |
---|
48 | // PicYuvs |
---|
49 | TComPicYuv* getPicYuvRecVideo ( InterViewReference eIVRef ) { return m_apcVideoRec[eIVRef]; }; |
---|
50 | TComPicYuv* getPicYuvOrgVideo ( InterViewReference eIVRef ) { return m_apcVideoOrg[eIVRef]; }; |
---|
51 | TComPicYuv* getPicYuvRecDepth ( InterViewReference eIVRef ) { return m_apcDepthRec[eIVRef]; }; |
---|
52 | TComPicYuv* getPicYuvOrgDepth ( InterViewReference eIVRef ) { return m_apcDepthOrg[eIVRef]; }; |
---|
53 | Void setPicYuvBaseView ( InterViewReference eView, Bool bDepth, TComPicYuv* pcOrgView, TComPicYuv* pcRecView ); |
---|
54 | |
---|
55 | |
---|
56 | TComPicYuv* getPicYuvVideo ( ) { return getPicYuvOrgVideo( CURRVIEW ); /*return getRecVideo( CURRVIEW ) ? getRecVideo( CURRVIEW ) : getOrgVideo( CURRVIEW ); */ }; |
---|
57 | //TComPicYuv* getVideoData( ) { return getRecVideo( CURRVIEW ) ? getRecVideo( CURRVIEW ) : getOrgVideo( CURRVIEW ); }; |
---|
58 | |
---|
59 | // Luts |
---|
60 | Int** getShiftLUTLBaseView ( InterViewReference eIVRef ) { return m_aiBaseViewShiftLUT[eIVRef]; }; |
---|
61 | Double** getShiftLUTDBaseView ( InterViewReference eIVRef ) { return m_adBaseViewShiftLUT[eIVRef]; }; |
---|
62 | Void setShiftLUTsBaseView ( InterViewReference eView, Double** adInLut, Int** alInLut ); |
---|
63 | |
---|
64 | // Void render( TComPicYuv* pOut ) /*Void render( TComPicYuv* pIn, TComPicYuv* pOut, TComPicYuv* pDepth, Long** aalLUT ) */; |
---|
65 | |
---|
66 | //GT VSO |
---|
67 | // PicYuvs |
---|
68 | TComPicYuv* getPicYuvERView ( UInt uiReferenceNum ) { return m_apcExternalReferenceViews[uiReferenceNum]; }; |
---|
69 | Void setPicYuvERViews ( std::vector<TComPicYuv*> papcIn ) { m_apcExternalReferenceViews = papcIn; }; |
---|
70 | |
---|
71 | // LUTS |
---|
72 | Int** getShiftLUTIERView ( UInt uiTargetViewNum ) { return (m_aiERViewShiftLUT)[uiTargetViewNum]; }; |
---|
73 | Double** getShiftLUTDERView ( UInt uiTargetViewNum ) { return (m_adERViewShiftLUT)[uiTargetViewNum]; }; |
---|
74 | |
---|
75 | Void setShiftLUTsERView ( Double*** adInLut, Int*** aiInLut) { m_aiERViewShiftLUT = aiInLut; m_adERViewShiftLUT = adInLut; }; |
---|
76 | |
---|
77 | // Set Indices of actually used references |
---|
78 | Void setRefViewInd ( std::vector<Int> aiBaseViewRef, std::vector<Int> aiEViewRef, std::vector<Int> aiEViewRefLut ) { m_aiBaseViewRefInd = aiBaseViewRef, m_aiExtViewRefInd = aiEViewRef, m_aiExtViewRefLUTInd = aiEViewRefLut; } |
---|
79 | |
---|
80 | // Get Reference Data |
---|
81 | UInt getNumOfRefViews() { return (UInt) m_aiExtViewRefInd.size() + (UInt) m_aiBaseViewRefInd.size(); }; |
---|
82 | Void getRefPicYuvAndLUT ( TComPicYuv**& rpacDistRefPicList, Int***& rppalShiftLut); |
---|
83 | Void getRefPicYuvAndLUTMode1( TComPicYuv**& rpacDistRefPicList, Int***& rppaiShiftLut ); |
---|
84 | //GT VSO end |
---|
85 | |
---|
86 | }; // END CLASS DEFINITION TComMVDRefData |
---|
87 | |
---|
88 | #endif // __TCOMMVDREFDATA__ |
---|
89 | |
---|
90 | |
---|
91 | |
---|