2006-09-10 14:02:42 +00:00
/*
* copyright ( c ) 2002 Mark Hills < mark @ pogo . org . uk >
*
2011-03-18 17:35:10 +00:00
* This file is part of Libav .
2006-10-07 15:30:46 +00:00
*
2011-03-18 17:35:10 +00:00
* Libav is free software ; you can redistribute it and / or
2006-09-10 14:02:42 +00:00
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation ; either
2006-10-07 15:30:46 +00:00
* version 2.1 of the License , or ( at your option ) any later version .
2006-09-10 14:02:42 +00:00
*
2011-03-18 17:35:10 +00:00
* Libav is distributed in the hope that it will be useful ,
2006-09-10 14:02:42 +00:00
* but WITHOUT ANY WARRANTY ; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE . See the GNU
* Lesser General Public License for more details .
*
* You should have received a copy of the GNU Lesser General Public
2011-03-18 17:35:10 +00:00
* License along with Libav ; if not , write to the Free Software
2007-07-05 10:40:25 +00:00
* Foundation , Inc . , 51 Franklin Street , Fifth Floor , Boston , MA 02110 - 1301 USA
2006-09-10 14:02:42 +00:00
*/
2003-03-06 11:32:04 +00:00
/**
2010-04-20 14:45:34 +00:00
* @ file
2003-03-06 11:32:04 +00:00
* Ogg Vorbis codec support via libvorbisenc .
* @ author Mark Hills < mark @ pogo . org . uk >
2002-09-01 18:07:56 +00:00
*/
# include <vorbis/vorbisenc.h>
2010-09-29 15:09:38 +00:00
# include "libavutil/opt.h"
2002-09-01 18:07:56 +00:00
# include "avcodec.h"
2007-06-02 01:41:07 +00:00
# include "bytestream.h"
2010-06-27 09:25:05 +00:00
# include "vorbis.h"
2011-07-04 06:44:49 +00:00
# include "libavutil/mathematics.h"
2002-09-01 18:07:56 +00:00
2004-04-04 17:06:30 +00:00
# undef NDEBUG
# include <assert.h>
2004-04-04 14:39:20 +00:00
# define OGGVORBIS_FRAME_SIZE 64
2002-09-01 18:07:56 +00:00
2004-04-04 14:39:20 +00:00
# define BUFFER_SIZE (1024*64)
2002-09-01 18:07:56 +00:00
typedef struct OggVorbisContext {
2010-09-29 15:09:38 +00:00
AVClass * av_class ;
2002-09-01 18:07:56 +00:00
vorbis_info vi ;
vorbis_dsp_state vd ;
vorbis_block vb ;
2004-04-04 14:39:20 +00:00
uint8_t buffer [ BUFFER_SIZE ] ;
int buffer_index ;
2008-10-15 07:29:37 +00:00
int eof ;
2002-11-22 07:27:13 +00:00
/* decoder */
vorbis_comment vc ;
2004-04-04 02:07:15 +00:00
ogg_packet op ;
2010-09-29 15:09:38 +00:00
double iblock ;
2002-09-01 18:07:56 +00:00
} OggVorbisContext ;
2010-09-29 15:09:38 +00:00
static const AVOption options [ ] = {
2011-04-30 05:43:26 +00:00
{ " iblock " , " Sets the impulse block bias " , offsetof ( OggVorbisContext , iblock ) , FF_OPT_TYPE_DOUBLE , { . dbl = 0 } , - 15 , 0 , AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_ENCODING_PARAM } ,
2010-09-29 15:09:38 +00:00
{ NULL }
} ;
2010-10-14 07:47:49 +00:00
static const AVClass class = { " libvorbis " , av_default_item_name , options , LIBAVUTIL_VERSION_INT } ;
2002-09-01 18:07:56 +00:00
2009-02-22 13:48:55 +00:00
static av_cold int oggvorbis_init_encoder ( vorbis_info * vi , AVCodecContext * avccontext ) {
2010-09-29 15:09:38 +00:00
OggVorbisContext * context = avccontext - > priv_data ;
2006-01-21 17:09:23 +00:00
double cfreq ;
2002-12-21 15:54:21 +00:00
2005-09-02 11:08:49 +00:00
if ( avccontext - > flags & CODEC_FLAG_QSCALE ) {
2006-01-21 17:09:23 +00:00
/* variable bitrate */
if ( vorbis_encode_setup_vbr ( vi , avccontext - > channels ,
2005-09-02 11:08:49 +00:00
avccontext - > sample_rate ,
2009-02-09 22:11:28 +00:00
avccontext - > global_quality / ( float ) FF_QP2LAMBDA / 10.0 ) )
2006-01-21 17:09:23 +00:00
return - 1 ;
} else {
2010-06-04 22:40:31 +00:00
int minrate = avccontext - > rc_min_rate > 0 ? avccontext - > rc_min_rate : - 1 ;
int maxrate = avccontext - > rc_min_rate > 0 ? avccontext - > rc_max_rate : - 1 ;
2006-01-21 17:09:23 +00:00
/* constant bitrate */
if ( vorbis_encode_setup_managed ( vi , avccontext - > channels ,
2010-06-04 22:40:31 +00:00
avccontext - > sample_rate , minrate , avccontext - > bit_rate , maxrate ) )
2006-01-21 17:09:23 +00:00
return - 1 ;
2010-06-04 22:40:40 +00:00
/* variable bitrate by estimate, disable slow rate management */
if ( minrate = = - 1 & & maxrate = = - 1 )
if ( vorbis_encode_ctl ( vi , OV_ECTL_RATEMANAGE2_SET , NULL ) )
return - 1 ;
2006-01-21 17:09:23 +00:00
}
2002-12-21 15:54:21 +00:00
2006-01-21 17:09:23 +00:00
/* cutoff frequency */
if ( avccontext - > cutoff > 0 ) {
cfreq = avccontext - > cutoff / 1000.0 ;
if ( vorbis_encode_ctl ( vi , OV_ECTL_LOWPASS_SET , & cfreq ) )
return - 1 ;
}
2002-09-01 18:07:56 +00:00
2010-09-29 15:09:38 +00:00
if ( context - > iblock ) {
vorbis_encode_ctl ( vi , OV_ECTL_IBLOCK_SET , & context - > iblock ) ;
}
2006-01-21 17:09:23 +00:00
return vorbis_encode_setup_init ( vi ) ;
2002-09-01 18:07:56 +00:00
}
2010-07-20 21:54:46 +00:00
/* How many bytes are needed for a buffer of length 'l' */
static int xiph_len ( int l ) { return ( 1 + l / 255 + l ) ; }
2008-03-21 03:11:20 +00:00
static av_cold int oggvorbis_encode_init ( AVCodecContext * avccontext ) {
2002-09-01 18:07:56 +00:00
OggVorbisContext * context = avccontext - > priv_data ;
2004-04-04 15:19:20 +00:00
ogg_packet header , header_comm , header_code ;
uint8_t * p ;
2010-07-20 21:54:46 +00:00
unsigned int offset ;
2002-09-01 18:07:56 +00:00
vorbis_info_init ( & context - > vi ) ;
if ( oggvorbis_init_encoder ( & context - > vi , avccontext ) < 0 ) {
2009-07-01 06:48:27 +00:00
av_log ( avccontext , AV_LOG_ERROR , " oggvorbis_encode_init: init_encoder failed \n " ) ;
2005-12-22 01:10:11 +00:00
return - 1 ;
2002-09-01 18:07:56 +00:00
}
vorbis_analysis_init ( & context - > vd , & context - > vi ) ;
vorbis_block_init ( & context - > vd , & context - > vb ) ;
2004-04-04 15:19:20 +00:00
vorbis_comment_init ( & context - > vc ) ;
vorbis_comment_add_tag ( & context - > vc , " encoder " , LIBAVCODEC_IDENT ) ;
vorbis_analysis_headerout ( & context - > vd , & context - > vc , & header ,
& header_comm , & header_code ) ;
2005-12-17 18:14:38 +00:00
2010-07-20 21:54:46 +00:00
avccontext - > extradata_size =
1 + xiph_len ( header . bytes ) + xiph_len ( header_comm . bytes ) +
header_code . bytes ;
p = avccontext - > extradata =
av_malloc ( avccontext - > extradata_size + FF_INPUT_BUFFER_PADDING_SIZE ) ;
2005-05-13 18:10:23 +00:00
p [ 0 ] = 2 ;
offset = 1 ;
offset + = av_xiphlacing ( & p [ offset ] , header . bytes ) ;
offset + = av_xiphlacing ( & p [ offset ] , header_comm . bytes ) ;
memcpy ( & p [ offset ] , header . packet , header . bytes ) ;
offset + = header . bytes ;
memcpy ( & p [ offset ] , header_comm . packet , header_comm . bytes ) ;
offset + = header_comm . bytes ;
memcpy ( & p [ offset ] , header_code . packet , header_code . bytes ) ;
offset + = header_code . bytes ;
2010-07-20 21:54:46 +00:00
assert ( offset = = avccontext - > extradata_size ) ;
2005-12-17 18:14:38 +00:00
2004-04-04 15:19:20 +00:00
/* vorbis_block_clear(&context->vb);
vorbis_dsp_clear ( & context - > vd ) ;
vorbis_info_clear ( & context - > vi ) ; */
vorbis_comment_clear ( & context - > vc ) ;
2005-12-17 18:14:38 +00:00
2002-09-01 18:07:56 +00:00
avccontext - > frame_size = OGGVORBIS_FRAME_SIZE ;
2005-12-17 18:14:38 +00:00
2002-12-09 12:03:43 +00:00
avccontext - > coded_frame = avcodec_alloc_frame ( ) ;
avccontext - > coded_frame - > key_frame = 1 ;
2005-12-17 18:14:38 +00:00
2002-09-01 18:07:56 +00:00
return 0 ;
}
2002-11-22 07:27:13 +00:00
static int oggvorbis_encode_frame ( AVCodecContext * avccontext ,
2005-12-22 01:10:11 +00:00
unsigned char * packets ,
int buf_size , void * data )
2002-09-01 18:07:56 +00:00
{
OggVorbisContext * context = avccontext - > priv_data ;
ogg_packet op ;
2004-12-18 16:20:42 +00:00
signed short * audio = data ;
2008-10-15 07:29:37 +00:00
int l ;
if ( data ) {
2010-07-11 06:40:05 +00:00
const int samples = avccontext - > frame_size ;
2008-10-15 07:29:37 +00:00
float * * buffer ;
2010-06-27 09:25:05 +00:00
int c , channels = context - > vi . channels ;
2002-09-01 18:07:56 +00:00
2008-10-15 07:31:06 +00:00
buffer = vorbis_analysis_buffer ( & context - > vd , samples ) ;
2010-06-27 09:25:05 +00:00
for ( c = 0 ; c < channels ; c + + ) {
int co = ( channels > 8 ) ? c :
ff_vorbis_encoding_channel_layout_offsets [ channels - 1 ] [ c ] ;
2008-10-15 07:31:06 +00:00
for ( l = 0 ; l < samples ; l + + )
2010-06-27 09:25:05 +00:00
buffer [ c ] [ l ] = audio [ l * channels + co ] / 32768.f ;
2005-12-22 01:10:11 +00:00
}
2008-10-15 07:31:06 +00:00
vorbis_analysis_wrote ( & context - > vd , samples ) ;
2008-10-15 07:29:37 +00:00
} else {
if ( ! context - > eof )
vorbis_analysis_wrote ( & context - > vd , 0 ) ;
context - > eof = 1 ;
}
2002-09-01 18:07:56 +00:00
while ( vorbis_analysis_blockout ( & context - > vd , & context - > vb ) = = 1 ) {
2005-12-22 01:10:11 +00:00
vorbis_analysis ( & context - > vb , NULL ) ;
vorbis_bitrate_addblock ( & context - > vb ) ;
2002-09-01 18:07:56 +00:00
2005-12-22 01:10:11 +00:00
while ( vorbis_bitrate_flushpacket ( & context - > vd , & op ) ) {
2007-02-21 10:15:08 +00:00
/* i'd love to say the following line is a hack, but sadly it's
* not , apparently the end of stream decision is in libogg . */
2010-06-22 06:53:06 +00:00
if ( op . bytes = = 1 & & op . e_o_s )
2004-06-22 21:14:01 +00:00
continue ;
2010-07-11 06:59:21 +00:00
if ( context - > buffer_index + sizeof ( ogg_packet ) + op . bytes > BUFFER_SIZE ) {
av_log ( avccontext , AV_LOG_ERROR , " libvorbis: buffer overflow. " ) ;
return - 1 ;
}
2004-04-04 14:39:20 +00:00
memcpy ( context - > buffer + context - > buffer_index , & op , sizeof ( ogg_packet ) ) ;
context - > buffer_index + = sizeof ( ogg_packet ) ;
memcpy ( context - > buffer + context - > buffer_index , op . packet , op . bytes ) ;
context - > buffer_index + = op . bytes ;
// av_log(avccontext, AV_LOG_DEBUG, "e%d / %d\n", context->buffer_index, op.bytes);
2005-12-22 01:10:11 +00:00
}
2002-09-01 18:07:56 +00:00
}
2004-04-04 17:06:30 +00:00
l = 0 ;
2004-04-04 14:39:20 +00:00
if ( context - > buffer_index ) {
2004-04-04 15:19:20 +00:00
ogg_packet * op2 = ( ogg_packet * ) context - > buffer ;
2004-04-04 14:39:20 +00:00
op2 - > packet = context - > buffer + sizeof ( ogg_packet ) ;
2004-04-04 17:06:30 +00:00
2004-06-22 21:14:01 +00:00
l = op2 - > bytes ;
2005-09-04 09:03:01 +00:00
avccontext - > coded_frame - > pts = av_rescale_q ( op2 - > granulepos , ( AVRational ) { 1 , avccontext - > sample_rate } , avccontext - > time_base ) ;
2005-09-04 09:04:52 +00:00
//FIXME we should reorder the user supplied pts and not assume that they are spaced by 1/sample_rate
2004-04-04 17:06:30 +00:00
2010-07-11 06:59:21 +00:00
if ( l > buf_size ) {
av_log ( avccontext , AV_LOG_ERROR , " libvorbis: buffer overflow. " ) ;
return - 1 ;
}
2004-06-22 21:14:01 +00:00
memcpy ( packets , op2 - > packet , l ) ;
context - > buffer_index - = l + sizeof ( ogg_packet ) ;
2010-06-16 19:03:54 +00:00
memmove ( context - > buffer , context - > buffer + l + sizeof ( ogg_packet ) , context - > buffer_index ) ;
2004-04-04 14:39:20 +00:00
// av_log(avccontext, AV_LOG_DEBUG, "E%d\n", l);
}
2004-04-04 17:06:30 +00:00
return l ;
2002-09-01 18:07:56 +00:00
}
2008-03-21 03:11:20 +00:00
static av_cold int oggvorbis_encode_close ( AVCodecContext * avccontext ) {
2002-09-01 18:07:56 +00:00
OggVorbisContext * context = avccontext - > priv_data ;
/* ogg_packet op ; */
2005-12-17 18:14:38 +00:00
2002-09-01 18:07:56 +00:00
vorbis_analysis_wrote ( & context - > vd , 0 ) ; /* notify vorbisenc this is EOF */
vorbis_block_clear ( & context - > vb ) ;
vorbis_dsp_clear ( & context - > vd ) ;
vorbis_info_clear ( & context - > vi ) ;
2002-12-09 12:03:43 +00:00
av_freep ( & avccontext - > coded_frame ) ;
2004-04-04 15:19:20 +00:00
av_freep ( & avccontext - > extradata ) ;
2005-12-17 18:14:38 +00:00
2002-09-01 18:07:56 +00:00
return 0 ;
}
2011-01-25 21:40:11 +00:00
AVCodec ff_libvorbis_encoder = {
2007-05-21 12:54:12 +00:00
" libvorbis " ,
2010-03-30 23:30:55 +00:00
AVMEDIA_TYPE_AUDIO ,
2002-09-01 18:07:56 +00:00
CODEC_ID_VORBIS ,
sizeof ( OggVorbisContext ) ,
oggvorbis_encode_init ,
oggvorbis_encode_frame ,
2004-06-22 21:14:01 +00:00
oggvorbis_encode_close ,
. capabilities = CODEC_CAP_DELAY ,
2010-11-12 11:04:40 +00:00
. sample_fmts = ( const enum AVSampleFormat [ ] ) { AV_SAMPLE_FMT_S16 , AV_SAMPLE_FMT_NONE } ,
2008-06-12 21:50:13 +00:00
. long_name = NULL_IF_CONFIG_SMALL ( " libvorbis Vorbis " ) ,
2010-09-29 15:09:38 +00:00
. priv_class = & class ,
2002-11-22 07:27:13 +00:00
} ;