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