source: trunk/src/Utils/GaussSmooth1D.hh @ 1393

// -----------------------------------------------------------------------
// GaussSmooth1D.hh: Definition of GaussSmooth1D 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 GAUSSSMOOTH1D_H
#define GAUSSSMOOTH1D_H
#include <vector>
/// @brief
///  Define a Gaussian to smooth a 2D array.
/// @details
///  A simple class to define a Gaussian kernel that can be used to
///  smooth a two-dimensional array.

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

  /// @brief Specific constructor that sets up kernel.
  GaussSmooth1D(float fwhm);  

  /// @brief Define the size and the array of coefficients. 
  void   define(float fwhm); 

  /// @brief Smooth an array with the Gaussian kernel
  Type *smooth(Type *input, int dim, bool normalise=false, bool scaleByCoverage=false);  
  /// @brief Smooth an array with the Gaussian kernel, using a mask to define blank pixels
  Type *smooth(Type *input, int dim, std::vector<bool> mask, bool normalise=false, bool scaleByCoverage=false);  
  
  void   setKernFWHM(float f){kernFWHM=f;};

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

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

private:
  float  kernFWHM;     ///< The FWHM of the 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  // GAUSSSMOOTH1D_H