AudioEncoded.h

#pragma once
 
#include "AudioConfig.h"
#include "AudioLogger.h"
#include "AudioTools/CoreAudio/AudioIO.h"
#include "AudioTools/CoreAudio/AudioOutput.h"
#include "AudioTools/CoreAudio/AudioStreams.h"
#include "AudioTools/CoreAudio/AudioTypes.h"
#include "AudioCodecsBase.h"
 
namespace audio_tools {
 
class EncodedAudioOutput : public ModifyingOutput {
 public:
  EncodedAudioOutput() {
    active = false;
  }
 
  EncodedAudioOutput(AudioDecoder *decoder) {
    setDecoder(decoder);
    active = false;
  }
 
  EncodedAudioOutput(AudioEncoder *encoder) {
    setEncoder(encoder);
    active = false;
  }
 
  EncodedAudioOutput(AudioStream *outputStream, AudioDecoder *decoder) {
    setDecoder(decoder);
    setOutput(outputStream);
    active = false;
  }
 
  EncodedAudioOutput(AudioOutput *outputStream, AudioDecoder *decoder) {
    setDecoder(decoder);
    setOutput(outputStream);
    active = false;
  }
 
  EncodedAudioOutput(Print *outputStream, AudioDecoder *decoder) {
    setDecoder(decoder);
    setOutput(outputStream);
    active = false;
  }
 
  EncodedAudioOutput(Print *outputStream, AudioEncoder *encoder) {
    setEncoder(encoder);
    setOutput(outputStream);
    active = false;
  }
 
  EncodedAudioOutput(AudioOutput *outputStream, AudioEncoder *encoder) {
    setEncoder(encoder);
    setOutput(outputStream);
    active = false;
  }
 
  EncodedAudioOutput(AudioStream *outputStream, AudioEncoder *encoder) {
    setEncoder(encoder);
    setOutput(outputStream);
    active = false;
  }
 
  void addNotifyAudioChange(AudioInfoSupport &bi) override {
    TRACEI();
    decoder_ptr->addNotifyAudioChange(bi);
  }
 
  AudioInfo defaultConfig() {
    AudioInfo cfg;
    cfg.channels = 2;
    cfg.sample_rate = 44100;
    cfg.bits_per_sample = 16;
    return cfg;
  }
 
  virtual void setAudioInfo(AudioInfo newInfo) override {
    TRACED();
    if (this->cfg != newInfo && newInfo.channels != 0 && newInfo.sample_rate != 0) {
      this->cfg = newInfo;
      decoder_ptr->setAudioInfo(cfg);
      encoder_ptr->setAudioInfo(cfg);
    }
  }
 
  void setOutput(Print &outputStream) { setOutput(&outputStream); }
 
  void setOutput(Print *outputStream) {
    ptr_out = outputStream;
    if (decoder_ptr != nullptr) {
      decoder_ptr->setOutput(*ptr_out);
    }
    if (encoder_ptr != nullptr) {
      encoder_ptr->setOutput(*ptr_out);
    }
  }
 
  void setEncoder(AudioEncoder *encoder) {
    if (encoder == nullptr) {
      encoder = CodecNOP::instance();
    }
    encoder_ptr = encoder;
    writer_ptr = encoder;
    if (ptr_out != nullptr) {
      encoder_ptr->setOutput(*ptr_out);
    }
  }
 
  AudioEncoder *getEncoder() { return encoder_ptr; }
 
  void setDecoder(AudioDecoder *decoder) {
    if (decoder == nullptr) {
      decoder = CodecNOP::instance();
    }
    decoder_ptr = decoder;
    writer_ptr = decoder;
    if (ptr_out != nullptr) {
      decoder_ptr->setOutput(*ptr_out);
    }
  }
 
  AudioDecoder *getDecoder() { return decoder_ptr; }
 
  bool begin() override {
#if USE_AUDIO_LOGGING
    custom_log_level.set();
#endif
    TRACED();
    if (!active) {
      TRACED();
      const CodecNOP *nop = CodecNOP::instance();
      if (decoder_ptr != nop || encoder_ptr != nop) {
        active = true;
        if (!decoder_ptr->begin(cfg)) active = false;
        if (!encoder_ptr->begin(cfg)) active = false;
      } else {
        LOGW("no decoder or encoder defined");
      }
    }
#if USE_AUDIO_LOGGING
    custom_log_level.reset();
#endif
    return active;
  }
 
  virtual bool begin(AudioInfo newInfo) {
    cfg = newInfo;
    return begin();
  }
 
  void end() override {
#if USE_AUDIO_LOGGING
    custom_log_level.set();
#endif
    TRACEI();
    decoder_ptr->end();
    encoder_ptr->end();
    active = false;
#if USE_AUDIO_LOGGING
    custom_log_level.reset();
#endif
  }
 
  virtual size_t write(const uint8_t *data, size_t len) override {
    if (len == 0) {
      //LOGI("write: %d", 0);
      return 0;
    }
#if USE_AUDIO_LOGGING
    custom_log_level.set();
#endif
    LOGD("EncodedAudioOutput::write: %d", (int)len);
 
    if (writer_ptr == nullptr || data == nullptr) {
      LOGE("NPE");
      return 0;
    }
 
    if (check_available_for_write && availableForWrite() == 0) {
      return 0;
    }
 
    size_t result = writer_ptr->write(data, len);
    LOGD("EncodedAudioOutput::write: %d -> %d", (int)len, (int)result);
#if USE_AUDIO_LOGGING
    custom_log_level.reset();
#endif
    return result;
  }
 
  int availableForWrite() override {
    if (!check_available_for_write) return frame_size;
    return min(ptr_out->availableForWrite(), frame_size);
  }
 
  operator bool() { return active; }
 
  AudioDecoder &decoder() { return *decoder_ptr; }
 
  AudioEncoder &encoder() { return *encoder_ptr; }
 
#if USE_AUDIO_LOGGING
  void setLogLevel(AudioLogger::LogLevel level) { custom_log_level.set(level); }
#endif
  bool isCheckAvailableForWrite() { return check_available_for_write; }
 
  void setFrameSize(int size) { frame_size = size; }
 
 protected:
  // AudioInfo info;
  AudioDecoder *decoder_ptr = CodecNOP::instance();  // decoder
  AudioEncoder *encoder_ptr = CodecNOP::instance();  // decoder
  AudioWriter *writer_ptr = nullptr;
  Print *ptr_out = nullptr;
  bool active = false;
  bool check_available_for_write = false;
#if USE_AUDIO_LOGGING
  CustomLogLevel custom_log_level;
#endif
  int frame_size = DEFAULT_BUFFER_SIZE;
};
 
// legacy name
using EncodedAudioPrint = EncodedAudioOutput;
 
class EncodedAudioStream : public ReformatBaseStream {
 public:
  EncodedAudioStream() = default;
 
  EncodedAudioStream(AudioStream *ioStream, AudioDecoder *decoder) {
    setDecoder(decoder);
    setStream(*ioStream);
  }
 
  EncodedAudioStream(Stream *ioStream, AudioDecoder *decoder) {
    setDecoder(decoder);
    setStream(*ioStream);
  }
 
  EncodedAudioStream(AudioOutput *outputStream, AudioDecoder *decoder) {
    setOutput(*outputStream);
    setDecoder(decoder);
  }
 
  EncodedAudioStream(Print *outputStream, AudioDecoder *decoder) {
    setOutput(*outputStream);
    setDecoder(decoder);
  }
 
  EncodedAudioStream(Print *outputStream, AudioEncoder *encoder) {
    setOutput(*outputStream);
    setEncoder(encoder);
  }
 
  EncodedAudioStream(AudioDecoder *decoder) { setDecoder(decoder); }
 
  EncodedAudioStream(AudioEncoder *encoder) { setEncoder(encoder); }
 
  void setEncoder(AudioEncoder *encoder) { enc_out.setEncoder(encoder); }
 
  void setDecoder(AudioDecoder *decoder) { enc_out.setDecoder(decoder); }
 
  AudioEncoder *getEncoder() { return enc_out.getEncoder(); }
 
  AudioDecoder *getDecoder() { return enc_out.getDecoder(); }
 
  AudioDecoder &decoder() { return *getDecoder(); }
 
  AudioEncoder &encoder() { return *getEncoder(); }
 
  void setStream(Stream *stream) {
    setStream(*stream);
  }
 
  void setStream(AudioStream *stream) {
    setStream(*stream);
  }
 
  void setOutput(AudioOutput *stream) {
    setOutput(*stream);
  }
 
  void setOutput(Print *stream) {
    setOutput(*stream);
  }
 
  void setStream(AudioStream &stream) {
    ReformatBaseStream::setStream(stream);
    enc_out.setOutput(&stream);
  }
 
  void setStream(Stream &stream) {
    ReformatBaseStream::setStream(stream);
    enc_out.setOutput(&stream);
  }
 
  void setOutput(AudioOutput &stream) {
    ReformatBaseStream::setOutput(stream);
    enc_out.setOutput(&stream);
  }
 
  void setOutput(Print &out) {
    ReformatBaseStream::setOutput(out);
    enc_out.setOutput(&out);
  }
 
  AudioInfo defaultConfig() {
    AudioInfo ai;
    return ai;
  }
 
  bool begin(AudioInfo info) {
    setAudioInfo(info);
    return begin();
  }
 
  bool begin() {
    //is_output_notify = false;
    reader.setByteCountFactor(10);
    setupReader();
    ReformatBaseStream::begin();
    return enc_out.begin(audioInfo());
  }
 
  void end() {
    enc_out.end();
    reader.end();
  }
 
  int availableForWrite() { return enc_out.availableForWrite(); }
 
  size_t write(const uint8_t *data, size_t len) {
    //addNotifyOnFirstWrite();
    return enc_out.write(data, len);
  }
 
  size_t readBytes(uint8_t *data, size_t len) {
    return reader.readBytes(data, len);
  }
 
  void addNotifyAudioChange(AudioInfoSupport &bi) override {
    enc_out.addNotifyAudioChange(bi);
  }
 
  float getByteFactor() { return 1.0f; }
 
#if USE_AUDIO_LOGGING
  void setLogLevel(AudioLogger::LogLevel level) { enc_out.setLogLevel(level); }
#endif
 
  void setFrameSize(int size) { enc_out.setFrameSize(size); }
 
 protected:
  EncodedAudioOutput enc_out;
 
};
 
class AudioWriterToAudioOutput : public AudioOutputAdapter {
 public:
  void setWriter(AudioWriter *writer) { p_writer = writer; }
  size_t write(const uint8_t *data, size_t len) {
    return p_writer->write(data, len);
  };
 
 protected:
  AudioWriter *p_writer = nullptr;
};
 
class ContainerTarget {
 public:
  virtual bool begin() = 0;
  virtual void end() = 0;
  virtual void setAudioInfo(AudioInfo info) {
    if (this->info != info && info.channels != 0 && info.sample_rate != 0) {
      this->info = info;
      if (p_writer1 != nullptr) p_writer1->setAudioInfo(info);
      if (p_writer2 != nullptr) p_writer2->setAudioInfo(info);
    }
  }
  virtual size_t write(uint8_t *data, size_t size) = 0;
 
 protected:
  AudioInfo info;
  AudioWriter *p_writer1 = nullptr;
  AudioWriter *p_writer2 = nullptr;
  AudioWriterToAudioOutput print2;
  bool active = false;
};
 
class ContainerTargetPrint : public ContainerTarget {
 public:
  void setupOutput(AudioWriter *writer1, AudioWriter *writer2, Print &print) {
    p_print = &print;
    p_writer1 = writer1;
    p_writer2 = writer2;
    print2.setWriter(p_writer2);
  }
 
  void setupOutput(AudioWriter *writer1, Print &print) {
    p_print = &print;
    p_writer1 = writer1;
  }
 
  virtual bool begin() {
    if (!active) {
      active = true;
      if (p_writer2 != nullptr) {
        p_writer1->setOutput(print2);
        p_writer2->setOutput(*p_print);
        p_writer1->begin();
        p_writer2->begin();
      } else {
        p_writer1->setOutput(*p_print);
        p_writer1->begin();
      }
    }
    return true;
  }
  virtual void end() {
    if (active) {
      if (p_writer1 != nullptr) p_writer1->end();
      if (p_writer2 != nullptr) p_writer2->end();
    }
    active = false;
  }
  virtual size_t write(uint8_t *data, size_t size) {
    TRACED();
    return p_writer1->write(data, size);
  }
 
 protected:
  Print *p_print = nullptr;
  AudioWriterToAudioOutput print2;
};
 
}  // namespace audio_tools