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

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: PCM data vector blocking, windowing and dis/reassembly
  14.  ********************************************************************/
  15. #include <stdio.h>
  16. #include <stdlib.h>
  17. #include <string.h>
  18. #include "ogg.h"
  19. #include "ivorbiscodec.h"
  20. #include "codec_internal.h"
  21. #include "window.h"
  22. #include "registry.h"
  23. #include "misc.h"
  24. static int ilog(unsigned int v){
  25.   int ret=0;
  26.   if(v)--v;
  27.   while(v){
  28.     ret++;
  29.     v>>=1;
  30.   }
  31.   return(ret);
  32. }
  33. /* pcm accumulator examples (not exhaustive):
  34.  <-------------- lW ---------------->
  35.                    <--------------- W ---------------->
  36. :            .....|.....       _______________         |
  37. :        .'''     |     '''_---      |       |        |
  38. :.....'''         |_____--- '''......|       | _______|
  39. :.................|__________________|_______|__|______|
  40.                   |<------ Sl ------>|      > Sr <     |endW
  41.                   |beginSl           |endSl  |  |endSr   
  42.                   |beginW            |endlW  |beginSr
  43.                       |< lW >|       
  44.                    <--------------- W ---------------->
  45.                   |   |  ..  ______________            |
  46.                   |   | '  `/        |     ---_        |
  47.                   |___.'___/`.       |         ---_____| 
  48.                   |_______|__|_______|_________________|
  49.                   |      >|Sl|<      |<------ Sr ----->|endW
  50.                   |       |  |endSl  |beginSr          |endSr
  51.                   |beginW |  |endlW                     
  52.                   mult[0] |beginSl                     mult[n]
  53.  <-------------- lW ----------------->
  54.                           |<--W-->|                               
  55. :            ..............  ___  |   |                    
  56. :        .'''             |`/    |   |                       
  57. :.....'''                 |/`....|...|                    
  58. :.........................|___|___|___|                  
  59.                           |Sl |Sr |endW    
  60.                           |   |   |endSr
  61.                           |   |beginSr
  62.                           |   |endSl
  63.   |beginSl
  64.   |beginW
  65. */
  66. /* block abstraction setup *********************************************/
  67. #ifndef WORD_ALIGN
  68. #define WORD_ALIGN 8
  69. #endif
  70. int vorbis_block_init(vorbis_dsp_state *v, vorbis_block *vb){
  71.   memset(vb,0,sizeof(*vb));
  72.   vb->vd=v;
  73.   vb->localalloc=0;
  74.   vb->localstore=NULL;
  75.   
  76.   return(0);
  77. }
  78. void *_vorbis_block_alloc(vorbis_block *vb,long bytes){
  79.   bytes=(bytes+(WORD_ALIGN-1)) & ~(WORD_ALIGN-1);
  80.   if(bytes+vb->localtop>vb->localalloc){
  81.     /* can't just _ogg_realloc... there are outstanding pointers */
  82.     if(vb->localstore){
  83.       struct alloc_chain *link=(struct alloc_chain *)_ogg_malloc(sizeof(*link));
  84.       vb->totaluse+=vb->localtop;
  85.       link->next=vb->reap;
  86.       link->ptr=vb->localstore;
  87.       vb->reap=link;
  88.     }
  89.     /* highly conservative */
  90.     vb->localalloc=bytes;
  91.     vb->localstore=_ogg_malloc(vb->localalloc);
  92.     vb->localtop=0;
  93.   }
  94.   {
  95.     void *ret=(void *)(((char *)vb->localstore)+vb->localtop);
  96.     vb->localtop+=bytes;
  97.     return ret;
  98.   }
  99. }
  100. /* reap the chain, pull the ripcord */
  101. void _vorbis_block_ripcord(vorbis_block *vb){
  102.   /* reap the chain */
  103.   struct alloc_chain *reap=vb->reap;
  104.   while(reap){
  105.     struct alloc_chain *next=reap->next;
  106.     _ogg_free(reap->ptr);
  107.     memset(reap,0,sizeof(*reap));
  108.     _ogg_free(reap);
  109.     reap=next;
  110.   }
  111.   /* consolidate storage */
  112.   if(vb->totaluse){
  113.     vb->localstore=_ogg_realloc(vb->localstore,vb->totaluse+vb->localalloc);
  114.     vb->localalloc+=vb->totaluse;
  115.     vb->totaluse=0;
  116.   }
  117.   /* pull the ripcord */
  118.   vb->localtop=0;
  119.   vb->reap=NULL;
  120. }
  121. int vorbis_block_clear(vorbis_block *vb){
  122.   _vorbis_block_ripcord(vb);
  123.   if(vb->localstore)_ogg_free(vb->localstore);
  124.   memset(vb,0,sizeof(*vb));
  125.   return(0);
  126. }
  127. static int _vds_init(vorbis_dsp_state *v,vorbis_info *vi){
  128.   int i;
  129.   codec_setup_info *ci=(codec_setup_info *)vi->codec_setup;
  130.   private_state *b=NULL;
  131.   memset(v,0,sizeof(*v));
  132.   b=(private_state *)(v->backend_state=_ogg_calloc(1,sizeof(*b)));
  133.   v->vi=vi;
  134.   b->modebits=ilog(ci->modes);
  135.   /* Vorbis I uses only window type 0 */
  136.   b->window[0]=_vorbis_window(0,ci->blocksizes[0]/2);
  137.   b->window[1]=_vorbis_window(0,ci->blocksizes[1]/2);
  138.   /* finish the codebooks */
  139.   if(!ci->fullbooks){
  140.     ci->fullbooks=(codebook *)_ogg_calloc(ci->books,sizeof(*ci->fullbooks));
  141.     for(i=0;i<ci->books;i++){
  142.       vorbis_book_init_decode(ci->fullbooks+i,ci->book_param[i]);
  143.       /* decode codebooks are now standalone after init */
  144.       vorbis_staticbook_destroy(ci->book_param[i]);
  145.       ci->book_param[i]=NULL;
  146.     }
  147.   }
  148.   v->pcm_storage=ci->blocksizes[1];
  149.   v->pcm=(ogg_int32_t **)_ogg_malloc(vi->channels*sizeof(*v->pcm));
  150.   v->pcmret=(ogg_int32_t **)_ogg_malloc(vi->channels*sizeof(*v->pcmret));
  151.   for(i=0;i<vi->channels;i++)
  152.     v->pcm[i]=(ogg_int32_t *)_ogg_calloc(v->pcm_storage,sizeof(*v->pcm[i]));
  153.   /* all 1 (large block) or 0 (small block) */
  154.   /* explicitly set for the sake of clarity */
  155.   v->lW=0; /* previous window size */
  156.   v->W=0;  /* current window size */
  157.   /* initialize all the mapping/backend lookups */
  158.   b->mode=(vorbis_look_mapping **)_ogg_calloc(ci->modes,sizeof(*b->mode));
  159.   for(i=0;i<ci->modes;i++){
  160.     int mapnum=ci->mode_param[i]->mapping;
  161.     int maptype=ci->map_type[mapnum];
  162.     b->mode[i]=_mapping_P[maptype]->look(v,ci->mode_param[i],
  163.  ci->map_param[mapnum]);
  164.   }
  165.   return(0);
  166. }
  167. int vorbis_synthesis_restart(vorbis_dsp_state *v){
  168.   vorbis_info *vi=v->vi;
  169.   codec_setup_info *ci;
  170.   if(!v->backend_state)return -1;
  171.   if(!vi)return -1;
  172.   ci=vi->codec_setup;
  173.   if(!ci)return -1;
  174.   v->centerW=ci->blocksizes[1]/2;
  175.   v->pcm_current=v->centerW;
  176.   
  177.   v->pcm_returned=-1;
  178.   v->granulepos=-1;
  179.   v->sequence=-1;
  180.   ((private_state *)(v->backend_state))->sample_count=-1;
  181.   return(0);
  182. }
  183. int vorbis_synthesis_init(vorbis_dsp_state *v,vorbis_info *vi){
  184.   _vds_init(v,vi);
  185.   vorbis_synthesis_restart(v);
  186.   return(0);
  187. }
  188. void vorbis_dsp_clear(vorbis_dsp_state *v){
  189.   int i;
  190.   if(v){
  191.     vorbis_info *vi=v->vi;
  192.     codec_setup_info *ci=(codec_setup_info *)(vi?vi->codec_setup:NULL);
  193.     private_state *b=(private_state *)v->backend_state;
  194.     if(v->pcm){
  195.       for(i=0;i<vi->channels;i++)
  196. if(v->pcm[i])_ogg_free(v->pcm[i]);
  197.       _ogg_free(v->pcm);
  198.       if(v->pcmret)_ogg_free(v->pcmret);
  199.     }
  200.     /* free mode lookups; these are actually vorbis_look_mapping structs */
  201.     if(ci){
  202.       for(i=0;i<ci->modes;i++){
  203. int mapnum=ci->mode_param[i]->mapping;
  204. int maptype=ci->map_type[mapnum];
  205. if(b && b->mode)_mapping_P[maptype]->free_look(b->mode[i]);
  206.       }
  207.     }
  208.     if(b){
  209.       if(b->mode)_ogg_free(b->mode);    
  210.       _ogg_free(b);
  211.     }
  212.     
  213.     memset(v,0,sizeof(*v));
  214.   }
  215. }
  216. /* Unlike in analysis, the window is only partially applied for each
  217.    block.  The time domain envelope is not yet handled at the point of
  218.    calling (as it relies on the previous block). */
  219. int vorbis_synthesis_blockin(vorbis_dsp_state *v,vorbis_block *vb){
  220.   vorbis_info *vi=v->vi;
  221.   codec_setup_info *ci=(codec_setup_info *)vi->codec_setup;
  222.   private_state *b=v->backend_state;
  223.   int i,j;
  224.   if(v->pcm_current>v->pcm_returned  && v->pcm_returned!=-1)return(OV_EINVAL);
  225.   v->lW=v->W;
  226.   v->W=vb->W;
  227.   v->nW=-1;
  228.   if((v->sequence==-1)||
  229.      (v->sequence+1 != vb->sequence)){
  230.     v->granulepos=-1; /* out of sequence; lose count */
  231.     b->sample_count=-1;
  232.   }
  233.   v->sequence=vb->sequence;
  234.   
  235.   if(vb->pcm){  /* no pcm to process if vorbis_synthesis_trackonly 
  236.                    was called on block */
  237.     int n=ci->blocksizes[v->W]/2;
  238.     int n0=ci->blocksizes[0]/2;
  239.     int n1=ci->blocksizes[1]/2;
  240.     
  241.     int thisCenter;
  242.     int prevCenter;
  243.     
  244.     if(v->centerW){
  245.       thisCenter=n1;
  246.       prevCenter=0;
  247.     }else{
  248.       thisCenter=0;
  249.       prevCenter=n1;
  250.     }
  251.     
  252.     /* v->pcm is now used like a two-stage double buffer.  We don't want
  253.        to have to constantly shift *or* adjust memory usage.  Don't
  254.        accept a new block until the old is shifted out */
  255.     
  256.     /* overlap/add PCM */
  257.     
  258.     for(j=0;j<vi->channels;j++){
  259.       /* the overlap/add section */
  260.       if(v->lW){
  261. if(v->W){
  262.   /* large/large */
  263.   ogg_int32_t *pcm=v->pcm[j]+prevCenter;
  264.   ogg_int32_t *p=vb->pcm[j];
  265.   for(i=0;i<n1;i++)
  266.     pcm[i]+=p[i];
  267. }else{
  268.   /* large/small */
  269.   ogg_int32_t *pcm=v->pcm[j]+prevCenter+n1/2-n0/2;
  270.   ogg_int32_t *p=vb->pcm[j];
  271.   for(i=0;i<n0;i++)
  272.     pcm[i]+=p[i];
  273. }
  274.       }else{
  275. if(v->W){
  276.   /* small/large */
  277.   ogg_int32_t *pcm=v->pcm[j]+prevCenter;
  278.   ogg_int32_t *p=vb->pcm[j]+n1/2-n0/2;
  279.   for(i=0;i<n0;i++)
  280.     pcm[i]+=p[i];
  281.   for(;i<n1/2+n0/2;i++)
  282.     pcm[i]=p[i];
  283. }else{
  284.   /* small/small */
  285.   ogg_int32_t *pcm=v->pcm[j]+prevCenter;
  286.   ogg_int32_t *p=vb->pcm[j];
  287.   for(i=0;i<n0;i++)
  288.     pcm[i]+=p[i];
  289. }
  290.       }
  291.       
  292.       /* the copy section */
  293.       {
  294. ogg_int32_t *pcm=v->pcm[j]+thisCenter;
  295. ogg_int32_t *p=vb->pcm[j]+n;
  296. for(i=0;i<n;i++)
  297.   pcm[i]=p[i];
  298.       }
  299.     }
  300.     
  301.     if(v->centerW)
  302.       v->centerW=0;
  303.     else
  304.       v->centerW=n1;
  305.     
  306.     /* deal with initial packet state; we do this using the explicit
  307.        pcm_returned==-1 flag otherwise we're sensitive to first block
  308.        being short or long */
  309.     if(v->pcm_returned==-1){
  310.       v->pcm_returned=thisCenter;
  311.       v->pcm_current=thisCenter;
  312.     }else{
  313.       v->pcm_returned=prevCenter;
  314.       v->pcm_current=prevCenter+
  315. ci->blocksizes[v->lW]/4+
  316. ci->blocksizes[v->W]/4;
  317.     }
  318.   }
  319.     
  320.   /* track the frame number... This is for convenience, but also
  321.      making sure our last packet doesn't end with added padding.  If
  322.      the last packet is partial, the number of samples we'll have to
  323.      return will be past the vb->granulepos.
  324.      
  325.      This is not foolproof!  It will be confused if we begin
  326.      decoding at the last page after a seek or hole.  In that case,
  327.      we don't have a starting point to judge where the last frame
  328.      is.  For this reason, vorbisfile will always try to make sure
  329.      it reads the last two marked pages in proper sequence */
  330.   
  331.   if(b->sample_count==-1){
  332.     b->sample_count=0;
  333.   }else{
  334.     b->sample_count+=ci->blocksizes[v->lW]/4+ci->blocksizes[v->W]/4;
  335.   }
  336.     
  337.   if(v->granulepos==-1){
  338.     if(vb->granulepos!=-1){ /* only set if we have a position to set to */
  339.       
  340.       v->granulepos=vb->granulepos;
  341.       
  342.       /* is this a short page? */
  343.       if(b->sample_count>v->granulepos){
  344. /* corner case; if this is both the first and last audio page,
  345.    then spec says the end is cut, not beginning */
  346. if(vb->eofflag){
  347.   /* trim the end */
  348.   /* no preceeding granulepos; assume we started at zero (we'd
  349.      have to in a short single-page stream) */
  350.   /* granulepos could be -1 due to a seek, but that would result
  351.      in a long coun`t, not short count */
  352.   
  353.   v->pcm_current-=(b->sample_count-v->granulepos);
  354. }else{
  355.   /* trim the beginning */
  356.   v->pcm_returned+=(b->sample_count-v->granulepos);
  357.   if(v->pcm_returned>v->pcm_current)
  358.     v->pcm_returned=v->pcm_current;
  359. }
  360.       }
  361.       
  362.     }
  363.   }else{
  364.     v->granulepos+=ci->blocksizes[v->lW]/4+ci->blocksizes[v->W]/4;
  365.     if(vb->granulepos!=-1 && v->granulepos!=vb->granulepos){
  366.       
  367.       if(v->granulepos>vb->granulepos){
  368. long extra=v->granulepos-vb->granulepos;
  369. if(extra)
  370.   if(vb->eofflag){
  371.     /* partial last frame.  Strip the extra samples off */
  372.     v->pcm_current-=extra;
  373.   } /* else {Shouldn't happen *unless* the bitstream is out of
  374.        spec.  Either way, believe the bitstream } */
  375.       } /* else {Shouldn't happen *unless* the bitstream is out of
  376.    spec.  Either way, believe the bitstream } */
  377.       v->granulepos=vb->granulepos;
  378.     }
  379.   }
  380.   
  381.   /* Update, cleanup */
  382.   
  383.   if(vb->eofflag)v->eofflag=1;
  384.   return(0);
  385. }
  386. /* pcm==NULL indicates we just want the pending samples, no more */
  387. int vorbis_synthesis_pcmout(vorbis_dsp_state *v,ogg_int32_t ***pcm){
  388.   vorbis_info *vi=v->vi;
  389.   if(v->pcm_returned>-1 && v->pcm_returned<v->pcm_current){
  390.     if(pcm){
  391.       int i;
  392.       for(i=0;i<vi->channels;i++)
  393. v->pcmret[i]=v->pcm[i]+v->pcm_returned;
  394.       *pcm=v->pcmret;
  395.     }
  396.     return(v->pcm_current-v->pcm_returned);
  397.   }
  398.   return(0);
  399. }
  400. int vorbis_synthesis_read(vorbis_dsp_state *v,int bytes){
  401.   if(bytes && v->pcm_returned+bytes>v->pcm_current)return(OV_EINVAL);
  402.   v->pcm_returned+=bytes;
  403.   return(0);
  404. }