sensorsettreeitem.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "sensorsettreeitem.h"
#include "sensorsurfacetreeitem.h"
#include "sensorpositiontreeitem.h"
 
#include "../../3dhelpers/renderable3Dentity.h"
 
#include <mne/mne_bem.h>
#include <fiff/fiff_ch_info.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace FSLIB;
using namespace MNELIB;
using namespace DISP3DLIB;
using namespace FIFFLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
SensorSetTreeItem::SensorSetTreeItem(int iType, const QString& text)
: AbstractTreeItem(iType, text)
{
    initItem();
}
 
//=============================================================================================================
 
void SensorSetTreeItem::initItem()
{
    this->setEditable(false);
    this->setCheckable(true);
    this->setCheckState(Qt::Checked);
    this->setToolTip("Sensor item");
}
 
//=============================================================================================================
 
void SensorSetTreeItem::addData(const MNEBem &tSensor,
                                const QList<FiffChInfo>& lChInfo,
                                const QString& sDataType,
                                const QStringList& bads,
                                Qt3DCore::QEntity* p3DEntityParent)
{
    if(!m_pRenderable3DEntity) {
        m_pRenderable3DEntity = new Renderable3DEntity(p3DEntityParent);
    }
 
    //Generate sensor surfaces as childs
    for(int i = 0; i < tSensor.size(); ++i) {
        SensorSurfaceTreeItem* pSurfItem = new SensorSurfaceTreeItem(m_pRenderable3DEntity, Data3DTreeModelItemTypes::SensorSurfaceItem);
        pSurfItem->addData(tSensor[i]);
 
        QList<QStandardItem*> list;
        list << pSurfItem;
        list << new QStandardItem(pSurfItem->toolTip());
        this->appendRow(list);
    }
 
    //Sort MEG channel types
    QList<FiffChInfo> lChInfoGrad;
    QList<FiffChInfo> lChInfoMag;
    QList<FiffChInfo> lChInfoEEG;
 
    for(int i = 0; i < lChInfo.size(); ++i) {
        if(lChInfo.at(i).unit == FIFF_UNIT_T_M) {
            lChInfoGrad << lChInfo.at(i);
        } else if(lChInfo.at(i).unit == FIFF_UNIT_T) {
            lChInfoMag << lChInfo.at(i);
        } else if(lChInfo.at(i).kind == FIFFV_EEG_CH) {
            lChInfoEEG << lChInfo.at(i);
        }
    }
 
    //Add sensor locations as child items
    if(!lChInfoGrad.isEmpty() && sDataType == "MEG") {
        SensorPositionTreeItem* pSensorPosItem = new SensorPositionTreeItem(m_pRenderable3DEntity, Data3DTreeModelItemTypes::SensorPositionItem, "Grad");
        pSensorPosItem->addData(lChInfoGrad, "MEG", bads);
 
        QList<QStandardItem*> list;
        list << pSensorPosItem;
        list << new QStandardItem(pSensorPosItem->toolTip());
        this->appendRow(list);
    }
 
    if(!lChInfoMag.isEmpty() && sDataType == "MEG") {
        SensorPositionTreeItem* pSensorPosItem = new SensorPositionTreeItem(m_pRenderable3DEntity, Data3DTreeModelItemTypes::SensorPositionItem, "Mag");
        pSensorPosItem->addData(lChInfoMag, "MEG", bads);
 
        QList<QStandardItem*> list;
        list << pSensorPosItem;
        list << new QStandardItem(pSensorPosItem->toolTip());
        this->appendRow(list);
    }
 
    if(!lChInfoEEG.isEmpty() && sDataType == "EEG") {
        SensorPositionTreeItem* pSensorPosItem = new SensorPositionTreeItem(m_pRenderable3DEntity, Data3DTreeModelItemTypes::SensorPositionItem, "EEG");
        pSensorPosItem->addData(lChInfoEEG, "EEG", bads);
 
        QList<QStandardItem*> list;
        list << pSensorPosItem;
        list << new QStandardItem(pSensorPosItem->toolTip());
        this->appendRow(list);
    }
}
 
//=============================================================================================================
 
void SensorSetTreeItem::setTransform(const Qt3DCore::QTransform& transform)
{
    if(m_pRenderable3DEntity) {
        m_pRenderable3DEntity->setTransform(transform);
    }
}
 
//=============================================================================================================
 
void SensorSetTreeItem::setTransform(const FiffCoordTrans& transform, bool bApplyInverse)
{
    if(m_pRenderable3DEntity) {
        m_pRenderable3DEntity->setTransform(transform, bApplyInverse);
    }
}
 
//=============================================================================================================
 
void SensorSetTreeItem::applyTransform(const Qt3DCore::QTransform& transform)
{
    if(m_pRenderable3DEntity) {
        m_pRenderable3DEntity->applyTransform(transform);
    }
}
 
//=============================================================================================================
 
void SensorSetTreeItem::applyTransform(const FiffCoordTrans& transform, bool bApplyInverse)
{
    if(m_pRenderable3DEntity) {
        m_pRenderable3DEntity->applyTransform(transform, bApplyInverse);
    }
}