Add analyze_fft1024_F32

4 years ago · a6ccd8ece9
parent ce44604c83
commit a6ccd8ece9
3 changed files with 260 additions and 0 deletions
--- 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
--- a/analyze_fft1024_F32.cpp
+++ 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 <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
+}
--- a/analyze_fft1024_F32.h
+++ 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