From a6ccd8ece9186cfd12e2b2ce7925f751ea526cf0 Mon Sep 17 00:00:00 2001 From: boblark Date: Thu, 18 Feb 2021 10:46:11 -0800 Subject: [PATCH] Add analyze_fft1024_F32 --- OpenAudio_ArduinoLibrary.h | 1 + analyze_fft1024_F32.cpp | 137 +++++++++++++++++++++++++++++++++++++ analyze_fft1024_F32.h | 122 +++++++++++++++++++++++++++++++++ 3 files changed, 260 insertions(+) create mode 100644 analyze_fft1024_F32.cpp create mode 100644 analyze_fft1024_F32.h diff --git a/OpenAudio_ArduinoLibrary.h b/OpenAudio_ArduinoLibrary.h index bfa29a5..4b66b68 100644 --- a/OpenAudio_ArduinoLibrary.h +++ b/OpenAudio_ArduinoLibrary.h @@ -27,6 +27,7 @@ #include "synth_waveform_F32.h" #include "synth_whitenoise_f32.h" #include "synth_GaussianWhiteNoise_F32.h" +#include "analyze_fft1024_F32.h" #include "analyze_peak_f32.h" #include "analyze_rms_f32.h" // #include "control_tlv320aic3206.h" collides much with Teensy Audio diff --git a/analyze_fft1024_F32.cpp b/analyze_fft1024_F32.cpp new file mode 100644 index 0000000..814be64 --- /dev/null +++ b/analyze_fft1024_F32.cpp @@ -0,0 +1,137 @@ +/* analyze_fft1024_F32.cpp Converted from Teensy I16 Audio Library + * This version uses floating point F32 inputs + * + * Conversion parts copyright (c) Bob Larkin 2021 + * + * Audio Library for Teensy 3.X + * Copyright (c) 2014, Paul Stoffregen, paul@pjrc.com + * + * Development of this audio library was funded by PJRC.COM, LLC by sales of + * Teensy and Audio Adaptor boards. Please support PJRC's efforts to develop + * open source software by purchasing Teensy or other PJRC products. + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice, development funding notice, and this permission + * notice shall be included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN + * THE SOFTWARE. + */ + +#include +#include "analyze_fft1024_F32.h" +// #include "utility/dspinst.h" + +// Move audio data from an audio_block_f32_t to the FFT instance buffer. +static void copy_to_fft_buffer(void *destination, const void *source) +{ + const float *src = (const float *)source; + float *dst = (float *)destination; + + for (int i=0; i < AUDIO_BLOCK_SAMPLES; i++) { + *dst++ = *src++; // real sample + *dst++ = 0.0f; // 0 for Imag + } +} + +static void apply_window_to_fft_buffer(void *buffer, const void *window) +{ + float *buf = (float *)buffer; // 0th entry is real (do window) 1th is imag + const float *win = (float *)window; + + for (int i=0; i < 1024; i++) + buf[2*i] *= *win++; +} + +void AudioAnalyzeFFT1024_F32::update(void) +{ + audio_block_f32_t *block; + + block = receiveReadOnly_f32(); + if (!block) return; + +// What all does 7EM cover?? +#if defined(__ARM_ARCH_7EM__) + switch (state) { + case 0: + blocklist[0] = block; + state = 1; + break; + case 1: + blocklist[1] = block; + state = 2; + break; + case 2: + blocklist[2] = block; + state = 3; + break; + case 3: + blocklist[3] = block; + state = 4; + break; + case 4: + blocklist[4] = block; + state = 5; + break; + case 5: + blocklist[5] = block; + state = 6; + break; + case 6: + blocklist[6] = block; + state = 7; + break; + case 7: + blocklist[7] = block; + copy_to_fft_buffer(fft_buffer+0x000, blocklist[0]->data); + copy_to_fft_buffer(fft_buffer+0x100, blocklist[1]->data); + copy_to_fft_buffer(fft_buffer+0x200, blocklist[2]->data); + copy_to_fft_buffer(fft_buffer+0x300, blocklist[3]->data); + copy_to_fft_buffer(fft_buffer+0x400, blocklist[4]->data); + copy_to_fft_buffer(fft_buffer+0x500, blocklist[5]->data); + copy_to_fft_buffer(fft_buffer+0x600, blocklist[6]->data); + copy_to_fft_buffer(fft_buffer+0x700, blocklist[7]->data); + + if (window) + apply_window_to_fft_buffer(fft_buffer, window); + + arm_cfft_radix4_f32(&fft_inst, fft_buffer); + + for (int i=0; i < 512; i++) { + float magsq = fft_buffer[2*i]*fft_buffer[2*i] + fft_buffer[2*i+1]*fft_buffer[2*i+1]; + if(outputType==FFT_RMS) + output[i] = sqrtf(magsq); + else if(outputType==FFT_POWER) + output[i] = magsq; + else if(outputType==FFT_DBFS) + output[i] = 10.0f*log10f(magsq)-54.1854f; // Scaled to FS sine wave + else + output[i] = 0.0f; + } + outputflag = true; + release(blocklist[0]); + release(blocklist[1]); + release(blocklist[2]); + release(blocklist[3]); + blocklist[0] = blocklist[4]; + blocklist[1] = blocklist[5]; + blocklist[2] = blocklist[6]; + blocklist[3] = blocklist[7]; + state = 4; + break; + } +#else + release(block); +#endif +} diff --git a/analyze_fft1024_F32.h b/analyze_fft1024_F32.h new file mode 100644 index 0000000..83217a4 --- /dev/null +++ b/analyze_fft1024_F32.h @@ -0,0 +1,122 @@ +/* analyze_fft1024_F32.h Converted from Teensy I16 Audio Library + * + * Audio Library for Teensy 3.X + * Copyright (c) 2014, Paul Stoffregen, paul@pjrc.com + * + * Development of this audio library was funded by PJRC.COM, LLC by sales of + * Teensy and Audio Adaptor boards. Please support PJRC's efforts to develop + * open source software by purchasing Teensy or other PJRC products. + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice, development funding notice, and this permission + * notice shall be included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN + * THE SOFTWARE. + */ + + /* Moved directly I16 to F32. Bob Larkin 16 Feb 2021 + * Only Hann window for now. + */ + +#ifndef analyze_fft1024_F32_h_ +#define analyze_fft1024_F32_h_ + +#include "Arduino.h" +#include "AudioStream_F32.h" +#include "arm_math.h" +#define FFT_RMS 0 +#define FFT_POWER 1 +#define FFT_DBFS 2 + +/* // windows.c +extern "C" { +extern const int16_t AudioWindowHanning1024[]; +extern const int16_t AudioWindowBartlett1024[]; +extern const int16_t AudioWindowBlackman1024[]; +extern const int16_t AudioWindowFlattop1024[]; +extern const int16_t AudioWindowBlackmanHarris1024[]; +extern const int16_t AudioWindowNuttall1024[]; +extern const int16_t AudioWindowBlackmanNuttall1024[]; +extern const int16_t AudioWindowWelch1024[]; +extern const int16_t AudioWindowHamming1024[]; +extern const int16_t AudioWindowCosine1024[]; +extern const int16_t AudioWindowTukey1024[]; ) +*/ + +class AudioAnalyzeFFT1024_F32 : public AudioStream_F32 { +//GUI: inputs:1, outputs:0 //this line used for automatic generation of GUI node +//GUI: shortName:AnalyzeFFT1024 +public: + AudioAnalyzeFFT1024_F32() : AudioStream_F32(1, inputQueueArray) { + arm_cfft_radix4_init_f32(&fft_inst, 1024, 0, 1); + useHanningWindow(); // Revisit this for more flexibility <<<<< + } + bool available() { + if (outputflag == true) { + outputflag = false; + return true; + } + return false; + } + float read(unsigned int binNumber) { + if (binNumber>511 || binNumber<0) return 0.0; + return output[binNumber]; + } + float read(unsigned int binFirst, unsigned int binLast) { + if (binFirst > binLast) { + unsigned int tmp = binLast; + binLast = binFirst; + binFirst = tmp; + } + if (binFirst > 511) return 0.0; + if (binLast > 511) binLast = 511; + uint32_t sum = 0; + do { + sum += output[binFirst++]; + } while (binFirst <= binLast); + return (float)sum * (1.0 / 16384.0); + } + + void useHanningWindow(void) { + for (int i=0; i < 1024; i++) { + // 2*PI/1023 = 0.006141921 + window[i] = 0.5*(1.0 - cosf(0.006141921f*(float)i)); + } + } + +// void windowFunction(const float *w) { +// window = w; +// } + + void setOutputType(int _type) { + outputType = _type; + } + + virtual void update(void); + + float output[512]; +private: + // void init(void); + float window[1024]; int doPrint = 0; + audio_block_f32_t *blocklist[8]; + float fft_buffer[2048]; + uint8_t state = 0; + bool outputflag = false; + audio_block_f32_t *inputQueueArray[1]; + arm_cfft_radix4_instance_f32 fft_inst; + int outputType = FFT_RMS; //Same type as I16 version has +}; + +#endif