channeldatamodel.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef CHANNELDATAMODEL_H
#define CHANNELDATAMODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../../disp_global.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QObject>
#include <QColor>
#include <QMap>
#include <QVector>
#include <QReadWriteLock>
#include <QSharedPointer>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace FIFFLIB {
    class FiffInfo;
}
 
//=============================================================================================================
// DEFINE NAMESPACE DISPLIB
//=============================================================================================================
 
namespace DISPLIB
{
 
//=============================================================================================================
enum class DetrendMode {
    None   = 0,  
    Mean   = 1,  
    Linear = 2   
};
 
//=============================================================================================================
struct ChannelDisplayInfo {
    QString name;           
    QString typeLabel;      
    QColor  color;          
    float   amplitudeMax;   
    bool    bad;            
    bool    isVirtualChannel = false; 
};
 
//=============================================================================================================
class DISPSHARED_EXPORT ChannelDataModel : public QObject
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<ChannelDataModel>       SPtr;
    typedef QSharedPointer<const ChannelDataModel> ConstSPtr;
 
    explicit ChannelDataModel(QObject *parent = nullptr);
 
    //=========================================================================================================
    void init(QSharedPointer<FIFFLIB::FiffInfo> pFiffInfo);
 
    //=========================================================================================================
    void setData(const Eigen::MatrixXd &data, int firstSample = 0);
 
    //=========================================================================================================
    void appendData(const Eigen::MatrixXd &data);
 
    //=========================================================================================================
    //=========================================================================================================
    void clearData();
 
    void setScaleMap(const QMap<qint32, float> &scaleMap);
    void setScaleMapFromStrings(const QMap<QString, double> &scaleMap);
 
    //=========================================================================================================
    void setVirtualChannels(const QVector<ChannelDisplayInfo> &virtualChannels);
 
    //=========================================================================================================
    void setSignalColor(const QColor &color);
 
    //=========================================================================================================
    void setMaxStoredSamples(int n);
 
    //=========================================================================================================
    void setChannelBad(int channelIdx, bool bad);
 
    //=========================================================================================================
    void setRemoveDC(bool remove);
    bool removeDC() const { return m_detrendMode != DetrendMode::None; }
 
    void setDetrendMode(DetrendMode mode);
    DetrendMode detrendMode() const { return m_detrendMode; }
 
    // ── Accessors (all thread-safe read) ──────────────────────────────
 
    int     channelCount()  const;
    int     firstSample()   const;
    int     totalSamples()  const;
    float   sfreq()         const; 
 
    //=========================================================================================================
    ChannelDisplayInfo channelInfo(int channelIdx) const;
 
    //=========================================================================================================
    float channelRms(int channelIdx, int firstSample, int lastSample) const;
 
    //=========================================================================================================
    QVector<float> decimatedVertices(int   channelIdx,
                                     int   firstSample,
                                     int   lastSample,
                                     int   pixelWidth,
                                     int  &vboFirstSample) const;
 
    //=========================================================================================================
    float sampleValueAt(int channelIdx, int sample) const;
 
signals:
    //=========================================================================================================
    void dataChanged();
 
    //=========================================================================================================
    void metaChanged();
 
private:
    void    rebuildDisplayInfo();
    float   amplitudeMaxForChannel(int ch) const;
    QColor  colorForChannel(int ch) const;
    QString typeLabelForChannel(int ch) const;
 
    mutable QReadWriteLock                    m_lock;
 
    QSharedPointer<FIFFLIB::FiffInfo>         m_pFiffInfo;
    QVector<QVector<float>>                   m_channelData;   // [ch][sample]
    int                                       m_firstSample = 0;
    int                                       m_maxStoredSamples = 0; // 0 = unlimited
 
    QMap<qint32, float>                       m_scaleMap;
    QColor                                    m_signalColor { Qt::darkGreen };
 
    QVector<ChannelDisplayInfo>               m_virtualDisplayInfo;
    QVector<ChannelDisplayInfo>               m_displayInfo;  // pre-computed, rebuild on meta change
    DetrendMode                                m_detrendMode = DetrendMode::None;
};
 
} // namespace DISPLIB
 
#endif // CHANNELDATAMODEL_H