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