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.29 2001/12/19 00:48:41 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   /* we cheat on the WAV header; we just bypass the header and never
  77      verify that it matches 16bit/stereo/44.1kHz.  This is just an
  78      example, after all. */
  79
  80   readbuffer[0] = '\0';
  81   for (i=0, founddata=0; i<30 && ! feof(stdin) && ! ferror(stdin); i++)
  82   {
  83     fread(readbuffer,1,2,stdin);
  84
  85     if ( ! strncmp(readbuffer, "da", 2) )
  86     {
  87       founddata = 1;
  88       fread(readbuffer,1,6,stdin);
  89       break;
  90     }
  91   }
  92
  93   /********** Encode setup ************/
  94
  95   /* choose an encoding mode */
  96   /* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
  97   vorbis_info_init(&vi);
  98
  99   vorbis_encode_init_vbr(&vi,2,44100,.4);
 100
 101   /* add a comment */
 102   vorbis_comment_init(&vc);
 103   vorbis_comment_add(&vc,"ENCODER=encoder_example.c");
 104
 105   /* set up the analysis state and auxiliary encoding storage */
 106   vorbis_analysis_init(&vd,&vi);
 107   vorbis_block_init(&vd,&vb);
 108
 109   /* set up our packet->stream encoder */
 110   /* pick a random serial number; that way we can more likely build
 111      chained streams just by concatenation */
 112   srand(time(NULL));
 113   ogg_stream_init(&os,rand());
 114
 115   /* Vorbis streams begin with three headers; the initial header (with
 116      most of the codec setup parameters) which is mandated by the Ogg
 117      bitstream spec.  The second header holds any comment fields.  The
 118      third header holds the bitstream codebook.  We merely need to
 119      make the headers, then pass them to libvorbis one at a time;
 120      libvorbis handles the additional Ogg bitstream constraints */
 121
 122   {
 123     ogg_packet header;
 124     ogg_packet header_comm;
 125     ogg_packet header_code;
 126
 127     vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
 128     ogg_stream_packetin(&os,&header); /* automatically placed in its own
 129                                          page */
 130     ogg_stream_packetin(&os,&header_comm);
 131     ogg_stream_packetin(&os,&header_code);
 132
 133         /* We don't have to write out here, but doing so makes streaming
 134          * much easier, so we do, flushing ALL pages. This ensures the actual
 135          * audio data will start on a new page
 136          */
 137         while(!eos){
 138                 int result=ogg_stream_flush(&os,&og);
 139                 if(result==0)break;
 140                 fwrite(og.header,1,og.header_len,stdout);
 141                 fwrite(og.body,1,og.body_len,stdout);
 142         }
 143
 144   }
 145
 146   while(!eos){
 147     long i;
 148     long bytes=fread(readbuffer,1,READ*4,stdin); /* stereo hardwired here */
 149
 150     if(bytes==0){
 151       /* end of file.  this can be done implicitly in the mainline,
 152          but it's easier to see here in non-clever fashion.
 153          Tell the library we're at end of stream so that it can handle
 154          the last frame and mark end of stream in the output properly */
 155       vorbis_analysis_wrote(&vd,0);
 156
 157     }else{
 158       /* data to encode */
 159
 160       /* expose the buffer to submit data */
 161       float **buffer=vorbis_analysis_buffer(&vd,READ);
 162
 163       /* uninterleave samples */
 164       for(i=0;i<bytes/4;i++){
 165         buffer[0][i]=((readbuffer[i*4+1]<<8)|
 166                       (0x00ff&(int)readbuffer[i*4]))/32768.f;
 167         buffer[1][i]=((readbuffer[i*4+3]<<8)|
 168                       (0x00ff&(int)readbuffer[i*4+2]))/32768.f;
 169       }
 170
 171       /* tell the library how much we actually submitted */
 172       vorbis_analysis_wrote(&vd,i);
 173     }
 174
 175     /* vorbis does some data preanalysis, then divvies up blocks for
 176        more involved (potentially parallel) processing.  Get a single
 177        block for encoding now */
 178     while(vorbis_analysis_blockout(&vd,&vb)==1){
 179
 180       /* analysis, assume we want to use bitrate management */
 181       vorbis_analysis(&vb,NULL);
 182       vorbis_bitrate_addblock(&vb);
 183
 184       while(vorbis_bitrate_flushpacket(&vd,&op)){
 185
 186         /* weld the packet into the bitstream */
 187         ogg_stream_packetin(&os,&op);
 188
 189         /* write out pages (if any) */
 190         while(!eos){
 191           int result=ogg_stream_pageout(&os,&og);
 192           if(result==0)break;
 193           fwrite(og.header,1,og.header_len,stdout);
 194           fwrite(og.body,1,og.body_len,stdout);
 195
 196           /* this could be set above, but for illustrative purposes, I do
 197              it here (to show that vorbis does know where the stream ends) */
 198
 199           if(ogg_page_eos(&og))eos=1;
 200         }
 201       }
 202     }
 203   }
 204
 205   /* clean up and exit.  vorbis_info_clear() must be called last */
 206
 207   ogg_stream_clear(&os);
 208   vorbis_block_clear(&vb);
 209   vorbis_dsp_clear(&vd);
 210   vorbis_comment_clear(&vc);
 211   vorbis_info_clear(&vi);
 212
 213   /* ogg_page and ogg_packet structs always point to storage in
 214      libvorbis.  They're never freed or manipulated directly */
 215
 216   fprintf(stderr,"Done.\n");
 217   return(0);
 218 }