AudioSourceIdxSDFAT.h

#pragma once
 
#include <SPI.h>
#include <SdFat.h>
 
#include "AudioToolsConfig.h"
#include "AudioLogger.h"
#include "AudioTools/Disk/AudioSource.h"
 
#define USE_SDFAT
#include "AudioTools/Disk/SDIndex.h"
 
// Try max SPI clock for an SD. Reduce SPI_CLOCK if errors occur. (40?)
#define SPI_CLOCK SD_SCK_MHZ(50)
 
namespace audio_tools {
template <typename AudioFs = SdFat32, typename AudioFile = File32>
class AudioSourceIdxSDFAT : public AudioSource {
 public:
  AudioSourceIdxSDFAT(const char *startFilePath = "/", const char *ext = ".mp3",
                      int chipSelect = PIN_CS, int speedMHz = 10,
                      bool setupIndex = true) {
    TRACED();
    LOGI("SD chipSelect: %d", chipSelect);
    LOGI("SD speedMHz: %d", speedMHz);
    LOGI("ext: %s", ext);
    p_cfg = new SdSpiConfig(chipSelect, DEDICATED_SPI, SD_SCK_MHZ(speedMHz));
    owns_cfg = true;
    start_path = startFilePath;
    exension = ext;
    setup_index = setupIndex;
  }
 
  AudioSourceIdxSDFAT(const char *startFilePath, const char *ext,
                      SdSpiConfig &config, bool setupIndex = true) {
    TRACED();
    p_cfg = &config;
    owns_cfg = false;
    start_path = startFilePath;
    exension = ext;
    setup_index = setupIndex;
  }
 
  AudioSourceIdxSDFAT(AudioFs fs, const char *startFilePath="/", const char *ext="", bool setupIndex = true){
    TRACED();
    sd = fs;
    p_cfg = nullptr;
    owns_cfg = false;
    start_path = startFilePath;
    exension = ext;
    setup_index = setupIndex;
    is_sd_setup = true;
    // since we expect an open fs we do not close it
    is_close_sd = false;
  }
 
  virtual ~AudioSourceIdxSDFAT() { end(); }
 
  virtual bool begin() override {
    TRACED();
    if (!is_sd_setup) {
      if (!sd.begin(*p_cfg)) {
        LOGE("sd.begin failed");
        return false;
      }
      is_sd_setup = true;
    }
    idx.begin(start_path, exension, file_name_pattern, setup_index);
    idx_pos = 0;
    return is_sd_setup;
  }
 
  void end() {
    if (is_sd_setup) {
#ifdef ESP32
      if (is_close_sd) sd.end();
#endif
      if (owns_cfg) delete (p_cfg);
      is_sd_setup = false;
    }
  }
 
  virtual Stream *nextStream(int offset = 1) override {
    LOGI("nextStream: %d", offset);
    return selectStream(idx_pos + offset);
  }
 
  virtual Stream *selectStream(int index) override {
    LOGI("selectStream: %d", index);
    idx_pos = index;
    return selectStream(idx[index]);
  }
 
  virtual Stream *selectStream(const char *path) override {
    file.close();
    if (path == nullptr) {
      LOGE("Filename is null")
      return nullptr;
    }
 
    // AudioFile new_file;
    if (!file.open(path, O_RDONLY)) {
      LOGE("Open error: '%s'", path);
    }
 
    LOGI("-> selectStream: %s", path);
    // file = new_file;
    return &file;
  }
 
  void setFileFilter(const char *filter) { file_name_pattern = filter; }
 
  int index() { return idx_pos; }
 
  const char *toStr() { return file_name; }
 
  // provides default setting go to the next
  virtual bool isAutoNext() { return true; }
 
  virtual void setPath(const char *p) { start_path = p; }
 
  long size() { return idx.size(); }
 
  int indexOf(const char* filename) {
    return idx.indexOf(filename);
  }
 
  const char* name(int pos) {
    return idx[pos];
  }
 
 protected:
  SdSpiConfig *p_cfg = nullptr;
  AudioFs sd;
  AudioFile file;
  SDIndex<AudioFs, AudioFile> idx{sd};
  size_t idx_pos = 0;
  char file_name[MAX_FILE_LEN];
  const char *exension = nullptr;
  const char *start_path = nullptr;
  const char *file_name_pattern = "*";
  bool setup_index = true;
  bool is_close_sd = true;
  bool is_sd_setup = false;
  int cs;
  bool owns_cfg = false;
 
  const char *getFileName(AudioFile &file) {
    static char name[MAX_FILE_LEN];
    file.getName(name, MAX_FILE_LEN);
    return name;
  }
};
 
}  // namespace audio_tools