diff options
author | Campbell Barton <ideasman42@gmail.com> | 2017-10-09 12:49:27 +0300 |
---|---|---|
committer | Campbell Barton <ideasman42@gmail.com> | 2017-10-09 12:49:27 +0300 |
commit | a5b4b0f21c1ae8c96e4fea9abdcfac2fab1cf300 (patch) | |
tree | 0658d8bdfb8ec03652aa04f82ee8a4d243ec6370 /extern/audaspace/src/util/FFTPlan.cpp | |
parent | d68f698cf0321477c0734474150eb4bc43c4e85f (diff) | |
parent | abcda06934aba054de8540b66b13c2bbc5f8f515 (diff) |
Merge branch '28' into custom-manipulatorscustom-manipulators
Diffstat (limited to 'extern/audaspace/src/util/FFTPlan.cpp')
-rw-r--r-- | extern/audaspace/src/util/FFTPlan.cpp | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/extern/audaspace/src/util/FFTPlan.cpp b/extern/audaspace/src/util/FFTPlan.cpp new file mode 100644 index 00000000000..5e99dbff247 --- /dev/null +++ b/extern/audaspace/src/util/FFTPlan.cpp @@ -0,0 +1,66 @@ +/******************************************************************************* +* Copyright 2015-2016 Juan Francisco Crespo Galán +* +* Licensed under the Apache License, Version 2.0 (the "License"); +* you may not use this file except in compliance with the License. +* You may obtain a copy of the License at +* +* http://www.apache.org/licenses/LICENSE-2.0 +* +* Unless required by applicable law or agreed to in writing, software +* distributed under the License is distributed on an "AS IS" BASIS, +* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +* See the License for the specific language governing permissions and +* limitations under the License. +******************************************************************************/ + +#include "util/FFTPlan.h" + +AUD_NAMESPACE_BEGIN +FFTPlan::FFTPlan(double measureTime) : + FFTPlan(DEFAULT_N, measureTime) +{ +} + +FFTPlan::FFTPlan(int n, double measureTime) : + m_N(n), m_bufferSize(((n/2)+1)*2*sizeof(fftwf_complex)) +{ + fftwf_set_timelimit(measureTime); + void* buf = fftwf_malloc(m_bufferSize); + m_fftPlanR2C = fftwf_plan_dft_r2c_1d(m_N, (float*)buf, (fftwf_complex*)buf, FFTW_EXHAUSTIVE); + m_fftPlanC2R = fftwf_plan_dft_c2r_1d(m_N, (fftwf_complex*)buf, (float*)buf, FFTW_EXHAUSTIVE); + fftwf_free(buf); +} + +FFTPlan::~FFTPlan() +{ + fftwf_destroy_plan(m_fftPlanC2R); + fftwf_destroy_plan(m_fftPlanR2C); +} + +int FFTPlan::getSize() +{ + return m_N; +} + +void FFTPlan::FFT(void* buffer) +{ + fftwf_execute_dft_r2c(m_fftPlanR2C, (float*)buffer, (fftwf_complex*)buffer); +} + +void FFTPlan::IFFT(void* buffer) +{ + fftwf_execute_dft_c2r(m_fftPlanC2R, (fftwf_complex*)buffer, (float*)buffer); +} + +void* FFTPlan::getBuffer() +{ + return fftwf_malloc(m_bufferSize); +} + +void FFTPlan::freeBuffer(void* buffer) +{ + fftwf_free(buffer); +} + +AUD_NAMESPACE_END |