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.

278 lines
8.7KB

  1. ;******************************************************************************
  2. ;* VP9 SIMD optimizations
  3. ;*
  4. ;* Copyright (c) 2013 Ronald S. Bultje <rsbultje gmail com>
  5. ;*
  6. ;* This file is part of Libav.
  7. ;*
  8. ;* Libav is free software; you can redistribute it and/or
  9. ;* modify it under the terms of the GNU Lesser General Public
  10. ;* License as published by the Free Software Foundation; either
  11. ;* version 2.1 of the License, or (at your option) any later version.
  12. ;*
  13. ;* Libav is distributed in the hope that it will be useful,
  14. ;* but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. ;* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. ;* Lesser General Public License for more details.
  17. ;*
  18. ;* You should have received a copy of the GNU Lesser General Public
  19. ;* License along with Libav; if not, write to the Free Software
  20. ;* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. ;******************************************************************************
  22. %include "libavutil/x86/x86util.asm"
  23. SECTION_RODATA
  24. ; FIXME share with vp8dsp.asm
  25. pw_256: times 8 dw 256
  26. %macro F8_TAPS 8
  27. times 8 db %1, %2
  28. times 8 db %3, %4
  29. times 8 db %5, %6
  30. times 8 db %7, %8
  31. %endmacro
  32. ; int8_t ff_filters_ssse3[3][15][4][16]
  33. const filters_ssse3 ; smooth
  34. F8_TAPS -3, -1, 32, 64, 38, 1, -3, 0
  35. F8_TAPS -2, -2, 29, 63, 41, 2, -3, 0
  36. F8_TAPS -2, -2, 26, 63, 43, 4, -4, 0
  37. F8_TAPS -2, -3, 24, 62, 46, 5, -4, 0
  38. F8_TAPS -2, -3, 21, 60, 49, 7, -4, 0
  39. F8_TAPS -1, -4, 18, 59, 51, 9, -4, 0
  40. F8_TAPS -1, -4, 16, 57, 53, 12, -4, -1
  41. F8_TAPS -1, -4, 14, 55, 55, 14, -4, -1
  42. F8_TAPS -1, -4, 12, 53, 57, 16, -4, -1
  43. F8_TAPS 0, -4, 9, 51, 59, 18, -4, -1
  44. F8_TAPS 0, -4, 7, 49, 60, 21, -3, -2
  45. F8_TAPS 0, -4, 5, 46, 62, 24, -3, -2
  46. F8_TAPS 0, -4, 4, 43, 63, 26, -2, -2
  47. F8_TAPS 0, -3, 2, 41, 63, 29, -2, -2
  48. F8_TAPS 0, -3, 1, 38, 64, 32, -1, -3
  49. ; regular
  50. F8_TAPS 0, 1, -5, 126, 8, -3, 1, 0
  51. F8_TAPS -1, 3, -10, 122, 18, -6, 2, 0
  52. F8_TAPS -1, 4, -13, 118, 27, -9, 3, -1
  53. F8_TAPS -1, 4, -16, 112, 37, -11, 4, -1
  54. F8_TAPS -1, 5, -18, 105, 48, -14, 4, -1
  55. F8_TAPS -1, 5, -19, 97, 58, -16, 5, -1
  56. F8_TAPS -1, 6, -19, 88, 68, -18, 5, -1
  57. F8_TAPS -1, 6, -19, 78, 78, -19, 6, -1
  58. F8_TAPS -1, 5, -18, 68, 88, -19, 6, -1
  59. F8_TAPS -1, 5, -16, 58, 97, -19, 5, -1
  60. F8_TAPS -1, 4, -14, 48, 105, -18, 5, -1
  61. F8_TAPS -1, 4, -11, 37, 112, -16, 4, -1
  62. F8_TAPS -1, 3, -9, 27, 118, -13, 4, -1
  63. F8_TAPS 0, 2, -6, 18, 122, -10, 3, -1
  64. F8_TAPS 0, 1, -3, 8, 126, -5, 1, 0
  65. ; sharp
  66. F8_TAPS -1, 3, -7, 127, 8, -3, 1, 0
  67. F8_TAPS -2, 5, -13, 125, 17, -6, 3, -1
  68. F8_TAPS -3, 7, -17, 121, 27, -10, 5, -2
  69. F8_TAPS -4, 9, -20, 115, 37, -13, 6, -2
  70. F8_TAPS -4, 10, -23, 108, 48, -16, 8, -3
  71. F8_TAPS -4, 10, -24, 100, 59, -19, 9, -3
  72. F8_TAPS -4, 11, -24, 90, 70, -21, 10, -4
  73. F8_TAPS -4, 11, -23, 80, 80, -23, 11, -4
  74. F8_TAPS -4, 10, -21, 70, 90, -24, 11, -4
  75. F8_TAPS -3, 9, -19, 59, 100, -24, 10, -4
  76. F8_TAPS -3, 8, -16, 48, 108, -23, 10, -4
  77. F8_TAPS -2, 6, -13, 37, 115, -20, 9, -4
  78. F8_TAPS -2, 5, -10, 27, 121, -17, 7, -3
  79. F8_TAPS -1, 3, -6, 17, 125, -13, 5, -2
  80. F8_TAPS 0, 1, -3, 8, 127, -7, 3, -1
  81. SECTION .text
  82. %macro filter_h_fn 1
  83. %assign %%px mmsize/2
  84. cglobal %1_8tap_1d_h_ %+ %%px, 6, 6, 11, dst, src, dstride, sstride, h, filtery
  85. mova m6, [pw_256]
  86. mova m7, [filteryq+ 0]
  87. %if ARCH_X86_64 && mmsize > 8
  88. mova m8, [filteryq+16]
  89. mova m9, [filteryq+32]
  90. mova m10, [filteryq+48]
  91. %endif
  92. .loop:
  93. movh m0, [srcq-3]
  94. movh m1, [srcq-2]
  95. movh m2, [srcq-1]
  96. movh m3, [srcq+0]
  97. movh m4, [srcq+1]
  98. movh m5, [srcq+2]
  99. punpcklbw m0, m1
  100. punpcklbw m2, m3
  101. movh m1, [srcq+3]
  102. movh m3, [srcq+4]
  103. add srcq, sstrideq
  104. punpcklbw m4, m5
  105. punpcklbw m1, m3
  106. pmaddubsw m0, m7
  107. %if ARCH_X86_64 && mmsize > 8
  108. pmaddubsw m2, m8
  109. pmaddubsw m4, m9
  110. pmaddubsw m1, m10
  111. %else
  112. pmaddubsw m2, [filteryq+16]
  113. pmaddubsw m4, [filteryq+32]
  114. pmaddubsw m1, [filteryq+48]
  115. %endif
  116. paddw m0, m2
  117. paddw m4, m1
  118. paddsw m0, m4
  119. pmulhrsw m0, m6
  120. %ifidn %1, avg
  121. movh m1, [dstq]
  122. %endif
  123. packuswb m0, m0
  124. %ifidn %1, avg
  125. pavgb m0, m1
  126. %endif
  127. movh [dstq], m0
  128. add dstq, dstrideq
  129. dec hd
  130. jg .loop
  131. RET
  132. %endmacro
  133. INIT_MMX ssse3
  134. filter_h_fn put
  135. filter_h_fn avg
  136. INIT_XMM ssse3
  137. filter_h_fn put
  138. filter_h_fn avg
  139. %macro filter_v_fn 1
  140. %assign %%px mmsize/2
  141. %if ARCH_X86_64
  142. cglobal %1_8tap_1d_v_ %+ %%px, 6, 8, 11, dst, src, dstride, sstride, h, filtery, src4, sstride3
  143. %else
  144. cglobal %1_8tap_1d_v_ %+ %%px, 4, 7, 11, dst, src, dstride, sstride, filtery, src4, sstride3
  145. mov filteryq, r5mp
  146. %define hd r4mp
  147. %endif
  148. sub srcq, sstrideq
  149. lea sstride3q, [sstrideq*3]
  150. sub srcq, sstrideq
  151. mova m6, [pw_256]
  152. sub srcq, sstrideq
  153. mova m7, [filteryq+ 0]
  154. lea src4q, [srcq+sstrideq*4]
  155. %if ARCH_X86_64 && mmsize > 8
  156. mova m8, [filteryq+16]
  157. mova m9, [filteryq+32]
  158. mova m10, [filteryq+48]
  159. %endif
  160. .loop:
  161. ; FIXME maybe reuse loads from previous rows, or just more generally
  162. ; unroll this to prevent multiple loads of the same data?
  163. movh m0, [srcq]
  164. movh m1, [srcq+sstrideq]
  165. movh m2, [srcq+sstrideq*2]
  166. movh m3, [srcq+sstride3q]
  167. movh m4, [src4q]
  168. movh m5, [src4q+sstrideq]
  169. punpcklbw m0, m1
  170. punpcklbw m2, m3
  171. movh m1, [src4q+sstrideq*2]
  172. movh m3, [src4q+sstride3q]
  173. add srcq, sstrideq
  174. add src4q, sstrideq
  175. punpcklbw m4, m5
  176. punpcklbw m1, m3
  177. pmaddubsw m0, m7
  178. %if ARCH_X86_64 && mmsize > 8
  179. pmaddubsw m2, m8
  180. pmaddubsw m4, m9
  181. pmaddubsw m1, m10
  182. %else
  183. pmaddubsw m2, [filteryq+16]
  184. pmaddubsw m4, [filteryq+32]
  185. pmaddubsw m1, [filteryq+48]
  186. %endif
  187. paddw m0, m2
  188. paddw m4, m1
  189. paddsw m0, m4
  190. pmulhrsw m0, m6
  191. %ifidn %1, avg
  192. movh m1, [dstq]
  193. %endif
  194. packuswb m0, m0
  195. %ifidn %1, avg
  196. pavgb m0, m1
  197. %endif
  198. movh [dstq], m0
  199. add dstq, dstrideq
  200. dec hd
  201. jg .loop
  202. RET
  203. %endmacro
  204. INIT_MMX ssse3
  205. filter_v_fn put
  206. filter_v_fn avg
  207. INIT_XMM ssse3
  208. filter_v_fn put
  209. filter_v_fn avg
  210. %macro fpel_fn 6
  211. %if %2 == 4
  212. %define %%srcfn movh
  213. %define %%dstfn movh
  214. %else
  215. %define %%srcfn movu
  216. %define %%dstfn mova
  217. %endif
  218. %if %2 <= 16
  219. cglobal %1%2, 5, 7, 4, dst, src, dstride, sstride, h, dstride3, sstride3
  220. lea sstride3q, [sstrideq*3]
  221. lea dstride3q, [dstrideq*3]
  222. %else
  223. cglobal %1%2, 5, 5, 4, dst, src, dstride, sstride, h
  224. %endif
  225. .loop:
  226. %%srcfn m0, [srcq]
  227. %%srcfn m1, [srcq+s%3]
  228. %%srcfn m2, [srcq+s%4]
  229. %%srcfn m3, [srcq+s%5]
  230. lea srcq, [srcq+sstrideq*%6]
  231. %ifidn %1, avg
  232. pavgb m0, [dstq]
  233. pavgb m1, [dstq+d%3]
  234. pavgb m2, [dstq+d%4]
  235. pavgb m3, [dstq+d%5]
  236. %endif
  237. %%dstfn [dstq], m0
  238. %%dstfn [dstq+d%3], m1
  239. %%dstfn [dstq+d%4], m2
  240. %%dstfn [dstq+d%5], m3
  241. lea dstq, [dstq+dstrideq*%6]
  242. sub hd, %6
  243. jnz .loop
  244. RET
  245. %endmacro
  246. %define d16 16
  247. %define s16 16
  248. INIT_MMX mmx
  249. fpel_fn put, 4, strideq, strideq*2, stride3q, 4
  250. fpel_fn put, 8, strideq, strideq*2, stride3q, 4
  251. INIT_MMX sse
  252. fpel_fn avg, 4, strideq, strideq*2, stride3q, 4
  253. fpel_fn avg, 8, strideq, strideq*2, stride3q, 4
  254. INIT_XMM sse
  255. fpel_fn put, 16, strideq, strideq*2, stride3q, 4
  256. fpel_fn put, 32, mmsize, strideq, strideq+mmsize, 2
  257. fpel_fn put, 64, mmsize, mmsize*2, mmsize*3, 1
  258. INIT_XMM sse2
  259. fpel_fn avg, 16, strideq, strideq*2, stride3q, 4
  260. fpel_fn avg, 32, mmsize, strideq, strideq+mmsize, 2
  261. fpel_fn avg, 64, mmsize, mmsize*2, mmsize*3, 1
  262. %undef s16
  263. %undef d16