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,
59 Asset *asset, BC_WindowBase* &format_window,
60 int audio_options, int video_options, EDL *edl)
64 VorbisConfigAudio *window = new VorbisConfigAudio(parent_window, asset);
65 format_window = window;
66 window->create_objects();
72 int FileVorbis::check_sig(Asset *asset)
74 FILE *fd = fopen(asset->path, "rb");
77 // Test for Quicktime since OGG misinterprets it
80 fseek(fd, 4, SEEK_SET);
82 (void)fread(data, 4, 1, fd);
92 fseek(fd, 0, SEEK_SET);
94 if(ov_open(fd, &vf, NULL, 0) < 0)
96 // OGG failed. Close file handle manually.
111 int FileVorbis::reset_parameters_derived()
114 bzero(&vf, sizeof(vf));
119 int FileVorbis::open_file(int rd, int wr)
124 //printf("FileVorbis::open_file 1\n");
127 //printf("FileVorbis::open_file 1\n");
128 if(!(fd = fopen(asset->path, "rb")))
130 eprintf("Error while opening \"%s\" for reading. \n%m\n", asset->path);
135 //printf("FileVorbis::open_file 2 %p %p\n", fd, vf);
136 if(ov_open(fd, &vf, NULL, 0) < 0)
138 eprintf(_("FileVorbis::open_file %s: invalid bitstream.\n"), asset->path);
143 //printf("FileVorbis::open_file 1\n");
144 vorbis_info *vi = ov_info(&vf, -1);
145 asset->channels = vi->channels;
146 if(!asset->sample_rate)
147 asset->sample_rate = vi->rate;
148 //printf("FileVorbis::open_file 1\n");
149 asset->audio_length = ov_pcm_total(&vf,-1);
150 //printf("FileVorbis::open_file 1\n");
151 asset->audio_data = 1;
152 // printf("FileVorbis::open_file 1 %d %d %d\n",
154 // asset->sample_rate,
155 // asset->audio_length);
162 if(!(fd = fopen(asset->path, "wb")))
164 eprintf(_("Error while opening \"%s\" for writing. \n%m\n"), asset->path);
169 vorbis_info_init(&vi);
170 if(!asset->vorbis_vbr)
171 result = vorbis_encode_init(&vi,
174 asset->vorbis_max_bitrate,
175 asset->vorbis_bitrate,
176 asset->vorbis_min_bitrate);
179 result = vorbis_encode_setup_managed(&vi,
182 asset->vorbis_max_bitrate,
183 asset->vorbis_bitrate,
184 asset->vorbis_min_bitrate);
185 result |= vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_AVG, NULL);
186 result |= vorbis_encode_setup_init(&vi);
191 vorbis_analysis_init(&vd, &vi);
192 vorbis_block_init(&vd, &vb);
193 vorbis_comment_init(&vc);
195 ogg_stream_init(&os, rand());
198 ogg_packet header_comm;
199 ogg_packet header_code;
200 vorbis_analysis_headerout(&vd,
205 ogg_stream_packetin(&os,
207 ogg_stream_packetin(&os,
209 ogg_stream_packetin(&os,
214 int result = ogg_stream_flush(&os, &og);
215 if(result == 0) break;
216 fwrite(og.header, 1, og.header_len, fd);
217 fwrite(og.body, 1, og.body_len, fd);
223 //printf("FileVorbis::open_file 2\n");
227 #define FLUSH_VORBIS \
228 while(vorbis_analysis_blockout(&vd, &vb) == 1) \
230 vorbis_analysis(&vb, NULL); \
231 vorbis_bitrate_addblock(&vb); \
232 while(vorbis_bitrate_flushpacket(&vd, &op)) \
234 ogg_stream_packetin(&os, &op); \
237 int result = ogg_stream_pageout(&os, &og); \
239 fwrite(og.header, 1, og.header_len, fd); \
240 fwrite(og.body, 1, og.body_len, fd); \
241 if(ogg_page_eos(&og)) break; \
247 int FileVorbis::close_file_derived()
253 vorbis_analysis_wrote(&vd, 0);
256 ogg_stream_clear(&os);
257 vorbis_block_clear(&vb);
258 vorbis_dsp_clear(&vd);
259 vorbis_comment_clear(&vc);
260 vorbis_info_clear(&vi);
266 // This also closes the file handle.
276 int FileVorbis::write_samples(double **buffer, int64_t len)
280 float **vorbis_buffer = vorbis_analysis_buffer(&vd, len);
281 for(int i = 0; i < asset->channels; i++)
283 float *output = vorbis_buffer[i];
284 double *input = buffer[i];
285 for(int j = 0; j < len; j++)
287 output[j] = input[j];
291 vorbis_analysis_wrote(&vd, len);
297 int FileVorbis::read_samples(double *buffer, int64_t len)
301 // printf("FileVorbis::read_samples 1 %d %d %d %d\n",
304 // file->current_sample,
306 float **vorbis_output;
308 int accumulation = 0;
311 update_pcm_history(len);
314 // Fill history buffer
315 if(decode_start != decode_end)
317 ov_pcm_seek(&vf, decode_start);
318 decode_end = decode_start;
321 while(accumulation < decode_len)
323 int result = ov_read_float(&vf,
325 decode_len - accumulation,
327 //printf("FileVorbis::read_samples 1 %d %d %d\n", result, len, accumulation);
330 append_history(vorbis_output, result);
331 accumulation += result;
336 file->current_sample,
337 file->current_channel,
340 // printf("FileVorbis::read_samples 2 %d %d %d %d\n",
343 // file->current_sample,
358 VorbisConfigAudio::VorbisConfigAudio(BC_WindowBase *parent_window,
360 : BC_Window(_(PROGRAM_NAME ": Audio Compression"),
361 parent_window->get_abs_cursor_x(1),
362 parent_window->get_abs_cursor_y(1),
371 this->parent_window = parent_window;
375 VorbisConfigAudio::~VorbisConfigAudio()
379 void VorbisConfigAudio::create_objects()
383 char string[BCTEXTLEN];
385 lock_window("VorbisConfigAudio::create_objects");
386 add_tool(fixed_bitrate = new VorbisFixedBitrate(x, y, this));
387 add_tool(variable_bitrate = new VorbisVariableBitrate(x1, y, this));
390 sprintf(string, "%d", asset->vorbis_min_bitrate);
391 add_tool(new BC_Title(x, y, _("Min bitrate:")));
392 add_tool(new VorbisMinBitrate(x1, y, this, string));
395 add_tool(new BC_Title(x, y, _("Avg bitrate:")));
396 sprintf(string, "%d", asset->vorbis_bitrate);
397 add_tool(new VorbisAvgBitrate(x1, y, this, string));
400 add_tool(new BC_Title(x, y, _("Max bitrate:")));
401 sprintf(string, "%d", asset->vorbis_max_bitrate);
402 add_tool(new VorbisMaxBitrate(x1, y, this, string));
405 add_subwindow(new BC_OKButton(this));
411 int VorbisConfigAudio::close_event()
421 VorbisFixedBitrate::VorbisFixedBitrate(int x, int y, VorbisConfigAudio *gui)
422 : BC_Radial(x, y, !gui->asset->vorbis_vbr, _("Fixed bitrate"))
426 int VorbisFixedBitrate::handle_event()
428 gui->asset->vorbis_vbr = 0;
429 gui->variable_bitrate->update(0);
433 VorbisVariableBitrate::VorbisVariableBitrate(int x, int y, VorbisConfigAudio *gui)
434 : BC_Radial(x, y, gui->asset->vorbis_vbr, _("Variable bitrate"))
438 int VorbisVariableBitrate::handle_event()
440 gui->asset->vorbis_vbr = 1;
441 gui->fixed_bitrate->update(0);
446 VorbisMinBitrate::VorbisMinBitrate(int x,
448 VorbisConfigAudio *gui,
450 : BC_TextBox(x, y, 180, 1, text)
454 int VorbisMinBitrate::handle_event()
456 gui->asset->vorbis_min_bitrate = atol(get_text());
462 VorbisMaxBitrate::VorbisMaxBitrate(int x,
464 VorbisConfigAudio *gui,
466 : BC_TextBox(x, y, 180, 1, text)
470 int VorbisMaxBitrate::handle_event()
472 gui->asset->vorbis_max_bitrate = atol(get_text());
478 VorbisAvgBitrate::VorbisAvgBitrate(int x, int y, VorbisConfigAudio *gui, char *text)
479 : BC_TextBox(x, y, 180, 1, text)
483 int VorbisAvgBitrate::handle_event()
485 gui->asset->vorbis_bitrate = atol(get_text());