CodecChain

Author: pschatzmann

src/AudioTools/AudioCodecs/AudioEncoded.h

@@ -208,6 +208,18 @@ class EncodedAudioOutput : public ModifyingOutput {
   /// defines the size of the decoded frame in bytes
   void setFrameSize(int size) { frame_size = size; }
 
+  EncodedAudioOutput& operator=(EncodedAudioOutput const& src) {
+    decoder_ptr = src.decoder_ptr;
+    encoder_ptr = src.encoder_ptr;
+    ptr_out = src.ptr_out;
+    active = src.active;
+    check_available_for_write = src.check_available_for_write;
+    frame_size = src.frame_size;
+    cfg = src.cfg;
+    is_active = src.is_active;
+    return *this;
+  }
+
  protected:
   // AudioInfo info;
   AudioDecoder *decoder_ptr = CodecNOP::instance();  // decoder
@@ -350,6 +362,15 @@ class EncodedAudioStream : public ReformatBaseStream {
   /// defines the size of the decoded frame in bytes
   void setFrameSize(int size) { enc_out.setFrameSize(size); }
 
+  EncodedAudioStream& operator=(EncodedAudioStream const& src) {
+    enc_out = src.enc_out;
+    byte_factor = src.byte_factor;
+    p_stream = src.p_stream;
+    p_print = src.p_print;
+    info = src.info;
+    return *this;
+  };
+
  protected:
   EncodedAudioOutput enc_out;
   float byte_factor = 2.0f;

src/AudioTools/AudioCodecs/CodecChain.h

@@ -0,0 +1,93 @@
+#pragma once
+#include "AudioTools/AudioCodecs/AudioCodecsBase.h"
+#include "AudioTools/AudioCodecs/AudioEncoded.h"
+
+namespace audio_tools {
+
+/**
+ * @brief CodecChain - allows to chain multiple decoders and encoders together
+ * @ingroup codecs
+ * @ingroup decoder
+ * @ingroup encoder
+ * @author Phil Schatzmann
+ * @copyright GPLv3
+ */
+class CodecChain : public AudioDecoder, AudioEncoder {
+ public:
+  CodecChain() = default;
+  CodecChain(AudioDecoder &decoder) { addDecoder(decoder); }
+  CodecChain(AudioEncoder &encoder) { addEncoder(encoder); }
+
+  /// Adds a decoder to the chain
+  void addDecoder(AudioDecoder &decoder) {
+    EncodedAudioStream stream;
+    stream.setDecoder(&decoder);
+    streams.push_back(stream);
+    if (streams.size() > 1) {
+      streams[streams.size() - 2].setOutput(streams[streams.size() - 1]);
+    }
+  }
+
+  /// Adds an encoder to the chain
+  void addEncoder(AudioEncoder &encoder) {
+    EncodedAudioStream stream;
+    stream.setEncoder(&encoder);
+    streams.push_back(stream);
+    if (streams.size() > 1) {
+      streams[streams.size() - 2].setOutput(streams[streams.size() - 1]);
+    }
+  }
+
+  void setOutput(Print &out_stream) override {
+    p_print = &out_stream;
+    if (streams.size() > 0) streams[streams.size() - 1].setOutput(out_stream);
+  }
+
+  void setOutput(AudioStream &out_stream) override {
+    p_print = &out_stream;
+    if (streams.size() > 0) streams[streams.size() - 1].setOutput(out_stream);
+  }
+
+  void setOutput(AudioOutput &out_stream) override {
+    p_print = &out_stream;
+    if (streams.size() > 0) streams[streams.size() - 1].setOutput(out_stream);
+  }
+
+  void setAudioInfo(AudioInfo from) override {
+    AudioDecoder::setAudioInfo(from);
+    for (auto &stream : streams) {
+      stream.setAudioInfo(from);
+    }
+  }
+
+  size_t write(const uint8_t *data, size_t len) override {
+    if (streams.size() == 0) return 0;
+    return streams[0].write(data, len);
+  }
+
+  operator bool() { return is_active; }
+
+  bool begin() {
+    is_active = true;
+    for (auto &stream : streams) {
+      stream.begin();
+    }
+    return true;
+  }
+
+  void end() override {
+    is_active = false;
+    for (auto &stream : streams) {
+      stream.end();
+    }
+  };
+
+  /// Returns nullptr
+  const char *mime() { return nullptr; }
+
+ protected:
+  Vector<EncodedAudioStream> streams;
+  bool is_active = false;
+};
+
+}  // namespace audio_tools