rt_noise.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef RT_NOISE_RTPROCESSING_H
#define RT_NOISE_RTPROCESSING_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../dsp_global.h"
 
#include <fiff/fiff_info.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QThread>
#include <QMutex>
#include <QSharedPointer>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// DEFINE NAMESPACE RTPROCESSINGLIB
//=============================================================================================================
 
namespace RTPROCESSINGLIB
{
 
//=============================================================================================================
class DSPSHARED_EXPORT RtNoiseWorker : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    explicit RtNoiseWorker(qint32 iFftLength,
                           FIFFLIB::FiffInfo::SPtr pFiffInfo,
                           qint32 iDataLength);
 
    //=========================================================================================================
    void doWork(const Eigen::MatrixXd& matData);
 
signals:
    void resultReady(const Eigen::MatrixXd& matSpecData);
 
private:
    static QVector<float> hanning(int N, short itype);
 
    qint32              m_iFftLength;       
    double              m_dFs;              
    qint32              m_iDataLength;      
    QVector<float>      m_fWin;             
 
    int                 m_iNumOfBlocks = 0; 
    int                 m_iBlockSize = 0;   
    int                 m_iSensors = 0;     
    int                 m_iBlockIndex = 0;  
    bool                m_bFirstBlock = true;
    Eigen::MatrixXd     m_matCircBuf;       
};
 
//=============================================================================================================
class DSPSHARED_EXPORT RtNoise : public QObject
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<RtNoise> SPtr;             
    typedef QSharedPointer<const RtNoise> ConstSPtr;  
 
    //=========================================================================================================
    explicit RtNoise(qint32 iFftLength,
                     FIFFLIB::FiffInfo::SPtr pFiffInfo,
                     qint32 iDataLength,
                     QObject *parent = nullptr);
 
    //=========================================================================================================
    ~RtNoise();
 
    //=========================================================================================================
    void append(const Eigen::MatrixXd& matData);
 
    //=========================================================================================================
    bool isRunning() const;
 
    //=========================================================================================================
    bool start();
 
    //=========================================================================================================
    bool stop();
 
    //=========================================================================================================
    bool wait(unsigned long time = ULONG_MAX);
 
signals:
    void SpecCalculated(const Eigen::MatrixXd& matSpecData);

    void operate(const Eigen::MatrixXd& matData);
 
private:
    QThread             m_workerThread;     
    bool                m_bIsRunning = false;
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
 
} // NAMESPACE
 
#endif // RT_NOISE_RTPROCESSING_H