correlation.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "correlation.h"
#include "../network/networknode.h"
#include "../network/networkedge.h"
#include "../network/network.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QDebug>
#include <QElapsedTimer>
#include <QtConcurrent>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <unsupported/Eigen/FFT>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace CONNECTIVITYLIB;
using namespace Eigen;
 
//=============================================================================================================
// DEFINE GLOBAL METHODS
//=============================================================================================================
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
Correlation::Correlation()
{
}
 
//=============================================================================================================
 
Network Correlation::calculate(ConnectivitySettings& connectivitySettings)
{
    //    QElapsedTimer timer;
    //    qint64 iTime = 0;
    //    timer.start();
 
    Network finalNetwork("COR");
 
    if (connectivitySettings.isEmpty()) {
        qDebug() << "Correlation::calculate - Input data is empty";
        return finalNetwork;
    }
 
    finalNetwork.setSamplingFrequency(connectivitySettings.getSamplingFrequency());
 
    //Create nodes
    int rows = connectivitySettings.at(0).matData.rows();
    RowVectorXf rowVert = RowVectorXf::Zero(3);
 
    for (int i = 0; i < rows; ++i) {
        rowVert = RowVectorXf::Zero(3);
 
        if (connectivitySettings.getNodePositions().rows() != 0 && i < connectivitySettings.getNodePositions().rows()) {
            rowVert(0) = connectivitySettings.getNodePositions().row(i)(0);
            rowVert(1) = connectivitySettings.getNodePositions().row(i)(1);
            rowVert(2) = connectivitySettings.getNodePositions().row(i)(2);
        }
 
        finalNetwork.append(NetworkNode::SPtr(new NetworkNode(i, rowVert)));
    }
 
    //    iTime = timer.elapsed();
    //    qWarning() << "Preparation" << iTime;
    //    timer.restart();
 
    // Calculate connectivity matrix over epochs and average afterwards
    //    double dScalingStep = 1.0/matDataList.size();
    //    dataTemp.matInputData = dScalingStep * (i+1) * matDataList.at(i);
 
    QFuture<MatrixXd> resultMat = QtConcurrent::mappedReduced(connectivitySettings.getTrialData(),
                                                              compute,
                                                              reduce);
    resultMat.waitForFinished();
 
    MatrixXd matDist = resultMat.result() / connectivitySettings.size();
 
    //    MatrixXd matDist;
 
    //    for(int i = 0; i < connectivitySettings.getTrialData().size(); ++i) {
    //        reduce(matDist, compute(connectivitySettings.getTrialData().at(i)));
    //    }
 
    //    matDist /= connectivitySettings.size();
 
    //    iTime = timer.elapsed();
    //    qWarning() << "ComputeSpectraPSDCSD" << iTime;
    //    timer.restart();
 
    //Add edges to network
    MatrixXd matWeight(1, 1);
    QSharedPointer<NetworkEdge> pEdge;
    int j;
 
    for (int i = 0; i < matDist.rows(); ++i) {
        for (j = i; j < matDist.cols(); ++j) {
            matWeight << matDist(i, j);
 
            pEdge = QSharedPointer<NetworkEdge>(new NetworkEdge(i, j, matWeight));
 
            finalNetwork.getNodeAt(i)->append(pEdge);
            finalNetwork.getNodeAt(j)->append(pEdge);
            finalNetwork.append(pEdge);
        }
    }
 
    //    iTime = timer.elapsed();
    //    qWarning() << "Compute" << iTime;
    //    timer.restart();
 
    return finalNetwork;
}
 
//=============================================================================================================
 
MatrixXd Correlation::compute(const ConnectivitySettings::IntermediateTrialData& inputData)
{
    const MatrixXd centered = inputData.matData.colwise() - inputData.matData.rowwise().mean();
    const VectorXd norms = centered.rowwise().norm();
    MatrixXd matDist = centered * centered.transpose();
 
    for (int i = 0; i < matDist.rows(); ++i) {
        for (int j = 0; j < matDist.cols(); ++j) {
            const double denom = norms(i) * norms(j);
            matDist(i, j) = denom > 0.0 ? matDist(i, j) / denom : 0.0;
        }
    }
 
    return matDist;
}
 
//=============================================================================================================
 
void Correlation::reduce(MatrixXd& resultData,
                         const MatrixXd& data)
{
    if (resultData.rows() != data.rows() || resultData.cols() != data.cols()) {
        resultData.resize(data.rows(), data.cols());
        resultData.setZero();
    }
 
    resultData += data;
}