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 SOURCE IS GOVERNED BY *
   5  * THE GNU LESSER/LIBRARY PUBLIC LICENSE, WHICH IS INCLUDED WITH    *
   6  * THIS SOURCE. PLEASE READ THESE TERMS BEFORE DISTRIBUTING.        *
   7  *                                                                  *
   8  * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2000             *
   9  * by Monty <monty@xiph.org> and the XIPHOPHORUS Company            *
  10  * http://www.xiph.org/                                             *
  11  *                                                                  *
  12  ********************************************************************
  13
  14  function: simple example encoder
  15  last mod: $Id: encoder_example.c,v 1.17 2000/12/21 21:04:37 xiphmont Exp $
  16
  17  ********************************************************************/
  18
  19 /* takes a stereo 16bit 44.1kHz WAV file from stdin and encodes it into
  20    a Vorbis bitstream */
  21
  22 /* Note that this is POSIX, not ANSI, code */
  23
  24 #include <stdio.h>
  25 #include <stdlib.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
  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
  76   fread(readbuffer,1,44,stdin);
  77
  78   /********** Encode setup ************/
  79
  80   /* choose an encoding mode */
  81   /* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
  82   vorbis_info_init(&vi);
  83   vorbis_encode_init(&vi,2,44100, -1, 160000, -1);
  84
  85   /* add a comment */
  86   vorbis_comment_init(&vc);
  87   vorbis_comment_add(&vc,"Track encoded by encoder_example.c");
  88
  89   /* set up the analysis state and auxiliary encoding storage */
  90   vorbis_analysis_init(&vd,&vi);
  91   vorbis_block_init(&vd,&vb);
  92
  93   /* set up our packet->stream encoder */
  94   /* pick a random serial number; that way we can more likely build
  95      chained streams just by concatenation */
  96   srand(time(NULL));
  97   ogg_stream_init(&os,rand());
  98
  99   /* Vorbis streams begin with three headers; the initial header (with
 100      most of the codec setup parameters) which is mandated by the Ogg
 101      bitstream spec.  The second header holds any comment fields.  The
 102      third header holds the bitstream codebook.  We merely need to
 103      make the headers, then pass them to libvorbis one at a time;
 104      libvorbis handles the additional Ogg bitstream constraints */
 105
 106   {
 107     ogg_packet header;
 108     ogg_packet header_comm;
 109     ogg_packet header_code;
 110
 111     vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
 112     ogg_stream_packetin(&os,&header); /* automatically placed in its own
 113                                          page */
 114     ogg_stream_packetin(&os,&header_comm);
 115     ogg_stream_packetin(&os,&header_code);
 116
 117         /* We don't have to write out here, but doing so makes streaming
 118          * much easier, so we do, flushing ALL pages. This ensures the actual
 119          * audio data will start on a new page
 120          */
 121         while(!eos){
 122                 int result=ogg_stream_flush(&os,&og);
 123                 if(result==0)break;
 124                 fwrite(og.header,1,og.header_len,stdout);
 125                 fwrite(og.body,1,og.body_len,stdout);
 126         }
 127
 128   }
 129
 130   while(!eos){
 131     long i;
 132     long bytes=fread(readbuffer,1,READ*4,stdin); /* stereo hardwired here */
 133
 134     if(bytes==0){
 135       /* end of file.  this can be done implicitly in the mainline,
 136          but it's easier to see here in non-clever fashion.
 137          Tell the library we're at end of stream so that it can handle
 138          the last frame and mark end of stream in the output properly */
 139       vorbis_analysis_wrote(&vd,0);
 140
 141     }else{
 142       /* data to encode */
 143
 144       /* expose the buffer to submit data */
 145       float **buffer=vorbis_analysis_buffer(&vd,READ);
 146
 147       /* uninterleave samples */
 148       for(i=0;i<bytes/4;i++){
 149         buffer[0][i]=((readbuffer[i*4+1]<<8)|
 150                       (0x00ff&(int)readbuffer[i*4]))/32768.f;
 151         buffer[1][i]=((readbuffer[i*4+3]<<8)|
 152                       (0x00ff&(int)readbuffer[i*4+2]))/32768.f;
 153       }
 154
 155       /* tell the library how much we actually submitted */
 156       vorbis_analysis_wrote(&vd,i);
 157     }
 158
 159     /* vorbis does some data preanalysis, then divvies up blocks for
 160        more involved (potentially parallel) processing.  Get a single
 161        block for encoding now */
 162     while(vorbis_analysis_blockout(&vd,&vb)==1){
 163
 164       /* analysis */
 165       vorbis_analysis(&vb,&op);
 166
 167       /* weld the packet into the bitstream */
 168       ogg_stream_packetin(&os,&op);
 169
 170       /* write out pages (if any) */
 171       while(!eos){
 172         int result=ogg_stream_pageout(&os,&og);
 173         if(result==0)break;
 174         fwrite(og.header,1,og.header_len,stdout);
 175         fwrite(og.body,1,og.body_len,stdout);
 176
 177         /* this could be set above, but for illustrative purposes, I do
 178            it here (to show that vorbis does know where the stream ends) */
 179
 180         if(ogg_page_eos(&og))eos=1;
 181
 182       }
 183     }
 184   }
 185
 186   /* clean up and exit.  vorbis_info_clear() must be called last */
 187
 188   ogg_stream_clear(&os);
 189   vorbis_block_clear(&vb);
 190   vorbis_dsp_clear(&vd);
 191   vorbis_info_clear(&vi);
 192
 193   /* ogg_page and ogg_packet structs always point to storage in
 194      libvorbis.  They're never freed or manipulated directly */
 195
 196   fprintf(stderr,"Done.\n");
 197   return(0);
 198 }
 199