falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75716 Commits
32 Branches
303MB
Tree: 26ece7a511
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '26ece7a511'
${ noResults }
FFmpeg/libavcodec/x86/vp9dsp_init_16bpp_template.c

190 lines
6.5KB
Raw Blame History 

  1. /*

  2.  * VP9 SIMD optimizations

  3.  *

  4.  * Copyright (c) 2013 Ronald S. Bultje <rsbultje gmail com>

  5.  *

  6.  * This file is part of FFmpeg.

  7.  *

  8.  * FFmpeg is free software; you can redistribute it and/or

  9.  * modify it under the terms of the GNU Lesser General Public

  10.  * License as published by the Free Software Foundation; either

  11.  * version 2.1 of the License, or (at your option) any later version.

  12.  *

  13.  * FFmpeg is distributed in the hope that it will be useful,

  14.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  15.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  16.  * Lesser General Public License for more details.

  17.  *

  18.  * You should have received a copy of the GNU Lesser General Public

  19.  * License along with FFmpeg; if not, write to the Free Software

  20.  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

  21.  */

  22. 

  23. #include "libavutil/attributes.h"

  24. #include "libavutil/cpu.h"

  25. #include "libavutil/mem.h"

  26. #include "libavutil/x86/cpu.h"

  27. #include "libavcodec/vp9dsp.h"

  28. #include "libavcodec/x86/vp9dsp_init.h"

  29. 

  30. #if HAVE_YASM

  31. 

  32. extern const int16_t ff_filters_16bpp[3][15][4][16];

  33. 

  34. decl_mc_funcs(4, sse2, int16_t, 16, BPC);

  35. decl_mc_funcs(8, sse2, int16_t, 16, BPC);

  36. decl_mc_funcs(16, avx2, int16_t, 16, BPC);

  37. 

  38. mc_rep_funcs(16,  8, 16, sse2, int16_t, 16, BPC);

  39. mc_rep_funcs(32, 16, 32, sse2, int16_t, 16, BPC);

  40. mc_rep_funcs(64, 32, 64, sse2, int16_t, 16, BPC);

  41. #if HAVE_AVX2_EXTERNAL

  42. mc_rep_funcs(32, 16, 32, avx2, int16_t, 16, BPC);

  43. mc_rep_funcs(64, 32, 64, avx2, int16_t, 16, BPC);

  44. #endif

  45. 

  46. filters_8tap_2d_fn2(put, 16, BPC, 2, sse2, sse2, 16bpp)

  47. filters_8tap_2d_fn2(avg, 16, BPC, 2, sse2, sse2, 16bpp)

  48. #if HAVE_AVX2_EXTERNAL

  49. filters_8tap_2d_fn(put, 64, 32, BPC, 2, avx2, 16bpp)

  50. filters_8tap_2d_fn(avg, 64, 32, BPC, 2, avx2, 16bpp)

  51. filters_8tap_2d_fn(put, 32, 32, BPC, 2, avx2, 16bpp)

  52. filters_8tap_2d_fn(avg, 32, 32, BPC, 2, avx2, 16bpp)

  53. filters_8tap_2d_fn(put, 16, 32, BPC, 2, avx2, 16bpp)

  54. filters_8tap_2d_fn(avg, 16, 32, BPC, 2, avx2, 16bpp)

  55. #endif

  56. 

  57. filters_8tap_1d_fn3(put, BPC, sse2, sse2, 16bpp)

  58. filters_8tap_1d_fn3(avg, BPC, sse2, sse2, 16bpp)

  59. #if HAVE_AVX2_EXTERNAL

  60. filters_8tap_1d_fn2(put, 64, BPC, avx2, 16bpp)

  61. filters_8tap_1d_fn2(avg, 64, BPC, avx2, 16bpp)

  62. filters_8tap_1d_fn2(put, 32, BPC, avx2, 16bpp)

  63. filters_8tap_1d_fn2(avg, 32, BPC, avx2, 16bpp)

  64. filters_8tap_1d_fn2(put, 16, BPC, avx2, 16bpp)

  65. filters_8tap_1d_fn2(avg, 16, BPC, avx2, 16bpp)

  66. #endif

  67. 

  68. #define decl_lpf_func(dir, wd, bpp, opt) \

  69. void ff_vp9_loop_filter_##dir##_##wd##_##bpp##_##opt(uint8_t *dst, ptrdiff_t stride, \

  70.                                                      int E, int I, int H)

  71. 

  72. #define decl_lpf_funcs(dir, wd, bpp) \

  73. decl_lpf_func(dir, wd, bpp, sse2); \

  74. decl_lpf_func(dir, wd, bpp, ssse3); \

  75. decl_lpf_func(dir, wd, bpp, avx)

  76. 

  77. #define decl_lpf_funcs_wd(dir) \

  78. decl_lpf_funcs(dir,  4, BPC); \

  79. decl_lpf_funcs(dir,  8, BPC); \

  80. decl_lpf_funcs(dir, 16, BPC)

  81. 

  82. decl_lpf_funcs_wd(h);

  83. decl_lpf_funcs_wd(v);

  84. 

  85. #define lpf_16_wrapper(dir, off, bpp, opt) \

  86. static void loop_filter_##dir##_16_##bpp##_##opt(uint8_t *dst, ptrdiff_t stride, \

  87.                                                  int E, int I, int H) \

  88. { \

  89.     ff_vp9_loop_filter_##dir##_16_##bpp##_##opt(dst,       stride, E, I, H); \

  90.     ff_vp9_loop_filter_##dir##_16_##bpp##_##opt(dst + off, stride, E, I, H); \

  91. }

  92. 

  93. #define lpf_16_wrappers(bpp, opt) \

  94. lpf_16_wrapper(h, 8 * stride, bpp, opt); \

  95. lpf_16_wrapper(v, 16,         bpp, opt)

  96. 

  97. lpf_16_wrappers(BPC, sse2);

  98. lpf_16_wrappers(BPC, ssse3);

  99. lpf_16_wrappers(BPC, avx);

  100. 

  101. #define lpf_mix2_wrapper(dir, off, wd1, wd2, bpp, opt) \

  102. static void loop_filter_##dir##_##wd1##wd2##_##bpp##_##opt(uint8_t *dst, ptrdiff_t stride, \

  103.                                                            int E, int I, int H) \

  104. { \

  105.     ff_vp9_loop_filter_##dir##_##wd1##_##bpp##_##opt(dst,       stride, \

  106.                                                      E & 0xff, I & 0xff, H & 0xff); \

  107.     ff_vp9_loop_filter_##dir##_##wd2##_##bpp##_##opt(dst + off, stride, \

  108.                                                      E >> 8,   I >> 8,   H >> 8); \

  109. }

  110. 

  111. #define lpf_mix2_wrappers(wd1, wd2, bpp, opt) \

  112. lpf_mix2_wrapper(h, 8 * stride, wd1, wd2, bpp, opt); \

  113. lpf_mix2_wrapper(v, 16,         wd1, wd2, bpp, opt)

  114. 

  115. #define lpf_mix2_wrappers_set(bpp, opt) \

  116. lpf_mix2_wrappers(4, 4, bpp, opt); \

  117. lpf_mix2_wrappers(4, 8, bpp, opt); \

  118. lpf_mix2_wrappers(8, 4, bpp, opt); \

  119. lpf_mix2_wrappers(8, 8, bpp, opt); \

  120. 

  121. lpf_mix2_wrappers_set(BPC, sse2);

  122. lpf_mix2_wrappers_set(BPC, ssse3);

  123. lpf_mix2_wrappers_set(BPC, avx);

  124. 

  125. decl_ipred_fns(tm, BPC, mmxext, sse2);

  126. #endif /* HAVE_YASM */

  127. 

  128. av_cold void INIT_FUNC(VP9DSPContext *dsp)

  129. {

  130. #if HAVE_YASM

  131.     int cpu_flags = av_get_cpu_flags();

  132. 

  133. #define init_lpf_8_func(idx1, idx2, dir, wd, bpp, opt) \

  134.     dsp->loop_filter_8[idx1][idx2] = ff_vp9_loop_filter_##dir##_##wd##_##bpp##_##opt

  135. #define init_lpf_16_func(idx, dir, bpp, opt) \

  136.     dsp->loop_filter_16[idx] = loop_filter_##dir##_16_##bpp##_##opt

  137. #define init_lpf_mix2_func(idx1, idx2, idx3, dir, wd1, wd2, bpp, opt) \

  138.     dsp->loop_filter_mix2[idx1][idx2][idx3] = loop_filter_##dir##_##wd1##wd2##_##bpp##_##opt

  139. 

  140. #define init_lpf_funcs(bpp, opt) \

  141.     init_lpf_8_func(0, 0, h,  4, bpp, opt); \

  142.     init_lpf_8_func(0, 1, v,  4, bpp, opt); \

  143.     init_lpf_8_func(1, 0, h,  8, bpp, opt); \

  144.     init_lpf_8_func(1, 1, v,  8, bpp, opt); \

  145.     init_lpf_8_func(2, 0, h, 16, bpp, opt); \

  146.     init_lpf_8_func(2, 1, v, 16, bpp, opt); \

  147.     init_lpf_16_func(0, h, bpp, opt); \

  148.     init_lpf_16_func(1, v, bpp, opt); \

  149.     init_lpf_mix2_func(0, 0, 0, h, 4, 4, bpp, opt); \

  150.     init_lpf_mix2_func(0, 1, 0, h, 4, 8, bpp, opt); \

  151.     init_lpf_mix2_func(1, 0, 0, h, 8, 4, bpp, opt); \

  152.     init_lpf_mix2_func(1, 1, 0, h, 8, 8, bpp, opt); \

  153.     init_lpf_mix2_func(0, 0, 1, v, 4, 4, bpp, opt); \

  154.     init_lpf_mix2_func(0, 1, 1, v, 4, 8, bpp, opt); \

  155.     init_lpf_mix2_func(1, 0, 1, v, 8, 4, bpp, opt); \

  156.     init_lpf_mix2_func(1, 1, 1, v, 8, 8, bpp, opt)

  157. 

  158.     if (EXTERNAL_MMXEXT(cpu_flags)) {

  159.         init_ipred_func(tm, TM_VP8, 4, BPC, mmxext);

  160.     }

  161. 

  162.     if (EXTERNAL_SSE2(cpu_flags)) {

  163.         init_subpel3(0, put, BPC, sse2);

  164.         init_subpel3(1, avg, BPC, sse2);

  165.         init_lpf_funcs(BPC, sse2);

  166.         init_8_16_32_ipred_funcs(tm, TM_VP8, BPC, sse2);

  167.     }

  168. 

  169.     if (EXTERNAL_SSSE3(cpu_flags)) {

  170.         init_lpf_funcs(BPC, ssse3);

  171.     }

  172. 

  173.     if (EXTERNAL_AVX(cpu_flags)) {

  174.         init_lpf_funcs(BPC, avx);

  175.     }

  176. 

  177.     if (EXTERNAL_AVX2(cpu_flags)) {

  178. #if HAVE_AVX2_EXTERNAL

  179.         init_subpel3_32_64(0,  put, BPC, avx2);

  180.         init_subpel3_32_64(1,  avg, BPC, avx2);

  181.         init_subpel2(2, 0, 16, put, BPC, avx2);

  182.         init_subpel2(2, 1, 16, avg, BPC, avx2);

  183. #endif

  184.     }

  185. 

  186. #endif /* HAVE_YASM */

  187. 

  188.     ff_vp9dsp_init_16bpp_x86(dsp);

  189. }