bids_electrode.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "bids_electrode.h"
#include "bids_tsv.h"
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace BIDSLIB;
 
//=============================================================================================================
// LOCAL HELPERS
//=============================================================================================================
 
namespace
{
static const QString NA = QStringLiteral("n/a");
 
QString naToEmpty(const QString& s)
{
    return (s == NA) ? QString() : s;
}
} // anonymous namespace
 
//=============================================================================================================
// STATIC METHODS
//=============================================================================================================
 
QList<BidsElectrode> BidsElectrode::readTsv(const QString& sFilePath)
{
    QStringList headers;
    QList<BidsTsvRow> rawRows = BidsTsv::readTsv(sFilePath, headers);
 
    QList<BidsElectrode> electrodes;
    electrodes.reserve(rawRows.size());
 
    for(const auto& row : rawRows) {
        BidsElectrode elec;
        elec.name       = row.value(QStringLiteral("name"));
        elec.x          = row.value(QStringLiteral("x"), NA);
        elec.y          = row.value(QStringLiteral("y"), NA);
        elec.z          = row.value(QStringLiteral("z"), NA);
        elec.size       = naToEmpty(row.value(QStringLiteral("size")));
        elec.type       = naToEmpty(row.value(QStringLiteral("type")));
        elec.material   = naToEmpty(row.value(QStringLiteral("material")));
        elec.impedance  = naToEmpty(row.value(QStringLiteral("impedance")));
        electrodes.append(elec);
    }
 
    return electrodes;
}
 
//=============================================================================================================
 
bool BidsElectrode::writeTsv(const QString& sFilePath,
                             const QList<BidsElectrode>& electrodes)
{
    QStringList headers = {
        QStringLiteral("name"),
        QStringLiteral("x"),
        QStringLiteral("y"),
        QStringLiteral("z"),
        QStringLiteral("size"),
        QStringLiteral("type"),
        QStringLiteral("material"),
        QStringLiteral("impedance"),
    };
 
    QList<BidsTsvRow> rows;
    rows.reserve(electrodes.size());
 
    for(const auto& elec : electrodes) {
        BidsTsvRow row;
        row[QStringLiteral("name")]      = elec.name;
        row[QStringLiteral("x")]         = elec.x;
        row[QStringLiteral("y")]         = elec.y;
        row[QStringLiteral("z")]         = elec.z;
        row[QStringLiteral("size")]      = elec.size;
        row[QStringLiteral("type")]      = elec.type;
        row[QStringLiteral("material")]  = elec.material;
        row[QStringLiteral("impedance")] = elec.impedance;
        rows.append(row);
    }
 
    return BidsTsv::writeTsv(sFilePath, headers, rows);
}