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