6d6c1ec4fa4af3c82c1b409113fc99eada6f719f
[ffmpeg.git] / libavcodec / vc1dec.c
1 /*
2  * VC-1 and WMV3 decoder
3  * Copyright (c) 2011 Mashiat Sarker Shakkhar
4  * Copyright (c) 2006-2007 Konstantin Shishkov
5  * Partly based on vc9.c (c) 2005 Anonymous, Alex Beregszaszi, Michael Niedermayer
6  *
7  * This file is part of Libav.
8  *
9  * Libav is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * Libav is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with Libav; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23
24 /**
25  * @file
26  * VC-1 and WMV3 decoder
27  */
28
29 #include "internal.h"
30 #include "avcodec.h"
31 #include "error_resilience.h"
32 #include "mpeg_er.h"
33 #include "mpegutils.h"
34 #include "mpegvideo.h"
35 #include "h263.h"
36 #include "h264chroma.h"
37 #include "qpeldsp.h"
38 #include "vc1.h"
39 #include "vc1data.h"
40 #include "vc1acdata.h"
41 #include "msmpeg4data.h"
42 #include "unary.h"
43 #include "mathops.h"
44
45 #undef NDEBUG
46 #include <assert.h>
47
48 #define MB_INTRA_VLC_BITS 9
49 #define DC_VLC_BITS 9
50
51
52 // offset tables for interlaced picture MVDATA decoding
53 static const int offset_table1[9] = {  0,  1,  2,  4,  8, 16, 32,  64, 128 };
54 static const int offset_table2[9] = {  0,  1,  3,  7, 15, 31, 63, 127, 255 };
55
56 /***********************************************************************/
57 /**
58  * @name VC-1 Bitplane decoding
59  * @see 8.7, p56
60  * @{
61  */
62
63 /**
64  * Imode types
65  * @{
66  */
67 enum Imode {
68     IMODE_RAW,
69     IMODE_NORM2,
70     IMODE_DIFF2,
71     IMODE_NORM6,
72     IMODE_DIFF6,
73     IMODE_ROWSKIP,
74     IMODE_COLSKIP
75 };
76 /** @} */ //imode defines
77
78 static void init_block_index(VC1Context *v)
79 {
80     MpegEncContext *s = &v->s;
81     ff_init_block_index(s);
82     if (v->field_mode && !(v->second_field ^ v->tff)) {
83         s->dest[0] += s->current_picture_ptr->f->linesize[0];
84         s->dest[1] += s->current_picture_ptr->f->linesize[1];
85         s->dest[2] += s->current_picture_ptr->f->linesize[2];
86     }
87 }
88
89 /** @} */ //Bitplane group
90
91 static void vc1_put_signed_blocks_clamped(VC1Context *v)
92 {
93     MpegEncContext *s = &v->s;
94     int topleft_mb_pos, top_mb_pos;
95     int stride_y, fieldtx = 0;
96     int v_dist;
97
98     /* The put pixels loop is always one MB row behind the decoding loop,
99      * because we can only put pixels when overlap filtering is done, and
100      * for filtering of the bottom edge of a MB, we need the next MB row
101      * present as well.
102      * Within the row, the put pixels loop is also one MB col behind the
103      * decoding loop. The reason for this is again, because for filtering
104      * of the right MB edge, we need the next MB present. */
105     if (!s->first_slice_line) {
106         if (s->mb_x) {
107             topleft_mb_pos = (s->mb_y - 1) * s->mb_stride + s->mb_x - 1;
108             if (v->fcm == ILACE_FRAME)
109                 fieldtx = v->fieldtx_plane[topleft_mb_pos];
110             stride_y       = s->linesize << fieldtx;
111             v_dist         = (16 - fieldtx) >> (fieldtx == 0);
112             s->dsp.put_signed_pixels_clamped(v->block[v->topleft_blk_idx][0],
113                                              s->dest[0] - 16 * s->linesize - 16,
114                                              stride_y);
115             s->dsp.put_signed_pixels_clamped(v->block[v->topleft_blk_idx][1],
116                                              s->dest[0] - 16 * s->linesize - 8,
117                                              stride_y);
118             s->dsp.put_signed_pixels_clamped(v->block[v->topleft_blk_idx][2],
119                                              s->dest[0] - v_dist * s->linesize - 16,
120                                              stride_y);
121             s->dsp.put_signed_pixels_clamped(v->block[v->topleft_blk_idx][3],
122                                              s->dest[0] - v_dist * s->linesize - 8,
123                                              stride_y);
124             s->dsp.put_signed_pixels_clamped(v->block[v->topleft_blk_idx][4],
125                                              s->dest[1] - 8 * s->uvlinesize - 8,
126                                              s->uvlinesize);
127             s->dsp.put_signed_pixels_clamped(v->block[v->topleft_blk_idx][5],
128                                              s->dest[2] - 8 * s->uvlinesize - 8,
129                                              s->uvlinesize);
130         }
131         if (s->mb_x == s->mb_width - 1) {
132             top_mb_pos = (s->mb_y - 1) * s->mb_stride + s->mb_x;
133             if (v->fcm == ILACE_FRAME)
134                 fieldtx = v->fieldtx_plane[top_mb_pos];
135             stride_y   = s->linesize << fieldtx;
136             v_dist     = fieldtx ? 15 : 8;
137             s->dsp.put_signed_pixels_clamped(v->block[v->top_blk_idx][0],
138                                              s->dest[0] - 16 * s->linesize,
139                                              stride_y);
140             s->dsp.put_signed_pixels_clamped(v->block[v->top_blk_idx][1],
141                                              s->dest[0] - 16 * s->linesize + 8,
142                                              stride_y);
143             s->dsp.put_signed_pixels_clamped(v->block[v->top_blk_idx][2],
144                                              s->dest[0] - v_dist * s->linesize,
145                                              stride_y);
146             s->dsp.put_signed_pixels_clamped(v->block[v->top_blk_idx][3],
147                                              s->dest[0] - v_dist * s->linesize + 8,
148                                              stride_y);
149             s->dsp.put_signed_pixels_clamped(v->block[v->top_blk_idx][4],
150                                              s->dest[1] - 8 * s->uvlinesize,
151                                              s->uvlinesize);
152             s->dsp.put_signed_pixels_clamped(v->block[v->top_blk_idx][5],
153                                              s->dest[2] - 8 * s->uvlinesize,
154                                              s->uvlinesize);
155         }
156     }
157
158 #define inc_blk_idx(idx) do { \
159         idx++; \
160         if (idx >= v->n_allocated_blks) \
161             idx = 0; \
162     } while (0)
163
164     inc_blk_idx(v->topleft_blk_idx);
165     inc_blk_idx(v->top_blk_idx);
166     inc_blk_idx(v->left_blk_idx);
167     inc_blk_idx(v->cur_blk_idx);
168 }
169
170 static void vc1_loop_filter_iblk(VC1Context *v, int pq)
171 {
172     MpegEncContext *s = &v->s;
173     int j;
174     if (!s->first_slice_line) {
175         v->vc1dsp.vc1_v_loop_filter16(s->dest[0], s->linesize, pq);
176         if (s->mb_x)
177             v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 16 * s->linesize, s->linesize, pq);
178         v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 16 * s->linesize + 8, s->linesize, pq);
179         for (j = 0; j < 2; j++) {
180             v->vc1dsp.vc1_v_loop_filter8(s->dest[j + 1], s->uvlinesize, pq);
181             if (s->mb_x)
182                 v->vc1dsp.vc1_h_loop_filter8(s->dest[j + 1] - 8 * s->uvlinesize, s->uvlinesize, pq);
183         }
184     }
185     v->vc1dsp.vc1_v_loop_filter16(s->dest[0] + 8 * s->linesize, s->linesize, pq);
186
187     if (s->mb_y == s->end_mb_y - 1) {
188         if (s->mb_x) {
189             v->vc1dsp.vc1_h_loop_filter16(s->dest[0], s->linesize, pq);
190             v->vc1dsp.vc1_h_loop_filter8(s->dest[1], s->uvlinesize, pq);
191             v->vc1dsp.vc1_h_loop_filter8(s->dest[2], s->uvlinesize, pq);
192         }
193         v->vc1dsp.vc1_h_loop_filter16(s->dest[0] + 8, s->linesize, pq);
194     }
195 }
196
197 static void vc1_loop_filter_iblk_delayed(VC1Context *v, int pq)
198 {
199     MpegEncContext *s = &v->s;
200     int j;
201
202     /* The loopfilter runs 1 row and 1 column behind the overlap filter, which
203      * means it runs two rows/cols behind the decoding loop. */
204     if (!s->first_slice_line) {
205         if (s->mb_x) {
206             if (s->mb_y >= s->start_mb_y + 2) {
207                 v->vc1dsp.vc1_v_loop_filter16(s->dest[0] - 16 * s->linesize - 16, s->linesize, pq);
208
209                 if (s->mb_x >= 2)
210                     v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 32 * s->linesize - 16, s->linesize, pq);
211                 v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 32 * s->linesize - 8, s->linesize, pq);
212                 for (j = 0; j < 2; j++) {
213                     v->vc1dsp.vc1_v_loop_filter8(s->dest[j + 1] - 8 * s->uvlinesize - 8, s->uvlinesize, pq);
214                     if (s->mb_x >= 2) {
215                         v->vc1dsp.vc1_h_loop_filter8(s->dest[j + 1] - 16 * s->uvlinesize - 8, s->uvlinesize, pq);
216                     }
217                 }
218             }
219             v->vc1dsp.vc1_v_loop_filter16(s->dest[0] - 8 * s->linesize - 16, s->linesize, pq);
220         }
221
222         if (s->mb_x == s->mb_width - 1) {
223             if (s->mb_y >= s->start_mb_y + 2) {
224                 v->vc1dsp.vc1_v_loop_filter16(s->dest[0] - 16 * s->linesize, s->linesize, pq);
225
226                 if (s->mb_x)
227                     v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 32 * s->linesize, s->linesize, pq);
228                 v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 32 * s->linesize + 8, s->linesize, pq);
229                 for (j = 0; j < 2; j++) {
230                     v->vc1dsp.vc1_v_loop_filter8(s->dest[j + 1] - 8 * s->uvlinesize, s->uvlinesize, pq);
231                     if (s->mb_x >= 2) {
232                         v->vc1dsp.vc1_h_loop_filter8(s->dest[j + 1] - 16 * s->uvlinesize, s->uvlinesize, pq);
233                     }
234                 }
235             }
236             v->vc1dsp.vc1_v_loop_filter16(s->dest[0] - 8 * s->linesize, s->linesize, pq);
237         }
238
239         if (s->mb_y == s->end_mb_y) {
240             if (s->mb_x) {
241                 if (s->mb_x >= 2)
242                     v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 16 * s->linesize - 16, s->linesize, pq);
243                 v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 16 * s->linesize - 8, s->linesize, pq);
244                 if (s->mb_x >= 2) {
245                     for (j = 0; j < 2; j++) {
246                         v->vc1dsp.vc1_h_loop_filter8(s->dest[j + 1] - 8 * s->uvlinesize - 8, s->uvlinesize, pq);
247                     }
248                 }
249             }
250
251             if (s->mb_x == s->mb_width - 1) {
252                 if (s->mb_x)
253                     v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 16 * s->linesize, s->linesize, pq);
254                 v->vc1dsp.vc1_h_loop_filter16(s->dest[0] - 16 * s->linesize + 8, s->linesize, pq);
255                 if (s->mb_x) {
256                     for (j = 0; j < 2; j++) {
257                         v->vc1dsp.vc1_h_loop_filter8(s->dest[j + 1] - 8 * s->uvlinesize, s->uvlinesize, pq);
258                     }
259                 }
260             }
261         }
262     }
263 }
264
265 static void vc1_smooth_overlap_filter_iblk(VC1Context *v)
266 {
267     MpegEncContext *s = &v->s;
268     int mb_pos;
269
270     if (v->condover == CONDOVER_NONE)
271         return;
272
273     mb_pos = s->mb_x + s->mb_y * s->mb_stride;
274
275     /* Within a MB, the horizontal overlap always runs before the vertical.
276      * To accomplish that, we run the H on left and internal borders of the
277      * currently decoded MB. Then, we wait for the next overlap iteration
278      * to do H overlap on the right edge of this MB, before moving over and
279      * running the V overlap. Therefore, the V overlap makes us trail by one
280      * MB col and the H overlap filter makes us trail by one MB row. This
281      * is reflected in the time at which we run the put_pixels loop. */
282     if (v->condover == CONDOVER_ALL || v->pq >= 9 || v->over_flags_plane[mb_pos]) {
283         if (s->mb_x && (v->condover == CONDOVER_ALL || v->pq >= 9 ||
284                         v->over_flags_plane[mb_pos - 1])) {
285             v->vc1dsp.vc1_h_s_overlap(v->block[v->left_blk_idx][1],
286                                       v->block[v->cur_blk_idx][0]);
287             v->vc1dsp.vc1_h_s_overlap(v->block[v->left_blk_idx][3],
288                                       v->block[v->cur_blk_idx][2]);
289             if (!(s->flags & CODEC_FLAG_GRAY)) {
290                 v->vc1dsp.vc1_h_s_overlap(v->block[v->left_blk_idx][4],
291                                           v->block[v->cur_blk_idx][4]);
292                 v->vc1dsp.vc1_h_s_overlap(v->block[v->left_blk_idx][5],
293                                           v->block[v->cur_blk_idx][5]);
294             }
295         }
296         v->vc1dsp.vc1_h_s_overlap(v->block[v->cur_blk_idx][0],
297                                   v->block[v->cur_blk_idx][1]);
298         v->vc1dsp.vc1_h_s_overlap(v->block[v->cur_blk_idx][2],
299                                   v->block[v->cur_blk_idx][3]);
300
301         if (s->mb_x == s->mb_width - 1) {
302             if (!s->first_slice_line && (v->condover == CONDOVER_ALL || v->pq >= 9 ||
303                                          v->over_flags_plane[mb_pos - s->mb_stride])) {
304                 v->vc1dsp.vc1_v_s_overlap(v->block[v->top_blk_idx][2],
305                                           v->block[v->cur_blk_idx][0]);
306                 v->vc1dsp.vc1_v_s_overlap(v->block[v->top_blk_idx][3],
307                                           v->block[v->cur_blk_idx][1]);
308                 if (!(s->flags & CODEC_FLAG_GRAY)) {
309                     v->vc1dsp.vc1_v_s_overlap(v->block[v->top_blk_idx][4],
310                                               v->block[v->cur_blk_idx][4]);
311                     v->vc1dsp.vc1_v_s_overlap(v->block[v->top_blk_idx][5],
312                                               v->block[v->cur_blk_idx][5]);
313                 }
314             }
315             v->vc1dsp.vc1_v_s_overlap(v->block[v->cur_blk_idx][0],
316                                       v->block[v->cur_blk_idx][2]);
317             v->vc1dsp.vc1_v_s_overlap(v->block[v->cur_blk_idx][1],
318                                       v->block[v->cur_blk_idx][3]);
319         }
320     }
321     if (s->mb_x && (v->condover == CONDOVER_ALL || v->over_flags_plane[mb_pos - 1])) {
322         if (!s->first_slice_line && (v->condover == CONDOVER_ALL || v->pq >= 9 ||
323                                      v->over_flags_plane[mb_pos - s->mb_stride - 1])) {
324             v->vc1dsp.vc1_v_s_overlap(v->block[v->topleft_blk_idx][2],
325                                       v->block[v->left_blk_idx][0]);
326             v->vc1dsp.vc1_v_s_overlap(v->block[v->topleft_blk_idx][3],
327                                       v->block[v->left_blk_idx][1]);
328             if (!(s->flags & CODEC_FLAG_GRAY)) {
329                 v->vc1dsp.vc1_v_s_overlap(v->block[v->topleft_blk_idx][4],
330                                           v->block[v->left_blk_idx][4]);
331                 v->vc1dsp.vc1_v_s_overlap(v->block[v->topleft_blk_idx][5],
332                                           v->block[v->left_blk_idx][5]);
333             }
334         }
335         v->vc1dsp.vc1_v_s_overlap(v->block[v->left_blk_idx][0],
336                                   v->block[v->left_blk_idx][2]);
337         v->vc1dsp.vc1_v_s_overlap(v->block[v->left_blk_idx][1],
338                                   v->block[v->left_blk_idx][3]);
339     }
340 }
341
342 /** Do motion compensation over 1 macroblock
343  * Mostly adapted hpel_motion and qpel_motion from mpegvideo.c
344  */
345 static void vc1_mc_1mv(VC1Context *v, int dir)
346 {
347     MpegEncContext *s = &v->s;
348     H264ChromaContext *h264chroma = &v->h264chroma;
349     uint8_t *srcY, *srcU, *srcV;
350     int dxy, mx, my, uvmx, uvmy, src_x, src_y, uvsrc_x, uvsrc_y;
351     int v_edge_pos = s->v_edge_pos >> v->field_mode;
352     int i;
353     uint8_t (*luty)[256], (*lutuv)[256];
354     int use_ic;
355
356     if ((!v->field_mode ||
357          (v->ref_field_type[dir] == 1 && v->cur_field_type == 1)) &&
358         !v->s.last_picture.f->data[0])
359         return;
360
361     mx = s->mv[dir][0][0];
362     my = s->mv[dir][0][1];
363
364     // store motion vectors for further use in B frames
365     if (s->pict_type == AV_PICTURE_TYPE_P) {
366         for (i = 0; i < 4; i++) {
367             s->current_picture.motion_val[1][s->block_index[i] + v->blocks_off][0] = mx;
368             s->current_picture.motion_val[1][s->block_index[i] + v->blocks_off][1] = my;
369         }
370     }
371
372     uvmx = (mx + ((mx & 3) == 3)) >> 1;
373     uvmy = (my + ((my & 3) == 3)) >> 1;
374     v->luma_mv[s->mb_x][0] = uvmx;
375     v->luma_mv[s->mb_x][1] = uvmy;
376
377     if (v->field_mode &&
378         v->cur_field_type != v->ref_field_type[dir]) {
379         my   = my   - 2 + 4 * v->cur_field_type;
380         uvmy = uvmy - 2 + 4 * v->cur_field_type;
381     }
382
383     // fastuvmc shall be ignored for interlaced frame picture
384     if (v->fastuvmc && (v->fcm != ILACE_FRAME)) {
385         uvmx = uvmx + ((uvmx < 0) ? (uvmx & 1) : -(uvmx & 1));
386         uvmy = uvmy + ((uvmy < 0) ? (uvmy & 1) : -(uvmy & 1));
387     }
388     if (!dir) {
389         if (v->field_mode && (v->cur_field_type != v->ref_field_type[dir]) && v->second_field) {
390             srcY = s->current_picture.f->data[0];
391             srcU = s->current_picture.f->data[1];
392             srcV = s->current_picture.f->data[2];
393             luty  = v->curr_luty;
394             lutuv = v->curr_lutuv;
395             use_ic = v->curr_use_ic;
396         } else {
397             srcY = s->last_picture.f->data[0];
398             srcU = s->last_picture.f->data[1];
399             srcV = s->last_picture.f->data[2];
400             luty  = v->last_luty;
401             lutuv = v->last_lutuv;
402             use_ic = v->last_use_ic;
403         }
404     } else {
405         srcY = s->next_picture.f->data[0];
406         srcU = s->next_picture.f->data[1];
407         srcV = s->next_picture.f->data[2];
408         luty  = v->next_luty;
409         lutuv = v->next_lutuv;
410         use_ic = v->next_use_ic;
411     }
412
413     if (!srcY || !srcU) {
414         av_log(v->s.avctx, AV_LOG_ERROR, "Referenced frame missing.\n");
415         return;
416     }
417
418     src_x   = s->mb_x * 16 + (mx   >> 2);
419     src_y   = s->mb_y * 16 + (my   >> 2);
420     uvsrc_x = s->mb_x *  8 + (uvmx >> 2);
421     uvsrc_y = s->mb_y *  8 + (uvmy >> 2);
422
423     if (v->profile != PROFILE_ADVANCED) {
424         src_x   = av_clip(  src_x, -16, s->mb_width  * 16);
425         src_y   = av_clip(  src_y, -16, s->mb_height * 16);
426         uvsrc_x = av_clip(uvsrc_x,  -8, s->mb_width  *  8);
427         uvsrc_y = av_clip(uvsrc_y,  -8, s->mb_height *  8);
428     } else {
429         src_x   = av_clip(  src_x, -17, s->avctx->coded_width);
430         src_y   = av_clip(  src_y, -18, s->avctx->coded_height + 1);
431         uvsrc_x = av_clip(uvsrc_x,  -8, s->avctx->coded_width  >> 1);
432         uvsrc_y = av_clip(uvsrc_y,  -8, s->avctx->coded_height >> 1);
433     }
434
435     srcY += src_y   * s->linesize   + src_x;
436     srcU += uvsrc_y * s->uvlinesize + uvsrc_x;
437     srcV += uvsrc_y * s->uvlinesize + uvsrc_x;
438
439     if (v->field_mode && v->ref_field_type[dir]) {
440         srcY += s->current_picture_ptr->f->linesize[0];
441         srcU += s->current_picture_ptr->f->linesize[1];
442         srcV += s->current_picture_ptr->f->linesize[2];
443     }
444
445     /* for grayscale we should not try to read from unknown area */
446     if (s->flags & CODEC_FLAG_GRAY) {
447         srcU = s->edge_emu_buffer + 18 * s->linesize;
448         srcV = s->edge_emu_buffer + 18 * s->linesize;
449     }
450
451     if (v->rangeredfrm || use_ic
452         || s->h_edge_pos < 22 || v_edge_pos < 22
453         || (unsigned)(src_x - s->mspel) > s->h_edge_pos - (mx&3) - 16 - s->mspel * 3
454         || (unsigned)(src_y - 1)        > v_edge_pos    - (my&3) - 16 - 3) {
455         uint8_t *uvbuf = s->edge_emu_buffer + 19 * s->linesize;
456
457         srcY -= s->mspel * (1 + s->linesize);
458         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, srcY,
459                                  s->linesize, s->linesize,
460                                  17 + s->mspel * 2, 17 + s->mspel * 2,
461                                  src_x - s->mspel, src_y - s->mspel,
462                                  s->h_edge_pos, v_edge_pos);
463         srcY = s->edge_emu_buffer;
464         s->vdsp.emulated_edge_mc(uvbuf, srcU,
465                                  s->uvlinesize, s->uvlinesize,
466                                  8 + 1, 8 + 1,
467                                  uvsrc_x, uvsrc_y, s->h_edge_pos >> 1, v_edge_pos >> 1);
468         s->vdsp.emulated_edge_mc(uvbuf + 16, srcV,
469                                  s->uvlinesize, s->uvlinesize,
470                                  8 + 1, 8 + 1,
471                                  uvsrc_x, uvsrc_y, s->h_edge_pos >> 1, v_edge_pos >> 1);
472         srcU = uvbuf;
473         srcV = uvbuf + 16;
474         /* if we deal with range reduction we need to scale source blocks */
475         if (v->rangeredfrm) {
476             int i, j;
477             uint8_t *src, *src2;
478
479             src = srcY;
480             for (j = 0; j < 17 + s->mspel * 2; j++) {
481                 for (i = 0; i < 17 + s->mspel * 2; i++)
482                     src[i] = ((src[i] - 128) >> 1) + 128;
483                 src += s->linesize;
484             }
485             src  = srcU;
486             src2 = srcV;
487             for (j = 0; j < 9; j++) {
488                 for (i = 0; i < 9; i++) {
489                     src[i]  = ((src[i]  - 128) >> 1) + 128;
490                     src2[i] = ((src2[i] - 128) >> 1) + 128;
491                 }
492                 src  += s->uvlinesize;
493                 src2 += s->uvlinesize;
494             }
495         }
496         /* if we deal with intensity compensation we need to scale source blocks */
497         if (use_ic) {
498             int i, j;
499             uint8_t *src, *src2;
500
501             src = srcY;
502             for (j = 0; j < 17 + s->mspel * 2; j++) {
503                 int f = v->field_mode ? v->ref_field_type[dir] : ((j + src_y - s->mspel) & 1) ;
504                 for (i = 0; i < 17 + s->mspel * 2; i++)
505                     src[i] = luty[f][src[i]];
506                 src += s->linesize;
507             }
508             src  = srcU;
509             src2 = srcV;
510             for (j = 0; j < 9; j++) {
511                 int f = v->field_mode ? v->ref_field_type[dir] : ((j + uvsrc_y) & 1);
512                 for (i = 0; i < 9; i++) {
513                     src[i]  = lutuv[f][src[i]];
514                     src2[i] = lutuv[f][src2[i]];
515                 }
516                 src  += s->uvlinesize;
517                 src2 += s->uvlinesize;
518             }
519         }
520         srcY += s->mspel * (1 + s->linesize);
521     }
522
523     if (s->mspel) {
524         dxy = ((my & 3) << 2) | (mx & 3);
525         v->vc1dsp.put_vc1_mspel_pixels_tab[dxy](s->dest[0]    , srcY    , s->linesize, v->rnd);
526         v->vc1dsp.put_vc1_mspel_pixels_tab[dxy](s->dest[0] + 8, srcY + 8, s->linesize, v->rnd);
527         srcY += s->linesize * 8;
528         v->vc1dsp.put_vc1_mspel_pixels_tab[dxy](s->dest[0] + 8 * s->linesize    , srcY    , s->linesize, v->rnd);
529         v->vc1dsp.put_vc1_mspel_pixels_tab[dxy](s->dest[0] + 8 * s->linesize + 8, srcY + 8, s->linesize, v->rnd);
530     } else { // hpel mc - always used for luma
531         dxy = (my & 2) | ((mx & 2) >> 1);
532         if (!v->rnd)
533             s->hdsp.put_pixels_tab[0][dxy](s->dest[0], srcY, s->linesize, 16);
534         else
535             s->hdsp.put_no_rnd_pixels_tab[0][dxy](s->dest[0], srcY, s->linesize, 16);
536     }
537
538     if (s->flags & CODEC_FLAG_GRAY) return;
539     /* Chroma MC always uses qpel bilinear */
540     uvmx = (uvmx & 3) << 1;
541     uvmy = (uvmy & 3) << 1;
542     if (!v->rnd) {
543         h264chroma->put_h264_chroma_pixels_tab[0](s->dest[1], srcU, s->uvlinesize, 8, uvmx, uvmy);
544         h264chroma->put_h264_chroma_pixels_tab[0](s->dest[2], srcV, s->uvlinesize, 8, uvmx, uvmy);
545     } else {
546         v->vc1dsp.put_no_rnd_vc1_chroma_pixels_tab[0](s->dest[1], srcU, s->uvlinesize, 8, uvmx, uvmy);
547         v->vc1dsp.put_no_rnd_vc1_chroma_pixels_tab[0](s->dest[2], srcV, s->uvlinesize, 8, uvmx, uvmy);
548     }
549 }
550
551 static inline int median4(int a, int b, int c, int d)
552 {
553     if (a < b) {
554         if (c < d) return (FFMIN(b, d) + FFMAX(a, c)) / 2;
555         else       return (FFMIN(b, c) + FFMAX(a, d)) / 2;
556     } else {
557         if (c < d) return (FFMIN(a, d) + FFMAX(b, c)) / 2;
558         else       return (FFMIN(a, c) + FFMAX(b, d)) / 2;
559     }
560 }
561
562 /** Do motion compensation for 4-MV macroblock - luminance block
563  */
564 static void vc1_mc_4mv_luma(VC1Context *v, int n, int dir, int avg)
565 {
566     MpegEncContext *s = &v->s;
567     uint8_t *srcY;
568     int dxy, mx, my, src_x, src_y;
569     int off;
570     int fieldmv = (v->fcm == ILACE_FRAME) ? v->blk_mv_type[s->block_index[n]] : 0;
571     int v_edge_pos = s->v_edge_pos >> v->field_mode;
572     uint8_t (*luty)[256];
573     int use_ic;
574
575     if ((!v->field_mode ||
576          (v->ref_field_type[dir] == 1 && v->cur_field_type == 1)) &&
577         !v->s.last_picture.f->data[0])
578         return;
579
580     mx = s->mv[dir][n][0];
581     my = s->mv[dir][n][1];
582
583     if (!dir) {
584         if (v->field_mode && (v->cur_field_type != v->ref_field_type[dir]) && v->second_field) {
585             srcY = s->current_picture.f->data[0];
586             luty = v->curr_luty;
587             use_ic = v->curr_use_ic;
588         } else {
589             srcY = s->last_picture.f->data[0];
590             luty = v->last_luty;
591             use_ic = v->last_use_ic;
592         }
593     } else {
594         srcY = s->next_picture.f->data[0];
595         luty = v->next_luty;
596         use_ic = v->next_use_ic;
597     }
598
599     if (!srcY) {
600         av_log(v->s.avctx, AV_LOG_ERROR, "Referenced frame missing.\n");
601         return;
602     }
603
604     if (v->field_mode) {
605         if (v->cur_field_type != v->ref_field_type[dir])
606             my = my - 2 + 4 * v->cur_field_type;
607     }
608
609     if (s->pict_type == AV_PICTURE_TYPE_P && n == 3 && v->field_mode) {
610         int same_count = 0, opp_count = 0, k;
611         int chosen_mv[2][4][2], f;
612         int tx, ty;
613         for (k = 0; k < 4; k++) {
614             f = v->mv_f[0][s->block_index[k] + v->blocks_off];
615             chosen_mv[f][f ? opp_count : same_count][0] = s->mv[0][k][0];
616             chosen_mv[f][f ? opp_count : same_count][1] = s->mv[0][k][1];
617             opp_count  += f;
618             same_count += 1 - f;
619         }
620         f = opp_count > same_count;
621         switch (f ? opp_count : same_count) {
622         case 4:
623             tx = median4(chosen_mv[f][0][0], chosen_mv[f][1][0],
624                          chosen_mv[f][2][0], chosen_mv[f][3][0]);
625             ty = median4(chosen_mv[f][0][1], chosen_mv[f][1][1],
626                          chosen_mv[f][2][1], chosen_mv[f][3][1]);
627             break;
628         case 3:
629             tx = mid_pred(chosen_mv[f][0][0], chosen_mv[f][1][0], chosen_mv[f][2][0]);
630             ty = mid_pred(chosen_mv[f][0][1], chosen_mv[f][1][1], chosen_mv[f][2][1]);
631             break;
632         case 2:
633             tx = (chosen_mv[f][0][0] + chosen_mv[f][1][0]) / 2;
634             ty = (chosen_mv[f][0][1] + chosen_mv[f][1][1]) / 2;
635             break;
636         }
637         s->current_picture.motion_val[1][s->block_index[0] + v->blocks_off][0] = tx;
638         s->current_picture.motion_val[1][s->block_index[0] + v->blocks_off][1] = ty;
639         for (k = 0; k < 4; k++)
640             v->mv_f[1][s->block_index[k] + v->blocks_off] = f;
641     }
642
643     if (v->fcm == ILACE_FRAME) {  // not sure if needed for other types of picture
644         int qx, qy;
645         int width  = s->avctx->coded_width;
646         int height = s->avctx->coded_height >> 1;
647         if (s->pict_type == AV_PICTURE_TYPE_P) {
648             s->current_picture.motion_val[1][s->block_index[n] + v->blocks_off][0] = mx;
649             s->current_picture.motion_val[1][s->block_index[n] + v->blocks_off][1] = my;
650         }
651         qx = (s->mb_x * 16) + (mx >> 2);
652         qy = (s->mb_y *  8) + (my >> 3);
653
654         if (qx < -17)
655             mx -= 4 * (qx + 17);
656         else if (qx > width)
657             mx -= 4 * (qx - width);
658         if (qy < -18)
659             my -= 8 * (qy + 18);
660         else if (qy > height + 1)
661             my -= 8 * (qy - height - 1);
662     }
663
664     if ((v->fcm == ILACE_FRAME) && fieldmv)
665         off = ((n > 1) ? s->linesize : 0) + (n & 1) * 8;
666     else
667         off = s->linesize * 4 * (n & 2) + (n & 1) * 8;
668
669     src_x = s->mb_x * 16 + (n & 1) * 8 + (mx >> 2);
670     if (!fieldmv)
671         src_y = s->mb_y * 16 + (n & 2) * 4 + (my >> 2);
672     else
673         src_y = s->mb_y * 16 + ((n > 1) ? 1 : 0) + (my >> 2);
674
675     if (v->profile != PROFILE_ADVANCED) {
676         src_x = av_clip(src_x, -16, s->mb_width  * 16);
677         src_y = av_clip(src_y, -16, s->mb_height * 16);
678     } else {
679         src_x = av_clip(src_x, -17, s->avctx->coded_width);
680         if (v->fcm == ILACE_FRAME) {
681             if (src_y & 1)
682                 src_y = av_clip(src_y, -17, s->avctx->coded_height + 1);
683             else
684                 src_y = av_clip(src_y, -18, s->avctx->coded_height);
685         } else {
686             src_y = av_clip(src_y, -18, s->avctx->coded_height + 1);
687         }
688     }
689
690     srcY += src_y * s->linesize + src_x;
691     if (v->field_mode && v->ref_field_type[dir])
692         srcY += s->current_picture_ptr->f->linesize[0];
693
694     if (fieldmv && !(src_y & 1))
695         v_edge_pos--;
696     if (fieldmv && (src_y & 1) && src_y < 4)
697         src_y--;
698     if (v->rangeredfrm || use_ic
699         || s->h_edge_pos < 13 || v_edge_pos < 23
700         || (unsigned)(src_x - s->mspel) > s->h_edge_pos - (mx & 3) - 8 - s->mspel * 2
701         || (unsigned)(src_y - (s->mspel << fieldmv)) > v_edge_pos - (my & 3) - ((8 + s->mspel * 2) << fieldmv)) {
702         srcY -= s->mspel * (1 + (s->linesize << fieldmv));
703         /* check emulate edge stride and offset */
704         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, srcY,
705                                  s->linesize, s->linesize,
706                                  9 + s->mspel * 2, (9 + s->mspel * 2) << fieldmv,
707                                  src_x - s->mspel, src_y - (s->mspel << fieldmv),
708                                  s->h_edge_pos, v_edge_pos);
709         srcY = s->edge_emu_buffer;
710         /* if we deal with range reduction we need to scale source blocks */
711         if (v->rangeredfrm) {
712             int i, j;
713             uint8_t *src;
714
715             src = srcY;
716             for (j = 0; j < 9 + s->mspel * 2; j++) {
717                 for (i = 0; i < 9 + s->mspel * 2; i++)
718                     src[i] = ((src[i] - 128) >> 1) + 128;
719                 src += s->linesize << fieldmv;
720             }
721         }
722         /* if we deal with intensity compensation we need to scale source blocks */
723         if (use_ic) {
724             int i, j;
725             uint8_t *src;
726
727             src = srcY;
728             for (j = 0; j < 9 + s->mspel * 2; j++) {
729                 int f = v->field_mode ? v->ref_field_type[dir] : (((j<<fieldmv)+src_y - (s->mspel << fieldmv)) & 1);
730                 for (i = 0; i < 9 + s->mspel * 2; i++)
731                     src[i] = luty[f][src[i]];
732                 src += s->linesize << fieldmv;
733             }
734         }
735         srcY += s->mspel * (1 + (s->linesize << fieldmv));
736     }
737
738     if (s->mspel) {
739         dxy = ((my & 3) << 2) | (mx & 3);
740         if (avg)
741             v->vc1dsp.avg_vc1_mspel_pixels_tab[dxy](s->dest[0] + off, srcY, s->linesize << fieldmv, v->rnd);
742         else
743             v->vc1dsp.put_vc1_mspel_pixels_tab[dxy](s->dest[0] + off, srcY, s->linesize << fieldmv, v->rnd);
744     } else { // hpel mc - always used for luma
745         dxy = (my & 2) | ((mx & 2) >> 1);
746         if (!v->rnd)
747             s->hdsp.put_pixels_tab[1][dxy](s->dest[0] + off, srcY, s->linesize, 8);
748         else
749             s->hdsp.put_no_rnd_pixels_tab[1][dxy](s->dest[0] + off, srcY, s->linesize, 8);
750     }
751 }
752
753 static av_always_inline int get_chroma_mv(int *mvx, int *mvy, int *a, int flag, int *tx, int *ty)
754 {
755     int idx, i;
756     static const int count[16] = { 0, 1, 1, 2, 1, 2, 2, 3, 1, 2, 2, 3, 2, 3, 3, 4};
757
758     idx =  ((a[3] != flag) << 3)
759          | ((a[2] != flag) << 2)
760          | ((a[1] != flag) << 1)
761          |  (a[0] != flag);
762     if (!idx) {
763         *tx = median4(mvx[0], mvx[1], mvx[2], mvx[3]);
764         *ty = median4(mvy[0], mvy[1], mvy[2], mvy[3]);
765         return 4;
766     } else if (count[idx] == 1) {
767         switch (idx) {
768         case 0x1:
769             *tx = mid_pred(mvx[1], mvx[2], mvx[3]);
770             *ty = mid_pred(mvy[1], mvy[2], mvy[3]);
771             return 3;
772         case 0x2:
773             *tx = mid_pred(mvx[0], mvx[2], mvx[3]);
774             *ty = mid_pred(mvy[0], mvy[2], mvy[3]);
775             return 3;
776         case 0x4:
777             *tx = mid_pred(mvx[0], mvx[1], mvx[3]);
778             *ty = mid_pred(mvy[0], mvy[1], mvy[3]);
779             return 3;
780         case 0x8:
781             *tx = mid_pred(mvx[0], mvx[1], mvx[2]);
782             *ty = mid_pred(mvy[0], mvy[1], mvy[2]);
783             return 3;
784         }
785     } else if (count[idx] == 2) {
786         int t1 = 0, t2 = 0;
787         for (i = 0; i < 3; i++)
788             if (!a[i]) {
789                 t1 = i;
790                 break;
791             }
792         for (i = t1 + 1; i < 4; i++)
793             if (!a[i]) {
794                 t2 = i;
795                 break;
796             }
797         *tx = (mvx[t1] + mvx[t2]) / 2;
798         *ty = (mvy[t1] + mvy[t2]) / 2;
799         return 2;
800     } else {
801         return 0;
802     }
803     return -1;
804 }
805
806 /** Do motion compensation for 4-MV macroblock - both chroma blocks
807  */
808 static void vc1_mc_4mv_chroma(VC1Context *v, int dir)
809 {
810     MpegEncContext *s = &v->s;
811     H264ChromaContext *h264chroma = &v->h264chroma;
812     uint8_t *srcU, *srcV;
813     int uvmx, uvmy, uvsrc_x, uvsrc_y;
814     int k, tx = 0, ty = 0;
815     int mvx[4], mvy[4], intra[4], mv_f[4];
816     int valid_count;
817     int chroma_ref_type = v->cur_field_type;
818     int v_edge_pos = s->v_edge_pos >> v->field_mode;
819     uint8_t (*lutuv)[256];
820     int use_ic;
821
822     if (!v->field_mode && !v->s.last_picture.f->data[0])
823         return;
824     if (s->flags & CODEC_FLAG_GRAY)
825         return;
826
827     for (k = 0; k < 4; k++) {
828         mvx[k] = s->mv[dir][k][0];
829         mvy[k] = s->mv[dir][k][1];
830         intra[k] = v->mb_type[0][s->block_index[k]];
831         if (v->field_mode)
832             mv_f[k] = v->mv_f[dir][s->block_index[k] + v->blocks_off];
833     }
834
835     /* calculate chroma MV vector from four luma MVs */
836     if (!v->field_mode || (v->field_mode && !v->numref)) {
837         valid_count = get_chroma_mv(mvx, mvy, intra, 0, &tx, &ty);
838         chroma_ref_type = v->reffield;
839         if (!valid_count) {
840             s->current_picture.motion_val[1][s->block_index[0] + v->blocks_off][0] = 0;
841             s->current_picture.motion_val[1][s->block_index[0] + v->blocks_off][1] = 0;
842             v->luma_mv[s->mb_x][0] = v->luma_mv[s->mb_x][1] = 0;
843             return; //no need to do MC for intra blocks
844         }
845     } else {
846         int dominant = 0;
847         if (mv_f[0] + mv_f[1] + mv_f[2] + mv_f[3] > 2)
848             dominant = 1;
849         valid_count = get_chroma_mv(mvx, mvy, mv_f, dominant, &tx, &ty);
850         if (dominant)
851             chroma_ref_type = !v->cur_field_type;
852     }
853     if (v->field_mode && chroma_ref_type == 1 && v->cur_field_type == 1 && !v->s.last_picture.f->data[0])
854         return;
855     s->current_picture.motion_val[1][s->block_index[0] + v->blocks_off][0] = tx;
856     s->current_picture.motion_val[1][s->block_index[0] + v->blocks_off][1] = ty;
857     uvmx = (tx + ((tx & 3) == 3)) >> 1;
858     uvmy = (ty + ((ty & 3) == 3)) >> 1;
859
860     v->luma_mv[s->mb_x][0] = uvmx;
861     v->luma_mv[s->mb_x][1] = uvmy;
862
863     if (v->fastuvmc) {
864         uvmx = uvmx + ((uvmx < 0) ? (uvmx & 1) : -(uvmx & 1));
865         uvmy = uvmy + ((uvmy < 0) ? (uvmy & 1) : -(uvmy & 1));
866     }
867     // Field conversion bias
868     if (v->cur_field_type != chroma_ref_type)
869         uvmy += 2 - 4 * chroma_ref_type;
870
871     uvsrc_x = s->mb_x * 8 + (uvmx >> 2);
872     uvsrc_y = s->mb_y * 8 + (uvmy >> 2);
873
874     if (v->profile != PROFILE_ADVANCED) {
875         uvsrc_x = av_clip(uvsrc_x, -8, s->mb_width  * 8);
876         uvsrc_y = av_clip(uvsrc_y, -8, s->mb_height * 8);
877     } else {
878         uvsrc_x = av_clip(uvsrc_x, -8, s->avctx->coded_width  >> 1);
879         uvsrc_y = av_clip(uvsrc_y, -8, s->avctx->coded_height >> 1);
880     }
881
882     if (!dir) {
883         if (v->field_mode && (v->cur_field_type != chroma_ref_type) && v->second_field) {
884             srcU = s->current_picture.f->data[1];
885             srcV = s->current_picture.f->data[2];
886             lutuv = v->curr_lutuv;
887             use_ic = v->curr_use_ic;
888         } else {
889             srcU = s->last_picture.f->data[1];
890             srcV = s->last_picture.f->data[2];
891             lutuv = v->last_lutuv;
892             use_ic = v->last_use_ic;
893         }
894     } else {
895         srcU = s->next_picture.f->data[1];
896         srcV = s->next_picture.f->data[2];
897         lutuv = v->next_lutuv;
898         use_ic = v->next_use_ic;
899     }
900
901     if (!srcU) {
902         av_log(v->s.avctx, AV_LOG_ERROR, "Referenced frame missing.\n");
903         return;
904     }
905
906     srcU += uvsrc_y * s->uvlinesize + uvsrc_x;
907     srcV += uvsrc_y * s->uvlinesize + uvsrc_x;
908
909     if (v->field_mode) {
910         if (chroma_ref_type) {
911             srcU += s->current_picture_ptr->f->linesize[1];
912             srcV += s->current_picture_ptr->f->linesize[2];
913         }
914     }
915
916     if (v->rangeredfrm || use_ic
917         || s->h_edge_pos < 18 || v_edge_pos < 18
918         || (unsigned)uvsrc_x > (s->h_edge_pos >> 1) - 9
919         || (unsigned)uvsrc_y > (v_edge_pos    >> 1) - 9) {
920         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, srcU,
921                                  s->uvlinesize, s->uvlinesize,
922                                  8 + 1, 8 + 1, uvsrc_x, uvsrc_y,
923                                  s->h_edge_pos >> 1, v_edge_pos >> 1);
924         s->vdsp.emulated_edge_mc(s->edge_emu_buffer + 16, srcV,
925                                  s->uvlinesize, s->uvlinesize,
926                                  8 + 1, 8 + 1, uvsrc_x, uvsrc_y,
927                                  s->h_edge_pos >> 1, v_edge_pos >> 1);
928         srcU = s->edge_emu_buffer;
929         srcV = s->edge_emu_buffer + 16;
930
931         /* if we deal with range reduction we need to scale source blocks */
932         if (v->rangeredfrm) {
933             int i, j;
934             uint8_t *src, *src2;
935
936             src  = srcU;
937             src2 = srcV;
938             for (j = 0; j < 9; j++) {
939                 for (i = 0; i < 9; i++) {
940                     src[i]  = ((src[i]  - 128) >> 1) + 128;
941                     src2[i] = ((src2[i] - 128) >> 1) + 128;
942                 }
943                 src  += s->uvlinesize;
944                 src2 += s->uvlinesize;
945             }
946         }
947         /* if we deal with intensity compensation we need to scale source blocks */
948         if (use_ic) {
949             int i, j;
950             uint8_t *src, *src2;
951
952             src  = srcU;
953             src2 = srcV;
954             for (j = 0; j < 9; j++) {
955                 int f = v->field_mode ? chroma_ref_type : ((j + uvsrc_y) & 1);
956                 for (i = 0; i < 9; i++) {
957                     src[i]  = lutuv[f][src[i]];
958                     src2[i] = lutuv[f][src2[i]];
959                 }
960                 src  += s->uvlinesize;
961                 src2 += s->uvlinesize;
962             }
963         }
964     }
965
966     /* Chroma MC always uses qpel bilinear */
967     uvmx = (uvmx & 3) << 1;
968     uvmy = (uvmy & 3) << 1;
969     if (!v->rnd) {
970         h264chroma->put_h264_chroma_pixels_tab[0](s->dest[1], srcU, s->uvlinesize, 8, uvmx, uvmy);
971         h264chroma->put_h264_chroma_pixels_tab[0](s->dest[2], srcV, s->uvlinesize, 8, uvmx, uvmy);
972     } else {
973         v->vc1dsp.put_no_rnd_vc1_chroma_pixels_tab[0](s->dest[1], srcU, s->uvlinesize, 8, uvmx, uvmy);
974         v->vc1dsp.put_no_rnd_vc1_chroma_pixels_tab[0](s->dest[2], srcV, s->uvlinesize, 8, uvmx, uvmy);
975     }
976 }
977
978 /** Do motion compensation for 4-MV interlaced frame chroma macroblock (both U and V)
979  */
980 static void vc1_mc_4mv_chroma4(VC1Context *v, int dir, int dir2, int avg)
981 {
982     MpegEncContext *s = &v->s;
983     H264ChromaContext *h264chroma = &v->h264chroma;
984     uint8_t *srcU, *srcV;
985     int uvsrc_x, uvsrc_y;
986     int uvmx_field[4], uvmy_field[4];
987     int i, off, tx, ty;
988     int fieldmv = v->blk_mv_type[s->block_index[0]];
989     static const int s_rndtblfield[16] = { 0, 0, 1, 2, 4, 4, 5, 6, 2, 2, 3, 8, 6, 6, 7, 12 };
990     int v_dist = fieldmv ? 1 : 4; // vertical offset for lower sub-blocks
991     int v_edge_pos = s->v_edge_pos >> 1;
992     int use_ic;
993     uint8_t (*lutuv)[256];
994
995     if (s->flags & CODEC_FLAG_GRAY)
996         return;
997
998     for (i = 0; i < 4; i++) {
999         int d = i < 2 ? dir: dir2;
1000         tx = s->mv[d][i][0];
1001         uvmx_field[i] = (tx + ((tx & 3) == 3)) >> 1;
1002         ty = s->mv[d][i][1];
1003         if (fieldmv)
1004             uvmy_field[i] = (ty >> 4) * 8 + s_rndtblfield[ty & 0xF];
1005         else
1006             uvmy_field[i] = (ty + ((ty & 3) == 3)) >> 1;
1007     }
1008
1009     for (i = 0; i < 4; i++) {
1010         off = (i & 1) * 4 + ((i & 2) ? v_dist * s->uvlinesize : 0);
1011         uvsrc_x = s->mb_x * 8 +  (i & 1) * 4           + (uvmx_field[i] >> 2);
1012         uvsrc_y = s->mb_y * 8 + ((i & 2) ? v_dist : 0) + (uvmy_field[i] >> 2);
1013         // FIXME: implement proper pull-back (see vc1cropmv.c, vc1CROPMV_ChromaPullBack())
1014         uvsrc_x = av_clip(uvsrc_x, -8, s->avctx->coded_width  >> 1);
1015         uvsrc_y = av_clip(uvsrc_y, -8, s->avctx->coded_height >> 1);
1016         if (i < 2 ? dir : dir2) {
1017             srcU = s->next_picture.f->data[1] + uvsrc_y * s->uvlinesize + uvsrc_x;
1018             srcV = s->next_picture.f->data[2] + uvsrc_y * s->uvlinesize + uvsrc_x;
1019             lutuv  = v->next_lutuv;
1020             use_ic = v->next_use_ic;
1021         } else {
1022             srcU = s->last_picture.f->data[1] + uvsrc_y * s->uvlinesize + uvsrc_x;
1023             srcV = s->last_picture.f->data[2] + uvsrc_y * s->uvlinesize + uvsrc_x;
1024             lutuv  = v->last_lutuv;
1025             use_ic = v->last_use_ic;
1026         }
1027         uvmx_field[i] = (uvmx_field[i] & 3) << 1;
1028         uvmy_field[i] = (uvmy_field[i] & 3) << 1;
1029
1030         if (fieldmv && !(uvsrc_y & 1))
1031             v_edge_pos--;
1032         if (fieldmv && (uvsrc_y & 1) && uvsrc_y < 2)
1033             uvsrc_y--;
1034         if (use_ic
1035             || s->h_edge_pos < 10 || v_edge_pos < (5 << fieldmv)
1036             || (unsigned)uvsrc_x > (s->h_edge_pos >> 1) - 5
1037             || (unsigned)uvsrc_y > v_edge_pos - (5 << fieldmv)) {
1038             s->vdsp.emulated_edge_mc(s->edge_emu_buffer, srcU,
1039                                      s->uvlinesize, s->uvlinesize,
1040                                      5, (5 << fieldmv), uvsrc_x, uvsrc_y,
1041                                      s->h_edge_pos >> 1, v_edge_pos);
1042             s->vdsp.emulated_edge_mc(s->edge_emu_buffer + 16, srcV,
1043                                      s->uvlinesize, s->uvlinesize,
1044                                      5, (5 << fieldmv), uvsrc_x, uvsrc_y,
1045                                      s->h_edge_pos >> 1, v_edge_pos);
1046             srcU = s->edge_emu_buffer;
1047             srcV = s->edge_emu_buffer + 16;
1048
1049             /* if we deal with intensity compensation we need to scale source blocks */
1050             if (use_ic) {
1051                 int i, j;
1052                 uint8_t *src, *src2;
1053
1054                 src  = srcU;
1055                 src2 = srcV;
1056                 for (j = 0; j < 5; j++) {
1057                     int f = (uvsrc_y + (j << fieldmv)) & 1;
1058                     for (i = 0; i < 5; i++) {
1059                         src[i]  = lutuv[f][src[i]];
1060                         src2[i] = lutuv[f][src2[i]];
1061                     }
1062                     src  += s->uvlinesize << fieldmv;
1063                     src2 += s->uvlinesize << fieldmv;
1064                 }
1065             }
1066         }
1067         if (avg) {
1068             if (!v->rnd) {
1069                 h264chroma->avg_h264_chroma_pixels_tab[1](s->dest[1] + off, srcU, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1070                 h264chroma->avg_h264_chroma_pixels_tab[1](s->dest[2] + off, srcV, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1071             } else {
1072                 v->vc1dsp.avg_no_rnd_vc1_chroma_pixels_tab[1](s->dest[1] + off, srcU, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1073                 v->vc1dsp.avg_no_rnd_vc1_chroma_pixels_tab[1](s->dest[2] + off, srcV, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1074             }
1075         } else {
1076             if (!v->rnd) {
1077                 h264chroma->put_h264_chroma_pixels_tab[1](s->dest[1] + off, srcU, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1078                 h264chroma->put_h264_chroma_pixels_tab[1](s->dest[2] + off, srcV, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1079             } else {
1080                 v->vc1dsp.put_no_rnd_vc1_chroma_pixels_tab[1](s->dest[1] + off, srcU, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1081                 v->vc1dsp.put_no_rnd_vc1_chroma_pixels_tab[1](s->dest[2] + off, srcV, s->uvlinesize << fieldmv, 4, uvmx_field[i], uvmy_field[i]);
1082             }
1083         }
1084     }
1085 }
1086
1087 /***********************************************************************/
1088 /**
1089  * @name VC-1 Block-level functions
1090  * @see 7.1.4, p91 and 8.1.1.7, p(1)04
1091  * @{
1092  */
1093
1094 /**
1095  * @def GET_MQUANT
1096  * @brief Get macroblock-level quantizer scale
1097  */
1098 #define GET_MQUANT()                                           \
1099     if (v->dquantfrm) {                                        \
1100         int edges = 0;                                         \
1101         if (v->dqprofile == DQPROFILE_ALL_MBS) {               \
1102             if (v->dqbilevel) {                                \
1103                 mquant = (get_bits1(gb)) ? v->altpq : v->pq;   \
1104             } else {                                           \
1105                 mqdiff = get_bits(gb, 3);                      \
1106                 if (mqdiff != 7)                               \
1107                     mquant = v->pq + mqdiff;                   \
1108                 else                                           \
1109                     mquant = get_bits(gb, 5);                  \
1110             }                                                  \
1111         }                                                      \
1112         if (v->dqprofile == DQPROFILE_SINGLE_EDGE)             \
1113             edges = 1 << v->dqsbedge;                          \
1114         else if (v->dqprofile == DQPROFILE_DOUBLE_EDGES)       \
1115             edges = (3 << v->dqsbedge) % 15;                   \
1116         else if (v->dqprofile == DQPROFILE_FOUR_EDGES)         \
1117             edges = 15;                                        \
1118         if ((edges&1) && !s->mb_x)                             \
1119             mquant = v->altpq;                                 \
1120         if ((edges&2) && s->first_slice_line)                  \
1121             mquant = v->altpq;                                 \
1122         if ((edges&4) && s->mb_x == (s->mb_width - 1))         \
1123             mquant = v->altpq;                                 \
1124         if ((edges&8) && s->mb_y == (s->mb_height - 1))        \
1125             mquant = v->altpq;                                 \
1126         if (!mquant || mquant > 31) {                          \
1127             av_log(v->s.avctx, AV_LOG_ERROR,                   \
1128                    "Overriding invalid mquant %d\n", mquant);  \
1129             mquant = 1;                                        \
1130         }                                                      \
1131     }
1132
1133 /**
1134  * @def GET_MVDATA(_dmv_x, _dmv_y)
1135  * @brief Get MV differentials
1136  * @see MVDATA decoding from 8.3.5.2, p(1)20
1137  * @param _dmv_x Horizontal differential for decoded MV
1138  * @param _dmv_y Vertical differential for decoded MV
1139  */
1140 #define GET_MVDATA(_dmv_x, _dmv_y)                                      \
1141     index = 1 + get_vlc2(gb, ff_vc1_mv_diff_vlc[s->mv_table_index].table, \
1142                          VC1_MV_DIFF_VLC_BITS, 2);                      \
1143     if (index > 36) {                                                   \
1144         mb_has_coeffs = 1;                                              \
1145         index -= 37;                                                    \
1146     } else                                                              \
1147         mb_has_coeffs = 0;                                              \
1148     s->mb_intra = 0;                                                    \
1149     if (!index) {                                                       \
1150         _dmv_x = _dmv_y = 0;                                            \
1151     } else if (index == 35) {                                           \
1152         _dmv_x = get_bits(gb, v->k_x - 1 + s->quarter_sample);          \
1153         _dmv_y = get_bits(gb, v->k_y - 1 + s->quarter_sample);          \
1154     } else if (index == 36) {                                           \
1155         _dmv_x = 0;                                                     \
1156         _dmv_y = 0;                                                     \
1157         s->mb_intra = 1;                                                \
1158     } else {                                                            \
1159         index1 = index % 6;                                             \
1160         if (!s->quarter_sample && index1 == 5) val = 1;                 \
1161         else                                   val = 0;                 \
1162         if (size_table[index1] - val > 0)                               \
1163             val = get_bits(gb, size_table[index1] - val);               \
1164         else                                   val = 0;                 \
1165         sign = 0 - (val&1);                                             \
1166         _dmv_x = (sign ^ ((val>>1) + offset_table[index1])) - sign;     \
1167                                                                         \
1168         index1 = index / 6;                                             \
1169         if (!s->quarter_sample && index1 == 5) val = 1;                 \
1170         else                                   val = 0;                 \
1171         if (size_table[index1] - val > 0)                               \
1172             val = get_bits(gb, size_table[index1] - val);               \
1173         else                                   val = 0;                 \
1174         sign = 0 - (val & 1);                                           \
1175         _dmv_y = (sign ^ ((val >> 1) + offset_table[index1])) - sign;   \
1176     }
1177
1178 static av_always_inline void get_mvdata_interlaced(VC1Context *v, int *dmv_x,
1179                                                    int *dmv_y, int *pred_flag)
1180 {
1181     int index, index1;
1182     int extend_x = 0, extend_y = 0;
1183     GetBitContext *gb = &v->s.gb;
1184     int bits, esc;
1185     int val, sign;
1186     const int* offs_tab;
1187
1188     if (v->numref) {
1189         bits = VC1_2REF_MVDATA_VLC_BITS;
1190         esc  = 125;
1191     } else {
1192         bits = VC1_1REF_MVDATA_VLC_BITS;
1193         esc  = 71;
1194     }
1195     switch (v->dmvrange) {
1196     case 1:
1197         extend_x = 1;
1198         break;
1199     case 2:
1200         extend_y = 1;
1201         break;
1202     case 3:
1203         extend_x = extend_y = 1;
1204         break;
1205     }
1206     index = get_vlc2(gb, v->imv_vlc->table, bits, 3);
1207     if (index == esc) {
1208         *dmv_x = get_bits(gb, v->k_x);
1209         *dmv_y = get_bits(gb, v->k_y);
1210         if (v->numref) {
1211             if (pred_flag) {
1212                 *pred_flag = *dmv_y & 1;
1213                 *dmv_y     = (*dmv_y + *pred_flag) >> 1;
1214             } else {
1215                 *dmv_y     = (*dmv_y + (*dmv_y & 1)) >> 1;
1216             }
1217         }
1218     }
1219     else {
1220         if (extend_x)
1221             offs_tab = offset_table2;
1222         else
1223             offs_tab = offset_table1;
1224         index1 = (index + 1) % 9;
1225         if (index1 != 0) {
1226             val    = get_bits(gb, index1 + extend_x);
1227             sign   = 0 -(val & 1);
1228             *dmv_x = (sign ^ ((val >> 1) + offs_tab[index1])) - sign;
1229         } else
1230             *dmv_x = 0;
1231         if (extend_y)
1232             offs_tab = offset_table2;
1233         else
1234             offs_tab = offset_table1;
1235         index1 = (index + 1) / 9;
1236         if (index1 > v->numref) {
1237             val    = get_bits(gb, (index1 + (extend_y << v->numref)) >> v->numref);
1238             sign   = 0 - (val & 1);
1239             *dmv_y = (sign ^ ((val >> 1) + offs_tab[index1 >> v->numref])) - sign;
1240         } else
1241             *dmv_y = 0;
1242         if (v->numref && pred_flag)
1243             *pred_flag = index1 & 1;
1244     }
1245 }
1246
1247 static av_always_inline int scaleforsame_x(VC1Context *v, int n /* MV */, int dir)
1248 {
1249     int scaledvalue, refdist;
1250     int scalesame1, scalesame2;
1251     int scalezone1_x, zone1offset_x;
1252     int table_index = dir ^ v->second_field;
1253
1254     if (v->s.pict_type != AV_PICTURE_TYPE_B)
1255         refdist = v->refdist;
1256     else
1257         refdist = dir ? v->brfd : v->frfd;
1258     if (refdist > 3)
1259         refdist = 3;
1260     scalesame1    = ff_vc1_field_mvpred_scales[table_index][1][refdist];
1261     scalesame2    = ff_vc1_field_mvpred_scales[table_index][2][refdist];
1262     scalezone1_x  = ff_vc1_field_mvpred_scales[table_index][3][refdist];
1263     zone1offset_x = ff_vc1_field_mvpred_scales[table_index][5][refdist];
1264
1265     if (FFABS(n) > 255)
1266         scaledvalue = n;
1267     else {
1268         if (FFABS(n) < scalezone1_x)
1269             scaledvalue = (n * scalesame1) >> 8;
1270         else {
1271             if (n < 0)
1272                 scaledvalue = ((n * scalesame2) >> 8) - zone1offset_x;
1273             else
1274                 scaledvalue = ((n * scalesame2) >> 8) + zone1offset_x;
1275         }
1276     }
1277     return av_clip(scaledvalue, -v->range_x, v->range_x - 1);
1278 }
1279
1280 static av_always_inline int scaleforsame_y(VC1Context *v, int i, int n /* MV */, int dir)
1281 {
1282     int scaledvalue, refdist;
1283     int scalesame1, scalesame2;
1284     int scalezone1_y, zone1offset_y;
1285     int table_index = dir ^ v->second_field;
1286
1287     if (v->s.pict_type != AV_PICTURE_TYPE_B)
1288         refdist = v->refdist;
1289     else
1290         refdist = dir ? v->brfd : v->frfd;
1291     if (refdist > 3)
1292         refdist = 3;
1293     scalesame1    = ff_vc1_field_mvpred_scales[table_index][1][refdist];
1294     scalesame2    = ff_vc1_field_mvpred_scales[table_index][2][refdist];
1295     scalezone1_y  = ff_vc1_field_mvpred_scales[table_index][4][refdist];
1296     zone1offset_y = ff_vc1_field_mvpred_scales[table_index][6][refdist];
1297
1298     if (FFABS(n) > 63)
1299         scaledvalue = n;
1300     else {
1301         if (FFABS(n) < scalezone1_y)
1302             scaledvalue = (n * scalesame1) >> 8;
1303         else {
1304             if (n < 0)
1305                 scaledvalue = ((n * scalesame2) >> 8) - zone1offset_y;
1306             else
1307                 scaledvalue = ((n * scalesame2) >> 8) + zone1offset_y;
1308         }
1309     }
1310
1311     if (v->cur_field_type && !v->ref_field_type[dir])
1312         return av_clip(scaledvalue, -v->range_y / 2 + 1, v->range_y / 2);
1313     else
1314         return av_clip(scaledvalue, -v->range_y / 2, v->range_y / 2 - 1);
1315 }
1316
1317 static av_always_inline int scaleforopp_x(VC1Context *v, int n /* MV */)
1318 {
1319     int scalezone1_x, zone1offset_x;
1320     int scaleopp1, scaleopp2, brfd;
1321     int scaledvalue;
1322
1323     brfd = FFMIN(v->brfd, 3);
1324     scalezone1_x  = ff_vc1_b_field_mvpred_scales[3][brfd];
1325     zone1offset_x = ff_vc1_b_field_mvpred_scales[5][brfd];
1326     scaleopp1     = ff_vc1_b_field_mvpred_scales[1][brfd];
1327     scaleopp2     = ff_vc1_b_field_mvpred_scales[2][brfd];
1328
1329     if (FFABS(n) > 255)
1330         scaledvalue = n;
1331     else {
1332         if (FFABS(n) < scalezone1_x)
1333             scaledvalue = (n * scaleopp1) >> 8;
1334         else {
1335             if (n < 0)
1336                 scaledvalue = ((n * scaleopp2) >> 8) - zone1offset_x;
1337             else
1338                 scaledvalue = ((n * scaleopp2) >> 8) + zone1offset_x;
1339         }
1340     }
1341     return av_clip(scaledvalue, -v->range_x, v->range_x - 1);
1342 }
1343
1344 static av_always_inline int scaleforopp_y(VC1Context *v, int n /* MV */, int dir)
1345 {
1346     int scalezone1_y, zone1offset_y;
1347     int scaleopp1, scaleopp2, brfd;
1348     int scaledvalue;
1349
1350     brfd = FFMIN(v->brfd, 3);
1351     scalezone1_y  = ff_vc1_b_field_mvpred_scales[4][brfd];
1352     zone1offset_y = ff_vc1_b_field_mvpred_scales[6][brfd];
1353     scaleopp1     = ff_vc1_b_field_mvpred_scales[1][brfd];
1354     scaleopp2     = ff_vc1_b_field_mvpred_scales[2][brfd];
1355
1356     if (FFABS(n) > 63)
1357         scaledvalue = n;
1358     else {
1359         if (FFABS(n) < scalezone1_y)
1360             scaledvalue = (n * scaleopp1) >> 8;
1361         else {
1362             if (n < 0)
1363                 scaledvalue = ((n * scaleopp2) >> 8) - zone1offset_y;
1364             else
1365                 scaledvalue = ((n * scaleopp2) >> 8) + zone1offset_y;
1366         }
1367     }
1368     if (v->cur_field_type && !v->ref_field_type[dir]) {
1369         return av_clip(scaledvalue, -v->range_y / 2 + 1, v->range_y / 2);
1370     } else {
1371         return av_clip(scaledvalue, -v->range_y / 2, v->range_y / 2 - 1);
1372     }
1373 }
1374
1375 static av_always_inline int scaleforsame(VC1Context *v, int i, int n /* MV */,
1376                                          int dim, int dir)
1377 {
1378     int brfd, scalesame;
1379     int hpel = 1 - v->s.quarter_sample;
1380
1381     n >>= hpel;
1382     if (v->s.pict_type != AV_PICTURE_TYPE_B || v->second_field || !dir) {
1383         if (dim)
1384             n = scaleforsame_y(v, i, n, dir) << hpel;
1385         else
1386             n = scaleforsame_x(v, n, dir) << hpel;
1387         return n;
1388     }
1389     brfd      = FFMIN(v->brfd, 3);
1390     scalesame = ff_vc1_b_field_mvpred_scales[0][brfd];
1391
1392     n = (n * scalesame >> 8) << hpel;
1393     return n;
1394 }
1395
1396 static av_always_inline int scaleforopp(VC1Context *v, int n /* MV */,
1397                                         int dim, int dir)
1398 {
1399     int refdist, scaleopp;
1400     int hpel = 1 - v->s.quarter_sample;
1401
1402     n >>= hpel;
1403     if (v->s.pict_type == AV_PICTURE_TYPE_B && !v->second_field && dir == 1) {
1404         if (dim)
1405             n = scaleforopp_y(v, n, dir) << hpel;
1406         else
1407             n = scaleforopp_x(v, n) << hpel;
1408         return n;
1409     }
1410     if (v->s.pict_type != AV_PICTURE_TYPE_B)
1411         refdist = FFMIN(v->refdist, 3);
1412     else
1413         refdist = dir ? v->brfd : v->frfd;
1414     scaleopp = ff_vc1_field_mvpred_scales[dir ^ v->second_field][0][refdist];
1415
1416     n = (n * scaleopp >> 8) << hpel;
1417     return n;
1418 }
1419
1420 /** Predict and set motion vector
1421  */
1422 static inline void vc1_pred_mv(VC1Context *v, int n, int dmv_x, int dmv_y,
1423                                int mv1, int r_x, int r_y, uint8_t* is_intra,
1424                                int pred_flag, int dir)
1425 {
1426     MpegEncContext *s = &v->s;
1427     int xy, wrap, off = 0;
1428     int16_t *A, *B, *C;
1429     int px, py;
1430     int sum;
1431     int mixedmv_pic, num_samefield = 0, num_oppfield = 0;
1432     int opposite, a_f, b_f, c_f;
1433     int16_t field_predA[2];
1434     int16_t field_predB[2];
1435     int16_t field_predC[2];
1436     int a_valid, b_valid, c_valid;
1437     int hybridmv_thresh, y_bias = 0;
1438
1439     if (v->mv_mode == MV_PMODE_MIXED_MV ||
1440         ((v->mv_mode == MV_PMODE_INTENSITY_COMP) && (v->mv_mode2 == MV_PMODE_MIXED_MV)))
1441         mixedmv_pic = 1;
1442     else
1443         mixedmv_pic = 0;
1444     /* scale MV difference to be quad-pel */
1445     dmv_x <<= 1 - s->quarter_sample;
1446     dmv_y <<= 1 - s->quarter_sample;
1447
1448     wrap = s->b8_stride;
1449     xy   = s->block_index[n];
1450
1451     if (s->mb_intra) {
1452         s->mv[0][n][0] = s->current_picture.motion_val[0][xy + v->blocks_off][0] = 0;
1453         s->mv[0][n][1] = s->current_picture.motion_val[0][xy + v->blocks_off][1] = 0;
1454         s->current_picture.motion_val[1][xy + v->blocks_off][0] = 0;
1455         s->current_picture.motion_val[1][xy + v->blocks_off][1] = 0;
1456         if (mv1) { /* duplicate motion data for 1-MV block */
1457             s->current_picture.motion_val[0][xy + 1 + v->blocks_off][0]        = 0;
1458             s->current_picture.motion_val[0][xy + 1 + v->blocks_off][1]        = 0;
1459             s->current_picture.motion_val[0][xy + wrap + v->blocks_off][0]     = 0;
1460             s->current_picture.motion_val[0][xy + wrap + v->blocks_off][1]     = 0;
1461             s->current_picture.motion_val[0][xy + wrap + 1 + v->blocks_off][0] = 0;
1462             s->current_picture.motion_val[0][xy + wrap + 1 + v->blocks_off][1] = 0;
1463             v->luma_mv[s->mb_x][0] = v->luma_mv[s->mb_x][1] = 0;
1464             s->current_picture.motion_val[1][xy + 1 + v->blocks_off][0]        = 0;
1465             s->current_picture.motion_val[1][xy + 1 + v->blocks_off][1]        = 0;
1466             s->current_picture.motion_val[1][xy + wrap][0]                     = 0;
1467             s->current_picture.motion_val[1][xy + wrap + v->blocks_off][1]     = 0;
1468             s->current_picture.motion_val[1][xy + wrap + 1 + v->blocks_off][0] = 0;
1469             s->current_picture.motion_val[1][xy + wrap + 1 + v->blocks_off][1] = 0;
1470         }
1471         return;
1472     }
1473
1474     C = s->current_picture.motion_val[dir][xy -    1 + v->blocks_off];
1475     A = s->current_picture.motion_val[dir][xy - wrap + v->blocks_off];
1476     if (mv1) {
1477         if (v->field_mode && mixedmv_pic)
1478             off = (s->mb_x == (s->mb_width - 1)) ? -2 : 2;
1479         else
1480             off = (s->mb_x == (s->mb_width - 1)) ? -1 : 2;
1481     } else {
1482         //in 4-MV mode different blocks have different B predictor position
1483         switch (n) {
1484         case 0:
1485             off = (s->mb_x > 0) ? -1 : 1;
1486             break;
1487         case 1:
1488             off = (s->mb_x == (s->mb_width - 1)) ? -1 : 1;
1489             break;
1490         case 2:
1491             off = 1;
1492             break;
1493         case 3:
1494             off = -1;
1495         }
1496     }
1497     B = s->current_picture.motion_val[dir][xy - wrap + off + v->blocks_off];
1498
1499     a_valid = !s->first_slice_line || (n == 2 || n == 3);
1500     b_valid = a_valid && (s->mb_width > 1);
1501     c_valid = s->mb_x || (n == 1 || n == 3);
1502     if (v->field_mode) {
1503         a_valid = a_valid && !is_intra[xy - wrap];
1504         b_valid = b_valid && !is_intra[xy - wrap + off];
1505         c_valid = c_valid && !is_intra[xy - 1];
1506     }
1507
1508     if (a_valid) {
1509         a_f = v->mv_f[dir][xy - wrap + v->blocks_off];
1510         num_oppfield  += a_f;
1511         num_samefield += 1 - a_f;
1512         field_predA[0] = A[0];
1513         field_predA[1] = A[1];
1514     } else {
1515         field_predA[0] = field_predA[1] = 0;
1516         a_f = 0;
1517     }
1518     if (b_valid) {
1519         b_f = v->mv_f[dir][xy - wrap + off + v->blocks_off];
1520         num_oppfield  += b_f;
1521         num_samefield += 1 - b_f;
1522         field_predB[0] = B[0];
1523         field_predB[1] = B[1];
1524     } else {
1525         field_predB[0] = field_predB[1] = 0;
1526         b_f = 0;
1527     }
1528     if (c_valid) {
1529         c_f = v->mv_f[dir][xy - 1 + v->blocks_off];
1530         num_oppfield  += c_f;
1531         num_samefield += 1 - c_f;
1532         field_predC[0] = C[0];
1533         field_predC[1] = C[1];
1534     } else {
1535         field_predC[0] = field_predC[1] = 0;
1536         c_f = 0;
1537     }
1538
1539     if (v->field_mode) {
1540         if (!v->numref)
1541             // REFFIELD determines if the last field or the second-last field is
1542             // to be used as reference
1543             opposite = 1 - v->reffield;
1544         else {
1545             if (num_samefield <= num_oppfield)
1546                 opposite = 1 - pred_flag;
1547             else
1548                 opposite = pred_flag;
1549         }
1550     } else
1551         opposite = 0;
1552     if (opposite) {
1553         if (a_valid && !a_f) {
1554             field_predA[0] = scaleforopp(v, field_predA[0], 0, dir);
1555             field_predA[1] = scaleforopp(v, field_predA[1], 1, dir);
1556         }
1557         if (b_valid && !b_f) {
1558             field_predB[0] = scaleforopp(v, field_predB[0], 0, dir);
1559             field_predB[1] = scaleforopp(v, field_predB[1], 1, dir);
1560         }
1561         if (c_valid && !c_f) {
1562             field_predC[0] = scaleforopp(v, field_predC[0], 0, dir);
1563             field_predC[1] = scaleforopp(v, field_predC[1], 1, dir);
1564         }
1565         v->mv_f[dir][xy + v->blocks_off] = 1;
1566         v->ref_field_type[dir] = !v->cur_field_type;
1567     } else {
1568         if (a_valid && a_f) {
1569             field_predA[0] = scaleforsame(v, n, field_predA[0], 0, dir);
1570             field_predA[1] = scaleforsame(v, n, field_predA[1], 1, dir);
1571         }
1572         if (b_valid && b_f) {
1573             field_predB[0] = scaleforsame(v, n, field_predB[0], 0, dir);
1574             field_predB[1] = scaleforsame(v, n, field_predB[1], 1, dir);
1575         }
1576         if (c_valid && c_f) {
1577             field_predC[0] = scaleforsame(v, n, field_predC[0], 0, dir);
1578             field_predC[1] = scaleforsame(v, n, field_predC[1], 1, dir);
1579         }
1580         v->mv_f[dir][xy + v->blocks_off] = 0;
1581         v->ref_field_type[dir] = v->cur_field_type;
1582     }
1583
1584     if (a_valid) {
1585         px = field_predA[0];
1586         py = field_predA[1];
1587     } else if (c_valid) {
1588         px = field_predC[0];
1589         py = field_predC[1];
1590     } else if (b_valid) {
1591         px = field_predB[0];
1592         py = field_predB[1];
1593     } else {
1594         px = 0;
1595         py = 0;
1596     }
1597
1598     if (num_samefield + num_oppfield > 1) {
1599         px = mid_pred(field_predA[0], field_predB[0], field_predC[0]);
1600         py = mid_pred(field_predA[1], field_predB[1], field_predC[1]);
1601     }
1602
1603     /* Pullback MV as specified in 8.3.5.3.4 */
1604     if (!v->field_mode) {
1605         int qx, qy, X, Y;
1606         qx = (s->mb_x << 6) + ((n == 1 || n == 3) ? 32 : 0);
1607         qy = (s->mb_y << 6) + ((n == 2 || n == 3) ? 32 : 0);
1608         X  = (s->mb_width  << 6) - 4;
1609         Y  = (s->mb_height << 6) - 4;
1610         if (mv1) {
1611             if (qx + px < -60) px = -60 - qx;
1612             if (qy + py < -60) py = -60 - qy;
1613         } else {
1614             if (qx + px < -28) px = -28 - qx;
1615             if (qy + py < -28) py = -28 - qy;
1616         }
1617         if (qx + px > X) px = X - qx;
1618         if (qy + py > Y) py = Y - qy;
1619     }
1620
1621     if (!v->field_mode || s->pict_type != AV_PICTURE_TYPE_B) {
1622         /* Calculate hybrid prediction as specified in 8.3.5.3.5 (also 10.3.5.4.3.5) */
1623         hybridmv_thresh = 32;
1624         if (a_valid && c_valid) {
1625             if (is_intra[xy - wrap])
1626                 sum = FFABS(px) + FFABS(py);
1627             else
1628                 sum = FFABS(px - field_predA[0]) + FFABS(py - field_predA[1]);
1629             if (sum > hybridmv_thresh) {
1630                 if (get_bits1(&s->gb)) {     // read HYBRIDPRED bit
1631                     px = field_predA[0];
1632                     py = field_predA[1];
1633                 } else {
1634                     px = field_predC[0];
1635                     py = field_predC[1];
1636                 }
1637             } else {
1638                 if (is_intra[xy - 1])
1639                     sum = FFABS(px) + FFABS(py);
1640                 else
1641                     sum = FFABS(px - field_predC[0]) + FFABS(py - field_predC[1]);
1642                 if (sum > hybridmv_thresh) {
1643                     if (get_bits1(&s->gb)) {
1644                         px = field_predA[0];
1645                         py = field_predA[1];
1646                     } else {
1647                         px = field_predC[0];
1648                         py = field_predC[1];
1649                     }
1650                 }
1651             }
1652         }
1653     }
1654
1655     if (v->field_mode && v->numref)
1656         r_y >>= 1;
1657     if (v->field_mode && v->cur_field_type && v->ref_field_type[dir] == 0)
1658         y_bias = 1;
1659     /* store MV using signed modulus of MV range defined in 4.11 */
1660     s->mv[dir][n][0] = s->current_picture.motion_val[dir][xy + v->blocks_off][0] = ((px + dmv_x + r_x) & ((r_x << 1) - 1)) - r_x;
1661     s->mv[dir][n][1] = s->current_picture.motion_val[dir][xy + v->blocks_off][1] = ((py + dmv_y + r_y - y_bias) & ((r_y << 1) - 1)) - r_y + y_bias;
1662     if (mv1) { /* duplicate motion data for 1-MV block */
1663         s->current_picture.motion_val[dir][xy +    1 +     v->blocks_off][0] = s->current_picture.motion_val[dir][xy + v->blocks_off][0];
1664         s->current_picture.motion_val[dir][xy +    1 +     v->blocks_off][1] = s->current_picture.motion_val[dir][xy + v->blocks_off][1];
1665         s->current_picture.motion_val[dir][xy + wrap +     v->blocks_off][0] = s->current_picture.motion_val[dir][xy + v->blocks_off][0];
1666         s->current_picture.motion_val[dir][xy + wrap +     v->blocks_off][1] = s->current_picture.motion_val[dir][xy + v->blocks_off][1];
1667         s->current_picture.motion_val[dir][xy + wrap + 1 + v->blocks_off][0] = s->current_picture.motion_val[dir][xy + v->blocks_off][0];
1668         s->current_picture.motion_val[dir][xy + wrap + 1 + v->blocks_off][1] = s->current_picture.motion_val[dir][xy + v->blocks_off][1];
1669         v->mv_f[dir][xy +    1 + v->blocks_off] = v->mv_f[dir][xy +            v->blocks_off];
1670         v->mv_f[dir][xy + wrap + v->blocks_off] = v->mv_f[dir][xy + wrap + 1 + v->blocks_off] = v->mv_f[dir][xy + v->blocks_off];
1671     }
1672 }
1673
1674 /** Predict and set motion vector for interlaced frame picture MBs
1675  */
1676 static inline void vc1_pred_mv_intfr(VC1Context *v, int n, int dmv_x, int dmv_y,
1677                                      int mvn, int r_x, int r_y, uint8_t* is_intra, int dir)
1678 {
1679     MpegEncContext *s = &v->s;
1680     int xy, wrap, off = 0;
1681     int A[2], B[2], C[2];
1682     int px, py;
1683     int a_valid = 0, b_valid = 0, c_valid = 0;
1684     int field_a, field_b, field_c; // 0: same, 1: opposit
1685     int total_valid, num_samefield, num_oppfield;
1686     int pos_c, pos_b, n_adj;
1687
1688     wrap = s->b8_stride;
1689     xy = s->block_index[n];
1690
1691     if (s->mb_intra) {
1692         s->mv[0][n][0] = s->current_picture.motion_val[0][xy][0] = 0;
1693         s->mv[0][n][1] = s->current_picture.motion_val[0][xy][1] = 0;
1694         s->current_picture.motion_val[1][xy][0] = 0;
1695         s->current_picture.motion_val[1][xy][1] = 0;
1696         if (mvn == 1) { /* duplicate motion data for 1-MV block */
1697             s->current_picture.motion_val[0][xy + 1][0]        = 0;
1698             s->current_picture.motion_val[0][xy + 1][1]        = 0;
1699             s->current_picture.motion_val[0][xy + wrap][0]     = 0;
1700             s->current_picture.motion_val[0][xy + wrap][1]     = 0;
1701             s->current_picture.motion_val[0][xy + wrap + 1][0] = 0;
1702             s->current_picture.motion_val[0][xy + wrap + 1][1] = 0;
1703             v->luma_mv[s->mb_x][0] = v->luma_mv[s->mb_x][1] = 0;
1704             s->current_picture.motion_val[1][xy + 1][0]        = 0;
1705             s->current_picture.motion_val[1][xy + 1][1]        = 0;
1706             s->current_picture.motion_val[1][xy + wrap][0]     = 0;
1707             s->current_picture.motion_val[1][xy + wrap][1]     = 0;
1708             s->current_picture.motion_val[1][xy + wrap + 1][0] = 0;
1709             s->current_picture.motion_val[1][xy + wrap + 1][1] = 0;
1710         }
1711         return;
1712     }
1713
1714     off = ((n == 0) || (n == 1)) ? 1 : -1;
1715     /* predict A */
1716     if (s->mb_x || (n == 1) || (n == 3)) {
1717         if ((v->blk_mv_type[xy]) // current block (MB) has a field MV
1718             || (!v->blk_mv_type[xy] && !v->blk_mv_type[xy - 1])) { // or both have frame MV
1719             A[0] = s->current_picture.motion_val[dir][xy - 1][0];
1720             A[1] = s->current_picture.motion_val[dir][xy - 1][1];
1721             a_valid = 1;
1722         } else { // current block has frame mv and cand. has field MV (so average)
1723             A[0] = (s->current_picture.motion_val[dir][xy - 1][0]
1724                     + s->current_picture.motion_val[dir][xy - 1 + off * wrap][0] + 1) >> 1;
1725             A[1] = (s->current_picture.motion_val[dir][xy - 1][1]
1726                     + s->current_picture.motion_val[dir][xy - 1 + off * wrap][1] + 1) >> 1;
1727             a_valid = 1;
1728         }
1729         if (!(n & 1) && v->is_intra[s->mb_x - 1]) {
1730             a_valid = 0;
1731             A[0] = A[1] = 0;
1732         }
1733     } else
1734         A[0] = A[1] = 0;
1735     /* Predict B and C */
1736     B[0] = B[1] = C[0] = C[1] = 0;
1737     if (n == 0 || n == 1 || v->blk_mv_type[xy]) {
1738         if (!s->first_slice_line) {
1739             if (!v->is_intra[s->mb_x - s->mb_stride]) {
1740                 b_valid = 1;
1741                 n_adj   = n | 2;
1742                 pos_b   = s->block_index[n_adj] - 2 * wrap;
1743                 if (v->blk_mv_type[pos_b] && v->blk_mv_type[xy]) {
1744                     n_adj = (n & 2) | (n & 1);
1745                 }
1746                 B[0] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap][0];
1747                 B[1] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap][1];
1748                 if (v->blk_mv_type[pos_b] && !v->blk_mv_type[xy]) {
1749                     B[0] = (B[0] + s->current_picture.motion_val[dir][s->block_index[n_adj ^ 2] - 2 * wrap][0] + 1) >> 1;
1750                     B[1] = (B[1] + s->current_picture.motion_val[dir][s->block_index[n_adj ^ 2] - 2 * wrap][1] + 1) >> 1;
1751                 }
1752             }
1753             if (s->mb_width > 1) {
1754                 if (!v->is_intra[s->mb_x - s->mb_stride + 1]) {
1755                     c_valid = 1;
1756                     n_adj   = 2;
1757                     pos_c   = s->block_index[2] - 2 * wrap + 2;
1758                     if (v->blk_mv_type[pos_c] && v->blk_mv_type[xy]) {
1759                         n_adj = n & 2;
1760                     }
1761                     C[0] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap + 2][0];
1762                     C[1] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap + 2][1];
1763                     if (v->blk_mv_type[pos_c] && !v->blk_mv_type[xy]) {
1764                         C[0] = (1 + C[0] + (s->current_picture.motion_val[dir][s->block_index[n_adj ^ 2] - 2 * wrap + 2][0])) >> 1;
1765                         C[1] = (1 + C[1] + (s->current_picture.motion_val[dir][s->block_index[n_adj ^ 2] - 2 * wrap + 2][1])) >> 1;
1766                     }
1767                     if (s->mb_x == s->mb_width - 1) {
1768                         if (!v->is_intra[s->mb_x - s->mb_stride - 1]) {
1769                             c_valid = 1;
1770                             n_adj   = 3;
1771                             pos_c   = s->block_index[3] - 2 * wrap - 2;
1772                             if (v->blk_mv_type[pos_c] && v->blk_mv_type[xy]) {
1773                                 n_adj = n | 1;
1774                             }
1775                             C[0] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap - 2][0];
1776                             C[1] = s->current_picture.motion_val[dir][s->block_index[n_adj] - 2 * wrap - 2][1];
1777                             if (v->blk_mv_type[pos_c] && !v->blk_mv_type[xy]) {
1778                                 C[0] = (1 + C[0] + s->current_picture.motion_val[dir][s->block_index[1] - 2 * wrap - 2][0]) >> 1;
1779                                 C[1] = (1 + C[1] + s->current_picture.motion_val[dir][s->block_index[1] - 2 * wrap - 2][1]) >> 1;
1780                             }
1781                         } else
1782                             c_valid = 0;
1783                     }
1784                 }
1785             }
1786         }
1787     } else {
1788         pos_b   = s->block_index[1];
1789         b_valid = 1;
1790         B[0]    = s->current_picture.motion_val[dir][pos_b][0];
1791         B[1]    = s->current_picture.motion_val[dir][pos_b][1];
1792         pos_c   = s->block_index[0];
1793         c_valid = 1;
1794         C[0]    = s->current_picture.motion_val[dir][pos_c][0];
1795         C[1]    = s->current_picture.motion_val[dir][pos_c][1];
1796     }
1797
1798     total_valid = a_valid + b_valid + c_valid;
1799     // check if predictor A is out of bounds
1800     if (!s->mb_x && !(n == 1 || n == 3)) {
1801         A[0] = A[1] = 0;
1802     }
1803     // check if predictor B is out of bounds
1804     if ((s->first_slice_line && v->blk_mv_type[xy]) || (s->first_slice_line && !(n & 2))) {
1805         B[0] = B[1] = C[0] = C[1] = 0;
1806     }
1807     if (!v->blk_mv_type[xy]) {
1808         if (s->mb_width == 1) {
1809             px = B[0];
1810             py = B[1];
1811         } else {
1812             if (total_valid >= 2) {
1813                 px = mid_pred(A[0], B[0], C[0]);
1814                 py = mid_pred(A[1], B[1], C[1]);
1815             } else if (total_valid) {
1816                 if (a_valid) { px = A[0]; py = A[1]; }
1817                 if (b_valid) { px = B[0]; py = B[1]; }
1818                 if (c_valid) { px = C[0]; py = C[1]; }
1819             } else
1820                 px = py = 0;
1821         }
1822     } else {
1823         if (a_valid)
1824             field_a = (A[1] & 4) ? 1 : 0;
1825         else
1826             field_a = 0;
1827         if (b_valid)
1828             field_b = (B[1] & 4) ? 1 : 0;
1829         else
1830             field_b = 0;
1831         if (c_valid)
1832             field_c = (C[1] & 4) ? 1 : 0;
1833         else
1834             field_c = 0;
1835
1836         num_oppfield  = field_a + field_b + field_c;
1837         num_samefield = total_valid - num_oppfield;
1838         if (total_valid == 3) {
1839             if ((num_samefield == 3) || (num_oppfield == 3)) {
1840                 px = mid_pred(A[0], B[0], C[0]);
1841                 py = mid_pred(A[1], B[1], C[1]);
1842             } else if (num_samefield >= num_oppfield) {
1843                 /* take one MV from same field set depending on priority
1844                 the check for B may not be necessary */
1845                 px = !field_a ? A[0] : B[0];
1846                 py = !field_a ? A[1] : B[1];
1847             } else {
1848                 px =  field_a ? A[0] : B[0];
1849                 py =  field_a ? A[1] : B[1];
1850             }
1851         } else if (total_valid == 2) {
1852             if (num_samefield >= num_oppfield) {
1853                 if (!field_a && a_valid) {
1854                     px = A[0];
1855                     py = A[1];
1856                 } else if (!field_b && b_valid) {
1857                     px = B[0];
1858                     py = B[1];
1859                 } else if (c_valid) {
1860                     px = C[0];
1861                     py = C[1];
1862                 } else px = py = 0;
1863             } else {
1864                 if (field_a && a_valid) {
1865                     px = A[0];
1866                     py = A[1];
1867                 } else if (field_b && b_valid) {
1868                     px = B[0];
1869                     py = B[1];
1870                 } else if (c_valid) {
1871                     px = C[0];
1872                     py = C[1];
1873                 } else
1874                     px = py = 0;
1875             }
1876         } else if (total_valid == 1) {
1877             px = (a_valid) ? A[0] : ((b_valid) ? B[0] : C[0]);
1878             py = (a_valid) ? A[1] : ((b_valid) ? B[1] : C[1]);
1879         } else
1880             px = py = 0;
1881     }
1882
1883     /* store MV using signed modulus of MV range defined in 4.11 */
1884     s->mv[dir][n][0] = s->current_picture.motion_val[dir][xy][0] = ((px + dmv_x + r_x) & ((r_x << 1) - 1)) - r_x;
1885     s->mv[dir][n][1] = s->current_picture.motion_val[dir][xy][1] = ((py + dmv_y + r_y) & ((r_y << 1) - 1)) - r_y;
1886     if (mvn == 1) { /* duplicate motion data for 1-MV block */
1887         s->current_picture.motion_val[dir][xy +    1    ][0] = s->current_picture.motion_val[dir][xy][0];
1888         s->current_picture.motion_val[dir][xy +    1    ][1] = s->current_picture.motion_val[dir][xy][1];
1889         s->current_picture.motion_val[dir][xy + wrap    ][0] = s->current_picture.motion_val[dir][xy][0];
1890         s->current_picture.motion_val[dir][xy + wrap    ][1] = s->current_picture.motion_val[dir][xy][1];
1891         s->current_picture.motion_val[dir][xy + wrap + 1][0] = s->current_picture.motion_val[dir][xy][0];
1892         s->current_picture.motion_val[dir][xy + wrap + 1][1] = s->current_picture.motion_val[dir][xy][1];
1893     } else if (mvn == 2) { /* duplicate motion data for 2-Field MV block */
1894         s->current_picture.motion_val[dir][xy + 1][0] = s->current_picture.motion_val[dir][xy][0];
1895         s->current_picture.motion_val[dir][xy + 1][1] = s->current_picture.motion_val[dir][xy][1];
1896         s->mv[dir][n + 1][0] = s->mv[dir][n][0];
1897         s->mv[dir][n + 1][1] = s->mv[dir][n][1];
1898     }
1899 }
1900
1901 /** Motion compensation for direct or interpolated blocks in B-frames
1902  */
1903 static void vc1_interp_mc(VC1Context *v)
1904 {
1905     MpegEncContext *s = &v->s;
1906     H264ChromaContext *h264chroma = &v->h264chroma;
1907     uint8_t *srcY, *srcU, *srcV;
1908     int dxy, mx, my, uvmx, uvmy, src_x, src_y, uvsrc_x, uvsrc_y;
1909     int off, off_uv;
1910     int v_edge_pos = s->v_edge_pos >> v->field_mode;
1911     int use_ic = v->next_use_ic;
1912
1913     if (!v->field_mode && !v->s.next_picture.f->data[0])
1914         return;
1915
1916     mx   = s->mv[1][0][0];
1917     my   = s->mv[1][0][1];
1918     uvmx = (mx + ((mx & 3) == 3)) >> 1;
1919     uvmy = (my + ((my & 3) == 3)) >> 1;
1920     if (v->field_mode) {
1921         if (v->cur_field_type != v->ref_field_type[1])
1922             my   = my   - 2 + 4 * v->cur_field_type;
1923             uvmy = uvmy - 2 + 4 * v->cur_field_type;
1924     }
1925     if (v->fastuvmc) {
1926         uvmx = uvmx + ((uvmx < 0) ? -(uvmx & 1) : (uvmx & 1));
1927         uvmy = uvmy + ((uvmy < 0) ? -(uvmy & 1) : (uvmy & 1));
1928     }
1929     srcY = s->next_picture.f->data[0];
1930     srcU = s->next_picture.f->data[1];
1931     srcV = s->next_picture.f->data[2];
1932
1933     src_x   = s->mb_x * 16 + (mx   >> 2);
1934     src_y   = s->mb_y * 16 + (my   >> 2);
1935     uvsrc_x = s->mb_x *  8 + (uvmx >> 2);
1936     uvsrc_y = s->mb_y *  8 + (uvmy >> 2);
1937
1938     if (v->profile != PROFILE_ADVANCED) {
1939         src_x   = av_clip(  src_x, -16, s->mb_width  * 16);
1940         src_y   = av_clip(  src_y, -16, s->mb_height * 16);
1941         uvsrc_x = av_clip(uvsrc_x,  -8, s->mb_width  *  8);
1942         uvsrc_y = av_clip(uvsrc_y,  -8, s->mb_height *  8);
1943     } else {
1944         src_x   = av_clip(  src_x, -17, s->avctx->coded_width);
1945         src_y   = av_clip(  src_y, -18, s->avctx->coded_height + 1);
1946         uvsrc_x = av_clip(uvsrc_x,  -8, s->avctx->coded_width  >> 1);
1947         uvsrc_y = av_clip(uvsrc_y,  -8, s->avctx->coded_height >> 1);
1948     }
1949
1950     srcY += src_y   * s->linesize   + src_x;
1951     srcU += uvsrc_y * s->uvlinesize + uvsrc_x;
1952     srcV += uvsrc_y * s->uvlinesize + uvsrc_x;
1953
1954     if (v->field_mode && v->ref_field_type[1]) {
1955         srcY += s->current_picture_ptr->f->linesize[0];
1956         srcU += s->current_picture_ptr->f->linesize[1];
1957         srcV += s->current_picture_ptr->f->linesize[2];
1958     }
1959
1960     /* for grayscale we should not try to read from unknown area */
1961     if (s->flags & CODEC_FLAG_GRAY) {
1962         srcU = s->edge_emu_buffer + 18 * s->linesize;
1963         srcV = s->edge_emu_buffer + 18 * s->linesize;
1964     }
1965
1966     if (v->rangeredfrm || s->h_edge_pos < 22 || v_edge_pos < 22 || use_ic
1967         || (unsigned)(src_x - 1) > s->h_edge_pos - (mx & 3) - 16 - 3
1968         || (unsigned)(src_y - 1) > v_edge_pos    - (my & 3) - 16 - 3) {
1969         uint8_t *uvbuf = s->edge_emu_buffer + 19 * s->linesize;
1970
1971         srcY -= s->mspel * (1 + s->linesize);
1972         s->vdsp.emulated_edge_mc(s->edge_emu_buffer, srcY,
1973                                  s->linesize, s->linesize,
1974                                  17 + s->mspel * 2, 17 + s->mspel * 2,
1975                                  src_x - s->mspel, src_y - s->mspel,
1976                                  s->h_edge_pos, v_edge_pos);
1977         srcY = s->edge_emu_buffer;
1978         s->vdsp.emulated_edge_mc(uvbuf, srcU,
1979                                  s->uvlinesize, s->uvlinesize,
1980                                  8 + 1, 8 + 1,
1981                                  uvsrc_x, uvsrc_y, s->h_edge_pos >> 1, v_edge_pos >> 1);
1982         s->vdsp.emulated_edge_mc(uvbuf + 16, srcV,
1983                                  s->uvlinesize, s->uvlinesize,
1984                                  8 + 1, 8 + 1,
1985                                  uvsrc_x, uvsrc_y, s->h_edge_pos >> 1, v_edge_pos >> 1);
1986         srcU = uvbuf;
1987         srcV = uvbuf + 16;
1988         /* if we deal with range reduction we need to scale source blocks */
1989         if (v->rangeredfrm) {
1990             int i, j;
1991             uint8_t *src, *src2;
1992
1993             src = srcY;
1994             for (j = 0; j < 17 + s->mspel * 2; j++) {
1995                 for (i = 0; i < 17 + s->mspel * 2; i++)
1996                     src[i] = ((src[i] - 128) >> 1) + 128;
1997                 src += s->linesize;
1998             }
1999             src = srcU;
2000             src2 = srcV;
2001             for (j = 0; j < 9; j++) {
2002                 for (i = 0; i < 9; i++) {
2003                     src[i]  = ((src[i]  - 128) >> 1) + 128;
2004                     src2[i] = ((src2[i] - 128) >> 1) + 128;
2005                 }
2006                 src  += s->uvlinesize;
2007                 src2 += s->uvlinesize;
2008             }
2009         }
2010
2011         if (use_ic) {
2012             uint8_t (*luty )[256] = v->next_luty;
2013             uint8_t (*lutuv)[256] = v->next_lutuv;
2014             int i, j;
2015             uint8_t *src, *src2;
2016
2017             src = srcY;
2018             for (j = 0; j < 17 + s->mspel * 2; j++) {
2019                 int f = v->field_mode ? v->ref_field_type[1] : ((j+src_y - s->mspel) & 1);
2020                 for (i = 0; i < 17 + s->mspel * 2; i++)
2021                     src[i] = luty[f][src[i]];
2022                 src += s->linesize;
2023             }
2024             src  = srcU;
2025             src2 = srcV;
2026             for (j = 0; j < 9; j++) {
2027                 int f = v->field_mode ? v->ref_field_type[1] : ((j+uvsrc_y) & 1);
2028                 for (i = 0; i < 9; i++) {
2029                     src[i]  = lutuv[f][src[i]];
2030                     src2[i] = lutuv[f][src2[i]];
2031                 }
2032                 src  += s->uvlinesize;
2033                 src2 += s->uvlinesize;
2034             }
2035         }
2036         srcY += s->mspel * (1 + s->linesize);
2037     }
2038
2039     off    = 0;
2040     off_uv = 0;
2041
2042     if (s->mspel) {
2043         dxy = ((my & 3) << 2) | (mx & 3);
2044         v->vc1dsp.avg_vc1_mspel_pixels_tab[dxy](s->dest[0] + off    , srcY    , s->linesize, v->rnd);
2045         v->vc1dsp.avg_vc1_mspel_pixels_tab[dxy](s->dest[0] + off + 8, srcY + 8, s->linesize, v->rnd);
2046         srcY += s->linesize * 8;
2047         v->vc1dsp.avg_vc1_mspel_pixels_tab[dxy](s->dest[0] + off + 8 * s->linesize    , srcY    , s->linesize, v->rnd);
2048         v->vc1dsp.avg_vc1_mspel_pixels_tab[dxy](s->dest[0] + off + 8 * s->linesize + 8, srcY + 8, s->linesize, v->rnd);
2049     } else { // hpel mc
2050         dxy = (my & 2) | ((mx & 2) >> 1);
2051
2052         if (!v->rnd)
2053             s->hdsp.avg_pixels_tab[0][dxy](s->dest[0] + off, srcY, s->linesize, 16);
2054         else
2055             s->hdsp.avg_no_rnd_pixels_tab[dxy](s->dest[0] + off, srcY, s->linesize, 16);
2056     }
2057
2058     if (s->flags & CODEC_FLAG_GRAY) return;
2059     /* Chroma MC always uses qpel blilinear */
2060     uvmx = (uvmx & 3) << 1;
2061     uvmy = (uvmy & 3) << 1;
2062     if (!v->rnd) {
2063         h264chroma->avg_h264_chroma_pixels_tab[0](s->dest[1] + off_uv, srcU, s->uvlinesize, 8, uvmx, uvmy);
2064         h264chroma->avg_h264_chroma_pixels_tab[0](s->dest[2] + off_uv, srcV, s->uvlinesize, 8, uvmx, uvmy);
2065     } else {
2066         v->vc1dsp.avg_no_rnd_vc1_chroma_pixels_tab[0](s->dest[1] + off_uv, srcU, s->uvlinesize, 8, uvmx, uvmy);
2067         v->vc1dsp.avg_no_rnd_vc1_chroma_pixels_tab[0](s->dest[2] + off_uv, srcV, s->uvlinesize, 8, uvmx, uvmy);
2068     }
2069 }
2070
2071 static av_always_inline int scale_mv(int value, int bfrac, int inv, int qs)
2072 {
2073     int n = bfrac;
2074
2075 #if B_FRACTION_DEN==256
2076     if (inv)
2077         n -= 256;
2078     if (!qs)
2079         return 2 * ((value * n + 255) >> 9);
2080     return (value * n + 128) >> 8;
2081 #else
2082     if (inv)
2083         n -= B_FRACTION_DEN;
2084     if (!qs)
2085         return 2 * ((value * n + B_FRACTION_DEN - 1) / (2 * B_FRACTION_DEN));
2086     return (value * n + B_FRACTION_DEN/2) / B_FRACTION_DEN;
2087 #endif
2088 }
2089
2090 /** Reconstruct motion vector for B-frame and do motion compensation
2091  */
2092 static inline void vc1_b_mc(VC1Context *v, int dmv_x[2], int dmv_y[2],
2093                             int direct, int mode)
2094 {
2095     if (direct) {
2096         vc1_mc_1mv(v, 0);
2097         vc1_interp_mc(v);
2098         return;
2099     }
2100     if (mode == BMV_TYPE_INTERPOLATED) {
2101         vc1_mc_1mv(v, 0);
2102         vc1_interp_mc(v);
2103         return;
2104     }
2105
2106     vc1_mc_1mv(v, (mode == BMV_TYPE_BACKWARD));
2107 }
2108
2109 static inline void vc1_pred_b_mv(VC1Context *v, int dmv_x[2], int dmv_y[2],
2110                                  int direct, int mvtype)
2111 {
2112     MpegEncContext *s = &v->s;
2113     int xy, wrap, off = 0;
2114     int16_t *A, *B, *C;
2115     int px, py;
2116     int sum;
2117     int r_x, r_y;
2118     const uint8_t *is_intra = v->mb_type[0];
2119
2120     r_x = v->range_x;
2121     r_y = v->range_y;
2122     /* scale MV difference to be quad-pel */
2123     dmv_x[0] <<= 1 - s->quarter_sample;
2124     dmv_y[0] <<= 1 - s->quarter_sample;
2125     dmv_x[1] <<= 1 - s->quarter_sample;
2126     dmv_y[1] <<= 1 - s->quarter_sample;
2127
2128     wrap = s->b8_stride;
2129     xy = s->block_index[0];
2130
2131     if (s->mb_intra) {
2132         s->current_picture.motion_val[0][xy + v->blocks_off][0] =
2133         s->current_picture.motion_val[0][xy + v->blocks_off][1] =
2134         s->current_picture.motion_val[1][xy + v->blocks_off][0] =
2135         s->current_picture.motion_val[1][xy + v->blocks_off][1] = 0;
2136         return;
2137     }
2138     if (!v->field_mode) {
2139         s->mv[0][0][0] = scale_mv(s->next_picture.motion_val[1][xy][0], v->bfraction, 0, s->quarter_sample);
2140         s->mv[0][0][1] = scale_mv(s->next_picture.motion_val[1][xy][1], v->bfraction, 0, s->quarter_sample);
2141         s->mv[1][0][0] = scale_mv(s->next_picture.motion_val[1][xy][0], v->bfraction, 1, s->quarter_sample);
2142         s->mv[1][0][1] = scale_mv(s->next_picture.motion_val[1][xy][1], v->bfraction, 1, s->quarter_sample);
2143
2144         /* Pullback predicted motion vectors as specified in 8.4.5.4 */
2145         s->mv[0][0][0] = av_clip(s->mv[0][0][0], -60 - (s->mb_x << 6), (s->mb_width  << 6) - 4 - (s->mb_x << 6));
2146         s->mv[0][0][1] = av_clip(s->mv[0][0][1], -60 - (s->mb_y << 6), (s->mb_height << 6) - 4 - (s->mb_y << 6));
2147         s->mv[1][0][0] = av_clip(s->mv[1][0][0], -60 - (s->mb_x << 6), (s->mb_width  << 6) - 4 - (s->mb_x << 6));
2148         s->mv[1][0][1] = av_clip(s->mv[1][0][1], -60 - (s->mb_y << 6), (s->mb_height << 6) - 4 - (s->mb_y << 6));
2149     }
2150     if (direct) {
2151         s->current_picture.motion_val[0][xy + v->blocks_off][0] = s->mv[0][0][0];
2152         s->current_picture.motion_val[0][xy + v->blocks_off][1] = s->mv[0][0][1];
2153         s->current_picture.motion_val[1][xy + v->blocks_off][0] = s->mv[1][0][0];
2154         s->current_picture.motion_val[1][xy + v->blocks_off][1] = s->mv[1][0][1];
2155         return;
2156     }
2157
2158     if ((mvtype == BMV_TYPE_FORWARD) || (mvtype == BMV_TYPE_INTERPOLATED)) {
2159         C   = s->current_picture.motion_val[0][xy - 2];
2160         A   = s->current_picture.motion_val[0][xy - wrap * 2];
2161         off = (s->mb_x == (s->mb_width - 1)) ? -2 : 2;
2162         B   = s->current_picture.motion_val[0][xy - wrap * 2 + off];
2163
2164         if (!s->mb_x) C[0] = C[1] = 0;
2165         if (!s->first_slice_line) { // predictor A is not out of bounds
2166             if (s->mb_width == 1) {
2167                 px = A[0];
2168                 py = A[1];
2169             } else {
2170                 px = mid_pred(A[0], B[0], C[0]);
2171                 py = mid_pred(A[1], B[1], C[1]);
2172             }
2173         } else if (s->mb_x) { // predictor C is not out of bounds
2174             px = C[0];
2175             py = C[1];
2176         } else {
2177             px = py = 0;
2178         }
2179         /* Pullback MV as specified in 8.3.5.3.4 */
2180         {
2181             int qx, qy, X, Y;
2182             if (v->profile < PROFILE_ADVANCED) {
2183                 qx = (s->mb_x << 5);
2184                 qy = (s->mb_y << 5);
2185                 X  = (s->mb_width  << 5) - 4;
2186                 Y  = (s->mb_height << 5) - 4;
2187                 if (qx + px < -28) px = -28 - qx;
2188                 if (qy + py < -28) py = -28 - qy;
2189                 if (qx + px > X) px = X - qx;
2190                 if (qy + py > Y) py = Y - qy;
2191             } else {
2192                 qx = (s->mb_x << 6);
2193                 qy = (s->mb_y << 6);
2194                 X  = (s->mb_width  << 6) - 4;
2195                 Y  = (s->mb_height << 6) - 4;
2196                 if (qx + px < -60) px = -60 - qx;
2197                 if (qy + py < -60) py = -60 - qy;
2198                 if (qx + px > X) px = X - qx;
2199                 if (qy + py > Y) py = Y - qy;
2200             }
2201         }
2202         /* Calculate hybrid prediction as specified in 8.3.5.3.5 */
2203         if (0 && !s->first_slice_line && s->mb_x) {
2204             if (is_intra[xy - wrap])
2205                 sum = FFABS(px) + FFABS(py);
2206             else
2207                 sum = FFABS(px - A[0]) + FFABS(py - A[1]);
2208             if (sum > 32) {
2209                 if (get_bits1(&s->gb)) {
2210                     px = A[0];
2211                     py = A[1];
2212                 } else {
2213                     px = C[0];
2214                     py = C[1];
2215                 }
2216             } else {
2217                 if (is_intra[xy - 2])
2218                     sum = FFABS(px) + FFABS(py);
2219                 else
2220                     sum = FFABS(px - C[0]) + FFABS(py - C[1]);
2221                 if (sum > 32) {
2222                     if (get_bits1(&s->gb)) {
2223                         px = A[0];
2224                         py = A[1];
2225                     } else {
2226                         px = C[0];
2227                         py = C[1];
2228                     }
2229                 }
2230             }
2231         }
2232         /* store MV using signed modulus of MV range defined in 4.11 */
2233         s->mv[0][0][0] = ((px + dmv_x[0] + r_x) & ((r_x << 1) - 1)) - r_x;
2234         s->mv[0][0][1] = ((py + dmv_y[0] + r_y) & ((r_y << 1) - 1)) - r_y;
2235     }
2236     if ((mvtype == BMV_TYPE_BACKWARD) || (mvtype == BMV_TYPE_INTERPOLATED)) {
2237         C   = s->current_picture.motion_val[1][xy - 2];
2238         A   = s->current_picture.motion_val[1][xy - wrap * 2];
2239         off = (s->mb_x == (s->mb_width - 1)) ? -2 : 2;
2240         B   = s->current_picture.motion_val[1][xy - wrap * 2 + off];
2241
2242         if (!s->mb_x)
2243             C[0] = C[1] = 0;
2244         if (!s->first_slice_line) { // predictor A is not out of bounds
2245             if (s->mb_width == 1) {
2246                 px = A[0];
2247                 py = A[1];
2248             } else {
2249                 px = mid_pred(A[0], B[0], C[0]);
2250                 py = mid_pred(A[1], B[1], C[1]);
2251             }
2252         } else if (s->mb_x) { // predictor C is not out of bounds
2253             px = C[0];
2254             py = C[1];
2255         } else {
2256             px = py = 0;
2257         }
2258         /* Pullback MV as specified in 8.3.5.3.4 */
2259         {
2260             int qx, qy, X, Y;
2261             if (v->profile < PROFILE_ADVANCED) {
2262                 qx = (s->mb_x << 5);
2263                 qy = (s->mb_y << 5);
2264                 X  = (s->mb_width  << 5) - 4;
2265                 Y  = (s->mb_height << 5) - 4;
2266                 if (qx + px < -28) px = -28 - qx;
2267                 if (qy + py < -28) py = -28 - qy;
2268                 if (qx + px > X) px = X - qx;
2269                 if (qy + py > Y) py = Y - qy;
2270             } else {
2271                 qx = (s->mb_x << 6);
2272                 qy = (s->mb_y << 6);
2273                 X  = (s->mb_width  << 6) - 4;
2274                 Y  = (s->mb_height << 6) - 4;
2275                 if (qx + px < -60) px = -60 - qx;
2276                 if (qy + py < -60) py = -60 - qy;
2277                 if (qx + px > X) px = X - qx;
2278                 if (qy + py > Y) py = Y - qy;
2279             }
2280         }
2281         /* Calculate hybrid prediction as specified in 8.3.5.3.5 */
2282         if (0 && !s->first_slice_line && s->mb_x) {
2283             if (is_intra[xy - wrap])
2284                 sum = FFABS(px) + FFABS(py);
2285             else
2286                 sum = FFABS(px - A[0]) + FFABS(py - A[1]);
2287             if (sum > 32) {
2288                 if (get_bits1(&s->gb)) {
2289                     px = A[0];
2290                     py = A[1];
2291                 } else {
2292                     px = C[0];
2293                     py = C[1];
2294                 }
2295             } else {
2296                 if (is_intra[xy - 2])
2297                     sum = FFABS(px) + FFABS(py);
2298                 else
2299                     sum = FFABS(px - C[0]) + FFABS(py - C[1]);
2300                 if (sum > 32) {
2301                     if (get_bits1(&s->gb)) {
2302                         px = A[0];
2303                         py = A[1];
2304                     } else {
2305                         px = C[0];
2306                         py = C[1];
2307                     }
2308                 }
2309             }
2310         }
2311         /* store MV using signed modulus of MV range defined in 4.11 */
2312
2313         s->mv[1][0][0] = ((px + dmv_x[1] + r_x) & ((r_x << 1) - 1)) - r_x;
2314         s->mv[1][0][1] = ((py + dmv_y[1] + r_y) & ((r_y << 1) - 1)) - r_y;
2315     }
2316     s->current_picture.motion_val[0][xy][0] = s->mv[0][0][0];
2317     s->current_picture.motion_val[0][xy][1] = s->mv[0][0][1];
2318     s->current_picture.motion_val[1][xy][0] = s->mv[1][0][0];
2319     s->current_picture.motion_val[1][xy][1] = s->mv[1][0][1];
2320 }
2321
2322 static inline void vc1_pred_b_mv_intfi(VC1Context *v, int n, int *dmv_x, int *dmv_y, int mv1, int *pred_flag)
2323 {
2324     int dir = (v->bmvtype == BMV_TYPE_BACKWARD) ? 1 : 0;
2325     MpegEncContext *s = &v->s;
2326     int mb_pos = s->mb_x + s->mb_y * s->mb_stride;
2327
2328     if (v->bmvtype == BMV_TYPE_DIRECT) {
2329         int total_opp, k, f;
2330         if (s->next_picture.mb_type[mb_pos + v->mb_off] != MB_TYPE_INTRA) {
2331             s->mv[0][0][0] = scale_mv(s->next_picture.motion_val[1][s->block_index[0] + v->blocks_off][0],
2332                                       v->bfraction, 0, s->quarter_sample);
2333             s->mv[0][0][1] = scale_mv(s->next_picture.motion_val[1][s->block_index[0] + v->blocks_off][1],
2334                                       v->bfraction, 0, s->quarter_sample);
2335             s->mv[1][0][0] = scale_mv(s->next_picture.motion_val[1][s->block_index[0] + v->blocks_off][0],
2336                                       v->bfraction, 1, s->quarter_sample);
2337             s->mv[1][0][1] = scale_mv(s->next_picture.motion_val[1][s->block_index[0] + v->blocks_off][1],
2338                                       v->bfraction, 1, s->quarter_sample);
2339
2340             total_opp = v->mv_f_next[0][s->block_index[0] + v->blocks_off]
2341                       + v->mv_f_next[0][s->block_index[1] + v->blocks_off]
2342                       + v->mv_f_next[0][s->block_index[2] + v->blocks_off]
2343                       + v->mv_f_next[0][s->block_index[3] + v->blocks_off];
2344             f = (total_opp > 2) ? 1 : 0;
2345         } else {
2346             s->mv[0][0][0] = s->mv[0][0][1] = 0;
2347             s->mv[1][0][0] = s->mv[1][0][1] = 0;
2348             f = 0;
2349         }
2350         v->ref_field_type[0] = v->ref_field_type[1] = v->cur_field_type ^ f;
2351         for (k = 0; k < 4; k++) {
2352             s->current_picture.motion_val[0][s->block_index[k] + v->blocks_off][0] = s->mv[0][0][0];
2353             s->current_picture.motion_val[0][s->block_index[k] + v->blocks_off][1] = s->mv[0][0][1];
2354             s->current_picture.motion_val[1][s->block_index[k] + v->blocks_off][0] = s->mv[1][0][0];
2355             s->current_picture.motion_val[1][s->block_index[k] + v->blocks_off][1] = s->mv[1][0][1];
2356             v->mv_f[0][s->block_index[k] + v->blocks_off] = f;
2357             v->mv_f[1][s->block_index[k] + v->blocks_off] = f;
2358         }
2359         return;
2360     }
2361     if (v->bmvtype == BMV_TYPE_INTERPOLATED) {
2362         vc1_pred_mv(v, 0, dmv_x[0], dmv_y[0],   1, v->range_x, v->range_y, v->mb_type[0], pred_flag[0], 0);
2363         vc1_pred_mv(v, 0, dmv_x[1], dmv_y[1],   1, v->range_x, v->range_y, v->mb_type[0], pred_flag[1], 1);
2364         return;
2365     }
2366     if (dir) { // backward
2367         vc1_pred_mv(v, n, dmv_x[1], dmv_y[1], mv1, v->range_x, v->range_y, v->mb_type[0], pred_flag[1], 1);
2368         if (n == 3 || mv1) {
2369             vc1_pred_mv(v, 0, dmv_x[0], dmv_y[0],   1, v->range_x, v->range_y, v->mb_type[0], 0, 0);
2370         }
2371     } else { // forward
2372         vc1_pred_mv(v, n, dmv_x[0], dmv_y[0], mv1, v->range_x, v->range_y, v->mb_type[0], pred_flag[0], 0);
2373         if (n == 3 || mv1) {
2374             vc1_pred_mv(v, 0, dmv_x[1], dmv_y[1],   1, v->range_x, v->range_y, v->mb_type[0], 0, 1);
2375         }
2376     }
2377 }
2378
2379 /** Get predicted DC value for I-frames only
2380  * prediction dir: left=0, top=1
2381  * @param s MpegEncContext
2382  * @param overlap flag indicating that overlap filtering is used
2383  * @param pq integer part of picture quantizer
2384  * @param[in] n block index in the current MB
2385  * @param dc_val_ptr Pointer to DC predictor
2386  * @param dir_ptr Prediction direction for use in AC prediction
2387  */
2388 static inline int vc1_i_pred_dc(MpegEncContext *s, int overlap, int pq, int n,
2389                                 int16_t **dc_val_ptr, int *dir_ptr)
2390 {
2391     int a, b, c, wrap, pred, scale;
2392     int16_t *dc_val;
2393     static const uint16_t dcpred[32] = {
2394         -1, 1024,  512,  341,  256,  205,  171,  146,  128,
2395              114,  102,   93,   85,   79,   73,   68,   64,
2396               60,   57,   54,   51,   49,   47,   45,   43,
2397               41,   39,   38,   37,   35,   34,   33
2398     };
2399
2400     /* find prediction - wmv3_dc_scale always used here in fact */
2401     if (n < 4) scale = s->y_dc_scale;
2402     else       scale = s->c_dc_scale;
2403
2404     wrap   = s->block_wrap[n];
2405     dc_val = s->dc_val[0] + s->block_index[n];
2406
2407     /* B A
2408      * C X
2409      */
2410     c = dc_val[ - 1];
2411     b = dc_val[ - 1 - wrap];
2412     a = dc_val[ - wrap];
2413
2414     if (pq < 9 || !overlap) {
2415         /* Set outer values */
2416         if (s->first_slice_line && (n != 2 && n != 3))
2417             b = a = dcpred[scale];
2418         if (s->mb_x == 0 && (n != 1 && n != 3))
2419             b = c = dcpred[scale];
2420     } else {
2421         /* Set outer values */
2422         if (s->first_slice_line && (n != 2 && n != 3))
2423             b = a = 0;
2424         if (s->mb_x == 0 && (n != 1 && n != 3))
2425             b = c = 0;
2426     }
2427
2428     if (abs(a - b) <= abs(b - c)) {
2429         pred     = c;
2430         *dir_ptr = 1; // left
2431     } else {
2432         pred     = a;
2433         *dir_ptr = 0; // top
2434     }
2435
2436     /* update predictor */
2437     *dc_val_ptr = &dc_val[0];
2438     return pred;
2439 }
2440
2441
2442 /** Get predicted DC value
2443  * prediction dir: left=0, top=1
2444  * @param s MpegEncContext
2445  * @param overlap flag indicating that overlap filtering is used
2446  * @param pq integer part of picture quantizer
2447  * @param[in] n block index in the current MB
2448  * @param a_avail flag indicating top block availability
2449  * @param c_avail flag indicating left block availability
2450  * @param dc_val_ptr Pointer to DC predictor
2451  * @param dir_ptr Prediction direction for use in AC prediction
2452  */
2453 static inline int vc1_pred_dc(MpegEncContext *s, int overlap, int pq, int n,
2454                               int a_avail, int c_avail,
2455                               int16_t **dc_val_ptr, int *dir_ptr)
2456 {
2457     int a, b, c, wrap, pred;
2458     int16_t *dc_val;
2459     int mb_pos = s->mb_x + s->mb_y * s->mb_stride;
2460     int q1, q2 = 0;
2461     int dqscale_index;
2462
2463     wrap = s->block_wrap[n];
2464     dc_val = s->dc_val[0] + s->block_index[n];
2465
2466     /* B A
2467      * C X
2468      */
2469     c = dc_val[ - 1];
2470     b = dc_val[ - 1 - wrap];
2471     a = dc_val[ - wrap];
2472     /* scale predictors if needed */
2473     q1 = s->current_picture.qscale_table[mb_pos];
2474     dqscale_index = s->y_dc_scale_table[q1] - 1;
2475     if (dqscale_index < 0)
2476         return 0;
2477     if (c_avail && (n != 1 && n != 3)) {
2478         q2 = s->current_picture.qscale_table[mb_pos - 1];
2479         if (q2 && q2 != q1)
2480             c = (c * s->y_dc_scale_table[q2] * ff_vc1_dqscale[dqscale_index] + 0x20000) >> 18;
2481     }
2482     if (a_avail && (n != 2 && n != 3)) {
2483         q2 = s->current_picture.qscale_table[mb_pos - s->mb_stride];
2484         if (q2 && q2 != q1)
2485             a = (a * s->y_dc_scale_table[q2] * ff_vc1_dqscale[dqscale_index] + 0x20000) >> 18;
2486     }
2487     if (a_avail && c_avail && (n != 3)) {
2488         int off = mb_pos;
2489         if (n != 1)
2490             off--;
2491         if (n != 2)
2492             off -= s->mb_stride;
2493         q2 = s->current_picture.qscale_table[off];
2494         if (q2 && q2 != q1)
2495             b = (b * s->y_dc_scale_table[q2] * ff_vc1_dqscale[dqscale_index] + 0x20000) >> 18;
2496     }
2497
2498     if (a_avail && c_avail) {
2499         if (abs(a - b) <= abs(b - c)) {
2500             pred     = c;
2501             *dir_ptr = 1; // left
2502         } else {
2503             pred     = a;
2504             *dir_ptr = 0; // top
2505         }
2506     } else if (a_avail) {
2507         pred     = a;
2508         *dir_ptr = 0; // top
2509     } else if (c_avail) {
2510         pred     = c;
2511         *dir_ptr = 1; // left
2512     } else {
2513         pred     = 0;
2514         *dir_ptr = 1; // left
2515     }
2516
2517     /* update predictor */
2518     *dc_val_ptr = &dc_val[0];
2519     return pred;
2520 }
2521
2522 /** @} */ // Block group
2523
2524 /**
2525  * @name VC1 Macroblock-level functions in Simple/Main Profiles
2526  * @see 7.1.4, p91 and 8.1.1.7, p(1)04
2527  * @{
2528  */
2529
2530 static inline int vc1_coded_block_pred(MpegEncContext * s, int n,
2531                                        uint8_t **coded_block_ptr)
2532 {
2533     int xy, wrap, pred, a, b, c;
2534
2535     xy   = s->block_index[n];
2536     wrap = s->b8_stride;
2537
2538     /* B C
2539      * A X
2540      */
2541     a = s->coded_block[xy - 1       ];
2542     b = s->coded_block[xy - 1 - wrap];
2543     c = s->coded_block[xy     - wrap];
2544
2545     if (b == c) {
2546         pred = a;
2547     } else {
2548         pred = c;
2549     }
2550
2551     /* store value */
2552     *coded_block_ptr = &s->coded_block[xy];
2553
2554     return pred;
2555 }
2556
2557 /**
2558  * Decode one AC coefficient
2559  * @param v The VC1 context
2560  * @param last Last coefficient
2561  * @param skip How much zero coefficients to skip
2562  * @param value Decoded AC coefficient value
2563  * @param codingset set of VLC to decode data
2564  * @see 8.1.3.4
2565  */
2566 static void vc1_decode_ac_coeff(VC1Context *v, int *last, int *skip,
2567                                 int *value, int codingset)
2568 {
2569     GetBitContext *gb = &v->s.gb;
2570     int index, escape, run = 0, level = 0, lst = 0;
2571
2572     index = get_vlc2(gb, ff_vc1_ac_coeff_table[codingset].table, AC_VLC_BITS, 3);
2573     if (index != ff_vc1_ac_sizes[codingset] - 1) {
2574         run   = vc1_index_decode_table[codingset][index][0];
2575         level = vc1_index_decode_table[codingset][index][1];
2576         lst   = index >= vc1_last_decode_table[codingset] || get_bits_left(gb) < 0;
2577         if (get_bits1(gb))
2578             level = -level;
2579     } else {
2580         escape = decode210(gb);
2581         if (escape != 2) {
2582             index = get_vlc2(gb, ff_vc1_ac_coeff_table[codingset].table, AC_VLC_BITS, 3);
2583             run   = vc1_index_decode_table[codingset][index][0];
2584             level = vc1_index_decode_table[codingset][index][1];
2585             lst   = index >= vc1_last_decode_table[codingset];
2586             if (escape == 0) {
2587                 if (lst)
2588                     level += vc1_last_delta_level_table[codingset][run];
2589                 else
2590                     level += vc1_delta_level_table[codingset][run];
2591             } else {
2592                 if (lst)
2593                     run += vc1_last_delta_run_table[codingset][level] + 1;
2594                 else
2595                     run += vc1_delta_run_table[codingset][level] + 1;
2596             }
2597             if (get_bits1(gb))
2598                 level = -level;
2599         } else {
2600             int sign;
2601             lst = get_bits1(gb);
2602             if (v->s.esc3_level_length == 0) {
2603                 if (v->pq < 8 || v->dquantfrm) { // table 59
2604                     v->s.esc3_level_length = get_bits(gb, 3);
2605                     if (!v->s.esc3_level_length)
2606                         v->s.esc3_level_length = get_bits(gb, 2) + 8;
2607                 } else { // table 60
2608                     v->s.esc3_level_length = get_unary(gb, 1, 6) + 2;
2609                 }
2610                 v->s.esc3_run_length = 3 + get_bits(gb, 2);
2611             }
2612             run   = get_bits(gb, v->s.esc3_run_length);
2613             sign  = get_bits1(gb);
2614             level = get_bits(gb, v->s.esc3_level_length);
2615             if (sign)
2616                 level = -level;
2617         }
2618     }
2619
2620     *last  = lst;
2621     *skip  = run;
2622     *value = level;
2623 }
2624
2625 /** Decode intra block in intra frames - should be faster than decode_intra_block
2626  * @param v VC1Context
2627  * @param block block to decode
2628  * @param[in] n subblock index
2629  * @param coded are AC coeffs present or not
2630  * @param codingset set of VLC to decode data
2631  */
2632 static int vc1_decode_i_block(VC1Context *v, int16_t block[64], int n,
2633                               int coded, int codingset)
2634 {
2635     GetBitContext *gb = &v->s.gb;
2636     MpegEncContext *s = &v->s;
2637     int dc_pred_dir = 0; /* Direction of the DC prediction used */
2638     int i;
2639     int16_t *dc_val;
2640     int16_t *ac_val, *ac_val2;
2641     int dcdiff;
2642
2643     /* Get DC differential */
2644     if (n < 4) {
2645         dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_luma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
2646     } else {
2647         dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_chroma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
2648     }
2649     if (dcdiff < 0) {
2650         av_log(s->avctx, AV_LOG_ERROR, "Illegal DC VLC\n");
2651         return -1;
2652     }
2653     if (dcdiff) {
2654         if (dcdiff == 119 /* ESC index value */) {
2655             /* TODO: Optimize */
2656             if (v->pq == 1)      dcdiff = get_bits(gb, 10);
2657             else if (v->pq == 2) dcdiff = get_bits(gb, 9);
2658             else                 dcdiff = get_bits(gb, 8);
2659         } else {
2660             if (v->pq == 1)
2661                 dcdiff = (dcdiff << 2) + get_bits(gb, 2) - 3;
2662             else if (v->pq == 2)
2663                 dcdiff = (dcdiff << 1) + get_bits1(gb)   - 1;
2664         }
2665         if (get_bits1(gb))
2666             dcdiff = -dcdiff;
2667     }
2668
2669     /* Prediction */
2670     dcdiff += vc1_i_pred_dc(&v->s, v->overlap, v->pq, n, &dc_val, &dc_pred_dir);
2671     *dc_val = dcdiff;
2672
2673     /* Store the quantized DC coeff, used for prediction */
2674     if (n < 4) {
2675         block[0] = dcdiff * s->y_dc_scale;
2676     } else {
2677         block[0] = dcdiff * s->c_dc_scale;
2678     }
2679     /* Skip ? */
2680     if (!coded) {
2681         goto not_coded;
2682     }
2683
2684     // AC Decoding
2685     i = 1;
2686
2687     {
2688         int last = 0, skip, value;
2689         const uint8_t *zz_table;
2690         int scale;
2691         int k;
2692
2693         scale = v->pq * 2 + v->halfpq;
2694
2695         if (v->s.ac_pred) {
2696             if (!dc_pred_dir)
2697                 zz_table = v->zz_8x8[2];
2698             else
2699                 zz_table = v->zz_8x8[3];
2700         } else
2701             zz_table = v->zz_8x8[1];
2702
2703         ac_val  = s->ac_val[0][0] + s->block_index[n] * 16;
2704         ac_val2 = ac_val;
2705         if (dc_pred_dir) // left
2706             ac_val -= 16;
2707         else // top
2708             ac_val -= 16 * s->block_wrap[n];
2709
2710         while (!last) {
2711             vc1_decode_ac_coeff(v, &last, &skip, &value, codingset);
2712             i += skip;
2713             if (i > 63)
2714                 break;
2715             block[zz_table[i++]] = value;
2716         }
2717
2718         /* apply AC prediction if needed */
2719         if (s->ac_pred) {
2720             if (dc_pred_dir) { // left
2721                 for (k = 1; k < 8; k++)
2722                     block[k << v->left_blk_sh] += ac_val[k];
2723             } else { // top
2724                 for (k = 1; k < 8; k++)
2725                     block[k << v->top_blk_sh] += ac_val[k + 8];
2726             }
2727         }
2728         /* save AC coeffs for further prediction */
2729         for (k = 1; k < 8; k++) {
2730             ac_val2[k]     = block[k << v->left_blk_sh];
2731             ac_val2[k + 8] = block[k << v->top_blk_sh];
2732         }
2733
2734         /* scale AC coeffs */
2735         for (k = 1; k < 64; k++)
2736             if (block[k]) {
2737                 block[k] *= scale;
2738                 if (!v->pquantizer)
2739                     block[k] += (block[k] < 0) ? -v->pq : v->pq;
2740             }
2741
2742         if (s->ac_pred) i = 63;
2743     }
2744
2745 not_coded:
2746     if (!coded) {
2747         int k, scale;
2748         ac_val  = s->ac_val[0][0] + s->block_index[n] * 16;
2749         ac_val2 = ac_val;
2750
2751         i = 0;
2752         scale = v->pq * 2 + v->halfpq;
2753         memset(ac_val2, 0, 16 * 2);
2754         if (dc_pred_dir) { // left
2755             ac_val -= 16;
2756             if (s->ac_pred)
2757                 memcpy(ac_val2, ac_val, 8 * 2);
2758         } else { // top
2759             ac_val -= 16 * s->block_wrap[n];
2760             if (s->ac_pred)
2761                 memcpy(ac_val2 + 8, ac_val + 8, 8 * 2);
2762         }
2763
2764         /* apply AC prediction if needed */
2765         if (s->ac_pred) {
2766             if (dc_pred_dir) { //left
2767                 for (k = 1; k < 8; k++) {
2768                     block[k << v->left_blk_sh] = ac_val[k] * scale;
2769                     if (!v->pquantizer && block[k << v->left_blk_sh])
2770                         block[k << v->left_blk_sh] += (block[k << v->left_blk_sh] < 0) ? -v->pq : v->pq;
2771                 }
2772             } else { // top
2773                 for (k = 1; k < 8; k++) {
2774                     block[k << v->top_blk_sh] = ac_val[k + 8] * scale;
2775                     if (!v->pquantizer && block[k << v->top_blk_sh])
2776                         block[k << v->top_blk_sh] += (block[k << v->top_blk_sh] < 0) ? -v->pq : v->pq;
2777                 }
2778             }
2779             i = 63;
2780         }
2781     }
2782     s->block_last_index[n] = i;
2783
2784     return 0;
2785 }
2786
2787 /** Decode intra block in intra frames - should be faster than decode_intra_block
2788  * @param v VC1Context
2789  * @param block block to decode
2790  * @param[in] n subblock number
2791  * @param coded are AC coeffs present or not
2792  * @param codingset set of VLC to decode data
2793  * @param mquant quantizer value for this macroblock
2794  */
2795 static int vc1_decode_i_block_adv(VC1Context *v, int16_t block[64], int n,
2796                                   int coded, int codingset, int mquant)
2797 {
2798     GetBitContext *gb = &v->s.gb;
2799     MpegEncContext *s = &v->s;
2800     int dc_pred_dir = 0; /* Direction of the DC prediction used */
2801     int i;
2802     int16_t *dc_val;
2803     int16_t *ac_val, *ac_val2;
2804     int dcdiff;
2805     int a_avail = v->a_avail, c_avail = v->c_avail;
2806     int use_pred = s->ac_pred;
2807     int scale;
2808     int q1, q2 = 0;
2809     int mb_pos = s->mb_x + s->mb_y * s->mb_stride;
2810
2811     /* Get DC differential */
2812     if (n < 4) {
2813         dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_luma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
2814     } else {
2815         dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_chroma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
2816     }
2817     if (dcdiff < 0) {
2818         av_log(s->avctx, AV_LOG_ERROR, "Illegal DC VLC\n");
2819         return -1;
2820     }
2821     if (dcdiff) {
2822         if (dcdiff == 119 /* ESC index value */) {
2823             /* TODO: Optimize */
2824             if (mquant == 1)      dcdiff = get_bits(gb, 10);
2825             else if (mquant == 2) dcdiff = get_bits(gb, 9);
2826             else                  dcdiff = get_bits(gb, 8);
2827         } else {
2828             if (mquant == 1)
2829                 dcdiff = (dcdiff << 2) + get_bits(gb, 2) - 3;
2830             else if (mquant == 2)
2831                 dcdiff = (dcdiff << 1) + get_bits1(gb)   - 1;
2832         }
2833         if (get_bits1(gb))
2834             dcdiff = -dcdiff;
2835     }
2836
2837     /* Prediction */
2838     dcdiff += vc1_pred_dc(&v->s, v->overlap, mquant, n, v->a_avail, v->c_avail, &dc_val, &dc_pred_dir);
2839     *dc_val = dcdiff;
2840
2841     /* Store the quantized DC coeff, used for prediction */
2842     if (n < 4) {
2843         block[0] = dcdiff * s->y_dc_scale;
2844     } else {
2845         block[0] = dcdiff * s->c_dc_scale;
2846     }
2847
2848     //AC Decoding
2849     i = 1;
2850
2851     /* check if AC is needed at all */
2852     if (!a_avail && !c_avail)
2853         use_pred = 0;
2854     ac_val  = s->ac_val[0][0] + s->block_index[n] * 16;
2855     ac_val2 = ac_val;
2856
2857     scale = mquant * 2 + ((mquant == v->pq) ? v->halfpq : 0);
2858
2859     if (dc_pred_dir) // left
2860         ac_val -= 16;
2861     else // top
2862         ac_val -= 16 * s->block_wrap[n];
2863
2864     q1 = s->current_picture.qscale_table[mb_pos];
2865     if ( dc_pred_dir && c_avail && mb_pos)
2866         q2 = s->current_picture.qscale_table[mb_pos - 1];
2867     if (!dc_pred_dir && a_avail && mb_pos >= s->mb_stride)
2868         q2 = s->current_picture.qscale_table[mb_pos - s->mb_stride];
2869     if ( dc_pred_dir && n == 1)
2870         q2 = q1;
2871     if (!dc_pred_dir && n == 2)
2872         q2 = q1;
2873     if (n == 3)
2874         q2 = q1;
2875
2876     if (coded) {
2877         int last = 0, skip, value;
2878         const uint8_t *zz_table;
2879         int k;
2880
2881         if (v->s.ac_pred) {
2882             if (!use_pred && v->fcm == ILACE_FRAME) {
2883                 zz_table = v->zzi_8x8;
2884             } else {
2885                 if (!dc_pred_dir) // top
2886                     zz_table = v->zz_8x8[2];
2887                 else // left
2888                     zz_table = v->zz_8x8[3];
2889             }
2890         } else {
2891             if (v->fcm != ILACE_FRAME)
2892                 zz_table = v->zz_8x8[1];
2893             else
2894                 zz_table = v->zzi_8x8;
2895         }
2896
2897         while (!last) {
2898             vc1_decode_ac_coeff(v, &last, &skip, &value, codingset);
2899             i += skip;
2900             if (i > 63)
2901                 break;
2902             block[zz_table[i++]] = value;
2903         }
2904
2905         /* apply AC prediction if needed */
2906         if (use_pred) {
2907             /* scale predictors if needed*/
2908             if (q2 && q1 != q2) {
2909                 q1 = q1 * 2 + ((q1 == v->pq) ? v->halfpq : 0) - 1;
2910                 q2 = q2 * 2 + ((q2 == v->pq) ? v->halfpq : 0) - 1;
2911
2912                 if (q1 < 1)
2913                     return AVERROR_INVALIDDATA;
2914                 if (dc_pred_dir) { // left
2915                     for (k = 1; k < 8; k++)
2916                         block[k << v->left_blk_sh] += (ac_val[k] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
2917                 } else { // top
2918                     for (k = 1; k < 8; k++)
2919                         block[k << v->top_blk_sh] += (ac_val[k + 8] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
2920                 }
2921             } else {
2922                 if (dc_pred_dir) { //left
2923                     for (k = 1; k < 8; k++)
2924                         block[k << v->left_blk_sh] += ac_val[k];
2925                 } else { //top
2926                     for (k = 1; k < 8; k++)
2927                         block[k << v->top_blk_sh] += ac_val[k + 8];
2928                 }
2929             }
2930         }
2931         /* save AC coeffs for further prediction */
2932         for (k = 1; k < 8; k++) {
2933             ac_val2[k    ] = block[k << v->left_blk_sh];
2934             ac_val2[k + 8] = block[k << v->top_blk_sh];
2935         }
2936
2937         /* scale AC coeffs */
2938         for (k = 1; k < 64; k++)
2939             if (block[k]) {
2940                 block[k] *= scale;
2941                 if (!v->pquantizer)
2942                     block[k] += (block[k] < 0) ? -mquant : mquant;
2943             }
2944
2945         if (use_pred) i = 63;
2946     } else { // no AC coeffs
2947         int k;
2948
2949         memset(ac_val2, 0, 16 * 2);
2950         if (dc_pred_dir) { // left
2951             if (use_pred) {
2952                 memcpy(ac_val2, ac_val, 8 * 2);
2953                 if (q2 && q1 != q2) {
2954                     q1 = q1 * 2 + ((q1 == v->pq) ? v->halfpq : 0) - 1;
2955                     q2 = q2 * 2 + ((q2 == v->pq) ? v->halfpq : 0) - 1;
2956                     if (q1 < 1)
2957                         return AVERROR_INVALIDDATA;
2958                     for (k = 1; k < 8; k++)
2959                         ac_val2[k] = (ac_val2[k] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
2960                 }
2961             }
2962         } else { // top
2963             if (use_pred) {
2964                 memcpy(ac_val2 + 8, ac_val + 8, 8 * 2);
2965                 if (q2 && q1 != q2) {
2966                     q1 = q1 * 2 + ((q1 == v->pq) ? v->halfpq : 0) - 1;
2967                     q2 = q2 * 2 + ((q2 == v->pq) ? v->halfpq : 0) - 1;
2968                     if (q1 < 1)
2969                         return AVERROR_INVALIDDATA;
2970                     for (k = 1; k < 8; k++)
2971                         ac_val2[k + 8] = (ac_val2[k + 8] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
2972                 }
2973             }
2974         }
2975
2976         /* apply AC prediction if needed */
2977         if (use_pred) {
2978             if (dc_pred_dir) { // left
2979                 for (k = 1; k < 8; k++) {
2980                     block[k << v->left_blk_sh] = ac_val2[k] * scale;
2981                     if (!v->pquantizer && block[k << v->left_blk_sh])
2982                         block[k << v->left_blk_sh] += (block[k << v->left_blk_sh] < 0) ? -mquant : mquant;
2983                 }
2984             } else { // top
2985                 for (k = 1; k < 8; k++) {
2986                     block[k << v->top_blk_sh] = ac_val2[k + 8] * scale;
2987                     if (!v->pquantizer && block[k << v->top_blk_sh])
2988                         block[k << v->top_blk_sh] += (block[k << v->top_blk_sh] < 0) ? -mquant : mquant;
2989                 }
2990             }
2991             i = 63;
2992         }
2993     }
2994     s->block_last_index[n] = i;
2995
2996     return 0;
2997 }
2998
2999 /** Decode intra block in inter frames - more generic version than vc1_decode_i_block
3000  * @param v VC1Context
3001  * @param block block to decode
3002  * @param[in] n subblock index
3003  * @param coded are AC coeffs present or not
3004  * @param mquant block quantizer
3005  * @param codingset set of VLC to decode data
3006  */
3007 static int vc1_decode_intra_block(VC1Context *v, int16_t block[64], int n,
3008                                   int coded, int mquant, int codingset)
3009 {
3010     GetBitContext *gb = &v->s.gb;
3011     MpegEncContext *s = &v->s;
3012     int dc_pred_dir = 0; /* Direction of the DC prediction used */
3013     int i;
3014     int16_t *dc_val;
3015     int16_t *ac_val, *ac_val2;
3016     int dcdiff;
3017     int mb_pos = s->mb_x + s->mb_y * s->mb_stride;
3018     int a_avail = v->a_avail, c_avail = v->c_avail;
3019     int use_pred = s->ac_pred;
3020     int scale;
3021     int q1, q2 = 0;
3022
3023     s->bdsp.clear_block(block);
3024
3025     /* XXX: Guard against dumb values of mquant */
3026     mquant = (mquant < 1) ? 0 : ((mquant > 31) ? 31 : mquant);
3027
3028     /* Set DC scale - y and c use the same */
3029     s->y_dc_scale = s->y_dc_scale_table[mquant];
3030     s->c_dc_scale = s->c_dc_scale_table[mquant];
3031
3032     /* Get DC differential */
3033     if (n < 4) {
3034         dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_luma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
3035     } else {
3036         dcdiff = get_vlc2(&s->gb, ff_msmp4_dc_chroma_vlc[s->dc_table_index].table, DC_VLC_BITS, 3);
3037     }
3038     if (dcdiff < 0) {
3039         av_log(s->avctx, AV_LOG_ERROR, "Illegal DC VLC\n");
3040         return -1;
3041     }
3042     if (dcdiff) {
3043         if (dcdiff == 119 /* ESC index value */) {
3044             /* TODO: Optimize */
3045             if (mquant == 1)      dcdiff = get_bits(gb, 10);
3046             else if (mquant == 2) dcdiff = get_bits(gb, 9);
3047             else                  dcdiff = get_bits(gb, 8);
3048         } else {
3049             if (mquant == 1)
3050                 dcdiff = (dcdiff << 2) + get_bits(gb, 2) - 3;
3051             else if (mquant == 2)
3052                 dcdiff = (dcdiff << 1) + get_bits1(gb)   - 1;
3053         }
3054         if (get_bits1(gb))
3055             dcdiff = -dcdiff;
3056     }
3057
3058     /* Prediction */
3059     dcdiff += vc1_pred_dc(&v->s, v->overlap, mquant, n, a_avail, c_avail, &dc_val, &dc_pred_dir);
3060     *dc_val = dcdiff;
3061
3062     /* Store the quantized DC coeff, used for prediction */
3063
3064     if (n < 4) {
3065         block[0] = dcdiff * s->y_dc_scale;
3066     } else {
3067         block[0] = dcdiff * s->c_dc_scale;
3068     }
3069
3070     //AC Decoding
3071     i = 1;
3072
3073     /* check if AC is needed at all and adjust direction if needed */
3074     if (!a_avail) dc_pred_dir = 1;
3075     if (!c_avail) dc_pred_dir = 0;
3076     if (!a_avail && !c_avail) use_pred = 0;
3077     ac_val = s->ac_val[0][0] + s->block_index[n] * 16;
3078     ac_val2 = ac_val;
3079
3080     scale = mquant * 2 + v->halfpq;
3081
3082     if (dc_pred_dir) //left
3083         ac_val -= 16;
3084     else //top
3085         ac_val -= 16 * s->block_wrap[n];
3086
3087     q1 = s->current_picture.qscale_table[mb_pos];
3088     if (dc_pred_dir && c_avail && mb_pos)
3089         q2 = s->current_picture.qscale_table[mb_pos - 1];
3090     if (!dc_pred_dir && a_avail && mb_pos >= s->mb_stride)
3091         q2 = s->current_picture.qscale_table[mb_pos - s->mb_stride];
3092     if ( dc_pred_dir && n == 1)
3093         q2 = q1;
3094     if (!dc_pred_dir && n == 2)
3095         q2 = q1;
3096     if (n == 3) q2 = q1;
3097
3098     if (coded) {
3099         int last = 0, skip, value;
3100         int k;
3101
3102         while (!last) {
3103             vc1_decode_ac_coeff(v, &last, &skip, &value, codingset);
3104             i += skip;
3105             if (i > 63)
3106                 break;
3107             if (v->fcm == PROGRESSIVE)
3108                 block[v->zz_8x8[0][i++]] = value;
3109             else {
3110                 if (use_pred && (v->fcm == ILACE_FRAME)) {
3111                     if (!dc_pred_dir) // top
3112                         block[v->zz_8x8[2][i++]] = value;
3113                     else // left
3114                         block[v->zz_8x8[3][i++]] = value;
3115                 } else {
3116                     block[v->zzi_8x8[i++]] = value;
3117                 }
3118             }
3119         }
3120
3121         /* apply AC prediction if needed */
3122         if (use_pred) {
3123             /* scale predictors if needed*/
3124             if (q2 && q1 != q2) {
3125                 q1 = q1 * 2 + ((q1 == v->pq) ? v->halfpq : 0) - 1;
3126                 q2 = q2 * 2 + ((q2 == v->pq) ? v->halfpq : 0) - 1;
3127
3128                 if (q1 < 1)
3129                     return AVERROR_INVALIDDATA;
3130                 if (dc_pred_dir) { // left
3131                     for (k = 1; k < 8; k++)
3132                         block[k << v->left_blk_sh] += (ac_val[k] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
3133                 } else { //top
3134                     for (k = 1; k < 8; k++)
3135                         block[k << v->top_blk_sh] += (ac_val[k + 8] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
3136                 }
3137             } else {
3138                 if (dc_pred_dir) { // left
3139                     for (k = 1; k < 8; k++)
3140                         block[k << v->left_blk_sh] += ac_val[k];
3141                 } else { // top
3142                     for (k = 1; k < 8; k++)
3143                         block[k << v->top_blk_sh] += ac_val[k + 8];
3144                 }
3145             }
3146         }
3147         /* save AC coeffs for further prediction */
3148         for (k = 1; k < 8; k++) {
3149             ac_val2[k    ] = block[k << v->left_blk_sh];
3150             ac_val2[k + 8] = block[k << v->top_blk_sh];
3151         }
3152
3153         /* scale AC coeffs */
3154         for (k = 1; k < 64; k++)
3155             if (block[k]) {
3156                 block[k] *= scale;
3157                 if (!v->pquantizer)
3158                     block[k] += (block[k] < 0) ? -mquant : mquant;
3159             }
3160
3161         if (use_pred) i = 63;
3162     } else { // no AC coeffs
3163         int k;
3164
3165         memset(ac_val2, 0, 16 * 2);
3166         if (dc_pred_dir) { // left
3167             if (use_pred) {
3168                 memcpy(ac_val2, ac_val, 8 * 2);
3169                 if (q2 && q1 != q2) {
3170                     q1 = q1 * 2 + ((q1 == v->pq) ? v->halfpq : 0) - 1;
3171                     q2 = q2 * 2 + ((q2 == v->pq) ? v->halfpq : 0) - 1;
3172                     if (q1 < 1)
3173                         return AVERROR_INVALIDDATA;
3174                     for (k = 1; k < 8; k++)
3175                         ac_val2[k] = (ac_val2[k] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
3176                 }
3177             }
3178         } else { // top
3179             if (use_pred) {
3180                 memcpy(ac_val2 + 8, ac_val + 8, 8 * 2);
3181                 if (q2 && q1 != q2) {
3182                     q1 = q1 * 2 + ((q1 == v->pq) ? v->halfpq : 0) - 1;
3183                     q2 = q2 * 2 + ((q2 == v->pq) ? v->halfpq : 0) - 1;
3184                     if (q1 < 1)
3185                         return AVERROR_INVALIDDATA;
3186                     for (k = 1; k < 8; k++)
3187                         ac_val2[k + 8] = (ac_val2[k + 8] * q2 * ff_vc1_dqscale[q1 - 1] + 0x20000) >> 18;
3188                 }
3189             }
3190         }
3191
3192         /* apply AC prediction if needed */
3193         if (use_pred) {
3194             if (dc_pred_dir) { // left
3195                 for (k = 1; k < 8; k++) {
3196                     block[k << v->left_blk_sh] = ac_val2[k] * scale;
3197                     if (!v->pquantizer && block[k << v->left_blk_sh])
3198                         block[k << v->left_blk_sh] += (block[k << v->left_blk_sh] < 0) ? -mquant : mquant;
3199                 }
3200             } else { // top
3201                 for (k = 1; k < 8; k++) {
3202                     block[k << v->top_blk_sh] = ac_val2[k + 8] * scale;
3203                     if (!v->pquantizer && block[k << v->top_blk_sh])
3204                         block[k << v->top_blk_sh] += (block[k << v->top_blk_sh] < 0) ? -mquant : mquant;
3205                 }
3206             }
3207             i = 63;
3208         }
3209     }
3210     s->block_last_index[n] = i;
3211
3212     return 0;
3213 }
3214
3215 /** Decode P block
3216  */
3217 static int vc1_decode_p_block(VC1Context *v, int16_t block[64], int n,
3218                               int mquant, int ttmb, int first_block,
3219                               uint8_t *dst, int linesize, int skip_block,
3220                               int *ttmb_out)
3221 {
3222     MpegEncContext *s = &v->s;
3223     GetBitContext *gb = &s->gb;
3224     int i, j;
3225     int subblkpat = 0;
3226     int scale, off, idx, last, skip, value;
3227     int ttblk = ttmb & 7;
3228     int pat = 0;
3229
3230     s->bdsp.clear_block(block);
3231
3232     if (ttmb == -1) {
3233         ttblk = ff_vc1_ttblk_to_tt[v->tt_index][get_vlc2(gb, ff_vc1_ttblk_vlc[v->tt_index].table, VC1_TTBLK_VLC_BITS, 1)];
3234     }
3235     if (ttblk == TT_4X4) {
3236         subblkpat = ~(get_vlc2(gb, ff_vc1_subblkpat_vlc[v->tt_index].table, VC1_SUBBLKPAT_VLC_BITS, 1) + 1);
3237     }
3238     if ((ttblk != TT_8X8 && ttblk != TT_4X4)
3239         && ((v->ttmbf || (ttmb != -1 && (ttmb & 8) && !first_block))
3240             || (!v->res_rtm_flag && !first_block))) {
3241         subblkpat = decode012(gb);
3242         if (subblkpat)
3243             subblkpat ^= 3; // swap decoded pattern bits
3244         if (ttblk == TT_8X4_TOP || ttblk == TT_8X4_BOTTOM)
3245             ttblk = TT_8X4;
3246         if (ttblk == TT_4X8_RIGHT || ttblk == TT_4X8_LEFT)
3247             ttblk = TT_4X8;
3248     }
3249     scale = 2 * mquant + ((v->pq == mquant) ? v->halfpq : 0);
3250
3251     // convert transforms like 8X4_TOP to generic TT and SUBBLKPAT
3252     if (ttblk == TT_8X4_TOP || ttblk == TT_8X4_BOTTOM) {
3253         subblkpat = 2 - (ttblk == TT_8X4_TOP);
3254         ttblk     = TT_8X4;
3255     }
3256     if (ttblk == TT_4X8_RIGHT || ttblk == TT_4X8_LEFT) {
3257         subblkpat = 2 - (ttblk == TT_4X8_LEFT);
3258         ttblk     = TT_4X8;
3259     }
3260     switch (ttblk) {
3261     case TT_8X8:
3262         pat  = 0xF;
3263         i    = 0;
3264         last = 0;
3265         while (!last) {
3266             vc1_decode_ac_coeff(v, &last, &skip, &value, v->codingset2);
3267             i += skip;
3268             if (i > 63)
3269                 break;
3270             if (!v->fcm)
3271                 idx = v->zz_8x8[0][i++];
3272             else
3273                 idx = v->zzi_8x8[i++];
3274             block[idx] = value * scale;
3275             if (!v->pquantizer)
3276                 block[idx] += (block[idx] < 0) ? -mquant : mquant;
3277         }
3278         if (!skip_block) {
3279             if (i == 1)
3280                 v->vc1dsp.vc1_inv_trans_8x8_dc(dst, linesize, block);
3281             else {
3282                 v->vc1dsp.vc1_inv_trans_8x8(block);
3283                 s->dsp.add_pixels_clamped(block, dst, linesize);
3284             }
3285         }
3286         break;
3287     case TT_4X4:
3288         pat = ~subblkpat & 0xF;
3289         for (j = 0; j < 4; j++) {
3290             last = subblkpat & (1 << (3 - j));
3291             i    = 0;
3292             off  = (j & 1) * 4 + (j & 2) * 16;
3293             while (!last) {
3294                 vc1_decode_ac_coeff(v, &last, &skip, &value, v->codingset2);
3295                 i += skip;
3296                 if (i > 15)
3297                     break;
3298                 if (!v->fcm)
3299                     idx = ff_vc1_simple_progressive_4x4_zz[i++];
3300                 else
3301                     idx = ff_vc1_adv_interlaced_4x4_zz[i++];
3302                 block[idx + off] = value * scale;
3303                 if (!v->pquantizer)
3304                     block[idx + off] += (block[idx + off] < 0) ? -mquant : mquant;
3305             }
3306             if (!(subblkpat & (1 << (3 - j))) && !skip_block) {
3307                 if (i == 1)
3308                     v->vc1dsp.vc1_inv_trans_4x4_dc(dst + (j & 1) * 4 + (j & 2) * 2 * linesize, linesize, block + off);
3309                 else
3310                     v->vc1dsp.vc1_inv_trans_4x4(dst + (j & 1) * 4 + (j & 2) *  2 * linesize, linesize, block + off);
3311             }
3312         }
3313         break;
3314     case TT_8X4:
3315         pat = ~((subblkpat & 2) * 6 + (subblkpat & 1) * 3) & 0xF;
3316         for (j = 0; j < 2; j++) {
3317             last = subblkpat & (1 << (1 - j));
3318             i    = 0;
3319             off  = j * 32;
3320             while (!last) {
3321                 vc1_decode_ac_coeff(v, &last, &skip, &value, v->codingset2);
3322                 i += skip;
3323                 if (i > 31)
3324                     break;
3325                 if (!v->fcm)
3326                     idx = v->zz_8x4[i++] + off;
3327                 else
3328                     idx = ff_vc1_adv_interlaced_8x4_zz[i++] + off;
3329                 block[idx] = value * scale;
3330                 if (!v->pquantizer)
3331                     block[idx] += (block[idx] < 0) ? -mquant : mquant;
3332             }
3333             if (!(subblkpat & (1 << (1 - j))) && !skip_block) {
3334                 if (i == 1)
3335                     v->vc1dsp.vc1_inv_trans_8x4_dc(dst + j * 4 * linesize, linesize, block + off);
3336                 else
3337                     v->vc1dsp.vc1_inv_trans_8x4(dst + j * 4 * linesize, linesize, block + off);
3338             }
3339         }
3340         break;
3341     case TT_4X8:
3342         pat = ~(subblkpat * 5) & 0xF;
3343         for (j = 0; j < 2; j++) {
3344             last = subblkpat & (1 << (1 - j));
3345             i    = 0;
3346             off  = j * 4;
3347             while (!last) {
3348                 vc1_decode_ac_coeff(v, &last, &skip, &value, v->codingset2);
3349                 i += skip;
3350                 if (i > 31)
3351                     break;
3352                 if (!v->fcm)
3353                     idx = v->zz_4x8[i++] + off;
3354                 else
3355                     idx = ff_vc1_adv_interlaced_4x8_zz[i++] + off;
3356                 block[idx] = value * scale;
3357                 if (!v->pquantizer)
3358                     block[idx] += (block[idx] < 0) ? -mquant : mquant;
3359             }
3360             if (!(subblkpat & (1 << (1 - j))) && !skip_block) {
3361                 if (i == 1)
3362                     v->vc1dsp.vc1_inv_trans_4x8_dc(dst + j * 4, linesize, block + off);
3363                 else
3364                     v->vc1dsp.vc1_inv_trans_4x8(dst + j*4, linesize, block + off);
3365             }
3366         }
3367         break;
3368     }
3369     if (ttmb_out)
3370         *ttmb_out |= ttblk << (n * 4);
3371     return pat;
3372 }
3373
3374 /** @} */ // Macroblock group
3375
3376 static const int size_table  [6] = { 0, 2, 3, 4,  5,  8 };
3377 static const int offset_table[6] = { 0, 1, 3, 7, 15, 31 };
3378
3379 static av_always_inline void vc1_apply_p_v_loop_filter(VC1Context *v, int block_num)
3380 {
3381     MpegEncContext *s  = &v->s;
3382     int mb_cbp         = v->cbp[s->mb_x - s->mb_stride],
3383         block_cbp      = mb_cbp      >> (block_num * 4), bottom_cbp,
3384         mb_is_intra    = v->is_intra[s->mb_x - s->mb_stride],
3385         block_is_intra = mb_is_intra >> (block_num * 4), bottom_is_intra;
3386     int idx, linesize  = block_num > 3 ? s->uvlinesize : s->linesize, ttblk;
3387     uint8_t *dst;
3388
3389     if (block_num > 3) {
3390         dst      = s->dest[block_num - 3];
3391     } else {
3392         dst      = s->dest[0] + (block_num & 1) * 8 + ((block_num & 2) * 4 - 8) * linesize;
3393     }
3394     if (s->mb_y != s->end_mb_y || block_num < 2) {
3395         int16_t (*mv)[2];
3396         int mv_stride;
3397
3398         if (block_num > 3) {
3399             bottom_cbp      = v->cbp[s->mb_x]      >> (block_num * 4);
3400             bottom_is_intra = v->is_intra[s->mb_x] >> (block_num * 4);
3401             mv              = &v->luma_mv[s->mb_x - s->mb_stride];
3402             mv_stride       = s->mb_stride;
3403         } else {
3404             bottom_cbp      = (block_num < 2) ? (mb_cbp               >> ((block_num + 2) * 4))
3405                                               : (v->cbp[s->mb_x]      >> ((block_num - 2) * 4));
3406             bottom_is_intra = (block_num < 2) ? (mb_is_intra          >> ((block_num + 2) * 4))
3407                                               : (v->is_intra[s->mb_x] >> ((block_num - 2) * 4));
3408             mv_stride       = s->b8_stride;
3409             mv              = &s->current_picture.motion_val[0][s->block_index[block_num] - 2 * mv_stride];
3410         }
3411
3412         if (bottom_is_intra & 1 || block_is_intra & 1 ||
3413             mv[0][0] != mv[mv_stride][0] || mv[0][1] != mv[mv_stride][1]) {
3414             v->vc1dsp.vc1_v_loop_filter8(dst, linesize, v->pq);
3415         } else {
3416             idx = ((bottom_cbp >> 2) | block_cbp) & 3;
3417             if (idx == 3) {
3418                 v->vc1dsp.vc1_v_loop_filter8(dst, linesize, v->pq);
3419             } else if (idx) {
3420                 if (idx == 1)
3421                     v->vc1dsp.vc1_v_loop_filter4(dst + 4, linesize, v->pq);
3422                 else
3423                     v->vc1dsp.vc1_v_loop_filter4(dst,     linesize, v->pq);
3424             }
3425         }
3426     }
3427
3428     dst -= 4 * linesize;
3429     ttblk = (v->ttblk[s->mb_x - s->mb_stride] >> (block_num * 4)) & 0xF;
3430     if (ttblk == TT_4X4 || ttblk == TT_8X4) {
3431         idx = (block_cbp | (block_cbp >> 2)) & 3;
3432         if (idx == 3) {
3433             v->vc1dsp.vc1_v_loop_filter8(dst, linesize, v->pq);
3434         } else if (idx) {
3435             if (idx == 1)
3436                 v->vc1dsp.vc1_v_loop_filter4(dst + 4, linesize, v->pq);
3437             else
3438                 v->vc1dsp.vc1_v_loop_filter4(dst,     linesize, v->pq);
3439         }
3440     }
3441 }
3442
3443 static av_always_inline void vc1_apply_p_h_loop_filter(VC1Context *v, int block_num)
3444 {
3445     MpegEncContext *s  = &v->s;
3446     int mb_cbp         = v->cbp[s->mb_x - 1 - s->mb_stride],
3447         block_cbp      = mb_cbp      >> (block_num * 4), right_cbp,
3448         mb_is_intra    = v->is_intra[s->mb_x - 1 - s->mb_stride],
3449         block_is_intra = mb_is_intra >> (block_num * 4), right_is_intra;
3450     int idx, linesize  = block_num > 3 ? s->uvlinesize : s->linesize, ttblk;
3451     uint8_t *dst;
3452
3453     if (block_num > 3) {
3454         dst = s->dest[block_num - 3] - 8 * linesize;
3455     } else {
3456         dst = s->dest[0] + (block_num & 1) * 8 + ((block_num & 2) * 4 - 16) * linesize - 8;
3457     }
3458
3459     if (s->mb_x != s->mb_width || !(block_num & 5)) {
3460         int16_t (*mv)[2];
3461
3462         if (block_num > 3) {
3463             right_cbp      = v->cbp[s->mb_x - s->mb_stride] >> (block_num * 4);
3464             right_is_intra = v->is_intra[s->mb_x - s->mb_stride] >> (block_num * 4);
3465             mv             = &v->luma_mv[s->mb_x - s->mb_stride - 1];
3466         } else {
3467             right_cbp      = (block_num & 1) ? (v->cbp[s->mb_x - s->mb_stride]      >> ((block_num - 1) * 4))
3468                                              : (mb_cbp                              >> ((block_num + 1) * 4));
3469             right_is_intra = (block_num & 1) ? (v->is_intra[s->mb_x - s->mb_stride] >> ((block_num - 1) * 4))
3470                                              : (mb_is_intra                         >> ((block_num + 1) * 4));
3471             mv             = &s->current_picture.motion_val[0][s->block_index[block_num] - s->b8_stride * 2 - 2];
3472         }
3473         if (block_is_intra & 1 || right_is_intra & 1 || mv[0][0] != mv[1][0] || mv[0][1] != mv[1][1]) {
3474             v->vc1dsp.vc1_h_loop_filter8(dst, linesize, v->pq);
3475         } else {
3476             idx = ((right_cbp >> 1) | block_cbp) & 5; // FIXME check
3477             if (idx == 5) {
3478                 v->vc1dsp.vc1_h_loop_filter8(dst, linesize, v->pq);
3479             } else if (idx) {
3480                 if (idx == 1)
3481                     v->vc1dsp.vc1_h_loop_filter4(dst + 4 * linesize, linesize, v->pq);
3482                 else
3483                     v->vc1dsp.vc1_h_loop_filter4(dst,                linesize, v->pq);
3484             }
3485         }
3486     }
3487
3488     dst -= 4;
3489     ttblk = (v->ttblk[s->mb_x - s->mb_stride - 1] >> (block_num * 4)) & 0xf;
3490     if (ttblk == TT_4X4 || ttblk == TT_4X8) {
3491         idx = (block_cbp | (block_cbp >> 1)) & 5;
3492         if (idx == 5) {
3493             v->vc1dsp.vc1_h_loop_filter8(dst, linesize, v->pq);
3494         } else if (idx) {
3495             if (idx == 1)
3496                 v->vc1dsp.vc1_h_loop_filter4(dst + linesize * 4, linesize, v->pq);
3497             else
3498                 v->vc1dsp.vc1_h_loop_filter4(dst,                linesize, v->pq);
3499         }
3500     }
3501 }
3502
3503 static void vc1_apply_p_loop_filter(VC1Context *v)
3504 {
3505     MpegEncContext *s = &v->s;
3506     int i;
3507
3508     for (i = 0; i < 6; i++) {
3509         vc1_apply_p_v_loop_filter(v, i);
3510     }
3511
3512     /* V always precedes H, therefore we run H one MB before V;
3513      * at the end of a row, we catch up to complete the row */
3514     if (s->mb_x) {
3515         for (i = 0; i < 6; i++) {
3516             vc1_apply_p_h_loop_filter(v, i);
3517         }
3518         if (s->mb_x == s->mb_width - 1) {
3519             s->mb_x++;
3520             ff_update_block_index(s);
3521             for (i = 0; i < 6; i++) {
3522                 vc1_apply_p_h_loop_filter(v, i);
3523             }
3524         }
3525     }
3526 }
3527
3528 /** Decode one P-frame MB
3529  */
3530 static int vc1_decode_p_mb(VC1Context *v)
3531 {
3532     MpegEncContext *s = &v->s;
3533     GetBitContext *gb = &s->gb;
3534     int i, j;
3535     int mb_pos = s->mb_x + s->mb_y * s->mb_stride;
3536     int cbp; /* cbp decoding stuff */
3537     int mqdiff, mquant; /* MB quantization */
3538     int ttmb = v->ttfrm; /* MB Transform type */
3539
3540     int mb_has_coeffs = 1; /* last_flag */
3541     int dmv_x, dmv_y; /* Differential MV components */
3542     int index, index1; /* LUT indexes */
3543     int val, sign; /* temp values */
3544     int first_block = 1;
3545     int dst_idx, off;
3546     int skipped, fourmv;
3547     int block_cbp = 0, pat, block_tt = 0, block_intra = 0;
3548
3549     mquant = v->pq; /* lossy initialization */
3550
3551     if (v->mv_type_is_raw)
3552         fourmv = get_bits1(gb);
3553     else
3554         fourmv = v->mv_type_mb_plane[mb_pos];
3555     if (v->skip_is_raw)
3556         skipped = get_bits1(gb);
3557     else
3558         skipped = v->s.mbskip_table[mb_pos];
3559
3560     if (!fourmv) { /* 1MV mode */
3561         if (!skipped) {
3562             GET_MVDATA(dmv_x, dmv_y);
3563
3564             if (s->mb_intra) {
3565                 s->current_picture.motion_val[1][s->block_index[0]][0] = 0;
3566                 s->current_picture.motion_val[1][s->block_index[0]][1] = 0;
3567             }
3568             s->current_picture.mb_type[mb_pos] = s->mb_intra ? MB_TYPE_INTRA : MB_TYPE_16x16;
3569             vc1_pred_mv(v, 0, dmv_x, dmv_y, 1, v->range_x, v->range_y, v->mb_type[0], 0, 0);
3570
3571             /* FIXME Set DC val for inter block ? */
3572             if (s->mb_intra && !mb_has_coeffs) {
3573                 GET_MQUANT();
3574                 s->ac_pred = get_bits1(gb);
3575                 cbp        = 0;
3576             } else if (mb_has_coeffs) {
3577                 if (s->mb_intra)
3578                     s->ac_pred = get_bits1(gb);