static GstBuffer *
_create_codebook_header_buffer (void)
{
GstBuffer *buffer;
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_info_init (&vi);
vorbis_encode_setup_vbr (&vi, 1, 44000, 0.5);
vorbis_encode_setup_init (&vi);
vorbis_analysis_init (&vd, &vi);
vorbis_block_init (&vd, &vb);
vorbis_comment_init (&vc);
vorbis_analysis_headerout (&vd, &vc, &header, &header_comm, &header_code);
buffer = gst_buffer_new_and_alloc (header_code.bytes);
gst_buffer_fill (buffer, 0, header_code.packet, header_code.bytes);
return buffer;
}
encoder_instance encoder_create_vbr(int ch, int bitrate, float quality) {
encoder_instance state = malloc(sizeof(struct encoder_state));
state->data = NULL;
state->data_len = 0;
vorbis_info_init(&state->vi);
if (vorbis_encode_init_vbr(&state->vi, ch, bitrate, quality) != 0) {
free(state);
return NULL;
}
vorbis_comment_init(&state->vc);
vorbis_comment_add_tag(&state->vc, "ENCODER", "libvorbis.js");
vorbis_analysis_init(&state->vd, &state->vi);
vorbis_block_init(&state->vd, &state->vb);
srand(time(NULL));
ogg_stream_init(&state->os, rand());
return state;
}
/* Create an ogg stream and vorbis encoder, with the configuration
* specified in the encoder_state.
*/
static gboolean
xmms_ices_encoder_create (encoder_state *s, vorbis_comment *vc)
{
ogg_packet header[3];
if (s->encoder_inited) {
XMMS_DBG ("OOPS: xmms_ices_encoder_create called "
"with s->encoder_inited == TRUE !");
}
XMMS_DBG ("Creating encoder in ABR mode: min/avg/max bitrate %d/%d/%d",
s->min_br, s->nom_br, s->max_br);
/* Create the Vorbis encoder. */
vorbis_info_init (&s->vi);
if (vorbis_encode_init (&s->vi, s->channels, s->rate,
s->max_br, s->nom_br, s->min_br) < 0)
return FALSE;
vorbis_analysis_init (&s->vd, &s->vi);
vorbis_block_init (&s->vd, &s->vb);
/* Initialize the ogg stream and input the vorbis header
* packets. */
ogg_stream_init (&s->os, s->serial++);
vorbis_analysis_headerout (&s->vd, vc, &header[0], &header[1], &header[2]);
ogg_stream_packetin (&s->os, &header[0]);
ogg_stream_packetin (&s->os, &header[1]);
ogg_stream_packetin (&s->os, &header[2]);
s->in_header = TRUE;
s->flushing = FALSE;
s->samples_in_current_page = 0;
s->previous_granulepos = 0;
s->encoder_inited = TRUE;
return TRUE;
}
static bool
vorbis_encoder_reinit(struct vorbis_encoder *encoder, GError **error)
{
vorbis_info_init(&encoder->vi);
if (encoder->quality >= -1.0) {
/* a quality was configured (VBR) */
if (0 != vorbis_encode_init_vbr(&encoder->vi,
encoder->audio_format.channels,
encoder->audio_format.sample_rate,
encoder->quality * 0.1)) {
g_set_error(error, vorbis_encoder_quark(), 0,
"error initializing vorbis vbr");
vorbis_info_clear(&encoder->vi);
return false;
}
} else {
/* a bit rate was configured */
if (0 != vorbis_encode_init(&encoder->vi,
encoder->audio_format.channels,
encoder->audio_format.sample_rate, -1.0,
encoder->bitrate * 1000, -1.0)) {
g_set_error(error, vorbis_encoder_quark(), 0,
"error initializing vorbis encoder");
vorbis_info_clear(&encoder->vi);
return false;
}
}
vorbis_analysis_init(&encoder->vd, &encoder->vi);
vorbis_block_init(&encoder->vd, &encoder->vb);
ogg_stream_init(&encoder->os, g_random_int());
return true;
}
StreamEncoder::StreamEncoder()
{
//Initialize the info
vorbis_info_init(&mVorbisInfo);
if (ErrorCheck(vorbis_encode_init(&mVorbisInfo, 2, 44100, 100, 80, 60)) == true)
{
//Error
Write("vorbis_encode_init error");
return;
}
if (ErrorCheck(vorbis_analysis_init(&mVorbisDspState, &mVorbisInfo)) == true)
{
//Error
Write("vorbis_analysis_init error");
return;
}
vorbis_comment_init(&mVorbisComment);
//vorbis_comment_add(&mVorbisComment, "Comments");
int vahCode = vorbis_analysis_headerout(&mVorbisDspState, &mVorbisComment, &mOggPacketIdentification, &mOggPacketComment, &mOggPacketCodes);
if (ErrorCheck(vahCode) == true)
{
//Error
Write("vorbis_analysis_init error");
return;
}
if (ErrorCheck(vorbis_block_init(&mVorbisDspState, &mVorbisBlock)) == true)
{
//Error
Write("vorbis_block_init error");
return;
}
}
void EncoderVorbis::initStream() {
// set up analysis state and auxiliary encoding storage
vorbis_analysis_init(&m_vdsp, &m_vinfo);
vorbis_block_init(&m_vdsp, &m_vblock);
// set up packet-to-stream encoder; attach a random serial number
srand(time(0));
ogg_stream_init(&m_oggs, getSerial());
// add comment
vorbis_comment_init(&m_vcomment);
vorbis_comment_add_tag(&m_vcomment, "ENCODER", "mixxx/libvorbis");
if (m_metaDataArtist != NULL) {
vorbis_comment_add_tag(&m_vcomment, "ARTIST", m_metaDataArtist);
}
if (m_metaDataTitle != NULL) {
vorbis_comment_add_tag(&m_vcomment, "TITLE", m_metaDataTitle);
}
if (m_metaDataAlbum != NULL) {
vorbis_comment_add_tag(&m_vcomment, "ALBUM", m_metaDataAlbum);
}
// set up the vorbis headers
ogg_packet headerInit;
ogg_packet headerComment;
ogg_packet headerCode;
vorbis_analysis_headerout(&m_vdsp, &m_vcomment, &headerInit, &headerComment, &headerCode);
ogg_stream_packetin(&m_oggs, &headerInit);
ogg_stream_packetin(&m_oggs, &headerComment);
ogg_stream_packetin(&m_oggs, &headerCode);
// The encoder is now inialized. The encode method will start streaming by
// sending the header first.
m_header_write = true;
m_bStreamInitialized = true;
}
int Encode_Ogg(void *stream,void(*writefunc)(void *bytes,int count,void *stream),int freq,int channels,float *samples,int length,float compression) {
oggwriter *ogg;
int eos;
int result;
ogg=(oggwriter*)malloc(sizeof(oggwriter));
vorbis_info_init(&ogg->vi);
result=vorbis_encode_init_vbr(&ogg->vi,channels,freq,compression);
if(result) return -1; //error format not supported...
// add a comment
vorbis_comment_init(&ogg->vc);
vorbis_comment_add_tag(&ogg->vc,"ENCODER","encoder_example.c");
// set up the analysis state and auxiliary encoding storage
vorbis_analysis_init(&ogg->vd,&ogg->vi);
vorbis_block_init(&ogg->vd,&ogg->vb);
srand(time(NULL));
ogg_stream_init(&ogg->os,rand());
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&ogg->vd,&ogg->vc,&header,&header_comm,&header_code);
ogg_stream_packetin(&ogg->os,&header); //automatically placed in its own page
ogg_stream_packetin(&ogg->os,&header_comm);
ogg_stream_packetin(&ogg->os,&header_code);
//This ensures the actual audio data will start on a new page, as per spec
while(1) {
result=ogg_stream_flush(&ogg->os,&ogg->og);
if(result==0)break;
writefunc(ogg->og.header,ogg->og.header_len,stream);
writefunc(ogg->og.body,ogg->og.body_len,stream);
}
int i,c,n;
eos=0;
while(!eos) {
if (length>0) {
float **buffer=vorbis_analysis_buffer(&ogg->vd,READ);
n=length;
if (n>READ) n=READ;
//uninterleave samples
for (i=0; i<n; i++) {
for (c=0; c<channels; c++) {
buffer[c][i]=*samples++;
}
}
length=length-n;
//tell the library how much we actually submitted
vorbis_analysis_wrote(&ogg->vd,i);
} else {
vorbis_analysis_wrote(&ogg->vd,0);
}
while(vorbis_analysis_blockout(&ogg->vd,&ogg->vb)==1) {
//analysis, assume we want to use bitrate management
vorbis_analysis(&ogg->vb,NULL);
vorbis_bitrate_addblock(&ogg->vb);
while(vorbis_bitrate_flushpacket(&ogg->vd,&ogg->op)) {
//weld the packet into the bitstream
ogg_stream_packetin(&ogg->os,&ogg->op);
//write out pages (if any)
while(!eos) {
result=ogg_stream_pageout(&ogg->os,&ogg->og);
if(result==0)break;
writefunc(ogg->og.header,ogg->og.header_len,stream);
writefunc(ogg->og.body,ogg->og.body_len,stream);
if (ogg_page_eos(&ogg->og)) eos=1;
}
}
}
}
//clean up and exit. vorbis_info_clear() must be called last
ogg_stream_clear(&ogg->os);
vorbis_block_clear(&ogg->vb);
vorbis_dsp_clear(&ogg->vd);
vorbis_comment_clear(&ogg->vc);
vorbis_info_clear(&ogg->vi);
free(ogg);
return 0;
}
static gboolean
gst_vorbis_enc_setup (GstVorbisEnc * vorbisenc)
{
GST_LOG_OBJECT (vorbisenc, "setup");
if (vorbisenc->bitrate < 0 && vorbisenc->min_bitrate < 0
&& vorbisenc->max_bitrate < 0) {
vorbisenc->quality_set = TRUE;
}
update_start_message (vorbisenc);
/* choose an encoding mode */
/* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
vorbis_info_init (&vorbisenc->vi);
if (vorbisenc->quality_set) {
if (vorbis_encode_setup_vbr (&vorbisenc->vi,
vorbisenc->channels, vorbisenc->frequency,
vorbisenc->quality) != 0) {
GST_ERROR_OBJECT (vorbisenc,
"vorbisenc: initialisation failed: invalid parameters for quality");
vorbis_info_clear (&vorbisenc->vi);
return FALSE;
}
/* do we have optional hard quality restrictions? */
if (vorbisenc->max_bitrate > 0 || vorbisenc->min_bitrate > 0) {
struct ovectl_ratemanage_arg ai;
vorbis_encode_ctl (&vorbisenc->vi, OV_ECTL_RATEMANAGE_GET, &ai);
ai.bitrate_hard_min = vorbisenc->min_bitrate;
ai.bitrate_hard_max = vorbisenc->max_bitrate;
ai.management_active = 1;
vorbis_encode_ctl (&vorbisenc->vi, OV_ECTL_RATEMANAGE_SET, &ai);
}
} else {
long min_bitrate, max_bitrate;
min_bitrate = vorbisenc->min_bitrate > 0 ? vorbisenc->min_bitrate : -1;
max_bitrate = vorbisenc->max_bitrate > 0 ? vorbisenc->max_bitrate : -1;
if (vorbis_encode_setup_managed (&vorbisenc->vi,
vorbisenc->channels,
vorbisenc->frequency,
max_bitrate, vorbisenc->bitrate, min_bitrate) != 0) {
GST_ERROR_OBJECT (vorbisenc,
"vorbis_encode_setup_managed "
"(c %d, rate %d, max br %ld, br %d, min br %ld) failed",
vorbisenc->channels, vorbisenc->frequency, max_bitrate,
vorbisenc->bitrate, min_bitrate);
vorbis_info_clear (&vorbisenc->vi);
return FALSE;
}
}
if (vorbisenc->managed && vorbisenc->bitrate < 0) {
vorbis_encode_ctl (&vorbisenc->vi, OV_ECTL_RATEMANAGE_AVG, NULL);
} else if (!vorbisenc->managed) {
/* Turn off management entirely (if it was turned on). */
vorbis_encode_ctl (&vorbisenc->vi, OV_ECTL_RATEMANAGE_SET, NULL);
}
vorbis_encode_setup_init (&vorbisenc->vi);
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init (&vorbisenc->vd, &vorbisenc->vi);
vorbis_block_init (&vorbisenc->vd, &vorbisenc->vb);
/* samples == granulepos start at 0 again */
vorbisenc->samples_out = 0;
/* fresh encoder available */
vorbisenc->setup = TRUE;
return TRUE;
}
bool ExportOGG(AudacityProject *project,
bool stereo, wxString fName,
bool selectionOnly, double t0, double t1)
{
double rate = project->GetRate();
wxWindow *parent = project;
TrackList *tracks = project->GetTracks();
double quality = (gPrefs->Read("/FileFormats/OggExportQuality", 50)/(float)100.0);
wxLogNull logNo; // temporarily disable wxWindows error messages
bool cancelling = false;
wxFFile outFile(fName, "wb");
if(!outFile.IsOpened()) {
wxMessageBox(_("Unable to open target file for writing"));
return false;
}
// All the Ogg and Vorbis encoding data
ogg_stream_state stream;
ogg_page page;
ogg_packet packet;
vorbis_info info;
vorbis_comment comment;
vorbis_dsp_state dsp;
vorbis_block block;
// Encoding setup
vorbis_info_init(&info);
vorbis_encode_init_vbr(&info, stereo ? 2 : 1, int(rate + 0.5), quality);
vorbis_comment_init(&comment);
// If we wanted to add comments, we would do it here
// Set up analysis state and auxiliary encoding storage
vorbis_analysis_init(&dsp, &info);
vorbis_block_init(&dsp, &block);
// Set up packet->stream encoder. According to encoder example,
// a random serial number makes it more likely that you can make
// chained streams with concatenation.
srand(time(NULL));
ogg_stream_init(&stream, rand());
// First we need to write the required headers:
// 1. The Ogg bitstream header, which contains codec setup params
// 2. The Vorbis comment header
// 3. The bitstream codebook.
//
// After we create those our responsibility is complete, libvorbis will
// take care of any other ogg bistream constraints (again, according
// to the example encoder source)
ogg_packet bitstream_header;
ogg_packet comment_header;
ogg_packet codebook_header;
vorbis_analysis_headerout(&dsp, &comment, &bitstream_header, &comment_header,
&codebook_header);
// Place these headers into the stream
ogg_stream_packetin(&stream, &bitstream_header);
ogg_stream_packetin(&stream, &comment_header);
ogg_stream_packetin(&stream, &codebook_header);
// Flushing these headers now guarentees that audio data will
// start on a new page, which apparently makes streaming easier
ogg_stream_flush(&stream, &page);
outFile.Write(page.header, page.header_len);
outFile.Write(page.body, page.body_len);
double t = t0;
bool done = false;
wxProgressDialog *progress = NULL;
wxYield();
wxStartTimer();
while(!done && !cancelling){
float deltat = (float)SAMPLES_PER_RUN / rate;
sampleCount samplesThisRun = SAMPLES_PER_RUN;
Mixer *mixer = new Mixer(stereo ? 2 : 1, SAMPLES_PER_RUN,
/* interleaved = */ false, rate, floatSample);
if(t + deltat > t1) {
done = true;
deltat = t1 - t;
samplesThisRun = int(deltat * rate + 0.5);
}
mixer->Clear();
TrackListIterator iter(tracks);
Track *tr = iter.First();
while (tr) {
if (tr->GetKind() == Track::Wave) {
if (tr->GetSelected() || !selectionOnly) {
if (tr->GetChannel() == Track::MonoChannel)
mixer->MixMono((WaveTrack *) tr, t, t + deltat);
else if (tr->GetChannel() == Track::LeftChannel)
mixer->MixLeft((WaveTrack *) tr, t, t + deltat);
else if (tr->GetChannel() == Track::RightChannel)
mixer->MixRight((WaveTrack *) tr, t, t + deltat);
}
}
tr = iter.Next();
}
float **vorbis_buffer = vorbis_analysis_buffer(&dsp, SAMPLES_PER_RUN);
float *left = (float *)mixer->GetBuffer(0);
memcpy(vorbis_buffer[0], left, sizeof(float)*SAMPLES_PER_RUN);
if(stereo) {
float *right = (float *)mixer->GetBuffer(1);
memcpy(vorbis_buffer[1], right, sizeof(float)*SAMPLES_PER_RUN);
}
// tell the encoder how many samples we have
vorbis_analysis_wrote(&dsp, samplesThisRun);
// I don't understand what this call does, so here is the comment
// from the example, verbatim:
//
// vorbis does some data preanalysis, then divvies up blocks
// for more involved (potentially parallel) processing. Get
// a single block for encoding now
while(vorbis_analysis_blockout(&dsp, &block) == 1) {
// analysis, assume we want to use bitrate management
vorbis_analysis(&block, NULL);
vorbis_bitrate_addblock(&block);
while(vorbis_bitrate_flushpacket(&dsp, &packet)) {
// add the packet to the bitstream
ogg_stream_packetin(&stream, &packet);
int result = ogg_stream_pageout(&stream, &page);
if(result != 0) {
outFile.Write(page.header, page.header_len);
outFile.Write(page.body, page.body_len);
}
}
}
if(progress)
cancelling = !progress->Update(int (((t - t0) * 1000) / (t1 - t0) + 0.5));
else if(wxGetElapsedTime(false) > 500) {
wxString message = selectionOnly ?
_("Exporting the selected audio as Ogg Vorbis") :
_("Exporting the entire project as Ogg Vorbis");
progress = new wxProgressDialog(
_("Export"),
message,
1000,
parent,
wxPD_CAN_ABORT | wxPD_REMAINING_TIME | wxPD_AUTO_HIDE);
}
delete mixer;
t += deltat;
}
outFile.Close();
if(progress)
delete progress;
return true;
}
tbool CVorbisEncoder::Process_Init_Descendant()
{
// Setup encoder
vorbis_info_init(&vi);
//mfQuality = -0.1;
if (vorbis_encode_init_vbr(&vi, miOutputChannels, miInputSampleFreq, mfQuality) != 0) {
AppendErrMsg("Function vorbis_encode_init_vbr failed for unknown reason.");
return false;
}
// Note: Using vorbis_encode_init rather than vorbis_encode_init_vbr makes encoding 4 times slower
// because it will try harder at targetting the bitrate you ask of it.
// Also I think the guy with the unpronouncable name that tuned vorbis (the aoTuV versions)
// only bothered to optimize the "_vbr" method.. and that's quite ok.
// Lasse
//
//if (vorbis_encode_init(&vi, miOutputChannels, miInputSampleFreq, 256000, 192000, 32000) != 0) {
// AppendErrMsg("Function vorbis_encode_init(..) failed for unknown reason.");
// return false;
//}
// TODO: add comments
vorbis_comment_init(&vc);
vorbis_comment_add_tag(&vc,"ENCODER","Koblo 1.0 // aoTuV b5");
vorbis_comment_add_tag(&vc,"KOBLO_ENC_QUAL", (char*)GetQualityName(meQuality));
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init(&vd,&vi);
vorbis_block_init(&vd,&vb);
/* set up our packet->stream encoder */
/* pick a random serial number; that way we can more likely build
chained streams just by concatenation */
srand((unsigned)time(NULL));
ogg_stream_init(&os,rand());
/* Vorbis streams begin with three headers; the initial header (with
most of the codec setup parameters) which is mandated by the Ogg
bitstream spec. The second header holds any comment fields. The
third header holds the bitstream codebook. We merely need to
make the headers, then pass them to libvorbis one at a time;
libvorbis handles the additional Ogg bitstream constraints */
{
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
ogg_stream_packetin(&os,&header); /* automatically placed in its own
page */
ogg_stream_packetin(&os,&header_comm);
ogg_stream_packetin(&os,&header_code);
/* This ensures the actual
* audio data will start on a new page, as per spec
*/
while(1){
int result=ogg_stream_flush(&os,&og);
if(result==0)break;
//fwrite(og.header,1,og.header_len,stdout);
//fwrite(og.body,1,og.body_len,stdout);
WriteOutput((char*)og.header, og.header_len);
WriteOutput((char*)og.body, og.body_len);
}
}
return true;
} // Process_Init_Secendant
int main(){
ogg_stream_state os; /* take physical pages, weld into a logical
stream of packets */
ogg_page og; /* one Ogg bitstream page. Vorbis packets are inside */
ogg_packet op; /* one raw packet of data for decode */
vorbis_info vi; /* struct that stores all the static vorbis bitstream
settings */
vorbis_comment vc; /* struct that stores all the user comments */
vorbis_dsp_state vd; /* central working state for the packet->PCM decoder */
vorbis_block vb; /* local working space for packet->PCM decode */
int eos=0,ret;
int i, founddata;
#if defined(macintosh) && defined(__MWERKS__)
int argc = 0;
char **argv = NULL;
argc = ccommand(&argv); /* get a "command line" from the Mac user */
/* this also lets the user set stdin and stdout */
#endif
/* we cheat on the WAV header; we just bypass 44 bytes and never
verify that it matches 16bit/stereo/44.1kHz. This is just an
example, after all. */
#ifdef _WIN32 /* We need to set stdin/stdout to binary mode. Damn windows. */
/* if we were reading/writing a file, it would also need to in
binary mode, eg, fopen("file.wav","wb"); */
/* Beware the evil ifdef. We avoid these where we can, but this one we
cannot. Don't add any more, you'll probably go to hell if you do. */
_setmode( _fileno( stdin ), _O_BINARY );
_setmode( _fileno( stdout ), _O_BINARY );
#endif
/* we cheat on the WAV header; we just bypass the header and never
verify that it matches 16bit/stereo/44.1kHz. This is just an
example, after all. */
readbuffer[0] = '\0';
for (i=0, founddata=0; i<30 && ! feof(stdin) && ! ferror(stdin); i++)
{
fread(readbuffer,1,2,stdin);
if ( ! strncmp((char*)readbuffer, "da", 2) )
{
founddata = 1;
fread(readbuffer,1,6,stdin);
break;
}
}
/********** Encode setup ************/
vorbis_info_init(&vi);
/* choose an encoding mode. A few possibilities commented out, one
actually used: */
/*********************************************************************
Encoding using a VBR quality mode. The usable range is -.1
(lowest quality, smallest file) to 1. (highest quality, largest file).
Example quality mode .4: 44kHz stereo coupled, roughly 128kbps VBR
ret = vorbis_encode_init_vbr(&vi,2,44100,.4);
---------------------------------------------------------------------
Encoding using an average bitrate mode (ABR).
example: 44kHz stereo coupled, average 128kbps VBR
ret = vorbis_encode_init(&vi,2,44100,-1,128000,-1);
---------------------------------------------------------------------
Encode using a quality mode, but select that quality mode by asking for
an approximate bitrate. This is not ABR, it is true VBR, but selected
using the bitrate interface, and then turning bitrate management off:
ret = ( vorbis_encode_setup_managed(&vi,2,44100,-1,128000,-1) ||
vorbis_encode_ctl(&vi,OV_ECTL_RATEMANAGE2_SET,NULL) ||
vorbis_encode_setup_init(&vi));
*********************************************************************/
ret=vorbis_encode_init_vbr(&vi,2,44100,0.1);
/* do not continue if setup failed; this can happen if we ask for a
mode that libVorbis does not support (eg, too low a bitrate, etc,
will return 'OV_EIMPL') */
if(ret)exit(1);
/* add a comment */
vorbis_comment_init(&vc);
vorbis_comment_add_tag(&vc,"ENCODER","encoder_example.c");
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init(&vd,&vi);
vorbis_block_init(&vd,&vb);
/* set up our packet->stream encoder */
/* pick a random serial number; that way we can more likely build
chained streams just by concatenation */
srand(time(NULL));
ogg_stream_init(&os,rand());
/* Vorbis streams begin with three headers; the initial header (with
most of the codec setup parameters) which is mandated by the Ogg
bitstream spec. The second header holds any comment fields. The
third header holds the bitstream codebook. We merely need to
make the headers, then pass them to libvorbis one at a time;
libvorbis handles the additional Ogg bitstream constraints */
{
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
ogg_stream_packetin(&os,&header); /* automatically placed in its own
page */
ogg_stream_packetin(&os,&header_comm);
ogg_stream_packetin(&os,&header_code);
/* This ensures the actual
* audio data will start on a new page, as per spec
*/
while(!eos){
int result=ogg_stream_flush(&os,&og);
if(result==0)break;
fwrite(og.header,1,og.header_len,stdout);
fwrite(og.body,1,og.body_len,stdout);
}
}
while(!eos){
long i;
long bytes=fread(readbuffer,1,READ*4,stdin); /* stereo hardwired here */
if(bytes==0){
/* end of file. this can be done implicitly in the mainline,
but it's easier to see here in non-clever fashion.
Tell the library we're at end of stream so that it can handle
the last frame and mark end of stream in the output properly */
vorbis_analysis_wrote(&vd,0);
}else{
/* data to encode */
/* expose the buffer to submit data */
float **buffer=vorbis_analysis_buffer(&vd,READ);
/* uninterleave samples */
for(i=0;i<bytes/4;i++){
buffer[0][i]=((readbuffer[i*4+1]<<8)|
(0x00ff&(int)readbuffer[i*4]))/32768.f;
buffer[1][i]=((readbuffer[i*4+3]<<8)|
(0x00ff&(int)readbuffer[i*4+2]))/32768.f;
}
/* tell the library how much we actually submitted */
vorbis_analysis_wrote(&vd,i);
}
/* vorbis does some data preanalysis, then divvies up blocks for
more involved (potentially parallel) processing. Get a single
block for encoding now */
while(vorbis_analysis_blockout(&vd,&vb)==1){
/* analysis, assume we want to use bitrate management */
vorbis_analysis(&vb,NULL);
vorbis_bitrate_addblock(&vb);
while(vorbis_bitrate_flushpacket(&vd,&op)){
/* weld the packet into the bitstream */
ogg_stream_packetin(&os,&op);
/* write out pages (if any) */
while(!eos){
int result=ogg_stream_pageout(&os,&og);
if(result==0)break;
fwrite(og.header,1,og.header_len,stdout);
fwrite(og.body,1,og.body_len,stdout);
/* this could be set above, but for illustrative purposes, I do
it here (to show that vorbis does know where the stream ends) */
if(ogg_page_eos(&og))eos=1;
}
}
}
}
/* clean up and exit. vorbis_info_clear() must be called last */
ogg_stream_clear(&os);
vorbis_block_clear(&vb);
vorbis_dsp_clear(&vd);
vorbis_comment_clear(&vc);
vorbis_info_clear(&vi);
/* ogg_page and ogg_packet structs always point to storage in
libvorbis. They're never freed or manipulated directly */
fprintf(stderr,"Done.\n");
return(0);
}
void rmdInitEncoder(ProgData *pdata,EncData *enc_data_t,int buffer_ready){
int y0,
y1,
y2,
fname_length;
ogg_stream_state m_ogg_skel;
ogg_page skel_og_pg;
fisbone_packet skel_fbv, //video fisbone packet
skel_fba ; //audio fisbone packet
(pdata)->enc_data=enc_data_t;
fname_length=strlen(pdata->args.filename);
if(!(fname_length>4 &&
pdata->args.filename[fname_length-4] == '.' &&
(pdata->args.filename[fname_length-3] == 'o' ||
pdata->args.filename[fname_length-3] == 'O') &&
(pdata->args.filename[fname_length-2] == 'g' ||
pdata->args.filename[fname_length-2] == 'G') &&
(pdata->args.filename[fname_length-1] == 'v' ||
pdata->args.filename[fname_length-1] == 'V'))){
char *new_name=malloc(fname_length+5);
strcpy(new_name,pdata->args.filename);
strcat(new_name,".ogv");
free(pdata->args.filename);
pdata->args.filename=new_name;
}
if (!pdata->args.overwrite) {
rmdIncrementalNaming(&(pdata)->args.filename);
fprintf(stderr, "Output file: %s\n", pdata->args.filename);
}
enc_data_t->fp=fopen((pdata)->args.filename,"w");
if(enc_data_t->fp==NULL){
fprintf(stderr,"Cannot open file %s for writting!\n",
(pdata)->args.filename);
exit(13);
}
//each stream must have a unique
srand(time(NULL));
y0=rand()+1;
y1=rand()+1;
y2=rand()+1;
y2+=(y1==y2);
y0=(((y0==y1)||(y0==y2))?(y1+y2):y0);
//init ogg streams
//skeleton first
ogg_stream_init(&m_ogg_skel,y0);
m_add_fishead_packet(&m_ogg_skel);
if(ogg_stream_pageout(&m_ogg_skel,&skel_og_pg)!= 1){
fprintf (stderr, "Internal Ogg library error.\n");
exit (2);
}
fwrite(skel_og_pg.header,1,skel_og_pg.header_len,enc_data_t->fp);
fwrite(skel_og_pg.body,1,skel_og_pg.body_len,enc_data_t->fp);
ogg_stream_init(&enc_data_t->m_ogg_ts,y1);
if(!pdata->args.nosound)
ogg_stream_init(&enc_data_t->m_ogg_vs,y2);
theora_info_init(&enc_data_t->m_th_inf);
enc_data_t->m_th_inf.frame_width = pdata->brwin.rrect.width;
enc_data_t->m_th_inf.frame_height = pdata->brwin.rrect.height;
enc_data_t->m_th_inf.width = ((enc_data_t->m_th_inf.frame_width + 15) >> 4) << 4;
enc_data_t->m_th_inf.height = ((enc_data_t->m_th_inf.frame_height + 15) >> 4) << 4;
enc_data_t->m_th_inf.offset_x = 0;
enc_data_t->m_th_inf.offset_y = 0;
enc_data_t->m_th_inf.fps_numerator = pdata->args.fps * 100.0;
enc_data_t->m_th_inf.fps_denominator = 100;
enc_data_t->m_th_inf.aspect_numerator = 1;
enc_data_t->m_th_inf.aspect_denominator = 1;
enc_data_t->m_th_inf.colorspace = OC_CS_UNSPECIFIED;
enc_data_t->m_th_inf.pixelformat = OC_PF_420;
enc_data_t->m_th_inf.target_bitrate = pdata->args.v_bitrate;
enc_data_t->m_th_inf.quality = pdata->args.v_quality;
enc_data_t->m_th_inf.dropframes_p = 0;
enc_data_t->m_th_inf.quick_p = 1;
enc_data_t->m_th_inf.keyframe_auto_p = 1;
enc_data_t->m_th_inf.keyframe_frequency = 64;
enc_data_t->m_th_inf.keyframe_frequency_force = 64;
enc_data_t->m_th_inf.keyframe_data_target_bitrate = enc_data_t->m_th_inf.quality * 1.5;
enc_data_t->m_th_inf.keyframe_auto_threshold = 80;
enc_data_t->m_th_inf.keyframe_mindistance = 8;
enc_data_t->m_th_inf.noise_sensitivity = 1;
enc_data_t->m_th_inf.sharpness = 2;
theora_encode_init(&enc_data_t->m_th_st,&enc_data_t->m_th_inf);
if(!pdata->args.nosound){
int ret;
vorbis_info_init(&enc_data_t->m_vo_inf);
ret = vorbis_encode_init_vbr(&enc_data_t->m_vo_inf,
pdata->args.channels,
pdata->args.frequency,
(float)pdata->args.s_quality*0.1);
if(ret){
fprintf(stderr,"Error while setting up vorbis stream quality!\n");
exit(2);
}
vorbis_comment_init(&enc_data_t->m_vo_cmmnt);
vorbis_analysis_init(&enc_data_t->m_vo_dsp,&enc_data_t->m_vo_inf);
vorbis_block_init(&enc_data_t->m_vo_dsp,&enc_data_t->m_vo_block);
}
theora_encode_header(&enc_data_t->m_th_st,&enc_data_t->m_ogg_pckt1);
ogg_stream_packetin(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pckt1);
if(ogg_stream_pageout(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pg)!=1){
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
fwrite(enc_data_t->m_ogg_pg.header,1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
fwrite(enc_data_t->m_ogg_pg.body,1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
theora_comment_init(&enc_data_t->m_th_cmmnt);
theora_comment_add_tag(&enc_data_t->m_th_cmmnt,"recordMyDesktop",VERSION);
theora_encode_comment(&enc_data_t->m_th_cmmnt,&enc_data_t->m_ogg_pckt1);
ogg_stream_packetin(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pckt1);
theora_encode_tables(&enc_data_t->m_th_st,&enc_data_t->m_ogg_pckt1);
ogg_stream_packetin(&enc_data_t->m_ogg_ts,&enc_data_t->m_ogg_pckt1);
if(!pdata->args.nosound){
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&enc_data_t->m_vo_dsp,
&enc_data_t->m_vo_cmmnt,
&header,&header_comm,
&header_code);
ogg_stream_packetin(&enc_data_t->m_ogg_vs,&header);
if(ogg_stream_pageout(&enc_data_t->m_ogg_vs,&enc_data_t->m_ogg_pg)!=1){
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
fwrite(enc_data_t->m_ogg_pg.header,1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
fwrite(enc_data_t->m_ogg_pg.body,1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
ogg_stream_packetin(&enc_data_t->m_ogg_vs,&header_comm);
ogg_stream_packetin(&enc_data_t->m_ogg_vs,&header_code);
}
//fishbone packets go here
memset(&skel_fbv,0,sizeof(skel_fbv));
skel_fbv.serial_no=enc_data_t->m_ogg_ts.serialno;
skel_fbv.nr_header_packet=3;
skel_fbv.granule_rate_n=enc_data_t->m_th_inf.fps_numerator;
skel_fbv.granule_rate_d=enc_data_t->m_th_inf.fps_denominator;
skel_fbv.start_granule=0;
skel_fbv.preroll=0;
skel_fbv.granule_shift=theora_granule_shift(&enc_data_t->m_th_inf);
add_message_header_field(&skel_fbv,
"Content-Type",
"video/theora");
add_fisbone_to_stream(&m_ogg_skel,&skel_fbv);
if(!pdata->args.nosound){
memset(&skel_fba,0,sizeof(skel_fba));
skel_fba.serial_no=enc_data_t->m_ogg_vs.serialno;
skel_fba.nr_header_packet=3;
skel_fba.granule_rate_n=pdata->args.frequency;
skel_fba.granule_rate_d=(ogg_int64_t)1;
skel_fba.start_granule=0;
skel_fba.preroll=2;
skel_fba.granule_shift=0;
add_message_header_field(&skel_fba,
"Content-Type",
"audio/vorbis");
add_fisbone_to_stream(&m_ogg_skel,&skel_fba);
}
while(1){
int result = ogg_stream_flush(&m_ogg_skel, &skel_og_pg);
if(result<0){
fprintf (stderr, "Internal Ogg library error.\n");
exit(2);
}
if(result==0)
break;
fwrite(skel_og_pg.header,1,skel_og_pg.header_len,enc_data_t->fp);
fwrite(skel_og_pg.body,1,skel_og_pg.body_len,enc_data_t->fp);
}
while(1){
int result = ogg_stream_flush(&enc_data_t->m_ogg_ts,
&enc_data_t->m_ogg_pg);
if(result<0){
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
if(result==0)break;
fwrite(enc_data_t->m_ogg_pg.header,1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
fwrite(enc_data_t->m_ogg_pg.body,1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
}
if(!pdata->args.nosound){
while(1){
int result=ogg_stream_flush(&enc_data_t->m_ogg_vs,
&enc_data_t->m_ogg_pg);
if(result<0){
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
if(result==0)break;
fwrite(enc_data_t->m_ogg_pg.header,1,
enc_data_t->m_ogg_pg.header_len,
enc_data_t->fp);
fwrite(enc_data_t->m_ogg_pg.body,1,
enc_data_t->m_ogg_pg.body_len,
enc_data_t->fp);
}
}
//skeleton eos
add_eos_packet_to_stream(&m_ogg_skel);
if(ogg_stream_flush(&m_ogg_skel,&skel_og_pg)<0){
fprintf(stderr,"Internal Ogg library error.\n");
exit(2);
}
fwrite(skel_og_pg.header,1,skel_og_pg.header_len,enc_data_t->fp);
fwrite(skel_og_pg.body,1,skel_og_pg.body_len,enc_data_t->fp);
//theora buffer allocation, if any
if(!buffer_ready){
enc_data_t->yuv.y=(unsigned char *)malloc(enc_data_t->m_th_inf.height*
enc_data_t->m_th_inf.width);
enc_data_t->yuv.u=(unsigned char *)malloc(enc_data_t->m_th_inf.height*
enc_data_t->m_th_inf.width/4);
enc_data_t->yuv.v=(unsigned char *)malloc(enc_data_t->m_th_inf.height*
enc_data_t->m_th_inf.width/4);
enc_data_t->yuv.y_width=enc_data_t->m_th_inf.width;
enc_data_t->yuv.y_height=enc_data_t->m_th_inf.height;
enc_data_t->yuv.y_stride=enc_data_t->m_th_inf.width;
enc_data_t->yuv.uv_width=enc_data_t->m_th_inf.width/2;
enc_data_t->yuv.uv_height=enc_data_t->m_th_inf.height/2;
enc_data_t->yuv.uv_stride=enc_data_t->m_th_inf.width/2;
enc_data_t->x_offset=enc_data_t->m_th_inf.offset_x;
enc_data_t->y_offset=enc_data_t->m_th_inf.offset_y;
}
theora_info_clear(&enc_data_t->m_th_inf);
}
/*****************************************************************************
* OpenEncoder: probe the encoder and return score
*****************************************************************************/
static int OpenEncoder( vlc_object_t *p_this )
{
encoder_t *p_enc = (encoder_t *)p_this;
encoder_sys_t *p_sys;
int i_quality, i_min_bitrate, i_max_bitrate;
ogg_packet header[3];
if( p_enc->fmt_out.i_codec != VLC_CODEC_VORBIS &&
!p_enc->b_force )
{
return VLC_EGENERIC;
}
/* Allocate the memory needed to store the decoder's structure */
if( ( p_sys = (encoder_sys_t *)malloc(sizeof(encoder_sys_t)) ) == NULL )
return VLC_ENOMEM;
p_enc->p_sys = p_sys;
p_enc->pf_encode_audio = Encode;
p_enc->fmt_in.i_codec = VLC_CODEC_FL32;
p_enc->fmt_out.i_codec = VLC_CODEC_VORBIS;
config_ChainParse( p_enc, ENC_CFG_PREFIX, ppsz_enc_options, p_enc->p_cfg );
i_quality = var_GetInteger( p_enc, ENC_CFG_PREFIX "quality" );
if( i_quality > 10 ) i_quality = 10;
if( i_quality < 0 ) i_quality = 0;
if( var_GetBool( p_enc, ENC_CFG_PREFIX "cbr" ) ) i_quality = 0;
i_max_bitrate = var_GetInteger( p_enc, ENC_CFG_PREFIX "max-bitrate" );
i_min_bitrate = var_GetInteger( p_enc, ENC_CFG_PREFIX "min-bitrate" );
/* Initialize vorbis encoder */
vorbis_info_init( &p_sys->vi );
if( i_quality > 0 )
{
/* VBR mode */
if( vorbis_encode_setup_vbr( &p_sys->vi,
p_enc->fmt_in.audio.i_channels, p_enc->fmt_in.audio.i_rate,
i_quality * 0.1 ) )
{
vorbis_info_clear( &p_sys->vi );
free( p_enc->p_sys );
msg_Err( p_enc, "VBR mode initialisation failed" );
return VLC_EGENERIC;
}
/* Do we have optional hard quality restrictions? */
if( i_max_bitrate > 0 || i_min_bitrate > 0 )
{
struct ovectl_ratemanage_arg ai;
vorbis_encode_ctl( &p_sys->vi, OV_ECTL_RATEMANAGE_GET, &ai );
ai.bitrate_hard_min = i_min_bitrate;
ai.bitrate_hard_max = i_max_bitrate;
ai.management_active = 1;
vorbis_encode_ctl( &p_sys->vi, OV_ECTL_RATEMANAGE_SET, &ai );
}
else
{
/* Turn off management entirely */
vorbis_encode_ctl( &p_sys->vi, OV_ECTL_RATEMANAGE_SET, NULL );
}
}
else
{
if( vorbis_encode_setup_managed( &p_sys->vi,
p_enc->fmt_in.audio.i_channels, p_enc->fmt_in.audio.i_rate,
i_min_bitrate > 0 ? i_min_bitrate * 1000: -1,
p_enc->fmt_out.i_bitrate,
i_max_bitrate > 0 ? i_max_bitrate * 1000: -1 ) )
{
vorbis_info_clear( &p_sys->vi );
msg_Err( p_enc, "CBR mode initialisation failed" );
free( p_enc->p_sys );
return VLC_EGENERIC;
}
}
vorbis_encode_setup_init( &p_sys->vi );
/* Add a comment */
vorbis_comment_init( &p_sys->vc);
vorbis_comment_add_tag( &p_sys->vc, "ENCODER", "VLC media player");
/* Set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init( &p_sys->vd, &p_sys->vi );
vorbis_block_init( &p_sys->vd, &p_sys->vb );
/* Create and store headers */
vorbis_analysis_headerout( &p_sys->vd, &p_sys->vc,
&header[0], &header[1], &header[2]);
for( int i = 0; i < 3; i++ )
{
if( xiph_AppendHeaders( &p_enc->fmt_out.i_extra, &p_enc->fmt_out.p_extra,
header[i].bytes, header[i].packet ) )
{
p_enc->fmt_out.i_extra = 0;
p_enc->fmt_out.p_extra = NULL;
}
}
p_sys->i_channels = p_enc->fmt_in.audio.i_channels;
p_sys->i_last_block_size = 0;
p_sys->i_samples_delay = 0;
ConfigureChannelOrder(p_sys->pi_chan_table, p_sys->vi.channels,
p_enc->fmt_in.audio.i_physical_channels, true);
return VLC_SUCCESS;
}
int oe_encode ( oe_enc_opt* opt )
{
ogg_stream_state os;
ogg_page og;
ogg_packet op;
vorbis_dsp_state vd;
vorbis_block vb;
vorbis_info vi;
long samplesdone = 0;
int eos;
long bytes_written = 0;
long packetsdone = 0;
int ret = 0;
vorbis_info_init ( &vi );
if ( opt->quality >= 0.0f )
{
if ( vorbis_encode_init_vbr ( &vi, opt->channels, opt->rate, opt->quality ) )
{
vorbis_info_clear ( &vi );
return 1;
}
}
else
{
if ( vorbis_encode_init (
&vi,
opt->channels,
opt->rate,
opt->max_bitrate > 0 ? opt->max_bitrate * 1000 : -1,
opt->bitrate * 1000,
opt->min_bitrate > 0 ? opt->min_bitrate * 1000 : -1
) )
{
vorbis_info_clear ( &vi );
return 1;
}
}
vorbis_analysis_init ( &vd, &vi );
vorbis_block_init ( &vd, &vb );
ogg_stream_init ( &os, opt->serialno );
ogg_packet header_main;
ogg_packet header_comments;
ogg_packet header_codebooks;
int result;
vorbis_analysis_headerout ( &vd,opt->comments, &header_main, &header_comments, &header_codebooks );
ogg_stream_packetin ( &os, &header_main );
ogg_stream_packetin ( &os, &header_comments );
ogg_stream_packetin ( &os, &header_codebooks );
while ( ( result = ogg_stream_flush ( &os, &og ) ) )
{
if ( !result )
break;
ret = oe_write_page ( &og, opt->out );
if ( ret != og.header_len + og.body_len )
{
ret = 1;
goto cleanup;
}
else
bytes_written += ret;
}
eos = 0;
while ( !eos )
{
float** buffer = vorbis_analysis_buffer ( &vd, READSIZE );
long samples_read = opt->read_samples ( opt->readdata, buffer, READSIZE );
if ( samples_read == 0 )
vorbis_analysis_wrote ( &vd, 0 );
else
{
samplesdone += samples_read;
vorbis_analysis_wrote ( &vd, samples_read );
}
while ( vorbis_analysis_blockout ( &vd, &vb ) == 1 )
{
vorbis_analysis ( &vb, NULL );
vorbis_bitrate_addblock ( &vb );
while ( vorbis_bitrate_flushpacket ( &vd, &op ) )
{
ogg_stream_packetin ( &os,&op );
packetsdone++;
while ( !eos )
{
int result = ogg_stream_pageout ( &os, &og );
if ( !result )
break;
ret = oe_write_page ( &og, opt->out );
if ( ret != og.header_len + og.body_len )
{
ret = 1;
goto cleanup;
}
else
bytes_written += ret;
if ( ogg_page_eos ( &og ) )
eos = 1;
}
}
}
}
ret = 0;
cleanup:
ogg_stream_clear ( &os );
vorbis_block_clear ( &vb );
vorbis_dsp_clear ( &vd );
vorbis_info_clear ( &vi );
return ret;
}
void CDemoVideoRecorder::Init(int Width, int Height, int FPS, int Format, const char *pName)
{
m_pSound = Kernel()->RequestInterface<ISound>();
m_FPS = FPS;
m_ScreenWidth = Width;
m_ScreenHeight = Height;
m_Format = Format;
if (m_Format == IClient::DEMO_RECORD_FORMAT_OGV)
{
ogg_stream_init(&m_TheoraOggStreamState, rand());
ogg_stream_init(&m_VorbisOggStreamState, rand());
char aBuf[1024];
if (str_find_rev(pName, "/"))
str_format(aBuf, sizeof(aBuf), "%s.ogv", str_find_rev(pName, "/"));
else if (str_find_rev(aBuf, "\\"))
str_format(aBuf, sizeof(pName), "%s.ogv", str_find_rev(pName, "\\"));
else
str_format(aBuf, sizeof(aBuf), "%s.ogv", pName);
m_OggFile = io_open(aBuf, IOFLAG_WRITE);
//thread_sleep(10000);
vorbis_info_init(&m_VorbisEncodingInfo);
vorbis_encode_init_vbr(&m_VorbisEncodingInfo, 2, g_Config.m_SndRate, 1.0f); //2 ch - samplerate - quality 1
vorbis_analysis_init(&m_VorbisState, &m_VorbisEncodingInfo);
vorbis_block_init(&m_VorbisState, &m_VorbisBlock);
vorbis_comment_init(&m_VorbisComment);
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&m_VorbisState, &m_VorbisComment, &header, &header_comm, &header_code);
ogg_stream_packetin(&m_VorbisOggStreamState, &header);
ogg_stream_packetin(&m_VorbisOggStreamState, &header_comm);
ogg_stream_packetin(&m_VorbisOggStreamState, &header_code);
th_info_init(&m_TheoraEncodingInfo);
m_TheoraEncodingInfo.frame_width = m_ScreenWidth+15&~0xF;
m_TheoraEncodingInfo.frame_height = m_ScreenHeight+15&~0xF;
m_TheoraEncodingInfo.pic_width = m_ScreenWidth;
m_TheoraEncodingInfo.pic_height = m_ScreenHeight;
m_TheoraEncodingInfo.pic_x = m_TheoraEncodingInfo.frame_width - m_ScreenWidth>>1&~1;
m_TheoraEncodingInfo.pic_y = m_TheoraEncodingInfo.frame_height - m_ScreenHeight>>1&~1;
m_TheoraEncodingInfo.colorspace = TH_CS_UNSPECIFIED;
m_TheoraEncodingInfo.fps_numerator = FPS; //fps
m_TheoraEncodingInfo.fps_denominator = 1;
m_TheoraEncodingInfo.aspect_numerator = -1;
m_TheoraEncodingInfo.aspect_denominator = -1;
m_TheoraEncodingInfo.pixel_fmt = TH_PF_444;
m_TheoraEncodingInfo.target_bitrate = (int)(64870*(ogg_int64_t)48000>>16);
m_TheoraEncodingInfo.quality = 32;
m_TheoraEncodingInfo.keyframe_granule_shift = 0;
m_pThreoraContext = th_encode_alloc(&m_TheoraEncodingInfo);
int arg = TH_RATECTL_CAP_UNDERFLOW;
th_encode_ctl(m_pThreoraContext, TH_ENCCTL_SET_RATE_FLAGS, &arg, sizeof(arg));
th_comment CommentHeader;
ogg_packet OggPacket;
th_comment_init(&CommentHeader);
mem_zero(&OggPacket, sizeof(OggPacket));
//Flush
//Step 1
th_encode_flushheader(m_pThreoraContext, &CommentHeader, &OggPacket); // first header
ogg_stream_packetin(&m_TheoraOggStreamState, &OggPacket);
//
ogg_page OggPage;
ogg_stream_pageout(&m_TheoraOggStreamState, &OggPage);
io_write(m_OggFile, OggPage.header, OggPage.header_len);
io_write(m_OggFile, OggPage.body, OggPage.body_len);
while(1)
{
ogg_page OggPage;
if (ogg_stream_flush(&m_VorbisOggStreamState,&OggPage) == 0)
break;
io_write(m_OggFile, OggPage.header, OggPage.header_len);
io_write(m_OggFile, OggPage.body, OggPage.body_len);
}
while(th_encode_flushheader(m_pThreoraContext, &CommentHeader, &OggPacket))
{
ogg_stream_packetin(&m_TheoraOggStreamState, &OggPacket);
}
ogg_stream_flush(&m_TheoraOggStreamState, &OggPage);
io_write(m_OggFile, OggPage.header, OggPage.header_len);
io_write(m_OggFile, OggPage.body, OggPage.body_len);
}
//________________________________________________
// Init lame encoder
// frequence : Impose frequency , 0 means reuse the incoming fq
// mode : ADM_STEREO etc...
// bitrate : Bitrate in kbps (96,192...)
// return 0 : init failed
// 1 : init succeeded
//_______________________________________________
uint8_t AUDMEncoder_Vorbis::init(ADM_audioEncoderDescriptor *config)
{
int ret;
VORBIS_encoderParam *vorbisConf=(VORBIS_encoderParam *)config->param;
ADM_assert(config->paramSize==sizeof(VORBIS_encoderParam));
ogg_packet header1,header2,header3;
int err;
vorbis_info_init(&VI) ;
switch(vorbisConf->mode)
{
case ADM_VORBIS_VBR:
err=vorbis_encode_init(&VI,
_wavheader->channels,
_wavheader->frequency,
-1, // Max bitrate
config->bitrate*1000, //long nominal_bitrate,
-1 //long min_bitrate))
);
break;
case ADM_VORBIS_QUALITY :
err=vorbis_encode_init_vbr(&VI,
_wavheader->channels,
_wavheader->frequency,
vorbisConf->quality/10
);
break;
default:
ADM_assert(0);
}
if (err!=0)
{
delete (vorbisStruct*)_handle;
_handle = NULL;
printf("[vorbis] init error %d\n",err);
return 0;
}
vorbis_analysis_init(&VD, &VI) ;
vorbis_block_init(&VD, &VB);
vorbis_comment_init(&VC);
vorbis_comment_add_tag(&VC, "encoder", "AVIDEMUX2") ;
vorbis_analysis_headerout(&VD, &VC, &header1,
&header2, &header3);
// Store all headers as extra data
// see ogg vorbis decode for details
// we need 3 packets
_extraSize=header1.bytes+header2.bytes+header3.bytes+3*sizeof(uint32_t);
_extraData=new uint8_t[_extraSize];
uint32_t *ex=(uint32_t *)_extraData;
uint8_t *d;
d=_extraData+sizeof(uint32_t)*3;
ex[0]=header1.bytes;
ex[1]=header2.bytes;
ex[2]=header3.bytes;
memcpy(d,header1.packet,ex[0]);
d+=ex[0];
memcpy(d,header2.packet,ex[1]);
d+=ex[1];
memcpy(d,header3.packet,ex[2]);
vorbis_comment_clear(&VC);
printf("\n[Vorbis]Vorbis encoder initialized\n");
switch(vorbisConf->mode)
{
case ADM_VORBIS_VBR:
printf("[Vorbis]CBR Bitrate:%lu\n",config->bitrate);
break;
case ADM_VORBIS_QUALITY: //FIXME FIXME FIXME
printf("[Vorbis]VBR Quality:%.1f\n",vorbisConf->quality);
break;
default:
ADM_assert(0);
}
printf("[Vorbis]Channels :%lu\n",_wavheader->channels);
printf("[Vorbis]Frequency :%lu\n",_wavheader->frequency);
return 1;
}
void CompressionTool::encodeRaw(const char *rawData, int length, int samplerate, const char *outname, AudioFormat compmode) {
print(" - len=%ld, ch=%d, rate=%d, %dbits", length, (rawAudioType.isStereo ? 2 : 1), samplerate, rawAudioType.bitsPerSample);
#ifdef USE_VORBIS
if (compmode == AUDIO_VORBIS) {
char outputString[256] = "";
int numChannels = (rawAudioType.isStereo ? 2 : 1);
int totalSamples = length / ((rawAudioType.bitsPerSample / 8) * numChannels);
int samplesLeft = totalSamples;
int eos = 0;
int totalBytes = 0;
vorbis_info vi;
vorbis_comment vc;
vorbis_dsp_state vd;
vorbis_block vb;
ogg_stream_state os;
ogg_page og;
ogg_packet op;
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
Common::File outputOgg(outname, "wb");
vorbis_info_init(&vi);
if (oggparms.nominalBitr > 0) {
int result = 0;
/* Input is in kbps, function takes bps */
result = vorbis_encode_setup_managed(&vi, numChannels, samplerate, (oggparms.maxBitr > 0 ? 1000 * oggparms.maxBitr : -1), (1000 * oggparms.nominalBitr), (oggparms.minBitr > 0 ? 1000 * oggparms.minBitr : -1));
if (result == OV_EFAULT) {
vorbis_info_clear(&vi);
error("Error: Internal Logic Fault");
} else if ((result == OV_EINVAL) || (result == OV_EIMPL)) {
vorbis_info_clear(&vi);
error("Error: Invalid bitrate parameters");
}
if (!oggparms.silent) {
sprintf(outputString, "Encoding to\n \"%s\"\nat average bitrate %i kbps (", outname, oggparms.nominalBitr);
if (oggparms.minBitr > 0) {
sprintf(outputString + strlen(outputString), "min %i kbps, ", oggparms.minBitr);
} else {
sprintf(outputString + strlen(outputString), "no min, ");
}
if (oggparms.maxBitr > 0) {
sprintf(outputString + strlen(outputString), "max %i kbps),\nusing full bitrate management engine\nSet optional hard quality restrictions\n", oggparms.maxBitr);
} else {
sprintf(outputString + strlen(outputString), "no max),\nusing full bitrate management engine\nSet optional hard quality restrictions\n");
}
}
} else {
int result = 0;
/* Quality input is -1 - 10, function takes -0.1 through 1.0 */
result = vorbis_encode_setup_vbr(&vi, numChannels, samplerate, oggparms.quality * 0.1f);
if (result == OV_EFAULT) {
vorbis_info_clear(&vi);
error("Internal Logic Fault");
} else if ((result == OV_EINVAL) || (result == OV_EIMPL)) {
vorbis_info_clear(&vi);
error("Invalid bitrate parameters");
}
if (!oggparms.silent) {
sprintf(outputString, "Encoding to\n \"%s\"\nat quality %2.2f", outname, oggparms.quality);
}
if ((oggparms.minBitr > 0) || (oggparms.maxBitr > 0)) {
struct ovectl_ratemanage_arg extraParam;
vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_GET, &extraParam);
extraParam.bitrate_hard_min = (oggparms.minBitr > 0 ? (1000 * oggparms.minBitr) : -1);
extraParam.bitrate_hard_max = (oggparms.maxBitr > 0 ? (1000 * oggparms.maxBitr) : -1);
extraParam.management_active = 1;
vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_SET, &extraParam);
if (!oggparms.silent) {
sprintf(outputString + strlen(outputString), " using constrained VBR (");
if (oggparms.minBitr != -1) {
sprintf(outputString + strlen(outputString), "min %i kbps, ", oggparms.minBitr);
} else {
sprintf(outputString + strlen(outputString), "no min, ");
}
if (oggparms.maxBitr != -1) {
sprintf(outputString + strlen(outputString), "max %i kbps)\nSet optional hard quality restrictions\n", oggparms.maxBitr);
} else {
sprintf(outputString + strlen(outputString), "no max)\nSet optional hard quality restrictions\n");
}
}
} else {
sprintf(outputString + strlen(outputString), "\n");
}
}
puts(outputString);
vorbis_encode_setup_init(&vi);
vorbis_comment_init(&vc);
vorbis_analysis_init(&vd, &vi);
vorbis_block_init(&vd, &vb);
ogg_stream_init(&os, 0);
vorbis_analysis_headerout(&vd, &vc, &header, &header_comm, &header_code);
ogg_stream_packetin(&os, &header);
ogg_stream_packetin(&os, &header_comm);
ogg_stream_packetin(&os, &header_code);
while (!eos) {
int result = ogg_stream_flush(&os,&og);
if (result == 0) {
break;
}
outputOgg.write(og.header, og.header_len);
outputOgg.write(og.body, og.body_len);
}
while (!eos) {
int numSamples = ((samplesLeft < 2048) ? samplesLeft : 2048);
float **buffer = vorbis_analysis_buffer(&vd, numSamples);
/* We must tell the encoder that we have reached the end of the stream */
if (numSamples == 0) {
vorbis_analysis_wrote(&vd, 0);
} else {
/* Adapted from oggenc 1.1.1 */
if (rawAudioType.bitsPerSample == 8) {
const byte *rawDataUnsigned = (const byte *)rawData;
for (int i = 0; i < numSamples; i++) {
for (int j = 0; j < numChannels; j++) {
buffer[j][i] = ((int)(rawDataUnsigned[i * numChannels + j]) - 128) / 128.0f;
}
}
} else if (rawAudioType.bitsPerSample == 16) {
if (rawAudioType.isLittleEndian) {
for (int i = 0; i < numSamples; i++) {
for (int j = 0; j < numChannels; j++) {
buffer[j][i] = ((rawData[(i * 2 * numChannels) + (2 * j) + 1] << 8) | (rawData[(i * 2 * numChannels) + (2 * j)] & 0xff)) / 32768.0f;
}
}
} else {
for (int i = 0; i < numSamples; i++) {
for (int j = 0; j < numChannels; j++) {
buffer[j][i] = ((rawData[(i * 2 * numChannels) + (2 * j)] << 8) | (rawData[(i * 2 * numChannels) + (2 * j) + 1] & 0xff)) / 32768.0f;
}
}
}
}
vorbis_analysis_wrote(&vd, numSamples);
}
while (vorbis_analysis_blockout(&vd, &vb) == 1) {
vorbis_analysis(&vb, NULL);
vorbis_bitrate_addblock(&vb);
while (vorbis_bitrate_flushpacket(&vd, &op)) {
ogg_stream_packetin(&os, &op);
while (!eos) {
int result = ogg_stream_pageout(&os, &og);
if (result == 0) {
break;
}
totalBytes += outputOgg.write(og.header, og.header_len);
totalBytes += outputOgg.write(og.body, og.body_len);
if (ogg_page_eos(&og)) {
eos = 1;
}
}
}
}
rawData += 2048 * (rawAudioType.bitsPerSample / 8) * numChannels;
samplesLeft -= 2048;
}
ogg_stream_clear(&os);
vorbis_block_clear(&vb);
vorbis_dsp_clear(&vd);
vorbis_info_clear(&vi);
if (!oggparms.silent) {
print("\nDone encoding file \"%s\"", outname);
print("\n\tFile length: %dm %ds", (int)(totalSamples / samplerate / 60), (totalSamples / samplerate % 60));
print("\tAverage bitrate: %.1f kb/s\n", (8.0 * (double)totalBytes / 1000.0) / ((double)totalSamples / (double)samplerate));
}
}
#endif
#ifdef USE_FLAC
if (compmode == AUDIO_FLAC) {
int i;
int numChannels = (rawAudioType.isStereo ? 2 : 1);
int samplesPerChannel = length / ((rawAudioType.bitsPerSample / 8) * numChannels);
FLAC__StreamEncoder *encoder;
FLAC__StreamEncoderInitStatus initStatus;
FLAC__int32 *flacData;
flacData = (FLAC__int32 *)malloc(samplesPerChannel * numChannels * sizeof(FLAC__int32));
if (rawAudioType.bitsPerSample == 8) {
for (i = 0; i < samplesPerChannel * numChannels; i++) {
FLAC__uint8 *rawDataUnsigned;
rawDataUnsigned = (FLAC__uint8 *)rawData;
flacData[i] = (FLAC__int32)rawDataUnsigned[i] - 0x80;
}
} else if (rawAudioType.bitsPerSample == 16) {
/* The rawData pointer is an 8-bit char so we must create a new pointer to access 16-bit samples */
FLAC__int16 *rawData16;
rawData16 = (FLAC__int16 *)rawData;
for (i = 0; i < samplesPerChannel * numChannels; i++) {
flacData[i] = (FLAC__int32)rawData16[i];
}
}
if (!flacparms.silent) {
print("Encoding to\n \"%s\"\nat compression level %d using blocksize %d\n", outname, flacparms.compressionLevel, flacparms.blocksize);
}
encoder = FLAC__stream_encoder_new();
FLAC__stream_encoder_set_bits_per_sample(encoder, rawAudioType.bitsPerSample);
FLAC__stream_encoder_set_blocksize(encoder, flacparms.blocksize);
FLAC__stream_encoder_set_channels(encoder, numChannels);
FLAC__stream_encoder_set_compression_level(encoder, flacparms.compressionLevel);
FLAC__stream_encoder_set_sample_rate(encoder, samplerate);
FLAC__stream_encoder_set_streamable_subset(encoder, false);
FLAC__stream_encoder_set_total_samples_estimate(encoder, samplesPerChannel);
FLAC__stream_encoder_set_verify(encoder, flacparms.verify);
initStatus = FLAC__stream_encoder_init_file(encoder, outname, NULL, NULL);
if (initStatus != FLAC__STREAM_ENCODER_INIT_STATUS_OK) {
char buf[2048];
sprintf(buf, "Error in FLAC encoder. (check the parameters)\nExact error was:%s", FLAC__StreamEncoderInitStatusString[initStatus]);
free(flacData);
throw ToolException(buf);
} else {
FLAC__stream_encoder_process_interleaved(encoder, flacData, samplesPerChannel);
}
FLAC__stream_encoder_finish(encoder);
FLAC__stream_encoder_delete(encoder);
free(flacData);
if (!flacparms.silent) {
print("\nDone encoding file \"%s\"", outname);
print("\n\tFile length: %dm %ds\n", (int)(samplesPerChannel / samplerate / 60), (samplesPerChannel / samplerate % 60));
}
}
#endif
}
static int encode_ogg (cdrom_drive *drive, rip_opts_s *rip_opts,
text_tag_s *text_tag, int track,
int tracktot, char *filename, char **filenames)
{
ogg_stream_state os;
ogg_page og;
ogg_packet op;
vorbis_dsp_state vd;
vorbis_block vb;
vorbis_info vi;
long samplesdone = 0;
int sector = 0, last_sector = 0;
long bytes_written = 0, packetsdone = 0;
double time_elapsed = 0.0;
int ret = 0;
time_t *timer;
double time;
int serialno = rand ();
vorbis_comment vc;
long total_samples_per_channel = 0;
int channels = 2;
int eos = 0;
long rate = 44100;
FILE *out = fopen (filename, "w+");
timer = timer_start ();
if (!rip_opts->managed && (rip_opts->min_bitrate > 0 || rip_opts->max_bitrate > 0)) {
log_msg ("Min or max bitrate requires managed", FL, FN, LN);
return -1;
}
if (rip_opts->bitrate < 0 && rip_opts->min_bitrate < 0 && rip_opts->max_bitrate < 0) {
rip_opts->quality_set = 1;
}
start_func (filename, rip_opts->bitrate, rip_opts->quality, rip_opts->quality_set,
rip_opts->managed, rip_opts->min_bitrate, rip_opts->max_bitrate);
vorbis_info_init (&vi);
if (rip_opts->quality_set > 0) {
if (vorbis_encode_setup_vbr (&vi, channels, rate, rip_opts->quality)) {
log_msg ("Couldn't initialize vorbis_info", FL, FN, LN);
vorbis_info_clear (&vi);
return -1;
}
/* two options here, max or min bitrate */
if (rip_opts->max_bitrate > 0 || rip_opts->min_bitrate > 0) {
struct ovectl_ratemanage_arg ai;
vorbis_encode_ctl (&vi, OV_ECTL_RATEMANAGE_GET, &ai);
ai.bitrate_hard_min = rip_opts->min_bitrate;
ai.bitrate_hard_max = rip_opts->max_bitrate;
ai.management_active = 1;
vorbis_encode_ctl (&vi, OV_ECTL_RATEMANAGE_SET, &ai);
}
} else {
if (vorbis_encode_setup_managed (&vi, channels, rate,
rip_opts->max_bitrate > 0 ? rip_opts->max_bitrate * 1000 : -1,
rip_opts->bitrate * 1000,
rip_opts->min_bitrate > 0 ? rip_opts->min_bitrate * 1000 : -1)) {
log_msg ("Mode init failed, encode setup managed", FL, FN, LN);
vorbis_info_clear (&vi);
return -1;
}
}
if (rip_opts->managed && rip_opts->bitrate < 0) {
vorbis_encode_ctl (&vi, OV_ECTL_RATEMANAGE_AVG, NULL);
} else if (!rip_opts->managed) {
vorbis_encode_ctl (&vi, OV_ECTL_RATEMANAGE_SET, NULL);
}
/* set advanced encoder options */
vorbis_encode_setup_init (&vi);
vorbis_analysis_init (&vd, &vi);
vorbis_block_init (&vd, &vb);
ogg_stream_init (&os, serialno);
{
ogg_packet header_main;
ogg_packet header_comments;
ogg_packet header_codebooks;
int result;
char buf[32];
vorbis_comment_init (&vc);
vorbis_comment_add_tag (&vc, "title", text_tag->songname);
vorbis_comment_add_tag (&vc, "artist", text_tag->artistname);
vorbis_comment_add_tag (&vc, "album", text_tag->albumname);
vorbis_comment_add_tag (&vc, "genre", text_tag->genre);
snprintf (buf, 32, "%d", text_tag->year);
vorbis_comment_add_tag (&vc, "date", buf);
snprintf (buf, 32, "%02d", text_tag->track);
vorbis_comment_add_tag (&vc, "tracknumber", buf);
vorbis_analysis_headerout (&vd, &vc, &header_main, &header_comments, &header_codebooks);
ogg_stream_packetin (&os, &header_main);
ogg_stream_packetin (&os, &header_comments);
ogg_stream_packetin (&os, &header_codebooks);
while ((result = ogg_stream_flush (&os, &og))) {
if (result == 0)
break;
ret = write_page (&og, out);
if (ret != og.header_len + og.body_len) {
log_msg ("Failed writing data to output stream", FL, FN, LN);
ret = -1;
}
}
sector = cdda_track_firstsector (drive, track);
last_sector = cdda_track_lastsector (drive, track);
total_samples_per_channel = (last_sector - sector) * (CD_FRAMESAMPLES / 2);
int eos = 0;
while (!eos) {
signed char *buffer = (signed char *)malloc (CD_FRAMESIZE_RAW * READ_SECTORS);
//use this variable as a s**t
long sectors_read = last_sector - sector;
if (sectors_read > READ_SECTORS)
sectors_read = READ_SECTORS;
sectors_read = cdda_read (drive, (signed char *)buffer, sector, sectors_read);
int i;
if (sectors_read == 0) {
vorbis_analysis_wrote (&vd, 0);
} else {
float **vorbbuf = vorbis_analysis_buffer (&vd, CD_FRAMESIZE_RAW * sectors_read);
for (i = 0; i < (CD_FRAMESIZE_RAW * sectors_read) / 4; i++) {
vorbbuf[0][i] = ((buffer[i * 4 + 1] << 8) | (0x00ff&(int)buffer[i * 4])) / 32768.f;
vorbbuf[1][i] = ((buffer[i * 4 + 3] << 8) | (0x00ff&(int)buffer[i * 4 + 2])) / 32768.f;
}
int samples_read = sectors_read * (CD_FRAMESAMPLES / 2);
samplesdone += samples_read;
// progress every 60 pages
if (packetsdone >= 60) {
packetsdone = 0;
time = timer_time (timer);
update_statistics (total_samples_per_channel, samplesdone, time, track,
tracktot, 0, filenames);
}
vorbis_analysis_wrote (&vd, i);
}
free (buffer);
sector += sectors_read;
while (vorbis_analysis_blockout (&vd, &vb) == 1) {
vorbis_analysis (&vb, &op);
vorbis_bitrate_addblock (&vb);
while (vorbis_bitrate_flushpacket (&vd, &op)) {
ogg_stream_packetin (&os, &op);
packetsdone++;
while (!eos) {
int result = ogg_stream_pageout (&os, &og);
if (result == 0) {
break;
}
ret = write_page (&og, out);
if (ret != og.header_len + og.body_len) {
log_msg ("Failed writing data to output stream", FL, FN, LN);
ret = -1;
} else
bytes_written += ret;
if (ogg_page_eos (&og)) {
eos = 1;
}
}
}
}
}
}
ret = 0;
update_statistics (total_samples_per_channel, samplesdone, time, track,
tracktot, 0, filenames);
ogg_stream_clear (&os);
vorbis_block_clear (&vb);
vorbis_dsp_clear (&vd);
vorbis_comment_clear (&vc);
vorbis_info_clear (&vi);
vorbis_comment_clear (&vc);
time_elapsed = timer_time (timer);
end_func (time_elapsed, rate, samplesdone, bytes_written);
timer_clear (timer);
fclose (out);
return ret;
}
int ExportOGG::Export(AudacityProject *project,
int numChannels,
const wxString &fName,
bool selectionOnly,
double t0,
double t1,
MixerSpec *mixerSpec,
const Tags *metadata,
int WXUNUSED(subformat))
{
double rate = project->GetRate();
const TrackList *tracks = project->GetTracks();
double quality = (gPrefs->Read(wxT("/FileFormats/OggExportQuality"), 50)/(float)100.0);
wxLogNull logNo; // temporarily disable wxWidgets error messages
int updateResult = eProgressSuccess;
int eos = 0;
FileIO outFile(fName, FileIO::Output);
if (!outFile.IsOpened()) {
wxMessageBox(_("Unable to open target file for writing"));
return false;
}
// All the Ogg and Vorbis encoding data
ogg_stream_state stream;
ogg_page page;
ogg_packet packet;
vorbis_info info;
vorbis_comment comment;
vorbis_dsp_state dsp;
vorbis_block block;
// Encoding setup
vorbis_info_init(&info);
vorbis_encode_init_vbr(&info, numChannels, int(rate + 0.5), quality);
// Retrieve tags
if (!FillComment(project, &comment, metadata)) {
return false;
}
// Set up analysis state and auxiliary encoding storage
vorbis_analysis_init(&dsp, &info);
vorbis_block_init(&dsp, &block);
// Set up packet->stream encoder. According to encoder example,
// a random serial number makes it more likely that you can make
// chained streams with concatenation.
srand(time(NULL));
ogg_stream_init(&stream, rand());
// First we need to write the required headers:
// 1. The Ogg bitstream header, which contains codec setup params
// 2. The Vorbis comment header
// 3. The bitstream codebook.
//
// After we create those our responsibility is complete, libvorbis will
// take care of any other ogg bistream constraints (again, according
// to the example encoder source)
ogg_packet bitstream_header;
ogg_packet comment_header;
ogg_packet codebook_header;
vorbis_analysis_headerout(&dsp, &comment, &bitstream_header, &comment_header,
&codebook_header);
// Place these headers into the stream
ogg_stream_packetin(&stream, &bitstream_header);
ogg_stream_packetin(&stream, &comment_header);
ogg_stream_packetin(&stream, &codebook_header);
// Flushing these headers now guarentees that audio data will
// start on a NEW page, which apparently makes streaming easier
while (ogg_stream_flush(&stream, &page)) {
outFile.Write(page.header, page.header_len);
outFile.Write(page.body, page.body_len);
}
const WaveTrackConstArray waveTracks =
tracks->GetWaveTrackConstArray(selectionOnly, false);
{
auto mixer = CreateMixer(waveTracks,
tracks->GetTimeTrack(),
t0, t1,
numChannels, SAMPLES_PER_RUN, false,
rate, floatSample, true, mixerSpec);
ProgressDialog progress(wxFileName(fName).GetName(),
selectionOnly ?
_("Exporting the selected audio as Ogg Vorbis") :
_("Exporting the entire project as Ogg Vorbis"));
while (updateResult == eProgressSuccess && !eos) {
float **vorbis_buffer = vorbis_analysis_buffer(&dsp, SAMPLES_PER_RUN);
sampleCount samplesThisRun = mixer->Process(SAMPLES_PER_RUN);
if (samplesThisRun == 0) {
// Tell the library that we wrote 0 bytes - signalling the end.
vorbis_analysis_wrote(&dsp, 0);
}
else {
for (int i = 0; i < numChannels; i++) {
float *temp = (float *)mixer->GetBuffer(i);
memcpy(vorbis_buffer[i], temp, sizeof(float)*SAMPLES_PER_RUN);
}
// tell the encoder how many samples we have
vorbis_analysis_wrote(&dsp, samplesThisRun);
}
// I don't understand what this call does, so here is the comment
// from the example, verbatim:
//
// vorbis does some data preanalysis, then divvies up blocks
// for more involved (potentially parallel) processing. Get
// a single block for encoding now
while (vorbis_analysis_blockout(&dsp, &block) == 1) {
// analysis, assume we want to use bitrate management
vorbis_analysis(&block, NULL);
vorbis_bitrate_addblock(&block);
while (vorbis_bitrate_flushpacket(&dsp, &packet)) {
// add the packet to the bitstream
ogg_stream_packetin(&stream, &packet);
// From vorbis-tools-1.0/oggenc/encode.c:
// If we've gone over a page boundary, we can do actual output,
// so do so (for however many pages are available).
while (!eos) {
int result = ogg_stream_pageout(&stream, &page);
if (!result) {
break;
}
outFile.Write(page.header, page.header_len);
outFile.Write(page.body, page.body_len);
if (ogg_page_eos(&page)) {
eos = 1;
}
}
}
}
updateResult = progress.Update(mixer->MixGetCurrentTime() - t0, t1 - t0);
}
}
ogg_stream_clear(&stream);
vorbis_block_clear(&block);
vorbis_dsp_clear(&dsp);
vorbis_info_clear(&info);
vorbis_comment_clear(&comment);
outFile.Close();
return updateResult;
}
// Just create the Quicktime objects since this routine is also called
// for reopening.
int FileVorbis::open_file(int rd, int wr)
{
int result = 0;
this->rd = rd;
this->wr = wr;
//printf("FileVorbis::open_file 1\n");
if(rd)
{
//printf("FileVorbis::open_file 1\n");
if(!(fd = fopen(asset->path, "rb")))
{
eprintf("Error while opening \"%s\" for reading. \n%m\n", asset->path);
result = 1;
}
else
{
//printf("FileVorbis::open_file 2 %p %p\n", fd, vf);
if(ov_open(fd, &vf, NULL, 0) < 0)
{
eprintf("Invalid bitstream in %s\n", asset->path);
result = 1;
}
else
{
//printf("FileVorbis::open_file 1\n");
vorbis_info *vi = ov_info(&vf, -1);
asset->channels = vi->channels;
if(!asset->sample_rate)
asset->sample_rate = vi->rate;
//printf("FileVorbis::open_file 1\n");
asset->audio_length = ov_pcm_total(&vf,-1);
//printf("FileVorbis::open_file 1\n");
asset->audio_data = 1;
// printf("FileVorbis::open_file 1 %d %d %d\n",
// asset->channels,
// asset->sample_rate,
// asset->audio_length);
}
}
}
if(wr)
{
if(!(fd = fopen(asset->path, "wb")))
{
eprintf("Error while opening \"%s\" for writing. \n%m\n", asset->path);
result = 1;
}
else
{
vorbis_info_init(&vi);
if(!asset->vorbis_vbr)
result = vorbis_encode_init(&vi,
asset->channels,
asset->sample_rate,
asset->vorbis_max_bitrate,
asset->vorbis_bitrate,
asset->vorbis_min_bitrate);
else
{
result = vorbis_encode_setup_managed(&vi,
asset->channels,
asset->sample_rate,
asset->vorbis_max_bitrate,
asset->vorbis_bitrate,
asset->vorbis_min_bitrate);
result |= vorbis_encode_ctl(&vi, OV_ECTL_RATEMANAGE_AVG, NULL);
result |= vorbis_encode_setup_init(&vi);
}
if(!result)
{
vorbis_analysis_init(&vd, &vi);
vorbis_block_init(&vd, &vb);
vorbis_comment_init(&vc);
srand(time(NULL));
ogg_stream_init(&os, rand());
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd,
&vc,
&header,
&header_comm,
&header_code);
ogg_stream_packetin(&os,
&header);
ogg_stream_packetin(&os,
&header_comm);
ogg_stream_packetin(&os,
&header_code);
while(1)
{
int result = ogg_stream_flush(&os, &og);
if(result == 0) break;
fwrite(og.header, 1, og.header_len, fd);
fwrite(og.body, 1, og.body_len, fd);
}
}
}
}
//printf("FileVorbis::open_file 2\n");
return result;
}
bool Start(void *ctx, int iInChannels, int iInRate, int iInBits,
const char* title, const char* artist,
const char* albumartist, const char* album,
const char* year, const char* track, const char* genre,
const char* comment, int iTrackLength)
{
ogg_context *context = (ogg_context *)ctx;
if (!context || !context->callbacks.write)
return false;
// we accept only 2 ch 16bit atm
if (iInChannels != 2 || iInBits != 16)
return false;
if (preset == -1)
vorbis_encode_init(&context->vorbisInfo, iInChannels, iInRate, -1, bitrate*1000, -1);
else
vorbis_encode_init_vbr(&context->vorbisInfo, iInChannels, iInRate, float(preset)/10.0f);
/* add a comment */
vorbis_comment comm;
vorbis_comment_init(&comm);
vorbis_comment_add_tag(&comm, (char*)"comment", (char*)comment);
vorbis_comment_add_tag(&comm, (char*)"artist", (char*)artist);
vorbis_comment_add_tag(&comm, (char*)"title", (char*)title);
vorbis_comment_add_tag(&comm, (char*)"album", (char*)album);
vorbis_comment_add_tag(&comm, (char*)"albumartist", (char*)albumartist);
vorbis_comment_add_tag(&comm, (char*)"genre", (char*)genre);
vorbis_comment_add_tag(&comm, (char*)"tracknumber", (char*)track);
vorbis_comment_add_tag(&comm, (char*)"date", (char*)year);
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init(&context->vorbisDspState, &context->vorbisInfo);
vorbis_block_init(&context->vorbisDspState, &context->vorbisBlock);
/* set up our packet->stream encoder */
/* pick a random serial number; that way we can more likely build
chained streams just by concatenation */
srand((unsigned int)time(NULL));
ogg_stream_init(&context->oggStreamState, rand());
/* write out the metadata */
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
ogg_page page;
vorbis_analysis_headerout(&context->vorbisDspState, &comm,
&header, &header_comm, &header_code);
ogg_stream_packetin(&context->oggStreamState, &header);
ogg_stream_packetin(&context->oggStreamState, &header_comm);
ogg_stream_packetin(&context->oggStreamState, &header_code);
while (1)
{
/* This ensures the actual
* audio data will start on a new page, as per spec
*/
int result = ogg_stream_flush(&context->oggStreamState, &page);
if (result == 0)
break;
context->callbacks.write(context->callbacks.opaque, page.header, page.header_len);
context->callbacks.write(context->callbacks.opaque, page.body, page.body_len);
}
vorbis_comment_clear(&comm);
context->inited = true;
return true;
}
//________________________________________________
// Init Vorbis encoder
// frequence : Impose frequency , 0 means reuse the incoming fq
// mode : VBR CBR
// bitrate for CBR : bitrate in kbps (96,192...)
// for VBR : quality (0, 1, 2, ... 11)
//
// return 0 : init failed
// 1 : init succeeded
//_______________________________________________
uint8_t AVDMProcessAudio_Vorbis::init( uint32_t bitrate, uint8_t mode)
{
ogg_packet header1,header2,header3;
int err;
vorbis_info_init(&VI) ;
if (mode==0) { //VBR
err=vorbis_encode_init_vbr(&VI,
_wavheader->channels,
_wavheader->frequency,
((float)bitrate-1)/10
);
} else { //CBR
err=vorbis_encode_init(&VI,
_wavheader->channels,
_wavheader->frequency,
-1, // Max bitrate
bitrate*1000, //long nominal_bitrate,
-1 //long min_bitrate))
);
}
if (err!=0) {
printf("vorbis init error\n");
return 0;
}
vorbis_analysis_init(&VD, &VI) ;
vorbis_block_init(&VD, &VB);
vorbis_comment_init(&VC);
vorbis_comment_add_tag(&VC, "encoder", "AVIDEMUX2") ;
vorbis_analysis_headerout(&VD, &VC, &header1,
&header2, &header3);
// Store all headers as extra data
// see ogg vorbis decode for details
// we need 3 packets
_extraLen=header1.bytes+header2.bytes+header3.bytes+3*sizeof(uint32_t);
_extraData=new uint8_t[_extraLen];
uint32_t *ex=(uint32_t *)_extraData;
uint8_t *d;
d=_extraData+sizeof(uint32_t)*3;
ex[0]=header1.bytes;
ex[1]=header2.bytes;
ex[2]=header3.bytes;
memcpy(d,header1.packet,ex[0]);
d+=ex[0];
memcpy(d,header2.packet,ex[1]);
d+=ex[1];
memcpy(d,header3.packet,ex[2]);
vorbis_comment_clear(&VC);
printf("\nVorbis encoder initialized\n");
if (mode==0)
printf("VBR Quality:%i\n",bitrate-1);
else
printf("CBR Bitrate:%lu\n",bitrate);
printf("Channels:%lu\n",_wavheader->channels);
printf("Frequenc:%lu\n",_wavheader->frequency);
return 1;
}
S32 encode_vorbis_file(const std::string& in_fname, const std::string& out_fname)
{
#define READ_BUFFER 1024
unsigned char readbuffer[READ_BUFFER*4+44]; /* out of the data segment, not the stack */ /*Flawfinder: ignore*/
ogg_stream_state os; /* take physical pages, weld into a logical stream of packets */
ogg_page og; /* one Ogg bitstream page. Vorbis packets are inside */
ogg_packet op; /* one raw packet of data for decode */
vorbis_info vi; /* struct that stores all the static vorbis bitstream settings */
vorbis_comment vc; /* struct that stores all the user comments */
vorbis_dsp_state vd; /* central working state for the packet->PCM decoder */
vorbis_block vb; /* local working space for packet->PCM decode */
int eos=0;
int result;
U16 num_channels = 0;
U32 sample_rate = 0;
U32 bits_per_sample = 0;
S32 format_error = 0;
std::string error_msg;
if ((format_error = check_for_invalid_wav_formats(in_fname, error_msg)))
{
llwarns << error_msg << ": " << in_fname << llendl;
return(format_error);
}
#if 1
unsigned char wav_header[44]; /*Flawfinder: ignore*/
S32 data_left = 0;
LLAPRFile infile ;
infile.open(in_fname,LL_APR_RB);
if (!infile.getFileHandle())
{
llwarns << "Couldn't open temporary ogg file for writing: " << in_fname
<< llendl;
return(LLVORBISENC_SOURCE_OPEN_ERR);
}
LLAPRFile outfile ;
outfile.open(out_fname,LL_APR_WPB);
if (!outfile.getFileHandle())
{
llwarns << "Couldn't open upload sound file for reading: " << in_fname
<< llendl;
return(LLVORBISENC_DEST_OPEN_ERR);
}
// parse the chunks
U32 chunk_length = 0;
U32 file_pos = 12; // start at the first chunk (usually fmt but not always)
while (infile.eof() != APR_EOF)
{
infile.seek(APR_SET,file_pos);
infile.read(wav_header, 44);
chunk_length = ((U32) wav_header[7] << 24)
+ ((U32) wav_header[6] << 16)
+ ((U32) wav_header[5] << 8)
+ wav_header[4];
// llinfos << "chunk found: '" << wav_header[0] << wav_header[1] << wav_header[2] << wav_header[3] << "'" << llendl;
if (!(strncmp((char *)&(wav_header[0]),"fmt ",4)))
{
num_channels = ((U16) wav_header[11] << 8) + wav_header[10];
sample_rate = ((U32) wav_header[15] << 24)
+ ((U32) wav_header[14] << 16)
+ ((U32) wav_header[13] << 8)
+ wav_header[12];
bits_per_sample = ((U16) wav_header[23] << 8) + wav_header[22];
}
else if (!(strncmp((char *)&(wav_header[0]),"data",4)))
{
infile.seek(APR_SET,file_pos+8);
// leave the file pointer at the beginning of the data chunk data
data_left = chunk_length;
break;
}
file_pos += (chunk_length + 8);
chunk_length = 0;
}
/********** Encode setup ************/
/* choose an encoding mode */
/* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
vorbis_info_init(&vi);
// always encode to mono
// SL-52913 & SL-53779 determined this quality level to be our 'good
// enough' general-purpose quality level with a nice low bitrate.
// Equivalent to oggenc -q0.5
F32 quality = 0.05f;
// quality = (bitrate==128000 ? 0.4f : 0.1);
// if (vorbis_encode_init(&vi, /* num_channels */ 1 ,sample_rate, -1, bitrate, -1))
if (vorbis_encode_init_vbr(&vi, /* num_channels */ 1 ,sample_rate, quality))
// if (vorbis_encode_setup_managed(&vi,1,sample_rate,-1,bitrate,-1) ||
// vorbis_encode_ctl(&vi,OV_ECTL_RATEMANAGE_AVG,NULL) ||
// vorbis_encode_setup_init(&vi))
{
llwarns << "unable to initialize vorbis codec at quality " << quality << llendl;
// llwarns << "unable to initialize vorbis codec at bitrate " << bitrate << llendl;
return(LLVORBISENC_DEST_OPEN_ERR);
}
/* add a comment */
vorbis_comment_init(&vc);
// vorbis_comment_add(&vc,"Linden");
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init(&vd,&vi);
vorbis_block_init(&vd,&vb);
/* set up our packet->stream encoder */
/* pick a random serial number; that way we can more likely build
chained streams just by concatenation */
ogg_stream_init(&os, ll_rand());
/* Vorbis streams begin with three headers; the initial header (with
most of the codec setup parameters) which is mandated by the Ogg
bitstream spec. The second header holds any comment fields. The
third header holds the bitstream codebook. We merely need to
make the headers, then pass them to libvorbis one at a time;
libvorbis handles the additional Ogg bitstream constraints */
{
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd,&vc,&header,&header_comm,&header_code);
ogg_stream_packetin(&os,&header); /* automatically placed in its own
page */
ogg_stream_packetin(&os,&header_comm);
ogg_stream_packetin(&os,&header_code);
/* We don't have to write out here, but doing so makes streaming
* much easier, so we do, flushing ALL pages. This ensures the actual
* audio data will start on a new page
*/
while(!eos){
int result=ogg_stream_flush(&os,&og);
if(result==0)break;
outfile.write(og.header, og.header_len);
outfile.write(og.body, og.body_len);
}
}
while(!eos)
{
long bytes_per_sample = bits_per_sample/8;
long bytes=(long)infile.read(readbuffer,llclamp((S32)(READ_BUFFER*num_channels*bytes_per_sample),0,data_left)); /* stereo hardwired here */
if (bytes==0)
{
/* end of file. this can be done implicitly in the mainline,
but it's easier to see here in non-clever fashion.
Tell the library we're at end of stream so that it can handle
the last frame and mark end of stream in the output properly */
vorbis_analysis_wrote(&vd,0);
// eos = 1;
}
else
{
long i;
long samples;
int temp;
data_left -= bytes;
/* data to encode */
/* expose the buffer to submit data */
float **buffer=vorbis_analysis_buffer(&vd,READ_BUFFER);
i = 0;
samples = bytes / (num_channels * bytes_per_sample);
if (num_channels == 2)
{
if (bytes_per_sample == 2)
{
/* uninterleave samples */
for(i=0; i<samples ;i++)
{
temp = ((signed char *)readbuffer)[i*4+1]; /*Flawfinder: ignore*/
temp += ((signed char *)readbuffer)[i*4+3]; /*Flawfinder: ignore*/
temp <<= 8;
temp += readbuffer[i*4];
temp += readbuffer[i*4+2];
buffer[0][i] = ((float)temp) / 65536.f;
}
}
else // presume it's 1 byte per which is unsigned (F#@%ing wav "standard")
{
/* uninterleave samples */
for(i=0; i<samples ;i++)
{
temp = readbuffer[i*2+0];
temp += readbuffer[i*2+1];
temp -= 256;
buffer[0][i] = ((float)temp) / 256.f;
}
}
}
else if (num_channels == 1)
{
if (bytes_per_sample == 2)
{
for(i=0; i < samples ;i++)
{
temp = ((signed char*)readbuffer)[i*2+1];
temp <<= 8;
temp += readbuffer[i*2];
buffer[0][i] = ((float)temp) / 32768.f;
}
}
else // presume it's 1 byte per which is unsigned (F#@%ing wav "standard")
{
for(i=0; i < samples ;i++)
{
temp = readbuffer[i];
temp -= 128;
buffer[0][i] = ((float)temp) / 128.f;
}
}
}
/* tell the library how much we actually submitted */
vorbis_analysis_wrote(&vd,i);
}
/* vorbis does some data preanalysis, then divvies up blocks for
more involved (potentially parallel) processing. Get a single
block for encoding now */
while(vorbis_analysis_blockout(&vd,&vb)==1)
{
/* analysis */
/* Do the main analysis, creating a packet */
vorbis_analysis(&vb, NULL);
vorbis_bitrate_addblock(&vb);
while(vorbis_bitrate_flushpacket(&vd, &op))
{
/* weld the packet into the bitstream */
ogg_stream_packetin(&os,&op);
/* write out pages (if any) */
while(!eos)
{
result = ogg_stream_pageout(&os,&og);
if(result==0)
break;
outfile.write(og.header, og.header_len);
outfile.write(og.body, og.body_len);
/* this could be set above, but for illustrative purposes, I do
it here (to show that vorbis does know where the stream ends) */
if(ogg_page_eos(&og))
eos=1;
}
}
}
}
/* clean up and exit. vorbis_info_clear() must be called last */
ogg_stream_clear(&os);
vorbis_block_clear(&vb);
vorbis_dsp_clear(&vd);
vorbis_comment_clear(&vc);
vorbis_info_clear(&vi);
/* ogg_page and ogg_packet structs always point to storage in
libvorbis. They're never freed or manipulated directly */
// fprintf(stderr,"Vorbis encoding: Done.\n");
llinfos << "Vorbis encoding: Done." << llendl;
#endif
return(LLVORBISENC_NOERR);
}
int encvorbisInit( hb_work_object_t * w, hb_job_t * job )
{
hb_audio_t * audio = w->audio;
int i;
ogg_packet header[3];
hb_work_private_t * pv = calloc( 1, sizeof( hb_work_private_t ) );
w->private_data = pv;
pv->out_discrete_channels = HB_AMIXDOWN_GET_DISCRETE_CHANNEL_COUNT(audio->config.out.mixdown);
pv->job = job;
hb_log( "encvorbis: opening libvorbis" );
/* init */
for( i = 0; i < 3; i++ )
{
// Zero vorbis headers so that we don't crash in mk_laceXiph
// when vorbis_encode_setup_managed fails.
memset( w->config->vorbis.headers[i], 0, sizeof( ogg_packet ) );
}
vorbis_info_init( &pv->vi );
if( audio->config.out.bitrate > 0 )
{
/* 28kbps/channel seems to be the minimum for 6ch vorbis. */
int min_bitrate = 28 * pv->out_discrete_channels;
if (pv->out_discrete_channels > 2 && audio->config.out.bitrate < min_bitrate)
{
hb_log( "encvorbis: Selected bitrate (%d kbps) too low for %d channel audio.", audio->config.out.bitrate, pv->out_discrete_channels);
hb_log( "encvorbis: Resetting bitrate to %d kbps", min_bitrate);
/* Naughty! We shouldn't modify the audio from here. */
audio->config.out.bitrate = min_bitrate;
}
if( vorbis_encode_setup_managed( &pv->vi, pv->out_discrete_channels,
audio->config.out.samplerate, -1, 1000 * audio->config.out.bitrate, -1 ) )
{
hb_error( "encvorbis: vorbis_encode_setup_managed failed.\n" );
*job->die = 1;
return -1;
}
}
else if( audio->config.out.quality != -1 )
{
// map VBR quality to Vorbis API (divide by 10)
if( vorbis_encode_setup_vbr( &pv->vi, pv->out_discrete_channels,
audio->config.out.samplerate, audio->config.out.quality/10 ) )
{
hb_error( "encvorbis: vorbis_encode_setup_vbr failed.\n" );
*job->die = 1;
return -1;
}
}
if( vorbis_encode_ctl( &pv->vi, OV_ECTL_RATEMANAGE2_SET, NULL ) ||
vorbis_encode_setup_init( &pv->vi ) )
{
hb_error( "encvorbis: vorbis_encode_ctl( ratemanage2_set ) OR vorbis_encode_setup_init failed.\n" );
*job->die = 1;
return -1;
}
/* add a comment */
vorbis_comment_init( &pv->vc );
vorbis_comment_add_tag( &pv->vc, "Encoder", "HandBrake");
vorbis_comment_add_tag( &pv->vc, "LANGUAGE", w->config->vorbis.language);
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init( &pv->vd, &pv->vi);
vorbis_block_init( &pv->vd, &pv->vb);
/* get the 3 headers */
vorbis_analysis_headerout( &pv->vd, &pv->vc,
&header[0], &header[1], &header[2] );
for( i = 0; i < 3; i++ )
{
memcpy( w->config->vorbis.headers[i], &header[i],
sizeof( ogg_packet ) );
memcpy( w->config->vorbis.headers[i] + sizeof( ogg_packet ),
header[i].packet, header[i].bytes );
}
pv->input_samples = pv->out_discrete_channels * OGGVORBIS_FRAME_SIZE;
audio->config.out.samples_per_frame = OGGVORBIS_FRAME_SIZE;
pv->buf = malloc( pv->input_samples * sizeof( float ) );
pv->list = hb_list_init();
switch (pv->out_discrete_channels) {
case 1:
pv->channel_map[0] = 0;
break;
case 6:
// Vorbis uses the following channel map = L C R Ls Rs Lfe
if( audio->config.in.channel_map == &hb_ac3_chan_map )
{
pv->channel_map[0] = 1;
pv->channel_map[1] = 2;
pv->channel_map[2] = 3;
pv->channel_map[3] = 4;
pv->channel_map[4] = 5;
pv->channel_map[5] = 0;
}
else if( audio->config.in.channel_map == &hb_smpte_chan_map )
{
pv->channel_map[0] = 0;
pv->channel_map[1] = 2;
pv->channel_map[2] = 1;
pv->channel_map[3] = 4;
pv->channel_map[4] = 5;
pv->channel_map[5] = 3;
}
else // &hb_qt_chan_map
{
pv->channel_map[0] = 1;
pv->channel_map[1] = 0;
pv->channel_map[2] = 2;
pv->channel_map[3] = 3;
pv->channel_map[4] = 4;
pv->channel_map[5] = 5;
}
break;
default:
hb_log("encvorbis.c: Unable to correctly proccess %d channels, assuming stereo.", pv->out_discrete_channels);
case 2:
// Assume stereo
pv->channel_map[0] = 0;
pv->channel_map[1] = 1;
break;
}
return 0;
}
int lame_encode_ogg_init(lame_global_flags *gfp)
{
lame_internal_flags *gfc=gfp->internal_flags;
char comment[MAX_COMMENT_LENGTH+1];
/********** Encode setup ************/
/* choose an encoding mode */
/* (mode 0: 44kHz stereo uncoupled, roughly 128kbps VBR) */
if (gfp->compression_ratio < 5.01) {
memcpy(&vi2,&info_E,sizeof(vi2));
MSGF( gfc, "Encoding with Vorbis mode info_E \n" );
} else if (gfp->compression_ratio < 6) {
memcpy(&vi2,&info_D,sizeof(vi2));
MSGF( gfc, "Encoding with Vorbis mode info_D \n" );
} else if (gfp->compression_ratio < 8) {
memcpy(&vi2,&info_C,sizeof(vi2));
MSGF( gfc, "Encoding with Vorbis mode info_C \n" );
} else if (gfp->compression_ratio < 10) {
memcpy(&vi2,&info_B,sizeof(vi2));
MSGF( gfc, "Encoding with Vorbis mode info_B \n" );
} else if (gfp->compression_ratio < 12) {
memcpy(&vi2,&info_A,sizeof(vi2));
MSGF( gfc, "Encoding with Vorbis mode info_A \n" );
} else {
memcpy(&vi2,&info_A,sizeof(vi2));
MSGF( gfc, "Encoding with Vorbis mode info_A \n" );
}
vi2.channels = gfc->channels_out;
vi2.rate = gfp->out_samplerate;
/* add a comment */
vorbis_comment_init(&vc2);
vorbis_comment_add(&vc2,"Track encoded using L.A.M.E. libvorbis interface.");
/* Add ID3-style comments to the output using (for the time being) the
"private data members" in the "id3tag_spec" data structure. This was
from a patch by Ralph Giles <*****@*****.**> */
#ifdef THIS_CODE_IS_NOT_BROKEN_ANYMORE
if(gfp->tag_spec.title) {
strcpy(comment,"TITLE=");
strncat(comment,gfp->tag_spec.title,MAX_COMMENT_LENGTH-strlen(comment));
vorbis_comment_add(&vc2,comment);
}
if(gfp->tag_spec.artist) {
strcpy(comment,"ARTIST=");
strncat(comment,gfp->tag_spec.artist,MAX_COMMENT_LENGTH-strlen(comment));
vorbis_comment_add(&vc2,comment);
}
if(gfp->tag_spec.album) {
strcpy(comment,"ALBUM=");
strncat(comment,gfp->tag_spec.album,MAX_COMMENT_LENGTH-strlen(comment));
vorbis_comment_add(&vc2,comment);
}
/* pretend that the ID3 fields are equivalent to the Vorbis fields */
if(gfp->tag_spec.year) {
sprintf(comment, "DATE=%d", gfp->tag_spec.year);
vorbis_comment_add(&vc2,comment);
}
if(gfp->tag_spec.comment) {
strcpy(comment,"DESCRIPTION=");
strncat(comment,gfp->tag_spec.comment,MAX_COMMENT_LENGTH-strlen(comment));
vorbis_comment_add(&vc2,comment);
}
/* TODO -- support for track and genre */
#endif
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init(&vd2,&vi2);
vorbis_block_init(&vd2,&vb2);
/* set up our packet->stream encoder */
/* pick a random serial number; that way we can more likely build
chained streams just by concatenation */
srand(time(NULL));
ogg_stream_init(&os2,rand());
/* Vorbis streams begin with three headers; the initial header (with
most of the codec setup parameters) which is mandated by the Ogg
bitstream spec. The second header holds any comment fields. The
third header holds the bitstream codebook. We merely need to
make the headers, then pass them to libvorbis one at a time;
libvorbis handles the additional Ogg bitstream constraints */
{
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd2,&vc2,&header,&header_comm,&header_code);
ogg_stream_packetin(&os2,&header); /* automatically placed in its own
page */
ogg_stream_packetin(&os2,&header_comm);
ogg_stream_packetin(&os2,&header_code);
/* no need to write out here. We'll get to that in the main loop */
}
return 0;
}
void* vorbisEncoder(void *obj)
{
VS::setThreadName("vorbisEncoder");
VorbisEncoderData* ved = (VorbisEncoderData*)obj;
vorbis_info vi;
vorbis_dsp_state vd;
vorbis_block vb;
vorbis_info_init(&vi);
vorbis_encode_init(&vi, ved->m_channels, ved->m_sample_rate, -1,
112000, -1);
vorbis_analysis_init(&vd, &vi);
vorbis_block_init(&vd, &vb);
vorbis_comment vc;
vorbis_comment_init(&vc);
vorbis_comment_add_tag(&vc, "ENCODER", "STK vorbis encoder");
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd, &vc, &header, &header_comm,
&header_code);
if (header.bytes > 255 || header_comm.bytes > 255)
{
Log::error("vorbisEncoder", "Header is too long.");
return NULL;
}
FILE* vb_data = fopen((getRecordingName() + ".audio").c_str(), "wb");
if (vb_data == NULL)
{
Log::error("vorbisEncoder", "Failed to open file for encoding"
" vorbis.");
return NULL;
}
fwrite(&ved->m_sample_rate, 1, sizeof(uint32_t), vb_data);
fwrite(&ved->m_channels, 1, sizeof(uint32_t), vb_data);
const uint32_t all = header.bytes + header_comm.bytes +
header_code.bytes + 3;
fwrite(&all, 1, sizeof(uint32_t), vb_data);
uint8_t size = 2;
fwrite(&size, 1, sizeof(uint8_t), vb_data);
size = (uint8_t)header.bytes;
fwrite(&size, 1, sizeof(uint8_t), vb_data);
size = (uint8_t)header_comm.bytes;
fwrite(&size, 1, sizeof(uint8_t), vb_data);
fwrite(header.packet, 1, header.bytes, vb_data);
fwrite(header_comm.packet, 1, header_comm.bytes, vb_data);
fwrite(header_code.packet, 1, header_code.bytes, vb_data);
Synchronised<std::list<int8_t*> >* audio_data =
(Synchronised<std::list<int8_t*> >*)ved->m_data;
pthread_cond_t* cond_request = ved->m_enc_request;
ogg_packet op;
int64_t last_timestamp = 0;
bool eos = false;
while (eos == false)
{
audio_data->lock();
bool waiting = audio_data->getData().empty();
while (waiting)
{
pthread_cond_wait(cond_request, audio_data->getMutex());
waiting = audio_data->getData().empty();
}
int8_t* audio_buf = audio_data->getData().front();
audio_data->getData().pop_front();
audio_data->unlock();
if (audio_buf == NULL)
{
vorbis_analysis_wrote(&vd, 0);
eos = true;
}
else
{
float **buffer = vorbis_analysis_buffer(&vd, 1024);
const unsigned channels = ved->m_channels;
if (ved->m_audio_type == VorbisEncoderData::AT_PCM)
{
for (unsigned j = 0; j < channels; j++)
{
for (unsigned i = 0; i < 1024; i++)
{
int8_t* each_channel =
&audio_buf[i * channels * 2 + j * 2];
buffer[j][i] = float((each_channel[1] << 8) |
(0x00ff & (int)each_channel[0])) / 32768.0f;
}
}
}
else
{
for (unsigned j = 0; j < channels; j++)
{
for (unsigned i = 0; i < 1024; i++)
{
float* fbuf = reinterpret_cast<float*>(audio_buf);
buffer[j][i] = fbuf[i * channels + j];
}
}
}
vorbis_analysis_wrote(&vd, 1024);
}
while (vorbis_analysis_blockout(&vd, &vb) == 1)
{
vorbis_analysis(&vb, NULL);
vorbis_bitrate_addblock(&vb);
while (vorbis_bitrate_flushpacket(&vd, &op))
{
if (op.granulepos > 0)
{
uint32_t frame_size = (uint32_t)op.bytes;
fwrite(&frame_size, 1, sizeof(uint32_t), vb_data);
fwrite(&last_timestamp, 1, sizeof(int64_t), vb_data);
fwrite(op.packet, 1, frame_size, vb_data);
double s = (double)op.granulepos /
(double)ved->m_sample_rate * 1000000000.;
last_timestamp = (int64_t)s;
}
}
}
delete[] audio_buf;
}
vorbis_block_clear(&vb);
vorbis_dsp_clear(&vd);
vorbis_comment_clear(&vc);
vorbis_info_clear(&vi);
fclose(vb_data);
return NULL;
} // vorbisEncoder
static int
vorbis_write_header (SF_PRIVATE *psf, int UNUSED (calc_length))
{ OGG_PRIVATE *odata = (OGG_PRIVATE *) psf->container_data ;
VORBIS_PRIVATE *vdata = (VORBIS_PRIVATE *) psf->codec_data ;
int k, ret ;
vorbis_info_init (&vdata->vinfo) ;
/* The style of encoding should be selectable here, VBR quality mode. */
ret = vorbis_encode_init_vbr (&vdata->vinfo, psf->sf.channels, psf->sf.samplerate, vdata->quality) ;
#if 0
ret = vorbis_encode_init (&vdata->vinfo, psf->sf.channels, psf->sf.samplerate, -1, 128000, -1) ; /* average bitrate mode */
ret = ( vorbis_encode_setup_managed (&vdata->vinfo, psf->sf.channels, psf->sf.samplerate, -1, 128000, -1)
|| vorbis_encode_ctl (&vdata->vinfo, OV_ECTL_RATEMANAGE_AVG, NULL)
|| vorbis_encode_setup_init (&vdata->vinfo)
) ;
#endif
if (ret)
return SFE_BAD_OPEN_FORMAT ;
vdata->loc = 0 ;
/* add a comment */
vorbis_comment_init (&vdata->vcomment) ;
vorbis_comment_add_tag (&vdata->vcomment, "ENCODER", "libsndfile") ;
for (k = 0 ; k < SF_MAX_STRINGS ; k++)
{ const char * name ;
if (psf->strings.data [k].type == 0)
break ;
switch (psf->strings.data [k].type)
{ case SF_STR_TITLE :
name = "TITLE" ;
break ;
case SF_STR_COPYRIGHT :
name = "COPYRIGHT" ;
break ;
case SF_STR_SOFTWARE :
name = "SOFTWARE" ;
break ;
case SF_STR_ARTIST :
name = "ARTIST" ;
break ;
case SF_STR_COMMENT :
name = "COMMENT" ;
break ;
case SF_STR_DATE :
name = "DATE" ;
break ;
case SF_STR_ALBUM :
name = "ALBUM" ;
break ;
case SF_STR_LICENSE :
name = "LICENSE" ;
break ;
case SF_STR_TRACKNUMBER :
name = "Tracknumber" ;
break ;
case SF_STR_GENRE :
name = "Genre" ;
break ;
default :
continue ;
} ;
vorbis_comment_add_tag (&vdata->vcomment, name, psf->strings.storage + psf->strings.data [k].offset) ;
} ;
/* set up the analysis state and auxiliary encoding storage */
vorbis_analysis_init (&vdata->vdsp, &vdata->vinfo) ;
vorbis_block_init (&vdata->vdsp, &vdata->vblock) ;
/*
** Set up our packet->stream encoder.
** Pick a random serial number ; that way we can more likely build
** chained streams just by concatenation.
*/
ogg_stream_init (&odata->ostream, psf_rand_int32 ()) ;
/* Vorbis streams begin with three headers ; the initial header (with
most of the codec setup parameters) which is mandated by the Ogg
bitstream spec. The second header holds any comment fields. The
third header holds the bitstream codebook. We merely need to
make the headers, then pass them to libvorbis one at a time ;
libvorbis handles the additional Ogg bitstream constraints */
{ ogg_packet header ;
ogg_packet header_comm ;
ogg_packet header_code ;
int result ;
vorbis_analysis_headerout (&vdata->vdsp, &vdata->vcomment, &header, &header_comm, &header_code) ;
ogg_stream_packetin (&odata->ostream, &header) ; /* automatically placed in its own page */
ogg_stream_packetin (&odata->ostream, &header_comm) ;
ogg_stream_packetin (&odata->ostream, &header_code) ;
/* This ensures the actual
* audio data will start on a new page, as per spec
*/
while ((result = ogg_stream_flush (&odata->ostream, &odata->opage)) != 0)
{ psf_fwrite (odata->opage.header, 1, odata->opage.header_len, psf) ;
psf_fwrite (odata->opage.body, 1, odata->opage.body_len, psf) ;
} ;
}
return 0 ;
} /* vorbis_write_header */
int audev_init_device(char *devname, long ratewanted, int verbose, extraopt_t *extra)
{
int ret;
int channels, format, rate;
int fragsize;
extraopt_t *opt;
double maxsecs = 5.0;
double quality = 0.5;
char *title = NULL;
if (verbose) {
printf("Boodler: VORBIS sound driver.\n");
}
if (device) {
fprintf(stderr, "Sound device is already open.\n");
return FALSE;
}
format = FALSE; /* always little-endian */
for (opt=extra; opt->key; opt++) {
if (!strcmp(opt->key, "time") && opt->val) {
maxsecs = atof(opt->val);
}
else if (!strcmp(opt->key, "quality") && opt->val) {
quality = atof(opt->val);
}
else if (!strcmp(opt->key, "title") && opt->val) {
title = opt->val;
}
else if (!strcmp(opt->key, "listdevices")) {
printf("Device list: give any writable file as a device name.\n");
}
}
if (!ratewanted)
ratewanted = DEFAULT_SOUNDRATE;
if (!devname)
devname = DEFAULT_FILENAME;
device = fopen(devname, "wb");
if (!device) {
fprintf(stderr, "Error opening file %s\n", devname);
return FALSE;
}
if (verbose) {
printf("Opened file %s.\n", devname);
}
rate = ratewanted;
channels = 2;
fragsize = 16384;
if (verbose) {
printf("%d channels, %d frames per second, 16-bit samples (signed, %s)\n",
channels, rate, (format?"big-endian":"little-endian"));
printf("vorbis VBR encoding quality %f\n", quality);
}
maxtime = (long)(maxsecs * (double)rate);
curtime = 0;
if (verbose) {
printf("%g seconds of output (%ld frames)\n", maxsecs, maxtime);
}
sound_rate = rate;
sound_channels = channels;
sound_format = format;
sound_buffersize = fragsize;
samplesperbuf = sound_buffersize / 2;
framesperbuf = sound_buffersize / (2 * sound_channels);
rawbuffer = (char *)malloc(sound_buffersize);
if (!rawbuffer) {
fprintf(stderr, "Unable to allocate sound buffer.\n");
fclose(device);
device = NULL;
return FALSE;
}
valbuffer = (long *)malloc(sizeof(long) * samplesperbuf);
if (!valbuffer) {
fprintf(stderr, "Unable to allocate sound buffer.\n");
free(rawbuffer);
rawbuffer = NULL;
fclose(device);
device = NULL;
return FALSE;
}
vorbis_info_init(&vi);
ret = vorbis_encode_init_vbr(&vi, 2, sound_rate, quality);
if (ret) {
fprintf(stderr, "Unable to initialize Vorbis encoder.\n");
free(valbuffer);
valbuffer = NULL;
free(rawbuffer);
rawbuffer = NULL;
fclose(device);
device = NULL;
return FALSE;
}
/* See <http://xiph.org/vorbis/doc/v-comment.html> */
vorbis_comment_init(&vc);
{
char commentbuf[256];
time_t nowtime;
struct tm *now;
if (title) {
strcpy(commentbuf, "Boodler: ");
strncat(commentbuf, title, 255-strlen(commentbuf));
vorbis_comment_add_tag(&vc, "TITLE", commentbuf);
}
nowtime = time(NULL);
now = localtime(&nowtime);
strftime(commentbuf, 255, "%Y-%m-%d (generated)", now);
vorbis_comment_add_tag(&vc, "DATE", commentbuf);
vorbis_comment_add_tag(&vc, "ENCODER", "Boodler");
}
vorbis_analysis_init(&vd, &vi);
vorbis_block_init(&vd, &vb);
srand(time(NULL));
ret = ogg_stream_init(&os, rand());
if (ret) {
fprintf(stderr, "Unable to initialize Ogg stream.\n");
free(valbuffer);
valbuffer = NULL;
free(rawbuffer);
rawbuffer = NULL;
fclose(device);
device = NULL;
return FALSE;
}
{
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_analysis_headerout(&vd, &vc, &header, &header_comm, &header_code);
ogg_stream_packetin(&os, &header);
ogg_stream_packetin(&os, &header_comm);
ogg_stream_packetin(&os, &header_code);
while (!eos) {
int result = ogg_stream_flush(&os, &og);
if (result==0) break;
fwrite(og.header, 1, og.header_len, device);
fwrite(og.body, 1, og.body_len, device);
}
}
return TRUE;
}
static gint vorbis_open(void)
{
ogg_packet header;
ogg_packet header_comm;
ogg_packet header_code;
vorbis_init(NULL);
vorbis_info_init(&vi);
vorbis_comment_init(&vc);
if (tuple)
{
gchar tmpstr[32];
gint scrint;
add_string_from_tuple (& vc, "title", tuple, FIELD_TITLE);
add_string_from_tuple (& vc, "artist", tuple, FIELD_ARTIST);
add_string_from_tuple (& vc, "album", tuple, FIELD_ALBUM);
add_string_from_tuple (& vc, "genre", tuple, FIELD_GENRE);
add_string_from_tuple (& vc, "date", tuple, FIELD_DATE);
add_string_from_tuple (& vc, "comment", tuple, FIELD_COMMENT);
if ((scrint = tuple_get_int(tuple, FIELD_TRACK_NUMBER, NULL)))
{
g_snprintf(tmpstr, sizeof(tmpstr), "%d", scrint);
vorbis_comment_add_tag(&vc, "tracknumber", tmpstr);
}
if ((scrint = tuple_get_int(tuple, FIELD_YEAR, NULL)))
{
g_snprintf(tmpstr, sizeof(tmpstr), "%d", scrint);
vorbis_comment_add_tag(&vc, "year", tmpstr);
}
}
if (vorbis_encode_init_vbr (& vi, input.channels, input.frequency,
v_base_quality))
{
vorbis_info_clear(&vi);
return 0;
}
vorbis_analysis_init(&vd, &vi);
vorbis_block_init(&vd, &vb);
srand(time(NULL));
ogg_stream_init(&os, rand());
vorbis_analysis_headerout(&vd, &vc, &header, &header_comm, &header_code);
ogg_stream_packetin(&os, &header);
ogg_stream_packetin(&os, &header_comm);
ogg_stream_packetin(&os, &header_code);
while (ogg_stream_flush (& os, & og))
{
write_output(og.header, og.header_len);
write_output(og.body, og.body_len);
}
return 1;
}
static SCM open_vorbis_stream(SCM server, SCM port, SCM password,
SCM mount, SCM config) {
SCM smob;
double qual;
char stream_name[32];
VORBIS_STREAM *stream;
stream = (VORBIS_STREAM *)my_gc_malloc(sizeof(VORBIS_STREAM),
"vorbis_stream", "vorbis stream");
stream->header.shutdown = vorbis_shutdown;
vorbis_info_init(&(stream->vi));
SCM quality = get_property(config, "quality");
if (scm_is_number(quality)) qual = scm_to_double(quality);
else qual = 0.6;
scm_remember_upto_here_1(quality);
stream->header.shoutbuf = NULL;
stream->header.shoutbuf_size = 0;
stream->header.rb_size = sizeof(sample_t) * RB_SECS * sampling_rate
* QMX_CHANNELS + 0.5;
stream->header.ringbuf = jack_ringbuffer_create(stream->header.rb_size);
if (vorbis_encode_init_vbr(&(stream->vi), QMX_CHANNELS,
sampling_rate, qual) != 0) {
log_msg("vorbis init failed\n");
return SCM_BOOL_F;
}
if ((stream->header.shout = shout_new()) == NULL) {
log_msg("can't create shout instance\n");
return SCM_BOOL_F;
}
vorbis_analysis_init(&(stream->vd), &(stream->vi));
vorbis_block_init(&(stream->vd), &(stream->vb));
ogg_stream_init(&(stream->os), rand());
vorbis_comment_init(&(stream->vc));
vorbis_comment_add_tag(&(stream->vc), "ENCODER", "QMX");
shout_set_protocol(stream->header.shout, SHOUT_PROTOCOL_HTTP);
shout_set_format(stream->header.shout, SHOUT_FORMAT_OGG);
char *server_s = scm_to_locale_string(server); // server
shout_set_host(stream->header.shout, server_s);
free(server_s);
shout_set_port(stream->header.shout, scm_to_int(port)); // port
shout_set_user(stream->header.shout, "source"); // user
char *password_s = scm_to_locale_string(password); // password
shout_set_password(stream->header.shout, password_s);
free(password_s);
char *mount_s = scm_to_locale_string(mount); // mount pt
shout_set_mount(stream->header.shout, mount_s);
SCM title = get_property(config, "title");
if (scm_is_string(title)) {
char *title_s = scm_to_locale_string(title);
if (strlen(title_s) > 0)
vorbis_comment_add_tag(&(stream->vc), "TITLE", title_s);
free(title_s);
}
scm_remember_upto_here_1(title);
SCM name = get_property(config, "name");
stream_name[0] = '\0';
if (scm_is_string(name)) {
char *name_s = scm_to_locale_string(name);
shout_set_name(stream->header.shout, name_s);
snprintf(stream_name, sizeof(stream_name), "'%s' ", name_s);
free(name_s);
}
scm_remember_upto_here_1(name);
SCM description = get_property(config, "description");
if (scm_is_string(description)) {
char *description_s = scm_to_locale_string(description);
shout_set_description(stream->header.shout, description_s);
free(description_s);
}
scm_remember_upto_here_1(description);
SCM genre = get_property(config, "genre");
if (scm_is_string(genre)) {
char *genre_s = scm_to_locale_string(genre);
shout_set_genre(stream->header.shout, genre_s);
free(genre_s);
}
scm_remember_upto_here_1(genre);
shout_set_audio_info(stream->header.shout,
SHOUT_AI_CHANNELS, encode_int(QMX_CHANNELS));
shout_set_audio_info(stream->header.shout,
SHOUT_AI_SAMPLERATE, encode_int(sampling_rate));
if (shout_open(stream->header.shout) == SHOUTERR_SUCCESS) {
log_msg("vorbis stream %smounted at %s\n", stream_name, mount_s);
free(mount_s);
}
else {
log_msg("vorbis stream %sfailed to open: %s\n", stream_name,
shout_get_error(stream->header.shout));
free(mount_s);
shout_free(stream->header.shout);
return SCM_BOOL_F;
}
vorbis_streams += 1;
stream->header.eos = 0;
SCM_NEWSMOB(smob, vorbis_stream_tag, stream);
streams = link_in(streams, smob);
stream->header.do_prep = need_prepper;
need_prepper = 0;
stream->header.bookmark = 0;
add_outbuf_client(load_frames, process_frames, (void *)stream,
"Ogg-Vorbis stream");
spawn_detached_thread(shout_thread, (void *)&(stream->header));
scm_remember_upto_here_1(smob);
return smob;
}