evokedsetmodel.h

Go to the documentation of this file.

//=============================================================================================================
#ifndef EVOKEDSETMODEL_H
#define EVOKEDSETMODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../../disp_global.h"
 
#include <rtprocessing/helpers/filterkernel.h>
#include <fiff/fiff_types.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QAbstractTableModel>
#include <QSharedPointer>
#include <QColor>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
#include <Eigen/SparseCore>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace FIFFLIB {
    class FiffEvokedSet;
    class FiffProj;
}
 
//=============================================================================================================
// DEFINE NAMESPACE DISPLIB
//=============================================================================================================
 
namespace DISPLIB
{
 
//=============================================================================================================
// DISPLIB FORWARD DECLARATIONS
//=============================================================================================================
 
namespace EvokedSetModelRoles {
    enum ItemRole{GetAverageData = Qt::UserRole + 1020};
}
 
//=============================================================================================================
// DEFINE TYPEDEFS
//=============================================================================================================
 
typedef Eigen::Matrix<double,Eigen::Dynamic,Eigen::Dynamic,Eigen::RowMajor> MatrixXdR;
typedef QPair<const double*,qint32> RowVectorPair;
typedef QPair<QString, Eigen::RowVectorXd> AvrTypeRowVector;
typedef QPair<QString, DISPLIB::RowVectorPair> AvrTypeRowVectorPair;
 
//=============================================================================================================
class DISPSHARED_EXPORT EvokedSetModel : public QAbstractTableModel
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<EvokedSetModel> SPtr;              
    typedef QSharedPointer<const EvokedSetModel> ConstSPtr;   
    //=========================================================================================================
    EvokedSetModel(QObject *parent = 0);
 
    ~EvokedSetModel();
 
    //=========================================================================================================
    bool isInit() const;
 
    //=========================================================================================================
    qint32 getNumSamples() const;
 
    //=========================================================================================================
    virtual int rowCount(const QModelIndex &parent = QModelIndex()) const ;
 
    //=========================================================================================================
    virtual int columnCount(const QModelIndex &parent = QModelIndex()) const;
 
    //=========================================================================================================
    QVariant data(int row,
                  int column,
                  int role = Qt::DisplayRole) const;
 
    //=========================================================================================================
    virtual QVariant data(const QModelIndex &index,
                          int role = Qt::DisplayRole) const;
 
    //=========================================================================================================
    virtual QVariant headerData(int section,
                                Qt::Orientation orientation,
                                int role = Qt::DisplayRole) const;
 
    void init();
 
    //=========================================================================================================
    void setEvokedSet(QSharedPointer<FIFFLIB::FiffEvokedSet> pEvokedSet);
 
    //=========================================================================================================
    QSharedPointer<FIFFLIB::FiffEvokedSet> getEvokedSet();
 
    //=========================================================================================================
    void updateData();
 
    //=========================================================================================================
    QSharedPointer<QMap<QString, QColor> > getAverageColor() const;
 
    //=========================================================================================================
    QSharedPointer<QMap<QString, bool> > getAverageActivation() const;
 
    //=========================================================================================================
    void setAverageColor(const QSharedPointer<QMap<QString, QColor> > qMapAverageColor);
 
    //=========================================================================================================
    void setAverageActivation(const QSharedPointer<QMap<QString, bool> > qMapAverageActivation);
 
    //=========================================================================================================
    FIFFLIB::fiff_int_t getKind(qint32 row) const;
 
    //=========================================================================================================
    bool getIsChannelBad(qint32 row) const;
 
    //=========================================================================================================
    FIFFLIB::fiff_int_t getUnit(qint32 row) const;
 
    //=========================================================================================================
    FIFFLIB::fiff_int_t getCoil(qint32 row) const;
 
    //=========================================================================================================
    const QMap<qint32,qint32>& getIdxSelMap() const;
 
    //=========================================================================================================
    int getNumberOfTimeSpacers() const;
 
    //=========================================================================================================
    QPair<QVariant,QVariant> getBaselineInfo() const;
 
    //=========================================================================================================
    int getNumAverages() const;
 
    //=========================================================================================================
    qint32 getNumPreStimSamples() const;
 
    //=========================================================================================================
    float getSamplingFrequency() const;
 
    //=========================================================================================================
    void selectRows(const QList<qint32> &selection);
 
    //=========================================================================================================
    void resetSelection();
 
    //=========================================================================================================
    qint32 numVLines() const;
 
    //=========================================================================================================
    bool isFreezed() const;
 
    //=========================================================================================================
    void updateProjection(const QList<FIFFLIB::FiffProj>& projs);
 
    //=========================================================================================================
    void updateCompensator(int to);
 
    //=========================================================================================================
    void toggleFreeze();
 
private:
    QSharedPointer<FIFFLIB::FiffEvokedSet>  m_pEvokedSet;                   
    QMap<qint32,qint32>                     m_qMapIdxRowSelection;          
    QSharedPointer<QMap<QString, QColor> >  m_qMapAverageColor;             
    QSharedPointer<QMap<QString, bool> >    m_qMapAverageActivation;        
    QSharedPointer<QMap<QString, QColor> >  m_qMapAverageColorOld;          
    QSharedPointer<QMap<QString, bool> >    m_qMapAverageActivationOld;     
    QList<Eigen::MatrixXd>                  m_matData;                      
    QList<Eigen::MatrixXd>                  m_matDataFreeze;                
    QStringList                             m_lAvrTypes;                    
    Eigen::MatrixXd                         m_matProj;                      
    Eigen::MatrixXd                         m_matComp;                      
    Eigen::SparseMatrix<double>             m_matSparseProjCompMult;        
    Eigen::SparseMatrix<double>             m_matSparseProjMult;            
    Eigen::SparseMatrix<double>             m_matSparseCompMult;            
    Eigen::RowVectorXi                      m_vecBadIdcs;                   
    QPair<QVariant,QVariant>                m_pairBaseline;                 
    bool                                    m_bIsInit;                      
    bool                                    m_bIsFreezed;                   
    bool                                    m_bProjActivated;               
    bool                                    m_bCompActivated;               
    float                                   m_fSps;                         
signals:
    //=========================================================================================================
    void newSelection(QList<qint32> selection);
 
    //=========================================================================================================
    void newAverageColorMap(const QSharedPointer<QMap<QString, QColor> > qMapAverageColor);
 
    //=========================================================================================================
    void newAverageActivationMap(const QSharedPointer<QMap<QString, bool> > qMapAverageActivation);
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
} // NAMESPACE
 
#ifndef metatype_listrowvectorxd
#define metatype_listrowvectorxd
Q_DECLARE_METATYPE(DISPLIB::AvrTypeRowVector);    
#endif
 
#ifndef metatype_listrowvectorpair
#define metatype_listrowvectorpair
Q_DECLARE_METATYPE(DISPLIB::AvrTypeRowVectorPair);
#endif
 
#endif // EVOKEDSETMODEL_H