1 /********************************************************************
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. *
8 * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2007 *
9 * by the Xiph.Org Foundation http://www.xiph.org/ *
11 ********************************************************************
13 function: utility functions for vorbis codec test suite.
14 last mod: $Id: util.c 13293 2007-07-24 00:09:47Z erikd $
16 ********************************************************************/
24 #include <vorbis/codec.h>
25 #include <vorbis/vorbisenc.h>
27 #include "write_read.h"
29 /* The following function is basically a hacked version of the code in
30 * examples/encoder_example.c */
32 write_vorbis_data_or_die (const char *filename, int srate, float q, const float * data, int count)
45 if ((file = fopen (filename, "wb")) == NULL) {
46 printf("\n\nError : fopen failed : %s\n", strerror (errno)) ;
50 /********** Encode setup ************/
52 vorbis_info_init (&vi);
54 ret = vorbis_encode_init_vbr (&vi,1,srate,q);
56 printf ("vorbis_encode_init_vbr return %d\n", ret) ;
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);
65 ogg_stream_init (&os,12345678);
69 ogg_packet header_comm;
70 ogg_packet header_code;
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);
77 /* Ensures the audio data will start on a new page. */
79 int result = ogg_stream_flush (&os,&og);
82 fwrite (og.header,1,og.header_len,file);
83 fwrite (og.body,1,og.body_len,file);
89 /* expose the buffer to submit data */
90 float **buffer = vorbis_analysis_buffer (&vd,count);
92 memcpy (buffer [0], data, count * sizeof (float)) ;
94 /* tell the library how much we actually submitted */
95 vorbis_analysis_wrote (&vd,count);
96 vorbis_analysis_wrote (&vd,0);
99 while (vorbis_analysis_blockout (&vd,&vb) == 1) {
100 vorbis_analysis (&vb,NULL);
101 vorbis_bitrate_addblock (&vb);
103 while (vorbis_bitrate_flushpacket (&vd,&op)) {
104 ogg_stream_packetin (&os,&op);
107 int result = ogg_stream_pageout (&os,&og);
110 fwrite (og.header,1,og.header_len,file);
111 fwrite (og.body,1,og.body_len,file);
113 if (ogg_page_eos (&og))
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);
128 /* The following function is basically a hacked version of the code in
129 * examples/decoder_example.c */
131 read_vorbis_data_or_die (const char *filename, int srate, float * data, int count)
150 if ((file = fopen (filename, "rb")) == NULL) {
151 printf("\n\nError : fopen failed : %s\n", strerror (errno)) ;
158 buffer = ogg_sync_buffer (&oy,4096);
159 bytes = fread (buffer,1,4096,file);
160 ogg_sync_wrote (&oy,bytes);
162 if(ogg_sync_pageout (&oy,&og) != 1) {
164 printf ("Out of data.\n") ;
168 fprintf (stderr,"Input does not appear to be an Ogg bitstream.\n");
172 ogg_stream_init (&os,ogg_page_serialno(&og));
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");
181 if (ogg_stream_packetout(&os,&op) != 1) {
182 fprintf (stderr,"Error reading initial header packet.\n");
186 if (vorbis_synthesis_headerin (&vi,&vc,&op) < 0) {
187 fprintf (stderr,"This Ogg bitstream does not contain Vorbis "
195 int result = ogg_sync_pageout (&oy,&og);
199 ogg_stream_pagein(&os,&og);
202 result = ogg_stream_packetout (&os,&op);
206 fprintf (stderr,"Corrupt secondary header. Exiting.\n");
209 vorbis_synthesis_headerin (&vi,&vc,&op);
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");
222 ogg_sync_wrote (&oy,bytes);
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);
229 if (vi.channels != 1) {
230 printf ("\n\nError : File '%s' has %d channels, but should be mono.\n\n", filename, vi.channels);
234 vorbis_synthesis_init (&vd,&vi);
235 vorbis_block_init (&vd,&vb);
239 int result = ogg_sync_pageout (&oy,&og);
243 fprintf (stderr,"Corrupt or missing data in bitstream; "
246 ogg_stream_pagein (&os,&og);
248 result = ogg_stream_packetout (&os,&op);
253 /* no reason to complain; already complained above */
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;
264 memcpy (data + read_total, pcm[0], bout * sizeof (float)) ;
266 vorbis_synthesis_read (&vd,bout);
272 if (ogg_page_eos (&og)) eos = 1;
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;
284 ogg_stream_clear (&os);
286 vorbis_block_clear (&vb);
287 vorbis_dsp_clear (&vd);
288 vorbis_comment_clear (&vc);
289 vorbis_info_clear (&vi);
294 /* OK, clean up the framer */
295 ogg_sync_clear (&oy);