codebook.c
上传用户:wstnjxml
上传日期:2014-04-03
资源大小:7248k
文件大小:9k
源码类别:

Windows CE

开发平台:

C/C++

  1. /********************************************************************
  2.  *                                                                  *
  3.  * THIS FILE IS PART OF THE OggVorbis 'TREMOR' CODEC SOURCE CODE.   *
  4.  *                                                                  *
  5.  * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *
  6.  * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
  7.  * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *
  8.  *                                                                  *
  9.  * THE OggVorbis 'TREMOR' SOURCE CODE IS (C) COPYRIGHT 1994-2002    *
  10.  * BY THE Xiph.Org FOUNDATION http://www.xiph.org/                  *
  11.  *                                                                  *
  12.  ********************************************************************
  13.  function: basic codebook pack/unpack/code/decode operations
  14.  ********************************************************************/
  15. #include <stdlib.h>
  16. #include <string.h>
  17. #include "ogg.h"
  18. #include "ivorbiscodec.h"
  19. #include "codebook.h"
  20. #include "misc.h"
  21. #include "os.h"
  22. /* unpacks a codebook from the packet buffer into the codebook struct,
  23.    readies the codebook auxiliary structures for decode *************/
  24. int vorbis_staticbook_unpack(oggpack_buffer *opb,static_codebook *s){
  25.   long i,j;
  26.   memset(s,0,sizeof(*s));
  27.   /* make sure alignment is correct */
  28.   if(oggpack_read(opb,24)!=0x564342)goto _eofout;
  29.   /* first the basic parameters */
  30.   s->dim=oggpack_read(opb,16);
  31.   s->entries=oggpack_read(opb,24);
  32.   if(s->entries==-1)goto _eofout;
  33.   /* codeword ordering.... length ordered or unordered? */
  34.   switch((int)oggpack_read(opb,1)){
  35.   case 0:
  36.     /* unordered */
  37.     s->lengthlist=(long *)_ogg_malloc(sizeof(*s->lengthlist)*s->entries);
  38.     /* allocated but unused entries? */
  39.     if(oggpack_read(opb,1)){
  40.       /* yes, unused entries */
  41.       for(i=0;i<s->entries;i++){
  42. if(oggpack_read(opb,1)){
  43.   long num=oggpack_read(opb,5);
  44.   if(num==-1)goto _eofout;
  45.   s->lengthlist[i]=num+1;
  46. }else
  47.   s->lengthlist[i]=0;
  48.       }
  49.     }else{
  50.       /* all entries used; no tagging */
  51.       for(i=0;i<s->entries;i++){
  52. long num=oggpack_read(opb,5);
  53. if(num==-1)goto _eofout;
  54. s->lengthlist[i]=num+1;
  55.       }
  56.     }
  57.     
  58.     break;
  59.   case 1:
  60.     /* ordered */
  61.     {
  62.       long length=oggpack_read(opb,5)+1;
  63.       s->lengthlist=(long *)_ogg_malloc(sizeof(*s->lengthlist)*s->entries);
  64.       for(i=0;i<s->entries;){
  65. long num=oggpack_read(opb,_ilog(s->entries-i));
  66. if(num==-1)goto _eofout;
  67. for(j=0;j<num && i<s->entries;j++,i++)
  68.   s->lengthlist[i]=length;
  69. length++;
  70.       }
  71.     }
  72.     break;
  73.   default:
  74.     /* EOF */
  75.     return(-1);
  76.   }
  77.   
  78.   /* Do we have a mapping to unpack? */
  79.   switch((s->maptype=oggpack_read(opb,4))){
  80.   case 0:
  81.     /* no mapping */
  82.     break;
  83.   case 1: case 2:
  84.     /* implicitly populated value mapping */
  85.     /* explicitly populated value mapping */
  86.     s->q_min=oggpack_read(opb,32);
  87.     s->q_delta=oggpack_read(opb,32);
  88.     s->q_quant=oggpack_read(opb,4)+1;
  89.     s->q_sequencep=oggpack_read(opb,1);
  90.     {
  91.       int quantvals=0;
  92.       switch(s->maptype){
  93.       case 1:
  94. quantvals=_book_maptype1_quantvals(s);
  95. break;
  96.       case 2:
  97. quantvals=s->entries*s->dim;
  98. break;
  99.       }
  100.       
  101.       /* quantized values */
  102.       s->quantlist=(long *)_ogg_malloc(sizeof(*s->quantlist)*quantvals);
  103.       for(i=0;i<quantvals;i++)
  104. s->quantlist[i]=oggpack_read(opb,s->q_quant);
  105.       
  106.       if(quantvals&&s->quantlist[quantvals-1]==-1)goto _eofout;
  107.     }
  108.     break;
  109.   default:
  110.     goto _errout;
  111.   }
  112.   /* all set */
  113.   return(0);
  114.   
  115.  _errout:
  116.  _eofout:
  117.   vorbis_staticbook_clear(s);
  118.   return(-1); 
  119. }
  120. /* the 'eliminate the decode tree' optimization actually requires the
  121.    codewords to be MSb first, not LSb.  This is an annoying inelegancy
  122.    (and one of the first places where carefully thought out design
  123.    turned out to be wrong; Vorbis II and future Ogg codecs should go
  124.    to an MSb bitpacker), but not actually the huge hit it appears to
  125.    be.  The first-stage decode table catches most words so that
  126.    bitreverse is not in the main execution path. */
  127. static ogg_uint32_t bitreverse(ogg_uint32_t x){
  128.   x=    ((x>>16)&0x0000ffff) | ((x<<16)&0xffff0000);
  129.   x=    ((x>> 8)&0x00ff00ff) | ((x<< 8)&0xff00ff00);
  130.   x=    ((x>> 4)&0x0f0f0f0f) | ((x<< 4)&0xf0f0f0f0);
  131.   x=    ((x>> 2)&0x33333333) | ((x<< 2)&0xcccccccc);
  132.   return((x>> 1)&0x55555555) | ((x<< 1)&0xaaaaaaaa);
  133. }
  134. static inline long decode_packed_entry_number(codebook *book, 
  135.       oggpack_buffer *b){
  136.   int  read=book->dec_maxlength;
  137.   long lo,hi;
  138.   long lok = oggpack_look(b,book->dec_firsttablen);
  139.  
  140.   if (lok >= 0) {
  141.     long entry = book->dec_firsttable[lok];
  142.     if(entry&0x80000000UL){
  143.       lo=(entry>>15)&0x7fff;
  144.       hi=book->used_entries-(entry&0x7fff);
  145.     }else{
  146.       oggpack_adv(b, book->dec_codelengths[entry-1]);
  147.       return(entry-1);
  148.     }
  149.   }else{
  150.     lo=0;
  151.     hi=book->used_entries;
  152.   }
  153.   lok = oggpack_look(b, read);
  154.   while(lok<0 && read>1)
  155.     lok = oggpack_look(b, --read);
  156.   if(lok<0)return -1;
  157.   /* bisect search for the codeword in the ordered list */
  158.   {
  159.     ogg_uint32_t testword=bitreverse((ogg_uint32_t)lok);
  160.     while(hi-lo>1){
  161.       long p=(hi-lo)>>1;
  162.       long test=book->codelist[lo+p]>testword;    
  163.       lo+=p&(test-1);
  164.       hi-=p&(-test);
  165.     }
  166.     if(book->dec_codelengths[lo]<=read){
  167.       oggpack_adv(b, book->dec_codelengths[lo]);
  168.       return(lo);
  169.     }
  170.   }
  171.   
  172.   oggpack_adv(b, read);
  173.   return(-1);
  174. }
  175. /* Decode side is specced and easier, because we don't need to find
  176.    matches using different criteria; we simply read and map.  There are
  177.    two things we need to do 'depending':
  178.    
  179.    We may need to support interleave.  We don't really, but it's
  180.    convenient to do it here rather than rebuild the vector later.
  181.    Cascades may be additive or multiplicitive; this is not inherent in
  182.    the codebook, but set in the code using the codebook.  Like
  183.    interleaving, it's easiest to do it here.  
  184.    addmul==0 -> declarative (set the value)
  185.    addmul==1 -> additive
  186.    addmul==2 -> multiplicitive */
  187. /* returns the [original, not compacted] entry number or -1 on eof *********/
  188. long vorbis_book_decode(codebook *book, oggpack_buffer *b){
  189.   long packed_entry=decode_packed_entry_number(book,b);
  190.   if(packed_entry>=0)
  191.     return(book->dec_index[packed_entry]);
  192.   
  193.   /* if there's no dec_index, the codebook unpacking isn't collapsed */
  194.   return(packed_entry);
  195. }
  196. /* returns 0 on OK or -1 on eof *************************************/
  197. long vorbis_book_decodevs_add(codebook *book,ogg_int32_t *a,
  198.       oggpack_buffer *b,int n,int point){
  199.   int step=n/book->dim;
  200.   long *entry = (long *)alloca(sizeof(*entry)*step);
  201.   ogg_int32_t **t = (ogg_int32_t **)alloca(sizeof(*t)*step);
  202.   int i,j,o;
  203.   int shift=point-book->binarypoint;
  204.   if(shift>=0){
  205.     for (i = 0; i < step; i++) {
  206.       entry[i]=decode_packed_entry_number(book,b);
  207.       if(entry[i]==-1)return(-1);
  208.       t[i] = book->valuelist+entry[i]*book->dim;
  209.     }
  210.     for(i=0,o=0;i<book->dim;i++,o+=step)
  211.       for (j=0;j<step;j++)
  212. a[o+j]+=t[j][i]>>shift;
  213.   }else{
  214.     for (i = 0; i < step; i++) {
  215.       entry[i]=decode_packed_entry_number(book,b);
  216.       if(entry[i]==-1)return(-1);
  217.       t[i] = book->valuelist+entry[i]*book->dim;
  218.     }
  219.     for(i=0,o=0;i<book->dim;i++,o+=step)
  220.       for (j=0;j<step;j++)
  221. a[o+j]+=t[j][i]<<-shift;
  222.   }
  223.   return(0);
  224. }
  225. long vorbis_book_decodev_add(codebook *book,ogg_int32_t *a,
  226.      oggpack_buffer *b,int n,int point){
  227.   int i,j,entry;
  228.   ogg_int32_t *t;
  229.   int shift=point-book->binarypoint;
  230.   
  231.   if(shift>=0){
  232.     for(i=0;i<n;){
  233.       entry = decode_packed_entry_number(book,b);
  234.       if(entry==-1)return(-1);
  235.       t     = book->valuelist+entry*book->dim;
  236.       for (j=0;j<book->dim;)
  237. a[i++]+=t[j++]>>shift;
  238.     }
  239.   }else{
  240.     for(i=0;i<n;){
  241.       entry = decode_packed_entry_number(book,b);
  242.       if(entry==-1)return(-1);
  243.       t     = book->valuelist+entry*book->dim;
  244.       for (j=0;j<book->dim;)
  245. a[i++]+=t[j++]<<-shift;
  246.     }
  247.   }
  248.   return(0);
  249. }
  250. long vorbis_book_decodev_set(codebook *book,ogg_int32_t *a,
  251.      oggpack_buffer *b,int n,int point){
  252.   int i,j,entry;
  253.   ogg_int32_t *t;
  254.   int shift=point-book->binarypoint;
  255.   
  256.   if(shift>=0){
  257.     for(i=0;i<n;){
  258.       entry = decode_packed_entry_number(book,b);
  259.       if(entry==-1)return(-1);
  260.       t     = book->valuelist+entry*book->dim;
  261.       for (j=0;j<book->dim;){
  262. a[i++]=t[j++]>>shift;
  263.       }
  264.     }
  265.   }else{
  266.     for(i=0;i<n;){
  267.       entry = decode_packed_entry_number(book,b);
  268.       if(entry==-1)return(-1);
  269.       t     = book->valuelist+entry*book->dim;
  270.       for (j=0;j<book->dim;){
  271. a[i++]=t[j++]<<-shift;
  272.       }
  273.     }
  274.   }
  275.   return(0);
  276. }
  277. long vorbis_book_decodevv_add(codebook *book,ogg_int32_t **a,
  278.       long offset,int ch,
  279.       oggpack_buffer *b,int n,int point){
  280.   long i,j,entry;
  281.   int chptr=0;
  282.   int shift=point-book->binarypoint;
  283.   
  284.   if(shift>=0){
  285.     
  286.     for(i=offset;i<offset+n;){
  287.       entry = decode_packed_entry_number(book,b);
  288.       if(entry==-1)return(-1);
  289.       {
  290. const ogg_int32_t *t = book->valuelist+entry*book->dim;
  291. for (j=0;j<book->dim;j++){
  292.   a[chptr++][i]+=t[j]>>shift;
  293.   if(chptr==ch){
  294.     chptr=0;
  295.     i++;
  296.   }
  297. }
  298.       }
  299.     }
  300.   }else{
  301.     for(i=offset;i<offset+n;){
  302.       entry = decode_packed_entry_number(book,b);
  303.       if(entry==-1)return(-1);
  304.       {
  305. const ogg_int32_t *t = book->valuelist+entry*book->dim;
  306. for (j=0;j<book->dim;j++){
  307.   a[chptr++][i]+=t[j]<<-shift;
  308.   if(chptr==ch){
  309.     chptr=0;
  310.     i++;
  311.   }
  312. }
  313.       }
  314.     }
  315.   }
  316.   return(0);
  317. }