Add btr support for the compress filter.

diff --git a/compress_filter.c b/compress_filter.c
index 90af8abdcc8bec3faac008eef4e7720cc27cc761..b17549c3a24b51d74e96c9caa63a30851e4e0094 100644 (file)
--- 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,