AudioTypes.h

#pragma once
 
#include "AudioToolsConfig.h"
#ifdef USE_TYPETRAITS
#include <limits>
#include <type_traits>
#endif
 
#include "AudioTools/CoreAudio/AudioBasic/Collections/Vector.h"
#include "AudioTools/CoreAudio/AudioLogger.h"
 
// fix compile error for ESP32 C3
#undef HZ
 
// MIN
#undef MIN
#define MIN(A, B) ((A) < (B) ? (A) : (B))
 
namespace audio_tools {
 
using sample_rate_t = uint32_t;
 
enum RxTxMode { UNDEFINED_MODE = 0, TX_MODE = 1, RX_MODE = 2, RXTX_MODE = 3 };
 
enum MemoryType { RAM, PS_RAM, FLASH_RAM };
 
static const char* RxTxModeNames[4] = {"UNDEFINED_MODE", "TX_MODE", "RX_MODE",
                                       "RXTX_MODE"};
enum TimeUnit { MS, US, HZ };
static const char* TimeUnitStr[3]{"MS", "US", "HZ"};
 
struct AudioInfo {
  sample_rate_t sample_rate = DEFAULT_SAMPLE_RATE;
  uint16_t channels = DEFAULT_CHANNELS;
  uint8_t bits_per_sample = DEFAULT_BITS_PER_SAMPLE;
 
  AudioInfo() = default;
 
  AudioInfo(sample_rate_t sampleRate, uint16_t channelCount,
            uint8_t bitsPerSample) {
    sample_rate = sampleRate;
    channels = channelCount;
    bits_per_sample = bitsPerSample;
  };
 
  AudioInfo(const AudioInfo&) = default;
 
  bool operator==(AudioInfo alt) {
    return sample_rate == alt.sample_rate && channels == alt.channels &&
           bits_per_sample == alt.bits_per_sample;
  }
 
  bool operator!=(AudioInfo alt) { return !(*this == alt); }
 
  bool equals(AudioInfo alt) { return *this == alt; }
 
  bool equalsExSampleRate(AudioInfo alt) {
    return channels == alt.channels && bits_per_sample == alt.bits_per_sample;
  }
 
  void set(AudioInfo info) {
    sample_rate = info.sample_rate;
    channels = info.channels;
    bits_per_sample = info.bits_per_sample;
  }
 
  void setAudioInfo(AudioInfo info) { set(info); }
 
  void copyFrom(AudioInfo info) { setAudioInfo(info); }
 
#ifndef SWIG
  AudioInfo& operator=(const AudioInfo& info) {
    setAudioInfo(info);
    return *this;
  }
#endif
  operator bool() {
    return sample_rate > 0 && sample_rate <= 100000 && channels > 0 && channels < 20 && bits_per_sample > 0 && bits_per_sample <= 64;
  }
 
  virtual void clear() {
    sample_rate = 0;
    channels = 0;
    bits_per_sample = 0;
  }
 
  virtual void logInfo(const char* source = "") {
    LOGI("%s sample_rate: %d / channels: %d / bits_per_sample: %d", source,
         (int)sample_rate, (int)channels, (int)bits_per_sample);
  }
};
 
class AudioInfoSupport {
 public:
  virtual void setAudioInfo(AudioInfo info) = 0;
  virtual AudioInfo audioInfo() = 0;
  virtual AudioInfo audioInfoOut() { return audioInfo(); }
};
 
class AudioInfoSource {
 public:
  virtual void addNotifyAudioChange(AudioInfoSupport& bi) {
    if (!notify_vector.contains(&bi)) notify_vector.push_back(&bi);
  }
 
  virtual bool removeNotifyAudioChange(AudioInfoSupport& bi) {
    int pos = notify_vector.indexOf(&bi);
    if (pos < 0) return false;
    notify_vector.erase(pos);
    return true;
  }
 
  virtual void clearNotifyAudioChange() { notify_vector.clear(); }
 
  void setNotifyActive(bool flag) { is_notify_active = flag; }
 
  bool isNotifyActive() { return is_notify_active; }
 
 protected:
  Vector<AudioInfoSupport*> notify_vector;
  bool is_notify_active = true;
 
  void notifyAudioChange(AudioInfo info) {
    if (isNotifyActive()) {
      for (auto n : notify_vector) {
        n->setAudioInfo(info);
      }
    }
  }
};
 
class VolumeSupport {
 public:
  virtual float volume() { return volume_value; }
  virtual bool setVolume(float volume) {
    volume_value = volume;
    return true;
  }
 
 protected:
  float volume_value = 1.0f;
};
 
class AudioWriter : public AudioInfoSupport {
 public:
  virtual size_t write(const uint8_t* data, size_t len) = 0;
  virtual void setAudioInfo(AudioInfo from) = 0;
  virtual void setOutput(Print& out_stream) = 0;
  virtual operator bool() = 0;
  virtual bool begin() = 0;
  virtual bool begin(AudioInfo info) {
    setAudioInfo(info);
    return begin();
  }
  virtual void end() = 0;
 
 protected:
  void writeBlocking(Print* out, uint8_t* data, size_t len) {
    TRACED();
    int open = len;
    int written = 0;
    while (open > 0) {
      int result = out->write(data + written, open);
      open -= result;
      written += result;
    }
  }
};
 
class AudioTime {
 public:
  static uint32_t toTimeUs(uint32_t samplingRate, uint8_t limit = 10) {
    uint32_t result = 1000000l / samplingRate;
    if (1000000l % samplingRate != 0) {
      result++;
    }
    if (result <= limit) {
      LOGW("Time for samplingRate %u -> %u is < %u μs - we rounded up",
           (unsigned int)samplingRate, (unsigned int)result,
           (unsigned int)limit);
      result = limit;
    }
    return result;
  }
 
  static uint32_t toBytes(uint32_t millis, AudioInfo info) {
    size_t samples = info.sample_rate * millis / 1000;
    size_t bytes = samples * info.channels * info.bits_per_sample / 8;
    return bytes;
  }
 
  static uint32_t toTimeMs(uint32_t samplingRate, uint8_t limit = 1) {
    uint32_t result = 1000l / samplingRate;
    if (1000000l % samplingRate != 0) {
      result++;
    }
    if (result <= limit) {
      LOGW("Time for samplingRate %u -> %u is < %u μs - we rounded up",
           (unsigned int)samplingRate, (unsigned int)result,
           (unsigned int)limit);
      result = limit;
    }
    return result;
  }
 
  static float toRateUs(uint32_t time_us) { return 1000000.0 / time_us; }
 
  static float toRateMs(uint32_t time_ms) { return 1000.0 / time_ms; }
};
 
template <typename T>
inline T mapT(T x, T in_min, T in_max, T out_min, T out_max) {
  return (x - in_min) * (out_max - out_min) / (in_max - in_min) + out_min;
}
 
class NumberConverter {
 public:
  static int64_t maxValue(int value_bits_per_sample) {
    switch (value_bits_per_sample) {
      case 8:
        return 127;
      case 16:
        return 32767;
      case 24:
        return 8388607;
      case 32:
        return 2147483647;
    }
    return 32767;
  }
 
  template <typename T>
  static float maxValueT() {
#ifdef USE_TYPETRAITS
    // int24_t uses 4 bytes instead of 3!
    // return (std::is_same<T, int24_t>::value ) ? 8388607 :
    // maxValue(sizeof(T)*8);
    return std::numeric_limits<T>::max();
#else
    return maxValue(sizeof(T) * 8);
#endif
  }
 
  template <typename T>
  static float minValueT() {
#ifdef USE_TYPETRAITS
    // int24_t uses 4 bytes instead of 3!
    // return (std::is_same<T, int24_t>::value ) ? 8388607 :
    // maxValue(sizeof(T)*8);
    return std::numeric_limits<T>::min();
#else
    return -maxValue(sizeof(T) * 8);
#endif
  }
 
  template <typename T>
  static T clipT(float value) {
    float mv = maxValueT<T>();
    if (value > mv) {
      return mv;
    } else if (value < -mv) {
      return -mv;
    }
    return value;
  }
 
  inline static int32_t clip(float value, int bits) {
    float mv = maxValue(bits);
    if (value > mv) {
      return mv;
    } else if (value < -mv) {
      return -mv;
    }
    return value;
  }
 
  template <typename T>
  static float toFloatT(T value) {
    return static_cast<float>(value) / maxValueT<T>();
  }
 
  template <typename T>
  static T fromFloatT(float value) {
    return value * maxValueT<T>();
  }
 
  inline static float toFloat(int32_t value, int bits) {
    return static_cast<float>(value) / maxValue(bits);
  }
 
  inline static int32_t fromFloat(float value, int bits) {
    return clip(value * maxValue(bits), bits);
  }
 
  template <typename FromT, typename ToT>
  static ToT convert(FromT value) {
    float value1 = value;
    float minTo = minValueT<ToT>();
    float maxTo = maxValueT<ToT>();
    float maxFrom = maxValueT<FromT>();
    float minFrom = minValueT<FromT>();
 
    if (maxTo - minTo > 1.0f || maxFrom - minFrom > 1.0f) {
      return mapT<float>(value1, minFrom, maxFrom, minTo, maxTo);
    }
 
    return value1 * maxValueT<ToT>() / maxValueT<FromT>();
  }
 
  template <typename FromT, typename ToT>
  static void convertArray(FromT* from, ToT* to, int samples,
                           float vol = 1.0f) {
    float factor = static_cast<float>(maxValueT<ToT>()) / maxValueT<FromT>();
    float vol_factor = factor * vol;
    for (int j = 0; j < samples; j++) {
      to[j] = clipT<ToT>(vol_factor * convert<FromT, ToT>(from[j]));
    }
  }
};
 
#if defined(USE_I2S)
 
enum I2SFormat {
  I2S_STD_FORMAT,
  I2S_LSB_FORMAT,
  I2S_MSB_FORMAT,
  I2S_PHILIPS_FORMAT,
  I2S_RIGHT_JUSTIFIED_FORMAT,
  I2S_LEFT_JUSTIFIED_FORMAT,
  I2S_PCM,
};
 
static const char* i2s_formats[] = {"I2S_STD_FORMAT",
                                    "I2S_LSB_FORMAT",
                                    "I2S_MSB_FORMAT",
                                    "I2S_PHILIPS_FORMAT",
                                    "I2S_RIGHT_JUSTIFIED_FORMAT",
                                    "I2S_LEFT_JUSTIFIED_FORMAT",
                                    "I2S_PCM"};
 
enum I2SSignalType {
  Digital,
  Analog,
  PDM,
  TDM,
};
 
static const char* i2s_signal_types[] = {"Digital", "Analog", "PDM", "TDM"};
 
#endif
 
template <typename T>
T readSample(Stream* p_stream) {
  T result = 0;
  uint8_t* p_result = (uint8_t*)&result;
  int open = sizeof(T);
  int total = 0;
  // copy missing data
  while (open > 0) {
    int read = p_stream->readBytes(p_result + total, open);
    open -= read;
    total += read;
  }
  return result;
}
 
template <typename T>
size_t readSamples(Stream* p_stream, T* data, int samples,
                   int retryCount = -1) {
  uint8_t* p_result = (uint8_t*)data;
  int open = samples * sizeof(T);
  int total = 0;
  int count0 = 0;
  // copy missing data - we abort after 5 x read returned 0 bytes
  while (open > 0) {
    int read = p_stream->readBytes(p_result + total, open);
    open -= read;
    total += read;
    if (read == 0) {
      count0++;
      delay(1);
    } else {
      count0 = 0;
    }
    // abort loop
    if (retryCount >= 0 && count0 > retryCount) break;
  }
  // convert bytes to samples
  return total / sizeof(T);
}
 
template <typename T, class P>
size_t writeDataT(P* p_out, T* data, int samples, int maxSamples = 512) {
  uint8_t* p_result = (uint8_t*)data;
  int open = samples * sizeof(T);
  int total = 0;
  // copy missing data
  while (open > 0) {
    int to_write = min(open, static_cast<int>(maxSamples * sizeof(T)));
    int written = p_out->write(p_result + total, to_write);
    open -= written;
    total += written;
  }
  // convert bytes to samples
  return total / sizeof(T);
}
 
template <typename T>
size_t writeData(Print* p_out, T* data, int samples, int maxSamples = 512) {
  return writeDataT<T, Print>(p_out, data, samples, maxSamples);
}
 
static const char* mime_pcm = "audio/pcm";
 
#ifndef IS_DESKTOP
inline void waitFor(HardwareSerial& out) { while (!out); }
#endif
 
inline void waitFor(bool& flag) { while (!flag); }
 
using Pins = Vector<int>;
 
}  // namespace audio_tools