diff --git a/libavcodec/vorbis.c b/libavcodec/vorbis.c index 01f1301e51..29206cc881 100644 --- a/libavcodec/vorbis.c +++ b/libavcodec/vorbis.c @@ -56,6 +56,9 @@ static unsigned int get_bits_long_le(GetBitContext *s, int n){ #define ilog(i) av_log2(2*(i)) +#define BARK(x) \ + (13.1f*atan(0.00074f*(x))+2.24f*atan(1.85e-8f*(x)*(x))+1e-4f*(x)) + static unsigned int nth_root(unsigned int x, unsigned int n) { // x^(1/n) unsigned int ret=0, i, j; @@ -166,10 +169,17 @@ static void vorbis_free(vorbis_context *vc) { av_freep(&vc->codebooks); for(i=0;ifloor_count;++i) { - av_free(vc->floors[i].x_list); - av_free(vc->floors[i].x_list_order); - av_free(vc->floors[i].low_neighbour); - av_free(vc->floors[i].high_neighbour); + if(vc->floors[i].floor_type==0) { + av_free(vc->floors[i].data.t0.map); + av_free(vc->floors[i].data.t0.book_list); + av_free(vc->floors[i].data.t0.lsp); + } + else { + av_free(vc->floors[i].data.t1.x_list); + av_free(vc->floors[i].data.t1.x_list_order); + av_free(vc->floors[i].data.t1.low_neighbour); + av_free(vc->floors[i].data.t1.high_neighbour); + } } av_freep(&vc->floors); @@ -404,8 +414,12 @@ static int vorbis_parse_setup_hdr_tdtransforms(vorbis_context *vc) { return 0; } -// Process floors part - only floor type 1 is supported +// Process floors part +static uint_fast8_t vorbis_floor0_decode(vorbis_context *vc, + vorbis_floor_data *vfu, float *vec); +static uint_fast8_t vorbis_floor1_decode(vorbis_context *vc, + vorbis_floor_data *vfu, float *vec); static int vorbis_parse_setup_hdr_floors(vorbis_context *vc) { GetBitContext *gb=&vc->gb; uint_fast16_t i,j,k; @@ -426,102 +440,168 @@ static int vorbis_parse_setup_hdr_floors(vorbis_context *vc) { uint_fast8_t rangebits; uint_fast16_t floor1_values=2; - floor_setup->partitions=get_bits(gb, 5); + floor_setup->decode=vorbis_floor1_decode; - AV_DEBUG(" %d.floor: %d partitions \n", i, floor_setup->partitions); + floor_setup->data.t1.partitions=get_bits(gb, 5); - for(j=0;jpartitions;++j) { - floor_setup->partition_class[j]=get_bits(gb, 4); - if (floor_setup->partition_class[j]>maximum_class) maximum_class=floor_setup->partition_class[j]; + AV_DEBUG(" %d.floor: %d partitions \n", i, floor_setup->data.t1.partitions); - AV_DEBUG(" %d. floor %d partition class %d \n", i, j, floor_setup->partition_class[j]); + for(j=0;jdata.t1.partitions;++j) { + floor_setup->data.t1.partition_class[j]=get_bits(gb, 4); + if (floor_setup->data.t1.partition_class[j]>maximum_class) maximum_class=floor_setup->data.t1.partition_class[j]; + + AV_DEBUG(" %d. floor %d partition class %d \n", i, j, floor_setup->data.t1.partition_class[j]); } AV_DEBUG(" maximum class %d \n", maximum_class); - floor_setup->maximum_class=maximum_class; + floor_setup->data.t1.maximum_class=maximum_class; for(j=0;j<=maximum_class;++j) { - floor_setup->class_dimensions[j]=get_bits(gb, 3)+1; - floor_setup->class_subclasses[j]=get_bits(gb, 2); + floor_setup->data.t1.class_dimensions[j]=get_bits(gb, 3)+1; + floor_setup->data.t1.class_subclasses[j]=get_bits(gb, 2); - AV_DEBUG(" %d floor %d class dim: %d subclasses %d \n", i, j, floor_setup->class_dimensions[j], floor_setup->class_subclasses[j]); + AV_DEBUG(" %d floor %d class dim: %d subclasses %d \n", i, j, floor_setup->data.t1.class_dimensions[j], floor_setup->data.t1.class_subclasses[j]); - if (floor_setup->class_subclasses[j]) { - floor_setup->class_masterbook[j]=get_bits(gb, 8); + if (floor_setup->data.t1.class_subclasses[j]) { + floor_setup->data.t1.class_masterbook[j]=get_bits(gb, 8); - AV_DEBUG(" masterbook: %d \n", floor_setup->class_masterbook[j]); + AV_DEBUG(" masterbook: %d \n", floor_setup->data.t1.class_masterbook[j]); } - for(k=0;k<(1<class_subclasses[j]);++k) { - floor_setup->subclass_books[j][k]=get_bits(gb, 8)-1; + for(k=0;k<(1<data.t1.class_subclasses[j]);++k) { + floor_setup->data.t1.subclass_books[j][k]=get_bits(gb, 8)-1; - AV_DEBUG(" book %d. : %d \n", k, floor_setup->subclass_books[j][k]); + AV_DEBUG(" book %d. : %d \n", k, floor_setup->data.t1.subclass_books[j][k]); } } - floor_setup->multiplier=get_bits(gb, 2)+1; - floor_setup->x_list_dim=2; + floor_setup->data.t1.multiplier=get_bits(gb, 2)+1; + floor_setup->data.t1.x_list_dim=2; - for(j=0;jpartitions;++j) { - floor_setup->x_list_dim+=floor_setup->class_dimensions[floor_setup->partition_class[j]]; + for(j=0;jdata.t1.partitions;++j) { + floor_setup->data.t1.x_list_dim+=floor_setup->data.t1.class_dimensions[floor_setup->data.t1.partition_class[j]]; } - floor_setup->x_list=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t)); - floor_setup->x_list_order=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t)); - floor_setup->low_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t)); - floor_setup->high_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t)); + floor_setup->data.t1.x_list=(uint_fast16_t *)av_mallocz(floor_setup->data.t1.x_list_dim * sizeof(uint_fast16_t)); + floor_setup->data.t1.x_list_order=(uint_fast16_t *)av_mallocz(floor_setup->data.t1.x_list_dim * sizeof(uint_fast16_t)); + floor_setup->data.t1.low_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->data.t1.x_list_dim * sizeof(uint_fast16_t)); + floor_setup->data.t1.high_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->data.t1.x_list_dim * sizeof(uint_fast16_t)); rangebits=get_bits(gb, 4); - floor_setup->x_list[0] = 0; - floor_setup->x_list[1] = (1<data.t1.x_list[0] = 0; + floor_setup->data.t1.x_list[1] = (1<partitions;++j) { - for(k=0;kclass_dimensions[floor_setup->partition_class[j]];++k,++floor1_values) { - floor_setup->x_list[floor1_values]=get_bits(gb, rangebits); + for(j=0;jdata.t1.partitions;++j) { + for(k=0;kdata.t1.class_dimensions[floor_setup->data.t1.partition_class[j]];++k,++floor1_values) { + floor_setup->data.t1.x_list[floor1_values]=get_bits(gb, rangebits); - AV_DEBUG(" %d. floor1 Y coord. %d \n", floor1_values, floor_setup->x_list[floor1_values]); + AV_DEBUG(" %d. floor1 Y coord. %d \n", floor1_values, floor_setup->data.t1.x_list[floor1_values]); } } // Precalculate order of x coordinates - needed for decode - for(k=0;kx_list_dim;++k) { - floor_setup->x_list_order[k]=k; + for(k=0;kdata.t1.x_list_dim;++k) { + floor_setup->data.t1.x_list_order[k]=k; } - for(k=0;kx_list_dim-1;++k) { // FIXME optimize sorting ? - for(j=k+1;jx_list_dim;++j) { - if(floor_setup->x_list[floor_setup->x_list_order[k]]>floor_setup->x_list[floor_setup->x_list_order[j]]) { - uint_fast16_t tmp=floor_setup->x_list_order[k]; - floor_setup->x_list_order[k]=floor_setup->x_list_order[j]; - floor_setup->x_list_order[j]=tmp; + for(k=0;kdata.t1.x_list_dim-1;++k) { // FIXME optimize sorting ? + for(j=k+1;jdata.t1.x_list_dim;++j) { + if(floor_setup->data.t1.x_list[floor_setup->data.t1.x_list_order[k]]>floor_setup->data.t1.x_list[floor_setup->data.t1.x_list_order[j]]) { + uint_fast16_t tmp=floor_setup->data.t1.x_list_order[k]; + floor_setup->data.t1.x_list_order[k]=floor_setup->data.t1.x_list_order[j]; + floor_setup->data.t1.x_list_order[j]=tmp; } } } // Precalculate low and high neighbours - for(k=2;kx_list_dim;++k) { - floor_setup->low_neighbour[k]=0; - floor_setup->high_neighbour[k]=1; // correct according to SPEC requirements + for(k=2;kdata.t1.x_list_dim;++k) { + floor_setup->data.t1.low_neighbour[k]=0; + floor_setup->data.t1.high_neighbour[k]=1; // correct according to SPEC requirements for (j=0;jx_list[j]x_list[k]) && - (floor_setup->x_list[j]>floor_setup->x_list[floor_setup->low_neighbour[k]])) { - floor_setup->low_neighbour[k]=j; + if ((floor_setup->data.t1.x_list[j]data.t1.x_list[k]) && + (floor_setup->data.t1.x_list[j]>floor_setup->data.t1.x_list[floor_setup->data.t1.low_neighbour[k]])) { + floor_setup->data.t1.low_neighbour[k]=j; } - if ((floor_setup->x_list[j]>floor_setup->x_list[k]) && - (floor_setup->x_list[j]x_list[floor_setup->high_neighbour[k]])) { - floor_setup->high_neighbour[k]=j; + if ((floor_setup->data.t1.x_list[j]>floor_setup->data.t1.x_list[k]) && + (floor_setup->data.t1.x_list[j]data.t1.x_list[floor_setup->data.t1.high_neighbour[k]])) { + floor_setup->data.t1.high_neighbour[k]=j; } } } } + else if(floor_setup->floor_type==0) { + uint_fast8_t max_codebook_dim=0; + + floor_setup->decode=vorbis_floor0_decode; + + floor_setup->data.t0.order=get_bits(gb, 8); + floor_setup->data.t0.rate=get_bits(gb, 16); + floor_setup->data.t0.bark_map_size=get_bits(gb, 16); + floor_setup->data.t0.amplitude_bits=get_bits(gb, 6); + floor_setup->data.t0.amplitude_offset=get_bits(gb, 8); + floor_setup->data.t0.num_books=get_bits(gb, 4)+1; + + /* allocate mem for booklist */ + floor_setup->data.t0.book_list= + av_malloc(floor_setup->data.t0.num_books); + if(!floor_setup->data.t0.book_list) { return 1; } + /* read book indexes */ + { + int idx; + uint_fast8_t book_idx; + for (idx=0;idxdata.t0.num_books;++idx) { + book_idx=get_bits(gb, 8); + floor_setup->data.t0.book_list[idx]=book_idx; + if (vc->codebooks[book_idx].dimensions > max_codebook_dim) + max_codebook_dim=vc->codebooks[book_idx].dimensions; + + if (floor_setup->data.t0.book_list[idx]>vc->codebook_count) + return 1; + } + } + + /* allocate mem for lsp coefficients */ + { + /* codebook dim is for padding if codebook dim doesn't * + * divide order+1 then we need to read more data */ + floor_setup->data.t0.lsp= + av_malloc((floor_setup->data.t0.order+1 + max_codebook_dim) + * sizeof(float)); + if(!floor_setup->data.t0.book_list) { return 1; } + } + +#ifdef V_DEBUG /* debug output parsed headers */ + AV_DEBUG("floor0 order: %u\n", floor_setup->data.t0.order); + AV_DEBUG("floor0 rate: %u\n", floor_setup->data.t0.rate); + AV_DEBUG("floor0 bark map size: %u\n", + floor_setup->data.t0.bark_map_size); + AV_DEBUG("floor0 amplitude bits: %u\n", + floor_setup->data.t0.amplitude_bits); + AV_DEBUG("floor0 amplitude offset: %u\n", + floor_setup->data.t0.amplitude_offset); + AV_DEBUG("floor0 number of books: %u\n", + floor_setup->data.t0.num_books); + AV_DEBUG("floor0 book list pointer: %p\n", + floor_setup->data.t0.book_list); + { + int idx; + for (idx=0;idxdata.t0.num_books;++idx) { + AV_DEBUG( " Book %d: %u\n", + idx+1, + floor_setup->data.t0.book_list[idx] ); + } + } +#endif + } else { - av_log(vc->avccontext, AV_LOG_ERROR, "Only floor type 1 supported. \n"); + av_log(vc->avccontext, AV_LOG_ERROR, "Invalid floor type!\n"); return 1; } } @@ -653,6 +733,41 @@ static int vorbis_parse_setup_hdr_mappings(vorbis_context *vc) { // Process modes part +static void create_map( vorbis_context * vc, uint_fast8_t mode_number ) +{ + vorbis_floor * floors=vc->floors; + vorbis_floor0 * vf; + int idx; + int_fast32_t * map; + int_fast32_t n; //TODO: could theoretically be smaller? + + n=(vc->modes[mode_number].blockflag ? + vc->blocksize_1 : vc->blocksize_0) / 2; + floors[mode_number].data.t0.map= + av_malloc((n+1) * sizeof(int_fast32_t)); // n+sentinel + + map=floors[mode_number].data.t0.map; + vf=&floors[mode_number].data.t0; + + for (idx=0; idxrate*idx)/(2.0f*n)) * + ((vf->bark_map_size)/ + BARK(vf->rate/2.0f )) ); + if (vf->bark_map_size-1 < map[idx]) { + map[idx]=vf->bark_map_size-1; + } + } + map[n]=-1; + vf->map_size=n; + +# ifdef V_DEBUG + for(idx=0;idx<=n;++idx) { + AV_DEBUG("floor0 map: map at pos %d is %d\n", + idx, map[idx]); + } +# endif +} + static int vorbis_parse_setup_hdr_modes(vorbis_context *vc) { GetBitContext *gb=&vc->gb; uint_fast8_t i; @@ -671,6 +786,8 @@ static int vorbis_parse_setup_hdr_modes(vorbis_context *vc) { mode_setup->mapping=get_bits(gb, 8); //FIXME check AV_DEBUG(" %d mode: blockflag %d, windowtype %d, transformtype %d, mapping %d \n", i, mode_setup->blockflag, mode_setup->windowtype, mode_setup->transformtype, mode_setup->mapping); + + if (vc->floors[i].floor_type == 0) { create_map( vc, i ); } } return 0; } @@ -859,9 +976,125 @@ static int vorbis_decode_init(AVCodecContext *avccontext) { // Decode audiopackets ------------------------------------------------- -// Read and decode floor (type 1 only) +// Read and decode floor -static uint_fast8_t vorbis_floor1_decode(vorbis_context *vc, vorbis_floor *vf, float *vec) { +static uint_fast8_t vorbis_floor0_decode(vorbis_context *vc, + vorbis_floor_data *vfu, float *vec) { + vorbis_floor0 * vf=&vfu->t0; + float * lsp=vf->lsp; + uint_fast32_t amplitude; + uint_fast32_t book_idx; + + amplitude=get_bits(&vc->gb, vf->amplitude_bits); + if (amplitude>0) { + float last = 0; + uint_fast16_t lsp_len = 0; + uint_fast16_t idx; + vorbis_codebook codebook; + + book_idx=get_bits(&vc->gb, ilog(vf->num_books)); + if ( book_idx >= vf->num_books ) { + av_log( vc->avccontext, AV_LOG_ERROR, + "floor0 dec: booknumber too high!\n" ); + //FIXME: look above + } + AV_DEBUG( "floor0 dec: booknumber: %u\n", book_idx ); + codebook=vc->codebooks[vf->book_list[book_idx]]; + + while (lsp_lenorder) { + int vec_off; + + AV_DEBUG( "floor0 dec: book dimension: %d\n", codebook.dimensions ); + AV_DEBUG( "floor0 dec: maximum depth: %d\n", codebook.maxdepth ); + /* read temp vector */ + vec_off=get_vlc2(&vc->gb, + codebook.vlc.table, + codebook.nb_bits, + codebook.maxdepth ) * + codebook.dimensions; + AV_DEBUG( "floor0 dec: vector offset: %d\n", vec_off ); + /* copy each vector component and add last to it */ + for (idx=0; idxorder; + float wstep=M_PI/vf->bark_map_size; + + for(i=0;imap_size, order, wstep); + + i=0; + while(imap_size) { + int j, iter_cond=vf->map[i]; + float p=0.5f; + float q=0.5f; + float two_cos_w=2.0f*cos(wstep*iter_cond); // needed all times + + /* similar part for the q and p products */ + for(j=0;jamplitude_bits; + if ( vf->amplitude_bits ) { + while ( --exponent ) { pow_of_two <<= 1; } + } + else { pow_of_two=1; } + q=exp( ( + ( (amplitude*vf->amplitude_offset)/ + ((pow_of_two-1) * sqrt(p+q)) ) + - vf->amplitude_offset ) * .11512925f + ); + } + + /* fill vector */ + do { vec[i]=q; ++i; }while(vf->map[i]==iter_cond); + } + } + } + else { + /* this channel is unused */ + return 1; + } + + AV_DEBUG(" Floor0 decoded\n"); + + return 0; +} +static uint_fast8_t vorbis_floor1_decode(vorbis_context *vc, vorbis_floor_data *vfu, float *vec) { + vorbis_floor1 * vf=&vfu->t1; GetBitContext *gb=&vc->gb; uint_fast16_t range_v[4]={ 256, 128, 86, 64 }; uint_fast16_t range=range_v[vf->multiplier-1]; @@ -1234,7 +1467,7 @@ static int vorbis_parse_audio_packet(vorbis_context *vc) { memset(ch_res_ptr, 0, sizeof(float)*vc->audio_channels*blocksize/2); //FIXME can this be removed ? memset(ch_floor_ptr, 0, sizeof(float)*vc->audio_channels*blocksize/2); //FIXME can this be removed ? -// Decode floor(1) +// Decode floor for(i=0;iaudio_channels;++i) { vorbis_floor *floor; @@ -1244,7 +1477,7 @@ static int vorbis_parse_audio_packet(vorbis_context *vc) { floor=&vc->floors[mapping->submap_floor[0]]; } - no_residue[i]=vorbis_floor1_decode(vc, floor, ch_floor_ptr); + no_residue[i]=floor->decode(vc, &floor->data, ch_floor_ptr); ch_floor_ptr+=blocksize/2; } diff --git a/libavcodec/vorbis.h b/libavcodec/vorbis.h index 27274a126e..5247d3414b 100644 --- a/libavcodec/vorbis.h +++ b/libavcodec/vorbis.h @@ -12,8 +12,33 @@ typedef struct { unsigned int nb_bits; } vorbis_codebook; +typedef union vorbis_floor_u vorbis_floor_data; +typedef struct vorbis_floor0_s vorbis_floor0; +typedef struct vorbis_floor1_s vorbis_floor1; +struct vorbis_context_s; +typedef +uint_fast8_t (* vorbis_floor_decode_func) + (struct vorbis_context_s *, vorbis_floor_data *, float *); typedef struct { uint_fast8_t floor_type; + vorbis_floor_decode_func decode; + union vorbis_floor_u + { + struct vorbis_floor0_s + { + uint_fast8_t order; + uint_fast16_t rate; + uint_fast16_t bark_map_size; + int_fast32_t * map; + uint_fast32_t map_size; + uint_fast8_t amplitude_bits; + uint_fast8_t amplitude_offset; + uint_fast8_t num_books; + uint_fast8_t * book_list; + float * lsp; + } t0; + struct vorbis_floor1_s + { uint_fast8_t partitions; uint_fast8_t maximum_class; uint_fast8_t partition_class[32]; @@ -27,6 +52,8 @@ typedef struct { uint_fast16_t *x_list_order; uint_fast16_t *low_neighbour; uint_fast16_t *high_neighbour; + } t1; + } data; } vorbis_floor; typedef struct { @@ -57,7 +84,7 @@ typedef struct { uint_fast8_t mapping; } vorbis_mode; -typedef struct { +typedef struct vorbis_context_s { AVCodecContext *avccontext; GetBitContext gb;