sensordatatreeitem.h

Go to the documentation of this file.

//=============================================================================================================
#ifndef DISP3DLIB_SENSORDATATREEITEM_H
#define DISP3DLIB_SENSORDATATREEITEM_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../../../../disp3D_global.h"
#include "../common/abstracttreeitem.h"
#include "../measurement/measurementtreeitem.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QPointer>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace FIFFLIB {
    class FiffEvoked;
    class FiffInfo;
}
 
namespace MNELIB{
    class MNEBemSurface;
}
 
namespace Qt3DCore {
    class QTransform;
}
 
//=============================================================================================================
// DEFINE NAMESPACE DISP3DLIB
//=============================================================================================================
 
namespace DISP3DLIB {
 
//=============================================================================================================
// DISP3DLIB FORWARD DECLARATIONS
//=============================================================================================================
 
class RtSensorDataController;
class GpuInterpolationItem;
class AbstractMeshTreeItem;
 
//=============================================================================================================
class DISP3DSHARED_EXPORT SensorDataTreeItem : public AbstractTreeItem
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<SensorDataTreeItem> SPtr;            
    typedef QSharedPointer<const SensorDataTreeItem> ConstSPtr; 
    //=========================================================================================================
    explicit SensorDataTreeItem(int iType = Data3DTreeModelItemTypes::SensorDataItem,
                                const QString& text = "Sensor Data",
                                bool bUseGPU = false);
 
    //=========================================================================================================
    virtual ~SensorDataTreeItem();
 
    //=========================================================================================================
    virtual void initData(const MNELIB::MNEBemSurface& tBemSurface,
                          const FIFFLIB::FiffInfo &tFiffInfo,
                          const QString& sSensorType,
                          Qt3DCore::QEntity *p3DEntityParent);
 
    //=========================================================================================================
    virtual void addData(const Eigen::MatrixXd& tSensorData);
 
    //=========================================================================================================
    virtual inline bool isDataInit() const;
 
    //=========================================================================================================
    virtual void setLoopState(bool bState);
 
    //=========================================================================================================
    virtual void setStreamingState(bool bState);
 
    //=========================================================================================================
    virtual void setTimeInterval(int iMSec);
 
    //=========================================================================================================
    virtual void setNumberAverages(int iNumberAverages);
 
    //=========================================================================================================
    virtual void setColormapType(const QString& sColormap);
 
    //=========================================================================================================
    virtual void setThresholds(const QVector3D& vecThresholds);
    
    //=========================================================================================================
    virtual void setCancelDistance(double dCancelDist);
    
    //=========================================================================================================
    virtual void setInterpolationFunction(const QString &sInterpolationFunction);
 
    //=========================================================================================================
    virtual void setSFreq(const double dSFreq);
 
    //=========================================================================================================
    virtual void setBadChannels(const FIFFLIB::FiffInfo& info);
 
    //=========================================================================================================
    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:
    //=========================================================================================================
    virtual void initItem() override;
 
    //=========================================================================================================
    virtual void onNewInterpolationMatrixAvailable(QSharedPointer<Eigen::SparseMatrix<float> > pMatInterpolationMatrixLeftHemi);
 
    //=========================================================================================================
    void virtual onNewRtRawDataAvailable(const Eigen::VectorXd &vecDataVector);
 
    //=========================================================================================================
    virtual void onNewRtSmoothedDataAvailable(const Eigen::MatrixX4f &matColorMatrix);
 
    //=========================================================================================================
    virtual void onStreamingStateChanged(const Qt::CheckState& checkState);
 
    //=========================================================================================================
    virtual void onColormapTypeChanged(const QVariant& sColormapType);
 
    //=========================================================================================================
    virtual void onTimeIntervalChanged(const QVariant &iMSec);
 
    //=========================================================================================================
    virtual void onDataThresholdChanged(const QVariant &vecThresholds);
 
    //=========================================================================================================
    virtual void onLoopStateChanged(const Qt::CheckState& checkState);
 
    //=========================================================================================================
    virtual void onNumberAveragesChanged(const QVariant& iNumAvr);
 
    //=========================================================================================================
    virtual void onCancelDistanceChanged(const QVariant& dCancelDist);
 
    //=========================================================================================================
    virtual void onInterpolationFunctionChanged(const QVariant& sInterpolationFunction);
 
    bool                                m_bIsDataInit;                     
    bool                                m_bUseGPU;                         
    QVector<int>                        m_iUsedSensors;                    
    QVector<int>                        m_iSensorsBad;                     
    QPointer<RtSensorDataController>    m_pSensorRtDataWorkController;     
    QPointer<GpuInterpolationItem>      m_pInterpolationItemGPU;           
    QPointer<AbstractMeshTreeItem>      m_pInterpolationItemCPU;           
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
 
inline bool SensorDataTreeItem::isDataInit() const
{
    return m_bIsDataInit;
}
} // namespace DISP3DLIB
 
#endif // DISP3DLIB_SENSORDATATREEITEM_H