ImageDenoiser.hh

Go to the documentation of this file.

/*
 * class ImageDenoiser
 *
 * $Date$
 * $Revision$
 *
 */

#ifndef IMAGE_DENOISER_HH__
#define IMAGE_DENOISER_HH__

#include "Wave/PyramidTransform.hh"
#include "WTools/PyramidTree.hh"

#define DENOISE_HL 0x01

#define DENOISE_LH 0x02

#define DENOISE_HH 0x04

#define SIGNIFICANT_COEFF   0
#define SIGNIFICANT_REGION  1
#define SIGNIFICANT_CHANNEL 2

#define REPLACE_SIMPLE  0
#define REPLACE_CHANNEL 1

class ImageDenoiser 
{
public:
  ImageDenoiser (Image &img, unsigned areas, int fromSub, int toSub = 1, 
                 double alpha = 1.0, FilterSet &filter = Daub4, 
                 int isSingleSignificant = 0, int replacementFunction = 0);
  
  virtual ~ImageDenoiser (void);

  virtual void denoise (void);
  
protected:

  coeff threshold (int subband, area channel);

  virtual void processFrom (PyramidTree &tree, coeff *thresholds, 
                            area channel, int level);

  virtual bool isCoeffSingleSignificant (CoeffInformation &c, area channel, 
                                         coeff threshold, bool &stop);
  virtual bool isSingleSignificantInRegion (CoeffInformation &c, area channel, 
                                            coeff threshold, bool &stop);
  virtual bool isSingleSignificantInChannel (CoeffInformation &c, area channel,
                                             coeff threshold, bool &stop);
  virtual void replaceSimple (int y, int x, area channel, coeff threshold);
  virtual void replaceChannel (int y, int x, area channel, coeff threshold);

  Image *m_image;
  PyramidTransform *m_transform;
  FilterSet *m_filter;
  int m_from;
  int m_to;
  unsigned m_areas;
  double m_alpha;
  int m_significanceFunction;
  int m_replacementFunction;
  
}; /* class ImageDenoiser */

#endif /* IMAGE_DENOISER_HH__ */