lpc.h 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212
  1. /*
  2. * LPC utility code
  3. * Copyright (c) 2006 Justin Ruggles <justin.ruggles@gmail.com>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg 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. * FFmpeg 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 FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. #ifndef AVCODEC_LPC_H
  22. #define AVCODEC_LPC_H
  23. #include <stdint.h>
  24. #include "libavutil/avassert.h"
  25. #include "libavutil/lls.h"
  26. #include "aac_defines.h"
  27. #define ORDER_METHOD_EST 0
  28. #define ORDER_METHOD_2LEVEL 1
  29. #define ORDER_METHOD_4LEVEL 2
  30. #define ORDER_METHOD_8LEVEL 3
  31. #define ORDER_METHOD_SEARCH 4
  32. #define ORDER_METHOD_LOG 5
  33. #define MIN_LPC_ORDER 1
  34. #define MAX_LPC_ORDER 32
  35. /**
  36. * LPC analysis type
  37. */
  38. enum FFLPCType {
  39. FF_LPC_TYPE_DEFAULT = -1, ///< use the codec default LPC type
  40. FF_LPC_TYPE_NONE = 0, ///< do not use LPC prediction or use all zero coefficients
  41. FF_LPC_TYPE_FIXED = 1, ///< fixed LPC coefficients
  42. FF_LPC_TYPE_LEVINSON = 2, ///< Levinson-Durbin recursion
  43. FF_LPC_TYPE_CHOLESKY = 3, ///< Cholesky factorization
  44. FF_LPC_TYPE_NB , ///< Not part of ABI
  45. };
  46. typedef struct LPCContext {
  47. int blocksize;
  48. int max_order;
  49. enum FFLPCType lpc_type;
  50. double *windowed_buffer;
  51. double *windowed_samples;
  52. /**
  53. * Apply a Welch window to an array of input samples.
  54. * The output samples have the same scale as the input, but are in double
  55. * sample format.
  56. * @param data input samples
  57. * @param len number of input samples
  58. * @param w_data output samples
  59. */
  60. void (*lpc_apply_welch_window)(const int32_t *data, int len,
  61. double *w_data);
  62. /**
  63. * Perform autocorrelation on input samples with delay of 0 to lag.
  64. * @param data input samples.
  65. * constraints: no alignment needed, but must have at
  66. * least lag*sizeof(double) valid bytes preceding it, and
  67. * size must be at least (len+1)*sizeof(double) if data is
  68. * 16-byte aligned or (len+2)*sizeof(double) if data is
  69. * unaligned.
  70. * @param len number of input samples to process
  71. * @param lag maximum delay to calculate
  72. * @param autoc output autocorrelation coefficients.
  73. * constraints: array size must be at least lag+1.
  74. */
  75. void (*lpc_compute_autocorr)(const double *data, int len, int lag,
  76. double *autoc);
  77. // TODO: these should be allocated to reduce ABI compatibility issues
  78. LLSModel lls_models[2];
  79. } LPCContext;
  80. /**
  81. * Calculate LPC coefficients for multiple orders
  82. */
  83. int ff_lpc_calc_coefs(LPCContext *s,
  84. const int32_t *samples, int blocksize, int min_order,
  85. int max_order, int precision,
  86. int32_t coefs[][MAX_LPC_ORDER], int *shift,
  87. enum FFLPCType lpc_type, int lpc_passes,
  88. int omethod, int min_shift, int max_shift, int zero_shift);
  89. int ff_lpc_calc_ref_coefs(LPCContext *s,
  90. const int32_t *samples, int order, double *ref);
  91. double ff_lpc_calc_ref_coefs_f(LPCContext *s, const float *samples, int len,
  92. int order, double *ref);
  93. /**
  94. * Initialize LPCContext.
  95. */
  96. int ff_lpc_init(LPCContext *s, int blocksize, int max_order,
  97. enum FFLPCType lpc_type);
  98. void ff_lpc_init_x86(LPCContext *s);
  99. /**
  100. * Uninitialize LPCContext.
  101. */
  102. void ff_lpc_end(LPCContext *s);
  103. #if USE_FIXED
  104. typedef int LPC_TYPE;
  105. typedef unsigned LPC_TYPE_U;
  106. #else
  107. #ifdef LPC_USE_DOUBLE
  108. typedef double LPC_TYPE;
  109. typedef double LPC_TYPE_U;
  110. #else
  111. typedef float LPC_TYPE;
  112. typedef float LPC_TYPE_U;
  113. #endif
  114. #endif // USE_FIXED
  115. /**
  116. * Schur recursion.
  117. * Produces reflection coefficients from autocorrelation data.
  118. */
  119. static inline void compute_ref_coefs(const LPC_TYPE *autoc, int max_order,
  120. LPC_TYPE *ref, LPC_TYPE *error)
  121. {
  122. int i, j;
  123. LPC_TYPE err;
  124. LPC_TYPE gen0[MAX_LPC_ORDER], gen1[MAX_LPC_ORDER];
  125. for (i = 0; i < max_order; i++)
  126. gen0[i] = gen1[i] = autoc[i + 1];
  127. err = autoc[0];
  128. ref[0] = -gen1[0] / err;
  129. err += gen1[0] * ref[0];
  130. if (error)
  131. error[0] = err;
  132. for (i = 1; i < max_order; i++) {
  133. for (j = 0; j < max_order - i; j++) {
  134. gen1[j] = gen1[j + 1] + ref[i - 1] * gen0[j];
  135. gen0[j] = gen1[j + 1] * ref[i - 1] + gen0[j];
  136. }
  137. ref[i] = -gen1[0] / err;
  138. err += gen1[0] * ref[i];
  139. if (error)
  140. error[i] = err;
  141. }
  142. }
  143. /**
  144. * Levinson-Durbin recursion.
  145. * Produce LPC coefficients from autocorrelation data.
  146. */
  147. static inline int AAC_RENAME(compute_lpc_coefs)(const LPC_TYPE *autoc, int max_order,
  148. LPC_TYPE *lpc, int lpc_stride, int fail,
  149. int normalize)
  150. {
  151. int i, j;
  152. LPC_TYPE err = 0;
  153. LPC_TYPE *lpc_last = lpc;
  154. av_assert2(normalize || !fail);
  155. if (normalize)
  156. err = *autoc++;
  157. if (fail && (autoc[max_order - 1] == 0 || err <= 0))
  158. return -1;
  159. for(i=0; i<max_order; i++) {
  160. LPC_TYPE r = AAC_SRA_R(-autoc[i], 5);
  161. if (normalize) {
  162. for(j=0; j<i; j++)
  163. r -= lpc_last[j] * autoc[i-j-1];
  164. r /= err;
  165. err *= FIXR(1.0) - (r * r);
  166. }
  167. lpc[i] = r;
  168. for(j=0; j < (i+1)>>1; j++) {
  169. LPC_TYPE f = lpc_last[ j];
  170. LPC_TYPE b = lpc_last[i-1-j];
  171. lpc[ j] = f + (LPC_TYPE_U)AAC_MUL26(r, b);
  172. lpc[i-1-j] = b + (LPC_TYPE_U)AAC_MUL26(r, f);
  173. }
  174. if (fail && err < 0)
  175. return -1;
  176. lpc_last = lpc;
  177. lpc += lpc_stride;
  178. }
  179. return 0;
  180. }
  181. #endif /* AVCODEC_LPC_H */