wavelet_decomp.cpp

Go to the documentation of this file.

#include "CTimeSeriesData.h"
#include <iostream>
#include <string>
#include <vector>
#include <fstream>
#include <gsl/gsl_wavelet.h>

using namespace std;

int main(void)
{
        string infilename, outfilename;
        int componentindex = 0;
        ptsdata currentdata;
        double *dataarray;
        const int seglength = 1024;
        
        
        
        cout << "Input - Filename: ";
        cin >> infilename;
        CTimeSeriesData TsData;
        TsData.GetData(infilename);
        cout << "Component: ";
        cin >> componentindex;
        switch (componentindex)
        {
                case    0:
                        currentdata = &(TsData.Data->Ex.data);
                break;
                case 1:
                        currentdata = &(TsData.Data->Ey.data);
                break;
                case 2:
                        currentdata = &(TsData.Data->Hx.data);
                break;
                case 3:
                        currentdata = &(TsData.Data->Hy.data);
                break;
                case 4:
                        currentdata = &(TsData.Data->Hz.data);
                break;
        }       
        outfilename = infilename + ".wav";
        const int size = currentdata->size();
        const int nsegs = size/seglength;
        vector<double> stackedcoeffs(seglength,0);
        dataarray = new double[seglength];
        
        
        gsl_wavelet *w = gsl_wavelet_alloc(gsl_wavelet_daubechies,4);
        gsl_wavelet_workspace *work = gsl_wavelet_workspace_alloc(seglength);
        
        for (int i = 0; i < nsegs; ++i)
        {
                for (int j = 0; j < seglength; ++j)
                        dataarray[j] = currentdata->at(j+i*seglength);
                gsl_wavelet_transform_forward(w, dataarray, 1, seglength, work);
                for (int j = 0; j < seglength; ++j)
                        stackedcoeffs.at(j) += dataarray[j];
        }
        ofstream outfile(outfilename.c_str());
        for (int j = 0; j < seglength; ++j)
                outfile << stackedcoeffs.at(j) << endl;
        delete []dataarray;
        
        
}

---