falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
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.
4597 Commits
32 Branches
283MB
Tree: 6ad1fa5a49
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '6ad1fa5a49'
${ noResults }
FFmpeg/libavcodec/vorbis.c

1459 lines
50KB
Raw Blame History 

  1. /**

  2.  * @file vorbis.c

  3.  * Vorbis I decoder

  4.  * @author Denes Balatoni  ( dbalatoni programozo hu )

  5. 

  6.  * This library is free software; you can redistribute it and/or

  7.  * modify it under the terms of the GNU Lesser General Public

  8.  * License as published by the Free Software Foundation; either

  9.  * version 2 of the License, or (at your option) any later version.

  10.  *

  11.  * This library is distributed in the hope that it will be useful,

  12.  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  13.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

  14.  * Lesser General Public License for more details.

  15.  *

  16.  * You should have received a copy of the GNU Lesser General Public

  17.  * License along with this library; if not, write to the Free Software

  18.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

  19.  *

  20.  */

  21. 

  22. #undef V_DEBUG

  23. 

  24. #include <math.h>

  25. 

  26. #define ALT_BITSTREAM_READER_LE

  27. #include "avcodec.h"

  28. #include "bitstream.h"

  29. #include "dsputil.h"

  30. 

  31. #include "vorbis.h"

  32. 

  33. #define V_NB_BITS 11

  34. #define V_MAX_VLCS (1<<16)

  35. 

  36. #ifndef V_DEBUG

  37. #define AV_DEBUG(...)

  38. #endif

  39. 

  40. 

  41. 

  42. /* Helper functions */

  43. 

  44. /**

  45.  *  reads 0-32 bits when using the ALT_BITSTREAM_READER_LE bitstream reader

  46.  */

  47. unsigned int get_bits_long_le(GetBitContext *s, int n){

  48.     if(n<=17) return get_bits(s, n);

  49.     else{

  50.         int ret= get_bits(s, 16);

  51.         return ret | (get_bits(s, n-16) << 16);

  52.     }

  53. }

  54. 

  55. static unsigned int ilog(unsigned int i) { // unfortunatelly av_log2 uses different rounding

  56.     unsigned int ret=0;

  57.     while (i!=0) {

  58.         ++ret;

  59.         i>>=1;

  60.     }

  61.     return ret;

  62. }

  63. 

  64. static unsigned int nth_root(unsigned int x, unsigned int n) {   // x^(1/n)

  65.     unsigned int ret=0, i, j;

  66. 

  67.     do {

  68.         ++ret;

  69.         for(i=0,j=ret;i<n-1;i++) j*=ret;

  70.     } while (j<=x);

  71. 

  72.     return (ret-1);

  73. }

  74. 

  75. static float vorbisfloat2float(uint_fast32_t val) {

  76.     double mant=val&0x1fffff;

  77.     long exp=(val&0x7fe00000L)>>21;

  78.     if (val&0x80000000) mant=-mant;

  79.     return(ldexp(mant, exp-20-768));

  80. }

  81. 

  82. 

  83. // Generate vlc codes from vorbis huffman code lengths

  84. 

  85. static int vorbis_len2vlc(vorbis_context *vc, uint_fast8_t *bits, uint_fast32_t *codes, uint_fast32_t num) {

  86.     uint_fast32_t exit_at_level[33]={404,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,

  87.         0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};

  88. 

  89.     uint_fast8_t i,j;

  90.     uint_fast32_t code,p;

  91. 

  92. #ifdef V_DEBUG

  93.     GetBitContext gb;

  94. #endif

  95. 

  96.     for(p=0;(bits[p]==0) && (p<num);++p);

  97.     if (p==num) {

  98. //        av_log(vc->avccontext, AV_LOG_INFO, "An empty codebook. Heh?! \n");

  99.         return 0;

  100.     }

  101. 

  102.     codes[p]=0;

  103.     for(i=0;i<bits[p];++i) {

  104.         exit_at_level[i+1]=1<<i;

  105.     }

  106. 

  107. #ifdef V_DEBUG

  108.     av_log(vc->avccontext, AV_LOG_INFO, " %d. of %d code len %d code %d - ", p, num, bits[p], codes[p]);

  109.     init_get_bits(&gb, (uint_fast8_t *)&codes[p], bits[p]);

  110.     for(i=0;i<bits[p];++i) {

  111.         av_log(vc->avccontext, AV_LOG_INFO, "%s", get_bits1(&gb) ? "1" : "0");

  112.     }

  113.     av_log(vc->avccontext, AV_LOG_INFO, "\n");

  114. #endif

  115. 

  116.     ++p;

  117. 

  118.     for(;p<num;++p) {

  119.         if (bits[p]==0) continue;

  120.         // find corresponding exit(node which the tree can grow further from)

  121.         for(i=bits[p];i>0;--i) {

  122.             if (exit_at_level[i]) break;

  123.         }

  124.         if (!i) return 1; // overspecified tree

  125.         code=exit_at_level[i];

  126.         exit_at_level[i]=0;

  127.         // construct code (append 0s to end) and introduce new exits

  128.         for(j=i+1;j<=bits[p];++j) {

  129.             exit_at_level[j]=code+(1<<(j-1));

  130.         }

  131.         codes[p]=code;

  132. 

  133. #ifdef V_DEBUG

  134.         av_log(vc->avccontext, AV_LOG_INFO, " %d. code len %d code %d - ", p, bits[p], codes[p]);

  135.         init_get_bits(&gb, (uint_fast8_t *)&codes[p], bits[p]);

  136.         for(i=0;i<bits[p];++i) {

  137.             av_log(vc->avccontext, AV_LOG_INFO, "%s", get_bits1(&gb) ? "1" : "0");

  138.         }

  139.         av_log(vc->avccontext, AV_LOG_INFO, "\n");

  140. #endif

  141. 

  142.     }

  143. 

  144.     //FIXME no exits should be left (underspecified tree - ie. unused valid vlcs - not allowed by SPEC)

  145. 

  146.     return 0;

  147. }

  148. 

  149. // Free all allocated memory -----------------------------------------

  150. 

  151. static void vorbis_free(vorbis_context *vc) {

  152.     int_fast16_t i;

  153. 

  154.     av_freep(&vc->channel_residues);

  155.     av_freep(&vc->channel_floors);

  156.     av_freep(&vc->saved);

  157.     av_freep(&vc->ret);

  158. 

  159.     av_freep(&vc->residues);

  160.     av_freep(&vc->modes);

  161. 

  162.     ff_mdct_end(&vc->mdct0);

  163.     ff_mdct_end(&vc->mdct1);

  164. 

  165.     for(i=0;i<vc->codebook_count;++i) {

  166.         av_free(vc->codebooks[i].codevectors);

  167.         free_vlc(&vc->codebooks[i].vlc);

  168.     }

  169.     av_freep(&vc->codebooks);

  170. 

  171.     for(i=0;i<vc->floor_count;++i) {

  172.         av_free(vc->floors[i].x_list);

  173.         av_free(vc->floors[i].x_list_order);

  174.         av_free(vc->floors[i].low_neighbour);

  175.         av_free(vc->floors[i].high_neighbour);

  176.     }

  177.     av_freep(&vc->floors);

  178. 

  179.     for(i=0;i<vc->mapping_count;++i) {

  180.         av_free(vc->mappings[i].magnitude);

  181.         av_free(vc->mappings[i].angle);

  182.         av_free(vc->mappings[i].mux);

  183.     }

  184.     av_freep(&vc->mappings);

  185. }

  186. 

  187. // Parse setup header -------------------------------------------------

  188. 

  189. // Process codebooks part

  190. 

  191. static int vorbis_parse_setup_hdr_codebooks(vorbis_context *vc) {

  192.     uint_fast16_t cb;

  193.     uint_fast8_t *tmp_vlc_bits;

  194.     uint_fast32_t *tmp_vlc_codes;

  195.     GetBitContext *gb=&vc->gb;

  196. 

  197.     vc->codebook_count=get_bits(gb,8)+1;

  198. 

  199.     AV_DEBUG(" Codebooks: %d \n", vc->codebook_count);

  200. 

  201.     vc->codebooks=(vorbis_codebook *)av_mallocz(vc->codebook_count * sizeof(vorbis_codebook));

  202.     tmp_vlc_bits=(uint_fast8_t *)av_mallocz(V_MAX_VLCS * sizeof(uint_fast8_t));

  203.     tmp_vlc_codes=(uint_fast32_t *)av_mallocz(V_MAX_VLCS * sizeof(uint_fast32_t));

  204. 

  205.     for(cb=0;cb<vc->codebook_count;++cb) {

  206.         vorbis_codebook *codebook_setup=&vc->codebooks[cb];

  207.         uint_fast8_t ordered;

  208.         uint_fast32_t t, used_entries=0;

  209.         uint_fast32_t entries;

  210. 

  211.         AV_DEBUG(" %d. Codebook \n", cb);

  212. 

  213.         if (get_bits(gb, 24)!=0x564342) {

  214.             av_log(vc->avccontext, AV_LOG_ERROR, " %d. Codebook setup data corrupt. \n", cb);

  215.             goto error;

  216.         }

  217. 

  218.         codebook_setup->dimensions=get_bits(gb, 16);

  219.         if (codebook_setup->dimensions>16) {

  220.             av_log(vc->avccontext, AV_LOG_ERROR, " %d. Codebook's dimension is too large (%d). \n", cb, codebook_setup->dimensions);

  221.             goto error;

  222.         }

  223.         entries=get_bits(gb, 24);

  224.         if (entries>V_MAX_VLCS) {

  225.             av_log(vc->avccontext, AV_LOG_ERROR, " %d. Codebook has too many entries (%d). \n", cb, entries);

  226.             goto error;

  227.         }

  228. 

  229.         ordered=get_bits1(gb);

  230. 

  231.         AV_DEBUG(" codebook_dimensions %d, codebook_entries %d \n", codebook_setup->dimensions, entries);

  232. 

  233.         if (!ordered) {

  234.             uint_fast16_t ce;

  235.             uint_fast8_t flag;

  236.             uint_fast8_t sparse=get_bits1(gb);

  237. 

  238.             AV_DEBUG(" not ordered \n");

  239. 

  240.             if (sparse) {

  241.                 AV_DEBUG(" sparse \n");

  242. 

  243.                 used_entries=0;

  244.                 for(ce=0;ce<entries;++ce) {

  245.                     flag=get_bits1(gb);

  246.                     if (flag) {

  247.                         tmp_vlc_bits[ce]=get_bits(gb, 5)+1;

  248.                         ++used_entries;

  249.                     }

  250.                     else tmp_vlc_bits[ce]=0;

  251.                 }

  252.             } else {

  253.                 AV_DEBUG(" not sparse \n");

  254. 

  255.                 used_entries=entries;

  256.                 for(ce=0;ce<entries;++ce) {

  257.                     tmp_vlc_bits[ce]=get_bits(gb, 5)+1;

  258.                 }

  259.             }

  260.         } else {

  261.             uint_fast16_t current_entry=0;

  262.             uint_fast8_t current_length=get_bits(gb, 5)+1;

  263. 

  264.             AV_DEBUG(" ordered, current length: %d \n", current_length);  //FIXME

  265. 

  266.             used_entries=entries;

  267.             for(;current_entry<used_entries;++current_length) {

  268.                 uint_fast16_t i, number;

  269. 

  270.                 AV_DEBUG(" number bits: %d ", ilog(entries - current_entry));

  271. 

  272.                 number=get_bits(gb, ilog(entries - current_entry));

  273. 

  274.                 AV_DEBUG(" number: %d \n", number);

  275. 

  276.                 for(i=current_entry;i<number+current_entry;++i) {

  277.                     if (i<used_entries) tmp_vlc_bits[i]=current_length;

  278.                 }

  279. 

  280.                 current_entry+=number;

  281.             }

  282.             if (current_entry>used_entries) {

  283.                 av_log(vc->avccontext, AV_LOG_ERROR, " More codelengths than codes in codebook. \n");

  284.                 goto error;

  285.             }

  286.         }

  287. 

  288.         codebook_setup->lookup_type=get_bits(gb, 4);

  289. 

  290.         AV_DEBUG(" lookup type: %d : %s \n", codebook_setup->lookup_type, codebook_setup->lookup_type ? "vq" : "no lookup" );

  291. 

  292. // If the codebook is used for (inverse) VQ, calculate codevectors.

  293. 

  294.         if (codebook_setup->lookup_type==1) {

  295.             uint_fast16_t i, j, k;

  296.             uint_fast16_t codebook_lookup_values=nth_root(entries, codebook_setup->dimensions);

  297.             uint_fast16_t codebook_multiplicands[codebook_lookup_values];

  298. 

  299.             float codebook_minimum_value=vorbisfloat2float(get_bits_long_le(gb, 32));

  300.             float codebook_delta_value=vorbisfloat2float(get_bits_long_le(gb, 32));

  301.             uint_fast8_t codebook_value_bits=get_bits(gb, 4)+1;

  302.             uint_fast8_t codebook_sequence_p=get_bits1(gb);

  303. 

  304.             AV_DEBUG(" We expect %d numbers for building the codevectors. \n", codebook_lookup_values);

  305.             AV_DEBUG("  delta %f minmum %f \n", codebook_delta_value, codebook_minimum_value);

  306. 

  307.             for(i=0;i<codebook_lookup_values;++i) {

  308.                 codebook_multiplicands[i]=get_bits(gb, codebook_value_bits);

  309. 

  310.                 AV_DEBUG(" multiplicands*delta+minmum : %e \n", (float)codebook_multiplicands[i]*codebook_delta_value+codebook_minimum_value);

  311.                 AV_DEBUG(" multiplicand %d \n", codebook_multiplicands[i]);

  312.             }

  313. 

  314. // Weed out unused vlcs and build codevector vector

  315.             codebook_setup->codevectors=(float *)av_mallocz(used_entries*codebook_setup->dimensions * sizeof(float));

  316.             for(j=0, i=0;i<entries;++i) {

  317.                 uint_fast8_t dim=codebook_setup->dimensions;

  318. 

  319.                 if (tmp_vlc_bits[i]) {

  320.                     float last=0.0;

  321.                     uint_fast32_t lookup_offset=i;

  322. 

  323. #ifdef V_DEBUG

  324.                     av_log(vc->avccontext, AV_LOG_INFO, "Lookup offset %d ,", i);

  325. #endif

  326. 

  327.                     for(k=0;k<dim;++k) {

  328.                         uint_fast32_t multiplicand_offset = lookup_offset % codebook_lookup_values;

  329.                         codebook_setup->codevectors[j*dim+k]=codebook_multiplicands[multiplicand_offset]*codebook_delta_value+codebook_minimum_value+last;

  330.                         if (codebook_sequence_p) {

  331.                             last=codebook_setup->codevectors[j*dim+k];

  332.                         }

  333.                         lookup_offset/=codebook_lookup_values;

  334.                     }

  335.                     tmp_vlc_bits[j]=tmp_vlc_bits[i];

  336. 

  337. #ifdef V_DEBUG

  338.                     av_log(vc->avccontext, AV_LOG_INFO, "real lookup offset %d, vector: ", j);

  339.                     for(k=0;k<dim;++k) {

  340.                         av_log(vc->avccontext, AV_LOG_INFO, " %f ", codebook_setup->codevectors[j*dim+k]);

  341.                     }

  342.                     av_log(vc->avccontext, AV_LOG_INFO, "\n");

  343. #endif

  344. 

  345.                     ++j;

  346.                 }

  347.             }

  348.             if (j!=used_entries) {

  349.                 av_log(vc->avccontext, AV_LOG_ERROR, "Bug in codevector vector building code. \n");

  350.                 goto error;

  351.             }

  352.             entries=used_entries;

  353.         }

  354.         else if (codebook_setup->lookup_type>=2) {

  355.             av_log(vc->avccontext, AV_LOG_ERROR, "Codebook lookup type not supported. \n");

  356.             goto error;

  357.         }

  358. 

  359. // Initialize VLC table

  360.         if (vorbis_len2vlc(vc, tmp_vlc_bits, tmp_vlc_codes, entries)) {

  361.             av_log(vc->avccontext, AV_LOG_ERROR, " Invalid code lengths while generating vlcs. \n");

  362.             goto error;

  363.         }

  364.         codebook_setup->maxdepth=0;

  365.         for(t=0;t<entries;++t)

  366.             if (tmp_vlc_bits[t]>=codebook_setup->maxdepth) codebook_setup->maxdepth=tmp_vlc_bits[t];

  367. 

  368.         codebook_setup->maxdepth=(codebook_setup->maxdepth+V_NB_BITS-1)/V_NB_BITS;

  369. 

  370.         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)) {

  371.             av_log(vc->avccontext, AV_LOG_ERROR, " Error generating vlc tables. \n");

  372.             goto error;

  373.         }

  374.     }

  375. 

  376.     av_free(tmp_vlc_bits);

  377.     av_free(tmp_vlc_codes);

  378.     return 0;

  379. 

  380. // Error:

  381. error:

  382.     av_free(tmp_vlc_bits);

  383.     av_free(tmp_vlc_codes);

  384.     return 1;

  385. }

  386. 

  387. // Process time domain transforms part (unused in Vorbis I)

  388. 

  389. static int vorbis_parse_setup_hdr_tdtransforms(vorbis_context *vc) {

  390.     GetBitContext *gb=&vc->gb;

  391.     uint_fast8_t i;

  392.     uint_fast8_t vorbis_time_count=get_bits(gb, 6)+1;

  393. 

  394.     for(i=0;i<vorbis_time_count;++i) {

  395.         uint_fast16_t vorbis_tdtransform=get_bits(gb, 16);

  396. 

  397.         AV_DEBUG(" Vorbis time domain transform %d: %d \n", vorbis_time_count, vorbis_tdtransform);

  398. 

  399.         if (vorbis_tdtransform) {

  400.             av_log(vc->avccontext, AV_LOG_ERROR, "Vorbis time domain transform data nonzero. \n");

  401.             return 1;

  402.         }

  403.     }

  404.     return 0;

  405. }

  406. 

  407. // Process floors part - only floor type 1 is supported

  408. 

  409. static int vorbis_parse_setup_hdr_floors(vorbis_context *vc) {

  410.     GetBitContext *gb=&vc->gb;

  411.     uint_fast16_t i,j,k;

  412. 

  413.     vc->floor_count=get_bits(gb, 6)+1;

  414. 

  415.     vc->floors=(vorbis_floor *)av_mallocz(vc->floor_count * sizeof(vorbis_floor));

  416. 

  417.     for (i=0;i<vc->floor_count;++i) {

  418.         vorbis_floor *floor_setup=&vc->floors[i];

  419. 

  420.         floor_setup->floor_type=get_bits(gb, 16);

  421. 

  422.         AV_DEBUG(" %d. floor type %d \n", i, floor_setup->floor_type);

  423. 

  424.         if (floor_setup->floor_type==1) {

  425.             uint_fast8_t maximum_class=0;

  426.             uint_fast8_t rangebits;

  427.             uint_fast16_t floor1_values=2;

  428. 

  429.             floor_setup->partitions=get_bits(gb, 5);

  430. 

  431.             AV_DEBUG(" %d.floor: %d partitions \n", i, floor_setup->partitions);

  432. 

  433.             for(j=0;j<floor_setup->partitions;++j) {

  434.                 floor_setup->partition_class[j]=get_bits(gb, 4);

  435.                 if (floor_setup->partition_class[j]>maximum_class) maximum_class=floor_setup->partition_class[j];

  436. 

  437.                 AV_DEBUG(" %d. floor %d partition class %d \n", i, j, floor_setup->partition_class[j]);

  438. 

  439.             }

  440. 

  441.             AV_DEBUG(" maximum class %d \n", maximum_class);

  442. 

  443.             floor_setup->maximum_class=maximum_class;

  444. 

  445.             for(j=0;j<=maximum_class;++j) {

  446.                 floor_setup->class_dimensions[j]=get_bits(gb, 3)+1;

  447.                 floor_setup->class_subclasses[j]=get_bits(gb, 2);

  448. 

  449.                 AV_DEBUG(" %d floor %d class dim: %d subclasses %d \n", i, j, floor_setup->class_dimensions[j], floor_setup->class_subclasses[j]);

  450. 

  451.                 if (floor_setup->class_subclasses[j]) {

  452.                     floor_setup->class_masterbook[j]=get_bits(gb, 8);

  453. 

  454.                     AV_DEBUG("   masterbook: %d \n", floor_setup->class_masterbook[j]);

  455.                 }

  456. 

  457.                 for(k=0;k<(1<<floor_setup->class_subclasses[j]);++k) {

  458.                     floor_setup->subclass_books[j][k]=get_bits(gb, 8)-1;

  459. 

  460.                     AV_DEBUG("    book %d. : %d \n", k, floor_setup->subclass_books[j][k]);

  461.                 }

  462.             }

  463. 

  464.             floor_setup->multiplier=get_bits(gb, 2)+1;

  465.             floor_setup->x_list_dim=2;

  466. 

  467.             for(j=0;j<floor_setup->partitions;++j) {

  468.                 floor_setup->x_list_dim+=floor_setup->class_dimensions[floor_setup->partition_class[j]];

  469.             }

  470. 

  471.             floor_setup->x_list=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));

  472.             floor_setup->x_list_order=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));

  473.             floor_setup->low_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));

  474.             floor_setup->high_neighbour=(uint_fast16_t *)av_mallocz(floor_setup->x_list_dim * sizeof(uint_fast16_t));

  475. 

  476. 

  477.             rangebits=get_bits(gb, 4);

  478.             floor_setup->x_list[0] = 0;

  479.             floor_setup->x_list[1] = (1<<rangebits);

  480. 

  481.             for(j=0;j<floor_setup->partitions;++j) {

  482.                 for(k=0;k<floor_setup->class_dimensions[floor_setup->partition_class[j]];++k,++floor1_values) {

  483.                     floor_setup->x_list[floor1_values]=get_bits(gb, rangebits);

  484. 

  485.                     AV_DEBUG(" %d. floor1 Y coord. %d \n", floor1_values, floor_setup->x_list[floor1_values]);

  486.                 }

  487.             }

  488. 

  489. // Precalculate order of x coordinates - needed for decode

  490. 

  491.             for(k=0;k<floor_setup->x_list_dim;++k) {

  492.                 floor_setup->x_list_order[k]=k;

  493.             }

  494. 

  495.             for(k=0;k<floor_setup->x_list_dim-1;++k) {   // FIXME optimize sorting ?

  496.                 for(j=k+1;j<floor_setup->x_list_dim;++j) {

  497.                     if(floor_setup->x_list[floor_setup->x_list_order[k]]>floor_setup->x_list[floor_setup->x_list_order[j]]) {

  498.                         uint_fast16_t tmp=floor_setup->x_list_order[k];

  499.                         floor_setup->x_list_order[k]=floor_setup->x_list_order[j];

  500.                         floor_setup->x_list_order[j]=tmp;

  501.                     }

  502.                 }

  503.             }

  504. 

  505. // Precalculate low and high neighbours

  506. 

  507.             for(k=2;k<floor_setup->x_list_dim;++k) {

  508.                 floor_setup->low_neighbour[k]=0;

  509.                 floor_setup->high_neighbour[k]=1;  // correct according to SPEC requirements

  510. 

  511.                 for (j=0;j<k;++j) {

  512.                     if ((floor_setup->x_list[j]<floor_setup->x_list[k]) &&

  513.                       (floor_setup->x_list[j]>floor_setup->x_list[floor_setup->low_neighbour[k]])) {

  514.                         floor_setup->low_neighbour[k]=j;

  515.                     }

  516.                     if ((floor_setup->x_list[j]>floor_setup->x_list[k]) &&

  517.                       (floor_setup->x_list[j]<floor_setup->x_list[floor_setup->high_neighbour[k]])) {

  518.                         floor_setup->high_neighbour[k]=j;

  519.                     }

  520.                 }

  521.             }

  522.         }

  523.         else {

  524.             av_log(vc->avccontext, AV_LOG_ERROR, "Only floor type 1 supported. \n");

  525.             return 1;

  526.         }

  527.     }

  528.     return 0;

  529. }

  530. 

  531. // Process residues part

  532. 

  533. static int vorbis_parse_setup_hdr_residues(vorbis_context *vc){

  534.     GetBitContext *gb=&vc->gb;

  535.     uint_fast8_t i, j, k;

  536. 

  537.     vc->residue_count=get_bits(gb, 6)+1;

  538.     vc->residues=(vorbis_residue *)av_mallocz(vc->residue_count * sizeof(vorbis_residue));

  539. 

  540.     AV_DEBUG(" There are %d residues. \n", vc->residue_count);

  541. 

  542.     for(i=0;i<vc->residue_count;++i) {

  543.         vorbis_residue *res_setup=&vc->residues[i];

  544.         uint_fast8_t cascade[64];

  545.         uint_fast8_t high_bits;

  546.         uint_fast8_t low_bits;

  547. 

  548.         res_setup->type=get_bits(gb, 16);

  549. 

  550.         AV_DEBUG(" %d. residue type %d \n", i, res_setup->type);

  551. 

  552.         res_setup->begin=get_bits(gb, 24);

  553.         res_setup->end=get_bits(gb, 24);

  554.         res_setup->partition_size=get_bits(gb, 24)+1;

  555.         res_setup->classifications=get_bits(gb, 6)+1;

  556.         res_setup->classbook=get_bits(gb, 8);

  557. 

  558.         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,

  559.           res_setup->classifications, res_setup->classbook);

  560. 

  561.         for(j=0;j<res_setup->classifications;++j) {

  562.             high_bits=0;

  563.             low_bits=get_bits(gb, 3);

  564.             if (get_bits1(gb)) {

  565.                 high_bits=get_bits(gb, 5);

  566.             }

  567.             cascade[j]=(high_bits<<3)+low_bits;

  568. 

  569.             AV_DEBUG("     %d class casscade depth: %d \n", j, ilog(cascade[j]));

  570.         }

  571. 

  572.         res_setup->maxpass=0;

  573.         for(j=0;j<res_setup->classifications;++j) {

  574.             for(k=0;k<8;++k) {

  575.                 if (cascade[j]&(1<<k)) {

  576.                         res_setup->books[j][k]=get_bits(gb, 8);

  577. 

  578.                     AV_DEBUG("     %d class casscade depth %d book: %d \n", j, k, res_setup->books[j][k]);

  579. 

  580.                     if (k>res_setup->maxpass) {

  581.                         res_setup->maxpass=k;

  582.                     }

  583.                 } else {

  584.                     res_setup->books[j][k]=-1;

  585.                 }

  586.             }

  587.         }

  588.     }

  589.     return 0;

  590. }

  591. 

  592. // Process mappings part

  593. 

  594. static int vorbis_parse_setup_hdr_mappings(vorbis_context *vc) {

  595.     GetBitContext *gb=&vc->gb;

  596.     uint_fast8_t i, j;

  597. 

  598.     vc->mapping_count=get_bits(gb, 6)+1;

  599.     vc->mappings=(vorbis_mapping *)av_mallocz(vc->mapping_count * sizeof(vorbis_mapping));

  600. 

  601.     AV_DEBUG(" There are %d mappings. \n", vc->mapping_count);

  602. 

  603.     for(i=0;i<vc->mapping_count;++i) {

  604.         vorbis_mapping *mapping_setup=&vc->mappings[i];

  605. 

  606.         if (get_bits(gb, 16)) {

  607.             av_log(vc->avccontext, AV_LOG_ERROR, "Other mappings than type 0 are not compliant with the Vorbis I specification. \n");

  608.             return 1;

  609.         }

  610.         if (get_bits1(gb)) {

  611.             mapping_setup->submaps=get_bits(gb, 4)+1;

  612.         } else {

  613.             mapping_setup->submaps=1;

  614.         }

  615. 

  616.         if (get_bits1(gb)) {

  617.             mapping_setup->coupling_steps=get_bits(gb, 8)+1;

  618.             mapping_setup->magnitude=(uint_fast8_t *)av_mallocz(mapping_setup->coupling_steps * sizeof(uint_fast8_t));

  619.             mapping_setup->angle=(uint_fast8_t *)av_mallocz(mapping_setup->coupling_steps * sizeof(uint_fast8_t));

  620.             for(j=0;j<mapping_setup->coupling_steps;++j) {

  621.                 mapping_setup->magnitude[j]=get_bits(gb, ilog(vc->audio_channels-1));

  622.                 mapping_setup->angle[j]=get_bits(gb, ilog(vc->audio_channels-1));

  623.                 // FIXME: sanity checks

  624.             }

  625.         } else {

  626.             mapping_setup->coupling_steps=0;

  627.         }

  628. 

  629.         AV_DEBUG("   %d mapping coupling steps: %d \n", i, mapping_setup->coupling_steps);

  630. 

  631.         if(get_bits(gb, 2)) {

  632.             av_log(vc->avccontext, AV_LOG_ERROR, "%d. mapping setup data invalid. \n", i);

  633.             return 1; // following spec.

  634.         }

  635. 

  636.         if (mapping_setup->submaps>1) {

  637.             mapping_setup->mux=(uint_fast8_t *)av_mallocz(vc->audio_channels * sizeof(uint_fast8_t));

  638.             for(j=0;j<vc->audio_channels;++j) {

  639.                 mapping_setup->mux[j]=get_bits(gb, 4);

  640.             }

  641.         }

  642. 

  643.         for(j=0;j<mapping_setup->submaps;++j) {

  644.             get_bits(gb, 8); // FIXME check?

  645.             mapping_setup->submap_floor[j]=get_bits(gb, 8);

  646.             mapping_setup->submap_residue[j]=get_bits(gb, 8);

  647. 

  648.             AV_DEBUG("   %d mapping %d submap : floor %d, residue %d \n", i, j, mapping_setup->submap_floor[j], mapping_setup->submap_residue[j]);

  649.         }

  650.     }

  651.     return 0;

  652. }

  653. 

  654. // Process modes part

  655. 

  656. static int vorbis_parse_setup_hdr_modes(vorbis_context *vc) {

  657.     GetBitContext *gb=&vc->gb;

  658.     uint_fast8_t i;

  659. 

  660.     vc->mode_count=get_bits(gb, 6)+1;

  661.     vc->modes=(vorbis_mode *)av_mallocz(vc->mode_count * sizeof(vorbis_mode));

  662. 

  663.     AV_DEBUG(" There are %d modes.\n", vc->mode_count);

  664. 

  665.     for(i=0;i<vc->mode_count;++i) {

  666.         vorbis_mode *mode_setup=&vc->modes[i];

  667. 

  668.         mode_setup->blockflag=get_bits(gb, 1);

  669.         mode_setup->windowtype=get_bits(gb, 16); //FIXME check

  670.         mode_setup->transformtype=get_bits(gb, 16); //FIXME check

  671.         mode_setup->mapping=get_bits(gb, 8); //FIXME check

  672. 

  673.         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);

  674.     }

  675.     return 0;

  676. }

  677. 

  678. // Process the whole setup header using the functions above

  679. 

  680. static int vorbis_parse_setup_hdr(vorbis_context *vc) {

  681.     GetBitContext *gb=&vc->gb;

  682. 

  683.     if ((get_bits(gb, 8)!='v') || (get_bits(gb, 8)!='o') ||

  684.     (get_bits(gb, 8)!='r') || (get_bits(gb, 8)!='b') ||

  685.     (get_bits(gb, 8)!='i') || (get_bits(gb, 8)!='s')) {

  686.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (no vorbis signature). \n");

  687.         return 1;

  688.     }

  689. 

  690.     if (vorbis_parse_setup_hdr_codebooks(vc)) {

  691.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (codebooks). \n");

  692.         return 2;

  693.     }

  694.     if (vorbis_parse_setup_hdr_tdtransforms(vc)) {

  695.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (time domain transforms). \n");

  696.         return 3;

  697.     }

  698.     if (vorbis_parse_setup_hdr_floors(vc)) {

  699.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (floors). \n");

  700.         return 4;

  701.     }

  702.     if (vorbis_parse_setup_hdr_residues(vc)) {

  703.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (residues). \n");

  704.         return 5;

  705.     }

  706.     if (vorbis_parse_setup_hdr_mappings(vc)) {

  707.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (mappings). \n");

  708.         return 6;

  709.     }

  710.     if (vorbis_parse_setup_hdr_modes(vc)) {

  711.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (modes). \n");

  712.         return 7;

  713.     }

  714.     if (!get_bits1(gb)) {

  715.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis setup header packet corrupt (framing flag). \n");

  716.         return 8; // framing flag bit unset error

  717.     }

  718. 

  719.     return 0;

  720. }

  721. 

  722. // Process the identification header

  723. 

  724. static int vorbis_parse_id_hdr(vorbis_context *vc){

  725.     GetBitContext *gb=&vc->gb;

  726.     uint_fast8_t bl0, bl1;

  727.     const float *vwin[8]={ vwin64, vwin128, vwin256, vwin512, vwin1024, vwin2048, vwin4096, vwin8192 };

  728. 

  729.     if ((get_bits(gb, 8)!='v') || (get_bits(gb, 8)!='o') ||

  730.     (get_bits(gb, 8)!='r') || (get_bits(gb, 8)!='b') ||

  731.     (get_bits(gb, 8)!='i') || (get_bits(gb, 8)!='s')) {

  732.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (no vorbis signature). \n");

  733.         return 1;

  734.     }

  735. 

  736.     vc->version=get_bits_long_le(gb, 32);    //FIXME check 0

  737.     vc->audio_channels=get_bits(gb, 8);   //FIXME check >0

  738.     vc->audio_samplerate=get_bits_long_le(gb, 32);   //FIXME check >0

  739.     vc->bitrate_maximum=get_bits_long_le(gb, 32);

  740.     vc->bitrate_nominal=get_bits_long_le(gb, 32);

  741.     vc->bitrate_minimum=get_bits_long_le(gb, 32);

  742.     bl0=get_bits(gb, 4);

  743.     bl1=get_bits(gb, 4);

  744.     vc->blocksize_0=(1<<bl0);

  745.     vc->blocksize_1=(1<<bl1);

  746.     if (bl0>13 || bl0<6 || bl1>13 || bl1<6) {

  747.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (illegal blocksize). \n");

  748.         return 3;

  749.     }

  750.     vc->swin=vwin[bl0-6];

  751.     vc->lwin=vwin[bl1-6];

  752. 

  753.     if ((get_bits1(gb)) == 0) {

  754.         av_log(vc->avccontext, AV_LOG_ERROR, " Vorbis id header packet corrupt (framing flag not set). \n");

  755.         return 2;

  756.     }

  757. 

  758.     vc->channel_residues=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));

  759.     vc->channel_floors=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));

  760.     vc->saved=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));

  761.     vc->ret=(float *)av_malloc((vc->blocksize_1/2)*vc->audio_channels * sizeof(float));

  762.     vc->saved_start=0;

  763. 

  764.     ff_mdct_init(&vc->mdct0, bl0, 1);

  765.     ff_mdct_init(&vc->mdct1, bl1, 1);

  766. 

  767.     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 ",

  768.             vc->version, vc->audio_channels, vc->audio_samplerate, vc->bitrate_maximum, vc->bitrate_nominal, vc->bitrate_minimum, vc->blocksize_0, vc->blocksize_1);

  769. 

  770. /*

  771.     BLK=vc->blocksize_0;

  772.     for(i=0;i<BLK/2;++i) {

  773.         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)));

  774.     }

  775. */

  776. 

  777.     return 0;

  778. }

  779. 

  780. // Process the extradata using the functions above (identification header, setup header)

  781. 

  782. static int vorbis_decode_init(AVCodecContext *avccontext) {

  783.     vorbis_context *vc = avccontext->priv_data ;

  784.     uint8_t *headers = avccontext->extradata;

  785.     int headers_len=avccontext->extradata_size;

  786.     uint8_t *header_start[3];

  787.     int header_len[3];

  788.     GetBitContext *gb = &(vc->gb);

  789.     int i, j, hdr_type;

  790. 

  791.     vc->avccontext = avccontext;

  792. 

  793.     if (!headers_len) {

  794.         av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");

  795.         return -1;

  796.     }

  797. 

  798.     if(headers[0] == 0 && headers[1] == 30) {

  799.         for(i = 0; i < 3; i++){

  800.             header_len[i] = *headers++ << 8;

  801.             header_len[i] += *headers++;

  802.             header_start[i] = headers;

  803.             headers += header_len[i];

  804.         }

  805.     } else if(headers[0] == 2) {

  806.         for(j=1,i=0;i<2;++i, ++j) {

  807.             header_len[i]=0;

  808.             while(j<headers_len && headers[j]==0xff) {

  809.                 header_len[i]+=0xff;

  810.                 ++j;

  811.             }

  812.             if (j>=headers_len) {

  813.                 av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");

  814.                 return -1;

  815.             }

  816.             header_len[i]+=headers[j];

  817.         }

  818.         header_len[2]=headers_len-header_len[0]-header_len[1]-j;

  819.         headers+=j;

  820.         header_start[0] = headers;

  821.         header_start[1] = header_start[0] + header_len[0];

  822.         header_start[2] = header_start[1] + header_len[1];

  823.     } else {

  824.         av_log(avccontext, AV_LOG_ERROR, "Extradata corrupt.\n");

  825.         return -1;

  826.     }

  827. 

  828.     init_get_bits(gb, header_start[0], header_len[0]*8);

  829.     hdr_type=get_bits(gb, 8);

  830.     if (hdr_type!=1) {

  831.         av_log(avccontext, AV_LOG_ERROR, "First header is not the id header.\n");

  832.         return -1;

  833.     }

  834.     if (vorbis_parse_id_hdr(vc)) {

  835.         av_log(avccontext, AV_LOG_ERROR, "Id header corrupt.\n");

  836.         vorbis_free(vc);

  837.         return -1;

  838.     }

  839. 

  840.     init_get_bits(gb, header_start[2], header_len[2]*8);

  841.     hdr_type=get_bits(gb, 8);

  842.     if (hdr_type!=5) {

  843.         av_log(avccontext, AV_LOG_ERROR, "Third header is not the setup header.\n");

  844.         return -1;

  845.     }

  846.     if (vorbis_parse_setup_hdr(vc)) {

  847.         av_log(avccontext, AV_LOG_ERROR, "Setup header corrupt.\n");

  848.         vorbis_free(vc);

  849.         return -1;

  850.     }

  851. 

  852.     avccontext->channels = vc->audio_channels;

  853.     avccontext->sample_rate = vc->audio_samplerate;

  854. 

  855.     return 0 ;

  856. }

  857. 

  858. // Decode audiopackets -------------------------------------------------

  859. 

  860. // Read and decode floor (type 1 only)

  861. 

  862. static uint_fast8_t vorbis_floor1_decode(vorbis_context *vc, vorbis_floor *vf, float *vec) {

  863.     GetBitContext *gb=&vc->gb;

  864.     uint_fast16_t range_v[4]={ 256, 128, 86, 64 };

  865.     uint_fast16_t range=range_v[vf->multiplier-1];

  866.     uint_fast16_t floor1_Y[vf->x_list_dim];

  867.     uint_fast16_t floor1_Y_final[vf->x_list_dim];

  868.     uint_fast8_t floor1_flag[vf->x_list_dim];

  869.     uint_fast8_t class_;

  870.     uint_fast8_t cdim;

  871.     uint_fast8_t cbits;

  872.     uint_fast8_t csub;

  873.     uint_fast8_t cval;

  874.     int_fast16_t book;

  875.     uint_fast16_t offset;

  876.     uint_fast16_t i,j;

  877.     uint_fast16_t *floor_x_sort=vf->x_list_order;

  878.     /*u*/int_fast16_t adx, ady, off, predicted; // WTF ? dy/adx= (unsigned)dy/adx ?

  879.     int_fast16_t dy, err;

  880.     uint_fast16_t lx,hx, ly, hy=0;

  881. 

  882. 

  883.     if (!get_bits1(gb)) return 1; // silence

  884. 

  885. // Read values (or differences) for the floor's points

  886. 

  887.     floor1_Y[0]=get_bits(gb, ilog(range-1));

  888.     floor1_Y[1]=get_bits(gb, ilog(range-1));

  889. 

  890.     AV_DEBUG("floor 0 Y %d floor 1 Y %d \n", floor1_Y[0], floor1_Y[1]);

  891. 

  892.     offset=2;

  893.     for(i=0;i<vf->partitions;++i) {

  894.         class_=vf->partition_class[i];

  895.         cdim=vf->class_dimensions[class_];

  896.         cbits=vf->class_subclasses[class_];

  897.         csub=(1<<cbits)-1;

  898.         cval=0;

  899. 

  900.         AV_DEBUG("Cbits %d \n", cbits);

  901. 

  902.         if (cbits) { // this reads all subclasses for this partition's class

  903.             cval=get_vlc2(gb, vc->codebooks[vf->class_masterbook[class_]].vlc.table,

  904.             V_NB_BITS, vc->codebooks[vf->class_masterbook[class_]].maxdepth);

  905.         }

  906. 

  907.         for(j=0;j<cdim;++j) {

  908.             book=vf->subclass_books[class_][cval & csub];

  909. 

  910.             AV_DEBUG("book %d Cbits %d cval %d  bits:%d \n", book, cbits, cval, get_bits_count(gb));

  911. 

  912.             cval=cval>>cbits;

  913.             if (book>0) {

  914.                 floor1_Y[offset+j]=get_vlc2(gb, vc->codebooks[book].vlc.table,

  915.                 V_NB_BITS, vc->codebooks[book].maxdepth);

  916.             } else {

  917.                 floor1_Y[offset+j]=0;

  918.             }

  919. 

  920.             AV_DEBUG(" floor(%d) = %d \n", vf->x_list[offset+j], floor1_Y[offset+j]);

  921.         }

  922.         offset+=cdim;

  923.     }

  924. 

  925. // Amplitude calculation from the differences

  926. 

  927.     floor1_flag[0]=1;

  928.     floor1_flag[1]=1;

  929.     floor1_Y_final[0]=floor1_Y[0];

  930.     floor1_Y_final[1]=floor1_Y[1];

  931. 

  932.     for(i=2;i<vf->x_list_dim;++i) {

  933.         uint_fast16_t val, highroom, lowroom, room;

  934.         uint_fast16_t high_neigh_offs;

  935.         uint_fast16_t low_neigh_offs;

  936. 

  937.         low_neigh_offs=vf->low_neighbour[i];

  938.         high_neigh_offs=vf->high_neighbour[i];

  939.         dy=floor1_Y_final[high_neigh_offs]-floor1_Y_final[low_neigh_offs];  // render_point begin

  940.         adx=vf->x_list[high_neigh_offs]-vf->x_list[low_neigh_offs];

  941.         ady= ABS(dy);

  942.         err=ady*(vf->x_list[i]-vf->x_list[low_neigh_offs]);

  943.         off=err/adx;

  944.         if (dy<0) {

  945.             predicted=floor1_Y_final[low_neigh_offs]-off;

  946.         } else {

  947.             predicted=floor1_Y_final[low_neigh_offs]+off;

  948.         } // render_point end

  949. 

  950.         val=floor1_Y[i];

  951.         highroom=range-predicted;

  952.         lowroom=predicted;

  953.         if (highroom < lowroom) {

  954.             room=highroom*2;

  955.         } else {

  956.             room=lowroom*2;   // SPEC mispelling

  957.         }

  958.         if (val) {

  959.             floor1_flag[low_neigh_offs]=1;

  960.             floor1_flag[high_neigh_offs]=1;

  961.             floor1_flag[i]=1;

  962.             if (val>=room) {

  963.                 if (highroom > lowroom) {

  964.                     floor1_Y_final[i]=val-lowroom+predicted;

  965.                 } else {

  966.                     floor1_Y_final[i]=predicted-val+highroom-1;

  967.                 }

  968.             } else {

  969.                 if (val & 1) {

  970.                     floor1_Y_final[i]=predicted-(val+1)/2;

  971.                 } else {

  972.                     floor1_Y_final[i]=predicted+val/2;

  973.                 }

  974.             }

  975.         } else {

  976.             floor1_flag[i]=0;

  977.             floor1_Y_final[i]=predicted;

  978.         }

  979. 

  980.         AV_DEBUG(" Decoded floor(%d) = %d / val %d \n", vf->x_list[i], floor1_Y_final[i], val);

  981.     }

  982. 

  983. // Curve synth - connect the calculated dots and convert from dB scale FIXME optimize ?

  984. 

  985.     hx=0;

  986.     lx=0;

  987.     ly=floor1_Y_final[0]*vf->multiplier;  // conforms to SPEC

  988. 

  989.     vec[0]=floor1_inverse_db_table[ly];

  990. 

  991.     for(i=1;i<vf->x_list_dim;++i) {

  992.         AV_DEBUG(" Looking at post %d \n", i);

  993. 

  994.         if (floor1_flag[floor_x_sort[i]]) {   // SPEC mispelled

  995.             int_fast16_t x, y, dy, base, sy; // if uncommented: dy = -32 adx = 2  base = 2blablabla ?????

  996. 

  997.             hy=floor1_Y_final[floor_x_sort[i]]*vf->multiplier;

  998.             hx=vf->x_list[floor_x_sort[i]];

  999. 

  1000.             dy=hy-ly;

  1001.             adx=hx-lx;

  1002.             ady= (dy<0) ? -dy:dy;//ABS(dy);

  1003.             base=dy/adx;

  1004. 

  1005.             AV_DEBUG(" dy %d  adx %d base %d = %d \n", dy, adx, base, dy/adx);

  1006. 

  1007.             x=lx;

  1008.             y=ly;

  1009.             err=0;

  1010.             if (dy<0) {

  1011.                 sy=base-1;

  1012.             } else {

  1013.                 sy=base+1;

  1014.             }

  1015.             ady=ady-(base<0 ? -base : base)*adx;

  1016.             vec[x]=floor1_inverse_db_table[y];

  1017. 

  1018.             AV_DEBUG(" vec[ %d ] = %d \n", x, y);

  1019. 

  1020.             for(x=lx+1;(x<hx) && (x<vf->x_list[1]);++x) {

  1021.                 err+=ady;

  1022.                 if (err>=adx) {

  1023.                     err-=adx;

  1024.                     y+=sy;

  1025.                 } else {

  1026.                     y+=base;

  1027.                 }

  1028.                 vec[x]=floor1_inverse_db_table[y];

  1029. 

  1030.                 AV_DEBUG(" vec[ %d ] = %d \n", x, y);

  1031.             }

  1032. 

  1033. /*            for(j=1;j<hx-lx+1;++j) {  // iterating render_point

  1034.                 dy=hy-ly;

  1035.                 adx=hx-lx;

  1036.                 ady= dy<0 ? -dy : dy;

  1037.                 err=ady*j;

  1038.                 off=err/adx;

  1039.                 if (dy<0) {

  1040.                     predicted=ly-off;

  1041.                 } else {

  1042.                     predicted=ly+off;

  1043.                 }

  1044.                 if (lx+j < vf->x_list[1]) {

  1045.                     vec[lx+j]=floor1_inverse_db_table[predicted];

  1046.                 }

  1047.             }*/

  1048. 

  1049.             lx=hx;

  1050.             ly=hy;

  1051.         }

  1052.     }

  1053. 

  1054.     if (hx<vf->x_list[1]) {

  1055.         for(i=hx;i<vf->x_list[1];++i) {

  1056.             vec[i]=floor1_inverse_db_table[hy];

  1057.         }

  1058.     }

  1059. 

  1060.     AV_DEBUG(" Floor decoded\n");

  1061. 

  1062.     return 0;

  1063. }

  1064. 

  1065. // Read and decode residue

  1066. 

  1067. 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) {

  1068.     GetBitContext *gb=&vc->gb;

  1069.     uint_fast8_t c_p_c=vc->codebooks[vr->classbook].dimensions;

  1070.     uint_fast16_t n_to_read=vr->end-vr->begin;

  1071.     uint_fast16_t ptns_to_read=n_to_read/vr->partition_size;

  1072.     uint_fast8_t classifs[ptns_to_read*vc->audio_channels];

  1073.     uint_fast8_t pass;

  1074.     uint_fast8_t ch_used;

  1075.     uint_fast8_t i,j,l;

  1076.     uint_fast16_t k;

  1077. 

  1078.     if (vr->type==2) {

  1079.         for(j=1;j<ch;++j) {

  1080.                 do_not_decode[0]&=do_not_decode[j];  // FIXME - clobbering input

  1081.         }

  1082.         if (do_not_decode[0]) return 0;

  1083.         ch_used=1;

  1084.     } else {

  1085.         ch_used=ch;

  1086.     }

  1087. 

  1088.     AV_DEBUG(" residue type 0/1/2 decode begin, ch: %d  cpc %d  \n", ch, c_p_c);

  1089. 

  1090.     for(pass=0;pass<=vr->maxpass;++pass) { // FIXME OPTIMIZE?

  1091.         uint_fast16_t voffset;

  1092.         uint_fast16_t partition_count;

  1093.         uint_fast16_t j_times_ptns_to_read;

  1094. 

  1095.         voffset=vr->begin;

  1096.         for(partition_count=0;partition_count<ptns_to_read;) {  // SPEC        error

  1097.             if (!pass) {

  1098.                 for(j_times_ptns_to_read=0, j=0;j<ch_used;++j) {

  1099.                     if (!do_not_decode[j]) {

  1100.                         uint_fast32_t temp=get_vlc2(gb, vc->codebooks[vr->classbook].vlc.table,

  1101.                         V_NB_BITS, vc->codebooks[vr->classbook].maxdepth);

  1102. 

  1103.                         AV_DEBUG("Classword: %d \n", temp);

  1104. 

  1105.                         for(i=0;i<c_p_c;++i) {

  1106.                             uint_fast32_t temp2;

  1107. 

  1108.                             temp2=temp/vr->classifications;

  1109.                             classifs[j_times_ptns_to_read+partition_count+c_p_c-1-i]=temp-temp2*vr->classifications;

  1110.                             temp=temp2;

  1111.                         }

  1112.                     }

  1113.                     j_times_ptns_to_read+=ptns_to_read;

  1114.                 }

  1115.             }

  1116.             for(i=0;(i<c_p_c) && (partition_count<ptns_to_read);++i) {

  1117.                 for(j_times_ptns_to_read=0, j=0;j<ch_used;++j) {

  1118.                     uint_fast16_t voffs;

  1119. 

  1120.                     if (!do_not_decode[j]) {

  1121.                         uint_fast8_t vqclass=classifs[j_times_ptns_to_read+partition_count];

  1122.                         int_fast16_t vqbook=vr->books[vqclass][pass];

  1123. 

  1124.                         if (vqbook>=0) {

  1125.                             uint_fast16_t coffs;

  1126. 

  1127.                             if (vr->type==0) {

  1128.                                 uint_fast16_t step=vr->partition_size/vc->codebooks[vqbook].dimensions;

  1129. 

  1130.                                 voffs=voffset+j*vlen;

  1131.                                 for(k=0;k<step;++k) {

  1132.                                     coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,

  1133.                                     V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;

  1134.                                     for(l=0;l<vc->codebooks[vqbook].dimensions;++l) {

  1135.                                         vec[voffs+k+l*step]+=vc->codebooks[vqbook].codevectors[coffs+l];  // FPMATH

  1136.                                     }

  1137.                                 }

  1138.                             }

  1139.                             else if (vr->type==1) {

  1140.                                 voffs=voffset+j*vlen;

  1141.                                 for(k=0;k<vr->partition_size/vc->codebooks[vqbook].dimensions;++k) {

  1142.                                     coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,

  1143.                                     V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;

  1144.                                     for(l=0;l<vc->codebooks[vqbook].dimensions;++l, ++voffs) {

  1145.                                         vec[voffs]+=vc->codebooks[vqbook].codevectors[coffs+l];  // FPMATH

  1146. 

  1147.                                         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);

  1148.                                     }

  1149.                                 }

  1150.                             }

  1151.                             else if (vr->type==2 && ch==2) { // most frequent case optimized

  1152.                                 voffs=voffset;

  1153. 

  1154.                                 for(k=0;k<vr->partition_size/vc->codebooks[vqbook].dimensions;++k) {

  1155.                                     coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,

  1156.                                     V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;

  1157.                                     for(l=0;l<vc->codebooks[vqbook].dimensions;++l, ++voffs) {

  1158.                                         vec[(voffs>>1)+((voffs&1) ? vlen : 0)]+=vc->codebooks[vqbook].codevectors[coffs+l];  // FPMATH

  1159. 

  1160.                                         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);

  1161.                                     }

  1162.                                 }

  1163. 

  1164.                             }

  1165.                             else if (vr->type==2) {

  1166.                                 voffs=voffset;

  1167. 

  1168.                                 for(k=0;k<vr->partition_size/vc->codebooks[vqbook].dimensions;++k) {

  1169.                                     coffs=get_vlc2(gb, vc->codebooks[vqbook].vlc.table,

  1170.                                     V_NB_BITS, vc->codebooks[vr->classbook].maxdepth) * vc->codebooks[vqbook].dimensions;

  1171.                                     for(l=0;l<vc->codebooks[vqbook].dimensions;++l, ++voffs) {

  1172.                                         vec[voffs/ch+(voffs%ch)*vlen]+=vc->codebooks[vqbook].codevectors[coffs+l];  // FPMATH FIXME use if and counter instead of / and %

  1173. 

  1174.                                         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);

  1175.                                     }

  1176.                                 }

  1177.                             } else {

  1178.                                 av_log(vc->avccontext, AV_LOG_ERROR, " Invalid residue type while residue decode?! \n");

  1179.                                 return 1;

  1180.                             }

  1181.                         }

  1182.                     }

  1183.                     j_times_ptns_to_read+=ptns_to_read;

  1184.                 }

  1185.                 ++partition_count;

  1186.                 voffset+=vr->partition_size;

  1187.             }

  1188.         }

  1189.     }

  1190.     return 0;

  1191. }

  1192. 

  1193. // Decode the audio packet using the functions above

  1194. 

  1195. static int vorbis_parse_audio_packet(vorbis_context *vc) {

  1196.     GetBitContext *gb=&vc->gb;

  1197. 

  1198.     uint_fast8_t previous_window=0,next_window=0;

  1199.     uint_fast8_t mode_number;

  1200.     uint_fast16_t blocksize;

  1201.     int_fast32_t i,j;

  1202.     uint_fast8_t no_residue[vc->audio_channels];

  1203.     uint_fast8_t do_not_decode[vc->audio_channels];

  1204.     vorbis_mapping *mapping;

  1205.     float *ch_res_ptr=vc->channel_residues;

  1206.     float *ch_floor_ptr=vc->channel_floors;

  1207.     uint_fast8_t res_chan[vc->audio_channels];

  1208.     uint_fast8_t res_num=0;

  1209.     int_fast16_t retlen=0;

  1210.     uint_fast16_t saved_start=0;

  1211. 

  1212.     if (get_bits1(gb)) {

  1213.         av_log(vc->avccontext, AV_LOG_ERROR, "Not a Vorbis I audio packet.\n");

  1214.         return -1; // packet type not audio

  1215.     }

  1216. 

  1217.     if (vc->mode_count==1) {

  1218.         mode_number=0;

  1219.     } else {

  1220.         mode_number=get_bits(gb, ilog(vc->mode_count-1));

  1221.     }

  1222.     mapping=&vc->mappings[vc->modes[mode_number].mapping];

  1223. 

  1224.     AV_DEBUG(" Mode number: %d , mapping: %d , blocktype %d \n", mode_number, vc->modes[mode_number].mapping, vc->modes[mode_number].blockflag);

  1225. 

  1226.     if (vc->modes[mode_number].blockflag) {

  1227.         previous_window=get_bits1(gb);

  1228.         next_window=get_bits1(gb);

  1229.     }

  1230. 

  1231.     blocksize=vc->modes[mode_number].blockflag ? vc->blocksize_1 : vc->blocksize_0;

  1232.     memset(ch_res_ptr, 0, sizeof(float)*vc->audio_channels*blocksize/2); //FIXME can this be removed ?

  1233.     memset(ch_floor_ptr, 0, sizeof(float)*vc->audio_channels*blocksize/2); //FIXME can this be removed ?

  1234. 

  1235. // Decode floor(1)

  1236. 

  1237.     for(i=0;i<vc->audio_channels;++i) {

  1238.         vorbis_floor *floor;

  1239.         if (mapping->submaps>1) {

  1240.             floor=&vc->floors[mapping->submap_floor[mapping->mux[i]]];

  1241.         } else {

  1242.             floor=&vc->floors[mapping->submap_floor[0]];

  1243.         }

  1244. 

  1245.         no_residue[i]=vorbis_floor1_decode(vc, floor, ch_floor_ptr);

  1246.         ch_floor_ptr+=blocksize/2;

  1247.     }

  1248. 

  1249. // Nonzero vector propagate

  1250. 

  1251.     for(i=mapping->coupling_steps-1;i>=0;--i) {

  1252.         if (!(no_residue[mapping->magnitude[i]] & no_residue[mapping->angle[i]])) {

  1253.             no_residue[mapping->magnitude[i]]=0;

  1254.             no_residue[mapping->angle[i]]=0;

  1255.         }

  1256.     }

  1257. 

  1258. // Decode residue

  1259. 

  1260.     for(i=0;i<mapping->submaps;++i) {

  1261.         vorbis_residue *residue;

  1262.         uint_fast8_t ch=0;

  1263. 

  1264.         for(j=0;j<vc->audio_channels;++j) {

  1265.             if ((mapping->submaps==1) || (i=mapping->mux[j])) {

  1266.                 res_chan[j]=res_num;

  1267.                 if (no_residue[j]) {

  1268.                     do_not_decode[ch]=1;

  1269.                 } else {

  1270.                     do_not_decode[ch]=0;

  1271.                 }

  1272.                 ++ch;

  1273.                 ++res_num;

  1274.             }

  1275.         }

  1276.         residue=&vc->residues[mapping->submap_residue[i]];

  1277.         vorbis_residue_decode(vc, residue, ch, do_not_decode, ch_res_ptr, blocksize/2);

  1278. 

  1279.         ch_res_ptr+=ch*blocksize/2;

  1280.     }

  1281. 

  1282. // Inverse coupling

  1283. 

  1284.     for(i=mapping->coupling_steps-1;i>=0;--i) { //warning: i has to be signed

  1285.         float *mag, *ang;

  1286. 

  1287.         mag=vc->channel_residues+res_chan[mapping->magnitude[i]]*blocksize/2;

  1288.         ang=vc->channel_residues+res_chan[mapping->angle[i]]*blocksize/2;

  1289.         for(j=0;j<blocksize/2;++j) {

  1290.             float temp;

  1291.             if (mag[j]>0.0) {

  1292.                 if (ang[j]>0.0) {

  1293.                     ang[j]=mag[j]-ang[j];

  1294.                 } else {

  1295.                     temp=ang[j];

  1296.                     ang[j]=mag[j];

  1297.                     mag[j]+=temp;

  1298.                 }

  1299.             } else {

  1300.                 if (ang[j]>0.0) {

  1301.                     ang[j]+=mag[j];

  1302.                 } else {

  1303.                     temp=ang[j];

  1304.                     ang[j]=mag[j];

  1305.                     mag[j]-=temp;

  1306.                 }

  1307.             }

  1308.         }

  1309.     }

  1310. 

  1311. // Dotproduct

  1312. 

  1313.     for(j=0, ch_floor_ptr=vc->channel_floors;j<vc->audio_channels;++j,ch_floor_ptr+=blocksize/2) {

  1314.         ch_res_ptr=vc->channel_residues+res_chan[j]*blocksize/2;

  1315. 

  1316.         for(i=0;i<blocksize/2;++i) {

  1317.             ch_floor_ptr[i]*=ch_res_ptr[i]; //FPMATH

  1318.         }

  1319.     }

  1320. 

  1321. // MDCT, overlap/add, save data for next overlapping  FPMATH

  1322. 

  1323.     for(j=0;j<vc->audio_channels;++j) {

  1324.         uint_fast8_t step=vc->audio_channels;

  1325.         uint_fast16_t k;

  1326.         float *saved=vc->saved+j*vc->blocksize_1/2;

  1327.         float *ret=vc->ret;

  1328.         const float *lwin=vc->lwin;

  1329.         const float *swin=vc->swin;

  1330.         float buf[blocksize];

  1331.         float buf_tmp[blocksize];

  1332. 

  1333.         ch_floor_ptr=vc->channel_floors+j*blocksize/2;

  1334. 

  1335.         saved_start=vc->saved_start;

  1336. 

  1337.         ff_imdct_calc(vc->modes[mode_number].blockflag ? &vc->mdct1 : &vc->mdct0, buf, ch_floor_ptr, buf_tmp);

  1338. 

  1339.         if (vc->modes[mode_number].blockflag) {

  1340.             // -- overlap/add

  1341.             if (previous_window) {

  1342.                 for(k=j, i=0;i<vc->blocksize_1/2;++i, k+=step) {

  1343.                     ret[k]=saved[i]+buf[i]*lwin[i];

  1344.                 }

  1345.                 retlen=vc->blocksize_1/2;

  1346.             } else {

  1347.                 for(k=j, i=0;i<vc->blocksize_0/2;++i, k+=step) {

  1348.                     ret[k]=saved[i]+buf[(vc->blocksize_1-vc->blocksize_0)/4+i]*swin[i];

  1349.                 }

  1350.                 for(i=0;i<(vc->blocksize_1-vc->blocksize_0)/4;++i, k+=step) {

  1351.                     ret[k]=buf[vc->blocksize_0/2+(vc->blocksize_1-vc->blocksize_0)/4+i];

  1352.                 }

  1353.                 retlen=vc->blocksize_0/2+(vc->blocksize_1-vc->blocksize_0)/4;

  1354.             }

  1355.             // -- save

  1356.             if (next_window) {

  1357.                 for(i=0;i<vc->blocksize_1/2;++i) {

  1358.                     saved[i]=buf[vc->blocksize_1/2+i]*lwin[vc->blocksize_1/2-1-i];

  1359.                 }

  1360.                 saved_start=0;

  1361.             } else {

  1362.                 saved_start=(vc->blocksize_1-vc->blocksize_0)/4;

  1363.                 for(i=0;i<saved_start;++i) {

  1364.                     saved[i]=buf[vc->blocksize_1/2+i];

  1365.                 }

  1366.                 for(i=0;i<vc->blocksize_0/2;++i) {

  1367.                     saved[saved_start+i]=buf[vc->blocksize_1/2+saved_start+i]*swin[vc->blocksize_0/2-1-i];

  1368.                 }

  1369.             }

  1370.         } else {

  1371.             // --overlap/add

  1372.             for(k=j, i=0;i<saved_start;++i, k+=step) {

  1373.                 ret[k]=saved[i];

  1374.             }

  1375.             for(i=0;i<vc->blocksize_0/2;++i, k+=step) {

  1376.                 ret[k]=saved[saved_start+i]+buf[i]*swin[i];

  1377.             }

  1378.             retlen=saved_start+vc->blocksize_0/2;

  1379.             // -- save

  1380.             for(i=0;i<vc->blocksize_0/2;++i) {

  1381.                 saved[i]=buf[vc->blocksize_0/2+i]*swin[vc->blocksize_0/2-1-i];

  1382.             }

  1383.             saved_start=0;

  1384.         }

  1385.     }

  1386.     vc->saved_start=saved_start;

  1387. 

  1388.     return retlen*vc->audio_channels;

  1389. }

  1390. 

  1391. // Return the decoded audio packet through the standard api

  1392. 

  1393. static int vorbis_decode_frame(AVCodecContext *avccontext,

  1394.                         void *data, int *data_size,

  1395.                         uint8_t *buf, int buf_size)

  1396. {

  1397.     vorbis_context *vc = avccontext->priv_data ;

  1398.     GetBitContext *gb = &(vc->gb);

  1399. 

  1400.     int_fast16_t i, len;

  1401. 

  1402.     if(!buf_size){

  1403.         return 0;

  1404.     }

  1405. 

  1406.     AV_DEBUG("packet length %d \n", buf_size);

  1407. 

  1408.     init_get_bits(gb, buf, buf_size*8);

  1409. 

  1410.     len=vorbis_parse_audio_packet(vc);

  1411. 

  1412.     if (len<=0) {

  1413.         *data_size=0;

  1414.         return buf_size;

  1415.     }

  1416. 

  1417.     if (!vc->first_frame) {

  1418.         vc->first_frame=1;

  1419.         *data_size=0;

  1420.         return buf_size ;

  1421.     }

  1422. 

  1423.     AV_DEBUG("parsed %d bytes %d bits, returned %d samples (*ch*bits) \n", get_bits_count(gb)/8, get_bits_count(gb)%8, len);

  1424. 

  1425.     for(i=0;i<len;++i) {

  1426.         int_fast32_t tmp;

  1427. 

  1428.         tmp=vc->ret[i]*32768;

  1429.         if (tmp>32767) tmp=32767;

  1430.         if (tmp<-32768) tmp=-32768;

  1431.         ((int16_t*)data)[i]=tmp;

  1432.     }

  1433.     *data_size=len*2;

  1434. 

  1435.     return buf_size ;

  1436. }

  1437. 

  1438. // Close decoder

  1439. 

  1440. static int vorbis_decode_close(AVCodecContext *avccontext) {

  1441.     vorbis_context *vc = avccontext->priv_data;

  1442. 

  1443.     vorbis_free(vc);

  1444. 

  1445.     return 0 ;

  1446. }

  1447. 

  1448. AVCodec vorbis_decoder = {

  1449.     "vorbis",

  1450.     CODEC_TYPE_AUDIO,

  1451.     CODEC_ID_VORBIS,

  1452.     sizeof(vorbis_context),

  1453.     vorbis_decode_init,

  1454.     NULL,

  1455.     vorbis_decode_close,

  1456.     vorbis_decode_frame,

  1457. };