Mercurial > libavcodec.hg
comparison g729dec.c @ 9865:498ceea90017 libavcodec
G.729 decoding routine (skeleton, including parameters decoding).
author | voroshil |
---|---|
date | Thu, 18 Jun 2009 18:14:38 +0000 |
parents | ae38e13a63d6 |
children | 2719d29359c6 |
comparison
equal
deleted
inserted
replaced
9864:f5ffd813dc7f | 9865:498ceea90017 |
---|---|
70 * 13017 (equals to 0.7945) instead of it. | 70 * 13017 (equals to 0.7945) instead of it. |
71 */ | 71 */ |
72 #define SHARP_MAX 13017 | 72 #define SHARP_MAX 13017 |
73 | 73 |
74 typedef struct { | 74 typedef struct { |
75 int sample_rate; | 75 uint8_t ac_index_bits[2]; ///< adaptive codebook index for second subframe (size in bits) |
76 uint8_t packed_frame_size; ///< input frame size(in bytes) | 76 uint8_t parity_bit; ///< parity bit for pitch delay |
77 uint8_t unpacked_frame_size;///< output frame size (in bytes) | 77 uint8_t gc_1st_index_bits; ///< gain codebook (first stage) index (size in bits) |
78 uint8_t gc_2nd_index_bits; ///< gain codebook (second stage) index (size in bits) | |
79 uint8_t fc_signs_bits; ///< number of pulses in fixed-codebook vector | |
78 uint8_t fc_indexes_bits; ///< size (in bits) of fixed-codebook index entry | 80 uint8_t fc_indexes_bits; ///< size (in bits) of fixed-codebook index entry |
79 | |
80 /// mr_energy = mean_energy + 10 * log10(2^26 * subframe_size) in (7.13) | |
81 int mr_energy; | |
82 } G729FormatDescription; | 81 } G729FormatDescription; |
82 | |
83 static const G729FormatDescription format_g729_8k = { | |
84 .ac_index_bits = {8,5}, | |
85 .parity_bit = 1, | |
86 .gc_1st_index_bits = GC_1ST_IDX_BITS_8K, | |
87 .gc_2nd_index_bits = GC_2ND_IDX_BITS_8K, | |
88 .fc_signs_bits = 4, | |
89 .fc_indexes_bits = 13, | |
90 }; | |
91 | |
92 static const G729FormatDescription format_g729d_6k4 = { | |
93 .ac_index_bits = {8,4}, | |
94 .parity_bit = 0, | |
95 .gc_1st_index_bits = GC_1ST_IDX_BITS_6K4, | |
96 .gc_2nd_index_bits = GC_2ND_IDX_BITS_6K4, | |
97 .fc_signs_bits = 2, | |
98 .fc_indexes_bits = 9, | |
99 }; | |
83 | 100 |
84 /** | 101 /** |
85 * \brief pseudo random number generator | 102 * \brief pseudo random number generator |
86 */ | 103 */ |
87 static inline uint16_t g729_prng(uint16_t value) | 104 static inline uint16_t g729_prng(uint16_t value) |
107 /* Both 8kbit/s and 6.4kbit/s modes uses two subframes per frame. */ | 124 /* Both 8kbit/s and 6.4kbit/s modes uses two subframes per frame. */ |
108 avctx->frame_size = SUBFRAME_SIZE << 1; | 125 avctx->frame_size = SUBFRAME_SIZE << 1; |
109 | 126 |
110 return 0; | 127 return 0; |
111 } | 128 } |
129 | |
130 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, | |
131 AVPacket *avpkt) | |
132 { | |
133 const uint8_t *buf = avpkt->data; | |
134 int buf_size = avpkt->size; | |
135 int16_t *out_frame = data; | |
136 GetBitContext gb; | |
137 G729FormatDescription format; | |
138 int frame_erasure = 0; ///< frame erasure detected during decoding | |
139 int bad_pitch = 0; ///< parity check failed | |
140 int i; | |
141 uint8_t ma_predictor; ///< switched MA predictor of LSP quantizer | |
142 uint8_t quantizer_1st; ///< first stage vector of quantizer | |
143 uint8_t quantizer_2nd_lo; ///< second stage lower vector of quantizer (size in bits) | |
144 uint8_t quantizer_2nd_hi; ///< second stage higher vector of quantizer (size in bits) | |
145 | |
146 if (*data_size < SUBFRAME_SIZE << 2) { | |
147 av_log(avctx, AV_LOG_ERROR, "Error processing packet: output buffer too small\n"); | |
148 return AVERROR(EIO); | |
149 } | |
150 | |
151 if (buf_size == 10) { | |
152 format = format_g729_8k; | |
153 av_log(avctx, AV_LOG_DEBUG, "Packet type: %s\n", "G.729 @ 8kbit/s"); | |
154 } else if (buf_size == 8) { | |
155 format = format_g729d_6k4; | |
156 av_log(avctx, AV_LOG_DEBUG, "Packet type: %s\n", "G.729D @ 6.4kbit/s"); | |
157 } else { | |
158 av_log(avctx, AV_LOG_ERROR, "Packet size %d is unknown.\n", buf_size); | |
159 return (AVERROR_NOFMT); | |
160 } | |
161 | |
162 for (i=0; i < buf_size; i++) | |
163 frame_erasure |= buf[i]; | |
164 frame_erasure = !frame_erasure; | |
165 | |
166 init_get_bits(&gb, buf, buf_size); | |
167 | |
168 ma_predictor = get_bits(&gb, 1); | |
169 quantizer_1st = get_bits(&gb, VQ_1ST_BITS); | |
170 quantizer_2nd_lo = get_bits(&gb, VQ_2ND_BITS); | |
171 quantizer_2nd_hi = get_bits(&gb, VQ_2ND_BITS); | |
172 | |
173 for (i = 0; i < 2; i++) { | |
174 uint8_t ac_index; ///< adaptive codebook index | |
175 uint8_t pulses_signs; ///< fixed-codebook vector pulse signs | |
176 int fc_indexes; ///< fixed-codebook indexes | |
177 uint8_t gc_1st_index; ///< gain codebook (first stage) index | |
178 uint8_t gc_2nd_index; ///< gain codebook (second stage) index | |
179 | |
180 ac_index = get_bits(&gb, format.ac_index_bits[i]); | |
181 if(!i && format.parity_bit) | |
182 bad_pitch = get_parity(ac_index) == get_bits1(&gb); | |
183 fc_indexes = get_bits(&gb, format.fc_indexes_bits); | |
184 pulses_signs = get_bits(&gb, format.fc_signs_bits); | |
185 gc_1st_index = get_bits(&gb, format.gc_1st_index_bits); | |
186 gc_2nd_index = get_bits(&gb, format.gc_2nd_index_bits); | |
112 | 187 |
113 ff_acelp_weighted_vector_sum(fc + pitch_delay_int[i], | 188 ff_acelp_weighted_vector_sum(fc + pitch_delay_int[i], |
114 fc + pitch_delay_int[i], | 189 fc + pitch_delay_int[i], |
115 fc, 1 << 14, | 190 fc, 1 << 14, |
116 av_clip(ctx->gain_pitch, SHARP_MIN, SHARP_MAX), | 191 av_clip(ctx->gain_pitch, SHARP_MIN, SHARP_MAX), |
117 0, 14, | 192 0, 14, |
118 SUBFRAME_SIZE - pitch_delay_int[i]); | 193 SUBFRAME_SIZE - pitch_delay_int[i]); |
119 | 194 |
120 if (ctx->frame_erasure) { | 195 if (frame_erasure) { |
121 ctx->gain_pitch = (29491 * ctx->gain_pitch) >> 15; // 0.90 (0.15) | 196 ctx->gain_pitch = (29491 * ctx->gain_pitch) >> 15; // 0.90 (0.15) |
122 ctx->gain_code = ( 2007 * ctx->gain_code ) >> 11; // 0.98 (0.11) | 197 ctx->gain_code = ( 2007 * ctx->gain_code ) >> 11; // 0.98 (0.11) |
123 | 198 |
124 gain_corr_factor = 0; | 199 gain_corr_factor = 0; |
125 } else { | 200 } else { |
126 ctx->gain_pitch = cb_gain_1st_8k[parm->gc_1st_index[i]][0] + | 201 ctx->gain_pitch = cb_gain_1st_8k[gc_1st_index][0] + |
127 cb_gain_2nd_8k[parm->gc_2nd_index[i]][0]; | 202 cb_gain_2nd_8k[gc_2nd_index][0]; |
128 gain_corr_factor = cb_gain_1st_8k[parm->gc_1st_index[i]][1] + | 203 gain_corr_factor = cb_gain_1st_8k[gc_1st_index][1] + |
129 cb_gain_2nd_8k[parm->gc_2nd_index[i]][1]; | 204 cb_gain_2nd_8k[gc_2nd_index][1]; |
130 | 205 |
131 ff_acelp_weighted_vector_sum(ctx->exc + i * SUBFRAME_SIZE, | 206 ff_acelp_weighted_vector_sum(ctx->exc + i * SUBFRAME_SIZE, |
132 ctx->exc + i * SUBFRAME_SIZE, fc, | 207 ctx->exc + i * SUBFRAME_SIZE, fc, |
133 (!voicing && ctx->frame_erasure) ? 0 : ctx->gain_pitch, | 208 (!voicing && frame_erasure) ? 0 : ctx->gain_pitch, |
134 ( voicing && ctx->frame_erasure) ? 0 : ctx->gain_code, | 209 ( voicing && frame_erasure) ? 0 : ctx->gain_code, |
135 1 << 13, 14, SUBFRAME_SIZE); | 210 1 << 13, 14, SUBFRAME_SIZE); |
136 | 211 } |
137 if (buf_size < packed_frame_size) { | 212 |
138 av_log(avctx, AV_LOG_ERROR, "Error processing packet: packet size too small\n"); | 213 *data_size = SUBFRAME_SIZE << 2; |
139 return AVERROR(EIO); | 214 return buf_size; |
140 } | 215 } |
141 if (*data_size < unpacked_frame_size) { | |
142 av_log(avctx, AV_LOG_ERROR, "Error processing packet: output buffer too small\n"); | |
143 return AVERROR(EIO); | |
144 } | |
145 | 216 |
146 AVCodec g729_decoder = | 217 AVCodec g729_decoder = |
147 { | 218 { |
148 "g729", | 219 "g729", |
149 CODEC_TYPE_AUDIO, | 220 CODEC_TYPE_AUDIO, |