Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 1 | /* Copyright (c) 2007-2008 CSIRO |
| 2 | Copyright (c) 2007-2009 Xiph.Org Foundation |
| 3 | Copyright (c) 2008-2009 Gregory Maxwell |
| 4 | Written by Jean-Marc Valin and Gregory Maxwell */ |
| 5 | /* |
| 6 | Redistribution and use in source and binary forms, with or without |
| 7 | modification, are permitted provided that the following conditions |
| 8 | are met: |
| 9 | |
| 10 | - Redistributions of source code must retain the above copyright |
| 11 | notice, this list of conditions and the following disclaimer. |
| 12 | |
| 13 | - Redistributions in binary form must reproduce the above copyright |
| 14 | notice, this list of conditions and the following disclaimer in the |
| 15 | documentation and/or other materials provided with the distribution. |
| 16 | |
| 17 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 18 | ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 19 | LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 20 | A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER |
| 21 | OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
| 22 | EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
| 23 | PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
| 24 | PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
| 25 | LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
| 26 | NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
| 27 | SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 28 | */ |
| 29 | |
| 30 | #ifndef BANDS_H |
| 31 | #define BANDS_H |
| 32 | |
| 33 | #include "arch.h" |
| 34 | #include "modes.h" |
| 35 | #include "entenc.h" |
| 36 | #include "entdec.h" |
| 37 | #include "rate.h" |
| 38 | |
| 39 | /** Compute the amplitude (sqrt energy) in each of the bands |
| 40 | * @param m Mode data |
| 41 | * @param X Spectrum |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 42 | * @param bandE Square root of the energy for each band (returned) |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 43 | */ |
| 44 | void compute_band_energies(const CELTMode *m, const celt_sig *X, celt_ener *bandE, int end, int C, int M); |
| 45 | |
| 46 | /*void compute_noise_energies(const CELTMode *m, const celt_sig *X, const opus_val16 *tonality, celt_ener *bandE);*/ |
| 47 | |
| 48 | /** Normalise each band of X such that the energy in each band is |
| 49 | equal to 1 |
| 50 | * @param m Mode data |
| 51 | * @param X Spectrum (returned normalised) |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 52 | * @param bandE Square root of the energy for each band |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 53 | */ |
| 54 | void normalise_bands(const CELTMode *m, const celt_sig * OPUS_RESTRICT freq, celt_norm * OPUS_RESTRICT X, const celt_ener *bandE, int end, int C, int M); |
| 55 | |
| 56 | /** Denormalise each band of X to restore full amplitude |
| 57 | * @param m Mode data |
| 58 | * @param X Spectrum (returned de-normalised) |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 59 | * @param bandE Square root of the energy for each band |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 60 | */ |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 61 | void denormalise_bands(const CELTMode *m, const celt_norm * OPUS_RESTRICT X, |
| 62 | celt_sig * OPUS_RESTRICT freq, const opus_val16 *bandE, int start, int end, int C, int M); |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 63 | |
| 64 | #define SPREAD_NONE (0) |
| 65 | #define SPREAD_LIGHT (1) |
| 66 | #define SPREAD_NORMAL (2) |
| 67 | #define SPREAD_AGGRESSIVE (3) |
| 68 | |
| 69 | int spreading_decision(const CELTMode *m, celt_norm *X, int *average, |
| 70 | int last_decision, int *hf_average, int *tapset_decision, int update_hf, |
| 71 | int end, int C, int M); |
| 72 | |
| 73 | #ifdef MEASURE_NORM_MSE |
| 74 | void measure_norm_mse(const CELTMode *m, float *X, float *X0, float *bandE, float *bandE0, int M, int N, int C); |
| 75 | #endif |
| 76 | |
| 77 | void haar1(celt_norm *X, int N0, int stride); |
| 78 | |
| 79 | /** Quantisation/encoding of the residual spectrum |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 80 | * @param encode flag that indicates whether we're encoding (1) or decoding (0) |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 81 | * @param m Mode data |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 82 | * @param start First band to process |
| 83 | * @param end Last band to process + 1 |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 84 | * @param X Residual (normalised) |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 85 | * @param Y Residual (normalised) for second channel (or NULL for mono) |
| 86 | * @param collapse_masks Anti-collapse tracking mask |
| 87 | * @param bandE Square root of the energy for each band |
| 88 | * @param pulses Bit allocation (per band) for PVQ |
| 89 | * @param shortBlocks Zero for long blocks, non-zero for short blocks |
| 90 | * @param spread Amount of spreading to use |
| 91 | * @param dual_stereo Zero for MS stereo, non-zero for dual stereo |
| 92 | * @param intensity First band to use intensity stereo |
| 93 | * @param tf_res Time-frequency resolution change |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 94 | * @param total_bits Total number of bits that can be used for the frame (including the ones already spent) |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 95 | * @param balance Number of unallocated bits |
| 96 | * @param en Entropy coder state |
| 97 | * @param LM log2() of the number of 2.5 subframes in the frame |
| 98 | * @param codedBands Last band to receive bits + 1 |
| 99 | * @param seed Random generator seed |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 100 | */ |
| 101 | void quant_all_bands(int encode, const CELTMode *m, int start, int end, |
| 102 | celt_norm * X, celt_norm * Y, unsigned char *collapse_masks, const celt_ener *bandE, int *pulses, |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 103 | int shortBlocks, int spread, int dual_stereo, int intensity, int *tf_res, |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 104 | opus_int32 total_bits, opus_int32 balance, ec_ctx *ec, int M, int codedBands, opus_uint32 *seed); |
| 105 | |
| 106 | void anti_collapse(const CELTMode *m, celt_norm *X_, unsigned char *collapse_masks, int LM, int C, int size, |
| 107 | int start, int end, opus_val16 *logE, opus_val16 *prev1logE, |
| 108 | opus_val16 *prev2logE, int *pulses, opus_uint32 seed); |
| 109 | |
| 110 | opus_uint32 celt_lcg_rand(opus_uint32 seed); |
| 111 | |
Alexandre Lision | 8538238 | 2014-01-27 15:54:16 -0500 | [diff] [blame] | 112 | int hysteresis_decision(opus_val16 val, const opus_val16 *thresholds, const opus_val16 *hysteresis, int N, int prev); |
| 113 | |
Alexandre Lision | 744f742 | 2013-09-25 11:39:37 -0400 | [diff] [blame] | 114 | #endif /* BANDS_H */ |