---

fill_2D.h

Go to the documentation of this file.

#ifndef __FILL_2D__
#define __FILL_2D__

namespace Function_2D{

  template <typename Function,typename Array>
  void fill(const Function& f,
            Array* const    array,
            const bool      normalize_sum = false,
            const bool      zero_in_the_middle = false);





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




  
  template <typename Function,typename Array>
  void fill(const Function& f,
            Array* const    array,
            const bool      normalize_sum,
            const bool      zero_in_the_middle){

    typedef unsigned int size_type;
    typedef double       real_type;

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

    const size_type half_width  = width/2;
    const size_type half_height = height/2;

    real_type sum = 0;

    if (zero_in_the_middle){

      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) - 0.5*width;
          const real_type Y = static_cast<real_type>(y) - 0.5*height;

          const real_type v = f(X,Y);
          (*array)(x,y) = v;
          sum += v;
        }
      }

    }
    else{ // if (zero_in_the_middle)

      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) -
            height*((y+half_height)/height);

          const real_type v = f(X,Y);
          (*array)(x,y) = v;
          sum += v;
        }
      }
    
    } // else !zero_in_the_middle

    if (normalize_sum){

      for(size_type x=0;x<width;x++){
        for(size_type y=0;y<height;y++){
          (*array)(x,y) /= sum;
        }
      }

    }
    
  }
  
}


#endif

---