Replace 2010 in copyright message by 2011.

[paraslash.git] / amp_filter.c

diff --git a/amp_filter.c b/amp_filter.c
index b3aa27000f4e0039c8137674715ba06777ebe04c..21fd9fd179a3885e7f25664fbfae60029922536b 100644 (file)
--- a/amp_filter.c
+++ b/amp_filter.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2009 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2009-2011 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
@@ -19,9 +19,6 @@
 #include "string.h"
 #include "error.h"
 
-/** The size of the output data buffer. */
-#define AMP_CHUNK_SIZE 40960
-
 extern char *stat_item_values[NUM_STAT_ITEMS];
 
 /** Data specific to the amplify filter. */
@@ -32,37 +29,9 @@ struct private_amp_data {
        unsigned amp;
 };
 
-static ssize_t amp_convert(char *inbuf, size_t inbuf_len, struct filter_node *fn)
-{
-       size_t i, length = PARA_MIN((inbuf_len / 2),
-               (fn->bufsize - fn->loaded) / 2);
-       struct private_amp_data *pad = fn->private_data;
-       int16_t *ip = (int16_t *)inbuf, *op = (int16_t *)(fn->buf + fn->loaded);
-       int factor = 64 + pad->amp;
-
-       if (!length)
-               return 0;
-
-       if (pad->amp == 0) {
-               memcpy(op, ip, length * 2);
-               goto out;
-       }
-       for (i = 0; i < length; i++) {
-               int x = (ip[i] * factor) >> 6;
-
-               op[i] = x;
-               if (op[i] != x)
-                       op[i] = (x >= 32768)? 32767 : -32768;
-       }
-out:
-       fn->loaded += length * 2;
-       return length * 2;
-}
-
 static void amp_close(struct filter_node *fn)
 {
        free(fn->private_data);
-       free(fn->buf);
 }
 
 static int amp_parse_config(int argc, char **argv, void **config)
@@ -75,8 +44,6 @@ static int amp_parse_config(int argc, char **argv, void **config)
        ret = -ERRNO_TO_PARA_ERROR(EINVAL);
        if (amp_conf->amp_arg < 0)
                goto err;
-       PARA_NOTICE_LOG("amplification: %u (scaling factor: %1.2f)\n",
-               amp_conf->amp_arg, amp_conf->amp_arg / 64.0 + 1.0);
        *config = amp_conf;
        return 1;
 err:
@@ -90,12 +57,74 @@ static void amp_open(struct filter_node *fn)
 
        pad->conf = fn->conf;
        fn->private_data = pad;
+       fn->min_iqs = 2;
        if (!pad->conf->amp_given && stat_item_values[SI_AMPLIFICATION])
                sscanf(stat_item_values[SI_AMPLIFICATION], "%u", &pad->amp);
        else
                pad->amp = pad->conf->amp_arg;
-       fn->bufsize = AMP_CHUNK_SIZE;
-       fn->buf = para_malloc(fn->bufsize);
+       PARA_NOTICE_LOG("amplification: %u (scaling factor: %1.2f)\n",
+               pad->amp, pad->amp / 64.0 + 1.0);
+}
+
+static void amp_post_select(__a_unused struct sched *s, struct task *t)
+{
+       struct filter_node *fn = container_of(t, struct filter_node, task);
+       struct private_amp_data *pad = fn->private_data;
+       struct btr_node *btrn = fn->btrn;
+       int ret, factor = 64 + pad->amp;
+       size_t i, in_bytes, len;
+       int16_t *in, *out;
+       bool inplace = btr_inplace_ok(btrn);
+
+       if (pad->amp == 0) { /* no amplification */
+               t->error = -E_AMP_ZERO_AMP;
+               btr_splice_out_node(btrn);
+               return;
+       }
+next_buffer:
+       ret = btr_node_status(btrn, fn->min_iqs, BTR_NT_INTERNAL);
+       if (ret < 0)
+               goto err;
+       if (ret == 0)
+               return;
+       btr_merge(btrn, fn->min_iqs);
+       in_bytes = btr_next_buffer(btrn, (char **)&in);
+       len = in_bytes / 2;
+       if (len == 0) { /* eof and in_bytes == 1 */
+               ret = -E_AMP_EOF;
+               goto err;
+       }
+
+       if (inplace)
+               out = in;
+       else
+               out = para_malloc(len * 2);
+
+       for (i = 0; i < len; i++) {
+               int x = (in[i] * factor) >> 6;
+
+               out[i] = x;
+               if (out[i] != x) /* overflow, clip */
+                       out[i] = (x >= 0)? 32767 : -32768;
+       }
+
+       if (inplace)
+               btr_pushdown_one(btrn);
+       else {
+               btr_consume(btrn, len * 2);
+               btr_add_output((char *)out, len * 2, btrn);
+       }
+       t->error = 0;
+       goto next_buffer;
+err:
+       assert(ret < 0);
+       t->error = ret;
+       btr_remove_node(btrn);
+}
+
+static void amp_free_config(void *conf)
+{
+       amp_cmdline_parser_free(conf);
 }
 
 /**
@@ -110,8 +139,10 @@ void amp_filter_init(struct filter *f)
        amp_cmdline_parser_init(&dummy);
        f->open = amp_open;
        f->close = amp_close;
-       f->convert = amp_convert;
+       f->pre_select = generic_filter_pre_select;
+       f->post_select = amp_post_select;
        f->parse_config = amp_parse_config;
+       f->free_config = amp_free_config;
        f->help = (struct ggo_help) {
                .short_help = amp_filter_args_info_help,
                .detailed_help = amp_filter_args_info_detailed_help