| @@ -37,63 +37,65 @@ | |||||
| #define OGGVORBIS_FRAME_SIZE 64 | #define OGGVORBIS_FRAME_SIZE 64 | ||||
| #define BUFFER_SIZE (1024*64) | |||||
| #define BUFFER_SIZE (1024 * 64) | |||||
| typedef struct OggVorbisContext { | typedef struct OggVorbisContext { | ||||
| AVClass *av_class; | AVClass *av_class; | ||||
| vorbis_info vi ; | |||||
| vorbis_dsp_state vd ; | |||||
| vorbis_block vb ; | |||||
| vorbis_info vi; | |||||
| vorbis_dsp_state vd; | |||||
| vorbis_block vb; | |||||
| uint8_t buffer[BUFFER_SIZE]; | uint8_t buffer[BUFFER_SIZE]; | ||||
| int buffer_index; | int buffer_index; | ||||
| int eof; | int eof; | ||||
| /* decoder */ | /* decoder */ | ||||
| vorbis_comment vc ; | |||||
| vorbis_comment vc; | |||||
| ogg_packet op; | ogg_packet op; | ||||
| double iblock; | double iblock; | ||||
| } OggVorbisContext ; | |||||
| } OggVorbisContext; | |||||
| 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} | |||||
| 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 } | |||||
| }; | }; | ||||
| static const AVClass class = { "libvorbis", av_default_item_name, options, LIBAVUTIL_VERSION_INT }; | static const AVClass class = { "libvorbis", av_default_item_name, options, LIBAVUTIL_VERSION_INT }; | ||||
| static av_cold int oggvorbis_init_encoder(vorbis_info *vi, AVCodecContext *avccontext) { | |||||
| OggVorbisContext *context = avccontext->priv_data ; | |||||
| static av_cold int oggvorbis_init_encoder(vorbis_info *vi, AVCodecContext *avccontext) | |||||
| { | |||||
| OggVorbisContext *context = avccontext->priv_data; | |||||
| double cfreq; | double cfreq; | ||||
| if(avccontext->flags & CODEC_FLAG_QSCALE) { | |||||
| if (avccontext->flags & CODEC_FLAG_QSCALE) { | |||||
| /* variable bitrate */ | /* variable bitrate */ | ||||
| if(vorbis_encode_setup_vbr(vi, avccontext->channels, | |||||
| avccontext->sample_rate, | |||||
| avccontext->global_quality / (float)FF_QP2LAMBDA / 10.0)) | |||||
| if (vorbis_encode_setup_vbr(vi, avccontext->channels, | |||||
| avccontext->sample_rate, | |||||
| avccontext->global_quality / (float)FF_QP2LAMBDA / 10.0)) | |||||
| return -1; | return -1; | ||||
| } else { | } else { | ||||
| int minrate = avccontext->rc_min_rate > 0 ? avccontext->rc_min_rate : -1; | int minrate = avccontext->rc_min_rate > 0 ? avccontext->rc_min_rate : -1; | ||||
| int maxrate = avccontext->rc_min_rate > 0 ? avccontext->rc_max_rate : -1; | int maxrate = avccontext->rc_min_rate > 0 ? avccontext->rc_max_rate : -1; | ||||
| /* constant bitrate */ | /* constant bitrate */ | ||||
| if(vorbis_encode_setup_managed(vi, avccontext->channels, | |||||
| avccontext->sample_rate, minrate, avccontext->bit_rate, maxrate)) | |||||
| if (vorbis_encode_setup_managed(vi, avccontext->channels, | |||||
| avccontext->sample_rate, minrate, | |||||
| avccontext->bit_rate, maxrate)) | |||||
| return -1; | return -1; | ||||
| /* variable bitrate by estimate, disable slow rate management */ | /* variable bitrate by estimate, disable slow rate management */ | ||||
| if(minrate == -1 && maxrate == -1) | |||||
| if(vorbis_encode_ctl(vi, OV_ECTL_RATEMANAGE2_SET, NULL)) | |||||
| if (minrate == -1 && maxrate == -1) | |||||
| if (vorbis_encode_ctl(vi, OV_ECTL_RATEMANAGE2_SET, NULL)) | |||||
| return -1; | return -1; | ||||
| } | } | ||||
| /* cutoff frequency */ | /* cutoff frequency */ | ||||
| if(avccontext->cutoff > 0) { | |||||
| if (avccontext->cutoff > 0) { | |||||
| cfreq = avccontext->cutoff / 1000.0; | cfreq = avccontext->cutoff / 1000.0; | ||||
| if(vorbis_encode_ctl(vi, OV_ECTL_LOWPASS_SET, &cfreq)) | |||||
| if (vorbis_encode_ctl(vi, OV_ECTL_LOWPASS_SET, &cfreq)) | |||||
| return -1; | return -1; | ||||
| } | } | ||||
| if(context->iblock){ | |||||
| if (context->iblock) { | |||||
| vorbis_encode_ctl(vi, OV_ECTL_IBLOCK_SET, &context->iblock); | vorbis_encode_ctl(vi, OV_ECTL_IBLOCK_SET, &context->iblock); | ||||
| } | } | ||||
| @@ -101,35 +103,39 @@ static av_cold int oggvorbis_init_encoder(vorbis_info *vi, AVCodecContext *avcco | |||||
| } | } | ||||
| /* How many bytes are needed for a buffer of length 'l' */ | /* How many bytes are needed for a buffer of length 'l' */ | ||||
| static int xiph_len(int l) { return (1 + l / 255 + l); } | |||||
| static int xiph_len(int l) | |||||
| { | |||||
| return (1 + l / 255 + l); | |||||
| } | |||||
| static av_cold int oggvorbis_encode_init(AVCodecContext *avccontext) { | |||||
| OggVorbisContext *context = avccontext->priv_data ; | |||||
| static av_cold int oggvorbis_encode_init(AVCodecContext *avccontext) | |||||
| { | |||||
| OggVorbisContext *context = avccontext->priv_data; | |||||
| ogg_packet header, header_comm, header_code; | ogg_packet header, header_comm, header_code; | ||||
| uint8_t *p; | uint8_t *p; | ||||
| unsigned int offset; | unsigned int offset; | ||||
| 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 ; | |||||
| 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; | |||||
| } | } | ||||
| vorbis_analysis_init(&context->vd, &context->vi) ; | |||||
| vorbis_block_init(&context->vd, &context->vb) ; | |||||
| vorbis_analysis_init(&context->vd, &context->vi); | |||||
| vorbis_block_init(&context->vd, &context->vb); | |||||
| vorbis_comment_init(&context->vc); | vorbis_comment_init(&context->vc); | ||||
| vorbis_comment_add_tag(&context->vc, "encoder", LIBAVCODEC_IDENT) ; | |||||
| vorbis_comment_add_tag(&context->vc, "encoder", LIBAVCODEC_IDENT); | |||||
| vorbis_analysis_headerout(&context->vd, &context->vc, &header, | vorbis_analysis_headerout(&context->vd, &context->vc, &header, | ||||
| &header_comm, &header_code); | |||||
| &header_comm, &header_code); | |||||
| avccontext->extradata_size= | |||||
| avccontext->extradata_size = | |||||
| 1 + xiph_len(header.bytes) + xiph_len(header_comm.bytes) + | 1 + xiph_len(header.bytes) + xiph_len(header_comm.bytes) + | ||||
| header_code.bytes; | header_code.bytes; | ||||
| p = avccontext->extradata = | p = avccontext->extradata = | ||||
| av_malloc(avccontext->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); | |||||
| p[0] = 2; | |||||
| offset = 1; | |||||
| av_malloc(avccontext->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE); | |||||
| p[0] = 2; | |||||
| offset = 1; | |||||
| offset += av_xiphlacing(&p[offset], header.bytes); | offset += av_xiphlacing(&p[offset], header.bytes); | ||||
| offset += av_xiphlacing(&p[offset], header_comm.bytes); | offset += av_xiphlacing(&p[offset], header_comm.bytes); | ||||
| memcpy(&p[offset], header.packet, header.bytes); | memcpy(&p[offset], header.packet, header.bytes); | ||||
| @@ -140,56 +146,57 @@ static av_cold int oggvorbis_encode_init(AVCodecContext *avccontext) { | |||||
| offset += header_code.bytes; | offset += header_code.bytes; | ||||
| assert(offset == avccontext->extradata_size); | assert(offset == avccontext->extradata_size); | ||||
| /* vorbis_block_clear(&context->vb); | |||||
| #if 0 | |||||
| vorbis_block_clear(&context->vb); | |||||
| vorbis_dsp_clear(&context->vd); | vorbis_dsp_clear(&context->vd); | ||||
| vorbis_info_clear(&context->vi);*/ | |||||
| vorbis_info_clear(&context->vi); | |||||
| #endif | |||||
| vorbis_comment_clear(&context->vc); | vorbis_comment_clear(&context->vc); | ||||
| avccontext->frame_size = OGGVORBIS_FRAME_SIZE ; | |||||
| avccontext->frame_size = OGGVORBIS_FRAME_SIZE; | |||||
| avccontext->coded_frame= avcodec_alloc_frame(); | |||||
| avccontext->coded_frame->key_frame= 1; | |||||
| avccontext->coded_frame = avcodec_alloc_frame(); | |||||
| avccontext->coded_frame->key_frame = 1; | |||||
| return 0 ; | |||||
| return 0; | |||||
| } | } | ||||
| static int oggvorbis_encode_frame(AVCodecContext *avccontext, | static int oggvorbis_encode_frame(AVCodecContext *avccontext, | ||||
| unsigned char *packets, | unsigned char *packets, | ||||
| int buf_size, void *data) | |||||
| int buf_size, void *data) | |||||
| { | { | ||||
| OggVorbisContext *context = avccontext->priv_data ; | |||||
| ogg_packet op ; | |||||
| signed short *audio = data ; | |||||
| OggVorbisContext *context = avccontext->priv_data; | |||||
| ogg_packet op; | |||||
| signed short *audio = data; | |||||
| int l; | int l; | ||||
| if(data) { | |||||
| if (data) { | |||||
| const int samples = avccontext->frame_size; | const int samples = avccontext->frame_size; | ||||
| float **buffer ; | |||||
| float **buffer; | |||||
| int c, channels = context->vi.channels; | int c, channels = context->vi.channels; | ||||
| buffer = vorbis_analysis_buffer(&context->vd, samples) ; | |||||
| buffer = vorbis_analysis_buffer(&context->vd, samples); | |||||
| for (c = 0; c < channels; c++) { | for (c = 0; c < channels; c++) { | ||||
| int co = (channels > 8) ? c : | int co = (channels > 8) ? c : | ||||
| ff_vorbis_encoding_channel_layout_offsets[channels-1][c]; | |||||
| for(l = 0 ; l < samples ; l++) | |||||
| buffer[c][l]=audio[l*channels+co]/32768.f; | |||||
| ff_vorbis_encoding_channel_layout_offsets[channels - 1][c]; | |||||
| for (l = 0; l < samples; l++) | |||||
| buffer[c][l] = audio[l * channels + co] / 32768.f; | |||||
| } | } | ||||
| vorbis_analysis_wrote(&context->vd, samples) ; | |||||
| vorbis_analysis_wrote(&context->vd, samples); | |||||
| } else { | } else { | ||||
| if(!context->eof) | |||||
| vorbis_analysis_wrote(&context->vd, 0) ; | |||||
| if (!context->eof) | |||||
| vorbis_analysis_wrote(&context->vd, 0); | |||||
| context->eof = 1; | context->eof = 1; | ||||
| } | } | ||||
| while(vorbis_analysis_blockout(&context->vd, &context->vb) == 1) { | |||||
| while (vorbis_analysis_blockout(&context->vd, &context->vb) == 1) { | |||||
| vorbis_analysis(&context->vb, NULL); | vorbis_analysis(&context->vb, NULL); | ||||
| vorbis_bitrate_addblock(&context->vb) ; | |||||
| vorbis_bitrate_addblock(&context->vb); | |||||
| while(vorbis_bitrate_flushpacket(&context->vd, &op)) { | |||||
| while (vorbis_bitrate_flushpacket(&context->vd, &op)) { | |||||
| /* i'd love to say the following line is a hack, but sadly it's | /* 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. */ | * not, apparently the end of stream decision is in libogg. */ | ||||
| if(op.bytes==1 && op.e_o_s) | |||||
| if (op.bytes == 1 && op.e_o_s) | |||||
| continue; | continue; | ||||
| if (context->buffer_index + sizeof(ogg_packet) + op.bytes > BUFFER_SIZE) { | if (context->buffer_index + sizeof(ogg_packet) + op.bytes > BUFFER_SIZE) { | ||||
| av_log(avccontext, AV_LOG_ERROR, "libvorbis: buffer overflow."); | av_log(avccontext, AV_LOG_ERROR, "libvorbis: buffer overflow."); | ||||
| @@ -203,13 +210,13 @@ static int oggvorbis_encode_frame(AVCodecContext *avccontext, | |||||
| } | } | ||||
| } | } | ||||
| l=0; | |||||
| if(context->buffer_index){ | |||||
| ogg_packet *op2= (ogg_packet*)context->buffer; | |||||
| l = 0; | |||||
| if (context->buffer_index) { | |||||
| ogg_packet *op2 = (ogg_packet *)context->buffer; | |||||
| op2->packet = context->buffer + sizeof(ogg_packet); | op2->packet = context->buffer + sizeof(ogg_packet); | ||||
| l= op2->bytes; | |||||
| avccontext->coded_frame->pts= av_rescale_q(op2->granulepos, (AVRational){1, avccontext->sample_rate}, avccontext->time_base); | |||||
| l = op2->bytes; | |||||
| avccontext->coded_frame->pts = av_rescale_q(op2->granulepos, (AVRational) { 1, avccontext->sample_rate }, avccontext->time_base); | |||||
| //FIXME we should reorder the user supplied pts and not assume that they are spaced by 1/sample_rate | //FIXME we should reorder the user supplied pts and not assume that they are spaced by 1/sample_rate | ||||
| if (l > buf_size) { | if (l > buf_size) { | ||||
| @@ -226,12 +233,12 @@ static int oggvorbis_encode_frame(AVCodecContext *avccontext, | |||||
| return l; | return l; | ||||
| } | } | ||||
| static av_cold int oggvorbis_encode_close(AVCodecContext *avccontext) { | |||||
| OggVorbisContext *context = avccontext->priv_data ; | |||||
| static av_cold int oggvorbis_encode_close(AVCodecContext *avccontext) | |||||
| { | |||||
| OggVorbisContext *context = avccontext->priv_data; | |||||
| /* ogg_packet op ; */ | /* ogg_packet op ; */ | ||||
| vorbis_analysis_wrote(&context->vd, 0) ; /* notify vorbisenc this is EOF */ | |||||
| vorbis_analysis_wrote(&context->vd, 0); /* notify vorbisenc this is EOF */ | |||||
| vorbis_block_clear(&context->vb); | vorbis_block_clear(&context->vb); | ||||
| vorbis_dsp_clear(&context->vd); | vorbis_dsp_clear(&context->vd); | ||||
| @@ -240,10 +247,9 @@ static av_cold int oggvorbis_encode_close(AVCodecContext *avccontext) { | |||||
| av_freep(&avccontext->coded_frame); | av_freep(&avccontext->coded_frame); | ||||
| av_freep(&avccontext->extradata); | av_freep(&avccontext->extradata); | ||||
| return 0 ; | |||||
| return 0; | |||||
| } | } | ||||
| AVCodec ff_libvorbis_encoder = { | AVCodec ff_libvorbis_encoder = { | ||||
| .name = "libvorbis", | .name = "libvorbis", | ||||
| .type = AVMEDIA_TYPE_AUDIO, | .type = AVMEDIA_TYPE_AUDIO, | ||||
| @@ -253,7 +259,7 @@ AVCodec ff_libvorbis_encoder = { | |||||
| .encode = oggvorbis_encode_frame, | .encode = oggvorbis_encode_frame, | ||||
| .close = oggvorbis_encode_close, | .close = oggvorbis_encode_close, | ||||
| .capabilities = CODEC_CAP_DELAY, | .capabilities = CODEC_CAP_DELAY, | ||||
| .sample_fmts = (const enum AVSampleFormat[]){AV_SAMPLE_FMT_S16,AV_SAMPLE_FMT_NONE}, | |||||
| .sample_fmts = (const enum AVSampleFormat[]) { AV_SAMPLE_FMT_S16, AV_SAMPLE_FMT_NONE }, | |||||
| .long_name = NULL_IF_CONFIG_SMALL("libvorbis Vorbis"), | .long_name = NULL_IF_CONFIG_SMALL("libvorbis Vorbis"), | ||||
| .priv_class = &class, | .priv_class = &class, | ||||
| }; | }; | ||||