tangara-fw/src/audio/fatfs_audio_input.cpp

#include "fatfs_audio_input.hpp"

#include <algorithm>
#include <cstdint>
#include <memory>
#include <string>

#include "esp_heap_caps.h"
#include "freertos/portmacro.h"

#include "audio_element.hpp"
#include "chunk.hpp"
#include "stream_buffer.hpp"
#include "stream_event.hpp"
#include "stream_message.hpp"

static const char* kTag = "SRC";

namespace audio {

// 32KiB to match the minimum himen region size.
static const std::size_t kChunkSize = 1024;

FatfsAudioInput::FatfsAudioInput(std::shared_ptr<drivers::SdStorage> storage)
    : IAudioElement(),
      storage_(storage),
      current_file_(),
      is_file_open_(false) {}

FatfsAudioInput::~FatfsAudioInput() {}

auto FatfsAudioInput::HasUnprocessedInput() -> bool {
  return is_file_open_;
}

auto FatfsAudioInput::ProcessStreamInfo(const StreamInfo& info)
    -> cpp::result<void, AudioProcessingError> {
  if (is_file_open_) {
    f_close(&current_file_);
    is_file_open_ = false;
  }

  if (!info.path) {
    return cpp::fail(UNSUPPORTED_STREAM);
  }
  std::string path = *info.path;
  FRESULT res = f_open(&current_file_, path.c_str(), FA_READ);
  if (res != FR_OK) {
    return cpp::fail(IO_ERROR);
  }

  is_file_open_ = true;

  std::unique_ptr<StreamInfo> new_info = std::make_unique<StreamInfo>(info);
  new_info->chunk_size = kChunkSize;

  auto event =
      StreamEvent::CreateStreamInfo(input_events_, std::move(new_info));
  SendOrBufferEvent(std::move(event));

  return {};
}

auto FatfsAudioInput::ProcessChunk(const cpp::span<std::byte>& chunk)
    -> cpp::result<size_t, AudioProcessingError> {
  return cpp::fail(UNSUPPORTED_STREAM);
}

auto FatfsAudioInput::Process() -> cpp::result<void, AudioProcessingError> {
  if (is_file_open_) {
    auto dest_event = StreamEvent::CreateChunkData(input_events_, kChunkSize);
    UINT bytes_read = 0;

    FRESULT result =
        f_read(&current_file_, dest_event->chunk_data.raw_bytes.get(),
               kChunkSize, &bytes_read);
    if (result != FR_OK) {
      ESP_LOGE(kTag, "file I/O error %d", result);
      return cpp::fail(IO_ERROR);
    }

    dest_event->chunk_data.bytes =
        dest_event->chunk_data.bytes.first(bytes_read);
    SendOrBufferEvent(std::move(dest_event));

    if (f_eof(&current_file_)) {
      f_close(&current_file_);
      is_file_open_ = false;
    }
  }
  return {};
}

}  // namespace audio