FFmpeg  4.1.4
atrac9dec.c
Go to the documentation of this file.
1 /*
2  * ATRAC9 decoder
3  * Copyright (c) 2018 Rostislav Pehlivanov <atomnuker@gmail.com>
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "internal.h"
23 #include "get_bits.h"
24 #include "fft.h"
25 #include "atrac9tab.h"
26 #include "libavutil/lfg.h"
27 #include "libavutil/float_dsp.h"
28 
29 typedef struct ATRAC9ChannelData {
30  int band_ext;
32  int band_ext_data[4];
35 
37  int precision_fine[30];
38  int precision_mask[30];
39 
40  int codebookset[30];
41 
44 
45  DECLARE_ALIGNED(32, float, coeffs )[256];
46  DECLARE_ALIGNED(32, float, prev_win)[128];
48 
49 typedef struct ATRAC9BlockData {
51 
52  /* Base */
56 
57  /* Stereo block only */
59 
60  /* Band extension only */
64 
65  /* Gradient */
66  int grad_mode;
68  int gradient[31];
69 
70  /* Stereo */
72  int is_signs[30];
73 
74  int reuseable;
75 
77 
78 typedef struct ATRAC9Context {
84 
85  /* Set on init */
91 
92  /* Generated on init */
93  VLC sf_vlc[2][8]; /* Signed/unsigned, length */
94  VLC coeff_vlc[2][8][4]; /* Cookbook, precision, cookbook index */
95  uint8_t alloc_curve[48][48];
96  DECLARE_ALIGNED(32, float, imdct_win)[256];
97 
98  DECLARE_ALIGNED(32, float, temp)[256];
100 
102  GetBitContext *gb)
103 {
104  int grad_range[2];
105  int grad_value[2];
106  int values, sign, base;
107  uint8_t *curve;
108  float scale;
109 
110  b->grad_mode = get_bits(gb, 2);
111  if (b->grad_mode) {
112  grad_range[0] = get_bits(gb, 5);
113  grad_range[1] = 31;
114  grad_value[0] = get_bits(gb, 5);
115  grad_value[1] = 31;
116  } else {
117  grad_range[0] = get_bits(gb, 6);
118  grad_range[1] = get_bits(gb, 6) + 1;
119  grad_value[0] = get_bits(gb, 5);
120  grad_value[1] = get_bits(gb, 5);
121  }
122  b->grad_boundary = get_bits(gb, 4);
123 
124  if (grad_range[0] >= grad_range[1] || grad_range[1] > 47)
125  return AVERROR_INVALIDDATA;
126 
127  if (grad_value[0] > 31 || grad_value[1] > 31)
128  return AVERROR_INVALIDDATA;
129 
130  if (b->grad_boundary > b->q_unit_cnt)
131  return AVERROR_INVALIDDATA;
132 
133  values = grad_value[1] - grad_value[0];
134  sign = 1 - 2*(values < 0);
135  base = grad_value[0] + sign;
136  scale = (FFABS(values) - 1) / 31.0f;
137  curve = s->alloc_curve[grad_range[1] - grad_range[0] - 1];
138 
139  for (int i = 0; i <= b->q_unit_cnt; i++)
140  b->gradient[i] = grad_value[i >= grad_range[0]];
141 
142  for (int i = grad_range[0]; i < grad_range[1]; i++)
143  b->gradient[i] = base + sign*((int)(scale*curve[i - grad_range[0]]));
144 
145  return 0;
146 }
147 
150 {
151  memset(c->precision_mask, 0, sizeof(c->precision_mask));
152  for (int i = 1; i < b->q_unit_cnt; i++) {
153  const int delta = FFABS(c->scalefactors[i] - c->scalefactors[i - 1]) - 1;
154  if (delta > 0) {
155  const int neg = c->scalefactors[i - 1] > c->scalefactors[i];
156  c->precision_mask[i - neg] += FFMIN(delta, 5);
157  }
158  }
159 
160  if (b->grad_mode) {
161  for (int i = 0; i < b->q_unit_cnt; i++) {
162  c->precision_coarse[i] = c->scalefactors[i];
163  c->precision_coarse[i] += c->precision_mask[i] - b->gradient[i];
164  if (c->precision_coarse[i] < 0)
165  continue;
166  switch (b->grad_mode) {
167  case 1:
168  c->precision_coarse[i] >>= 1;
169  break;
170  case 2:
171  c->precision_coarse[i] = (3 * c->precision_coarse[i]) >> 3;
172  break;
173  case 3:
174  c->precision_coarse[i] >>= 2;
175  break;
176  }
177  }
178  } else {
179  for (int i = 0; i < b->q_unit_cnt; i++)
180  c->precision_coarse[i] = c->scalefactors[i] - b->gradient[i];
181  }
182 
183 
184  for (int i = 0; i < b->q_unit_cnt; i++)
185  c->precision_coarse[i] = FFMAX(c->precision_coarse[i], 1);
186 
187  for (int i = 0; i < b->grad_boundary; i++)
188  c->precision_coarse[i]++;
189 
190  for (int i = 0; i < b->q_unit_cnt; i++) {
191  c->precision_fine[i] = 0;
192  if (c->precision_coarse[i] > 15) {
193  c->precision_fine[i] = c->precision_coarse[i] - 15;
194  c->precision_coarse[i] = 15;
195  }
196  }
197 }
198 
200  GetBitContext *gb, int stereo)
201 {
202  int ext_band = 0;
203 
204  if (b->has_band_ext) {
205  if (b->q_unit_cnt < 13)
206  return AVERROR_INVALIDDATA;
207  ext_band = at9_tab_band_ext_group[b->q_unit_cnt - 13][2];
208  if (stereo) {
209  b->channel[1].band_ext = get_bits(gb, 2);
210  b->channel[1].band_ext = ext_band > 2 ? b->channel[1].band_ext : 4;
211  } else {
212  skip_bits1(gb);
213  }
214  }
215 
216  b->has_band_ext_data = get_bits1(gb);
217  if (!b->has_band_ext_data)
218  return 0;
219 
220  if (!b->has_band_ext) {
221  skip_bits(gb, 2);
222  skip_bits_long(gb, get_bits(gb, 5));
223  return 0;
224  }
225 
226  b->channel[0].band_ext = get_bits(gb, 2);
227  b->channel[0].band_ext = ext_band > 2 ? b->channel[0].band_ext : 4;
228 
229  if (!get_bits(gb, 5))
230  return 0;
231 
232  for (int i = 0; i <= stereo; i++) {
233  ATRAC9ChannelData *c = &b->channel[i];
234  const int count = at9_tab_band_ext_cnt[c->band_ext][ext_band];
235  for (int j = 0; j < count; j++) {
236  int len = at9_tab_band_ext_lengths[c->band_ext][ext_band][j];
237  c->band_ext_data[j] = get_bits(gb, len);
238  }
239  }
240 
241  return 0;
242 }
243 
246  int channel_idx, int first_in_pkt)
247 {
248  static const int mode_map[2][4] = { { 0, 1, 2, 3 }, { 0, 2, 3, 4 } };
249  const int mode = mode_map[channel_idx][get_bits(gb, 2)];
250 
251  memset(c->scalefactors, 0, sizeof(c->scalefactors));
252 
253  if (first_in_pkt && (mode == 4 || ((mode == 3) && !channel_idx))) {
254  av_log(s->avctx, AV_LOG_ERROR, "Invalid scalefactor coding mode!\n");
255  return AVERROR_INVALIDDATA;
256  }
257 
258  switch (mode) {
259  case 0: { /* VLC delta offset */
260  const uint8_t *sf_weights = at9_tab_sf_weights[get_bits(gb, 3)];
261  const int base = get_bits(gb, 5);
262  const int len = get_bits(gb, 2) + 3;
263  const VLC *tab = &s->sf_vlc[0][len];
264 
265  c->scalefactors[0] = get_bits(gb, len);
266 
267  for (int i = 1; i < b->band_ext_q_unit; i++) {
268  int val = c->scalefactors[i - 1] + get_vlc2(gb, tab->table, 9, 2);
269  c->scalefactors[i] = val & ((1 << len) - 1);
270  }
271 
272  for (int i = 0; i < b->band_ext_q_unit; i++)
273  c->scalefactors[i] += base - sf_weights[i];
274 
275  break;
276  }
277  case 1: { /* CLC offset */
278  const int len = get_bits(gb, 2) + 2;
279  const int base = len < 5 ? get_bits(gb, 5) : 0;
280  for (int i = 0; i < b->band_ext_q_unit; i++)
281  c->scalefactors[i] = base + get_bits(gb, len);
282  break;
283  }
284  case 2:
285  case 4: { /* VLC dist to baseline */
286  const int *baseline = mode == 4 ? c->scalefactors_prev :
287  channel_idx ? b->channel[0].scalefactors :
289  const int baseline_len = mode == 4 ? b->q_unit_cnt_prev :
290  channel_idx ? b->band_ext_q_unit :
291  b->q_unit_cnt_prev;
292 
293  const int len = get_bits(gb, 2) + 2;
294  const int unit_cnt = FFMIN(b->band_ext_q_unit, baseline_len);
295  const VLC *tab = &s->sf_vlc[1][len];
296 
297  for (int i = 0; i < unit_cnt; i++) {
298  int dist = get_vlc2(gb, tab->table, 9, 2);
299  c->scalefactors[i] = baseline[i] + dist;
300  }
301 
302  for (int i = unit_cnt; i < b->band_ext_q_unit; i++)
303  c->scalefactors[i] = get_bits(gb, 5);
304 
305  break;
306  }
307  case 3: { /* VLC offset with baseline */
308  const int *baseline = channel_idx ? b->channel[0].scalefactors :
310  const int baseline_len = channel_idx ? b->band_ext_q_unit :
311  b->q_unit_cnt_prev;
312 
313  const int base = get_bits(gb, 5) - (1 << (5 - 1));
314  const int len = get_bits(gb, 2) + 1;
315  const int unit_cnt = FFMIN(b->band_ext_q_unit, baseline_len);
316  const VLC *tab = &s->sf_vlc[0][len];
317 
318  c->scalefactors[0] = get_bits(gb, len);
319 
320  for (int i = 1; i < unit_cnt; i++) {
321  int val = c->scalefactors[i - 1] + get_vlc2(gb, tab->table, 9, 2);
322  c->scalefactors[i] = val & ((1 << len) - 1);
323  }
324 
325  for (int i = 0; i < unit_cnt; i++)
326  c->scalefactors[i] += base + baseline[i];
327 
328  for (int i = unit_cnt; i < b->band_ext_q_unit; i++)
329  c->scalefactors[i] = get_bits(gb, 5);
330  break;
331  }
332  }
333 
334  for (int i = 0; i < b->band_ext_q_unit; i++)
335  if (c->scalefactors[i] < 0 || c->scalefactors[i] > 31)
336  return AVERROR_INVALIDDATA;
337 
338  memcpy(c->scalefactors_prev, c->scalefactors, sizeof(c->scalefactors));
339 
340  return 0;
341 }
342 
345 {
346  int avg = 0;
347  const int last_sf = c->scalefactors[c->q_unit_cnt];
348 
349  memset(c->codebookset, 0, sizeof(c->codebookset));
350 
351  if (c->q_unit_cnt <= 1)
352  return;
353  if (s->samplerate_idx > 7)
354  return;
355 
356  c->scalefactors[c->q_unit_cnt] = c->scalefactors[c->q_unit_cnt - 1];
357 
358  if (c->q_unit_cnt > 12) {
359  for (int i = 0; i < 12; i++)
360  avg += c->scalefactors[i];
361  avg = (avg + 6) / 12;
362  }
363 
364  for (int i = 8; i < c->q_unit_cnt; i++) {
365  const int prev = c->scalefactors[i - 1];
366  const int cur = c->scalefactors[i ];
367  const int next = c->scalefactors[i + 1];
368  const int min = FFMIN(prev, next);
369  if ((cur - min >= 3 || 2*cur - prev - next >= 3))
370  c->codebookset[i] = 1;
371  }
372 
373 
374  for (int i = 12; i < c->q_unit_cnt; i++) {
375  const int cur = c->scalefactors[i];
376  const int cnd = at9_q_unit_to_coeff_cnt[i] == 16;
377  const int min = FFMIN(c->scalefactors[i + 1], c->scalefactors[i - 1]);
378  if (c->codebookset[i])
379  continue;
380 
381  c->codebookset[i] = (((cur - min) >= 2) && (cur >= (avg - cnd)));
382  }
383 
384  c->scalefactors[c->q_unit_cnt] = last_sf;
385 }
386 
389 {
390  const int max_prec = s->samplerate_idx > 7 ? 1 : 7;
391 
392  memset(c->q_coeffs_coarse, 0, sizeof(c->q_coeffs_coarse));
393 
394  for (int i = 0; i < c->q_unit_cnt; i++) {
396  const int bands = at9_q_unit_to_coeff_cnt[i];
397  const int prec = c->precision_coarse[i] + 1;
398 
399  if (prec <= max_prec) {
400  const int cb = c->codebookset[i];
401  const int cbi = at9_q_unit_to_codebookidx[i];
402  const VLC *tab = &s->coeff_vlc[cb][prec][cbi];
403  const HuffmanCodebook *huff = &at9_huffman_coeffs[cb][prec][cbi];
404  const int groups = bands >> huff->value_cnt_pow;
405 
406  for (int j = 0; j < groups; j++) {
407  uint16_t val = get_vlc2(gb, tab->table, 9, huff->max_bit_size);
408 
409  for (int k = 0; k < huff->value_cnt; k++) {
410  coeffs[k] = sign_extend(val, huff->value_bits);
411  val >>= huff->value_bits;
412  }
413 
414  coeffs += huff->value_cnt;
415  }
416  } else {
417  for (int j = 0; j < bands; j++)
418  coeffs[j] = sign_extend(get_bits(gb, prec), prec);
419  }
420  }
421 }
422 
425 {
426  memset(c->q_coeffs_fine, 0, sizeof(c->q_coeffs_fine));
427 
428  for (int i = 0; i < c->q_unit_cnt; i++) {
429  const int start = at9_q_unit_to_coeff_idx[i + 0];
430  const int end = at9_q_unit_to_coeff_idx[i + 1];
431  const int len = c->precision_fine[i] + 1;
432 
433  if (c->precision_fine[i] <= 0)
434  continue;
435 
436  for (int j = start; j < end; j++)
437  c->q_coeffs_fine[j] = sign_extend(get_bits(gb, len), len);
438  }
439 }
440 
443 {
444  memset(c->coeffs, 0, sizeof(c->coeffs));
445 
446  for (int i = 0; i < c->q_unit_cnt; i++) {
447  const int start = at9_q_unit_to_coeff_idx[i + 0];
448  const int end = at9_q_unit_to_coeff_idx[i + 1];
449 
450  const float coarse_c = at9_quant_step_coarse[c->precision_coarse[i]];
451  const float fine_c = at9_quant_step_fine[c->precision_fine[i]];
452 
453  for (int j = start; j < end; j++) {
454  const float vc = c->q_coeffs_coarse[j] * coarse_c;
455  const float vf = c->q_coeffs_fine[j] * fine_c;
456  c->coeffs[j] = vc + vf;
457  }
458  }
459 }
460 
462  const int stereo)
463 {
464  float *src = b->channel[ b->cpe_base_channel].coeffs;
465  float *dst = b->channel[!b->cpe_base_channel].coeffs;
466 
467  if (!stereo)
468  return;
469 
470  if (b->q_unit_cnt <= b->stereo_q_unit)
471  return;
472 
473  for (int i = b->stereo_q_unit; i < b->q_unit_cnt; i++) {
474  const int sign = b->is_signs[i];
475  const int start = at9_q_unit_to_coeff_idx[i + 0];
476  const int end = at9_q_unit_to_coeff_idx[i + 1];
477  for (int j = start; j < end; j++)
478  dst[j] = sign*src[j];
479  }
480 }
481 
483  const int stereo)
484 {
485  for (int i = 0; i <= stereo; i++) {
486  float *coeffs = b->channel[i].coeffs;
487  for (int j = 0; j < b->q_unit_cnt; j++) {
488  const int start = at9_q_unit_to_coeff_idx[j + 0];
489  const int end = at9_q_unit_to_coeff_idx[j + 1];
490  const int scalefactor = b->channel[i].scalefactors[j];
491  const float scale = at9_scalefactor_c[scalefactor];
492  for (int k = start; k < end; k++)
493  coeffs[k] *= scale;
494  }
495  }
496 }
497 
499  int start, int count)
500 {
501  float maxval = 0.0f;
502  for (int i = 0; i < count; i += 2) {
503  double tmp[2];
504  av_bmg_get(&s->lfg, tmp);
505  c->coeffs[start + i + 0] = tmp[0];
506  c->coeffs[start + i + 1] = tmp[1];
507  maxval = FFMAX(FFMAX(FFABS(tmp[0]), FFABS(tmp[1])), maxval);
508  }
509  /* Normalize */
510  for (int i = 0; i < count; i++)
511  c->coeffs[start + i] /= maxval;
512 }
513 
514 static inline void scale_band_ext_coeffs(ATRAC9ChannelData *c, float sf[6],
515  const int s_unit, const int e_unit)
516 {
517  for (int i = s_unit; i < e_unit; i++) {
518  const int start = at9_q_unit_to_coeff_idx[i + 0];
519  const int end = at9_q_unit_to_coeff_idx[i + 1];
520  for (int j = start; j < end; j++)
521  c->coeffs[j] *= sf[i - s_unit];
522  }
523 }
524 
526  const int stereo)
527 {
528  const int g_units[4] = { /* A, B, C, total units */
529  b->q_unit_cnt,
532  FFMAX(g_units[2], 22),
533  };
534 
535  const int g_bins[4] = { /* A, B, C, total bins */
536  at9_q_unit_to_coeff_idx[g_units[0]],
537  at9_q_unit_to_coeff_idx[g_units[1]],
538  at9_q_unit_to_coeff_idx[g_units[2]],
539  at9_q_unit_to_coeff_idx[g_units[3]],
540  };
541 
542  if (!b->has_band_ext || !b->has_band_ext_data)
543  return;
544 
545  for (int ch = 0; ch <= stereo; ch++) {
546  ATRAC9ChannelData *c = &b->channel[ch];
547 
548  /* Mirror the spectrum */
549  for (int i = 0; i < 3; i++)
550  for (int j = 0; j < (g_bins[i + 1] - g_bins[i + 0]); j++)
551  c->coeffs[g_bins[i] + j] = c->coeffs[g_bins[i] - j - 1];
552 
553  switch (c->band_ext) {
554  case 0: {
555  float sf[6] = { 0.0f };
556  const int l = g_units[3] - g_units[0] - 1;
557  const int n_start = at9_q_unit_to_coeff_idx[g_units[3] - 1];
558  const int n_cnt = at9_q_unit_to_coeff_cnt[g_units[3] - 1];
559  switch (at9_tab_band_ext_group[b->q_unit_cnt - 13][2]) {
560  case 3:
561  sf[0] = at9_band_ext_scales_m0[0][0][c->band_ext_data[0]];
562  sf[1] = at9_band_ext_scales_m0[0][1][c->band_ext_data[0]];
563  sf[2] = at9_band_ext_scales_m0[0][2][c->band_ext_data[1]];
564  sf[3] = at9_band_ext_scales_m0[0][3][c->band_ext_data[2]];
565  sf[4] = at9_band_ext_scales_m0[0][4][c->band_ext_data[3]];
566  break;
567  case 4:
568  sf[0] = at9_band_ext_scales_m0[1][0][c->band_ext_data[0]];
569  sf[1] = at9_band_ext_scales_m0[1][1][c->band_ext_data[0]];
570  sf[2] = at9_band_ext_scales_m0[1][2][c->band_ext_data[1]];
571  sf[3] = at9_band_ext_scales_m0[1][3][c->band_ext_data[2]];
572  sf[4] = at9_band_ext_scales_m0[1][4][c->band_ext_data[3]];
573  break;
574  case 5:
575  sf[0] = at9_band_ext_scales_m0[2][0][c->band_ext_data[0]];
576  sf[1] = at9_band_ext_scales_m0[2][1][c->band_ext_data[1]];
577  sf[2] = at9_band_ext_scales_m0[2][2][c->band_ext_data[1]];
578  break;
579  }
580 
581  sf[l] = at9_scalefactor_c[c->scalefactors[g_units[0]]];
582 
583  fill_with_noise(s, c, n_start, n_cnt);
584  scale_band_ext_coeffs(c, sf, g_units[0], g_units[3]);
585  break;
586  }
587  case 1: {
588  float sf[6];
589  for (int i = g_units[0]; i < g_units[3]; i++)
590  sf[i - g_units[0]] = at9_scalefactor_c[c->scalefactors[i]];
591 
592  fill_with_noise(s, c, g_bins[0], g_bins[3] - g_bins[0]);
593  scale_band_ext_coeffs(c, sf, g_units[0], g_units[3]);
594  break;
595  }
596  case 2: {
597  const float g_sf[2] = {
600  };
601 
602  for (int i = 0; i < 2; i++)
603  for (int j = g_bins[i + 0]; j < g_bins[i + 1]; j++)
604  c->coeffs[j] *= g_sf[i];
605  break;
606  }
607  case 3: {
608  float scale = at9_band_ext_scales_m3[c->band_ext_data[0]][0];
609  float rate = at9_band_ext_scales_m3[c->band_ext_data[1]][1];
610  rate = pow(2, rate);
611  for (int i = g_bins[0]; i < g_bins[3]; i++) {
612  scale *= rate;
613  c->coeffs[i] *= scale;
614  }
615  break;
616  }
617  case 4: {
618  const float m = at9_band_ext_scales_m4[c->band_ext_data[0]];
619  const float g_sf[3] = { 0.7079468f*m, 0.5011902f*m, 0.3548279f*m };
620 
621  for (int i = 0; i < 3; i++)
622  for (int j = g_bins[i + 0]; j < g_bins[i + 1]; j++)
623  c->coeffs[j] *= g_sf[i];
624  break;
625  }
626  }
627  }
628 }
629 
632  int frame_idx, int block_idx)
633 {
634  const int first_in_pkt = !get_bits1(gb);
635  const int reuse_params = get_bits1(gb);
636  const int stereo = s->block_config->type[block_idx] == ATRAC9_BLOCK_TYPE_CPE;
637 
638  if (s->block_config->type[block_idx] == ATRAC9_BLOCK_TYPE_LFE) {
639  ATRAC9ChannelData *c = &b->channel[0];
640  const int precision = reuse_params ? 8 : 4;
641  c->q_unit_cnt = b->q_unit_cnt = 2;
642 
643  memset(c->scalefactors, 0, sizeof(c->scalefactors));
644  memset(c->q_coeffs_fine, 0, sizeof(c->q_coeffs_fine));
645  memset(c->q_coeffs_coarse, 0, sizeof(c->q_coeffs_coarse));
646 
647  for (int i = 0; i < b->q_unit_cnt; i++) {
648  c->scalefactors[i] = get_bits(gb, 5);
649  c->precision_coarse[i] = precision;
650  c->precision_fine[i] = 0;
651  }
652 
653  for (int i = 0; i < c->q_unit_cnt; i++) {
654  const int start = at9_q_unit_to_coeff_idx[i + 0];
655  const int end = at9_q_unit_to_coeff_idx[i + 1];
656  for (int j = start; j < end; j++)
657  c->q_coeffs_coarse[j] = get_bits(gb, c->precision_coarse[i] + 1);
658  }
659 
660  dequantize (s, b, c);
661  apply_scalefactors(s, b, 0);
662 
663  goto imdct;
664  }
665 
666  if (first_in_pkt && reuse_params) {
667  av_log(s->avctx, AV_LOG_ERROR, "Invalid block flags!\n");
668  return AVERROR_INVALIDDATA;
669  }
670 
671  /* Band parameters */
672  if (!reuse_params) {
673  int stereo_band, ext_band;
674  const int min_band_count = s->samplerate_idx > 7 ? 1 : 3;
675  b->reuseable = 0;
676  b->band_count = get_bits(gb, 4) + min_band_count;
678 
680 
682  av_log(s->avctx, AV_LOG_ERROR, "Invalid band count %i!\n",
683  b->band_count);
684  return AVERROR_INVALIDDATA;
685  }
686 
687  if (stereo) {
688  stereo_band = get_bits(gb, 4) + min_band_count;
689  if (stereo_band > b->band_count) {
690  av_log(s->avctx, AV_LOG_ERROR, "Invalid stereo band %i!\n",
691  stereo_band);
692  return AVERROR_INVALIDDATA;
693  }
694  b->stereo_q_unit = at9_tab_band_q_unit_map[stereo_band];
695  }
696 
697  b->has_band_ext = get_bits1(gb);
698  if (b->has_band_ext) {
699  ext_band = get_bits(gb, 4) + min_band_count;
700  if (ext_band < b->band_count) {
701  av_log(s->avctx, AV_LOG_ERROR, "Invalid extension band %i!\n",
702  ext_band);
703  return AVERROR_INVALIDDATA;
704  }
706  }
707  b->reuseable = 1;
708  }
709  if (!b->reuseable) {
710  av_log(s->avctx, AV_LOG_ERROR, "invalid block reused!\n");
711  return AVERROR_INVALIDDATA;
712  }
713 
714  /* Calculate bit alloc gradient */
715  if (parse_gradient(s, b, gb))
716  return AVERROR_INVALIDDATA;
717 
718  /* IS data */
719  b->cpe_base_channel = 0;
720  if (stereo) {
721  b->cpe_base_channel = get_bits1(gb);
722  if (get_bits1(gb)) {
723  for (int i = b->stereo_q_unit; i < b->q_unit_cnt; i++)
724  b->is_signs[i] = 1 - 2*get_bits1(gb);
725  } else {
726  for (int i = 0; i < FF_ARRAY_ELEMS(b->is_signs); i++)
727  b->is_signs[i] = 1;
728  }
729  }
730 
731  /* Band extension */
732  if (parse_band_ext(s, b, gb, stereo))
733  return AVERROR_INVALIDDATA;
734 
735  /* Scalefactors */
736  for (int i = 0; i <= stereo; i++) {
737  ATRAC9ChannelData *c = &b->channel[i];
738  c->q_unit_cnt = i == b->cpe_base_channel ? b->q_unit_cnt :
739  b->stereo_q_unit;
740  if (read_scalefactors(s, b, c, gb, i, first_in_pkt))
741  return AVERROR_INVALIDDATA;
742 
743  calc_precision (s, b, c);
744  calc_codebook_idx (s, b, c);
745  read_coeffs_coarse(s, b, c, gb);
746  read_coeffs_fine (s, b, c, gb);
747  dequantize (s, b, c);
748  }
749 
751 
752  apply_intensity_stereo(s, b, stereo);
753  apply_scalefactors (s, b, stereo);
754  apply_band_extension (s, b, stereo);
755 
756 imdct:
757  for (int i = 0; i <= stereo; i++) {
758  ATRAC9ChannelData *c = &b->channel[i];
759  const int dst_idx = s->block_config->plane_map[block_idx][i];
760  const int wsize = 1 << s->frame_log2;
761  const ptrdiff_t offset = wsize*frame_idx*sizeof(float);
762  float *dst = (float *)(frame->extended_data[dst_idx] + offset);
763 
764  s->imdct.imdct_half(&s->imdct, s->temp, c->coeffs);
765  s->fdsp->vector_fmul_window(dst, c->prev_win, s->temp,
766  s->imdct_win, wsize >> 1);
767  memcpy(c->prev_win, s->temp + (wsize >> 1), sizeof(float)*wsize >> 1);
768  }
769 
770  return 0;
771 }
772 
773 static int atrac9_decode_frame(AVCodecContext *avctx, void *data,
774  int *got_frame_ptr, AVPacket *avpkt)
775 {
776  int ret;
777  GetBitContext gb;
778  AVFrame *frame = data;
779  ATRAC9Context *s = avctx->priv_data;
780  const int frames = FFMIN(avpkt->size / s->avg_frame_size, s->frame_count);
781 
782  frame->nb_samples = (1 << s->frame_log2) * frames;
783  ret = ff_get_buffer(avctx, frame, 0);
784  if (ret < 0)
785  return ret;
786 
787  init_get_bits8(&gb, avpkt->data, avpkt->size);
788 
789  for (int i = 0; i < frames; i++) {
790  for (int j = 0; j < s->block_config->count; j++) {
791  ret = atrac9_decode_block(s, &gb, &s->block[j], frame, i, j);
792  if (ret)
793  return ret;
794  align_get_bits(&gb);
795  }
796  }
797 
798  *got_frame_ptr = 1;
799 
800  return avctx->block_align;
801 }
802 
804 {
805  ATRAC9Context *s = avctx->priv_data;
806 
807  for (int j = 0; j < s->block_config->count; j++) {
808  ATRAC9BlockData *b = &s->block[j];
809  const int stereo = s->block_config->type[j] == ATRAC9_BLOCK_TYPE_CPE;
810  for (int i = 0; i <= stereo; i++) {
811  ATRAC9ChannelData *c = &b->channel[i];
812  memset(c->prev_win, 0, sizeof(c->prev_win));
813  }
814  }
815 }
816 
818 {
819  ATRAC9Context *s = avctx->priv_data;
820 
821  for (int i = 1; i < 7; i++)
822  ff_free_vlc(&s->sf_vlc[0][i]);
823  for (int i = 2; i < 6; i++)
824  ff_free_vlc(&s->sf_vlc[1][i]);
825  for (int i = 0; i < 2; i++)
826  for (int j = 0; j < 8; j++)
827  for (int k = 0; k < 4; k++)
828  ff_free_vlc(&s->coeff_vlc[i][j][k]);
829 
830  ff_mdct_end(&s->imdct);
831  av_free(s->fdsp);
832 
833  return 0;
834 }
835 
837 {
838  GetBitContext gb;
839  ATRAC9Context *s = avctx->priv_data;
840  int version, block_config_idx, superframe_idx, alloc_c_len;
841 
842  s->avctx = avctx;
843 
844  av_lfg_init(&s->lfg, 0xFBADF00D);
845 
846  if (avctx->extradata_size != 12) {
847  av_log(avctx, AV_LOG_ERROR, "Invalid extradata length!\n");
848  return AVERROR_INVALIDDATA;
849  }
850 
851  version = AV_RL32(avctx->extradata);
852  if (version > 2) {
853  av_log(avctx, AV_LOG_ERROR, "Unsupported version (%i)!\n", version);
854  return AVERROR_INVALIDDATA;
855  }
856 
857  init_get_bits8(&gb, avctx->extradata + 4, avctx->extradata_size);
858 
859  if (get_bits(&gb, 8) != 0xFE) {
860  av_log(avctx, AV_LOG_ERROR, "Incorrect magic byte!\n");
861  return AVERROR_INVALIDDATA;
862  }
863 
864  s->samplerate_idx = get_bits(&gb, 4);
866 
867  block_config_idx = get_bits(&gb, 3);
868  if (block_config_idx > 5) {
869  av_log(avctx, AV_LOG_ERROR, "Incorrect block config!\n");
870  return AVERROR_INVALIDDATA;
871  }
872  s->block_config = &at9_block_layout[block_config_idx];
873 
876 
877  if (get_bits1(&gb)) {
878  av_log(avctx, AV_LOG_ERROR, "Incorrect verification bit!\n");
879  return AVERROR_INVALIDDATA;
880  }
881 
882  /* Average frame size in bytes */
883  s->avg_frame_size = get_bits(&gb, 11) + 1;
884 
885  superframe_idx = get_bits(&gb, 2);
886  if (superframe_idx & 1) {
887  av_log(avctx, AV_LOG_ERROR, "Invalid superframe index!\n");
888  return AVERROR_INVALIDDATA;
889  }
890 
891  s->frame_count = 1 << superframe_idx;
893 
894  if (ff_mdct_init(&s->imdct, s->frame_log2 + 1, 1, 1.0f / 32768.0f))
895  return AVERROR(ENOMEM);
896 
898  if (!s->fdsp)
899  return AVERROR(ENOMEM);
900 
901  /* iMDCT window */
902  for (int i = 0; i < (1 << s->frame_log2); i++) {
903  const int len = 1 << s->frame_log2;
904  const float sidx = ( i + 0.5f) / len;
905  const float eidx = (len - i - 0.5f) / len;
906  const float s_c = sinf(sidx*M_PI - M_PI_2)*0.5f + 0.5f;
907  const float e_c = sinf(eidx*M_PI - M_PI_2)*0.5f + 0.5f;
908  s->imdct_win[i] = s_c / ((s_c * s_c) + (e_c * e_c));
909  }
910 
911  /* Allocation curve */
912  alloc_c_len = FF_ARRAY_ELEMS(at9_tab_b_dist);
913  for (int i = 1; i <= alloc_c_len; i++)
914  for (int j = 0; j < i; j++)
915  s->alloc_curve[i - 1][j] = at9_tab_b_dist[(j * alloc_c_len) / i];
916 
917  /* Unsigned scalefactor VLCs */
918  for (int i = 1; i < 7; i++) {
920 
921  init_vlc(&s->sf_vlc[0][i], 9, hf->size, hf->bits, 1, 1, hf->codes,
922  2, 2, 0);
923  }
924 
925  /* Signed scalefactor VLCs */
926  for (int i = 2; i < 6; i++) {
927  const HuffmanCodebook *hf = &at9_huffman_sf_signed[i];
928 
929  int nums = hf->size;
930  int16_t sym[32];
931  for (int j = 0; j < nums; j++)
932  sym[j] = sign_extend(j, hf->value_bits);
933 
934  ff_init_vlc_sparse(&s->sf_vlc[1][i], 9, hf->size, hf->bits, 1, 1,
935  hf->codes, 2, 2, sym, sizeof(*sym), sizeof(*sym), 0);
936  }
937 
938  /* Coefficient VLCs */
939  for (int i = 0; i < 2; i++) {
940  for (int j = 0; j < 8; j++) {
941  for (int k = 0; k < 4; k++) {
942  const HuffmanCodebook *hf = &at9_huffman_coeffs[i][j][k];
943  init_vlc(&s->coeff_vlc[i][j][k], 9, hf->size, hf->bits, 1, 1,
944  hf->codes, 2, 2, 0);
945  }
946  }
947  }
948 
949  return 0;
950 }
951 
953  .name = "atrac9",
954  .long_name = NULL_IF_CONFIG_SMALL("ATRAC9 (Adaptive TRansform Acoustic Coding 9)"),
955  .type = AVMEDIA_TYPE_AUDIO,
956  .id = AV_CODEC_ID_ATRAC9,
957  .priv_data_size = sizeof(ATRAC9Context),
959  .close = atrac9_decode_close,
963  .capabilities = AV_CODEC_CAP_SUBFRAMES | AV_CODEC_CAP_DR1,
964 };
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
Definition: internal.h:48
Definition: lfg.h:27
static const uint8_t at9_tab_sri_max_bands[]
Definition: atrac9tab.h:125
float, planar
Definition: samplefmt.h:69
const char const char void * val
Definition: avisynth_c.h:771
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
AVCodec ff_atrac9_decoder
Definition: atrac9dec.c:952
int precision_mask[30]
Definition: atrac9dec.c:38
This structure describes decoded (raw) audio or video data.
Definition: frame.h:226
int precision_coarse[30]
Definition: atrac9dec.c:36
static void apply_band_extension(ATRAC9Context *s, ATRAC9BlockData *b, const int stereo)
Definition: atrac9dec.c:525
static void flush(AVCodecContext *avctx)
static unsigned int get_bits(GetBitContext *s, int n)
Read 1-25 bits.
Definition: get_bits.h:381
static void dequantize(ATRAC9Context *s, ATRAC9BlockData *b, ATRAC9ChannelData *c)
Definition: atrac9dec.c:441
else temp
Definition: vf_mcdeint.c:256
static void skip_bits_long(GetBitContext *s, int n)
Skips the specified number of bits.
Definition: get_bits.h:293
int band_ext_data[4]
Definition: atrac9dec.c:32
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
int ff_init_vlc_sparse(VLC *vlc_arg, int nb_bits, int nb_codes, const void *bits, int bits_wrap, int bits_size, const void *codes, int codes_wrap, int codes_size, const void *symbols, int symbols_wrap, int symbols_size, int flags)
Definition: bitstream.c:273
int size
Definition: avcodec.h:1446
uint8_t pi<< 24) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_U8,(uint64_t)((*(const uint8_t *) pi - 0x80U))<< 56) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_U8,(*(const uint8_t *) pi - 0x80) *(1.0f/(1<< 7))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_U8,(*(const uint8_t *) pi - 0x80) *(1.0/(1<< 7))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S16,(*(const int16_t *) pi >>8)+0x80) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_S16,(uint64_t)(*(const int16_t *) pi)<< 48) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S16, *(const int16_t *) pi *(1.0f/(1<< 15))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S16, *(const int16_t *) pi *(1.0/(1<< 15))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S32,(*(const int32_t *) pi >>24)+0x80) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_S32,(uint64_t)(*(const int32_t *) pi)<< 32) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S32, *(const int32_t *) pi *(1.0f/(1U<< 31))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S32, *(const int32_t *) pi *(1.0/(1U<< 31))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_S64,(*(const int64_t *) pi >>56)+0x80) CONV_FUNC(AV_SAMPLE_FMT_FLT, float, AV_SAMPLE_FMT_S64, *(const int64_t *) pi *(1.0f/(INT64_C(1)<< 63))) CONV_FUNC(AV_SAMPLE_FMT_DBL, double, AV_SAMPLE_FMT_S64, *(const int64_t *) pi *(1.0/(INT64_C(1)<< 63))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_FLT, av_clip_uint8(lrintf(*(const float *) pi *(1<< 7))+0x80)) CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_FLT, av_clip_int16(lrintf(*(const float *) pi *(1<< 15)))) CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_FLT, av_clipl_int32(llrintf(*(const float *) pi *(1U<< 31)))) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_FLT, llrintf(*(const float *) pi *(INT64_C(1)<< 63))) CONV_FUNC(AV_SAMPLE_FMT_U8, uint8_t, AV_SAMPLE_FMT_DBL, av_clip_uint8(lrint(*(const double *) pi *(1<< 7))+0x80)) CONV_FUNC(AV_SAMPLE_FMT_S16, int16_t, AV_SAMPLE_FMT_DBL, av_clip_int16(lrint(*(const double *) pi *(1<< 15)))) CONV_FUNC(AV_SAMPLE_FMT_S32, int32_t, AV_SAMPLE_FMT_DBL, av_clipl_int32(llrint(*(const double *) pi *(1U<< 31)))) CONV_FUNC(AV_SAMPLE_FMT_S64, int64_t, AV_SAMPLE_FMT_DBL, llrint(*(const double *) pi *(INT64_C(1)<< 63))) #define FMT_PAIR_FUNC(out, in) static conv_func_type *const fmt_pair_to_conv_functions[AV_SAMPLE_FMT_NB *AV_SAMPLE_FMT_NB]={ FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_U8), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_S16), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_S32), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_FLT), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_DBL), FMT_PAIR_FUNC(AV_SAMPLE_FMT_U8, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S32, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_FLT, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_DBL, AV_SAMPLE_FMT_S64), FMT_PAIR_FUNC(AV_SAMPLE_FMT_S64, AV_SAMPLE_FMT_S64), };static void cpy1(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, len);} static void cpy2(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, 2 *len);} static void cpy4(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, 4 *len);} static void cpy8(uint8_t **dst, const uint8_t **src, int len){ memcpy(*dst, *src, 8 *len);} AudioConvert *swri_audio_convert_alloc(enum AVSampleFormat out_fmt, enum AVSampleFormat in_fmt, int channels, const int *ch_map, int flags) { AudioConvert *ctx;conv_func_type *f=fmt_pair_to_conv_functions[av_get_packed_sample_fmt(out_fmt)+AV_SAMPLE_FMT_NB *av_get_packed_sample_fmt(in_fmt)];if(!f) return NULL;ctx=av_mallocz(sizeof(*ctx));if(!ctx) return NULL;if(channels==1){ in_fmt=av_get_planar_sample_fmt(in_fmt);out_fmt=av_get_planar_sample_fmt(out_fmt);} ctx->channels=channels;ctx->conv_f=f;ctx->ch_map=ch_map;if(in_fmt==AV_SAMPLE_FMT_U8||in_fmt==AV_SAMPLE_FMT_U8P) memset(ctx->silence, 0x80, sizeof(ctx->silence));if(out_fmt==in_fmt &&!ch_map) { switch(av_get_bytes_per_sample(in_fmt)){ case 1:ctx->simd_f=cpy1;break;case 2:ctx->simd_f=cpy2;break;case 4:ctx->simd_f=cpy4;break;case 8:ctx->simd_f=cpy8;break;} } if(HAVE_X86ASM &&HAVE_MMX) swri_audio_convert_init_x86(ctx, out_fmt, in_fmt, channels);if(ARCH_ARM) swri_audio_convert_init_arm(ctx, out_fmt, in_fmt, channels);if(ARCH_AARCH64) swri_audio_convert_init_aarch64(ctx, out_fmt, in_fmt, channels);return ctx;} void swri_audio_convert_free(AudioConvert **ctx) { av_freep(ctx);} int swri_audio_convert(AudioConvert *ctx, AudioData *out, AudioData *in, int len) { int ch;int off=0;const int os=(out->planar ? 1 :out->ch_count) *out->bps;unsigned misaligned=0;av_assert0(ctx->channels==out->ch_count);if(ctx->in_simd_align_mask) { int planes=in->planar ? in->ch_count :1;unsigned m=0;for(ch=0;ch< planes;ch++) m|=(intptr_t) in->ch[ch];misaligned|=m &ctx->in_simd_align_mask;} if(ctx->out_simd_align_mask) { int planes=out->planar ? out->ch_count :1;unsigned m=0;for(ch=0;ch< planes;ch++) m|=(intptr_t) out->ch[ch];misaligned|=m &ctx->out_simd_align_mask;} if(ctx->simd_f &&!ctx->ch_map &&!misaligned){ off=len &~15;av_assert1(off >=0);av_assert1(off<=len);av_assert2(ctx->channels==SWR_CH_MAX||!in->ch[ctx->channels]);if(off >0){ if(out->planar==in->planar){ int planes=out->planar ? out->ch_count :1;for(ch=0;ch< planes;ch++){ ctx->simd_f(out-> ch ch
Definition: audioconvert.c:56
const char * b
Definition: vf_curves.c:116
int q_unit_cnt_prev
Definition: atrac9dec.c:55
float prev_win[128]
Definition: atrac9dec.c:46
static void read_coeffs_coarse(ATRAC9Context *s, ATRAC9BlockData *b, ATRAC9ChannelData *c, GetBitContext *gb)
Definition: atrac9dec.c:387
VLC coeff_vlc[2][8][4]
Definition: atrac9dec.c:94
int version
Definition: avisynth_c.h:766
#define init_vlc(vlc, nb_bits, nb_codes, bits, bits_wrap, bits_size, codes, codes_wrap, codes_size, flags)
Definition: vlc.h:38
float coeffs[256]
Definition: atrac9dec.c:45
#define src
Definition: vp8dsp.c:254
AVCodec.
Definition: avcodec.h:3424
int block_align
number of bytes per packet if constant and known or 0 Used by some WAV based audio codecs...
Definition: avcodec.h:2226
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:42
int is_signs[30]
Definition: atrac9dec.c:72
AVCodecContext * avctx
Definition: atrac9dec.c:79
static const float at9_band_ext_scales_m4[]
Definition: atrac9tab.h:314
uint64_t channel_layout
Definition: atrac9tab.h:36
static int16_t block[64]
Definition: dct.c:115
const int max_bit_size
Definition: atrac9tab.h:492
static const HuffmanCodebook at9_huffman_coeffs[][8][4]
Definition: atrac9tab.h:1550
static double cb(void *priv, double x, double y)
Definition: vf_geq.c:112
static const int at9_tab_samplerates[]
Definition: atrac9tab.h:129
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
Definition: internal.h:40
ATRAC9ChannelData channel[2]
Definition: atrac9dec.c:50
enum AVSampleFormat sample_fmt
audio sample format
Definition: avcodec.h:2197
uint8_t
#define av_cold
Definition: attributes.h:82
float delta
int avg_frame_size
Definition: atrac9dec.c:87
static const OptionGroupDef groups[]
Definition: ffmpeg_opt.c:3210
#define f(width, name)
Definition: cbs_vp9.c:255
float imdct_win[256]
Definition: atrac9dec.c:96
static av_cold int end(AVCodecContext *avctx)
Definition: avrndec.c:90
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1634
static AVFrame * frame
const char data[16]
Definition: mxf.c:91
#define DECLARE_ALIGNED(n, t, v)
Declare a variable that is aligned in memory.
Definition: mem.h:112
uint8_t * data
Definition: avcodec.h:1445
bitstream reader API header.
static void fill_with_noise(ATRAC9Context *s, ATRAC9ChannelData *c, int start, int count)
Definition: atrac9dec.c:498
int frame_count
Definition: atrac9dec.c:88
static int read_scalefactors(ATRAC9Context *s, ATRAC9BlockData *b, ATRAC9ChannelData *c, GetBitContext *gb, int channel_idx, int first_in_pkt)
Definition: atrac9dec.c:244
void av_bmg_get(AVLFG *lfg, double out[2])
Get the next two numbers generated by a Box-Muller Gaussian generator using the random numbers issued...
Definition: lfg.c:49
static void apply_scalefactors(ATRAC9Context *s, ATRAC9BlockData *b, const int stereo)
Definition: atrac9dec.c:482
static const uint8_t at9_tab_sf_weights[][32]
Definition: atrac9tab.h:348
int32_t q_coeffs_coarse[256]
Definition: atrac9dec.c:42
#define av_log(a,...)
static void calc_precision(ATRAC9Context *s, ATRAC9BlockData *b, ATRAC9ChannelData *c)
Definition: atrac9dec.c:148
static int atrac9_decode_block(ATRAC9Context *s, GetBitContext *gb, ATRAC9BlockData *b, AVFrame *frame, int frame_idx, int block_idx)
Definition: atrac9dec.c:630
static const float at9_quant_step_fine[]
Definition: atrac9tab.h:328
const uint8_t * bits
Definition: atrac9tab.h:486
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:119
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
static void read_coeffs_fine(ATRAC9Context *s, ATRAC9BlockData *b, ATRAC9ChannelData *c, GetBitContext *gb)
Definition: atrac9dec.c:423
av_cold AVFloatDSPContext * avpriv_float_dsp_alloc(int bit_exact)
Allocate a float DSP context.
Definition: float_dsp.c:135
static const uint8_t at9_tab_band_q_unit_map[]
Definition: atrac9tab.h:106
#define AVERROR(e)
Definition: error.h:43
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
const uint8_t at9_q_unit_to_codebookidx[]
Definition: atrac9tab.h:120
int flags
AV_CODEC_FLAG_*.
Definition: avcodec.h:1613
static int parse_band_ext(ATRAC9Context *s, ATRAC9BlockData *b, GetBitContext *gb, int stereo)
Definition: atrac9dec.c:199
const char * name
Name of the codec implementation.
Definition: avcodec.h:3431
const int size
Definition: atrac9tab.h:488
static void calc_codebook_idx(ATRAC9Context *s, ATRAC9BlockData *b, ATRAC9ChannelData *c)
Definition: atrac9dec.c:343
#define ff_mdct_init
Definition: fft.h:169
int has_band_ext_data
Definition: atrac9dec.c:62
static const uint8_t offset[127][2]
Definition: vf_spp.c:92
VLC sf_vlc[2][8]
Definition: atrac9dec.c:93
#define FFMAX(a, b)
Definition: common.h:94
FFTContext imdct
Definition: atrac9dec.c:81
Definition: vlc.h:26
uint64_t channel_layout
Audio channel layout.
Definition: avcodec.h:2240
enum ATRAC9BlockType type[5]
Definition: atrac9tab.h:37
static VLC sf_vlc
scale factor DPCM vlc
Definition: wmaprodec.c:128
int frame_log2
Definition: atrac9dec.c:86
Definition: fft.h:88
#define AV_CODEC_FLAG_BITEXACT
Use only bitexact stuff (except (I)DCT).
Definition: avcodec.h:895
#define FFMIN(a, b)
Definition: common.h:96
int32_t scalefactors_prev[31]
Definition: atrac9dec.c:34
int band_ext_q_unit
Definition: atrac9dec.c:63
#define M_PI_2
Definition: mathematics.h:55
int32_t
static const float at9_band_ext_scales_m0[][5][32]
Definition: atrac9tab.h:197
ATRAC9BlockData block[5]
Definition: atrac9dec.c:82
#define FFABS(a)
Absolute value, Note, INT_MIN / INT64_MIN result in undefined behavior as they are not representable ...
Definition: common.h:72
#define s(width, name)
Definition: cbs_vp9.c:257
static const float at9_scalefactor_c[]
Definition: atrac9tab.h:337
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:762
#define AV_RL32
Definition: intreadwrite.h:146
int frames
Definition: movenc.c:65
#define FF_ARRAY_ELEMS(a)
static int parse_gradient(ATRAC9Context *s, ATRAC9BlockData *b, GetBitContext *gb)
Definition: atrac9dec.c:101
static av_cold int atrac9_decode_init(AVCodecContext *avctx)
Definition: atrac9dec.c:836
#define sinf(x)
Definition: libm.h:419
const int value_bits
Definition: atrac9tab.h:491
int sample_rate
samples per second
Definition: avcodec.h:2189
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
Definition: get_bits.h:650
int cpe_base_channel
Definition: atrac9dec.c:71
main external API structure.
Definition: avcodec.h:1533
const int value_cnt_pow
Definition: atrac9tab.h:490
static const float bands[]
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: decode.c:1918
int extradata_size
Definition: avcodec.h:1635
static void apply_intensity_stereo(ATRAC9Context *s, ATRAC9BlockData *b, const int stereo)
Definition: atrac9dec.c:461
static unsigned int get_bits1(GetBitContext *s)
Definition: get_bits.h:487
static const uint8_t at9_q_unit_to_coeff_cnt[]
Definition: atrac9tab.h:110
static void skip_bits1(GetBitContext *s)
Definition: get_bits.h:523
static void skip_bits(GetBitContext *s, int n)
Definition: get_bits.h:460
int precision_fine[30]
Definition: atrac9dec.c:37
#define AV_CODEC_CAP_SUBFRAMES
Codec can output multiple frames per AVPacket Normally demuxers return one frame at a time...
Definition: avcodec.h:1011
uint8_t alloc_curve[48][48]
Definition: atrac9dec.c:95
int samplerate_idx
Definition: atrac9dec.c:89
static av_cold int atrac9_decode_close(AVCodecContext *avctx)
Definition: atrac9dec.c:817
av_cold void av_lfg_init(AVLFG *c, unsigned int seed)
Definition: lfg.c:32
static const uint8_t at9_tab_band_ext_group[][3]
Definition: atrac9tab.h:143
void(* imdct_half)(struct FFTContext *s, FFTSample *output, const FFTSample *input)
Definition: fft.h:108
static const uint8_t at9_tab_band_ext_cnt[][6]
Definition: atrac9tab.h:134
static av_const int sign_extend(int val, unsigned bits)
Definition: mathops.h:130
static const float at9_band_ext_scales_m2[]
Definition: atrac9tab.h:284
static void scale_band_ext_coeffs(ATRAC9ChannelData *c, float sf[6], const int s_unit, const int e_unit)
Definition: atrac9dec.c:514
AVFloatDSPContext * fdsp
Definition: atrac9dec.c:80
static const float at9_quant_step_coarse[]
Definition: atrac9tab.h:319
#define avg(a, b, c, d)
static int atrac9_decode_frame(AVCodecContext *avctx, void *data, int *got_frame_ptr, AVPacket *avpkt)
Definition: atrac9dec.c:773
static const HuffmanCodebook at9_huffman_sf_unsigned[]
Definition: atrac9tab.h:495
common internal api header.
const ATRAC9BlockConfig * block_config
Definition: atrac9dec.c:90
#define ff_mdct_end
Definition: fft.h:170
const int value_cnt
Definition: atrac9tab.h:489
static double c[64]
channel
Use these values when setting the channel map with ebur128_set_channel().
Definition: ebur128.h:39
static const uint8_t at9_tab_sri_frame_log2[]
Definition: atrac9tab.h:102
int codebookset[30]
Definition: atrac9dec.c:40
int plane_map[5][2]
Definition: atrac9tab.h:38
static const float at9_band_ext_scales_m3[][2]
Definition: atrac9tab.h:303
const uint16_t * codes
Definition: atrac9tab.h:487
void * priv_data
Definition: avcodec.h:1560
#define av_free(p)
static const HuffmanCodebook at9_huffman_sf_signed[]
Definition: atrac9tab.h:505
int gradient[31]
Definition: atrac9dec.c:68
int len
VLC_TYPE(* table)[2]
code, bits
Definition: vlc.h:28
static const uint8_t * align_get_bits(GetBitContext *s)
Definition: get_bits.h:658
static const struct twinvq_data tab
int32_t scalefactors[31]
Definition: atrac9dec.c:33
static const ATRAC9BlockConfig at9_block_layout[]
Definition: atrac9tab.h:42
static const uint8_t at9_tab_b_dist[]
Definition: atrac9tab.h:383
static void atrac9_decode_flush(AVCodecContext *avctx)
Definition: atrac9dec.c:803
void INT64 INT64 count
Definition: avisynth_c.h:690
void INT64 start
Definition: avisynth_c.h:690
#define M_PI
Definition: mathematics.h:52
uint8_t ** extended_data
pointers to the data planes/channels.
Definition: frame.h:273
float min
This structure stores compressed data.
Definition: avcodec.h:1422
void ff_free_vlc(VLC *vlc)
Definition: bitstream.c:359
mode
Use these values in ebur128_init (or&#39;ed).
Definition: ebur128.h:83
int nb_samples
number of audio samples (per channel) described by this frame
Definition: frame.h:292
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:968
static const int at9_q_unit_to_coeff_idx[]
Definition: atrac9tab.h:115
int32_t q_coeffs_fine[256]
Definition: atrac9dec.c:43
for(j=16;j >0;--j)
static const uint8_t at9_tab_band_ext_lengths[][6][4]
Definition: atrac9tab.h:154
float temp[256]
Definition: atrac9dec.c:98
static uint8_t tmp[11]
Definition: aes_ctr.c:26