asclmscancel.cpp

Go to the documentation of this file.

#include <fstream>
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include "CLMSCanceller.h"
#include "CTimeSeriesData.h"
#include "CStackedSpectrum.h"
#include "CLmsConfig.h"

using namespace std;

int main()
{
        CLmsConfig Configuration;
        CLMSCanceller Canceller;
        CTimeSeriesData TsData;
        StackedSpectrum WeightSpectrum;
        //CMTStation Output;
        
        int i,j;
        string tsfilename, noisefilename;
        ifstream noisefile,tsfile;
        ofstream outfile, weightfile,epsfile;
        double temp;
        int ntimeseries,minlength = 1e15;
        vector<double> noise, epsilon, output,ts;
        
        Configuration.GetData("lms.conf");
        cout << "Reference Filename: ";
        cin >> noisefilename;
        noisefile.open(noisefilename.c_str());
        while (noisefile.good())
        {
                noisefile >> temp; 
                if (noisefile.good())
                 noise.push_back(temp);
        }
        noisefile.close();
        cout << "Number of Time Series: ";
        cin >> ntimeseries;
        minlength = noise.size();
        for (i=0; i < ntimeseries; ++i)
        {
                cout << "Time Series Filename: ";
                cin >> tsfilename;
                Canceller.Input.push_back(new ttsdata);
        
                tsfile.open(tsfilename.c_str());
                while (tsfile.good())
                {
                        tsfile >> temp; 
                        if (tsfile.good())
                                Canceller.Input.at(i)->push_back(temp);
                }
                tsfile.close();
                tsfile.clear();
                if (Canceller.Input.at(i)->size() < minlength)
                        minlength = Canceller.Input.at(i)->size();
        }
        for (i=0; i < ntimeseries; ++i)
                Canceller.Input.at(i)->resize(minlength);
        cout << "Minlength: " << minlength << endl; 
        Canceller.filterlength = Configuration.filterlength;
        Canceller.mu.assign(ntimeseries,Configuration.mu);
        Canceller.secfactor = Configuration.secfactor;
        //Canceller.input = &TsData.Data->Ex.data;
        Canceller.FilterOutput = &noise;
        Canceller.Epsilon = &epsilon;
        Canceller.Desired = &output;
        const int shift = Configuration.shift;
        Canceller.FilterData(shift);
        
        cout << "Mu: " ;
        for (i=0 ; i < ntimeseries; ++i) 
                cout << Canceller.mu.at(i) << " ";
        cout <<endl;
        epsfile.open((noisefilename+".eps").c_str());
        outfile.open( (noisefilename+".clean").c_str());
        for (i = shift; i < output.size(); ++i)
        {
                outfile << output.at(i) << endl;
                epsfile << Canceller.Epsilon->at(i) << endl;
        }
        outfile.close();
        epsfile.close();
        
        /*WeightSpectrum.TimeSeries.assign(Canceller.Weights.size(),0);
        copy(Canceller.Weights.begin(),Canceller.Weights.end(),WeightSpectrum.TimeSeries.begin());
        WeightSpectrum.CalcSpectrum(filterlength);*/
        
        weightfile.open((noisefilename+".weights").c_str());
        for (i = 0; i < Canceller.WeightHistory.size(); ++i)
        {
                for (j = 0; j < Configuration.filterlength*ntimeseries; ++j)
                {
                        weightfile << j << " " << i << " " << Canceller.WeightHistory.at(i).at(j) << " " << endl;
                        //weightfile << Canceller.WeightHistory.at(i).at(j) << " ";
                }
                //weightfile << endl;
        }
        weightfile.close();
}