From 3b636f21daa6996e20ef97131a1c3649c3043304 Mon Sep 17 00:00:00 2001 From: David Conrad Date: Tue, 22 Jun 2010 19:24:09 +0000 Subject: Native VP8 decoder. Patch by David Conrad and myself. Originally committed as revision 23719 to svn://svn.ffmpeg.org/ffmpeg/trunk --- libavcodec/vp8dsp.c | 373 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 373 insertions(+) create mode 100644 libavcodec/vp8dsp.c (limited to 'libavcodec/vp8dsp.c') diff --git a/libavcodec/vp8dsp.c b/libavcodec/vp8dsp.c new file mode 100644 index 0000000000..50b029776c --- /dev/null +++ b/libavcodec/vp8dsp.c @@ -0,0 +1,373 @@ +/** + * VP8 compatible video decoder + * + * Copyright (C) 2010 David Conrad + * Copyright (C) 2010 Ronald S. Bultje + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "dsputil.h" +#include "vp8dsp.h" + +// TODO: Maybe add dequant +static void vp8_luma_dc_wht_c(DCTELEM block[4][4][16], DCTELEM dc[16]) +{ + int i, t0, t1, t2, t3; + + for (i = 0; i < 4; i++) { + t0 = dc[0*4+i] + dc[3*4+i]; + t1 = dc[1*4+i] + dc[2*4+i]; + t2 = dc[1*4+i] - dc[2*4+i]; + t3 = dc[0*4+i] - dc[3*4+i]; + + dc[0*4+i] = t0 + t1; + dc[1*4+i] = t3 + t2; + dc[2*4+i] = t0 - t1; + dc[3*4+i] = t3 - t2; + } + + for (i = 0; i < 4; i++) { + t0 = dc[i*4+0] + dc[i*4+3] + 3; // rounding + t1 = dc[i*4+1] + dc[i*4+2]; + t2 = dc[i*4+1] - dc[i*4+2]; + t3 = dc[i*4+0] - dc[i*4+3] + 3; // rounding + + *block[i][0] = (t0 + t1) >> 3; + *block[i][1] = (t3 + t2) >> 3; + *block[i][2] = (t0 - t1) >> 3; + *block[i][3] = (t3 - t2) >> 3; + } +} + + +#define MUL_20091(a) ((((a)*20091) >> 16) + (a)) +#define MUL_35468(a) (((a)*35468) >> 16) + +static void vp8_idct_add_c(uint8_t *dst, DCTELEM block[16], int stride) +{ + int i, t0, t1, t2, t3; + DCTELEM tmp[16]; + + for (i = 0; i < 4; i++) { + t0 = block[0*4+i] + block[2*4+i]; + t1 = block[0*4+i] - block[2*4+i]; + t2 = MUL_35468(block[1*4+i]) - MUL_20091(block[3*4+i]); + t3 = MUL_20091(block[1*4+i]) + MUL_35468(block[3*4+i]); + + tmp[i*4+0] = t0 + t3; + tmp[i*4+1] = t1 + t2; + tmp[i*4+2] = t1 - t2; + tmp[i*4+3] = t0 - t3; + } + + for (i = 0; i < 4; i++) { + t0 = tmp[0*4+i] + tmp[2*4+i]; + t1 = tmp[0*4+i] - tmp[2*4+i]; + t2 = MUL_35468(tmp[1*4+i]) - MUL_20091(tmp[3*4+i]); + t3 = MUL_20091(tmp[1*4+i]) + MUL_35468(tmp[3*4+i]); + + dst[0] = av_clip_uint8(dst[0] + ((t0 + t3 + 4) >> 3)); + dst[1] = av_clip_uint8(dst[1] + ((t1 + t2 + 4) >> 3)); + dst[2] = av_clip_uint8(dst[2] + ((t1 - t2 + 4) >> 3)); + dst[3] = av_clip_uint8(dst[3] + ((t0 - t3 + 4) >> 3)); + dst += stride; + } +} + +static void vp8_idct_dc_add_c(uint8_t *dst, DCTELEM block[16], int stride) +{ + int i, dc = (block[0] + 4) >> 3; + + for (i = 0; i < 4; i++) { + dst[0] = av_clip_uint8(dst[0] + dc); + dst[1] = av_clip_uint8(dst[1] + dc); + dst[2] = av_clip_uint8(dst[2] + dc); + dst[3] = av_clip_uint8(dst[3] + dc); + dst += stride; + } +} + + +// because I like only having two parameters to pass functions... +#define LOAD_PIXELS\ + int av_unused p3 = p[-4*stride];\ + int av_unused p2 = p[-3*stride];\ + int av_unused p1 = p[-2*stride];\ + int av_unused p0 = p[-1*stride];\ + int av_unused q0 = p[ 0*stride];\ + int av_unused q1 = p[ 1*stride];\ + int av_unused q2 = p[ 2*stride];\ + int av_unused q3 = p[ 3*stride]; + +static av_always_inline void filter_common(uint8_t *p, int stride, int is4tap) +{ + LOAD_PIXELS + int a, f1, f2; + + a = 3*(q0 - p0); + + if (is4tap) + a += av_clip_int8(p1 - q1); + + a = av_clip_int8(a); + + // We deviate from the spec here with c(a+3) >> 3 + // since that's what libvpx does. + f1 = FFMIN(a+4, 127) >> 3; + f2 = FFMIN(a+3, 127) >> 3; + + // Despite what the spec says, we do need to clamp here to + // be bitexact with libvpx. + p[-1*stride] = av_clip_uint8(p0 + f2); + p[ 0*stride] = av_clip_uint8(q0 - f1); + + // only used for _inner on blocks without high edge variance + if (!is4tap) { + a = (f1+1)>>1; + p[-2*stride] = av_clip_uint8(p1 + a); + p[ 1*stride] = av_clip_uint8(q1 - a); + } +} + +static av_always_inline int simple_limit(uint8_t *p, int stride, int flim) +{ + LOAD_PIXELS + return 2*FFABS(p0-q0) + (FFABS(p1-q1) >> 1) <= flim; +} + +/** + * E - limit at the macroblock edge + * I - limit for interior difference + */ +static av_always_inline int normal_limit(uint8_t *p, int stride, int E, int I) +{ + LOAD_PIXELS + return simple_limit(p, stride, 2*E+I) + && FFABS(p3-p2) <= I && FFABS(p2-p1) <= I && FFABS(p1-p0) <= I + && FFABS(q3-q2) <= I && FFABS(q2-q1) <= I && FFABS(q1-q0) <= I; +} + +// high edge variance +static av_always_inline int hev(uint8_t *p, int stride, int thresh) +{ + LOAD_PIXELS + return FFABS(p1-p0) > thresh || FFABS(q1-q0) > thresh; +} + +static av_always_inline void filter_mbedge(uint8_t *p, int stride) +{ + int a0, a1, a2, w; + + LOAD_PIXELS + + w = av_clip_int8(p1-q1); + w = av_clip_int8(w + 3*(q0-p0)); + + a0 = (27*w + 63) >> 7; + a1 = (18*w + 63) >> 7; + a2 = ( 9*w + 63) >> 7; + + p[-3*stride] = av_clip_uint8(p2 + a2); + p[-2*stride] = av_clip_uint8(p1 + a1); + p[-1*stride] = av_clip_uint8(p0 + a0); + p[ 0*stride] = av_clip_uint8(q0 - a0); + p[ 1*stride] = av_clip_uint8(q1 - a1); + p[ 2*stride] = av_clip_uint8(q2 - a2); +} + +#define LOOP_FILTER(dir, size, stridea, strideb) \ +static void vp8_ ## dir ## _loop_filter ## size ## _c(uint8_t *dst, int stride,\ + int flim_E, int flim_I, int hev_thresh)\ +{\ + int i;\ +\ + for (i = 0; i < size; i++)\ + if (normal_limit(dst+i*stridea, strideb, flim_E, flim_I)) {\ + if (hev(dst+i*stridea, strideb, hev_thresh))\ + filter_common(dst+i*stridea, strideb, 1);\ + else\ + filter_mbedge(dst+i*stridea, strideb);\ + }\ +}\ +\ +static void vp8_ ## dir ## _loop_filter ## size ## _inner_c(uint8_t *dst, int stride,\ + int flim_E, int flim_I, int hev_thresh)\ +{\ + int i, hv;\ +\ + for (i = 0; i < size; i++)\ + if (normal_limit(dst+i*stridea, strideb, flim_E, flim_I)) {\ + hv = hev(dst+i*stridea, strideb, hev_thresh);\ + filter_common(dst+i*stridea, strideb, hv);\ + }\ +} + +LOOP_FILTER(v, 16, 1, stride) +LOOP_FILTER(h, 16, stride, 1) +LOOP_FILTER(v, 8, 1, stride) +LOOP_FILTER(h, 8, stride, 1) + +static void vp8_v_loop_filter_simple_c(uint8_t *dst, int stride, int flim) +{ + int i; + + for (i = 0; i < 16; i++) + if (simple_limit(dst+i, stride, flim)) + filter_common(dst+i, stride, 1); +} + +static void vp8_h_loop_filter_simple_c(uint8_t *dst, int stride, int flim) +{ + int i; + + for (i = 0; i < 16; i++) + if (simple_limit(dst+i*stride, 1, flim)) + filter_common(dst+i*stride, 1, 1); +} + +static const uint8_t subpel_filters[7][6] = { + { 0, 6, 123, 12, 1, 0 }, + { 2, 11, 108, 36, 8, 1 }, + { 0, 9, 93, 50, 6, 0 }, + { 3, 16, 77, 77, 16, 3 }, + { 0, 6, 50, 93, 9, 0 }, + { 1, 8, 36, 108, 11, 2 }, + { 0, 1, 12, 123, 6, 0 }, +}; + + +#define FILTER_6TAP(src, F, stride) \ + av_clip_uint8((F[2]*src[x+0*stride] - F[1]*src[x-1*stride] + F[0]*src[x-2*stride] + \ + F[3]*src[x+1*stride] - F[4]*src[x+2*stride] + F[5]*src[x+3*stride] + 64) >> 7) + +#define FILTER_4TAP(src, F, stride) \ + av_clip_uint8((F[2]*src[x+0*stride] - F[1]*src[x-1*stride] + \ + F[3]*src[x+1*stride] - F[4]*src[x+2*stride] + 64) >> 7) + +#define VP8_EPEL_H(SIZE, FILTER, FILTERNAME) \ +static void put_vp8_epel ## SIZE ## _ ## FILTERNAME ## _c(uint8_t *dst, uint8_t *src, int stride, int h, int mx, int my) \ +{ \ + const uint8_t *filter = subpel_filters[mx-1]; \ + int x, y; \ +\ + for (y = 0; y < h; y++) { \ + for (x = 0; x < SIZE; x++) \ + dst[x] = FILTER(src, filter, 1); \ + dst += stride; \ + src += stride; \ + } \ +} +#define VP8_EPEL_V(SIZE, FILTER, FILTERNAME) \ +static void put_vp8_epel ## SIZE ## _ ## FILTERNAME ## _c(uint8_t *dst, uint8_t *src, int stride, int h, int mx, int my) \ +{ \ + const uint8_t *filter = subpel_filters[my-1]; \ + int x, y; \ +\ + for (y = 0; y < h; y++) { \ + for (x = 0; x < SIZE; x++) \ + dst[x] = FILTER(src, filter, stride); \ + dst += stride; \ + src += stride; \ + } \ +} +#define VP8_EPEL_HV(SIZE, FILTERX, FILTERY, FILTERNAME) \ +static void put_vp8_epel ## SIZE ## _ ## FILTERNAME ## _c(uint8_t *dst, uint8_t *src, int stride, int h, int mx, int my) \ +{ \ + const uint8_t *filter = subpel_filters[mx-1]; \ + int x, y; \ + uint8_t tmp_array[(2*SIZE+5)*SIZE]; \ + uint8_t *tmp = tmp_array; \ + src -= 2*stride; \ +\ + for (y = 0; y < h+5; y++) { \ + for (x = 0; x < SIZE; x++) \ + tmp[x] = FILTERX(src, filter, 1); \ + tmp += SIZE; \ + src += stride; \ + } \ +\ + tmp = tmp_array + 2*SIZE; \ + filter = subpel_filters[my-1]; \ +\ + for (y = 0; y < h; y++) { \ + for (x = 0; x < SIZE; x++) \ + dst[x] = FILTERY(tmp, filter, SIZE); \ + dst += stride; \ + tmp += SIZE; \ + } \ +} + +VP8_EPEL_H(16, FILTER_4TAP, h4) +VP8_EPEL_H(8, FILTER_4TAP, h4) +VP8_EPEL_H(4, FILTER_4TAP, h4) +VP8_EPEL_H(16, FILTER_6TAP, h6) +VP8_EPEL_H(8, FILTER_6TAP, h6) +VP8_EPEL_H(4, FILTER_6TAP, h6) +VP8_EPEL_V(16, FILTER_4TAP, v4) +VP8_EPEL_V(8, FILTER_4TAP, v4) +VP8_EPEL_V(4, FILTER_4TAP, v4) +VP8_EPEL_V(16, FILTER_6TAP, v6) +VP8_EPEL_V(8, FILTER_6TAP, v6) +VP8_EPEL_V(4, FILTER_6TAP, v6) +VP8_EPEL_HV(16, FILTER_4TAP, FILTER_4TAP, h4v4) +VP8_EPEL_HV(8, FILTER_4TAP, FILTER_4TAP, h4v4) +VP8_EPEL_HV(4, FILTER_4TAP, FILTER_4TAP, h4v4) +VP8_EPEL_HV(16, FILTER_4TAP, FILTER_6TAP, h4v6) +VP8_EPEL_HV(8, FILTER_4TAP, FILTER_6TAP, h4v6) +VP8_EPEL_HV(4, FILTER_4TAP, FILTER_6TAP, h4v6) +VP8_EPEL_HV(16, FILTER_6TAP, FILTER_4TAP, h6v4) +VP8_EPEL_HV(8, FILTER_6TAP, FILTER_4TAP, h6v4) +VP8_EPEL_HV(4, FILTER_6TAP, FILTER_4TAP, h6v4) +VP8_EPEL_HV(16, FILTER_6TAP, FILTER_6TAP, h6v6) +VP8_EPEL_HV(8, FILTER_6TAP, FILTER_6TAP, h6v6) +VP8_EPEL_HV(4, FILTER_6TAP, FILTER_6TAP, h6v6) + +#define VP8_MC_FUNC(IDX, SIZE) \ + dsp->put_vp8_epel_pixels_tab[IDX][0][0] = ff_put_vp8_pixels ## SIZE ## _c; \ + dsp->put_vp8_epel_pixels_tab[IDX][0][1] = put_vp8_epel ## SIZE ## _h4_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][0][2] = put_vp8_epel ## SIZE ## _h6_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][1][0] = put_vp8_epel ## SIZE ## _v4_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][1][1] = put_vp8_epel ## SIZE ## _h4v4_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][1][2] = put_vp8_epel ## SIZE ## _h6v4_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][2][0] = put_vp8_epel ## SIZE ## _v6_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][2][1] = put_vp8_epel ## SIZE ## _h4v6_c; \ + dsp->put_vp8_epel_pixels_tab[IDX][2][2] = put_vp8_epel ## SIZE ## _h6v6_c + +av_cold void ff_vp8dsp_init(VP8DSPContext *dsp) +{ + dsp->vp8_luma_dc_wht = vp8_luma_dc_wht_c; + dsp->vp8_idct_add = vp8_idct_add_c; + dsp->vp8_idct_dc_add = vp8_idct_dc_add_c; + + dsp->vp8_v_loop_filter16 = vp8_v_loop_filter16_c; + dsp->vp8_h_loop_filter16 = vp8_h_loop_filter16_c; + dsp->vp8_v_loop_filter8 = vp8_v_loop_filter8_c; + dsp->vp8_h_loop_filter8 = vp8_h_loop_filter8_c; + + dsp->vp8_v_loop_filter16_inner = vp8_v_loop_filter16_inner_c; + dsp->vp8_h_loop_filter16_inner = vp8_h_loop_filter16_inner_c; + dsp->vp8_v_loop_filter8_inner = vp8_v_loop_filter8_inner_c; + dsp->vp8_h_loop_filter8_inner = vp8_h_loop_filter8_inner_c; + + dsp->vp8_v_loop_filter_simple = vp8_v_loop_filter_simple_c; + dsp->vp8_h_loop_filter_simple = vp8_h_loop_filter_simple_c; + + VP8_MC_FUNC(0, 16); + VP8_MC_FUNC(1, 8); + VP8_MC_FUNC(2, 4); +} -- cgit v1.2.3