annotate wmaenc.c @ 11219:872413b133f4 libavcodec

msmpeg4v* encoders depend on h263dec
author daniel
date Fri, 19 Feb 2010 21:34:50 +0000
parents 964d01b50f17
children 8a4984c5cacc
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
1 /*
0efc832d9102 wma encoder
michael
parents:
diff changeset
2 * WMA compatible encoder
0efc832d9102 wma encoder
michael
parents:
diff changeset
3 * Copyright (c) 2007 Michael Niedermayer
0efc832d9102 wma encoder
michael
parents:
diff changeset
4 *
0efc832d9102 wma encoder
michael
parents:
diff changeset
5 * This file is part of FFmpeg.
0efc832d9102 wma encoder
michael
parents:
diff changeset
6 *
0efc832d9102 wma encoder
michael
parents:
diff changeset
7 * FFmpeg is free software; you can redistribute it and/or
0efc832d9102 wma encoder
michael
parents:
diff changeset
8 * modify it under the terms of the GNU Lesser General Public
0efc832d9102 wma encoder
michael
parents:
diff changeset
9 * License as published by the Free Software Foundation; either
0efc832d9102 wma encoder
michael
parents:
diff changeset
10 * version 2.1 of the License, or (at your option) any later version.
0efc832d9102 wma encoder
michael
parents:
diff changeset
11 *
0efc832d9102 wma encoder
michael
parents:
diff changeset
12 * FFmpeg is distributed in the hope that it will be useful,
0efc832d9102 wma encoder
michael
parents:
diff changeset
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
0efc832d9102 wma encoder
michael
parents:
diff changeset
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
0efc832d9102 wma encoder
michael
parents:
diff changeset
15 * Lesser General Public License for more details.
0efc832d9102 wma encoder
michael
parents:
diff changeset
16 *
0efc832d9102 wma encoder
michael
parents:
diff changeset
17 * You should have received a copy of the GNU Lesser General Public
0efc832d9102 wma encoder
michael
parents:
diff changeset
18 * License along with FFmpeg; if not, write to the Free Software
0efc832d9102 wma encoder
michael
parents:
diff changeset
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
0efc832d9102 wma encoder
michael
parents:
diff changeset
20 */
0efc832d9102 wma encoder
michael
parents:
diff changeset
21
0efc832d9102 wma encoder
michael
parents:
diff changeset
22 #include "avcodec.h"
0efc832d9102 wma encoder
michael
parents:
diff changeset
23 #include "wma.h"
0efc832d9102 wma encoder
michael
parents:
diff changeset
24
0efc832d9102 wma encoder
michael
parents:
diff changeset
25 #undef NDEBUG
0efc832d9102 wma encoder
michael
parents:
diff changeset
26 #include <assert.h>
0efc832d9102 wma encoder
michael
parents:
diff changeset
27
0efc832d9102 wma encoder
michael
parents:
diff changeset
28
0efc832d9102 wma encoder
michael
parents:
diff changeset
29 static int encode_init(AVCodecContext * avctx){
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
30 WMACodecContext *s = avctx->priv_data;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
31 int i, flags1, flags2;
0efc832d9102 wma encoder
michael
parents:
diff changeset
32 uint8_t *extradata;
0efc832d9102 wma encoder
michael
parents:
diff changeset
33
4600
6ac364a4ce2b Supply context to tprintf
mbardiaux
parents: 4493
diff changeset
34 s->avctx = avctx;
6ac364a4ce2b Supply context to tprintf
mbardiaux
parents: 4493
diff changeset
35
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
36 if(avctx->channels > MAX_CHANNELS)
0efc832d9102 wma encoder
michael
parents:
diff changeset
37 return -1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
38
4635
2ba8d13a66bc bitrate sanity check (fixes assertion failure)
michael
parents: 4601
diff changeset
39 if(avctx->bit_rate < 24*1000)
2ba8d13a66bc bitrate sanity check (fixes assertion failure)
michael
parents: 4601
diff changeset
40 return -1;
2ba8d13a66bc bitrate sanity check (fixes assertion failure)
michael
parents: 4601
diff changeset
41
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
42 /* extract flag infos */
0efc832d9102 wma encoder
michael
parents:
diff changeset
43 flags1 = 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
44 flags2 = 1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
45 if (avctx->codec->id == CODEC_ID_WMAV1) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
46 extradata= av_malloc(4);
0efc832d9102 wma encoder
michael
parents:
diff changeset
47 avctx->extradata_size= 4;
5089
bff60ecc02f9 Use AV_xx throughout libavcodec
ramiro
parents: 4737
diff changeset
48 AV_WL16(extradata, flags1);
bff60ecc02f9 Use AV_xx throughout libavcodec
ramiro
parents: 4737
diff changeset
49 AV_WL16(extradata+2, flags2);
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
50 } else if (avctx->codec->id == CODEC_ID_WMAV2) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
51 extradata= av_mallocz(10);
0efc832d9102 wma encoder
michael
parents:
diff changeset
52 avctx->extradata_size= 10;
5089
bff60ecc02f9 Use AV_xx throughout libavcodec
ramiro
parents: 4737
diff changeset
53 AV_WL32(extradata, flags1);
bff60ecc02f9 Use AV_xx throughout libavcodec
ramiro
parents: 4737
diff changeset
54 AV_WL16(extradata+4, flags2);
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
55 }else
0efc832d9102 wma encoder
michael
parents:
diff changeset
56 assert(0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
57 avctx->extradata= extradata;
0efc832d9102 wma encoder
michael
parents:
diff changeset
58 s->use_exp_vlc = flags2 & 0x0001;
0efc832d9102 wma encoder
michael
parents:
diff changeset
59 s->use_bit_reservoir = flags2 & 0x0002;
0efc832d9102 wma encoder
michael
parents:
diff changeset
60 s->use_variable_block_len = flags2 & 0x0004;
0efc832d9102 wma encoder
michael
parents:
diff changeset
61
0efc832d9102 wma encoder
michael
parents:
diff changeset
62 ff_wma_init(avctx, flags2);
0efc832d9102 wma encoder
michael
parents:
diff changeset
63
0efc832d9102 wma encoder
michael
parents:
diff changeset
64 /* init MDCT */
0efc832d9102 wma encoder
michael
parents:
diff changeset
65 for(i = 0; i < s->nb_block_sizes; i++)
9658
67a20f0eb42c Support for getting (i)MDCT output multiplied by a constant scaling factor.
serge
parents: 9431
diff changeset
66 ff_mdct_init(&s->mdct_ctx[i], s->frame_len_bits - i + 1, 0, 1.0);
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
67
0efc832d9102 wma encoder
michael
parents:
diff changeset
68 avctx->block_align=
0efc832d9102 wma encoder
michael
parents:
diff changeset
69 s->block_align= avctx->bit_rate*(int64_t)s->frame_len / (avctx->sample_rate*8);
0efc832d9102 wma encoder
michael
parents:
diff changeset
70 //av_log(NULL, AV_LOG_ERROR, "%d %d %d %d\n", s->block_align, avctx->bit_rate, s->frame_len, avctx->sample_rate);
0efc832d9102 wma encoder
michael
parents:
diff changeset
71 avctx->frame_size= s->frame_len;
0efc832d9102 wma encoder
michael
parents:
diff changeset
72
0efc832d9102 wma encoder
michael
parents:
diff changeset
73 return 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
74 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
75
0efc832d9102 wma encoder
michael
parents:
diff changeset
76
0efc832d9102 wma encoder
michael
parents:
diff changeset
77 static void apply_window_and_mdct(AVCodecContext * avctx, signed short * audio, int len) {
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
78 WMACodecContext *s = avctx->priv_data;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
79 int window_index= s->frame_len_bits - s->block_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
80 int i, j, channel;
0efc832d9102 wma encoder
michael
parents:
diff changeset
81 const float * win = s->windows[window_index];
0efc832d9102 wma encoder
michael
parents:
diff changeset
82 int window_len = 1 << s->block_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
83 float n = window_len/2;
0efc832d9102 wma encoder
michael
parents:
diff changeset
84
0efc832d9102 wma encoder
michael
parents:
diff changeset
85 for (channel = 0; channel < avctx->channels; channel++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
86 memcpy(s->output, s->frame_out[channel], sizeof(float)*window_len);
0efc832d9102 wma encoder
michael
parents:
diff changeset
87 j = channel;
0efc832d9102 wma encoder
michael
parents:
diff changeset
88 for (i = 0; i < len; i++, j += avctx->channels){
4737
99d9dd34903b Optimize by building the mdct window and multipying/adding at the same time.
banan
parents: 4635
diff changeset
89 s->output[i+window_len] = audio[j] / n * win[window_len - i - 1];
99d9dd34903b Optimize by building the mdct window and multipying/adding at the same time.
banan
parents: 4635
diff changeset
90 s->frame_out[channel][i] = audio[j] / n * win[i];
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
91 }
7546
97383e012cb9 remove mdct tmp buffer
lorenm
parents: 7451
diff changeset
92 ff_mdct_calc(&s->mdct_ctx[window_index], s->coefs[channel], s->output);
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
93 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
94 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
95
0efc832d9102 wma encoder
michael
parents:
diff changeset
96 //FIXME use for decoding too
5258
4372aeade5dc trivial warning fixes
mru
parents: 5089
diff changeset
97 static void init_exp(WMACodecContext *s, int ch, const int *exp_param){
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
98 int n;
0efc832d9102 wma encoder
michael
parents:
diff changeset
99 const uint16_t *ptr;
0efc832d9102 wma encoder
michael
parents:
diff changeset
100 float v, *q, max_scale, *q_end;
0efc832d9102 wma encoder
michael
parents:
diff changeset
101
0efc832d9102 wma encoder
michael
parents:
diff changeset
102 ptr = s->exponent_bands[s->frame_len_bits - s->block_len_bits];
0efc832d9102 wma encoder
michael
parents:
diff changeset
103 q = s->exponents[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
104 q_end = q + s->block_len;
0efc832d9102 wma encoder
michael
parents:
diff changeset
105 max_scale = 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
106 while (q < q_end) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
107 /* XXX: use a table */
0efc832d9102 wma encoder
michael
parents:
diff changeset
108 v = pow(10, *exp_param++ * (1.0 / 16.0));
0efc832d9102 wma encoder
michael
parents:
diff changeset
109 max_scale= FFMAX(max_scale, v);
0efc832d9102 wma encoder
michael
parents:
diff changeset
110 n = *ptr++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
111 do {
0efc832d9102 wma encoder
michael
parents:
diff changeset
112 *q++ = v;
0efc832d9102 wma encoder
michael
parents:
diff changeset
113 } while (--n);
0efc832d9102 wma encoder
michael
parents:
diff changeset
114 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
115 s->max_exponent[ch] = max_scale;
0efc832d9102 wma encoder
michael
parents:
diff changeset
116 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
117
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
118 static void encode_exp_vlc(WMACodecContext *s, int ch, const int *exp_param){
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
119 int last_exp;
0efc832d9102 wma encoder
michael
parents:
diff changeset
120 const uint16_t *ptr;
0efc832d9102 wma encoder
michael
parents:
diff changeset
121 float *q, *q_end;
0efc832d9102 wma encoder
michael
parents:
diff changeset
122
0efc832d9102 wma encoder
michael
parents:
diff changeset
123 ptr = s->exponent_bands[s->frame_len_bits - s->block_len_bits];
0efc832d9102 wma encoder
michael
parents:
diff changeset
124 q = s->exponents[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
125 q_end = q + s->block_len;
0efc832d9102 wma encoder
michael
parents:
diff changeset
126 if (s->version == 1) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
127 last_exp= *exp_param++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
128 assert(last_exp-10 >= 0 && last_exp-10 < 32);
0efc832d9102 wma encoder
michael
parents:
diff changeset
129 put_bits(&s->pb, 5, last_exp - 10);
0efc832d9102 wma encoder
michael
parents:
diff changeset
130 q+= *ptr++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
131 }else
0efc832d9102 wma encoder
michael
parents:
diff changeset
132 last_exp = 36;
0efc832d9102 wma encoder
michael
parents:
diff changeset
133 while (q < q_end) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
134 int exp = *exp_param++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
135 int code = exp - last_exp + 60;
0efc832d9102 wma encoder
michael
parents:
diff changeset
136 assert(code >= 0 && code < 120);
11215
964d01b50f17 remove a Huffman table from WMA which also exists in AAC
stefang
parents: 10145
diff changeset
137 put_bits(&s->pb, ff_aac_scalefactor_bits[code], ff_aac_scalefactor_code[code]);
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
138 /* XXX: use a table */
0efc832d9102 wma encoder
michael
parents:
diff changeset
139 q+= *ptr++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
140 last_exp= exp;
0efc832d9102 wma encoder
michael
parents:
diff changeset
141 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
142 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
143
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
144 static int encode_block(WMACodecContext *s, float (*src_coefs)[BLOCK_MAX_SIZE], int total_gain){
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
145 int v, bsize, ch, coef_nb_bits, parse_exponents;
0efc832d9102 wma encoder
michael
parents:
diff changeset
146 float mdct_norm;
0efc832d9102 wma encoder
michael
parents:
diff changeset
147 int nb_coefs[MAX_CHANNELS];
0efc832d9102 wma encoder
michael
parents:
diff changeset
148 static const int fixed_exp[25]={20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20,20};
0efc832d9102 wma encoder
michael
parents:
diff changeset
149
0efc832d9102 wma encoder
michael
parents:
diff changeset
150 //FIXME remove duplication relative to decoder
0efc832d9102 wma encoder
michael
parents:
diff changeset
151 if (s->use_variable_block_len) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
152 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
153 }else{
0efc832d9102 wma encoder
michael
parents:
diff changeset
154 /* fixed block len */
0efc832d9102 wma encoder
michael
parents:
diff changeset
155 s->next_block_len_bits = s->frame_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
156 s->prev_block_len_bits = s->frame_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
157 s->block_len_bits = s->frame_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
158 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
159
0efc832d9102 wma encoder
michael
parents:
diff changeset
160 s->block_len = 1 << s->block_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
161 // assert((s->block_pos + s->block_len) <= s->frame_len);
0efc832d9102 wma encoder
michael
parents:
diff changeset
162 bsize = s->frame_len_bits - s->block_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
163
0efc832d9102 wma encoder
michael
parents:
diff changeset
164 //FIXME factor
0efc832d9102 wma encoder
michael
parents:
diff changeset
165 v = s->coefs_end[bsize] - s->coefs_start;
0efc832d9102 wma encoder
michael
parents:
diff changeset
166 for(ch = 0; ch < s->nb_channels; ch++)
0efc832d9102 wma encoder
michael
parents:
diff changeset
167 nb_coefs[ch] = v;
0efc832d9102 wma encoder
michael
parents:
diff changeset
168 {
0efc832d9102 wma encoder
michael
parents:
diff changeset
169 int n4 = s->block_len / 2;
0efc832d9102 wma encoder
michael
parents:
diff changeset
170 mdct_norm = 1.0 / (float)n4;
0efc832d9102 wma encoder
michael
parents:
diff changeset
171 if (s->version == 1) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
172 mdct_norm *= sqrt(n4);
0efc832d9102 wma encoder
michael
parents:
diff changeset
173 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
174 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
175
0efc832d9102 wma encoder
michael
parents:
diff changeset
176 if (s->nb_channels == 2) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
177 put_bits(&s->pb, 1, s->ms_stereo= 1);
0efc832d9102 wma encoder
michael
parents:
diff changeset
178 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
179
0efc832d9102 wma encoder
michael
parents:
diff changeset
180 for(ch = 0; ch < s->nb_channels; ch++) {
7985
5be50ba0ce24 Intel C compiler warns on this assignment in this if(), probably
gpoirier
parents: 7546
diff changeset
181 s->channel_coded[ch] = 1; //FIXME only set channel_coded when needed, instead of always
5be50ba0ce24 Intel C compiler warns on this assignment in this if(), probably
gpoirier
parents: 7546
diff changeset
182 if (s->channel_coded[ch]) {
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
183 init_exp(s, ch, fixed_exp);
0efc832d9102 wma encoder
michael
parents:
diff changeset
184 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
185 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
186
0efc832d9102 wma encoder
michael
parents:
diff changeset
187 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
188 if (s->channel_coded[ch]) {
9868
5cc32c474caf Introduce WMACoef typedef for decoded coefficients
faust3
parents: 9658
diff changeset
189 WMACoef *coefs1;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
190 float *coefs, *exponents, mult;
0efc832d9102 wma encoder
michael
parents:
diff changeset
191 int i, n;
0efc832d9102 wma encoder
michael
parents:
diff changeset
192
0efc832d9102 wma encoder
michael
parents:
diff changeset
193 coefs1 = s->coefs1[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
194 exponents = s->exponents[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
195 mult = pow(10, total_gain * 0.05) / s->max_exponent[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
196 mult *= mdct_norm;
0efc832d9102 wma encoder
michael
parents:
diff changeset
197 coefs = src_coefs[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
198 if (s->use_noise_coding && 0) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
199 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
200 } else {
0efc832d9102 wma encoder
michael
parents:
diff changeset
201 coefs += s->coefs_start;
0efc832d9102 wma encoder
michael
parents:
diff changeset
202 n = nb_coefs[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
203 for(i = 0;i < n; i++){
0efc832d9102 wma encoder
michael
parents:
diff changeset
204 double t= *coefs++ / (exponents[i] * mult);
0efc832d9102 wma encoder
michael
parents:
diff changeset
205 if(t<-32768 || t>32767)
0efc832d9102 wma encoder
michael
parents:
diff changeset
206 return -1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
207
0efc832d9102 wma encoder
michael
parents:
diff changeset
208 coefs1[i] = lrint(t);
0efc832d9102 wma encoder
michael
parents:
diff changeset
209 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
210 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
211 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
212 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
213
0efc832d9102 wma encoder
michael
parents:
diff changeset
214 v = 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
215 for(ch = 0; ch < s->nb_channels; ch++) {
4492
245a46e76b00 10l use of uninitalized var
michael
parents: 4490
diff changeset
216 int a = s->channel_coded[ch];
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
217 put_bits(&s->pb, 1, a);
0efc832d9102 wma encoder
michael
parents:
diff changeset
218 v |= a;
0efc832d9102 wma encoder
michael
parents:
diff changeset
219 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
220
0efc832d9102 wma encoder
michael
parents:
diff changeset
221 if (!v)
0efc832d9102 wma encoder
michael
parents:
diff changeset
222 return 1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
223
0efc832d9102 wma encoder
michael
parents:
diff changeset
224 for(v= total_gain-1; v>=127; v-= 127)
0efc832d9102 wma encoder
michael
parents:
diff changeset
225 put_bits(&s->pb, 7, 127);
0efc832d9102 wma encoder
michael
parents:
diff changeset
226 put_bits(&s->pb, 7, v);
0efc832d9102 wma encoder
michael
parents:
diff changeset
227
0efc832d9102 wma encoder
michael
parents:
diff changeset
228 coef_nb_bits= ff_wma_total_gain_to_bits(total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
229
0efc832d9102 wma encoder
michael
parents:
diff changeset
230 if (s->use_noise_coding) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
231 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
232 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
233 int i, n;
0efc832d9102 wma encoder
michael
parents:
diff changeset
234 n = s->exponent_high_sizes[bsize];
0efc832d9102 wma encoder
michael
parents:
diff changeset
235 for(i=0;i<n;i++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
236 put_bits(&s->pb, 1, s->high_band_coded[ch][i]= 0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
237 if (0)
0efc832d9102 wma encoder
michael
parents:
diff changeset
238 nb_coefs[ch] -= s->exponent_high_bands[bsize][i];
0efc832d9102 wma encoder
michael
parents:
diff changeset
239 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
240 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
241 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
242 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
243
0efc832d9102 wma encoder
michael
parents:
diff changeset
244 parse_exponents = 1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
245 if (s->block_len_bits != s->frame_len_bits) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
246 put_bits(&s->pb, 1, parse_exponents);
0efc832d9102 wma encoder
michael
parents:
diff changeset
247 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
248
0efc832d9102 wma encoder
michael
parents:
diff changeset
249 if (parse_exponents) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
250 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
251 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
252 if (s->use_exp_vlc) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
253 encode_exp_vlc(s, ch, fixed_exp);
0efc832d9102 wma encoder
michael
parents:
diff changeset
254 } else {
0efc832d9102 wma encoder
michael
parents:
diff changeset
255 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
256 // encode_exp_lsp(s, ch);
0efc832d9102 wma encoder
michael
parents:
diff changeset
257 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
258 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
259 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
260 } else {
0efc832d9102 wma encoder
michael
parents:
diff changeset
261 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
262 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
263
0efc832d9102 wma encoder
michael
parents:
diff changeset
264 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
265 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
266 int run, tindex;
9868
5cc32c474caf Introduce WMACoef typedef for decoded coefficients
faust3
parents: 9658
diff changeset
267 WMACoef *ptr, *eptr;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
268 tindex = (ch == 1 && s->ms_stereo);
0efc832d9102 wma encoder
michael
parents:
diff changeset
269 ptr = &s->coefs1[ch][0];
0efc832d9102 wma encoder
michael
parents:
diff changeset
270 eptr = ptr + nb_coefs[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
271
0efc832d9102 wma encoder
michael
parents:
diff changeset
272 run=0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
273 for(;ptr < eptr; ptr++){
0efc832d9102 wma encoder
michael
parents:
diff changeset
274 if(*ptr){
0efc832d9102 wma encoder
michael
parents:
diff changeset
275 int level= *ptr;
0efc832d9102 wma encoder
michael
parents:
diff changeset
276 int abs_level= FFABS(level);
0efc832d9102 wma encoder
michael
parents:
diff changeset
277 int code= 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
278 if(abs_level <= s->coef_vlcs[tindex]->max_level){
0efc832d9102 wma encoder
michael
parents:
diff changeset
279 if(run < s->coef_vlcs[tindex]->levels[abs_level-1])
0efc832d9102 wma encoder
michael
parents:
diff changeset
280 code= run + s->int_table[tindex][abs_level-1];
0efc832d9102 wma encoder
michael
parents:
diff changeset
281 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
282
0efc832d9102 wma encoder
michael
parents:
diff changeset
283 assert(code < s->coef_vlcs[tindex]->n);
0efc832d9102 wma encoder
michael
parents:
diff changeset
284 put_bits(&s->pb, s->coef_vlcs[tindex]->huffbits[code], s->coef_vlcs[tindex]->huffcodes[code]);
0efc832d9102 wma encoder
michael
parents:
diff changeset
285
0efc832d9102 wma encoder
michael
parents:
diff changeset
286 if(code == 0){
0efc832d9102 wma encoder
michael
parents:
diff changeset
287 if(1<<coef_nb_bits <= abs_level)
0efc832d9102 wma encoder
michael
parents:
diff changeset
288 return -1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
289
8985
028c9cf95b95 A try to fix the regressions.
michael
parents: 7985
diff changeset
290
028c9cf95b95 A try to fix the regressions.
michael
parents: 7985
diff changeset
291 //Workaround minor rounding differences for the regression tests, FIXME we should find and replace the problematic float by fixpoint for reg tests
028c9cf95b95 A try to fix the regressions.
michael
parents: 7985
diff changeset
292 if(abs_level == 0x71B && (s->avctx->flags & CODEC_FLAG_BITEXACT)) abs_level=0x71A;
028c9cf95b95 A try to fix the regressions.
michael
parents: 7985
diff changeset
293
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
294 put_bits(&s->pb, coef_nb_bits, abs_level);
0efc832d9102 wma encoder
michael
parents:
diff changeset
295 put_bits(&s->pb, s->frame_len_bits, run);
0efc832d9102 wma encoder
michael
parents:
diff changeset
296 }
4493
e78c415815f3 workaround sign bug
michael
parents: 4492
diff changeset
297 put_bits(&s->pb, 1, level < 0); //FIXME the sign is fliped somewhere
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
298 run=0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
299 }else{
0efc832d9102 wma encoder
michael
parents:
diff changeset
300 run++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
301 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
302 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
303 if(run)
0efc832d9102 wma encoder
michael
parents:
diff changeset
304 put_bits(&s->pb, s->coef_vlcs[tindex]->huffbits[1], s->coef_vlcs[tindex]->huffcodes[1]);
0efc832d9102 wma encoder
michael
parents:
diff changeset
305 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
306 if (s->version == 1 && s->nb_channels >= 2) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
307 align_put_bits(&s->pb);
0efc832d9102 wma encoder
michael
parents:
diff changeset
308 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
309 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
310 return 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
311 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
312
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
313 static int encode_frame(WMACodecContext *s, float (*src_coefs)[BLOCK_MAX_SIZE], uint8_t *buf, int buf_size, int total_gain){
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
314 init_put_bits(&s->pb, buf, buf_size);
0efc832d9102 wma encoder
michael
parents:
diff changeset
315
0efc832d9102 wma encoder
michael
parents:
diff changeset
316 if (s->use_bit_reservoir) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
317 assert(0);//FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
318 }else{
0efc832d9102 wma encoder
michael
parents:
diff changeset
319 if(encode_block(s, src_coefs, total_gain) < 0)
0efc832d9102 wma encoder
michael
parents:
diff changeset
320 return INT_MAX;
0efc832d9102 wma encoder
michael
parents:
diff changeset
321 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
322
0efc832d9102 wma encoder
michael
parents:
diff changeset
323 align_put_bits(&s->pb);
0efc832d9102 wma encoder
michael
parents:
diff changeset
324
0efc832d9102 wma encoder
michael
parents:
diff changeset
325 return put_bits_count(&s->pb)/8 - s->block_align;
0efc832d9102 wma encoder
michael
parents:
diff changeset
326 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
327
0efc832d9102 wma encoder
michael
parents:
diff changeset
328 static int encode_superframe(AVCodecContext *avctx,
0efc832d9102 wma encoder
michael
parents:
diff changeset
329 unsigned char *buf, int buf_size, void *data){
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
330 WMACodecContext *s = avctx->priv_data;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
331 short *samples = data;
5258
4372aeade5dc trivial warning fixes
mru
parents: 5089
diff changeset
332 int i, total_gain;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
333
0efc832d9102 wma encoder
michael
parents:
diff changeset
334 s->block_len_bits= s->frame_len_bits; //required by non variable block len
0efc832d9102 wma encoder
michael
parents:
diff changeset
335 s->block_len = 1 << s->block_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
336
0efc832d9102 wma encoder
michael
parents:
diff changeset
337 apply_window_and_mdct(avctx, samples, avctx->frame_size);
0efc832d9102 wma encoder
michael
parents:
diff changeset
338
0efc832d9102 wma encoder
michael
parents:
diff changeset
339 if (s->ms_stereo) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
340 float a, b;
0efc832d9102 wma encoder
michael
parents:
diff changeset
341 int i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
342
0efc832d9102 wma encoder
michael
parents:
diff changeset
343 for(i = 0; i < s->block_len; i++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
344 a = s->coefs[0][i]*0.5;
0efc832d9102 wma encoder
michael
parents:
diff changeset
345 b = s->coefs[1][i]*0.5;
0efc832d9102 wma encoder
michael
parents:
diff changeset
346 s->coefs[0][i] = a + b;
0efc832d9102 wma encoder
michael
parents:
diff changeset
347 s->coefs[1][i] = a - b;
0efc832d9102 wma encoder
michael
parents:
diff changeset
348 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
349 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
350
0efc832d9102 wma encoder
michael
parents:
diff changeset
351 #if 1
0efc832d9102 wma encoder
michael
parents:
diff changeset
352 total_gain= 128;
0efc832d9102 wma encoder
michael
parents:
diff changeset
353 for(i=64; i; i>>=1){
0efc832d9102 wma encoder
michael
parents:
diff changeset
354 int error= encode_frame(s, s->coefs, buf, buf_size, total_gain-i);
0efc832d9102 wma encoder
michael
parents:
diff changeset
355 if(error<0)
0efc832d9102 wma encoder
michael
parents:
diff changeset
356 total_gain-= i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
357 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
358 #else
0efc832d9102 wma encoder
michael
parents:
diff changeset
359 total_gain= 90;
0efc832d9102 wma encoder
michael
parents:
diff changeset
360 best= encode_frame(s, s->coefs, buf, buf_size, total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
361 for(i=32; i; i>>=1){
0efc832d9102 wma encoder
michael
parents:
diff changeset
362 int scoreL= encode_frame(s, s->coefs, buf, buf_size, total_gain-i);
0efc832d9102 wma encoder
michael
parents:
diff changeset
363 int scoreR= encode_frame(s, s->coefs, buf, buf_size, total_gain+i);
0efc832d9102 wma encoder
michael
parents:
diff changeset
364 av_log(NULL, AV_LOG_ERROR, "%d %d %d (%d)\n", scoreL, best, scoreR, total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
365 if(scoreL < FFMIN(best, scoreR)){
0efc832d9102 wma encoder
michael
parents:
diff changeset
366 best = scoreL;
0efc832d9102 wma encoder
michael
parents:
diff changeset
367 total_gain -= i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
368 }else if(scoreR < best){
0efc832d9102 wma encoder
michael
parents:
diff changeset
369 best = scoreR;
0efc832d9102 wma encoder
michael
parents:
diff changeset
370 total_gain += i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
371 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
372 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
373 #endif
0efc832d9102 wma encoder
michael
parents:
diff changeset
374
0efc832d9102 wma encoder
michael
parents:
diff changeset
375 encode_frame(s, s->coefs, buf, buf_size, total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
376 assert((put_bits_count(&s->pb) & 7) == 0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
377 i= s->block_align - (put_bits_count(&s->pb)+7)/8;
0efc832d9102 wma encoder
michael
parents:
diff changeset
378 assert(i>=0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
379 while(i--)
0efc832d9102 wma encoder
michael
parents:
diff changeset
380 put_bits(&s->pb, 8, 'N');
0efc832d9102 wma encoder
michael
parents:
diff changeset
381
0efc832d9102 wma encoder
michael
parents:
diff changeset
382 flush_put_bits(&s->pb);
9431
932543edc1d2 Rename pbBufPtr() to put_bits_ptr().
stefano
parents: 8985
diff changeset
383 return put_bits_ptr(&s->pb) - s->pb.buf;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
384 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
385
0efc832d9102 wma encoder
michael
parents:
diff changeset
386 AVCodec wmav1_encoder =
0efc832d9102 wma encoder
michael
parents:
diff changeset
387 {
0efc832d9102 wma encoder
michael
parents:
diff changeset
388 "wmav1",
0efc832d9102 wma encoder
michael
parents:
diff changeset
389 CODEC_TYPE_AUDIO,
0efc832d9102 wma encoder
michael
parents:
diff changeset
390 CODEC_ID_WMAV1,
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
391 sizeof(WMACodecContext),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
392 encode_init,
0efc832d9102 wma encoder
michael
parents:
diff changeset
393 encode_superframe,
0efc832d9102 wma encoder
michael
parents:
diff changeset
394 ff_wma_end,
10145
7955db355703 Make sample_fmts and channel_layouts compound literals const to reduce size of
reimar
parents: 9868
diff changeset
395 .sample_fmts = (const enum SampleFormat[]){SAMPLE_FMT_S16,SAMPLE_FMT_NONE},
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 6710
diff changeset
396 .long_name = NULL_IF_CONFIG_SMALL("Windows Media Audio 1"),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
397 };
0efc832d9102 wma encoder
michael
parents:
diff changeset
398
0efc832d9102 wma encoder
michael
parents:
diff changeset
399 AVCodec wmav2_encoder =
0efc832d9102 wma encoder
michael
parents:
diff changeset
400 {
0efc832d9102 wma encoder
michael
parents:
diff changeset
401 "wmav2",
0efc832d9102 wma encoder
michael
parents:
diff changeset
402 CODEC_TYPE_AUDIO,
0efc832d9102 wma encoder
michael
parents:
diff changeset
403 CODEC_ID_WMAV2,
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
404 sizeof(WMACodecContext),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
405 encode_init,
0efc832d9102 wma encoder
michael
parents:
diff changeset
406 encode_superframe,
0efc832d9102 wma encoder
michael
parents:
diff changeset
407 ff_wma_end,
10145
7955db355703 Make sample_fmts and channel_layouts compound literals const to reduce size of
reimar
parents: 9868
diff changeset
408 .sample_fmts = (const enum SampleFormat[]){SAMPLE_FMT_S16,SAMPLE_FMT_NONE},
7040
e943e1409077 Make AVCodec long_names definition conditional depending on CONFIG_SMALL.
stefano
parents: 6710
diff changeset
409 .long_name = NULL_IF_CONFIG_SMALL("Windows Media Audio 2"),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
410 };