diff options
Diffstat (limited to 'libhb/encvorbis.c')
| -rw-r--r-- | libhb/encvorbis.c | 205 |
1 files changed, 205 insertions, 0 deletions
diff --git a/libhb/encvorbis.c b/libhb/encvorbis.c new file mode 100644 index 000000000..86b824372 --- /dev/null +++ b/libhb/encvorbis.c @@ -0,0 +1,205 @@ +/* $Id: encvorbis.c,v 1.6 2005/03/05 15:08:32 titer Exp $ + + This file is part of the HandBrake source code. + Homepage: <http://handbrake.m0k.org/>. + It may be used under the terms of the GNU General Public License. */ + +#include "hb.h" + +#include "vorbis/vorbisenc.h" + +#define OGGVORBIS_FRAME_SIZE 1024 + +struct hb_work_object_s +{ + HB_WORK_COMMON; + + hb_job_t * job; + hb_audio_t * audio; + + vorbis_info vi; + vorbis_comment vc; + vorbis_dsp_state vd; + vorbis_block vb; + + unsigned long input_samples; + uint8_t * buf; + uint64_t pts; + + hb_list_t * list; +}; + +/*********************************************************************** + * Local prototypes + **********************************************************************/ +static void Close( hb_work_object_t ** _w ); +static int Work( hb_work_object_t * w, hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out ); + +/*********************************************************************** + * hb_work_encvorbis_init + *********************************************************************** + * + **********************************************************************/ +hb_work_object_t * hb_work_encvorbis_init( hb_job_t * job, hb_audio_t * audio ) +{ + int i; + ogg_packet header[3]; + + hb_work_object_t * w = calloc( sizeof( hb_work_object_t ), 1 ); + w->name = strdup( "Vorbis encoder (libvorbis)" ); + w->work = Work; + w->close = Close; + + w->job = job; + w->audio = audio; + + hb_log( "encvorbis: opening libvorbis" ); + + /* init */ + vorbis_info_init( &w->vi ); + if( vorbis_encode_setup_managed( &w->vi, 2, + job->arate, -1, 1000 * job->abitrate, -1 ) || + vorbis_encode_ctl( &w->vi, OV_ECTL_RATEMANAGE_AVG, NULL ) || + vorbis_encode_setup_init( &w->vi ) ) + { + hb_log( "encvorbis: vorbis_encode_setup_managed failed" ); + } + + /* add a comment */ + vorbis_comment_init( &w->vc ); + vorbis_comment_add_tag( &w->vc, "Encoder", "HandBrake"); + + /* set up the analysis state and auxiliary encoding storage */ + vorbis_analysis_init( &w->vd, &w->vi); + vorbis_block_init( &w->vd, &w->vb); + + /* get the 3 headers */ + vorbis_analysis_headerout( &w->vd, &w->vc, + &header[0], &header[1], &header[2] ); + for( i = 0; i < 3; i++ ) + { + audio->config.vorbis.headers[i] = + malloc( sizeof( ogg_packet ) + header[i].bytes ); + memcpy( audio->config.vorbis.headers[i], &header[i], + sizeof( ogg_packet ) ); + memcpy( audio->config.vorbis.headers[i] + sizeof( ogg_packet ), + header[i].packet, header[i].bytes ); + } + + w->input_samples = 2 * OGGVORBIS_FRAME_SIZE; + w->buf = malloc( w->input_samples * sizeof( float ) ); + + w->list = hb_list_init(); + + return w; +} + +/*********************************************************************** + * Close + *********************************************************************** + * + **********************************************************************/ +static void Close( hb_work_object_t ** _w ) +{ + hb_work_object_t * w = *_w; + free( w->name ); + free( w ); + *_w = NULL; +} + +/*********************************************************************** + * Flush + *********************************************************************** + * + **********************************************************************/ +static hb_buffer_t * Flush( hb_work_object_t * w ) +{ + hb_buffer_t * buf; + + if( vorbis_analysis_blockout( &w->vd, &w->vb ) == 1 ) + { + ogg_packet op; + + vorbis_analysis( &w->vb, NULL ); + vorbis_bitrate_addblock( &w->vb ); + + if( vorbis_bitrate_flushpacket( &w->vd, &op ) ) + { + buf = hb_buffer_init( sizeof( ogg_packet ) + op.bytes ); + memcpy( buf->data, &op, sizeof( ogg_packet ) ); + memcpy( buf->data + sizeof( ogg_packet ), op.packet, + op.bytes ); + buf->key = 1; + buf->start = w->pts; /* No exact, but who cares - the OGM + muxer doesn't use it */ + buf->stop = buf->start + + 90000 * OGGVORBIS_FRAME_SIZE + w->job->arate; + + return buf; + } + } + + return NULL; +} + +/*********************************************************************** + * Encode + *********************************************************************** + * + **********************************************************************/ +static hb_buffer_t * Encode( hb_work_object_t * w ) +{ + hb_buffer_t * buf; + float ** buffer; + int i; + + /* Try to extract more data */ + if( ( buf = Flush( w ) ) ) + { + return buf; + } + + if( hb_list_bytes( w->list ) < w->input_samples * sizeof( float ) ) + { + return NULL; + } + + /* Process more samples */ + hb_list_getbytes( w->list, w->buf, w->input_samples * sizeof( float ), + &w->pts, NULL ); + buffer = vorbis_analysis_buffer( &w->vd, OGGVORBIS_FRAME_SIZE ); + for( i = 0; i < OGGVORBIS_FRAME_SIZE; i++ ) + { + buffer[0][i] = ((float *) w->buf)[2*i] / 32768.f; + buffer[1][i] = ((float *) w->buf)[2*i+1] / 32768.f; + } + vorbis_analysis_wrote( &w->vd, OGGVORBIS_FRAME_SIZE ); + + /* Try to extract again */ + return Flush( w ); +} + +/*********************************************************************** + * Work + *********************************************************************** + * + **********************************************************************/ +static int Work( hb_work_object_t * w, hb_buffer_t ** buf_in, + hb_buffer_t ** buf_out ) +{ + hb_buffer_t * buf; + + hb_list_add( w->list, *buf_in ); + *buf_in = NULL; + + *buf_out = buf = Encode( w ); + + while( buf ) + { + buf->next = Encode( w ); + buf = buf->next; + } + + return HB_WORK_OK; +} |