git.sesse.net Git - ffmpeg/blob - libavcodec/arm/fft_fixed_neon.S

   1 /*
   2  * Copyright (c) 2011 Mans Rullgard <mans@mansr.com>
   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 #include "libavutil/arm/asm.S"
  22
  23 .macro  bflies          d0,  d1,  r0,  r1
  24         vrev64.32       \r0, \d1                @ t5, t6, t1, t2
  25         vhsub.s16       \r1, \d1, \r0           @ t1-t5, t2-t6, t5-t1, t6-t2
  26         vhadd.s16       \r0, \d1, \r0           @ t1+t5, t2+t6, t5+t1, t6+t2
  27         vext.16         \r1, \r1, \r1, #1       @ t2-t6, t5-t1, t6-t2, t1-t5
  28         vtrn.32         \r0, \r1                @ t1+t5, t2+t6, t2-t6, t5-t1
  29                                                 @ t5,    t6,    t4,    t3
  30         vhsub.s16       \d1, \d0, \r0
  31         vhadd.s16       \d0, \d0, \r0
  32 .endm
  33
  34 .macro  transform01     q0,  q1,  d3,  c0,  c1,  r0,  w0,  w1
  35         vrev32.16       \r0, \d3
  36         vmull.s16       \w0, \d3, \c0
  37         vmlal.s16       \w0, \r0, \c1
  38         vshrn.s32       \d3, \w0, #15
  39         bflies          \q0, \q1, \w0, \w1
  40 .endm
  41
  42 .macro  transform2      d0,  d1,  d2,  d3,  q0,  q1,  c0,  c1,  c2,  c3, \
  43                         r0,  r1,  w0,  w1
  44         vrev32.16       \r0, \d1
  45         vrev32.16       \r1, \d3
  46         vmull.s16       \w0, \d1, \c0
  47         vmlal.s16       \w0, \r0, \c1
  48         vmull.s16       \w1, \d3, \c2
  49         vmlal.s16       \w1, \r1, \c3
  50         vshrn.s32       \d1, \w0, #15
  51         vshrn.s32       \d3, \w1, #15
  52         bflies          \q0, \q1, \w0, \w1
  53 .endm
  54
  55 .macro  fft4            d0,  d1,  r0,  r1
  56         vhsub.s16       \r0, \d0, \d1           @ t3, t4, t8, t7
  57         vhsub.s16       \r1, \d1, \d0
  58         vhadd.s16       \d0, \d0, \d1           @ t1, t2, t6, t5
  59         vmov.i64        \d1, #0xffff00000000
  60         vbit            \r0, \r1, \d1
  61         vrev64.16       \r1, \r0                @ t7, t8, t4, t3
  62         vtrn.32         \r0, \r1                @ t3, t4, t7, t8
  63         vtrn.32         \d0, \r0                @ t1, t2, t3, t4, t6, t5, t8, t7
  64         vhsub.s16       \d1, \d0, \r0           @ r2, i2, r3, i1
  65         vhadd.s16       \d0, \d0, \r0           @ r0, i0, r1, i3
  66 .endm
  67
  68 .macro  fft8            d0,  d1,  d2,  d3,  q0,  q1,  c0,  c1,  r0,  r1, w0, w1
  69         fft4            \d0, \d1, \r0, \r1
  70         vtrn.32         \d0, \d1                @ z0, z2, z1, z3
  71         vhadd.s16       \r0, \d2, \d3           @ t1, t2, t3, t4
  72         vhsub.s16       \d3, \d2, \d3           @ z5, z7
  73         vmov            \d2, \r0
  74         transform01     \q0, \q1, \d3, \c0, \c1, \r0, \w0, \w1
  75 .endm
  76
  77 function fft4_neon
  78         vld1.16         {d0-d1},  [r0]
  79         fft4            d0,  d1,  d2,  d3
  80         vst1.16         {d0-d1},  [r0]
  81         bx              lr
  82 endfunc
  83
  84 function fft8_neon
  85         vld1.16         {d0-d3},  [r0,:128]
  86         movrel          r1,  coefs
  87         vld1.16         {d30},    [r1,:64]
  88         vdup.16         d31, d30[0]
  89         fft8            d0,  d1,  d2,  d3,  q0,  q1,  d31, d30, d20, d21, q8, q9
  90         vtrn.32         d0,  d1
  91         vtrn.32         d2,  d3
  92         vst1.16         {d0-d3},  [r0,:128]
  93         bx              lr
  94 endfunc
  95
  96 function fft16_neon
  97         vld1.16         {d0-d3},  [r0,:128]!
  98         vld1.16         {d4-d7},  [r0,:128]
  99         movrel          r1,  coefs
 100         sub             r0,  r0,  #32
 101         vld1.16         {d28-d31},[r1,:128]
 102         vdup.16         d31, d28[0]
 103         fft8            d0,  d1,  d2,  d3,  q0,  q1,  d31, d28, d20, d21, q8, q9
 104         vswp            d5,  d6
 105         fft4            q2,  q3,  q8,  q9
 106         vswp            d5,  d6
 107         vtrn.32         q0,  q1             @ z0, z4, z2, z6, z1, z5, z3, z7
 108         vtrn.32         q2,  q3             @ z8, z12,z10,z14,z9, z13,z11,z15
 109         vswp            d1,  d2
 110         vdup.16         d31, d28[0]
 111         transform01     q0,  q2,  d5,  d31, d28, d20, q8, q9
 112         vdup.16         d26, d29[0]
 113         vdup.16         d27, d30[0]
 114         transform2      d2,  d6,  d3,  d7,  q1,  q3,  d26, d30, d27, d29, \
 115                         d20, d21, q8,  q9
 116         vtrn.32         q0,  q1
 117         vtrn.32         q2,  q3
 118         vst1.16         {d0-d3},  [r0,:128]!
 119         vst1.16         {d4-d7},  [r0,:128]
 120         bx              lr
 121 endfunc
 122
 123 function fft_pass_neon
 124         push            {r4,lr}
 125         movrel          lr,  coefs + 24
 126         vld1.16         {d30},    [lr,:64]
 127         lsl             r12, r2,  #3
 128         vmov            d31, d30
 129         add             r3,  r1,  r2,  lsl #2
 130         mov             lr,  #-8
 131         sub             r3,  r3,  #2
 132         mov             r4,  r0
 133         vld1.16         {d27[]},  [r3,:16]
 134         sub             r3,  r3,  #6
 135         vld1.16         {q0},     [r4,:128], r12
 136         vld1.16         {q1},     [r4,:128], r12
 137         vld1.16         {q2},     [r4,:128], r12
 138         vld1.16         {q3},     [r4,:128], r12
 139         vld1.16         {d28},    [r1,:64]!
 140         vld1.16         {d29},    [r3,:64], lr
 141         vswp            d1,  d2
 142         vswp            d5,  d6
 143         vtrn.32         d0,  d1
 144         vtrn.32         d4,  d5
 145         vdup.16         d25, d28[1]
 146         vmul.s16        d27, d27, d31
 147         transform01     q0,  q2,  d5,  d25, d27, d20, q8,  q9
 148         b               2f
 149 1:
 150         mov             r4,  r0
 151         vdup.16         d26, d29[0]
 152         vld1.16         {q0},     [r4,:128], r12
 153         vld1.16         {q1},     [r4,:128], r12
 154         vld1.16         {q2},     [r4,:128], r12
 155         vld1.16         {q3},     [r4,:128], r12
 156         vld1.16         {d28},    [r1,:64]!
 157         vld1.16         {d29},    [r3,:64], lr
 158         vswp            d1,  d2
 159         vswp            d5,  d6
 160         vtrn.32         d0,  d1
 161         vtrn.32         d4,  d5
 162         vdup.16         d24, d28[0]
 163         vdup.16         d25, d28[1]
 164         vdup.16         d27, d29[3]
 165         vmul.s16        q13, q13, q15
 166         transform2      d0,  d4,  d1,  d5,  q0,  q2,  d24, d26, d25, d27, \
 167                         d16, d17, q9,  q10
 168 2:
 169         vtrn.32         d2,  d3
 170         vtrn.32         d6,  d7
 171         vdup.16         d24, d28[2]
 172         vdup.16         d26, d29[2]
 173         vdup.16         d25, d28[3]
 174         vdup.16         d27, d29[1]
 175         vmul.s16        q13, q13, q15
 176         transform2      d2,  d6,  d3,  d7,  q1,  q3,  d24, d26, d25, d27, \
 177                         d16, d17, q9,  q10
 178         vtrn.32         d0,  d1
 179         vtrn.32         d2,  d3
 180         vtrn.32         d4,  d5
 181         vtrn.32         d6,  d7
 182         vswp            d1,  d2
 183         vswp            d5,  d6
 184         mov             r4,  r0
 185         vst1.16         {q0},     [r4,:128], r12
 186         vst1.16         {q1},     [r4,:128], r12
 187         vst1.16         {q2},     [r4,:128], r12
 188         vst1.16         {q3},     [r4,:128], r12
 189         add             r0,  r0,  #16
 190         subs            r2,  r2,  #2
 191         bgt             1b
 192         pop             {r4,pc}
 193 endfunc
 194
 195 #define F_SQRT1_2   23170
 196 #define F_COS_16_1  30274
 197 #define F_COS_16_3  12540
 198
 199 const   coefs, align=4
 200         .short          F_SQRT1_2, -F_SQRT1_2, -F_SQRT1_2,  F_SQRT1_2
 201         .short          F_COS_16_1,-F_COS_16_1,-F_COS_16_1, F_COS_16_1
 202         .short          F_COS_16_3,-F_COS_16_3,-F_COS_16_3, F_COS_16_3
 203         .short          1,         -1,         -1,          1
 204 endconst
 205
 206 .macro  def_fft n, n2, n4
 207 function fft\n\()_neon
 208         push            {r4, lr}
 209         mov             r4,  r0
 210         bl              fft\n2\()_neon
 211         add             r0,  r4,  #\n4*2*4
 212         bl              fft\n4\()_neon
 213         add             r0,  r4,  #\n4*3*4
 214         bl              fft\n4\()_neon
 215         mov             r0,  r4
 216         pop             {r4, lr}
 217         movrelx         r1,  X(ff_cos_\n\()_fixed)
 218         mov             r2,  #\n4/2
 219         b               fft_pass_neon
 220 endfunc
 221 .endm
 222
 223         def_fft    32,    16,     8
 224         def_fft    64,    32,    16
 225         def_fft   128,    64,    32
 226         def_fft   256,   128,    64
 227         def_fft   512,   256,   128
 228         def_fft  1024,   512,   256
 229         def_fft  2048,  1024,   512
 230         def_fft  4096,  2048,  1024
 231         def_fft  8192,  4096,  2048
 232         def_fft 16384,  8192,  4096
 233         def_fft 32768, 16384,  8192
 234         def_fft 65536, 32768, 16384
 235
 236 function ff_fft_fixed_calc_neon, export=1
 237         ldr             r2,  [r0]
 238         sub             r2,  r2,  #2
 239         movrel          r3,  fft_fixed_tab_neon
 240         ldr             r3,  [r3, r2, lsl #2]
 241         mov             r0,  r1
 242         bx              r3
 243 endfunc
 244
 245 const   fft_fixed_tab_neon, relocate=1
 246         .word fft4_neon
 247         .word fft8_neon
 248         .word fft16_neon
 249         .word fft32_neon
 250         .word fft64_neon
 251         .word fft128_neon
 252         .word fft256_neon
 253         .word fft512_neon
 254         .word fft1024_neon
 255         .word fft2048_neon
 256         .word fft4096_neon
 257         .word fft8192_neon
 258         .word fft16384_neon
 259         .word fft32768_neon
 260         .word fft65536_neon
 261 endconst