对armadillo矩阵进行fft与ifft的注意点

东方耀

对armadillo矩阵进行fft与ifft的注意点


//警告：当变换长度是 2 的幂时，变换速度最快
    //ifft()变换只能对复矩阵
    //如果给定一个矩阵，则对矩阵的每个列向量进行变换

1. #include <iostream>
   
2. #include <armadillo>
   
3. #include <vector>
   
4. #include <math.h>
   
5. 
   
6. using namespace std;
   
7. 
   
8. //对armadillo矩阵进行fft与ifft的注意点
   
9. 
   
10. int main(){
    
11.     float f0 = 100.0;
    
12.     float fs = 1000.0;
    
13.     std::cout << "信号频率=" << f0 << ";采样率=" << fs << endl;
    
14.     std::cout << "信号周期=" << 1/f0 << ";采样间隔=" << 1/fs << endl;
    
15.     std::cout << "一个周期采多少个点=" << fs/f0 << endl;
    
16.     int sample_num = 128;
    
17.     std::cout << "采样点数确定后的最大时间=" << sample_num*1/fs << endl;
    
18.     vector<float> signal_Rx;
    
19.     for(int i=0;i<sample_num;i++){
    
20.         signal_Rx.emplace_back(cos(2*M_PI*f0 * i / fs));
    
21.     }
    
22. 
    
23.     arma::fmat A(2, sample_num, arma::fill::zeros);   // float实矩阵
    
24.     arma::cx_fmat B, C;                               // complex_float 复矩阵
    
25.     // arma::frowvec r=signal_Rx;
    
26.     // A.row(0) = r;
    
27.     A.row(0) = (arma::frowvec)signal_Rx;
    
28.     //A.row(1) = (arma::frowvec)signal_Rx;
    
29. 
    
30.     A.print("原始A矩阵：");
    
31. 
    
32.     //警告：当变换长度是 2 的幂时，变换速度最快
    
33.     //ifft()变换只能对复矩阵
    
34.     //如果给定一个矩阵，则对矩阵的每个列向量进行变换
    
35.     //arma::fft(A);
    
36.     B = arma::fft(A.t());
    
37.     B = B.t();
    
38.     cout << "B.n_rows行数: " << B.n_rows << ";B.n_cols列数: " << B.n_cols << endl;
    
39.     B.print("fft变换后的B矩阵：");
    
40. 
    
41.     C = arma::ifft(B.t());
    
42.     C = C.t();
    
43. 
    
44.     cout << "C.n_rows行数: " << C.n_rows << ";C.n_cols列数: " << C.n_cols << endl;
    
45.     C.print("ifft变换后的C矩阵：");
    
46. 
    
47. 
    
48.     return 0;
    
49. }

复制代码