comparison wmadec.c @ 1342:f574934c4219 libavcodec

uniformization (now it uses the same trace functions as h264, defined in common.h)

1341:6e5d4ec4f3ab

 * WMA compatible decoder.
 */

#include "avcodec.h"
#include "dsputil.h"

//#define DEBUG_PARAMS
//#define DEBUG_TRACE

/* size of blocks */
#define BLOCK_MIN_BITS 7
#define BLOCK_MAX_BITS 11
#define BLOCK_MAX_SIZE (1 << BLOCK_MAX_BITS)

static void wma_lsp_to_curve_init(WMADecodeContext *s, int frame_len);

#include "wmadata.h"

#ifdef DEBUG_TRACE
#include <stdarg.h>
int frame_count;

static FILE *flog;

void trace(const char *fmt, ...)
{
    va_list ap;
    

    if (!flog) {
        flog = fopen("/tmp/out.log", "w");
        setlinebuf(flog);
    }

    va_start(ap, fmt);
    vfprintf(flog, fmt, ap);
    va_end(ap);
}

#define get_bits(s, n) get_bits_trace(s, n)
#define get_vlc(s, vlc) get_vlc_trace(s, vlc)

unsigned int get_bits_trace(GetBitContext *s, int n)
{
    unsigned int val;
    val = (get_bits)(s, n);
    trace("get_bits(%d) : 0x%x\n", n, val);
    return val;
}

static int get_vlc_trace(GetBitContext *s, VLC *vlc)
{
    int code;
    code = (get_vlc)(s, vlc);
    trace("get_vlc() : %d\n", code);
    return code;
}

static void dump_shorts(const char *name, const short *tab, int n)
{
    int i;

    trace("%s[%d]:\n", name, n);
    for(i=0;i<n;i++) {
        if ((i & 7) == 0)
            trace("%4d: ", i);
        trace(" %5d.0", tab[i]);
        if ((i & 7) == 7)
            trace("\n");
    }
}

static void dump_floats(const char *name, int prec, const float *tab, int n)
{
    int i;

    trace("%s[%d]:\n", name, n);
    for(i=0;i<n;i++) {
        if ((i & 7) == 0)
            trace("%4d: ", i);
        trace(" %8.*f", prec, tab[i]);
        if ((i & 7) == 7)
            trace("\n");
    }
    if ((i & 7) != 0)
        trace("\n");
}

#else

#define trace(fmt, args...)

#endif

/* XXX: use same run/length optimization as mpeg decoders */
static void init_coef_vlc(VLC *vlc, 
                          uint16_t **prun_table, uint16_t **plevel_table,

            high_freq = high_freq * 0.6;
        } else {
            high_freq = high_freq * 0.5;
        }
    }
#ifdef DEBUG_PARAMS
    printf("flags1=0x%x flags2=0x%x\n", flags1, flags2);
    printf("version=%d channels=%d sample_rate=%d bitrate=%d block_align=%d\n",
           s->version, s->nb_channels, s->sample_rate, s->bit_rate, 
           s->block_align);
    printf("bps=%f bps1=%f high_freq=%f bitoffset=%d\n", 
           bps, bps1, high_freq, s->byte_offset_bits);
    printf("use_noise_coding=%d use_exp_vlc=%d nb_block_sizes=%d\n",
           s->use_noise_coding, s->use_exp_vlc, s->nb_block_sizes);
#endif

    /* compute the scale factor band sizes for each MDCT block size */
    {
        int a, b, pos, lpos, k, block_len, i, j, n;
        const uint8_t *table;

                if (end > start)
                    s->exponent_high_bands[k][j++] = end - start;
            }
            s->exponent_high_sizes[k] = j;
#if 0
            trace("%5d: coefs_end=%d high_band_start=%d nb_high_bands=%d: ",
                  s->frame_len >> k, 
                  s->coefs_end[k],
                  s->high_band_start[k],
                  s->exponent_high_sizes[k]);
            for(j=0;j<s->exponent_high_sizes[k];j++)
                trace(" %d", s->exponent_high_bands[k][j]);
            trace("\n");
#endif
        }
    }

#ifdef DEBUG_TRACE
    {
        int i, j;
        for(i = 0; i < s->nb_block_sizes; i++) {
            trace("%5d: n=%2d:", 
                   s->frame_len >> i, 
                   s->exponent_sizes[i]);
            for(j=0;j<s->exponent_sizes[i];j++)
                trace(" %d", s->exponent_bands[i][j]);
            trace("\n");
        }
    }
#endif

    /* init MDCT */

        if (s->use_exp_vlc)
            s->noise_mult = 0.02;
        else
            s->noise_mult = 0.04;
               
#if defined(DEBUG_TRACE)
        for(i=0;i<NOISE_TAB_SIZE;i++)
            s->noise_table[i] = 1.0 * s->noise_mult;
#else
        {
            unsigned int seed;

    int coef_nb_bits, total_gain, parse_exponents;
    float window[BLOCK_MAX_SIZE * 2];
    int nb_coefs[MAX_CHANNELS];
    float mdct_norm;

    trace("***decode_block: %d:%d\n", frame_count - 1, s->block_num);

    /* compute current block length */
    if (s->use_variable_block_len) {
        n = av_log2(s->nb_block_sizes - 1) + 1;
    

                            v = exp_ptr[i];
                            e2 += v * v;
                        }
                        exp_power[j] = e2 / n;
                        last_high_band = j;
                        trace("%d: power=%f (%d)\n", j, exp_power[j], n);
                    }
                    exp_ptr += n;
                }

                /* main freqs and high freqs */

                    *coefs++ = 0.0;
            }
        }
    }

#ifdef DEBUG_TRACE
    for(ch = 0; ch < s->nb_channels; ch++) {
        if (s->channel_coded[ch]) {
            dump_floats("exponents", 3, s->exponents[ch], s->block_len);
            dump_floats("coefs", 1, s->coefs[ch], s->block_len);
        }

        /* nominal case for ms stereo: we do it before mdct */
        /* no need to optimize this case because it should almost
           never happen */
        if (!s->channel_coded[0]) {
#ifdef DEBUG_TRACE
            trace("rare ms-stereo case happened\n");
#endif
            memset(s->coefs[0], 0, sizeof(float) * s->block_len);
            s->channel_coded[0] = 1;
        }
        
        for(i = 0; i < s->block_len; i++) {

{
    int ret, i, n, a, ch, incr;
    int16_t *ptr;
    float *iptr;

    trace("***decode_frame: %d size=%d\n", frame_count++, s->frame_len);

    /* read each block */
    s->block_num = 0;
    s->block_pos = 0;
    for(;;) {

        /* XXX: suppress this */
        memset(&s->frame_out[ch][s->frame_len], 0, 
               s->frame_len * sizeof(float));
    }

#ifdef DEBUG_TRACE
    dump_shorts("samples", samples, n * s->nb_channels);
#endif
    return 0;
}

    WMADecodeContext *s = avctx->priv_data;
    int nb_frames, bit_offset, i, pos, len;
    uint8_t *q;
    int16_t *samples;
    
    trace("***decode_superframe:\n");

    samples = data;

    init_get_bits(&s->gb, buf, buf_size*8);
    

1342:f574934c4219

 * WMA compatible decoder.
 */

#include "avcodec.h"
#include "dsputil.h"

/* size of blocks */
#define BLOCK_MIN_BITS 7
#define BLOCK_MAX_BITS 11
#define BLOCK_MAX_SIZE (1 << BLOCK_MAX_BITS)

static void wma_lsp_to_curve_init(WMADecodeContext *s, int frame_len);

#include "wmadata.h"

#ifdef TRACE
int frame_count = 0;

static void dump_shorts(const char *name, const short *tab, int n)
{
    int i;

    tprintf("%s[%d]:\n", name, n);
    for(i=0;i<n;i++) {
        if ((i & 7) == 0)
            tprintf("%4d: ", i);
        tprintf(" %5d.0", tab[i]);
        if ((i & 7) == 7)
            tprintf("\n");
    }
}

static void dump_floats(const char *name, int prec, const float *tab, int n)
{
    int i;

    tprintf("%s[%d]:\n", name, n);
    for(i=0;i<n;i++) {
        if ((i & 7) == 0)
            tprintf("%4d: ", i);
        tprintf(" %8.*f", prec, tab[i]);
        if ((i & 7) == 7)
            tprintf("\n");
    }
    if ((i & 7) != 0)
        tprintf("\n");
}
#endif

/* XXX: use same run/length optimization as mpeg decoders */
static void init_coef_vlc(VLC *vlc, 
                          uint16_t **prun_table, uint16_t **plevel_table,

            high_freq = high_freq * 0.6;
        } else {
            high_freq = high_freq * 0.5;
        }
    }
    dprintf("flags1=0x%x flags2=0x%x\n", flags1, flags2);
    dprintf("version=%d channels=%d sample_rate=%d bitrate=%d block_align=%d\n",
           s->version, s->nb_channels, s->sample_rate, s->bit_rate, 
           s->block_align);
    dprintf("bps=%f bps1=%f high_freq=%f bitoffset=%d\n", 
           bps, bps1, high_freq, s->byte_offset_bits);
    dprintf("use_noise_coding=%d use_exp_vlc=%d nb_block_sizes=%d\n",
           s->use_noise_coding, s->use_exp_vlc, s->nb_block_sizes);

    /* compute the scale factor band sizes for each MDCT block size */
    {
        int a, b, pos, lpos, k, block_len, i, j, n;
        const uint8_t *table;

                if (end > start)
                    s->exponent_high_bands[k][j++] = end - start;
            }
            s->exponent_high_sizes[k] = j;
#if 0
            tprintf("%5d: coefs_end=%d high_band_start=%d nb_high_bands=%d: ",
                  s->frame_len >> k, 
                  s->coefs_end[k],
                  s->high_band_start[k],
                  s->exponent_high_sizes[k]);
            for(j=0;j<s->exponent_high_sizes[k];j++)
                tprintf(" %d", s->exponent_high_bands[k][j]);
            tprintf("\n");
#endif
        }
    }

#ifdef TRACE
    {
        int i, j;
        for(i = 0; i < s->nb_block_sizes; i++) {
            tprintf("%5d: n=%2d:", 
                   s->frame_len >> i, 
                   s->exponent_sizes[i]);
            for(j=0;j<s->exponent_sizes[i];j++)
                tprintf(" %d", s->exponent_bands[i][j]);
            tprintf("\n");
        }
    }
#endif

    /* init MDCT */

        if (s->use_exp_vlc)
            s->noise_mult = 0.02;
        else
            s->noise_mult = 0.04;
               
#ifdef TRACE
        for(i=0;i<NOISE_TAB_SIZE;i++)
            s->noise_table[i] = 1.0 * s->noise_mult;
#else
        {
            unsigned int seed;

    int coef_nb_bits, total_gain, parse_exponents;
    float window[BLOCK_MAX_SIZE * 2];
    int nb_coefs[MAX_CHANNELS];
    float mdct_norm;

    tprintf("***decode_block: %d:%d\n", frame_count - 1, s->block_num);

    /* compute current block length */
    if (s->use_variable_block_len) {
        n = av_log2(s->nb_block_sizes - 1) + 1;
    

                            v = exp_ptr[i];
                            e2 += v * v;
                        }
                        exp_power[j] = e2 / n;
                        last_high_band = j;
                        tprintf("%d: power=%f (%d)\n", j, exp_power[j], n);
                    }
                    exp_ptr += n;
                }

                /* main freqs and high freqs */

                    *coefs++ = 0.0;
            }
        }
    }

#ifdef TRACE
    for(ch = 0; ch < s->nb_channels; ch++) {
        if (s->channel_coded[ch]) {
            dump_floats("exponents", 3, s->exponents[ch], s->block_len);
            dump_floats("coefs", 1, s->coefs[ch], s->block_len);
        }

        /* nominal case for ms stereo: we do it before mdct */
        /* no need to optimize this case because it should almost
           never happen */
        if (!s->channel_coded[0]) {
            tprintf("rare ms-stereo case happened\n");
            memset(s->coefs[0], 0, sizeof(float) * s->block_len);
            s->channel_coded[0] = 1;
        }
        
        for(i = 0; i < s->block_len; i++) {

{
    int ret, i, n, a, ch, incr;
    int16_t *ptr;
    float *iptr;

    tprintf("***decode_frame: %d size=%d\n", frame_count++, s->frame_len);

    /* read each block */
    s->block_num = 0;
    s->block_pos = 0;
    for(;;) {

        /* XXX: suppress this */
        memset(&s->frame_out[ch][s->frame_len], 0, 
               s->frame_len * sizeof(float));
    }

#ifdef TRACE
    dump_shorts("samples", samples, n * s->nb_channels);
#endif
    return 0;
}

    WMADecodeContext *s = avctx->priv_data;
    int nb_frames, bit_offset, i, pos, len;
    uint8_t *q;
    int16_t *samples;
    
    tprintf("***decode_superframe:\n");

    samples = data;

    init_get_bits(&s->gb, buf, buf_size*8);