#!/usr/bin/env python
# -*- coding: utf-8 -*-


# This script is a modified version to support Linux TTS fiel genration using PicoTTS

# Sound pack maintainers (incomplete list) by language alphabetical order
# Czech      : Martin Hotar
# French     : Bertrand Songis & André Bernet
# English    : Rob Thompson & Martin Hotar
# German     : Romolo Manfredini (Some corrections by Peer)
# Italian    : Romolo Manfredini
# Portuguese : Romolo Manfredini
# Spanish    : Romolo Manfredini (With the help of Jose Moreno)

# from __future__ import print_function

import os
import sys
import subprocess
import zipfile
from tts_common import *
board = "taranis"

SOURCE_DIRECTORY = os.path.dirname(os.path.realpath(__file__))
lib_path = os.path.abspath(os.path.join(SOURCE_DIRECTORY, '..', '..', 'radio', 'util'))
sys.path.append(lib_path)

def generate(str, filename):
    if 0:
        output = "output.wav"
        command = 'pico2wave -l=%s -w=%s "%s"' % (voice, output, str)
        os.system(command.encode('utf-8'))
        command = "sox %s -r 32000 %s reverse silence 1 0.1 0.1%% reverse" % (output, filename)
        os.system(command.encode('utf-8'))
    else:
        output = "output.mp3"
        command = 'gtts-cli -l %s -o %s "%s"' % (voice[:2], output, str)
        os.system(command.encode('utf-8'))
        command = "sox %s -r 32000 %s" % (output, filename)
        os.system(command.encode('utf-8'))

################################################################

if __name__ == "__main__":
    if "en" in sys.argv:
        from tts_en import systemSounds, sounds

        directory = "en"
        voice = "en-US"

    elif "fr" in sys.argv:
        from tts_fr import systemSounds, sounds

        directory = "fr"
        voice = "fr-FR"

    elif "it" in sys.argv:
        from tts_it import systemSounds, sounds

        directory = "it"
        voice = "it-IT"

    elif "de" in sys.argv:
        from tts_de import systemSounds, sounds

        directory = "de"
        voice = "de-DE"

    elif "es" in sys.argv:
        from tts_es import systemSounds, sounds

        directory = "es"
        voice = "es-ES"

    elif "cz" in sys.argv:
        from tts_cz import systemSounds, sounds

        directory = "cz"
        voice = "cs-CZ"

    elif "pt" in sys.argv:
        from tts_pt import systemSounds, sounds

        directory = "pt"
        voice = "pt-PT"

    else:
        print("which language?")
        exit()

    if "csv" in sys.argv:
        path = "/tmp/SOUNDS/" + directory + "/SYSTEM/"
        if not os.path.exists(path):
          os.makedirs(path)
        os.chdir(path)
        with open("%s-%s.csv" % (voice, board), "wb") as csvFile:
            for s, f in systemSounds:
                if s and f:
                    l = u""
                    if board in ("sky9x", "taranis"):
                        l += u"SOUNDS/%s/SYSTEM;" % directory
                    l += f + u";" + s + u"\n"
                    csvFile.write(l.encode("utf-8"))
            for s, f in sounds:
                if s and f:
                    l = u""
                    if board in ("sky9x", "taranis"):
                        l += u"SOUNDS/%s;" % directory
                    l += f + u";" + s + u"\n"
                    csvFile.write(l.encode("utf-8"))

    if "files" in sys.argv:
              path = "/tmp/SOUNDS/" + directory + "/SYSTEM/"
              if not os.path.exists(path):
                os.makedirs(path)
              os.chdir(path)
              for s, f in systemSounds:
                if s and f:
                    generate(s, f)
              os.chdir("..")
              for s, f in sounds:
                if s and f:
                    generate(s, f)