#ifndef AudioInI2S_H
#define AudioInI2S_H

#include "Arduino.h"
#include <driver/i2s.h>

class AudioInI2S
{
public:
  AudioInI2S(int bck_pin, int ws_pin, int data_pin, int channel_pin = -1, i2s_channel_fmt_t channel_format = I2S_CHANNEL_FMT_ONLY_RIGHT);
  void read(int32_t _samples[]);
  void begin(int sample_size, int sample_rate = 44100, i2s_port_t i2s_port_number = I2S_NUM_0);

private:
  int _bck_pin;
  int _ws_pin;
  int _data_pin;
  int _channel_pin;
  i2s_channel_fmt_t _channel_format;
  int _sample_size;
  int _sample_rate;
  i2s_port_t _i2s_port_number;

  i2s_config_t _i2s_config = {
      .mode = (i2s_mode_t)(I2S_MODE_MASTER | I2S_MODE_RX),
      .sample_rate = 0, // set in begin()
      .bits_per_sample = I2S_BITS_PER_SAMPLE_32BIT,
      .channel_format = I2S_CHANNEL_FMT_ONLY_RIGHT,
      .communication_format = I2S_COMM_FORMAT_I2S,
      .intr_alloc_flags = ESP_INTR_FLAG_LEVEL1,
      .dma_buf_count = 4,
      .dma_buf_len = 0, // set in begin()
      .use_apll = false,
      .tx_desc_auto_clear = false,
      .fixed_mclk = 0};

  i2s_pin_config_t _i2s_mic_pins = {
      .bck_io_num = I2S_PIN_NO_CHANGE, // set in begin()
      .ws_io_num = I2S_PIN_NO_CHANGE,  // set in begin()
      .data_out_num = I2S_PIN_NO_CHANGE,
      .data_in_num = I2S_PIN_NO_CHANGE // set in begin()
  };
};

AudioInI2S::AudioInI2S(int bck_pin, int ws_pin, int data_pin, int channel_pin, i2s_channel_fmt_t channel_format)
{
  _bck_pin = bck_pin;
  _ws_pin = ws_pin;
  _data_pin = data_pin;
  _channel_pin = channel_pin;
  _channel_format = channel_format;
}

void AudioInI2S::begin(int sample_size, int sample_rate, i2s_port_t i2s_port_number)
{
  if (_channel_pin >= 0)
  {
    pinMode(_channel_pin, OUTPUT);
    digitalWrite(_channel_pin, _channel_format == I2S_CHANNEL_FMT_ONLY_RIGHT ? LOW : HIGH);
  }

  _sample_rate = sample_rate;
  _sample_size = sample_size;
  _i2s_port_number = i2s_port_number;

  _i2s_mic_pins.bck_io_num = _bck_pin;
  _i2s_mic_pins.ws_io_num = _ws_pin;
  _i2s_mic_pins.data_in_num = _data_pin;

  _i2s_config.sample_rate = _sample_rate;
  _i2s_config.dma_buf_len = _sample_size;
  _i2s_config.channel_format = _channel_format;

  // start up the I2S peripheral
  i2s_driver_install(_i2s_port_number, &_i2s_config, 0, NULL);
  i2s_set_pin(_i2s_port_number, &_i2s_mic_pins);
}

void AudioInI2S::read(int32_t _samples[])
{
  // read I2S stream data into the samples buffer
  size_t bytes_read = 0;
  i2s_read(_i2s_port_number, _samples, sizeof(int32_t) * _sample_size, &bytes_read, portMAX_DELAY);
  int samples_read = bytes_read / sizeof(int32_t);
}

#endif // AudioInI2S_H