Parent Directory | Revision Log
Revision 229 - (view) (download)
1 : | edgomez | 195 | // 01.05.2002 updated MBMotionCompensationBVOP |
2 : | // 14.04.2002 bframe compensation | ||
3 : | suxen_drol | 118 | |
4 : | Isibaar | 3 | #include "../encoder.h" |
5 : | #include "../utils/mbfunctions.h" | ||
6 : | #include "../image/interpolate8x8.h" | ||
7 : | #include "../utils/timer.h" | ||
8 : | suxen_drol | 118 | #include "motion.h" |
9 : | Isibaar | 3 | |
10 : | suxen_drol | 229 | #undef BFRAMES |
11 : | |||
12 : | Isibaar | 3 | #define ABS(X) (((X)>0)?(X):-(X)) |
13 : | #define SIGN(X) (((X)>0)?1:-1) | ||
14 : | |||
15 : | edgomez | 195 | static __inline void |
16 : | compensate8x8_halfpel(int16_t * const dct_codes, | ||
17 : | uint8_t * const cur, | ||
18 : | const uint8_t * const ref, | ||
19 : | const uint8_t * const refh, | ||
20 : | const uint8_t * const refv, | ||
21 : | const uint8_t * const refhv, | ||
22 : | const uint32_t x, | ||
23 : | const uint32_t y, | ||
24 : | const int32_t dx, | ||
25 : | const int dy, | ||
26 : | const uint32_t stride) | ||
27 : | Isibaar | 3 | { |
28 : | edgomez | 195 | int32_t ddx, ddy; |
29 : | Isibaar | 3 | |
30 : | edgomez | 195 | switch (((dx & 1) << 1) + (dy & 1)) // ((dx%2)?2:0)+((dy%2)?1:0) |
31 : | { | ||
32 : | case 0: | ||
33 : | ddx = dx / 2; | ||
34 : | ddy = dy / 2; | ||
35 : | transfer_8to16sub(dct_codes, cur + y * stride + x, | ||
36 : | Isibaar | 209 | ref + (int) ((y + ddy) * stride + x + ddx), stride); |
37 : | Isibaar | 3 | break; |
38 : | |||
39 : | edgomez | 195 | case 1: |
40 : | ddx = dx / 2; | ||
41 : | ddy = (dy - 1) / 2; | ||
42 : | transfer_8to16sub(dct_codes, cur + y * stride + x, | ||
43 : | Isibaar | 209 | refv + (int) ((y + ddy) * stride + x + ddx), stride); |
44 : | Isibaar | 3 | break; |
45 : | |||
46 : | edgomez | 195 | case 2: |
47 : | ddx = (dx - 1) / 2; | ||
48 : | ddy = dy / 2; | ||
49 : | transfer_8to16sub(dct_codes, cur + y * stride + x, | ||
50 : | Isibaar | 209 | refh + (int) ((y + ddy) * stride + x + ddx), stride); |
51 : | Isibaar | 3 | break; |
52 : | |||
53 : | edgomez | 195 | default: // case 3: |
54 : | ddx = (dx - 1) / 2; | ||
55 : | ddy = (dy - 1) / 2; | ||
56 : | transfer_8to16sub(dct_codes, cur + y * stride + x, | ||
57 : | Isibaar | 209 | refhv + (int) ((y + ddy) * stride + x + ddx), stride); |
58 : | Isibaar | 3 | break; |
59 : | edgomez | 195 | } |
60 : | Isibaar | 3 | } |
61 : | |||
62 : | |||
63 : | |||
64 : | edgomez | 195 | void |
65 : | MBMotionCompensation(MACROBLOCK * const mb, | ||
66 : | const uint32_t i, | ||
67 : | const uint32_t j, | ||
68 : | const IMAGE * const ref, | ||
69 : | const IMAGE * const refh, | ||
70 : | const IMAGE * const refv, | ||
71 : | const IMAGE * const refhv, | ||
72 : | IMAGE * const cur, | ||
73 : | int16_t * dct_codes, | ||
74 : | const uint32_t width, | ||
75 : | const uint32_t height, | ||
76 : | const uint32_t edged_width, | ||
77 : | const uint32_t rounding) | ||
78 : | Isibaar | 3 | { |
79 : | static const uint32_t roundtab[16] = | ||
80 : | { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 }; | ||
81 : | |||
82 : | |||
83 : | edgomez | 195 | if (mb->mode == MODE_INTER || mb->mode == MODE_INTER_Q) { |
84 : | Isibaar | 3 | int32_t dx = mb->mvs[0].x; |
85 : | int32_t dy = mb->mvs[0].y; | ||
86 : | |||
87 : | edgomez | 195 | compensate8x8_halfpel(&dct_codes[0 * 64], cur->y, ref->y, refh->y, |
88 : | refv->y, refhv->y, 16 * i, 16 * j, dx, dy, | ||
89 : | edged_width); | ||
90 : | compensate8x8_halfpel(&dct_codes[1 * 64], cur->y, ref->y, refh->y, | ||
91 : | refv->y, refhv->y, 16 * i + 8, 16 * j, dx, dy, | ||
92 : | edged_width); | ||
93 : | compensate8x8_halfpel(&dct_codes[2 * 64], cur->y, ref->y, refh->y, | ||
94 : | refv->y, refhv->y, 16 * i, 16 * j + 8, dx, dy, | ||
95 : | edged_width); | ||
96 : | compensate8x8_halfpel(&dct_codes[3 * 64], cur->y, ref->y, refh->y, | ||
97 : | refv->y, refhv->y, 16 * i + 8, 16 * j + 8, dx, | ||
98 : | dy, edged_width); | ||
99 : | Isibaar | 3 | |
100 : | dx = (dx & 3) ? (dx >> 1) | 1 : dx / 2; | ||
101 : | dy = (dy & 3) ? (dy >> 1) | 1 : dy / 2; | ||
102 : | |||
103 : | suxen_drol | 152 | /* uv-image-based compensation */ |
104 : | #ifdef BFRAMES | ||
105 : | edgomez | 195 | compensate8x8_halfpel(&dct_codes[4 * 64], cur->u, ref->u, refh->u, |
106 : | refv->u, refhv->u, 8 * i, 8 * j, dx, dy, | ||
107 : | edged_width / 2); | ||
108 : | compensate8x8_halfpel(&dct_codes[5 * 64], cur->v, ref->v, refh->v, | ||
109 : | refv->v, refhv->v, 8 * i, 8 * j, dx, dy, | ||
110 : | edged_width / 2); | ||
111 : | suxen_drol | 152 | #else |
112 : | Isibaar | 3 | /* uv-block-based compensation */ |
113 : | edgomez | 195 | interpolate8x8_switch(refv->u, ref->u, 8 * i, 8 * j, dx, dy, |
114 : | edged_width / 2, rounding); | ||
115 : | transfer_8to16sub(&dct_codes[4 * 64], | ||
116 : | cur->u + 8 * j * edged_width / 2 + 8 * i, | ||
117 : | refv->u + 8 * j * edged_width / 2 + 8 * i, | ||
118 : | edged_width / 2); | ||
119 : | Isibaar | 3 | |
120 : | edgomez | 195 | interpolate8x8_switch(refv->v, ref->v, 8 * i, 8 * j, dx, dy, |
121 : | edged_width / 2, rounding); | ||
122 : | transfer_8to16sub(&dct_codes[5 * 64], | ||
123 : | cur->v + 8 * j * edged_width / 2 + 8 * i, | ||
124 : | refv->v + 8 * j * edged_width / 2 + 8 * i, | ||
125 : | edged_width / 2); | ||
126 : | suxen_drol | 152 | #endif |
127 : | edgomez | 195 | } else // mode == MODE_INTER4V |
128 : | Isibaar | 3 | { |
129 : | int32_t sum, dx, dy; | ||
130 : | |||
131 : | edgomez | 195 | compensate8x8_halfpel(&dct_codes[0 * 64], cur->y, ref->y, refh->y, |
132 : | refv->y, refhv->y, 16 * i, 16 * j, mb->mvs[0].x, | ||
133 : | mb->mvs[0].y, edged_width); | ||
134 : | compensate8x8_halfpel(&dct_codes[1 * 64], cur->y, ref->y, refh->y, | ||
135 : | refv->y, refhv->y, 16 * i + 8, 16 * j, | ||
136 : | mb->mvs[1].x, mb->mvs[1].y, edged_width); | ||
137 : | compensate8x8_halfpel(&dct_codes[2 * 64], cur->y, ref->y, refh->y, | ||
138 : | refv->y, refhv->y, 16 * i, 16 * j + 8, | ||
139 : | mb->mvs[2].x, mb->mvs[2].y, edged_width); | ||
140 : | compensate8x8_halfpel(&dct_codes[3 * 64], cur->y, ref->y, refh->y, | ||
141 : | refv->y, refhv->y, 16 * i + 8, 16 * j + 8, | ||
142 : | mb->mvs[3].x, mb->mvs[3].y, edged_width); | ||
143 : | Isibaar | 3 | |
144 : | sum = mb->mvs[0].x + mb->mvs[1].x + mb->mvs[2].x + mb->mvs[3].x; | ||
145 : | edgomez | 195 | dx = (sum ? SIGN(sum) * |
146 : | (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) : 0); | ||
147 : | Isibaar | 3 | |
148 : | sum = mb->mvs[0].y + mb->mvs[1].y + mb->mvs[2].y + mb->mvs[3].y; | ||
149 : | edgomez | 195 | dy = (sum ? SIGN(sum) * |
150 : | (roundtab[ABS(sum) % 16] + (ABS(sum) / 16) * 2) : 0); | ||
151 : | Isibaar | 3 | |
152 : | suxen_drol | 152 | /* uv-image-based compensation */ |
153 : | #ifdef BFRAMES | ||
154 : | edgomez | 195 | compensate8x8_halfpel(&dct_codes[4 * 64], cur->u, ref->u, refh->u, |
155 : | refv->u, refhv->u, 8 * i, 8 * j, dx, dy, | ||
156 : | edged_width / 2); | ||
157 : | compensate8x8_halfpel(&dct_codes[5 * 64], cur->v, ref->v, refh->v, | ||
158 : | refv->v, refhv->v, 8 * i, 8 * j, dx, dy, | ||
159 : | edged_width / 2); | ||
160 : | suxen_drol | 152 | #else |
161 : | Isibaar | 3 | /* uv-block-based compensation */ |
162 : | edgomez | 195 | interpolate8x8_switch(refv->u, ref->u, 8 * i, 8 * j, dx, dy, |
163 : | edged_width / 2, rounding); | ||
164 : | transfer_8to16sub(&dct_codes[4 * 64], | ||
165 : | cur->u + 8 * j * edged_width / 2 + 8 * i, | ||
166 : | refv->u + 8 * j * edged_width / 2 + 8 * i, | ||
167 : | edged_width / 2); | ||
168 : | Isibaar | 3 | |
169 : | edgomez | 195 | interpolate8x8_switch(refv->v, ref->v, 8 * i, 8 * j, dx, dy, |
170 : | edged_width / 2, rounding); | ||
171 : | transfer_8to16sub(&dct_codes[5 * 64], | ||
172 : | cur->v + 8 * j * edged_width / 2 + 8 * i, | ||
173 : | refv->v + 8 * j * edged_width / 2 + 8 * i, | ||
174 : | edged_width / 2); | ||
175 : | suxen_drol | 152 | #endif |
176 : | Isibaar | 3 | } |
177 : | } | ||
178 : | suxen_drol | 118 | |
179 : | |||
180 : | edgomez | 195 | void |
181 : | MBMotionCompensationBVOP(MBParam * pParam, | ||
182 : | MACROBLOCK * const mb, | ||
183 : | const uint32_t i, | ||
184 : | const uint32_t j, | ||
185 : | IMAGE * const cur, | ||
186 : | const IMAGE * const f_ref, | ||
187 : | const IMAGE * const f_refh, | ||
188 : | const IMAGE * const f_refv, | ||
189 : | const IMAGE * const f_refhv, | ||
190 : | const IMAGE * const b_ref, | ||
191 : | const IMAGE * const b_refh, | ||
192 : | const IMAGE * const b_refv, | ||
193 : | const IMAGE * const b_refhv, | ||
194 : | int16_t * dct_codes) | ||
195 : | suxen_drol | 118 | { |
196 : | static const uint32_t roundtab[16] = | ||
197 : | { 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2 }; | ||
198 : | |||
199 : | const int32_t edged_width = pParam->edged_width; | ||
200 : | int32_t dx, dy; | ||
201 : | int32_t b_dx, b_dy; | ||
202 : | int x = i; | ||
203 : | int y = j; | ||
204 : | |||
205 : | |||
206 : | |||
207 : | edgomez | 195 | switch (mb->mode) { |
208 : | case MODE_FORWARD: | ||
209 : | suxen_drol | 118 | dx = mb->mvs[0].x; |
210 : | dy = mb->mvs[0].y; | ||
211 : | |||
212 : | edgomez | 195 | transfer_8to16sub_c(&dct_codes[0 * 64], |
213 : | cur->y + (j * 16) * edged_width + (i * 16), | ||
214 : | get_ref(f_ref->y, f_refh->y, f_refv->y, f_refhv->y, | ||
215 : | i * 16, j * 16, 1, dx, dy, edged_width), | ||
216 : | edged_width); | ||
217 : | suxen_drol | 118 | |
218 : | edgomez | 195 | transfer_8to16sub(&dct_codes[1 * 64], |
219 : | cur->y + (j * 16) * edged_width + (i * 16 + 8), | ||
220 : | get_ref(f_ref->y, f_refh->y, f_refv->y, f_refhv->y, | ||
221 : | i * 16 + 8, j * 16, 1, dx, dy, edged_width), | ||
222 : | edged_width); | ||
223 : | suxen_drol | 118 | |
224 : | edgomez | 195 | transfer_8to16sub_c(&dct_codes[2 * 64], |
225 : | cur->y + (j * 16 + 8) * edged_width + (i * 16), | ||
226 : | get_ref(f_ref->y, f_refh->y, f_refv->y, f_refhv->y, | ||
227 : | i * 16, j * 16 + 8, 1, dx, dy, | ||
228 : | edged_width), edged_width); | ||
229 : | suxen_drol | 118 | |
230 : | edgomez | 195 | transfer_8to16sub(&dct_codes[3 * 64], |
231 : | cur->y + (j * 16 + 8) * edged_width + (i * 16 + 8), | ||
232 : | get_ref(f_ref->y, f_refh->y, f_refv->y, f_refhv->y, | ||
233 : | i * 16 + 8, j * 16 + 8, 1, dx, dy, | ||
234 : | edged_width), edged_width); | ||
235 : | suxen_drol | 118 | |
236 : | |||
237 : | dx = (dx & 3) ? (dx >> 1) | 1 : dx / 2; | ||
238 : | dy = (dy & 3) ? (dy >> 1) | 1 : dy / 2; | ||
239 : | |||
240 : | /* uv-image-based compensation */ | ||
241 : | edgomez | 195 | compensate8x8_halfpel(&dct_codes[4 * 64], cur->u, f_ref->u, f_refh->u, |
242 : | f_refv->u, f_refhv->u, 8 * i, 8 * j, dx, dy, | ||
243 : | edged_width / 2); | ||
244 : | compensate8x8_halfpel(&dct_codes[5 * 64], cur->v, f_ref->v, f_refh->v, | ||
245 : | f_refv->v, f_refhv->v, 8 * i, 8 * j, dx, dy, | ||
246 : | edged_width / 2); | ||
247 : | suxen_drol | 118 | |
248 : | break; | ||
249 : | |||
250 : | edgomez | 195 | case MODE_BACKWARD: |
251 : | suxen_drol | 118 | b_dx = mb->b_mvs[0].x; |
252 : | b_dy = mb->b_mvs[0].y; | ||
253 : | |||
254 : | edgomez | 195 | transfer_8to16sub_c(&dct_codes[0 * 64], |
255 : | cur->y + (j * 16) * edged_width + (i * 16), | ||
256 : | get_ref(b_ref->y, b_refh->y, b_refv->y, b_refhv->y, | ||
257 : | i * 16, j * 16, 1, b_dx, b_dy, | ||
258 : | edged_width), edged_width); | ||
259 : | suxen_drol | 118 | |
260 : | edgomez | 195 | transfer_8to16sub(&dct_codes[1 * 64], |
261 : | cur->y + (j * 16) * edged_width + (i * 16 + 8), | ||
262 : | get_ref(b_ref->y, b_refh->y, b_refv->y, b_refhv->y, | ||
263 : | i * 16 + 8, j * 16, 1, b_dx, b_dy, | ||
264 : | edged_width), edged_width); | ||
265 : | suxen_drol | 118 | |
266 : | edgomez | 195 | transfer_8to16sub_c(&dct_codes[2 * 64], |
267 : | cur->y + (j * 16 + 8) * edged_width + (i * 16), | ||
268 : | get_ref(b_ref->y, b_refh->y, b_refv->y, b_refhv->y, | ||
269 : | i * 16, j * 16 + 8, 1, b_dx, b_dy, | ||
270 : | edged_width), edged_width); | ||
271 : | suxen_drol | 118 | |
272 : | edgomez | 195 | transfer_8to16sub(&dct_codes[3 * 64], |
273 : | cur->y + (j * 16 + 8) * edged_width + (i * 16 + 8), | ||
274 : | get_ref(b_ref->y, b_refh->y, b_refv->y, b_refhv->y, | ||
275 : | i * 16 + 8, j * 16 + 8, 1, b_dx, b_dy, | ||
276 : | edged_width), edged_width); | ||
277 : | suxen_drol | 118 | |
278 : | b_dx = (b_dx & 3) ? (b_dx >> 1) | 1 : b_dx / 2; | ||
279 : | b_dy = (b_dy & 3) ? (b_dy >> 1) | 1 : b_dy / 2; | ||
280 : | |||
281 : | /* uv-image-based compensation */ | ||
282 : | edgomez | 195 | compensate8x8_halfpel(&dct_codes[4 * 64], cur->u, b_ref->u, b_refh->u, |
283 : | b_refv->u, b_refhv->u, 8 * i, 8 * j, b_dx, b_dy, | ||
284 : | edged_width / 2); | ||
285 : | compensate8x8_halfpel(&dct_codes[5 * 64], cur->v, b_ref->v, b_refh->v, | ||
286 : | b_refv->v, b_refhv->v, 8 * i, 8 * j, b_dx, b_dy, | ||
287 : | edged_width / 2); | ||
288 : | suxen_drol | 118 | |
289 : | break; | ||
290 : | |||
291 : | |||
292 : | edgomez | 195 | case MODE_INTERPOLATE: |
293 : | suxen_drol | 118 | dx = mb->mvs[0].x; |
294 : | dy = mb->mvs[0].y; | ||
295 : | b_dx = mb->b_mvs[0].x; | ||
296 : | b_dy = mb->b_mvs[0].y; | ||
297 : | |||
298 : | edgomez | 195 | transfer_8to16sub2_c(&dct_codes[0 * 64], |
299 : | cur->y + (i * 16) + (j * 16) * edged_width, | ||
300 : | get_ref(f_ref->y, f_refh->y, f_refv->y, | ||
301 : | f_refhv->y, 16 * i, 16 * j, 1, dx, dy, | ||
302 : | edged_width), get_ref(b_ref->y, b_refh->y, | ||
303 : | b_refv->y, | ||
304 : | b_refhv->y, 16 * i, | ||
305 : | 16 * j, 1, b_dx, | ||
306 : | b_dy, edged_width), | ||
307 : | edged_width); | ||
308 : | suxen_drol | 118 | |
309 : | edgomez | 195 | transfer_8to16sub2_c(&dct_codes[1 * 64], |
310 : | cur->y + (i * 16 + 8) + (j * 16) * edged_width, | ||
311 : | get_ref(f_ref->y, f_refh->y, f_refv->y, | ||
312 : | f_refhv->y, 16 * i + 8, 16 * j, 1, dx, dy, | ||
313 : | edged_width), get_ref(b_ref->y, b_refh->y, | ||
314 : | b_refv->y, | ||
315 : | b_refhv->y, | ||
316 : | 16 * i + 8, 16 * j, | ||
317 : | 1, b_dx, b_dy, | ||
318 : | edged_width), | ||
319 : | edged_width); | ||
320 : | suxen_drol | 118 | |
321 : | edgomez | 195 | transfer_8to16sub2_c(&dct_codes[2 * 64], |
322 : | cur->y + (i * 16) + (j * 16 + 8) * edged_width, | ||
323 : | get_ref(f_ref->y, f_refh->y, f_refv->y, | ||
324 : | f_refhv->y, 16 * i, 16 * j + 8, 1, dx, dy, | ||
325 : | edged_width), get_ref(b_ref->y, b_refh->y, | ||
326 : | b_refv->y, | ||
327 : | b_refhv->y, 16 * i, | ||
328 : | 16 * j + 8, 1, b_dx, | ||
329 : | b_dy, edged_width), | ||
330 : | edged_width); | ||
331 : | suxen_drol | 118 | |
332 : | edgomez | 195 | transfer_8to16sub2_c(&dct_codes[3 * 64], |
333 : | cur->y + (i * 16 + 8) + (j * 16 + | ||
334 : | 8) * edged_width, | ||
335 : | get_ref(f_ref->y, f_refh->y, f_refv->y, | ||
336 : | f_refhv->y, 16 * i + 8, 16 * j + 8, 1, dx, | ||
337 : | dy, edged_width), get_ref(b_ref->y, | ||
338 : | b_refh->y, | ||
339 : | b_refv->y, | ||
340 : | b_refhv->y, | ||
341 : | 16 * i + 8, | ||
342 : | 16 * j + 8, 1, | ||
343 : | b_dx, b_dy, | ||
344 : | edged_width), | ||
345 : | edged_width); | ||
346 : | suxen_drol | 118 | |
347 : | edgomez | 195 | |
348 : | suxen_drol | 118 | dx = (dx & 3) ? (dx >> 1) | 1 : dx / 2; |
349 : | dy = (dy & 3) ? (dy >> 1) | 1 : dy / 2; | ||
350 : | |||
351 : | b_dx = (b_dx & 3) ? (b_dx >> 1) | 1 : b_dx / 2; | ||
352 : | b_dy = (b_dy & 3) ? (b_dy >> 1) | 1 : b_dy / 2; | ||
353 : | |||
354 : | edgomez | 195 | transfer_8to16sub2_c(&dct_codes[4 * 64], |
355 : | cur->u + (y * 8) * edged_width / 2 + (x * 8), | ||
356 : | get_ref(f_ref->u, f_refh->u, f_refv->u, | ||
357 : | f_refhv->u, 8 * i, 8 * j, 1, dx, dy, | ||
358 : | edged_width / 2), get_ref(b_ref->u, | ||
359 : | b_refh->u, | ||
360 : | b_refv->u, | ||
361 : | b_refhv->u, | ||
362 : | 8 * i, 8 * j, 1, | ||
363 : | b_dx, b_dy, | ||
364 : | edged_width / | ||
365 : | 2), | ||
366 : | edged_width / 2); | ||
367 : | suxen_drol | 118 | |
368 : | edgomez | 195 | transfer_8to16sub2_c(&dct_codes[5 * 64], |
369 : | cur->v + (y * 8) * edged_width / 2 + (x * 8), | ||
370 : | get_ref(f_ref->v, f_refh->v, f_refv->v, | ||
371 : | f_refhv->v, 8 * i, 8 * j, 1, dx, dy, | ||
372 : | edged_width / 2), get_ref(b_ref->v, | ||
373 : | b_refh->v, | ||
374 : | b_refv->v, | ||
375 : | b_refhv->v, | ||
376 : | 8 * i, 8 * j, 1, | ||
377 : | b_dx, b_dy, | ||
378 : | edged_width / | ||
379 : | 2), | ||
380 : | edged_width / 2); | ||
381 : | |||
382 : | suxen_drol | 118 | break; |
383 : | |||
384 : | edgomez | 195 | case MODE_DIRECT: |
385 : | suxen_drol | 118 | // todo |
386 : | break; | ||
387 : | } | ||
388 : | edgomez | 195 | |
389 : | suxen_drol | 118 | } |
No admin address has been configured | ViewVC Help |
Powered by ViewVC 1.0.4 |