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.

642 lines
30KB

  1. /*
  2. * Copyright (c) 2004-2005 Michael Niedermayer, Loren Merritt
  3. * Copyright (c) 2011 Daniel Kang
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * Libav is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #include "libavutil/attributes.h"
  22. #include "libavutil/cpu.h"
  23. #include "libavutil/x86/asm.h"
  24. #include "libavutil/x86/cpu.h"
  25. #include "libavcodec/h264qpel.h"
  26. #include "libavcodec/mpegvideo.h"
  27. #include "dsputil_mmx.h"
  28. #if HAVE_YASM
  29. void ff_put_pixels4_mmxext(uint8_t *block, const uint8_t *pixels,
  30. ptrdiff_t line_size, int h);
  31. void ff_avg_pixels4_mmxext(uint8_t *block, const uint8_t *pixels,
  32. ptrdiff_t line_size, int h);
  33. static void ff_put_pixels16_mmxext(uint8_t *block, const uint8_t *pixels,
  34. ptrdiff_t line_size, int h)
  35. {
  36. ff_put_pixels8_mmxext(block, pixels, line_size, h);
  37. ff_put_pixels8_mmxext(block + 8, pixels + 8, line_size, h);
  38. }
  39. static void ff_avg_pixels16_mmxext(uint8_t *block, const uint8_t *pixels,
  40. ptrdiff_t line_size, int h)
  41. {
  42. ff_avg_pixels8_mmxext(block, pixels, line_size, h);
  43. ff_avg_pixels8_mmxext(block + 8, pixels + 8, line_size, h);
  44. }
  45. void ff_put_pixels4_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
  46. int dstStride, int src1Stride, int h);
  47. void ff_avg_pixels4_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
  48. int dstStride, int src1Stride, int h);
  49. void ff_put_pixels8_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
  50. int dstStride, int src1Stride, int h);
  51. void ff_avg_pixels8_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
  52. int dstStride, int src1Stride, int h);
  53. void ff_put_pixels16_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
  54. int dstStride, int src1Stride, int h);
  55. void ff_avg_pixels16_l2_mmxext(uint8_t *dst, uint8_t *src1, uint8_t *src2,
  56. int dstStride, int src1Stride, int h);
  57. #define ff_put_pixels8_l2_sse2 ff_put_pixels8_l2_mmxext
  58. #define ff_avg_pixels8_l2_sse2 ff_avg_pixels8_l2_mmxext
  59. #define ff_put_pixels16_l2_sse2 ff_put_pixels16_l2_mmxext
  60. #define ff_avg_pixels16_l2_sse2 ff_avg_pixels16_l2_mmxext
  61. #define DEF_QPEL(OPNAME)\
  62. void ff_ ## OPNAME ## _h264_qpel4_h_lowpass_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
  63. void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
  64. void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_ssse3(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
  65. void ff_ ## OPNAME ## _h264_qpel4_h_lowpass_l2_mmxext(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);\
  66. void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_l2_mmxext(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);\
  67. void ff_ ## OPNAME ## _h264_qpel8_h_lowpass_l2_ssse3(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);\
  68. void ff_ ## OPNAME ## _h264_qpel4_v_lowpass_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride);\
  69. void ff_ ## OPNAME ## _h264_qpel8or16_v_lowpass_op_mmxext(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h);\
  70. void ff_ ## OPNAME ## _h264_qpel8or16_v_lowpass_sse2(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h);\
  71. void ff_ ## OPNAME ## _h264_qpel4_hv_lowpass_v_mmxext(uint8_t *src, int16_t *tmp, int srcStride);\
  72. void ff_ ## OPNAME ## _h264_qpel4_hv_lowpass_h_mmxext(int16_t *tmp, uint8_t *dst, int dstStride);\
  73. void ff_ ## OPNAME ## _h264_qpel8or16_hv1_lowpass_op_mmxext(uint8_t *src, int16_t *tmp, int srcStride, int size);\
  74. void ff_ ## OPNAME ## _h264_qpel8or16_hv1_lowpass_op_sse2(uint8_t *src, int16_t *tmp, int srcStride, int size);\
  75. void ff_ ## OPNAME ## _h264_qpel8or16_hv2_lowpass_op_mmxext(uint8_t *dst, int16_t *tmp, int dstStride, int unused, int h);\
  76. void ff_ ## OPNAME ## _h264_qpel8or16_hv2_lowpass_ssse3(uint8_t *dst, int16_t *tmp, int dstStride, int tmpStride, int size);\
  77. void ff_ ## OPNAME ## _pixels4_l2_shift5_mmxext(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h);\
  78. void ff_ ## OPNAME ## _pixels8_l2_shift5_mmxext(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h);
  79. DEF_QPEL(avg)
  80. DEF_QPEL(put)
  81. #define QPEL_H264(OPNAME, OP, MMX)\
  82. static av_always_inline void ff_ ## OPNAME ## h264_qpel4_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  83. int w=3;\
  84. src -= 2*srcStride+2;\
  85. while(w--){\
  86. ff_ ## OPNAME ## h264_qpel4_hv_lowpass_v_mmxext(src, tmp, srcStride);\
  87. tmp += 4;\
  88. src += 4;\
  89. }\
  90. tmp -= 3*4;\
  91. ff_ ## OPNAME ## h264_qpel4_hv_lowpass_h_mmxext(tmp, dst, dstStride);\
  92. }\
  93. \
  94. static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
  95. src -= 2*srcStride;\
  96. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_op_mmxext(dst, src, dstStride, srcStride, h);\
  97. src += 4;\
  98. dst += 4;\
  99. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_op_mmxext(dst, src, dstStride, srcStride, h);\
  100. }\
  101. static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv1_lowpass_ ## MMX(int16_t *tmp, uint8_t *src, int tmpStride, int srcStride, int size){\
  102. int w = (size+8)>>2;\
  103. src -= 2*srcStride+2;\
  104. while(w--){\
  105. ff_ ## OPNAME ## h264_qpel8or16_hv1_lowpass_op_mmxext(src, tmp, srcStride, size);\
  106. tmp += 4;\
  107. src += 4;\
  108. }\
  109. }\
  110. static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, int dstStride, int tmpStride, int size){\
  111. int w = size>>4;\
  112. do{\
  113. ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_op_mmxext(dst, tmp, dstStride, 0, size);\
  114. tmp += 8;\
  115. dst += 8;\
  116. }while(w--);\
  117. }\
  118. \
  119. static av_always_inline void ff_ ## OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  120. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 8);\
  121. }\
  122. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  123. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 16);\
  124. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  125. }\
  126. \
  127. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  128. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  129. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  130. src += 8*srcStride;\
  131. dst += 8*dstStride;\
  132. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  133. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  134. }\
  135. \
  136. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  137. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  138. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  139. src += 8*dstStride;\
  140. dst += 8*dstStride;\
  141. src2 += 8*src2Stride;\
  142. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  143. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  144. }\
  145. \
  146. static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
  147. ff_put_h264_qpel8or16_hv1_lowpass_ ## MMX(tmp, src, tmpStride, srcStride, size);\
  148. ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_ ## MMX(dst, tmp, dstStride, tmpStride, size);\
  149. }\
  150. static av_always_inline void ff_ ## OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  151. ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst , tmp , src , dstStride, tmpStride, srcStride, 8);\
  152. }\
  153. \
  154. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  155. ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst , tmp , src , dstStride, tmpStride, srcStride, 16);\
  156. }\
  157. \
  158. static av_always_inline void ff_ ## OPNAME ## pixels16_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  159. {\
  160. ff_ ## OPNAME ## pixels8_l2_shift5_ ## MMX(dst , src16 , src8 , dstStride, src8Stride, h);\
  161. ff_ ## OPNAME ## pixels8_l2_shift5_ ## MMX(dst+8, src16+8, src8+8, dstStride, src8Stride, h);\
  162. }\
  163. #if ARCH_X86_64
  164. #define QPEL_H264_H16_XMM(OPNAME, OP, MMX)\
  165. void ff_avg_h264_qpel16_h_lowpass_l2_ssse3(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);
  166. void ff_put_h264_qpel16_h_lowpass_l2_ssse3(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride);
  167. #else // ARCH_X86_64
  168. #define QPEL_H264_H16_XMM(OPNAME, OP, MMX)\
  169. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  170. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  171. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  172. src += 8*dstStride;\
  173. dst += 8*dstStride;\
  174. src2 += 8*src2Stride;\
  175. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  176. ff_ ## OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  177. }
  178. #endif // ARCH_X86_64
  179. #define QPEL_H264_H_XMM(OPNAME, OP, MMX)\
  180. QPEL_H264_H16_XMM(OPNAME, OP, MMX)\
  181. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  182. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  183. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  184. src += 8*srcStride;\
  185. dst += 8*dstStride;\
  186. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  187. ff_ ## OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  188. }\
  189. #define QPEL_H264_V_XMM(OPNAME, OP, MMX)\
  190. static av_always_inline void ff_ ## OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  191. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 8);\
  192. }\
  193. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  194. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 16);\
  195. ff_ ## OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  196. }
  197. static av_always_inline void ff_put_h264_qpel8or16_hv1_lowpass_sse2(int16_t *tmp, uint8_t *src, int tmpStride, int srcStride, int size){
  198. int w = (size+8)>>3;
  199. src -= 2*srcStride+2;
  200. while(w--){
  201. ff_put_h264_qpel8or16_hv1_lowpass_op_sse2(src, tmp, srcStride, size);
  202. tmp += 8;
  203. src += 8;
  204. }
  205. }
  206. #define QPEL_H264_HV_XMM(OPNAME, OP, MMX)\
  207. static av_always_inline void ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
  208. ff_put_h264_qpel8or16_hv1_lowpass_sse2(tmp, src, tmpStride, srcStride, size);\
  209. ff_ ## OPNAME ## h264_qpel8or16_hv2_lowpass_ ## MMX(dst, tmp, dstStride, tmpStride, size);\
  210. }\
  211. static av_always_inline void ff_ ## OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  212. ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst, tmp, src, dstStride, tmpStride, srcStride, 8);\
  213. }\
  214. static av_always_inline void ff_ ## OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  215. ff_ ## OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst, tmp, src, dstStride, tmpStride, srcStride, 16);\
  216. }\
  217. #define ff_put_h264_qpel8_h_lowpass_l2_sse2 ff_put_h264_qpel8_h_lowpass_l2_mmxext
  218. #define ff_avg_h264_qpel8_h_lowpass_l2_sse2 ff_avg_h264_qpel8_h_lowpass_l2_mmxext
  219. #define ff_put_h264_qpel16_h_lowpass_l2_sse2 ff_put_h264_qpel16_h_lowpass_l2_mmxext
  220. #define ff_avg_h264_qpel16_h_lowpass_l2_sse2 ff_avg_h264_qpel16_h_lowpass_l2_mmxext
  221. #define ff_put_h264_qpel8_v_lowpass_ssse3 ff_put_h264_qpel8_v_lowpass_sse2
  222. #define ff_avg_h264_qpel8_v_lowpass_ssse3 ff_avg_h264_qpel8_v_lowpass_sse2
  223. #define ff_put_h264_qpel16_v_lowpass_ssse3 ff_put_h264_qpel16_v_lowpass_sse2
  224. #define ff_avg_h264_qpel16_v_lowpass_ssse3 ff_avg_h264_qpel16_v_lowpass_sse2
  225. #define ff_put_h264_qpel8or16_hv2_lowpass_sse2 ff_put_h264_qpel8or16_hv2_lowpass_mmxext
  226. #define ff_avg_h264_qpel8or16_hv2_lowpass_sse2 ff_avg_h264_qpel8or16_hv2_lowpass_mmxext
  227. #define H264_MC(OPNAME, SIZE, MMX, ALIGN) \
  228. H264_MC_C(OPNAME, SIZE, MMX, ALIGN)\
  229. H264_MC_V(OPNAME, SIZE, MMX, ALIGN)\
  230. H264_MC_H(OPNAME, SIZE, MMX, ALIGN)\
  231. H264_MC_HV(OPNAME, SIZE, MMX, ALIGN)\
  232. static void put_h264_qpel16_mc00_sse2 (uint8_t *dst, uint8_t *src,
  233. ptrdiff_t stride)
  234. {
  235. ff_put_pixels16_sse2(dst, src, stride, 16);
  236. }
  237. static void avg_h264_qpel16_mc00_sse2 (uint8_t *dst, uint8_t *src,
  238. ptrdiff_t stride)
  239. {
  240. ff_avg_pixels16_sse2(dst, src, stride, 16);
  241. }
  242. #define put_h264_qpel8_mc00_sse2 put_h264_qpel8_mc00_mmxext
  243. #define avg_h264_qpel8_mc00_sse2 avg_h264_qpel8_mc00_mmxext
  244. #define H264_MC_C(OPNAME, SIZE, MMX, ALIGN) \
  245. static void OPNAME ## h264_qpel ## SIZE ## _mc00_ ## MMX (uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  246. {\
  247. ff_ ## OPNAME ## pixels ## SIZE ## _ ## MMX(dst, src, stride, SIZE);\
  248. }\
  249. #define H264_MC_H(OPNAME, SIZE, MMX, ALIGN) \
  250. static void OPNAME ## h264_qpel ## SIZE ## _mc10_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  251. {\
  252. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src, stride, stride);\
  253. }\
  254. \
  255. static void OPNAME ## h264_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  256. {\
  257. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_ ## MMX(dst, src, stride, stride);\
  258. }\
  259. \
  260. static void OPNAME ## h264_qpel ## SIZE ## _mc30_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  261. {\
  262. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src+1, stride, stride);\
  263. }\
  264. #define H264_MC_V(OPNAME, SIZE, MMX, ALIGN) \
  265. static void OPNAME ## h264_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  266. {\
  267. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
  268. ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
  269. ff_ ## OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src, temp, stride, stride, SIZE);\
  270. }\
  271. \
  272. static void OPNAME ## h264_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  273. {\
  274. ff_ ## OPNAME ## h264_qpel ## SIZE ## _v_lowpass_ ## MMX(dst, src, stride, stride);\
  275. }\
  276. \
  277. static void OPNAME ## h264_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  278. {\
  279. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
  280. ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
  281. ff_ ## OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src+stride, temp, stride, stride, SIZE);\
  282. }\
  283. #define H264_MC_HV(OPNAME, SIZE, MMX, ALIGN) \
  284. static void OPNAME ## h264_qpel ## SIZE ## _mc11_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  285. {\
  286. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
  287. ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
  288. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, temp, stride, SIZE);\
  289. }\
  290. \
  291. static void OPNAME ## h264_qpel ## SIZE ## _mc31_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  292. {\
  293. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
  294. ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src+1, SIZE, stride);\
  295. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, temp, stride, SIZE);\
  296. }\
  297. \
  298. static void OPNAME ## h264_qpel ## SIZE ## _mc13_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  299. {\
  300. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
  301. ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src, SIZE, stride);\
  302. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, temp, stride, SIZE);\
  303. }\
  304. \
  305. static void OPNAME ## h264_qpel ## SIZE ## _mc33_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  306. {\
  307. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*SIZE];\
  308. ff_put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(temp, src+1, SIZE, stride);\
  309. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, temp, stride, SIZE);\
  310. }\
  311. \
  312. static void OPNAME ## h264_qpel ## SIZE ## _mc22_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  313. {\
  314. DECLARE_ALIGNED(ALIGN, uint16_t, temp)[SIZE*(SIZE<8?12:24)];\
  315. ff_ ## OPNAME ## h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(dst, temp, src, stride, SIZE, stride);\
  316. }\
  317. \
  318. static void OPNAME ## h264_qpel ## SIZE ## _mc21_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  319. {\
  320. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
  321. uint8_t * const halfHV= temp;\
  322. int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
  323. assert(((int)temp & 7) == 0);\
  324. ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  325. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfHV, stride, SIZE);\
  326. }\
  327. \
  328. static void OPNAME ## h264_qpel ## SIZE ## _mc23_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  329. {\
  330. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
  331. uint8_t * const halfHV= temp;\
  332. int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
  333. assert(((int)temp & 7) == 0);\
  334. ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  335. ff_ ## OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfHV, stride, SIZE);\
  336. }\
  337. \
  338. static void OPNAME ## h264_qpel ## SIZE ## _mc12_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  339. {\
  340. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
  341. uint8_t * const halfHV= temp;\
  342. int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
  343. assert(((int)temp & 7) == 0);\
  344. ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  345. ff_ ## OPNAME ## pixels ## SIZE ## _l2_shift5_mmxext(dst, halfV+2, halfHV, stride, SIZE, SIZE);\
  346. }\
  347. \
  348. static void OPNAME ## h264_qpel ## SIZE ## _mc32_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride)\
  349. {\
  350. DECLARE_ALIGNED(ALIGN, uint8_t, temp)[SIZE*(SIZE<8?12:24)*2 + SIZE*SIZE];\
  351. uint8_t * const halfHV= temp;\
  352. int16_t * const halfV= (int16_t*)(temp + SIZE*SIZE);\
  353. assert(((int)temp & 7) == 0);\
  354. ff_put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  355. ff_ ## OPNAME ## pixels ## SIZE ## _l2_shift5_mmxext(dst, halfV+3, halfHV, stride, SIZE, SIZE);\
  356. }\
  357. #define H264_MC_4816(MMX)\
  358. H264_MC(put_, 4, MMX, 8)\
  359. H264_MC(put_, 8, MMX, 8)\
  360. H264_MC(put_, 16,MMX, 8)\
  361. H264_MC(avg_, 4, MMX, 8)\
  362. H264_MC(avg_, 8, MMX, 8)\
  363. H264_MC(avg_, 16,MMX, 8)\
  364. #define H264_MC_816(QPEL, XMM)\
  365. QPEL(put_, 8, XMM, 16)\
  366. QPEL(put_, 16,XMM, 16)\
  367. QPEL(avg_, 8, XMM, 16)\
  368. QPEL(avg_, 16,XMM, 16)\
  369. #undef PAVGB
  370. #define PAVGB "pavgb"
  371. QPEL_H264(put_, PUT_OP, mmxext)
  372. QPEL_H264(avg_, AVG_MMXEXT_OP, mmxext)
  373. QPEL_H264_V_XMM(put_, PUT_OP, sse2)
  374. QPEL_H264_V_XMM(avg_,AVG_MMXEXT_OP, sse2)
  375. QPEL_H264_HV_XMM(put_, PUT_OP, sse2)
  376. QPEL_H264_HV_XMM(avg_,AVG_MMXEXT_OP, sse2)
  377. QPEL_H264_H_XMM(put_, PUT_OP, ssse3)
  378. QPEL_H264_H_XMM(avg_,AVG_MMXEXT_OP, ssse3)
  379. QPEL_H264_HV_XMM(put_, PUT_OP, ssse3)
  380. QPEL_H264_HV_XMM(avg_,AVG_MMXEXT_OP, ssse3)
  381. #undef PAVGB
  382. H264_MC_4816(mmxext)
  383. H264_MC_816(H264_MC_V, sse2)
  384. H264_MC_816(H264_MC_HV, sse2)
  385. H264_MC_816(H264_MC_H, ssse3)
  386. H264_MC_816(H264_MC_HV, ssse3)
  387. //10bit
  388. #define LUMA_MC_OP(OP, NUM, DEPTH, TYPE, OPT) \
  389. void ff_ ## OP ## _h264_qpel ## NUM ## _ ## TYPE ## _ ## DEPTH ## _ ## OPT \
  390. (uint8_t *dst, uint8_t *src, ptrdiff_t stride);
  391. #define LUMA_MC_ALL(DEPTH, TYPE, OPT) \
  392. LUMA_MC_OP(put, 4, DEPTH, TYPE, OPT) \
  393. LUMA_MC_OP(avg, 4, DEPTH, TYPE, OPT) \
  394. LUMA_MC_OP(put, 8, DEPTH, TYPE, OPT) \
  395. LUMA_MC_OP(avg, 8, DEPTH, TYPE, OPT) \
  396. LUMA_MC_OP(put, 16, DEPTH, TYPE, OPT) \
  397. LUMA_MC_OP(avg, 16, DEPTH, TYPE, OPT)
  398. #define LUMA_MC_816(DEPTH, TYPE, OPT) \
  399. LUMA_MC_OP(put, 8, DEPTH, TYPE, OPT) \
  400. LUMA_MC_OP(avg, 8, DEPTH, TYPE, OPT) \
  401. LUMA_MC_OP(put, 16, DEPTH, TYPE, OPT) \
  402. LUMA_MC_OP(avg, 16, DEPTH, TYPE, OPT)
  403. LUMA_MC_ALL(10, mc00, mmxext)
  404. LUMA_MC_ALL(10, mc10, mmxext)
  405. LUMA_MC_ALL(10, mc20, mmxext)
  406. LUMA_MC_ALL(10, mc30, mmxext)
  407. LUMA_MC_ALL(10, mc01, mmxext)
  408. LUMA_MC_ALL(10, mc11, mmxext)
  409. LUMA_MC_ALL(10, mc21, mmxext)
  410. LUMA_MC_ALL(10, mc31, mmxext)
  411. LUMA_MC_ALL(10, mc02, mmxext)
  412. LUMA_MC_ALL(10, mc12, mmxext)
  413. LUMA_MC_ALL(10, mc22, mmxext)
  414. LUMA_MC_ALL(10, mc32, mmxext)
  415. LUMA_MC_ALL(10, mc03, mmxext)
  416. LUMA_MC_ALL(10, mc13, mmxext)
  417. LUMA_MC_ALL(10, mc23, mmxext)
  418. LUMA_MC_ALL(10, mc33, mmxext)
  419. LUMA_MC_816(10, mc00, sse2)
  420. LUMA_MC_816(10, mc10, sse2)
  421. LUMA_MC_816(10, mc10, sse2_cache64)
  422. LUMA_MC_816(10, mc10, ssse3_cache64)
  423. LUMA_MC_816(10, mc20, sse2)
  424. LUMA_MC_816(10, mc20, sse2_cache64)
  425. LUMA_MC_816(10, mc20, ssse3_cache64)
  426. LUMA_MC_816(10, mc30, sse2)
  427. LUMA_MC_816(10, mc30, sse2_cache64)
  428. LUMA_MC_816(10, mc30, ssse3_cache64)
  429. LUMA_MC_816(10, mc01, sse2)
  430. LUMA_MC_816(10, mc11, sse2)
  431. LUMA_MC_816(10, mc21, sse2)
  432. LUMA_MC_816(10, mc31, sse2)
  433. LUMA_MC_816(10, mc02, sse2)
  434. LUMA_MC_816(10, mc12, sse2)
  435. LUMA_MC_816(10, mc22, sse2)
  436. LUMA_MC_816(10, mc32, sse2)
  437. LUMA_MC_816(10, mc03, sse2)
  438. LUMA_MC_816(10, mc13, sse2)
  439. LUMA_MC_816(10, mc23, sse2)
  440. LUMA_MC_816(10, mc33, sse2)
  441. #define QPEL16_OPMC(OP, MC, MMX)\
  442. void ff_ ## OP ## _h264_qpel16_ ## MC ## _10_ ## MMX(uint8_t *dst, uint8_t *src, ptrdiff_t stride){\
  443. ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst , src , stride);\
  444. ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst+16, src+16, stride);\
  445. src += 8*stride;\
  446. dst += 8*stride;\
  447. ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst , src , stride);\
  448. ff_ ## OP ## _h264_qpel8_ ## MC ## _10_ ## MMX(dst+16, src+16, stride);\
  449. }
  450. #define QPEL16_OP(MC, MMX)\
  451. QPEL16_OPMC(put, MC, MMX)\
  452. QPEL16_OPMC(avg, MC, MMX)
  453. #define QPEL16(MMX)\
  454. QPEL16_OP(mc00, MMX)\
  455. QPEL16_OP(mc01, MMX)\
  456. QPEL16_OP(mc02, MMX)\
  457. QPEL16_OP(mc03, MMX)\
  458. QPEL16_OP(mc10, MMX)\
  459. QPEL16_OP(mc11, MMX)\
  460. QPEL16_OP(mc12, MMX)\
  461. QPEL16_OP(mc13, MMX)\
  462. QPEL16_OP(mc20, MMX)\
  463. QPEL16_OP(mc21, MMX)\
  464. QPEL16_OP(mc22, MMX)\
  465. QPEL16_OP(mc23, MMX)\
  466. QPEL16_OP(mc30, MMX)\
  467. QPEL16_OP(mc31, MMX)\
  468. QPEL16_OP(mc32, MMX)\
  469. QPEL16_OP(mc33, MMX)
  470. #if ARCH_X86_32 && HAVE_YASM && CONFIG_H264QPEL // ARCH_X86_64 implies SSE2+
  471. QPEL16(mmxext)
  472. #endif
  473. #endif /* HAVE_YASM */
  474. #define SET_QPEL_FUNCS(PFX, IDX, SIZE, CPU, PREFIX) \
  475. do { \
  476. c->PFX ## _pixels_tab[IDX][ 0] = PREFIX ## PFX ## SIZE ## _mc00_ ## CPU; \
  477. c->PFX ## _pixels_tab[IDX][ 1] = PREFIX ## PFX ## SIZE ## _mc10_ ## CPU; \
  478. c->PFX ## _pixels_tab[IDX][ 2] = PREFIX ## PFX ## SIZE ## _mc20_ ## CPU; \
  479. c->PFX ## _pixels_tab[IDX][ 3] = PREFIX ## PFX ## SIZE ## _mc30_ ## CPU; \
  480. c->PFX ## _pixels_tab[IDX][ 4] = PREFIX ## PFX ## SIZE ## _mc01_ ## CPU; \
  481. c->PFX ## _pixels_tab[IDX][ 5] = PREFIX ## PFX ## SIZE ## _mc11_ ## CPU; \
  482. c->PFX ## _pixels_tab[IDX][ 6] = PREFIX ## PFX ## SIZE ## _mc21_ ## CPU; \
  483. c->PFX ## _pixels_tab[IDX][ 7] = PREFIX ## PFX ## SIZE ## _mc31_ ## CPU; \
  484. c->PFX ## _pixels_tab[IDX][ 8] = PREFIX ## PFX ## SIZE ## _mc02_ ## CPU; \
  485. c->PFX ## _pixels_tab[IDX][ 9] = PREFIX ## PFX ## SIZE ## _mc12_ ## CPU; \
  486. c->PFX ## _pixels_tab[IDX][10] = PREFIX ## PFX ## SIZE ## _mc22_ ## CPU; \
  487. c->PFX ## _pixels_tab[IDX][11] = PREFIX ## PFX ## SIZE ## _mc32_ ## CPU; \
  488. c->PFX ## _pixels_tab[IDX][12] = PREFIX ## PFX ## SIZE ## _mc03_ ## CPU; \
  489. c->PFX ## _pixels_tab[IDX][13] = PREFIX ## PFX ## SIZE ## _mc13_ ## CPU; \
  490. c->PFX ## _pixels_tab[IDX][14] = PREFIX ## PFX ## SIZE ## _mc23_ ## CPU; \
  491. c->PFX ## _pixels_tab[IDX][15] = PREFIX ## PFX ## SIZE ## _mc33_ ## CPU; \
  492. } while (0)
  493. #define H264_QPEL_FUNCS(x, y, CPU) \
  494. do { \
  495. c->put_h264_qpel_pixels_tab[0][x + y * 4] = put_h264_qpel16_mc ## x ## y ## _ ## CPU; \
  496. c->put_h264_qpel_pixels_tab[1][x + y * 4] = put_h264_qpel8_mc ## x ## y ## _ ## CPU; \
  497. c->avg_h264_qpel_pixels_tab[0][x + y * 4] = avg_h264_qpel16_mc ## x ## y ## _ ## CPU; \
  498. c->avg_h264_qpel_pixels_tab[1][x + y * 4] = avg_h264_qpel8_mc ## x ## y ## _ ## CPU; \
  499. } while (0)
  500. #define H264_QPEL_FUNCS_10(x, y, CPU) \
  501. do { \
  502. c->put_h264_qpel_pixels_tab[0][x + y * 4] = ff_put_h264_qpel16_mc ## x ## y ## _10_ ## CPU; \
  503. c->put_h264_qpel_pixels_tab[1][x + y * 4] = ff_put_h264_qpel8_mc ## x ## y ## _10_ ## CPU; \
  504. c->avg_h264_qpel_pixels_tab[0][x + y * 4] = ff_avg_h264_qpel16_mc ## x ## y ## _10_ ## CPU; \
  505. c->avg_h264_qpel_pixels_tab[1][x + y * 4] = ff_avg_h264_qpel8_mc ## x ## y ## _10_ ## CPU; \
  506. } while (0)
  507. av_cold void ff_h264qpel_init_x86(H264QpelContext *c, int bit_depth)
  508. {
  509. #if HAVE_YASM
  510. int high_bit_depth = bit_depth > 8;
  511. int mm_flags = av_get_cpu_flags();
  512. if (EXTERNAL_MMXEXT(mm_flags)) {
  513. if (!high_bit_depth) {
  514. SET_QPEL_FUNCS(put_h264_qpel, 0, 16, mmxext, );
  515. SET_QPEL_FUNCS(put_h264_qpel, 1, 8, mmxext, );
  516. SET_QPEL_FUNCS(put_h264_qpel, 2, 4, mmxext, );
  517. SET_QPEL_FUNCS(avg_h264_qpel, 0, 16, mmxext, );
  518. SET_QPEL_FUNCS(avg_h264_qpel, 1, 8, mmxext, );
  519. SET_QPEL_FUNCS(avg_h264_qpel, 2, 4, mmxext, );
  520. } else if (bit_depth == 10) {
  521. #if ARCH_X86_32
  522. SET_QPEL_FUNCS(avg_h264_qpel, 0, 16, 10_mmxext, ff_);
  523. SET_QPEL_FUNCS(put_h264_qpel, 0, 16, 10_mmxext, ff_);
  524. SET_QPEL_FUNCS(put_h264_qpel, 1, 8, 10_mmxext, ff_);
  525. SET_QPEL_FUNCS(avg_h264_qpel, 1, 8, 10_mmxext, ff_);
  526. #endif
  527. SET_QPEL_FUNCS(put_h264_qpel, 2, 4, 10_mmxext, ff_);
  528. SET_QPEL_FUNCS(avg_h264_qpel, 2, 4, 10_mmxext, ff_);
  529. }
  530. }
  531. if (EXTERNAL_SSE2(mm_flags)) {
  532. if (!(mm_flags & AV_CPU_FLAG_SSE2SLOW) && !high_bit_depth) {
  533. // these functions are slower than mmx on AMD, but faster on Intel
  534. H264_QPEL_FUNCS(0, 0, sse2);
  535. }
  536. if (!high_bit_depth) {
  537. H264_QPEL_FUNCS(0, 1, sse2);
  538. H264_QPEL_FUNCS(0, 2, sse2);
  539. H264_QPEL_FUNCS(0, 3, sse2);
  540. H264_QPEL_FUNCS(1, 1, sse2);
  541. H264_QPEL_FUNCS(1, 2, sse2);
  542. H264_QPEL_FUNCS(1, 3, sse2);
  543. H264_QPEL_FUNCS(2, 1, sse2);
  544. H264_QPEL_FUNCS(2, 2, sse2);
  545. H264_QPEL_FUNCS(2, 3, sse2);
  546. H264_QPEL_FUNCS(3, 1, sse2);
  547. H264_QPEL_FUNCS(3, 2, sse2);
  548. H264_QPEL_FUNCS(3, 3, sse2);
  549. }
  550. if (bit_depth == 10) {
  551. SET_QPEL_FUNCS(put_h264_qpel, 0, 16, 10_sse2, ff_);
  552. SET_QPEL_FUNCS(put_h264_qpel, 1, 8, 10_sse2, ff_);
  553. SET_QPEL_FUNCS(avg_h264_qpel, 0, 16, 10_sse2, ff_);
  554. SET_QPEL_FUNCS(avg_h264_qpel, 1, 8, 10_sse2, ff_);
  555. H264_QPEL_FUNCS_10(1, 0, sse2_cache64);
  556. H264_QPEL_FUNCS_10(2, 0, sse2_cache64);
  557. H264_QPEL_FUNCS_10(3, 0, sse2_cache64);
  558. }
  559. }
  560. if (EXTERNAL_SSSE3(mm_flags)) {
  561. if (!high_bit_depth) {
  562. H264_QPEL_FUNCS(1, 0, ssse3);
  563. H264_QPEL_FUNCS(1, 1, ssse3);
  564. H264_QPEL_FUNCS(1, 2, ssse3);
  565. H264_QPEL_FUNCS(1, 3, ssse3);
  566. H264_QPEL_FUNCS(2, 0, ssse3);
  567. H264_QPEL_FUNCS(2, 1, ssse3);
  568. H264_QPEL_FUNCS(2, 2, ssse3);
  569. H264_QPEL_FUNCS(2, 3, ssse3);
  570. H264_QPEL_FUNCS(3, 0, ssse3);
  571. H264_QPEL_FUNCS(3, 1, ssse3);
  572. H264_QPEL_FUNCS(3, 2, ssse3);
  573. H264_QPEL_FUNCS(3, 3, ssse3);
  574. }
  575. if (bit_depth == 10) {
  576. H264_QPEL_FUNCS_10(1, 0, ssse3_cache64);
  577. H264_QPEL_FUNCS_10(2, 0, ssse3_cache64);
  578. H264_QPEL_FUNCS_10(3, 0, ssse3_cache64);
  579. }
  580. }
  581. if (EXTERNAL_AVX(mm_flags)) {
  582. /* AVX implies 64 byte cache lines without the need to avoid unaligned
  583. * memory accesses that cross the boundary between two cache lines.
  584. * TODO: Port X264_CPU_CACHELINE_32/64 detection from x264 to avoid
  585. * having to treat SSE2 functions with such properties as AVX. */
  586. if (bit_depth == 10) {
  587. H264_QPEL_FUNCS_10(1, 0, sse2);
  588. H264_QPEL_FUNCS_10(2, 0, sse2);
  589. H264_QPEL_FUNCS_10(3, 0, sse2);
  590. }
  591. }
  592. #endif
  593. }