dtsdec.c
上传用户:wstnjxml
上传日期:2014-04-03
资源大小:7248k
文件大小:8k
源码类别:

Windows CE

开发平台:

C/C++

  1. /*
  2.  * dtsdec.c : free DTS Coherent Acoustics stream decoder.
  3.  * Copyright (C) 2004 Benjamin Zores <ben@geexbox.org>
  4.  *
  5.  * This file is part of libavcodec.
  6.  *
  7.  * This library is free software; you can redistribute it and/or modify
  8.  * it under the terms of the GNU General Public License as published by
  9.  * the Free Software Foundation; either version 2 of the License, or
  10.  * (at your option) any later version.
  11.  *  
  12.  * This library 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
  15.  * GNU General Public License for more details.
  16.  *  
  17.  * You should have received a copy of the GNU General Public License
  18.  * along with this program; if not, write to the Free Software
  19.  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20.  */
  21. #ifdef HAVE_AV_CONFIG_H
  22. #undef HAVE_AV_CONFIG_H
  23. #endif
  24. #include "avcodec.h"
  25. #include <dts.h>
  26. #include <stdlib.h>
  27. #include <string.h>
  28. #ifdef HAVE_MALLOC_H
  29. #include <malloc.h>
  30. #endif
  31. #define INBUF_SIZE 4096
  32. #define BUFFER_SIZE 4096
  33. #define HEADER_SIZE 14
  34. #ifdef LIBDTS_FIXED
  35. #define CONVERT_LEVEL (1 << 26)
  36. #define CONVERT_BIAS 0
  37. #else
  38. #define CONVERT_LEVEL 1
  39. #define CONVERT_BIAS 384
  40. #endif
  41. static inline
  42. int16_t convert (int32_t i)
  43. {
  44. #ifdef LIBDTS_FIXED
  45.     i >>= 15;
  46. #else
  47.     i -= 0x43c00000;
  48. #endif
  49.     return (i > 32767) ? 32767 : ((i < -32768) ? -32768 : i);
  50. }
  51. void
  52. convert2s16_2 (sample_t * _f, int16_t * s16)
  53. {
  54.   int i;
  55.   int32_t * f = (int32_t *) _f;
  56.   for (i = 0; i < 256; i++)
  57.     {
  58.       s16[2*i] = convert (f[i]);
  59.       s16[2*i+1] = convert (f[i+256]);
  60.     }
  61. }
  62. void
  63. convert2s16_4 (sample_t * _f, int16_t * s16)
  64. {
  65.   int i;
  66.   int32_t * f = (int32_t *) _f;
  67.   for (i = 0; i < 256; i++)
  68.     {
  69.       s16[4*i] = convert (f[i]);
  70.       s16[4*i+1] = convert (f[i+256]);
  71.       s16[4*i+2] = convert (f[i+512]);
  72.       s16[4*i+3] = convert (f[i+768]);
  73.     }
  74. }
  75. void
  76. convert2s16_5 (sample_t * _f, int16_t * s16)
  77. {
  78.   int i;
  79.   int32_t * f = (int32_t *) _f;
  80.   for (i = 0; i < 256; i++)
  81.     {
  82.       s16[5*i] = convert (f[i]);
  83.       s16[5*i+1] = convert (f[i+256]);
  84.       s16[5*i+2] = convert (f[i+512]);
  85.       s16[5*i+3] = convert (f[i+768]);
  86.       s16[5*i+4] = convert (f[i+1024]);
  87.     }
  88. }
  89. static void
  90. convert2s16_multi (sample_t * _f, int16_t * s16, int flags)
  91. {
  92.   int i;
  93.   int32_t * f = (int32_t *) _f;
  94.   switch (flags)
  95.     {
  96.     case DTS_MONO:
  97.       for (i = 0; i < 256; i++)
  98.         {
  99.           s16[5*i] = s16[5*i+1] = s16[5*i+2] = s16[5*i+3] = 0;
  100.           s16[5*i+4] = convert (f[i]);
  101.         }
  102.       break;
  103.     case DTS_CHANNEL:
  104.     case DTS_STEREO:
  105.     case DTS_DOLBY:
  106.       convert2s16_2 (_f, s16);
  107.       break;
  108.     case DTS_3F:
  109.       for (i = 0; i < 256; i++)
  110.         {
  111.           s16[5*i] = convert (f[i]);
  112.           s16[5*i+1] = convert (f[i+512]);
  113.           s16[5*i+2] = s16[5*i+3] = 0;
  114.           s16[5*i+4] = convert (f[i+256]);
  115.         }
  116.       break;
  117.     case DTS_2F2R:
  118.       convert2s16_4 (_f, s16);
  119.       break;
  120.     case DTS_3F2R:
  121.       convert2s16_5 (_f, s16);
  122.       break;
  123.     case DTS_MONO | DTS_LFE:
  124.       for (i = 0; i < 256; i++)
  125.         {
  126.           s16[6*i] = s16[6*i+1] = s16[6*i+2] = s16[6*i+3] = 0;
  127.           s16[6*i+4] = convert (f[i+256]);
  128.           s16[6*i+5] = convert (f[i]);
  129.         }
  130.       break;
  131.     case DTS_CHANNEL | DTS_LFE:
  132.     case DTS_STEREO | DTS_LFE:
  133.     case DTS_DOLBY | DTS_LFE:
  134.       for (i = 0; i < 256; i++)
  135.         {
  136.           s16[6*i] = convert (f[i+256]);
  137.           s16[6*i+1] = convert (f[i+512]);
  138.           s16[6*i+2] = s16[6*i+3] = s16[6*i+4] = 0;
  139.           s16[6*i+5] = convert (f[i]);
  140.         }
  141.       break;
  142.     case DTS_3F | DTS_LFE:
  143.       for (i = 0; i < 256; i++)
  144.         {
  145.           s16[6*i] = convert (f[i+256]);
  146.           s16[6*i+1] = convert (f[i+768]);
  147.           s16[6*i+2] = s16[6*i+3] = 0;
  148.           s16[6*i+4] = convert (f[i+512]);
  149.           s16[6*i+5] = convert (f[i]);
  150.         }
  151.       break;
  152.     case DTS_2F2R | DTS_LFE:
  153.       for (i = 0; i < 256; i++)
  154.         {
  155.           s16[6*i] = convert (f[i+256]);
  156.           s16[6*i+1] = convert (f[i+512]);
  157.           s16[6*i+2] = convert (f[i+768]);
  158.           s16[6*i+3] = convert (f[i+1024]);
  159.           s16[6*i+4] = 0;
  160.           s16[6*i+5] = convert (f[i]);
  161.         }
  162.       break;
  163.     case DTS_3F2R | DTS_LFE:
  164.       for (i = 0; i < 256; i++)
  165.         {
  166.           s16[6*i] = convert (f[i+256]);
  167.           s16[6*i+1] = convert (f[i+768]);
  168.           s16[6*i+2] = convert (f[i+1024]);
  169.           s16[6*i+3] = convert (f[i+1280]);
  170.           s16[6*i+4] = convert (f[i+512]);
  171.           s16[6*i+5] = convert (f[i]);
  172.         }
  173.       break;
  174.     }
  175. }
  176. static int
  177. channels_multi (int flags)
  178. {
  179.   if (flags & DTS_LFE)
  180.     return 6;
  181.   else if (flags & 1) /* center channel */
  182.     return 5;
  183.   else if ((flags & DTS_CHANNEL_MASK) == DTS_2F2R)
  184.     return 4;
  185.   else
  186.     return 2;
  187. }
  188. static int
  189. dts_decode_frame (AVCodecContext *avctx, void *data, int *data_size,
  190.                   uint8_t *buff, int buff_size)
  191. {
  192.   uint8_t * start = buff;
  193.   uint8_t * end = buff + buff_size;
  194.   static uint8_t buf[BUFFER_SIZE];
  195.   static uint8_t * bufptr = buf;
  196.   static uint8_t * bufpos = buf + HEADER_SIZE;
  197.   static int sample_rate;
  198.   static int frame_length;
  199.   static int flags;
  200.   int bit_rate;
  201.   int len;
  202.   dts_state_t *state = avctx->priv_data;
  203.   *data_size = 0;
  204.   while (1)
  205.     {
  206.       len = end - start;
  207.       if (!len)
  208.         break;
  209.       if (len > bufpos - bufptr)
  210.         len = bufpos - bufptr;
  211.       memcpy (bufptr, start, len);
  212.       bufptr += len;
  213.       start += len;
  214.       if (bufptr == bufpos)
  215.         {
  216.           if (bufpos == buf + HEADER_SIZE)
  217.             {
  218.               int length;
  219.               length = dts_syncinfo (state, buf, &flags, &sample_rate,
  220.                                      &bit_rate, &frame_length);
  221.               if (!length)
  222.                 {
  223.                   av_log (NULL, AV_LOG_INFO, "skipn");
  224.                   for (bufptr = buf; bufptr < buf + HEADER_SIZE-1; bufptr++)
  225.                     bufptr[0] = bufptr[1];
  226.                   continue;
  227.                 }
  228.               bufpos = buf + length;
  229.             }
  230.           else
  231.             {
  232.               level_t level;
  233.               sample_t bias;
  234.               int i;
  235.               flags = 2; /* ???????????? */
  236.               level = CONVERT_LEVEL;
  237.               bias = CONVERT_BIAS;
  238.               flags |= DTS_ADJUST_LEVEL;
  239.               if (dts_frame (state, buf, &flags, &level, bias))
  240.                 goto error;
  241.               avctx->sample_rate = sample_rate;
  242.               avctx->channels = channels_multi (flags);
  243.               avctx->bit_rate = bit_rate;
  244.               for (i = 0; i < dts_blocks_num (state); i++)
  245.                 {
  246.                   if (dts_block (state))
  247.                     goto error;
  248.                   {
  249.                     int chans;
  250.                     chans = channels_multi (flags);
  251.                     convert2s16_multi (dts_samples (state), data,
  252.                                        flags & (DTS_CHANNEL_MASK | DTS_LFE));
  253.                     data += 256 * sizeof (int16_t) * chans;
  254.                     *data_size += 256 * sizeof (int16_t) * chans;
  255.                   }
  256.                 }
  257.               bufptr = buf;
  258.               bufpos = buf + HEADER_SIZE;
  259.               continue;
  260.             error:
  261.               av_log (NULL, AV_LOG_ERROR, "errorn");
  262.               bufptr = buf;
  263.               bufpos = buf + HEADER_SIZE;
  264.             }
  265.         }
  266.     }
  267.   return buff_size;
  268. }
  269. static int
  270. dts_decode_init (AVCodecContext *avctx)
  271. {
  272.   avctx->priv_data = dts_init (0);
  273.   if (avctx->priv_data == NULL)
  274.     return 1;
  275.   return 0;
  276. }
  277. static int
  278. dts_decode_end (AVCodecContext *s)
  279. {
  280.   return 0;
  281. }
  282. AVCodec dts_decoder = {
  283.   "dts", 
  284.   CODEC_TYPE_AUDIO,
  285.   CODEC_ID_DTS,
  286.   sizeof (dts_state_t *),
  287.   dts_decode_init,
  288.   NULL,
  289.   dts_decode_end,
  290.   dts_decode_frame,
  291. };