You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

530 lines
17KB

  1. /*
  2. * PCM codecs
  3. * Copyright (c) 2001 Fabrice Bellard
  4. *
  5. * This file is part of Libav.
  6. *
  7. * Libav 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.
  11. *
  12. * Libav 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.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with Libav; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file
  23. * PCM codecs
  24. */
  25. #include "avcodec.h"
  26. #include "libavutil/common.h" /* for av_reverse */
  27. #include "bytestream.h"
  28. #include "pcm_tablegen.h"
  29. #define MAX_CHANNELS 64
  30. static av_cold int pcm_encode_init(AVCodecContext *avctx)
  31. {
  32. avctx->frame_size = 1;
  33. switch(avctx->codec->id) {
  34. case CODEC_ID_PCM_ALAW:
  35. pcm_alaw_tableinit();
  36. break;
  37. case CODEC_ID_PCM_MULAW:
  38. pcm_ulaw_tableinit();
  39. break;
  40. default:
  41. break;
  42. }
  43. avctx->bits_per_coded_sample = av_get_bits_per_sample(avctx->codec->id);
  44. avctx->block_align = avctx->channels * avctx->bits_per_coded_sample/8;
  45. avctx->coded_frame= avcodec_alloc_frame();
  46. avctx->coded_frame->key_frame= 1;
  47. return 0;
  48. }
  49. static av_cold int pcm_encode_close(AVCodecContext *avctx)
  50. {
  51. av_freep(&avctx->coded_frame);
  52. return 0;
  53. }
  54. /**
  55. * Write PCM samples macro
  56. * @param type Datatype of native machine format
  57. * @param endian bytestream_put_xxx() suffix
  58. * @param src Source pointer (variable name)
  59. * @param dst Destination pointer (variable name)
  60. * @param n Total number of samples (variable name)
  61. * @param shift Bitshift (bits)
  62. * @param offset Sample value offset
  63. */
  64. #define ENCODE(type, endian, src, dst, n, shift, offset) \
  65. samples_##type = (const type*) src; \
  66. for(;n>0;n--) { \
  67. register type v = (*samples_##type++ >> shift) + offset; \
  68. bytestream_put_##endian(&dst, v); \
  69. }
  70. static int pcm_encode_frame(AVCodecContext *avctx,
  71. unsigned char *frame, int buf_size, void *data)
  72. {
  73. int n, sample_size, v;
  74. const short *samples;
  75. unsigned char *dst;
  76. const uint8_t *srcu8;
  77. const int16_t *samples_int16_t;
  78. const int32_t *samples_int32_t;
  79. const int64_t *samples_int64_t;
  80. const uint16_t *samples_uint16_t;
  81. const uint32_t *samples_uint32_t;
  82. sample_size = av_get_bits_per_sample(avctx->codec->id)/8;
  83. n = buf_size / sample_size;
  84. samples = data;
  85. dst = frame;
  86. switch(avctx->codec->id) {
  87. case CODEC_ID_PCM_U32LE:
  88. ENCODE(uint32_t, le32, samples, dst, n, 0, 0x80000000)
  89. break;
  90. case CODEC_ID_PCM_U32BE:
  91. ENCODE(uint32_t, be32, samples, dst, n, 0, 0x80000000)
  92. break;
  93. case CODEC_ID_PCM_S24LE:
  94. ENCODE(int32_t, le24, samples, dst, n, 8, 0)
  95. break;
  96. case CODEC_ID_PCM_S24BE:
  97. ENCODE(int32_t, be24, samples, dst, n, 8, 0)
  98. break;
  99. case CODEC_ID_PCM_U24LE:
  100. ENCODE(uint32_t, le24, samples, dst, n, 8, 0x800000)
  101. break;
  102. case CODEC_ID_PCM_U24BE:
  103. ENCODE(uint32_t, be24, samples, dst, n, 8, 0x800000)
  104. break;
  105. case CODEC_ID_PCM_S24DAUD:
  106. for(;n>0;n--) {
  107. uint32_t tmp = av_reverse[(*samples >> 8) & 0xff] +
  108. (av_reverse[*samples & 0xff] << 8);
  109. tmp <<= 4; // sync flags would go here
  110. bytestream_put_be24(&dst, tmp);
  111. samples++;
  112. }
  113. break;
  114. case CODEC_ID_PCM_U16LE:
  115. ENCODE(uint16_t, le16, samples, dst, n, 0, 0x8000)
  116. break;
  117. case CODEC_ID_PCM_U16BE:
  118. ENCODE(uint16_t, be16, samples, dst, n, 0, 0x8000)
  119. break;
  120. case CODEC_ID_PCM_S8:
  121. srcu8= data;
  122. for(;n>0;n--) {
  123. v = *srcu8++;
  124. *dst++ = v - 128;
  125. }
  126. break;
  127. #if HAVE_BIGENDIAN
  128. case CODEC_ID_PCM_F64LE:
  129. ENCODE(int64_t, le64, samples, dst, n, 0, 0)
  130. break;
  131. case CODEC_ID_PCM_S32LE:
  132. case CODEC_ID_PCM_F32LE:
  133. ENCODE(int32_t, le32, samples, dst, n, 0, 0)
  134. break;
  135. case CODEC_ID_PCM_S16LE:
  136. ENCODE(int16_t, le16, samples, dst, n, 0, 0)
  137. break;
  138. case CODEC_ID_PCM_F64BE:
  139. case CODEC_ID_PCM_F32BE:
  140. case CODEC_ID_PCM_S32BE:
  141. case CODEC_ID_PCM_S16BE:
  142. #else
  143. case CODEC_ID_PCM_F64BE:
  144. ENCODE(int64_t, be64, samples, dst, n, 0, 0)
  145. break;
  146. case CODEC_ID_PCM_F32BE:
  147. case CODEC_ID_PCM_S32BE:
  148. ENCODE(int32_t, be32, samples, dst, n, 0, 0)
  149. break;
  150. case CODEC_ID_PCM_S16BE:
  151. ENCODE(int16_t, be16, samples, dst, n, 0, 0)
  152. break;
  153. case CODEC_ID_PCM_F64LE:
  154. case CODEC_ID_PCM_F32LE:
  155. case CODEC_ID_PCM_S32LE:
  156. case CODEC_ID_PCM_S16LE:
  157. #endif /* HAVE_BIGENDIAN */
  158. case CODEC_ID_PCM_U8:
  159. memcpy(dst, samples, n*sample_size);
  160. dst += n*sample_size;
  161. break;
  162. case CODEC_ID_PCM_ALAW:
  163. for(;n>0;n--) {
  164. v = *samples++;
  165. *dst++ = linear_to_alaw[(v + 32768) >> 2];
  166. }
  167. break;
  168. case CODEC_ID_PCM_MULAW:
  169. for(;n>0;n--) {
  170. v = *samples++;
  171. *dst++ = linear_to_ulaw[(v + 32768) >> 2];
  172. }
  173. break;
  174. default:
  175. return -1;
  176. }
  177. //avctx->frame_size = (dst - frame) / (sample_size * avctx->channels);
  178. return dst - frame;
  179. }
  180. typedef struct PCMDecode {
  181. AVFrame frame;
  182. short table[256];
  183. } PCMDecode;
  184. static av_cold int pcm_decode_init(AVCodecContext * avctx)
  185. {
  186. PCMDecode *s = avctx->priv_data;
  187. int i;
  188. if (avctx->channels <= 0 || avctx->channels > MAX_CHANNELS) {
  189. av_log(avctx, AV_LOG_ERROR, "PCM channels out of bounds\n");
  190. return AVERROR(EINVAL);
  191. }
  192. switch(avctx->codec->id) {
  193. case CODEC_ID_PCM_ALAW:
  194. for(i=0;i<256;i++)
  195. s->table[i] = alaw2linear(i);
  196. break;
  197. case CODEC_ID_PCM_MULAW:
  198. for(i=0;i<256;i++)
  199. s->table[i] = ulaw2linear(i);
  200. break;
  201. default:
  202. break;
  203. }
  204. avctx->sample_fmt = avctx->codec->sample_fmts[0];
  205. if (avctx->sample_fmt == AV_SAMPLE_FMT_S32)
  206. avctx->bits_per_raw_sample = av_get_bits_per_sample(avctx->codec->id);
  207. avcodec_get_frame_defaults(&s->frame);
  208. avctx->coded_frame = &s->frame;
  209. return 0;
  210. }
  211. /**
  212. * Read PCM samples macro
  213. * @param size Data size of native machine format
  214. * @param endian bytestream_get_xxx() endian suffix
  215. * @param src Source pointer (variable name)
  216. * @param dst Destination pointer (variable name)
  217. * @param n Total number of samples (variable name)
  218. * @param shift Bitshift (bits)
  219. * @param offset Sample value offset
  220. */
  221. #define DECODE(size, endian, src, dst, n, shift, offset) \
  222. for(;n>0;n--) { \
  223. uint##size##_t v = bytestream_get_##endian(&src); \
  224. AV_WN##size##A(dst, (v - offset) << shift); \
  225. dst += size / 8; \
  226. }
  227. static int pcm_decode_frame(AVCodecContext *avctx, void *data,
  228. int *got_frame_ptr, AVPacket *avpkt)
  229. {
  230. const uint8_t *src = avpkt->data;
  231. int buf_size = avpkt->size;
  232. PCMDecode *s = avctx->priv_data;
  233. int sample_size, c, n, ret, samples_per_block;
  234. uint8_t *samples;
  235. int32_t *dst_int32_t;
  236. sample_size = av_get_bits_per_sample(avctx->codec_id)/8;
  237. /* av_get_bits_per_sample returns 0 for CODEC_ID_PCM_DVD */
  238. samples_per_block = 1;
  239. if (CODEC_ID_PCM_DVD == avctx->codec_id) {
  240. if (avctx->bits_per_coded_sample != 20 &&
  241. avctx->bits_per_coded_sample != 24) {
  242. av_log(avctx, AV_LOG_ERROR, "PCM DVD unsupported sample depth\n");
  243. return AVERROR(EINVAL);
  244. }
  245. /* 2 samples are interleaved per block in PCM_DVD */
  246. samples_per_block = 2;
  247. sample_size = avctx->bits_per_coded_sample * 2 / 8;
  248. } else if (avctx->codec_id == CODEC_ID_PCM_LXF) {
  249. /* we process 40-bit blocks per channel for LXF */
  250. samples_per_block = 2;
  251. sample_size = 5;
  252. }
  253. if (sample_size == 0) {
  254. av_log(avctx, AV_LOG_ERROR, "Invalid sample_size\n");
  255. return AVERROR(EINVAL);
  256. }
  257. n = avctx->channels * sample_size;
  258. if(n && buf_size % n){
  259. if (buf_size < n) {
  260. av_log(avctx, AV_LOG_ERROR, "invalid PCM packet\n");
  261. return -1;
  262. }else
  263. buf_size -= buf_size % n;
  264. }
  265. n = buf_size/sample_size;
  266. /* get output buffer */
  267. s->frame.nb_samples = n * samples_per_block / avctx->channels;
  268. if ((ret = avctx->get_buffer(avctx, &s->frame)) < 0) {
  269. av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n");
  270. return ret;
  271. }
  272. samples = s->frame.data[0];
  273. switch(avctx->codec->id) {
  274. case CODEC_ID_PCM_U32LE:
  275. DECODE(32, le32, src, samples, n, 0, 0x80000000)
  276. break;
  277. case CODEC_ID_PCM_U32BE:
  278. DECODE(32, be32, src, samples, n, 0, 0x80000000)
  279. break;
  280. case CODEC_ID_PCM_S24LE:
  281. DECODE(32, le24, src, samples, n, 8, 0)
  282. break;
  283. case CODEC_ID_PCM_S24BE:
  284. DECODE(32, be24, src, samples, n, 8, 0)
  285. break;
  286. case CODEC_ID_PCM_U24LE:
  287. DECODE(32, le24, src, samples, n, 8, 0x800000)
  288. break;
  289. case CODEC_ID_PCM_U24BE:
  290. DECODE(32, be24, src, samples, n, 8, 0x800000)
  291. break;
  292. case CODEC_ID_PCM_S24DAUD:
  293. for(;n>0;n--) {
  294. uint32_t v = bytestream_get_be24(&src);
  295. v >>= 4; // sync flags are here
  296. AV_WN16A(samples, av_reverse[(v >> 8) & 0xff] +
  297. (av_reverse[v & 0xff] << 8));
  298. samples += 2;
  299. }
  300. break;
  301. case CODEC_ID_PCM_S16LE_PLANAR:
  302. {
  303. const uint8_t *src2[MAX_CHANNELS];
  304. n /= avctx->channels;
  305. for(c=0;c<avctx->channels;c++)
  306. src2[c] = &src[c*n*2];
  307. for(;n>0;n--)
  308. for(c=0;c<avctx->channels;c++) {
  309. AV_WN16A(samples, bytestream_get_le16(&src2[c]));
  310. samples += 2;
  311. }
  312. break;
  313. }
  314. case CODEC_ID_PCM_U16LE:
  315. DECODE(16, le16, src, samples, n, 0, 0x8000)
  316. break;
  317. case CODEC_ID_PCM_U16BE:
  318. DECODE(16, be16, src, samples, n, 0, 0x8000)
  319. break;
  320. case CODEC_ID_PCM_S8:
  321. for(;n>0;n--) {
  322. *samples++ = *src++ + 128;
  323. }
  324. break;
  325. #if HAVE_BIGENDIAN
  326. case CODEC_ID_PCM_F64LE:
  327. DECODE(64, le64, src, samples, n, 0, 0)
  328. break;
  329. case CODEC_ID_PCM_S32LE:
  330. case CODEC_ID_PCM_F32LE:
  331. DECODE(32, le32, src, samples, n, 0, 0)
  332. break;
  333. case CODEC_ID_PCM_S16LE:
  334. DECODE(16, le16, src, samples, n, 0, 0)
  335. break;
  336. case CODEC_ID_PCM_F64BE:
  337. case CODEC_ID_PCM_F32BE:
  338. case CODEC_ID_PCM_S32BE:
  339. case CODEC_ID_PCM_S16BE:
  340. #else
  341. case CODEC_ID_PCM_F64BE:
  342. DECODE(64, be64, src, samples, n, 0, 0)
  343. break;
  344. case CODEC_ID_PCM_F32BE:
  345. case CODEC_ID_PCM_S32BE:
  346. DECODE(32, be32, src, samples, n, 0, 0)
  347. break;
  348. case CODEC_ID_PCM_S16BE:
  349. DECODE(16, be16, src, samples, n, 0, 0)
  350. break;
  351. case CODEC_ID_PCM_F64LE:
  352. case CODEC_ID_PCM_F32LE:
  353. case CODEC_ID_PCM_S32LE:
  354. case CODEC_ID_PCM_S16LE:
  355. #endif /* HAVE_BIGENDIAN */
  356. case CODEC_ID_PCM_U8:
  357. memcpy(samples, src, n*sample_size);
  358. break;
  359. case CODEC_ID_PCM_ZORK:
  360. for (; n > 0; n--) {
  361. int v = *src++;
  362. if (v < 128)
  363. v = 128 - v;
  364. *samples++ = v;
  365. }
  366. break;
  367. case CODEC_ID_PCM_ALAW:
  368. case CODEC_ID_PCM_MULAW:
  369. for(;n>0;n--) {
  370. AV_WN16A(samples, s->table[*src++]);
  371. samples += 2;
  372. }
  373. break;
  374. case CODEC_ID_PCM_DVD:
  375. {
  376. const uint8_t *src8;
  377. dst_int32_t = (int32_t *)s->frame.data[0];
  378. n /= avctx->channels;
  379. switch (avctx->bits_per_coded_sample) {
  380. case 20:
  381. while (n--) {
  382. c = avctx->channels;
  383. src8 = src + 4*c;
  384. while (c--) {
  385. *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8 &0xf0) << 8);
  386. *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++ &0x0f) << 12);
  387. }
  388. src = src8;
  389. }
  390. break;
  391. case 24:
  392. while (n--) {
  393. c = avctx->channels;
  394. src8 = src + 4*c;
  395. while (c--) {
  396. *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
  397. *dst_int32_t++ = (bytestream_get_be16(&src) << 16) + ((*src8++) << 8);
  398. }
  399. src = src8;
  400. }
  401. break;
  402. }
  403. break;
  404. }
  405. case CODEC_ID_PCM_LXF:
  406. {
  407. int i;
  408. const uint8_t *src8;
  409. dst_int32_t = (int32_t *)s->frame.data[0];
  410. n /= avctx->channels;
  411. //unpack and de-planerize
  412. for (i = 0; i < n; i++) {
  413. for (c = 0, src8 = src + i*5; c < avctx->channels; c++, src8 += n*5) {
  414. //extract low 20 bits and expand to 32 bits
  415. *dst_int32_t++ = (src8[2] << 28) | (src8[1] << 20) | (src8[0] << 12) |
  416. ((src8[2] & 0xF) << 8) | src8[1];
  417. }
  418. for (c = 0, src8 = src + i*5; c < avctx->channels; c++, src8 += n*5) {
  419. //extract high 20 bits and expand to 32 bits
  420. *dst_int32_t++ = (src8[4] << 24) | (src8[3] << 16) |
  421. ((src8[2] & 0xF0) << 8) | (src8[4] << 4) | (src8[3] >> 4);
  422. }
  423. }
  424. break;
  425. }
  426. default:
  427. return -1;
  428. }
  429. *got_frame_ptr = 1;
  430. *(AVFrame *)data = s->frame;
  431. return buf_size;
  432. }
  433. #if CONFIG_ENCODERS
  434. #define PCM_ENCODER(id_,sample_fmt_,name_,long_name_) \
  435. AVCodec ff_ ## name_ ## _encoder = { \
  436. .name = #name_, \
  437. .type = AVMEDIA_TYPE_AUDIO, \
  438. .id = id_, \
  439. .init = pcm_encode_init, \
  440. .encode = pcm_encode_frame, \
  441. .close = pcm_encode_close, \
  442. .sample_fmts = (const enum AVSampleFormat[]){sample_fmt_,AV_SAMPLE_FMT_NONE}, \
  443. .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
  444. }
  445. #else
  446. #define PCM_ENCODER(id,sample_fmt_,name,long_name_)
  447. #endif
  448. #if CONFIG_DECODERS
  449. #define PCM_DECODER(id_,sample_fmt_,name_,long_name_) \
  450. AVCodec ff_ ## name_ ## _decoder = { \
  451. .name = #name_, \
  452. .type = AVMEDIA_TYPE_AUDIO, \
  453. .id = id_, \
  454. .priv_data_size = sizeof(PCMDecode), \
  455. .init = pcm_decode_init, \
  456. .decode = pcm_decode_frame, \
  457. .capabilities = CODEC_CAP_DR1, \
  458. .sample_fmts = (const enum AVSampleFormat[]){sample_fmt_,AV_SAMPLE_FMT_NONE}, \
  459. .long_name = NULL_IF_CONFIG_SMALL(long_name_), \
  460. }
  461. #else
  462. #define PCM_DECODER(id,sample_fmt_,name,long_name_)
  463. #endif
  464. #define PCM_CODEC(id, sample_fmt_, name, long_name_) \
  465. PCM_ENCODER(id,sample_fmt_,name,long_name_); PCM_DECODER(id,sample_fmt_,name,long_name_)
  466. /* Note: Do not forget to add new entries to the Makefile as well. */
  467. PCM_CODEC (CODEC_ID_PCM_ALAW, AV_SAMPLE_FMT_S16, pcm_alaw, "PCM A-law");
  468. PCM_DECODER(CODEC_ID_PCM_DVD, AV_SAMPLE_FMT_S32, pcm_dvd, "PCM signed 20|24-bit big-endian");
  469. PCM_CODEC (CODEC_ID_PCM_F32BE, AV_SAMPLE_FMT_FLT, pcm_f32be, "PCM 32-bit floating point big-endian");
  470. PCM_CODEC (CODEC_ID_PCM_F32LE, AV_SAMPLE_FMT_FLT, pcm_f32le, "PCM 32-bit floating point little-endian");
  471. PCM_CODEC (CODEC_ID_PCM_F64BE, AV_SAMPLE_FMT_DBL, pcm_f64be, "PCM 64-bit floating point big-endian");
  472. PCM_CODEC (CODEC_ID_PCM_F64LE, AV_SAMPLE_FMT_DBL, pcm_f64le, "PCM 64-bit floating point little-endian");
  473. PCM_DECODER(CODEC_ID_PCM_LXF, AV_SAMPLE_FMT_S32, pcm_lxf, "PCM signed 20-bit little-endian planar");
  474. PCM_CODEC (CODEC_ID_PCM_MULAW, AV_SAMPLE_FMT_S16, pcm_mulaw, "PCM mu-law");
  475. PCM_CODEC (CODEC_ID_PCM_S8, AV_SAMPLE_FMT_U8, pcm_s8, "PCM signed 8-bit");
  476. PCM_CODEC (CODEC_ID_PCM_S16BE, AV_SAMPLE_FMT_S16, pcm_s16be, "PCM signed 16-bit big-endian");
  477. PCM_CODEC (CODEC_ID_PCM_S16LE, AV_SAMPLE_FMT_S16, pcm_s16le, "PCM signed 16-bit little-endian");
  478. PCM_DECODER(CODEC_ID_PCM_S16LE_PLANAR, AV_SAMPLE_FMT_S16, pcm_s16le_planar, "PCM 16-bit little-endian planar");
  479. PCM_CODEC (CODEC_ID_PCM_S24BE, AV_SAMPLE_FMT_S32, pcm_s24be, "PCM signed 24-bit big-endian");
  480. PCM_CODEC (CODEC_ID_PCM_S24DAUD, AV_SAMPLE_FMT_S16, pcm_s24daud, "PCM D-Cinema audio signed 24-bit");
  481. PCM_CODEC (CODEC_ID_PCM_S24LE, AV_SAMPLE_FMT_S32, pcm_s24le, "PCM signed 24-bit little-endian");
  482. PCM_CODEC (CODEC_ID_PCM_S32BE, AV_SAMPLE_FMT_S32, pcm_s32be, "PCM signed 32-bit big-endian");
  483. PCM_CODEC (CODEC_ID_PCM_S32LE, AV_SAMPLE_FMT_S32, pcm_s32le, "PCM signed 32-bit little-endian");
  484. PCM_CODEC (CODEC_ID_PCM_U8, AV_SAMPLE_FMT_U8, pcm_u8, "PCM unsigned 8-bit");
  485. PCM_CODEC (CODEC_ID_PCM_U16BE, AV_SAMPLE_FMT_S16, pcm_u16be, "PCM unsigned 16-bit big-endian");
  486. PCM_CODEC (CODEC_ID_PCM_U16LE, AV_SAMPLE_FMT_S16, pcm_u16le, "PCM unsigned 16-bit little-endian");
  487. PCM_CODEC (CODEC_ID_PCM_U24BE, AV_SAMPLE_FMT_S32, pcm_u24be, "PCM unsigned 24-bit big-endian");
  488. PCM_CODEC (CODEC_ID_PCM_U24LE, AV_SAMPLE_FMT_S32, pcm_u24le, "PCM unsigned 24-bit little-endian");
  489. PCM_CODEC (CODEC_ID_PCM_U32BE, AV_SAMPLE_FMT_S32, pcm_u32be, "PCM unsigned 32-bit big-endian");
  490. PCM_CODEC (CODEC_ID_PCM_U32LE, AV_SAMPLE_FMT_S32, pcm_u32le, "PCM unsigned 32-bit little-endian");
  491. PCM_DECODER(CODEC_ID_PCM_ZORK, AV_SAMPLE_FMT_U8, pcm_zork, "PCM Zork");