1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
/*
* Copyright (C) 2020 Linux Studio Plugins Project <https://lsp-plug.in/>
* (C) 2020 Vladimir Sadovnikov <sadko4u@gmail.com>
*
* This file is part of lsp-dsp-lib
* Created on: 31 мар. 2020 г.
*
* lsp-dsp-lib is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* any later version.
*
* lsp-dsp-lib is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with lsp-dsp-lib. If not, see <https://www.gnu.org/licenses/>.
*/
#ifndef PRIVATE_DSP_ARCH_X86_SSE_FFT_P_BUTTERFLY_H_
#define PRIVATE_DSP_ARCH_X86_SSE_FFT_P_BUTTERFLY_H_
#ifndef PRIVATE_DSP_ARCH_X86_SSE_IMPL
#error "This header should not be included directly"
#endif /* PRIVATE_DSP_ARCH_X86_SSE_IMPL */
namespace lsp
{
namespace sse
{
#define FFT_BUTTERFLY_BODY(add_b, add_a) \
/* Init pointers */ \
float *a = &dst[blk*pairs*2]; \
float *b = &a[pairs]; \
size_t p = pairs; \
__IF_32(size_t tmp1); \
\
ARCH_X86_ASM \
( \
/* Prepare angle */ \
__ASM_EMIT32("mov %[a], %[tmp1]") \
__ASM_EMIT32("mov %[XFFT_A], %[a]") \
__ASM_EMIT32("movaps 0x00(%[a], %[rank]), %%xmm6") /* xmm6 = angle_re[0..3] */ \
__ASM_EMIT32("movaps 0x10(%[a], %[rank]), %%xmm7") /* xmm7 = angle_im[0..3] */ \
__ASM_EMIT32("mov %[tmp1], %[a]") \
__ASM_EMIT64("movaps 0x00(%[XFFT_A], %[rank]), %%xmm6") /* xmm6 = angle_re[0..3] */ \
__ASM_EMIT64("movaps 0x10(%[XFFT_A], %[rank]), %%xmm7") /* xmm7 = angle_im[0..3] */ \
/* Start loop */ \
__ASM_EMIT(".align 16") \
__ASM_EMIT("1:") \
\
/* Load complex values */ \
/* predicate: xmm6 = w_re[0..3] */ \
/* predicate: xmm7 = w_im[0..3] */ \
__ASM_EMIT("movups 0x00(%[a]), %%xmm0") /* xmm0 = a_re[0..3] */ \
__ASM_EMIT("movups 0x10(%[a]), %%xmm1") /* xmm1 = a_im[0..3] */ \
__ASM_EMIT("movups 0x00(%[b]), %%xmm2") /* xmm2 = b_re[0..3] */ \
__ASM_EMIT("movups 0x10(%[b]), %%xmm3") /* xmm3 = b_im[0..3] */ \
\
/* Calculate complex multiplication */ \
__ASM_EMIT("movaps %%xmm2, %%xmm4") /* xmm4 = b_re[0..3] */ \
__ASM_EMIT("movaps %%xmm3, %%xmm5") /* xmm5 = b_im[0..3] */ \
__ASM_EMIT("mulps %%xmm6, %%xmm2") /* xmm2 = w_re[0..3] * b_re[0..3] */ \
__ASM_EMIT("mulps %%xmm7, %%xmm4") /* xmm4 = w_im[0..3] * b_re[0..3] */ \
__ASM_EMIT("mulps %%xmm6, %%xmm3") /* xmm3 = w_re[0..3] * b_im[0..3] */ \
__ASM_EMIT("mulps %%xmm7, %%xmm5") /* xmm5 = w_im[0..3] * b_im[0..3] */ \
__ASM_EMIT(add_a " %%xmm4, %%xmm3") /* xmm3 = c_im[0..3] = w_re[0..3] * b_im[0..3] -+ w_im[0..3] * b_re[0..3] */ \
__ASM_EMIT(add_b " %%xmm5, %%xmm2") /* xmm2 = c_re[0..3] = w_re[0..3] * b_re[0..3] +- w_im[0..3] * b_im[0..3] */ \
\
/* Perform butterfly */ \
__ASM_EMIT("movaps %%xmm0, %%xmm4") /* xmm4 = a_re[0..3] */ \
__ASM_EMIT("movaps %%xmm1, %%xmm5") /* xmm5 = a_im[0..3] */ \
__ASM_EMIT("subps %%xmm2, %%xmm0") /* xmm0 = a_re[0..3] - c_re[0..3] */ \
__ASM_EMIT("subps %%xmm3, %%xmm1") /* xmm1 = a_im[0..3] - c_im[0..3] */ \
__ASM_EMIT("addps %%xmm4, %%xmm2") /* xmm2 = a_re[0..3] + c_re[0..3] */ \
__ASM_EMIT("addps %%xmm5, %%xmm3") /* xmm3 = a_im[0..3] + c_im[0..3] */ \
\
/* Store values */ \
__ASM_EMIT("movups %%xmm2, 0x00(%[a])") \
__ASM_EMIT("movups %%xmm3, 0x10(%[a])") \
__ASM_EMIT("movups %%xmm0, 0x00(%[b])") \
__ASM_EMIT("movups %%xmm1, 0x10(%[b])") \
\
/* Update pointers */ \
__ASM_EMIT("add $0x20, %[a]") \
__ASM_EMIT("add $0x20, %[b]") \
\
/* Repeat loop */ \
__ASM_EMIT("sub $8, %[p]") \
__ASM_EMIT("jz 2f") \
\
/* Rotate angle */ \
__ASM_EMIT("movaps 0x00(%[XFFT_W], %[rank]), %%xmm0") /* xmm0 = w_re[0..3] */ \
__ASM_EMIT("movaps 0x10(%[XFFT_W], %[rank]), %%xmm1") /* xmm1 = w_im[0..3] */ \
\
__ASM_EMIT("movaps %%xmm0, %%xmm2") /* xmm2 = w_re[0..3] */ \
__ASM_EMIT("movaps %%xmm1, %%xmm3") /* xmm3 = w_im[0..3] */ \
\
__ASM_EMIT("mulps %%xmm6, %%xmm3") /* xmm3 = a_re[0..3] * w_im[0..3] */ \
__ASM_EMIT("mulps %%xmm7, %%xmm1") /* xmm1 = a_im[0..3] * w_im[0..3] */ \
__ASM_EMIT("mulps %%xmm0, %%xmm6") /* xmm6 = a_re[0..3] * w_re[0..3] */ \
__ASM_EMIT("mulps %%xmm2, %%xmm7") /* xmm7 = a_im[0..3] * w_re[0..3] */ \
__ASM_EMIT("subps %%xmm1, %%xmm6") /* xmm6 = a_re[0..3] * w_re[0..3] + a_im[0..3] * w_im[0..3] */ \
__ASM_EMIT("addps %%xmm3, %%xmm7") /* xmm7 = a_im[0..3] * w_re[0..3] - a_re[0..3] * w_im[0..3] */ \
\
/* Repeat loop */ \
__ASM_EMIT("jmp 1b") \
__ASM_EMIT("2:") \
\
: [a] "+r" (a), [b] "+r"(b), [p] "+r" (p) \
: [rank] "r" (rank), [XFFT_W] "r"(XFFT_W), \
__IF_64([XFFT_A] "r" (XFFT_A)) \
__IF_32([XFFT_A] "g" (&XFFT_A[0]), [tmp1] "g" (&tmp1)) \
: "cc", "memory", \
"%xmm0", "%xmm1", "%xmm2", "%xmm3", \
"%xmm4", "%xmm5", "%xmm6", "%xmm7" \
);
static inline void packed_butterfly_direct(float *dst, size_t rank, size_t blocks)
{
size_t pairs = 1 << (rank + 1);
rank = (rank - 2) << 5;
for (size_t blk=0; blk<blocks; ++blk)
{
FFT_BUTTERFLY_BODY("addps", "subps");
}
}
static inline void packed_butterfly_reverse(float *dst, size_t rank, size_t blocks)
{
size_t pairs = 1 << (rank + 1);
rank = (rank - 2) << 5;
for (size_t blk=0; blk<blocks; ++blk)
{
FFT_BUTTERFLY_BODY("subps", "addps");
}
}
#undef FFT_BUTTERFLY_BODY
} /* namespace sse */
} /* namespace lsp */
#endif /* PRIVATE_DSP_ARCH_X86_SSE_FFT_P_BUTTERFLY_H_ */
|
