119 lines
3.7KB

  1. /* Audio Library for Teensy 3.X
  2. * Copyright (c) 2014, Paul Stoffregen, paul@pjrc.com
  3. *
  4. * Development of this audio library was funded by PJRC.COM, LLC by sales of
  5. * Teensy and Audio Adaptor boards. Please support PJRC's efforts to develop
  6. * open source software by purchasing Teensy or other PJRC products.
  7. *
  8. * Permission is hereby granted, free of charge, to any person obtaining a copy
  9. * of this software and associated documentation files (the "Software"), to deal
  10. * in the Software without restriction, including without limitation the rights
  11. * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12. * copies of the Software, and to permit persons to whom the Software is
  13. * furnished to do so, subject to the following conditions:
  14. *
  15. * The above copyright notice, development funding notice, and this permission
  16. * notice shall be included in all copies or substantial portions of the Software.
  17. *
  18. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21. * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23. * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  24. * THE SOFTWARE.
  25. */
  26. #ifndef analyze_fft256_h_
  27. #define analyze_fft256_h_
  28. #include "Arduino.h"
  29. #include "AudioStream.h"
  30. #include "arm_math.h"
  31. // windows.c
  32. extern "C" {
  33. extern const int16_t AudioWindowHanning256[];
  34. extern const int16_t AudioWindowBartlett256[];
  35. extern const int16_t AudioWindowBlackman256[];
  36. extern const int16_t AudioWindowFlattop256[];
  37. extern const int16_t AudioWindowBlackmanHarris256[];
  38. extern const int16_t AudioWindowNuttall256[];
  39. extern const int16_t AudioWindowBlackmanNuttall256[];
  40. extern const int16_t AudioWindowWelch256[];
  41. extern const int16_t AudioWindowHamming256[];
  42. extern const int16_t AudioWindowCosine256[];
  43. extern const int16_t AudioWindowTukey256[];
  44. }
  45. class AudioAnalyzeFFT256 : public AudioStream
  46. {
  47. public:
  48. AudioAnalyzeFFT256() : AudioStream(1, inputQueueArray),
  49. window(AudioWindowHanning256), count(0), outputflag(false) {
  50. arm_cfft_radix4_init_q15(&fft_inst, 256, 0, 1);
  51. #if AUDIO_BLOCK_SAMPLES == 128
  52. prevblock = NULL;
  53. naverage = 8;
  54. #elif AUDIO_BLOCK_SAMPLES == 64
  55. prevblocks[0] = NULL;
  56. prevblocks[1] = NULL;
  57. prevblocks[2] = NULL;
  58. #endif
  59. }
  60. bool available() {
  61. if (outputflag == true) {
  62. outputflag = false;
  63. return true;
  64. }
  65. return false;
  66. }
  67. float read(unsigned int binNumber) {
  68. if (binNumber > 127) return 0.0;
  69. return (float)(output[binNumber]) * (1.0 / 16384.0);
  70. }
  71. float read(unsigned int binFirst, unsigned int binLast) {
  72. if (binFirst > binLast) {
  73. unsigned int tmp = binLast;
  74. binLast = binFirst;
  75. binFirst = tmp;
  76. }
  77. if (binFirst > 127) return 0.0;
  78. if (binLast > 127) binLast = 127;
  79. uint32_t sum = 0;
  80. do {
  81. sum += output[binFirst++];
  82. } while (binFirst <= binLast);
  83. return (float)sum * (1.0 / 16384.0);
  84. }
  85. void averageTogether(uint8_t n) {
  86. #if AUDIO_BLOCK_SAMPLES == 128
  87. if (n == 0) n = 1;
  88. naverage = n;
  89. #endif
  90. }
  91. void windowFunction(const int16_t *w) {
  92. window = w;
  93. }
  94. virtual void update(void);
  95. uint16_t output[128] __attribute__ ((aligned (4)));
  96. private:
  97. const int16_t *window;
  98. #if AUDIO_BLOCK_SAMPLES == 128
  99. audio_block_t *prevblock;
  100. #elif AUDIO_BLOCK_SAMPLES == 64
  101. audio_block_t *prevblocks[3];
  102. #endif
  103. int16_t buffer[512] __attribute__ ((aligned (4)));
  104. #if AUDIO_BLOCK_SAMPLES == 128
  105. uint32_t sum[128];
  106. uint8_t naverage;
  107. #endif
  108. uint8_t count;
  109. volatile bool outputflag;
  110. audio_block_t *inputQueueArray[1];
  111. arm_cfft_radix4_instance_q15 fft_inst;
  112. };
  113. #endif