predict.c
上传用户:lctgjx
上传日期:2022-06-04
资源大小:8887k
文件大小:7k
源码类别:

流媒体/Mpeg4/MP4

开发平台:

Visual C++

  1. /*****************************************************************************
  2.  * predict.c: h264 encoder
  3.  *****************************************************************************
  4.  * Copyright (C) 2007-2009 Guillaume Poirier <gpoirier@mplayerhq.hu>
  5.  *
  6.  * This program is free software; you can redistribute it and/or modify
  7.  * it under the terms of the GNU General Public License as published by
  8.  * the Free Software Foundation; either version 2 of the License, or
  9.  * (at your option) any later version.
  10.  *
  11.  * This program is distributed in the hope that it will be useful,
  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14.  * GNU General Public License for more details.
  15.  *
  16.  * You should have received a copy of the GNU General Public License
  17.  * along with this program; if not, write to the Free Software
  18.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  19.  *****************************************************************************/
  20. #include "common/common.h"
  21. #include "predict.h"
  22. #include "pixel.h"
  23. #include "ppccommon.h"
  24. static void predict_8x8c_p_altivec( uint8_t *src )
  25. {
  26.     int i;
  27.     int a, b, c;
  28.     int H = 0;
  29.     int V = 0;
  30.     int i00;
  31.     for( i = 0; i < 4; i++ )
  32.     {
  33.         H += ( i + 1 ) * ( src[4+i - FDEC_STRIDE] - src[2 - i -FDEC_STRIDE] );
  34.         V += ( i + 1 ) * ( src[-1 +(i+4)*FDEC_STRIDE] - src[-1+(2-i)*FDEC_STRIDE] );
  35.     }
  36.     a = 16 * ( src[-1+7*FDEC_STRIDE] + src[7 - FDEC_STRIDE] );
  37.     b = ( 17 * H + 16 ) >> 5;
  38.     c = ( 17 * V + 16 ) >> 5;
  39.     i00 = a -3*b -3*c + 16;
  40.     vec_s16_u i00_u, b_u, c_u;
  41.     i00_u.s[0] = i00;
  42.     b_u.s[0]   = b;
  43.     c_u.s[0]   = c;
  44.     vec_u16_t val5_v = vec_splat_u16(5);
  45.     vec_s16_t i00_v, b_v, c_v;
  46.     i00_v = vec_splat(i00_u.v, 0);
  47.     b_v = vec_splat(b_u.v, 0);
  48.     c_v = vec_splat(c_u.v, 0);
  49.     vec_s16_t induc_v  = (vec_s16_t) CV(0, 1, 2, 3, 4, 5, 6, 7);
  50.     vec_s16_t add_i0_b_0v = vec_mladd(induc_v, b_v, i00_v);
  51.     PREP_STORE8;
  52.     for( i = 0; i < 8; ++i )
  53.     {
  54.         vec_s16_t shift_0_v = vec_sra(add_i0_b_0v, val5_v);
  55.         vec_u8_t com_sat_v = vec_packsu(shift_0_v, shift_0_v);
  56.         VEC_STORE8(com_sat_v, &src[0]);
  57.         src += FDEC_STRIDE;
  58.         add_i0_b_0v = vec_adds(add_i0_b_0v, c_v);
  59.     }
  60. }
  61. /****************************************************************************
  62.  * 16x16 prediction for intra luma block
  63.  ****************************************************************************/
  64. static void predict_16x16_p_altivec( uint8_t *src )
  65. {
  66.     int16_t a, b, c, i;
  67.     int H = 0;
  68.     int V = 0;
  69.     int16_t i00;
  70.     for( i = 1; i <= 8; i++ )
  71.     {
  72.         H += i * ( src[7+i - FDEC_STRIDE ]  - src[7-i - FDEC_STRIDE ] );
  73.         V += i * ( src[(7+i)*FDEC_STRIDE -1] - src[(7-i)*FDEC_STRIDE -1] );
  74.     }
  75.     a = 16 * ( src[15*FDEC_STRIDE -1] + src[15 - FDEC_STRIDE] );
  76.     b = ( 5 * H + 32 ) >> 6;
  77.     c = ( 5 * V + 32 ) >> 6;
  78.     i00 = a - b * 7 - c * 7 + 16;
  79.     vec_s16_u i00_u, b_u, c_u;
  80.     i00_u.s[0] = i00;
  81.     b_u.s[0]   = b;
  82.     c_u.s[0]   = c;
  83.     vec_u16_t val5_v = vec_splat_u16(5);
  84.     vec_s16_t i00_v, b_v, c_v;
  85.     i00_v = vec_splat(i00_u.v, 0);
  86.     b_v = vec_splat(b_u.v, 0);
  87.     c_v = vec_splat(c_u.v, 0);
  88.     vec_s16_t induc_v  = (vec_s16_t) CV(0,  1,  2,  3,  4,  5,  6,  7);
  89.     vec_s16_t b8_v = vec_sl(b_v, vec_splat_u16(3));
  90.     vec_s16_t add_i0_b_0v = vec_mladd(induc_v, b_v, i00_v);
  91.     vec_s16_t add_i0_b_8v = vec_adds(b8_v, add_i0_b_0v);
  92.     int y;
  93.     for( y = 0; y < 16; y++ )
  94.     {
  95.         vec_s16_t shift_0_v = vec_sra(add_i0_b_0v, val5_v);
  96.         vec_s16_t shift_8_v = vec_sra(add_i0_b_8v, val5_v);
  97.         vec_u8_t com_sat_v = vec_packsu(shift_0_v, shift_8_v);
  98.         vec_st( com_sat_v, 0, &src[0]);
  99.         src += FDEC_STRIDE;
  100.         add_i0_b_0v = vec_adds(add_i0_b_0v, c_v);
  101.         add_i0_b_8v = vec_adds(add_i0_b_8v, c_v);
  102.     }
  103. }
  104. #define PREDICT_16x16_DC_ALTIVEC(v) 
  105. for (i=0; i<16; i+=2)               
  106. {                                   
  107.     vec_st(v, 0, src);              
  108.     vec_st(v, FDEC_STRIDE, src);    
  109.     src += FDEC_STRIDE*2;           
  110. }
  111. static void predict_16x16_dc_altivec( uint8_t *src )
  112. {
  113.     uint32_t dc = 0;
  114.     int i;
  115.     for( i = 0; i < 16; i++ )
  116.     {
  117.         dc += src[-1 + i * FDEC_STRIDE];
  118.         dc += src[i - FDEC_STRIDE];
  119.     }
  120.     vec_u8_u v ; v.s[0] = (( dc + 16 ) >> 5);
  121.     vec_u8_t bc_v = vec_splat(v.v, 0);
  122.     PREDICT_16x16_DC_ALTIVEC(bc_v);
  123. }
  124. static void predict_16x16_dc_left_altivec( uint8_t *src )
  125. {
  126.     uint32_t dc = 0;
  127.     int i;
  128.     for( i = 0; i < 16; i++ )
  129.     {
  130.         dc += src[-1 + i * FDEC_STRIDE];
  131.     }
  132.     vec_u8_u v ; v.s[0] = (( dc + 8 ) >> 4);
  133.     vec_u8_t bc_v = vec_splat(v.v, 0);
  134.     PREDICT_16x16_DC_ALTIVEC(bc_v);
  135. }
  136. static void predict_16x16_dc_top_altivec( uint8_t *src )
  137. {
  138.     uint32_t dc = 0;
  139.     int i;
  140.     for( i = 0; i < 16; i++ )
  141.     {
  142.         dc += src[i - FDEC_STRIDE];
  143.     }
  144.     vec_u8_u v ; v.s[0] = (( dc + 8 ) >> 4);
  145.     vec_u8_t bc_v = vec_splat(v.v, 0);
  146.     PREDICT_16x16_DC_ALTIVEC(bc_v);
  147. }
  148. static void predict_16x16_dc_128_altivec( uint8_t *src )
  149. {
  150.     int i;
  151.     /* test if generating the constant is faster than loading it.
  152.     vector unsigned int bc_v = (vector unsigned int)CV(0x80808080, 0x80808080, 0x80808080, 0x80808080);
  153.     */
  154.     vec_u8_t bc_v = vec_vslb((vec_u8_t)vec_splat_u8(1),(vec_u8_t)vec_splat_u8(7));
  155.     PREDICT_16x16_DC_ALTIVEC(bc_v);
  156. }
  157. static void predict_16x16_h_altivec( uint8_t *src )
  158. {
  159.     int i;
  160.     for( i = 0; i < 16; i++ )
  161.     {
  162.         vec_u8_t v = vec_ld(-1, src);
  163.         vec_u8_t v_v = vec_splat(v, 15);
  164.         vec_st(v_v, 0, src);
  165.         src += FDEC_STRIDE;
  166.     }
  167. }
  168. static void predict_16x16_v_altivec( uint8_t *src )
  169. {
  170.     vec_u32_u v;
  171.     v.s[0] = *(uint32_t*)&src[ 0-FDEC_STRIDE];
  172.     v.s[1] = *(uint32_t*)&src[ 4-FDEC_STRIDE];
  173.     v.s[2] = *(uint32_t*)&src[ 8-FDEC_STRIDE];
  174.     v.s[3] = *(uint32_t*)&src[12-FDEC_STRIDE];
  175.     int i;
  176.     for( i = 0; i < 16; i++ )
  177.     {
  178.         vec_st(v.v, 0, (uint32_t*)src);
  179.         src += FDEC_STRIDE;
  180.     }
  181. }
  182. /****************************************************************************
  183.  * Exported functions:
  184.  ****************************************************************************/
  185. void x264_predict_16x16_init_altivec( x264_predict_t pf[7] )
  186. {
  187.     pf[I_PRED_16x16_V ]      = predict_16x16_v_altivec;
  188.     pf[I_PRED_16x16_H ]      = predict_16x16_h_altivec;
  189.     pf[I_PRED_16x16_DC]      = predict_16x16_dc_altivec;
  190.     pf[I_PRED_16x16_P ]      = predict_16x16_p_altivec;
  191.     pf[I_PRED_16x16_DC_LEFT] = predict_16x16_dc_left_altivec;
  192.     pf[I_PRED_16x16_DC_TOP ] = predict_16x16_dc_top_altivec;
  193.     pf[I_PRED_16x16_DC_128 ] = predict_16x16_dc_128_altivec;
  194. }
  195. void x264_predict_8x8c_init_altivec( x264_predict_t pf[7] )
  196. {
  197.     pf[I_PRED_CHROMA_P]       = predict_8x8c_p_altivec;
  198. }