rlscancel.cpp

Go to the documentation of this file.

#include <fstream>
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include "CRLSCanceller.h"
#include "CTimeSeriesData.h"
#include "CStackedSpectrum.h"
#include "CRLSConfig.h"

using namespace std;

int main()
{
        CRLSConfig       Config;
        CRLSCanceller Canceller;
        CTimeSeriesData TsData;
        StackedSpectrum WeightSpectrum;
        //CMTStation Output;
        
        Config.GetData("rls.conf");
        const int filterlength = Config.filterlength;
        //const double mu = 1e-17;
        const double delta = Config.delta;
        const double lambda = Config.lambda;
        const int shift = Config.shift;
        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;
        
        cout << "Reference Filename: ";
        cin >> noisefilename;
        noisefile.open(noisefilename.c_str());
        while (noisefile.good())
        {
                noisefile >> temp; 
                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; 
                        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 = filterlength;
        Canceller.delta = delta;
        Canceller.lambda = lambda;
        //Canceller.input = &TsData.Data->Ex.data;
        Canceller.FilterOutput = &noise;
        Canceller.Epsilon = &epsilon;
        Canceller.Desired = &output;
        Canceller.FilterData(shift);
        
        cout <<endl;
        epsfile.open((noisefilename+".eps").c_str());
        outfile.open( (noisefilename+".clean").c_str());
        for (i = 0; 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 < filterlength * ntimeseries; ++j)
                {
                        weightfile << j << " " << i << " " << Canceller.WeightHistory.at(i).at(j) << " " << endl;
                }
        }
        weightfile.close();
}