1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
/** @file
* This is the header file for the FFTTools class.
* @author André Offringa <offringa@gmail.com>
*/
#ifndef FFTTOOLS_H
#define FFTTOOLS_H
#include "../structures/image2d.h"
#include "../structures/samplerow.h"
/**
* This is a wrapper around the fftw3 library that is able to calculate fast Fourier transformations.
*/
class FFTTools{
public:
/**
* What will be outputted by the FFT method.
*/
enum FFTOutputMethod {
/**
* The transformation to real data.
*/
Real,
/**
* The transformation to imaginary data.
*/
Imaginary,
/**
* The transformation to the absolute value of real and imaginary.
*/
Absolute,
/**
* The transformation to the real and the imaginary data.
*/
Both
};
/**
* Create the FFT of an image.
* @param original The original image.
* @param method Which method to use.
* @return A new image that contains the FFT. The caller should @c delete the image when
* it is no longer required.
*/
static Image2D *CreateFFTImage(const Image2D &original, FFTOutputMethod method);
static Image2DPtr CreateFFTImagePtr(Image2DCPtr original, FFTOutputMethod method)
{
return Image2DPtr(CreateFFTImage(*original, method));
}
static void CreateFFTImage(const Image2D &real, const Image2D &imaginary, Image2D &realOut, Image2D &imaginaryOut, bool centerAfter=true, bool negate=false);
static Image2D *CreateFullImageFromFFT(const Image2D &fft);
static Image2D *CreateShiftedImageFromFFT(const Image2D &fft);
static Image2D *CreateAbsoluteImage(const Image2D &real, const Image2D &imaginary);
static Image2DPtr CreateAbsoluteImage(Image2DCPtr real, Image2DCPtr imaginary)
{
return Image2DPtr(CreateAbsoluteImage(*real, *imaginary));
}
static Image2DPtr CreatePhaseImage(Image2DCPtr real, Image2DCPtr imaginary);
static void FFTConvolve(const Image2D &realIn, const Image2D &imaginaryIn, const Image2D &realKernel, const Image2D &imaginaryKernel, Image2D &outReal, Image2D &outImaginary);
static void FFTConvolveFFTKernel(const Image2D &realIn, const Image2D &imaginaryIn, const Image2D &realFFTKernel, const Image2D &imaginaryFFTKernel, Image2D &outReal, Image2D &outImaginary);
static void Multiply(Image2D &left, const Image2D &right);
static void Divide(Image2D &left, const Image2D &right);
static void Multiply(Image2D &leftReal, Image2D &leftImaginary, const Image2D &rightReal, const Image2D &rightImaginary);
static void Sqrt(Image2D &image);
static void Sqrt(Image2DPtr image) { Sqrt(*image); }
static void SignedSqrt(Image2D &image);
static void SignedSqrt(Image2DPtr image) { SignedSqrt(*image); }
static void CreateHorizontalFFTImage(Image2D &real, Image2D &imaginary, bool inverse=false);
static void CreateDynamicHorizontalFFTImage(Image2DPtr real, Image2DPtr imaginary, unsigned sections, bool inverse=false);
static Image2DPtr AngularTransform(Image2DCPtr input);
static void FFT(SampleRow& realRow, SampleRow& imaginaryRow);
//static void FFTConvolve(num_t *realValues, num_t *imagValues, num_t *realKernel, num_t *imagKernel, size_t count);
private:
FFTTools() { }
~FFTTools() { };
};
#endif
|
