mne_ctf_comp_data.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "mne_ctf_comp_data.h"
#include "mne_types.h"
 
#include <fiff/fiff_constants.h>
#include <fiff/fiff_tag.h>
 
#include <QFile>
#include <QDebug>
 
#include <Eigen/Core>
 
constexpr int FAIL = -1;
constexpr int OK = 0;
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace Eigen;
using namespace FIFFLIB;
using namespace MNELIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
MNECTFCompData::MNECTFCompData()
: kind(MNE_CTFV_COMP_UNKNOWN)
, mne_kind(MNE_CTFV_COMP_UNKNOWN)
, calibrated(false)
{
}
 
//=============================================================================================================
 
MNECTFCompData::MNECTFCompData(const MNECTFCompData& comp)
: kind(MNE_CTFV_COMP_UNKNOWN)
, mne_kind(MNE_CTFV_COMP_UNKNOWN)
, calibrated(false)
{
    kind = comp.kind;
    mne_kind = comp.mne_kind;
    calibrated = comp.calibrated;
    if (comp.data)
        data = std::make_unique<MNENamedMatrix>(*comp.data);
    if (comp.presel)
        presel = std::make_unique<FiffSparseMatrix>(*comp.presel);
    if (comp.postsel)
        postsel = std::make_unique<FiffSparseMatrix>(*comp.postsel);
}
 
//=============================================================================================================
 
MNECTFCompData::~MNECTFCompData()
{
}
 
//=============================================================================================================
 
int MNECTFCompData::calibrate(const QList<FIFFLIB::FiffChInfo>& chs, int nch, bool do_it)
{
    Eigen::VectorXf col_cals(this->data->ncol);
    Eigen::VectorXf row_cals(this->data->nrow);
    int j, k, p, found;
    QString name;
 
    if (calibrated)
        return OK;
 
    for (j = 0; j < this->data->nrow; j++) {
        name = this->data->rowlist[j];
        found = false;
        for (p = 0; p < nch; p++)
            if (QString::compare(name, chs[p].ch_name) == 0) {
                row_cals[j] = chs[p].range * chs[p].cal;
                found = true;
                break;
            }
        if (!found) {
            qCritical("Channel %s not found. Cannot calibrate the compensation matrix.", name.toUtf8().constData());
            return FAIL;
        }
    }
    for (k = 0; k < this->data->ncol; k++) {
        name = this->data->collist[k];
        found = false;
        for (p = 0; p < nch; p++)
            if (QString::compare(name, chs[p].ch_name) == 0) {
                col_cals[k] = chs[p].range * chs[p].cal;
                found = true;
                break;
            }
        if (!found) {
            qCritical("Channel %s not found. Cannot calibrate the compensation matrix.", name.toUtf8().constData());
            return FAIL;
        }
    }
    if (do_it) {
        for (j = 0; j < this->data->nrow; j++)
            for (k = 0; k < this->data->ncol; k++)
                this->data->data(j, k) = row_cals[j] * this->data->data(j, k) / col_cals[k];
    } else {
        for (j = 0; j < this->data->nrow; j++)
            for (k = 0; k < this->data->ncol; k++)
                this->data->data(j, k) = col_cals[k] * this->data->data(j, k) / row_cals[j];
    }
    return OK;
}