test/write_read.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-2007             *
   9  * by the Xiph.Org Foundation http://www.xiph.org/                  *
  10  *                                                                  *
  11  ********************************************************************
  12
  13  function: utility functions for vorbis codec test suite.
  14  last mod: $Id: util.c 13293 2007-07-24 00:09:47Z erikd $
  15
  16  ********************************************************************/
  17
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <math.h>
  21 #include <string.h>
  22 #include <errno.h>
  23
  24 #include <vorbis/codec.h>
  25 #include <vorbis/vorbisenc.h>
  26
  27 #include "write_read.h"
  28
  29 /* The following function is basically a hacked version of the code in
  30  * examples/encoder_example.c */
  31 void
  32 write_vorbis_data_or_die (const char *filename, int srate, const float * data, int count)
  33 {
  34   FILE * file ;
  35   ogg_stream_state os;
  36   ogg_page         og;
  37   ogg_packet       op;
  38   vorbis_info      vi;
  39   vorbis_comment   vc;
  40   vorbis_dsp_state vd;
  41   vorbis_block     vb;
  42
  43   int eos = 0, ret;
  44
  45   if ((file = fopen (filename, "wb")) == NULL) {
  46     printf("\n\nError : fopen failed : %s\n", strerror (errno)) ;
  47     exit (1) ;
  48   }
  49
  50   /********** Encode setup ************/
  51
  52   vorbis_info_init (&vi);
  53
  54   ret = vorbis_encode_init_vbr (&vi,1,srate,0.8);
  55   if (ret) {
  56     printf ("vorbis_encode_init_vbr return %d\n", ret) ;
  57     exit (1) ;
  58   }
  59
  60   vorbis_comment_init (&vc);
  61   vorbis_comment_add_tag (&vc,"ENCODER","test/util.c");
  62   vorbis_analysis_init (&vd,&vi);
  63   vorbis_block_init (&vd,&vb);
  64
  65   ogg_stream_init (&os,12345678);
  66
  67   {
  68     ogg_packet header;
  69     ogg_packet header_comm;
  70     ogg_packet header_code;
  71
  72     vorbis_analysis_headerout (&vd,&vc,&header,&header_comm,&header_code);
  73     ogg_stream_packetin (&os,&header);
  74     ogg_stream_packetin (&os,&header_comm);
  75     ogg_stream_packetin (&os,&header_code);
  76
  77     /* Ensures the audio data will start on a new page. */
  78     while (!eos){
  79         int result = ogg_stream_flush (&os,&og);
  80         if (result == 0)
  81             break;
  82         fwrite (og.header,1,og.header_len,file);
  83         fwrite (og.body,1,og.body_len,file);
  84     }
  85
  86   }
  87
  88   {
  89     /* expose the buffer to submit data */
  90     float **buffer = vorbis_analysis_buffer (&vd,count);
  91
  92     memcpy (buffer [0], data, count * sizeof (float)) ;
  93
  94     /* tell the library how much we actually submitted */
  95     vorbis_analysis_wrote (&vd,count);
  96     vorbis_analysis_wrote (&vd,0);
  97   }
  98
  99   while (vorbis_analysis_blockout (&vd,&vb) == 1) {
 100     vorbis_analysis (&vb,NULL);
 101     vorbis_bitrate_addblock (&vb);
 102
 103     while (vorbis_bitrate_flushpacket (&vd,&op)) {
 104       ogg_stream_packetin (&os,&op);
 105
 106       while (!eos) {
 107           int result = ogg_stream_pageout (&os,&og);
 108           if (result == 0)
 109               break;
 110           fwrite (og.header,1,og.header_len,file);
 111           fwrite (og.body,1,og.body_len,file);
 112
 113           if (ogg_page_eos (&og))
 114               eos = 1;
 115       }
 116     }
 117   }
 118
 119   ogg_stream_clear (&os);
 120   vorbis_block_clear (&vb);
 121   vorbis_dsp_clear (&vd);
 122   vorbis_comment_clear (&vc);
 123   vorbis_info_clear (&vi);
 124
 125  fclose (file) ;
 126 }
 127
 128 /* The following function is basically a hacked version of the code in
 129  * examples/decoder_example.c */
 130 void
 131 read_vorbis_data_or_die (const char *filename, int srate, float * data, int count)
 132 {
 133   ogg_sync_state   oy;
 134   ogg_stream_state os;
 135   ogg_page         og;
 136   ogg_packet       op;
 137
 138   vorbis_info      vi;
 139   vorbis_comment   vc;
 140   vorbis_dsp_state vd;
 141   vorbis_block     vb;
 142
 143   FILE *file;
 144   char *buffer;
 145   int  bytes;
 146   int eos = 0;
 147   int i;
 148   int read_total = 0 ;
 149
 150   if ((file = fopen (filename, "rb")) == NULL) {
 151     printf("\n\nError : fopen failed : %s\n", strerror (errno)) ;
 152     exit (1) ;
 153   }
 154
 155   ogg_sync_init (&oy);
 156
 157   {
 158     buffer = ogg_sync_buffer (&oy,4096);
 159     bytes = fread (buffer,1,4096,file);
 160     ogg_sync_wrote (&oy,bytes);
 161
 162     if(ogg_sync_pageout (&oy,&og) != 1) {
 163       if(bytes < 4096) {
 164         printf ("Out of data.\n") ;
 165           goto done_decode ;
 166       }
 167
 168       fprintf (stderr,"Input does not appear to be an Ogg bitstream.\n");
 169       exit (1);
 170     }
 171
 172     ogg_stream_init (&os,ogg_page_serialno(&og));
 173
 174     vorbis_info_init (&vi);
 175     vorbis_comment_init (&vc);
 176     if (ogg_stream_pagein (&os,&og) < 0) {
 177       fprintf (stderr,"Error reading first page of Ogg bitstream data.\n");
 178       exit (1);
 179     }
 180
 181     if (ogg_stream_packetout(&os,&op) != 1) {
 182       fprintf (stderr,"Error reading initial header packet.\n");
 183       exit (1);
 184     }
 185
 186     if (vorbis_synthesis_headerin (&vi,&vc,&op) < 0) {
 187       fprintf (stderr,"This Ogg bitstream does not contain Vorbis "
 188           "audio data.\n");
 189       exit (1);
 190     }
 191
 192     i = 0;
 193     while ( i < 2) {
 194       while (i < 2) {
 195         int result = ogg_sync_pageout (&oy,&og);
 196         if(result == 0)
 197           break;
 198         if(result==1) {
 199           ogg_stream_pagein(&os,&og);
 200
 201         while (i < 2) {
 202           result = ogg_stream_packetout (&os,&op);
 203           if (result == 0)
 204             goto done_decode;
 205           if (result < 0) {
 206             fprintf (stderr,"Corrupt secondary header.  Exiting.\n");
 207             exit(1);
 208           }
 209         vorbis_synthesis_headerin (&vi,&vc,&op);
 210         i++;
 211         }
 212       }
 213     }
 214
 215     buffer = ogg_sync_buffer (&oy,4096);
 216     bytes = fread (buffer,1,4096,file);
 217     if (bytes == 0 && i < 2) {
 218       fprintf (stderr,"End of file before finding all Vorbis headers!\n");
 219       exit (1);
 220     }
 221
 222     ogg_sync_wrote (&oy,bytes);
 223   }
 224
 225   if (vi.rate != srate) {
 226     printf ("\n\nError : File '%s' has sample rate of %ld when it should be %d.\n\n", filename, vi.rate, srate);
 227     exit (1) ;
 228   }
 229   if (vi.channels != 1) {
 230     printf ("\n\nError : File '%s' has %d channels, but should be mono.\n\n", filename, vi.channels);
 231     exit (1) ;
 232   }
 233
 234   vorbis_synthesis_init (&vd,&vi);
 235   vorbis_block_init (&vd,&vb);
 236
 237   while(!eos) {
 238     while (!eos) {
 239       int result = ogg_sync_pageout (&oy,&og);
 240       if (result == 0)
 241         break;
 242       if (result < 0) {
 243         fprintf (stderr,"Corrupt or missing data in bitstream; "
 244             "continuing...\n");
 245       } else {
 246         ogg_stream_pagein (&os,&og);
 247         while (1) {
 248           result = ogg_stream_packetout (&os,&op);
 249
 250           if (result == 0)
 251             break;
 252           if (result < 0) {
 253             /* no reason to complain; already complained above */
 254           } else {
 255               float **pcm;
 256             int samples;
 257
 258             if (vorbis_synthesis (&vb,&op) == 0)
 259               vorbis_synthesis_blockin(&vd,&vb);
 260             while ((samples = vorbis_synthesis_pcmout (&vd,&pcm)) > 0 && read_total < count) {
 261               int bout = samples < count ? samples : count;
 262               bout = read_total + bout > count ? count - read_total : bout;
 263
 264               memcpy (data + read_total, pcm[0], bout * sizeof (float)) ;
 265
 266               vorbis_synthesis_read (&vd,bout);
 267               read_total += bout ;
 268             }
 269           }
 270         }
 271
 272         if (ogg_page_eos (&og)) eos = 1;
 273       }
 274     }
 275
 276       if (!eos) {
 277         buffer = ogg_sync_buffer (&oy,4096);
 278         bytes = fread (buffer,1,4096,file);
 279         ogg_sync_wrote (&oy,bytes);
 280         if (bytes == 0) eos = 1;
 281       }
 282     }
 283
 284     ogg_stream_clear (&os);
 285
 286     vorbis_block_clear (&vb);
 287     vorbis_dsp_clear (&vd);
 288     vorbis_comment_clear (&vc);
 289     vorbis_info_clear (&vi);
 290
 291   }
 292 done_decode:
 293
 294   /* OK, clean up the framer */
 295   ogg_sync_clear (&oy);
 296
 297   fclose (file) ;
 298 }
 299