2 * Ogg Vorbis codec support via libvorbisenc
3 * Mark Hills <mark@pogo.org.uk>
8 #include <vorbis/vorbisenc.h>
11 #include "oggvorbis.h"
13 #define OGGVORBIS_FRAME_SIZE 1024
16 typedef struct OggVorbisContext {
23 int oggvorbis_init_encoder(vorbis_info *vi, AVCodecContext *avccontext) {
24 if(avccontext->quality) { /* VBR requested */
26 fprintf(stderr, "init_encode: channels=%d quality=%d\n",
27 avccontext->channels, avccontext->quality) ;
29 return vorbis_encode_init_vbr(vi, avccontext->channels,
30 avccontext->sample_rate, (float)avccontext->quality / 1000) ;
33 fprintf(stderr, "init_encoder: channels=%d bitrate=%d tolerance=%d\n",
34 avccontext->channels, avccontext->bit_rate,
35 avccontext->bit_rate_tolerance) ;
37 return vorbis_encode_init(vi, avccontext->channels,
38 avccontext->sample_rate, -1, avccontext->bit_rate, -1) ;
42 static int oggvorbis_encode_init(AVCodecContext *avccontext) {
43 OggVorbisContext *context = avccontext->priv_data ;
45 fprintf(stderr, "oggvorbis_encode_init\n") ;
47 vorbis_info_init(&context->vi) ;
49 if(oggvorbis_init_encoder(&context->vi, avccontext) < 0) {
50 fprintf(stderr, "oggvorbis_encode_init: init_encoder failed") ;
54 vorbis_analysis_init(&context->vd, &context->vi) ;
55 vorbis_block_init(&context->vd, &context->vb) ;
57 avccontext->frame_size = OGGVORBIS_FRAME_SIZE ;
63 int oggvorbis_encode_frame(AVCodecContext *avccontext, unsigned char *packets,
64 int buf_size, void *data)
66 OggVorbisContext *context = avccontext->priv_data ;
69 signed char *audio = data ;
70 int l, samples = buf_size / 16 ; /* samples = OGGVORBIS_FRAME_SIZE */ ;
72 buffer = vorbis_analysis_buffer(&context->vd, samples) ;
74 if(context->vi.channels == 1) {
75 for(l = 0 ; l < samples ; l++)
76 buffer[0][l]=((audio[l*2+1]<<8)|(0x00ff&(int)audio[l*2]))/32768.f;
78 for(l = 0 ; l < samples ; l++){
79 buffer[0][l]=((audio[l*4+1]<<8)|(0x00ff&(int)audio[l*4]))/32768.f;
80 buffer[1][l]=((audio[l*4+3]<<8)|(0x00ff&(int)audio[l*4+2]))/32768.f;
84 vorbis_analysis_wrote(&context->vd, samples) ;
88 while(vorbis_analysis_blockout(&context->vd, &context->vb) == 1) {
89 vorbis_analysis(&context->vb, NULL);
90 vorbis_bitrate_addblock(&context->vb) ;
92 while(vorbis_bitrate_flushpacket(&context->vd, &op)) {
93 memcpy(packets + l, &op, sizeof(ogg_packet)) ;
94 memcpy(packets + l + sizeof(ogg_packet), op.packet, op.bytes) ;
95 l += sizeof(ogg_packet) + op.bytes ;
103 int oggvorbis_encode_close(AVCodecContext *avccontext) {
104 OggVorbisContext *context = avccontext->priv_data ;
105 /* ogg_packet op ; */
107 fprintf(stderr, "oggvorbis_encode_close\n") ;
109 vorbis_analysis_wrote(&context->vd, 0) ; /* notify vorbisenc this is EOF */
111 /* We need to write all the remaining packets into the stream
115 while(vorbis_bitrate_flushpacket(&context->vd, &op)) {
116 memcpy(packets + l, &op, sizeof(ogg_packet)) ;
117 memcpy(packets + l + sizeof(ogg_packet), op.packet, op.bytes) ;
118 l += sizeof(ogg_packet) + op.bytes ;
122 vorbis_block_clear(&context->vb);
123 vorbis_dsp_clear(&context->vd);
124 vorbis_info_clear(&context->vi);
130 AVCodec oggvorbis_encoder = {
134 sizeof(OggVorbisContext),
135 oggvorbis_encode_init,
136 oggvorbis_encode_frame,
137 oggvorbis_encode_close