---

wavefreq.cpp

Go to the documentation of this file.

#include <assert.h>
#include <stdio.h>

#include <vector>

#include "signalUtil.h"
#include "spectrum.h"

#include "haar.h"
#include "daub.h"
#include "line.h"

#include "poly.h"


void genSawTooth( double *vec, size_t N)
{
  double *A = new double[N];
  double *B = new double[N];

  signalUtil::sawToothWave( A, N, 8, 1.5 );  
  signalUtil::sawToothWave( B, N, 32, 0.25 );
  signalUtil::addSignal( vec, A, B, N );
  delete [] A;
  delete [] B;
}


int
main()
{
  const size_t N = 1024;
  // const size_t N = sizeof( data ) / sizeof( double );
  double vecX[N], vecY[N];

  // signalUtil::gen_freqMix( vecX, vecY, N );
  signalUtil::gen_sinCombo( vecX, vecY, N );

  // genSawTooth( vecY, N );
  // signalUtil::prVec( vecY, N );

  // signalUtil::prVec( vecY, N );
  // signalUtil::prCoords( vecX, vecY, N );

  // The "Haar" transform
  // haar<double * > w;

  // Daubechies D4 wavelet
  // Daubechies<double * > w;

  // linear interpolation wavelet
  // line<double *> w;

  //
  // Polynomial interpolation wavelet (not recommended)
  //
  poly<double *> w;

  double *ptr = vecY;  
  w.forwardTrans( ptr, N );

  std::vector<double> powerVec;
  spectrum::spectralCalc( vecY, N, powerVec );

  //
  // Print out the power spectrum
  //
  size_t len = powerVec.size();
  for (size_t i = 0; i < len; i++) {
    // printf("%2d, %7.4f\n", i, powerVec[i] );
  }

  double d1[N], d2[N];

  spectrum::copyBands( d1, vecY, N, 0, 6 );
  spectrum::copyBands( d2, vecY, N, 7, 10 );

  ptr = d1;
  w.inverseTrans( ptr, N );

  ptr = d2;
  w.inverseTrans( ptr, N );

  //
  // Print out the lower half of the spectrum
  //
  // signalUtil::prCoords( vecX, d1, N );
  // signalUtil::prVec( d1, N );

  //
  // Print out the upper half of the spectrum
  //
  signalUtil::prCoords( vecX, d2, N );
  // signalUtil::prVec( d2, N );

  return 0;
}

---