ffmpeg / libavcodec / vorbis_dec.c @ 925aa969
History | View | Annotate | Download (61.1 KB)
1 |
/**
|
---|---|
2 |
* @file
|
3 |
* Vorbis I decoder
|
4 |
* @author Denes Balatoni ( dbalatoni programozo hu )
|
5 |
*
|
6 |
* This file is part of FFmpeg.
|
7 |
*
|
8 |
* FFmpeg 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 |
* FFmpeg 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 FFmpeg; if not, write to the Free Software
|
20 |
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
21 |
*/
|
22 |
|
23 |
#undef V_DEBUG
|
24 |
//#define V_DEBUG
|
25 |
//#define AV_DEBUG(...) av_log(NULL, AV_LOG_INFO, __VA_ARGS__)
|
26 |
|
27 |
#include <math.h> |
28 |
|
29 |
#define ALT_BITSTREAM_READER_LE
|
30 |
#include "avcodec.h" |
31 |
#include "get_bits.h" |
32 |
#include "dsputil.h" |
33 |
#include "fft.h" |
34 |
|
35 |
#include "vorbis.h" |
36 |
#include "xiph.h" |
37 |
|
38 |
#define V_NB_BITS 8 |
39 |
#define V_NB_BITS2 11 |
40 |
#define V_MAX_VLCS (1 << 16) |
41 |
#define V_MAX_PARTITIONS (1 << 20) |
42 |
|
43 |
#ifndef V_DEBUG
|
44 |
#define AV_DEBUG(...)
|
45 |
#endif
|
46 |
|
47 |
#undef NDEBUG
|
48 |
#include <assert.h> |
49 |
|
50 |
typedef struct { |
51 |
uint_fast8_t dimensions; |
52 |
uint_fast8_t lookup_type; |
53 |
uint_fast8_t maxdepth; |
54 |
VLC vlc; |
55 |
float *codevectors;
|
56 |
unsigned int nb_bits; |
57 |
} vorbis_codebook; |
58 |
|
59 |
typedef union vorbis_floor_u vorbis_floor_data; |
60 |
typedef struct vorbis_floor0_s vorbis_floor0; |
61 |
typedef struct vorbis_floor1_s vorbis_floor1; |
62 |
struct vorbis_context_s;
|
63 |
typedef
|
64 |
int (* vorbis_floor_decode_func)
|
65 |
(struct vorbis_context_s *, vorbis_floor_data *, float *); |
66 |
typedef struct { |
67 |
uint_fast8_t floor_type; |
68 |
vorbis_floor_decode_func decode; |
69 |
union vorbis_floor_u {
|
70 |
struct vorbis_floor0_s {
|
71 |
uint_fast8_t order; |
72 |
uint_fast16_t rate; |
73 |
uint_fast16_t bark_map_size; |
74 |
int_fast32_t *map[2];
|
75 |
uint_fast32_t map_size[2];
|
76 |
uint_fast8_t amplitude_bits; |
77 |
uint_fast8_t amplitude_offset; |
78 |
uint_fast8_t num_books; |
79 |
uint_fast8_t *book_list; |
80 |
float *lsp;
|
81 |
} t0; |
82 |
struct vorbis_floor1_s {
|
83 |
uint_fast8_t partitions; |
84 |
uint8_t partition_class[32];
|
85 |
uint_fast8_t class_dimensions[16];
|
86 |
uint_fast8_t class_subclasses[16];
|
87 |
uint_fast8_t class_masterbook[16];
|
88 |
int_fast16_t subclass_books[16][8]; |
89 |
uint_fast8_t multiplier; |
90 |
uint_fast16_t x_list_dim; |
91 |
vorbis_floor1_entry *list; |
92 |
} t1; |
93 |
} data; |
94 |
} vorbis_floor; |
95 |
|
96 |
typedef struct { |
97 |
uint_fast16_t type; |
98 |
uint_fast32_t begin; |
99 |
uint_fast32_t end; |
100 |
unsigned partition_size;
|
101 |
uint_fast8_t classifications; |
102 |
uint_fast8_t classbook; |
103 |
int_fast16_t books[64][8]; |
104 |
uint_fast8_t maxpass; |
105 |
uint_fast16_t ptns_to_read; |
106 |
uint8_t *classifs; |
107 |
} vorbis_residue; |
108 |
|
109 |
typedef struct { |
110 |
uint_fast8_t submaps; |
111 |
uint_fast16_t coupling_steps; |
112 |
uint_fast8_t *magnitude; |
113 |
uint_fast8_t *angle; |
114 |
uint_fast8_t *mux; |
115 |
uint_fast8_t submap_floor[16];
|
116 |
uint_fast8_t submap_residue[16];
|
117 |
} vorbis_mapping; |
118 |
|
119 |
typedef struct { |
120 |
uint_fast8_t blockflag; |
121 |
uint_fast16_t windowtype; |
122 |
uint_fast16_t transformtype; |
123 |
uint_fast8_t mapping; |
124 |
} vorbis_mode; |
125 |
|
126 |
typedef struct vorbis_context_s { |
127 |
AVCodecContext *avccontext; |
128 |
GetBitContext gb; |
129 |
DSPContext dsp; |
130 |
|
131 |
FFTContext mdct[2];
|
132 |
uint_fast8_t first_frame; |
133 |
uint_fast32_t version; |
134 |
uint_fast8_t audio_channels; |
135 |
uint_fast32_t audio_samplerate; |
136 |
uint_fast32_t bitrate_maximum; |
137 |
uint_fast32_t bitrate_nominal; |
138 |
uint_fast32_t bitrate_minimum; |
139 |
uint_fast32_t blocksize[2];
|
140 |
const float *win[2]; |
141 |
uint_fast16_t codebook_count; |
142 |
vorbis_codebook *codebooks; |
143 |
uint_fast8_t floor_count; |
144 |
vorbis_floor *floors; |
145 |
uint_fast8_t residue_count; |
146 |
vorbis_residue *residues; |
147 |
uint_fast8_t mapping_count; |
148 |
vorbis_mapping *mappings; |
149 |
uint_fast8_t mode_count; |
150 |
vorbis_mode *modes; |
151 |
uint_fast8_t mode_number; // mode number for the current packet
|
152 |
uint_fast8_t previous_window; |
153 |
float *channel_residues;
|
154 |
float *channel_floors;
|
155 |
float *saved;
|
156 |
uint_fast32_t add_bias; // for float->int conversion
|
157 |
uint_fast32_t exp_bias; |
158 |
} vorbis_context; |
159 |
|
160 |
/* Helper functions */
|
161 |
|
162 |
#define BARK(x) \
|
163 |
(13.1f * atan(0.00074f * (x)) + 2.24f * atan(1.85e-8f * (x) * (x)) + 1e-4f * (x)) |
164 |
|
165 |
static const char idx_err_str[] = "Index value %d out of range (0 - %d) for %s at %s:%i\n"; |
166 |
#define VALIDATE_INDEX(idx, limit) \
|
167 |
if (idx >= limit) {\
|
168 |
av_log(vc->avccontext, AV_LOG_ERROR,\ |
169 |
idx_err_str,\ |
170 |
(int)(idx), (int)(limit - 1), #idx, __FILE__, __LINE__);\ |
171 |
return -1;\ |
172 |
} |
173 |
#define GET_VALIDATED_INDEX(idx, bits, limit) \
|
174 |
{\ |
175 |
idx = get_bits(gb, bits);\ |
176 |
VALIDATE_INDEX(idx, limit)\ |
177 |
} |
178 |
|
179 |
static float vorbisfloat2float(uint_fast32_t val) |
180 |
{ |
181 |
double mant = val & 0x1fffff; |
182 |
long exp = (val & 0x7fe00000L) >> 21; |
183 |
if (val & 0x80000000) |
184 |
mant = -mant; |
185 |
return ldexp(mant, exp - 20 - 768); |
186 |
} |
187 |
|
188 |
|
189 |
// Free all allocated memory -----------------------------------------
|
190 |
|
191 |
static void vorbis_free(vorbis_context *vc) |
192 |
{ |
193 |
int_fast16_t i; |
194 |
|
195 |
av_freep(&vc->channel_residues); |
196 |
av_freep(&vc->channel_floors); |
197 |
av_freep(&vc->saved); |
198 |
|
199 |
for (i = 0; i < vc->residue_count; i++) |
200 |
av_free(vc->residues[i].classifs); |
201 |
av_freep(&vc->residues); |
202 |
av_freep(&vc->modes); |
203 |
|
204 |
ff_mdct_end(&vc->mdct[0]);
|
205 |
ff_mdct_end(&vc->mdct[1]);
|
206 |
|
207 |
for (i = 0; i < vc->codebook_count; ++i) { |
208 |
av_free(vc->codebooks[i].codevectors); |
209 |
free_vlc(&vc->codebooks[i].vlc); |
210 |
} |
211 |
av_freep(&vc->codebooks); |
212 |
|
213 |
for (i = 0; i < vc->floor_count; ++i) { |
214 |
if (vc->floors[i].floor_type == 0) { |
215 |
av_free(vc->floors[i].data.t0.map[0]);
|
216 |
av_free(vc->floors[i].data.t0.map[1]);
|
217 |
av_free(vc->floors[i].data.t0.book_list); |
218 |
av_free(vc->floors[i].data.t0.lsp); |
219 |
} else {
|
220 |
av_free(vc->floors[i].data.t1.list); |
221 |
} |
222 |
} |
223 |
av_freep(&vc->floors); |
224 |
|
225 |
for (i = 0; i < vc->mapping_count; ++i) { |
226 |
av_free(vc->mappings[i].magnitude); |
227 |
av_free(vc->mappings[i].angle); |
228 |
av_free(vc->mappings[i].mux); |
229 |
} |
230 |
av_freep(&vc->mappings); |
231 |
} |
232 |
|
233 |
// Parse setup header -------------------------------------------------
|
234 |
|
235 |
// Process codebooks part
|
236 |
|
237 |
static int vorbis_parse_setup_hdr_codebooks(vorbis_context *vc) |
238 |
{ |
239 |
uint_fast16_t cb; |
240 |
uint8_t *tmp_vlc_bits; |
241 |
uint32_t *tmp_vlc_codes; |
242 |
GetBitContext *gb = &vc->gb; |
243 |
uint_fast16_t *codebook_multiplicands; |
244 |
|
245 |
vc->codebook_count = get_bits(gb, 8) + 1; |
246 |
|
247 |
AV_DEBUG(" Codebooks: %d \n", vc->codebook_count);
|
248 |
|
249 |
vc->codebooks = av_mallocz(vc->codebook_count * sizeof(vorbis_codebook));
|
250 |
tmp_vlc_bits = av_mallocz(V_MAX_VLCS * sizeof(uint8_t));
|
251 |
tmp_vlc_codes = av_mallocz(V_MAX_VLCS * sizeof(uint32_t));
|
252 |
codebook_multiplicands = av_malloc(V_MAX_VLCS * sizeof(*codebook_multiplicands));
|
253 |
|
254 |
for (cb = 0; cb < vc->codebook_count; ++cb) { |
255 |
vorbis_codebook *codebook_setup = &vc->codebooks[cb]; |
256 |
uint_fast8_t ordered; |
257 |
uint_fast32_t t, used_entries = 0;
|
258 |
uint_fast32_t entries; |
259 |
|
260 |
AV_DEBUG(" %d. Codebook \n", cb);
|
261 |
|
262 |
if (get_bits(gb, 24) != 0x564342) { |
263 |
av_log(vc->avccontext, AV_LOG_ERROR, " %"PRIdFAST16". Codebook setup data corrupt. \n", cb); |
264 |
goto error;
|
265 |
} |
266 |
|
267 |
codebook_setup->dimensions=get_bits(gb, 16);
|
268 |
if (codebook_setup->dimensions > 16 || codebook_setup->dimensions == 0) { |
269 |
av_log(vc->avccontext, AV_LOG_ERROR, " %"PRIdFAST16". Codebook's dimension is invalid (%d). \n", cb, codebook_setup->dimensions); |
270 |
goto error;
|
271 |
} |
272 |
entries = get_bits(gb, 24);
|
273 |
if (entries > V_MAX_VLCS) {
|
274 |
av_log(vc->avccontext, AV_LOG_ERROR, " %"PRIdFAST16". Codebook has too many entries (%"PRIdFAST32"). \n", cb, entries); |
275 |
goto error;
|
276 |
} |
277 |
|
278 |
ordered = get_bits1(gb); |
279 |
|
280 |
AV_DEBUG(" codebook_dimensions %d, codebook_entries %d \n", codebook_setup->dimensions, entries);
|
281 |
|
282 |
if (!ordered) {
|
283 |
uint_fast16_t ce; |
284 |
uint_fast8_t flag; |
285 |
uint_fast8_t sparse = get_bits1(gb); |
286 |
|
287 |
AV_DEBUG(" not ordered \n");
|
288 |
|
289 |
if (sparse) {
|
290 |
AV_DEBUG(" sparse \n");
|
291 |
|
292 |
used_entries = 0;
|
293 |
for (ce = 0; ce < entries; ++ce) { |
294 |
flag = get_bits1(gb); |
295 |
if (flag) {
|
296 |
tmp_vlc_bits[ce] = get_bits(gb, 5) + 1; |
297 |
++used_entries; |
298 |
} else
|
299 |
tmp_vlc_bits[ce] = 0;
|
300 |
} |
301 |
} else {
|
302 |
AV_DEBUG(" not sparse \n");
|
303 |
|
304 |
used_entries = entries; |
305 |
for (ce = 0; ce < entries; ++ce) |
306 |
tmp_vlc_bits[ce] = get_bits(gb, 5) + 1; |
307 |
} |
308 |
} else {
|
309 |
uint_fast16_t current_entry = 0;
|
310 |
uint_fast8_t current_length = get_bits(gb, 5)+1; |
311 |
|
312 |
AV_DEBUG(" ordered, current length: %d \n", current_length); //FIXME |
313 |
|
314 |
used_entries = entries; |
315 |
for (; current_entry < used_entries && current_length <= 32; ++current_length) { |
316 |
uint_fast16_t i, number; |
317 |
|
318 |
AV_DEBUG(" number bits: %d ", ilog(entries - current_entry));
|
319 |
|
320 |
number = get_bits(gb, ilog(entries - current_entry)); |
321 |
|
322 |
AV_DEBUG(" number: %d \n", number);
|
323 |
|
324 |
for (i = current_entry; i < number+current_entry; ++i)
|
325 |
if (i < used_entries)
|
326 |
tmp_vlc_bits[i] = current_length; |
327 |
|
328 |
current_entry+=number; |
329 |
} |
330 |
if (current_entry>used_entries) {
|
331 |
av_log(vc->avccontext, AV_LOG_ERROR, " More codelengths than codes in codebook. \n");
|
332 |
goto error;
|
333 |
} |
334 |
} |
335 |
|
336 |
codebook_setup->lookup_type = get_bits(gb, 4);
|
337 |
|
338 |
AV_DEBUG(" lookup type: %d : %s \n", codebook_setup->lookup_type, codebook_setup->lookup_type ? "vq" : "no lookup"); |
339 |
|
340 |
// If the codebook is used for (inverse) VQ, calculate codevectors.
|
341 |
|
342 |
if (codebook_setup->lookup_type == 1) { |
343 |
uint_fast16_t i, j, k; |
344 |
uint_fast16_t codebook_lookup_values = ff_vorbis_nth_root(entries, codebook_setup->dimensions); |
345 |
|
346 |
float codebook_minimum_value = vorbisfloat2float(get_bits_long(gb, 32)); |
347 |
float codebook_delta_value = vorbisfloat2float(get_bits_long(gb, 32)); |
348 |
uint_fast8_t codebook_value_bits = get_bits(gb, 4)+1; |
349 |
uint_fast8_t codebook_sequence_p = get_bits1(gb); |
350 |
|
351 |
AV_DEBUG(" We expect %d numbers for building the codevectors. \n", codebook_lookup_values);
|
352 |
AV_DEBUG(" delta %f minmum %f \n", codebook_delta_value, codebook_minimum_value);
|
353 |
|
354 |
for (i = 0; i < codebook_lookup_values; ++i) { |
355 |
codebook_multiplicands[i] = get_bits(gb, codebook_value_bits); |
356 |
|
357 |
AV_DEBUG(" multiplicands*delta+minmum : %e \n", (float)codebook_multiplicands[i]*codebook_delta_value+codebook_minimum_value); |
358 |
AV_DEBUG(" multiplicand %d \n", codebook_multiplicands[i]);
|
359 |
} |
360 |
|
361 |
// Weed out unused vlcs and build codevector vector
|
362 |
codebook_setup->codevectors = used_entries ? av_mallocz(used_entries*codebook_setup->dimensions * sizeof(float)) : NULL; |
363 |
for (j = 0, i = 0; i < entries; ++i) { |
364 |
uint_fast8_t dim = codebook_setup->dimensions; |
365 |
|
366 |
if (tmp_vlc_bits[i]) {
|
367 |
float last = 0.0; |
368 |
uint_fast32_t lookup_offset = i; |
369 |
|
370 |
#ifdef V_DEBUG
|
371 |
av_log(vc->avccontext, AV_LOG_INFO, "Lookup offset %d ,", i);
|
372 |
#endif
|
373 |
|
374 |
for (k = 0; k < dim; ++k) { |
375 |
uint_fast32_t multiplicand_offset = lookup_offset % codebook_lookup_values; |
376 |
codebook_setup->codevectors[j * dim + k] = codebook_multiplicands[multiplicand_offset] * codebook_delta_value + codebook_minimum_value + last; |
377 |
if (codebook_sequence_p)
|
378 |
last = codebook_setup->codevectors[j * dim + k]; |
379 |
lookup_offset/=codebook_lookup_values; |
380 |
} |
381 |
tmp_vlc_bits[j] = tmp_vlc_bits[i]; |
382 |
|
383 |
#ifdef V_DEBUG
|
384 |
av_log(vc->avccontext, AV_LOG_INFO, "real lookup offset %d, vector: ", j);
|
385 |
for (k = 0; k < dim; ++k) |
386 |
av_log(vc->avccontext, AV_LOG_INFO, " %f ", codebook_setup->codevectors[j * dim + k]);
|
387 |
av_log(vc->avccontext, AV_LOG_INFO, "\n");
|
388 |
#endif
|
389 |
|
390 |
++j; |
391 |
} |
392 |
} |
393 |
if (j != used_entries) {
|
394 |
av_log(vc->avccontext, AV_LOG_ERROR, "Bug in codevector vector building code. \n");
|
395 |
goto error;
|
396 |
} |
397 |
entries = used_entries; |
398 |
} else if (codebook_setup->lookup_type >= 2) { |
399 |
av_log(vc->avccontext, AV_LOG_ERROR, "Codebook lookup type not supported. \n");
|
400 |
goto error;
|
401 |
} |
402 |
|
403 |
// Initialize VLC table
|
404 |
if (ff_vorbis_len2vlc(tmp_vlc_bits, tmp_vlc_codes, entries)) {
|
405 |
av_log(vc->avccontext, AV_LOG_ERROR, " Invalid code lengths while generating vlcs. \n");
|
406 |
goto error;
|
407 |
} |
408 |
codebook_setup->maxdepth = 0;
|
409 |
for (t = 0; t < entries; ++t) |
410 |
if (tmp_vlc_bits[t] >= codebook_setup->maxdepth)
|
411 |
codebook_setup->maxdepth = tmp_vlc_bits[t]; |
412 |
|
413 |
if (codebook_setup->maxdepth > 3 * V_NB_BITS) |
414 |
codebook_setup->nb_bits = V_NB_BITS2; |
415 |
else
|
416 |
codebook_setup->nb_bits = V_NB_BITS; |
417 |
|
418 |
codebook_setup->maxdepth = (codebook_setup->maxdepth+codebook_setup->nb_bits - 1) / codebook_setup->nb_bits;
|
419 |
|
420 |
if (init_vlc(&codebook_setup->vlc, codebook_setup->nb_bits, entries, tmp_vlc_bits, sizeof(*tmp_vlc_bits), sizeof(*tmp_vlc_bits), tmp_vlc_codes, sizeof(*tmp_vlc_codes), sizeof(*tmp_vlc_codes), INIT_VLC_LE)) { |
421 |
av_log(vc->avccontext, AV_LOG_ERROR, " Error generating vlc tables. \n");
|
422 |
goto error;
|
423 |
} |
424 |
} |
425 |
|
426 |
av_free(tmp_vlc_bits); |
427 |
av_free(tmp_vlc_codes); |
428 |
av_free(codebook_multiplicands); |
429 |
return 0; |
430 |
|
431 |
// Error:
|
432 |
error:
|
433 |
av_free(tmp_vlc_bits); |
434 |
av_free(tmp_vlc_codes); |
435 |
av_free(codebook_multiplicands); |
436 |
return -1; |
437 |
} |
438 |
|
439 |
// Process time domain transforms part (unused in Vorbis I)
|
440 |
|
441 |
static int vorbis_parse_setup_hdr_tdtransforms(vorbis_context *vc) |
442 |
{ |
443 |
GetBitContext *gb = &vc->gb; |
444 |
uint_fast8_t i; |
445 |
uint_fast8_t vorbis_time_count = get_bits(gb, 6) + 1; |
446 |
|
447 |
for (i = 0; i < vorbis_time_count; ++i) { |
448 |
uint_fast16_t vorbis_tdtransform = get_bits(gb, 16);
|
449 |
|
450 |
AV_DEBUG(" Vorbis time domain transform %d: %d \n", vorbis_time_count, vorbis_tdtransform);
|
451 |
|
452 |
if (vorbis_tdtransform) {
|
453 |
av_log(vc->avccontext, AV_LOG_ERROR, "Vorbis time domain transform data nonzero. \n");
|
454 |
return -1; |
455 |
} |
456 |
} |
457 |
return 0; |
458 |
} |
459 |
|
460 |
// Process floors part
|
461 |
|
462 |
static int vorbis_floor0_decode(vorbis_context *vc, |
463 |
vorbis_floor_data *vfu, float *vec);
|
464 |
static void create_map(vorbis_context *vc, uint_fast8_t floor_number); |
465 |
static int vorbis_floor1_decode(vorbis_context *vc, |
466 |
vorbis_floor_data *vfu, float *vec);
|
467 |
static int vorbis_parse_setup_hdr_floors(vorbis_context *vc) |
468 |
{ |
469 |
GetBitContext *gb = &vc->gb; |
470 |
int i,j,k;
|
471 |
|
472 |
vc->floor_count = get_bits(gb, 6) + 1; |
473 |
|
474 |
vc->floors = av_mallocz(vc->floor_count * sizeof(vorbis_floor));
|
475 |
|
476 |
for (i = 0; i < vc->floor_count; ++i) { |
477 |
vorbis_floor *floor_setup = &vc->floors[i]; |
478 |
|
479 |
floor_setup->floor_type = get_bits(gb, 16);
|
480 |
|
481 |
AV_DEBUG(" %d. floor type %d \n", i, floor_setup->floor_type);
|
482 |
|
483 |
if (floor_setup->floor_type == 1) { |
484 |
int maximum_class = -1; |
485 |
uint_fast8_t rangebits; |
486 |
uint_fast32_t rangemax; |
487 |
uint_fast16_t floor1_values = 2;
|
488 |
|
489 |
floor_setup->decode = vorbis_floor1_decode; |
490 |
|
491 |
floor_setup->data.t1.partitions = get_bits(gb, 5);
|
492 |
|
493 |
AV_DEBUG(" %d.floor: %d partitions \n", i, floor_setup->data.t1.partitions);
|
494 |
|
495 |
for (j = 0; j < floor_setup->data.t1.partitions; ++j) { |
496 |
floor_setup->data.t1.partition_class[j] = get_bits(gb, 4);
|
497 |
if (floor_setup->data.t1.partition_class[j] > maximum_class)
|
498 |
maximum_class = floor_setup->data.t1.partition_class[j]; |
499 |
|
500 |
AV_DEBUG(" %d. floor %d partition class %d \n", i, j, floor_setup->data.t1.partition_class[j]);
|
501 |
|
502 |
} |
503 |
|
504 |
AV_DEBUG(" maximum class %d \n", maximum_class);
|
505 |
|
506 |
for (j = 0; j <= maximum_class; ++j) { |
507 |
floor_setup->data.t1.class_dimensions[j] = get_bits(gb, 3) + 1; |
508 |
floor_setup->data.t1.class_subclasses[j] = get_bits(gb, 2);
|
509 |
|
510 |
AV_DEBUG(" %d floor %d class dim: %d subclasses %d \n", i, j, floor_setup->data.t1.class_dimensions[j], floor_setup->data.t1.class_subclasses[j]);
|
511 |
|
512 |
if (floor_setup->data.t1.class_subclasses[j]) {
|
513 |
GET_VALIDATED_INDEX(floor_setup->data.t1.class_masterbook[j], 8, vc->codebook_count)
|
514 |
|
515 |
AV_DEBUG(" masterbook: %d \n", floor_setup->data.t1.class_masterbook[j]);
|
516 |
} |
517 |
|
518 |
for (k = 0; k < (1 << floor_setup->data.t1.class_subclasses[j]); ++k) { |
519 |
int16_t bits = get_bits(gb, 8) - 1; |
520 |
if (bits != -1) |
521 |
VALIDATE_INDEX(bits, vc->codebook_count) |
522 |
floor_setup->data.t1.subclass_books[j][k] = bits; |
523 |
|
524 |
AV_DEBUG(" book %d. : %d \n", k, floor_setup->data.t1.subclass_books[j][k]);
|
525 |
} |
526 |
} |
527 |
|
528 |
floor_setup->data.t1.multiplier = get_bits(gb, 2) + 1; |
529 |
floor_setup->data.t1.x_list_dim = 2;
|
530 |
|
531 |
for (j = 0; j < floor_setup->data.t1.partitions; ++j) |
532 |
floor_setup->data.t1.x_list_dim+=floor_setup->data.t1.class_dimensions[floor_setup->data.t1.partition_class[j]]; |
533 |
|
534 |
floor_setup->data.t1.list = av_mallocz(floor_setup->data.t1.x_list_dim * sizeof(vorbis_floor1_entry));
|
535 |
|
536 |
|
537 |
rangebits = get_bits(gb, 4);
|
538 |
rangemax = (1 << rangebits);
|
539 |
if (rangemax > vc->blocksize[1] / 2) { |
540 |
av_log(vc->avccontext, AV_LOG_ERROR, |
541 |
"Floor value is too large for blocksize: %d (%d)\n",
|
542 |
rangemax, vc->blocksize[1] / 2); |
543 |
return -1; |
544 |
} |
545 |
floor_setup->data.t1.list[0].x = 0; |
546 |
floor_setup->data.t1.list[1].x = rangemax;
|
547 |
|
548 |
for (j = 0; j < floor_setup->data.t1.partitions; ++j) { |
549 |
for (k = 0; k < floor_setup->data.t1.class_dimensions[floor_setup->data.t1.partition_class[j]]; ++k, ++floor1_values) { |
550 |
floor_setup->data.t1.list[floor1_values].x = get_bits(gb, rangebits); |
551 |
|
552 |
AV_DEBUG(" %d. floor1 Y coord. %d \n", floor1_values, floor_setup->data.t1.list[floor1_values].x);
|
553 |
} |
554 |
} |
555 |
|
556 |
// Precalculate order of x coordinates - needed for decode
|
557 |
ff_vorbis_ready_floor1_list(floor_setup->data.t1.list, floor_setup->data.t1.x_list_dim); |
558 |
} else if (floor_setup->floor_type == 0) { |
559 |
uint_fast8_t max_codebook_dim = 0;
|
560 |
|
561 |
floor_setup->decode = vorbis_floor0_decode; |
562 |
|
563 |
floor_setup->data.t0.order = get_bits(gb, 8);
|
564 |
floor_setup->data.t0.rate = get_bits(gb, 16);
|
565 |
floor_setup->data.t0.bark_map_size = get_bits(gb, 16);
|
566 |
floor_setup->data.t0.amplitude_bits = get_bits(gb, 6);
|
567 |
/* zero would result in a div by zero later *
|
568 |
* 2^0 - 1 == 0 */
|
569 |
if (floor_setup->data.t0.amplitude_bits == 0) { |
570 |
av_log(vc->avccontext, AV_LOG_ERROR, |
571 |
"Floor 0 amplitude bits is 0.\n");
|
572 |
return -1; |
573 |
} |
574 |
floor_setup->data.t0.amplitude_offset = get_bits(gb, 8);
|
575 |
floor_setup->data.t0.num_books = get_bits(gb, 4) + 1; |
576 |
|
577 |
/* allocate mem for booklist */
|
578 |
floor_setup->data.t0.book_list = |
579 |
av_malloc(floor_setup->data.t0.num_books); |
580 |
if (!floor_setup->data.t0.book_list)
|
581 |
return -1; |
582 |
/* read book indexes */
|
583 |
{ |
584 |
int idx;
|
585 |
uint_fast8_t book_idx; |
586 |
for (idx = 0; idx < floor_setup->data.t0.num_books; ++idx) { |
587 |
GET_VALIDATED_INDEX(book_idx, 8, vc->codebook_count)
|
588 |
floor_setup->data.t0.book_list[idx] = book_idx; |
589 |
if (vc->codebooks[book_idx].dimensions > max_codebook_dim)
|
590 |
max_codebook_dim = vc->codebooks[book_idx].dimensions; |
591 |
} |
592 |
} |
593 |
|
594 |
create_map(vc, i); |
595 |
|
596 |
/* allocate mem for lsp coefficients */
|
597 |
{ |
598 |
/* codebook dim is for padding if codebook dim doesn't *
|
599 |
* divide order+1 then we need to read more data */
|
600 |
floor_setup->data.t0.lsp = |
601 |
av_malloc((floor_setup->data.t0.order+1 + max_codebook_dim)
|
602 |
* sizeof(float)); |
603 |
if (!floor_setup->data.t0.lsp)
|
604 |
return -1; |
605 |
} |
606 |
|
607 |
#ifdef V_DEBUG /* debug output parsed headers */ |
608 |
AV_DEBUG("floor0 order: %u\n", floor_setup->data.t0.order);
|
609 |
AV_DEBUG("floor0 rate: %u\n", floor_setup->data.t0.rate);
|
610 |
AV_DEBUG("floor0 bark map size: %u\n",
|
611 |
floor_setup->data.t0.bark_map_size); |
612 |
AV_DEBUG("floor0 amplitude bits: %u\n",
|
613 |
floor_setup->data.t0.amplitude_bits); |
614 |
AV_DEBUG("floor0 amplitude offset: %u\n",
|
615 |
floor_setup->data.t0.amplitude_offset); |
616 |
AV_DEBUG("floor0 number of books: %u\n",
|
617 |
floor_setup->data.t0.num_books); |
618 |
AV_DEBUG("floor0 book list pointer: %p\n",
|
619 |
floor_setup->data.t0.book_list); |
620 |
{ |
621 |
int idx;
|
622 |
for (idx = 0; idx < floor_setup->data.t0.num_books; ++idx) { |
623 |
AV_DEBUG(" Book %d: %u\n",
|
624 |
idx+1,
|
625 |
floor_setup->data.t0.book_list[idx]); |
626 |
} |
627 |
} |
628 |
#endif
|
629 |
} else {
|
630 |
av_log(vc->avccontext, AV_LOG_ERROR, "Invalid floor type!\n");
|
631 |
return -1; |
632 |
} |
633 |
} |
634 |
return 0; |
635 |
} |
636 |
|
637 |
// Process residues part
|
638 |
|
639 |
static int vorbis_parse_setup_hdr_residues(vorbis_context *vc) |
640 |
{ |
641 |
GetBitContext *gb = &vc->gb; |
642 |
uint_fast8_t i, j, k; |
643 |
|
644 |
vc->residue_count = get_bits(gb, 6)+1; |
645 |
vc->residues = av_mallocz(vc->residue_count * sizeof(vorbis_residue));
|
646 |
|
647 |
AV_DEBUG(" There are %d residues. \n", vc->residue_count);
|
648 |
|
649 |
for (i = 0; i < vc->residue_count; ++i) { |
650 |
vorbis_residue *res_setup = &vc->residues[i]; |
651 |
uint_fast8_t cascade[64];
|
652 |
uint_fast8_t high_bits; |
653 |
uint_fast8_t low_bits; |
654 |
|
655 |
res_setup->type = get_bits(gb, 16);
|
656 |
|
657 |
AV_DEBUG(" %d. residue type %d \n", i, res_setup->type);
|
658 |
|
659 |
res_setup->begin = get_bits(gb, 24);
|
660 |
res_setup->end = get_bits(gb, 24);
|
661 |
res_setup->partition_size = get_bits(gb, 24) + 1; |
662 |
/* Validations to prevent a buffer overflow later. */
|
663 |
if (res_setup->begin>res_setup->end ||
|
664 |
res_setup->end > vc->avccontext->channels * vc->blocksize[1] / 2 || |
665 |
(res_setup->end-res_setup->begin) / res_setup->partition_size > V_MAX_PARTITIONS) { |
666 |
av_log(vc->avccontext, AV_LOG_ERROR, "partition out of bounds: type, begin, end, size, blocksize: %"PRIdFAST16", %"PRIdFAST32", %"PRIdFAST32", %u, %"PRIdFAST32"\n", res_setup->type, res_setup->begin, res_setup->end, res_setup->partition_size, vc->blocksize[1] / 2); |
667 |
return -1; |
668 |
} |
669 |
|
670 |
res_setup->classifications = get_bits(gb, 6) + 1; |
671 |
GET_VALIDATED_INDEX(res_setup->classbook, 8, vc->codebook_count)
|
672 |
|
673 |
res_setup->ptns_to_read = |
674 |
(res_setup->end - res_setup->begin) / res_setup->partition_size; |
675 |
res_setup->classifs = av_malloc(res_setup->ptns_to_read * |
676 |
vc->audio_channels * |
677 |
sizeof(*res_setup->classifs));
|
678 |
if (!res_setup->classifs)
|
679 |
return AVERROR(ENOMEM);
|
680 |
|
681 |
AV_DEBUG(" begin %d end %d part.size %d classif.s %d classbook %d \n", res_setup->begin, res_setup->end, res_setup->partition_size,
|
682 |
res_setup->classifications, res_setup->classbook); |
683 |
|
684 |
for (j = 0; j < res_setup->classifications; ++j) { |
685 |
high_bits = 0;
|
686 |
low_bits = get_bits(gb, 3);
|
687 |
if (get_bits1(gb))
|
688 |
high_bits = get_bits(gb, 5);
|
689 |
cascade[j] = (high_bits << 3) + low_bits;
|
690 |
|
691 |
AV_DEBUG(" %d class casscade depth: %d \n", j, ilog(cascade[j]));
|
692 |
} |
693 |
|
694 |
res_setup->maxpass = 0;
|
695 |
for (j = 0; j < res_setup->classifications; ++j) { |
696 |
for (k = 0; k < 8; ++k) { |
697 |
if (cascade[j]&(1 << k)) { |
698 |
GET_VALIDATED_INDEX(res_setup->books[j][k], 8, vc->codebook_count)
|
699 |
|
700 |
AV_DEBUG(" %d class casscade depth %d book: %d \n", j, k, res_setup->books[j][k]);
|
701 |
|
702 |
if (k>res_setup->maxpass)
|
703 |
res_setup->maxpass = k; |
704 |
} else {
|
705 |
res_setup->books[j][k] = -1;
|
706 |
} |
707 |
} |
708 |
} |
709 |
} |
710 |
return 0; |
711 |
} |
712 |
|
713 |
// Process mappings part
|
714 |
|
715 |
static int vorbis_parse_setup_hdr_mappings(vorbis_context *vc) |
716 |
{ |
717 |
GetBitContext *gb = &vc->gb; |
718 |
uint_fast8_t i, j; |
719 |
|
720 |
vc->mapping_count = get_bits(gb, 6)+1; |
721 |
vc->mappings = av_mallocz(vc->mapping_count * sizeof(vorbis_mapping));
|
722 |
|
723 |
AV_DEBUG(" There are %d mappings. \n", vc->mapping_count);
|
724 |
|
725 |
for (i = 0; i < vc->mapping_count; ++i) { |
726 |
vorbis_mapping *mapping_setup = &vc->mappings[i]; |
727 |
|
728 |
if (get_bits(gb, 16)) { |
729 |
av_log(vc->avccontext, AV_LOG_ERROR, "Other mappings than type 0 are not compliant with the Vorbis I specification. \n");
|
730 |
return -1; |
731 |
} |
732 |
if (get_bits1(gb)) {
|
733 |
mapping_setup->submaps = get_bits(gb, 4) + 1; |
734 |
} else {
|
735 |
mapping_setup->submaps = 1;
|
736 |
} |
737 |
|
738 |
if (get_bits1(gb)) {
|
739 |
mapping_setup->coupling_steps = get_bits(gb, 8) + 1; |
740 |
mapping_setup->magnitude = av_mallocz(mapping_setup->coupling_steps * sizeof(uint_fast8_t));
|
741 |
mapping_setup->angle = av_mallocz(mapping_setup->coupling_steps * sizeof(uint_fast8_t));
|
742 |
for (j = 0; j < mapping_setup->coupling_steps; ++j) { |
743 |
GET_VALIDATED_INDEX(mapping_setup->magnitude[j], ilog(vc->audio_channels - 1), vc->audio_channels)
|
744 |
GET_VALIDATED_INDEX(mapping_setup->angle[j], ilog(vc->audio_channels - 1), vc->audio_channels)
|
745 |
} |
746 |
} else {
|
747 |
mapping_setup->coupling_steps = 0;
|
748 |
} |
749 |
|
750 |
AV_DEBUG(" %d mapping coupling steps: %d \n", i, mapping_setup->coupling_steps);
|
751 |
|
752 |
if (get_bits(gb, 2)) { |
753 |
av_log(vc->avccontext, AV_LOG_ERROR, "%d. mapping setup data invalid. \n", i);
|
754 |
return -1; // following spec. |
755 |
} |
756 |
|
757 |
if (mapping_setup->submaps>1) { |
758 |
mapping_setup->mux = av_mallocz(vc->audio_channels * sizeof(uint_fast8_t));
|
759 |
for (j = 0; j < vc->audio_channels; ++j) |
760 |
mapping_setup->mux[j] = get_bits(gb, 4);
|
761 |
} |
762 |
|
763 |
for (j = 0; j < mapping_setup->submaps; ++j) { |
764 |
skip_bits(gb, 8); // FIXME check? |
765 |
GET_VALIDATED_INDEX(mapping_setup->submap_floor[j], 8, vc->floor_count)
|
766 |
GET_VALIDATED_INDEX(mapping_setup->submap_residue[j], 8, vc->residue_count)
|
767 |
|
768 |
AV_DEBUG(" %d mapping %d submap : floor %d, residue %d \n", i, j, mapping_setup->submap_floor[j], mapping_setup->submap_residue[j]);
|
769 |
} |
770 |
} |
771 |
return 0; |
772 |
} |
773 |
|
774 |
// Process modes part
|
775 |
|
776 |
static void create_map(vorbis_context *vc, uint_fast8_t floor_number) |
777 |
{ |
778 |
vorbis_floor *floors = vc->floors; |
779 |
vorbis_floor0 *vf; |
780 |
int idx;
|
781 |
int_fast8_t blockflag; |
782 |
int_fast32_t *map; |
783 |
int_fast32_t n; //TODO: could theoretically be smaller?
|
784 |
|
785 |
for (blockflag = 0; blockflag < 2; ++blockflag) { |
786 |
n = vc->blocksize[blockflag] / 2;
|
787 |
floors[floor_number].data.t0.map[blockflag] = |
788 |
av_malloc((n+1) * sizeof(int_fast32_t)); // n + sentinel |
789 |
|
790 |
map = floors[floor_number].data.t0.map[blockflag]; |
791 |
vf = &floors[floor_number].data.t0; |
792 |
|
793 |
for (idx = 0; idx < n; ++idx) { |
794 |
map[idx] = floor(BARK((vf->rate * idx) / (2.0f * n)) * |
795 |
((vf->bark_map_size) / |
796 |
BARK(vf->rate / 2.0f))); |
797 |
if (vf->bark_map_size-1 < map[idx]) |
798 |
map[idx] = vf->bark_map_size - 1;
|
799 |
} |
800 |
map[n] = -1;
|
801 |
vf->map_size[blockflag] = n; |
802 |
} |
803 |
|
804 |
# ifdef V_DEBUG
|
805 |
for (idx = 0; idx <= n; ++idx) { |
806 |
AV_DEBUG("floor0 map: map at pos %d is %d\n",
|
807 |
idx, map[idx]); |
808 |
} |
809 |
# endif
|
810 |
} |
811 |
|
812 |
static int vorbis_parse_setup_hdr_modes(vorbis_context *vc) |
813 |
{ |
814 |
GetBitContext *gb = &vc->gb; |
815 |
uint_fast8_t i; |
816 |
|
817 |
vc->mode_count = get_bits(gb, 6) + 1; |
818 |
vc->modes = av_mallocz(vc->mode_count * sizeof(vorbis_mode));
|
819 |
|
820 |
AV_DEBUG(" There are %d modes.\n", vc->mode_count);
|
821 |
|
822 |
for (i = 0; i < vc->mode_count; ++i) { |
823 |
vorbis_mode *mode_setup = &vc->modes[i]; |
824 |
|
825 |
mode_setup->blockflag = get_bits1(gb); |
826 |
mode_setup->windowtype = get_bits(gb, 16); //FIXME check |
827 |
mode_setup->transformtype = get_bits(gb, 16); //FIXME check |
828 |
GET_VALIDATED_INDEX(mode_setup->mapping, 8, vc->mapping_count);
|
829 |
|
830 |
AV_DEBUG(" %d mode: blockflag %d, windowtype %d, transformtype %d, mapping %d \n", i, mode_setup->blockflag, mode_setup->windowtype, mode_setup->transformtype, mode_setup->mapping);
|
831 |
} |
832 |
return 0; |
833 |
} |
834 |
|
835 |
// Process the whole setup header using the functions above
|
836 |
|
837 |
static int vorbis_parse_setup_hdr(vorbis_context *vc) |
838 |
{ |
839 |
GetBitContext *gb = &vc->gb; |
840 |
|
841 |
if ((get_bits(gb, 8) != 'v') || (get_bits(gb, 8) != 'o') || |
842 |
(get_bits(gb, 8) != 'r') || (get_bits(gb, 8) != 'b') || |
843 |
(get_bits(gb, 8) != 'i') || (get_bits(gb, 8) != 's')) { |
844 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (no vorbis signature). \n");
|
845 |
return -1; |
846 |
} |
847 |
|
848 |
if (vorbis_parse_setup_hdr_codebooks(vc)) {
|
849 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (codebooks). \n");
|
850 |
return -2; |
851 |
} |
852 |
if (vorbis_parse_setup_hdr_tdtransforms(vc)) {
|
853 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (time domain transforms). \n");
|
854 |
return -3; |
855 |
} |
856 |
if (vorbis_parse_setup_hdr_floors(vc)) {
|
857 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (floors). \n");
|
858 |
return -4; |
859 |
} |
860 |
if (vorbis_parse_setup_hdr_residues(vc)) {
|
861 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (residues). \n");
|
862 |
return -5; |
863 |
} |
864 |
if (vorbis_parse_setup_hdr_mappings(vc)) {
|
865 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (mappings). \n");
|
866 |
return -6; |
867 |
} |
868 |
if (vorbis_parse_setup_hdr_modes(vc)) {
|
869 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (modes). \n");
|
870 |
return -7; |
871 |
} |
872 |
if (!get_bits1(gb)) {
|
873 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (framing flag). \n");
|
874 |
return -8; // framing flag bit unset error |
875 |
} |
876 |
|
877 |
return 0; |
878 |
} |
879 |
|
880 |
// Process the identification header
|
881 |
|
882 |
static int vorbis_parse_id_hdr(vorbis_context *vc) |
883 |
{ |
884 |
GetBitContext *gb = &vc->gb; |
885 |
uint_fast8_t bl0, bl1; |
886 |
|
887 |
if ((get_bits(gb, 8) != 'v') || (get_bits(gb, 8) != 'o') || |
888 |
(get_bits(gb, 8) != 'r') || (get_bits(gb, 8) != 'b') || |
889 |
(get_bits(gb, 8) != 'i') || (get_bits(gb, 8) != 's')) { |
890 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (no vorbis signature). \n");
|
891 |
return -1; |
892 |
} |
893 |
|
894 |
vc->version = get_bits_long(gb, 32); //FIXME check 0 |
895 |
vc->audio_channels = get_bits(gb, 8);
|
896 |
if (vc->audio_channels <= 0) { |
897 |
av_log(vc->avccontext, AV_LOG_ERROR, "Invalid number of channels\n");
|
898 |
return -1; |
899 |
} |
900 |
vc->audio_samplerate = get_bits_long(gb, 32);
|
901 |
if (vc->audio_samplerate <= 0) { |
902 |
av_log(vc->avccontext, AV_LOG_ERROR, "Invalid samplerate\n");
|
903 |
return -1; |
904 |
} |
905 |
vc->bitrate_maximum = get_bits_long(gb, 32);
|
906 |
vc->bitrate_nominal = get_bits_long(gb, 32);
|
907 |
vc->bitrate_minimum = get_bits_long(gb, 32);
|
908 |
bl0 = get_bits(gb, 4);
|
909 |
bl1 = get_bits(gb, 4);
|
910 |
vc->blocksize[0] = (1 << bl0); |
911 |
vc->blocksize[1] = (1 << bl1); |
912 |
if (bl0 > 13 || bl0 < 6 || bl1 > 13 || bl1 < 6 || bl1 < bl0) { |
913 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (illegal blocksize). \n");
|
914 |
return -3; |
915 |
} |
916 |
// output format int16
|
917 |
if (vc->blocksize[1] / 2 * vc->audio_channels * 2 > AVCODEC_MAX_AUDIO_FRAME_SIZE) { |
918 |
av_log(vc->avccontext, AV_LOG_ERROR, "Vorbis channel count makes "
|
919 |
"output packets too large.\n");
|
920 |
return -4; |
921 |
} |
922 |
vc->win[0] = ff_vorbis_vwin[bl0 - 6]; |
923 |
vc->win[1] = ff_vorbis_vwin[bl1 - 6]; |
924 |
|
925 |
if ((get_bits1(gb)) == 0) { |
926 |
av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (framing flag not set). \n");
|
927 |
return -2; |
928 |
} |
929 |
|
930 |
vc->channel_residues = av_malloc((vc->blocksize[1] / 2) * vc->audio_channels * sizeof(float)); |
931 |
vc->channel_floors = av_malloc((vc->blocksize[1] / 2) * vc->audio_channels * sizeof(float)); |
932 |
vc->saved = av_mallocz((vc->blocksize[1] / 4) * vc->audio_channels * sizeof(float)); |
933 |
vc->previous_window = 0;
|
934 |
|
935 |
ff_mdct_init(&vc->mdct[0], bl0, 1, vc->exp_bias ? -(1 << 15) : -1.0); |
936 |
ff_mdct_init(&vc->mdct[1], bl1, 1, vc->exp_bias ? -(1 << 15) : -1.0); |
937 |
|
938 |
AV_DEBUG(" vorbis version %d \n audio_channels %d \n audio_samplerate %d \n bitrate_max %d \n bitrate_nom %d \n bitrate_min %d \n blk_0 %d blk_1 %d \n ",
|
939 |
vc->version, vc->audio_channels, vc->audio_samplerate, vc->bitrate_maximum, vc->bitrate_nominal, vc->bitrate_minimum, vc->blocksize[0], vc->blocksize[1]); |
940 |
|
941 |
/*
|
942 |
BLK = vc->blocksize[0];
|
943 |
for (i = 0; i < BLK / 2; ++i) {
|
944 |
vc->win[0][i] = sin(0.5*3.14159265358*(sin(((float)i + 0.5) / (float)BLK*3.14159265358))*(sin(((float)i + 0.5) / (float)BLK*3.14159265358)));
|
945 |
}
|
946 |
*/
|
947 |
|
948 |
return 0; |
949 |
} |
950 |
|
951 |
// Process the extradata using the functions above (identification header, setup header)
|
952 |
|
953 |
static av_cold int vorbis_decode_init(AVCodecContext *avccontext) |
954 |
{ |
955 |
vorbis_context *vc = avccontext->priv_data ; |
956 |
uint8_t *headers = avccontext->extradata; |
957 |
int headers_len = avccontext->extradata_size;
|
958 |
uint8_t *header_start[3];
|
959 |
int header_len[3]; |
960 |
GetBitContext *gb = &(vc->gb); |
961 |
int hdr_type;
|
962 |
|
963 |
vc->avccontext = avccontext; |
964 |
dsputil_init(&vc->dsp, avccontext); |
965 |
|
966 |
if (vc->dsp.float_to_int16_interleave == ff_float_to_int16_interleave_c) {
|
967 |
vc->add_bias = 385;
|
968 |
vc->exp_bias = 0;
|
969 |
} else {
|
970 |
vc->add_bias = 0;
|
971 |
vc->exp_bias = 15 << 23; |
972 |
} |
973 |
|
974 |
if (!headers_len) {
|
975 |
av_log(avccontext, AV_LOG_ERROR, "Extradata missing.\n");
|
976 |
return -1; |
977 |
} |
978 |
|
979 |
if (ff_split_xiph_headers(headers, headers_len, 30, header_start, header_len) < 0) { |
980 |
av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");
|
981 |
return -1; |
982 |
} |
983 |
|
984 |
init_get_bits(gb, header_start[0], header_len[0]*8); |
985 |
hdr_type = get_bits(gb, 8);
|
986 |
if (hdr_type != 1) { |
987 |
av_log(avccontext, AV_LOG_ERROR, "First header is not the id header.\n");
|
988 |
return -1; |
989 |
} |
990 |
if (vorbis_parse_id_hdr(vc)) {
|
991 |
av_log(avccontext, AV_LOG_ERROR, "Id header corrupt.\n");
|
992 |
vorbis_free(vc); |
993 |
return -1; |
994 |
} |
995 |
|
996 |
init_get_bits(gb, header_start[2], header_len[2]*8); |
997 |
hdr_type = get_bits(gb, 8);
|
998 |
if (hdr_type != 5) { |
999 |
av_log(avccontext, AV_LOG_ERROR, "Third header is not the setup header.\n");
|
1000 |
vorbis_free(vc); |
1001 |
return -1; |
1002 |
} |
1003 |
if (vorbis_parse_setup_hdr(vc)) {
|
1004 |
av_log(avccontext, AV_LOG_ERROR, "Setup header corrupt.\n");
|
1005 |
vorbis_free(vc); |
1006 |
return -1; |
1007 |
} |
1008 |
|
1009 |
if (vc->audio_channels > 8) |
1010 |
avccontext->channel_layout = 0;
|
1011 |
else
|
1012 |
avccontext->channel_layout = ff_vorbis_channel_layouts[vc->audio_channels - 1];
|
1013 |
|
1014 |
avccontext->channels = vc->audio_channels; |
1015 |
avccontext->sample_rate = vc->audio_samplerate; |
1016 |
avccontext->frame_size = FFMIN(vc->blocksize[0], vc->blocksize[1]) >> 2; |
1017 |
avccontext->sample_fmt = AV_SAMPLE_FMT_S16; |
1018 |
|
1019 |
return 0 ; |
1020 |
} |
1021 |
|
1022 |
// Decode audiopackets -------------------------------------------------
|
1023 |
|
1024 |
// Read and decode floor
|
1025 |
|
1026 |
static int vorbis_floor0_decode(vorbis_context *vc, |
1027 |
vorbis_floor_data *vfu, float *vec)
|
1028 |
{ |
1029 |
vorbis_floor0 *vf = &vfu->t0; |
1030 |
float *lsp = vf->lsp;
|
1031 |
uint_fast32_t amplitude; |
1032 |
uint_fast32_t book_idx; |
1033 |
uint_fast8_t blockflag = vc->modes[vc->mode_number].blockflag; |
1034 |
|
1035 |
amplitude = get_bits(&vc->gb, vf->amplitude_bits); |
1036 |
if (amplitude > 0) { |
1037 |
float last = 0; |
1038 |
uint_fast16_t lsp_len = 0;
|
1039 |
uint_fast16_t idx; |
1040 |
vorbis_codebook codebook; |
1041 |
|
1042 |
book_idx = get_bits(&vc->gb, ilog(vf->num_books)); |
1043 |
if (book_idx >= vf->num_books) {
|
1044 |
av_log(vc->avccontext, AV_LOG_ERROR, |
1045 |
"floor0 dec: booknumber too high!\n");
|
1046 |
book_idx = 0;
|
1047 |
//FIXME: look above
|
1048 |
} |
1049 |
AV_DEBUG("floor0 dec: booknumber: %u\n", book_idx);
|
1050 |
codebook = vc->codebooks[vf->book_list[book_idx]]; |
1051 |
/* Invalid codebook! */
|
1052 |
if (!codebook.codevectors)
|
1053 |
return -1; |
1054 |
|
1055 |
while (lsp_len<vf->order) {
|
1056 |
int vec_off;
|
1057 |
|
1058 |
AV_DEBUG("floor0 dec: book dimension: %d\n", codebook.dimensions);
|
1059 |
AV_DEBUG("floor0 dec: maximum depth: %d\n", codebook.maxdepth);
|
1060 |
/* read temp vector */
|
1061 |
vec_off = get_vlc2(&vc->gb, codebook.vlc.table, |
1062 |
codebook.nb_bits, codebook.maxdepth) |
1063 |
* codebook.dimensions; |
1064 |
AV_DEBUG("floor0 dec: vector offset: %d\n", vec_off);
|
1065 |
/* copy each vector component and add last to it */
|
1066 |
for (idx = 0; idx < codebook.dimensions; ++idx) |
1067 |
lsp[lsp_len+idx] = codebook.codevectors[vec_off+idx] + last; |
1068 |
last = lsp[lsp_len+idx-1]; /* set last to last vector component */ |
1069 |
|
1070 |
lsp_len += codebook.dimensions; |
1071 |
} |
1072 |
#ifdef V_DEBUG
|
1073 |
/* DEBUG: output lsp coeffs */
|
1074 |
{ |
1075 |
int idx;
|
1076 |
for (idx = 0; idx < lsp_len; ++idx) |
1077 |
AV_DEBUG("floor0 dec: coeff at %d is %f\n", idx, lsp[idx]);
|
1078 |
} |
1079 |
#endif
|
1080 |
|
1081 |
/* synthesize floor output vector */
|
1082 |
{ |
1083 |
int i;
|
1084 |
int order = vf->order;
|
1085 |
float wstep = M_PI / vf->bark_map_size;
|
1086 |
|
1087 |
for (i = 0; i < order; i++) |
1088 |
lsp[i] = 2.0f * cos(lsp[i]); |
1089 |
|
1090 |
AV_DEBUG("floor0 synth: map_size = %d; m = %d; wstep = %f\n",
|
1091 |
vf->map_size, order, wstep); |
1092 |
|
1093 |
i = 0;
|
1094 |
while (i < vf->map_size[blockflag]) {
|
1095 |
int j, iter_cond = vf->map[blockflag][i];
|
1096 |
float p = 0.5f; |
1097 |
float q = 0.5f; |
1098 |
float two_cos_w = 2.0f * cos(wstep * iter_cond); // needed all times |
1099 |
|
1100 |
/* similar part for the q and p products */
|
1101 |
for (j = 0; j + 1 < order; j += 2) { |
1102 |
q *= lsp[j] - two_cos_w; |
1103 |
p *= lsp[j + 1] - two_cos_w;
|
1104 |
} |
1105 |
if (j == order) { // even order |
1106 |
p *= p * (2.0f - two_cos_w); |
1107 |
q *= q * (2.0f + two_cos_w); |
1108 |
} else { // odd order |
1109 |
q *= two_cos_w-lsp[j]; // one more time for q
|
1110 |
|
1111 |
/* final step and square */
|
1112 |
p *= p * (4.f - two_cos_w * two_cos_w);
|
1113 |
q *= q; |
1114 |
} |
1115 |
|
1116 |
/* calculate linear floor value */
|
1117 |
{ |
1118 |
q = exp((((amplitude*vf->amplitude_offset) / |
1119 |
(((1 << vf->amplitude_bits) - 1) * sqrt(p + q))) |
1120 |
- vf->amplitude_offset) * .11512925f);
|
1121 |
} |
1122 |
|
1123 |
/* fill vector */
|
1124 |
do {
|
1125 |
vec[i] = q; ++i; |
1126 |
} while (vf->map[blockflag][i] == iter_cond);
|
1127 |
} |
1128 |
} |
1129 |
} else {
|
1130 |
/* this channel is unused */
|
1131 |
return 1; |
1132 |
} |
1133 |
|
1134 |
AV_DEBUG(" Floor0 decoded\n");
|
1135 |
|
1136 |
return 0; |
1137 |
} |
1138 |
|
1139 |
static int vorbis_floor1_decode(vorbis_context *vc, |
1140 |
vorbis_floor_data *vfu, float *vec)
|
1141 |
{ |
1142 |
vorbis_floor1 *vf = &vfu->t1; |
1143 |
GetBitContext *gb = &vc->gb; |
1144 |
uint_fast16_t range_v[4] = { 256, 128, 86, 64 }; |
1145 |
uint_fast16_t range = range_v[vf->multiplier-1];
|
1146 |
uint_fast16_t floor1_Y[258];
|
1147 |
uint_fast16_t floor1_Y_final[258];
|
1148 |
int floor1_flag[258]; |
1149 |
uint_fast8_t class_; |
1150 |
uint_fast8_t cdim; |
1151 |
uint_fast8_t cbits; |
1152 |
uint_fast8_t csub; |
1153 |
uint_fast8_t cval; |
1154 |
int_fast16_t book; |
1155 |
uint_fast16_t offset; |
1156 |
uint_fast16_t i,j; |
1157 |
int_fast16_t adx, ady, dy, off, predicted; |
1158 |
int_fast32_t err; |
1159 |
|
1160 |
|
1161 |
if (!get_bits1(gb)) // silence |
1162 |
return 1; |
1163 |
|
1164 |
// Read values (or differences) for the floor's points
|
1165 |
|
1166 |
floor1_Y[0] = get_bits(gb, ilog(range - 1)); |
1167 |
floor1_Y[1] = get_bits(gb, ilog(range - 1)); |
1168 |
|
1169 |
AV_DEBUG("floor 0 Y %d floor 1 Y %d \n", floor1_Y[0], floor1_Y[1]); |
1170 |
|
1171 |
offset = 2;
|
1172 |
for (i = 0; i < vf->partitions; ++i) { |
1173 |
class_ = vf->partition_class[i]; |
1174 |
cdim = vf->class_dimensions[class_]; |
1175 |
cbits = vf->class_subclasses[class_]; |
1176 |
csub = (1 << cbits) - 1; |
1177 |
cval = 0;
|
1178 |
|
1179 |
AV_DEBUG("Cbits %d \n", cbits);
|
1180 |
|
1181 |
if (cbits) // this reads all subclasses for this partition's class |
1182 |
cval = get_vlc2(gb, vc->codebooks[vf->class_masterbook[class_]].vlc.table, |
1183 |
vc->codebooks[vf->class_masterbook[class_]].nb_bits, 3);
|
1184 |
|
1185 |
for (j = 0; j < cdim; ++j) { |
1186 |
book = vf->subclass_books[class_][cval & csub]; |
1187 |
|
1188 |
AV_DEBUG("book %d Cbits %d cval %d bits:%d \n", book, cbits, cval, get_bits_count(gb));
|
1189 |
|
1190 |
cval = cval >> cbits; |
1191 |
if (book > -1) { |
1192 |
floor1_Y[offset+j] = get_vlc2(gb, vc->codebooks[book].vlc.table, |
1193 |
vc->codebooks[book].nb_bits, 3);
|
1194 |
} else {
|
1195 |
floor1_Y[offset+j] = 0;
|
1196 |
} |
1197 |
|
1198 |
AV_DEBUG(" floor(%d) = %d \n", vf->list[offset+j].x, floor1_Y[offset+j]);
|
1199 |
} |
1200 |
offset+=cdim; |
1201 |
} |
1202 |
|
1203 |
// Amplitude calculation from the differences
|
1204 |
|
1205 |
floor1_flag[0] = 1; |
1206 |
floor1_flag[1] = 1; |
1207 |
floor1_Y_final[0] = floor1_Y[0]; |
1208 |
floor1_Y_final[1] = floor1_Y[1]; |
1209 |
|
1210 |
for (i = 2; i < vf->x_list_dim; ++i) { |
1211 |
uint_fast16_t val, highroom, lowroom, room; |
1212 |
uint_fast16_t high_neigh_offs; |
1213 |
uint_fast16_t low_neigh_offs; |
1214 |
|
1215 |
low_neigh_offs = vf->list[i].low; |
1216 |
high_neigh_offs = vf->list[i].high; |
1217 |
dy = floor1_Y_final[high_neigh_offs] - floor1_Y_final[low_neigh_offs]; // render_point begin
|
1218 |
adx = vf->list[high_neigh_offs].x - vf->list[low_neigh_offs].x; |
1219 |
ady = FFABS(dy); |
1220 |
err = ady * (vf->list[i].x - vf->list[low_neigh_offs].x); |
1221 |
off = err / adx; |
1222 |
if (dy < 0) { |
1223 |
predicted = floor1_Y_final[low_neigh_offs] - off; |
1224 |
} else {
|
1225 |
predicted = floor1_Y_final[low_neigh_offs] + off; |
1226 |
} // render_point end
|
1227 |
|
1228 |
val = floor1_Y[i]; |
1229 |
highroom = range-predicted; |
1230 |
lowroom = predicted; |
1231 |
if (highroom < lowroom) {
|
1232 |
room = highroom * 2;
|
1233 |
} else {
|
1234 |
room = lowroom * 2; // SPEC mispelling |
1235 |
} |
1236 |
if (val) {
|
1237 |
floor1_flag[low_neigh_offs] = 1;
|
1238 |
floor1_flag[high_neigh_offs] = 1;
|
1239 |
floor1_flag[i] = 1;
|
1240 |
if (val >= room) {
|
1241 |
if (highroom > lowroom) {
|
1242 |
floor1_Y_final[i] = val - lowroom + predicted; |
1243 |
} else {
|
1244 |
floor1_Y_final[i] = predicted - val + highroom - 1;
|
1245 |
} |
1246 |
} else {
|
1247 |
if (val & 1) { |
1248 |
floor1_Y_final[i] = predicted - (val + 1) / 2; |
1249 |
} else {
|
1250 |
floor1_Y_final[i] = predicted + val / 2;
|
1251 |
} |
1252 |
} |
1253 |
} else {
|
1254 |
floor1_flag[i] = 0;
|
1255 |
floor1_Y_final[i] = predicted; |
1256 |
} |
1257 |
|
1258 |
AV_DEBUG(" Decoded floor(%d) = %d / val %d \n", vf->list[i].x, floor1_Y_final[i], val);
|
1259 |
} |
1260 |
|
1261 |
// Curve synth - connect the calculated dots and convert from dB scale FIXME optimize ?
|
1262 |
|
1263 |
ff_vorbis_floor1_render_list(vf->list, vf->x_list_dim, floor1_Y_final, floor1_flag, vf->multiplier, vec, vf->list[1].x);
|
1264 |
|
1265 |
AV_DEBUG(" Floor decoded\n");
|
1266 |
|
1267 |
return 0; |
1268 |
} |
1269 |
|
1270 |
// Read and decode residue
|
1271 |
|
1272 |
static av_always_inline int vorbis_residue_decode_internal(vorbis_context *vc, |
1273 |
vorbis_residue *vr, |
1274 |
uint_fast8_t ch, |
1275 |
uint_fast8_t *do_not_decode, |
1276 |
float *vec,
|
1277 |
uint_fast16_t vlen, |
1278 |
int vr_type)
|
1279 |
{ |
1280 |
GetBitContext *gb = &vc->gb; |
1281 |
uint_fast8_t c_p_c = vc->codebooks[vr->classbook].dimensions; |
1282 |
uint_fast16_t ptns_to_read = vr->ptns_to_read; |
1283 |
uint8_t *classifs = vr->classifs; |
1284 |
uint_fast8_t pass; |
1285 |
uint_fast8_t ch_used; |
1286 |
uint_fast8_t i,j,l; |
1287 |
uint_fast16_t k; |
1288 |
|
1289 |
if (vr_type == 2) { |
1290 |
for (j = 1; j < ch; ++j) |
1291 |
do_not_decode[0] &= do_not_decode[j]; // FIXME - clobbering input |
1292 |
if (do_not_decode[0]) |
1293 |
return 0; |
1294 |
ch_used = 1;
|
1295 |
} else {
|
1296 |
ch_used = ch; |
1297 |
} |
1298 |
|
1299 |
AV_DEBUG(" residue type 0/1/2 decode begin, ch: %d cpc %d \n", ch, c_p_c);
|
1300 |
|
1301 |
for (pass = 0; pass <= vr->maxpass; ++pass) { // FIXME OPTIMIZE? |
1302 |
uint_fast16_t voffset; |
1303 |
uint_fast16_t partition_count; |
1304 |
uint_fast16_t j_times_ptns_to_read; |
1305 |
|
1306 |
voffset = vr->begin; |
1307 |
for (partition_count = 0; partition_count < ptns_to_read;) { // SPEC error |
1308 |
if (!pass) {
|
1309 |
uint_fast32_t inverse_class = ff_inverse[vr->classifications]; |
1310 |
for (j_times_ptns_to_read = 0, j = 0; j < ch_used; ++j) { |
1311 |
if (!do_not_decode[j]) {
|
1312 |
uint_fast32_t temp = get_vlc2(gb, vc->codebooks[vr->classbook].vlc.table, |
1313 |
vc->codebooks[vr->classbook].nb_bits, 3);
|
1314 |
|
1315 |
AV_DEBUG("Classword: %d \n", temp);
|
1316 |
|
1317 |
assert(vr->classifications > 1 && temp <= 65536); //needed for inverse[] |
1318 |
for (i = 0; i < c_p_c; ++i) { |
1319 |
uint_fast32_t temp2; |
1320 |
|
1321 |
temp2 = (((uint_fast64_t)temp) * inverse_class) >> 32;
|
1322 |
if (partition_count + c_p_c - 1 - i < ptns_to_read) |
1323 |
classifs[j_times_ptns_to_read + partition_count + c_p_c - 1 - i] = temp - temp2 * vr->classifications;
|
1324 |
temp = temp2; |
1325 |
} |
1326 |
} |
1327 |
j_times_ptns_to_read += ptns_to_read; |
1328 |
} |
1329 |
} |
1330 |
for (i = 0; (i < c_p_c) && (partition_count < ptns_to_read); ++i) { |
1331 |
for (j_times_ptns_to_read = 0, j = 0; j < ch_used; ++j) { |
1332 |
uint_fast16_t voffs; |
1333 |
|
1334 |
if (!do_not_decode[j]) {
|
1335 |
uint_fast8_t vqclass = classifs[j_times_ptns_to_read+partition_count]; |
1336 |
int_fast16_t vqbook = vr->books[vqclass][pass]; |
1337 |
|
1338 |
if (vqbook >= 0 && vc->codebooks[vqbook].codevectors) { |
1339 |
uint_fast16_t coffs; |
1340 |
unsigned dim = vc->codebooks[vqbook].dimensions; // not uint_fast8_t: 64bit is slower here on amd64 |
1341 |
uint_fast16_t step = dim == 1 ? vr->partition_size
|
1342 |
: FASTDIV(vr->partition_size, dim); |
1343 |
vorbis_codebook codebook = vc->codebooks[vqbook]; |
1344 |
|
1345 |
if (vr_type == 0) { |
1346 |
|
1347 |
voffs = voffset+j*vlen; |
1348 |
for (k = 0; k < step; ++k) { |
1349 |
coffs = get_vlc2(gb, codebook.vlc.table, codebook.nb_bits, 3) * dim;
|
1350 |
for (l = 0; l < dim; ++l) |
1351 |
vec[voffs + k + l * step] += codebook.codevectors[coffs + l]; // FPMATH
|
1352 |
} |
1353 |
} else if (vr_type == 1) { |
1354 |
voffs = voffset + j * vlen; |
1355 |
for (k = 0; k < step; ++k) { |
1356 |
coffs = get_vlc2(gb, codebook.vlc.table, codebook.nb_bits, 3) * dim;
|
1357 |
for (l = 0; l < dim; ++l, ++voffs) { |
1358 |
vec[voffs]+=codebook.codevectors[coffs+l]; // FPMATH
|
1359 |
|
1360 |
AV_DEBUG(" pass %d offs: %d curr: %f change: %f cv offs.: %d \n", pass, voffs, vec[voffs], codebook.codevectors[coffs+l], coffs);
|
1361 |
} |
1362 |
} |
1363 |
} else if (vr_type == 2 && ch == 2 && (voffset & 1) == 0 && (dim & 1) == 0) { // most frequent case optimized |
1364 |
voffs = voffset >> 1;
|
1365 |
|
1366 |
if (dim == 2) { |
1367 |
for (k = 0; k < step; ++k) { |
1368 |
coffs = get_vlc2(gb, codebook.vlc.table, codebook.nb_bits, 3) * 2; |
1369 |
vec[voffs + k ] += codebook.codevectors[coffs ]; // FPMATH
|
1370 |
vec[voffs + k + vlen] += codebook.codevectors[coffs + 1]; // FPMATH |
1371 |
} |
1372 |
} else if (dim == 4) { |
1373 |
for (k = 0; k < step; ++k, voffs += 2) { |
1374 |
coffs = get_vlc2(gb, codebook.vlc.table, codebook.nb_bits, 3) * 4; |
1375 |
vec[voffs ] += codebook.codevectors[coffs ]; // FPMATH
|
1376 |
vec[voffs + 1 ] += codebook.codevectors[coffs + 2]; // FPMATH |
1377 |
vec[voffs + vlen ] += codebook.codevectors[coffs + 1]; // FPMATH |
1378 |
vec[voffs + vlen + 1] += codebook.codevectors[coffs + 3]; // FPMATH |
1379 |
} |
1380 |
} else
|
1381 |
for (k = 0; k < step; ++k) { |
1382 |
coffs = get_vlc2(gb, codebook.vlc.table, codebook.nb_bits, 3) * dim;
|
1383 |
for (l = 0; l < dim; l += 2, voffs++) { |
1384 |
vec[voffs ] += codebook.codevectors[coffs + l ]; // FPMATH
|
1385 |
vec[voffs + vlen] += codebook.codevectors[coffs + l + 1]; // FPMATH |
1386 |
|
1387 |
AV_DEBUG(" pass %d offs: %d curr: %f change: %f cv offs.: %d+%d \n", pass, voffset / ch + (voffs % ch) * vlen, vec[voffset / ch + (voffs % ch) * vlen], codebook.codevectors[coffs + l], coffs, l);
|
1388 |
} |
1389 |
} |
1390 |
|
1391 |
} else if (vr_type == 2) { |
1392 |
voffs = voffset; |
1393 |
|
1394 |
for (k = 0; k < step; ++k) { |
1395 |
coffs = get_vlc2(gb, codebook.vlc.table, codebook.nb_bits, 3) * dim;
|
1396 |
for (l = 0; l < dim; ++l, ++voffs) { |
1397 |
vec[voffs / ch + (voffs % ch) * vlen] += codebook.codevectors[coffs + l]; // FPMATH FIXME use if and counter instead of / and %
|
1398 |
|
1399 |
AV_DEBUG(" pass %d offs: %d curr: %f change: %f cv offs.: %d+%d \n", pass, voffset / ch + (voffs % ch) * vlen, vec[voffset / ch + (voffs % ch) * vlen], codebook.codevectors[coffs + l], coffs, l);
|
1400 |
} |
1401 |
} |
1402 |
} |
1403 |
} |
1404 |
} |
1405 |
j_times_ptns_to_read += ptns_to_read; |
1406 |
} |
1407 |
++partition_count; |
1408 |
voffset += vr->partition_size; |
1409 |
} |
1410 |
} |
1411 |
} |
1412 |
return 0; |
1413 |
} |
1414 |
|
1415 |
static inline int vorbis_residue_decode(vorbis_context *vc, vorbis_residue *vr, |
1416 |
uint_fast8_t ch, |
1417 |
uint_fast8_t *do_not_decode, |
1418 |
float *vec, uint_fast16_t vlen)
|
1419 |
{ |
1420 |
if (vr->type == 2) |
1421 |
return vorbis_residue_decode_internal(vc, vr, ch, do_not_decode, vec, vlen, 2); |
1422 |
else if (vr->type == 1) |
1423 |
return vorbis_residue_decode_internal(vc, vr, ch, do_not_decode, vec, vlen, 1); |
1424 |
else if (vr->type == 0) |
1425 |
return vorbis_residue_decode_internal(vc, vr, ch, do_not_decode, vec, vlen, 0); |
1426 |
else {
|
1427 |
av_log(vc->avccontext, AV_LOG_ERROR, " Invalid residue type while residue decode?! \n");
|
1428 |
return -1; |
1429 |
} |
1430 |
} |
1431 |
|
1432 |
void vorbis_inverse_coupling(float *mag, float *ang, int blocksize) |
1433 |
{ |
1434 |
int i;
|
1435 |
for (i = 0; i < blocksize; i++) { |
1436 |
if (mag[i] > 0.0) { |
1437 |
if (ang[i] > 0.0) { |
1438 |
ang[i] = mag[i] - ang[i]; |
1439 |
} else {
|
1440 |
float temp = ang[i];
|
1441 |
ang[i] = mag[i]; |
1442 |
mag[i] += temp; |
1443 |
} |
1444 |
} else {
|
1445 |
if (ang[i] > 0.0) { |
1446 |
ang[i] += mag[i]; |
1447 |
} else {
|
1448 |
float temp = ang[i];
|
1449 |
ang[i] = mag[i]; |
1450 |
mag[i] -= temp; |
1451 |
} |
1452 |
} |
1453 |
} |
1454 |
} |
1455 |
|
1456 |
static void copy_normalize(float *dst, float *src, int len, int exp_bias, |
1457 |
float add_bias)
|
1458 |
{ |
1459 |
int i;
|
1460 |
if (exp_bias) {
|
1461 |
memcpy(dst, src, len * sizeof(float)); |
1462 |
} else {
|
1463 |
for (i = 0; i < len; i++) |
1464 |
dst[i] = src[i] + add_bias; |
1465 |
} |
1466 |
} |
1467 |
|
1468 |
// Decode the audio packet using the functions above
|
1469 |
|
1470 |
static int vorbis_parse_audio_packet(vorbis_context *vc) |
1471 |
{ |
1472 |
GetBitContext *gb = &vc->gb; |
1473 |
|
1474 |
uint_fast8_t previous_window = vc->previous_window; |
1475 |
uint_fast8_t mode_number; |
1476 |
uint_fast8_t blockflag; |
1477 |
uint_fast16_t blocksize; |
1478 |
int_fast32_t i,j; |
1479 |
uint_fast8_t no_residue[255];
|
1480 |
uint_fast8_t do_not_decode[255];
|
1481 |
vorbis_mapping *mapping; |
1482 |
float *ch_res_ptr = vc->channel_residues;
|
1483 |
float *ch_floor_ptr = vc->channel_floors;
|
1484 |
uint_fast8_t res_chan[255];
|
1485 |
uint_fast8_t res_num = 0;
|
1486 |
int_fast16_t retlen = 0;
|
1487 |
float fadd_bias = vc->add_bias;
|
1488 |
|
1489 |
if (get_bits1(gb)) {
|
1490 |
av_log(vc->avccontext, AV_LOG_ERROR, "Not a Vorbis I audio packet.\n");
|
1491 |
return -1; // packet type not audio |
1492 |
} |
1493 |
|
1494 |
if (vc->mode_count == 1) { |
1495 |
mode_number = 0;
|
1496 |
} else {
|
1497 |
GET_VALIDATED_INDEX(mode_number, ilog(vc->mode_count-1), vc->mode_count)
|
1498 |
} |
1499 |
vc->mode_number = mode_number; |
1500 |
mapping = &vc->mappings[vc->modes[mode_number].mapping]; |
1501 |
|
1502 |
AV_DEBUG(" Mode number: %d , mapping: %d , blocktype %d \n", mode_number, vc->modes[mode_number].mapping, vc->modes[mode_number].blockflag);
|
1503 |
|
1504 |
blockflag = vc->modes[mode_number].blockflag; |
1505 |
blocksize = vc->blocksize[blockflag]; |
1506 |
if (blockflag)
|
1507 |
skip_bits(gb, 2); // previous_window, next_window |
1508 |
|
1509 |
memset(ch_res_ptr, 0, sizeof(float) * vc->audio_channels * blocksize / 2); //FIXME can this be removed ? |
1510 |
memset(ch_floor_ptr, 0, sizeof(float) * vc->audio_channels * blocksize / 2); //FIXME can this be removed ? |
1511 |
|
1512 |
// Decode floor
|
1513 |
|
1514 |
for (i = 0; i < vc->audio_channels; ++i) { |
1515 |
vorbis_floor *floor; |
1516 |
int ret;
|
1517 |
if (mapping->submaps > 1) { |
1518 |
floor = &vc->floors[mapping->submap_floor[mapping->mux[i]]]; |
1519 |
} else {
|
1520 |
floor = &vc->floors[mapping->submap_floor[0]];
|
1521 |
} |
1522 |
|
1523 |
ret = floor->decode(vc, &floor->data, ch_floor_ptr); |
1524 |
|
1525 |
if (ret < 0) { |
1526 |
av_log(vc->avccontext, AV_LOG_ERROR, "Invalid codebook in vorbis_floor_decode.\n");
|
1527 |
return -1; |
1528 |
} |
1529 |
no_residue[i] = ret; |
1530 |
ch_floor_ptr += blocksize / 2;
|
1531 |
} |
1532 |
|
1533 |
// Nonzero vector propagate
|
1534 |
|
1535 |
for (i = mapping->coupling_steps - 1; i >= 0; --i) { |
1536 |
if (!(no_residue[mapping->magnitude[i]] & no_residue[mapping->angle[i]])) {
|
1537 |
no_residue[mapping->magnitude[i]] = 0;
|
1538 |
no_residue[mapping->angle[i]] = 0;
|
1539 |
} |
1540 |
} |
1541 |
|
1542 |
// Decode residue
|
1543 |
|
1544 |
for (i = 0; i < mapping->submaps; ++i) { |
1545 |
vorbis_residue *residue; |
1546 |
uint_fast8_t ch = 0;
|
1547 |
|
1548 |
for (j = 0; j < vc->audio_channels; ++j) { |
1549 |
if ((mapping->submaps == 1) || (i == mapping->mux[j])) { |
1550 |
res_chan[j] = res_num; |
1551 |
if (no_residue[j]) {
|
1552 |
do_not_decode[ch] = 1;
|
1553 |
} else {
|
1554 |
do_not_decode[ch] = 0;
|
1555 |
} |
1556 |
++ch; |
1557 |
++res_num; |
1558 |
} |
1559 |
} |
1560 |
residue = &vc->residues[mapping->submap_residue[i]]; |
1561 |
vorbis_residue_decode(vc, residue, ch, do_not_decode, ch_res_ptr, blocksize/2);
|
1562 |
|
1563 |
ch_res_ptr += ch * blocksize / 2;
|
1564 |
} |
1565 |
|
1566 |
// Inverse coupling
|
1567 |
|
1568 |
for (i = mapping->coupling_steps - 1; i >= 0; --i) { //warning: i has to be signed |
1569 |
float *mag, *ang;
|
1570 |
|
1571 |
mag = vc->channel_residues+res_chan[mapping->magnitude[i]] * blocksize / 2;
|
1572 |
ang = vc->channel_residues+res_chan[mapping->angle[i]] * blocksize / 2;
|
1573 |
vc->dsp.vorbis_inverse_coupling(mag, ang, blocksize / 2);
|
1574 |
} |
1575 |
|
1576 |
// Dotproduct, MDCT
|
1577 |
|
1578 |
for (j = vc->audio_channels-1;j >= 0; j--) { |
1579 |
ch_floor_ptr = vc->channel_floors + j * blocksize / 2;
|
1580 |
ch_res_ptr = vc->channel_residues + res_chan[j] * blocksize / 2;
|
1581 |
vc->dsp.vector_fmul(ch_floor_ptr, ch_res_ptr, blocksize / 2);
|
1582 |
ff_imdct_half(&vc->mdct[blockflag], ch_res_ptr, ch_floor_ptr); |
1583 |
} |
1584 |
|
1585 |
// Overlap/add, save data for next overlapping FPMATH
|
1586 |
|
1587 |
retlen = (blocksize + vc->blocksize[previous_window]) / 4;
|
1588 |
for (j = 0; j < vc->audio_channels; j++) { |
1589 |
uint_fast16_t bs0 = vc->blocksize[0];
|
1590 |
uint_fast16_t bs1 = vc->blocksize[1];
|
1591 |
float *residue = vc->channel_residues + res_chan[j] * blocksize / 2; |
1592 |
float *saved = vc->saved + j * bs1 / 4; |
1593 |
float *ret = vc->channel_floors + j * retlen;
|
1594 |
float *buf = residue;
|
1595 |
const float *win = vc->win[blockflag & previous_window]; |
1596 |
|
1597 |
if (blockflag == previous_window) {
|
1598 |
vc->dsp.vector_fmul_window(ret, saved, buf, win, fadd_bias, blocksize / 4);
|
1599 |
} else if (blockflag > previous_window) { |
1600 |
vc->dsp.vector_fmul_window(ret, saved, buf, win, fadd_bias, bs0 / 4);
|
1601 |
copy_normalize(ret+bs0/2, buf+bs0/4, (bs1-bs0)/4, vc->exp_bias, fadd_bias); |
1602 |
} else {
|
1603 |
copy_normalize(ret, saved, (bs1 - bs0) / 4, vc->exp_bias, fadd_bias);
|
1604 |
vc->dsp.vector_fmul_window(ret + (bs1 - bs0) / 4, saved + (bs1 - bs0) / 4, buf, win, fadd_bias, bs0 / 4); |
1605 |
} |
1606 |
memcpy(saved, buf + blocksize / 4, blocksize / 4 * sizeof(float)); |
1607 |
} |
1608 |
|
1609 |
vc->previous_window = blockflag; |
1610 |
return retlen;
|
1611 |
} |
1612 |
|
1613 |
// Return the decoded audio packet through the standard api
|
1614 |
|
1615 |
static int vorbis_decode_frame(AVCodecContext *avccontext, |
1616 |
void *data, int *data_size, |
1617 |
AVPacket *avpkt) |
1618 |
{ |
1619 |
const uint8_t *buf = avpkt->data;
|
1620 |
int buf_size = avpkt->size;
|
1621 |
vorbis_context *vc = avccontext->priv_data ; |
1622 |
GetBitContext *gb = &(vc->gb); |
1623 |
const float *channel_ptrs[255]; |
1624 |
int i;
|
1625 |
|
1626 |
int_fast16_t len; |
1627 |
|
1628 |
if (!buf_size)
|
1629 |
return 0; |
1630 |
|
1631 |
AV_DEBUG("packet length %d \n", buf_size);
|
1632 |
|
1633 |
init_get_bits(gb, buf, buf_size*8);
|
1634 |
|
1635 |
len = vorbis_parse_audio_packet(vc); |
1636 |
|
1637 |
if (len <= 0) { |
1638 |
*data_size = 0;
|
1639 |
return buf_size;
|
1640 |
} |
1641 |
|
1642 |
if (!vc->first_frame) {
|
1643 |
vc->first_frame = 1;
|
1644 |
*data_size = 0;
|
1645 |
return buf_size ;
|
1646 |
} |
1647 |
|
1648 |
AV_DEBUG("parsed %d bytes %d bits, returned %d samples (*ch*bits) \n", get_bits_count(gb)/8, get_bits_count(gb)%8, len); |
1649 |
|
1650 |
if (vc->audio_channels > 8) { |
1651 |
for (i = 0; i < vc->audio_channels; i++) |
1652 |
channel_ptrs[i] = vc->channel_floors + i * len; |
1653 |
} else {
|
1654 |
for (i = 0; i < vc->audio_channels; i++) |
1655 |
channel_ptrs[i] = vc->channel_floors + |
1656 |
len * ff_vorbis_channel_layout_offsets[vc->audio_channels - 1][i];
|
1657 |
} |
1658 |
|
1659 |
vc->dsp.float_to_int16_interleave(data, channel_ptrs, len, vc->audio_channels); |
1660 |
*data_size = len * 2 * vc->audio_channels;
|
1661 |
|
1662 |
return buf_size ;
|
1663 |
} |
1664 |
|
1665 |
// Close decoder
|
1666 |
|
1667 |
static av_cold int vorbis_decode_close(AVCodecContext *avccontext) |
1668 |
{ |
1669 |
vorbis_context *vc = avccontext->priv_data; |
1670 |
|
1671 |
vorbis_free(vc); |
1672 |
|
1673 |
return 0 ; |
1674 |
} |
1675 |
|
1676 |
AVCodec vorbis_decoder = { |
1677 |
"vorbis",
|
1678 |
AVMEDIA_TYPE_AUDIO, |
1679 |
CODEC_ID_VORBIS, |
1680 |
sizeof(vorbis_context),
|
1681 |
vorbis_decode_init, |
1682 |
NULL,
|
1683 |
vorbis_decode_close, |
1684 |
vorbis_decode_frame, |
1685 |
.long_name = NULL_IF_CONFIG_SMALL("Vorbis"),
|
1686 |
.channel_layouts = ff_vorbis_channel_layouts, |
1687 |
}; |
1688 |
|