WaveletSmooth.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2008 by Paul Demorest
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __Pulsar_WaveletSmooth_h
#define __Pulsar_WaveletSmooth_h
 
#include "Pulsar/Transformation.h"
#include "Pulsar/WaveletTransform.h"
 
namespace Pulsar {
 
  class Profile;

  class WaveletSmooth : public Transformation<Profile> {
 
  public:

    WaveletSmooth ();

    ~WaveletSmooth ();

    unsigned get_ncoeff() const { return ncoeff; };

    void transform(Profile *);

    void set_wavelet(const gsl_wavelet_type *t, int order, bool decimate=true);

    void set_wavelet(std::string s);

    enum Threshold { Hard, Soft };

    void set_threshold(Threshold t) { thresh = t; };

    void set_cutoff(double cut) { cutoff_factor = cut; }
 
  protected:

    int ncoeff;

    Threshold thresh;

    double cutoff;

    double cutoff_factor;

    double sigma;

    WaveletTransform wt;
 
  private:
    
    double thresh_hard(double in);

    double thresh_soft(double in);
 
  };
 
}
 
#endif