/*
 * * Copyright (C) 2006-2011 Anders Brander <anders@brander.dk>, 
 * * Anders Kvist <akv@lnxbx.dk> and Klaus Post <klauspost@gmail.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */

#ifndef complexfilter_h__
#define complexfilter_h__
#include "complexblock.h"
#include "floatimageplane.h"

namespace RawStudio {
namespace FFTFilter {

class FFTWindow;

class ComplexFilter
{
public:
  ComplexFilter(int block_width, int block_height);
  ComplexFilter(const ComplexFilter& p);
  virtual ~ComplexFilter(void);
  void process(ComplexBlock* block);
  virtual void setSharpen( float sharpen, float sigmaSharpenMin, float sigmaSharpenMax, float scutoff );
  virtual gboolean skipBlock();
protected:
  virtual void processNoSharpen(ComplexBlock* block) = 0;
  virtual void processSharpen(ComplexBlock* block) = 0;  
  const int bw;
  const int bh;
  const float norm; // Normalization factor
  float lowlimit;
  float sharpen;
  float sigmaSquaredSharpenMin;
  float sigmaSquaredSharpenMax;
  FloatImagePlane *sharpenWindow;
};

class DeGridComplexFilter : public ComplexFilter
{
public:
  DeGridComplexFilter(int block_width, int block_height, float degrid, FFTWindow *window, fftwf_plan plan_forward);
  virtual ~DeGridComplexFilter(void);
protected:
  virtual void processSharpenOnly(ComplexBlock* block);
#if defined (__i386__) || defined (__x86_64__)
  void processSharpenOnlySSE(ComplexBlock* block);
  void processSharpenOnlySSE3(ComplexBlock* block);
#endif
  const float degrid;
  FFTWindow *window;
  ComplexBlock* grid;
};

class ComplexWienerFilter : public ComplexFilter
{
public:
  ComplexWienerFilter(int block_width, int block_height, float beta, float sigma);
  virtual ~ComplexWienerFilter(void);
  virtual gboolean skipBlock();
protected:
  virtual void processNoSharpen(ComplexBlock* block);
  virtual void processSharpen(ComplexBlock* block);
  float sigmaSquaredNoiseNormed;
};

class ComplexWienerFilterDeGrid : public DeGridComplexFilter
{
public:
  ComplexWienerFilterDeGrid(int block_width, int block_height, float beta, float sigma, float degrid, fftwf_plan plan, FFTWindow *window);
  virtual ~ComplexWienerFilterDeGrid(void);
  virtual gboolean skipBlock();
protected:
  virtual void processNoSharpen(ComplexBlock* block);
  virtual void processSharpen(ComplexBlock* block);
#if defined (__i386__) || defined (__x86_64__)
  virtual void processSharpen_SSE3(ComplexBlock* block);
  virtual void processSharpen_SSE(ComplexBlock* block);
  virtual void processNoSharpen_SSE(ComplexBlock* block);
  virtual void processNoSharpen_SSE3(ComplexBlock* block);
#endif
  float sigmaSquaredNoiseNormed;
  FFTWindow *window;
};

class ComplexPatternFilter : public ComplexFilter
{
public:
  ComplexPatternFilter(int block_width, int block_height, float beta, FloatImagePlane* pattern, float pattern_strength );
  virtual ~ComplexPatternFilter(void);
  virtual gboolean skipBlock();
protected:
  virtual void processNoSharpen(ComplexBlock* block);
  virtual void processSharpen(ComplexBlock* block);
  FloatImagePlane* pattern;
  const float pfactor;
};


class ComplexFilterPatternDeGrid : public DeGridComplexFilter
{
public:
  ComplexFilterPatternDeGrid(int block_width, int block_height, float beta, float sigma, float degrid, fftwf_plan plan, FFTWindow *window, FloatImagePlane *pattern);
  virtual ~ComplexFilterPatternDeGrid(void);
  virtual gboolean skipBlock();
protected:
  virtual void processNoSharpen(ComplexBlock* block);
  virtual void processSharpen(ComplexBlock* block);
  float sigmaSquaredNoiseNormed;
  FloatImagePlane *pattern;
};

}} // namespace RawStudio::FFTFilter

#endif // complexfilter_h__