Original Commit: r7 | ods15 | 2006-09-17 08:50:47 +0300 (Sun, 17 Sep 2006) | 2 lines
[ffmpeg.git] / libavcodec / vorbis_enc.c
1 /*
2  * copyright (c) 2006 Oded Shimon <ods15@ods15.dyndns.org>
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17  */
18
19 /**
20  * @file vorbis_enc.c
21  * Native Vorbis encoder.
22  * @author Oded Shimon <ods15@ods15.dyndns.org>
23  */
24
25 #include "avcodec.h"
26
27 #undef NDEBUG
28 #include <assert.h>
29
30 #define ALT_BITSTREAM_READER_LE
31 #include "bitstream.h"
32
33 #define VORBIS_FRAME_SIZE 64
34
35 #define BUFFER_SIZE (1024*64)
36
37 typedef struct {
38     int len;
39     uint32_t codeword;
40 } entry_t;
41
42 typedef struct {
43     int nentries;
44     entry_t * entries;
45     int ndimentions;
46     float min;
47     float delta;
48     int seq_p;
49     int lookup;
50     //float * dimentions;
51     int * quantlist;
52 } codebook_t;
53
54 typedef struct {
55 } floor_t;
56
57 typedef struct {
58 } residue_t;
59
60 typedef struct {
61 } mapping_t;
62
63 typedef struct {
64     int channels;
65     int sample_rate;
66     int blocksize[2];
67
68     int ncodebooks;
69     codebook_t * codebooks;
70
71     int nfloors;
72     floor_t * floors;
73
74     int nresidues;
75     residue_t * residues;
76
77     int nmappings;
78     mapping_t * mappings;
79 } venc_context_t;
80
81 static inline int ilog(unsigned int a) {
82     int i;
83     for (i = 0; a >> i; i++);
84     return i;
85 }
86
87 static void put_float(PutBitContext * pb, float f) {
88     int exp, mant;
89     uint32_t res = 0;
90     mant = (int)ldexp(frexp(f, &exp), 20);
91     exp += 788 - 20;
92     if (mant < 0) { res |= (1 << 31); mant = -mant; }
93     res |= mant | (exp << 21);
94     put_bits(pb, 32, res);
95 }
96
97 static void put_codebook_header(PutBitContext * pb, codebook_t * cb) {
98     int i;
99     int ordered = 0;
100
101     put_bits(pb, 24, 0x564342); //magic
102     put_bits(pb, 16, cb->ndimentions);
103     put_bits(pb, 24, cb->nentries);
104
105     for (i = 1; i < cb->nentries; i++) if (cb->entries[i].len < cb->entries[i-1].len) break;
106     if (i == cb->nentries) ordered = 1;
107
108     put_bits(pb, 1, ordered);
109     if (ordered) {
110         int len = cb->entries[0].len;
111         put_bits(pb, 5, len);
112         i = 0;
113         while (i < cb->nentries) {
114             int j;
115             for (j = 0; j+i < cb->nentries; j++) if (cb->entries[j+i].len != len) break;
116             put_bits(pb, 5, j);
117             i += j;
118             len++;
119         }
120     } else {
121         int sparse = 0;
122         for (i = 0; i < cb->nentries; i++) if (!cb->entries[i].len) break;
123         if (i != cb->nentries) sparse = 1;
124         put_bits(pb, 1, sparse);
125
126         for (i = 0; i < cb->nentries; i++) {
127             if (sparse) put_bits(pb, 1, !!cb->entries[i].len);
128             if (cb->entries[i].len) put_bits(pb, 5, cb->entries[i].len);
129         }
130     }
131
132     put_bits(pb, 4, cb->lookup);
133     if (cb->lookup) {
134         int tmp, bits = ilog(cb->quantlist[0]);
135
136         if (cb->lookup == 1) {
137             for (tmp = 0; ; tmp++) {
138                 int n = 1;
139                 for (i = 0; i < cb->ndimentions; i++) n *= tmp;
140                 if (n > cb->nentries) break;
141             }
142             tmp--;
143         } else tmp = cb->ndimentions * cb->nentries;
144
145         for (i = 1; i < tmp; i++) bits = FFMIN(bits, ilog(cb->quantlist[i]));
146
147         put_float(pb, cb->min);
148         put_float(pb, cb->delta);
149
150         put_bits(pb, 4, bits - 1);
151         put_bits(pb, 1, cb->seq_p);
152
153         for (i = 0; i < tmp; i++) put_bits(pb, bits, cb->quantlist[i]);
154     }
155 }
156
157 static void put_floor_header(PutBitContext * pb, floor_t * fl) {
158 }
159
160 static void put_residue_header(PutBitContext * pb, residue_t * r) {
161 }
162
163 static int put_main_header(venc_context_t * venc, uint8_t ** out) {
164     int i;
165     PutBitContext pb;
166     uint8_t buffer[50000] = {0}, * p = buffer;
167     int buffer_len = sizeof buffer;
168     int len, hlens[3];
169
170     // identification header
171     init_put_bits(&pb, p, buffer_len);
172     put_bits(&pb, 8, 1); //magic
173     for (i = 0; "vorbis"[i]; i++) put_bits(&pb, 8, "vorbis"[i]);
174     put_bits(&pb, 32, 0); // version
175     put_bits(&pb, 8, venc->channels);
176     put_bits(&pb, 32, venc->sample_rate);
177     put_bits(&pb, 32, 0); // bitrate
178     put_bits(&pb, 32, 0); // bitrate
179     put_bits(&pb, 32, 0); // bitrate
180     put_bits(&pb, 4, venc->blocksize[0]);
181     put_bits(&pb, 4, venc->blocksize[1]);
182     put_bits(&pb, 1, 1); // framing
183
184     flush_put_bits(&pb);
185     hlens[0] = (put_bits_count(&pb) + 7) / 8;
186     buffer_len -= hlens[0];
187     p += hlens[0];
188
189     // comment header
190     init_put_bits(&pb, p, buffer_len);
191     put_bits(&pb, 8, 3); //magic
192     for (i = 0; "vorbis"[i]; i++) put_bits(&pb, 8, "vorbis"[i]);
193     put_bits(&pb, 32, 0); // vendor length TODO
194     put_bits(&pb, 32, 0); // amount of comments
195     put_bits(&pb, 1, 1); // framing
196
197     flush_put_bits(&pb);
198     hlens[1] = (put_bits_count(&pb) + 7) / 8;
199     buffer_len -= hlens[1];
200     p += hlens[1];
201
202     // setup header
203     init_put_bits(&pb, p, buffer_len);
204     put_bits(&pb, 8, 5); //magic
205     for (i = 0; "vorbis"[i]; i++) put_bits(&pb, 8, "vorbis"[i]);
206
207     // codebooks
208     put_bits(&pb, 8, venc->ncodebooks - 1);
209     for (i = 0; i < venc->ncodebooks; i++) put_codebook_header(&pb, &venc->codebooks[0]);
210
211     // time domain, reserved, zero
212     put_bits(&pb, 6, 0);
213     put_bits(&pb, 16, 0);
214
215     // floors
216     put_bits(&pb, 6, venc->nfloors - 1);
217     for (i = 0; i < venc->nfloors; i++) put_floor_header(&pb, &venc->floors[0]);
218
219     // residues
220     put_bits(&pb, 6, venc->nresidues - 1);
221     for (i = 0; i < venc->nresidues; i++) put_residue_header(&pb, &venc->residues[0]);
222
223     // mappings
224     put_bits(&pb, 6, venc->nmappings - 1);
225     for (i = 0; i < venc->nmappings; i++) {
226     }
227
228     flush_put_bits(&pb);
229     hlens[2] = (put_bits_count(&pb) + 7) / 8;
230
231     len = hlens[0] + hlens[1] + hlens[2];
232     p = *out = av_mallocz(64 + len + len/255);
233
234     *p++ = 2;
235     p += av_xiphlacing(p, hlens[0]);
236     p += av_xiphlacing(p, hlens[1]);
237     buffer_len = 0;
238     for (i = 0; i < 3; i++) {
239         memcpy(p, buffer + buffer_len, hlens[i]);
240         p += hlens[i];
241         buffer_len += hlens[i];
242     }
243
244     return p - *out;
245 }
246
247 static int vorbis_encode_init(AVCodecContext * avccontext)
248 {
249     venc_context_t * venc = avccontext->priv_data;
250
251     venc->channels = avccontext->channels;
252     venc->sample_rate = avccontext->sample_rate;
253
254     //if (avccontext->flags & CODEC_FLAG_QSCALE) avccontext->global_quality / (float)FF_QP2LAMBDA); else avccontext->bit_rate;
255     //if(avccontext->cutoff > 0) cfreq = avccontext->cutoff / 1000.0;
256
257     avccontext->extradata_size = put_main_header(venc, (uint8_t**)&avccontext->extradata);
258
259     avccontext->frame_size = VORBIS_FRAME_SIZE;
260
261     avccontext->coded_frame = avcodec_alloc_frame();
262     avccontext->coded_frame->key_frame = 1;
263
264     return 0;
265 }
266
267
268 static int vorbis_encode_frame(AVCodecContext * avccontext, unsigned char * packets, int buf_size, void *data)
269 {
270 #if 0
271     venc_context_t * venc = avccontext->priv_data;
272     signed short * audio = data;
273     int samples = data ? VORBIS_FRAME_SIZE : 0;
274
275     avccontext->coded_frame->pts = av_rescale_q(op2->granulepos, (AVRational){1, avccontext->sample_rate}, avccontext->time_base);
276     memcpy(packets, compressed_frame, l);
277 #endif
278     return 0;
279 }
280
281
282 static int vorbis_encode_close(AVCodecContext * avccontext)
283 {
284     venc_context_t * venc = avccontext->priv_data;
285
286     av_freep(&avccontext->coded_frame);
287     av_freep(&avccontext->extradata);
288
289     return 0 ;
290 }
291
292 AVCodec oggvorbis_encoder = {
293     "vorbis",
294     CODEC_TYPE_AUDIO,
295     CODEC_ID_VORBIS,
296     sizeof(venc_context_t),
297     vorbis_encode_init,
298     vorbis_encode_frame,
299     vorbis_encode_close,
300     .capabilities= CODEC_CAP_DELAY,
301 };