Further para_play abstraction: struct writer_node_group
[paraslash.git] / play.c
1 /*
2 * Copyright (C) 2005-2006 Andre Noll <maan@systemlinux.org>
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; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
17 */
18
19 /*
20 * Based in parts on aplay.c from the alsa-utils-1.0.8 package,
21 * Copyright (c) by Jaroslav Kysela <perex@suse.cz>, which is
22 * based on the vplay program by Michael Beck.
23 */
24
25 #define WAV_HEADER_LEN 44
26 #include <sys/time.h> /* gettimeofday */
27 #include "para.h"
28 #include "fd.h"
29 #include "play.cmdline.h"
30 #include <alsa/asoundlib.h>
31 #include "string.h"
32 #include "error.h"
33
34 #define FORMAT SND_PCM_FORMAT_S16_LE
35
36 struct private_alsa_data {
37 snd_pcm_t *handle;
38 size_t bytes_per_frame;
39 };
40
41 struct writer_node {
42 struct writer *writer;
43 void *private_data;
44 int chunk_bytes;
45 };
46
47 struct writer {
48 int (*open)(struct writer_node *);
49 int (*write)(char *data, size_t nbytes, struct writer_node *);
50 void (*close)(struct writer_node *);
51 void (*shutdown)(struct writer_node *);
52 };
53
54 struct writer_node_group {
55 unsigned num_writers;
56 struct writer_node *writer_nodes;
57 int *written;
58 size_t max_chunk_bytes;
59 int eof;
60 };
61
62
63 #define FOR_EACH_WRITER_NODE(i, wng) for (i = 0; i < (wng)->num_writers; i++)
64
65 #define NUM_WRITERS 1
66 static struct writer writers[NUM_WRITERS];
67 #define FOR_EACH_WRITER(i) for (i = 0; i < NUM_WRITERS, i++)
68
69
70 static unsigned char *audiobuf;
71 static struct timeval *start_time;
72 static struct gengetopt_args_info conf;
73
74 INIT_PLAY_ERRLISTS;
75
76 void para_log(int ll, const char* fmt,...)
77 {
78 va_list argp;
79
80 if (ll < conf.loglevel_arg)
81 return;
82 va_start(argp, fmt);
83 vfprintf(stderr, fmt, argp);
84 va_end(argp);
85 }
86
87 /**
88 * read WAV_HEADER_LEN bytes from stdin to audio buffer
89 *
90 * \return -E_READ_HDR on errors and on eof before WAV_HEADER_LEN could be
91 * read. A positive return value indicates success.
92 */
93 static int read_wav_header(void)
94 {
95 ssize_t ret, count = 0;
96
97 while (count < WAV_HEADER_LEN) {
98 ret = read(STDIN_FILENO, audiobuf + count, WAV_HEADER_LEN - count);
99 if (ret <= 0)
100 return -E_READ_HDR;
101 count += ret;
102 }
103 return 1;
104 }
105
106 /*
107 * open and prepare the PCM handle for writing
108 *
109 * Install PCM software and hardware configuration. Exit on errors.
110 */
111 static int alsa_open(struct writer_node *w)
112 {
113 snd_pcm_hw_params_t *hwparams;
114 snd_pcm_sw_params_t *swparams;
115 snd_pcm_uframes_t buffer_size, xfer_align, start_threshold,
116 stop_threshold;
117 unsigned buffer_time = 0;
118 int err;
119 snd_pcm_info_t *info;
120 snd_output_t *log;
121 snd_pcm_uframes_t period_size;
122 struct private_alsa_data *pad = para_malloc(sizeof(struct private_alsa_data));
123 w->private_data = pad;
124
125 snd_pcm_info_alloca(&info);
126 if (snd_output_stdio_attach(&log, stderr, 0) < 0)
127 return -E_ALSA_LOG;
128 err = snd_pcm_open(&pad->handle, conf.device_arg,
129 SND_PCM_STREAM_PLAYBACK, 0);
130 if (err < 0)
131 return -E_PCM_OPEN;
132 if ((err = snd_pcm_info(pad->handle, info)) < 0)
133 return -E_SND_PCM_INFO;
134
135 snd_pcm_hw_params_alloca(&hwparams);
136 snd_pcm_sw_params_alloca(&swparams);
137 if (snd_pcm_hw_params_any(pad->handle, hwparams) < 0)
138 return -E_BROKEN_CONF;
139 if (snd_pcm_hw_params_set_access(pad->handle, hwparams,
140 SND_PCM_ACCESS_RW_INTERLEAVED) < 0)
141 return -E_ACCESS_TYPE;
142 if (snd_pcm_hw_params_set_format(pad->handle, hwparams, FORMAT) < 0)
143 return -E_SAMPLE_FORMAT;
144 if (snd_pcm_hw_params_set_channels(pad->handle, hwparams,
145 conf.channels_arg) < 0)
146 return -E_CHANNEL_COUNT;
147 if (snd_pcm_hw_params_set_rate_near(pad->handle, hwparams,
148 (unsigned int*) &conf.sample_rate_arg, 0) < 0)
149 return -E_SET_RATE;
150 err = snd_pcm_hw_params_get_buffer_time_max(hwparams, &buffer_time, 0);
151 if (err < 0 || !buffer_time)
152 return -E_GET_BUFFER_TIME;
153 PARA_DEBUG_LOG("buffer time: %d\n", buffer_time);
154 if (snd_pcm_hw_params_set_buffer_time_near(pad->handle, hwparams,
155 &buffer_time, 0) < 0)
156 return -E_SET_BUFFER_TIME;
157 if (snd_pcm_hw_params(pad->handle, hwparams) < 0)
158 return -E_HW_PARAMS;
159 snd_pcm_hw_params_get_period_size(hwparams, &period_size, 0);
160 snd_pcm_hw_params_get_buffer_size(hwparams, &buffer_size);
161 PARA_DEBUG_LOG("buffer size: %lu, period_size: %lu\n", buffer_size,
162 period_size);
163 if (period_size == buffer_size)
164 return -E_BAD_PERIOD;
165 snd_pcm_sw_params_current(pad->handle, swparams);
166 err = snd_pcm_sw_params_get_xfer_align(swparams, &xfer_align);
167 if (err < 0 || !xfer_align)
168 return -E_GET_XFER;
169 snd_pcm_sw_params_set_avail_min(pad->handle, swparams, period_size);
170 /* round to closest transfer boundary */
171 start_threshold = (buffer_size / xfer_align) * xfer_align;
172 if (start_threshold < 1)
173 start_threshold = 1;
174 if (snd_pcm_sw_params_set_start_threshold(pad->handle, swparams,
175 start_threshold) < 0)
176 return -E_START_THRESHOLD;
177 stop_threshold = buffer_size;
178 if (snd_pcm_sw_params_set_stop_threshold(pad->handle, swparams,
179 stop_threshold) < 0)
180 return -E_STOP_THRESHOLD;
181 if (snd_pcm_sw_params_set_xfer_align(pad->handle, swparams,
182 xfer_align) < 0)
183 return -E_SET_XFER;
184 if (snd_pcm_sw_params(pad->handle, swparams) < 0)
185 return -E_SW_PARAMS;
186 pad->bytes_per_frame = snd_pcm_format_physical_width(FORMAT)
187 * conf.channels_arg / 8;
188 return period_size * pad->bytes_per_frame;
189 }
190
191 /**
192 * push out pcm frames
193 * \param data pointer do data to be written
194 * \param nbytes number of bytes (not frames)
195 *
196 * \return Number of bytes written, -E_ALSA_WRITE on errors.
197 */
198 static int alsa_write(char *data, size_t nbytes, struct writer_node *wn)
199 {
200 struct private_alsa_data *pad = wn->private_data;
201 size_t frames = nbytes / pad->bytes_per_frame;
202 unsigned char *d = data;
203 snd_pcm_sframes_t r, result = 0;
204
205 while (frames > 0) {
206 /* write interleaved frames */
207 r = snd_pcm_writei(pad->handle, d, frames);
208 if (r < 0)
209 PARA_ERROR_LOG("write error: %s\n", snd_strerror(r));
210 if (r == -EAGAIN || (r >= 0 && r < frames))
211 snd_pcm_wait(pad->handle, 1);
212 else if (r == -EPIPE)
213 snd_pcm_prepare(pad->handle);
214 else if (r < 0)
215 return -E_ALSA_WRITE;
216 if (r > 0) {
217 result += r;
218 frames -= r;
219 d += r * pad->bytes_per_frame;
220 }
221 }
222 return result * pad->bytes_per_frame;
223 }
224
225 static void alsa_close(struct writer_node *wn)
226 {
227 struct private_alsa_data *pad = wn->private_data;
228 snd_pcm_drain(pad->handle);
229 snd_pcm_close(pad->handle);
230 snd_config_update_free_global();
231 free(pad);
232 }
233
234 void alsa_writer_init(struct writer *w)
235 {
236 w->open = alsa_open;
237 w->write = alsa_write;
238 w->close = alsa_close;
239 w->shutdown = NULL; /* nothing to do */
240 }
241
242
243 /**
244 * check if current time is later than start_time
245 * \param diff pointer to write remaining time to
246 *
247 * If start_time was not given, or current time is later than given
248 * start_time, return 0. Otherwise, return 1 and write the time
249 * difference between current time and start_time to diff. diff may be
250 * NULL.
251 *
252 */
253 static int start_time_in_future(struct timeval *diff)
254 {
255 struct timeval now;
256
257 if (!conf.start_time_given)
258 return 0;
259 gettimeofday(&now, NULL);
260 return tv_diff(start_time, &now, diff) > 0? 1 : 0;
261 }
262
263 /**
264 * sleep until time given at command line
265 *
266 * This is called if the initial buffer is filled. It returns
267 * immediately if no start_time was given at the command line
268 * or if the given start time is in the past.
269 *
270 */
271 static void do_initial_delay(struct timeval *delay)
272 {
273 do
274 para_select(1, NULL, NULL, delay);
275 while (start_time_in_future(delay));
276 }
277
278 static int read_stdin(char *buf, size_t bytes_to_load, size_t *loaded)
279 {
280 ssize_t ret;
281
282 while (*loaded < bytes_to_load) {
283 ret = read(STDIN_FILENO, buf + *loaded, bytes_to_load - *loaded);
284 if (ret <= 0) {
285 if (ret < 0)
286 ret = -E_READ_STDIN;
287 return ret;
288 }
289 *loaded += ret;
290 }
291 return 1;
292 }
293
294 int wng_write(struct writer_node_group *g, char *buf, size_t *loaded)
295 {
296 int ret, i, need_more_writes = 1;
297 size_t min_written = 0;
298
299 while (need_more_writes) {
300 need_more_writes = 0;
301 FOR_EACH_WRITER_NODE(i, g) {
302 size_t w = g->written[i];
303 unsigned char *p = buf + w;
304 int bytes_to_write;
305 struct writer_node *wn = &g->writer_nodes[i];
306 if (!i)
307 min_written = w;
308 else
309 min_written = PARA_MIN(min_written, w);
310 if (w == *loaded)
311 continue;
312 if (!g->eof && (*loaded < wn->chunk_bytes + w))
313 continue;
314 bytes_to_write = PARA_MIN(wn->chunk_bytes,
315 *loaded - w);
316 ret = wn->writer->write(p, bytes_to_write, wn);
317 if (ret < 0)
318 goto out;
319 if (ret != bytes_to_write)
320 PARA_WARNING_LOG("short write: %d/%d\n", ret,
321 bytes_to_write);
322 g->written[i] += ret;
323 need_more_writes = 1;
324 }
325 }
326 *loaded -= min_written;
327 ret = 0;
328 if (g->eof)
329 goto out;
330 if (*loaded)
331 memmove(buf, buf + min_written, *loaded);
332 FOR_EACH_WRITER_NODE(i, g)
333 g->written[i] -= min_written;
334 ret = 1;
335 out:
336 return ret;
337 }
338
339 int wng_open(struct writer_node_group *g)
340 {
341 int i, ret = 1;
342
343 FOR_EACH_WRITER_NODE(i, g) {
344 struct writer_node *wn = &g->writer_nodes[i];
345 ret = wn->writer->open(wn);
346 if (ret < 0)
347 goto out;
348 wn->chunk_bytes = ret;
349 g->max_chunk_bytes = PARA_MAX(g->max_chunk_bytes, ret);
350 }
351 out:
352 return ret;
353 }
354
355 void wng_close(struct writer_node_group *g)
356 {
357 int i;
358
359 FOR_EACH_WRITER_NODE(i, g) {
360 struct writer_node *wn = &g->writer_nodes[i];
361 wn->writer->close(wn);
362 }
363 }
364
365 /**
366 * play raw pcm data
367 * \param loaded number of bytes already loaded
368 *
369 * If start_time was given, prebuffer data until buffer is full or
370 * start_time is reached. In any case, do not start playing before
371 * start_time.
372 *
373 * \return positive on success, negative on errors.
374 */
375 static int pcm_write(struct writer_node_group *wng, size_t loaded)
376 {
377 size_t bufsize, prebuf_size, bytes_to_load;
378 struct timeval delay;
379 int ret, not_yet_started = 1;
380
381 ret = wng_open(wng);
382 if (ret < 0)
383 goto out;
384 PARA_INFO_LOG("max chunk_bytes: %d\n", wng->max_chunk_bytes);
385 bufsize = (conf.bufsize_arg * 1024 / wng->max_chunk_bytes)
386 * wng->max_chunk_bytes;
387 audiobuf = para_realloc(audiobuf, bufsize);
388 prebuf_size = conf.prebuffer_arg * bufsize / 100;
389 bytes_to_load = PARA_MAX(prebuf_size, wng->max_chunk_bytes);
390 ret = read_stdin(audiobuf, bytes_to_load, &loaded);
391 if (ret <= 0 || loaded < bytes_to_load) {
392 if (ret >= 0)
393 ret = -E_PREMATURE_END;
394 goto out;
395 }
396 if (not_yet_started && start_time && start_time_in_future(&delay))
397 do_initial_delay(&delay);
398 not_yet_started = 0;
399 again:
400 ret = wng_write(wng, audiobuf, &loaded);
401 if (ret <= 0)
402 goto out;
403 ret = -E_PLAY_OVERRUN;
404 if (loaded >= bufsize)
405 goto out;
406 bytes_to_load = PARA_MIN(wng->max_chunk_bytes, bufsize);
407 ret = read_stdin(audiobuf, bytes_to_load, &loaded);
408 if (ret < 0)
409 goto out;
410 if (!ret)
411 wng->eof = 1;
412 goto again;
413 out:
414 wng_close(wng);
415 return ret;
416 }
417
418 struct writer_node_group *wng_new(unsigned num_writers)
419 {
420 struct writer_node_group *g = para_calloc(sizeof(struct writer_node_group));
421 g->num_writers = num_writers;
422 g->writer_nodes = para_calloc(num_writers
423 * sizeof(struct writer_node));
424 g->written = para_calloc(num_writers * sizeof(size_t));
425 return g;
426 }
427
428 void wng_destroy(struct writer_node_group *g)
429 {
430 if (!g)
431 return;
432 free(g->written);
433 free(g->writer_nodes);
434 free(g);
435 }
436
437 /**
438 * test if audio buffer contains a valid wave header
439 *
440 * \return If not, return 0, otherwise, store number of channels and sample rate
441 * in struct conf and return WAV_HEADER_LEN.
442 */
443 static size_t check_wave(void)
444 {
445 unsigned char *a = audiobuf;
446 if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F')
447 return WAV_HEADER_LEN;
448 conf.channels_arg = (unsigned) a[22];
449 conf.sample_rate_arg = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24);
450 return 0;
451 }
452
453 int main(int argc, char *argv[])
454 {
455 struct timeval tv;
456 int ret = -E_PLAY_SYNTAX;
457 struct writer_node_group *wng = NULL;
458
459 cmdline_parser(argc, argv, &conf);
460 if (conf.prebuffer_arg < 0 || conf.prebuffer_arg > 100)
461 goto out;
462 if (conf.start_time_given) {
463 if (sscanf(conf.start_time_arg, "%lu:%lu",
464 &tv.tv_sec, &tv.tv_usec) != 2)
465 goto out;
466 start_time = &tv;
467 }
468 /* call init for each supported writer */
469 alsa_writer_init(&writers[0]);
470
471 wng = wng_new(1);
472 wng->writer_nodes[0].writer = &writers[0]; /* alsa */
473
474 audiobuf = para_malloc(WAV_HEADER_LEN);
475 ret = read_wav_header();
476 if (ret < 0)
477 goto out;
478 ret = pcm_write(wng, check_wave());
479 out:
480 wng_destroy(wng);
481 free(audiobuf);
482 if (ret < 0)
483 PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));
484 return ret;
485 }