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.

244 lines
8.7KB

  1. /*
  2. * adaptive and fixed codebook vector operations for ACELP-based codecs
  3. *
  4. * Copyright (c) 2008 Vladimir Voroshilov
  5. *
  6. * This file is part of FFmpeg.
  7. *
  8. * FFmpeg is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU Lesser General Public
  10. * License as published by the Free Software Foundation; either
  11. * version 2.1 of the License, or (at your option) any later version.
  12. *
  13. * FFmpeg is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. * Lesser General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU Lesser General Public
  19. * License along with FFmpeg; if not, write to the Free Software
  20. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. */
  22. #ifndef AVCODEC_ACELP_VECTORS_H
  23. #define AVCODEC_ACELP_VECTORS_H
  24. #include <stdint.h>
  25. /** Sparse representation for the algebraic codebook (fixed) vector */
  26. typedef struct {
  27. int n;
  28. int x[10];
  29. float y[10];
  30. int pitch_lag;
  31. float pitch_fac;
  32. } AMRFixed;
  33. /**
  34. * Track|Pulse| Positions
  35. * -------------------------------------------------------------------------
  36. * 1 | 0 | 0, 5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75
  37. * -------------------------------------------------------------------------
  38. * 2 | 1 | 1, 6, 11, 16, 21, 26, 31, 36, 41, 46, 51, 56, 61, 66, 71, 76
  39. * -------------------------------------------------------------------------
  40. * 3 | 2 | 2, 7, 12, 17, 22, 27, 32, 37, 42, 47, 52, 57, 62, 67, 72, 77
  41. * -------------------------------------------------------------------------
  42. *
  43. * Table contains only first the pulse indexes.
  44. *
  45. * Used in G.729 @8k, G.729 @4.4k, AMR @7.95k, AMR @7.40k
  46. */
  47. extern const uint8_t ff_fc_4pulses_8bits_tracks_13[16];
  48. /**
  49. * Track|Pulse| Positions
  50. * -------------------------------------------------------------------------
  51. * 4 | 3 | 3, 8, 13, 18, 23, 28, 33, 38, 43, 48, 53, 58, 63, 68, 73, 78
  52. * | | 4, 9, 14, 19, 24, 29, 34, 39, 44, 49, 54, 59, 64, 69, 74, 79
  53. * -------------------------------------------------------------------------
  54. *
  55. * @remark Track in the table should be read top-to-bottom, left-to-right.
  56. *
  57. * Used in G.729 @8k, G.729 @4.4k, AMR @7.95k, AMR @7.40k
  58. */
  59. extern const uint8_t ff_fc_4pulses_8bits_track_4[32];
  60. /**
  61. * Track|Pulse| Positions
  62. * -----------------------------------------
  63. * 1 | 0 | 1, 6, 11, 16, 21, 26, 31, 36
  64. * | | 3, 8, 13, 18, 23, 28, 33, 38
  65. * -----------------------------------------
  66. *
  67. * @remark Track in the table should be read top-to-bottom, left-to-right.
  68. *
  69. * @note (EE) Reference G.729D code also uses gray decoding for each
  70. * pulse index before looking up the value in the table.
  71. *
  72. * Used in G.729 @6.4k (with gray coding), AMR @5.9k (without gray coding)
  73. */
  74. extern const uint8_t ff_fc_2pulses_9bits_track1[16];
  75. extern const uint8_t ff_fc_2pulses_9bits_track1_gray[16];
  76. /**
  77. * Track|Pulse| Positions
  78. * -----------------------------------------
  79. * 2 | 1 | 0, 7, 14, 20, 27, 34, 1, 21
  80. * | | 2, 9, 15, 22, 29, 35, 6, 26
  81. * | | 4,10, 17, 24, 30, 37, 11, 31
  82. * | | 5,12, 19, 25, 32, 39, 16, 36
  83. * -----------------------------------------
  84. *
  85. * @remark Track in the table should be read top-to-bottom, left-to-right.
  86. *
  87. * @note (EE.1) This table (from the reference code) does not comply with
  88. * the specification.
  89. * The specification contains the following table:
  90. *
  91. * Track|Pulse| Positions
  92. * -----------------------------------------
  93. * 2 | 1 | 0, 5, 10, 15, 20, 25, 30, 35
  94. * | | 1, 6, 11, 16, 21, 26, 31, 36
  95. * | | 2, 7, 12, 17, 22, 27, 32, 37
  96. * | | 4, 9, 14, 19, 24, 29, 34, 39
  97. *
  98. * -----------------------------------------
  99. *
  100. * @note (EE.2) Reference G.729D code also uses gray decoding for each
  101. * pulse index before looking up the value in the table.
  102. *
  103. * Used in G.729 @6.4k (with gray coding)
  104. */
  105. extern const uint8_t ff_fc_2pulses_9bits_track2_gray[32];
  106. /**
  107. * Decode fixed-codebook vector (3.8 and D.5.8 of G.729, 5.7.1 of AMR).
  108. * @param fc_v [out] decoded fixed codebook vector (2.13)
  109. * @param tab1 table used for first pulse_count pulses
  110. * @param tab2 table used for last pulse
  111. * @param pulse_indexes fixed codebook indexes
  112. * @param pulse_signs signs of the excitation pulses (0 bit value
  113. * means negative sign)
  114. * @param bits number of bits per one pulse index
  115. * @param pulse_count number of pulses decoded using first table
  116. * @param bits length of one pulse index in bits
  117. *
  118. * Used in G.729 @8k, G.729 @4.4k, G.729 @6.4k, AMR @7.95k, AMR @7.40k
  119. */
  120. void ff_acelp_fc_pulse_per_track(
  121. int16_t* fc_v,
  122. const uint8_t *tab1,
  123. const uint8_t *tab2,
  124. int pulse_indexes,
  125. int pulse_signs,
  126. int pulse_count,
  127. int bits);
  128. /**
  129. * Decode the algebraic codebook index to pulse positions and signs and
  130. * construct the algebraic codebook vector for MODE_12k2.
  131. *
  132. * @note: The positions and signs are explicitly coded in MODE_12k2.
  133. *
  134. * @param fixed_index positions of the ten pulses
  135. * @param fixed_sparse pointer to the algebraic codebook vector
  136. * @param gray_decode gray decoding table
  137. * @param half_pulse_count number of couples of pulses
  138. * @param bits length of one pulse index in bits
  139. */
  140. void ff_decode_10_pulses_35bits(const int16_t *fixed_index,
  141. AMRFixed *fixed_sparse,
  142. const uint8_t *gray_decode,
  143. int half_pulse_count, int bits);
  144. /**
  145. * weighted sum of two vectors with rounding.
  146. * @param out [out] result of addition
  147. * @param in_a first vector
  148. * @param in_b second vector
  149. * @param weight_coeff_a first vector weight coefficient
  150. * @param weight_coeff_a second vector weight coefficient
  151. * @param rounder this value will be added to the sum of the two vectors
  152. * @param shift result will be shifted to right by this value
  153. * @param length vectors length
  154. *
  155. * @note It is safe to pass the same buffer for out and in_a or in_b.
  156. *
  157. * out[i] = (in_a[i]*weight_a + in_b[i]*weight_b + rounder) >> shift
  158. */
  159. void ff_acelp_weighted_vector_sum(
  160. int16_t* out,
  161. const int16_t *in_a,
  162. const int16_t *in_b,
  163. int16_t weight_coeff_a,
  164. int16_t weight_coeff_b,
  165. int16_t rounder,
  166. int shift,
  167. int length);
  168. /**
  169. * float implementation of weighted sum of two vectors.
  170. * @param out [out] result of addition
  171. * @param in_a first vector
  172. * @param in_b second vector
  173. * @param weight_coeff_a first vector weight coefficient
  174. * @param weight_coeff_a second vector weight coefficient
  175. * @param length vectors length
  176. *
  177. * @note It is safe to pass the same buffer for out and in_a or in_b.
  178. */
  179. void ff_weighted_vector_sumf(float *out, const float *in_a, const float *in_b,
  180. float weight_coeff_a, float weight_coeff_b, int length);
  181. /**
  182. * Adaptative gain control (as used in AMR postfiltering)
  183. *
  184. * @param buf_out the input speech buffer
  185. * @param speech_energ input energy
  186. * @param size the input buffer size
  187. * @param alpha exponential filter factor
  188. * @param gain_mem a pointer to the filter memory (single float of size)
  189. */
  190. void ff_adaptative_gain_control(float *buf_out, float speech_energ,
  191. int size, float alpha, float *gain_mem);
  192. /**
  193. * Set the sum of squares of a signal by scaling
  194. *
  195. * @param out output samples
  196. * @param in input samples
  197. * @param sum_of_squares new sum of squares
  198. * @param n number of samples
  199. *
  200. * @note If the input is zero (or its energy underflows), the output is zero.
  201. * This is the behavior of AGC in the AMR reference decoder. The QCELP
  202. * reference decoder seems to have undefined behavior.
  203. *
  204. * TIA/EIA/IS-733 2.4.8.3-2/3/4/5, 2.4.8.6
  205. * 3GPP TS 26.090 6.1 (6)
  206. */
  207. void ff_scale_vector_to_given_sum_of_squares(float *out, const float *in,
  208. float sum_of_squares, const int n);
  209. /**
  210. * Add fixed vector to an array from a sparse representation
  211. *
  212. * @param out fixed vector with pitch sharpening
  213. * @param in sparse fixed vector
  214. * @param scale number to multiply the fixed vector by
  215. * @param size the output vector size
  216. */
  217. void ff_set_fixed_vector(float *out, const AMRFixed *in, float scale, int size);
  218. /**
  219. * Clear array values set by set_fixed_vector
  220. *
  221. * @param out fixed vector to be cleared
  222. * @param in sparse fixed vector
  223. * @param size the output vector size
  224. */
  225. void ff_clear_fixed_vector(float *out, const AMRFixed *in, int size);
  226. #endif /* AVCODEC_ACELP_VECTORS_H */