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.

559 lines
21KB

  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 Libav.
  9. *
  10. * Libav 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. * Libav 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 Libav; if not, write to the Free Software
  22. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23. */
  24. #include "libavutil/attributes.h"
  25. #include "libavutil/common.h"
  26. #include "libavutil/cpu.h"
  27. #include "libavutil/x86/asm.h"
  28. #include "libavutil/x86/cpu.h"
  29. #include "libavcodec/cavsdsp.h"
  30. #include "libavcodec/idctdsp.h"
  31. #include "constants.h"
  32. #include "fpel.h"
  33. #include "idctdsp.h"
  34. #include "config.h"
  35. #if HAVE_MMX_INLINE
  36. /* in/out: mma=mma+mmb, mmb=mmb-mma */
  37. #define SUMSUB_BA( a, b ) \
  38. "paddw "#b", "#a" \n\t"\
  39. "paddw "#b", "#b" \n\t"\
  40. "psubw "#a", "#b" \n\t"
  41. /*****************************************************************************
  42. *
  43. * inverse transform
  44. *
  45. ****************************************************************************/
  46. static inline void cavs_idct8_1d(int16_t *block, uint64_t bias)
  47. {
  48. __asm__ volatile(
  49. "movq 112(%0), %%mm4 \n\t" /* mm4 = src7 */
  50. "movq 16(%0), %%mm5 \n\t" /* mm5 = src1 */
  51. "movq 80(%0), %%mm2 \n\t" /* mm2 = src5 */
  52. "movq 48(%0), %%mm7 \n\t" /* mm7 = src3 */
  53. "movq %%mm4, %%mm0 \n\t"
  54. "movq %%mm5, %%mm3 \n\t"
  55. "movq %%mm2, %%mm6 \n\t"
  56. "movq %%mm7, %%mm1 \n\t"
  57. "paddw %%mm4, %%mm4 \n\t" /* mm4 = 2*src7 */
  58. "paddw %%mm3, %%mm3 \n\t" /* mm3 = 2*src1 */
  59. "paddw %%mm6, %%mm6 \n\t" /* mm6 = 2*src5 */
  60. "paddw %%mm1, %%mm1 \n\t" /* mm1 = 2*src3 */
  61. "paddw %%mm4, %%mm0 \n\t" /* mm0 = 3*src7 */
  62. "paddw %%mm3, %%mm5 \n\t" /* mm5 = 3*src1 */
  63. "paddw %%mm6, %%mm2 \n\t" /* mm2 = 3*src5 */
  64. "paddw %%mm1, %%mm7 \n\t" /* mm7 = 3*src3 */
  65. "psubw %%mm4, %%mm5 \n\t" /* mm5 = 3*src1 - 2*src7 = a0 */
  66. "paddw %%mm6, %%mm7 \n\t" /* mm7 = 3*src3 + 2*src5 = a1 */
  67. "psubw %%mm2, %%mm1 \n\t" /* mm1 = 2*src3 - 3*src5 = a2 */
  68. "paddw %%mm0, %%mm3 \n\t" /* mm3 = 2*src1 + 3*src7 = a3 */
  69. "movq %%mm5, %%mm4 \n\t"
  70. "movq %%mm7, %%mm6 \n\t"
  71. "movq %%mm3, %%mm0 \n\t"
  72. "movq %%mm1, %%mm2 \n\t"
  73. SUMSUB_BA( %%mm7, %%mm5 ) /* mm7 = a0 + a1 mm5 = a0 - a1 */
  74. "paddw %%mm3, %%mm7 \n\t" /* mm7 = a0 + a1 + a3 */
  75. "paddw %%mm1, %%mm5 \n\t" /* mm5 = a0 - a1 + a2 */
  76. "paddw %%mm7, %%mm7 \n\t"
  77. "paddw %%mm5, %%mm5 \n\t"
  78. "paddw %%mm6, %%mm7 \n\t" /* mm7 = b4 */
  79. "paddw %%mm4, %%mm5 \n\t" /* mm5 = b5 */
  80. SUMSUB_BA( %%mm1, %%mm3 ) /* mm1 = a3 + a2 mm3 = a3 - a2 */
  81. "psubw %%mm1, %%mm4 \n\t" /* mm4 = a0 - a2 - a3 */
  82. "movq %%mm4, %%mm1 \n\t" /* mm1 = a0 - a2 - a3 */
  83. "psubw %%mm6, %%mm3 \n\t" /* mm3 = a3 - a2 - a1 */
  84. "paddw %%mm1, %%mm1 \n\t"
  85. "paddw %%mm3, %%mm3 \n\t"
  86. "psubw %%mm2, %%mm1 \n\t" /* mm1 = b7 */
  87. "paddw %%mm0, %%mm3 \n\t" /* mm3 = b6 */
  88. "movq 32(%0), %%mm2 \n\t" /* mm2 = src2 */
  89. "movq 96(%0), %%mm6 \n\t" /* mm6 = src6 */
  90. "movq %%mm2, %%mm4 \n\t"
  91. "movq %%mm6, %%mm0 \n\t"
  92. "psllw $2, %%mm4 \n\t" /* mm4 = 4*src2 */
  93. "psllw $2, %%mm6 \n\t" /* mm6 = 4*src6 */
  94. "paddw %%mm4, %%mm2 \n\t" /* mm2 = 5*src2 */
  95. "paddw %%mm6, %%mm0 \n\t" /* mm0 = 5*src6 */
  96. "paddw %%mm2, %%mm2 \n\t"
  97. "paddw %%mm0, %%mm0 \n\t"
  98. "psubw %%mm0, %%mm4 \n\t" /* mm4 = 4*src2 - 10*src6 = a7 */
  99. "paddw %%mm2, %%mm6 \n\t" /* mm6 = 4*src6 + 10*src2 = a6 */
  100. "movq (%0), %%mm2 \n\t" /* mm2 = src0 */
  101. "movq 64(%0), %%mm0 \n\t" /* mm0 = src4 */
  102. SUMSUB_BA( %%mm0, %%mm2 ) /* mm0 = src0+src4 mm2 = src0-src4 */
  103. "psllw $3, %%mm0 \n\t"
  104. "psllw $3, %%mm2 \n\t"
  105. "paddw %1, %%mm0 \n\t" /* add rounding bias */
  106. "paddw %1, %%mm2 \n\t" /* add rounding bias */
  107. SUMSUB_BA( %%mm6, %%mm0 ) /* mm6 = a4 + a6 mm0 = a4 - a6 */
  108. SUMSUB_BA( %%mm4, %%mm2 ) /* mm4 = a5 + a7 mm2 = a5 - a7 */
  109. SUMSUB_BA( %%mm7, %%mm6 ) /* mm7 = dst0 mm6 = dst7 */
  110. SUMSUB_BA( %%mm5, %%mm4 ) /* mm5 = dst1 mm4 = dst6 */
  111. SUMSUB_BA( %%mm3, %%mm2 ) /* mm3 = dst2 mm2 = dst5 */
  112. SUMSUB_BA( %%mm1, %%mm0 ) /* mm1 = dst3 mm0 = dst4 */
  113. :: "r"(block), "m"(bias)
  114. );
  115. }
  116. #define SBUTTERFLY(a,b,t,n,m)\
  117. "mov" #m " " #a ", " #t " \n\t" /* abcd */\
  118. "punpckl" #n " " #b ", " #a " \n\t" /* aebf */\
  119. "punpckh" #n " " #b ", " #t " \n\t" /* cgdh */\
  120. #define TRANSPOSE4(a,b,c,d,t)\
  121. SBUTTERFLY(a,b,t,wd,q) /* a=aebf t=cgdh */\
  122. SBUTTERFLY(c,d,b,wd,q) /* c=imjn b=kolp */\
  123. SBUTTERFLY(a,c,d,dq,q) /* a=aeim d=bfjn */\
  124. SBUTTERFLY(t,b,c,dq,q) /* t=cgko c=dhlp */
  125. static void cavs_idct8_add_mmx(uint8_t *dst, int16_t *block, int stride)
  126. {
  127. int i;
  128. DECLARE_ALIGNED(8, int16_t, b2)[64];
  129. for(i=0; i<2; i++){
  130. cavs_idct8_1d(block + 4 * i, ff_pw_4.a);
  131. __asm__ volatile(
  132. "psraw $3, %%mm7 \n\t"
  133. "psraw $3, %%mm6 \n\t"
  134. "psraw $3, %%mm5 \n\t"
  135. "psraw $3, %%mm4 \n\t"
  136. "psraw $3, %%mm3 \n\t"
  137. "psraw $3, %%mm2 \n\t"
  138. "psraw $3, %%mm1 \n\t"
  139. "psraw $3, %%mm0 \n\t"
  140. "movq %%mm7, (%0) \n\t"
  141. TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
  142. "movq %%mm0, 8(%0) \n\t"
  143. "movq %%mm6, 24(%0) \n\t"
  144. "movq %%mm7, 40(%0) \n\t"
  145. "movq %%mm4, 56(%0) \n\t"
  146. "movq (%0), %%mm7 \n\t"
  147. TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
  148. "movq %%mm7, (%0) \n\t"
  149. "movq %%mm1, 16(%0) \n\t"
  150. "movq %%mm0, 32(%0) \n\t"
  151. "movq %%mm3, 48(%0) \n\t"
  152. :
  153. : "r"(b2 + 32 * i)
  154. : "memory"
  155. );
  156. }
  157. for(i=0; i<2; i++){
  158. cavs_idct8_1d(b2+4*i, ff_pw_64.a);
  159. __asm__ volatile(
  160. "psraw $7, %%mm7 \n\t"
  161. "psraw $7, %%mm6 \n\t"
  162. "psraw $7, %%mm5 \n\t"
  163. "psraw $7, %%mm4 \n\t"
  164. "psraw $7, %%mm3 \n\t"
  165. "psraw $7, %%mm2 \n\t"
  166. "psraw $7, %%mm1 \n\t"
  167. "psraw $7, %%mm0 \n\t"
  168. "movq %%mm7, (%0) \n\t"
  169. "movq %%mm5, 16(%0) \n\t"
  170. "movq %%mm3, 32(%0) \n\t"
  171. "movq %%mm1, 48(%0) \n\t"
  172. "movq %%mm0, 64(%0) \n\t"
  173. "movq %%mm2, 80(%0) \n\t"
  174. "movq %%mm4, 96(%0) \n\t"
  175. "movq %%mm6, 112(%0) \n\t"
  176. :: "r"(b2+4*i)
  177. : "memory"
  178. );
  179. }
  180. ff_add_pixels_clamped_mmx(b2, dst, stride);
  181. }
  182. #endif /* HAVE_MMX_INLINE */
  183. #if (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE)
  184. /*****************************************************************************
  185. *
  186. * motion compensation
  187. *
  188. ****************************************************************************/
  189. /* vertical filter [-1 -2 96 42 -7 0] */
  190. #define QPEL_CAVSV1(A,B,C,D,E,F,OP,MUL2) \
  191. "movd (%0), "#F" \n\t"\
  192. "movq "#C", %%mm6 \n\t"\
  193. "pmullw %5, %%mm6 \n\t"\
  194. "movq "#D", %%mm7 \n\t"\
  195. "pmullw "MANGLE(MUL2)", %%mm7\n\t"\
  196. "psllw $3, "#E" \n\t"\
  197. "psubw "#E", %%mm6 \n\t"\
  198. "psraw $3, "#E" \n\t"\
  199. "paddw %%mm7, %%mm6 \n\t"\
  200. "paddw "#E", %%mm6 \n\t"\
  201. "paddw "#B", "#B" \n\t"\
  202. "pxor %%mm7, %%mm7 \n\t"\
  203. "add %2, %0 \n\t"\
  204. "punpcklbw %%mm7, "#F" \n\t"\
  205. "psubw "#B", %%mm6 \n\t"\
  206. "psraw $1, "#B" \n\t"\
  207. "psubw "#A", %%mm6 \n\t"\
  208. "paddw %4, %%mm6 \n\t"\
  209. "psraw $7, %%mm6 \n\t"\
  210. "packuswb %%mm6, %%mm6 \n\t"\
  211. OP(%%mm6, (%1), A, d) \
  212. "add %3, %1 \n\t"
  213. /* vertical filter [ 0 -1 5 5 -1 0] */
  214. #define QPEL_CAVSV2(A,B,C,D,E,F,OP,MUL2) \
  215. "movd (%0), "#F" \n\t"\
  216. "movq "#C", %%mm6 \n\t"\
  217. "paddw "#D", %%mm6 \n\t"\
  218. "pmullw %5, %%mm6 \n\t"\
  219. "add %2, %0 \n\t"\
  220. "punpcklbw %%mm7, "#F" \n\t"\
  221. "psubw "#B", %%mm6 \n\t"\
  222. "psubw "#E", %%mm6 \n\t"\
  223. "paddw %4, %%mm6 \n\t"\
  224. "psraw $3, %%mm6 \n\t"\
  225. "packuswb %%mm6, %%mm6 \n\t"\
  226. OP(%%mm6, (%1), A, d) \
  227. "add %3, %1 \n\t"
  228. /* vertical filter [ 0 -7 42 96 -2 -1] */
  229. #define QPEL_CAVSV3(A,B,C,D,E,F,OP,MUL2) \
  230. "movd (%0), "#F" \n\t"\
  231. "movq "#C", %%mm6 \n\t"\
  232. "pmullw "MANGLE(MUL2)", %%mm6\n\t"\
  233. "movq "#D", %%mm7 \n\t"\
  234. "pmullw %5, %%mm7 \n\t"\
  235. "psllw $3, "#B" \n\t"\
  236. "psubw "#B", %%mm6 \n\t"\
  237. "psraw $3, "#B" \n\t"\
  238. "paddw %%mm7, %%mm6 \n\t"\
  239. "paddw "#B", %%mm6 \n\t"\
  240. "paddw "#E", "#E" \n\t"\
  241. "pxor %%mm7, %%mm7 \n\t"\
  242. "add %2, %0 \n\t"\
  243. "punpcklbw %%mm7, "#F" \n\t"\
  244. "psubw "#E", %%mm6 \n\t"\
  245. "psraw $1, "#E" \n\t"\
  246. "psubw "#F", %%mm6 \n\t"\
  247. "paddw %4, %%mm6 \n\t"\
  248. "psraw $7, %%mm6 \n\t"\
  249. "packuswb %%mm6, %%mm6 \n\t"\
  250. OP(%%mm6, (%1), A, d) \
  251. "add %3, %1 \n\t"
  252. #define QPEL_CAVSVNUM(VOP,OP,ADD,MUL1,MUL2)\
  253. int w= 2;\
  254. src -= 2*srcStride;\
  255. \
  256. while(w--){\
  257. __asm__ volatile(\
  258. "pxor %%mm7, %%mm7 \n\t"\
  259. "movd (%0), %%mm0 \n\t"\
  260. "add %2, %0 \n\t"\
  261. "movd (%0), %%mm1 \n\t"\
  262. "add %2, %0 \n\t"\
  263. "movd (%0), %%mm2 \n\t"\
  264. "add %2, %0 \n\t"\
  265. "movd (%0), %%mm3 \n\t"\
  266. "add %2, %0 \n\t"\
  267. "movd (%0), %%mm4 \n\t"\
  268. "add %2, %0 \n\t"\
  269. "punpcklbw %%mm7, %%mm0 \n\t"\
  270. "punpcklbw %%mm7, %%mm1 \n\t"\
  271. "punpcklbw %%mm7, %%mm2 \n\t"\
  272. "punpcklbw %%mm7, %%mm3 \n\t"\
  273. "punpcklbw %%mm7, %%mm4 \n\t"\
  274. VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
  275. VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
  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. \
  283. : "+a"(src), "+c"(dst)\
  284. : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD), "m"(MUL1)\
  285. : "memory"\
  286. );\
  287. if(h==16){\
  288. __asm__ volatile(\
  289. VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
  290. VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
  291. VOP(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP, MUL2)\
  292. VOP(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP, MUL2)\
  293. VOP(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP, MUL2)\
  294. VOP(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP, MUL2)\
  295. VOP(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP, MUL2)\
  296. VOP(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP, MUL2)\
  297. \
  298. : "+a"(src), "+c"(dst)\
  299. : "S"((x86_reg)srcStride), "r"((x86_reg)dstStride), "m"(ADD), "m"(MUL1)\
  300. : "memory"\
  301. );\
  302. }\
  303. src += 4-(h+5)*srcStride;\
  304. dst += 4-h*dstStride;\
  305. }
  306. #define QPEL_CAVS(OPNAME, OP, MMX)\
  307. static void OPNAME ## cavs_qpel8_h_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  308. int h=8;\
  309. __asm__ volatile(\
  310. "pxor %%mm7, %%mm7 \n\t"\
  311. "movq %5, %%mm6 \n\t"\
  312. "1: \n\t"\
  313. "movq (%0), %%mm0 \n\t"\
  314. "movq 1(%0), %%mm2 \n\t"\
  315. "movq %%mm0, %%mm1 \n\t"\
  316. "movq %%mm2, %%mm3 \n\t"\
  317. "punpcklbw %%mm7, %%mm0 \n\t"\
  318. "punpckhbw %%mm7, %%mm1 \n\t"\
  319. "punpcklbw %%mm7, %%mm2 \n\t"\
  320. "punpckhbw %%mm7, %%mm3 \n\t"\
  321. "paddw %%mm2, %%mm0 \n\t"\
  322. "paddw %%mm3, %%mm1 \n\t"\
  323. "pmullw %%mm6, %%mm0 \n\t"\
  324. "pmullw %%mm6, %%mm1 \n\t"\
  325. "movq -1(%0), %%mm2 \n\t"\
  326. "movq 2(%0), %%mm4 \n\t"\
  327. "movq %%mm2, %%mm3 \n\t"\
  328. "movq %%mm4, %%mm5 \n\t"\
  329. "punpcklbw %%mm7, %%mm2 \n\t"\
  330. "punpckhbw %%mm7, %%mm3 \n\t"\
  331. "punpcklbw %%mm7, %%mm4 \n\t"\
  332. "punpckhbw %%mm7, %%mm5 \n\t"\
  333. "paddw %%mm4, %%mm2 \n\t"\
  334. "paddw %%mm3, %%mm5 \n\t"\
  335. "psubw %%mm2, %%mm0 \n\t"\
  336. "psubw %%mm5, %%mm1 \n\t"\
  337. "movq %6, %%mm5 \n\t"\
  338. "paddw %%mm5, %%mm0 \n\t"\
  339. "paddw %%mm5, %%mm1 \n\t"\
  340. "psraw $3, %%mm0 \n\t"\
  341. "psraw $3, %%mm1 \n\t"\
  342. "packuswb %%mm1, %%mm0 \n\t"\
  343. OP(%%mm0, (%1),%%mm5, q) \
  344. "add %3, %0 \n\t"\
  345. "add %4, %1 \n\t"\
  346. "decl %2 \n\t"\
  347. " jnz 1b \n\t"\
  348. : "+a"(src), "+c"(dst), "+m"(h)\
  349. : "d"((x86_reg)srcStride), "S"((x86_reg)dstStride), "m"(ff_pw_5), "m"(ff_pw_4)\
  350. : "memory"\
  351. );\
  352. }\
  353. \
  354. static inline void OPNAME ## cavs_qpel8or16_v1_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h){\
  355. QPEL_CAVSVNUM(QPEL_CAVSV1,OP,ff_pw_64,ff_pw_96,ff_pw_42) \
  356. }\
  357. \
  358. static inline void OPNAME ## cavs_qpel8or16_v2_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h){\
  359. QPEL_CAVSVNUM(QPEL_CAVSV2,OP,ff_pw_4,ff_pw_5,ff_pw_5) \
  360. }\
  361. \
  362. static inline void OPNAME ## cavs_qpel8or16_v3_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride, int h){\
  363. QPEL_CAVSVNUM(QPEL_CAVSV3,OP,ff_pw_64,ff_pw_96,ff_pw_42) \
  364. }\
  365. \
  366. static void OPNAME ## cavs_qpel8_v1_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  367. OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst , src , dstStride, srcStride, 8);\
  368. }\
  369. static void OPNAME ## cavs_qpel16_v1_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  370. OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst , src , dstStride, srcStride, 16);\
  371. OPNAME ## cavs_qpel8or16_v1_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  372. }\
  373. \
  374. static void OPNAME ## cavs_qpel8_v2_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  375. OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst , src , dstStride, srcStride, 8);\
  376. }\
  377. static void OPNAME ## cavs_qpel16_v2_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  378. OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst , src , dstStride, srcStride, 16);\
  379. OPNAME ## cavs_qpel8or16_v2_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  380. }\
  381. \
  382. static void OPNAME ## cavs_qpel8_v3_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  383. OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst , src , dstStride, srcStride, 8);\
  384. }\
  385. static void OPNAME ## cavs_qpel16_v3_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  386. OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst , src , dstStride, srcStride, 16);\
  387. OPNAME ## cavs_qpel8or16_v3_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  388. }\
  389. \
  390. static void OPNAME ## cavs_qpel16_h_ ## MMX(uint8_t *dst, const uint8_t *src, int dstStride, int srcStride){\
  391. OPNAME ## cavs_qpel8_h_ ## MMX(dst , src , dstStride, srcStride);\
  392. OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  393. src += 8*srcStride;\
  394. dst += 8*dstStride;\
  395. OPNAME ## cavs_qpel8_h_ ## MMX(dst , src , dstStride, srcStride);\
  396. OPNAME ## cavs_qpel8_h_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  397. }\
  398. #define CAVS_MC(OPNAME, SIZE, MMX) \
  399. static void OPNAME ## cavs_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
  400. {\
  401. OPNAME ## cavs_qpel ## SIZE ## _h_ ## MMX(dst, src, stride, stride);\
  402. }\
  403. \
  404. static void OPNAME ## cavs_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
  405. {\
  406. OPNAME ## cavs_qpel ## SIZE ## _v1_ ## MMX(dst, src, stride, stride);\
  407. }\
  408. \
  409. static void OPNAME ## cavs_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
  410. {\
  411. OPNAME ## cavs_qpel ## SIZE ## _v2_ ## MMX(dst, src, stride, stride);\
  412. }\
  413. \
  414. static void OPNAME ## cavs_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, const uint8_t *src, ptrdiff_t stride)\
  415. {\
  416. OPNAME ## cavs_qpel ## SIZE ## _v3_ ## MMX(dst, src, stride, stride);\
  417. }\
  418. #define PUT_OP(a,b,temp, size) "mov" #size " " #a ", " #b " \n\t"
  419. #define AVG_3DNOW_OP(a,b,temp, size) \
  420. "mov" #size " " #b ", " #temp " \n\t"\
  421. "pavgusb " #temp ", " #a " \n\t"\
  422. "mov" #size " " #a ", " #b " \n\t"
  423. #define AVG_MMXEXT_OP(a, b, temp, size) \
  424. "mov" #size " " #b ", " #temp " \n\t"\
  425. "pavgb " #temp ", " #a " \n\t"\
  426. "mov" #size " " #a ", " #b " \n\t"
  427. #endif /* (HAVE_MMXEXT_INLINE || HAVE_AMD3DNOW_INLINE) */
  428. #if HAVE_MMX_INLINE
  429. static void put_cavs_qpel8_mc00_mmx(uint8_t *dst, const uint8_t *src,
  430. ptrdiff_t stride)
  431. {
  432. ff_put_pixels8_mmx(dst, src, stride, 8);
  433. }
  434. static void avg_cavs_qpel8_mc00_mmx(uint8_t *dst, const uint8_t *src,
  435. ptrdiff_t stride)
  436. {
  437. ff_avg_pixels8_mmx(dst, src, stride, 8);
  438. }
  439. static void put_cavs_qpel16_mc00_mmx(uint8_t *dst, const uint8_t *src,
  440. ptrdiff_t stride)
  441. {
  442. ff_put_pixels16_mmx(dst, src, stride, 16);
  443. }
  444. static void avg_cavs_qpel16_mc00_mmx(uint8_t *dst, const uint8_t *src,
  445. ptrdiff_t stride)
  446. {
  447. ff_avg_pixels16_mmx(dst, src, stride, 16);
  448. }
  449. static av_cold void cavsdsp_init_mmx(CAVSDSPContext *c,
  450. AVCodecContext *avctx)
  451. {
  452. c->put_cavs_qpel_pixels_tab[0][0] = put_cavs_qpel16_mc00_mmx;
  453. c->put_cavs_qpel_pixels_tab[1][0] = put_cavs_qpel8_mc00_mmx;
  454. c->avg_cavs_qpel_pixels_tab[0][0] = avg_cavs_qpel16_mc00_mmx;
  455. c->avg_cavs_qpel_pixels_tab[1][0] = avg_cavs_qpel8_mc00_mmx;
  456. c->cavs_idct8_add = cavs_idct8_add_mmx;
  457. c->idct_perm = FF_IDCT_PERM_TRANSPOSE;
  458. }
  459. #endif /* HAVE_MMX_INLINE */
  460. #define DSPFUNC(PFX, IDX, NUM, EXT) \
  461. c->PFX ## _cavs_qpel_pixels_tab[IDX][ 2] = PFX ## _cavs_qpel ## NUM ## _mc20_ ## EXT; \
  462. c->PFX ## _cavs_qpel_pixels_tab[IDX][ 4] = PFX ## _cavs_qpel ## NUM ## _mc01_ ## EXT; \
  463. c->PFX ## _cavs_qpel_pixels_tab[IDX][ 8] = PFX ## _cavs_qpel ## NUM ## _mc02_ ## EXT; \
  464. c->PFX ## _cavs_qpel_pixels_tab[IDX][12] = PFX ## _cavs_qpel ## NUM ## _mc03_ ## EXT; \
  465. #if HAVE_MMXEXT_INLINE
  466. QPEL_CAVS(put_, PUT_OP, mmxext)
  467. QPEL_CAVS(avg_, AVG_MMXEXT_OP, mmxext)
  468. CAVS_MC(put_, 8, mmxext)
  469. CAVS_MC(put_, 16, mmxext)
  470. CAVS_MC(avg_, 8, mmxext)
  471. CAVS_MC(avg_, 16, mmxext)
  472. static av_cold void cavsdsp_init_mmxext(CAVSDSPContext *c,
  473. AVCodecContext *avctx)
  474. {
  475. DSPFUNC(put, 0, 16, mmxext);
  476. DSPFUNC(put, 1, 8, mmxext);
  477. DSPFUNC(avg, 0, 16, mmxext);
  478. DSPFUNC(avg, 1, 8, mmxext);
  479. }
  480. #endif /* HAVE_MMXEXT_INLINE */
  481. #if HAVE_AMD3DNOW_INLINE
  482. QPEL_CAVS(put_, PUT_OP, 3dnow)
  483. QPEL_CAVS(avg_, AVG_3DNOW_OP, 3dnow)
  484. CAVS_MC(put_, 8, 3dnow)
  485. CAVS_MC(put_, 16,3dnow)
  486. CAVS_MC(avg_, 8, 3dnow)
  487. CAVS_MC(avg_, 16,3dnow)
  488. static av_cold void cavsdsp_init_3dnow(CAVSDSPContext *c,
  489. AVCodecContext *avctx)
  490. {
  491. DSPFUNC(put, 0, 16, 3dnow);
  492. DSPFUNC(put, 1, 8, 3dnow);
  493. DSPFUNC(avg, 0, 16, 3dnow);
  494. DSPFUNC(avg, 1, 8, 3dnow);
  495. }
  496. #endif /* HAVE_AMD3DNOW_INLINE */
  497. av_cold void ff_cavsdsp_init_x86(CAVSDSPContext *c, AVCodecContext *avctx)
  498. {
  499. #if HAVE_MMX_INLINE
  500. int cpu_flags = av_get_cpu_flags();
  501. if (INLINE_MMX(cpu_flags))
  502. cavsdsp_init_mmx(c, avctx);
  503. #endif /* HAVE_MMX_INLINE */
  504. #if HAVE_AMD3DNOW_INLINE
  505. if (INLINE_AMD3DNOW(cpu_flags))
  506. cavsdsp_init_3dnow(c, avctx);
  507. #endif /* HAVE_AMD3DNOW_INLINE */
  508. #if HAVE_MMXEXT_INLINE
  509. if (INLINE_MMXEXT(cpu_flags))
  510. cavsdsp_init_mmxext(c, avctx);
  511. #endif /* HAVE_MMXEXT_INLINE */
  512. }