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 "libavutil/common.h"
00029 #include "bytestream.h"
00030 #include "pcm_tablegen.h"
00031
00032 #define MAX_CHANNELS 64
00033
00034 static av_cold int pcm_encode_init(AVCodecContext *avctx)
00035 {
00036 avctx->frame_size = 1;
00037 switch(avctx->codec->id) {
00038 case CODEC_ID_PCM_ALAW:
00039 pcm_alaw_tableinit();
00040 break;
00041 case CODEC_ID_PCM_MULAW:
00042 pcm_ulaw_tableinit();
00043 break;
00044 default:
00045 break;
00046 }
00047
00048 avctx->bits_per_coded_sample = av_get_bits_per_sample(avctx->codec->id);
00049 avctx->block_align = avctx->channels * avctx->bits_per_coded_sample/8;
00050 avctx->coded_frame= avcodec_alloc_frame();
00051 avctx->coded_frame->key_frame= 1;
00052
00053 return 0;
00054 }
00055
00056 static av_cold int pcm_encode_close(AVCodecContext *avctx)
00057 {
00058 av_freep(&avctx->coded_frame);
00059
00060 return 0;
00061 }
00062
00073 #define ENCODE(type, endian, src, dst, n, shift, offset) \
00074 samples_##type = (const type*) src; \
00075 for(;n>0;n--) { \
00076 register type v = (*samples_##type++ >> shift) + offset; \
00077 bytestream_put_##endian(&dst, v); \
00078 }
00079
00080 static int pcm_encode_frame(AVCodecContext *avctx,
00081 unsigned char *frame, int buf_size, void *data)
00082 {
00083 int n, sample_size, v;
00084 const short *samples;
00085 unsigned char *dst;
00086 const uint8_t *srcu8;
00087 const int16_t *samples_int16_t;
00088 const int32_t *samples_int32_t;
00089 const int64_t *samples_int64_t;
00090 const uint16_t *samples_uint16_t;
00091 const uint32_t *samples_uint32_t;
00092
00093 sample_size = av_get_bits_per_sample(avctx->codec->id)/8;
00094 n = buf_size / sample_size;
00095 samples = data;
00096 dst = frame;
00097
00098 switch(avctx->codec->id) {
00099 case CODEC_ID_PCM_U32LE:
00100 ENCODE(uint32_t, le32, samples, dst, n, 0, 0x80000000)
00101 break;
00102 case CODEC_ID_PCM_U32BE:
00103 ENCODE(uint32_t, be32, samples, dst, n, 0, 0x80000000)
00104 break;
00105 case CODEC_ID_PCM_S24LE:
00106 ENCODE(int32_t, le24, samples, dst, n, 8, 0)
00107 break;
00108 case CODEC_ID_PCM_S24BE:
00109 ENCODE(int32_t, be24, samples, dst, n, 8, 0)
00110 break;
00111 case CODEC_ID_PCM_U24LE:
00112 ENCODE(uint32_t, le24, samples, dst, n, 8, 0x800000)
00113 break;
00114 case CODEC_ID_PCM_U24BE:
00115 ENCODE(uint32_t, be24, samples, dst, n, 8, 0x800000)
00116 break;
00117 case CODEC_ID_PCM_S24DAUD:
00118 for(;n>0;n--) {
00119 uint32_t tmp = av_reverse[(*samples >> 8) & 0xff] +
00120 (av_reverse[*samples & 0xff] << 8);
00121 tmp <<= 4;
00122 bytestream_put_be24(&dst, tmp);
00123 samples++;
00124 }
00125 break;
00126 case CODEC_ID_PCM_U16LE:
00127 ENCODE(uint16_t, le16, samples, dst, n, 0, 0x8000)
00128 break;
00129 case CODEC_ID_PCM_U16BE:
00130 ENCODE(uint16_t, be16, samples, dst, n, 0, 0x8000)
00131 break;
00132 case CODEC_ID_PCM_S8:
00133 srcu8= data;
00134 for(;n>0;n--) {
00135 v = *srcu8++;
00136 *dst++ = v - 128;
00137 }
00138 break;
00139 #if HAVE_BIGENDIAN
00140 case CODEC_ID_PCM_F64LE:
00141 ENCODE(int64_t, le64, samples, dst, n, 0, 0)
00142 break;
00143 case CODEC_ID_PCM_S32LE:
00144 case CODEC_ID_PCM_F32LE:
00145 ENCODE(int32_t, le32, samples, dst, n, 0, 0)
00146 break;
00147 case CODEC_ID_PCM_S16LE:
00148 ENCODE(int16_t, le16, samples, dst, n, 0, 0)
00149 break;
00150 case CODEC_ID_PCM_F64BE:
00151 case CODEC_ID_PCM_F32BE:
00152 case CODEC_ID_PCM_S32BE:
00153 case CODEC_ID_PCM_S16BE:
00154 #else
00155 case CODEC_ID_PCM_F64BE:
00156 ENCODE(int64_t, be64, samples, dst, n, 0, 0)
00157 break;
00158 case CODEC_ID_PCM_F32BE:
00159 case CODEC_ID_PCM_S32BE:
00160 ENCODE(int32_t, be32, samples, dst, n, 0, 0)
00161 break;
00162 case CODEC_ID_PCM_S16BE:
00163 ENCODE(int16_t, be16, samples, dst, n, 0, 0)
00164 break;
00165 case CODEC_ID_PCM_F64LE:
00166 case CODEC_ID_PCM_F32LE:
00167 case CODEC_ID_PCM_S32LE:
00168 case CODEC_ID_PCM_S16LE:
00169 #endif
00170 case CODEC_ID_PCM_U8:
00171 memcpy(dst, samples, n*sample_size);
00172 dst += n*sample_size;
00173 break;
00174 case CODEC_ID_PCM_ALAW:
00175 for(;n>0;n--) {
00176 v = *samples++;
00177 *dst++ = linear_to_alaw[(v + 32768) >> 2];
00178 }
00179 break;
00180 case CODEC_ID_PCM_MULAW:
00181 for(;n>0;n--) {
00182 v = *samples++;
00183 *dst++ = linear_to_ulaw[(v + 32768) >> 2];
00184 }
00185 break;
00186 default:
00187 return -1;
00188 }
00189
00190
00191 return dst - frame;
00192 }
00193
00194 typedef struct PCMDecode {
00195 AVFrame frame;
00196 short table[256];
00197 } PCMDecode;
00198
00199 static av_cold int pcm_decode_init(AVCodecContext * avctx)
00200 {
00201 PCMDecode *s = avctx->priv_data;
00202 int i;
00203
00204 if (avctx->channels <= 0 || avctx->channels > MAX_CHANNELS) {
00205 av_log(avctx, AV_LOG_ERROR, "PCM channels out of bounds\n");
00206 return AVERROR(EINVAL);
00207 }
00208
00209 switch(avctx->codec->id) {
00210 case CODEC_ID_PCM_ALAW:
00211 for(i=0;i<256;i++)
00212 s->table[i] = alaw2linear(i);
00213 break;
00214 case CODEC_ID_PCM_MULAW:
00215 for(i=0;i<256;i++)
00216 s->table[i] = ulaw2linear(i);
00217 break;
00218 default:
00219 break;
00220 }
00221
00222 avctx->sample_fmt = avctx->codec->sample_fmts[0];
00223
00224 if (avctx->sample_fmt == AV_SAMPLE_FMT_S32)
00225 avctx->bits_per_raw_sample = av_get_bits_per_sample(avctx->codec->id);
00226
00227 avcodec_get_frame_defaults(&s->frame);
00228 avctx->coded_frame = &s->frame;
00229
00230 return 0;
00231 }
00232
00243 #define DECODE(size, endian, src, dst, n, shift, offset) \
00244 for(;n>0;n--) { \
00245 uint##size##_t v = bytestream_get_##endian(&src); \
00246 AV_WN##size##A(dst, (v - offset) << shift); \
00247 dst += size / 8; \
00248 }
00249
00250 static int pcm_decode_frame(AVCodecContext *avctx, void *data,
00251 int *got_frame_ptr, AVPacket *avpkt)
00252 {
00253 const uint8_t *src = avpkt->data;
00254 int buf_size = avpkt->size;
00255 PCMDecode *s = avctx->priv_data;
00256 int sample_size, c, n, ret, samples_per_block;
00257 uint8_t *samples;
00258 int32_t *dst_int32_t;
00259
00260 sample_size = av_get_bits_per_sample(avctx->codec_id)/8;
00261
00262
00263 samples_per_block = 1;
00264 if (CODEC_ID_PCM_DVD == avctx->codec_id) {
00265 if (avctx->bits_per_coded_sample != 20 &&
00266 avctx->bits_per_coded_sample != 24) {
00267 av_log(avctx, AV_LOG_ERROR,
00268 "PCM DVD unsupported sample depth %i\n",
00269 avctx->bits_per_coded_sample);
00270 return AVERROR(EINVAL);
00271 }
00272
00273 samples_per_block = 2;
00274 sample_size = avctx->bits_per_coded_sample * 2 / 8;
00275 } else if (avctx->codec_id == CODEC_ID_PCM_LXF) {
00276
00277 samples_per_block = 2;
00278 sample_size = 5;
00279 }
00280
00281 if (sample_size == 0) {
00282 av_log(avctx, AV_LOG_ERROR, "Invalid sample_size\n");
00283 return AVERROR(EINVAL);
00284 }
00285
00286 n = avctx->channels * sample_size;
00287
00288 if(n && buf_size % n){
00289 if (buf_size < n) {
00290 av_log(avctx, AV_LOG_ERROR, "invalid PCM packet\n");
00291 return -1;
00292 }else
00293 buf_size -= buf_size % n;
00294 }
00295
00296 n = buf_size/sample_size;
00297
00298
00299 s->frame.nb_samples = n * samples_per_block / avctx->channels;
00300 if ((ret = avctx->get_buffer(avctx, &s->frame)) < 0) {
00301 av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
00302 return ret;
00303 }
00304 samples = s->frame.data[0];
00305
00306 switch(avctx->codec->id) {
00307 case CODEC_ID_PCM_U32LE:
00308 DECODE(32, le32, src, samples, n, 0, 0x80000000)
00309 break;
00310 case CODEC_ID_PCM_U32BE:
00311 DECODE(32, be32, src, samples, n, 0, 0x80000000)
00312 break;
00313 case CODEC_ID_PCM_S24LE:
00314 DECODE(32, le24, src, samples, n, 8, 0)
00315 break;
00316 case CODEC_ID_PCM_S24BE:
00317 DECODE(32, be24, src, samples, n, 8, 0)
00318 break;
00319 case CODEC_ID_PCM_U24LE:
00320 DECODE(32, le24, src, samples, n, 8, 0x800000)
00321 break;
00322 case CODEC_ID_PCM_U24BE:
00323 DECODE(32, be24, src, samples, n, 8, 0x800000)
00324 break;
00325 case CODEC_ID_PCM_S24DAUD:
00326 for(;n>0;n--) {
00327 uint32_t v = bytestream_get_be24(&src);
00328 v >>= 4;
00329 AV_WN16A(samples, av_reverse[(v >> 8) & 0xff] +
00330 (av_reverse[v & 0xff] << 8));
00331 samples += 2;
00332 }
00333 break;
00334 case CODEC_ID_PCM_S16LE_PLANAR:
00335 {
00336 const uint8_t *src2[MAX_CHANNELS];
00337 n /= avctx->channels;
00338 for(c=0;c<avctx->channels;c++)
00339 src2[c] = &src[c*n*2];
00340 for(;n>0;n--)
00341 for(c=0;c<avctx->channels;c++) {
00342 AV_WN16A(samples, bytestream_get_le16(&src2[c]));
00343 samples += 2;
00344 }
00345 break;
00346 }
00347 case CODEC_ID_PCM_U16LE:
00348 DECODE(16, le16, src, samples, n, 0, 0x8000)
00349 break;
00350 case CODEC_ID_PCM_U16BE:
00351 DECODE(16, be16, src, samples, n, 0, 0x8000)
00352 break;
00353 case CODEC_ID_PCM_S8:
00354 for(;n>0;n--) {
00355 *samples++ = *src++ + 128;
00356 }
00357 break;
00358 #if HAVE_BIGENDIAN
00359 case CODEC_ID_PCM_F64LE:
00360 DECODE(64, le64, src, samples, n, 0, 0)
00361 break;
00362 case CODEC_ID_PCM_S32LE:
00363 case CODEC_ID_PCM_F32LE:
00364 DECODE(32, le32, src, samples, n, 0, 0)
00365 break;
00366 case CODEC_ID_PCM_S16LE:
00367 DECODE(16, le16, src, samples, n, 0, 0)
00368 break;
00369 case CODEC_ID_PCM_F64BE:
00370 case CODEC_ID_PCM_F32BE:
00371 case CODEC_ID_PCM_S32BE:
00372 case CODEC_ID_PCM_S16BE:
00373 #else
00374 case CODEC_ID_PCM_F64BE:
00375 DECODE(64, be64, src, samples, n, 0, 0)
00376 break;
00377 case CODEC_ID_PCM_F32BE:
00378 case CODEC_ID_PCM_S32BE:
00379 DECODE(32, be32, src, samples, n, 0, 0)
00380 break;
00381 case CODEC_ID_PCM_S16BE:
00382 DECODE(16, be16, src, samples, n, 0, 0)
00383 break;
00384 case CODEC_ID_PCM_F64LE:
00385 case CODEC_ID_PCM_F32LE:
00386 case CODEC_ID_PCM_S32LE:
00387 case CODEC_ID_PCM_S16LE:
00388 #endif
00389 case CODEC_ID_PCM_U8:
00390 memcpy(samples, src, n*sample_size);
00391 break;
00392 case CODEC_ID_PCM_ZORK:
00393 for (; n > 0; n--) {
00394 int v = *src++;
00395 if (v < 128)
00396 v = 128 - v;
00397 *samples++ = v;
00398 }
00399 break;
00400 case CODEC_ID_PCM_ALAW:
00401 case CODEC_ID_PCM_MULAW:
00402 for(;n>0;n--) {
00403 AV_WN16A(samples, s->table[*src++]);
00404 samples += 2;
00405 }
00406 break;
00407 case CODEC_ID_PCM_DVD:
00408 {
00409 const uint8_t *src8;
00410 dst_int32_t = (int32_t *)s->frame.data[0];
00411 n /= avctx->channels;
00412 switch (avctx->bits_per_coded_sample) {
00413 case 20:
00414 while (n--) {
00415 c = avctx->channels;
00416 src8 = src + 4*c;
00417 while (c--) {
00418 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8 &0xf0) << 8);
00419 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++ &0x0f) << 12);
00420 }
00421 src = src8;
00422 }
00423 break;
00424 case 24:
00425 while (n--) {
00426 c = avctx->channels;
00427 src8 = src + 4*c;
00428 while (c--) {
00429 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
00430 *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
00431 }
00432 src = src8;
00433 }
00434 break;
00435 }
00436 break;
00437 }
00438 case CODEC_ID_PCM_LXF:
00439 {
00440 int i;
00441 const uint8_t *src8;
00442 dst_int32_t = (int32_t *)s->frame.data[0];
00443 n /= avctx->channels;
00444
00445 for (i = 0; i < n; i++) {
00446 for (c = 0, src8 = src + i*5; c < avctx->channels; c++, src8 += n*5) {
00447
00448 *dst_int32_t++ = (src8[2] << 28) | (src8[1] << 20) | (src8[0] << 12) |
00449 ((src8[2] & 0xF) << 8) | src8[1];
00450 }
00451
00452 for (c = 0, src8 = src + i*5; c < avctx->channels; c++, src8 += n*5) {
00453
00454 *dst_int32_t++ = (src8[4] << 24) | (src8[3] << 16) |
00455 ((src8[2] & 0xF0) << 8) | (src8[4] << 4) | (src8[3] >> 4);
00456 }
00457 }
00458 break;
00459 }
00460 default:
00461 return -1;
00462 }
00463
00464 *got_frame_ptr = 1;
00465 *(AVFrame *)data = s->frame;
00466
00467 return buf_size;
00468 }
00469
00470 #if CONFIG_ENCODERS
00471 #define PCM_ENCODER(id_,sample_fmt_,name_,long_name_) \
00472 AVCodec ff_ ## name_ ## _encoder = { \
00473 .name = #name_, \
00474 .type = AVMEDIA_TYPE_AUDIO, \
00475 .id = id_, \
00476 .init = pcm_encode_init, \
00477 .encode = pcm_encode_frame, \
00478 .close = pcm_encode_close, \
00479 .sample_fmts = (const enum AVSampleFormat[]){sample_fmt_,AV_SAMPLE_FMT_NONE}, \
00480 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
00481 }
00482 #else
00483 #define PCM_ENCODER(id,sample_fmt_,name,long_name_)
00484 #endif
00485
00486 #if CONFIG_DECODERS
00487 #define PCM_DECODER(id_,sample_fmt_,name_,long_name_) \
00488 AVCodec ff_ ## name_ ## _decoder = { \
00489 .name = #name_, \
00490 .type = AVMEDIA_TYPE_AUDIO, \
00491 .id = id_, \
00492 .priv_data_size = sizeof(PCMDecode), \
00493 .init = pcm_decode_init, \
00494 .decode = pcm_decode_frame, \
00495 .capabilities = CODEC_CAP_DR1, \
00496 .sample_fmts = (const enum AVSampleFormat[]){sample_fmt_,AV_SAMPLE_FMT_NONE}, \
00497 .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
00498 }
00499 #else
00500 #define PCM_DECODER(id,sample_fmt_,name,long_name_)
00501 #endif
00502
00503 #define PCM_CODEC(id, sample_fmt_, name, long_name_) \
00504 PCM_ENCODER(id,sample_fmt_,name,long_name_); PCM_DECODER(id,sample_fmt_,name,long_name_)
00505
00506
00507 PCM_CODEC (CODEC_ID_PCM_ALAW, AV_SAMPLE_FMT_S16, pcm_alaw, "PCM A-law");
00508 PCM_DECODER(CODEC_ID_PCM_DVD, AV_SAMPLE_FMT_S32, pcm_dvd, "PCM signed 20|24-bit big-endian");
00509 PCM_CODEC (CODEC_ID_PCM_F32BE, AV_SAMPLE_FMT_FLT, pcm_f32be, "PCM 32-bit floating point big-endian");
00510 PCM_CODEC (CODEC_ID_PCM_F32LE, AV_SAMPLE_FMT_FLT, pcm_f32le, "PCM 32-bit floating point little-endian");
00511 PCM_CODEC (CODEC_ID_PCM_F64BE, AV_SAMPLE_FMT_DBL, pcm_f64be, "PCM 64-bit floating point big-endian");
00512 PCM_CODEC (CODEC_ID_PCM_F64LE, AV_SAMPLE_FMT_DBL, pcm_f64le, "PCM 64-bit floating point little-endian");
00513 PCM_DECODER(CODEC_ID_PCM_LXF, AV_SAMPLE_FMT_S32, pcm_lxf, "PCM signed 20-bit little-endian planar");
00514 PCM_CODEC (CODEC_ID_PCM_MULAW, AV_SAMPLE_FMT_S16, pcm_mulaw, "PCM mu-law");
00515 PCM_CODEC (CODEC_ID_PCM_S8, AV_SAMPLE_FMT_U8, pcm_s8, "PCM signed 8-bit");
00516 PCM_CODEC (CODEC_ID_PCM_S16BE, AV_SAMPLE_FMT_S16, pcm_s16be, "PCM signed 16-bit big-endian");
00517 PCM_CODEC (CODEC_ID_PCM_S16LE, AV_SAMPLE_FMT_S16, pcm_s16le, "PCM signed 16-bit little-endian");
00518 PCM_DECODER(CODEC_ID_PCM_S16LE_PLANAR, AV_SAMPLE_FMT_S16, pcm_s16le_planar, "PCM 16-bit little-endian planar");
00519 PCM_CODEC (CODEC_ID_PCM_S24BE, AV_SAMPLE_FMT_S32, pcm_s24be, "PCM signed 24-bit big-endian");
00520 PCM_CODEC (CODEC_ID_PCM_S24DAUD, AV_SAMPLE_FMT_S16, pcm_s24daud, "PCM D-Cinema audio signed 24-bit");
00521 PCM_CODEC (CODEC_ID_PCM_S24LE, AV_SAMPLE_FMT_S32, pcm_s24le, "PCM signed 24-bit little-endian");
00522 PCM_CODEC (CODEC_ID_PCM_S32BE, AV_SAMPLE_FMT_S32, pcm_s32be, "PCM signed 32-bit big-endian");
00523 PCM_CODEC (CODEC_ID_PCM_S32LE, AV_SAMPLE_FMT_S32, pcm_s32le, "PCM signed 32-bit little-endian");
00524 PCM_CODEC (CODEC_ID_PCM_U8, AV_SAMPLE_FMT_U8, pcm_u8, "PCM unsigned 8-bit");
00525 PCM_CODEC (CODEC_ID_PCM_U16BE, AV_SAMPLE_FMT_S16, pcm_u16be, "PCM unsigned 16-bit big-endian");
00526 PCM_CODEC (CODEC_ID_PCM_U16LE, AV_SAMPLE_FMT_S16, pcm_u16le, "PCM unsigned 16-bit little-endian");
00527 PCM_CODEC (CODEC_ID_PCM_U24BE, AV_SAMPLE_FMT_S32, pcm_u24be, "PCM unsigned 24-bit big-endian");
00528 PCM_CODEC (CODEC_ID_PCM_U24LE, AV_SAMPLE_FMT_S32, pcm_u24le, "PCM unsigned 24-bit little-endian");
00529 PCM_CODEC (CODEC_ID_PCM_U32BE, AV_SAMPLE_FMT_S32, pcm_u32be, "PCM unsigned 32-bit big-endian");
00530 PCM_CODEC (CODEC_ID_PCM_U32LE, AV_SAMPLE_FMT_S32, pcm_u32le, "PCM unsigned 32-bit little-endian");
00531 PCM_DECODER(CODEC_ID_PCM_ZORK, AV_SAMPLE_FMT_U8, pcm_zork, "PCM Zork");