GBT6323标准中针对汽车频域特性的角脉冲试验的简单示例

GBT6323标准中针对汽车频域特性的角脉冲试验的简单示例

这段代码实现了一个简单的卷积计算，输入数据从文件中读取。首先读入两个整数n和m，分别表示两个向量的长度，然后读入n个double型数据作为第一个向量x，再读入m个double型数据作为第二个向量y。程序对两个向量进行卷积运算，最后将结果输出到文件中。其中，卷积运算使用了快速傅里叶变换（FFT）算法来加速计算。

#include <iostream>
#include <fstream>
#include <vector>
#include <complex>
#include <cmath>

using namespace std;

const double pi = 3.14159265358979323846;

vector<complex<double>> fft(const vector<complex<double>> &x)
{
    int n = x.size();
    if (n == 1) return vector<complex<double>> {x[0]};
    vector<complex<double>> x0, x1;
    for (int i = 0; i < n; i++)
        (i % 2 == 0 ? x0 : x1).push_back(x[i]);
    auto y0 = fft(x0), y1 = fft(x1);
    vector<complex<double>> y(n);
    for (int i = 0; i < n / 2; i++)
    {
        complex<double> t = polar(1.0, -2 * pi * i / n) * y1[i];
        y[i] = y0[i] + t;
        y[i + n / 2] = y0[i] - t;
    }
    return y;
}

vector<double> conv(const vector<double> &x, const vector<double> &y)
{
    int n = x.size(), m = y.size();
    vector<complex<double>> a(n), b(m);
    for (int i = 0; i < n; i++) a[i] = complex<double>(x[i], 0);
    for (int i = 0; i < m; i++) b[i] = complex<double>(y[i], 0);
    int k = 1;
    while (k < n + m) k *= 2;
    a.resize(k), b.resize(k);
    auto c = fft(a), d = fft(b);
    for (int i = 0; i < k; i++) c[i] *= d[i];
    auto e = fft(c);
    vector<double> z(n + m - 1);
    for (int i = 0; i < n + m - 1; i++) z[i] = e[i].real() / k;
    return z;
}

int main()
{
    ifstream fin("input.txt");
    ofstream fout("output.txt");

    int n, m;
    fin >> n >> m;

    vector<double> x(n), y(m);
    for (int i = 0; i < n; i++) fin >> x[i];
    for (int i = 0; i < m; i++) fin >> y[i];

    auto z = conv(x, y);

    for (int i = 0; i < z.size(); i++) fout << z[i] << endl;

    return 0;
}

这段代码实现了两个向量之间的卷积计算，通过离散傅里叶变换（Discrete Fourier Transform, DFT）的方法实现了高效的卷积运算。具体实现流程如下：

1. 定义复数类型和常量 pi（圆周率）；
2. 实现离散傅里叶变换函数 fft，该函数接受一个复数向量 x，使用递归的方法将其分为两部分 x0 和 x1，分别对其进行递归调用 fft，并使用快速傅里叶变换算法（FFT）计算两个向量的 DFT；
3. 将两个向量的 DFT 相乘，得到卷积结果的 DFT，然后对其进行反向离散傅里叶变换，得到卷积结果向量；
4. 实现卷积函数 conv，该函数接受两个向量 x 和 y，将其转换为复数向量 a 和 b，然后使用 fft 计算它们的卷积结果；
5. 从文件 input.txt 中读入两个向量的长度和值，调用 conv 函数计算它们的卷积，将结果写入到文件 output.txt 中。