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.

1525 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. 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. // in: mm0=p1 mm1=p0 mm2=q0 mm3=q1
  285. // out: mm5=beta-1, mm7=mask
  286. // clobbers: mm4,mm6
  287. #define H264_DEBLOCK_MASK(alpha1, beta1) \
  288. "pshufw $0, "#alpha1", %%mm4 \n\t"\
  289. "pshufw $0, "#beta1 ", %%mm5 \n\t"\
  290. "packuswb %%mm4, %%mm4 \n\t"\
  291. "packuswb %%mm5, %%mm5 \n\t"\
  292. DIFF_GT_MMX(%%mm1, %%mm2, %%mm4, %%mm7, %%mm6) /* |p0-q0| > alpha-1 */\
  293. DIFF_GT_MMX(%%mm0, %%mm1, %%mm5, %%mm4, %%mm6) /* |p1-p0| > beta-1 */\
  294. "por %%mm4, %%mm7 \n\t"\
  295. DIFF_GT_MMX(%%mm3, %%mm2, %%mm5, %%mm4, %%mm6) /* |q1-q0| > beta-1 */\
  296. "por %%mm4, %%mm7 \n\t"\
  297. "pxor %%mm6, %%mm6 \n\t"\
  298. "pcmpeqb %%mm6, %%mm7 \n\t"
  299. // in: mm0=p1 mm1=p0 mm2=q0 mm3=q1 mm7=(tc&mask)
  300. // out: mm1=p0' mm2=q0'
  301. // clobbers: mm0,3-6
  302. #define H264_DEBLOCK_P0_Q0(pb_01, pb_3f)\
  303. /* a = q0^p0^((p1-q1)>>2) */\
  304. "movq %%mm0, %%mm4 \n\t"\
  305. "psubb %%mm3, %%mm4 \n\t"\
  306. "psrlw $2, %%mm4 \n\t"\
  307. "pxor %%mm1, %%mm4 \n\t"\
  308. "pxor %%mm2, %%mm4 \n\t"\
  309. /* b = p0^(q1>>2) */\
  310. "psrlw $2, %%mm3 \n\t"\
  311. "pand "#pb_3f", %%mm3 \n\t"\
  312. "movq %%mm1, %%mm5 \n\t"\
  313. "pxor %%mm3, %%mm5 \n\t"\
  314. /* c = q0^(p1>>2) */\
  315. "psrlw $2, %%mm0 \n\t"\
  316. "pand "#pb_3f", %%mm0 \n\t"\
  317. "movq %%mm2, %%mm6 \n\t"\
  318. "pxor %%mm0, %%mm6 \n\t"\
  319. /* d = (c^b) & ~(b^a) & 1 */\
  320. "pxor %%mm5, %%mm6 \n\t"\
  321. "pxor %%mm4, %%mm5 \n\t"\
  322. "pandn %%mm6, %%mm5 \n\t"\
  323. "pand "#pb_01", %%mm5 \n\t"\
  324. /* delta = (avg(q0, p1>>2) + (d&a))
  325. * - (avg(p0, q1>>2) + (d&~a)) */\
  326. "pavgb %%mm2, %%mm0 \n\t"\
  327. "pand %%mm5, %%mm4 \n\t"\
  328. "paddusb %%mm4, %%mm0 \n\t"\
  329. "pavgb %%mm1, %%mm3 \n\t"\
  330. "pxor %%mm5, %%mm4 \n\t"\
  331. "paddusb %%mm4, %%mm3 \n\t"\
  332. /* p0 += clip(delta, -tc0, tc0)
  333. * q0 -= clip(delta, -tc0, tc0) */\
  334. "movq %%mm0, %%mm4 \n\t"\
  335. "psubusb %%mm3, %%mm0 \n\t"\
  336. "psubusb %%mm4, %%mm3 \n\t"\
  337. "pminub %%mm7, %%mm0 \n\t"\
  338. "pminub %%mm7, %%mm3 \n\t"\
  339. "paddusb %%mm0, %%mm1 \n\t"\
  340. "paddusb %%mm3, %%mm2 \n\t"\
  341. "psubusb %%mm3, %%mm1 \n\t"\
  342. "psubusb %%mm0, %%mm2 \n\t"
  343. // in: mm0=p1 mm1=p0 mm2=q0 mm3=q1 mm7=(tc&mask) %8=mm_bone
  344. // out: (q1addr) = clip( (q2+((p0+q0+1)>>1))>>1, q1-tc0, q1+tc0 )
  345. // clobbers: q2, tmp, tc0
  346. #define H264_DEBLOCK_Q1(p1, q2, q2addr, q1addr, tc0, tmp)\
  347. "movq %%mm1, "#tmp" \n\t"\
  348. "pavgb %%mm2, "#tmp" \n\t"\
  349. "pavgb "#tmp", "#q2" \n\t" /* avg(p2,avg(p0,q0)) */\
  350. "pxor "q2addr", "#tmp" \n\t"\
  351. "pand %8, "#tmp" \n\t" /* (p2^avg(p0,q0))&1 */\
  352. "psubusb "#tmp", "#q2" \n\t" /* (p2+((p0+q0+1)>>1))>>1 */\
  353. "movq "#p1", "#tmp" \n\t"\
  354. "psubusb "#tc0", "#tmp" \n\t"\
  355. "paddusb "#p1", "#tc0" \n\t"\
  356. "pmaxub "#tmp", "#q2" \n\t"\
  357. "pminub "#tc0", "#q2" \n\t"\
  358. "movq "#q2", "q1addr" \n\t"
  359. static inline void h264_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha1, int beta1, int8_t *tc0)
  360. {
  361. uint64_t tmp0;
  362. uint64_t tc = (uint8_t)tc0[1]*0x01010000 | (uint8_t)tc0[0]*0x0101;
  363. // with luma, tc0=0 doesn't mean no filtering, so we need a separate input mask
  364. uint32_t mask[2] = { (tc0[0]>=0)*0xffffffff, (tc0[1]>=0)*0xffffffff };
  365. asm volatile(
  366. "movq (%1,%3), %%mm0 \n\t" //p1
  367. "movq (%1,%3,2), %%mm1 \n\t" //p0
  368. "movq (%2), %%mm2 \n\t" //q0
  369. "movq (%2,%3), %%mm3 \n\t" //q1
  370. H264_DEBLOCK_MASK(%6, %7)
  371. "pand %5, %%mm7 \n\t"
  372. "movq %%mm7, %0 \n\t"
  373. /* filter p1 */
  374. "movq (%1), %%mm3 \n\t" //p2
  375. DIFF_GT_MMX(%%mm1, %%mm3, %%mm5, %%mm6, %%mm4) // |p2-p0|>beta-1
  376. "pandn %%mm7, %%mm6 \n\t"
  377. "pcmpeqb %%mm7, %%mm6 \n\t"
  378. "pand %%mm7, %%mm6 \n\t" // mask & |p2-p0|<beta
  379. "pshufw $80, %4, %%mm4 \n\t"
  380. "pand %%mm7, %%mm4 \n\t" // mask & tc0
  381. "movq %8, %%mm7 \n\t"
  382. "pand %%mm6, %%mm7 \n\t" // mask & |p2-p0|<beta & 1
  383. "pand %%mm4, %%mm6 \n\t" // mask & |p2-p0|<beta & tc0
  384. "paddb %%mm4, %%mm7 \n\t" // tc++
  385. H264_DEBLOCK_Q1(%%mm0, %%mm3, "(%1)", "(%1,%3)", %%mm6, %%mm4)
  386. /* filter q1 */
  387. "movq (%2,%3,2), %%mm4 \n\t" //q2
  388. DIFF_GT_MMX(%%mm2, %%mm4, %%mm5, %%mm6, %%mm3) // |q2-q0|>beta-1
  389. "pandn %0, %%mm6 \n\t"
  390. "pcmpeqb %0, %%mm6 \n\t"
  391. "pand %0, %%mm6 \n\t"
  392. "pshufw $80, %4, %%mm5 \n\t"
  393. "pand %%mm6, %%mm5 \n\t"
  394. "pand %8, %%mm6 \n\t"
  395. "paddb %%mm6, %%mm7 \n\t"
  396. "movq (%2,%3), %%mm3 \n\t"
  397. H264_DEBLOCK_Q1(%%mm3, %%mm4, "(%2,%3,2)", "(%2,%3)", %%mm5, %%mm6)
  398. /* filter p0, q0 */
  399. H264_DEBLOCK_P0_Q0(%8, %9)
  400. "movq %%mm1, (%1,%3,2) \n\t"
  401. "movq %%mm2, (%2) \n\t"
  402. : "=m"(tmp0)
  403. : "r"(pix-3*stride), "r"(pix), "r"((long)stride),
  404. "m"(tc), "m"(*(uint64_t*)mask), "m"(alpha1), "m"(beta1),
  405. "m"(mm_bone), "m"(ff_pb_3F)
  406. );
  407. }
  408. static void h264_v_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  409. {
  410. if((tc0[0] & tc0[1]) >= 0)
  411. h264_loop_filter_luma_mmx2(pix, stride, alpha-1, beta-1, tc0);
  412. if((tc0[2] & tc0[3]) >= 0)
  413. h264_loop_filter_luma_mmx2(pix+8, stride, alpha-1, beta-1, tc0+2);
  414. }
  415. static void h264_h_loop_filter_luma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  416. {
  417. //FIXME: could cut some load/stores by merging transpose with filter
  418. // also, it only needs to transpose 6x8
  419. uint8_t trans[8*8];
  420. int i;
  421. for(i=0; i<2; i++, pix+=8*stride, tc0+=2) {
  422. if((tc0[0] & tc0[1]) < 0)
  423. continue;
  424. transpose4x4(trans, pix-4, 8, stride);
  425. transpose4x4(trans +4*8, pix, 8, stride);
  426. transpose4x4(trans+4, pix-4+4*stride, 8, stride);
  427. transpose4x4(trans+4+4*8, pix +4*stride, 8, stride);
  428. h264_loop_filter_luma_mmx2(trans+4*8, 8, alpha-1, beta-1, tc0);
  429. transpose4x4(pix-2, trans +2*8, stride, 8);
  430. transpose4x4(pix-2+4*stride, trans+4+2*8, stride, 8);
  431. }
  432. }
  433. static inline void h264_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha1, int beta1, int8_t *tc0)
  434. {
  435. asm volatile(
  436. "movq (%0), %%mm0 \n\t" //p1
  437. "movq (%0,%2), %%mm1 \n\t" //p0
  438. "movq (%1), %%mm2 \n\t" //q0
  439. "movq (%1,%2), %%mm3 \n\t" //q1
  440. H264_DEBLOCK_MASK(%4, %5)
  441. "movd %3, %%mm6 \n\t"
  442. "punpcklbw %%mm6, %%mm6 \n\t"
  443. "pand %%mm6, %%mm7 \n\t" // mm7 = tc&mask
  444. H264_DEBLOCK_P0_Q0(%6, %7)
  445. "movq %%mm1, (%0,%2) \n\t"
  446. "movq %%mm2, (%1) \n\t"
  447. :: "r"(pix-2*stride), "r"(pix), "r"((long)stride),
  448. "r"(*(uint32_t*)tc0),
  449. "m"(alpha1), "m"(beta1), "m"(mm_bone), "m"(ff_pb_3F)
  450. );
  451. }
  452. static void h264_v_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  453. {
  454. h264_loop_filter_chroma_mmx2(pix, stride, alpha-1, beta-1, tc0);
  455. }
  456. static void h264_h_loop_filter_chroma_mmx2(uint8_t *pix, int stride, int alpha, int beta, int8_t *tc0)
  457. {
  458. //FIXME: could cut some load/stores by merging transpose with filter
  459. uint8_t trans[8*4];
  460. transpose4x4(trans, pix-2, 8, stride);
  461. transpose4x4(trans+4, pix-2+4*stride, 8, stride);
  462. h264_loop_filter_chroma_mmx2(trans+2*8, 8, alpha-1, beta-1, tc0);
  463. transpose4x4(pix-2, trans, stride, 8);
  464. transpose4x4(pix-2+4*stride, trans+4, stride, 8);
  465. }
  466. // p0 = (p0 + q1 + 2*p1 + 2) >> 2
  467. #define H264_FILTER_CHROMA4(p0, p1, q1, one) \
  468. "movq "#p0", %%mm4 \n\t"\
  469. "pxor "#q1", %%mm4 \n\t"\
  470. "pand "#one", %%mm4 \n\t" /* mm4 = (p0^q1)&1 */\
  471. "pavgb "#q1", "#p0" \n\t"\
  472. "psubusb %%mm4, "#p0" \n\t"\
  473. "pavgb "#p1", "#p0" \n\t" /* dst = avg(p1, avg(p0,q1) - ((p0^q1)&1)) */\
  474. static inline void h264_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha1, int beta1)
  475. {
  476. asm volatile(
  477. "movq (%0), %%mm0 \n\t"
  478. "movq (%0,%2), %%mm1 \n\t"
  479. "movq (%1), %%mm2 \n\t"
  480. "movq (%1,%2), %%mm3 \n\t"
  481. H264_DEBLOCK_MASK(%3, %4)
  482. "movq %%mm1, %%mm5 \n\t"
  483. "movq %%mm2, %%mm6 \n\t"
  484. H264_FILTER_CHROMA4(%%mm1, %%mm0, %%mm3, %5) //p0'
  485. H264_FILTER_CHROMA4(%%mm2, %%mm3, %%mm0, %5) //q0'
  486. "psubb %%mm5, %%mm1 \n\t"
  487. "psubb %%mm6, %%mm2 \n\t"
  488. "pand %%mm7, %%mm1 \n\t"
  489. "pand %%mm7, %%mm2 \n\t"
  490. "paddb %%mm5, %%mm1 \n\t"
  491. "paddb %%mm6, %%mm2 \n\t"
  492. "movq %%mm1, (%0,%2) \n\t"
  493. "movq %%mm2, (%1) \n\t"
  494. :: "r"(pix-2*stride), "r"(pix), "r"((long)stride),
  495. "m"(alpha1), "m"(beta1), "m"(mm_bone)
  496. );
  497. }
  498. static void h264_v_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha, int beta)
  499. {
  500. h264_loop_filter_chroma_intra_mmx2(pix, stride, alpha-1, beta-1);
  501. }
  502. static void h264_h_loop_filter_chroma_intra_mmx2(uint8_t *pix, int stride, int alpha, int beta)
  503. {
  504. //FIXME: could cut some load/stores by merging transpose with filter
  505. uint8_t trans[8*4];
  506. transpose4x4(trans, pix-2, 8, stride);
  507. transpose4x4(trans+4, pix-2+4*stride, 8, stride);
  508. h264_loop_filter_chroma_intra_mmx2(trans+2*8, 8, alpha-1, beta-1);
  509. transpose4x4(pix-2, trans, stride, 8);
  510. transpose4x4(pix-2+4*stride, trans+4, stride, 8);
  511. }
  512. 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],
  513. int bidir, int edges, int step, int mask_mv0, int mask_mv1 ) {
  514. int dir;
  515. asm volatile(
  516. "pxor %%mm7, %%mm7 \n\t"
  517. "movq %0, %%mm6 \n\t"
  518. "movq %1, %%mm5 \n\t"
  519. "movq %2, %%mm4 \n\t"
  520. ::"m"(ff_pb_1), "m"(ff_pb_3), "m"(ff_pb_7)
  521. );
  522. // could do a special case for dir==0 && edges==1, but it only reduces the
  523. // average filter time by 1.2%
  524. for( dir=1; dir>=0; dir-- ) {
  525. const int d_idx = dir ? -8 : -1;
  526. const int mask_mv = dir ? mask_mv1 : mask_mv0;
  527. const uint64_t mask_dir = dir ? 0 : 0xffffffffffffffffULL;
  528. int b_idx, edge, l;
  529. for( b_idx=12, edge=0; edge<edges; edge+=step, b_idx+=8*step ) {
  530. asm volatile(
  531. "pand %0, %%mm0 \n\t"
  532. ::"m"(mask_dir)
  533. );
  534. if(!(mask_mv & edge)) {
  535. asm volatile("pxor %%mm0, %%mm0 \n\t":);
  536. for( l = bidir; l >= 0; l-- ) {
  537. asm volatile(
  538. "movd %0, %%mm1 \n\t"
  539. "punpckldq %1, %%mm1 \n\t"
  540. "movq %%mm1, %%mm2 \n\t"
  541. "psrlw $7, %%mm2 \n\t"
  542. "pand %%mm6, %%mm2 \n\t"
  543. "por %%mm2, %%mm1 \n\t" // ref_cache with -2 mapped to -1
  544. "punpckldq %%mm1, %%mm2 \n\t"
  545. "pcmpeqb %%mm2, %%mm1 \n\t"
  546. "paddb %%mm6, %%mm1 \n\t"
  547. "punpckhbw %%mm7, %%mm1 \n\t" // ref[b] != ref[bn]
  548. "por %%mm1, %%mm0 \n\t"
  549. "movq %2, %%mm1 \n\t"
  550. "movq %3, %%mm2 \n\t"
  551. "psubw %4, %%mm1 \n\t"
  552. "psubw %5, %%mm2 \n\t"
  553. "packsswb %%mm2, %%mm1 \n\t"
  554. "paddb %%mm5, %%mm1 \n\t"
  555. "pminub %%mm4, %%mm1 \n\t"
  556. "pcmpeqb %%mm4, %%mm1 \n\t" // abs(mv[b] - mv[bn]) >= limit
  557. "por %%mm1, %%mm0 \n\t"
  558. ::"m"(ref[l][b_idx]),
  559. "m"(ref[l][b_idx+d_idx]),
  560. "m"(mv[l][b_idx][0]),
  561. "m"(mv[l][b_idx+2][0]),
  562. "m"(mv[l][b_idx+d_idx][0]),
  563. "m"(mv[l][b_idx+d_idx+2][0])
  564. );
  565. }
  566. }
  567. asm volatile(
  568. "movd %0, %%mm1 \n\t"
  569. "por %1, %%mm1 \n\t"
  570. "punpcklbw %%mm7, %%mm1 \n\t"
  571. "pcmpgtw %%mm7, %%mm1 \n\t" // nnz[b] || nnz[bn]
  572. ::"m"(nnz[b_idx]),
  573. "m"(nnz[b_idx+d_idx])
  574. );
  575. asm volatile(
  576. "pcmpeqw %%mm7, %%mm0 \n\t"
  577. "pcmpeqw %%mm7, %%mm0 \n\t"
  578. "psrlw $15, %%mm0 \n\t" // nonzero -> 1
  579. "psrlw $14, %%mm1 \n\t"
  580. "movq %%mm0, %%mm2 \n\t"
  581. "por %%mm1, %%mm2 \n\t"
  582. "psrlw $1, %%mm1 \n\t"
  583. "pandn %%mm2, %%mm1 \n\t"
  584. "movq %%mm1, %0 \n\t"
  585. :"=m"(*bS[dir][edge])
  586. ::"memory"
  587. );
  588. }
  589. edges = 4;
  590. step = 1;
  591. }
  592. asm volatile(
  593. "movq (%0), %%mm0 \n\t"
  594. "movq 8(%0), %%mm1 \n\t"
  595. "movq 16(%0), %%mm2 \n\t"
  596. "movq 24(%0), %%mm3 \n\t"
  597. TRANSPOSE4(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4)
  598. "movq %%mm0, (%0) \n\t"
  599. "movq %%mm3, 8(%0) \n\t"
  600. "movq %%mm4, 16(%0) \n\t"
  601. "movq %%mm2, 24(%0) \n\t"
  602. ::"r"(bS[0])
  603. :"memory"
  604. );
  605. }
  606. /***********************************/
  607. /* motion compensation */
  608. #define QPEL_H264V(A,B,C,D,E,F,OP)\
  609. "movd (%0), "#F" \n\t"\
  610. "movq "#C", %%mm6 \n\t"\
  611. "paddw "#D", %%mm6 \n\t"\
  612. "psllw $2, %%mm6 \n\t"\
  613. "psubw "#B", %%mm6 \n\t"\
  614. "psubw "#E", %%mm6 \n\t"\
  615. "pmullw %4, %%mm6 \n\t"\
  616. "add %2, %0 \n\t"\
  617. "punpcklbw %%mm7, "#F" \n\t"\
  618. "paddw %5, "#A" \n\t"\
  619. "paddw "#F", "#A" \n\t"\
  620. "paddw "#A", %%mm6 \n\t"\
  621. "psraw $5, %%mm6 \n\t"\
  622. "packuswb %%mm6, %%mm6 \n\t"\
  623. OP(%%mm6, (%1), A, d)\
  624. "add %3, %1 \n\t"
  625. #define QPEL_H264HV(A,B,C,D,E,F,OF)\
  626. "movd (%0), "#F" \n\t"\
  627. "movq "#C", %%mm6 \n\t"\
  628. "paddw "#D", %%mm6 \n\t"\
  629. "psllw $2, %%mm6 \n\t"\
  630. "psubw "#B", %%mm6 \n\t"\
  631. "psubw "#E", %%mm6 \n\t"\
  632. "pmullw %3, %%mm6 \n\t"\
  633. "add %2, %0 \n\t"\
  634. "punpcklbw %%mm7, "#F" \n\t"\
  635. "paddw "#F", "#A" \n\t"\
  636. "paddw "#A", %%mm6 \n\t"\
  637. "movq %%mm6, "#OF"(%1) \n\t"
  638. #define QPEL_H264(OPNAME, OP, MMX)\
  639. static void OPNAME ## h264_qpel4_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  640. int h=4;\
  641. \
  642. asm volatile(\
  643. "pxor %%mm7, %%mm7 \n\t"\
  644. "movq %5, %%mm4 \n\t"\
  645. "movq %6, %%mm5 \n\t"\
  646. "1: \n\t"\
  647. "movd -1(%0), %%mm1 \n\t"\
  648. "movd (%0), %%mm2 \n\t"\
  649. "movd 1(%0), %%mm3 \n\t"\
  650. "movd 2(%0), %%mm0 \n\t"\
  651. "punpcklbw %%mm7, %%mm1 \n\t"\
  652. "punpcklbw %%mm7, %%mm2 \n\t"\
  653. "punpcklbw %%mm7, %%mm3 \n\t"\
  654. "punpcklbw %%mm7, %%mm0 \n\t"\
  655. "paddw %%mm0, %%mm1 \n\t"\
  656. "paddw %%mm3, %%mm2 \n\t"\
  657. "movd -2(%0), %%mm0 \n\t"\
  658. "movd 3(%0), %%mm3 \n\t"\
  659. "punpcklbw %%mm7, %%mm0 \n\t"\
  660. "punpcklbw %%mm7, %%mm3 \n\t"\
  661. "paddw %%mm3, %%mm0 \n\t"\
  662. "psllw $2, %%mm2 \n\t"\
  663. "psubw %%mm1, %%mm2 \n\t"\
  664. "pmullw %%mm4, %%mm2 \n\t"\
  665. "paddw %%mm5, %%mm0 \n\t"\
  666. "paddw %%mm2, %%mm0 \n\t"\
  667. "psraw $5, %%mm0 \n\t"\
  668. "packuswb %%mm0, %%mm0 \n\t"\
  669. OP(%%mm0, (%1),%%mm6, d)\
  670. "add %3, %0 \n\t"\
  671. "add %4, %1 \n\t"\
  672. "decl %2 \n\t"\
  673. " jnz 1b \n\t"\
  674. : "+a"(src), "+c"(dst), "+m"(h)\
  675. : "d"((long)srcStride), "S"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  676. : "memory"\
  677. );\
  678. }\
  679. static void OPNAME ## h264_qpel4_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  680. int h=4;\
  681. asm volatile(\
  682. "pxor %%mm7, %%mm7 \n\t"\
  683. "movq %0, %%mm4 \n\t"\
  684. "movq %1, %%mm5 \n\t"\
  685. :: "m"(ff_pw_5), "m"(ff_pw_16)\
  686. );\
  687. do{\
  688. asm volatile(\
  689. "movd -1(%0), %%mm1 \n\t"\
  690. "movd (%0), %%mm2 \n\t"\
  691. "movd 1(%0), %%mm3 \n\t"\
  692. "movd 2(%0), %%mm0 \n\t"\
  693. "punpcklbw %%mm7, %%mm1 \n\t"\
  694. "punpcklbw %%mm7, %%mm2 \n\t"\
  695. "punpcklbw %%mm7, %%mm3 \n\t"\
  696. "punpcklbw %%mm7, %%mm0 \n\t"\
  697. "paddw %%mm0, %%mm1 \n\t"\
  698. "paddw %%mm3, %%mm2 \n\t"\
  699. "movd -2(%0), %%mm0 \n\t"\
  700. "movd 3(%0), %%mm3 \n\t"\
  701. "punpcklbw %%mm7, %%mm0 \n\t"\
  702. "punpcklbw %%mm7, %%mm3 \n\t"\
  703. "paddw %%mm3, %%mm0 \n\t"\
  704. "psllw $2, %%mm2 \n\t"\
  705. "psubw %%mm1, %%mm2 \n\t"\
  706. "pmullw %%mm4, %%mm2 \n\t"\
  707. "paddw %%mm5, %%mm0 \n\t"\
  708. "paddw %%mm2, %%mm0 \n\t"\
  709. "movd (%2), %%mm3 \n\t"\
  710. "psraw $5, %%mm0 \n\t"\
  711. "packuswb %%mm0, %%mm0 \n\t"\
  712. PAVGB" %%mm3, %%mm0 \n\t"\
  713. OP(%%mm0, (%1),%%mm6, d)\
  714. "add %4, %0 \n\t"\
  715. "add %4, %1 \n\t"\
  716. "add %3, %2 \n\t"\
  717. : "+a"(src), "+c"(dst), "+d"(src2)\
  718. : "D"((long)src2Stride), "S"((long)dstStride)\
  719. : "memory"\
  720. );\
  721. }while(--h);\
  722. }\
  723. static void OPNAME ## h264_qpel4_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  724. src -= 2*srcStride;\
  725. asm volatile(\
  726. "pxor %%mm7, %%mm7 \n\t"\
  727. "movd (%0), %%mm0 \n\t"\
  728. "add %2, %0 \n\t"\
  729. "movd (%0), %%mm1 \n\t"\
  730. "add %2, %0 \n\t"\
  731. "movd (%0), %%mm2 \n\t"\
  732. "add %2, %0 \n\t"\
  733. "movd (%0), %%mm3 \n\t"\
  734. "add %2, %0 \n\t"\
  735. "movd (%0), %%mm4 \n\t"\
  736. "add %2, %0 \n\t"\
  737. "punpcklbw %%mm7, %%mm0 \n\t"\
  738. "punpcklbw %%mm7, %%mm1 \n\t"\
  739. "punpcklbw %%mm7, %%mm2 \n\t"\
  740. "punpcklbw %%mm7, %%mm3 \n\t"\
  741. "punpcklbw %%mm7, %%mm4 \n\t"\
  742. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  743. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  744. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  745. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  746. \
  747. : "+a"(src), "+c"(dst)\
  748. : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  749. : "memory"\
  750. );\
  751. }\
  752. static void OPNAME ## h264_qpel4_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  753. int h=4;\
  754. int w=3;\
  755. src -= 2*srcStride+2;\
  756. while(w--){\
  757. asm volatile(\
  758. "pxor %%mm7, %%mm7 \n\t"\
  759. "movd (%0), %%mm0 \n\t"\
  760. "add %2, %0 \n\t"\
  761. "movd (%0), %%mm1 \n\t"\
  762. "add %2, %0 \n\t"\
  763. "movd (%0), %%mm2 \n\t"\
  764. "add %2, %0 \n\t"\
  765. "movd (%0), %%mm3 \n\t"\
  766. "add %2, %0 \n\t"\
  767. "movd (%0), %%mm4 \n\t"\
  768. "add %2, %0 \n\t"\
  769. "punpcklbw %%mm7, %%mm0 \n\t"\
  770. "punpcklbw %%mm7, %%mm1 \n\t"\
  771. "punpcklbw %%mm7, %%mm2 \n\t"\
  772. "punpcklbw %%mm7, %%mm3 \n\t"\
  773. "punpcklbw %%mm7, %%mm4 \n\t"\
  774. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 0*8*3)\
  775. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 1*8*3)\
  776. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 2*8*3)\
  777. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 3*8*3)\
  778. \
  779. : "+a"(src)\
  780. : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
  781. : "memory"\
  782. );\
  783. tmp += 4;\
  784. src += 4 - 9*srcStride;\
  785. }\
  786. tmp -= 3*4;\
  787. asm volatile(\
  788. "movq %4, %%mm6 \n\t"\
  789. "1: \n\t"\
  790. "movq (%0), %%mm0 \n\t"\
  791. "paddw 10(%0), %%mm0 \n\t"\
  792. "movq 2(%0), %%mm1 \n\t"\
  793. "paddw 8(%0), %%mm1 \n\t"\
  794. "movq 4(%0), %%mm2 \n\t"\
  795. "paddw 6(%0), %%mm2 \n\t"\
  796. "psubw %%mm1, %%mm0 \n\t"/*a-b (abccba)*/\
  797. "psraw $2, %%mm0 \n\t"/*(a-b)/4 */\
  798. "psubw %%mm1, %%mm0 \n\t"/*(a-b)/4-b */\
  799. "paddsw %%mm2, %%mm0 \n\t"\
  800. "psraw $2, %%mm0 \n\t"/*((a-b)/4-b+c)/4 */\
  801. "paddw %%mm6, %%mm2 \n\t"\
  802. "paddw %%mm2, %%mm0 \n\t"/*(a-5*b+20*c)/16 +32 */\
  803. "psraw $6, %%mm0 \n\t"\
  804. "packuswb %%mm0, %%mm0 \n\t"\
  805. OP(%%mm0, (%1),%%mm7, d)\
  806. "add $24, %0 \n\t"\
  807. "add %3, %1 \n\t"\
  808. "decl %2 \n\t"\
  809. " jnz 1b \n\t"\
  810. : "+a"(tmp), "+c"(dst), "+m"(h)\
  811. : "S"((long)dstStride), "m"(ff_pw_32)\
  812. : "memory"\
  813. );\
  814. }\
  815. \
  816. static void OPNAME ## h264_qpel8_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  817. int h=8;\
  818. asm volatile(\
  819. "pxor %%mm7, %%mm7 \n\t"\
  820. "movq %5, %%mm6 \n\t"\
  821. "1: \n\t"\
  822. "movq (%0), %%mm0 \n\t"\
  823. "movq 1(%0), %%mm2 \n\t"\
  824. "movq %%mm0, %%mm1 \n\t"\
  825. "movq %%mm2, %%mm3 \n\t"\
  826. "punpcklbw %%mm7, %%mm0 \n\t"\
  827. "punpckhbw %%mm7, %%mm1 \n\t"\
  828. "punpcklbw %%mm7, %%mm2 \n\t"\
  829. "punpckhbw %%mm7, %%mm3 \n\t"\
  830. "paddw %%mm2, %%mm0 \n\t"\
  831. "paddw %%mm3, %%mm1 \n\t"\
  832. "psllw $2, %%mm0 \n\t"\
  833. "psllw $2, %%mm1 \n\t"\
  834. "movq -1(%0), %%mm2 \n\t"\
  835. "movq 2(%0), %%mm4 \n\t"\
  836. "movq %%mm2, %%mm3 \n\t"\
  837. "movq %%mm4, %%mm5 \n\t"\
  838. "punpcklbw %%mm7, %%mm2 \n\t"\
  839. "punpckhbw %%mm7, %%mm3 \n\t"\
  840. "punpcklbw %%mm7, %%mm4 \n\t"\
  841. "punpckhbw %%mm7, %%mm5 \n\t"\
  842. "paddw %%mm4, %%mm2 \n\t"\
  843. "paddw %%mm3, %%mm5 \n\t"\
  844. "psubw %%mm2, %%mm0 \n\t"\
  845. "psubw %%mm5, %%mm1 \n\t"\
  846. "pmullw %%mm6, %%mm0 \n\t"\
  847. "pmullw %%mm6, %%mm1 \n\t"\
  848. "movd -2(%0), %%mm2 \n\t"\
  849. "movd 7(%0), %%mm5 \n\t"\
  850. "punpcklbw %%mm7, %%mm2 \n\t"\
  851. "punpcklbw %%mm7, %%mm5 \n\t"\
  852. "paddw %%mm3, %%mm2 \n\t"\
  853. "paddw %%mm5, %%mm4 \n\t"\
  854. "movq %6, %%mm5 \n\t"\
  855. "paddw %%mm5, %%mm2 \n\t"\
  856. "paddw %%mm5, %%mm4 \n\t"\
  857. "paddw %%mm2, %%mm0 \n\t"\
  858. "paddw %%mm4, %%mm1 \n\t"\
  859. "psraw $5, %%mm0 \n\t"\
  860. "psraw $5, %%mm1 \n\t"\
  861. "packuswb %%mm1, %%mm0 \n\t"\
  862. OP(%%mm0, (%1),%%mm5, q)\
  863. "add %3, %0 \n\t"\
  864. "add %4, %1 \n\t"\
  865. "decl %2 \n\t"\
  866. " jnz 1b \n\t"\
  867. : "+a"(src), "+c"(dst), "+m"(h)\
  868. : "d"((long)srcStride), "S"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  869. : "memory"\
  870. );\
  871. }\
  872. \
  873. static void OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  874. int h=8;\
  875. asm volatile(\
  876. "pxor %%mm7, %%mm7 \n\t"\
  877. "movq %0, %%mm6 \n\t"\
  878. :: "m"(ff_pw_5)\
  879. );\
  880. do{\
  881. asm volatile(\
  882. "movq (%0), %%mm0 \n\t"\
  883. "movq 1(%0), %%mm2 \n\t"\
  884. "movq %%mm0, %%mm1 \n\t"\
  885. "movq %%mm2, %%mm3 \n\t"\
  886. "punpcklbw %%mm7, %%mm0 \n\t"\
  887. "punpckhbw %%mm7, %%mm1 \n\t"\
  888. "punpcklbw %%mm7, %%mm2 \n\t"\
  889. "punpckhbw %%mm7, %%mm3 \n\t"\
  890. "paddw %%mm2, %%mm0 \n\t"\
  891. "paddw %%mm3, %%mm1 \n\t"\
  892. "psllw $2, %%mm0 \n\t"\
  893. "psllw $2, %%mm1 \n\t"\
  894. "movq -1(%0), %%mm2 \n\t"\
  895. "movq 2(%0), %%mm4 \n\t"\
  896. "movq %%mm2, %%mm3 \n\t"\
  897. "movq %%mm4, %%mm5 \n\t"\
  898. "punpcklbw %%mm7, %%mm2 \n\t"\
  899. "punpckhbw %%mm7, %%mm3 \n\t"\
  900. "punpcklbw %%mm7, %%mm4 \n\t"\
  901. "punpckhbw %%mm7, %%mm5 \n\t"\
  902. "paddw %%mm4, %%mm2 \n\t"\
  903. "paddw %%mm3, %%mm5 \n\t"\
  904. "psubw %%mm2, %%mm0 \n\t"\
  905. "psubw %%mm5, %%mm1 \n\t"\
  906. "pmullw %%mm6, %%mm0 \n\t"\
  907. "pmullw %%mm6, %%mm1 \n\t"\
  908. "movd -2(%0), %%mm2 \n\t"\
  909. "movd 7(%0), %%mm5 \n\t"\
  910. "punpcklbw %%mm7, %%mm2 \n\t"\
  911. "punpcklbw %%mm7, %%mm5 \n\t"\
  912. "paddw %%mm3, %%mm2 \n\t"\
  913. "paddw %%mm5, %%mm4 \n\t"\
  914. "movq %5, %%mm5 \n\t"\
  915. "paddw %%mm5, %%mm2 \n\t"\
  916. "paddw %%mm5, %%mm4 \n\t"\
  917. "paddw %%mm2, %%mm0 \n\t"\
  918. "paddw %%mm4, %%mm1 \n\t"\
  919. "psraw $5, %%mm0 \n\t"\
  920. "psraw $5, %%mm1 \n\t"\
  921. "movq (%2), %%mm4 \n\t"\
  922. "packuswb %%mm1, %%mm0 \n\t"\
  923. PAVGB" %%mm4, %%mm0 \n\t"\
  924. OP(%%mm0, (%1),%%mm5, q)\
  925. "add %4, %0 \n\t"\
  926. "add %4, %1 \n\t"\
  927. "add %3, %2 \n\t"\
  928. : "+a"(src), "+c"(dst), "+d"(src2)\
  929. : "D"((long)src2Stride), "S"((long)dstStride),\
  930. "m"(ff_pw_16)\
  931. : "memory"\
  932. );\
  933. }while(--h);\
  934. }\
  935. \
  936. static inline void OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride, int h){\
  937. int w= 2;\
  938. src -= 2*srcStride;\
  939. \
  940. while(w--){\
  941. asm volatile(\
  942. "pxor %%mm7, %%mm7 \n\t"\
  943. "movd (%0), %%mm0 \n\t"\
  944. "add %2, %0 \n\t"\
  945. "movd (%0), %%mm1 \n\t"\
  946. "add %2, %0 \n\t"\
  947. "movd (%0), %%mm2 \n\t"\
  948. "add %2, %0 \n\t"\
  949. "movd (%0), %%mm3 \n\t"\
  950. "add %2, %0 \n\t"\
  951. "movd (%0), %%mm4 \n\t"\
  952. "add %2, %0 \n\t"\
  953. "punpcklbw %%mm7, %%mm0 \n\t"\
  954. "punpcklbw %%mm7, %%mm1 \n\t"\
  955. "punpcklbw %%mm7, %%mm2 \n\t"\
  956. "punpcklbw %%mm7, %%mm3 \n\t"\
  957. "punpcklbw %%mm7, %%mm4 \n\t"\
  958. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  959. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  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. \
  967. : "+a"(src), "+c"(dst)\
  968. : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  969. : "memory"\
  970. );\
  971. if(h==16){\
  972. asm volatile(\
  973. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  974. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  975. QPEL_H264V(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, OP)\
  976. QPEL_H264V(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, OP)\
  977. QPEL_H264V(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, OP)\
  978. QPEL_H264V(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, OP)\
  979. QPEL_H264V(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, OP)\
  980. QPEL_H264V(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, OP)\
  981. \
  982. : "+a"(src), "+c"(dst)\
  983. : "S"((long)srcStride), "D"((long)dstStride), "m"(ff_pw_5), "m"(ff_pw_16)\
  984. : "memory"\
  985. );\
  986. }\
  987. src += 4-(h+5)*srcStride;\
  988. dst += 4-h*dstStride;\
  989. }\
  990. }\
  991. static inline void OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride, int size){\
  992. int h = size;\
  993. int w = (size+8)>>2;\
  994. src -= 2*srcStride+2;\
  995. while(w--){\
  996. asm volatile(\
  997. "pxor %%mm7, %%mm7 \n\t"\
  998. "movd (%0), %%mm0 \n\t"\
  999. "add %2, %0 \n\t"\
  1000. "movd (%0), %%mm1 \n\t"\
  1001. "add %2, %0 \n\t"\
  1002. "movd (%0), %%mm2 \n\t"\
  1003. "add %2, %0 \n\t"\
  1004. "movd (%0), %%mm3 \n\t"\
  1005. "add %2, %0 \n\t"\
  1006. "movd (%0), %%mm4 \n\t"\
  1007. "add %2, %0 \n\t"\
  1008. "punpcklbw %%mm7, %%mm0 \n\t"\
  1009. "punpcklbw %%mm7, %%mm1 \n\t"\
  1010. "punpcklbw %%mm7, %%mm2 \n\t"\
  1011. "punpcklbw %%mm7, %%mm3 \n\t"\
  1012. "punpcklbw %%mm7, %%mm4 \n\t"\
  1013. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 0*48)\
  1014. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 1*48)\
  1015. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 2*48)\
  1016. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 3*48)\
  1017. QPEL_H264HV(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, 4*48)\
  1018. QPEL_H264HV(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, 5*48)\
  1019. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 6*48)\
  1020. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 7*48)\
  1021. : "+a"(src)\
  1022. : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
  1023. : "memory"\
  1024. );\
  1025. if(size==16){\
  1026. asm volatile(\
  1027. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 8*48)\
  1028. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 9*48)\
  1029. QPEL_H264HV(%%mm4, %%mm5, %%mm0, %%mm1, %%mm2, %%mm3, 10*48)\
  1030. QPEL_H264HV(%%mm5, %%mm0, %%mm1, %%mm2, %%mm3, %%mm4, 11*48)\
  1031. QPEL_H264HV(%%mm0, %%mm1, %%mm2, %%mm3, %%mm4, %%mm5, 12*48)\
  1032. QPEL_H264HV(%%mm1, %%mm2, %%mm3, %%mm4, %%mm5, %%mm0, 13*48)\
  1033. QPEL_H264HV(%%mm2, %%mm3, %%mm4, %%mm5, %%mm0, %%mm1, 14*48)\
  1034. QPEL_H264HV(%%mm3, %%mm4, %%mm5, %%mm0, %%mm1, %%mm2, 15*48)\
  1035. : "+a"(src)\
  1036. : "c"(tmp), "S"((long)srcStride), "m"(ff_pw_5)\
  1037. : "memory"\
  1038. );\
  1039. }\
  1040. tmp += 4;\
  1041. src += 4 - (size+5)*srcStride;\
  1042. }\
  1043. tmp -= size+8;\
  1044. w = size>>4;\
  1045. do{\
  1046. h = size;\
  1047. asm volatile(\
  1048. "movq %4, %%mm6 \n\t"\
  1049. "1: \n\t"\
  1050. "movq (%0), %%mm0 \n\t"\
  1051. "movq 8(%0), %%mm3 \n\t"\
  1052. "movq 2(%0), %%mm1 \n\t"\
  1053. "movq 10(%0), %%mm4 \n\t"\
  1054. "paddw %%mm4, %%mm0 \n\t"\
  1055. "paddw %%mm3, %%mm1 \n\t"\
  1056. "paddw 18(%0), %%mm3 \n\t"\
  1057. "paddw 16(%0), %%mm4 \n\t"\
  1058. "movq 4(%0), %%mm2 \n\t"\
  1059. "movq 12(%0), %%mm5 \n\t"\
  1060. "paddw 6(%0), %%mm2 \n\t"\
  1061. "paddw 14(%0), %%mm5 \n\t"\
  1062. "psubw %%mm1, %%mm0 \n\t"\
  1063. "psubw %%mm4, %%mm3 \n\t"\
  1064. "psraw $2, %%mm0 \n\t"\
  1065. "psraw $2, %%mm3 \n\t"\
  1066. "psubw %%mm1, %%mm0 \n\t"\
  1067. "psubw %%mm4, %%mm3 \n\t"\
  1068. "paddsw %%mm2, %%mm0 \n\t"\
  1069. "paddsw %%mm5, %%mm3 \n\t"\
  1070. "psraw $2, %%mm0 \n\t"\
  1071. "psraw $2, %%mm3 \n\t"\
  1072. "paddw %%mm6, %%mm2 \n\t"\
  1073. "paddw %%mm6, %%mm5 \n\t"\
  1074. "paddw %%mm2, %%mm0 \n\t"\
  1075. "paddw %%mm5, %%mm3 \n\t"\
  1076. "psraw $6, %%mm0 \n\t"\
  1077. "psraw $6, %%mm3 \n\t"\
  1078. "packuswb %%mm3, %%mm0 \n\t"\
  1079. OP(%%mm0, (%1),%%mm7, q)\
  1080. "add $48, %0 \n\t"\
  1081. "add %3, %1 \n\t"\
  1082. "decl %2 \n\t"\
  1083. " jnz 1b \n\t"\
  1084. : "+a"(tmp), "+c"(dst), "+m"(h)\
  1085. : "S"((long)dstStride), "m"(ff_pw_32)\
  1086. : "memory"\
  1087. );\
  1088. tmp += 8 - size*24;\
  1089. dst += 8 - size*dstStride;\
  1090. }while(w--);\
  1091. }\
  1092. \
  1093. static void OPNAME ## h264_qpel8_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  1094. OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 8);\
  1095. }\
  1096. static void OPNAME ## h264_qpel16_v_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  1097. OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst , src , dstStride, srcStride, 16);\
  1098. OPNAME ## h264_qpel8or16_v_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride, 16);\
  1099. }\
  1100. \
  1101. static void OPNAME ## h264_qpel16_h_lowpass_ ## MMX(uint8_t *dst, uint8_t *src, int dstStride, int srcStride){\
  1102. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  1103. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  1104. src += 8*srcStride;\
  1105. dst += 8*dstStride;\
  1106. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst , src , dstStride, srcStride);\
  1107. OPNAME ## h264_qpel8_h_lowpass_ ## MMX(dst+8, src+8, dstStride, srcStride);\
  1108. }\
  1109. \
  1110. static void OPNAME ## h264_qpel16_h_lowpass_l2_ ## MMX(uint8_t *dst, uint8_t *src, uint8_t *src2, int dstStride, int src2Stride){\
  1111. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  1112. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  1113. src += 8*dstStride;\
  1114. dst += 8*dstStride;\
  1115. src2 += 8*src2Stride;\
  1116. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst , src , src2 , dstStride, src2Stride);\
  1117. OPNAME ## h264_qpel8_h_lowpass_l2_ ## MMX(dst+8, src+8, src2+8, dstStride, src2Stride);\
  1118. }\
  1119. \
  1120. static void OPNAME ## h264_qpel8_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  1121. OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst , tmp , src , dstStride, tmpStride, srcStride, 8);\
  1122. }\
  1123. \
  1124. static void OPNAME ## h264_qpel16_hv_lowpass_ ## MMX(uint8_t *dst, int16_t *tmp, uint8_t *src, int dstStride, int tmpStride, int srcStride){\
  1125. OPNAME ## h264_qpel8or16_hv_lowpass_ ## MMX(dst , tmp , src , dstStride, tmpStride, srcStride, 16);\
  1126. }\
  1127. \
  1128. static void OPNAME ## pixels4_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  1129. {\
  1130. asm volatile(\
  1131. "movq %5, %%mm6 \n\t"\
  1132. "movq (%1), %%mm0 \n\t"\
  1133. "movq 24(%1), %%mm1 \n\t"\
  1134. "paddw %%mm6, %%mm0 \n\t"\
  1135. "paddw %%mm6, %%mm1 \n\t"\
  1136. "psraw $5, %%mm0 \n\t"\
  1137. "psraw $5, %%mm1 \n\t"\
  1138. "packuswb %%mm0, %%mm0 \n\t"\
  1139. "packuswb %%mm1, %%mm1 \n\t"\
  1140. PAVGB" (%0), %%mm0 \n\t"\
  1141. PAVGB" (%0,%3), %%mm1 \n\t"\
  1142. OP(%%mm0, (%2), %%mm4, d)\
  1143. OP(%%mm1, (%2,%4), %%mm5, d)\
  1144. "lea (%0,%3,2), %0 \n\t"\
  1145. "lea (%2,%4,2), %2 \n\t"\
  1146. "movq 48(%1), %%mm0 \n\t"\
  1147. "movq 72(%1), %%mm1 \n\t"\
  1148. "paddw %%mm6, %%mm0 \n\t"\
  1149. "paddw %%mm6, %%mm1 \n\t"\
  1150. "psraw $5, %%mm0 \n\t"\
  1151. "psraw $5, %%mm1 \n\t"\
  1152. "packuswb %%mm0, %%mm0 \n\t"\
  1153. "packuswb %%mm1, %%mm1 \n\t"\
  1154. PAVGB" (%0), %%mm0 \n\t"\
  1155. PAVGB" (%0,%3), %%mm1 \n\t"\
  1156. OP(%%mm0, (%2), %%mm4, d)\
  1157. OP(%%mm1, (%2,%4), %%mm5, d)\
  1158. :"+a"(src8), "+c"(src16), "+d"(dst)\
  1159. :"S"((long)src8Stride), "D"((long)dstStride), "m"(ff_pw_16)\
  1160. :"memory");\
  1161. }\
  1162. static void OPNAME ## pixels8_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  1163. {\
  1164. asm volatile(\
  1165. "movq %0, %%mm6 \n\t"\
  1166. ::"m"(ff_pw_16)\
  1167. );\
  1168. while(h--){\
  1169. asm volatile(\
  1170. "movq (%1), %%mm0 \n\t"\
  1171. "movq 8(%1), %%mm1 \n\t"\
  1172. "paddw %%mm6, %%mm0 \n\t"\
  1173. "paddw %%mm6, %%mm1 \n\t"\
  1174. "psraw $5, %%mm0 \n\t"\
  1175. "psraw $5, %%mm1 \n\t"\
  1176. "packuswb %%mm1, %%mm0 \n\t"\
  1177. PAVGB" (%0), %%mm0 \n\t"\
  1178. OP(%%mm0, (%2), %%mm5, q)\
  1179. ::"a"(src8), "c"(src16), "d"(dst)\
  1180. :"memory");\
  1181. src8 += src8Stride;\
  1182. src16 += 24;\
  1183. dst += dstStride;\
  1184. }\
  1185. }\
  1186. static void OPNAME ## pixels16_l2_shift5_ ## MMX(uint8_t *dst, int16_t *src16, uint8_t *src8, int dstStride, int src8Stride, int h)\
  1187. {\
  1188. OPNAME ## pixels8_l2_shift5_ ## MMX(dst , src16 , src8 , dstStride, src8Stride, h);\
  1189. OPNAME ## pixels8_l2_shift5_ ## MMX(dst+8, src16+8, src8+8, dstStride, src8Stride, h);\
  1190. }\
  1191. #define H264_MC(OPNAME, SIZE, MMX) \
  1192. static void OPNAME ## h264_qpel ## SIZE ## _mc00_ ## MMX (uint8_t *dst, uint8_t *src, int stride){\
  1193. OPNAME ## pixels ## SIZE ## _mmx(dst, src, stride, SIZE);\
  1194. }\
  1195. \
  1196. static void OPNAME ## h264_qpel ## SIZE ## _mc10_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1197. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src, stride, stride);\
  1198. }\
  1199. \
  1200. static void OPNAME ## h264_qpel ## SIZE ## _mc20_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1201. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_ ## MMX(dst, src, stride, stride);\
  1202. }\
  1203. \
  1204. static void OPNAME ## h264_qpel ## SIZE ## _mc30_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1205. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, src+1, stride, stride);\
  1206. }\
  1207. \
  1208. static void OPNAME ## h264_qpel ## SIZE ## _mc01_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1209. uint64_t temp[SIZE*SIZE/8];\
  1210. uint8_t * const half= (uint8_t*)temp;\
  1211. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(half, src, SIZE, stride);\
  1212. OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src, half, stride, stride, SIZE);\
  1213. }\
  1214. \
  1215. static void OPNAME ## h264_qpel ## SIZE ## _mc02_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1216. OPNAME ## h264_qpel ## SIZE ## _v_lowpass_ ## MMX(dst, src, stride, stride);\
  1217. }\
  1218. \
  1219. static void OPNAME ## h264_qpel ## SIZE ## _mc03_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1220. uint64_t temp[SIZE*SIZE/8];\
  1221. uint8_t * const half= (uint8_t*)temp;\
  1222. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(half, src, SIZE, stride);\
  1223. OPNAME ## pixels ## SIZE ## _l2_ ## MMX(dst, src+stride, half, stride, stride, SIZE);\
  1224. }\
  1225. \
  1226. static void OPNAME ## h264_qpel ## SIZE ## _mc11_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1227. uint64_t temp[SIZE*SIZE/8];\
  1228. uint8_t * const halfV= (uint8_t*)temp;\
  1229. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src, SIZE, stride);\
  1230. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfV, stride, SIZE);\
  1231. }\
  1232. \
  1233. static void OPNAME ## h264_qpel ## SIZE ## _mc31_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1234. uint64_t temp[SIZE*SIZE/8];\
  1235. uint8_t * const halfV= (uint8_t*)temp;\
  1236. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src+1, SIZE, stride);\
  1237. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfV, stride, SIZE);\
  1238. }\
  1239. \
  1240. static void OPNAME ## h264_qpel ## SIZE ## _mc13_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1241. uint64_t temp[SIZE*SIZE/8];\
  1242. uint8_t * const halfV= (uint8_t*)temp;\
  1243. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src, SIZE, stride);\
  1244. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfV, stride, SIZE);\
  1245. }\
  1246. \
  1247. static void OPNAME ## h264_qpel ## SIZE ## _mc33_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1248. uint64_t temp[SIZE*SIZE/8];\
  1249. uint8_t * const halfV= (uint8_t*)temp;\
  1250. put_h264_qpel ## SIZE ## _v_lowpass_ ## MMX(halfV, src+1, SIZE, stride);\
  1251. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfV, stride, SIZE);\
  1252. }\
  1253. \
  1254. static void OPNAME ## h264_qpel ## SIZE ## _mc22_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1255. uint64_t temp[SIZE*(SIZE<8?12:24)/4];\
  1256. int16_t * const tmp= (int16_t*)temp;\
  1257. OPNAME ## h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(dst, tmp, src, stride, SIZE, stride);\
  1258. }\
  1259. \
  1260. static void OPNAME ## h264_qpel ## SIZE ## _mc21_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1261. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1262. uint8_t * const halfHV= (uint8_t*)temp;\
  1263. int16_t * const tmp= ((int16_t*)temp) + SIZE*SIZE/2;\
  1264. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, tmp, src, SIZE, SIZE, stride);\
  1265. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src, halfHV, stride, SIZE);\
  1266. }\
  1267. \
  1268. static void OPNAME ## h264_qpel ## SIZE ## _mc23_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1269. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1270. uint8_t * const halfHV= (uint8_t*)temp;\
  1271. int16_t * const tmp= ((int16_t*)temp) + SIZE*SIZE/2;\
  1272. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, tmp, src, SIZE, SIZE, stride);\
  1273. OPNAME ## h264_qpel ## SIZE ## _h_lowpass_l2_ ## MMX(dst, src+stride, halfHV, stride, SIZE);\
  1274. }\
  1275. \
  1276. static void OPNAME ## h264_qpel ## SIZE ## _mc12_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1277. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1278. int16_t * const halfV= ((int16_t*)temp) + SIZE*SIZE/2;\
  1279. uint8_t * const halfHV= ((uint8_t*)temp);\
  1280. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  1281. OPNAME ## pixels ## SIZE ## _l2_shift5_ ## MMX(dst, halfV+2, halfHV, stride, SIZE, SIZE);\
  1282. }\
  1283. \
  1284. static void OPNAME ## h264_qpel ## SIZE ## _mc32_ ## MMX(uint8_t *dst, uint8_t *src, int stride){\
  1285. uint64_t temp[SIZE*(SIZE<8?12:24)/4 + SIZE*SIZE/8];\
  1286. int16_t * const halfV= ((int16_t*)temp) + SIZE*SIZE/2;\
  1287. uint8_t * const halfHV= ((uint8_t*)temp);\
  1288. put_h264_qpel ## SIZE ## _hv_lowpass_ ## MMX(halfHV, halfV, src, SIZE, SIZE, stride);\
  1289. OPNAME ## pixels ## SIZE ## _l2_shift5_ ## MMX(dst, halfV+3, halfHV, stride, SIZE, SIZE);\
  1290. }\
  1291. #define AVG_3DNOW_OP(a,b,temp, size) \
  1292. "mov" #size " " #b ", " #temp " \n\t"\
  1293. "pavgusb " #temp ", " #a " \n\t"\
  1294. "mov" #size " " #a ", " #b " \n\t"
  1295. #define AVG_MMX2_OP(a,b,temp, size) \
  1296. "mov" #size " " #b ", " #temp " \n\t"\
  1297. "pavgb " #temp ", " #a " \n\t"\
  1298. "mov" #size " " #a ", " #b " \n\t"
  1299. #define PAVGB "pavgusb"
  1300. QPEL_H264(put_, PUT_OP, 3dnow)
  1301. QPEL_H264(avg_, AVG_3DNOW_OP, 3dnow)
  1302. #undef PAVGB
  1303. #define PAVGB "pavgb"
  1304. QPEL_H264(put_, PUT_OP, mmx2)
  1305. QPEL_H264(avg_, AVG_MMX2_OP, mmx2)
  1306. #undef PAVGB
  1307. H264_MC(put_, 4, 3dnow)
  1308. H264_MC(put_, 8, 3dnow)
  1309. H264_MC(put_, 16,3dnow)
  1310. H264_MC(avg_, 4, 3dnow)
  1311. H264_MC(avg_, 8, 3dnow)
  1312. H264_MC(avg_, 16,3dnow)
  1313. H264_MC(put_, 4, mmx2)
  1314. H264_MC(put_, 8, mmx2)
  1315. H264_MC(put_, 16,mmx2)
  1316. H264_MC(avg_, 4, mmx2)
  1317. H264_MC(avg_, 8, mmx2)
  1318. H264_MC(avg_, 16,mmx2)
  1319. #define H264_CHROMA_OP(S,D)
  1320. #define H264_CHROMA_OP4(S,D,T)
  1321. #define H264_CHROMA_MC8_TMPL put_h264_chroma_mc8_mmx
  1322. #define H264_CHROMA_MC4_TMPL put_h264_chroma_mc4_mmx
  1323. #define H264_CHROMA_MC2_TMPL put_h264_chroma_mc2_mmx2
  1324. #define H264_CHROMA_MC8_MV0 put_pixels8_mmx
  1325. #include "dsputil_h264_template_mmx.c"
  1326. #undef H264_CHROMA_OP
  1327. #undef H264_CHROMA_OP4
  1328. #undef H264_CHROMA_MC8_TMPL
  1329. #undef H264_CHROMA_MC4_TMPL
  1330. #undef H264_CHROMA_MC2_TMPL
  1331. #undef H264_CHROMA_MC8_MV0
  1332. #define H264_CHROMA_OP(S,D) "pavgb " #S ", " #D " \n\t"
  1333. #define H264_CHROMA_OP4(S,D,T) "movd " #S ", " #T " \n\t"\
  1334. "pavgb " #T ", " #D " \n\t"
  1335. #define H264_CHROMA_MC8_TMPL avg_h264_chroma_mc8_mmx2
  1336. #define H264_CHROMA_MC4_TMPL avg_h264_chroma_mc4_mmx2
  1337. #define H264_CHROMA_MC2_TMPL avg_h264_chroma_mc2_mmx2
  1338. #define H264_CHROMA_MC8_MV0 avg_pixels8_mmx2
  1339. #include "dsputil_h264_template_mmx.c"
  1340. #undef H264_CHROMA_OP
  1341. #undef H264_CHROMA_OP4
  1342. #undef H264_CHROMA_MC8_TMPL
  1343. #undef H264_CHROMA_MC4_TMPL
  1344. #undef H264_CHROMA_MC2_TMPL
  1345. #undef H264_CHROMA_MC8_MV0
  1346. #define H264_CHROMA_OP(S,D) "pavgusb " #S ", " #D " \n\t"
  1347. #define H264_CHROMA_OP4(S,D,T) "movd " #S ", " #T " \n\t"\
  1348. "pavgusb " #T ", " #D " \n\t"
  1349. #define H264_CHROMA_MC8_TMPL avg_h264_chroma_mc8_3dnow
  1350. #define H264_CHROMA_MC4_TMPL avg_h264_chroma_mc4_3dnow
  1351. #define H264_CHROMA_MC8_MV0 avg_pixels8_3dnow
  1352. #include "dsputil_h264_template_mmx.c"
  1353. #undef H264_CHROMA_OP
  1354. #undef H264_CHROMA_OP4
  1355. #undef H264_CHROMA_MC8_TMPL
  1356. #undef H264_CHROMA_MC4_TMPL
  1357. #undef H264_CHROMA_MC8_MV0
  1358. /***********************************/
  1359. /* weighted prediction */
  1360. static inline void ff_h264_weight_WxH_mmx2(uint8_t *dst, int stride, int log2_denom, int weight, int offset, int w, int h)
  1361. {
  1362. int x, y;
  1363. offset <<= log2_denom;
  1364. offset += (1 << log2_denom) >> 1;
  1365. asm volatile(
  1366. "movd %0, %%mm4 \n\t"
  1367. "movd %1, %%mm5 \n\t"
  1368. "movd %2, %%mm6 \n\t"
  1369. "pshufw $0, %%mm4, %%mm4 \n\t"
  1370. "pshufw $0, %%mm5, %%mm5 \n\t"
  1371. "pxor %%mm7, %%mm7 \n\t"
  1372. :: "g"(weight), "g"(offset), "g"(log2_denom)
  1373. );
  1374. for(y=0; y<h; y+=2){
  1375. for(x=0; x<w; x+=4){
  1376. asm volatile(
  1377. "movd %0, %%mm0 \n\t"
  1378. "movd %1, %%mm1 \n\t"
  1379. "punpcklbw %%mm7, %%mm0 \n\t"
  1380. "punpcklbw %%mm7, %%mm1 \n\t"
  1381. "pmullw %%mm4, %%mm0 \n\t"
  1382. "pmullw %%mm4, %%mm1 \n\t"
  1383. "paddsw %%mm5, %%mm0 \n\t"
  1384. "paddsw %%mm5, %%mm1 \n\t"
  1385. "psraw %%mm6, %%mm0 \n\t"
  1386. "psraw %%mm6, %%mm1 \n\t"
  1387. "packuswb %%mm7, %%mm0 \n\t"
  1388. "packuswb %%mm7, %%mm1 \n\t"
  1389. "movd %%mm0, %0 \n\t"
  1390. "movd %%mm1, %1 \n\t"
  1391. : "+m"(*(uint32_t*)(dst+x)),
  1392. "+m"(*(uint32_t*)(dst+x+stride))
  1393. );
  1394. }
  1395. dst += 2*stride;
  1396. }
  1397. }
  1398. 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)
  1399. {
  1400. int x, y;
  1401. offset = ((offset + 1) | 1) << log2_denom;
  1402. asm volatile(
  1403. "movd %0, %%mm3 \n\t"
  1404. "movd %1, %%mm4 \n\t"
  1405. "movd %2, %%mm5 \n\t"
  1406. "movd %3, %%mm6 \n\t"
  1407. "pshufw $0, %%mm3, %%mm3 \n\t"
  1408. "pshufw $0, %%mm4, %%mm4 \n\t"
  1409. "pshufw $0, %%mm5, %%mm5 \n\t"
  1410. "pxor %%mm7, %%mm7 \n\t"
  1411. :: "g"(weightd), "g"(weights), "g"(offset), "g"(log2_denom+1)
  1412. );
  1413. for(y=0; y<h; y++){
  1414. for(x=0; x<w; x+=4){
  1415. asm volatile(
  1416. "movd %0, %%mm0 \n\t"
  1417. "movd %1, %%mm1 \n\t"
  1418. "punpcklbw %%mm7, %%mm0 \n\t"
  1419. "punpcklbw %%mm7, %%mm1 \n\t"
  1420. "pmullw %%mm3, %%mm0 \n\t"
  1421. "pmullw %%mm4, %%mm1 \n\t"
  1422. "paddsw %%mm1, %%mm0 \n\t"
  1423. "paddsw %%mm5, %%mm0 \n\t"
  1424. "psraw %%mm6, %%mm0 \n\t"
  1425. "packuswb %%mm0, %%mm0 \n\t"
  1426. "movd %%mm0, %0 \n\t"
  1427. : "+m"(*(uint32_t*)(dst+x))
  1428. : "m"(*(uint32_t*)(src+x))
  1429. );
  1430. }
  1431. src += stride;
  1432. dst += stride;
  1433. }
  1434. }
  1435. #define H264_WEIGHT(W,H) \
  1436. 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){ \
  1437. ff_h264_biweight_WxH_mmx2(dst, src, stride, log2_denom, weightd, weights, offset, W, H); \
  1438. } \
  1439. static void ff_h264_weight_ ## W ## x ## H ## _mmx2(uint8_t *dst, int stride, int log2_denom, int weight, int offset){ \
  1440. ff_h264_weight_WxH_mmx2(dst, stride, log2_denom, weight, offset, W, H); \
  1441. }
  1442. H264_WEIGHT(16,16)
  1443. H264_WEIGHT(16, 8)
  1444. H264_WEIGHT( 8,16)
  1445. H264_WEIGHT( 8, 8)
  1446. H264_WEIGHT( 8, 4)
  1447. H264_WEIGHT( 4, 8)
  1448. H264_WEIGHT( 4, 4)
  1449. H264_WEIGHT( 4, 2)