From: Andre Noll Date: Mon, 4 Jan 2010 06:26:56 +0000 (+0100) Subject: Add btr support for the compress filter. X-Git-Tag: v0.4.2~207 X-Git-Url: http://git.tue.mpg.de/?a=commitdiff_plain;h=aa3c8a3f067dafda3e2f4c45b43e58246d0eb587;p=paraslash.git Add btr support for the compress filter. --- diff --git a/compress_filter.c b/compress_filter.c index 90af8abd..b17549c3 100644 --- a/compress_filter.c +++ b/compress_filter.c @@ -87,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 = prepare_filter_node(fn); + if (ret < 0) + goto err; + if (ret == 0) + return; + 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); + close_compress(fn); + t->error = ret; + btr_del_node(btrn); +} + /** TODO: Add sanity checks */ static int compress_parse_config(int argc, char **argv, void **config) { @@ -110,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; @@ -129,6 +197,8 @@ void compress_filter_init(struct filter *f) f->open = open_compress; f->close = close_compress; f->convert = compress; + 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,