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