Add support for intra-coding of the coarse energy.
[opus.git] / libcelt / bands.h
index 914ef96..ccaa7ee 100644 (file)
@@ -32,7 +32,7 @@
 #ifndef BANDS_H
 #define BANDS_H
 
 #ifndef BANDS_H
 #define BANDS_H
 
-
+#include "arch.h"
 #include "modes.h"
 #include "entenc.h"
 #include "entdec.h"
 #include "modes.h"
 #include "entenc.h"
 #include "entdec.h"
@@ -43,7 +43,9 @@
  * @param X Spectrum
  * @param bands Square root of the energy for each band (returned)
  */
  * @param X Spectrum
  * @param bands Square root of the energy for each band (returned)
  */
-void compute_band_energies(const CELTMode *m, float *X, float *bands);
+void compute_band_energies(const CELTMode *m, const celt_sig_t *X, celt_ener_t *bands);
+
+void compute_noise_energies(const CELTMode *m, const celt_sig_t *X, const celt_word16_t *tonality, celt_ener_t *bank);
 
 /** Normalise each band of X such that the energy in each band is 
     equal to 1
 
 /** Normalise each band of X such that the energy in each band is 
     equal to 1
@@ -51,16 +53,16 @@ void compute_band_energies(const CELTMode *m, float *X, float *bands);
  * @param X Spectrum (returned normalised)
  * @param bands Square root of the energy for each band
  */
  * @param X Spectrum (returned normalised)
  * @param bands Square root of the energy for each band
  */
-void normalise_bands(const CELTMode *m, float *X, float *bands);
+void normalise_bands(const CELTMode *m, const celt_sig_t * restrict freq, celt_norm_t * restrict X, const celt_ener_t *bands);
 
 
-void renormalise_bands(const CELTMode *m, float *X);
+void renormalise_bands(const CELTMode *m, celt_norm_t * restrict X);
 
 /** Denormalise each band of X to restore full amplitude
  * @param m Mode data 
  * @param X Spectrum (returned de-normalised)
  * @param bands Square root of the energy for each band
  */
 
 /** Denormalise each band of X to restore full amplitude
  * @param m Mode data 
  * @param X Spectrum (returned de-normalised)
  * @param bands Square root of the energy for each band
  */
-void denormalise_bands(const CELTMode *m, float *X, float *bands);
+void denormalise_bands(const CELTMode *m, const celt_norm_t * restrict X, celt_sig_t * restrict freq, const celt_ener_t *bands);
 
 /** Compute the pitch predictor gain for each pitch band
  * @param m Mode data 
 
 /** Compute the pitch predictor gain for each pitch band
  * @param m Mode data 
@@ -69,27 +71,31 @@ void denormalise_bands(const CELTMode *m, float *X, float *bands);
  * @param gains Gain computed for each pitch band (returned)
  * @param bank Square root of the energy for each band
  */
  * @param gains Gain computed for each pitch band (returned)
  * @param bank Square root of the energy for each band
  */
-void compute_pitch_gain(const CELTMode *m, float *X, float *P, float *gains, float *bank);
-
-void pitch_quant_bands(const CELTMode *m, float *X, float *P, float *gains);
+int compute_pitch_gain(const CELTMode *m, const celt_norm_t *X, const celt_norm_t *P, celt_pgain_t *gains);
 
 /** Quantisation/encoding of the residual spectrum
  * @param m Mode data 
  * @param X Residual (normalised)
  * @param P Pitch vector (normalised)
  * @param W Perceptual weighting
 
 /** Quantisation/encoding of the residual spectrum
  * @param m Mode data 
  * @param X Residual (normalised)
  * @param P Pitch vector (normalised)
  * @param W Perceptual weighting
+ * @param total_bits Total number of bits that can be used for the frame (including the ones already spent)
  * @param enc Entropy encoder
  */
  * @param enc Entropy encoder
  */
-void quant_bands(const CELTMode *m, float *X, float *P, float *W, struct alloc_data *alloc, int total_bits, ec_enc *enc);
+void quant_bands(const CELTMode *m, celt_norm_t * restrict X, celt_norm_t *P, celt_mask_t *W, int pitch_used, celt_pgain_t *pgains, const celt_ener_t *bandE, int *pulses, int time_domain, int fold, int total_bits, ec_enc *enc);
+
+void quant_bands_stereo(const CELTMode *m, celt_norm_t * restrict X, celt_norm_t *P, celt_mask_t *W, int pitch_used, celt_pgain_t *pgains, const celt_ener_t *bandE, int *pulses, int time_domain, int fold, int total_bits, ec_enc *enc);
 
 /** Decoding of the residual spectrum
  * @param m Mode data 
  * @param X Residual (normalised)
  * @param P Pitch vector (normalised)
 
 /** Decoding of the residual spectrum
  * @param m Mode data 
  * @param X Residual (normalised)
  * @param P Pitch vector (normalised)
+ * @param total_bits Total number of bits that can be used for the frame (including the ones already spent)
  * @param dec Entropy decoder
 */
  * @param dec Entropy decoder
 */
-void unquant_bands(const CELTMode *m, float *X, float *P, struct alloc_data *alloc, int total_bits, ec_dec *dec);
+void unquant_bands(const CELTMode *m, celt_norm_t * restrict X, celt_norm_t *P, int pitch_used, celt_pgain_t *pgains, const celt_ener_t *bandE, int *pulses, int time_domain, int fold, int total_bits, ec_dec *dec);
+
+void unquant_bands_stereo(const CELTMode *m, celt_norm_t * restrict X, celt_norm_t *P, int pitch_used, celt_pgain_t *pgains, const celt_ener_t *bandE, int *pulses, int time_domain, int fold, int total_bits, ec_dec *dec);
 
 
-void stereo_mix(const CELTMode *m, float *X, float *bank, int dir);
+void stereo_decision(const CELTMode *m, celt_norm_t * restrict X, int *stereo_mode, int len);
 
 #endif /* BANDS_H */
 
 #endif /* BANDS_H */