sss.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef SSS_DSP_H
#define SSS_DSP_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
#include <fiff/fiff_info.h>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QVector>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
struct DSPSHARED_EXPORT SSSParams
{
    int iOrderIn = 8;                       
    int iOrderOut = 3;                      
    Eigen::Vector3d origin{0.0, 0.0, 0.04}; 
    double dRegIn = 1e-5;                   
};
 
//=============================================================================================================
class DSPSHARED_EXPORT SSS
{
public:
    using Params = SSSParams; 
 
    //=========================================================================================================
    struct Basis
    {
        Eigen::MatrixXd matSin;     
        Eigen::MatrixXd matSout;    
        Eigen::MatrixXd matProjIn;  
        Eigen::MatrixXd matPinvAll; 
        QVector<int> megChannelIdx; 
        int iOrderIn = 8;           
        int iOrderOut = 3;          
        int iNin = 0;               
        int iNout = 0;              
    };
 
    //=========================================================================================================
    static Basis computeBasis(const FIFFLIB::FiffInfo& fiffInfo,
                              const Params& params = Params());
 
    //=========================================================================================================
    static Eigen::MatrixXd apply(const Eigen::MatrixXd& matData,
                                 const Basis& basis);
 
    //=========================================================================================================
    static Eigen::MatrixXd applyTemporal(const Eigen::MatrixXd& matData,
                                         const Basis& basis,
                                         int iBufferLength = 10000,
                                         double dCorrLimit = 0.98);
 
private:
    //=========================================================================================================
    static void computeNormALP(int lmax,
                               double cosTheta,
                               double sinTheta,
                               Eigen::MatrixXd& P,
                               Eigen::MatrixXd& dP);
 
    //=========================================================================================================
    static Eigen::Vector3d basisGradCart(int l,
                                         int m,
                                         bool bInternal,
                                         const Eigen::Vector3d& rPos,
                                         const Eigen::MatrixXd& P,
                                         const Eigen::MatrixXd& dP,
                                         double cosTheta,
                                         double sinTheta,
                                         double cosPhi,
                                         double sinPhi);
};
 
} // namespace UTILSLIB
 
#endif // SSS_DSP_H