You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1463 lines
50KB

  1. /**
  2. * @file vorbis.c
  3. * Vorbis I decoder
  4. * @author Denes Balatoni ( dbalatoni programozo hu )
  5. * This library is free software; you can redistribute it and/or
  6. * modify it under the terms of the GNU Lesser General Public
  7. * License as published by the Free Software Foundation; either
  8. * version 2 of the License, or (at your option) any later version.
  9. *
  10. * This library is distributed in the hope that it will be useful,
  11. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  13. * Lesser General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU Lesser General Public
  16. * License along with this library; if not, write to the Free Software
  17. * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  18. *
  19. */
  20. #undef V_DEBUG
  21. #include <math.h>
  22. #define ALT_BITSTREAM_READER_LE
  23. #include "avcodec.h"
  24. #include "bitstream.h"
  25. #include "dsputil.h"
  26. #include "vorbis.h"
  27. #define V_NB_BITS 11
  28. #define V_MAX_VLCS (1<<16)
  29. #ifndef V_DEBUG
  30. #define AV_DEBUG(...)
  31. #endif
  32. /* Helper functions */
  33. /**
  34. * reads 0-32 bits when using the ALT_BITSTREAM_READER_LE bitstream reader
  35. */
  36. unsigned int get_bits_long_le(GetBitContext *s, int n){
  37. if(n<=17) return get_bits(s, n);
  38. else{
  39. int ret= get_bits(s, 16);
  40. return ret | (get_bits(s, n-16) << 16);
  41. }
  42. }
  43. static unsigned int ilog(unsigned int i) { // unfortunatelly av_log2 uses different rounding
  44. unsigned int ret=0;
  45. while (i!=0) {
  46. ++ret;
  47. i>>=1;
  48. }
  49. return ret;
  50. }
  51. static unsigned int nth_root(unsigned int x, unsigned int n) { // x^(1/n)
  52. unsigned int ret=0, i, j;
  53. do {
  54. ++ret;
  55. for(i=0,j=ret;i<n-1;i++) j*=ret;
  56. } while (j<=x);
  57. return (ret-1);
  58. }
  59. static float vorbisfloat2float(uint_fast32_t val) {
  60. double mant=val&0x1fffff;
  61. long exp=(val&0x7fe00000L)>>21;
  62. if (val&0x80000000) mant=-mant;
  63. return(ldexp(mant, exp-20-768));
  64. }
  65. // Generate vlc codes from vorbis huffman code lengths
  66. static int vorbis_len2vlc(vorbis_context *vc, uint_fast8_t *bits, uint_fast32_t *codes, uint_fast32_t num) {
  67. uint_fast32_t exit_at_level[33]={404,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,
  68. 0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
  69. uint_fast8_t i,j;
  70. uint_fast32_t code,p;
  71. #ifdef V_DEBUG
  72. GetBitContext gb;
  73. #endif
  74. for(p=0;(bits[p]==0) && (p<num);++p);
  75. if (p==num) {
  76. // av_log(vc->avccontext, AV_LOG_INFO, "An empty codebook. Heh?! \n");
  77. return 0;
  78. }
  79. codes[p]=0;
  80. for(i=0;i<bits[p];++i) {
  81. exit_at_level[i+1]=1<<i;
  82. }
  83. #ifdef V_DEBUG
  84. av_log(vc->avccontext, AV_LOG_INFO, " %d. of %d code len %d code %d - ", p, num, bits[p], codes[p]);
  85. init_get_bits(&gb, (uint_fast8_t *)&codes[p], bits[p]);
  86. for(i=0;i<bits[p];++i) {
  87. av_log(vc->avccontext, AV_LOG_INFO, "%s", get_bits1(&gb) ? "1" : "0");
  88. }
  89. av_log(vc->avccontext, AV_LOG_INFO, "\n");
  90. #endif
  91. ++p;
  92. for(;p<num;++p) {
  93. if (bits[p]==0) continue;
  94. // find corresponding exit(node which the tree can grow further from)
  95. for(i=bits[p];i>0;--i) {
  96. if (exit_at_level[i]) break;
  97. }
  98. if (!i) return 1; // overspecified tree
  99. code=exit_at_level[i];
  100. exit_at_level[i]=0;
  101. // construct code (append 0s to end) and introduce new exits
  102. for(j=i+1;j<=bits[p];++j) {
  103. exit_at_level[j]=code+(1<<(j-1));
  104. }
  105. codes[p]=code;
  106. #ifdef V_DEBUG
  107. av_log(vc->avccontext, AV_LOG_INFO, " %d. code len %d code %d - ", p, bits[p], codes[p]);
  108. init_get_bits(&gb, (uint_fast8_t *)&codes[p], bits[p]);
  109. for(i=0;i<bits[p];++i) {
  110. av_log(vc->avccontext, AV_LOG_INFO, "%s", get_bits1(&gb) ? "1" : "0");
  111. }
  112. av_log(vc->avccontext, AV_LOG_INFO, "\n");
  113. #endif
  114. }
  115. //FIXME no exits should be left (underspecified tree - ie. unused valid vlcs - not allowed by SPEC)
  116. return 0;
  117. }
  118. // Free all allocated memory -----------------------------------------
  119. static void vorbis_free(vorbis_context *vc) {
  120. int_fast16_t i;
  121. av_freep(&vc->channel_residues);
  122. av_freep(&vc->channel_floors);
  123. av_freep(&vc->saved);
  124. av_freep(&vc->ret);
  125. av_freep(&vc->buf);
  126. av_freep(&vc->buf_tmp);
  127. av_freep(&vc->residues);
  128. av_freep(&vc->modes);
  129. ff_mdct_end(&vc->mdct0);
  130. ff_mdct_end(&vc->mdct1);
  131. for(i=0;i<vc->codebook_count;++i) {
  132. av_free(vc->codebooks[i].codevectors);
  133. free_vlc(&vc->codebooks[i].vlc);
  134. }
  135. av_freep(&vc->codebooks);
  136. for(i=0;i<vc->floor_count;++i) {
  137. av_free(vc->floors[i].x_list);
  138. av_free(vc->floors[i].x_list_order);
  139. av_free(vc->floors[i].low_neighbour);
  140. av_free(vc->floors[i].high_neighbour);
  141. }
  142. av_freep(&vc->floors);
  143. for(i=0;i<vc->mapping_count;++i) {
  144. av_free(vc->mappings[i].magnitude);
  145. av_free(vc->mappings[i].angle);
  146. av_free(vc->mappings[i].mux);
  147. }
  148. av_freep(&vc->mappings);
  149. }
  150. // Parse setup header -------------------------------------------------
  151. // Process codebooks part
  152. static int vorbis_parse_setup_hdr_codebooks(vorbis_context *vc) {
  153. uint_fast16_t cb;
  154. uint_fast8_t *tmp_vlc_bits;
  155. uint_fast32_t *tmp_vlc_codes;
  156. GetBitContext *gb=&vc->gb;
  157. vc->codebook_count=get_bits(gb,8)+1;
  158. AV_DEBUG(" Codebooks: %d \n", vc->codebook_count);
  159. vc->codebooks=(vorbis_codebook *)av_mallocz(vc->codebook_count * sizeof(vorbis_codebook));
  160. tmp_vlc_bits=(uint_fast8_t *)av_mallocz(V_MAX_VLCS * sizeof(uint_fast8_t));
  161. tmp_vlc_codes=(uint_fast32_t *)av_mallocz(V_MAX_VLCS * sizeof(uint_fast32_t));
  162. for(cb=0;cb<vc->codebook_count;++cb) {
  163. vorbis_codebook *codebook_setup=&vc->codebooks[cb];
  164. uint_fast8_t ordered;
  165. uint_fast32_t t, used_entries=0;
  166. uint_fast32_t entries;
  167. AV_DEBUG(" %d. Codebook \n", cb);
  168. if (get_bits(gb, 24)!=0x564342) {
  169. av_log(vc->avccontext, AV_LOG_ERROR, " %d. Codebook setup data corrupt. \n", cb);
  170. goto error;
  171. }
  172. codebook_setup->dimensions=get_bits(gb, 16);
  173. if (codebook_setup->dimensions>16) {
  174. av_log(vc->avccontext, AV_LOG_ERROR, " %d. Codebook's dimension is too large (%d). \n", cb, codebook_setup->dimensions);
  175. goto error;
  176. }
  177. entries=get_bits(gb, 24);
  178. if (entries>V_MAX_VLCS) {
  179. av_log(vc->avccontext, AV_LOG_ERROR, " %d. Codebook has too many entries (%d). \n", cb, entries);
  180. goto error;
  181. }
  182. ordered=get_bits1(gb);
  183. AV_DEBUG(" codebook_dimensions %d, codebook_entries %d \n", codebook_setup->dimensions, entries);
  184. if (!ordered) {
  185. uint_fast16_t ce;
  186. uint_fast8_t flag;
  187. uint_fast8_t sparse=get_bits1(gb);
  188. AV_DEBUG(" not ordered \n");
  189. if (sparse) {
  190. AV_DEBUG(" sparse \n");
  191. used_entries=0;
  192. for(ce=0;ce<entries;++ce) {
  193. flag=get_bits1(gb);
  194. if (flag) {
  195. tmp_vlc_bits[ce]=get_bits(gb, 5)+1;
  196. ++used_entries;
  197. }
  198. else tmp_vlc_bits[ce]=0;
  199. }
  200. } else {
  201. AV_DEBUG(" not sparse \n");
  202. used_entries=entries;
  203. for(ce=0;ce<entries;++ce) {
  204. tmp_vlc_bits[ce]=get_bits(gb, 5)+1;
  205. }
  206. }
  207. } else {
  208. uint_fast16_t current_entry=0;
  209. uint_fast8_t current_length=get_bits(gb, 5)+1;
  210. AV_DEBUG(" ordered, current length: %d \n", current_length); //FIXME
  211. used_entries=entries;
  212. for(;current_entry<used_entries;++current_length) {
  213. uint_fast16_t i, number;
  214. AV_DEBUG(" number bits: %d ", ilog(entries - current_entry));
  215. number=get_bits(gb, ilog(entries - current_entry));
  216. AV_DEBUG(" number: %d \n", number);
  217. for(i=current_entry;i<number+current_entry;++i) {
  218. if (i<used_entries) tmp_vlc_bits[i]=current_length;
  219. }
  220. current_entry+=number;
  221. }
  222. if (current_entry>used_entries) {
  223. av_log(vc->avccontext, AV_LOG_ERROR, " More codelengths than codes in codebook. \n");
  224. goto error;
  225. }
  226. }
  227. codebook_setup->lookup_type=get_bits(gb, 4);
  228. AV_DEBUG(" lookup type: %d : %s \n", codebook_setup->lookup_type, codebook_setup->lookup_type ? "vq" : "no lookup" );
  229. // If the codebook is used for (inverse) VQ, calculate codevectors.
  230. if (codebook_setup->lookup_type==1) {
  231. uint_fast16_t i, j, k;
  232. uint_fast16_t codebook_lookup_values=nth_root(entries, codebook_setup->dimensions);
  233. uint_fast16_t codebook_multiplicands[codebook_lookup_values];
  234. float codebook_minimum_value=vorbisfloat2float(get_bits_long_le(gb, 32));
  235. float codebook_delta_value=vorbisfloat2float(get_bits_long_le(gb, 32));
  236. uint_fast8_t codebook_value_bits=get_bits(gb, 4)+1;
  237. uint_fast8_t codebook_sequence_p=get_bits1(gb);
  238. AV_DEBUG(" We expect %d numbers for building the codevectors. \n", codebook_lookup_values);
  239. AV_DEBUG(" delta %f minmum %f \n", codebook_delta_value, codebook_minimum_value);
  240. for(i=0;i<codebook_lookup_values;++i) {
  241. codebook_multiplicands[i]=get_bits(gb, codebook_value_bits);
  242. AV_DEBUG(" multiplicands*delta+minmum : %e \n", (float)codebook_multiplicands[i]*codebook_delta_value+codebook_minimum_value);
  243. AV_DEBUG(" multiplicand %d \n", codebook_multiplicands[i]);
  244. }
  245. // Weed out unused vlcs and build codevector vector
  246. codebook_setup->codevectors=(float *)av_mallocz(used_entries*codebook_setup->dimensions * sizeof(float));
  247. for(j=0, i=0;i<entries;++i) {
  248. uint_fast8_t dim=codebook_setup->dimensions;
  249. if (tmp_vlc_bits[i]) {
  250. float last=0.0;
  251. uint_fast32_t lookup_offset=i;
  252. #ifdef V_DEBUG
  253. av_log(vc->avccontext, AV_LOG_INFO, "Lookup offset %d ,", i);
  254. #endif
  255. for(k=0;k<dim;++k) {
  256. uint_fast32_t multiplicand_offset = lookup_offset % codebook_lookup_values;
  257. codebook_setup->codevectors[j*dim+k]=codebook_multiplicands[multiplicand_offset]*codebook_delta_value+codebook_minimum_value+last;
  258. if (codebook_sequence_p) {
  259. last=codebook_setup->codevectors[j*dim+k];
  260. }
  261. lookup_offset/=codebook_lookup_values;
  262. }
  263. tmp_vlc_bits[j]=tmp_vlc_bits[i];
  264. #ifdef V_DEBUG
  265. av_log(vc->avccontext, AV_LOG_INFO, "real lookup offset %d, vector: ", j);
  266. for(k=0;k<dim;++k) {
  267. av_log(vc->avccontext, AV_LOG_INFO, " %f ", codebook_setup->codevectors[j*dim+k]);
  268. }
  269. av_log(vc->avccontext, AV_LOG_INFO, "\n");
  270. #endif
  271. ++j;
  272. }
  273. }
  274. if (j!=used_entries) {
  275. av_log(vc->avccontext, AV_LOG_ERROR, "Bug in codevector vector building code. \n");
  276. goto error;
  277. }
  278. entries=used_entries;
  279. }
  280. else if (codebook_setup->lookup_type>=2) {
  281. av_log(vc->avccontext, AV_LOG_ERROR, "Codebook lookup type not supported. \n");
  282. goto error;
  283. }
  284. // Initialize VLC table
  285. if (vorbis_len2vlc(vc, tmp_vlc_bits, tmp_vlc_codes, entries)) {
  286. av_log(vc->avccontext, AV_LOG_ERROR, " Invalid code lengths while generating vlcs. \n");
  287. goto error;
  288. }
  289. codebook_setup->maxdepth=0;
  290. for(t=0;t<entries;++t)
  291. if (tmp_vlc_bits[t]>=codebook_setup->maxdepth) codebook_setup->maxdepth=tmp_vlc_bits[t];
  292. codebook_setup->maxdepth=(codebook_setup->maxdepth+V_NB_BITS-1)/V_NB_BITS;
  293. if (init_vlc(&codebook_setup->vlc, V_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)) {
  294. av_log(vc->avccontext, AV_LOG_ERROR, " Error generating vlc tables. \n");
  295. goto error;
  296. }
  297. }
  298. av_free(tmp_vlc_bits);
  299. av_free(tmp_vlc_codes);
  300. return 0;
  301. // Error:
  302. error:
  303. av_free(tmp_vlc_bits);
  304. av_free(tmp_vlc_codes);
  305. return 1;
  306. }
  307. // Process time domain transforms part (unused in Vorbis I)
  308. static int vorbis_parse_setup_hdr_tdtransforms(vorbis_context *vc) {
  309. GetBitContext *gb=&vc->gb;
  310. uint_fast8_t i;
  311. uint_fast8_t vorbis_time_count=get_bits(gb, 6)+1;
  312. for(i=0;i<vorbis_time_count;++i) {
  313. uint_fast16_t vorbis_tdtransform=get_bits(gb, 16);
  314. AV_DEBUG(" Vorbis time domain transform %d: %d \n", vorbis_time_count, vorbis_tdtransform);
  315. if (vorbis_tdtransform) {
  316. av_log(vc->avccontext, AV_LOG_ERROR, "Vorbis time domain transform data nonzero. \n");
  317. return 1;
  318. }
  319. }
  320. return 0;
  321. }
  322. // Process floors part - only floor type 1 is supported
  323. static int vorbis_parse_setup_hdr_floors(vorbis_context *vc) {
  324. GetBitContext *gb=&vc->gb;
  325. uint_fast16_t i,j,k;
  326. vc->floor_count=get_bits(gb, 6)+1;
  327. vc->floors=(vorbis_floor *)av_mallocz(vc->floor_count * sizeof(vorbis_floor));
  328. for (i=0;i<vc->floor_count;++i) {
  329. vorbis_floor *floor_setup=&vc->floors[i];
  330. floor_setup->floor_type=get_bits(gb, 16);
  331. AV_DEBUG(" %d. floor type %d \n", i, floor_setup->floor_type);
  332. if (floor_setup->floor_type==1) {
  333. uint_fast8_t maximum_class=0;
  334. uint_fast8_t rangebits;
  335. uint_fast16_t floor1_values=2;
  336. floor_setup->partitions=get_bits(gb, 5);
  337. AV_DEBUG(" %d.floor: %d partitions \n", i, floor_setup->partitions);
  338. for(j=0;j<floor_setup->partitions;++j) {
  339. floor_setup->partition_class[j]=get_bits(gb, 4);
  340. if (floor_setup->partition_class[j]>maximum_class) maximum_class=floor_setup->partition_class[j];
  341. AV_DEBUG(" %d. floor %d partition class %d \n", i, j, floor_setup->partition_class[j]);
  342. }
  343. AV_DEBUG(" maximum class %d \n", maximum_class);
  344. floor_setup->maximum_class=maximum_class;
  345. for(j=0;j<=maximum_class;++j) {
  346. floor_setup->class_dimensions[j]=get_bits(gb, 3)+1;
  347. floor_setup->class_subclasses[j]=get_bits(gb, 2);
  348. AV_DEBUG(" %d floor %d class dim: %d subclasses %d \n", i, j, floor_setup->class_dimensions[j], floor_setup->class_subclasses[j]);
  349. if (floor_setup->class_subclasses[j]) {
  350. floor_setup->class_masterbook[j]=get_bits(gb, 8);
  351. AV_DEBUG(" masterbook: %d \n", floor_setup->class_masterbook[j]);
  352. }
  353. for(k=0;k<(1<<floor_setup->class_subclasses[j]);++k) {
  354. floor_setup->subclass_books[j][k]=get_bits(gb, 8)-1;
  355. AV_DEBUG(" book %d. : %d \n", k, floor_setup->subclass_books[j][k]);
  356. }
  357. }
  358. floor_setup->multiplier=get_bits(gb, 2)+1;
  359. floor_setup->x_list_dim=2;
  360. for(j=0;j<floor_setup->partitions;++j) {
  361. floor_setup->x_list_dim+=floor_setup->class_dimensions[floor_setup->partition_class[j]];
  362. }
  363. floor_setup->x_list=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));
  364. floor_setup->x_list_order=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));
  365. floor_setup->low_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));
  366. floor_setup->high_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));
  367. rangebits=get_bits(gb, 4);
  368. floor_setup->x_list[0] = 0;
  369. floor_setup->x_list[1] = (1<<rangebits);
  370. for(j=0;j<floor_setup->partitions;++j) {
  371. for(k=0;k<floor_setup->class_dimensions[floor_setup->partition_class[j]];++k,++floor1_values) {
  372. floor_setup->x_list[floor1_values]=get_bits(gb, rangebits);
  373. AV_DEBUG(" %d. floor1 Y coord. %d \n", floor1_values, floor_setup->x_list[floor1_values]);
  374. }
  375. }
  376. // Precalculate order of x coordinates - needed for decode
  377. for(k=0;k<floor_setup->x_list_dim;++k) {
  378. floor_setup->x_list_order[k]=k;
  379. }
  380. for(k=0;k<floor_setup->x_list_dim-1;++k) { // FIXME optimize sorting ?
  381. for(j=k+1;j<floor_setup->x_list_dim;++j) {
  382. if(floor_setup->x_list[floor_setup->x_list_order[k]]>floor_setup->x_list[floor_setup->x_list_order[j]]) {
  383. uint_fast16_t tmp=floor_setup->x_list_order[k];
  384. floor_setup->x_list_order[k]=floor_setup->x_list_order[j];
  385. floor_setup->x_list_order[j]=tmp;
  386. }
  387. }
  388. }
  389. // Precalculate low and high neighbours
  390. for(k=2;k<floor_setup->x_list_dim;++k) {
  391. floor_setup->low_neighbour[k]=0;
  392. floor_setup->high_neighbour[k]=1; // correct according to SPEC requirements
  393. for (j=0;j<k;++j) {
  394. if ((floor_setup->x_list[j]<floor_setup->x_list[k]) &&
  395. (floor_setup->x_list[j]>floor_setup->x_list[floor_setup->low_neighbour[k]])) {
  396. floor_setup->low_neighbour[k]=j;
  397. }
  398. if ((floor_setup->x_list[j]>floor_setup->x_list[k]) &&
  399. (floor_setup->x_list[j]<floor_setup->x_list[floor_setup->high_neighbour[k]])) {
  400. floor_setup->high_neighbour[k]=j;
  401. }
  402. }
  403. }
  404. }
  405. else {
  406. av_log(vc->avccontext, AV_LOG_ERROR, "Only floor type 1 supported. \n");
  407. return 1;
  408. }
  409. }
  410. return 0;
  411. }
  412. // Process residues part
  413. static int vorbis_parse_setup_hdr_residues(vorbis_context *vc){
  414. GetBitContext *gb=&vc->gb;
  415. uint_fast8_t i, j, k;
  416. vc->residue_count=get_bits(gb, 6)+1;
  417. vc->residues=(vorbis_residue *)av_mallocz(vc->residue_count * sizeof(vorbis_residue));
  418. AV_DEBUG(" There are %d residues. \n", vc->residue_count);
  419. for(i=0;i<vc->residue_count;++i) {
  420. vorbis_residue *res_setup=&vc->residues[i];
  421. uint_fast8_t cascade[64];
  422. uint_fast8_t high_bits;
  423. uint_fast8_t low_bits;
  424. res_setup->type=get_bits(gb, 16);
  425. AV_DEBUG(" %d. residue type %d \n", i, res_setup->type);
  426. res_setup->begin=get_bits(gb, 24);
  427. res_setup->end=get_bits(gb, 24);
  428. res_setup->partition_size=get_bits(gb, 24)+1;
  429. res_setup->classifications=get_bits(gb, 6)+1;
  430. res_setup->classbook=get_bits(gb, 8);
  431. 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,
  432. res_setup->classifications, res_setup->classbook);
  433. for(j=0;j<res_setup->classifications;++j) {
  434. high_bits=0;
  435. low_bits=get_bits(gb, 3);
  436. if (get_bits1(gb)) {
  437. high_bits=get_bits(gb, 5);
  438. }
  439. cascade[j]=(high_bits<<3)+low_bits;
  440. AV_DEBUG(" %d class casscade depth: %d \n", j, ilog(cascade[j]));
  441. }
  442. res_setup->maxpass=0;
  443. for(j=0;j<res_setup->classifications;++j) {
  444. for(k=0;k<8;++k) {
  445. if (cascade[j]&(1<<k)) {
  446. res_setup->books[j][k]=get_bits(gb, 8);
  447. AV_DEBUG(" %d class casscade depth %d book: %d \n", j, k, res_setup->books[j][k]);
  448. if (k>res_setup->maxpass) {
  449. res_setup->maxpass=k;
  450. }
  451. } else {
  452. res_setup->books[j][k]=-1;
  453. }
  454. }
  455. }
  456. }
  457. return 0;
  458. }
  459. // Process mappings part
  460. static int vorbis_parse_setup_hdr_mappings(vorbis_context *vc) {
  461. GetBitContext *gb=&vc->gb;
  462. uint_fast8_t i, j;
  463. vc->mapping_count=get_bits(gb, 6)+1;
  464. vc->mappings=(vorbis_mapping *)av_mallocz(vc->mapping_count * sizeof(vorbis_mapping));
  465. AV_DEBUG(" There are %d mappings. \n", vc->mapping_count);
  466. for(i=0;i<vc->mapping_count;++i) {
  467. vorbis_mapping *mapping_setup=&vc->mappings[i];
  468. if (get_bits(gb, 16)) {
  469. av_log(vc->avccontext, AV_LOG_ERROR, "Other mappings than type 0 are not compliant with the Vorbis I specification. \n");
  470. return 1;
  471. }
  472. if (get_bits1(gb)) {
  473. mapping_setup->submaps=get_bits(gb, 4)+1;
  474. } else {
  475. mapping_setup->submaps=1;
  476. }
  477. if (get_bits1(gb)) {
  478. mapping_setup->coupling_steps=get_bits(gb, 8)+1;
  479. mapping_setup->magnitude=(uint_fast8_t *)av_mallocz(mapping_setup->coupling_steps * sizeof(uint_fast8_t));
  480. mapping_setup->angle=(uint_fast8_t *)av_mallocz(mapping_setup->coupling_steps * sizeof(uint_fast8_t));
  481. for(j=0;j<mapping_setup->coupling_steps;++j) {
  482. mapping_setup->magnitude[j]=get_bits(gb, ilog(vc->audio_channels-1));
  483. mapping_setup->angle[j]=get_bits(gb, ilog(vc->audio_channels-1));
  484. // FIXME: sanity checks
  485. }
  486. } else {
  487. mapping_setup->coupling_steps=0;
  488. }
  489. AV_DEBUG(" %d mapping coupling steps: %d \n", i, mapping_setup->coupling_steps);
  490. if(get_bits(gb, 2)) {
  491. av_log(vc->avccontext, AV_LOG_ERROR, "%d. mapping setup data invalid. \n", i);
  492. return 1; // following spec.
  493. }
  494. if (mapping_setup->submaps>1) {
  495. mapping_setup->mux=(uint_fast8_t *)av_mallocz(vc->audio_channels * sizeof(uint_fast8_t));
  496. for(j=0;j<vc->audio_channels;++j) {
  497. mapping_setup->mux[j]=get_bits(gb, 4);
  498. }
  499. }
  500. for(j=0;j<mapping_setup->submaps;++j) {
  501. get_bits(gb, 8); // FIXME check?
  502. mapping_setup->submap_floor[j]=get_bits(gb, 8);
  503. mapping_setup->submap_residue[j]=get_bits(gb, 8);
  504. AV_DEBUG(" %d mapping %d submap : floor %d, residue %d \n", i, j, mapping_setup->submap_floor[j], mapping_setup->submap_residue[j]);
  505. }
  506. }
  507. return 0;
  508. }
  509. // Process modes part
  510. static int vorbis_parse_setup_hdr_modes(vorbis_context *vc) {
  511. GetBitContext *gb=&vc->gb;
  512. uint_fast8_t i;
  513. vc->mode_count=get_bits(gb, 6)+1;
  514. vc->modes=(vorbis_mode *)av_mallocz(vc->mode_count * sizeof(vorbis_mode));
  515. AV_DEBUG(" There are %d modes.\n", vc->mode_count);
  516. for(i=0;i<vc->mode_count;++i) {
  517. vorbis_mode *mode_setup=&vc->modes[i];
  518. mode_setup->blockflag=get_bits(gb, 1);
  519. mode_setup->windowtype=get_bits(gb, 16); //FIXME check
  520. mode_setup->transformtype=get_bits(gb, 16); //FIXME check
  521. mode_setup->mapping=get_bits(gb, 8); //FIXME check
  522. 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);
  523. }
  524. return 0;
  525. }
  526. // Process the whole setup header using the functions above
  527. static int vorbis_parse_setup_hdr(vorbis_context *vc) {
  528. GetBitContext *gb=&vc->gb;
  529. if ((get_bits(gb, 8)!='v') || (get_bits(gb, 8)!='o') ||
  530. (get_bits(gb, 8)!='r') || (get_bits(gb, 8)!='b') ||
  531. (get_bits(gb, 8)!='i') || (get_bits(gb, 8)!='s')) {
  532. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (no vorbis signature). \n");
  533. return 1;
  534. }
  535. if (vorbis_parse_setup_hdr_codebooks(vc)) {
  536. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (codebooks). \n");
  537. return 2;
  538. }
  539. if (vorbis_parse_setup_hdr_tdtransforms(vc)) {
  540. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (time domain transforms). \n");
  541. return 3;
  542. }
  543. if (vorbis_parse_setup_hdr_floors(vc)) {
  544. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (floors). \n");
  545. return 4;
  546. }
  547. if (vorbis_parse_setup_hdr_residues(vc)) {
  548. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (residues). \n");
  549. return 5;
  550. }
  551. if (vorbis_parse_setup_hdr_mappings(vc)) {
  552. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (mappings). \n");
  553. return 6;
  554. }
  555. if (vorbis_parse_setup_hdr_modes(vc)) {
  556. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (modes). \n");
  557. return 7;
  558. }
  559. if (!get_bits1(gb)) {
  560. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (framing flag). \n");
  561. return 8; // framing flag bit unset error
  562. }
  563. return 0;
  564. }
  565. // Process the identification header
  566. static int vorbis_parse_id_hdr(vorbis_context *vc){
  567. GetBitContext *gb=&vc->gb;
  568. uint_fast8_t bl0, bl1;
  569. const float *vwin[8]={ vwin64, vwin128, vwin256, vwin512, vwin1024, vwin2048, vwin4096, vwin8192 };
  570. if ((get_bits(gb, 8)!='v') || (get_bits(gb, 8)!='o') ||
  571. (get_bits(gb, 8)!='r') || (get_bits(gb, 8)!='b') ||
  572. (get_bits(gb, 8)!='i') || (get_bits(gb, 8)!='s')) {
  573. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (no vorbis signature). \n");
  574. return 1;
  575. }
  576. vc->version=get_bits_long_le(gb, 32); //FIXME check 0
  577. vc->audio_channels=get_bits(gb, 8); //FIXME check >0
  578. vc->audio_samplerate=get_bits_long_le(gb, 32); //FIXME check >0
  579. vc->bitrate_maximum=get_bits_long_le(gb, 32);
  580. vc->bitrate_nominal=get_bits_long_le(gb, 32);
  581. vc->bitrate_minimum=get_bits_long_le(gb, 32);
  582. bl0=get_bits(gb, 4);
  583. bl1=get_bits(gb, 4);
  584. vc->blocksize_0=(1<<bl0);
  585. vc->blocksize_1=(1<<bl1);
  586. if (bl0>13 || bl0<6 || bl1>13 || bl1<6) {
  587. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (illegal blocksize). \n");
  588. return 3;
  589. }
  590. vc->swin=vwin[bl0-6];
  591. vc->lwin=vwin[bl1-6];
  592. if ((get_bits1(gb)) == 0) {
  593. av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (framing flag not set). \n");
  594. return 2;
  595. }
  596. vc->channel_residues=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));
  597. vc->channel_floors=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));
  598. vc->saved=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));
  599. vc->ret=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));
  600. vc->buf=(float *)av_malloc(vc->blocksize_1 * sizeof(float));
  601. vc->buf_tmp=(float *)av_malloc(vc->blocksize_1 * sizeof(float));
  602. vc->saved_start=0;
  603. ff_mdct_init(&vc->mdct0, bl0, 1);
  604. ff_mdct_init(&vc->mdct1, bl1, 1);
  605. 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 ",
  606. vc->version, vc->audio_channels, vc->audio_samplerate, vc->bitrate_maximum, vc->bitrate_nominal, vc->bitrate_minimum, vc->blocksize_0, vc->blocksize_1);
  607. /*
  608. BLK=vc->blocksize_0;
  609. for(i=0;i<BLK/2;++i) {
  610. vc->swin[i]=sin(0.5*3.14159265358*(sin(((float)i+0.5)/(float)BLK*3.14159265358))*(sin(((float)i+0.5)/(float)BLK*3.14159265358)));
  611. }
  612. */
  613. return 0;
  614. }
  615. // Process the extradata using the functions above (identification header, setup header)
  616. static int vorbis_decode_init(AVCodecContext *avccontext) {
  617. vorbis_context *vc = avccontext->priv_data ;
  618. uint8_t *headers = avccontext->extradata;
  619. int headers_len=avccontext->extradata_size;
  620. uint8_t *header_start[3];
  621. int header_len[3];
  622. GetBitContext *gb = &(vc->gb);
  623. int i, j, hdr_type;
  624. vc->avccontext = avccontext;
  625. if (!headers_len) {
  626. av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");
  627. return -1;
  628. }
  629. if(headers[0] == 0 && headers[1] == 30) {
  630. for(i = 0; i < 3; i++){
  631. header_len[i] = *headers++ << 8;
  632. header_len[i] += *headers++;
  633. header_start[i] = headers;
  634. headers += header_len[i];
  635. }
  636. } else if(headers[0] == 2) {
  637. for(j=1,i=0;i<2;++i, ++j) {
  638. header_len[i]=0;
  639. while(j<headers_len && headers[j]==0xff) {
  640. header_len[i]+=0xff;
  641. ++j;
  642. }
  643. if (j>=headers_len) {
  644. av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");
  645. return -1;
  646. }
  647. header_len[i]+=headers[j];
  648. }
  649. header_len[2]=headers_len-header_len[0]-header_len[1]-j;
  650. headers+=j;
  651. header_start[0] = headers;
  652. header_start[1] = header_start[0] + header_len[0];
  653. header_start[2] = header_start[1] + header_len[1];
  654. } else {
  655. av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");
  656. return -1;
  657. }
  658. init_get_bits(gb, header_start[0], header_len[0]*8);
  659. hdr_type=get_bits(gb, 8);
  660. if (hdr_type!=1) {
  661. av_log(avccontext, AV_LOG_ERROR, "First header is not the id header.\n");
  662. return -1;
  663. }
  664. if (vorbis_parse_id_hdr(vc)) {
  665. av_log(avccontext, AV_LOG_ERROR, "Id header corrupt.\n");
  666. vorbis_free(vc);
  667. return -1;
  668. }
  669. init_get_bits(gb, header_start[2], header_len[2]*8);
  670. hdr_type=get_bits(gb, 8);
  671. if (hdr_type!=5) {
  672. av_log(avccontext, AV_LOG_ERROR, "Third header is not the setup header.\n");
  673. return -1;
  674. }
  675. if (vorbis_parse_setup_hdr(vc)) {
  676. av_log(avccontext, AV_LOG_ERROR, "Setup header corrupt.\n");
  677. vorbis_free(vc);
  678. return -1;
  679. }
  680. avccontext->channels = vc->audio_channels;
  681. avccontext->sample_rate = vc->audio_samplerate;
  682. return 0 ;
  683. }
  684. // Decode audiopackets -------------------------------------------------
  685. // Read and decode floor (type 1 only)
  686. static uint_fast8_t vorbis_floor1_decode(vorbis_context *vc, vorbis_floor *vf, float *vec) {
  687. GetBitContext *gb=&vc->gb;
  688. uint_fast16_t range_v[4]={ 256, 128, 86, 64 };
  689. uint_fast16_t range=range_v[vf->multiplier-1];
  690. uint_fast16_t floor1_Y[vf->x_list_dim];
  691. uint_fast16_t floor1_Y_final[vf->x_list_dim];
  692. uint_fast8_t floor1_flag[vf->x_list_dim];
  693. uint_fast8_t class_;
  694. uint_fast8_t cdim;
  695. uint_fast8_t cbits;
  696. uint_fast8_t csub;
  697. uint_fast8_t cval;
  698. int_fast16_t book;
  699. uint_fast16_t offset;
  700. uint_fast16_t i,j;
  701. uint_fast16_t *floor_x_sort=vf->x_list_order;
  702. /*u*/int_fast16_t adx, ady, off, predicted; // WTF ? dy/adx= (unsigned)dy/adx ?
  703. int_fast16_t dy, err;
  704. uint_fast16_t lx,hx, ly, hy=0;
  705. if (!get_bits1(gb)) return 1; // silence
  706. // Read values (or differences) for the floor's points
  707. floor1_Y[0]=get_bits(gb, ilog(range-1));
  708. floor1_Y[1]=get_bits(gb, ilog(range-1));
  709. AV_DEBUG("floor 0 Y %d floor 1 Y %d \n", floor1_Y[0], floor1_Y[1]);
  710. offset=2;
  711. for(i=0;i<vf->partitions;++i) {
  712. class_=vf->partition_class[i];
  713. cdim=vf->class_dimensions[class_];
  714. cbits=vf->class_subclasses[class_];
  715. csub=(1<<cbits)-1;
  716. cval=0;
  717. AV_DEBUG("Cbits %d \n", cbits);
  718. if (cbits) { // this reads all subclasses for this partition's class
  719. cval=get_vlc2(gb, vc->codebooks[vf->class_masterbook[class_]].vlc.table,
  720. V_NB_BITS, vc->codebooks[vf->class_masterbook[class_]].maxdepth);
  721. }
  722. for(j=0;j<cdim;++j) {
  723. book=vf->subclass_books[class_][cval & csub];
  724. AV_DEBUG("book %d Cbits %d cval %d bits:%d \n", book, cbits, cval, get_bits_count(gb));
  725. cval=cval>>cbits;
  726. if (book>0) {
  727. floor1_Y[offset+j]=get_vlc2(gb, vc->codebooks[book].vlc.table,
  728. V_NB_BITS, vc->codebooks[book].maxdepth);
  729. } else {
  730. floor1_Y[offset+j]=0;
  731. }
  732. AV_DEBUG(" floor(%d) = %d \n", vf->x_list[offset+j], floor1_Y[offset+j]);
  733. }
  734. offset+=cdim;
  735. }
  736. // Amplitude calculation from the differences
  737. floor1_flag[0]=1;
  738. floor1_flag[1]=1;
  739. floor1_Y_final[0]=floor1_Y[0];
  740. floor1_Y_final[1]=floor1_Y[1];
  741. for(i=2;i<vf->x_list_dim;++i) {
  742. uint_fast16_t val, highroom, lowroom, room;
  743. uint_fast16_t high_neigh_offs;
  744. uint_fast16_t low_neigh_offs;
  745. low_neigh_offs=vf->low_neighbour[i];
  746. high_neigh_offs=vf->high_neighbour[i];
  747. dy=floor1_Y_final[high_neigh_offs]-floor1_Y_final[low_neigh_offs]; // render_point begin
  748. adx=vf->x_list[high_neigh_offs]-vf->x_list[low_neigh_offs];
  749. ady= ABS(dy);
  750. err=ady*(vf->x_list[i]-vf->x_list[low_neigh_offs]);
  751. off=err/adx;
  752. if (dy<0) {
  753. predicted=floor1_Y_final[low_neigh_offs]-off;
  754. } else {
  755. predicted=floor1_Y_final[low_neigh_offs]+off;
  756. } // render_point end
  757. val=floor1_Y[i];
  758. highroom=range-predicted;
  759. lowroom=predicted;
  760. if (highroom < lowroom) {
  761. room=highroom*2;
  762. } else {
  763. room=lowroom*2; // SPEC mispelling
  764. }
  765. if (val) {
  766. floor1_flag[low_neigh_offs]=1;
  767. floor1_flag[high_neigh_offs]=1;
  768. floor1_flag[i]=1;
  769. if (val>=room) {
  770. if (highroom > lowroom) {
  771. floor1_Y_final[i]=val-lowroom+predicted;
  772. } else {
  773. floor1_Y_final[i]=predicted-val+highroom-1;
  774. }
  775. } else {
  776. if (val & 1) {
  777. floor1_Y_final[i]=predicted-(val+1)/2;
  778. } else {
  779. floor1_Y_final[i]=predicted+val/2;
  780. }
  781. }
  782. } else {
  783. floor1_flag[i]=0;
  784. floor1_Y_final[i]=predicted;
  785. }
  786. AV_DEBUG(" Decoded floor(%d) = %d / val %d \n", vf->x_list[i], floor1_Y_final[i], val);
  787. }
  788. // Curve synth - connect the calculated dots and convert from dB scale FIXME optimize ?
  789. hx=0;
  790. lx=0;
  791. ly=floor1_Y_final[0]*vf->multiplier; // conforms to SPEC
  792. vec[0]=floor1_inverse_db_table[ly];
  793. for(i=1;i<vf->x_list_dim;++i) {
  794. AV_DEBUG(" Looking at post %d \n", i);
  795. if (floor1_flag[floor_x_sort[i]]) { // SPEC mispelled
  796. int_fast16_t x, y, dy, base, sy; // if uncommented: dy = -32 adx = 2 base = 2blablabla ?????
  797. hy=floor1_Y_final[floor_x_sort[i]]*vf->multiplier;
  798. hx=vf->x_list[floor_x_sort[i]];
  799. dy=hy-ly;
  800. adx=hx-lx;
  801. ady= (dy<0) ? -dy:dy;//ABS(dy);
  802. base=dy/adx;
  803. AV_DEBUG(" dy %d adx %d base %d = %d \n", dy, adx, base, dy/adx);
  804. x=lx;
  805. y=ly;
  806. err=0;
  807. if (dy<0) {
  808. sy=base-1;
  809. } else {
  810. sy=base+1;
  811. }
  812. ady=ady-(base<0 ? -base : base)*adx;
  813. vec[x]=floor1_inverse_db_table[y];
  814. AV_DEBUG(" vec[ %d ] = %d \n", x, y);
  815. for(x=lx+1;(x<hx) && (x<vf->x_list[1]);++x) {
  816. err+=ady;
  817. if (err>=adx) {
  818. err-=adx;
  819. y+=sy;
  820. } else {
  821. y+=base;
  822. }
  823. vec[x]=floor1_inverse_db_table[y];
  824. AV_DEBUG(" vec[ %d ] = %d \n", x, y);
  825. }
  826. /* for(j=1;j<hx-lx+1;++j) { // iterating render_point
  827. dy=hy-ly;
  828. adx=hx-lx;
  829. ady= dy<0 ? -dy : dy;
  830. err=ady*j;
  831. off=err/adx;
  832. if (dy<0) {
  833. predicted=ly-off;
  834. } else {
  835. predicted=ly+off;
  836. }
  837. if (lx+j < vf->x_list[1]) {
  838. vec[lx+j]=floor1_inverse_db_table[predicted];
  839. }
  840. }*/
  841. lx=hx;
  842. ly=hy;
  843. }
  844. }
  845. if (hx<vf->x_list[1]) {
  846. for(i=hx;i<vf->x_list[1];++i) {
  847. vec[i]=floor1_inverse_db_table[hy];
  848. }
  849. }
  850. AV_DEBUG(" Floor decoded\n");
  851. return 0;
  852. }
  853. // Read and decode residue
  854. 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) {
  855. GetBitContext *gb=&vc->gb;
  856. uint_fast8_t c_p_c=vc->codebooks[vr->classbook].dimensions;
  857. uint_fast16_t n_to_read=vr->end-vr->begin;
  858. uint_fast16_t ptns_to_read=n_to_read/vr->partition_size;
  859. uint_fast8_t classifs[ptns_to_read*vc->audio_channels];
  860. uint_fast8_t pass;
  861. uint_fast8_t ch_used;
  862. uint_fast8_t i,j,l;
  863. uint_fast16_t k;
  864. if (vr->type==2) {
  865. for(j=1;j<ch;++j) {
  866. do_not_decode[0]&=do_not_decode[j]; // FIXME - clobbering input
  867. }
  868. if (do_not_decode[0]) return 0;
  869. ch_used=1;
  870. } else {
  871. ch_used=ch;
  872. }
  873. AV_DEBUG(" residue type 0/1/2 decode begin, ch: %d cpc %d \n", ch, c_p_c);
  874. for(pass=0;pass<=vr->maxpass;++pass) { // FIXME OPTIMIZE?
  875. uint_fast16_t voffset;
  876. uint_fast16_t partition_count;
  877. uint_fast16_t j_times_ptns_to_read;
  878. voffset=vr->begin;
  879. for(partition_count=0;partition_count<ptns_to_read;) { // SPEC error
  880. if (!pass) {
  881. for(j_times_ptns_to_read=0, j=0;j<ch_used;++j) {
  882. if (!do_not_decode[j]) {
  883. uint_fast32_t temp=get_vlc2(gb, vc->codebooks[vr->classbook].vlc.table,
  884. V_NB_BITS, vc->codebooks[vr->classbook].maxdepth);
  885. AV_DEBUG("Classword: %d \n", temp);
  886. for(i=0;i<c_p_c;++i) {
  887. uint_fast32_t temp2;
  888. temp2=temp/vr->classifications;
  889. classifs[j_times_ptns_to_read+partition_count+c_p_c-1-i]=temp-temp2*vr->classifications;
  890. temp=temp2;
  891. }
  892. }
  893. j_times_ptns_to_read+=ptns_to_read;
  894. }
  895. }
  896. for(i=0;(i<c_p_c) && (partition_count<ptns_to_read);++i) {
  897. for(j_times_ptns_to_read=0, j=0;j<ch_used;++j) {
  898. uint_fast16_t voffs;
  899. if (!do_not_decode[j]) {
  900. uint_fast8_t vqclass=classifs[j_times_ptns_to_read+partition_count];
  901. int_fast16_t vqbook=vr->books[vqclass][pass];
  902. if (vqbook>=0) {
  903. uint_fast16_t coffs;
  904. if (vr->type==0) {
  905. uint_fast16_t step=vr->partition_size/vc->codebooks[vqbook].dimensions;
  906. voffs=voffset+j*vlen;
  907. for(k=0;k<step;++k) {
  908. coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,
  909. V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;
  910. for(l=0;l<vc->codebooks[vqbook].dimensions;++l) {
  911. vec[voffs+k+l*step]+=vc->codebooks[vqbook].codevectors[coffs+l]; // FPMATH
  912. }
  913. }
  914. }
  915. else if (vr->type==1) {
  916. voffs=voffset+j*vlen;
  917. for(k=0;k<vr->partition_size/vc->codebooks[vqbook].dimensions;++k) {
  918. coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,
  919. V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;
  920. for(l=0;l<vc->codebooks[vqbook].dimensions;++l, ++voffs) {
  921. vec[voffs]+=vc->codebooks[vqbook].codevectors[coffs+l]; // FPMATH
  922. AV_DEBUG(" pass %d offs: %d curr: %f change: %f cv offs.: %d \n", pass, voffs, vec[voffs], vc->codebooks[vqbook].codevectors[coffs+l], coffs);
  923. }
  924. }
  925. }
  926. else if (vr->type==2 && ch==2) { // most frequent case optimized
  927. voffs=voffset;
  928. for(k=0;k<vr->partition_size/vc->codebooks[vqbook].dimensions;++k) {
  929. coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,
  930. V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;
  931. for(l=0;l<vc->codebooks[vqbook].dimensions;++l, ++voffs) {
  932. vec[(voffs>>1)+((voffs&1) ? vlen : 0)]+=vc->codebooks[vqbook].codevectors[coffs+l]; // FPMATH
  933. 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], vc->codebooks[vqbook].codevectors[coffs+l], coffs, l);
  934. }
  935. }
  936. }
  937. else if (vr->type==2) {
  938. voffs=voffset;
  939. for(k=0;k<vr->partition_size/vc->codebooks[vqbook].dimensions;++k) {
  940. coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,
  941. V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;
  942. for(l=0;l<vc->codebooks[vqbook].dimensions;++l, ++voffs) {
  943. vec[voffs/ch+(voffs%ch)*vlen]+=vc->codebooks[vqbook].codevectors[coffs+l]; // FPMATH FIXME use if and counter instead of / and %
  944. 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], vc->codebooks[vqbook].codevectors[coffs+l], coffs, l);
  945. }
  946. }
  947. } else {
  948. av_log(vc->avccontext, AV_LOG_ERROR, " Invalid residue type while residue decode?! \n");
  949. return 1;
  950. }
  951. }
  952. }
  953. j_times_ptns_to_read+=ptns_to_read;
  954. }
  955. ++partition_count;
  956. voffset+=vr->partition_size;
  957. }
  958. }
  959. }
  960. return 0;
  961. }
  962. // Decode the audio packet using the functions above
  963. static int vorbis_parse_audio_packet(vorbis_context *vc) {
  964. GetBitContext *gb=&vc->gb;
  965. uint_fast8_t previous_window=0,next_window=0;
  966. uint_fast8_t mode_number;
  967. uint_fast16_t blocksize;
  968. int_fast32_t i,j;
  969. uint_fast8_t no_residue[vc->audio_channels];
  970. uint_fast8_t do_not_decode[vc->audio_channels];
  971. vorbis_mapping *mapping;
  972. float *ch_res_ptr=vc->channel_residues;
  973. float *ch_floor_ptr=vc->channel_floors;
  974. uint_fast8_t res_chan[vc->audio_channels];
  975. uint_fast8_t res_num=0;
  976. int_fast16_t retlen=0;
  977. uint_fast16_t saved_start=0;
  978. if (get_bits1(gb)) {
  979. av_log(vc->avccontext, AV_LOG_ERROR, "Not a Vorbis I audio packet.\n");
  980. return -1; // packet type not audio
  981. }
  982. if (vc->mode_count==1) {
  983. mode_number=0;
  984. } else {
  985. mode_number=get_bits(gb, ilog(vc->mode_count-1));
  986. }
  987. mapping=&vc->mappings[vc->modes[mode_number].mapping];
  988. AV_DEBUG(" Mode number: %d , mapping: %d , blocktype %d \n", mode_number, vc->modes[mode_number].mapping, vc->modes[mode_number].blockflag);
  989. if (vc->modes[mode_number].blockflag) {
  990. previous_window=get_bits1(gb);
  991. next_window=get_bits1(gb);
  992. }
  993. blocksize=vc->modes[mode_number].blockflag ? vc->blocksize_1 : vc->blocksize_0;
  994. memset(ch_res_ptr, 0, sizeof(float)*vc->audio_channels*blocksize/2); //FIXME can this be removed ?
  995. memset(ch_floor_ptr, 0, sizeof(float)*vc->audio_channels*blocksize/2); //FIXME can this be removed ?
  996. // Decode floor(1)
  997. for(i=0;i<vc->audio_channels;++i) {
  998. vorbis_floor *floor;
  999. if (mapping->submaps>1) {
  1000. floor=&vc->floors[mapping->submap_floor[mapping->mux[i]]];
  1001. } else {
  1002. floor=&vc->floors[mapping->submap_floor[0]];
  1003. }
  1004. no_residue[i]=vorbis_floor1_decode(vc, floor, ch_floor_ptr);
  1005. ch_floor_ptr+=blocksize/2;
  1006. }
  1007. // Nonzero vector propagate
  1008. for(i=mapping->coupling_steps-1;i>=0;--i) {
  1009. if (!(no_residue[mapping->magnitude[i]] & no_residue[mapping->angle[i]])) {
  1010. no_residue[mapping->magnitude[i]]=0;
  1011. no_residue[mapping->angle[i]]=0;
  1012. }
  1013. }
  1014. // Decode residue
  1015. for(i=0;i<mapping->submaps;++i) {
  1016. vorbis_residue *residue;
  1017. uint_fast8_t ch=0;
  1018. for(j=0;j<vc->audio_channels;++j) {
  1019. if ((mapping->submaps==1) || (i=mapping->mux[j])) {
  1020. res_chan[j]=res_num;
  1021. if (no_residue[j]) {
  1022. do_not_decode[ch]=1;
  1023. } else {
  1024. do_not_decode[ch]=0;
  1025. }
  1026. ++ch;
  1027. ++res_num;
  1028. }
  1029. }
  1030. residue=&vc->residues[mapping->submap_residue[i]];
  1031. vorbis_residue_decode(vc, residue, ch, do_not_decode, ch_res_ptr, blocksize/2);
  1032. ch_res_ptr+=ch*blocksize/2;
  1033. }
  1034. // Inverse coupling
  1035. for(i=mapping->coupling_steps-1;i>=0;--i) { //warning: i has to be signed
  1036. float *mag, *ang;
  1037. mag=vc->channel_residues+res_chan[mapping->magnitude[i]]*blocksize/2;
  1038. ang=vc->channel_residues+res_chan[mapping->angle[i]]*blocksize/2;
  1039. for(j=0;j<blocksize/2;++j) {
  1040. float temp;
  1041. if (mag[j]>0.0) {
  1042. if (ang[j]>0.0) {
  1043. ang[j]=mag[j]-ang[j];
  1044. } else {
  1045. temp=ang[j];
  1046. ang[j]=mag[j];
  1047. mag[j]+=temp;
  1048. }
  1049. } else {
  1050. if (ang[j]>0.0) {
  1051. ang[j]+=mag[j];
  1052. } else {
  1053. temp=ang[j];
  1054. ang[j]=mag[j];
  1055. mag[j]-=temp;
  1056. }
  1057. }
  1058. }
  1059. }
  1060. // Dotproduct
  1061. for(j=0, ch_floor_ptr=vc->channel_floors;j<vc->audio_channels;++j,ch_floor_ptr+=blocksize/2) {
  1062. ch_res_ptr=vc->channel_residues+res_chan[j]*blocksize/2;
  1063. for(i=0;i<blocksize/2;++i) {
  1064. ch_floor_ptr[i]*=ch_res_ptr[i]; //FPMATH
  1065. }
  1066. }
  1067. // MDCT, overlap/add, save data for next overlapping FPMATH
  1068. for(j=0;j<vc->audio_channels;++j) {
  1069. uint_fast8_t step=vc->audio_channels;
  1070. uint_fast16_t k;
  1071. float *saved=vc->saved+j*vc->blocksize_1/2;
  1072. float *ret=vc->ret;
  1073. const float *lwin=vc->lwin;
  1074. const float *swin=vc->swin;
  1075. float *buf=vc->buf;
  1076. float *buf_tmp=vc->buf_tmp;
  1077. ch_floor_ptr=vc->channel_floors+j*blocksize/2;
  1078. saved_start=vc->saved_start;
  1079. ff_imdct_calc(vc->modes[mode_number].blockflag ? &vc->mdct1 : &vc->mdct0, buf, ch_floor_ptr, buf_tmp);
  1080. if (vc->modes[mode_number].blockflag) {
  1081. // -- overlap/add
  1082. if (previous_window) {
  1083. for(k=j, i=0;i<vc->blocksize_1/2;++i, k+=step) {
  1084. ret[k]=saved[i]+buf[i]*lwin[i];
  1085. }
  1086. retlen=vc->blocksize_1/2;
  1087. } else {
  1088. for(k=j, i=0;i<vc->blocksize_0/2;++i, k+=step) {
  1089. ret[k]=saved[i]+buf[(vc->blocksize_1-vc->blocksize_0)/4+i]*swin[i];
  1090. }
  1091. for(i=0;i<(vc->blocksize_1-vc->blocksize_0)/4;++i, k+=step) {
  1092. ret[k]=buf[vc->blocksize_0/2+(vc->blocksize_1-vc->blocksize_0)/4+i];
  1093. }
  1094. retlen=vc->blocksize_0/2+(vc->blocksize_1-vc->blocksize_0)/4;
  1095. }
  1096. // -- save
  1097. if (next_window) {
  1098. for(i=0;i<vc->blocksize_1/2;++i) {
  1099. saved[i]=buf[vc->blocksize_1/2+i]*lwin[vc->blocksize_1/2-1-i];
  1100. }
  1101. saved_start=0;
  1102. } else {
  1103. saved_start=(vc->blocksize_1-vc->blocksize_0)/4;
  1104. for(i=0;i<saved_start;++i) {
  1105. saved[i]=buf[vc->blocksize_1/2+i];
  1106. }
  1107. for(i=0;i<vc->blocksize_0/2;++i) {
  1108. saved[saved_start+i]=buf[vc->blocksize_1/2+saved_start+i]*swin[vc->blocksize_0/2-1-i];
  1109. }
  1110. }
  1111. } else {
  1112. // --overlap/add
  1113. for(k=j, i=0;i<saved_start;++i, k+=step) {
  1114. ret[k]=saved[i];
  1115. }
  1116. for(i=0;i<vc->blocksize_0/2;++i, k+=step) {
  1117. ret[k]=saved[saved_start+i]+buf[i]*swin[i];
  1118. }
  1119. retlen=saved_start+vc->blocksize_0/2;
  1120. // -- save
  1121. for(i=0;i<vc->blocksize_0/2;++i) {
  1122. saved[i]=buf[vc->blocksize_0/2+i]*swin[vc->blocksize_0/2-1-i];
  1123. }
  1124. saved_start=0;
  1125. }
  1126. }
  1127. vc->saved_start=saved_start;
  1128. return retlen*vc->audio_channels;
  1129. }
  1130. // Return the decoded audio packet through the standard api
  1131. static int vorbis_decode_frame(AVCodecContext *avccontext,
  1132. void *data, int *data_size,
  1133. uint8_t *buf, int buf_size)
  1134. {
  1135. vorbis_context *vc = avccontext->priv_data ;
  1136. GetBitContext *gb = &(vc->gb);
  1137. int_fast16_t i, len;
  1138. if(!buf_size){
  1139. return 0;
  1140. }
  1141. AV_DEBUG("packet length %d \n", buf_size);
  1142. init_get_bits(gb, buf, buf_size*8);
  1143. len=vorbis_parse_audio_packet(vc);
  1144. if (len<=0) {
  1145. *data_size=0;
  1146. return buf_size;
  1147. }
  1148. if (!vc->first_frame) {
  1149. vc->first_frame=1;
  1150. *data_size=0;
  1151. return buf_size ;
  1152. }
  1153. AV_DEBUG("parsed %d bytes %d bits, returned %d samples (*ch*bits) \n", get_bits_count(gb)/8, get_bits_count(gb)%8, len);
  1154. for(i=0;i<len;++i) {
  1155. int_fast32_t tmp;
  1156. tmp=vc->ret[i]*32768;
  1157. if (tmp>32767) tmp=32767;
  1158. if (tmp<-32768) tmp=-32768;
  1159. ((int16_t*)data)[i]=tmp;
  1160. }
  1161. *data_size=len*2;
  1162. return buf_size ;
  1163. }
  1164. // Close decoder
  1165. static int vorbis_decode_close(AVCodecContext *avccontext) {
  1166. vorbis_context *vc = avccontext->priv_data;
  1167. vorbis_free(vc);
  1168. return 0 ;
  1169. }
  1170. AVCodec vorbis_decoder = {
  1171. "vorbis",
  1172. CODEC_TYPE_AUDIO,
  1173. CODEC_ID_VORBIS,
  1174. sizeof(vorbis_context),
  1175. vorbis_decode_init,
  1176. NULL,
  1177. vorbis_decode_close,
  1178. vorbis_decode_frame,
  1179. };