Feat/mod dub v2 (#22)

* support tts elevenlabs to mod_dub Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * wip Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * wip * wip * fix aws race condition when 2 start transcribes are sent at the same instant * wip * wip * wip * allow queue play on track * wip Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * revert change for aws transcribe Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> * fix type * wip * wip * rename parameters * rename paramters * wip * wip * wip * wip * wip * bug: there exists scenarios where callback is not defined * wip * wip * revert unintended changes to mod_google_transcribe * fix bugs w/ streaming tts simplified arg parsing to use freeswitch conventions --------- Signed-off-by: Quan HL <quan.luuhoang8@gmail.com> Co-authored-by: Quan HL <quan.luuhoang8@gmail.com>
2026-01-25 02:08:27 +00:00 · 2024-03-23 15:37:27 -04:00
parent 4e57f73c7e
commit 9cdc5fdfca
23 changed files with 1907 additions and 1612 deletions
@@ -0,0 +1,143 @@
+#include "track.h"
+#include "ap_file.h"
+#include "ap_http.h"
+#include "switch.h"
+
+#define INIT_BUFFER_SIZE (80000)
+
+Track::Track(const std::string& trackName, int sampleRate) : _trackName(trackName), _sampleRate(sampleRate), 
+  _buffer(INIT_BUFFER_SIZE), _stopping(false)
+{
+}
+
+Track::~Track() {
+  removeAllAudio();
+  switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Track::~Track: track %s\n", _trackName.c_str());
+}
+
+/**
+ * @brief called when an audio producer has finished retrieving the audio.
+ * If we have another producer queued, then start it.
+ * 
+ * @param hasError 
+ * @param errMsg 
+ */
+void Track::onPlayDone(bool hasError, const std::string& errMsg) {
+  switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "Track::onPlayDone %s\n", _trackName.c_str());
+
+  if (!_stopping) {
+    if (hasError) switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "onPlayDone: error: %s\n", errMsg.c_str());
+    std::lock_guard<std::mutex> lock(_mutex);
+    _apQueue.pop();
+    if (!_apQueue.empty()) {
+      switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "onPlayDone: starting queued audio on track %s\n", _trackName.c_str());
+      _apQueue.front()->start(std::bind(&Track::onPlayDone, this, std::placeholders::_1, std::placeholders::_2));
+    }
+  }
+  else {
+    switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "onPlayDone: track %s stopping\n", _trackName.c_str());
+  }
+}
+
+void Track::queueFileAudio(const std::string& path, int gain, bool loop) {
+  bool startIt = false;
+  if (_stopping) return;
+
+  auto ap = std::make_shared<AudioProducerFile>(_mutex, _buffer, _sampleRate);
+  ap->queueFileAudio(path, gain, loop);
+  {
+    std::lock_guard<std::mutex> lock(_mutex);
+    _apQueue.push(ap);
+    startIt = _apQueue.size() == 1;
+  }
+
+  if (startIt) {
+    try {
+      ap->start(std::bind(&Track::onPlayDone, this, std::placeholders::_1, std::placeholders::_2));
+    } catch (std::exception& e) {
+      onPlayDone(true, e.what());
+    }
+  }
+}
+
+void Track::queueHttpGetAudio(const std::string& url, int gain, bool loop) {
+  bool startIt = false;
+  if (_stopping) return;
+  auto ap = std::make_shared<AudioProducerHttp>(_mutex, _buffer, _sampleRate);
+  ap->queueHttpGetAudio(url, gain, loop);
+  {
+    std::lock_guard<std::mutex> lock(_mutex);
+    _apQueue.push(ap);
+    startIt = _apQueue.size() == 1;
+  }
+
+  if (startIt) {
+    try {
+      ap->start(std::bind(&Track::onPlayDone, this, std::placeholders::_1, std::placeholders::_2));
+    } catch (std::exception& e) {
+      onPlayDone(true, e.what());
+    }
+  }
+}
+
+void Track::queueHttpPostAudio(const std::string& url, int gain, bool loop) {
+  bool startIt = false;
+  if (_stopping) return;
+  auto ap = std::make_shared<AudioProducerHttp>(_mutex, _buffer, _sampleRate);
+  ap->queueHttpPostAudio(url, gain, loop);
+  {
+    std::lock_guard<std::mutex> lock(_mutex);
+    _apQueue.push(ap);
+    startIt = _apQueue.size() == 1;
+  }
+
+  if (startIt) {
+    try {
+      ap->start(std::bind(&Track::onPlayDone, this, std::placeholders::_1, std::placeholders::_2));
+    } catch (std::exception& e) {
+      onPlayDone(true, e.what());
+    }
+  }
+}
+
+void Track::queueHttpPostAudio(const std::string& url, const std::string& body, std::vector<std::string>& headers, int gain, bool loop) {
+  bool startIt = false;
+  if (_stopping) return;
+  auto ap = std::make_shared<AudioProducerHttp>(_mutex, _buffer, _sampleRate);
+    ap->queueHttpPostAudio(url, body, headers, gain, loop);
+  {
+    std::lock_guard<std::mutex> lock(_mutex);
+    _apQueue.push(ap);
+    startIt = _apQueue.size() == 1;
+  }
+
+  if (startIt) {
+    try {
+      ap->start(std::bind(&Track::onPlayDone, this, std::placeholders::_1, std::placeholders::_2));
+    } catch (std::exception& e) {
+      onPlayDone(true, e.what());
+    }
+  }
+}
+
+ bool Track::hasAudio() {
+  if (_stopping) return false;
+  std::lock_guard<std::mutex> lock(_mutex); 
+  return hasAudio_NoLock();
+}
+
+void Track::removeAllAudio() {
+  _stopping = true;
+  std::queue<std::shared_ptr<AudioProducer>> apQueueCopy;
+  {
+    std::lock_guard<std::mutex> lock(_mutex); 
+    apQueueCopy = _apQueue;
+    _apQueue = std::queue<std::shared_ptr<AudioProducer>>();
+  }
+  
+  while (!apQueueCopy.empty()) {
+    auto ap = apQueueCopy.front();
+    apQueueCopy.pop();
+    ap->stop();
+  }
+}