Signal Processing/testlmscancel.cpp

Go to the documentation of this file.

#include <fstream>
#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include "LMSCanceller.h"
#include "RLSCanceller.h"
#include "ApplyFilter.h"
#include "TimeSeriesComponent.h"
using namespace std;

int main()
{
        ofstream weightfile;

        
        const int filterlength = 2;
        const double mu = 0.1;
        const int nsamples = 500;
        const double period = 5.2;
        const double PI = acos(-1.0);
        LMSCanceller Filter(filterlength); 
        Filter.SetMu(mu);
        
        ApplyFilter Canceller(Filter);
        
        TimeSeriesComponent  reference, input;
        
        for (int i = 0; i < nsamples; ++i)
        {
                reference.GetData().push_back(2.0*cos( 2.0 * PI * i/period));
                input.GetData().push_back(sin( 2.0 * PI * i/period));
        }

        Canceller.AddReferenceChannel(reference);
        Canceller.AddInputChannel(input);
        Canceller.SetShift(0);
        Canceller.FilterData();
        
        
        //*RefComp = *Canceller.GetOutChannels().front();
        //ReferenceData.WriteAsMtu(noisefilename+".clean");
        ofstream cleanfile("test.clean");
        copy(Canceller.GetOutChannels().front()->GetData().begin(),Canceller.GetOutChannels().front()->GetData().end(),ostream_iterator<double>(cleanfile,"\n"));
        //copy(Canceller.GetEpsValues().front().begin(),Canceller.GetEpsValues().front().end(),RefComp->GetData().begin());
        //ReferenceData.WriteAsMtu(noisefilename+".eps");
        ofstream epsfile("test.eps");
        copy(Canceller.GetEpsValues().front().begin(),Canceller.GetEpsValues().front().end(),ostream_iterator<double>(epsfile,"\n"));
        
        cout << "Last Weights: " << endl;
        Filter.PrintWeights(cout);
        cout << endl;
        cout << "Optimum: " << endl;
        cout << 2 * 1./tan(2*PI/period) << " " << - 2 * 1./sin(2*PI/period) << endl;
        cout << endl;
        //copy(Canceller.Epsilon->begin(),Canceller.Epsilon->end(),prefcomponent->begin());
        //ReferenceData.WriteAsMtu(noisefilename+".eps");
        //copy(Canceller.Desired->begin(),Canceller.Desired->end(),prefcomponent->begin());
        //ReferenceData.WriteAsMtu(noisefilename+".clean");
        
        /*
        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();*/
}

---