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 setScaleMap(const QMap<qint32, float>& scaleMap);
 
    //=============================================================================================================
    void setScaleMapFromStrings(const QMap<QString, double>& scaleMap);
 
    //=============================================================================================================
    void clearData();
 
    //=========================================================================================================
    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