Normalize triplets in gain_val_tab[][] so gain_exp_tab[] can be just an
[ffmpeg-lucabe.git] / libavcodec / pcm.c
blob84aa098750a36ff1dec5623adaa62c721e60270f
1 /*
2 * PCM codecs
3 * Copyright (c) 2001 Fabrice Bellard.
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22 /**
23 * @file pcm.c
24 * PCM codecs
27 #include "avcodec.h"
28 #include "bitstream.h" // for ff_reverse
29 #include "bytestream.h"
31 #define MAX_CHANNELS 64
33 /* from g711.c by SUN microsystems (unrestricted use) */
35 #define SIGN_BIT (0x80) /* Sign bit for a A-law byte. */
36 #define QUANT_MASK (0xf) /* Quantization field mask. */
37 #define NSEGS (8) /* Number of A-law segments. */
38 #define SEG_SHIFT (4) /* Left shift for segment number. */
39 #define SEG_MASK (0x70) /* Segment field mask. */
41 #define BIAS (0x84) /* Bias for linear code. */
44 * alaw2linear() - Convert an A-law value to 16-bit linear PCM
47 static av_cold int alaw2linear(unsigned char a_val)
49 int t;
50 int seg;
52 a_val ^= 0x55;
54 t = a_val & QUANT_MASK;
55 seg = ((unsigned)a_val & SEG_MASK) >> SEG_SHIFT;
56 if(seg) t= (t + t + 1 + 32) << (seg + 2);
57 else t= (t + t + 1 ) << 3;
59 return (a_val & SIGN_BIT) ? t : -t;
62 static av_cold int ulaw2linear(unsigned char u_val)
64 int t;
66 /* Complement to obtain normal u-law value. */
67 u_val = ~u_val;
70 * Extract and bias the quantization bits. Then
71 * shift up by the segment number and subtract out the bias.
73 t = ((u_val & QUANT_MASK) << 3) + BIAS;
74 t <<= ((unsigned)u_val & SEG_MASK) >> SEG_SHIFT;
76 return (u_val & SIGN_BIT) ? (BIAS - t) : (t - BIAS);
79 /* 16384 entries per table */
80 static uint8_t linear_to_alaw[16384];
81 static uint8_t linear_to_ulaw[16384];
83 static av_cold void build_xlaw_table(uint8_t *linear_to_xlaw,
84 int (*xlaw2linear)(unsigned char),
85 int mask)
87 int i, j, v, v1, v2;
89 j = 0;
90 for(i=0;i<128;i++) {
91 if (i != 127) {
92 v1 = xlaw2linear(i ^ mask);
93 v2 = xlaw2linear((i + 1) ^ mask);
94 v = (v1 + v2 + 4) >> 3;
95 } else {
96 v = 8192;
98 for(;j<v;j++) {
99 linear_to_xlaw[8192 + j] = (i ^ mask);
100 if (j > 0)
101 linear_to_xlaw[8192 - j] = (i ^ (mask ^ 0x80));
104 linear_to_xlaw[0] = linear_to_xlaw[1];
107 static av_cold int pcm_encode_init(AVCodecContext *avctx)
109 avctx->frame_size = 1;
110 switch(avctx->codec->id) {
111 case CODEC_ID_PCM_ALAW:
112 build_xlaw_table(linear_to_alaw, alaw2linear, 0xd5);
113 break;
114 case CODEC_ID_PCM_MULAW:
115 build_xlaw_table(linear_to_ulaw, ulaw2linear, 0xff);
116 break;
117 default:
118 break;
121 avctx->bits_per_coded_sample = av_get_bits_per_sample(avctx->codec->id);
122 avctx->block_align = avctx->channels * avctx->bits_per_coded_sample/8;
123 avctx->coded_frame= avcodec_alloc_frame();
124 avctx->coded_frame->key_frame= 1;
126 return 0;
129 static av_cold int pcm_encode_close(AVCodecContext *avctx)
131 av_freep(&avctx->coded_frame);
133 return 0;
137 * Write PCM samples macro
138 * @param type Datatype of native machine format
139 * @param endian bytestream_put_xxx() suffix
140 * @param src Source pointer (variable name)
141 * @param dst Destination pointer (variable name)
142 * @param n Total number of samples (variable name)
143 * @param shift Bitshift (bits)
144 * @param offset Sample value offset
146 #define ENCODE(type, endian, src, dst, n, shift, offset) \
147 samples_##type = (type*)src; \
148 for(;n>0;n--) { \
149 register type v = (*samples_##type++ >> shift) + offset; \
150 bytestream_put_##endian(&dst, v); \
153 static int pcm_encode_frame(AVCodecContext *avctx,
154 unsigned char *frame, int buf_size, void *data)
156 int n, sample_size, v;
157 short *samples;
158 unsigned char *dst;
159 uint8_t *srcu8;
160 int16_t *samples_int16_t;
161 int32_t *samples_int32_t;
162 int64_t *samples_int64_t;
163 uint16_t *samples_uint16_t;
164 uint32_t *samples_uint32_t;
166 sample_size = av_get_bits_per_sample(avctx->codec->id)/8;
167 n = buf_size / sample_size;
168 samples = data;
169 dst = frame;
171 if (avctx->sample_fmt!=avctx->codec->sample_fmts[0]) {
172 av_log(avctx, AV_LOG_ERROR, "invalid sample_fmt\n");
173 return -1;
176 switch(avctx->codec->id) {
177 case CODEC_ID_PCM_U32LE:
178 ENCODE(uint32_t, le32, samples, dst, n, 0, 0x80000000)
179 break;
180 case CODEC_ID_PCM_U32BE:
181 ENCODE(uint32_t, be32, samples, dst, n, 0, 0x80000000)
182 break;
183 case CODEC_ID_PCM_S24LE:
184 ENCODE(int32_t, le24, samples, dst, n, 8, 0)
185 break;
186 case CODEC_ID_PCM_S24BE:
187 ENCODE(int32_t, be24, samples, dst, n, 8, 0)
188 break;
189 case CODEC_ID_PCM_U24LE:
190 ENCODE(uint32_t, le24, samples, dst, n, 8, 0x800000)
191 break;
192 case CODEC_ID_PCM_U24BE:
193 ENCODE(uint32_t, be24, samples, dst, n, 8, 0x800000)
194 break;
195 case CODEC_ID_PCM_S24DAUD:
196 for(;n>0;n--) {
197 uint32_t tmp = ff_reverse[(*samples >> 8) & 0xff] +
198 (ff_reverse[*samples & 0xff] << 8);
199 tmp <<= 4; // sync flags would go here
200 bytestream_put_be24(&dst, tmp);
201 samples++;
203 break;
204 case CODEC_ID_PCM_U16LE:
205 ENCODE(uint16_t, le16, samples, dst, n, 0, 0x8000)
206 break;
207 case CODEC_ID_PCM_U16BE:
208 ENCODE(uint16_t, be16, samples, dst, n, 0, 0x8000)
209 break;
210 case CODEC_ID_PCM_S8:
211 srcu8= data;
212 for(;n>0;n--) {
213 v = *srcu8++;
214 *dst++ = v - 128;
216 break;
217 #if WORDS_BIGENDIAN
218 case CODEC_ID_PCM_F64LE:
219 ENCODE(int64_t, le64, samples, dst, n, 0, 0)
220 break;
221 case CODEC_ID_PCM_S32LE:
222 case CODEC_ID_PCM_F32LE:
223 ENCODE(int32_t, le32, samples, dst, n, 0, 0)
224 break;
225 case CODEC_ID_PCM_S16LE:
226 ENCODE(int16_t, le16, samples, dst, n, 0, 0)
227 break;
228 case CODEC_ID_PCM_F64BE:
229 case CODEC_ID_PCM_F32BE:
230 case CODEC_ID_PCM_S32BE:
231 case CODEC_ID_PCM_S16BE:
232 #else
233 case CODEC_ID_PCM_F64BE:
234 ENCODE(int64_t, be64, samples, dst, n, 0, 0)
235 break;
236 case CODEC_ID_PCM_F32BE:
237 case CODEC_ID_PCM_S32BE:
238 ENCODE(int32_t, be32, samples, dst, n, 0, 0)
239 break;
240 case CODEC_ID_PCM_S16BE:
241 ENCODE(int16_t, be16, samples, dst, n, 0, 0)
242 break;
243 case CODEC_ID_PCM_F64LE:
244 case CODEC_ID_PCM_F32LE:
245 case CODEC_ID_PCM_S32LE:
246 case CODEC_ID_PCM_S16LE:
247 #endif /* WORDS_BIGENDIAN */
248 case CODEC_ID_PCM_U8:
249 memcpy(dst, samples, n*sample_size);
250 dst += n*sample_size;
251 break;
252 case CODEC_ID_PCM_ZORK:
253 for(;n>0;n--) {
254 v= *samples++ >> 8;
255 if(v<0) v = -v;
256 else v+= 128;
257 *dst++ = v;
259 break;
260 case CODEC_ID_PCM_ALAW:
261 for(;n>0;n--) {
262 v = *samples++;
263 *dst++ = linear_to_alaw[(v + 32768) >> 2];
265 break;
266 case CODEC_ID_PCM_MULAW:
267 for(;n>0;n--) {
268 v = *samples++;
269 *dst++ = linear_to_ulaw[(v + 32768) >> 2];
271 break;
272 default:
273 return -1;
275 //avctx->frame_size = (dst - frame) / (sample_size * avctx->channels);
277 return dst - frame;
280 typedef struct PCMDecode {
281 short table[256];
282 } PCMDecode;
284 static av_cold int pcm_decode_init(AVCodecContext * avctx)
286 PCMDecode *s = avctx->priv_data;
287 int i;
289 switch(avctx->codec->id) {
290 case CODEC_ID_PCM_ALAW:
291 for(i=0;i<256;i++)
292 s->table[i] = alaw2linear(i);
293 break;
294 case CODEC_ID_PCM_MULAW:
295 for(i=0;i<256;i++)
296 s->table[i] = ulaw2linear(i);
297 break;
298 default:
299 break;
302 avctx->sample_fmt = avctx->codec->sample_fmts[0];
303 return 0;
307 * Read PCM samples macro
308 * @param type Datatype of native machine format
309 * @param endian bytestream_get_xxx() endian suffix
310 * @param src Source pointer (variable name)
311 * @param dst Destination pointer (variable name)
312 * @param n Total number of samples (variable name)
313 * @param shift Bitshift (bits)
314 * @param offset Sample value offset
316 #define DECODE(type, endian, src, dst, n, shift, offset) \
317 dst_##type = (type*)dst; \
318 for(;n>0;n--) { \
319 register type v = bytestream_get_##endian(&src); \
320 *dst_##type++ = (v - offset) << shift; \
322 dst = (short*)dst_##type;
324 static int pcm_decode_frame(AVCodecContext *avctx,
325 void *data, int *data_size,
326 const uint8_t *buf, int buf_size)
328 PCMDecode *s = avctx->priv_data;
329 int sample_size, c, n;
330 short *samples;
331 const uint8_t *src, *src8, *src2[MAX_CHANNELS];
332 uint8_t *dstu8;
333 int16_t *dst_int16_t;
334 int32_t *dst_int32_t;
335 int64_t *dst_int64_t;
336 uint16_t *dst_uint16_t;
337 uint32_t *dst_uint32_t;
339 samples = data;
340 src = buf;
342 if (avctx->sample_fmt!=avctx->codec->sample_fmts[0]) {
343 av_log(avctx, AV_LOG_ERROR, "invalid sample_fmt\n");
344 return -1;
347 if(avctx->channels <= 0 || avctx->channels > MAX_CHANNELS){
348 av_log(avctx, AV_LOG_ERROR, "PCM channels out of bounds\n");
349 return -1;
352 sample_size = av_get_bits_per_sample(avctx->codec_id)/8;
354 /* av_get_bits_per_sample returns 0 for CODEC_ID_PCM_DVD */
355 if (CODEC_ID_PCM_DVD == avctx->codec_id)
356 /* 2 samples are interleaved per block in PCM_DVD */
357 sample_size = avctx->bits_per_coded_sample * 2 / 8;
359 n = avctx->channels * sample_size;
361 if(n && buf_size % n){
362 av_log(avctx, AV_LOG_ERROR, "invalid PCM packet\n");
363 return -1;
366 buf_size= FFMIN(buf_size, *data_size/2);
367 *data_size=0;
369 n = buf_size/sample_size;
371 switch(avctx->codec->id) {
372 case CODEC_ID_PCM_U32LE:
373 DECODE(uint32_t, le32, src, samples, n, 0, 0x80000000)
374 break;
375 case CODEC_ID_PCM_U32BE:
376 DECODE(uint32_t, be32, src, samples, n, 0, 0x80000000)
377 break;
378 case CODEC_ID_PCM_S24LE:
379 DECODE(int32_t, le24, src, samples, n, 8, 0)
380 break;
381 case CODEC_ID_PCM_S24BE:
382 DECODE(int32_t, be24, src, samples, n, 8, 0)
383 break;
384 case CODEC_ID_PCM_U24LE:
385 DECODE(uint32_t, le24, src, samples, n, 8, 0x800000)
386 break;
387 case CODEC_ID_PCM_U24BE:
388 DECODE(uint32_t, be24, src, samples, n, 8, 0x800000)
389 break;
390 case CODEC_ID_PCM_S24DAUD:
391 for(;n>0;n--) {
392 uint32_t v = bytestream_get_be24(&src);
393 v >>= 4; // sync flags are here
394 *samples++ = ff_reverse[(v >> 8) & 0xff] +
395 (ff_reverse[v & 0xff] << 8);
397 break;
398 case CODEC_ID_PCM_S16LE_PLANAR:
399 n /= avctx->channels;
400 for(c=0;c<avctx->channels;c++)
401 src2[c] = &src[c*n*2];
402 for(;n>0;n--)
403 for(c=0;c<avctx->channels;c++)
404 *samples++ = bytestream_get_le16(&src2[c]);
405 src = src2[avctx->channels-1];
406 break;
407 case CODEC_ID_PCM_U16LE:
408 DECODE(uint16_t, le16, src, samples, n, 0, 0x8000)
409 break;
410 case CODEC_ID_PCM_U16BE:
411 DECODE(uint16_t, be16, src, samples, n, 0, 0x8000)
412 break;
413 case CODEC_ID_PCM_S8:
414 dstu8= (uint8_t*)samples;
415 for(;n>0;n--) {
416 *dstu8++ = *src++ + 128;
418 samples= (short*)dstu8;
419 break;
420 #if WORDS_BIGENDIAN
421 case CODEC_ID_PCM_F64LE:
422 DECODE(int64_t, le64, src, samples, n, 0, 0)
423 break;
424 case CODEC_ID_PCM_S32LE:
425 case CODEC_ID_PCM_F32LE:
426 DECODE(int32_t, le32, src, samples, n, 0, 0)
427 break;
428 case CODEC_ID_PCM_S16LE:
429 DECODE(int16_t, le16, src, samples, n, 0, 0)
430 break;
431 case CODEC_ID_PCM_F64BE:
432 case CODEC_ID_PCM_F32BE:
433 case CODEC_ID_PCM_S32BE:
434 case CODEC_ID_PCM_S16BE:
435 #else
436 case CODEC_ID_PCM_F64BE:
437 DECODE(int64_t, be64, src, samples, n, 0, 0)
438 break;
439 case CODEC_ID_PCM_F32BE:
440 case CODEC_ID_PCM_S32BE:
441 DECODE(int32_t, be32, src, samples, n, 0, 0)
442 break;
443 case CODEC_ID_PCM_S16BE:
444 DECODE(int16_t, be16, src, samples, n, 0, 0)
445 break;
446 case CODEC_ID_PCM_F64LE:
447 case CODEC_ID_PCM_F32LE:
448 case CODEC_ID_PCM_S32LE:
449 case CODEC_ID_PCM_S16LE:
450 #endif /* WORDS_BIGENDIAN */
451 case CODEC_ID_PCM_U8:
452 memcpy(samples, src, n*sample_size);
453 src += n*sample_size;
454 samples = (short*)((uint8_t*)data + n*sample_size);
455 break;
456 case CODEC_ID_PCM_ZORK:
457 for(;n>0;n--) {
458 int x= *src++;
459 if(x&128) x-= 128;
460 else x = -x;
461 *samples++ = x << 8;
463 break;
464 case CODEC_ID_PCM_ALAW:
465 case CODEC_ID_PCM_MULAW:
466 for(;n>0;n--) {
467 *samples++ = s->table[*src++];
469 break;
470 case CODEC_ID_PCM_DVD:
471 dst_int32_t = data;
472 n /= avctx->channels;
473 switch (avctx->bits_per_coded_sample) {
474 case 20:
475 while (n--) {
476 c = avctx->channels;
477 src8 = src + 4*c;
478 while (c--) {
479 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8 &0xf0) << 8);
480 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++ &0x0f) << 12);
482 src = src8;
484 break;
485 case 24:
486 while (n--) {
487 c = avctx->channels;
488 src8 = src + 4*c;
489 while (c--) {
490 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
491 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
493 src = src8;
495 break;
496 default:
497 av_log(avctx, AV_LOG_ERROR, "PCM DVD unsupported sample depth\n");
498 return -1;
499 break;
501 samples = (short *) dst_int32_t;
502 break;
503 default:
504 return -1;
506 *data_size = (uint8_t *)samples - (uint8_t *)data;
507 return src - buf;
510 #ifdef CONFIG_ENCODERS
511 #define PCM_ENCODER(id,sample_fmt_,name,long_name_) \
512 AVCodec name ## _encoder = { \
513 #name, \
514 CODEC_TYPE_AUDIO, \
515 id, \
516 0, \
517 pcm_encode_init, \
518 pcm_encode_frame, \
519 pcm_encode_close, \
520 NULL, \
521 .sample_fmts = (enum SampleFormat[]){sample_fmt_,SAMPLE_FMT_NONE}, \
522 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
524 #else
525 #define PCM_ENCODER(id,sample_fmt_,name,long_name_)
526 #endif
528 #ifdef CONFIG_DECODERS
529 #define PCM_DECODER(id,sample_fmt_,name,long_name_) \
530 AVCodec name ## _decoder = { \
531 #name, \
532 CODEC_TYPE_AUDIO, \
533 id, \
534 sizeof(PCMDecode), \
535 pcm_decode_init, \
536 NULL, \
537 NULL, \
538 pcm_decode_frame, \
539 .sample_fmts = (enum SampleFormat[]){sample_fmt_,SAMPLE_FMT_NONE}, \
540 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
542 #else
543 #define PCM_DECODER(id,sample_fmt_,name,long_name_)
544 #endif
546 #define PCM_CODEC(id, sample_fmt_, name, long_name_) \
547 PCM_ENCODER(id,sample_fmt_,name,long_name_) PCM_DECODER(id,sample_fmt_,name,long_name_)
549 /* Note: Do not forget to add new entries to the Makefile as well. */
550 PCM_CODEC (CODEC_ID_PCM_ALAW, SAMPLE_FMT_S16, pcm_alaw, "A-law PCM");
551 PCM_CODEC (CODEC_ID_PCM_DVD, SAMPLE_FMT_S32, pcm_dvd, "signed 20|24-bit big-endian PCM");
552 PCM_CODEC (CODEC_ID_PCM_F32BE, SAMPLE_FMT_FLT, pcm_f32be, "32-bit floating point big-endian PCM");
553 PCM_CODEC (CODEC_ID_PCM_F32LE, SAMPLE_FMT_FLT, pcm_f32le, "32-bit floating point little-endian PCM");
554 PCM_CODEC (CODEC_ID_PCM_F64BE, SAMPLE_FMT_DBL, pcm_f64be, "64-bit floating point big-endian PCM");
555 PCM_CODEC (CODEC_ID_PCM_F64LE, SAMPLE_FMT_DBL, pcm_f64le, "64-bit floating point little-endian PCM");
556 PCM_CODEC (CODEC_ID_PCM_MULAW, SAMPLE_FMT_S16, pcm_mulaw, "mu-law PCM");
557 PCM_CODEC (CODEC_ID_PCM_S8, SAMPLE_FMT_U8, pcm_s8, "signed 8-bit PCM");
558 PCM_CODEC (CODEC_ID_PCM_S16BE, SAMPLE_FMT_S16, pcm_s16be, "signed 16-bit big-endian PCM");
559 PCM_CODEC (CODEC_ID_PCM_S16LE, SAMPLE_FMT_S16, pcm_s16le, "signed 16-bit little-endian PCM");
560 PCM_DECODER(CODEC_ID_PCM_S16LE_PLANAR, SAMPLE_FMT_S16, pcm_s16le_planar, "16-bit little-endian planar PCM");
561 PCM_CODEC (CODEC_ID_PCM_S24BE, SAMPLE_FMT_S32, pcm_s24be, "signed 24-bit big-endian PCM");
562 PCM_CODEC (CODEC_ID_PCM_S24DAUD, SAMPLE_FMT_S16, pcm_s24daud, "D-Cinema audio signed 24-bit PCM");
563 PCM_CODEC (CODEC_ID_PCM_S24LE, SAMPLE_FMT_S32, pcm_s24le, "signed 24-bit little-endian PCM");
564 PCM_CODEC (CODEC_ID_PCM_S32BE, SAMPLE_FMT_S32, pcm_s32be, "signed 32-bit big-endian PCM");
565 PCM_CODEC (CODEC_ID_PCM_S32LE, SAMPLE_FMT_S32, pcm_s32le, "signed 32-bit little-endian PCM");
566 PCM_CODEC (CODEC_ID_PCM_U8, SAMPLE_FMT_U8, pcm_u8, "unsigned 8-bit PCM");
567 PCM_CODEC (CODEC_ID_PCM_U16BE, SAMPLE_FMT_S16, pcm_u16be, "unsigned 16-bit big-endian PCM");
568 PCM_CODEC (CODEC_ID_PCM_U16LE, SAMPLE_FMT_S16, pcm_u16le, "unsigned 16-bit little-endian PCM");
569 PCM_CODEC (CODEC_ID_PCM_U24BE, SAMPLE_FMT_S32, pcm_u24be, "unsigned 24-bit big-endian PCM");
570 PCM_CODEC (CODEC_ID_PCM_U24LE, SAMPLE_FMT_S32, pcm_u24le, "unsigned 24-bit little-endian PCM");
571 PCM_CODEC (CODEC_ID_PCM_U32BE, SAMPLE_FMT_S32, pcm_u32be, "unsigned 32-bit big-endian PCM");
572 PCM_CODEC (CODEC_ID_PCM_U32LE, SAMPLE_FMT_S32, pcm_u32le, "unsigned 32-bit little-endian PCM");
573 PCM_CODEC (CODEC_ID_PCM_ZORK, SAMPLE_FMT_S16, pcm_zork, "Zork PCM");