300 lines
		
	
	
		
			9.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			300 lines
		
	
	
		
			9.6 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * Copyright (C) 2004 the ffmpeg project
 | 
						|
 *
 | 
						|
 * This file is part of Libav.
 | 
						|
 *
 | 
						|
 * Libav 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.
 | 
						|
 *
 | 
						|
 * Libav 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 Libav; 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/intreadwrite.h"
 | 
						|
#include "libavutil/common.h"
 | 
						|
#include "libavutil/intreadwrite.h"
 | 
						|
 | 
						|
#include "avcodec.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]) +
 | 
						|
                       (first_pixel[0] - first_pixel[nstride]) * 3;
 | 
						|
        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]) +
 | 
						|
                       (first_pixel[ 0] - first_pixel[-1]) * 3;
 | 
						|
        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);
 | 
						|
}
 |