annotate wmaenc.c @ 7951:f6c77b264d9b libavcodec

Remove useless initializations.
author benoit
date Tue, 30 Sep 2008 11:46:37 +0000
parents 97383e012cb9
children 5be50ba0ce24
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++)
0efc832d9102 wma encoder
michael
parents:
diff changeset
66 ff_mdct_init(&s->mdct_ctx[i], s->frame_len_bits - i + 1, 0);
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);
0efc832d9102 wma encoder
michael
parents:
diff changeset
137 put_bits(&s->pb, ff_wma_scale_huffbits[code], ff_wma_scale_huffcodes[code]);
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++) {
6355
5a2c7290ddd3 silence wmaenc.c:181: warning:suggestparentheses around assignment used as truth value
banan
parents: 6035
diff changeset
181 if ((s->channel_coded[ch]= 1)) { //FIXME only set channel_coded when needed, instead of always
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
182 init_exp(s, ch, fixed_exp);
0efc832d9102 wma encoder
michael
parents:
diff changeset
183 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
184 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
185
0efc832d9102 wma encoder
michael
parents:
diff changeset
186 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
187 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
188 int16_t *coefs1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
189 float *coefs, *exponents, mult;
0efc832d9102 wma encoder
michael
parents:
diff changeset
190 int i, n;
0efc832d9102 wma encoder
michael
parents:
diff changeset
191
0efc832d9102 wma encoder
michael
parents:
diff changeset
192 coefs1 = s->coefs1[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
193 exponents = s->exponents[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
194 mult = pow(10, total_gain * 0.05) / s->max_exponent[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
195 mult *= mdct_norm;
0efc832d9102 wma encoder
michael
parents:
diff changeset
196 coefs = src_coefs[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
197 if (s->use_noise_coding && 0) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
198 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
199 } else {
0efc832d9102 wma encoder
michael
parents:
diff changeset
200 coefs += s->coefs_start;
0efc832d9102 wma encoder
michael
parents:
diff changeset
201 n = nb_coefs[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
202 for(i = 0;i < n; i++){
0efc832d9102 wma encoder
michael
parents:
diff changeset
203 double t= *coefs++ / (exponents[i] * mult);
0efc832d9102 wma encoder
michael
parents:
diff changeset
204 if(t<-32768 || t>32767)
0efc832d9102 wma encoder
michael
parents:
diff changeset
205 return -1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
206
0efc832d9102 wma encoder
michael
parents:
diff changeset
207 coefs1[i] = lrint(t);
0efc832d9102 wma encoder
michael
parents:
diff changeset
208 }
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 v = 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
214 for(ch = 0; ch < s->nb_channels; ch++) {
4492
245a46e76b00 10l use of uninitalized var
michael
parents: 4490
diff changeset
215 int a = s->channel_coded[ch];
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
216 put_bits(&s->pb, 1, a);
0efc832d9102 wma encoder
michael
parents:
diff changeset
217 v |= a;
0efc832d9102 wma encoder
michael
parents:
diff changeset
218 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
219
0efc832d9102 wma encoder
michael
parents:
diff changeset
220 if (!v)
0efc832d9102 wma encoder
michael
parents:
diff changeset
221 return 1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
222
0efc832d9102 wma encoder
michael
parents:
diff changeset
223 for(v= total_gain-1; v>=127; v-= 127)
0efc832d9102 wma encoder
michael
parents:
diff changeset
224 put_bits(&s->pb, 7, 127);
0efc832d9102 wma encoder
michael
parents:
diff changeset
225 put_bits(&s->pb, 7, v);
0efc832d9102 wma encoder
michael
parents:
diff changeset
226
0efc832d9102 wma encoder
michael
parents:
diff changeset
227 coef_nb_bits= ff_wma_total_gain_to_bits(total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
228
0efc832d9102 wma encoder
michael
parents:
diff changeset
229 if (s->use_noise_coding) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
230 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
231 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
232 int i, n;
0efc832d9102 wma encoder
michael
parents:
diff changeset
233 n = s->exponent_high_sizes[bsize];
0efc832d9102 wma encoder
michael
parents:
diff changeset
234 for(i=0;i<n;i++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
235 put_bits(&s->pb, 1, s->high_band_coded[ch][i]= 0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
236 if (0)
0efc832d9102 wma encoder
michael
parents:
diff changeset
237 nb_coefs[ch] -= s->exponent_high_bands[bsize][i];
0efc832d9102 wma encoder
michael
parents:
diff changeset
238 }
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 parse_exponents = 1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
244 if (s->block_len_bits != s->frame_len_bits) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
245 put_bits(&s->pb, 1, parse_exponents);
0efc832d9102 wma encoder
michael
parents:
diff changeset
246 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
247
0efc832d9102 wma encoder
michael
parents:
diff changeset
248 if (parse_exponents) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
249 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
250 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
251 if (s->use_exp_vlc) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
252 encode_exp_vlc(s, ch, fixed_exp);
0efc832d9102 wma encoder
michael
parents:
diff changeset
253 } else {
0efc832d9102 wma encoder
michael
parents:
diff changeset
254 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
255 // encode_exp_lsp(s, ch);
0efc832d9102 wma encoder
michael
parents:
diff changeset
256 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
257 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
258 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
259 } else {
0efc832d9102 wma encoder
michael
parents:
diff changeset
260 assert(0); //FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
261 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
262
0efc832d9102 wma encoder
michael
parents:
diff changeset
263 for(ch = 0; ch < s->nb_channels; ch++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
264 if (s->channel_coded[ch]) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
265 int run, tindex;
0efc832d9102 wma encoder
michael
parents:
diff changeset
266 int16_t *ptr, *eptr;
0efc832d9102 wma encoder
michael
parents:
diff changeset
267 tindex = (ch == 1 && s->ms_stereo);
0efc832d9102 wma encoder
michael
parents:
diff changeset
268 ptr = &s->coefs1[ch][0];
0efc832d9102 wma encoder
michael
parents:
diff changeset
269 eptr = ptr + nb_coefs[ch];
0efc832d9102 wma encoder
michael
parents:
diff changeset
270
0efc832d9102 wma encoder
michael
parents:
diff changeset
271 run=0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
272 for(;ptr < eptr; ptr++){
0efc832d9102 wma encoder
michael
parents:
diff changeset
273 if(*ptr){
0efc832d9102 wma encoder
michael
parents:
diff changeset
274 int level= *ptr;
0efc832d9102 wma encoder
michael
parents:
diff changeset
275 int abs_level= FFABS(level);
0efc832d9102 wma encoder
michael
parents:
diff changeset
276 int code= 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
277 if(abs_level <= s->coef_vlcs[tindex]->max_level){
0efc832d9102 wma encoder
michael
parents:
diff changeset
278 if(run < s->coef_vlcs[tindex]->levels[abs_level-1])
0efc832d9102 wma encoder
michael
parents:
diff changeset
279 code= run + s->int_table[tindex][abs_level-1];
0efc832d9102 wma encoder
michael
parents:
diff changeset
280 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
281
0efc832d9102 wma encoder
michael
parents:
diff changeset
282 assert(code < s->coef_vlcs[tindex]->n);
0efc832d9102 wma encoder
michael
parents:
diff changeset
283 put_bits(&s->pb, s->coef_vlcs[tindex]->huffbits[code], s->coef_vlcs[tindex]->huffcodes[code]);
0efc832d9102 wma encoder
michael
parents:
diff changeset
284
0efc832d9102 wma encoder
michael
parents:
diff changeset
285 if(code == 0){
0efc832d9102 wma encoder
michael
parents:
diff changeset
286 if(1<<coef_nb_bits <= abs_level)
0efc832d9102 wma encoder
michael
parents:
diff changeset
287 return -1;
0efc832d9102 wma encoder
michael
parents:
diff changeset
288
0efc832d9102 wma encoder
michael
parents:
diff changeset
289 put_bits(&s->pb, coef_nb_bits, abs_level);
0efc832d9102 wma encoder
michael
parents:
diff changeset
290 put_bits(&s->pb, s->frame_len_bits, run);
0efc832d9102 wma encoder
michael
parents:
diff changeset
291 }
4493
e78c415815f3 workaround sign bug
michael
parents: 4492
diff changeset
292 put_bits(&s->pb, 1, level < 0); //FIXME the sign is fliped somewhere
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
293 run=0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
294 }else{
0efc832d9102 wma encoder
michael
parents:
diff changeset
295 run++;
0efc832d9102 wma encoder
michael
parents:
diff changeset
296 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
297 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
298 if(run)
0efc832d9102 wma encoder
michael
parents:
diff changeset
299 put_bits(&s->pb, s->coef_vlcs[tindex]->huffbits[1], s->coef_vlcs[tindex]->huffcodes[1]);
0efc832d9102 wma encoder
michael
parents:
diff changeset
300 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
301 if (s->version == 1 && s->nb_channels >= 2) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
302 align_put_bits(&s->pb);
0efc832d9102 wma encoder
michael
parents:
diff changeset
303 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
304 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
305 return 0;
0efc832d9102 wma encoder
michael
parents:
diff changeset
306 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
307
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
308 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
309 init_put_bits(&s->pb, buf, buf_size);
0efc832d9102 wma encoder
michael
parents:
diff changeset
310
0efc832d9102 wma encoder
michael
parents:
diff changeset
311 if (s->use_bit_reservoir) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
312 assert(0);//FIXME not implemented
0efc832d9102 wma encoder
michael
parents:
diff changeset
313 }else{
0efc832d9102 wma encoder
michael
parents:
diff changeset
314 if(encode_block(s, src_coefs, total_gain) < 0)
0efc832d9102 wma encoder
michael
parents:
diff changeset
315 return INT_MAX;
0efc832d9102 wma encoder
michael
parents:
diff changeset
316 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
317
0efc832d9102 wma encoder
michael
parents:
diff changeset
318 align_put_bits(&s->pb);
0efc832d9102 wma encoder
michael
parents:
diff changeset
319
0efc832d9102 wma encoder
michael
parents:
diff changeset
320 return put_bits_count(&s->pb)/8 - s->block_align;
0efc832d9102 wma encoder
michael
parents:
diff changeset
321 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
322
0efc832d9102 wma encoder
michael
parents:
diff changeset
323 static int encode_superframe(AVCodecContext *avctx,
0efc832d9102 wma encoder
michael
parents:
diff changeset
324 unsigned char *buf, int buf_size, void *data){
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
325 WMACodecContext *s = avctx->priv_data;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
326 short *samples = data;
5258
4372aeade5dc trivial warning fixes
mru
parents: 5089
diff changeset
327 int i, total_gain;
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
328
0efc832d9102 wma encoder
michael
parents:
diff changeset
329 s->block_len_bits= s->frame_len_bits; //required by non variable block len
0efc832d9102 wma encoder
michael
parents:
diff changeset
330 s->block_len = 1 << s->block_len_bits;
0efc832d9102 wma encoder
michael
parents:
diff changeset
331
0efc832d9102 wma encoder
michael
parents:
diff changeset
332 apply_window_and_mdct(avctx, samples, avctx->frame_size);
0efc832d9102 wma encoder
michael
parents:
diff changeset
333
0efc832d9102 wma encoder
michael
parents:
diff changeset
334 if (s->ms_stereo) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
335 float a, b;
0efc832d9102 wma encoder
michael
parents:
diff changeset
336 int i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
337
0efc832d9102 wma encoder
michael
parents:
diff changeset
338 for(i = 0; i < s->block_len; i++) {
0efc832d9102 wma encoder
michael
parents:
diff changeset
339 a = s->coefs[0][i]*0.5;
0efc832d9102 wma encoder
michael
parents:
diff changeset
340 b = s->coefs[1][i]*0.5;
0efc832d9102 wma encoder
michael
parents:
diff changeset
341 s->coefs[0][i] = a + b;
0efc832d9102 wma encoder
michael
parents:
diff changeset
342 s->coefs[1][i] = a - b;
0efc832d9102 wma encoder
michael
parents:
diff changeset
343 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
344 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
345
0efc832d9102 wma encoder
michael
parents:
diff changeset
346 #if 1
0efc832d9102 wma encoder
michael
parents:
diff changeset
347 total_gain= 128;
0efc832d9102 wma encoder
michael
parents:
diff changeset
348 for(i=64; i; i>>=1){
0efc832d9102 wma encoder
michael
parents:
diff changeset
349 int error= encode_frame(s, s->coefs, buf, buf_size, total_gain-i);
0efc832d9102 wma encoder
michael
parents:
diff changeset
350 if(error<0)
0efc832d9102 wma encoder
michael
parents:
diff changeset
351 total_gain-= i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
352 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
353 #else
0efc832d9102 wma encoder
michael
parents:
diff changeset
354 total_gain= 90;
0efc832d9102 wma encoder
michael
parents:
diff changeset
355 best= encode_frame(s, s->coefs, buf, buf_size, total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
356 for(i=32; i; i>>=1){
0efc832d9102 wma encoder
michael
parents:
diff changeset
357 int scoreL= encode_frame(s, s->coefs, buf, buf_size, total_gain-i);
0efc832d9102 wma encoder
michael
parents:
diff changeset
358 int scoreR= encode_frame(s, s->coefs, buf, buf_size, total_gain+i);
0efc832d9102 wma encoder
michael
parents:
diff changeset
359 av_log(NULL, AV_LOG_ERROR, "%d %d %d (%d)\n", scoreL, best, scoreR, total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
360 if(scoreL < FFMIN(best, scoreR)){
0efc832d9102 wma encoder
michael
parents:
diff changeset
361 best = scoreL;
0efc832d9102 wma encoder
michael
parents:
diff changeset
362 total_gain -= i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
363 }else if(scoreR < best){
0efc832d9102 wma encoder
michael
parents:
diff changeset
364 best = scoreR;
0efc832d9102 wma encoder
michael
parents:
diff changeset
365 total_gain += i;
0efc832d9102 wma encoder
michael
parents:
diff changeset
366 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
367 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
368 #endif
0efc832d9102 wma encoder
michael
parents:
diff changeset
369
0efc832d9102 wma encoder
michael
parents:
diff changeset
370 encode_frame(s, s->coefs, buf, buf_size, total_gain);
0efc832d9102 wma encoder
michael
parents:
diff changeset
371 assert((put_bits_count(&s->pb) & 7) == 0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
372 i= s->block_align - (put_bits_count(&s->pb)+7)/8;
0efc832d9102 wma encoder
michael
parents:
diff changeset
373 assert(i>=0);
0efc832d9102 wma encoder
michael
parents:
diff changeset
374 while(i--)
0efc832d9102 wma encoder
michael
parents:
diff changeset
375 put_bits(&s->pb, 8, 'N');
0efc832d9102 wma encoder
michael
parents:
diff changeset
376
0efc832d9102 wma encoder
michael
parents:
diff changeset
377 flush_put_bits(&s->pb);
0efc832d9102 wma encoder
michael
parents:
diff changeset
378 return pbBufPtr(&s->pb) - s->pb.buf;
0efc832d9102 wma encoder
michael
parents:
diff changeset
379 }
0efc832d9102 wma encoder
michael
parents:
diff changeset
380
0efc832d9102 wma encoder
michael
parents:
diff changeset
381 AVCodec wmav1_encoder =
0efc832d9102 wma encoder
michael
parents:
diff changeset
382 {
0efc832d9102 wma encoder
michael
parents:
diff changeset
383 "wmav1",
0efc832d9102 wma encoder
michael
parents:
diff changeset
384 CODEC_TYPE_AUDIO,
0efc832d9102 wma encoder
michael
parents:
diff changeset
385 CODEC_ID_WMAV1,
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
386 sizeof(WMACodecContext),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
387 encode_init,
0efc832d9102 wma encoder
michael
parents:
diff changeset
388 encode_superframe,
0efc832d9102 wma encoder
michael
parents:
diff changeset
389 ff_wma_end,
7451
85ab7655ad4d Modify all codecs to report their supported input and output sample format(s).
pross
parents: 7040
diff changeset
390 .sample_fmts = (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
391 .long_name = NULL_IF_CONFIG_SMALL("Windows Media Audio 1"),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
392 };
0efc832d9102 wma encoder
michael
parents:
diff changeset
393
0efc832d9102 wma encoder
michael
parents:
diff changeset
394 AVCodec wmav2_encoder =
0efc832d9102 wma encoder
michael
parents:
diff changeset
395 {
0efc832d9102 wma encoder
michael
parents:
diff changeset
396 "wmav2",
0efc832d9102 wma encoder
michael
parents:
diff changeset
397 CODEC_TYPE_AUDIO,
0efc832d9102 wma encoder
michael
parents:
diff changeset
398 CODEC_ID_WMAV2,
4601
7b9ce6f729ae Rename WMADecodeContext to WMACodecContext
mbardiaux
parents: 4600
diff changeset
399 sizeof(WMACodecContext),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
400 encode_init,
0efc832d9102 wma encoder
michael
parents:
diff changeset
401 encode_superframe,
0efc832d9102 wma encoder
michael
parents:
diff changeset
402 ff_wma_end,
7451
85ab7655ad4d Modify all codecs to report their supported input and output sample format(s).
pross
parents: 7040
diff changeset
403 .sample_fmts = (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
404 .long_name = NULL_IF_CONFIG_SMALL("Windows Media Audio 2"),
4490
0efc832d9102 wma encoder
michael
parents:
diff changeset
405 };