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