/*
- * Copyright (C) 2006-2007 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2006-2013 Andre Noll <maan@systemlinux.org>
*
- * 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"
/** the array of supported writers */
struct writer writers[NUM_SUPPORTED_WRITERS] = {WRITER_ARRAY};
-static void wng_pre_select(__a_unused 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;
- FOR_EACH_WRITER_NODE(i, g) {
- struct writer_node *wn = &g->writer_nodes[i];
- t->ret = wn->writer->pre_select(s, wn);
- if (t->ret < 0) {
- g->eof = 1;
- return;
- }
- }
-}
-
-static void wng_post_select(struct sched *s, struct task *t)
-{
- 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->writer->post_select(s, wn);
- if (t->ret < 0) {
- g->eof = 1;
- return;
- }
- if (!i)
- min_written = wn->written;
- else
- min_written = PARA_MIN(min_written, wn->written);
- }
-// PARA_INFO_LOG("loaded: %zd, min_written: %zd bytes\n", *g->loaded, min_written);
- if (min_written) {
- *g->loaded -= min_written;
- FOR_EACH_WRITER_NODE(i, g)
- g->writer_nodes[i].written -= min_written;
- }
- if (!*g->loaded && *g->input_eof) {
- g->eof = 1;
- t->ret = -E_WNG_EOF;
- return;
- }
- t->ret = 1;
- if (*g->loaded && min_written) {
-// PARA_INFO_LOG("moving %zd bytes\n", *g->loaded);
- memmove(g->buf, g->buf + min_written, *g->loaded);
- }
+ FOR_EACH_WRITER(i)
+ writers[i].init(&writers[i]);
}
-
/**
- * call the open function of each writer in the group
+ * Check if given string is a valid command line for any writer.
*
- * \param g the writer node group
+ * \param \wa String of the form writer_name:options.
+ * \param writer_num Contains the number of the writer upon success.
*
- * \return If at least one open function returned an error, all successful
- * writer notes get closed and this error value is returned. Upon success, a
- * task associated with \a g is registered to the scheduler and the function
- * returnes a positive value.
- * */
-int wng_open(struct writer_node_group *g)
+ * 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;
-
- PARA_NOTICE_LOG("opening wng %p with %d writer(s)\n", g, g->num_writers);
- register_task(&g->task);
- FOR_EACH_WRITER_NODE(i, g) {
- struct writer_node *wn = &g->writer_nodes[i];
- wn->wng = g;
- ret = wn->writer->open(wn);
- if (ret < 0)
- goto err_out;
- wn->chunk_bytes = ret;
- g->max_chunk_bytes = PARA_MAX(g->max_chunk_bytes, ret);
+ 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;
+ }
}
- sprintf(g->task.status, "%s", "writer node group");
- g->eof = 0;
- return 1;
-err_out:
- unregister_task(&g->task);
- while (i > 0) {
- struct writer_node *wn = &g->writer_nodes[--i];
- wn->writer->close(wn);
+ 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);
}
- g->num_writers = 0;
- g->eof = 1;
- return ret;
+ 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;
}
-
/**
- * unregister a writer node group task
+ * Open a writer node and register the corresponding task.
*
- * \param g the group whose task is to be closed
+ * \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 wng_unregister(struct writer_node_group *g)
+void register_writer_node(struct writer_node *wn, struct btr_node *parent,
+ struct sched *s)
{
- unregister_task(&g->task);
- g->eof = 1;
+ struct writer *w = writers + wn->writer_num;
+ char *name = make_message("%s writer", writer_names[wn->writer_num]);
+
+ wn->btrn = btr_new_node(&(struct btr_node_description)
+ EMBRACE(.name = name, .parent = parent,
+ .handler = w->execute, .context = wn));
+ strcpy(wn->task.status, name);
+ free(name);
+ wn->task.pre_select = w->pre_select;
+ wn->task.post_select = w->post_select;
+ register_task(s, &wn->task);
}
/**
- * call the close function of each writer in the given group
- *
- * \param g the writer node group to close
+ * Print the help text of all writers to stdout.
*
- * This function also frees all resources of the given group.
+ * \param detailed Whether to print the detailed help text.
*/
-void wng_close(struct writer_node_group *g)
+void print_writer_helps(int detailed)
{
int i;
- if (!g)
+ printf_or_die("\nAvailable writers: \n\t");
+ FOR_EACH_WRITER(i)
+ printf_or_die("%s%s", i? " " : "", writer_names[i]);
+ printf_or_die("\n\n");
+ FOR_EACH_WRITER(i) {
+ struct writer *w = writers + i;
+
+ if (!w->help.short_help)
+ continue;
+ printf_or_die("Options for %s:\n", writer_names[i]);
+ ggo_print_help(&w->help, detailed);
+ }
+}
+
+static void get_btr_value(struct btr_node *btrn, const char *cmd,
+ int32_t *result)
+{
+ 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;
- PARA_NOTICE_LOG("closing wng with %d writer(s)\n", g->num_writers);
- FOR_EACH_WRITER_NODE(i, g) {
- struct writer_node *wn = &g->writer_nodes[i];
- wn->writer->close(wn);
}
- free(g->writer_nodes);
- free(g);
+ ret = para_atoi32(buf, result);
+ assert(ret >= 0);
+ free(buf);
}
/**
- * allocate and initialize a new writer_node_group struct
+ * Ask parent btr nodes for the sample rate of the current stream.
*
- * \param num_writers the number of writer nodes for the new group
+ * \param btrn Where to start the search.
+ * \param result Filled in by this function.
*
- * \return Pointer to the new writer node group
+ * This function is assumed to succeed and terminates on errors.
*/
-struct writer_node_group *wng_new(unsigned num_writers)
+void get_btr_sample_rate(struct btr_node *btrn, 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->task.private_data = g;
- g->task.post_select = wng_post_select;
- g->task.pre_select = wng_pre_select;
- return g;
+ get_btr_value(btrn, "sample_rate", result);
}
/**
- * call the init function of each supported paraslash writer
- */
-void init_supported_writers(void)
-{
- int i;
-
- FOR_EACH_WRITER(i)
- writers[i].init(&writers[i]);
-}
-/**
- * check if given string is a valid command line for any writer
+ * Ask parent btr nodes for the channel count of the current stream.
*
- * \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, optinally 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
- * returns \p NULL.
+ * \param btrn See \ref get_btr_sample_rate.
+ * \param result See \ref get_btr_sample_rate.
*/
-void *check_writer_arg(const char *wa, int *writer_num)
+void get_btr_channels(struct btr_node *btrn, int32_t *result)
{
- int i;
-
- *writer_num = -E_WRITE_COMMON_SYNTAX;
- 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 != ' ')
- continue;
- if (c && !writers[i].parse_config)
- return NULL;
- *writer_num = i;
- return writers[i].parse_config(c? wa + len + 1 : "");
- }
- PARA_ERROR_LOG("%s", "writer not found\n");
- return NULL;
+ get_btr_value(btrn, "channels", result);
}
/**
- * setup a writer node group with only one writer, the default writer
- *
- * The writer which is set up depends on the OS. It defaults to alsa for Linux,
- * osx_write for OS X, file writer if neither of these is supported.
+ * Ask parent btr nodes for the number of bits per sample and the byte sex.
*
- * \return pointer to the allocated writer node group
+ * \param btrn See \ref get_btr_sample_rate.
+ * \param result Contains the sample format as an enum sample_format type.
*/
-struct writer_node_group *setup_default_wng(void)
+void get_btr_sample_format(struct btr_node *btrn, int32_t *result)
{
- struct writer_node_group *wng = wng_new(1);
- wng->writer_nodes[0].writer = &writers[DEFAULT_WRITER];
- PARA_INFO_LOG("using default writer: %s %p\n",
- writer_names[DEFAULT_WRITER], writers[DEFAULT_WRITER].parse_config);
- wng->writer_nodes[0].conf = writers[DEFAULT_WRITER].parse_config("");
- return wng;
+ get_btr_value(btrn, "sample_format", result);
}