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"
38 FileVorbis::FileVorbis(Asset *asset, File *file)
39 : FileBase(asset, file)
42 if(asset->format == FILE_UNKNOWN) asset->format = FILE_VORBIS;
43 asset->byte_order = 0;
46 FileVorbis::~FileVorbis()
51 void FileVorbis::get_parameters(BC_WindowBase *parent_window,
53 BC_WindowBase* &format_window,
59 VorbisConfigAudio *window = new VorbisConfigAudio(parent_window, asset);
60 format_window = window;
61 window->create_objects();
67 int FileVorbis::check_sig(Asset *asset)
69 FILE *fd = fopen(asset->path, "rb");
72 // Test for Quicktime since OGG misinterprets it
75 fseek(fd, 4, SEEK_SET);
77 (void)fread(data, 4, 1, fd);
87 fseek(fd, 0, SEEK_SET);
89 if(ov_open(fd, &vf, NULL, 0) < 0)
91 // OGG failed. Close file handle manually.
106 int FileVorbis::reset_parameters_derived()
109 bzero(&vf, sizeof(vf));
114 int FileVorbis::open_file(int rd, int wr)
119 //printf("FileVorbis::open_file 1\n");
122 //printf("FileVorbis::open_file 1\n");
123 if(!(fd = fopen(asset->path, "rb")))
125 eprintf("FileVorbis::open_file %s: %s\n", asset->path, strerror(errno));
130 //printf("FileVorbis::open_file 2 %p %p\n", fd, vf);
131 if(ov_open(fd, &vf, NULL, 0) < 0)
133 eprintf(_("FileVorbis::open_file %s: invalid bitstream.\n"), asset->path);
138 //printf("FileVorbis::open_file 1\n");
139 vorbis_info *vi = ov_info(&vf, -1);
140 asset->channels = vi->channels;
141 if(!asset->sample_rate)
142 asset->sample_rate = vi->rate;
143 //printf("FileVorbis::open_file 1\n");
144 asset->audio_length = ov_pcm_total(&vf,-1);
145 //printf("FileVorbis::open_file 1\n");
146 asset->audio_data = 1;
147 // printf("FileVorbis::open_file 1 %d %d %d\n",
149 // asset->sample_rate,
150 // asset->audio_length);
157 if(!(fd = fopen(asset->path, "wb")))
159 eprintf("FileVorbis::open_file %s: %s\n", asset->path, strerror(errno));
164 vorbis_info_init(&vi);
165 if(!asset->vorbis_vbr)
166 result = vorbis_encode_init(&vi,
169 asset->vorbis_max_bitrate,
170 asset->vorbis_bitrate,
171 asset->vorbis_min_bitrate);
174 result = vorbis_encode_setup_managed(&vi,
177 asset->vorbis_max_bitrate,
178 asset->vorbis_bitrate,
179 asset->vorbis_min_bitrate);
180 result |= vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_AVG, NULL);
181 result |= vorbis_encode_setup_init(&vi);
186 vorbis_analysis_init(&vd, &vi);
187 vorbis_block_init(&vd, &vb);
188 vorbis_comment_init(&vc);
190 ogg_stream_init(&os, rand());
193 ogg_packet header_comm;
194 ogg_packet header_code;
195 vorbis_analysis_headerout(&vd,
200 ogg_stream_packetin(&os,
202 ogg_stream_packetin(&os,
204 ogg_stream_packetin(&os,
209 int result = ogg_stream_flush(&os, &og);
210 if(result == 0) break;
211 fwrite(og.header, 1, og.header_len, fd);
212 fwrite(og.body, 1, og.body_len, fd);
218 //printf("FileVorbis::open_file 2\n");
222 #define FLUSH_VORBIS \
223 while(vorbis_analysis_blockout(&vd, &vb) == 1) \
225 vorbis_analysis(&vb, NULL); \
226 vorbis_bitrate_addblock(&vb); \
227 while(vorbis_bitrate_flushpacket(&vd, &op)) \
229 ogg_stream_packetin(&os, &op); \
232 int result = ogg_stream_pageout(&os, &og); \
234 fwrite(og.header, 1, og.header_len, fd); \
235 fwrite(og.body, 1, og.body_len, fd); \
236 if(ogg_page_eos(&og)) break; \
242 int FileVorbis::close_file_derived()
248 vorbis_analysis_wrote(&vd, 0);
251 ogg_stream_clear(&os);
252 vorbis_block_clear(&vb);
253 vorbis_dsp_clear(&vd);
254 vorbis_comment_clear(&vc);
255 vorbis_info_clear(&vi);
261 // This also closes the file handle.
270 int FileVorbis::write_samples(double **buffer, int64_t len)
274 float **vorbis_buffer = vorbis_analysis_buffer(&vd, len);
275 for(int i = 0; i < asset->channels; i++)
277 float *output = vorbis_buffer[i];
278 double *input = buffer[i];
279 for(int j = 0; j < len; j++)
281 output[j] = input[j];
285 vorbis_analysis_wrote(&vd, len);
291 int FileVorbis::read_samples(double *buffer, int64_t len)
295 // printf("FileVorbis::read_samples 1 %d %d %d %d\n",
298 // file->current_sample,
300 float **vorbis_output;
302 int accumulation = 0;
305 update_pcm_history(len);
308 // Fill history buffer
309 if(decode_start != decode_end)
311 ov_pcm_seek(&vf, decode_start);
312 decode_end = decode_start;
315 while(accumulation < decode_len)
317 int result = ov_read_float(&vf,
319 decode_len - accumulation,
321 //printf("FileVorbis::read_samples 1 %d %d %d\n", result, len, accumulation);
324 append_history(vorbis_output, result);
325 accumulation += result;
330 file->current_sample,
331 file->current_channel,
334 // printf("FileVorbis::read_samples 2 %d %d %d %d\n",
337 // file->current_sample,
352 VorbisConfigAudio::VorbisConfigAudio(BC_WindowBase *parent_window,
354 : BC_Window(PROGRAM_NAME ": Audio Compression",
355 parent_window->get_abs_cursor_x(1),
356 parent_window->get_abs_cursor_y(1),
365 this->parent_window = parent_window;
369 VorbisConfigAudio::~VorbisConfigAudio()
373 void VorbisConfigAudio::create_objects()
377 char string[BCTEXTLEN];
379 lock_window("VorbisConfigAudio::create_objects");
380 add_tool(fixed_bitrate = new VorbisFixedBitrate(x, y, this));
381 add_tool(variable_bitrate = new VorbisVariableBitrate(x1, y, this));
384 sprintf(string, "%d", asset->vorbis_min_bitrate);
385 add_tool(new BC_Title(x, y, _("Min bitrate:")));
386 add_tool(new VorbisMinBitrate(x1, y, this, string));
389 add_tool(new BC_Title(x, y, _("Avg bitrate:")));
390 sprintf(string, "%d", asset->vorbis_bitrate);
391 add_tool(new VorbisAvgBitrate(x1, y, this, string));
394 add_tool(new BC_Title(x, y, _("Max bitrate:")));
395 sprintf(string, "%d", asset->vorbis_max_bitrate);
396 add_tool(new VorbisMaxBitrate(x1, y, this, string));
399 add_subwindow(new BC_OKButton(this));
405 int VorbisConfigAudio::close_event()
415 VorbisFixedBitrate::VorbisFixedBitrate(int x, int y, VorbisConfigAudio *gui)
416 : BC_Radial(x, y, !gui->asset->vorbis_vbr, _("Fixed bitrate"))
420 int VorbisFixedBitrate::handle_event()
422 gui->asset->vorbis_vbr = 0;
423 gui->variable_bitrate->update(0);
427 VorbisVariableBitrate::VorbisVariableBitrate(int x, int y, VorbisConfigAudio *gui)
428 : BC_Radial(x, y, gui->asset->vorbis_vbr, _("Variable bitrate"))
432 int VorbisVariableBitrate::handle_event()
434 gui->asset->vorbis_vbr = 1;
435 gui->fixed_bitrate->update(0);
440 VorbisMinBitrate::VorbisMinBitrate(int x,
442 VorbisConfigAudio *gui,
444 : BC_TextBox(x, y, 180, 1, text)
448 int VorbisMinBitrate::handle_event()
450 gui->asset->vorbis_min_bitrate = atol(get_text());
456 VorbisMaxBitrate::VorbisMaxBitrate(int x,
458 VorbisConfigAudio *gui,
460 : BC_TextBox(x, y, 180, 1, text)
464 int VorbisMaxBitrate::handle_event()
466 gui->asset->vorbis_max_bitrate = atol(get_text());
472 VorbisAvgBitrate::VorbisAvgBitrate(int x, int y, VorbisConfigAudio *gui, char *text)
473 : BC_TextBox(x, y, 180, 1, text)
477 int VorbisAvgBitrate::handle_event()
479 gui->asset->vorbis_bitrate = atol(get_text());