rename vector_fmul_add() vector_fmul_reverse_c() and make them static.
[paraslash.git] / wmadec_filter.c
index 8298296df3cc5fc3f42006a09abe5d0569725138..c49489b70eea8d1d9ae271f318cefbb3a349e1e8 100644 (file)
@@ -19,7 +19,6 @@
 
 #include <sys/time.h>
 #include <inttypes.h>
-#include <math.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <math.h>
@@ -34,7 +33,7 @@
 #include "sched.h"
 #include "filter.h"
 #include "bitstream.h"
-#include "mdct.h"
+#include "imdct.h"
 #include "wma.h"
 #include "wmadata.h"
 
@@ -96,7 +95,6 @@ struct private_wmadec_data {
        int next_block_len_bits;        ///< log2 of next block length
        int prev_block_len_bits;        ///< log2 of prev block length
        int block_len;          ///< block length in samples
-       int block_num;          ///< block number in current frame
        int block_pos;          ///< current position in frame
        uint8_t ms_stereo;      ///< true if mid/side stereo mode
        uint8_t channel_coded[MAX_CHANNELS];    ///< true if channel is coded
@@ -133,12 +131,33 @@ struct private_wmadec_data {
 #define VLCBITS 9
 #define VLCMAX ((22+VLCBITS-1)/VLCBITS)
 
+DECLARE_ALIGNED(16, float, ff_sine_128[128]);
+DECLARE_ALIGNED(16, float, ff_sine_256[256]);
+DECLARE_ALIGNED(16, float, ff_sine_512[512]);
+DECLARE_ALIGNED(16, float, ff_sine_1024[1024]);
+DECLARE_ALIGNED(16, float, ff_sine_2048[2048]);
+DECLARE_ALIGNED(16, float, ff_sine_4096[4096]);
+
+static float *ff_sine_windows[6] = {
+       ff_sine_128, ff_sine_256, ff_sine_512, ff_sine_1024,
+       ff_sine_2048, ff_sine_4096
+};
+
+/* Generate a sine window. */
+static void sine_window_init(float *window, int n)
+{
+       int i;
+
+       for (i = 0; i < n; i++)
+               window[i] = sinf((i + 0.5) * (M_PI / (2.0 * n)));
+}
+
 static int wmadec_cleanup(struct private_wmadec_data *s)
 {
        int i;
 
        for (i = 0; i < s->nb_block_sizes; i++)
-               mdct_end(s->mdct_ctx[i]);
+               imdct_end(s->mdct_ctx[i]);
 
        if (s->use_exp_vlc)
                free_vlc(&s->exp_vlc);
@@ -168,9 +187,9 @@ static void init_coef_vlc(struct vlc *vlc, uint16_t **prun_table,
 
        init_vlc(vlc, VLCBITS, n, table_bits, 1, 1, table_codes, 4, 4);
 
-       run_table = para_malloc(n * sizeof (uint16_t));
-       level_table = para_malloc(n * sizeof (uint16_t));
-       int_table = para_malloc(n * sizeof (uint16_t));
+       run_table = para_malloc(n * sizeof(uint16_t));
+       level_table = para_malloc(n * sizeof(uint16_t));
+       int_table = para_malloc(n * sizeof(uint16_t));
        i = 2;
        level = 1;
        k = 0;
@@ -446,14 +465,13 @@ static int wma_decode_init(char *initial_buf, int len, struct private_wmadec_dat
        struct private_wmadec_data *s;
        int ret, i;
 
-       if (len < 18)
-               return 0;
-
        PARA_NOTICE_LOG("initial buf: %d bytes\n", len);
        s = para_calloc(sizeof(*s));
        ret = read_asf_header(initial_buf, len, &s->ahi);
-       if (ret < 0)
+       if (ret <= 0) {
+               free(s);
                return ret;
+       }
 
        s->use_exp_vlc = s->ahi.flags2 & 0x0001;
        s->use_bit_reservoir = s->ahi.flags2 & 0x0002;
@@ -464,21 +482,21 @@ static int wma_decode_init(char *initial_buf, int len, struct private_wmadec_dat
                return ret;
        /* init MDCT */
        for (i = 0; i < s->nb_block_sizes; i++) {
-               ret = mdct_init(s->frame_len_bits - i + 1, 1, &s->mdct_ctx[i]);
+               ret = imdct_init(s->frame_len_bits - i + 1, &s->mdct_ctx[i]);
                if (ret < 0)
                        return ret;
        }
        if (s->use_noise_coding) {
                PARA_INFO_LOG("using noise coding\n");
                init_vlc(&s->hgain_vlc, HGAINVLCBITS,
-                       sizeof (ff_wma_hgain_huffbits), ff_wma_hgain_huffbits,
+                       sizeof(ff_wma_hgain_huffbits), ff_wma_hgain_huffbits,
                        1, 1, ff_wma_hgain_huffcodes, 2, 2);
        }
 
        if (s->use_exp_vlc) {
                PARA_INFO_LOG("using exp_vlc\n");
                init_vlc(&s->exp_vlc, EXPVLCBITS,
-               sizeof (ff_wma_scale_huffbits), ff_wma_scale_huffbits,
+               sizeof(ff_wma_scale_huffbits), ff_wma_scale_huffbits,
                1, 1, ff_wma_scale_huffcodes, 4, 4);
        } else {
                PARA_INFO_LOG("using curve\n");
@@ -611,18 +629,21 @@ static int decode_exp_vlc(struct private_wmadec_data *s, int ch)
        return 0;
 }
 
-static void vector_fmul_add(float *dst, const float *src0, const float *src1,
-               const float *src2, int src3, int len, int step)
+/* compute src0 * src1 + src2 */
+static inline void vector_mult_add(float *dst, const float *src0, const float *src1,
+               const float *src2, int len)
 {
        int i;
+
        for (i = 0; i < len; i++)
-               dst[i * step] = src0[i] * src1[i] + src2[i] + src3;
+               dst[i] = src0[i] * src1[i] + src2[i];
 }
 
-static void vector_fmul_reverse_c(float *dst, const float *src0,
+static inline void vector_mult_reverse(float *dst, const float *src0,
                const float *src1, int len)
 {
        int i;
+
        src1 += len - 1;
        for (i = 0; i < len; i++)
                dst[i] = src0[i] * src1[-i];
@@ -644,19 +665,18 @@ static void wma_window(struct private_wmadec_data *s, float *out)
                block_len = s->block_len;
                bsize = s->frame_len_bits - s->block_len_bits;
 
-               vector_fmul_add(out, in, s->windows[bsize],
-                                        out, 0, block_len, 1);
+               vector_mult_add(out, in, s->windows[bsize], out, block_len);
 
        } else {
                block_len = 1 << s->prev_block_len_bits;
                n = (s->block_len - block_len) / 2;
                bsize = s->frame_len_bits - s->prev_block_len_bits;
 
-               vector_fmul_add(out + n, in + n, s->windows[bsize],
-                                        out + n, 0, block_len, 1);
+               vector_mult_add(out + n, in + n, s->windows[bsize], out + n,
+                       block_len);
 
                memcpy(out + n + block_len, in + n + block_len,
-                      n * sizeof (float));
+                      n * sizeof(float));
        }
 
        out += s->block_len;
@@ -667,19 +687,19 @@ static void wma_window(struct private_wmadec_data *s, float *out)
                block_len = s->block_len;
                bsize = s->frame_len_bits - s->block_len_bits;
 
-               vector_fmul_reverse_c(out, in, s->windows[bsize], block_len);
+               vector_mult_reverse(out, in, s->windows[bsize], block_len);
 
        } else {
                block_len = 1 << s->next_block_len_bits;
                n = (s->block_len - block_len) / 2;
                bsize = s->frame_len_bits - s->next_block_len_bits;
 
-               memcpy(out, in, n * sizeof (float));
+               memcpy(out, in, n * sizeof(float));
 
-               vector_fmul_reverse_c(out + n, in + n, s->windows[bsize],
+               vector_mult_reverse(out + n, in + n, s->windows[bsize],
                                      block_len);
 
-               memset(out + n + block_len, 0, n * sizeof (float));
+               memset(out + n + block_len, 0, n * sizeof(float));
        }
 }
 
@@ -779,7 +799,6 @@ static int wma_decode_block(struct private_wmadec_data *s)
 
        /* complex coding */
        if (s->use_noise_coding) {
-
                for (ch = 0; ch < s->ahi.channels; ch++) {
                        if (s->channel_coded[ch]) {
                                int i, m, a;
@@ -1091,7 +1110,7 @@ next:
                if (s->channel_coded[ch])
                        imdct(s->mdct_ctx[bsize], s->output, s->coefs[ch]);
                else if (!(s->ms_stereo && ch == 1))
-                       memset(s->output, 0, sizeof (s->output));
+                       memset(s->output, 0, sizeof(s->output));
 
                /* multiply by the window and add in the frame */
                index = (s->frame_len / 2) + s->block_pos - n4;
@@ -1099,7 +1118,6 @@ next:
        }
 
        /* update block number */
-       s->block_num++;
        s->block_pos += s->block_len;
        if (s->block_pos >= s->frame_len)
                return 1;
@@ -1130,7 +1148,6 @@ static int wma_decode_frame(struct private_wmadec_data *s, int16_t * samples)
        float *iptr;
 
        /* read each block */
-       s->block_num = 0;
        s->block_pos = 0;
        for (;;) {
                ret = wma_decode_block(s);
@@ -1153,7 +1170,7 @@ static int wma_decode_frame(struct private_wmadec_data *s, int16_t * samples)
                }
                /* prepare for next block */
                memmove(&s->frame_out[ch][0], &s->frame_out[ch][s->frame_len],
-                       s->frame_len * sizeof (float));
+                       s->frame_len * sizeof(float));
        }
        return 0;
 }
@@ -1286,6 +1303,8 @@ static ssize_t wmadec_convert(char *inbuffer, size_t len,
                if (ret <= 0)
                        return ret;
                fn->private_data = pwd;
+               fn->fc->channels = pwd->ahi.channels;
+               fn->fc->samplerate = pwd->ahi.sample_rate;
                return pwd->ahi.header_len;
        }
        /* skip 31 bytes */