rtinvop.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================

#include "rtinvop.h"

#include <fiff/fiff_info.h>

#include <mne/mne_forwardsolution.h>
#include <mne/mne_inverse_operator.h>

//=============================================================================================================
// QT INCLUDES
//=============================================================================================================

#include <QDebug>

//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================

#include <Eigen/Core>

//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================

using namespace RTPROCESSINGLIB;
using namespace Eigen;
using namespace MNELIB;
using namespace FIFFLIB;

//=============================================================================================================
// DEFINE MEMBER METHODS RtInvOpWorker
//=============================================================================================================

void RtInvOpWorker::doWork(const RtInvOpInput &inputData)
{
    if(this->thread()->isInterruptionRequested()) {
        return;
    }

    // Restrict forward solution as necessary for MEG
    MNEForwardSolution forwardMeg = inputData.pFwd->pick_types(true, false);

    MNEInverseOperator invOpMeg(*inputData.pFiffInfo.data(),
                                forwardMeg,
                                inputData.noiseCov,
                                0.2f,
                                0.8f);

    emit resultReady(invOpMeg);
}

//=============================================================================================================
// DEFINE MEMBER METHODS RtInvOp
//=============================================================================================================

RtInvOp::RtInvOp(FiffInfo::SPtr &p_pFiffInfo,
                 MNEForwardSolution::SPtr &p_pFwd,
                 QObject *parent)
: QObject(parent)
, m_pFiffInfo(p_pFiffInfo)
, m_pFwd(p_pFwd)
{
    RtInvOpWorker *worker = new RtInvOpWorker;
    worker->moveToThread(&m_workerThread);

    connect(&m_workerThread, &QThread::finished,
            worker, &QObject::deleteLater);

    connect(this, &RtInvOp::operate,
            worker, &RtInvOpWorker::doWork);

    connect(worker, &RtInvOpWorker::resultReady,
            this, &RtInvOp::handleResults);

    m_workerThread.start();

    qRegisterMetaType<RtInvOpInput>("RtInvOpInput");
}

//=============================================================================================================

RtInvOp::~RtInvOp()
{
    stop();
}

//=============================================================================================================

void RtInvOp::append(const FIFFLIB::FiffCov &noiseCov)
{
    RtInvOpInput inputData;
    inputData.noiseCov = noiseCov;
    inputData.pFiffInfo = m_pFiffInfo;
    inputData.pFwd = m_pFwd;

    emit operate(inputData);
}

//=============================================================================================================

void RtInvOp::setFwdSolution(QSharedPointer<MNELIB::MNEForwardSolution> pFwd)
{
    m_pFwd = pFwd;
}

//=============================================================================================================

void RtInvOp::handleResults(const MNELIB::MNEInverseOperator& invOp)
{
    emit invOperatorCalculated(invOp);
}

//=============================================================================================================

void RtInvOp::restart()
{
    stop();

    RtInvOpWorker *worker = new RtInvOpWorker;
    worker->moveToThread(&m_workerThread);

    connect(&m_workerThread, &QThread::finished,
            worker, &QObject::deleteLater);

    connect(this, &RtInvOp::operate,
            worker, &RtInvOpWorker::doWork);

    connect(worker, &RtInvOpWorker::resultReady,
            this, &RtInvOp::handleResults);

    m_workerThread.start();
}

//=============================================================================================================

void RtInvOp::stop()
{
    m_workerThread.requestInterruption();
    m_workerThread.quit();
    m_workerThread.wait();
}