Blame liba52/parse.c

Packit 64f477
/*
Packit 64f477
 * parse.c
Packit 64f477
 * Copyright (C) 2000-2002 Michel Lespinasse <walken@zoy.org>
Packit 64f477
 * Copyright (C) 1999-2000 Aaron Holtzman <aholtzma@ess.engr.uvic.ca>
Packit 64f477
 *
Packit 64f477
 * This file is part of a52dec, a free ATSC A-52 stream decoder.
Packit 64f477
 * See http://liba52.sourceforge.net/ for updates.
Packit 64f477
 *
Packit 64f477
 * a52dec is free software; you can redistribute it and/or modify
Packit 64f477
 * it under the terms of the GNU General Public License as published by
Packit 64f477
 * the Free Software Foundation; either version 2 of the License, or
Packit 64f477
 * (at your option) any later version.
Packit 64f477
 *
Packit 64f477
 * a52dec is distributed in the hope that it will be useful,
Packit 64f477
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit 64f477
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit 64f477
 * GNU General Public License for more details.
Packit 64f477
 *
Packit 64f477
 * You should have received a copy of the GNU General Public License
Packit 64f477
 * along with this program; if not, write to the Free Software
Packit 64f477
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
Packit 64f477
 */
Packit 64f477
Packit 64f477
#include "config.h"
Packit 64f477
Packit 64f477
#include <stdlib.h>
Packit 64f477
#include <string.h>
Packit 64f477
#include <inttypes.h>
Packit 64f477
Packit 64f477
#include "a52.h"
Packit 64f477
#include "a52_internal.h"
Packit 64f477
#include "bitstream.h"
Packit 64f477
#include "tables.h"
Packit 64f477
Packit 64f477
#ifdef HAVE_MEMALIGN
Packit 64f477
/* some systems have memalign() but no declaration for it */
Packit 64f477
void * memalign (size_t align, size_t size);
Packit 64f477
#else
Packit 64f477
/* assume malloc alignment is sufficient */
Packit 64f477
#define memalign(align,size) malloc (size)
Packit 64f477
#endif
Packit 64f477
Packit 64f477
typedef struct {
Packit 64f477
    sample_t q1[2];
Packit 64f477
    sample_t q2[2];
Packit 64f477
    sample_t q4;
Packit 64f477
    int q1_ptr;
Packit 64f477
    int q2_ptr;
Packit 64f477
    int q4_ptr;
Packit 64f477
} quantizer_t;
Packit 64f477
Packit 64f477
static uint8_t halfrate[12] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3};
Packit 64f477
Packit 64f477
a52_state_t * a52_init (uint32_t mm_accel)
Packit 64f477
{
Packit 64f477
    a52_state_t * state;
Packit 64f477
    int i;
Packit 64f477
Packit 64f477
    state = malloc (sizeof (a52_state_t));
Packit 64f477
    if (state == NULL)
Packit 64f477
	return NULL;
Packit 64f477
Packit 64f477
    state->samples = memalign (16, 256 * 12 * sizeof (sample_t));
Packit 64f477
    if (state->samples == NULL) {
Packit 64f477
	free (state);
Packit 64f477
	return NULL;
Packit 64f477
    }
Packit 64f477
Packit 64f477
    for (i = 0; i < 256 * 12; i++)
Packit 64f477
	state->samples[i] = 0;
Packit 64f477
Packit 64f477
    state->downmixed = 1;
Packit 64f477
Packit 64f477
    state->lfsr_state = 1;
Packit 64f477
Packit 64f477
    a52_imdct_init (mm_accel);
Packit 64f477
Packit 64f477
    return state;
Packit 64f477
}
Packit 64f477
Packit 64f477
sample_t * a52_samples (a52_state_t * state)
Packit 64f477
{
Packit 64f477
    return state->samples;
Packit 64f477
}
Packit 64f477
Packit 64f477
int a52_syncinfo (uint8_t * buf, int * flags,
Packit 64f477
		  int * sample_rate, int * bit_rate)
Packit 64f477
{
Packit 64f477
    static int rate[] = { 32,  40,  48,  56,  64,  80,  96, 112,
Packit 64f477
			 128, 160, 192, 224, 256, 320, 384, 448,
Packit 64f477
			 512, 576, 640};
Packit 64f477
    static uint8_t lfeon[8] = {0x10, 0x10, 0x04, 0x04, 0x04, 0x01, 0x04, 0x01};
Packit 64f477
    int frmsizecod;
Packit 64f477
    int bitrate;
Packit 64f477
    int half;
Packit 64f477
    int acmod;
Packit 64f477
Packit 64f477
    if ((buf[0] != 0x0b) || (buf[1] != 0x77))	/* syncword */
Packit 64f477
	return 0;
Packit 64f477
Packit 64f477
    if (buf[5] >= 0x60)		/* bsid >= 12 */
Packit 64f477
	return 0;
Packit 64f477
    half = halfrate[buf[5] >> 3];
Packit 64f477
Packit 64f477
    /* acmod, dsurmod and lfeon */
Packit 64f477
    acmod = buf[6] >> 5;
Packit 64f477
    *flags = ((((buf[6] & 0xf8) == 0x50) ? A52_DOLBY : acmod) |
Packit 64f477
	      ((buf[6] & lfeon[acmod]) ? A52_LFE : 0));
Packit 64f477
Packit 64f477
    frmsizecod = buf[4] & 63;
Packit 64f477
    if (frmsizecod >= 38)
Packit 64f477
	return 0;
Packit 64f477
    bitrate = rate [frmsizecod >> 1];
Packit 64f477
    *bit_rate = (bitrate * 1000) >> half;
Packit 64f477
Packit 64f477
    switch (buf[4] & 0xc0) {
Packit 64f477
    case 0:
Packit 64f477
	*sample_rate = 48000 >> half;
Packit 64f477
	return 4 * bitrate;
Packit 64f477
    case 0x40:
Packit 64f477
	*sample_rate = 44100 >> half;
Packit 64f477
	return 2 * (320 * bitrate / 147 + (frmsizecod & 1));
Packit 64f477
    case 0x80:
Packit 64f477
	*sample_rate = 32000 >> half;
Packit 64f477
	return 6 * bitrate;
Packit 64f477
    default:
Packit 64f477
	return 0;
Packit 64f477
    }
Packit 64f477
}
Packit 64f477
Packit 64f477
int a52_frame (a52_state_t * state, uint8_t * buf, int * flags,
Packit 64f477
	       sample_t * level, sample_t bias)
Packit 64f477
{
Packit 64f477
    static sample_t clev[4] = {LEVEL_3DB, LEVEL_45DB, LEVEL_6DB, LEVEL_45DB};
Packit 64f477
    static sample_t slev[4] = {LEVEL_3DB, LEVEL_6DB, 0, LEVEL_6DB};
Packit 64f477
    int chaninfo;
Packit 64f477
    int acmod;
Packit 64f477
Packit 64f477
    state->fscod = buf[4] >> 6;
Packit 64f477
    state->halfrate = halfrate[buf[5] >> 3];
Packit 64f477
    state->acmod = acmod = buf[6] >> 5;
Packit 64f477
Packit 64f477
    a52_bitstream_set_ptr (state, buf + 6);
Packit 64f477
    bitstream_get (state, 3);	/* skip acmod we already parsed */
Packit 64f477
Packit 64f477
    if ((acmod == 2) && (bitstream_get (state, 2) == 2))	/* dsurmod */
Packit 64f477
	acmod = A52_DOLBY;
Packit 64f477
Packit 64f477
    if ((acmod & 1) && (acmod != 1))
Packit 64f477
	state->clev = clev[bitstream_get (state, 2)];	/* cmixlev */
Packit 64f477
Packit 64f477
    if (acmod & 4)
Packit 64f477
	state->slev = slev[bitstream_get (state, 2)];	/* surmixlev */
Packit 64f477
Packit 64f477
    state->lfeon = bitstream_get (state, 1);
Packit 64f477
Packit 64f477
    state->output = a52_downmix_init (acmod, *flags, level,
Packit 64f477
				      state->clev, state->slev);
Packit 64f477
    if (state->output < 0)
Packit 64f477
	return 1;
Packit 64f477
    if (state->lfeon && (*flags & A52_LFE))
Packit 64f477
	state->output |= A52_LFE;
Packit 64f477
    *flags = state->output;
Packit 64f477
    /* the 2* compensates for differences in imdct */
Packit 64f477
    state->dynrng = state->level = 2 * *level;
Packit 64f477
    state->bias = bias;
Packit 64f477
    state->dynrnge = 1;
Packit 64f477
    state->dynrngcall = NULL;
Packit 64f477
    state->cplba.deltbae = DELTA_BIT_NONE;
Packit 64f477
    state->ba[0].deltbae = state->ba[1].deltbae = state->ba[2].deltbae =
Packit 64f477
	state->ba[3].deltbae = state->ba[4].deltbae = DELTA_BIT_NONE;
Packit 64f477
Packit 64f477
    chaninfo = !acmod;
Packit 64f477
    do {
Packit 64f477
	bitstream_get (state, 5);	/* dialnorm */
Packit 64f477
	if (bitstream_get (state, 1))	/* compre */
Packit 64f477
	    bitstream_get (state, 8);	/* compr */
Packit 64f477
	if (bitstream_get (state, 1))	/* langcode */
Packit 64f477
	    bitstream_get (state, 8);	/* langcod */
Packit 64f477
	if (bitstream_get (state, 1))	/* audprodie */
Packit 64f477
	    bitstream_get (state, 7);	/* mixlevel + roomtyp */
Packit 64f477
    } while (chaninfo--);
Packit 64f477
Packit 64f477
    bitstream_get (state, 2);		/* copyrightb + origbs */
Packit 64f477
Packit 64f477
    if (bitstream_get (state, 1))	/* timecod1e */
Packit 64f477
	bitstream_get (state, 14);	/* timecod1 */
Packit 64f477
    if (bitstream_get (state, 1))	/* timecod2e */
Packit 64f477
	bitstream_get (state, 14);	/* timecod2 */
Packit 64f477
Packit 64f477
    if (bitstream_get (state, 1)) {	/* addbsie */
Packit 64f477
	int addbsil;
Packit 64f477
Packit 64f477
	addbsil = bitstream_get (state, 6);
Packit 64f477
	do {
Packit 64f477
	    bitstream_get (state, 8);	/* addbsi */
Packit 64f477
	} while (addbsil--);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    return 0;
Packit 64f477
}
Packit 64f477
Packit 64f477
void a52_dynrng (a52_state_t * state,
Packit 64f477
		 sample_t (* call) (sample_t, void *), void * data)
Packit 64f477
{
Packit 64f477
    state->dynrnge = 0;
Packit 64f477
    if (call) {
Packit 64f477
	state->dynrnge = 1;
Packit 64f477
	state->dynrngcall = call;
Packit 64f477
	state->dynrngdata = data;
Packit 64f477
    }
Packit 64f477
}
Packit 64f477
Packit 64f477
static int parse_exponents (a52_state_t * state, int expstr, int ngrps,
Packit 64f477
			    uint8_t exponent, uint8_t * dest)
Packit 64f477
{
Packit 64f477
    int exps;
Packit 64f477
Packit 64f477
    while (ngrps--) {
Packit 64f477
	exps = bitstream_get (state, 7);
Packit 64f477
Packit 64f477
	exponent += exp_1[exps];
Packit 64f477
	if (exponent > 24)
Packit 64f477
	    return 1;
Packit 64f477
Packit 64f477
	switch (expstr) {
Packit 64f477
	case EXP_D45:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	case EXP_D25:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	case EXP_D15:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	}
Packit 64f477
Packit 64f477
	exponent += exp_2[exps];
Packit 64f477
	if (exponent > 24)
Packit 64f477
	    return 1;
Packit 64f477
Packit 64f477
	switch (expstr) {
Packit 64f477
	case EXP_D45:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	case EXP_D25:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	case EXP_D15:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	}
Packit 64f477
Packit 64f477
	exponent += exp_3[exps];
Packit 64f477
	if (exponent > 24)
Packit 64f477
	    return 1;
Packit 64f477
Packit 64f477
	switch (expstr) {
Packit 64f477
	case EXP_D45:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	case EXP_D25:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	case EXP_D15:
Packit 64f477
	    *(dest++) = exponent;
Packit 64f477
	}
Packit 64f477
    }	
Packit 64f477
Packit 64f477
    return 0;
Packit 64f477
}
Packit 64f477
Packit 64f477
static int parse_deltba (a52_state_t * state, int8_t * deltba)
Packit 64f477
{
Packit 64f477
    int deltnseg, deltlen, delta, j;
Packit 64f477
Packit 64f477
    memset (deltba, 0, 50);
Packit 64f477
Packit 64f477
    deltnseg = bitstream_get (state, 3);
Packit 64f477
    j = 0;
Packit 64f477
    do {
Packit 64f477
	j += bitstream_get (state, 5);
Packit 64f477
	deltlen = bitstream_get (state, 4);
Packit 64f477
	delta = bitstream_get (state, 3);
Packit 64f477
	delta -= (delta >= 4) ? 3 : 4;
Packit 64f477
	if (!deltlen)
Packit 64f477
	    continue;
Packit 64f477
	if (j + deltlen >= 50)
Packit 64f477
	    return 1;
Packit 64f477
	while (deltlen--)
Packit 64f477
	    deltba[j++] = delta;
Packit 64f477
    } while (deltnseg--);
Packit 64f477
Packit 64f477
    return 0;
Packit 64f477
}
Packit 64f477
Packit 64f477
static inline int zero_snr_offsets (int nfchans, a52_state_t * state)
Packit 64f477
{
Packit 64f477
    int i;
Packit 64f477
Packit 64f477
    if ((state->csnroffst) ||
Packit 64f477
	(state->chincpl && state->cplba.bai >> 3) ||	/* cplinu, fsnroffst */
Packit 64f477
	(state->lfeon && state->lfeba.bai >> 3))	/* fsnroffst */
Packit 64f477
	return 0;
Packit 64f477
    for (i = 0; i < nfchans; i++)
Packit 64f477
	if (state->ba[i].bai >> 3)			/* fsnroffst */
Packit 64f477
	    return 0;
Packit 64f477
    return 1;
Packit 64f477
}
Packit 64f477
Packit 64f477
static inline int16_t dither_gen (a52_state_t * state)
Packit 64f477
{
Packit 64f477
    int16_t nstate;
Packit 64f477
Packit 64f477
    nstate = dither_lut[state->lfsr_state >> 8] ^ (state->lfsr_state << 8);
Packit 64f477
	
Packit 64f477
    state->lfsr_state = (uint16_t) nstate;
Packit 64f477
Packit 64f477
    return nstate;
Packit 64f477
}
Packit 64f477
Packit 64f477
static void coeff_get (a52_state_t * state, sample_t * coeff,
Packit 64f477
		       expbap_t * expbap, quantizer_t * quantizer,
Packit 64f477
		       sample_t level, int dither, int end)
Packit 64f477
{
Packit 64f477
    int i;
Packit 64f477
    uint8_t * exp;
Packit 64f477
    int8_t * bap;
Packit 64f477
    sample_t factor[25];
Packit 64f477
Packit 64f477
    for (i = 0; i <= 24; i++)
Packit 64f477
	factor[i] = scale_factor[i] * level;
Packit 64f477
Packit 64f477
    exp = expbap->exp;
Packit 64f477
    bap = expbap->bap;
Packit 64f477
Packit 64f477
    for (i = 0; i < end; i++) {
Packit 64f477
	int bapi;
Packit 64f477
Packit 64f477
	bapi = bap[i];
Packit 64f477
	switch (bapi) {
Packit 64f477
	case 0:
Packit 64f477
	    if (dither) {
Packit 64f477
		coeff[i] = dither_gen (state) * LEVEL_3DB * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    } else {
Packit 64f477
		coeff[i] = 0;
Packit 64f477
		continue;
Packit 64f477
	    }
Packit 64f477
Packit 64f477
	case -1:
Packit 64f477
	    if (quantizer->q1_ptr >= 0) {
Packit 64f477
		coeff[i] = quantizer->q1[quantizer->q1_ptr--] * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    } else {
Packit 64f477
		int code;
Packit 64f477
Packit 64f477
		code = bitstream_get (state, 5);
Packit 64f477
Packit 64f477
		quantizer->q1_ptr = 1;
Packit 64f477
		quantizer->q1[0] = q_1_2[code];
Packit 64f477
		quantizer->q1[1] = q_1_1[code];
Packit 64f477
		coeff[i] = q_1_0[code] * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    }
Packit 64f477
Packit 64f477
	case -2:
Packit 64f477
	    if (quantizer->q2_ptr >= 0) {
Packit 64f477
		coeff[i] = quantizer->q2[quantizer->q2_ptr--] * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    } else {
Packit 64f477
		int code;
Packit 64f477
Packit 64f477
		code = bitstream_get (state, 7);
Packit 64f477
Packit 64f477
		quantizer->q2_ptr = 1;
Packit 64f477
		quantizer->q2[0] = q_2_2[code];
Packit 64f477
		quantizer->q2[1] = q_2_1[code];
Packit 64f477
		coeff[i] = q_2_0[code] * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    }
Packit 64f477
Packit 64f477
	case 3:
Packit 64f477
	    coeff[i] = q_3[bitstream_get (state, 3)] * factor[exp[i]];
Packit 64f477
	    continue;
Packit 64f477
Packit 64f477
	case -3:
Packit 64f477
	    if (quantizer->q4_ptr == 0) {
Packit 64f477
		quantizer->q4_ptr = -1;
Packit 64f477
		coeff[i] = quantizer->q4 * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    } else {
Packit 64f477
		int code;
Packit 64f477
Packit 64f477
		code = bitstream_get (state, 7);
Packit 64f477
Packit 64f477
		quantizer->q4_ptr = 0;
Packit 64f477
		quantizer->q4 = q_4_1[code];
Packit 64f477
		coeff[i] = q_4_0[code] * factor[exp[i]];
Packit 64f477
		continue;
Packit 64f477
	    }
Packit 64f477
Packit 64f477
	case 4:
Packit 64f477
	    coeff[i] = q_5[bitstream_get (state, 4)] * factor[exp[i]];
Packit 64f477
	    continue;
Packit 64f477
Packit 64f477
	default:
Packit 64f477
	    coeff[i] = ((bitstream_get_2 (state, bapi) << (16 - bapi)) *
Packit 64f477
			  factor[exp[i]]);
Packit 64f477
	}
Packit 64f477
    }
Packit 64f477
}
Packit 64f477
Packit 64f477
static void coeff_get_coupling (a52_state_t * state, int nfchans,
Packit 64f477
				sample_t * coeff, sample_t (* samples)[256],
Packit 64f477
				quantizer_t * quantizer, uint8_t dithflag[5])
Packit 64f477
{
Packit 64f477
    int cplbndstrc, bnd, i, i_end, ch;
Packit 64f477
    uint8_t * exp;
Packit 64f477
    int8_t * bap;
Packit 64f477
    sample_t cplco[5];
Packit 64f477
Packit 64f477
    exp = state->cpl_expbap.exp;
Packit 64f477
    bap = state->cpl_expbap.bap;
Packit 64f477
    bnd = 0;
Packit 64f477
    cplbndstrc = state->cplbndstrc;
Packit 64f477
    i = state->cplstrtmant;
Packit 64f477
    while (i < state->cplendmant) {
Packit 64f477
	i_end = i + 12;
Packit 64f477
	while (cplbndstrc & 1) {
Packit 64f477
	    cplbndstrc >>= 1;
Packit 64f477
	    i_end += 12;
Packit 64f477
	}
Packit 64f477
	cplbndstrc >>= 1;
Packit 64f477
	for (ch = 0; ch < nfchans; ch++)
Packit 64f477
	    cplco[ch] = state->cplco[ch][bnd] * coeff[ch];
Packit 64f477
	bnd++;
Packit 64f477
Packit 64f477
	while (i < i_end) {
Packit 64f477
	    sample_t cplcoeff;
Packit 64f477
	    int bapi;
Packit 64f477
Packit 64f477
	    bapi = bap[i];
Packit 64f477
	    switch (bapi) {
Packit 64f477
	    case 0:
Packit 64f477
		cplcoeff = LEVEL_3DB * scale_factor[exp[i]];
Packit 64f477
		for (ch = 0; ch < nfchans; ch++)
Packit 64f477
		    if ((state->chincpl >> ch) & 1) {
Packit 64f477
			if (dithflag[ch])
Packit 64f477
			    samples[ch][i] = (cplcoeff * cplco[ch] *
Packit 64f477
					      dither_gen (state));
Packit 64f477
			else
Packit 64f477
			    samples[ch][i] = 0;
Packit 64f477
		    }
Packit 64f477
		i++;
Packit 64f477
		continue;
Packit 64f477
Packit 64f477
	    case -1:
Packit 64f477
		if (quantizer->q1_ptr >= 0) {
Packit 64f477
		    cplcoeff = quantizer->q1[quantizer->q1_ptr--];
Packit 64f477
		    break;
Packit 64f477
		} else {
Packit 64f477
		    int code;
Packit 64f477
Packit 64f477
		    code = bitstream_get (state, 5);
Packit 64f477
Packit 64f477
		    quantizer->q1_ptr = 1;
Packit 64f477
		    quantizer->q1[0] = q_1_2[code];
Packit 64f477
		    quantizer->q1[1] = q_1_1[code];
Packit 64f477
		    cplcoeff = q_1_0[code];
Packit 64f477
		    break;
Packit 64f477
		}
Packit 64f477
Packit 64f477
	    case -2:
Packit 64f477
		if (quantizer->q2_ptr >= 0) {
Packit 64f477
		    cplcoeff = quantizer->q2[quantizer->q2_ptr--];
Packit 64f477
		    break;
Packit 64f477
		} else {
Packit 64f477
		    int code;
Packit 64f477
Packit 64f477
		    code = bitstream_get (state, 7);
Packit 64f477
Packit 64f477
		    quantizer->q2_ptr = 1;
Packit 64f477
		    quantizer->q2[0] = q_2_2[code];
Packit 64f477
		    quantizer->q2[1] = q_2_1[code];
Packit 64f477
		    cplcoeff = q_2_0[code];
Packit 64f477
		    break;
Packit 64f477
		}
Packit 64f477
Packit 64f477
	    case 3:
Packit 64f477
		cplcoeff = q_3[bitstream_get (state, 3)];
Packit 64f477
		break;
Packit 64f477
Packit 64f477
	    case -3:
Packit 64f477
		if (quantizer->q4_ptr == 0) {
Packit 64f477
		    quantizer->q4_ptr = -1;
Packit 64f477
		    cplcoeff = quantizer->q4;
Packit 64f477
		    break;
Packit 64f477
		} else {
Packit 64f477
		    int code;
Packit 64f477
Packit 64f477
		    code = bitstream_get (state, 7);
Packit 64f477
Packit 64f477
		    quantizer->q4_ptr = 0;
Packit 64f477
		    quantizer->q4 = q_4_1[code];
Packit 64f477
		    cplcoeff = q_4_0[code];
Packit 64f477
		    break;
Packit 64f477
		}
Packit 64f477
Packit 64f477
	    case 4:
Packit 64f477
		cplcoeff = q_5[bitstream_get (state, 4)];
Packit 64f477
		break;
Packit 64f477
Packit 64f477
	    default:
Packit 64f477
		cplcoeff = bitstream_get_2 (state, bapi) << (16 - bapi);
Packit 64f477
	    }
Packit 64f477
Packit 64f477
	    cplcoeff *= scale_factor[exp[i]];
Packit 64f477
	    for (ch = 0; ch < nfchans; ch++)
Packit 64f477
		if ((state->chincpl >> ch) & 1)
Packit 64f477
		    samples[ch][i] = cplcoeff * cplco[ch];
Packit 64f477
	    i++;
Packit 64f477
	}
Packit 64f477
    }
Packit 64f477
}
Packit 64f477
Packit 64f477
int a52_block (a52_state_t * state)
Packit 64f477
{
Packit 64f477
    static const uint8_t nfchans_tbl[] = {2, 1, 2, 3, 3, 4, 4, 5, 1, 1, 2};
Packit 64f477
    static int rematrix_band[4] = {25, 37, 61, 253};
Packit 64f477
    int i, nfchans, chaninfo;
Packit 64f477
    uint8_t cplexpstr, chexpstr[5], lfeexpstr, do_bit_alloc, done_cpl;
Packit 64f477
    uint8_t blksw[5], dithflag[5];
Packit 64f477
    sample_t coeff[5];
Packit 64f477
    int chanbias;
Packit 64f477
    quantizer_t quantizer;
Packit 64f477
    sample_t * samples;
Packit 64f477
Packit 64f477
    nfchans = nfchans_tbl[state->acmod];
Packit 64f477
Packit 64f477
    for (i = 0; i < nfchans; i++)
Packit 64f477
	blksw[i] = bitstream_get (state, 1);
Packit 64f477
Packit 64f477
    for (i = 0; i < nfchans; i++)
Packit 64f477
	dithflag[i] = bitstream_get (state, 1);
Packit 64f477
Packit 64f477
    chaninfo = !state->acmod;
Packit 64f477
    do {
Packit 64f477
	if (bitstream_get (state, 1)) {	/* dynrnge */
Packit 64f477
	    int dynrng;
Packit 64f477
Packit 64f477
	    dynrng = bitstream_get_2 (state, 8);
Packit 64f477
	    if (state->dynrnge) {
Packit 64f477
		sample_t range;
Packit 64f477
Packit 64f477
		range = ((((dynrng & 0x1f) | 0x20) << 13) *
Packit 64f477
			 scale_factor[3 - (dynrng >> 5)]);
Packit 64f477
		if (state->dynrngcall)
Packit 64f477
		    range = state->dynrngcall (range, state->dynrngdata);
Packit 64f477
		state->dynrng = state->level * range;
Packit 64f477
	    }
Packit 64f477
	}
Packit 64f477
    } while (chaninfo--);
Packit 64f477
Packit 64f477
    if (bitstream_get (state, 1)) {	/* cplstre */
Packit 64f477
	state->chincpl = 0;
Packit 64f477
	if (bitstream_get (state, 1)) {	/* cplinu */
Packit 64f477
	    static uint8_t bndtab[16] = {31, 35, 37, 39, 41, 42, 43, 44,
Packit 64f477
					 45, 45, 46, 46, 47, 47, 48, 48};
Packit 64f477
	    int cplbegf;
Packit 64f477
	    int cplendf;
Packit 64f477
	    int ncplsubnd;
Packit 64f477
Packit 64f477
	    for (i = 0; i < nfchans; i++)
Packit 64f477
		state->chincpl |= bitstream_get (state, 1) << i;
Packit 64f477
	    switch (state->acmod) {
Packit 64f477
	    case 0: case 1:
Packit 64f477
		return 1;
Packit 64f477
	    case 2:
Packit 64f477
		state->phsflginu = bitstream_get (state, 1);
Packit 64f477
	    }
Packit 64f477
	    cplbegf = bitstream_get (state, 4);
Packit 64f477
	    cplendf = bitstream_get (state, 4);
Packit 64f477
Packit 64f477
	    if (cplendf + 3 - cplbegf < 0)
Packit 64f477
		return 1;
Packit 64f477
	    state->ncplbnd = ncplsubnd = cplendf + 3 - cplbegf;
Packit 64f477
	    state->cplstrtbnd = bndtab[cplbegf];
Packit 64f477
	    state->cplstrtmant = cplbegf * 12 + 37;
Packit 64f477
	    state->cplendmant = cplendf * 12 + 73;
Packit 64f477
Packit 64f477
	    state->cplbndstrc = 0;
Packit 64f477
	    for (i = 0; i < ncplsubnd - 1; i++)
Packit 64f477
		if (bitstream_get (state, 1)) {
Packit 64f477
		    state->cplbndstrc |= 1 << i;
Packit 64f477
		    state->ncplbnd--;
Packit 64f477
		}
Packit 64f477
	}
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (state->chincpl) {	/* cplinu */
Packit 64f477
	int j, cplcoe;
Packit 64f477
Packit 64f477
	cplcoe = 0;
Packit 64f477
	for (i = 0; i < nfchans; i++)
Packit 64f477
	    if ((state->chincpl) >> i & 1)
Packit 64f477
		if (bitstream_get (state, 1)) {	/* cplcoe */
Packit 64f477
		    int mstrcplco, cplcoexp, cplcomant;
Packit 64f477
Packit 64f477
		    cplcoe = 1;
Packit 64f477
		    mstrcplco = 3 * bitstream_get (state, 2);
Packit 64f477
		    for (j = 0; j < state->ncplbnd; j++) {
Packit 64f477
			cplcoexp = bitstream_get (state, 4);
Packit 64f477
			cplcomant = bitstream_get (state, 4);
Packit 64f477
			if (cplcoexp == 15)
Packit 64f477
			    cplcomant <<= 14;
Packit 64f477
			else
Packit 64f477
			    cplcomant = (cplcomant | 0x10) << 13;
Packit 64f477
			state->cplco[i][j] =
Packit 64f477
			    cplcomant * scale_factor[cplcoexp + mstrcplco];
Packit 64f477
		    }
Packit 64f477
		}
Packit 64f477
	if ((state->acmod == 2) && state->phsflginu && cplcoe)
Packit 64f477
	    for (j = 0; j < state->ncplbnd; j++)
Packit 64f477
		if (bitstream_get (state, 1))	/* phsflg */
Packit 64f477
		    state->cplco[1][j] = -state->cplco[1][j];
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if ((state->acmod == 2) && (bitstream_get (state, 1))) {	/* rematstr */
Packit 64f477
	int end;
Packit 64f477
Packit 64f477
	state->rematflg = 0;
Packit 64f477
	end = (state->chincpl) ? state->cplstrtmant : 253;	/* cplinu */
Packit 64f477
	i = 0;
Packit 64f477
	do
Packit 64f477
	    state->rematflg |= bitstream_get (state, 1) << i;
Packit 64f477
	while (rematrix_band[i++] < end);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    cplexpstr = EXP_REUSE;
Packit 64f477
    lfeexpstr = EXP_REUSE;
Packit 64f477
    if (state->chincpl)	/* cplinu */
Packit 64f477
	cplexpstr = bitstream_get (state, 2);
Packit 64f477
    for (i = 0; i < nfchans; i++)
Packit 64f477
	chexpstr[i] = bitstream_get (state, 2);
Packit 64f477
    if (state->lfeon) 
Packit 64f477
	lfeexpstr = bitstream_get (state, 1);
Packit 64f477
Packit 64f477
    for (i = 0; i < nfchans; i++)
Packit 64f477
	if (chexpstr[i] != EXP_REUSE) {
Packit 64f477
	    if ((state->chincpl >> i) & 1)
Packit 64f477
		state->endmant[i] = state->cplstrtmant;
Packit 64f477
	    else {
Packit 64f477
		int chbwcod;
Packit 64f477
Packit 64f477
		chbwcod = bitstream_get (state, 6);
Packit 64f477
		if (chbwcod > 60)
Packit 64f477
		    return 1;
Packit 64f477
		state->endmant[i] = chbwcod * 3 + 73;
Packit 64f477
	    }
Packit 64f477
	}
Packit 64f477
Packit 64f477
    do_bit_alloc = 0;
Packit 64f477
Packit 64f477
    if (cplexpstr != EXP_REUSE) {
Packit 64f477
	int cplabsexp, ncplgrps;
Packit 64f477
Packit 64f477
	do_bit_alloc = 64;
Packit 64f477
	ncplgrps = ((state->cplendmant - state->cplstrtmant) /
Packit 64f477
		    (3 << (cplexpstr - 1)));
Packit 64f477
	cplabsexp = bitstream_get (state, 4) << 1;
Packit 64f477
	if (parse_exponents (state, cplexpstr, ncplgrps, cplabsexp,
Packit 64f477
			     state->cpl_expbap.exp + state->cplstrtmant))
Packit 64f477
	    return 1;
Packit 64f477
    }
Packit 64f477
    for (i = 0; i < nfchans; i++)
Packit 64f477
	if (chexpstr[i] != EXP_REUSE) {
Packit 64f477
	    int grp_size, nchgrps;
Packit 64f477
Packit 64f477
	    do_bit_alloc |= 1 << i;
Packit 64f477
	    grp_size = 3 << (chexpstr[i] - 1);
Packit 64f477
	    nchgrps = (state->endmant[i] + grp_size - 4) / grp_size;
Packit 64f477
	    state->fbw_expbap[i].exp[0] = bitstream_get (state, 4);
Packit 64f477
	    if (parse_exponents (state, chexpstr[i], nchgrps,
Packit 64f477
				 state->fbw_expbap[i].exp[0],
Packit 64f477
				 state->fbw_expbap[i].exp + 1))
Packit 64f477
		return 1;
Packit 64f477
	    bitstream_get (state, 2);	/* gainrng */
Packit 64f477
	}
Packit 64f477
    if (lfeexpstr != EXP_REUSE) {
Packit 64f477
	do_bit_alloc |= 32;
Packit 64f477
	state->lfe_expbap.exp[0] = bitstream_get (state, 4);
Packit 64f477
	if (parse_exponents (state, lfeexpstr, 2, state->lfe_expbap.exp[0],
Packit 64f477
			     state->lfe_expbap.exp + 1))
Packit 64f477
	    return 1;
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (bitstream_get (state, 1)) {	/* baie */
Packit 64f477
	do_bit_alloc = -1;
Packit 64f477
	state->bai = bitstream_get (state, 11);
Packit 64f477
    }
Packit 64f477
    if (bitstream_get (state, 1)) {	/* snroffste */
Packit 64f477
	do_bit_alloc = -1;
Packit 64f477
	state->csnroffst = bitstream_get (state, 6);
Packit 64f477
	if (state->chincpl)	/* cplinu */
Packit 64f477
	    state->cplba.bai = bitstream_get (state, 7);
Packit 64f477
	for (i = 0; i < nfchans; i++)
Packit 64f477
	    state->ba[i].bai = bitstream_get (state, 7);
Packit 64f477
	if (state->lfeon)
Packit 64f477
	    state->lfeba.bai = bitstream_get (state, 7);
Packit 64f477
    }
Packit 64f477
    if ((state->chincpl) && (bitstream_get (state, 1))) { /* cplleake */
Packit 64f477
	do_bit_alloc |= 64;
Packit 64f477
	state->cplfleak = 9 - bitstream_get (state, 3);
Packit 64f477
	state->cplsleak = 9 - bitstream_get (state, 3);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (bitstream_get (state, 1)) {	/* deltbaie */
Packit 64f477
	do_bit_alloc = -1;
Packit 64f477
	if (state->chincpl)	/* cplinu */
Packit 64f477
	    state->cplba.deltbae = bitstream_get (state, 2);
Packit 64f477
	for (i = 0; i < nfchans; i++)
Packit 64f477
	    state->ba[i].deltbae = bitstream_get (state, 2);
Packit 64f477
	if (state->chincpl &&	/* cplinu */
Packit 64f477
	    (state->cplba.deltbae == DELTA_BIT_NEW) &&
Packit 64f477
	    parse_deltba (state, state->cplba.deltba))
Packit 64f477
	    return 1;
Packit 64f477
	for (i = 0; i < nfchans; i++)
Packit 64f477
	    if ((state->ba[i].deltbae == DELTA_BIT_NEW) &&
Packit 64f477
		parse_deltba (state, state->ba[i].deltba))
Packit 64f477
		return 1;
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (do_bit_alloc) {
Packit 64f477
	if (zero_snr_offsets (nfchans, state)) {
Packit 64f477
	    memset (state->cpl_expbap.bap, 0, sizeof (state->cpl_expbap.bap));
Packit 64f477
	    for (i = 0; i < nfchans; i++)
Packit 64f477
		memset (state->fbw_expbap[i].bap, 0,
Packit 64f477
			sizeof (state->fbw_expbap[i].bap));
Packit 64f477
	    memset (state->lfe_expbap.bap, 0, sizeof (state->lfe_expbap.bap));
Packit 64f477
	} else {
Packit 64f477
	    if (state->chincpl && (do_bit_alloc & 64))	/* cplinu */
Packit 64f477
		a52_bit_allocate (state, &state->cplba, state->cplstrtbnd,
Packit 64f477
				  state->cplstrtmant, state->cplendmant,
Packit 64f477
				  state->cplfleak << 8, state->cplsleak << 8,
Packit 64f477
				  &state->cpl_expbap);
Packit 64f477
	    for (i = 0; i < nfchans; i++)
Packit 64f477
		if (do_bit_alloc & (1 << i))
Packit 64f477
		    a52_bit_allocate (state, state->ba + i, 0, 0,
Packit 64f477
				      state->endmant[i], 0, 0,
Packit 64f477
				      state->fbw_expbap +i);
Packit 64f477
	    if (state->lfeon && (do_bit_alloc & 32)) {
Packit 64f477
		state->lfeba.deltbae = DELTA_BIT_NONE;
Packit 64f477
		a52_bit_allocate (state, &state->lfeba, 0, 0, 7, 0, 0,
Packit 64f477
				  &state->lfe_expbap);
Packit 64f477
	    }
Packit 64f477
	}
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (bitstream_get (state, 1)) {	/* skiple */
Packit 64f477
	i = bitstream_get (state, 9);	/* skipl */
Packit 64f477
	while (i--)
Packit 64f477
	    bitstream_get (state, 8);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    samples = state->samples;
Packit 64f477
    if (state->output & A52_LFE)
Packit 64f477
	samples += 256;	/* shift for LFE channel */
Packit 64f477
Packit 64f477
    chanbias = a52_downmix_coeff (coeff, state->acmod, state->output,
Packit 64f477
				  state->dynrng, state->clev, state->slev);
Packit 64f477
Packit 64f477
    quantizer.q1_ptr = quantizer.q2_ptr = quantizer.q4_ptr = -1;
Packit 64f477
    done_cpl = 0;
Packit 64f477
Packit 64f477
    for (i = 0; i < nfchans; i++) {
Packit 64f477
	int j;
Packit 64f477
Packit 64f477
	coeff_get (state, samples + 256 * i, state->fbw_expbap +i, &quantizer,
Packit 64f477
		   coeff[i], dithflag[i], state->endmant[i]);
Packit 64f477
Packit 64f477
	if ((state->chincpl >> i) & 1) {
Packit 64f477
	    if (!done_cpl) {
Packit 64f477
		done_cpl = 1;
Packit 64f477
		coeff_get_coupling (state, nfchans, coeff,
Packit 64f477
				    (sample_t (*)[256])samples, &quantizer,
Packit 64f477
				    dithflag);
Packit 64f477
	    }
Packit 64f477
	    j = state->cplendmant;
Packit 64f477
	} else
Packit 64f477
	    j = state->endmant[i];
Packit 64f477
	do
Packit 64f477
	    (samples + 256 * i)[j] = 0;
Packit 64f477
	while (++j < 256);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (state->acmod == 2) {
Packit 64f477
	int j, end, band, rematflg;
Packit 64f477
Packit 64f477
	end = ((state->endmant[0] < state->endmant[1]) ?
Packit 64f477
	       state->endmant[0] : state->endmant[1]);
Packit 64f477
Packit 64f477
	i = 0;
Packit 64f477
	j = 13;
Packit 64f477
	rematflg = state->rematflg;
Packit 64f477
	do {
Packit 64f477
	    if (! (rematflg & 1)) {
Packit 64f477
		rematflg >>= 1;
Packit 64f477
		j = rematrix_band[i++];
Packit 64f477
		continue;
Packit 64f477
	    }
Packit 64f477
	    rematflg >>= 1;
Packit 64f477
	    band = rematrix_band[i++];
Packit 64f477
	    if (band > end)
Packit 64f477
		band = end;
Packit 64f477
	    do {
Packit 64f477
		sample_t tmp0, tmp1;
Packit 64f477
Packit 64f477
		tmp0 = samples[j];
Packit 64f477
		tmp1 = (samples+256)[j];
Packit 64f477
		samples[j] = tmp0 + tmp1;
Packit 64f477
		(samples+256)[j] = tmp0 - tmp1;
Packit 64f477
	    } while (++j < band);
Packit 64f477
	} while (j < end);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    if (state->lfeon) {
Packit 64f477
	if (state->output & A52_LFE) {
Packit 64f477
	    coeff_get (state, samples - 256, &state->lfe_expbap, &quantizer,
Packit 64f477
		       state->dynrng, 0, 7);
Packit 64f477
	    for (i = 7; i < 256; i++)
Packit 64f477
		(samples-256)[i] = 0;
Packit 64f477
	    a52_imdct_512 (samples - 256, samples + 1536 - 256, state->bias);
Packit 64f477
	} else {
Packit 64f477
	    /* just skip the LFE coefficients */
Packit 64f477
	    coeff_get (state, samples + 1280, &state->lfe_expbap, &quantizer,
Packit 64f477
		       0, 0, 7);
Packit 64f477
	}
Packit 64f477
    }
Packit 64f477
Packit 64f477
    i = 0;
Packit 64f477
    if (nfchans_tbl[state->output & A52_CHANNEL_MASK] < nfchans)
Packit 64f477
	for (i = 1; i < nfchans; i++)
Packit 64f477
	    if (blksw[i] != blksw[0])
Packit 64f477
		break;
Packit 64f477
Packit 64f477
    if (i < nfchans) {
Packit 64f477
	if (state->downmixed) {
Packit 64f477
	    state->downmixed = 0;
Packit 64f477
	    a52_upmix (samples + 1536, state->acmod, state->output);
Packit 64f477
	}
Packit 64f477
Packit 64f477
	for (i = 0; i < nfchans; i++) {
Packit 64f477
	    sample_t bias;
Packit 64f477
Packit 64f477
	    bias = 0;
Packit 64f477
	    if (!(chanbias & (1 << i)))
Packit 64f477
		bias = state->bias;
Packit 64f477
Packit 64f477
	    if (coeff[i]) {
Packit 64f477
		if (blksw[i])
Packit 64f477
		    a52_imdct_256 (samples + 256 * i, samples + 1536 + 256 * i,
Packit 64f477
				   bias);
Packit 64f477
		else 
Packit 64f477
		    a52_imdct_512 (samples + 256 * i, samples + 1536 + 256 * i,
Packit 64f477
				   bias);
Packit 64f477
	    } else {
Packit 64f477
		int j;
Packit 64f477
Packit 64f477
		for (j = 0; j < 256; j++)
Packit 64f477
		    (samples + 256 * i)[j] = bias;
Packit 64f477
	    }
Packit 64f477
	}
Packit 64f477
Packit 64f477
	a52_downmix (samples, state->acmod, state->output, state->bias,
Packit 64f477
		     state->clev, state->slev);
Packit 64f477
    } else {
Packit 64f477
	nfchans = nfchans_tbl[state->output & A52_CHANNEL_MASK];
Packit 64f477
Packit 64f477
	a52_downmix (samples, state->acmod, state->output, 0,
Packit 64f477
		     state->clev, state->slev);
Packit 64f477
Packit 64f477
	if (!state->downmixed) {
Packit 64f477
	    state->downmixed = 1;
Packit 64f477
	    a52_downmix (samples + 1536, state->acmod, state->output, 0,
Packit 64f477
			 state->clev, state->slev);
Packit 64f477
	}
Packit 64f477
Packit 64f477
	if (blksw[0])
Packit 64f477
	    for (i = 0; i < nfchans; i++)
Packit 64f477
		a52_imdct_256 (samples + 256 * i, samples + 1536 + 256 * i,
Packit 64f477
			       state->bias);
Packit 64f477
	else 
Packit 64f477
	    for (i = 0; i < nfchans; i++)
Packit 64f477
		a52_imdct_512 (samples + 256 * i, samples + 1536 + 256 * i,
Packit 64f477
			       state->bias);
Packit 64f477
    }
Packit 64f477
Packit 64f477
    return 0;
Packit 64f477
}
Packit 64f477
Packit 64f477
void a52_free (a52_state_t * state)
Packit 64f477
{
Packit 64f477
    free (state->samples);
Packit 64f477
    free (state);
Packit 64f477
}