4 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 #include "bcsignals.h"
24 #include "byteorder.h"
27 #include "filevorbis.h"
30 #include "mainerror.h"
31 #include "mwindow.inc"
32 #include "mainerror.h"
39 //suppress noref warning
40 void *vorbis1_ov_callbacks[] = {
41 &OV_CALLBACKS_DEFAULT, &OV_CALLBACKS_NOCLOSE,
42 &OV_CALLBACKS_STREAMONLY, &OV_CALLBACKS_STREAMONLY_NOCLOSE,
45 FileVorbis::FileVorbis(Asset *asset, File *file)
46 : FileBase(asset, file)
49 if(asset->format == FILE_UNKNOWN) asset->format = FILE_VORBIS;
50 asset->byte_order = 0;
53 FileVorbis::~FileVorbis()
58 void FileVorbis::get_parameters(BC_WindowBase *parent_window,
60 BC_WindowBase* &format_window,
66 VorbisConfigAudio *window = new VorbisConfigAudio(parent_window, asset);
67 format_window = window;
68 window->create_objects();
74 int FileVorbis::check_sig(Asset *asset)
76 FILE *fd = fopen(asset->path, "rb");
79 // Test for Quicktime since OGG misinterprets it
82 fseek(fd, 4, SEEK_SET);
84 (void)fread(data, 4, 1, fd);
94 fseek(fd, 0, SEEK_SET);
96 if(ov_open(fd, &vf, NULL, 0) < 0)
98 // OGG failed. Close file handle manually.
113 int FileVorbis::reset_parameters_derived()
116 bzero(&vf, sizeof(vf));
121 int FileVorbis::open_file(int rd, int wr)
126 //printf("FileVorbis::open_file 1\n");
129 //printf("FileVorbis::open_file 1\n");
130 if(!(fd = fopen(asset->path, "rb")))
132 eprintf("Error while opening \"%s\" for reading. \n%m\n", asset->path);
137 //printf("FileVorbis::open_file 2 %p %p\n", fd, vf);
138 if(ov_open(fd, &vf, NULL, 0) < 0)
140 eprintf(_("FileVorbis::open_file %s: invalid bitstream.\n"), asset->path);
145 //printf("FileVorbis::open_file 1\n");
146 vorbis_info *vi = ov_info(&vf, -1);
147 asset->channels = vi->channels;
148 if(!asset->sample_rate)
149 asset->sample_rate = vi->rate;
150 //printf("FileVorbis::open_file 1\n");
151 asset->audio_length = ov_pcm_total(&vf,-1);
152 //printf("FileVorbis::open_file 1\n");
153 asset->audio_data = 1;
154 // printf("FileVorbis::open_file 1 %d %d %d\n",
156 // asset->sample_rate,
157 // asset->audio_length);
164 if(!(fd = fopen(asset->path, "wb")))
166 eprintf("Error while opening \"%s\" for writing. \n%m\n", asset->path);
171 vorbis_info_init(&vi);
172 if(!asset->vorbis_vbr)
173 result = vorbis_encode_init(&vi,
176 asset->vorbis_max_bitrate,
177 asset->vorbis_bitrate,
178 asset->vorbis_min_bitrate);
181 result = vorbis_encode_setup_managed(&vi,
184 asset->vorbis_max_bitrate,
185 asset->vorbis_bitrate,
186 asset->vorbis_min_bitrate);
187 result |= vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_AVG, NULL);
188 result |= vorbis_encode_setup_init(&vi);
193 vorbis_analysis_init(&vd, &vi);
194 vorbis_block_init(&vd, &vb);
195 vorbis_comment_init(&vc);
197 ogg_stream_init(&os, rand());
200 ogg_packet header_comm;
201 ogg_packet header_code;
202 vorbis_analysis_headerout(&vd,
207 ogg_stream_packetin(&os,
209 ogg_stream_packetin(&os,
211 ogg_stream_packetin(&os,
216 int result = ogg_stream_flush(&os, &og);
217 if(result == 0) break;
218 fwrite(og.header, 1, og.header_len, fd);
219 fwrite(og.body, 1, og.body_len, fd);
225 //printf("FileVorbis::open_file 2\n");
229 #define FLUSH_VORBIS \
230 while(vorbis_analysis_blockout(&vd, &vb) == 1) \
232 vorbis_analysis(&vb, NULL); \
233 vorbis_bitrate_addblock(&vb); \
234 while(vorbis_bitrate_flushpacket(&vd, &op)) \
236 ogg_stream_packetin(&os, &op); \
239 int result = ogg_stream_pageout(&os, &og); \
241 fwrite(og.header, 1, og.header_len, fd); \
242 fwrite(og.body, 1, og.body_len, fd); \
243 if(ogg_page_eos(&og)) break; \
249 int FileVorbis::close_file_derived()
255 vorbis_analysis_wrote(&vd, 0);
258 ogg_stream_clear(&os);
259 vorbis_block_clear(&vb);
260 vorbis_dsp_clear(&vd);
261 vorbis_comment_clear(&vc);
262 vorbis_info_clear(&vi);
268 // This also closes the file handle.
278 int FileVorbis::write_samples(double **buffer, int64_t len)
282 float **vorbis_buffer = vorbis_analysis_buffer(&vd, len);
283 for(int i = 0; i < asset->channels; i++)
285 float *output = vorbis_buffer[i];
286 double *input = buffer[i];
287 for(int j = 0; j < len; j++)
289 output[j] = input[j];
293 vorbis_analysis_wrote(&vd, len);
299 int FileVorbis::read_samples(double *buffer, int64_t len)
303 // printf("FileVorbis::read_samples 1 %d %d %d %d\n",
306 // file->current_sample,
308 float **vorbis_output;
310 int accumulation = 0;
313 update_pcm_history(len);
316 // Fill history buffer
317 if(decode_start != decode_end)
319 ov_pcm_seek(&vf, decode_start);
320 decode_end = decode_start;
323 while(accumulation < decode_len)
325 int result = ov_read_float(&vf,
327 decode_len - accumulation,
329 //printf("FileVorbis::read_samples 1 %d %d %d\n", result, len, accumulation);
332 append_history(vorbis_output, result);
333 accumulation += result;
338 file->current_sample,
339 file->current_channel,
342 // printf("FileVorbis::read_samples 2 %d %d %d %d\n",
345 // file->current_sample,
360 VorbisConfigAudio::VorbisConfigAudio(BC_WindowBase *parent_window,
362 : BC_Window(_(PROGRAM_NAME ": Audio Compression"),
363 parent_window->get_abs_cursor_x(1),
364 parent_window->get_abs_cursor_y(1),
373 this->parent_window = parent_window;
377 VorbisConfigAudio::~VorbisConfigAudio()
381 void VorbisConfigAudio::create_objects()
385 char string[BCTEXTLEN];
387 lock_window("VorbisConfigAudio::create_objects");
388 add_tool(fixed_bitrate = new VorbisFixedBitrate(x, y, this));
389 add_tool(variable_bitrate = new VorbisVariableBitrate(x1, y, this));
392 sprintf(string, "%d", asset->vorbis_min_bitrate);
393 add_tool(new BC_Title(x, y, _("Min bitrate:")));
394 add_tool(new VorbisMinBitrate(x1, y, this, string));
397 add_tool(new BC_Title(x, y, _("Avg bitrate:")));
398 sprintf(string, "%d", asset->vorbis_bitrate);
399 add_tool(new VorbisAvgBitrate(x1, y, this, string));
402 add_tool(new BC_Title(x, y, _("Max bitrate:")));
403 sprintf(string, "%d", asset->vorbis_max_bitrate);
404 add_tool(new VorbisMaxBitrate(x1, y, this, string));
407 add_subwindow(new BC_OKButton(this));
413 int VorbisConfigAudio::close_event()
423 VorbisFixedBitrate::VorbisFixedBitrate(int x, int y, VorbisConfigAudio *gui)
424 : BC_Radial(x, y, !gui->asset->vorbis_vbr, _("Fixed bitrate"))
428 int VorbisFixedBitrate::handle_event()
430 gui->asset->vorbis_vbr = 0;
431 gui->variable_bitrate->update(0);
435 VorbisVariableBitrate::VorbisVariableBitrate(int x, int y, VorbisConfigAudio *gui)
436 : BC_Radial(x, y, gui->asset->vorbis_vbr, _("Variable bitrate"))
440 int VorbisVariableBitrate::handle_event()
442 gui->asset->vorbis_vbr = 1;
443 gui->fixed_bitrate->update(0);
448 VorbisMinBitrate::VorbisMinBitrate(int x,
450 VorbisConfigAudio *gui,
452 : BC_TextBox(x, y, 180, 1, text)
456 int VorbisMinBitrate::handle_event()
458 gui->asset->vorbis_min_bitrate = atol(get_text());
464 VorbisMaxBitrate::VorbisMaxBitrate(int x,
466 VorbisConfigAudio *gui,
468 : BC_TextBox(x, y, 180, 1, text)
472 int VorbisMaxBitrate::handle_event()
474 gui->asset->vorbis_max_bitrate = atol(get_text());
480 VorbisAvgBitrate::VorbisAvgBitrate(int x, int y, VorbisConfigAudio *gui, char *text)
481 : BC_TextBox(x, y, 180, 1, text)
485 int VorbisAvgBitrate::handle_event()
487 gui->asset->vorbis_bitrate = atol(get_text());