tx.h 4.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. /*
  2. * This file is part of FFmpeg.
  3. *
  4. * FFmpeg is free software; you can redistribute it and/or
  5. * modify it under the terms of the GNU Lesser General Public
  6. * License as published by the Free Software Foundation; either
  7. * version 2.1 of the License, or (at your option) any later version.
  8. *
  9. * FFmpeg is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  12. * Lesser General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU Lesser General Public
  15. * License along with FFmpeg; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. #ifndef AVUTIL_TX_H
  19. #define AVUTIL_TX_H
  20. #include <stdint.h>
  21. #include <stddef.h>
  22. typedef struct AVTXContext AVTXContext;
  23. typedef struct AVComplexFloat {
  24. float re, im;
  25. } AVComplexFloat;
  26. typedef struct AVComplexDouble {
  27. double re, im;
  28. } AVComplexDouble;
  29. typedef struct AVComplexInt32 {
  30. int32_t re, im;
  31. } AVComplexInt32;
  32. enum AVTXType {
  33. /**
  34. * Standard complex to complex FFT with sample data type AVComplexFloat.
  35. * Output is not 1/len normalized. Scaling currently unsupported.
  36. * The stride parameter is ignored.
  37. */
  38. AV_TX_FLOAT_FFT = 0,
  39. /**
  40. * Standard MDCT with sample data type of float and a scale type of
  41. * float. Length is the frame size, not the window size (which is 2x frame)
  42. * For forward transforms, the stride specifies the spacing between each
  43. * sample in the output array in bytes. The input must be a flat array.
  44. *
  45. * For inverse transforms, the stride specifies the spacing between each
  46. * sample in the input array in bytes. The output will be a flat array.
  47. * Stride must be a non-zero multiple of sizeof(float).
  48. *
  49. * NOTE: the inverse transform is half-length, meaning the output will not
  50. * contain redundant data. This is what most codecs work with. To do a full
  51. * inverse transform, set the AV_TX_FULL_IMDCT flag on init.
  52. */
  53. AV_TX_FLOAT_MDCT = 1,
  54. /**
  55. * Same as AV_TX_FLOAT_FFT with a data type of AVComplexDouble.
  56. */
  57. AV_TX_DOUBLE_FFT = 2,
  58. /**
  59. * Same as AV_TX_FLOAT_MDCT with data and scale type of double.
  60. * Stride must be a non-zero multiple of sizeof(double).
  61. */
  62. AV_TX_DOUBLE_MDCT = 3,
  63. /**
  64. * Same as AV_TX_FLOAT_FFT with a data type of AVComplexInt32.
  65. */
  66. AV_TX_INT32_FFT = 4,
  67. /**
  68. * Same as AV_TX_FLOAT_MDCT with data type of int32_t and scale type of float.
  69. * Only scale values less than or equal to 1.0 are supported.
  70. * Stride must be a non-zero multiple of sizeof(int32_t).
  71. */
  72. AV_TX_INT32_MDCT = 5,
  73. };
  74. /**
  75. * Function pointer to a function to perform the transform.
  76. *
  77. * @note Using a different context than the one allocated during av_tx_init()
  78. * is not allowed.
  79. *
  80. * @param s the transform context
  81. * @param out the output array
  82. * @param in the input array
  83. * @param stride the input or output stride in bytes
  84. *
  85. * The out and in arrays must be aligned to the maximum required by the CPU
  86. * architecture unless the AV_TX_UNALIGNED flag was set in av_tx_init().
  87. * The stride must follow the constraints the transform type has specified.
  88. */
  89. typedef void (*av_tx_fn)(AVTXContext *s, void *out, void *in, ptrdiff_t stride);
  90. /**
  91. * Flags for av_tx_init()
  92. */
  93. enum AVTXFlags {
  94. /**
  95. * Performs an in-place transformation on the input. The output argument
  96. * of av_tn_fn() MUST match the input. May be unsupported or slower for some
  97. * transform types.
  98. */
  99. AV_TX_INPLACE = 1ULL << 0,
  100. /**
  101. * Relaxes alignment requirement for the in and out arrays of av_tx_fn().
  102. * May be slower with certain transform types.
  103. */
  104. AV_TX_UNALIGNED = 1ULL << 1,
  105. /**
  106. * Performs a full inverse MDCT rather than leaving out samples that can be
  107. * derived through symmetry. Requires an output array of 'len' floats,
  108. * rather than the usual 'len/2' floats.
  109. * Ignored for all transforms but inverse MDCTs.
  110. */
  111. AV_TX_FULL_IMDCT = 1ULL << 2,
  112. };
  113. /**
  114. * Initialize a transform context with the given configuration
  115. * (i)MDCTs with an odd length are currently not supported.
  116. *
  117. * @param ctx the context to allocate, will be NULL on error
  118. * @param tx pointer to the transform function pointer to set
  119. * @param type type the type of transform
  120. * @param inv whether to do an inverse or a forward transform
  121. * @param len the size of the transform in samples
  122. * @param scale pointer to the value to scale the output if supported by type
  123. * @param flags a bitmask of AVTXFlags or 0
  124. *
  125. * @return 0 on success, negative error code on failure
  126. */
  127. int av_tx_init(AVTXContext **ctx, av_tx_fn *tx, enum AVTXType type,
  128. int inv, int len, const void *scale, uint64_t flags);
  129. /**
  130. * Frees a context and sets ctx to NULL, does nothing when ctx == NULL
  131. */
  132. void av_tx_uninit(AVTXContext **ctx);
  133. #endif /* AVUTIL_TX_H */