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.

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