tftest.cpp

Go to the documentation of this file.

#include "TimeFrequency.h"
#include "StackedSpectrum.h"
#include "UniformRNG.h"
#include "types.h"
#include "WFunc.h"
#include <iostream>
#include "VecMat.h"
#include <gsl/gsl_math.h>
using namespace std;
int main()
{
        const double tolerance = 1e-6;
        const int runs = 100;
        const int maxseg = 1000;

        UniformRNG Random;

        for (int r = 0; r < runs; ++r)
        {
                const int length = Random.GetNumber(maxseg * 500);
                const int seglength = Random.GetNumber(maxseg) +10;
                const int speclength = seglength/2+1;
                ttsdata Input(length);
                tcompdata Spectrum(seglength/2+1);

                cout << "Run: " << r << "       ";
                cout << "Ts length: " << length << " Seglength: " << seglength << "     ";
                bool runok = true;
                for (int i = 0; i < length; ++i)
                        Input.at(i) = Random.GetNumber();
                try
                {
                        gplib::cmat result =
                                TimeFrequency(Input.begin(), Input.end(), seglength, Hamming());
                        StackedSpectrum(Input.begin(), Input.end(), Spectrum.begin(),seglength, Hamming());
                        gplib::cvec tfstack
                                = prod( boost::numeric::ublas::scalar_vector<double>(length/seglength,1.0),result);
                        cout << "Matrix   Rows: " << result.size1() << " Columns: " << result.size2() << endl;
                        cout << "tfstack size: " << tfstack.size() << endl;
                        for (int i = 0; i < speclength; ++i)
                        {
                                if ((gsl_fcmp(Spectrum.at(i).real(),tfstack(i).real(),tolerance) != 0) ||
                                        (gsl_fcmp(Spectrum.at(i).imag(),tfstack(i).imag(),tolerance) != 0))
                                {
                                cout << endl;
                                cout << i << " " << Spectrum.at(i) << " " << tfstack(i) << endl;
                                runok = false;
                                }
                        }
                }
                catch(CFatalException &e)
                {
                        cout << e.what() << "     ";
                        runok = false;
                }
                if (runok)
                        cout << "Run OK !" << endl;
                else
                        cout << "Run failed !" << endl;
        }
        cout << "All finished !" << endl;
}

---