Chris@69: /* Copyright (c) 2007-2008 CSIRO
Chris@69:    Copyright (c) 2007-2009 Xiph.Org Foundation
Chris@69:    Copyright (c) 2008 Gregory Maxwell
Chris@69:    Written by Jean-Marc Valin and Gregory Maxwell */
Chris@69: /**
Chris@69:   @file celt.h
Chris@69:   @brief Contains all the functions for encoding and decoding audio
Chris@69:  */
Chris@69: 
Chris@69: /*
Chris@69:    Redistribution and use in source and binary forms, with or without
Chris@69:    modification, are permitted provided that the following conditions
Chris@69:    are met:
Chris@69: 
Chris@69:    - Redistributions of source code must retain the above copyright
Chris@69:    notice, this list of conditions and the following disclaimer.
Chris@69: 
Chris@69:    - Redistributions in binary form must reproduce the above copyright
Chris@69:    notice, this list of conditions and the following disclaimer in the
Chris@69:    documentation and/or other materials provided with the distribution.
Chris@69: 
Chris@69:    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Chris@69:    ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Chris@69:    LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Chris@69:    A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
Chris@69:    OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
Chris@69:    EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
Chris@69:    PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
Chris@69:    PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
Chris@69:    LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
Chris@69:    NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
Chris@69:    SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Chris@69: */
Chris@69: 
Chris@69: #ifndef CELT_H
Chris@69: #define CELT_H
Chris@69: 
Chris@69: #include "opus_types.h"
Chris@69: #include "opus_defines.h"
Chris@69: #include "opus_custom.h"
Chris@69: #include "entenc.h"
Chris@69: #include "entdec.h"
Chris@69: #include "arch.h"
Chris@69: 
Chris@69: #ifdef __cplusplus
Chris@69: extern "C" {
Chris@69: #endif
Chris@69: 
Chris@69: #define CELTEncoder OpusCustomEncoder
Chris@69: #define CELTDecoder OpusCustomDecoder
Chris@69: #define CELTMode OpusCustomMode
Chris@69: 
Chris@69: #define LEAK_BANDS 19
Chris@69: 
Chris@69: typedef struct {
Chris@69:    int valid;
Chris@69:    float tonality;
Chris@69:    float tonality_slope;
Chris@69:    float noisiness;
Chris@69:    float activity;
Chris@69:    float music_prob;
Chris@69:    float music_prob_min;
Chris@69:    float music_prob_max;
Chris@69:    int   bandwidth;
Chris@69:    float activity_probability;
Chris@69:    float max_pitch_ratio;
Chris@69:    /* Store as Q6 char to save space. */
Chris@69:    unsigned char leak_boost[LEAK_BANDS];
Chris@69: } AnalysisInfo;
Chris@69: 
Chris@69: typedef struct {
Chris@69:    int signalType;
Chris@69:    int offset;
Chris@69: } SILKInfo;
Chris@69: 
Chris@69: #define __celt_check_mode_ptr_ptr(ptr) ((ptr) + ((ptr) - (const CELTMode**)(ptr)))
Chris@69: 
Chris@69: #define __celt_check_analysis_ptr(ptr) ((ptr) + ((ptr) - (const AnalysisInfo*)(ptr)))
Chris@69: 
Chris@69: #define __celt_check_silkinfo_ptr(ptr) ((ptr) + ((ptr) - (const SILKInfo*)(ptr)))
Chris@69: 
Chris@69: /* Encoder/decoder Requests */
Chris@69: 
Chris@69: 
Chris@69: #define CELT_SET_PREDICTION_REQUEST    10002
Chris@69: /** Controls the use of interframe prediction.
Chris@69:     0=Independent frames
Chris@69:     1=Short term interframe prediction allowed
Chris@69:     2=Long term prediction allowed
Chris@69:  */
Chris@69: #define CELT_SET_PREDICTION(x) CELT_SET_PREDICTION_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define CELT_SET_INPUT_CLIPPING_REQUEST    10004
Chris@69: #define CELT_SET_INPUT_CLIPPING(x) CELT_SET_INPUT_CLIPPING_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define CELT_GET_AND_CLEAR_ERROR_REQUEST   10007
Chris@69: #define CELT_GET_AND_CLEAR_ERROR(x) CELT_GET_AND_CLEAR_ERROR_REQUEST, __opus_check_int_ptr(x)
Chris@69: 
Chris@69: #define CELT_SET_CHANNELS_REQUEST    10008
Chris@69: #define CELT_SET_CHANNELS(x) CELT_SET_CHANNELS_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: 
Chris@69: /* Internal */
Chris@69: #define CELT_SET_START_BAND_REQUEST    10010
Chris@69: #define CELT_SET_START_BAND(x) CELT_SET_START_BAND_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define CELT_SET_END_BAND_REQUEST    10012
Chris@69: #define CELT_SET_END_BAND(x) CELT_SET_END_BAND_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define CELT_GET_MODE_REQUEST    10015
Chris@69: /** Get the CELTMode used by an encoder or decoder */
Chris@69: #define CELT_GET_MODE(x) CELT_GET_MODE_REQUEST, __celt_check_mode_ptr_ptr(x)
Chris@69: 
Chris@69: #define CELT_SET_SIGNALLING_REQUEST    10016
Chris@69: #define CELT_SET_SIGNALLING(x) CELT_SET_SIGNALLING_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define CELT_SET_TONALITY_REQUEST    10018
Chris@69: #define CELT_SET_TONALITY(x) CELT_SET_TONALITY_REQUEST, __opus_check_int(x)
Chris@69: #define CELT_SET_TONALITY_SLOPE_REQUEST    10020
Chris@69: #define CELT_SET_TONALITY_SLOPE(x) CELT_SET_TONALITY_SLOPE_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define CELT_SET_ANALYSIS_REQUEST    10022
Chris@69: #define CELT_SET_ANALYSIS(x) CELT_SET_ANALYSIS_REQUEST, __celt_check_analysis_ptr(x)
Chris@69: 
Chris@69: #define OPUS_SET_LFE_REQUEST    10024
Chris@69: #define OPUS_SET_LFE(x) OPUS_SET_LFE_REQUEST, __opus_check_int(x)
Chris@69: 
Chris@69: #define OPUS_SET_ENERGY_MASK_REQUEST    10026
Chris@69: #define OPUS_SET_ENERGY_MASK(x) OPUS_SET_ENERGY_MASK_REQUEST, __opus_check_val16_ptr(x)
Chris@69: 
Chris@69: #define CELT_SET_SILK_INFO_REQUEST    10028
Chris@69: #define CELT_SET_SILK_INFO(x) CELT_SET_SILK_INFO_REQUEST, __celt_check_silkinfo_ptr(x)
Chris@69: 
Chris@69: /* Encoder stuff */
Chris@69: 
Chris@69: int celt_encoder_get_size(int channels);
Chris@69: 
Chris@69: int celt_encode_with_ec(OpusCustomEncoder * OPUS_RESTRICT st, const opus_val16 * pcm, int frame_size, unsigned char *compressed, int nbCompressedBytes, ec_enc *enc);
Chris@69: 
Chris@69: int celt_encoder_init(CELTEncoder *st, opus_int32 sampling_rate, int channels,
Chris@69:                       int arch);
Chris@69: 
Chris@69: 
Chris@69: 
Chris@69: /* Decoder stuff */
Chris@69: 
Chris@69: int celt_decoder_get_size(int channels);
Chris@69: 
Chris@69: 
Chris@69: int celt_decoder_init(CELTDecoder *st, opus_int32 sampling_rate, int channels);
Chris@69: 
Chris@69: int celt_decode_with_ec(OpusCustomDecoder * OPUS_RESTRICT st, const unsigned char *data,
Chris@69:       int len, opus_val16 * OPUS_RESTRICT pcm, int frame_size, ec_dec *dec, int accum);
Chris@69: 
Chris@69: #define celt_encoder_ctl opus_custom_encoder_ctl
Chris@69: #define celt_decoder_ctl opus_custom_decoder_ctl
Chris@69: 
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69: #define OPUS_CUSTOM_NOSTATIC
Chris@69: #else
Chris@69: #define OPUS_CUSTOM_NOSTATIC static OPUS_INLINE
Chris@69: #endif
Chris@69: 
Chris@69: static const unsigned char trim_icdf[11] = {126, 124, 119, 109, 87, 41, 19, 9, 4, 2, 0};
Chris@69: /* Probs: NONE: 21.875%, LIGHT: 6.25%, NORMAL: 65.625%, AGGRESSIVE: 6.25% */
Chris@69: static const unsigned char spread_icdf[4] = {25, 23, 2, 0};
Chris@69: 
Chris@69: static const unsigned char tapset_icdf[3]={2,1,0};
Chris@69: 
Chris@69: #ifdef CUSTOM_MODES
Chris@69: static const unsigned char toOpusTable[20] = {
Chris@69:       0xE0, 0xE8, 0xF0, 0xF8,
Chris@69:       0xC0, 0xC8, 0xD0, 0xD8,
Chris@69:       0xA0, 0xA8, 0xB0, 0xB8,
Chris@69:       0x00, 0x00, 0x00, 0x00,
Chris@69:       0x80, 0x88, 0x90, 0x98,
Chris@69: };
Chris@69: 
Chris@69: static const unsigned char fromOpusTable[16] = {
Chris@69:       0x80, 0x88, 0x90, 0x98,
Chris@69:       0x40, 0x48, 0x50, 0x58,
Chris@69:       0x20, 0x28, 0x30, 0x38,
Chris@69:       0x00, 0x08, 0x10, 0x18
Chris@69: };
Chris@69: 
Chris@69: static OPUS_INLINE int toOpus(unsigned char c)
Chris@69: {
Chris@69:    int ret=0;
Chris@69:    if (c<0xA0)
Chris@69:       ret = toOpusTable[c>>3];
Chris@69:    if (ret == 0)
Chris@69:       return -1;
Chris@69:    else
Chris@69:       return ret|(c&0x7);
Chris@69: }
Chris@69: 
Chris@69: static OPUS_INLINE int fromOpus(unsigned char c)
Chris@69: {
Chris@69:    if (c<0x80)
Chris@69:       return -1;
Chris@69:    else
Chris@69:       return fromOpusTable[(c>>3)-16] | (c&0x7);
Chris@69: }
Chris@69: #endif /* CUSTOM_MODES */
Chris@69: 
Chris@69: #define COMBFILTER_MAXPERIOD 1024
Chris@69: #define COMBFILTER_MINPERIOD 15
Chris@69: 
Chris@69: extern const signed char tf_select_table[4][8];
Chris@69: 
Chris@69: #if defined(ENABLE_HARDENING) || defined(ENABLE_ASSERTIONS)
Chris@69: void validate_celt_decoder(CELTDecoder *st);
Chris@69: #define VALIDATE_CELT_DECODER(st) validate_celt_decoder(st)
Chris@69: #else
Chris@69: #define VALIDATE_CELT_DECODER(st)
Chris@69: #endif
Chris@69: 
Chris@69: int resampling_factor(opus_int32 rate);
Chris@69: 
Chris@69: void celt_preemphasis(const opus_val16 * OPUS_RESTRICT pcmp, celt_sig * OPUS_RESTRICT inp,
Chris@69:                         int N, int CC, int upsample, const opus_val16 *coef, celt_sig *mem, int clip);
Chris@69: 
Chris@69: void comb_filter(opus_val32 *y, opus_val32 *x, int T0, int T1, int N,
Chris@69:       opus_val16 g0, opus_val16 g1, int tapset0, int tapset1,
Chris@69:       const opus_val16 *window, int overlap, int arch);
Chris@69: 
Chris@69: #ifdef NON_STATIC_COMB_FILTER_CONST_C
Chris@69: void comb_filter_const_c(opus_val32 *y, opus_val32 *x, int T, int N,
Chris@69:                          opus_val16 g10, opus_val16 g11, opus_val16 g12);
Chris@69: #endif
Chris@69: 
Chris@69: #ifndef OVERRIDE_COMB_FILTER_CONST
Chris@69: # define comb_filter_const(y, x, T, N, g10, g11, g12, arch) \
Chris@69:     ((void)(arch),comb_filter_const_c(y, x, T, N, g10, g11, g12))
Chris@69: #endif
Chris@69: 
Chris@69: void init_caps(const CELTMode *m,int *cap,int LM,int C);
Chris@69: 
Chris@69: #ifdef RESYNTH
Chris@69: void deemphasis(celt_sig *in[], opus_val16 *pcm, int N, int C, int downsample, const opus_val16 *coef, celt_sig *mem);
Chris@69: void celt_synthesis(const CELTMode *mode, celt_norm *X, celt_sig * out_syn[],
Chris@69:       opus_val16 *oldBandE, int start, int effEnd, int C, int CC, int isTransient,
Chris@69:       int LM, int downsample, int silence);
Chris@69: #endif
Chris@69: 
Chris@69: #ifdef __cplusplus
Chris@69: }
Chris@69: #endif
Chris@69: 
Chris@69: #endif /* CELT_H */