examples/encoder_example.c

   1 /********************************************************************
   2  *                                                                  *
   3  * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE.   *
   4  * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS     *
   5  * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
   6  * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.       *
   7  *                                                                  *
   8  * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2001             *
   9  * by the XIPHOPHORUS Company http://www.xiph.org/                  *
  10  *                                                                  *
  11  ********************************************************************
  12
  13  function: simple example encoder
  14  last mod: $Id: encoder_example.c,v 1.33 2001/12/23 10:12:02 xiphmont Exp $
  15
  16  ********************************************************************/
  17
  18 /* takes a stereo 16bit 44.1kHz WAV file from stdin and encodes it into
  19    a Vorbis bitstream */
  20
  21 /* Note that this is POSIX, not ANSI, code */
  22
  23 #include <stdio.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <time.h>
  27 #include <math.h>
  28 #include <vorbis/vorbisenc.h>
  29
  30 #ifdef _WIN32 /* We need the following two to set stdin/stdout to binary */
  31 #include <io.h>
  32 #include <fcntl.h>
  33 #endif
  34
  35 #if defined(macintosh) && defined(__MWERKS__)
  36 #include <console.h>      /* CodeWarrior's Mac "command-line" support */
  37 #endif
  38
  39 #define READ 1024
  40 signed char readbuffer[READ*4+44]; /* out of the data segment, not the stack */
  41
  42 int main(){
  43   ogg_stream_state os; /* take physical pages, weld into a logical
  44                           stream of packets */
  45   ogg_page         og; /* one Ogg bitstream page.  Vorbis packets are inside */
  46   ogg_packet       op; /* one raw packet of data for decode */
  47
  48   vorbis_info      vi; /* struct that stores all the static vorbis bitstream
  49                           settings */
  50   vorbis_comment   vc; /* struct that stores all the user comments */
  51
  52   vorbis_dsp_state vd; /* central working state for the packet->PCM decoder */
  53   vorbis_block     vb; /* local working space for packet->PCM decode */
  54
  55   int eos=0;
  56   int i, founddata;
  57
  58 #if defined(macintosh) && defined(__MWERKS__)
  59   int argc = 0;
  60   char **argv = NULL;
  61   argc = ccommand(&argv); /* get a "command line" from the Mac user */
  62                           /* this also lets the user set stdin and stdout */
  63 #endif
  64
  65   /* we cheat on the WAV header; we just bypass 44 bytes and never
  66      verify that it matches 16bit/stereo/44.1kHz.  This is just an
  67      example, after all. */
  68
  69 #ifdef _WIN32 /* We need to set stdin/stdout to binary mode. Damn windows. */
  70   /* Beware the evil ifdef. We avoid these where we can, but this one we
  71      cannot. Don't add any more, you'll probably go to hell if you do. */
  72   _setmode( _fileno( stdin ), _O_BINARY );
  73   _setmode( _fileno( stdout ), _O_BINARY );
  74 #endif
  75
  76
  77 #  include <fpu_control.h>
  78     unsigned int mask;
  79     _FPU_GETCW(mask);
  80     /* Set the Linux mask to abort on most FPE's */
  81     /* if bit is set, we _mask_ SIGFPE on that error! */
  82     /*  mask &= ~( _FPU_MASK_IM | _FPU_MASK_ZM | _FPU_MASK_OM | _FPU_MASK_UM
  83 +);*/
  84     mask &= ~( _FPU_MASK_IM | _FPU_MASK_ZM | _FPU_MASK_OM );
  85     _FPU_SETCW(mask);
  86
  87
  88   /* we cheat on the WAV header; we just bypass the header and never
  89      verify that it matches 16bit/stereo/44.1kHz.  This is just an
  90      example, after all. */
  91
  92   readbuffer[0] = '\0';
  93   for (i=0, founddata=0; i<30 && ! feof(stdin) && ! ferror(stdin); i++)
  94   {
  95     fread(readbuffer,1,2,stdin);
  96
  97     if ( ! strncmp(readbuffer, "da", 2) )
  98     {
  99       founddata = 1;
 100       fread(readbuffer,1,6,stdin);
 101       break;
 102     }
 103   }
 104
 105   /********** Encode setup ************/
 106
 107   /* choose an encoding mode */
 108   /* (quality mode .4: 44kHz stereo coupled, roughly 128kbps VBR) */
 109   vorbis_info_init(&vi);
 110
 111   vorbis_encode_init_vbr(&vi,1,44100,.4);
 112   /*vorbis_encode_init(&vi,2,44100,64000,-1,-1);*/
 113
 114   /* add a comment */
 115   vorbis_comment_init(&vc);
 116   vorbis_comment_add_tag(&vc,"ENCODER","encoder_example.c");
 117
 118   /* set up the analysis state and auxiliary encoding storage */
 119   vorbis_analysis_init(&vd,&vi);
 120   vorbis_block_init(&vd,&vb);
 121
 122   /* set up our packet->stream encoder */
 123   /* pick a random serial number; that way we can more likely build
 124      chained streams just by concatenation */
 125   srand(time(NULL));
 126   ogg_stream_init(&os,rand());
 127
 128   /* Vorbis streams begin with three headers; the initial header (with
 129      most of the codec setup parameters) which is mandated by the Ogg
 130      bitstream spec.  The second header holds any comment fields.  The
 131      third header holds the bitstream codebook.  We merely need to
 132      make the headers, then pass them to libvorbis one at a time;
 133      libvorbis handles the additional Ogg bitstream constraints */
 134
 135   {
 136     ogg_packet header;
 137     ogg_packet header_comm;
 138     ogg_packet header_code;
 139
 140     vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
 141     ogg_stream_packetin(&os,&header); /* automatically placed in its own
 142                                          page */
 143     ogg_stream_packetin(&os,&header_comm);
 144     ogg_stream_packetin(&os,&header_code);
 145
 146         /* We don't have to write out here, but doing so makes streaming
 147          * much easier, so we do, flushing ALL pages. This ensures the actual
 148          * audio data will start on a new page
 149          */
 150         while(!eos){
 151                 int result=ogg_stream_flush(&os,&og);
 152                 if(result==0)break;
 153                 fwrite(og.header,1,og.header_len,stdout);
 154                 fwrite(og.body,1,og.body_len,stdout);
 155         }
 156
 157   }
 158
 159   while(!eos){
 160     long i;
 161     long bytes=fread(readbuffer,1,READ*4,stdin); /* stereo hardwired here */
 162
 163     if(bytes==0){
 164       /* end of file.  this can be done implicitly in the mainline,
 165          but it's easier to see here in non-clever fashion.
 166          Tell the library we're at end of stream so that it can handle
 167          the last frame and mark end of stream in the output properly */
 168       vorbis_analysis_wrote(&vd,0);
 169
 170     }else{
 171       /* data to encode */
 172
 173       /* expose the buffer to submit data */
 174       float **buffer=vorbis_analysis_buffer(&vd,READ);
 175
 176       /* uninterleave samples */
 177       for(i=0;i<bytes/4;i++){
 178         buffer[0][i]=((readbuffer[i*4+1]<<8)|
 179                       (0x00ff&(int)readbuffer[i*4]))/32768.f;
 180         buffer[1][i]=((readbuffer[i*4+3]<<8)|
 181                       (0x00ff&(int)readbuffer[i*4+2]))/32768.f;
 182       }
 183
 184       /* tell the library how much we actually submitted */
 185       vorbis_analysis_wrote(&vd,i);
 186     }
 187
 188     /* vorbis does some data preanalysis, then divvies up blocks for
 189        more involved (potentially parallel) processing.  Get a single
 190        block for encoding now */
 191     while(vorbis_analysis_blockout(&vd,&vb)==1){
 192
 193       /* analysis, assume we want to use bitrate management */
 194       vorbis_analysis(&vb,NULL);
 195       vorbis_bitrate_addblock(&vb);
 196
 197       while(vorbis_bitrate_flushpacket(&vd,&op)){
 198
 199         /* weld the packet into the bitstream */
 200         ogg_stream_packetin(&os,&op);
 201
 202         /* write out pages (if any) */
 203         while(!eos){
 204           int result=ogg_stream_pageout(&os,&og);
 205           if(result==0)break;
 206           fwrite(og.header,1,og.header_len,stdout);
 207           fwrite(og.body,1,og.body_len,stdout);
 208
 209           /* this could be set above, but for illustrative purposes, I do
 210              it here (to show that vorbis does know where the stream ends) */
 211
 212           if(ogg_page_eos(&og))eos=1;
 213         }
 214       }
 215     }
 216   }
 217
 218   /* clean up and exit.  vorbis_info_clear() must be called last */
 219
 220   ogg_stream_clear(&os);
 221   vorbis_block_clear(&vb);
 222   vorbis_dsp_clear(&vd);
 223   vorbis_comment_clear(&vc);
 224   vorbis_info_clear(&vi);
 225
 226   /* ogg_page and ogg_packet structs always point to storage in
 227      libvorbis.  They're never freed or manipulated directly */
 228
 229   fprintf(stderr,"Done.\n");
 230   return(0);
 231 }