X-Git-Url: http://git.tuebingen.mpg.de/?p=paraslash.git;a=blobdiff_plain;f=compress_filter.c;h=7c392746c5c1e02bc03408876e8af65f1a8118da;hp=bf129b4ad13b939ec7ceb7cbbd2cf6d3e9563225;hb=aa234b7afe223879a7bd7274ce05a3a315a2ec49;hpb=3062b9da4f71ea236f46efda9a51add1b15cda1a

diff --git a/compress_filter.c b/compress_filter.c
index bf129b4a..7c392746 100644
--- a/compress_filter.c
+++ b/compress_filter.c
@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2005-2008 Andre Noll <maan@systemlinux.org>
+ * Copyright (C) 2005-2009 Andre Noll <maan@systemlinux.org>
  *
  * Licensed under the GPL v2. For licencing details see COPYING.
  */
@@ -10,10 +10,15 @@
  * Uses ideas of AudioCompress, (C) 2002-2004  M. Hari Nezumi <magenta@trikuare.cx>
  */
 
+#include <regex.h>
+#include <stdbool.h>
+
 #include "para.h"
 #include "compress_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"
@@ -21,8 +26,6 @@
 /** The size of the output data buffer. */
 #define COMPRESS_CHUNK_SIZE 40960
 
-extern char *stat_item_values[NUM_STAT_ITEMS];
-
 /** Data specific to the compress filter. */
 struct private_compress_data {
 	/** The current multiplier. */
@@ -52,7 +55,7 @@ static ssize_t compress(char *inbuf, size_t inbuf_len, struct filter_node *fn)
 		/* be careful in that heat, my dear */
 		int sample = *ip++, adjusted_sample = (PARA_ABS(sample) *
 			pcd->current_gain) >> gain_shift;
-		if (unlikely(adjusted_sample > 32767)) { /* clip */
+		if (adjusted_sample > 32767) { /* clip */
 			PARA_NOTICE_LOG("clip: sample: %d, adjusted sample: %d\n",
 				sample, adjusted_sample);
 			adjusted_sample = 32767;
@@ -62,7 +65,7 @@ static ssize_t compress(char *inbuf, size_t inbuf_len, struct filter_node *fn)
 		} else
 			pcd->peak = PARA_MAX(pcd->peak, adjusted_sample);
 		*op++ = sample >= 0? adjusted_sample : -adjusted_sample;
-		if (likely(++pcd->num_samples & mask))
+		if (++pcd->num_samples & mask)
 			continue;
 //		PARA_DEBUG_LOG("gain: %u, peak: %u\n", pcd->current_gain,
 //			pcd->peak);
@@ -84,6 +87,73 @@ static void close_compress(struct filter_node *fn)
 	free(fn->buf);
 }
 
+static void compress_post_select(__a_unused struct sched *s, struct task *t)
+{
+	struct filter_node *fn = container_of(t, struct filter_node, task);
+	struct private_compress_data *pcd = fn->private_data;
+	struct btr_node *btrn = fn->btrn;
+	bool inplace = btr_inplace_ok(btrn);
+	int ret;
+	char *inbuf;
+	size_t length, i;
+	int16_t *ip, *op;
+	unsigned gain_shift = pcd->conf->inertia_arg + pcd->conf->damp_arg,
+		mask = (1 << pcd->conf->blocksize_arg) - 1;
+
+	//inplace = false;
+next_buffer:
+	t->error = 0;
+	ret = btr_node_status(btrn, fn->min_iqs);
+	if (ret < 0)
+		goto err;
+	if (ret == 0)
+		return;
+	btr_merge(btrn, fn->min_iqs);
+	length = btr_next_buffer(btrn, &inbuf) & ~(size_t)1;
+	ip = (int16_t *)inbuf;
+	if (inplace)
+		op = ip;
+	else
+		op = para_malloc(length);
+	for (i = 0; i < length / 2; i++) {
+		/* be careful in that heat, my dear */
+		int sample = *ip++, adjusted_sample = (PARA_ABS(sample) *
+			pcd->current_gain) >> gain_shift;
+		if (adjusted_sample > 32767) { /* clip */
+			PARA_NOTICE_LOG("clip: sample: %d, adjusted sample: %d\n",
+				sample, adjusted_sample);
+			adjusted_sample = 32767;
+			pcd->current_gain = (3 * pcd->current_gain +
+				(1 << pcd->conf->inertia_arg)) / 4;
+			pcd->peak = 0;
+		} else
+			pcd->peak = PARA_MAX(pcd->peak, adjusted_sample);
+		op[i] = sample >= 0? adjusted_sample : -adjusted_sample;
+		if (++pcd->num_samples & mask)
+			continue;
+//		PARA_DEBUG_LOG("gain: %u, peak: %u\n", pcd->current_gain,
+//			pcd->peak);
+		if (pcd->peak < pcd->conf->target_level_arg) {
+			if (pcd->current_gain < pcd->max_gain)
+				pcd->current_gain++;
+		} else
+			pcd->current_gain = PARA_MAX(pcd->current_gain - 2,
+				1U << pcd->conf->inertia_arg);
+		pcd->peak = 0;
+	}
+	if (inplace)
+		btr_pushdown_one(btrn);
+	else {
+		btr_consume(btrn, length);
+		btr_add_output((char *)op, length, btrn);
+	}
+	goto next_buffer;
+err:
+	assert(ret < 0);
+	t->error = ret;
+	btr_remove_node(btrn);
+}
+
 /** TODO: Add sanity checks */
 static int compress_parse_config(int argc, char **argv, void **config)
 {
@@ -107,6 +177,7 @@ static void open_compress(struct filter_node *fn)
 		sizeof(struct private_compress_data));
 	pcd->conf = fn->conf;
 	fn->private_data = pcd;
+	fn->min_iqs = 2; /* 16 bit audio */
 	fn->bufsize = COMPRESS_CHUNK_SIZE;
 	fn->buf = para_malloc(fn->bufsize);
 	pcd->current_gain = 1 << pcd->conf->inertia_arg;
@@ -120,9 +191,17 @@ static void open_compress(struct filter_node *fn)
  */
 void compress_filter_init(struct filter *f)
 {
+	struct compress_filter_args_info dummy;
+
+	compress_cmdline_parser_init(&dummy);
 	f->open = open_compress;
 	f->close = close_compress;
 	f->convert = compress;
-	f->print_help = compress_cmdline_parser_print_help;
+	f->pre_select = generic_filter_pre_select;
+	f->post_select = compress_post_select;
 	f->parse_config = compress_parse_config;
+	f->help = (struct ggo_help) {
+		.short_help = compress_filter_args_info_help,
+		.detailed_help = compress_filter_args_info_detailed_help
+	};
 }