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;
}