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

多媒体编程

开发平台:

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: tns.c,v 1.3 2005/11/01 21:41:43 gabest Exp $
  31. **/
  32. #include "common.h"
  33. #include "structs.h"
  34. #include "syntax.h"
  35. #include "tns.h"
  36. /* static function declarations */
  37. static void tns_decode_coef(uint8_t order, uint8_t coef_res_bits, uint8_t coef_compress,
  38.                             uint8_t *coef, real_t *a);
  39. static void tns_ar_filter(real_t *spectrum, uint16_t size, int8_t inc, real_t *lpc,
  40.                           uint8_t order);
  41. static void tns_ma_filter(real_t *spectrum, uint16_t size, int8_t inc, real_t *lpc,
  42.                           uint8_t order);
  43. #ifdef _MSC_VER
  44. #pragma warning(disable:4305)
  45. #pragma warning(disable:4244)
  46. #endif
  47. static real_t tns_coef_0_3[] =
  48. {
  49.     COEF_CONST(0.0), COEF_CONST(0.4338837391), COEF_CONST(0.7818314825), COEF_CONST(0.9749279122),
  50.     COEF_CONST(-0.9848077530), COEF_CONST(-0.8660254038), COEF_CONST(-0.6427876097), COEF_CONST(-0.3420201433),
  51.     COEF_CONST(-0.4338837391), COEF_CONST(-0.7818314825), COEF_CONST(-0.9749279122), COEF_CONST(-0.9749279122),
  52.     COEF_CONST(-0.9848077530), COEF_CONST(-0.8660254038), COEF_CONST(-0.6427876097), COEF_CONST(-0.3420201433)
  53. };
  54. static real_t tns_coef_0_4[] =
  55. {
  56.     COEF_CONST(0.0), COEF_CONST(0.2079116908), COEF_CONST(0.4067366431), COEF_CONST(0.5877852523),
  57.     COEF_CONST(0.7431448255), COEF_CONST(0.8660254038), COEF_CONST(0.9510565163), COEF_CONST(0.9945218954),
  58.     COEF_CONST(-0.9957341763), COEF_CONST(-0.9618256432), COEF_CONST(-0.8951632914), COEF_CONST(-0.7980172273),
  59.     COEF_CONST(-0.6736956436), COEF_CONST(-0.5264321629), COEF_CONST(-0.3612416662), COEF_CONST(-0.1837495178)
  60. };
  61. static real_t tns_coef_1_3[] =
  62. {
  63.     COEF_CONST(0.0), COEF_CONST(0.4338837391), COEF_CONST(-0.6427876097), COEF_CONST(-0.3420201433),
  64.     COEF_CONST(0.9749279122), COEF_CONST(0.7818314825), COEF_CONST(-0.6427876097), COEF_CONST(-0.3420201433),
  65.     COEF_CONST(-0.4338837391), COEF_CONST(-0.7818314825), COEF_CONST(-0.6427876097), COEF_CONST(-0.3420201433),
  66.     COEF_CONST(-0.7818314825), COEF_CONST(-0.4338837391), COEF_CONST(-0.6427876097), COEF_CONST(-0.3420201433)
  67. };
  68. static real_t tns_coef_1_4[] =
  69. {
  70.     COEF_CONST(0.0), COEF_CONST(0.2079116908), COEF_CONST(0.4067366431), COEF_CONST(0.5877852523),
  71.     COEF_CONST(-0.6736956436), COEF_CONST(-0.5264321629), COEF_CONST(-0.3612416662), COEF_CONST(-0.1837495178),
  72.     COEF_CONST(0.9945218954), COEF_CONST(0.9510565163), COEF_CONST(0.8660254038), COEF_CONST(0.7431448255),
  73.     COEF_CONST(-0.6736956436), COEF_CONST(-0.5264321629), COEF_CONST(-0.3612416662), COEF_CONST(-0.1837495178)
  74. };
  75. /* TNS decoding for one channel and frame */
  76. void tns_decode_frame(ic_stream *ics, tns_info *tns, uint8_t sr_index,
  77.                       uint8_t object_type, real_t *spec, uint16_t frame_len)
  78. {
  79.     uint8_t w, f, tns_order;
  80.     int8_t inc;
  81.     int16_t size;
  82.     uint16_t bottom, top, start, end;
  83.     uint16_t nshort = frame_len/8;
  84.     real_t lpc[TNS_MAX_ORDER+1];
  85.     if (!ics->tns_data_present)
  86.         return;
  87.     for (w = 0; w < ics->num_windows; w++)
  88.     {
  89.         bottom = ics->num_swb;
  90.         for (f = 0; f < tns->n_filt[w]; f++)
  91.         {
  92.             top = bottom;
  93.             bottom = max(top - tns->length[w][f], 0);
  94.             tns_order = min(tns->order[w][f], TNS_MAX_ORDER);
  95.             if (!tns_order)
  96.                 continue;
  97.             tns_decode_coef(tns_order, tns->coef_res[w]+3,
  98.                 tns->coef_compress[w][f], tns->coef[w][f], lpc);
  99.             start = min(bottom, max_tns_sfb(sr_index, object_type, (ics->window_sequence == EIGHT_SHORT_SEQUENCE)));
  100.             start = min(start, ics->max_sfb);
  101.             start = ics->swb_offset[start];
  102.             end = min(top, max_tns_sfb(sr_index, object_type, (ics->window_sequence == EIGHT_SHORT_SEQUENCE)));
  103.             end = min(end, ics->max_sfb);
  104.             end = ics->swb_offset[end];
  105.             size = end - start;
  106.             if (size <= 0)
  107.                 continue;
  108.             if (tns->direction[w][f])
  109.             {
  110.                 inc = -1;
  111.                 start = end - 1;
  112.             } else {
  113.                 inc = 1;
  114.             }
  115.             tns_ar_filter(&spec[(w*nshort)+start], size, inc, lpc, tns_order);
  116.         }
  117.     }
  118. }
  119. /* TNS encoding for one channel and frame */
  120. void tns_encode_frame(ic_stream *ics, tns_info *tns, uint8_t sr_index,
  121.                       uint8_t object_type, real_t *spec, uint16_t frame_len)
  122. {
  123.     uint8_t w, f, tns_order;
  124.     int8_t inc;
  125.     int16_t size;
  126.     uint16_t bottom, top, start, end;
  127.     uint16_t nshort = frame_len/8;
  128.     real_t lpc[TNS_MAX_ORDER+1];
  129.     if (!ics->tns_data_present)
  130.         return;
  131.     for (w = 0; w < ics->num_windows; w++)
  132.     {
  133.         bottom = ics->num_swb;
  134.         for (f = 0; f < tns->n_filt[w]; f++)
  135.         {
  136.             top = bottom;
  137.             bottom = max(top - tns->length[w][f], 0);
  138.             tns_order = min(tns->order[w][f], TNS_MAX_ORDER);
  139.             if (!tns_order)
  140.                 continue;
  141.             tns_decode_coef(tns_order, tns->coef_res[w]+3,
  142.                 tns->coef_compress[w][f], tns->coef[w][f], lpc);
  143.             start = min(bottom, max_tns_sfb(sr_index, object_type, (ics->window_sequence == EIGHT_SHORT_SEQUENCE)));
  144.             start = min(start, ics->max_sfb);
  145.             start = ics->swb_offset[start];
  146.             end = min(top, max_tns_sfb(sr_index, object_type, (ics->window_sequence == EIGHT_SHORT_SEQUENCE)));
  147.             end = min(end, ics->max_sfb);
  148.             end = ics->swb_offset[end];
  149.             size = end - start;
  150.             if (size <= 0)
  151.                 continue;
  152.             if (tns->direction[w][f])
  153.             {
  154.                 inc = -1;
  155.                 start = end - 1;
  156.             } else {
  157.                 inc = 1;
  158.             }
  159.             tns_ma_filter(&spec[(w*nshort)+start], size, inc, lpc, tns_order);
  160.         }
  161.     }
  162. }
  163. /* Decoder transmitted coefficients for one TNS filter */
  164. static void tns_decode_coef(uint8_t order, uint8_t coef_res_bits, uint8_t coef_compress,
  165.                             uint8_t *coef, real_t *a)
  166. {
  167.     uint8_t i, m;
  168.     real_t tmp2[TNS_MAX_ORDER+1], b[TNS_MAX_ORDER+1];
  169.     /* Conversion to signed integer */
  170.     for (i = 0; i < order; i++)
  171.     {
  172.         if (coef_compress == 0)
  173.         {
  174.             if (coef_res_bits == 3)
  175.             {
  176.                 tmp2[i] = tns_coef_0_3[coef[i]];
  177.             } else {
  178.                 tmp2[i] = tns_coef_0_4[coef[i]];
  179.             }
  180.         } else {
  181.             if (coef_res_bits == 3)
  182.             {
  183.                 tmp2[i] = tns_coef_1_3[coef[i]];
  184.             } else {
  185.                 tmp2[i] = tns_coef_1_4[coef[i]];
  186.             }
  187.         }
  188.     }
  189.     /* Conversion to LPC coefficients */
  190.     a[0] = COEF_CONST(1.0);
  191.     for (m = 1; m <= order; m++)
  192.     {
  193.         for (i = 1; i < m; i++) /* loop only while i<m */
  194.             b[i] = a[i] + MUL_C(tmp2[m-1], a[m-i]);
  195.         for (i = 1; i < m; i++) /* loop only while i<m */
  196.             a[i] = b[i];
  197.         a[m] = tmp2[m-1]; /* changed */
  198.     }
  199. }
  200. static void tns_ar_filter(real_t *spectrum, uint16_t size, int8_t inc, real_t *lpc,
  201.                           uint8_t order)
  202. {
  203.     /*
  204.      - Simple all-pole filter of order "order" defined by
  205.        y(n) = x(n) - lpc[1]*y(n-1) - ... - lpc[order]*y(n-order)
  206.      - The state variables of the filter are initialized to zero every time
  207.      - The output data is written over the input data ("in-place operation")
  208.      - An input vector of "size" samples is processed and the index increment
  209.        to the next data sample is given by "inc"
  210.     */
  211.     uint8_t j;
  212.     uint16_t i;
  213.     real_t y;
  214.     /* state is stored as a double ringbuffer */
  215.     real_t state[2*TNS_MAX_ORDER] = {0};
  216.     int8_t state_index = 0;
  217.     for (i = 0; i < size; i++)
  218.     {
  219.         y = *spectrum;
  220.         for (j = 0; j < order; j++)
  221.             y -= MUL_C(state[state_index+j], lpc[j+1]);
  222.         /* double ringbuffer state */
  223.         state_index--;
  224.         if (state_index < 0)
  225.             state_index = order-1;
  226.         state[state_index] = state[state_index + order] = y;
  227.         *spectrum = y;
  228.         spectrum += inc;
  229. //#define TNS_PRINT
  230. #ifdef TNS_PRINT
  231.         //printf("%dn", y);
  232.         printf("0x%.8Xn", y);
  233. #endif
  234.     }
  235. }
  236. static void tns_ma_filter(real_t *spectrum, uint16_t size, int8_t inc, real_t *lpc,
  237.                           uint8_t order)
  238. {
  239.     /*
  240.      - Simple all-zero filter of order "order" defined by
  241.        y(n) =  x(n) + a(2)*x(n-1) + ... + a(order+1)*x(n-order)
  242.      - The state variables of the filter are initialized to zero every time
  243.      - The output data is written over the input data ("in-place operation")
  244.      - An input vector of "size" samples is processed and the index increment
  245.        to the next data sample is given by "inc"
  246.     */
  247.     uint8_t j;
  248.     uint16_t i;
  249.     real_t y;
  250.     /* state is stored as a double ringbuffer */
  251.     real_t state[2*TNS_MAX_ORDER] = {0};
  252.     int8_t state_index = 0;
  253.     for (i = 0; i < size; i++)
  254.     {
  255.         y = *spectrum;
  256.         for (j = 0; j < order; j++)
  257.             y += MUL_C(state[j], lpc[j+1]);
  258.         /* double ringbuffer state */
  259.         state_index--;
  260.         if (state_index < 0)
  261.             state_index = order-1;
  262.         state[state_index] = state[state_index + order] = *spectrum;
  263.         *spectrum = y;
  264.         spectrum += inc;
  265.     }
  266. }