ImageComparison.hh

Go to the documentation of this file.

/*
 * class ImageComparison
 *
 * $Date$
 * $Revision$
 *
 */

#ifndef IMAGE_COMPARISON_HH__
#define IMAGE_COMPARISON_HH__

#include "WImage/Image.hh"
#include "Wave/Filter.hh"
#include "WTools/ImageInformation.hh"

typedef enum { DRAWN = 0, SCANNED } imgtype;

typedef enum { PYRAMID = 0, STANDARD } decomp;

typedef struct lq {
  double colors;
  double details;
} lq;

typedef struct logvals {
  int val;
  bool init;
} logvals;

class ImageComparison 
{
public:
  ImageComparison (const Image &img1, const Image &img2);
  ~ImageComparison (void);

  inline int rows (void) const { return m_ysize; }
  inline int cols (void) const { return m_xsize; }
  inline int size (void) const { return m_xysize; }

  Image *difference (coeff magnify = 1, bool reverse = false);

  double cmpsnr (void);
  
  double cmpmse (void);
  
  lq distlq (double percent, FilterSet &flt, imgtype type);

  double distlqd (double percent, FilterSet &flt, imgtype type);

  static Image *truncateForLq (const Image &image, double percent,
                               FilterSet &flt, const decomp t = STANDARD);
  static double calcLqDetailScore (const Image &truncated1,
                                   const Image &truncated2,
                                   int colorChannel = 0);
  static double calcLqAverageScore (const Image &truncated1,
                                    const Image &truncated2,
                                    int colorChannel = 0);
  static Image *truncateForLq (const Image &image, int nKeptCoeffs,
                               FilterSet &flt, const decomp t = STANDARD);
  static ImageInformation *imageInfoForLq (const Image &image, int nKeptCoeffs,
                                           FilterSet &flt, const decomp t = STANDARD);

#define N_WEIGHTS 6

  static const float *getWeights (const imgtype imgType, int channel);

protected:
  int m_ysize;
  int m_xsize;
  int m_xysize;
  
  Image *m_images[2];
  
  logvals * m_lqcache;

  double m_lqDWeights[2][N_WEIGHTS];

  static const float m_scWeights[][N_WEIGHTS];
  static const float m_drWeights[][N_WEIGHTS];

  int m_lqcachesize;

  int level (int pos);

  double weight (int y, int x, imgtype type);
  
  double lq_sum (imgtype type, bool reverse, bool oldLqd = false);
};

#endif // IMAGE_COMPARISON_HH__