source_morph.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "source_morph.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QDebug>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace INVLIB;
using namespace Eigen;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
void SourceMorph::compute(const VectorXi& verticesFrom,
                           const VectorXi& verticesTo,
                           const SparseMatrix<double>& morphMap)
{
    if (verticesFrom.size() == 0 || verticesTo.size() == 0) {
        qWarning() << "[SourceMorph::compute] Empty vertex vectors.";
        return;
    }
 
    if (morphMap.rows() != verticesTo.size() || morphMap.cols() != verticesFrom.size()) {
        qWarning() << "[SourceMorph::compute] Morph map dimensions mismatch:"
                   << morphMap.rows() << "x" << morphMap.cols()
                   << "but expected" << verticesTo.size() << "x" << verticesFrom.size();
        return;
    }
 
    m_verticesFrom = verticesFrom;
    m_verticesTo = verticesTo;
    m_morphMatrix = morphMap;
    m_bComputed = true;
}
 
//=============================================================================================================
 
InvSourceEstimate SourceMorph::apply(const InvSourceEstimate& stcFrom) const
{
    if (!m_bComputed) {
        qWarning() << "[SourceMorph::apply] Morph not computed. Call compute() first.";
        return InvSourceEstimate();
    }
 
    if (stcFrom.isEmpty()) {
        qWarning() << "[SourceMorph::apply] Source estimate is empty.";
        return InvSourceEstimate();
    }
 
    if (stcFrom.data.rows() != m_morphMatrix.cols()) {
        qWarning() << "[SourceMorph::apply] Source estimate rows" << stcFrom.data.rows()
                   << "!= morph matrix cols" << m_morphMatrix.cols();
        return InvSourceEstimate();
    }
 
    // Apply morph: data_to = morphMatrix * data_from
    MatrixXd morphedData = m_morphMatrix * stcFrom.data;
 
    InvSourceEstimate stcTo(morphedData, m_verticesTo, stcFrom.tmin, stcFrom.tstep);
    stcTo.method = stcFrom.method;
    stcTo.sourceSpaceType = stcFrom.sourceSpaceType;
    stcTo.orientationType = stcFrom.orientationType;
 
    return stcTo;
}