You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
122 lines
4.1 KiB
122 lines
4.1 KiB
/* 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
|
|
|