HEVC Test Model (HM)  HM-16.3
TComMotionInfo.h
Go to the documentation of this file.
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-2015, 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 
39 #ifndef __TCOMMOTIONINFO__
40 #define __TCOMMOTIONINFO__
41 
42 #include <memory.h>
43 #include "CommonDef.h"
44 #include "TComMv.h"
45 
48 
49 // ====================================================================================================================
50 // Type definition
51 // ====================================================================================================================
52 
54 typedef struct _AMVPInfo
55 {
57  Int iN;
58 } AMVPInfo;
59 
60 // ====================================================================================================================
61 // Class definition
62 // ====================================================================================================================
63 
66 {
67 private:
70 
71 public:
72  TComMvField() : m_iRefIdx( NOT_VALID ) {}
73 
74  Void setMvField( TComMv const & cMv, Int iRefIdx )
75  {
76  m_acMv = cMv;
77  m_iRefIdx = iRefIdx;
78  }
79 
80  Void setRefIdx( Int refIdx ) { m_iRefIdx = refIdx; }
81 
82  TComMv const & getMv() const { return m_acMv; }
83  TComMv & getMv() { return m_acMv; }
84 
85  Int getRefIdx() const { return m_iRefIdx; }
86  Int getHor () const { return m_acMv.getHor(); }
87  Int getVer () const { return m_acMv.getVer(); }
88 };
89 
92 {
93 private:
99 
100  template <typename T>
101  Void setAll( T *p, T const & val, PartSize eCUMode, Int iPartAddr, UInt uiDepth, Int iPartIdx );
102 
103 public:
104  TComCUMvField() : m_pcMv(NULL), m_pcMvd(NULL), m_piRefIdx(NULL), m_uiNumPartition(0) {}
106 
107  // ------------------------------------------------------------------------------------------------------------------
108  // create / destroy
109  // ------------------------------------------------------------------------------------------------------------------
110 
111  Void create( UInt uiNumPartition );
112  Void destroy();
113 
114  // ------------------------------------------------------------------------------------------------------------------
115  // clear / copy
116  // ------------------------------------------------------------------------------------------------------------------
117 
118  Void clearMvField();
119 
120  Void copyFrom( TComCUMvField const * pcCUMvFieldSrc, Int iNumPartSrc, Int iPartAddrDst );
121  Void copyTo ( TComCUMvField* pcCUMvFieldDst, Int iPartAddrDst ) const;
122  Void copyTo ( TComCUMvField* pcCUMvFieldDst, Int iPartAddrDst, UInt uiOffset, UInt uiNumPart ) const;
123 
124  // ------------------------------------------------------------------------------------------------------------------
125  // get
126  // ------------------------------------------------------------------------------------------------------------------
127 
128  TComMv const & getMv ( Int iIdx ) const { return m_pcMv [iIdx]; }
129  TComMv const & getMvd ( Int iIdx ) const { return m_pcMvd [iIdx]; }
130  Int getRefIdx( Int iIdx ) const { return m_piRefIdx[iIdx]; }
131 
133 
134  // ------------------------------------------------------------------------------------------------------------------
135  // set
136  // ------------------------------------------------------------------------------------------------------------------
137 
138  Void setAllMv ( TComMv const & rcMv, PartSize eCUMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0 );
139  Void setAllMvd ( TComMv const & rcMvd, PartSize eCUMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0 );
140  Void setAllRefIdx ( Int iRefIdx, PartSize eMbMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0 );
141  Void setAllMvField( TComMvField const & mvField, PartSize eMbMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0 );
142 
144  {
145  m_uiNumPartition = iNumPart;
146  }
147 
148  Void linkToWithOffset( TComCUMvField const * src, Int offset )
149  {
150  m_pcMv = src->m_pcMv + offset;
151  m_pcMvd = src->m_pcMvd + offset;
152  m_piRefIdx = src->m_piRefIdx + offset;
153  }
154 
155  Void compress(Char* pePredMode, Int scale);
156 };
157 
159 
160 #endif // __TCOMMOTIONINFO__
Void setRefIdx(Int refIdx)
Int getRefIdx(Int iIdx) const
Void linkToWithOffset(TComCUMvField const *src, Int offset)
Int getHor() const
Definition: TComMv.h:88
Defines constants, macros and tool parameters.
void Void
Definition: TypeDef.h:285
Void setAllRefIdx(Int iRefIdx, PartSize eMbMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0)
#define NULL
Definition: CommonDef.h:100
TComMv const & getMv() const
class for motion vector with reference index
char Char
Definition: TypeDef.h:291
TComMv * m_pcMvd
unsigned int UInt
Definition: TypeDef.h:297
TComMv const & getMvd(Int iIdx) const
Void copyFrom(TComCUMvField const *pcCUMvFieldSrc, Int iNumPartSrc, Int iPartAddrDst)
class for motion information in one CU
PartSize
supported partition shape
Definition: TypeDef.h:393
Void setMvField(TComMv const &cMv, Int iRefIdx)
Void compress(Char *pePredMode, Int scale)
#define NOT_VALID
Definition: CommonDef.h:129
Int iN
number of motion vector predictor candidates
AMVPInfo m_cAMVPInfo
Void setAllMvField(TComMvField const &mvField, PartSize eMbMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0)
Void setAll(T *p, T const &val, PartSize eCUMode, Int iPartAddr, UInt uiDepth, Int iPartIdx)
Void setNumPartition(Int iNumPart)
parameters for AMVP
TComMv const & getMv(Int iIdx) const
motion vector class (header)
struct _AMVPInfo AMVPInfo
parameters for AMVP
Void copyTo(TComCUMvField *pcCUMvFieldDst, Int iPartAddrDst) const
Int getVer() const
Definition: TComMv.h:89
Int getHor() const
Int getVer() const
TComMv * m_pcMv
AMVPInfo * getAMVPInfo()
Void setAllMvd(TComMv const &rcMvd, PartSize eCUMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0)
TComMv & getMv()
int Int
Definition: TypeDef.h:296
basic motion vector class
Definition: TComMv.h:51
#define AMVP_MAX_NUM_CANDS_MEM
max number of candidates
Definition: CommonDef.h:184
Void create(UInt uiNumPartition)
Void setAllMv(TComMv const &rcMv, PartSize eCUMode, Int iPartAddr, UInt uiDepth, Int iPartIdx=0)
Int getRefIdx() const
TComMv m_acMvCand[3]
array of motion vector predictor candidates