我有一个输入信号,我想存储在ArrayList中,然后将其转换为Complex,就像这样
-0.03480425839330703
0.07910192950176387
0.7233322451735928
0.1659819820667019
并输出这样的FFT
0.9336118983487516
-0.7581365035668999 + 0.08688005256493803i
0.44344407521182005
-0.7581365035668999 - 0.08688005256493803i
这是一个复杂的结构,我想将其转换为ArrayList类型。同时删除+ 0.08688005256493803i
值。
所以我需要的只是这些价值
0.9336118983487516
-0.7581365035668999
0.44344407521182005
-0.7581365035668999
解决这个问题的最佳方式是什么?
这是我正在使用的代码
public static Complex[] fft(Complex[] x) {
int N = x.length;
// base case
if (N == 1) return new Complex[] { x[0] };
// radix 2 Cooley-Tukey FFT
if (N % 2 != 0) { throw new RuntimeException("N is not a power of 2"); }
// fft of even terms
Complex[] even = new Complex[N/2];
for (int k = 0; k < N/2; k++) {
even[k] = x[2*k];
}
Complex[] q = fft(even);
// fft of odd terms
Complex[] odd = even; // reuse the array
for (int k = 0; k < N/2; k++) {
odd[k] = x[2*k + 1];
}
Complex[] r = fft(odd);
// combine
Complex[] y = new Complex[N];
for (int k = 0; k < N/2; k++) {
double kth = -2 * k * Math.PI / N;
Complex wk = new Complex(Math.cos(kth), Math.sin(kth));
y[k] = q[k].plus(wk.times(r[k]));
y[k + N/2] = q[k].minus(wk.times(r[k]));
}
return y;
}
答案 0 :(得分:1)
您要做的只是删除Complex
数据结构的虚构部分。
因为你没有告诉我们Complex
类假设它有实际成员(例如double real;
)
要删除虚部,只需调用类似complex.getRealPart()
的内容,或访问complex.real
(替换为您的真实会员名称)。
要撰写ArrayList<Double>
,请使用以下代码段:
ArrayList<Double> list = new ArrayList<Double>();
for (Complex c : complexes) { // complexes your array of complexes returned from for fft
list.add(c.getRealpart());
}
注意:以防万一,我可能错了,但我认为不是真正的部分,你需要复数的绝对值。计算它使用:
Math.sqrt(c.getRealPart() * c.getRealPart() + c.getImPart() * c.getImPart());
答案 1 :(得分:0)
据我所知,你只想要复杂价值的真实部分。如果是这种情况,可能你的Complex类也有getReal()和getImaginary()(或类似的)方法 - 所以只需使用getReal()。