blob: 9acda5494c963d044011742d61dac77b09223caf [file] [log] [blame]
andrew@webrtc.org04c50982015-03-19 20:06:291/*
2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Mirko Bonadei92ea95e2017-09-15 04:47:3111#include "common_audio/real_fourier_ooura.h"
andrew@webrtc.org04c50982015-03-19 20:06:2912
andrew@webrtc.org04c50982015-03-19 20:06:2913#include <algorithm>
Yves Gerey665174f2018-06-19 13:03:0514#include <cmath>
andrew@webrtc.org04c50982015-03-19 20:06:2915
Mirko Bonadeif0d64a52020-04-17 10:25:1916#include "common_audio/third_party/ooura/fft_size_256/fft4g.h"
Mirko Bonadei92ea95e2017-09-15 04:47:3117#include "rtc_base/checks.h"
andrew@webrtc.org04c50982015-03-19 20:06:2918
19namespace webrtc {
20
21using std::complex;
22
23namespace {
24
Peter Kastingdce40cf2015-08-24 21:52:2325void Conjugate(complex<float>* array, size_t complex_length) {
andrew@webrtc.org04c50982015-03-19 20:06:2926 std::for_each(array, array + complex_length,
27 [=](complex<float>& v) { v = std::conj(v); });
28}
29
Peter Kastingdce40cf2015-08-24 21:52:2330size_t ComputeWorkIpSize(size_t fft_length) {
Yves Gerey665174f2018-06-19 13:03:0531 return static_cast<size_t>(
32 2 + std::ceil(std::sqrt(static_cast<float>(fft_length))));
andrew@webrtc.org04c50982015-03-19 20:06:2933}
34
35} // namespace
36
37RealFourierOoura::RealFourierOoura(int fft_order)
38 : order_(fft_order),
39 length_(FftLength(order_)),
40 complex_length_(ComplexLength(order_)),
41 // Zero-initializing work_ip_ will cause rdft to initialize these work
42 // arrays on the first call.
Peter Kastingdce40cf2015-08-24 21:52:2343 work_ip_(new size_t[ComputeWorkIpSize(length_)]()),
andrew@webrtc.org04c50982015-03-19 20:06:2944 work_w_(new float[complex_length_]()) {
henrikg91d6ede2015-09-17 07:24:3445 RTC_CHECK_GE(fft_order, 1);
andrew@webrtc.org04c50982015-03-19 20:06:2946}
47
Alex Loiko0520b0e2018-05-08 11:11:1248RealFourierOoura::~RealFourierOoura() = default;
49
andrew@webrtc.org04c50982015-03-19 20:06:2950void RealFourierOoura::Forward(const float* src, complex<float>* dest) const {
51 {
52 // This cast is well-defined since C++11. See "Non-static data members" at:
53 // http://en.cppreference.com/w/cpp/numeric/complex
Mirko Bonadei91df0912018-07-17 09:08:1554 auto* dest_float = reinterpret_cast<float*>(dest);
andrew@webrtc.org04c50982015-03-19 20:06:2955 std::copy(src, src + length_, dest_float);
56 WebRtc_rdft(length_, 1, dest_float, work_ip_.get(), work_w_.get());
57 }
58
59 // Ooura places real[n/2] in imag[0].
60 dest[complex_length_ - 1] = complex<float>(dest[0].imag(), 0.0f);
61 dest[0] = complex<float>(dest[0].real(), 0.0f);
62 // Ooura returns the conjugate of the usual Fourier definition.
63 Conjugate(dest, complex_length_);
64}
65
66void RealFourierOoura::Inverse(const complex<float>* src, float* dest) const {
67 {
Mirko Bonadei91df0912018-07-17 09:08:1568 auto* dest_complex = reinterpret_cast<complex<float>*>(dest);
andrew@webrtc.org04c50982015-03-19 20:06:2969 // The real output array is shorter than the input complex array by one
70 // complex element.
Peter Kastingdce40cf2015-08-24 21:52:2371 const size_t dest_complex_length = complex_length_ - 1;
andrew@webrtc.org04c50982015-03-19 20:06:2972 std::copy(src, src + dest_complex_length, dest_complex);
73 // Restore Ooura's conjugate definition.
74 Conjugate(dest_complex, dest_complex_length);
75 // Restore real[n/2] to imag[0].
Yves Gerey665174f2018-06-19 13:03:0576 dest_complex[0] =
77 complex<float>(dest_complex[0].real(), src[complex_length_ - 1].real());
andrew@webrtc.org04c50982015-03-19 20:06:2978 }
79
80 WebRtc_rdft(length_, -1, dest, work_ip_.get(), work_w_.get());
81
82 // Ooura returns a scaled version.
83 const float scale = 2.0f / length_;
84 std::for_each(dest, dest + length_, [scale](float& v) { v *= scale; });
85}
86
Alex Loiko0520b0e2018-05-08 11:11:1287int RealFourierOoura::order() const {
88 return order_;
89}
90
andrew@webrtc.org04c50982015-03-19 20:06:2991} // namespace webrtc