aboutsummaryrefslogtreecommitdiffstats
path: root/common/alcomplex.h
blob: f730ba9e659ad014b549848938fb53fa5979db9f (plain)
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
#ifndef ALCOMPLEX_H
#define ALCOMPLEX_H

#include <complex>
#include <type_traits>

#include "alspan.h"

/**
 * Iterative implementation of 2-radix FFT (In-place algorithm). Sign = -1 is
 * FFT and 1 is inverse FFT. Applies the Discrete Fourier Transform (DFT) to
 * the data supplied in the buffer, which MUST BE power of two.
 */
void complex_fft(const al::span<std::complex<double>> buffer, const double sign);

/**
 * Calculate the frequency-domain response of the time-domain signal in the
 * provided buffer, which MUST BE power of two.
 */
template<size_t N>
void forward_fft(const al::span<std::complex<double>,N> buffer)
{ complex_fft(al::span<std::complex<double>>{buffer}, -1.0); }

/**
 * Calculate the time-domain signal of the frequency-domain response in the
 * provided buffer, which MUST BE power of two.
 */
template<size_t N>
void inverse_fft(const al::span<std::complex<double>,N> buffer)
{ complex_fft(al::span<std::complex<double>>{buffer}, +1.0); }

/**
 * Calculate the complex helical sequence (discrete-time analytical signal) of
 * the given input using the discrete Hilbert transform (In-place algorithm).
 * Fills the buffer with the discrete-time analytical signal stored in the
 * buffer. The buffer is an array of complex numbers and MUST BE power of two,
 * and the imaginary components should be cleared to 0.
 */
void complex_hilbert(const al::span<std::complex<double>> buffer);

#endif /* ALCOMPLEX_H */