diff options
Diffstat (limited to 'contrib/vorbis/test/write_read.c')
| -rw-r--r-- | contrib/vorbis/test/write_read.c | 297 |
1 files changed, 297 insertions, 0 deletions
diff --git a/contrib/vorbis/test/write_read.c b/contrib/vorbis/test/write_read.c new file mode 100644 index 0000000..f1190dc --- /dev/null +++ b/contrib/vorbis/test/write_read.c @@ -0,0 +1,297 @@ +/******************************************************************** + * * + * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE. * + * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS * + * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE * + * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING. * + * * + * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2007 * + * by the Xiph.Org Foundation http://www.xiph.org/ * + * * + ******************************************************************** + + function: utility functions for vorbis codec test suite. + + ********************************************************************/ + +#include <stdio.h> +#include <stdlib.h> +#include <math.h> +#include <string.h> +#include <errno.h> + +#include <vorbis/codec.h> +#include <vorbis/vorbisenc.h> + +#include "write_read.h" + +/* The following function is basically a hacked version of the code in + * examples/encoder_example.c */ +void +write_vorbis_data_or_die (const char *filename, int srate, float q, const float * data, int count, int ch) +{ + FILE * file ; + ogg_stream_state os; + ogg_page og; + ogg_packet op; + vorbis_info vi; + vorbis_comment vc; + vorbis_dsp_state vd; + vorbis_block vb; + + int eos = 0, ret; + + if ((file = fopen (filename, "wb")) == NULL) { + printf("\n\nError : fopen failed : %s\n", strerror (errno)) ; + exit (1) ; + } + + /********** Encode setup ************/ + + vorbis_info_init (&vi); + + ret = vorbis_encode_init_vbr (&vi,ch,srate,q); + if (ret) { + printf ("vorbis_encode_init_vbr return %d\n", ret) ; + exit (1) ; + } + + vorbis_comment_init (&vc); + vorbis_comment_add_tag (&vc,"ENCODER","test/util.c"); + vorbis_analysis_init (&vd,&vi); + vorbis_block_init (&vd,&vb); + + ogg_stream_init (&os,12345678); + + { + ogg_packet header; + ogg_packet header_comm; + ogg_packet header_code; + + vorbis_analysis_headerout (&vd,&vc,&header,&header_comm,&header_code); + ogg_stream_packetin (&os,&header); + ogg_stream_packetin (&os,&header_comm); + ogg_stream_packetin (&os,&header_code); + + /* Ensures the audio data will start on a new page. */ + while (!eos){ + int result = ogg_stream_flush (&os,&og); + if (result == 0) + break; + fwrite (og.header,1,og.header_len,file); + fwrite (og.body,1,og.body_len,file); + } + + } + + { + /* expose the buffer to submit data */ + float **buffer = vorbis_analysis_buffer (&vd,count); + int i; + + for(i=0;i<ch;i++) + memcpy (buffer [i], data, count * sizeof (float)) ; + + /* tell the library how much we actually submitted */ + vorbis_analysis_wrote (&vd,count); + vorbis_analysis_wrote (&vd,0); + } + + while (vorbis_analysis_blockout (&vd,&vb) == 1) { + vorbis_analysis (&vb,NULL); + vorbis_bitrate_addblock (&vb); + + while (vorbis_bitrate_flushpacket (&vd,&op)) { + ogg_stream_packetin (&os,&op); + + while (!eos) { + int result = ogg_stream_pageout (&os,&og); + if (result == 0) + break; + fwrite (og.header,1,og.header_len,file); + fwrite (og.body,1,og.body_len,file); + + if (ogg_page_eos (&og)) + eos = 1; + } + } + } + + ogg_stream_clear (&os); + vorbis_block_clear (&vb); + vorbis_dsp_clear (&vd); + vorbis_comment_clear (&vc); + vorbis_info_clear (&vi); + + fclose (file) ; +} + +/* The following function is basically a hacked version of the code in + * examples/decoder_example.c */ +void +read_vorbis_data_or_die (const char *filename, int srate, float * data, int count) +{ + ogg_sync_state oy; + ogg_stream_state os; + ogg_page og; + ogg_packet op; + + vorbis_info vi; + vorbis_comment vc; + vorbis_dsp_state vd; + vorbis_block vb; + + FILE *file; + char *buffer; + int bytes; + int eos = 0; + int i; + int read_total = 0 ; + + if ((file = fopen (filename, "rb")) == NULL) { + printf("\n\nError : fopen failed : %s\n", strerror (errno)) ; + exit (1) ; + } + + ogg_sync_init (&oy); + + { + /* fragile! Assumes all of our headers will fit in the first 8kB, + which currently they will */ + buffer = ogg_sync_buffer (&oy,8192); + bytes = fread (buffer,1,8192,file); + ogg_sync_wrote (&oy,bytes); + + if(ogg_sync_pageout (&oy,&og) != 1) { + if(bytes < 8192) { + printf ("Out of data.\n") ; + goto done_decode ; + } + + fprintf (stderr,"Input does not appear to be an Ogg bitstream.\n"); + exit (1); + } + + ogg_stream_init (&os,ogg_page_serialno(&og)); + + vorbis_info_init (&vi); + vorbis_comment_init (&vc); + if (ogg_stream_pagein (&os,&og) < 0) { + fprintf (stderr,"Error reading first page of Ogg bitstream data.\n"); + exit (1); + } + + if (ogg_stream_packetout(&os,&op) != 1) { + fprintf (stderr,"Error reading initial header packet.\n"); + exit (1); + } + + if (vorbis_synthesis_headerin (&vi,&vc,&op) < 0) { + fprintf (stderr,"This Ogg bitstream does not contain Vorbis " + "audio data.\n"); + exit (1); + } + + i = 0; + while ( i < 2) { + while (i < 2) { + + int result = ogg_sync_pageout (&oy,&og); + if(result == 0) + break; + if(result==1) { + ogg_stream_pagein(&os,&og); + + while (i < 2) { + result = ogg_stream_packetout (&os,&op); + if (result == 0) break; + if (result < 0) { + fprintf (stderr,"Corrupt secondary header. Exiting.\n"); + exit(1); + } + vorbis_synthesis_headerin (&vi,&vc,&op); + i++; + } + } + } + + buffer = ogg_sync_buffer (&oy,4096); + bytes = fread (buffer,1,4096,file); + if (bytes == 0 && i < 2) { + fprintf (stderr,"End of file before finding all Vorbis headers!\n"); + exit (1); + } + + ogg_sync_wrote (&oy,bytes); + } + + if (vi.rate != srate) { + printf ("\n\nError : File '%s' has sample rate of %ld when it should be %d.\n\n", filename, vi.rate, srate); + exit (1) ; + } + + vorbis_synthesis_init (&vd,&vi); + vorbis_block_init (&vd,&vb); + + while(!eos) { + while (!eos) { + int result = ogg_sync_pageout (&oy,&og); + if (result == 0) + break; + if (result < 0) { + fprintf (stderr,"Corrupt or missing data in bitstream; " + "continuing...\n"); + } else { + ogg_stream_pagein (&os,&og); + while (1) { + result = ogg_stream_packetout (&os,&op); + + if (result == 0) + break; + if (result < 0) { + /* no reason to complain; already complained above */ + } else { + float **pcm; + int samples; + + if (vorbis_synthesis (&vb,&op) == 0) + vorbis_synthesis_blockin(&vd,&vb); + while ((samples = vorbis_synthesis_pcmout (&vd,&pcm)) > 0 && read_total < count) { + int bout = samples < count ? samples : count; + bout = read_total + bout > count ? count - read_total : bout; + + memcpy (data + read_total, pcm[0], bout * sizeof (float)) ; + + vorbis_synthesis_read (&vd,bout); + read_total += bout ; + } + } + } + + if (ogg_page_eos (&og)) eos = 1; + } + } + + if (!eos) { + buffer = ogg_sync_buffer (&oy,4096); + bytes = fread (buffer,1,4096,file); + ogg_sync_wrote (&oy,bytes); + if (bytes == 0) eos = 1; + } + } + + ogg_stream_clear (&os); + + vorbis_block_clear (&vb); + vorbis_dsp_clear (&vd); + vorbis_comment_clear (&vc); + vorbis_info_clear (&vi); + } +done_decode: + + /* OK, clean up the framer */ + ogg_sync_clear (&oy); + + fclose (file) ; +} + |