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