FFTHelper.h

Go to the documentation of this file.

 
#ifndef _SEFRAMEWORK_FFT_FFTHELPER_H
#define _SEFRAMEWORK_FFT_FFTHELPER_H
 
#include "SEFramework/Image/Image.h"
#include <cassert>
#include <vector>
 
namespace SourceXtractor {
 
template <typename T, template <typename> class Img>
static void copyImageToFFTWorkArea(Img<T>& origin, std::vector<T>& buffer) {
  int width  = origin.getWidth();
  int height = origin.getHeight();
  int pad    = 2 * (width / 2 + 1) - width;
  int stride = width + pad;
  assert(buffer.size() == static_cast<size_t>(height * (width + pad)));
  for (int y = 0; y < height; ++y) {
    for (int x = 0; x < width; ++x) {
      buffer[x + y * stride] = origin.getValue(x, y);
    }
  }
}
 
template <typename T, template <typename> class Img>
static void copyFFTWorkAreaToImage(std::vector<T>& buffer, Img<T>& dest, int rpad = 0, int lpad = 0, int tpad = 0, int bpad = 0,
                                   bool normalize = true) {
  const int width         = dest.getWidth();
  const int height        = dest.getHeight();
  const int padded_width  = width + rpad + lpad;
  const int padded_height = height + tpad + bpad;
  const int fftw_pad      = 2 * (padded_width / 2 + 1) - padded_width;
  const int stride        = padded_width + fftw_pad;
  const int total_size    = normalize ? padded_width * padded_height : 1;
  assert(buffer.size() == static_cast<size_t>(padded_height * (padded_width + fftw_pad)));
  for (int y = 0; y < height; ++y) {
    for (int x = 0; x < width; ++x) {
      dest.setValue(x, y, buffer[x + lpad + (y + tpad) * stride] / total_size);
    }
  }
}
 
}  // namespace SourceXtractor
 
#endif  // _SEFRAMEWORK_FFT_FFTHELPER_H