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.

304 lines
12KB

  1. /*
  2. * ARM NEON optimised MDCT
  3. * Copyright (c) 2009 Mans Rullgard <mans@mansr.com>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg 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. * FFmpeg 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 FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #include "asm.S"
  22. preserve8
  23. .text
  24. #define ff_fft_calc_neon X(ff_fft_calc_neon)
  25. function ff_imdct_half_neon, export=1
  26. push {r4-r8,lr}
  27. mov r12, #1
  28. ldr lr, [r0, #28] @ mdct_bits
  29. ldr r4, [r0, #32] @ tcos
  30. ldr r3, [r0, #8] @ revtab
  31. lsl r12, r12, lr @ n = 1 << nbits
  32. lsr lr, r12, #2 @ n4 = n >> 2
  33. add r7, r2, r12, lsl #1
  34. mov r12, #-16
  35. sub r7, r7, #16
  36. vld2.32 {d16-d17},[r7,:128],r12 @ d16=x,n1 d17=x,n0
  37. vld2.32 {d0-d1}, [r2,:128]! @ d0 =m0,x d1 =m1,x
  38. vrev64.32 d17, d17
  39. vld2.32 {d2,d3}, [r4,:128]! @ d2=c0,c1 d3=s0,s2
  40. vmul.f32 d6, d17, d2
  41. vmul.f32 d7, d0, d2
  42. 1:
  43. subs lr, lr, #2
  44. ldr r6, [r3], #4
  45. vmul.f32 d4, d0, d3
  46. vmul.f32 d5, d17, d3
  47. vsub.f32 d4, d6, d4
  48. vadd.f32 d5, d5, d7
  49. uxth r8, r6, ror #16
  50. uxth r6, r6
  51. add r8, r1, r8, lsl #3
  52. add r6, r1, r6, lsl #3
  53. beq 1f
  54. vld2.32 {d16-d17},[r7,:128],r12
  55. vld2.32 {d0-d1}, [r2,:128]!
  56. vrev64.32 d17, d17
  57. vld2.32 {d2,d3}, [r4,:128]! @ d2=c0,c1 d3=s0,s2
  58. vmul.f32 d6, d17, d2
  59. vmul.f32 d7, d0, d2
  60. vst2.32 {d4[0],d5[0]}, [r6,:64]
  61. vst2.32 {d4[1],d5[1]}, [r8,:64]
  62. b 1b
  63. 1:
  64. vst2.32 {d4[0],d5[0]}, [r6,:64]
  65. vst2.32 {d4[1],d5[1]}, [r8,:64]
  66. mov r4, r0
  67. mov r6, r1
  68. bl ff_fft_calc_neon
  69. mov r12, #1
  70. ldr lr, [r4, #28] @ mdct_bits
  71. ldr r4, [r4, #32] @ tcos
  72. lsl r12, r12, lr @ n = 1 << nbits
  73. lsr lr, r12, #3 @ n8 = n >> 3
  74. add r4, r4, lr, lsl #3
  75. add r6, r6, lr, lsl #3
  76. sub r1, r4, #16
  77. sub r3, r6, #16
  78. mov r7, #-16
  79. mov r8, r6
  80. mov r0, r3
  81. vld2.32 {d0-d1}, [r3,:128], r7 @ d0 =i1,r1 d1 =i0,r0
  82. vld2.32 {d20-d21},[r6,:128]! @ d20=i2,r2 d21=i3,r3
  83. vld2.32 {d16,d18},[r1,:128], r7 @ d16=c1,c0 d18=s1,s0
  84. 1:
  85. subs lr, lr, #2
  86. vmul.f32 d7, d0, d18
  87. vld2.32 {d17,d19},[r4,:128]! @ d17=c2,c3 d19=s2,s3
  88. vmul.f32 d4, d1, d18
  89. vmul.f32 d5, d21, d19
  90. vmul.f32 d6, d20, d19
  91. vmul.f32 d22, d1, d16
  92. vmul.f32 d23, d21, d17
  93. vmul.f32 d24, d0, d16
  94. vmul.f32 d25, d20, d17
  95. vadd.f32 d7, d7, d22
  96. vadd.f32 d6, d6, d23
  97. vsub.f32 d4, d4, d24
  98. vsub.f32 d5, d5, d25
  99. beq 1f
  100. vld2.32 {d0-d1}, [r3,:128], r7
  101. vld2.32 {d20-d21},[r6,:128]!
  102. vld2.32 {d16,d18},[r1,:128], r7 @ d16=c1,c0 d18=s1,s0
  103. vrev64.32 q3, q3
  104. vst2.32 {d4,d6}, [r0,:128], r7
  105. vst2.32 {d5,d7}, [r8,:128]!
  106. b 1b
  107. 1:
  108. vrev64.32 q3, q3
  109. vst2.32 {d4,d6}, [r0,:128]
  110. vst2.32 {d5,d7}, [r8,:128]
  111. pop {r4-r8,pc}
  112. .endfunc
  113. function ff_imdct_calc_neon, export=1
  114. push {r4-r6,lr}
  115. ldr r3, [r0, #28]
  116. mov r4, #1
  117. mov r5, r1
  118. lsl r4, r4, r3
  119. add r1, r1, r4
  120. bl ff_imdct_half_neon
  121. add r0, r5, r4, lsl #2
  122. add r1, r5, r4, lsl #1
  123. sub r0, r0, #8
  124. sub r2, r1, #16
  125. mov r3, #-16
  126. mov r6, #-8
  127. vmov.i32 d30, #1<<31
  128. 1:
  129. vld1.32 {d0-d1}, [r2,:128], r3
  130. pld [r0, #-16]
  131. vrev64.32 q0, q0
  132. vld1.32 {d2-d3}, [r1,:128]!
  133. veor d4, d1, d30
  134. pld [r2, #-16]
  135. vrev64.32 q1, q1
  136. veor d5, d0, d30
  137. vst1.32 {d2}, [r0,:64], r6
  138. vst1.32 {d3}, [r0,:64], r6
  139. vst1.32 {d4-d5}, [r5,:128]!
  140. subs r4, r4, #16
  141. bgt 1b
  142. pop {r4-r6,pc}
  143. .endfunc
  144. function ff_mdct_calc_neon, export=1
  145. push {r4-r10,lr}
  146. mov r12, #1
  147. ldr lr, [r0, #28] @ mdct_bits
  148. ldr r4, [r0, #32] @ tcos
  149. ldr r3, [r0, #8] @ revtab
  150. lsl lr, r12, lr @ n = 1 << nbits
  151. add r7, r2, lr @ in4u
  152. sub r9, r7, #16 @ in4d
  153. add r2, r7, lr, lsl #1 @ in3u
  154. add r8, r9, lr, lsl #1 @ in3d
  155. add r5, r4, lr, lsl #1
  156. sub r5, r5, #16
  157. sub r3, r3, #4
  158. mov r12, #-16
  159. vld2.32 {d16,d18},[r9,:128],r12 @ in0u0,in0u1 in4d1,in4d0
  160. vld2.32 {d17,d19},[r8,:128],r12 @ in2u0,in2u1 in3d1,in3d0
  161. vld2.32 {d0, d2}, [r7,:128]! @ in4u0,in4u1 in2d1,in2d0
  162. vrev64.32 q9, q9 @ in4d0,in4d1 in3d0,in3d1
  163. vld2.32 {d1, d3}, [r2,:128]! @ in3u0,in3u1 in1d1,in1d0
  164. vsub.f32 d0, d18, d0 @ in4d-in4u I
  165. vld2.32 {d20,d21},[r4,:128]! @ c0,c1 s0,s1
  166. vrev64.32 q1, q1 @ in2d0,in2d1 in1d0,in1d1
  167. vld2.32 {d30,d31},[r5,:128],r12 @ c2,c3 s2,s3
  168. vadd.f32 d1, d1, d19 @ in3u+in3d -R
  169. vsub.f32 d16, d16, d2 @ in0u-in2d R
  170. vadd.f32 d17, d17, d3 @ in2u+in1d -I
  171. 1:
  172. vmul.f32 d7, d0, d21 @ I*s
  173. ldr r10, [r3, lr, lsr #1]
  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, #28] @ mdct_bits
  234. ldr r4, [r4, #32] @ 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