/*
- * Copyright (C) 2006 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2006-2014 Andre Noll <maan@tuebingen.mpg.de>
*
- * 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.
+ * Licensed under the GPL v2. For licencing details see COPYING.
*/
/** \file write_common.c common functions of para_audiod and para_write */
+#include <regex.h>
+
#include "para.h"
#include "string.h"
#include "list.h"
#include "sched.h"
+#include "ggo.h"
+#include "buffer_tree.h"
#include "write.h"
#include "error.h"
+#include "write_common.h"
+/** the array containing the names of all supported writers */
const char *writer_names[] ={WRITER_NAMES};
+
+/** the array of supported writers */
struct writer writers[NUM_SUPPORTED_WRITERS] = {WRITER_ARRAY};
-static void wng_post_select(struct sched *s, struct task *t)
+/**
+ * Call the init function of each supported paraslash writer.
+ */
+void writer_init(void)
{
- struct writer_node_group *g = t->private_data;
int i;
- size_t min_written = 0;
-
- FOR_EACH_WRITER_NODE(i, g) {
- struct writer_node *wn = &g->writer_nodes[i];
- t->ret = wn->task.ret;
- if (t->ret < 0)
- return;
- if (!i)
- min_written = t->ret;
- else
- min_written = PARA_MIN(min_written, t->ret);
- }
- *g->loaded -= min_written;
- if (!*g->loaded && *g->eof)
- t->ret = 0;
- else
- t->ret = 1;
- if (*g->loaded && min_written)
- memmove(g->buf, g->buf + min_written, *g->loaded);
-}
-int wng_open(struct writer_node_group *g)
+ FOR_EACH_WRITER(i)
+ writers[i].init(&writers[i]);
+}
+/**
+ * Check if given string is a valid command line for any writer.
+ *
+ * \param \wa String of the form writer_name:options.
+ * \param writer_num Contains the number of the writer upon success.
+ *
+ * This function checks whether \a wa starts with the name of a supported
+ * paraslash writer, optionally followed by a colon and any options for that
+ * writer. If a valid writer name was found and further are present, the
+ * remaining part of \a wa is passed to that writer's config parser.
+ *
+ * \return On success, a pointer to the gengetopt args info struct is returned
+ * and \a writer_num contains the number of the writer. Otherwise this function
+ * prints an error message and calls exit().
+ */
+void *check_writer_arg_or_die(const char *wa, int *writer_num)
{
- int i, ret = 1;
-
- FOR_EACH_WRITER_NODE(i, g) {
- struct writer_node *wn = &g->writer_nodes[i];
- ret = wn->writer->open(wn);
- if (ret < 0)
- goto out;
- wn->chunk_bytes = ret;
- g->max_chunk_bytes = PARA_MAX(g->max_chunk_bytes, ret);
- wn->wng = g;
- PARA_DEBUG_LOG("pre_select: %p\n", &wn->writer->pre_select);
- PARA_DEBUG_LOG("post_select: %p\n", &wn->writer->post_select);
- wn->task.pre_select = wn->writer->pre_select;
- wn->task.post_select = wn->writer->post_select;
- wn->task.private_data = wn;
- wn->task.flags = POST_EOF_IS_ERROR;
- register_task(&wn->task);
+ int i, ret, argc;
+ const char *cmdline;
+ char **argv;
+ void *conf;
+
+ if (!wa || !*wa) {
+ i = DEFAULT_WRITER;
+ cmdline = NULL;
+ goto check;
+ }
+ PARA_INFO_LOG("checking %s\n", wa);
+ FOR_EACH_WRITER(i) {
+ const char *name = writer_names[i];
+ size_t len = strlen(name);
+ char c;
+
+ if (strlen(wa) < len)
+ continue;
+ if (strncmp(name, wa, len))
+ continue;
+ c = wa[len];
+ if (!c || c == ' ') {
+ cmdline = c? wa + len + 1 : NULL;
+ goto check;
+ }
}
- register_task(&g->task);
-out:
- return ret;
+ PARA_EMERG_LOG("invalid writer %s\n", wa);
+ exit(EXIT_FAILURE);
+check:
+ ret = create_shifted_argv(cmdline, " \t", &argv);
+ if (ret < 0) {
+ PARA_EMERG_LOG("%s: %s\n", wa, para_strerror(-ret));
+ exit(EXIT_FAILURE);
+ }
+ argc = ret;
+ argv[0] = make_message("%s_write", writer_names[i]);
+ *writer_num = i;
+ conf = writers[i].parse_config_or_die(argc, argv);
+ free_argv(argv);
+ return conf;
}
-void wng_destroy(struct writer_node_group *g)
+/**
+ * Open a writer node and register the corresponding task.
+ *
+ * \param wn The writer node to open.
+ * \param parent The parent btr node (the source for the writer node).
+ * \param s The scheduler instance to register the task to.
+ *
+ * The configuration of the writer node stored in \p wn->conf must be
+ * initialized before this function may be called.
+ */
+void register_writer_node(struct writer_node *wn, struct btr_node *parent,
+ struct sched *s)
{
- if (!g)
- return;
- free(g->written);
- free(g->writer_nodes);
- free(g);
+ struct writer *w = writers + wn->writer_num;
+
+ wn->btrn = btr_new_node(&(struct btr_node_description)
+ EMBRACE(.name = writer_names[wn->writer_num], .parent = parent,
+ .handler = w->execute, .context = wn));
+ wn->task = task_register(&(struct task_info) {
+ .name = writer_names[wn->writer_num],
+ .pre_select = w->pre_select,
+ .post_select = w->post_select,
+ .context = wn,
+ }, s);
}
-void wng_close(struct writer_node_group *g)
+/**
+ * Print the help text of all writers to stdout.
+ *
+ * \param flags Passed to \ref ggo_print_help().
+ */
+void print_writer_helps(unsigned flags)
{
int i;
- FOR_EACH_WRITER_NODE(i, g) {
- struct writer_node *wn = &g->writer_nodes[i];
- unregister_task(&wn->task);
- wn->writer->close(wn);
- }
-}
-
-static void wng_error_handler(struct task *t)
-{
- struct writer_node_group *g = t->private_data;
+ printf_or_die("\nAvailable writers: ");
+ FOR_EACH_WRITER(i)
+ printf_or_die("%s%s", i? " " : "", writer_names[i]);
+ printf_or_die("\n");
+ FOR_EACH_WRITER(i) {
+ struct writer *w = writers + i;
- PARA_INFO_LOG("%p: ret = %d\n", t, t->ret);
- unregister_task(t);
- wng_close(g);
- wng_destroy(g);
+ if (!w->help.short_help)
+ continue;
+ printf_or_die("\n%s: %s", writer_names[i],
+ w->help.purpose);
+ ggo_print_help(&w->help, flags);
+ }
}
-struct writer_node_group *wng_new(unsigned num_writers)
+static void get_btr_value(struct btr_node *btrn, const char *cmd,
+ int32_t *result)
{
- struct writer_node_group *g = para_calloc(sizeof(struct writer_node_group));
- g->num_writers = num_writers;
- g->writer_nodes = para_calloc(num_writers
- * sizeof(struct writer_node));
- g->written = para_calloc(num_writers * sizeof(size_t));
- g->task.private_data = g;
- g->task.post_select = wng_post_select;
- g->task.error_handler = wng_error_handler;
- g->task.flags = POST_ADD_TAIL | POST_EOF_IS_ERROR;
- return g;
+ char *buf = NULL;
+ int ret = btr_exec_up(btrn, cmd, &buf);
+
+ if (ret < 0) {
+ /*
+ * This really should not happen. It means one of our parent
+ * nodes died unexpectedly. Proceed with fingers crossed.
+ */
+ PARA_CRIT_LOG("cmd %s: %s\n", cmd, para_strerror(-ret));
+ *result = 0;
+ return;
+ }
+ ret = para_atoi32(buf, result);
+ assert(ret >= 0);
+ free(buf);
}
-void init_supported_writers(void)
+/**
+ * Ask parent btr nodes for the sample rate of the current stream.
+ *
+ * \param btrn Where to start the search.
+ * \param result Filled in by this function.
+ *
+ * This function is assumed to succeed and terminates on errors.
+ */
+void get_btr_sample_rate(struct btr_node *btrn, int32_t *result)
{
- int i;
-
- FOR_EACH_WRITER(i)
- writers[i].init(&writers[i]);
+ get_btr_value(btrn, "sample_rate", result);
}
-int check_writer_arg(const char *arg)
+/**
+ * Ask parent btr nodes for the channel count of the current stream.
+ *
+ * \param btrn See \ref get_btr_sample_rate.
+ * \param result See \ref get_btr_sample_rate.
+ */
+void get_btr_channels(struct btr_node *btrn, int32_t *result)
{
- int i, ret = -E_WRITE_COMMON_SYNTAX;
- char *a = para_strdup(arg), *p = strchr(a, ':');
- if (p)
- *p = '\0';
- p++;
- FOR_EACH_WRITER(i) {
- if (strcmp(writer_names[i], a))
- continue;
- ret = i;
- goto out;
- }
-out:
- free(a);
- return ret;
+ get_btr_value(btrn, "channels", result);
}
-struct writer_node_group *setup_default_wng(void)
+/**
+ * Ask parent btr nodes for the number of bits per sample and the byte sex.
+ *
+ * \param btrn See \ref get_btr_sample_rate.
+ * \param result Contains the sample format as an enum sample_format type.
+ */
+void get_btr_sample_format(struct btr_node *btrn, int32_t *result)
{
- struct writer_node_group *wng = wng_new(1);
- enum writer_enum default_writer;
-
- if (NUM_SUPPORTED_WRITERS == 1)
- default_writer = FILE_WRITE;
- else
- default_writer = 1;
- wng->writer_nodes[0].writer = &writers[default_writer];
- PARA_INFO_LOG("using default writer: %s\n",
- writer_names[default_writer]);
- return wng;
+ get_btr_value(btrn, "sample_format", result);
}