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