AVVideoFrame -> AVFrame
[ffmpeg.git] / libavcodec / oggvorbis.c
1 /*
2  * Ogg Vorbis codec support via libvorbisenc
3  * Mark Hills <mark@pogo.org.uk>
4  */
5
6 #include <time.h>
7
8 #include <vorbis/vorbisenc.h>
9
10 #include "avcodec.h"
11 #include "oggvorbis.h"
12
13 #define OGGVORBIS_FRAME_SIZE 1024
14
15
16 typedef struct OggVorbisContext {
17     vorbis_info vi ;
18     vorbis_dsp_state vd ;
19     vorbis_block vb ;
20
21     /* decoder */
22     vorbis_comment vc ;
23 } OggVorbisContext ;
24
25
26 int oggvorbis_init_encoder(vorbis_info *vi, AVCodecContext *avccontext) {
27     if(avccontext->coded_frame->quality) /* VBR requested */
28         return vorbis_encode_init_vbr(vi, avccontext->channels,
29                   avccontext->sample_rate, (float)avccontext->coded_frame->quality / 1000) ;
30
31     return vorbis_encode_init(vi, avccontext->channels,
32                   avccontext->sample_rate, -1, avccontext->bit_rate, -1) ;
33 }
34
35
36 static int oggvorbis_encode_init(AVCodecContext *avccontext) {
37     OggVorbisContext *context = avccontext->priv_data ;
38
39     vorbis_info_init(&context->vi) ;
40     if(oggvorbis_init_encoder(&context->vi, avccontext) < 0) {
41         fprintf(stderr, "oggvorbis_encode_init: init_encoder failed") ;
42         return -1 ;
43     }
44     vorbis_analysis_init(&context->vd, &context->vi) ;
45     vorbis_block_init(&context->vd, &context->vb) ;
46
47     avccontext->frame_size = OGGVORBIS_FRAME_SIZE ;
48  
49     avccontext->coded_frame= avcodec_alloc_frame();
50     avccontext->coded_frame->key_frame= 1;
51     
52     return 0 ;
53 }
54
55
56 static int oggvorbis_encode_frame(AVCodecContext *avccontext,
57                                   unsigned char *packets,
58                            int buf_size, void *data)
59 {
60     OggVorbisContext *context = avccontext->priv_data ;
61     float **buffer ;
62     ogg_packet op ;
63     signed char *audio = data ;
64     int l, samples = OGGVORBIS_FRAME_SIZE ;
65
66     buffer = vorbis_analysis_buffer(&context->vd, samples) ;
67
68     if(context->vi.channels == 1) {
69         for(l = 0 ; l < samples ; l++)
70             buffer[0][l]=((audio[l*2+1]<<8)|(0x00ff&(int)audio[l*2]))/32768.f;
71     } else {
72         for(l = 0 ; l < samples ; l++){
73             buffer[0][l]=((audio[l*4+1]<<8)|(0x00ff&(int)audio[l*4]))/32768.f;
74             buffer[1][l]=((audio[l*4+3]<<8)|(0x00ff&(int)audio[l*4+2]))/32768.f;
75         }
76     }
77     
78     vorbis_analysis_wrote(&context->vd, samples) ; 
79
80     l = 0 ;
81
82     while(vorbis_analysis_blockout(&context->vd, &context->vb) == 1) {
83         vorbis_analysis(&context->vb, NULL);
84         vorbis_bitrate_addblock(&context->vb) ;
85
86         while(vorbis_bitrate_flushpacket(&context->vd, &op)) {
87             memcpy(packets + l, &op, sizeof(ogg_packet)) ;
88             memcpy(packets + l + sizeof(ogg_packet), op.packet, op.bytes) ;
89             l += sizeof(ogg_packet) + op.bytes ;
90         }
91     }
92
93     return l ;
94 }
95
96
97 static int oggvorbis_encode_close(AVCodecContext *avccontext) {
98     OggVorbisContext *context = avccontext->priv_data ;
99 /*  ogg_packet op ; */
100     
101     vorbis_analysis_wrote(&context->vd, 0) ; /* notify vorbisenc this is EOF */
102
103     /* We need to write all the remaining packets into the stream
104      * on closing */
105     
106     fprintf(stderr, "fixme: not all packets written on oggvorbis_encode_close()\n") ;
107
108 /*
109     while(vorbis_bitrate_flushpacket(&context->vd, &op)) {
110         memcpy(packets + l, &op, sizeof(ogg_packet)) ;
111         memcpy(packets + l + sizeof(ogg_packet), op.packet, op.bytes) ;
112         l += sizeof(ogg_packet) + op.bytes ;    
113     }
114 */
115
116     vorbis_block_clear(&context->vb);
117     vorbis_dsp_clear(&context->vd);
118     vorbis_info_clear(&context->vi);
119
120     av_freep(&avccontext->coded_frame);
121   
122     return 0 ;
123 }
124
125
126 AVCodec oggvorbis_encoder = {
127     "vorbis",
128     CODEC_TYPE_AUDIO,
129     CODEC_ID_VORBIS,
130     sizeof(OggVorbisContext),
131     oggvorbis_encode_init,
132     oggvorbis_encode_frame,
133     oggvorbis_encode_close
134 } ;
135
136
137 static int oggvorbis_decode_init(AVCodecContext *avccontext) {
138     OggVorbisContext *context = avccontext->priv_data ;
139
140     vorbis_info_init(&context->vi) ;
141     vorbis_comment_init(&context->vc) ;
142
143     return 0 ;
144 }
145
146
147 static inline int conv(int samples, float **pcm, char *buf, int channels) {
148     int i, j, val ;
149     ogg_int16_t *ptr, *data = (ogg_int16_t*)buf ;
150     float *mono ;
151  
152     for(i = 0 ; i < channels ; i++){
153         ptr = &data[i];
154         mono = pcm[i] ;
155         
156         for(j = 0 ; j < samples ; j++) {
157             
158             val = mono[j] * 32767.f;
159             
160             if(val > 32767) val = 32767 ;
161             if(val < -32768) val = -32768 ;
162                     
163             *ptr = val ;
164             ptr += channels;
165         }
166     }
167     
168     return 0 ;
169 }
170            
171         
172 static int oggvorbis_decode_frame(AVCodecContext *avccontext,
173                         void *data, int *data_size,
174                         UINT8 *buf, int buf_size)
175 {
176     OggVorbisContext *context = avccontext->priv_data ;
177     ogg_packet *op = (ogg_packet*)buf ;
178     float **pcm ;
179     int samples, total_samples, total_bytes ;
180  
181     op->packet = (char*)op + sizeof(ogg_packet) ; /* correct data pointer */
182
183     if(op->packetno < 3) {
184         vorbis_synthesis_headerin(&context->vi, &context->vc, op) ;
185         return buf_size ;
186     }
187
188     if(op->packetno == 3) {
189         fprintf(stderr, "vorbis_decode: %d channel, %ldHz, encoder `%s'\n",
190                 context->vi.channels, context->vi.rate, context->vc.vendor);
191
192         avccontext->channels = context->vi.channels ;
193         avccontext->sample_rate = context->vi.rate ;
194         
195         vorbis_synthesis_init(&context->vd, &context->vi) ;
196         vorbis_block_init(&context->vd, &context->vb); 
197     }
198
199     if(vorbis_synthesis(&context->vb, op) == 0)
200         vorbis_synthesis_blockin(&context->vd, &context->vb) ;
201     
202     total_samples = 0 ;
203     total_bytes = 0 ;
204
205     while((samples = vorbis_synthesis_pcmout(&context->vd, &pcm)) > 0) {
206         conv(samples, pcm, (char*)data + total_bytes, context->vi.channels) ;
207         total_bytes += samples * 2 * context->vi.channels ;
208         total_samples += samples ;
209         vorbis_synthesis_read(&context->vd, samples) ;
210     }
211
212     *data_size = total_bytes ;   
213     return buf_size ;
214 }
215
216
217 static int oggvorbis_decode_close(AVCodecContext *avccontext) {
218     OggVorbisContext *context = avccontext->priv_data ;
219    
220     vorbis_info_clear(&context->vi) ;
221     vorbis_comment_clear(&context->vc) ;
222
223     return 0 ;
224 }
225
226
227 AVCodec oggvorbis_decoder = {
228     "vorbis",
229     CODEC_TYPE_AUDIO,
230     CODEC_ID_VORBIS,
231     sizeof(OggVorbisContext),
232     oggvorbis_decode_init,
233     NULL,
234     oggvorbis_decode_close,
235     oggvorbis_decode_frame,
236 } ;