#include "CombFilterWorker.h"
#include "qdir.h"
#include "qfileinfo.h"
CombFilterWorker::CombFilterWorker(QString oStrDir, int iRow, QString oStrFileName, qint64 iTimestampStart, qint64 iTimestampEnd, QObject* parent)
: QObject{parent}, oStrDir(oStrDir), iRow(iRow), oStrFileName(oStrFileName), iTimestampStart(iTimestampStart), iTimestampEnd(iTimestampEnd)
{
oHead = PF::readHead(oStrFileName);
this->setAutoDelete(true);
}
void CombFilterWorker::run()
{
QElapsedTimer time;
time.start();
qDebugV0() << oStrFileName;
QFile oFileRaw(oStrFileName);
if(!oFileRaw.open(QIODevice::ReadOnly) )
{
emit sigMsg(oStrFileName + " 打开失败!");
return;
}
//获取新的文件名,梳妆滤波后的文件
QString oStrHead = PF::readHeadString(oStrFileName);
QStringList aoStrHead = oStrHead.split(',', Qt::KeepEmptyParts);
aoStrHead.replace(RAW_HEADER_DATA_CNT, QString::number(oHead.uiFS * (iTimestampEnd - iTimestampStart)));
aoStrHead.replace(RAW_HEADER_START_TIME, QString::number(iTimestampStart));
aoStrHead.replace(RAW_HEADER_SAMP_TIME, QString::number(iTimestampEnd - iTimestampStart));
QString oStrHeadNew = aoStrHead.join(",");
int iIndexStart = oStrFileName.lastIndexOf("_");
int iIndexEnd = oStrFileName.lastIndexOf(".");
QString oStrIndex = oStrFileName.mid(iIndexStart + 1, iIndexEnd - iIndexStart - 1);
QString oStrBaseName = QString("%1_L(%2)_S(%3)_D(%4)_CH%5_%6_%7_%8.dat")
.arg(oHead.oStrTaskName)
.arg(oHead.oStrLine)
.arg(oHead.oStrSite)
.arg(oHead.uiDev)
.arg(oHead.uiCh)
.arg(oHead.oStrRAW_COMPONENT)
.arg(QDateTime::fromSecsSinceEpoch(iTimestampStart).toString("yyyyMMddHHmmss"))
.arg(oStrIndex);
QString oStrFileNameNew = oStrDir + QDir::separator() + QDir::separator() + oStrBaseName;
qDebugV0() << oStrFileNameNew;
//写入首行信息
QFile oFileNew(oStrFileNameNew);
if(!oFileNew.open(QIODevice::WriteOnly | QIODevice::Truncate))
{
emit sigMsg(oStrFileNameNew + " 打开失败!");
return;
}
oFileNew.write(oStrHeadNew.toUtf8(), LENGTH_178_BYTE);
QDataStream oStreamNew(&oFileNew);
t_CMF cmf;
/*初始化*/
comb_filter_init(&cmf, -0.22059817649696845, CMF_TYPE_STOP_BANDS);
QDataStream oStreamRaw(&oFileRaw);
oStreamRaw.setByteOrder(QDataStream::BigEndian);
oStreamRaw.setFloatingPointPrecision(QDataStream::SinglePrecision);
oStreamRaw.skipRawData(LENGTH_178_BYTE);//作者是靠占位符来占据首行的。
for(int i = 0; i < (iTimestampStart - oHead.iStartTimestamp); ++i)
{
oStreamRaw.skipRawData(LENGTH_4_BYTE * oHead.uiFS);
}
oStreamNew.setByteOrder(QDataStream::BigEndian);
oStreamNew.setFloatingPointPrecision(QDataStream::SinglePrecision);
quint32 uiPercentage = PROGRESS_STEP;
oStreamRaw.startTransaction();
oStreamNew.startTransaction();
double in = 0.00f;
double out = 0.00f;
qDebugV0() << oHead.uiFS* iTimestampStart << " ~ " << oHead.uiFS* iTimestampEnd;
for(qint64 i = 0; i < (iTimestampEnd - iTimestampStart); ++i)
{
for(int j = 0; j < oHead.uiFS; ++j)
{
oStreamRaw >> in;
out = this->comb_filter(&cmf, in);
oStreamNew << out;
}
float fProgress = ((float)(i) / ((float)(iTimestampEnd - iTimestampStart))) * 100;
if(fProgress >= uiPercentage)
{
emit sigProgress(iRow, "处理中~ ", fProgress);
uiPercentage = uiPercentage + PROGRESS_STEP;
}
}
oStreamRaw.commitTransaction();
oFileRaw.close();
oStreamNew.commitTransaction();
oFileNew.flush();
oFileNew.close();
emit sigProgress(iRow, "梳妆滤波完毕!", 100);
qint64 milsec = time.elapsed();
emit sigMsg(QString("行号:%1\t加载采样点个数:%2\t耗时:%3s\t梳状滤波")
.arg(iRow + 1)
.arg(oHead.uiFS * (iTimestampEnd - iTimestampStart))
.arg(QString::number(milsec / 1000.00, 'f', 3)));
}
void CombFilterWorker::comb_filter_init(t_CMF* pCmf, double rho, E_CMF_TYPE type)
{
memset(pCmf, 0, sizeof(t_CMF));
pCmf->index = CMF_RANK - 1;
pCmf->r = rho;
pCmf->type = type;
if(type == CMF_TYPE_STOP_BANDS)
{
pCmf->b = (1 + rho) / 2;
}
else
{
pCmf->b = (1 - rho) / 2;
}
}
float CombFilterWorker::comb_filter(t_CMF* pCmf, double xn)
{
double yn = 0;
int n_N;
int i = 0;
n_N = pCmf->index;
if(pCmf->type == CMF_TYPE_STOP_BANDS)
{
/*y[n] = bx[n]-bx[n-N]+ry[n-N]*/
yn = pCmf->b * (xn - pCmf->x[n_N]) + pCmf->r * pCmf->y[n_N];
}
else
{
/*y[n] = bx[n]+bx[n-N]+ry[n-N]*/
yn = pCmf->b * (xn + pCmf->x[n_N]) + pCmf->r * pCmf->y[n_N];
}
/*存储yn为下次迭代准备*/
pCmf->y[n_N] = yn;
pCmf->x[n_N] = xn;
if(pCmf->index == 0)
{
pCmf->index = CMF_RANK - 1;
}
else
{
pCmf->index--;
}
return yn;
}
#ifndef COMBFILTERWORKER_H
#define COMBFILTERWORKER_H
#include "PF.h"
#include "qrunnable.h"
#include <QObject>
#include <stdio.h>
#include <math.h>
#include <string.h>
/*长度应为阶数+1*/
#define CMF_RANK 60
#define RHO -0.22059817649696845
typedef enum _E_CMF_TYPE
{
CMF_TYPE_STOP_BANDS,
CMF_TYPE_PASS_BANDS
} E_CMF_TYPE;
/*定义移动平均寄存器历史状态*/
typedef struct _t_CMF
{
float x[CMF_RANK];
float y[CMF_RANK];
float b;
float r;
E_CMF_TYPE type;
int index;
} t_CMF;
class CombFilterWorker : public QObject, public QRunnable
{
Q_OBJECT
public:
explicit CombFilterWorker(QString oStrDir, int iRow, QString oStrFileName, qint64 iTimestampStart, qint64 iTimestampEnd, QObject* parent = nullptr);
void run()override;
void comb_filter_init(t_CMF* pCmf, double rho, E_CMF_TYPE type);
float comb_filter(t_CMF* pCmf, double xn);
signals:
void sigMsg(QString);
void sigProgress(int iRow, QString, float fProgress);
private:
QString oStrDir;
int iRow;
QString oStrFileName;
qint64 iTimestampStart;
qint64 iTimestampEnd;
HEAD oHead;
};
#endif // COMBFILTERWORKER_H
抄袭了别个的代码:https://www.cnblogs.com/embInn/p/12940708.html