RTSPOutput.h

#pragma once
 
#include "AudioTools/CoreAudio/AudioPlayer.h"
#include "AudioTools/CoreAudio/AudioStreams.h"
#include "RTSPAudioSource.h"
#include "RTSPAudioStreamer.h"
#include "RTSPFormat.h"
 
namespace audio_tools {
 
template <typename Platform>
class RTSPOutput : public AudioOutput {
 public:
  RTSPOutput(RTSPFormat &format, AudioEncoder &encoder) {
    setFormat(format);
    rtsp_streamer.setAudioSource(&rtsp_source);
    p_encoder = &encoder;
  }
 
  void setFormat(RTSPFormat &format) {
    TRACEI();
    p_format = &format;
    rtsp_source.setFormat(format);
  }
 
  RTSPOutput() = default;
 
  RTSPAudioStreamer<Platform> &streamer() { return rtsp_streamer; }
 
  bool begin(AudioInfo info) {
    cfg = info;
    return begin();
  }
 
  bool begin() {
    TRACEI();
    if (p_encoder == nullptr) {
      LOGE("encoder is null");
      return false;
    }
    if (p_format == nullptr) {
      LOGE("format is null");
      return false;
    }
    // setup the RTSPAudioStreamer
    cfg.logInfo();
 
    p_encoder->setOutput(memory_stream);
    p_encoder->setAudioInfo(cfg);
    p_encoder->begin();
    // setup the RTSPSourceFromAudioStream
    rtsp_source.setInput(memory_stream);
    rtsp_source.setFormat(*p_format);
    rtsp_source.setAudioInfo(cfg);
    rtsp_source.start();
 
    memory_stream.setConsumeOnRead(true);
    memory_stream.begin();
 
    // Initialize format with audio info (PCM or codec-specific)
    p_format->begin(cfg);
 
    return true;
  }
 
  void end() {
    rtsp_source.stop();
    memory_stream.end();
  }
 
  int availableForWrite() {
    return rtsp_source.isStarted() ? memory_stream.availableForWrite() : 0;
  }
 
  size_t write(const uint8_t *data, size_t len) override {
    TRACED();
    size_t result = p_encoder->write(data, len);
    return result;
  }
 
  operator bool() { return rtsp_source.isActive() && memory_stream.availableForWrite() > 0; }
 
 protected:
  // Core Components
  CopyEncoder copy_encoder;           
  RTSPAudioSource rtsp_source;        
  DynamicMemoryStream memory_stream{false, 1024, 10};  
  AudioEncoder *p_encoder =
      &copy_encoder;            
  RTSPFormatPCM pcm;            
  RTSPFormat *p_format = &pcm;  
  RTSPAudioStreamer<Platform>
      rtsp_streamer;  
};
 
}  // namespace audio_tools