avcodec/vp8dsp: zeroise input coefficient array after use in vp7 idct functions
[ffmpeg.git] / libavcodec / vp8dsp.c
1 /*
2  * Copyright (C) 2010 David Conrad
3  * Copyright (C) 2010 Ronald S. Bultje
4  * Copyright (C) 2014 Peter Ross
5  *
6  * This file is part of FFmpeg.
7  *
8  * FFmpeg is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * FFmpeg is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with FFmpeg; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22
23 /**
24  * @file
25  * VP8 compatible video decoder
26  */
27
28 #include "dsputil.h"
29 #include "vp8dsp.h"
30 #include "libavutil/common.h"
31
32 static void vp7_luma_dc_wht_c(int16_t block[4][4][16], int16_t dc[16])
33 {
34     int i, a1, b1, c1, d1;
35     int16_t tmp[16];
36
37     for (i = 0; i < 4; i++) {
38         a1 = (dc[i*4+0] + dc[i*4+2]) * 23170;
39         b1 = (dc[i*4+0] - dc[i*4+2]) * 23170;
40         c1 = dc[i*4+1] * 12540 - dc[i*4+3] * 30274;
41         d1 = dc[i*4+1] * 30274 + dc[i*4+3] * 12540;
42         tmp[i*4+0] = (a1 + d1) >> 14;
43         tmp[i*4+3] = (a1 - d1) >> 14;
44         tmp[i*4+1] = (b1 + c1) >> 14;
45         tmp[i*4+2] = (b1 - c1) >> 14;
46     }
47
48     for (i = 0; i < 4; i++) {
49         a1 = (tmp[i + 0] + tmp[i + 8]) * 23170;
50         b1 = (tmp[i + 0] - tmp[i + 8]) * 23170;
51         c1 = tmp[i + 4] * 12540 - tmp[i + 12] * 30274;
52         d1 = tmp[i + 4] * 30274 + tmp[i + 12] * 12540;
53         dc[i*4+0] = 0;
54         dc[i*4+1] = 0;
55         dc[i*4+2] = 0;
56         dc[i*4+3] = 0;
57         block[0][i][0] = (a1 + d1 + 0x20000) >> 18;
58         block[3][i][0] = (a1 - d1 + 0x20000) >> 18;
59         block[1][i][0] = (b1 + c1 + 0x20000) >> 18;
60         block[2][i][0] = (b1 - c1 + 0x20000) >> 18;
61     }
62 }
63
64 static void vp7_luma_dc_wht_dc_c(int16_t block[4][4][16], int16_t dc[16])
65 {
66     int i, val = (23170 * (23170 * dc[0] >> 14) + 0x20000) >> 18;
67     dc[0] = 0;
68
69     for (i = 0; i < 4; i++) {
70         block[i][0][0] = val;
71         block[i][1][0] = val;
72         block[i][2][0] = val;
73         block[i][3][0] = val;
74     }
75 }
76
77 static void vp7_idct_add_c(uint8_t *dst, int16_t block[16], ptrdiff_t stride)
78 {
79     int i, a1, b1, c1, d1;
80     int16_t tmp[16];
81
82     for (i = 0; i < 4; i++) {
83         a1 = (block[i*4+0] + block[i*4+2]) * 23170;
84         b1 = (block[i*4+0] - block[i*4+2]) * 23170;
85         c1 = block[i*4+1] * 12540 - block[i*4+3] * 30274;
86         d1 = block[i*4+1] * 30274 + block[i*4+3] * 12540;
87         block[i*4+0] = 0;
88         block[i*4+1] = 0;
89         block[i*4+2] = 0;
90         block[i*4+3] = 0;
91         tmp[i*4+0] = (a1 + d1) >> 14;
92         tmp[i*4+3] = (a1 - d1) >> 14;
93         tmp[i*4+1] = (b1 + c1) >> 14;
94         tmp[i*4+2] = (b1 - c1) >> 14;
95     }
96
97     for (i = 0; i < 4; i++) {
98         a1 = (tmp[i + 0] + tmp[i + 8]) * 23170;
99         b1 = (tmp[i + 0] - tmp[i + 8]) * 23170;
100         c1 = tmp[i + 4] * 12540 - tmp[i + 12] * 30274;
101         d1 = tmp[i + 4] * 30274 + tmp[i + 12] * 12540;
102         dst[0*stride+i] = av_clip_uint8(dst[0*stride+i] + ((a1 + d1 + 0x20000) >> 18));
103         dst[3*stride+i] = av_clip_uint8(dst[3*stride+i] + ((a1 - d1 + 0x20000) >> 18));
104         dst[1*stride+i] = av_clip_uint8(dst[1*stride+i] + ((b1 + c1 + 0x20000) >> 18));
105         dst[2*stride+i] = av_clip_uint8(dst[2*stride+i] + ((b1 - c1 + 0x20000) >> 18));
106     }
107 }
108
109 static void vp7_idct_dc_add_c(uint8_t *dst, int16_t block[16], ptrdiff_t stride)
110 {
111     int i, dc = (23170 * (23170 * block[0] >> 14) + 0x20000) >> 18;
112     block[0] = 0;
113
114     for (i = 0; i < 4; i++) {
115         dst[0] = av_clip_uint8(dst[0] + dc);
116         dst[1] = av_clip_uint8(dst[1] + dc);
117         dst[2] = av_clip_uint8(dst[2] + dc);
118         dst[3] = av_clip_uint8(dst[3] + dc);
119         dst += stride;
120     }
121 }
122
123 // TODO: Maybe add dequant
124 static void vp8_luma_dc_wht_c(int16_t block[4][4][16], int16_t dc[16])
125 {
126     int i, t0, t1, t2, t3;
127
128     for (i = 0; i < 4; i++) {
129         t0 = dc[0*4+i] + dc[3*4+i];
130         t1 = dc[1*4+i] + dc[2*4+i];
131         t2 = dc[1*4+i] - dc[2*4+i];
132         t3 = dc[0*4+i] - dc[3*4+i];
133
134         dc[0*4+i] = t0 + t1;
135         dc[1*4+i] = t3 + t2;
136         dc[2*4+i] = t0 - t1;
137         dc[3*4+i] = t3 - t2;
138     }
139
140     for (i = 0; i < 4; i++) {
141         t0 = dc[i*4+0] + dc[i*4+3] + 3; // rounding
142         t1 = dc[i*4+1] + dc[i*4+2];
143         t2 = dc[i*4+1] - dc[i*4+2];
144         t3 = dc[i*4+0] - dc[i*4+3] + 3; // rounding
145         dc[i*4+0] = 0;
146         dc[i*4+1] = 0;
147         dc[i*4+2] = 0;
148         dc[i*4+3] = 0;
149
150         block[i][0][0] = (t0 + t1) >> 3;
151         block[i][1][0] = (t3 + t2) >> 3;
152         block[i][2][0] = (t0 - t1) >> 3;
153         block[i][3][0] = (t3 - t2) >> 3;
154     }
155 }
156
157 static void vp8_luma_dc_wht_dc_c(int16_t block[4][4][16], int16_t dc[16])
158 {
159     int i, val = (dc[0] + 3) >> 3;
160     dc[0] = 0;
161
162     for (i = 0; i < 4; i++) {
163         block[i][0][0] = val;
164         block[i][1][0] = val;
165         block[i][2][0] = val;
166         block[i][3][0] = val;
167     }
168 }
169
170 #define MUL_20091(a) ((((a)*20091) >> 16) + (a))
171 #define MUL_35468(a)  (((a)*35468) >> 16)
172
173 static void vp8_idct_add_c(uint8_t *dst, int16_t block[16], ptrdiff_t stride)
174 {
175     int i, t0, t1, t2, t3;
176     int16_t tmp[16];
177
178     for (i = 0; i < 4; i++) {
179         t0 = block[0*4+i] + block[2*4+i];
180         t1 = block[0*4+i] - block[2*4+i];
181         t2 = MUL_35468(block[1*4+i]) - MUL_20091(block[3*4+i]);
182         t3 = MUL_20091(block[1*4+i]) + MUL_35468(block[3*4+i]);
183         block[0*4+i] = 0;
184         block[1*4+i] = 0;
185         block[2*4+i] = 0;
186         block[3*4+i] = 0;
187
188         tmp[i*4+0] = t0 + t3;
189         tmp[i*4+1] = t1 + t2;
190         tmp[i*4+2] = t1 - t2;
191         tmp[i*4+3] = t0 - t3;
192     }
193
194     for (i = 0; i < 4; i++) {
195         t0 = tmp[0*4+i] + tmp[2*4+i];
196         t1 = tmp[0*4+i] - tmp[2*4+i];
197         t2 = MUL_35468(tmp[1*4+i]) - MUL_20091(tmp[3*4+i]);
198         t3 = MUL_20091(tmp[1*4+i]) + MUL_35468(tmp[3*4+i]);
199
200         dst[0] = av_clip_uint8(dst[0] + ((t0 + t3 + 4) >> 3));
201         dst[1] = av_clip_uint8(dst[1] + ((t1 + t2 + 4) >> 3));
202         dst[2] = av_clip_uint8(dst[2] + ((t1 - t2 + 4) >> 3));
203         dst[3] = av_clip_uint8(dst[3] + ((t0 - t3 + 4) >> 3));
204         dst += stride;
205     }
206 }
207
208 static void vp8_idct_dc_add_c(uint8_t *dst, int16_t block[16], ptrdiff_t stride)
209 {
210     int i, dc = (block[0] + 4) >> 3;
211     block[0] = 0;
212
213     for (i = 0; i < 4; i++) {
214         dst[0] = av_clip_uint8(dst[0] + dc);
215         dst[1] = av_clip_uint8(dst[1] + dc);
216         dst[2] = av_clip_uint8(dst[2] + dc);
217         dst[3] = av_clip_uint8(dst[3] + dc);
218         dst += stride;
219     }
220 }
221
222 #define MK_IDCT_DC_ADD4_C(name) \
223 static void name ## _idct_dc_add4uv_c(uint8_t *dst, int16_t block[4][16], ptrdiff_t stride)\
224 {\
225     name ## _idct_dc_add_c(dst+stride*0+0, block[0], stride);\
226     name ## _idct_dc_add_c(dst+stride*0+4, block[1], stride);\
227     name ## _idct_dc_add_c(dst+stride*4+0, block[2], stride);\
228     name ## _idct_dc_add_c(dst+stride*4+4, block[3], stride);\
229 }\
230 \
231 static void name ## _idct_dc_add4y_c(uint8_t *dst, int16_t block[4][16], ptrdiff_t stride)\
232 {\
233     name ## _idct_dc_add_c(dst+ 0, block[0], stride);\
234     name ## _idct_dc_add_c(dst+ 4, block[1], stride);\
235     name ## _idct_dc_add_c(dst+ 8, block[2], stride);\
236     name ## _idct_dc_add_c(dst+12, block[3], stride);\
237 }
238
239 MK_IDCT_DC_ADD4_C(vp7)
240 MK_IDCT_DC_ADD4_C(vp8)
241
242 // because I like only having two parameters to pass functions...
243 #define LOAD_PIXELS\
244     int av_unused p3 = p[-4*stride];\
245     int av_unused p2 = p[-3*stride];\
246     int av_unused p1 = p[-2*stride];\
247     int av_unused p0 = p[-1*stride];\
248     int av_unused q0 = p[ 0*stride];\
249     int av_unused q1 = p[ 1*stride];\
250     int av_unused q2 = p[ 2*stride];\
251     int av_unused q3 = p[ 3*stride];
252
253 #define clip_int8(n) (cm[n+0x80]-0x80)
254
255 static av_always_inline void filter_common(uint8_t *p, ptrdiff_t stride, int is4tap, int vpn)
256 {
257     LOAD_PIXELS
258     int a, f1, f2;
259     const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
260
261     a = 3*(q0 - p0);
262
263     if (is4tap)
264         a += clip_int8(p1 - q1);
265
266     a = clip_int8(a);
267
268     // We deviate from the spec here with c(a+3) >> 3
269     // since that's what libvpx does.
270     f1 = FFMIN(a+4, 127) >> 3;
271
272     if (vpn == 7)
273         f2 = f1 - ((a & 7) == 4);
274     else
275         f2 = FFMIN(a+3, 127) >> 3;
276
277     // Despite what the spec says, we do need to clamp here to
278     // be bitexact with libvpx.
279     p[-1*stride] = cm[p0 + f2];
280     p[ 0*stride] = cm[q0 - f1];
281
282     // only used for _inner on blocks without high edge variance
283     if (!is4tap) {
284         a = (f1+1)>>1;
285         p[-2*stride] = cm[p1 + a];
286         p[ 1*stride] = cm[q1 - a];
287     }
288 }
289
290 static av_always_inline int vp7_simple_limit(uint8_t *p, ptrdiff_t stride, int flim)
291 {
292     LOAD_PIXELS
293     return FFABS(p0-q0) <= flim;
294 }
295
296 static av_always_inline int vp8_simple_limit(uint8_t *p, ptrdiff_t stride, int flim)
297 {
298     LOAD_PIXELS
299     return 2*FFABS(p0-q0) + (FFABS(p1-q1) >> 1) <= flim;
300 }
301
302 /**
303  * E - limit at the macroblock edge
304  * I - limit for interior difference
305  */
306 #define NORMAL_LIMIT(vpn) \
307 static av_always_inline int vp ## vpn ## _normal_limit(uint8_t *p, ptrdiff_t stride, int E, int I)\
308 {\
309     LOAD_PIXELS\
310     return vp ## vpn ## _simple_limit(p, stride, E)\
311         && FFABS(p3-p2) <= I && FFABS(p2-p1) <= I && FFABS(p1-p0) <= I\
312         && FFABS(q3-q2) <= I && FFABS(q2-q1) <= I && FFABS(q1-q0) <= I;\
313 }
314
315 NORMAL_LIMIT(7)
316 NORMAL_LIMIT(8)
317
318 // high edge variance
319 static av_always_inline int hev(uint8_t *p, ptrdiff_t stride, int thresh)
320 {
321     LOAD_PIXELS
322     return FFABS(p1-p0) > thresh || FFABS(q1-q0) > thresh;
323 }
324
325 static av_always_inline void filter_mbedge(uint8_t *p, ptrdiff_t stride)
326 {
327     int a0, a1, a2, w;
328     const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
329
330     LOAD_PIXELS
331
332     w = clip_int8(p1-q1);
333     w = clip_int8(w + 3*(q0-p0));
334
335     a0 = (27*w + 63) >> 7;
336     a1 = (18*w + 63) >> 7;
337     a2 = ( 9*w + 63) >> 7;
338
339     p[-3*stride] = cm[p2 + a2];
340     p[-2*stride] = cm[p1 + a1];
341     p[-1*stride] = cm[p0 + a0];
342     p[ 0*stride] = cm[q0 - a0];
343     p[ 1*stride] = cm[q1 - a1];
344     p[ 2*stride] = cm[q2 - a2];
345 }
346
347 #define LOOP_FILTER(vpn, dir, size, stridea, strideb, maybe_inline) \
348 static maybe_inline void vp ## vpn ## _ ## dir ## _loop_filter ## size ## _c(uint8_t *dst, ptrdiff_t stride,\
349                                      int flim_E, int flim_I, int hev_thresh)\
350 {\
351     int i;\
352 \
353     for (i = 0; i < size; i++)\
354         if (vp ## vpn ## _normal_limit(dst+i*stridea, strideb, flim_E, flim_I)) {\
355             if (hev(dst+i*stridea, strideb, hev_thresh))\
356                 filter_common(dst+i*stridea, strideb, 1, vpn);\
357             else\
358                 filter_mbedge(dst+i*stridea, strideb);\
359         }\
360 }\
361 \
362 static maybe_inline void vp ## vpn ## _ ## dir ## _loop_filter ## size ## _inner_c(uint8_t *dst, ptrdiff_t stride,\
363                                       int flim_E, int flim_I, int hev_thresh)\
364 {\
365     int i;\
366 \
367     for (i = 0; i < size; i++)\
368         if (vp ## vpn ## _normal_limit(dst+i*stridea, strideb, flim_E, flim_I)) {\
369             int hv = hev(dst+i*stridea, strideb, hev_thresh);\
370             if (hv) \
371                 filter_common(dst+i*stridea, strideb, 1, vpn);\
372             else \
373                 filter_common(dst+i*stridea, strideb, 0, vpn);\
374         }\
375 }
376
377 LOOP_FILTER(7, v, 16, 1, stride,)
378 LOOP_FILTER(7, h, 16, stride, 1,)
379
380 LOOP_FILTER(8, v, 16, 1, stride,)
381 LOOP_FILTER(8, h, 16, stride, 1,)
382
383 #define UV_LOOP_FILTER(vpn, dir, stridea, strideb) \
384 LOOP_FILTER(vpn, dir, 8, stridea, strideb, av_always_inline) \
385 static void vp ## vpn ## _ ## dir ## _loop_filter8uv_c(uint8_t *dstU, uint8_t *dstV, ptrdiff_t stride,\
386                                       int fE, int fI, int hev_thresh)\
387 {\
388   vp ## vpn ## _ ## dir ## _loop_filter8_c(dstU, stride, fE, fI, hev_thresh);\
389   vp ## vpn ## _ ## dir ## _loop_filter8_c(dstV, stride, fE, fI, hev_thresh);\
390 }\
391 static void vp ## vpn ## _ ## dir ## _loop_filter8uv_inner_c(uint8_t *dstU, uint8_t *dstV, ptrdiff_t stride,\
392                                       int fE, int fI, int hev_thresh)\
393 {\
394   vp ## vpn ## _ ## dir ## _loop_filter8_inner_c(dstU, stride, fE, fI, hev_thresh);\
395   vp ## vpn ## _ ## dir ## _loop_filter8_inner_c(dstV, stride, fE, fI, hev_thresh);\
396 }
397
398 UV_LOOP_FILTER(7, v, 1, stride)
399 UV_LOOP_FILTER(7, h, stride, 1)
400
401 UV_LOOP_FILTER(8, v, 1, stride)
402 UV_LOOP_FILTER(8, h, stride, 1)
403
404 #define LOOP_FILTER_SIMPLE(vpn) \
405 static void vp ## vpn ## _v_loop_filter_simple_c(uint8_t *dst, ptrdiff_t stride, int flim)\
406 {\
407     int i;\
408 \
409     for (i = 0; i < 16; i++)\
410         if (vp ## vpn ## _simple_limit(dst+i, stride, flim))\
411             filter_common(dst+i, stride, 1, vpn);\
412 }\
413 \
414 static void vp ## vpn ## _h_loop_filter_simple_c(uint8_t *dst, ptrdiff_t stride, int flim)\
415 {\
416     int i;\
417 \
418     for (i = 0; i < 16; i++)\
419         if (vp ## vpn ## _simple_limit(dst+i*stride, 1, flim))\
420             filter_common(dst+i*stride, 1, 1, vpn);\
421 }
422
423 LOOP_FILTER_SIMPLE(7)
424 LOOP_FILTER_SIMPLE(8)
425
426 static const uint8_t subpel_filters[7][6] = {
427     { 0,   6, 123,  12,   1,   0 },
428     { 2,  11, 108,  36,   8,   1 },
429     { 0,   9,  93,  50,   6,   0 },
430     { 3,  16,  77,  77,  16,   3 },
431     { 0,   6,  50,  93,   9,   0 },
432     { 1,   8,  36, 108,  11,   2 },
433     { 0,   1,  12, 123,   6,   0 },
434 };
435
436 #define PUT_PIXELS(WIDTH) \
437 static void put_vp8_pixels ## WIDTH ##_c(uint8_t *dst, ptrdiff_t dststride, uint8_t *src, ptrdiff_t srcstride, int h, int x, int y) { \
438     int i; \
439     for (i = 0; i < h; i++, dst+= dststride, src+= srcstride) { \
440         memcpy(dst, src, WIDTH); \
441     } \
442 }
443
444 PUT_PIXELS(16)
445 PUT_PIXELS(8)
446 PUT_PIXELS(4)
447
448 #define FILTER_6TAP(src, F, stride) \
449     cm[(F[2]*src[x+0*stride] - F[1]*src[x-1*stride] + F[0]*src[x-2*stride] + \
450         F[3]*src[x+1*stride] - F[4]*src[x+2*stride] + F[5]*src[x+3*stride] + 64) >> 7]
451
452 #define FILTER_4TAP(src, F, stride) \
453     cm[(F[2]*src[x+0*stride] - F[1]*src[x-1*stride] + \
454         F[3]*src[x+1*stride] - F[4]*src[x+2*stride] + 64) >> 7]
455
456 #define VP8_EPEL_H(SIZE, TAPS) \
457 static void put_vp8_epel ## SIZE ## _h ## TAPS ## _c(uint8_t *dst, ptrdiff_t dststride, uint8_t *src, ptrdiff_t srcstride, int h, int mx, int my) \
458 { \
459     const uint8_t *filter = subpel_filters[mx-1]; \
460     const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP; \
461     int x, y; \
462 \
463     for (y = 0; y < h; y++) { \
464         for (x = 0; x < SIZE; x++) \
465             dst[x] = FILTER_ ## TAPS ## TAP(src, filter, 1); \
466         dst += dststride; \
467         src += srcstride; \
468     } \
469 }
470 #define VP8_EPEL_V(SIZE, TAPS) \
471 static void put_vp8_epel ## SIZE ## _v ## TAPS ## _c(uint8_t *dst, ptrdiff_t dststride, uint8_t *src, ptrdiff_t srcstride, int h, int mx, int my) \
472 { \
473     const uint8_t *filter = subpel_filters[my-1]; \
474     const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP; \
475     int x, y; \
476 \
477     for (y = 0; y < h; y++) { \
478         for (x = 0; x < SIZE; x++) \
479             dst[x] = FILTER_ ## TAPS ## TAP(src, filter, srcstride); \
480         dst += dststride; \
481         src += srcstride; \
482     } \
483 }
484 #define VP8_EPEL_HV(SIZE, HTAPS, VTAPS) \
485 static void put_vp8_epel ## SIZE ## _h ## HTAPS ## v ## VTAPS ## _c(uint8_t *dst, ptrdiff_t dststride, uint8_t *src, ptrdiff_t srcstride, int h, int mx, int my) \
486 { \
487     const uint8_t *filter = subpel_filters[mx-1]; \
488     const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP; \
489     int x, y; \
490     uint8_t tmp_array[(2*SIZE+VTAPS-1)*SIZE]; \
491     uint8_t *tmp = tmp_array; \
492     src -= (2-(VTAPS==4))*srcstride; \
493 \
494     for (y = 0; y < h+VTAPS-1; y++) { \
495         for (x = 0; x < SIZE; x++) \
496             tmp[x] = FILTER_ ## HTAPS ## TAP(src, filter, 1); \
497         tmp += SIZE; \
498         src += srcstride; \
499     } \
500 \
501     tmp = tmp_array + (2-(VTAPS==4))*SIZE; \
502     filter = subpel_filters[my-1]; \
503 \
504     for (y = 0; y < h; y++) { \
505         for (x = 0; x < SIZE; x++) \
506             dst[x] = FILTER_ ## VTAPS ## TAP(tmp, filter, SIZE); \
507         dst += dststride; \
508         tmp += SIZE; \
509     } \
510 }
511
512 VP8_EPEL_H(16, 4)
513 VP8_EPEL_H(8,  4)
514 VP8_EPEL_H(4,  4)
515 VP8_EPEL_H(16, 6)
516 VP8_EPEL_H(8,  6)
517 VP8_EPEL_H(4,  6)
518 VP8_EPEL_V(16, 4)
519 VP8_EPEL_V(8,  4)
520 VP8_EPEL_V(4,  4)
521 VP8_EPEL_V(16, 6)
522 VP8_EPEL_V(8,  6)
523 VP8_EPEL_V(4,  6)
524 VP8_EPEL_HV(16, 4, 4)
525 VP8_EPEL_HV(8,  4, 4)
526 VP8_EPEL_HV(4,  4, 4)
527 VP8_EPEL_HV(16, 4, 6)
528 VP8_EPEL_HV(8,  4, 6)
529 VP8_EPEL_HV(4,  4, 6)
530 VP8_EPEL_HV(16, 6, 4)
531 VP8_EPEL_HV(8,  6, 4)
532 VP8_EPEL_HV(4,  6, 4)
533 VP8_EPEL_HV(16, 6, 6)
534 VP8_EPEL_HV(8,  6, 6)
535 VP8_EPEL_HV(4,  6, 6)
536
537 #define VP8_BILINEAR(SIZE) \
538 static void put_vp8_bilinear ## SIZE ## _h_c(uint8_t *dst, ptrdiff_t dstride, uint8_t *src, ptrdiff_t sstride, int h, int mx, int my) \
539 { \
540     int a = 8-mx, b = mx; \
541     int x, y; \
542 \
543     for (y = 0; y < h; y++) { \
544         for (x = 0; x < SIZE; x++) \
545             dst[x] = (a*src[x] + b*src[x+1] + 4) >> 3; \
546         dst += dstride; \
547         src += sstride; \
548     } \
549 } \
550 static void put_vp8_bilinear ## SIZE ## _v_c(uint8_t *dst, ptrdiff_t dstride, uint8_t *src, ptrdiff_t sstride, int h, int mx, int my) \
551 { \
552     int c = 8-my, d = my; \
553     int x, y; \
554 \
555     for (y = 0; y < h; y++) { \
556         for (x = 0; x < SIZE; x++) \
557             dst[x] = (c*src[x] + d*src[x+sstride] + 4) >> 3; \
558         dst += dstride; \
559         src += sstride; \
560     } \
561 } \
562 \
563 static void put_vp8_bilinear ## SIZE ## _hv_c(uint8_t *dst, ptrdiff_t dstride, uint8_t *src, ptrdiff_t sstride, int h, int mx, int my) \
564 { \
565     int a = 8-mx, b = mx; \
566     int c = 8-my, d = my; \
567     int x, y; \
568     uint8_t tmp_array[(2*SIZE+1)*SIZE]; \
569     uint8_t *tmp = tmp_array; \
570 \
571     for (y = 0; y < h+1; y++) { \
572         for (x = 0; x < SIZE; x++) \
573             tmp[x] = (a*src[x] + b*src[x+1] + 4) >> 3; \
574         tmp += SIZE; \
575         src += sstride; \
576     } \
577 \
578     tmp = tmp_array; \
579 \
580     for (y = 0; y < h; y++) { \
581         for (x = 0; x < SIZE; x++) \
582             dst[x] = (c*tmp[x] + d*tmp[x+SIZE] + 4) >> 3; \
583         dst += dstride; \
584         tmp += SIZE; \
585     } \
586 }
587
588 VP8_BILINEAR(16)
589 VP8_BILINEAR(8)
590 VP8_BILINEAR(4)
591
592 #define VP8_MC_FUNC(IDX, SIZE) \
593     dsp->put_vp8_epel_pixels_tab[IDX][0][0] = put_vp8_pixels ## SIZE ## _c; \
594     dsp->put_vp8_epel_pixels_tab[IDX][0][1] = put_vp8_epel ## SIZE ## _h4_c; \
595     dsp->put_vp8_epel_pixels_tab[IDX][0][2] = put_vp8_epel ## SIZE ## _h6_c; \
596     dsp->put_vp8_epel_pixels_tab[IDX][1][0] = put_vp8_epel ## SIZE ## _v4_c; \
597     dsp->put_vp8_epel_pixels_tab[IDX][1][1] = put_vp8_epel ## SIZE ## _h4v4_c; \
598     dsp->put_vp8_epel_pixels_tab[IDX][1][2] = put_vp8_epel ## SIZE ## _h6v4_c; \
599     dsp->put_vp8_epel_pixels_tab[IDX][2][0] = put_vp8_epel ## SIZE ## _v6_c; \
600     dsp->put_vp8_epel_pixels_tab[IDX][2][1] = put_vp8_epel ## SIZE ## _h4v6_c; \
601     dsp->put_vp8_epel_pixels_tab[IDX][2][2] = put_vp8_epel ## SIZE ## _h6v6_c
602
603 #define VP8_BILINEAR_MC_FUNC(IDX, SIZE) \
604     dsp->put_vp8_bilinear_pixels_tab[IDX][0][0] = put_vp8_pixels ## SIZE ## _c; \
605     dsp->put_vp8_bilinear_pixels_tab[IDX][0][1] = put_vp8_bilinear ## SIZE ## _h_c; \
606     dsp->put_vp8_bilinear_pixels_tab[IDX][0][2] = put_vp8_bilinear ## SIZE ## _h_c; \
607     dsp->put_vp8_bilinear_pixels_tab[IDX][1][0] = put_vp8_bilinear ## SIZE ## _v_c; \
608     dsp->put_vp8_bilinear_pixels_tab[IDX][1][1] = put_vp8_bilinear ## SIZE ## _hv_c; \
609     dsp->put_vp8_bilinear_pixels_tab[IDX][1][2] = put_vp8_bilinear ## SIZE ## _hv_c; \
610     dsp->put_vp8_bilinear_pixels_tab[IDX][2][0] = put_vp8_bilinear ## SIZE ## _v_c; \
611     dsp->put_vp8_bilinear_pixels_tab[IDX][2][1] = put_vp8_bilinear ## SIZE ## _hv_c; \
612     dsp->put_vp8_bilinear_pixels_tab[IDX][2][2] = put_vp8_bilinear ## SIZE ## _hv_c
613
614 av_cold void ff_vp8dsp_init(VP8DSPContext *dsp, int vp7)
615 {
616 #define VPX(f) vp7 ? vp7 ## _ ## f : vp8 ## _ ## f
617
618     dsp->vp8_luma_dc_wht    = VPX(luma_dc_wht_c);
619     dsp->vp8_luma_dc_wht_dc = VPX(luma_dc_wht_dc_c);
620     dsp->vp8_idct_add       = VPX(idct_add_c);
621     dsp->vp8_idct_dc_add    = VPX(idct_dc_add_c);
622     dsp->vp8_idct_dc_add4y  = VPX(idct_dc_add4y_c);
623     dsp->vp8_idct_dc_add4uv = VPX(idct_dc_add4uv_c);
624
625     dsp->vp8_v_loop_filter16y = VPX(v_loop_filter16_c);
626     dsp->vp8_h_loop_filter16y = VPX(h_loop_filter16_c);
627     dsp->vp8_v_loop_filter8uv = VPX(v_loop_filter8uv_c);
628     dsp->vp8_h_loop_filter8uv = VPX(h_loop_filter8uv_c);
629
630     dsp->vp8_v_loop_filter16y_inner = VPX(v_loop_filter16_inner_c);
631     dsp->vp8_h_loop_filter16y_inner = VPX(h_loop_filter16_inner_c);
632     dsp->vp8_v_loop_filter8uv_inner = VPX(v_loop_filter8uv_inner_c);
633     dsp->vp8_h_loop_filter8uv_inner = VPX(h_loop_filter8uv_inner_c);
634
635     dsp->vp8_v_loop_filter_simple = VPX(v_loop_filter_simple_c);
636     dsp->vp8_h_loop_filter_simple = VPX(h_loop_filter_simple_c);
637
638     VP8_MC_FUNC(0, 16);
639     VP8_MC_FUNC(1, 8);
640     VP8_MC_FUNC(2, 4);
641
642     VP8_BILINEAR_MC_FUNC(0, 16);
643     VP8_BILINEAR_MC_FUNC(1, 8);
644     VP8_BILINEAR_MC_FUNC(2, 4);
645
646     if (ARCH_ARM)
647         ff_vp8dsp_init_arm(dsp, vp7);
648     if (ARCH_PPC)
649         ff_vp8dsp_init_ppc(dsp);
650     if (ARCH_X86)
651         ff_vp8dsp_init_x86(dsp, vp7);
652 }