evokedsetmodel.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef EVOKEDSETMODEL_H
#define EVOKEDSETMODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../../disp_global.h"
 
#include <dsp/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;                
    QStringList m_lAvrTypesFreeze;          
 
    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