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
24 #include "bcsignals.h"
25 #include "byteorder.h"
28 #include "filevorbis.h"
31 #include "mainerror.h"
32 #include "mwindow.inc"
33 #include "mainerror.h"
40 //suppress noref warning
41 void *vorbis1_ov_callbacks[] = {
42 &OV_CALLBACKS_DEFAULT, &OV_CALLBACKS_NOCLOSE,
43 &OV_CALLBACKS_STREAMONLY, &OV_CALLBACKS_STREAMONLY_NOCLOSE,
46 FileVorbis::FileVorbis(Asset *asset, File *file)
47 : FileBase(asset, file)
50 if(asset->format == FILE_UNKNOWN) asset->format = FILE_VORBIS;
51 asset->byte_order = 0;
54 FileVorbis::~FileVorbis()
59 void FileVorbis::get_parameters(BC_WindowBase *parent_window,
60 Asset *asset, BC_WindowBase* &format_window,
61 int audio_options, int video_options, EDL *edl)
65 VorbisConfigAudio *window = new VorbisConfigAudio(parent_window, asset);
66 format_window = window;
67 window->create_objects();
73 int FileVorbis::check_sig(Asset *asset)
75 FILE *fd = fopen(asset->path, "rb");
78 // Test for Quicktime since OGG misinterprets it
81 fseek(fd, 4, SEEK_SET);
83 (void)fread(data, 4, 1, fd);
93 fseek(fd, 0, SEEK_SET);
95 if(ov_open(fd, &vf, NULL, 0) < 0)
97 // OGG failed. Close file handle manually.
112 int FileVorbis::reset_parameters_derived()
115 bzero(&vf, sizeof(vf));
120 int FileVorbis::open_file(int rd, int wr)
125 //printf("FileVorbis::open_file 1\n");
128 //printf("FileVorbis::open_file 1\n");
129 if(!(fd = fopen(asset->path, "rb")))
131 eprintf("Error while opening \"%s\" for reading. \n%m\n", asset->path);
136 //printf("FileVorbis::open_file 2 %p %p\n", fd, vf);
137 if(ov_open(fd, &vf, NULL, 0) < 0)
139 eprintf(_("FileVorbis::open_file %s: invalid bitstream.\n"), asset->path);
144 //printf("FileVorbis::open_file 1\n");
145 vorbis_info *vi = ov_info(&vf, -1);
146 asset->channels = vi->channels;
147 if(!asset->sample_rate)
148 asset->sample_rate = vi->rate;
149 //printf("FileVorbis::open_file 1\n");
150 asset->audio_length = ov_pcm_total(&vf,-1);
151 //printf("FileVorbis::open_file 1\n");
152 asset->audio_data = 1;
153 // printf("FileVorbis::open_file 1 %d %d %d\n",
155 // asset->sample_rate,
156 // asset->audio_length);
163 if(!(fd = fopen(asset->path, "wb")))
165 eprintf(_("Error while opening \"%s\" for writing. \n%m\n"), asset->path);
170 vorbis_info_init(&vi);
171 if(!asset->vorbis_vbr)
172 result = vorbis_encode_init(&vi,
175 asset->vorbis_max_bitrate,
176 asset->vorbis_bitrate,
177 asset->vorbis_min_bitrate);
180 result = vorbis_encode_setup_managed(&vi,
183 asset->vorbis_max_bitrate,
184 asset->vorbis_bitrate,
185 asset->vorbis_min_bitrate);
186 result |= vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_AVG, NULL);
187 result |= vorbis_encode_setup_init(&vi);
192 vorbis_analysis_init(&vd, &vi);
193 vorbis_block_init(&vd, &vb);
194 vorbis_comment_init(&vc);
196 ogg_stream_init(&os, rand());
199 ogg_packet header_comm;
200 ogg_packet header_code;
201 vorbis_analysis_headerout(&vd,
206 ogg_stream_packetin(&os,
208 ogg_stream_packetin(&os,
210 ogg_stream_packetin(&os,
215 int result = ogg_stream_flush(&os, &og);
216 if(result == 0) break;
217 fwrite(og.header, 1, og.header_len, fd);
218 fwrite(og.body, 1, og.body_len, fd);
224 //printf("FileVorbis::open_file 2\n");
228 #define FLUSH_VORBIS \
229 while(vorbis_analysis_blockout(&vd, &vb) == 1) \
231 vorbis_analysis(&vb, NULL); \
232 vorbis_bitrate_addblock(&vb); \
233 while(vorbis_bitrate_flushpacket(&vd, &op)) \
235 ogg_stream_packetin(&os, &op); \
238 int result = ogg_stream_pageout(&os, &og); \
240 fwrite(og.header, 1, og.header_len, fd); \
241 fwrite(og.body, 1, og.body_len, fd); \
242 if(ogg_page_eos(&og)) break; \
248 int FileVorbis::close_file_derived()
254 vorbis_analysis_wrote(&vd, 0);
257 ogg_stream_clear(&os);
258 vorbis_block_clear(&vb);
259 vorbis_dsp_clear(&vd);
260 vorbis_comment_clear(&vc);
261 vorbis_info_clear(&vi);
267 // This also closes the file handle.
277 int FileVorbis::write_samples(double **buffer, int64_t len)
281 float **vorbis_buffer = vorbis_analysis_buffer(&vd, len);
282 for(int i = 0; i < asset->channels; i++)
284 float *output = vorbis_buffer[i];
285 double *input = buffer[i];
286 for(int j = 0; j < len; j++)
288 output[j] = input[j];
292 vorbis_analysis_wrote(&vd, len);
298 int FileVorbis::read_samples(double *buffer, int64_t len)
302 // printf("FileVorbis::read_samples 1 %d %d %d %d\n",
305 // file->current_sample,
307 float **vorbis_output;
309 int accumulation = 0;
312 update_pcm_history(len);
315 // Fill history buffer
316 if(decode_start != decode_end)
318 ov_pcm_seek(&vf, decode_start);
319 decode_end = decode_start;
322 while(accumulation < decode_len)
324 int result = ov_read_float(&vf,
326 decode_len - accumulation,
328 //printf("FileVorbis::read_samples 1 %d %d %d\n", result, len, accumulation);
331 append_history(vorbis_output, result);
332 accumulation += result;
337 file->current_sample,
338 file->current_channel,
341 // printf("FileVorbis::read_samples 2 %d %d %d %d\n",
344 // file->current_sample,
359 VorbisConfigAudio::VorbisConfigAudio(BC_WindowBase *parent_window,
361 : BC_Window(_(PROGRAM_NAME ": Audio Compression"),
362 parent_window->get_abs_cursor_x(1),
363 parent_window->get_abs_cursor_y(1),
372 this->parent_window = parent_window;
374 // *** CONTEXT_HELP ***
375 context_help_set_keyword("Single File Rendering");
378 VorbisConfigAudio::~VorbisConfigAudio()
382 void VorbisConfigAudio::create_objects()
386 char string[BCTEXTLEN];
388 lock_window("VorbisConfigAudio::create_objects");
389 add_tool(fixed_bitrate = new VorbisFixedBitrate(x, y, this));
390 add_tool(variable_bitrate = new VorbisVariableBitrate(x1, y, this));
393 sprintf(string, "%d", asset->vorbis_min_bitrate);
394 add_tool(new BC_Title(x, y, _("Min bitrate:")));
395 add_tool(new VorbisMinBitrate(x1, y, this, string));
398 add_tool(new BC_Title(x, y, _("Avg bitrate:")));
399 sprintf(string, "%d", asset->vorbis_bitrate);
400 add_tool(new VorbisAvgBitrate(x1, y, this, string));
403 add_tool(new BC_Title(x, y, _("Max bitrate:")));
404 sprintf(string, "%d", asset->vorbis_max_bitrate);
405 add_tool(new VorbisMaxBitrate(x1, y, this, string));
408 add_subwindow(new BC_OKButton(this));
414 int VorbisConfigAudio::close_event()
424 VorbisFixedBitrate::VorbisFixedBitrate(int x, int y, VorbisConfigAudio *gui)
425 : BC_Radial(x, y, !gui->asset->vorbis_vbr, _("Fixed bitrate"))
429 int VorbisFixedBitrate::handle_event()
431 gui->asset->vorbis_vbr = 0;
432 gui->variable_bitrate->update(0);
436 VorbisVariableBitrate::VorbisVariableBitrate(int x, int y, VorbisConfigAudio *gui)
437 : BC_Radial(x, y, gui->asset->vorbis_vbr, _("Variable bitrate"))
441 int VorbisVariableBitrate::handle_event()
443 gui->asset->vorbis_vbr = 1;
444 gui->fixed_bitrate->update(0);
449 VorbisMinBitrate::VorbisMinBitrate(int x,
451 VorbisConfigAudio *gui,
453 : BC_TextBox(x, y, 180, 1, text)
457 int VorbisMinBitrate::handle_event()
459 gui->asset->vorbis_min_bitrate = atol(get_text());
465 VorbisMaxBitrate::VorbisMaxBitrate(int x,
467 VorbisConfigAudio *gui,
469 : BC_TextBox(x, y, 180, 1, text)
473 int VorbisMaxBitrate::handle_event()
475 gui->asset->vorbis_max_bitrate = atol(get_text());
481 VorbisAvgBitrate::VorbisAvgBitrate(int x, int y, VorbisConfigAudio *gui, char *text)
482 : BC_TextBox(x, y, 180, 1, text)
486 int VorbisAvgBitrate::handle_event()
488 gui->asset->vorbis_bitrate = atol(get_text());