layoutloader.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "layoutloader.h"
#include <algorithm>
#include <fstream>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QIODevice>
#include <QTextStream>
#include <QFile>
#include <QDebug>
#include <QRegularExpression>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace UTILSLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================s
 
bool LayoutLoader::readAsaElcFile(const QString& path,
                                  QStringList &channelNames,
                                  QList<QVector<float> > &location3D,
                                  QList<QVector<float> > &location2D,
                                  QString &unit)
{
    //Open .elc file
    if(!path.contains(".elc"))
        return false;
 
    QFile file(path);
    if(!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
        qDebug()<<"Error opening elc file";
        return false;
    }
 
    //Start reading from file
    double numberElectrodes;
    QTextStream in(&file);
    bool read2D = false;
 
    while(!in.atEnd())
    {
        QString line = in.readLine();
 
        QStringList fields = line.split(QRegularExpression("\\s+"));
 
        //Delete last element if it is a blank character
        if(fields.at(fields.size()-1) == "")
            fields.removeLast();
 
        if(!line.contains("#")) //Skip commented areas in file
        {
            //Read number of electrodes
            if(line.contains("NumberPositions"))
                numberElectrodes = fields.at(1).toDouble();
 
            //Read the unit of the position values
            if(line.contains("UnitPosition"))
                unit = fields.at(1);
 
            //Read actual electrode positions
            if(line.contains("Positions2D"))
                read2D = true;
 
            if(line.contains(":") && !read2D) //Read 3D positions
            {
                channelNames.push_back(fields.at(0));
                QVector<float> posTemp;
 
                posTemp.push_back(fields.at(fields.size()-3).toDouble());    //x
                posTemp.push_back(fields.at(fields.size()-2).toDouble());    //y
                posTemp.push_back(fields.at(fields.size()-1).toDouble());    //z
 
                location3D.append(posTemp);
            }
 
            if(line.contains(":") && read2D) //Read 2D positions
            {
                QVector<float> posTemp;
                posTemp.push_back(fields.at(fields.size()-2).toDouble());    //x
                posTemp.push_back(fields.at(fields.size()-1).toDouble());    //y
                location2D.append(posTemp);
            }
 
            //Read channel names
            if(line.contains("Labels"))
            {
                line = in.readLine();
                fields = line.split(QRegularExpression("\\s+"));
 
                //Delete last element if it is a blank character
                if(fields.at(fields.size()-1) == "")
                    fields.removeLast();
 
                channelNames = fields;
            }
        }
    }
 
    Q_UNUSED(numberElectrodes);
 
    file.close();
 
    return true;
}
 
//=============================================================================================================
 
bool LayoutLoader::readAsaElcFile(const std::string &path,
                                  std::vector<std::string> &channelNames,
                                  std::vector<std::vector<float> > &location3D,
                                  std::vector<std::vector<float> > &location2D,
                                  std::string &unit)
{
 
    if(path.find(".elc") == std::string::npos){
        return false;
    }
 
    std::ifstream inFile(path);
 
    if(!inFile.is_open()){
        qDebug()<<"Error opening elc file";
        return false;
    }
 
    //Start reading from file
    double numberElectrodes;
    bool read2D = false;
 
    std::string line;
 
    while(std::getline(inFile, line)){
        if(line.find('#') == std::string::npos){
            std::vector<std::string> elements;
            std::stringstream stream{line};
            std::string element;
 
            stream >> std::ws;
            while(stream >> element){
                elements.push_back(std::move(element));
                stream >> std::ws;
            }
 
            //Read number of electrodes
            if(line.find("NumberPositions") != std::string::npos)
                numberElectrodes = std::stod(elements.at(1));
 
            //Read the unit of the position values
            if(line.find("UnitPosition") != std::string::npos)
                unit = elements.at(1);
 
            //Read actual electrode positions
            if(line.find("Positions2D") != std::string::npos)
                read2D = true;
 
            if(line.find(':') != std::string::npos && !read2D) //Read 3D positions
            {
                channelNames.push_back(elements.at(0));
                std::vector<float> posTemp;
 
                posTemp.push_back(std::stod(elements.at(elements.size()-3)));    //x
                posTemp.push_back(std::stod(elements.at(elements.size()-2)));    //y
                posTemp.push_back(std::stod(elements.at(elements.size()-1)));    //z
 
                location3D.push_back(std::move(posTemp));
            }
 
            if(line.find(":") != std::string::npos && read2D) //Read 2D positions
            {
                std::vector<float> posTemp;
                posTemp.push_back(std::stod(elements.at(elements.size()-2)));    //x
                posTemp.push_back(std::stod(elements.at(elements.size()-1)));    //y
                location2D.push_back(std::move(posTemp));
            }
 
            //Read channel names
            if(line.find("Labels") != std::string::npos)
            {
                std::getline(inFile, line);
                std::stringstream channels{line};
                std::vector<std::string> listOfNames;
 
                std::string channelName;
 
                channels >> std::ws;
                while(channels >> channelName){
                    listOfNames.push_back(std::move(channelName));
                    channels >> std::ws;
                }
 
                channelNames = std::move(listOfNames);
            }
        }
    }
 
    Q_UNUSED(numberElectrodes);
 
    return true;
}
 
//=============================================================================================================
 
bool LayoutLoader::readMNELoutFile(const QString &path, QMap<QString, QPointF> &channelData)
{
    //Open .elc file
    if(!path.contains(".lout"))
        return false;
 
    channelData.clear();
 
    QFile file(path);
    if(!file.open(QIODevice::ReadOnly | QIODevice::Text)) {
        qDebug()<<"Error opening mne lout file";
        return false;
    }
 
    //Start reading from file
    QTextStream in(&file);
 
    //skip first line
    in.readLine();
 
    while(!in.atEnd()) {
        QString line = in.readLine();
 
        QStringList fields = line.split(QRegularExpression("\\s+"));
 
        //Delete last element if it is a blank character
        if(fields.at(fields.size()-1) == "")
            fields.removeLast();
 
        QPointF posTemp;
        posTemp.setX(fields.at(1).toDouble());      //x
        posTemp.setY(fields.at(2).toDouble());      //y
 
        //Create channel data map entry
        QString key = QString("%1 %2").arg(fields.at(fields.size()-2)).arg(fields.at(fields.size()-1));
        channelData.insert(key, posTemp);
    }
 
    file.close();
 
    return true;
}
 
//=============================================================================================================
 
bool LayoutLoader::readMNELoutFile(const std::string &path, QMap<std::string, QPointF> &channelData)
{
 
    if(path.find(".lout") == std::string::npos){
        return false;
    }
 
    channelData.clear();
    std::ifstream inFile(path);
 
    if(!inFile.is_open()){
        qDebug()<<"Error opening mne lout file";
        return false;
    }
 
    std::string line;
 
    while(std::getline(inFile, line)){
        if(line.find('#') != std::string::npos){
            std::vector<std::string> elements;
            std::stringstream stream{line};
            std::string element;
 
            stream >> std::ws;
            while(stream >> element){
                elements.push_back(std::move(element));
                stream >> std::ws;
            }
 
            QPointF posTemp;
            posTemp.setX(std::stod(elements.at(1)));      //x
            posTemp.setY(std::stod(elements.at(2)));      //y
 
            //Create channel data map entry
            std::string key{elements.at(elements.size() - 2) + " " + elements.at(elements.size() - 1)};
            channelData.insert(key, posTemp);
        }
    }
 
    return true;
}