testwiener.cpp

Go to the documentation of this file.

#include <iostream>
#include <CUniformRNG.h>
#include <numeric>      
#include <algorithm>
#include <fstream>
#include "WienerFilter.h"
#include <boost/numeric/ublas/vector.hpp>

namespace ublas=boost::numeric::ublas;
int main()
{
        int length = 500; 
    double dt = 0;
    double b = 0;
    
        std::ifstream radial("smp.r"), vertical("smp.z");
        std::ofstream outfile("out");
        radial >> length >> dt >> b;
        //std::cout << "Length: " << length << " dt: " << dt << " b: " << b << std::endl;
        vertical >> length >> dt >> b;
        //std::cout << "Length: " << length << " dt: " << dt << " b: " << b << std::endl;
        gplib::rvec TimeSeries1(length);
        gplib::rvec TimeSeries2(length);
        gplib::rvec Output(length);
        for (int i = 0; i < length; ++i)
        {
                radial >> TimeSeries2(i);
                vertical >> TimeSeries1(i);
        }
        //std::cout << "lenght1: " << TimeSeries1.size() << " length2: " << TimeSeries2.size() << std::endl; 
        double mean1=std::accumulate(TimeSeries1.begin(),TimeSeries1.end(),0.)/length;
        double mean2=std::accumulate(TimeSeries2.begin(),TimeSeries2.end(),0.)/length;
        //std::cout << "Mean1: " << mean1 << " Mean2: " << mean2 << std::endl;
        std::transform(TimeSeries2.begin(),TimeSeries2.end(),TimeSeries2.begin(),bind2nd(std::minus<double>(),mean2));
        std::transform(TimeSeries1.begin(),TimeSeries1.end(),TimeSeries1.begin(),bind2nd(std::minus<double>(),mean1));
        
    WienerFilter Wiener(length);
    Wiener.SetLambda(1.);
    Wiener.AdaptFilter(TimeSeries1,TimeSeries2);
    Wiener.CalcOutput(TimeSeries1,Output);
    Wiener.PrintWeights(outfile);
        /*for (int i = 0; i < Output.size(); ++i)
            std::cout << Output(i) << " " << std::endl;
        std::cout << std::endl;*/
}

---