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.

506 lines
20KB

  1. /*
  2. * Chinese AVS video (AVS1-P2, JiZhun profile) decoder.
  3. * Copyright (c) 2006 Stefan Gehrer <stefan.gehrer@gmx.de>
  4. *
  5. * MMX-optimized DSP functions, based on H.264 optimizations by
  6. * Michael Niedermayer and Loren Merritt
  7. *
  8. * This file is part of FFmpeg.
  9. *
  10. * FFmpeg is free software; you can redistribute it and/or
  11. * modify it under the terms of the GNU Lesser General Public
  12. * License as published by the Free Software Foundation; either
  13. * version 2.1 of the License, or (at your option) any later version.
  14. *
  15. * FFmpeg is distributed in the hope that it will be useful,
  16. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18. * Lesser General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU Lesser General Public
  21. * License along with FFmpeg; if not, write to the Free Software
  22. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23. */
  24. #include "libavutil/common.h"
  25. #include "libavutil/cpu.h"
  26. #include "libavutil/x86/asm.h"
  27. #include "libavutil/x86/cpu.h"
  28. #include "libavcodec/dsputil.h"
  29. #include "libavcodec/cavsdsp.h"
  30. #include "dsputil_mmx.h"
  31. #include "config.h"
  32. #if (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE)
  33. /* in/out: mma=mma+mmb, mmb=mmb-mma */
  34. #define SUMSUB_BA( a, b ) \
  35. "paddw "#b", "#a" \n\t"\
  36. "paddw "#b", "#b" \n\t"\
  37. "psubw "#a", "#b" \n\t"
  38. /*****************************************************************************
  39. *
  40. * inverse transform
  41. *
  42. ****************************************************************************/
  43. static inline void cavs_idct8_1d(int16_t *block, uint64_t bias)
  44. {
  45. __asm__ volatile(
  46. "movq 112(%0), %%mm4 \n\t" /* mm4 = src7 */
  47. "movq 16(%0), %%mm5 \n\t" /* mm5 = src1 */
  48. "movq 80(%0), %%mm2 \n\t" /* mm2 = src5 */
  49. "movq 48(%0), %%mm7 \n\t" /* mm7 = src3 */
  50. "movq %%mm4, %%mm0 \n\t"
  51. "movq %%mm5, %%mm3 \n\t"
  52. "movq %%mm2, %%mm6 \n\t"
  53. "movq %%mm7, %%mm1 \n\t"
  54. "paddw %%mm4, %%mm4 \n\t" /* mm4 = 2*src7 */
  55. "paddw %%mm3, %%mm3 \n\t" /* mm3 = 2*src1 */
  56. "paddw %%mm6, %%mm6 \n\t" /* mm6 = 2*src5 */
  57. "paddw %%mm1, %%mm1 \n\t" /* mm1 = 2*src3 */
  58. "paddw %%mm4, %%mm0 \n\t" /* mm0 = 3*src7 */
  59. "paddw %%mm3, %%mm5 \n\t" /* mm5 = 3*src1 */
  60. "paddw %%mm6, %%mm2 \n\t" /* mm2 = 3*src5 */
  61. "paddw %%mm1, %%mm7 \n\t" /* mm7 = 3*src3 */
  62. "psubw %%mm4, %%mm5 \n\t" /* mm5 = 3*src1 - 2*src7 = a0 */
  63. "paddw %%mm6, %%mm7 \n\t" /* mm7 = 3*src3 + 2*src5 = a1 */
  64. "psubw %%mm2, %%mm1 \n\t" /* mm1 = 2*src3 - 3*src5 = a2 */
  65. "paddw %%mm0, %%mm3 \n\t" /* mm3 = 2*src1 + 3*src7 = a3 */
  66. "movq %%mm5, %%mm4 \n\t"
  67. "movq %%mm7, %%mm6 \n\t"
  68. "movq %%mm3, %%mm0 \n\t"
  69. "movq %%mm1, %%mm2 \n\t"
  70. SUMSUB_BA( %%mm7, %%mm5 ) /* mm7 = a0 + a1 mm5 = a0 - a1 */
  71. "paddw %%mm3, %%mm7 \n\t" /* mm7 = a0 + a1 + a3 */
  72. "paddw %%mm1, %%mm5 \n\t" /* mm5 = a0 - a1 + a2 */
  73. "paddw %%mm7, %%mm7 \n\t"
  74. "paddw %%mm5, %%mm5 \n\t"
  75. "paddw %%mm6, %%mm7 \n\t" /* mm7 = b4 */
  76. "paddw %%mm4, %%mm5 \n\t" /* mm5 = b5 */
  77. SUMSUB_BA( %%mm1, %%mm3 ) /* mm1 = a3 + a2 mm3 = a3 - a2 */
  78. "psubw %%mm1, %%mm4 \n\t" /* mm4 = a0 - a2 - a3 */
  79. "movq %%mm4, %%mm1 \n\t" /* mm1 = a0 - a2 - a3 */
  80. "psubw %%mm6, %%mm3 \n\t" /* mm3 = a3 - a2 - a1 */
  81. "paddw %%mm1, %%mm1 \n\t"
  82. "paddw %%mm3, %%mm3 \n\t"
  83. "psubw %%mm2, %%mm1 \n\t" /* mm1 = b7 */
  84. "paddw %%mm0, %%mm3 \n\t" /* mm3 = b6 */
  85. "movq 32(%0), %%mm2 \n\t" /* mm2 = src2 */
  86. "movq 96(%0), %%mm6 \n\t" /* mm6 = src6 */
  87. "movq %%mm2, %%mm4 \n\t"
  88. "movq %%mm6, %%mm0 \n\t"
  89. "psllw $2, %%mm4 \n\t" /* mm4 = 4*src2 */
  90. "psllw $2, %%mm6 \n\t" /* mm6 = 4*src6 */
  91. "paddw %%mm4, %%mm2 \n\t" /* mm2 = 5*src2 */
  92. "paddw %%mm6, %%mm0 \n\t" /* mm0 = 5*src6 */
  93. "paddw %%mm2, %%mm2 \n\t"
  94. "paddw %%mm0, %%mm0 \n\t"
  95. "psubw %%mm0, %%mm4 \n\t" /* mm4 = 4*src2 - 10*src6 = a7 */
  96. "paddw %%mm2, %%mm6 \n\t" /* mm6 = 4*src6 + 10*src2 = a6 */
  97. "movq (%0), %%mm2 \n\t" /* mm2 = src0 */
  98. "movq 64(%0), %%mm0 \n\t" /* mm0 = src4 */
  99. SUMSUB_BA( %%mm0, %%mm2 ) /* mm0 = src0+src4 mm2 = src0-src4 */
  100. "psllw $3, %%mm0 \n\t"
  101. "psllw $3, %%mm2 \n\t"
  102. "paddw %1, %%mm0 \n\t" /* add rounding bias */
  103. "paddw %1, %%mm2 \n\t" /* add rounding bias */
  104. SUMSUB_BA( %%mm6, %%mm0 ) /* mm6 = a4 + a6 mm0 = a4 - a6 */
  105. SUMSUB_BA( %%mm4, %%mm2 ) /* mm4 = a5 + a7 mm2 = a5 - a7 */
  106. SUMSUB_BA( %%mm7, %%mm6 ) /* mm7 = dst0 mm6 = dst7 */
  107. SUMSUB_BA( %%mm5, %%mm4 ) /* mm5 = dst1 mm4 = dst6 */
  108. SUMSUB_BA( %%mm3, %%mm2 ) /* mm3 = dst2 mm2 = dst5 */
  109. SUMSUB_BA( %%mm1, %%mm0 ) /* mm1 = dst3 mm0 = dst4 */
  110. :: "r"(block), "m"(bias)
  111. );
  112. }
  113. static void cavs_idct8_add_mmx(uint8_t *dst, int16_t *block, int stride)
  114. {
  115. int i;
  116. DECLARE_ALIGNED(8, int16_t, b2)[64];
  117. for(i=0; i<2; i++){
  118. DECLARE_ALIGNED(8, uint64_t, tmp);
  119. cavs_idct8_1d(block+4*i, ff_pw_4.a);
  120. __asm__ volatile(
  121. "psraw $3, %%mm7 \n\t"
  122. "psraw $3, %%mm6 \n\t"
  123. "psraw $3, %%mm5 \n\t"
  124. "psraw $3, %%mm4 \n\t"
  125. "psraw $3, %%mm3 \n\t"
  126. "psraw $3, %%mm2 \n\t"
  127. "psraw $3, %%mm1 \n\t"
  128. "psraw $3, %%mm0 \n\t"
  129. "movq %%mm7, %0 \n\t"
  130. TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
  131. "movq %%mm0, 8(%1) \n\t"
  132. "movq %%mm6, 24(%1) \n\t"
  133. "movq %%mm7, 40(%1) \n\t"
  134. "movq %%mm4, 56(%1) \n\t"
  135. "movq %0, %%mm7 \n\t"
  136. TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
  137. "movq %%mm7, (%1) \n\t"
  138. "movq %%mm1, 16(%1) \n\t"
  139. "movq %%mm0, 32(%1) \n\t"
  140. "movq %%mm3, 48(%1) \n\t"
  141. : "=m"(tmp)
  142. : "r"(b2+32*i)
  143. : "memory"
  144. );
  145. }
  146. for(i=0; i<2; i++){
  147. cavs_idct8_1d(b2+4*i, ff_pw_64.a);
  148. __asm__ volatile(
  149. "psraw $7, %%mm7 \n\t"
  150. "psraw $7, %%mm6 \n\t"
  151. "psraw $7, %%mm5 \n\t"
  152. "psraw $7, %%mm4 \n\t"
  153. "psraw $7, %%mm3 \n\t"
  154. "psraw $7, %%mm2 \n\t"
  155. "psraw $7, %%mm1 \n\t"
  156. "psraw $7, %%mm0 \n\t"
  157. "movq %%mm7, (%0) \n\t"
  158. "movq %%mm5, 16(%0) \n\t"
  159. "movq %%mm3, 32(%0) \n\t"
  160. "movq %%mm1, 48(%0) \n\t"
  161. "movq %%mm0, 64(%0) \n\t"
  162. "movq %%mm2, 80(%0) \n\t"
  163. "movq %%mm4, 96(%0) \n\t"
  164. "movq %%mm6, 112(%0) \n\t"
  165. :: "r"(b2+4*i)
  166. : "memory"
  167. );
  168. }
  169. ff_add_pixels_clamped_mmx(b2, dst, stride);
  170. }
  171. /*****************************************************************************
  172. *
  173. * motion compensation
  174. *
  175. ****************************************************************************/
  176. /* vertical filter [-1 -2 96 42 -7 0] */
  177. #define QPEL_CAVSV1(A,B,C,D,E,F,OP,MUL2) \
  178. "movd (%0), "#F" \n\t"\
  179. "movq "#C", %%mm6 \n\t"\
  180. "pmullw %5, %%mm6 \n\t"\
  181. "movq "#D", %%mm7 \n\t"\
  182. "pmullw "MANGLE(MUL2)", %%mm7\n\t"\
  183. "psllw $3, "#E" \n\t"\
  184. "psubw "#E", %%mm6 \n\t"\
  185. "psraw $3, "#E" \n\t"\
  186. "paddw %%mm7, %%mm6 \n\t"\
  187. "paddw "#E", %%mm6 \n\t"\
  188. "paddw "#B", "#B" \n\t"\
  189. "pxor %%mm7, %%mm7 \n\t"\
  190. "add %2, %0 \n\t"\
  191. "punpcklbw %%mm7, "#F" \n\t"\
  192. "psubw "#B", %%mm6 \n\t"\
  193. "psraw $1, "#B" \n\t"\
  194. "psubw "#A", %%mm6 \n\t"\
  195. "paddw %4, %%mm6 \n\t"\
  196. "psraw $7, %%mm6 \n\t"\
  197. "packuswb %%mm6, %%mm6 \n\t"\
  198. OP(%%mm6, (%1), A, d) \
  199. "add %3, %1 \n\t"
  200. /* vertical filter [ 0 -1 5 5 -1 0] */
  201. #define QPEL_CAVSV2(A,B,C,D,E,F,OP,MUL2) \
  202. "movd (%0), "#F" \n\t"\
  203. "movq "#C", %%mm6 \n\t"\
  204. "paddw "#D", %%mm6 \n\t"\
  205. "pmullw %5, %%mm6 \n\t"\
  206. "add %2, %0 \n\t"\
  207. "punpcklbw %%mm7, "#F" \n\t"\
  208. "psubw "#B", %%mm6 \n\t"\
  209. "psubw "#E", %%mm6 \n\t"\
  210. "paddw %4, %%mm6 \n\t"\
  211. "psraw $3, %%mm6 \n\t"\
  212. "packuswb %%mm6, %%mm6 \n\t"\
  213. OP(%%mm6, (%1), A, d) \
  214. "add %3, %1 \n\t"
  215. /* vertical filter [ 0 -7 42 96 -2 -1] */
  216. #define QPEL_CAVSV3(A,B,C,D,E,F,OP,MUL2) \
  217. "movd (%0), "#F" \n\t"\
  218. "movq "#C", %%mm6 \n\t"\
  219. "pmullw "MANGLE(MUL2)", %%mm6\n\t"\
  220. "movq "#D", %%mm7 \n\t"\
  221. "pmullw %5, %%mm7 \n\t"\
  222. "psllw $3, "#B" \n\t"\
  223. "psubw "#B", %%mm6 \n\t"\
  224. "psraw $3, "#B" \n\t"\
  225. "paddw %%mm7, %%mm6 \n\t"\
  226. "paddw "#B", %%mm6 \n\t"\
  227. "paddw "#E", "#E" \n\t"\
  228. "pxor %%mm7, %%mm7 \n\t"\
  229. "add %2, %0 \n\t"\
  230. "punpcklbw %%mm7, "#F" \n\t"\
  231. "psubw "#E", %%mm6 \n\t"\
  232. "psraw $1, "#E" \n\t"\
  233. "psubw "#F", %%mm6 \n\t"\
  234. "paddw %4, %%mm6 \n\t"\
  235. "psraw $7, %%mm6 \n\t"\
  236. "packuswb %%mm6, %%mm6 \n\t"\
  237. OP(%%mm6, (%1), A, d) \
  238. "add %3, %1 \n\t"
  239. #define QPEL_CAVSVNUM(VOP,OP,ADD,MUL1,MUL2)\
  240. int w= 2;\
  241. src -= 2*srcStride;\
  242. \
  243. while(w--){\
  244. __asm__ volatile(\
  245. "pxor %%mm7, %%mm7 \n\t"\
  246. "movd (%0), %%mm0 \n\t"\
  247. "add %2, %0 \n\t"\
  248. "movd (%0), %%mm1 \n\t"\
  249. "add %2, %0 \n\t"\
  250. "movd (%0), %%mm2 \n\t"\
  251. "add %2, %0 \n\t"\
  252. "movd (%0), %%mm3 \n\t"\
  253. "add %2, %0 \n\t"\
  254. "movd (%0), %%mm4 \n\t"\
  255. "add %2, %0 \n\t"\
  256. "punpcklbw %%mm7, %%mm0 \n\t"\
  257. "punpcklbw %%mm7, %%mm1 \n\t"\
  258. "punpcklbw %%mm7, %%mm2 \n\t"\
  259. "punpcklbw %%mm7, %%mm3 \n\t"\
  260. "punpcklbw %%mm7, %%mm4 \n\t"\
  261. VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
  262. VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
  263. VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
  264. VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
  265. VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
  266. VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
  267. VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
  268. VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
  269. \
  270. : "+a"(src), "+c"(dst)\
  271. : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD), "m"(MUL1)\
  272. : "memory"\
  273. );\
  274. if(h==16){\
  275. __asm__ volatile(\
  276. VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
  277. VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
  278. VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
  279. VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
  280. VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
  281. VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
  282. VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
  283. VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
  284. \
  285. : "+a"(src), "+c"(dst)\
  286. : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD), "m"(MUL1)\
  287. : "memory"\
  288. );\
  289. }\
  290. src += 4-(h+5)*srcStride;\
  291. dst += 4-h*dstStride;\
  292. }
  293. #define QPEL_CAVS(OPNAME, OP, MMX)\
  294. static void OPNAME ## cavs_qpel8_h_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  295. int h=8;\
  296. __asm__ volatile(\
  297. "pxor %%mm7, %%mm7 \n\t"\
  298. "movq %5, %%mm6 \n\t"\
  299. "1: \n\t"\
  300. "movq (%0), %%mm0 \n\t"\
  301. "movq 1(%0), %%mm2 \n\t"\
  302. "movq %%mm0, %%mm1 \n\t"\
  303. "movq %%mm2, %%mm3 \n\t"\
  304. "punpcklbw %%mm7, %%mm0 \n\t"\
  305. "punpckhbw %%mm7, %%mm1 \n\t"\
  306. "punpcklbw %%mm7, %%mm2 \n\t"\
  307. "punpckhbw %%mm7, %%mm3 \n\t"\
  308. "paddw %%mm2, %%mm0 \n\t"\
  309. "paddw %%mm3, %%mm1 \n\t"\
  310. "pmullw %%mm6, %%mm0 \n\t"\
  311. "pmullw %%mm6, %%mm1 \n\t"\
  312. "movq -1(%0), %%mm2 \n\t"\
  313. "movq 2(%0), %%mm4 \n\t"\
  314. "movq %%mm2, %%mm3 \n\t"\
  315. "movq %%mm4, %%mm5 \n\t"\
  316. "punpcklbw %%mm7, %%mm2 \n\t"\
  317. "punpckhbw %%mm7, %%mm3 \n\t"\
  318. "punpcklbw %%mm7, %%mm4 \n\t"\
  319. "punpckhbw %%mm7, %%mm5 \n\t"\
  320. "paddw %%mm4, %%mm2 \n\t"\
  321. "paddw %%mm3, %%mm5 \n\t"\
  322. "psubw %%mm2, %%mm0 \n\t"\
  323. "psubw %%mm5, %%mm1 \n\t"\
  324. "movq %6, %%mm5 \n\t"\
  325. "paddw %%mm5, %%mm0 \n\t"\
  326. "paddw %%mm5, %%mm1 \n\t"\
  327. "psraw $3, %%mm0 \n\t"\
  328. "psraw $3, %%mm1 \n\t"\
  329. "packuswb %%mm1, %%mm0 \n\t"\
  330. OP(%%mm0, (%1),%%mm5, q) \
  331. "add %3, %0 \n\t"\
  332. "add %4, %1 \n\t"\
  333. "decl %2 \n\t"\
  334. " jnz 1b \n\t"\
  335. : "+a"(src), "+c"(dst), "+m"(h)\
  336. : "d"((x86_reg)srcStride), "S"((x86_reg)dstStride), "m"(ff_pw_5), "m"(ff_pw_4)\
  337. : "memory"\
  338. );\
  339. }\
  340. \
  341. static inline void OPNAME ## cavs_qpel8or16_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
  342. QPEL_CAVSVNUM(QPEL_CAVSV1,OP,ff_pw_64,ff_pw_96,ff_pw_42) \
  343. }\
  344. \
  345. static inline void OPNAME ## cavs_qpel8or16_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
  346. QPEL_CAVSVNUM(QPEL_CAVSV2,OP,ff_pw_4,ff_pw_5,ff_pw_5) \
  347. }\
  348. \
  349. static inline void OPNAME ## cavs_qpel8or16_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
  350. QPEL_CAVSVNUM(QPEL_CAVSV3,OP,ff_pw_64,ff_pw_96,ff_pw_42) \
  351. }\
  352. \
  353. static void OPNAME ## cavs_qpel8_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  354. OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst , src , dstStride, srcStride, 8);\
  355. }\
  356. static void OPNAME ## cavs_qpel16_v1_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  357. OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst , src , dstStride, srcStride, 16);\
  358. OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  359. }\
  360. \
  361. static void OPNAME ## cavs_qpel8_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  362. OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst , src , dstStride, srcStride, 8);\
  363. }\
  364. static void OPNAME ## cavs_qpel16_v2_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  365. OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst , src , dstStride, srcStride, 16);\
  366. OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  367. }\
  368. \
  369. static void OPNAME ## cavs_qpel8_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  370. OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst , src , dstStride, srcStride, 8);\
  371. }\
  372. static void OPNAME ## cavs_qpel16_v3_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  373. OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst , src , dstStride, srcStride, 16);\
  374. OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  375. }\
  376. \
  377. static void OPNAME ## cavs_qpel16_h_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  378. OPNAME ## cavs_qpel8_h_ ## MMX(dst , src , dstStride, srcStride);\
  379. OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  380. src += 8*srcStride;\
  381. dst += 8*dstStride;\
  382. OPNAME ## cavs_qpel8_h_ ## MMX(dst , src , dstStride, srcStride);\
  383. OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  384. }\
  385. #define CAVS_MC(OPNAME, SIZE, MMX) \
  386. static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  387. OPNAME ## cavs_qpel ## SIZE ## _h_ ## MMX(dst, src, stride, stride);\
  388. }\
  389. \
  390. static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  391. OPNAME ## cavs_qpel ## SIZE ## _v1_ ## MMX(dst, src, stride, stride);\
  392. }\
  393. \
  394. static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  395. OPNAME ## cavs_qpel ## SIZE ## _v2_ ## MMX(dst, src, stride, stride);\
  396. }\
  397. \
  398. static void ff_ ## OPNAME ## cavs_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  399. OPNAME ## cavs_qpel ## SIZE ## _v3_ ## MMX(dst, src, stride, stride);\
  400. }\
  401. #define PUT_OP(a,b,temp, size) "mov" #size " " #a ", " #b " \n\t"
  402. #define AVG_3DNOW_OP(a,b,temp, size) \
  403. "mov" #size " " #b ", " #temp " \n\t"\
  404. "pavgusb " #temp ", " #a " \n\t"\
  405. "mov" #size " " #a ", " #b " \n\t"
  406. #define AVG_MMXEXT_OP(a, b, temp, size) \
  407. "mov" #size " " #b ", " #temp " \n\t"\
  408. "pavgb " #temp ", " #a " \n\t"\
  409. "mov" #size " " #a ", " #b " \n\t"
  410. #endif /* (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE) */
  411. #if HAVE_MMXEXT_INLINE
  412. QPEL_CAVS(put_, PUT_OP, mmxext)
  413. QPEL_CAVS(avg_, AVG_MMXEXT_OP, mmxext)
  414. CAVS_MC(put_, 8, mmxext)
  415. CAVS_MC(put_, 16, mmxext)
  416. CAVS_MC(avg_, 8, mmxext)
  417. CAVS_MC(avg_, 16, mmxext)
  418. static void ff_cavsdsp_init_mmxext(CAVSDSPContext *c, AVCodecContext *avctx)
  419. {
  420. #define dspfunc(PFX, IDX, NUM) \
  421. c->PFX ## _pixels_tab[IDX][ 0] = ff_ ## PFX ## NUM ## _mc00_mmxext; \
  422. c->PFX ## _pixels_tab[IDX][ 2] = ff_ ## PFX ## NUM ## _mc20_mmxext; \
  423. c->PFX ## _pixels_tab[IDX][ 4] = ff_ ## PFX ## NUM ## _mc01_mmxext; \
  424. c->PFX ## _pixels_tab[IDX][ 8] = ff_ ## PFX ## NUM ## _mc02_mmxext; \
  425. c->PFX ## _pixels_tab[IDX][12] = ff_ ## PFX ## NUM ## _mc03_mmxext; \
  426. dspfunc(put_cavs_qpel, 0, 16);
  427. dspfunc(put_cavs_qpel, 1, 8);
  428. dspfunc(avg_cavs_qpel, 0, 16);
  429. dspfunc(avg_cavs_qpel, 1, 8);
  430. #undef dspfunc
  431. c->cavs_idct8_add = cavs_idct8_add_mmx;
  432. c->idct_perm = FF_TRANSPOSE_IDCT_PERM;
  433. }
  434. #endif /* HAVE_MMXEXT_INLINE */
  435. #if HAVE_AMD3DNOW_INLINE
  436. QPEL_CAVS(put_, PUT_OP, 3dnow)
  437. QPEL_CAVS(avg_, AVG_3DNOW_OP, 3dnow)
  438. CAVS_MC(put_, 8, 3dnow)
  439. CAVS_MC(put_, 16,3dnow)
  440. CAVS_MC(avg_, 8, 3dnow)
  441. CAVS_MC(avg_, 16,3dnow)
  442. static void ff_cavsdsp_init_3dnow(CAVSDSPContext* c, AVCodecContext *avctx) {
  443. #define dspfunc(PFX, IDX, NUM) \
  444. c->PFX ## _pixels_tab[IDX][ 0] = ff_ ## PFX ## NUM ## _mc00_mmxext; \
  445. c->PFX ## _pixels_tab[IDX][ 2] = ff_ ## PFX ## NUM ## _mc20_3dnow; \
  446. c->PFX ## _pixels_tab[IDX][ 4] = ff_ ## PFX ## NUM ## _mc01_3dnow; \
  447. c->PFX ## _pixels_tab[IDX][ 8] = ff_ ## PFX ## NUM ## _mc02_3dnow; \
  448. c->PFX ## _pixels_tab[IDX][12] = ff_ ## PFX ## NUM ## _mc03_3dnow; \
  449. dspfunc(put_cavs_qpel, 0, 16);
  450. dspfunc(put_cavs_qpel, 1, 8);
  451. dspfunc(avg_cavs_qpel, 0, 16);
  452. dspfunc(avg_cavs_qpel, 1, 8);
  453. #undef dspfunc
  454. c->cavs_idct8_add = cavs_idct8_add_mmx;
  455. c->idct_perm = FF_TRANSPOSE_IDCT_PERM;
  456. }
  457. #endif /* HAVE_AMD3DNOW_INLINE */
  458. av_cold void ff_cavsdsp_init_x86(CAVSDSPContext *c, AVCodecContext *avctx)
  459. {
  460. int mm_flags = av_get_cpu_flags();
  461. #if HAVE_MMXEXT_INLINE
  462. if (mm_flags & AV_CPU_FLAG_MMXEXT) ff_cavsdsp_init_mmxext(c, avctx);
  463. #endif /* HAVE_MMXEXT_INLINE */
  464. #if HAVE_AMD3DNOW_INLINE
  465. if (mm_flags & AV_CPU_FLAG_3DNOW) ff_cavsdsp_init_3dnow(c, avctx);
  466. #endif /* HAVE_AMD3DNOW_INLINE */
  467. }