mnedatatreeitem.h

Go to the documentation of this file.

 //=============================================================================================================
#ifndef DISP3DLIB_MNEDATATREEITEM_H
#define DISP3DLIB_MNEDATATREEITEM_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../../../../disp3D_global.h"
#include "../common/abstracttreeitem.h"
 
#include <fiff/fiff_types.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QPointer>
#include <Qt3DCore/QTransform>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace MNELIB {
    class MNEForwardSolution;
    class MNESourceEstimate;
}
 
namespace FSLIB {
    class SurfaceSet;
    class AnnotationSet;
}
 
namespace Qt3DCore {
    class QEntity;
    class QTransform;
}
 
//=============================================================================================================
// DEFINE NAMESPACE DISP3DLIB
//=============================================================================================================
 
namespace DISP3DLIB
{
 
//=============================================================================================================
// DISP3DLIB FORWARD DECLARATIONS
//=============================================================================================================
 
class RtSourceDataController;
class AbstractMeshTreeItem;
class GpuInterpolationItem;
 
//=============================================================================================================
class DISP3DSHARED_EXPORT MneDataTreeItem : public AbstractTreeItem
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<MneDataTreeItem> SPtr;             
    typedef QSharedPointer<const MneDataTreeItem> ConstSPtr;  
    //=========================================================================================================
    explicit MneDataTreeItem(int iType = Data3DTreeModelItemTypes::MNEDataItem,
                             const QString& text = "MNE",
                             bool bUseGPU = false);
 
    //=========================================================================================================
    ~MneDataTreeItem();
 
    //=========================================================================================================
    void initData(const MNELIB::MNEForwardSolution& tForwardSolution,
                  const FSLIB::SurfaceSet& tSurfSet,
                  const FSLIB::AnnotationSet& tAnnotSet,
                  Qt3DCore::QEntity* p3DEntityParent);
 
    //=========================================================================================================
    void addData(const MNELIB::MNESourceEstimate& tSourceEstimate);
 
    //=========================================================================================================
    inline bool isDataInit() const;
 
    //=========================================================================================================
    void setLoopState(bool state);
 
    //=========================================================================================================
    void setStreamingState(bool state);
 
    //=========================================================================================================
    void setTimeInterval(int iMSec);
 
    //=========================================================================================================
    void setNumberAverages(int iNumberAverages);
 
    //=========================================================================================================
    void setColormapType(const QString& sColormap);
 
    //=========================================================================================================
    void setVisualizationType(const QString& sVisualizationType);
 
    //=========================================================================================================
    void setThresholds(const QVector3D& vecThresholds);
 
    //=========================================================================================================
    //virtual void setCancelDistance(double dCancelDist);
 
    //=========================================================================================================
    virtual void setInterpolationFunction(const QString &sInterpolationFunction);
 
    //=========================================================================================================
    void setSFreq(const double dSFreq);
 
    //=========================================================================================================
    void setAlpha(float fAlpha);
 
    //=========================================================================================================
    virtual void setTransform(const Qt3DCore::QTransform &transform);
 
    //=========================================================================================================
    virtual void setTransform(const FIFFLIB::FiffCoordTrans& transform, bool bApplyInverse = false);
 
    //=========================================================================================================
    virtual void applyTransform(const Qt3DCore::QTransform& transform);
 
    //=========================================================================================================
    virtual void applyTransform(const FIFFLIB::FiffCoordTrans& transform, bool bApplyInverse = false);
 
protected:
    //=========================================================================================================
    void initItem();
 
    //=========================================================================================================
    void onCheckStateWorkerChanged(const Qt::CheckState& checkState);
 
    //=========================================================================================================
    virtual void onNewInterpolationMatrixLeftAvailable(QSharedPointer<Eigen::SparseMatrix<float> > pMatInterpolationMatrixLeftHemi);
 
    //=========================================================================================================
    virtual void onNewInterpolationMatrixRightAvailable(QSharedPointer<Eigen::SparseMatrix<float> > pMatInterpolationMatrixRightHemi);
 
    //=========================================================================================================
    void virtual onNewRtRawData(const Eigen::VectorXd &vecDataVectorLeftHemi,
                                const Eigen::VectorXd &vecDataVectorRightHemi);
 
    //=========================================================================================================
    void onNewRtSmoothedDataAvailable(const Eigen::MatrixX4f &matColorMatrixLeftHemi,
                                      const Eigen::MatrixX4f &matColorMatrixRightHemi);
 
    //=========================================================================================================
    void onColormapTypeChanged(const QVariant& sColormapType);
 
    //=========================================================================================================
    void onTimeIntervalChanged(const QVariant &iMSec);
 
    //=========================================================================================================
    void onDataThresholdChanged(const QVariant &vecThresholds);
 
    //=========================================================================================================
    void onVisualizationTypeChanged(const QVariant& sVisType);
 
    //=========================================================================================================
    void onCheckStateLoopedStateChanged(const Qt::CheckState& checkState);
 
    //=========================================================================================================
    void onNumberAveragesChanged(const QVariant& iNumAvr);
 
    //=========================================================================================================
    //virtual void onCancelDistanceChanged(const QVariant& dCancelDist);
 
    //=========================================================================================================
    virtual void onInterpolationFunctionChanged(const QVariant& sInterpolationFunction);
 
    bool                                m_bIsDataInit;                      
    bool                                m_bUseGPU;                          
    QPointer<RtSourceDataController>    m_pRtSourceDataController;          
    QPointer<AbstractMeshTreeItem>      m_pInterpolationItemLeftCPU;        
    QPointer<GpuInterpolationItem>      m_pInterpolationItemLeftGPU;        
    QPointer<AbstractMeshTreeItem>      m_pInterpolationItemRightCPU;       
    QPointer<GpuInterpolationItem>      m_pInterpolationItemRightGPU;       
signals:
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
 
inline bool MneDataTreeItem::isDataInit() const
{
    return m_bIsDataInit;
}
} //NAMESPACE DISP3DLIB
 
#endif // DISP3DLIB_MNEESTIMATTREEITEM_H