mirror of https://github.com/FFmpeg/FFmpeg.git
Mans Rullgard
14 years ago
parent
7b4ee3a21d
commit
e7a972e113
17 changed files with 461 additions and 361 deletions
@ -0,0 +1,401 @@ |
||||
/*
|
||||
* Simple IDCT |
||||
* |
||||
* Copyright (c) 2001 Michael Niedermayer <michaelni@gmx.at> |
||||
* |
||||
* 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 |
||||
* simpleidct in C. |
||||
*/ |
||||
|
||||
/*
|
||||
based upon some outcommented c code from mpeg2dec (idct_mmx.c |
||||
written by Aaron Holtzman <aholtzma@ess.engr.uvic.ca>) |
||||
*/ |
||||
|
||||
#include "bit_depth_template.c" |
||||
|
||||
#undef W1 |
||||
#undef W2 |
||||
#undef W3 |
||||
#undef W4 |
||||
#undef W5 |
||||
#undef W6 |
||||
#undef W7 |
||||
#undef ROW_SHIFT |
||||
#undef COL_SHIFT |
||||
#undef DC_SHIFT |
||||
#undef MUL |
||||
#undef MAC |
||||
|
||||
#if BIT_DEPTH == 8 |
||||
|
||||
#define W1 22725 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
#define W2 21407 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
#define W3 19266 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
#define W4 16383 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
#define W5 12873 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
#define W6 8867 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
#define W7 4520 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
|
||||
|
||||
#define ROW_SHIFT 11 |
||||
#define COL_SHIFT 20 |
||||
#define DC_SHIFT 3 |
||||
|
||||
#define MUL(a, b) MUL16(a, b) |
||||
#define MAC(a, b, c) MAC16(a, b, c) |
||||
|
||||
#elif BIT_DEPTH == 10 |
||||
|
||||
#define W1 90901 |
||||
#define W2 85627 |
||||
#define W3 77062 |
||||
#define W4 65535 |
||||
#define W5 51491 |
||||
#define W6 35468 |
||||
#define W7 18081 |
||||
|
||||
#define ROW_SHIFT 15 |
||||
#define COL_SHIFT 20 |
||||
#define DC_SHIFT 1 |
||||
|
||||
#define MUL(a, b) ((a) * (b)) |
||||
#define MAC(a, b, c) ((a) += (b) * (c)) |
||||
|
||||
#else |
||||
|
||||
#error "Unsupported bitdepth" |
||||
|
||||
#endif |
||||
|
||||
static inline void FUNC(idctRowCondDC)(DCTELEM *row) |
||||
{ |
||||
int a0, a1, a2, a3, b0, b1, b2, b3; |
||||
|
||||
#if HAVE_FAST_64BIT |
||||
#define ROW0_MASK (0xffffLL << 48 * HAVE_BIGENDIAN) |
||||
if (((((uint64_t *)row)[0] & ~ROW0_MASK) | ((uint64_t *)row)[1]) == 0) { |
||||
uint64_t temp = (row[0] << DC_SHIFT) & 0xffff; |
||||
temp += temp << 16; |
||||
temp += temp << 32; |
||||
((uint64_t *)row)[0] = temp; |
||||
((uint64_t *)row)[1] = temp; |
||||
return; |
||||
} |
||||
#else |
||||
if (!(((uint32_t*)row)[1] | |
||||
((uint32_t*)row)[2] | |
||||
((uint32_t*)row)[3] | |
||||
row[1])) { |
||||
uint32_t temp = (row[0] << DC_SHIFT) & 0xffff; |
||||
temp += temp << 16; |
||||
((uint32_t*)row)[0]=((uint32_t*)row)[1] = |
||||
((uint32_t*)row)[2]=((uint32_t*)row)[3] = temp; |
||||
return; |
||||
} |
||||
#endif |
||||
|
||||
a0 = (W4 * row[0]) + (1 << (ROW_SHIFT - 1)); |
||||
a1 = a0; |
||||
a2 = a0; |
||||
a3 = a0; |
||||
|
||||
/* no need to optimize : gcc does it */ |
||||
a0 += W2 * row[2]; |
||||
a1 += W6 * row[2]; |
||||
a2 -= W6 * row[2]; |
||||
a3 -= W2 * row[2]; |
||||
|
||||
b0 = MUL(W1, row[1]); |
||||
MAC(b0, W3, row[3]); |
||||
b1 = MUL(W3, row[1]); |
||||
MAC(b1, -W7, row[3]); |
||||
b2 = MUL(W5, row[1]); |
||||
MAC(b2, -W1, row[3]); |
||||
b3 = MUL(W7, row[1]); |
||||
MAC(b3, -W5, row[3]); |
||||
|
||||
if (AV_RN64A(row + 4)) { |
||||
a0 += W4*row[4] + W6*row[6]; |
||||
a1 += - W4*row[4] - W2*row[6]; |
||||
a2 += - W4*row[4] + W2*row[6]; |
||||
a3 += W4*row[4] - W6*row[6]; |
||||
|
||||
MAC(b0, W5, row[5]); |
||||
MAC(b0, W7, row[7]); |
||||
|
||||
MAC(b1, -W1, row[5]); |
||||
MAC(b1, -W5, row[7]); |
||||
|
||||
MAC(b2, W7, row[5]); |
||||
MAC(b2, W3, row[7]); |
||||
|
||||
MAC(b3, W3, row[5]); |
||||
MAC(b3, -W1, row[7]); |
||||
} |
||||
|
||||
row[0] = (a0 + b0) >> ROW_SHIFT; |
||||
row[7] = (a0 - b0) >> ROW_SHIFT; |
||||
row[1] = (a1 + b1) >> ROW_SHIFT; |
||||
row[6] = (a1 - b1) >> ROW_SHIFT; |
||||
row[2] = (a2 + b2) >> ROW_SHIFT; |
||||
row[5] = (a2 - b2) >> ROW_SHIFT; |
||||
row[3] = (a3 + b3) >> ROW_SHIFT; |
||||
row[4] = (a3 - b3) >> ROW_SHIFT; |
||||
} |
||||
|
||||
static inline void FUNC(idctSparseColPut)(pixel *dest, int line_size, |
||||
DCTELEM *col) |
||||
{ |
||||
int a0, a1, a2, a3, b0, b1, b2, b3; |
||||
INIT_CLIP; |
||||
|
||||
/* XXX: I did that only to give same values as previous code */ |
||||
a0 = W4 * (col[8*0] + ((1<<(COL_SHIFT-1))/W4)); |
||||
a1 = a0; |
||||
a2 = a0; |
||||
a3 = a0; |
||||
|
||||
a0 += + W2*col[8*2]; |
||||
a1 += + W6*col[8*2]; |
||||
a2 += - W6*col[8*2]; |
||||
a3 += - W2*col[8*2]; |
||||
|
||||
b0 = MUL(W1, col[8*1]); |
||||
b1 = MUL(W3, col[8*1]); |
||||
b2 = MUL(W5, col[8*1]); |
||||
b3 = MUL(W7, col[8*1]); |
||||
|
||||
MAC(b0, + W3, col[8*3]); |
||||
MAC(b1, - W7, col[8*3]); |
||||
MAC(b2, - W1, col[8*3]); |
||||
MAC(b3, - W5, col[8*3]); |
||||
|
||||
if(col[8*4]){ |
||||
a0 += + W4*col[8*4]; |
||||
a1 += - W4*col[8*4]; |
||||
a2 += - W4*col[8*4]; |
||||
a3 += + W4*col[8*4]; |
||||
} |
||||
|
||||
if (col[8*5]) { |
||||
MAC(b0, + W5, col[8*5]); |
||||
MAC(b1, - W1, col[8*5]); |
||||
MAC(b2, + W7, col[8*5]); |
||||
MAC(b3, + W3, col[8*5]); |
||||
} |
||||
|
||||
if(col[8*6]){ |
||||
a0 += + W6*col[8*6]; |
||||
a1 += - W2*col[8*6]; |
||||
a2 += + W2*col[8*6]; |
||||
a3 += - W6*col[8*6]; |
||||
} |
||||
|
||||
if (col[8*7]) { |
||||
MAC(b0, + W7, col[8*7]); |
||||
MAC(b1, - W5, col[8*7]); |
||||
MAC(b2, + W3, col[8*7]); |
||||
MAC(b3, - W1, col[8*7]); |
||||
} |
||||
|
||||
dest[0] = CLIP((a0 + b0) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a1 + b1) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a2 + b2) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a3 + b3) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a3 - b3) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a2 - b2) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a1 - b1) >> COL_SHIFT); |
||||
dest += line_size; |
||||
dest[0] = CLIP((a0 - b0) >> COL_SHIFT); |
||||
} |
||||
|
||||
static inline void FUNC(idctSparseColAdd)(pixel *dest, int line_size, |
||||
DCTELEM *col) |
||||
{ |
||||
int a0, a1, a2, a3, b0, b1, b2, b3; |
||||
INIT_CLIP; |
||||
|
||||
/* XXX: I did that only to give same values as previous code */ |
||||
a0 = W4 * (col[8*0] + ((1<<(COL_SHIFT-1))/W4)); |
||||
a1 = a0; |
||||
a2 = a0; |
||||
a3 = a0; |
||||
|
||||
a0 += + W2*col[8*2]; |
||||
a1 += + W6*col[8*2]; |
||||
a2 += - W6*col[8*2]; |
||||
a3 += - W2*col[8*2]; |
||||
|
||||
b0 = MUL(W1, col[8*1]); |
||||
b1 = MUL(W3, col[8*1]); |
||||
b2 = MUL(W5, col[8*1]); |
||||
b3 = MUL(W7, col[8*1]); |
||||
|
||||
MAC(b0, + W3, col[8*3]); |
||||
MAC(b1, - W7, col[8*3]); |
||||
MAC(b2, - W1, col[8*3]); |
||||
MAC(b3, - W5, col[8*3]); |
||||
|
||||
if(col[8*4]){ |
||||
a0 += + W4*col[8*4]; |
||||
a1 += - W4*col[8*4]; |
||||
a2 += - W4*col[8*4]; |
||||
a3 += + W4*col[8*4]; |
||||
} |
||||
|
||||
if (col[8*5]) { |
||||
MAC(b0, + W5, col[8*5]); |
||||
MAC(b1, - W1, col[8*5]); |
||||
MAC(b2, + W7, col[8*5]); |
||||
MAC(b3, + W3, col[8*5]); |
||||
} |
||||
|
||||
if(col[8*6]){ |
||||
a0 += + W6*col[8*6]; |
||||
a1 += - W2*col[8*6]; |
||||
a2 += + W2*col[8*6]; |
||||
a3 += - W6*col[8*6]; |
||||
} |
||||
|
||||
if (col[8*7]) { |
||||
MAC(b0, + W7, col[8*7]); |
||||
MAC(b1, - W5, col[8*7]); |
||||
MAC(b2, + W3, col[8*7]); |
||||
MAC(b3, - W1, col[8*7]); |
||||
} |
||||
|
||||
dest[0] = CLIP(dest[0] + ((a0 + b0) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a1 + b1) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a2 + b2) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a3 + b3) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a3 - b3) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a2 - b2) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a1 - b1) >> COL_SHIFT)); |
||||
dest += line_size; |
||||
dest[0] = CLIP(dest[0] + ((a0 - b0) >> COL_SHIFT)); |
||||
} |
||||
|
||||
static inline void FUNC(idctSparseCol)(DCTELEM *col) |
||||
{ |
||||
int a0, a1, a2, a3, b0, b1, b2, b3; |
||||
|
||||
/* XXX: I did that only to give same values as previous code */ |
||||
a0 = W4 * (col[8*0] + ((1<<(COL_SHIFT-1))/W4)); |
||||
a1 = a0; |
||||
a2 = a0; |
||||
a3 = a0; |
||||
|
||||
a0 += + W2*col[8*2]; |
||||
a1 += + W6*col[8*2]; |
||||
a2 += - W6*col[8*2]; |
||||
a3 += - W2*col[8*2]; |
||||
|
||||
b0 = MUL(W1, col[8*1]); |
||||
b1 = MUL(W3, col[8*1]); |
||||
b2 = MUL(W5, col[8*1]); |
||||
b3 = MUL(W7, col[8*1]); |
||||
|
||||
MAC(b0, + W3, col[8*3]); |
||||
MAC(b1, - W7, col[8*3]); |
||||
MAC(b2, - W1, col[8*3]); |
||||
MAC(b3, - W5, col[8*3]); |
||||
|
||||
if(col[8*4]){ |
||||
a0 += + W4*col[8*4]; |
||||
a1 += - W4*col[8*4]; |
||||
a2 += - W4*col[8*4]; |
||||
a3 += + W4*col[8*4]; |
||||
} |
||||
|
||||
if (col[8*5]) { |
||||
MAC(b0, + W5, col[8*5]); |
||||
MAC(b1, - W1, col[8*5]); |
||||
MAC(b2, + W7, col[8*5]); |
||||
MAC(b3, + W3, col[8*5]); |
||||
} |
||||
|
||||
if(col[8*6]){ |
||||
a0 += + W6*col[8*6]; |
||||
a1 += - W2*col[8*6]; |
||||
a2 += + W2*col[8*6]; |
||||
a3 += - W6*col[8*6]; |
||||
} |
||||
|
||||
if (col[8*7]) { |
||||
MAC(b0, + W7, col[8*7]); |
||||
MAC(b1, - W5, col[8*7]); |
||||
MAC(b2, + W3, col[8*7]); |
||||
MAC(b3, - W1, col[8*7]); |
||||
} |
||||
|
||||
col[0 ] = ((a0 + b0) >> COL_SHIFT); |
||||
col[8 ] = ((a1 + b1) >> COL_SHIFT); |
||||
col[16] = ((a2 + b2) >> COL_SHIFT); |
||||
col[24] = ((a3 + b3) >> COL_SHIFT); |
||||
col[32] = ((a3 - b3) >> COL_SHIFT); |
||||
col[40] = ((a2 - b2) >> COL_SHIFT); |
||||
col[48] = ((a1 - b1) >> COL_SHIFT); |
||||
col[56] = ((a0 - b0) >> COL_SHIFT); |
||||
} |
||||
|
||||
void FUNC(ff_simple_idct_put)(uint8_t *dest_, int line_size, DCTELEM *block) |
||||
{ |
||||
pixel *dest = (pixel *)dest_; |
||||
int i; |
||||
for(i=0; i<8; i++) |
||||
FUNC(idctRowCondDC)(block + i*8); |
||||
|
||||
for(i=0; i<8; i++) |
||||
FUNC(idctSparseColPut)(dest + i, line_size, block + i); |
||||
} |
||||
|
||||
void FUNC(ff_simple_idct_add)(uint8_t *dest_, int line_size, DCTELEM *block) |
||||
{ |
||||
pixel *dest = (pixel *)dest_; |
||||
int i; |
||||
for(i=0; i<8; i++) |
||||
FUNC(idctRowCondDC)(block + i*8); |
||||
|
||||
for(i=0; i<8; i++) |
||||
FUNC(idctSparseColAdd)(dest + i, line_size, block + i); |
||||
} |
||||
|
||||
void FUNC(ff_simple_idct)(DCTELEM *block) |
||||
{ |
||||
int i; |
||||
for(i=0; i<8; i++) |
||||
FUNC(idctRowCondDC)(block + i*8); |
||||
|
||||
for(i=0; i<8; i++) |
||||
FUNC(idctSparseCol)(block + i); |
||||
} |
||||
Loading…
Reference in new issue