#include <fix_fft.h>

#define AUDIOPIN 0
const int leds = 5; // Количество светодиодов
const int DLed[leds] = {9, 8, 7, 6, 5}; // Номера выводов светодиодов
const int res = 48; // "Разрешение" одного канала (от 1 до 64/leds*10)
const int lev[leds] = {6, 4, 4, 4, 4}; // Уровень срабатывания светодиодов (подбирается по уровню входа)
char im[128], data[128];
char data_avgs[leds];
int i, val, barsdiv;


void setup() {
  ADCSRA = (1<<ADEN)|(1<<ADPS0)|(0<<ADPS1)|(1<<ADPS2); // sps=500k ~f=10k
  barsdiv = 64 / leds; // Отношение 64-х столбцов спектра к количеству светодиодов
  for (i = 0; i < leds; i++) {
    pinMode(DLed[i], OUTPUT);
  }
  Serial.begin(57600);
}


void loop() {

  for (i = 0; i < 128; i++) {
    val = analogRead(AUDIOPIN);
    data[i] = val;// / 4 - 128;
    im[i] = 0;
  };


  fix_fft(data, im, 7, 0);

  for (i = 0; i < 64; i++) {
    data[i] = sqrt(data[i] * data[i] + im[i] * im[i]);  // this gets the absolute value of the values in the array, so we're only dealing with positive numbers
  };


  // Усредняем 64 полученых значения до количества светодиодов

  for (i = 0; i < leds; i++) {
    for (val = 0; val < barsdiv; val++) {
      data_avgs[i] = data_avgs[i] + data[i * barsdiv + val]; // Суммируем значения спектра для одного светодиода
    }

    data_avgs[i] = map(data_avgs[i], 0, 10 * barsdiv, 0, res);

    Serial.println(data_avgs[1],DEC); // Рисуем в serial plotter значение уровня канала

    // Let be the light!

    if (data_avgs[i] > lev[i]) {
      digitalWrite(DLed[i], 1);       // set the LED on
    } else {
      digitalWrite(DLed[i], 0);      // set the LED off
    }
  }


}