add new writer to para_play: file
[paraslash.git] / play.c
diff --git a/play.c b/play.c
index c9cc56abb5374831979cb510bcc89ed337425257..e1fdb2ef9fb44f3e68554a80b79508456cf2cc07 100644 (file)
--- a/play.c
+++ b/play.c
 #include "string.h"
 #include "error.h"
 
-#define FORMAT SND_PCM_FORMAT_S16_LE
+/*
+files:
+~~~~~~
+write.c
+write.h wr
+write_common.c
+write_common.h: decratation of the wng funcs
+alsa_writer.c
+*/
 
-struct private_alsa_data {
-       snd_pcm_t *handle;
-       size_t bytes_per_frame;
-};
+/* write.h */
+enum writer_enum {WRITER_ENUM};
 
+/* write.h */
 struct writer_node {
        struct writer *writer;
        void *private_data;
+       int chunk_bytes;
 };
 
+/* write.h */
 struct writer {
-       int (*open)(struct writer_node **);
+       void (*init)(struct writer *w);
+       int (*open)(struct writer_node *);
        int (*write)(char *data, size_t nbytes, struct writer_node *);
        void (*close)(struct writer_node *);
        void (*shutdown)(struct writer_node *);
 };
 
-#define NUM_WRITERS 1
-static struct writer writers[NUM_WRITERS];
-#define FOR_EACH_WRITER(i) for (i = 0; i < NUM_WRITERS, i++)
-static struct writer_node **writer_nodes;
+/* write.h */
+struct writer_node_group {
+       unsigned num_writers;
+       struct writer_node *writer_nodes;
+       int *written;
+       size_t max_chunk_bytes;
+       int eof;
+};
+
+/* write.h */
+#define FOR_EACH_WRITER_NODE(i, wng) for (i = 0; i < (wng)->num_writers; i++)
+#define FOR_EACH_WRITER(i) for (i = 0; i < NUM_SUPPORTED_WRITERS; i++)
 
+DECLARE_WRITER_INITS;
 
+
+/* write.c */
 static unsigned char *audiobuf;
 static struct timeval *start_time;
 static struct gengetopt_args_info conf;
 
+/* write.c */
 INIT_PLAY_ERRLISTS;
 
-void para_log(__a_unused int ll, const char* fmt,...)
+/* write.c */
+void para_log(int ll, const char* fmt,...)
 {
        va_list argp;
 
+       if (ll < conf.loglevel_arg)
+               return;
        va_start(argp, fmt);
        vfprintf(stderr, fmt, argp);
        va_end(argp);
 }
 
+/* write.c */
 /**
  * read WAV_HEADER_LEN bytes from stdin to audio buffer
  *
@@ -90,12 +116,19 @@ static int read_wav_header(void)
        return 1;
 }
 
+/* alsa_writer.c */
+#define FORMAT SND_PCM_FORMAT_S16_LE
+struct private_alsa_data {
+       snd_pcm_t *handle;
+       size_t bytes_per_frame;
+};
+
 /*
  * open and prepare the PCM handle for writing
  *
  * Install PCM software and hardware configuration. Exit on errors.
  */
-static int alsa_open(void **private_data)
+static int alsa_open(struct writer_node *w)
 {
        snd_pcm_hw_params_t *hwparams;
        snd_pcm_sw_params_t *swparams;
@@ -107,7 +140,7 @@ static int alsa_open(void **private_data)
        snd_output_t *log;
        snd_pcm_uframes_t period_size;
        struct private_alsa_data *pad = para_malloc(sizeof(struct private_alsa_data));
-       *private_data = pad;
+       w->private_data = pad;
 
        snd_pcm_info_alloca(&info);
        if (snd_output_stdio_attach(&log, stderr, 0) < 0)
@@ -145,7 +178,8 @@ static int alsa_open(void **private_data)
                return -E_HW_PARAMS;
        snd_pcm_hw_params_get_period_size(hwparams, &period_size, 0);
        snd_pcm_hw_params_get_buffer_size(hwparams, &buffer_size);
-       PARA_DEBUG_LOG("buffer size: %lu, period_size: %lu\n", buffer_size, period_size);
+       PARA_DEBUG_LOG("buffer size: %lu, period_size: %lu\n", buffer_size,
+               period_size);
        if (period_size == buffer_size)
                return -E_BAD_PERIOD;
        snd_pcm_sw_params_current(pad->handle, swparams);
@@ -174,6 +208,7 @@ static int alsa_open(void **private_data)
        return period_size * pad->bytes_per_frame;
 }
 
+/* alsa_writer.c */
 /**
  * push out pcm frames
  * \param data pointer do data to be written
@@ -181,9 +216,9 @@ static int alsa_open(void **private_data)
  *
  * \return Number of bytes written, -E_ALSA_WRITE on errors.
  */
-static int alsa_write(char *data, size_t nbytes, void *private_data)
+static int alsa_write(char *data, size_t nbytes, struct writer_node *wn)
 {
-       struct private_alsa_data *pad = private_data;
+       struct private_alsa_data *pad = wn->private_data;
        size_t frames = nbytes / pad->bytes_per_frame;
        unsigned char *d = data;
        snd_pcm_sframes_t r, result = 0;
@@ -208,15 +243,17 @@ static int alsa_write(char *data, size_t nbytes, void *private_data)
        return result * pad->bytes_per_frame;
 }
 
-static void alsa_close(void *private_data)
+/* alsa_writer.c */
+static void alsa_close(struct writer_node *wn)
 {
-       struct private_alsa_data *pad = private_data;
+       struct private_alsa_data *pad = wn->private_data;
        snd_pcm_drain(pad->handle);
        snd_pcm_close(pad->handle);
        snd_config_update_free_global();
        free(pad);
 }
 
+/* alsa_writer.c */
 void alsa_writer_init(struct writer *w)
 {
        w->open = alsa_open;
@@ -226,6 +263,57 @@ void alsa_writer_init(struct writer *w)
 }
 
 
+
+
+/* file_writer.c */
+
+struct private_file_writer_data {
+       int fd;
+};
+static int file_writer_open(struct writer_node *w)
+{
+       struct private_file_writer_data *pfwd = para_calloc(
+               sizeof(struct private_file_writer_data));
+       char *tmp = para_tmpname(), *home = para_homedir(),
+               *filename = make_message("%s/.paraslash/%s", home, tmp);
+
+       free(home);
+       free(tmp);
+       w->private_data = pfwd;
+       pfwd->fd = open(filename, O_WRONLY | O_CREAT, S_IRUSR | S_IWUSR);
+       free(filename);
+       if (pfwd->fd >= 0)
+               return 8192;
+       free(pfwd);
+       return -E_FW_OPEN;
+}
+
+static int file_writer_write(char *data, size_t nbytes, struct writer_node *wn)
+{
+       struct private_file_writer_data *pfwd = wn->private_data;
+       int ret = write(pfwd->fd, data, nbytes);
+       if (ret < 0)
+               ret = -E_FW_WRITE;
+       return ret;
+}
+
+static void file_writer_close(struct writer_node *wn)
+{
+       struct private_file_writer_data *pfwd = wn->private_data;
+       close(pfwd->fd);
+       free(pfwd);
+}
+
+void file_writer_init(struct writer *w)
+{
+       w->open = file_writer_open;
+       w->write = file_writer_write;
+       w->close = file_writer_close;
+       w->shutdown = NULL; /* nothing to do */
+}
+
+
+/* write.c */
 /**
  * check if current time is later than start_time
  * \param diff pointer to write remaining time to
@@ -246,6 +334,7 @@ static int start_time_in_future(struct timeval *diff)
        return tv_diff(start_time, &now, diff) > 0? 1 : 0;
 }
 
+/* write.c */
 /**
  * sleep until time given at command line
  *
@@ -261,6 +350,102 @@ static void do_initial_delay(struct timeval *delay)
        while (start_time_in_future(delay));
 }
 
+/* write.c */
+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;
+}
+
+/* write_common.c */
+
+const char *writer_names[] ={WRITER_NAMES};
+static struct writer writers[NUM_SUPPORTED_WRITERS] = {WRITER_ARRAY};
+
+int wng_write(struct writer_node_group *g, char *buf, size_t *loaded)
+{
+       int ret, i, need_more_writes = 1;
+       size_t min_written = 0;
+
+       while (need_more_writes) {
+               need_more_writes = 0;
+               FOR_EACH_WRITER_NODE(i, g) {
+                       size_t w = g->written[i];
+                       unsigned char *p = buf + w;
+                       int bytes_to_write;
+                       struct writer_node *wn = &g->writer_nodes[i];
+                       if (!i)
+                               min_written = w;
+                       else
+                               min_written = PARA_MIN(min_written, w);
+                       if (w == *loaded)
+                               continue;
+                       if (!g->eof && (*loaded < wn->chunk_bytes + w))
+                               continue;
+                       bytes_to_write = PARA_MIN(wn->chunk_bytes,
+                               *loaded - w);
+                       ret = wn->writer->write(p, bytes_to_write, wn);
+                       if (ret < 0)
+                               goto out;
+                       if (ret != bytes_to_write)
+                               PARA_WARNING_LOG("short write: %d/%d\n", ret,
+                                       bytes_to_write);
+                       g->written[i] += ret;
+                       need_more_writes = 1;
+               }
+       }
+       *loaded -= min_written;
+       ret = 0;
+       if (g->eof)
+               goto out;
+       if (*loaded)
+               memmove(buf, buf + min_written, *loaded);
+       FOR_EACH_WRITER_NODE(i, g)
+               g->written[i] -= min_written;
+       ret = 1;
+out:
+       return ret;
+}
+
+/* write_common.c */
+int wng_open(struct writer_node_group *g)
+{
+       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);
+       }
+out:
+       return ret;
+}
+
+/* write_common.c */
+void wng_close(struct writer_node_group *g)
+{
+       int i;
+
+       FOR_EACH_WRITER_NODE(i, g) {
+               struct writer_node *wn = &g->writer_nodes[i];
+               wn->writer->close(wn);
+       }
+}
+
+/* write.c */
 /**
  * play raw pcm data
  * \param loaded number of bytes already loaded
@@ -271,54 +456,162 @@ static void do_initial_delay(struct timeval *delay)
  *
  * \return positive on success, negative on errors.
  */
-static int play_pcm(size_t loaded)
+static int pcm_write(struct writer_node_group *wng, size_t loaded)
 {
-       size_t bufsize, written = 0, prebuf_size;
-       unsigned char *p;
+       size_t bufsize, prebuf_size, bytes_to_load;
        struct timeval delay;
-       void *private_data;
-       int chunk_bytes, ret = alsa_open(&private_data);
+       int ret, not_yet_started = 1;
 
+       ret = wng_open(wng);
        if (ret < 0)
                goto out;
-       chunk_bytes = ret;
-       bufsize = (conf.bufsize_arg * 1024 / chunk_bytes) * chunk_bytes;
+       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;
+       }
+       if (not_yet_started && start_time && start_time_in_future(&delay))
+               do_initial_delay(&delay);
+       not_yet_started = 0;
 again:
-       if (!written) {
-               if (loaded < prebuf_size)
-                       goto read;
-               if (start_time && start_time_in_future(&delay)) {
-                       do_initial_delay(&delay);
-                       start_time = NULL;
+       ret = wng_write(wng, audiobuf, &loaded);
+       if (ret <= 0)
+               goto out;
+       ret = -E_PLAY_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;
+}
+
+/* writer_node.c */
+struct writer_node_group *wng_new(unsigned num_writers)
+{
+       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));
+       return g;
+}
+
+/* writer_node.c */
+void wng_destroy(struct writer_node_group *g)
+{
+       if (!g)
+               return;
+       free(g->written);
+       free(g->writer_nodes);
+       free(g);
+}
+
+void init_supported_writers(void)
+{
+       int i;
+
+       FOR_EACH_WRITER(i)
+               writers[i].init(&writers[i]);
+}
+
+int check_writer_arg(const char *arg)
+{
+       int i, ret = -E_PLAY_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;
+}
+
+struct writer_node_group *setup_default_wng(void)
+{
+       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;
+}
+
+/* write.c */
+
+struct writer_node_group *check_args(void)
+{
+       int i, ret = -E_PLAY_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;
                }
+               fprintf(stderr, "%s\n", msg);
+               free(msg);
+               exit(EXIT_SUCCESS);
        }
-       p = audiobuf;
-       while (loaded >= chunk_bytes) {
-               ret = alsa_write(p, chunk_bytes, private_data);
-               if (ret < 0)
+       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;
-               p += ret;
-               written += ret;
-               loaded -= ret;
+               tv.tv_sec = sec;
+               tv.tv_usec = usec;
+               start_time = &tv;
        }
-       if (loaded && p != audiobuf)
-               memmove(audiobuf, p, loaded);
-read:
-       ret = read(STDIN_FILENO, audiobuf + loaded, bufsize - loaded);
-       if (ret < 0) {
-               ret = -E_READ_STDIN;
+       if (!conf.writer_given) {
+               wng = setup_default_wng();
+               ret = 1;
                goto out;
        }
-       if (ret) {
-               loaded += ret;
-               goto again;
+       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];
        }
        ret = 1;
 out:
-       alsa_close(private_data);
-       return ret;
+       if (ret > 0)
+               return wng;
+       free(wng);
+       return NULL;
 }
 
 /**
@@ -337,30 +630,24 @@ static size_t check_wave(void)
        return 0;
 }
 
+/* write.c */
 int main(int argc, char *argv[])
 {
-       struct timeval tv;
-       int ret;
+       int ret = -E_PLAY_SYNTAX;
+       struct writer_node_group *wng = NULL;
 
        cmdline_parser(argc, argv, &conf);
-       if (conf.start_time_given) {
-               ret = -E_PLAY_SYNTAX;
-               if (sscanf(conf.start_time_arg, "%lu:%lu",
-                               &tv.tv_sec, &tv.tv_usec) != 2)
-                       goto out;
-               start_time = &tv;
-       }
-       /* call init for each supported writer */
-       alsa_writer_init(&writers[0]);
-       /* one for each given writer */
-       writer_nodes = para_calloc(2 * sizeof(struct writer_node));
-
+       wng = check_args();
+       if (!wng)
+               goto out;
+       init_supported_writers();
        audiobuf = para_malloc(WAV_HEADER_LEN);
        ret = read_wav_header();
        if (ret < 0)
                goto out;
-       ret = play_pcm(check_wave());
+       ret = pcm_write(wng, check_wave());
 out:
+       wng_destroy(wng);
        free(audiobuf);
        if (ret < 0)
                PARA_ERROR_LOG("%s\n", PARA_STRERROR(-ret));