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
23 * @file libavcodec/pcm.c
28 #include "libavutil/common.h" /* for av_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)
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)
66 /* Complement to obtain normal u-law value. */
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),
92 v1 = xlaw2linear(i ^ mask);
93 v2 = xlaw2linear((i + 1) ^ mask);
94 v = (v1 + v2 + 4) >> 3;
99 linear_to_xlaw[8192 + j] = (i ^ mask);
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);
114 case CODEC_ID_PCM_MULAW:
115 build_xlaw_table(linear_to_ulaw, ulaw2linear, 0xff);
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;
129 static av_cold int pcm_encode_close(AVCodecContext *avctx)
131 av_freep(&avctx->coded_frame);
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; \
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;
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;
171 if (avctx->sample_fmt!=avctx->codec->sample_fmts[0]) {
172 av_log(avctx, AV_LOG_ERROR, "invalid sample_fmt\n");
176 switch(avctx->codec->id) {
177 case CODEC_ID_PCM_U32LE:
178 ENCODE(uint32_t, le32, samples, dst, n, 0, 0x80000000)
180 case CODEC_ID_PCM_U32BE:
181 ENCODE(uint32_t, be32, samples, dst, n, 0, 0x80000000)
183 case CODEC_ID_PCM_S24LE:
184 ENCODE(int32_t, le24, samples, dst, n, 8, 0)
186 case CODEC_ID_PCM_S24BE:
187 ENCODE(int32_t, be24, samples, dst, n, 8, 0)
189 case CODEC_ID_PCM_U24LE:
190 ENCODE(uint32_t, le24, samples, dst, n, 8, 0x800000)
192 case CODEC_ID_PCM_U24BE:
193 ENCODE(uint32_t, be24, samples, dst, n, 8, 0x800000)
195 case CODEC_ID_PCM_S24DAUD:
197 uint32_t tmp = av_reverse[(*samples >> 8) & 0xff] +
198 (av_reverse[*samples & 0xff] << 8);
199 tmp <<= 4; // sync flags would go here
200 bytestream_put_be24(&dst, tmp);
204 case CODEC_ID_PCM_U16LE:
205 ENCODE(uint16_t, le16, samples, dst, n, 0, 0x8000)
207 case CODEC_ID_PCM_U16BE:
208 ENCODE(uint16_t, be16, samples, dst, n, 0, 0x8000)
210 case CODEC_ID_PCM_S8:
218 case CODEC_ID_PCM_F64LE:
219 ENCODE(int64_t, le64, samples, dst, n, 0, 0)
221 case CODEC_ID_PCM_S32LE:
222 case CODEC_ID_PCM_F32LE:
223 ENCODE(int32_t, le32, samples, dst, n, 0, 0)
225 case CODEC_ID_PCM_S16LE:
226 ENCODE(int16_t, le16, samples, dst, n, 0, 0)
228 case CODEC_ID_PCM_F64BE:
229 case CODEC_ID_PCM_F32BE:
230 case CODEC_ID_PCM_S32BE:
231 case CODEC_ID_PCM_S16BE:
233 case CODEC_ID_PCM_F64BE:
234 ENCODE(int64_t, be64, samples, dst, n, 0, 0)
236 case CODEC_ID_PCM_F32BE:
237 case CODEC_ID_PCM_S32BE:
238 ENCODE(int32_t, be32, samples, dst, n, 0, 0)
240 case CODEC_ID_PCM_S16BE:
241 ENCODE(int16_t, be16, samples, dst, n, 0, 0)
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 /* HAVE_BIGENDIAN */
248 case CODEC_ID_PCM_U8:
249 memcpy(dst, samples, n*sample_size);
250 dst += n*sample_size;
252 case CODEC_ID_PCM_ZORK:
260 case CODEC_ID_PCM_ALAW:
263 *dst++ = linear_to_alaw[(v + 32768) >> 2];
266 case CODEC_ID_PCM_MULAW:
269 *dst++ = linear_to_ulaw[(v + 32768) >> 2];
275 //avctx->frame_size = (dst - frame) / (sample_size * avctx->channels);
280 typedef struct PCMDecode {
284 static av_cold int pcm_decode_init(AVCodecContext * avctx)
286 PCMDecode *s = avctx->priv_data;
289 switch(avctx->codec->id) {
290 case CODEC_ID_PCM_ALAW:
292 s->table[i] = alaw2linear(i);
294 case CODEC_ID_PCM_MULAW:
296 s->table[i] = ulaw2linear(i);
302 avctx->sample_fmt = avctx->codec->sample_fmts[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; \
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,
328 const uint8_t *buf = avpkt->data;
329 int buf_size = avpkt->size;
330 PCMDecode *s = avctx->priv_data;
331 int sample_size, c, n;
333 const uint8_t *src, *src8, *src2[MAX_CHANNELS];
335 int16_t *dst_int16_t;
336 int32_t *dst_int32_t;
337 int64_t *dst_int64_t;
338 uint16_t *dst_uint16_t;
339 uint32_t *dst_uint32_t;
344 if (avctx->sample_fmt!=avctx->codec->sample_fmts[0]) {
345 av_log(avctx, AV_LOG_ERROR, "invalid sample_fmt\n");
349 if(avctx->channels <= 0 || avctx->channels > MAX_CHANNELS){
350 av_log(avctx, AV_LOG_ERROR, "PCM channels out of bounds\n");
354 sample_size = av_get_bits_per_sample(avctx->codec_id)/8;
356 /* av_get_bits_per_sample returns 0 for CODEC_ID_PCM_DVD */
357 if (CODEC_ID_PCM_DVD == avctx->codec_id)
358 /* 2 samples are interleaved per block in PCM_DVD */
359 sample_size = avctx->bits_per_coded_sample * 2 / 8;
361 n = avctx->channels * sample_size;
363 if(n && buf_size % n){
365 av_log(avctx, AV_LOG_ERROR, "invalid PCM packet\n");
368 buf_size -= buf_size % n;
371 buf_size= FFMIN(buf_size, *data_size/2);
374 n = buf_size/sample_size;
376 switch(avctx->codec->id) {
377 case CODEC_ID_PCM_U32LE:
378 DECODE(uint32_t, le32, src, samples, n, 0, 0x80000000)
380 case CODEC_ID_PCM_U32BE:
381 DECODE(uint32_t, be32, src, samples, n, 0, 0x80000000)
383 case CODEC_ID_PCM_S24LE:
384 DECODE(int32_t, le24, src, samples, n, 8, 0)
386 case CODEC_ID_PCM_S24BE:
387 DECODE(int32_t, be24, src, samples, n, 8, 0)
389 case CODEC_ID_PCM_U24LE:
390 DECODE(uint32_t, le24, src, samples, n, 8, 0x800000)
392 case CODEC_ID_PCM_U24BE:
393 DECODE(uint32_t, be24, src, samples, n, 8, 0x800000)
395 case CODEC_ID_PCM_S24DAUD:
397 uint32_t v = bytestream_get_be24(&src);
398 v >>= 4; // sync flags are here
399 *samples++ = av_reverse[(v >> 8) & 0xff] +
400 (av_reverse[v & 0xff] << 8);
403 case CODEC_ID_PCM_S16LE_PLANAR:
404 n /= avctx->channels;
405 for(c=0;c<avctx->channels;c++)
406 src2[c] = &src[c*n*2];
408 for(c=0;c<avctx->channels;c++)
409 *samples++ = bytestream_get_le16(&src2[c]);
410 src = src2[avctx->channels-1];
412 case CODEC_ID_PCM_U16LE:
413 DECODE(uint16_t, le16, src, samples, n, 0, 0x8000)
415 case CODEC_ID_PCM_U16BE:
416 DECODE(uint16_t, be16, src, samples, n, 0, 0x8000)
418 case CODEC_ID_PCM_S8:
419 dstu8= (uint8_t*)samples;
421 *dstu8++ = *src++ + 128;
423 samples= (short*)dstu8;
426 case CODEC_ID_PCM_F64LE:
427 DECODE(int64_t, le64, src, samples, n, 0, 0)
429 case CODEC_ID_PCM_S32LE:
430 case CODEC_ID_PCM_F32LE:
431 DECODE(int32_t, le32, src, samples, n, 0, 0)
433 case CODEC_ID_PCM_S16LE:
434 DECODE(int16_t, le16, src, samples, n, 0, 0)
436 case CODEC_ID_PCM_F64BE:
437 case CODEC_ID_PCM_F32BE:
438 case CODEC_ID_PCM_S32BE:
439 case CODEC_ID_PCM_S16BE:
441 case CODEC_ID_PCM_F64BE:
442 DECODE(int64_t, be64, src, samples, n, 0, 0)
444 case CODEC_ID_PCM_F32BE:
445 case CODEC_ID_PCM_S32BE:
446 DECODE(int32_t, be32, src, samples, n, 0, 0)
448 case CODEC_ID_PCM_S16BE:
449 DECODE(int16_t, be16, src, samples, n, 0, 0)
451 case CODEC_ID_PCM_F64LE:
452 case CODEC_ID_PCM_F32LE:
453 case CODEC_ID_PCM_S32LE:
454 case CODEC_ID_PCM_S16LE:
455 #endif /* HAVE_BIGENDIAN */
456 case CODEC_ID_PCM_U8:
457 memcpy(samples, src, n*sample_size);
458 src += n*sample_size;
459 samples = (short*)((uint8_t*)data + n*sample_size);
461 case CODEC_ID_PCM_ZORK:
469 case CODEC_ID_PCM_ALAW:
470 case CODEC_ID_PCM_MULAW:
472 *samples++ = s->table[*src++];
475 case CODEC_ID_PCM_DVD:
477 n /= avctx->channels;
478 switch (avctx->bits_per_coded_sample) {
484 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8 &0xf0) << 8);
485 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++ &0x0f) << 12);
495 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
496 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
502 av_log(avctx, AV_LOG_ERROR, "PCM DVD unsupported sample depth\n");
506 samples = (short *) dst_int32_t;
511 *data_size = (uint8_t *)samples - (uint8_t *)data;
516 #define PCM_ENCODER(id,sample_fmt_,name,long_name_) \
517 AVCodec name ## _encoder = { \
519 AVMEDIA_TYPE_AUDIO, \
526 .sample_fmts = (const enum SampleFormat[]){sample_fmt_,SAMPLE_FMT_NONE}, \
527 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
530 #define PCM_ENCODER(id,sample_fmt_,name,long_name_)
534 #define PCM_DECODER(id,sample_fmt_,name,long_name_) \
535 AVCodec name ## _decoder = { \
537 AVMEDIA_TYPE_AUDIO, \
544 .sample_fmts = (const enum SampleFormat[]){sample_fmt_,SAMPLE_FMT_NONE}, \
545 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
548 #define PCM_DECODER(id,sample_fmt_,name,long_name_)
551 #define PCM_CODEC(id, sample_fmt_, name, long_name_) \
552 PCM_ENCODER(id,sample_fmt_,name,long_name_) PCM_DECODER(id,sample_fmt_,name,long_name_)
554 /* Note: Do not forget to add new entries to the Makefile as well. */
555 PCM_CODEC (CODEC_ID_PCM_ALAW, SAMPLE_FMT_S16, pcm_alaw, "PCM A-law");
556 PCM_CODEC (CODEC_ID_PCM_DVD, SAMPLE_FMT_S32, pcm_dvd, "PCM signed 20|24-bit big-endian");
557 PCM_CODEC (CODEC_ID_PCM_F32BE, SAMPLE_FMT_FLT, pcm_f32be, "PCM 32-bit floating point big-endian");
558 PCM_CODEC (CODEC_ID_PCM_F32LE, SAMPLE_FMT_FLT, pcm_f32le, "PCM 32-bit floating point little-endian");
559 PCM_CODEC (CODEC_ID_PCM_F64BE, SAMPLE_FMT_DBL, pcm_f64be, "PCM 64-bit floating point big-endian");
560 PCM_CODEC (CODEC_ID_PCM_F64LE, SAMPLE_FMT_DBL, pcm_f64le, "PCM 64-bit floating point little-endian");
561 PCM_CODEC (CODEC_ID_PCM_MULAW, SAMPLE_FMT_S16, pcm_mulaw, "PCM mu-law");
562 PCM_CODEC (CODEC_ID_PCM_S8, SAMPLE_FMT_U8, pcm_s8, "PCM signed 8-bit");
563 PCM_CODEC (CODEC_ID_PCM_S16BE, SAMPLE_FMT_S16, pcm_s16be, "PCM signed 16-bit big-endian");
564 PCM_CODEC (CODEC_ID_PCM_S16LE, SAMPLE_FMT_S16, pcm_s16le, "PCM signed 16-bit little-endian");
565 PCM_DECODER(CODEC_ID_PCM_S16LE_PLANAR, SAMPLE_FMT_S16, pcm_s16le_planar, "PCM 16-bit little-endian planar");
566 PCM_CODEC (CODEC_ID_PCM_S24BE, SAMPLE_FMT_S32, pcm_s24be, "PCM signed 24-bit big-endian");
567 PCM_CODEC (CODEC_ID_PCM_S24DAUD, SAMPLE_FMT_S16, pcm_s24daud, "PCM D-Cinema audio signed 24-bit");
568 PCM_CODEC (CODEC_ID_PCM_S24LE, SAMPLE_FMT_S32, pcm_s24le, "PCM signed 24-bit little-endian");
569 PCM_CODEC (CODEC_ID_PCM_S32BE, SAMPLE_FMT_S32, pcm_s32be, "PCM signed 32-bit big-endian");
570 PCM_CODEC (CODEC_ID_PCM_S32LE, SAMPLE_FMT_S32, pcm_s32le, "PCM signed 32-bit little-endian");
571 PCM_CODEC (CODEC_ID_PCM_U8, SAMPLE_FMT_U8, pcm_u8, "PCM unsigned 8-bit");
572 PCM_CODEC (CODEC_ID_PCM_U16BE, SAMPLE_FMT_S16, pcm_u16be, "PCM unsigned 16-bit big-endian");
573 PCM_CODEC (CODEC_ID_PCM_U16LE, SAMPLE_FMT_S16, pcm_u16le, "PCM unsigned 16-bit little-endian");
574 PCM_CODEC (CODEC_ID_PCM_U24BE, SAMPLE_FMT_S32, pcm_u24be, "PCM unsigned 24-bit big-endian");
575 PCM_CODEC (CODEC_ID_PCM_U24LE, SAMPLE_FMT_S32, pcm_u24le, "PCM unsigned 24-bit little-endian");
576 PCM_CODEC (CODEC_ID_PCM_U32BE, SAMPLE_FMT_S32, pcm_u32be, "PCM unsigned 32-bit big-endian");
577 PCM_CODEC (CODEC_ID_PCM_U32LE, SAMPLE_FMT_S32, pcm_u32le, "PCM unsigned 32-bit little-endian");
578 PCM_CODEC (CODEC_ID_PCM_ZORK, SAMPLE_FMT_S16, pcm_zork, "PCM Zork");