Mercurial > libavcodec.hg
annotate motion_est.c @ 280:3dc1ca4ba717 libavcodec
fixing epzs & mpeg1 (hopefully now really ...)
author | michaelni |
---|---|
date | Fri, 22 Mar 2002 16:51:44 +0000 |
parents | 5cb2978e701f |
children | 1fc96b02142e |
rev | line source |
---|---|
0 | 1 /* |
2 * Motion estimation | |
3 * Copyright (c) 2000,2001 Gerard Lantau. | |
4 * | |
5 * | |
6 * This program is free software; you can redistribute it and/or modify | |
7 * it under the terms of the GNU General Public License as published by | |
8 * the Free Software Foundation; either version 2 of the License, or | |
9 * (at your option) any later version. | |
10 * | |
11 * This program is distributed in the hope that it will be useful, | |
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
14 * GNU General Public License for more details. | |
15 * | |
16 * You should have received a copy of the GNU General Public License | |
17 * along with this program; if not, write to the Free Software | |
18 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
19 * |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
20 * new Motion Estimation (X1/EPZS) by Michael Niedermayer <michaelni@gmx.at> |
0 | 21 */ |
22 #include <stdlib.h> | |
23 #include <stdio.h> | |
24 #include "avcodec.h" | |
25 #include "dsputil.h" | |
26 #include "mpegvideo.h" | |
27 | |
28 static void halfpel_motion_search(MpegEncContext * s, | |
29 int *mx_ptr, int *my_ptr, int dmin, | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
30 int xmin, int ymin, int xmax, int ymax, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
31 int pred_x, int pred_y); |
0 | 32 |
33 /* config it to test motion vector encoding (send random vectors) */ | |
34 //#define CONFIG_TEST_MV_ENCODE | |
35 | |
36 static int pix_sum(UINT8 * pix, int line_size) | |
37 { | |
38 int s, i, j; | |
39 | |
40 s = 0; | |
41 for (i = 0; i < 16; i++) { | |
42 for (j = 0; j < 16; j += 8) { | |
43 s += pix[0]; | |
44 s += pix[1]; | |
45 s += pix[2]; | |
46 s += pix[3]; | |
47 s += pix[4]; | |
48 s += pix[5]; | |
49 s += pix[6]; | |
50 s += pix[7]; | |
51 pix += 8; | |
52 } | |
53 pix += line_size - 16; | |
54 } | |
55 return s; | |
56 } | |
57 | |
58 static int pix_norm1(UINT8 * pix, int line_size) | |
59 { | |
60 int s, i, j; | |
61 UINT32 *sq = squareTbl + 256; | |
62 | |
63 s = 0; | |
64 for (i = 0; i < 16; i++) { | |
65 for (j = 0; j < 16; j += 8) { | |
66 s += sq[pix[0]]; | |
67 s += sq[pix[1]]; | |
68 s += sq[pix[2]]; | |
69 s += sq[pix[3]]; | |
70 s += sq[pix[4]]; | |
71 s += sq[pix[5]]; | |
72 s += sq[pix[6]]; | |
73 s += sq[pix[7]]; | |
74 pix += 8; | |
75 } | |
76 pix += line_size - 16; | |
77 } | |
78 return s; | |
79 } | |
80 | |
81 static int pix_norm(UINT8 * pix1, UINT8 * pix2, int line_size) | |
82 { | |
83 int s, i, j; | |
84 UINT32 *sq = squareTbl + 256; | |
85 | |
86 s = 0; | |
87 for (i = 0; i < 16; i++) { | |
88 for (j = 0; j < 16; j += 8) { | |
89 s += sq[pix1[0] - pix2[0]]; | |
90 s += sq[pix1[1] - pix2[1]]; | |
91 s += sq[pix1[2] - pix2[2]]; | |
92 s += sq[pix1[3] - pix2[3]]; | |
93 s += sq[pix1[4] - pix2[4]]; | |
94 s += sq[pix1[5] - pix2[5]]; | |
95 s += sq[pix1[6] - pix2[6]]; | |
96 s += sq[pix1[7] - pix2[7]]; | |
97 pix1 += 8; | |
98 pix2 += 8; | |
99 } | |
100 pix1 += line_size - 16; | |
101 pix2 += line_size - 16; | |
102 } | |
103 return s; | |
104 } | |
105 | |
106 static void no_motion_search(MpegEncContext * s, | |
107 int *mx_ptr, int *my_ptr) | |
108 { | |
109 *mx_ptr = 16 * s->mb_x; | |
110 *my_ptr = 16 * s->mb_y; | |
111 } | |
112 | |
113 static int full_motion_search(MpegEncContext * s, | |
114 int *mx_ptr, int *my_ptr, int range, | |
115 int xmin, int ymin, int xmax, int ymax) | |
116 { | |
117 int x1, y1, x2, y2, xx, yy, x, y; | |
118 int mx, my, dmin, d; | |
119 UINT8 *pix; | |
120 | |
121 xx = 16 * s->mb_x; | |
122 yy = 16 * s->mb_y; | |
123 x1 = xx - range + 1; /* we loose one pixel to avoid boundary pb with half pixel pred */ | |
124 if (x1 < xmin) | |
125 x1 = xmin; | |
126 x2 = xx + range - 1; | |
127 if (x2 > xmax) | |
128 x2 = xmax; | |
129 y1 = yy - range + 1; | |
130 if (y1 < ymin) | |
131 y1 = ymin; | |
132 y2 = yy + range - 1; | |
133 if (y2 > ymax) | |
134 y2 = ymax; | |
135 pix = s->new_picture[0] + (yy * s->linesize) + xx; | |
136 dmin = 0x7fffffff; | |
137 mx = 0; | |
138 my = 0; | |
139 for (y = y1; y <= y2; y++) { | |
140 for (x = x1; x <= x2; x++) { | |
141 d = pix_abs16x16(pix, s->last_picture[0] + (y * s->linesize) + x, | |
142 s->linesize, 16); | |
143 if (d < dmin || | |
144 (d == dmin && | |
145 (abs(x - xx) + abs(y - yy)) < | |
146 (abs(mx - xx) + abs(my - yy)))) { | |
147 dmin = d; | |
148 mx = x; | |
149 my = y; | |
150 } | |
151 } | |
152 } | |
153 | |
154 *mx_ptr = mx; | |
155 *my_ptr = my; | |
156 | |
157 #if 0 | |
158 if (*mx_ptr < -(2 * range) || *mx_ptr >= (2 * range) || | |
159 *my_ptr < -(2 * range) || *my_ptr >= (2 * range)) { | |
160 fprintf(stderr, "error %d %d\n", *mx_ptr, *my_ptr); | |
161 } | |
162 #endif | |
163 return dmin; | |
164 } | |
165 | |
166 | |
167 static int log_motion_search(MpegEncContext * s, | |
168 int *mx_ptr, int *my_ptr, int range, | |
169 int xmin, int ymin, int xmax, int ymax) | |
170 { | |
171 int x1, y1, x2, y2, xx, yy, x, y; | |
172 int mx, my, dmin, d; | |
173 UINT8 *pix; | |
174 | |
175 xx = s->mb_x << 4; | |
176 yy = s->mb_y << 4; | |
177 | |
178 /* Left limit */ | |
179 x1 = xx - range; | |
180 if (x1 < xmin) | |
181 x1 = xmin; | |
182 | |
183 /* Right limit */ | |
184 x2 = xx + range; | |
185 if (x2 > xmax) | |
186 x2 = xmax; | |
187 | |
188 /* Upper limit */ | |
189 y1 = yy - range; | |
190 if (y1 < ymin) | |
191 y1 = ymin; | |
192 | |
193 /* Lower limit */ | |
194 y2 = yy + range; | |
195 if (y2 > ymax) | |
196 y2 = ymax; | |
197 | |
198 pix = s->new_picture[0] + (yy * s->linesize) + xx; | |
199 dmin = 0x7fffffff; | |
200 mx = 0; | |
201 my = 0; | |
202 | |
203 do { | |
204 for (y = y1; y <= y2; y += range) { | |
205 for (x = x1; x <= x2; x += range) { | |
206 d = pix_abs16x16(pix, s->last_picture[0] + (y * s->linesize) + x, s->linesize, 16); | |
207 if (d < dmin || (d == dmin && (abs(x - xx) + abs(y - yy)) < (abs(mx - xx) + abs(my - yy)))) { | |
208 dmin = d; | |
209 mx = x; | |
210 my = y; | |
211 } | |
212 } | |
213 } | |
214 | |
215 range = range >> 1; | |
216 | |
217 x1 = mx - range; | |
218 if (x1 < xmin) | |
219 x1 = xmin; | |
220 | |
221 x2 = mx + range; | |
222 if (x2 > xmax) | |
223 x2 = xmax; | |
224 | |
225 y1 = my - range; | |
226 if (y1 < ymin) | |
227 y1 = ymin; | |
228 | |
229 y2 = my + range; | |
230 if (y2 > ymax) | |
231 y2 = ymax; | |
232 | |
233 } while (range >= 1); | |
234 | |
235 #ifdef DEBUG | |
236 fprintf(stderr, "log - MX: %d\tMY: %d\n", mx, my); | |
237 #endif | |
238 *mx_ptr = mx; | |
239 *my_ptr = my; | |
240 return dmin; | |
241 } | |
242 | |
243 static int phods_motion_search(MpegEncContext * s, | |
244 int *mx_ptr, int *my_ptr, int range, | |
245 int xmin, int ymin, int xmax, int ymax) | |
246 { | |
247 int x1, y1, x2, y2, xx, yy, x, y, lastx, d; | |
248 int mx, my, dminx, dminy; | |
249 UINT8 *pix; | |
250 | |
251 xx = s->mb_x << 4; | |
252 yy = s->mb_y << 4; | |
253 | |
254 /* Left limit */ | |
255 x1 = xx - range; | |
256 if (x1 < xmin) | |
257 x1 = xmin; | |
258 | |
259 /* Right limit */ | |
260 x2 = xx + range; | |
261 if (x2 > xmax) | |
262 x2 = xmax; | |
263 | |
264 /* Upper limit */ | |
265 y1 = yy - range; | |
266 if (y1 < ymin) | |
267 y1 = ymin; | |
268 | |
269 /* Lower limit */ | |
270 y2 = yy + range; | |
271 if (y2 > ymax) | |
272 y2 = ymax; | |
273 | |
274 pix = s->new_picture[0] + (yy * s->linesize) + xx; | |
275 mx = 0; | |
276 my = 0; | |
277 | |
278 x = xx; | |
279 y = yy; | |
280 do { | |
281 dminx = 0x7fffffff; | |
282 dminy = 0x7fffffff; | |
283 | |
284 lastx = x; | |
285 for (x = x1; x <= x2; x += range) { | |
286 d = pix_abs16x16(pix, s->last_picture[0] + (y * s->linesize) + x, s->linesize, 16); | |
287 if (d < dminx || (d == dminx && (abs(x - xx) + abs(y - yy)) < (abs(mx - xx) + abs(my - yy)))) { | |
288 dminx = d; | |
289 mx = x; | |
290 } | |
291 } | |
292 | |
293 x = lastx; | |
294 for (y = y1; y <= y2; y += range) { | |
295 d = pix_abs16x16(pix, s->last_picture[0] + (y * s->linesize) + x, s->linesize, 16); | |
296 if (d < dminy || (d == dminy && (abs(x - xx) + abs(y - yy)) < (abs(mx - xx) + abs(my - yy)))) { | |
297 dminy = d; | |
298 my = y; | |
299 } | |
300 } | |
301 | |
302 range = range >> 1; | |
303 | |
304 x = mx; | |
305 y = my; | |
306 x1 = mx - range; | |
307 if (x1 < xmin) | |
308 x1 = xmin; | |
309 | |
310 x2 = mx + range; | |
311 if (x2 > xmax) | |
312 x2 = xmax; | |
313 | |
314 y1 = my - range; | |
315 if (y1 < ymin) | |
316 y1 = ymin; | |
317 | |
318 y2 = my + range; | |
319 if (y2 > ymax) | |
320 y2 = ymax; | |
321 | |
322 } while (range >= 1); | |
323 | |
324 #ifdef DEBUG | |
325 fprintf(stderr, "phods - MX: %d\tMY: %d\n", mx, my); | |
326 #endif | |
327 | |
328 /* half pixel search */ | |
329 *mx_ptr = mx; | |
330 *my_ptr = my; | |
331 return dminy; | |
332 } | |
333 | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
334 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
335 #define Z_THRESHOLD 256 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
336 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
337 #define CHECK_MV(x,y)\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
338 d = pix_abs16x16(new_pic, old_pic + (x) + (y)*pic_stride, pic_stride, 16);\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
339 d += (mv_penalty[((x)<<shift)-pred_x] + mv_penalty[((y)<<shift)-pred_y])*quant;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
340 if(d<dmin){\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
341 best[0]=x;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
342 best[1]=y;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
343 dmin=d;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
344 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
345 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
346 #define CHECK_MV_DIR(x,y,new_dir)\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
347 {\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
348 d = pix_abs16x16(new_pic, old_pic + (x) + (y)*pic_stride, pic_stride, 16);\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
349 d += (mv_penalty[((x)<<shift)-pred_x] + mv_penalty[((y)<<shift)-pred_y])*quant;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
350 if(d<dmin){\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
351 best[0]=x;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
352 best[1]=y;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
353 dmin=d;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
354 next_dir= new_dir;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
355 }\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
356 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
357 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
358 static inline int small_diamond_search(MpegEncContext * s, int *best, int dmin, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
359 UINT8 *new_pic, UINT8 *old_pic, int pic_stride, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
360 int pred_x, int pred_y, UINT16 *mv_penalty, int quant, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
361 int xmin, int ymin, int xmax, int ymax, int shift) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
362 { |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
363 int next_dir=-1; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
364 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
365 for(;;){ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
366 int d; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
367 const int dir= next_dir; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
368 const int x= best[0]; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
369 const int y= best[1]; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
370 next_dir=-1; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
371 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
372 //printf("%d", dir); |
280 | 373 if(dir!=2 && x>xmin) CHECK_MV_DIR(x-1, y , 0) |
374 if(dir!=3 && y>ymin) CHECK_MV_DIR(x , y-1, 1) | |
375 if(dir!=0 && x<xmax) CHECK_MV_DIR(x+1, y , 2) | |
376 if(dir!=1 && y<ymax) CHECK_MV_DIR(x , y+1, 3) | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
377 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
378 if(next_dir==-1){ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
379 return dmin; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
380 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
381 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
382 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
383 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
384 static int epzs_motion_search(MpegEncContext * s, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
385 int *mx_ptr, int *my_ptr, |
280 | 386 int P[5][2], int pred_x, int pred_y, |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
387 int xmin, int ymin, int xmax, int ymax) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
388 { |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
389 int best[2]={0, 0}; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
390 int d, dmin; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
391 UINT8 *new_pic, *old_pic; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
392 const int pic_stride= s->linesize; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
393 const int pic_xy= (s->mb_y*pic_stride + s->mb_x)*16; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
394 UINT16 *mv_penalty= s->mv_penalty[s->f_code] + MAX_MV; // f_code of the prev frame |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
395 int quant= s->qscale; // qscale of the prev frame |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
396 const int shift= 1+s->quarter_sample; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
397 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
398 new_pic = s->new_picture[0] + pic_xy; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
399 old_pic = s->last_picture[0] + pic_xy; |
280 | 400 //printf("%d %d %d %d\n", xmin, ymin, xmax, ymax); |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
401 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
402 dmin = pix_abs16x16(new_pic, old_pic, pic_stride, 16); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
403 if(dmin<Z_THRESHOLD){ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
404 *mx_ptr= 0; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
405 *my_ptr= 0; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
406 //printf("Z"); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
407 return dmin; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
408 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
409 |
280 | 410 /* first line */ |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
411 if ((s->mb_y == 0 || s->first_slice_line || s->first_gob_line)) { |
280 | 412 CHECK_MV(P[1][0]>>shift, P[1][1]>>shift) |
413 }else{ | |
414 CHECK_MV(P[4][0]>>shift, P[4][1]>>shift) | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
415 if(dmin<Z_THRESHOLD){ |
280 | 416 *mx_ptr= P[4][0]>>shift; |
417 *my_ptr= P[4][1]>>shift; | |
418 //printf("M\n"); | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
419 return dmin; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
420 } |
280 | 421 CHECK_MV(P[1][0]>>shift, P[1][1]>>shift) |
422 CHECK_MV(P[2][0]>>shift, P[2][1]>>shift) | |
423 CHECK_MV(P[3][0]>>shift, P[3][1]>>shift) | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
424 } |
280 | 425 CHECK_MV(P[0][0]>>shift, P[0][1]>>shift) |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
426 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
427 dmin= small_diamond_search(s, best, dmin, new_pic, old_pic, pic_stride, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
428 pred_x, pred_y, mv_penalty, quant, xmin, ymin, xmax, ymax, shift); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
429 *mx_ptr= best[0]; |
280 | 430 *my_ptr= best[1]; |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
431 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
432 // printf("%d %d %d \n", best[0], best[1], dmin); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
433 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
434 return dmin; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
435 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
436 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
437 #define CHECK_HALF_MV(suffix, x, y) \ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
438 d= pix_abs16x16_ ## suffix(pix, ptr+((x)>>1), s->linesize, 16);\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
439 d += (mv_penalty[pen_x + x] + mv_penalty[pen_y + y])*quant;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
440 if(d<dminh){\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
441 dminh= d;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
442 mx= mx1 + x;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
443 my= my1 + y;\ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
444 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
445 |
0 | 446 /* The idea would be to make half pel ME after Inter/Intra decision to |
447 save time. */ | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
448 static inline void halfpel_motion_search(MpegEncContext * s, |
0 | 449 int *mx_ptr, int *my_ptr, int dmin, |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
450 int xmin, int ymin, int xmax, int ymax, |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
451 int pred_x, int pred_y) |
0 | 452 { |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
453 UINT16 *mv_penalty= s->mv_penalty[s->f_code] + MAX_MV; // f_code of the prev frame |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
454 const int quant= s->qscale; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
455 int pen_x, pen_y; |
0 | 456 int mx, my, mx1, my1, d, xx, yy, dminh; |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
457 UINT8 *pix, *ptr; |
0 | 458 |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
459 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
460 mx = *mx_ptr; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
461 my = *my_ptr; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
462 ptr = s->last_picture[0] + (my * s->linesize) + mx; |
0 | 463 |
464 xx = 16 * s->mb_x; | |
465 yy = 16 * s->mb_y; | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
466 pix = s->new_picture[0] + (yy * s->linesize) + xx; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
467 |
0 | 468 dminh = dmin; |
469 | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
470 if (mx > xmin && mx < xmax && |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
471 my > ymin && my < ymax) { |
0 | 472 |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
473 mx= mx1= 2*(mx - xx); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
474 my= my1= 2*(my - yy); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
475 if(dmin < Z_THRESHOLD && mx==0 && my==0){ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
476 *mx_ptr = 0; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
477 *my_ptr = 0; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
478 return; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
479 } |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
480 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
481 pen_x= pred_x + mx; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
482 pen_y= pred_y + my; |
0 | 483 |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
484 ptr-= s->linesize; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
485 CHECK_HALF_MV(xy2, -1, -1) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
486 CHECK_HALF_MV(y2 , 0, -1) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
487 CHECK_HALF_MV(xy2, +1, -1) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
488 |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
489 ptr+= s->linesize; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
490 CHECK_HALF_MV(x2 , -1, 0) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
491 CHECK_HALF_MV(x2 , +1, 0) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
492 CHECK_HALF_MV(xy2, -1, +1) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
493 CHECK_HALF_MV(y2 , 0, +1) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
494 CHECK_HALF_MV(xy2, +1, +1) |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
495 }else{ |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
496 mx= 2*(mx - xx); |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
497 my= 2*(my - yy); |
0 | 498 } |
499 | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
500 *mx_ptr = mx; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
501 *my_ptr = my; |
0 | 502 } |
503 | |
504 #ifndef CONFIG_TEST_MV_ENCODE | |
505 | |
506 int estimate_motion(MpegEncContext * s, | |
507 int mb_x, int mb_y, | |
508 int *mx_ptr, int *my_ptr) | |
509 { | |
510 UINT8 *pix, *ppix; | |
511 int sum, varc, vard, mx, my, range, dmin, xx, yy; | |
512 int xmin, ymin, xmax, ymax; | |
280 | 513 int rel_xmin, rel_ymin, rel_xmax, rel_ymax; |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
514 int pred_x=0, pred_y=0; |
280 | 515 int P[5][2]; |
516 const int shift= 1+s->quarter_sample; | |
233
3f5b72726118
- More work on preliminary bit rate control, just to be able to get an
pulento
parents:
232
diff
changeset
|
517 |
0 | 518 range = 8 * (1 << (s->f_code - 1)); |
519 /* XXX: temporary kludge to avoid overflow for msmpeg4 */ | |
520 if (s->out_format == FMT_H263 && !s->h263_msmpeg4) | |
521 range = range * 2; | |
522 | |
523 if (s->unrestricted_mv) { | |
524 xmin = -16; | |
525 ymin = -16; | |
275
7ebb3f9aaf3b
- Added video coding statistics for ffmpeg. Needs more work.
pulento
parents:
268
diff
changeset
|
526 if (s->h263_plus) |
7ebb3f9aaf3b
- Added video coding statistics for ffmpeg. Needs more work.
pulento
parents:
268
diff
changeset
|
527 range *= 2; |
218 | 528 if(s->avctx==NULL || s->avctx->codec->id!=CODEC_ID_MPEG4){ |
529 xmax = s->mb_width*16; | |
530 ymax = s->mb_height*16; | |
531 }else { | |
532 /* XXX: dunno if this is correct but ffmpeg4 decoder wont like it otherwise | |
533 (cuz the drawn edge isnt large enough))*/ | |
534 xmax = s->width; | |
535 ymax = s->height; | |
232
b640ec5948b0
- Now the ME is done for the entire picture when enconding, the
pulento
parents:
218
diff
changeset
|
536 } |
0 | 537 } else { |
538 xmin = 0; | |
539 ymin = 0; | |
218 | 540 xmax = s->mb_width*16 - 16; |
541 ymax = s->mb_height*16 - 16; | |
0 | 542 } |
543 switch(s->full_search) { | |
544 case ME_ZERO: | |
545 default: | |
546 no_motion_search(s, &mx, &my); | |
547 dmin = 0; | |
548 break; | |
549 case ME_FULL: | |
550 dmin = full_motion_search(s, &mx, &my, range, xmin, ymin, xmax, ymax); | |
551 break; | |
552 case ME_LOG: | |
553 dmin = log_motion_search(s, &mx, &my, range / 2, xmin, ymin, xmax, ymax); | |
554 break; | |
555 case ME_PHODS: | |
556 dmin = phods_motion_search(s, &mx, &my, range / 2, xmin, ymin, xmax, ymax); | |
557 break; | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
558 case ME_X1: // just reserving some space for experiments ... |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
559 case ME_EPZS: |
280 | 560 rel_xmin= xmin - s->mb_x*16; |
561 rel_xmax= xmax - s->mb_x*16; | |
562 rel_ymin= ymin - s->mb_y*16; | |
563 rel_ymax= ymax - s->mb_y*16; | |
564 if(s->out_format == FMT_H263){ | |
565 static const int off[4]= {2, 1, 1, -1}; | |
566 const int mot_stride = s->block_wrap[0]; | |
567 const int mot_xy = s->block_index[0]; | |
568 | |
569 P[0][0] = s->motion_val[mot_xy ][0]; | |
570 P[0][1] = s->motion_val[mot_xy ][1]; | |
571 P[1][0] = s->motion_val[mot_xy - 1][0]; | |
572 P[1][1] = s->motion_val[mot_xy - 1][1]; | |
573 if(P[1][0] > (rel_xmax<<shift)) P[1][0]= (rel_xmax<<shift); | |
574 | |
575 /* special case for first line */ | |
576 if ((s->mb_y == 0 || s->first_slice_line || s->first_gob_line)) { | |
577 pred_x = P[1][0]; | |
578 pred_y = P[1][1]; | |
579 } else { | |
580 P[2][0] = s->motion_val[mot_xy - mot_stride ][0]; | |
581 P[2][1] = s->motion_val[mot_xy - mot_stride ][1]; | |
582 P[3][0] = s->motion_val[mot_xy - mot_stride + off[0] ][0]; | |
583 P[3][1] = s->motion_val[mot_xy - mot_stride + off[0] ][1]; | |
584 if(P[2][1] > (rel_ymax<<shift)) P[2][1]= (rel_ymax<<shift); | |
585 if(P[3][0] < (rel_xmin<<shift)) P[3][0]= (rel_xmin<<shift); | |
586 if(P[3][1] > (rel_ymax<<shift)) P[3][1]= (rel_ymax<<shift); | |
587 | |
588 P[4][0]= pred_x = mid_pred(P[1][0], P[2][0], P[3][0]); | |
589 P[4][1]= pred_y = mid_pred(P[1][1], P[2][1], P[3][1]); | |
590 } | |
591 }else { | |
592 const int xy= s->mb_y*s->mb_width + s->mb_x; | |
593 pred_x= s->last_mv[0][0][0]; | |
594 pred_y= s->last_mv[0][0][1]; | |
595 | |
596 P[0][0]= s->mv_table[0][xy ]; | |
597 P[0][1]= s->mv_table[1][xy ]; | |
598 if(s->mb_x == 0){ | |
599 P[1][0]= 0; | |
600 P[1][1]= 0; | |
601 }else{ | |
602 P[1][0]= s->mv_table[0][xy-1]; | |
603 P[1][1]= s->mv_table[1][xy-1]; | |
604 if(P[1][0] > (rel_xmax<<shift)) P[1][0]= (rel_xmax<<shift); | |
605 } | |
606 | |
607 if (!(s->mb_y == 0 || s->first_slice_line || s->first_gob_line)) { | |
608 P[2][0] = s->mv_table[0][xy - s->mb_width]; | |
609 P[2][1] = s->mv_table[1][xy - s->mb_width]; | |
610 P[3][0] = s->mv_table[0][xy - s->mb_width+1]; | |
611 P[3][1] = s->mv_table[1][xy - s->mb_width+1]; | |
612 if(P[2][1] > (rel_ymax<<shift)) P[2][1]= (rel_ymax<<shift); | |
613 if(P[3][0] > (rel_xmax<<shift)) P[3][0]= (rel_xmax<<shift); | |
614 if(P[3][0] < (rel_xmin<<shift)) P[3][0]= (rel_xmin<<shift); | |
615 if(P[3][1] > (rel_ymax<<shift)) P[3][1]= (rel_ymax<<shift); | |
616 | |
617 P[4][0]= mid_pred(P[1][0], P[2][0], P[3][0]); | |
618 P[4][1]= mid_pred(P[1][1], P[2][1], P[3][1]); | |
619 } | |
620 } | |
621 dmin = epzs_motion_search(s, &mx, &my, P, pred_x, pred_y, rel_xmin, rel_ymin, rel_xmax, rel_ymax); | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
622 mx+= s->mb_x*16; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
623 my+= s->mb_y*16; |
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
624 break; |
0 | 625 } |
626 | |
627 /* intra / predictive decision */ | |
628 xx = mb_x * 16; | |
629 yy = mb_y * 16; | |
630 | |
631 pix = s->new_picture[0] + (yy * s->linesize) + xx; | |
632 /* At this point (mx,my) are full-pell and the absolute displacement */ | |
633 ppix = s->last_picture[0] + (my * s->linesize) + mx; | |
634 | |
635 sum = pix_sum(pix, s->linesize); | |
636 varc = pix_norm1(pix, s->linesize); | |
637 vard = pix_norm(pix, ppix, s->linesize); | |
638 | |
639 vard = vard >> 8; | |
640 sum = sum >> 8; | |
641 varc = (varc >> 8) - (sum * sum); | |
239 | 642 s->mb_var[s->mb_width * mb_y + mb_x] = varc; |
233
3f5b72726118
- More work on preliminary bit rate control, just to be able to get an
pulento
parents:
232
diff
changeset
|
643 s->avg_mb_var += varc; |
268 | 644 s->mc_mb_var += vard; |
233
3f5b72726118
- More work on preliminary bit rate control, just to be able to get an
pulento
parents:
232
diff
changeset
|
645 |
0 | 646 #if 0 |
233
3f5b72726118
- More work on preliminary bit rate control, just to be able to get an
pulento
parents:
232
diff
changeset
|
647 printf("varc=%4d avg_var=%4d (sum=%4d) vard=%4d mx=%2d my=%2d\n", |
3f5b72726118
- More work on preliminary bit rate control, just to be able to get an
pulento
parents:
232
diff
changeset
|
648 varc, s->avg_mb_var, sum, vard, mx - xx, my - yy); |
0 | 649 #endif |
650 if (vard <= 64 || vard < varc) { | |
651 if (s->full_search != ME_ZERO) { | |
277
5cb2978e701f
new motion estimation (epzs) not complete yet but allready pretty good :)
michaelni
parents:
275
diff
changeset
|
652 halfpel_motion_search(s, &mx, &my, dmin, xmin, ymin, xmax, ymax, pred_x, pred_y); |
0 | 653 } else { |
654 mx -= 16 * s->mb_x; | |
655 my -= 16 * s->mb_y; | |
656 } | |
657 *mx_ptr = mx; | |
658 *my_ptr = my; | |
659 return 0; | |
660 } else { | |
661 *mx_ptr = 0; | |
662 *my_ptr = 0; | |
663 return 1; | |
664 } | |
665 } | |
666 | |
667 #else | |
668 | |
669 /* test version which generates valid random vectors */ | |
670 int estimate_motion(MpegEncContext * s, | |
671 int mb_x, int mb_y, | |
672 int *mx_ptr, int *my_ptr) | |
673 { | |
674 int xx, yy, x1, y1, x2, y2, range; | |
675 | |
676 if ((random() % 10) >= 5) { | |
677 range = 8 * (1 << (s->f_code - 1)); | |
678 if (s->out_format == FMT_H263 && !s->h263_msmpeg4) | |
679 range = range * 2; | |
680 | |
681 xx = 16 * s->mb_x; | |
682 yy = 16 * s->mb_y; | |
683 x1 = xx - range; | |
684 if (x1 < 0) | |
685 x1 = 0; | |
686 x2 = xx + range - 1; | |
687 if (x2 > (s->width - 16)) | |
688 x2 = s->width - 16; | |
689 y1 = yy - range; | |
690 if (y1 < 0) | |
691 y1 = 0; | |
692 y2 = yy + range - 1; | |
693 if (y2 > (s->height - 16)) | |
694 y2 = s->height - 16; | |
695 | |
696 *mx_ptr = (random() % (2 * (x2 - x1 + 1))) + 2 * (x1 - xx); | |
697 *my_ptr = (random() % (2 * (y2 - y1 + 1))) + 2 * (y1 - yy); | |
698 return 0; | |
699 } else { | |
700 *mx_ptr = 0; | |
701 *my_ptr = 0; | |
702 return 1; | |
703 } | |
704 } | |
705 | |
706 #endif |