---

fill.h

Go to the documentation of this file.

#ifndef __FFT_FILL__
#define __FFT_FILL__


namespace FFT{

  template <typename Function,typename Complex_data_2D_array>
  void fill_real_part(const Function& f,
                      Complex_data_2D_array* const array);

  template <typename Function,typename Complex_data_2D_array>
  void fill_imaginary_part(const Function& f,
                           Complex_data_2D_array* const array);



  
/*
  ###########################################################
  ###########################################################
  ###########################################################
  ##############                               ##############
  ##############  I M P L E M E N T A T I O N  ##############
  ##############                               ##############
  ###########################################################
  ###########################################################
  ###########################################################
*/



  
  template <typename Function,typename Complex_data_2D_array>
  void fill_real_part(const Function& f,
                      Complex_data_2D_array* const array){

    typedef typename Complex_data_2D_array::value_type complex_type;
    typedef float                                      real_type;
    typedef unsigned int                               size_type;

    const size_type width  = array->x_size();
    const size_type height = array->y_size();
  
    const size_type half_width  = width/2;

    for(size_type x=0;x<width;x++){
      for(size_type y=0;y<height;y++){

        const real_type X = static_cast<real_type>(x) -
          width*((x+half_width)/width);
      
        const real_type Y = static_cast<real_type>(y);
 
        (*array)(x,y) = complex_type(f(X,Y),(*array)(x,y).imag());
      }
    }
  }


  
  template <typename Function,typename Complex_data_2D_array>
  void fill_imaginary_part(const Function& f,
                           Complex_data_2D_array* const array){

    typedef typename Complex_data_2D_array::value_type complex_type;
    typedef float                                      real_type;
    typedef unsigned int                               size_type;

    const size_type width  = array->x_size();
    const size_type height = array->y_size();
  
    const size_type half_width  = width/2;

    for(size_type x=0;x<width;x++){
      for(size_type y=0;y<height;y++){

        const real_type X = static_cast<real_type>(x) -
          width*((x+half_width)/width);
      
        const real_type Y = static_cast<real_type>(y) - 0.5*height;
 
        (*array)(x,y) = complex_type((*array)(x,y).real(),f(X,Y));
      }
    }
  }

}

#endif

---