Details | Last modification | View Log | RSS feed
Rev | Author | Line No. | Line |
---|---|---|---|
1085 | pj | 1 | /* Predict.c, motion compensation routines */ |
2 | |||
3 | /* Copyright (C) 1996, MPEG Software Simulation Group. All Rights Reserved. */ |
||
4 | |||
5 | /* |
||
6 | * Disclaimer of Warranty |
||
7 | * |
||
8 | * These software programs are available to the user without any license fee or |
||
9 | * royalty on an "as is" basis. The MPEG Software Simulation Group disclaims |
||
10 | * any and all warranties, whether express, implied, or statuary, including any |
||
11 | * implied warranties or merchantability or of fitness for a particular |
||
12 | * purpose. In no event shall the copyright-holder be liable for any |
||
13 | * incidental, punitive, or consequential damages of any kind whatsoever |
||
14 | * arising from the use of these programs. |
||
15 | * |
||
16 | * This disclaimer of warranty extends to the user of these programs and user's |
||
17 | * customers, employees, agents, transferees, successors, and assigns. |
||
18 | * |
||
19 | * The MPEG Software Simulation Group does not represent or warrant that the |
||
20 | * programs furnished hereunder are free of infringement of any third-party |
||
21 | * patents. |
||
22 | * |
||
23 | * Commercial implementations of MPEG-1 and MPEG-2 video, including shareware, |
||
24 | * are subject to royalty fees to patent holders. Many of these patents are |
||
25 | * general enough such that they are unavoidable regardless of implementation |
||
26 | * design. |
||
27 | * |
||
28 | */ |
||
29 | |||
30 | #include <stdio.h> |
||
31 | |||
32 | #include "config.h" |
||
33 | #include "global.h" |
||
34 | |||
35 | /* private prototypes */ |
||
36 | static void form_prediction _ANSI_ARGS_((unsigned char *src[], int sfield, |
||
37 | unsigned char *dst[], int dfield, |
||
38 | int lx, int lx2, int w, int h, int x, int y, int dx, int dy, |
||
39 | int average_flag)); |
||
40 | |||
41 | static void form_component_prediction _ANSI_ARGS_((unsigned char *src, unsigned char *dst, |
||
42 | int lx, int lx2, int w, int h, int x, int y, int dx, int dy, int average_flag)); |
||
43 | |||
44 | void form_predictions(bx,by,macroblock_type,motion_type,PMV,motion_vertical_field_select,dmvector,stwtype) |
||
45 | int bx, by; |
||
46 | int macroblock_type; |
||
47 | int motion_type; |
||
48 | int PMV[2][2][2], motion_vertical_field_select[2][2], dmvector[2]; |
||
49 | int stwtype; |
||
50 | { |
||
51 | int currentfield; |
||
52 | unsigned char **predframe; |
||
53 | int DMV[2][2]; |
||
54 | int stwtop, stwbot; |
||
55 | |||
56 | stwtop = stwtype%3; /* 0:temporal, 1:(spat+temp)/2, 2:spatial */ |
||
57 | stwbot = stwtype/3; |
||
58 | |||
59 | if ((macroblock_type & MACROBLOCK_MOTION_FORWARD) |
||
60 | || (picture_coding_type==P_TYPE)) |
||
61 | { |
||
62 | if (picture_structure==FRAME_PICTURE) |
||
63 | { |
||
64 | if ((motion_type==MC_FRAME) |
||
65 | || !(macroblock_type & MACROBLOCK_MOTION_FORWARD)) |
||
66 | { |
||
67 | /* frame-based prediction (broken into top and bottom halves |
||
68 | for spatial scalability prediction purposes) */ |
||
69 | if (stwtop<2) |
||
70 | form_prediction(forward_reference_frame,0,current_frame,0, |
||
71 | Coded_Picture_Width,Coded_Picture_Width<<1,16,8,bx,by, |
||
72 | PMV[0][0][0],PMV[0][0][1],stwtop); |
||
73 | |||
74 | if (stwbot<2) |
||
75 | form_prediction(forward_reference_frame,1,current_frame,1, |
||
76 | Coded_Picture_Width,Coded_Picture_Width<<1,16,8,bx,by, |
||
77 | PMV[0][0][0],PMV[0][0][1],stwbot); |
||
78 | } |
||
79 | else if (motion_type==MC_FIELD) /* field-based prediction */ |
||
80 | { |
||
81 | /* top field prediction */ |
||
82 | if (stwtop<2) |
||
83 | form_prediction(forward_reference_frame,motion_vertical_field_select[0][0], |
||
84 | current_frame,0,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8, |
||
85 | bx,by>>1,PMV[0][0][0],PMV[0][0][1]>>1,stwtop); |
||
86 | |||
87 | /* bottom field prediction */ |
||
88 | if (stwbot<2) |
||
89 | form_prediction(forward_reference_frame,motion_vertical_field_select[1][0], |
||
90 | current_frame,1,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8, |
||
91 | bx,by>>1,PMV[1][0][0],PMV[1][0][1]>>1,stwbot); |
||
92 | } |
||
93 | else if (motion_type==MC_DMV) /* dual prime prediction */ |
||
94 | { |
||
95 | /* calculate derived motion vectors */ |
||
96 | Dual_Prime_Arithmetic(DMV,dmvector,PMV[0][0][0],PMV[0][0][1]>>1); |
||
97 | |||
98 | if (stwtop<2) |
||
99 | { |
||
100 | /* predict top field from top field */ |
||
101 | form_prediction(forward_reference_frame,0,current_frame,0, |
||
102 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8,bx,by>>1, |
||
103 | PMV[0][0][0],PMV[0][0][1]>>1,0); |
||
104 | |||
105 | /* predict and add to top field from bottom field */ |
||
106 | form_prediction(forward_reference_frame,1,current_frame,0, |
||
107 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8,bx,by>>1, |
||
108 | DMV[0][0],DMV[0][1],1); |
||
109 | } |
||
110 | |||
111 | if (stwbot<2) |
||
112 | { |
||
113 | /* predict bottom field from bottom field */ |
||
114 | form_prediction(forward_reference_frame,1,current_frame,1, |
||
115 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8,bx,by>>1, |
||
116 | PMV[0][0][0],PMV[0][0][1]>>1,0); |
||
117 | |||
118 | /* predict and add to bottom field from top field */ |
||
119 | form_prediction(forward_reference_frame,0,current_frame,1, |
||
120 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8,bx,by>>1, |
||
121 | DMV[1][0],DMV[1][1],1); |
||
122 | } |
||
123 | } |
||
124 | else |
||
125 | /* invalid motion_type */ |
||
126 | printf("invalid motion_type\n"); |
||
127 | } |
||
128 | else /* TOP_FIELD or BOTTOM_FIELD */ |
||
129 | { |
||
130 | /* field picture */ |
||
131 | currentfield = (picture_structure==BOTTOM_FIELD); |
||
132 | |||
133 | /* determine which frame to use for prediction */ |
||
134 | if ((picture_coding_type==P_TYPE) && Second_Field |
||
135 | && (currentfield!=motion_vertical_field_select[0][0])) |
||
136 | predframe = backward_reference_frame; /* same frame */ |
||
137 | else |
||
138 | predframe = forward_reference_frame; /* previous frame */ |
||
139 | |||
140 | if ((motion_type==MC_FIELD) |
||
141 | || !(macroblock_type & MACROBLOCK_MOTION_FORWARD)) |
||
142 | { |
||
143 | /* field-based prediction */ |
||
144 | if (stwtop<2) |
||
145 | form_prediction(predframe,motion_vertical_field_select[0][0],current_frame,0, |
||
146 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,16,bx,by, |
||
147 | PMV[0][0][0],PMV[0][0][1],stwtop); |
||
148 | } |
||
149 | else if (motion_type==MC_16X8) |
||
150 | { |
||
151 | if (stwtop<2) |
||
152 | { |
||
153 | form_prediction(predframe,motion_vertical_field_select[0][0],current_frame,0, |
||
154 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8,bx,by, |
||
155 | PMV[0][0][0],PMV[0][0][1],stwtop); |
||
156 | |||
157 | /* determine which frame to use for lower half prediction */ |
||
158 | if ((picture_coding_type==P_TYPE) && Second_Field |
||
159 | && (currentfield!=motion_vertical_field_select[1][0])) |
||
160 | predframe = backward_reference_frame; /* same frame */ |
||
161 | else |
||
162 | predframe = forward_reference_frame; /* previous frame */ |
||
163 | |||
164 | form_prediction(predframe,motion_vertical_field_select[1][0],current_frame,0, |
||
165 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8,bx,by+8, |
||
166 | PMV[1][0][0],PMV[1][0][1],stwtop); |
||
167 | } |
||
168 | } |
||
169 | else if (motion_type==MC_DMV) /* dual prime prediction */ |
||
170 | { |
||
171 | if (Second_Field) |
||
172 | predframe = backward_reference_frame; /* same frame */ |
||
173 | else |
||
174 | predframe = forward_reference_frame; /* previous frame */ |
||
175 | |||
176 | /* calculate derived motion vectors */ |
||
177 | Dual_Prime_Arithmetic(DMV,dmvector,PMV[0][0][0],PMV[0][0][1]); |
||
178 | |||
179 | /* predict from field of same parity */ |
||
180 | form_prediction(forward_reference_frame,currentfield,current_frame,0, |
||
181 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,16,bx,by, |
||
182 | PMV[0][0][0],PMV[0][0][1],0); |
||
183 | |||
184 | /* predict from field of opposite parity */ |
||
185 | form_prediction(predframe,!currentfield,current_frame,0, |
||
186 | Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,16,bx,by, |
||
187 | DMV[0][0],DMV[0][1],1); |
||
188 | } |
||
189 | else |
||
190 | /* invalid motion_type */ |
||
191 | printf("invalid motion_type\n"); |
||
192 | } |
||
193 | stwtop = stwbot = 1; |
||
194 | } |
||
195 | |||
196 | if (macroblock_type & MACROBLOCK_MOTION_BACKWARD) |
||
197 | { |
||
198 | if (picture_structure==FRAME_PICTURE) |
||
199 | { |
||
200 | if (motion_type==MC_FRAME) |
||
201 | { |
||
202 | /* frame-based prediction */ |
||
203 | if (stwtop<2) |
||
204 | form_prediction(backward_reference_frame,0,current_frame,0, |
||
205 | Coded_Picture_Width,Coded_Picture_Width<<1,16,8,bx,by, |
||
206 | PMV[0][1][0],PMV[0][1][1],stwtop); |
||
207 | |||
208 | if (stwbot<2) |
||
209 | form_prediction(backward_reference_frame,1,current_frame,1, |
||
210 | Coded_Picture_Width,Coded_Picture_Width<<1,16,8,bx,by, |
||
211 | PMV[0][1][0],PMV[0][1][1],stwbot); |
||
212 | } |
||
213 | else /* field-based prediction */ |
||
214 | { |
||
215 | /* top field prediction */ |
||
216 | if (stwtop<2) |
||
217 | form_prediction(backward_reference_frame,motion_vertical_field_select[0][1], |
||
218 | current_frame,0,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8, |
||
219 | bx,by>>1,PMV[0][1][0],PMV[0][1][1]>>1,stwtop); |
||
220 | |||
221 | /* bottom field prediction */ |
||
222 | if (stwbot<2) |
||
223 | form_prediction(backward_reference_frame,motion_vertical_field_select[1][1], |
||
224 | current_frame,1,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8, |
||
225 | bx,by>>1,PMV[1][1][0],PMV[1][1][1]>>1,stwbot); |
||
226 | } |
||
227 | } |
||
228 | else /* TOP_FIELD or BOTTOM_FIELD */ |
||
229 | { |
||
230 | /* field picture */ |
||
231 | if (motion_type==MC_FIELD) |
||
232 | { |
||
233 | /* field-based prediction */ |
||
234 | form_prediction(backward_reference_frame,motion_vertical_field_select[0][1], |
||
235 | current_frame,0,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,16, |
||
236 | bx,by,PMV[0][1][0],PMV[0][1][1],stwtop); |
||
237 | } |
||
238 | else if (motion_type==MC_16X8) |
||
239 | { |
||
240 | form_prediction(backward_reference_frame,motion_vertical_field_select[0][1], |
||
241 | current_frame,0,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8, |
||
242 | bx,by,PMV[0][1][0],PMV[0][1][1],stwtop); |
||
243 | |||
244 | form_prediction(backward_reference_frame,motion_vertical_field_select[1][1], |
||
245 | current_frame,0,Coded_Picture_Width<<1,Coded_Picture_Width<<1,16,8, |
||
246 | bx,by+8,PMV[1][1][0],PMV[1][1][1],stwtop); |
||
247 | } |
||
248 | else |
||
249 | /* invalid motion_type */ |
||
250 | printf("invalid motion_type\n"); |
||
251 | } |
||
252 | } |
||
253 | } |
||
254 | |||
255 | static void form_prediction(src,sfield,dst,dfield,lx,lx2,w,h,x,y,dx,dy,average_flag) |
||
256 | unsigned char *src[]; /* prediction source buffer */ |
||
257 | int sfield; /* prediction source field number (0 or 1) */ |
||
258 | unsigned char *dst[]; /* prediction destination buffer */ |
||
259 | int dfield; /* prediction destination field number (0 or 1)*/ |
||
260 | int lx,lx2; /* line strides */ |
||
261 | int w,h; /* prediction block/sub-block width, height */ |
||
262 | int x,y; /* pixel co-ordinates of top-left sample in current MB */ |
||
263 | int dx,dy; /* horizontal, vertical prediction address */ |
||
264 | int average_flag; /* add prediction error to prediction ? */ |
||
265 | { |
||
266 | /* Y */ |
||
267 | form_component_prediction(src[0]+(sfield?lx2>>1:0),dst[0]+(dfield?lx2>>1:0), |
||
268 | lx,lx2,w,h,x,y,dx,dy,average_flag); |
||
269 | |||
270 | if (chroma_format!=CHROMA444) |
||
271 | { |
||
272 | lx>>=1; lx2>>=1; w>>=1; x>>=1; dx/=2; |
||
273 | } |
||
274 | |||
275 | if (chroma_format==CHROMA420) |
||
276 | { |
||
277 | h>>=1; y>>=1; dy/=2; |
||
278 | } |
||
279 | |||
280 | /* Cb */ |
||
281 | form_component_prediction(src[1]+(sfield?lx2>>1:0),dst[1]+(dfield?lx2>>1:0), |
||
282 | lx,lx2,w,h,x,y,dx,dy,average_flag); |
||
283 | |||
284 | /* Cr */ |
||
285 | form_component_prediction(src[2]+(sfield?lx2>>1:0),dst[2]+(dfield?lx2>>1:0), |
||
286 | lx,lx2,w,h,x,y,dx,dy,average_flag); |
||
287 | } |
||
288 | |||
289 | /* ISO/IEC 13818-2 section 7.6.4: Forming predictions */ |
||
290 | /* NOTE: the arithmetic below produces numerically equivalent results |
||
291 | * to 7.6.4, yet is more elegant. It differs in the following ways: |
||
292 | * |
||
293 | * 1. the vectors (dx, dy) are based on cartesian frame |
||
294 | * coordiantes along a half-pel grid (always positive numbers) |
||
295 | * In contrast, vector[r][s][t] are differential (with positive and |
||
296 | * negative values). As a result, deriving the integer vectors |
||
297 | * (int_vec[t]) from dx, dy is accomplished by a simple right shift. |
||
298 | * |
||
299 | * 2. Half pel flags (xh, yh) are equivalent to the LSB (Least |
||
300 | * Significant Bit) of the half-pel coordinates (dx,dy). |
||
301 | * |
||
302 | * |
||
303 | * NOTE: the work of combining predictions (ISO/IEC 13818-2 section 7.6.7) |
||
304 | * is distributed among several other stages. This is accomplished by |
||
305 | * folding line offsets into the source and destination (src,dst) |
||
306 | * addresses (note the call arguments to form_prediction() in Predict()), |
||
307 | * line stride variables lx and lx2, the block dimension variables (w,h), |
||
308 | * average_flag, and by the very order in which Predict() is called. |
||
309 | * This implementation design (implicitly different than the spec) |
||
310 | * was chosen for its elegance. |
||
311 | */ |
||
312 | |||
313 | static void form_component_prediction(src,dst,lx,lx2,w,h,x,y,dx,dy,average_flag) |
||
314 | unsigned char *src; |
||
315 | unsigned char *dst; |
||
316 | int lx; /* raster line increment */ |
||
317 | int lx2; |
||
318 | int w,h; |
||
319 | int x,y; |
||
320 | int dx,dy; |
||
321 | int average_flag; /* flag that signals bi-directional or Dual-Prime |
||
322 | averaging (7.6.7.1 and 7.6.7.4). if average_flag==1, |
||
323 | a previously formed prediction has been stored in |
||
324 | pel_pred[] */ |
||
325 | { |
||
326 | int xint; /* horizontal integer sample vector: analogous to int_vec[0] */ |
||
327 | int yint; /* vertical integer sample vectors: analogous to int_vec[1] */ |
||
328 | int xh; /* horizontal half sample flag: analogous to half_flag[0] */ |
||
329 | int yh; /* vertical half sample flag: analogous to half_flag[1] */ |
||
330 | int i, j, v; |
||
331 | unsigned char *s; /* source pointer: analogous to pel_ref[][] */ |
||
332 | unsigned char *d; /* destination pointer: analogous to pel_pred[][] */ |
||
333 | |||
334 | /* half pel scaling for integer vectors */ |
||
335 | xint = dx>>1; |
||
336 | yint = dy>>1; |
||
337 | |||
338 | /* derive half pel flags */ |
||
339 | xh = dx & 1; |
||
340 | yh = dy & 1; |
||
341 | |||
342 | /* compute the linear address of pel_ref[][] and pel_pred[][] |
||
343 | based on cartesian/raster cordinates provided */ |
||
344 | s = src + lx*(y+yint) + x + xint; |
||
345 | d = dst + lx*y + x; |
||
346 | |||
347 | if (!xh && !yh) /* no horizontal nor vertical half-pel */ |
||
348 | { |
||
349 | if (average_flag) |
||
350 | { |
||
351 | for (j=0; j<h; j++) |
||
352 | { |
||
353 | for (i=0; i<w; i++) |
||
354 | { |
||
355 | v = d[i]+s[i]; |
||
356 | d[i] = (v+(v>=0?1:0))>>1; |
||
357 | } |
||
358 | |||
359 | s+= lx2; |
||
360 | d+= lx2; |
||
361 | } |
||
362 | } |
||
363 | else |
||
364 | { |
||
365 | for (j=0; j<h; j++) |
||
366 | { |
||
367 | for (i=0; i<w; i++) |
||
368 | { |
||
369 | d[i] = s[i]; |
||
370 | } |
||
371 | |||
372 | s+= lx2; |
||
373 | d+= lx2; |
||
374 | } |
||
375 | } |
||
376 | } |
||
377 | else if (!xh && yh) /* no horizontal but vertical half-pel */ |
||
378 | { |
||
379 | if (average_flag) |
||
380 | { |
||
381 | for (j=0; j<h; j++) |
||
382 | { |
||
383 | for (i=0; i<w; i++) |
||
384 | { |
||
385 | v = d[i] + ((unsigned int)(s[i]+s[i+lx]+1)>>1); |
||
386 | d[i]=(v+(v>=0?1:0))>>1; |
||
387 | } |
||
388 | |||
389 | s+= lx2; |
||
390 | d+= lx2; |
||
391 | } |
||
392 | } |
||
393 | else |
||
394 | { |
||
395 | for (j=0; j<h; j++) |
||
396 | { |
||
397 | for (i=0; i<w; i++) |
||
398 | { |
||
399 | d[i] = (unsigned int)(s[i]+s[i+lx]+1)>>1; |
||
400 | } |
||
401 | |||
402 | s+= lx2; |
||
403 | d+= lx2; |
||
404 | } |
||
405 | } |
||
406 | } |
||
407 | else if (xh && !yh) /* horizontal but no vertical half-pel */ |
||
408 | { |
||
409 | if (average_flag) |
||
410 | { |
||
411 | for (j=0; j<h; j++) |
||
412 | { |
||
413 | for (i=0; i<w; i++) |
||
414 | { |
||
415 | v = d[i] + ((unsigned int)(s[i]+s[i+1]+1)>>1); |
||
416 | d[i] = (v+(v>=0?1:0))>>1; |
||
417 | } |
||
418 | |||
419 | s+= lx2; |
||
420 | d+= lx2; |
||
421 | } |
||
422 | } |
||
423 | else |
||
424 | { |
||
425 | for (j=0; j<h; j++) |
||
426 | { |
||
427 | for (i=0; i<w; i++) |
||
428 | { |
||
429 | d[i] = (unsigned int)(s[i]+s[i+1]+1)>>1; |
||
430 | } |
||
431 | |||
432 | s+= lx2; |
||
433 | d+= lx2; |
||
434 | } |
||
435 | } |
||
436 | } |
||
437 | else /* if (xh && yh) horizontal and vertical half-pel */ |
||
438 | { |
||
439 | if (average_flag) |
||
440 | { |
||
441 | for (j=0; j<h; j++) |
||
442 | { |
||
443 | for (i=0; i<w; i++) |
||
444 | { |
||
445 | v = d[i] + ((unsigned int)(s[i]+s[i+1]+s[i+lx]+s[i+lx+1]+2)>>2); |
||
446 | d[i] = (v+(v>=0?1:0))>>1; |
||
447 | } |
||
448 | |||
449 | s+= lx2; |
||
450 | d+= lx2; |
||
451 | } |
||
452 | } |
||
453 | else |
||
454 | { |
||
455 | for (j=0; j<h; j++) |
||
456 | { |
||
457 | for (i=0; i<w; i++) |
||
458 | { |
||
459 | d[i] = (unsigned int)(s[i]+s[i+1]+s[i+lx]+s[i+lx+1]+2)>>2; |
||
460 | } |
||
461 | |||
462 | s+= lx2; |
||
463 | d+= lx2; |
||
464 | } |
||
465 | } |
||
466 | } |
||
467 | } |