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, float q, const float * data, int count, int ch) 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,ch,srate,q); 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 int i; 92 93 for(i=0;i<ch;i++) 94 memcpy (buffer [i], data, count * sizeof (float)) ; 95 96 /* tell the library how much we actually submitted */ 97 vorbis_analysis_wrote (&vd,count); 98 vorbis_analysis_wrote (&vd,0); 99 } 100 101 while (vorbis_analysis_blockout (&vd,&vb) == 1) { 102 vorbis_analysis (&vb,NULL); 103 vorbis_bitrate_addblock (&vb); 104 105 while (vorbis_bitrate_flushpacket (&vd,&op)) { 106 ogg_stream_packetin (&os,&op); 107 108 while (!eos) { 109 int result = ogg_stream_pageout (&os,&og); 110 if (result == 0) 111 break; 112 fwrite (og.header,1,og.header_len,file); 113 fwrite (og.body,1,og.body_len,file); 114 115 if (ogg_page_eos (&og)) 116 eos = 1; 117 } 118 } 119 } 120 121 ogg_stream_clear (&os); 122 vorbis_block_clear (&vb); 123 vorbis_dsp_clear (&vd); 124 vorbis_comment_clear (&vc); 125 vorbis_info_clear (&vi); 126 127 fclose (file) ; 128 } 129 130 /* The following function is basically a hacked version of the code in 131 * examples/decoder_example.c */ 132 void 133 read_vorbis_data_or_die (const char *filename, int srate, float * data, int count) 134 { 135 ogg_sync_state oy; 136 ogg_stream_state os; 137 ogg_page og; 138 ogg_packet op; 139 140 vorbis_info vi; 141 vorbis_comment vc; 142 vorbis_dsp_state vd; 143 vorbis_block vb; 144 145 FILE *file; 146 char *buffer; 147 int bytes; 148 int eos = 0; 149 int i; 150 int read_total = 0 ; 151 152 if ((file = fopen (filename, "rb")) == NULL) { 153 printf("\n\nError : fopen failed : %s\n", strerror (errno)) ; 154 exit (1) ; 155 } 156 157 ogg_sync_init (&oy); 158 159 { 160 /* fragile! Assumes all of our headers will fit in the first 8kB, 161 which currently they will */ 162 buffer = ogg_sync_buffer (&oy,8192); 163 bytes = fread (buffer,1,8192,file); 164 ogg_sync_wrote (&oy,bytes); 165 166 if(ogg_sync_pageout (&oy,&og) != 1) { 167 if(bytes < 8192) { 168 printf ("Out of data.\n") ; 169 goto done_decode ; 170 } 171 172 fprintf (stderr,"Input does not appear to be an Ogg bitstream.\n"); 173 exit (1); 174 } 175 176 ogg_stream_init (&os,ogg_page_serialno(&og)); 177 178 vorbis_info_init (&vi); 179 vorbis_comment_init (&vc); 180 if (ogg_stream_pagein (&os,&og) < 0) { 181 fprintf (stderr,"Error reading first page of Ogg bitstream data.\n"); 182 exit (1); 183 } 184 185 if (ogg_stream_packetout(&os,&op) != 1) { 186 fprintf (stderr,"Error reading initial header packet.\n"); 187 exit (1); 188 } 189 190 if (vorbis_synthesis_headerin (&vi,&vc,&op) < 0) { 191 fprintf (stderr,"This Ogg bitstream does not contain Vorbis " 192 "audio data.\n"); 193 exit (1); 194 } 195 196 i = 0; 197 while ( i < 2) { 198 while (i < 2) { 199 200 int result = ogg_sync_pageout (&oy,&og); 201 if(result == 0) 202 break; 203 if(result==1) { 204 ogg_stream_pagein(&os,&og); 205 206 while (i < 2) { 207 result = ogg_stream_packetout (&os,&op); 208 if (result == 0) break; 209 if (result < 0) { 210 fprintf (stderr,"Corrupt secondary header. Exiting.\n"); 211 exit(1); 212 } 213 vorbis_synthesis_headerin (&vi,&vc,&op); 214 i++; 215 } 216 } 217 } 218 219 buffer = ogg_sync_buffer (&oy,4096); 220 bytes = fread (buffer,1,4096,file); 221 if (bytes == 0 && i < 2) { 222 fprintf (stderr,"End of file before finding all Vorbis headers!\n"); 223 exit (1); 224 } 225 226 ogg_sync_wrote (&oy,bytes); 227 } 228 229 if (vi.rate != srate) { 230 printf ("\n\nError : File '%s' has sample rate of %ld when it should be %d.\n\n", filename, vi.rate, srate); 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 done_decode: 292 293 /* OK, clean up the framer */ 294 ogg_sync_clear (&oy); 295 296 fclose (file) ; 297 } 298 299
