festival_plugin: add support for different encodings

ros-drivers · Apr 3, 2024 · 53bc94e · 53bc94e
1 parent f95f9c8
commit 53bc94e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/sound_play/src/sound_play/festival_plugin.py b/sound_play/src/sound_play/festival_plugin.py
@@ -16,6 +16,9 @@ def __init__(self):
     def sound_play_say_plugin(self, text, voice):
         if voice is None or voice == '':
             voice = self._default_voice
+        encoding = 'ISO-8859-15'
+        if ':' in voice:
+            voice, encoding = voice.split(':', maxsplit=1)
         txtfile = tempfile.NamedTemporaryFile(
             prefix='sound_play', suffix='.txt')
         (wavfile, wavfilename) = tempfile.mkstemp(
@@ -26,10 +29,10 @@ def sound_play_say_plugin(self, text, voice):
             try:
                 if hasattr(text, 'decode'):
                     txtfile.write(
-                        text.decode('UTF-8').encode('ISO-8859-15'))
+                        text.decode('UTF-8').encode(encoding))
                 else:
                     txtfile.write(
-                        text.encode('ISO-8859-15'))
+                        text.encode(encoding))
             except UnicodeEncodeError:
                 if hasattr(text, 'decode'):
                     txtfile.write(text)