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.

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