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