e327e2fd3e0e33f6a39de8529d7b6d766b1166de
[libav.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 } ;