Parent Directory | Revision Log
Revision 1.2 - (view) (download)
1 : | edgomez | 1.2 | /************************************************************************** |
2 : | * | ||
3 : | * XVID MPEG-4 VIDEO CODEC | ||
4 : | * - Motion estimation header - | ||
5 : | * | ||
6 : | * This program is an implementation of a part of one or more MPEG-4 | ||
7 : | * Video tools as specified in ISO/IEC 14496-2 standard. Those intending | ||
8 : | * to use this software module in hardware or software products are | ||
9 : | * advised that its use may infringe existing patents or copyrights, and | ||
10 : | * any such use would be at such party's own risk. The original | ||
11 : | * developer of this software module and his/her company, and subsequent | ||
12 : | * editors and their companies, will have no liability for use of this | ||
13 : | * software or modifications or derivatives thereof. | ||
14 : | * | ||
15 : | * This program is free software; you can redistribute it and/or modify | ||
16 : | * it under the terms of the GNU General Public License as published by | ||
17 : | * the Free Software Foundation; either version 2 of the License, or | ||
18 : | * (at your option) any later version. | ||
19 : | * | ||
20 : | * This program is distributed in the hope that it will be useful, | ||
21 : | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||
22 : | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||
23 : | * GNU General Public License for more details. | ||
24 : | * | ||
25 : | * You should have received a copy of the GNU General Public License | ||
26 : | * along with this program; if not, write to the Free Software | ||
27 : | * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | ||
28 : | * | ||
29 : | * $Id: motion_est.h,v 1.1.2.15 2003/02/12 12:57:27 syskin Exp $ | ||
30 : | * | ||
31 : | ***************************************************************************/ | ||
32 : | |||
33 : | #ifndef _MOTION_EST_H_ | ||
34 : | #define _MOTION_EST_H_ | ||
35 : | |||
36 : | #include "../portab.h" | ||
37 : | #include "../global.h" | ||
38 : | #include "../image/reduced.h" | ||
39 : | |||
40 : | /* hard coded motion search parameters for motion_est and smp_motion_est */ | ||
41 : | |||
42 : | // very large value | ||
43 : | #define MV_MAX_ERROR (4096 * 256) | ||
44 : | |||
45 : | /* INTER bias for INTER/INTRA decision; mpeg4 spec suggests 2*nb */ | ||
46 : | #define MV16_INTER_BIAS 512 | ||
47 : | |||
48 : | /* vector map (vlc delta size) smoother parameters ! float !*/ | ||
49 : | #define NEIGH_TEND_16X16 10.5 | ||
50 : | #define NEIGH_TEND_8X8 40.0 | ||
51 : | #define NEIGH_8X8_BIAS 30 | ||
52 : | |||
53 : | /* Parameters which control inter/inter4v decision */ | ||
54 : | #define IMV16X16 2 | ||
55 : | |||
56 : | static const int lambda_vec16[32] = | ||
57 : | { 0 ,(int)(1.00235 * NEIGH_TEND_16X16 + 0.5), | ||
58 : | (int)(1.15582*NEIGH_TEND_16X16 + 0.5), (int)(1.31976*NEIGH_TEND_16X16 + 0.5), | ||
59 : | (int)(1.49591*NEIGH_TEND_16X16 + 0.5), (int)(1.68601*NEIGH_TEND_16X16 + 0.5), | ||
60 : | (int)(1.89187*NEIGH_TEND_16X16 + 0.5), (int)(2.11542*NEIGH_TEND_16X16 + 0.5), | ||
61 : | (int)(2.35878*NEIGH_TEND_16X16 + 0.5), (int)(2.62429*NEIGH_TEND_16X16 + 0.5), | ||
62 : | (int)(2.91455*NEIGH_TEND_16X16 + 0.5), (int)(3.23253*NEIGH_TEND_16X16 + 0.5), | ||
63 : | (int)(3.58158*NEIGH_TEND_16X16 + 0.5), (int)(3.96555*NEIGH_TEND_16X16 + 0.5), | ||
64 : | (int)(4.38887*NEIGH_TEND_16X16 + 0.5), (int)(4.85673*NEIGH_TEND_16X16 + 0.5), | ||
65 : | (int)(5.37519*NEIGH_TEND_16X16 + 0.5), (int)(5.95144*NEIGH_TEND_16X16 + 0.5), | ||
66 : | (int)(6.59408*NEIGH_TEND_16X16 + 0.5), (int)(7.31349*NEIGH_TEND_16X16 + 0.5), | ||
67 : | (int)(8.12242*NEIGH_TEND_16X16 + 0.5), (int)(9.03669*NEIGH_TEND_16X16 + 0.5), | ||
68 : | (int)(10.0763*NEIGH_TEND_16X16 + 0.5), (int)(11.2669*NEIGH_TEND_16X16 + 0.5), | ||
69 : | (int)(12.6426*NEIGH_TEND_16X16 + 0.5), (int)(14.2493*NEIGH_TEND_16X16 + 0.5), | ||
70 : | (int)(16.1512*NEIGH_TEND_16X16 + 0.5), (int)(18.442*NEIGH_TEND_16X16 + 0.5), | ||
71 : | (int)(21.2656*NEIGH_TEND_16X16 + 0.5), (int)(24.8580*NEIGH_TEND_16X16 + 0.5), | ||
72 : | (int)(29.6436*NEIGH_TEND_16X16 + 0.5), (int)(36.4949*NEIGH_TEND_16X16 + 0.5) }; | ||
73 : | |||
74 : | static const int lambda_vec8[32] = | ||
75 : | { 0 ,(int)(1.00235 * NEIGH_TEND_8X8 + 0.5), | ||
76 : | (int)(1.15582 + NEIGH_TEND_8X8 + 0.5), (int)(1.31976*NEIGH_TEND_8X8 + 0.5), | ||
77 : | (int)(1.49591*NEIGH_TEND_8X8 + 0.5), (int)(1.68601*NEIGH_TEND_8X8 + 0.5), | ||
78 : | (int)(1.89187*NEIGH_TEND_8X8 + 0.5), (int)(2.11542*NEIGH_TEND_8X8 + 0.5), | ||
79 : | (int)(2.35878*NEIGH_TEND_8X8 + 0.5), (int)(2.62429*NEIGH_TEND_8X8 + 0.5), | ||
80 : | (int)(2.91455*NEIGH_TEND_8X8 + 0.5), (int)(3.23253*NEIGH_TEND_8X8 + 0.5), | ||
81 : | (int)(3.58158*NEIGH_TEND_8X8 + 0.5), (int)(3.96555*NEIGH_TEND_8X8 + 0.5), | ||
82 : | (int)(4.38887*NEIGH_TEND_8X8 + 0.5), (int)(4.85673*NEIGH_TEND_8X8 + 0.5), | ||
83 : | (int)(5.37519*NEIGH_TEND_8X8 + 0.5), (int)(5.95144*NEIGH_TEND_8X8 + 0.5), | ||
84 : | (int)(6.59408*NEIGH_TEND_8X8 + 0.5), (int)(7.31349*NEIGH_TEND_8X8 + 0.5), | ||
85 : | (int)(8.12242*NEIGH_TEND_8X8 + 0.5), (int)(9.03669*NEIGH_TEND_8X8 + 0.5), | ||
86 : | (int)(10.0763*NEIGH_TEND_8X8 + 0.5), (int)(11.2669*NEIGH_TEND_8X8 + 0.5), | ||
87 : | (int)(12.6426*NEIGH_TEND_8X8 + 0.5), (int)(14.2493*NEIGH_TEND_8X8 + 0.5), | ||
88 : | (int)(16.1512*NEIGH_TEND_8X8 + 0.5), (int)(18.442*NEIGH_TEND_8X8 + 0.5), | ||
89 : | (int)(21.2656*NEIGH_TEND_8X8 + 0.5), (int)(24.8580*NEIGH_TEND_8X8 + 0.5), | ||
90 : | (int)(29.6436*NEIGH_TEND_8X8 + 0.5), (int)(36.4949*NEIGH_TEND_8X8 + 0.5) }; | ||
91 : | |||
92 : | // mv.length table | ||
93 : | static const uint32_t mvtab[33] = { | ||
94 : | 1, 2, 3, 4, 6, 7, 7, 7, | ||
95 : | 9, 9, 9, 10, 10, 10, 10, 10, | ||
96 : | 10, 10, 10, 10, 10, 10, 10, 10, | ||
97 : | 10, 11, 11, 11, 11, 11, 11, 12, 12 | ||
98 : | }; | ||
99 : | |||
100 : | static const int DQtab[4] = { | ||
101 : | -1, -2, 1, 2 | ||
102 : | }; | ||
103 : | |||
104 : | #define RRV_MV_SCALEDOWN(a) ( (a)>=0 ? (a+1)/2 : (a-1)/2 ) | ||
105 : | |||
106 : | static const VECTOR zeroMV = {0,0}; | ||
107 : | |||
108 : | typedef struct | ||
109 : | { | ||
110 : | // general fields | ||
111 : | int max_dx, min_dx, max_dy, min_dy; | ||
112 : | uint32_t rounding; | ||
113 : | VECTOR predMV; | ||
114 : | VECTOR * currentMV; | ||
115 : | VECTOR * currentQMV; | ||
116 : | int32_t * iMinSAD; | ||
117 : | const uint8_t * Ref; | ||
118 : | const uint8_t * RefH; | ||
119 : | const uint8_t * RefV; | ||
120 : | const uint8_t * RefHV; | ||
121 : | const uint8_t * RefCU; | ||
122 : | const uint8_t * RefCV; | ||
123 : | const uint8_t * CurU; | ||
124 : | const uint8_t * CurV; | ||
125 : | uint8_t * RefQ; | ||
126 : | const uint8_t * Cur; | ||
127 : | uint32_t lambda16; | ||
128 : | uint32_t lambda8; | ||
129 : | uint32_t iEdgedWidth; | ||
130 : | uint32_t iFcode; | ||
131 : | int * temp; | ||
132 : | int qpel, qpel_precision; | ||
133 : | int chroma; | ||
134 : | int rrv; | ||
135 : | //fields for interpolate and direct modes | ||
136 : | const uint8_t * bRef; | ||
137 : | const uint8_t * bRefH; | ||
138 : | const uint8_t * bRefV; | ||
139 : | const uint8_t * bRefHV; | ||
140 : | const uint8_t * b_RefCU; | ||
141 : | const uint8_t * b_RefCV; | ||
142 : | |||
143 : | VECTOR bpredMV; | ||
144 : | uint32_t bFcode; | ||
145 : | // fields for direct mode | ||
146 : | VECTOR directmvF[4]; | ||
147 : | VECTOR directmvB[4]; | ||
148 : | const VECTOR * referencemv; | ||
149 : | |||
150 : | } SearchData; | ||
151 : | |||
152 : | |||
153 : | typedef void(CheckFunc)(const int x, const int y, | ||
154 : | const int Direction, int * const dir, | ||
155 : | const SearchData * const Data); | ||
156 : | CheckFunc *CheckCandidate; | ||
157 : | |||
158 : | /* | ||
159 : | * Calculate the min/max range | ||
160 : | * relative to the _MACROBLOCK_ position | ||
161 : | */ | ||
162 : | static void __inline | ||
163 : | get_range(int32_t * const min_dx, | ||
164 : | int32_t * const max_dx, | ||
165 : | int32_t * const min_dy, | ||
166 : | int32_t * const max_dy, | ||
167 : | const uint32_t x, | ||
168 : | const uint32_t y, | ||
169 : | uint32_t block_sz, /* block dimension, 8 or 16 */ | ||
170 : | const uint32_t width, | ||
171 : | const uint32_t height, | ||
172 : | const uint32_t fcode, | ||
173 : | const int qpel, /* 1 if the resulting range should be in qpel precision; otherwise 0 */ | ||
174 : | const int rrv) | ||
175 : | { | ||
176 : | int k, m = qpel ? 4 : 2; | ||
177 : | const int search_range = 32 << (fcode - 1); | ||
178 : | int high = search_range - 1; | ||
179 : | int low = -search_range; | ||
180 : | |||
181 : | if (rrv) { | ||
182 : | high = RRV_MV_SCALEUP(high); | ||
183 : | low = RRV_MV_SCALEUP(low); | ||
184 : | block_sz *= 2; | ||
185 : | } | ||
186 : | |||
187 : | k = m * (int)(width - x * block_sz); | ||
188 : | *max_dx = MIN(high, k); | ||
189 : | k = m * (int)(height - y * block_sz); | ||
190 : | *max_dy = MIN(high, k); | ||
191 : | |||
192 : | k = -m * (int)((x+1) * block_sz); | ||
193 : | *min_dx = MAX(low, k); | ||
194 : | k = -m * (int)((y+1) * block_sz); | ||
195 : | *min_dy = MAX(low, k); | ||
196 : | } | ||
197 : | |||
198 : | typedef void MainSearchFunc(int x, int y, const SearchData * const Data, int bDirection); | ||
199 : | |||
200 : | static MainSearchFunc DiamondSearch, AdvDiamondSearch, SquareSearch; | ||
201 : | |||
202 : | static void Search8(const SearchData * const OldData, | ||
203 : | const int x, const int y, | ||
204 : | const uint32_t MotionFlags, | ||
205 : | const MBParam * const pParam, | ||
206 : | MACROBLOCK * const pMB, | ||
207 : | const MACROBLOCK * const pMBs, | ||
208 : | const int block, | ||
209 : | SearchData * const Data); | ||
210 : | |||
211 : | bool | ||
212 : | MotionEstimation(MBParam * const pParam, | ||
213 : | FRAMEINFO * const current, | ||
214 : | FRAMEINFO * const reference, | ||
215 : | const IMAGE * const pRefH, | ||
216 : | const IMAGE * const pRefV, | ||
217 : | const IMAGE * const pRefHV, | ||
218 : | const uint32_t iLimit); | ||
219 : | |||
220 : | static void | ||
221 : | SearchP(const IMAGE * const pRef, | ||
222 : | const uint8_t * const pRefH, | ||
223 : | const uint8_t * const pRefV, | ||
224 : | const uint8_t * const pRefHV, | ||
225 : | const IMAGE * const pCur, | ||
226 : | const int x, | ||
227 : | const int y, | ||
228 : | const uint32_t MotionFlags, | ||
229 : | const uint32_t GlobalFlags, | ||
230 : | const uint32_t iQuant, | ||
231 : | SearchData * const Data, | ||
232 : | const MBParam * const pParam, | ||
233 : | const MACROBLOCK * const pMBs, | ||
234 : | const MACROBLOCK * const prevMBs, | ||
235 : | int inter4v, | ||
236 : | MACROBLOCK * const pMB); | ||
237 : | |||
238 : | |||
239 : | static WARPPOINTS | ||
240 : | GlobalMotionEst(const MACROBLOCK * const pMBs, | ||
241 : | const MBParam * const pParam, | ||
242 : | const FRAMEINFO * const current, | ||
243 : | const FRAMEINFO * const reference, | ||
244 : | const IMAGE * const pRefH, | ||
245 : | const IMAGE * const pRefV, | ||
246 : | const IMAGE * const pRefHV ); | ||
247 : | |||
248 : | #define iDiamondSize 2 | ||
249 : | |||
250 : | static __inline uint32_t | ||
251 : | MakeGoodMotionFlags(const uint32_t MotionFlags, const uint32_t GlobalFlags) | ||
252 : | { | ||
253 : | uint32_t Flags = MotionFlags; | ||
254 : | |||
255 : | if (!(GlobalFlags & XVID_MODEDECISION_BITS)) | ||
256 : | Flags &= ~(QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS+EXTSEARCH_BITS); | ||
257 : | |||
258 : | if (Flags & EXTSEARCH_BITS) | ||
259 : | Flags |= HALFPELREFINE16_BITS; | ||
260 : | |||
261 : | if (Flags & EXTSEARCH_BITS && MotionFlags & PMV_EXTSEARCH8) | ||
262 : | Flags |= HALFPELREFINE8_BITS; | ||
263 : | |||
264 : | if (Flags & HALFPELREFINE16_BITS) | ||
265 : | Flags |= QUARTERPELREFINE16_BITS; | ||
266 : | |||
267 : | if (Flags & HALFPELREFINE8_BITS) { | ||
268 : | Flags |= QUARTERPELREFINE8_BITS; | ||
269 : | Flags &= ~PMV_HALFPELREFINE8; | ||
270 : | } | ||
271 : | |||
272 : | if (Flags & QUARTERPELREFINE8_BITS) | ||
273 : | Flags &= ~PMV_QUARTERPELREFINE8; | ||
274 : | |||
275 : | if (!(GlobalFlags & XVID_QUARTERPEL)) | ||
276 : | Flags &= ~(PMV_QUARTERPELREFINE16+PMV_QUARTERPELREFINE8+QUARTERPELREFINE16_BITS+QUARTERPELREFINE8_BITS); | ||
277 : | |||
278 : | if (!(GlobalFlags & XVID_HALFPEL)) | ||
279 : | Flags &= ~(PMV_EXTSEARCH16+PMV_HALFPELREFINE16+PMV_HALFPELREFINE8+HALFPELREFINE16_BITS+HALFPELREFINE8_BITS); | ||
280 : | |||
281 : | if (GlobalFlags & (XVID_GREYSCALE + XVID_REDUCED)) | ||
282 : | Flags &= ~(PMV_CHROMA16 + PMV_CHROMA8); | ||
283 : | |||
284 : | return Flags; | ||
285 : | } | ||
286 : | |||
287 : | /* BITS mode decision and search */ | ||
288 : | |||
289 : | #include "../bitstream/zigzag.h" | ||
290 : | #include "../quant/quant_mpeg4.h" | ||
291 : | #include "../quant/quant_h263.h" | ||
292 : | #include "../bitstream/vlc_codes.h" | ||
293 : | |||
294 : | static int | ||
295 : | CountMBBitsInter(SearchData * const Data, | ||
296 : | const MACROBLOCK * const pMBs, const int x, const int y, | ||
297 : | const MBParam * const pParam, | ||
298 : | const uint32_t MotionFlags); | ||
299 : | |||
300 : | static int | ||
301 : | CountMBBitsInter4v(const SearchData * const Data, | ||
302 : | MACROBLOCK * const pMB, const MACROBLOCK * const pMBs, | ||
303 : | const int x, const int y, | ||
304 : | const MBParam * const pParam, const uint32_t MotionFlags, | ||
305 : | const VECTOR * const backup); | ||
306 : | |||
307 : | static int | ||
308 : | CountMBBitsIntra(const SearchData * const Data); | ||
309 : | |||
310 : | int CodeCoeffIntra_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); | ||
311 : | int CodeCoeffInter_CalcBits(const int16_t qcoeff[64], const uint16_t * zigzag); | ||
312 : | |||
313 : | #endif /* _MOTION_EST_H_ */ |
No admin address has been configured | ViewVC Help |
Powered by ViewVC 1.0.4 |