cascadelmscancel.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 Stage1Canceller, Stage2Canceller;
        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> stage1noise, stage1epsilon, stage1output;
        vector<double> stage2noise, stage2epsilon, stage2output;
        
        Configuration.GetData("lms.conf");
        cout << "Reference Filename: ";
        cin >> noisefilename;
        noisefile.open(noisefilename.c_str());
        while (noisefile.good())
        {
                noisefile >> temp; 
                stage1noise.push_back(temp);
        }
        noisefile.close();
        cout << "Number of Time Series: ";
        cin >> ntimeseries;
        minlength = stage1noise.size();
        for (i=0; i < ntimeseries; ++i)
        {
                cout << "Time Series Filename: ";
                cin >> tsfilename;
                Stage1Canceller.Input.push_back(new ttsdata);
        
                tsfile.open(tsfilename.c_str());
                while (tsfile.good())
                {
                        tsfile >> temp; 
                        Stage1Canceller.Input.at(i)->push_back(temp);
                }
                tsfile.close();
                tsfile.clear();
                if (Stage1Canceller.Input.at(i)->size() < minlength)
                        minlength = Stage1Canceller.Input.at(i)->size();
        }
        for (i=0; i < ntimeseries; ++i)
                Stage1Canceller.Input.at(i)->resize(minlength);
        cout << "Minlength: " << minlength << endl; 
        Stage1Canceller.filterlength = Configuration.filterlength;
        Stage1Canceller.mu.assign(ntimeseries,Configuration.mu);
        Stage1Canceller.secfactor = Configuration.secfactor;
        Stage2Canceller.filterlength = Configuration.filterlength;
        Stage2Canceller.mu.assign(ntimeseries,Configuration.mu);
        Stage2Canceller.secfactor = Configuration.secfactor;
        //Canceller.input = &TsData.Data->Ex.data;
        Stage1Canceller.FilterOutput = &stage1noise;
        Stage1Canceller.Epsilon = &stage1epsilon;
        Stage1Canceller.Desired = &stage1output;
        Stage2Canceller.FilterOutput = &stage1epsilon;
        Stage2Canceller.Desired = &stage2output;
        Stage2Canceller.Epsilon = &stage2epsilon;
        Stage2Canceller.Input = Stage1Canceller.Input;
        
        Stage1Canceller.FilterData(Configuration.shift);
        Stage2Canceller.FilterData(Configuration.shift);
        cout << "Mu: " ;
        for (i=0 ; i < ntimeseries; ++i) 
                cout << Stage1Canceller.mu.at(i) << " ";
        cout <<endl;
        epsfile.open((noisefilename+".eps").c_str());
        outfile.open( (noisefilename+".clean").c_str());
        for (i = 0; i < stage2output.size(); ++i)
        {
                outfile << stage2output.at(i) << endl;
                epsfile << Stage2Canceller.Epsilon->at(i) << endl;
        }
        outfile.close();
        epsfile.close();
        
}