testfastica.cpp

Go to the documentation of this file.

#include "FastICA.h"
#include "UniformRNG.h"
#include "miscfunc.h"
#include <itpp/base/fastica.h>
#include <itpp/base/mat.h>
#include <itpp/base/sigfun.h>
#include <fstream>
using namespace std;
void WriteToFile(const itpp::Vec<double> &in,const  itpp::Vec<double> &out,const string &base)
{
        ofstream ints((base+".in").c_str());
        ofstream outts((base+".out").c_str());
        const int size = in.length();
        for (int i = 0; i < size; ++i)
        {
                ints << in(i) <<  endl;
                outts << out(i) <<  endl;
        }
}

int main()
{
        const int nchannels = 2;
        const int nsamples = 2000;
        const float minrand = -0.5;
        const float maxrand = 0.5;
        UniformRNG Random;
        
        itpp::mat Input(nchannels,nsamples);
        itpp::mat Sources(nchannels,nsamples);
        rmat orig_source(nchannels,nsamples);
        rmat input(nchannels,nsamples);
        for (int j = 0; j < nsamples; ++j)
        {
                        orig_source(0,j) = 100.9*Random.GetNumber(minrand,maxrand); 
                        orig_source(1,j) = 13.0*sin(j/13.678);
        }
        rmat orig_mix(nchannels,nchannels);
        for (int i = 0; i < nchannels; ++i)
                for (int j = 0; j < nchannels; ++j)
                        orig_mix(i,j) = Random.GetNumber(minrand,maxrand); 
        
        noalias(input) = ublas::prod(orig_mix,orig_source);
                
         for (size_t i = 0; i < nsamples; ++i)
    {
        
        Input(0,i) = input(0,i);
        Input(1,i) = input(1,i);
    }
        
        itpp::Fast_ICA fastica(Input);
        fastica.set_nrof_independent_components(nchannels);
        fastica.set_non_linearity(FICA_NONLIN_POW3);
        fastica.set_approach(FICA_APPROACH_SYMM);
        fastica.separate();
        itpp::mat Output = fastica.get_independent_components();
        
        
        rmat source(nchannels,nsamples);
        rmat mix(nchannels,nchannels);
        
        
        //std::cout << "input: " << input << std::endl << std::endl;
        FastICA(input,source,mix);
        WriteToFile(Output.get_row(0),Output.get_row(1),"1");
        std::cout << std::endl << std::endl;
        //std::cout << "orig_source: " << orig_source << std::endl << std::endl;
        std::cout << "Itpp WhM: " << fastica.get_whitening_matrix() << std::endl;
        std::cout << "orig_mix: " << orig_mix << std::endl<< std::endl;
        std::cout << "FastICA MixMat: " << fastica.get_mixing_matrix() << std::endl;
        std::cout << "FastICA SepMat: " << fastica.get_separating_matrix() << std::endl;
        //std::cout << "est_source: " << source << std::endl<< std::endl;
        std::cout << "est_mix: " << mix << std::endl<< std::endl; 
        std::ofstream outfile("fastica.out");
        for (int i = 0; i < nsamples; ++i)
                outfile << i << " " << input(0,i) << " " << input(1,i) <<" " << source(0,i) << " " << source(1,i) << " " << Output(0,i) << " " << Output(1,i)  << std::endl;
}

---