开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 多媒体 > Audio > 查看源码
mc-c.c
资源名称：chapter15.rar　[点击查看]
上传用户：hjq518
上传日期：2021-12-09
资源大小：5084k
文件大小：13k
源码类别：

Audio

开发平台：

Visual C++

mc-c.c：源码内容
  1. /*****************************************************************************
  2.  * mc-c.c: h264 encoder library (Motion Compensation)
  3.  *****************************************************************************
  4.  * Copyright (C) 2003-2008 x264 project
  5.  *
  6.  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
  7.  *          Loren Merritt <lorenm@u.washington.edu>
  8.  *          Jason Garrett-Glaser <darkshikari@gmail.com>
  9.  *
  10.  * This program is free software; you can redistribute it and/or modify
  11.  * it under the terms of the GNU General Public License as published by
  12.  * the Free Software Foundation; either version 2 of the License, or
  13.  * (at your option) any later version.
  14.  *
  15.  * This program is distributed in the hope that it will be useful,
  16.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18.  * GNU General Public License for more details.
  19.  *
  20.  * You should have received a copy of the GNU General Public License
  21.  * along with this program; if not, write to the Free Software
  22.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111, USA.
  23.  *****************************************************************************/
  25. #include <stdlib.h>
  26. #include <stdio.h>
  27. #include <string.h>
  29. #include "common/common.h"
  30. #include "mc.h"
  32. /* NASM functions */
  33. extern void x264_pixel_avg_16x16_sse2( uint8_t *, int, uint8_t *, int );
  34. extern void x264_pixel_avg_16x8_sse2( uint8_t *, int, uint8_t *, int );
  35. extern void x264_pixel_avg_16x16_mmxext( uint8_t *, int, uint8_t *, int );
  36. extern void x264_pixel_avg_16x8_mmxext( uint8_t *, int, uint8_t *, int );
  37. extern void x264_pixel_avg_8x16_mmxext( uint8_t *, int, uint8_t *, int );
  38. extern void x264_pixel_avg_8x8_mmxext( uint8_t *, int, uint8_t *, int );
  39. extern void x264_pixel_avg_8x4_mmxext( uint8_t *, int, uint8_t *, int );
  40. extern void x264_pixel_avg_4x8_mmxext( uint8_t *, int, uint8_t *, int );
  41. extern void x264_pixel_avg_4x4_mmxext( uint8_t *, int, uint8_t *, int );
  42. extern void x264_pixel_avg_4x2_mmxext( uint8_t *, int, uint8_t *, int );
  43. extern void x264_mc_copy_w4_mmx( uint8_t *, int, uint8_t *, int, int );
  44. extern void x264_mc_copy_w8_mmx( uint8_t *, int, uint8_t *, int, int );
  45. extern void x264_mc_copy_w16_mmx( uint8_t *, int, uint8_t *, int, int );
  46. extern void x264_mc_copy_w16_sse2( uint8_t *, int, uint8_t *, int, int );
  47. extern void x264_mc_copy_w16_sse3( uint8_t *, int, uint8_t *, int, int );
  48. extern void x264_mc_copy_w16_aligned_sse2( uint8_t *, int, uint8_t *, int, int );
  49. extern void x264_pixel_avg_weight_4x4_mmxext( uint8_t *, int, uint8_t *, int, int );
  50. extern void x264_pixel_avg_weight_w8_mmxext( uint8_t *, int, uint8_t *, int, int, int );
  51. extern void x264_pixel_avg_weight_w8_sse2( uint8_t *, int, uint8_t *, int, int, int );
  52. extern void x264_pixel_avg_weight_w16_sse2( uint8_t *, int, uint8_t *, int, int, int );
  53. extern void x264_pixel_avg_weight_w16_mmxext( uint8_t *, int, uint8_t *, int, int, int );
  54. extern void x264_prefetch_fenc_mmxext( uint8_t *, int, uint8_t *, int, int );
  55. extern void x264_prefetch_ref_mmxext( uint8_t *, int, int );
  56. extern void x264_mc_chroma_mmxext( uint8_t *src, int i_src_stride,
  57.                                    uint8_t *dst, int i_dst_stride,
  58.                                    int dx, int dy, int i_width, int i_height );
  59. extern void x264_mc_chroma_sse2( uint8_t *src, int i_src_stride,
  60.                                  uint8_t *dst, int i_dst_stride,
  61.                                  int dx, int dy, int i_width, int i_height );
  62. extern void x264_mc_chroma_ssse3( uint8_t *src, int i_src_stride,
  63.                                   uint8_t *dst, int i_dst_stride,
  64.                                   int dx, int dy, int i_width, int i_height );
  65. extern void x264_plane_copy_mmxext( uint8_t *, int, uint8_t *, int, int w, int h);
  66. extern void *x264_memcpy_aligned_mmx( void * dst, const void * src, size_t n );
  67. extern void *x264_memcpy_aligned_sse2( void * dst, const void * src, size_t n );
  68. extern void x264_memzero_aligned_mmx( void * dst, int n );
  69. extern void x264_memzero_aligned_sse2( void * dst, int n );
  70. #define LOWRES(cpu) 
  71. extern void x264_frame_init_lowres_core_##cpu( uint8_t *src0, uint8_t *dst0, uint8_t *dsth, uint8_t *dstv, uint8_t *dstc,
  72.                                                int src_stride, int dst_stride, int width, int height );
  73. LOWRES(mmxext)
  74. LOWRES(cache32_mmxext)
  75. LOWRES(sse2)
  76. LOWRES(ssse3)
  78. #define PIXEL_AVG_W(width,cpu)
  79. extern void x264_pixel_avg2_w##width##_##cpu( uint8_t *, int, uint8_t *, int, uint8_t *, int );
  80. /* This declares some functions that don't exist, but that isn't a problem. */
  81. #define PIXEL_AVG_WALL(cpu)
  82. PIXEL_AVG_W(4,cpu); PIXEL_AVG_W(8,cpu); PIXEL_AVG_W(12,cpu); PIXEL_AVG_W(16,cpu); PIXEL_AVG_W(20,cpu);
  84. PIXEL_AVG_WALL(mmxext)
  85. PIXEL_AVG_WALL(cache32_mmxext)
  86. PIXEL_AVG_WALL(cache64_mmxext)
  87. PIXEL_AVG_WALL(cache64_sse2)
  88. PIXEL_AVG_WALL(sse2)
  90. #define AVG_WEIGHT(W,H,name) 
  91. static void x264_pixel_avg_weight_ ## W ## x ## H ## _##name( uint8_t *dst, int i_dst, uint8_t *src, int i_src, int i_weight_dst ) 
  92. { 
  93.     x264_pixel_avg_weight_w ## W ## _##name( dst, i_dst, src, i_src, i_weight_dst, H ); 
  94. }
  96. AVG_WEIGHT(16,16,mmxext)
  97. AVG_WEIGHT(16,8,mmxext)
  98. AVG_WEIGHT(8,16,mmxext)
  99. AVG_WEIGHT(8,8,mmxext)
  100. AVG_WEIGHT(8,4,mmxext)
  101. AVG_WEIGHT(16,16,sse2)
  102. AVG_WEIGHT(16,8,sse2)
  103. AVG_WEIGHT(8,16,sse2)
  104. AVG_WEIGHT(8,8,sse2)
  105. AVG_WEIGHT(8,4,sse2)
  107. #define PIXEL_AVG_WTAB(instr, name1, name2, name3, name4, name5)
  108. static void (* const x264_pixel_avg_wtab_##instr[6])( uint8_t *, int, uint8_t *, int, uint8_t *, int ) =
  109. {
  110.     NULL,
  111.     x264_pixel_avg2_w4_##name1,
  112.     x264_pixel_avg2_w8_##name2,
  113.     x264_pixel_avg2_w12_##name3,
  114.     x264_pixel_avg2_w16_##name4,
  115.     x264_pixel_avg2_w20_##name5,
  116. };
  118. /* w16 sse2 is faster than w12 mmx as long as the cacheline issue is resolved */
  119. #define x264_pixel_avg2_w12_cache64_sse2 x264_pixel_avg2_w16_cache64_sse2
  120. #define x264_pixel_avg2_w12_sse3         x264_pixel_avg2_w16_sse3
  122. PIXEL_AVG_WTAB(mmxext, mmxext, mmxext, mmxext, mmxext, mmxext)
  123. #ifdef ARCH_X86
  124. PIXEL_AVG_WTAB(cache32_mmxext, mmxext, cache32_mmxext, cache32_mmxext, cache32_mmxext, cache32_mmxext)
  125. #endif
  126. PIXEL_AVG_WTAB(cache64_mmxext, mmxext, cache64_mmxext, cache64_mmxext, cache64_mmxext, cache64_mmxext)
  127. PIXEL_AVG_WTAB(sse2, mmxext, mmxext, mmxext, sse2, sse2)
  128. PIXEL_AVG_WTAB(cache64_sse2, mmxext, cache64_mmxext, cache64_sse2, cache64_sse2, cache64_sse2)
  130. #define MC_COPY_WTAB(instr, name1, name2, name3)
  131. static void (* const x264_mc_copy_wtab_##instr[5])( uint8_t *, int, uint8_t *, int, int ) =
  132. {
  133.     NULL,
  134.     x264_mc_copy_w4_##name1,
  135.     x264_mc_copy_w8_##name2,
  136.     NULL,
  137.     x264_mc_copy_w16_##name3,
  138. };
  140. MC_COPY_WTAB(mmx,mmx,mmx,mmx)
  141. MC_COPY_WTAB(sse2,mmx,mmx,sse2)
  143. static const int hpel_ref0[16] = {0,1,1,1,0,1,1,1,2,3,3,3,0,1,1,1};
  144. static const int hpel_ref1[16] = {0,0,0,0,2,2,3,2,2,2,3,2,2,2,3,2};
  146. #define MC_LUMA(name,instr1,instr2)
  147. static void mc_luma_##name( uint8_t *dst,    int i_dst_stride,
  148.                   uint8_t *src[4], int i_src_stride,
  149.                   int mvx, int mvy,
  150.                   int i_width, int i_height )
  151. {
  152.     int qpel_idx = ((mvy&3)<<2) + (mvx&3);
  153.     int offset = (mvy>>2)*i_src_stride + (mvx>>2);
  154.     uint8_t *src1 = src[hpel_ref0[qpel_idx]] + offset + ((mvy&3) == 3) * i_src_stride;
  155.     if( qpel_idx & 5 ) /* qpel interpolation needed */
  156.     {
  157.         uint8_t *src2 = src[hpel_ref1[qpel_idx]] + offset + ((mvx&3) == 3);
  158.         x264_pixel_avg_wtab_##instr1[i_width>>2](
  159.                 dst, i_dst_stride, src1, i_src_stride,
  160.                 src2, i_height );
  161.     }
  162.     else
  163.     {
  164.         x264_mc_copy_wtab_##instr2[i_width>>2](
  165.                 dst, i_dst_stride, src1, i_src_stride, i_height );
  166.     }
  167. }
  169. MC_LUMA(mmxext,mmxext,mmx)
  170. #ifdef ARCH_X86
  171. MC_LUMA(cache32_mmxext,cache32_mmxext,mmx)
  172. MC_LUMA(cache64_mmxext,cache64_mmxext,mmx)
  173. #endif
  174. MC_LUMA(sse2,sse2,sse2)
  175. MC_LUMA(cache64_sse2,cache64_sse2,sse2)
  177. #define GET_REF(name)
  178. static uint8_t *get_ref_##name( uint8_t *dst,   int *i_dst_stride,
  179.                          uint8_t *src[4], int i_src_stride,
  180.                          int mvx, int mvy,
  181.                          int i_width, int i_height )
  182. {
  183.     int qpel_idx = ((mvy&3)<<2) + (mvx&3);
  184.     int offset = (mvy>>2)*i_src_stride + (mvx>>2);
  185.     uint8_t *src1 = src[hpel_ref0[qpel_idx]] + offset + ((mvy&3) == 3) * i_src_stride;
  186.     if( qpel_idx & 5 ) /* qpel interpolation needed */
  187.     {
  188.         uint8_t *src2 = src[hpel_ref1[qpel_idx]] + offset + ((mvx&3) == 3);
  189.         x264_pixel_avg_wtab_##name[i_width>>2](
  190.                 dst, *i_dst_stride, src1, i_src_stride,
  191.                 src2, i_height );
  192.         return dst;
  193.     }
  194.     else
  195.     {
  196.         *i_dst_stride = i_src_stride;
  197.         return src1;
  198.     }
  199. }
  201. GET_REF(mmxext)
  202. #ifdef ARCH_X86
  203. GET_REF(cache32_mmxext)
  204. GET_REF(cache64_mmxext)
  205. #endif
  206. GET_REF(sse2)
  207. GET_REF(cache64_sse2)
  209. #define HPEL(align, cpu, cpuv, cpuc, cpuh)
  210. void x264_hpel_filter_v_##cpuv( uint8_t *dst, uint8_t *src, int16_t *buf, int stride, int width);
  211. void x264_hpel_filter_c_##cpuc( uint8_t *dst, int16_t *buf, int width );
  212. void x264_hpel_filter_h_##cpuh( uint8_t *dst, uint8_t *src, int width );
  213. void x264_sfence( void );
  214. static void x264_hpel_filter_##cpu( uint8_t *dsth, uint8_t *dstv, uint8_t *dstc, uint8_t *src,
  215.                              int stride, int width, int height )
  216. {
  217.     int16_t *buf;
  218.     int realign = (long)src & (align-1);
  219.     src -= realign;
  220.     dstv -= realign;
  221.     dstc -= realign;
  222.     dsth -= realign;
  223.     width += realign;
  224.     buf = x264_malloc((width+16)*sizeof(int16_t));
  225.     while( height-- )
  226.     {
  227.         x264_hpel_filter_v_##cpuv( dstv, src, buf+8, stride, width );
  228.         x264_hpel_filter_c_##cpuc( dstc, buf+8, width );
  229.         x264_hpel_filter_h_##cpuh( dsth, src, width );
  230.         dsth += stride;
  231.         dstv += stride;
  232.         dstc += stride;
  233.         src  += stride;
  234.     }
  235.     x264_sfence();
  236.     x264_free(buf);
  237. }
  239. HPEL(8, mmxext, mmxext, mmxext, mmxext)
  240. HPEL(16, sse2_amd, mmxext, mmxext, sse2)
  241. HPEL(16, sse2, sse2, sse2, sse2)
  242. HPEL(16, ssse3, sse2, ssse3, sse2)
  244. void x264_mc_init_mmx( int cpu, x264_mc_functions_t *pf )
  245. {
  246.     if( !(cpu&X264_CPU_MMX) )
  247.         return;
  249.     pf->copy[PIXEL_16x16] = x264_mc_copy_w16_mmx;
  250.     pf->copy[PIXEL_8x8]   = x264_mc_copy_w8_mmx;
  251.     pf->copy[PIXEL_4x4]   = x264_mc_copy_w4_mmx;
  252.     pf->memcpy_aligned = x264_memcpy_aligned_mmx;
  253.     pf->memzero_aligned = x264_memzero_aligned_mmx;
  255.     if( !(cpu&X264_CPU_MMXEXT) )
  256.         return;
  258.     pf->mc_luma = mc_luma_mmxext;
  259.     pf->get_ref = get_ref_mmxext;
  260.     pf->mc_chroma = x264_mc_chroma_mmxext;
  262.     pf->avg[PIXEL_16x16] = x264_pixel_avg_16x16_mmxext;
  263.     pf->avg[PIXEL_16x8]  = x264_pixel_avg_16x8_mmxext;
  264.     pf->avg[PIXEL_8x16]  = x264_pixel_avg_8x16_mmxext;
  265.     pf->avg[PIXEL_8x8]   = x264_pixel_avg_8x8_mmxext;
  266.     pf->avg[PIXEL_8x4]   = x264_pixel_avg_8x4_mmxext;
  267.     pf->avg[PIXEL_4x8]   = x264_pixel_avg_4x8_mmxext;
  268.     pf->avg[PIXEL_4x4]   = x264_pixel_avg_4x4_mmxext;
  269.     pf->avg[PIXEL_4x2]   = x264_pixel_avg_4x2_mmxext;
  271.     pf->avg_weight[PIXEL_16x16] = x264_pixel_avg_weight_16x16_mmxext;
  272.     pf->avg_weight[PIXEL_16x8]  = x264_pixel_avg_weight_16x8_mmxext;
  273.     pf->avg_weight[PIXEL_8x16]  = x264_pixel_avg_weight_8x16_mmxext;
  274.     pf->avg_weight[PIXEL_8x8]   = x264_pixel_avg_weight_8x8_mmxext;
  275.     pf->avg_weight[PIXEL_8x4]   = x264_pixel_avg_weight_8x4_mmxext;
  276.     pf->avg_weight[PIXEL_4x4]   = x264_pixel_avg_weight_4x4_mmxext;
  277.     // avg_weight_4x8 is rare and 4x2 is not used
  279.     pf->plane_copy = x264_plane_copy_mmxext;
  280.     pf->hpel_filter = x264_hpel_filter_mmxext;
  281.     pf->frame_init_lowres_core = x264_frame_init_lowres_core_mmxext;
  283.     pf->prefetch_fenc = x264_prefetch_fenc_mmxext;
  284.     pf->prefetch_ref  = x264_prefetch_ref_mmxext;
  286. #ifdef ARCH_X86 // all x86_64 cpus with cacheline split issues use sse2 instead
  287.     if( cpu&X264_CPU_CACHELINE_32 )
  288.     {
  289.         pf->mc_luma = mc_luma_cache32_mmxext;
  290.         pf->get_ref = get_ref_cache32_mmxext;
  291.         pf->frame_init_lowres_core = x264_frame_init_lowres_core_cache32_mmxext;
  292.     }
  293.     else if( cpu&X264_CPU_CACHELINE_64 )
  294.     {
  295.         pf->mc_luma = mc_luma_cache64_mmxext;
  296.         pf->get_ref = get_ref_cache64_mmxext;
  297.         pf->frame_init_lowres_core = x264_frame_init_lowres_core_cache32_mmxext;
  298.     }
  299. #endif
  301.     if( !(cpu&X264_CPU_SSE2) )
  302.         return;
  304.     pf->memcpy_aligned = x264_memcpy_aligned_sse2;
  305.     pf->memzero_aligned = x264_memzero_aligned_sse2;
  306.     pf->hpel_filter = x264_hpel_filter_sse2_amd;
  308.     if( cpu&X264_CPU_SSE2_IS_SLOW )
  309.         return;
  311.     pf->copy[PIXEL_16x16] = x264_mc_copy_w16_aligned_sse2;
  312.     pf->avg[PIXEL_16x16] = x264_pixel_avg_16x16_sse2;
  313.     pf->avg[PIXEL_16x8]  = x264_pixel_avg_16x8_sse2;
  314.     if( !(cpu&X264_CPU_STACK_MOD4) )
  315.     {
  316.         pf->avg_weight[PIXEL_16x16] = x264_pixel_avg_weight_16x16_sse2;
  317.         pf->avg_weight[PIXEL_16x8]  = x264_pixel_avg_weight_16x8_sse2;
  318.         pf->avg_weight[PIXEL_8x16]  = x264_pixel_avg_weight_8x16_sse2;
  319.         pf->avg_weight[PIXEL_8x8]   = x264_pixel_avg_weight_8x8_sse2;
  320.         pf->avg_weight[PIXEL_8x4]   = x264_pixel_avg_weight_8x4_sse2;
  321.     }
  322.     pf->hpel_filter = x264_hpel_filter_sse2;
  323.     pf->frame_init_lowres_core = x264_frame_init_lowres_core_sse2;
  324.     pf->mc_chroma = x264_mc_chroma_sse2;
  326.     if( cpu&X264_CPU_SSE2_IS_FAST )
  327.     {
  328.         pf->mc_luma = mc_luma_sse2;
  329.         pf->get_ref = get_ref_sse2;
  330.         if( cpu&X264_CPU_CACHELINE_64 )
  331.         {
  332.             pf->mc_luma = mc_luma_cache64_sse2;
  333.             pf->get_ref = get_ref_cache64_sse2;
  334.         }
  335.     }
  337.     if( !(cpu&X264_CPU_SSSE3) )
  338.         return;
  340.     pf->hpel_filter = x264_hpel_filter_ssse3;
  341.     pf->frame_init_lowres_core = x264_frame_init_lowres_core_ssse3;
  342.     pf->mc_chroma = x264_mc_chroma_ssse3;
  343. }