examples/encoder_example.c

   1 /********************************************************************
   2  *                                                                  *
   3  * THIS FILE IS PART OF THE Ogg Vorbis SOFTWARE CODEC SOURCE CODE.  *
   4  * USE, DISTRIBUTION AND REPRODUCTION OF THIS SOURCE IS GOVERNED BY *
   5  * THE GNU PUBLIC LICENSE 2, WHICH IS INCLUDED WITH THIS SOURCE.    *
   6  * PLEASE READ THESE TERMS DISTRIBUTING.                            *
   7  *                                                                  *
   8  * THE OggSQUISH 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.6 2000/01/28 15:25:07 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/modes.h"
  29
  30 #define READ 1024
  31 signed char readbuffer[READ*4+44]; /* out of the data segment, not the stack */
  32
  33 int main(){
  34   ogg_stream_state os; /* take physical pages, weld into a logical
  35                           stream of packets */
  36   ogg_page         og; /* one Ogg bitstream page.  Vorbis packets are inside */
  37   ogg_packet       op; /* one raw packet of data for decode */
  38
  39   vorbis_info     *vi; /* struct that stores all the static vorbis bitstream
  40                           settings */
  41   vorbis_comment   vc; /* struct that stores all the user comments */
  42
  43   vorbis_dsp_state vd; /* central working state for the packet->PCM decoder */
  44   vorbis_block     vb; /* local working space for packet->PCM decode */
  45
  46   int eos=0;
  47
  48   /* we cheat on the WAV header; we just bypass 44 bytes and never
  49      verify that it matches 16bit/stereo/44.1kHz.  This is just an
  50      example, after all. */
  51
  52   fread(readbuffer,1,44,stdin);
  53
  54   /********** Encode setup ************/
  55
  56   /* choose an encoding mode */
  57   /* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
  58   vi=&info_A;
  59
  60   /* add a comment */
  61   vorbis_comment_init(&vc);
  62   vorbis_comment_add(&vc,"Track encoded by encoder_example.c");
  63
  64   /* set up the analysis state and auxiliary encoding storage */
  65   vorbis_analysis_init(&vd,vi);
  66   vorbis_block_init(&vd,&vb);
  67
  68   /* set up our packet->stream encoder */
  69   /* pick a random serial number; that way we can more likely build
  70      chained streams just by concatenation */
  71   srandom(time(NULL));
  72   ogg_stream_init(&os,random());
  73
  74   /* Vorbis streams begin with three headers; the initial header (with
  75      most of the codec setup parameters) which is mandated by the Ogg
  76      bitstream spec.  The second header holds any comment fields.  The
  77      third header holds the bitstream codebook.  We merely need to
  78      make the headers, then pass them to libvorbis one at a time;
  79      libvorbis handles the additional Ogg bitstream constraints */
  80
  81   {
  82     ogg_packet header;
  83     ogg_packet header_comm;
  84     ogg_packet header_code;
  85
  86     vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
  87     ogg_stream_packetin(&os,&header); /* automatically placed in its own
  88                                          page */
  89     ogg_stream_packetin(&os,&header_comm);
  90     ogg_stream_packetin(&os,&header_code);
  91
  92     /* no need to write out here.  We'll get to that in the main loop */
  93   }
  94
  95   while(!eos){
  96     long i;
  97     long bytes=fread(readbuffer,1,READ*4,stdin); /* stereo hardwired here */
  98
  99     if(bytes==0){
 100       /* end of file.  this can be done implicitly in the mainline,
 101          but it's easier to see here in non-clever fashion.
 102          Tell the library we're at end of stream so that it can handle
 103          the last frame and mark end of stream in the output properly */
 104       vorbis_analysis_wrote(&vd,0);
 105
 106     }else{
 107       /* data to encode */
 108
 109       /* expose the buffer to submit data */
 110       double **buffer=vorbis_analysis_buffer(&vd,READ);
 111
 112       /* uninterleave samples */
 113       for(i=0;i<bytes/4;i++){
 114         buffer[0][i]=((readbuffer[i*4+1]<<8)|
 115                       (0x00ff&(int)readbuffer[i*4]))/32768.;
 116         buffer[1][i]=((readbuffer[i*4+3]<<8)|
 117                       (0x00ff&(int)readbuffer[i*4+2]))/32768.;
 118       }
 119
 120       /* tell the library how much we actually submitted */
 121       vorbis_analysis_wrote(&vd,i);
 122     }
 123
 124     /* vorbis does some data preanalysis, then divvies up blocks for
 125        more involved (potentially parallel) processing.  Get a single
 126        block for encoding now */
 127     while(vorbis_analysis_blockout(&vd,&vb)==1){
 128
 129       /* analysis */
 130       vorbis_analysis(&vb,&op);
 131
 132       /* weld the packet into the bitstream */
 133       ogg_stream_packetin(&os,&op);
 134
 135       /* write out pages (if any) */
 136       while(!eos){
 137         int result=ogg_stream_pageout(&os,&og);
 138         if(result==0)break;
 139         fwrite(og.header,1,og.header_len,stdout);
 140         fwrite(og.body,1,og.body_len,stdout);
 141
 142         /* this could be set above, but for illustrative purposes, I do
 143            it here (to show that vorbis does know where the stream ends) */
 144
 145         if(ogg_page_eos(&og))eos=1;
 146
 147       }
 148     }
 149   }
 150
 151   /* clean up and exit.  vorbis_info_clear() must be called last */
 152
 153   ogg_stream_clear(&os);
 154   vorbis_block_clear(&vb);
 155   vorbis_dsp_clear(&vd);
 156
 157   /* ogg_page and ogg_packet structs always point to storage in
 158      libvorbis.  They're never freed or manipulated directly */
 159
 160   fprintf(stderr,"Done.\n");
 161   return(0);
 162 }
 163