X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=amp_filter.c;h=e91791bf8655e58cbc494901ad232fc1787cbe10;hp=e180db437c6a26fe8c978fefccdd51841cd97fd0;hb=c282c836791cedf57c128555af90af37c7c01c05;hpb=3ddbf5db98280cb09ef74726573b0b3f3b7e62d6

diff --git a/amp_filter.c b/amp_filter.c
index e180db43..e91791bf 100644
--- a/amp_filter.c
+++ b/amp_filter.c
@@ -7,12 +7,14 @@
 /** \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"
@@ -88,6 +90,7 @@ 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
@@ -96,6 +99,68 @@ static void amp_open(struct filter_node *fn)
 	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;
+	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);
+		amp_close(fn);
+		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);
+}
+
 /**
  * The init function of the amplify filter.
  *
@@ -109,7 +174,10 @@ void amp_filter_init(struct filter *f)
 	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