X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=write.c;h=acfb94605b53d259a89ea41e75507b81e2cb2ca4;hb=e32a0e7295c6386b421956553ae577a7df1abf8e;hp=79aaef3112931579fca863dd4484e1dc460817f1;hpb=19afe4270372d36eddc567dfc7d37db5806e02a7;p=paraslash.git diff --git a/write.c b/write.c index 79aaef31..acfb9460 100644 --- a/write.c +++ b/write.c @@ -1,259 +1,163 @@ -/* - * Copyright (C) 2005-2006 Andre Noll - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA. - */ +/* Copyright (C) 2005 Andre Noll , see file COPYING. */ +/** \file write.c Paraslash's standalone wav/raw player. */ + +#include +#include +#include + +#include "write_cmd.lsg.h" +#include "write.lsg.h" #include "para.h" #include "string.h" -#include "write.cmdline.h" +#include "list.h" +#include "sched.h" +#include "stdin.h" +#include "buffer_tree.h" #include "write.h" -#include "write_common.h" #include "fd.h" - -#include /* gettimeofday */ - #include "error.h" +#include "version.h" +#include "check_wav.h" -#define WAV_HEADER_LEN 44 +/** Array of error strings. */ +DEFINE_PARA_ERRLIST; -static unsigned char *audiobuf; -static struct timeval *start_time; -struct gengetopt_args_info conf; +#define CMD_PTR (lls_cmd(0, write_suite)) +#define OPT_RESULT(_name, _lpr) \ + (lls_opt_result(LSG_WRITE_PARA_WRITE_OPT_ ## _name, _lpr)) +#define OPT_GIVEN(_name, _lpr) (lls_opt_given(OPT_RESULT(_name, _lpr))) +#define OPT_UINT32_VAL(_name, _lpr) (lls_uint32_val(0, OPT_RESULT(_name, _lpr))) -INIT_WRITE_ERRLISTS; +static struct stdin_task sit; +static int loglevel; +INIT_STDERR_LOGGING(loglevel) -void para_log(int ll, const char* fmt,...) +static void handle_help_flag(struct lls_parse_result *lpr) { - va_list argp; + char *help; - if (ll < conf.loglevel_arg) + if (OPT_GIVEN(DETAILED_HELP, lpr)) + help = lls_long_help(CMD_PTR); + else if (OPT_GIVEN(HELP, lpr)) + help = lls_short_help(CMD_PTR); + else return; - va_start(argp, fmt); - vfprintf(stderr, fmt, argp); - va_end(argp); + printf("%s\n", help); + free(help); + print_writer_helps(OPT_GIVEN(DETAILED_HELP, lpr)); + exit(EXIT_SUCCESS); } -/** - * read WAV_HEADER_LEN bytes from stdin to audio buffer - * - * \return -E_READ_HDR on errors and on eof before WAV_HEADER_LEN could be - * read. A positive return value indicates success. - */ -static int read_wav_header(void) -{ - ssize_t ret, count = 0; +struct write_task { + struct task *task; + struct check_wav_context *cwc; +}; - while (count < WAV_HEADER_LEN) { - ret = read(STDIN_FILENO, audiobuf + count, WAV_HEADER_LEN - count); - if (ret <= 0) - return -E_READ_HDR; - count += ret; - } - return 1; -} - -/** - * check if current time is later than start_time - * \param diff pointer to write remaining time to - * - * If start_time was not given, or current time is later than given - * start_time, return 0. Otherwise, return 1 and write the time - * difference between current time and start_time to diff. diff may be - * NULL. - * - */ -static int start_time_in_future(struct timeval *diff) +static void write_pre_select(struct sched *s, void *context) { - struct timeval now; - - if (!conf.start_time_given) - return 0; - gettimeofday(&now, NULL); - return tv_diff(start_time, &now, diff) > 0? 1 : 0; + struct write_task *wt = context; + check_wav_pre_select(s, wt->cwc); } -/** - * sleep until time given at command line - * - * This is called if the initial buffer is filled. It returns - * immediately if no start_time was given at the command line - * or if the given start time is in the past. - * - */ -static void do_initial_delay(struct timeval *delay) +static int write_post_select(__a_unused struct sched *s, void *context) { - do - para_select(1, NULL, NULL, delay); - while (start_time_in_future(delay)); + struct write_task *wt = context; + return check_wav_post_select(wt->cwc); } -static int read_stdin(char *buf, size_t bytes_to_load, size_t *loaded) -{ - ssize_t ret; - - while (*loaded < bytes_to_load) { - ret = read(STDIN_FILENO, buf + *loaded, bytes_to_load - *loaded); - if (ret <= 0) { - if (ret < 0) - ret = -E_READ_STDIN; - return ret; - } - *loaded += ret; - } - return 1; -} -/** - * play raw pcm data - * \param loaded number of bytes already loaded - * - * If start_time was given, prebuffer data until buffer is full or - * start_time is reached. In any case, do not start playing before - * start_time. - * - * \return positive on success, negative on errors. - */ -static int pcm_write(struct writer_node_group *wng, size_t loaded) +static int setup_and_schedule(struct lls_parse_result *lpr) { - size_t bufsize, prebuf_size, bytes_to_load; - struct timeval delay; - int ret, not_yet_started = 1; - - ret = wng_open(wng); - if (ret < 0) - goto out; - PARA_INFO_LOG("max chunk_bytes: %d\n", wng->max_chunk_bytes); - bufsize = (conf.bufsize_arg * 1024 / wng->max_chunk_bytes) - * wng->max_chunk_bytes; - audiobuf = para_realloc(audiobuf, bufsize); - prebuf_size = conf.prebuffer_arg * bufsize / 100; - bytes_to_load = PARA_MAX(prebuf_size, wng->max_chunk_bytes); - ret = read_stdin(audiobuf, bytes_to_load, &loaded); - if (ret <= 0 || loaded < bytes_to_load) { - if (ret >= 0) - ret = -E_PREMATURE_END; - goto out; + int i, n, ret, writer_given = OPT_GIVEN(WRITER, lpr); + struct btr_node *cw_btrn; + struct writer_node *wns; + static struct sched s; + struct wav_params wp; + struct write_task wt; + + sit.btrn = btr_new_node(&(struct btr_node_description) + EMBRACE(.name = "stdin")); + stdin_task_register(&sit, &s); + + LLS_COPY_WAV_PARMS(&wp, LSG_WRITE_PARA_WRITE, lpr); + wt.cwc = check_wav_init(sit.btrn, NULL, &wp, &cw_btrn); + wt.task = task_register(&(struct task_info) { + .name = "write", + .pre_select = write_pre_select, + .post_select = write_post_select, + .context = &wt, + }, &s); + + n = writer_given? writer_given : 1; + wns = para_calloc(n * sizeof(*wns)); + for (i = 0; i < n; i++) { + const char *arg = i < writer_given? + lls_string_val(i, OPT_RESULT(WRITER, lpr)) : NULL; + wns[i].wid = check_writer_arg_or_die(arg, &wns[i].lpr); + register_writer_node(wns + i, cw_btrn, &s); } - if (not_yet_started && start_time && start_time_in_future(&delay)) - do_initial_delay(&delay); - not_yet_started = 0; -again: - ret = wng_write(wng, audiobuf, &loaded); - if (ret <= 0) - goto out; - ret = -E_WRITE_OVERRUN; - if (loaded >= bufsize) - goto out; - bytes_to_load = PARA_MIN(wng->max_chunk_bytes, bufsize); - ret = read_stdin(audiobuf, bytes_to_load, &loaded); - if (ret < 0) - goto out; - if (!ret) - wng->eof = 1; - goto again; -out: - wng_close(wng); - return ret; -} - -static struct writer_node_group *check_args(void) -{ - int i, ret = -E_WRITE_SYNTAX; - static struct timeval tv; - struct writer_node_group *wng = NULL; - - if (conf.list_writers_given) { - char *msg = NULL; - FOR_EACH_WRITER(i) { - char *tmp = make_message("%s%s%s", - i? msg : "", - i? " " : "", - writer_names[i]); - free(msg); - msg = tmp; + s.default_timeout.tv_sec = 10; + s.default_timeout.tv_usec = 50000; + ret = schedule(&s); + if (ret >= 0) { + int j, ts; + for (j = 0; j < n; j++) { + struct writer_node *wn = wns + j; + ts = task_status(wn->task); + assert(ts < 0); + if (ts != -E_WRITE_COMMON_EOF && ts != -E_BTR_EOF) { + const char *name = writer_name(wn->wid); + PARA_ERROR_LOG("%s: %s\n", name, + para_strerror(-ts)); + if (ret >= 0) + ret = ts; + } } - fprintf(stderr, "%s\n", msg); - free(msg); - exit(EXIT_SUCCESS); } - if (conf.prebuffer_arg < 0 || conf.prebuffer_arg > 100) - goto out; - if (conf.start_time_given) { - long unsigned sec, usec; - if (sscanf(conf.start_time_arg, "%lu:%lu", - &sec, &usec) != 2) - goto out; - tv.tv_sec = sec; - tv.tv_usec = usec; - start_time = &tv; - } - if (!conf.writer_given) { - wng = setup_default_wng(); - ret = 1; - goto out; - } - wng = wng_new(conf.writer_given); - for (i = 0; i < conf.writer_given; i++) { - ret = check_writer_arg(conf.writer_arg[i]); - if (ret < 0) - goto out; - wng->writer_nodes[i].writer = &writers[ret]; + for (i = n - 1; i >= 0; i--) { + struct writer_node *wn = wns + i; + writer_get(wn->wid)->close(wn); + btr_remove_node(&wn->btrn); + lls_free_parse_result(wns[i].lpr, + lls_cmd(wn->wid, write_cmd_suite)); } - ret = 1; -out: - if (ret > 0) - return wng; - free(wng); - return NULL; + free(wns); + check_wav_shutdown(wt.cwc); + sched_shutdown(&s); + return ret; } /** - * test if audio buffer contains a valid wave header + * Para_write's main function. + * + * \param argc The usual argument counter. + * \param argv The usual argument vector. + * + * It sets up and starts the tasks and the buffer tree nodes determined by + * command line options. * - * \return If not, return 0, otherwise, store number of channels and sample rate - * in struct conf and return WAV_HEADER_LEN. + * \return \p EXIT_SUCCESS or EXIT_FAILURE */ -static size_t check_wave(void) -{ - unsigned char *a = audiobuf; - if (a[0] != 'R' || a[1] != 'I' || a[2] != 'F' || a[3] != 'F') - return WAV_HEADER_LEN; - conf.channels_arg = (unsigned) a[22]; - conf.sample_rate_arg = a[24] + (a[25] << 8) + (a[26] << 16) + (a[27] << 24); - return 0; -} - int main(int argc, char *argv[]) { - int ret = -E_WRITE_SYNTAX; - struct writer_node_group *wng = NULL; + int ret; + struct lls_parse_result *lpr; + char *errctx; - cmdline_parser(argc, argv, &conf); - wng = check_args(); - if (!wng) - goto out; - init_supported_writers(); - audiobuf = para_malloc(WAV_HEADER_LEN); - ret = read_wav_header(); + ret = lls(lls_parse(argc, argv, CMD_PTR, &lpr, &errctx)); if (ret < 0) goto out; - ret = pcm_write(wng, check_wave()); + loglevel = OPT_UINT32_VAL(LOGLEVEL, lpr); + version_handle_flag("write", OPT_GIVEN(VERSION, lpr)); + handle_help_flag(lpr); + ret = setup_and_schedule(lpr); + lls_free_parse_result(lpr, CMD_PTR); out: - wng_destroy(wng); - free(audiobuf); - if (ret < 0) - PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret)); - return ret; + if (ret < 0) { + if (errctx) + PARA_ERROR_LOG("%s\n", errctx); + free(errctx); + PARA_ERROR_LOG("%s\n", para_strerror(-ret)); + } + return ret < 0? EXIT_FAILURE : EXIT_SUCCESS; }