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

流媒体/Mpeg4/MP4

开发平台:

Visual 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.  *****************************************************************************/
  24. #include <stdlib.h>
  25. #include <stdio.h>
  26. #include <string.h>
  27. #include "common/common.h"
  28. #include "mc.h"
  29. #define DECL_SUF( func, args )
  30.     void func##_mmxext args;
  31.     void func##_sse2 args;
  32.     void func##_ssse3 args;
  33. DECL_SUF( x264_pixel_avg_16x16, ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  34. DECL_SUF( x264_pixel_avg_16x8,  ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  35. DECL_SUF( x264_pixel_avg_8x16,  ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  36. DECL_SUF( x264_pixel_avg_8x8,   ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  37. DECL_SUF( x264_pixel_avg_8x4,   ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  38. DECL_SUF( x264_pixel_avg_4x8,   ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  39. DECL_SUF( x264_pixel_avg_4x4,   ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  40. DECL_SUF( x264_pixel_avg_4x2,   ( uint8_t *, int, uint8_t *, int, uint8_t *, int, int ))
  41. extern void x264_mc_copy_w4_mmx( uint8_t *, int, uint8_t *, int, int );
  42. extern void x264_mc_copy_w8_mmx( uint8_t *, int, uint8_t *, int, int );
  43. extern void x264_mc_copy_w16_mmx( uint8_t *, int, uint8_t *, int, int );
  44. extern void x264_mc_copy_w16_sse2( uint8_t *, int, uint8_t *, int, int );
  45. extern void x264_mc_copy_w16_sse3( uint8_t *, int, uint8_t *, int, int );
  46. extern void x264_mc_copy_w16_aligned_sse2( uint8_t *, int, uint8_t *, int, int );
  47. extern void x264_prefetch_fenc_mmxext( uint8_t *, int, uint8_t *, int, int );
  48. extern void x264_prefetch_ref_mmxext( uint8_t *, int, int );
  49. extern void x264_mc_chroma_mmxext( uint8_t *src, int i_src_stride,
  50.                                    uint8_t *dst, int i_dst_stride,
  51.                                    int dx, int dy, int i_width, int i_height );
  52. extern void x264_mc_chroma_sse2( uint8_t *src, int i_src_stride,
  53.                                  uint8_t *dst, int i_dst_stride,
  54.                                  int dx, int dy, int i_width, int i_height );
  55. extern void x264_mc_chroma_ssse3( uint8_t *src, int i_src_stride,
  56.                                   uint8_t *dst, int i_dst_stride,
  57.                                   int dx, int dy, int i_width, int i_height );
  58. extern void x264_mc_chroma_ssse3_cache64( uint8_t *src, int i_src_stride,
  59.                                   uint8_t *dst, int i_dst_stride,
  60.                                   int dx, int dy, int i_width, int i_height );
  61. extern void x264_plane_copy_mmxext( uint8_t *, int, uint8_t *, int, int w, int h);
  62. extern void *x264_memcpy_aligned_mmx( void * dst, const void * src, size_t n );
  63. extern void *x264_memcpy_aligned_sse2( void * dst, const void * src, size_t n );
  64. extern void x264_memzero_aligned_mmx( void * dst, int n );
  65. extern void x264_memzero_aligned_sse2( void * dst, int n );
  66. extern void x264_integral_init4h_sse4( uint16_t *sum, uint8_t *pix, int stride );
  67. extern void x264_integral_init8h_sse4( uint16_t *sum, uint8_t *pix, int stride );
  68. extern void x264_integral_init4v_mmx( uint16_t *sum8, uint16_t *sum4, int stride );
  69. extern void x264_integral_init4v_sse2( uint16_t *sum8, uint16_t *sum4, int stride );
  70. extern void x264_integral_init8v_mmx( uint16_t *sum8, int stride );
  71. extern void x264_integral_init8v_sse2( uint16_t *sum8, int stride );
  72. extern void x264_integral_init4v_ssse3( uint16_t *sum8, uint16_t *sum4, int stride );
  73. extern void x264_mbtree_propagate_cost_sse2( int *dst, uint16_t *propagate_in, uint16_t *intra_costs,
  74.                                              uint16_t *inter_costs, uint16_t *inv_qscales, int len );
  75. #define LOWRES(cpu) 
  76. extern void x264_frame_init_lowres_core_##cpu( uint8_t *src0, uint8_t *dst0, uint8_t *dsth, uint8_t *dstv, uint8_t *dstc,
  77.                                                int src_stride, int dst_stride, int width, int height );
  78. LOWRES(mmxext)
  79. LOWRES(cache32_mmxext)
  80. LOWRES(sse2)
  81. LOWRES(ssse3)
  82. #define PIXEL_AVG_W(width,cpu)
  83. extern void x264_pixel_avg2_w##width##_##cpu( uint8_t *, int, uint8_t *, int, uint8_t *, int );
  84. /* This declares some functions that don't exist, but that isn't a problem. */
  85. #define PIXEL_AVG_WALL(cpu)
  86. 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);
  87. PIXEL_AVG_WALL(mmxext)
  88. PIXEL_AVG_WALL(cache32_mmxext)
  89. PIXEL_AVG_WALL(cache64_mmxext)
  90. PIXEL_AVG_WALL(cache64_sse2)
  91. PIXEL_AVG_WALL(sse2)
  92. PIXEL_AVG_WALL(sse2_misalign)
  93. PIXEL_AVG_WALL(cache64_ssse3)
  94. #define PIXEL_AVG_WTAB(instr, name1, name2, name3, name4, name5)
  95. static void (* const x264_pixel_avg_wtab_##instr[6])( uint8_t *, int, uint8_t *, int, uint8_t *, int ) =
  96. {
  97.     NULL,
  98.     x264_pixel_avg2_w4_##name1,
  99.     x264_pixel_avg2_w8_##name2,
  100.     x264_pixel_avg2_w12_##name3,
  101.     x264_pixel_avg2_w16_##name4,
  102.     x264_pixel_avg2_w20_##name5,
  103. };
  104. /* w16 sse2 is faster than w12 mmx as long as the cacheline issue is resolved */
  105. #define x264_pixel_avg2_w12_cache64_sse2 x264_pixel_avg2_w16_cache64_sse2
  106. #define x264_pixel_avg2_w12_sse3         x264_pixel_avg2_w16_sse3
  107. #define x264_pixel_avg2_w12_sse2         x264_pixel_avg2_w16_sse2
  108. PIXEL_AVG_WTAB(mmxext, mmxext, mmxext, mmxext, mmxext, mmxext)
  109. #ifdef ARCH_X86
  110. PIXEL_AVG_WTAB(cache32_mmxext, mmxext, cache32_mmxext, cache32_mmxext, cache32_mmxext, cache32_mmxext)
  111. PIXEL_AVG_WTAB(cache64_mmxext, mmxext, cache64_mmxext, cache64_mmxext, cache64_mmxext, cache64_mmxext)
  112. #endif
  113. PIXEL_AVG_WTAB(sse2, mmxext, mmxext, sse2, sse2, sse2)
  114. PIXEL_AVG_WTAB(sse2_misalign, mmxext, mmxext, sse2, sse2, sse2_misalign)
  115. PIXEL_AVG_WTAB(cache64_sse2, mmxext, cache64_mmxext, cache64_sse2, cache64_sse2, cache64_sse2)
  116. PIXEL_AVG_WTAB(cache64_ssse3, mmxext, cache64_mmxext, cache64_sse2, cache64_ssse3, cache64_sse2)
  117. #define MC_COPY_WTAB(instr, name1, name2, name3)
  118. static void (* const x264_mc_copy_wtab_##instr[5])( uint8_t *, int, uint8_t *, int, int ) =
  119. {
  120.     NULL,
  121.     x264_mc_copy_w4_##name1,
  122.     x264_mc_copy_w8_##name2,
  123.     NULL,
  124.     x264_mc_copy_w16_##name3,
  125. };
  126. MC_COPY_WTAB(mmx,mmx,mmx,mmx)
  127. MC_COPY_WTAB(sse2,mmx,mmx,sse2)
  128. static const int hpel_ref0[16] = {0,1,1,1,0,1,1,1,2,3,3,3,0,1,1,1};
  129. static const int hpel_ref1[16] = {0,0,0,0,2,2,3,2,2,2,3,2,2,2,3,2};
  130. #define MC_LUMA(name,instr1,instr2)
  131. static void mc_luma_##name( uint8_t *dst,    int i_dst_stride,
  132.                   uint8_t *src[4], int i_src_stride,
  133.                   int mvx, int mvy,
  134.                   int i_width, int i_height )
  135. {
  136.     int qpel_idx = ((mvy&3)<<2) + (mvx&3);
  137.     int offset = (mvy>>2)*i_src_stride + (mvx>>2);
  138.     uint8_t *src1 = src[hpel_ref0[qpel_idx]] + offset + ((mvy&3) == 3) * i_src_stride;
  139.     if( qpel_idx & 5 ) /* qpel interpolation needed */
  140.     {
  141.         uint8_t *src2 = src[hpel_ref1[qpel_idx]] + offset + ((mvx&3) == 3);
  142.         x264_pixel_avg_wtab_##instr1[i_width>>2](
  143.                 dst, i_dst_stride, src1, i_src_stride,
  144.                 src2, i_height );
  145.     }
  146.     else
  147.     {
  148.         x264_mc_copy_wtab_##instr2[i_width>>2](
  149.                 dst, i_dst_stride, src1, i_src_stride, i_height );
  150.     }
  151. }
  152. MC_LUMA(mmxext,mmxext,mmx)
  153. #ifdef ARCH_X86
  154. MC_LUMA(cache32_mmxext,cache32_mmxext,mmx)
  155. MC_LUMA(cache64_mmxext,cache64_mmxext,mmx)
  156. #endif
  157. MC_LUMA(sse2,sse2,sse2)
  158. MC_LUMA(cache64_sse2,cache64_sse2,sse2)
  159. MC_LUMA(cache64_ssse3,cache64_ssse3,sse2)
  160. #define GET_REF(name)
  161. static uint8_t *get_ref_##name( uint8_t *dst,   int *i_dst_stride,
  162.                          uint8_t *src[4], int i_src_stride,
  163.                          int mvx, int mvy,
  164.                          int i_width, int i_height )
  165. {
  166.     int qpel_idx = ((mvy&3)<<2) + (mvx&3);
  167.     int offset = (mvy>>2)*i_src_stride + (mvx>>2);
  168.     uint8_t *src1 = src[hpel_ref0[qpel_idx]] + offset + ((mvy&3) == 3) * i_src_stride;
  169.     if( qpel_idx & 5 ) /* qpel interpolation needed */
  170.     {
  171.         uint8_t *src2 = src[hpel_ref1[qpel_idx]] + offset + ((mvx&3) == 3);
  172.         x264_pixel_avg_wtab_##name[i_width>>2](
  173.                 dst, *i_dst_stride, src1, i_src_stride,
  174.                 src2, i_height );
  175.         return dst;
  176.     }
  177.     else
  178.     {
  179.         *i_dst_stride = i_src_stride;
  180.         return src1;
  181.     }
  182. }
  183. GET_REF(mmxext)
  184. #ifdef ARCH_X86
  185. GET_REF(cache32_mmxext)
  186. GET_REF(cache64_mmxext)
  187. #endif
  188. GET_REF(sse2)
  189. GET_REF(sse2_misalign)
  190. GET_REF(cache64_sse2)
  191. GET_REF(cache64_ssse3)
  192. #define HPEL(align, cpu, cpuv, cpuc, cpuh)
  193. void x264_hpel_filter_v_##cpuv( uint8_t *dst, uint8_t *src, int16_t *buf, int stride, int width);
  194. void x264_hpel_filter_c_##cpuc( uint8_t *dst, int16_t *buf, int width );
  195. void x264_hpel_filter_h_##cpuh( uint8_t *dst, uint8_t *src, int width );
  196. void x264_sfence( void );
  197. static void x264_hpel_filter_##cpu( uint8_t *dsth, uint8_t *dstv, uint8_t *dstc, uint8_t *src,
  198.                              int stride, int width, int height, int16_t *buf )
  199. {
  200.     int realign = (intptr_t)src & (align-1);
  201.     src -= realign;
  202.     dstv -= realign;
  203.     dstc -= realign;
  204.     dsth -= realign;
  205.     width += realign;
  206.     while( height-- )
  207.     {
  208.         x264_hpel_filter_v_##cpuv( dstv, src, buf+8, stride, width );
  209.         x264_hpel_filter_c_##cpuc( dstc, buf+8, width );
  210.         x264_hpel_filter_h_##cpuh( dsth, src, width );
  211.         dsth += stride;
  212.         dstv += stride;
  213.         dstc += stride;
  214.         src  += stride;
  215.     }
  216.     x264_sfence();
  217. }
  218. HPEL(8, mmxext, mmxext, mmxext, mmxext)
  219. HPEL(16, sse2_amd, mmxext, mmxext, sse2)
  220. #ifdef ARCH_X86_64
  221. void x264_hpel_filter_sse2( uint8_t *dsth, uint8_t *dstv, uint8_t *dstc, uint8_t *src, int stride, int width, int height, int16_t *buf );
  222. void x264_hpel_filter_ssse3( uint8_t *dsth, uint8_t *dstv, uint8_t *dstc, uint8_t *src, int stride, int width, int height, int16_t *buf );
  223. #else
  224. HPEL(16, sse2, sse2, sse2, sse2)
  225. HPEL(16, ssse3, ssse3, ssse3, ssse3)
  226. #endif
  227. HPEL(16, sse2_misalign, sse2, sse2_misalign, sse2)
  228. void x264_mc_init_mmx( int cpu, x264_mc_functions_t *pf )
  229. {
  230.     if( !(cpu&X264_CPU_MMX) )
  231.         return;
  232.     pf->copy_16x16_unaligned = x264_mc_copy_w16_mmx;
  233.     pf->copy[PIXEL_16x16] = x264_mc_copy_w16_mmx;
  234.     pf->copy[PIXEL_8x8]   = x264_mc_copy_w8_mmx;
  235.     pf->copy[PIXEL_4x4]   = x264_mc_copy_w4_mmx;
  236.     pf->memcpy_aligned = x264_memcpy_aligned_mmx;
  237.     pf->memzero_aligned = x264_memzero_aligned_mmx;
  238.     pf->integral_init4v = x264_integral_init4v_mmx;
  239.     pf->integral_init8v = x264_integral_init8v_mmx;
  240.     if( !(cpu&X264_CPU_MMXEXT) )
  241.         return;
  242.     pf->mc_luma = mc_luma_mmxext;
  243.     pf->get_ref = get_ref_mmxext;
  244.     pf->mc_chroma = x264_mc_chroma_mmxext;
  245.     pf->avg[PIXEL_16x16] = x264_pixel_avg_16x16_mmxext;
  246.     pf->avg[PIXEL_16x8]  = x264_pixel_avg_16x8_mmxext;
  247.     pf->avg[PIXEL_8x16]  = x264_pixel_avg_8x16_mmxext;
  248.     pf->avg[PIXEL_8x8]   = x264_pixel_avg_8x8_mmxext;
  249.     pf->avg[PIXEL_8x4]   = x264_pixel_avg_8x4_mmxext;
  250.     pf->avg[PIXEL_4x8]   = x264_pixel_avg_4x8_mmxext;
  251.     pf->avg[PIXEL_4x4]   = x264_pixel_avg_4x4_mmxext;
  252.     pf->avg[PIXEL_4x2]   = x264_pixel_avg_4x2_mmxext;
  253.     pf->plane_copy = x264_plane_copy_mmxext;
  254.     pf->hpel_filter = x264_hpel_filter_mmxext;
  255.     pf->frame_init_lowres_core = x264_frame_init_lowres_core_mmxext;
  256.     pf->prefetch_fenc = x264_prefetch_fenc_mmxext;
  257.     pf->prefetch_ref  = x264_prefetch_ref_mmxext;
  258. #ifdef ARCH_X86 // all x86_64 cpus with cacheline split issues use sse2 instead
  259.     if( cpu&X264_CPU_CACHELINE_32 )
  260.     {
  261.         pf->mc_luma = mc_luma_cache32_mmxext;
  262.         pf->get_ref = get_ref_cache32_mmxext;
  263.         pf->frame_init_lowres_core = x264_frame_init_lowres_core_cache32_mmxext;
  264.     }
  265.     else if( cpu&X264_CPU_CACHELINE_64 )
  266.     {
  267.         pf->mc_luma = mc_luma_cache64_mmxext;
  268.         pf->get_ref = get_ref_cache64_mmxext;
  269.         pf->frame_init_lowres_core = x264_frame_init_lowres_core_cache32_mmxext;
  270.     }
  271. #endif
  272.     if( !(cpu&X264_CPU_SSE2) )
  273.         return;
  274.     pf->memcpy_aligned = x264_memcpy_aligned_sse2;
  275.     pf->memzero_aligned = x264_memzero_aligned_sse2;
  276.     pf->integral_init4v = x264_integral_init4v_sse2;
  277.     pf->integral_init8v = x264_integral_init8v_sse2;
  278.     pf->hpel_filter = x264_hpel_filter_sse2_amd;
  279.     pf->mbtree_propagate_cost = x264_mbtree_propagate_cost_sse2;
  280.     if( cpu&X264_CPU_SSE2_IS_SLOW )
  281.         return;
  282.     pf->copy[PIXEL_16x16] = x264_mc_copy_w16_aligned_sse2;
  283.     pf->avg[PIXEL_16x16] = x264_pixel_avg_16x16_sse2;
  284.     pf->avg[PIXEL_16x8]  = x264_pixel_avg_16x8_sse2;
  285.     pf->avg[PIXEL_8x16] = x264_pixel_avg_8x16_sse2;
  286.     pf->avg[PIXEL_8x8]  = x264_pixel_avg_8x8_sse2;
  287.     pf->avg[PIXEL_8x4]  = x264_pixel_avg_8x4_sse2;
  288.     pf->hpel_filter = x264_hpel_filter_sse2;
  289.     if( cpu&X264_CPU_SSE_MISALIGN )
  290.         pf->hpel_filter = x264_hpel_filter_sse2_misalign;
  291.     pf->frame_init_lowres_core = x264_frame_init_lowres_core_sse2;
  292.     pf->mc_chroma = x264_mc_chroma_sse2;
  293.     if( cpu&X264_CPU_SSE2_IS_FAST )
  294.     {
  295.         pf->mc_luma = mc_luma_sse2;
  296.         pf->get_ref = get_ref_sse2;
  297.         if( cpu&X264_CPU_CACHELINE_64 )
  298.         {
  299.             pf->mc_luma = mc_luma_cache64_sse2;
  300.             pf->get_ref = get_ref_cache64_sse2;
  301.         }
  302.         if( cpu&X264_CPU_SSE_MISALIGN )
  303.             pf->get_ref = get_ref_sse2_misalign;
  304.     }
  305.     if( !(cpu&X264_CPU_SSSE3) )
  306.         return;
  307.     pf->avg[PIXEL_16x16] = x264_pixel_avg_16x16_ssse3;
  308.     pf->avg[PIXEL_16x8]  = x264_pixel_avg_16x8_ssse3;
  309.     pf->avg[PIXEL_8x16]  = x264_pixel_avg_8x16_ssse3;
  310.     pf->avg[PIXEL_8x8]   = x264_pixel_avg_8x8_ssse3;
  311.     pf->avg[PIXEL_8x4]   = x264_pixel_avg_8x4_ssse3;
  312.     pf->avg[PIXEL_4x8]   = x264_pixel_avg_4x8_ssse3;
  313.     pf->avg[PIXEL_4x4]   = x264_pixel_avg_4x4_ssse3;
  314.     pf->avg[PIXEL_4x2]   = x264_pixel_avg_4x2_ssse3;
  315.     pf->hpel_filter = x264_hpel_filter_ssse3;
  316.     pf->frame_init_lowres_core = x264_frame_init_lowres_core_ssse3;
  317.     pf->mc_chroma = x264_mc_chroma_ssse3;
  318.     if( cpu&X264_CPU_CACHELINE_64 )
  319.     {
  320.         pf->mc_chroma = x264_mc_chroma_ssse3_cache64;
  321.         pf->mc_luma = mc_luma_cache64_ssse3;
  322.         pf->get_ref = get_ref_cache64_ssse3;
  323.     }
  324.     if( cpu&X264_CPU_SHUFFLE_IS_FAST )
  325.         pf->integral_init4v = x264_integral_init4v_ssse3;
  326.     if( !(cpu&X264_CPU_SSE4) )
  327.         return;
  328.     pf->integral_init4h = x264_integral_init4h_sse4;
  329.     pf->integral_init8h = x264_integral_init8h_sse4;
  330. }