voice: Improvements to the talk clip generation

This covers the voiced directory and filenames

 * Don't regenerate a talk clip if one is present (?)
 * Format awareness; ie if the TTS engine generates an mp3 file,
   convert it to a wav file so we can encode it properly
 * Use a global variable for the wavtrim threshold

Change-Id: I9f441b573704bdf7675794fd0e1984446308463b

1 files changed, 24 insertions, 10 deletions

diff --git a/tools/voice.pl b/tools/voice.pl
index 0718e517ff..9398b58b47 100755
--- a/tools/voice.pl
+++ b/tools/voice.pl
@@ -124,6 +124,8 @@ my %espeak_lang_map = (
         'turkce' => 'tr',
 );
 
+my $trim_thresh = 500;   # Trim silence if over this, in ms
+
 # Initialize TTS engine. May return an object or value which will be passed
 # to voicestring and shutdown_tts
 sub init_tts {
@@ -433,8 +435,7 @@ sub generateclips {
                     } else {
                         voicestring($voice, $wav, $tts_engine_opts, $tts_object);
                         if ($format eq "wav") {
-                            wavtrim($wav, 500, $tts_object);
-                            # 500 seems to be a reasonable default for now
+                            wavtrim($wav, $trim_thresh, $tts_object);
                         }
                     }
                     # Convert from mp3 to wav so we can use rbspeex
@@ -510,6 +511,8 @@ sub gentalkclips {
     my $d = new DirHandle $dir;
     while (my $file = $d->read) {
         my ($voice, $wav, $enc);
+        my $format = $tts_object->{'format'};
+
         # Print some progress information
         if (++$i % 10 == 0 and !$verbose) {
             print(".");
@@ -527,8 +530,8 @@ sub gentalkclips {
         }
         # Element is a dir
         if ( -d $path) {
+            $enc = sprintf("%s/_dirname.talk", $path);
             gentalkclips($path, $tts_object, $encoder, $encoder_opts, $tts_engine_opts, $i);
-            $enc = sprintf("%s/_dirname.talk", $path);
         }
         # Element is a file
         else {
@@ -537,13 +540,24 @@ sub gentalkclips {
         }
 
         printf("Talkclip %s: %s", $enc, $voice) if $verbose;
-
-        voicestring($voice, $wav, $tts_engine_opts, $tts_object);
-        wavtrim($wav, 500, $tts_object);
-        # 500 seems to be a reasonable default for now
-        encodewav($wav, $enc, $encoder, $encoder_opts, $tts_object);
-        synchronize($tts_object);
-        unlink($wav);
+        # Don't generate encoded file if it already exists
+        next if (-f $enc);
+
+        voicestring($voice, $wav, $tts_engine_opts, $tts_object);
+        wavtrim($wav, $trim_thresh, $tts_object);
+
+        if ($format eq "mp3") {
+            system("ffmpeg -loglevel 0 -i $wav $voice$wav");
+            rename("$voice$wav","$wav");
+            $format = "wav";
+        }
+        if ($format eq "wav") {
+            encodewav($wav, $enc, $encoder, $encoder_opts, $tts_object);
+        } else {
+            copy($wav, $enc);
+        }
+        synchronize($tts_object);
+        unlink($wav);
     }
 }