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"
2012-02-07 20:37:45 +00:00
# include "internal.h"
2010-06-27 09:25:05 +00:00
# include "vorbis.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
2011-12-29 20:37:05 +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 ;
2011-12-29 20:37:05 +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 */
2011-12-29 20:37:05 +00:00
vorbis_comment vc ;
2004-04-04 02:07:15 +00:00
ogg_packet op ;
2010-09-29 15:09:38 +00:00
double iblock ;
2011-12-29 20:37:05 +00:00
} OggVorbisContext ;
2002-09-01 18:07:56 +00:00
2011-12-29 20:37:05 +00:00
static const AVOption options [ ] = {
{ " iblock " , " Sets the impulse block bias " , offsetof ( OggVorbisContext , iblock ) , AV_OPT_TYPE_DOUBLE , { . dbl = 0 } , - 15 , 0 , AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_ENCODING_PARAM } ,
{ NULL }
2010-09-29 15:09:38 +00:00
} ;
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
2011-12-29 20:37:05 +00:00
static av_cold int oggvorbis_init_encoder ( vorbis_info * vi , AVCodecContext * avccontext )
{
OggVorbisContext * context = avccontext - > priv_data ;
2006-01-21 17:09:23 +00:00
double cfreq ;
2002-12-21 15:54:21 +00:00
2011-12-29 20:37:05 +00:00
if ( avccontext - > flags & CODEC_FLAG_QSCALE ) {
2006-01-21 17:09:23 +00:00
/* variable bitrate */
2011-12-29 20:37:05 +00:00
if ( vorbis_encode_setup_vbr ( vi , avccontext - > channels ,
avccontext - > sample_rate ,
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 */
2011-12-29 20:37:05 +00:00
if ( vorbis_encode_setup_managed ( vi , avccontext - > channels ,
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 */
2011-12-29 20:37:05 +00:00
if ( minrate = = - 1 & & maxrate = = - 1 )
if ( vorbis_encode_ctl ( vi , OV_ECTL_RATEMANAGE2_SET , NULL ) )
2010-06-04 22:40:40 +00:00
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 */
2011-12-29 20:37:05 +00:00
if ( avccontext - > cutoff > 0 ) {
2006-01-21 17:09:23 +00:00
cfreq = avccontext - > cutoff / 1000.0 ;
2011-12-29 20:37:05 +00:00
if ( vorbis_encode_ctl ( vi , OV_ECTL_LOWPASS_SET , & cfreq ) )
2006-01-21 17:09:23 +00:00
return - 1 ;
}
2002-09-01 18:07:56 +00:00
2011-12-29 20:37:05 +00:00
if ( context - > iblock ) {
2010-09-29 15:09:38 +00:00
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' */
2011-12-29 20:37:05 +00:00
static int xiph_len ( int l )
{
2011-12-29 21:23:16 +00:00
return 1 + l / 255 + l ;
2011-12-29 20:37:05 +00:00
}
2010-07-20 21:54:46 +00:00
2011-12-29 20:37:05 +00:00
static av_cold int oggvorbis_encode_init ( AVCodecContext * avccontext )
{
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
2011-12-29 20:37:05 +00:00
vorbis_info_init ( & context - > vi ) ;
if ( oggvorbis_init_encoder ( & context - > vi , avccontext ) < 0 ) {
av_log ( avccontext , AV_LOG_ERROR , " oggvorbis_encode_init: init_encoder failed \n " ) ;
return - 1 ;
2002-09-01 18:07:56 +00:00
}
2011-12-29 20:37:05 +00:00
vorbis_analysis_init ( & context - > vd , & context - > vi ) ;
vorbis_block_init ( & context - > vd , & context - > vb ) ;
2002-09-01 18:07:56 +00:00
2004-04-04 15:19:20 +00:00
vorbis_comment_init ( & context - > vc ) ;
2011-12-29 20:37:05 +00:00
vorbis_comment_add_tag ( & context - > vc , " encoder " , LIBAVCODEC_IDENT ) ;
2004-04-04 15:19:20 +00:00
vorbis_analysis_headerout ( & context - > vd , & context - > vc , & header ,
2011-12-29 20:37:05 +00:00
& header_comm , & header_code ) ;
2005-12-17 18:14:38 +00:00
2011-12-29 20:37:05 +00:00
avccontext - > extradata_size =
2010-07-20 21:54:46 +00:00
1 + xiph_len ( header . bytes ) + xiph_len ( header_comm . bytes ) +
header_code . bytes ;
p = avccontext - > extradata =
2011-12-29 20:37:05 +00:00
av_malloc ( avccontext - > extradata_size + FF_INPUT_BUFFER_PADDING_SIZE ) ;
p [ 0 ] = 2 ;
offset = 1 ;
2005-05-13 18:10:23 +00:00
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
2011-12-29 20:37:05 +00:00
#if 0
vorbis_block_clear ( & context - > vb ) ;
2004-04-04 15:19:20 +00:00
vorbis_dsp_clear ( & context - > vd ) ;
2011-12-29 20:37:05 +00:00
vorbis_info_clear ( & context - > vi ) ;
# endif
2004-04-04 15:19:20 +00:00
vorbis_comment_clear ( & context - > vc ) ;
2005-12-17 18:14:38 +00:00
2011-12-29 20:37:05 +00:00
avccontext - > frame_size = OGGVORBIS_FRAME_SIZE ;
2005-12-17 18:14:38 +00:00
2011-12-29 20:37:05 +00:00
avccontext - > coded_frame = avcodec_alloc_frame ( ) ;
avccontext - > coded_frame - > key_frame = 1 ;
2005-12-17 18:14:38 +00:00
2011-12-29 20:37:05 +00:00
return 0 ;
2002-09-01 18:07:56 +00:00
}
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 ,
2011-12-29 20:37:05 +00:00
int buf_size , void * data )
2002-09-01 18:07:56 +00:00
{
2011-12-29 20:37:05 +00:00
OggVorbisContext * context = avccontext - > priv_data ;
ogg_packet op ;
signed short * audio = data ;
2008-10-15 07:29:37 +00:00
int l ;
2011-12-29 20:37:05 +00:00
if ( data ) {
2010-07-11 06:40:05 +00:00
const int samples = avccontext - > frame_size ;
2011-12-29 20:37:05 +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
2011-12-29 20:37:05 +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 :
2011-12-29 20:37:05 +00:00
ff_vorbis_encoding_channel_layout_offsets [ channels - 1 ] [ c ] ;
for ( l = 0 ; l < samples ; l + + )
buffer [ c ] [ l ] = audio [ l * channels + co ] / 32768.f ;
2005-12-22 01:10:11 +00:00
}
2011-12-29 20:37:05 +00:00
vorbis_analysis_wrote ( & context - > vd , samples ) ;
2008-10-15 07:29:37 +00:00
} else {
2011-12-29 20:37:05 +00:00
if ( ! context - > eof )
vorbis_analysis_wrote ( & context - > vd , 0 ) ;
2008-10-15 07:29:37 +00:00
context - > eof = 1 ;
}
2002-09-01 18:07:56 +00:00
2011-12-29 20:37:05 +00:00
while ( vorbis_analysis_blockout ( & context - > vd , & context - > vb ) = = 1 ) {
2005-12-22 01:10:11 +00:00
vorbis_analysis ( & context - > vb , NULL ) ;
2011-12-29 20:37:05 +00:00
vorbis_bitrate_addblock ( & context - > vb ) ;
2002-09-01 18:07:56 +00:00
2011-12-29 20:37:05 +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 . */
2011-12-29 20:37:05 +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
}
2011-12-29 20:37:05 +00:00
l = 0 ;
if ( context - > buffer_index ) {
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
2011-12-29 20:37:05 +00:00
l = op2 - > bytes ;
2012-02-07 20:37:45 +00:00
avccontext - > coded_frame - > pts = ff_samples_to_time_base ( avccontext ,
op2 - > granulepos ) ;
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
}
2011-12-29 20:37:05 +00:00
static av_cold int oggvorbis_encode_close ( AVCodecContext * avccontext )
{
OggVorbisContext * context = avccontext - > priv_data ;
2002-09-01 18:07:56 +00:00
/* ogg_packet op ; */
2005-12-17 18:14:38 +00:00
2011-12-29 20:37:05 +00:00
vorbis_analysis_wrote ( & context - > vd , 0 ) ; /* notify vorbisenc this is EOF */
2002-09-01 18:07:56 +00:00
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
2011-12-29 20:37:05 +00:00
return 0 ;
2002-09-01 18:07:56 +00:00
}
2011-01-25 21:40:11 +00:00
AVCodec ff_libvorbis_encoder = {
2011-09-23 19:11:15 +00:00
. name = " libvorbis " ,
. type = AVMEDIA_TYPE_AUDIO ,
. id = CODEC_ID_VORBIS ,
. priv_data_size = sizeof ( OggVorbisContext ) ,
. init = oggvorbis_encode_init ,
. encode = oggvorbis_encode_frame ,
. close = oggvorbis_encode_close ,
. capabilities = CODEC_CAP_DELAY ,
2011-12-29 20:37:05 +00:00
. sample_fmts = ( const enum AVSampleFormat [ ] ) { AV_SAMPLE_FMT_S16 , AV_SAMPLE_FMT_NONE } ,
2011-09-23 19:11:15 +00:00
. long_name = NULL_IF_CONFIG_SMALL ( " libvorbis Vorbis " ) ,
. priv_class = & class ,
} ;