Bogushenko Sasha / Mbed 2 deprecated fir_f32

огшор

Dependencies:   CMSIS_DSP_401 mbed

Fork of fir_f32 by Nikolay Sergeev

main.cpp@5:bd3887263120, 2014-11-13 (annotated)

Committer:
Sasha_2004
Date:
Thu Nov 13 17:39:05 2014 +0000
Revision:
5:bd3887263120
Parent:
4:397e8f4699df 
????

Who changed what in which revision?

UserRevisionLine numberNew contents of line
Sergeev 0:b712470aea1d 1 #include "mbed.h"
Sergeev 1:9b1df0b2507d 2 #include <ctype.h>
Sergeev 0:b712470aea1d 3 #include "arm_math.h"
Sergeev 1:9b1df0b2507d 4 #include "arm_const_structs.h"
Sergeev 1:9b1df0b2507d 5
Sergeev 1:9b1df0b2507d 6 void arm_cfft_f32(
Sergeev 1:9b1df0b2507d 7 const arm_cfft_instance_f32 * S,
Sergeev 1:9b1df0b2507d 8 float32_t * p1,
Sergeev 1:9b1df0b2507d 9 uint8_t ifftFlag,
Sergeev 1:9b1df0b2507d 10 uint8_t bitReverseFlag);
Sergeev 0:b712470aea1d 11
Sasha_2004 5:bd3887263120 12 Serial pc(USBTX, USBRX);//:D ПРИВЕТ!!!!
Sasha_2004 5:bd3887263120 13
Sasha_2004 5:bd3887263120 14
Sergeev 1:9b1df0b2507d 15
Sergeev 1:9b1df0b2507d 16 AnalogIn left(A2);
Sergeev 1:9b1df0b2507d 17 AnalogIn right(A3);
Sasha_2004 5:bd3887263120 18 AnalogIn center(A3);
Sergeev 1:9b1df0b2507d 19
Sergeev 1:9b1df0b2507d 20 int SAMPLE_RATE_HZ = 40000; // Sample rate of the audio in hertz.
Sergeev 1:9b1df0b2507d 21 const int FFT_SIZE = 16; // Size of the FFT.
Sergeev 1:9b1df0b2507d 22
Sergeev 1:9b1df0b2507d 23 const static arm_cfft_instance_f32 *S;
Sergeev 3:1f56b8f439a1 24 //static arm_cfft_radix2_instance_f32 *S;
Sergeev 1:9b1df0b2507d 25 Ticker samplingTimer;
Sergeev 1:9b1df0b2507d 26 float samples[FFT_SIZE*2];
Sasha_2004 5:bd3887263120 27 float samples2[FFT_SIZE*2];
Sasha_2004 5:bd3887263120 28 float samples3[FFT_SIZE*2];
Sasha_2004 5:bd3887263120 29
Sergeev 1:9b1df0b2507d 30 float magnitudes[FFT_SIZE];
Sasha_2004 5:bd3887263120 31 float magnitudes2[FFT_SIZE];
Sasha_2004 5:bd3887263120 32 float magnitudes3[FFT_SIZE];
Sergeev 1:9b1df0b2507d 33 int sampleCounter = 0;
Sasha_2004 5:bd3887263120 34
Sergeev 1:9b1df0b2507d 35
Sergeev 1:9b1df0b2507d 36 void samplingCallback()
Sergeev 3:1f56b8f439a1 37 {
Sergeev 1:9b1df0b2507d 38 // Read from the ADC and store the sample data
Sergeev 3:1f56b8f439a1 39 samples[sampleCounter] = 1000*left.read();
Sasha_2004 5:bd3887263120 40 samples2[sampleCounter] = 1000*left.read();
Sasha_2004 5:bd3887263120 41 samples3[sampleCounter] = 1000*left.read();
Sergeev 1:9b1df0b2507d 42 // Complex FFT functions require a coefficient for the imaginary part of the input.
Sergeev 1:9b1df0b2507d 43 // Since we only have real data, set this coefficient to zero.
Sergeev 1:9b1df0b2507d 44 samples[sampleCounter+1] = 0.0;
Sasha_2004 5:bd3887263120 45 samples2[sampleCounter+1] = 0.0;
Sasha_2004 5:bd3887263120 46 samples3[sampleCounter+1] = 0.0;
Sergeev 1:9b1df0b2507d 47 // Update sample buffer position and stop after the buffer is filled
Sergeev 3:1f56b8f439a1 48
Sergeev 1:9b1df0b2507d 49 sampleCounter += 2;
Sergeev 2:707eb5bc5dfc 50
Sergeev 1:9b1df0b2507d 51 if (sampleCounter >= FFT_SIZE*2) {
Sergeev 1:9b1df0b2507d 52 samplingTimer.detach();
Sergeev 1:9b1df0b2507d 53 }
Sergeev 1:9b1df0b2507d 54 }
Sergeev 1:9b1df0b2507d 55
Sergeev 1:9b1df0b2507d 56 void samplingBegin()
Sergeev 1:9b1df0b2507d 57 {
Sergeev 1:9b1df0b2507d 58 // Reset sample buffer position and start callback at necessary rate.
Sergeev 1:9b1df0b2507d 59 sampleCounter = 0;
Sergeev 3:1f56b8f439a1 60 samplingTimer.attach_us(&samplingCallback, (float)(1000000/SAMPLE_RATE_HZ));
Sergeev 1:9b1df0b2507d 61 }
Sergeev 1:9b1df0b2507d 62
Sergeev 1:9b1df0b2507d 63 bool samplingIsDone()
Sergeev 1:9b1df0b2507d 64 {
Sergeev 1:9b1df0b2507d 65 return sampleCounter >= FFT_SIZE*2;
Sergeev 0:b712470aea1d 66 }
Sergeev 1:9b1df0b2507d 67
Sergeev 1:9b1df0b2507d 68 int main()
Sergeev 0:b712470aea1d 69 {
Sergeev 1:9b1df0b2507d 70 // Set up serial port.
Sergeev 3:1f56b8f439a1 71 //pc.baud (38400);
Sergeev 1:9b1df0b2507d 72
Sergeev 1:9b1df0b2507d 73 // Init arm_ccft_32
Sergeev 1:9b1df0b2507d 74 switch (FFT_SIZE)
Sergeev 1:9b1df0b2507d 75 {
Sergeev 1:9b1df0b2507d 76 case 16:
Sergeev 1:9b1df0b2507d 77 S = & arm_cfft_sR_f32_len16;
Sergeev 1:9b1df0b2507d 78 break;
Sergeev 1:9b1df0b2507d 79 case 32:
Sergeev 1:9b1df0b2507d 80 S = & arm_cfft_sR_f32_len32;
Sergeev 1:9b1df0b2507d 81 break;
Sergeev 1:9b1df0b2507d 82 case 64:
Sergeev 1:9b1df0b2507d 83 S = & arm_cfft_sR_f32_len64;
Sergeev 1:9b1df0b2507d 84 break;
Sergeev 1:9b1df0b2507d 85 case 128:
Sergeev 1:9b1df0b2507d 86 S = & arm_cfft_sR_f32_len128;
Sergeev 1:9b1df0b2507d 87 break;
Sergeev 1:9b1df0b2507d 88 case 256:
Sergeev 1:9b1df0b2507d 89 S = & arm_cfft_sR_f32_len256;
Sergeev 1:9b1df0b2507d 90 break;
Sergeev 1:9b1df0b2507d 91 case 512:
Sergeev 1:9b1df0b2507d 92 S = & arm_cfft_sR_f32_len512;
Sergeev 1:9b1df0b2507d 93 break;
Sergeev 1:9b1df0b2507d 94 case 1024:
Sergeev 1:9b1df0b2507d 95 S = & arm_cfft_sR_f32_len1024;
Sergeev 1:9b1df0b2507d 96 break;
Sergeev 1:9b1df0b2507d 97 case 2048:
Sergeev 1:9b1df0b2507d 98 S = & arm_cfft_sR_f32_len2048;
Sergeev 1:9b1df0b2507d 99 break;
Sergeev 1:9b1df0b2507d 100 case 4096:
Sergeev 1:9b1df0b2507d 101 S = & arm_cfft_sR_f32_len4096;
Sergeev 1:9b1df0b2507d 102 break;
Sergeev 1:9b1df0b2507d 103 }
Sergeev 1:9b1df0b2507d 104 float maxValue = 0.0f;
Sasha_2004 5:bd3887263120 105 float maxValue2 = 0.0f;
Sasha_2004 5:bd3887263120 106 float maxValue3 = 0.0f;
Sasha_2004 5:bd3887263120 107
Sergeev 1:9b1df0b2507d 108 unsigned int testIndex = 0;
Sasha_2004 5:bd3887263120 109 unsigned int testIndex2 = 0;
Sasha_2004 5:bd3887263120 110 unsigned int testIndex3 = 0;
Sergeev 1:9b1df0b2507d 111
Sergeev 1:9b1df0b2507d 112 // Begin sampling audio
Sergeev 1:9b1df0b2507d 113 samplingBegin();
Sasha_2004 5:bd3887263120 114
Sasha_2004 5:bd3887263120 115 pc.printf(" MAX value a");
Sergeev 1:9b1df0b2507d 116
Sergeev 1:9b1df0b2507d 117 while(1)
Sergeev 1:9b1df0b2507d 118 {
Sergeev 1:9b1df0b2507d 119 // Calculate FFT if a full sample is available.
Sergeev 1:9b1df0b2507d 120 if (samplingIsDone())
Sergeev 3:1f56b8f439a1 121 {
Sergeev 1:9b1df0b2507d 122 // Run FFT on sample data.
Sergeev 1:9b1df0b2507d 123 //arm_cfft_radix2_f32(arm_cfft_radix2_instance_f32*S, samples);
Sergeev 1:9b1df0b2507d 124 arm_cfft_f32(S, samples, 0, 1);
Sasha_2004 5:bd3887263120 125 samples[0]=0;
Sasha_2004 5:bd3887263120 126 samples2[0]=0;
Sasha_2004 5:bd3887263120 127 samples3[0]=0;
Sasha_2004 5:bd3887263120 128
Sergeev 3:1f56b8f439a1 129 /* Initialize the CFFT/CIFFT module */
Sergeev 3:1f56b8f439a1 130 //arm_cfft_radix2_init_f32(S, 128, 0, 1);
Sergeev 3:1f56b8f439a1 131 //arm_cfft_radix2_f32(S, samples);
Sasha_2004 5:bd3887263120 132 //for(int i = 0;i < FFT_SIZE*2;++i)
Sasha_2004 5:bd3887263120 133 // pc.printf(" Samples[%d]: %8.2f ",i,samples[i]);
Sasha_2004 5:bd3887263120 134 //pc.printf("\r\n");
Sergeev 1:9b1df0b2507d 135 // Calculate magnitude of complex numbers output by the FFT.
Sergeev 1:9b1df0b2507d 136 arm_cmplx_mag_f32(samples, magnitudes, FFT_SIZE);
Sasha_2004 5:bd3887263120 137 arm_cmplx_mag_f32(samples2, magnitudes2, FFT_SIZE);
Sasha_2004 5:bd3887263120 138 arm_cmplx_mag_f32(samples3, magnitudes3, FFT_SIZE);
Sasha_2004 5:bd3887263120 139
Sasha_2004 5:bd3887263120 140 //for(int i = 0;i < FFT_SIZE;++i)
Sasha_2004 5:bd3887263120 141 // pc.printf(" Magnitude: %d = %8.2f ;", i, magnitudes[i]);
Sasha_2004 5:bd3887263120 142 //pc.printf(" \r\n");
Sergeev 1:9b1df0b2507d 143 arm_max_f32(magnitudes, FFT_SIZE, &maxValue, &testIndex);
Sasha_2004 5:bd3887263120 144 arm_max_f32(magnitudes2, FFT_SIZE, &maxValue2, &testIndex2);
Sasha_2004 5:bd3887263120 145 arm_max_f32(magnitudes3, FFT_SIZE, &maxValue3, &testIndex3);
Sasha_2004 5:bd3887263120 146 pc.printf(" MAX value at magnitudes 1[%d] : %+8.2f\r\n", testIndex, maxValue);//Я НЯШКА А АНЯ ВЛАСЮК КОЗА:DDDDDDDDDDDDD
Sasha_2004 5:bd3887263120 147 //pc.printf(" MAX value at magnitudes 2[%d] : %+8.2f\r\n", testIndex2, maxValue2);
Sasha_2004 5:bd3887263120 148 //pc.printf(" MAX value at magnitudes 3[%d] : %+8.2f\r\n", testIndex2, maxValue3);
Sergeev 1:9b1df0b2507d 149 // Wait for user confirmation to restart audio sampling.
Sergeev 1:9b1df0b2507d 150 //pc.getc();
Sergeev 1:9b1df0b2507d 151 wait(1);
Sergeev 1:9b1df0b2507d 152 samplingBegin();
Sergeev 1:9b1df0b2507d 153 }
Sergeev 1:9b1df0b2507d 154 }
Sergeev 0:b712470aea1d 155 }