inv_trap_music.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef INV_TRAP_MUSIC_H
#define INV_TRAP_MUSIC_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../inv_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QList>
 
//=============================================================================================================
// DEFINE NAMESPACE INVLIB
//=============================================================================================================
 
namespace INVLIB
{
 
//=============================================================================================================
struct INVSHARED_EXPORT TrapMusicDipole
{
    int     sourceIdx = -1;      
    double  correlation = 0.0;   
    Eigen::Vector3d position;    
    Eigen::Vector3d orientation; 
};
 
//=============================================================================================================
class INVSHARED_EXPORT InvTrapMusic
{
public:
    //=========================================================================================================
    explicit InvTrapMusic(int iMaxSources = 5, double dThreshold = 0.85);
 
    //=========================================================================================================
    QList<TrapMusicDipole> compute(const Eigen::MatrixXd& matLeadField,
                                    const Eigen::MatrixXd& matData,
                                    const Eigen::MatrixXd& matSourcePos,
                                    int iNOrient = 3) const;
 
    //=========================================================================================================
    static Eigen::VectorXd scanCorrelations(const Eigen::MatrixXd& matLeadField,
                                             const Eigen::MatrixXd& matSignalSubspace,
                                             int iNOrient);
 
private:
    int     m_iMaxSources;
    double  m_dThreshold;
};
 
} // namespace INVLIB
 
#endif // INV_TRAP_MUSIC_H