parent
ce44604c83
commit
a6ccd8ece9
@ -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 <Arduino.h> |
||||||
|
#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 |
||||||
|
} |
@ -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 |
Loading…
Reference in new issue