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