00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00027 #include "avcodec.h"
00028 #include "bitstream.h"
00029 #include "bytestream.h"
00030
00031 #define MAX_CHANNELS 64
00032
00033
00034
00035 #define SIGN_BIT (0x80)
00036 #define QUANT_MASK (0xf)
00037 #define NSEGS (8)
00038 #define SEG_SHIFT (4)
00039 #define SEG_MASK (0x70)
00040
00041 #define BIAS (0x84)
00042
00043
00044
00045
00046
00047 static av_cold int alaw2linear(unsigned char a_val)
00048 {
00049 int t;
00050 int seg;
00051
00052 a_val ^= 0x55;
00053
00054 t = a_val & QUANT_MASK;
00055 seg = ((unsigned)a_val & SEG_MASK) >> SEG_SHIFT;
00056 if(seg) t= (t + t + 1 + 32) << (seg + 2);
00057 else t= (t + t + 1 ) << 3;
00058
00059 return (a_val & SIGN_BIT) ? t : -t;
00060 }
00061
00062 static av_cold int ulaw2linear(unsigned char u_val)
00063 {
00064 int t;
00065
00066
00067 u_val = ~u_val;
00068
00069
00070
00071
00072
00073 t = ((u_val & QUANT_MASK) << 3) + BIAS;
00074 t <<= ((unsigned)u_val & SEG_MASK) >> SEG_SHIFT;
00075
00076 return (u_val & SIGN_BIT) ? (BIAS - t) : (t - BIAS);
00077 }
00078
00079
00080 static uint8_t linear_to_alaw[16384];
00081 static uint8_t linear_to_ulaw[16384];
00082
00083 static av_cold void build_xlaw_table(uint8_t *linear_to_xlaw,
00084 int (*xlaw2linear)(unsigned char),
00085 int mask)
00086 {
00087 int i, j, v, v1, v2;
00088
00089 j = 0;
00090 for(i=0;i<128;i++) {
00091 if (i != 127) {
00092 v1 = xlaw2linear(i ^ mask);
00093 v2 = xlaw2linear((i + 1) ^ mask);
00094 v = (v1 + v2 + 4) >> 3;
00095 } else {
00096 v = 8192;
00097 }
00098 for(;j<v;j++) {
00099 linear_to_xlaw[8192 + j] = (i ^ mask);
00100 if (j > 0)
00101 linear_to_xlaw[8192 - j] = (i ^ (mask ^ 0x80));
00102 }
00103 }
00104 linear_to_xlaw[0] = linear_to_xlaw[1];
00105 }
00106
00107 static av_cold int pcm_encode_init(AVCodecContext *avctx)
00108 {
00109 avctx->frame_size = 1;
00110 switch(avctx->codec->id) {
00111 case CODEC_ID_PCM_ALAW:
00112 build_xlaw_table(linear_to_alaw, alaw2linear, 0xd5);
00113 break;
00114 case CODEC_ID_PCM_MULAW:
00115 build_xlaw_table(linear_to_ulaw, ulaw2linear, 0xff);
00116 break;
00117 default:
00118 break;
00119 }
00120
00121 avctx->bits_per_coded_sample = av_get_bits_per_sample(avctx->codec->id);
00122 avctx->block_align = avctx->channels * avctx->bits_per_coded_sample/8;
00123 avctx->coded_frame= avcodec_alloc_frame();
00124 avctx->coded_frame->key_frame= 1;
00125
00126 return 0;
00127 }
00128
00129 static av_cold int pcm_encode_close(AVCodecContext *avctx)
00130 {
00131 av_freep(&avctx->coded_frame);
00132
00133 return 0;
00134 }
00135
00146 #define ENCODE(type, endian, src, dst, n, shift, offset) \
00147 samples_##type = (type*)src; \
00148 for(;n>0;n--) { \
00149 register type v = (*samples_##type++ >> shift) + offset; \
00150 bytestream_put_##endian(&dst, v); \
00151 }
00152
00153 static int pcm_encode_frame(AVCodecContext *avctx,
00154 unsigned char *frame, int buf_size, void *data)
00155 {
00156 int n, sample_size, v;
00157 short *samples;
00158 unsigned char *dst;
00159 uint8_t *srcu8;
00160 int16_t *samples_int16_t;
00161 int32_t *samples_int32_t;
00162 int64_t *samples_int64_t;
00163 uint16_t *samples_uint16_t;
00164 uint32_t *samples_uint32_t;
00165
00166 sample_size = av_get_bits_per_sample(avctx->codec->id)/8;
00167 n = buf_size / sample_size;
00168 samples = data;
00169 dst = frame;
00170
00171 if (avctx->sample_fmt!=avctx->codec->sample_fmts[0]) {
00172 av_log(avctx, AV_LOG_ERROR, "invalid sample_fmt\n");
00173 return -1;
00174 }
00175
00176 switch(avctx->codec->id) {
00177 case CODEC_ID_PCM_U32LE:
00178 ENCODE(uint32_t, le32, samples, dst, n, 0, 0x80000000)
00179 break;
00180 case CODEC_ID_PCM_U32BE:
00181 ENCODE(uint32_t, be32, samples, dst, n, 0, 0x80000000)
00182 break;
00183 case CODEC_ID_PCM_S24LE:
00184 ENCODE(int32_t, le24, samples, dst, n, 8, 0)
00185 break;
00186 case CODEC_ID_PCM_S24BE:
00187 ENCODE(int32_t, be24, samples, dst, n, 8, 0)
00188 break;
00189 case CODEC_ID_PCM_U24LE:
00190 ENCODE(uint32_t, le24, samples, dst, n, 8, 0x800000)
00191 break;
00192 case CODEC_ID_PCM_U24BE:
00193 ENCODE(uint32_t, be24, samples, dst, n, 8, 0x800000)
00194 break;
00195 case CODEC_ID_PCM_S24DAUD:
00196 for(;n>0;n--) {
00197 uint32_t tmp = ff_reverse[(*samples >> 8) & 0xff] +
00198 (ff_reverse[*samples & 0xff] << 8);
00199 tmp <<= 4;
00200 bytestream_put_be24(&dst, tmp);
00201 samples++;
00202 }
00203 break;
00204 case CODEC_ID_PCM_U16LE:
00205 ENCODE(uint16_t, le16, samples, dst, n, 0, 0x8000)
00206 break;
00207 case CODEC_ID_PCM_U16BE:
00208 ENCODE(uint16_t, be16, samples, dst, n, 0, 0x8000)
00209 break;
00210 case CODEC_ID_PCM_S8:
00211 srcu8= data;
00212 for(;n>0;n--) {
00213 v = *srcu8++;
00214 *dst++ = v - 128;
00215 }
00216 break;
00217 #ifdef WORDS_BIGENDIAN
00218 case CODEC_ID_PCM_F64LE:
00219 ENCODE(int64_t, le64, samples, dst, n, 0, 0)
00220 break;
00221 case CODEC_ID_PCM_S32LE:
00222 case CODEC_ID_PCM_F32LE:
00223 ENCODE(int32_t, le32, samples, dst, n, 0, 0)
00224 break;
00225 case CODEC_ID_PCM_S16LE:
00226 ENCODE(int16_t, le16, samples, dst, n, 0, 0)
00227 break;
00228 case CODEC_ID_PCM_F64BE:
00229 case CODEC_ID_PCM_F32BE:
00230 case CODEC_ID_PCM_S32BE:
00231 case CODEC_ID_PCM_S16BE:
00232 #else
00233 case CODEC_ID_PCM_F64BE:
00234 ENCODE(int64_t, be64, samples, dst, n, 0, 0)
00235 break;
00236 case CODEC_ID_PCM_F32BE:
00237 case CODEC_ID_PCM_S32BE:
00238 ENCODE(int32_t, be32, samples, dst, n, 0, 0)
00239 break;
00240 case CODEC_ID_PCM_S16BE:
00241 ENCODE(int16_t, be16, samples, dst, n, 0, 0)
00242 break;
00243 case CODEC_ID_PCM_F64LE:
00244 case CODEC_ID_PCM_F32LE:
00245 case CODEC_ID_PCM_S32LE:
00246 case CODEC_ID_PCM_S16LE:
00247 #endif
00248 case CODEC_ID_PCM_U8:
00249 memcpy(dst, samples, n*sample_size);
00250 dst += n*sample_size;
00251 break;
00252 case CODEC_ID_PCM_ZORK:
00253 for(;n>0;n--) {
00254 v= *samples++ >> 8;
00255 if(v<0) v = -v;
00256 else v+= 128;
00257 *dst++ = v;
00258 }
00259 break;
00260 case CODEC_ID_PCM_ALAW:
00261 for(;n>0;n--) {
00262 v = *samples++;
00263 *dst++ = linear_to_alaw[(v + 32768) >> 2];
00264 }
00265 break;
00266 case CODEC_ID_PCM_MULAW:
00267 for(;n>0;n--) {
00268 v = *samples++;
00269 *dst++ = linear_to_ulaw[(v + 32768) >> 2];
00270 }
00271 break;
00272 default:
00273 return -1;
00274 }
00275
00276
00277 return dst - frame;
00278 }
00279
00280 typedef struct PCMDecode {
00281 short table[256];
00282 } PCMDecode;
00283
00284 static av_cold int pcm_decode_init(AVCodecContext * avctx)
00285 {
00286 PCMDecode *s = avctx->priv_data;
00287 int i;
00288
00289 switch(avctx->codec->id) {
00290 case CODEC_ID_PCM_ALAW:
00291 for(i=0;i<256;i++)
00292 s->table[i] = alaw2linear(i);
00293 break;
00294 case CODEC_ID_PCM_MULAW:
00295 for(i=0;i<256;i++)
00296 s->table[i] = ulaw2linear(i);
00297 break;
00298 default:
00299 break;
00300 }
00301
00302 avctx->sample_fmt = avctx->codec->sample_fmts[0];
00303 return 0;
00304 }
00305
00316 #define DECODE(type, endian, src, dst, n, shift, offset) \
00317 dst_##type = (type*)dst; \
00318 for(;n>0;n--) { \
00319 register type v = bytestream_get_##endian(&src); \
00320 *dst_##type++ = (v - offset) << shift; \
00321 } \
00322 dst = (short*)dst_##type;
00323
00324 static int pcm_decode_frame(AVCodecContext *avctx,
00325 void *data, int *data_size,
00326 const uint8_t *buf, int buf_size)
00327 {
00328 PCMDecode *s = avctx->priv_data;
00329 int sample_size, c, n;
00330 short *samples;
00331 const uint8_t *src, *src8, *src2[MAX_CHANNELS];
00332 uint8_t *dstu8;
00333 int16_t *dst_int16_t;
00334 int32_t *dst_int32_t;
00335 int64_t *dst_int64_t;
00336 uint16_t *dst_uint16_t;
00337 uint32_t *dst_uint32_t;
00338
00339 samples = data;
00340 src = buf;
00341
00342 if (avctx->sample_fmt!=avctx->codec->sample_fmts[0]) {
00343 av_log(avctx, AV_LOG_ERROR, "invalid sample_fmt\n");
00344 return -1;
00345 }
00346
00347 if(avctx->channels <= 0 || avctx->channels > MAX_CHANNELS){
00348 av_log(avctx, AV_LOG_ERROR, "PCM channels out of bounds\n");
00349 return -1;
00350 }
00351
00352 sample_size = av_get_bits_per_sample(avctx->codec_id)/8;
00353
00354
00355 if (CODEC_ID_PCM_DVD == avctx->codec_id)
00356
00357 sample_size = avctx->bits_per_coded_sample * 2 / 8;
00358
00359 n = avctx->channels * sample_size;
00360
00361 if(n && buf_size % n){
00362 av_log(avctx, AV_LOG_ERROR, "invalid PCM packet\n");
00363 return -1;
00364 }
00365
00366 buf_size= FFMIN(buf_size, *data_size/2);
00367 *data_size=0;
00368
00369 n = buf_size/sample_size;
00370
00371 switch(avctx->codec->id) {
00372 case CODEC_ID_PCM_U32LE:
00373 DECODE(uint32_t, le32, src, samples, n, 0, 0x80000000)
00374 break;
00375 case CODEC_ID_PCM_U32BE:
00376 DECODE(uint32_t, be32, src, samples, n, 0, 0x80000000)
00377 break;
00378 case CODEC_ID_PCM_S24LE:
00379 DECODE(int32_t, le24, src, samples, n, 8, 0)
00380 break;
00381 case CODEC_ID_PCM_S24BE:
00382 DECODE(int32_t, be24, src, samples, n, 8, 0)
00383 break;
00384 case CODEC_ID_PCM_U24LE:
00385 DECODE(uint32_t, le24, src, samples, n, 8, 0x800000)
00386 break;
00387 case CODEC_ID_PCM_U24BE:
00388 DECODE(uint32_t, be24, src, samples, n, 8, 0x800000)
00389 break;
00390 case CODEC_ID_PCM_S24DAUD:
00391 for(;n>0;n--) {
00392 uint32_t v = bytestream_get_be24(&src);
00393 v >>= 4;
00394 *samples++ = ff_reverse[(v >> 8) & 0xff] +
00395 (ff_reverse[v & 0xff] << 8);
00396 }
00397 break;
00398 case CODEC_ID_PCM_S16LE_PLANAR:
00399 n /= avctx->channels;
00400 for(c=0;c<avctx->channels;c++)
00401 src2[c] = &src[c*n*2];
00402 for(;n>0;n--)
00403 for(c=0;c<avctx->channels;c++)
00404 *samples++ = bytestream_get_le16(&src2[c]);
00405 src = src2[avctx->channels-1];
00406 break;
00407 case CODEC_ID_PCM_U16LE:
00408 DECODE(uint16_t, le16, src, samples, n, 0, 0x8000)
00409 break;
00410 case CODEC_ID_PCM_U16BE:
00411 DECODE(uint16_t, be16, src, samples, n, 0, 0x8000)
00412 break;
00413 case CODEC_ID_PCM_S8:
00414 dstu8= (uint8_t*)samples;
00415 for(;n>0;n--) {
00416 *dstu8++ = *src++ + 128;
00417 }
00418 samples= (short*)dstu8;
00419 break;
00420 #ifdef WORDS_BIGENDIAN
00421 case CODEC_ID_PCM_F64LE:
00422 DECODE(int64_t, le64, src, samples, n, 0, 0)
00423 break;
00424 case CODEC_ID_PCM_S32LE:
00425 case CODEC_ID_PCM_F32LE:
00426 DECODE(int32_t, le32, src, samples, n, 0, 0)
00427 break;
00428 case CODEC_ID_PCM_S16LE:
00429 DECODE(int16_t, le16, src, samples, n, 0, 0)
00430 break;
00431 case CODEC_ID_PCM_F64BE:
00432 case CODEC_ID_PCM_F32BE:
00433 case CODEC_ID_PCM_S32BE:
00434 case CODEC_ID_PCM_S16BE:
00435 #else
00436 case CODEC_ID_PCM_F64BE:
00437 DECODE(int64_t, be64, src, samples, n, 0, 0)
00438 break;
00439 case CODEC_ID_PCM_F32BE:
00440 case CODEC_ID_PCM_S32BE:
00441 DECODE(int32_t, be32, src, samples, n, 0, 0)
00442 break;
00443 case CODEC_ID_PCM_S16BE:
00444 DECODE(int16_t, be16, src, samples, n, 0, 0)
00445 break;
00446 case CODEC_ID_PCM_F64LE:
00447 case CODEC_ID_PCM_F32LE:
00448 case CODEC_ID_PCM_S32LE:
00449 case CODEC_ID_PCM_S16LE:
00450 #endif
00451 case CODEC_ID_PCM_U8:
00452 memcpy(samples, src, n*sample_size);
00453 src += n*sample_size;
00454 samples = (short*)((uint8_t*)data + n*sample_size);
00455 break;
00456 case CODEC_ID_PCM_ZORK:
00457 for(;n>0;n--) {
00458 int x= *src++;
00459 if(x&128) x-= 128;
00460 else x = -x;
00461 *samples++ = x << 8;
00462 }
00463 break;
00464 case CODEC_ID_PCM_ALAW:
00465 case CODEC_ID_PCM_MULAW:
00466 for(;n>0;n--) {
00467 *samples++ = s->table[*src++];
00468 }
00469 break;
00470 case CODEC_ID_PCM_DVD:
00471 dst_int32_t = data;
00472 n /= avctx->channels;
00473 switch (avctx->bits_per_coded_sample) {
00474 case 20:
00475 while (n--) {
00476 c = avctx->channels;
00477 src8 = src + 4*c;
00478 while (c--) {
00479 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8 &0xf0) << 8);
00480 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++ &0x0f) << 12);
00481 }
00482 src = src8;
00483 }
00484 break;
00485 case 24:
00486 while (n--) {
00487 c = avctx->channels;
00488 src8 = src + 4*c;
00489 while (c--) {
00490 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
00491 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
00492 }
00493 src = src8;
00494 }
00495 break;
00496 default:
00497 av_log(avctx, AV_LOG_ERROR, "PCM DVD unsupported sample depth\n");
00498 return -1;
00499 break;
00500 }
00501 samples = (short *) dst_int32_t;
00502 break;
00503 default:
00504 return -1;
00505 }
00506 *data_size = (uint8_t *)samples - (uint8_t *)data;
00507 return src - buf;
00508 }
00509
00510 #if CONFIG_ENCODERS
00511 #define PCM_ENCODER(id,sample_fmt_,name,long_name_) \
00512 AVCodec name ## _encoder = { \
00513 #name, \
00514 CODEC_TYPE_AUDIO, \
00515 id, \
00516 0, \
00517 pcm_encode_init, \
00518 pcm_encode_frame, \
00519 pcm_encode_close, \
00520 NULL, \
00521 .sample_fmts = (enum SampleFormat[]){sample_fmt_,SAMPLE_FMT_NONE}, \
00522 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
00523 };
00524 #else
00525 #define PCM_ENCODER(id,sample_fmt_,name,long_name_)
00526 #endif
00527
00528 #if CONFIG_DECODERS
00529 #define PCM_DECODER(id,sample_fmt_,name,long_name_) \
00530 AVCodec name ## _decoder = { \
00531 #name, \
00532 CODEC_TYPE_AUDIO, \
00533 id, \
00534 sizeof(PCMDecode), \
00535 pcm_decode_init, \
00536 NULL, \
00537 NULL, \
00538 pcm_decode_frame, \
00539 .sample_fmts = (enum SampleFormat[]){sample_fmt_,SAMPLE_FMT_NONE}, \
00540 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
00541 };
00542 #else
00543 #define PCM_DECODER(id,sample_fmt_,name,long_name_)
00544 #endif
00545
00546 #define PCM_CODEC(id, sample_fmt_, name, long_name_) \
00547 PCM_ENCODER(id,sample_fmt_,name,long_name_) PCM_DECODER(id,sample_fmt_,name,long_name_)
00548
00549
00550 PCM_CODEC (CODEC_ID_PCM_ALAW, SAMPLE_FMT_S16, pcm_alaw, "PCM A-law");
00551 PCM_CODEC (CODEC_ID_PCM_DVD, SAMPLE_FMT_S32, pcm_dvd, "PCM signed 20|24-bit big-endian");
00552 PCM_CODEC (CODEC_ID_PCM_F32BE, SAMPLE_FMT_FLT, pcm_f32be, "PCM 32-bit floating point big-endian");
00553 PCM_CODEC (CODEC_ID_PCM_F32LE, SAMPLE_FMT_FLT, pcm_f32le, "PCM 32-bit floating point little-endian");
00554 PCM_CODEC (CODEC_ID_PCM_F64BE, SAMPLE_FMT_DBL, pcm_f64be, "PCM 64-bit floating point big-endian");
00555 PCM_CODEC (CODEC_ID_PCM_F64LE, SAMPLE_FMT_DBL, pcm_f64le, "PCM 64-bit floating point little-endian");
00556 PCM_CODEC (CODEC_ID_PCM_MULAW, SAMPLE_FMT_S16, pcm_mulaw, "PCM mu-law");
00557 PCM_CODEC (CODEC_ID_PCM_S8, SAMPLE_FMT_U8, pcm_s8, "PCM signed 8-bit");
00558 PCM_CODEC (CODEC_ID_PCM_S16BE, SAMPLE_FMT_S16, pcm_s16be, "PCM signed 16-bit big-endian");
00559 PCM_CODEC (CODEC_ID_PCM_S16LE, SAMPLE_FMT_S16, pcm_s16le, "PCM signed 16-bit little-endian");
00560 PCM_DECODER(CODEC_ID_PCM_S16LE_PLANAR, SAMPLE_FMT_S16, pcm_s16le_planar, "PCM 16-bit little-endian planar");
00561 PCM_CODEC (CODEC_ID_PCM_S24BE, SAMPLE_FMT_S32, pcm_s24be, "PCM signed 24-bit big-endian");
00562 PCM_CODEC (CODEC_ID_PCM_S24DAUD, SAMPLE_FMT_S16, pcm_s24daud, "PCM D-Cinema audio signed 24-bit");
00563 PCM_CODEC (CODEC_ID_PCM_S24LE, SAMPLE_FMT_S32, pcm_s24le, "PCM signed 24-bit little-endian");
00564 PCM_CODEC (CODEC_ID_PCM_S32BE, SAMPLE_FMT_S32, pcm_s32be, "PCM signed 32-bit big-endian");
00565 PCM_CODEC (CODEC_ID_PCM_S32LE, SAMPLE_FMT_S32, pcm_s32le, "PCM signed 32-bit little-endian");
00566 PCM_CODEC (CODEC_ID_PCM_U8, SAMPLE_FMT_U8, pcm_u8, "PCM unsigned 8-bit");
00567 PCM_CODEC (CODEC_ID_PCM_U16BE, SAMPLE_FMT_S16, pcm_u16be, "PCM unsigned 16-bit big-endian");
00568 PCM_CODEC (CODEC_ID_PCM_U16LE, SAMPLE_FMT_S16, pcm_u16le, "PCM unsigned 16-bit little-endian");
00569 PCM_CODEC (CODEC_ID_PCM_U24BE, SAMPLE_FMT_S32, pcm_u24be, "PCM unsigned 24-bit big-endian");
00570 PCM_CODEC (CODEC_ID_PCM_U24LE, SAMPLE_FMT_S32, pcm_u24le, "PCM unsigned 24-bit little-endian");
00571 PCM_CODEC (CODEC_ID_PCM_U32BE, SAMPLE_FMT_S32, pcm_u32be, "PCM unsigned 32-bit big-endian");
00572 PCM_CODEC (CODEC_ID_PCM_U32LE, SAMPLE_FMT_S32, pcm_u32le, "PCM unsigned 32-bit little-endian");
00573 PCM_CODEC (CODEC_ID_PCM_ZORK, SAMPLE_FMT_S16, pcm_zork, "PCM Zork");