sts_source_metrics.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "sts_source_metrics.h"
 
//=============================================================================================================
// STL INCLUDES
//=============================================================================================================
 
#include <cmath>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace STSLIB;
using namespace Eigen;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
double StatsSourceMetrics::peakLocalizationError(
    const Vector3d& truePos,
    const Vector3d& estimatedPos)
{
    return (truePos - estimatedPos).norm();
}
 
//=============================================================================================================
 
double StatsSourceMetrics::spatialDispersion(
    const VectorXd& sourceAmplitudes,
    const MatrixXd& sourcePositions,
    int peakIndex)
{
    int nSources = sourceAmplitudes.size();
    Vector3d peakPos = sourcePositions.row(peakIndex);
 
    double weightedSum = 0.0;
    double totalWeight = 0.0;
 
    for (int k = 0; k < nSources; ++k) {
        double absAmp = std::abs(sourceAmplitudes(k));
        double dist = (sourcePositions.row(k).transpose() - peakPos).norm();
        weightedSum += dist * absAmp;
        totalWeight += absAmp;
    }
 
    if (totalWeight < 1e-10) {
        return 0.0;
    }
 
    return weightedSum / totalWeight;
}
 
//=============================================================================================================
 
int StatsSourceMetrics::findPeakIndex(const VectorXd& sourceAmplitudes)
{
    int peakIdx = 0;
    sourceAmplitudes.cwiseAbs().maxCoeff(&peakIdx);
    return peakIdx;
}