bits.h
上传用户:xjjlds
上传日期:2015-12-05
资源大小:22823k
文件大小:10k
源码类别:

多媒体编程

开发平台:

Visual C++

  1. /*
  2. ** FAAD2 - Freeware Advanced Audio (AAC) Decoder including SBR decoding
  3. ** Copyright (C) 2003-2005 M. Bakker, Ahead Software AG, http://www.nero.com
  4. **  
  5. ** This program is free software; you can redistribute it and/or modify
  6. ** it under the terms of the GNU General Public License as published by
  7. ** the Free Software Foundation; either version 2 of the License, or
  8. ** (at your option) any later version.
  9. ** 
  10. ** This program is distributed in the hope that it will be useful,
  11. ** but WITHOUT ANY WARRANTY; without even the implied warranty of
  12. ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13. ** GNU General Public License for more details.
  14. ** 
  15. ** You should have received a copy of the GNU General Public License
  16. ** along with this program; if not, write to the Free Software 
  17. ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18. **
  19. ** Any non-GPL usage of this software or parts of this software is strictly
  20. ** forbidden.
  21. **
  22. ** Software using this code must display the following message visibly in the
  23. ** software:
  24. ** "FAAD2 AAC/HE-AAC/HE-AACv2/DRM decoder (c) Ahead Software, www.nero.com"
  25. ** in, for example, the about-box or help/startup screen.
  26. **
  27. ** Commercial non-GPL licensing of this software is possible.
  28. ** For more info contact Ahead Software through Mpeg4AAClicense@nero.com.
  29. **
  30. ** $Id: bits.h,v 1.2 2005/11/01 21:41:43 gabest Exp $
  31. **/
  32. #ifndef __BITS_H__
  33. #define __BITS_H__
  34. #ifdef __cplusplus
  35. extern "C" {
  36. #endif
  37. #include "analysis.h"
  38. #ifdef ANALYSIS
  39. #include <stdio.h>
  40. #endif
  41. #define BYTE_NUMBIT 8
  42. #define bit2byte(a) ((a+7)/BYTE_NUMBIT)
  43. typedef struct _bitfile
  44. {
  45.     /* bit input */
  46.     uint32_t bufa;
  47.     uint32_t bufb;
  48.     uint32_t bits_left;
  49.     uint32_t buffer_size; /* size of the buffer in bytes */
  50.     uint32_t bytes_used;
  51.     uint8_t no_more_reading;
  52.     uint8_t error;
  53.     uint32_t *tail;
  54.     uint32_t *start;
  55.     void *buffer;
  56. } bitfile;
  57. #if defined (_WIN32) && !defined(_WIN32_WCE) && !defined(__MINGW32__)
  58. #define BSWAP(a) __asm mov eax,a __asm bswap eax __asm mov a, eax
  59. #elif defined(LINUX) || defined(DJGPP) || defined(__MINGW32__)
  60. #define BSWAP(a) __asm__ ( "bswapl %0n" : "=r" (a) : "0" (a) )
  61. #else
  62. #define BSWAP(a) 
  63.     ((a) = ( ((a)&0xff)<<24) | (((a)&0xff00)<<8) | (((a)>>8)&0xff00) | (((a)>>24)&0xff))
  64. #endif
  65. static uint32_t bitmask[] = {
  66.     0x0, 0x1, 0x3, 0x7, 0xF, 0x1F, 0x3F, 0x7F, 0xFF, 0x1FF,
  67.     0x3FF, 0x7FF, 0xFFF, 0x1FFF, 0x3FFF, 0x7FFF, 0xFFFF,
  68.     0x1FFFF, 0x3FFFF, 0x7FFFF, 0xFFFFF, 0x1FFFFF, 0x3FFFFF,
  69.     0x7FFFFF, 0xFFFFFF, 0x1FFFFFF, 0x3FFFFFF, 0x7FFFFFF,
  70.     0xFFFFFFF, 0x1FFFFFFF, 0x3FFFFFFF, 0x7FFFFFFF
  71.     /* added bitmask 32, correct?!?!?! */
  72.     , 0xFFFFFFFF
  73. };
  74. void faad_initbits(bitfile *ld, const void *buffer, const uint32_t buffer_size);
  75. void faad_endbits(bitfile *ld);
  76. void faad_initbits_rev(bitfile *ld, void *buffer,
  77.                        uint32_t bits_in_buffer);
  78. uint8_t faad_byte_align(bitfile *ld);
  79. uint32_t faad_get_processed_bits(bitfile *ld);
  80. void faad_flushbits_ex(bitfile *ld, uint32_t bits);
  81. void faad_rewindbits(bitfile *ld);
  82. uint8_t *faad_getbitbuffer(bitfile *ld, uint32_t bits
  83.                        DEBUGDEC);
  84. #ifdef DRM
  85. void *faad_origbitbuffer(bitfile *ld);
  86. uint32_t faad_origbitbuffer_size(bitfile *ld);
  87. #endif
  88. /* circumvent memory alignment errors on ARM */
  89. static INLINE uint32_t getdword(void *mem)
  90. {
  91. #ifdef ARM
  92.     uint32_t tmp;
  93. #ifndef ARCH_IS_BIG_ENDIAN
  94.     ((uint8_t*)&tmp)[0] = ((uint8_t*)mem)[3];
  95.     ((uint8_t*)&tmp)[1] = ((uint8_t*)mem)[2];
  96.     ((uint8_t*)&tmp)[2] = ((uint8_t*)mem)[1];
  97.     ((uint8_t*)&tmp)[3] = ((uint8_t*)mem)[0];
  98. #else
  99.     ((uint8_t*)&tmp)[0] = ((uint8_t*)mem)[0];
  100.     ((uint8_t*)&tmp)[1] = ((uint8_t*)mem)[1];
  101.     ((uint8_t*)&tmp)[2] = ((uint8_t*)mem)[2];
  102.     ((uint8_t*)&tmp)[3] = ((uint8_t*)mem)[3];
  103. #endif
  104.     return tmp;
  105. #else
  106.     uint32_t tmp;
  107.     tmp = *(uint32_t*)mem;
  108. #ifndef ARCH_IS_BIG_ENDIAN
  109.     BSWAP(tmp);
  110. #endif
  111.     return tmp;
  112. #endif
  113. }
  114. static INLINE uint32_t faad_showbits(bitfile *ld, uint32_t bits)
  115. {
  116.     if (bits <= ld->bits_left)
  117.     {
  118.         return (ld->bufa >> (ld->bits_left - bits)) & bitmask[bits];
  119.     }
  120.     bits -= ld->bits_left;
  121.     return ((ld->bufa & bitmask[ld->bits_left]) << bits) | (ld->bufb >> (32 - bits));
  122. }
  123. static INLINE void faad_flushbits(bitfile *ld, uint32_t bits)
  124. {
  125.     /* do nothing if error */
  126.     if (ld->error != 0)
  127.         return;
  128.     if (bits < ld->bits_left)
  129.     {
  130.         ld->bits_left -= bits;
  131.     } else {
  132.         faad_flushbits_ex(ld, bits);
  133.     }
  134. }
  135. /* return next n bits (right adjusted) */
  136. static INLINE uint32_t faad_getbits(bitfile *ld, uint32_t n DEBUGDEC)
  137. {
  138.     uint32_t ret;
  139.     if (ld->no_more_reading || n == 0)
  140.         return 0;
  141.     ret = faad_showbits(ld, n);
  142.     faad_flushbits(ld, n);
  143. #ifdef ANALYSIS
  144.     if (print)
  145.         fprintf(stdout, "%4d %2d bits, val: %4d, variable: %d %sn", dbg_count++, n, ret, var, dbg);
  146. #endif
  147.     return ret;
  148. }
  149. static INLINE uint8_t faad_get1bit(bitfile *ld DEBUGDEC)
  150. {
  151.     uint8_t r;
  152.     if (ld->bits_left > 0)
  153.     {
  154.         ld->bits_left--;
  155.         r = (uint8_t)((ld->bufa >> ld->bits_left) & 1);
  156.         return r;
  157.     }
  158.     /* bits_left == 0 */
  159. #if 0
  160.     r = (uint8_t)(ld->bufb >> 31);
  161.     faad_flushbits_ex(ld, 1);
  162. #else
  163.     r = (uint8_t)faad_getbits(ld, 1);
  164. #endif
  165.     return r;
  166. }
  167. /* reversed bitreading routines */
  168. static INLINE uint32_t faad_showbits_rev(bitfile *ld, uint32_t bits)
  169. {
  170.     uint8_t i;
  171.     uint32_t B = 0;
  172.     if (bits <= ld->bits_left)
  173.     {
  174.         for (i = 0; i < bits; i++)
  175.         {
  176.             if (ld->bufa & (1 << (i + (32 - ld->bits_left))))
  177.                 B |= (1 << (bits - i - 1));
  178.         }
  179.         return B;
  180.     } else {
  181.         for (i = 0; i < ld->bits_left; i++)
  182.         {
  183.             if (ld->bufa & (1 << (i + (32 - ld->bits_left))))
  184.                 B |= (1 << (bits - i - 1));
  185.         }
  186.         for (i = 0; i < bits - ld->bits_left; i++)
  187.         {
  188.             if (ld->bufb & (1 << (i + (32-ld->bits_left))))
  189.                 B |= (1 << (bits - ld->bits_left - i - 1));
  190.         }
  191.         return B;
  192.     }
  193. }
  194. static INLINE void faad_flushbits_rev(bitfile *ld, uint32_t bits)
  195. {
  196.     /* do nothing if error */
  197.     if (ld->error != 0)
  198.         return;
  199.     if (bits < ld->bits_left)
  200.     {
  201.         ld->bits_left -= bits;
  202.     } else {
  203.         uint32_t tmp;
  204.         ld->bufa = ld->bufb;
  205.         tmp = getdword(ld->start);
  206.         ld->bufb = tmp;
  207.         ld->start--;
  208.         ld->bits_left += (32 - bits);
  209.         ld->bytes_used += 4;
  210.         if (ld->bytes_used == ld->buffer_size)
  211.             ld->no_more_reading = 1;
  212.         if (ld->bytes_used > ld->buffer_size)
  213.             ld->error = 1;
  214.     }
  215. }
  216. static INLINE uint32_t faad_getbits_rev(bitfile *ld, uint32_t n
  217.                                         DEBUGDEC)
  218. {
  219.     uint32_t ret;
  220.     if (ld->no_more_reading)
  221.         return 0;
  222.     if (n == 0)
  223.         return 0;
  224.     ret = faad_showbits_rev(ld, n);
  225.     faad_flushbits_rev(ld, n);
  226. #ifdef ANALYSIS
  227.     if (print)
  228.         fprintf(stdout, "%4d %2d bits, val: %4d, variable: %d %sn", dbg_count++, n, ret, var, dbg);
  229. #endif
  230.     return ret;
  231. }
  232. #ifdef DRM
  233. static uint8_t faad_check_CRC(bitfile *ld, uint16_t len)
  234. {
  235.     uint8_t CRC;
  236.     uint16_t r=255;  /* Initialize to all ones */
  237.     /* CRC polynome used x^8 + x^4 + x^3 + x^2 +1 */
  238. #define GPOLY 0435
  239.     faad_rewindbits(ld);
  240.     CRC = (uint8_t) ~faad_getbits(ld, 8
  241.         DEBUGVAR(1,999,"faad_check_CRC(): CRC"));          /* CRC is stored inverted */
  242.     for (; len>0; len--)
  243.     {
  244.         r = ( (r << 1) ^ (( ( faad_get1bit(ld
  245.             DEBUGVAR(1,998,""))  & 1) ^ ((r >> 7) & 1)) * GPOLY )) & 0xFF;
  246.     }
  247.     if (r != CRC)
  248.   //  if (0)
  249.     {
  250.         return 28;
  251.     } else {
  252.         return 0;
  253.     }
  254. }
  255. static uint8_t tabFlipbits[256] = {
  256.     0,128,64,192,32,160,96,224,16,144,80,208,48,176,112,240,
  257.     8,136,72,200,40,168,104,232,24,152,88,216,56,184,120,248,
  258.     4,132,68,196,36,164,100,228,20,148,84,212,52,180,116,244,
  259.     12,140,76,204,44,172,108,236,28,156,92,220,60,188,124,252,
  260.     2,130,66,194,34,162,98,226,18,146,82,210,50,178,114,242,
  261.     10,138,74,202,42,170,106,234,26,154,90,218,58,186,122,250,
  262.     6,134,70,198,38,166,102,230,22,150,86,214,54,182,118,246,
  263.     14,142,78,206,46,174,110,238,30,158,94,222,62,190,126,254,
  264.     1,129,65,193,33,161,97,225,17,145,81,209,49,177,113,241,
  265.     9,137,73,201,41,169,105,233,25,153,89,217,57,185,121,249,
  266.     5,133,69,197,37,165,101,229,21,149,85,213,53,181,117,245,
  267.     13,141,77,205,45,173,109,237,29,157,93,221,61,189,125,253,
  268.     3,131,67,195,35,163,99,227,19,147,83,211,51,179,115,243,
  269.     11,139,75,203,43,171,107,235,27,155,91,219,59,187,123,251,
  270.     7,135,71,199,39,167,103,231,23,151,87,215,55,183,119,247,
  271.     15,143,79,207,47,175,111,239,31,159,95,223,63,191,127,255
  272. };
  273. #endif
  274. #ifdef ERROR_RESILIENCE
  275. /* Modified bit reading functions for HCR */
  276. typedef struct
  277. {
  278.     /* bit input */
  279.     uint32_t bufa;
  280.     uint32_t bufb;
  281.     int8_t len;
  282. } bits_t;
  283. static INLINE uint32_t showbits_hcr(bits_t *ld, uint8_t bits)
  284. {
  285.     if (bits == 0) return 0;
  286.     if (ld->len <= 32)
  287.     {
  288.         /* huffman_spectral_data_2 needs to read more than may be available, bits maybe
  289.            > ld->len, deliver 0 than */
  290.         if (ld->len >= bits)
  291.             return ((ld->bufa >> (ld->len - bits)) & (0xFFFFFFFF >> (32 - bits)));
  292.         else
  293.             return ((ld->bufa << (bits - ld->len)) & (0xFFFFFFFF >> (32 - bits)));
  294.     } else {
  295.         if ((ld->len - bits) < 32)
  296.         {
  297.             return ( (ld->bufb & (0xFFFFFFFF >> (64 - ld->len))) << (bits - ld->len + 32)) |
  298.                 (ld->bufa >> (ld->len - bits));
  299.         } else {
  300.             return ((ld->bufb >> (ld->len - bits - 32)) & (0xFFFFFFFF >> (32 - bits)));
  301.         }
  302.     }
  303. }
  304. /* return 1 if position is outside of buffer, 0 otherwise */
  305. static INLINE int8_t flushbits_hcr( bits_t *ld, uint8_t bits)
  306. {
  307.     ld->len -= bits;
  308.     if (ld->len <0)
  309.     {
  310.         ld->len = 0;
  311.         return 1;
  312.     } else {
  313.         return 0;
  314.     }
  315. }
  316. static INLINE int8_t getbits_hcr(bits_t *ld, uint8_t n, uint32_t *result)
  317. {
  318.     *result = showbits_hcr(ld, n);
  319.     return flushbits_hcr(ld, n);
  320. }
  321. static INLINE int8_t get1bit_hcr(bits_t *ld, uint8_t *result)
  322. {
  323.     uint32_t res;
  324.     int8_t ret;
  325.     ret = getbits_hcr(ld, 1, &res);
  326.     *result = (int8_t)(res & 1);
  327.     return ret;
  328. }
  329. #endif
  330. #ifdef __cplusplus
  331. }
  332. #endif
  333. #endif