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.

635 lines
29KB

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