ica.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ICA_DSP_H
#define ICA_DSP_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QVector>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
struct DSPSHARED_EXPORT IcaResult
{
    Eigen::MatrixXd matMixing;    
    Eigen::MatrixXd matUnmixing;  
    Eigen::MatrixXd matSources;   
    Eigen::VectorXd vecMean;      
    bool            bConverged;   
};
 
//=============================================================================================================
class DSPSHARED_EXPORT ICA
{
public:
    //=========================================================================================================
    static IcaResult run(const Eigen::MatrixXd& matData,
                         int    nComponents = -1,
                         int    maxIter     = 200,
                         double tol         = 1e-4,
                         int    randomSeed  = 42);
 
    //=========================================================================================================
    static Eigen::MatrixXd applyUnmixing(const Eigen::MatrixXd& matData,
                                          const IcaResult&       result);
 
    //=========================================================================================================
    static Eigen::MatrixXd excludeComponents(const Eigen::MatrixXd& matData,
                                              const IcaResult&       result,
                                              const QVector<int>&    excludedComponents);
 
private:
    //=========================================================================================================
    static Eigen::MatrixXd whiten(const Eigen::MatrixXd& matCentered,
                                   int                    nComponents,
                                   Eigen::MatrixXd&       matWhitening,
                                   Eigen::MatrixXd&       matDewhitening);
};
 
} // namespace UTILSLIB
 
#endif // ICA_DSP_H