TextToSpeech.h

#pragma once
#ifndef NO_SPEECH
#include "AudioCodecs/CodecMP3Helix.h"
#include "AudioTools/AudioOutput.h"
#include "AudioTools/AudioStreams.h"
#include "AudioTools/AudioTypes.h"
#include "NumberToText.h"
#include "AudioDictionary.h"
 
namespace simple_tts {
 
class TextToSpeech {
 public:
  TextToSpeech(SimpleTTSBase &tts, AudioStream &sink, AudioDecoder &decoder,
               AudioDictionaryBase &dict) {
    tts.registerCallback(callback, this);
    p_tts = &tts;
    p_dictionary = &dict;
    p_decoder = &decoder;
    p_sink = &sink;
    decodedStream = new audio_tools::EncodedAudioStream(&sink, &decoder);
    begin();
  }
 
  TextToSpeech(SimpleTTSBase &tts, AudioOutput &sink, AudioDecoder &decoder,
               AudioDictionaryBase &dict) {
    tts.registerCallback(callback, this);
    p_tts = &tts;
    p_dictionary = &dict;
    p_decoder = &decoder;
    p_sink = &sink;
    decodedStream = new audio_tools::EncodedAudioStream(&sink, &decoder);
    begin();
  }
 
  TextToSpeech(AudioStream &sink, AudioDecoder &decoder,
               AudioDictionaryBase &dict) {
    p_dictionary = &dict;
    p_decoder = &decoder;
    p_sink = &sink;
    decodedStream = new audio_tools::EncodedAudioStream(&sink, &decoder);
    begin();
  }
 
 
  ~TextToSpeech() { delete decodedStream; }
 
  void say(const char *word) {
    LOGI("say: %s",word);
    AudioStream *mp3Stream = p_dictionary->get(word);
    if (mp3Stream != nullptr) {
      if (!active) begin();
      mp3Stream->begin();
      copier.begin(*decodedStream, *mp3Stream);
      copier.copyAll(0, 0);
      copier.end();
      mp3Stream->end();
    } else {
      LOGE("Word not available in dictionary: %s", word);
    }
  }
 
  void begin() {
    p_decoder->begin();
    active = true;
  }
 
  void end() {
    p_decoder->end();
    active = false;
  }
 
  void say(audio_tools::Vector<const char *> words) {
    begin();
    for (auto word : words) {      
      say(word);
    }
    end();
  }
 
  void delay(uint32_t delay_ms){
      uint8_t buffer[1024] = {0};
      unsigned long timeout = millis()+delay_ms;
      while(timeout>millis()){
         p_sink->write((const uint8_t*)buffer,1024);
      }
  }
 
 protected:
  bool active = false;
  NumberToText ntt;
  audio_tools::AudioDecoder *p_decoder = nullptr;
  audio_tools::EncodedAudioStream *decodedStream = nullptr;  // Decoding stream
  SimpleTTSBase *p_tts = nullptr;               // Text source
  AudioDictionaryBase *p_dictionary = nullptr;  // Dictionary to access audio data
  audio_tools::StreamCopy copier;               // copy in to out
  Print *p_sink=nullptr;
 
  static void callback(audio_tools::Vector<const char *> words, void* ref) { 
    TextToSpeech* self = (TextToSpeech*)ref;
    self->say(words); }
};
 
}  // namespace simple_tts
 
#endif