Libav
wmaprodec.c
Go to the documentation of this file.
1 /*
2  * Wmapro compatible decoder
3  * Copyright (c) 2007 Baptiste Coudurier, Benjamin Larsson, Ulion
4  * Copyright (c) 2008 - 2011 Sascha Sommer, Benjamin Larsson
5  *
6  * This file is part of Libav.
7  *
8  * Libav is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public
10  * License as published by the Free Software Foundation; either
11  * version 2.1 of the License, or (at your option) any later version.
12  *
13  * Libav is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * Lesser General Public License for more details.
17  *
18  * You should have received a copy of the GNU Lesser General Public
19  * License along with Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
89 #include <inttypes.h>
90 
91 #include "libavutil/float_dsp.h"
92 #include "libavutil/intfloat.h"
93 #include "libavutil/intreadwrite.h"
94 #include "avcodec.h"
95 #include "internal.h"
96 #include "get_bits.h"
97 #include "put_bits.h"
98 #include "wmaprodata.h"
99 #include "sinewin.h"
100 #include "wma.h"
101 #include "wma_common.h"
102 
104 #define WMAPRO_MAX_CHANNELS 8
105 #define MAX_SUBFRAMES 32
106 #define MAX_BANDS 29
107 #define MAX_FRAMESIZE 32768
108 
109 #define WMAPRO_BLOCK_MIN_BITS 6
110 #define WMAPRO_BLOCK_MAX_BITS 13
111 #define WMAPRO_BLOCK_MIN_SIZE (1 << WMAPRO_BLOCK_MIN_BITS)
112 #define WMAPRO_BLOCK_MAX_SIZE (1 << WMAPRO_BLOCK_MAX_BITS)
113 #define WMAPRO_BLOCK_SIZES (WMAPRO_BLOCK_MAX_BITS - WMAPRO_BLOCK_MIN_BITS + 1)
114 
115 
116 #define VLCBITS 9
117 #define SCALEVLCBITS 8
118 #define VEC4MAXDEPTH ((HUFF_VEC4_MAXBITS+VLCBITS-1)/VLCBITS)
119 #define VEC2MAXDEPTH ((HUFF_VEC2_MAXBITS+VLCBITS-1)/VLCBITS)
120 #define VEC1MAXDEPTH ((HUFF_VEC1_MAXBITS+VLCBITS-1)/VLCBITS)
121 #define SCALEMAXDEPTH ((HUFF_SCALE_MAXBITS+SCALEVLCBITS-1)/SCALEVLCBITS)
122 #define SCALERLMAXDEPTH ((HUFF_SCALE_RL_MAXBITS+VLCBITS-1)/VLCBITS)
123 
124 static VLC sf_vlc;
125 static VLC sf_rl_vlc;
126 static VLC vec4_vlc;
127 static VLC vec2_vlc;
128 static VLC vec1_vlc;
129 static VLC coef_vlc[2];
130 static float sin64[33];
131 
135 typedef struct {
136  int16_t prev_block_len;
139  uint16_t subframe_len[MAX_SUBFRAMES];
140  uint16_t subframe_offset[MAX_SUBFRAMES];
142  uint16_t decoded_samples;
145  int8_t reuse_sf;
148  int saved_scale_factors[2][MAX_BANDS];
152  float* coeffs;
153  uint16_t num_vec_coeffs;
156 
160 typedef struct {
162  int8_t transform;
163  int8_t transform_band[MAX_BANDS];
164  float decorrelation_matrix[WMAPRO_MAX_CHANNELS*WMAPRO_MAX_CHANNELS];
165  float* channel_data[WMAPRO_MAX_CHANNELS];
167 
171 typedef struct WMAProDecodeCtx {
172  /* generic decoder variables */
181 
182  /* frame size dependent frame information (set during initialization) */
183  uint32_t decode_flags;
187  uint16_t samples_per_frame;
188  uint16_t log2_frame_size;
189  int8_t lfe_channel;
198 
199  /* packet decode state */
209 
210  /* frame decode state */
211  uint32_t frame_num;
215  int8_t skip_frame;
217 
218  /* subframe/block decode state */
219  int16_t subframe_len;
222  int8_t num_bands;
224  int16_t* cur_sfb_offsets;
226  int8_t esc_len;
227 
230 
233 
234 
240 {
241 #define PRINT(a, b) av_log(s->avctx, AV_LOG_DEBUG, " %s = %d\n", a, b);
242 #define PRINT_HEX(a, b) av_log(s->avctx, AV_LOG_DEBUG, " %s = %"PRIx32"\n", a, b);
243 
244  PRINT("ed sample bit depth", s->bits_per_sample);
245  PRINT_HEX("ed decode flags", s->decode_flags);
246  PRINT("samples per frame", s->samples_per_frame);
247  PRINT("log2 frame size", s->log2_frame_size);
248  PRINT("max num subframes", s->max_num_subframes);
249  PRINT("len prefix", s->len_prefix);
250  PRINT("num channels", s->avctx->channels);
251 }
252 
259 {
260  WMAProDecodeCtx *s = avctx->priv_data;
261  int i;
262 
263  for (i = 0; i < WMAPRO_BLOCK_SIZES; i++)
264  ff_mdct_end(&s->mdct_ctx[i]);
265 
266  return 0;
267 }
268 
275 {
276  WMAProDecodeCtx *s = avctx->priv_data;
277  uint8_t *edata_ptr = avctx->extradata;
278  unsigned int channel_mask;
279  int i, bits;
280  int log2_max_num_subframes;
281  int num_possible_block_sizes;
282 
283  if (!avctx->block_align) {
284  av_log(avctx, AV_LOG_ERROR, "block_align is not set\n");
285  return AVERROR(EINVAL);
286  }
287 
288  s->avctx = avctx;
290 
292 
294 
295  if (avctx->extradata_size >= 18) {
296  s->decode_flags = AV_RL16(edata_ptr+14);
297  channel_mask = AV_RL32(edata_ptr+2);
298  s->bits_per_sample = AV_RL16(edata_ptr);
300  for (i = 0; i < avctx->extradata_size; i++)
301  av_dlog(avctx, "[%x] ", avctx->extradata[i]);
302  av_dlog(avctx, "\n");
303 
304  } else {
305  avpriv_request_sample(avctx, "Unknown extradata size");
306  return AVERROR_PATCHWELCOME;
307  }
308 
310  s->log2_frame_size = av_log2(avctx->block_align) + 4;
311 
313  s->skip_frame = 1; /* skip first frame */
314  s->packet_loss = 1;
315  s->len_prefix = (s->decode_flags & 0x40);
316 
318  bits = ff_wma_get_frame_len_bits(avctx->sample_rate, 3, s->decode_flags);
319  if (bits > WMAPRO_BLOCK_MAX_BITS) {
320  avpriv_request_sample(avctx, "14-bit block sizes");
321  return AVERROR_PATCHWELCOME;
322  }
323  s->samples_per_frame = 1 << bits;
324 
326  log2_max_num_subframes = ((s->decode_flags & 0x38) >> 3);
327  s->max_num_subframes = 1 << log2_max_num_subframes;
328  if (s->max_num_subframes == 16 || s->max_num_subframes == 4)
329  s->max_subframe_len_bit = 1;
330  s->subframe_len_bits = av_log2(log2_max_num_subframes) + 1;
331 
332  num_possible_block_sizes = log2_max_num_subframes + 1;
334  s->dynamic_range_compression = (s->decode_flags & 0x80);
335 
336  if (s->max_num_subframes > MAX_SUBFRAMES) {
337  av_log(avctx, AV_LOG_ERROR, "invalid number of subframes %"PRId8"\n",
338  s->max_num_subframes);
339  return AVERROR_INVALIDDATA;
340  }
341 
343  av_log(avctx, AV_LOG_ERROR, "Invalid minimum block size %"PRId8"\n",
344  s->max_num_subframes);
345  return AVERROR_INVALIDDATA;
346  }
347 
348  if (s->avctx->sample_rate <= 0) {
349  av_log(avctx, AV_LOG_ERROR, "invalid sample rate\n");
350  return AVERROR_INVALIDDATA;
351  }
352 
353  if (avctx->channels < 0) {
354  av_log(avctx, AV_LOG_ERROR, "invalid number of channels %d\n",
355  avctx->channels);
356  return AVERROR_INVALIDDATA;
357  } else if (avctx->channels > WMAPRO_MAX_CHANNELS) {
358  avpriv_request_sample(avctx,
359  "More than %d channels", WMAPRO_MAX_CHANNELS);
360  return AVERROR_PATCHWELCOME;
361  }
362 
364  for (i = 0; i < avctx->channels; i++)
366 
368  s->lfe_channel = -1;
369 
370  if (channel_mask & 8) {
371  unsigned int mask;
372  for (mask = 1; mask < 16; mask <<= 1) {
373  if (channel_mask & mask)
374  ++s->lfe_channel;
375  }
376  }
377 
379  scale_huffbits, 1, 1,
380  scale_huffcodes, 2, 2, 616);
381 
383  scale_rl_huffbits, 1, 1,
384  scale_rl_huffcodes, 4, 4, 1406);
385 
386  INIT_VLC_STATIC(&coef_vlc[0], VLCBITS, HUFF_COEF0_SIZE,
387  coef0_huffbits, 1, 1,
388  coef0_huffcodes, 4, 4, 2108);
389 
390  INIT_VLC_STATIC(&coef_vlc[1], VLCBITS, HUFF_COEF1_SIZE,
391  coef1_huffbits, 1, 1,
392  coef1_huffcodes, 4, 4, 3912);
393 
395  vec4_huffbits, 1, 1,
396  vec4_huffcodes, 2, 2, 604);
397 
399  vec2_huffbits, 1, 1,
400  vec2_huffcodes, 2, 2, 562);
401 
403  vec1_huffbits, 1, 1,
404  vec1_huffcodes, 2, 2, 562);
405 
408  for (i = 0; i < num_possible_block_sizes; i++) {
409  int subframe_len = s->samples_per_frame >> i;
410  int x;
411  int band = 1;
412 
413  s->sfb_offsets[i][0] = 0;
414 
415  for (x = 0; x < MAX_BANDS-1 && s->sfb_offsets[i][band - 1] < subframe_len; x++) {
416  int offset = (subframe_len * 2 * critical_freq[x])
417  / s->avctx->sample_rate + 2;
418  offset &= ~3;
419  if (offset > s->sfb_offsets[i][band - 1])
420  s->sfb_offsets[i][band++] = offset;
421  }
422  s->sfb_offsets[i][band - 1] = subframe_len;
423  s->num_sfb[i] = band - 1;
424  }
425 
426 
432  for (i = 0; i < num_possible_block_sizes; i++) {
433  int b;
434  for (b = 0; b < s->num_sfb[i]; b++) {
435  int x;
436  int offset = ((s->sfb_offsets[i][b]
437  + s->sfb_offsets[i][b + 1] - 1) << i) >> 1;
438  for (x = 0; x < num_possible_block_sizes; x++) {
439  int v = 0;
440  while (s->sfb_offsets[x][v + 1] << x < offset)
441  if (++v >= MAX_BANDS)
442  return AVERROR_INVALIDDATA;
443  s->sf_offsets[i][x][b] = v;
444  }
445  }
446  }
447 
449  for (i = 0; i < WMAPRO_BLOCK_SIZES; i++)
451  1.0 / (1 << (WMAPRO_BLOCK_MIN_BITS + i - 1))
452  / (1 << (s->bits_per_sample - 1)));
453 
455  for (i = 0; i < WMAPRO_BLOCK_SIZES; i++) {
456  const int win_idx = WMAPRO_BLOCK_MAX_BITS - i;
457  ff_init_ff_sine_windows(win_idx);
458  s->windows[WMAPRO_BLOCK_SIZES - i - 1] = ff_sine_windows[win_idx];
459  }
460 
462  for (i = 0; i < num_possible_block_sizes; i++) {
463  int block_size = s->samples_per_frame >> i;
464  int cutoff = (440*block_size + 3 * (s->avctx->sample_rate >> 1) - 1)
465  / s->avctx->sample_rate;
466  s->subwoofer_cutoffs[i] = av_clip(cutoff, 4, block_size);
467  }
468 
470  for (i = 0; i < 33; i++)
471  sin64[i] = sin(i*M_PI / 64.0);
472 
473  if (avctx->debug & FF_DEBUG_BITSTREAM)
474  dump_context(s);
475 
476  avctx->channel_layout = channel_mask;
477 
478  return 0;
479 }
480 
487 static int decode_subframe_length(WMAProDecodeCtx *s, int offset)
488 {
489  int frame_len_shift = 0;
490  int subframe_len;
491 
493  if (offset == s->samples_per_frame - s->min_samples_per_subframe)
494  return s->min_samples_per_subframe;
495 
497  if (s->max_subframe_len_bit) {
498  if (get_bits1(&s->gb))
499  frame_len_shift = 1 + get_bits(&s->gb, s->subframe_len_bits-1);
500  } else
501  frame_len_shift = get_bits(&s->gb, s->subframe_len_bits);
502 
503  subframe_len = s->samples_per_frame >> frame_len_shift;
504 
506  if (subframe_len < s->min_samples_per_subframe ||
507  subframe_len > s->samples_per_frame) {
508  av_log(s->avctx, AV_LOG_ERROR, "broken frame: subframe_len %i\n",
509  subframe_len);
510  return AVERROR_INVALIDDATA;
511  }
512  return subframe_len;
513 }
514 
536 {
537  uint16_t num_samples[WMAPRO_MAX_CHANNELS] = { 0 };
538  uint8_t contains_subframe[WMAPRO_MAX_CHANNELS];
539  int channels_for_cur_subframe = s->avctx->channels;
540  int fixed_channel_layout = 0;
541  int min_channel_len = 0;
542  int c;
543 
544  /* Should never consume more than 3073 bits (256 iterations for the
545  * while loop when always the minimum amount of 128 samples is subtracted
546  * from missing samples in the 8 channel case).
547  * 1 + BLOCK_MAX_SIZE * MAX_CHANNELS / BLOCK_MIN_SIZE * (MAX_CHANNELS + 4)
548  */
549 
551  for (c = 0; c < s->avctx->channels; c++)
552  s->channel[c].num_subframes = 0;
553 
554  if (s->max_num_subframes == 1 || get_bits1(&s->gb))
555  fixed_channel_layout = 1;
556 
558  do {
559  int subframe_len;
560 
562  for (c = 0; c < s->avctx->channels; c++) {
563  if (num_samples[c] == min_channel_len) {
564  if (fixed_channel_layout || channels_for_cur_subframe == 1 ||
565  (min_channel_len == s->samples_per_frame - s->min_samples_per_subframe))
566  contains_subframe[c] = 1;
567  else
568  contains_subframe[c] = get_bits1(&s->gb);
569  } else
570  contains_subframe[c] = 0;
571  }
572 
574  if ((subframe_len = decode_subframe_length(s, min_channel_len)) <= 0)
575  return AVERROR_INVALIDDATA;
576 
578  min_channel_len += subframe_len;
579  for (c = 0; c < s->avctx->channels; c++) {
580  WMAProChannelCtx* chan = &s->channel[c];
581 
582  if (contains_subframe[c]) {
583  if (chan->num_subframes >= MAX_SUBFRAMES) {
585  "broken frame: num subframes > 31\n");
586  return AVERROR_INVALIDDATA;
587  }
588  chan->subframe_len[chan->num_subframes] = subframe_len;
589  num_samples[c] += subframe_len;
590  ++chan->num_subframes;
591  if (num_samples[c] > s->samples_per_frame) {
592  av_log(s->avctx, AV_LOG_ERROR, "broken frame: "
593  "channel len > samples_per_frame\n");
594  return AVERROR_INVALIDDATA;
595  }
596  } else if (num_samples[c] <= min_channel_len) {
597  if (num_samples[c] < min_channel_len) {
598  channels_for_cur_subframe = 0;
599  min_channel_len = num_samples[c];
600  }
601  ++channels_for_cur_subframe;
602  }
603  }
604  } while (min_channel_len < s->samples_per_frame);
605 
606  for (c = 0; c < s->avctx->channels; c++) {
607  int i;
608  int offset = 0;
609  for (i = 0; i < s->channel[c].num_subframes; i++) {
610  av_dlog(s->avctx, "frame[%i] channel[%i] subframe[%i]"
611  " len %i\n", s->frame_num, c, i,
612  s->channel[c].subframe_len[i]);
613  s->channel[c].subframe_offset[i] = offset;
614  offset += s->channel[c].subframe_len[i];
615  }
616  }
617 
618  return 0;
619 }
620 
627  WMAProChannelGrp *chgroup)
628 {
629  int i;
630  int offset = 0;
631  int8_t rotation_offset[WMAPRO_MAX_CHANNELS * WMAPRO_MAX_CHANNELS];
632  memset(chgroup->decorrelation_matrix, 0, s->avctx->channels *
633  s->avctx->channels * sizeof(*chgroup->decorrelation_matrix));
634 
635  for (i = 0; i < chgroup->num_channels * (chgroup->num_channels - 1) >> 1; i++)
636  rotation_offset[i] = get_bits(&s->gb, 6);
637 
638  for (i = 0; i < chgroup->num_channels; i++)
639  chgroup->decorrelation_matrix[chgroup->num_channels * i + i] =
640  get_bits1(&s->gb) ? 1.0 : -1.0;
641 
642  for (i = 1; i < chgroup->num_channels; i++) {
643  int x;
644  for (x = 0; x < i; x++) {
645  int y;
646  for (y = 0; y < i + 1; y++) {
647  float v1 = chgroup->decorrelation_matrix[x * chgroup->num_channels + y];
648  float v2 = chgroup->decorrelation_matrix[i * chgroup->num_channels + y];
649  int n = rotation_offset[offset + x];
650  float sinv;
651  float cosv;
652 
653  if (n < 32) {
654  sinv = sin64[n];
655  cosv = sin64[32 - n];
656  } else {
657  sinv = sin64[64 - n];
658  cosv = -sin64[n - 32];
659  }
660 
661  chgroup->decorrelation_matrix[y + x * chgroup->num_channels] =
662  (v1 * sinv) - (v2 * cosv);
663  chgroup->decorrelation_matrix[y + i * chgroup->num_channels] =
664  (v1 * cosv) + (v2 * sinv);
665  }
666  }
667  offset += i;
668  }
669 }
670 
677 {
678  int i;
679  /* should never consume more than 1921 bits for the 8 channel case
680  * 1 + MAX_CHANNELS * (MAX_CHANNELS + 2 + 3 * MAX_CHANNELS * MAX_CHANNELS
681  * + MAX_CHANNELS + MAX_BANDS + 1)
682  */
683 
685  s->num_chgroups = 0;
686  if (s->avctx->channels > 1) {
687  int remaining_channels = s->channels_for_cur_subframe;
688 
689  if (get_bits1(&s->gb)) {
691  "Channel transform bit");
692  return AVERROR_PATCHWELCOME;
693  }
694 
695  for (s->num_chgroups = 0; remaining_channels &&
697  WMAProChannelGrp* chgroup = &s->chgroup[s->num_chgroups];
698  float** channel_data = chgroup->channel_data;
699  chgroup->num_channels = 0;
700  chgroup->transform = 0;
701 
703  if (remaining_channels > 2) {
704  for (i = 0; i < s->channels_for_cur_subframe; i++) {
705  int channel_idx = s->channel_indexes_for_cur_subframe[i];
706  if (!s->channel[channel_idx].grouped
707  && get_bits1(&s->gb)) {
708  ++chgroup->num_channels;
709  s->channel[channel_idx].grouped = 1;
710  *channel_data++ = s->channel[channel_idx].coeffs;
711  }
712  }
713  } else {
714  chgroup->num_channels = remaining_channels;
715  for (i = 0; i < s->channels_for_cur_subframe; i++) {
716  int channel_idx = s->channel_indexes_for_cur_subframe[i];
717  if (!s->channel[channel_idx].grouped)
718  *channel_data++ = s->channel[channel_idx].coeffs;
719  s->channel[channel_idx].grouped = 1;
720  }
721  }
722 
724  if (chgroup->num_channels == 2) {
725  if (get_bits1(&s->gb)) {
726  if (get_bits1(&s->gb)) {
728  "Unknown channel transform type");
729  return AVERROR_PATCHWELCOME;
730  }
731  } else {
732  chgroup->transform = 1;
733  if (s->avctx->channels == 2) {
734  chgroup->decorrelation_matrix[0] = 1.0;
735  chgroup->decorrelation_matrix[1] = -1.0;
736  chgroup->decorrelation_matrix[2] = 1.0;
737  chgroup->decorrelation_matrix[3] = 1.0;
738  } else {
740  chgroup->decorrelation_matrix[0] = 0.70703125;
741  chgroup->decorrelation_matrix[1] = -0.70703125;
742  chgroup->decorrelation_matrix[2] = 0.70703125;
743  chgroup->decorrelation_matrix[3] = 0.70703125;
744  }
745  }
746  } else if (chgroup->num_channels > 2) {
747  if (get_bits1(&s->gb)) {
748  chgroup->transform = 1;
749  if (get_bits1(&s->gb)) {
750  decode_decorrelation_matrix(s, chgroup);
751  } else {
753  if (chgroup->num_channels > 6) {
755  "Coupled channels > 6");
756  } else {
757  memcpy(chgroup->decorrelation_matrix,
759  chgroup->num_channels * chgroup->num_channels *
760  sizeof(*chgroup->decorrelation_matrix));
761  }
762  }
763  }
764  }
765 
767  if (chgroup->transform) {
768  if (!get_bits1(&s->gb)) {
769  int i;
771  for (i = 0; i < s->num_bands; i++) {
772  chgroup->transform_band[i] = get_bits1(&s->gb);
773  }
774  } else {
775  memset(chgroup->transform_band, 1, s->num_bands);
776  }
777  }
778  remaining_channels -= chgroup->num_channels;
779  }
780  }
781  return 0;
782 }
783 
790 static int decode_coeffs(WMAProDecodeCtx *s, int c)
791 {
792  /* Integers 0..15 as single-precision floats. The table saves a
793  costly int to float conversion, and storing the values as
794  integers allows fast sign-flipping. */
795  static const uint32_t fval_tab[16] = {
796  0x00000000, 0x3f800000, 0x40000000, 0x40400000,
797  0x40800000, 0x40a00000, 0x40c00000, 0x40e00000,
798  0x41000000, 0x41100000, 0x41200000, 0x41300000,
799  0x41400000, 0x41500000, 0x41600000, 0x41700000,
800  };
801  int vlctable;
802  VLC* vlc;
803  WMAProChannelCtx* ci = &s->channel[c];
804  int rl_mode = 0;
805  int cur_coeff = 0;
806  int num_zeros = 0;
807  const uint16_t* run;
808  const float* level;
809 
810  av_dlog(s->avctx, "decode coefficients for channel %i\n", c);
811 
812  vlctable = get_bits1(&s->gb);
813  vlc = &coef_vlc[vlctable];
814 
815  if (vlctable) {
816  run = coef1_run;
817  level = coef1_level;
818  } else {
819  run = coef0_run;
820  level = coef0_level;
821  }
822 
825  while ((s->transmit_num_vec_coeffs || !rl_mode) &&
826  (cur_coeff + 3 < ci->num_vec_coeffs)) {
827  uint32_t vals[4];
828  int i;
829  unsigned int idx;
830 
831  idx = get_vlc2(&s->gb, vec4_vlc.table, VLCBITS, VEC4MAXDEPTH);
832 
833  if (idx == HUFF_VEC4_SIZE - 1) {
834  for (i = 0; i < 4; i += 2) {
835  idx = get_vlc2(&s->gb, vec2_vlc.table, VLCBITS, VEC2MAXDEPTH);
836  if (idx == HUFF_VEC2_SIZE - 1) {
837  uint32_t v0, v1;
838  v0 = get_vlc2(&s->gb, vec1_vlc.table, VLCBITS, VEC1MAXDEPTH);
839  if (v0 == HUFF_VEC1_SIZE - 1)
840  v0 += ff_wma_get_large_val(&s->gb);
841  v1 = get_vlc2(&s->gb, vec1_vlc.table, VLCBITS, VEC1MAXDEPTH);
842  if (v1 == HUFF_VEC1_SIZE - 1)
843  v1 += ff_wma_get_large_val(&s->gb);
844  vals[i ] = av_float2int(v0);
845  vals[i+1] = av_float2int(v1);
846  } else {
847  vals[i] = fval_tab[symbol_to_vec2[idx] >> 4 ];
848  vals[i+1] = fval_tab[symbol_to_vec2[idx] & 0xF];
849  }
850  }
851  } else {
852  vals[0] = fval_tab[ symbol_to_vec4[idx] >> 12 ];
853  vals[1] = fval_tab[(symbol_to_vec4[idx] >> 8) & 0xF];
854  vals[2] = fval_tab[(symbol_to_vec4[idx] >> 4) & 0xF];
855  vals[3] = fval_tab[ symbol_to_vec4[idx] & 0xF];
856  }
857 
859  for (i = 0; i < 4; i++) {
860  if (vals[i]) {
861  uint32_t sign = get_bits1(&s->gb) - 1;
862  AV_WN32A(&ci->coeffs[cur_coeff], vals[i] ^ sign << 31);
863  num_zeros = 0;
864  } else {
865  ci->coeffs[cur_coeff] = 0;
868  rl_mode |= (++num_zeros > s->subframe_len >> 8);
869  }
870  ++cur_coeff;
871  }
872  }
873 
875  if (cur_coeff < s->subframe_len) {
876  memset(&ci->coeffs[cur_coeff], 0,
877  sizeof(*ci->coeffs) * (s->subframe_len - cur_coeff));
878  if (ff_wma_run_level_decode(s->avctx, &s->gb, vlc,
879  level, run, 1, ci->coeffs,
880  cur_coeff, s->subframe_len,
881  s->subframe_len, s->esc_len, 0))
882  return AVERROR_INVALIDDATA;
883  }
884 
885  return 0;
886 }
887 
894 {
895  int i;
896 
901  for (i = 0; i < s->channels_for_cur_subframe; i++) {
902  int c = s->channel_indexes_for_cur_subframe[i];
903  int* sf;
904  int* sf_end;
906  sf_end = s->channel[c].scale_factors + s->num_bands;
907 
913  if (s->channel[c].reuse_sf) {
914  const int8_t* sf_offsets = s->sf_offsets[s->table_idx][s->channel[c].table_idx];
915  int b;
916  for (b = 0; b < s->num_bands; b++)
917  s->channel[c].scale_factors[b] =
918  s->channel[c].saved_scale_factors[s->channel[c].scale_factor_idx][*sf_offsets++];
919  }
920 
921  if (!s->channel[c].cur_subframe || get_bits1(&s->gb)) {
922 
923  if (!s->channel[c].reuse_sf) {
924  int val;
926  s->channel[c].scale_factor_step = get_bits(&s->gb, 2) + 1;
927  val = 45 / s->channel[c].scale_factor_step;
928  for (sf = s->channel[c].scale_factors; sf < sf_end; sf++) {
929  val += get_vlc2(&s->gb, sf_vlc.table, SCALEVLCBITS, SCALEMAXDEPTH) - 60;
930  *sf = val;
931  }
932  } else {
933  int i;
935  for (i = 0; i < s->num_bands; i++) {
936  int idx;
937  int skip;
938  int val;
939  int sign;
940 
941  idx = get_vlc2(&s->gb, sf_rl_vlc.table, VLCBITS, SCALERLMAXDEPTH);
942 
943  if (!idx) {
944  uint32_t code = get_bits(&s->gb, 14);
945  val = code >> 6;
946  sign = (code & 1) - 1;
947  skip = (code & 0x3f) >> 1;
948  } else if (idx == 1) {
949  break;
950  } else {
951  skip = scale_rl_run[idx];
952  val = scale_rl_level[idx];
953  sign = get_bits1(&s->gb)-1;
954  }
955 
956  i += skip;
957  if (i >= s->num_bands) {
959  "invalid scale factor coding\n");
960  return AVERROR_INVALIDDATA;
961  }
962  s->channel[c].scale_factors[i] += (val ^ sign) - sign;
963  }
964  }
967  s->channel[c].table_idx = s->table_idx;
968  s->channel[c].reuse_sf = 1;
969  }
970 
973  for (sf = s->channel[c].scale_factors + 1; sf < sf_end; sf++) {
974  s->channel[c].max_scale_factor =
975  FFMAX(s->channel[c].max_scale_factor, *sf);
976  }
977 
978  }
979  return 0;
980 }
981 
987 {
988  int i;
989 
990  for (i = 0; i < s->num_chgroups; i++) {
991  if (s->chgroup[i].transform) {
992  float data[WMAPRO_MAX_CHANNELS];
993  const int num_channels = s->chgroup[i].num_channels;
994  float** ch_data = s->chgroup[i].channel_data;
995  float** ch_end = ch_data + num_channels;
996  const int8_t* tb = s->chgroup[i].transform_band;
997  int16_t* sfb;
998 
1000  for (sfb = s->cur_sfb_offsets;
1001  sfb < s->cur_sfb_offsets + s->num_bands; sfb++) {
1002  int y;
1003  if (*tb++ == 1) {
1005  for (y = sfb[0]; y < FFMIN(sfb[1], s->subframe_len); y++) {
1006  const float* mat = s->chgroup[i].decorrelation_matrix;
1007  const float* data_end = data + num_channels;
1008  float* data_ptr = data;
1009  float** ch;
1010 
1011  for (ch = ch_data; ch < ch_end; ch++)
1012  *data_ptr++ = (*ch)[y];
1013 
1014  for (ch = ch_data; ch < ch_end; ch++) {
1015  float sum = 0;
1016  data_ptr = data;
1017  while (data_ptr < data_end)
1018  sum += *data_ptr++ * *mat++;
1019 
1020  (*ch)[y] = sum;
1021  }
1022  }
1023  } else if (s->avctx->channels == 2) {
1024  int len = FFMIN(sfb[1], s->subframe_len) - sfb[0];
1025  s->fdsp.vector_fmul_scalar(ch_data[0] + sfb[0],
1026  ch_data[0] + sfb[0],
1027  181.0 / 128, len);
1028  s->fdsp.vector_fmul_scalar(ch_data[1] + sfb[0],
1029  ch_data[1] + sfb[0],
1030  181.0 / 128, len);
1031  }
1032  }
1033  }
1034  }
1035 }
1036 
1042 {
1043  int i;
1044  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1045  int c = s->channel_indexes_for_cur_subframe[i];
1046  float* window;
1047  int winlen = s->channel[c].prev_block_len;
1048  float* start = s->channel[c].coeffs - (winlen >> 1);
1049 
1050  if (s->subframe_len < winlen) {
1051  start += (winlen - s->subframe_len) >> 1;
1052  winlen = s->subframe_len;
1053  }
1054 
1055  window = s->windows[av_log2(winlen) - WMAPRO_BLOCK_MIN_BITS];
1056 
1057  winlen >>= 1;
1058 
1059  s->fdsp.vector_fmul_window(start, start, start + winlen,
1060  window, winlen);
1061 
1062  s->channel[c].prev_block_len = s->subframe_len;
1063  }
1064 }
1065 
1072 {
1073  int offset = s->samples_per_frame;
1074  int subframe_len = s->samples_per_frame;
1075  int i;
1076  int total_samples = s->samples_per_frame * s->avctx->channels;
1077  int transmit_coeffs = 0;
1078  int cur_subwoofer_cutoff;
1079 
1080  s->subframe_offset = get_bits_count(&s->gb);
1081 
1086  for (i = 0; i < s->avctx->channels; i++) {
1087  s->channel[i].grouped = 0;
1088  if (offset > s->channel[i].decoded_samples) {
1089  offset = s->channel[i].decoded_samples;
1090  subframe_len =
1092  }
1093  }
1094 
1095  av_dlog(s->avctx,
1096  "processing subframe with offset %i len %i\n", offset, subframe_len);
1097 
1100  for (i = 0; i < s->avctx->channels; i++) {
1101  const int cur_subframe = s->channel[i].cur_subframe;
1103  total_samples -= s->channel[i].decoded_samples;
1104 
1106  if (offset == s->channel[i].decoded_samples &&
1107  subframe_len == s->channel[i].subframe_len[cur_subframe]) {
1108  total_samples -= s->channel[i].subframe_len[cur_subframe];
1109  s->channel[i].decoded_samples +=
1110  s->channel[i].subframe_len[cur_subframe];
1113  }
1114  }
1115 
1118  if (!total_samples)
1119  s->parsed_all_subframes = 1;
1120 
1121 
1122  av_dlog(s->avctx, "subframe is part of %i channels\n",
1124 
1126  s->table_idx = av_log2(s->samples_per_frame/subframe_len);
1127  s->num_bands = s->num_sfb[s->table_idx];
1129  cur_subwoofer_cutoff = s->subwoofer_cutoffs[s->table_idx];
1130 
1132  offset += s->samples_per_frame >> 1;
1133 
1134  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1135  int c = s->channel_indexes_for_cur_subframe[i];
1136 
1137  s->channel[c].coeffs = &s->channel[c].out[offset];
1138  }
1139 
1140  s->subframe_len = subframe_len;
1141  s->esc_len = av_log2(s->subframe_len - 1) + 1;
1142 
1144  if (get_bits1(&s->gb)) {
1145  int num_fill_bits;
1146  if (!(num_fill_bits = get_bits(&s->gb, 2))) {
1147  int len = get_bits(&s->gb, 4);
1148  num_fill_bits = get_bits(&s->gb, len) + 1;
1149  }
1150 
1151  if (num_fill_bits >= 0) {
1152  if (get_bits_count(&s->gb) + num_fill_bits > s->num_saved_bits) {
1153  av_log(s->avctx, AV_LOG_ERROR, "invalid number of fill bits\n");
1154  return AVERROR_INVALIDDATA;
1155  }
1156 
1157  skip_bits_long(&s->gb, num_fill_bits);
1158  }
1159  }
1160 
1162  if (get_bits1(&s->gb)) {
1163  avpriv_request_sample(s->avctx, "Reserved bit");
1164  return AVERROR_PATCHWELCOME;
1165  }
1166 
1167 
1168  if (decode_channel_transform(s) < 0)
1169  return AVERROR_INVALIDDATA;
1170 
1171 
1172  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1173  int c = s->channel_indexes_for_cur_subframe[i];
1174  if ((s->channel[c].transmit_coefs = get_bits1(&s->gb)))
1175  transmit_coeffs = 1;
1176  }
1177 
1178  if (transmit_coeffs) {
1179  int step;
1180  int quant_step = 90 * s->bits_per_sample >> 4;
1181 
1183  if ((s->transmit_num_vec_coeffs = get_bits1(&s->gb))) {
1184  int num_bits = av_log2((s->subframe_len + 3)/4) + 1;
1185  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1186  int c = s->channel_indexes_for_cur_subframe[i];
1187  int num_vec_coeffs = get_bits(&s->gb, num_bits) << 2;
1188  if (num_vec_coeffs + offset > FF_ARRAY_ELEMS(s->channel[c].out)) {
1189  av_log(s->avctx, AV_LOG_ERROR, "num_vec_coeffs %d is too large\n", num_vec_coeffs);
1190  return AVERROR_INVALIDDATA;
1191  }
1192  s->channel[c].num_vec_coeffs = num_vec_coeffs;
1193  }
1194  } else {
1195  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1196  int c = s->channel_indexes_for_cur_subframe[i];
1197  s->channel[c].num_vec_coeffs = s->subframe_len;
1198  }
1199  }
1201  step = get_sbits(&s->gb, 6);
1202  quant_step += step;
1203  if (step == -32 || step == 31) {
1204  const int sign = (step == 31) - 1;
1205  int quant = 0;
1206  while (get_bits_count(&s->gb) + 5 < s->num_saved_bits &&
1207  (step = get_bits(&s->gb, 5)) == 31) {
1208  quant += 31;
1209  }
1210  quant_step += ((quant + step) ^ sign) - sign;
1211  }
1212  if (quant_step < 0) {
1213  av_log(s->avctx, AV_LOG_DEBUG, "negative quant step\n");
1214  }
1215 
1218  if (s->channels_for_cur_subframe == 1) {
1219  s->channel[s->channel_indexes_for_cur_subframe[0]].quant_step = quant_step;
1220  } else {
1221  int modifier_len = get_bits(&s->gb, 3);
1222  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1223  int c = s->channel_indexes_for_cur_subframe[i];
1224  s->channel[c].quant_step = quant_step;
1225  if (get_bits1(&s->gb)) {
1226  if (modifier_len) {
1227  s->channel[c].quant_step += get_bits(&s->gb, modifier_len) + 1;
1228  } else
1229  ++s->channel[c].quant_step;
1230  }
1231  }
1232  }
1233 
1235  if (decode_scale_factors(s) < 0)
1236  return AVERROR_INVALIDDATA;
1237  }
1238 
1239  av_dlog(s->avctx, "BITSTREAM: subframe header length was %i\n",
1240  get_bits_count(&s->gb) - s->subframe_offset);
1241 
1243  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1244  int c = s->channel_indexes_for_cur_subframe[i];
1245  if (s->channel[c].transmit_coefs &&
1246  get_bits_count(&s->gb) < s->num_saved_bits) {
1247  decode_coeffs(s, c);
1248  } else
1249  memset(s->channel[c].coeffs, 0,
1250  sizeof(*s->channel[c].coeffs) * subframe_len);
1251  }
1252 
1253  av_dlog(s->avctx, "BITSTREAM: subframe length was %i\n",
1254  get_bits_count(&s->gb) - s->subframe_offset);
1255 
1256  if (transmit_coeffs) {
1257  FFTContext *mdct = &s->mdct_ctx[av_log2(subframe_len) - WMAPRO_BLOCK_MIN_BITS];
1260  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1261  int c = s->channel_indexes_for_cur_subframe[i];
1262  const int* sf = s->channel[c].scale_factors;
1263  int b;
1264 
1265  if (c == s->lfe_channel)
1266  memset(&s->tmp[cur_subwoofer_cutoff], 0, sizeof(*s->tmp) *
1267  (subframe_len - cur_subwoofer_cutoff));
1268 
1270  for (b = 0; b < s->num_bands; b++) {
1271  const int end = FFMIN(s->cur_sfb_offsets[b+1], s->subframe_len);
1272  const int exp = s->channel[c].quant_step -
1273  (s->channel[c].max_scale_factor - *sf++) *
1274  s->channel[c].scale_factor_step;
1275  const float quant = pow(10.0, exp / 20.0);
1276  int start = s->cur_sfb_offsets[b];
1277  s->fdsp.vector_fmul_scalar(s->tmp + start,
1278  s->channel[c].coeffs + start,
1279  quant, end - start);
1280  }
1281 
1283  mdct->imdct_half(mdct, s->channel[c].coeffs, s->tmp);
1284  }
1285  }
1286 
1288  wmapro_window(s);
1289 
1291  for (i = 0; i < s->channels_for_cur_subframe; i++) {
1292  int c = s->channel_indexes_for_cur_subframe[i];
1293  if (s->channel[c].cur_subframe >= s->channel[c].num_subframes) {
1294  av_log(s->avctx, AV_LOG_ERROR, "broken subframe\n");
1295  return AVERROR_INVALIDDATA;
1296  }
1297  ++s->channel[c].cur_subframe;
1298  }
1299 
1300  return 0;
1301 }
1302 
1309 static int decode_frame(WMAProDecodeCtx *s, AVFrame *frame, int *got_frame_ptr)
1310 {
1311  AVCodecContext *avctx = s->avctx;
1312  GetBitContext* gb = &s->gb;
1313  int more_frames = 0;
1314  int len = 0;
1315  int i, ret;
1316 
1318  if (s->len_prefix)
1319  len = get_bits(gb, s->log2_frame_size);
1320 
1321  av_dlog(s->avctx, "decoding frame with length %x\n", len);
1322 
1324  if (decode_tilehdr(s)) {
1325  s->packet_loss = 1;
1326  return 0;
1327  }
1328 
1330  if (s->avctx->channels > 1 && get_bits1(gb)) {
1331  if (get_bits1(gb)) {
1332  for (i = 0; i < avctx->channels * avctx->channels; i++)
1333  skip_bits(gb, 4);
1334  }
1335  }
1336 
1338  if (s->dynamic_range_compression) {
1339  s->drc_gain = get_bits(gb, 8);
1340  av_dlog(s->avctx, "drc_gain %i\n", s->drc_gain);
1341  }
1342 
1345  if (get_bits1(gb)) {
1346  int av_unused skip;
1347 
1349  if (get_bits1(gb)) {
1350  skip = get_bits(gb, av_log2(s->samples_per_frame * 2));
1351  av_dlog(s->avctx, "start skip: %i\n", skip);
1352  }
1353 
1355  if (get_bits1(gb)) {
1356  skip = get_bits(gb, av_log2(s->samples_per_frame * 2));
1357  av_dlog(s->avctx, "end skip: %i\n", skip);
1358  }
1359 
1360  }
1361 
1362  av_dlog(s->avctx, "BITSTREAM: frame header length was %i\n",
1363  get_bits_count(gb) - s->frame_offset);
1364 
1366  s->parsed_all_subframes = 0;
1367  for (i = 0; i < avctx->channels; i++) {
1368  s->channel[i].decoded_samples = 0;
1369  s->channel[i].cur_subframe = 0;
1370  s->channel[i].reuse_sf = 0;
1371  }
1372 
1374  while (!s->parsed_all_subframes) {
1375  if (decode_subframe(s) < 0) {
1376  s->packet_loss = 1;
1377  return 0;
1378  }
1379  }
1380 
1381  /* get output buffer */
1382  frame->nb_samples = s->samples_per_frame;
1383  if ((ret = ff_get_buffer(avctx, frame, 0)) < 0) {
1384  av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
1385  s->packet_loss = 1;
1386  return 0;
1387  }
1388 
1390  for (i = 0; i < avctx->channels; i++)
1391  memcpy(frame->extended_data[i], s->channel[i].out,
1392  s->samples_per_frame * sizeof(*s->channel[i].out));
1393 
1394  for (i = 0; i < avctx->channels; i++) {
1396  memcpy(&s->channel[i].out[0],
1397  &s->channel[i].out[s->samples_per_frame],
1398  s->samples_per_frame * sizeof(*s->channel[i].out) >> 1);
1399  }
1400 
1401  if (s->skip_frame) {
1402  s->skip_frame = 0;
1403  *got_frame_ptr = 0;
1404  av_frame_unref(frame);
1405  } else {
1406  *got_frame_ptr = 1;
1407  }
1408 
1409  if (s->len_prefix) {
1410  if (len != (get_bits_count(gb) - s->frame_offset) + 2) {
1413  "frame[%"PRIu32"] would have to skip %i bits\n",
1414  s->frame_num,
1415  len - (get_bits_count(gb) - s->frame_offset) - 1);
1416  s->packet_loss = 1;
1417  return 0;
1418  }
1419 
1421  skip_bits_long(gb, len - (get_bits_count(gb) - s->frame_offset) - 1);
1422  } else {
1423  while (get_bits_count(gb) < s->num_saved_bits && get_bits1(gb) == 0) {
1424  }
1425  }
1426 
1428  more_frames = get_bits1(gb);
1429 
1430  ++s->frame_num;
1431  return more_frames;
1432 }
1433 
1441 {
1442  return s->buf_bit_size - get_bits_count(gb);
1443 }
1444 
1452 static void save_bits(WMAProDecodeCtx *s, GetBitContext* gb, int len,
1453  int append)
1454 {
1455  int buflen;
1456 
1461  if (!append) {
1462  s->frame_offset = get_bits_count(gb) & 7;
1463  s->num_saved_bits = s->frame_offset;
1465  }
1466 
1467  buflen = (s->num_saved_bits + len + 8) >> 3;
1468 
1469  if (len <= 0 || buflen > MAX_FRAMESIZE) {
1470  avpriv_request_sample(s->avctx, "Too small input buffer");
1471  s->packet_loss = 1;
1472  return;
1473  }
1474 
1475  if (len > put_bits_left(&s->pb)) {
1477  "Cannot append %d bits, only %d bits available.\n",
1478  len, put_bits_left(&s->pb));
1479  s->packet_loss = 1;
1480  return;
1481  }
1482 
1483  s->num_saved_bits += len;
1484  if (!append) {
1485  avpriv_copy_bits(&s->pb, gb->buffer + (get_bits_count(gb) >> 3),
1486  s->num_saved_bits);
1487  } else {
1488  int align = 8 - (get_bits_count(gb) & 7);
1489  align = FFMIN(align, len);
1490  put_bits(&s->pb, align, get_bits(gb, align));
1491  len -= align;
1492  avpriv_copy_bits(&s->pb, gb->buffer + (get_bits_count(gb) >> 3), len);
1493  }
1494  skip_bits_long(gb, len);
1495 
1496  {
1497  PutBitContext tmp = s->pb;
1498  flush_put_bits(&tmp);
1499  }
1500 
1502  skip_bits(&s->gb, s->frame_offset);
1503 }
1504 
1512 static int decode_packet(AVCodecContext *avctx, void *data,
1513  int *got_frame_ptr, AVPacket* avpkt)
1514 {
1515  WMAProDecodeCtx *s = avctx->priv_data;
1516  GetBitContext* gb = &s->pgb;
1517  const uint8_t* buf = avpkt->data;
1518  int buf_size = avpkt->size;
1519  int num_bits_prev_frame;
1520  int packet_sequence_number;
1521 
1522  *got_frame_ptr = 0;
1523 
1524  if (s->packet_done || s->packet_loss) {
1525  s->packet_done = 0;
1526 
1528  if (buf_size < avctx->block_align) {
1529  av_log(avctx, AV_LOG_ERROR, "Input packet too small (%d < %d)\n",
1530  buf_size, avctx->block_align);
1531  return AVERROR_INVALIDDATA;
1532  }
1533 
1534  s->next_packet_start = buf_size - avctx->block_align;
1535  buf_size = avctx->block_align;
1536  s->buf_bit_size = buf_size << 3;
1537 
1539  init_get_bits(gb, buf, s->buf_bit_size);
1540  packet_sequence_number = get_bits(gb, 4);
1541  skip_bits(gb, 2);
1542 
1544  num_bits_prev_frame = get_bits(gb, s->log2_frame_size);
1545  av_dlog(avctx, "packet[%d]: nbpf %x\n", avctx->frame_number,
1546  num_bits_prev_frame);
1547 
1549  if (!s->packet_loss &&
1550  ((s->packet_sequence_number + 1) & 0xF) != packet_sequence_number) {
1551  s->packet_loss = 1;
1552  av_log(avctx, AV_LOG_ERROR,
1553  "Packet loss detected! seq %"PRIx8" vs %x\n",
1554  s->packet_sequence_number, packet_sequence_number);
1555  }
1556  s->packet_sequence_number = packet_sequence_number;
1557 
1558  if (num_bits_prev_frame > 0) {
1559  int remaining_packet_bits = s->buf_bit_size - get_bits_count(gb);
1560  if (num_bits_prev_frame >= remaining_packet_bits) {
1561  num_bits_prev_frame = remaining_packet_bits;
1562  s->packet_done = 1;
1563  }
1564 
1567  save_bits(s, gb, num_bits_prev_frame, 1);
1568  av_dlog(avctx, "accumulated %x bits of frame data\n",
1569  s->num_saved_bits - s->frame_offset);
1570 
1572  if (!s->packet_loss)
1573  decode_frame(s, data, got_frame_ptr);
1574  } else if (s->num_saved_bits - s->frame_offset) {
1575  av_dlog(avctx, "ignoring %x previously saved bits\n",
1576  s->num_saved_bits - s->frame_offset);
1577  }
1578 
1579  if (s->packet_loss) {
1583  s->num_saved_bits = 0;
1584  s->packet_loss = 0;
1585  }
1586 
1587  } else {
1588  int frame_size;
1589  s->buf_bit_size = (avpkt->size - s->next_packet_start) << 3;
1590  init_get_bits(gb, avpkt->data, s->buf_bit_size);
1591  skip_bits(gb, s->packet_offset);
1592  if (s->len_prefix && remaining_bits(s, gb) > s->log2_frame_size &&
1593  (frame_size = show_bits(gb, s->log2_frame_size)) &&
1594  frame_size <= remaining_bits(s, gb)) {
1595  save_bits(s, gb, frame_size, 0);
1596  s->packet_done = !decode_frame(s, data, got_frame_ptr);
1597  } else if (!s->len_prefix
1598  && s->num_saved_bits > get_bits_count(&s->gb)) {
1606  s->packet_done = !decode_frame(s, data, got_frame_ptr);
1607  } else
1608  s->packet_done = 1;
1609  }
1610 
1611  if (s->packet_done && !s->packet_loss &&
1612  remaining_bits(s, gb) > 0) {
1615  save_bits(s, gb, remaining_bits(s, gb), 0);
1616  }
1617 
1618  s->packet_offset = get_bits_count(gb) & 7;
1619  if (s->packet_loss)
1620  return AVERROR_INVALIDDATA;
1621 
1622  return get_bits_count(gb) >> 3;
1623 }
1624 
1629 static void flush(AVCodecContext *avctx)
1630 {
1631  WMAProDecodeCtx *s = avctx->priv_data;
1632  int i;
1635  for (i = 0; i < avctx->channels; i++)
1636  memset(s->channel[i].out, 0, s->samples_per_frame *
1637  sizeof(*s->channel[i].out));
1638  s->packet_loss = 1;
1639 }
1640 
1641 
1646  .name = "wmapro",
1647  .long_name = NULL_IF_CONFIG_SMALL("Windows Media Audio 9 Professional"),
1648  .type = AVMEDIA_TYPE_AUDIO,
1649  .id = AV_CODEC_ID_WMAPRO,
1650  .priv_data_size = sizeof(WMAProDecodeCtx),
1651  .init = decode_init,
1652  .close = decode_end,
1653  .decode = decode_packet,
1654  .capabilities = CODEC_CAP_SUBFRAMES | CODEC_CAP_DR1,
1655  .flush = flush,
1656  .sample_fmts = (const enum AVSampleFormat[]) { AV_SAMPLE_FMT_FLTP,
1658 };
float, planar
Definition: samplefmt.h:72
float * channel_data[WMAPRO_MAX_CHANNELS]
transformation coefficients
Definition: wmaprodec.c:165
uint8_t max_num_subframes
Definition: wmaprodec.c:190
static const uint16_t critical_freq[]
frequencies to divide the frequency spectrum into scale factor bands
Definition: wmaprodata.h:37
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:54
static int decode_tilehdr(WMAProDecodeCtx *s)
Decode how the data in the frame is split into subframes.
Definition: wmaprodec.c:535
This structure describes decoded (raw) audio or video data.
Definition: frame.h:135
#define VEC2MAXDEPTH
Definition: wmaprodec.c:119
int subframe_offset
subframe offset in the bit reservoir
Definition: wmaprodec.c:206
static const uint16_t vec1_huffcodes[HUFF_VEC1_SIZE]
Definition: wmaprodata.h:507
static const float coef0_level[HUFF_COEF0_SIZE]
Definition: wmaprodata.h:355
uint8_t table_idx
index for the num_sfb, sfb_offsets, sf_offsets and subwoofer_cutoffs tables
Definition: wmaprodec.c:225
static const uint32_t scale_rl_huffcodes[HUFF_SCALE_RL_SIZE]
Definition: wmaprodata.h:97
uint16_t num_vec_coeffs
number of vector coded coefficients
Definition: wmaprodec.c:153
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:240
#define WMAPRO_BLOCK_MIN_SIZE
minimum block size
Definition: wmaprodec.c:111
uint16_t min_samples_per_subframe
Definition: wmaprodec.c:193
static void skip_bits_long(GetBitContext *s, int n)
Definition: get_bits.h:199
AVCodecContext * avctx
codec context for av_log
Definition: wmaprodec.c:173
uint32_t decode_flags
used compression features
Definition: wmaprodec.c:183
#define FF_DEBUG_BITSTREAM
Definition: avcodec.h:2381
int size
Definition: avcodec.h:974
static const uint16_t vec4_huffcodes[HUFF_VEC4_SIZE]
Definition: wmaprodata.h:421
void avpriv_copy_bits(PutBitContext *pb, const uint8_t *src, int length)
Copy the content of src to the bitstream.
Definition: bitstream.c:61
int8_t scale_factor_step
scaling step for the current subframe
Definition: wmaprodec.c:146
const uint8_t * buffer
Definition: get_bits.h:54
#define DECLARE_ALIGNED(n, t, v)
Definition: mem.h:58
int ff_wma_run_level_decode(AVCodecContext *avctx, GetBitContext *gb, VLC *vlc, const float *level_table, const uint16_t *run_table, int version, WMACoef *ptr, int offset, int num_coefs, int block_len, int frame_len_bits, int coef_nb_bits)
Decode run level compressed coefficients.
Definition: wma.c:431
uint16_t log2_frame_size
Definition: wmaprodec.c:188
static const uint8_t scale_huffbits[HUFF_SCALE_SIZE]
Definition: wmaprodata.h:70
#define AV_RL16
Definition: intreadwrite.h:42
uint8_t table_idx
index in sf_offsets for the scale factor reference block
Definition: wmaprodec.c:151
#define FF_ARRAY_ELEMS(a)
av_dlog(ac->avr,"%d samples - audio_convert: %s to %s (%s)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt), use_generic?ac->func_descr_generic:ac->func_descr)
int16_t * cur_sfb_offsets
sfb offsets for the current block
Definition: wmaprodec.c:224
#define VLCBITS
Definition: wmaprodec.c:116
uint8_t run
Definition: svq3.c:146
WMAProChannelGrp chgroup[WMAPRO_MAX_CHANNELS]
channel group information
Definition: wmaprodec.c:229
PutBitContext pb
context for filling the frame_data buffer
Definition: wmaprodec.c:177
#define SCALEVLCBITS
Definition: wmaprodec.c:117
AVCodec.
Definition: avcodec.h:2812
#define WMAPRO_MAX_CHANNELS
current decoder limitations
Definition: wmaprodec.c:104
int block_align
number of bytes per packet if constant and known or 0 Used by some WAV based audio codecs...
Definition: avcodec.h:1844
#define AV_WN32A(p, v)
Definition: intreadwrite.h:458
static int get_sbits(GetBitContext *s, int n)
Definition: get_bits.h:226
static VLC vec1_vlc
1 coefficient per symbol
Definition: wmaprodec.c:128
static av_cold void dump_context(WMAProDecodeCtx *s)
helper function to print the most important members of the context
Definition: wmaprodec.c:239
static const uint8_t scale_rl_run[HUFF_SCALE_RL_SIZE]
Definition: wmaprodata.h:140
static int decode_packet(AVCodecContext *avctx, void *data, int *got_frame_ptr, AVPacket *avpkt)
Decode a single WMA packet.
Definition: wmaprodec.c:1512
int frame_offset
frame offset in the bit reservoir
Definition: wmaprodec.c:205
static int decode(MimicContext *ctx, int quality, int num_coeffs, int is_iframe)
Definition: mimic.c:275
void void avpriv_request_sample(void *avc, const char *msg,...) av_printf_format(2
Log a generic warning message about a missing feature.
void(* vector_fmul_window)(float *dst, const float *src0, const float *src1, const float *win, int len)
Overlap/add with window function.
Definition: float_dsp.h:103
static const float coef1_level[HUFF_COEF1_SIZE]
Definition: wmaprodata.h:396
uint16_t subframe_offset[MAX_SUBFRAMES]
subframe positions in the current frame
Definition: wmaprodec.c:140
uint8_t bits
Definition: crc.c:251
enum AVSampleFormat sample_fmt
audio sample format
Definition: avcodec.h:1815
static void inverse_channel_transform(WMAProDecodeCtx *s)
Reconstruct the individual channel data.
Definition: wmaprodec.c:986
int16_t sfb_offsets[WMAPRO_BLOCK_SIZES][MAX_BANDS]
scale factor band offsets (multiples of 4)
Definition: wmaprodec.c:195
uint8_t
#define av_cold
Definition: attributes.h:66
#define HUFF_COEF1_SIZE
Definition: wmaprodata.h:253
#define HUFF_VEC2_SIZE
Definition: wmaprodata.h:460
#define INIT_VLC_STATIC(vlc, bits, a, b, c, d, e, f, g, static_size)
Definition: get_bits.h:443
int8_t num_bands
number of scale factor bands
Definition: wmaprodec.c:222
#define SCALEMAXDEPTH
Definition: wmaprodec.c:121
uint8_t frame_data[MAX_FRAMESIZE+FF_INPUT_BUFFER_PADDING_SIZE]
compressed frame data
Definition: wmaprodec.c:176
#define b
Definition: input.c:52
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1164
uint8_t packet_sequence_number
current packet number
Definition: wmaprodec.c:203
uint8_t drc_gain
gain for the DRC tool
Definition: wmaprodec.c:214
uint8_t dynamic_range_compression
frame contains DRC data
Definition: wmaprodec.c:185
#define CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:684
const char data[16]
Definition: mxf.c:70
static uint8_t * append(uint8_t *buf, const uint8_t *src, int size)
#define MAX_FRAMESIZE
maximum compressed frame size
Definition: wmaprodec.c:107
int16_t prev_block_len
length of the previous block
Definition: wmaprodec.c:136
WMAProChannelCtx channel[WMAPRO_MAX_CHANNELS]
per channel data
Definition: wmaprodec.c:231
GetBitContext gb
bitstream reader context
Definition: wmaprodec.c:212
uint8_t * data
Definition: avcodec.h:973
uint8_t grouped
channel is part of a group
Definition: wmaprodec.c:143
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:194
#define WMAPRO_BLOCK_MAX_BITS
log2 of max block size
Definition: wmaprodec.c:110
#define HUFF_COEF0_SIZE
Definition: wmaprodata.h:166
int * scale_factors
pointer to the scale factor values used for decoding
Definition: wmaprodec.c:150
bitstream reader API header.
#define HUFF_VEC1_SIZE
Definition: wmaprodata.h:505
#define CODEC_FLAG_BITEXACT
Use only bitexact stuff (except (I)DCT).
Definition: avcodec.h:658
int next_packet_start
start offset of the next wma packet in the demuxer packet
Definition: wmaprodec.c:201
int num_saved_bits
saved number of bits
Definition: wmaprodec.c:204
static const uint16_t scale_huffcodes[HUFF_SCALE_SIZE]
Definition: wmaprodata.h:51
int buf_bit_size
buffer size in bits
Definition: wmaprodec.c:213
static const uint32_t coef1_huffcodes[555]
Definition: wmadata.h:269
#define SCALERLMAXDEPTH
Definition: wmaprodec.c:122
static VLC vec4_vlc
4 coefficients per symbol
Definition: wmaprodec.c:126
static const uint8_t frame_size[4]
Definition: g723_1_data.h:47
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:123
static int decode_subframe(WMAProDecodeCtx *s)
Decode a single subframe (block).
Definition: wmaprodec.c:1071
uint8_t packet_loss
set in case of bitstream error
Definition: wmaprodec.c:207
static int put_bits_left(PutBitContext *s)
Definition: put_bits.h:75
static const uint16_t mask[17]
Definition: lzw.c:38
#define AVERROR(e)
Definition: error.h:43
#define MAX_SUBFRAMES
max number of subframes per channel
Definition: wmaprodec.c:105
#define HUFF_VEC4_SIZE
Definition: wmaprodata.h:419
sample_fmts
Definition: avconv_filter.c:68
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:145
int8_t channel_indexes_for_cur_subframe[WMAPRO_MAX_CHANNELS]
Definition: wmaprodec.c:221
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:144
int flags
CODEC_FLAG_*.
Definition: avcodec.h:1144
static const uint8_t coef0_huffbits[666]
Definition: wmadata.h:182
static float sin64[33]
sine table for decorrelation
Definition: wmaprodec.c:130
void av_log(void *avcl, int level, const char *fmt,...)
Definition: log.c:169
#define HUFF_SCALE_SIZE
Definition: wmaprodata.h:49
int max_scale_factor
maximum scale factor for the current subframe
Definition: wmaprodec.c:147
const char * name
Name of the codec implementation.
Definition: avcodec.h:2819
static void put_bits(PutBitContext *s, int n, unsigned int value)
Write up to 31 bits into a bitstream.
Definition: put_bits.h:134
#define ff_mdct_init
Definition: fft.h:151
int8_t channels_for_cur_subframe
number of channels that contain the subframe
Definition: wmaprodec.c:220
AVCodec ff_wmapro_decoder
wmapro decoder
Definition: wmaprodec.c:1645
#define FFMAX(a, b)
Definition: common.h:55
static int decode_subframe_length(WMAProDecodeCtx *s, int offset)
Decode the subframe length.
Definition: wmaprodec.c:487
static const uint8_t coef1_huffbits[555]
Definition: wmadata.h:342
float out[WMAPRO_BLOCK_MAX_SIZE+WMAPRO_BLOCK_MAX_SIZE/2]
output buffer
Definition: wmaprodec.c:154
int quant_step
quantization step for the current subframe
Definition: wmaprodec.c:144
static const uint16_t coef1_run[HUFF_COEF1_SIZE]
Definition: wmaprodata.h:379
Definition: get_bits.h:64
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:1868
frame specific decoder context for a single channel
Definition: wmaprodec.c:135
static VLC sf_vlc
scale factor DPCM vlc
Definition: wmaprodec.c:124
uint8_t bits_per_sample
integer audio sample size for the unscaled IMDCT output (used to scale to [-1.0, 1.0])
Definition: wmaprodec.c:186
#define FF_INPUT_BUFFER_PADDING_SIZE
Required number of additionally allocated bytes at the end of the input bitstream for decoding...
Definition: avcodec.h:531
static const uint8_t scale_rl_huffbits[HUFF_SCALE_RL_SIZE]
Definition: wmaprodata.h:118
static void flush(AVCodecContext *avctx)
Clear decoder buffers (for seeking).
Definition: wmaprodec.c:1629
static const uint16_t symbol_to_vec4[HUFF_VEC4_SIZE]
Definition: wmaprodata.h:540
static int decode_coeffs(WMAProDecodeCtx *s, int c)
Extract the coefficients from the bitstream.
Definition: wmaprodec.c:790
uint8_t num_subframes
Definition: wmaprodec.c:138
Definition: fft.h:73
#define FFMIN(a, b)
Definition: common.h:57
uint32_t frame_num
current frame number (not used for decoding)
Definition: wmaprodec.c:211
int8_t transform
transform on / off
Definition: wmaprodec.c:162
float tmp[WMAPRO_BLOCK_MAX_SIZE]
IMDCT output buffer.
Definition: wmaprodec.c:179
main decoder context
Definition: wmaprodec.c:171
int8_t sf_offsets[WMAPRO_BLOCK_SIZES][WMAPRO_BLOCK_SIZES][MAX_BANDS]
scale factor resample matrix
Definition: wmaprodec.c:196
static unsigned int show_bits(GetBitContext *s, int n)
Show 1-25 bits.
Definition: get_bits.h:254
uint16_t decoded_samples
number of already processed samples
Definition: wmaprodec.c:142
static const float *const default_decorrelation[]
default decorrelation matrix offsets
Definition: wmaprodata.h:594
static void save_bits(WMAProDecodeCtx *s, GetBitContext *gb, int len, int append)
Fill the bit reservoir with a (partial) frame.
Definition: wmaprodec.c:1452
static av_always_inline int get_vlc2(GetBitContext *s, VLC_TYPE(*table)[2], int bits, int max_depth)
Parse a vlc code.
Definition: get_bits.h:522
int8_t transmit_num_vec_coeffs
number of vector coded coefficients is part of the bitstream
Definition: wmaprodec.c:223
void(* vector_fmul_scalar)(float *dst, const float *src, float mul, int len)
Multiply a vector of floats by a scalar float.
Definition: float_dsp.h:69
#define AV_RL32
Definition: intreadwrite.h:146
#define MAX_BANDS
max number of scale factor bands
Definition: wmaprodec.c:106
uint8_t subframe_len_bits
number of bits used for the subframe length
Definition: wmaprodec.c:191
int8_t transform_band[MAX_BANDS]
controls if the transform is enabled for a certain band
Definition: wmaprodec.c:163
if(ac->has_optimized_func)
#define AVERROR_PATCHWELCOME
Not yet implemented in Libav, patches welcome.
Definition: error.h:57
int8_t lfe_channel
lfe channel index
Definition: wmaprodec.c:189
int16_t subwoofer_cutoffs[WMAPRO_BLOCK_SIZES]
subwoofer cutoff values
Definition: wmaprodec.c:197
static VLC vec2_vlc
2 coefficients per symbol
Definition: wmaprodec.c:127
int8_t parsed_all_subframes
all subframes decoded?
Definition: wmaprodec.c:216
Libavcodec external API header.
channel group for channel transformations
Definition: wmaprodec.c:160
AVSampleFormat
Audio Sample Formats.
Definition: samplefmt.h:61
#define VEC1MAXDEPTH
Definition: wmaprodec.c:120
AV_SAMPLE_FMT_NONE
Definition: avconv_filter.c:68
int16_t subframe_len
current subframe length
Definition: wmaprodec.c:219
int sample_rate
samples per second
Definition: avcodec.h:1807
#define WMAPRO_BLOCK_SIZES
possible block sizes
Definition: wmaprodec.c:113
int debug
debug
Definition: avcodec.h:2378
static const uint8_t vec4_huffbits[HUFF_VEC4_SIZE]
Definition: wmaprodata.h:440
main external API structure.
Definition: avcodec.h:1050
static void close(AVCodecParserContext *s)
Definition: h264_parser.c:490
static const uint8_t symbol_to_vec2[HUFF_VEC2_SIZE]
Definition: wmaprodata.h:557
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: utils.c:621
tables for wmapro decoding
#define VEC4MAXDEPTH
Definition: wmaprodec.c:118
void(* imdct_half)(struct FFTContext *s, FFTSample *output, const FFTSample *input)
Definition: fft.h:93
int extradata_size
Definition: avcodec.h:1165
static VLC coef_vlc[2]
coefficient run length vlc codes
Definition: wmaprodec.c:129
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:271
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:263
int8_t reuse_sf
share scale factors between subframes
Definition: wmaprodec.c:145
SINETABLE_CONST float *const ff_sine_windows[14]
static int init_get_bits(GetBitContext *s, const uint8_t *buffer, int bit_size)
Initialize GetBitContext.
Definition: get_bits.h:375
uint8_t packet_offset
frame offset in the packet
Definition: wmaprodec.c:202
static int decode_scale_factors(WMAProDecodeCtx *s)
Extract scale factors from the bitstream.
Definition: wmaprodec.c:893
int saved_scale_factors[2][MAX_BANDS]
resampled and (previously) transmitted scale factor values
Definition: wmaprodec.c:148
uint8_t num_channels
number of channels in the group
Definition: wmaprodec.c:161
uint8_t transmit_coefs
Definition: wmaprodec.c:137
static VLC sf_rl_vlc
scale factor run length vlc
Definition: wmaprodec.c:125
float * coeffs
pointer to the subframe decode buffer
Definition: wmaprodec.c:152
static int step
Definition: avplay.c:247
float * windows[WMAPRO_BLOCK_SIZES]
windows for the different block sizes
Definition: wmaprodec.c:180
uint8_t cur_subframe
current subframe number
Definition: wmaprodec.c:141
static av_always_inline uint32_t av_float2int(float f)
Reinterpret a float as a 32-bit integer.
Definition: intfloat.h:50
static int remaining_bits(WMAProDecodeCtx *s, GetBitContext *gb)
Calculate remaining input buffer length.
Definition: wmaprodec.c:1440
const uint8_t * quant
void av_frame_unref(AVFrame *frame)
Unreference all the buffers referenced by frame and reset the frame fields.
Definition: frame.c:283
static int decode_channel_transform(WMAProDecodeCtx *s)
Decode channel transformation parameters.
Definition: wmaprodec.c:676
av_cold int ff_wma_get_frame_len_bits(int sample_rate, int version, unsigned int decode_flags)
Get the samples per frame for this stream.
Definition: wma_common.c:31
FFTContext mdct_ctx[WMAPRO_BLOCK_SIZES]
MDCT context per block size.
Definition: wmaprodec.c:178
av_cold void avpriv_float_dsp_init(AVFloatDSPContext *fdsp, int bit_exact)
Initialize a float DSP context.
Definition: float_dsp.c:115
int8_t scale_factor_idx
index for the transmitted scale factor values (used for resampling)
Definition: wmaprodec.c:149
uint8_t level
Definition: svq3.c:147
#define PRINT(a, b)
#define WMAPRO_BLOCK_MAX_SIZE
maximum block size
Definition: wmaprodec.c:112
static av_cold int decode_init(AVCodecContext *avctx)
Initialize the decoder.
Definition: wmaprodec.c:274
#define PRINT_HEX(a, b)
uint16_t subframe_len[MAX_SUBFRAMES]
subframe length in samples
Definition: wmaprodec.c:139
uint8_t pi<< 24) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0f/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8, uint8_t,(*(constuint8_t *) pi-0x80)*(1.0/(1<< 7))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16, int16_t,(*(constint16_t *) pi >>8)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, int16_t,*(constint16_t *) pi *(1.0/(1<< 15))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32, int32_t,(*(constint32_t *) pi >>24)+0x80) CONV_FUNC_GROUP(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, int32_t,*(constint32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, float, av_clip_uint8(lrintf(*(constfloat *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, float, av_clip_int16(lrintf(*(constfloat *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, float, av_clipl_int32(llrintf(*(constfloat *) pi *(1U<< 31)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, double, av_clip_uint8(lrint(*(constdouble *) pi *(1<< 7))+0x80)) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, double, av_clip_int16(lrint(*(constdouble *) pi *(1<< 15)))) CONV_FUNC_GROUP(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, double, av_clipl_int32(llrint(*(constdouble *) pi *(1U<< 31))))#defineSET_CONV_FUNC_GROUP(ofmt, ifmt) staticvoidset_generic_function(AudioConvert *ac){}voidff_audio_convert_free(AudioConvert **ac){if(!*ac) return;ff_dither_free(&(*ac) ->dc);av_freep(ac);}AudioConvert *ff_audio_convert_alloc(AVAudioResampleContext *avr, enumAVSampleFormatout_fmt, enumAVSampleFormatin_fmt, intchannels, intsample_rate, intapply_map){AudioConvert *ac;intin_planar, out_planar;ac=av_mallocz(sizeof(*ac));if(!ac) returnNULL;ac->avr=avr;ac->out_fmt=out_fmt;ac->in_fmt=in_fmt;ac->channels=channels;ac->apply_map=apply_map;if(avr->dither_method!=AV_RESAMPLE_DITHER_NONE &&av_get_packed_sample_fmt(out_fmt)==AV_SAMPLE_FMT_S16 &&av_get_bytes_per_sample(in_fmt)>2){ac->dc=ff_dither_alloc(avr, out_fmt, in_fmt, channels, sample_rate, apply_map);if(!ac->dc){av_free(ac);returnNULL;}returnac;}in_planar=ff_sample_fmt_is_planar(in_fmt, channels);out_planar=ff_sample_fmt_is_planar(out_fmt, channels);if(in_planar==out_planar){ac->func_type=CONV_FUNC_TYPE_FLAT;ac->planes=in_planar?ac->channels:1;}elseif(in_planar) ac->func_type=CONV_FUNC_TYPE_INTERLEAVE;elseac->func_type=CONV_FUNC_TYPE_DEINTERLEAVE;set_generic_function(ac);if(ARCH_AARCH64) ff_audio_convert_init_aarch64(ac);if(ARCH_ARM) ff_audio_convert_init_arm(ac);if(ARCH_X86) ff_audio_convert_init_x86(ac);returnac;}intff_audio_convert(AudioConvert *ac, AudioData *out, AudioData *in){intuse_generic=1;intlen=in->nb_samples;intp;if(ac->dc){av_dlog(ac->avr,"%dsamples-audio_convert:%sto%s(dithered)\n", len, av_get_sample_fmt_name(ac->in_fmt), av_get_sample_fmt_name(ac->out_fmt));returnff_convert_dither(ac-> out
#define CODEC_CAP_SUBFRAMES
Codec can output multiple frames per AVPacket Normally demuxers return one frame at a time...
Definition: avcodec.h:736
int8_t num_sfb[WMAPRO_BLOCK_SIZES]
scale factor bands per block size
Definition: wmaprodec.c:194
common internal api header.
static void decode_decorrelation_matrix(WMAProDecodeCtx *s, WMAProChannelGrp *chgroup)
Calculate a decorrelation matrix from the bitstream parameters.
Definition: wmaprodec.c:626
static void flush_put_bits(PutBitContext *s)
Pad the end of the output stream with zeros.
Definition: put_bits.h:83
#define ff_mdct_end
Definition: fft.h:152
static void init_put_bits(PutBitContext *s, uint8_t *buffer, int buffer_size)
Initialize the PutBitContext s.
Definition: put_bits.h:48
static av_cold int init(AVCodecParserContext *s)
Definition: h264_parser.c:499
static void wmapro_window(WMAProDecodeCtx *s)
Apply sine window and reconstruct the output buffer.
Definition: wmaprodec.c:1041
#define WMAPRO_BLOCK_MIN_BITS
log2 of min block size
Definition: wmaprodec.c:109
static const uint8_t scale_rl_level[HUFF_SCALE_RL_SIZE]
Definition: wmaprodata.h:150
void * priv_data
Definition: avcodec.h:1092
int len
GetBitContext pgb
bitstream reader context for the packet
Definition: wmaprodec.c:200
int channels
number of audio channels
Definition: avcodec.h:1808
#define av_log2
Definition: intmath.h:85
VLC_TYPE(* table)[2]
code, bits
Definition: get_bits.h:66
unsigned int ff_wma_get_large_val(GetBitContext *gb)
Decode an uncompressed coefficient.
Definition: wma.c:398
static const uint32_t coef0_huffcodes[666]
Definition: wmadata.h:95
static int decode_frame(WMAProDecodeCtx *s, AVFrame *frame, int *got_frame_ptr)
Decode one WMA frame.
Definition: wmaprodec.c:1309
int frame_number
Frame counter, set by libavcodec.
Definition: avcodec.h:1838
AVFloatDSPContext fdsp
Definition: wmaprodec.c:174
uint16_t samples_per_frame
number of samples to output
Definition: wmaprodec.c:187
static const uint8_t vec1_huffbits[HUFF_VEC1_SIZE]
Definition: wmaprodata.h:523
static const uint16_t coef0_run[HUFF_COEF0_SIZE]
Definition: wmaprodata.h:332
static const uint16_t vec2_huffcodes[HUFF_VEC2_SIZE]
Definition: wmaprodata.h:462
int8_t esc_len
length of escaped coefficients
Definition: wmaprodec.c:226
uint8_t len_prefix
frame is prefixed with its length
Definition: wmaprodec.c:184
float decorrelation_matrix[WMAPRO_MAX_CHANNELS *WMAPRO_MAX_CHANNELS]
Definition: wmaprodec.c:164
uint8_t ** extended_data
pointers to the data planes/channels.
Definition: frame.h:169
static av_cold int decode_end(AVCodecContext *avctx)
Uninitialize the decoder and free all resources.
Definition: wmaprodec.c:258
This structure stores compressed data.
Definition: avcodec.h:950
#define HUFF_SCALE_RL_SIZE
Definition: wmaprodata.h:95
uint8_t max_subframe_len_bit
flag indicating that the subframe is of maximum size when the first subframe length bit is 1 ...
Definition: wmaprodec.c:192
int nb_samples
number of audio samples (per channel) described by this frame
Definition: frame.h:179
int8_t skip_frame
skip output step
Definition: wmaprodec.c:215
void ff_init_ff_sine_windows(int index)
initialize the specified entry of ff_sine_windows
#define av_unused
Definition: attributes.h:86
static const uint8_t vec2_huffbits[HUFF_VEC2_SIZE]
Definition: wmaprodata.h:483
uint8_t num_chgroups
number of channel groups
Definition: wmaprodec.c:228
uint8_t packet_done
set when a packet is fully decoded
Definition: wmaprodec.c:208
bitstream writer API