| Index: third_party/openmax_dl/dl/sp/src/armSP_FFT_CToC_FC32_Radix4_unsafe_s.S
|
| diff --git a/third_party/openmax_dl/dl/sp/src/armSP_FFT_CToC_FC32_Radix4_unsafe_s.S b/third_party/openmax_dl/dl/sp/src/armSP_FFT_CToC_FC32_Radix4_unsafe_s.S
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..ae450c5f629793cecdd72f97f457bf6223e0ff77
|
| --- /dev/null
|
| +++ b/third_party/openmax_dl/dl/sp/src/armSP_FFT_CToC_FC32_Radix4_unsafe_s.S
|
| @@ -0,0 +1,331 @@
|
| +@//
|
| +@// Copyright (c) 2013 The WebRTC project authors. All Rights Reserved.
|
| +@//
|
| +@// Use of this source code is governed by a BSD-style license
|
| +@// that can be found in the LICENSE file in the root of the source
|
| +@// tree. An additional intellectual property rights grant can be found
|
| +@// in the file PATENTS. All contributing project authors may
|
| +@// be found in the AUTHORS file in the root of the source tree.
|
| +@//
|
| +@//
|
| +@// This is a modification of armSP_FFT_CToC_SC32_Radix4_unsafe_s.s
|
| +@// to support float instead of SC32.
|
| +@//
|
| +
|
| +@//
|
| +@// Description:
|
| +@// Compute a Radix 4 FFT stage for a N point complex signal
|
| +@//
|
| +@//
|
| +
|
| +
|
| +@// Include standard headers
|
| +
|
| +#include "dl/api/armCOMM_s.h"
|
| +#include "dl/api/omxtypes_s.h"
|
| +
|
| +
|
| +@// Import symbols required from other files
|
| +@// (For example tables)
|
| +
|
| +
|
| +
|
| +
|
| +@// Set debugging level
|
| +@//DEBUG_ON SETL {TRUE}
|
| +
|
| +
|
| +
|
| +@// Guarding implementation by the processor name
|
| +
|
| +
|
| +
|
| +
|
| +@// Guarding implementation by the processor name
|
| +
|
| +
|
| +@// Import symbols required from other files
|
| +@// (For example tables)
|
| +
|
| +
|
| +@//Input Registers
|
| +
|
| +#define pSrc r0
|
| +#define pDst r2
|
| +#define pTwiddle r1
|
| +#define subFFTNum r6
|
| +#define subFFTSize r7
|
| +
|
| +
|
| +
|
| +@//Output Registers
|
| +
|
| +
|
| +@//Local Scratch Registers
|
| +
|
| +#define grpCount r3
|
| +#define pointStep r4
|
| +#define outPointStep r5
|
| +#define stepTwiddle r12
|
| +#define setCount r14
|
| +#define srcStep r8
|
| +#define setStep r9
|
| +#define dstStep r10
|
| +#define twStep r11
|
| +#define t1 r3
|
| +
|
| +@// Neon Registers
|
| +
|
| +#define dW1 D0.F32
|
| +#define dW2 D1.F32
|
| +#define dW3 D2.F32
|
| +
|
| +#define dXr0 D4.F32
|
| +#define dXi0 D5.F32
|
| +#define dXr1 D6.F32
|
| +#define dXi1 D7.F32
|
| +#define dXr2 D8.F32
|
| +#define dXi2 D9.F32
|
| +#define dXr3 D10.F32
|
| +#define dXi3 D11.F32
|
| +#define dYr0 D12.F32
|
| +#define dYi0 D13.F32
|
| +#define dYr1 D14.F32
|
| +#define dYi1 D15.F32
|
| +#define dYr2 D16.F32
|
| +#define dYi2 D17.F32
|
| +#define dYr3 D18.F32
|
| +#define dYi3 D19.F32
|
| +#define qT0 d16.f32
|
| +#define qT1 d18.f32
|
| +#define qT2 d12.f32
|
| +#define qT3 d14.f32
|
| +#define dZr0 D20.F32
|
| +#define dZi0 D21.F32
|
| +#define dZr1 D22.F32
|
| +#define dZi1 D23.F32
|
| +#define dZr2 D24.F32
|
| +#define dZi2 D25.F32
|
| +#define dZr3 D26.F32
|
| +#define dZi3 D27.F32
|
| +
|
| +#define qY0 Q6.F32
|
| +#define qY1 Q7.F32
|
| +#define qY2 Q8.F32
|
| +#define qY3 Q9.F32
|
| +#define qX0 Q2.F32
|
| +#define qZ0 Q10.F32
|
| +#define qZ1 Q11.F32
|
| +#define qZ2 Q12.F32
|
| +#define qZ3 Q13.F32
|
| +
|
| + .MACRO FFTSTAGE scaled, inverse , name
|
| +
|
| + @// Define stack arguments
|
| +
|
| +
|
| + @// Update grpCount and grpSize rightaway inorder to reuse
|
| + @// pGrpCount and pGrpSize regs
|
| +
|
| + LSL grpCount,subFFTSize,#2
|
| + LSR subFFTNum,subFFTNum,#2
|
| + MOV subFFTSize,grpCount
|
| +
|
| + VLD1 dW1,[pTwiddle] @//[wi | wr]
|
| + @// pT0+1 increments pT0 by 8 bytes
|
| + @// pT0+pointStep = increment of 8*pointStep bytes = 2*grpSize bytes
|
| + MOV pointStep,subFFTNum,LSL #1
|
| +
|
| +
|
| + @// pOut0+1 increments pOut0 by 8 bytes
|
| + @// pOut0+outPointStep == increment of 8*outPointStep bytes
|
| + @// = 2*size bytes
|
| +
|
| + MOV stepTwiddle,#0
|
| + VLD1 dW2,[pTwiddle] @//[wi | wr]
|
| + SMULBB outPointStep,grpCount,pointStep
|
| + LSL pointStep,pointStep,#2 @// 2*grpSize
|
| +
|
| + VLD1 dW3,[pTwiddle] @//[wi | wr]
|
| + MOV srcStep,pointStep,LSL #1 @// srcStep = 2*pointStep
|
| + ADD setStep,srcStep,pointStep @// setStep = 3*pointStep
|
| +
|
| + RSB setStep,setStep,#0 @// setStep = - 3*pointStep
|
| + SUB srcStep,srcStep,#16 @// srcStep = 2*pointStep-16
|
| +
|
| + MOV dstStep,outPointStep,LSL #1
|
| + ADD dstStep,dstStep,outPointStep @// dstStep = 3*outPointStep
|
| + @// dstStep = - 3*outPointStep+16
|
| + RSB dstStep,dstStep,#16
|
| +
|
| +
|
| +
|
| +radix4GrpLoop\name :
|
| +
|
| + VLD2 {dXr0,dXi0},[pSrc],pointStep @// data[0]
|
| + ADD stepTwiddle,stepTwiddle,pointStep
|
| + VLD2 {dXr1,dXi1},[pSrc],pointStep @// data[1]
|
| + @// set pTwiddle to the first point
|
| + ADD pTwiddle,pTwiddle,stepTwiddle
|
| + VLD2 {dXr2,dXi2},[pSrc],pointStep @// data[2]
|
| + MOV twStep,stepTwiddle,LSL #2
|
| +
|
| + @// data[3] & update pSrc for the next set
|
| + VLD2 {dXr3,dXi3},[pSrc],setStep
|
| + SUB twStep,stepTwiddle,twStep @// twStep = -3*stepTwiddle
|
| +
|
| + MOV setCount,pointStep,LSR #3
|
| + @// set pSrc to data[0] of the next set
|
| + ADD pSrc,pSrc,#16
|
| + @// increment to data[1] of the next set
|
| + ADD pSrc,pSrc,pointStep
|
| +
|
| +
|
| + @// Loop on the sets
|
| +
|
| +radix4SetLoop\name :
|
| +
|
| +
|
| +
|
| + .ifeqs "\inverse", "TRUE"
|
| + VMUL dZr1,dXr1,dW1[0]
|
| + VMUL dZi1,dXi1,dW1[0]
|
| + VMUL dZr2,dXr2,dW2[0]
|
| + VMUL dZi2,dXi2,dW2[0]
|
| + VMUL dZr3,dXr3,dW3[0]
|
| + VMUL dZi3,dXi3,dW3[0]
|
| +
|
| + VMLA dZr1,dXi1,dW1[1] @// real part
|
| + VMLS dZi1,dXr1,dW1[1] @// imag part
|
| +
|
| + @// data[1] for next iteration
|
| + VLD2 {dXr1,dXi1},[pSrc],pointStep
|
| +
|
| + VMLA dZr2,dXi2,dW2[1] @// real part
|
| + VMLS dZi2,dXr2,dW2[1] @// imag part
|
| +
|
| + @// data[2] for next iteration
|
| + VLD2 {dXr2,dXi2},[pSrc],pointStep
|
| +
|
| + VMLA dZr3,dXi3,dW3[1] @// real part
|
| + VMLS dZi3,dXr3,dW3[1] @// imag part
|
| + .else
|
| + VMUL dZr1,dXr1,dW1[0]
|
| + VMUL dZi1,dXi1,dW1[0]
|
| + VMUL dZr2,dXr2,dW2[0]
|
| + VMUL dZi2,dXi2,dW2[0]
|
| + VMUL dZr3,dXr3,dW3[0]
|
| + VMUL dZi3,dXi3,dW3[0]
|
| +
|
| + VMLS dZr1,dXi1,dW1[1] @// real part
|
| + VMLA dZi1,dXr1,dW1[1] @// imag part
|
| +
|
| + @// data[1] for next iteration
|
| + VLD2 {dXr1,dXi1},[pSrc],pointStep
|
| +
|
| + VMLS dZr2,dXi2,dW2[1] @// real part
|
| + VMLA dZi2,dXr2,dW2[1] @// imag part
|
| +
|
| + @// data[2] for next iteration
|
| + VLD2 {dXr2,dXi2},[pSrc],pointStep
|
| +
|
| + VMLS dZr3,dXi3,dW3[1] @// real part
|
| + VMLA dZi3,dXr3,dW3[1] @// imag part
|
| + .endif
|
| +
|
| + @// data[3] & update pSrc to data[0]
|
| + @// But don't read on the very last iteration because that reads past
|
| + @// the end of pSrc. The last iteration is grpCount = 4, setCount = 2.
|
| + cmp grpCount, #4
|
| + cmpeq setCount, #2 @// Test setCount if grpCount = 4
|
| + @// These are executed only if both grpCount = 4 and setCount = 2
|
| + addeq pSrc, pSrc, setStep
|
| + beq radix4SkipRead\name
|
| + VLD2 {dXr3,dXi3},[pSrc],setStep
|
| +radix4SkipRead\name:
|
| + SUBS setCount,setCount,#2
|
| +
|
| + @// finish first stage of 4 point FFT
|
| + VADD qY0,qX0,qZ2
|
| + VSUB qY2,qX0,qZ2
|
| +
|
| + @// data[0] for next iteration
|
| + VLD2 {dXr0,dXi0},[pSrc :128]!
|
| + VADD qY1,qZ1,qZ3
|
| + VSUB qY3,qZ1,qZ3
|
| +
|
| + @// finish second stage of 4 point FFT
|
| +
|
| + VSUB qZ0,qY2,qY1
|
| +
|
| +
|
| + .ifeqs "\inverse", "TRUE"
|
| +
|
| + VADD dZr3,dYr0,dYi3
|
| + VST2 {dZr0,dZi0},[pDst :128],outPointStep
|
| + VSUB dZi3,dYi0,dYr3
|
| +
|
| + VADD qZ2,qY2,qY1
|
| + VST2 {dZr3,dZi3},[pDst :128],outPointStep
|
| +
|
| + VSUB dZr1,dYr0,dYi3
|
| + VST2 {dZr2,dZi2},[pDst :128],outPointStep
|
| + VADD dZi1,dYi0,dYr3
|
| +
|
| + VST2 {dZr1,dZi1},[pDst :128],dstStep
|
| +
|
| +
|
| + .else
|
| +
|
| + VSUB dZr1,dYr0,dYi3
|
| + VST2 {dZr0,dZi0},[pDst :128],outPointStep
|
| + VADD dZi1,dYi0,dYr3
|
| +
|
| + VADD qZ2,qY2,qY1
|
| + VST2 {dZr1,dZi1},[pDst :128],outPointStep
|
| +
|
| + VADD dZr3,dYr0,dYi3
|
| + VST2 {dZr2,dZi2},[pDst :128],outPointStep
|
| + VSUB dZi3,dYi0,dYr3
|
| +
|
| + VST2 {dZr3,dZi3},[pDst :128],dstStep
|
| +
|
| +
|
| + .endif
|
| +
|
| + @// increment to data[1] of the next set
|
| + ADD pSrc,pSrc,pointStep
|
| + BGT radix4SetLoop\name
|
| +
|
| +
|
| + VLD1 dW1,[pTwiddle :64],stepTwiddle @//[wi | wr]
|
| + @// subtract 4 since grpCount multiplied by 4
|
| + SUBS grpCount,grpCount,#4
|
| + VLD1 dW2,[pTwiddle :64],stepTwiddle @//[wi | wr]
|
| + @// increment pSrc for the next grp
|
| + ADD pSrc,pSrc,srcStep
|
| + VLD1 dW3,[pTwiddle :64],twStep @//[wi | wr]
|
| + BGT radix4GrpLoop\name
|
| +
|
| +
|
| + @// Reset and Swap pSrc and pDst for the next stage
|
| + MOV t1,pDst
|
| + @// pDst -= 2*size; pSrc -= 8*size bytes
|
| + SUB pDst,pSrc,outPointStep,LSL #2
|
| + SUB pSrc,t1,outPointStep
|
| +
|
| +
|
| + .endm
|
| +
|
| +
|
| + M_START armSP_FFTFwd_CToC_FC32_Radix4_OutOfPlace_unsafe,r4
|
| + FFTSTAGE "FALSE","FALSE",FWD
|
| + M_END
|
| +
|
| +
|
| + M_START armSP_FFTInv_CToC_FC32_Radix4_OutOfPlace_unsafe,r4
|
| + FFTSTAGE "FALSE","TRUE",INV
|
| + M_END
|
| +
|
| +
|
| + .end
|
|
|