QCustomPlot 取包络

QCustomPlot是一个强大的Qt绘图库，可以方便地进行数据可视化。如果您已经使用QCustomPlot绘制了信号数据，那么提取包络的方法与一般算法相同。

以下是一个简单的示例代码，用于在QCustomPlot中绘制信号数据并提取包络：

#include <QApplication>
#include <QCustomPlot>
#include <vector>
#include <cmath>

using namespace std;

vector<double> get_envelope(vector<double> signal, int window_size) {
    vector<double> envelope(signal.size() - window_size + 1);
    for (int i = 0; i < envelope.size(); i++) {
        double sum = 0;
        for (int j = 0; j < window_size; j++) {
            sum += pow(signal[i + j], 2);
        }
        envelope[i] = sqrt(sum / window_size);
    }
    return envelope;
}

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);

    // 创建QCustomPlot对象
    QCustomPlot *customPlot = new QCustomPlot();

    // 示例信号数据
    vector<double> signal = {0.1, 0.2, 0.3, 0.2, 0.1, -0.1, -0.2, -0.3, -0.2, -0.1};

    // 创建两个graph对象，一个用于绘制信号数据，一个用于绘制包络数据
    QCPGraph *signalGraph = customPlot->addGraph();
    QCPGraph *envelopeGraph = customPlot->addGraph();

    // 设置信号数据的颜色、线宽和线型
    signalGraph->setPen(QPen(Qt::blue));
    signalGraph->setLineStyle(QCPGraph::lsLine);
    signalGraph->setLineWidth(2);

    // 设置包络数据的颜色、线宽和线型
    envelopeGraph->setPen(QPen(Qt::red));
    envelopeGraph->setLineStyle(QCPGraph::lsLine);
    envelopeGraph->setLineWidth(2);

    // 设置X轴和Y轴范围
    customPlot->xAxis->setRange(0, signal.size());
    customPlot->yAxis->setRange(-0.5, 0.5);

    // 绘制信号数据
    signalGraph->setData(QVector<double>::fromStdVector(signal));

    // 提取包络数据并绘制
    vector<double> envelope = get_envelope(signal, 5);
    envelopeGraph->setData(QVector<double>::fromStdVector(envelope));

    // 显示图形
    customPlot->replot();
    customPlot->show();

    return a.exec();
}

在该示例代码中，首先创建了一个QCustomPlot对象，然后创建了两个graph对象，一个用于绘制信号数据，一个用于绘制包络数据。接着，设置了信号数据和包络数据的颜色、线宽和线型，并设置了X轴和Y轴的范围。绘制信号数据时，直接调用setData方法即可。提取包络数据时，调用get_envelope函数进行计算，并将结果传递给envelopeGraph的setData方法。最后，调用replot和show方法显示图形。