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.

892 lines
20KB

  1. ;*****************************************************************************
  2. ;* MMX/SSE2/AVX-optimized 10-bit H.264 qpel code
  3. ;*****************************************************************************
  4. ;* Copyright (C) 2011 x264 project
  5. ;*
  6. ;* Authors: Daniel Kang <daniel.d.kang@gmail.com>
  7. ;*
  8. ;* This file is part of Libav.
  9. ;*
  10. ;* Libav is free software; you can redistribute it and/or
  11. ;* modify it under the terms of the GNU Lesser General Public
  12. ;* License as published by the Free Software Foundation; either
  13. ;* version 2.1 of the License, or (at your option) any later version.
  14. ;*
  15. ;* Libav is distributed in the hope that it will be useful,
  16. ;* but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. ;* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18. ;* Lesser General Public License for more details.
  19. ;*
  20. ;* You should have received a copy of the GNU Lesser General Public
  21. ;* License along with Libav; if not, write to the Free Software
  22. ;* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  23. ;******************************************************************************
  24. %include "x86inc.asm"
  25. %include "x86util.asm"
  26. SECTION_RODATA 32
  27. cextern pw_16
  28. cextern pw_1
  29. cextern pb_0
  30. pw_pixel_max: times 8 dw ((1 << 10)-1)
  31. pad10: times 8 dw 10*1023
  32. pad20: times 8 dw 20*1023
  33. pad30: times 8 dw 30*1023
  34. depad: times 4 dd 32*20*1023 + 512
  35. depad2: times 8 dw 20*1023 + 16*1022 + 16
  36. unpad: times 8 dw 16*1022/32 ; needs to be mod 16
  37. tap1: times 4 dw 1, -5
  38. tap2: times 4 dw 20, 20
  39. tap3: times 4 dw -5, 1
  40. pd_0f: times 4 dd 0xffff
  41. SECTION .text
  42. %macro AVG_MOV 2
  43. pavgw %2, %1
  44. mova %1, %2
  45. %endmacro
  46. %macro ADDW 3
  47. %if mmsize == 8
  48. paddw %1, %2
  49. %else
  50. movu %3, %2
  51. paddw %1, %3
  52. %endif
  53. %endmacro
  54. %macro FILT_H 4
  55. paddw %1, %4
  56. psubw %1, %2 ; a-b
  57. psraw %1, 2 ; (a-b)/4
  58. psubw %1, %2 ; (a-b)/4-b
  59. paddw %1, %3 ; (a-b)/4-b+c
  60. psraw %1, 2 ; ((a-b)/4-b+c)/4
  61. paddw %1, %3 ; ((a-b)/4-b+c)/4+c = (a-5*b+20*c)/16
  62. %endmacro
  63. %macro PRELOAD_V 0
  64. lea r3, [r2*3]
  65. sub r1, r3
  66. movu m0, [r1+r2]
  67. movu m1, [r1+r2*2]
  68. add r1, r3
  69. movu m2, [r1]
  70. movu m3, [r1+r2]
  71. movu m4, [r1+r2*2]
  72. add r1, r3
  73. %endmacro
  74. %macro FILT_V 8
  75. movu %6, [r1]
  76. paddw %1, %6
  77. mova %7, %2
  78. paddw %7, %5
  79. mova %8, %3
  80. paddw %8, %4
  81. FILT_H %1, %7, %8, [pw_16]
  82. psraw %1, 1
  83. CLIPW %1, [pb_0], [pw_pixel_max]
  84. %endmacro
  85. %macro MC 1
  86. %define OP_MOV mova
  87. INIT_MMX
  88. %1 mmxext, put, 4
  89. INIT_XMM
  90. %1 sse2 , put, 8
  91. %define OP_MOV AVG_MOV
  92. INIT_MMX
  93. %1 mmxext, avg, 4
  94. INIT_XMM
  95. %1 sse2 , avg, 8
  96. %endmacro
  97. %macro MCAxA 8
  98. %ifdef ARCH_X86_64
  99. %ifnidn %1,mmxext
  100. MCAxA_OP %1,%2,%3,%4,%5,%6,%7,%8
  101. %endif
  102. %else
  103. MCAxA_OP %1,%2,%3,%4,%5,%6,%7,%8
  104. %endif
  105. %endmacro
  106. %macro MCAxA_OP 8
  107. cglobal %2_h264_qpel%5_%3_10_%1, %6,%7,%8
  108. %ifdef ARCH_X86_32
  109. call stub_%2_h264_qpel%4_%3_10_%1
  110. mov r0, r0m
  111. mov r1, r1m
  112. add r0, %4*2
  113. add r1, %4*2
  114. call stub_%2_h264_qpel%4_%3_10_%1
  115. mov r0, r0m
  116. mov r1, r1m
  117. lea r0, [r0+r2*%4]
  118. lea r1, [r1+r2*%4]
  119. call stub_%2_h264_qpel%4_%3_10_%1
  120. mov r0, r0m
  121. mov r1, r1m
  122. lea r0, [r0+r2*%4+%4*2]
  123. lea r1, [r1+r2*%4+%4*2]
  124. call stub_%2_h264_qpel%4_%3_10_%1
  125. RET
  126. %else ; ARCH_X86_64
  127. mov r10, r0
  128. mov r11, r1
  129. call stub_%2_h264_qpel%4_%3_10_%1
  130. lea r0, [r10+%4*2]
  131. lea r1, [r11+%4*2]
  132. call stub_%2_h264_qpel%4_%3_10_%1
  133. lea r0, [r10+r2*%4]
  134. lea r1, [r11+r2*%4]
  135. call stub_%2_h264_qpel%4_%3_10_%1
  136. lea r0, [r10+r2*%4+%4*2]
  137. lea r1, [r11+r2*%4+%4*2]
  138. %ifndef UNIX64 ; fall through to function
  139. call stub_%2_h264_qpel%4_%3_10_%1
  140. RET
  141. %endif
  142. %endif
  143. %endmacro
  144. ;cpu, put/avg, mc, 4/8, ...
  145. %macro cglobal_mc 7
  146. %assign i %4*2
  147. MCAxA %1, %2, %3, %4, i, %5,%6,%7
  148. cglobal %2_h264_qpel%4_%3_10_%1, %5,%6,%7
  149. %ifndef UNIX64 ; no prologue or epilogue for UNIX64
  150. call stub_%2_h264_qpel%4_%3_10_%1
  151. RET
  152. %endif
  153. stub_%2_h264_qpel%4_%3_10_%1:
  154. %endmacro
  155. ;-----------------------------------------------------------------------------
  156. ; void h264_qpel_mc00(uint8_t *dst, uint8_t *src, int stride)
  157. ;-----------------------------------------------------------------------------
  158. %macro COPY4 0
  159. movu m0, [r1 ]
  160. OP_MOV [r0 ], m0
  161. movu m0, [r1+r2 ]
  162. OP_MOV [r0+r2 ], m0
  163. movu m0, [r1+r2*2]
  164. OP_MOV [r0+r2*2], m0
  165. movu m0, [r1+r3 ]
  166. OP_MOV [r0+r3 ], m0
  167. %endmacro
  168. %macro MC00 1
  169. INIT_MMX
  170. cglobal_mc mmxext, %1, mc00, 4, 3,4,0
  171. lea r3, [r2*3]
  172. COPY4
  173. ret
  174. INIT_XMM
  175. cglobal %1_h264_qpel8_mc00_10_sse2, 3,4
  176. lea r3, [r2*3]
  177. COPY4
  178. lea r0, [r0+r2*4]
  179. lea r1, [r1+r2*4]
  180. COPY4
  181. RET
  182. cglobal %1_h264_qpel16_mc00_10_sse2, 3,4
  183. mov r3d, 8
  184. .loop:
  185. movu m0, [r1 ]
  186. movu m1, [r1 +16]
  187. OP_MOV [r0 ], m0
  188. OP_MOV [r0 +16], m1
  189. movu m0, [r1+r2 ]
  190. movu m1, [r1+r2+16]
  191. OP_MOV [r0+r2 ], m0
  192. OP_MOV [r0+r2+16], m1
  193. lea r0, [r0+r2*2]
  194. lea r1, [r1+r2*2]
  195. dec r3d
  196. jg .loop
  197. REP_RET
  198. %endmacro
  199. %define OP_MOV mova
  200. MC00 put
  201. %define OP_MOV AVG_MOV
  202. MC00 avg
  203. ;-----------------------------------------------------------------------------
  204. ; void h264_qpel_mc20(uint8_t *dst, uint8_t *src, int stride)
  205. ;-----------------------------------------------------------------------------
  206. %macro MC_CACHE 1
  207. %define OP_MOV mova
  208. %define PALIGNR PALIGNR_MMX
  209. INIT_MMX
  210. %1 mmxext , put, 4
  211. INIT_XMM
  212. %1 sse2_cache64 , put, 8
  213. %define PALIGNR PALIGNR_SSSE3
  214. %1 ssse3_cache64, put, 8
  215. %1 sse2 , put, 8, 0
  216. %define OP_MOV AVG_MOV
  217. %define PALIGNR PALIGNR_MMX
  218. INIT_MMX
  219. %1 mmxext , avg, 4
  220. INIT_XMM
  221. %1 sse2_cache64 , avg, 8
  222. %define PALIGNR PALIGNR_SSSE3
  223. %1 ssse3_cache64, avg, 8
  224. %1 sse2 , avg, 8, 0
  225. %endmacro
  226. %macro MC20 3-4
  227. cglobal_mc %1, %2, mc20, %3, 3,4,9
  228. mov r3d, %3
  229. mova m1, [pw_pixel_max]
  230. %if num_mmregs > 8
  231. mova m8, [pw_16]
  232. %define p16 m8
  233. %else
  234. %define p16 [pw_16]
  235. %endif
  236. .nextrow
  237. %if %0 == 4
  238. movu m2, [r1-4]
  239. movu m3, [r1-2]
  240. movu m4, [r1+0]
  241. ADDW m2, [r1+6], m5
  242. ADDW m3, [r1+4], m5
  243. ADDW m4, [r1+2], m5
  244. %else ; movu is slow on these processors
  245. %if mmsize==16
  246. movu m2, [r1-4]
  247. movu m0, [r1+6]
  248. mova m6, m0
  249. psrldq m0, 6
  250. paddw m6, m2
  251. PALIGNR m3, m0, m2, 2, m5
  252. PALIGNR m7, m0, m2, 8, m5
  253. paddw m3, m7
  254. PALIGNR m4, m0, m2, 4, m5
  255. PALIGNR m7, m0, m2, 6, m5
  256. paddw m4, m7
  257. SWAP 2, 6
  258. %else
  259. movu m2, [r1-4]
  260. movu m6, [r1+4]
  261. PALIGNR m3, m6, m2, 2, m5
  262. paddw m3, m6
  263. PALIGNR m4, m6, m2, 4, m5
  264. PALIGNR m7, m6, m2, 6, m5
  265. paddw m4, m7
  266. paddw m2, [r1+6]
  267. %endif
  268. %endif
  269. FILT_H m2, m3, m4, p16
  270. psraw m2, 1
  271. pxor m0, m0
  272. CLIPW m2, m0, m1
  273. OP_MOV [r0], m2
  274. add r0, r2
  275. add r1, r2
  276. dec r3d
  277. jg .nextrow
  278. rep ret
  279. %endmacro
  280. MC_CACHE MC20
  281. ;-----------------------------------------------------------------------------
  282. ; void h264_qpel_mc30(uint8_t *dst, uint8_t *src, int stride)
  283. ;-----------------------------------------------------------------------------
  284. %macro MC30 3-4
  285. cglobal_mc %1, %2, mc30, %3, 3,5,9
  286. lea r4, [r1+2]
  287. jmp stub_%2_h264_qpel%3_mc10_10_%1.body
  288. %endmacro
  289. MC_CACHE MC30
  290. ;-----------------------------------------------------------------------------
  291. ; void h264_qpel_mc10(uint8_t *dst, uint8_t *src, int stride)
  292. ;-----------------------------------------------------------------------------
  293. %macro MC10 3-4
  294. cglobal_mc %1, %2, mc10, %3, 3,5,9
  295. mov r4, r1
  296. .body
  297. mov r3d, %3
  298. mova m1, [pw_pixel_max]
  299. %if num_mmregs > 8
  300. mova m8, [pw_16]
  301. %define p16 m8
  302. %else
  303. %define p16 [pw_16]
  304. %endif
  305. .nextrow
  306. %if %0 == 4
  307. movu m2, [r1-4]
  308. movu m3, [r1-2]
  309. movu m4, [r1+0]
  310. ADDW m2, [r1+6], m5
  311. ADDW m3, [r1+4], m5
  312. ADDW m4, [r1+2], m5
  313. %else ; movu is slow on these processors
  314. %if mmsize==16
  315. movu m2, [r1-4]
  316. movu m0, [r1+6]
  317. mova m6, m0
  318. psrldq m0, 6
  319. paddw m6, m2
  320. PALIGNR m3, m0, m2, 2, m5
  321. PALIGNR m7, m0, m2, 8, m5
  322. paddw m3, m7
  323. PALIGNR m4, m0, m2, 4, m5
  324. PALIGNR m7, m0, m2, 6, m5
  325. paddw m4, m7
  326. SWAP 2, 6
  327. %else
  328. movu m2, [r1-4]
  329. movu m6, [r1+4]
  330. PALIGNR m3, m6, m2, 2, m5
  331. paddw m3, m6
  332. PALIGNR m4, m6, m2, 4, m5
  333. PALIGNR m7, m6, m2, 6, m5
  334. paddw m4, m7
  335. paddw m2, [r1+6]
  336. %endif
  337. %endif
  338. FILT_H m2, m3, m4, p16
  339. psraw m2, 1
  340. pxor m0, m0
  341. CLIPW m2, m0, m1
  342. movu m3, [r4]
  343. pavgw m2, m3
  344. OP_MOV [r0], m2
  345. add r0, r2
  346. add r1, r2
  347. add r4, r2
  348. dec r3d
  349. jg .nextrow
  350. rep ret
  351. %endmacro
  352. MC_CACHE MC10
  353. ;-----------------------------------------------------------------------------
  354. ; void h264_qpel_mc02(uint8_t *dst, uint8_t *src, int stride)
  355. ;-----------------------------------------------------------------------------
  356. %macro V_FILT 11
  357. v_filt%9_%10_10_%11:
  358. add r4, r2
  359. .no_addr4:
  360. FILT_V m0, m1, m2, m3, m4, m5, m6, m7
  361. add r1, r2
  362. add r0, r2
  363. ret
  364. %endmacro
  365. INIT_MMX
  366. RESET_MM_PERMUTATION
  367. %assign i 0
  368. %rep 4
  369. V_FILT m0, m1, m2, m3, m4, m5, m6, m7, 4, i, mmxext
  370. SWAP 0,1,2,3,4,5
  371. %assign i i+1
  372. %endrep
  373. INIT_XMM
  374. RESET_MM_PERMUTATION
  375. %assign i 0
  376. %rep 6
  377. V_FILT m0, m1, m2, m3, m4, m5, m6, m7, 8, i, sse2
  378. SWAP 0,1,2,3,4,5
  379. %assign i i+1
  380. %endrep
  381. %macro MC02 3
  382. cglobal_mc %1, %2, mc02, %3, 3,4,8
  383. PRELOAD_V
  384. sub r0, r2
  385. %assign j 0
  386. %rep %3
  387. %assign i (j % 6)
  388. call v_filt%3_ %+ i %+ _10_%1.no_addr4
  389. OP_MOV [r0], m0
  390. SWAP 0,1,2,3,4,5
  391. %assign j j+1
  392. %endrep
  393. ret
  394. %endmacro
  395. MC MC02
  396. ;-----------------------------------------------------------------------------
  397. ; void h264_qpel_mc01(uint8_t *dst, uint8_t *src, int stride)
  398. ;-----------------------------------------------------------------------------
  399. %macro MC01 3
  400. cglobal_mc %1, %2, mc01, %3, 3,5,8
  401. mov r4, r1
  402. .body
  403. PRELOAD_V
  404. sub r4, r2
  405. sub r0, r2
  406. %assign j 0
  407. %rep %3
  408. %assign i (j % 6)
  409. call v_filt%3_ %+ i %+ _10_%1
  410. movu m7, [r4]
  411. pavgw m0, m7
  412. OP_MOV [r0], m0
  413. SWAP 0,1,2,3,4,5
  414. %assign j j+1
  415. %endrep
  416. ret
  417. %endmacro
  418. MC MC01
  419. ;-----------------------------------------------------------------------------
  420. ; void h264_qpel_mc03(uint8_t *dst, uint8_t *src, int stride)
  421. ;-----------------------------------------------------------------------------
  422. %macro MC03 3
  423. cglobal_mc %1, %2, mc03, %3, 3,5,8
  424. lea r4, [r1+r2]
  425. jmp stub_%2_h264_qpel%3_mc01_10_%1.body
  426. %endmacro
  427. MC MC03
  428. ;-----------------------------------------------------------------------------
  429. ; void h264_qpel_mc11(uint8_t *dst, uint8_t *src, int stride)
  430. ;-----------------------------------------------------------------------------
  431. %macro H_FILT_AVG 3-4
  432. h_filt%2_%3_10_%1:
  433. ;FILT_H with fewer registers and averaged with the FILT_V result
  434. ;m6,m7 are tmp registers, m0 is the FILT_V result, the rest are to be used next in the next iteration
  435. ;unfortunately I need three registers, so m5 will have to be re-read from memory
  436. movu m5, [r4-4]
  437. ADDW m5, [r4+6], m7
  438. movu m6, [r4-2]
  439. ADDW m6, [r4+4], m7
  440. paddw m5, [pw_16]
  441. psubw m5, m6 ; a-b
  442. psraw m5, 2 ; (a-b)/4
  443. psubw m5, m6 ; (a-b)/4-b
  444. movu m6, [r4+0]
  445. ADDW m6, [r4+2], m7
  446. paddw m5, m6 ; (a-b)/4-b+c
  447. psraw m5, 2 ; ((a-b)/4-b+c)/4
  448. paddw m5, m6 ; ((a-b)/4-b+c)/4+c = (a-5*b+20*c)/16
  449. psraw m5, 1
  450. CLIPW m5, [pb_0], [pw_pixel_max]
  451. ;avg FILT_V, FILT_H
  452. pavgw m0, m5
  453. %if %0!=4
  454. movu m5, [r1+r5]
  455. %endif
  456. ret
  457. %endmacro
  458. INIT_MMX
  459. RESET_MM_PERMUTATION
  460. %assign i 0
  461. %rep 3
  462. H_FILT_AVG mmxext, 4, i
  463. SWAP 0,1,2,3,4,5
  464. %assign i i+1
  465. %endrep
  466. H_FILT_AVG mmxext, 4, i, 0
  467. INIT_XMM
  468. RESET_MM_PERMUTATION
  469. %assign i 0
  470. %rep 6
  471. %if i==1
  472. H_FILT_AVG sse2, 8, i, 0
  473. %else
  474. H_FILT_AVG sse2, 8, i
  475. %endif
  476. SWAP 0,1,2,3,4,5
  477. %assign i i+1
  478. %endrep
  479. %macro MC11 3
  480. ; this REALLY needs x86_64
  481. cglobal_mc %1, %2, mc11, %3, 3,6,8
  482. mov r4, r1
  483. .body
  484. PRELOAD_V
  485. sub r0, r2
  486. sub r4, r2
  487. mov r5, r2
  488. neg r5
  489. %assign j 0
  490. %rep %3
  491. %assign i (j % 6)
  492. call v_filt%3_ %+ i %+ _10_%1
  493. call h_filt%3_ %+ i %+ _10_%1
  494. %if %3==8 && i==1
  495. movu m5, [r1+r5]
  496. %endif
  497. OP_MOV [r0], m0
  498. SWAP 0,1,2,3,4,5
  499. %assign j j+1
  500. %endrep
  501. ret
  502. %endmacro
  503. MC MC11
  504. ;-----------------------------------------------------------------------------
  505. ; void h264_qpel_mc31(uint8_t *dst, uint8_t *src, int stride)
  506. ;-----------------------------------------------------------------------------
  507. %macro MC31 3
  508. cglobal_mc %1, %2, mc31, %3, 3,6,8
  509. mov r4, r1
  510. add r1, 2
  511. jmp stub_%2_h264_qpel%3_mc11_10_%1.body
  512. %endmacro
  513. MC MC31
  514. ;-----------------------------------------------------------------------------
  515. ; void h264_qpel_mc13(uint8_t *dst, uint8_t *src, int stride)
  516. ;-----------------------------------------------------------------------------
  517. %macro MC13 3
  518. cglobal_mc %1, %2, mc13, %3, 3,7,12
  519. lea r4, [r1+r2]
  520. jmp stub_%2_h264_qpel%3_mc11_10_%1.body
  521. %endmacro
  522. MC MC13
  523. ;-----------------------------------------------------------------------------
  524. ; void h264_qpel_mc33(uint8_t *dst, uint8_t *src, int stride)
  525. ;-----------------------------------------------------------------------------
  526. %macro MC33 3
  527. cglobal_mc %1, %2, mc33, %3, 3,6,8
  528. lea r4, [r1+r2]
  529. add r1, 2
  530. jmp stub_%2_h264_qpel%3_mc11_10_%1.body
  531. %endmacro
  532. MC MC33
  533. ;-----------------------------------------------------------------------------
  534. ; void h264_qpel_mc22(uint8_t *dst, uint8_t *src, int stride)
  535. ;-----------------------------------------------------------------------------
  536. %macro FILT_H2 3
  537. psubw %1, %2 ; a-b
  538. psubw %2, %3 ; b-c
  539. psllw %2, 2
  540. psubw %1, %2 ; a-5*b+4*c
  541. psllw %3, 4
  542. paddw %1, %3 ; a-5*b+20*c
  543. %endmacro
  544. %macro FILT_VNRD 8
  545. movu %6, [r1]
  546. paddw %1, %6
  547. mova %7, %2
  548. paddw %7, %5
  549. mova %8, %3
  550. paddw %8, %4
  551. FILT_H2 %1, %7, %8
  552. %endmacro
  553. %macro HV 2
  554. %ifidn %1,sse2
  555. %define PAD 12
  556. %define COUNT 2
  557. %else
  558. %define PAD 0
  559. %define COUNT 3
  560. %endif
  561. put_hv%2_10_%1:
  562. neg r2 ; This actually saves instructions
  563. lea r1, [r1+r2*2-mmsize+PAD]
  564. lea r4, [rsp+PAD+gprsize]
  565. mov r3d, COUNT
  566. .v_loop:
  567. movu m0, [r1]
  568. sub r1, r2
  569. movu m1, [r1]
  570. sub r1, r2
  571. movu m2, [r1]
  572. sub r1, r2
  573. movu m3, [r1]
  574. sub r1, r2
  575. movu m4, [r1]
  576. sub r1, r2
  577. %assign i 0
  578. %rep %2-1
  579. FILT_VNRD m0, m1, m2, m3, m4, m5, m6, m7
  580. psubw m0, [pad20]
  581. movu [r4+i*mmsize*3], m0
  582. sub r1, r2
  583. SWAP 0,1,2,3,4,5
  584. %assign i i+1
  585. %endrep
  586. FILT_VNRD m0, m1, m2, m3, m4, m5, m6, m7
  587. psubw m0, [pad20]
  588. movu [r4+i*mmsize*3], m0
  589. add r4, mmsize
  590. lea r1, [r1+r2*8+mmsize]
  591. %if %2==8
  592. lea r1, [r1+r2*4]
  593. %endif
  594. dec r3d
  595. jg .v_loop
  596. neg r2
  597. ret
  598. %endmacro
  599. INIT_MMX
  600. HV mmxext, 4
  601. INIT_XMM
  602. HV sse2 , 8
  603. %macro H_LOOP 2
  604. %if num_mmregs > 8
  605. %define s1 m8
  606. %define s2 m9
  607. %define s3 m10
  608. %define d1 m11
  609. %else
  610. %define s1 [tap1]
  611. %define s2 [tap2]
  612. %define s3 [tap3]
  613. %define d1 [depad]
  614. %endif
  615. h%2_loop_op_%1:
  616. movu m1, [r1+mmsize-4]
  617. movu m2, [r1+mmsize-2]
  618. mova m3, [r1+mmsize+0]
  619. movu m4, [r1+mmsize+2]
  620. movu m5, [r1+mmsize+4]
  621. movu m6, [r1+mmsize+6]
  622. %if num_mmregs > 8
  623. pmaddwd m1, s1
  624. pmaddwd m2, s1
  625. pmaddwd m3, s2
  626. pmaddwd m4, s2
  627. pmaddwd m5, s3
  628. pmaddwd m6, s3
  629. paddd m1, d1
  630. paddd m2, d1
  631. %else
  632. mova m0, s1
  633. pmaddwd m1, m0
  634. pmaddwd m2, m0
  635. mova m0, s2
  636. pmaddwd m3, m0
  637. pmaddwd m4, m0
  638. mova m0, s3
  639. pmaddwd m5, m0
  640. pmaddwd m6, m0
  641. mova m0, d1
  642. paddd m1, m0
  643. paddd m2, m0
  644. %endif
  645. paddd m3, m5
  646. paddd m4, m6
  647. paddd m1, m3
  648. paddd m2, m4
  649. psrad m1, 10
  650. psrad m2, 10
  651. pslld m2, 16
  652. pand m1, [pd_0f]
  653. por m1, m2
  654. %if num_mmregs <= 8
  655. pxor m0, m0
  656. %endif
  657. CLIPW m1, m0, m7
  658. add r1, mmsize*3
  659. ret
  660. %endmacro
  661. INIT_MMX
  662. H_LOOP mmxext, 4
  663. INIT_XMM
  664. H_LOOP sse2 , 8
  665. %macro MC22 3
  666. cglobal_mc %1, %2, mc22, %3, 3,7,12
  667. %define PAD mmsize*8*4*2 ; SIZE*16*4*sizeof(pixel)
  668. mov r6, rsp ; backup stack pointer
  669. and rsp, ~(mmsize-1) ; align stack
  670. sub rsp, PAD
  671. call put_hv%3_10_%1
  672. mov r3d, %3
  673. mova m7, [pw_pixel_max]
  674. %if num_mmregs > 8
  675. pxor m0, m0
  676. mova m8, [tap1]
  677. mova m9, [tap2]
  678. mova m10, [tap3]
  679. mova m11, [depad]
  680. %endif
  681. mov r1, rsp
  682. .h_loop:
  683. call h%3_loop_op_%1
  684. OP_MOV [r0], m1
  685. add r0, r2
  686. dec r3d
  687. jg .h_loop
  688. mov rsp, r6 ; restore stack pointer
  689. ret
  690. %endmacro
  691. MC MC22
  692. ;-----------------------------------------------------------------------------
  693. ; void h264_qpel_mc12(uint8_t *dst, uint8_t *src, int stride)
  694. ;-----------------------------------------------------------------------------
  695. %macro MC12 3
  696. cglobal_mc %1, %2, mc12, %3, 3,7,12
  697. %define PAD mmsize*8*4*2 ; SIZE*16*4*sizeof(pixel)
  698. mov r6, rsp ; backup stack pointer
  699. and rsp, ~(mmsize-1) ; align stack
  700. sub rsp, PAD
  701. call put_hv%3_10_%1
  702. xor r4d, r4d
  703. .body
  704. mov r3d, %3
  705. pxor m0, m0
  706. mova m7, [pw_pixel_max]
  707. %if num_mmregs > 8
  708. mova m8, [tap1]
  709. mova m9, [tap2]
  710. mova m10, [tap3]
  711. mova m11, [depad]
  712. %endif
  713. mov r1, rsp
  714. .h_loop:
  715. call h%3_loop_op_%1
  716. movu m3, [r1+r4-2*mmsize] ; movu needed for mc32, etc
  717. paddw m3, [depad2]
  718. psrlw m3, 5
  719. psubw m3, [unpad]
  720. CLIPW m3, m0, m7
  721. pavgw m1, m3
  722. OP_MOV [r0], m1
  723. add r0, r2
  724. dec r3d
  725. jg .h_loop
  726. mov rsp, r6 ; restore stack pointer
  727. ret
  728. %endmacro
  729. MC MC12
  730. ;-----------------------------------------------------------------------------
  731. ; void h264_qpel_mc32(uint8_t *dst, uint8_t *src, int stride)
  732. ;-----------------------------------------------------------------------------
  733. %macro MC32 3
  734. cglobal_mc %1, %2, mc32, %3, 3,7,12
  735. %define PAD mmsize*8*3*2 ; SIZE*16*4*sizeof(pixel)
  736. mov r6, rsp ; backup stack pointer
  737. and rsp, ~(mmsize-1) ; align stack
  738. sub rsp, PAD
  739. call put_hv%3_10_%1
  740. mov r4d, 2 ; sizeof(pixel)
  741. jmp stub_%2_h264_qpel%3_mc12_10_%1.body
  742. %endmacro
  743. MC MC32
  744. ;-----------------------------------------------------------------------------
  745. ; void h264_qpel_mc21(uint8_t *dst, uint8_t *src, int stride)
  746. ;-----------------------------------------------------------------------------
  747. %macro H_NRD 2
  748. put_h%2_10_%1:
  749. add rsp, gprsize
  750. mov r3d, %2
  751. xor r4d, r4d
  752. mova m6, [pad20]
  753. .nextrow
  754. movu m2, [r5-4]
  755. movu m3, [r5-2]
  756. movu m4, [r5+0]
  757. ADDW m2, [r5+6], m5
  758. ADDW m3, [r5+4], m5
  759. ADDW m4, [r5+2], m5
  760. FILT_H2 m2, m3, m4
  761. psubw m2, m6
  762. mova [rsp+r4], m2
  763. add r4d, mmsize*3
  764. add r5, r2
  765. dec r3d
  766. jg .nextrow
  767. sub rsp, gprsize
  768. ret
  769. %endmacro
  770. INIT_MMX
  771. H_NRD mmxext, 4
  772. INIT_XMM
  773. H_NRD sse2 , 8
  774. %macro MC21 3
  775. cglobal_mc %1, %2, mc21, %3, 3,7,12
  776. mov r5, r1
  777. .body
  778. %define PAD mmsize*8*3*2 ; SIZE*16*4*sizeof(pixel)
  779. mov r6, rsp ; backup stack pointer
  780. and rsp, ~(mmsize-1) ; align stack
  781. sub rsp, PAD
  782. call put_h%3_10_%1
  783. sub rsp, PAD
  784. call put_hv%3_10_%1
  785. mov r4d, PAD-mmsize ; H buffer
  786. jmp stub_%2_h264_qpel%3_mc12_10_%1.body
  787. %endmacro
  788. MC MC21
  789. ;-----------------------------------------------------------------------------
  790. ; void h264_qpel_mc23(uint8_t *dst, uint8_t *src, int stride)
  791. ;-----------------------------------------------------------------------------
  792. %macro MC23 3
  793. cglobal_mc %1, %2, mc23, %3, 3,7,12
  794. lea r5, [r1+r2]
  795. jmp stub_%2_h264_qpel%3_mc21_10_%1.body
  796. %endmacro
  797. MC MC23