From ae364d9bed0589bf1a22cd5f530c563462379e3e Mon Sep 17 00:00:00 2001 From: tpearson Date: Mon, 22 Feb 2010 18:23:26 +0000 Subject: Added old KDE3 version of kradio git-svn-id: svn://anonsvn.kde.org/home/kde/branches/trinity/applications/kradio@1094417 283d02a7-25f6-0310-bc7c-ecb5cbfe19da --- kradio3/plugins/recording/encoder_ogg.cpp | 250 ++++++++++++++++++++++++++++++ 1 file changed, 250 insertions(+) create mode 100644 kradio3/plugins/recording/encoder_ogg.cpp (limited to 'kradio3/plugins/recording/encoder_ogg.cpp') diff --git a/kradio3/plugins/recording/encoder_ogg.cpp b/kradio3/plugins/recording/encoder_ogg.cpp new file mode 100644 index 0000000..e3db6cc --- /dev/null +++ b/kradio3/plugins/recording/encoder_ogg.cpp @@ -0,0 +1,250 @@ +/*************************************************************************** + encoder_ogg.cpp + ------------------- + begin : Sat Aug 20 2005 + copyright : (C) 2005 by Martin Witte + email : witte@kawo1.rwth-aachen.de + ***************************************************************************/ + +/*************************************************************************** + * * + * This program is free software; you can redistribute it and/or modify * + * it under the terms of the GNU General Public License as published by * + * the Free Software Foundation; either version 2 of the License, or * + * (at your option) any later version. * + * * + ***************************************************************************/ + +#include "encoder_ogg.h" + +#include +#include + +RecordingEncodingOgg::RecordingEncodingOgg(QObject *parent, SoundStreamID ssid, + const RecordingConfig &cfg, const RadioStation *rs, + const QString &filename) + : RecordingEncoding(parent, ssid, cfg, rs, filename) +#ifdef HAVE_OGG + , + m_OggOutput(NULL), + m_OggExportBuffer(NULL), + m_OggExportBufferSize(0) +#endif +{ + m_config.m_OutputFormat = RecordingConfig::outputOGG; + m_config.m_SoundFormat.m_Encoding = "ogg"; + openOutput(filename); +} + + +RecordingEncodingOgg::~RecordingEncodingOgg() +{ + closeOutput(); +} + +void RecordingEncodingOgg::encode(const char *_buffer, size_t buffer_size, char *&export_buffer, size_t &export_buffer_size) +{ + if (m_error) + return; + +#ifdef HAVE_OGG + SoundFormat &sf = m_config.m_SoundFormat; + ogg_page ogg_pg; + ogg_packet ogg_pkt; + + size_t samples = buffer_size / sf.frameSize(); + + // buffer[channel][sample], normalized to -1..0..+1 + float **buffer = vorbis_analysis_buffer(&m_VorbisDSP, (samples < 512 ? 512 : samples)); + + sf.convertSamplesToFloat(_buffer, buffer, samples); + + /* Tell the library how many samples (per channel) we wrote + into the supplied buffer */ + vorbis_analysis_wrote(&m_VorbisDSP, samples); + + /* While we can get enough data from the library to analyse, one + block at a time... */ + + bool eos = false; + while(!m_error && !eos && vorbis_analysis_blockout(&m_VorbisDSP, &m_VorbisBlock) == 1) { + + /* Do the main analysis, creating a packet */ + vorbis_analysis(&m_VorbisBlock, NULL); + vorbis_bitrate_addblock(&m_VorbisBlock); + + while(!m_error && vorbis_bitrate_flushpacket(&m_VorbisDSP, &ogg_pkt)) { + /* Add packet to bitstream */ + ogg_stream_packetin(&m_OggStream,&ogg_pkt); + + /* If we've gone over a page boundary, we can do actual output, + so do so (for however many pages are available) */ + + while(!m_error && !eos) { + int result = ogg_stream_pageout(&m_OggStream, &ogg_pg); + if (!result) break; + + int n = fwrite(ogg_pg.header, 1, ogg_pg.header_len, m_OggOutput); + n += fwrite(ogg_pg.body, 1, ogg_pg.body_len, m_OggOutput); + + m_encodedSize += n; + + if (n != (ogg_pg.header_len + ogg_pg.body_len)) { + m_error = true; + m_errorString += i18n("Failed writing data to ogg/vorbis output stream. "); + break; + } else { + + if (m_OggExportBufferSize < export_buffer_size + n) { + m_OggExportBuffer = (char*)realloc(m_OggExportBuffer, m_OggExportBufferSize + 2 * n); + m_OggExportBufferSize += 2 * n; + } + + memcpy (m_OggExportBuffer + export_buffer_size, ogg_pg.header, ogg_pg.header_len); + export_buffer_size += ogg_pg.header_len; + memcpy (m_OggExportBuffer + export_buffer_size, ogg_pg.body, ogg_pg.body_len); + export_buffer_size += ogg_pg.body_len; + + } + if (ogg_page_eos(&ogg_pg)) + eos = 1; + } + } + } + + export_buffer = m_OggExportBuffer; +#endif +} + + +#ifdef HAVE_OGG +static void vorbis_comment_add_tag_new(vorbis_comment *vc, const QString &tag, const QString &value) +{ + char *stag = strdup(tag.ascii()); + char *svalue = strdup(value.utf8()); + vorbis_comment_add_tag(vc, stag, svalue); + delete stag; + delete svalue; +} +#endif + +bool RecordingEncodingOgg::openOutput(const QString &output) +{ +#ifdef HAVE_OGG + m_OggOutput = fopen(output.ascii(), "wb+"); + if (!m_OggOutput) { + m_errorString += i18n("Cannot open Ogg/Vorbis output file %1. ").arg(output); + m_error = true; + } + + m_OggExportBuffer = (char*)malloc(m_OggExportBufferSize = 65536); // start with a 64k buffer + + + /* Have vorbisenc choose a mode for us */ + vorbis_info_init(&m_VorbisInfo); + + SoundFormat &sf = m_config.m_SoundFormat; + if (vorbis_encode_setup_vbr(&m_VorbisInfo, sf.m_Channels, sf.m_SampleRate, m_config.m_oggQuality)) { + m_error = true; + m_errorString = i18n("Ogg/Vorbis Mode initialisation failed: invalid parameters for quality\n"); + vorbis_info_clear(&m_VorbisInfo); + return false; + } + + /* Turn off management entirely (if it was turned on). */ + vorbis_encode_ctl(&m_VorbisInfo, OV_ECTL_RATEMANAGE_SET, NULL); + vorbis_encode_setup_init(&m_VorbisInfo); + + /* Now, set up the analysis engine, stream encoder, and other + preparation before the encoding begins. + */ + + vorbis_analysis_init(&m_VorbisDSP, &m_VorbisInfo); + vorbis_block_init(&m_VorbisDSP, &m_VorbisBlock); + + ogg_stream_init (&m_OggStream, m_SoundStreamID.getID()); + + /* Now, build the three header packets and send through to the stream + output stage (but defer actual file output until the main encode loop) */ + + ogg_packet header_main; + ogg_packet header_comments; + ogg_packet header_codebooks; + + /* Build the packets */ + vorbis_comment vc; + vorbis_comment_init (&vc); + vorbis_comment_add_tag_new(&vc, "creator", "KRadio" VERSION); + vorbis_comment_add_tag_new(&vc, "title", m_RadioStation->longName().utf8()); + vorbis_comment_add_tag_new(&vc, "date", QDateTime::currentDateTime().toString(Qt::ISODate)); + + vorbis_analysis_headerout(&m_VorbisDSP, &vc, + &header_main, &header_comments, &header_codebooks); + + /* And stream them out */ + ogg_stream_packetin(&m_OggStream, &header_main); + ogg_stream_packetin(&m_OggStream, &header_comments); + ogg_stream_packetin(&m_OggStream, &header_codebooks); + + int result; + ogg_page ogg_page; + while((result = ogg_stream_flush(&m_OggStream, &ogg_page))) { + + if (!result) break; + + int n = fwrite(ogg_page.header, 1, ogg_page.header_len, m_OggOutput); + n += fwrite(ogg_page.body, 1, ogg_page.body_len, m_OggOutput); + + if(n != ogg_page.header_len + ogg_page.body_len) { + m_error = true; + m_errorString += i18n("Failed writing Ogg/Vorbis header to output stream\n"); + break; + } + } + + vorbis_comment_clear (&vc); + + if (m_error) { + if (m_OggOutput) fclose (m_OggOutput); + m_OggOutput = NULL; + free(m_OggExportBuffer); + m_OggExportBuffer = NULL; + m_OggExportBufferSize = 0; + + ogg_stream_clear(&m_OggStream); + vorbis_block_clear(&m_VorbisBlock); + vorbis_dsp_clear(&m_VorbisDSP); + vorbis_info_clear(&m_VorbisInfo); + } + + return !m_error; +#endif +} + + +void RecordingEncodingOgg::closeOutput() +{ +#ifdef HAVE_OGG + if (m_OggOutput) { + + char *tmp_buf = NULL; + size_t tmp_size = 0; + // flush buffer + encode(tmp_buf, tmp_size, tmp_buf, tmp_size); + + fclose(m_OggOutput); + m_OggOutput = NULL; + + free(m_OggExportBuffer); + m_OggExportBuffer = NULL; + m_OggExportBufferSize = 0; + + ogg_stream_clear(&m_OggStream); + vorbis_block_clear(&m_VorbisBlock); + vorbis_dsp_clear(&m_VorbisDSP); + vorbis_info_clear(&m_VorbisInfo); + } +#endif +} + + -- cgit v1.2.1