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.

302 lines
12KB

  1. /*
  2. * ARM NEON optimised MDCT
  3. * Copyright (c) 2009 Mans Rullgard <mans@mansr.com>
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * Libav is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #include "libavutil/arm/asm.S"
  22. #define ff_fft_calc_neon X(ff_fft_calc_neon)
  23. function ff_imdct_half_neon, export=1
  24. push {r4-r8,lr}
  25. mov r12, #1
  26. ldr lr, [r0, #20] @ mdct_bits
  27. ldr r4, [r0, #24] @ tcos
  28. ldr r3, [r0, #8] @ revtab
  29. lsl r12, r12, lr @ n = 1 << nbits
  30. lsr lr, r12, #2 @ n4 = n >> 2
  31. add r7, r2, r12, lsl #1
  32. mov r12, #-16
  33. sub r7, r7, #16
  34. vld2.32 {d16-d17},[r7,:128],r12 @ d16=x,n1 d17=x,n0
  35. vld2.32 {d0-d1}, [r2,:128]! @ d0 =m0,x d1 =m1,x
  36. vrev64.32 d17, d17
  37. vld2.32 {d2,d3}, [r4,:128]! @ d2=c0,c1 d3=s0,s2
  38. vmul.f32 d6, d17, d2
  39. vmul.f32 d7, d0, d2
  40. 1:
  41. subs lr, lr, #2
  42. ldr r6, [r3], #4
  43. vmul.f32 d4, d0, d3
  44. vmul.f32 d5, d17, d3
  45. vsub.f32 d4, d6, d4
  46. vadd.f32 d5, d5, d7
  47. uxth r8, r6, ror #16
  48. uxth r6, r6
  49. add r8, r1, r8, lsl #3
  50. add r6, r1, r6, lsl #3
  51. beq 1f
  52. vld2.32 {d16-d17},[r7,:128],r12
  53. vld2.32 {d0-d1}, [r2,:128]!
  54. vrev64.32 d17, d17
  55. vld2.32 {d2,d3}, [r4,:128]! @ d2=c0,c1 d3=s0,s2
  56. vmul.f32 d6, d17, d2
  57. vmul.f32 d7, d0, d2
  58. vst2.32 {d4[0],d5[0]}, [r6,:64]
  59. vst2.32 {d4[1],d5[1]}, [r8,:64]
  60. b 1b
  61. 1:
  62. vst2.32 {d4[0],d5[0]}, [r6,:64]
  63. vst2.32 {d4[1],d5[1]}, [r8,:64]
  64. mov r4, r0
  65. mov r6, r1
  66. bl ff_fft_calc_neon
  67. mov r12, #1
  68. ldr lr, [r4, #20] @ mdct_bits
  69. ldr r4, [r4, #24] @ tcos
  70. lsl r12, r12, lr @ n = 1 << nbits
  71. lsr lr, r12, #3 @ n8 = n >> 3
  72. add r4, r4, lr, lsl #3
  73. add r6, r6, lr, lsl #3
  74. sub r1, r4, #16
  75. sub r3, r6, #16
  76. mov r7, #-16
  77. mov r8, r6
  78. mov r0, r3
  79. vld2.32 {d0-d1}, [r3,:128], r7 @ d0 =i1,r1 d1 =i0,r0
  80. vld2.32 {d20-d21},[r6,:128]! @ d20=i2,r2 d21=i3,r3
  81. vld2.32 {d16,d18},[r1,:128], r7 @ d16=c1,c0 d18=s1,s0
  82. 1:
  83. subs lr, lr, #2
  84. vmul.f32 d7, d0, d18
  85. vld2.32 {d17,d19},[r4,:128]! @ d17=c2,c3 d19=s2,s3
  86. vmul.f32 d4, d1, d18
  87. vmul.f32 d5, d21, d19
  88. vmul.f32 d6, d20, d19
  89. vmul.f32 d22, d1, d16
  90. vmul.f32 d23, d21, d17
  91. vmul.f32 d24, d0, d16
  92. vmul.f32 d25, d20, d17
  93. vadd.f32 d7, d7, d22
  94. vadd.f32 d6, d6, d23
  95. vsub.f32 d4, d4, d24
  96. vsub.f32 d5, d5, d25
  97. beq 1f
  98. vld2.32 {d0-d1}, [r3,:128], r7
  99. vld2.32 {d20-d21},[r6,:128]!
  100. vld2.32 {d16,d18},[r1,:128], r7 @ d16=c1,c0 d18=s1,s0
  101. vrev64.32 q3, q3
  102. vst2.32 {d4,d6}, [r0,:128], r7
  103. vst2.32 {d5,d7}, [r8,:128]!
  104. b 1b
  105. 1:
  106. vrev64.32 q3, q3
  107. vst2.32 {d4,d6}, [r0,:128]
  108. vst2.32 {d5,d7}, [r8,:128]
  109. pop {r4-r8,pc}
  110. endfunc
  111. function ff_imdct_calc_neon, export=1
  112. push {r4-r6,lr}
  113. ldr r3, [r0, #20]
  114. mov r4, #1
  115. mov r5, r1
  116. lsl r4, r4, r3
  117. add r1, r1, r4
  118. bl X(ff_imdct_half_neon)
  119. add r0, r5, r4, lsl #2
  120. add r1, r5, r4, lsl #1
  121. sub r0, r0, #8
  122. sub r2, r1, #16
  123. mov r3, #-16
  124. mov r6, #-8
  125. vmov.i32 d30, #1<<31
  126. 1:
  127. vld1.32 {d0-d1}, [r2,:128], r3
  128. pld [r0, #-16]
  129. vrev64.32 q0, q0
  130. vld1.32 {d2-d3}, [r1,:128]!
  131. veor d4, d1, d30
  132. pld [r2, #-16]
  133. vrev64.32 q1, q1
  134. veor d5, d0, d30
  135. vst1.32 {d2}, [r0,:64], r6
  136. vst1.32 {d3}, [r0,:64], r6
  137. vst1.32 {d4-d5}, [r5,:128]!
  138. subs r4, r4, #16
  139. bgt 1b
  140. pop {r4-r6,pc}
  141. endfunc
  142. function ff_mdct_calc_neon, export=1
  143. push {r4-r10,lr}
  144. mov r12, #1
  145. ldr lr, [r0, #20] @ mdct_bits
  146. ldr r4, [r0, #24] @ tcos
  147. ldr r3, [r0, #8] @ revtab
  148. lsl lr, r12, lr @ n = 1 << nbits
  149. add r7, r2, lr @ in4u
  150. sub r9, r7, #16 @ in4d
  151. add r2, r7, lr, lsl #1 @ in3u
  152. add r8, r9, lr, lsl #1 @ in3d
  153. add r5, r4, lr, lsl #1
  154. sub r5, r5, #16
  155. sub r3, r3, #4
  156. mov r12, #-16
  157. vld2.32 {d16,d18},[r9,:128],r12 @ in0u0,in0u1 in4d1,in4d0
  158. vld2.32 {d17,d19},[r8,:128],r12 @ in2u0,in2u1 in3d1,in3d0
  159. vld2.32 {d0, d2}, [r7,:128]! @ in4u0,in4u1 in2d1,in2d0
  160. vrev64.32 q9, q9 @ in4d0,in4d1 in3d0,in3d1
  161. vld2.32 {d1, d3}, [r2,:128]! @ in3u0,in3u1 in1d1,in1d0
  162. vsub.f32 d0, d18, d0 @ in4d-in4u I
  163. vld2.32 {d20,d21},[r4,:128]! @ c0,c1 s0,s1
  164. vrev64.32 q1, q1 @ in2d0,in2d1 in1d0,in1d1
  165. vld2.32 {d30,d31},[r5,:128],r12 @ c2,c3 s2,s3
  166. vadd.f32 d1, d1, d19 @ in3u+in3d -R
  167. vsub.f32 d16, d16, d2 @ in0u-in2d R
  168. vadd.f32 d17, d17, d3 @ in2u+in1d -I
  169. 1:
  170. vmul.f32 d7, d0, d21 @ I*s
  171. A ldr r10, [r3, lr, lsr #1]
  172. T lsr r10, lr, #1
  173. T ldr r10, [r3, r10]
  174. vmul.f32 d6, d1, d20 @ -R*c
  175. ldr r6, [r3, #4]!
  176. vmul.f32 d4, d1, d21 @ -R*s
  177. vmul.f32 d5, d0, d20 @ I*c
  178. vmul.f32 d24, d16, d30 @ R*c
  179. vmul.f32 d25, d17, d31 @ -I*s
  180. vmul.f32 d22, d16, d31 @ R*s
  181. vmul.f32 d23, d17, d30 @ I*c
  182. subs lr, lr, #16
  183. vsub.f32 d6, d6, d7 @ -R*c-I*s
  184. vadd.f32 d7, d4, d5 @ -R*s+I*c
  185. vsub.f32 d24, d25, d24 @ I*s-R*c
  186. vadd.f32 d25, d22, d23 @ R*s-I*c
  187. beq 1f
  188. mov r12, #-16
  189. vld2.32 {d16,d18},[r9,:128],r12 @ in0u0,in0u1 in4d1,in4d0
  190. vld2.32 {d17,d19},[r8,:128],r12 @ in2u0,in2u1 in3d1,in3d0
  191. vneg.f32 d7, d7 @ R*s-I*c
  192. vld2.32 {d0, d2}, [r7,:128]! @ in4u0,in4u1 in2d1,in2d0
  193. vrev64.32 q9, q9 @ in4d0,in4d1 in3d0,in3d1
  194. vld2.32 {d1, d3}, [r2,:128]! @ in3u0,in3u1 in1d1,in1d0
  195. vsub.f32 d0, d18, d0 @ in4d-in4u I
  196. vld2.32 {d20,d21},[r4,:128]! @ c0,c1 s0,s1
  197. vrev64.32 q1, q1 @ in2d0,in2d1 in1d0,in1d1
  198. vld2.32 {d30,d31},[r5,:128],r12 @ c2,c3 s2,s3
  199. vadd.f32 d1, d1, d19 @ in3u+in3d -R
  200. vsub.f32 d16, d16, d2 @ in0u-in2d R
  201. vadd.f32 d17, d17, d3 @ in2u+in1d -I
  202. uxth r12, r6, ror #16
  203. uxth r6, r6
  204. add r12, r1, r12, lsl #3
  205. add r6, r1, r6, lsl #3
  206. vst2.32 {d6[0],d7[0]}, [r6,:64]
  207. vst2.32 {d6[1],d7[1]}, [r12,:64]
  208. uxth r6, r10, ror #16
  209. uxth r10, r10
  210. add r6 , r1, r6, lsl #3
  211. add r10, r1, r10, lsl #3
  212. vst2.32 {d24[0],d25[0]},[r10,:64]
  213. vst2.32 {d24[1],d25[1]},[r6,:64]
  214. b 1b
  215. 1:
  216. vneg.f32 d7, d7 @ R*s-I*c
  217. uxth r12, r6, ror #16
  218. uxth r6, r6
  219. add r12, r1, r12, lsl #3
  220. add r6, r1, r6, lsl #3
  221. vst2.32 {d6[0],d7[0]}, [r6,:64]
  222. vst2.32 {d6[1],d7[1]}, [r12,:64]
  223. uxth r6, r10, ror #16
  224. uxth r10, r10
  225. add r6 , r1, r6, lsl #3
  226. add r10, r1, r10, lsl #3
  227. vst2.32 {d24[0],d25[0]},[r10,:64]
  228. vst2.32 {d24[1],d25[1]},[r6,:64]
  229. mov r4, r0
  230. mov r6, r1
  231. bl ff_fft_calc_neon
  232. mov r12, #1
  233. ldr lr, [r4, #20] @ mdct_bits
  234. ldr r4, [r4, #24] @ tcos
  235. lsl r12, r12, lr @ n = 1 << nbits
  236. lsr lr, r12, #3 @ n8 = n >> 3
  237. add r4, r4, lr, lsl #3
  238. add r6, r6, lr, lsl #3
  239. sub r1, r4, #16
  240. sub r3, r6, #16
  241. mov r7, #-16
  242. mov r8, r6
  243. mov r0, r3
  244. vld2.32 {d0-d1}, [r3,:128], r7 @ d0 =r1,i1 d1 =r0,i0
  245. vld2.32 {d20-d21},[r6,:128]! @ d20=r2,i2 d21=r3,i3
  246. vld2.32 {d16,d18},[r1,:128], r7 @ c1,c0 s1,s0
  247. 1:
  248. subs lr, lr, #2
  249. vmul.f32 d7, d0, d18 @ r1*s1,r0*s0
  250. vld2.32 {d17,d19},[r4,:128]! @ c2,c3 s2,s3
  251. vmul.f32 d4, d1, d18 @ i1*s1,i0*s0
  252. vmul.f32 d5, d21, d19 @ i2*s2,i3*s3
  253. vmul.f32 d6, d20, d19 @ r2*s2,r3*s3
  254. vmul.f32 d24, d0, d16 @ r1*c1,r0*c0
  255. vmul.f32 d25, d20, d17 @ r2*c2,r3*c3
  256. vmul.f32 d22, d21, d17 @ i2*c2,i3*c3
  257. vmul.f32 d23, d1, d16 @ i1*c1,i0*c0
  258. vadd.f32 d4, d4, d24 @ i1*s1+r1*c1,i0*s0+r0*c0
  259. vadd.f32 d5, d5, d25 @ i2*s2+r2*c2,i3*s3+r3*c3
  260. vsub.f32 d6, d22, d6 @ i2*c2-r2*s2,i3*c3-r3*s3
  261. vsub.f32 d7, d23, d7 @ i1*c1-r1*s1,i0*c0-r0*s0
  262. vneg.f32 q2, q2
  263. beq 1f
  264. vld2.32 {d0-d1}, [r3,:128], r7
  265. vld2.32 {d20-d21},[r6,:128]!
  266. vld2.32 {d16,d18},[r1,:128], r7 @ c1,c0 s1,s0
  267. vrev64.32 q3, q3
  268. vst2.32 {d4,d6}, [r0,:128], r7
  269. vst2.32 {d5,d7}, [r8,:128]!
  270. b 1b
  271. 1:
  272. vrev64.32 q3, q3
  273. vst2.32 {d4,d6}, [r0,:128]
  274. vst2.32 {d5,d7}, [r8,:128]
  275. pop {r4-r10,pc}
  276. endfunc