mticatest.cpp

Go to the documentation of this file.

#include <iostream>
#include <itpp/base/fastica.h>
#include <itpp/base/mat.h>
#include <itpp/base/sigfun.h>
#include <string>
#include <fstream>
#include "CUniformRNG.h"
#include "TimeSeriesData.h"
#include "statutils.h"
#include <itpp/base/stat.h>
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 inspec((base+".spec.in").c_str());
        ofstream outts((base+".out").c_str());
        //ofstream outspec((base+".spec.out").c_str());
        const int size = in.length();
    //int speclength = 2;
    //while (speclength < size)
    //  speclength *= 2;
        //itpp::Vec<double> insp = itpp::spectrum(in,speclength,0);
        //itpp::Vec<double> outsp = itpp::spectrum(out,speclength,0);
        for (int i = 0; i < size; ++i)
        {
                ints << in(i) <<  endl;
                //inspec << insp(i) <<  endl;
                outts << out(i) <<  endl;
                //outspec << outsp(i) <<  endl;
        }
}


int main()
{
        string datafilename, reffilename;
        TimeSeriesData MTData, RefData;    
            
        cout << "Data filename: ";
        cin >> datafilename;
        MTData.GetData(datafilename);
        
        
        cout << "Reference filename: ";
        cin >> reffilename;
        RefData.GetData(reffilename);
        
        const size_t tslength = MTData.GetData().GetEx().GetData().size();
        
        SubMean(MTData.GetData().GetHx().GetData().begin(),MTData.GetData().GetHx().GetData().end());
        SubMean(MTData.GetData().GetHy().GetData().begin(),MTData.GetData().GetHy().GetData().end());
        SubMean(RefData.GetData().GetHx().GetData().begin(),RefData.GetData().GetHx().GetData().end());
        SubMean(RefData.GetData().GetHy().GetData().begin(),RefData.GetData().GetHy().GetData().end());
        
    const int nchannels =4;
        itpp::mat Input(nchannels,tslength);
    for (size_t i = 0; i < tslength; ++i)
    {
        Input(0,i) = MTData.GetData().GetHx().GetData().at(i);
        Input(1,i) = MTData.GetData().GetHy().GetData().at(i);
        Input(2,i) = RefData.GetData().GetHx().GetData().at(i);
        Input(3,i) = RefData.GetData().GetHy().GetData().at(i);
    }
    for (int i = 0; i < nchannels; ++i)
         cout << "Kurt " << i << ": "<< itpp::kurtosis(Input.get_row(i)) << endl;
        itpp::Fast_ICA fastica(Input);
        fastica.set_stabilization(true);
        fastica.set_nrof_independent_components(nchannels);
        fastica.set_non_linearity(FICA_NONLIN_GAUSS);
        fastica.set_approach(FICA_APPROACH_DEFL);
        fastica.separate();
        itpp::mat Output = fastica.get_independent_components();
        itpp::mat White = fastica.get_white_sig();
        WriteToFile(Input.get_row(0),Output.get_row(0),datafilename+".hx");
        WriteToFile(Input.get_row(1),Output.get_row(1),datafilename+".hy");
        WriteToFile(Input.get_row(2),Output.get_row(2),reffilename+".hx");
        WriteToFile(Input.get_row(3),Output.get_row(3),reffilename+".hy");
        cout << "Mix Mat: " << endl << fastica.get_mixing_matrix() << endl;
        cout << "Whit Mat: " << endl << fastica.get_whitening_matrix() << endl;
}