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