299 lines
9.4 KiB
C
299 lines
9.4 KiB
C
/*
|
|
* Copyright (C) 2004 the ffmpeg project
|
|
*
|
|
* 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
|
|
*/
|
|
|
|
/**
|
|
* @file
|
|
* Standard C DSP-oriented functions cribbed from the original VP3
|
|
* source code.
|
|
*/
|
|
|
|
#include "libavutil/attributes.h"
|
|
#include "libavutil/common.h"
|
|
#include "avcodec.h"
|
|
#include "dsputil.h"
|
|
#include "rnd_avg.h"
|
|
#include "vp3dsp.h"
|
|
|
|
#define IdctAdjustBeforeShift 8
|
|
#define xC1S7 64277
|
|
#define xC2S6 60547
|
|
#define xC3S5 54491
|
|
#define xC4S4 46341
|
|
#define xC5S3 36410
|
|
#define xC6S2 25080
|
|
#define xC7S1 12785
|
|
|
|
#define M(a,b) (((a) * (b))>>16)
|
|
|
|
static av_always_inline void idct(uint8_t *dst, int stride, int16_t *input, int type)
|
|
{
|
|
int16_t *ip = input;
|
|
|
|
int A, B, C, D, Ad, Bd, Cd, Dd, E, F, G, H;
|
|
int Ed, Gd, Add, Bdd, Fd, Hd;
|
|
|
|
int i;
|
|
|
|
/* Inverse DCT on the rows now */
|
|
for (i = 0; i < 8; i++) {
|
|
/* Check for non-zero values */
|
|
if ( ip[0 * 8] | ip[1 * 8] | ip[2 * 8] | ip[3 * 8] |
|
|
ip[4 * 8] | ip[5 * 8] | ip[6 * 8] | ip[7 * 8] ) {
|
|
A = M(xC1S7, ip[1 * 8]) + M(xC7S1, ip[7 * 8]);
|
|
B = M(xC7S1, ip[1 * 8]) - M(xC1S7, ip[7 * 8]);
|
|
C = M(xC3S5, ip[3 * 8]) + M(xC5S3, ip[5 * 8]);
|
|
D = M(xC3S5, ip[5 * 8]) - M(xC5S3, ip[3 * 8]);
|
|
|
|
Ad = M(xC4S4, (A - C));
|
|
Bd = M(xC4S4, (B - D));
|
|
|
|
Cd = A + C;
|
|
Dd = B + D;
|
|
|
|
E = M(xC4S4, (ip[0 * 8] + ip[4 * 8]));
|
|
F = M(xC4S4, (ip[0 * 8] - ip[4 * 8]));
|
|
|
|
G = M(xC2S6, ip[2 * 8]) + M(xC6S2, ip[6 * 8]);
|
|
H = M(xC6S2, ip[2 * 8]) - M(xC2S6, ip[6 * 8]);
|
|
|
|
Ed = E - G;
|
|
Gd = E + G;
|
|
|
|
Add = F + Ad;
|
|
Bdd = Bd - H;
|
|
|
|
Fd = F - Ad;
|
|
Hd = Bd + H;
|
|
|
|
/* Final sequence of operations over-write original inputs. */
|
|
ip[0 * 8] = Gd + Cd ;
|
|
ip[7 * 8] = Gd - Cd ;
|
|
|
|
ip[1 * 8] = Add + Hd;
|
|
ip[2 * 8] = Add - Hd;
|
|
|
|
ip[3 * 8] = Ed + Dd ;
|
|
ip[4 * 8] = Ed - Dd ;
|
|
|
|
ip[5 * 8] = Fd + Bdd;
|
|
ip[6 * 8] = Fd - Bdd;
|
|
}
|
|
|
|
ip += 1; /* next row */
|
|
}
|
|
|
|
ip = input;
|
|
|
|
for ( i = 0; i < 8; i++) {
|
|
/* Check for non-zero values (bitwise or faster than ||) */
|
|
if ( ip[1] | ip[2] | ip[3] |
|
|
ip[4] | ip[5] | ip[6] | ip[7] ) {
|
|
|
|
A = M(xC1S7, ip[1]) + M(xC7S1, ip[7]);
|
|
B = M(xC7S1, ip[1]) - M(xC1S7, ip[7]);
|
|
C = M(xC3S5, ip[3]) + M(xC5S3, ip[5]);
|
|
D = M(xC3S5, ip[5]) - M(xC5S3, ip[3]);
|
|
|
|
Ad = M(xC4S4, (A - C));
|
|
Bd = M(xC4S4, (B - D));
|
|
|
|
Cd = A + C;
|
|
Dd = B + D;
|
|
|
|
E = M(xC4S4, (ip[0] + ip[4])) + 8;
|
|
F = M(xC4S4, (ip[0] - ip[4])) + 8;
|
|
|
|
if(type==1){ //HACK
|
|
E += 16*128;
|
|
F += 16*128;
|
|
}
|
|
|
|
G = M(xC2S6, ip[2]) + M(xC6S2, ip[6]);
|
|
H = M(xC6S2, ip[2]) - M(xC2S6, ip[6]);
|
|
|
|
Ed = E - G;
|
|
Gd = E + G;
|
|
|
|
Add = F + Ad;
|
|
Bdd = Bd - H;
|
|
|
|
Fd = F - Ad;
|
|
Hd = Bd + H;
|
|
|
|
/* Final sequence of operations over-write original inputs. */
|
|
if (type == 1) {
|
|
dst[0*stride] = av_clip_uint8((Gd + Cd ) >> 4);
|
|
dst[7*stride] = av_clip_uint8((Gd - Cd ) >> 4);
|
|
|
|
dst[1*stride] = av_clip_uint8((Add + Hd ) >> 4);
|
|
dst[2*stride] = av_clip_uint8((Add - Hd ) >> 4);
|
|
|
|
dst[3*stride] = av_clip_uint8((Ed + Dd ) >> 4);
|
|
dst[4*stride] = av_clip_uint8((Ed - Dd ) >> 4);
|
|
|
|
dst[5*stride] = av_clip_uint8((Fd + Bdd ) >> 4);
|
|
dst[6*stride] = av_clip_uint8((Fd - Bdd ) >> 4);
|
|
}else{
|
|
dst[0*stride] = av_clip_uint8(dst[0*stride] + ((Gd + Cd ) >> 4));
|
|
dst[7*stride] = av_clip_uint8(dst[7*stride] + ((Gd - Cd ) >> 4));
|
|
|
|
dst[1*stride] = av_clip_uint8(dst[1*stride] + ((Add + Hd ) >> 4));
|
|
dst[2*stride] = av_clip_uint8(dst[2*stride] + ((Add - Hd ) >> 4));
|
|
|
|
dst[3*stride] = av_clip_uint8(dst[3*stride] + ((Ed + Dd ) >> 4));
|
|
dst[4*stride] = av_clip_uint8(dst[4*stride] + ((Ed - Dd ) >> 4));
|
|
|
|
dst[5*stride] = av_clip_uint8(dst[5*stride] + ((Fd + Bdd ) >> 4));
|
|
dst[6*stride] = av_clip_uint8(dst[6*stride] + ((Fd - Bdd ) >> 4));
|
|
}
|
|
|
|
} else {
|
|
if (type == 1) {
|
|
dst[0*stride]=
|
|
dst[1*stride]=
|
|
dst[2*stride]=
|
|
dst[3*stride]=
|
|
dst[4*stride]=
|
|
dst[5*stride]=
|
|
dst[6*stride]=
|
|
dst[7*stride]= av_clip_uint8(128 + ((xC4S4 * ip[0] + (IdctAdjustBeforeShift<<16))>>20));
|
|
}else{
|
|
if(ip[0]){
|
|
int v= ((xC4S4 * ip[0] + (IdctAdjustBeforeShift<<16))>>20);
|
|
dst[0*stride] = av_clip_uint8(dst[0*stride] + v);
|
|
dst[1*stride] = av_clip_uint8(dst[1*stride] + v);
|
|
dst[2*stride] = av_clip_uint8(dst[2*stride] + v);
|
|
dst[3*stride] = av_clip_uint8(dst[3*stride] + v);
|
|
dst[4*stride] = av_clip_uint8(dst[4*stride] + v);
|
|
dst[5*stride] = av_clip_uint8(dst[5*stride] + v);
|
|
dst[6*stride] = av_clip_uint8(dst[6*stride] + v);
|
|
dst[7*stride] = av_clip_uint8(dst[7*stride] + v);
|
|
}
|
|
}
|
|
}
|
|
|
|
ip += 8; /* next column */
|
|
dst++;
|
|
}
|
|
}
|
|
|
|
static void vp3_idct_put_c(uint8_t *dest/*align 8*/, int line_size,
|
|
int16_t *block/*align 16*/)
|
|
{
|
|
idct(dest, line_size, block, 1);
|
|
memset(block, 0, sizeof(*block) * 64);
|
|
}
|
|
|
|
static void vp3_idct_add_c(uint8_t *dest/*align 8*/, int line_size,
|
|
int16_t *block/*align 16*/)
|
|
{
|
|
idct(dest, line_size, block, 2);
|
|
memset(block, 0, sizeof(*block) * 64);
|
|
}
|
|
|
|
static void vp3_idct_dc_add_c(uint8_t *dest/*align 8*/, int line_size,
|
|
int16_t *block/*align 16*/)
|
|
{
|
|
int i, dc = (block[0] + 15) >> 5;
|
|
|
|
for(i = 0; i < 8; i++){
|
|
dest[0] = av_clip_uint8(dest[0] + dc);
|
|
dest[1] = av_clip_uint8(dest[1] + dc);
|
|
dest[2] = av_clip_uint8(dest[2] + dc);
|
|
dest[3] = av_clip_uint8(dest[3] + dc);
|
|
dest[4] = av_clip_uint8(dest[4] + dc);
|
|
dest[5] = av_clip_uint8(dest[5] + dc);
|
|
dest[6] = av_clip_uint8(dest[6] + dc);
|
|
dest[7] = av_clip_uint8(dest[7] + dc);
|
|
dest += line_size;
|
|
}
|
|
block[0] = 0;
|
|
}
|
|
|
|
static void vp3_v_loop_filter_c(uint8_t *first_pixel, int stride,
|
|
int *bounding_values)
|
|
{
|
|
unsigned char *end;
|
|
int filter_value;
|
|
const int nstride= -stride;
|
|
|
|
for (end= first_pixel + 8; first_pixel < end; first_pixel++) {
|
|
filter_value =
|
|
(first_pixel[2 * nstride] - first_pixel[ stride])
|
|
+3*(first_pixel[0 ] - first_pixel[nstride]);
|
|
filter_value = bounding_values[(filter_value + 4) >> 3];
|
|
first_pixel[nstride] = av_clip_uint8(first_pixel[nstride] + filter_value);
|
|
first_pixel[0] = av_clip_uint8(first_pixel[0] - filter_value);
|
|
}
|
|
}
|
|
|
|
static void vp3_h_loop_filter_c(uint8_t *first_pixel, int stride,
|
|
int *bounding_values)
|
|
{
|
|
unsigned char *end;
|
|
int filter_value;
|
|
|
|
for (end= first_pixel + 8*stride; first_pixel != end; first_pixel += stride) {
|
|
filter_value =
|
|
(first_pixel[-2] - first_pixel[ 1])
|
|
+3*(first_pixel[ 0] - first_pixel[-1]);
|
|
filter_value = bounding_values[(filter_value + 4) >> 3];
|
|
first_pixel[-1] = av_clip_uint8(first_pixel[-1] + filter_value);
|
|
first_pixel[ 0] = av_clip_uint8(first_pixel[ 0] - filter_value);
|
|
}
|
|
}
|
|
|
|
static void put_no_rnd_pixels_l2(uint8_t *dst, const uint8_t *src1,
|
|
const uint8_t *src2, ptrdiff_t stride, int h)
|
|
{
|
|
int i;
|
|
|
|
for (i = 0; i < h; i++) {
|
|
uint32_t a, b;
|
|
|
|
a = AV_RN32(&src1[i * stride]);
|
|
b = AV_RN32(&src2[i * stride]);
|
|
AV_WN32A(&dst[i * stride], no_rnd_avg32(a, b));
|
|
a = AV_RN32(&src1[i * stride + 4]);
|
|
b = AV_RN32(&src2[i * stride + 4]);
|
|
AV_WN32A(&dst[i * stride + 4], no_rnd_avg32(a, b));
|
|
}
|
|
}
|
|
|
|
av_cold void ff_vp3dsp_init(VP3DSPContext *c, int flags)
|
|
{
|
|
c->put_no_rnd_pixels_l2 = put_no_rnd_pixels_l2;
|
|
|
|
c->idct_put = vp3_idct_put_c;
|
|
c->idct_add = vp3_idct_add_c;
|
|
c->idct_dc_add = vp3_idct_dc_add_c;
|
|
c->v_loop_filter = vp3_v_loop_filter_c;
|
|
c->h_loop_filter = vp3_h_loop_filter_c;
|
|
|
|
if (ARCH_ARM)
|
|
ff_vp3dsp_init_arm(c, flags);
|
|
if (ARCH_BFIN)
|
|
ff_vp3dsp_init_bfin(c, flags);
|
|
if (ARCH_PPC)
|
|
ff_vp3dsp_init_ppc(c, flags);
|
|
if (ARCH_X86)
|
|
ff_vp3dsp_init_x86(c, flags);
|
|
}
|