00e6c9deeaae0f3a0acd207f6933f42178a6efe6
[fw/altos] / ao-tools / ao-view / aoview_flite.c
1 /*
2  * Copyright © 2009 Keith Packard <keithp@keithp.com>
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; version 2 of the License.
7  *
8  * This program is distributed in the hope that it will be useful, but
9  * WITHOUT ANY WARRANTY; without even the implied warranty of
10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11  * General Public License for more details.
12  *
13  * You should have received a copy of the GNU General Public License along
14  * with this program; if not, write to the Free Software Foundation, Inc.,
15  * 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.
16  */
17
18 #include <stdio.h>
19 #include <flite/flite.h>
20 #include "aoview.h"
21 #include <alsa/asoundlib.h>
22
23 cst_voice *register_cmu_us_kal16();
24 static cst_voice *voice;
25
26 static FILE *pipe_write;
27 static GThread *aoview_flite_thread;
28
29 static snd_pcm_t        *alsa_handle;
30
31 gpointer
32 aoview_flite_task(gpointer data)
33 {
34         FILE            *input = data;
35         char            line[1024];
36         cst_wave        *wave;
37         int             rate;
38         int             channels;
39         int             err;
40         char            *samples;
41         int             num_samples;
42
43         err = snd_pcm_open(&alsa_handle, "default",
44                            SND_PCM_STREAM_PLAYBACK, 0);
45         if (err < 0) {
46                 fprintf(stderr, "alsa open failed %s\n",
47                         strerror(-err));
48                 alsa_handle = NULL;
49         }
50         rate = 0;
51         channels = 0;
52         while (fgets(line, sizeof (line) - 1, input) != NULL) {
53                 if (!alsa_handle)
54                         continue;
55                 wave = flite_text_to_wave(line, voice);
56                 if (wave->sample_rate != rate ||
57                     wave->num_channels != channels)
58                 {
59                         rate = wave->sample_rate;
60                         channels = wave->num_channels;
61                         err = snd_pcm_set_params(alsa_handle,
62                                                  SND_PCM_FORMAT_S16,
63                                                  SND_PCM_ACCESS_RW_INTERLEAVED,
64                                                  channels,
65                                                  rate,
66                                                  1,
67                                                  100000);
68                         if (err < 0)
69                                 fprintf(stderr, "alsa set_params error %s\n",
70                                         strerror(-err));
71                 }
72                 err = snd_pcm_prepare(alsa_handle);
73                 if (err < 0)
74                         fprintf(stderr, "alsa pcm_prepare error %s\n",
75                                 strerror(-err));
76                 samples = (char *) wave->samples;
77                 num_samples = wave->num_samples;
78                 while (num_samples > 0) {
79                         err = snd_pcm_writei(alsa_handle,
80                                              samples, num_samples);
81                         if (err <= 0) {
82                                 fprintf(stderr, "alsa write error %s\n",
83                                         strerror(-err));
84                                 break;
85                         }
86                         num_samples -= err;
87                         samples += err * 2 * channels;
88                 }
89                 snd_pcm_drain(alsa_handle);
90                 delete_wave(wave);
91         }
92         snd_pcm_close(alsa_handle);
93         alsa_handle = 0;
94         return NULL;
95 }
96
97 void
98 aoview_flite_stop(void)
99 {
100         int status;
101         if (pipe_write) {
102                 fclose(pipe_write);
103                 pipe_write = NULL;
104         }
105         if (aoview_flite_thread) {
106                 g_thread_join(aoview_flite_thread);
107                 aoview_flite_thread = NULL;
108         }
109 }
110
111 FILE *
112 aoview_flite_start(void)
113 {
114         static once;
115         int     p[2];
116         GError  *error;
117         FILE    *pipe_read;
118
119         if (!once) {
120                 flite_init();
121                 voice = register_cmu_us_kal16();
122                 if (!voice) {
123                         perror("register voice");
124                         exit(1);
125                 }
126         }
127         aoview_flite_stop();
128         pipe(p);
129         pipe_read = fdopen(p[0], "r");
130         pipe_write = fdopen(p[1], "w");
131         g_thread_create(aoview_flite_task, pipe_read, TRUE, &error);
132         return pipe_write;
133 }