X-Git-Url: http://git.tuebingen.mpg.de/?a=blobdiff_plain;f=amp_filter.c;h=0ed2d8e0f13e5391698901fc9dd6e5af1af3463d;hb=e16a8a93779d7a94cf0c716f6dab3b42651d0205;hp=22784955513ead8b5ba21db76af799e5fadbf66a;hpb=b779fcfc62bca2e728348c70bab3c50df3fe63b8;p=paraslash.git

diff --git a/amp_filter.c b/amp_filter.c
index 22784955..0ed2d8e0 100644
--- a/amp_filter.c
+++ b/amp_filter.c
@@ -1,15 +1,20 @@
 /*
- * Copyright (C) 2008 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2009 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
 
 /** \file amp_filter.c Paraslash's amplify filter. */
 
+#include <regex.h>
+#include <stdbool.h>
+
 #include "para.h"
 #include "amp_filter.cmdline.h"
 #include "list.h"
 #include "sched.h"
+#include "ggo.h"
+#include "buffer_tree.h"
 #include "filter.h"
 #include "string.h"
 #include "error.h"
@@ -29,19 +34,29 @@ struct private_amp_data {
 
 static ssize_t amp_convert(char *inbuf, size_t inbuf_len, struct filter_node *fn)
 {
-	size_t i, length = PARA_MIN((inbuf_len / 2) * 2,
-		(fn->bufsize - fn->loaded) / 2 * 2);
+	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;
-	for (i = 0; i < length / 2; i++) {
-		int x = (PARA_ABS(*ip) * (64 + pad->amp)) >> 6;
-		*op++ = *ip++ > 0? PARA_MIN(x, 32767) : PARA_MAX(-x, -32768);
+
+	if (pad->amp == 0) {
+		memcpy(op, ip, length * 2);
+		goto out;
 	}
-	fn->loaded += length;
-	return length;
+	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)
@@ -75,26 +90,81 @@ static void amp_open(struct filter_node *fn)
 
 	pad->conf = fn->conf;
 	fn->private_data = pad;
-	if (!pad->conf->amp_given && stat_item_values[SI_AMPLIFICATION]) {
-		int i = SI_AMPLIFICATION;
-		char *s = stat_item_values[i] + strlen(status_item_list[i]) + 1;
-		sscanf(s, "%u", &pad->amp);
-	} else
+	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);
 }
 
+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;
+	char *in;
+	size_t in_len = btr_next_buffer(btrn, &in);
+	size_t i, length = in_len / 2;
+	int16_t *ip = (int16_t *)in, *op;
+	bool inplace = btr_inplace_ok(btrn);
+
+	ret = prepare_filter_node(fn);
+	if (ret < 0)
+		goto err;
+	if (ret == 0)
+		return;
+
+	if (inplace)
+		op = ip;
+	else
+		op = para_malloc(length * 2);
+	if (pad->amp == 0) {
+		memcpy(op, ip, length * 2);
+		goto success;
+	}
+	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;
+	}
+success:
+	t->error = 0;
+	if (inplace)
+		btr_pushdown_one(btrn);
+	else {
+		btr_consume(btrn, length * 2);
+		btr_add_output((char *)op, length * 2, btrn);
+	}
+	return;
+err:
+	assert(ret < 0);
+	amp_close(fn);
+	t->error = ret;
+	btr_del_node(btrn);
+}
+
 /**
  * The init function of the amplify filter.
  *
  * \param f Pointer to the struct to initialize.
  */
-void amp_init(struct filter *f)
+void amp_filter_init(struct filter *f)
 {
+	struct amp_filter_args_info dummy;
+
+	amp_cmdline_parser_init(&dummy);
 	f->open = amp_open;
 	f->close = amp_close;
 	f->convert = amp_convert;
-	f->print_help = amp_cmdline_parser_print_help;
+	f->pre_select = generic_filter_pre_select;
+	f->post_select = amp_post_select;
 	f->parse_config = amp_parse_config;
+	f->help = (struct ggo_help) {
+		.short_help = amp_filter_args_info_help,
+		.detailed_help = amp_filter_args_info_detailed_help
+	};
 }