mc.c
资源名称:chapter15.rar [点击查看]
上传用户:hjq518
上传日期:2021-12-09
资源大小:5084k
文件大小:23k
源码类别:
Audio
开发平台:
Visual C++
- /*****************************************************************************
- * mc.c: h264 encoder library (Motion Compensation)
- *****************************************************************************
- * Copyright (C) 2003-2008 x264 project
- *
- * Authors: Eric Petit <eric.petit@lapsus.org>
- * Guillaume Poirier <gpoirier@mplayerhq.hu>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program 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 General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02111, USA.
- *****************************************************************************/
- #include <stdlib.h>
- #include <stdio.h>
- #include <string.h>
- #include <stdint.h>
- #include <stdarg.h>
- #ifdef SYS_LINUX
- #include <altivec.h>
- #endif
- #include "x264.h"
- #include "common/common.h"
- #include "common/mc.h"
- #include "mc.h"
- #include "ppccommon.h"
- typedef void (*pf_mc_t)( uint8_t *src, int i_src,
- uint8_t *dst, int i_dst, int i_height );
- static inline int x264_tapfilter( uint8_t *pix, int i_pix_next )
- {
- return pix[-2*i_pix_next] - 5*pix[-1*i_pix_next] + 20*(pix[0] +
- pix[1*i_pix_next]) - 5*pix[ 2*i_pix_next] +
- pix[ 3*i_pix_next];
- }
- static inline int x264_tapfilter1( uint8_t *pix )
- {
- return pix[-2] - 5*pix[-1] + 20*(pix[0] + pix[1]) - 5*pix[ 2] +
- pix[ 3];
- }
- /* pixel_avg */
- static inline void pixel_avg_w4( uint8_t *dst, int i_dst,
- uint8_t *src1, int i_src1,
- uint8_t *src2, int i_src2,
- int i_height )
- {
- int x, y;
- for( y = 0; y < i_height; y++ )
- {
- for( x = 0; x < 4; x++ )
- {
- dst[x] = ( src1[x] + src2[x] + 1 ) >> 1;
- }
- dst += i_dst;
- src1 += i_src1;
- src2 += i_src2;
- }
- }
- static inline void pixel_avg_w8( uint8_t *dst, int i_dst,
- uint8_t *src1, int i_src1,
- uint8_t *src2, int i_src2,
- int i_height )
- {
- int y;
- vec_u8_t src1v, src2v;
- LOAD_ZERO;
- PREP_LOAD;
- PREP_STORE8;
- for( y = 0; y < i_height; y++ )
- {
- VEC_LOAD( src1, src1v, 8, vec_u8_t );
- VEC_LOAD( src2, src2v, 8, vec_u8_t );
- src1v = vec_avg( src1v, src2v );
- VEC_STORE8( src1v, dst );
- dst += i_dst;
- src1 += i_src1;
- src2 += i_src2;
- }
- }
- static inline void pixel_avg_w16( uint8_t *dst, int i_dst,
- uint8_t *src1, int i_src1,
- uint8_t *src2, int i_src2,
- int i_height )
- {
- int y;
- vec_u8_t src1v, src2v;
- PREP_LOAD;
- PREP_STORE16;
- for( y = 0; y < i_height; y++ )
- {
- VEC_LOAD( src1, src1v, 16, vec_u8_t );
- VEC_LOAD( src2, src2v, 16, vec_u8_t );
- src1v = vec_avg( src1v, src2v );
- VEC_STORE16( src1v, dst );
- dst += i_dst;
- src1 += i_src1;
- src2 += i_src2;
- }
- }
- /* mc_copy: plain c */
- #define MC_COPY( name, a )
- static void name( uint8_t *src, int i_src,
- uint8_t *dst, int i_dst, int i_height )
- {
- int y;
- for( y = 0; y < i_height; y++ )
- {
- memcpy( dst, src, a );
- src += i_src;
- dst += i_dst;
- }
- }
- MC_COPY( mc_copy_w4, 4 )
- MC_COPY( mc_copy_w8, 8 )
- MC_COPY( mc_copy_w16, 16 )
- void mc_luma_altivec( uint8_t *dst, int i_dst_stride,
- uint8_t *src[4], int i_src_stride,
- int mvx, int mvy,
- int i_width, int i_height )
- {
- uint8_t *src1, *src2;
- /* todo : fixme... */
- int correction = (((mvx&3) == 3 && (mvy&3) == 1) || ((mvx&3) == 1 && (mvy&3) == 3)) ? 1:0;
- int hpel1x = mvx>>1;
- int hpel1y = (mvy+1-correction)>>1;
- int filter1 = (hpel1x & 1) + ( (hpel1y & 1) << 1 );
- src1 = src[filter1] + (hpel1y >> 1) * i_src_stride + (hpel1x >> 1);
- if ( (mvx|mvy) & 1 ) /* qpel interpolation needed */
- {
- int hpel2x = (mvx+1)>>1;
- int hpel2y = (mvy+correction)>>1;
- int filter2 = (hpel2x & 1) + ( (hpel2y & 1) <<1 );
- src2 = src[filter2] + (hpel2y >> 1) * i_src_stride + (hpel2x >> 1);
- switch(i_width) {
- case 4:
- pixel_avg_w4( dst, i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- break;
- case 8:
- pixel_avg_w8( dst, i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- break;
- case 16:
- default:
- pixel_avg_w16( dst, i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- }
- }
- else
- {
- switch(i_width) {
- case 4:
- mc_copy_w4( src1, i_src_stride, dst, i_dst_stride, i_height );
- break;
- case 8:
- mc_copy_w8( src1, i_src_stride, dst, i_dst_stride, i_height );
- break;
- case 16:
- mc_copy_w16( src1, i_src_stride, dst, i_dst_stride, i_height );
- break;
- }
- }
- }
- uint8_t *get_ref_altivec( uint8_t *dst, int * i_dst_stride,
- uint8_t *src[4], int i_src_stride,
- int mvx, int mvy,
- int i_width, int i_height )
- {
- uint8_t *src1, *src2;
- /* todo : fixme... */
- int correction = (((mvx&3) == 3 && (mvy&3) == 1) || ((mvx&3) == 1 && (mvy&3) == 3)) ? 1:0;
- int hpel1x = mvx>>1;
- int hpel1y = (mvy+1-correction)>>1;
- int filter1 = (hpel1x & 1) + ( (hpel1y & 1) << 1 );
- src1 = src[filter1] + (hpel1y >> 1) * i_src_stride + (hpel1x >> 1);
- if ( (mvx|mvy) & 1 ) /* qpel interpolation needed */
- {
- int hpel2x = (mvx+1)>>1;
- int hpel2y = (mvy+correction)>>1;
- int filter2 = (hpel2x & 1) + ( (hpel2y & 1) <<1 );
- src2 = src[filter2] + (hpel2y >> 1) * i_src_stride + (hpel2x >> 1);
- switch(i_width) {
- case 4:
- pixel_avg_w4( dst, *i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- break;
- case 8:
- pixel_avg_w8( dst, *i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- break;
- case 12:
- case 16:
- default:
- pixel_avg_w16( dst, *i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- break;
- case 20:
- //FIXME suboptimal
- pixel_avg_w16( dst, *i_dst_stride, src1, i_src_stride,
- src2, i_src_stride, i_height );
- pixel_avg_w4( dst+16, *i_dst_stride, src1+16, i_src_stride,
- src2+16, i_src_stride, i_height );
- break;
- }
- return dst;
- }
- else
- {
- *i_dst_stride = i_src_stride;
- return src1;
- }
- }
- #define DO_PROCESS(a)
- src##a##v_16 = vec_u8_to_u16( src##a##v_8 );
- src##a##v_16 = vec_mladd( coeff##a##v, src##a##v_16, zero_u16v );
- dstv_16 = vec_add( dstv_16, src##a##v_16 )
- static void mc_chroma_altivec_4xh( uint8_t *dst, int i_dst_stride,
- uint8_t *src, int i_src_stride,
- int mvx, int mvy,
- int i_height )
- {
- uint8_t *srcp;
- int y;
- int d8x = mvx & 0x07;
- int d8y = mvy & 0x07;
- DECLARE_ALIGNED_16( uint16_t coeff[4] );
- coeff[0] = (8-d8x)*(8-d8y);
- coeff[1] = d8x *(8-d8y);
- coeff[2] = (8-d8x)*d8y;
- coeff[3] = d8x *d8y;
- src += (mvy >> 3) * i_src_stride + (mvx >> 3);
- srcp = &src[i_src_stride];
- LOAD_ZERO;
- PREP_LOAD;
- PREP_STORE4;
- vec_u16_t coeff0v, coeff1v, coeff2v, coeff3v;
- vec_u8_t src0v_8, src1v_8, src2v_8, src3v_8;
- vec_u16_t src0v_16, src1v_16, src2v_16, src3v_16;
- vec_u8_t dstv_8;
- vec_u16_t dstv_16;
- vec_u8_t permv;
- vec_u16_t shiftv;
- vec_u16_t k32v;
- coeff0v = vec_ld( 0, coeff );
- coeff3v = vec_splat( coeff0v, 3 );
- coeff2v = vec_splat( coeff0v, 2 );
- coeff1v = vec_splat( coeff0v, 1 );
- coeff0v = vec_splat( coeff0v, 0 );
- k32v = vec_sl( vec_splat_u16( 1 ), vec_splat_u16( 5 ) );
- permv = vec_lvsl( 0, (uint8_t *) 1 );
- shiftv = vec_splat_u16( 6 );
- VEC_LOAD( src, src2v_8, 5, vec_u8_t );
- src3v_8 = vec_perm( src2v_8, src2v_8, permv );
- for( y = 0; y < i_height; y++ )
- {
- src0v_8 = src2v_8;
- src1v_8 = src3v_8;
- VEC_LOAD( srcp, src2v_8, 5, vec_u8_t );
- src3v_8 = vec_perm( src2v_8, src2v_8, permv );
- dstv_16 = k32v;
- DO_PROCESS( 0 );
- DO_PROCESS( 1 );
- DO_PROCESS( 2 );
- DO_PROCESS( 3 );
- dstv_16 = vec_sr( dstv_16, shiftv );
- dstv_8 = vec_u16_to_u8( dstv_16 );
- VEC_STORE4( dstv_8, dst );
- dst += i_dst_stride;
- srcp += i_src_stride;
- }
- }
- static void mc_chroma_altivec_8xh( uint8_t *dst, int i_dst_stride,
- uint8_t *src, int i_src_stride,
- int mvx, int mvy,
- int i_height )
- {
- uint8_t *srcp;
- int y;
- int d8x = mvx & 0x07;
- int d8y = mvy & 0x07;
- DECLARE_ALIGNED_16( uint16_t coeff[4] );
- coeff[0] = (8-d8x)*(8-d8y);
- coeff[1] = d8x *(8-d8y);
- coeff[2] = (8-d8x)*d8y;
- coeff[3] = d8x *d8y;
- src += (mvy >> 3) * i_src_stride + (mvx >> 3);
- srcp = &src[i_src_stride];
- LOAD_ZERO;
- PREP_LOAD;
- PREP_STORE8;
- vec_u16_t coeff0v, coeff1v, coeff2v, coeff3v;
- vec_u8_t src0v_8, src1v_8, src2v_8, src3v_8;
- vec_u16_t src0v_16, src1v_16, src2v_16, src3v_16;
- vec_u8_t dstv_8;
- vec_u16_t dstv_16;
- vec_u8_t permv;
- vec_u16_t shiftv;
- vec_u16_t k32v;
- coeff0v = vec_ld( 0, coeff );
- coeff3v = vec_splat( coeff0v, 3 );
- coeff2v = vec_splat( coeff0v, 2 );
- coeff1v = vec_splat( coeff0v, 1 );
- coeff0v = vec_splat( coeff0v, 0 );
- k32v = vec_sl( vec_splat_u16( 1 ), vec_splat_u16( 5 ) );
- permv = vec_lvsl( 0, (uint8_t *) 1 );
- shiftv = vec_splat_u16( 6 );
- VEC_LOAD( src, src2v_8, 9, vec_u8_t );
- src3v_8 = vec_perm( src2v_8, src2v_8, permv );
- for( y = 0; y < i_height; y++ )
- {
- src0v_8 = src2v_8;
- src1v_8 = src3v_8;
- VEC_LOAD( srcp, src2v_8, 9, vec_u8_t );
- src3v_8 = vec_perm( src2v_8, src2v_8, permv );
- dstv_16 = k32v;
- DO_PROCESS( 0 );
- DO_PROCESS( 1 );
- DO_PROCESS( 2 );
- DO_PROCESS( 3 );
- dstv_16 = vec_sr( dstv_16, shiftv );
- dstv_8 = vec_u16_to_u8( dstv_16 );
- VEC_STORE8( dstv_8, dst );
- dst += i_dst_stride;
- srcp += i_src_stride;
- }
- }
- static void mc_chroma_altivec( uint8_t *dst, int i_dst_stride,
- uint8_t *src, int i_src_stride,
- int mvx, int mvy,
- int i_width, int i_height )
- {
- if( i_width == 8 )
- {
- mc_chroma_altivec_8xh( dst, i_dst_stride, src, i_src_stride,
- mvx, mvy, i_height );
- }
- else
- {
- mc_chroma_altivec_4xh( dst, i_dst_stride, src, i_src_stride,
- mvx, mvy, i_height );
- }
- }
- #define HPEL_FILTER_1( t1v, t2v, t3v, t4v, t5v, t6v )
- {
- t1v = vec_add( t1v, t6v );
- t2v = vec_add( t2v, t5v );
- t3v = vec_add( t3v, t4v );
- t1v = vec_sub( t1v, t2v ); /* (a-b) */
- t2v = vec_sub( t2v, t3v ); /* (b-c) */
- t2v = vec_sl( t2v, twov ); /* (b-c)*4 */
- t1v = vec_sub( t1v, t2v ); /* a-5*b+4*c */
- t3v = vec_sl( t3v, fourv ); /* 16*c */
- t1v = vec_add( t1v, t3v ); /* a-5*b+20*c */
- }
- #define HPEL_FILTER_2( t1v, t2v, t3v, t4v, t5v, t6v )
- {
- t1v = vec_add( t1v, t6v );
- t2v = vec_add( t2v, t5v );
- t3v = vec_add( t3v, t4v );
- t1v = vec_sub( t1v, t2v ); /* (a-b) */
- t1v = vec_sra( t1v, twov ); /* (a-b)/4 */
- t1v = vec_sub( t1v, t2v ); /* (a-b)/4-b */
- t1v = vec_add( t1v, t3v ); /* (a-b)/4-b+c */
- t1v = vec_sra( t1v, twov ); /* ((a-b)/4-b+c)/4 */
- t1v = vec_add( t1v, t3v ); /* ((a-b)/4-b+c)/4+c = (a-5*b+20*c)/16 */
- }
- #define HPEL_FILTER_HORIZONTAL()
- {
- VEC_LOAD( &src[x- 2+i_stride*y], src1v, 16, vec_u8_t );
- VEC_LOAD( &src[x+14+i_stride*y], src6v, 16, vec_u8_t );
- src2v = vec_sld( src1v, src6v, 1 );
- src3v = vec_sld( src1v, src6v, 2 );
- src4v = vec_sld( src1v, src6v, 3 );
- src5v = vec_sld( src1v, src6v, 4 );
- src6v = vec_sld( src1v, src6v, 5 );
- temp1v = vec_u8_to_s16_h( src1v );
- temp2v = vec_u8_to_s16_h( src2v );
- temp3v = vec_u8_to_s16_h( src3v );
- temp4v = vec_u8_to_s16_h( src4v );
- temp5v = vec_u8_to_s16_h( src5v );
- temp6v = vec_u8_to_s16_h( src6v );
- HPEL_FILTER_1( temp1v, temp2v, temp3v,
- temp4v, temp5v, temp6v );
- dest1v = vec_add( temp1v, sixteenv );
- dest1v = vec_sra( dest1v, fivev );
- temp1v = vec_u8_to_s16_l( src1v );
- temp2v = vec_u8_to_s16_l( src2v );
- temp3v = vec_u8_to_s16_l( src3v );
- temp4v = vec_u8_to_s16_l( src4v );
- temp5v = vec_u8_to_s16_l( src5v );
- temp6v = vec_u8_to_s16_l( src6v );
- HPEL_FILTER_1( temp1v, temp2v, temp3v,
- temp4v, temp5v, temp6v );
- dest2v = vec_add( temp1v, sixteenv );
- dest2v = vec_sra( dest2v, fivev );
- destv = vec_packsu( dest1v, dest2v );
- VEC_STORE16( destv, &dsth[x+i_stride*y] );
- }
- #define HPEL_FILTER_VERTICAL()
- {
- VEC_LOAD( &src[x+i_stride*(y-2)], src1v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y-1)], src2v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y-0)], src3v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y+1)], src4v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y+2)], src5v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y+3)], src6v, 16, vec_u8_t );
- temp1v = vec_u8_to_s16_h( src1v );
- temp2v = vec_u8_to_s16_h( src2v );
- temp3v = vec_u8_to_s16_h( src3v );
- temp4v = vec_u8_to_s16_h( src4v );
- temp5v = vec_u8_to_s16_h( src5v );
- temp6v = vec_u8_to_s16_h( src6v );
- HPEL_FILTER_1( temp1v, temp2v, temp3v,
- temp4v, temp5v, temp6v );
- dest1v = vec_add( temp1v, sixteenv );
- dest1v = vec_sra( dest1v, fivev );
- temp4v = vec_u8_to_s16_l( src1v );
- temp5v = vec_u8_to_s16_l( src2v );
- temp6v = vec_u8_to_s16_l( src3v );
- temp7v = vec_u8_to_s16_l( src4v );
- temp8v = vec_u8_to_s16_l( src5v );
- temp9v = vec_u8_to_s16_l( src6v );
- HPEL_FILTER_1( temp4v, temp5v, temp6v,
- temp7v, temp8v, temp9v );
- dest2v = vec_add( temp4v, sixteenv );
- dest2v = vec_sra( dest2v, fivev );
- destv = vec_packsu( dest1v, dest2v );
- VEC_STORE16( destv, &dstv[x+i_stride*y] );
- }
- #define HPEL_FILTER_CENTRAL()
- {
- temp1v = vec_sld( tempav, tempbv, 12 );
- temp2v = vec_sld( tempav, tempbv, 14 );
- temp3v = tempbv;
- temp4v = vec_sld( tempbv, tempcv, 2 );
- temp5v = vec_sld( tempbv, tempcv, 4 );
- temp6v = vec_sld( tempbv, tempcv, 6 );
- HPEL_FILTER_2( temp1v, temp2v, temp3v,
- temp4v, temp5v, temp6v );
- dest1v = vec_add( temp1v, thirtytwov );
- dest1v = vec_sra( dest1v, sixv );
- temp1v = vec_sld( tempbv, tempcv, 12 );
- temp2v = vec_sld( tempbv, tempcv, 14 );
- temp3v = tempcv;
- temp4v = vec_sld( tempcv, tempdv, 2 );
- temp5v = vec_sld( tempcv, tempdv, 4 );
- temp6v = vec_sld( tempcv, tempdv, 6 );
- HPEL_FILTER_2( temp1v, temp2v, temp3v,
- temp4v, temp5v, temp6v );
- dest2v = vec_add( temp1v, thirtytwov );
- dest2v = vec_sra( dest2v, sixv );
- destv = vec_packsu( dest1v, dest2v );
- VEC_STORE16( destv, &dstc[x-16+i_stride*y] );
- }
- void x264_hpel_filter_altivec( uint8_t *dsth, uint8_t *dstv, uint8_t *dstc, uint8_t *src,
- int i_stride, int i_width, int i_height )
- {
- int x, y;
- vec_u8_t destv;
- vec_u8_t src1v, src2v, src3v, src4v, src5v, src6v;
- vec_s16_t dest1v, dest2v;
- vec_s16_t temp1v, temp2v, temp3v, temp4v, temp5v, temp6v, temp7v, temp8v, temp9v;
- vec_s16_t tempav, tempbv, tempcv, tempdv, tempev;
- PREP_LOAD;
- PREP_STORE16;
- LOAD_ZERO;
- vec_u16_t twov, fourv, fivev, sixv;
- vec_s16_t sixteenv, thirtytwov;
- vect_ushort_u temp_u;
- temp_u.s[0]=2;
- twov = vec_splat( temp_u.v, 0 );
- temp_u.s[0]=4;
- fourv = vec_splat( temp_u.v, 0 );
- temp_u.s[0]=5;
- fivev = vec_splat( temp_u.v, 0 );
- temp_u.s[0]=6;
- sixv = vec_splat( temp_u.v, 0 );
- temp_u.s[0]=16;
- sixteenv = (vec_s16_t)vec_splat( temp_u.v, 0 );
- temp_u.s[0]=32;
- thirtytwov = (vec_s16_t)vec_splat( temp_u.v, 0 );
- for( y = 0; y < i_height; y++ )
- {
- x = 0;
- /* horizontal_filter */
- HPEL_FILTER_HORIZONTAL();
- /* vertical_filter */
- HPEL_FILTER_VERTICAL();
- /* central_filter */
- tempav = tempcv;
- tempbv = tempdv;
- tempcv = vec_splat( temp1v, 0 ); /* first only */
- tempdv = temp1v;
- tempev = temp4v;
- for( x = 16; x < i_width; x+=16 )
- {
- /* horizontal_filter */
- HPEL_FILTER_HORIZONTAL();
- /* vertical_filter */
- HPEL_FILTER_VERTICAL();
- /* central_filter */
- tempav = tempcv;
- tempbv = tempdv;
- tempcv = tempev;
- tempdv = temp1v;
- tempev = temp4v;
- HPEL_FILTER_CENTRAL();
- }
- /* Partial vertical filter */
- VEC_LOAD( &src[x+i_stride*(y-2)], src1v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y-1)], src2v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y-0)], src3v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y+1)], src4v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y+2)], src5v, 16, vec_u8_t );
- VEC_LOAD( &src[x+i_stride*(y+3)], src6v, 16, vec_u8_t );
- temp1v = vec_u8_to_s16_h( src1v );
- temp2v = vec_u8_to_s16_h( src2v );
- temp3v = vec_u8_to_s16_h( src3v );
- temp4v = vec_u8_to_s16_h( src4v );
- temp5v = vec_u8_to_s16_h( src5v );
- temp6v = vec_u8_to_s16_h( src6v );
- HPEL_FILTER_1( temp1v, temp2v, temp3v,
- temp4v, temp5v, temp6v );
- /* central_filter */
- tempav = tempcv;
- tempbv = tempdv;
- tempcv = tempev;
- tempdv = temp1v;
- /* tempev is not used */
- HPEL_FILTER_CENTRAL();
- }
- }
- void x264_mc_altivec_init( x264_mc_functions_t *pf )
- {
- pf->mc_luma = mc_luma_altivec;
- pf->get_ref = get_ref_altivec;
- pf->mc_chroma = mc_chroma_altivec;
- pf->hpel_filter = x264_hpel_filter_altivec;
- }