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.24 2001/09/15 04:47:48 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 <string.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(int argc, char *argv[]){
  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   FILE *fpin=NULL;
  57   FILE *fpout=NULL;
  58   char msg[512];
  59   int i, founddata;
  60
  61 #if defined(macintosh) && defined(__MWERKS__)
  62   int ac = 0;
  63   char **av = NULL;
  64   ac = ccommand(&av); /* get a "command line" from the Mac user */
  65                       /* this also lets the user set stdin and stdout */
  66 #endif
  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   /* If command line args were supplied, open the named file(s)
  76      for i/o, else maintain use of stdin/stdout.*/
  77   if (argc == 3)
  78   {
  79     if ((fpin = fopen(argv[1], "rb")) == (FILE*)NULL)
  80     {
  81       (void)sprintf(msg, "Can't open %s for reading.", argv[1]);
  82       perror(msg);
  83       return 1;
  84     }
  85
  86     if ((fpout = fopen(argv[2], "wb")) == (FILE*)NULL)
  87     {
  88       (void)sprintf(msg, "Can't open %s for writing.", argv[2]);
  89       perror(msg);
  90       return 1;
  91     }
  92   }
  93   else
  94   {
  95     fpin = stdin;
  96     fpout = stdout;
  97   }
  98
  99   /* we cheat on the WAV header; we just bypass the header and never
 100      verify that it matches 16bit/stereo/44.1kHz.  This is just an
 101      example, after all. */
 102
 103   readbuffer[0] = '\0';
 104   for (i=0, founddata=0; i<40 && ! feof(fpin) && ! ferror(fpin); i++)
 105   {
 106     fread(readbuffer,1,2,fpin);
 107
 108     if ( ! strncmp(readbuffer, "da", 2) )
 109     {
 110       founddata = 1;
 111       fread(readbuffer,1,6,fpin);
 112     }
 113   }
 114
 115   if ( feof(fpin) || ferror(fpin) )
 116   {
 117     (void)fprintf(stderr, "Error: Input WAV too short, or corrupt.\n");
 118     return(1);
 119   }
 120
 121   if ( ! founddata )
 122   {
 123     (void)fprintf(stderr, "Error: Can't find \"data\" chunk in WAV input.\n");
 124     return(1);
 125   }
 126
 127   /********** Encode setup ************/
 128
 129   /* choose an encoding mode */
 130   /* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
 131   vorbis_info_init(&vi);
 132   vorbis_encode_init(&vi,2,44100, -1, 128000, -1);
 133
 134   /* add a comment */
 135   vorbis_comment_init(&vc);
 136   vorbis_comment_add(&vc,"Track encoded by encoder_example.c");
 137
 138   /* set up the analysis state and auxiliary encoding storage */
 139   vorbis_analysis_init(&vd,&vi);
 140   vorbis_block_init(&vd,&vb);
 141
 142   /* set up our packet->stream encoder */
 143   /* pick a random serial number; that way we can more likely build
 144      chained streams just by concatenation */
 145   srand(time(NULL));
 146   ogg_stream_init(&os,rand());
 147
 148   /* Vorbis streams begin with three headers; the initial header (with
 149      most of the codec setup parameters) which is mandated by the Ogg
 150      bitstream spec.  The second header holds any comment fields.  The
 151      third header holds the bitstream codebook.  We merely need to
 152      make the headers, then pass them to libvorbis one at a time;
 153      libvorbis handles the additional Ogg bitstream constraints */
 154
 155   {
 156     ogg_packet header;
 157     ogg_packet header_comm;
 158     ogg_packet header_code;
 159
 160     vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
 161     ogg_stream_packetin(&os,&header); /* automatically placed in its own
 162                                          page */
 163     ogg_stream_packetin(&os,&header_comm);
 164     ogg_stream_packetin(&os,&header_code);
 165
 166         /* We don't have to write out here, but doing so makes streaming
 167          * much easier, so we do, flushing ALL pages. This ensures the actual
 168          * audio data will start on a new page
 169          */
 170         while(!eos){
 171                 int result=ogg_stream_flush(&os,&og);
 172                 if(result==0)break;
 173                 fwrite(og.header,1,og.header_len,fpout);
 174                 fwrite(og.body,1,og.body_len,fpout);
 175         }
 176
 177   }
 178
 179   while(!eos){
 180     long i;
 181     long bytes=fread(readbuffer,1,READ*4,fpin); /* stereo hardwired here */
 182
 183     if(bytes==0){
 184       /* end of file.  this can be done implicitly in the mainline,
 185          but it's easier to see here in non-clever fashion.
 186          Tell the library we're at end of stream so that it can handle
 187          the last frame and mark end of stream in the output properly */
 188       vorbis_analysis_wrote(&vd,0);
 189
 190     }else{
 191       /* data to encode */
 192
 193       /* expose the buffer to submit data */
 194       float **buffer=vorbis_analysis_buffer(&vd,READ);
 195
 196       /* uninterleave samples */
 197       for(i=0;i<bytes/4;i++){
 198         buffer[0][i]=((readbuffer[i*4+1]<<8)|
 199                       (0x00ff&(int)readbuffer[i*4]))/32768.f;
 200         buffer[1][i]=((readbuffer[i*4+3]<<8)|
 201                       (0x00ff&(int)readbuffer[i*4+2]))/32768.f;
 202       }
 203
 204       /* tell the library how much we actually submitted */
 205       vorbis_analysis_wrote(&vd,i);
 206     }
 207
 208     /* vorbis does some data preanalysis, then divvies up blocks for
 209        more involved (potentially parallel) processing.  Get a single
 210        block for encoding now */
 211     while(vorbis_analysis_blockout(&vd,&vb)==1){
 212
 213       /* analysis */
 214       vorbis_analysis(&vb,&op);
 215
 216       /* weld the packet into the bitstream */
 217       ogg_stream_packetin(&os,&op);
 218
 219       /* write out pages (if any) */
 220       while(!eos){
 221         int result=ogg_stream_pageout(&os,&og);
 222         if(result==0)break;
 223         fwrite(og.header,1,og.header_len,fpout);
 224         fwrite(og.body,1,og.body_len,fpout);
 225
 226         /* this could be set above, but for illustrative purposes, I do
 227            it here (to show that vorbis does know where the stream ends) */
 228
 229         if(ogg_page_eos(&og))eos=1;
 230
 231       }
 232     }
 233   }
 234
 235   /* clean up and exit.  vorbis_info_clear() must be called last */
 236
 237   ogg_stream_clear(&os);
 238   vorbis_block_clear(&vb);
 239   vorbis_dsp_clear(&vd);
 240   vorbis_comment_clear(&vc);
 241   vorbis_info_clear(&vi);
 242
 243   /* ogg_page and ogg_packet structs always point to storage in
 244      libvorbis.  They're never freed or manipulated directly */
 245
 246   fprintf(stderr,"Done.\n");
 247   return(0);
 248 }
 249