source: trunk/src/Utils/GaussSmooth.hh @ 526

// -----------------------------------------------------------------------
// GaussSmooth.hh: Definition of GaussSmooth class, used to smooth a
//                 2D image with a Gaussian kernel.
// -----------------------------------------------------------------------
// Copyright (C) 2006, Matthew Whiting, ATNF
//
// 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.
//
// Duchamp 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 Duchamp; if not, write to the Free Software Foundation,
// Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA
//
// Correspondence concerning Duchamp may be directed to:
//    Internet email: Matthew.Whiting [at] atnf.csiro.au
//    Postal address: Dr. Matthew Whiting
//                    Australia Telescope National Facility, CSIRO
//                    PO Box 76
//                    Epping NSW 1710
//                    AUSTRALIA
// -----------------------------------------------------------------------
#ifndef GAUSSSMOOTH_H
#define GAUSSSMOOTH_H

/**
 *  Define a Gaussian to smooth a 2D array.
 *
 *  A simple class to define a Gaussian kernel that can be used to
 *  smooth a two-dimensional array.
 */

template <class Type>
class GaussSmooth
{
public:
  GaussSmooth();          ///< Basic constructor: no kernel defined.
  virtual ~GaussSmooth(); ///< Destructor
  GaussSmooth(const GaussSmooth& g);
  GaussSmooth& operator=(const GaussSmooth& g);
  

  /** Specific constructor that sets up kernel.*/
  GaussSmooth(float maj, float min, float pa);  
  /** Specific constructor that sets up kernel: assuming circular gaussian.*/
  GaussSmooth(float maj);  

  /** Define the size and the array of coefficients. */
  void   define(float maj, float min, float pa); 

  /** Smooth an array with the Gaussian kernel*/
  Type *smooth(Type *input, int xdim, int ydim, bool scaleByCoverage=false);  
  /** Smooth an array with the Gaussian kernel, using a mask to define
      blank pixels*/
  Type *smooth(Type *input, int xdim, int ydim, bool *mask, bool scaleByCoverage=false);  
  
  void   setKernMaj(float f){kernMaj=f;};
  void   setKernMin(float f){kernMin=f;};
  void   setKernPA(float f){kernPA=f;};

  Type   getKernelPt(int i){return kernel[i];};
  Type  *getKernel(){return kernel;};

  int    getKernWidth(){return kernWidth;};
  float  getStddevScale(){return stddevScale;};

private:
  float  kernMaj;      ///< The FWHM of the major axis of the
                       ///   elliptical Gaussian.
  float  kernMin;      ///< The FWHM of the minor axis of the
                       ///   elliptical Gaussian.
  float  kernPA;       ///< The position angle of the elliptical
                       ///   Gaussian.
  int    kernWidth;    ///< The width of the kernel (in pixels).
  float  stddevScale;  ///< The factor by which the rms of the input
                       ///   array gets scaled by (assuming iid
                       ///   normally)
  Type  *kernel;       ///< The coefficients of the smoothing kernel
  bool   allocated;    ///< Have the coefficients been allocated in memory?

};

#endif  // HANNING_H