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.

1518 lines
58KB

  1. /*
  2. * Copyright (c) 2004-2005 Michael Niedermayer, Loren Merritt
  3. *
  4. * This file is part of FFmpeg.
  5. *
  6. * FFmpeg is free software; you can redistribute it and/or
  7. * modify it under the terms of the GNU Lesser General Public
  8. * License as published by the Free Software Foundation; either
  9. * version 2.1 of the License, or (at your option) any later version.
  10. *
  11. * FFmpeg is distributed in the hope that it will be useful,
  12. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  14. * Lesser General Public License for more details.
  15. *
  16. * You should have received a copy of the GNU Lesser General Public
  17. * License along with FFmpeg; if not, write to the Free Software
  18. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19. */
  20. /***********************************/
  21. /* IDCT */
  22. /* in/out: mma=mma+mmb, mmb=mmb-mma */
  23. #define SUMSUB_BA( a, b ) \
  24. "paddw "#b", "#a" \n\t"\
  25. "paddw "#b", "#b" \n\t"\
  26. "psubw "#a", "#b" \n\t"
  27. #define SUMSUB_BADC( a, b, c, d ) \
  28. "paddw "#b", "#a" \n\t"\
  29. "paddw "#d", "#c" \n\t"\
  30. "paddw "#b", "#b" \n\t"\
  31. "paddw "#d", "#d" \n\t"\
  32. "psubw "#a", "#b" \n\t"\
  33. "psubw "#c", "#d" \n\t"
  34. #define SUMSUBD2_AB( a, b, t ) \
  35. "movq "#b", "#t" \n\t"\
  36. "psraw $1 , "#b" \n\t"\
  37. "paddw "#a", "#b" \n\t"\
  38. "psraw $1 , "#a" \n\t"\
  39. "psubw "#t", "#a" \n\t"
  40. #define IDCT4_1D( s02, s13, d02, d13, t ) \
  41. SUMSUB_BA ( s02, d02 )\
  42. SUMSUBD2_AB( s13, d13, t )\
  43. SUMSUB_BADC( d13, s02, s13, d02 )
  44. #define TRANSPOSE4(a,b,c,d,t)\
  45. SBUTTERFLY(a,b,t,wd) /* a=aebf t=cgdh */\
  46. SBUTTERFLY(c,d,b,wd) /* c=imjn b=kolp */\
  47. SBUTTERFLY(a,c,d,dq) /* a=aeim d=bfjn */\
  48. SBUTTERFLY(t,b,c,dq) /* t=cgko c=dhlp */
  49. #define STORE_DIFF_4P( p, t, z ) \
  50. "psraw $6, "#p" \n\t"\
  51. "movd (%0), "#t" \n\t"\
  52. "punpcklbw "#z", "#t" \n\t"\
  53. "paddsw "#t", "#p" \n\t"\
  54. "packuswb "#z", "#p" \n\t"\
  55. "movd "#p", (%0) \n\t"
  56. static void ff_h264_idct_add_mmx(uint8_t *dst, int16_t *block, int stride)
  57. {
  58. /* Load dct coeffs */
  59. asm volatile(
  60. "movq (%0), %%mm0 \n\t"
  61. "movq 8(%0), %%mm1 \n\t"
  62. "movq 16(%0), %%mm2 \n\t"
  63. "movq 24(%0), %%mm3 \n\t"
  64. :: "r"(block) );
  65. asm volatile(
  66. /* mm1=s02+s13 mm2=s02-s13 mm4=d02+d13 mm0=d02-d13 */
  67. IDCT4_1D( %%mm2, %%mm1, %%mm0, %%mm3, %%mm4 )
  68. "movq %0, %%mm6 \n\t"
  69. /* in: 1,4,0,2 out: 1,2,3,0 */
  70. TRANSPOSE4( %%mm3, %%mm1, %%mm0, %%mm2, %%mm4 )
  71. "paddw %%mm6, %%mm3 \n\t"
  72. /* mm2=s02+s13 mm3=s02-s13 mm4=d02+d13 mm1=d02-d13 */
  73. IDCT4_1D( %%mm4, %%mm2, %%mm3, %%mm0, %%mm1 )
  74. "pxor %%mm7, %%mm7 \n\t"
  75. :: "m"(ff_pw_32));
  76. asm volatile(
  77. STORE_DIFF_4P( %%mm0, %%mm1, %%mm7)
  78. "add %1, %0 \n\t"
  79. STORE_DIFF_4P( %%mm2, %%mm1, %%mm7)
  80. "add %1, %0 \n\t"
  81. STORE_DIFF_4P( %%mm3, %%mm1, %%mm7)
  82. "add %1, %0 \n\t"
  83. STORE_DIFF_4P( %%mm4, %%mm1, %%mm7)
  84. : "+r"(dst)
  85. : "r" ((long)stride)
  86. );
  87. }
  88. static inline void h264_idct8_1d(int16_t *block)
  89. {
  90. asm volatile(
  91. "movq 112(%0), %%mm7 \n\t"
  92. "movq 80(%0), %%mm5 \n\t"
  93. "movq 48(%0), %%mm3 \n\t"
  94. "movq 16(%0), %%mm1 \n\t"
  95. "movq %%mm7, %%mm4 \n\t"
  96. "movq %%mm3, %%mm6 \n\t"
  97. "movq %%mm5, %%mm0 \n\t"
  98. "movq %%mm7, %%mm2 \n\t"
  99. "psraw $1, %%mm4 \n\t"
  100. "psraw $1, %%mm6 \n\t"
  101. "psubw %%mm7, %%mm0 \n\t"
  102. "psubw %%mm6, %%mm2 \n\t"
  103. "psubw %%mm4, %%mm0 \n\t"
  104. "psubw %%mm3, %%mm2 \n\t"
  105. "psubw %%mm3, %%mm0 \n\t"
  106. "paddw %%mm1, %%mm2 \n\t"
  107. "movq %%mm5, %%mm4 \n\t"
  108. "movq %%mm1, %%mm6 \n\t"
  109. "psraw $1, %%mm4 \n\t"
  110. "psraw $1, %%mm6 \n\t"
  111. "paddw %%mm5, %%mm4 \n\t"
  112. "paddw %%mm1, %%mm6 \n\t"
  113. "paddw %%mm7, %%mm4 \n\t"
  114. "paddw %%mm5, %%mm6 \n\t"
  115. "psubw %%mm1, %%mm4 \n\t"
  116. "paddw %%mm3, %%mm6 \n\t"
  117. "movq %%mm0, %%mm1 \n\t"
  118. "movq %%mm4, %%mm3 \n\t"
  119. "movq %%mm2, %%mm5 \n\t"
  120. "movq %%mm6, %%mm7 \n\t"
  121. "psraw $2, %%mm6 \n\t"
  122. "psraw $2, %%mm3 \n\t"
  123. "psraw $2, %%mm5 \n\t"
  124. "psraw $2, %%mm0 \n\t"
  125. "paddw %%mm6, %%mm1 \n\t"
  126. "paddw %%mm2, %%mm3 \n\t"
  127. "psubw %%mm4, %%mm5 \n\t"
  128. "psubw %%mm0, %%mm7 \n\t"
  129. "movq 32(%0), %%mm2 \n\t"
  130. "movq 96(%0), %%mm6 \n\t"
  131. "movq %%mm2, %%mm4 \n\t"
  132. "movq %%mm6, %%mm0 \n\t"
  133. "psraw $1, %%mm4 \n\t"
  134. "psraw $1, %%mm6 \n\t"
  135. "psubw %%mm0, %%mm4 \n\t"
  136. "paddw %%mm2, %%mm6 \n\t"
  137. "movq (%0), %%mm2 \n\t"
  138. "movq 64(%0), %%mm0 \n\t"
  139. SUMSUB_BA( %%mm0, %%mm2 )
  140. SUMSUB_BA( %%mm6, %%mm0 )
  141. SUMSUB_BA( %%mm4, %%mm2 )
  142. SUMSUB_BA( %%mm7, %%mm6 )
  143. SUMSUB_BA( %%mm5, %%mm4 )
  144. SUMSUB_BA( %%mm3, %%mm2 )
  145. SUMSUB_BA( %%mm1, %%mm0 )
  146. :: "r"(block)
  147. );
  148. }
  149. static void ff_h264_idct8_add_mmx(uint8_t *dst, int16_t *block, int stride)
  150. {
  151. int i;
  152. int16_t __attribute__ ((aligned(8))) b2[64];
  153. block[0] += 32;
  154. for(i=0; i<2; i++){
  155. DECLARE_ALIGNED_8(uint64_t, tmp);
  156. h264_idct8_1d(block+4*i);
  157. asm volatile(
  158. "movq %%mm7, %0 \n\t"
  159. TRANSPOSE4( %%mm0, %%mm2, %%mm4, %%mm6, %%mm7 )
  160. "movq %%mm0, 8(%1) \n\t"
  161. "movq %%mm6, 24(%1) \n\t"
  162. "movq %%mm7, 40(%1) \n\t"
  163. "movq %%mm4, 56(%1) \n\t"
  164. "movq %0, %%mm7 \n\t"
  165. TRANSPOSE4( %%mm7, %%mm5, %%mm3, %%mm1, %%mm0 )
  166. "movq %%mm7, (%1) \n\t"
  167. "movq %%mm1, 16(%1) \n\t"
  168. "movq %%mm0, 32(%1) \n\t"
  169. "movq %%mm3, 48(%1) \n\t"
  170. : "=m"(tmp)
  171. : "r"(b2+32*i)
  172. : "memory"
  173. );
  174. }
  175. for(i=0; i<2; i++){
  176. h264_idct8_1d(b2+4*i);
  177. asm volatile(
  178. "psraw $6, %%mm7 \n\t"
  179. "psraw $6, %%mm6 \n\t"
  180. "psraw $6, %%mm5 \n\t"
  181. "psraw $6, %%mm4 \n\t"
  182. "psraw $6, %%mm3 \n\t"
  183. "psraw $6, %%mm2 \n\t"
  184. "psraw $6, %%mm1 \n\t"
  185. "psraw $6, %%mm0 \n\t"
  186. "movq %%mm7, (%0) \n\t"
  187. "movq %%mm5, 16(%0) \n\t"
  188. "movq %%mm3, 32(%0) \n\t"
  189. "movq %%mm1, 48(%0) \n\t"
  190. "movq %%mm0, 64(%0) \n\t"
  191. "movq %%mm2, 80(%0) \n\t"
  192. "movq %%mm4, 96(%0) \n\t"
  193. "movq %%mm6, 112(%0) \n\t"
  194. :: "r"(b2+4*i)
  195. : "memory"
  196. );
  197. }
  198. add_pixels_clamped_mmx(b2, dst, stride);
  199. }
  200. static void ff_h264_idct_dc_add_mmx2(uint8_t *dst, int16_t *block, int stride)
  201. {
  202. int dc = (block[0] + 32) >> 6;
  203. asm volatile(
  204. "movd %0, %%mm0 \n\t"
  205. "pshufw $0, %%mm0, %%mm0 \n\t"
  206. "pxor %%mm1, %%mm1 \n\t"
  207. "psubw %%mm0, %%mm1 \n\t"
  208. "packuswb %%mm0, %%mm0 \n\t"
  209. "packuswb %%mm1, %%mm1 \n\t"
  210. ::"r"(dc)
  211. );
  212. asm volatile(
  213. "movd %0, %%mm2 \n\t"
  214. "movd %1, %%mm3 \n\t"
  215. "movd %2, %%mm4 \n\t"
  216. "movd %3, %%mm5 \n\t"
  217. "paddusb %%mm0, %%mm2 \n\t"
  218. "paddusb %%mm0, %%mm3 \n\t"
  219. "paddusb %%mm0, %%mm4 \n\t"
  220. "paddusb %%mm0, %%mm5 \n\t"
  221. "psubusb %%mm1, %%mm2 \n\t"
  222. "psubusb %%mm1, %%mm3 \n\t"
  223. "psubusb %%mm1, %%mm4 \n\t"
  224. "psubusb %%mm1, %%mm5 \n\t"
  225. "movd %%mm2, %0 \n\t"
  226. "movd %%mm3, %1 \n\t"
  227. "movd %%mm4, %2 \n\t"
  228. "movd %%mm5, %3 \n\t"
  229. :"+m"(*(uint32_t*)(dst+0*stride)),
  230. "+m"(*(uint32_t*)(dst+1*stride)),
  231. "+m"(*(uint32_t*)(dst+2*stride)),
  232. "+m"(*(uint32_t*)(dst+3*stride))
  233. );
  234. }
  235. static void ff_h264_idct8_dc_add_mmx2(uint8_t *dst, int16_t *block, int stride)
  236. {
  237. int dc = (block[0] + 32) >> 6;
  238. int y;
  239. asm volatile(
  240. "movd %0, %%mm0 \n\t"
  241. "pshufw $0, %%mm0, %%mm0 \n\t"
  242. "pxor %%mm1, %%mm1 \n\t"
  243. "psubw %%mm0, %%mm1 \n\t"
  244. "packuswb %%mm0, %%mm0 \n\t"
  245. "packuswb %%mm1, %%mm1 \n\t"
  246. ::"r"(dc)
  247. );
  248. for(y=2; y--; dst += 4*stride){
  249. asm volatile(
  250. "movq %0, %%mm2 \n\t"
  251. "movq %1, %%mm3 \n\t"
  252. "movq %2, %%mm4 \n\t"
  253. "movq %3, %%mm5 \n\t"
  254. "paddusb %%mm0, %%mm2 \n\t"
  255. "paddusb %%mm0, %%mm3 \n\t"
  256. "paddusb %%mm0, %%mm4 \n\t"
  257. "paddusb %%mm0, %%mm5 \n\t"
  258. "psubusb %%mm1, %%mm2 \n\t"
  259. "psubusb %%mm1, %%mm3 \n\t"
  260. "psubusb %%mm1, %%mm4 \n\t"
  261. "psubusb %%mm1, %%mm5 \n\t"
  262. "movq %%mm2, %0 \n\t"
  263. "movq %%mm3, %1 \n\t"
  264. "movq %%mm4, %2 \n\t"
  265. "movq %%mm5, %3 \n\t"
  266. :"+m"(*(uint64_t*)(dst+0*stride)),
  267. "+m"(*(uint64_t*)(dst+1*stride)),
  268. "+m"(*(uint64_t*)(dst+2*stride)),
  269. "+m"(*(uint64_t*)(dst+3*stride))
  270. );
  271. }
  272. }
  273. /***********************************/
  274. /* deblocking */
  275. // out: o = |x-y|>a
  276. // clobbers: t
  277. #define DIFF_GT_MMX(x,y,a,o,t)\
  278. "movq "#y", "#t" \n\t"\
  279. "movq "#x", "#o" \n\t"\
  280. "psubusb "#x", "#t" \n\t"\
  281. "psubusb "#y", "#o" \n\t"\
  282. "por "#t", "#o" \n\t"\
  283. "psubusb "#a", "#o" \n\t"
  284. // out: o = |x-y|>a
  285. // clobbers: t
  286. #define DIFF_GT2_MMX(x,y,a,o,t)\
  287. "movq "#y", "#t" \n\t"\
  288. "movq "#x", "#o" \n\t"\
  289. "psubusb "#x", "#t" \n\t"\
  290. "psubusb "#y", "#o" \n\t"\
  291. "psubusb "#a", "#t" \n\t"\
  292. "psubusb "#a", "#o" \n\t"\
  293. "pcmpeqb "#t", "#o" \n\t"\
  294. // in: mm0=p1 mm1=p0 mm2=q0 mm3=q1
  295. // out: mm5=beta-1, mm7=mask
  296. // clobbers: mm4,mm6
  297. #define H264_DEBLOCK_MASK(alpha1, beta1) \
  298. "pshufw $0, "#alpha1", %%mm4 \n\t"\
  299. "pshufw $0, "#beta1 ", %%mm5 \n\t"\
  300. "packuswb %%mm4, %%mm4 \n\t"\
  301. "packuswb %%mm5, %%mm5 \n\t"\
  302. DIFF_GT_MMX(%%mm1, %%mm2, %%mm4, %%mm7, %%mm6) /* |p0-q0| > alpha-1 */\
  303. DIFF_GT_MMX(%%mm0, %%mm1, %%mm5, %%mm4, %%mm6) /* |p1-p0| > beta-1 */\
  304. "por %%mm4, %%mm7 \n\t"\
  305. DIFF_GT_MMX(%%mm3, %%mm2, %%mm5, %%mm4, %%mm6) /* |q1-q0| > beta-1 */\
  306. "por %%mm4, %%mm7 \n\t"\
  307. "pxor %%mm6, %%mm6 \n\t"\
  308. "pcmpeqb %%mm6, %%mm7 \n\t"
  309. // in: mm0=p1 mm1=p0 mm2=q0 mm3=q1 mm7=(tc&mask)
  310. // out: mm1=p0' mm2=q0'
  311. // clobbers: mm0,3-6
  312. #define H264_DEBLOCK_P0_Q0(pb_01, pb_3f)\
  313. "movq %%mm1 , %%mm5 \n\t"\
  314. "pxor %%mm2 , %%mm5 \n\t" /* p0^q0*/\
  315. "pand "#pb_01" , %%mm5 \n\t" /* (p0^q0)&1*/\
  316. "pcmpeqb %%mm4 , %%mm4 \n\t"\
  317. "pxor %%mm4 , %%mm3 \n\t"\
  318. "pavgb %%mm0 , %%mm3 \n\t" /* (p1 - q1 + 256)>>1*/\
  319. "pavgb "MANGLE(ff_pb_3)" , %%mm3 \n\t" /*(((p1 - q1 + 256)>>1)+4)>>1 = 64+2+(p1-q1)>>2*/\
  320. "pxor %%mm1 , %%mm4 \n\t"\
  321. "pavgb %%mm2 , %%mm4 \n\t" /* (q0 - p0 + 256)>>1*/\
  322. "pavgb %%mm5 , %%mm3 \n\t"\
  323. "paddusb %%mm4 , %%mm3 \n\t" /* d+128+33*/\
  324. "movq "MANGLE(ff_pb_A1)" , %%mm6 \n\t"\
  325. "psubusb %%mm3 , %%mm6 \n\t"\
  326. "psubusb "MANGLE(ff_pb_A1)" , %%mm3 \n\t"\
  327. "pminub %%mm7 , %%mm6 \n\t"\
  328. "pminub %%mm7 , %%mm3 \n\t"\
  329. "psubusb %%mm6 , %%mm1 \n\t"\
  330. "psubusb %%mm3 , %%mm2 \n\t"\
  331. "paddusb %%mm3 , %%mm1 \n\t"\
  332. "paddusb %%mm6 , %%mm2 \n\t"
  333. // in: mm0=p1 mm1=p0 mm2=q0 mm3=q1 mm7=(tc&mask) %8=mm_bone
  334. // out: (q1addr) = clip( (q2+((p0+q0+1)>>1))>>1, q1-tc0, q1+tc0 )
  335. // clobbers: q2, tmp, tc0
  336. #define H264_DEBLOCK_Q1(p1, q2, q2addr, q1addr, tc0, tmp)\
  337. "movq %%mm1, "#tmp" \n\t"\
  338. "pavgb %%mm2, "#tmp" \n\t"\
  339. "pavgb "#tmp", "#q2" \n\t" /* avg(p2,avg(p0,q0)) */\
  340. "pxor "q2addr", "#tmp" \n\t"\
  341. "pand %8, "#tmp" \n\t" /* (p2^avg(p0,q0))&1 */\
  342. "psubusb "#tmp", "#q2" \n\t" /* (p2+((p0+q0+1)>>1))>>1 */\
  343. "movq "#p1", "#tmp" \n\t"\
  344. "psubusb "#tc0", "#tmp" \n\t"\
  345. "paddusb "#p1", "#tc0" \n\t"\
  346. "pmaxub "#tmp", "#q2" \n\t"\
  347. "pminub "#tc0", "#q2" \n\t"\
  348. "movq "#q2", "q1addr" \n\t"
  349. static inline void h264_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha1, int beta1, int8_t *tc0)
  350. {
  351. DECLARE_ALIGNED_8(uint64_t, tmp0[2]);
  352. asm volatile(
  353. "movq (%1,%3), %%mm0 \n\t" //p1
  354. "movq (%1,%3,2), %%mm1 \n\t" //p0
  355. "movq (%2), %%mm2 \n\t" //q0
  356. "movq (%2,%3), %%mm3 \n\t" //q1
  357. H264_DEBLOCK_MASK(%6, %7)
  358. "movd %5, %%mm4 \n\t"
  359. "punpcklbw %%mm4, %%mm4 \n\t"
  360. "punpcklwd %%mm4, %%mm4 \n\t"
  361. "pcmpeqb %%mm3, %%mm3 \n\t"
  362. "movq %%mm4, %%mm6 \n\t"
  363. "pcmpgtb %%mm3, %%mm4 \n\t"
  364. "movq %%mm6, 8+%0 \n\t"
  365. "pand %%mm4, %%mm7 \n\t"
  366. "movq %%mm7, %0 \n\t"
  367. /* filter p1 */
  368. "movq (%1), %%mm3 \n\t" //p2
  369. DIFF_GT2_MMX(%%mm1, %%mm3, %%mm5, %%mm6, %%mm4) // |p2-p0|>beta-1
  370. "pand %%mm7, %%mm6 \n\t" // mask & |p2-p0|<beta
  371. "pand 8+%0, %%mm7 \n\t" // mask & tc0
  372. "movq %%mm7, %%mm4 \n\t"
  373. "psubb %%mm6, %%mm7 \n\t"
  374. "pand %%mm4, %%mm6 \n\t" // mask & |p2-p0|<beta & tc0
  375. H264_DEBLOCK_Q1(%%mm0, %%mm3, "(%1)", "(%1,%3)", %%mm6, %%mm4)
  376. /* filter q1 */
  377. "movq (%2,%3,2), %%mm4 \n\t" //q2
  378. DIFF_GT2_MMX(%%mm2, %%mm4, %%mm5, %%mm6, %%mm3) // |q2-q0|>beta-1
  379. "pand %0, %%mm6 \n\t"
  380. "movq 8+%0, %%mm5 \n\t" // can be merged with the and below but is slower then
  381. "pand %%mm6, %%mm5 \n\t"
  382. "psubb %%mm6, %%mm7 \n\t"
  383. "movq (%2,%3), %%mm3 \n\t"
  384. H264_DEBLOCK_Q1(%%mm3, %%mm4, "(%2,%3,2)", "(%2,%3)", %%mm5, %%mm6)
  385. /* filter p0, q0 */
  386. H264_DEBLOCK_P0_Q0(%8, unused)
  387. "movq %%mm1, (%1,%3,2) \n\t"
  388. "movq %%mm2, (%2) \n\t"
  389. : "=m"(*tmp0)
  390. : "r"(pix-3*stride), "r"(pix), "r"((long)stride),
  391. "m"(*tmp0/*unused*/), "m"(*(uint32_t*)tc0), "m"(alpha1), "m"(beta1),
  392. "m"(mm_bone)
  393. );
  394. }
  395. static void h264_v_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  396. {
  397. if((tc0[0] & tc0[1]) >= 0)
  398. h264_loop_filter_luma_mmx2(pix, stride, alpha-1, beta-1, tc0);
  399. if((tc0[2] & tc0[3]) >= 0)
  400. h264_loop_filter_luma_mmx2(pix+8, stride, alpha-1, beta-1, tc0+2);
  401. }
  402. static void h264_h_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  403. {
  404. //FIXME: could cut some load/stores by merging transpose with filter
  405. // also, it only needs to transpose 6x8
  406. DECLARE_ALIGNED_8(uint8_t, trans[8*8]);
  407. int i;
  408. for(i=0; i<2; i++, pix+=8*stride, tc0+=2) {
  409. if((tc0[0] & tc0[1]) < 0)
  410. continue;
  411. transpose4x4(trans, pix-4, 8, stride);
  412. transpose4x4(trans +4*8, pix, 8, stride);
  413. transpose4x4(trans+4, pix-4+4*stride, 8, stride);
  414. transpose4x4(trans+4+4*8, pix +4*stride, 8, stride);
  415. h264_loop_filter_luma_mmx2(trans+4*8, 8, alpha-1, beta-1, tc0);
  416. transpose4x4(pix-2, trans +2*8, stride, 8);
  417. transpose4x4(pix-2+4*stride, trans+4+2*8, stride, 8);
  418. }
  419. }
  420. static inline void h264_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha1, int beta1, int8_t *tc0)
  421. {
  422. asm volatile(
  423. "movq (%0), %%mm0 \n\t" //p1
  424. "movq (%0,%2), %%mm1 \n\t" //p0
  425. "movq (%1), %%mm2 \n\t" //q0
  426. "movq (%1,%2), %%mm3 \n\t" //q1
  427. H264_DEBLOCK_MASK(%4, %5)
  428. "movd %3, %%mm6 \n\t"
  429. "punpcklbw %%mm6, %%mm6 \n\t"
  430. "pand %%mm6, %%mm7 \n\t" // mm7 = tc&mask
  431. H264_DEBLOCK_P0_Q0(%6, %7)
  432. "movq %%mm1, (%0,%2) \n\t"
  433. "movq %%mm2, (%1) \n\t"
  434. :: "r"(pix-2*stride), "r"(pix), "r"((long)stride),
  435. "r"(*(uint32_t*)tc0),
  436. "m"(alpha1), "m"(beta1), "m"(mm_bone), "m"(ff_pb_3F)
  437. );
  438. }
  439. static void h264_v_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  440. {
  441. h264_loop_filter_chroma_mmx2(pix, stride, alpha-1, beta-1, tc0);
  442. }
  443. static void h264_h_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  444. {
  445. //FIXME: could cut some load/stores by merging transpose with filter
  446. DECLARE_ALIGNED_8(uint8_t, trans[8*4]);
  447. transpose4x4(trans, pix-2, 8, stride);
  448. transpose4x4(trans+4, pix-2+4*stride, 8, stride);
  449. h264_loop_filter_chroma_mmx2(trans+2*8, 8, alpha-1, beta-1, tc0);
  450. transpose4x4(pix-2, trans, stride, 8);
  451. transpose4x4(pix-2+4*stride, trans+4, stride, 8);
  452. }
  453. // p0 = (p0 + q1 + 2*p1 + 2) >> 2
  454. #define H264_FILTER_CHROMA4(p0, p1, q1, one) \
  455. "movq "#p0", %%mm4 \n\t"\
  456. "pxor "#q1", %%mm4 \n\t"\
  457. "pand "#one", %%mm4 \n\t" /* mm4 = (p0^q1)&1 */\
  458. "pavgb "#q1", "#p0" \n\t"\
  459. "psubusb %%mm4, "#p0" \n\t"\
  460. "pavgb "#p1", "#p0" \n\t" /* dst = avg(p1, avg(p0,q1) - ((p0^q1)&1)) */\
  461. static inline void h264_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha1, int beta1)
  462. {
  463. asm volatile(
  464. "movq (%0), %%mm0 \n\t"
  465. "movq (%0,%2), %%mm1 \n\t"
  466. "movq (%1), %%mm2 \n\t"
  467. "movq (%1,%2), %%mm3 \n\t"
  468. H264_DEBLOCK_MASK(%3, %4)
  469. "movq %%mm1, %%mm5 \n\t"
  470. "movq %%mm2, %%mm6 \n\t"
  471. H264_FILTER_CHROMA4(%%mm1, %%mm0, %%mm3, %5) //p0'
  472. H264_FILTER_CHROMA4(%%mm2, %%mm3, %%mm0, %5) //q0'
  473. "psubb %%mm5, %%mm1 \n\t"
  474. "psubb %%mm6, %%mm2 \n\t"
  475. "pand %%mm7, %%mm1 \n\t"
  476. "pand %%mm7, %%mm2 \n\t"
  477. "paddb %%mm5, %%mm1 \n\t"
  478. "paddb %%mm6, %%mm2 \n\t"
  479. "movq %%mm1, (%0,%2) \n\t"
  480. "movq %%mm2, (%1) \n\t"
  481. :: "r"(pix-2*stride), "r"(pix), "r"((long)stride),
  482. "m"(alpha1), "m"(beta1), "m"(mm_bone)
  483. );
  484. }
  485. static void h264_v_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha, int beta)
  486. {
  487. h264_loop_filter_chroma_intra_mmx2(pix, stride, alpha-1, beta-1);
  488. }
  489. static void h264_h_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha, int beta)
  490. {
  491. //FIXME: could cut some load/stores by merging transpose with filter
  492. DECLARE_ALIGNED_8(uint8_t, trans[8*4]);
  493. transpose4x4(trans, pix-2, 8, stride);
  494. transpose4x4(trans+4, pix-2+4*stride, 8, stride);
  495. h264_loop_filter_chroma_intra_mmx2(trans+2*8, 8, alpha-1, beta-1);
  496. transpose4x4(pix-2, trans, stride, 8);
  497. transpose4x4(pix-2+4*stride, trans+4, stride, 8);
  498. }
  499. static void h264_loop_filter_strength_mmx2( int16_t bS[2][4][4], uint8_t nnz[40], int8_t ref[2][40], int16_t mv[2][40][2],
  500. int bidir, int edges, int step, int mask_mv0, int mask_mv1 ) {
  501. int dir;
  502. asm volatile(
  503. "pxor %%mm7, %%mm7 \n\t"
  504. "movq %0, %%mm6 \n\t"
  505. "movq %1, %%mm5 \n\t"
  506. "movq %2, %%mm4 \n\t"
  507. ::"m"(ff_pb_1), "m"(ff_pb_3), "m"(ff_pb_7)
  508. );
  509. // could do a special case for dir==0 && edges==1, but it only reduces the
  510. // average filter time by 1.2%
  511. for( dir=1; dir>=0; dir-- ) {
  512. const int d_idx = dir ? -8 : -1;
  513. const int mask_mv = dir ? mask_mv1 : mask_mv0;
  514. DECLARE_ALIGNED_8(const uint64_t, mask_dir) = dir ? 0 : 0xffffffffffffffffULL;
  515. int b_idx, edge, l;
  516. for( b_idx=12, edge=0; edge<edges; edge+=step, b_idx+=8*step ) {
  517. asm volatile(
  518. "pand %0, %%mm0 \n\t"
  519. ::"m"(mask_dir)
  520. );
  521. if(!(mask_mv & edge)) {
  522. asm volatile("pxor %%mm0, %%mm0 \n\t":);
  523. for( l = bidir; l >= 0; l-- ) {
  524. asm volatile(
  525. "movd %0, %%mm1 \n\t"
  526. "punpckldq %1, %%mm1 \n\t"
  527. "movq %%mm1, %%mm2 \n\t"
  528. "psrlw $7, %%mm2 \n\t"
  529. "pand %%mm6, %%mm2 \n\t"
  530. "por %%mm2, %%mm1 \n\t" // ref_cache with -2 mapped to -1
  531. "punpckldq %%mm1, %%mm2 \n\t"
  532. "pcmpeqb %%mm2, %%mm1 \n\t"
  533. "paddb %%mm6, %%mm1 \n\t"
  534. "punpckhbw %%mm7, %%mm1 \n\t" // ref[b] != ref[bn]
  535. "por %%mm1, %%mm0 \n\t"
  536. "movq %2, %%mm1 \n\t"
  537. "movq %3, %%mm2 \n\t"
  538. "psubw %4, %%mm1 \n\t"
  539. "psubw %5, %%mm2 \n\t"
  540. "packsswb %%mm2, %%mm1 \n\t"
  541. "paddb %%mm5, %%mm1 \n\t"
  542. "pminub %%mm4, %%mm1 \n\t"
  543. "pcmpeqb %%mm4, %%mm1 \n\t" // abs(mv[b] - mv[bn]) >= limit
  544. "por %%mm1, %%mm0 \n\t"
  545. ::"m"(ref[l][b_idx]),
  546. "m"(ref[l][b_idx+d_idx]),
  547. "m"(mv[l][b_idx][0]),
  548. "m"(mv[l][b_idx+2][0]),
  549. "m"(mv[l][b_idx+d_idx][0]),
  550. "m"(mv[l][b_idx+d_idx+2][0])
  551. );
  552. }
  553. }
  554. asm volatile(
  555. "movd %0, %%mm1 \n\t"
  556. "por %1, %%mm1 \n\t"
  557. "punpcklbw %%mm7, %%mm1 \n\t"
  558. "pcmpgtw %%mm7, %%mm1 \n\t" // nnz[b] || nnz[bn]
  559. ::"m"(nnz[b_idx]),
  560. "m"(nnz[b_idx+d_idx])
  561. );
  562. asm volatile(
  563. "pcmpeqw %%mm7, %%mm0 \n\t"
  564. "pcmpeqw %%mm7, %%mm0 \n\t"
  565. "psrlw $15, %%mm0 \n\t" // nonzero -> 1
  566. "psrlw $14, %%mm1 \n\t"
  567. "movq %%mm0, %%mm2 \n\t"
  568. "por %%mm1, %%mm2 \n\t"
  569. "psrlw $1, %%mm1 \n\t"
  570. "pandn %%mm2, %%mm1 \n\t"
  571. "movq %%mm1, %0 \n\t"
  572. :"=m"(*bS[dir][edge])
  573. ::"memory"
  574. );
  575. }
  576. edges = 4;
  577. step = 1;
  578. }
  579. asm volatile(
  580. "movq (%0), %%mm0 \n\t"
  581. "movq 8(%0), %%mm1 \n\t"
  582. "movq 16(%0), %%mm2 \n\t"
  583. "movq 24(%0), %%mm3 \n\t"
  584. TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4)
  585. "movq %%mm0, (%0) \n\t"
  586. "movq %%mm3, 8(%0) \n\t"
  587. "movq %%mm4, 16(%0) \n\t"
  588. "movq %%mm2, 24(%0) \n\t"
  589. ::"r"(bS[0])
  590. :"memory"
  591. );
  592. }
  593. /***********************************/
  594. /* motion compensation */
  595. #define QPEL_H264V(A,B,C,D,E,F,OP)\
  596. "movd (%0), "#F" \n\t"\
  597. "movq "#C", %%mm6 \n\t"\
  598. "paddw "#D", %%mm6 \n\t"\
  599. "psllw $2, %%mm6 \n\t"\
  600. "psubw "#B", %%mm6 \n\t"\
  601. "psubw "#E", %%mm6 \n\t"\
  602. "pmullw %4, %%mm6 \n\t"\
  603. "add %2, %0 \n\t"\
  604. "punpcklbw %%mm7, "#F" \n\t"\
  605. "paddw %5, "#A" \n\t"\
  606. "paddw "#F", "#A" \n\t"\
  607. "paddw "#A", %%mm6 \n\t"\
  608. "psraw $5, %%mm6 \n\t"\
  609. "packuswb %%mm6, %%mm6 \n\t"\
  610. OP(%%mm6, (%1), A, d)\
  611. "add %3, %1 \n\t"
  612. #define QPEL_H264HV(A,B,C,D,E,F,OF)\
  613. "movd (%0), "#F" \n\t"\
  614. "movq "#C", %%mm6 \n\t"\
  615. "paddw "#D", %%mm6 \n\t"\
  616. "psllw $2, %%mm6 \n\t"\
  617. "psubw "#B", %%mm6 \n\t"\
  618. "psubw "#E", %%mm6 \n\t"\
  619. "pmullw %3, %%mm6 \n\t"\
  620. "add %2, %0 \n\t"\
  621. "punpcklbw %%mm7, "#F" \n\t"\
  622. "paddw "#F", "#A" \n\t"\
  623. "paddw "#A", %%mm6 \n\t"\
  624. "movq %%mm6, "#OF"(%1) \n\t"
  625. #define QPEL_H264(OPNAME, OP, MMX)\
  626. static av_noinline void OPNAME ## h264_qpel4_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  627. int h=4;\
  628. \
  629. asm volatile(\
  630. "pxor %%mm7, %%mm7 \n\t"\
  631. "movq %5, %%mm4 \n\t"\
  632. "movq %6, %%mm5 \n\t"\
  633. "1: \n\t"\
  634. "movd -1(%0), %%mm1 \n\t"\
  635. "movd (%0), %%mm2 \n\t"\
  636. "movd 1(%0), %%mm3 \n\t"\
  637. "movd 2(%0), %%mm0 \n\t"\
  638. "punpcklbw %%mm7, %%mm1 \n\t"\
  639. "punpcklbw %%mm7, %%mm2 \n\t"\
  640. "punpcklbw %%mm7, %%mm3 \n\t"\
  641. "punpcklbw %%mm7, %%mm0 \n\t"\
  642. "paddw %%mm0, %%mm1 \n\t"\
  643. "paddw %%mm3, %%mm2 \n\t"\
  644. "movd -2(%0), %%mm0 \n\t"\
  645. "movd 3(%0), %%mm3 \n\t"\
  646. "punpcklbw %%mm7, %%mm0 \n\t"\
  647. "punpcklbw %%mm7, %%mm3 \n\t"\
  648. "paddw %%mm3, %%mm0 \n\t"\
  649. "psllw $2, %%mm2 \n\t"\
  650. "psubw %%mm1, %%mm2 \n\t"\
  651. "pmullw %%mm4, %%mm2 \n\t"\
  652. "paddw %%mm5, %%mm0 \n\t"\
  653. "paddw %%mm2, %%mm0 \n\t"\
  654. "psraw $5, %%mm0 \n\t"\
  655. "packuswb %%mm0, %%mm0 \n\t"\
  656. OP(%%mm0, (%1),%%mm6, d)\
  657. "add %3, %0 \n\t"\
  658. "add %4, %1 \n\t"\
  659. "decl %2 \n\t"\
  660. " jnz 1b \n\t"\
  661. : "+a"(src), "+c"(dst), "+m"(h)\
  662. : "d"((long)srcStride), "S"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  663. : "memory"\
  664. );\
  665. }\
  666. static av_noinline void OPNAME ## h264_qpel4_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  667. int h=4;\
  668. asm volatile(\
  669. "pxor %%mm7, %%mm7 \n\t"\
  670. "movq %0, %%mm4 \n\t"\
  671. "movq %1, %%mm5 \n\t"\
  672. :: "m"(ff_pw_5), "m"(ff_pw_16)\
  673. );\
  674. do{\
  675. asm volatile(\
  676. "movd -1(%0), %%mm1 \n\t"\
  677. "movd (%0), %%mm2 \n\t"\
  678. "movd 1(%0), %%mm3 \n\t"\
  679. "movd 2(%0), %%mm0 \n\t"\
  680. "punpcklbw %%mm7, %%mm1 \n\t"\
  681. "punpcklbw %%mm7, %%mm2 \n\t"\
  682. "punpcklbw %%mm7, %%mm3 \n\t"\
  683. "punpcklbw %%mm7, %%mm0 \n\t"\
  684. "paddw %%mm0, %%mm1 \n\t"\
  685. "paddw %%mm3, %%mm2 \n\t"\
  686. "movd -2(%0), %%mm0 \n\t"\
  687. "movd 3(%0), %%mm3 \n\t"\
  688. "punpcklbw %%mm7, %%mm0 \n\t"\
  689. "punpcklbw %%mm7, %%mm3 \n\t"\
  690. "paddw %%mm3, %%mm0 \n\t"\
  691. "psllw $2, %%mm2 \n\t"\
  692. "psubw %%mm1, %%mm2 \n\t"\
  693. "pmullw %%mm4, %%mm2 \n\t"\
  694. "paddw %%mm5, %%mm0 \n\t"\
  695. "paddw %%mm2, %%mm0 \n\t"\
  696. "movd (%2), %%mm3 \n\t"\
  697. "psraw $5, %%mm0 \n\t"\
  698. "packuswb %%mm0, %%mm0 \n\t"\
  699. PAVGB" %%mm3, %%mm0 \n\t"\
  700. OP(%%mm0, (%1),%%mm6, d)\
  701. "add %4, %0 \n\t"\
  702. "add %4, %1 \n\t"\
  703. "add %3, %2 \n\t"\
  704. : "+a"(src), "+c"(dst), "+d"(src2)\
  705. : "D"((long)src2Stride), "S"((long)dstStride)\
  706. : "memory"\
  707. );\
  708. }while(--h);\
  709. }\
  710. static av_noinline void OPNAME ## h264_qpel4_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  711. src -= 2*srcStride;\
  712. asm volatile(\
  713. "pxor %%mm7, %%mm7 \n\t"\
  714. "movd (%0), %%mm0 \n\t"\
  715. "add %2, %0 \n\t"\
  716. "movd (%0), %%mm1 \n\t"\
  717. "add %2, %0 \n\t"\
  718. "movd (%0), %%mm2 \n\t"\
  719. "add %2, %0 \n\t"\
  720. "movd (%0), %%mm3 \n\t"\
  721. "add %2, %0 \n\t"\
  722. "movd (%0), %%mm4 \n\t"\
  723. "add %2, %0 \n\t"\
  724. "punpcklbw %%mm7, %%mm0 \n\t"\
  725. "punpcklbw %%mm7, %%mm1 \n\t"\
  726. "punpcklbw %%mm7, %%mm2 \n\t"\
  727. "punpcklbw %%mm7, %%mm3 \n\t"\
  728. "punpcklbw %%mm7, %%mm4 \n\t"\
  729. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  730. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  731. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  732. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  733. \
  734. : "+a"(src), "+c"(dst)\
  735. : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  736. : "memory"\
  737. );\
  738. }\
  739. static av_noinline void OPNAME ## h264_qpel4_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  740. int h=4;\
  741. int w=3;\
  742. src -= 2*srcStride+2;\
  743. while(w--){\
  744. asm volatile(\
  745. "pxor %%mm7, %%mm7 \n\t"\
  746. "movd (%0), %%mm0 \n\t"\
  747. "add %2, %0 \n\t"\
  748. "movd (%0), %%mm1 \n\t"\
  749. "add %2, %0 \n\t"\
  750. "movd (%0), %%mm2 \n\t"\
  751. "add %2, %0 \n\t"\
  752. "movd (%0), %%mm3 \n\t"\
  753. "add %2, %0 \n\t"\
  754. "movd (%0), %%mm4 \n\t"\
  755. "add %2, %0 \n\t"\
  756. "punpcklbw %%mm7, %%mm0 \n\t"\
  757. "punpcklbw %%mm7, %%mm1 \n\t"\
  758. "punpcklbw %%mm7, %%mm2 \n\t"\
  759. "punpcklbw %%mm7, %%mm3 \n\t"\
  760. "punpcklbw %%mm7, %%mm4 \n\t"\
  761. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 0*8*3)\
  762. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 1*8*3)\
  763. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 2*8*3)\
  764. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 3*8*3)\
  765. \
  766. : "+a"(src)\
  767. : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
  768. : "memory"\
  769. );\
  770. tmp += 4;\
  771. src += 4 - 9*srcStride;\
  772. }\
  773. tmp -= 3*4;\
  774. asm volatile(\
  775. "movq %4, %%mm6 \n\t"\
  776. "1: \n\t"\
  777. "movq (%0), %%mm0 \n\t"\
  778. "paddw 10(%0), %%mm0 \n\t"\
  779. "movq 2(%0), %%mm1 \n\t"\
  780. "paddw 8(%0), %%mm1 \n\t"\
  781. "movq 4(%0), %%mm2 \n\t"\
  782. "paddw 6(%0), %%mm2 \n\t"\
  783. "psubw %%mm1, %%mm0 \n\t"/*a-b (abccba)*/\
  784. "psraw $2, %%mm0 \n\t"/*(a-b)/4 */\
  785. "psubw %%mm1, %%mm0 \n\t"/*(a-b)/4-b */\
  786. "paddsw %%mm2, %%mm0 \n\t"\
  787. "psraw $2, %%mm0 \n\t"/*((a-b)/4-b+c)/4 */\
  788. "paddw %%mm6, %%mm2 \n\t"\
  789. "paddw %%mm2, %%mm0 \n\t"/*(a-5*b+20*c)/16 +32 */\
  790. "psraw $6, %%mm0 \n\t"\
  791. "packuswb %%mm0, %%mm0 \n\t"\
  792. OP(%%mm0, (%1),%%mm7, d)\
  793. "add $24, %0 \n\t"\
  794. "add %3, %1 \n\t"\
  795. "decl %2 \n\t"\
  796. " jnz 1b \n\t"\
  797. : "+a"(tmp), "+c"(dst), "+m"(h)\
  798. : "S"((long)dstStride), "m"(ff_pw_32)\
  799. : "memory"\
  800. );\
  801. }\
  802. \
  803. static av_noinline void OPNAME ## h264_qpel8_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  804. int h=8;\
  805. asm volatile(\
  806. "pxor %%mm7, %%mm7 \n\t"\
  807. "movq %5, %%mm6 \n\t"\
  808. "1: \n\t"\
  809. "movq (%0), %%mm0 \n\t"\
  810. "movq 1(%0), %%mm2 \n\t"\
  811. "movq %%mm0, %%mm1 \n\t"\
  812. "movq %%mm2, %%mm3 \n\t"\
  813. "punpcklbw %%mm7, %%mm0 \n\t"\
  814. "punpckhbw %%mm7, %%mm1 \n\t"\
  815. "punpcklbw %%mm7, %%mm2 \n\t"\
  816. "punpckhbw %%mm7, %%mm3 \n\t"\
  817. "paddw %%mm2, %%mm0 \n\t"\
  818. "paddw %%mm3, %%mm1 \n\t"\
  819. "psllw $2, %%mm0 \n\t"\
  820. "psllw $2, %%mm1 \n\t"\
  821. "movq -1(%0), %%mm2 \n\t"\
  822. "movq 2(%0), %%mm4 \n\t"\
  823. "movq %%mm2, %%mm3 \n\t"\
  824. "movq %%mm4, %%mm5 \n\t"\
  825. "punpcklbw %%mm7, %%mm2 \n\t"\
  826. "punpckhbw %%mm7, %%mm3 \n\t"\
  827. "punpcklbw %%mm7, %%mm4 \n\t"\
  828. "punpckhbw %%mm7, %%mm5 \n\t"\
  829. "paddw %%mm4, %%mm2 \n\t"\
  830. "paddw %%mm3, %%mm5 \n\t"\
  831. "psubw %%mm2, %%mm0 \n\t"\
  832. "psubw %%mm5, %%mm1 \n\t"\
  833. "pmullw %%mm6, %%mm0 \n\t"\
  834. "pmullw %%mm6, %%mm1 \n\t"\
  835. "movd -2(%0), %%mm2 \n\t"\
  836. "movd 7(%0), %%mm5 \n\t"\
  837. "punpcklbw %%mm7, %%mm2 \n\t"\
  838. "punpcklbw %%mm7, %%mm5 \n\t"\
  839. "paddw %%mm3, %%mm2 \n\t"\
  840. "paddw %%mm5, %%mm4 \n\t"\
  841. "movq %6, %%mm5 \n\t"\
  842. "paddw %%mm5, %%mm2 \n\t"\
  843. "paddw %%mm5, %%mm4 \n\t"\
  844. "paddw %%mm2, %%mm0 \n\t"\
  845. "paddw %%mm4, %%mm1 \n\t"\
  846. "psraw $5, %%mm0 \n\t"\
  847. "psraw $5, %%mm1 \n\t"\
  848. "packuswb %%mm1, %%mm0 \n\t"\
  849. OP(%%mm0, (%1),%%mm5, q)\
  850. "add %3, %0 \n\t"\
  851. "add %4, %1 \n\t"\
  852. "decl %2 \n\t"\
  853. " jnz 1b \n\t"\
  854. : "+a"(src), "+c"(dst), "+m"(h)\
  855. : "d"((long)srcStride), "S"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  856. : "memory"\
  857. );\
  858. }\
  859. \
  860. static av_noinline void OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  861. int h=8;\
  862. asm volatile(\
  863. "pxor %%mm7, %%mm7 \n\t"\
  864. "movq %0, %%mm6 \n\t"\
  865. :: "m"(ff_pw_5)\
  866. );\
  867. do{\
  868. asm volatile(\
  869. "movq (%0), %%mm0 \n\t"\
  870. "movq 1(%0), %%mm2 \n\t"\
  871. "movq %%mm0, %%mm1 \n\t"\
  872. "movq %%mm2, %%mm3 \n\t"\
  873. "punpcklbw %%mm7, %%mm0 \n\t"\
  874. "punpckhbw %%mm7, %%mm1 \n\t"\
  875. "punpcklbw %%mm7, %%mm2 \n\t"\
  876. "punpckhbw %%mm7, %%mm3 \n\t"\
  877. "paddw %%mm2, %%mm0 \n\t"\
  878. "paddw %%mm3, %%mm1 \n\t"\
  879. "psllw $2, %%mm0 \n\t"\
  880. "psllw $2, %%mm1 \n\t"\
  881. "movq -1(%0), %%mm2 \n\t"\
  882. "movq 2(%0), %%mm4 \n\t"\
  883. "movq %%mm2, %%mm3 \n\t"\
  884. "movq %%mm4, %%mm5 \n\t"\
  885. "punpcklbw %%mm7, %%mm2 \n\t"\
  886. "punpckhbw %%mm7, %%mm3 \n\t"\
  887. "punpcklbw %%mm7, %%mm4 \n\t"\
  888. "punpckhbw %%mm7, %%mm5 \n\t"\
  889. "paddw %%mm4, %%mm2 \n\t"\
  890. "paddw %%mm3, %%mm5 \n\t"\
  891. "psubw %%mm2, %%mm0 \n\t"\
  892. "psubw %%mm5, %%mm1 \n\t"\
  893. "pmullw %%mm6, %%mm0 \n\t"\
  894. "pmullw %%mm6, %%mm1 \n\t"\
  895. "movd -2(%0), %%mm2 \n\t"\
  896. "movd 7(%0), %%mm5 \n\t"\
  897. "punpcklbw %%mm7, %%mm2 \n\t"\
  898. "punpcklbw %%mm7, %%mm5 \n\t"\
  899. "paddw %%mm3, %%mm2 \n\t"\
  900. "paddw %%mm5, %%mm4 \n\t"\
  901. "movq %5, %%mm5 \n\t"\
  902. "paddw %%mm5, %%mm2 \n\t"\
  903. "paddw %%mm5, %%mm4 \n\t"\
  904. "paddw %%mm2, %%mm0 \n\t"\
  905. "paddw %%mm4, %%mm1 \n\t"\
  906. "psraw $5, %%mm0 \n\t"\
  907. "psraw $5, %%mm1 \n\t"\
  908. "movq (%2), %%mm4 \n\t"\
  909. "packuswb %%mm1, %%mm0 \n\t"\
  910. PAVGB" %%mm4, %%mm0 \n\t"\
  911. OP(%%mm0, (%1),%%mm5, q)\
  912. "add %4, %0 \n\t"\
  913. "add %4, %1 \n\t"\
  914. "add %3, %2 \n\t"\
  915. : "+a"(src), "+c"(dst), "+d"(src2)\
  916. : "D"((long)src2Stride), "S"((long)dstStride),\
  917. "m"(ff_pw_16)\
  918. : "memory"\
  919. );\
  920. }while(--h);\
  921. }\
  922. \
  923. static av_noinline void OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
  924. int w= 2;\
  925. src -= 2*srcStride;\
  926. \
  927. while(w--){\
  928. asm volatile(\
  929. "pxor %%mm7, %%mm7 \n\t"\
  930. "movd (%0), %%mm0 \n\t"\
  931. "add %2, %0 \n\t"\
  932. "movd (%0), %%mm1 \n\t"\
  933. "add %2, %0 \n\t"\
  934. "movd (%0), %%mm2 \n\t"\
  935. "add %2, %0 \n\t"\
  936. "movd (%0), %%mm3 \n\t"\
  937. "add %2, %0 \n\t"\
  938. "movd (%0), %%mm4 \n\t"\
  939. "add %2, %0 \n\t"\
  940. "punpcklbw %%mm7, %%mm0 \n\t"\
  941. "punpcklbw %%mm7, %%mm1 \n\t"\
  942. "punpcklbw %%mm7, %%mm2 \n\t"\
  943. "punpcklbw %%mm7, %%mm3 \n\t"\
  944. "punpcklbw %%mm7, %%mm4 \n\t"\
  945. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  946. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  947. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  948. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  949. QPEL_H264V(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP)\
  950. QPEL_H264V(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP)\
  951. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  952. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  953. \
  954. : "+a"(src), "+c"(dst)\
  955. : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  956. : "memory"\
  957. );\
  958. if(h==16){\
  959. asm volatile(\
  960. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  961. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  962. QPEL_H264V(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP)\
  963. QPEL_H264V(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP)\
  964. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  965. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  966. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  967. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  968. \
  969. : "+a"(src), "+c"(dst)\
  970. : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  971. : "memory"\
  972. );\
  973. }\
  974. src += 4-(h+5)*srcStride;\
  975. dst += 4-h*dstStride;\
  976. }\
  977. }\
  978. static av_noinline void OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
  979. int h = size;\
  980. int w = (size+8)>>2;\
  981. src -= 2*srcStride+2;\
  982. while(w--){\
  983. asm volatile(\
  984. "pxor %%mm7, %%mm7 \n\t"\
  985. "movd (%0), %%mm0 \n\t"\
  986. "add %2, %0 \n\t"\
  987. "movd (%0), %%mm1 \n\t"\
  988. "add %2, %0 \n\t"\
  989. "movd (%0), %%mm2 \n\t"\
  990. "add %2, %0 \n\t"\
  991. "movd (%0), %%mm3 \n\t"\
  992. "add %2, %0 \n\t"\
  993. "movd (%0), %%mm4 \n\t"\
  994. "add %2, %0 \n\t"\
  995. "punpcklbw %%mm7, %%mm0 \n\t"\
  996. "punpcklbw %%mm7, %%mm1 \n\t"\
  997. "punpcklbw %%mm7, %%mm2 \n\t"\
  998. "punpcklbw %%mm7, %%mm3 \n\t"\
  999. "punpcklbw %%mm7, %%mm4 \n\t"\
  1000. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 0*48)\
  1001. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 1*48)\
  1002. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 2*48)\
  1003. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 3*48)\
  1004. QPEL_H264HV(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, 4*48)\
  1005. QPEL_H264HV(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, 5*48)\
  1006. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 6*48)\
  1007. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 7*48)\
  1008. : "+a"(src)\
  1009. : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
  1010. : "memory"\
  1011. );\
  1012. if(size==16){\
  1013. asm volatile(\
  1014. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 8*48)\
  1015. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 9*48)\
  1016. QPEL_H264HV(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, 10*48)\
  1017. QPEL_H264HV(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, 11*48)\
  1018. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 12*48)\
  1019. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 13*48)\
  1020. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 14*48)\
  1021. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 15*48)\
  1022. : "+a"(src)\
  1023. : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
  1024. : "memory"\
  1025. );\
  1026. }\
  1027. tmp += 4;\
  1028. src += 4 - (size+5)*srcStride;\
  1029. }\
  1030. tmp -= size+8;\
  1031. w = size>>4;\
  1032. do{\
  1033. h = size;\
  1034. asm volatile(\
  1035. "movq %4, %%mm6 \n\t"\
  1036. "1: \n\t"\
  1037. "movq (%0), %%mm0 \n\t"\
  1038. "movq 8(%0), %%mm3 \n\t"\
  1039. "movq 2(%0), %%mm1 \n\t"\
  1040. "movq 10(%0), %%mm4 \n\t"\
  1041. "paddw %%mm4, %%mm0 \n\t"\
  1042. "paddw %%mm3, %%mm1 \n\t"\
  1043. "paddw 18(%0), %%mm3 \n\t"\
  1044. "paddw 16(%0), %%mm4 \n\t"\
  1045. "movq 4(%0), %%mm2 \n\t"\
  1046. "movq 12(%0), %%mm5 \n\t"\
  1047. "paddw 6(%0), %%mm2 \n\t"\
  1048. "paddw 14(%0), %%mm5 \n\t"\
  1049. "psubw %%mm1, %%mm0 \n\t"\
  1050. "psubw %%mm4, %%mm3 \n\t"\
  1051. "psraw $2, %%mm0 \n\t"\
  1052. "psraw $2, %%mm3 \n\t"\
  1053. "psubw %%mm1, %%mm0 \n\t"\
  1054. "psubw %%mm4, %%mm3 \n\t"\
  1055. "paddsw %%mm2, %%mm0 \n\t"\
  1056. "paddsw %%mm5, %%mm3 \n\t"\
  1057. "psraw $2, %%mm0 \n\t"\
  1058. "psraw $2, %%mm3 \n\t"\
  1059. "paddw %%mm6, %%mm2 \n\t"\
  1060. "paddw %%mm6, %%mm5 \n\t"\
  1061. "paddw %%mm2, %%mm0 \n\t"\
  1062. "paddw %%mm5, %%mm3 \n\t"\
  1063. "psraw $6, %%mm0 \n\t"\
  1064. "psraw $6, %%mm3 \n\t"\
  1065. "packuswb %%mm3, %%mm0 \n\t"\
  1066. OP(%%mm0, (%1),%%mm7, q)\
  1067. "add $48, %0 \n\t"\
  1068. "add %3, %1 \n\t"\
  1069. "decl %2 \n\t"\
  1070. " jnz 1b \n\t"\
  1071. : "+a"(tmp), "+c"(dst), "+m"(h)\
  1072. : "S"((long)dstStride), "m"(ff_pw_32)\
  1073. : "memory"\
  1074. );\
  1075. tmp += 8 - size*24;\
  1076. dst += 8 - size*dstStride;\
  1077. }while(w--);\
  1078. }\
  1079. \
  1080. static void OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  1081. OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 8);\
  1082. }\
  1083. static av_noinline void OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  1084. OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 16);\
  1085. OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  1086. }\
  1087. \
  1088. static av_noinline void OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  1089. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  1090. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  1091. src += 8*srcStride;\
  1092. dst += 8*dstStride;\
  1093. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  1094. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  1095. }\
  1096. \
  1097. static av_noinline void OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  1098. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  1099. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  1100. src += 8*dstStride;\
  1101. dst += 8*dstStride;\
  1102. src2 += 8*src2Stride;\
  1103. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  1104. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  1105. }\
  1106. \
  1107. static void OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  1108. OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst , tmp , src , dstStride, tmpStride, srcStride, 8);\
  1109. }\
  1110. \
  1111. static void OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  1112. OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst , tmp , src , dstStride, tmpStride, srcStride, 16);\
  1113. }\
  1114. \
  1115. static av_noinline void OPNAME ## pixels4_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  1116. {\
  1117. asm volatile(\
  1118. "movq %5, %%mm6 \n\t"\
  1119. "movq (%1), %%mm0 \n\t"\
  1120. "movq 24(%1), %%mm1 \n\t"\
  1121. "paddw %%mm6, %%mm0 \n\t"\
  1122. "paddw %%mm6, %%mm1 \n\t"\
  1123. "psraw $5, %%mm0 \n\t"\
  1124. "psraw $5, %%mm1 \n\t"\
  1125. "packuswb %%mm0, %%mm0 \n\t"\
  1126. "packuswb %%mm1, %%mm1 \n\t"\
  1127. PAVGB" (%0), %%mm0 \n\t"\
  1128. PAVGB" (%0,%3), %%mm1 \n\t"\
  1129. OP(%%mm0, (%2), %%mm4, d)\
  1130. OP(%%mm1, (%2,%4), %%mm5, d)\
  1131. "lea (%0,%3,2), %0 \n\t"\
  1132. "lea (%2,%4,2), %2 \n\t"\
  1133. "movq 48(%1), %%mm0 \n\t"\
  1134. "movq 72(%1), %%mm1 \n\t"\
  1135. "paddw %%mm6, %%mm0 \n\t"\
  1136. "paddw %%mm6, %%mm1 \n\t"\
  1137. "psraw $5, %%mm0 \n\t"\
  1138. "psraw $5, %%mm1 \n\t"\
  1139. "packuswb %%mm0, %%mm0 \n\t"\
  1140. "packuswb %%mm1, %%mm1 \n\t"\
  1141. PAVGB" (%0), %%mm0 \n\t"\
  1142. PAVGB" (%0,%3), %%mm1 \n\t"\
  1143. OP(%%mm0, (%2), %%mm4, d)\
  1144. OP(%%mm1, (%2,%4), %%mm5, d)\
  1145. :"+a"(src8), "+c"(src16), "+d"(dst)\
  1146. :"S"((long)src8Stride), "D"((long)dstStride), "m"(ff_pw_16)\
  1147. :"memory");\
  1148. }\
  1149. static av_noinline void OPNAME ## pixels8_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  1150. {\
  1151. asm volatile(\
  1152. "movq %0, %%mm6 \n\t"\
  1153. ::"m"(ff_pw_16)\
  1154. );\
  1155. while(h--){\
  1156. asm volatile(\
  1157. "movq (%1), %%mm0 \n\t"\
  1158. "movq 8(%1), %%mm1 \n\t"\
  1159. "paddw %%mm6, %%mm0 \n\t"\
  1160. "paddw %%mm6, %%mm1 \n\t"\
  1161. "psraw $5, %%mm0 \n\t"\
  1162. "psraw $5, %%mm1 \n\t"\
  1163. "packuswb %%mm1, %%mm0 \n\t"\
  1164. PAVGB" (%0), %%mm0 \n\t"\
  1165. OP(%%mm0, (%2), %%mm5, q)\
  1166. ::"a"(src8), "c"(src16), "d"(dst)\
  1167. :"memory");\
  1168. src8 += src8Stride;\
  1169. src16 += 24;\
  1170. dst += dstStride;\
  1171. }\
  1172. }\
  1173. static void OPNAME ## pixels16_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  1174. {\
  1175. OPNAME ## pixels8_l2_shift5_ ## MMX(dst , src16 , src8 , dstStride, src8Stride, h);\
  1176. OPNAME ## pixels8_l2_shift5_ ## MMX(dst+8, src16+8, src8+8, dstStride, src8Stride, h);\
  1177. }\
  1178. #define H264_MC(OPNAME, SIZE, MMX) \
  1179. static void OPNAME ## h264_qpel ## SIZE ## _mc00_ ## MMX (uint8_t *dst, uint8_t *src, int stride){\
  1180. OPNAME ## pixels ## SIZE ## _mmx(dst, src, stride, SIZE);\
  1181. }\
  1182. \
  1183. static void OPNAME ## h264_qpel ## SIZE ## _mc10_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1184. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src, stride, stride);\
  1185. }\
  1186. \
  1187. static void OPNAME ## h264_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1188. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_ ## MMX(dst, src, stride, stride);\
  1189. }\
  1190. \
  1191. static void OPNAME ## h264_qpel ## SIZE ## _mc30_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1192. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src+1, stride, stride);\
  1193. }\
  1194. \
  1195. static void OPNAME ## h264_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1196. uint64_t temp[SIZE*SIZE/8];\
  1197. uint8_t * const half= (uint8_t*)temp;\
  1198. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(half, src, SIZE, stride);\
  1199. OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src, half, stride, stride, SIZE);\
  1200. }\
  1201. \
  1202. static void OPNAME ## h264_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1203. OPNAME ## h264_qpel ## SIZE ## _v_lowpass_ ## MMX(dst, src, stride, stride);\
  1204. }\
  1205. \
  1206. static void OPNAME ## h264_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1207. uint64_t temp[SIZE*SIZE/8];\
  1208. uint8_t * const half= (uint8_t*)temp;\
  1209. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(half, src, SIZE, stride);\
  1210. OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src+stride, half, stride, stride, SIZE);\
  1211. }\
  1212. \
  1213. static void OPNAME ## h264_qpel ## SIZE ## _mc11_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1214. uint64_t temp[SIZE*SIZE/8];\
  1215. uint8_t * const halfV= (uint8_t*)temp;\
  1216. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src, SIZE, stride);\
  1217. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfV, stride, SIZE);\
  1218. }\
  1219. \
  1220. static void OPNAME ## h264_qpel ## SIZE ## _mc31_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1221. uint64_t temp[SIZE*SIZE/8];\
  1222. uint8_t * const halfV= (uint8_t*)temp;\
  1223. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src+1, SIZE, stride);\
  1224. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfV, stride, SIZE);\
  1225. }\
  1226. \
  1227. static void OPNAME ## h264_qpel ## SIZE ## _mc13_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1228. uint64_t temp[SIZE*SIZE/8];\
  1229. uint8_t * const halfV= (uint8_t*)temp;\
  1230. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src, SIZE, stride);\
  1231. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfV, stride, SIZE);\
  1232. }\
  1233. \
  1234. static void OPNAME ## h264_qpel ## SIZE ## _mc33_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1235. uint64_t temp[SIZE*SIZE/8];\
  1236. uint8_t * const halfV= (uint8_t*)temp;\
  1237. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src+1, SIZE, stride);\
  1238. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfV, stride, SIZE);\
  1239. }\
  1240. \
  1241. static void OPNAME ## h264_qpel ## SIZE ## _mc22_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1242. uint64_t temp[SIZE*(SIZE<8?12:24)/4];\
  1243. int16_t * const tmp= (int16_t*)temp;\
  1244. OPNAME ## h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(dst, tmp, src, stride, SIZE, stride);\
  1245. }\
  1246. \
  1247. static void OPNAME ## h264_qpel ## SIZE ## _mc21_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1248. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1249. uint8_t * const halfHV= (uint8_t*)temp;\
  1250. int16_t * const tmp= ((int16_t*)temp) + SIZE*SIZE/2;\
  1251. assert((int)temp & 7 == 0);\
  1252. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, tmp, src, SIZE, SIZE, stride);\
  1253. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfHV, stride, SIZE);\
  1254. }\
  1255. \
  1256. static void OPNAME ## h264_qpel ## SIZE ## _mc23_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1257. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1258. uint8_t * const halfHV= (uint8_t*)temp;\
  1259. int16_t * const tmp= ((int16_t*)temp) + SIZE*SIZE/2;\
  1260. assert((int)temp & 7 == 0);\
  1261. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, tmp, src, SIZE, SIZE, stride);\
  1262. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfHV, stride, SIZE);\
  1263. }\
  1264. \
  1265. static void OPNAME ## h264_qpel ## SIZE ## _mc12_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1266. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1267. int16_t * const halfV= ((int16_t*)temp) + SIZE*SIZE/2;\
  1268. uint8_t * const halfHV= ((uint8_t*)temp);\
  1269. assert((int)temp & 7 == 0);\
  1270. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  1271. OPNAME ## pixels ## SIZE ## _l2_shift5_ ## MMX(dst, halfV+2, halfHV, stride, SIZE, SIZE);\
  1272. }\
  1273. \
  1274. static void OPNAME ## h264_qpel ## SIZE ## _mc32_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1275. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1276. int16_t * const halfV= ((int16_t*)temp) + SIZE*SIZE/2;\
  1277. uint8_t * const halfHV= ((uint8_t*)temp);\
  1278. assert((int)temp & 7 == 0);\
  1279. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  1280. OPNAME ## pixels ## SIZE ## _l2_shift5_ ## MMX(dst, halfV+3, halfHV, stride, SIZE, SIZE);\
  1281. }\
  1282. #define AVG_3DNOW_OP(a,b,temp, size) \
  1283. "mov" #size " " #b ", " #temp " \n\t"\
  1284. "pavgusb " #temp ", " #a " \n\t"\
  1285. "mov" #size " " #a ", " #b " \n\t"
  1286. #define AVG_MMX2_OP(a,b,temp, size) \
  1287. "mov" #size " " #b ", " #temp " \n\t"\
  1288. "pavgb " #temp ", " #a " \n\t"\
  1289. "mov" #size " " #a ", " #b " \n\t"
  1290. #define PAVGB "pavgusb"
  1291. QPEL_H264(put_, PUT_OP, 3dnow)
  1292. QPEL_H264(avg_, AVG_3DNOW_OP, 3dnow)
  1293. #undef PAVGB
  1294. #define PAVGB "pavgb"
  1295. QPEL_H264(put_, PUT_OP, mmx2)
  1296. QPEL_H264(avg_, AVG_MMX2_OP, mmx2)
  1297. #undef PAVGB
  1298. H264_MC(put_, 4, 3dnow)
  1299. H264_MC(put_, 8, 3dnow)
  1300. H264_MC(put_, 16,3dnow)
  1301. H264_MC(avg_, 4, 3dnow)
  1302. H264_MC(avg_, 8, 3dnow)
  1303. H264_MC(avg_, 16,3dnow)
  1304. H264_MC(put_, 4, mmx2)
  1305. H264_MC(put_, 8, mmx2)
  1306. H264_MC(put_, 16,mmx2)
  1307. H264_MC(avg_, 4, mmx2)
  1308. H264_MC(avg_, 8, mmx2)
  1309. H264_MC(avg_, 16,mmx2)
  1310. #define H264_CHROMA_OP(S,D)
  1311. #define H264_CHROMA_OP4(S,D,T)
  1312. #define H264_CHROMA_MC8_TMPL put_h264_chroma_mc8_mmx
  1313. #define H264_CHROMA_MC4_TMPL put_h264_chroma_mc4_mmx
  1314. #define H264_CHROMA_MC2_TMPL put_h264_chroma_mc2_mmx2
  1315. #define H264_CHROMA_MC8_MV0 put_pixels8_mmx
  1316. #include "dsputil_h264_template_mmx.c"
  1317. #undef H264_CHROMA_OP
  1318. #undef H264_CHROMA_OP4
  1319. #undef H264_CHROMA_MC8_TMPL
  1320. #undef H264_CHROMA_MC4_TMPL
  1321. #undef H264_CHROMA_MC2_TMPL
  1322. #undef H264_CHROMA_MC8_MV0
  1323. #define H264_CHROMA_OP(S,D) "pavgb " #S ", " #D " \n\t"
  1324. #define H264_CHROMA_OP4(S,D,T) "movd " #S ", " #T " \n\t"\
  1325. "pavgb " #T ", " #D " \n\t"
  1326. #define H264_CHROMA_MC8_TMPL avg_h264_chroma_mc8_mmx2
  1327. #define H264_CHROMA_MC4_TMPL avg_h264_chroma_mc4_mmx2
  1328. #define H264_CHROMA_MC2_TMPL avg_h264_chroma_mc2_mmx2
  1329. #define H264_CHROMA_MC8_MV0 avg_pixels8_mmx2
  1330. #include "dsputil_h264_template_mmx.c"
  1331. #undef H264_CHROMA_OP
  1332. #undef H264_CHROMA_OP4
  1333. #undef H264_CHROMA_MC8_TMPL
  1334. #undef H264_CHROMA_MC4_TMPL
  1335. #undef H264_CHROMA_MC2_TMPL
  1336. #undef H264_CHROMA_MC8_MV0
  1337. #define H264_CHROMA_OP(S,D) "pavgusb " #S ", " #D " \n\t"
  1338. #define H264_CHROMA_OP4(S,D,T) "movd " #S ", " #T " \n\t"\
  1339. "pavgusb " #T ", " #D " \n\t"
  1340. #define H264_CHROMA_MC8_TMPL avg_h264_chroma_mc8_3dnow
  1341. #define H264_CHROMA_MC4_TMPL avg_h264_chroma_mc4_3dnow
  1342. #define H264_CHROMA_MC8_MV0 avg_pixels8_3dnow
  1343. #include "dsputil_h264_template_mmx.c"
  1344. #undef H264_CHROMA_OP
  1345. #undef H264_CHROMA_OP4
  1346. #undef H264_CHROMA_MC8_TMPL
  1347. #undef H264_CHROMA_MC4_TMPL
  1348. #undef H264_CHROMA_MC8_MV0
  1349. /***********************************/
  1350. /* weighted prediction */
  1351. static inline void ff_h264_weight_WxH_mmx2(uint8_t *dst, int stride, int log2_denom, int weight, int offset, int w, int h)
  1352. {
  1353. int x, y;
  1354. offset <<= log2_denom;
  1355. offset += (1 << log2_denom) >> 1;
  1356. asm volatile(
  1357. "movd %0, %%mm4 \n\t"
  1358. "movd %1, %%mm5 \n\t"
  1359. "movd %2, %%mm6 \n\t"
  1360. "pshufw $0, %%mm4, %%mm4 \n\t"
  1361. "pshufw $0, %%mm5, %%mm5 \n\t"
  1362. "pxor %%mm7, %%mm7 \n\t"
  1363. :: "g"(weight), "g"(offset), "g"(log2_denom)
  1364. );
  1365. for(y=0; y<h; y+=2){
  1366. for(x=0; x<w; x+=4){
  1367. asm volatile(
  1368. "movd %0, %%mm0 \n\t"
  1369. "movd %1, %%mm1 \n\t"
  1370. "punpcklbw %%mm7, %%mm0 \n\t"
  1371. "punpcklbw %%mm7, %%mm1 \n\t"
  1372. "pmullw %%mm4, %%mm0 \n\t"
  1373. "pmullw %%mm4, %%mm1 \n\t"
  1374. "paddsw %%mm5, %%mm0 \n\t"
  1375. "paddsw %%mm5, %%mm1 \n\t"
  1376. "psraw %%mm6, %%mm0 \n\t"
  1377. "psraw %%mm6, %%mm1 \n\t"
  1378. "packuswb %%mm7, %%mm0 \n\t"
  1379. "packuswb %%mm7, %%mm1 \n\t"
  1380. "movd %%mm0, %0 \n\t"
  1381. "movd %%mm1, %1 \n\t"
  1382. : "+m"(*(uint32_t*)(dst+x)),
  1383. "+m"(*(uint32_t*)(dst+x+stride))
  1384. );
  1385. }
  1386. dst += 2*stride;
  1387. }
  1388. }
  1389. static inline void ff_h264_biweight_WxH_mmx2(uint8_t *dst, uint8_t *src, int stride, int log2_denom, int weightd, int weights, int offset, int w, int h)
  1390. {
  1391. int x, y;
  1392. offset = ((offset + 1) | 1) << log2_denom;
  1393. asm volatile(
  1394. "movd %0, %%mm3 \n\t"
  1395. "movd %1, %%mm4 \n\t"
  1396. "movd %2, %%mm5 \n\t"
  1397. "movd %3, %%mm6 \n\t"
  1398. "pshufw $0, %%mm3, %%mm3 \n\t"
  1399. "pshufw $0, %%mm4, %%mm4 \n\t"
  1400. "pshufw $0, %%mm5, %%mm5 \n\t"
  1401. "pxor %%mm7, %%mm7 \n\t"
  1402. :: "g"(weightd), "g"(weights), "g"(offset), "g"(log2_denom+1)
  1403. );
  1404. for(y=0; y<h; y++){
  1405. for(x=0; x<w; x+=4){
  1406. asm volatile(
  1407. "movd %0, %%mm0 \n\t"
  1408. "movd %1, %%mm1 \n\t"
  1409. "punpcklbw %%mm7, %%mm0 \n\t"
  1410. "punpcklbw %%mm7, %%mm1 \n\t"
  1411. "pmullw %%mm3, %%mm0 \n\t"
  1412. "pmullw %%mm4, %%mm1 \n\t"
  1413. "paddsw %%mm1, %%mm0 \n\t"
  1414. "paddsw %%mm5, %%mm0 \n\t"
  1415. "psraw %%mm6, %%mm0 \n\t"
  1416. "packuswb %%mm0, %%mm0 \n\t"
  1417. "movd %%mm0, %0 \n\t"
  1418. : "+m"(*(uint32_t*)(dst+x))
  1419. : "m"(*(uint32_t*)(src+x))
  1420. );
  1421. }
  1422. src += stride;
  1423. dst += stride;
  1424. }
  1425. }
  1426. #define H264_WEIGHT(W,H) \
  1427. static void ff_h264_biweight_ ## W ## x ## H ## _mmx2(uint8_t *dst, uint8_t *src, int stride, int log2_denom, int weightd, int weights, int offset){ \
  1428. ff_h264_biweight_WxH_mmx2(dst, src, stride, log2_denom, weightd, weights, offset, W, H); \
  1429. } \
  1430. static void ff_h264_weight_ ## W ## x ## H ## _mmx2(uint8_t *dst, int stride, int log2_denom, int weight, int offset){ \
  1431. ff_h264_weight_WxH_mmx2(dst, stride, log2_denom, weight, offset, W, H); \
  1432. }
  1433. H264_WEIGHT(16,16)
  1434. H264_WEIGHT(16, 8)
  1435. H264_WEIGHT( 8,16)
  1436. H264_WEIGHT( 8, 8)
  1437. H264_WEIGHT( 8, 4)
  1438. H264_WEIGHT( 4, 8)
  1439. H264_WEIGHT( 4, 4)
  1440. H264_WEIGHT( 4, 2)