0,0 → 1,170 |
/* |
* Copyright (c) 2000, 2001 Fabrice Bellard |
* Copyright (c) 2002-2004 Michael Niedermayer <michaelni@gmx.at> |
* |
* 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 "libavutil/avassert.h" |
|
#include "bit_depth_template.c" |
|
#define H264_CHROMA_MC(OPNAME, OP)\ |
static void FUNCC(OPNAME ## h264_chroma_mc1)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\ |
pixel *dst = (pixel*)_dst;\ |
pixel *src = (pixel*)_src;\ |
const int A=(8-x)*(8-y);\ |
const int B=( x)*(8-y);\ |
const int C=(8-x)*( y);\ |
const int D=( x)*( y);\ |
int i;\ |
stride >>= sizeof(pixel)-1;\ |
\ |
av_assert2(x<8 && y<8 && x>=0 && y>=0);\ |
\ |
if(D){\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}else{\ |
const int E= B+C;\ |
const int step= C ? stride : 1;\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + E*src[step+0]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}\ |
}\ |
static void FUNCC(OPNAME ## h264_chroma_mc2)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\ |
pixel *dst = (pixel*)_dst;\ |
pixel *src = (pixel*)_src;\ |
const int A=(8-x)*(8-y);\ |
const int B=( x)*(8-y);\ |
const int C=(8-x)*( y);\ |
const int D=( x)*( y);\ |
int i;\ |
stride >>= sizeof(pixel)-1;\ |
\ |
av_assert2(x<8 && y<8 && x>=0 && y>=0);\ |
\ |
if(D){\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\ |
OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}else{\ |
const int E= B+C;\ |
const int step= C ? stride : 1;\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + E*src[step+0]));\ |
OP(dst[1], (A*src[1] + E*src[step+1]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}\ |
}\ |
\ |
static void FUNCC(OPNAME ## h264_chroma_mc4)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\ |
pixel *dst = (pixel*)_dst;\ |
pixel *src = (pixel*)_src;\ |
const int A=(8-x)*(8-y);\ |
const int B=( x)*(8-y);\ |
const int C=(8-x)*( y);\ |
const int D=( x)*( y);\ |
int i;\ |
stride >>= sizeof(pixel)-1;\ |
\ |
av_assert2(x<8 && y<8 && x>=0 && y>=0);\ |
\ |
if(D){\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\ |
OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\ |
OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\ |
OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}else{\ |
const int E= B+C;\ |
const int step= C ? stride : 1;\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + E*src[step+0]));\ |
OP(dst[1], (A*src[1] + E*src[step+1]));\ |
OP(dst[2], (A*src[2] + E*src[step+2]));\ |
OP(dst[3], (A*src[3] + E*src[step+3]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}\ |
}\ |
\ |
static void FUNCC(OPNAME ## h264_chroma_mc8)(uint8_t *_dst/*align 8*/, uint8_t *_src/*align 1*/, int stride, int h, int x, int y){\ |
pixel *dst = (pixel*)_dst;\ |
pixel *src = (pixel*)_src;\ |
const int A=(8-x)*(8-y);\ |
const int B=( x)*(8-y);\ |
const int C=(8-x)*( y);\ |
const int D=( x)*( y);\ |
int i;\ |
stride >>= sizeof(pixel)-1;\ |
\ |
av_assert2(x<8 && y<8 && x>=0 && y>=0);\ |
\ |
if(D){\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + B*src[1] + C*src[stride+0] + D*src[stride+1]));\ |
OP(dst[1], (A*src[1] + B*src[2] + C*src[stride+1] + D*src[stride+2]));\ |
OP(dst[2], (A*src[2] + B*src[3] + C*src[stride+2] + D*src[stride+3]));\ |
OP(dst[3], (A*src[3] + B*src[4] + C*src[stride+3] + D*src[stride+4]));\ |
OP(dst[4], (A*src[4] + B*src[5] + C*src[stride+4] + D*src[stride+5]));\ |
OP(dst[5], (A*src[5] + B*src[6] + C*src[stride+5] + D*src[stride+6]));\ |
OP(dst[6], (A*src[6] + B*src[7] + C*src[stride+6] + D*src[stride+7]));\ |
OP(dst[7], (A*src[7] + B*src[8] + C*src[stride+7] + D*src[stride+8]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}else{\ |
const int E= B+C;\ |
const int step= C ? stride : 1;\ |
for(i=0; i<h; i++){\ |
OP(dst[0], (A*src[0] + E*src[step+0]));\ |
OP(dst[1], (A*src[1] + E*src[step+1]));\ |
OP(dst[2], (A*src[2] + E*src[step+2]));\ |
OP(dst[3], (A*src[3] + E*src[step+3]));\ |
OP(dst[4], (A*src[4] + E*src[step+4]));\ |
OP(dst[5], (A*src[5] + E*src[step+5]));\ |
OP(dst[6], (A*src[6] + E*src[step+6]));\ |
OP(dst[7], (A*src[7] + E*src[step+7]));\ |
dst+= stride;\ |
src+= stride;\ |
}\ |
}\ |
} |
|
#define op_avg(a, b) a = (((a)+(((b) + 32)>>6)+1)>>1) |
#define op_put(a, b) a = (((b) + 32)>>6) |
|
H264_CHROMA_MC(put_ , op_put) |
H264_CHROMA_MC(avg_ , op_avg) |
#undef op_avg |
#undef op_put |