WaveletTransform.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2008 by Paul Demorest
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __Pulsar_Wavelet_h
#define __Pulsar_Wavelet_h
 
#include "ReferenceAble.h"
 
#include <gsl/gsl_wavelet.h>
 
namespace Pulsar {
 
  class Profile;
 
 
  class WaveletTransform : public Reference::Able {
 
    public:
 
      WaveletTransform();
 
      ~WaveletTransform();
 
 
      void set_type(const gsl_wavelet_type *t) { type=t; };
 
 
      void set_order(int o) { order=o; };
 
      void set_wavelet(std::string s);
 
      void set_decimate(bool _dec=true) { decimate = _dec; }
 
      void transform(unsigned npts, const float *in);
 
      void transform(const std::vector<float>& in);
 
      void transform(const Pulsar::Profile* in);
 
      void invert();
 
      enum State { Wavelet, Time, Empty };
 
      const bool get_decimate() const { return decimate; }
 
      State get_state() { return state; }
 
      double get_data(int level, int k);
 
      double get_data(int n);
 
      double* get_data();
 
      const double* get_data() const;
 
      const int get_npts() const { return npts; };
 
      const int get_log2_npts() const { return log2_npts; };
 
      const int get_ncoeff() const { return decimate ? npts : log2_npts*npts; }
 
      const int get_ncoeff(int level) const;
 
    protected:
 
      const gsl_wavelet_type *type;
 
      bool decimate;
 
      int order;
 
      gsl_wavelet *wave;
 
      int npts;
 
      int log2_npts;
 
      double *data;
 
      double mean;
 
      State state;
 
      gsl_wavelet_workspace *work;
 
    private:
 
      void free_mem();
 
      void init_mem();
 
 
  };
 
}
 
#endif